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