]> git.stg.codes - stg.git/blob - stglibs/conffiles.lib/conffiles.cpp
Full month stats patches added (derived from GTS version)
[stg.git] / stglibs / conffiles.lib / conffiles.cpp
1 /*
2  *    This program is free software; you can redistribute it and/or modify
3  *    it under the terms of the GNU General Public License as published by
4  *    the Free Software Foundation; either version 2 of the License, or
5  *    (at your option) any later version.
6  *
7  *    This program is distributed in the hope that it will be useful,
8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  *    GNU General Public License for more details.
11  *
12  *    You should have received a copy of the GNU General Public License
13  *    along with this program; if not, write to the Free Software
14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
15  */
16
17 /*
18  *    Date: 27.10.2002
19  */
20
21 /*
22  *    Author : Boris Mikhailenko <stg34@ua.fm>
23  */
24
25  /*
26  $Revision: 1.5 $
27  $Date: 2009/10/22 11:40:22 $
28  */
29
30 //---------------------------------------------------------------------------
31
32 // getpid
33 #include <sys/types.h>
34 #include <unistd.h>
35
36 #include <cerrno> // E*
37 #include <cstring>
38 #include <cstdlib>
39 #include <cstdio>
40
41 #include <fstream>
42
43 #include "conffiles.h"
44 #include "stg/common.h"
45
46 using namespace std;
47
48 //---------------------------------------------------------------------------
49 bool StringCaseCmp(const string & str1, const string & str2)
50 {
51 return (strcasecmp(str1.c_str(), str2.c_str()) < 0);
52 }
53 //---------------------------------------------------------------------------
54 CONFIGFILE::CONFIGFILE(const string & fn, bool nook)
55     : param_val(StringCaseCmp),
56       fileName(fn),
57       error(0),
58       changed(false)
59 {
60 ifstream f(fileName.c_str());
61
62 if (!f)
63     {
64     if (!nook)
65         error = -1;
66     return;
67     }
68
69 string line;
70 while (getline(f, line))
71     {
72     size_t pos = line.find('#');
73     if (pos != string::npos)
74         line.resize(pos);
75
76     if (line.find_first_not_of(" \t\r") == string::npos)
77         continue;
78
79     pos = line.find_first_of('=');
80     if (pos == string::npos)
81         {
82         error = -1;
83         return;
84         }
85
86     string parameter = line.substr(0, pos);
87     string value = line.substr(pos + 1);
88     param_val[parameter] = value;
89     }
90 }
91 //---------------------------------------------------------------------------
92 CONFIGFILE::~CONFIGFILE()
93 {
94 Flush();
95 }
96 //---------------------------------------------------------------------------
97 const string & CONFIGFILE::GetFileName() const
98 {
99 return fileName;
100 }
101 //---------------------------------------------------------------------------
102 int CONFIGFILE::Error() const
103 {
104 int e = error;
105 error = 0;
106 return e;
107 }
108 /*//---------------------------------------------------------------------------
109 int CONFIGFILE::ReadString(const string & param, char * str, int * maxLen, const char * defaultVal) const
110 {
111 it = param_val.find(param);
112 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
113
114 if (it != param_val.end())
115     {
116     // þÔÏ-ÔÏ ÓÔÏÉÔ
117     strncpy(str, param_val[param].c_str(), *maxLen);
118     *maxLen = param_val[param].size();
119     return 0;
120     }
121
122 strncpy(str, defaultVal, *maxLen);
123 *maxLen = strlen(defaultVal);
124 return -1;
125 }*/
126 //---------------------------------------------------------------------------
127 int CONFIGFILE::ReadString(const string & param, string * val, const string & defaultVal) const
128 {
129 const map<string, string>::const_iterator it(param_val.find(param));
130 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
131
132 if (it != param_val.end())
133     {
134     // þÔÏ-ÔÏ ÓÔÏÉÔ
135     *val = it->second;
136     return 0;
137     }
138
139 *val = defaultVal;
140 return -1;
141 }
142 //---------------------------------------------------------------------------
143 void CONFIGFILE::WriteString(const string & param, const string &val)
144 {
145 param_val[param] = val;
146 changed = true;
147 }
148 //---------------------------------------------------------------------------
149 int CONFIGFILE::ReadTime(const string & param, time_t * val, time_t defaultVal) const
150 {
151 const map<string, string>::const_iterator it(param_val.find(param));
152
153 if (it != param_val.end())
154     {
155     char *res;
156     *val = strtol(it->second.c_str(), &res, 10);
157     if (*res != 0)
158         {
159         *val = defaultVal; //Error!
160         return EINVAL;
161         }
162     return 0;
163     }
164
165 *val = defaultVal;
166 return -1;
167 }
168 //---------------------------------------------------------------------------
169 int CONFIGFILE::ReadInt(const string & param, int * val, int defaultVal) const
170 {
171 const map<string, string>::const_iterator it(param_val.find(param));
172 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
173
174 if (it != param_val.end())
175     {
176     // þÔÏ-ÔÏ ÓÔÏÉÔ
177     char *res;
178     *val = strtol(it->second.c_str(), &res, 10);
179     if (*res != 0)
180         {
181         *val = defaultVal; //Error!
182         return EINVAL;
183         }
184     return 0;
185     }
186
187 *val = defaultVal;
188 return -1;
189 }
190 //---------------------------------------------------------------------------
191 int CONFIGFILE::ReadUInt(const string & param, unsigned int * val, unsigned int defaultVal) const
192 {
193 const map<string, string>::const_iterator it(param_val.find(param));
194 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
195
196 if (it != param_val.end())
197     {
198     // þÔÏ-ÔÏ ÓÔÏÉÔ
199     char *res;
200     *val = strtoul(it->second.c_str(), &res, 10);
201     if (*res != 0)
202         {
203         *val = defaultVal; //Error!
204         return EINVAL;
205         }
206     return 0;
207     }
208
209 *val = defaultVal;
210 return -1;
211 }
212 //---------------------------------------------------------------------------
213 int CONFIGFILE::ReadLongInt(const string & param, long int * val, long int defaultVal) const
214 {
215 const map<string, string>::const_iterator it(param_val.find(param));
216 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
217
218 if (it != param_val.end())
219     {
220     // þÔÏ-ÔÏ ÓÔÏÉÔ
221     char *res;
222     *val = strtol(it->second.c_str(), &res, 10);
223     if (*res != 0)
224         {
225         *val = defaultVal; //Error!
226         return EINVAL;
227         }
228     return 0;
229     }
230
231 *val = defaultVal;
232 return -1;
233 }
234 //---------------------------------------------------------------------------
235 int CONFIGFILE::ReadULongInt(const string & param, unsigned long int * val, unsigned long int defaultVal) const
236 {
237 const map<string, string>::const_iterator it(param_val.find(param));
238 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
239
240 if (it != param_val.end())
241     {
242     // þÔÏ-ÔÏ ÓÔÏÉÔ
243     char *res;
244     *val = strtoul(it->second.c_str(), &res, 10);
245     if (*res != 0)
246         {
247         *val = defaultVal; //Error!
248         return EINVAL;
249         }
250     return 0;
251     }
252
253 *val = defaultVal;
254 return -1;
255 }
256 //---------------------------------------------------------------------------
257 int CONFIGFILE::ReadLongLongInt(const string & param, int64_t * val, int64_t defaultVal) const
258 {
259 const map<string, string>::const_iterator it(param_val.find(param));
260 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
261
262 if (it != param_val.end())
263     {
264     // þÔÏ-ÔÏ ÓÔÏÉÔ
265     char *res;
266     *val = strtoll(it->second.c_str(), &res, 10);
267     if (*res != 0)
268         {
269         *val = defaultVal; //Error!
270         return EINVAL;
271         }
272     return 0;
273     }
274
275 *val = defaultVal;
276 return -1;
277 }
278 //---------------------------------------------------------------------------
279 int CONFIGFILE::ReadULongLongInt(const string & param, uint64_t * val, uint64_t defaultVal) const
280 {
281 const map<string, string>::const_iterator it(param_val.find(param));
282 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
283
284 if (it != param_val.end())
285     {
286     // þÔÏ-ÔÏ ÓÔÏÉÔ
287     char *res;
288     *val = strtoull(it->second.c_str(), &res, 10);
289     if (*res != 0)
290         {
291         *val = defaultVal; //Error!
292         return EINVAL;
293         }
294     return 0;
295     }
296
297 *val = defaultVal;
298 return -1;
299 }
300 //---------------------------------------------------------------------------
301 int CONFIGFILE::ReadShortInt(const string & param, short int * val, short int defaultVal) const
302 {
303 const map<string, string>::const_iterator it(param_val.find(param));
304 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
305
306 if (it != param_val.end())
307     {
308     // þÔÏ-ÔÏ ÓÔÏÉÔ
309     char *res;
310     *val = (short)strtol(it->second.c_str(), &res, 10);
311     if (*res != 0)
312         {
313         *val = defaultVal; //Error!
314         return EINVAL;
315         }
316     return 0;
317     }
318
319 *val = defaultVal;
320 return -1;
321 }
322 //---------------------------------------------------------------------------
323 int CONFIGFILE::ReadUShortInt(const string & param, unsigned short int * val, unsigned short int defaultVal) const
324 {
325 const map<string, string>::const_iterator it(param_val.find(param));
326 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
327
328 if (it != param_val.end())
329     {
330     // þÔÏ-ÔÏ ÓÔÏÉÔ
331     char *res;
332     *val = (short)strtoul(it->second.c_str(), &res, 10);
333     if (*res != 0)
334         {
335         *val = defaultVal; //Error!
336         return EINVAL;
337         }
338     return 0;
339     }
340
341 *val = defaultVal;
342 return -1;
343 }
344 //---------------------------------------------------------------------------
345 void CONFIGFILE::WriteInt(const string & param, int64_t val)
346 {
347 string s;
348 x2str(val, s);
349 param_val[param] = s;
350 changed = true;
351 }
352 //---------------------------------------------------------------------------
353 int CONFIGFILE::ReadDouble(const string & param, double * val, double defaultVal) const
354 {
355 const map<string, string>::const_iterator it(param_val.find(param));
356 // îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
357
358 if (it != param_val.end())
359     {
360     // þÔÏ-ÔÏ ÓÔÏÉÔ
361     char *res;
362     *val = strtod(it->second.c_str(), &res);
363     if (*res != 0)
364         {
365         *val = defaultVal; //Error!
366         return EINVAL;
367         }
368     return 0;
369     }
370
371 *val = defaultVal;
372 return -1;
373 }
374 //---------------------------------------------------------------------------
375 void CONFIGFILE::WriteDouble(const string & param, double val)
376 {
377 char s[30];
378 snprintf(s, 30, "%f", val);
379 param_val[param] = s;
380 changed = true;
381 }
382 //---------------------------------------------------------------------------
383 int CONFIGFILE::Flush(const std::string & path) const
384 {
385 ofstream f(path.c_str());
386 if (!f.is_open())
387     {
388     error = EIO;
389     return EIO;
390     }
391
392 map<string, string>::const_iterator it = param_val.begin();
393 while (it != param_val.end())
394     {
395     f << it->first << "=" << it->second << "\n";
396     ++it;
397     }
398
399 f.close();
400 return 0;
401 }
402 //---------------------------------------------------------------------------
403 int CONFIGFILE::Flush() const
404 {
405 if (!changed)
406     return 0;
407
408 std::string pid;
409 x2str(getpid(), pid);
410
411 if (Flush(fileName + "." + pid))
412     return -1;
413
414 if (rename((fileName + "." + pid).c_str(), fileName.c_str()))
415     return -1;
416
417 changed = false;
418
419 return 0;
420 }
421 //---------------------------------------------------------------------------