diff options
Diffstat (limited to 'src/ilasm/main.cpp')
| -rw-r--r-- | src/ilasm/main.cpp | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/ilasm/main.cpp b/src/ilasm/main.cpp index 50131b2b01..811c65640e 100644 --- a/src/ilasm/main.cpp +++ b/src/ilasm/main.cpp @@ -95,7 +95,7 @@ WCHAR *pwzDeltaFiles[1024]; char szInputFilename[MAX_FILENAME_LENGTH*3]; WCHAR wzInputFilename[MAX_FILENAME_LENGTH]; WCHAR wzOutputFilename[MAX_FILENAME_LENGTH]; -WCHAR wzIncludePathBuffer[MAX_FILENAME_LENGTH]; + #ifdef _PREFAST_ #pragma warning(push) @@ -628,8 +628,12 @@ extern "C" int _cdecl wmain(int argc, __in WCHAR **argv) } if(wzIncludePath == NULL) { - if(0!=WszGetEnvironmentVariable(W("ILASM_INCLUDE"),wzIncludePathBuffer,MAX_FILENAME_LENGTH)) - wzIncludePath = wzIncludePathBuffer; + PathString wzIncludePathBuffer; + if (0 != WszGetEnvironmentVariable(W("ILASM_INCLUDE"), wzIncludePathBuffer)) + { + wzIncludePath = wzIncludePathBuffer.GetCopyOfUnicodeString(); + + } } //------------ Assembler initialization done. Now, to business ----------------------- if((pParser = new AsmParse(NULL, pAsm))) |