source: protocols/jabber/jabber.c @ 4be0e34

Last change on this file since 4be0e34 was 64b6635, checked in by Wilmer van der Gaast <wilmer@…>, at 2011-12-18T21:56:44Z

Restructured and updated code a little bit to support new-style (much better
and "proper" OAuth2) Facebook OAuth support. (And, add wl.offline scope to get
tokens that don't expire after an hour.)

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