Запуск приложения из-под консоли в отдельном потоке с записью времени запуска и закрытия в консоль

//
// Waiting for the process to complete.
//
// Copyright (c) Карбид
// (_:_)
 
#include <windows.h>
#include <tchar.h>
#include <stdio.h>
 
#define TM_ENTRYTHREAD_DESTROY (WM_APP + 1)
 
DWORD WINAPI EntryThread(LPVOID param)
{
    TCHAR* Filename = (TCHAR*)param;
 
    STARTUPINFO si;
    PROCESS_INFORMATION pi;
    memset(&si, 0, sizeof(STARTUPINFO));
    si.cb           = sizeof(STARTUPINFO);
    si.wShowWindow  = SW_SHOWNORMAL;
    si.dwFlags      = STARTF_FORCEOFFFEEDBACK;
    if( !CreateProcess(NULL,
        Filename, NULL, NULL, 0, 0, NULL, NULL, &si, &pi) ) {
        HINSTANCE ShellExError = ShellExecute(
            NULL, TEXT("open"), Filename, NULL, NULL, si.wShowWindow);
        if( INT_PTR(ShellExError)<=32 ) {
            wprintf(TEXT("Could not spawn process '%s'\n"), Filename);
            return EXIT_FAILURE;
        }
    } else {
        wprintf(TEXT("Process spawn '%s'\n\
                     Wait for the end process...\n"), Filename);
    }
 
 
    errno_t Result;
    FILE* FileHandle = NULL;
    Result = fopen_s(&FileHandle, "c:\\Mylog.txt", "at");
    if( Result==0 ) {
        DWORD Offset = 0;
        fseek(FileHandle, Offset, SEEK_SET);
        SYSTEMTIME st = {0};
        GetLocalTime(&st);
        fwprintf(FileHandle,
            TEXT("START %.4lu.%.2lu.%.2lu-%.2lu:%.2lu:%.2lu %s\n"),
            st.wYear, st.wMonth, st.wDay, st.wHour, st.wMinute, st.wSecond,
            Filename);
    }
 
    MSG msg;
    BOOL bDone = TRUE;
    DWORD dwStatus = 0;
    DWORD dwExitCode = 0;
 
    // Cycle waiting for the end of the process.
 
    while( bDone ) {
        
        Sleep(20);  // Give up time.
 
        dwStatus = MsgWaitForMultipleObjects(
            1, &pi.hProcess, FALSE, INFINITE, QS_ALLINPUT);
        if( dwStatus==WAIT_OBJECT_0 ) {
            bDone = FALSE;
        }
 
        while( PeekMessage(&msg, NULL, 0, 0, PM_REMOVE) ) {
            DispatchMessage(&msg);
        }
    }
 
    if ( !bDone )
    {
        switch( WaitForSingleObject(pi.hProcess, 500 /* msec. */) )
        {
        case WAIT_OBJECT_0: /* Process terminated. */
            if( GetExitCodeProcess(pi.hProcess, &dwExitCode) )
                printf("Exit code %d\n", dwExitCode);
            break;
        case STILL_ACTIVE:
            dwExitCode = STILL_ACTIVE;
            break;
        case WAIT_FAILED:
            dwExitCode = WAIT_FAILED;
            break;
        case WAIT_TIMEOUT:
            dwExitCode = WAIT_TIMEOUT;
            break;
        default:
            printf("Fall through unexpected WaitForSingleObject()\n");
            break;
        }
    }
    else
    {
        // Just to suppress the warnings about uninitialized variable.
        dwExitCode = 0;
    }
 
    if( FileHandle )
    {
        SYSTEMTIME st = {0};
        GetLocalTime(&st);
        fwprintf(FileHandle,
            TEXT("STOP %.4lu.%.2lu.%.2lu-%.2lu:%.2lu:%.2lu %s\nExit code %d\n"),
            st.wYear, st.wMonth, st.wDay, st.wHour, st.wMinute, st.wSecond,
            Filename, dwExitCode);
        Result = fclose(FileHandle);
        FileHandle = NULL;
    }
 
    return 0;
}
 
//
// Entry Point.
//
int _tmain()
{
    TCHAR Filename[MAX_PATH];
    TCHAR Filter[] = TEXT("Programs (*.exe)\0*.exe\0 All files\0*.*\0" );
    OPENFILENAME ofn;
    memset(&ofn, 0, sizeof(OPENFILENAME));
    TCHAR TempFilename[MAX_PATH];
    TempFilename[0]     = 0;
    ofn.lpstrFile       = TempFilename;
    ofn.nMaxFile        = MAX_PATH;
    ofn.lpstrFilter     = Filter;
    ofn.lpstrTitle      = TEXT("Please select an executable file" );
    ofn.Flags           = OFN_HIDEREADONLY | OFN_PATHMUSTEXIST;
    ofn.lStructSize     = sizeof(OPENFILENAME);
    ofn.hwndOwner       = GetActiveWindow();
    ofn.lpstrInitialDir = TEXT("");
    if( GetOpenFileName(&ofn)==0 )
    {
        wprintf(TEXT("The file is not selected.\n"));
        return EXIT_FAILURE;
    }
    else
    {
        lstrcpy(Filename, ofn.lpstrFile);
    }
 
    DWORD dwTID = 0;
    HANDLE HandleThread = NULL;
    HandleThread = CreateThread(NULL, 0, EntryThread, (LPVOID)Filename, 0, &dwTID);
    if( HandleThread==NULL )
    {
        wprintf(TEXT("Could not create thread.\n"));
        return EXIT_FAILURE;
    }
 
    MSG msg;
    BOOL bDone = FALSE;
    
    while( !bDone ) {
        
        Sleep(20);  // Give up time.
        
        if( WaitForSingleObject(HandleThread, 0)!=WAIT_TIMEOUT ) {
            bDone = TRUE;
        }
        
        if( PeekMessage(&msg, (HWND)-1, 0, 0, PM_REMOVE) ) {
            if( msg.message==TM_ENTRYTHREAD_DESTROY ) {
                bDone = TRUE;
            }
        }
    }
 
    if( CloseHandle(HandleThread)==TRUE )
        HandleThread = NULL;
 
    system("pause");
 
    return EXIT_SUCCESS;
}

// Приложение, читает параметры командной строки, окно скрыто.
// Остановить программу принудительно, нажать Ctrl + Alt + F12.
//
// [+]Опции проекта[+] Настроить:
// [Configuration] General [Character set] "Use Unicode Character Set"
//
 
#include <windows.h>
 
TCHAR g_Wndname[] = TEXT("WAITING_EXIT_PROCESS");   // Класс,
TCHAR g_Appname[] = TEXT("Waiting Exit Process");   // заголовок Окна.
 
// Идентификатор горячей клавиши.
#define ID_HOTKEY 0x0ABCD2016
 
#define MY_THREAD_DONE      (WM_APP + 1)
#define MY_THREAD_DESTROY   (WM_APP + 2)
 
typedef struct {
    HWND Parent;
    TCHAR Cmd[MAX_PATH];
    TCHAR LogFilename[MAX_PATH];
} THREAD_ARGS;
 
TCHAR g_Cmd[MAX_PATH];
TCHAR g_LogFilename[MAX_PATH];
 
const TCHAR* StampDateTime()
{
    static TCHAR Result[520];
    TCHAR Date[260] = TEXT("");
    TCHAR Time[260] = TEXT("");
    const TCHAR* TimeFormat = TEXT("hh':'mm':'ss");
    const TCHAR* DateFormat = TEXT("dddd dd-MM-yyyy");
    GetDateFormat(LOCALE_SYSTEM_DEFAULT, 0, NULL, DateFormat, Date, 260);
    GetTimeFormat(LOCALE_SYSTEM_DEFAULT, TIME_FORCE24HOURFORMAT,
        NULL, TimeFormat, Time, 260);
    wsprintf(Result, TEXT("%.100s %.100s"), Date, Time);
    return Result;
}
 
char* TCHAR_To_ANSI(const TCHAR* Str)
{
    HGLOBAL GlobalMem;
    int Count = WideCharToMultiByte(CP_ACP, 0, Str, -1, NULL, 0, NULL, NULL);
    GlobalMem = GlobalAlloc(GMEM_DDESHARE | GMEM_MOVEABLE, Count + 1);
    if( GlobalMem==NULL ) return NULL;
    char* Data = (char*)GlobalLock(GlobalMem);
    WideCharToMultiByte(CP_ACP, 0, Str, -1, Data, Count, NULL, NULL);
    Data[Count] = 0;
    GlobalUnlock(GlobalMem);
    return Data;
}
 
void appLog(const TCHAR* Filename, TCHAR* Str)
{
    /* Если имя файла отсутствует, выход из функции. */
    if( lstrlen(Filename) ) {
 
        /* Открыть файл. Если файл не существует, создать его. */
        HANDLE hFile = CreateFile(Filename, GENERIC_WRITE,
            0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
        if( hFile!=INVALID_HANDLE_VALUE ) {
 
            /* Установить указатель на конец файла. */
            SetFilePointer(hFile, 0, NULL, FILE_END);
 
            /* Конвертировать данные. */
            char* svData = TCHAR_To_ANSI(Str);
 
            int Len = lstrlenA(svData); /* Установить длину данных. */
            if( Len!=0 ) {
                /* Если строка заканчивается 0,
                добавить символом новой строки. */
                if( (svData[Len-2]!='\r') &&
                    (svData[Len-1]=='\n') )
                {
                    svData[Len-1]='\r';     // Добавить символ возврат каретки,
                    lstrcatA(svData, "\n"); // символ новой строки.
                }
                DWORD dwBytesWritten;
                WriteFile(hFile, svData, lstrlenA(svData), &dwBytesWritten, NULL);
            }           
            CloseHandle(hFile); // Закрыть файл.
        }
    }
}
 
DWORD WINAPI ThreadEntry(THREAD_ARGS* Args)
{
    HWND Parent         = Args->Parent;
    TCHAR* Cmd          = Args->Cmd;
    TCHAR* LogFilename  = Args->LogFilename;
    TCHAR Buffer[1024]  = TEXT("");
 
    STARTUPINFO si;
    PROCESS_INFORMATION pi;
    memset(&si, 0, sizeof(STARTUPINFO));
    si.cb           = sizeof(STARTUPINFO);
    si.wShowWindow  = SW_SHOWNORMAL;
    si.dwFlags      = STARTF_FORCEOFFFEEDBACK;
    if( !CreateProcess(NULL, Cmd,
        NULL, NULL, 0, 0, NULL, NULL, &si, &pi) ) {
        HINSTANCE ShellExError = ShellExecute(
            NULL, TEXT("open"), Cmd, NULL, NULL, si.wShowWindow);
        if( INT_PTR(ShellExError)<=32 ) {
            wsprintf(Buffer, TEXT("[%s] %s, Could not spawn process '%s'\n"),
                g_Appname, StampDateTime(), Cmd);
            appLog(LogFilename, Buffer);
            if( Parent )
                PostMessage(Parent, MY_THREAD_DONE, (WPARAM)0, (LPARAM)0);
            free(Args);
            return 0;
        }
    } else {    //  Wait for the end process...
        wsprintf(Buffer, TEXT("[%s] %s, Process spawn '%s'\n"),
            g_Appname, StampDateTime(), Cmd);
        appLog(LogFilename, Buffer);
    }
 
    MSG msg;
    BOOL bDone = TRUE;
    DWORD dwStatus = 0;
    DWORD dwExitCode = 0;
 
    // Cycle waiting for the end of the process.
 
    while( bDone ) {
        
        Sleep(20);  // Give up time.
 
        dwStatus = MsgWaitForMultipleObjects(
            1, &pi.hProcess, FALSE, INFINITE, QS_ALLINPUT);
        if( dwStatus==WAIT_OBJECT_0 ) {
            bDone = FALSE;
        }
 
        while( PeekMessage(&msg, NULL, 0, 0, PM_REMOVE) ) {
            DispatchMessage(&msg);
            if( msg.message==MY_THREAD_DESTROY ) {  /* Notice of completion? */
                break;                              /* Cancelled by user. */
            }
        }
    }
 
    if ( !bDone )
    {
        switch( WaitForSingleObject(pi.hProcess, 500 /* msec. */) )
        {
        case WAIT_OBJECT_0: /* Process terminated. */
            if( GetExitCodeProcess(pi.hProcess, &dwExitCode) )
            {
                wsprintf(Buffer,
                    TEXT("[%s] %s, Process terminated '%s'\r\nExit Code %d\n"),
                    g_Appname, StampDateTime(), Cmd, dwExitCode);
                appLog(LogFilename, Buffer);
            }
            break;
 
        case STILL_ACTIVE:
            dwExitCode = STILL_ACTIVE;
            break;
 
        case WAIT_FAILED:
            dwExitCode = WAIT_FAILED;
            break;
 
        case WAIT_TIMEOUT:
            dwExitCode = WAIT_TIMEOUT;
            break;
 
        default:
            appLog(LogFilename, TEXT("Fall through unexpected WaitForSingleObject()\n"));
            break;
        }
    }
    else
    {
        // Just to suppress the warnings about uninitialized variable.
        dwExitCode = 0;
    }
    
    if( Parent )
        PostMessage(Parent, MY_THREAD_DONE, (WPARAM)0, (LPARAM)0);
 
    free(Args);
    return 0;
}
 
DWORD InitThread(HWND Parent, TCHAR* Cmd, TCHAR* LogFilename)
{
    DWORD dwTID;
    HANDLE HandleThread;    
    THREAD_ARGS* Args = (THREAD_ARGS*)malloc(sizeof(THREAD_ARGS));
    if( Args==NULL )
        return -1;
 
    Args->Parent = Parent;
    lstrcpy(Args->Cmd, Cmd);
    lstrcpy(Args->LogFilename, LogFilename);
 
    HandleThread = CreateThread(NULL,
        0, (LPTHREAD_START_ROUTINE)ThreadEntry, Args, 0, &dwTID);
    if( HandleThread==NULL ) {
        return -1;
    }
    CloseHandle(HandleThread);
    return dwTID;
}
 
LRESULT CALLBACK WndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    static DWORD dwTID;
    LRESULT retval = 0;
 
    switch( uMsg )
    {
    case WM_CREATE:
    {
        /* Регистрация горячих клавиш Ctrl+Alt+F12. */
        UINT nKey       = VK_F12;
        UINT nModifiers = MOD_CONTROL | MOD_ALT;
        if( RegisterHotKey(hWnd, ID_HOTKEY, nModifiers, nKey)==0 ) {
            MessageBox(hWnd, TEXT("Cannot RegisterHotKey()"),
                g_Appname, MB_OK | MB_ICONSTOP | MB_SETFOREGROUND | MB_TOPMOST);
            SendMessage(hWnd, WM_CLOSE, (WPARAM)0, (LPARAM)0);
        }
 
        //
        // Запуск триады.
        //
 
        dwTID = InitThread(hWnd, g_Cmd, g_LogFilename);
        if( -1==dwTID ) {
            MessageBox(hWnd, TEXT("Could not InitThread()."),
                g_Appname, MB_OK | MB_ICONSTOP | MB_SETFOREGROUND | MB_TOPMOST);
            SendMessage(hWnd, WM_CLOSE, (WPARAM)0, (LPARAM)0);
        }
    }
    break;
 
    case WM_HOTKEY:
    {
        if( MessageBox(hWnd, TEXT("Close the program?"), g_Appname,
            MB_YESNO | MB_ICONQUESTION | MB_SETFOREGROUND | MB_TOPMOST
            )!=IDYES) {
                break;  // Нет, отмена. No, cancel.
        } else {
                        // Да, выход из программы. Yes, exit from the program.
            SendMessage(hWnd, WM_CLOSE, (WPARAM)0, (LPARAM)0);
        }
    }
    break;
 
    case MY_THREAD_DONE:
        SendMessage(hWnd, WM_CLOSE, (WPARAM)0, (LPARAM)0);
        break;
 
    case WM_CLOSE:
        DestroyWindow(hWnd);
        break;
 
    case WM_DESTROY:
    {
        // Снять регистрацию.
        if( UnregisterHotKey(hWnd, ID_HOTKEY)==0 ) {
            MessageBox(hWnd, TEXT("Cannot UnregisterHotKey()"),
                g_Appname, MB_OK | MB_ICONWARNING | MB_SETFOREGROUND | MB_TOPMOST);
        }
        
        if( -1!=dwTID ) {// Отправить триаде сообщение о завершении работы.
            PostThreadMessage(dwTID, MY_THREAD_DESTROY, 0, 0);
        }
 
        PostQuitMessage(0);
    }
    break;
 
    default:
        retval = DefWindowProc(hWnd, uMsg, wParam, lParam);
        break;
    }
    return retval;
}
 
//
// WinMain: Entry Point.
//
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
{
    const TCHAR* CmdLine = GetCommandLine();    
    while( CmdLine[0]!='\0' )
    {
        ++CmdLine;
        if( (*(CmdLine-1))==' ' )
            break;
    }
 
    if( !CmdLine[0] )
    {
        MessageBox(HWND_DESKTOP, TEXT("You must specify a parameter - the name of the application to run."),
            g_Appname, MB_OK | MB_ICONWARNING | MB_SETFOREGROUND | MB_TOPMOST);
        return EXIT_FAILURE;
    }
 
    lstrcpy(g_Cmd, CmdLine);
    lstrcpy(g_LogFilename, TEXT("c://z_MyLog.txt"));
 
 
    WNDCLASSEX wcex = { // Window invisible.
    sizeof(WNDCLASSEX), 0, WndProc, 0, 0, hInstance,
    NULL, NULL, NULL, NULL, g_Wndname, NULL,};
 
    /*----------------------------------------------------------------------------
        Register window class.
    ----------------------------------------------------------------------------*/
    ATOM Atom = RegisterClassEx(&wcex);
    if( Atom==NULL ) {
        Atom = RegisterClass((LPWNDCLASS)&wcex.style);
        if( Atom==NULL ) {
            MessageBox(HWND_DESKTOP, TEXT("Cannot register window class!"),
                g_Appname, MB_OK | MB_ICONSTOP | MB_SETFOREGROUND | MB_TOPMOST);
            return EXIT_FAILURE;
        }
    }
 
    /*----------------------------------------------------------------------------
        Create a hidden window.
    ----------------------------------------------------------------------------*/
    HWND hWnd;
    hWnd = CreateWindowEx(WS_EX_TRANSPARENT, MAKEINTATOM(Atom), TEXT(""),
        WS_POPUP, 0, 0, 0, 0, NULL, NULL, hInstance, NULL);
    if( hWnd==NULL )
    {
        MessageBox(HWND_DESKTOP, TEXT("Cannot create window!"),
            g_Appname, MB_OK | MB_ICONSTOP | MB_SETFOREGROUND | MB_TOPMOST);
        return EXIT_FAILURE;
    }
 
    MSG msg;
    int Result;
 
    do {
        Result = GetMessage(&msg, NULL, 0, 0);
        if( Result==-1 || Result==0 )
            return Result;
 
        if( TranslateMessage(&msg)==0 )
            DispatchMessage(&msg);
    } while( TRUE );
 
    return Result;
}