diff options
Diffstat (limited to 'src/pacsync.c')
| -rw-r--r-- | src/pacsync.c | 183 |
1 files changed, 183 insertions, 0 deletions
diff --git a/src/pacsync.c b/src/pacsync.c new file mode 100644 index 00000000..224528f8 --- /dev/null +++ b/src/pacsync.c @@ -0,0 +1,183 @@ +/* + * pacman + * + * Copyright (c) 2002 by Judd Vinet <jvinet@zeroflux.org> + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, + * USA. + */ + +#include "config.h" +#include <stdio.h> +#include <stdlib.h> +#include <limits.h> +#include <errno.h> +#include <string.h> +#include <sys/types.h> +#include <unistd.h> +#include <ftplib.h> +/* pacman */ +#include "list.h" +#include "package.h" +#include "db.h" +#include "util.h" +#include "pacsync.h" +#include "pacman.h" + +static int log_progress(netbuf *ctl, int xfered, void *arg); +static char sync_fnm[25]; + +/* pacman options */ +extern char *pmo_root; +/* configuration options */ +extern char pmc_syncserver[512]; +extern char pmc_syncpath[512]; +extern char pmc_syncname[512]; + +int sync_synctree() +{ + char ldir[PATH_MAX] = ""; + char path[PATH_MAX]; + mode_t oldmask; + char *str; + PMList *files = NULL; + + snprintf(ldir, PATH_MAX, "%s%s/%s", pmo_root, PKGDIR, pmc_syncname); + + /* remove the old dir */ + vprint("Removing %s (if it exists)\n", ldir); + rmrf(ldir); + + /* make the new dir */ + oldmask = umask(0000); + mkdir(ldir, 0755); + umask(oldmask); + + /* build out list of one */ + snprintf(path, PATH_MAX, "%s.db.tar.gz", pmc_syncname); + str = strdup(path); + files = list_add(files, str); + if(downloadfiles(pmc_syncserver, pmc_syncpath, ldir, files)) { + list_free(files); + return(1); + } + + /* uncompress the sync database */ + snprintf(path, PATH_MAX, "%s/%s", ldir, (char*)files->data); + list_free(files); + vprint("Unpacking %s...\n", path); + if(unpack(path, ldir)) { + return(1); + } + + /* remove the .tar.gz */ + unlink(path); + + return(0); +} + +int downloadfiles(char *server, char *remotepath, char *localpath, PMList *files) +{ + int fsz; + netbuf *control = NULL; + PMList *lp; + int ret = 0; + + if(files == NULL) { + return(0); + } + + FtpInit(); + if(!FtpConnect(server, &control)) { + fprintf(stderr, "error: cannot connect to %s\n", server); + return(1); + } + if(!FtpLogin("anonymous", "arch@guest", control)) { + fprintf(stderr, "error: anonymous login failed\n"); + FtpQuit(control); + return(1); + } + + + if(!FtpChdir(remotepath, control)) { + fprintf(stderr, "error: could not cwd to %s: %s\n", remotepath, + FtpLastResponse(control)); + return(1); + } + + /* get each file in the list */ + for(lp = files; lp; lp = lp->next) { + char output[PATH_MAX]; + int j; + + snprintf(output, PATH_MAX, "%s/%s", localpath, (char*)lp->data); + + /* passive mode */ + /* TODO: make passive ftp an option */ + if(!FtpOptions(FTPLIB_CONNMODE, FTPLIB_PASSIVE, control)) { + fprintf(stderr, "warning: failed to set passive mode\n"); + } + + if(!FtpSize((char*)lp->data, &fsz, FTPLIB_IMAGE, control)) { + fprintf(stderr, "warning: failed to get filesize for %s\n", (char*)lp->data); + } + + /* set up our progress bar's callback */ + FtpOptions(FTPLIB_CALLBACK, (long)log_progress, control); + FtpOptions(FTPLIB_IDLETIME, (long)1000, control); + FtpOptions(FTPLIB_CALLBACKARG, (long)&fsz, control); + FtpOptions(FTPLIB_CALLBACKBYTES, (10*1024), control); + + strncpy(sync_fnm, lp->data, 24); + for(j = strlen(sync_fnm); j < 24; j++) { + sync_fnm[j] = ' '; + } + sync_fnm[24] = '\0'; + if(!FtpGet(output, lp->data, FTPLIB_IMAGE, control)) { + fprintf(stderr, "\nerror: could not download %s: %s\n", (char*)lp->data, + FtpLastResponse(control)); + /* unlink the file */ + unlink(output); + ret = 1; + } else { + log_progress(control, fsz, &fsz); + } + printf("\n"); + fflush(stdout); + } + + FtpQuit(control); + return(ret); +} + +static int log_progress(netbuf *ctl, int xfered, void *arg) +{ + int fsz = *(int*)arg; + int pct = (unsigned int)(xfered * 100) / fsz; + int i; + + printf("%s [", sync_fnm); + for(i = 0; i < (int)(pct/3); i++) { + printf("#"); + } + for(i = (int)(pct/3); i < (int)(100/3); i++) { + printf(" "); + } + printf("] %3d%% | %6dK\r", pct, (xfered/1024)); + fflush(stdout); + return(1); +} + +/* vim: set ts=2 sw=2 noet: */ |