X-Git-Url: https://git.distorted.org.uk/~mdw/sgt/putty/blobdiff_plain/65857773290a7bacd6f5d84ab51a9c1228e61e99..54018d95d47e4aff427c4e76f2d952c063ae077b:/windows/winsftp.c diff --git a/windows/winsftp.c b/windows/winsftp.c index c0712571..ff471dd5 100644 --- a/windows/winsftp.c +++ b/windows/winsftp.c @@ -306,20 +306,26 @@ DirHandle *open_directory(char *name) char *read_filename(DirHandle *dir) { - while (!dir->name) { - WIN32_FIND_DATA fdat; - int ok = FindNextFile(dir->h, &fdat); + do { - if (!ok) - return NULL; + if (!dir->name) { + WIN32_FIND_DATA fdat; + int ok = FindNextFile(dir->h, &fdat); + if (!ok) + return NULL; + else + dir->name = dupstr(fdat.cFileName); + } - if (fdat.cFileName[0] == '.' && - (fdat.cFileName[1] == '\0' || - (fdat.cFileName[1] == '.' && fdat.cFileName[2] == '\0'))) + assert(dir->name); + if (dir->name[0] == '.' && + (dir->name[1] == '\0' || + (dir->name[1] == '.' && dir->name[2] == '\0'))) { + sfree(dir->name); dir->name = NULL; - else - dir->name = dupstr(fdat.cFileName); - } + } + + } while (!dir->name); if (dir->name) { char *ret = dir->name; @@ -445,6 +451,17 @@ void finish_wildcard_matching(WildcardMatcher *dir) sfree(dir); } +int vet_filename(char *name) +{ + if (strchr(name, '/') || strchr(name, '\\') || strchr(name, ':')) + return FALSE; + + if (!name[strspn(name, ".")]) /* entirely composed of dots */ + return FALSE; + + return TRUE; +} + int create_directory(char *name) { return CreateDirectory(name, NULL) != 0;