正文

揭秘C语言异步读取CMD，轻松判断命令执行完成！

/0 浏览量
0321

在C语言编程中，异步读取命令行（CMD）输出是一个常见的需求，特别是在开发需要与系统命令交互的程序时。异步读取允许程序在等待命令执行的同时继续执行其他任务。本文将详细介绍如何在C语言中实现异步读取CMD，并轻松判断命令执行完成。

1. 异步读取CMD的原理

异步读取CMD主要依赖于操作系统提供的API，如Windows中的CreatePipe和CreateProcess函数，以及Linux中的fork、pipe和exec函数。这些API允许程序创建一个管道，并通过该管道与子进程通信。

2. Windows平台下的实现

在Windows平台下，我们可以使用CreatePipe和CreateProcess函数来创建一个管道，并通过该管道异步读取CMD的输出。

#include <windows.h>
#include <stdio.h>

int main() {
    SECURITY_ATTRIBUTES saAttr;
    HANDLE hStdinRd, hStdinWr, hStdoutRd, hStdoutWr;
    STARTUPINFO siStartInfo;
    PROCESS_INFORMATION piProcInfo;
    char command[] = "echo Hello, World!";

    // 初始化安全属性
    saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
    saAttr.bInheritHandle = TRUE;
    saAttr.lpSecurityDescriptor = NULL;

    // 创建管道
    if (!CreatePipe(&hStdinRd, &hStdinWr, &saAttr, 0) ||
        !CreatePipe(&hStdoutRd, &hStdoutWr, &saAttr, 0)) {
        // 创建管道失败
        return 1;
    }

    // 设置进程信息
    ZeroMemory(&siStartInfo, sizeof(STARTUPINFO));
    siStartInfo.cb = sizeof(STARTUPINFO);
    siStartInfo.hStdError = hStdoutWr;
    siStartInfo.hStdOutput = hStdoutWr;
    siStartInfo.hStdInput = hStdinRd;
    siStartInfo.dwFlags |= STARTF_USESTDHANDLES;

    // 创建进程
    if (!CreateProcess(NULL, command, NULL, NULL, TRUE, 0, NULL, NULL, &siStartInfo, &piProcInfo)) {
        // 创建进程失败
        return 1;
    }

    // 关闭不必要的句柄
    CloseHandle(hStdinWr);
    CloseHandle(hStdoutWr);

    // 异步读取输出
    char buffer[1024];
    DWORD bytesRead;
    while (ReadFile(hStdoutRd, buffer, sizeof(buffer), &bytesRead, NULL) && bytesRead > 0) {
        printf("%s", buffer);
    }

    // 关闭句柄
    CloseHandle(hStdoutRd);
    CloseHandle(hStdinRd);
    CloseHandle(piProcInfo.hProcess);
    CloseHandle(piProcInfo.hThread);

    return 0;
}

3. Linux平台下的实现

在Linux平台下，我们可以使用fork、pipe和exec函数来实现异步读取CMD。

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/wait.h>

int main() {
    int pipefd[2];
    pid_t cpid;
    char buffer[1024];
    ssize_t num_bytes;

    // 创建管道
    if (pipe(pipefd) == -1) {
        perror("pipe");
        exit(EXIT_FAILURE);
    }

    // 创建子进程
    cpid = fork();
    if (cpid == -1) {
        perror("fork");
        exit(EXIT_FAILURE);
    }

    if (cpid == 0) { // 子进程
        // 关闭不需要的管道句柄
        close(pipefd[1]);

        // 将标准输出重定向到管道
        dup2(pipefd[0], STDOUT_FILENO);
        close(pipefd[0]);

        // 执行命令
        execlp("echo", "echo", "Hello, World!", NULL);
        // 如果execlp返回，则表示出错
        perror("execlp");
        exit(EXIT_FAILURE);
    } else { // 父进程
        // 关闭不需要的管道句柄
        close(pipefd[0]);

        // 等待子进程结束
        wait(NULL);

        // 异步读取输出
        while ((num_bytes = read(pipefd[1], buffer, sizeof(buffer) - 1)) > 0) {
            buffer[num_bytes] = '\0';
            printf("%s", buffer);
        }

        // 关闭管道句柄
        close(pipefd[1]);
    }

    return 0;
}

4. 判断命令执行完成

在上述代码中，通过wait(NULL)函数可以等待子进程结束。在Linux平台下，还可以使用waitpid函数来等待特定的子进程结束。

在Windows平台下，可以通过检查ReadFile函数的返回值来判断命令是否执行完成。如果返回值小于0，则表示读取失败或命令执行完成。

5. 总结

本文介绍了在C语言中实现异步读取CMD的方法，并通过示例代码展示了在Windows和Linux平台下的具体实现。通过这些方法，我们可以轻松地判断命令是否执行完成，并获取命令的输出结果。

-- 展开阅读全文 --