diff options
author | yugui <yugui@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-08-25 15:13:14 +0000 |
---|---|---|
committer | yugui <yugui@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-08-25 15:13:14 +0000 |
commit | d0233291bc8a5068e52c69c210e5979e5324b5bc (patch) | |
tree | 7d9459449c33792c63eeb7baa071e76352e0baab /trunk/win32/win32.c | |
parent | 0dc342de848a642ecce8db697b8fecd83a63e117 (diff) | |
parent | 72eaacaa15256ab95c3b52ea386f88586fb9da40 (diff) |
re-adding tag v1_9_0_4 as an alias of trunk@18848v1_9_0_4
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/tags/v1_9_0_4@18849 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'trunk/win32/win32.c')
-rw-r--r-- | trunk/win32/win32.c | 4279 |
1 files changed, 0 insertions, 4279 deletions
diff --git a/trunk/win32/win32.c b/trunk/win32/win32.c deleted file mode 100644 index 04a11f3fb8..0000000000 --- a/trunk/win32/win32.c +++ /dev/null @@ -1,4279 +0,0 @@ -/* - * Copyright (c) 1993, Intergraph Corporation - * - * You may distribute under the terms of either the GNU General Public - * License or the Artistic License, as specified in the perl README file. - * - * Various Unix compatibility functions and NT specific functions. - * - * Some of this code was derived from the MSDOS port(s) and the OS/2 port. - * - */ - -#include "ruby/ruby.h" -#include "ruby/signal.h" -#include "dln.h" -#include <fcntl.h> -#include <process.h> -#include <sys/stat.h> -/* #include <sys/wait.h> */ -#include <stdio.h> -#include <stdlib.h> -#include <errno.h> -#include <assert.h> -#include <ctype.h> - -#include <windows.h> -#include <winbase.h> -#include <wincon.h> -#include <share.h> -#include <shlobj.h> -#include <mbstring.h> -#ifdef __MINGW32__ -#include <mswsock.h> -#endif -#include "ruby/win32.h" -#include "win32/dir.h" -#ifdef _WIN32_WCE -#include "wince.h" -#endif -#ifndef index -#define index(x, y) strchr((x), (y)) -#endif -#define isdirsep(x) ((x) == '/' || (x) == '\\') - -#undef stat -#undef fclose -#undef close -#undef setsockopt - -#if defined __BORLANDC__ || defined _WIN32_WCE -# define _filbuf _fgetc -# define _flsbuf _fputc -# define enough_to_get(n) (--(n) >= 0) -# define enough_to_put(n) (++(n) < 0) -#else -# define enough_to_get(n) (--(n) >= 0) -# define enough_to_put(n) (--(n) >= 0) -#endif - -#ifdef WIN32_DEBUG -#define Debug(something) something -#else -#define Debug(something) /* nothing */ -#endif - -#define TO_SOCKET(x) _get_osfhandle(x) - -static struct ChildRecord *CreateChild(const char *, const char *, SECURITY_ATTRIBUTES *, HANDLE, HANDLE, HANDLE); -static int has_redirection(const char *); -int rb_w32_wait_events(HANDLE *events, int num, DWORD timeout); -#if !defined(_WIN32_WCE) -static int rb_w32_open_osfhandle(intptr_t osfhandle, int flags); -#else -#define rb_w32_open_osfhandle(osfhandle, flags) _open_osfhandle(osfhandle, flags) -#endif - -/* errno mapping */ -static struct { - DWORD winerr; - int err; -} errmap[] = { - { ERROR_INVALID_FUNCTION, EINVAL }, - { ERROR_FILE_NOT_FOUND, ENOENT }, - { ERROR_PATH_NOT_FOUND, ENOENT }, - { ERROR_TOO_MANY_OPEN_FILES, EMFILE }, - { ERROR_ACCESS_DENIED, EACCES }, - { ERROR_INVALID_HANDLE, EBADF }, - { ERROR_ARENA_TRASHED, ENOMEM }, - { ERROR_NOT_ENOUGH_MEMORY, ENOMEM }, - { ERROR_INVALID_BLOCK, ENOMEM }, - { ERROR_BAD_ENVIRONMENT, E2BIG }, - { ERROR_BAD_FORMAT, ENOEXEC }, - { ERROR_INVALID_ACCESS, EINVAL }, - { ERROR_INVALID_DATA, EINVAL }, - { ERROR_INVALID_DRIVE, ENOENT }, - { ERROR_CURRENT_DIRECTORY, EACCES }, - { ERROR_NOT_SAME_DEVICE, EXDEV }, - { ERROR_NO_MORE_FILES, ENOENT }, - { ERROR_WRITE_PROTECT, EROFS }, - { ERROR_BAD_UNIT, ENODEV }, - { ERROR_NOT_READY, ENXIO }, - { ERROR_BAD_COMMAND, EACCES }, - { ERROR_CRC, EACCES }, - { ERROR_BAD_LENGTH, EACCES }, - { ERROR_SEEK, EIO }, - { ERROR_NOT_DOS_DISK, EACCES }, - { ERROR_SECTOR_NOT_FOUND, EACCES }, - { ERROR_OUT_OF_PAPER, EACCES }, - { ERROR_WRITE_FAULT, EIO }, - { ERROR_READ_FAULT, EIO }, - { ERROR_GEN_FAILURE, EACCES }, - { ERROR_LOCK_VIOLATION, EACCES }, - { ERROR_SHARING_VIOLATION, EACCES }, - { ERROR_WRONG_DISK, EACCES }, - { ERROR_SHARING_BUFFER_EXCEEDED, EACCES }, - { ERROR_BAD_NETPATH, ENOENT }, - { ERROR_NETWORK_ACCESS_DENIED, EACCES }, - { ERROR_BAD_NET_NAME, ENOENT }, - { ERROR_FILE_EXISTS, EEXIST }, - { ERROR_CANNOT_MAKE, EACCES }, - { ERROR_FAIL_I24, EACCES }, - { ERROR_INVALID_PARAMETER, EINVAL }, - { ERROR_NO_PROC_SLOTS, EAGAIN }, - { ERROR_DRIVE_LOCKED, EACCES }, - { ERROR_BROKEN_PIPE, EPIPE }, - { ERROR_DISK_FULL, ENOSPC }, - { ERROR_INVALID_TARGET_HANDLE, EBADF }, - { ERROR_INVALID_HANDLE, EINVAL }, - { ERROR_WAIT_NO_CHILDREN, ECHILD }, - { ERROR_CHILD_NOT_COMPLETE, ECHILD }, - { ERROR_DIRECT_ACCESS_HANDLE, EBADF }, - { ERROR_NEGATIVE_SEEK, EINVAL }, - { ERROR_SEEK_ON_DEVICE, EACCES }, - { ERROR_DIR_NOT_EMPTY, ENOTEMPTY }, - { ERROR_DIRECTORY, ENOTDIR }, - { ERROR_NOT_LOCKED, EACCES }, - { ERROR_BAD_PATHNAME, ENOENT }, - { ERROR_MAX_THRDS_REACHED, EAGAIN }, - { ERROR_LOCK_FAILED, EACCES }, - { ERROR_ALREADY_EXISTS, EEXIST }, - { ERROR_INVALID_STARTING_CODESEG, ENOEXEC }, - { ERROR_INVALID_STACKSEG, ENOEXEC }, - { ERROR_INVALID_MODULETYPE, ENOEXEC }, - { ERROR_INVALID_EXE_SIGNATURE, ENOEXEC }, - { ERROR_EXE_MARKED_INVALID, ENOEXEC }, - { ERROR_BAD_EXE_FORMAT, ENOEXEC }, - { ERROR_ITERATED_DATA_EXCEEDS_64k,ENOEXEC }, - { ERROR_INVALID_MINALLOCSIZE, ENOEXEC }, - { ERROR_DYNLINK_FROM_INVALID_RING,ENOEXEC }, - { ERROR_IOPL_NOT_ENABLED, ENOEXEC }, - { ERROR_INVALID_SEGDPL, ENOEXEC }, - { ERROR_AUTODATASEG_EXCEEDS_64k, ENOEXEC }, - { ERROR_RING2SEG_MUST_BE_MOVABLE, ENOEXEC }, - { ERROR_RELOC_CHAIN_XEEDS_SEGLIM, ENOEXEC }, - { ERROR_INFLOOP_IN_RELOC_CHAIN, ENOEXEC }, - { ERROR_FILENAME_EXCED_RANGE, ENOENT }, - { ERROR_NESTING_NOT_ALLOWED, EAGAIN }, -#ifndef ERROR_PIPE_LOCAL -#define ERROR_PIPE_LOCAL 229L -#endif - { ERROR_PIPE_LOCAL, EPIPE }, - { ERROR_BAD_PIPE, EPIPE }, - { ERROR_PIPE_BUSY, EAGAIN }, - { ERROR_NO_DATA, EPIPE }, - { ERROR_PIPE_NOT_CONNECTED, EPIPE }, - { ERROR_OPERATION_ABORTED, EINTR }, - { ERROR_NOT_ENOUGH_QUOTA, ENOMEM }, - { WSAENAMETOOLONG, ENAMETOOLONG }, - { WSAENOTEMPTY, ENOTEMPTY }, - { WSAEINTR, EINTR }, - { WSAEBADF, EBADF }, - { WSAEACCES, EACCES }, - { WSAEFAULT, EFAULT }, - { WSAEINVAL, EINVAL }, - { WSAEMFILE, EMFILE }, -}; - -int -rb_w32_map_errno(DWORD winerr) -{ - int i; - - if (winerr == 0) { - return 0; - } - - for (i = 0; i < sizeof(errmap) / sizeof(*errmap); i++) { - if (errmap[i].winerr == winerr) { - return errmap[i].err; - } - } - - if (winerr >= WSABASEERR) { - return winerr; - } - return EINVAL; -} - -#define map_errno rb_w32_map_errno - -static const char *NTLoginName; - -static OSVERSIONINFO osver; -#ifdef WIN95 -static DWORD Win32System = (DWORD)-1; - -DWORD -rb_w32_osid(void) -{ - if (osver.dwPlatformId != Win32System) { - memset(&osver, 0, sizeof(OSVERSIONINFO)); - osver.dwOSVersionInfoSize = sizeof(OSVERSIONINFO); - GetVersionEx(&osver); - Win32System = osver.dwPlatformId; - } - return (Win32System); -} -#endif -static DWORD Win32Version = (DWORD)-1; - -static DWORD -rb_w32_osver(void) -{ - if (osver.dwMajorVersion != Win32Version) { - memset(&osver, 0, sizeof(OSVERSIONINFO)); - osver.dwOSVersionInfoSize = sizeof(OSVERSIONINFO); - GetVersionEx(&osver); - Win32Version = osver.dwMajorVersion; - } - return (Win32Version); -} - -#define IsWinNT() rb_w32_iswinnt() -#define IsWin95() rb_w32_iswin95() - -HANDLE -GetCurrentThreadHandle(void) -{ - static HANDLE current_process_handle = NULL; - HANDLE h; - - if (!current_process_handle) - current_process_handle = GetCurrentProcess(); - if (!DuplicateHandle(current_process_handle, GetCurrentThread(), - current_process_handle, &h, - 0, FALSE, DUPLICATE_SAME_ACCESS)) - return NULL; - return h; -} - -/* simulate flock by locking a range on the file */ - - -#define LK_ERR(f,i) \ - do { \ - if (f) \ - i = 0; \ - else { \ - DWORD err = GetLastError(); \ - if (err == ERROR_LOCK_VIOLATION) \ - errno = EWOULDBLOCK; \ - else if (err == ERROR_NOT_LOCKED) \ - i = 0; \ - else \ - errno = map_errno(err); \ - } \ - } while (0) -#define LK_LEN ULONG_MAX - -static uintptr_t -flock_winnt(uintptr_t self, int argc, uintptr_t* argv) -{ - OVERLAPPED o; - int i = -1; - const HANDLE fh = (HANDLE)self; - const int oper = argc; - - memset(&o, 0, sizeof(o)); - - switch(oper) { - case LOCK_SH: /* shared lock */ - LK_ERR(LockFileEx(fh, 0, 0, LK_LEN, LK_LEN, &o), i); - break; - case LOCK_EX: /* exclusive lock */ - LK_ERR(LockFileEx(fh, LOCKFILE_EXCLUSIVE_LOCK, 0, LK_LEN, LK_LEN, &o), i); - break; - case LOCK_SH|LOCK_NB: /* non-blocking shared lock */ - LK_ERR(LockFileEx(fh, LOCKFILE_FAIL_IMMEDIATELY, 0, LK_LEN, LK_LEN, &o), i); - break; - case LOCK_EX|LOCK_NB: /* non-blocking exclusive lock */ - LK_ERR(LockFileEx(fh, - LOCKFILE_EXCLUSIVE_LOCK|LOCKFILE_FAIL_IMMEDIATELY, - 0, LK_LEN, LK_LEN, &o), i); - break; - case LOCK_UN: /* unlock lock */ - case LOCK_UN|LOCK_NB: /* unlock is always non-blocking, I hope */ - LK_ERR(UnlockFileEx(fh, 0, LK_LEN, LK_LEN, &o), i); - break; - default: /* unknown */ - errno = EINVAL; - break; - } - return i; -} - -#ifdef WIN95 -static uintptr_t -flock_win95(uintptr_t self, int argc, uintptr_t* argv) -{ - int i = -1; - const HANDLE fh = (HANDLE)self; - const int oper = argc; - - switch(oper) { - case LOCK_EX: - do { - LK_ERR(LockFile(fh, 0, 0, LK_LEN, LK_LEN), i); - } while (i && errno == EWOULDBLOCK); - break; - case LOCK_EX|LOCK_NB: - LK_ERR(LockFile(fh, 0, 0, LK_LEN, LK_LEN), i); - break; - case LOCK_UN: - case LOCK_UN|LOCK_NB: - LK_ERR(UnlockFile(fh, 0, 0, LK_LEN, LK_LEN), i); - break; - default: - errno = EINVAL; - break; - } - return i; -} -#endif - -#undef LK_ERR - -int -flock(int fd, int oper) -{ -#ifdef WIN95 - static asynchronous_func_t locker = NULL; - - if (!locker) { - if (IsWinNT()) - locker = flock_winnt; - else - locker = flock_win95; - } -#else - const asynchronous_func_t locker = flock_winnt; -#endif - - return rb_w32_asynchronize(locker, - (VALUE)_get_osfhandle(fd), oper, NULL, - (DWORD)-1); -} - -static void -init_env(void) -{ - char env[_MAX_PATH]; - DWORD len; - BOOL f; - LPITEMIDLIST pidl; - - if (!GetEnvironmentVariable("HOME", env, sizeof(env))) { - f = FALSE; - if (GetEnvironmentVariable("HOMEDRIVE", env, sizeof(env))) - len = strlen(env); - else - len = 0; - if (GetEnvironmentVariable("HOMEPATH", env + len, sizeof(env) - len) || len) { - f = TRUE; - } - else if (GetEnvironmentVariable("USERPROFILE", env, sizeof(env))) { - f = TRUE; - } - else if (SHGetSpecialFolderLocation(NULL, CSIDL_PERSONAL, &pidl) == 0) { - LPMALLOC alloc; - f = SHGetPathFromIDList(pidl, env); - SHGetMalloc(&alloc); - alloc->lpVtbl->Free(alloc, pidl); - alloc->lpVtbl->Release(alloc); - } - if (f) { - char *p = env; - while (*p) { - if (*p == '\\') *p = '/'; - p = CharNext(p); - } - if (p - env == 2 && env[1] == ':') { - *p++ = '/'; - *p = 0; - } - SetEnvironmentVariable("HOME", env); - } - } - - if (!GetEnvironmentVariable("USER", env, sizeof env)) { - if (GetEnvironmentVariable("USERNAME", env, sizeof env)) { - SetEnvironmentVariable("USER", env); - } - else if (!GetUserName(env, (len = sizeof env, &len))) { - NTLoginName = "<Unknown>"; - return; - } - } - NTLoginName = strdup(env); -} - - -typedef BOOL (WINAPI *cancel_io_t)(HANDLE); -static cancel_io_t cancel_io = NULL; - -static void -init_func(void) -{ - if (!cancel_io) - cancel_io = (cancel_io_t)GetProcAddress(GetModuleHandle("kernel32"), - "CancelIo"); -} - -static void init_stdhandle(void); - -#if _MSC_VER >= 1400 -static void invalid_parameter(const wchar_t *expr, const wchar_t *func, const wchar_t *file, unsigned int line, uintptr_t dummy) -{ - // nothing to do -} -#endif - -static CRITICAL_SECTION select_mutex; -static int NtSocketsInitialized = 0; -static st_table *socklist = NULL; -static char *envarea; - -static void -exit_handler(void) -{ - if (NtSocketsInitialized) { - WSACleanup(); - st_free_table(socklist); - socklist = NULL; - NtSocketsInitialized = 0; - } - if (envarea) { - FreeEnvironmentStrings(envarea); - envarea = NULL; - } - DeleteCriticalSection(&select_mutex); -} - -static void -StartSockets(void) -{ - WORD version; - WSADATA retdata; - - // - // initalize the winsock interface and insure that it's - // cleaned up at exit. - // - version = MAKEWORD(2, 0); - if (WSAStartup(version, &retdata)) - rb_fatal ("Unable to locate winsock library!\n"); - if (LOBYTE(retdata.wVersion) != 2) - rb_fatal("could not find version 2 of winsock dll\n"); - - socklist = st_init_numtable(); - - NtSocketsInitialized = 1; -} - -// -// Initialization stuff -// -void -rb_w32_sysinit(int *argc, char ***argv) -{ -#if _MSC_VER >= 1400 - static void set_pioinfo_extra(void); - - _set_invalid_parameter_handler(invalid_parameter); - set_pioinfo_extra(); -#endif - - // - // subvert cmd.exe's feeble attempt at command line parsing - // - *argc = rb_w32_cmdvector(GetCommandLine(), argv); - - // - // Now set up the correct time stuff - // - - tzset(); - - init_env(); - - init_func(); - - init_stdhandle(); - - InitializeCriticalSection(&select_mutex); - - atexit(exit_handler); - - // Initialize Winsock - StartSockets(); - -#ifdef _WIN32_WCE - // free commandline buffer - wce_FreeCommandLine(); -#endif -} - -char * -getlogin(void) -{ - return (char *)NTLoginName; -} - -#define MAXCHILDNUM 256 /* max num of child processes */ - -static struct ChildRecord { - HANDLE hProcess; /* process handle */ - rb_pid_t pid; /* process id */ -} ChildRecord[MAXCHILDNUM]; - -#define FOREACH_CHILD(v) do { \ - struct ChildRecord* v; \ - for (v = ChildRecord; v < ChildRecord + sizeof(ChildRecord) / sizeof(ChildRecord[0]); ++v) -#define END_FOREACH_CHILD } while (0) - -static struct ChildRecord * -FindChildSlot(rb_pid_t pid) -{ - - FOREACH_CHILD(child) { - if (child->pid == pid) { - return child; - } - } END_FOREACH_CHILD; - return NULL; -} - -static void -CloseChildHandle(struct ChildRecord *child) -{ - HANDLE h = child->hProcess; - child->hProcess = NULL; - child->pid = 0; - CloseHandle(h); -} - -static struct ChildRecord * -FindFreeChildSlot(void) -{ - FOREACH_CHILD(child) { - if (!child->pid) { - child->pid = -1; /* lock the slot */ - child->hProcess = NULL; - return child; - } - } END_FOREACH_CHILD; - return NULL; -} - - -/* - ruby -lne 'BEGIN{$cmds = Hash.new(0); $mask = 1}' - -e '$cmds[$_.downcase] |= $mask' -e '$mask <<= 1 if ARGF.eof' - -e 'END{$cmds.sort.each{|n,f|puts " \"\\#{f.to_s(8)}\" #{n.dump} + 1,"}}' - 98cmd ntcmd - */ -static const char *const szInternalCmds[] = { - "\2" "assoc" + 1, - "\3" "break" + 1, - "\3" "call" + 1, - "\3" "cd" + 1, - "\1" "chcp" + 1, - "\3" "chdir" + 1, - "\3" "cls" + 1, - "\2" "color" + 1, - "\3" "copy" + 1, - "\1" "ctty" + 1, - "\3" "date" + 1, - "\3" "del" + 1, - "\3" "dir" + 1, - "\3" "echo" + 1, - "\2" "endlocal" + 1, - "\3" "erase" + 1, - "\3" "exit" + 1, - "\3" "for" + 1, - "\2" "ftype" + 1, - "\3" "goto" + 1, - "\3" "if" + 1, - "\1" "lfnfor" + 1, - "\1" "lh" + 1, - "\1" "lock" + 1, - "\3" "md" + 1, - "\3" "mkdir" + 1, - "\2" "move" + 1, - "\3" "path" + 1, - "\3" "pause" + 1, - "\2" "popd" + 1, - "\3" "prompt" + 1, - "\2" "pushd" + 1, - "\3" "rd" + 1, - "\3" "rem" + 1, - "\3" "ren" + 1, - "\3" "rename" + 1, - "\3" "rmdir" + 1, - "\3" "set" + 1, - "\2" "setlocal" + 1, - "\3" "shift" + 1, - "\2" "start" + 1, - "\3" "time" + 1, - "\2" "title" + 1, - "\1" "truename" + 1, - "\3" "type" + 1, - "\1" "unlock" + 1, - "\3" "ver" + 1, - "\3" "verify" + 1, - "\3" "vol" + 1, -}; - -static int -internal_match(const void *key, const void *elem) -{ - return strcmp(key, *(const char *const *)elem); -} - -static int -is_command_com(const char *interp) -{ - int i = strlen(interp) - 11; - - if ((i == 0 || i > 0 && isdirsep(interp[i-1])) && - strcasecmp(interp+i, "command.com") == 0) { - return 1; - } - return 0; -} - -static int -is_internal_cmd(const char *cmd, int nt) -{ - char cmdname[9], *b = cmdname, c, **nm; - - do { - if (!(c = *cmd++)) return 0; - } while (isspace(c)); - while (isalpha(c)) { - *b++ = tolower(c); - if (b == cmdname + sizeof(cmdname)) return 0; - c = *cmd++; - } - if (c == '.') c = *cmd; - switch (c) { - case '<': case '>': case '|': - return 1; - case '\0': case ' ': case '\t': case '\n': - break; - default: - return 0; - } - *b = 0; - nm = bsearch(cmdname, szInternalCmds, - sizeof(szInternalCmds) / sizeof(*szInternalCmds), - sizeof(*szInternalCmds), - internal_match); - if (!nm || !(nm[0][-1] & (nt ? 2 : 1))) - return 0; - return 1; -} - -SOCKET -rb_w32_get_osfhandle(int fh) -{ - return _get_osfhandle(fh); -} - -int -rb_w32_argv_size(char *const *argv) -{ - const char *p; - char *const *t; - int len, n, bs, quote; - - for (t = argv, len = 0; *t; t++) { - for (p = *t, n = quote = bs = 0; *p; ++p) { - switch (*p) { - case '\\': - ++bs; - break; - case '"': - n += bs + 1; bs = 0; - quote = 1; - break; - case ' ': case '\t': - quote = 1; - default: - bs = 0; - p = CharNext(p) - 1; - break; - } - } - len += p - *t + n + 1; - if (quote) len += 2; - } - return len; -} - -char * -rb_w32_join_argv(char *cmd, char *const *argv) -{ - const char *p, *s; - char *q, *const *t; - int n, bs, quote; - - for (t = argv, q = cmd; p = *t; t++) { - quote = 0; - s = p; - if (!*p || strpbrk(p, " \t\"")) { - quote = 1; - *q++ = '"'; - } - for (bs = 0; *p; ++p) { - switch (*p) { - case '\\': - ++bs; - break; - case '"': - memcpy(q, s, n = p - s); q += n; s = p; - memset(q, '\\', ++bs); q += bs; bs = 0; - break; - default: - bs = 0; - p = CharNext(p) - 1; - break; - } - } - memcpy(q, s, n = p - s); - q += n; - if (quote) *q++ = '"'; - *q++ = ' '; - } - if (q > cmd) --q; - *q = '\0'; - return cmd; -} - -rb_pid_t -rb_w32_pipe_exec(const char *cmd, const char *prog, int mode, int *pipe, - int *write_pipe) -{ - struct ChildRecord* child; - HANDLE hIn, hOut; - HANDLE hDupIn, hDupOut; - HANDLE hCurProc; - SECURITY_ATTRIBUTES sa; - BOOL reading, writing; - int ret; - - /* Figure out what we're doing... */ - if (mode & O_RDWR) { - reading = writing = TRUE; - } - else if (mode & O_WRONLY) { - reading = FALSE; - writing = TRUE; - } - else { - reading = TRUE; - writing = FALSE; - } - mode &= ~(O_RDWR|O_RDONLY|O_WRONLY); - if (!(mode & O_BINARY)) - mode |= O_TEXT; - - sa.nLength = sizeof (SECURITY_ATTRIBUTES); - sa.lpSecurityDescriptor = NULL; - sa.bInheritHandle = TRUE; - ret = -1; - - RUBY_CRITICAL(do { - /* create pipe */ - hCurProc = GetCurrentProcess(); - hIn = hOut = hDupIn = hDupOut = NULL; - if (reading) { - HANDLE hTmpIn; - if (!CreatePipe(&hTmpIn, &hOut, &sa, 2048L)) { - errno = map_errno(GetLastError()); - break; - } - if (!DuplicateHandle(hCurProc, hTmpIn, hCurProc, &hDupIn, 0, - FALSE, DUPLICATE_SAME_ACCESS)) { - errno = map_errno(GetLastError()); - CloseHandle(hTmpIn); - CloseHandle(hOut); - break; - } - CloseHandle(hTmpIn); - hTmpIn = NULL; - } - if (writing) { - HANDLE hTmpOut; - if (!CreatePipe(&hIn, &hTmpOut, &sa, 2048L)) { - errno = map_errno(GetLastError()); - break; - } - if (!DuplicateHandle(hCurProc, hTmpOut, hCurProc, &hDupOut, 0, - FALSE, DUPLICATE_SAME_ACCESS)) { - errno = map_errno(GetLastError()); - CloseHandle(hIn); - CloseHandle(hTmpOut); - break; - } - CloseHandle(hTmpOut); - hTmpOut = NULL; - } - - /* create child process */ - child = CreateChild(cmd, prog, &sa, hIn, hOut, NULL); - if (!child) { - if (hIn) - CloseHandle(hIn); - if (hOut) - CloseHandle(hOut); - if (hDupIn) - CloseHandle(hDupIn); - if (hDupOut) - CloseHandle(hDupOut); - break; - } - - /* associate handle to file descritor */ - if (reading) { - *pipe = rb_w32_open_osfhandle((intptr_t)hDupIn, O_RDONLY | mode); - if (writing) - *write_pipe = rb_w32_open_osfhandle((intptr_t)hDupOut, - O_WRONLY | mode); - } - else { - *pipe = rb_w32_open_osfhandle((intptr_t)hDupOut, O_WRONLY | mode); - } - if (hIn) - CloseHandle(hIn); - if (hOut) - CloseHandle(hOut); - if (reading && writing && *write_pipe == -1) { - if (*pipe != -1) - rb_w32_close(*pipe); - else - CloseHandle(hDupIn); - CloseHandle(hDupOut); - CloseChildHandle(child); - break; - } - else if (*pipe == -1) { - if (reading) - CloseHandle(hDupIn); - else - CloseHandle(hDupOut); - CloseChildHandle(child); - break; - } - - ret = child->pid; - } while (0)); - - return ret; -} - -rb_pid_t -rb_w32_spawn(int mode, const char *cmd, const char *prog) -{ - struct ChildRecord *child; - DWORD exitcode; - - switch (mode) { - case P_NOWAIT: - case P_OVERLAY: - break; - default: - errno = EINVAL; - return -1; - } - - child = CreateChild(cmd, prog, NULL, NULL, NULL, NULL); - if (!child) { - return -1; - } - - switch (mode) { - case P_NOWAIT: - return child->pid; - case P_OVERLAY: - WaitForSingleObject(child->hProcess, INFINITE); - GetExitCodeProcess(child->hProcess, &exitcode); - CloseChildHandle(child); - _exit(exitcode); - default: - return -1; /* not reached */ - } -} - -rb_pid_t -rb_w32_aspawn(int mode, const char *prog, char *const *argv) -{ - int len = rb_w32_argv_size(argv); - char *cmd = ALLOCA_N(char, len); - - if (!prog) prog = argv[0]; - return rb_w32_spawn(mode, rb_w32_join_argv(cmd, argv), prog); -} - -#ifdef HAVE_SYS_PARAM_H -# include <sys/param.h> -#else -# define MAXPATHLEN 512 -#endif - -static struct ChildRecord * -CreateChild(const char *cmd, const char *prog, SECURITY_ATTRIBUTES *psa, - HANDLE hInput, HANDLE hOutput, HANDLE hError) -{ - BOOL fRet; - DWORD dwCreationFlags; - STARTUPINFO aStartupInfo; - PROCESS_INFORMATION aProcessInformation; - SECURITY_ATTRIBUTES sa; - const char *shell; - struct ChildRecord *child; - char *p = NULL; - char fbuf[MAXPATHLEN]; - - if (!cmd && !prog) { - errno = EFAULT; - return NULL; - } - - child = FindFreeChildSlot(); - if (!child) { - errno = EAGAIN; - return NULL; - } - - if (!psa) { - sa.nLength = sizeof (SECURITY_ATTRIBUTES); - sa.lpSecurityDescriptor = NULL; - sa.bInheritHandle = TRUE; - psa = &sa; - } - - memset(&aStartupInfo, 0, sizeof (STARTUPINFO)); - memset(&aProcessInformation, 0, sizeof (PROCESS_INFORMATION)); - aStartupInfo.cb = sizeof (STARTUPINFO); - if (hInput || hOutput || hError) { - aStartupInfo.dwFlags = STARTF_USESTDHANDLES; - if (hInput) { - aStartupInfo.hStdInput = hInput; - } - else { - aStartupInfo.hStdInput = GetStdHandle(STD_INPUT_HANDLE); - } - if (hOutput) { - aStartupInfo.hStdOutput = hOutput; - } - else { - aStartupInfo.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE); - } - if (hError) { - aStartupInfo.hStdError = hError; - } - else { - aStartupInfo.hStdError = GetStdHandle(STD_ERROR_HANDLE); - } - } - - dwCreationFlags = (NORMAL_PRIORITY_CLASS); - - if (prog) { - if (!(p = dln_find_exe_r(prog, NULL, fbuf, sizeof(fbuf)))) { - shell = prog; - } - } - else { - int redir = -1; - int len = 0; - int nt; - while (ISSPACE(*cmd)) cmd++; - for (prog = cmd; *prog; prog = CharNext(prog)) { - if (ISSPACE(*prog)) { - len = prog - cmd; - do ++prog; while (ISSPACE(*prog)); - if (!*prog--) break; - } - else { - len = 0; - } - } - if (!len) len = strlen(cmd); - if ((shell = getenv("RUBYSHELL")) && (redir = has_redirection(cmd))) { - char *tmp = ALLOCA_N(char, strlen(shell) + len + sizeof(" -c ") + 2); - sprintf(tmp, "%s -c \"%.*s\"", shell, len, cmd); - cmd = tmp; - } - else if ((shell = getenv("COMSPEC")) && - (nt = !is_command_com(shell), - (redir < 0 ? has_redirection(cmd) : redir) || - is_internal_cmd(cmd, nt))) { - char *tmp = ALLOCA_N(char, strlen(shell) + len + sizeof(" /c ") - + (nt ? 2 : 0)); - sprintf(tmp, nt ? "%s /c \"%.*s\"" : "%s /c %.*s", shell, len, cmd); - cmd = tmp; - } - else { - shell = NULL; - prog = cmd; - for (;;) { - if (!*prog) { - p = dln_find_exe_r(cmd, NULL, fbuf, sizeof(fbuf)); - break; - } - if (strchr(".:*?\"/\\", *prog)) { - if (cmd[len]) { - char *tmp = ALLOCA_N(char, len + 1); - memcpy(tmp, cmd, len); - tmp[len] = 0; - cmd = tmp; - } - break; - } - if (ISSPACE(*prog) || strchr("<>|", *prog)) { - len = prog - cmd; - p = ALLOCA_N(char, len + 1); - memcpy(p, cmd, len); - p[len] = 0; - p = dln_find_exe_r(p, NULL, fbuf, sizeof(fbuf)); - break; - } - prog++; - } - } - } - if (p) { - shell = p; - while (*p) { - if ((unsigned char)*p == '/') - *p = '\\'; - p = CharNext(p); - } - } - - RUBY_CRITICAL({ - fRet = CreateProcess(shell, (char *)cmd, psa, psa, - psa->bInheritHandle, dwCreationFlags, NULL, NULL, - &aStartupInfo, &aProcessInformation); - errno = map_errno(GetLastError()); - }); - - if (!fRet) { - child->pid = 0; /* release the slot */ - return NULL; - } - - CloseHandle(aProcessInformation.hThread); - - child->hProcess = aProcessInformation.hProcess; - child->pid = (rb_pid_t)aProcessInformation.dwProcessId; - - if (!IsWinNT()) { - /* On Win9x, make pid positive similarly to cygwin and perl */ - child->pid = -child->pid; - } - - return child; -} - -typedef struct _NtCmdLineElement { - struct _NtCmdLineElement *next; - char *str; - int len; - int flags; -} NtCmdLineElement; - -// -// Possible values for flags -// - -#define NTGLOB 0x1 // element contains a wildcard -#define NTMALLOC 0x2 // string in element was malloc'ed -#define NTSTRING 0x4 // element contains a quoted string - -static int -insert(const char *path, VALUE vinfo, void *enc) -{ - NtCmdLineElement *tmpcurr; - NtCmdLineElement ***tail = (NtCmdLineElement ***)vinfo; - - tmpcurr = (NtCmdLineElement *)malloc(sizeof(NtCmdLineElement)); - if (!tmpcurr) return -1; - MEMZERO(tmpcurr, NtCmdLineElement, 1); - tmpcurr->len = strlen(path); - tmpcurr->str = strdup(path); - if (!tmpcurr->str) return -1; - tmpcurr->flags |= NTMALLOC; - **tail = tmpcurr; - *tail = &tmpcurr->next; - - return 0; -} - - -static NtCmdLineElement ** -cmdglob(NtCmdLineElement *patt, NtCmdLineElement **tail) -{ - char buffer[MAXPATHLEN], *buf = buffer; - char *p; - NtCmdLineElement **last = tail; - int status; - - if (patt->len >= MAXPATHLEN) - if (!(buf = malloc(patt->len + 1))) return 0; - - strlcpy(buf, patt->str, patt->len + 1); - buf[patt->len] = '\0'; - for (p = buf; *p; p = CharNext(p)) - if (*p == '\\') - *p = '/'; - status = ruby_brace_glob(buf, 0, insert, (VALUE)&tail); - if (buf != buffer) - free(buf); - - if (status || last == tail) return 0; - if (patt->flags & NTMALLOC) - free(patt->str); - free(patt); - return tail; -} - -// -// Check a command string to determine if it has I/O redirection -// characters that require it to be executed by a command interpreter -// - -static int -has_redirection(const char *cmd) -{ - char quote = '\0'; - const char *ptr; - - // - // Scan the string, looking for redirection (< or >) or pipe - // characters (|) that are not in a quoted string - // - - for (ptr = cmd; *ptr;) { - switch (*ptr) { - case '\'': - case '\"': - if (!quote) - quote = *ptr; - else if (quote == *ptr) - quote = '\0'; - ptr++; - break; - - case '>': - case '<': - case '|': - if (!quote) - return TRUE; - ptr++; - break; - - case '\\': - ptr++; - default: - ptr = CharNext(ptr); - break; - } - } - return FALSE; -} - -static inline char * -skipspace(char *ptr) -{ - while (ISSPACE(*ptr)) - ptr++; - return ptr; -} - -int -rb_w32_cmdvector(const char *cmd, char ***vec) -{ - int globbing, len; - int elements, strsz, done; - int slashes, escape; - char *ptr, *base, *buffer, *cmdline; - char **vptr; - char quote; - NtCmdLineElement *curr, **tail; - NtCmdLineElement *cmdhead = NULL, **cmdtail = &cmdhead; - - // - // just return if we don't have a command line - // - - while (ISSPACE(*cmd)) - cmd++; - if (!*cmd) { - *vec = NULL; - return 0; - } - - ptr = cmdline = strdup(cmd); - - // - // Ok, parse the command line, building a list of CmdLineElements. - // When we've finished, and it's an input command (meaning that it's - // the processes argv), we'll do globing and then build the argument - // vector. - // The outer loop does one interation for each element seen. - // The inner loop does one interation for each character in the element. - // - - while (*(ptr = skipspace(ptr))) { - base = ptr; - quote = slashes = globbing = escape = 0; - for (done = 0; !done && *ptr; ) { - // - // Switch on the current character. We only care about the - // white-space characters, the wild-card characters, and the - // quote characters. - // - - switch (*ptr) { - case '\\': - if (quote != '\'') slashes++; - break; - - case ' ': - case '\t': - case '\n': - // - // if we're not in a string, then we're finished with this - // element - // - - if (!quote) { - *ptr = 0; - done = 1; - } - break; - - case '*': - case '?': - case '[': - case '{': - // - // record the fact that this element has a wildcard character - // N.B. Don't glob if inside a single quoted string - // - - if (quote != '\'') - globbing++; - slashes = 0; - break; - - case '\'': - case '\"': - // - // if we're already in a string, see if this is the - // terminating close-quote. If it is, we're finished with - // the string, but not neccessarily with the element. - // If we're not already in a string, start one. - // - - if (!(slashes & 1)) { - if (!quote) - quote = *ptr; - else if (quote == *ptr) { - if (quote == '"' && quote == ptr[1]) - ptr++; - quote = '\0'; - } - } - escape++; - slashes = 0; - break; - - default: - ptr = CharNext(ptr); - slashes = 0; - continue; - } - ptr++; - } - - // - // when we get here, we've got a pair of pointers to the element, - // base and ptr. Base points to the start of the element while ptr - // points to the character following the element. - // - - len = ptr - base; - if (done) --len; - - // - // if it's an input vector element and it's enclosed by quotes, - // we can remove them. - // - - if (escape) { - char *p = base, c; - slashes = quote = 0; - while (p < base + len) { - switch (c = *p) { - case '\\': - p++; - if (quote != '\'') slashes++; - break; - - case '\'': - case '"': - if (!(slashes & 1) && quote && quote != c) { - p++; - slashes = 0; - break; - } - memcpy(p - ((slashes + 1) >> 1), p + (~slashes & 1), - base + len - p); - len -= ((slashes + 1) >> 1) + (~slashes & 1); - p -= (slashes + 1) >> 1; - if (!(slashes & 1)) { - if (quote) { - if (quote == '"' && quote == *p) - p++; - quote = '\0'; - } - else - quote = c; - } - else - p++; - slashes = 0; - break; - - default: - p = CharNext(p); - slashes = 0; - break; - } - } - } - - curr = (NtCmdLineElement *)calloc(sizeof(NtCmdLineElement), 1); - if (!curr) goto do_nothing; - curr->str = base; - curr->len = len; - - if (globbing && (tail = cmdglob(curr, cmdtail))) { - cmdtail = tail; - } - else { - *cmdtail = curr; - cmdtail = &curr->next; - } - } - - // - // Almost done! - // Count up the elements, then allocate space for a vector of pointers - // (argv) and a string table for the elements. - // - - for (elements = 0, strsz = 0, curr = cmdhead; curr; curr = curr->next) { - elements++; - strsz += (curr->len + 1); - } - - len = (elements+1)*sizeof(char *) + strsz; - buffer = (char *)malloc(len); - if (!buffer) { - do_nothing: - while (curr = cmdhead) { - cmdhead = curr->next; - if (curr->flags & NTMALLOC) free(curr->str); - free(curr); - } - free(cmdline); - for (vptr = *vec; *vptr; ++vptr); - return vptr - *vec; - } - - // - // make vptr point to the start of the buffer - // and ptr point to the area we'll consider the string table. - // - // buffer (*vec) - // | - // V ^---------------------V - // +---+---+---+---+---+---+---+---+---+---+---+---+---+---+---+ - // | | | .... | NULL | | ..... |\0 | | ..... |\0 |... - // +---+---+---+---+---+---+---+---+---+---+---+---+---+---+---+ - // |- elements+1 -| ^ 1st element ^ 2nd element - - vptr = (char **) buffer; - - ptr = buffer + (elements+1) * sizeof(char *); - - while (curr = cmdhead) { - strlcpy(ptr, curr->str, curr->len + 1); - *vptr++ = ptr; - ptr += curr->len + 1; - cmdhead = curr->next; - if (curr->flags & NTMALLOC) free(curr->str); - free(curr); - } - *vptr = 0; - - *vec = (char **) buffer; - free(cmdline); - return elements; -} - -// -// UNIX compatible directory access functions for NT -// - -#define PATHLEN 1024 - -// -// The idea here is to read all the directory names into a string table -// (separated by nulls) and when one of the other dir functions is called -// return the pointer to the current file name. -// - -#define GetBit(bits, i) ((bits)[(i) / CHAR_BIT] & (1 << (i) % CHAR_BIT)) -#define SetBit(bits, i) ((bits)[(i) / CHAR_BIT] |= (1 << (i) % CHAR_BIT)) - -#define BitOfIsDir(n) ((n) * 2) -#define BitOfIsRep(n) ((n) * 2 + 1) -#define DIRENT_PER_CHAR (CHAR_BIT / 2) - -DIR * -rb_w32_opendir(const char *filename) -{ - DIR *p; - long len; - long idx; - char *scanname; - char *tmp; - struct stati64 sbuf; - WIN32_FIND_DATA fd; - HANDLE fh; - - // - // check to see if we've got a directory - // - if (rb_w32_stati64(filename, &sbuf) < 0) - return NULL; - if (!(sbuf.st_mode & S_IFDIR) && - (!ISALPHA(filename[0]) || filename[1] != ':' || filename[2] != '\0' || - ((1 << (filename[0] & 0x5f) - 'A') & GetLogicalDrives()) == 0)) { - errno = ENOTDIR; - return NULL; - } - - // - // Get us a DIR structure - // - p = calloc(sizeof(DIR), 1); - if (p == NULL) - return NULL; - - // - // Create the search pattern - // - len = strlen(filename) + 2 + 1; - if (!(scanname = malloc(len))) { - free(p); - return NULL; - } - strlcpy(scanname, filename, len); - - if (index("/\\:", *CharPrev(scanname, scanname + strlen(scanname))) == NULL) - strlcat(scanname, "/*", len); - else - strlcat(scanname, "*", len); - - // - // do the FindFirstFile call - // - fh = FindFirstFile(scanname, &fd); - free(scanname); - if (fh == INVALID_HANDLE_VALUE) { - errno = map_errno(GetLastError()); - free(p); - return NULL; - } - - idx = 0; - - // - // loop finding all the files that match the wildcard - // (which should be all of them in this directory!). - // the variable idx should point one past the null terminator - // of the previous string found. - // - do { - len = strlen(fd.cFileName) + 1; - - // - // bump the string table size by enough for the - // new name and it's null terminator - // - tmp = realloc(p->start, idx + len); - if (!tmp) { - error: - rb_w32_closedir(p); - FindClose(fh); - errno = ENOMEM; - return NULL; - } - - p->start = tmp; - strlcpy(&p->start[idx], fd.cFileName, len); - - if (p->nfiles % DIRENT_PER_CHAR == 0) { - tmp = realloc(p->bits, p->nfiles / DIRENT_PER_CHAR + 1); - if (!tmp) - goto error; - p->bits = tmp; - p->bits[p->nfiles / DIRENT_PER_CHAR] = 0; - } - if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) - SetBit(p->bits, BitOfIsDir(p->nfiles)); - if (fd.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) - SetBit(p->bits, BitOfIsRep(p->nfiles)); - - p->nfiles++; - idx += len; - } while (FindNextFile(fh, &fd)); - FindClose(fh); - p->size = idx; - p->curr = p->start; - return p; -} - -// -// Move to next entry -// - -static void -move_to_next_entry(DIR *dirp) -{ - if (dirp->curr) { - dirp->loc++; - dirp->curr += strlen(dirp->curr) + 1; - if (dirp->curr >= (dirp->start + dirp->size)) { - dirp->curr = NULL; - } - } -} - -// -// Readdir just returns the current string pointer and bumps the -// string pointer to the next entry. -// - -struct direct * -rb_w32_readdir(DIR *dirp) -{ - static int dummy = 0; - - if (dirp->curr) { - - // - // first set up the structure to return - // - dirp->dirstr.d_namlen = strlen(dirp->curr); - if (!(dirp->dirstr.d_name = malloc(dirp->dirstr.d_namlen + 1))) - return NULL; - strlcpy(dirp->dirstr.d_name, dirp->curr, dirp->dirstr.d_namlen + 1); - - // - // Fake inode - // - dirp->dirstr.d_ino = dummy++; - - // - // Attributes - // - dirp->dirstr.d_isdir = GetBit(dirp->bits, BitOfIsDir(dirp->loc)); - dirp->dirstr.d_isrep = GetBit(dirp->bits, BitOfIsRep(dirp->loc)); - - // - // Now set up for the next call to readdir - // - - move_to_next_entry(dirp); - - return &(dirp->dirstr); - - } else - return NULL; -} - -// -// Telldir returns the current string pointer position -// - -off_t -rb_w32_telldir(DIR *dirp) -{ - return dirp->loc; -} - -// -// Seekdir moves the string pointer to a previously saved position -// (Saved by telldir). - -void -rb_w32_seekdir(DIR *dirp, off_t loc) -{ - if (dirp->loc > loc) rb_w32_rewinddir(dirp); - - while (dirp->curr && dirp->loc < loc) { - move_to_next_entry(dirp); - } -} - -// -// Rewinddir resets the string pointer to the start -// - -void -rb_w32_rewinddir(DIR *dirp) -{ - dirp->curr = dirp->start; - dirp->loc = 0; -} - -// -// This just free's the memory allocated by opendir -// - -void -rb_w32_closedir(DIR *dirp) -{ - if (dirp) { - if (dirp->dirstr.d_name) - free(dirp->dirstr.d_name); - if (dirp->start) - free(dirp->start); - if (dirp->bits) - free(dirp->bits); - free(dirp); - } -} - -#if (defined _MT || defined __MSVCRT__) && !defined __BORLANDC__ -#define MSVCRT_THREADS -#endif -#ifdef MSVCRT_THREADS -# define MTHREAD_ONLY(x) x -# define STHREAD_ONLY(x) -#elif defined(__BORLANDC__) -# define MTHREAD_ONLY(x) -# define STHREAD_ONLY(x) -#else -# define MTHREAD_ONLY(x) -# define STHREAD_ONLY(x) x -#endif - -typedef struct { - intptr_t osfhnd; /* underlying OS file HANDLE */ - char osfile; /* attributes of file (e.g., open in text mode?) */ - char pipech; /* one char buffer for handles opened on pipes */ -#ifdef MSVCRT_THREADS - int lockinitflag; - CRITICAL_SECTION lock; -#endif -#if _MSC_VER >= 1400 - char textmode; - char pipech2[2]; -#endif -} ioinfo; - -#if !defined _CRTIMP || defined __MINGW32__ -#undef _CRTIMP -#define _CRTIMP __declspec(dllimport) -#endif - -#if !defined(__BORLANDC__) && !defined(_WIN32_WCE) -EXTERN_C _CRTIMP ioinfo * __pioinfo[]; - -#define IOINFO_L2E 5 -#define IOINFO_ARRAY_ELTS (1 << IOINFO_L2E) -#define _pioinfo(i) ((ioinfo*)((char*)(__pioinfo[i >> IOINFO_L2E]) + (i & (IOINFO_ARRAY_ELTS - 1)) * (sizeof(ioinfo) + pioinfo_extra))) -#define _osfhnd(i) (_pioinfo(i)->osfhnd) -#define _osfile(i) (_pioinfo(i)->osfile) -#define _pipech(i) (_pioinfo(i)->pipech) - -#if _MSC_VER >= 1400 -static size_t pioinfo_extra = 0; /* workaround for VC++8 SP1 */ - -static void -set_pioinfo_extra(void) -{ - int fd; - - fd = open("NUL", O_RDONLY); - for (pioinfo_extra = 0; pioinfo_extra <= 64; pioinfo_extra += sizeof(void *)) { - if (_osfhnd(fd) == _get_osfhandle(fd)) { - break; - } - } - close(fd); - - if (pioinfo_extra > 64) { - /* not found, maybe something wrong... */ - pioinfo_extra = 0; - } -} -#else -#define pioinfo_extra 0 -#endif - -#define _set_osfhnd(fh, osfh) (void)(_osfhnd(fh) = osfh) -#define _set_osflags(fh, flags) (_osfile(fh) = (flags)) - -#define FOPEN 0x01 /* file handle open */ -#define FNOINHERIT 0x10 /* file handle opened O_NOINHERIT */ -#define FAPPEND 0x20 /* file handle opened O_APPEND */ -#define FDEV 0x40 /* file handle refers to device */ -#define FTEXT 0x80 /* file handle is in text mode */ - -static int -rb_w32_open_osfhandle(intptr_t osfhandle, int flags) -{ - int fh; - char fileflags; /* _osfile flags */ - HANDLE hF; - - /* copy relevant flags from second parameter */ - fileflags = FDEV; - - if (flags & O_APPEND) - fileflags |= FAPPEND; - - if (flags & O_TEXT) - fileflags |= FTEXT; - - if (flags & O_NOINHERIT) - fileflags |= FNOINHERIT; - - /* attempt to allocate a C Runtime file handle */ - hF = CreateFile("NUL", 0, 0, NULL, OPEN_ALWAYS, 0, NULL); - fh = _open_osfhandle((long)hF, 0); - CloseHandle(hF); - if (fh == -1) { - errno = EMFILE; /* too many open files */ - _doserrno = 0L; /* not an OS error */ - } - else { - - MTHREAD_ONLY(EnterCriticalSection(&(_pioinfo(fh)->lock))); - /* the file is open. now, set the info in _osfhnd array */ - _set_osfhnd(fh, osfhandle); - - fileflags |= FOPEN; /* mark as open */ - - _set_osflags(fh, fileflags); /* set osfile entry */ - MTHREAD_ONLY(LeaveCriticalSection(&_pioinfo(fh)->lock)); - } - return fh; /* return handle */ -} - -static void -init_stdhandle(void) -{ - int nullfd = -1; - int keep = 0; -#define open_null(fd) \ - (((nullfd < 0) ? \ - (nullfd = open("NUL", O_RDWR|O_BINARY)) : 0), \ - ((nullfd == (fd)) ? (keep = 1) : dup2(nullfd, fd)), \ - (fd)) - - if (fileno(stdin) < 0) { - stdin->_file = open_null(0); - } - if (fileno(stdout) < 0) { - stdout->_file = open_null(1); - } - if (fileno(stderr) < 0) { - stderr->_file = open_null(2); - } - if (nullfd >= 0 && !keep) close(nullfd); - setvbuf(stderr, NULL, _IONBF, 0); -} -#else - -#define _set_osfhnd(fh, osfh) (void)((fh), (osfh)) -#define _set_osflags(fh, flags) (void)((fh), (flags)) - -static void -init_stdhandle(void) -{ -} -#endif - -#ifdef __BORLANDC__ -static int -rb_w32_open_osfhandle(intptr_t osfhandle, int flags) -{ - int fd = _open_osfhandle(osfhandle, flags); - if (fd == -1) { - errno = EMFILE; /* too many open files */ - _doserrno = 0L; /* not an OS error */ - } - return fd; -} -#endif - -#undef getsockopt - -static int -is_socket(SOCKET sock) -{ - if (st_lookup(socklist, (st_data_t)sock, NULL)) - return TRUE; - else - return FALSE; -} - -int -rb_w32_is_socket(int fd) -{ - return is_socket(TO_SOCKET(fd)); -} - -// -// Since the errors returned by the socket error function -// WSAGetLastError() are not known by the library routine strerror -// we have to roll our own. -// - -#undef strerror - -char * -rb_w32_strerror(int e) -{ - static char buffer[512]; -#if !defined __MINGW32__ - extern int sys_nerr; -#endif - DWORD source = 0; - char *p; - -#if defined __BORLANDC__ && defined ENOTEMPTY // _sys_errlist is broken - switch (e) { - case ENAMETOOLONG: - return "Filename too long"; - case ENOTEMPTY: - return "Directory not empty"; - } -#endif - - if (e < 0 || e > sys_nerr) { - if (e < 0) - e = GetLastError(); - if (FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | - FORMAT_MESSAGE_IGNORE_INSERTS, &source, e, 0, - buffer, sizeof(buffer), NULL) == 0) - strlcpy(buffer, "Unknown Error", sizeof(buffer)); - } - else - strlcpy(buffer, strerror(e), sizeof(buffer)); - - p = buffer; - while ((p = strpbrk(p, "\r\n")) != NULL) { - memmove(p, p + 1, strlen(p)); - } - return buffer; -} - -// -// various stubs -// - - -// Ownership -// -// Just pretend that everyone is a superuser. NT will let us know if -// we don't really have permission to do something. -// - -#define ROOT_UID 0 -#define ROOT_GID 0 - -rb_uid_t -getuid(void) -{ - return ROOT_UID; -} - -rb_uid_t -geteuid(void) -{ - return ROOT_UID; -} - -rb_gid_t -getgid(void) -{ - return ROOT_GID; -} - -rb_gid_t -getegid(void) -{ - return ROOT_GID; -} - -int -setuid(rb_uid_t uid) -{ - return (uid == ROOT_UID ? 0 : -1); -} - -int -setgid(rb_gid_t gid) -{ - return (gid == ROOT_GID ? 0 : -1); -} - -// -// File system stuff -// - -int -ioctl(int i, int u, ...) -{ - errno = EINVAL; - return -1; -} - -#undef FD_SET - -void -rb_w32_fdset(int fd, fd_set *set) -{ - unsigned int i; - SOCKET s = TO_SOCKET(fd); - - for (i = 0; i < set->fd_count; i++) { - if (set->fd_array[i] == s) { - return; - } - } - if (i == set->fd_count) { - if (set->fd_count < FD_SETSIZE) { - set->fd_array[i] = s; - set->fd_count++; - } - } -} - -#undef FD_CLR - -void -rb_w32_fdclr(int fd, fd_set *set) -{ - unsigned int i; - SOCKET s = TO_SOCKET(fd); - - for (i = 0; i < set->fd_count; i++) { - if (set->fd_array[i] == s) { - while (i < set->fd_count - 1) { - set->fd_array[i] = set->fd_array[i + 1]; - i++; - } - set->fd_count--; - break; - } - } -} - -#undef FD_ISSET - -int -rb_w32_fdisset(int fd, fd_set *set) -{ - int ret; - SOCKET s = TO_SOCKET(fd); - if (s == (SOCKET)INVALID_HANDLE_VALUE) - return 0; - RUBY_CRITICAL(ret = __WSAFDIsSet(s, set)); - return ret; -} - -// -// Networking trampolines -// These are used to avoid socket startup/shutdown overhead in case -// the socket routines aren't used. -// - -#undef select - -static int -extract_fd(fd_set *dst, fd_set *src, int (*func)(SOCKET)) -{ - int s = 0; - if (!src || !dst) return 0; - - while (s < src->fd_count) { - SOCKET fd = src->fd_array[s]; - - if (!func || (*func)(fd)) { /* move it to dst */ - int d; - - for (d = 0; d < dst->fd_count; d++) { - if (dst->fd_array[d] == fd) break; - } - if (d == dst->fd_count && dst->fd_count < FD_SETSIZE) { - dst->fd_array[dst->fd_count++] = fd; - } - memmove( - &src->fd_array[s], - &src->fd_array[s+1], - sizeof(src->fd_array[0]) * (--src->fd_count - s)); - } - else s++; - } - - return dst->fd_count; -} - -static int -is_not_socket(SOCKET sock) -{ - return !is_socket(sock); -} - -static int -is_pipe(SOCKET sock) /* DONT call this for SOCKET! it clains it is PIPE. */ -{ - int ret; - - RUBY_CRITICAL({ - ret = (GetFileType((HANDLE)sock) == FILE_TYPE_PIPE); - }); - - return ret; -} - -static int -is_readable_pipe(SOCKET sock) /* call this for pipe only */ -{ - int ret; - DWORD n = 0; - - RUBY_CRITICAL( - if (PeekNamedPipe((HANDLE)sock, NULL, 0, NULL, &n, NULL)) { - ret = (n > 0); - } - else { - ret = (GetLastError() == ERROR_BROKEN_PIPE); /* pipe was closed */ - } - ); - - return ret; -} - -static int -is_console(SOCKET sock) /* DONT call this for SOCKET! */ -{ - int ret; - DWORD n = 0; - INPUT_RECORD ir; - - RUBY_CRITICAL( - ret = (PeekConsoleInput((HANDLE)sock, &ir, 1, &n)) - ); - - return ret; -} - -static int -is_readable_console(SOCKET sock) /* call this for console only */ -{ - int ret = 0; - DWORD n = 0; - INPUT_RECORD ir; - - RUBY_CRITICAL( - if (PeekConsoleInput((HANDLE)sock, &ir, 1, &n) && n > 0) { - if (ir.EventType == KEY_EVENT && ir.Event.KeyEvent.bKeyDown && - ir.Event.KeyEvent.uChar.AsciiChar) { - ret = 1; - } - else { - ReadConsoleInput((HANDLE)sock, &ir, 1, &n); - } - } - ); - - return ret; -} - -static int -do_select(int nfds, fd_set *rd, fd_set *wr, fd_set *ex, - struct timeval *timeout) -{ - int r = 0; - - if (nfds == 0) { - if (timeout) - rb_w32_sleep(timeout->tv_sec * 1000 + timeout->tv_usec / 1000); - else - rb_w32_sleep(INFINITE); - } - else { - RUBY_CRITICAL( - EnterCriticalSection(&select_mutex); - r = select(nfds, rd, wr, ex, timeout); - LeaveCriticalSection(&select_mutex); - if (r == SOCKET_ERROR) { - errno = map_errno(WSAGetLastError()); - r = -1; - } - ); - } - - return r; -} - -static inline int -subtract(struct timeval *rest, const struct timeval *wait) -{ - while (rest->tv_usec < wait->tv_usec) { - if (rest->tv_sec <= wait->tv_sec) { - return 0; - } - rest->tv_sec -= 1; - rest->tv_usec += 1000 * 1000; - } - rest->tv_sec -= wait->tv_sec; - rest->tv_usec -= wait->tv_usec; - return 1; -} - -static inline int -compare(const struct timeval *t1, const struct timeval *t2) -{ - if (t1->tv_sec < t2->tv_sec) - return -1; - if (t1->tv_sec > t2->tv_sec) - return 1; - if (t1->tv_usec < t2->tv_usec) - return -1; - if (t1->tv_usec > t2->tv_usec) - return 1; - return 0; -} - -#undef Sleep -int WSAAPI -rb_w32_select(int nfds, fd_set *rd, fd_set *wr, fd_set *ex, - struct timeval *timeout) -{ - int r; - fd_set pipe_rd; - fd_set cons_rd; - fd_set else_rd; - fd_set else_wr; - fd_set except; - int nonsock = 0; - struct timeval limit; - - if (nfds < 0 || (timeout && (timeout->tv_sec < 0 || timeout->tv_usec < 0))) { - errno = EINVAL; - return -1; - } - - if (timeout) { - if (timeout->tv_sec < 0 || - timeout->tv_usec < 0 || - timeout->tv_usec >= 1000000) { - errno = EINVAL; - return -1; - } - gettimeofday(&limit, NULL); - limit.tv_sec += timeout->tv_sec; - limit.tv_usec += timeout->tv_usec; - if (limit.tv_usec >= 1000000) { - limit.tv_usec -= 1000000; - limit.tv_sec++; - } - } - - if (!NtSocketsInitialized) { - StartSockets(); - } - - // assume else_{rd,wr} (other than socket, pipe reader, console reader) - // are always readable/writable. but this implementation still has - // problem. if pipe's buffer is full, writing to pipe will block - // until some data is read from pipe. but ruby is single threaded system, - // so whole system will be blocked forever. - - else_rd.fd_count = 0; - nonsock += extract_fd(&else_rd, rd, is_not_socket); - - pipe_rd.fd_count = 0; - extract_fd(&pipe_rd, &else_rd, is_pipe); // should not call is_pipe for socket - - cons_rd.fd_count = 0; - extract_fd(&cons_rd, &else_rd, is_console); // ditto - - else_wr.fd_count = 0; - nonsock += extract_fd(&else_wr, wr, is_not_socket); - - except.fd_count = 0; - extract_fd(&except, ex, is_not_socket); // drop only - - r = 0; - if (rd && rd->fd_count > r) r = rd->fd_count; - if (wr && wr->fd_count > r) r = wr->fd_count; - if (ex && ex->fd_count > r) r = ex->fd_count; - if (nfds > r) nfds = r; - - { - struct timeval rest; - struct timeval wait; - struct timeval zero; - wait.tv_sec = 0; wait.tv_usec = 10 * 1000; // 10ms - zero.tv_sec = 0; zero.tv_usec = 0; // 0ms - for (;;) { - if (nonsock) { - // modifying {else,pipe,cons}_rd is safe because - // if they are modified, function returns immediately. - extract_fd(&else_rd, &pipe_rd, is_readable_pipe); - extract_fd(&else_rd, &cons_rd, is_readable_console); - } - - if (else_rd.fd_count || else_wr.fd_count) { - r = do_select(nfds, rd, wr, ex, &zero); // polling - if (r < 0) break; // XXX: should I ignore error and return signaled handles? - r += extract_fd(rd, &else_rd, NULL); // move all - r += extract_fd(wr, &else_wr, NULL); // move all - break; - } - else { - struct timeval *dowait = &wait; - - fd_set orig_rd; - fd_set orig_wr; - fd_set orig_ex; - if (rd) orig_rd = *rd; - if (wr) orig_wr = *wr; - if (ex) orig_ex = *ex; - r = do_select(nfds, rd, wr, ex, &zero); // polling - if (r != 0) break; // signaled or error - if (rd) *rd = orig_rd; - if (wr) *wr = orig_wr; - if (ex) *ex = orig_ex; - - if (timeout) { - struct timeval now; - gettimeofday(&now, NULL); - rest = limit; - if (!subtract(&rest, &now)) break; - if (compare(&rest, &wait) < 0) dowait = &rest; - } - Sleep(dowait->tv_sec * 1000 + dowait->tv_usec / 1000); - } - } - } - - return r; -} - -#undef accept - -int WSAAPI -rb_w32_accept(int s, struct sockaddr *addr, int *addrlen) -{ - SOCKET r; - - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = accept(TO_SOCKET(s), addr, addrlen); - if (r == INVALID_SOCKET) { - errno = map_errno(WSAGetLastError()); - s = -1; - } - else { - s = rb_w32_open_osfhandle(r, O_RDWR|O_BINARY|O_NOINHERIT); - if (s != -1) - st_insert(socklist, (st_data_t)r, (st_data_t)0); - else - closesocket(r); - } - }); - return s; -} - -#undef bind - -int WSAAPI -rb_w32_bind(int s, const struct sockaddr *addr, int addrlen) -{ - int r; - - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = bind(TO_SOCKET(s), addr, addrlen); - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef connect - -int WSAAPI -rb_w32_connect(int s, const struct sockaddr *addr, int addrlen) -{ - int r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = connect(TO_SOCKET(s), addr, addrlen); - if (r == SOCKET_ERROR) { - r = WSAGetLastError(); - if (r != WSAEWOULDBLOCK) { - errno = map_errno(r); - } - else { - errno = EINPROGRESS; - r = -1; - } - } - }); - return r; -} - - -#undef getpeername - -int WSAAPI -rb_w32_getpeername(int s, struct sockaddr *addr, int *addrlen) -{ - int r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = getpeername(TO_SOCKET(s), addr, addrlen); - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef getsockname - -int WSAAPI -rb_w32_getsockname(int s, struct sockaddr *addr, int *addrlen) -{ - int r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = getsockname(TO_SOCKET(s), addr, addrlen); - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -int WSAAPI -rb_w32_getsockopt(int s, int level, int optname, char *optval, int *optlen) -{ - int r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = getsockopt(TO_SOCKET(s), level, optname, optval, optlen); - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef ioctlsocket - -int WSAAPI -rb_w32_ioctlsocket(int s, long cmd, u_long *argp) -{ - int r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = ioctlsocket(TO_SOCKET(s), cmd, argp); - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef listen - -int WSAAPI -rb_w32_listen(int s, int backlog) -{ - int r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = listen(TO_SOCKET(s), backlog); - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef recv -#undef recvfrom -#undef send -#undef sendto - -static int -overlapped_socket_io(BOOL input, int fd, char *buf, int len, int flags, - struct sockaddr *addr, int *addrlen) -{ - int r; - int ret; - int mode; - st_data_t data; - DWORD flg; - WSAOVERLAPPED wol; - WSABUF wbuf; - int err; - SOCKET s; - - if (!NtSocketsInitialized) - StartSockets(); - - s = TO_SOCKET(fd); - st_lookup(socklist, (st_data_t)s, &data); - mode = (int)data; - if (!cancel_io || (mode & O_NONBLOCK)) { - RUBY_CRITICAL({ - if (input) { - if (addr && addrlen) - r = recvfrom(s, buf, len, flags, addr, addrlen); - else - r = recv(s, buf, len, flags); - } - else { - if (addr && addrlen) - r = sendto(s, buf, len, flags, addr, *addrlen); - else - r = send(s, buf, len, flags); - } - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - } - else { - DWORD size; - wbuf.len = len; - wbuf.buf = buf; - memset(&wol, 0, sizeof(wol)); - RUBY_CRITICAL({ - wol.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL); - if (input) { - flg = flags; - if (addr && addrlen) - ret = WSARecvFrom(s, &wbuf, 1, &size, &flg, addr, addrlen, - &wol, NULL); - else - ret = WSARecv(s, &wbuf, 1, &size, &flg, &wol, NULL); - } - else { - if (addr && addrlen) - ret = WSASendTo(s, &wbuf, 1, &size, flags, addr, *addrlen, - &wol, NULL); - else - ret = WSASend(s, &wbuf, 1, &size, flags, &wol, NULL); - } - }); - - if (ret != SOCKET_ERROR) { - r = size; - } - else if ((err = WSAGetLastError()) == WSA_IO_PENDING) { - switch (rb_w32_wait_events_blocking(&wol.hEvent, 1, INFINITE)) { - case WAIT_OBJECT_0: - RUBY_CRITICAL( - ret = WSAGetOverlappedResult(s, &wol, &size, TRUE, &flg) - ); - if (ret) { - r = size; - break; - } - /* thru */ - default: - errno = map_errno(err); - /* thru */ - case WAIT_OBJECT_0 + 1: - /* interrupted */ - r = -1; - cancel_io((HANDLE)s); - break; - } - } - else { - errno = map_errno(err); - r = -1; - } - CloseHandle(&wol.hEvent); - } - - return r; -} - -int WSAAPI -rb_w32_recv(int fd, char *buf, int len, int flags) -{ - return overlapped_socket_io(TRUE, fd, buf, len, flags, NULL, NULL); -} - -int WSAAPI -rb_w32_recvfrom(int fd, char *buf, int len, int flags, - struct sockaddr *from, int *fromlen) -{ - return overlapped_socket_io(TRUE, fd, buf, len, flags, from, fromlen); -} - -int WSAAPI -rb_w32_send(int fd, const char *buf, int len, int flags) -{ - return overlapped_socket_io(FALSE, fd, (char *)buf, len, flags, NULL, NULL); -} - -int WSAAPI -rb_w32_sendto(int fd, const char *buf, int len, int flags, - const struct sockaddr *to, int tolen) -{ - return overlapped_socket_io(FALSE, fd, (char *)buf, len, flags, - (struct sockaddr *)to, &tolen); -} - -#undef setsockopt - -int WSAAPI -rb_w32_setsockopt(int s, int level, int optname, const char *optval, int optlen) -{ - int r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = setsockopt(TO_SOCKET(s), level, optname, optval, optlen); - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef shutdown - -int WSAAPI -rb_w32_shutdown(int s, int how) -{ - int r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = shutdown(TO_SOCKET(s), how); - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -static SOCKET -open_ifs_socket(int af, int type, int protocol) -{ - unsigned long proto_buffers_len = 0; - int error_code; - SOCKET out = INVALID_SOCKET; - - if (WSAEnumProtocols(NULL, NULL, &proto_buffers_len) == SOCKET_ERROR) { - error_code = WSAGetLastError(); - if (error_code == WSAENOBUFS) { - WSAPROTOCOL_INFO *proto_buffers; - int protocols_available = 0; - - proto_buffers = (WSAPROTOCOL_INFO *)malloc(proto_buffers_len); - if (!proto_buffers) { - WSASetLastError(WSA_NOT_ENOUGH_MEMORY); - return INVALID_SOCKET; - } - - protocols_available = - WSAEnumProtocols(NULL, proto_buffers, &proto_buffers_len); - if (protocols_available != SOCKET_ERROR) { - int i; - for (i = 0; i < protocols_available; i++) { - if ((af != AF_UNSPEC && af != proto_buffers[i].iAddressFamily) || - (type != proto_buffers[i].iSocketType) || - (protocol != 0 && protocol != proto_buffers[i].iProtocol)) - continue; - - if ((proto_buffers[i].dwServiceFlags1 & XP1_IFS_HANDLES) == 0) - continue; - - out = WSASocket(af, type, protocol, &(proto_buffers[i]), 0, - WSA_FLAG_OVERLAPPED); - break; - } - } - - free(proto_buffers); - } - } - - return out; -} - -#undef socket - -int WSAAPI -rb_w32_socket(int af, int type, int protocol) -{ - SOCKET s; - int fd; - - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - s = open_ifs_socket(af, type, protocol); - if (s == INVALID_SOCKET) { - errno = map_errno(WSAGetLastError()); - fd = -1; - } - else { - fd = rb_w32_open_osfhandle(s, O_RDWR|O_BINARY|O_NOINHERIT); - if (fd != -1) - st_insert(socklist, (st_data_t)s, (st_data_t)0); - else - closesocket(s); - } - }); - return fd; -} - -#undef gethostbyaddr - -struct hostent * WSAAPI -rb_w32_gethostbyaddr(const char *addr, int len, int type) -{ - struct hostent *r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = gethostbyaddr(addr, len, type); - if (r == NULL) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef gethostbyname - -struct hostent * WSAAPI -rb_w32_gethostbyname(const char *name) -{ - struct hostent *r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = gethostbyname(name); - if (r == NULL) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef gethostname - -int WSAAPI -rb_w32_gethostname(char *name, int len) -{ - int r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = gethostname(name, len); - if (r == SOCKET_ERROR) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef getprotobyname - -struct protoent * WSAAPI -rb_w32_getprotobyname(const char *name) -{ - struct protoent *r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = getprotobyname(name); - if (r == NULL) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef getprotobynumber - -struct protoent * WSAAPI -rb_w32_getprotobynumber(int num) -{ - struct protoent *r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = getprotobynumber(num); - if (r == NULL) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef getservbyname - -struct servent * WSAAPI -rb_w32_getservbyname(const char *name, const char *proto) -{ - struct servent *r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = getservbyname(name, proto); - if (r == NULL) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -#undef getservbyport - -struct servent * WSAAPI -rb_w32_getservbyport(int port, const char *proto) -{ - struct servent *r; - if (!NtSocketsInitialized) { - StartSockets(); - } - RUBY_CRITICAL({ - r = getservbyport(port, proto); - if (r == NULL) - errno = map_errno(WSAGetLastError()); - }); - return r; -} - -static int -socketpair_internal(int af, int type, int protocol, SOCKET *sv) -{ - SOCKET svr = INVALID_SOCKET, r = INVALID_SOCKET, w = INVALID_SOCKET; - struct sockaddr_in sock_in4; -#ifdef INET6 - struct sockaddr_in6 sock_in6; -#endif - struct sockaddr *addr; - int ret = -1; - int len; - - if (!NtSocketsInitialized) { - StartSockets(); - } - - switch (af) { - case AF_INET: -#if defined PF_INET && PF_INET != AF_INET - case PF_INET: -#endif - sock_in4.sin_family = AF_INET; - sock_in4.sin_port = 0; - sock_in4.sin_addr.s_addr = htonl(INADDR_LOOPBACK); - addr = (struct sockaddr *)&sock_in4; - len = sizeof(sock_in4); - break; -#ifdef INET6 - case AF_INET6: - memset(&sock_in6, 0, sizeof(sock_in6)); - sock_in6.sin6_family = AF_INET6; - sock_in6.sin6_addr = IN6ADDR_LOOPBACK_INIT; - addr = (struct sockaddr *)&sock_in6; - len = sizeof(sock_in6); - break; -#endif - default: - errno = EAFNOSUPPORT; - return -1; - } - if (type != SOCK_STREAM) { - errno = EPROTOTYPE; - return -1; - } - - RUBY_CRITICAL({ - do { - svr = open_ifs_socket(af, type, protocol); - if (svr == INVALID_SOCKET) - break; - if (bind(svr, addr, len) < 0) - break; - if (getsockname(svr, addr, &len) < 0) - break; - if (type == SOCK_STREAM) - listen(svr, 5); - - w = open_ifs_socket(af, type, protocol); - if (w == INVALID_SOCKET) - break; - if (connect(w, addr, len) < 0) - break; - - r = accept(svr, addr, &len); - if (r == INVALID_SOCKET) - break; - - ret = 0; - } while (0); - - if (ret < 0) { - errno = map_errno(WSAGetLastError()); - if (r != INVALID_SOCKET) - closesocket(r); - if (w != INVALID_SOCKET) - closesocket(w); - } - else { - sv[0] = r; - sv[1] = w; - } - if (svr != INVALID_SOCKET) - closesocket(svr); - }); - - return ret; -} - -int -rb_w32_socketpair(int af, int type, int protocol, int *sv) -{ - SOCKET pair[2]; - - if (socketpair_internal(af, type, protocol, pair) < 0) - return -1; - sv[0] = rb_w32_open_osfhandle(pair[0], O_RDWR|O_BINARY|O_NOINHERIT); - if (sv[0] == -1) { - closesocket(pair[0]); - closesocket(pair[1]); - return -1; - } - sv[1] = rb_w32_open_osfhandle(pair[1], O_RDWR|O_BINARY|O_NOINHERIT); - if (sv[1] == -1) { - rb_w32_close(sv[0]); - closesocket(pair[1]); - return -1; - } - st_insert(socklist, (st_data_t)pair[0], (st_data_t)0); - st_insert(socklist, (st_data_t)pair[1], (st_data_t)0); - - return 0; -} - -// -// Networking stubs -// - -void endhostent(void) {} -void endnetent(void) {} -void endprotoent(void) {} -void endservent(void) {} - -struct netent *getnetent (void) {return (struct netent *) NULL;} - -struct netent *getnetbyaddr(long net, int type) {return (struct netent *)NULL;} - -struct netent *getnetbyname(const char *name) {return (struct netent *)NULL;} - -struct protoent *getprotoent (void) {return (struct protoent *) NULL;} - -struct servent *getservent (void) {return (struct servent *) NULL;} - -void sethostent (int stayopen) {} - -void setnetent (int stayopen) {} - -void setprotoent (int stayopen) {} - -void setservent (int stayopen) {} - -int -fcntl(int fd, int cmd, ...) -{ - SOCKET sock = TO_SOCKET(fd); - va_list va; - int arg; - int ret; - int flag = 0; - st_data_t data; - u_long ioctlArg; - - if (!is_socket(sock)) { - errno = EBADF; - return -1; - } - if (cmd != F_SETFL) { - errno = EINVAL; - return -1; - } - - va_start(va, cmd); - arg = va_arg(va, int); - va_end(va); - st_lookup(socklist, (st_data_t)sock, &data); - flag = (int)data; - if (arg & O_NONBLOCK) { - flag |= O_NONBLOCK; - ioctlArg = 1; - } - else { - flag &= ~O_NONBLOCK; - ioctlArg = 0; - } - RUBY_CRITICAL({ - ret = ioctlsocket(sock, FIONBIO, &ioctlArg); - if (ret == 0) - st_insert(socklist, (st_data_t)sock, (st_data_t)flag); - else - errno = map_errno(WSAGetLastError()); - }); - - return ret; -} - -#ifndef WNOHANG -#define WNOHANG -1 -#endif - -static rb_pid_t -poll_child_status(struct ChildRecord *child, int *stat_loc) -{ - DWORD exitcode; - DWORD err; - - if (!GetExitCodeProcess(child->hProcess, &exitcode)) { - /* If an error occured, return immediatly. */ - err = GetLastError(); - if (err == ERROR_INVALID_PARAMETER) - errno = ECHILD; - else { - if (GetLastError() == ERROR_INVALID_HANDLE) - errno = EINVAL; - else - errno = map_errno(GetLastError()); - } - CloseChildHandle(child); - return -1; - } - if (exitcode != STILL_ACTIVE) { - /* If already died, return immediatly. */ - rb_pid_t pid = child->pid; - CloseChildHandle(child); - if (stat_loc) *stat_loc = exitcode << 8; - return pid; - } - return 0; -} - -rb_pid_t -waitpid(rb_pid_t pid, int *stat_loc, int options) -{ - DWORD timeout; - - if (options == WNOHANG) { - timeout = 0; - } else { - timeout = INFINITE; - } - - if (pid == -1) { - int count = 0; - DWORD ret; - HANDLE events[MAXCHILDNUM]; - - FOREACH_CHILD(child) { - if (!child->pid || child->pid < 0) continue; - if ((pid = poll_child_status(child, stat_loc))) return pid; - events[count++] = child->hProcess; - } END_FOREACH_CHILD; - if (!count) { - errno = ECHILD; - return -1; - } - - ret = rb_w32_wait_events_blocking(events, count, timeout); - if (ret == WAIT_TIMEOUT) return 0; - if ((ret -= WAIT_OBJECT_0) == count) { - return -1; - } - if (ret > count) { - errno = map_errno(GetLastError()); - return -1; - } - - return poll_child_status(ChildRecord + ret, stat_loc); - } - else { - struct ChildRecord* child = FindChildSlot(pid); - if (!child) { - errno = ECHILD; - return -1; - } - - while (!(pid = poll_child_status(child, stat_loc))) { - /* wait... */ - if (rb_w32_wait_events_blocking(&child->hProcess, 1, timeout) != WAIT_OBJECT_0) { - /* still active */ - pid = 0; - break; - } - } - } - - return pid; -} - -#include <sys/timeb.h> - -int _cdecl -gettimeofday(struct timeval *tv, struct timezone *tz) -{ - SYSTEMTIME st; - time_t t; - struct tm tm; - - GetLocalTime(&st); - tm.tm_sec = st.wSecond; - tm.tm_min = st.wMinute; - tm.tm_hour = st.wHour; - tm.tm_mday = st.wDay; - tm.tm_mon = st.wMonth - 1; - tm.tm_year = st.wYear - 1900; - tm.tm_isdst = -1; - t = mktime(&tm); - tv->tv_sec = t; - tv->tv_usec = st.wMilliseconds * 1000; - - return 0; -} - -char * -rb_w32_getcwd(char *buffer, int size) -{ - char *p = buffer; - char *bp; - int len; - - len = GetCurrentDirectory(0, NULL); - if (!len) { - errno = map_errno(GetLastError()); - return NULL; - } - - if (p) { - if (size < len) { - errno = ERANGE; - return NULL; - } - } - else { - p = malloc(len); - size = len; - if (!p) { - errno = ENOMEM; - return NULL; - } - } - - if (!GetCurrentDirectory(size, p)) { - errno = map_errno(GetLastError()); - if (!buffer) - free(p); - return NULL; - } - - for (bp = p; *bp != '\0'; bp = CharNext(bp)) { - if (*bp == '\\') { - *bp = '/'; - } - } - - return p; -} - -int -chown(const char *path, int owner, int group) -{ - return 0; -} - -int -kill(int pid, int sig) -{ - int ret = 0; - DWORD err; - - if (pid <= 0) { - errno = EINVAL; - return -1; - } - - if (IsWin95()) pid = -pid; - if ((unsigned int)pid == GetCurrentProcessId() && - (sig != 0 && sig != SIGKILL)) { - if ((ret = raise(sig)) != 0) { - /* MSVCRT doesn't set errno... */ - errno = EINVAL; - } - return ret; - } - - switch (sig) { - case 0: - RUBY_CRITICAL({ - HANDLE hProc = - OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, (DWORD)pid); - if (hProc == NULL || hProc == INVALID_HANDLE_VALUE) { - if (GetLastError() == ERROR_INVALID_PARAMETER) { - errno = ESRCH; - } - else { - errno = EPERM; - } - ret = -1; - } - else { - CloseHandle(hProc); - } - }); - break; - - case SIGINT: - RUBY_CRITICAL({ - if (!GenerateConsoleCtrlEvent(CTRL_C_EVENT, (DWORD)pid)) { - if ((err = GetLastError()) == 0) - errno = EPERM; - else - errno = map_errno(GetLastError()); - ret = -1; - } - }); - break; - - case SIGKILL: - RUBY_CRITICAL({ - HANDLE hProc = OpenProcess(PROCESS_TERMINATE, FALSE, (DWORD)pid); - if (hProc == NULL || hProc == INVALID_HANDLE_VALUE) { - if (GetLastError() == ERROR_INVALID_PARAMETER) { - errno = ESRCH; - } - else { - errno = EPERM; - } - ret = -1; - } - else { - if (!TerminateProcess(hProc, 0)) { - errno = EPERM; - ret = -1; - } - CloseHandle(hProc); - } - }); - break; - - default: - errno = EINVAL; - ret = -1; - break; - } - - return ret; -} - -int -link(const char *from, const char *to) -{ - static BOOL (WINAPI *pCreateHardLink)(LPCTSTR, LPCTSTR, LPSECURITY_ATTRIBUTES) = NULL; - static int myerrno = 0; - - if (!pCreateHardLink && !myerrno) { - HANDLE hKernel; - - hKernel = GetModuleHandle("kernel32.dll"); - if (hKernel) { - pCreateHardLink = (BOOL (WINAPI *)(LPCTSTR, LPCTSTR, LPSECURITY_ATTRIBUTES))GetProcAddress(hKernel, "CreateHardLinkA"); - if (!pCreateHardLink) { - rb_notimplement(); - } - } - else { - myerrno = map_errno(GetLastError()); - } - } - if (!pCreateHardLink) { - errno = myerrno; - return -1; - } - - if (!pCreateHardLink(to, from, NULL)) { - errno = map_errno(GetLastError()); - return -1; - } - - return 0; -} - -int -wait(int *status) -{ - return waitpid(-1, status, 0); -} - -char * -rb_w32_getenv(const char *name) -{ - int len = strlen(name); - char *env; - - if (envarea) - FreeEnvironmentStrings(envarea); - envarea = GetEnvironmentStrings(); - if (!envarea) { - map_errno(GetLastError()); - return NULL; - } - - for (env = envarea; *env; env += strlen(env) + 1) - if (strncasecmp(env, name, len) == 0 && *(env + len) == '=') - return env + len + 1; - - return NULL; -} - -int -rb_w32_rename(const char *oldpath, const char *newpath) -{ - int res = 0; - int oldatts; - int newatts; - - oldatts = GetFileAttributes(oldpath); - newatts = GetFileAttributes(newpath); - - if (oldatts == -1) { - errno = map_errno(GetLastError()); - return -1; - } - - RUBY_CRITICAL({ - if (newatts != -1 && newatts & FILE_ATTRIBUTE_READONLY) - SetFileAttributesA(newpath, newatts & ~ FILE_ATTRIBUTE_READONLY); - - if (!MoveFile(oldpath, newpath)) - res = -1; - - if (res) { - switch (GetLastError()) { - case ERROR_ALREADY_EXISTS: - case ERROR_FILE_EXISTS: - if (IsWinNT()) { - if (MoveFileEx(oldpath, newpath, MOVEFILE_REPLACE_EXISTING)) - res = 0; - } else { - for (;;) { - if (!DeleteFile(newpath) && GetLastError() != ERROR_FILE_NOT_FOUND) - break; - else if (MoveFile(oldpath, newpath)) { - res = 0; - break; - } - } - } - } - } - - if (res) - errno = map_errno(GetLastError()); - else - SetFileAttributes(newpath, oldatts); - }); - - return res; -} - -static int -isUNCRoot(const char *path) -{ - if (path[0] == '\\' && path[1] == '\\') { - const char *p; - for (p = path + 2; *p; p = CharNext(p)) { - if (*p == '\\') - break; - } - if (p[0] && p[1]) { - for (p++; *p; p = CharNext(p)) { - if (*p == '\\') - break; - } - if (!p[0] || !p[1] || (p[1] == '.' && !p[2])) - return 1; - } - } - return 0; -} - -#define COPY_STAT(src, dest) do { \ - (dest).st_dev = (src).st_dev; \ - (dest).st_ino = (src).st_ino; \ - (dest).st_mode = (src).st_mode; \ - (dest).st_nlink = (src).st_nlink; \ - (dest).st_uid = (src).st_uid; \ - (dest).st_gid = (src).st_gid; \ - (dest).st_rdev = (src).st_rdev; \ - (dest).st_size = (src).st_size; \ - (dest).st_atime = (src).st_atime; \ - (dest).st_mtime = (src).st_mtime; \ - (dest).st_ctime = (src).st_ctime; \ - } while (0) - -#ifdef __BORLANDC__ -#undef fstat -int -rb_w32_fstat(int fd, struct stat *st) -{ - BY_HANDLE_FILE_INFORMATION info; - int ret = fstat(fd, st); - - if (ret) return ret; - st->st_mode &= ~(S_IWGRP | S_IWOTH); - if (GetFileInformationByHandle((HANDLE)_get_osfhandle(fd), &info) && - !(info.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) { - st->st_mode |= S_IWUSR; - } - return ret; -} - -int -rb_w32_fstati64(int fd, struct stati64 *st) -{ - BY_HANDLE_FILE_INFORMATION info; - struct stat tmp; - int ret = fstat(fd, &tmp); - - if (ret) return ret; - tmp.st_mode &= ~(S_IWGRP | S_IWOTH); - COPY_STAT(tmp, *st); - if (GetFileInformationByHandle((HANDLE)_get_osfhandle(fd), &info)) { - if (!(info.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) { - st->st_mode |= S_IWUSR; - } - st->st_size = ((__int64)info.nFileSizeHigh << 32) | info.nFileSizeLow; - } - return ret; -} -#endif - -static time_t -filetime_to_unixtime(const FILETIME *ft) -{ - FILETIME loc; - SYSTEMTIME st; - struct tm tm; - time_t t; - - if (!FileTimeToLocalFileTime(ft, &loc)) { - return 0; - } - if (!FileTimeToSystemTime(&loc, &st)) { - return 0; - } - memset(&tm, 0, sizeof(tm)); - tm.tm_year = st.wYear - 1900; - tm.tm_mon = st.wMonth - 1; - tm.tm_mday = st.wDay; - tm.tm_hour = st.wHour; - tm.tm_min = st.wMinute; - tm.tm_sec = st.wSecond; - tm.tm_isdst = -1; - t = mktime(&tm); - return t == -1 ? 0 : t; -} - -static unsigned -fileattr_to_unixmode(DWORD attr, const char *path) -{ - unsigned mode = 0; - - if (attr & FILE_ATTRIBUTE_READONLY) { - mode |= S_IREAD; - } - else { - mode |= S_IREAD | S_IWRITE | S_IWUSR; - } - - if (attr & FILE_ATTRIBUTE_DIRECTORY) { - mode |= S_IFDIR | S_IEXEC; - } - else { - mode |= S_IFREG; - } - - if (path && (mode & S_IFREG)) { - const char *end = path + strlen(path); - while (path < end) { - end = CharPrev(path, end); - if (*end == '.') { - if ((strcmpi(end, ".bat") == 0) || - (strcmpi(end, ".cmd") == 0) || - (strcmpi(end, ".com") == 0) || - (strcmpi(end, ".exe") == 0)) { - mode |= S_IEXEC; - } - break; - } - } - } - - mode |= (mode & 0700) >> 3; - mode |= (mode & 0700) >> 6; - - return mode; -} - -static int -winnt_stat(const char *path, struct stati64 *st) -{ - HANDLE h; - WIN32_FIND_DATA wfd; - - memset(st, 0, sizeof(*st)); - st->st_nlink = 1; - - if (_mbspbrk(path, "?*")) { - errno = ENOENT; - return -1; - } - h = FindFirstFile(path, &wfd); - if (h != INVALID_HANDLE_VALUE) { - FindClose(h); - st->st_mode = fileattr_to_unixmode(wfd.dwFileAttributes, path); - st->st_atime = filetime_to_unixtime(&wfd.ftLastAccessTime); - st->st_mtime = filetime_to_unixtime(&wfd.ftLastWriteTime); - st->st_ctime = filetime_to_unixtime(&wfd.ftCreationTime); - st->st_size = ((__int64)wfd.nFileSizeHigh << 32) | wfd.nFileSizeLow; - } - else { - // If runtime stat(2) is called for network shares, it fails on WinNT. - // Because GetDriveType returns 1 for network shares. (Win98 returns 4) - DWORD attr = GetFileAttributes(path); - if (attr == -1) { - errno = map_errno(GetLastError()); - return -1; - } - st->st_mode = fileattr_to_unixmode(attr, path); - } - - st->st_dev = st->st_rdev = (isalpha(path[0]) && path[1] == ':') ? - toupper(path[0]) - 'A' : _getdrive() - 1; - - return 0; -} - -int -rb_w32_stat(const char *path, struct stat *st) -{ - struct stati64 tmp; - - if (rb_w32_stati64(path, &tmp)) return -1; - COPY_STAT(tmp, *st); - return 0; -} - -int -rb_w32_stati64(const char *path, struct stati64 *st) -{ - const char *p; - char *buf1, *s, *end; - int len, size; - int ret; - - if (!path || !st) { - errno = EFAULT; - return -1; - } - size = strlen(path) + 2; - buf1 = ALLOCA_N(char, size); - for (p = path, s = buf1; *p; p++, s++) { - if (*p == '/') - *s = '\\'; - else - *s = *p; - } - *s = '\0'; - len = s - buf1; - if (!len || '\"' == *(--s)) { - errno = ENOENT; - return -1; - } - end = CharPrev(buf1, buf1 + len); - - if (isUNCRoot(buf1)) { - if (*end == '.') - *end = '\0'; - else if (*end != '\\') - strlcat(buf1, "\\", size); - } - else if (*end == '\\' || (buf1 + 1 == end && *end == ':')) - strlcat(buf1, ".", size); - - ret = IsWinNT() ? winnt_stat(buf1, st) : stati64(buf1, st); - if (ret == 0) { - st->st_mode &= ~(S_IWGRP | S_IWOTH); - } - return ret; -} - -static int -rb_chsize(HANDLE h, off_t size) -{ - long upos, lpos, usize, lsize, uend, lend; - off_t end; - int ret = -1; - DWORD e; - - if (((lpos = SetFilePointer(h, 0, (upos = 0, &upos), SEEK_CUR)) == -1L && - (e = GetLastError())) || - ((lend = GetFileSize(h, (DWORD *)&uend)) == -1L && (e = GetLastError()))) { - errno = map_errno(e); - return -1; - } - end = ((off_t)uend << 32) | (unsigned long)lend; - usize = (long)(size >> 32); - lsize = (long)size; - if (SetFilePointer(h, lsize, &usize, SEEK_SET) == -1L && - (e = GetLastError())) { - errno = map_errno(e); - } - else if (!SetEndOfFile(h)) { - errno = map_errno(GetLastError()); - } - else { - ret = 0; - } - SetFilePointer(h, lpos, &upos, SEEK_SET); - return ret; -} - -int -truncate(const char *path, off_t length) -{ - HANDLE h; - int ret; - if (IsWin95()) { - int fd = open(path, O_WRONLY), e = 0; - if (fd == -1) return -1; - ret = chsize(fd, (unsigned long)length); - if (ret == -1) e = errno; - close(fd); - if (ret == -1) errno = e; - return ret; - } - h = CreateFile(path, GENERIC_WRITE, 0, 0, OPEN_EXISTING, 0, 0); - if (h == INVALID_HANDLE_VALUE) { - errno = map_errno(GetLastError()); - return -1; - } - ret = rb_chsize(h, length); - CloseHandle(h); - return ret; -} - -int -ftruncate(int fd, off_t length) -{ - long h; - - if (IsWin95()) { - return chsize(fd, (unsigned long)length); - } - h = _get_osfhandle(fd); - if (h == -1) return -1; - return rb_chsize((HANDLE)h, length); -} - -#ifdef __BORLANDC__ -off_t -_filelengthi64(int fd) -{ - DWORD u, l; - int e; - - l = GetFileSize((HANDLE)_get_osfhandle(fd), &u); - if (l == (DWORD)-1L && (e = GetLastError())) { - errno = map_errno(e); - return (off_t)-1; - } - return ((off_t)u << 32) | l; -} - -off_t -_lseeki64(int fd, off_t offset, int whence) -{ - long u, l; - int e; - HANDLE h = (HANDLE)_get_osfhandle(fd); - - if (!h) { - errno = EBADF; - return -1; - } - u = (long)(offset >> 32); - if ((l = SetFilePointer(h, (long)offset, &u, whence)) == -1L && - (e = GetLastError())) { - errno = map_errno(e); - return -1; - } - return ((off_t)u << 32) | l; -} -#endif - -int -fseeko(FILE *stream, off_t offset, int whence) -{ - off_t pos; - switch (whence) { - case SEEK_CUR: - if (fgetpos(stream, (fpos_t *)&pos)) - return -1; - pos += offset; - break; - case SEEK_END: - if ((pos = _filelengthi64(fileno(stream))) == (off_t)-1) - return -1; - pos += offset; - break; - default: - pos = offset; - break; - } - return fsetpos(stream, (fpos_t *)&pos); -} - -off_t -ftello(FILE *stream) -{ - off_t pos; - if (fgetpos(stream, (fpos_t *)&pos)) return (off_t)-1; - return pos; -} - -static long -filetime_to_clock(FILETIME *ft) -{ - __int64 qw = ft->dwHighDateTime; - qw <<= 32; - qw |= ft->dwLowDateTime; - qw /= 10000; /* File time ticks at 0.1uS, clock at 1mS */ - return (long) qw; -} - -int -rb_w32_times(struct tms *tmbuf) -{ - FILETIME create, exit, kernel, user; - - if (GetProcessTimes(GetCurrentProcess(),&create, &exit, &kernel, &user)) { - tmbuf->tms_utime = filetime_to_clock(&user); - tmbuf->tms_stime = filetime_to_clock(&kernel); - tmbuf->tms_cutime = 0; - tmbuf->tms_cstime = 0; - } - else { - tmbuf->tms_utime = clock(); - tmbuf->tms_stime = 0; - tmbuf->tms_cutime = 0; - tmbuf->tms_cstime = 0; - } - return 0; -} - -#define yield_once() Sleep(0) -#define yield_until(condition) do yield_once(); while (!(condition)) - -static void -catch_interrupt(void) -{ - yield_once(); - RUBY_CRITICAL(rb_w32_wait_events(NULL, 0, 0)); -} - -#if defined __BORLANDC__ || defined _WIN32_WCE -#undef read -int -read(int fd, void *buf, size_t size) -{ - int trap_immediate = rb_trap_immediate; - int ret = _read(fd, buf, size); - if ((ret < 0) && (errno == EPIPE)) { - errno = 0; - ret = 0; - } - rb_trap_immediate = trap_immediate; - catch_interrupt(); - return ret; -} -#endif - -#undef fgetc -int -rb_w32_getc(FILE* stream) -{ - int c, trap_immediate = rb_trap_immediate; -#ifndef _WIN32_WCE - if (enough_to_get(stream->FILE_COUNT)) { - c = (unsigned char)*stream->FILE_READPTR++; - rb_trap_immediate = trap_immediate; - } - else -#endif - { - c = _filbuf(stream); -#if defined __BORLANDC__ || defined _WIN32_WCE - if ((c == EOF) && (errno == EPIPE)) { - clearerr(stream); - } -#endif - rb_trap_immediate = trap_immediate; - catch_interrupt(); - } - return c; -} - -#undef fputc -int -rb_w32_putc(int c, FILE* stream) -{ - int trap_immediate = rb_trap_immediate; -#ifndef _WIN32_WCE - if (enough_to_put(stream->FILE_COUNT)) { - c = (unsigned char)(*stream->FILE_READPTR++ = (char)c); - rb_trap_immediate = trap_immediate; - } - else -#endif - { - c = _flsbuf(c, stream); - rb_trap_immediate = trap_immediate; - catch_interrupt(); - } - return c; -} - -struct asynchronous_arg_t { - /* output field */ - void* stackaddr; - int errnum; - - /* input field */ - uintptr_t (*func)(uintptr_t self, int argc, uintptr_t* argv); - uintptr_t self; - int argc; - uintptr_t* argv; -}; - -static DWORD WINAPI -call_asynchronous(PVOID argp) -{ - DWORD ret; - struct asynchronous_arg_t *arg = argp; - arg->stackaddr = &argp; - ret = (DWORD)arg->func(arg->self, arg->argc, arg->argv); - arg->errnum = errno; - return ret; -} - -uintptr_t -rb_w32_asynchronize(asynchronous_func_t func, uintptr_t self, - int argc, uintptr_t* argv, uintptr_t intrval) -{ - DWORD val; - BOOL interrupted = FALSE; - HANDLE thr; - - RUBY_CRITICAL({ - struct asynchronous_arg_t arg; - - arg.stackaddr = NULL; - arg.errnum = 0; - arg.func = func; - arg.self = self; - arg.argc = argc; - arg.argv = argv; - - thr = CreateThread(NULL, 0, call_asynchronous, &arg, 0, &val); - - if (thr) { - yield_until(arg.stackaddr); - - if (rb_w32_wait_events_blocking(&thr, 1, INFINITE) != WAIT_OBJECT_0) { - interrupted = TRUE; - - if (TerminateThread(thr, intrval)) { - yield_once(); - } - } - - GetExitCodeThread(thr, &val); - CloseHandle(thr); - - if (interrupted) { - /* must release stack of killed thread, why doesn't Windows? */ - MEMORY_BASIC_INFORMATION m; - - memset(&m, 0, sizeof(m)); - if (!VirtualQuery(arg.stackaddr, &m, sizeof(m))) { - Debug(fprintf(stderr, "couldn't get stack base:%p:%d\n", - arg.stackaddr, GetLastError())); - } - else if (!VirtualFree(m.AllocationBase, 0, MEM_RELEASE)) { - Debug(fprintf(stderr, "couldn't release stack:%p:%d\n", - m.AllocationBase, GetLastError())); - } - errno = EINTR; - } - else { - errno = arg.errnum; - } - } - }); - - if (!thr) { - rb_fatal("failed to launch waiter thread:%ld", GetLastError()); - } - - return val; -} - -char ** -rb_w32_get_environ(void) -{ - char *envtop, *env; - char **myenvtop, **myenv; - int num; - - /* - * We avoid values started with `='. If you want to deal those values, - * change this function, and some functions in hash.c which recognize - * `=' as delimiter or rb_w32_getenv() and ruby_setenv(). - * CygWin deals these values by changing first `=' to '!'. But we don't - * use such trick and follow cmd.exe's way that just doesn't show these - * values. - * (U.N. 2001-11-15) - */ - envtop = GetEnvironmentStrings(); - for (env = envtop, num = 0; *env; env += strlen(env) + 1) - if (*env != '=') num++; - - myenvtop = (char **)malloc(sizeof(char *) * (num + 1)); - for (env = envtop, myenv = myenvtop; *env; env += strlen(env) + 1) { - if (*env != '=') { - if (!(*myenv = strdup(env))) { - break; - } - myenv++; - } - } - *myenv = NULL; - FreeEnvironmentStrings(envtop); - - return myenvtop; -} - -void -rb_w32_free_environ(char **env) -{ - char **t = env; - - while (*t) free(*t++); - free(env); -} - -#undef getpid -rb_pid_t -rb_w32_getpid(void) -{ - rb_pid_t pid; - - pid = getpid(); - - if (IsWin95()) pid = -pid; - - return pid; -} - - -rb_pid_t -rb_w32_getppid(void) -{ - static long (WINAPI *pNtQueryInformationProcess)(HANDLE, int, void *, ULONG, ULONG *) = NULL; - rb_pid_t ppid = 0; - - if (!IsWin95() && rb_w32_osver() >= 5) { - if (!pNtQueryInformationProcess) { - HANDLE hNtDll = GetModuleHandle("ntdll.dll"); - if (hNtDll) { - pNtQueryInformationProcess = (long (WINAPI *)(HANDLE, int, void *, ULONG, ULONG *))GetProcAddress(hNtDll, "NtQueryInformationProcess"); - if (pNtQueryInformationProcess) { - struct { - long ExitStatus; - void* PebBaseAddress; - ULONG AffinityMask; - ULONG BasePriority; - ULONG UniqueProcessId; - ULONG ParentProcessId; - } pbi; - ULONG len; - long ret = pNtQueryInformationProcess(GetCurrentProcess(), 0, &pbi, sizeof(pbi), &len); - if (!ret) { - ppid = pbi.ParentProcessId; - } - } - } - } - } - - return ppid; -} - -int -rb_w32_fclose(FILE *fp) -{ - int fd = fileno(fp); - SOCKET sock = TO_SOCKET(fd); - int save_errno = errno; - - if (fflush(fp)) return -1; - if (!is_socket(sock)) { - UnlockFile((HANDLE)sock, 0, 0, LK_LEN, LK_LEN); - return fclose(fp); - } - _set_osfhnd(fd, (SOCKET)INVALID_HANDLE_VALUE); - fclose(fp); - errno = save_errno; - if (closesocket(sock) == SOCKET_ERROR) { - errno = map_errno(WSAGetLastError()); - return -1; - } - return 0; -} - -int -rb_w32_close(int fd) -{ - SOCKET sock = TO_SOCKET(fd); - int save_errno = errno; - st_data_t key; - - if (!is_socket(sock)) { - UnlockFile((HANDLE)sock, 0, 0, LK_LEN, LK_LEN); - return _close(fd); - } - _set_osfhnd(fd, (SOCKET)INVALID_HANDLE_VALUE); - key = (st_data_t)sock; - st_delete(socklist, &key, NULL); - sock = (SOCKET)key; - _close(fd); - errno = save_errno; - if (closesocket(sock) == SOCKET_ERROR) { - errno = map_errno(WSAGetLastError()); - return -1; - } - return 0; -} - -#undef read -size_t -rb_w32_read(int fd, void *buf, size_t size) -{ - SOCKET sock = TO_SOCKET(fd); - - if (!is_socket(sock)) - return read(fd, buf, size); - else - return rb_w32_recv(fd, buf, size, 0); -} - -#undef write -size_t -rb_w32_write(int fd, const void *buf, size_t size) -{ - SOCKET sock = TO_SOCKET(fd); - - if (!is_socket(sock)) { - size_t ret = write(fd, buf, size); - if ((int)ret < 0 && errno == EINVAL) - errno = map_errno(GetLastError()); - return ret; - } - else - return rb_w32_send(fd, buf, size, 0); -} - -static int -unixtime_to_filetime(time_t time, FILETIME *ft) -{ - struct tm *tm; - SYSTEMTIME st; - FILETIME lt; - - tm = localtime(&time); - st.wYear = tm->tm_year + 1900; - st.wMonth = tm->tm_mon + 1; - st.wDayOfWeek = tm->tm_wday; - st.wDay = tm->tm_mday; - st.wHour = tm->tm_hour; - st.wMinute = tm->tm_min; - st.wSecond = tm->tm_sec; - st.wMilliseconds = 0; - if (!SystemTimeToFileTime(&st, <) || - !LocalFileTimeToFileTime(<, ft)) { - errno = map_errno(GetLastError()); - return -1; - } - return 0; -} - -int -rb_w32_utime(const char *path, const struct utimbuf *times) -{ - HANDLE hFile; - FILETIME atime, mtime; - struct stati64 stat; - int ret = 0; - - if (rb_w32_stati64(path, &stat)) { - return -1; - } - - if (times) { - if (unixtime_to_filetime(times->actime, &atime)) { - return -1; - } - if (unixtime_to_filetime(times->modtime, &mtime)) { - return -1; - } - } - else { - GetSystemTimeAsFileTime(&atime); - mtime = atime; - } - - RUBY_CRITICAL({ - const DWORD attr = GetFileAttributes(path); - if (attr != (DWORD)-1 && (attr & FILE_ATTRIBUTE_READONLY)) - SetFileAttributes(path, attr & ~FILE_ATTRIBUTE_READONLY); - hFile = CreateFile(path, GENERIC_WRITE, 0, 0, OPEN_EXISTING, - IsWin95() ? 0 : FILE_FLAG_BACKUP_SEMANTICS, 0); - if (hFile == INVALID_HANDLE_VALUE) { - errno = map_errno(GetLastError()); - ret = -1; - } - else { - if (!SetFileTime(hFile, NULL, &atime, &mtime)) { - errno = map_errno(GetLastError()); - ret = -1; - } - CloseHandle(hFile); - } - if (attr != (DWORD)-1 && (attr & FILE_ATTRIBUTE_READONLY)) - SetFileAttributes(path, attr); - }); - - return ret; -} - -int -rb_w32_vsnprintf(char *buf, size_t size, const char *format, va_list va) -{ - int ret = _vsnprintf(buf, size, format, va); - if (size > 0) buf[size - 1] = 0; - return ret; -} - -int -rb_w32_snprintf(char *buf, size_t size, const char *format, ...) -{ - int ret; - va_list va; - - va_start(va, format); - ret = vsnprintf(buf, size, format, va); - va_end(va); - return ret; -} - -int -rb_w32_mkdir(const char *path, int mode) -{ - int ret = -1; - RUBY_CRITICAL(do { - if (CreateDirectory(path, NULL) == FALSE) { - errno = map_errno(GetLastError()); - break; - } - if (chmod(path, mode) == -1) { - RemoveDirectory(path); - break; - } - ret = 0; - } while (0)); - return ret; -} - -int -rb_w32_rmdir(const char *path) -{ - int ret = 0; - RUBY_CRITICAL({ - const DWORD attr = GetFileAttributes(path); - if (attr != (DWORD)-1 && (attr & FILE_ATTRIBUTE_READONLY)) { - SetFileAttributes(path, attr & ~FILE_ATTRIBUTE_READONLY); - } - if (RemoveDirectory(path) == FALSE) { - errno = map_errno(GetLastError()); - ret = -1; - if (attr != (DWORD)-1 && (attr & FILE_ATTRIBUTE_READONLY)) { - SetFileAttributes(path, attr); - } - } - }); - return ret; -} - -int -rb_w32_unlink(const char *path) -{ - int ret = 0; - RUBY_CRITICAL({ - const DWORD attr = GetFileAttributes(path); - if (attr != (DWORD)-1 && (attr & FILE_ATTRIBUTE_READONLY)) { - SetFileAttributes(path, attr & ~FILE_ATTRIBUTE_READONLY); - } - if (DeleteFile(path) == FALSE) { - errno = map_errno(GetLastError()); - ret = -1; - if (attr != (DWORD)-1 && (attr & FILE_ATTRIBUTE_READONLY)) { - SetFileAttributes(path, attr); - } - } - }); - return ret; -} - -#if !defined(__BORLANDC__) && !defined(_WIN32_WCE) -int -rb_w32_isatty(int fd) -{ - if (!(_osfile(fd) & FOPEN)) { - errno = EBADF; - return 0; - } - if (!(_osfile(fd) & FDEV)) { - errno = ENOTTY; - return 0; - } - return 1; -} -#endif - -// -// Fix bcc32's stdio bug -// - -#ifdef __BORLANDC__ -static int -too_many_files(void) -{ - FILE *f; - for (f = _streams; f < _streams + _nfile; f++) { - if (f->fd < 0) return 0; - } - return 1; -} - -#undef fopen -FILE * -rb_w32_fopen(const char *path, const char *mode) -{ - FILE *f = (errno = 0, fopen(path, mode)); - if (f == NULL && errno == 0) { - if (too_many_files()) - errno = EMFILE; - } - return f; -} - -FILE * -rb_w32_fdopen(int handle, const char *type) -{ - FILE *f = (errno = 0, _fdopen(handle, (char *)type)); - if (f == NULL && errno == 0) { - if (handle < 0) - errno = EBADF; - else if (too_many_files()) - errno = EMFILE; - } - return f; -} - -FILE * -rb_w32_fsopen(const char *path, const char *mode, int shflags) -{ - FILE *f = (errno = 0, _fsopen(path, mode, shflags)); - if (f == NULL && errno == 0) { - if (too_many_files()) - errno = EMFILE; - } - return f; -} -#endif |