etc/cnid_dbd/cmd_dbd.c

   1 /*
   2    $Id: cmd_dbd.c,v 1.3 2009-05-22 20:48:44 franklahm Exp $
   3
   4    Copyright (c) 2009 Frank Lahm <franklahm@gmail.com>
   5
   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 2 of the License, or
   9    (at your option) any later version.
  10
  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.
  15 */
  16
  17 /*
  18   dbd specs and implementation progress
  19   =====================================
  20
  21   St := Status
  22
  23   Force option
  24   ------------
  25
  26   St Spec
  27   -- ----
  28   OK If -f is requested, ensure -e is too.
  29      Check if volumes is using AFPVOL_CACHE, then wipe db from disk. Rebuild from ad-files.
  30
  31   1st pass: Scan volume
  32   --------------------
  33
  34   St Type Check
  35   -- ---- -----
  36   OK F/D  Make sure ad file exists
  37   OK D    Make sure .AppleDouble dir exist, create if missing. Error creating
  38           it is fatal as that shouldn't happen as root.
  39   OK F/D  Delete orphaned ad-files, log dirs in ad-dir
  40   OK F/D  Check name encoding by roundtripping, log on error
  41   OK F/D  try: read CNID from ad file (if cnid caching is on)
  42           try: fetch CNID from database
  43           -> on mismatch: use CNID from file, update database (deleting both found CNIDs first)
  44           -> if no CNID in ad file: write CNID from database to ad file
  45           -> if no CNID in database: add CNID from ad file to database
  46           -> on no CNID at all: create one and store in both places
  47   OK F/D  Add found CNID, DID, filename, dev/inode, stamp to rebuild database
  48   OK F/D  Check/update stamp (implicitly done while checking CNIDs)
  49
  50
  51   2nd pass: Delete unused CNIDs
  52   -----------------------------
  53
  54   St Spec
  55   -- ----
  56   OK Step through dbd (the one on disk) and rebuild-db from pass 1 and delete any CNID from
  57      dbd not in rebuild db. This in only done in exclusive mode.
  58 */
  59
  60 #ifdef HAVE_CONFIG_H
  61 #include "config.h"
  62 #endif /* HAVE_CONFIG_H */
  63
  64 #include <unistd.h>
  65 #include <sys/types.h>
  66 #include <stdlib.h>
  67 #include <stdio.h>
  68 #include <stdarg.h>
  69 #include <limits.h>
  70 #include <signal.h>
  71 #include <string.h>
  72 #include <errno.h>
  73
  74 #include <atalk/logger.h>
  75 #include <atalk/cnid_dbd_private.h>
  76 #include <atalk/volinfo.h>
  77 #include "cmd_dbd.h"
  78 #include "dbd.h"
  79 #include "dbif.h"
  80 #include "db_param.h"
  81
  82 #define LOCKFILENAME  "lock"
  83 #define DBOPTIONS (DB_CREATE | DB_INIT_LOCK | DB_INIT_LOG | DB_INIT_MPOOL | DB_INIT_TXN)
  84
  85 static DBD *dbd;
  86
  87 volatile sig_atomic_t alarmed;
  88 static int verbose;             /* Logging flag */
  89 static int exclusive;           /* Exclusive volume access */
  90 static struct db_param db_param = {
  91     NULL,                       /* Volume dirpath */
  92     1,                          /* bdb logfile autoremove */
  93     16384,                      /* bdb cachesize */
  94     -1,                         /* not used ... */
  95     -1,
  96     "",
  97     -1,
  98     -1,
  99     -1
 100 };
 101
 102 /*
 103    Provide some logging
 104  */
 105 void dbd_log(enum logtype lt, char *fmt, ...)
 106 {
 107     int len;
 108     static char logbuffer[1024];
 109     va_list args;
 110
 111     if ( (lt == LOGSTD) || (verbose == 1)) {
 112         va_start(args, fmt);
 113         len = vsnprintf(logbuffer, 1023, fmt, args);
 114         va_end(args);
 115         logbuffer[1023] = 0;
 116
 117         printf("%s\n", logbuffer);
 118     }
 119 }
 120
 121 /*
 122    SIGNAL handling:
 123    catch SIGINT and SIGTERM which cause clean exit. Ignore anything else.
 124  */
 125
 126 static void sig_handler(int signo)
 127 {
 128     alarmed = 1;
 129     return;
 130 }
 131
 132 void set_signal(void)
 133 {
 134     struct sigaction sv;
 135
 136     sv.sa_handler = sig_handler;
 137     sv.sa_flags = SA_RESTART;
 138     sigemptyset(&sv.sa_mask);
 139     if (sigaction(SIGTERM, &sv, NULL) < 0) {
 140         dbd_log( LOGSTD, "error in sigaction(SIGTERM): %s", strerror(errno));
 141         exit(EXIT_FAILURE);
 142     }
 143     if (sigaction(SIGINT, &sv, NULL) < 0) {
 144         dbd_log( LOGSTD, "error in sigaction(SIGINT): %s", strerror(errno));
 145         exit(EXIT_FAILURE);
 146     }
 147
 148     memset(&sv, 0, sizeof(struct sigaction));
 149     sv.sa_handler = SIG_IGN;
 150     sigemptyset(&sv.sa_mask);
 151
 152     if (sigaction(SIGABRT, &sv, NULL) < 0) {
 153         dbd_log( LOGSTD, "error in sigaction(SIGABRT): %s", strerror(errno));
 154         exit(EXIT_FAILURE);
 155     }
 156     if (sigaction(SIGHUP, &sv, NULL) < 0) {
 157         dbd_log( LOGSTD, "error in sigaction(SIGHUP): %s", strerror(errno));
 158         exit(EXIT_FAILURE);
 159     }
 160     if (sigaction(SIGQUIT, &sv, NULL) < 0) {
 161         dbd_log( LOGSTD, "error in sigaction(SIGQUIT): %s", strerror(errno));
 162         exit(EXIT_FAILURE);
 163     }
 164 }
 165
 166 int get_lock(const char *dbpath)
 167 {
 168     int lockfd;
 169     char lockpath[PATH_MAX];
 170     struct flock lock;
 171     struct stat st;
 172
 173     if ( (strlen(dbpath) + strlen(LOCKFILENAME+1)) > (PATH_MAX - 1) ) {
 174         dbd_log( LOGSTD, ".AppleDB pathname too long");
 175         exit(EXIT_FAILURE);
 176     }
 177     strncpy(lockpath, dbpath, PATH_MAX - 1);
 178     strcat(lockpath, "/");
 179     strcat(lockpath, LOCKFILENAME);
 180
 181     if ((lockfd = open(lockpath, O_RDWR | O_CREAT, 0644)) < 0) {
 182         dbd_log( LOGSTD, "Error opening lockfile: %s", strerror(errno));
 183         exit(EXIT_FAILURE);
 184     }
 185
 186     if ((stat(dbpath, &st)) != 0) {
 187         dbd_log( LOGSTD, "Error statting lockfile: %s", strerror(errno));
 188         exit(EXIT_FAILURE);
 189     }
 190
 191     if ((chown(lockpath, st.st_uid, st.st_gid)) != 0) {
 192         dbd_log( LOGSTD, "Error inheriting lockfile permissions: %s", strerror(errno));
 193         exit(EXIT_FAILURE);
 194     }
 195
 196     lock.l_start  = 0;
 197     lock.l_whence = SEEK_SET;
 198     lock.l_len    = 0;
 199     lock.l_type   = F_WRLCK;
 200
 201     if (fcntl(lockfd, F_SETLK, &lock) < 0) {
 202         if (errno == EACCES || errno == EAGAIN) {
 203             if (exclusive) {
 204                 dbd_log( LOGSTD, "Database is in use and exlusive was requested", strerror(errno));
 205                 exit(EXIT_FAILURE);
 206             };
 207         } else {
 208             dbd_log( LOGSTD, "Error getting fcntl F_WRLCK on lockfile: %s", strerror(errno));
 209             exit(EXIT_FAILURE);
 210        }
 211     }
 212
 213     return lockfd;
 214 }
 215
 216 void free_lock(int lockfd)
 217 {
 218     struct flock lock;
 219
 220     lock.l_start  = 0;
 221     lock.l_whence = SEEK_SET;
 222     lock.l_len    = 0;
 223     lock.l_type = F_UNLCK;
 224     fcntl(lockfd, F_SETLK, &lock);
 225     close(lockfd);
 226 }
 227
 228 static void usage ()
 229 {
 230     printf("Usage: dbd [-e|-v|-x] -d [-i] | -s | -r [-f] <path to netatalk volume>\n"
 231            "dbd can dump, scan, reindex and rebuild Netatalk dbd CNID databases.\n"
 232            "dbd must be run with appropiate permissions i.e. as root.\n\n"
 233            "Main commands are:\n"
 234            "   -d Dump CNID database\n"
 235            "      Option: -i dump indexes too\n"
 236            "   -s Scan volume:\n"
 237            "      1. Compare database with volume\n"
 238            "      2. Check if .AppleDouble dirs exist\n"
 239            "      3. Check if  AppleDouble file exist\n"
 240            "      4. Report orphaned AppleDouble files\n"
 241            "      5. Check for directories inside AppleDouble directories\n"
 242            "      6. Check name encoding by roundtripping, log on error\n"
 243            "   -r Rebuild volume:\n"
 244            "      1. Sync database with volume\n"
 245            "      2. Make sure .AppleDouble dir exist, create if missing\n"
 246            "      3. Make sure AppleDouble file exists, create if missing\n"
 247            "      4. Delete orphaned AppleDouble files\n"
 248            "      5. Check for directories inside AppleDouble directories\n"
 249            "      6. Check name encoding by roundtripping, log on error\n"
 250            "      Option: -f wipe database and rebuild from IDs stored in AppleDouble files,\n"
 251            "                 only available for volumes with 'cachecnid' option.\n"
 252            "                 Implies -e."
 253            "General options:\n"
 254            "   -e only work on inactive volumes and lock them (exclusive)\n"
 255            "   -x rebuild indexes (just for completeness, mostly useless!)\n"
 256            "   -v verbose\n"
 257         );
 258 }
 259
 260 int main(int argc, char **argv)
 261 {
 262     int c, lockfd;
 263     int dump=0, scan=0, rebuild=0, rebuildindexes=0, dumpindexes=0, force=0;
 264     dbd_flags_t flags = 0;
 265     char *volpath;
 266     struct volinfo volinfo;
 267
 268     if (geteuid() != 0) {
 269         usage();
 270         exit(EXIT_FAILURE);
 271     }
 272
 273     while ((c = getopt(argc, argv, ":dsrvxife")) != -1) {
 274         switch(c) {
 275         case 'd':
 276             dump = 1;
 277             break;
 278         case 'i':
 279             dumpindexes = 1;
 280             break;
 281         case 's':
 282             scan = 1;
 283             flags |= DBD_FLAGS_SCAN;
 284             break;
 285         case 'r':
 286             rebuild = 1;
 287             break;
 288         case 'v':
 289             verbose = 1;
 290             break;
 291         case 'e':
 292             exclusive = 1;
 293             flags |= DBD_FLAGS_EXCL;
 294             break;
 295         case 'x':
 296             rebuildindexes = 1;
 297             break;
 298         case 'f':
 299             force = 1;
 300             flags |= DBD_FLAGS_FORCE | DBD_FLAGS_EXCL;
 301             break;
 302         case ':':
 303         case '?':
 304             usage();
 305             exit(EXIT_FAILURE);
 306             break;
 307         }
 308     }
 309
 310     if ((dump + scan + rebuild) != 1) {
 311         usage();
 312         exit(EXIT_FAILURE);
 313     }
 314
 315     if ( (optind + 1) != argc ) {
 316         usage();
 317         exit(EXIT_FAILURE);
 318     }
 319     volpath = argv[optind];
 320
 321     /* Put "/.AppleDB" at end of volpath */
 322     if ( (strlen(volpath) + strlen("/.AppleDB")) > (PATH_MAX - 1) ) {
 323         dbd_log( LOGSTD, "Volume pathname too long");
 324         exit(EXIT_FAILURE);
 325     }
 326     char dbpath[PATH_MAX];
 327     strncpy(dbpath, volpath, PATH_MAX - 1);
 328     strcat(dbpath, "/.AppleDB");
 329
 330     /* Remember cwd */
 331     int cdir;
 332     if ((cdir = open(".", O_RDONLY)) < 0) {
 333         dbd_log( LOGSTD, "Can't open dir: %s", strerror(errno));
 334         exit(EXIT_FAILURE);
 335     }
 336
 337     /*
 338        Before we do anything else, check if there is an instance of cnid_dbd
 339        running already and silently exit if yes.
 340     */
 341     lockfd = get_lock(dbpath);
 342
 343     /* Setup signal handling */
 344     set_signal();
 345
 346     /* Setup logging. Should be portable among *NIXes */
 347     if (!verbose)
 348         setuplog("default log_info /dev/tty");
 349     else
 350         setuplog("default log_debug /dev/tty");
 351
 352     /* Load .volinfo file */
 353     if (loadvolinfo(volpath, &volinfo) == -1) {
 354         dbd_log( LOGSTD, "Unkown volume options!");
 355         exit(EXIT_FAILURE);
 356     }
 357     if (vol_load_charsets(&volinfo) == -1) {
 358         dbd_log( LOGSTD, "Error loading charsets!");
 359         exit(EXIT_FAILURE);
 360     }
 361
 362     /* Check if -f is requested and wipe db if yes */
 363     if ((flags & DBD_FLAGS_FORCE) && (volinfo.v_flags & AFPVOL_CACHE)) {
 364         char cmd[8 + MAXPATHLEN];
 365         snprintf(cmd, 8 + MAXPATHLEN, "rm -f %s/cnid2.db", dbpath);
 366         dbd_log( LOGSTD, "Removing old database of volume: '%s'", volpath);
 367         system(cmd);
 368         dbd_log( LOGSTD, "Removed old database.");
 369     }
 370
 371     /*
 372        Lets start with the BerkeleyDB stuff
 373     */
 374     if ((dbd = dbif_init(dbpath, "cnid2.db")) == NULL)
 375         exit(EXIT_FAILURE);
 376
 377     if (dbif_env_open(dbd, &db_param, exclusive ? (DBOPTIONS | DB_RECOVER) : DBOPTIONS) < 0) {
 378         dbd_log( LOGSTD, "error opening database!");
 379         exit(EXIT_FAILURE);
 380     }
 381
 382     if (exclusive)
 383         dbd_log( LOGDEBUG, "Finished recovery.");
 384
 385     if (dbif_open(dbd, &db_param, rebuildindexes) < 0) {
 386         dbif_close(dbd);
 387         exit(EXIT_FAILURE);
 388     }
 389
 390     if (dbd_stamp(dbd) < 0) {
 391         dbif_close(dbd);
 392         exit(EXIT_FAILURE);
 393     }
 394
 395     if (dump) {
 396         if (dbif_dump(dbd, dumpindexes) < 0) {
 397             dbd_log( LOGSTD, "Error dumping database");
 398         }
 399     } else if (rebuild || scan) {
 400         if (cmd_dbd_scanvol(dbd, &volinfo, flags) < 0) {
 401             dbd_log( LOGSTD, "Error repairing database.");
 402         }
 403     }
 404
 405     if (dbif_close(dbd) < 0) {
 406         dbd_log( LOGSTD, "Error closing database");
 407         exit(EXIT_FAILURE);
 408     }
 409
 410     free_lock(lockfd);
 411
 412     if ((fchdir(cdir)) < 0)
 413         dbd_log(LOGSTD, "fchdir: %s", strerror(errno));
 414
 415     return 0;
 416 }