2 smbget: a wget-like utility with support for recursive downloading and
4 Copyright (C) 2003-2004 Jelmer Vernooij <jelmer@samba.org>
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>. */
20 #include "popt_common.h"
21 #include "libsmbclient.h"
23 #if _FILE_OFFSET_BITS==64
24 #define OFF_T_FORMAT "%lld"
25 #define OFF_T_FORMAT_CAST long long
27 #define OFF_T_FORMAT "%ld"
28 #define OFF_T_FORMAT_CAST long
31 static int columns
= 0;
33 static int debuglevel
, update
;
34 static char *outputfile
;
37 static time_t total_start_time
= 0;
38 static off_t total_bytes
= 0;
40 #define SMB_MAXPATHLEN MAXPATHLEN
42 /* Number of bytes to read when checking whether local and remote file are really the same file */
43 #define RESUME_CHECK_SIZE 512
44 #define RESUME_DOWNLOAD_OFFSET 1024
45 #define RESUME_CHECK_OFFSET RESUME_DOWNLOAD_OFFSET+RESUME_CHECK_SIZE
46 /* Number of bytes to read at once */
47 #define SMB_DEFAULT_BLOCKSIZE 64000
49 static const char *username
= NULL
, *password
= NULL
, *workgroup
= NULL
;
50 static int nonprompt
= 0, quiet
= 0, dots
= 0, keep_permissions
= 0, verbose
= 0, send_stdout
= 0;
51 static int blocksize
= SMB_DEFAULT_BLOCKSIZE
;
53 static int smb_download_file(const char *base
, const char *name
, int recursive
,
54 int resume
, int toplevel
, char *outfile
);
56 static int get_num_cols(void)
60 if(ioctl(STDOUT_FILENO
, TIOCGWINSZ
, &ws
) < 0) {
65 #warning No support for TIOCGWINSZ
66 char *cols
= getenv("COLUMNS");
72 static void change_columns(int sig
)
74 columns
= get_num_cols();
77 static void human_readable(off_t s
, char *buffer
, int l
)
79 if(s
> 1024 * 1024 * 1024) snprintf(buffer
, l
, "%.2fGb", 1.0 * s
/ (1024 * 1024 * 1024));
80 else if(s
> 1024 * 1024) snprintf(buffer
, l
, "%.2fMb", 1.0 * s
/ (1024 * 1024));
81 else if(s
> 1024) snprintf(buffer
, l
, "%.2fkb", 1.0 * s
/ 1024);
82 else snprintf(buffer
, l
, OFF_T_FORMAT
"b", (OFF_T_FORMAT_CAST
)s
);
85 static void get_auth_data(const char *srv
, const char *shr
, char *wg
, int wglen
, char *un
, int unlen
, char *pw
, int pwlen
)
87 static char hasasked
= 0;
88 char *wgtmp
, *usertmp
;
94 if(!nonprompt
&& !username
) {
95 printf("Username for %s at %s [guest] ", shr
, srv
);
96 if (fgets(tmp
, sizeof(tmp
), stdin
) == NULL
) {
99 if ((strlen(tmp
) > 0) && (tmp
[strlen(tmp
)-1] == '\n')) {
100 tmp
[strlen(tmp
)-1] = '\0';
102 strncpy(un
, tmp
, unlen
-1);
103 } else if(username
) strncpy(un
, username
, unlen
-1);
105 if(!nonprompt
&& !password
) {
107 if (asprintf(&prompt
, "Password for %s at %s: ", shr
, srv
) == -1) {
110 pass
= getpass(prompt
);
112 strncpy(pw
, pass
, pwlen
-1);
113 } else if(password
) strncpy(pw
, password
, pwlen
-1);
115 if(workgroup
)strncpy(wg
, workgroup
, wglen
-1);
117 wgtmp
= SMB_STRNDUP(wg
, wglen
);
118 usertmp
= SMB_STRNDUP(un
, unlen
);
119 if(!quiet
)printf("Using workgroup %s, %s%s\n", wgtmp
, *usertmp
?"user ":"guest user", usertmp
);
120 free(wgtmp
); free(usertmp
);
123 /* Return 1 on error, 0 on success. */
125 static int smb_download_dir(const char *base
, const char *name
, int resume
)
127 char path
[SMB_MAXPATHLEN
];
129 struct smbc_dirent
*dirent
;
130 const char *relname
= name
;
132 struct stat remotestat
;
135 snprintf(path
, SMB_MAXPATHLEN
-1, "%s%s%s", base
, (base
[0] && name
[0] && name
[0] != '/' && base
[strlen(base
)-1] != '/')?"/":"", name
);
137 /* List files in directory and call smb_download_file on them */
138 dirhandle
= smbc_opendir(path
);
140 if (errno
== ENOTDIR
) {
141 return smb_download_file(base
, name
, 1, resume
,
144 fprintf(stderr
, "Can't open directory %s: %s\n", path
, strerror(errno
));
148 while(*relname
== '/')relname
++;
149 mkdir(relname
, 0755);
151 tmpname
= SMB_STRDUP(name
);
153 while((dirent
= smbc_readdir(dirhandle
))) {
155 if(!strcmp(dirent
->name
, ".") || !strcmp(dirent
->name
, ".."))continue;
156 if (asprintf(&newname
, "%s/%s", tmpname
, dirent
->name
) == -1) {
159 switch(dirent
->smbc_type
) {
161 ret
= smb_download_dir(base
, newname
, resume
);
165 ret
= smb_download_dir("smb://", dirent
->name
, resume
);
169 ret
= smb_download_dir("smb://", dirent
->name
, resume
);
173 ret
= smb_download_file(base
, newname
, 1, resume
, 0,
177 case SMBC_FILE_SHARE
:
178 ret
= smb_download_dir(base
, newname
, resume
);
181 case SMBC_PRINTER_SHARE
:
182 if(!quiet
)printf("Ignoring printer share %s\n", dirent
->name
);
185 case SMBC_COMMS_SHARE
:
186 if(!quiet
)printf("Ignoring comms share %s\n", dirent
->name
);
190 if(!quiet
)printf("Ignoring ipc$ share %s\n", dirent
->name
);
194 fprintf(stderr
, "Ignoring file '%s' of type '%d'\n", newname
, dirent
->smbc_type
);
201 if(keep_permissions
) {
202 if(smbc_fstat(dirhandle
, &remotestat
) < 0) {
203 fprintf(stderr
, "Unable to get stats on %s on remote server\n", path
);
204 smbc_closedir(dirhandle
);
208 if(chmod(relname
, remotestat
.st_mode
) < 0) {
209 fprintf(stderr
, "Unable to change mode of local dir %s to %o\n", relname
,
210 (unsigned int)remotestat
.st_mode
);
211 smbc_closedir(dirhandle
);
216 smbc_closedir(dirhandle
);
220 static char *print_time(long t
)
222 static char buffer
[100];
223 int secs
, mins
, hours
;
225 strncpy(buffer
, "Unknown", sizeof(buffer
));
230 mins
= (int)t
/ 60 % 60;
231 hours
= (int)t
/ (60 * 60);
232 snprintf(buffer
, sizeof(buffer
)-1, "%02d:%02d:%02d", hours
, mins
, secs
);
236 static void print_progress(const char *name
, time_t start
, time_t now
, off_t start_pos
, off_t pos
, off_t total
)
241 char hpos
[20], htotal
[20], havg
[20];
242 char *status
, *filename
;
244 if(now
- start
)avg
= 1.0 * (pos
- start_pos
) / (now
- start
);
245 eta
= (total
- pos
) / avg
;
246 if(total
)prcnt
= 100.0 * pos
/ total
;
248 human_readable(pos
, hpos
, sizeof(hpos
));
249 human_readable(total
, htotal
, sizeof(htotal
));
250 human_readable(avg
, havg
, sizeof(havg
));
252 len
= asprintf(&status
, "%s of %s (%.2f%%) at %s/s ETA: %s", hpos
, htotal
, prcnt
, havg
, print_time(eta
));
258 int required
= strlen(name
), available
= columns
- len
- strlen("[] ");
259 if(required
> available
) {
260 if (asprintf(&filename
, "...%s", name
+ required
- available
+ 3) == -1) {
264 filename
= SMB_STRNDUP(name
, available
);
266 } else filename
= SMB_STRDUP(name
);
268 fprintf(stderr
, "\r[%s] %s", filename
, status
);
270 free(filename
); free(status
);
273 /* Return 1 on error, 0 on success. */
275 static int smb_download_file(const char *base
, const char *name
, int recursive
,
276 int resume
, int toplevel
, char *outfile
)
278 int remotehandle
, localhandle
;
279 time_t start_time
= time_mono(NULL
);
281 char path
[SMB_MAXPATHLEN
];
282 char checkbuf
[2][RESUME_CHECK_SIZE
];
283 char *readbuf
= NULL
;
284 off_t offset_download
= 0, offset_check
= 0, curpos
= 0, start_offset
= 0;
285 struct stat localstat
, remotestat
;
287 snprintf(path
, SMB_MAXPATHLEN
-1, "%s%s%s", base
, (*base
&& *name
&& name
[0] != '/' && base
[strlen(base
)-1] != '/')?"/":"", name
);
289 remotehandle
= smbc_open(path
, O_RDONLY
, 0755);
291 if(remotehandle
< 0) {
295 fprintf(stderr
, "%s is a directory. Specify -R to download recursively\n", path
);
298 return smb_download_dir(base
, name
, resume
);
301 fprintf(stderr
, "%s can't be found on the remote server\n", path
);
305 fprintf(stderr
, "Not enough memory\n");
309 fprintf(stderr
, "The share name used in %s does not exist\n", path
);
313 fprintf(stderr
, "You don't have enough permissions to access %s\n", path
);
322 if(smbc_fstat(remotehandle
, &remotestat
) < 0) {
323 fprintf(stderr
, "Can't stat %s: %s\n", path
, strerror(errno
));
327 if(outfile
) newpath
= outfile
;
329 newpath
= strrchr(base
, '/');
330 if(newpath
)newpath
++; else newpath
= base
;
331 } else newpath
= name
;
333 if (!toplevel
&& (newpath
[0] == '/')) {
337 /* Open local file according to the mode */
339 /* if it is up-to-date, skip */
340 if(stat(newpath
, &localstat
) == 0 &&
341 localstat
.st_mtime
>= remotestat
.st_mtime
) {
343 printf("%s is up-to-date, skipping\n", newpath
);
344 smbc_close(remotehandle
);
347 /* else open it for writing and truncate if it exists */
348 localhandle
= open(newpath
, O_CREAT
| O_NONBLOCK
| O_RDWR
| O_TRUNC
, 0775);
349 if(localhandle
< 0) {
350 fprintf(stderr
, "Can't open %s : %s\n", newpath
,
352 smbc_close(remotehandle
);
356 } else if(!send_stdout
) {
357 localhandle
= open(newpath
, O_CREAT
| O_NONBLOCK
| O_RDWR
| (!resume
?O_EXCL
:0), 0755);
358 if(localhandle
< 0) {
359 fprintf(stderr
, "Can't open %s: %s\n", newpath
, strerror(errno
));
360 smbc_close(remotehandle
);
364 if (fstat(localhandle
, &localstat
) != 0) {
365 fprintf(stderr
, "Can't fstat %s: %s\n", newpath
, strerror(errno
));
366 smbc_close(remotehandle
);
371 start_offset
= localstat
.st_size
;
373 if(localstat
.st_size
&& localstat
.st_size
== remotestat
.st_size
) {
374 if(verbose
)fprintf(stderr
, "%s is already downloaded completely.\n", path
);
375 else if(!quiet
)fprintf(stderr
, "%s\n", path
);
376 smbc_close(remotehandle
);
381 if(localstat
.st_size
> RESUME_CHECK_OFFSET
&& remotestat
.st_size
> RESUME_CHECK_OFFSET
) {
382 offset_download
= localstat
.st_size
- RESUME_DOWNLOAD_OFFSET
;
383 offset_check
= localstat
.st_size
- RESUME_CHECK_OFFSET
;
384 if(verbose
)printf("Trying to start resume of %s at "OFF_T_FORMAT
"\n"
385 "At the moment "OFF_T_FORMAT
" of "OFF_T_FORMAT
" bytes have been retrieved\n",
386 newpath
, (OFF_T_FORMAT_CAST
)offset_check
,
387 (OFF_T_FORMAT_CAST
)localstat
.st_size
,
388 (OFF_T_FORMAT_CAST
)remotestat
.st_size
);
393 /* First, check all bytes from offset_check to offset_download */
394 off1
= lseek(localhandle
, offset_check
, SEEK_SET
);
396 fprintf(stderr
, "Can't seek to "OFF_T_FORMAT
" in local file %s\n",
397 (OFF_T_FORMAT_CAST
)offset_check
, newpath
);
398 smbc_close(remotehandle
); close(localhandle
);
402 off2
= smbc_lseek(remotehandle
, offset_check
, SEEK_SET
);
404 fprintf(stderr
, "Can't seek to "OFF_T_FORMAT
" in remote file %s\n",
405 (OFF_T_FORMAT_CAST
)offset_check
, newpath
);
406 smbc_close(remotehandle
); close(localhandle
);
411 fprintf(stderr
, "Offset in local and remote files is different (local: "OFF_T_FORMAT
", remote: "OFF_T_FORMAT
")\n",
412 (OFF_T_FORMAT_CAST
)off1
,
413 (OFF_T_FORMAT_CAST
)off2
);
417 if(smbc_read(remotehandle
, checkbuf
[0], RESUME_CHECK_SIZE
) != RESUME_CHECK_SIZE
) {
418 fprintf(stderr
, "Can't read %d bytes from remote file %s\n", RESUME_CHECK_SIZE
, path
);
419 smbc_close(remotehandle
); close(localhandle
);
423 if(read(localhandle
, checkbuf
[1], RESUME_CHECK_SIZE
) != RESUME_CHECK_SIZE
) {
424 fprintf(stderr
, "Can't read %d bytes from local file %s\n", RESUME_CHECK_SIZE
, name
);
425 smbc_close(remotehandle
); close(localhandle
);
429 if(memcmp(checkbuf
[0], checkbuf
[1], RESUME_CHECK_SIZE
) == 0) {
430 if(verbose
)printf("Current local and remote file appear to be the same. Starting download from offset "OFF_T_FORMAT
"\n", (OFF_T_FORMAT_CAST
)offset_download
);
432 fprintf(stderr
, "Local and remote file appear to be different, not doing resume for %s\n", path
);
433 smbc_close(remotehandle
); close(localhandle
);
438 localhandle
= STDOUT_FILENO
;
444 readbuf
= (char *)SMB_MALLOC(blocksize
);
449 /* Now, download all bytes from offset_download to the end */
450 for(curpos
= offset_download
; curpos
< remotestat
.st_size
; curpos
+=blocksize
) {
451 ssize_t bytesread
= smbc_read(remotehandle
, readbuf
, blocksize
);
453 fprintf(stderr
, "Can't read %u bytes at offset "OFF_T_FORMAT
", file %s\n", (unsigned int)blocksize
, (OFF_T_FORMAT_CAST
)curpos
, path
);
454 smbc_close(remotehandle
);
455 if (localhandle
!= STDOUT_FILENO
) close(localhandle
);
460 total_bytes
+= bytesread
;
462 if(write(localhandle
, readbuf
, bytesread
) < 0) {
463 fprintf(stderr
, "Can't write %u bytes to local file %s at offset "OFF_T_FORMAT
"\n", (unsigned int)bytesread
, path
, (OFF_T_FORMAT_CAST
)curpos
);
465 smbc_close(remotehandle
);
466 if (localhandle
!= STDOUT_FILENO
) close(localhandle
);
470 if(dots
)fputc('.', stderr
);
472 print_progress(newpath
, start_time
, time_mono(NULL
),
473 start_offset
, curpos
, remotestat
.st_size
);
481 printf("%s downloaded\n", path
);
484 fprintf(stderr
, "\r%s", path
);
486 for(i
= strlen(path
); i
< columns
; i
++) {
493 if(keep_permissions
&& !send_stdout
) {
494 if(fchmod(localhandle
, remotestat
.st_mode
) < 0) {
495 fprintf(stderr
, "Unable to change mode of local file %s to %o\n", path
,
496 (unsigned int)remotestat
.st_mode
);
497 smbc_close(remotehandle
);
503 smbc_close(remotehandle
);
504 if (localhandle
!= STDOUT_FILENO
) close(localhandle
);
508 static void clean_exit(void)
511 human_readable(total_bytes
, bs
, sizeof(bs
));
512 if(!quiet
)fprintf(stderr
, "Downloaded %s in %lu seconds\n", bs
,
513 (unsigned long)(time_mono(NULL
) - total_start_time
));
517 static void signal_quit(int v
)
522 static int readrcfile(const char *name
, const struct poptOption long_options
[])
524 FILE *fd
= fopen(name
, "r");
526 char var
[101], val
[101];
528 int *intdata
; char **stringdata
;
530 fprintf(stderr
, "Can't open RC file %s\n", name
);
536 if(fscanf(fd
, "%100s %100s\n", var
, val
) < 2) {
537 fprintf(stderr
, "Can't parse line %d of %s, ignoring.\n", lineno
, name
);
543 for(i
= 0; long_options
[i
].shortName
; i
++) {
544 if(!long_options
[i
].longName
)continue;
545 if(strcmp(long_options
[i
].longName
, var
)) continue;
546 if(!long_options
[i
].arg
)continue;
548 switch(long_options
[i
].argInfo
) {
550 intdata
= (int *)long_options
[i
].arg
;
551 if(!strcmp(val
, "on")) *intdata
= 1;
552 else if(!strcmp(val
, "off")) *intdata
= 0;
553 else fprintf(stderr
, "Illegal value %s for %s at line %d in %s\n", val
, var
, lineno
, name
);
556 intdata
= (int *)long_options
[i
].arg
;
557 *intdata
= atoi(val
);
559 case POPT_ARG_STRING
:
560 stringdata
= (char **)long_options
[i
].arg
;
561 *stringdata
= SMB_STRDUP(val
);
564 fprintf(stderr
, "Invalid variable %s at line %d in %s\n", var
, lineno
, name
);
571 fprintf(stderr
, "Invalid variable %s at line %d in %s\n", var
, lineno
, name
);
579 int main(int argc
, const char **argv
)
582 const char *file
= NULL
;
584 bool smb_encrypt
= false;
585 int resume
= 0, recursive
= 0;
586 TALLOC_CTX
*frame
= talloc_stackframe();
588 struct poptOption long_options
[] = {
589 {"guest", 'a', POPT_ARG_NONE
, NULL
, 'a', "Work as user guest" },
590 {"encrypt", 'e', POPT_ARG_NONE
, NULL
, 'e', "Encrypt SMB transport (UNIX extended servers only)" },
591 {"resume", 'r', POPT_ARG_NONE
, &resume
, 0, "Automatically resume aborted files" },
592 {"update", 'U', POPT_ARG_NONE
, &update
, 0, "Download only when remote file is newer than local file or local file is missing"},
593 {"recursive", 'R', POPT_ARG_NONE
, &recursive
, 0, "Recursively download files" },
594 {"username", 'u', POPT_ARG_STRING
, &username
, 'u', "Username to use" },
595 {"password", 'p', POPT_ARG_STRING
, &password
, 'p', "Password to use" },
596 {"workgroup", 'w', POPT_ARG_STRING
, &workgroup
, 'w', "Workgroup to use (optional)" },
597 {"nonprompt", 'n', POPT_ARG_NONE
, &nonprompt
, 'n', "Don't ask anything (non-interactive)" },
598 {"debuglevel", 'd', POPT_ARG_INT
, &debuglevel
, 'd', "Debuglevel to use" },
599 {"outputfile", 'o', POPT_ARG_STRING
, &outputfile
, 'o', "Write downloaded data to specified file" },
600 {"stdout", 'O', POPT_ARG_NONE
, &send_stdout
, 'O', "Write data to stdout" },
601 {"dots", 'D', POPT_ARG_NONE
, &dots
, 'D', "Show dots as progress indication" },
602 {"quiet", 'q', POPT_ARG_NONE
, &quiet
, 'q', "Be quiet" },
603 {"verbose", 'v', POPT_ARG_NONE
, &verbose
, 'v', "Be verbose" },
604 {"keep-permissions", 'P', POPT_ARG_NONE
, &keep_permissions
, 'P', "Keep permissions" },
605 {"blocksize", 'b', POPT_ARG_INT
, &blocksize
, 'b', "Change number of bytes in a block"},
606 {"rcfile", 'f', POPT_ARG_STRING
, NULL
, 'f', "Use specified rc file"},
614 /* only read rcfile if it exists */
615 if (asprintf(&rcfile
, "%s/.smbgetrc", getenv("HOME")) == -1) {
618 if(access(rcfile
, F_OK
) == 0)
619 readrcfile(rcfile
, long_options
);
623 signal(SIGWINCH
, change_columns
);
625 signal(SIGINT
, signal_quit
);
626 signal(SIGTERM
, signal_quit
);
628 pc
= poptGetContext(argv
[0], argc
, argv
, long_options
, 0);
630 while((c
= poptGetNextOpt(pc
)) >= 0) {
633 readrcfile(poptGetOptArg(pc
), long_options
);
636 username
= ""; password
= "";
644 if((send_stdout
|| resume
|| outputfile
) && update
) {
645 fprintf(stderr
, "The -o, -R or -O and -U options can not be used together.\n");
648 if((send_stdout
|| outputfile
) && recursive
) {
649 fprintf(stderr
, "The -o or -O and -R options can not be used together.\n");
653 if(outputfile
&& send_stdout
) {
654 fprintf(stderr
, "The -o and -O options cannot be used together.\n");
658 if(smbc_init(get_auth_data
, debuglevel
) < 0) {
659 fprintf(stderr
, "Unable to initialize libsmbclient\n");
664 SMBCCTX
*smb_ctx
= smbc_set_context(NULL
);
665 smbc_option_set(smb_ctx
,
666 CONST_DISCARD(char *, "smb_encrypt_level"),
670 columns
= get_num_cols();
672 total_start_time
= time_mono(NULL
);
674 while ( (file
= poptGetArg(pc
)) ) {
676 ret
= smb_download_file(file
, "", recursive
, resume
,
679 ret
= smb_download_dir(file
, "", resume
);