[RAPPS] improve ConInitStdStreams initialization logic (#3098)

This commit is contained in:
He Yang 2020-08-26 17:10:51 +08:00 committed by Mark Jansen
parent 7a11c65561
commit ce0110195b
No known key found for this signature in database
GPG key ID: B39240EE84BEAE8B
2 changed files with 10 additions and 6 deletions

View file

@ -40,6 +40,7 @@ BOOL HandleInstallCommand(LPWSTR szCommand, int argcLeft, LPWSTR * argvLeft)
{
if (argcLeft == 0)
{
ConInitStdStreams(); // Initialize the Console Standard Streams
ConResMsgPrintf(StdOut, NULL, IDS_CMD_NEED_PACKAGE_NAME, szCommand);
return FALSE;
}
@ -72,9 +73,11 @@ BOOL HandleSetupCommand(LPWSTR szCommand, int argcLeft, LPWSTR * argvLeft)
{
if (argcLeft != 1)
{
ConInitStdStreams(); // Initialize the Console Standard Streams
ConResMsgPrintf(StdOut, NULL, IDS_CMD_NEED_FILE_NAME, szCommand);
return FALSE;
}
FreeConsole();
ATL::CSimpleArray<ATL::CStringW> PkgNameList;
HINF InfHandle = SetupOpenInfFileW(argvLeft[0], NULL, INF_STYLE_WIN4, NULL);
@ -272,6 +275,8 @@ BOOL ParseCmdAndExecute(LPWSTR lpCmdLine, BOOL bIsFirstLaunch, int nCmdShow)
if (hMutex)
CloseHandle(hMutex);
return TRUE;
}
else if (MatchCmdOption(argv[1], CMD_KEY_INSTALL))
{
@ -281,7 +286,11 @@ BOOL ParseCmdAndExecute(LPWSTR lpCmdLine, BOOL bIsFirstLaunch, int nCmdShow)
{
return HandleSetupCommand(argv[1], argc - 2, argv + 2);
}
else if (MatchCmdOption(argv[1], CMD_KEY_FIND))
ConInitStdStreams(); // Initialize the Console Standard Streams
if (MatchCmdOption(argv[1], CMD_KEY_FIND))
{
return HandleFindCommand(argv[1], argc - 2, argv + 2);
}
@ -299,7 +308,4 @@ BOOL ParseCmdAndExecute(LPWSTR lpCmdLine, BOOL bIsFirstLaunch, int nCmdShow)
ConResPuts(StdOut, IDS_CMD_INVALID_OPTION);
return FALSE;
}
return TRUE;
}

View file

@ -145,8 +145,6 @@ VOID SaveSettings(HWND hwnd)
int wmain(int argc, wchar_t *argv[])
{
ConInitStdStreams(); // Initialize the Console Standard Streams
BOOL bIsFirstLaunch;
InitializeAtlModule(GetModuleHandle(NULL), TRUE);