source: storage_xml.c @ 88d2208

Last change on this file since 88d2208 was 88d2208, checked in by Wilmer van der Gaast <wilmer@…>, at 2008-08-24T16:45:40Z

Now really fixing #429 by including unistd.h (which defines F_OK).

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