stglibs/srvconf.lib/servconf.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  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
  19  */
  20
  21 #include "stg/servconf.h"
  22
  23 #include "stg/common.h"
  24
  25 #include <cstdio>
  26 #include <cstring>
  27
  28 namespace
  29 {
  30
  31 //-----------------------------------------------------------------------------
  32 void ElementStart(void *data, const char *el, const char **attr)
  33 {
  34 SERVCONF * sc = static_cast<SERVCONF *>(data);
  35 sc->Start(el, attr);
  36 }
  37 //-----------------------------------------------------------------------------
  38 void ElementEnd(void * data, const char * el)
  39 {
  40 SERVCONF * sc = static_cast<SERVCONF *>(data);
  41 sc->End(el);
  42 }
  43
  44 } // namespace anonymous
  45
  46 bool AnsRecv(void * data, const std::string & chunk, bool final)
  47 {
  48 SERVCONF * sc = static_cast<SERVCONF *>(data);
  49 printf("Chunk: '%s', length: %d, final: %d\n", chunk.c_str(), chunk.length(), final);
  50
  51 if (XML_Parse(sc->parser, chunk.c_str(), chunk.length(), final) == XML_STATUS_ERROR)
  52     {
  53     strprintf(&sc->errorMsg, "XML parse error at line %d: %s",
  54               static_cast<int>(XML_GetCurrentLineNumber(sc->parser)),
  55               XML_ErrorString(XML_GetErrorCode(sc->parser)));
  56     printf("%s\n", sc->errorMsg.c_str());
  57     return false;
  58     }
  59
  60 return true;
  61 }
  62
  63 //-----------------------------------------------------------------------------
  64 SERVCONF::SERVCONF(const std::string & server, uint16_t port,
  65                    const std::string & login, const std::string & password)
  66     : currParser(NULL),
  67       nt( server, port, login, password )
  68 {
  69 parser = XML_ParserCreate(NULL);
  70 nt.SetRxCallback(this, AnsRecv);
  71 }
  72 //-----------------------------------------------------------------------------
  73 int SERVCONF::GetUser(const char * l)
  74 {
  75 char request[255];
  76 snprintf(request, 255, "<GetUser login=\"%s\"/>", l);
  77
  78 currParser = &parserGetUser;
  79
  80 return Exec(request);
  81 }
  82 //-----------------------------------------------------------------------------
  83 int SERVCONF::AuthBy(const char * l)
  84 {
  85 char request[255];
  86 snprintf(request, 255, "<GetUserAuthBy login=\"%s\"/>", l);
  87
  88 currParser = &parserAuthBy;
  89
  90 return Exec(request);
  91 }
  92 //-----------------------------------------------------------------------------
  93 int SERVCONF::GetUsers()
  94 {
  95 char request[] = "<GetUsers/>";
  96
  97 currParser = &parserGetUsers;
  98
  99 return Exec(request);
 100 }
 101 //-----------------------------------------------------------------------------
 102 int SERVCONF::ServerInfo()
 103 {
 104 char request[] = "<GetServerInfo/>";
 105
 106 currParser = &parserServerInfo;
 107
 108 return Exec(request);
 109 }
 110 //-----------------------------------------------------------------------------
 111 int SERVCONF::ChgUser(const char * request)
 112 {
 113 currParser = &parserChgUser;
 114
 115 return Exec(request);
 116 }
 117 //-----------------------------------------------------------------------------
 118 int SERVCONF::SendMessage(const char * request)
 119 {
 120 currParser = &parserSendMessage;
 121
 122 return Exec(request);
 123 }
 124 //-----------------------------------------------------------------------------
 125 int SERVCONF::CheckUser(const char * login, const char * password)
 126 {
 127 char request[255];
 128 snprintf(request, 255, "<CheckUser login=\"%s\" password=\"%s\"/>", login, password);
 129
 130 currParser = &parserCheckUser;
 131
 132 return Exec(request);
 133 }
 134 //-----------------------------------------------------------------------------
 135 int SERVCONF::Start(const char * el, const char ** attr)
 136 {
 137 currParser->ParseStart(el, attr);
 138 return 0;
 139 }
 140 //-----------------------------------------------------------------------------
 141 void SERVCONF::End(const char * el)
 142 {
 143 currParser->ParseEnd(el);
 144 }
 145 //-----------------------------------------------------------------------------
 146 void SERVCONF::SetGetUsersCallback(PARSER_GET_USERS::CALLBACK f, void * data)
 147 {
 148 parserGetUsers.SetCallback(f, data);
 149 }
 150 //-----------------------------------------------------------------------------
 151 void SERVCONF::SetGetUserCallback(PARSER_GET_USER::CALLBACK f, void * data)
 152 {
 153 parserGetUser.SetCallback(f, data);
 154 }
 155 //-----------------------------------------------------------------------------
 156 void SERVCONF::SetAuthByCallback(PARSER_AUTH_BY::CALLBACK f, void * data)
 157 {
 158 parserAuthBy.SetCallback(f, data);
 159 }
 160 //-----------------------------------------------------------------------------
 161 void SERVCONF::SetServerInfoCallback(PARSER_SERVER_INFO::CALLBACK f, void * data)
 162 {
 163 parserServerInfo.SetCallback(f, data);
 164 }
 165 //-----------------------------------------------------------------------------
 166 void SERVCONF::SetChgUserCallback(PARSER_CHG_USER::CALLBACK f, void * data)
 167 {
 168 parserChgUser.SetCallback(f, data);
 169 }
 170 //-----------------------------------------------------------------------------
 171 void SERVCONF::SetCheckUserCallback(PARSER_CHECK_USER::CALLBACK f, void * data)
 172 {
 173 parserCheckUser.SetCallback(f, data);
 174 }
 175 //-----------------------------------------------------------------------------
 176 void SERVCONF::SetSendMessageCallback(PARSER_SEND_MESSAGE::CALLBACK f, void * data)
 177 {
 178 parserSendMessage.SetCallback(f, data);
 179 }
 180 //-----------------------------------------------------------------------------
 181 const std::string & SERVCONF::GetStrError() const
 182 {
 183 return errorMsg;
 184 }
 185 //-----------------------------------------------------------------------------
 186 int SERVCONF::Exec(const char * request)
 187 {
 188 XML_ParserReset(parser, NULL);
 189 XML_SetElementHandler(parser, ElementStart, ElementEnd);
 190 XML_SetUserData(parser, this);
 191
 192 int ret = 0;
 193 if ((ret = nt.Connect()) != st_ok)
 194     {
 195     errorMsg = nt.GetError();
 196     return ret;
 197     }
 198 if ((ret = nt.Transact(request)) != st_ok)
 199     {
 200     errorMsg = nt.GetError();
 201     return ret;
 202     }
 203 if ((ret = nt.Disconnect()) != st_ok)
 204     {
 205     errorMsg = nt.GetError();
 206     return ret;
 207     }
 208
 209 return st_ok;
 210 }