1 files changed, 29 insertions, 8 deletions

diff --git a/scp.c b/scp.c
index 1971c80cd..035037bcc 100644
--- a/scp.c
+++ b/scp.c
@@ -94,6 +94,7 @@
 #include <dirent.h>
 #include <errno.h>
 #include <fcntl.h>
+#include <fnmatch.h>
 #include <limits.h>
 #include <locale.h>
 #include <pwd.h>
@@ -383,14 +384,14 @@ void verifydir(char *);
 struct passwd *pwd;
 uid_t userid;
 int errs, remin, remout;
-int pflag, iamremote, iamrecursive, targetshouldbedirectory;
+int Tflag, pflag, iamremote, iamrecursive, targetshouldbedirectory;
 
 #define CMDNEEDS        64
 char cmd[CMDNEEDS];             /* must hold "rcp -r -p -d\0" */
 
 int response(void);
 void rsource(char *, struct stat *);
-void sink(int, char *[]);
+void sink(int, char *[], const char *);
 void source(int, char *[]);
 void tolocal(int, char *[]);
 void toremote(int, char *[]);
@@ -429,8 +430,9 @@ main(int argc, char **argv)
         addargs(&args, "-oRemoteCommand=none");
         addargs(&args, "-oRequestTTY=no");
 
-        fflag = tflag = 0;
-        while ((ch = getopt(argc, argv, "dfl:prtvBCc:i:P:q12346S:o:F:")) != -1)
+        fflag = Tflag = tflag = 0;
+        while ((ch = getopt(argc, argv,
+            "dfl:prtTvBCc:i:P:q12346S:o:F:")) != -1) {
                 switch (ch) {
                 /* User-visible flags. */
                 case '1':
@@ -509,9 +511,13 @@ main(int argc, char **argv)
                         setmode(0, O_BINARY);
 #endif
                         break;
+                case 'T':
+                        Tflag = 1;
+                        break;
                 default:
                         usage();
                 }
+        }
         argc -= optind;
         argv += optind;
 
@@ -542,7 +548,7 @@ main(int argc, char **argv)
         }
         if (tflag) {
                 /* Receive data. */
-                sink(argc, argv);
+                sink(argc, argv, NULL);
                 exit(errs != 0);
         }
         if (argc < 2)
@@ -800,7 +806,7 @@ tolocal(int argc, char **argv)
                         continue;
                 }
                 free(bp);
-                sink(1, argv + argc - 1);
+                sink(1, argv + argc - 1, src);
                 (void) close(remin);
                 remin = remout = -1;
         }
@@ -976,7 +982,7 @@ rsource(char *name, struct stat *statp)
          (sizeof(type) != 4 && sizeof(type) != 8))
 
 void
-sink(int argc, char **argv)
+sink(int argc, char **argv, const char *src)
 {
         static BUF buffer;
         struct stat stb;
@@ -992,6 +998,7 @@ sink(int argc, char **argv)
         unsigned long long ull;
         int setimes, targisdir, wrerrno = 0;
         char ch, *cp, *np, *targ, *why, *vect[1], buf[2048], visbuf[2048];
+        char *src_copy = NULL, *restrict_pattern = NULL;
         struct timeval tv[2];
 
 #define atime   tv[0]
@@ -1016,6 +1023,17 @@ sink(int argc, char **argv)
         (void) atomicio(vwrite, remout, "", 1);
         if (stat(targ, &stb) == 0 && S_ISDIR(stb.st_mode))
                 targisdir = 1;
+        if (src != NULL && !iamrecursive && !Tflag) {
+                /*
+                 * Prepare to try to restrict incoming filenames to match
+                 * the requested destination file glob.
+                 */
+                if ((src_copy = strdup(src)) == NULL)
+                        fatal("strdup failed");
+                if ((restrict_pattern = strrchr(src_copy, '/')) != NULL) {
+                        *restrict_pattern++ = '\0';
+                }
+        }
         for (first = 1;; first = 0) {
                 cp = buf;
                 if (atomicio(read, remin, cp, 1) != 1)
@@ -1120,6 +1138,9 @@ sink(int argc, char **argv)
                         run_err("error: unexpected filename: %s", cp);
                         exit(1);
                 }
+                if (restrict_pattern != NULL &&
+                    fnmatch(restrict_pattern, cp, 0) != 0)
+                        SCREWUP("filename does not match request");
                 if (targisdir) {
                         static char *namebuf;
                         static size_t cursize;
@@ -1157,7 +1178,7 @@ sink(int argc, char **argv)
                                         goto bad;
                         }
                         vect[0] = xstrdup(np);
-                        sink(1, vect);
+                        sink(1, vect, src);
                         if (setimes) {
                                 setimes = 0;
                                 if (utimes(vect[0], tv) < 0)