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