Check End-Of-Transmission read result

[stg.git] / stglibs / conffiles.lib / conffiles.cpp

diff --git a/stglibs/conffiles.lib/conffiles.cpp b/stglibs/conffiles.lib/conffiles.cpp
index 549cb22d4e6d6533cdc7af55a9eaa0f1128e41ce..263c9933daa156add001ef7b611927fe7a5b9938 100644 (file)
--- a/stglibs/conffiles.lib/conffiles.cpp
+++ b/stglibs/conffiles.lib/conffiles.cpp
@@ -28,13 +28,19 @@
  */
 
 //---------------------------------------------------------------------------
  */
 
 //---------------------------------------------------------------------------

-#include <errno.h>
-#include <string.h>
-#include <stdlib.h>
+
+// getpid
+#include <sys/types.h>
+#include <unistd.h>
+
+#include <cerrno> // E*
+#include <cstring>
+#include <cstdlib>
+#include <cstdio>
+

 #include <fstream>
 #include <fstream>

-#include <algorithm>
-#include "conffiles.h"
-#include "common.h"
+
+#include "stg/conffiles.h"

 
 using namespace std;
 
 
 using namespace std;
 


@@ -44,81 +50,47 @@ bool StringCaseCmp(const string & str1, const string & str2)
 return (strcasecmp(str1.c_str(), str2.c_str()) < 0);
 }
 //---------------------------------------------------------------------------
 return (strcasecmp(str1.c_str(), str2.c_str()) < 0);
 }
 //---------------------------------------------------------------------------

-CONFIGFILE::CONFIGFILE(const string &fn):
-param_val(StringCaseCmp)
+CONFIGFILE::CONFIGFILE(const string & fn, bool nook)
+    : param_val(StringCaseCmp),
+      fileName(fn),
+      error(0),
+      changed(false)

 {
 {

-fileName = fn;
-f = fopen(fn.c_str(), "rt");
-
-error = 0;
-param_val.clear();
+ifstream f(fileName.c_str());

 
 if (!f)
     {
 
 if (!f)
     {

-    error = -1;
+    if (!nook)
+        error = -1;

     return;
     }
 
     return;
     }
 

-string line, parameter, value;
-
-unsigned long pos;
-bool emptyLine;
-unsigned char c;
-
-while (!feof(f))
+string line;
+while (getline(f, line))

     {
     {

-    line.erase(line.begin(), line.end());
-
-    c = fgetc(f);
-    while (!feof(f))
-        {
-        //printf("%c", c);
-        if (c == '\n')
-            break;
-        line.push_back(c);
-        c = fgetc(f);
-        }
-
-    pos = line.find('#');
+    size_t pos = line.find('#');

     if (pos != string::npos)
         line.resize(pos);
 
     if (pos != string::npos)
         line.resize(pos);
 

-    emptyLine = true;
-    for (unsigned int i = 0; i < line.size(); i++)
-        {
-        if (line[i] != ' ' && line[i] != '\t' && line[i] != '\n' && line[i] != '\r')
-            {
-            emptyLine = false;
-            break;
-            }
-        }
-    if (emptyLine)
-        {
+    if (line.find_first_not_of(" \t\r") == string::npos)

         continue;
         continue;

-        }

 
 

-    pos = line.find("=");
+    pos = line.find_first_of('=');

     if (pos == string::npos)
         {
     if (pos == string::npos)
         {

-        fclose(f);

         error = -1;
         error = -1;

-        //printf("%s find(=) error\n", __FILE__);

         return;
         }
         return;
         }

-    parameter = line.substr(0, pos);
-    //transform(parameter.begin(), parameter.end(), parameter.begin(), tolower);
-    value = line.substr(pos + 1);
-    //cout << parameter << "==" << value << endl;
+
+    string parameter = line.substr(0, pos);
+    string value = line.substr(pos + 1);

     param_val[parameter] = value;
     param_val[parameter] = value;

-    //cout << parameter << "==" << param_val[parameter] << endl;

     }
     }

-
-fclose(f);

 }
 //---------------------------------------------------------------------------
 CONFIGFILE::~CONFIGFILE()
 {
 }
 //---------------------------------------------------------------------------
 CONFIGFILE::~CONFIGFILE()
 {

-
+Flush();

 }
 //---------------------------------------------------------------------------
 const string & CONFIGFILE::GetFileName() const
 }
 //---------------------------------------------------------------------------
 const string & CONFIGFILE::GetFileName() const


@@ -126,71 +98,20 @@ const string & CONFIGFILE::GetFileName() const
 return fileName;
 }
 //---------------------------------------------------------------------------
 return fileName;
 }
 //---------------------------------------------------------------------------

-int CONFIGFILE::Error()
+int CONFIGFILE::Error() const

 {
 int e = error;
 error = 0;
 return e;
 }
 //---------------------------------------------------------------------------
 {
 int e = error;
 error = 0;
 return e;
 }
 //---------------------------------------------------------------------------

-int CONFIGFILE::FindParameter(const string &parameter, string * value) const
-{
-it = param_val.find(parameter);
-if (it == param_val.end())
-    return -1;
-
-*value = param_val[parameter];
-return 0;
-}
-//---------------------------------------------------------------------------
-int CONFIGFILE::Flush()
-{
-fstream f(fileName.c_str(), ios::out);
-if (!f.is_open())
-    {
-    error = EIO;
-    return EIO;
-    }
-
-it = param_val.begin();
-while (it != param_val.end())
-    {
-    f << it->first << "=" << it->second << endl;
-    it++;
-    }
-
-f.close();
-
-return 0;
-}
-//---------------------------------------------------------------------------
-int CONFIGFILE::ReadString(const string & param, char * str, int * maxLen, const char * defaultVal) const
-{
-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
-
-if (it != param_val.end())
-    {
-    // þÔÏ-ÔÏ ÓÔÏÉÔ
-    strncpy(str, param_val[param].c_str(), *maxLen);
-    *maxLen = param_val[param].size();
-    return 0;
-    }
-
-strncpy(str, defaultVal, *maxLen);
-*maxLen = strlen(defaultVal);
-return -1;
-}
-//---------------------------------------------------------------------------

 int CONFIGFILE::ReadString(const string & param, string * val, const string & defaultVal) const
 {
 int CONFIGFILE::ReadString(const string & param, string * val, const string & defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ
-    *val = param_val[param];
+    *val = it->second;

     return 0;
     }
 
     return 0;
     }
 


@@ -198,27 +119,20 @@ if (it != param_val.end())
 return -1;
 }
 //---------------------------------------------------------------------------
 return -1;
 }
 //---------------------------------------------------------------------------

-int CONFIGFILE::WriteString(const string & param, const char * val)
-{
-WriteString(param, string(val));
-return 0;
-}
-//---------------------------------------------------------------------------
-int CONFIGFILE::WriteString(const string & param, const string &val)
+void CONFIGFILE::WriteString(const string & param, const string &val)

 {
 param_val[param] = val;
 {
 param_val[param] = val;

-Flush();
-return 0;
+changed = true;

 }
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadTime(const string & param, time_t * val, time_t defaultVal) const
 {
 }
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadTime(const string & param, time_t * val, time_t defaultVal) const
 {

-it = param_val.find(param);
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
     char *res;
 
 if (it != param_val.end())
     {
     char *res;

-    *val = strtol(param_val[param].c_str(), &res, 10);
+    *val = strtol(it->second.c_str(), &res, 10);

     if (*res != 0)
         {
         *val = defaultVal; //Error!
     if (*res != 0)
         {
         *val = defaultVal; //Error!


@@ -233,14 +147,12 @@ return -1;
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadInt(const string & param, int * val, int defaultVal) const
 {
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadInt(const string & param, int * val, int defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ

     char *res;
     char *res;

-    *val = strtol(param_val[param].c_str(), &res, 10);
+    *val = strtol(it->second.c_str(), &res, 10);

     if (*res != 0)
         {
         *val = defaultVal; //Error!
     if (*res != 0)
         {
         *val = defaultVal; //Error!


@@ -255,14 +167,12 @@ return -1;
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadUInt(const string & param, unsigned int * val, unsigned int defaultVal) const
 {
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadUInt(const string & param, unsigned int * val, unsigned int defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ

     char *res;
     char *res;

-    *val = strtoul(param_val[param].c_str(), &res, 10);
+    *val = strtoul(it->second.c_str(), &res, 10);

     if (*res != 0)
         {
         *val = defaultVal; //Error!
     if (*res != 0)
         {
         *val = defaultVal; //Error!


@@ -277,14 +187,12 @@ return -1;
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadLongInt(const string & param, long int * val, long int defaultVal) const
 {
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadLongInt(const string & param, long int * val, long int defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ

     char *res;
     char *res;

-    *val = strtol(param_val[param].c_str(), &res, 10);
+    *val = strtol(it->second.c_str(), &res, 10);

     if (*res != 0)
         {
         *val = defaultVal; //Error!
     if (*res != 0)
         {
         *val = defaultVal; //Error!


@@ -299,14 +207,12 @@ return -1;
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadULongInt(const string & param, unsigned long int * val, unsigned long int defaultVal) const
 {
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadULongInt(const string & param, unsigned long int * val, unsigned long int defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ

     char *res;
     char *res;

-    *val = strtoul(param_val[param].c_str(), &res, 10);
+    *val = strtoul(it->second.c_str(), &res, 10);

     if (*res != 0)
         {
         *val = defaultVal; //Error!
     if (*res != 0)
         {
         *val = defaultVal; //Error!


@@ -321,14 +227,12 @@ return -1;
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadLongLongInt(const string & param, int64_t * val, int64_t defaultVal) const
 {
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadLongLongInt(const string & param, int64_t * val, int64_t defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ

     char *res;
     char *res;

-    *val = strtoll(param_val[param].c_str(), &res, 10);
+    *val = strtoll(it->second.c_str(), &res, 10);

     if (*res != 0)
         {
         *val = defaultVal; //Error!
     if (*res != 0)
         {
         *val = defaultVal; //Error!


@@ -343,14 +247,12 @@ return -1;
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadULongLongInt(const string & param, uint64_t * val, uint64_t defaultVal) const
 {
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadULongLongInt(const string & param, uint64_t * val, uint64_t defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ

     char *res;
     char *res;

-    *val = strtoull(param_val[param].c_str(), &res, 10);
+    *val = strtoull(it->second.c_str(), &res, 10);

     if (*res != 0)
         {
         *val = defaultVal; //Error!
     if (*res != 0)
         {
         *val = defaultVal; //Error!


@@ -365,14 +267,12 @@ return -1;
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadShortInt(const string & param, short int * val, short int defaultVal) const
 {
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadShortInt(const string & param, short int * val, short int defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ

     char *res;
     char *res;

-    *val = (short)strtol(param_val[param].c_str(), &res, 10);
+    *val = (short)strtol(it->second.c_str(), &res, 10);

     if (*res != 0)
         {
         *val = defaultVal; //Error!
     if (*res != 0)
         {
         *val = defaultVal; //Error!


@@ -387,14 +287,12 @@ return -1;
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadUShortInt(const string & param, unsigned short int * val, unsigned short int defaultVal) const
 {
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadUShortInt(const string & param, unsigned short int * val, unsigned short int defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ

     char *res;
     char *res;

-    *val = (short)strtoul(param_val[param].c_str(), &res, 10);
+    *val = (short)strtoul(it->second.c_str(), &res, 10);

     if (*res != 0)
         {
         *val = defaultVal; //Error!
     if (*res != 0)
         {
         *val = defaultVal; //Error!


@@ -407,49 +305,78 @@ if (it != param_val.end())
 return -1;
 }
 //---------------------------------------------------------------------------
 return -1;
 }
 //---------------------------------------------------------------------------

-int CONFIGFILE::WriteInt(const string & param, int64_t val)
+void CONFIGFILE::WriteInt(const string & param, int64_t val)

 {
 {

-string s;
-//sprintf(s, "%lld", val);
-x2str(val, s);
-param_val[param] = s;
-Flush();
-return 0;
+char buf[32];
+snprintf(buf, sizeof(buf), "%lld", static_cast<long long int>(val));
+param_val[param] = buf;
+changed = true;

 }
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadDouble(const string & param, double * val, double defaultVal) const
 {
 }
 //---------------------------------------------------------------------------
 int CONFIGFILE::ReadDouble(const string & param, double * val, double defaultVal) const
 {

-it = param_val.find(param);
-// îÁÛÌÉ ÎÕÖÎÕÀ ÐÅÒÅÍÅÎÎÕÀ
+const map<string, string>::const_iterator it(param_val.find(param));

 
 if (it != param_val.end())
     {
 
 if (it != param_val.end())
     {

-    // þÔÏ-ÔÏ ÓÔÏÉÔ

     char *res;
     char *res;

-    *val = strtod(param_val[param].c_str(), &res);
+    *val = strtod(it->second.c_str(), &res);

     if (*res != 0)
         {
     if (*res != 0)
         {

-        //cout << param << "=" << param_val[param] << " Error!!!\n";

         *val = defaultVal; //Error!
         return EINVAL;
         }
     return 0;
     }
 
         *val = defaultVal; //Error!
         return EINVAL;
         }
     return 0;
     }
 

-//cout << "îÉÞÅÇÏ ÎÅÔ!!!\n";
-

 *val = defaultVal;
 return -1;
 }
 //---------------------------------------------------------------------------
 *val = defaultVal;
 return -1;
 }
 //---------------------------------------------------------------------------

-int CONFIGFILE::WriteDouble(const string & param, double val)
+void CONFIGFILE::WriteDouble(const string & param, double val)

 {
 char s[30];
 {
 char s[30];

-sprintf(s, "%f", val);
+snprintf(s, 30, "%f", val);

 param_val[param] = s;
 param_val[param] = s;

-Flush();
+changed = true;
+}
+//---------------------------------------------------------------------------
+int CONFIGFILE::Flush(const std::string & path) const
+{
+ofstream f(path.c_str());
+if (!f.is_open())
+    {
+    error = EIO;
+    return EIO;
+    }
+
+map<string, string>::const_iterator it = param_val.begin();
+while (it != param_val.end())
+    {
+    f << it->first << "=" << it->second << "\n";
+    ++it;
+    }
+
+f.close();

 return 0;
 }
 //---------------------------------------------------------------------------
 return 0;
 }
 //---------------------------------------------------------------------------

+int CONFIGFILE::Flush() const
+{
+if (!changed)
+    return 0;

 
 

+char pid[6];
+snprintf(pid, sizeof(pid), "%d", getpid());

 
 

+if (Flush(fileName + "." + pid))
+    return -1;
+
+if (rename((fileName + "." + pid).c_str(), fileName.c_str()))
+    return -1;
+
+changed = false;
+
+return 0;
+}
+//---------------------------------------------------------------------------