source: protocols/jabber/jabber.c @ 5513f3e

Last change on this file since 5513f3e was 792a93b, checked in by Wilmer van der Gaast <wilmer@…>, at 2011-12-23T12:44:08Z

Merging SSL certificate verification for GnuTLS, with help from AopicieR.

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