diff options
| author | Sean Hall <r.sean.hall@gmail.com> | 2020-08-29 21:29:28 -0500 |
|---|---|---|
| committer | Sean Hall <r.sean.hall@gmail.com> | 2020-10-24 20:03:38 -0500 |
| commit | d1d31466bb9f2e887a277807d60378afef9cc57d (patch) | |
| tree | 90b0185ce83f4ec5d986205aff2ce2487d3260de /src/dutil | |
| parent | 281ad838c5001f988aeea06a6f06ce2cc6c0991d (diff) | |
| download | wix-d1d31466bb9f2e887a277807d60378afef9cc57d.tar.gz wix-d1d31466bb9f2e887a277807d60378afef9cc57d.tar.bz2 wix-d1d31466bb9f2e887a277807d60378afef9cc57d.zip | |
WIXFEAT:6210 Parse and compare bundle versions kind of like SemVer.
Diffstat (limited to 'src/dutil')
| -rw-r--r-- | src/dutil/dutil.vcxproj | 2 | ||||
| -rw-r--r-- | src/dutil/dutil.vcxproj.filters | 6 | ||||
| -rw-r--r-- | src/dutil/inc/dutilsources.h | 1 | ||||
| -rw-r--r-- | src/dutil/inc/verutil.h | 93 | ||||
| -rw-r--r-- | src/dutil/precomp.h | 1 | ||||
| -rw-r--r-- | src/dutil/verutil.cpp | 631 |
6 files changed, 734 insertions, 0 deletions
diff --git a/src/dutil/dutil.vcxproj b/src/dutil/dutil.vcxproj index e9bbb98b..017f7a6f 100644 --- a/src/dutil/dutil.vcxproj +++ b/src/dutil/dutil.vcxproj | |||
| @@ -111,6 +111,7 @@ | |||
| 111 | <ClCompile Include="uncutil.cpp" /> | 111 | <ClCompile Include="uncutil.cpp" /> |
| 112 | <ClCompile Include="uriutil.cpp" /> | 112 | <ClCompile Include="uriutil.cpp" /> |
| 113 | <ClCompile Include="userutil.cpp" /> | 113 | <ClCompile Include="userutil.cpp" /> |
| 114 | <ClCompile Include="verutil.cpp" /> | ||
| 114 | <ClCompile Include="wiutil.cpp" /> | 115 | <ClCompile Include="wiutil.cpp" /> |
| 115 | <ClCompile Include="wuautil.cpp" /> | 116 | <ClCompile Include="wuautil.cpp" /> |
| 116 | <ClCompile Include="xmlutil.cpp" /> | 117 | <ClCompile Include="xmlutil.cpp" /> |
| @@ -167,6 +168,7 @@ | |||
| 167 | <ClInclude Include="inc\timeutil.h" /> | 168 | <ClInclude Include="inc\timeutil.h" /> |
| 168 | <ClInclude Include="inc\uriutil.h" /> | 169 | <ClInclude Include="inc\uriutil.h" /> |
| 169 | <ClInclude Include="inc\userutil.h" /> | 170 | <ClInclude Include="inc\userutil.h" /> |
| 171 | <ClInclude Include="inc\verutil.h" /> | ||
| 170 | <ClInclude Include="inc\wiutil.h" /> | 172 | <ClInclude Include="inc\wiutil.h" /> |
| 171 | <ClInclude Include="inc\wuautil.h" /> | 173 | <ClInclude Include="inc\wuautil.h" /> |
| 172 | <ClInclude Include="inc\xmlutil.h" /> | 174 | <ClInclude Include="inc\xmlutil.h" /> |
diff --git a/src/dutil/dutil.vcxproj.filters b/src/dutil/dutil.vcxproj.filters index 01dd6661..b93d166b 100644 --- a/src/dutil/dutil.vcxproj.filters +++ b/src/dutil/dutil.vcxproj.filters | |||
| @@ -159,6 +159,9 @@ | |||
| 159 | <ClCompile Include="userutil.cpp"> | 159 | <ClCompile Include="userutil.cpp"> |
| 160 | <Filter>Source Files</Filter> | 160 | <Filter>Source Files</Filter> |
| 161 | </ClCompile> | 161 | </ClCompile> |
| 162 | <ClCompile Include="verutil.cpp"> | ||
| 163 | <Filter>Source Files</Filter> | ||
| 164 | </ClCompile> | ||
| 162 | <ClCompile Include="wiutil.cpp"> | 165 | <ClCompile Include="wiutil.cpp"> |
| 163 | <Filter>Source Files</Filter> | 166 | <Filter>Source Files</Filter> |
| 164 | </ClCompile> | 167 | </ClCompile> |
| @@ -329,6 +332,9 @@ | |||
| 329 | <ClInclude Include="inc\userutil.h"> | 332 | <ClInclude Include="inc\userutil.h"> |
| 330 | <Filter>Header Files</Filter> | 333 | <Filter>Header Files</Filter> |
| 331 | </ClInclude> | 334 | </ClInclude> |
| 335 | <ClInclude Include="inc\verutil.h"> | ||
| 336 | <Filter>Header Files</Filter> | ||
| 337 | </ClInclude> | ||
| 332 | <ClInclude Include="inc\wiutil.h"> | 338 | <ClInclude Include="inc\wiutil.h"> |
| 333 | <Filter>Header Files</Filter> | 339 | <Filter>Header Files</Filter> |
| 334 | </ClInclude> | 340 | </ClInclude> |
diff --git a/src/dutil/inc/dutilsources.h b/src/dutil/inc/dutilsources.h index b03013ca..7d512cb3 100644 --- a/src/dutil/inc/dutilsources.h +++ b/src/dutil/inc/dutilsources.h | |||
| @@ -60,6 +60,7 @@ typedef enum DUTIL_SOURCE | |||
| 60 | DUTIL_SOURCE_WIUTIL, | 60 | DUTIL_SOURCE_WIUTIL, |
| 61 | DUTIL_SOURCE_WUAUTIL, | 61 | DUTIL_SOURCE_WUAUTIL, |
| 62 | DUTIL_SOURCE_XMLUTIL, | 62 | DUTIL_SOURCE_XMLUTIL, |
| 63 | DUTIL_SOURCE_VERUTIL, | ||
| 63 | 64 | ||
| 64 | DUTIL_SOURCE_EXTERNAL = 256, | 65 | DUTIL_SOURCE_EXTERNAL = 256, |
| 65 | } DUTIL_SOURCE; | 66 | } DUTIL_SOURCE; |
diff --git a/src/dutil/inc/verutil.h b/src/dutil/inc/verutil.h new file mode 100644 index 00000000..d3715049 --- /dev/null +++ b/src/dutil/inc/verutil.h | |||
| @@ -0,0 +1,93 @@ | |||
| 1 | #pragma once | ||
| 2 | // 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. | ||
| 3 | |||
| 4 | |||
| 5 | #ifdef __cplusplus | ||
| 6 | extern "C" { | ||
| 7 | #endif | ||
| 8 | |||
| 9 | #define ReleaseVerutilVersion(p) if (p) { VerFreeVersion(p); p = NULL; } | ||
| 10 | |||
| 11 | typedef struct _VERUTIL_VERSION_RELEASE_LABEL | ||
| 12 | { | ||
| 13 | BOOL fNumeric; | ||
| 14 | DWORD dwValue; | ||
| 15 | DWORD_PTR cchLabelOffset; | ||
| 16 | int cchLabel; | ||
| 17 | } VERUTIL_VERSION_RELEASE_LABEL; | ||
| 18 | |||
| 19 | typedef struct _VERUTIL_VERSION | ||
| 20 | { | ||
| 21 | LPWSTR sczVersion; | ||
| 22 | DWORD dwMajor; | ||
| 23 | DWORD dwMinor; | ||
| 24 | DWORD dwPatch; | ||
| 25 | DWORD dwRevision; | ||
| 26 | DWORD cReleaseLabels; | ||
| 27 | VERUTIL_VERSION_RELEASE_LABEL* rgReleaseLabels; | ||
| 28 | DWORD_PTR cchMetadataOffset; | ||
| 29 | BOOL fInvalid; | ||
| 30 | } VERUTIL_VERSION; | ||
| 31 | |||
| 32 | /******************************************************************* | ||
| 33 | VerCompareParsedVersions - compares the Verutil versions. | ||
| 34 | |||
| 35 | *******************************************************************/ | ||
| 36 | HRESULT DAPI VerCompareParsedVersions( | ||
| 37 | __in VERUTIL_VERSION* pVersion1, | ||
| 38 | __in VERUTIL_VERSION* pVersion2, | ||
| 39 | __out int* pnResult | ||
| 40 | ); | ||
| 41 | |||
| 42 | /******************************************************************* | ||
| 43 | VerCompareStringVersions - parses the strings with VerParseVersion and then | ||
| 44 | compares the Verutil versions with VerCompareParsedVersions. | ||
| 45 | |||
| 46 | *******************************************************************/ | ||
| 47 | HRESULT DAPI VerCompareStringVersions( | ||
| 48 | __in_z LPCWSTR wzVersion1, | ||
| 49 | __in_z LPCWSTR wzVersion2, | ||
| 50 | __in BOOL fStrict, | ||
| 51 | __out int* pnResult | ||
| 52 | ); | ||
| 53 | |||
| 54 | /******************************************************************** | ||
| 55 | VerCopyVersion - copies the given Verutil version. | ||
| 56 | |||
| 57 | *******************************************************************/ | ||
| 58 | HRESULT DAPI VerCopyVersion( | ||
| 59 | __in VERUTIL_VERSION* pSource, | ||
| 60 | __out VERUTIL_VERSION** ppVersion | ||
| 61 | ); | ||
| 62 | |||
| 63 | /******************************************************************** | ||
| 64 | VerFreeVersion - frees any memory associated with a Verutil version. | ||
| 65 | |||
| 66 | *******************************************************************/ | ||
| 67 | void DAPI VerFreeVersion( | ||
| 68 | __in VERUTIL_VERSION* pVersion | ||
| 69 | ); | ||
| 70 | |||
| 71 | /******************************************************************* | ||
| 72 | VerParseVersion - parses the string into a Verutil version. | ||
| 73 | |||
| 74 | *******************************************************************/ | ||
| 75 | HRESULT DAPI VerParseVersion( | ||
| 76 | __in_z LPCWSTR wzVersion, | ||
| 77 | __in DWORD cchVersion, | ||
| 78 | __in BOOL fStrict, | ||
| 79 | __out VERUTIL_VERSION** ppVersion | ||
| 80 | ); | ||
| 81 | |||
| 82 | /******************************************************************* | ||
| 83 | VerParseVersion - parses the QWORD into a Verutil version. | ||
| 84 | |||
| 85 | *******************************************************************/ | ||
| 86 | HRESULT DAPI VerVersionFromQword( | ||
| 87 | __in DWORD64 qwVersion, | ||
| 88 | __out VERUTIL_VERSION** ppVersion | ||
| 89 | ); | ||
| 90 | |||
| 91 | #ifdef __cplusplus | ||
| 92 | } | ||
| 93 | #endif | ||
diff --git a/src/dutil/precomp.h b/src/dutil/precomp.h index 7fdc83ae..be58860c 100644 --- a/src/dutil/precomp.h +++ b/src/dutil/precomp.h | |||
| @@ -89,6 +89,7 @@ | |||
| 89 | #include "uncutil.h" | 89 | #include "uncutil.h" |
| 90 | #include "uriutil.h" | 90 | #include "uriutil.h" |
| 91 | #include "userutil.h" | 91 | #include "userutil.h" |
| 92 | #include "verutil.h" | ||
| 92 | #include "wiutil.h" | 93 | #include "wiutil.h" |
| 93 | #include "wuautil.h" | 94 | #include "wuautil.h" |
| 94 | #include <comutil.h> // This header is needed for msxml2.h to compile correctly | 95 | #include <comutil.h> // This header is needed for msxml2.h to compile correctly |
diff --git a/src/dutil/verutil.cpp b/src/dutil/verutil.cpp new file mode 100644 index 00000000..f362f413 --- /dev/null +++ b/src/dutil/verutil.cpp | |||
| @@ -0,0 +1,631 @@ | |||
| 1 | // 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. | ||
| 2 | |||
| 3 | #include "precomp.h" | ||
| 4 | |||
| 5 | // Exit macros | ||
| 6 | #define VerExitOnLastError(x, s, ...) ExitOnLastErrorSource(DUTIL_SOURCE_VERUTIL, x, s, __VA_ARGS__) | ||
| 7 | #define VerExitOnLastErrorDebugTrace(x, s, ...) ExitOnLastErrorDebugTraceSource(DUTIL_SOURCE_VERUTIL, x, s, __VA_ARGS__) | ||
| 8 | #define VerExitWithLastError(x, s, ...) ExitWithLastErrorSource(DUTIL_SOURCE_VERUTIL, x, s, __VA_ARGS__) | ||
| 9 | #define VerExitOnFailure(x, s, ...) ExitOnFailureSource(DUTIL_SOURCE_VERUTIL, x, s, __VA_ARGS__) | ||
| 10 | #define VerExitOnRootFailure(x, s, ...) ExitOnRootFailureSource(DUTIL_SOURCE_VERUTIL, x, s, __VA_ARGS__) | ||
| 11 | #define VerExitOnFailureDebugTrace(x, s, ...) ExitOnFailureDebugTraceSource(DUTIL_SOURCE_VERUTIL, x, s, __VA_ARGS__) | ||
| 12 | #define VerExitOnNull(p, x, e, s, ...) ExitOnNullSource(DUTIL_SOURCE_VERUTIL, p, x, e, s, __VA_ARGS__) | ||
| 13 | #define VerExitOnNullWithLastError(p, x, s, ...) ExitOnNullWithLastErrorSource(DUTIL_SOURCE_VERUTIL, p, x, s, __VA_ARGS__) | ||
| 14 | #define VerExitOnNullDebugTrace(p, x, e, s, ...) ExitOnNullDebugTraceSource(DUTIL_SOURCE_VERUTIL, p, x, e, s, __VA_ARGS__) | ||
| 15 | #define VerExitOnInvalidHandleWithLastError(p, x, s, ...) ExitOnInvalidHandleWithLastErrorSource(DUTIL_SOURCE_VERUTIL, p, x, s, __VA_ARGS__) | ||
| 16 | #define VerExitOnWin32Error(e, x, s, ...) ExitOnWin32ErrorSource(DUTIL_SOURCE_VERUTIL, e, x, s, __VA_ARGS__) | ||
| 17 | |||
| 18 | // constants | ||
| 19 | const DWORD GROW_RELEASE_LABELS = 3; | ||
| 20 | |||
| 21 | // Forward declarations. | ||
| 22 | static int CompareDword( | ||
| 23 | __in const DWORD& dw1, | ||
| 24 | __in const DWORD& dw2 | ||
| 25 | ); | ||
| 26 | static HRESULT CompareReleaseLabel( | ||
| 27 | __in const VERUTIL_VERSION_RELEASE_LABEL* p1, | ||
| 28 | __in LPCWSTR wzVersion1, | ||
| 29 | __in const VERUTIL_VERSION_RELEASE_LABEL* p2, | ||
| 30 | __in LPCWSTR wzVersion2, | ||
| 31 | __out int* pnResult | ||
| 32 | ); | ||
| 33 | static HRESULT CompareVersionSubstring( | ||
| 34 | __in LPCWSTR wzString1, | ||
| 35 | __in int cchCount1, | ||
| 36 | __in LPCWSTR wzString2, | ||
| 37 | __in int cchCount2, | ||
| 38 | __out int* pnResult | ||
| 39 | ); | ||
| 40 | |||
| 41 | |||
| 42 | DAPI_(HRESULT) VerCompareParsedVersions( | ||
| 43 | __in VERUTIL_VERSION* pVersion1, | ||
| 44 | __in VERUTIL_VERSION* pVersion2, | ||
| 45 | __out int* pnResult | ||
| 46 | ) | ||
| 47 | { | ||
| 48 | HRESULT hr = S_OK; | ||
| 49 | int nResult = 0; | ||
| 50 | DWORD cMaxReleaseLabels = 0; | ||
| 51 | BOOL fCompareMetadata = FALSE; | ||
| 52 | |||
| 53 | if (!pVersion1 || !pVersion1->sczVersion || | ||
| 54 | !pVersion2 || !pVersion2->sczVersion) | ||
| 55 | { | ||
| 56 | ExitFunction1(hr = E_INVALIDARG); | ||
| 57 | } | ||
| 58 | |||
| 59 | if (pVersion1 == pVersion2) | ||
| 60 | { | ||
| 61 | ExitFunction1(nResult = 0); | ||
| 62 | } | ||
| 63 | |||
| 64 | nResult = CompareDword(pVersion1->dwMajor, pVersion2->dwMajor); | ||
| 65 | if (0 != nResult) | ||
| 66 | { | ||
| 67 | ExitFunction(); | ||
| 68 | } | ||
| 69 | |||
| 70 | nResult = CompareDword(pVersion1->dwMinor, pVersion2->dwMinor); | ||
| 71 | if (0 != nResult) | ||
| 72 | { | ||
| 73 | ExitFunction(); | ||
| 74 | } | ||
| 75 | |||
| 76 | nResult = CompareDword(pVersion1->dwPatch, pVersion2->dwPatch); | ||
| 77 | if (0 != nResult) | ||
| 78 | { | ||
| 79 | ExitFunction(); | ||
| 80 | } | ||
| 81 | |||
| 82 | nResult = CompareDword(pVersion1->dwRevision, pVersion2->dwRevision); | ||
| 83 | if (0 != nResult) | ||
| 84 | { | ||
| 85 | ExitFunction(); | ||
| 86 | } | ||
| 87 | |||
| 88 | if (pVersion1->fInvalid) | ||
| 89 | { | ||
| 90 | if (!pVersion2->fInvalid) | ||
| 91 | { | ||
| 92 | ExitFunction1(nResult = -1); | ||
| 93 | } | ||
| 94 | else | ||
| 95 | { | ||
| 96 | fCompareMetadata = TRUE; | ||
| 97 | } | ||
| 98 | } | ||
| 99 | else if (pVersion2->fInvalid) | ||
| 100 | { | ||
| 101 | ExitFunction1(nResult = 1); | ||
| 102 | } | ||
| 103 | |||
| 104 | if (pVersion1->cReleaseLabels) | ||
| 105 | { | ||
| 106 | if (pVersion2->cReleaseLabels) | ||
| 107 | { | ||
| 108 | cMaxReleaseLabels = max(pVersion1->cReleaseLabels, pVersion2->cReleaseLabels); | ||
| 109 | } | ||
| 110 | else | ||
| 111 | { | ||
| 112 | ExitFunction1(nResult = -1); | ||
| 113 | } | ||
| 114 | } | ||
| 115 | else if (pVersion2->cReleaseLabels) | ||
| 116 | { | ||
| 117 | ExitFunction1(nResult = 1); | ||
| 118 | } | ||
| 119 | |||
| 120 | if (cMaxReleaseLabels) | ||
| 121 | { | ||
| 122 | for (DWORD i = 0; i < cMaxReleaseLabels; ++i) | ||
| 123 | { | ||
| 124 | VERUTIL_VERSION_RELEASE_LABEL* pReleaseLabel1 = pVersion1->cReleaseLabels > i ? pVersion1->rgReleaseLabels + i : NULL; | ||
| 125 | VERUTIL_VERSION_RELEASE_LABEL* pReleaseLabel2 = pVersion2->cReleaseLabels > i ? pVersion2->rgReleaseLabels + i : NULL; | ||
| 126 | |||
| 127 | hr = CompareReleaseLabel(pReleaseLabel1, pVersion1->sczVersion, pReleaseLabel2, pVersion2->sczVersion, &nResult); | ||
| 128 | if (FAILED(hr) || 0 != nResult) | ||
| 129 | { | ||
| 130 | ExitFunction(); | ||
| 131 | } | ||
| 132 | } | ||
| 133 | } | ||
| 134 | |||
| 135 | if (fCompareMetadata) | ||
| 136 | { | ||
| 137 | hr = CompareVersionSubstring(pVersion1->sczVersion + pVersion1->cchMetadataOffset, -1, pVersion2->sczVersion + pVersion2->cchMetadataOffset, -1, &nResult); | ||
| 138 | } | ||
| 139 | |||
| 140 | LExit: | ||
| 141 | *pnResult = nResult; | ||
| 142 | return hr; | ||
| 143 | } | ||
| 144 | |||
| 145 | DAPI_(HRESULT) VerCompareStringVersions( | ||
| 146 | __in_z LPCWSTR wzVersion1, | ||
| 147 | __in_z LPCWSTR wzVersion2, | ||
| 148 | __in BOOL fStrict, | ||
| 149 | __out int* pnResult | ||
| 150 | ) | ||
| 151 | { | ||
| 152 | HRESULT hr = S_OK; | ||
| 153 | VERUTIL_VERSION* pVersion1 = NULL; | ||
| 154 | VERUTIL_VERSION* pVersion2 = NULL; | ||
| 155 | int nResult = 0; | ||
| 156 | |||
| 157 | hr = VerParseVersion(wzVersion1, 0, fStrict, &pVersion1); | ||
| 158 | VerExitOnFailure(hr, "Failed to parse Verutil version '%ls'", wzVersion1); | ||
| 159 | |||
| 160 | hr = VerParseVersion(wzVersion2, 0, fStrict, &pVersion2); | ||
| 161 | VerExitOnFailure(hr, "Failed to parse Verutil version '%ls'", wzVersion2); | ||
| 162 | |||
| 163 | hr = VerCompareParsedVersions(pVersion1, pVersion2, &nResult); | ||
| 164 | VerExitOnFailure(hr, "Failed to compare parsed Verutil versions '%ls' and '%ls'.", wzVersion1, wzVersion2); | ||
| 165 | |||
| 166 | LExit: | ||
| 167 | *pnResult = nResult; | ||
| 168 | |||
| 169 | ReleaseVerutilVersion(pVersion1); | ||
| 170 | ReleaseVerutilVersion(pVersion2); | ||
| 171 | |||
| 172 | return hr; | ||
| 173 | } | ||
| 174 | |||
| 175 | DAPI_(HRESULT) VerCopyVersion( | ||
| 176 | __in VERUTIL_VERSION* pSource, | ||
| 177 | __out VERUTIL_VERSION** ppVersion | ||
| 178 | ) | ||
| 179 | { | ||
| 180 | HRESULT hr = S_OK; | ||
| 181 | VERUTIL_VERSION* pCopy = NULL; | ||
| 182 | |||
| 183 | pCopy = reinterpret_cast<VERUTIL_VERSION*>(MemAlloc(sizeof(VERUTIL_VERSION), TRUE)); | ||
| 184 | VerExitOnNull(pCopy, hr, E_OUTOFMEMORY, "Failed to allocate memory for Verutil version copy."); | ||
| 185 | |||
| 186 | hr = StrAllocString(&pCopy->sczVersion, pSource->sczVersion, 0); | ||
| 187 | VerExitOnFailure(hr, "Failed to copy Verutil version string '%ls'.", pSource->sczVersion); | ||
| 188 | |||
| 189 | pCopy->dwMajor = pSource->dwMajor; | ||
| 190 | pCopy->dwMinor = pSource->dwMinor; | ||
| 191 | pCopy->dwPatch = pSource->dwPatch; | ||
| 192 | pCopy->dwRevision = pSource->dwRevision; | ||
| 193 | |||
| 194 | hr = MemEnsureArraySize(reinterpret_cast<LPVOID*>(&pCopy->rgReleaseLabels), 0, sizeof(VERUTIL_VERSION_RELEASE_LABEL), pSource->cReleaseLabels); | ||
| 195 | VerExitOnFailure(hr, "Failed to allocate memory for Verutil version release labels copies."); | ||
| 196 | |||
| 197 | pCopy->cReleaseLabels = pSource->cReleaseLabels; | ||
| 198 | |||
| 199 | for (DWORD i = 0; i < pCopy->cReleaseLabels; ++i) | ||
| 200 | { | ||
| 201 | VERUTIL_VERSION_RELEASE_LABEL* pSourceLabel = pSource->rgReleaseLabels + i; | ||
| 202 | VERUTIL_VERSION_RELEASE_LABEL* pCopyLabel = pCopy->rgReleaseLabels + i; | ||
| 203 | |||
| 204 | pCopyLabel->cchLabelOffset = pSourceLabel->cchLabelOffset; | ||
| 205 | pCopyLabel->cchLabel = pSourceLabel->cchLabel; | ||
| 206 | pCopyLabel->fNumeric = pSourceLabel->fNumeric; | ||
| 207 | pCopyLabel->dwValue = pSourceLabel->dwValue; | ||
| 208 | } | ||
| 209 | |||
| 210 | pCopy->cchMetadataOffset = pSource->cchMetadataOffset; | ||
| 211 | pCopy->fInvalid = pSource->fInvalid; | ||
| 212 | |||
| 213 | *ppVersion = pCopy; | ||
| 214 | pCopy = NULL; | ||
| 215 | |||
| 216 | LExit: | ||
| 217 | ReleaseVerutilVersion(pCopy); | ||
| 218 | |||
| 219 | return hr; | ||
| 220 | } | ||
| 221 | |||
| 222 | DAPI_(void) VerFreeVersion( | ||
| 223 | __in VERUTIL_VERSION* pVersion | ||
| 224 | ) | ||
| 225 | { | ||
| 226 | if (pVersion) | ||
| 227 | { | ||
| 228 | ReleaseStr(pVersion->sczVersion); | ||
| 229 | ReleaseMem(pVersion->rgReleaseLabels); | ||
| 230 | ReleaseMem(pVersion); | ||
| 231 | } | ||
| 232 | } | ||
| 233 | |||
| 234 | DAPI_(HRESULT) VerParseVersion( | ||
| 235 | __in_z LPCWSTR wzVersion, | ||
| 236 | __in DWORD cchVersion, | ||
| 237 | __in BOOL fStrict, | ||
| 238 | __out VERUTIL_VERSION** ppVersion | ||
| 239 | ) | ||
| 240 | { | ||
| 241 | HRESULT hr = S_OK; | ||
| 242 | VERUTIL_VERSION* pVersion = NULL; | ||
| 243 | LPCWSTR wzEnd = NULL; | ||
| 244 | LPCWSTR wzPartBegin = NULL; | ||
| 245 | LPCWSTR wzPartEnd = NULL; | ||
| 246 | BOOL fInvalid = FALSE; | ||
| 247 | BOOL fLastPart = FALSE; | ||
| 248 | BOOL fTrailingDot = FALSE; | ||
| 249 | BOOL fParsedVersionNumber = FALSE; | ||
| 250 | BOOL fExpectedReleaseLabels = FALSE; | ||
| 251 | DWORD iPart = 0; | ||
| 252 | |||
| 253 | if (!wzVersion || !ppVersion) | ||
| 254 | { | ||
| 255 | ExitFunction1(hr = E_INVALIDARG); | ||
| 256 | } | ||
| 257 | |||
| 258 | // Get string length if not provided. | ||
| 259 | if (0 == cchVersion) | ||
| 260 | { | ||
| 261 | cchVersion = lstrlenW(wzVersion); | ||
| 262 | } | ||
| 263 | |||
| 264 | if (L'v' == *wzVersion || L'V' == *wzVersion) | ||
| 265 | { | ||
| 266 | ++wzVersion; | ||
| 267 | --cchVersion; | ||
| 268 | } | ||
| 269 | |||
| 270 | pVersion = reinterpret_cast<VERUTIL_VERSION*>(MemAlloc(sizeof(VERUTIL_VERSION), TRUE)); | ||
| 271 | VerExitOnNull(pVersion, hr, E_OUTOFMEMORY, "Failed to allocate memory for Verutil version '%ls'.", wzVersion); | ||
| 272 | |||
| 273 | hr = StrAllocString(&pVersion->sczVersion, wzVersion, cchVersion); | ||
| 274 | VerExitOnFailure(hr, "Failed to copy Verutil version string '%ls'.", wzVersion); | ||
| 275 | |||
| 276 | wzVersion = wzPartBegin = wzPartEnd = pVersion->sczVersion; | ||
| 277 | |||
| 278 | // Save end pointer. | ||
| 279 | wzEnd = wzVersion + cchVersion; | ||
| 280 | |||
| 281 | // Parse version number | ||
| 282 | while (wzPartBegin < wzEnd) | ||
| 283 | { | ||
| 284 | fTrailingDot = FALSE; | ||
| 285 | |||
| 286 | // Find end of part. | ||
| 287 | for (;;) | ||
| 288 | { | ||
| 289 | if (wzPartEnd >= wzEnd) | ||
| 290 | { | ||
| 291 | fLastPart = TRUE; | ||
| 292 | break; | ||
| 293 | } | ||
| 294 | |||
| 295 | switch (*wzPartEnd) | ||
| 296 | { | ||
| 297 | case L'0': | ||
| 298 | case L'1': | ||
| 299 | case L'2': | ||
| 300 | case L'3': | ||
| 301 | case L'4': | ||
| 302 | case L'5': | ||
| 303 | case L'6': | ||
| 304 | case L'7': | ||
| 305 | case L'8': | ||
| 306 | case L'9': | ||
| 307 | ++wzPartEnd; | ||
| 308 | continue; | ||
| 309 | case L'.': | ||
| 310 | fTrailingDot = TRUE; | ||
| 311 | break; | ||
| 312 | case L'-': | ||
| 313 | case L'+': | ||
| 314 | fLastPart = TRUE; | ||
| 315 | break; | ||
| 316 | default: | ||
| 317 | fInvalid = TRUE; | ||
| 318 | break; | ||
| 319 | } | ||
| 320 | |||
| 321 | break; | ||
| 322 | } | ||
| 323 | |||
| 324 | if (wzPartBegin == wzPartEnd) | ||
| 325 | { | ||
| 326 | fInvalid = TRUE; | ||
| 327 | } | ||
| 328 | |||
| 329 | if (fInvalid) | ||
| 330 | { | ||
| 331 | break; | ||
| 332 | } | ||
| 333 | |||
| 334 | DWORD cchPart = 0; | ||
| 335 | hr = ::PtrdiffTToDWord(wzPartEnd - wzPartBegin, &cchPart); | ||
| 336 | if (FAILED(hr)) | ||
| 337 | { | ||
| 338 | fInvalid = TRUE; | ||
| 339 | break; | ||
| 340 | } | ||
| 341 | |||
| 342 | // Parse version part. | ||
| 343 | UINT uPart = 0; | ||
| 344 | hr = StrStringToUInt32(wzPartBegin, cchPart, &uPart); | ||
| 345 | if (FAILED(hr)) | ||
| 346 | { | ||
| 347 | fInvalid = TRUE; | ||
| 348 | break; | ||
| 349 | } | ||
| 350 | |||
| 351 | switch (iPart) | ||
| 352 | { | ||
| 353 | case 0: | ||
| 354 | pVersion->dwMajor = uPart; | ||
| 355 | break; | ||
| 356 | case 1: | ||
| 357 | pVersion->dwMinor = uPart; | ||
| 358 | break; | ||
| 359 | case 2: | ||
| 360 | pVersion->dwPatch = uPart; | ||
| 361 | break; | ||
| 362 | case 3: | ||
| 363 | pVersion->dwRevision = uPart; | ||
| 364 | break; | ||
| 365 | } | ||
| 366 | |||
| 367 | if (fTrailingDot) | ||
| 368 | { | ||
| 369 | ++wzPartEnd; | ||
| 370 | } | ||
| 371 | wzPartBegin = wzPartEnd; | ||
| 372 | ++iPart; | ||
| 373 | |||
| 374 | if (4 <= iPart || fLastPart) | ||
| 375 | { | ||
| 376 | fParsedVersionNumber = TRUE; | ||
| 377 | break; | ||
| 378 | } | ||
| 379 | } | ||
| 380 | |||
| 381 | fInvalid |= !fParsedVersionNumber || fTrailingDot; | ||
| 382 | |||
| 383 | if (!fInvalid && wzPartBegin < wzEnd && *wzPartBegin == L'-') | ||
| 384 | { | ||
| 385 | wzPartBegin = wzPartEnd = wzPartBegin + 1; | ||
| 386 | fExpectedReleaseLabels = TRUE; | ||
| 387 | fLastPart = FALSE; | ||
| 388 | } | ||
| 389 | |||
| 390 | while (fExpectedReleaseLabels && wzPartBegin < wzEnd) | ||
| 391 | { | ||
| 392 | fTrailingDot = FALSE; | ||
| 393 | |||
| 394 | // Find end of part. | ||
| 395 | for (;;) | ||
| 396 | { | ||
| 397 | if (wzPartEnd >= wzEnd) | ||
| 398 | { | ||
| 399 | fLastPart = TRUE; | ||
| 400 | break; | ||
| 401 | } | ||
| 402 | |||
| 403 | if (*wzPartEnd >= L'0' && *wzPartEnd <= L'9' || | ||
| 404 | *wzPartEnd >= L'A' && *wzPartEnd <= L'Z' || | ||
| 405 | *wzPartEnd >= L'a' && *wzPartEnd <= L'z' || | ||
| 406 | *wzPartEnd == L'-') | ||
| 407 | { | ||
| 408 | ++wzPartEnd; | ||
| 409 | continue; | ||
| 410 | } | ||
| 411 | else if (*wzPartEnd == L'+') | ||
| 412 | { | ||
| 413 | fLastPart = TRUE; | ||
| 414 | } | ||
| 415 | else if (*wzPartEnd == L'.') | ||
| 416 | { | ||
| 417 | fTrailingDot = TRUE; | ||
| 418 | } | ||
| 419 | else | ||
| 420 | { | ||
| 421 | fInvalid = TRUE; | ||
| 422 | } | ||
| 423 | |||
| 424 | break; | ||
| 425 | } | ||
| 426 | |||
| 427 | if (wzPartBegin == wzPartEnd) | ||
| 428 | { | ||
| 429 | fInvalid = TRUE; | ||
| 430 | } | ||
| 431 | |||
| 432 | if (fInvalid) | ||
| 433 | { | ||
| 434 | break; | ||
| 435 | } | ||
| 436 | |||
| 437 | int cchLabel = 0; | ||
| 438 | hr = ::PtrdiffTToInt32(wzPartEnd - wzPartBegin, &cchLabel); | ||
| 439 | if (FAILED(hr) || 0 > cchLabel) | ||
| 440 | { | ||
| 441 | fInvalid = TRUE; | ||
| 442 | break; | ||
| 443 | } | ||
| 444 | |||
| 445 | hr = MemReAllocArray(reinterpret_cast<LPVOID*>(&pVersion->rgReleaseLabels), pVersion->cReleaseLabels, sizeof(VERUTIL_VERSION_RELEASE_LABEL), GROW_RELEASE_LABELS - (pVersion->cReleaseLabels % GROW_RELEASE_LABELS)); | ||
| 446 | VerExitOnFailure(hr, "Failed to allocate memory for Verutil version release labels '%ls'", wzVersion); | ||
| 447 | |||
| 448 | VERUTIL_VERSION_RELEASE_LABEL* pReleaseLabel = pVersion->rgReleaseLabels + pVersion->cReleaseLabels; | ||
| 449 | ++pVersion->cReleaseLabels; | ||
| 450 | |||
| 451 | // Try to parse as number. | ||
| 452 | UINT uLabel = 0; | ||
| 453 | hr = StrStringToUInt32(wzPartBegin, cchLabel, &uLabel); | ||
| 454 | if (SUCCEEDED(hr)) | ||
| 455 | { | ||
| 456 | pReleaseLabel->fNumeric = TRUE; | ||
| 457 | pReleaseLabel->dwValue = uLabel; | ||
| 458 | } | ||
| 459 | |||
| 460 | pReleaseLabel->cchLabelOffset = wzPartBegin - pVersion->sczVersion; | ||
| 461 | pReleaseLabel->cchLabel = cchLabel; | ||
| 462 | |||
| 463 | if (fTrailingDot) | ||
| 464 | { | ||
| 465 | ++wzPartEnd; | ||
| 466 | } | ||
| 467 | wzPartBegin = wzPartEnd; | ||
| 468 | |||
| 469 | if (fLastPart) | ||
| 470 | { | ||
| 471 | break; | ||
| 472 | } | ||
| 473 | } | ||
| 474 | |||
| 475 | fInvalid |= fExpectedReleaseLabels && (!pVersion->cReleaseLabels || fTrailingDot); | ||
| 476 | |||
| 477 | if (!fInvalid && wzPartBegin < wzEnd) | ||
| 478 | { | ||
| 479 | if (*wzPartBegin == L'+') | ||
| 480 | { | ||
| 481 | wzPartBegin = wzPartEnd = wzPartBegin + 1; | ||
| 482 | } | ||
| 483 | else | ||
| 484 | { | ||
| 485 | fInvalid = TRUE; | ||
| 486 | } | ||
| 487 | } | ||
| 488 | |||
| 489 | if (fInvalid && fStrict) | ||
| 490 | { | ||
| 491 | ExitFunction1(hr = E_INVALIDARG); | ||
| 492 | } | ||
| 493 | |||
| 494 | pVersion->cchMetadataOffset = min(wzPartBegin, wzEnd) - pVersion->sczVersion; | ||
| 495 | pVersion->fInvalid = fInvalid; | ||
| 496 | |||
| 497 | *ppVersion = pVersion; | ||
| 498 | pVersion = NULL; | ||
| 499 | hr = S_OK; | ||
| 500 | |||
| 501 | LExit: | ||
| 502 | ReleaseVerutilVersion(pVersion); | ||
| 503 | |||
| 504 | return hr; | ||
| 505 | } | ||
| 506 | |||
| 507 | DAPI_(HRESULT) VerVersionFromQword( | ||
| 508 | __in DWORD64 qwVersion, | ||
| 509 | __out VERUTIL_VERSION** ppVersion | ||
| 510 | ) | ||
| 511 | { | ||
| 512 | HRESULT hr = S_OK; | ||
| 513 | VERUTIL_VERSION* pVersion = NULL; | ||
| 514 | |||
| 515 | pVersion = reinterpret_cast<VERUTIL_VERSION*>(MemAlloc(sizeof(VERUTIL_VERSION), TRUE)); | ||
| 516 | VerExitOnNull(pVersion, hr, E_OUTOFMEMORY, "Failed to allocate memory for Verutil version from QWORD."); | ||
| 517 | |||
| 518 | pVersion->dwMajor = (WORD)(qwVersion >> 48 & 0xffff); | ||
| 519 | pVersion->dwMinor = (WORD)(qwVersion >> 32 & 0xffff); | ||
| 520 | pVersion->dwPatch = (WORD)(qwVersion >> 16 & 0xffff); | ||
| 521 | pVersion->dwRevision = (WORD)(qwVersion & 0xffff); | ||
| 522 | |||
| 523 | hr = StrAllocFormatted(&pVersion->sczVersion, L"%lu.%lu.%lu.%lu", pVersion->dwMajor, pVersion->dwMinor, pVersion->dwPatch, pVersion->dwRevision); | ||
| 524 | ExitOnFailure(hr, "Failed to allocate and format the version string."); | ||
| 525 | |||
| 526 | pVersion->cchMetadataOffset = lstrlenW(pVersion->sczVersion); | ||
| 527 | |||
| 528 | *ppVersion = pVersion; | ||
| 529 | pVersion = NULL; | ||
| 530 | |||
| 531 | LExit: | ||
| 532 | ReleaseVerutilVersion(pVersion); | ||
| 533 | |||
| 534 | return hr; | ||
| 535 | } | ||
| 536 | |||
| 537 | |||
| 538 | static int CompareDword( | ||
| 539 | __in const DWORD& dw1, | ||
| 540 | __in const DWORD& dw2 | ||
| 541 | ) | ||
| 542 | { | ||
| 543 | int nResult = 0; | ||
| 544 | |||
| 545 | if (dw1 > dw2) | ||
| 546 | { | ||
| 547 | nResult = 1; | ||
| 548 | } | ||
| 549 | else if (dw1 < dw2) | ||
| 550 | { | ||
| 551 | nResult = -1; | ||
| 552 | } | ||
| 553 | |||
| 554 | return nResult; | ||
| 555 | } | ||
| 556 | |||
| 557 | static HRESULT CompareReleaseLabel( | ||
| 558 | __in const VERUTIL_VERSION_RELEASE_LABEL* p1, | ||
| 559 | __in LPCWSTR wzVersion1, | ||
| 560 | __in const VERUTIL_VERSION_RELEASE_LABEL* p2, | ||
| 561 | __in LPCWSTR wzVersion2, | ||
| 562 | __out int* pnResult | ||
| 563 | ) | ||
| 564 | { | ||
| 565 | HRESULT hr = S_OK; | ||
| 566 | int nResult = 0; | ||
| 567 | |||
| 568 | if (p1 == p2) | ||
| 569 | { | ||
| 570 | ExitFunction(); | ||
| 571 | } | ||
| 572 | else if (p1 && !p2) | ||
| 573 | { | ||
| 574 | ExitFunction1(nResult = 1); | ||
| 575 | } | ||
| 576 | else if (!p1 && p2) | ||
| 577 | { | ||
| 578 | ExitFunction1(nResult = -1); | ||
| 579 | } | ||
| 580 | |||
| 581 | if (p1->fNumeric) | ||
| 582 | { | ||
| 583 | if (p2->fNumeric) | ||
| 584 | { | ||
| 585 | nResult = CompareDword(p1->dwValue, p2->dwValue); | ||
| 586 | } | ||
| 587 | else | ||
| 588 | { | ||
| 589 | nResult = -1; | ||
| 590 | } | ||
| 591 | } | ||
| 592 | else | ||
| 593 | { | ||
| 594 | if (p2->fNumeric) | ||
| 595 | { | ||
| 596 | nResult = 1; | ||
| 597 | } | ||
| 598 | else | ||
| 599 | { | ||
| 600 | hr = CompareVersionSubstring(wzVersion1 + p1->cchLabelOffset, p1->cchLabel, wzVersion2 + p2->cchLabelOffset, p2->cchLabel, &nResult); | ||
| 601 | } | ||
| 602 | } | ||
| 603 | |||
| 604 | LExit: | ||
| 605 | *pnResult = nResult; | ||
| 606 | |||
| 607 | return hr; | ||
| 608 | } | ||
| 609 | |||
| 610 | static HRESULT CompareVersionSubstring( | ||
| 611 | __in LPCWSTR wzString1, | ||
| 612 | __in int cchCount1, | ||
| 613 | __in LPCWSTR wzString2, | ||
| 614 | __in int cchCount2, | ||
| 615 | __out int* pnResult | ||
| 616 | ) | ||
| 617 | { | ||
| 618 | HRESULT hr = S_OK; | ||
| 619 | int nResult = 0; | ||
| 620 | |||
| 621 | nResult = ::CompareStringW(LOCALE_INVARIANT, NORM_IGNORECASE, wzString1, cchCount1, wzString2, cchCount2); | ||
| 622 | if (!nResult) | ||
| 623 | { | ||
| 624 | VerExitOnLastError(hr, "Failed to compare version substrings"); | ||
| 625 | } | ||
| 626 | |||
| 627 | LExit: | ||
| 628 | *pnResult = nResult - 2; | ||
| 629 | |||
| 630 | return hr; | ||
| 631 | } | ||
