1 #define _POSIX_C_SOURCE 200809L
17 /* needed for S_IFMT and AT_FDCWD */
26 struct zpm *log; /* logging db will be attached as "log" */
30 int errabort, errors, verbose, dryrun, conflicts;
31 int setuser, setgroup;
32 int reverse, exitonerror;
33 int overwrite, accept, acceptdir, ignoredirmd;
34 int ops_total, ops_completed;
35 int ops_remove, ops_remove_completed;
36 int ops_update, ops_update_completed;
37 int ops_install, ops_install_completed;
38 int progress; /* type of progress meter */
51 char *pkglist; /* space separated */
55 int configuration, oldwasconf;
56 struct timespec times[2];
60 printf("usage: zpm $scriptname [-fncC] args ...\n");
63 static void pdots(int len, int ch, int was, int now, int total) {
64 was = len * was / total;
68 now = len * now / total;
75 static int seterror(struct config *conf, char *msgfmt, ...) {
82 vsnprintf(msg, sizeof msg, msgfmt, ap);
86 if (conf->log->errmsg) {
87 free(conf->log->errmsg);
90 conf->log->errmsg = strdup(msg);
93 fprintf(stderr, "%s\n", msg);
96 return conf->errabort;
99 static int setsyserr(struct config *conf, char *msgfmt, ...) {
106 va_start(ap, msgfmt);
107 printed = vsnprintf(msg, sizeof msg, msgfmt, ap);
111 /* nothing we can really do */
112 return conf->errabort;
115 if ((size_t)printed < sizeof msg) {
116 snprintf(msg+printed, sizeof msg - printed, ": %s",
121 if (conf->log->errmsg) {
122 free(conf->log->errmsg);
125 conf->log->errmsg = strdup(msg);
128 fprintf(stderr, "%s\n", msg);
131 return conf->errabort;
134 static int exists(char *path, mode_t *mode) {
137 if (lstat(path, &st) == -1) {
140 if (mode) *mode = st.st_mode;
144 /* TODO maintain a list of already created directories */
145 static int create_leading_dirs(char *path) {
148 char pcopy[ZPM_PATH_MAX];
153 delim = strrchr(pcopy, '/');
154 if (!delim) return 1; /* not an error, but no leading dirs */
156 /* cut off last component */
165 delim = strchr(s, '/');
171 /* try to create the directory, if it exists
172 * and is a directory or a symlink, that's ok
173 * should be (eventually) a symlink to a directory
174 * so we want stat here, not lstat
176 if (mkdir(pcopy, 0755) == -1) {
179 if (stat(pcopy, &st) == -1) {
183 switch (st.st_mode & S_IFMT) {
203 static char *column(char *col, int ncols, char **vals, char **cols) {
207 for (i=0; i < ncols; i++) {
208 // fprintf(stderr, "checking '%s' = '%s'\n", cols[i], vals[i]);
210 if (!strcmp(col, cols[i])) {
218 #define COL(x) column(x, ncols, vals, cols)
219 #define SYSERR(x) do { conf->log->error = 2; return conf->errabort; } while (0)
222 static char *ops[] = { "new", "remove", "update", 0 };
230 static int getop(char *opstr) {
233 if (!opstr) return 0;
234 for (i=0;ops[i];i++) {
235 if (!strcmp(opstr, ops[i])) {
242 static int report_conflicts(void *f, int ncols, char **vals, char **cols) {
243 struct config *conf = f;
244 char *path, *hash, *pkg, *conflict_type, *mds;
248 conflict_type = COL("conflict");
249 if (!strcmp(conflict_type, "hash")) {
251 fprintf(stderr, "hash conflict: package %s path %s hash %.8s\n",
254 if (!strcmp(conflict_type, "md")) {
256 fprintf(stderr, "md conflict: package %s path %s md %s\n",
259 fprintf(stderr, "%s conflict: package %s path %s\n",
260 conflict_type, pkg, path);
267 static int read_item(struct config *conf, int ncols, char **vals, char **cols,
273 struct nitem zero = { 0 };
279 seterror(conf, "can't determine op");
285 seterror(conf, "can't determine op");
289 n->path = COL("path");
291 seterror(conf, "no file path");
294 if (strlen(n->path) == 0) {
295 seterror(conf, "zero length path not allowed");
299 /* TODO config to dishonor setuid/setgid */
300 n->dest = COL("dest");
302 seterror(conf, "no file dest");
306 if (strlen(n->dest) == 0) {
307 seterror(conf, "zero length dest not allowed");
314 seterror(conf, "can't determine mode");
318 n->mode = strtoul(val, NULL, 8);
320 val = COL("configuration");
322 seterror(conf, "can't determine config status");
325 lval = strtol(val, NULL, 10);
327 n->configuration = ((lval & 1) != 0);
328 n->oldwasconf = ((lval & 2) != 0);
330 val = COL("filetype");
331 if (!val || strlen(val) == 0) {
332 seterror(conf, "can't determine file type");
337 /* these can be null */
338 n->ohash = COL("ohash");
340 n->omds = COL("omds");
341 n->pkglist = COL("pkglist");
343 if (n->ftype == 'r') {
344 n->hash = COL("hash");
346 seterror(conf, "can't get hash");
349 } else if (n->ftype == 'l') {
350 n->target = COL("target");
352 seterror(conf, "can't get target");
355 if (strlen(n->target) == 0) {
356 seterror(conf, "zero length target not allowed");
363 val = COL("username");
365 seterror(conf, "no username");
370 seterror(conf, "no passwd entry");
378 if (conf->setgroup) {
379 val = COL("groupname");
381 seterror(conf, "no groupname");
386 seterror(conf, "no group entry for %s", val);
395 double mtime = strtod(COL("mtime"),NULL);
397 mtime = (double)time(NULL);
400 n->mtime = (time_t)mtime;
402 n->times[0].tv_sec = 0;
403 n->times[0].tv_nsec = UTIME_OMIT;
404 n->times[1].tv_sec = (time_t)llrint(floor(mtime));
405 n->times[1].tv_nsec = lrint(floor(fmod(mtime,1.0)*1000000000));
410 /* file does not exist */
411 #define D_NOEXIST 0x1
412 /* files are different types */
414 /* metadata is different */
416 /* content or link target is different */
418 /* file to be installed is a directory */
420 /* path on disk is a directory */
421 #define D_EISDIR 0x20
422 /* usernames different */
424 /* group names different */
426 /* file mode is different */
428 /* mtimes are different */
429 #define D_MTIME 0x200
430 /* the hash of the file we are supposedly replacing is different than
431 * the the hash of the file on disk
433 #define D_OHASH 0x400
434 /* an error occurred trying to compare the file (other than it doesn't exist */
435 #define D_ERROR 0x1000
436 /* there was a stat error */
437 #define D_STATERROR 0x2000
438 /* there was an error calling readlink */
439 #define D_RLERROR 0x4000
441 /* 1 = file doesn't exist, 2 = file is a directory, target isn't */
442 /* 4 == ftype different */
443 /* 8 = hash different when both are regular files */
444 static unsigned int file_compare(struct nitem *n, struct stat *st) {
445 int etype = 0, stat_type;
446 char ehash[ZPM_HASH_STRLEN+1];
447 unsigned int diff = 0;
452 case 'd': etype = S_IFDIR; diff |= D_ISDIR ; break;
453 case 'r': etype = S_IFREG; break;
454 case 'l': etype = S_IFLNK; break;
455 default: etype = 0; break;
459 /* new file, so check type, hash, etc */
460 if (lstat(n->dest, st) == 0) {
461 stat_type = st->st_mode & S_IFMT;
462 if (stat_type != etype) {
465 if (stat_type == S_IFDIR) {
469 if (n->hash && etype == S_IFREG && stat_type == S_IFREG) {
470 zpm_hash(n->dest, ehash, 0);
471 if (strcmp(n->hash, ehash) != 0) {
474 if (n->ohash && strcmp(n->ohash, ehash) != 0) {
478 if (n->hash && etype == S_IFLNK && stat_type == S_IFLNK) {
479 lsize = readlink(n->dest, link, sizeof link);
481 if (lsize == -1 || lsize == sizeof link) {
486 if (strcmp(n->target, link) != 0) {
491 if (n->uid != st->st_uid) {
495 if (n->gid != st->st_gid) {
499 if (n->mode != (st->st_mode & 07777)) {
505 case ENOENT: diff |= D_NOEXIST; break;
506 default: diff |= (D_STATERROR|D_ERROR); break;
514 /* 0 = not acceptable
515 * 1 = accept and create/update/remove
517 * 3 = remove and overwrite
518 * 4 = update metadata
520 static int acceptable(struct config *conf, unsigned int diffs, int op) {
521 int exist = (!(diffs & D_NOEXIST));
522 int sametype = (!(diffs & D_TYPE));
523 int mdsame = (!(diffs & D_MD));
524 int hashsame = (!(diffs & D_HASH));
525 int isdir = (diffs & D_ISDIR);
528 return op == OP_REMOVE ? 2 : 1;
531 if (op == OP_UPDATE) {
532 return sametype ? 4 : 3;
535 if (op == OP_REMOVE) {
539 /* the hard cases, should be installing new, but already exists */
542 return conf->overwrite ? 3 : 0;
545 if (mdsame && (conf->accept || conf->overwrite)) {
550 if (mdsame || conf->ignoredirmd) {
551 return conf->acceptdir ? 2 : 0;
553 if (conf->overwrite) {
558 if (hashsame && (conf->accept || conf->overwrite)) {
562 return conf->overwrite ? 3 : 0;
565 static int check_existing(void *f, int ncols, char **vals, char **cols) {
566 struct config *conf = f;
570 if (!read_item(conf, ncols, vals, cols, &nitem)) {
571 fprintf(stderr, "can't read item\n");
572 return conf->errabort;
575 if (conf->verbose > 1) {
576 fprintf(stderr, "check for existing %s\n", nitem.path);
579 if (lstat(nitem.path, &st) == -1) {
581 /* not an error, file shouldn't exist*/
584 fprintf(stderr, "unable to check %s: %s\n",
585 nitem.path, strerror(errno));
592 unsigned int diffs = file_compare(&nitem, &st);
593 if (diffs >= D_ERROR) {
594 return seterror(conf, "can't check %s", nitem.dest);
597 int action = acceptable(conf, diffs, nitem.op);
600 fprintf(stderr, "%s exists and is not acceptable\n", nitem.path);
602 fprintf(stderr, "%s exists\n", nitem.path);
610 static int remove_files(void *f, int ncols, char **vals, char **cols) {
611 struct config *conf = f;
617 if (!dest) return seterror(conf,"no file dest");
620 char *ftype = COL("filetype");
624 case 'd': printf("rmdir %s\n", dest); break;
625 default: printf("unlink %s\n", dest); break;
631 if (lstat(dest, &st) == -1) {
632 return seterror(conf,"can't stat");
635 if (S_ISDIR(st.st_mode)) {
636 flags = AT_REMOVEDIR;
638 /* TODO check that expected filetype matches actual filetype */
641 if (conf->progress == 2) {
642 fprintf(stderr, "%s(%s)\n", flags ? "rmdir" : "unlink", dest);
643 } else if (conf->progress == 1) {
645 pdots(50, '.', conf->ops_completed, conf->ops_completed + 1, conf->ops_total);
646 conf->ops_completed++;
647 conf->ops_completed++;
649 pdots(50, '.', conf->ops_completed, conf->ops_completed + 1, conf->ops_total);
650 conf->ops_completed++;
656 if (unlinkat(AT_FDCWD, dest, flags) == -1) {
661 return seterror(conf, "can't unlink");
668 #define MARK fprintf(stderr, "%s %d: mark\n", __func__, __LINE__)
671 static int remove_dir(struct config *conf, char *path) {
676 setsyserr(conf, "can't rmdir %s", path);
682 static int remove_existing(struct config *conf, char *path) {
687 setsyserr(conf, "can't unlink %s", path);
693 static int set_md(struct config *conf, struct nitem *item) {
698 if (item->ftype != 'l') {
699 printf("chmod %o %s\n", item->mode, item->dest);
701 if (conf->setuser && conf->setgroup) {
702 printf("lchown %d:%d %s\n", item->uid, item->gid,
705 printf("mtime %.0f %s\n", (double)item->mtime, item->dest);
710 /* can't chmod a symlink */
711 if (item->ftype != 'l') {
712 rv = chmod(item->dest, item->mode);
715 setsyserr(conf, "can't chmod %o %s", item->mode, item->dest);
716 return conf->errabort;
720 if (conf->setuser && conf->setgroup) {
721 rv = lchown(item->dest, item->uid, item->gid);
723 setsyserr(conf, "can't lchown %s", item->dest);
724 return conf->errabort;
728 rv = utimensat(AT_FDCWD, item->dest, item->times, AT_SYMLINK_NOFOLLOW);
730 setsyserr(conf, "can't set mtime %.0f %s", (double)item->mtime,
732 return conf->errabort;
737 /* install a file or create a directory or symlink. path should not exist
740 /* flags: 1 = set md, 2 = create leading dirs, 4 = unlink existing file,
741 * 8 = rmdir existing dir, 16 = return true/false
745 #define INS_UNLINK 0x4
746 #define INS_RMDIR 0x8
748 #define INS_ZPMNEW 0x20
749 static int install(struct config *conf, struct nitem *item, unsigned int flags) {
753 int mkleading = (flags & 2);
754 int setmd = (flags & 1);
755 int unlink_file = (flags & 4);
756 int rm_dir = (flags & 8);
757 int failure = conf->errabort;
767 printf("unlink %s\n", item->dest);
769 printf("rmdir %s\n", item->dest);
772 printf("install %c%o %d:%d %s", item->ftype,
773 item->mode, item->uid, item->gid,
775 if (item->ftype == 'l') {
776 printf(" -> %s", item->target);
783 source = conf->src ? conf->src : conf->log;
786 rv = remove_existing(conf, item->dest);
788 rv = remove_dir(conf, item->dest);
796 rv = create_leading_dirs(item->dest);
798 setsyserr(conf, "can't create leading dirs for %s", item->dest);
803 if (item->ftype == 'r') {
804 rv = zpm_extract(source, item->hash, item->dest, item->mode);
806 seterror(conf, "can't extract %s", item->dest);
812 switch (item->ftype) {
813 case 'd': rv = mkdir(item->dest, item->mode);
815 case 'l': rv = symlink(item->target, item->dest);
822 setsyserr(conf, "installing %s failed", item->dest);
827 return set_md(conf, item) == 0 ? success : failure;
834 * figure out what the difference is for a config file, only called
835 * for an update of a configuration file
836 * return -1 on an error
837 * return 1 if the new file should not be installed
838 * return 0 if the new file should be installed
840 static int adjust_for_config(struct config *conf, struct nitem *n, unsigned int
843 if (!n->oldwasconf) {
847 /* TODO what if old was a directory? */
848 if (!n->configuration) {
849 /* replacing conf with non-conf */
850 /* absorb file, mark todo */
851 char hash[ZPM_HASH_STRLEN+1];
854 fprintf(stderr, "importing old conf file\n");
856 if (zpm_import(conf->log, n->dest, 0, hash)) {
857 zpm_note_add(conf->log, n->pkglist, n->dest, hash,
858 "replaced config file with non-config. zpm-cat %.8s", hash);
860 fprintf(stderr, "unable to import existing config file %s\n", n->dest);
864 fprintf(stderr, "dry-run: would replace config file %s with non-config file\n", n->dest);
869 int sametype = (!(diffs & D_TYPE));
870 int isdir = (diffs & D_ISDIR);
871 int eisdir = (diffs & D_EISDIR);
873 /* both old and new are config files */
874 if (isdir && sametype) {
875 /* both config directories, can only be changing
876 * metadata, so no adjustment needed
879 fprintf(stderr, "both config dirs, ok to update\n");
885 char hash[ZPM_HASH_STRLEN+1];
887 /* replacing old file with new directory */
888 /* absorb, make note */
890 if (zpm_import(conf->log, n->dest, 0, hash)) {
891 zpm_note_add(conf->log, n->pkglist, n->dest, hash,
892 "replaced config file with config directory. zpm-cat %.8s", hash);
894 fprintf(stderr, "unable to import existing config file %s\n", n->dest);
898 fprintf(stderr, "dry-run: would replace config file %s with config directory\n", n->dest);
904 /* replacing old conf directory with a conf file.
905 * nothing needs to be done, if the directory
906 * is empty, it's ok to remove. if it's not empty,
907 * the install will fail
912 /* replacing old file with new file */
913 /* new is same as on disk */
914 if (!(diffs & D_HASH)) {
916 fprintf(stderr, "new config file is already on disk, probably shouldn't happen\n");
921 /* new is different than on disk, but on disk is same as old */
922 if (!(diffs & D_OHASH)) {
924 fprintf(stderr, "old config file not changed from default, replacing with new default\n");
926 /* ok to do the update, since same as default */
927 fprintf(stderr, "updating default config %s\n", n->dest);
931 /* new is different than on disk, and disk different than old */
934 fprintf(stderr, "new default config file is different than on disk, and old default was changed, should keep on-disk config\n");
937 zpm_note_add(conf->log, n->pkglist, n->dest, n->hash,
938 "default config file update. zpm-cat %.8s", n->hash);
939 /* TODO check for note error */
941 fprintf(stderr, "dry-run: default config file %s update\n",
948 static int install_files(void *f, int ncols, char **vals, char **cols) {
949 struct config *conf = f;
951 struct stat existing;
954 /* TODO put the result row in a hash table. May not actually
957 if (!read_item(conf, ncols, vals, cols, &nitem)) {
958 fprintf(stderr, "can't read item\n");
959 return conf->errabort;
964 used = sqlite3_memory_used()/1024/1024;
965 high = sqlite3_memory_highwater(0)/1024/1024;
966 fprintf(stderr, "memory = %ld MB / %ld MB\n", used, high);
968 if (conf->verbose && !conf->dryrun) {
969 if (conf->progress == 2) {
970 fprintf(stderr, "%s '%c' %s\n", nitem.opstr, nitem.ftype,
972 } else if (conf->progress == 1) {
974 pdots(50, '.', conf->ops_completed, conf->ops_completed + 1, conf->ops_total);
975 conf->ops_completed++;
976 conf->ops_completed++;
978 pdots(50, '.', conf->ops_completed, conf->ops_completed + 1, conf->ops_total);
979 conf->ops_completed++;
983 unsigned int diffs = file_compare(&nitem, &existing);
984 if (diffs >= D_ERROR) {
985 return seterror(conf, "can't check %s", nitem.dest);
989 * exist & same type & md same & hash same: do nothing, but warn bug
990 * exist & same type & md diff & hash same: fix md
991 * exist & same type & md same & hash diff: replace
992 * exist & same type & md diff & hash diff: replace & fix
993 * no exist: install and warn
994 * dir & not dir : remove, mkdir
995 * not dir & not dir & diff type: remove, install
996 * not dir & dir : remove dir if empty, error if not empty, install
999 * no exist: create leading dirs, install
1001 * exist & same type & md same & hash same & accept or over: do nothing
1002 * exist & same & md diff or hash diff & overwrite : update
1003 * exist & same & md diff or hash diff & accept : error, can't accept
1004 * exist & same & md diff or hash diff & not accept : error
1006 * exist & different type & not overwrite : error
1007 * not dir & not dir & overwrite : remove and install
1008 * not dir & dir & overwrite: remove empty or error, install
1009 * dir & dir & overwrite: fix md
1010 * dir & not dir & overwrite: remove and mkdir
1012 int exist = (!(diffs & D_NOEXIST));
1013 int sametype = (!(diffs & D_TYPE));
1014 int mdsame = (!(diffs & D_MD));
1015 int hashsame = (!(diffs & D_HASH));
1016 int isdir = (diffs & D_ISDIR);
1017 int eisdir = (diffs & D_EISDIR);
1018 int accept = conf->accept;
1019 int overwrite = conf->overwrite;
1020 int installing = (nitem.op == OP_NEW);
1021 update = (nitem.op == OP_UPDATE);
1025 /* warn, it should exist */
1026 fprintf(stderr, "%s missing, installing", nitem.dest);
1027 return install(conf, &nitem, 3);
1030 switch (adjust_for_config(conf, &nitem, diffs)) {
1031 case -1: return conf->errabort; break;
1033 fprintf(stderr, "skipping changed default config file: %s\n", nitem.dest);
1038 /* file exists in filesystem */
1040 if (mdsame && hashsame) {
1041 /* warn, bug in logic. This shouldn't occur,
1042 * because if there is nothing to do, it
1043 * shouldn't be listed as an update
1045 /* could be an update. We're checking against
1046 * what's actually on disk, not what was
1047 * expected to have been on disk. So, if
1048 * the admin has modified the file, or if
1049 * it had been installed ignoring the user
1050 * and group, it might be correct on disk
1051 * but not as in the local database
1053 /* TODO detect whether this a logic bug or
1054 * an on-disk difference
1057 fprintf(stderr, "%s should not be an update\n", nitem.dest);
1058 fprintf(stderr, "old hash: %s\n", nitem.ohash);
1059 fprintf(stderr, "new hash: %s\n", nitem.hash);
1060 fprintf(stderr, "old mds: %s\n", nitem.omds);
1061 fprintf(stderr, "new mds: %s\n", nitem.mds);
1066 if (!mdsame && hashsame) {
1068 return set_md(conf, &nitem);
1070 if (mdsame && !hashsame) {
1072 return install(conf, &nitem, 3);
1074 if (!mdsame && !hashsame) {
1076 return install(conf, &nitem, 3);
1080 /* file exists, and is not the same type */
1082 if (isdir && !eisdir) {
1083 /* remove existing */
1085 return install(conf, &nitem, 7);
1087 if (!isdir && eisdir) {
1088 /* remove dir, or error */
1090 return install(conf, &nitem, 11);
1092 if (!isdir && !isdir) {
1093 /* necessarily !sametype, sametype handled above */
1094 /* remove existing */
1096 return install(conf, &nitem, 7);
1098 /* error, should not be possible, assert(0)? */
1099 fprintf(stderr,"impossible state: %s:%d\n", __func__, __LINE__);
1104 return install(conf, &nitem, 3);
1107 /* file exists in filesystem */
1109 if (mdsame && hashsame && (accept || overwrite)) {
1111 if (conf->dryrun || conf->verbose) {
1112 fprintf(stderr, "accept %s: %s\n",
1113 eisdir ? "directory" : "file", nitem.dest);
1118 if (mdsame && isdir && conf->acceptdir) {
1122 if (!mdsame && isdir && conf->ignoredirmd) {
1123 /* TODO warn ignoring dir md */
1127 if (mdsame && hashsame && !(accept || overwrite)) {
1129 return seterror(conf, "file exists: %s", nitem.dest);
1132 if (mdsame && !hashsame && overwrite) {
1134 return install(conf, &nitem, eisdir ? 11 : 7);
1137 if (nitem.configuration && accept) {
1138 /* accept a changed config file */
1139 if (conf->dryrun || conf->verbose) {
1140 fprintf(stderr, "accept %smodified config %s: %s\n", (!mdsame || !hashsame) ? "" : "un",
1141 eisdir ? "directory" : "file", nitem.dest);
1146 if (mdsame && !hashsame && !overwrite) {
1147 /* accept doesn't matter, since it's
1148 * not an acceptable file */
1150 if (nitem.ftype == 'l') {
1153 lsize = readlink(nitem.dest, link, sizeof link);
1154 if (lsize == -1 || (size_t)lsize >= sizeof link) {
1155 return seterror(conf, "%s (linkdiff): expecting %s -> %s, unable to read link", accept ? "existing file not acceptable" : "file exists", nitem.dest, nitem.target, link);
1158 /* links must be different */
1159 return seterror(conf, "%s (linkdiff): expecting %s -> %s, have -> %s", accept ? "existing file not acceptable" : "file exists", nitem.dest, nitem.target, link);
1162 return seterror(conf, "%s (hashdiff): %s", accept ? "existing file not acceptable" : "file exists", nitem.dest);
1165 if (!mdsame && hashsame && overwrite) {
1167 return set_md(conf, &nitem);
1169 if (!mdsame && hashsame && !overwrite) {
1170 /* accept doesn't matter, since it's
1171 * not an acceptable file */
1173 return seterror(conf, "%s (mddiff): %s", accept ? "existing file not acceptable" : "file exists", nitem.dest);
1175 if (!mdsame && !hashsame && overwrite) {
1177 return install(conf, &nitem, eisdir ? 11 : 7);
1179 if (!mdsame && !hashsame && !overwrite) {
1180 /* accept doesn't matter, since it's
1181 * not an acceptable file */
1183 return seterror(conf, "%s (md+hash): %s", accept ? "existing file not acceptable" : "file exists", nitem.dest);
1185 /* TODO error, should be impossible */
1186 return seterror(conf, "impossible state reached");
1189 /* file exists, and is not the same type */
1192 return seterror(conf, "%s (difftype): %s", accept ? "existing file not acceptable" : "file exists", nitem.dest);
1195 /* not the same type, but ok to overwrite */
1197 /* remove existing */
1198 return install(conf, &nitem, 7);
1201 /* existing path is a directory */
1204 /* impossible, if isdir and eisdir, would
1208 return set_md(conf, &nitem);
1210 /* remove empty dir or error */
1212 return install(conf, &nitem, 11);
1214 /* if we get here, we missed a case */
1216 return seterror(conf, "impossible state 2 reached");
1219 /* TODO extra verbose print perms, mtime, etc, probably ls -l
1222 if (conf->verbose) {
1223 printf("%s\n", nitem.path);
1229 static void check_conflicts(struct config *conf, char *conflict_type,
1230 int (callback)(void *, int, char **, char **)) {
1236 s = sqlite3_str_new(conf->log->db);
1237 sqlite3_str_appendall(s, "select *, ");
1238 if (conf->rootdir) {
1239 sqlite3_str_appendf(s, "printf('%%s/%%s',rtrim(%Q,'/'),ltrim(path,'/'))", conf->rootdir);
1241 sqlite3_str_appendf(s, "printf('/%%s', trim(path, '/'))");
1243 sqlite3_str_appendall(s, " as dest from syncconflicts");
1245 if (conflict_type) {
1246 sqlite3_str_appendf(s," where conflict = %Q", conflict_type);
1248 if (conf->reverse) {
1249 sqlite3_str_appendall(s," order by length(path) desc, path desc,pkgid collate vercmp desc, conflict desc");
1251 sqlite3_str_appendall(s," order by length(path), path, pkgid collate vercmp, conflict");
1255 sql = sqlite3_str_value(s);
1257 rv = zpm_exec(conf->log, sql, callback, conf, &errmsg);
1259 sqlite3_str_finish(s);
1262 fprintf(stderr, "exec fail: %s\n", sqlite3_errstr(rv));
1264 fprintf(stderr, "database error: %s\n", errmsg);
1267 if (conf->log->error == 1) {
1268 fprintf(stderr, "unable to allocate memory\n");
1270 fprintf(stderr, "zpm_exec failure: %s\n",
1271 conf->log->errmsg ? conf->log->errmsg : "unknown");
1274 if (conf->log->errmsg) {
1275 fprintf(stderr, "error: %s\n", conf->log->errmsg);
1277 if (conf->errors && conf->exitonerror) {
1278 zpm_close(conf->log);
1279 zpm_close(conf->src);
1282 /* TODO final report function in conf var */
1285 static void runstage(struct config *conf, char *stage,
1286 int (callback)(void *, int, char **, char **)) {
1292 s = sqlite3_str_new(conf->log->db);
1293 sqlite3_str_appendall(s, "select *, ");
1294 if (conf->rootdir) {
1295 sqlite3_str_appendf(s, "printf('%%s/%%s',rtrim(%Q,'/'),ltrim(path,'/'))", conf->rootdir);
1297 sqlite3_str_appendf(s, "printf('/%%s', trim(path, '/'))");
1299 sqlite3_str_appendall(s, " as dest from syncinfo");
1302 sqlite3_str_appendf(s," where op = %Q", stage);
1304 if (conf->reverse) {
1305 sqlite3_str_appendall(s," order by length(path) desc, path desc");
1308 sql = sqlite3_str_value(s);
1310 rv = zpm_exec(conf->log, sql, callback, conf, &errmsg);
1312 sqlite3_str_finish(s);
1315 fprintf(stderr, "exec fail: %s\n", sqlite3_errstr(rv));
1317 fprintf(stderr, "database error: %s\n", errmsg);
1320 if (conf->log->error == 1) {
1321 fprintf(stderr, "unable to allocate memory\n");
1323 fprintf(stderr, "zpm_exec failure: %s\n",
1324 conf->log->errmsg ? conf->log->errmsg : "unknown");
1328 if (conf->log->errmsg) {
1329 fprintf(stderr, "error: %s\n", conf->log->errmsg);
1332 if (conf->errors && conf->exitonerror) {
1333 zpm_close(conf->log);
1334 zpm_close(conf->src);
1337 /* TODO final report function in conf var */
1340 static void count_ops(struct config *conf) {
1341 conf->ops_remove = zpm_db_int(conf->log, "select count(*) from syncinfo where op = 'remove'");
1342 conf->ops_update = zpm_db_int(conf->log, "select count(*) from syncinfo where op = 'update'");
1343 conf->ops_install = zpm_db_int(conf->log, "select count(*) from syncinfo where op = 'new'");
1344 conf->ops_total = conf->ops_remove + conf->ops_update + conf->ops_install;
1347 int main(int ac, char **av) {
1351 char *pkgdbfile = 0, *localdbfile = 0;
1354 struct config conf = { 0 };
1371 if (geteuid() != 0) {
1376 localdbfile = ZPM_LOCAL_DB;
1377 if ((s = getenv("ZPMDB"))) {
1378 /* TODO does this need to be copied ? */
1382 if ((s = getenv("ZPM_ROOT_DIR"))) {
1383 /* TODO does this need to be copied ? */
1388 * -d localdb or ZPMDB * or /var/lib/zpm/zpm.db, or die
1389 * -f 'package database', otherwise regular default of env
1390 * ZPM_PACKAGE_FILE, or use pkgdb if otherwise not found
1391 * -R root of pkg, will just chdir there
1393 * args are pkgid triple, but will do a pkg find on the pkgdb
1396 while ((opt = getopt(ac, av, "f:d:c:nCR:vOAMDp")) != -1) {
1398 case 'd': localdbfile = optarg; break;
1399 case 'f': pkgdbfile = optarg; break;
1400 case 'n': conf.dryrun = 1; break;
1401 case 'v': conf.verbose++; break;
1402 case 'C': conf.errabort = 0; break;
1403 case 'R': conf.rootdir = optarg; break;
1404 case 'N': conf.setuser = 0; conf.setgroup = 0; break;
1405 case 'O': conf.overwrite = 1; break;
1406 case 'A': conf.accept = 1; break;
1407 case 'M': conf.ignoredirmd = 1;
1408 case 'D': conf.acceptdir = 0;
1409 case 'p': conf.progress++;
1417 /* verify root dir exists */
1418 if (conf.rootdir && !exists(conf.rootdir, NULL)) {
1419 fprintf(stderr, "rootdir %s does not exist\n", conf.rootdir);
1422 if (!zpm_open(&localdb, localdbfile)) {
1423 fprintf(stderr, "can't open zpm db %s\n", localdbfile);
1426 conf.log = &localdb;
1429 /* TODO open read-only */
1430 if (!zpm_open(&pkgdb, pkgdbfile)) {
1431 fprintf(stderr, "can't open src db %s\n", localdbfile);
1438 /* TODO find pkgid from arg */
1440 /* TODO set conf var to finalize error reporting */
1442 fprintf(stderr, "syncing filesystem %s (ldb %s) from %s\n",
1443 conf.rootdir ? conf.rootdir : "/",
1444 localdbfile, pkgdbfile);
1448 conf.exitonerror = 0;
1449 check_conflicts(&conf, NULL, report_conflicts);
1451 if (conf.conflicts) {
1452 fprintf(stderr, "%d conflicts reported, aborting sync\n",
1456 /* no point in running it if we're just going to
1457 * overwrite everything
1459 if (!conf.overwrite && !conf.accept && !conf.dryrun) {
1460 runstage(&conf, "new", check_existing);
1464 fprintf(stderr, "beginning %ssync\n", conf.dryrun ?
1467 /* have to do the removes first otherwise
1468 * old files may conflict with update file
1472 conf.exitonerror = conf.dryrun ? 0 : 1;
1473 conf.errabort = conf.dryrun ? 0 : 1;
1475 fprintf(stderr, "file ops: %d\n", conf.ops_total);
1476 if (conf.ops_remove > 0) {
1478 fprintf(stderr, "removing %d file%s\n", conf.ops_remove, conf.ops_remove > 0 ? "s" : "");
1481 conf.ops_completed = 0;
1482 conf.ops_total = conf.ops_remove;
1483 runstage(&conf, "remove", remove_files);
1484 if (conf.verbose && conf.progress < 2) {
1485 fprintf(stderr, " done\n");
1490 if (conf.ops_update > 0) {
1492 fprintf(stderr, "updating %d file%s\n", conf.ops_update, conf.ops_update > 0 ? "s" : "");
1495 conf.ops_completed = 0;
1496 conf.ops_total = conf.ops_update;
1497 runstage(&conf, "update", install_files);
1498 if (conf.verbose && conf.progress < 2) {
1499 fprintf(stderr, " done\n");
1504 if (conf.ops_install > 0) {
1506 fprintf(stderr, "installing %d file%s\n", conf.ops_install, conf.ops_install > 0 ? "s" : "");
1509 conf.ops_completed = 0;
1510 conf.ops_total = conf.ops_install;
1511 runstage(&conf, "new", install_files);
1512 if (conf.verbose && conf.progress < 2) {
1513 fprintf(stderr, " done\n");
1520 zpm_close(&localdb);
1521 zpm_close(conf.src);
1522 return conf.errors ? 1 : 0;