]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/configuration/sgconfig/configproto.cpp
Optimized sending answers in sgconfig.
[stg.git] / projects / stargazer / plugins / configuration / sgconfig / configproto.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@stargazer.dp.ua>
23  */
24
25  /*
26  $Revision: 1.22 $
27  $Date: 2010/10/04 20:24:14 $
28  $Author: faust $
29  */
30
31 #include "configproto.h"
32
33 #include "parser.h"
34 #include "parser_auth_by.h"
35 #include "parser_user_info.h"
36
37 #include "stg/users.h"
38 #include "stg/admins.h"
39 #include "stg/tariffs.h"
40 #include "stg/logger.h"
41
42 #include <unistd.h>
43
44 //-----------------------------------------------------------------------------
45 void ParseXMLStart(void *data, const char *el, const char **attr)
46 {
47 CONFIGPROTO * cp = static_cast<CONFIGPROTO *>(data);
48
49 if (cp->currParser)
50     {
51     cp->currParser->SetCurrAdmin(*cp->currAdmin);
52     cp->currParser->ParseStart(data, el, attr);
53     }
54 else
55     {
56     for (size_t i = 0; i < cp->dataParser.size(); i++)
57         {
58         cp->dataParser[i]->SetCurrAdmin(*cp->currAdmin);
59         cp->dataParser[i]->Reset();
60         if (cp->dataParser[i]->ParseStart(data, el, attr) == 0)
61             {
62             cp->currParser = cp->dataParser[i];
63             break;
64             }
65         else
66             {
67             cp->dataParser[i]->Reset();
68             }
69         }
70     }
71 }
72 //-----------------------------------------------------------------------------
73 void ParseXMLEnd(void *data, const char *el)
74 {
75 CONFIGPROTO * cp = static_cast<CONFIGPROTO *>(data);
76 if (cp->currParser)
77     {
78     if (cp->currParser->ParseEnd(data, el) == 0)
79         {
80         cp->currParser = NULL;
81         }
82     }
83 else
84     {
85     for (size_t i = 0; i < cp->dataParser.size(); i++)
86         {
87         if (cp->dataParser[i]->ParseEnd(data, el) == 0)
88             {
89             break;
90             }
91         }
92     }
93 }
94 //-----------------------------------------------------------------------------
95 CONFIGPROTO::CONFIGPROTO(PLUGIN_LOGGER & l)
96     : adminIP(0),
97       port(0),
98       nonstop(true),
99       state(0),
100       currAdmin(NULL),
101       logger(l),
102       listenSocket(-1),
103       admins(NULL),
104       currParser(NULL)
105 {
106 dataParser.push_back(new PARSER_GET_SERVER_INFO);
107
108 dataParser.push_back(new PARSER_GET_USERS);
109 dataParser.push_back(new PARSER_GET_USER);
110 dataParser.push_back(new PARSER_CHG_USER);
111 dataParser.push_back(new PARSER_ADD_USER);
112 dataParser.push_back(new PARSER_DEL_USER);
113 dataParser.push_back(new PARSER_CHECK_USER);
114 dataParser.push_back(new PARSER_SEND_MESSAGE);
115 dataParser.push_back(new PARSER_AUTH_BY);
116 dataParser.push_back(new PARSER_USER_INFO);
117
118 dataParser.push_back(new PARSER_GET_TARIFFS);
119 dataParser.push_back(new PARSER_ADD_TARIFF);
120 dataParser.push_back(new PARSER_DEL_TARIFF);
121 dataParser.push_back(new PARSER_CHG_TARIFF);
122
123 dataParser.push_back(new PARSER_GET_ADMINS);
124 dataParser.push_back(new PARSER_CHG_ADMIN);
125 dataParser.push_back(new PARSER_DEL_ADMIN);
126 dataParser.push_back(new PARSER_ADD_ADMIN);
127
128 xmlParser = XML_ParserCreate(NULL);
129
130 if (!xmlParser)
131     {
132     logger("Couldn't allocate memory for parser.");
133     exit(1);
134     }
135
136 }
137 //-----------------------------------------------------------------------------
138 CONFIGPROTO::~CONFIGPROTO()
139 {
140 for (size_t i = 0; i < dataParser.size(); ++i)
141     delete dataParser[i];
142 XML_ParserFree(xmlParser);
143 }
144 //-----------------------------------------------------------------------------
145 int CONFIGPROTO::ParseCommand()
146 {
147 std::list<std::string>::iterator n;
148 int done = 0;
149 char str[9];
150
151 if (requestList.empty())
152     return 0;
153
154 n = requestList.begin();
155
156 strncpy(str, (*n).c_str(), 8);
157 str[8] = 0;
158
159 XML_ParserReset(xmlParser, NULL);
160 XML_SetElementHandler(xmlParser, ParseXMLStart, ParseXMLEnd);
161 XML_SetUserData(xmlParser, this);
162
163 while(nonstop)
164     {
165     strncpy(str, (*n).c_str(), 8);
166     str[8] = 0;
167     size_t len = strlen(str);
168
169     ++n;
170     if (n == requestList.end())
171         done = 1;
172     --n;
173
174     if (XML_Parse(xmlParser, (*n).c_str(), static_cast<int>(len), done) == XML_STATUS_ERROR)
175         {
176         logger("Invalid configuration request");
177         printfd(__FILE__, "Parse error at line %d:\n%s\n",
178            XML_GetCurrentLineNumber(xmlParser),
179            XML_ErrorString(XML_GetErrorCode(xmlParser)));
180         if (currParser)
181             {
182             printfd(__FILE__, "Parser reset\n");
183             currParser->Reset();
184             currParser = NULL;
185             }
186
187         return -1;
188         }
189
190     if (done)
191         return 0;
192
193     ++n;
194     }
195
196 return 0;
197 }
198 //-----------------------------------------------------------------------------
199 void CONFIGPROTO::SetAdmins(ADMINS * a)
200 {
201 admins = a;
202 for (size_t i = 0; i < dataParser.size(); i++)
203     dataParser[i]->SetAdmins(a);
204 }
205 //-----------------------------------------------------------------------------
206 void CONFIGPROTO::SetUsers(USERS * u)
207 {
208 for (size_t i = 0; i < dataParser.size(); i++)
209     dataParser[i]->SetUsers(u);
210 }
211 //-----------------------------------------------------------------------------
212 void CONFIGPROTO::SetTariffs(TARIFFS * t)
213 {
214 for (size_t i = 0; i < dataParser.size(); i++)
215     dataParser[i]->SetTariffs(t);
216 }
217 //-----------------------------------------------------------------------------
218 void CONFIGPROTO::SetStore(STORE * s)
219 {
220 for (size_t i = 0; i < dataParser.size(); i++)
221     dataParser[i]->SetStore(s);
222 }
223 //-----------------------------------------------------------------------------
224 void CONFIGPROTO::SetStgSettings(const SETTINGS * s)
225 {
226 for (size_t i = 0; i < dataParser.size(); i++)
227     dataParser[i]->SetStgSettings(s);
228 }
229 //-----------------------------------------------------------------------------