]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/other/userstat/datathread.cpp
Added encoding of some strings.
[stg.git] / projects / stargazer / plugins / other / userstat / datathread.cpp
1 #include "datathread.h"
2
3 DataThread::DataThread()
4         : tid(-1),
5           users(NULL),
6           store(NULL),
7           sock(-1),
8           done(true),
9           pvList(NULL),
10           data(NULL),
11           dataSize(0)
12 {
13 xmlParser = XML_ParserCreate(NULL);
14 if (!xmlParser)
15     {
16     printfd(__FILE__, "DataThread::DataThread() Failed to create parser\n");
17     return;
18     }
19 XML_SetElementHandler(xmlParser, DTXMLStart, DTXMLEnd);
20 }
21
22 DataThread::~DataThread()
23 {
24 XML_ParserFree(xmlParser);
25 }
26
27 bool DataThread::Handle(int s)
28 {
29 if (users == NULL)
30     {
31     printfd(__FILE__, "DataThread::Handle() Users not set\n");
32     return false;
33     }
34 if (store == NULL)
35     {
36     printfd(__FILE__, "DataThread::Handle() Storage not set\n");
37     return false;
38     }
39
40 sock = s;
41
42 if (pthread_create(&tid, NULL, Run, this))
43     {
44     printfd(__FILE__, "DataThread::Handle() Failed to create thread\n");
45     return false;
46     }
47 if (pthread_detach(tid))
48     {
49     printfd(__FILE__, "DataThread::Handle() Cannot detach the thread\n");
50     }
51 return true;
52 }
53
54 void * DataThread::Run(void * self)
55 {
56 DataThread * dt = reinterpret_cast<DataThread *>(self);
57
58 dt->done = false;
59
60 if (dt->ReadRequest())
61     {
62     if (dt->DecodeRequest())
63         {
64         if (dt->ParseRequest())
65             {
66             if (dt->MakeAnswer())
67                 {
68                 printfd(__FILE__, "DataThread::Run() All done\n");
69                 }
70             else
71                 {
72                 printfd(__FILE__, "DataThread::Run() Failed to answer the request\n");
73                 }
74             }
75         else
76             {
77             printfd(__FILE__, "DataThread::Run() Cannot parse the request\n");
78             }
79         }
80     else
81         {
82         printfd(__FILE__, "DataThread::Run() Cannot decode the request\n");
83         }
84     }
85 else
86     {
87     printfd(__FILE__, "DataThread::Run() Cannot read the request\n");
88     }
89
90 dt->Cleanup();
91
92 return NULL;
93 }
94
95 bool DataThread::ReadRequest()
96 {
97 int32_t size;
98 char * buf;
99
100 int res = read(sock, &size, sizeof(size));
101 if (res != sizeof(size))
102     {
103     printfd(__FILE__, "DataThread::ReadRequest() Reading login size failed! Wanted %d bytes, got %d bytes.\n", sizeof(size), res);
104     done = true;
105     return false;
106     }
107
108 if (size < 0)
109     {
110     printfd(__FILE__, "DataThread::ReadRequest() Invalid login size.\n");
111     done = true;
112     return false;
113     }
114
115 buf = new char[size];
116
117 res = read(sock, buf, size);
118 if (res != size)
119     {
120     printfd(__FILE__, "DataThread::ReadRequest() Reading login failed! Wanted %d bytes, got %d bytes.\n", size, res);
121     delete[] buf;
122     done = true;
123     return false;
124     }
125
126 login.assign(buf, size);
127 delete[] buf;
128
129 res = read(sock, &size, sizeof(size));
130 if (res != sizeof(size))
131     {
132     printfd(__FILE__, "DataThread::ReadRequest() Reading request size failed! Wanted %d bytes, got %d bytes.\n", sizeof(size), res);
133     done = true;
134     return false;
135     }
136
137 if (size < 0)
138     {
139     printfd(__FILE__, "DataThread::ReadRequest() Invalid request size.\n");
140     done = true;
141     return false;
142     }
143
144 data = new char[size + 1];
145 dataSize = size;
146
147 res = read(sock, data, size);
148 if (res != size)
149     {
150     printfd(__FILE__, "DataThread::ReadRequest() Reading request failed! Wanted %d bytes, got %d bytes.\n", size, res);
151     done = true;
152     return false;
153     }
154 data[res] = 0;
155
156 return true;
157 }
158
159 bool DataThread::DecodeRequest()
160 {
161 if (users->FindByName(login, &(uit)))
162     {
163     printfd(__FILE__, "DataThread::DecodeRequest() User '%s' not found.\n", login.c_str());
164     done = true;
165     return false;
166     }
167
168 std::string password = uit->property.password;
169
170 BLOWFISH_CTX ctx;
171 char * key = new char[password.length()];
172 strncpy(key, password.c_str(), password.length());
173
174 Blowfish_Init(&ctx,
175               reinterpret_cast<unsigned char *>(key),
176               password.length());
177
178 for (int i = 0; i < dataSize / 8; ++i)
179     {
180     uint32_t a;
181     uint32_t b;
182     a = n2l(reinterpret_cast<unsigned char *>(data + i * 8));
183     b = n2l(reinterpret_cast<unsigned char *>(data + i * 8 + 4));
184     Blowfish_Decrypt(&ctx,
185                      &a,
186                      &b);
187     l2n(a, reinterpret_cast<unsigned char *>(data + i * 8));
188     l2n(b, reinterpret_cast<unsigned char *>(data + i * 8 + 4));
189     }
190
191 delete[] key;
192
193 return true;
194 }
195
196 bool DataThread::ParseRequest()
197 {
198 if (XML_Parse(xmlParser, data, dataSize, 1) != XML_STATUS_OK)
199     {
200     printfd(__FILE__, "DataThread::ParseRequest() Failed to parse the request\n");
201     request.isBad = true;
202     return false;
203     }
204 return true;
205 }
206
207 bool DataThread::MakeAnswer()
208 {
209 if (MakeConf())
210     {
211     if (MakeStat())
212         {
213         if (SendAnswer())
214             {
215             // All is ok
216             }
217         else
218             {
219             printfd(__FILE__, "DataThread::MakeAnswer() Failed to send answer");
220             return false;
221             }
222         }
223     else
224         {
225         printfd(__FILE__, "DataThread::MakeAnswer() Failed to make stat answer\n");
226         return false;
227         }
228     }
229 else
230     {
231     printfd(__FILE__, "DataThread::MakeAnswer() Failed to make conf answer\n");
232     return false;
233     }
234
235 return true;
236 }
237
238 void DataThread::Cleanup()
239 {
240 delete[] data;
241 dataSize = 0;
242 login = "";
243 done = false;
244 data = NULL;
245 request.conf.erase(request.conf.begin(), request.conf.end());
246 request.stat.erase(request.stat.begin(), request.stat.end());
247 request.login = "";
248 request.isOk = true;
249 request.isBad = false;
250 }
251
252 void DataThread::ParseTag(const std::string & name, const char ** attr)
253 {
254 if (request.isBad)
255     return;
256 if (name == "request")
257     {
258     if (attr == NULL)
259         {
260         printfd(__FILE__, "DataThread::ParseTag() 'request' tag require an attribute\n");
261         request.isBad = true;
262         return;
263         }
264     else
265         {
266         std::string attrName(*attr++);
267         std::string attrValue(*attr++);
268         if (attr != NULL)
269             {
270             printfd(__FILE__, "DataThread::ParseTag() Extra attributes on tag 'request'\n");
271             }
272         if (attrName == "login")
273             {
274             if (attrValue != login)
275                 {
276                 printfd(__FILE__, "DataThread::ParseTag() Logins doesn't match\n");
277                 request.isBad = true;
278                 return;
279                 }
280             }
281         else
282             {
283             printfd(__FILE__, "DataThread::ParseTag() Unexpected attribute '%s'\n", attrName.c_str());
284             request.isBad = true;
285             return;
286             }
287         }
288     }
289 else if (name == "conf")
290     {
291     pvList = &(request.conf);
292     }
293 else if (name == "stat")
294     {
295     pvList = &(request.stat);
296     }
297 else
298     {
299     (*pvList)[name];
300     }
301 }
302
303 bool DataThread::MakeConf()
304 {
305 return false;
306 }
307
308 bool DataThread::MakeStat()
309 {
310 return false;
311 }
312
313 bool DataThread::SendAnswer()
314 {
315 return false;
316 }
317
318 void DTXMLStart(void * data, const char * name, const char ** attr)
319 {
320 DataThread * dt = reinterpret_cast<DataThread *>(data);
321 dt->ParseTag(name, attr);
322 }
323
324 void DTXMLEnd(void * data, const char * name)
325 {
326 //DataThread * dt = reinterpret_cast<DataThread *>(data);
327 }