Context Navigation

source: protocols/jabber/jabber.c @ 0e788f5

Last change on this file since 0e788f5 was 0e788f5, checked in by Wilmer van der Gaast <wilmer@…>, at 2013-02-21T19:15:59Z
I'm still bored on a long flight. Wrote a script to automatically update my copyright mentions since some were getting pretty stale. Left files not touched since before 2012 alone so that this change doesn't touch almost EVERY source file.
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 \|= 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", "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: