← Back to branch summary

~alobbs/+junk/libaccounts-glib

« back to all changes in this revision

Viewing changes to tools/backup.c

  • Committer: Alberto Mardegan
  • Date: 2011-05-23 06:52:34 UTC
  • Revision ID: git-v1:6e1bf3bf3af9ba483c8ae66be8cd965ce9298bdc
Add backup tool

Show diffs side-by-side

added added

removed removed

Lines of Context:
tools/backup.c
 
1
/* vi: set et sw=4 ts=4 cino=t0,(0: */
 
2
/* -*- Mode: C; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
 
3
/*
 
4
 * This file is part of libaccounts-glib
 
5
 *
 
6
 * Copyright (C) 2011 Nokia Corporation.
 
7
 *
 
8
 * Contact: Alberto Mardegan <alberto.mardegan@nokia.com>
 
9
 *
 
10
 * This library is free software; you can redistribute it and/or
 
11
 * modify it under the terms of the GNU Lesser General Public License
 
12
 * version 2.1 as published by the Free Software Foundation.
 
13
 *
 
14
 * This library is distributed in the hope that it will be useful, but
 
15
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 
16
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 
17
 * Lesser General Public License for more details.
 
18
 *
 
19
 * You should have received a copy of the GNU Lesser General Public
 
20
 * License along with this library; if not, write to the Free Software
 
21
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 
22
 * 02110-1301 USA
 
23
 */
 
24
 
 
25
#include <glib.h>
 
26
#include <sched.h>
 
27
#include <sqlite3.h>
 
28
#include <stdio.h>
 
29
#include <stdlib.h>
 
30
#include <string.h>
 
31
 
 
32
static void
 
33
show_help ()
 
34
{
 
35
    printf ("\nUsage:\n"
 
36
            "   %1$s\n"
 
37
            "Backups the accounts from ~/.accounts/accounts.db into\n"
 
38
            "~/.accounts/accounts.db.bak\n\n",
 
39
            g_get_prgname());
 
40
}
 
41
 
 
42
static gboolean
 
43
write_backup (sqlite3 *src, const gchar *filename)
 
44
{
 
45
    sqlite3_backup *backup;
 
46
    sqlite3 *dest;
 
47
    gint n_retries;
 
48
    int ret;
 
49
 
 
50
    ret = sqlite3_open (filename, &dest);
 
51
    if (ret != SQLITE_OK) return FALSE;
 
52
 
 
53
    backup = sqlite3_backup_init (dest, "main", src, "main");
 
54
    if (!backup)
 
55
    {
 
56
        g_warning ("Couldn't start backup");
 
57
        sqlite3_close (dest);
 
58
        return FALSE;
 
59
    }
 
60
 
 
61
    n_retries = 0;
 
62
    do
 
63
    {
 
64
        ret = sqlite3_backup_step (backup, -1);
 
65
        if (ret == SQLITE_BUSY || ret == SQLITE_LOCKED)
 
66
            sqlite3_sleep(250);
 
67
        n_retries++;
 
68
    }
 
69
    while ((ret == SQLITE_BUSY || ret == SQLITE_LOCKED) && n_retries < 5);
 
70
 
 
71
    sqlite3_backup_finish (backup);
 
72
 
 
73
    sqlite3_close (dest);
 
74
    return ret == SQLITE_OK;
 
75
}
 
76
 
 
77
static gboolean
 
78
backup ()
 
79
{
 
80
    gchar *filename, *filename_bak;
 
81
    sqlite3 *db;
 
82
    gint n_retries;
 
83
    int ret;
 
84
    gboolean success = FALSE;;
 
85
 
 
86
    filename = g_build_filename (g_get_home_dir (),
 
87
                                 ".accounts",
 
88
                                 "accounts.db",
 
89
                                 NULL);
 
90
    filename_bak = g_strdup_printf ("%s.bak", filename);
 
91
 
 
92
    g_debug ("Opening %s", filename);
 
93
 
 
94
    n_retries = 0;
 
95
    do
 
96
    {
 
97
        ret = sqlite3_open (filename, &db);
 
98
        if (ret == SQLITE_BUSY)
 
99
            sched_yield ();
 
100
        n_retries++;
 
101
    }
 
102
    while (ret == SQLITE_BUSY && n_retries < 5);
 
103
 
 
104
    if (G_UNLIKELY (ret != SQLITE_OK))
 
105
    {
 
106
        g_warning ("Couldn't open accounts DB: %s", sqlite3_errmsg (db));
 
107
        goto error_open;
 
108
    }
 
109
 
 
110
    n_retries = 0;
 
111
    do
 
112
    {
 
113
        ret = sqlite3_wal_checkpoint (db, NULL);
 
114
        if (ret == SQLITE_BUSY)
 
115
            sched_yield ();
 
116
        n_retries++;
 
117
    }
 
118
    while (ret == SQLITE_BUSY && n_retries < 5);
 
119
 
 
120
    if (G_UNLIKELY (ret != SQLITE_OK))
 
121
        g_warning ("Checkpoint failed: %s", sqlite3_errmsg (db));
 
122
 
 
123
    success = write_backup (db, filename_bak);
 
124
 
 
125
    sqlite3_close (db);
 
126
    success = TRUE;
 
127
 
 
128
error_open:
 
129
    g_free (filename_bak);
 
130
    g_free (filename);
 
131
    return success;
 
132
}
 
133
 
 
134
int
 
135
main (int argc, char **argv)
 
136
{
 
137
    gboolean success;
 
138
 
 
139
    g_set_prgname (g_path_get_basename (argv[0]));
 
140
 
 
141
    if (argc > 1)
 
142
    {
 
143
        show_help ();
 
144
        return 0;
 
145
    }
 
146
 
 
147
    success = backup ();
 
148
 
 
149
    return success ? EXIT_SUCCESS : EXIT_FAILURE;
 
150
}
 
151