source: storage_xml.c @ d7f8500

Last change on this file since d7f8500 was 7a6ba50, checked in by Wilmer van der Gaast <wilmer@…>, at 2010-06-29T23:44:17Z

Also allow selecting contacts for a channel by protocol instead of account.
If someone has two MSN accts and wants contacts from both in one channel,
this is now possible.

  • Property mode set to 100644
File size: 17.0 KB
Line 
1  /********************************************************************\
2  * BitlBee -- An IRC to other IM-networks gateway                     *
3  *                                                                    *
4  * Copyright 2002-2006 Wilmer van der Gaast and others                *
5  \********************************************************************/
6
7/* Storage backend that uses an XMLish format for all data. */
8
9/*
10  This program is free software; you can redistribute it and/or modify
11  it under the terms of the GNU General Public License as published by
12  the Free Software Foundation; either version 2 of the License, or
13  (at your option) any later version.
14
15  This program is distributed in the hope that it will be useful,
16  but WITHOUT ANY WARRANTY; without even the implied warranty of
17  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  GNU General Public License for more details.
19
20  You should have received a copy of the GNU General Public License with
21  the Debian GNU/Linux distribution in /usr/share/common-licenses/GPL;
22  if not, write to the Free Software Foundation, Inc., 59 Temple Place,
23  Suite 330, Boston, MA  02111-1307  USA
24*/
25
26#define BITLBEE_CORE
27#include "bitlbee.h"
28#include "base64.h"
29#include "arc.h"
30#include "md5.h"
31
32#if GLIB_CHECK_VERSION(2,8,0)
33#include <glib/gstdio.h>
34#else
35/* GLib < 2.8.0 doesn't have g_access, so just use the system access(). */
36#include <unistd.h>
37#define g_access access
38#endif
39
40typedef enum
41{
42        XML_PASS_CHECK_ONLY = -1,
43        XML_PASS_UNKNOWN = 0,
44        XML_PASS_WRONG,
45        XML_PASS_OK
46} xml_pass_st;
47
48/* To make it easier later when extending the format: */
49#define XML_FORMAT_VERSION 1
50
51struct xml_parsedata
52{
53        irc_t *irc;
54        char *current_setting;
55        account_t *current_account;
56        irc_channel_t *current_channel;
57        set_t **current_set_head;
58        char *given_nick;
59        char *given_pass;
60        xml_pass_st pass_st;
61};
62
63static char *xml_attr( const gchar **attr_names, const gchar **attr_values, const gchar *key )
64{
65        int i;
66       
67        for( i = 0; attr_names[i]; i ++ )
68                if( g_strcasecmp( attr_names[i], key ) == 0 )
69                        return (char*) attr_values[i];
70       
71        return NULL;
72}
73
74static void xml_destroy_xd( gpointer data )
75{
76        struct xml_parsedata *xd = data;
77       
78        g_free( xd->given_nick );
79        g_free( xd->given_pass );
80        g_free( xd );
81}
82
83static void xml_start_element( GMarkupParseContext *ctx, const gchar *element_name, const gchar **attr_names, const gchar **attr_values, gpointer data, GError **error )
84{
85        struct xml_parsedata *xd = data;
86        irc_t *irc = xd->irc;
87       
88        if( g_strcasecmp( element_name, "user" ) == 0 )
89        {
90                char *nick = xml_attr( attr_names, attr_values, "nick" );
91                char *pass = xml_attr( attr_names, attr_values, "password" );
92                int st;
93               
94                if( !nick || !pass )
95                {
96                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
97                                     "Missing attributes for %s element", element_name );
98                }
99                else if( ( st = md5_verify_password( xd->given_pass, pass ) ) == -1 )
100                {
101                        xd->pass_st = XML_PASS_WRONG;
102                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
103                                     "Error while decoding password attribute" );
104                }
105                else if( st == 0 )
106                {
107                        if( xd->pass_st != XML_PASS_CHECK_ONLY )
108                                xd->pass_st = XML_PASS_OK;
109                }
110                else
111                {
112                        xd->pass_st = XML_PASS_WRONG;
113                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
114                                     "Password mismatch" );
115                }
116        }
117        else if( xd->pass_st < XML_PASS_OK )
118        {
119                /* Let's not parse anything else if we only have to check
120                   the password. */
121        }
122        else if( g_strcasecmp( element_name, "account" ) == 0 )
123        {
124                char *protocol, *handle, *server, *password = NULL, *autoconnect;
125                char *pass_b64 = NULL;
126                unsigned char *pass_cr = NULL;
127                int pass_len;
128                struct prpl *prpl = NULL;
129               
130                handle = xml_attr( attr_names, attr_values, "handle" );
131                pass_b64 = xml_attr( attr_names, attr_values, "password" );
132                server = xml_attr( attr_names, attr_values, "server" );
133                autoconnect = xml_attr( attr_names, attr_values, "autoconnect" );
134               
135                protocol = xml_attr( attr_names, attr_values, "protocol" );
136                if( protocol )
137                        prpl = find_protocol( protocol );
138               
139                if( !handle || !pass_b64 || !protocol )
140                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
141                                     "Missing attributes for %s element", element_name );
142                else if( !prpl )
143                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
144                                     "Unknown protocol: %s", protocol );
145                else if( ( pass_len = base64_decode( pass_b64, (unsigned char**) &pass_cr ) ) &&
146                                         arc_decode( pass_cr, pass_len, &password, xd->given_pass ) )
147                {
148                        xd->current_account = account_add( irc->b, prpl, handle, password );
149                        if( server )
150                                set_setstr( &xd->current_account->set, "server", server );
151                        if( autoconnect )
152                                set_setstr( &xd->current_account->set, "auto_connect", autoconnect );
153                }
154                else
155                {
156                        /* Actually the _decode functions don't even return error codes,
157                           but maybe they will later... */
158                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
159                                     "Error while decrypting account password" );
160                }
161               
162                g_free( pass_cr );
163                g_free( password );
164        }
165        else if( g_strcasecmp( element_name, "setting" ) == 0 )
166        {
167                char *setting;
168               
169                if( xd->current_setting )
170                {
171                        g_free( xd->current_setting );
172                        xd->current_setting = NULL;
173                }
174               
175                if( ( setting = xml_attr( attr_names, attr_values, "name" ) ) )
176                {
177                        if( xd->current_channel != NULL )
178                                xd->current_set_head = &xd->current_channel->set;
179                        else if( xd->current_account != NULL )
180                                xd->current_set_head = &xd->current_account->set;
181                        else
182                                xd->current_set_head = &xd->irc->b->set;
183                       
184                        xd->current_setting = g_strdup( setting );
185                }
186                else
187                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
188                                     "Missing attributes for %s element", element_name );
189        }
190        else if( g_strcasecmp( element_name, "buddy" ) == 0 )
191        {
192                char *handle, *nick;
193               
194                handle = xml_attr( attr_names, attr_values, "handle" );
195                nick = xml_attr( attr_names, attr_values, "nick" );
196               
197                if( xd->current_account && handle && nick )
198                {
199                        nick_set( xd->current_account, handle, nick );
200                }
201                else
202                {
203                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
204                                     "Missing attributes for %s element", element_name );
205                }
206        }
207        else if( g_strcasecmp( element_name, "channel" ) == 0 )
208        {
209                char *name, *type;
210               
211                name = xml_attr( attr_names, attr_values, "name" );
212                type = xml_attr( attr_names, attr_values, "type" );
213               
214                if( !name || !type )
215                {
216                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
217                                     "Missing attributes for %s element", element_name );
218                        return;
219                }
220               
221                /* The channel may exist already, for example if it's &bitlbee.
222                   Also, it's possible that the user just reconnected and the
223                   IRC client already rejoined all channels it was in. They
224                   should still get the right settings. */
225                if( ( xd->current_channel = irc_channel_by_name( irc, name ) ) ||
226                    ( xd->current_channel = irc_channel_new( irc, name ) ) )
227                        set_setstr( &xd->current_channel->set, "type", type );
228        }
229        /* Backward compatibility: Keep this around for a while for people
230           switching from BitlBee 1.2.4+. */
231        else if( g_strcasecmp( element_name, "chat" ) == 0 )
232        {
233                char *handle, *channel;
234               
235                handle = xml_attr( attr_names, attr_values, "handle" );
236                channel = xml_attr( attr_names, attr_values, "channel" );
237               
238                if( xd->current_account && handle && channel )
239                {
240                        irc_channel_t *ic;
241                        char *acc;
242                       
243                        acc = g_strdup_printf( "%s(%s)",
244                                               xd->current_account->prpl->name,
245                                               xd->current_account->user );
246                       
247                        if( ( ic = irc_channel_new( irc, channel ) ) &&
248                            set_setstr( &ic->set, "type", "chat" ) &&
249                            set_setstr( &ic->set, "chat_type", "room" ) &&
250                            set_setstr( &ic->set, "account", acc ) &&
251                            set_setstr( &ic->set, "room", handle ) )
252                        {
253                                /* Try to pick up some settings where possible. */
254                                xd->current_channel = ic;
255                        }
256                        else if( ic )
257                                irc_channel_free( ic );
258                       
259                        g_free( acc );
260                }
261                else
262                {
263                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
264                                     "Missing attributes for %s element", element_name );
265                }
266        }
267        else
268        {
269                g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_UNKNOWN_ELEMENT,
270                             "Unkown element: %s", element_name );
271        }
272}
273
274static void xml_end_element( GMarkupParseContext *ctx, const gchar *element_name, gpointer data, GError **error )
275{
276        struct xml_parsedata *xd = data;
277       
278        if( g_strcasecmp( element_name, "setting" ) == 0 && xd->current_setting )
279        {
280                g_free( xd->current_setting );
281                xd->current_setting = NULL;
282        }
283        else if( g_strcasecmp( element_name, "account" ) == 0 )
284        {
285                xd->current_account = NULL;
286        }
287        else if( g_strcasecmp( element_name, "channel" ) == 0 ||
288                 g_strcasecmp( element_name, "chat" ) == 0 )
289        {
290                xd->current_channel = NULL;
291        }
292}
293
294static void xml_text( GMarkupParseContext *ctx, const gchar *text_orig, gsize text_len, gpointer data, GError **error )
295{
296        char text[text_len+1];
297        struct xml_parsedata *xd = data;
298       
299        strncpy( text, text_orig, text_len );
300        text[text_len] = 0;
301       
302        if( xd->pass_st < XML_PASS_OK )
303        {
304                /* Let's not parse anything else if we only have to check
305                   the password, or if we didn't get the chance to check it
306                   yet. */
307        }
308        else if( g_strcasecmp( g_markup_parse_context_get_element( ctx ), "setting" ) == 0 && xd->current_setting )
309        {
310                set_setstr( xd->current_set_head, xd->current_setting, (char*) text );
311                g_free( xd->current_setting );
312                xd->current_setting = NULL;
313        }
314}
315
316GMarkupParser xml_parser =
317{
318        xml_start_element,
319        xml_end_element,
320        xml_text,
321        NULL,
322        NULL
323};
324
325static void xml_init( void )
326{
327        if( g_access( global.conf->configdir, F_OK ) != 0 )
328                log_message( LOGLVL_WARNING, "The configuration directory `%s' does not exist. Configuration won't be saved.", global.conf->configdir );
329        else if( g_access( global.conf->configdir, F_OK ) != 0 || 
330                 g_access( global.conf->configdir, W_OK ) != 0 )
331                log_message( LOGLVL_WARNING, "Permission problem: Can't read/write from/to `%s'.", global.conf->configdir );
332}
333
334static storage_status_t xml_load_real( irc_t *irc, const char *my_nick, const char *password, xml_pass_st action )
335{
336        GMarkupParseContext *ctx;
337        struct xml_parsedata *xd;
338        char *fn, buf[512];
339        GError *gerr = NULL;
340        int fd, st;
341       
342        xd = g_new0( struct xml_parsedata, 1 );
343        xd->irc = irc;
344        xd->given_nick = g_strdup( my_nick );
345        xd->given_pass = g_strdup( password );
346        xd->pass_st = action;
347        nick_lc( xd->given_nick );
348       
349        fn = g_strdup_printf( "%s%s%s", global.conf->configdir, xd->given_nick, ".xml" );
350        if( ( fd = open( fn, O_RDONLY ) ) < 0 )
351        {
352                xml_destroy_xd( xd );
353                g_free( fn );
354                return STORAGE_NO_SUCH_USER;
355        }
356        g_free( fn );
357       
358        ctx = g_markup_parse_context_new( &xml_parser, 0, xd, xml_destroy_xd );
359       
360        while( ( st = read( fd, buf, sizeof( buf ) ) ) > 0 )
361        {
362                if( !g_markup_parse_context_parse( ctx, buf, st, &gerr ) || gerr )
363                {
364                        xml_pass_st pass_st = xd->pass_st;
365                       
366                        g_markup_parse_context_free( ctx );
367                        close( fd );
368                       
369                        if( pass_st == XML_PASS_WRONG )
370                        {
371                                g_clear_error( &gerr );
372                                return STORAGE_INVALID_PASSWORD;
373                        }
374                        else
375                        {
376                                if( gerr && irc )
377                                        irc_usermsg( irc, "Error from XML-parser: %s", gerr->message );
378                               
379                                g_clear_error( &gerr );
380                                return STORAGE_OTHER_ERROR;
381                        }
382                }
383        }
384        /* Just to be sure... */
385        g_clear_error( &gerr );
386       
387        g_markup_parse_context_free( ctx );
388        close( fd );
389       
390        if( action == XML_PASS_CHECK_ONLY )
391                return STORAGE_OK;
392       
393        return STORAGE_OK;
394}
395
396static storage_status_t xml_load( irc_t *irc, const char *password )
397{
398        return xml_load_real( irc, irc->user->nick, password, XML_PASS_UNKNOWN );
399}
400
401static storage_status_t xml_check_pass( const char *my_nick, const char *password )
402{
403        /* This is a little bit risky because we have to pass NULL for the
404           irc_t argument. This *should* be fine, if I didn't miss anything... */
405        return xml_load_real( NULL, my_nick, password, XML_PASS_CHECK_ONLY );
406}
407
408static int xml_printf( int fd, int indent, char *fmt, ... )
409{
410        va_list params;
411        char *out;
412        char tabs[9] = "\t\t\t\t\t\t\t\t";
413        int len;
414       
415        /* Maybe not very clean, but who needs more than 8 levels of indentation anyway? */
416        if( write( fd, tabs, indent <= 8 ? indent : 8 ) != indent )
417                return 0;
418       
419        va_start( params, fmt );
420        out = g_markup_vprintf_escaped( fmt, params );
421        va_end( params );
422       
423        len = strlen( out );
424        len -= write( fd, out, len );
425        g_free( out );
426       
427        return len == 0;
428}
429
430static gboolean xml_save_nick( gpointer key, gpointer value, gpointer data );
431
432static storage_status_t xml_save( irc_t *irc, int overwrite )
433{
434        char path[512], *path2, *pass_buf = NULL;
435        set_t *set;
436        account_t *acc;
437        int fd;
438        md5_byte_t pass_md5[21];
439        md5_state_t md5_state;
440        GSList *l;
441       
442        path2 = g_strdup( irc->user->nick );
443        nick_lc( path2 );
444        g_snprintf( path, sizeof( path ) - 2, "%s%s%s", global.conf->configdir, path2, ".xml" );
445        g_free( path2 );
446       
447        if( !overwrite && g_access( path, F_OK ) == 0 )
448                return STORAGE_ALREADY_EXISTS;
449       
450        strcat( path, ".XXXXXX" );
451        if( ( fd = mkstemp( path ) ) < 0 )
452        {
453                irc_usermsg( irc, "Error while opening configuration file." );
454                return STORAGE_OTHER_ERROR;
455        }
456       
457        /* Generate a salted md5sum of the password. Use 5 bytes for the salt
458           (to prevent dictionary lookups of passwords) to end up with a 21-
459           byte password hash, more convenient for base64 encoding. */
460        random_bytes( pass_md5 + 16, 5 );
461        md5_init( &md5_state );
462        md5_append( &md5_state, (md5_byte_t*) irc->password, strlen( irc->password ) );
463        md5_append( &md5_state, pass_md5 + 16, 5 ); /* Add the salt. */
464        md5_finish( &md5_state, pass_md5 );
465        /* Save the hash in base64-encoded form. */
466        pass_buf = base64_encode( pass_md5, 21 );
467       
468        if( !xml_printf( fd, 0, "<user nick=\"%s\" password=\"%s\" version=\"%d\">\n", irc->user->nick, pass_buf, XML_FORMAT_VERSION ) )
469                goto write_error;
470       
471        g_free( pass_buf );
472       
473        for( set = irc->b->set; set; set = set->next )
474                if( set->value )
475                        if( !xml_printf( fd, 1, "<setting name=\"%s\">%s</setting>\n", set->key, set->value ) )
476                                goto write_error;
477       
478        for( acc = irc->b->accounts; acc; acc = acc->next )
479        {
480                unsigned char *pass_cr;
481                char *pass_b64;
482                int pass_len;
483               
484                pass_len = arc_encode( acc->pass, strlen( acc->pass ), (unsigned char**) &pass_cr, irc->password, 12 );
485                pass_b64 = base64_encode( pass_cr, pass_len );
486                g_free( pass_cr );
487               
488                if( !xml_printf( fd, 1, "<account protocol=\"%s\" handle=\"%s\" password=\"%s\" autoconnect=\"%d\"", acc->prpl->name, acc->user, pass_b64, acc->auto_connect ) )
489                {
490                        g_free( pass_b64 );
491                        goto write_error;
492                }
493                g_free( pass_b64 );
494               
495                if( acc->server && acc->server[0] && !xml_printf( fd, 0, " server=\"%s\"", acc->server ) )
496                        goto write_error;
497                if( !xml_printf( fd, 0, ">\n" ) )
498                        goto write_error;
499               
500                for( set = acc->set; set; set = set->next )
501                        if( set->value && !( set->flags & ACC_SET_NOSAVE ) )
502                                if( !xml_printf( fd, 2, "<setting name=\"%s\">%s</setting>\n", set->key, set->value ) )
503                                        goto write_error;
504               
505                /* This probably looks pretty strange. g_hash_table_foreach
506                   is quite a PITA already (but it can't get much better in
507                   C without using #define, I'm afraid), and since it
508                   doesn't seem to be possible to abort the foreach on write
509                   errors, so instead let's use the _find function and
510                   return TRUE on write errors. Which means, if we found
511                   something, there was an error. :-) */
512                if( g_hash_table_find( acc->nicks, xml_save_nick, & fd ) )
513                        goto write_error;
514               
515                if( !xml_printf( fd, 1, "</account>\n" ) )
516                        goto write_error;
517        }
518       
519        for( l = irc->channels; l; l = l->next )
520        {
521                irc_channel_t *ic = l->data;
522               
523                if( ic->flags & IRC_CHANNEL_TEMP )
524                        continue;
525               
526                if( !xml_printf( fd, 1, "<channel name=\"%s\" type=\"%s\">\n",
527                                 ic->name, set_getstr( &ic->set, "type" ) ) )
528                        goto write_error;
529               
530                for( set = ic->set; set; set = set->next )
531                        if( set->value && strcmp( set->key, "type" ) != 0 )
532                                if( !xml_printf( fd, 2, "<setting name=\"%s\">%s</setting>\n", set->key, set->value ) )
533                                        goto write_error;
534               
535                if( !xml_printf( fd, 1, "</channel>\n" ) )
536                        goto write_error;
537        }
538       
539        if( !xml_printf( fd, 0, "</user>\n" ) )
540                goto write_error;
541       
542        fsync( fd );
543        close( fd );
544       
545        path2 = g_strndup( path, strlen( path ) - 7 );
546        if( rename( path, path2 ) != 0 )
547        {
548                irc_usermsg( irc, "Error while renaming temporary configuration file." );
549               
550                g_free( path2 );
551                unlink( path );
552               
553                return STORAGE_OTHER_ERROR;
554        }
555       
556        g_free( path2 );
557       
558        return STORAGE_OK;
559
560write_error:
561        g_free( pass_buf );
562       
563        irc_usermsg( irc, "Write error. Disk full?" );
564        close( fd );
565       
566        return STORAGE_OTHER_ERROR;
567}
568
569static gboolean xml_save_nick( gpointer key, gpointer value, gpointer data )
570{
571        return !xml_printf( *( (int*) data ), 2, "<buddy handle=\"%s\" nick=\"%s\" />\n", key, value );
572}
573
574static storage_status_t xml_remove( const char *nick, const char *password )
575{
576        char s[512], *lc;
577        storage_status_t status;
578
579        status = xml_check_pass( nick, password );
580        if( status != STORAGE_OK )
581                return status;
582
583        lc = g_strdup( nick );
584        nick_lc( lc );
585        g_snprintf( s, 511, "%s%s%s", global.conf->configdir, lc, ".xml" );
586        g_free( lc );
587       
588        if( unlink( s ) == -1 )
589                return STORAGE_OTHER_ERROR;
590       
591        return STORAGE_OK;
592}
593
594storage_t storage_xml = {
595        .name = "xml",
596        .init = xml_init,
597        .check_pass = xml_check_pass,
598        .remove = xml_remove,
599        .load = xml_load,
600        .save = xml_save
601};
Note: See TracBrowser for help on using the repository browser.