blob: cf446e2eded46bc38b1a42c31de62c6071412e12 [file] [log] [blame]
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001/*
2 * Copyright (C) 2007 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
Dan Albert76649012015-02-24 15:51:19 -080017#include <dirent.h>
18#include <errno.h>
19#include <limits.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080020#include <stdio.h>
21#include <stdlib.h>
22#include <string.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080023#include <sys/stat.h>
24#include <sys/time.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080025#include <sys/types.h>
Dan Albert76649012015-02-24 15:51:19 -080026#include <time.h>
Greg Hackmann7a5e2bd2014-05-06 08:48:18 -070027#include <utime.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080028
29#include "sysdeps.h"
Dan Albert76649012015-02-24 15:51:19 -080030
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080031#include "adb.h"
32#include "adb_client.h"
33#include "file_sync_service.h"
Dan Albert76649012015-02-24 15:51:19 -080034#include "transport.h"
35#include "zipfile/zipfile.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080036
Jeff Smithd9a14302013-06-15 15:32:05 -050037static unsigned long long total_bytes;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080038static long long start_time;
39
40static long long NOW()
41{
42 struct timeval tv;
43 gettimeofday(&tv, 0);
44 return ((long long) tv.tv_usec) +
45 1000000LL * ((long long) tv.tv_sec);
46}
47
48static void BEGIN()
49{
50 total_bytes = 0;
51 start_time = NOW();
52}
53
54static void END()
55{
56 long long t = NOW() - start_time;
57 if(total_bytes == 0) return;
58
59 if (t == 0) /* prevent division by 0 :-) */
60 t = 1000000;
61
Mike Lockwoodee878752010-12-14 23:07:32 -080062 fprintf(stderr,"%lld KB/s (%lld bytes in %lld.%03llds)\n",
Jeff Smithd9a14302013-06-15 15:32:05 -050063 ((total_bytes * 1000000LL) / t) / 1024LL,
64 total_bytes, (t / 1000000LL), (t % 1000000LL) / 1000LL);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080065}
66
Mark Lindner76f2a932014-03-11 17:55:59 -070067static const char* transfer_progress_format = "\rTransferring: %llu/%llu (%d%%)";
68
69static void print_transfer_progress(unsigned long long bytes_current,
70 unsigned long long bytes_total) {
71 if (bytes_total == 0) return;
72
73 fprintf(stderr, transfer_progress_format, bytes_current, bytes_total,
74 (int) (bytes_current * 100 / bytes_total));
75
76 if (bytes_current == bytes_total) {
77 fputc('\n', stderr);
78 }
79
80 fflush(stderr);
81}
82
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080083void sync_quit(int fd)
84{
85 syncmsg msg;
86
87 msg.req.id = ID_QUIT;
88 msg.req.namelen = 0;
89
90 writex(fd, &msg.req, sizeof(msg.req));
91}
92
93typedef void (*sync_ls_cb)(unsigned mode, unsigned size, unsigned time, const char *name, void *cookie);
94
95int sync_ls(int fd, const char *path, sync_ls_cb func, void *cookie)
96{
97 syncmsg msg;
98 char buf[257];
99 int len;
100
101 len = strlen(path);
102 if(len > 1024) goto fail;
103
104 msg.req.id = ID_LIST;
105 msg.req.namelen = htoll(len);
106
107 if(writex(fd, &msg.req, sizeof(msg.req)) ||
108 writex(fd, path, len)) {
109 goto fail;
110 }
111
112 for(;;) {
113 if(readx(fd, &msg.dent, sizeof(msg.dent))) break;
114 if(msg.dent.id == ID_DONE) return 0;
115 if(msg.dent.id != ID_DENT) break;
116
117 len = ltohl(msg.dent.namelen);
118 if(len > 256) break;
119
120 if(readx(fd, buf, len)) break;
121 buf[len] = 0;
122
123 func(ltohl(msg.dent.mode),
124 ltohl(msg.dent.size),
125 ltohl(msg.dent.time),
126 buf, cookie);
127 }
128
129fail:
130 adb_close(fd);
131 return -1;
132}
133
134typedef struct syncsendbuf syncsendbuf;
135
136struct syncsendbuf {
137 unsigned id;
138 unsigned size;
139 char data[SYNC_DATA_MAX];
140};
141
142static syncsendbuf send_buffer;
143
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700144int sync_readtime(int fd, const char *path, unsigned int *timestamp,
145 unsigned int *mode)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800146{
147 syncmsg msg;
148 int len = strlen(path);
149
150 msg.req.id = ID_STAT;
151 msg.req.namelen = htoll(len);
152
153 if(writex(fd, &msg.req, sizeof(msg.req)) ||
154 writex(fd, path, len)) {
155 return -1;
156 }
157
158 if(readx(fd, &msg.stat, sizeof(msg.stat))) {
159 return -1;
160 }
161
162 if(msg.stat.id != ID_STAT) {
163 return -1;
164 }
165
166 *timestamp = ltohl(msg.stat.time);
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700167 *mode = ltohl(msg.stat.mode);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800168 return 0;
169}
170
171static int sync_start_readtime(int fd, const char *path)
172{
173 syncmsg msg;
174 int len = strlen(path);
175
176 msg.req.id = ID_STAT;
177 msg.req.namelen = htoll(len);
178
179 if(writex(fd, &msg.req, sizeof(msg.req)) ||
180 writex(fd, path, len)) {
181 return -1;
182 }
183
184 return 0;
185}
186
187static int sync_finish_readtime(int fd, unsigned int *timestamp,
David 'Digit' Turnerf6330a22009-05-18 17:36:28 +0200188 unsigned int *mode, unsigned int *size)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800189{
190 syncmsg msg;
191
192 if(readx(fd, &msg.stat, sizeof(msg.stat)))
193 return -1;
194
195 if(msg.stat.id != ID_STAT)
196 return -1;
197
198 *timestamp = ltohl(msg.stat.time);
199 *mode = ltohl(msg.stat.mode);
200 *size = ltohl(msg.stat.size);
201
202 return 0;
203}
204
205int sync_readmode(int fd, const char *path, unsigned *mode)
206{
207 syncmsg msg;
208 int len = strlen(path);
209
210 msg.req.id = ID_STAT;
211 msg.req.namelen = htoll(len);
212
213 if(writex(fd, &msg.req, sizeof(msg.req)) ||
214 writex(fd, path, len)) {
215 return -1;
216 }
217
218 if(readx(fd, &msg.stat, sizeof(msg.stat))) {
219 return -1;
220 }
221
222 if(msg.stat.id != ID_STAT) {
223 return -1;
224 }
225
226 *mode = ltohl(msg.stat.mode);
227 return 0;
228}
229
Mark Lindner76f2a932014-03-11 17:55:59 -0700230static int write_data_file(int fd, const char *path, syncsendbuf *sbuf, int show_progress)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800231{
232 int lfd, err = 0;
Mark Lindner76f2a932014-03-11 17:55:59 -0700233 unsigned long long size = 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800234
235 lfd = adb_open(path, O_RDONLY);
236 if(lfd < 0) {
237 fprintf(stderr,"cannot open '%s': %s\n", path, strerror(errno));
238 return -1;
239 }
240
Mark Lindner76f2a932014-03-11 17:55:59 -0700241 if (show_progress) {
242 // Determine local file size.
243 struct stat st;
eric.yana8592192015-02-03 22:16:29 +0800244 if (stat(path, &st)) {
Mark Lindner76f2a932014-03-11 17:55:59 -0700245 fprintf(stderr,"cannot stat '%s': %s\n", path, strerror(errno));
246 return -1;
247 }
248
249 size = st.st_size;
250 }
251
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800252 sbuf->id = ID_DATA;
253 for(;;) {
254 int ret;
255
256 ret = adb_read(lfd, sbuf->data, SYNC_DATA_MAX);
257 if(!ret)
258 break;
259
260 if(ret < 0) {
261 if(errno == EINTR)
262 continue;
263 fprintf(stderr,"cannot read '%s': %s\n", path, strerror(errno));
264 break;
265 }
266
267 sbuf->size = htoll(ret);
268 if(writex(fd, sbuf, sizeof(unsigned) * 2 + ret)){
269 err = -1;
270 break;
271 }
272 total_bytes += ret;
Mark Lindner76f2a932014-03-11 17:55:59 -0700273
274 if (show_progress) {
275 print_transfer_progress(total_bytes, size);
276 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800277 }
278
279 adb_close(lfd);
280 return err;
281}
282
Mark Lindner76f2a932014-03-11 17:55:59 -0700283static int write_data_buffer(int fd, char* file_buffer, int size, syncsendbuf *sbuf,
284 int show_progress)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800285{
286 int err = 0;
287 int total = 0;
288
289 sbuf->id = ID_DATA;
290 while (total < size) {
291 int count = size - total;
292 if (count > SYNC_DATA_MAX) {
293 count = SYNC_DATA_MAX;
294 }
295
296 memcpy(sbuf->data, &file_buffer[total], count);
297 sbuf->size = htoll(count);
298 if(writex(fd, sbuf, sizeof(unsigned) * 2 + count)){
299 err = -1;
300 break;
301 }
302 total += count;
303 total_bytes += count;
Mark Lindner76f2a932014-03-11 17:55:59 -0700304
305 if (show_progress) {
306 print_transfer_progress(total, size);
307 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800308 }
309
310 return err;
311}
312
Elliott Hughes0a049b12015-01-12 14:26:36 -0800313#if defined(_WIN32)
314extern int write_data_link(int fd, const char *path, syncsendbuf *sbuf) __attribute__((error("no symlinks on Windows")));
315#else
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800316static int write_data_link(int fd, const char *path, syncsendbuf *sbuf)
317{
318 int len, ret;
319
320 len = readlink(path, sbuf->data, SYNC_DATA_MAX-1);
321 if(len < 0) {
322 fprintf(stderr, "error reading link '%s': %s\n", path, strerror(errno));
323 return -1;
324 }
325 sbuf->data[len] = '\0';
326
327 sbuf->size = htoll(len + 1);
328 sbuf->id = ID_DATA;
329
330 ret = writex(fd, sbuf, sizeof(unsigned) * 2 + len + 1);
331 if(ret)
332 return -1;
333
334 total_bytes += len + 1;
335
336 return 0;
337}
338#endif
339
340static int sync_send(int fd, const char *lpath, const char *rpath,
Jeff Sharkey960df972014-06-09 17:30:57 -0700341 unsigned mtime, mode_t mode, int show_progress)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800342{
343 syncmsg msg;
344 int len, r;
345 syncsendbuf *sbuf = &send_buffer;
346 char* file_buffer = NULL;
347 int size = 0;
348 char tmp[64];
349
350 len = strlen(rpath);
351 if(len > 1024) goto fail;
352
353 snprintf(tmp, sizeof(tmp), ",%d", mode);
354 r = strlen(tmp);
355
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800356 msg.req.id = ID_SEND;
357 msg.req.namelen = htoll(len + r);
358
359 if(writex(fd, &msg.req, sizeof(msg.req)) ||
360 writex(fd, rpath, len) || writex(fd, tmp, r)) {
361 free(file_buffer);
362 goto fail;
363 }
364
365 if (file_buffer) {
Mark Lindner76f2a932014-03-11 17:55:59 -0700366 write_data_buffer(fd, file_buffer, size, sbuf, show_progress);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800367 free(file_buffer);
368 } else if (S_ISREG(mode))
Mark Lindner76f2a932014-03-11 17:55:59 -0700369 write_data_file(fd, lpath, sbuf, show_progress);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800370 else if (S_ISLNK(mode))
371 write_data_link(fd, lpath, sbuf);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800372 else
373 goto fail;
374
375 msg.data.id = ID_DONE;
376 msg.data.size = htoll(mtime);
377 if(writex(fd, &msg.data, sizeof(msg.data)))
378 goto fail;
379
380 if(readx(fd, &msg.status, sizeof(msg.status)))
381 return -1;
382
383 if(msg.status.id != ID_OKAY) {
384 if(msg.status.id == ID_FAIL) {
385 len = ltohl(msg.status.msglen);
386 if(len > 256) len = 256;
387 if(readx(fd, sbuf->data, len)) {
388 return -1;
389 }
390 sbuf->data[len] = 0;
391 } else
392 strcpy(sbuf->data, "unknown reason");
393
394 fprintf(stderr,"failed to copy '%s' to '%s': %s\n", lpath, rpath, sbuf->data);
395 return -1;
396 }
397
398 return 0;
399
400fail:
401 fprintf(stderr,"protocol failure\n");
402 adb_close(fd);
403 return -1;
404}
405
Mark Salyzyn60299df2014-04-30 09:10:31 -0700406static int mkdirs(const char *name)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800407{
408 int ret;
Mark Salyzyn60299df2014-04-30 09:10:31 -0700409 char *x = (char *)name + 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800410
411 for(;;) {
412 x = adb_dirstart(x);
413 if(x == 0) return 0;
414 *x = 0;
415 ret = adb_mkdir(name, 0775);
416 *x = OS_PATH_SEPARATOR;
417 if((ret < 0) && (errno != EEXIST)) {
418 return ret;
419 }
420 x++;
421 }
422 return 0;
423}
424
Mark Lindner76f2a932014-03-11 17:55:59 -0700425int sync_recv(int fd, const char *rpath, const char *lpath, int show_progress)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800426{
427 syncmsg msg;
428 int len;
429 int lfd = -1;
430 char *buffer = send_buffer.data;
431 unsigned id;
Mark Lindner76f2a932014-03-11 17:55:59 -0700432 unsigned long long size = 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800433
434 len = strlen(rpath);
435 if(len > 1024) return -1;
436
Mark Lindner76f2a932014-03-11 17:55:59 -0700437 if (show_progress) {
438 // Determine remote file size.
439 syncmsg stat_msg;
440 stat_msg.req.id = ID_STAT;
441 stat_msg.req.namelen = htoll(len);
442
443 if (writex(fd, &stat_msg.req, sizeof(stat_msg.req)) ||
444 writex(fd, rpath, len)) {
445 return -1;
446 }
447
448 if (readx(fd, &stat_msg.stat, sizeof(stat_msg.stat))) {
449 return -1;
450 }
451
452 if (stat_msg.stat.id != ID_STAT) return -1;
453
454 size = ltohl(stat_msg.stat.size);
455 }
456
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800457 msg.req.id = ID_RECV;
458 msg.req.namelen = htoll(len);
459 if(writex(fd, &msg.req, sizeof(msg.req)) ||
460 writex(fd, rpath, len)) {
461 return -1;
462 }
463
464 if(readx(fd, &msg.data, sizeof(msg.data))) {
465 return -1;
466 }
467 id = msg.data.id;
468
469 if((id == ID_DATA) || (id == ID_DONE)) {
470 adb_unlink(lpath);
Mark Salyzyn60299df2014-04-30 09:10:31 -0700471 mkdirs(lpath);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800472 lfd = adb_creat(lpath, 0644);
473 if(lfd < 0) {
474 fprintf(stderr,"cannot create '%s': %s\n", lpath, strerror(errno));
475 return -1;
476 }
477 goto handle_data;
478 } else {
479 goto remote_error;
480 }
481
482 for(;;) {
483 if(readx(fd, &msg.data, sizeof(msg.data))) {
484 return -1;
485 }
486 id = msg.data.id;
487
488 handle_data:
489 len = ltohl(msg.data.size);
490 if(id == ID_DONE) break;
491 if(id != ID_DATA) goto remote_error;
492 if(len > SYNC_DATA_MAX) {
493 fprintf(stderr,"data overrun\n");
494 adb_close(lfd);
495 return -1;
496 }
497
498 if(readx(fd, buffer, len)) {
499 adb_close(lfd);
500 return -1;
501 }
502
503 if(writex(lfd, buffer, len)) {
504 fprintf(stderr,"cannot write '%s': %s\n", rpath, strerror(errno));
505 adb_close(lfd);
506 return -1;
507 }
508
509 total_bytes += len;
Mark Lindner76f2a932014-03-11 17:55:59 -0700510
511 if (show_progress) {
512 print_transfer_progress(total_bytes, size);
513 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800514 }
515
516 adb_close(lfd);
517 return 0;
518
519remote_error:
520 adb_close(lfd);
521 adb_unlink(lpath);
522
523 if(id == ID_FAIL) {
524 len = ltohl(msg.data.size);
525 if(len > 256) len = 256;
526 if(readx(fd, buffer, len)) {
527 return -1;
528 }
529 buffer[len] = 0;
530 } else {
531 memcpy(buffer, &id, 4);
532 buffer[4] = 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800533 }
534 fprintf(stderr,"failed to copy '%s' to '%s': %s\n", rpath, lpath, buffer);
535 return 0;
536}
537
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800538/* --- */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800539static void do_sync_ls_cb(unsigned mode, unsigned size, unsigned time,
540 const char *name, void *cookie)
541{
542 printf("%08x %08x %08x %s\n", mode, size, time, name);
543}
544
545int do_sync_ls(const char *path)
546{
547 int fd = adb_connect("sync:");
548 if(fd < 0) {
549 fprintf(stderr,"error: %s\n", adb_error());
550 return 1;
551 }
552
553 if(sync_ls(fd, path, do_sync_ls_cb, 0)) {
554 return 1;
555 } else {
556 sync_quit(fd);
557 return 0;
558 }
559}
560
561typedef struct copyinfo copyinfo;
562
563struct copyinfo
564{
565 copyinfo *next;
566 const char *src;
567 const char *dst;
568 unsigned int time;
569 unsigned int mode;
570 unsigned int size;
571 int flag;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800572};
573
574copyinfo *mkcopyinfo(const char *spath, const char *dpath,
575 const char *name, int isdir)
576{
577 int slen = strlen(spath);
578 int dlen = strlen(dpath);
579 int nlen = strlen(name);
580 int ssize = slen + nlen + 2;
581 int dsize = dlen + nlen + 2;
582
583 copyinfo *ci = malloc(sizeof(copyinfo) + ssize + dsize);
584 if(ci == 0) {
585 fprintf(stderr,"out of memory\n");
586 abort();
587 }
588
589 ci->next = 0;
590 ci->time = 0;
591 ci->mode = 0;
592 ci->size = 0;
593 ci->flag = 0;
594 ci->src = (const char*)(ci + 1);
595 ci->dst = ci->src + ssize;
596 snprintf((char*) ci->src, ssize, isdir ? "%s%s/" : "%s%s", spath, name);
597 snprintf((char*) ci->dst, dsize, isdir ? "%s%s/" : "%s%s", dpath, name);
598
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800599 return ci;
600}
601
602
603static int local_build_list(copyinfo **filelist,
604 const char *lpath, const char *rpath)
605{
606 DIR *d;
607 struct dirent *de;
608 struct stat st;
609 copyinfo *dirlist = 0;
610 copyinfo *ci, *next;
611
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800612 d = opendir(lpath);
613 if(d == 0) {
614 fprintf(stderr,"cannot open '%s': %s\n", lpath, strerror(errno));
615 return -1;
616 }
617
618 while((de = readdir(d))) {
619 char stat_path[PATH_MAX];
620 char *name = de->d_name;
621
622 if(name[0] == '.') {
623 if(name[1] == 0) continue;
624 if((name[1] == '.') && (name[2] == 0)) continue;
625 }
626
627 /*
628 * We could use d_type if HAVE_DIRENT_D_TYPE is defined, but reiserfs
629 * always returns DT_UNKNOWN, so we just use stat() for all cases.
630 */
631 if (strlen(lpath) + strlen(de->d_name) + 1 > sizeof(stat_path))
632 continue;
633 strcpy(stat_path, lpath);
634 strcat(stat_path, de->d_name);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800635
Daniel Rosenberg686bce62014-06-30 20:29:40 -0700636 if(!lstat(stat_path, &st)) {
637 if (S_ISDIR(st.st_mode)) {
638 ci = mkcopyinfo(lpath, rpath, name, 1);
639 ci->next = dirlist;
640 dirlist = ci;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800641 } else {
Daniel Rosenberg686bce62014-06-30 20:29:40 -0700642 ci = mkcopyinfo(lpath, rpath, name, 0);
643 if(lstat(ci->src, &st)) {
644 fprintf(stderr,"cannot stat '%s': %s\n", ci->src, strerror(errno));
645 free(ci);
646 closedir(d);
647 return -1;
648 }
649 if(!S_ISREG(st.st_mode) && !S_ISLNK(st.st_mode)) {
650 fprintf(stderr, "skipping special file '%s'\n", ci->src);
651 free(ci);
652 } else {
653 ci->time = st.st_mtime;
654 ci->mode = st.st_mode;
655 ci->size = st.st_size;
656 ci->next = *filelist;
657 *filelist = ci;
658 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800659 }
Daniel Rosenberg686bce62014-06-30 20:29:40 -0700660 } else {
661 fprintf(stderr, "cannot lstat '%s': %s\n",stat_path , strerror(errno));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800662 }
663 }
664
665 closedir(d);
666
667 for(ci = dirlist; ci != 0; ci = next) {
668 next = ci->next;
669 local_build_list(filelist, ci->src, ci->dst);
670 free(ci);
671 }
672
673 return 0;
674}
675
676
Anthony Newnam705c9442010-02-22 08:36:49 -0600677static int copy_local_dir_remote(int fd, const char *lpath, const char *rpath, int checktimestamps, int listonly)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800678{
679 copyinfo *filelist = 0;
680 copyinfo *ci, *next;
681 int pushed = 0;
682 int skipped = 0;
683
684 if((lpath[0] == 0) || (rpath[0] == 0)) return -1;
685 if(lpath[strlen(lpath) - 1] != '/') {
686 int tmplen = strlen(lpath)+2;
687 char *tmp = malloc(tmplen);
688 if(tmp == 0) return -1;
689 snprintf(tmp, tmplen, "%s/",lpath);
690 lpath = tmp;
691 }
692 if(rpath[strlen(rpath) - 1] != '/') {
693 int tmplen = strlen(rpath)+2;
694 char *tmp = malloc(tmplen);
695 if(tmp == 0) return -1;
696 snprintf(tmp, tmplen, "%s/",rpath);
697 rpath = tmp;
698 }
699
700 if(local_build_list(&filelist, lpath, rpath)) {
701 return -1;
702 }
703
704 if(checktimestamps){
705 for(ci = filelist; ci != 0; ci = ci->next) {
706 if(sync_start_readtime(fd, ci->dst)) {
707 return 1;
708 }
709 }
710 for(ci = filelist; ci != 0; ci = ci->next) {
711 unsigned int timestamp, mode, size;
712 if(sync_finish_readtime(fd, &timestamp, &mode, &size))
713 return 1;
714 if(size == ci->size) {
715 /* for links, we cannot update the atime/mtime */
716 if((S_ISREG(ci->mode & mode) && timestamp == ci->time) ||
717 (S_ISLNK(ci->mode & mode) && timestamp >= ci->time))
718 ci->flag = 1;
719 }
720 }
721 }
722 for(ci = filelist; ci != 0; ci = next) {
723 next = ci->next;
724 if(ci->flag == 0) {
Anthony Newnam705c9442010-02-22 08:36:49 -0600725 fprintf(stderr,"%spush: %s -> %s\n", listonly ? "would " : "", ci->src, ci->dst);
726 if(!listonly &&
Jeff Sharkey960df972014-06-09 17:30:57 -0700727 sync_send(fd, ci->src, ci->dst, ci->time, ci->mode,
Mark Lindner76f2a932014-03-11 17:55:59 -0700728 0 /* no show progress */)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800729 return 1;
730 }
731 pushed++;
732 } else {
733 skipped++;
734 }
735 free(ci);
736 }
737
738 fprintf(stderr,"%d file%s pushed. %d file%s skipped.\n",
739 pushed, (pushed == 1) ? "" : "s",
740 skipped, (skipped == 1) ? "" : "s");
741
742 return 0;
743}
744
745
Jeff Sharkey960df972014-06-09 17:30:57 -0700746int do_sync_push(const char *lpath, const char *rpath, int show_progress)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800747{
748 struct stat st;
749 unsigned mode;
750 int fd;
751
752 fd = adb_connect("sync:");
753 if(fd < 0) {
754 fprintf(stderr,"error: %s\n", adb_error());
755 return 1;
756 }
757
758 if(stat(lpath, &st)) {
759 fprintf(stderr,"cannot stat '%s': %s\n", lpath, strerror(errno));
760 sync_quit(fd);
761 return 1;
762 }
763
764 if(S_ISDIR(st.st_mode)) {
765 BEGIN();
Anthony Newnam705c9442010-02-22 08:36:49 -0600766 if(copy_local_dir_remote(fd, lpath, rpath, 0, 0)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800767 return 1;
768 } else {
769 END();
770 sync_quit(fd);
771 }
772 } else {
773 if(sync_readmode(fd, rpath, &mode)) {
774 return 1;
775 }
776 if((mode != 0) && S_ISDIR(mode)) {
777 /* if we're copying a local file to a remote directory,
778 ** we *really* want to copy to remotedir + "/" + localfilename
779 */
780 const char *name = adb_dirstop(lpath);
781 if(name == 0) {
782 name = lpath;
783 } else {
784 name++;
785 }
786 int tmplen = strlen(name) + strlen(rpath) + 2;
787 char *tmp = malloc(strlen(name) + strlen(rpath) + 2);
788 if(tmp == 0) return 1;
789 snprintf(tmp, tmplen, "%s/%s", rpath, name);
790 rpath = tmp;
791 }
792 BEGIN();
Jeff Sharkey960df972014-06-09 17:30:57 -0700793 if(sync_send(fd, lpath, rpath, st.st_mtime, st.st_mode, show_progress)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800794 return 1;
795 } else {
796 END();
797 sync_quit(fd);
798 return 0;
799 }
800 }
801
802 return 0;
803}
804
805
806typedef struct {
807 copyinfo **filelist;
808 copyinfo **dirlist;
809 const char *rpath;
810 const char *lpath;
811} sync_ls_build_list_cb_args;
812
813void
814sync_ls_build_list_cb(unsigned mode, unsigned size, unsigned time,
815 const char *name, void *cookie)
816{
817 sync_ls_build_list_cb_args *args = (sync_ls_build_list_cb_args *)cookie;
818 copyinfo *ci;
819
820 if (S_ISDIR(mode)) {
821 copyinfo **dirlist = args->dirlist;
822
823 /* Don't try recursing down "." or ".." */
824 if (name[0] == '.') {
825 if (name[1] == '\0') return;
826 if ((name[1] == '.') && (name[2] == '\0')) return;
827 }
828
829 ci = mkcopyinfo(args->rpath, args->lpath, name, 1);
830 ci->next = *dirlist;
831 *dirlist = ci;
832 } else if (S_ISREG(mode) || S_ISLNK(mode)) {
833 copyinfo **filelist = args->filelist;
834
835 ci = mkcopyinfo(args->rpath, args->lpath, name, 0);
836 ci->time = time;
837 ci->mode = mode;
838 ci->size = size;
839 ci->next = *filelist;
840 *filelist = ci;
841 } else {
842 fprintf(stderr, "skipping special file '%s'\n", name);
843 }
844}
845
846static int remote_build_list(int syncfd, copyinfo **filelist,
847 const char *rpath, const char *lpath)
848{
849 copyinfo *dirlist = NULL;
850 sync_ls_build_list_cb_args args;
851
852 args.filelist = filelist;
853 args.dirlist = &dirlist;
854 args.rpath = rpath;
855 args.lpath = lpath;
856
857 /* Put the files/dirs in rpath on the lists. */
858 if (sync_ls(syncfd, rpath, sync_ls_build_list_cb, (void *)&args)) {
859 return 1;
860 }
861
862 /* Recurse into each directory we found. */
863 while (dirlist != NULL) {
864 copyinfo *next = dirlist->next;
865 if (remote_build_list(syncfd, filelist, dirlist->src, dirlist->dst)) {
866 return 1;
867 }
868 free(dirlist);
869 dirlist = next;
870 }
871
872 return 0;
873}
874
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700875static int set_time_and_mode(const char *lpath, unsigned int time, unsigned int mode)
876{
Greg Hackmann7a5e2bd2014-05-06 08:48:18 -0700877 struct utimbuf times = { time, time };
878 int r1 = utime(lpath, &times);
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700879
880 /* use umask for permissions */
881 mode_t mask=umask(0000);
882 umask(mask);
883 int r2 = chmod(lpath, mode & ~mask);
884
885 return r1 ? : r2;
886}
887
Riley Andrews4d04d242014-12-12 13:12:36 -0800888/* Return a copy of the path string with / appended if needed */
889static char *add_slash_to_path(const char *path)
890{
891 if (path[strlen(path) - 1] != '/') {
892 size_t len = strlen(path) + 2;
893 char *path_with_slash = malloc(len);
894 if (path_with_slash == NULL)
895 return NULL;
896 snprintf(path_with_slash, len, "%s/", path);
897 return path_with_slash;
898 } else {
899 return strdup(path);
900 }
901}
902
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800903static int copy_remote_dir_local(int fd, const char *rpath, const char *lpath,
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700904 int copy_attrs)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800905{
906 copyinfo *filelist = 0;
907 copyinfo *ci, *next;
908 int pulled = 0;
909 int skipped = 0;
Riley Andrews4d04d242014-12-12 13:12:36 -0800910 char *rpath_clean = NULL;
911 char *lpath_clean = NULL;
912 int ret = 0;
913
914 if (rpath[0] == '\0' || lpath[0] == '\0') {
915 ret = -1;
916 goto finish;
917 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800918
919 /* Make sure that both directory paths end in a slash. */
Riley Andrews4d04d242014-12-12 13:12:36 -0800920 rpath_clean = add_slash_to_path(rpath);
921 if (!rpath_clean) {
922 ret = -1;
923 goto finish;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800924 }
Riley Andrews4d04d242014-12-12 13:12:36 -0800925 lpath_clean = add_slash_to_path(lpath);
926 if (!lpath_clean) {
927 ret = -1;
928 goto finish;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800929 }
930
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800931 /* Recursively build the list of files to copy. */
Riley Andrews4d04d242014-12-12 13:12:36 -0800932 fprintf(stderr, "pull: building file list...\n");
933 if (remote_build_list(fd, &filelist, rpath_clean, lpath_clean)) {
934 ret = -1;
935 goto finish;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800936 }
937
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800938 for (ci = filelist; ci != 0; ci = next) {
939 next = ci->next;
940 if (ci->flag == 0) {
941 fprintf(stderr, "pull: %s -> %s\n", ci->src, ci->dst);
Mark Lindner76f2a932014-03-11 17:55:59 -0700942 if (sync_recv(fd, ci->src, ci->dst, 0 /* no show progress */)) {
Riley Andrews4d04d242014-12-12 13:12:36 -0800943 ret = -1;
944 goto finish;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800945 }
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700946
947 if (copy_attrs && set_time_and_mode(ci->dst, ci->time, ci->mode)) {
Riley Andrews4d04d242014-12-12 13:12:36 -0800948 ret = -1;
949 goto finish;
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700950 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800951 pulled++;
952 } else {
953 skipped++;
954 }
955 free(ci);
956 }
957
958 fprintf(stderr, "%d file%s pulled. %d file%s skipped.\n",
959 pulled, (pulled == 1) ? "" : "s",
960 skipped, (skipped == 1) ? "" : "s");
961
Riley Andrews4d04d242014-12-12 13:12:36 -0800962finish:
963 free(lpath_clean);
964 free(rpath_clean);
965 return ret;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800966}
967
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700968int do_sync_pull(const char *rpath, const char *lpath, int show_progress, int copy_attrs)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800969{
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700970 unsigned mode, time;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800971 struct stat st;
972
973 int fd;
974
975 fd = adb_connect("sync:");
976 if(fd < 0) {
977 fprintf(stderr,"error: %s\n", adb_error());
978 return 1;
979 }
980
Lajos Molnarde8ff4a2013-04-19 12:41:09 -0700981 if(sync_readtime(fd, rpath, &time, &mode)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800982 return 1;
983 }
984 if(mode == 0) {
985 fprintf(stderr,"remote object '%s' does not exist\n", rpath);
986 return 1;
987 }
988
Matt Fischer457d81c2010-01-04 16:18:50 -0600989 if(S_ISREG(mode) || S_ISLNK(mode) || S_ISCHR(mode) || S_ISBLK(mode)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800990 if(stat(lpath, &st) == 0) {
991 if(S_ISDIR(st.st_mode)) {
992 /* if we're copying a remote file to a local directory,
993 ** we *really* want to copy to localdir + "/" + remotefilename
994 */
995 const char *name = adb_dirstop(rpath);
996 if(name == 0) {
997 name = rpath;
998 } else {
999 name++;
1000 }
1001 int tmplen = strlen(name) + strlen(lpath) + 2;
1002 char *tmp = malloc(tmplen);
1003 if(tmp == 0) return 1;
1004 snprintf(tmp, tmplen, "%s/%s", lpath, name);
1005 lpath = tmp;
1006 }
1007 }
1008 BEGIN();
Mark Lindner76f2a932014-03-11 17:55:59 -07001009 if (sync_recv(fd, rpath, lpath, show_progress)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001010 return 1;
1011 } else {
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001012 if (copy_attrs && set_time_and_mode(lpath, time, mode))
1013 return 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001014 END();
1015 sync_quit(fd);
1016 return 0;
1017 }
1018 } else if(S_ISDIR(mode)) {
1019 BEGIN();
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001020 if (copy_remote_dir_local(fd, rpath, lpath, copy_attrs)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001021 return 1;
1022 } else {
1023 END();
1024 sync_quit(fd);
1025 return 0;
1026 }
1027 } else {
1028 fprintf(stderr,"remote object '%s' not a file or directory\n", rpath);
1029 return 1;
1030 }
1031}
1032
Anthony Newnam705c9442010-02-22 08:36:49 -06001033int do_sync_sync(const char *lpath, const char *rpath, int listonly)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001034{
1035 fprintf(stderr,"syncing %s...\n",rpath);
1036
1037 int fd = adb_connect("sync:");
1038 if(fd < 0) {
1039 fprintf(stderr,"error: %s\n", adb_error());
1040 return 1;
1041 }
1042
1043 BEGIN();
Anthony Newnam705c9442010-02-22 08:36:49 -06001044 if(copy_local_dir_remote(fd, lpath, rpath, 1, listonly)){
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001045 return 1;
1046 } else {
1047 END();
1048 sync_quit(fd);
1049 return 0;
1050 }
1051}