| /* |
| * Copyright (C) 2007 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <dirent.h> |
| #include <errno.h> |
| #include <inttypes.h> |
| #include <limits.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <sys/stat.h> |
| #include <sys/time.h> |
| #include <sys/types.h> |
| #include <time.h> |
| #include <utime.h> |
| |
| #include "sysdeps.h" |
| |
| #include "adb.h" |
| #include "adb_client.h" |
| #include "adb_io.h" |
| #include "file_sync_service.h" |
| |
| static unsigned long long total_bytes; |
| static long long start_time; |
| |
| static long long NOW() |
| { |
| struct timeval tv; |
| gettimeofday(&tv, 0); |
| return ((long long) tv.tv_usec) + |
| 1000000LL * ((long long) tv.tv_sec); |
| } |
| |
| static void BEGIN() |
| { |
| total_bytes = 0; |
| start_time = NOW(); |
| } |
| |
| static void END() |
| { |
| long long t = NOW() - start_time; |
| if(total_bytes == 0) return; |
| |
| if (t == 0) /* prevent division by 0 :-) */ |
| t = 1000000; |
| |
| fprintf(stderr,"%lld KB/s (%lld bytes in %lld.%03llds)\n", |
| ((total_bytes * 1000000LL) / t) / 1024LL, |
| total_bytes, (t / 1000000LL), (t % 1000000LL) / 1000LL); |
| } |
| |
| static void print_transfer_progress(uint64_t bytes_current, |
| uint64_t bytes_total) { |
| if (bytes_total == 0) return; |
| |
| fprintf(stderr, "\rTransferring: %" PRIu64 "/%" PRIu64 " (%d%%)", |
| bytes_current, bytes_total, |
| (int) (bytes_current * 100 / bytes_total)); |
| |
| if (bytes_current == bytes_total) { |
| fputc('\n', stderr); |
| } |
| |
| fflush(stderr); |
| } |
| |
| static void sync_quit(int fd) { |
| syncmsg msg; |
| |
| msg.req.id = ID_QUIT; |
| msg.req.namelen = 0; |
| |
| WriteFdExactly(fd, &msg.req, sizeof(msg.req)); |
| } |
| |
| typedef void (*sync_ls_cb)(unsigned mode, unsigned size, unsigned time, const char *name, void *cookie); |
| |
| static int sync_ls(int fd, const char* path, sync_ls_cb func, void* cookie) { |
| syncmsg msg; |
| char buf[257]; |
| int len; |
| |
| len = strlen(path); |
| if(len > 1024) goto fail; |
| |
| msg.req.id = ID_LIST; |
| msg.req.namelen = htoll(len); |
| |
| if(!WriteFdExactly(fd, &msg.req, sizeof(msg.req)) || |
| !WriteFdExactly(fd, path, len)) { |
| goto fail; |
| } |
| |
| for(;;) { |
| if(!ReadFdExactly(fd, &msg.dent, sizeof(msg.dent))) break; |
| if(msg.dent.id == ID_DONE) return 0; |
| if(msg.dent.id != ID_DENT) break; |
| |
| len = ltohl(msg.dent.namelen); |
| if(len > 256) break; |
| |
| if(!ReadFdExactly(fd, buf, len)) break; |
| buf[len] = 0; |
| |
| func(ltohl(msg.dent.mode), |
| ltohl(msg.dent.size), |
| ltohl(msg.dent.time), |
| buf, cookie); |
| } |
| |
| fail: |
| adb_close(fd); |
| return -1; |
| } |
| |
| struct syncsendbuf { |
| unsigned id; |
| unsigned size; |
| char data[SYNC_DATA_MAX]; |
| }; |
| |
| static syncsendbuf send_buffer; |
| |
| static int sync_readtime(int fd, const char* path, unsigned int* timestamp, unsigned int* mode) { |
| syncmsg msg; |
| int len = strlen(path); |
| |
| msg.req.id = ID_STAT; |
| msg.req.namelen = htoll(len); |
| |
| if(!WriteFdExactly(fd, &msg.req, sizeof(msg.req)) || |
| !WriteFdExactly(fd, path, len)) { |
| return -1; |
| } |
| |
| if(!ReadFdExactly(fd, &msg.stat, sizeof(msg.stat))) { |
| return -1; |
| } |
| |
| if(msg.stat.id != ID_STAT) { |
| return -1; |
| } |
| |
| *timestamp = ltohl(msg.stat.time); |
| *mode = ltohl(msg.stat.mode); |
| return 0; |
| } |
| |
| static int sync_start_readtime(int fd, const char *path) |
| { |
| syncmsg msg; |
| int len = strlen(path); |
| |
| msg.req.id = ID_STAT; |
| msg.req.namelen = htoll(len); |
| |
| if(!WriteFdExactly(fd, &msg.req, sizeof(msg.req)) || |
| !WriteFdExactly(fd, path, len)) { |
| return -1; |
| } |
| |
| return 0; |
| } |
| |
| static int sync_finish_readtime(int fd, unsigned int *timestamp, |
| unsigned int *mode, unsigned int *size) |
| { |
| syncmsg msg; |
| |
| if(!ReadFdExactly(fd, &msg.stat, sizeof(msg.stat))) |
| return -1; |
| |
| if(msg.stat.id != ID_STAT) |
| return -1; |
| |
| *timestamp = ltohl(msg.stat.time); |
| *mode = ltohl(msg.stat.mode); |
| *size = ltohl(msg.stat.size); |
| |
| return 0; |
| } |
| |
| static int sync_readmode(int fd, const char* path, unsigned* mode) { |
| syncmsg msg; |
| int len = strlen(path); |
| |
| msg.req.id = ID_STAT; |
| msg.req.namelen = htoll(len); |
| |
| if(!WriteFdExactly(fd, &msg.req, sizeof(msg.req)) || |
| !WriteFdExactly(fd, path, len)) { |
| return -1; |
| } |
| |
| if(!ReadFdExactly(fd, &msg.stat, sizeof(msg.stat))) { |
| return -1; |
| } |
| |
| if(msg.stat.id != ID_STAT) { |
| return -1; |
| } |
| |
| *mode = ltohl(msg.stat.mode); |
| return 0; |
| } |
| |
| static int write_data_file(int fd, const char *path, syncsendbuf *sbuf, int show_progress) |
| { |
| int lfd, err = 0; |
| unsigned long long size = 0; |
| |
| lfd = adb_open(path, O_RDONLY); |
| if(lfd < 0) { |
| fprintf(stderr,"cannot open '%s': %s\n", path, strerror(errno)); |
| return -1; |
| } |
| |
| if (show_progress) { |
| // Determine local file size. |
| struct stat st; |
| if (stat(path, &st)) { |
| fprintf(stderr,"cannot stat '%s': %s\n", path, strerror(errno)); |
| return -1; |
| } |
| |
| size = st.st_size; |
| } |
| |
| sbuf->id = ID_DATA; |
| for(;;) { |
| int ret; |
| |
| ret = adb_read(lfd, sbuf->data, SYNC_DATA_MAX); |
| if(!ret) |
| break; |
| |
| if(ret < 0) { |
| if(errno == EINTR) |
| continue; |
| fprintf(stderr,"cannot read '%s': %s\n", path, strerror(errno)); |
| break; |
| } |
| |
| sbuf->size = htoll(ret); |
| if(!WriteFdExactly(fd, sbuf, sizeof(unsigned) * 2 + ret)){ |
| err = -1; |
| break; |
| } |
| total_bytes += ret; |
| |
| if (show_progress) { |
| print_transfer_progress(total_bytes, size); |
| } |
| } |
| |
| adb_close(lfd); |
| return err; |
| } |
| |
| static int write_data_buffer(int fd, char* file_buffer, int size, syncsendbuf *sbuf, |
| int show_progress) |
| { |
| int err = 0; |
| int total = 0; |
| |
| sbuf->id = ID_DATA; |
| while (total < size) { |
| int count = size - total; |
| if (count > SYNC_DATA_MAX) { |
| count = SYNC_DATA_MAX; |
| } |
| |
| memcpy(sbuf->data, &file_buffer[total], count); |
| sbuf->size = htoll(count); |
| if(!WriteFdExactly(fd, sbuf, sizeof(unsigned) * 2 + count)){ |
| err = -1; |
| break; |
| } |
| total += count; |
| total_bytes += count; |
| |
| if (show_progress) { |
| print_transfer_progress(total, size); |
| } |
| } |
| |
| return err; |
| } |
| |
| #if defined(_WIN32) |
| extern int write_data_link(int fd, const char *path, syncsendbuf *sbuf) __attribute__((error("no symlinks on Windows"))); |
| #else |
| static int write_data_link(int fd, const char *path, syncsendbuf *sbuf) |
| { |
| int len, ret; |
| |
| len = readlink(path, sbuf->data, SYNC_DATA_MAX-1); |
| if(len < 0) { |
| fprintf(stderr, "error reading link '%s': %s\n", path, strerror(errno)); |
| return -1; |
| } |
| sbuf->data[len] = '\0'; |
| |
| sbuf->size = htoll(len + 1); |
| sbuf->id = ID_DATA; |
| |
| ret = !WriteFdExactly(fd, sbuf, sizeof(unsigned) * 2 + len + 1); |
| if(ret) |
| return -1; |
| |
| total_bytes += len + 1; |
| |
| return 0; |
| } |
| #endif |
| |
| static int sync_send(int fd, const char *lpath, const char *rpath, |
| unsigned mtime, mode_t mode, int show_progress) |
| { |
| syncmsg msg; |
| int len, r; |
| syncsendbuf *sbuf = &send_buffer; |
| char* file_buffer = NULL; |
| int size = 0; |
| char tmp[64]; |
| |
| len = strlen(rpath); |
| if(len > 1024) goto fail; |
| |
| snprintf(tmp, sizeof(tmp), ",%d", mode); |
| r = strlen(tmp); |
| |
| msg.req.id = ID_SEND; |
| msg.req.namelen = htoll(len + r); |
| |
| if(!WriteFdExactly(fd, &msg.req, sizeof(msg.req)) || |
| !WriteFdExactly(fd, rpath, len) || !WriteFdExactly(fd, tmp, r)) { |
| free(file_buffer); |
| goto fail; |
| } |
| |
| if (file_buffer) { |
| write_data_buffer(fd, file_buffer, size, sbuf, show_progress); |
| free(file_buffer); |
| } else if (S_ISREG(mode)) |
| write_data_file(fd, lpath, sbuf, show_progress); |
| else if (S_ISLNK(mode)) |
| write_data_link(fd, lpath, sbuf); |
| else |
| goto fail; |
| |
| msg.data.id = ID_DONE; |
| msg.data.size = htoll(mtime); |
| if(!WriteFdExactly(fd, &msg.data, sizeof(msg.data))) |
| goto fail; |
| |
| if(!ReadFdExactly(fd, &msg.status, sizeof(msg.status))) |
| return -1; |
| |
| if(msg.status.id != ID_OKAY) { |
| if(msg.status.id == ID_FAIL) { |
| len = ltohl(msg.status.msglen); |
| if(len > 256) len = 256; |
| if(!ReadFdExactly(fd, sbuf->data, len)) { |
| return -1; |
| } |
| sbuf->data[len] = 0; |
| } else |
| strcpy(sbuf->data, "unknown reason"); |
| |
| fprintf(stderr,"failed to copy '%s' to '%s': %s\n", lpath, rpath, sbuf->data); |
| return -1; |
| } |
| |
| return 0; |
| |
| fail: |
| fprintf(stderr,"protocol failure\n"); |
| adb_close(fd); |
| return -1; |
| } |
| |
| static int mkdirs(const char *name) |
| { |
| int ret; |
| char *x = (char *)name + 1; |
| |
| for(;;) { |
| x = adb_dirstart(x); |
| if(x == 0) return 0; |
| *x = 0; |
| ret = adb_mkdir(name, 0775); |
| *x = OS_PATH_SEPARATOR; |
| if((ret < 0) && (errno != EEXIST)) { |
| return ret; |
| } |
| x++; |
| } |
| return 0; |
| } |
| |
| static int sync_recv(int fd, const char* rpath, const char* lpath, int show_progress) { |
| syncmsg msg; |
| int len; |
| int lfd = -1; |
| char *buffer = send_buffer.data; |
| unsigned id; |
| unsigned long long size = 0; |
| |
| len = strlen(rpath); |
| if(len > 1024) return -1; |
| |
| if (show_progress) { |
| // Determine remote file size. |
| syncmsg stat_msg; |
| stat_msg.req.id = ID_STAT; |
| stat_msg.req.namelen = htoll(len); |
| |
| if (!WriteFdExactly(fd, &stat_msg.req, sizeof(stat_msg.req)) || |
| !WriteFdExactly(fd, rpath, len)) { |
| return -1; |
| } |
| |
| if (!ReadFdExactly(fd, &stat_msg.stat, sizeof(stat_msg.stat))) { |
| return -1; |
| } |
| |
| if (stat_msg.stat.id != ID_STAT) return -1; |
| |
| size = ltohl(stat_msg.stat.size); |
| } |
| |
| msg.req.id = ID_RECV; |
| msg.req.namelen = htoll(len); |
| if(!WriteFdExactly(fd, &msg.req, sizeof(msg.req)) || |
| !WriteFdExactly(fd, rpath, len)) { |
| return -1; |
| } |
| |
| if(!ReadFdExactly(fd, &msg.data, sizeof(msg.data))) { |
| return -1; |
| } |
| id = msg.data.id; |
| |
| if((id == ID_DATA) || (id == ID_DONE)) { |
| adb_unlink(lpath); |
| mkdirs(lpath); |
| lfd = adb_creat(lpath, 0644); |
| if(lfd < 0) { |
| fprintf(stderr,"cannot create '%s': %s\n", lpath, strerror(errno)); |
| return -1; |
| } |
| goto handle_data; |
| } else { |
| goto remote_error; |
| } |
| |
| for(;;) { |
| if(!ReadFdExactly(fd, &msg.data, sizeof(msg.data))) { |
| return -1; |
| } |
| id = msg.data.id; |
| |
| handle_data: |
| len = ltohl(msg.data.size); |
| if(id == ID_DONE) break; |
| if(id != ID_DATA) goto remote_error; |
| if(len > SYNC_DATA_MAX) { |
| fprintf(stderr,"data overrun\n"); |
| adb_close(lfd); |
| return -1; |
| } |
| |
| if(!ReadFdExactly(fd, buffer, len)) { |
| adb_close(lfd); |
| return -1; |
| } |
| |
| if(!WriteFdExactly(lfd, buffer, len)) { |
| fprintf(stderr,"cannot write '%s': %s\n", rpath, strerror(errno)); |
| adb_close(lfd); |
| return -1; |
| } |
| |
| total_bytes += len; |
| |
| if (show_progress) { |
| print_transfer_progress(total_bytes, size); |
| } |
| } |
| |
| adb_close(lfd); |
| return 0; |
| |
| remote_error: |
| adb_close(lfd); |
| adb_unlink(lpath); |
| |
| if(id == ID_FAIL) { |
| len = ltohl(msg.data.size); |
| if(len > 256) len = 256; |
| if(!ReadFdExactly(fd, buffer, len)) { |
| return -1; |
| } |
| buffer[len] = 0; |
| } else { |
| memcpy(buffer, &id, 4); |
| buffer[4] = 0; |
| } |
| fprintf(stderr,"failed to copy '%s' to '%s': %s\n", rpath, lpath, buffer); |
| return 0; |
| } |
| |
| /* --- */ |
| static void do_sync_ls_cb(unsigned mode, unsigned size, unsigned time, |
| const char *name, void *cookie) |
| { |
| printf("%08x %08x %08x %s\n", mode, size, time, name); |
| } |
| |
| int do_sync_ls(const char* path) { |
| std::string error; |
| int fd = adb_connect("sync:", &error); |
| if (fd < 0) { |
| fprintf(stderr,"error: %s\n", error.c_str()); |
| return 1; |
| } |
| |
| if(sync_ls(fd, path, do_sync_ls_cb, 0)) { |
| return 1; |
| } else { |
| sync_quit(fd); |
| return 0; |
| } |
| } |
| |
| struct copyinfo |
| { |
| copyinfo *next; |
| const char *src; |
| const char *dst; |
| unsigned int time; |
| unsigned int mode; |
| unsigned int size; |
| int flag; |
| }; |
| |
| static copyinfo* mkcopyinfo(const char* spath, const char* dpath, const char* name, int isdir) { |
| int slen = strlen(spath); |
| int dlen = strlen(dpath); |
| int nlen = strlen(name); |
| int ssize = slen + nlen + 2; |
| int dsize = dlen + nlen + 2; |
| |
| copyinfo *ci = reinterpret_cast<copyinfo*>(malloc(sizeof(copyinfo) + ssize + dsize)); |
| if(ci == 0) { |
| fprintf(stderr,"out of memory\n"); |
| abort(); |
| } |
| |
| ci->next = 0; |
| ci->time = 0; |
| ci->mode = 0; |
| ci->size = 0; |
| ci->flag = 0; |
| ci->src = (const char*)(ci + 1); |
| ci->dst = ci->src + ssize; |
| snprintf((char*) ci->src, ssize, isdir ? "%s%s/" : "%s%s", spath, name); |
| snprintf((char*) ci->dst, dsize, isdir ? "%s%s/" : "%s%s", dpath, name); |
| |
| return ci; |
| } |
| |
| |
| static int local_build_list(copyinfo **filelist, |
| const char *lpath, const char *rpath) |
| { |
| DIR *d; |
| struct dirent *de; |
| struct stat st; |
| copyinfo *dirlist = 0; |
| copyinfo *ci, *next; |
| |
| d = opendir(lpath); |
| if(d == 0) { |
| fprintf(stderr,"cannot open '%s': %s\n", lpath, strerror(errno)); |
| return -1; |
| } |
| |
| while((de = readdir(d))) { |
| char stat_path[PATH_MAX]; |
| char *name = de->d_name; |
| |
| if(name[0] == '.') { |
| if(name[1] == 0) continue; |
| if((name[1] == '.') && (name[2] == 0)) continue; |
| } |
| |
| /* |
| * We could use d_type if HAVE_DIRENT_D_TYPE is defined, but reiserfs |
| * always returns DT_UNKNOWN, so we just use stat() for all cases. |
| */ |
| if (strlen(lpath) + strlen(de->d_name) + 1 > sizeof(stat_path)) |
| continue; |
| strcpy(stat_path, lpath); |
| strcat(stat_path, de->d_name); |
| |
| if(!lstat(stat_path, &st)) { |
| if (S_ISDIR(st.st_mode)) { |
| ci = mkcopyinfo(lpath, rpath, name, 1); |
| ci->next = dirlist; |
| dirlist = ci; |
| } else { |
| ci = mkcopyinfo(lpath, rpath, name, 0); |
| if(lstat(ci->src, &st)) { |
| fprintf(stderr,"cannot stat '%s': %s\n", ci->src, strerror(errno)); |
| free(ci); |
| closedir(d); |
| return -1; |
| } |
| if(!S_ISREG(st.st_mode) && !S_ISLNK(st.st_mode)) { |
| fprintf(stderr, "skipping special file '%s'\n", ci->src); |
| free(ci); |
| } else { |
| ci->time = st.st_mtime; |
| ci->mode = st.st_mode; |
| ci->size = st.st_size; |
| ci->next = *filelist; |
| *filelist = ci; |
| } |
| } |
| } else { |
| fprintf(stderr, "cannot lstat '%s': %s\n",stat_path , strerror(errno)); |
| } |
| } |
| |
| closedir(d); |
| |
| for(ci = dirlist; ci != 0; ci = next) { |
| next = ci->next; |
| local_build_list(filelist, ci->src, ci->dst); |
| free(ci); |
| } |
| |
| return 0; |
| } |
| |
| |
| static int copy_local_dir_remote(int fd, const char *lpath, const char *rpath, int checktimestamps, int listonly) |
| { |
| copyinfo *filelist = 0; |
| copyinfo *ci, *next; |
| int pushed = 0; |
| int skipped = 0; |
| |
| if((lpath[0] == 0) || (rpath[0] == 0)) return -1; |
| if(lpath[strlen(lpath) - 1] != '/') { |
| int tmplen = strlen(lpath)+2; |
| char *tmp = reinterpret_cast<char*>(malloc(tmplen)); |
| if(tmp == 0) return -1; |
| snprintf(tmp, tmplen, "%s/",lpath); |
| lpath = tmp; |
| } |
| if(rpath[strlen(rpath) - 1] != '/') { |
| int tmplen = strlen(rpath)+2; |
| char *tmp = reinterpret_cast<char*>(malloc(tmplen)); |
| if(tmp == 0) return -1; |
| snprintf(tmp, tmplen, "%s/",rpath); |
| rpath = tmp; |
| } |
| |
| if(local_build_list(&filelist, lpath, rpath)) { |
| return -1; |
| } |
| |
| if(checktimestamps){ |
| for(ci = filelist; ci != 0; ci = ci->next) { |
| if(sync_start_readtime(fd, ci->dst)) { |
| return 1; |
| } |
| } |
| for(ci = filelist; ci != 0; ci = ci->next) { |
| unsigned int timestamp, mode, size; |
| if(sync_finish_readtime(fd, ×tamp, &mode, &size)) |
| return 1; |
| if(size == ci->size) { |
| /* for links, we cannot update the atime/mtime */ |
| if((S_ISREG(ci->mode & mode) && timestamp == ci->time) || |
| (S_ISLNK(ci->mode & mode) && timestamp >= ci->time)) |
| ci->flag = 1; |
| } |
| } |
| } |
| for(ci = filelist; ci != 0; ci = next) { |
| next = ci->next; |
| if(ci->flag == 0) { |
| fprintf(stderr,"%spush: %s -> %s\n", listonly ? "would " : "", ci->src, ci->dst); |
| if(!listonly && |
| sync_send(fd, ci->src, ci->dst, ci->time, ci->mode, |
| 0 /* no show progress */)) { |
| return 1; |
| } |
| pushed++; |
| } else { |
| skipped++; |
| } |
| free(ci); |
| } |
| |
| fprintf(stderr,"%d file%s pushed. %d file%s skipped.\n", |
| pushed, (pushed == 1) ? "" : "s", |
| skipped, (skipped == 1) ? "" : "s"); |
| |
| return 0; |
| } |
| |
| |
| int do_sync_push(const char *lpath, const char *rpath, int show_progress) |
| { |
| struct stat st; |
| unsigned mode; |
| |
| std::string error; |
| int fd = adb_connect("sync:", &error); |
| if (fd < 0) { |
| fprintf(stderr,"error: %s\n", error.c_str()); |
| return 1; |
| } |
| |
| if(stat(lpath, &st)) { |
| fprintf(stderr,"cannot stat '%s': %s\n", lpath, strerror(errno)); |
| sync_quit(fd); |
| return 1; |
| } |
| |
| if(S_ISDIR(st.st_mode)) { |
| BEGIN(); |
| if(copy_local_dir_remote(fd, lpath, rpath, 0, 0)) { |
| return 1; |
| } else { |
| END(); |
| sync_quit(fd); |
| } |
| } else { |
| if(sync_readmode(fd, rpath, &mode)) { |
| return 1; |
| } |
| if((mode != 0) && S_ISDIR(mode)) { |
| /* if we're copying a local file to a remote directory, |
| ** we *really* want to copy to remotedir + "/" + localfilename |
| */ |
| const char *name = adb_dirstop(lpath); |
| if(name == 0) { |
| name = lpath; |
| } else { |
| name++; |
| } |
| int tmplen = strlen(name) + strlen(rpath) + 2; |
| char *tmp = reinterpret_cast<char*>( |
| malloc(strlen(name) + strlen(rpath) + 2)); |
| if(tmp == 0) return 1; |
| snprintf(tmp, tmplen, "%s/%s", rpath, name); |
| rpath = tmp; |
| } |
| BEGIN(); |
| if(sync_send(fd, lpath, rpath, st.st_mtime, st.st_mode, show_progress)) { |
| return 1; |
| } else { |
| END(); |
| sync_quit(fd); |
| return 0; |
| } |
| } |
| |
| return 0; |
| } |
| |
| |
| struct sync_ls_build_list_cb_args { |
| copyinfo **filelist; |
| copyinfo **dirlist; |
| const char *rpath; |
| const char *lpath; |
| }; |
| |
| static void sync_ls_build_list_cb(unsigned mode, unsigned size, unsigned time, |
| const char* name, void* cookie) |
| { |
| sync_ls_build_list_cb_args *args = (sync_ls_build_list_cb_args *)cookie; |
| copyinfo *ci; |
| |
| if (S_ISDIR(mode)) { |
| copyinfo **dirlist = args->dirlist; |
| |
| /* Don't try recursing down "." or ".." */ |
| if (name[0] == '.') { |
| if (name[1] == '\0') return; |
| if ((name[1] == '.') && (name[2] == '\0')) return; |
| } |
| |
| ci = mkcopyinfo(args->rpath, args->lpath, name, 1); |
| ci->next = *dirlist; |
| *dirlist = ci; |
| } else if (S_ISREG(mode) || S_ISLNK(mode)) { |
| copyinfo **filelist = args->filelist; |
| |
| ci = mkcopyinfo(args->rpath, args->lpath, name, 0); |
| ci->time = time; |
| ci->mode = mode; |
| ci->size = size; |
| ci->next = *filelist; |
| *filelist = ci; |
| } else { |
| fprintf(stderr, "skipping special file '%s'\n", name); |
| } |
| } |
| |
| static int remote_build_list(int syncfd, copyinfo **filelist, |
| const char *rpath, const char *lpath) |
| { |
| copyinfo *dirlist = NULL; |
| sync_ls_build_list_cb_args args; |
| |
| args.filelist = filelist; |
| args.dirlist = &dirlist; |
| args.rpath = rpath; |
| args.lpath = lpath; |
| |
| /* Put the files/dirs in rpath on the lists. */ |
| if (sync_ls(syncfd, rpath, sync_ls_build_list_cb, (void *)&args)) { |
| return 1; |
| } |
| |
| /* Recurse into each directory we found. */ |
| while (dirlist != NULL) { |
| copyinfo *next = dirlist->next; |
| if (remote_build_list(syncfd, filelist, dirlist->src, dirlist->dst)) { |
| return 1; |
| } |
| free(dirlist); |
| dirlist = next; |
| } |
| |
| return 0; |
| } |
| |
| static int set_time_and_mode(const char *lpath, time_t time, unsigned int mode) |
| { |
| struct utimbuf times = { time, time }; |
| int r1 = utime(lpath, ×); |
| |
| /* use umask for permissions */ |
| mode_t mask=umask(0000); |
| umask(mask); |
| int r2 = chmod(lpath, mode & ~mask); |
| |
| return r1 ? : r2; |
| } |
| |
| /* Return a copy of the path string with / appended if needed */ |
| static char *add_slash_to_path(const char *path) |
| { |
| if (path[strlen(path) - 1] != '/') { |
| size_t len = strlen(path) + 2; |
| char *path_with_slash = reinterpret_cast<char*>(malloc(len)); |
| if (path_with_slash == NULL) |
| return NULL; |
| snprintf(path_with_slash, len, "%s/", path); |
| return path_with_slash; |
| } else { |
| return strdup(path); |
| } |
| } |
| |
| static int copy_remote_dir_local(int fd, const char *rpath, const char *lpath, |
| int copy_attrs) |
| { |
| copyinfo *filelist = 0; |
| copyinfo *ci, *next; |
| int pulled = 0; |
| int skipped = 0; |
| char *rpath_clean = NULL; |
| char *lpath_clean = NULL; |
| int ret = 0; |
| |
| if (rpath[0] == '\0' || lpath[0] == '\0') { |
| ret = -1; |
| goto finish; |
| } |
| |
| /* Make sure that both directory paths end in a slash. */ |
| rpath_clean = add_slash_to_path(rpath); |
| if (!rpath_clean) { |
| ret = -1; |
| goto finish; |
| } |
| lpath_clean = add_slash_to_path(lpath); |
| if (!lpath_clean) { |
| ret = -1; |
| goto finish; |
| } |
| |
| /* Recursively build the list of files to copy. */ |
| fprintf(stderr, "pull: building file list...\n"); |
| if (remote_build_list(fd, &filelist, rpath_clean, lpath_clean)) { |
| ret = -1; |
| goto finish; |
| } |
| |
| for (ci = filelist; ci != 0; ci = next) { |
| next = ci->next; |
| if (ci->flag == 0) { |
| fprintf(stderr, "pull: %s -> %s\n", ci->src, ci->dst); |
| if (sync_recv(fd, ci->src, ci->dst, 0 /* no show progress */)) { |
| ret = -1; |
| goto finish; |
| } |
| |
| if (copy_attrs && set_time_and_mode(ci->dst, ci->time, ci->mode)) { |
| ret = -1; |
| goto finish; |
| } |
| pulled++; |
| } else { |
| skipped++; |
| } |
| free(ci); |
| } |
| |
| fprintf(stderr, "%d file%s pulled. %d file%s skipped.\n", |
| pulled, (pulled == 1) ? "" : "s", |
| skipped, (skipped == 1) ? "" : "s"); |
| |
| finish: |
| free(lpath_clean); |
| free(rpath_clean); |
| return ret; |
| } |
| |
| int do_sync_pull(const char *rpath, const char *lpath, int show_progress, int copy_attrs) |
| { |
| unsigned mode, time; |
| struct stat st; |
| |
| std::string error; |
| int fd = adb_connect("sync:", &error); |
| if (fd < 0) { |
| fprintf(stderr,"error: %s\n", error.c_str()); |
| return 1; |
| } |
| |
| if(sync_readtime(fd, rpath, &time, &mode)) { |
| return 1; |
| } |
| if(mode == 0) { |
| fprintf(stderr,"remote object '%s' does not exist\n", rpath); |
| return 1; |
| } |
| |
| if(S_ISREG(mode) || S_ISLNK(mode) || S_ISCHR(mode) || S_ISBLK(mode)) { |
| if(stat(lpath, &st) == 0) { |
| if(S_ISDIR(st.st_mode)) { |
| /* if we're copying a remote file to a local directory, |
| ** we *really* want to copy to localdir + "/" + remotefilename |
| */ |
| const char *name = adb_dirstop(rpath); |
| if(name == 0) { |
| name = rpath; |
| } else { |
| name++; |
| } |
| int tmplen = strlen(name) + strlen(lpath) + 2; |
| char *tmp = reinterpret_cast<char*>(malloc(tmplen)); |
| if(tmp == 0) return 1; |
| snprintf(tmp, tmplen, "%s/%s", lpath, name); |
| lpath = tmp; |
| } |
| } |
| BEGIN(); |
| if (sync_recv(fd, rpath, lpath, show_progress)) { |
| return 1; |
| } else { |
| if (copy_attrs && set_time_and_mode(lpath, time, mode)) |
| return 1; |
| END(); |
| sync_quit(fd); |
| return 0; |
| } |
| } else if(S_ISDIR(mode)) { |
| BEGIN(); |
| if (copy_remote_dir_local(fd, rpath, lpath, copy_attrs)) { |
| return 1; |
| } else { |
| END(); |
| sync_quit(fd); |
| return 0; |
| } |
| } else { |
| fprintf(stderr,"remote object '%s' not a file or directory\n", rpath); |
| return 1; |
| } |
| } |
| |
| int do_sync_sync(const std::string& lpath, const std::string& rpath, bool list_only) |
| { |
| fprintf(stderr, "syncing %s...\n", rpath.c_str()); |
| |
| std::string error; |
| int fd = adb_connect("sync:", &error); |
| if (fd < 0) { |
| fprintf(stderr, "error: %s\n", error.c_str()); |
| return 1; |
| } |
| |
| BEGIN(); |
| if (copy_local_dir_remote(fd, lpath.c_str(), rpath.c_str(), 1, list_only)) { |
| return 1; |
| } else { |
| END(); |
| sync_quit(fd); |
| return 0; |
| } |
| } |