C与脚本语言awk混合编程[转载]

作者：    陈轶飞

最后更新：   2003-09-09

关键词：  c、脚本、awk、shell、perl

 

在linux上写程序、做网管的人，或多或少都会几种脚本。脚本语言灵活的变量类型、强大的正则表达式处理能力，再加上linux系统本身的管道、重定向以及丰富的命令行工具，让你编程起来游刃有余。

而C语言固然有种种优势，但不可否认，很多场合下，用脚本语言更为方便，比如我们将举例说明的对配置文件的处理。

 

先看看我们示例程序的任务：

 

假设我们有一个用c写的程序，它有一个配置文件 user.conf，保存了一些用户信息，user.conf定义如下：

1）、以 # 开头的行为注释行，不做处理

2）、允许空行

3）、如果不是1和2，那么就是有效的数据，格式如下

# user.conf: configure file for user
# username  age   sex   country
tom   20    male        us
chen  22    female      cn

 

每一列分为4个字段，字段之间用一个或多个空白字符（空格或者制表符）隔开，字段依次是 姓名、年龄、性别、国家

 

我们的c程序要完成对 user.conf的添加、删除、编辑、查询

 

这样一个简单的任务，用c处理起来不算复杂，不过也是要花点功夫的，而如果用脚本语言来做，却很简单，能不能在c中调用脚本来完成任务了？

 

Awk是linux上一种脚本语言，它的长处在于处理有一定格式规则的文件，例如咱们的user.conf。关于 awk 的资料有很多，oreilly公司出了专门的 awk 编程的书籍，网上也是可以下载到的。你也可以直接 man  awk看看。

 

我们先看看如何用 shell 结合 awk来完成上述任务：

1）  添加一条记录

例如，要添加 jack  18  male  us 这样一条记录，可以简单的用重定向功能

Echo  –e  “jack  18  male  us” >>  user.conf

 

现在，这条记录被添加到 user.conf末尾了。

 

2）  删除一条记录

例如，现在要删除用户 chen 的信息

 

cat user.conf | awk ‘!/^chen[[:blank:]]+/ {print}’ > tmp.conf; mv –f tmp.conf user.conf

 

3）、编辑一条记录
现在，想把 tom的性别改为 female

 

cat user.conf | awk ‘{if($0 ~ /^tom[[:blank:]]+/) print $1 $2 female $3; else print}’
 
通过 system()这个函数，我们就可以在 c 中调用以上脚本，完成任务了。
但是，system() 用起来还是觉得不爽，它的不足是只能执行脚本，却无法获得脚本的输出数据，而这通常是我们进一步处理的数据来源。（在shell和perl中，可以通过反引号( `` )来取得命令的输出结果）。 一个解决办法是把输出结果重定向到一个临时文件中，然后在c中读取文件，获取数据，最后当然还要删除这个文件。不过，这个方法总是让人觉得有一点点不爽，如果能直接把脚本执行中输出的数据输到我们的缓冲区来就更好了。

我写了个小函数，叫 my_system()，通过管道以及重定向，实现了以上想法。函数原型如下：

 

int my_system(const char* pCmd, char* pResult, int size)；

 

输出数据被保存到 pResult所指向的缓冲区中，缓冲区大小为 size，最多可以保存 size-1的数据。

 

函数的实现放在本文的最后

 

有了这个函数以后，在 c中调用脚本就更方便了，我们可以通过它来实现对 user.conf的查询。

 

4）、查询一个记录
例如，我们要获取 tom 的性别
可以用脚本这样来实现：

cat user.conf | awk ‘/^tom[[:blank:]]+/ {print $3}’
 
脚本的执行结果是 tom的性别 male被输出到屏幕上

 

在我们的 c程序中，如此调用 my_system()，

 

char  buf[101];
my_system(“cat user.conf | awk ‘/^tom[[:blank:]]+/ {print $3}’”, buf, 101);
 

调用完以后，buf中的数据就是 “male”了，怎么样，还算方便吧？

 

以上只是用结合脚本完成了一个比较简单的任务，所以我没有把这些脚本单独形成脚本文件。如果你善于使用 perl、shell、awk，那么可以写出更强大的脚本文件来处理更复杂的问题，然后通过类似 my_system( )的方法，在 c/c++等其它语言中取得脚本的输出结果，实现有趣的“混合编程”。

 

希望你能从中得到乐趣！

＃i nclude
＃i nclude
＃i nclude
＃i nclude
＃i nclude

static int my_system(const char* pCmd, char* pResult, int size)

{

   int fd[2];

   int pid;

   int count;

   int left;

   char* p = 0;

   int maxlen = size – 1;

   memset(pResult, 0, size);

 

   if(pipe(fd))

   {

       printf("pipe error\n");

       return –1;

   }

   if((pid = fork()) == 0)

   {// chile process

 

       int fd2[2];

       if(pipe(fd2))

       {

            printf("pipe2 error\n");

            return –1;

       }

       close(1);

       dup2(fd2[1],1);

       close(fd[0]);

       close(fd2[1]);

       system(pCmd);

       read(fd2[0], pResult, maxlen);

       pResult[strlen(pResult)-1] = 0;

       write(fd[1], pResult, strlen(pResult));

       close(fd2[0]);

       exit(0);

     }

 

   // parent process

   close(fd[1]);

   p = pResult; 

   left = maxlen;

   while((count = read(fd[0], p, left))) {

       p += count;

       left -= count;

       if(left == 0)

            break;

   }

   close(fd[0]);

   return 0;

}

 

int main(void)

{

   char result[1025];

   my_system("/sbin/ifconfig", result, 1025);

   printf("the result is\n\n%s\n", result);

       return 0;

}