Context Navigation

source: protocols/jabber/jabber.c @ 934dddf3

Last change on this file since 934dddf3 was 89d736a, checked in by Wilmer van der Gaast <wilmer@…>, at 2008-06-22T19:21:06Z
From the department of over-engineering, now cached packet IDs are full MD5 hashes instead of a known MD5 hash with a number. Just to make it harder to confuse BitlBee by sending it faked responses to packets.
Property mode set to `100644`
File size: 15.3 KB

Line
1	/***************************************************************************\
2	* *
3	* BitlBee - An IRC to IM gateway *
4	* Jabber module - Main file *
5	* *
6	* Copyright 2006 Wilmer van der Gaast <wilmer@gaast.net> *
7	* *
8	* This program is free software; you can redistribute it and/or modify *
9	* it under the terms of the GNU General Public License as published by *
10	* the Free Software Foundation; either version 2 of the License, or *
11	* (at your option) any later version. *
12	* *
13	* This program is distributed in the hope that it will be useful, *
14	* but WITHOUT ANY WARRANTY; without even the implied warranty of *
15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
16	* GNU General Public License for more details. *
17	* *
18	* You should have received a copy of the GNU General Public License along *
19	* with this program; if not, write to the Free Software Foundation, Inc., *
20	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. *
21	* *
22	\***************************************************************************/
23
24	#include <glib.h>
25	#include <string.h>
26	#include <unistd.h>
27	#include <ctype.h>
28	#include <stdio.h>
29
30	#include "ssl_client.h"
31	#include "xmltree.h"
32	#include "bitlbee.h"
33	#include "jabber.h"
34	#include "md5.h"
35
36	GSList *jabber_connections;
37
38	/* First enty is the default */
39	static const int jabber_port_list[] = {
40	5222,
41	5223,
42	5220,
43	5221,
44	5224,
45	5225,
46	5226,
47	5227,
48	5228,
49	5229,
50	80,
51	443,
52	0
53	};
54
55	static void jabber_init( account_t *acc )
56	{
57	set_t *s;
58	char str[16];
59
60	g_snprintf( str, sizeof( str ), "%d", jabber_port_list[0] );
61	s = set_add( &acc->set, "port", str, set_eval_int, acc );
62	s->flags \|= ACC_SET_OFFLINE_ONLY;
63
64	s = set_add( &acc->set, "priority", "0", set_eval_priority, acc );
65
66	s = set_add( &acc->set, "resource", "BitlBee", NULL, acc );
67	s->flags \|= ACC_SET_OFFLINE_ONLY;
68
69	s = set_add( &acc->set, "resource_select", "priority", NULL, acc );
70
71	s = set_add( &acc->set, "server", NULL, set_eval_account, acc );
72	s->flags \|= ACC_SET_NOSAVE \| ACC_SET_OFFLINE_ONLY;
73
74	s = set_add( &acc->set, "ssl", "false", set_eval_bool, acc );
75	s->flags \|= ACC_SET_OFFLINE_ONLY;
76
77	s = set_add( &acc->set, "tls", "try", set_eval_tls, acc );
78	s->flags \|= ACC_SET_OFFLINE_ONLY;
79
80	s = set_add( &acc->set, "xmlconsole", "false", set_eval_bool, acc );
81	s->flags \|= ACC_SET_OFFLINE_ONLY;
82	}
83
84	static void jabber_generate_id_hash( struct jabber_data *jd );
85
86	static void jabber_login( account_t *acc )
87	{
88	struct im_connection *ic = imcb_new( acc );
89	struct jabber_data *jd = g_new0( struct jabber_data, 1 );
90	struct ns_srv_reply *srv = NULL;
91	char connect_to, s;
92	int i;
93
94	/* For now this is needed in the _connected() handlers if using
95	GLib event handling, to make sure we're not handling events
96	on dead connections. */
97	jabber_connections = g_slist_prepend( jabber_connections, ic );
98
99	jd->ic = ic;
100	ic->proto_data = jd;
101
102	jd->username = g_strdup( acc->user );
103	jd->server = strchr( jd->username, '@' );
104
105	jd->fd = jd->r_inpa = jd->w_inpa = -1;
106
107	if( jd->server == NULL )
108	{
109	imcb_error( ic, "Incomplete account name (format it like <username@jabberserver.name>)" );
110	imc_logout( ic, FALSE );
111	return;
112	}
113
114	/* So don't think of free()ing jd->server.. :-) */
115	*jd->server = 0;
116	jd->server ++;
117
118	if( ( s = strchr( jd->server, '/' ) ) )
119	{
120	*s = 0;
121	set_setstr( &acc->set, "resource", s + 1 );
122
123	/* Also remove the /resource from the original variable so we
124	won't have to do this again every time. */
125	s = strchr( acc->user, '/' );
126	*s = 0;
127	}
128
129	/* This code isn't really pretty. Backwards compatibility never is... */
130	s = acc->server;
131	while( s )
132	{
133	static int had_port = 0;
134
135	if( strncmp( s, "ssl", 3 ) == 0 )
136	{
137	set_setstr( &acc->set, "ssl", "true" );
138
139	/* Flush this part so that (if this was the first
140	part of the server string) acc->server gets
141	flushed. We don't want to have to do this another
142	time. :-) */
143	*s = 0;
144	s ++;
145
146	/* Only set this if the user didn't specify a custom
147	port number already... */
148	if( !had_port )
149	set_setint( &acc->set, "port", 5223 );
150	}
151	else if( isdigit( *s ) )
152	{
153	int i;
154
155	/* The first character is a digit. It could be an
156	IP address though. Only accept this as a port#
157	if there are only digits. */
158	for( i = 0; isdigit( s[i] ); i ++ );
159
160	/* If the first non-digit character is a colon or
161	the end of the string, save the port number
162	where it should be. */
163	if( s[i] == ':' \|\| s[i] == 0 )
164	{
165	sscanf( s, "%d", &i );
166	set_setint( &acc->set, "port", i );
167
168	/* See above. */
169	*s = 0;
170	s ++;
171	}
172
173	had_port = 1;
174	}
175
176	s = strchr( s, ':' );
177	if( s )
178	{
179	*s = 0;
180	s ++;
181	}
182	}
183
184	jd->node_cache = g_hash_table_new_full( g_str_hash, g_str_equal, NULL, jabber_cache_entry_free );
185	jd->buddies = g_hash_table_new( g_str_hash, g_str_equal );
186
187	/* Figure out the hostname to connect to. */
188	if( acc->server && *acc->server )
189	connect_to = acc->server;
190	else if( ( srv = srv_lookup( "xmpp-client", "tcp", jd->server ) ) \|\|
191	( srv = srv_lookup( "jabber-client", "tcp", jd->server ) ) )
192	connect_to = srv->name;
193	else
194	connect_to = jd->server;
195
196	imcb_log( ic, "Connecting" );
197
198	for( i = 0; jabber_port_list[i] > 0; i ++ )
199	if( set_getint( &acc->set, "port" ) == jabber_port_list[i] )
200	break;
201
202	if( jabber_port_list[i] == 0 )
203	{
204	imcb_log( ic, "Illegal port number" );
205	imc_logout( ic, FALSE );
206	return;
207	}
208
209	/* For non-SSL connections we can try to use the port # from the SRV
210	reply, but let's not do that when using SSL, SSL usually runs on
211	non-standard ports... */
212	if( set_getbool( &acc->set, "ssl" ) )
213	{
214	jd->ssl = ssl_connect( connect_to, set_getint( &acc->set, "port" ), jabber_connected_ssl, ic );
215	jd->fd = jd->ssl ? ssl_getfd( jd->ssl ) : -1;
216	}
217	else
218	{
219	jd->fd = proxy_connect( connect_to, srv ? srv->port : set_getint( &acc->set, "port" ), jabber_connected_plain, ic );
220	}
221	g_free( srv );
222
223	if( jd->fd == -1 )
224	{
225	imcb_error( ic, "Could not connect to server" );
226	imc_logout( ic, TRUE );
227
228	return;
229	}
230
231	if( set_getbool( &acc->set, "xmlconsole" ) )
232	{
233	jd->flags \|= JFLAG_XMLCONSOLE;
234	/* Shouldn't really do this at this stage already, maybe. But
235	I think this shouldn't break anything. */
236	imcb_add_buddy( ic, JABBER_XMLCONSOLE_HANDLE, NULL );
237	}
238
239	jabber_generate_id_hash( jd );
240	}
241
242	/* This generates an unfinished md5_state_t variable. Every time we generate
243	an ID, we finish the state by adding a sequence number and take the hash. */
244	static void jabber_generate_id_hash( struct jabber_data *jd )
245	{
246	md5_byte_t binbuf[4];
247	char *s;
248
249	md5_init( &jd->cached_id_prefix );
250	md5_append( &jd->cached_id_prefix, (unsigned char *) jd->username, strlen( jd->username ) );
251	md5_append( &jd->cached_id_prefix, (unsigned char *) jd->server, strlen( jd->server ) );
252	s = set_getstr( &jd->ic->acc->set, "resource" );
253	md5_append( &jd->cached_id_prefix, (unsigned char *) s, strlen( s ) );
254	random_bytes( binbuf, 4 );
255	md5_append( &jd->cached_id_prefix, binbuf, 4 );
256	}
257
258	static void jabber_logout( struct im_connection *ic )
259	{
260	struct jabber_data *jd = ic->proto_data;
261
262	if( jd->fd >= 0 )
263	jabber_end_stream( ic );
264
265	while( ic->groupchats )
266	jabber_chat_free( ic->groupchats );
267
268	if( jd->r_inpa >= 0 )
269	b_event_remove( jd->r_inpa );
270	if( jd->w_inpa >= 0 )
271	b_event_remove( jd->w_inpa );
272
273	if( jd->ssl )
274	ssl_disconnect( jd->ssl );
275	if( jd->fd >= 0 )
276	closesocket( jd->fd );
277
278	if( jd->tx_len )
279	g_free( jd->txq );
280
281	if( jd->node_cache )
282	g_hash_table_destroy( jd->node_cache );
283
284	xt_free( jd->xt );
285
286	g_free( jd->away_message );
287	g_free( jd->username );
288	g_free( jd );
289
290	jabber_connections = g_slist_remove( jabber_connections, ic );
291	}
292
293	static int jabber_buddy_msg( struct im_connection ic, char who, char *message, int flags )
294	{
295	struct jabber_data *jd = ic->proto_data;
296	struct jabber_buddy *bud;
297	struct xt_node *node;
298	char *s;
299	int st;
300
301	if( g_strcasecmp( who, JABBER_XMLCONSOLE_HANDLE ) == 0 )
302	return jabber_write( ic, message, strlen( message ) );
303
304	if( ( s = strchr( who, '=' ) ) && jabber_chat_by_jid( ic, s + 1 ) )
305	bud = jabber_buddy_by_ext_jid( ic, who, 0 );
306	else
307	bud = jabber_buddy_by_jid( ic, who, 0 );
308
309	node = xt_new_node( "body", message, NULL );
310	node = jabber_make_packet( "message", "chat", bud ? bud->full_jid : who, node );
311
312	if( bud && ( jd->flags & JFLAG_WANT_TYPING ) &&
313	( ( bud->flags & JBFLAG_DOES_XEP85 ) \|\|
314	!( bud->flags & JBFLAG_PROBED_XEP85 ) ) )
315	{
316	struct xt_node *act;
317
318	/* If the user likes typing notification and if we don't know
319	(and didn't probe before) if this resource supports XEP85,
320	include a probe in this packet now. Also, if we know this
321	buddy does support XEP85, we have to send this <active/>
322	tag to tell that the user stopped typing (well, that's what
323	we guess when s/he pressed Enter...). */
324	act = xt_new_node( "active", NULL, NULL );
325	xt_add_attr( act, "xmlns", XMLNS_CHATSTATES );
326	xt_add_child( node, act );
327
328	/* Just make sure we do this only once. */
329	bud->flags \|= JBFLAG_PROBED_XEP85;
330	}
331
332	st = jabber_write_packet( ic, node );
333	xt_free_node( node );
334
335	return st;
336	}
337
338	static GList jabber_away_states( struct im_connection ic )
339	{
340	static GList *l = NULL;
341	int i;
342
343	if( l == NULL )
344	for( i = 0; jabber_away_state_list[i].full_name; i ++ )
345	l = g_list_append( l, (void*) jabber_away_state_list[i].full_name );
346
347	return l;
348	}
349
350	static void jabber_get_info( struct im_connection ic, char who )
351	{
352	struct jabber_data *jd = ic->proto_data;
353	struct jabber_buddy *bud;
354
355	if( strchr( who, '/' ) )
356	bud = jabber_buddy_by_jid( ic, who, 0 );
357	else
358	{
359	char *s = jabber_normalize( who );
360	bud = g_hash_table_lookup( jd->buddies, s );
361	g_free( s );
362	}
363
364	while( bud )
365	{
366	imcb_log( ic, "Buddy %s (%d) information:\nAway state: %s\nAway message: %s",
367	bud->full_jid, bud->priority,
368	bud->away_state ? bud->away_state->full_name : "(none)",
369	bud->away_message ? : "(none)" );
370	bud = bud->next;
371	}
372
373	jabber_get_vcard( ic, bud ? bud->full_jid : who );
374	}
375
376	static void jabber_set_away( struct im_connection ic, char state_txt, char *message )
377	{
378	struct jabber_data *jd = ic->proto_data;
379	struct jabber_away_state *state;
380
381	/* Save all this info. We need it, for example, when changing the priority setting. */
382	state = (void *) jabber_away_state_by_name( state_txt );
383	jd->away_state = state ? state : (void ) jabber_away_state_list; / Fall back to "Away" if necessary. */
384	g_free( jd->away_message );
385	jd->away_message = ( message && *message ) ? g_strdup( message ) : NULL;
386
387	presence_send_update( ic );
388	}
389
390	static void jabber_add_buddy( struct im_connection ic, char who, char *group )
391	{
392	struct jabber_data *jd = ic->proto_data;
393
394	if( g_strcasecmp( who, JABBER_XMLCONSOLE_HANDLE ) == 0 )
395	{
396	jd->flags \|= JFLAG_XMLCONSOLE;
397	imcb_add_buddy( ic, JABBER_XMLCONSOLE_HANDLE, NULL );
398	return;
399	}
400
401	if( jabber_add_to_roster( ic, who, NULL ) )
402	presence_send_request( ic, who, "subscribe" );
403	}
404
405	static void jabber_remove_buddy( struct im_connection ic, char who, char *group )
406	{
407	struct jabber_data *jd = ic->proto_data;
408
409	if( g_strcasecmp( who, JABBER_XMLCONSOLE_HANDLE ) == 0 )
410	{
411	jd->flags &= ~JFLAG_XMLCONSOLE;
412	/* Not necessary for now. And for now the code isn't too
413	happy if the buddy is completely gone right after calling
414	this function already.
415	imcb_remove_buddy( ic, JABBER_XMLCONSOLE_HANDLE, NULL );
416	*/
417	return;
418	}
419
420	/* We should always do this part. Clean up our administration a little bit. */
421	jabber_buddy_remove_bare( ic, who );
422
423	if( jabber_remove_from_roster( ic, who ) )
424	presence_send_request( ic, who, "unsubscribe" );
425	}
426
427	static struct groupchat jabber_chat_join_( struct im_connection ic, char room, char nick, char *password )
428	{
429	if( strchr( room, '@' ) == NULL )
430	imcb_error( ic, "Invalid room name: %s", room );
431	else if( jabber_chat_by_jid( ic, room ) )
432	imcb_error( ic, "Already present in chat `%s'", room );
433	else
434	return jabber_chat_join( ic, room, nick, password );
435
436	return NULL;
437	}
438
439	static void jabber_chat_msg_( struct groupchat c, char message, int flags )
440	{
441	if( c && message )
442	jabber_chat_msg( c, message, flags );
443	}
444
445	static void jabber_chat_topic_( struct groupchat c, char topic )
446	{
447	if( c && topic )
448	jabber_chat_topic( c, topic );
449	}
450
451	static void jabber_chat_leave_( struct groupchat *c )
452	{
453	if( c )
454	jabber_chat_leave( c, NULL );
455	}
456
457	static void jabber_chat_invite_( struct groupchat c, char who, char *msg )
458	{
459	struct jabber_chat *jc = c->data;
460	gchar *msg_alt = NULL;
461
462	if( msg == NULL )
463	msg_alt = g_strdup_printf( "%s invited you to %s", c->ic->acc->user, jc->name );
464
465	if( c && who )
466	jabber_chat_invite( c, who, msg ? msg : msg_alt );
467
468	g_free( msg_alt );
469	}
470
471	static void jabber_keepalive( struct im_connection *ic )
472	{
473	/* Just any whitespace character is enough as a keepalive for XMPP sessions. */
474	jabber_write( ic, "\n", 1 );
475
476	/* This runs the garbage collection every minute, which means every packet
477	is in the cache for about a minute (which should be enough AFAIK). */
478	jabber_cache_clean( ic );
479	}
480
481	static int jabber_send_typing( struct im_connection ic, char who, int typing )
482	{
483	struct jabber_data *jd = ic->proto_data;
484	struct jabber_buddy *bud;
485
486	/* Enable typing notification related code from now. */
487	jd->flags \|= JFLAG_WANT_TYPING;
488
489	if( ( bud = jabber_buddy_by_jid( ic, who, 0 ) ) == NULL )
490	{
491	/* Sending typing notifications to unknown buddies is
492	unsupported for now. Shouldn't be a problem, I think. */
493	return 0;
494	}
495
496	if( bud->flags & JBFLAG_DOES_XEP85 )
497	{
498	/* We're only allowed to send this stuff if we know the other
499	side supports it. */
500
501	struct xt_node *node;
502	char *type;
503	int st;
504
505	if( typing & OPT_TYPING )
506	type = "composing";
507	else if( typing & OPT_THINKING )
508	type = "paused";
509	else
510	type = "active";
511
512	node = xt_new_node( type, NULL, NULL );
513	xt_add_attr( node, "xmlns", XMLNS_CHATSTATES );
514	node = jabber_make_packet( "message", "chat", bud->full_jid, node );
515
516	st = jabber_write_packet( ic, node );
517	xt_free_node( node );
518
519	return st;
520	}
521
522	return 1;
523	}
524
525	void jabber_initmodule()
526	{
527	struct prpl *ret = g_new0( struct prpl, 1 );
528
529	ret->name = "jabber";
530	ret->login = jabber_login;
531	ret->init = jabber_init;
532	ret->logout = jabber_logout;
533	ret->buddy_msg = jabber_buddy_msg;
534	ret->away_states = jabber_away_states;
535	ret->set_away = jabber_set_away;
536	// ret->set_info = jabber_set_info;
537	ret->get_info = jabber_get_info;
538	ret->add_buddy = jabber_add_buddy;
539	ret->remove_buddy = jabber_remove_buddy;
540	ret->chat_msg = jabber_chat_msg_;
541	ret->chat_topic = jabber_chat_topic_;
542	ret->chat_invite = jabber_chat_invite_;
543	ret->chat_leave = jabber_chat_leave_;
544	ret->chat_join = jabber_chat_join_;
545	ret->keepalive = jabber_keepalive;
546	ret->send_typing = jabber_send_typing;
547	ret->handle_cmp = g_strcasecmp;
548
549	register_protocol( ret );
550	}

Note: See TracBrowser for help on using the repository browser.

Download in other formats: