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