Mercurial > hg
annotate mcabber/libjabber/jconn.c @ 398:ec82223ea4d5
Post-0.6.6 changes
author | Mikael Berthe <mikael@lilotux.net> |
---|---|
date | Thu, 18 Aug 2005 05:59:32 +0100 |
parents | bd5638c21834 |
children | 17aa60c6dc63 |
rev | line source |
---|---|
25 | 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 * Jabber | |
17 * Copyright (C) 1998-1999 The Jabber Team http://jabber.org/ | |
18 */ | |
19 | |
20 #include "jabber.h" | |
21 #include "connwrap.h" | |
22 | |
374
bd5638c21834
Improve logging system (traces)
Mikael Berthe <mikael@lilotux.net>
parents:
176
diff
changeset
|
23 #include "../src/logprint.h" /* For logging */ |
176 | 24 |
25 | 25 /* local macros for launching event handlers */ |
26 #define STATE_EVT(arg) if(j->on_state) { (j->on_state)(j, (arg) ); } | |
27 | |
28 /* prototypes of the local functions */ | |
29 static void startElement(void *userdata, const char *name, const char **attribs); | |
30 static void endElement(void *userdata, const char *name); | |
31 static void charData(void *userdata, const char *s, int slen); | |
32 | |
33 /* | |
34 * jab_new -- initialize a new jabber connection | |
35 * | |
36 * parameters | |
37 * user -- jabber id of the user | |
38 * pass -- password of the user | |
39 * | |
40 * results | |
41 * a pointer to the connection structure | |
42 * or NULL if allocations failed | |
43 */ | |
44 jconn jab_new(char *user, char *pass, int port, int ssl) | |
45 { | |
46 pool p; | |
47 jconn j; | |
48 | |
49 if(!user) return(NULL); | |
50 | |
51 p = pool_new(); | |
52 if(!p) return(NULL); | |
53 j = pmalloc_x(p, sizeof(jconn_struct), 0); | |
54 if(!j) return(NULL); | |
55 j->p = p; | |
56 | |
57 j->user = jid_new(p, user); | |
58 j->pass = pstrdup(p, pass); | |
59 j->port = port; | |
60 | |
61 j->state = JCONN_STATE_OFF; | |
62 j->cw_state = 0; | |
63 j->id = 1; | |
64 j->fd = -1; | |
65 j->ssl = ssl; | |
66 | |
67 return j; | |
68 } | |
69 | |
70 /* | |
71 * jab_delete -- free a jabber connection | |
72 * | |
73 * parameters | |
74 * j -- connection | |
75 * | |
76 */ | |
77 void jab_delete(jconn j) | |
78 { | |
79 if(!j) return; | |
80 | |
81 jab_stop(j); | |
82 pool_free(j->p); | |
83 } | |
84 | |
85 /* | |
86 * jab_state_handler -- set callback handler for state change | |
87 * | |
88 * parameters | |
89 * j -- connection | |
90 * h -- name of the handler function | |
91 */ | |
92 void jab_state_handler(jconn j, jconn_state_h h) | |
93 { | |
94 if(!j) return; | |
95 | |
96 j->on_state = h; | |
97 } | |
98 | |
99 /* | |
100 * jab_packet_handler -- set callback handler for incoming packets | |
101 * | |
102 * parameters | |
103 * j -- connection | |
104 * h -- name of the handler function | |
105 */ | |
106 void jab_packet_handler(jconn j, jconn_packet_h h) | |
107 { | |
108 if(!j) return; | |
109 | |
110 j->on_packet = h; | |
111 } | |
112 | |
113 void jab_logger(jconn j, jconn_logger h) | |
114 { | |
115 if(!j) return; | |
116 | |
117 j->logger = h; | |
118 } | |
119 | |
120 | |
121 /* | |
122 * jab_start -- start connection | |
123 * | |
124 * parameters | |
125 * j -- connection | |
126 * | |
127 */ | |
128 void jab_start(jconn j) | |
129 { | |
130 xmlnode x; | |
131 char *t,*t2; | |
132 | |
133 if(!j || (j->state != JCONN_STATE_OFF && j->state != JCONN_STATE_CONNECTING) ) return; | |
134 | |
135 if (!(j->cw_state & CW_CONNECT_WANT_SOMETHING)) { /* same as state != JCONN_STATE_CONNECTING */ | |
136 j->parser = XML_ParserCreate(NULL); | |
137 XML_SetUserData(j->parser, (void *)j); | |
138 XML_SetElementHandler(j->parser, startElement, endElement); | |
139 XML_SetCharacterDataHandler(j->parser, charData); | |
140 | |
141 if (j->cw_state & CW_CONNECT_BLOCKING) | |
142 j->fd = make_netsocket(j->port, j->user->server, NETSOCKET_CLIENT, j->ssl); | |
143 else | |
144 j->fd = make_nb_netsocket(j->port, j->user->server, NETSOCKET_CLIENT, j->ssl, &j->cw_state); | |
145 | |
146 if(j->fd < 0) { | |
147 STATE_EVT(JCONN_STATE_OFF); | |
148 return; | |
149 } | |
150 } | |
151 else { /* subsequent calls to cw_nb_connect until it finishes negociation */ | |
152 if (cw_nb_connect(j->fd, 0, 0, j->ssl, &j->cw_state)) { | |
153 STATE_EVT(JCONN_STATE_OFF); | |
154 return; | |
155 } | |
156 } | |
157 if (j->cw_state & CW_CONNECT_WANT_SOMETHING){ /* check if it finished negociation */ | |
158 j->state = JCONN_STATE_CONNECTING; | |
159 STATE_EVT(JCONN_STATE_CONNECTING); | |
160 return; | |
161 } | |
162 change_socket_to_blocking(j->fd); | |
163 | |
164 j->state = JCONN_STATE_CONNECTED; | |
165 STATE_EVT(JCONN_STATE_CONNECTED) | |
166 | |
167 /* start stream */ | |
168 x = jutil_header(NS_CLIENT, j->user->server); | |
169 t = xmlnode2str(x); | |
170 /* this is ugly, we can create the string here instead of jutil_header */ | |
171 /* what do you think about it? -madcat */ | |
172 t2 = strstr(t,"/>"); | |
173 *t2++ = '>'; | |
174 *t2 = '\0'; | |
175 jab_send_raw(j,"<?xml version='1.0'?>"); | |
176 jab_send_raw(j,t); | |
177 xmlnode_free(x); | |
178 | |
179 j->state = JCONN_STATE_ON; | |
180 STATE_EVT(JCONN_STATE_ON) | |
181 | |
182 } | |
183 | |
184 /* | |
185 * jab_stop -- stop connection | |
186 * | |
187 * parameters | |
188 * j -- connection | |
189 */ | |
190 void jab_stop(jconn j) | |
191 { | |
192 if(!j || j->state == JCONN_STATE_OFF) return; | |
193 | |
194 j->state = JCONN_STATE_OFF; | |
195 cw_close(j->fd); | |
196 j->fd = -1; | |
197 XML_ParserFree(j->parser); | |
198 } | |
199 | |
200 /* | |
201 * jab_getfd -- get file descriptor of connection socket | |
202 * | |
203 * parameters | |
204 * j -- connection | |
205 * | |
206 * returns | |
207 * fd of the socket or -1 if socket was not connected | |
208 */ | |
209 int jab_getfd(jconn j) | |
210 { | |
211 if(j) | |
212 return j->fd; | |
213 else | |
214 return -1; | |
215 } | |
216 | |
217 /* | |
218 * jab_getjid -- get jid structure of user | |
219 * | |
220 * parameters | |
221 * j -- connection | |
222 */ | |
223 jid jab_getjid(jconn j) | |
224 { | |
225 if(j) | |
226 return(j->user); | |
227 else | |
228 return NULL; | |
229 } | |
230 | |
231 /* jab_getsid -- get stream id | |
232 * This is the id of server's <stream:stream> tag and used for | |
233 * digest authorization. | |
234 * | |
235 * parameters | |
236 * j -- connection | |
237 */ | |
238 char *jab_getsid(jconn j) | |
239 { | |
240 if(j) | |
241 return(j->sid); | |
242 else | |
243 return NULL; | |
244 } | |
245 | |
246 /* | |
247 * jab_getid -- get a unique id | |
248 * | |
249 * parameters | |
250 * j -- connection | |
251 */ | |
252 char *jab_getid(jconn j) | |
253 { | |
254 snprintf(j->idbuf, 8, "%d", j->id++); | |
255 return &j->idbuf[0]; | |
256 } | |
257 | |
258 /* | |
259 * jab_send -- send xml data | |
260 * | |
261 * parameters | |
262 * j -- connection | |
263 * x -- xmlnode structure | |
264 */ | |
265 void jab_send(jconn j, xmlnode x) | |
266 { | |
267 if (j && j->state != JCONN_STATE_OFF) | |
268 { | |
269 char *buf = xmlnode2str(x); | |
270 if (buf) { | |
271 cw_write(j->fd, buf, strlen(buf), j->ssl); | |
272 if (j->logger) | |
273 (j->logger)(j, 0, buf); | |
274 } | |
275 | |
276 #ifdef JDEBUG | |
277 printf ("out: %s\n", buf); | |
278 #endif | |
279 } | |
280 } | |
281 | |
282 /* | |
283 * jab_send_raw -- send a string | |
284 * | |
285 * parameters | |
286 * j -- connection | |
287 * str -- xml string | |
288 */ | |
289 void jab_send_raw(jconn j, const char *str) | |
290 { | |
291 if (j && j->state != JCONN_STATE_OFF) { | |
292 cw_write(j->fd, str, strlen(str), j->ssl); | |
293 | |
294 if (j->logger) | |
295 (j->logger)(j, 0, str); | |
296 } | |
297 | |
298 #ifdef JDEBUG | |
299 printf ("out: %s\n", str); | |
300 #endif | |
301 } | |
302 | |
303 /* | |
304 * jab_recv -- read and parse incoming data | |
305 * | |
306 * parameters | |
307 * j -- connection | |
308 */ | |
309 void jab_recv(jconn j) | |
310 { | |
311 static char buf[32768]; | |
312 int len; | |
313 | |
314 if(!j || j->state == JCONN_STATE_OFF) | |
315 return; | |
316 | |
317 len = cw_read(j->fd, buf, sizeof(buf)-1, j->ssl); | |
318 if(len>0) | |
319 { | |
320 buf[len] = '\0'; | |
321 | |
322 if (j->logger) | |
323 (j->logger)(j, 1, buf); | |
324 | |
325 #ifdef JDEBUG | |
326 printf (" in: %s\n", buf); | |
327 #endif | |
328 XML_Parse(j->parser, buf, len, 0); | |
329 } | |
330 else if(len<=0) | |
331 { | |
332 STATE_EVT(JCONN_STATE_OFF); | |
333 jab_stop(j); | |
334 } | |
335 } | |
336 | |
337 /* | |
338 * jab_poll -- check socket for incoming data | |
339 * | |
340 * parameters | |
341 * j -- connection | |
342 * timeout -- poll timeout | |
343 */ | |
344 void jab_poll(jconn j, int timeout) | |
345 { | |
346 fd_set fds; | |
347 struct timeval tv; | |
348 int r; | |
349 | |
350 if (!j || j->state == JCONN_STATE_OFF) | |
351 return; | |
352 | |
353 FD_ZERO(&fds); | |
354 FD_SET(j->fd, &fds); | |
355 | |
356 if(timeout <= 0) { | |
357 r = select(j->fd + 1, &fds, NULL, NULL, NULL); | |
358 | |
359 } else { | |
360 tv.tv_sec = 0; | |
361 tv.tv_usec = timeout; | |
362 r = select(j->fd + 1, &fds, NULL, NULL, &tv); | |
363 | |
364 } | |
365 | |
366 if(r > 0) { | |
367 jab_recv(j); | |
368 | |
369 } else if(r) { | |
172 | 370 /* Don't disconnect for interrupted system call */ |
371 if(errno == EINTR) return; | |
171 | 372 |
374
bd5638c21834
Improve logging system (traces)
Mikael Berthe <mikael@lilotux.net>
parents:
176
diff
changeset
|
373 scr_LogPrint(LPRINT_LOGNORM, "jab_poll: select returned errno=%d", |
bd5638c21834
Improve logging system (traces)
Mikael Berthe <mikael@lilotux.net>
parents:
176
diff
changeset
|
374 errno); |
25 | 375 STATE_EVT(JCONN_STATE_OFF); |
376 jab_stop(j); | |
377 | |
378 } | |
379 } | |
380 | |
381 /* | |
382 * jab_auth -- authorize user | |
383 * | |
384 * parameters | |
385 * j -- connection | |
386 * | |
387 * returns | |
388 * id of the iq packet | |
389 */ | |
390 char *jab_auth(jconn j) | |
391 { | |
392 xmlnode x,y,z; | |
393 char *hash, *user, *id; | |
394 | |
395 if(!j) return(NULL); | |
396 | |
397 x = jutil_iqnew(JPACKET__SET, NS_AUTH); | |
398 id = jab_getid(j); | |
399 xmlnode_put_attrib(x, "id", id); | |
400 y = xmlnode_get_tag(x,"query"); | |
401 | |
402 user = j->user->user; | |
403 | |
404 if (user) | |
405 { | |
406 z = xmlnode_insert_tag(y, "username"); | |
407 xmlnode_insert_cdata(z, user, -1); | |
408 } | |
409 | |
410 z = xmlnode_insert_tag(y, "resource"); | |
411 xmlnode_insert_cdata(z, j->user->resource, -1); | |
412 | |
413 if (j->sid) | |
414 { | |
415 z = xmlnode_insert_tag(y, "digest"); | |
416 hash = pmalloc(x->p, strlen(j->sid)+strlen(j->pass)+1); | |
417 strcpy(hash, j->sid); | |
418 strcat(hash, j->pass); | |
419 hash = shahash(hash); | |
420 xmlnode_insert_cdata(z, hash, 40); | |
421 } | |
422 else | |
423 { | |
424 z = xmlnode_insert_tag(y, "password"); | |
425 xmlnode_insert_cdata(z, j->pass, -1); | |
426 } | |
427 | |
428 jab_send(j, x); | |
429 xmlnode_free(x); | |
430 return id; | |
431 } | |
432 | |
433 /* | |
434 * jab_reg -- register user | |
435 * | |
436 * parameters | |
437 * j -- connection | |
438 * | |
439 * returns | |
440 * id of the iq packet | |
441 */ | |
442 char *jab_reg(jconn j) | |
443 { | |
444 xmlnode x,y,z; | |
176 | 445 char *user, *id; |
25 | 446 |
447 if (!j) return(NULL); | |
448 | |
449 x = jutil_iqnew(JPACKET__SET, NS_REGISTER); | |
450 id = jab_getid(j); | |
451 xmlnode_put_attrib(x, "id", id); | |
452 y = xmlnode_get_tag(x,"query"); | |
453 | |
454 user = j->user->user; | |
455 | |
456 if (user) | |
457 { | |
458 z = xmlnode_insert_tag(y, "username"); | |
459 xmlnode_insert_cdata(z, user, -1); | |
460 } | |
461 | |
462 z = xmlnode_insert_tag(y, "resource"); | |
463 xmlnode_insert_cdata(z, j->user->resource, -1); | |
464 | |
465 if (j->pass) | |
466 { | |
467 z = xmlnode_insert_tag(y, "password"); | |
468 xmlnode_insert_cdata(z, j->pass, -1); | |
469 } | |
470 | |
471 jab_send(j, x); | |
472 xmlnode_free(x); | |
473 j->state = JCONN_STATE_ON; | |
474 STATE_EVT(JCONN_STATE_ON) | |
475 return id; | |
476 } | |
477 | |
478 | |
479 /* local functions */ | |
480 | |
481 static void startElement(void *userdata, const char *name, const char **attribs) | |
482 { | |
483 xmlnode x; | |
484 jconn j = (jconn)userdata; | |
485 | |
486 if(j->current) | |
487 { | |
488 /* Append the node to the current one */ | |
489 x = xmlnode_insert_tag(j->current, name); | |
490 xmlnode_put_expat_attribs(x, attribs); | |
491 | |
492 j->current = x; | |
493 } | |
494 else | |
495 { | |
496 x = xmlnode_new_tag(name); | |
497 xmlnode_put_expat_attribs(x, attribs); | |
498 if(strcmp(name, "stream:stream") == 0) { | |
499 /* special case: name == stream:stream */ | |
500 /* id attrib of stream is stored for digest auth */ | |
501 j->sid = xmlnode_get_attrib(x, "id"); | |
502 /* STATE_EVT(JCONN_STATE_AUTH) */ | |
503 } else { | |
504 j->current = x; | |
505 } | |
506 } | |
507 } | |
508 | |
509 static void endElement(void *userdata, const char *name) | |
510 { | |
511 jconn j = (jconn)userdata; | |
512 xmlnode x; | |
513 jpacket p; | |
514 | |
515 if(j->current == NULL) { | |
516 /* we got </stream:stream> */ | |
517 STATE_EVT(JCONN_STATE_OFF) | |
518 return; | |
519 } | |
520 | |
521 x = xmlnode_get_parent(j->current); | |
522 | |
523 if(x == NULL) | |
524 { | |
525 /* it is time to fire the event */ | |
526 p = jpacket_new(j->current); | |
527 | |
528 if(j->on_packet) | |
529 (j->on_packet)(j, p); | |
530 else | |
531 xmlnode_free(j->current); | |
532 } | |
533 | |
534 j->current = x; | |
535 } | |
536 | |
537 static void charData(void *userdata, const char *s, int slen) | |
538 { | |
539 jconn j = (jconn)userdata; | |
540 | |
541 if (j->current) | |
542 xmlnode_insert_cdata(j->current, s, slen); | |
543 } |