Context Navigation

source: protocols/jabber/jabber.c @ 8d96b78

Last change on this file since 8d96b78 was 34ded90, checked in by Wilmer van der Gaast <wilmer@…>, at 2011-12-24T18:09:05Z
Slight cleanup: Use a constant instead of just "jabber_oauth" everywhere, and added some safeguards to keep the user from messaging it when we're not actually doing OAuth setup.
Property mode set to `100644`
File size: 18.1 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 "oauth.h"
35	#include "md5.h"
36
37	GSList *jabber_connections;
38
39	/* First enty is the default */
40	static const int jabber_port_list[] = {
41	5222,
42	5223,
43	5220,
44	5221,
45	5224,
46	5225,
47	5226,
48	5227,
49	5228,
50	5229,
51	80,
52	443,
53	0
54	};
55
56	static void jabber_init( account_t *acc )
57	{
58	set_t *s;
59	char str[16];
60
61	s = set_add( &acc->set, "activity_timeout", "600", set_eval_int, acc );
62
63	s = set_add( &acc->set, "oauth", "false", set_eval_oauth, acc );
64
65	g_snprintf( str, sizeof( str ), "%d", jabber_port_list[0] );
66	s = set_add( &acc->set, "port", str, set_eval_int, acc );
67	s->flags \|= ACC_SET_OFFLINE_ONLY;
68
69	s = set_add( &acc->set, "priority", "0", set_eval_priority, acc );
70
71	s = set_add( &acc->set, "proxy", "<local>;<auto>", NULL, acc );
72
73	s = set_add( &acc->set, "resource", "BitlBee", NULL, acc );
74	s->flags \|= ACC_SET_OFFLINE_ONLY;
75
76	s = set_add( &acc->set, "resource_select", "activity", NULL, acc );
77
78	s = set_add( &acc->set, "sasl", "true", set_eval_bool, acc );
79	s->flags \|= ACC_SET_OFFLINE_ONLY \| SET_HIDDEN_DEFAULT;
80
81	s = set_add( &acc->set, "server", NULL, set_eval_account, acc );
82	s->flags \|= ACC_SET_NOSAVE \| ACC_SET_OFFLINE_ONLY \| SET_NULL_OK;
83
84	s = set_add( &acc->set, "ssl", "false", set_eval_bool, acc );
85	s->flags \|= ACC_SET_OFFLINE_ONLY;
86
87	s = set_add( &acc->set, "tls", "try", set_eval_tls, acc );
88	s->flags \|= ACC_SET_OFFLINE_ONLY;
89
90	s = set_add( &acc->set, "tls_verify", "true", set_eval_bool, acc );
91	s->flags \|= ACC_SET_OFFLINE_ONLY;
92
93	s = set_add( &acc->set, "user_agent", "BitlBee", NULL, acc );
94
95	s = set_add( &acc->set, "xmlconsole", "false", set_eval_bool, acc );
96	s->flags \|= ACC_SET_OFFLINE_ONLY;
97
98	acc->flags \|= ACC_FLAG_AWAY_MESSAGE \| ACC_FLAG_STATUS_MESSAGE;
99	}
100
101	static void jabber_generate_id_hash( struct jabber_data *jd );
102
103	static void jabber_login( account_t *acc )
104	{
105	struct im_connection *ic = imcb_new( acc );
106	struct jabber_data *jd = g_new0( struct jabber_data, 1 );
107	char *s;
108
109	/* For now this is needed in the _connected() handlers if using
110	GLib event handling, to make sure we're not handling events
111	on dead connections. */
112	jabber_connections = g_slist_prepend( jabber_connections, ic );
113
114	jd->ic = ic;
115	ic->proto_data = jd;
116
117	jabber_set_me( ic, acc->user );
118
119	jd->fd = jd->r_inpa = jd->w_inpa = -1;
120
121	if( jd->server == NULL )
122	{
123	imcb_error( ic, "Incomplete account name (format it like <username@jabberserver.name>)" );
124	imc_logout( ic, FALSE );
125	return;
126	}
127
128	if( ( s = strchr( jd->server, '/' ) ) )
129	{
130	*s = 0;
131	set_setstr( &acc->set, "resource", s + 1 );
132
133	/* Also remove the /resource from the original variable so we
134	won't have to do this again every time. */
135	s = strchr( acc->user, '/' );
136	*s = 0;
137	}
138
139	jd->node_cache = g_hash_table_new_full( g_str_hash, g_str_equal, NULL, jabber_cache_entry_free );
140	jd->buddies = g_hash_table_new( g_str_hash, g_str_equal );
141
142	if( set_getbool( &acc->set, "oauth" ) )
143	{
144	GSList *p_in = NULL;
145	const char *tok;
146
147	jd->fd = jd->r_inpa = jd->w_inpa = -1;
148
149	if( strstr( jd->server, ".live.com" ) )
150	jd->oauth2_service = &oauth2_service_mslive;
151	else if( strstr( jd->server, ".facebook.com" ) )
152	jd->oauth2_service = &oauth2_service_facebook;
153	else
154	jd->oauth2_service = &oauth2_service_google;
155
156	oauth_params_parse( &p_in, ic->acc->pass );
157
158	/* First see if we have a refresh token, in which case any
159	access token we might have has probably expired already
160	anyway. */
161	if( ( tok = oauth_params_get( &p_in, "refresh_token" ) ) )
162	{
163	sasl_oauth2_refresh( ic, tok );
164	}
165	/* If we don't have a refresh token, let's hope the access
166	token is still usable. */
167	else if( ( tok = oauth_params_get( &p_in, "access_token" ) ) )
168	{
169	jd->oauth2_access_token = g_strdup( tok );
170	jabber_connect( ic );
171	}
172	/* If we don't have any, start the OAuth process now. Don't
173	even open an XMPP connection yet. */
174	else
175	{
176	sasl_oauth2_init( ic );
177	ic->flags \|= OPT_SLOW_LOGIN;
178	}
179
180	oauth_params_free( &p_in );
181	}
182	else
183	jabber_connect( ic );
184	}
185
186	/* Separate this from jabber_login() so we can do OAuth first if necessary.
187	Putting this in io.c would probably be more correct. */
188	void jabber_connect( struct im_connection *ic )
189	{
190	account_t *acc = ic->acc;
191	struct jabber_data *jd = ic->proto_data;
192	int i;
193	char *connect_to;
194	struct ns_srv_reply *srvl = NULL, srv = NULL;
195
196	/* Figure out the hostname to connect to. */
197	if( acc->server && *acc->server )
198	connect_to = acc->server;
199	else if( ( srvl = srv_lookup( "xmpp-client", "tcp", jd->server ) ) \|\|
200	( srvl = srv_lookup( "jabber-client", "tcp", jd->server ) ) )
201	{
202	/* Find the lowest-priority one. These usually come
203	back in random/shuffled order. Not looking at
204	weights etc for now. */
205	srv = *srvl;
206	for( i = 1; srvl[i]; i ++ )
207	if( srvl[i]->prio < srv->prio )
208	srv = srvl[i];
209
210	connect_to = srv->name;
211	}
212	else
213	connect_to = jd->server;
214
215	imcb_log( ic, "Connecting" );
216
217	for( i = 0; jabber_port_list[i] > 0; i ++ )
218	if( set_getint( &acc->set, "port" ) == jabber_port_list[i] )
219	break;
220
221	if( jabber_port_list[i] == 0 )
222	{
223	imcb_log( ic, "Illegal port number" );
224	imc_logout( ic, FALSE );
225	return;
226	}
227
228	/* For non-SSL connections we can try to use the port # from the SRV
229	reply, but let's not do that when using SSL, SSL usually runs on
230	non-standard ports... */
231	if( set_getbool( &acc->set, "ssl" ) )
232	{
233	jd->ssl = ssl_connect( connect_to, set_getint( &acc->set, "port" ), FALSE, jabber_connected_ssl, ic );
234	jd->fd = jd->ssl ? ssl_getfd( jd->ssl ) : -1;
235	}
236	else
237	{
238	jd->fd = proxy_connect( connect_to, srv ? srv->port : set_getint( &acc->set, "port" ), jabber_connected_plain, ic );
239	}
240	srv_free( srvl );
241
242	if( jd->fd == -1 )
243	{
244	imcb_error( ic, "Could not connect to server" );
245	imc_logout( ic, TRUE );
246
247	return;
248	}
249
250	if( set_getbool( &acc->set, "xmlconsole" ) )
251	{
252	jd->flags \|= JFLAG_XMLCONSOLE;
253	/* Shouldn't really do this at this stage already, maybe. But
254	I think this shouldn't break anything. */
255	imcb_add_buddy( ic, JABBER_XMLCONSOLE_HANDLE, NULL );
256	}
257
258	jabber_generate_id_hash( jd );
259	}
260
261	/* This generates an unfinished md5_state_t variable. Every time we generate
262	an ID, we finish the state by adding a sequence number and take the hash. */
263	static void jabber_generate_id_hash( struct jabber_data *jd )
264	{
265	md5_byte_t binbuf[4];
266	char *s;
267
268	md5_init( &jd->cached_id_prefix );
269	md5_append( &jd->cached_id_prefix, (unsigned char *) jd->username, strlen( jd->username ) );
270	md5_append( &jd->cached_id_prefix, (unsigned char *) jd->server, strlen( jd->server ) );
271	s = set_getstr( &jd->ic->acc->set, "resource" );
272	md5_append( &jd->cached_id_prefix, (unsigned char *) s, strlen( s ) );
273	random_bytes( binbuf, 4 );
274	md5_append( &jd->cached_id_prefix, binbuf, 4 );
275	}
276
277	static void jabber_logout( struct im_connection *ic )
278	{
279	struct jabber_data *jd = ic->proto_data;
280
281	while( jd->filetransfers )
282	imcb_file_canceled( ic, ( ( struct jabber_transfer *) jd->filetransfers->data )->ft, "Logging out" );
283
284	while( jd->streamhosts )
285	{
286	jabber_streamhost_t *sh = jd->streamhosts->data;
287	jd->streamhosts = g_slist_remove( jd->streamhosts, sh );
288	g_free( sh->jid );
289	g_free( sh->host );
290	g_free( sh );
291	}
292
293	if( jd->fd >= 0 )
294	jabber_end_stream( ic );
295
296	while( ic->groupchats )
297	jabber_chat_free( ic->groupchats->data );
298
299	if( jd->r_inpa >= 0 )
300	b_event_remove( jd->r_inpa );
301	if( jd->w_inpa >= 0 )
302	b_event_remove( jd->w_inpa );
303
304	if( jd->ssl )
305	ssl_disconnect( jd->ssl );
306	if( jd->fd >= 0 )
307	closesocket( jd->fd );
308
309	if( jd->tx_len )
310	g_free( jd->txq );
311
312	if( jd->node_cache )
313	g_hash_table_destroy( jd->node_cache );
314
315	jabber_buddy_remove_all( ic );
316
317	xt_free( jd->xt );
318
319	g_free( jd->oauth2_access_token );
320	g_free( jd->away_message );
321	g_free( jd->username );
322	g_free( jd->me );
323	g_free( jd );
324
325	jabber_connections = g_slist_remove( jabber_connections, ic );
326	}
327
328	static int jabber_buddy_msg( struct im_connection ic, char who, char *message, int flags )
329	{
330	struct jabber_data *jd = ic->proto_data;
331	struct jabber_buddy *bud;
332	struct xt_node *node;
333	char *s;
334	int st;
335
336	if( g_strcasecmp( who, JABBER_XMLCONSOLE_HANDLE ) == 0 )
337	return jabber_write( ic, message, strlen( message ) );
338
339	if( g_strcasecmp( who, JABBER_OAUTH_HANDLE ) == 0 &&
340	!( jd->flags & OPT_LOGGED_IN ) && jd->fd == -1 )
341	{
342	if( sasl_oauth2_get_refresh_token( ic, message ) )
343	{
344	return 1;
345	}
346	else
347	{
348	imcb_error( ic, "OAuth failure" );
349	imc_logout( ic, TRUE );
350	return 0;
351	}
352	}
353
354	if( ( s = strchr( who, '=' ) ) && jabber_chat_by_jid( ic, s + 1 ) )
355	bud = jabber_buddy_by_ext_jid( ic, who, 0 );
356	else
357	bud = jabber_buddy_by_jid( ic, who, GET_BUDDY_BARE_OK );
358
359	node = xt_new_node( "body", message, NULL );
360	node = jabber_make_packet( "message", "chat", bud ? bud->full_jid : who, node );
361
362	if( bud && ( jd->flags & JFLAG_WANT_TYPING ) &&
363	( ( bud->flags & JBFLAG_DOES_XEP85 ) \|\|
364	!( bud->flags & JBFLAG_PROBED_XEP85 ) ) )
365	{
366	struct xt_node *act;
367
368	/* If the user likes typing notification and if we don't know
369	(and didn't probe before) if this resource supports XEP85,
370	include a probe in this packet now. Also, if we know this
371	buddy does support XEP85, we have to send this <active/>
372	tag to tell that the user stopped typing (well, that's what
373	we guess when s/he pressed Enter...). */
374	act = xt_new_node( "active", NULL, NULL );
375	xt_add_attr( act, "xmlns", XMLNS_CHATSTATES );
376	xt_add_child( node, act );
377
378	/* Just make sure we do this only once. */
379	bud->flags \|= JBFLAG_PROBED_XEP85;
380	}
381
382	st = jabber_write_packet( ic, node );
383	xt_free_node( node );
384
385	return st;
386	}
387
388	static GList jabber_away_states( struct im_connection ic )
389	{
390	static GList *l = NULL;
391	int i;
392
393	if( l == NULL )
394	for( i = 0; jabber_away_state_list[i].full_name; i ++ )
395	l = g_list_append( l, (void*) jabber_away_state_list[i].full_name );
396
397	return l;
398	}
399
400	static void jabber_get_info( struct im_connection ic, char who )
401	{
402	struct jabber_buddy *bud;
403
404	bud = jabber_buddy_by_jid( ic, who, GET_BUDDY_FIRST );
405
406	while( bud )
407	{
408	imcb_log( ic, "Buddy %s (%d) information:", bud->full_jid, bud->priority );
409	if( bud->away_state )
410	imcb_log( ic, "Away state: %s", bud->away_state->full_name );
411	imcb_log( ic, "Status message: %s", bud->away_message ? bud->away_message : "(none)" );
412
413	bud = bud->next;
414	}
415
416	jabber_get_vcard( ic, bud ? bud->full_jid : who );
417	}
418
419	static void jabber_set_away( struct im_connection ic, char state_txt, char *message )
420	{
421	struct jabber_data *jd = ic->proto_data;
422
423	/* state_txt == NULL -> Not away.
424	Unknown state -> fall back to the first defined away state. */
425	if( state_txt == NULL )
426	jd->away_state = NULL;
427	else if( ( jd->away_state = jabber_away_state_by_name( state_txt ) ) == NULL )
428	jd->away_state = jabber_away_state_list;
429
430	g_free( jd->away_message );
431	jd->away_message = ( message && *message ) ? g_strdup( message ) : NULL;
432
433	presence_send_update( ic );
434	}
435
436	static void jabber_add_buddy( struct im_connection ic, char who, char *group )
437	{
438	struct jabber_data *jd = ic->proto_data;
439
440	if( g_strcasecmp( who, JABBER_XMLCONSOLE_HANDLE ) == 0 )
441	{
442	jd->flags \|= JFLAG_XMLCONSOLE;
443	imcb_add_buddy( ic, JABBER_XMLCONSOLE_HANDLE, NULL );
444	return;
445	}
446
447	if( jabber_add_to_roster( ic, who, NULL, group ) )
448	presence_send_request( ic, who, "subscribe" );
449	}
450
451	static void jabber_remove_buddy( struct im_connection ic, char who, char *group )
452	{
453	struct jabber_data *jd = ic->proto_data;
454
455	if( g_strcasecmp( who, JABBER_XMLCONSOLE_HANDLE ) == 0 )
456	{
457	jd->flags &= ~JFLAG_XMLCONSOLE;
458	/* Not necessary for now. And for now the code isn't too
459	happy if the buddy is completely gone right after calling
460	this function already.
461	imcb_remove_buddy( ic, JABBER_XMLCONSOLE_HANDLE, NULL );
462	*/
463	return;
464	}
465
466	/* We should always do this part. Clean up our administration a little bit. */
467	jabber_buddy_remove_bare( ic, who );
468
469	if( jabber_remove_from_roster( ic, who ) )
470	presence_send_request( ic, who, "unsubscribe" );
471	}
472
473	static struct groupchat jabber_chat_join_( struct im_connection ic, const char room, const char nick, const char password, set_t *sets )
474	{
475	if( strchr( room, '@' ) == NULL )
476	imcb_error( ic, "Invalid room name: %s", room );
477	else if( jabber_chat_by_jid( ic, room ) )
478	imcb_error( ic, "Already present in chat `%s'", room );
479	else
480	return jabber_chat_join( ic, room, nick, set_getstr( sets, "password" ) );
481
482	return NULL;
483	}
484
485	static void jabber_chat_msg_( struct groupchat c, char message, int flags )
486	{
487	if( c && message )
488	jabber_chat_msg( c, message, flags );
489	}
490
491	static void jabber_chat_topic_( struct groupchat c, char topic )
492	{
493	if( c && topic )
494	jabber_chat_topic( c, topic );
495	}
496
497	static void jabber_chat_leave_( struct groupchat *c )
498	{
499	if( c )
500	jabber_chat_leave( c, NULL );
501	}
502
503	static void jabber_chat_invite_( struct groupchat c, char who, char *msg )
504	{
505	struct jabber_data *jd = c->ic->proto_data;
506	struct jabber_chat *jc = c->data;
507	gchar *msg_alt = NULL;
508
509	if( msg == NULL )
510	msg_alt = g_strdup_printf( "%s invited you to %s", jd->me, jc->name );
511
512	if( c && who )
513	jabber_chat_invite( c, who, msg ? msg : msg_alt );
514
515	g_free( msg_alt );
516	}
517
518	static void jabber_keepalive( struct im_connection *ic )
519	{
520	/* Just any whitespace character is enough as a keepalive for XMPP sessions. */
521	if( !jabber_write( ic, "\n", 1 ) )
522	return;
523
524	/* This runs the garbage collection every minute, which means every packet
525	is in the cache for about a minute (which should be enough AFAIK). */
526	jabber_cache_clean( ic );
527	}
528
529	static int jabber_send_typing( struct im_connection ic, char who, int typing )
530	{
531	struct jabber_data *jd = ic->proto_data;
532	struct jabber_buddy *bud;
533
534	/* Enable typing notification related code from now. */
535	jd->flags \|= JFLAG_WANT_TYPING;
536
537	if( ( bud = jabber_buddy_by_jid( ic, who, 0 ) ) == NULL )
538	{
539	/* Sending typing notifications to unknown buddies is
540	unsupported for now. Shouldn't be a problem, I think. */
541	return 0;
542	}
543
544	if( bud->flags & JBFLAG_DOES_XEP85 )
545	{
546	/* We're only allowed to send this stuff if we know the other
547	side supports it. */
548
549	struct xt_node *node;
550	char *type;
551	int st;
552
553	if( typing & OPT_TYPING )
554	type = "composing";
555	else if( typing & OPT_THINKING )
556	type = "paused";
557	else
558	type = "active";
559
560	node = xt_new_node( type, NULL, NULL );
561	xt_add_attr( node, "xmlns", XMLNS_CHATSTATES );
562	node = jabber_make_packet( "message", "chat", bud->full_jid, node );
563
564	st = jabber_write_packet( ic, node );
565	xt_free_node( node );
566
567	return st;
568	}
569
570	return 1;
571	}
572
573	void jabber_chat_add_settings( account_t acc, set_t *head )
574	{
575	/* Meh. Stupid room passwords. Not trying to obfuscate/hide
576	them from the user for now. */
577	set_add( head, "password", NULL, NULL, NULL );
578	}
579
580	void jabber_chat_free_settings( account_t acc, set_t *head )
581	{
582	set_del( head, "password" );
583	}
584
585	GList jabber_buddy_action_list( bee_user_t bu )
586	{
587	static GList *ret = NULL;
588
589	if( ret == NULL )
590	{
591	static const struct buddy_action ba[2] = {
592	{ "VERSION", "Get client (version) information" },
593	};
594
595	ret = g_list_prepend( ret, (void*) ba + 0 );
596	}
597
598	return ret;
599	}
600
601	void jabber_buddy_action( struct bee_user bu, const char action, char const args[], void *data )
602	{
603	if( g_strcasecmp( action, "VERSION" ) == 0 )
604	{
605	struct jabber_buddy *bud;
606
607	if( ( bud = jabber_buddy_by_ext_jid( bu->ic, bu->handle, 0 ) ) == NULL )
608	bud = jabber_buddy_by_jid( bu->ic, bu->handle, GET_BUDDY_FIRST );
609	for( ; bud; bud = bud->next )
610	jabber_iq_version_send( bu->ic, bud, data );
611	}
612
613	return NULL;
614	}
615
616	void jabber_initmodule()
617	{
618	struct prpl *ret = g_new0( struct prpl, 1 );
619
620	ret->name = "jabber";
621	ret->mms = 0; /* no limit */
622	ret->login = jabber_login;
623	ret->init = jabber_init;
624	ret->logout = jabber_logout;
625	ret->buddy_msg = jabber_buddy_msg;
626	ret->away_states = jabber_away_states;
627	ret->set_away = jabber_set_away;
628	// ret->set_info = jabber_set_info;
629	ret->get_info = jabber_get_info;
630	ret->add_buddy = jabber_add_buddy;
631	ret->remove_buddy = jabber_remove_buddy;
632	ret->chat_msg = jabber_chat_msg_;
633	ret->chat_topic = jabber_chat_topic_;
634	ret->chat_invite = jabber_chat_invite_;
635	ret->chat_leave = jabber_chat_leave_;
636	ret->chat_join = jabber_chat_join_;
637	ret->chat_add_settings = jabber_chat_add_settings;
638	ret->chat_free_settings = jabber_chat_free_settings;
639	ret->keepalive = jabber_keepalive;
640	ret->send_typing = jabber_send_typing;
641	ret->handle_cmp = g_strcasecmp;
642	ret->transfer_request = jabber_si_transfer_request;
643	ret->buddy_action_list = jabber_buddy_action_list;
644	ret->buddy_action = jabber_buddy_action;
645
646	register_protocol( ret );
647	}

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

Download in other formats: