登 录
源自发布跨平台INI文件读写API(C++版本) V0.2.1
使用说明请参考该链接。
下面的版本对源码做了改进：
1. 修改了等号两边不能有空格或~t的问题
2. 增加了readBool方法
/**<br /> * @file iniFile.h<br /> * @brief initialization file read and write API<br /> * @author Deng Yangjun<br /> * @date 2007-12-9<br /> * @version 0.2.1<br /> * <pre><b>email: </b></pre><br /> * (C)2007 Midapex<br /> * This program is free software; you can redistribute it and/or modify it<br /> * under the terms of the GNU Library General Public License as published<br /> * by the Free Software Foundation; either version 2 of the License,<br /> * or (at your option) any later version.<br /> * <pre>Don Hao修改了等号两边不能有空格或~t的问题</b></pre><br /> * <pre>Don Hao增加了readBool方法</b></pre><br /> */<br /> #ifndef _INIFILECPP_H<br /> #define _INIFILECPP_H</p> <p>#include <string></p> <p>class IniFile<br /> {</p> <p>public:<br /> IniFile(const std::string & fileName);</p> <p>public:<br /> virtual ~IniFile(void);</p> <p> const std::string &getFileName() const;</p> <p> const std::string &getSection() const;</p> <p> void setSection(const std::string §ion);</p> <p> bool write(const std::string &key, const std::string & value) const ;<br /> bool write(const std::string &key, int value) const ;</p> <p> std::string readStr(const std::string &key, const std::string &default_value) const;<br /> int readInt(const std::string &key, int default_value) const;<br /> bool readBool(const std::string &key, bool default_value) const;</p> <p>public:<br /> static int read_profile_string(const char *section, const char *key, char *value,<br /> int size, const char *default_value, const char *file);</p> <p> static int read_profile_int(const char *section, const char *key, int default_value,<br /> const char *file);</p> <p> static bool read_profile_bool(const char *section, const char *key, bool default_value,<br /> const char *file);</p> <p> static int write_profile_string(const char *section, const char *key,<br /> const char *value, const char *file);</p> <p>private:<br /> static int load_ini_file(const char *file, char *buf, int *file_size);<br /> static int newline(char c);<br /> static int end_of_string(char c);<br /> static int left_barce(char c);<br /> static int right_brace(char c);<br /> static int parse_file(const char *section, const char *key, const char *buf, int *sec_s, int *sec_e,<br /> int *key_s, int *key_e, int *value_s, int *value_e);</p> <p>private:<br /> std::string m_fileName;<br /> std::string m_section;<br /> };</p> <p>#endif<br />
/**<br /> * @file iniFile.cpp<br /> * @brief initialization file read and write API implementation<br /> * @author Deng Yangjun<br /> * @date 2008 - 11 - 13<br /> * @version 0.2.1<br /> * <pre><b>email: </b></pre><br /> * (C)2007 Midapex<br /> * This program is free software; you can redistribute it and/or modify it<br /> * under the terms of the GNU Library General Public License as published<br /> * by the Free Software Foundation; either version 2 of the License,<br /> * or(at your option) any later version.<br /> * <pre><b>Moficication: </b></pre><br /> * <pre>Don Hao修改了等号两边不能有空格或~t的问题</b></pre><br /> * <pre>Don Hao增加了readBool方法</b></pre><br /> */<br /> #include "IniFile.h"<br /> #include <fstream><br /> #include <cassert><br /> #include <cstdio><br /> #include <cstdlib></p> <p>const int MAX_INI_FILE_SIZE = 1024 * 16;</p> <p>IniFile::IniFile(const std::string & fileName)<br /> {<br /> m_fileName = fileName;<br /> }</p> <p>IniFile::~IniFile(void)<br /> {<br /> }</p> <p>const std::string & IniFile::getFileName() const<br /> {<br /> return m_fileName;<br /> }</p> <p>void IniFile::setSection(const std::string §ion)<br /> {<br /> m_section = section;<br /> }</p> <p>bool IniFile::write(const std::string &key, const std::string & value) const<br /> {<br /> return write_profile_string(m_section.c_str(), key.c_str(), value.c_str(), m_fileName.c_str()) == 1 ? true : false;<br /> }</p> <p>bool IniFile::write(const std::string &key, int value) const<br /> {<br /> char tmp[64];<br /> _snprintf(tmp, strlen(tmp) - 1, "%d", value);<br /> return write(key, tmp);<br /> }</p> <p>std::string IniFile::readStr(const std::string &key, const std::string &default_value) const<br /> {<br /> char buf[4096];<br /> read_profile_string(m_section.c_str(), key.c_str(), buf, sizeof(buf), default_value.c_str(), m_fileName.c_str());<br /> return buf;<br /> }</p> <p>int IniFile::readInt(const std::string &key, int default_value) const<br /> {<br /> return read_profile_int(m_section.c_str(), key.c_str(), default_value, m_fileName.c_str());<br /> }</p> <p>bool IniFile::readBool(const std::string &key, bool default_value) const<br /> {<br /> return read_profile_bool(m_section.c_str(), key.c_str(), default_value, m_fileName.c_str());<br /> }</p> <p>const std::string &IniFile::getSection() const<br /> {<br /> return m_section;<br /> }</p> <p>int IniFile::load_ini_file(const char *file, char *buf, int *file_size)<br /> {<br /> FILE *in = NULL;<br /> int i = 0;<br /> *file_size = 0;</p> <p> assert(file != NULL);<br /> assert(buf != NULL);</p> <p> in = fopen(file, "r");</p> <p> if (NULL == in)<br /> {<br /> return 0;<br /> }</p> <p> buf[i] = fgetc(in);</p> <p> // load initialization file</p> <p> while (buf[i] != (char)EOF)<br /> {<br /> ++i;<br /> assert(i < MAX_INI_FILE_SIZE); // file too big, you can redefine MAX_INI_FILE_SIZE to fit the big file<br /> buf[i] = fgetc(in);<br /> }</p> <p> buf[i] = '/0';</p> <p> *file_size = i;</p> <p> fclose(in);<br /> return 1;<br /> }</p> <p>int IniFile::newline(char c)<br /> {<br /> return ('/n' == c || '/r' == c) ? 1 : 0;<br /> }</p> <p>int IniFile::end_of_string(char c)<br /> {<br /> return '/0' == c ? 1 : 0;<br /> }</p> <p>int IniFile::left_barce(char c)<br /> {<br /> return '[' == c ? 1 : 0;<br /> }</p> <p>int IniFile::right_brace(char c)<br /> {<br /> return ']' == c ? 1 : 0;<br /> }</p> <p>int IniFile::parse_file(const char *section, const char *key, const char *buf, int *sec_s, int *sec_e,<br /> int *key_s, int *key_e, int *value_s, int *value_e)<br /> {<br /> const char *p = buf;<br /> int i = 0;</p> <p> assert(buf != NULL);<br /> assert(section != NULL && strlen(section));<br /> assert(key != NULL && strlen(key));</p> <p> *sec_e = *sec_s = *key_e = *key_s = *value_s = *value_e = - 1;</p> <p> while (!end_of_string(p[i]))<br /> {<br /> // find the section<br /> if ((0 == i || newline(p[i - 1])) && left_barce(p[i]))<br /> {<br /> int section_start = i + 1;</p> <p> // find the ']'</p> <p> do<br /> {<br /> ++i;<br /> }<br /> while (!right_brace(p[i]) && !end_of_string(p[i]));<br /> //////////////////////////////////////////////////////////////////////////<br /> if (i - section_start == strlen(section) && 0 == strncmp(p + section_start, section, strlen(section)))<br /> {<br /> int newline_start = 0;</p> <p> ++i;</p> <p> // Skip over space char after ']'</p> <p> while (isspace(p[i]))<br /> {<br /> ++i;<br /> }</p> <p> // find the section<br /> *sec_s = section_start;</p> <p> *sec_e = i;</p> <p> while (!(newline(p[i - 1]) && left_barce(p[i]))<br /> && !end_of_string(p[i]))<br /> {<br /> int j = 0;<br /> // get a new line<br /> newline_start = i;</p> <p> while (!newline(p[i]) && !end_of_string(p[i]))<br /> {<br /> ++i;<br /> }</p> <p> // now i is equal to end of the line<br /> j = newline_start;<br /> int valid = j;</p> <p> if ('#' != p[j]) // skip over comment<br /> {<br /> while (j < i && p[j] != '=')<br /> {<br /> ++j;<br /> if (' ' != p[j] && '/t' != p[j] && '=' != p[j])<br /> valid = j;</p> <p> if ('=' == p[j])<br /> {<br /> if (j - newline_start == strlen(key) && strncmp(key, p + newline_start, strlen(key)) == 0)<br /> {<br /> // find the key ok<br /> *key_s = newline_start;<br /> *key_e = j - 1;</p> <p> valid = j + 1;<br /> while (' ' == p[valid] || '/t' == p[valid])<br /> valid++;<br /> *value_s = valid;<br /> *value_e = i;</p> <p> return 1;<br /> }<br /> }<br /> }<br /> }</p> <p> ++i;<br /> }<br /> }<br /> }<br /> else<br /> {<br /> ++i;<br /> }<br /> }</p> <p> return 0;<br /> }</p> <p>/**<br /> *@brief read string in initialization file/n<br /> * retrieves a string from the specified section in an initialization file<br /> *@param section[in] name of the section containing the key name<br /> *@param key[in] name of the key pairs to value<br /> *@param value[in] pointer to the buffer that receives the retrieved string<br /> *@param size[in] size of result's buffer<br /> *@param default_value[in] default value of result<br /> *@param file[in] path of the initialization file<br /> *@return 1 : read success; /n 0 : read fail<br /> */<br /> int IniFile::read_profile_string(const char *section, const char *key, char *value,<br /> int size, const char *default_value, const char *file)<br /> {<br /> char buf[MAX_INI_FILE_SIZE] = {0};<br /> int file_size;<br /> int sec_s, sec_e, key_s, key_e, value_s, value_e;</p> <p> // check parameters<br /> assert(section != NULL && strlen(section));<br /> assert(key != NULL && strlen(key));<br /> assert(value != NULL);<br /> assert(size > 0);<br /> assert(file != NULL && strlen(key));</p> <p> if (!load_ini_file(file, buf, &file_size))<br /> {<br /> if (default_value != NULL)<br /> {<br /> strncpy(value, default_value, size);<br /> }</p> <p> return 0;<br /> }</p> <p> if (!parse_file(section, key, buf, &sec_s, &sec_e, &key_s, &key_e, &value_s, &value_e))<br /> {<br /> if (default_value != NULL)<br /> {<br /> strncpy(value, default_value, size);<br /> }</p> <p> return 0; // not find the key<br /> }<br /> else<br /> {<br /> int cpcount = value_e - value_s;</p> <p> if (size - 1 < cpcount)<br /> {<br /> cpcount = size - 1;<br /> }</p> <p> memset(value, 0, size);</p> <p> memcpy(value, buf + value_s, cpcount);<br /> value[cpcount] = '/0';</p> <p> return 1;<br /> }<br /> }</p> <p>/**<br /> *@brief read int value in initialization file/n<br /> * retrieves int value from the specified section in an initialization file<br /> *@param section[in] name of the section containing the key name<br /> *@param key[in] name of the key pairs to value<br /> *@param default_value[in] default value of result<br /> *@param file[in] path of the initialization file<br /> *@return profile int value, if read fail, return default value<br /> */<br /> int IniFile::read_profile_int(const char *section, const char *key, int default_value,<br /> const char *file)<br /> {<br /> char value[32] = {0};</p> <p> if (!read_profile_string(section, key, value, sizeof(value), NULL, file))<br /> {<br /> return default_value;<br /> }<br /> else<br /> {<br /> return atoi(value);<br /> }<br /> }</p> <p>bool IniFile::read_profile_bool(const char *section, const char *key, bool default_value,<br /> const char *file)<br /> {<br /> char value[32] = {0};</p> <p> if (!read_profile_string(section, key, value, sizeof(value), NULL, file))<br /> {<br /> return default_value;<br /> }<br /> else<br /> {<br /> for(int i = 0; i < sizeof(value); ++i)<br /> value[i]=tolower(value[i]);<br /> std::string boolValue = static_cast<std::string>(value);<br /> if ("true" == boolValue)<br /> {<br /> return true;<br /> }<br /> else<br /> {<br /> return false;<br /> }<br /> }<br /> }</p> <p>/**<br /> * @brief write a profile string to a ini file<br /> * @param section[in] name of the section, can't be NULL and empty string<br /> * @param key[in] name of the key pairs to value, can't be NULL and empty string<br /> * @param value[in] profile string value<br /> * @param file[in] path of ini file<br /> * @return 1 : success/n 0 : failure<br /> */<br /> int IniFile::write_profile_string(const char *section, const char *key,<br /> const char *value, const char *file)<br /> {<br /> char buf[MAX_INI_FILE_SIZE] = {0};<br /> char w_buf[MAX_INI_FILE_SIZE] = {0};<br /> int sec_s, sec_e, key_s, key_e, value_s, value_e;<br /> int value_len = (int)strlen(value);<br /> int file_size;<br /> FILE *out;</p> <p> // check parameters<br /> assert(section != NULL && strlen(section));<br /> assert(key != NULL && strlen(key));<br /> assert(value != NULL);<br /> assert(file != NULL && strlen(key));</p> <p> if (!load_ini_file(file, buf, &file_size))<br /> {<br /> sec_s = - 1;<br /> }<br /> else<br /> {<br /> parse_file(section, key, buf, &sec_s, &sec_e, &key_s, &key_e, &value_s, &value_e);<br /> }</p> <p> if (-1 == sec_s)<br /> {<br /> if (0 == file_size)<br /> {<br /> sprintf(w_buf + file_size, "[%s]/n%s=%s/n", section, key, value);<br /> }<br /> else<br /> {<br /> // not find the section, then add the new section at end of the file<br /> memcpy(w_buf, buf, file_size);<br /> sprintf(w_buf + file_size, "/n[%s]/n%s=%s/n", section, key, value);<br /> }<br /> }<br /> else if (-1 == key_s)<br /> {<br /> // not find the key, then add the new key=value at end of the section<br /> memcpy(w_buf, buf, sec_e);<br /> sprintf(w_buf + sec_e, "%s=%s/n", key, value);<br /> sprintf(w_buf + sec_e + strlen(key) + strlen(value) + 2, buf + sec_e, file_size - sec_e);<br /> }<br /> else<br /> {<br /> // update value with new value<br /> memcpy(w_buf, buf, value_s);<br /> memcpy(w_buf + value_s, value, value_len);<br /> memcpy(w_buf + value_s + value_len, buf + value_e, file_size - value_e);<br /> }</p> <p> out = fopen(file, "w");</p> <p> if (NULL == out)<br /> {<br /> return 0;<br /> }</p> <p> if (-1 == fputs(w_buf, out))<br /> {<br /> fclose(out);<br /> return 0;<br /> }</p> <p> fclose(out);</p> <p> return 1;<br /> }
抱歉!评论已关闭.