int main(int argc, char **argv)
{
int c, lockfd, ret = -1;
int dump=0, scan=0, rebuild=0, prep_upgrade=0, rebuildindexes=0, dumpindexes=0, force=0;
dbd_flags_t flags = 0;
char *volpath;
int cdir;
if (geteuid() != 0) {
usage();
exit(EXIT_FAILURE);
}
/* Inhereting perms in ad_mkdir etc requires this */
ad_setfuid(0);
while ((c = getopt(argc, argv, ":cdefinrstuvx")) != -1) {
switch(c) {
case 'c':
flags |= DBD_FLAGS_CLEANUP;
break;
case 'd':
dump = 1;
break;
case 'i':
dumpindexes = 1;
break;
case 's':
scan = 1;
flags |= DBD_FLAGS_SCAN;
break;
case 'n':
nocniddb = 1; /* FIXME: this could/should be a flag too for consistency */
break;
case 'r':
rebuild = 1;
break;
case 't':
flags |= DBD_FLAGS_STATS;
break;
case 'u':
prep_upgrade = 1;
break;
case 'v':
verbose = 1;
break;
case 'e':
exclusive = 1;
flags |= DBD_FLAGS_EXCL;
break;
case 'x':
rebuildindexes = 1;
break;
case 'f':
force = 1;
exclusive = 1;
flags |= DBD_FLAGS_FORCE | DBD_FLAGS_EXCL;
break;
case ':':
case '?':
usage();
exit(EXIT_FAILURE);
break;
}
}
if ((dump + scan + rebuild + prep_upgrade) != 1) {
usage();
exit(EXIT_FAILURE);
}
if ( (optind + 1) != argc ) {
usage();
exit(EXIT_FAILURE);
}
volpath = argv[optind];
setvbuf(stdout, (char *) NULL, _IONBF, 0);
/* Remember cwd */
if ((cdir = open(".", O_RDONLY)) < 0) {
dbd_log( LOGSTD, "Can't open dir: %s", strerror(errno));
exit(EXIT_FAILURE);
}
/* Setup signal handling */
set_signal();
/* Setup logging. Should be portable among *NIXes */
if (!verbose)
setuplog("default log_info /dev/tty");
else
setuplog("default log_debug /dev/tty");
/* Load .volinfo file */
if (loadvolinfo(volpath, &volinfo) == -1) {
dbd_log( LOGSTD, "Not a Netatalk volume at '%s', no .volinfo file at '%s/.AppleDesktop/.volinfo' or unknown volume options", volpath, volpath);
exit(EXIT_FAILURE);
}
if (vol_load_charsets(&volinfo) == -1) {
dbd_log( LOGSTD, "Error loading charsets!");
exit(EXIT_FAILURE);
}
/* Sanity checks to ensure we can touch this volume */
if (volinfo.v_vfs_ea != AFPVOL_EA_AD && volinfo.v_vfs_ea != AFPVOL_EA_SYS) {
dbd_log( LOGSTD, "Unknown Extended Attributes option: %u", volinfo.v_vfs_ea);
exit(EXIT_FAILURE);
}
/* Enuser dbpath is there, create if necessary */
struct stat st;
if (stat(volinfo.v_dbpath, &st) != 0) {
if (errno != ENOENT) {
dbd_log( LOGSTD, "Can't stat dbpath \"%s\": %s", volinfo.v_dbpath, strerror(errno));
exit(EXIT_FAILURE);
}
if ((mkdir(volinfo.v_dbpath, 0755)) != 0) {
dbd_log( LOGSTD, "Can't create dbpath \"%s\": %s", dbpath, strerror(errno));
exit(EXIT_FAILURE);
}
}
/* Put "/.AppleDB" at end of volpath, get path from volinfo file */
if ( (strlen(volinfo.v_dbpath) + strlen("/.AppleDB")) > MAXPATHLEN ) {
dbd_log( LOGSTD, "Volume pathname too long");
exit(EXIT_FAILURE);
}
strncpy(dbpath, volinfo.v_dbpath, MAXPATHLEN - strlen("/.AppleDB"));
strcat(dbpath, "/.AppleDB");
/* Check or create dbpath */
int dbdirfd = open(dbpath, O_RDONLY);
if (dbdirfd == -1 && errno == ENOENT) {
if (errno == ENOENT) {
if ((mkdir(dbpath, 0755)) != 0) {
dbd_log( LOGSTD, "Can't create .AppleDB for \"%s\": %s", dbpath, strerror(errno));
exit(EXIT_FAILURE);
}
} else {
dbd_log( LOGSTD, "Somethings wrong with .AppleDB for \"%s\", giving up: %s", dbpath, strerror(errno));
exit(EXIT_FAILURE);
}
} else {
close(dbdirfd);
}
/* Get db lock */
if ((db_locked = get_lock(LOCK_EXCL, dbpath)) == -1)
goto exit_failure;
if (db_locked != LOCK_EXCL) {
/* Couldn't get exclusive lock, try shared lock if -e wasn't requested */
if (exclusive) {
dbd_log(LOGSTD, "Database is in use and exlusive was requested");
goto exit_failure;
}
if ((db_locked = get_lock(LOCK_SHRD, NULL)) != LOCK_SHRD)
goto exit_failure;
}
/* Prepare upgrade ? */
if (prep_upgrade) {
if (dbif_env_remove(dbpath))
goto exit_failure;
goto exit_success;
}
/* Check if -f is requested and wipe db if yes */
if ((flags & DBD_FLAGS_FORCE) && rebuild && (volinfo.v_flags & AFPVOL_CACHE)) {
char cmd[8 + MAXPATHLEN];
if ((db_locked = get_lock(LOCK_FREE, NULL)) != 0)
goto exit_failure;
snprintf(cmd, 8 + MAXPATHLEN, "rm -rf \"%s\"", dbpath);
dbd_log( LOGDEBUG, "Removing old database of volume: '%s'", volpath);
system(cmd);
if ((mkdir(dbpath, 0755)) != 0) {
dbd_log( LOGSTD, "Can't create dbpath \"%s\": %s", dbpath, strerror(errno));
exit(EXIT_FAILURE);
}
dbd_log( LOGDEBUG, "Removed old database.");
if ((db_locked = get_lock(LOCK_EXCL, dbpath)) == -1)
goto exit_failure;
}
/*
Lets start with the BerkeleyDB stuff
*/
if ( ! nocniddb) {
if ((dbd = dbif_init(dbpath, "cnid2.db")) == NULL)
goto exit_failure;
if (dbif_env_open(dbd,
&db_param,
(db_locked == LOCK_EXCL) ? (DBOPTIONS | DB_RECOVER) : DBOPTIONS) < 0) {
dbd_log( LOGSTD, "error opening database!");
goto exit_failure;
}
if (db_locked == LOCK_EXCL)
dbd_log( LOGDEBUG, "Finished recovery.");
if (dbif_open(dbd, NULL, rebuildindexes) < 0) {
dbif_close(dbd);
goto exit_failure;
}
}
/* Downgrade db lock if not running exclusive */
if (!exclusive && (db_locked == LOCK_EXCL)) {
if (get_lock(LOCK_UNLOCK, NULL) != 0)
goto exit_failure;
if (get_lock(LOCK_SHRD, NULL) != LOCK_SHRD)
goto exit_failure;
}
/* Now execute given command scan|rebuild|dump */
if (dump && ! nocniddb) {
if (dbif_dump(dbd, dumpindexes) < 0) {
dbd_log( LOGSTD, "Error dumping database");
}
} else if ((rebuild && ! nocniddb) || scan) {
if (cmd_dbd_scanvol(dbd, &volinfo, flags) < 0) {
dbd_log( LOGSTD, "Error repairing database.");
}
}
/* Cleanup */
dbd_log(LOGDEBUG, "Closing db");
if (! nocniddb) {
if (dbif_close(dbd) < 0) {
dbd_log( LOGSTD, "Error closing database");
goto exit_failure;
}
}
exit_success:
ret = 0;
exit_failure:
get_lock(0, NULL);
if ((fchdir(cdir)) < 0)
dbd_log(LOGSTD, "fchdir: %s", strerror(errno));
if (ret == 0)
exit(EXIT_SUCCESS);
else
exit(EXIT_FAILURE);
}
int ad_cp(int argc, char *argv[])
{
struct stat to_stat, tmp_stat;
int r, ch, have_trailing_slash;
char *target;
#if 0
afpvol_t srcvol;
afpvol_t dstvol;
#endif
ppdid = pdid = htonl(1);
did = htonl(2);
while ((ch = getopt(argc, argv, "afinpRvx")) != -1)
switch (ch) {
case 'a':
pflag = 1;
Rflag = 1;
break;
case 'f':
fflag = 1;
iflag = nflag = 0;
break;
case 'i':
iflag = 1;
fflag = nflag = 0;
break;
case 'n':
nflag = 1;
fflag = iflag = 0;
break;
case 'p':
pflag = 1;
break;
case 'R':
Rflag = 1;
break;
case 'v':
vflag = 1;
break;
case 'x':
ftw_options |= FTW_MOUNT;
break;
default:
usage_cp();
break;
}
argc -= optind;
argv += optind;
if (argc < 2)
usage_cp();
set_signal();
cnid_init();
/* Save the target base in "to". */
target = argv[--argc];
if ((strlcpy(to.p_path, target, PATH_MAX)) >= PATH_MAX)
ERROR("%s: name too long", target);
to.p_end = to.p_path + strlen(to.p_path);
if (to.p_path == to.p_end) {
*to.p_end++ = '.';
*to.p_end = 0;
}
have_trailing_slash = (to.p_end[-1] == '/');
if (have_trailing_slash)
STRIP_TRAILING_SLASH(to);
to.target_end = to.p_end;
/* Set end of argument list */
argv[argc] = NULL;
/*
* Cp has two distinct cases:
*
* cp [-R] source target
* cp [-R] source1 ... sourceN directory
*
* In both cases, source can be either a file or a directory.
*
* In (1), the target becomes a copy of the source. That is, if the
* source is a file, the target will be a file, and likewise for
* directories.
*
* In (2), the real target is not directory, but "directory/source".
*/
r = stat(to.p_path, &to_stat);
if (r == -1 && errno != ENOENT)
ERROR("%s", to.p_path);
if (r == -1 || !S_ISDIR(to_stat.st_mode)) {
/*
* Case (1). Target is not a directory.
*/
if (argc > 1)
ERROR("%s is not a directory", to.p_path);
/*
* Need to detect the case:
*cp -R dir foo
* Where dir is a directory and foo does not exist, where
* we want pathname concatenations turned on but not for
* the initial mkdir().
*/
if (r == -1) {
lstat(*argv, &tmp_stat);
if (S_ISDIR(tmp_stat.st_mode) && Rflag)
type = DIR_TO_DNE;
else
type = FILE_TO_FILE;
} else
type = FILE_TO_FILE;
if (have_trailing_slash && type == FILE_TO_FILE) {
if (r == -1)
ERROR("directory %s does not exist", to.p_path);
else
ERROR("%s is not a directory", to.p_path);
}
} else
/*
* Case (2). Target is a directory.
*/
type = FILE_TO_DIR;
/*
* Keep an inverted copy of the umask, for use in correcting
* permissions on created directories when not using -p.
*/
mask = ~umask(0777);
umask(~mask);
#if 0
/* Inhereting perms in ad_mkdir etc requires this */
ad_setfuid(0);
#endif
/* Load .volinfo file for destination*/
openvol(to.p_path, &dvolume);
for (int i = 0; argv[i] != NULL; i++) {
/* Load .volinfo file for source */
openvol(argv[i], &svolume);
if (nftw(argv[i], copy, upfunc, 20, ftw_options) == -1) {
if (alarmed) {
SLOG("...break");
} else {
SLOG("Error: %s: %s", argv[i], strerror(errno));
}
closevol(&svolume);
closevol(&dvolume);
}
}
return rval;
}
