linux使用管道命令执行ps获取cpu与内存占用率

彼岸随笔 2014-02-20

代码如下:

#include <stdio.h>
#include <unistd.h>
int main()
{
    char caStdOutLine[1024]; // ps 命令的标准输出中的一行信息
    char* pcTmp = NULL;      // 指向以空格拆分后的字符串

    char caSelfPID[10];      // 自身进程的PID字符串
    char caPSCmd[24];        // "ps aux | grep PID"命令字符串

    memset( caSelfPID, 0, sizeof( caSelfPID ) );
    sprintf( caSelfPID, 
             "%d", 
             getpid() );

    memset( caPSCmd, 0, sizeof( caPSCmd ) );
    sprintf( caPSCmd, 
             "ps aux | grep %d", 
             getpid() );

    do // 非循环,只是为了方便控制分支层次,便于控制分支流向
    {
        // 通过创建一个管道,调用 fork 产生一个子进程,
        // 执行一个 shell 以运行命令来开启一个进程。
        // 这个进程必须由 pclose() 函数关闭。
        FILE* fp = popen( caPSCmd, // 一个指向以 NULL 结束的 shell 命令字符串的指针,
                                   // 这行命令将被传到 bin/sh 并使用 -c 标志,
                                   // 那么 shell 将执行这个命令从这个字符串中读取。
                          "r" );   // 文件指针连接到 shell 命令的标准输出


        if ( NULL == fp )
        {
            printf( "call popen is failed\n" );
            break;
        }


        memset( caStdOutLine, 0, sizeof( caStdOutLine ) );
        while ( NULL != fgets( caStdOutLine,
                               sizeof( caStdOutLine ),
                               fp ) )
        {
            // 再以空格分隔符拆分字符串
            pcTmp = strtok( caStdOutLine, " " );

            // 用户名跳过,直接匹配 PID ,不匹配跳过
            pcTmp = strtok( NULL, " " );
            if ( 0 != strncasecmp( caSelfPID, 
                                   pcTmp, 
                                   strlen( caSelfPID ) ) ) 
            {
                continue;
            }


            // 读出进程自身 CPU 占用率
            pcTmp = strtok( NULL, " " );
            printf( "CPU = %s %%\n", pcTmp );


            // 读出进程自身 MEM 占用率
            pcTmp = strtok( NULL, " " );
            printf( "MEM = %s %%\n", pcTmp );

            break;
        }


        // 关闭标准 I/O 流,等待命令执行结束,然后返回 shell 的终止状态。
        // 如果 shell 不能被执行,
        // 则 pclose() 返回的终止状态与 shell 已执行 exit 一样。
        pclose( fp );

    }while ( 0 );
}

代码如下:

$ gcc main.c -o test
$ ./test
CPU = 1.0 %
MEM = 0.0 %

$ ps  aux
USER       PID %CPU %MEM    VSZ   RSS TTY      STAT START   TIME COMMAND
nsc      24505  1.0  0.0   2004   232 pts/0    S+   09:46   0:00 ./test

相关推荐