建站
咨询
售后
建站咨询
C/C++中一次性执行多个DOS命令-创新互联
起因
最近给公司的一个系统写了个启动的脚本，但是领导说批处理这样的脚本太low了，要使用EXE来启动，未来还要使用加密工具对EXE进行加密。
好吧，我就在网上到处找bat转exe的工具，找了很久，都没有找到合适的，只有一个用winrar制作自解压包的方法还算可以，但是这玩意儿有两个坑爹的问题：
- 使用了自定义图标后，安装时会被360报告有***；
- 用winrar制作的exe，其本质还是解压后执行，解压后的文件其实可以在系统临时目录下找到，因此以后想要加密其实很容易就会被破解；
所以最好的办法看来就是自己写一个exe了，考虑到我以前用过C，因此下载了Dev-Cpp这个工具来编写代码。
思路
在C语言中执行DOS命令的方法很多，如：ShellExecute, WinExec, CreateProcess等，但是这些接口都是只能一次执行一条命令，在我的启动脚本里有很多命令，有一些是设置环境变量的，这样就没法在代码中一条条执行脚本中的命令，必须要找到一个办法可以一次性执行多条命令。
在网上找了很久，最终确定使用CreateProcess，同时要使用管道技术。也就是使用CreateProcess创建一个cmd进程，然后通过输入管道将待执行的命令传递给cmd进程，通过输出管道获取cmd进程的输出信息，因为是通过管道进行，所以可以模拟在DOS窗口一行行输入命令，从而实现执行多条DOS命令了。
实现
从MSDN上找到管道的示例代码，简单修改了一下。
首先，将CreateProcess的参数改为启动cmd：
char cmdLine[] = "cmd";
// Create the child process.
bFuncRetn = CreateProcess(NULL,
cmdLine, // command line
NULL, // process security attributes
NULL, // primary thread security attributes
TRUE, // handles are inherited
0, // creation flags
NULL, // use parent's environment
NULL, // use parent's current directory
&siStartInfo, // STARTUPINFO pointer
&piProcInfo); // receives PROCESS_INFORMATION
然后，将原来批处理里面的脚本复制一下，放到一个变量里（这里我改了一下，没有用我实际的脚本，因为那个不通用，不适合做例子），注意，每一行最后要加上回车符\n，这样才能正确模拟DOS窗口中输入命令的情况：
CHAR cmds[] = "@ECHO OFF\n"
"cd..\n"
"dir\n"再然后，原来的示例代码中是把批处理文件作为EXE的参数传递进来的，既然上面改为将批处理文件内容放到脚本里，代码中从文件中读取命令的那部分就要去掉了，这部分代码就不多说了。
完整的示例代码如下：
#include
#include
#define BUFSIZE 4096
HANDLE hChildStdinRd, hChildStdinWr, hChildStdinWrDup,
hChildStdoutRd, hChildStdoutWr, hChildStdoutRdDup,
hInputFile, hStdout;
BOOL CreateChildProcess(VOID);
VOID WriteToPipe(VOID);
VOID ReadFromPipe(VOID);
VOID ErrorExit(const char *);
VOID ErrMsg(LPTSTR, BOOL);
int main(int argc, char *argv[]) {
// SECURITY_ATTRIBUTES结构包含一个对象的安全描述符，并指定检索到指定这个结构的句柄是否是可继承的。
// 这个结构为很多函数创建对象时提供安全性设置
SECURITY_ATTRIBUTES saAttr;
BOOL fSuccess;
// Set the bInheritHandle flag so pipe handles are inherited.
// 设置句柄为可继承的，使得子线程可以使用父线程
saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
saAttr.bInheritHandle = TRUE;
saAttr.lpSecurityDescriptor = NULL;
// Get the handle to the current STDOUT.
// 取得当前应用的标准输出句柄，对于Windows控制台应用来说，一般是输出到屏幕
hStdout = GetStdHandle(STD_OUTPUT_HANDLE);
// Create a pipe for the child process's STDOUT.
// 创建一个用于输出操作的匿名管道。
if (! CreatePipe(&hChildStdoutRd, &hChildStdoutWr, &saAttr, 0))
ErrorExit("Stdout pipe creation failed\n");
// Create noninheritable read handle and close the inheritable read handle.
// 将输出管道的句柄绑定到当前进程
fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdoutRd,
GetCurrentProcess(), &hChildStdoutRdDup , 0,
FALSE,
DUPLICATE_SAME_ACCESS);
if( !fSuccess )
ErrorExit("DuplicateHandle failed");
CloseHandle(hChildStdoutRd);
// Create a pipe for the child process's STDIN.
// 创建一个用于输入操作的匿名管道。
if (! CreatePipe(&hChildStdinRd, &hChildStdinWr, &saAttr, 0))
ErrorExit("Stdin pipe creation failed\n");
// Duplicate the write handle to the pipe so it is not inherited.
// 将输入管道的句柄绑定到当前进程
fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdinWr,
GetCurrentProcess(), &hChildStdinWrDup, 0,
FALSE, // not inherited
DUPLICATE_SAME_ACCESS);
if (! fSuccess)
ErrorExit("DuplicateHandle failed");
CloseHandle(hChildStdinWr);
// Now create the child process.
// 创建DOS子进程
fSuccess = CreateChildProcess();
if (! fSuccess)
ErrorExit("Create process failed");
// Write to pipe that is the standard input for a child process.
writeToPipe();
// Read from pipe that is the standard output for child process.
ReadFromPipe();
return 0;
}
BOOL CreateChildProcess() {
PROCESS_INFORMATION piProcInfo;
STARTUPINFO siStartInfo;
BOOL bFuncRetn = FALSE;
// Set up members of the PROCESS_INFORMATION structure.
ZeroMemory( &piProcInfo, sizeof(PROCESS_INFORMATION) );
// Set up members of the STARTUPINFO structure.
// 设定DOS进程的标准输入、输出和错误信息的管道
// 使用前面创建的值，DOS窗口的输入输出都会被定向到本应用中
ZeroMemory( &siStartInfo, sizeof(STARTUPINFO) );
siStartInfo.cb = sizeof(STARTUPINFO);
siStartInfo.hStdError = hChildStdoutWr;
siStartInfo.hStdOutput = hChildStdoutWr;
siStartInfo.hStdInput = hChildStdinRd;
siStartInfo.dwFlags |= STARTF_USESTDHANDLES;
char cmdLine[] = "cmd";
// Create the child process.
bFuncRetn = CreateProcess(NULL,
cmdLine, // command line
NULL, // process security attributes
NULL, // primary thread security attributes
TRUE, // handles are inherited
0, // creation flags
NULL, // use parent's environment
NULL, // use parent's current directory
&siStartInfo, // STARTUPINFO pointer
&piProcInfo); // receives PROCESS_INFORMATION
if (bFuncRetn == 0)
ErrorExit("CreateProcess failed");
else {
CloseHandle(piProcInfo.hProcess);
CloseHandle(piProcInfo.hThread);
return bFuncRetn;
}
}
VOID WriteToPipe(VOID) {
DWORD dwRead, dwWritten;
CHAR chBuf[BUFSIZE];
CHAR cmds[] = "@ECHO ON\n"
"cd..\n"
"dir\n";
writeFile(hChildStdinWrDup, cmds, sizeof(cmds), &dwWritten, NULL);
// Close the pipe handle so the child process stops reading.
if (! CloseHandle(hChildStdinWrDup))
ErrorExit("Close pipe failed");
}
VOID ReadFromPipe(VOID) {
DWORD dwRead, dwWritten;
CHAR chBuf[BUFSIZE];
// Close the write end of the pipe before reading from the
// read end of the pipe.
if (!CloseHandle(hChildStdoutWr))
ErrorExit("CloseHandle failed");
// Read output from the child process, and write to parent's STDOUT.
// 获取子线程，即DOS窗口的输出，显示到标准输出设备上
for (;;) {
if( !ReadFile( hChildStdoutRdDup, chBuf, BUFSIZE, &dwRead,
NULL) || dwRead == 0) break;
if (! WriteFile(hStdout, chBuf, dwRead, &dwWritten, NULL))
break;
}
}
VOID ErrorExit (const char *lpszMessage) {
fprintf(stderr, "%s\n", lpszMessage);
ExitProcess(0);
} 执行效果如下图：
main.exe的原始目录是D:\Workspace\research\C\Chrome\，执行时，首先执行了cd..，退到上一层目录，然后执行dir，显示上一层目录的内容，证明上面的代码确实可以一次执行多条DOS命令。
另外有需要云服务器可以了解下创新互联scvps.cn，海内外云服务器15元起步，三天无理由+7*72小时售后在线，公司持有idc许可证，提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案，具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势，专为企业上云打造定制，能够满足用户丰富、多元化的应用场景需求。
分享标题：C/C++中一次性执行多个DOS命令-创新互联
本文地址：http://scpingwu.com/article/dedsji.html
