]> andersk Git - openssh.git/blob - sftp-int.c
46cbd4a129eb59cbf66cce1b39626537d0d50a41
[openssh.git] / sftp-int.c
1 /*
2  * Copyright (c) 2001,2002 Damien Miller.  All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
14  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
15  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
16  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
17  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
18  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
22  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23  */
24
25 /* XXX: globbed ls */
26 /* XXX: recursive operations */
27
28 #include "includes.h"
29 RCSID("$OpenBSD: sftp-int.c,v 1.45 2002/03/19 06:32:56 mpech Exp $");
30
31 #include "buffer.h"
32 #include "xmalloc.h"
33 #include "log.h"
34 #include "pathnames.h"
35
36 #include "sftp.h"
37 #include "sftp-common.h"
38 #include "sftp-glob.h"
39 #include "sftp-client.h"
40 #include "sftp-int.h"
41
42 /* File to read commands from */
43 extern FILE *infile;
44
45 /* Size of buffer used when copying files */
46 extern size_t copy_buffer_len;
47
48 /* Number of concurrent outstanding requests */
49 extern int num_requests;
50
51 /* Seperators for interactive commands */
52 #define WHITESPACE " \t\r\n"
53
54 /* Commands for interactive mode */
55 #define I_CHDIR         1
56 #define I_CHGRP         2
57 #define I_CHMOD         3
58 #define I_CHOWN         4
59 #define I_GET           5
60 #define I_HELP          6
61 #define I_LCHDIR        7
62 #define I_LLS           8
63 #define I_LMKDIR        9
64 #define I_LPWD          10
65 #define I_LS            11
66 #define I_LUMASK        12
67 #define I_MKDIR         13
68 #define I_PUT           14
69 #define I_PWD           15
70 #define I_QUIT          16
71 #define I_RENAME        17
72 #define I_RM            18
73 #define I_RMDIR         19
74 #define I_SHELL         20
75 #define I_SYMLINK       21
76 #define I_VERSION       22
77
78 struct CMD {
79         const char *c;
80         const int n;
81 };
82
83 const struct CMD cmds[] = {
84         { "bye",        I_QUIT },
85         { "cd",         I_CHDIR },
86         { "chdir",      I_CHDIR },
87         { "chgrp",      I_CHGRP },
88         { "chmod",      I_CHMOD },
89         { "chown",      I_CHOWN },
90         { "dir",        I_LS },
91         { "exit",       I_QUIT },
92         { "get",        I_GET },
93         { "mget",       I_GET },
94         { "help",       I_HELP },
95         { "lcd",        I_LCHDIR },
96         { "lchdir",     I_LCHDIR },
97         { "lls",        I_LLS },
98         { "lmkdir",     I_LMKDIR },
99         { "ln",         I_SYMLINK },
100         { "lpwd",       I_LPWD },
101         { "ls",         I_LS },
102         { "lumask",     I_LUMASK },
103         { "mkdir",      I_MKDIR },
104         { "put",        I_PUT },
105         { "mput",       I_PUT },
106         { "pwd",        I_PWD },
107         { "quit",       I_QUIT },
108         { "rename",     I_RENAME },
109         { "rm",         I_RM },
110         { "rmdir",      I_RMDIR },
111         { "symlink",    I_SYMLINK },
112         { "version",    I_VERSION },
113         { "!",          I_SHELL },
114         { "?",          I_HELP },
115         { NULL,                 -1}
116 };
117
118 static void
119 help(void)
120 {
121         printf("Available commands:\n");
122         printf("cd path                       Change remote directory to 'path'\n");
123         printf("lcd path                      Change local directory to 'path'\n");
124         printf("chgrp grp path                Change group of file 'path' to 'grp'\n");
125         printf("chmod mode path               Change permissions of file 'path' to 'mode'\n");
126         printf("chown own path                Change owner of file 'path' to 'own'\n");
127         printf("help                          Display this help text\n");
128         printf("get remote-path [local-path]  Download file\n");
129         printf("lls [ls-options [path]]       Display local directory listing\n");
130         printf("ln oldpath newpath            Symlink remote file\n");
131         printf("lmkdir path                   Create local directory\n");
132         printf("lpwd                          Print local working directory\n");
133         printf("ls [path]                     Display remote directory listing\n");
134         printf("lumask umask                  Set local umask to 'umask'\n");
135         printf("mkdir path                    Create remote directory\n");
136         printf("put local-path [remote-path]  Upload file\n");
137         printf("pwd                           Display remote working directory\n");
138         printf("exit                          Quit sftp\n");
139         printf("quit                          Quit sftp\n");
140         printf("rename oldpath newpath        Rename remote file\n");
141         printf("rmdir path                    Remove remote directory\n");
142         printf("rm path                       Delete remote file\n");
143         printf("symlink oldpath newpath       Symlink remote file\n");
144         printf("version                       Show SFTP version\n");
145         printf("!command                      Execute 'command' in local shell\n");
146         printf("!                             Escape to local shell\n");
147         printf("?                             Synonym for help\n");
148 }
149
150 static void
151 local_do_shell(const char *args)
152 {
153         int status;
154         char *shell;
155         pid_t pid;
156
157         if (!*args)
158                 args = NULL;
159
160         if ((shell = getenv("SHELL")) == NULL)
161                 shell = _PATH_BSHELL;
162
163         if ((pid = fork()) == -1)
164                 fatal("Couldn't fork: %s", strerror(errno));
165
166         if (pid == 0) {
167                 /* XXX: child has pipe fds to ssh subproc open - issue? */
168                 if (args) {
169                         debug3("Executing %s -c \"%s\"", shell, args);
170                         execl(shell, shell, "-c", args, (char *)NULL);
171                 } else {
172                         debug3("Executing %s", shell);
173                         execl(shell, shell, (char *)NULL);
174                 }
175                 fprintf(stderr, "Couldn't execute \"%s\": %s\n", shell,
176                     strerror(errno));
177                 _exit(1);
178         }
179         if (waitpid(pid, &status, 0) == -1)
180                 fatal("Couldn't wait for child: %s", strerror(errno));
181         if (!WIFEXITED(status))
182                 error("Shell exited abormally");
183         else if (WEXITSTATUS(status))
184                 error("Shell exited with status %d", WEXITSTATUS(status));
185 }
186
187 static void
188 local_do_ls(const char *args)
189 {
190         if (!args || !*args)
191                 local_do_shell(_PATH_LS);
192         else {
193                 int len = strlen(_PATH_LS " ") + strlen(args) + 1;
194                 char *buf = xmalloc(len);
195
196                 /* XXX: quoting - rip quoting code from ftp? */
197                 snprintf(buf, len, _PATH_LS " %s", args);
198                 local_do_shell(buf);
199                 xfree(buf);
200         }
201 }
202
203 static char *
204 path_append(char *p1, char *p2)
205 {
206         char *ret;
207         int len = strlen(p1) + strlen(p2) + 2;
208
209         ret = xmalloc(len);
210         strlcpy(ret, p1, len);
211         if (strcmp(p1, "/") != 0)
212                 strlcat(ret, "/", len);
213         strlcat(ret, p2, len);
214
215         return(ret);
216 }
217
218 static char *
219 make_absolute(char *p, char *pwd)
220 {
221         char *abs;
222
223         /* Derelativise */
224         if (p && p[0] != '/') {
225                 abs = path_append(pwd, p);
226                 xfree(p);
227                 return(abs);
228         } else
229                 return(p);
230 }
231
232 static int
233 infer_path(const char *p, char **ifp)
234 {
235         char *cp;
236
237         cp = strrchr(p, '/');
238         if (cp == NULL) {
239                 *ifp = xstrdup(p);
240                 return(0);
241         }
242
243         if (!cp[1]) {
244                 error("Invalid path");
245                 return(-1);
246         }
247
248         *ifp = xstrdup(cp + 1);
249         return(0);
250 }
251
252 static int
253 parse_getput_flags(const char **cpp, int *pflag)
254 {
255         const char *cp = *cpp;
256
257         /* Check for flags */
258         if (cp[0] == '-' && cp[1] && strchr(WHITESPACE, cp[2])) {
259                 switch (cp[1]) {
260                 case 'p':
261                 case 'P':
262                         *pflag = 1;
263                         break;
264                 default:
265                         error("Invalid flag -%c", cp[1]);
266                         return(-1);
267                 }
268                 cp += 2;
269                 *cpp = cp + strspn(cp, WHITESPACE);
270         }
271
272         return(0);
273 }
274
275 static int
276 get_pathname(const char **cpp, char **path)
277 {
278         const char *cp = *cpp, *end;
279         char quot;
280         int i;
281
282         cp += strspn(cp, WHITESPACE);
283         if (!*cp) {
284                 *cpp = cp;
285                 *path = NULL;
286                 return (0);
287         }
288
289         /* Check for quoted filenames */
290         if (*cp == '\"' || *cp == '\'') {
291                 quot = *cp++;
292
293                 end = strchr(cp, quot);
294                 if (end == NULL) {
295                         error("Unterminated quote");
296                         goto fail;
297                 }
298                 if (cp == end) {
299                         error("Empty quotes");
300                         goto fail;
301                 }
302                 *cpp = end + 1 + strspn(end + 1, WHITESPACE);
303         } else {
304                 /* Read to end of filename */
305                 end = strpbrk(cp, WHITESPACE);
306                 if (end == NULL)
307                         end = strchr(cp, '\0');
308                 *cpp = end + strspn(end, WHITESPACE);
309         }
310
311         i = end - cp;
312
313         *path = xmalloc(i + 1);
314         memcpy(*path, cp, i);
315         (*path)[i] = '\0';
316         return(0);
317
318  fail:
319         *path = NULL;
320         return (-1);
321 }
322
323 static int
324 is_dir(char *path)
325 {
326         struct stat sb;
327
328         /* XXX: report errors? */
329         if (stat(path, &sb) == -1)
330                 return(0);
331
332         return(sb.st_mode & S_IFDIR);
333 }
334
335 static int
336 remote_is_dir(struct sftp_conn *conn, char *path)
337 {
338         Attrib *a;
339
340         /* XXX: report errors? */
341         if ((a = do_stat(conn, path, 1)) == NULL)
342                 return(0);
343         if (!(a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS))
344                 return(0);
345         return(a->perm & S_IFDIR);
346 }
347
348 static int
349 process_get(struct sftp_conn *conn, char *src, char *dst, char *pwd, int pflag)
350 {
351         char *abs_src = NULL;
352         char *abs_dst = NULL;
353         char *tmp;
354         glob_t g;
355         int err = 0;
356         int i;
357
358         abs_src = xstrdup(src);
359         abs_src = make_absolute(abs_src, pwd);
360
361         memset(&g, 0, sizeof(g));
362         debug3("Looking up %s", abs_src);
363         if (remote_glob(conn, abs_src, 0, NULL, &g)) {
364                 error("File \"%s\" not found.", abs_src);
365                 err = -1;
366                 goto out;
367         }
368
369         /* Only one match, dst may be file, directory or unspecified */
370         if (g.gl_pathv[0] && g.gl_matchc == 1) {
371                 if (dst) {
372                         /* If directory specified, append filename */
373                         if (is_dir(dst)) {
374                                 if (infer_path(g.gl_pathv[0], &tmp)) {
375                                         err = 1;
376                                         goto out;
377                                 }
378                                 abs_dst = path_append(dst, tmp);
379                                 xfree(tmp);
380                         } else
381                                 abs_dst = xstrdup(dst);
382                 } else if (infer_path(g.gl_pathv[0], &abs_dst)) {
383                         err = -1;
384                         goto out;
385                 }
386                 printf("Fetching %s to %s\n", g.gl_pathv[0], abs_dst);
387                 err = do_download(conn, g.gl_pathv[0], abs_dst, pflag);
388                 goto out;
389         }
390
391         /* Multiple matches, dst may be directory or unspecified */
392         if (dst && !is_dir(dst)) {
393                 error("Multiple files match, but \"%s\" is not a directory",
394                     dst);
395                 err = -1;
396                 goto out;
397         }
398
399         for (i = 0; g.gl_pathv[i]; i++) {
400                 if (infer_path(g.gl_pathv[i], &tmp)) {
401                         err = -1;
402                         goto out;
403                 }
404                 if (dst) {
405                         abs_dst = path_append(dst, tmp);
406                         xfree(tmp);
407                 } else
408                         abs_dst = tmp;
409
410                 printf("Fetching %s to %s\n", g.gl_pathv[i], abs_dst);
411                 if (do_download(conn, g.gl_pathv[i], abs_dst, pflag) == -1)
412                         err = -1;
413                 xfree(abs_dst);
414                 abs_dst = NULL;
415         }
416
417 out:
418         xfree(abs_src);
419         if (abs_dst)
420                 xfree(abs_dst);
421         globfree(&g);
422         return(err);
423 }
424
425 static int
426 process_put(struct sftp_conn *conn, char *src, char *dst, char *pwd, int pflag)
427 {
428         char *tmp_dst = NULL;
429         char *abs_dst = NULL;
430         char *tmp;
431         glob_t g;
432         int err = 0;
433         int i;
434
435         if (dst) {
436                 tmp_dst = xstrdup(dst);
437                 tmp_dst = make_absolute(tmp_dst, pwd);
438         }
439
440         memset(&g, 0, sizeof(g));
441         debug3("Looking up %s", src);
442         if (glob(src, 0, NULL, &g)) {
443                 error("File \"%s\" not found.", src);
444                 err = -1;
445                 goto out;
446         }
447
448         /* Only one match, dst may be file, directory or unspecified */
449         if (g.gl_pathv[0] && g.gl_matchc == 1) {
450                 if (tmp_dst) {
451                         /* If directory specified, append filename */
452                         if (remote_is_dir(conn, tmp_dst)) {
453                                 if (infer_path(g.gl_pathv[0], &tmp)) {
454                                         err = 1;
455                                         goto out;
456                                 }
457                                 abs_dst = path_append(tmp_dst, tmp);
458                                 xfree(tmp);
459                         } else
460                                 abs_dst = xstrdup(tmp_dst);
461                 } else {
462                         if (infer_path(g.gl_pathv[0], &abs_dst)) {
463                                 err = -1;
464                                 goto out;
465                         }
466                         abs_dst = make_absolute(abs_dst, pwd);
467                 }
468                 printf("Uploading %s to %s\n", g.gl_pathv[0], abs_dst);
469                 err = do_upload(conn, g.gl_pathv[0], abs_dst, pflag);
470                 goto out;
471         }
472
473         /* Multiple matches, dst may be directory or unspecified */
474         if (tmp_dst && !remote_is_dir(conn, tmp_dst)) {
475                 error("Multiple files match, but \"%s\" is not a directory",
476                     tmp_dst);
477                 err = -1;
478                 goto out;
479         }
480
481         for (i = 0; g.gl_pathv[i]; i++) {
482                 if (infer_path(g.gl_pathv[i], &tmp)) {
483                         err = -1;
484                         goto out;
485                 }
486                 if (tmp_dst) {
487                         abs_dst = path_append(tmp_dst, tmp);
488                         xfree(tmp);
489                 } else
490                         abs_dst = make_absolute(tmp, pwd);
491
492                 printf("Uploading %s to %s\n", g.gl_pathv[i], abs_dst);
493                 if (do_upload(conn, g.gl_pathv[i], abs_dst, pflag) == -1)
494                         err = -1;
495         }
496
497 out:
498         if (abs_dst)
499                 xfree(abs_dst);
500         if (tmp_dst)
501                 xfree(tmp_dst);
502         return(err);
503 }
504
505 static int
506 parse_args(const char **cpp, int *pflag, unsigned long *n_arg,
507     char **path1, char **path2)
508 {
509         const char *cmd, *cp = *cpp;
510         char *cp2;
511         int base = 0;
512         long l;
513         int i, cmdnum;
514
515         /* Skip leading whitespace */
516         cp = cp + strspn(cp, WHITESPACE);
517
518         /* Ignore blank lines */
519         if (!*cp)
520                 return(-1);
521
522         /* Figure out which command we have */
523         for (i = 0; cmds[i].c; i++) {
524                 int cmdlen = strlen(cmds[i].c);
525
526                 /* Check for command followed by whitespace */
527                 if (!strncasecmp(cp, cmds[i].c, cmdlen) &&
528                     strchr(WHITESPACE, cp[cmdlen])) {
529                         cp += cmdlen;
530                         cp = cp + strspn(cp, WHITESPACE);
531                         break;
532                 }
533         }
534         cmdnum = cmds[i].n;
535         cmd = cmds[i].c;
536
537         /* Special case */
538         if (*cp == '!') {
539                 cp++;
540                 cmdnum = I_SHELL;
541         } else if (cmdnum == -1) {
542                 error("Invalid command.");
543                 return(-1);
544         }
545
546         /* Get arguments and parse flags */
547         *pflag = *n_arg = 0;
548         *path1 = *path2 = NULL;
549         switch (cmdnum) {
550         case I_GET:
551         case I_PUT:
552                 if (parse_getput_flags(&cp, pflag))
553                         return(-1);
554                 /* Get first pathname (mandatory) */
555                 if (get_pathname(&cp, path1))
556                         return(-1);
557                 if (*path1 == NULL) {
558                         error("You must specify at least one path after a "
559                             "%s command.", cmd);
560                         return(-1);
561                 }
562                 /* Try to get second pathname (optional) */
563                 if (get_pathname(&cp, path2))
564                         return(-1);
565                 break;
566         case I_RENAME:
567         case I_SYMLINK:
568                 if (get_pathname(&cp, path1))
569                         return(-1);
570                 if (get_pathname(&cp, path2))
571                         return(-1);
572                 if (!*path1 || !*path2) {
573                         error("You must specify two paths after a %s "
574                             "command.", cmd);
575                         return(-1);
576                 }
577                 break;
578         case I_RM:
579         case I_MKDIR:
580         case I_RMDIR:
581         case I_CHDIR:
582         case I_LCHDIR:
583         case I_LMKDIR:
584                 /* Get pathname (mandatory) */
585                 if (get_pathname(&cp, path1))
586                         return(-1);
587                 if (*path1 == NULL) {
588                         error("You must specify a path after a %s command.",
589                             cmd);
590                         return(-1);
591                 }
592                 break;
593         case I_LS:
594                 /* Path is optional */
595                 if (get_pathname(&cp, path1))
596                         return(-1);
597                 break;
598         case I_LLS:
599         case I_SHELL:
600                 /* Uses the rest of the line */
601                 break;
602         case I_LUMASK:
603                 base = 8;
604         case I_CHMOD:
605                 base = 8;
606         case I_CHOWN:
607         case I_CHGRP:
608                 /* Get numeric arg (mandatory) */
609                 l = strtol(cp, &cp2, base);
610                 if (cp2 == cp || ((l == LONG_MIN || l == LONG_MAX) &&
611                     errno == ERANGE) || l < 0) {
612                         error("You must supply a numeric argument "
613                             "to the %s command.", cmd);
614                         return(-1);
615                 }
616                 cp = cp2;
617                 *n_arg = l;
618                 if (cmdnum == I_LUMASK && strchr(WHITESPACE, *cp))
619                         break;
620                 if (cmdnum == I_LUMASK || !strchr(WHITESPACE, *cp)) {
621                         error("You must supply a numeric argument "
622                             "to the %s command.", cmd);
623                         return(-1);
624                 }
625                 cp += strspn(cp, WHITESPACE);
626
627                 /* Get pathname (mandatory) */
628                 if (get_pathname(&cp, path1))
629                         return(-1);
630                 if (*path1 == NULL) {
631                         error("You must specify a path after a %s command.",
632                             cmd);
633                         return(-1);
634                 }
635                 break;
636         case I_QUIT:
637         case I_PWD:
638         case I_LPWD:
639         case I_HELP:
640         case I_VERSION:
641                 break;
642         default:
643                 fatal("Command not implemented");
644         }
645
646         *cpp = cp;
647         return(cmdnum);
648 }
649
650 static int
651 parse_dispatch_command(struct sftp_conn *conn, const char *cmd, char **pwd)
652 {
653         char *path1, *path2, *tmp;
654         int pflag, cmdnum, i;
655         unsigned long n_arg;
656         Attrib a, *aa;
657         char path_buf[MAXPATHLEN];
658         int err = 0;
659         glob_t g;
660
661         path1 = path2 = NULL;
662         cmdnum = parse_args(&cmd, &pflag, &n_arg, &path1, &path2);
663
664         memset(&g, 0, sizeof(g));
665
666         /* Perform command */
667         switch (cmdnum) {
668         case -1:
669                 break;
670         case I_GET:
671                 err = process_get(conn, path1, path2, *pwd, pflag);
672                 break;
673         case I_PUT:
674                 err = process_put(conn, path1, path2, *pwd, pflag);
675                 break;
676         case I_RENAME:
677                 path1 = make_absolute(path1, *pwd);
678                 path2 = make_absolute(path2, *pwd);
679                 err = do_rename(conn, path1, path2);
680                 break;
681         case I_SYMLINK:
682                 path2 = make_absolute(path2, *pwd);
683                 err = do_symlink(conn, path1, path2);
684                 break;
685         case I_RM:
686                 path1 = make_absolute(path1, *pwd);
687                 remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
688                 for (i = 0; g.gl_pathv[i]; i++) {
689                         printf("Removing %s\n", g.gl_pathv[i]);
690                         if (do_rm(conn, g.gl_pathv[i]) == -1)
691                                 err = -1;
692                 }
693                 break;
694         case I_MKDIR:
695                 path1 = make_absolute(path1, *pwd);
696                 attrib_clear(&a);
697                 a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
698                 a.perm = 0777;
699                 err = do_mkdir(conn, path1, &a);
700                 break;
701         case I_RMDIR:
702                 path1 = make_absolute(path1, *pwd);
703                 err = do_rmdir(conn, path1);
704                 break;
705         case I_CHDIR:
706                 path1 = make_absolute(path1, *pwd);
707                 if ((tmp = do_realpath(conn, path1)) == NULL) {
708                         err = 1;
709                         break;
710                 }
711                 if ((aa = do_stat(conn, tmp, 0)) == NULL) {
712                         xfree(tmp);
713                         err = 1;
714                         break;
715                 }
716                 if (!(aa->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)) {
717                         error("Can't change directory: Can't check target");
718                         xfree(tmp);
719                         err = 1;
720                         break;
721                 }
722                 if (!S_ISDIR(aa->perm)) {
723                         error("Can't change directory: \"%s\" is not "
724                             "a directory", tmp);
725                         xfree(tmp);
726                         err = 1;
727                         break;
728                 }
729                 xfree(*pwd);
730                 *pwd = tmp;
731                 break;
732         case I_LS:
733                 if (!path1) {
734                         do_ls(conn, *pwd);
735                         break;
736                 }
737                 path1 = make_absolute(path1, *pwd);
738                 if ((tmp = do_realpath(conn, path1)) == NULL)
739                         break;
740                 xfree(path1);
741                 path1 = tmp;
742                 if ((aa = do_stat(conn, path1, 0)) == NULL)
743                         break;
744                 if ((aa->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
745                     !S_ISDIR(aa->perm)) {
746                         error("Can't ls: \"%s\" is not a directory", path1);
747                         break;
748                 }
749                 do_ls(conn, path1);
750                 break;
751         case I_LCHDIR:
752                 if (chdir(path1) == -1) {
753                         error("Couldn't change local directory to "
754                             "\"%s\": %s", path1, strerror(errno));
755                         err = 1;
756                 }
757                 break;
758         case I_LMKDIR:
759                 if (mkdir(path1, 0777) == -1) {
760                         error("Couldn't create local directory "
761                             "\"%s\": %s", path1, strerror(errno));
762                         err = 1;
763                 }
764                 break;
765         case I_LLS:
766                 local_do_ls(cmd);
767                 break;
768         case I_SHELL:
769                 local_do_shell(cmd);
770                 break;
771         case I_LUMASK:
772                 umask(n_arg);
773                 printf("Local umask: %03lo\n", n_arg);
774                 break;
775         case I_CHMOD:
776                 path1 = make_absolute(path1, *pwd);
777                 attrib_clear(&a);
778                 a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
779                 a.perm = n_arg;
780                 remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
781                 for (i = 0; g.gl_pathv[i]; i++) {
782                         printf("Changing mode on %s\n", g.gl_pathv[i]);
783                         do_setstat(conn, g.gl_pathv[i], &a);
784                 }
785                 break;
786         case I_CHOWN:
787                 path1 = make_absolute(path1, *pwd);
788                 remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
789                 for (i = 0; g.gl_pathv[i]; i++) {
790                         if (!(aa = do_stat(conn, g.gl_pathv[i], 0)))
791                                 continue;
792                         if (!(aa->flags & SSH2_FILEXFER_ATTR_UIDGID)) {
793                                 error("Can't get current ownership of "
794                                     "remote file \"%s\"", g.gl_pathv[i]);
795                                 continue;
796                         }
797                         printf("Changing owner on %s\n", g.gl_pathv[i]);
798                         aa->flags &= SSH2_FILEXFER_ATTR_UIDGID;
799                         aa->uid = n_arg;
800                         do_setstat(conn, g.gl_pathv[i], aa);
801                 }
802                 break;
803         case I_CHGRP:
804                 path1 = make_absolute(path1, *pwd);
805                 remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
806                 for (i = 0; g.gl_pathv[i]; i++) {
807                         if (!(aa = do_stat(conn, g.gl_pathv[i], 0)))
808                                 continue;
809                         if (!(aa->flags & SSH2_FILEXFER_ATTR_UIDGID)) {
810                                 error("Can't get current ownership of "
811                                     "remote file \"%s\"", g.gl_pathv[i]);
812                                 continue;
813                         }
814                         printf("Changing group on %s\n", g.gl_pathv[i]);
815                         aa->flags &= SSH2_FILEXFER_ATTR_UIDGID;
816                         aa->gid = n_arg;
817                         do_setstat(conn, g.gl_pathv[i], aa);
818                 }
819                 break;
820         case I_PWD:
821                 printf("Remote working directory: %s\n", *pwd);
822                 break;
823         case I_LPWD:
824                 if (!getcwd(path_buf, sizeof(path_buf)))
825                         error("Couldn't get local cwd: %s",
826                             strerror(errno));
827                 else
828                         printf("Local working directory: %s\n",
829                             path_buf);
830                 break;
831         case I_QUIT:
832                 return(-1);
833         case I_HELP:
834                 help();
835                 break;
836         case I_VERSION:
837                 printf("SFTP protocol version %d\n", sftp_proto_version(conn));
838                 break;
839         default:
840                 fatal("%d is not implemented", cmdnum);
841         }
842
843         if (g.gl_pathc)
844                 globfree(&g);
845         if (path1)
846                 xfree(path1);
847         if (path2)
848                 xfree(path2);
849
850         /* If an error occurs in batch mode we should abort. */
851         if (infile != stdin && err > 0)
852                 return -1;
853
854         return(0);
855 }
856
857 void
858 interactive_loop(int fd_in, int fd_out, char *file1, char *file2)
859 {
860         char *pwd;
861         char *dir = NULL;
862         char cmd[2048];
863         struct sftp_conn *conn;
864
865         conn = do_init(fd_in, fd_out, copy_buffer_len, num_requests);
866         if (conn == NULL)
867                 fatal("Couldn't initialise connection to server");
868
869         pwd = do_realpath(conn, ".");
870         if (pwd == NULL)
871                 fatal("Need cwd");
872
873         if (file1 != NULL) {
874                 dir = xstrdup(file1);
875                 dir = make_absolute(dir, pwd);
876
877                 if (remote_is_dir(conn, dir) && file2 == NULL) {
878                         printf("Changing to: %s\n", dir);
879                         snprintf(cmd, sizeof cmd, "cd \"%s\"", dir);
880                         parse_dispatch_command(conn, cmd, &pwd);
881                 } else {
882                         if (file2 == NULL)
883                                 snprintf(cmd, sizeof cmd, "get %s", dir);
884                         else
885                                 snprintf(cmd, sizeof cmd, "get %s %s", dir,
886                                     file2);
887
888                         parse_dispatch_command(conn, cmd, &pwd);
889                         xfree(dir);
890                         return;
891                 }
892                 xfree(dir);
893         }
894 #if HAVE_SETVBUF
895         setvbuf(stdout, NULL, _IOLBF, 0);
896         setvbuf(infile, NULL, _IOLBF, 0);
897 #else
898         setlinebuf(stdout);
899         setlinebuf(infile);
900 #endif
901
902         for (;;) {
903                 char *cp;
904
905                 printf("sftp> ");
906
907                 /* XXX: use libedit */
908                 if (fgets(cmd, sizeof(cmd), infile) == NULL) {
909                         printf("\n");
910                         break;
911                 } else if (infile != stdin) /* Bluff typing */
912                         printf("%s", cmd);
913
914                 cp = strrchr(cmd, '\n');
915                 if (cp)
916                         *cp = '\0';
917
918                 if (parse_dispatch_command(conn, cmd, &pwd))
919                         break;
920         }
921         xfree(pwd);
922 }
This page took 0.102095 seconds and 3 git commands to generate.