【转载】在LoadRunner中执行命令行程序之：popen()取代system()

我想大家应该都知道在LoadRunner可以使用函数system()来调用系统指令，结果同在批处理里执行一样。

但是system()有个缺陷：无法获取命令的返回结果。

也许你可以用`echo command > file`将结果输出到文件，然后再读出来。多痛苦的一件事啊！当虚拟用户多的时候，你还得考虑重复写同一个文件存在的风险。或是将文件名进行参数化以保证不同相同的文件，但这样又加大了IO的负载。

实际上有更好的解决办法：使用popen()

popen()用法上类似于fopen()，但不需要打开文件，直接读取数据流。全内存操作，不需要考虑文件覆盖，更不存在磁盘IO。

举例如下：

#define BUFFER_SIZE 10240 // 初始给10KB 
Action(){ 
    long fp; // 流的指针 
    int count; // 流的长度 
    char buffer[BUFFER_SIZE]; // 给输出分配内存空间 
    /* 
     * 下来我们来获取当前系统的用户名称 
     */ 
    fp = popen("echo %USERDOMAIN%", "r"); 
    if (fp == NULL) { 
        lr_error_message("命令执行失败"); 
        return -1; 
    } 
    count = fread(buffer, sizeof(char), BUFFER_SIZE, fp); // 读取返回数据 
    if (feof(fp) == 0) { 
        lr_error_message("请给流分配更多的内存空间，谢谢！"); 
        return -1; 
    } 
    if (ferror(fp)) { 
        lr_error_message ("流读取失败"); 
        return -1; 
    } 
    buffer[count] = NULL; 
    lr_output_message("Notify:当前用户: %s", buffer); 
    pclose(fp); 
    return 0; 
}

输出：

Running Vuser... 
Starting iteration 1. 
Starting action Action. 
Action.c(28): Notify:当前用户: higkoo 
Ending action Action. 
Ending iteration 1. 
Ending Vuser...

OK，下面再举个例子，逐行获取结果并逐行输出：

extern char* strtok(char *token, const char *delimiter); // 提前申明 
#define BUFFER_SIZE 20480 // 初始给它20 KB先 
Action(){ 
    long fp; // 流指针 
    int count; //流长度 
    char buffer[BUFFER_SIZE]; // 给流分配内存 
    char * token; 
    char param_buf[10]; // 用于保存参数名称 
    int i; 
    /* 
     * 显示已安装设备驱动程序列表 
     */ 
    fp = popen("DRIVERQUERY", "r"); 
    if (fp == NULL) { 
        lr_error_message("命令执行失败"); 
        return -1; 
    } 
    count = fread(buffer, sizeof(char), BUFFER_SIZE, fp); // 拿返回结果 
    if (feof(fp) == 0) { 
        lr_error_message("返回值过多，请加大流长度"); 
        return -1; 
    } 
    if (ferror(fp)) { 
        lr_error_message ("结果读取失败"); 
        return -1; 
    } 
    buffer[count] = NULL; 
    token = (char*) strtok(buffer, "\n"); // 按换行符分割 
    if (token == NULL) { 
        lr_error_message ("返回值里没有换行"); 
        return -1; 
    } 
    i = 1; 
    while (token != NULL) { // 逐行的读结果 
        sprintf(param_buf, "output_%d", i); 
        lr_save_string(token, param_buf); 
        i++; 
        token = (char*) strtok(NULL, "\n"); 
    } 
    lr_save_int(i-1, "output_count"); 
    for (i=1; i<=lr_paramarr_len("output"); i++) {// 把所有行都打出来 
        lr_output_message("设备信息: %s", lr_paramarr_idx("output", i)); 
    } 
    pclose(fp); 
    return 0; 
}

这个结果比较多，略。