fix version number (#173)

yskim1501 · Ryan Lai · commit ce988c96d88b · 2019-02-21T17:38:51.000-08:00
diff --git a/Tools/WinMLRunner/src/CommandLineArgs.cpp b/Tools/WinMLRunner/src/CommandLineArgs.cpp
@@ -212,9 +212,11 @@ CommandLineArgs::CommandLineArgs(const std::vector<std::wstring> &args)
         }
         else if (_wcsicmp(args[i].c_str(), L"-version") == 0)
         {
-            uint32_t versionInfoSize = GetFileVersionInfoSize(args[0].c_str(), 0);
+            TCHAR szExeFileName[MAX_PATH];
+            GetModuleFileName(NULL, szExeFileName, MAX_PATH);
+            uint32_t versionInfoSize = GetFileVersionInfoSize(szExeFileName, 0);
             wchar_t *pVersionData = new wchar_t[versionInfoSize / sizeof(wchar_t)];
-            GetFileVersionInfo(args[0].c_str(), 0, versionInfoSize, pVersionData);
+            GetFileVersionInfo(szExeFileName, 0, versionInfoSize, pVersionData);
 
             wchar_t *pOriginalFilename;
             uint32_t originalFilenameSize;

Original file line number	Diff line number	Diff line change
`@@ -212,9 +212,11 @@ CommandLineArgs::CommandLineArgs(const std::vector<std::wstring> &args)`
`212`	`212`	`}`
`213`	`213`	`else if (_wcsicmp(args[i].c_str(), L"-version") == 0)`
`214`	`214`	`{`
`215`		`- uint32_t versionInfoSize = GetFileVersionInfoSize(args[0].c_str(), 0);`
	`215`	`+ TCHAR szExeFileName[MAX_PATH];`
	`216`	`+ GetModuleFileName(NULL, szExeFileName, MAX_PATH);`
	`217`	`+ uint32_t versionInfoSize = GetFileVersionInfoSize(szExeFileName, 0);`
`216`	`218`	`wchar_t *pVersionData = new wchar_t[versionInfoSize / sizeof(wchar_t)];`
`217`		`- GetFileVersionInfo(args[0].c_str(), 0, versionInfoSize, pVersionData);`
	`219`	`+ GetFileVersionInfo(szExeFileName, 0, versionInfoSize, pVersionData);`
`218`	`220`
`219`	`221`	`wchar_t *pOriginalFilename;`
`220`	`222`	`uint32_t originalFilenameSize;`