pd.if.org Git - zpackage/blob - zpm-syncfs.c

   1 #define _POSIX_C_SOURCE 200809L
   2
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <sys/types.h>
   6 #include <sys/stat.h>
   7 #include <unistd.h>
   8 #include <limits.h>
   9 #include <errno.h>
  10 #include <ctype.h>
  11 #include <pwd.h>
  12 #include <grp.h>
  13 #include <math.h>
  14 #include <stdarg.h>
  15 #include <time.h>
  16
  17 /* needed for S_IFMT and AT_FDCWD */
  18 #include <fcntl.h>
  19
  20 #include <string.h>
  21
  22 #include "sqlite3.h"
  23 #include "zpm.h"
  24
  25 struct config {
  26         struct zpm *log; /* logging db will be attached as "log" */
  27         struct zpm *src;
  28         char *dbfile;
  29         char *rootdir;
  30         int errabort, errors, verbose, dryrun, conflicts;
  31         int setuser, setgroup;
  32         int reverse, exitonerror;
  33         int overwrite, absorb;
  34 };
  35
  36 struct nitem {
  37         int op;
  38         char *opstr;
  39         uid_t uid;
  40         gid_t gid;
  41         char *dest;
  42         char *path;
  43         char *hash, *ohash;
  44         char *mds, *omds;
  45         char *target;
  46         char *pkglist; /* space separated */
  47         time_t mtime;
  48         mode_t mode;
  49         int ftype;
  50         int configuration, oldwasconf;
  51         struct timespec times[2];
  52 };
  53
  54 static void usage() {
  55         printf("usage: zpm $scriptname [-fncC] args ...\n");
  56 }
  57
  58 static int seterror(struct config *conf, char *msgfmt, ...) {
  59         char msg[1024];
  60         va_list ap;
  61
  62         conf->errors++;
  63
  64         va_start(ap, msgfmt);
  65         vsnprintf(msg, sizeof msg, msgfmt, ap);
  66         va_end(ap);
  67
  68         msg[1023] = 0;
  69         if (conf->log->errmsg) {
  70                 free(conf->log->errmsg);
  71         }
  72
  73         conf->log->errmsg = strdup(msg);
  74
  75         if (conf->verbose) {
  76                 fprintf(stderr, "%s\n", msg);
  77         }
  78
  79         return conf->errabort;
  80 }
  81
  82 static int setsyserr(struct config *conf, char *msgfmt, ...) {
  83         char msg[1024];
  84         va_list ap;
  85         int printed;
  86
  87         conf->errors++;
  88
  89         va_start(ap, msgfmt);
  90         printed = vsnprintf(msg, sizeof msg, msgfmt, ap);
  91         va_end(ap);
  92
  93         if (printed < 1) {
  94                 /* nothing we can really do */
  95                 return conf->errabort;
  96         }
  97
  98         if ((size_t)printed < sizeof msg) {
  99                 snprintf(msg+printed, sizeof msg - printed, ": %s",
 100                                 strerror(errno));
 101         }
 102
 103         msg[1023] = 0;
 104         if (conf->log->errmsg) {
 105                 free(conf->log->errmsg);
 106         }
 107
 108         conf->log->errmsg = strdup(msg);
 109
 110         if (conf->verbose) {
 111                 fprintf(stderr, "%s\n", msg);
 112         }
 113
 114         return conf->errabort;
 115 }
 116
 117 static int exists(char *path, mode_t *mode) {
 118         struct stat st;
 119
 120         if (lstat(path, &st) == -1) {
 121                 return 0;
 122         }
 123         if (mode) *mode = st.st_mode;
 124         return 1;
 125 }
 126
 127 /* TODO maintain a list of already created directories */
 128 static int create_leading_dirs(char *path) {
 129         char *delim, *s;
 130         int ch = 0;
 131         char pcopy[ZPM_PATH_MAX];
 132         struct stat st;
 133
 134         strcpy(pcopy, path);
 135
 136         delim = strrchr(pcopy, '/');
 137         if (!delim) return 1; /* not an error, but no leading dirs */
 138
 139         /* cut off last component */
 140         *delim = 0;
 141
 142         s = pcopy;
 143         do {
 144                 while (*s == '/') {
 145                         s++;
 146                 }
 147
 148                 delim = strchr(s, '/');
 149                 if (delim) {
 150                         ch = *delim;
 151                         *delim = 0;
 152                 }
 153
 154                 /* try to create the directory, if it exists
 155                  * and is a directory or a symlink, that's ok
 156                  * should be (eventually) a symlink to a directory
 157                  * so we want stat here, not lstat
 158                  */
 159                 if (mkdir(pcopy, 0755) == -1) {
 160                         switch (errno) {
 161                                 case EEXIST:
 162                                         if (stat(pcopy, &st) == -1) {
 163                                                 /* can't stat? */
 164                                                 return 0;
 165                                         }
 166                                         switch (st.st_mode & S_IFMT) {
 167                                                 case S_IFDIR:
 168                                                         break;
 169                                                 default:
 170                                                         return 0;
 171                                         }
 172                                         break;
 173                                 default:
 174                                         return 0;
 175                         }
 176                 }
 177                 if (delim) {
 178                         *delim = ch;
 179                 }
 180                 s = delim;
 181         } while (delim);
 182
 183         return 1;
 184 }
 185
 186 static char *column(char *col, int ncols, char **vals, char **cols) {
 187         int i = 0;
 188         char *val = NULL;
 189
 190         for (i=0; i < ncols; i++) {
 191 //              fprintf(stderr, "checking '%s' = '%s'\n", cols[i], vals[i]);
 192
 193                 if (!strcmp(col, cols[i])) {
 194                         val = vals[i];
 195                         break;
 196                 }
 197         }
 198         return val;
 199 }
 200
 201 #define COL(x) column(x, ncols, vals, cols)
 202 #define SYSERR(x) do { conf->log->error = 2; return conf->errabort; } while (0)
 203
 204
 205 static char *ops[] = { "new", "remove", "update", 0 };
 206
 207 enum op {
 208         OP_NEW = 1,
 209         OP_REMOVE = 2,
 210         OP_UPDATE = 3
 211 };
 212
 213 static int getop(char *opstr) {
 214         int i;
 215
 216         if (!opstr) return 0;
 217         for (i=0;ops[i];i++) {
 218                 if (!strcmp(opstr, ops[i])) {
 219                         return i+1;
 220                 }
 221         }
 222         return 0;
 223 }
 224
 225 static int report_conflicts(void *f, int ncols, char **vals, char **cols) {
 226         struct config *conf = f;
 227         char *path, *hash, *pkg, *conflict_type, *mds;
 228
 229         pkg = COL("pkgid");
 230         path = COL("path");
 231         conflict_type = COL("conflict");
 232         if (!strcmp(conflict_type, "hash")) {
 233                 hash = COL("hash");
 234                 fprintf(stderr, "hash conflict: package %s path %s hash %.8s\n",
 235                                 pkg, path, hash);
 236         } else
 237         if (!strcmp(conflict_type, "md")) {
 238                 mds = COL("mds");
 239                 fprintf(stderr, "md conflict: package %s path %s md %s\n",
 240                                 pkg, path, mds);
 241         } else {
 242                 fprintf(stderr, "%s conflict: package %s path %s\n",
 243                                 conflict_type, pkg, path);
 244         }
 245
 246         conf->conflicts++;
 247         return 0;
 248 }
 249
 250 static int check_existing(void *f, int ncols, char **vals, char **cols) {
 251         struct config *conf = f;
 252         char *path;
 253         struct stat st;
 254
 255         path = COL("dest");
 256         if (!path) {
 257                 return seterror(conf, "no path");
 258         }
 259
 260         if (conf->dryrun) {
 261                 printf("checkfor %s\n", path);
 262                 fflush(stdout);
 263                 return 0;
 264         }
 265
 266         if (conf->verbose) {
 267                 fprintf(stderr, "check for existing %s\n", path);
 268         }
 269
 270         if (lstat(path, &st) == 0) {
 271                 fprintf(stderr, "%s exists\n", path);
 272                 conf->errors++;
 273         } else {
 274                 switch(errno) {
 275                         /* not an error, file shouldn't exist*/
 276                         case ENOENT: break;
 277                         default:
 278                                 fprintf(stderr, "unable to check %s: %s\n",
 279                                                 path, strerror(errno));
 280                                 conf->errors++;
 281                                 break;
 282                 }
 283         }
 284         return 0;
 285 }
 286
 287 static int remove_files(void *f, int ncols, char **vals, char **cols) {
 288         struct config *conf = f;
 289         char *dest;
 290         struct stat st;
 291         int flags = 0;
 292
 293         dest = COL("dest");
 294         if (!dest) return seterror(conf,"no file dest");
 295
 296         if (conf->dryrun) {
 297                 char *ftype = COL("filetype");
 298                 int t = *ftype;
 299
 300                 switch(t) {
 301                         case 'd': printf("rmdir %s\n", dest); break;
 302                         default: printf("unlink %s\n", dest); break;
 303                 }
 304                 fflush(stdout);
 305                 return 0;
 306         }
 307
 308         if (lstat(dest, &st) == -1) {
 309                 return seterror(conf,"can't stat");
 310         }
 311
 312         if (S_ISDIR(st.st_mode)) {
 313                 flags = AT_REMOVEDIR;
 314         }
 315         /* TODO check that expected filetype matches actual filetype */
 316
 317         if (conf->verbose) {
 318                 fprintf(stderr, "%s(%s)\n", flags ? "rmdir" : "unlink", dest);
 319         }
 320
 321         errno = 0;
 322
 323         if (unlinkat(AT_FDCWD, dest, flags) == -1) {
 324                 switch (errno) {
 325                         case ENOENT:
 326                                 break;
 327                         default:
 328                                 return seterror(conf, "can't unlink");
 329                 }
 330         }
 331
 332         return 0;
 333 }
 334
 335 #define MARK fprintf(stderr, "%s %d: mark\n", __func__, __LINE__)
 336
 337 /* file does not exist */
 338 #define D_NOEXIST 0x1
 339 /* files are different types */
 340 #define D_TYPE 0x2
 341 /* metadata is different */
 342 #define D_MD 0x4
 343 /* content or link target is different */
 344 #define D_HASH 0x8
 345 /* file to be installed is a directory */
 346 #define D_ISDIR  0x10
 347 /* path on disk is a directory */
 348 #define D_EISDIR 0x20
 349 /* usernames different */
 350 #define D_UID 0x40
 351 /* group names different */
 352 #define D_GID 0x80
 353 /* file mode is different */
 354 #define D_MODE 0x100
 355 /* mtimes are different */
 356 #define D_MTIME 0x200
 357 /* the hash of the file we are supposedly replacing is different than
 358  * the the hash of the file on disk
 359  */
 360 #define D_OHASH 0x400
 361 /* an error occurred trying to compare the file (other than it doesn't exist */
 362 #define D_ERROR 0x1000
 363 /* there was a stat error */
 364 #define D_STATERROR 0x2000
 365 /* there was an error calling readlink */
 366 #define D_RLERROR 0x4000
 367
 368 /* 1 = file doesn't exist, 2 = file is a directory, target isn't */
 369 /* 4 == ftype different */
 370 /* 8 = hash different when both are regular files */
 371 static unsigned int file_compare(struct nitem *n, struct stat *st) {
 372         int etype = 0, stat_type;
 373         char ehash[ZPM_HASH_STRLEN+1];
 374         unsigned int diff = 0;
 375         char link[1024];
 376         ssize_t lsize;
 377
 378         switch (n->ftype) {
 379                 case 'd': etype = S_IFDIR; diff |= D_ISDIR ; break;
 380                 case 'r': etype = S_IFREG; break;
 381                 case 'l': etype = S_IFLNK; break;
 382                 default: etype = 0; break;
 383         }
 384
 385         errno = 0;
 386         /* new file, so check type, hash, etc */
 387         if (lstat(n->dest, st) == 0) {
 388                 stat_type = st->st_mode & S_IFMT;
 389                 if (stat_type != etype) {
 390                         diff |= D_TYPE;
 391                 }
 392                 if (stat_type == S_IFDIR) {
 393                         diff |= D_EISDIR;
 394                 }
 395
 396                 if (n->hash && etype == S_IFREG && stat_type == S_IFREG) {
 397                         zpm_hash(n->dest, ehash, 0);
 398                         if (strcmp(n->hash, ehash) != 0) {
 399                                 diff |= D_HASH;
 400                         }
 401                         if (n->ohash && strcmp(n->ohash, ehash) != 0) {
 402                                 diff |= D_OHASH;
 403                         }
 404                 }
 405                 if (n->hash && etype == S_IFLNK && stat_type == S_IFLNK) {
 406                         lsize = readlink(n->dest, link, sizeof link);
 407
 408                         if (lsize == -1 || lsize == sizeof link) {
 409                                 diff |= D_RLERROR;
 410                                 diff |= D_ERROR;
 411                         } else {
 412                                 link[lsize] = 0;
 413                                 if (strcmp(n->target, link) != 0) {
 414                                         diff |= D_HASH;
 415                                 }
 416                         }
 417                 }
 418                 if (n->uid != st->st_uid) {
 419                         diff |= D_UID;
 420                         diff |= D_MD;
 421                 }
 422                 if (n->gid != st->st_gid) {
 423                         diff |= D_GID;
 424                         diff |= D_MD;
 425                 }
 426                 if (n->mode != (st->st_mode & 07777)) {
 427                         diff |= D_MODE;
 428                         diff |= D_MD;
 429                 }
 430         } else {
 431                 switch(errno) {
 432                         case ENOENT: diff |= D_NOEXIST; break;
 433                         default: diff |= (D_STATERROR|D_ERROR); break;
 434                 }
 435         }
 436
 437         return diff;
 438 }
 439
 440 static int read_item(struct config *conf, int ncols, char **vals, char **cols,
 441                 struct nitem *n) {
 442         char *val;
 443         long lval;
 444         struct passwd *pw;
 445         struct group *gr;
 446         struct nitem zero = { 0 };
 447
 448         *n = zero;
 449
 450         val = COL("op");
 451         if (!val) {
 452                 seterror(conf, "can't determine op");
 453                 return 0;
 454         }
 455         n->opstr = val;
 456         n->op = getop(val);
 457         if (!n->op) {
 458                 seterror(conf, "can't determine op");
 459                 return 0;
 460         }
 461
 462         n->path = COL("path");
 463         if (!n->path) {
 464                 seterror(conf, "no file path");
 465                 return 0;
 466         }
 467         if (strlen(n->path) == 0) {
 468                 seterror(conf, "zero length path not allowed");
 469                 return 0;
 470         }
 471
 472         /* TODO config to dishonor setuid/setgid */
 473         n->dest = COL("dest");
 474         if (!n->dest) {
 475                 seterror(conf, "no file dest");
 476                 return 0;
 477         }
 478
 479         if (strlen(n->dest) == 0) {
 480                 seterror(conf, "zero length dest not allowed");
 481                 return 0;
 482         }
 483
 484         val = COL("mode");
 485
 486         if (!val) {
 487                 seterror(conf, "can't determine mode");
 488                 return 0;
 489         }
 490
 491         n->mode = strtoul(val, NULL, 8);
 492
 493         val = COL("configuration");
 494         if (!val) {
 495                 seterror(conf, "can't determine config status");
 496                 return 0;
 497         }
 498         lval = strtol(val, NULL, 10);
 499
 500         n->configuration = ((lval & 1) != 0);
 501         n->oldwasconf = ((lval & 2) != 0);
 502
 503         val = COL("filetype");
 504         if (!val || strlen(val) == 0) {
 505                 seterror(conf, "can't determine file type");
 506                 return 0;
 507         }
 508         n->ftype = *val;
 509
 510         /* these can be null */
 511         n->ohash = COL("ohash");
 512         n->mds = COL("mds");
 513         n->omds = COL("omds");
 514         n->pkglist = COL("pkglist");
 515
 516         if (n->ftype == 'r') {
 517                 n->hash = COL("hash");
 518                 if (!n->hash) {
 519                         seterror(conf, "can't get hash");
 520                         return 0;
 521                 }
 522         } else if (n->ftype == 'l') {
 523                 n->target = COL("target");
 524                 if (!n->target) {
 525                         seterror(conf, "can't get target");
 526                         return 0;
 527                 }
 528                 if (strlen(n->target) == 0) {
 529                         seterror(conf, "zero length target not allowed");
 530                         return 0;
 531                 }
 532                 n->hash = n->target;
 533         }
 534
 535         if (conf->setuser) {
 536                 val = COL("username");
 537                 if (!val) {
 538                         seterror(conf, "no username");
 539                         return 0;
 540                 }
 541                 pw = getpwnam(val);
 542                 if (!pw) {
 543                         seterror(conf, "no passwd entry");
 544                         return 0;
 545                 }
 546                 n->uid = pw->pw_uid;
 547         } else {
 548                 n->uid = geteuid();
 549         }
 550
 551         if (conf->setgroup) {
 552                 val = COL("groupname");
 553                 if (!val) {
 554                         seterror(conf, "no groupname");
 555                         return 0;
 556                 }
 557                 gr = getgrnam(val);
 558                 if (!gr) {
 559                         seterror(conf, "no group entry");
 560                         return 0;
 561                 }
 562                 n->gid = gr->gr_gid;
 563         } else {
 564                 n->gid = getegid();
 565         }
 566
 567         errno = 0;
 568         double mtime = strtod(COL("mtime"),NULL);
 569         if (errno) {
 570                 mtime = (double)time(NULL);
 571         }
 572
 573         n->mtime = (time_t)mtime;
 574
 575         n->times[0].tv_sec = 0;
 576         n->times[0].tv_nsec = UTIME_OMIT;
 577         n->times[1].tv_sec = (time_t)llrint(floor(mtime));
 578         n->times[1].tv_nsec = lrint(floor(fmod(mtime,1.0)*1000000000));
 579
 580         return 1;
 581 }
 582
 583 static int remove_dir(struct config *conf, char *path) {
 584         int rv;
 585
 586         rv = rmdir(path);
 587         if (rv == -1) {
 588                 setsyserr(conf, "can't rmdir %s", path);
 589                 return 0;
 590         }
 591         return 1;
 592 }
 593
 594 static int remove_existing(struct config *conf, char *path) {
 595         int rv;
 596
 597         rv = unlink(path);
 598         if (rv == -1) {
 599                 setsyserr(conf, "can't unlink %s", path);
 600                 return 0;
 601         }
 602         return 1;
 603 }
 604
 605 static int set_md(struct config *conf, struct nitem *item) {
 606         int rv;
 607         int success = 0;
 608
 609         if (conf->dryrun) {
 610                 if (item->ftype != 'l') {
 611                         printf("chmod %o %s\n", item->mode, item->dest);
 612                 }
 613                 if (conf->setuser && conf->setgroup) {
 614                         printf("lchown %d:%d %s\n", item->uid, item->gid,
 615                                         item->dest);
 616                 }
 617                 printf("mtime %.0f %s\n", (double)item->mtime, item->dest);
 618                 fflush(stdout);
 619                 return success;
 620         }
 621
 622         /* can't chmod a symlink */
 623         if (item->ftype != 'l') {
 624                 rv = chmod(item->dest, item->mode);
 625
 626                 if (rv == -1) {
 627                         setsyserr(conf, "can't chmod %o %s", item->mode, item->dest);
 628                         return conf->errabort;
 629                 }
 630         }
 631
 632         if (conf->setuser && conf->setgroup) {
 633                 rv = lchown(item->dest, item->uid, item->gid);
 634                 if (rv == -1) {
 635                         setsyserr(conf, "can't lchown %s", item->dest);
 636                         return conf->errabort;
 637                 }
 638         }
 639
 640         rv = utimensat(AT_FDCWD, item->dest, item->times, AT_SYMLINK_NOFOLLOW);
 641         if (rv == -1) {
 642                 setsyserr(conf, "can't set mtime %.0f %s", (double)item->mtime,
 643                                 item->dest);
 644                 return conf->errabort;
 645         }
 646         return 0;
 647 }
 648
 649 /* install a file or create a directory or symlink.  path should not exist
 650  * at this point.
 651  */
 652 /* flags: 1 = set md, 2 = create leading dirs, 4 = unlink existing file,
 653  * 8 = rmdir existing dir, 16 = return true/false
 654  */
 655 #define INS_MD 0x1
 656 #define INS_CLD 0x2
 657 #define INS_UNLINK 0x4
 658 #define INS_RMDIR 0x8
 659 #define INS_RTF 0x10
 660 #define INS_ZPMNEW 0x20
 661 static int install(struct config *conf, struct nitem *item, unsigned int flags) {
 662         int rv = 1;
 663         struct zpm *source;
 664
 665         int mkleading = (flags & 2);
 666         int setmd = (flags & 1);
 667         int unlink_file = (flags & 4);
 668         int rm_dir = (flags & 8);
 669         int failure = conf->errabort;
 670         int success = 0;
 671
 672         if (flags & 16) {
 673                 failure = 0;
 674                 success = 1;
 675         }
 676
 677         if (conf->dryrun) {
 678                 if (unlink_file) {
 679                         printf("unlink %s\n", item->dest);
 680                 } else if (rm_dir) {
 681                         printf("rmdir %s\n", item->dest);
 682                 }
 683
 684                 printf("install %c%o %d:%d %s", item->ftype,
 685                                 item->mode, item->uid, item->gid,
 686                                 item->dest);
 687                 if (item->ftype == 'l') {
 688                         printf(" -> %s", item->target);
 689                 }
 690                 printf("\n");
 691                 fflush(stdout);
 692                 return success;
 693         }
 694
 695         source = conf->src ? conf->src : conf->log;
 696
 697         if (unlink_file) {
 698                 rv = remove_existing(conf, item->dest);
 699         } else if (rm_dir) {
 700                 rv = remove_dir(conf, item->dest);
 701         }
 702
 703         if (rv != 1) {
 704                 return failure;
 705         }
 706
 707         if (mkleading) {
 708                 rv = create_leading_dirs(item->dest);
 709                 if (!rv) {
 710                         setsyserr(conf, "can't create leading dirs for %s", item->dest);
 711                         return failure;
 712                 }
 713         }
 714
 715         if (item->ftype == 'r') {
 716                 rv = zpm_extract(source, item->hash, item->dest, item->mode);
 717                 if (rv == 0) {
 718                         seterror(conf, "can't extract %s", item->dest);
 719                         return failure;
 720                 }
 721                 return success;
 722         }
 723
 724         switch (item->ftype) {
 725                 case 'd': rv = mkdir(item->dest, item->mode);
 726                           break;
 727                 case 'l': rv = symlink(item->target, item->dest);
 728                           break;
 729                 default: /* error */
 730                           break;
 731         }
 732
 733         if (rv == -1) {
 734                 setsyserr(conf, "installing %s failed", item->dest);
 735                 return failure;
 736         }
 737
 738         if (setmd) {
 739                 return set_md(conf, item) == 0 ? success : failure;
 740         }
 741
 742         return success;
 743 }
 744
 745 /*
 746  * figure out what the difference is for a config file, only called
 747  * for an update of a configuration file
 748  * return -1 on an error
 749  * return 1 if the new file should not be installed
 750  * return 0 if the new file should be installed
 751  */
 752 static int adjust_for_config(struct config *conf, struct nitem *n, unsigned int
 753                 diffs) {
 754
 755         if (!n->oldwasconf) {
 756                 return 0;
 757         }
 758
 759         /* TODO what if old was a directory? */
 760         if (!n->configuration) {
 761                 /* replacing conf with non-conf */
 762                 /* absorb file, mark todo */
 763                 char hash[ZPM_HASH_STRLEN+1];
 764                 if (!conf->dryrun) {
 765                         if (conf->verbose) {
 766                                 fprintf(stderr, "importing old conf file\n");
 767                         }
 768                         if (zpm_import(conf->log, n->dest, 0, hash)) {
 769                                 zpm_note_add(conf->log, n->pkglist, n->dest, hash,
 770                                                 "replaced config file with non-config.  zpm-cat %.8s", hash);
 771                         } else {
 772                                 fprintf(stderr, "unable to import existing config file %s\n", n->dest);
 773                                 return -1;
 774                         }
 775                 } else {
 776                         fprintf(stderr, "dry-run: would replace config file %s with non-config file\n", n->dest);
 777                 }
 778                 return 0;
 779         }
 780
 781         int sametype = (!(diffs & D_TYPE));
 782         int isdir = (diffs & D_ISDIR);
 783         int eisdir = (diffs & D_EISDIR);
 784
 785         /* both old and new are config files */
 786         if (isdir && sametype) {
 787                 /* both config directories, can only be changing
 788                  * metadata, so no adjustment needed
 789                  */
 790                         if (conf->verbose) {
 791                                 fprintf(stderr, "both config dirs, ok to update\n");
 792                         }
 793                 return 0;
 794         }
 795
 796         if (isdir) {
 797                 char hash[ZPM_HASH_STRLEN+1];
 798
 799                 /* replacing old file with new directory */
 800                 /* absorb, make note */
 801                 if (!conf->dryrun) {
 802                         if (zpm_import(conf->log, n->dest, 0, hash)) {
 803                                 zpm_note_add(conf->log, n->pkglist, n->dest, hash,
 804                                                 "replaced config file with config directory.  zpm-cat %.8s", hash);
 805                         } else {
 806                                 fprintf(stderr, "unable to import existing config file %s\n", n->dest);
 807                                 return -1;
 808                         }
 809                 } else {
 810                         fprintf(stderr, "dry-run: would replace config file %s with config directory\n", n->dest);
 811                 }
 812                         return 0;
 813         }
 814
 815         if (eisdir) {
 816                 /* replacing old conf directory with a conf file.
 817                  * nothing needs to be done, if the directory
 818                  * is empty, it's ok to remove.  if it's not empty,
 819                  * the install will fail
 820                  */
 821                 return 0;
 822         }
 823
 824         /* replacing old file with new file */
 825         /* new is same as on disk */
 826         if (!(diffs & D_HASH)) {
 827                         if (conf->verbose) {
 828                                 fprintf(stderr, "new config file is already on disk, probably shouldn't happen\n");
 829                         }
 830                 return 0;
 831         }
 832
 833         /* new is different than on disk, but on disk is same as old */
 834         if (!(diffs & D_OHASH)) {
 835                         if (conf->verbose) {
 836                                 fprintf(stderr, "old config file not changed from default, replacing with new default\n");
 837                         }
 838                 /* ok to do the update, since same as default */
 839                 fprintf(stderr, "updating default config %s\n", n->dest);
 840                 return 0;
 841         }
 842
 843         /* new is different than on disk, and disk different than old */
 844         /* log */
 845         if (conf->verbose) {
 846                 fprintf(stderr, "new default config file is different than on disk, and old default was changed, should keep on-disk config\n");
 847         }
 848         if (!conf->dryrun) {
 849                 zpm_note_add(conf->log, n->pkglist, n->dest, n->hash,
 850                                 "default config file update.  zpm-cat %.8s", n->hash);
 851                 /* TODO check for note error */
 852         } else {
 853                 fprintf(stderr, "dry-run: default config file %s update\n",
 854                                 n->dest);
 855         }
 856         return 1;
 857
 858 }
 859
 860 static int install_files(void *f, int ncols, char **vals, char **cols) {
 861         struct config *conf = f;
 862         struct nitem nitem;
 863         struct stat existing;
 864         int update = 0;
 865
 866         /* TODO put the result row in a hash table.  May not actually
 867          * be faster
 868          */
 869         if (!read_item(conf, ncols, vals, cols, &nitem)) {
 870                 fprintf(stderr, "can't read item\n");
 871                 return conf->errabort;
 872         }
 873
 874         if (conf->verbose && !conf->dryrun) {
 875                 fprintf(stderr, "%s '%c' %s\n", nitem.opstr, nitem.ftype,
 876                                 nitem.dest);
 877         }
 878
 879         unsigned int diffs = file_compare(&nitem, &existing);
 880         if (diffs >= D_ERROR) {
 881                 return seterror(conf, "can't check %s", nitem.dest);
 882         }
 883
 884         /* updates:
 885          * exist & same type & md same & hash same: do nothing, but warn bug
 886          * exist & same type & md diff & hash same: fix md
 887          * exist & same type & md same & hash diff: replace
 888          * exist & same type & md diff & hash diff: replace & fix
 889          * no exist: install and warn
 890          * dir & not dir : remove, mkdir
 891          * not dir & not dir & diff type: remove, install
 892          * not dir & dir : remove dir if empty, error if not empty, install
 893          *
 894          * installs:
 895          * no exist: create leading dirs, install
 896          *
 897          * exist & same type & md same & hash same & accept or over: do nothing
 898          * exist & same & md diff or hash diff & overwrite : update
 899          * exist & same & md diff or hash diff & accept : error, can't accept
 900          * exist & same & md diff or hash diff & not accept : error
 901          *
 902          * exist & different type & not overwrite : error
 903          * not dir & not dir & overwrite : remove and install
 904          * not dir & dir & overwrite: remove empty or error, install
 905          * dir & dir & overwrite: fix md
 906          * dir & not dir & overwrite: remove and mkdir
 907          */
 908         int exist = (!(diffs & D_NOEXIST));
 909         int sametype = (!(diffs & D_TYPE));
 910         int mdsame = (!(diffs & D_MD));
 911         int hashsame = (!(diffs & D_HASH));
 912         int isdir = (diffs & D_ISDIR);
 913         int eisdir = (diffs & D_EISDIR);
 914         int accept = conf->absorb;
 915         int overwrite = conf->overwrite;
 916         int installing = (nitem.op == OP_NEW);
 917         update = (nitem.op == OP_UPDATE);
 918
 919         if (update) {
 920                 if (!exist) {
 921                         /* warn, it should exist */
 922                         fprintf(stderr, "%s missing, installing", nitem.dest);
 923                         return install(conf, &nitem, 3);
 924                 }
 925
 926                 switch (adjust_for_config(conf, &nitem, diffs)) {
 927                         case -1: return conf->errabort; break;
 928                         case 1:
 929                         fprintf(stderr, "skipping changed default config file: %s\n", nitem.dest);
 930                         return 0; break;
 931                         default: break;
 932                 }
 933
 934                 /* file exists in filesystem */
 935                 if (sametype) {
 936                         if (mdsame && hashsame) {
 937                                 /* warn, bug in logic.  This shouldn't occur,
 938                                  * because if there is nothing to do, it
 939                                  * shouldn't be listed as an update
 940                                  */
 941                                 /* could be an update.  We're checking against
 942                                  * what's actually on disk, not what was
 943                                  * expected to have been on disk.  So, if
 944                                  * the admin has modified the file, or if
 945                                  * it had been installed ignoring the user
 946                                  * and group, it might be correct on disk
 947                                  * but not as in the local database
 948                                  */
 949                                 /* TODO detect whether this a logic bug or
 950                                  * an on-disk difference
 951                                  */
 952 #if 0
 953                                 fprintf(stderr, "%s should not be an update\n", nitem.dest);
 954                                 fprintf(stderr, "old hash: %s\n", nitem.ohash);
 955                                 fprintf(stderr, "new hash: %s\n", nitem.hash);
 956                                 fprintf(stderr, "old mds: %s\n", nitem.omds);
 957                                 fprintf(stderr, "new mds: %s\n", nitem.mds);
 958 #endif
 959                                 /* do nothing */
 960                                 return 0;
 961                         }
 962                         if (!mdsame && hashsame) {
 963                                 /* fix md */
 964                                 return set_md(conf, &nitem);
 965                         }
 966                         if (mdsame && !hashsame) {
 967                                 /* install */
 968                                 return install(conf, &nitem, 3);
 969                         }
 970                         if (!mdsame && !hashsame) {
 971                                 /* install */
 972                                 return install(conf, &nitem, 3);
 973                         }
 974                 }
 975
 976                 /* file exists, and is not the same type */
 977
 978                 if (isdir && !eisdir) {
 979                         /* remove existing */
 980                         /* mkdir */
 981                         return install(conf, &nitem, 7);
 982                 }
 983                 if (!isdir && eisdir) {
 984                         /* remove dir, or error */
 985                         /* install */
 986                         return install(conf, &nitem, 11);
 987                 }
 988                 if (!isdir && !isdir) {
 989                         /* necessarily !sametype, sametype handled above */
 990                         /* remove existing */
 991                         /* install */
 992                         return install(conf, &nitem, 7);
 993                 }
 994                 /* error, should not be possible, assert(0)? */
 995                 fprintf(stderr,"impossible state: %s:%d\n", __func__, __LINE__);
 996         }
 997
 998         if (installing) {
 999                 if (!exist) {
1000                         return install(conf, &nitem, 3);
1001                 }
1002
1003                 /* file exists in filesystem */
1004                 if (sametype) {
1005                         if (mdsame && hashsame && (accept || overwrite)) {
1006                                 /* do nothing */
1007                                 if (conf->dryrun || conf->verbose) {
1008                                         fprintf(stderr, "accept %s: %s\n",
1009                                                         eisdir ? "directory" : "file", nitem.dest);
1010                                 }
1011                                 return 0;
1012                         }
1013
1014                         if (mdsame && hashsame && !(accept || overwrite)) {
1015                                 /* error */
1016                                 return seterror(conf, "file exists: %s", nitem.dest);
1017                         }
1018
1019                         if (mdsame && !hashsame && overwrite) {
1020                                 /* install */
1021                                 return install(conf, &nitem, eisdir ? 11 : 7);
1022                         }
1023
1024                         if (nitem.configuration && accept) {
1025                                 /* accept a changed config file */
1026                                 if (conf->dryrun || conf->verbose) {
1027                                         fprintf(stderr, "accept %smodified config %s: %s\n", (!mdsame || !hashsame) ? "" : "un",
1028                                                         eisdir ? "directory" : "file", nitem.dest);
1029                                 }
1030                                 return 0;
1031                         }
1032
1033                         if (mdsame && !hashsame && !overwrite) {
1034                                 /* accept doesn't matter, since it's
1035                                  * not an acceptable file */
1036                                 /* error */
1037                                 if (nitem.ftype == 'l') {
1038                                         char link[1024];
1039                                         ssize_t lsize;
1040                                         lsize = readlink(nitem.dest, link, sizeof link);
1041                                         if (lsize == -1 || (size_t)lsize >= sizeof link) {
1042                                                 return seterror(conf, "%s (linkdiff): expecting %s -> %s, unable to read link", accept ? "existing file not acceptable" : "file exists", nitem.dest, nitem.target, link);
1043                                         } else {
1044                                                 link[lsize] = 0;
1045                                                 /* links must be different */
1046                                                 return seterror(conf, "%s (linkdiff): expecting %s -> %s, have -> %s", accept ? "existing file not acceptable" : "file exists", nitem.dest, nitem.target, link);
1047                                         }
1048                                 } else {
1049                                         return seterror(conf, "%s (hashdiff): %s", accept ? "existing file not acceptable" : "file exists", nitem.dest);
1050                                 }
1051                         }
1052                         if (!mdsame && hashsame && overwrite) {
1053                                 /* fix md */
1054                                 return set_md(conf, &nitem);
1055                         }
1056                         if (!mdsame && hashsame && !overwrite) {
1057                                 /* accept doesn't matter, since it's
1058                                  * not an acceptable file */
1059                                 /* error */
1060                                 return seterror(conf, "%s (mddiff): %s", accept ? "existing file not acceptable" : "file exists", nitem.dest);
1061                         }
1062                         if (!mdsame && !hashsame && overwrite) {
1063                                 /* install */
1064                                 return install(conf, &nitem, eisdir ? 11 : 7);
1065                         }
1066                         if (!mdsame && !hashsame && !overwrite) {
1067                                 /* accept doesn't matter, since it's
1068                                  * not an acceptable file */
1069                                 /* error */
1070                                 return seterror(conf, "%s (md+hash): %s", accept ? "existing file not acceptable" : "file exists", nitem.dest);
1071                         }
1072                         /* TODO error, should be impossible */
1073                         return seterror(conf, "impossible state reached");
1074                 }
1075
1076                 /* file exists, and is not the same type */
1077                 if (!overwrite) {
1078                         /* error */
1079                                 return seterror(conf, "%s (difftype): %s", accept ? "existing file not acceptable" : "file exists", nitem.dest);
1080                 }
1081
1082                 /* not the same type, but ok to overwrite */
1083                 if (!eisdir) {
1084                         /* remove existing */
1085                         return install(conf, &nitem, 7);
1086                 }
1087
1088                 /* existing path is a directory */
1089                 if (isdir) {
1090                         /* fix md */
1091                         /* impossible, if isdir and eisdir, would
1092                          * be same type
1093                          * TODO
1094                          */
1095                         return set_md(conf, &nitem);
1096                 } else {
1097                         /* remove empty dir or error */
1098                         /* install */
1099                         return install(conf, &nitem, 11);
1100                 }
1101                 /* if we get here, we missed a case */
1102                 /* TODO error */
1103                 return seterror(conf, "impossible state 2 reached");
1104         }
1105
1106         /* TODO extra verbose print perms, mtime, etc, probably ls -l
1107          * format
1108          */
1109         if (conf->verbose) {
1110                 printf("%s\n", nitem.path);
1111         }
1112
1113         return 0;
1114 }
1115
1116 static void check_conflicts(struct config *conf, char *conflict_type,
1117                 int (callback)(void *, int, char **, char **)) {
1118         int rv;
1119         char *errmsg;
1120         sqlite3_str *s;
1121         char *sql;
1122
1123         s = sqlite3_str_new(conf->log->db);
1124         sqlite3_str_appendall(s, "select *, ");
1125         if (conf->rootdir) {
1126                 sqlite3_str_appendf(s, "printf('%%s/%%s',rtrim(%Q,'/'),ltrim(path,'/'))", conf->rootdir);
1127         } else {
1128                 sqlite3_str_appendf(s, "printf('/%%s', trim(path, '/'))");
1129         }
1130         sqlite3_str_appendall(s, " as dest from syncconflicts");
1131
1132         if (conflict_type) {
1133                 sqlite3_str_appendf(s," where conflict = %Q", conflict_type);
1134         }
1135         if (conf->reverse) {
1136                 sqlite3_str_appendall(s," order by length(path) desc, path desc,pkgid collate vercmp desc, conflict desc");
1137         } else {
1138                 sqlite3_str_appendall(s," order by length(path), path, pkgid collate vercmp, conflict");
1139
1140         }
1141
1142         sql = sqlite3_str_value(s);
1143
1144         rv = zpm_exec(conf->log, sql, callback, conf, &errmsg);
1145
1146         sqlite3_str_finish(s);
1147
1148         if (rv) {
1149                 fprintf(stderr, "exec fail: %s\n", sqlite3_errstr(rv));
1150                 if (errmsg) {
1151                         fprintf(stderr, "database error: %s\n", errmsg);
1152                         conf->errors++;
1153                 }
1154                 if (conf->log->error == 1) {
1155                         fprintf(stderr, "unable to allocate memory\n");
1156                 }
1157                 fprintf(stderr, "zpm_exec failure: %s\n",
1158                                 conf->log->errmsg ? conf->log->errmsg : "unknown");
1159                 conf->errors++;
1160         }
1161         if (conf->log->errmsg) {
1162                 fprintf(stderr, "error: %s\n", conf->log->errmsg);
1163         }
1164         if (conf->errors && conf->exitonerror) {
1165                 zpm_close(conf->log);
1166                 zpm_close(conf->src);
1167                 exit(EXIT_FAILURE);
1168         }
1169         /* TODO final report function in conf var */
1170 }
1171
1172 static void runstage(struct config *conf, char *stage,
1173                 int (callback)(void *, int, char **, char **)) {
1174         int rv;
1175         char *errmsg;
1176         sqlite3_str *s;
1177         char *sql;
1178
1179         s = sqlite3_str_new(conf->log->db);
1180         sqlite3_str_appendall(s, "select *, ");
1181         if (conf->rootdir) {
1182                 sqlite3_str_appendf(s, "printf('%%s/%%s',rtrim(%Q,'/'),ltrim(path,'/'))", conf->rootdir);
1183         } else {
1184                 sqlite3_str_appendf(s, "printf('/%%s', trim(path, '/'))");
1185         }
1186         sqlite3_str_appendall(s, " as dest from syncinfo");
1187
1188         if (stage) {
1189                 sqlite3_str_appendf(s," where op = %Q", stage);
1190         }
1191         if (conf->reverse) {
1192                 sqlite3_str_appendall(s," order by length(path) desc, path desc");
1193         }
1194
1195         sql = sqlite3_str_value(s);
1196
1197         rv = zpm_exec(conf->log, sql, callback, conf, &errmsg);
1198
1199         sqlite3_str_finish(s);
1200
1201         if (rv) {
1202                 fprintf(stderr, "exec fail: %s\n", sqlite3_errstr(rv));
1203                 if (errmsg) {
1204                         fprintf(stderr, "database error: %s\n", errmsg);
1205                         conf->errors++;
1206                 }
1207                 if (conf->log->error == 1) {
1208                         fprintf(stderr, "unable to allocate memory\n");
1209                 }
1210                 fprintf(stderr, "zpm_exec failure: %s\n",
1211                                 conf->log->errmsg ? conf->log->errmsg : "unknown");
1212                 conf->errors++;
1213         }
1214 #if 0
1215         if (conf->log->errmsg) {
1216                 fprintf(stderr, "error: %s\n", conf->log->errmsg);
1217         }
1218 #endif
1219         if (conf->errors && conf->exitonerror) {
1220                 zpm_close(conf->log);
1221                 zpm_close(conf->src);
1222                 exit(EXIT_FAILURE);
1223         }
1224         /* TODO final report function in conf var */
1225 }
1226
1227 int main(int ac, char **av){
1228         struct zpm localdb;
1229         struct zpm pkgdb;
1230         int opt;
1231         char *pkgdbfile = 0, *localdbfile = 0;
1232         char *s;
1233
1234         struct config conf;
1235
1236         conf.errabort = 1;
1237         conf.errors = 0;
1238         conf.conflicts = 0;
1239         conf.verbose = 0;
1240         conf.dryrun = 0;
1241         conf.setuser = 1;
1242         conf.setgroup = 1;
1243         conf.log = 0;
1244         conf.src = 0;
1245         conf.rootdir = 0;
1246         conf.reverse = 0;
1247         conf.overwrite = 0;
1248         conf.absorb = 0;
1249
1250         if (geteuid() != 0) {
1251                 conf.setuser = 0;
1252                 conf.setgroup = 0;
1253         }
1254
1255         localdbfile = ZPM_LOCAL_DB;
1256         if ((s = getenv("ZPMDB"))) {
1257                 /* TODO does this need to be copied ? */
1258                 localdbfile = s;
1259         }
1260
1261         if ((s = getenv("ZPM_ROOT_DIR"))) {
1262                 /* TODO does this need to be copied ? */
1263                 conf.rootdir = s;
1264         }
1265
1266         /*
1267          * -d localdb or ZPMDB * or /var/lib/zpm/zpm.db, or die
1268          * -f 'package database', otherwise regular default of env
1269          *  ZPM_PACKAGE_FILE, or use pkgdb if otherwise not found
1270          * -R root of pkg, will just chdir there
1271          *
1272          *  args are pkgid triple, but will do a pkg find on the pkgdb
1273          */
1274
1275         while ((opt = getopt(ac, av, "f:d:c:nCR:vOA")) != -1) {
1276                 switch (opt) {
1277                         case 'd': localdbfile = optarg; break;
1278                         case 'f': pkgdbfile = optarg; break;
1279                         case 'n': conf.dryrun = 1; break;
1280                         case 'v': conf.verbose++; break;
1281                         case 'C': conf.errabort = 0; break;
1282                         case 'R': conf.rootdir = optarg; break;
1283                         case 'N': conf.setuser = 0; conf.setgroup = 0; break;
1284                         case 'O': conf.overwrite = 1; break;
1285                         case 'A': conf.absorb = 1; break;
1286                         default:
1287                                   usage();
1288                                   exit(EXIT_FAILURE);
1289                                   break;
1290                 }
1291         }
1292
1293         /* verify root dir exists */
1294         if (conf.rootdir && !exists(conf.rootdir, NULL)) {
1295                 fprintf(stderr, "rootdir %s does not exist\n", conf.rootdir);
1296         }
1297
1298         if (!zpm_open(&localdb, localdbfile)) {
1299                 fprintf(stderr, "can't open zpm db %s\n", localdbfile);
1300                 exit(EXIT_FAILURE);
1301         }
1302         conf.log = &localdb;
1303
1304         if (pkgdbfile) {
1305                 /* TODO open read-only */
1306                 if (!zpm_open(&pkgdb, pkgdbfile)) {
1307                         fprintf(stderr, "can't open src db %s\n", localdbfile);
1308                         exit(EXIT_FAILURE);
1309                 } else {
1310                         conf.src = &pkgdb;
1311                 }
1312         }
1313
1314         /* TODO find pkgid from arg */
1315
1316         /* TODO set conf var to finalize error reporting */
1317         if (conf.verbose) {
1318                 fprintf(stderr, "syncing filesystem %s (ldb %s) from %s\n",
1319                                 conf.rootdir ? conf.rootdir : "/",
1320                                 localdbfile, pkgdbfile);
1321         }
1322
1323         conf.errors = 0;
1324         conf.exitonerror = 0;
1325         check_conflicts(&conf, NULL, report_conflicts);
1326
1327         if (conf.conflicts) {
1328                 fprintf(stderr, "%d conflicts reported, aborting sync\n",
1329                                 conf.conflicts);
1330                 conf.errors++;
1331         } else {
1332                 /* no point in running it if we're just going to
1333                  * overwrite everything
1334                  */
1335                 if (!conf.overwrite && !conf.absorb && !conf.dryrun) {
1336                         runstage(&conf, "new", check_existing);
1337                 }
1338
1339                 if (conf.verbose) {
1340                         fprintf(stderr, "beginning %ssync\n", conf.dryrun ?
1341                                         "dryrun " : "");
1342                 }
1343                 /* have to do the removes first otherwise
1344                  * old files may conflict with update file
1345                  * type changes
1346                  */
1347                 if (!conf.errors) {
1348                         conf.exitonerror = conf.dryrun ? 0 : 1;
1349                         conf.errabort = conf.dryrun ? 0 : 1;
1350                         conf.reverse = 1;
1351                         if (conf.verbose) {
1352                                 fprintf(stderr, "removing old files\n");
1353                         }
1354                         runstage(&conf, "remove", remove_files);
1355                         conf.reverse = 0;
1356                         if (conf.verbose) {
1357                                 fprintf(stderr, "updating files\n");
1358                         }
1359                         runstage(&conf, "update", install_files);
1360                         if (conf.verbose) {
1361                                 fprintf(stderr, "installing files\n");
1362                         }
1363                         runstage(&conf, "new", install_files);
1364                 }
1365         }
1366
1367         zpm_close(&localdb);
1368         zpm_close(conf.src);
1369         return conf.errors ? 1 : 0;
1370 }