source: storage_xml.c @ 47b571d

Last change on this file since 47b571d was 47b571d, checked in by Jelmer Vernooij <jelmer@…>, at 2008-06-28T17:35:34Z

Avoid g_access on GLib < 2.8.0.

  • Property mode set to 100644
File size: 14.5 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#include <glib/gstdio.h>
32
33#if !GLIB_CHECK_VERSION(2,8,0)
34/* GLib < 2.8.0 doesn't have g_access, so just use the system access(). */
35#define g_access access
36#endif
37
38typedef enum
39{
40        XML_PASS_CHECK_ONLY = -1,
41        XML_PASS_UNKNOWN = 0,
42        XML_PASS_WRONG,
43        XML_PASS_OK
44} xml_pass_st;
45
46/* To make it easier later when extending the format: */
47#define XML_FORMAT_VERSION 1
48
49struct xml_parsedata
50{
51        irc_t *irc;
52        char *current_setting;
53        account_t *current_account;
54        char *given_nick;
55        char *given_pass;
56        xml_pass_st pass_st;
57};
58
59static char *xml_attr( const gchar **attr_names, const gchar **attr_values, const gchar *key )
60{
61        int i;
62       
63        for( i = 0; attr_names[i]; i ++ )
64                if( g_strcasecmp( attr_names[i], key ) == 0 )
65                        return (char*) attr_values[i];
66       
67        return NULL;
68}
69
70static void xml_destroy_xd( gpointer data )
71{
72        struct xml_parsedata *xd = data;
73       
74        g_free( xd->given_nick );
75        g_free( xd->given_pass );
76        g_free( xd );
77}
78
79static void xml_start_element( GMarkupParseContext *ctx, const gchar *element_name, const gchar **attr_names, const gchar **attr_values, gpointer data, GError **error )
80{
81        struct xml_parsedata *xd = data;
82        irc_t *irc = xd->irc;
83       
84        if( g_strcasecmp( element_name, "user" ) == 0 )
85        {
86                char *nick = xml_attr( attr_names, attr_values, "nick" );
87                char *pass = xml_attr( attr_names, attr_values, "password" );
88                int st;
89               
90                if( !nick || !pass )
91                {
92                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
93                                     "Missing attributes for %s element", element_name );
94                }
95                else if( ( st = md5_verify_password( xd->given_pass, pass ) ) == -1 )
96                {
97                        xd->pass_st = XML_PASS_WRONG;
98                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
99                                     "Error while decoding password attribute" );
100                }
101                else if( st == 0 )
102                {
103                        if( xd->pass_st != XML_PASS_CHECK_ONLY )
104                                xd->pass_st = XML_PASS_OK;
105                }
106                else
107                {
108                        xd->pass_st = XML_PASS_WRONG;
109                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
110                                     "Password mismatch" );
111                }
112        }
113        else if( xd->pass_st < XML_PASS_OK )
114        {
115                /* Let's not parse anything else if we only have to check
116                   the password. */
117        }
118        else if( g_strcasecmp( element_name, "account" ) == 0 )
119        {
120                char *protocol, *handle, *server, *password = NULL, *autoconnect;
121                char *pass_b64 = NULL;
122                unsigned char *pass_cr = NULL;
123                int pass_len;
124                struct prpl *prpl = NULL;
125               
126                handle = xml_attr( attr_names, attr_values, "handle" );
127                pass_b64 = xml_attr( attr_names, attr_values, "password" );
128                server = xml_attr( attr_names, attr_values, "server" );
129                autoconnect = xml_attr( attr_names, attr_values, "autoconnect" );
130               
131                protocol = xml_attr( attr_names, attr_values, "protocol" );
132                if( protocol )
133                        prpl = find_protocol( protocol );
134               
135                if( !handle || !pass_b64 || !protocol )
136                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
137                                     "Missing attributes for %s element", element_name );
138                else if( !prpl )
139                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
140                                     "Unknown protocol: %s", protocol );
141                else if( ( pass_len = base64_decode( pass_b64, (unsigned char**) &pass_cr ) ) &&
142                                         arc_decode( pass_cr, pass_len, &password, xd->given_pass ) )
143                {
144                        xd->current_account = account_add( irc, prpl, handle, password );
145                        if( server )
146                                set_setstr( &xd->current_account->set, "server", server );
147                        if( autoconnect )
148                                set_setstr( &xd->current_account->set, "auto_connect", autoconnect );
149                }
150                else
151                {
152                        /* Actually the _decode functions don't even return error codes,
153                           but maybe they will later... */
154                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
155                                     "Error while decrypting account password" );
156                }
157               
158                g_free( pass_cr );
159                g_free( password );
160        }
161        else if( g_strcasecmp( element_name, "setting" ) == 0 )
162        {
163                char *setting;
164               
165                if( xd->current_setting )
166                {
167                        g_free( xd->current_setting );
168                        xd->current_setting = NULL;
169                }
170               
171                if( ( setting = xml_attr( attr_names, attr_values, "name" ) ) )
172                        xd->current_setting = g_strdup( setting );
173                else
174                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
175                                     "Missing attributes for %s element", element_name );
176        }
177        else if( g_strcasecmp( element_name, "buddy" ) == 0 )
178        {
179                char *handle, *nick;
180               
181                handle = xml_attr( attr_names, attr_values, "handle" );
182                nick = xml_attr( attr_names, attr_values, "nick" );
183               
184                if( xd->current_account && handle && nick )
185                {
186                        nick_set( xd->current_account, handle, nick );
187                }
188                else
189                {
190                        g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
191                                     "Missing attributes for %s element", element_name );
192                }
193        }
194        else
195        {
196                g_set_error( error, G_MARKUP_ERROR, G_MARKUP_ERROR_UNKNOWN_ELEMENT,
197                             "Unkown element: %s", element_name );
198        }
199}
200
201static void xml_end_element( GMarkupParseContext *ctx, const gchar *element_name, gpointer data, GError **error )
202{
203        struct xml_parsedata *xd = data;
204       
205        if( g_strcasecmp( element_name, "setting" ) == 0 && xd->current_setting )
206        {
207                g_free( xd->current_setting );
208                xd->current_setting = NULL;
209        }
210        else if( g_strcasecmp( element_name, "account" ) == 0 )
211        {
212                xd->current_account = NULL;
213        }
214}
215
216static void xml_text( GMarkupParseContext *ctx, const gchar *text_orig, gsize text_len, gpointer data, GError **error )
217{
218        char text[text_len+1];
219        struct xml_parsedata *xd = data;
220        irc_t *irc = xd->irc;
221       
222        strncpy( text, text_orig, text_len );
223        text[text_len] = 0;
224       
225        if( xd->pass_st < XML_PASS_OK )
226        {
227                /* Let's not parse anything else if we only have to check
228                   the password, or if we didn't get the chance to check it
229                   yet. */
230        }
231        else if( g_strcasecmp( g_markup_parse_context_get_element( ctx ), "setting" ) == 0 && xd->current_setting )
232        {
233                set_setstr( xd->current_account ? &xd->current_account->set : &irc->set,
234                            xd->current_setting, (char*) text );
235                g_free( xd->current_setting );
236                xd->current_setting = NULL;
237        }
238}
239
240GMarkupParser xml_parser =
241{
242        xml_start_element,
243        xml_end_element,
244        xml_text,
245        NULL,
246        NULL
247};
248
249static void xml_init( void )
250{
251        if( ! g_file_test( global.conf->configdir, G_FILE_TEST_EXISTS ) )
252                log_message( LOGLVL_WARNING, "The configuration directory `%s' does not exist. Configuration won't be saved.", global.conf->configdir );
253        else if( ! g_file_test( global.conf->configdir, G_FILE_TEST_EXISTS ) || 
254                         g_access( global.conf->configdir, W_OK ) != 0 )
255                log_message( LOGLVL_WARNING, "Permission problem: Can't read/write from/to `%s'.", global.conf->configdir );
256}
257
258static storage_status_t xml_load_real( const char *my_nick, const char *password, irc_t *irc, xml_pass_st action )
259{
260        GMarkupParseContext *ctx;
261        struct xml_parsedata *xd;
262        char *fn, buf[512];
263        GError *gerr = NULL;
264        int fd, st;
265       
266        if( irc && irc->status & USTATUS_IDENTIFIED )
267                return( 1 );
268       
269        xd = g_new0( struct xml_parsedata, 1 );
270        xd->irc = irc;
271        xd->given_nick = g_strdup( my_nick );
272        xd->given_pass = g_strdup( password );
273        xd->pass_st = action;
274        nick_lc( xd->given_nick );
275       
276        fn = g_strdup_printf( "%s%s%s", global.conf->configdir, xd->given_nick, ".xml" );
277        if( ( fd = open( fn, O_RDONLY ) ) < 0 )
278        {
279                xml_destroy_xd( xd );
280                g_free( fn );
281                return STORAGE_NO_SUCH_USER;
282        }
283        g_free( fn );
284       
285        ctx = g_markup_parse_context_new( &xml_parser, 0, xd, xml_destroy_xd );
286       
287        while( ( st = read( fd, buf, sizeof( buf ) ) ) > 0 )
288        {
289                if( !g_markup_parse_context_parse( ctx, buf, st, &gerr ) || gerr )
290                {
291                        xml_pass_st pass_st = xd->pass_st;
292                       
293                        g_markup_parse_context_free( ctx );
294                        close( fd );
295                       
296                        if( pass_st == XML_PASS_WRONG )
297                        {
298                                g_clear_error( &gerr );
299                                return STORAGE_INVALID_PASSWORD;
300                        }
301                        else
302                        {
303                                if( gerr && irc )
304                                        irc_usermsg( irc, "Error from XML-parser: %s", gerr->message );
305                               
306                                g_clear_error( &gerr );
307                                return STORAGE_OTHER_ERROR;
308                        }
309                }
310        }
311        /* Just to be sure... */
312        g_clear_error( &gerr );
313       
314        g_markup_parse_context_free( ctx );
315        close( fd );
316       
317        if( action == XML_PASS_CHECK_ONLY )
318                return STORAGE_OK;
319       
320        irc->status |= USTATUS_IDENTIFIED;
321       
322        return STORAGE_OK;
323}
324
325static storage_status_t xml_load( const char *my_nick, const char *password, irc_t *irc )
326{
327        return xml_load_real( my_nick, password, irc, XML_PASS_UNKNOWN );
328}
329
330static storage_status_t xml_check_pass( const char *my_nick, const char *password )
331{
332        /* This is a little bit risky because we have to pass NULL for the
333           irc_t argument. This *should* be fine, if I didn't miss anything... */
334        return xml_load_real( my_nick, password, NULL, XML_PASS_CHECK_ONLY );
335}
336
337static int xml_printf( int fd, int indent, char *fmt, ... )
338{
339        va_list params;
340        char *out;
341        char tabs[9] = "\t\t\t\t\t\t\t\t";
342        int len;
343       
344        /* Maybe not very clean, but who needs more than 8 levels of indentation anyway? */
345        if( write( fd, tabs, indent <= 8 ? indent : 8 ) != indent )
346                return 0;
347       
348        va_start( params, fmt );
349        out = g_markup_vprintf_escaped( fmt, params );
350        va_end( params );
351       
352        len = strlen( out );
353        len -= write( fd, out, len );
354        g_free( out );
355       
356        return len == 0;
357}
358
359static gboolean xml_save_nick( gpointer key, gpointer value, gpointer data );
360
361static storage_status_t xml_save( irc_t *irc, int overwrite )
362{
363        char path[512], *path2, *pass_buf = NULL;
364        set_t *set;
365        account_t *acc;
366        int fd;
367        md5_byte_t pass_md5[21];
368        md5_state_t md5_state;
369       
370        if( irc->password == NULL )
371        {
372                irc_usermsg( irc, "Please register yourself if you want to save your settings." );
373                return STORAGE_OTHER_ERROR;
374        }
375       
376        path2 = g_strdup( irc->nick );
377        nick_lc( path2 );
378        g_snprintf( path, sizeof( path ) - 2, "%s%s%s", global.conf->configdir, path2, ".xml" );
379        g_free( path2 );
380       
381        if( !overwrite && g_file_test( path, G_FILE_TEST_EXISTS ) )
382                return STORAGE_ALREADY_EXISTS;
383       
384        strcat( path, "~" );
385        if( ( fd = open( path, O_WRONLY | O_CREAT | O_TRUNC, 0600 ) ) < 0 )
386        {
387                irc_usermsg( irc, "Error while opening configuration file." );
388                return STORAGE_OTHER_ERROR;
389        }
390       
391        /* Generate a salted md5sum of the password. Use 5 bytes for the salt
392           (to prevent dictionary lookups of passwords) to end up with a 21-
393           byte password hash, more convenient for base64 encoding. */
394        random_bytes( pass_md5 + 16, 5 );
395        md5_init( &md5_state );
396        md5_append( &md5_state, (md5_byte_t*) irc->password, strlen( irc->password ) );
397        md5_append( &md5_state, pass_md5 + 16, 5 ); /* Add the salt. */
398        md5_finish( &md5_state, pass_md5 );
399        /* Save the hash in base64-encoded form. */
400        pass_buf = base64_encode( pass_md5, 21 );
401       
402        if( !xml_printf( fd, 0, "<user nick=\"%s\" password=\"%s\" version=\"%d\">\n", irc->nick, pass_buf, XML_FORMAT_VERSION ) )
403                goto write_error;
404       
405        g_free( pass_buf );
406       
407        for( set = irc->set; set; set = set->next )
408                if( set->value && set->def )
409                        if( !xml_printf( fd, 1, "<setting name=\"%s\">%s</setting>\n", set->key, set->value ) )
410                                goto write_error;
411       
412        for( acc = irc->accounts; acc; acc = acc->next )
413        {
414                unsigned char *pass_cr;
415                char *pass_b64;
416                int pass_len;
417               
418                pass_len = arc_encode( acc->pass, strlen( acc->pass ), (unsigned char**) &pass_cr, irc->password, 12 );
419                pass_b64 = base64_encode( pass_cr, pass_len );
420                g_free( pass_cr );
421               
422                if( !xml_printf( fd, 1, "<account protocol=\"%s\" handle=\"%s\" password=\"%s\" autoconnect=\"%d\"", acc->prpl->name, acc->user, pass_b64, acc->auto_connect ) )
423                {
424                        g_free( pass_b64 );
425                        goto write_error;
426                }
427                g_free( pass_b64 );
428               
429                if( acc->server && acc->server[0] && !xml_printf( fd, 0, " server=\"%s\"", acc->server ) )
430                        goto write_error;
431                if( !xml_printf( fd, 0, ">\n" ) )
432                        goto write_error;
433               
434                for( set = acc->set; set; set = set->next )
435                        if( set->value && set->def && !( set->flags & ACC_SET_NOSAVE ) )
436                                if( !xml_printf( fd, 2, "<setting name=\"%s\">%s</setting>\n", set->key, set->value ) )
437                                        goto write_error;
438               
439                /* This probably looks pretty strange. g_hash_table_foreach
440                   is quite a PITA already (but it can't get much better in
441                   C without using #define, I'm afraid), and since it
442                   doesn't seem to be possible to abort the foreach on write
443                   errors, so instead let's use the _find function and
444                   return TRUE on write errors. Which means, if we found
445                   something, there was an error. :-) */
446                if( g_hash_table_find( acc->nicks, xml_save_nick, & fd ) )
447                        goto write_error;
448               
449                if( !xml_printf( fd, 1, "</account>\n" ) )
450                        goto write_error;
451        }
452       
453        if( !xml_printf( fd, 0, "</user>\n" ) )
454                goto write_error;
455       
456        close( fd );
457       
458        path2 = g_strndup( path, strlen( path ) - 1 );
459        if( rename( path, path2 ) != 0 )
460        {
461                irc_usermsg( irc, "Error while renaming temporary configuration file." );
462               
463                g_free( path2 );
464                unlink( path );
465               
466                return STORAGE_OTHER_ERROR;
467        }
468       
469        g_free( path2 );
470       
471        return STORAGE_OK;
472
473write_error:
474        g_free( pass_buf );
475       
476        irc_usermsg( irc, "Write error. Disk full?" );
477        close( fd );
478       
479        return STORAGE_OTHER_ERROR;
480}
481
482static gboolean xml_save_nick( gpointer key, gpointer value, gpointer data )
483{
484        return !xml_printf( *( (int*) data ), 2, "<buddy handle=\"%s\" nick=\"%s\" />\n", key, value );
485}
486
487static storage_status_t xml_remove( const char *nick, const char *password )
488{
489        char s[512], *lc;
490        storage_status_t status;
491
492        status = xml_check_pass( nick, password );
493        if( status != STORAGE_OK )
494                return status;
495
496        lc = g_strdup( nick );
497        nick_lc( lc );
498        g_snprintf( s, 511, "%s%s%s", global.conf->configdir, lc, ".xml" );
499        g_free( lc );
500       
501        if( unlink( s ) == -1 )
502                return STORAGE_OTHER_ERROR;
503       
504        return STORAGE_OK;
505}
506
507storage_t storage_xml = {
508        .name = "xml",
509        .init = xml_init,
510        .check_pass = xml_check_pass,
511        .remove = xml_remove,
512        .load = xml_load,
513        .save = xml_save
514};
Note: See TracBrowser for help on using the repository browser.