source: storage_xml.c @ bb5ce568

Last change on this file since bb5ce568 was bb5ce568, checked in by Wilmer van der Gaast <wilmer@…>, at 2012-06-07T22:41:17Z

s/ACC_SET_NOSAVE/SET_NOSAVE/.

  • Property mode set to 100644
File size: 11.8 KB
RevLine 
[a312b6b]1  /********************************************************************\
2  * BitlBee -- An IRC to other IM-networks gateway                     *
3  *                                                                    *
[a338faa]4  * Copyright 2002-2012 Wilmer van der Gaast and others                *
[a312b6b]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"
[6e1fed7]28#include "base64.h"
[a7b5925]29#include "arc.h"
[d28f3b35]30#include "md5.h"
[a338faa]31#include "xmltree.h"
[a312b6b]32
[88d2208]33#include <glib/gstdio.h>
[47b571d]34
[c121f89]35typedef enum
36{
37        XML_PASS_CHECK_ONLY = -1,
38        XML_PASS_UNKNOWN = 0,
[c9f0c79]39        XML_PASS_WRONG,
[c121f89]40        XML_PASS_OK
41} xml_pass_st;
42
[c9f0c79]43/* To make it easier later when extending the format: */
[a338faa]44#define XML_FORMAT_VERSION "1"
[c121f89]45
[a312b6b]46struct xml_parsedata
47{
48        irc_t *irc;
[00f1e93]49        char given_nick[MAX_NICK_LENGTH+1];
[c121f89]50        char *given_pass;
[a312b6b]51};
52
[00f1e93]53static void xml_init( void )
[c121f89]54{
[00f1e93]55        if( g_access( global.conf->configdir, F_OK ) != 0 )
56                log_message( LOGLVL_WARNING, "The configuration directory `%s' does not exist. Configuration won't be saved.", global.conf->configdir );
57        else if( g_access( global.conf->configdir, F_OK ) != 0 || 
58                 g_access( global.conf->configdir, W_OK ) != 0 )
59                log_message( LOGLVL_WARNING, "Permission problem: Can't read/write from/to `%s'.", global.conf->configdir );
[c121f89]60}
61
[00f1e93]62static void handle_settings( struct xt_node *node, set_t **head )
[a312b6b]63{
[00f1e93]64        struct xt_node *c;
[a312b6b]65       
[00f1e93]66        for( c = node->children; ( c = xt_find_node( c, "setting" ) ); c = c->next )
[a312b6b]67        {
[00f1e93]68                char *name = xt_find_attr( c, "name" );
[a312b6b]69               
[15581c1]70                if( !name )
71                        continue;
72               
[00f1e93]73                if( strcmp( node->name, "account" ) == 0 )
[a312b6b]74                {
[00f1e93]75                        set_t *s = set_find( head, name );
76                        if( s && ( s->flags & ACC_SET_ONLINE_ONLY ) )
77                                continue; /* U can't touch this! */
[1917a1ec]78                }
[00f1e93]79                set_setstr( head, name, c->text );
[a312b6b]80        }
81}
82
[00f1e93]83static xt_status handle_account( struct xt_node *node, gpointer data )
[a312b6b]84{
[5898ef8]85        struct xml_parsedata *xd = data;
[00f1e93]86        char *protocol, *handle, *server, *password = NULL, *autoconnect, *tag;
87        char *pass_b64 = NULL;
88        unsigned char *pass_cr = NULL;
89        int pass_len;
90        struct prpl *prpl = NULL;
91        account_t *acc;
92        struct xt_node *c;
[5898ef8]93       
[00f1e93]94        handle = xt_find_attr( node, "handle" );
95        pass_b64 = xt_find_attr( node, "password" );
96        server = xt_find_attr( node, "server" );
97        autoconnect = xt_find_attr( node, "autoconnect" );
98        tag = xt_find_attr( node, "tag" );
99       
100        protocol = xt_find_attr( node, "protocol" );
101        if( protocol )
102                prpl = find_protocol( protocol );
103       
104        if( !handle || !pass_b64 || !protocol || !prpl )
105                return XT_ABORT;
106        else if( ( pass_len = base64_decode( pass_b64, (unsigned char**) &pass_cr ) ) &&
107                 arc_decode( pass_cr, pass_len, &password, xd->given_pass ) >= 0 )
108        {
109                acc = account_add( xd->irc->b, prpl, handle, password );
110                if( server )
111                        set_setstr( &acc->set, "server", server );
112                if( autoconnect )
113                        set_setstr( &acc->set, "auto_connect", autoconnect );
114                if( tag )
115                        set_setstr( &acc->set, "tag", tag );
[5898ef8]116        }
[00f1e93]117        else
118                return XT_ABORT;
119       
120        g_free( pass_cr );
121        g_free( password );
122       
123        handle_settings( node, &acc->set );
124       
125        for( c = node->children; ( c = xt_find_node( c, "buddy" ) ); c = c->next )
[1917a1ec]126        {
[00f1e93]127                char *handle, *nick;
128               
129                handle = xt_find_attr( c, "handle" );
130                nick = xt_find_attr( c, "nick" );
131               
132                if( handle && nick )
133                        nick_set_raw( acc, handle, nick );
134                else
135                        return XT_ABORT;
136        }       
137        return XT_HANDLED;
[a312b6b]138}
139
[00f1e93]140static xt_status handle_channel( struct xt_node *node, gpointer data )
[a312b6b]141{
142        struct xml_parsedata *xd = data;
[00f1e93]143        irc_channel_t *ic;
144        char *name, *type;
[a312b6b]145       
[00f1e93]146        name = xt_find_attr( node, "name" );
147        type = xt_find_attr( node, "type" );
[8320a7a]148       
[00f1e93]149        if( !name || !type )
150                return XT_ABORT;
151       
152        /* The channel may exist already, for example if it's &bitlbee.
153           Also, it's possible that the user just reconnected and the
154           IRC client already rejoined all channels it was in. They
155           should still get the right settings. */
156        if( ( ic = irc_channel_by_name( xd->irc, name ) ) ||
157            ( ic = irc_channel_new( xd->irc, name ) ) )
158                set_setstr( &ic->set, "type", type );
159       
160        handle_settings( node, &ic->set );
161       
162        return XT_HANDLED;
[a312b6b]163}
164
[00f1e93]165static const struct xt_handler_entry handlers[] = {
166        { "account", "user", handle_account, },
167        { "channel", "user", handle_channel, },
168        { NULL,      NULL,   NULL, },
[a312b6b]169};
170
[3183c21]171static storage_status_t xml_load_real( irc_t *irc, const char *my_nick, const char *password, xml_pass_st action )
[a312b6b]172{
[00f1e93]173        struct xml_parsedata xd[1];
[15581c1]174        char *fn, buf[2048];
[c121f89]175        int fd, st;
[00f1e93]176        struct xt_parser *xp;
177        struct xt_node *node;
178        storage_status_t ret = STORAGE_OTHER_ERROR;
[a312b6b]179       
[c121f89]180        xd->irc = irc;
[00f1e93]181        strncpy( xd->given_nick, my_nick, MAX_NICK_LENGTH );
182        xd->given_nick[MAX_NICK_LENGTH] = '\0';
[c121f89]183        nick_lc( xd->given_nick );
[00f1e93]184        xd->given_pass = password;
[a312b6b]185       
[00f1e93]186        fn = g_strconcat( global.conf->configdir, xd->given_nick, ".xml", NULL );
[c121f89]187        if( ( fd = open( fn, O_RDONLY ) ) < 0 )
[a312b6b]188        {
[00f1e93]189                ret = STORAGE_NO_SUCH_USER;
190                goto error;
[a312b6b]191        }
192       
[00f1e93]193        xp = xt_new( handlers, xd );
[c121f89]194        while( ( st = read( fd, buf, sizeof( buf ) ) ) > 0 )
[a312b6b]195        {
[00f1e93]196                st = xt_feed( xp, buf, st );
197                if( st != 1 )
198                        break;
199        }
200        close( fd );
201        if( st != 0 )
202                goto error;
203       
204        node = xp->root;
205        if( node == NULL || node->next != NULL || strcmp( node->name, "user" ) != 0 )
206                goto error;
207       
208        {
209                char *nick = xt_find_attr( node, "nick" );
210                char *pass = xt_find_attr( node, "password" );
211               
212                if( !nick || !pass )
213                {
214                        goto error;
215                }
216                else if( ( st = md5_verify_password( xd->given_pass, pass ) ) != 0 )
[c121f89]217                {
[00f1e93]218                        ret = STORAGE_INVALID_PASSWORD;
219                        goto error;
[c121f89]220                }
[a312b6b]221        }
[c121f89]222       
[84e9cea]223        if( action == XML_PASS_CHECK_ONLY )
[00f1e93]224        {
225                ret = STORAGE_OK;
226                goto error;
227        }
228       
229        /* DO NOT call xt_handle() before verifying the password! */
[15581c1]230        if( xt_handle( xp, NULL, 1 ) == XT_HANDLED )
[00f1e93]231                ret = STORAGE_OK;
[84e9cea]232       
[00f1e93]233        handle_settings( node, &xd->irc->b->set );
234       
235error:
[15581c1]236        xt_free( xp );
237        g_free( fn );
[00f1e93]238        return ret;
[a312b6b]239}
240
[3183c21]241static storage_status_t xml_load( irc_t *irc, const char *password )
[84e9cea]242{
[1f92a58]243        return xml_load_real( irc, irc->user->nick, password, XML_PASS_UNKNOWN );
[84e9cea]244}
245
246static storage_status_t xml_check_pass( const char *my_nick, const char *password )
247{
[3183c21]248        return xml_load_real( NULL, my_nick, password, XML_PASS_CHECK_ONLY );
[84e9cea]249}
250
[00f1e93]251
[d219296]252static gboolean xml_generate_nick( gpointer key, gpointer value, gpointer data );
[5b52a48]253
[a338faa]254struct xt_node *xml_generate( irc_t *irc )
[a312b6b]255{
[a338faa]256        char *pass_buf = NULL;
[5898ef8]257        set_t *set;
258        account_t *acc;
[6e1fed7]259        md5_byte_t pass_md5[21];
[ece2cd2]260        md5_state_t md5_state;
[547ea68]261        GSList *l;
[a338faa]262        struct xt_node *root, *cur;
[5898ef8]263       
[6e1fed7]264        /* Generate a salted md5sum of the password. Use 5 bytes for the salt
265           (to prevent dictionary lookups of passwords) to end up with a 21-
266           byte password hash, more convenient for base64 encoding. */
[1719464]267        random_bytes( pass_md5 + 16, 5 );
[ece2cd2]268        md5_init( &md5_state );
269        md5_append( &md5_state, (md5_byte_t*) irc->password, strlen( irc->password ) );
[6e1fed7]270        md5_append( &md5_state, pass_md5 + 16, 5 ); /* Add the salt. */
[ece2cd2]271        md5_finish( &md5_state, pass_md5 );
[6e1fed7]272        /* Save the hash in base64-encoded form. */
[3b6eadc]273        pass_buf = base64_encode( pass_md5, 21 );
[ece2cd2]274       
[a338faa]275        root = cur = xt_new_node( "user", NULL, NULL );
276        xt_add_attr( cur, "nick", irc->user->nick );
277        xt_add_attr( cur, "password", pass_buf );
278        xt_add_attr( cur, "version", XML_FORMAT_VERSION );
[5898ef8]279       
[6e1fed7]280        g_free( pass_buf );
281       
[1f92a58]282        for( set = irc->b->set; set; set = set->next )
[5df65bd]283                if( set->value && !( set->flags & SET_NOSAVE ) )
[a338faa]284                {
285                        struct xt_node *xset;
286                        xt_add_child( cur, xset = xt_new_node( "setting", set->value, NULL ) );
287                        xt_add_attr( xset, "name", set->key );
288                }
[5898ef8]289       
[1f92a58]290        for( acc = irc->b->accounts; acc; acc = acc->next )
[5898ef8]291        {
[a7b5925]292                unsigned char *pass_cr;
[3b6eadc]293                char *pass_b64;
[6e1fed7]294                int pass_len;
295               
[ddcf491f]296                pass_len = arc_encode( acc->pass, strlen( acc->pass ), (unsigned char**) &pass_cr, irc->password, 12 );
[a7b5925]297                pass_b64 = base64_encode( pass_cr, pass_len );
298                g_free( pass_cr );
[6e1fed7]299               
[a338faa]300                cur = xt_new_node( "account", NULL, NULL );
301                xt_add_attr( cur, "protocol", acc->prpl->name );
302                xt_add_attr( cur, "handle", acc->user );
303                xt_add_attr( cur, "password", pass_b64 );
304                xt_add_attr( cur, "autoconnect", acc->auto_connect ? "true" : "false" );
305                xt_add_attr( cur, "tag", acc->tag );
306                if( acc->server && acc->server[0] )
307                        xt_add_attr( cur, "server", acc->server );
[6e1fed7]308               
[a338faa]309                g_free( pass_b64 );
[5898ef8]310               
[5100caa]311                for( set = acc->set; set; set = set->next )
[bb5ce568]312                        if( set->value && !( set->flags & SET_NOSAVE ) )
[a338faa]313                        {
314                                struct xt_node *xset;
315                                xt_add_child( cur, xset = xt_new_node( "setting", set->value, NULL ) );
316                                xt_add_attr( xset, "name", set->key );
317                        }
[5100caa]318               
[5b52a48]319                /* This probably looks pretty strange. g_hash_table_foreach
320                   is quite a PITA already (but it can't get much better in
321                   C without using #define, I'm afraid), and since it
322                   doesn't seem to be possible to abort the foreach on write
323                   errors, so instead let's use the _find function and
324                   return TRUE on write errors. Which means, if we found
325                   something, there was an error. :-) */
[d219296]326                g_hash_table_find( acc->nicks, xml_generate_nick, cur );
[5898ef8]327               
[a338faa]328                xt_add_child( root, cur );
[5898ef8]329        }
330       
[547ea68]331        for( l = irc->channels; l; l = l->next )
332        {
333                irc_channel_t *ic = l->data;
334               
[1c40aa7]335                if( ic->flags & IRC_CHANNEL_TEMP )
336                        continue;
337               
[a338faa]338                cur = xt_new_node( "channel", NULL, NULL );
339                xt_add_attr( cur, "name", ic->name );
340                xt_add_attr( cur, "type", set_getstr( &ic->set, "type" ) );
[547ea68]341               
342                for( set = ic->set; set; set = set->next )
343                        if( set->value && strcmp( set->key, "type" ) != 0 )
[a338faa]344                        {
345                                struct xt_node *xset;
346                                xt_add_child( cur, xset = xt_new_node( "setting", set->value, NULL ) );
347                                xt_add_attr( xset, "name", set->key );
348                        }
[547ea68]349               
[a338faa]350                xt_add_child( root, cur );
351        }
352       
353        return root;
354}
355
[15581c1]356static gboolean xml_generate_nick( gpointer key, gpointer value, gpointer data )
357{
358        struct xt_node *node = xt_new_node( "buddy", NULL, NULL );
359        xt_add_attr( node, "handle", key );
360        xt_add_attr( node, "nick", value );
361        xt_add_child( (struct xt_node *) data, node );
362       
363        return FALSE;
364}
365
[a338faa]366static storage_status_t xml_save( irc_t *irc, int overwrite )
367{
[d219296]368        storage_status_t ret = STORAGE_OK;
369        char path[512], *path2 = NULL, *xml = NULL;
370        struct xt_node *tree = NULL;
371        size_t len;
[a338faa]372        int fd;
373       
374        path2 = g_strdup( irc->user->nick );
375        nick_lc( path2 );
[d219296]376        g_snprintf( path, sizeof( path ) - 20, "%s%s%s", global.conf->configdir, path2, ".xml" );
[a338faa]377        g_free( path2 );
378       
379        if( !overwrite && g_access( path, F_OK ) == 0 )
380                return STORAGE_ALREADY_EXISTS;
381       
382        strcat( path, ".XXXXXX" );
383        if( ( fd = mkstemp( path ) ) < 0 )
384        {
385                irc_rootmsg( irc, "Error while opening configuration file." );
386                return STORAGE_OTHER_ERROR;
[547ea68]387        }
388       
[a338faa]389        tree = xml_generate( irc );
[222b440]390        xml = xt_to_string_i( tree );
[d219296]391        len = strlen( xml );
392        if( write( fd, xml, len ) != len ||
393            fsync( fd ) != 0 || /* #559 */
394            close( fd ) != 0 )
395                goto error;
[5898ef8]396       
[f277225]397        path2 = g_strndup( path, strlen( path ) - 7 );
[5898ef8]398        if( rename( path, path2 ) != 0 )
399        {
400                g_free( path2 );
[d219296]401                goto error;
[5898ef8]402        }
403        g_free( path2 );
404       
[d219296]405        goto finish;
[5898ef8]406
[d219296]407error:
[e67e513]408        irc_rootmsg( irc, "Write error. Disk full?" );
[d219296]409        ret = STORAGE_OTHER_ERROR;
410
411finish: 
[5898ef8]412        close( fd );
[d219296]413        unlink( path );
414        g_free( xml );
415        xt_free_node( tree );
[5898ef8]416       
[d219296]417        return ret;
[a312b6b]418}
419
[d219296]420
[84e9cea]421static storage_status_t xml_remove( const char *nick, const char *password )
422{
[e0f9170]423        char s[512], *lc;
[84e9cea]424        storage_status_t status;
425
426        status = xml_check_pass( nick, password );
427        if( status != STORAGE_OK )
428                return status;
429
[e0f9170]430        lc = g_strdup( nick );
431        nick_lc( lc );
432        g_snprintf( s, 511, "%s%s%s", global.conf->configdir, lc, ".xml" );
433        g_free( lc );
434       
[84e9cea]435        if( unlink( s ) == -1 )
436                return STORAGE_OTHER_ERROR;
437       
438        return STORAGE_OK;
439}
440
[a312b6b]441storage_t storage_xml = {
442        .name = "xml",
443        .init = xml_init,
[84e9cea]444        .check_pass = xml_check_pass,
445        .remove = xml_remove,
[a312b6b]446        .load = xml_load,
447        .save = xml_save
448};
Note: See TracBrowser for help on using the repository browser.