+++ /dev/null
-/*\r
- * uxsftp.c: the Unix-specific parts of PSFTP and PSCP.\r
- */\r
-\r
-#include <sys/time.h>\r
-#include <sys/types.h>\r
-#include <sys/stat.h>\r
-#include <stdlib.h>\r
-#include <fcntl.h>\r
-#include <dirent.h>\r
-#include <unistd.h>\r
-#include <utime.h>\r
-#include <errno.h>\r
-#include <assert.h>\r
-#include <glob.h>\r
-#ifndef HAVE_NO_SYS_SELECT_H\r
-#include <sys/select.h>\r
-#endif\r
-\r
-#include "putty.h"\r
-#include "ssh.h"\r
-#include "psftp.h"\r
-#include "int64.h"\r
-\r
-/*\r
- * In PSFTP our selects are synchronous, so these functions are\r
- * empty stubs.\r
- */\r
-int uxsel_input_add(int fd, int rwx) { return 0; }\r
-void uxsel_input_remove(int id) { }\r
-\r
-char *x_get_default(const char *key)\r
-{\r
- return NULL; /* this is a stub */\r
-}\r
-\r
-void platform_get_x11_auth(struct X11Display *display, const Config *cfg)\r
-{\r
- /* Do nothing, therefore no auth. */\r
-}\r
-const int platform_uses_x11_unix_by_default = TRUE;\r
-\r
-/*\r
- * Default settings that are specific to PSFTP.\r
- */\r
-char *platform_default_s(const char *name)\r
-{\r
- return NULL;\r
-}\r
-\r
-int platform_default_i(const char *name, int def)\r
-{\r
- return def;\r
-}\r
-\r
-FontSpec platform_default_fontspec(const char *name)\r
-{\r
- FontSpec ret;\r
- *ret.name = '\0';\r
- return ret;\r
-}\r
-\r
-Filename platform_default_filename(const char *name)\r
-{\r
- Filename ret;\r
- if (!strcmp(name, "LogFileName"))\r
- strcpy(ret.path, "putty.log");\r
- else\r
- *ret.path = '\0';\r
- return ret;\r
-}\r
-\r
-char *get_ttymode(void *frontend, const char *mode) { return NULL; }\r
-\r
-int get_userpass_input(prompts_t *p, unsigned char *in, int inlen)\r
-{\r
- int ret;\r
- ret = cmdline_get_passwd_input(p, in, inlen);\r
- if (ret == -1)\r
- ret = console_get_userpass_input(p, in, inlen);\r
- return ret;\r
-}\r
-\r
-/*\r
- * Set local current directory. Returns NULL on success, or else an\r
- * error message which must be freed after printing.\r
- */\r
-char *psftp_lcd(char *dir)\r
-{\r
- if (chdir(dir) < 0)\r
- return dupprintf("%s: chdir: %s", dir, strerror(errno));\r
- else\r
- return NULL;\r
-}\r
-\r
-/*\r
- * Get local current directory. Returns a string which must be\r
- * freed.\r
- */\r
-char *psftp_getcwd(void)\r
-{\r
- char *buffer, *ret;\r
- int size = 256;\r
-\r
- buffer = snewn(size, char);\r
- while (1) {\r
- ret = getcwd(buffer, size);\r
- if (ret != NULL)\r
- return ret;\r
- if (errno != ERANGE) {\r
- sfree(buffer);\r
- return dupprintf("[cwd unavailable: %s]", strerror(errno));\r
- }\r
- /*\r
- * Otherwise, ERANGE was returned, meaning the buffer\r
- * wasn't big enough.\r
- */\r
- size = size * 3 / 2;\r
- buffer = sresize(buffer, size, char);\r
- }\r
-}\r
-\r
-struct RFile {\r
- int fd;\r
-};\r
-\r
-RFile *open_existing_file(char *name, uint64 *size,\r
- unsigned long *mtime, unsigned long *atime)\r
-{\r
- int fd;\r
- RFile *ret;\r
-\r
- fd = open(name, O_RDONLY);\r
- if (fd < 0)\r
- return NULL;\r
-\r
- ret = snew(RFile);\r
- ret->fd = fd;\r
-\r
- if (size || mtime || atime) {\r
- struct stat statbuf;\r
- if (fstat(fd, &statbuf) < 0) {\r
- fprintf(stderr, "%s: stat: %s\n", name, strerror(errno));\r
- memset(&statbuf, 0, sizeof(statbuf));\r
- }\r
-\r
- if (size)\r
- *size = uint64_make((statbuf.st_size >> 16) >> 16,\r
- statbuf.st_size);\r
- \r
- if (mtime)\r
- *mtime = statbuf.st_mtime;\r
-\r
- if (atime)\r
- *atime = statbuf.st_atime;\r
- }\r
-\r
- return ret;\r
-}\r
-\r
-int read_from_file(RFile *f, void *buffer, int length)\r
-{\r
- return read(f->fd, buffer, length);\r
-}\r
-\r
-void close_rfile(RFile *f)\r
-{\r
- close(f->fd);\r
- sfree(f);\r
-}\r
-\r
-struct WFile {\r
- int fd;\r
- char *name;\r
-};\r
-\r
-WFile *open_new_file(char *name)\r
-{\r
- int fd;\r
- WFile *ret;\r
-\r
- fd = open(name, O_CREAT | O_TRUNC | O_WRONLY, 0666);\r
- if (fd < 0)\r
- return NULL;\r
-\r
- ret = snew(WFile);\r
- ret->fd = fd;\r
- ret->name = dupstr(name);\r
-\r
- return ret;\r
-}\r
-\r
-\r
-WFile *open_existing_wfile(char *name, uint64 *size)\r
-{\r
- int fd;\r
- WFile *ret;\r
-\r
- fd = open(name, O_APPEND | O_WRONLY);\r
- if (fd < 0)\r
- return NULL;\r
-\r
- ret = snew(WFile);\r
- ret->fd = fd;\r
- ret->name = dupstr(name);\r
-\r
- if (size) {\r
- struct stat statbuf;\r
- if (fstat(fd, &statbuf) < 0) {\r
- fprintf(stderr, "%s: stat: %s\n", name, strerror(errno));\r
- memset(&statbuf, 0, sizeof(statbuf));\r
- }\r
-\r
- *size = uint64_make((statbuf.st_size >> 16) >> 16,\r
- statbuf.st_size);\r
- }\r
-\r
- return ret;\r
-}\r
-\r
-int write_to_file(WFile *f, void *buffer, int length)\r
-{\r
- char *p = (char *)buffer;\r
- int so_far = 0;\r
-\r
- /* Keep trying until we've really written as much as we can. */\r
- while (length > 0) {\r
- int ret = write(f->fd, p, length);\r
-\r
- if (ret < 0)\r
- return ret;\r
-\r
- if (ret == 0)\r
- break;\r
-\r
- p += ret;\r
- length -= ret;\r
- so_far += ret;\r
- }\r
-\r
- return so_far;\r
-}\r
-\r
-void set_file_times(WFile *f, unsigned long mtime, unsigned long atime)\r
-{\r
- struct utimbuf ut;\r
-\r
- ut.actime = atime;\r
- ut.modtime = mtime;\r
-\r
- utime(f->name, &ut);\r
-}\r
-\r
-/* Closes and frees the WFile */\r
-void close_wfile(WFile *f)\r
-{\r
- close(f->fd);\r
- sfree(f->name);\r
- sfree(f);\r
-}\r
-\r
-/* Seek offset bytes through file, from whence, where whence is\r
- FROM_START, FROM_CURRENT, or FROM_END */\r
-int seek_file(WFile *f, uint64 offset, int whence)\r
-{\r
- off_t fileofft;\r
- int lseek_whence;\r
- \r
- fileofft = (((off_t) offset.hi << 16) << 16) + offset.lo;\r
-\r
- switch (whence) {\r
- case FROM_START:\r
- lseek_whence = SEEK_SET;\r
- break;\r
- case FROM_CURRENT:\r
- lseek_whence = SEEK_CUR;\r
- break;\r
- case FROM_END:\r
- lseek_whence = SEEK_END;\r
- break;\r
- default:\r
- return -1;\r
- }\r
-\r
- return lseek(f->fd, fileofft, lseek_whence) >= 0 ? 0 : -1;\r
-}\r
-\r
-uint64 get_file_posn(WFile *f)\r
-{\r
- off_t fileofft;\r
- uint64 ret;\r
-\r
- fileofft = lseek(f->fd, (off_t) 0, SEEK_CUR);\r
-\r
- ret = uint64_make((fileofft >> 16) >> 16, fileofft);\r
-\r
- return ret;\r
-}\r
-\r
-int file_type(char *name)\r
-{\r
- struct stat statbuf;\r
-\r
- if (stat(name, &statbuf) < 0) {\r
- if (errno != ENOENT)\r
- fprintf(stderr, "%s: stat: %s\n", name, strerror(errno));\r
- return FILE_TYPE_NONEXISTENT;\r
- }\r
-\r
- if (S_ISREG(statbuf.st_mode))\r
- return FILE_TYPE_FILE;\r
-\r
- if (S_ISDIR(statbuf.st_mode))\r
- return FILE_TYPE_DIRECTORY;\r
-\r
- return FILE_TYPE_WEIRD;\r
-}\r
-\r
-struct DirHandle {\r
- DIR *dir;\r
-};\r
-\r
-DirHandle *open_directory(char *name)\r
-{\r
- DIR *dir;\r
- DirHandle *ret;\r
-\r
- dir = opendir(name);\r
- if (!dir)\r
- return NULL;\r
-\r
- ret = snew(DirHandle);\r
- ret->dir = dir;\r
- return ret;\r
-}\r
-\r
-char *read_filename(DirHandle *dir)\r
-{\r
- struct dirent *de;\r
-\r
- do {\r
- de = readdir(dir->dir);\r
- if (de == NULL)\r
- return NULL;\r
- } while ((de->d_name[0] == '.' &&\r
- (de->d_name[1] == '\0' ||\r
- (de->d_name[1] == '.' && de->d_name[2] == '\0'))));\r
-\r
- return dupstr(de->d_name);\r
-}\r
-\r
-void close_directory(DirHandle *dir)\r
-{\r
- closedir(dir->dir);\r
- sfree(dir);\r
-}\r
-\r
-int test_wildcard(char *name, int cmdline)\r
-{\r
- struct stat statbuf;\r
-\r
- if (stat(name, &statbuf) == 0) {\r
- return WCTYPE_FILENAME;\r
- } else if (cmdline) {\r
- /*\r
- * On Unix, we never need to parse wildcards coming from\r
- * the command line, because the shell will have expanded\r
- * them into a filename list already.\r
- */\r
- return WCTYPE_NONEXISTENT;\r
- } else {\r
- glob_t globbed;\r
- int ret = WCTYPE_NONEXISTENT;\r
-\r
- if (glob(name, GLOB_ERR, NULL, &globbed) == 0) {\r
- if (globbed.gl_pathc > 0)\r
- ret = WCTYPE_WILDCARD;\r
- globfree(&globbed);\r
- }\r
-\r
- return ret;\r
- }\r
-}\r
-\r
-/*\r
- * Actually return matching file names for a local wildcard.\r
- */\r
-struct WildcardMatcher {\r
- glob_t globbed;\r
- int i;\r
-};\r
-WildcardMatcher *begin_wildcard_matching(char *name) {\r
- WildcardMatcher *ret = snew(WildcardMatcher);\r
-\r
- if (glob(name, 0, NULL, &ret->globbed) < 0) {\r
- sfree(ret);\r
- return NULL;\r
- }\r
-\r
- ret->i = 0;\r
-\r
- return ret;\r
-}\r
-char *wildcard_get_filename(WildcardMatcher *dir) {\r
- if (dir->i < dir->globbed.gl_pathc) {\r
- return dupstr(dir->globbed.gl_pathv[dir->i++]);\r
- } else\r
- return NULL;\r
-}\r
-void finish_wildcard_matching(WildcardMatcher *dir) {\r
- globfree(&dir->globbed);\r
- sfree(dir);\r
-}\r
-\r
-int vet_filename(char *name)\r
-{\r
- if (strchr(name, '/'))\r
- return FALSE;\r
-\r
- if (name[0] == '.' && (!name[1] || (name[1] == '.' && !name[2])))\r
- return FALSE;\r
-\r
- return TRUE;\r
-}\r
-\r
-int create_directory(char *name)\r
-{\r
- return mkdir(name, 0777) == 0;\r
-}\r
-\r
-char *dir_file_cat(char *dir, char *file)\r
-{\r
- return dupcat(dir, "/", file, NULL);\r
-}\r
-\r
-/*\r
- * Do a select() between all currently active network fds and\r
- * optionally stdin.\r
- */\r
-static int ssh_sftp_do_select(int include_stdin, int no_fds_ok)\r
-{\r
- fd_set rset, wset, xset;\r
- int i, fdcount, fdsize, *fdlist;\r
- int fd, fdstate, rwx, ret, maxfd;\r
- long now = GETTICKCOUNT();\r
-\r
- fdlist = NULL;\r
- fdcount = fdsize = 0;\r
-\r
- do {\r
-\r
- /* Count the currently active fds. */\r
- i = 0;\r
- for (fd = first_fd(&fdstate, &rwx); fd >= 0;\r
- fd = next_fd(&fdstate, &rwx)) i++;\r
-\r
- if (i < 1 && !no_fds_ok)\r
- return -1; /* doom */\r
-\r
- /* Expand the fdlist buffer if necessary. */\r
- if (i > fdsize) {\r
- fdsize = i + 16;\r
- fdlist = sresize(fdlist, fdsize, int);\r
- }\r
-\r
- FD_ZERO(&rset);\r
- FD_ZERO(&wset);\r
- FD_ZERO(&xset);\r
- maxfd = 0;\r
-\r
- /*\r
- * Add all currently open fds to the select sets, and store\r
- * them in fdlist as well.\r
- */\r
- fdcount = 0;\r
- for (fd = first_fd(&fdstate, &rwx); fd >= 0;\r
- fd = next_fd(&fdstate, &rwx)) {\r
- fdlist[fdcount++] = fd;\r
- if (rwx & 1)\r
- FD_SET_MAX(fd, maxfd, rset);\r
- if (rwx & 2)\r
- FD_SET_MAX(fd, maxfd, wset);\r
- if (rwx & 4)\r
- FD_SET_MAX(fd, maxfd, xset);\r
- }\r
-\r
- if (include_stdin)\r
- FD_SET_MAX(0, maxfd, rset);\r
-\r
- do {\r
- long next, ticks;\r
- struct timeval tv, *ptv;\r
-\r
- if (run_timers(now, &next)) {\r
- ticks = next - GETTICKCOUNT();\r
- if (ticks <= 0)\r
- ticks = 1; /* just in case */\r
- tv.tv_sec = ticks / 1000;\r
- tv.tv_usec = ticks % 1000 * 1000;\r
- ptv = &tv;\r
- } else {\r
- ptv = NULL;\r
- }\r
- ret = select(maxfd, &rset, &wset, &xset, ptv);\r
- if (ret == 0)\r
- now = next;\r
- else {\r
- long newnow = GETTICKCOUNT();\r
- /*\r
- * Check to see whether the system clock has\r
- * changed massively during the select.\r
- */\r
- if (newnow - now < 0 || newnow - now > next - now) {\r
- /*\r
- * If so, look at the elapsed time in the\r
- * select and use it to compute a new\r
- * tickcount_offset.\r
- */\r
- long othernow = now + tv.tv_sec * 1000 + tv.tv_usec / 1000;\r
- /* So we'd like GETTICKCOUNT to have returned othernow,\r
- * but instead it return newnow. Hence ... */\r
- tickcount_offset += othernow - newnow;\r
- now = othernow;\r
- } else {\r
- now = newnow;\r
- }\r
- }\r
- } while (ret < 0 && errno != EINTR);\r
- } while (ret == 0);\r
-\r
- if (ret < 0) {\r
- perror("select");\r
- exit(1);\r
- }\r
-\r
- for (i = 0; i < fdcount; i++) {\r
- fd = fdlist[i];\r
- /*\r
- * We must process exceptional notifications before\r
- * ordinary readability ones, or we may go straight\r
- * past the urgent marker.\r
- */\r
- if (FD_ISSET(fd, &xset))\r
- select_result(fd, 4);\r
- if (FD_ISSET(fd, &rset))\r
- select_result(fd, 1);\r
- if (FD_ISSET(fd, &wset))\r
- select_result(fd, 2);\r
- }\r
-\r
- sfree(fdlist);\r
-\r
- return FD_ISSET(0, &rset) ? 1 : 0;\r
-}\r
-\r
-/*\r
- * Wait for some network data and process it.\r
- */\r
-int ssh_sftp_loop_iteration(void)\r
-{\r
- return ssh_sftp_do_select(FALSE, FALSE);\r
-}\r
-\r
-/*\r
- * Read a PSFTP command line from stdin.\r
- */\r
-char *ssh_sftp_get_cmdline(char *prompt, int no_fds_ok)\r
-{\r
- char *buf;\r
- int buflen, bufsize, ret;\r
-\r
- fputs(prompt, stdout);\r
- fflush(stdout);\r
-\r
- buf = NULL;\r
- buflen = bufsize = 0;\r
-\r
- while (1) {\r
- ret = ssh_sftp_do_select(TRUE, no_fds_ok);\r
- if (ret < 0) {\r
- printf("connection died\n");\r
- return NULL; /* woop woop */\r
- }\r
- if (ret > 0) {\r
- if (buflen >= bufsize) {\r
- bufsize = buflen + 512;\r
- buf = sresize(buf, bufsize, char);\r
- }\r
- ret = read(0, buf+buflen, 1);\r
- if (ret < 0) {\r
- perror("read");\r
- return NULL;\r
- }\r
- if (ret == 0) {\r
- /* eof on stdin; no error, but no answer either */\r
- return NULL;\r
- }\r
-\r
- if (buf[buflen++] == '\n') {\r
- /* we have a full line */\r
- return buf;\r
- }\r
- }\r
- }\r
-}\r
-\r
-/*\r
- * Main program: do platform-specific initialisation and then call\r
- * psftp_main().\r
- */\r
-int main(int argc, char *argv[])\r
-{\r
- uxsel_init();\r
- return psftp_main(argc, argv);\r
-}\r
OSDN Git Service
