2 Unix SMB/CIFS implementation.
3 low level tdb backup and restore utility
4 Copyright (C) Andrew Tridgell 2002
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23 This program is meant for backup/restore of tdb databases. Typical usage would be:
25 when Samba shuts down cleanly, which will make a backup of all the local databases
26 to *.bak files. Then on Samba startup you would use:
28 and this will check the databases for corruption and if corruption is detected then
29 the backup will be restored.
31 You may also like to do a backup on a regular basis while Samba is
32 running, perhaps using cron.
34 The reason this program is needed is to cope with power failures
35 while Samba is running. A power failure could lead to database
36 corruption and Samba will then not start correctly.
38 Note that many of the databases in Samba are transient and thus
39 don't need to be backed up, so you can optimise the above a little
40 by only running the backup on the critical databases.
45 #include "system/locale.h"
46 #include "system/time.h"
47 #include "system/filesys.h"
56 static char *add_suffix(const char *name, const char *suffix)
59 int len = strlen(name) + strlen(suffix) + 1;
60 ret = (char *)malloc(len);
62 fprintf(stderr,"Out of memory!\n");
65 snprintf(ret, len, "%s%s", name, suffix);
69 static int copy_fn(TDB_CONTEXT *tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
71 TDB_CONTEXT *tdb_new = (TDB_CONTEXT *)state;
73 if (tdb_store(tdb_new, key, dbuf, TDB_INSERT) != 0) {
74 fprintf(stderr,"Failed to insert into %s\n", tdb_name(tdb));
82 static int test_fn(TDB_CONTEXT *tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
88 carefully backup a tdb, validating the contents and
89 only doing the backup if its OK
90 this function is also used for restore
92 static int backup_tdb(const char *old_name, const char *new_name, int hash_size)
100 tmp_name = add_suffix(new_name, ".tmp");
102 /* stat the old tdb to find its permissions */
103 if (stat(old_name, &st) != 0) {
109 /* open the old tdb */
110 tdb = tdb_open(old_name, 0, 0, O_RDWR, 0);
112 printf("Failed to open %s\n", old_name);
117 /* create the new tdb */
119 tdb_new = tdb_open(tmp_name,
120 hash_size ? hash_size : tdb_hash_size(tdb),
121 TDB_DEFAULT, O_RDWR|O_CREAT|O_EXCL,
129 /* lock the old tdb */
130 if (tdb_lockall(tdb) != 0) {
131 fprintf(stderr,"Failed to lock %s\n", old_name);
141 /* traverse and copy */
142 count1 = tdb_traverse(tdb, copy_fn, (void *)tdb_new);
143 if (count1 < 0 || failed) {
144 fprintf(stderr,"failed to copy %s\n", old_name);
152 /* close the old tdb */
155 /* close the new tdb and re-open read-only */
157 tdb_new = tdb_open(tmp_name, 0, TDB_DEFAULT, O_RDONLY, 0);
159 fprintf(stderr,"failed to reopen %s\n", tmp_name);
166 /* traverse the new tdb to confirm */
167 count2 = tdb_traverse(tdb_new, test_fn, 0);
168 if (count2 != count1) {
169 fprintf(stderr,"failed to copy %s\n", old_name);
176 /* make sure the new tdb has reached stable storage */
177 fsync(tdb_fd(tdb_new));
179 /* close the new tdb and rename it to .bak */
182 if (rename(tmp_name, new_name) != 0) {
194 verify a tdb and if it is corrupt then restore from *.bak
196 static int verify_tdb(const char *fname, const char *bak_name)
202 tdb = tdb_open(fname, 0, 0, O_RDONLY, 0);
204 /* traverse the tdb, then close it */
206 count = tdb_traverse(tdb, test_fn, NULL);
210 /* count is < 0 means an error */
212 printf("restoring %s\n", fname);
213 return backup_tdb(bak_name, fname, 0);
216 printf("%s : %d records\n", fname, count);
222 see if one file is newer than another
224 static int file_newer(const char *fname1, const char *fname2)
226 struct stat st1, st2;
227 if (stat(fname1, &st1) != 0) {
230 if (stat(fname2, &st2) != 0) {
233 return (st1.st_mtime > st2.st_mtime);
236 static void usage(void)
238 printf("Usage: tdbbackup [options] <fname...>\n\n");
239 printf(" -h this help message\n");
240 printf(" -s suffix set the backup suffix\n");
241 printf(" -v verify mode (restore if corrupt)\n");
242 printf(" -n hashsize set the new hash size for the backup\n");
246 int main(int argc, char *argv[])
253 const char *suffix = ".bak";
255 while ((c = getopt(argc, argv, "vhs:n:")) != -1) {
267 hashsize = atoi(optarg);
280 for (i=0; i<argc; i++) {
281 const char *fname = argv[i];
284 bak_name = add_suffix(fname, suffix);
287 if (verify_tdb(fname, bak_name) != 0) {
291 if (file_newer(fname, bak_name) &&
292 backup_tdb(fname, bak_name, hashsize) != 0) {