C语言实现<读取>和<写入> *.ini文件（转）

原地址：https://blog.csdn.net/niha1993825jian/article/details/41086403

#include <stdio.h>  
#include <errno.h>  
#include <ctype.h>  
#include <string.h>  
#include <stdlib.h>  
#include <stdbool.h>  
  
#define  MAX_VALUE  64 /* 定义section,key,value字符串最大长度 */  
// printf("File = %s\nLine = %d\nFunc=%s\nDate=%s\nTime=%s\n", __FILE__, __LINE__, __FUNCTION__, __DATE__, __TIME__);  
#define  PRINT_ERRMSG(STR) fprintf(stderr,"line:%d,msg:%s,eMsg:%s\n", __LINE__, STR, strerror(errno))  
  
typedef struct _option {  
  char    key[MAX_VALUE];   /* 对应键 */  
  char    value[MAX_VALUE]; /* 对应值 */  
  struct  _option *next;    /* 链表连接标识 */  
}Option;  
  
typedef struct _data {  
  char    section[MAX_VALUE]; /* 保存section值          */  
  Option  *option;            /* option链表头           */  
  struct  _data *next;        /* 链表连接标识           */  
}Data;  
  
typedef struct {  
  char    comment;              /* 表示注释的符号    */  
  char    separator;            /* 表示分隔符        */  
  char    re_string[MAX_VALUE]; /* 返回值字符串的值  */  
  int     re_int;               /* 返回int的值       */  
  bool    re_bool;              /* 返回bool的值      */  
  double  re_double ;           /* 返回double类型    */  
  Data    *data;                /* 保存数据的头      */  
}Config;  
  
/** 
* 判断字符串是否为空 
* 为空返回true,不为空返回false 
**/  
bool str_empty(const char *string)  
{  
  return NULL == string || 0 == strlen(string);  
}  
  
/** 
* 向链表添加section,key,value 
* 如果添加时不存在section则新增一个 
* 如果对应section的key不存在则新增一个 
* 如果section已存在则不会重复创建 
* 如果对应section的key已存在则只会覆盖key的值 
**/  
bool cnf_add_option(Config *cnf, const char *section, const char *key, const char *value)  
{  
  if (NULL == cnf || str_empty(section) || str_empty(key) || str_empty(value)) {  
    return false; /* 参数不正确,返回false */  
  }  
    
  Data *p = cnf->data; /* 让变量p循环遍历data,找到对应section */  
  while (NULL != p && 0 != strcmp(p->section, section)) {  
    p = p->next;  
  }  
    
  if (NULL == p) { /* 说明没有找到section,需要加一个 */  
    Data *ps = (Data*)malloc(sizeof(Data));  
    if (NULL == ps) {  
      exit(-1); /* 申请内存错误 */  
    }  
    strcpy(ps->section, section);  
    ps->option = NULL;    /* 初始的option要为空 */  
    ps->next = cnf->data; /* cnf->data可能为NULL */  
    cnf->data = p = ps;   /* 头插法插入链表 */  
  }  
    
  Option *q = p->option;  
  while (NULL != q && 0 != strcmp(q->key, key)) {  
    q = q->next; /* 遍历option,检查key是否已经存在 */  
  }  
    
  if (NULL == q) { /* 不存在option,则新建一个 */  
    q = (Option*)malloc(sizeof(Option));  
    if (NULL == q) {  
      exit(-1); /* 申请内存错误 */  
    }  
    strcpy(q->key, key);  
    q->next = p->option; /*这里p->option可能为NULL,不过也没关系 */  
    p->option = q; /* 头插法插入链表 */  
  }  
  strcpy(q->value, value); /* 无论如何要把值改了 */  
    
  return true;  
}  
  
/** 
* 去掉字符串内所有空白 
* 且忽略注释部分 
* 最终得到没有空白的字符串 
**/  
bool strip_comments(char *string, char comment)  
{  
  if (NULL == string || '\n' == *string || '\r' == *string) {  
    return false; /* 第一个字符为回车或换行,表示空行 */  
  }  
    
  char *p, *q; /* 下面去掉字符串中所有空白字符 */  
  for (p = q = string; *p != '\0' && *p != comment; p++) {  
    if (0 == isspace(*p)) {  
      *q++ = *p; /* 不是空白字符则重写字符串 */  
    }  
  }  
  *q = '\0';  
    
  return 0 != strlen(string); /* 字符串长度不为0,表示数据可用 */  
}  
  
/** 
* 传递配置文件路径 
* 参数有文件路径,注释字符,分隔符 
* 返回Config结构体 
**/  
Config *cnf_read_config(const char *filename, char comment, char separator)  
{  
  Config *cnf     = (Config*)malloc(sizeof(Config));  
  cnf->comment    = comment; /* 每一行该字符及以后的字符将丢弃 */  
  cnf->separator  = separator; /* 用来分隔Section 和 数据 */  
  cnf->data       = NULL; /* 初始数据为空 */  
    
  if (str_empty(filename)) {  
    return cnf; /* 空字符串则直接返回对象 */  
  }  
    
  char *p, sLine[MAX_VALUE];    /* 保存一行数据到字符串 */  
  char section[MAX_VALUE], key[MAX_VALUE], value[MAX_VALUE]; /* 缓存section,key,value */  
  FILE *fp = fopen(filename, "r");  
  if(NULL == fp) {  
    PRINT_ERRMSG("fopen");  
    exit(errno); /* 读文件错误直接按照错误码退出 */  
  }  
    
  while (NULL != fgets(sLine, MAX_VALUE, fp)) {  
    if (strip_comments(sLine, cnf->comment)) { /* 去掉字符串所有空白,注释也忽略 */  
      if ('[' == sLine[0] && ']' == sLine[strlen(sLine)-1]) {  
        memset(section, '\0', MAX_VALUE); /* 清空section,因为strncpy不追加'\0' */  
        strncpy(section, sLine+1, strlen(sLine)-2);  
      } else if (NULL != (p = strchr(sLine, cnf->separator))) {  /* 存在分隔符 */  
        memset(key,   '\0', MAX_VALUE); /* 清空key,因为strncpy不追加'\0' */  
        strncpy(key,  sLine, p - sLine);  
        strcpy(value, p + 1); /* strcpy会追加'\0',所以妥妥哒 */  
        cnf_add_option(cnf, section, key, value); /* 添加section,key,value */  
      } /* 如果该行不存在分隔符则忽略这一行 */  
    } /* end strip_comments */  
  } /* end while */  
    
  fclose(fp);  
  return cnf;  
}  
  
/** 
* 获取指定类型的值 
* 根据不同类型会赋值给对应值 
* 本方法需要注意,int和double的转换,不满足就是0 
*     需要自己写代码时判断好 
**/  
bool cnf_get_value(Config *cnf, const char *section, const char *key)  
{  
  Data *p = cnf->data; /* 让变量p循环遍历data,找到对应section */  
  while (NULL != p && 0 != strcmp(p->section, section)) {  
    p = p->next;  
  }  
    
  if (NULL == p) {  
    PRINT_ERRMSG("section not find!");  
    return false;  
  }  
    
  Option *q = p->option;  
  while (NULL != q && 0 != strcmp(q->key, key)) {  
    q = q->next; /* 遍历option,检查key是否已经存在 */  
  }  
    
  if (NULL == q) {  
    PRINT_ERRMSG("key not find!");  
    return false;  
  }  
    
  strcpy(cnf->re_string, q->value);       /* 将结果字符串赋值 */  
  cnf->re_int    = atoi(cnf->re_string);  /* 转换为整形 */  
  cnf->re_bool   = 0 == strcmp ("true", cnf->re_string); /* 转换为bool型 */  
  cnf->re_double = atof(cnf->re_string);  /* 转换为double型 */  
    
  return true;  
}  
  
/** 
* 判断section是否存在 
* 不存在返回空指针 
* 存在则返回包含那个section的Data指针 
**/  
Data *cnf_has_section(Config *cnf, const char *section)  
{  
  Data *p = cnf->data; /* 让变量p循环遍历data,找到对应section */  
  while (NULL != p && 0 != strcmp(p->section, section)) {  
    p = p->next;  
  }  
    
  if (NULL == p) { /* 没找到则不存在 */  
    return NULL;  
  }  
    
  return p;  
}  
  
/** 
* 判断指定option是否存在 
* 不存在返回空指针 
* 存在则返回包含那个section下key的Option指针 
**/  
Option *cnf_has_option(Config *cnf, const char *section, const char *key)  
{  
  Data *p = cnf_has_section(cnf, section);  
  if (NULL == p) { /* 没找到则不存在 */  
    return NULL;  
  }  
    
  Option *q = p->option;  
  while (NULL != q && 0 != strcmp(q->key, key)) {  
    q = q->next; /* 遍历option,检查key是否已经存在 */  
  }  
  if (NULL == q) { /* 没找到则不存在 */  
    return NULL;  
  }  
    
  return q;  
}  
  
/** 
* 将Config对象写入指定文件中 
* header表示在文件开头加一句注释 
* 写入成功则返回true 
**/  
bool cnf_write_file(Config *cnf, const char *filename, const char *header)  
{  
  FILE *fp = fopen(filename, "w");  
  if(NULL == fp) {  
    PRINT_ERRMSG("fopen");  
    exit(errno); /* 读文件错误直接按照错误码退出 */  
  }  
    
  if (0 < strlen(header)) { /* 文件注释不为空,则写注释到文件 */  
    fprintf(fp, "%c %s\n\n", cnf->comment, header);  
  }  
    
  Option *q;  
  Data   *p = cnf->data;  
  while (NULL != p) {  
    fprintf(fp, "[%s]\n", p->section);  
    q = p->option;  
    while (NULL != q) {  
      fprintf(fp, "%s %c %s\n", q->key, cnf->separator, q->value);  
      q = q->next;  
    }  
    p = p->next;  
  }  
    
  fclose(fp);  
  return true;  
}  
  
/** 
* 删除option 
**/  
bool cnf_remove_option(Config *cnf, const char *section, const char *key)  
{  
  Data *ps = cnf_has_section(cnf, section);  
  if (NULL == ps) { /* 没找到则不存在 */  
    return NULL;  
  }  
    
  Option *p, *q;  
  q = p = ps->option;  
  while (NULL != p && 0 != strcmp(p->key, key)) {  
    if (p != q) { q = q->next; } /* 始终让q处于p的上一个节点 */  
    p = p->next;  
  }  
    
  if (NULL == p) { /* 没找到则不存在 */  
    return NULL;  
  }  
    
  if (p == q) { /* 第一个option就匹配了 */  
    ps->option = p->next;  
  } else {  
    q->next = p->next;  
  }  
    
  free(p);  
  q = p = NULL; // 避免野指针  
    
  return true;  
}  
  
/** 
* 删除section 
**/  
bool cnf_remove_section(Config *cnf, const char *section)  
{  
  if (str_empty(section)) {  
    return false;  
  }  
    
  Data *p, *q;  
  q = p = cnf->data; /* 让变量p循环遍历data,找到对应section */  
  while (NULL != p && 0 != strcmp(p->section, section)) {  
    if (p != q) { q = q->next; } /* 始终让q处于p的上一个节点 */  
    p = p->next;  
  }  
    
  if (NULL == p) { /* 没有找到section */  
    return false;  
  }  
    
  if (p == q) { /* 这里表示第一个section,因此链表头位置改变 */  
    cnf->data = p->next;  
  } else { /* 此时是中间或尾部节点 */  
    q->next = p->next;  
  }  
    
  Option *o = p->option;  
  while (NULL != o) {  
    free(o); /* 循环释放所有option */  
    o = o->next;  
  }  
  p->option = NULL; // 避免野指针  
  free(p); /* 释放删除的section */  
  q = p = NULL;  // 避免野指针  
    
  return true;  
}  
  
/** 
* 打印当前Config对象 
**/  
void print_config(Config *cnf)  
{  
  Data *p = cnf->data; // 循环打印结果  
  while (NULL != p) {  
    printf("[%s]\n",p->section);  
      
    Option *q = p->option;  
    while (NULL != q) {  
      printf("  %s %c %s\n", q->key, cnf->separator, q->value);  
      q = q->next;  
    }  
    p = p->next;  
  }  
}  
  
/** 
* 主程序,放在最底下 
* 避免重复声明其他函数 
**/  
int main(int argc, char *argv[])  
{  
  // 读取配置文件cnf.ini,注释字符为#,分隔键值字符为=  
  Config *cnf = cnf_read_config("cnf.ini", '#', '=');  
  if (NULL == cnf) {  
    return -1; /* 创建对象失败 */  
  }  
    
  printf("-------------- After Read File --------------\n");  
  print_config(cnf); // 打印cnf对象    
  cnf_remove_section(cnf,"AAA"); // 删除AAA的section  
  cnf_remove_option(cnf, "CC","df");  // 删除CC下的df  
  printf("-------------- After remove --------------\n");  
  print_config(cnf); // 打印cnf对象    
  cnf_add_option(cnf, "NEW", "new_1", "true"); // 新增NEW下的new_1的值  
  cnf_add_option(cnf, "NEW", "new_2", "asdas"); // 新增NEW下的new_2的值  
  cnf_add_option(cnf, "NEW1", "new_2", "true");  
  printf("-------------- After add --------------\n");  
  print_config(cnf); // 打印cnf对象    
    
  cnf_get_value(cnf, "NEW1", "new_2"); // 获取NEW1下的new_2值  
  printf("cnf_get_value:%s,%d,%d,%f\n",cnf->re_string,cnf->re_int,cnf->re_bool,cnf->re_double);  
    
  cnf->separator = ':'; // 将分隔符改成 : ,冒号  
  cnf_write_file(cnf, "cnf_new.ini", "write a new ini file!"); // 将对象写入cnf_new.ini文件  
  return 0;  
}