Context Navigation

source: protocols/jabber/jabber.c @ 7b05842

Last change on this file since 7b05842 was e31e5b8, checked in by Wilmer van der Gaast <wilmer@…>, at 2013-04-20T13:05:55Z
Merging "storage" branch which I wrote long ago. It separates generation of XML-formatted user configs from disk I/O so we can try to start using other mechanisms to store them (a REST API or something, for example).
Property mode set to `100644`
File size: 18.5 KB

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

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

Download in other formats: