Shell脚本调用C语言之字符串切分函数——strtok

今天上午在写一个需求，要求的比较急，要求当天完成，我大致分析了一下，可以采用从shell脚本中插入一连串的日期，通过调用proc生成的可执行文件，将日期传入后台数据库，在数据库中进行计算。需要切分日期的字符串，转化成整数，插入int 数组中，手工实现太慢，就直接借用系统的strtok函数来用了。

场景模拟：

1. shell脚本：

#diao.sh 
#!/bin/bash
date1="20170622,20170623,20170626,20170627,20170628,20170629,20170627"
date2="20170628,20170629,20170630"

if [ $1 -eq 0 ]
       then
                compute $date1
else
                compute $date2
~                             

2. 后台proc代码，这里用C代码来模拟，重点讲述用strtok函数实现字符串的切分。

#include<string.h>
#include<stdlib.h>
#include<stdio.h> int main(int argv ,char * argc[])
{
 char buf[100];
 char * p = NULL;
 char buf2[100][9];
 int  data[100];
 int len  = 0;
 int i = 0;
  memset(buf,0x00,sizeof(buf));
  memset(buf2,0x00,sizeof(buf2)); memset(data,0x00,sizeof(data));
  memcpy(buf,argc[1],strlen(argc[1]));

  printf("buf=%s\n",buf);
/* 下面代码按照","切分字符串，然后转化成整数，存入整数数组中*/
  p= strtok(buf, ",");             
 while( p!= NULL){ 
        strcpy(buf2[len],p);
        data[len]= atoi(buf2[len]);       
        printf("buf2[%d]=%s\n",len,buf2[len]);
        len++;
        p= strtok(NULL, ",");          // 再次调用strtok函数 
    }
 /* 上面的代码按照","切分字符串，然后转化成整数，存入整数数组中*/



for ( i = 0 ; i < len ; ++i){
      printf ("data[%d]=%d\n",i,data[i]);
  }

}

编译运行情况：

 思考:将上述代码中字符串切割，并转化为整数，存入整数数组部分做成一个独立的函数，进行调用，通用性一下子就上来了。

3. 将切分过程做成一个独立的函数

函数名称为：mystrtok，里面还是调用系统的strtok,如果直接用系统的strtok不做任何处理，是试用不了的，因为strtok出来的都是char*类型的。

#include<string.h>
#include<stdlib.h>
#include<stdio.h>
int mystrtok(char * str,const char * delim, char  buf[][9],int * len, int data[])
{
  char * p = NULL;
  int i = 0;
   p = strtok(str, delim);              
  while( p!= NULL){  
        strcpy(buf[i],p);
        data[i] = atoi(buf[i]);         
        i++;
        p = strtok(NULL, delim);          // 再次调用strtok函数  
    } 
    *len = i;
   return 0;
}
int main(int argv ,char * argc[])
{
  char buf[100];
  char * p = NULL;
  char buf2[100][9];
  int   data[100];
  int len  = 0;
  int i = 0;
  memset(buf,0x00,sizeof(buf));
  memset(buf2,0x00,sizeof(buf2));
  memset(data,0x00,sizeof(data));
  memcpy(buf,argc[1],strlen(argc[1]));
   
  printf("buf=%s\n",buf);  
 /* 下面代码按照","切分字符串，然后转化成整数，存入整数数组中*/
/*   p = strtok(buf, ",");              
  while( p!= NULL){  
        strcpy(buf2[len],p);
        data[len] = atoi(buf2[len]);         
        printf("buf2[%d]=%s\n",len,buf2[len]);
        len++;
        p = strtok(NULL, ",");          // 再次调用strtok函数  
    } */ 
  /* 上面的代码按照","切分字符串，然后转化成整数，存入整数数组中*/ 
  /* 思考，将上述代码写成一个独立的函数，进行调用*/
  mystrtok(buf,",",buf2,&len,data); 
  for ( i = 0 ; i < len ; ++i){
      printf ("data[%d]=%d\n",i,data[i]);
   }
  
   
}

运行新的代码：

上述函数可以在任何字符串切割的场景中用到，尤其是数字字符串按照某种方式切割时。

另外一个值得注意的地方就是:shell脚本调用C程��时，main函数的参数中接受到shell脚本的参数，然后进行处理。

特别是字符串类型 char * ，字符数组 char buf[][]，字符数组指针 char *p[], const char * 这些类型一定要搞清楚，之间是否可以转，怎么转，互相之间如何赋值的，都要非常清楚。