diff options
Diffstat (limited to 'src/inc/assemblyfilehash.h')
-rw-r--r-- | src/inc/assemblyfilehash.h | 92 |
1 files changed, 92 insertions, 0 deletions
diff --git a/src/inc/assemblyfilehash.h b/src/inc/assemblyfilehash.h new file mode 100644 index 0000000000..5f6fb23f68 --- /dev/null +++ b/src/inc/assemblyfilehash.h @@ -0,0 +1,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 |