blob: 5f6fb23f689e135fad8294aec8ee60236d711b87 (
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
90
91
92
|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
#ifndef __ASSEMBLYHASH_H__
#define __ASSEMBLYHASH_H__
#include "wincrypt.h"
#include "ex.h"
class AssemblyFileHash
{
protected:
BOOL m_bDataOwner;
PBYTE m_pbData;
DWORD m_cbData;
PBYTE m_pbHash;
DWORD m_cbHash;
HRESULT HashData(HCRYPTHASH);
HANDLE m_hFile;
BOOL m_NeedToReadData;
HRESULT ReadData();
public:
HRESULT SetFileName(LPCWSTR wszFileName);
HRESULT ReleaseFileHandle()
{
WRAPPER_NO_CONTRACT;
return SetFileHandle(INVALID_HANDLE_VALUE);
};
HRESULT SetFileHandle(HANDLE hFile)
{
LIMITED_METHOD_CONTRACT;
m_hFile=hFile;
return S_OK;
};
HRESULT SetData(PBYTE pbData, DWORD cbData) // Owned by owners context. Will not be deleted
{
CONTRACTL
{
NOTHROW;
GC_NOTRIGGER;
}
CONTRACTL_END;
m_pbData = pbData;
m_cbData = cbData;
m_bDataOwner = FALSE;
m_NeedToReadData=FALSE;
return S_OK;
}
PBYTE GetHash() { LIMITED_METHOD_CONTRACT; _ASSERTE(!m_NeedToReadData);return m_pbHash; }
DWORD GetHashSize() { LIMITED_METHOD_CONTRACT;_ASSERTE(!m_NeedToReadData); return m_cbHash; }
HRESULT CalculateHash(DWORD algid);
AssemblyFileHash()
: m_pbData( NULL ),
m_cbData( 0 ),
m_pbHash( NULL ),
m_cbHash( 0 ),
m_hFile(INVALID_HANDLE_VALUE),
m_NeedToReadData(TRUE)
{
CONTRACTL
{
CONSTRUCTOR_CHECK;
NOTHROW;
GC_NOTRIGGER;
}
CONTRACTL_END;
}
~AssemblyFileHash()
{
delete [] m_pbHash;
if (m_bDataOwner)
delete [] m_pbData;
if (m_hFile!=INVALID_HANDLE_VALUE)
CloseHandle(m_hFile);
m_hFile=INVALID_HANDLE_VALUE;
}
};
#endif
|