blob: 0b4055d655a1419076bf71c42afda74cbec248db (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
namespace WixToolset.Core.Cab
{
using System;
using System.Collections.Generic;
using WixToolset.Core.Native;
using Handle = System.Int32;
/// <summary>
/// Wrapper class around interop with wixcab.dll to enumerate files from a cabinet.
/// </summary>
public sealed class WixEnumerateCab : IDisposable
{
private bool disposed;
private List<CabinetFileInfo> fileInfoList;
private CabInterop.PFNNOTIFY pfnNotify;
/// <summary>
/// Creates a cabinet enumerator.
/// </summary>
public WixEnumerateCab()
{
this.pfnNotify = new CabInterop.PFNNOTIFY(this.Notify);
NativeMethods.EnumerateCabBegin();
}
/// <summary>
/// Destructor for cabinet enumeration.
/// </summary>
~WixEnumerateCab()
{
this.Dispose();
}
/// <summary>
/// Enumerates all files in a cabinet.
/// </summary>
/// <param name="cabinetFile">path to cabinet</param>
/// <returns>list of CabinetFileInfo</returns>
public List<CabinetFileInfo> Enumerate(string cabinetFile)
{
this.fileInfoList = new List<CabinetFileInfo>();
// the callback (this.Notify) will populate the list for each file in cabinet
NativeMethods.EnumerateCab(cabinetFile, this.pfnNotify);
return this.fileInfoList;
}
/// <summary>
/// Disposes the managed and unmanaged objects in this object.
/// </summary>
public void Dispose()
{
if (!this.disposed)
{
NativeMethods.EnumerateCabFinish();
GC.SuppressFinalize(this);
this.disposed = true;
}
}
/// <summary>
/// Delegate that's called for every file in cabinet.
/// </summary>
/// <param name="fdint">NOTIFICATIONTYPE</param>
/// <param name="pfdin">NOTIFICATION</param>
/// <returns>System.Int32</returns>
internal Handle Notify(CabInterop.NOTIFICATIONTYPE fdint, CabInterop.NOTIFICATION pfdin)
{
// This is FDI's way of notifying us of how many files total are in the cab, accurate even
// if the files are split into multiple folders - use it to allocate the precise size we need
if (CabInterop.NOTIFICATIONTYPE.ENUMERATE == fdint && 0 == this.fileInfoList.Count)
{
this.fileInfoList.Capacity = pfdin.Folder;
}
if (fdint == CabInterop.NOTIFICATIONTYPE.COPY_FILE)
{
CabinetFileInfo fileInfo = new CabinetFileInfo(pfdin.Psz1, pfdin.Date, pfdin.Time, pfdin.Cb);
this.fileInfoList.Add(fileInfo);
}
return 0; // tell cabinet api to skip this file.
}
}
}
|
