source: protocols/jabber/jabber.c @ 2d93a51e

Last change on this file since 2d93a51e was ad46e4d, checked in by Wilmer van der Gaast <wilmer@…>, at 2011-12-22T11:23:18Z

Use initgroups() as well when dropping privileges. Closes bug #852.

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