Removed out-of-date comment in _install_handlers and
[python.git] / Modules / posixmodule.c
blob8f32fd43497a6560f40204cd377e6e995212e87e
2 /* POSIX module implementation */
4 /* This file is also used for Windows NT/MS-Win and OS/2. In that case the
5 module actually calls itself 'nt' or 'os2', not 'posix', and a few
6 functions are either unimplemented or implemented differently. The source
7 assumes that for Windows NT, the macro 'MS_WINDOWS' is defined independent
8 of the compiler used. Different compilers define their own feature
9 test macro, e.g. '__BORLANDC__' or '_MSC_VER'. For OS/2, the compiler
10 independent macro PYOS_OS2 should be defined. On OS/2 the default
11 compiler is assumed to be IBM's VisualAge C++ (VACPP). PYCC_GCC is used
12 as the compiler specific macro for the EMX port of gcc to OS/2. */
14 /* See also ../Dos/dosmodule.c */
16 #ifdef __APPLE__
18 * Step 1 of support for weak-linking a number of symbols existing on
19 * OSX 10.4 and later, see the comment in the #ifdef __APPLE__ block
20 * at the end of this file for more information.
22 # pragma weak lchown
23 # pragma weak statvfs
24 # pragma weak fstatvfs
26 #endif /* __APPLE__ */
28 #define PY_SSIZE_T_CLEAN
30 #include "Python.h"
31 #include "structseq.h"
33 #if defined(__VMS)
34 # include <unixio.h>
35 #endif /* defined(__VMS) */
37 #ifdef __cplusplus
38 extern "C" {
39 #endif
41 PyDoc_STRVAR(posix__doc__,
42 "This module provides access to operating system functionality that is\n\
43 standardized by the C Standard and the POSIX standard (a thinly\n\
44 disguised Unix interface). Refer to the library manual and\n\
45 corresponding Unix manual entries for more information on calls.");
47 #ifndef Py_USING_UNICODE
48 /* This is used in signatures of functions. */
49 #define Py_UNICODE void
50 #endif
52 #if defined(PYOS_OS2)
53 #define INCL_DOS
54 #define INCL_DOSERRORS
55 #define INCL_DOSPROCESS
56 #define INCL_NOPMAPI
57 #include <os2.h>
58 #if defined(PYCC_GCC)
59 #include <ctype.h>
60 #include <io.h>
61 #include <stdio.h>
62 #include <process.h>
63 #endif
64 #include "osdefs.h"
65 #endif
67 #ifdef HAVE_SYS_TYPES_H
68 #include <sys/types.h>
69 #endif /* HAVE_SYS_TYPES_H */
71 #ifdef HAVE_SYS_STAT_H
72 #include <sys/stat.h>
73 #endif /* HAVE_SYS_STAT_H */
75 #ifdef HAVE_SYS_WAIT_H
76 #include <sys/wait.h> /* For WNOHANG */
77 #endif
79 #ifdef HAVE_SIGNAL_H
80 #include <signal.h>
81 #endif
83 #ifdef HAVE_FCNTL_H
84 #include <fcntl.h>
85 #endif /* HAVE_FCNTL_H */
87 #ifdef HAVE_GRP_H
88 #include <grp.h>
89 #endif
91 #ifdef HAVE_SYSEXITS_H
92 #include <sysexits.h>
93 #endif /* HAVE_SYSEXITS_H */
95 #ifdef HAVE_SYS_LOADAVG_H
96 #include <sys/loadavg.h>
97 #endif
99 /* Various compilers have only certain posix functions */
100 /* XXX Gosh I wish these were all moved into pyconfig.h */
101 #if defined(PYCC_VACPP) && defined(PYOS_OS2)
102 #include <process.h>
103 #else
104 #if defined(__WATCOMC__) && !defined(__QNX__) /* Watcom compiler */
105 #define HAVE_GETCWD 1
106 #define HAVE_OPENDIR 1
107 #define HAVE_SYSTEM 1
108 #if defined(__OS2__)
109 #define HAVE_EXECV 1
110 #define HAVE_WAIT 1
111 #endif
112 #include <process.h>
113 #else
114 #ifdef __BORLANDC__ /* Borland compiler */
115 #define HAVE_EXECV 1
116 #define HAVE_GETCWD 1
117 #define HAVE_OPENDIR 1
118 #define HAVE_PIPE 1
119 #define HAVE_POPEN 1
120 #define HAVE_SYSTEM 1
121 #define HAVE_WAIT 1
122 #else
123 #ifdef _MSC_VER /* Microsoft compiler */
124 #define HAVE_GETCWD 1
125 #define HAVE_SPAWNV 1
126 #define HAVE_EXECV 1
127 #define HAVE_PIPE 1
128 #define HAVE_POPEN 1
129 #define HAVE_SYSTEM 1
130 #define HAVE_CWAIT 1
131 #define HAVE_FSYNC 1
132 #define fsync _commit
133 #else
134 #if defined(PYOS_OS2) && defined(PYCC_GCC) || defined(__VMS)
135 /* Everything needed is defined in PC/os2emx/pyconfig.h or vms/pyconfig.h */
136 #else /* all other compilers */
137 /* Unix functions that the configure script doesn't check for */
138 #define HAVE_EXECV 1
139 #define HAVE_FORK 1
140 #if defined(__USLC__) && defined(__SCO_VERSION__) /* SCO UDK Compiler */
141 #define HAVE_FORK1 1
142 #endif
143 #define HAVE_GETCWD 1
144 #define HAVE_GETEGID 1
145 #define HAVE_GETEUID 1
146 #define HAVE_GETGID 1
147 #define HAVE_GETPPID 1
148 #define HAVE_GETUID 1
149 #define HAVE_KILL 1
150 #define HAVE_OPENDIR 1
151 #define HAVE_PIPE 1
152 #ifndef __rtems__
153 #define HAVE_POPEN 1
154 #endif
155 #define HAVE_SYSTEM 1
156 #define HAVE_WAIT 1
157 #define HAVE_TTYNAME 1
158 #endif /* PYOS_OS2 && PYCC_GCC && __VMS */
159 #endif /* _MSC_VER */
160 #endif /* __BORLANDC__ */
161 #endif /* ! __WATCOMC__ || __QNX__ */
162 #endif /* ! __IBMC__ */
164 #ifndef _MSC_VER
166 #if defined(__sgi)&&_COMPILER_VERSION>=700
167 /* declare ctermid_r if compiling with MIPSPro 7.x in ANSI C mode
168 (default) */
169 extern char *ctermid_r(char *);
170 #endif
172 #ifndef HAVE_UNISTD_H
173 #if defined(PYCC_VACPP)
174 extern int mkdir(char *);
175 #else
176 #if ( defined(__WATCOMC__) || defined(_MSC_VER) ) && !defined(__QNX__)
177 extern int mkdir(const char *);
178 #else
179 extern int mkdir(const char *, mode_t);
180 #endif
181 #endif
182 #if defined(__IBMC__) || defined(__IBMCPP__)
183 extern int chdir(char *);
184 extern int rmdir(char *);
185 #else
186 extern int chdir(const char *);
187 extern int rmdir(const char *);
188 #endif
189 #ifdef __BORLANDC__
190 extern int chmod(const char *, int);
191 #else
192 extern int chmod(const char *, mode_t);
193 #endif
194 /*#ifdef HAVE_FCHMOD
195 extern int fchmod(int, mode_t);
196 #endif*/
197 /*#ifdef HAVE_LCHMOD
198 extern int lchmod(const char *, mode_t);
199 #endif*/
200 extern int chown(const char *, uid_t, gid_t);
201 extern char *getcwd(char *, int);
202 extern char *strerror(int);
203 extern int link(const char *, const char *);
204 extern int rename(const char *, const char *);
205 extern int stat(const char *, struct stat *);
206 extern int unlink(const char *);
207 extern int pclose(FILE *);
208 #ifdef HAVE_SYMLINK
209 extern int symlink(const char *, const char *);
210 #endif /* HAVE_SYMLINK */
211 #ifdef HAVE_LSTAT
212 extern int lstat(const char *, struct stat *);
213 #endif /* HAVE_LSTAT */
214 #endif /* !HAVE_UNISTD_H */
216 #endif /* !_MSC_VER */
218 #ifdef HAVE_UTIME_H
219 #include <utime.h>
220 #endif /* HAVE_UTIME_H */
222 #ifdef HAVE_SYS_UTIME_H
223 #include <sys/utime.h>
224 #define HAVE_UTIME_H /* pretend we do for the rest of this file */
225 #endif /* HAVE_SYS_UTIME_H */
227 #ifdef HAVE_SYS_TIMES_H
228 #include <sys/times.h>
229 #endif /* HAVE_SYS_TIMES_H */
231 #ifdef HAVE_SYS_PARAM_H
232 #include <sys/param.h>
233 #endif /* HAVE_SYS_PARAM_H */
235 #ifdef HAVE_SYS_UTSNAME_H
236 #include <sys/utsname.h>
237 #endif /* HAVE_SYS_UTSNAME_H */
239 #ifdef HAVE_DIRENT_H
240 #include <dirent.h>
241 #define NAMLEN(dirent) strlen((dirent)->d_name)
242 #else
243 #if defined(__WATCOMC__) && !defined(__QNX__)
244 #include <direct.h>
245 #define NAMLEN(dirent) strlen((dirent)->d_name)
246 #else
247 #define dirent direct
248 #define NAMLEN(dirent) (dirent)->d_namlen
249 #endif
250 #ifdef HAVE_SYS_NDIR_H
251 #include <sys/ndir.h>
252 #endif
253 #ifdef HAVE_SYS_DIR_H
254 #include <sys/dir.h>
255 #endif
256 #ifdef HAVE_NDIR_H
257 #include <ndir.h>
258 #endif
259 #endif
261 #ifdef _MSC_VER
262 #ifdef HAVE_DIRECT_H
263 #include <direct.h>
264 #endif
265 #ifdef HAVE_IO_H
266 #include <io.h>
267 #endif
268 #ifdef HAVE_PROCESS_H
269 #include <process.h>
270 #endif
271 #include "osdefs.h"
272 #include <windows.h>
273 #include <shellapi.h> /* for ShellExecute() */
274 #define popen _popen
275 #define pclose _pclose
276 #endif /* _MSC_VER */
278 #if defined(PYCC_VACPP) && defined(PYOS_OS2)
279 #include <io.h>
280 #endif /* OS2 */
282 #ifndef MAXPATHLEN
283 #if defined(PATH_MAX) && PATH_MAX > 1024
284 #define MAXPATHLEN PATH_MAX
285 #else
286 #define MAXPATHLEN 1024
287 #endif
288 #endif /* MAXPATHLEN */
290 #ifdef UNION_WAIT
291 /* Emulate some macros on systems that have a union instead of macros */
293 #ifndef WIFEXITED
294 #define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
295 #endif
297 #ifndef WEXITSTATUS
298 #define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
299 #endif
301 #ifndef WTERMSIG
302 #define WTERMSIG(u_wait) ((u_wait).w_termsig)
303 #endif
305 #define WAIT_TYPE union wait
306 #define WAIT_STATUS_INT(s) (s.w_status)
308 #else /* !UNION_WAIT */
309 #define WAIT_TYPE int
310 #define WAIT_STATUS_INT(s) (s)
311 #endif /* UNION_WAIT */
313 /* Don't use the "_r" form if we don't need it (also, won't have a
314 prototype for it, at least on Solaris -- maybe others as well?). */
315 #if defined(HAVE_CTERMID_R) && defined(WITH_THREAD)
316 #define USE_CTERMID_R
317 #endif
319 #if defined(HAVE_TMPNAM_R) && defined(WITH_THREAD)
320 #define USE_TMPNAM_R
321 #endif
323 /* choose the appropriate stat and fstat functions and return structs */
324 #undef STAT
325 #if defined(MS_WIN64) || defined(MS_WINDOWS)
326 # define STAT win32_stat
327 # define FSTAT win32_fstat
328 # define STRUCT_STAT struct win32_stat
329 #else
330 # define STAT stat
331 # define FSTAT fstat
332 # define STRUCT_STAT struct stat
333 #endif
335 #if defined(MAJOR_IN_MKDEV)
336 #include <sys/mkdev.h>
337 #else
338 #if defined(MAJOR_IN_SYSMACROS)
339 #include <sys/sysmacros.h>
340 #endif
341 #if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
342 #include <sys/mkdev.h>
343 #endif
344 #endif
346 /* Return a dictionary corresponding to the POSIX environment table */
347 #ifdef WITH_NEXT_FRAMEWORK
348 /* On Darwin/MacOSX a shared library or framework has no access to
349 ** environ directly, we must obtain it with _NSGetEnviron().
351 #include <crt_externs.h>
352 static char **environ;
353 #elif !defined(_MSC_VER) && ( !defined(__WATCOMC__) || defined(__QNX__) )
354 extern char **environ;
355 #endif /* !_MSC_VER */
357 static PyObject *
358 convertenviron(void)
360 PyObject *d;
361 char **e;
362 d = PyDict_New();
363 if (d == NULL)
364 return NULL;
365 #ifdef WITH_NEXT_FRAMEWORK
366 if (environ == NULL)
367 environ = *_NSGetEnviron();
368 #endif
369 if (environ == NULL)
370 return d;
371 /* This part ignores errors */
372 for (e = environ; *e != NULL; e++) {
373 PyObject *k;
374 PyObject *v;
375 char *p = strchr(*e, '=');
376 if (p == NULL)
377 continue;
378 k = PyString_FromStringAndSize(*e, (int)(p-*e));
379 if (k == NULL) {
380 PyErr_Clear();
381 continue;
383 v = PyString_FromString(p+1);
384 if (v == NULL) {
385 PyErr_Clear();
386 Py_DECREF(k);
387 continue;
389 if (PyDict_GetItem(d, k) == NULL) {
390 if (PyDict_SetItem(d, k, v) != 0)
391 PyErr_Clear();
393 Py_DECREF(k);
394 Py_DECREF(v);
396 #if defined(PYOS_OS2)
398 APIRET rc;
399 char buffer[1024]; /* OS/2 Provides a Documented Max of 1024 Chars */
401 rc = DosQueryExtLIBPATH(buffer, BEGIN_LIBPATH);
402 if (rc == NO_ERROR) { /* (not a type, envname is NOT 'BEGIN_LIBPATH') */
403 PyObject *v = PyString_FromString(buffer);
404 PyDict_SetItemString(d, "BEGINLIBPATH", v);
405 Py_DECREF(v);
407 rc = DosQueryExtLIBPATH(buffer, END_LIBPATH);
408 if (rc == NO_ERROR) { /* (not a typo, envname is NOT 'END_LIBPATH') */
409 PyObject *v = PyString_FromString(buffer);
410 PyDict_SetItemString(d, "ENDLIBPATH", v);
411 Py_DECREF(v);
414 #endif
415 return d;
419 /* Set a POSIX-specific error from errno, and return NULL */
421 static PyObject *
422 posix_error(void)
424 return PyErr_SetFromErrno(PyExc_OSError);
426 static PyObject *
427 posix_error_with_filename(char* name)
429 return PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
432 #ifdef Py_WIN_WIDE_FILENAMES
433 static PyObject *
434 posix_error_with_unicode_filename(Py_UNICODE* name)
436 return PyErr_SetFromErrnoWithUnicodeFilename(PyExc_OSError, name);
438 #endif /* Py_WIN_WIDE_FILENAMES */
441 static PyObject *
442 posix_error_with_allocated_filename(char* name)
444 PyObject *rc = PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
445 PyMem_Free(name);
446 return rc;
449 #ifdef MS_WINDOWS
450 static PyObject *
451 win32_error(char* function, char* filename)
453 /* XXX We should pass the function name along in the future.
454 (_winreg.c also wants to pass the function name.)
455 This would however require an additional param to the
456 Windows error object, which is non-trivial.
458 errno = GetLastError();
459 if (filename)
460 return PyErr_SetFromWindowsErrWithFilename(errno, filename);
461 else
462 return PyErr_SetFromWindowsErr(errno);
465 #ifdef Py_WIN_WIDE_FILENAMES
466 static PyObject *
467 win32_error_unicode(char* function, Py_UNICODE* filename)
469 /* XXX - see win32_error for comments on 'function' */
470 errno = GetLastError();
471 if (filename)
472 return PyErr_SetFromWindowsErrWithUnicodeFilename(errno, filename);
473 else
474 return PyErr_SetFromWindowsErr(errno);
477 static PyObject *_PyUnicode_FromFileSystemEncodedObject(register PyObject *obj)
481 /* Function suitable for O& conversion */
482 static int
483 convert_to_unicode(PyObject *arg, void* _param)
485 PyObject **param = (PyObject**)_param;
486 if (PyUnicode_CheckExact(arg)) {
487 Py_INCREF(arg);
488 *param = arg;
490 else if (PyUnicode_Check(arg)) {
491 /* For a Unicode subtype that's not a Unicode object,
492 return a true Unicode object with the same data. */
493 *param = PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(arg),
494 PyUnicode_GET_SIZE(arg));
495 return *param != NULL;
497 else
498 *param = PyUnicode_FromEncodedObject(arg,
499 Py_FileSystemDefaultEncoding,
500 "strict");
501 return (*param) != NULL;
504 #endif /* Py_WIN_WIDE_FILENAMES */
506 #endif
508 #if defined(PYOS_OS2)
509 /**********************************************************************
510 * Helper Function to Trim and Format OS/2 Messages
511 **********************************************************************/
512 static void
513 os2_formatmsg(char *msgbuf, int msglen, char *reason)
515 msgbuf[msglen] = '\0'; /* OS/2 Doesn't Guarantee a Terminator */
517 if (strlen(msgbuf) > 0) { /* If Non-Empty Msg, Trim CRLF */
518 char *lastc = &msgbuf[ strlen(msgbuf)-1 ];
520 while (lastc > msgbuf && isspace(Py_CHARMASK(*lastc)))
521 *lastc-- = '\0'; /* Trim Trailing Whitespace (CRLF) */
524 /* Add Optional Reason Text */
525 if (reason) {
526 strcat(msgbuf, " : ");
527 strcat(msgbuf, reason);
531 /**********************************************************************
532 * Decode an OS/2 Operating System Error Code
534 * A convenience function to lookup an OS/2 error code and return a
535 * text message we can use to raise a Python exception.
537 * Notes:
538 * The messages for errors returned from the OS/2 kernel reside in
539 * the file OSO001.MSG in the \OS2 directory hierarchy.
541 **********************************************************************/
542 static char *
543 os2_strerror(char *msgbuf, int msgbuflen, int errorcode, char *reason)
545 APIRET rc;
546 ULONG msglen;
548 /* Retrieve Kernel-Related Error Message from OSO001.MSG File */
549 Py_BEGIN_ALLOW_THREADS
550 rc = DosGetMessage(NULL, 0, msgbuf, msgbuflen,
551 errorcode, "oso001.msg", &msglen);
552 Py_END_ALLOW_THREADS
554 if (rc == NO_ERROR)
555 os2_formatmsg(msgbuf, msglen, reason);
556 else
557 PyOS_snprintf(msgbuf, msgbuflen,
558 "unknown OS error #%d", errorcode);
560 return msgbuf;
563 /* Set an OS/2-specific error and return NULL. OS/2 kernel
564 errors are not in a global variable e.g. 'errno' nor are
565 they congruent with posix error numbers. */
567 static PyObject * os2_error(int code)
569 char text[1024];
570 PyObject *v;
572 os2_strerror(text, sizeof(text), code, "");
574 v = Py_BuildValue("(is)", code, text);
575 if (v != NULL) {
576 PyErr_SetObject(PyExc_OSError, v);
577 Py_DECREF(v);
579 return NULL; /* Signal to Python that an Exception is Pending */
582 #endif /* OS2 */
584 /* POSIX generic methods */
586 static PyObject *
587 posix_fildes(PyObject *fdobj, int (*func)(int))
589 int fd;
590 int res;
591 fd = PyObject_AsFileDescriptor(fdobj);
592 if (fd < 0)
593 return NULL;
594 Py_BEGIN_ALLOW_THREADS
595 res = (*func)(fd);
596 Py_END_ALLOW_THREADS
597 if (res < 0)
598 return posix_error();
599 Py_INCREF(Py_None);
600 return Py_None;
603 #ifdef Py_WIN_WIDE_FILENAMES
604 static int
605 unicode_file_names(void)
607 static int canusewide = -1;
608 if (canusewide == -1) {
609 /* As per doc for ::GetVersion(), this is the correct test for
610 the Windows NT family. */
611 canusewide = (GetVersion() < 0x80000000) ? 1 : 0;
613 return canusewide;
615 #endif
617 static PyObject *
618 posix_1str(PyObject *args, char *format, int (*func)(const char*))
620 char *path1 = NULL;
621 int res;
622 if (!PyArg_ParseTuple(args, format,
623 Py_FileSystemDefaultEncoding, &path1))
624 return NULL;
625 Py_BEGIN_ALLOW_THREADS
626 res = (*func)(path1);
627 Py_END_ALLOW_THREADS
628 if (res < 0)
629 return posix_error_with_allocated_filename(path1);
630 PyMem_Free(path1);
631 Py_INCREF(Py_None);
632 return Py_None;
635 static PyObject *
636 posix_2str(PyObject *args,
637 char *format,
638 int (*func)(const char *, const char *))
640 char *path1 = NULL, *path2 = NULL;
641 int res;
642 if (!PyArg_ParseTuple(args, format,
643 Py_FileSystemDefaultEncoding, &path1,
644 Py_FileSystemDefaultEncoding, &path2))
645 return NULL;
646 Py_BEGIN_ALLOW_THREADS
647 res = (*func)(path1, path2);
648 Py_END_ALLOW_THREADS
649 PyMem_Free(path1);
650 PyMem_Free(path2);
651 if (res != 0)
652 /* XXX how to report both path1 and path2??? */
653 return posix_error();
654 Py_INCREF(Py_None);
655 return Py_None;
658 #ifdef Py_WIN_WIDE_FILENAMES
659 static PyObject*
660 win32_1str(PyObject* args, char* func,
661 char* format, BOOL (__stdcall *funcA)(LPCSTR),
662 char* wformat, BOOL (__stdcall *funcW)(LPWSTR))
664 PyObject *uni;
665 char *ansi;
666 BOOL result;
667 if (unicode_file_names()) {
668 if (!PyArg_ParseTuple(args, wformat, &uni))
669 PyErr_Clear();
670 else {
671 Py_BEGIN_ALLOW_THREADS
672 result = funcW(PyUnicode_AsUnicode(uni));
673 Py_END_ALLOW_THREADS
674 if (!result)
675 return win32_error_unicode(func, PyUnicode_AsUnicode(uni));
676 Py_INCREF(Py_None);
677 return Py_None;
680 if (!PyArg_ParseTuple(args, format, &ansi))
681 return NULL;
682 Py_BEGIN_ALLOW_THREADS
683 result = funcA(ansi);
684 Py_END_ALLOW_THREADS
685 if (!result)
686 return win32_error(func, ansi);
687 Py_INCREF(Py_None);
688 return Py_None;
692 /* This is a reimplementation of the C library's chdir function,
693 but one that produces Win32 errors instead of DOS error codes.
694 chdir is essentially a wrapper around SetCurrentDirectory; however,
695 it also needs to set "magic" environment variables indicating
696 the per-drive current directory, which are of the form =<drive>: */
697 BOOL __stdcall
698 win32_chdir(LPCSTR path)
700 char new_path[MAX_PATH+1];
701 int result;
702 char env[4] = "=x:";
704 if(!SetCurrentDirectoryA(path))
705 return FALSE;
706 result = GetCurrentDirectoryA(MAX_PATH+1, new_path);
707 if (!result)
708 return FALSE;
709 /* In the ANSI API, there should not be any paths longer
710 than MAX_PATH. */
711 assert(result <= MAX_PATH+1);
712 if (strncmp(new_path, "\\\\", 2) == 0 ||
713 strncmp(new_path, "//", 2) == 0)
714 /* UNC path, nothing to do. */
715 return TRUE;
716 env[1] = new_path[0];
717 return SetEnvironmentVariableA(env, new_path);
720 /* The Unicode version differs from the ANSI version
721 since the current directory might exceed MAX_PATH characters */
722 BOOL __stdcall
723 win32_wchdir(LPCWSTR path)
725 wchar_t _new_path[MAX_PATH+1], *new_path = _new_path;
726 int result;
727 wchar_t env[4] = L"=x:";
729 if(!SetCurrentDirectoryW(path))
730 return FALSE;
731 result = GetCurrentDirectoryW(MAX_PATH+1, new_path);
732 if (!result)
733 return FALSE;
734 if (result > MAX_PATH+1) {
735 new_path = malloc(result);
736 if (!new_path) {
737 SetLastError(ERROR_OUTOFMEMORY);
738 return FALSE;
741 if (wcsncmp(new_path, L"\\\\", 2) == 0 ||
742 wcsncmp(new_path, L"//", 2) == 0)
743 /* UNC path, nothing to do. */
744 return TRUE;
745 env[1] = new_path[0];
746 result = SetEnvironmentVariableW(env, new_path);
747 if (new_path != _new_path)
748 free(new_path);
749 return result;
751 #endif
753 #ifdef MS_WINDOWS
754 /* The CRT of Windows has a number of flaws wrt. its stat() implementation:
755 - time stamps are restricted to second resolution
756 - file modification times suffer from forth-and-back conversions between
757 UTC and local time
758 Therefore, we implement our own stat, based on the Win32 API directly.
760 #define HAVE_STAT_NSEC 1
762 struct win32_stat{
763 int st_dev;
764 __int64 st_ino;
765 unsigned short st_mode;
766 int st_nlink;
767 int st_uid;
768 int st_gid;
769 int st_rdev;
770 __int64 st_size;
771 int st_atime;
772 int st_atime_nsec;
773 int st_mtime;
774 int st_mtime_nsec;
775 int st_ctime;
776 int st_ctime_nsec;
779 static __int64 secs_between_epochs = 11644473600; /* Seconds between 1.1.1601 and 1.1.1970 */
781 static void
782 FILE_TIME_to_time_t_nsec(FILETIME *in_ptr, int *time_out, int* nsec_out)
784 /* XXX endianness. Shouldn't matter, as all Windows implementations are little-endian */
785 /* Cannot simply cast and dereference in_ptr,
786 since it might not be aligned properly */
787 __int64 in;
788 memcpy(&in, in_ptr, sizeof(in));
789 *nsec_out = (int)(in % 10000000) * 100; /* FILETIME is in units of 100 nsec. */
790 /* XXX Win32 supports time stamps past 2038; we currently don't */
791 *time_out = Py_SAFE_DOWNCAST((in / 10000000) - secs_between_epochs, __int64, int);
794 static void
795 time_t_to_FILE_TIME(int time_in, int nsec_in, FILETIME *out_ptr)
797 /* XXX endianness */
798 __int64 out;
799 out = time_in + secs_between_epochs;
800 out = out * 10000000 + nsec_in / 100;
801 memcpy(out_ptr, &out, sizeof(out));
804 /* Below, we *know* that ugo+r is 0444 */
805 #if _S_IREAD != 0400
806 #error Unsupported C library
807 #endif
808 static int
809 attributes_to_mode(DWORD attr)
811 int m = 0;
812 if (attr & FILE_ATTRIBUTE_DIRECTORY)
813 m |= _S_IFDIR | 0111; /* IFEXEC for user,group,other */
814 else
815 m |= _S_IFREG;
816 if (attr & FILE_ATTRIBUTE_READONLY)
817 m |= 0444;
818 else
819 m |= 0666;
820 return m;
823 static int
824 attribute_data_to_stat(WIN32_FILE_ATTRIBUTE_DATA *info, struct win32_stat *result)
826 memset(result, 0, sizeof(*result));
827 result->st_mode = attributes_to_mode(info->dwFileAttributes);
828 result->st_size = (((__int64)info->nFileSizeHigh)<<32) + info->nFileSizeLow;
829 FILE_TIME_to_time_t_nsec(&info->ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
830 FILE_TIME_to_time_t_nsec(&info->ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
831 FILE_TIME_to_time_t_nsec(&info->ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
833 return 0;
836 /* Emulate GetFileAttributesEx[AW] on Windows 95 */
837 static int checked = 0;
838 static BOOL (CALLBACK *gfaxa)(LPCSTR, GET_FILEEX_INFO_LEVELS, LPVOID);
839 static BOOL (CALLBACK *gfaxw)(LPCWSTR, GET_FILEEX_INFO_LEVELS, LPVOID);
840 static void
841 check_gfax()
843 HINSTANCE hKernel32;
844 if (checked)
845 return;
846 checked = 1;
847 hKernel32 = GetModuleHandle("KERNEL32");
848 *(FARPROC*)&gfaxa = GetProcAddress(hKernel32, "GetFileAttributesExA");
849 *(FARPROC*)&gfaxw = GetProcAddress(hKernel32, "GetFileAttributesExW");
852 static BOOL
853 attributes_from_dir(LPCSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
855 HANDLE hFindFile;
856 WIN32_FIND_DATAA FileData;
857 hFindFile = FindFirstFileA(pszFile, &FileData);
858 if (hFindFile == INVALID_HANDLE_VALUE)
859 return FALSE;
860 FindClose(hFindFile);
861 pfad->dwFileAttributes = FileData.dwFileAttributes;
862 pfad->ftCreationTime = FileData.ftCreationTime;
863 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
864 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
865 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
866 pfad->nFileSizeLow = FileData.nFileSizeLow;
867 return TRUE;
870 static BOOL
871 attributes_from_dir_w(LPCWSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
873 HANDLE hFindFile;
874 WIN32_FIND_DATAW FileData;
875 hFindFile = FindFirstFileW(pszFile, &FileData);
876 if (hFindFile == INVALID_HANDLE_VALUE)
877 return FALSE;
878 FindClose(hFindFile);
879 pfad->dwFileAttributes = FileData.dwFileAttributes;
880 pfad->ftCreationTime = FileData.ftCreationTime;
881 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
882 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
883 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
884 pfad->nFileSizeLow = FileData.nFileSizeLow;
885 return TRUE;
888 static BOOL WINAPI
889 Py_GetFileAttributesExA(LPCSTR pszFile,
890 GET_FILEEX_INFO_LEVELS level,
891 LPVOID pv)
893 BOOL result;
894 LPWIN32_FILE_ATTRIBUTE_DATA pfad = pv;
895 /* First try to use the system's implementation, if that is
896 available and either succeeds to gives an error other than
897 that it isn't implemented. */
898 check_gfax();
899 if (gfaxa) {
900 result = gfaxa(pszFile, level, pv);
901 if (result || GetLastError() != ERROR_CALL_NOT_IMPLEMENTED)
902 return result;
904 /* It's either not present, or not implemented.
905 Emulate using FindFirstFile. */
906 if (level != GetFileExInfoStandard) {
907 SetLastError(ERROR_INVALID_PARAMETER);
908 return FALSE;
910 /* Use GetFileAttributes to validate that the file name
911 does not contain wildcards (which FindFirstFile would
912 accept). */
913 if (GetFileAttributesA(pszFile) == 0xFFFFFFFF)
914 return FALSE;
915 return attributes_from_dir(pszFile, pfad);
918 static BOOL WINAPI
919 Py_GetFileAttributesExW(LPCWSTR pszFile,
920 GET_FILEEX_INFO_LEVELS level,
921 LPVOID pv)
923 BOOL result;
924 LPWIN32_FILE_ATTRIBUTE_DATA pfad = pv;
925 /* First try to use the system's implementation, if that is
926 available and either succeeds to gives an error other than
927 that it isn't implemented. */
928 check_gfax();
929 if (gfaxa) {
930 result = gfaxw(pszFile, level, pv);
931 if (result || GetLastError() != ERROR_CALL_NOT_IMPLEMENTED)
932 return result;
934 /* It's either not present, or not implemented.
935 Emulate using FindFirstFile. */
936 if (level != GetFileExInfoStandard) {
937 SetLastError(ERROR_INVALID_PARAMETER);
938 return FALSE;
940 /* Use GetFileAttributes to validate that the file name
941 does not contain wildcards (which FindFirstFile would
942 accept). */
943 if (GetFileAttributesW(pszFile) == 0xFFFFFFFF)
944 return FALSE;
945 return attributes_from_dir_w(pszFile, pfad);
948 static int
949 win32_stat(const char* path, struct win32_stat *result)
951 WIN32_FILE_ATTRIBUTE_DATA info;
952 int code;
953 char *dot;
954 /* XXX not supported on Win95 and NT 3.x */
955 if (!Py_GetFileAttributesExA(path, GetFileExInfoStandard, &info)) {
956 if (GetLastError() != ERROR_SHARING_VIOLATION) {
957 /* Protocol violation: we explicitly clear errno, instead of
958 setting it to a POSIX error. Callers should use GetLastError. */
959 errno = 0;
960 return -1;
961 } else {
962 /* Could not get attributes on open file. Fall back to
963 reading the directory. */
964 if (!attributes_from_dir(path, &info)) {
965 /* Very strange. This should not fail now */
966 errno = 0;
967 return -1;
971 code = attribute_data_to_stat(&info, result);
972 if (code != 0)
973 return code;
974 /* Set S_IFEXEC if it is an .exe, .bat, ... */
975 dot = strrchr(path, '.');
976 if (dot) {
977 if (stricmp(dot, ".bat") == 0 ||
978 stricmp(dot, ".cmd") == 0 ||
979 stricmp(dot, ".exe") == 0 ||
980 stricmp(dot, ".com") == 0)
981 result->st_mode |= 0111;
983 return code;
986 static int
987 win32_wstat(const wchar_t* path, struct win32_stat *result)
989 int code;
990 const wchar_t *dot;
991 WIN32_FILE_ATTRIBUTE_DATA info;
992 /* XXX not supported on Win95 and NT 3.x */
993 if (!Py_GetFileAttributesExW(path, GetFileExInfoStandard, &info)) {
994 if (GetLastError() != ERROR_SHARING_VIOLATION) {
995 /* Protocol violation: we explicitly clear errno, instead of
996 setting it to a POSIX error. Callers should use GetLastError. */
997 errno = 0;
998 return -1;
999 } else {
1000 /* Could not get attributes on open file. Fall back to
1001 reading the directory. */
1002 if (!attributes_from_dir_w(path, &info)) {
1003 /* Very strange. This should not fail now */
1004 errno = 0;
1005 return -1;
1009 code = attribute_data_to_stat(&info, result);
1010 if (code < 0)
1011 return code;
1012 /* Set IFEXEC if it is an .exe, .bat, ... */
1013 dot = wcsrchr(path, '.');
1014 if (dot) {
1015 if (_wcsicmp(dot, L".bat") == 0 ||
1016 _wcsicmp(dot, L".cmd") == 0 ||
1017 _wcsicmp(dot, L".exe") == 0 ||
1018 _wcsicmp(dot, L".com") == 0)
1019 result->st_mode |= 0111;
1021 return code;
1024 static int
1025 win32_fstat(int file_number, struct win32_stat *result)
1027 BY_HANDLE_FILE_INFORMATION info;
1028 HANDLE h;
1029 int type;
1031 h = (HANDLE)_get_osfhandle(file_number);
1033 /* Protocol violation: we explicitly clear errno, instead of
1034 setting it to a POSIX error. Callers should use GetLastError. */
1035 errno = 0;
1037 if (h == INVALID_HANDLE_VALUE) {
1038 /* This is really a C library error (invalid file handle).
1039 We set the Win32 error to the closes one matching. */
1040 SetLastError(ERROR_INVALID_HANDLE);
1041 return -1;
1043 memset(result, 0, sizeof(*result));
1045 type = GetFileType(h);
1046 if (type == FILE_TYPE_UNKNOWN) {
1047 DWORD error = GetLastError();
1048 if (error != 0) {
1049 return -1;
1051 /* else: valid but unknown file */
1054 if (type != FILE_TYPE_DISK) {
1055 if (type == FILE_TYPE_CHAR)
1056 result->st_mode = _S_IFCHR;
1057 else if (type == FILE_TYPE_PIPE)
1058 result->st_mode = _S_IFIFO;
1059 return 0;
1062 if (!GetFileInformationByHandle(h, &info)) {
1063 return -1;
1066 /* similar to stat() */
1067 result->st_mode = attributes_to_mode(info.dwFileAttributes);
1068 result->st_size = (((__int64)info.nFileSizeHigh)<<32) + info.nFileSizeLow;
1069 FILE_TIME_to_time_t_nsec(&info.ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1070 FILE_TIME_to_time_t_nsec(&info.ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1071 FILE_TIME_to_time_t_nsec(&info.ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
1072 /* specific to fstat() */
1073 result->st_nlink = info.nNumberOfLinks;
1074 result->st_ino = (((__int64)info.nFileIndexHigh)<<32) + info.nFileIndexLow;
1075 return 0;
1078 #endif /* MS_WINDOWS */
1080 PyDoc_STRVAR(stat_result__doc__,
1081 "stat_result: Result from stat or lstat.\n\n\
1082 This object may be accessed either as a tuple of\n\
1083 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
1084 or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
1086 Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
1087 or st_flags, they are available as attributes only.\n\
1089 See os.stat for more information.");
1091 static PyStructSequence_Field stat_result_fields[] = {
1092 {"st_mode", "protection bits"},
1093 {"st_ino", "inode"},
1094 {"st_dev", "device"},
1095 {"st_nlink", "number of hard links"},
1096 {"st_uid", "user ID of owner"},
1097 {"st_gid", "group ID of owner"},
1098 {"st_size", "total size, in bytes"},
1099 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
1100 {NULL, "integer time of last access"},
1101 {NULL, "integer time of last modification"},
1102 {NULL, "integer time of last change"},
1103 {"st_atime", "time of last access"},
1104 {"st_mtime", "time of last modification"},
1105 {"st_ctime", "time of last change"},
1106 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
1107 {"st_blksize", "blocksize for filesystem I/O"},
1108 #endif
1109 #ifdef HAVE_STRUCT_STAT_ST_BLOCKS
1110 {"st_blocks", "number of blocks allocated"},
1111 #endif
1112 #ifdef HAVE_STRUCT_STAT_ST_RDEV
1113 {"st_rdev", "device type (if inode device)"},
1114 #endif
1115 #ifdef HAVE_STRUCT_STAT_ST_FLAGS
1116 {"st_flags", "user defined flags for file"},
1117 #endif
1118 #ifdef HAVE_STRUCT_STAT_ST_GEN
1119 {"st_gen", "generation number"},
1120 #endif
1121 #ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1122 {"st_birthtime", "time of creation"},
1123 #endif
1127 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
1128 #define ST_BLKSIZE_IDX 13
1129 #else
1130 #define ST_BLKSIZE_IDX 12
1131 #endif
1133 #ifdef HAVE_STRUCT_STAT_ST_BLOCKS
1134 #define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
1135 #else
1136 #define ST_BLOCKS_IDX ST_BLKSIZE_IDX
1137 #endif
1139 #ifdef HAVE_STRUCT_STAT_ST_RDEV
1140 #define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
1141 #else
1142 #define ST_RDEV_IDX ST_BLOCKS_IDX
1143 #endif
1145 #ifdef HAVE_STRUCT_STAT_ST_FLAGS
1146 #define ST_FLAGS_IDX (ST_RDEV_IDX+1)
1147 #else
1148 #define ST_FLAGS_IDX ST_RDEV_IDX
1149 #endif
1151 #ifdef HAVE_STRUCT_STAT_ST_GEN
1152 #define ST_GEN_IDX (ST_FLAGS_IDX+1)
1153 #else
1154 #define ST_GEN_IDX ST_FLAGS_IDX
1155 #endif
1157 #ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1158 #define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
1159 #else
1160 #define ST_BIRTHTIME_IDX ST_GEN_IDX
1161 #endif
1163 static PyStructSequence_Desc stat_result_desc = {
1164 "stat_result", /* name */
1165 stat_result__doc__, /* doc */
1166 stat_result_fields,
1170 PyDoc_STRVAR(statvfs_result__doc__,
1171 "statvfs_result: Result from statvfs or fstatvfs.\n\n\
1172 This object may be accessed either as a tuple of\n\
1173 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
1174 or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
1176 See os.statvfs for more information.");
1178 static PyStructSequence_Field statvfs_result_fields[] = {
1179 {"f_bsize", },
1180 {"f_frsize", },
1181 {"f_blocks", },
1182 {"f_bfree", },
1183 {"f_bavail", },
1184 {"f_files", },
1185 {"f_ffree", },
1186 {"f_favail", },
1187 {"f_flag", },
1188 {"f_namemax",},
1192 static PyStructSequence_Desc statvfs_result_desc = {
1193 "statvfs_result", /* name */
1194 statvfs_result__doc__, /* doc */
1195 statvfs_result_fields,
1199 static int initialized;
1200 static PyTypeObject StatResultType;
1201 static PyTypeObject StatVFSResultType;
1202 static newfunc structseq_new;
1204 static PyObject *
1205 statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1207 PyStructSequence *result;
1208 int i;
1210 result = (PyStructSequence*)structseq_new(type, args, kwds);
1211 if (!result)
1212 return NULL;
1213 /* If we have been initialized from a tuple,
1214 st_?time might be set to None. Initialize it
1215 from the int slots. */
1216 for (i = 7; i <= 9; i++) {
1217 if (result->ob_item[i+3] == Py_None) {
1218 Py_DECREF(Py_None);
1219 Py_INCREF(result->ob_item[i]);
1220 result->ob_item[i+3] = result->ob_item[i];
1223 return (PyObject*)result;
1228 /* If true, st_?time is float. */
1229 static int _stat_float_times = 1;
1231 PyDoc_STRVAR(stat_float_times__doc__,
1232 "stat_float_times([newval]) -> oldval\n\n\
1233 Determine whether os.[lf]stat represents time stamps as float objects.\n\
1234 If newval is True, future calls to stat() return floats, if it is False,\n\
1235 future calls return ints. \n\
1236 If newval is omitted, return the current setting.\n");
1238 static PyObject*
1239 stat_float_times(PyObject* self, PyObject *args)
1241 int newval = -1;
1242 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
1243 return NULL;
1244 if (newval == -1)
1245 /* Return old value */
1246 return PyBool_FromLong(_stat_float_times);
1247 _stat_float_times = newval;
1248 Py_INCREF(Py_None);
1249 return Py_None;
1252 static void
1253 fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
1255 PyObject *fval,*ival;
1256 #if SIZEOF_TIME_T > SIZEOF_LONG
1257 ival = PyLong_FromLongLong((PY_LONG_LONG)sec);
1258 #else
1259 ival = PyInt_FromLong((long)sec);
1260 #endif
1261 if (!ival)
1262 return;
1263 if (_stat_float_times) {
1264 fval = PyFloat_FromDouble(sec + 1e-9*nsec);
1265 } else {
1266 fval = ival;
1267 Py_INCREF(fval);
1269 PyStructSequence_SET_ITEM(v, index, ival);
1270 PyStructSequence_SET_ITEM(v, index+3, fval);
1273 /* pack a system stat C structure into the Python stat tuple
1274 (used by posix_stat() and posix_fstat()) */
1275 static PyObject*
1276 _pystat_fromstructstat(STRUCT_STAT *st)
1278 unsigned long ansec, mnsec, cnsec;
1279 PyObject *v = PyStructSequence_New(&StatResultType);
1280 if (v == NULL)
1281 return NULL;
1283 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long)st->st_mode));
1284 #ifdef HAVE_LARGEFILE_SUPPORT
1285 PyStructSequence_SET_ITEM(v, 1,
1286 PyLong_FromLongLong((PY_LONG_LONG)st->st_ino));
1287 #else
1288 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long)st->st_ino));
1289 #endif
1290 #if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
1291 PyStructSequence_SET_ITEM(v, 2,
1292 PyLong_FromLongLong((PY_LONG_LONG)st->st_dev));
1293 #else
1294 PyStructSequence_SET_ITEM(v, 2, PyInt_FromLong((long)st->st_dev));
1295 #endif
1296 PyStructSequence_SET_ITEM(v, 3, PyInt_FromLong((long)st->st_nlink));
1297 PyStructSequence_SET_ITEM(v, 4, PyInt_FromLong((long)st->st_uid));
1298 PyStructSequence_SET_ITEM(v, 5, PyInt_FromLong((long)st->st_gid));
1299 #ifdef HAVE_LARGEFILE_SUPPORT
1300 PyStructSequence_SET_ITEM(v, 6,
1301 PyLong_FromLongLong((PY_LONG_LONG)st->st_size));
1302 #else
1303 PyStructSequence_SET_ITEM(v, 6, PyInt_FromLong(st->st_size));
1304 #endif
1306 #if defined(HAVE_STAT_TV_NSEC)
1307 ansec = st->st_atim.tv_nsec;
1308 mnsec = st->st_mtim.tv_nsec;
1309 cnsec = st->st_ctim.tv_nsec;
1310 #elif defined(HAVE_STAT_TV_NSEC2)
1311 ansec = st->st_atimespec.tv_nsec;
1312 mnsec = st->st_mtimespec.tv_nsec;
1313 cnsec = st->st_ctimespec.tv_nsec;
1314 #elif defined(HAVE_STAT_NSEC)
1315 ansec = st->st_atime_nsec;
1316 mnsec = st->st_mtime_nsec;
1317 cnsec = st->st_ctime_nsec;
1318 #else
1319 ansec = mnsec = cnsec = 0;
1320 #endif
1321 fill_time(v, 7, st->st_atime, ansec);
1322 fill_time(v, 8, st->st_mtime, mnsec);
1323 fill_time(v, 9, st->st_ctime, cnsec);
1325 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
1326 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
1327 PyInt_FromLong((long)st->st_blksize));
1328 #endif
1329 #ifdef HAVE_STRUCT_STAT_ST_BLOCKS
1330 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
1331 PyInt_FromLong((long)st->st_blocks));
1332 #endif
1333 #ifdef HAVE_STRUCT_STAT_ST_RDEV
1334 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
1335 PyInt_FromLong((long)st->st_rdev));
1336 #endif
1337 #ifdef HAVE_STRUCT_STAT_ST_GEN
1338 PyStructSequence_SET_ITEM(v, ST_GEN_IDX,
1339 PyInt_FromLong((long)st->st_gen));
1340 #endif
1341 #ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1343 PyObject *val;
1344 unsigned long bsec,bnsec;
1345 bsec = (long)st->st_birthtime;
1346 #ifdef HAVE_STAT_TV_NSEC2
1347 bnsec = st->st_birthtimespec.tv_nsec;
1348 #else
1349 bnsec = 0;
1350 #endif
1351 if (_stat_float_times) {
1352 val = PyFloat_FromDouble(bsec + 1e-9*bnsec);
1353 } else {
1354 val = PyInt_FromLong((long)bsec);
1356 PyStructSequence_SET_ITEM(v, ST_BIRTHTIME_IDX,
1357 val);
1359 #endif
1360 #ifdef HAVE_STRUCT_STAT_ST_FLAGS
1361 PyStructSequence_SET_ITEM(v, ST_FLAGS_IDX,
1362 PyInt_FromLong((long)st->st_flags));
1363 #endif
1365 if (PyErr_Occurred()) {
1366 Py_DECREF(v);
1367 return NULL;
1370 return v;
1373 #ifdef MS_WINDOWS
1375 /* IsUNCRoot -- test whether the supplied path is of the form \\SERVER\SHARE\,
1376 where / can be used in place of \ and the trailing slash is optional.
1377 Both SERVER and SHARE must have at least one character.
1380 #define ISSLASHA(c) ((c) == '\\' || (c) == '/')
1381 #define ISSLASHW(c) ((c) == L'\\' || (c) == L'/')
1382 #ifndef ARRAYSIZE
1383 #define ARRAYSIZE(a) (sizeof(a) / sizeof(a[0]))
1384 #endif
1386 static BOOL
1387 IsUNCRootA(char *path, int pathlen)
1389 #define ISSLASH ISSLASHA
1391 int i, share;
1393 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1394 /* minimum UNCRoot is \\x\y */
1395 return FALSE;
1396 for (i = 2; i < pathlen ; i++)
1397 if (ISSLASH(path[i])) break;
1398 if (i == 2 || i == pathlen)
1399 /* do not allow \\\SHARE or \\SERVER */
1400 return FALSE;
1401 share = i+1;
1402 for (i = share; i < pathlen; i++)
1403 if (ISSLASH(path[i])) break;
1404 return (i != share && (i == pathlen || i == pathlen-1));
1406 #undef ISSLASH
1409 #ifdef Py_WIN_WIDE_FILENAMES
1410 static BOOL
1411 IsUNCRootW(Py_UNICODE *path, int pathlen)
1413 #define ISSLASH ISSLASHW
1415 int i, share;
1417 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1418 /* minimum UNCRoot is \\x\y */
1419 return FALSE;
1420 for (i = 2; i < pathlen ; i++)
1421 if (ISSLASH(path[i])) break;
1422 if (i == 2 || i == pathlen)
1423 /* do not allow \\\SHARE or \\SERVER */
1424 return FALSE;
1425 share = i+1;
1426 for (i = share; i < pathlen; i++)
1427 if (ISSLASH(path[i])) break;
1428 return (i != share && (i == pathlen || i == pathlen-1));
1430 #undef ISSLASH
1432 #endif /* Py_WIN_WIDE_FILENAMES */
1433 #endif /* MS_WINDOWS */
1435 static PyObject *
1436 posix_do_stat(PyObject *self, PyObject *args,
1437 char *format,
1438 #ifdef __VMS
1439 int (*statfunc)(const char *, STRUCT_STAT *, ...),
1440 #else
1441 int (*statfunc)(const char *, STRUCT_STAT *),
1442 #endif
1443 char *wformat,
1444 int (*wstatfunc)(const Py_UNICODE *, STRUCT_STAT *))
1446 STRUCT_STAT st;
1447 char *path = NULL; /* pass this to stat; do not free() it */
1448 char *pathfree = NULL; /* this memory must be free'd */
1449 int res;
1450 PyObject *result;
1452 #ifdef Py_WIN_WIDE_FILENAMES
1453 /* If on wide-character-capable OS see if argument
1454 is Unicode and if so use wide API. */
1455 if (unicode_file_names()) {
1456 PyUnicodeObject *po;
1457 if (PyArg_ParseTuple(args, wformat, &po)) {
1458 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
1460 Py_BEGIN_ALLOW_THREADS
1461 /* PyUnicode_AS_UNICODE result OK without
1462 thread lock as it is a simple dereference. */
1463 res = wstatfunc(wpath, &st);
1464 Py_END_ALLOW_THREADS
1466 if (res != 0)
1467 return win32_error_unicode("stat", wpath);
1468 return _pystat_fromstructstat(&st);
1470 /* Drop the argument parsing error as narrow strings
1471 are also valid. */
1472 PyErr_Clear();
1474 #endif
1476 if (!PyArg_ParseTuple(args, format,
1477 Py_FileSystemDefaultEncoding, &path))
1478 return NULL;
1479 pathfree = path;
1481 Py_BEGIN_ALLOW_THREADS
1482 res = (*statfunc)(path, &st);
1483 Py_END_ALLOW_THREADS
1485 if (res != 0) {
1486 #ifdef MS_WINDOWS
1487 result = win32_error("stat", pathfree);
1488 #else
1489 result = posix_error_with_filename(pathfree);
1490 #endif
1492 else
1493 result = _pystat_fromstructstat(&st);
1495 PyMem_Free(pathfree);
1496 return result;
1499 /* POSIX methods */
1501 PyDoc_STRVAR(posix_access__doc__,
1502 "access(path, mode) -> True if granted, False otherwise\n\n\
1503 Use the real uid/gid to test for access to a path. Note that most\n\
1504 operations will use the effective uid/gid, therefore this routine can\n\
1505 be used in a suid/sgid environment to test if the invoking user has the\n\
1506 specified access to the path. The mode argument can be F_OK to test\n\
1507 existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
1509 static PyObject *
1510 posix_access(PyObject *self, PyObject *args)
1512 char *path;
1513 int mode;
1515 #ifdef Py_WIN_WIDE_FILENAMES
1516 DWORD attr;
1517 if (unicode_file_names()) {
1518 PyUnicodeObject *po;
1519 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
1520 Py_BEGIN_ALLOW_THREADS
1521 /* PyUnicode_AS_UNICODE OK without thread lock as
1522 it is a simple dereference. */
1523 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1524 Py_END_ALLOW_THREADS
1525 goto finish;
1527 /* Drop the argument parsing error as narrow strings
1528 are also valid. */
1529 PyErr_Clear();
1531 if (!PyArg_ParseTuple(args, "eti:access",
1532 Py_FileSystemDefaultEncoding, &path, &mode))
1533 return 0;
1534 Py_BEGIN_ALLOW_THREADS
1535 attr = GetFileAttributesA(path);
1536 Py_END_ALLOW_THREADS
1537 PyMem_Free(path);
1538 finish:
1539 if (attr == 0xFFFFFFFF)
1540 /* File does not exist, or cannot read attributes */
1541 return PyBool_FromLong(0);
1542 /* Access is possible if either write access wasn't requested, or
1543 the file isn't read-only, or if it's a directory, as there are
1544 no read-only directories on Windows. */
1545 return PyBool_FromLong(!(mode & 2)
1546 || !(attr & FILE_ATTRIBUTE_READONLY)
1547 || (attr & FILE_ATTRIBUTE_DIRECTORY));
1548 #else
1549 int res;
1550 if (!PyArg_ParseTuple(args, "eti:access",
1551 Py_FileSystemDefaultEncoding, &path, &mode))
1552 return NULL;
1553 Py_BEGIN_ALLOW_THREADS
1554 res = access(path, mode);
1555 Py_END_ALLOW_THREADS
1556 PyMem_Free(path);
1557 return PyBool_FromLong(res == 0);
1558 #endif
1561 #ifndef F_OK
1562 #define F_OK 0
1563 #endif
1564 #ifndef R_OK
1565 #define R_OK 4
1566 #endif
1567 #ifndef W_OK
1568 #define W_OK 2
1569 #endif
1570 #ifndef X_OK
1571 #define X_OK 1
1572 #endif
1574 #ifdef HAVE_TTYNAME
1575 PyDoc_STRVAR(posix_ttyname__doc__,
1576 "ttyname(fd) -> string\n\n\
1577 Return the name of the terminal device connected to 'fd'.");
1579 static PyObject *
1580 posix_ttyname(PyObject *self, PyObject *args)
1582 int id;
1583 char *ret;
1585 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
1586 return NULL;
1588 #if defined(__VMS)
1589 /* file descriptor 0 only, the default input device (stdin) */
1590 if (id == 0) {
1591 ret = ttyname();
1593 else {
1594 ret = NULL;
1596 #else
1597 ret = ttyname(id);
1598 #endif
1599 if (ret == NULL)
1600 return posix_error();
1601 return PyString_FromString(ret);
1603 #endif
1605 #ifdef HAVE_CTERMID
1606 PyDoc_STRVAR(posix_ctermid__doc__,
1607 "ctermid() -> string\n\n\
1608 Return the name of the controlling terminal for this process.");
1610 static PyObject *
1611 posix_ctermid(PyObject *self, PyObject *noargs)
1613 char *ret;
1614 char buffer[L_ctermid];
1616 #ifdef USE_CTERMID_R
1617 ret = ctermid_r(buffer);
1618 #else
1619 ret = ctermid(buffer);
1620 #endif
1621 if (ret == NULL)
1622 return posix_error();
1623 return PyString_FromString(buffer);
1625 #endif
1627 PyDoc_STRVAR(posix_chdir__doc__,
1628 "chdir(path)\n\n\
1629 Change the current working directory to the specified path.");
1631 static PyObject *
1632 posix_chdir(PyObject *self, PyObject *args)
1634 #ifdef MS_WINDOWS
1635 return win32_1str(args, "chdir", "s:chdir", win32_chdir, "U:chdir", win32_wchdir);
1636 #elif defined(PYOS_OS2) && defined(PYCC_GCC)
1637 return posix_1str(args, "et:chdir", _chdir2);
1638 #elif defined(__VMS)
1639 return posix_1str(args, "et:chdir", (int (*)(const char *))chdir);
1640 #else
1641 return posix_1str(args, "et:chdir", chdir);
1642 #endif
1645 #ifdef HAVE_FCHDIR
1646 PyDoc_STRVAR(posix_fchdir__doc__,
1647 "fchdir(fildes)\n\n\
1648 Change to the directory of the given file descriptor. fildes must be\n\
1649 opened on a directory, not a file.");
1651 static PyObject *
1652 posix_fchdir(PyObject *self, PyObject *fdobj)
1654 return posix_fildes(fdobj, fchdir);
1656 #endif /* HAVE_FCHDIR */
1659 PyDoc_STRVAR(posix_chmod__doc__,
1660 "chmod(path, mode)\n\n\
1661 Change the access permissions of a file.");
1663 static PyObject *
1664 posix_chmod(PyObject *self, PyObject *args)
1666 char *path = NULL;
1667 int i;
1668 int res;
1669 #ifdef Py_WIN_WIDE_FILENAMES
1670 DWORD attr;
1671 if (unicode_file_names()) {
1672 PyUnicodeObject *po;
1673 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
1674 Py_BEGIN_ALLOW_THREADS
1675 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1676 if (attr != 0xFFFFFFFF) {
1677 if (i & _S_IWRITE)
1678 attr &= ~FILE_ATTRIBUTE_READONLY;
1679 else
1680 attr |= FILE_ATTRIBUTE_READONLY;
1681 res = SetFileAttributesW(PyUnicode_AS_UNICODE(po), attr);
1683 else
1684 res = 0;
1685 Py_END_ALLOW_THREADS
1686 if (!res)
1687 return win32_error_unicode("chmod",
1688 PyUnicode_AS_UNICODE(po));
1689 Py_INCREF(Py_None);
1690 return Py_None;
1692 /* Drop the argument parsing error as narrow strings
1693 are also valid. */
1694 PyErr_Clear();
1696 if (!PyArg_ParseTuple(args, "eti:chmod", Py_FileSystemDefaultEncoding,
1697 &path, &i))
1698 return NULL;
1699 Py_BEGIN_ALLOW_THREADS
1700 attr = GetFileAttributesA(path);
1701 if (attr != 0xFFFFFFFF) {
1702 if (i & _S_IWRITE)
1703 attr &= ~FILE_ATTRIBUTE_READONLY;
1704 else
1705 attr |= FILE_ATTRIBUTE_READONLY;
1706 res = SetFileAttributesA(path, attr);
1708 else
1709 res = 0;
1710 Py_END_ALLOW_THREADS
1711 if (!res) {
1712 win32_error("chmod", path);
1713 PyMem_Free(path);
1714 return NULL;
1716 PyMem_Free(path);
1717 Py_INCREF(Py_None);
1718 return Py_None;
1719 #else /* Py_WIN_WIDE_FILENAMES */
1720 if (!PyArg_ParseTuple(args, "eti:chmod", Py_FileSystemDefaultEncoding,
1721 &path, &i))
1722 return NULL;
1723 Py_BEGIN_ALLOW_THREADS
1724 res = chmod(path, i);
1725 Py_END_ALLOW_THREADS
1726 if (res < 0)
1727 return posix_error_with_allocated_filename(path);
1728 PyMem_Free(path);
1729 Py_INCREF(Py_None);
1730 return Py_None;
1731 #endif
1734 #ifdef HAVE_FCHMOD
1735 PyDoc_STRVAR(posix_fchmod__doc__,
1736 "fchmod(fd, mode)\n\n\
1737 Change the access permissions of the file given by file\n\
1738 descriptor fd.");
1740 static PyObject *
1741 posix_fchmod(PyObject *self, PyObject *args)
1743 int fd, mode, res;
1744 if (!PyArg_ParseTuple(args, "ii:fchmod", &fd, &mode))
1745 return NULL;
1746 Py_BEGIN_ALLOW_THREADS
1747 res = fchmod(fd, mode);
1748 Py_END_ALLOW_THREADS
1749 if (res < 0)
1750 return posix_error();
1751 Py_RETURN_NONE;
1753 #endif /* HAVE_FCHMOD */
1755 #ifdef HAVE_LCHMOD
1756 PyDoc_STRVAR(posix_lchmod__doc__,
1757 "lchmod(path, mode)\n\n\
1758 Change the access permissions of a file. If path is a symlink, this\n\
1759 affects the link itself rather than the target.");
1761 static PyObject *
1762 posix_lchmod(PyObject *self, PyObject *args)
1764 char *path = NULL;
1765 int i;
1766 int res;
1767 if (!PyArg_ParseTuple(args, "eti:lchmod", Py_FileSystemDefaultEncoding,
1768 &path, &i))
1769 return NULL;
1770 Py_BEGIN_ALLOW_THREADS
1771 res = lchmod(path, i);
1772 Py_END_ALLOW_THREADS
1773 if (res < 0)
1774 return posix_error_with_allocated_filename(path);
1775 PyMem_Free(path);
1776 Py_RETURN_NONE;
1778 #endif /* HAVE_LCHMOD */
1781 #ifdef HAVE_CHFLAGS
1782 PyDoc_STRVAR(posix_chflags__doc__,
1783 "chflags(path, flags)\n\n\
1784 Set file flags.");
1786 static PyObject *
1787 posix_chflags(PyObject *self, PyObject *args)
1789 char *path;
1790 unsigned long flags;
1791 int res;
1792 if (!PyArg_ParseTuple(args, "etk:chflags",
1793 Py_FileSystemDefaultEncoding, &path, &flags))
1794 return NULL;
1795 Py_BEGIN_ALLOW_THREADS
1796 res = chflags(path, flags);
1797 Py_END_ALLOW_THREADS
1798 if (res < 0)
1799 return posix_error_with_allocated_filename(path);
1800 PyMem_Free(path);
1801 Py_INCREF(Py_None);
1802 return Py_None;
1804 #endif /* HAVE_CHFLAGS */
1806 #ifdef HAVE_LCHFLAGS
1807 PyDoc_STRVAR(posix_lchflags__doc__,
1808 "lchflags(path, flags)\n\n\
1809 Set file flags.\n\
1810 This function will not follow symbolic links.");
1812 static PyObject *
1813 posix_lchflags(PyObject *self, PyObject *args)
1815 char *path;
1816 unsigned long flags;
1817 int res;
1818 if (!PyArg_ParseTuple(args, "etk:lchflags",
1819 Py_FileSystemDefaultEncoding, &path, &flags))
1820 return NULL;
1821 Py_BEGIN_ALLOW_THREADS
1822 res = lchflags(path, flags);
1823 Py_END_ALLOW_THREADS
1824 if (res < 0)
1825 return posix_error_with_allocated_filename(path);
1826 PyMem_Free(path);
1827 Py_INCREF(Py_None);
1828 return Py_None;
1830 #endif /* HAVE_LCHFLAGS */
1832 #ifdef HAVE_CHROOT
1833 PyDoc_STRVAR(posix_chroot__doc__,
1834 "chroot(path)\n\n\
1835 Change root directory to path.");
1837 static PyObject *
1838 posix_chroot(PyObject *self, PyObject *args)
1840 return posix_1str(args, "et:chroot", chroot);
1842 #endif
1844 #ifdef HAVE_FSYNC
1845 PyDoc_STRVAR(posix_fsync__doc__,
1846 "fsync(fildes)\n\n\
1847 force write of file with filedescriptor to disk.");
1849 static PyObject *
1850 posix_fsync(PyObject *self, PyObject *fdobj)
1852 return posix_fildes(fdobj, fsync);
1854 #endif /* HAVE_FSYNC */
1856 #ifdef HAVE_FDATASYNC
1858 #ifdef __hpux
1859 extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
1860 #endif
1862 PyDoc_STRVAR(posix_fdatasync__doc__,
1863 "fdatasync(fildes)\n\n\
1864 force write of file with filedescriptor to disk.\n\
1865 does not force update of metadata.");
1867 static PyObject *
1868 posix_fdatasync(PyObject *self, PyObject *fdobj)
1870 return posix_fildes(fdobj, fdatasync);
1872 #endif /* HAVE_FDATASYNC */
1875 #ifdef HAVE_CHOWN
1876 PyDoc_STRVAR(posix_chown__doc__,
1877 "chown(path, uid, gid)\n\n\
1878 Change the owner and group id of path to the numeric uid and gid.");
1880 static PyObject *
1881 posix_chown(PyObject *self, PyObject *args)
1883 char *path = NULL;
1884 long uid, gid;
1885 int res;
1886 if (!PyArg_ParseTuple(args, "etll:chown",
1887 Py_FileSystemDefaultEncoding, &path,
1888 &uid, &gid))
1889 return NULL;
1890 Py_BEGIN_ALLOW_THREADS
1891 res = chown(path, (uid_t) uid, (gid_t) gid);
1892 Py_END_ALLOW_THREADS
1893 if (res < 0)
1894 return posix_error_with_allocated_filename(path);
1895 PyMem_Free(path);
1896 Py_INCREF(Py_None);
1897 return Py_None;
1899 #endif /* HAVE_CHOWN */
1901 #ifdef HAVE_FCHOWN
1902 PyDoc_STRVAR(posix_fchown__doc__,
1903 "fchown(fd, uid, gid)\n\n\
1904 Change the owner and group id of the file given by file descriptor\n\
1905 fd to the numeric uid and gid.");
1907 static PyObject *
1908 posix_fchown(PyObject *self, PyObject *args)
1910 int fd, uid, gid;
1911 int res;
1912 if (!PyArg_ParseTuple(args, "iii:chown", &fd, &uid, &gid))
1913 return NULL;
1914 Py_BEGIN_ALLOW_THREADS
1915 res = fchown(fd, (uid_t) uid, (gid_t) gid);
1916 Py_END_ALLOW_THREADS
1917 if (res < 0)
1918 return posix_error();
1919 Py_RETURN_NONE;
1921 #endif /* HAVE_FCHOWN */
1923 #ifdef HAVE_LCHOWN
1924 PyDoc_STRVAR(posix_lchown__doc__,
1925 "lchown(path, uid, gid)\n\n\
1926 Change the owner and group id of path to the numeric uid and gid.\n\
1927 This function will not follow symbolic links.");
1929 static PyObject *
1930 posix_lchown(PyObject *self, PyObject *args)
1932 char *path = NULL;
1933 int uid, gid;
1934 int res;
1935 if (!PyArg_ParseTuple(args, "etii:lchown",
1936 Py_FileSystemDefaultEncoding, &path,
1937 &uid, &gid))
1938 return NULL;
1939 Py_BEGIN_ALLOW_THREADS
1940 res = lchown(path, (uid_t) uid, (gid_t) gid);
1941 Py_END_ALLOW_THREADS
1942 if (res < 0)
1943 return posix_error_with_allocated_filename(path);
1944 PyMem_Free(path);
1945 Py_INCREF(Py_None);
1946 return Py_None;
1948 #endif /* HAVE_LCHOWN */
1951 #ifdef HAVE_GETCWD
1952 PyDoc_STRVAR(posix_getcwd__doc__,
1953 "getcwd() -> path\n\n\
1954 Return a string representing the current working directory.");
1956 static PyObject *
1957 posix_getcwd(PyObject *self, PyObject *noargs)
1959 int bufsize_incr = 1024;
1960 int bufsize = 0;
1961 char *tmpbuf = NULL;
1962 char *res = NULL;
1963 PyObject *dynamic_return;
1965 Py_BEGIN_ALLOW_THREADS
1966 do {
1967 bufsize = bufsize + bufsize_incr;
1968 tmpbuf = malloc(bufsize);
1969 if (tmpbuf == NULL) {
1970 break;
1972 #if defined(PYOS_OS2) && defined(PYCC_GCC)
1973 res = _getcwd2(tmpbuf, bufsize);
1974 #else
1975 res = getcwd(tmpbuf, bufsize);
1976 #endif
1978 if (res == NULL) {
1979 free(tmpbuf);
1981 } while ((res == NULL) && (errno == ERANGE));
1982 Py_END_ALLOW_THREADS
1984 if (res == NULL)
1985 return posix_error();
1987 dynamic_return = PyString_FromString(tmpbuf);
1988 free(tmpbuf);
1990 return dynamic_return;
1993 #ifdef Py_USING_UNICODE
1994 PyDoc_STRVAR(posix_getcwdu__doc__,
1995 "getcwdu() -> path\n\n\
1996 Return a unicode string representing the current working directory.");
1998 static PyObject *
1999 posix_getcwdu(PyObject *self, PyObject *noargs)
2001 char buf[1026];
2002 char *res;
2004 #ifdef Py_WIN_WIDE_FILENAMES
2005 DWORD len;
2006 if (unicode_file_names()) {
2007 wchar_t wbuf[1026];
2008 wchar_t *wbuf2 = wbuf;
2009 PyObject *resobj;
2010 Py_BEGIN_ALLOW_THREADS
2011 len = GetCurrentDirectoryW(sizeof wbuf/ sizeof wbuf[0], wbuf);
2012 /* If the buffer is large enough, len does not include the
2013 terminating \0. If the buffer is too small, len includes
2014 the space needed for the terminator. */
2015 if (len >= sizeof wbuf/ sizeof wbuf[0]) {
2016 wbuf2 = malloc(len * sizeof(wchar_t));
2017 if (wbuf2)
2018 len = GetCurrentDirectoryW(len, wbuf2);
2020 Py_END_ALLOW_THREADS
2021 if (!wbuf2) {
2022 PyErr_NoMemory();
2023 return NULL;
2025 if (!len) {
2026 if (wbuf2 != wbuf) free(wbuf2);
2027 return win32_error("getcwdu", NULL);
2029 resobj = PyUnicode_FromWideChar(wbuf2, len);
2030 if (wbuf2 != wbuf) free(wbuf2);
2031 return resobj;
2033 #endif
2035 Py_BEGIN_ALLOW_THREADS
2036 #if defined(PYOS_OS2) && defined(PYCC_GCC)
2037 res = _getcwd2(buf, sizeof buf);
2038 #else
2039 res = getcwd(buf, sizeof buf);
2040 #endif
2041 Py_END_ALLOW_THREADS
2042 if (res == NULL)
2043 return posix_error();
2044 return PyUnicode_Decode(buf, strlen(buf), Py_FileSystemDefaultEncoding,"strict");
2046 #endif
2047 #endif
2050 #ifdef HAVE_LINK
2051 PyDoc_STRVAR(posix_link__doc__,
2052 "link(src, dst)\n\n\
2053 Create a hard link to a file.");
2055 static PyObject *
2056 posix_link(PyObject *self, PyObject *args)
2058 return posix_2str(args, "etet:link", link);
2060 #endif /* HAVE_LINK */
2063 PyDoc_STRVAR(posix_listdir__doc__,
2064 "listdir(path) -> list_of_strings\n\n\
2065 Return a list containing the names of the entries in the directory.\n\
2067 path: path of directory to list\n\
2069 The list is in arbitrary order. It does not include the special\n\
2070 entries '.' and '..' even if they are present in the directory.");
2072 static PyObject *
2073 posix_listdir(PyObject *self, PyObject *args)
2075 /* XXX Should redo this putting the (now four) versions of opendir
2076 in separate files instead of having them all here... */
2077 #if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
2079 PyObject *d, *v;
2080 HANDLE hFindFile;
2081 BOOL result;
2082 WIN32_FIND_DATA FileData;
2083 char namebuf[MAX_PATH+5]; /* Overallocate for \\*.*\0 */
2084 char *bufptr = namebuf;
2085 Py_ssize_t len = sizeof(namebuf)-5; /* only claim to have space for MAX_PATH */
2087 #ifdef Py_WIN_WIDE_FILENAMES
2088 /* If on wide-character-capable OS see if argument
2089 is Unicode and if so use wide API. */
2090 if (unicode_file_names()) {
2091 PyObject *po;
2092 if (PyArg_ParseTuple(args, "U:listdir", &po)) {
2093 WIN32_FIND_DATAW wFileData;
2094 Py_UNICODE *wnamebuf;
2095 Py_UNICODE wch;
2096 /* Overallocate for \\*.*\0 */
2097 len = PyUnicode_GET_SIZE(po);
2098 wnamebuf = malloc((len + 5) * sizeof(wchar_t));
2099 if (!wnamebuf) {
2100 PyErr_NoMemory();
2101 return NULL;
2103 wcscpy(wnamebuf, PyUnicode_AS_UNICODE(po));
2104 wch = len > 0 ? wnamebuf[len-1] : '\0';
2105 if (wch != L'/' && wch != L'\\' && wch != L':')
2106 wnamebuf[len++] = L'\\';
2107 wcscpy(wnamebuf + len, L"*.*");
2108 if ((d = PyList_New(0)) == NULL) {
2109 free(wnamebuf);
2110 return NULL;
2112 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
2113 if (hFindFile == INVALID_HANDLE_VALUE) {
2114 int error = GetLastError();
2115 if (error == ERROR_FILE_NOT_FOUND) {
2116 free(wnamebuf);
2117 return d;
2119 Py_DECREF(d);
2120 win32_error_unicode("FindFirstFileW", wnamebuf);
2121 free(wnamebuf);
2122 return NULL;
2124 do {
2125 /* Skip over . and .. */
2126 if (wcscmp(wFileData.cFileName, L".") != 0 &&
2127 wcscmp(wFileData.cFileName, L"..") != 0) {
2128 v = PyUnicode_FromUnicode(wFileData.cFileName, wcslen(wFileData.cFileName));
2129 if (v == NULL) {
2130 Py_DECREF(d);
2131 d = NULL;
2132 break;
2134 if (PyList_Append(d, v) != 0) {
2135 Py_DECREF(v);
2136 Py_DECREF(d);
2137 d = NULL;
2138 break;
2140 Py_DECREF(v);
2142 Py_BEGIN_ALLOW_THREADS
2143 result = FindNextFileW(hFindFile, &wFileData);
2144 Py_END_ALLOW_THREADS
2145 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2146 it got to the end of the directory. */
2147 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2148 Py_DECREF(d);
2149 win32_error_unicode("FindNextFileW", wnamebuf);
2150 FindClose(hFindFile);
2151 free(wnamebuf);
2152 return NULL;
2154 } while (result == TRUE);
2156 if (FindClose(hFindFile) == FALSE) {
2157 Py_DECREF(d);
2158 win32_error_unicode("FindClose", wnamebuf);
2159 free(wnamebuf);
2160 return NULL;
2162 free(wnamebuf);
2163 return d;
2165 /* Drop the argument parsing error as narrow strings
2166 are also valid. */
2167 PyErr_Clear();
2169 #endif
2171 if (!PyArg_ParseTuple(args, "et#:listdir",
2172 Py_FileSystemDefaultEncoding, &bufptr, &len))
2173 return NULL;
2174 if (len > 0) {
2175 char ch = namebuf[len-1];
2176 if (ch != SEP && ch != ALTSEP && ch != ':')
2177 namebuf[len++] = '/';
2179 strcpy(namebuf + len, "*.*");
2181 if ((d = PyList_New(0)) == NULL)
2182 return NULL;
2184 hFindFile = FindFirstFile(namebuf, &FileData);
2185 if (hFindFile == INVALID_HANDLE_VALUE) {
2186 int error = GetLastError();
2187 if (error == ERROR_FILE_NOT_FOUND)
2188 return d;
2189 Py_DECREF(d);
2190 return win32_error("FindFirstFile", namebuf);
2192 do {
2193 /* Skip over . and .. */
2194 if (strcmp(FileData.cFileName, ".") != 0 &&
2195 strcmp(FileData.cFileName, "..") != 0) {
2196 v = PyString_FromString(FileData.cFileName);
2197 if (v == NULL) {
2198 Py_DECREF(d);
2199 d = NULL;
2200 break;
2202 if (PyList_Append(d, v) != 0) {
2203 Py_DECREF(v);
2204 Py_DECREF(d);
2205 d = NULL;
2206 break;
2208 Py_DECREF(v);
2210 Py_BEGIN_ALLOW_THREADS
2211 result = FindNextFile(hFindFile, &FileData);
2212 Py_END_ALLOW_THREADS
2213 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2214 it got to the end of the directory. */
2215 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2216 Py_DECREF(d);
2217 win32_error("FindNextFile", namebuf);
2218 FindClose(hFindFile);
2219 return NULL;
2221 } while (result == TRUE);
2223 if (FindClose(hFindFile) == FALSE) {
2224 Py_DECREF(d);
2225 return win32_error("FindClose", namebuf);
2228 return d;
2230 #elif defined(PYOS_OS2)
2232 #ifndef MAX_PATH
2233 #define MAX_PATH CCHMAXPATH
2234 #endif
2235 char *name, *pt;
2236 Py_ssize_t len;
2237 PyObject *d, *v;
2238 char namebuf[MAX_PATH+5];
2239 HDIR hdir = 1;
2240 ULONG srchcnt = 1;
2241 FILEFINDBUF3 ep;
2242 APIRET rc;
2244 if (!PyArg_ParseTuple(args, "t#:listdir", &name, &len))
2245 return NULL;
2246 if (len >= MAX_PATH) {
2247 PyErr_SetString(PyExc_ValueError, "path too long");
2248 return NULL;
2250 strcpy(namebuf, name);
2251 for (pt = namebuf; *pt; pt++)
2252 if (*pt == ALTSEP)
2253 *pt = SEP;
2254 if (namebuf[len-1] != SEP)
2255 namebuf[len++] = SEP;
2256 strcpy(namebuf + len, "*.*");
2258 if ((d = PyList_New(0)) == NULL)
2259 return NULL;
2261 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
2262 &hdir, /* Handle to Use While Search Directory */
2263 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
2264 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
2265 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
2266 FIL_STANDARD); /* Format of Entry (EAs or Not) */
2268 if (rc != NO_ERROR) {
2269 errno = ENOENT;
2270 return posix_error_with_filename(name);
2273 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
2274 do {
2275 if (ep.achName[0] == '.'
2276 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
2277 continue; /* Skip Over "." and ".." Names */
2279 strcpy(namebuf, ep.achName);
2281 /* Leave Case of Name Alone -- In Native Form */
2282 /* (Removed Forced Lowercasing Code) */
2284 v = PyString_FromString(namebuf);
2285 if (v == NULL) {
2286 Py_DECREF(d);
2287 d = NULL;
2288 break;
2290 if (PyList_Append(d, v) != 0) {
2291 Py_DECREF(v);
2292 Py_DECREF(d);
2293 d = NULL;
2294 break;
2296 Py_DECREF(v);
2297 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
2300 return d;
2301 #else
2303 char *name = NULL;
2304 PyObject *d, *v;
2305 DIR *dirp;
2306 struct dirent *ep;
2307 int arg_is_unicode = 1;
2309 errno = 0;
2310 if (!PyArg_ParseTuple(args, "U:listdir", &v)) {
2311 arg_is_unicode = 0;
2312 PyErr_Clear();
2314 if (!PyArg_ParseTuple(args, "et:listdir", Py_FileSystemDefaultEncoding, &name))
2315 return NULL;
2316 if ((dirp = opendir(name)) == NULL) {
2317 return posix_error_with_allocated_filename(name);
2319 if ((d = PyList_New(0)) == NULL) {
2320 closedir(dirp);
2321 PyMem_Free(name);
2322 return NULL;
2324 for (;;) {
2325 Py_BEGIN_ALLOW_THREADS
2326 ep = readdir(dirp);
2327 Py_END_ALLOW_THREADS
2328 if (ep == NULL)
2329 break;
2330 if (ep->d_name[0] == '.' &&
2331 (NAMLEN(ep) == 1 ||
2332 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
2333 continue;
2334 v = PyString_FromStringAndSize(ep->d_name, NAMLEN(ep));
2335 if (v == NULL) {
2336 Py_DECREF(d);
2337 d = NULL;
2338 break;
2340 #ifdef Py_USING_UNICODE
2341 if (arg_is_unicode) {
2342 PyObject *w;
2344 w = PyUnicode_FromEncodedObject(v,
2345 Py_FileSystemDefaultEncoding,
2346 "strict");
2347 if (w != NULL) {
2348 Py_DECREF(v);
2349 v = w;
2351 else {
2352 /* fall back to the original byte string, as
2353 discussed in patch #683592 */
2354 PyErr_Clear();
2357 #endif
2358 if (PyList_Append(d, v) != 0) {
2359 Py_DECREF(v);
2360 Py_DECREF(d);
2361 d = NULL;
2362 break;
2364 Py_DECREF(v);
2366 if (errno != 0 && d != NULL) {
2367 /* readdir() returned NULL and set errno */
2368 closedir(dirp);
2369 Py_DECREF(d);
2370 return posix_error_with_allocated_filename(name);
2372 closedir(dirp);
2373 PyMem_Free(name);
2375 return d;
2377 #endif /* which OS */
2378 } /* end of posix_listdir */
2380 #ifdef MS_WINDOWS
2381 /* A helper function for abspath on win32 */
2382 static PyObject *
2383 posix__getfullpathname(PyObject *self, PyObject *args)
2385 /* assume encoded strings wont more than double no of chars */
2386 char inbuf[MAX_PATH*2];
2387 char *inbufp = inbuf;
2388 Py_ssize_t insize = sizeof(inbuf);
2389 char outbuf[MAX_PATH*2];
2390 char *temp;
2391 #ifdef Py_WIN_WIDE_FILENAMES
2392 if (unicode_file_names()) {
2393 PyUnicodeObject *po;
2394 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po)) {
2395 Py_UNICODE woutbuf[MAX_PATH*2];
2396 Py_UNICODE *wtemp;
2397 if (!GetFullPathNameW(PyUnicode_AS_UNICODE(po),
2398 sizeof(woutbuf)/sizeof(woutbuf[0]),
2399 woutbuf, &wtemp))
2400 return win32_error("GetFullPathName", "");
2401 return PyUnicode_FromUnicode(woutbuf, wcslen(woutbuf));
2403 /* Drop the argument parsing error as narrow strings
2404 are also valid. */
2405 PyErr_Clear();
2407 #endif
2408 if (!PyArg_ParseTuple (args, "et#:_getfullpathname",
2409 Py_FileSystemDefaultEncoding, &inbufp,
2410 &insize))
2411 return NULL;
2412 if (!GetFullPathName(inbuf, sizeof(outbuf)/sizeof(outbuf[0]),
2413 outbuf, &temp))
2414 return win32_error("GetFullPathName", inbuf);
2415 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
2416 return PyUnicode_Decode(outbuf, strlen(outbuf),
2417 Py_FileSystemDefaultEncoding, NULL);
2419 return PyString_FromString(outbuf);
2420 } /* end of posix__getfullpathname */
2421 #endif /* MS_WINDOWS */
2423 PyDoc_STRVAR(posix_mkdir__doc__,
2424 "mkdir(path [, mode=0777])\n\n\
2425 Create a directory.");
2427 static PyObject *
2428 posix_mkdir(PyObject *self, PyObject *args)
2430 int res;
2431 char *path = NULL;
2432 int mode = 0777;
2434 #ifdef Py_WIN_WIDE_FILENAMES
2435 if (unicode_file_names()) {
2436 PyUnicodeObject *po;
2437 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode)) {
2438 Py_BEGIN_ALLOW_THREADS
2439 /* PyUnicode_AS_UNICODE OK without thread lock as
2440 it is a simple dereference. */
2441 res = CreateDirectoryW(PyUnicode_AS_UNICODE(po), NULL);
2442 Py_END_ALLOW_THREADS
2443 if (!res)
2444 return win32_error_unicode("mkdir", PyUnicode_AS_UNICODE(po));
2445 Py_INCREF(Py_None);
2446 return Py_None;
2448 /* Drop the argument parsing error as narrow strings
2449 are also valid. */
2450 PyErr_Clear();
2452 if (!PyArg_ParseTuple(args, "et|i:mkdir",
2453 Py_FileSystemDefaultEncoding, &path, &mode))
2454 return NULL;
2455 Py_BEGIN_ALLOW_THREADS
2456 /* PyUnicode_AS_UNICODE OK without thread lock as
2457 it is a simple dereference. */
2458 res = CreateDirectoryA(path, NULL);
2459 Py_END_ALLOW_THREADS
2460 if (!res) {
2461 win32_error("mkdir", path);
2462 PyMem_Free(path);
2463 return NULL;
2465 PyMem_Free(path);
2466 Py_INCREF(Py_None);
2467 return Py_None;
2468 #else
2470 if (!PyArg_ParseTuple(args, "et|i:mkdir",
2471 Py_FileSystemDefaultEncoding, &path, &mode))
2472 return NULL;
2473 Py_BEGIN_ALLOW_THREADS
2474 #if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
2475 res = mkdir(path);
2476 #else
2477 res = mkdir(path, mode);
2478 #endif
2479 Py_END_ALLOW_THREADS
2480 if (res < 0)
2481 return posix_error_with_allocated_filename(path);
2482 PyMem_Free(path);
2483 Py_INCREF(Py_None);
2484 return Py_None;
2485 #endif
2489 /* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
2490 #if defined(HAVE_SYS_RESOURCE_H)
2491 #include <sys/resource.h>
2492 #endif
2495 #ifdef HAVE_NICE
2496 PyDoc_STRVAR(posix_nice__doc__,
2497 "nice(inc) -> new_priority\n\n\
2498 Decrease the priority of process by inc and return the new priority.");
2500 static PyObject *
2501 posix_nice(PyObject *self, PyObject *args)
2503 int increment, value;
2505 if (!PyArg_ParseTuple(args, "i:nice", &increment))
2506 return NULL;
2508 /* There are two flavours of 'nice': one that returns the new
2509 priority (as required by almost all standards out there) and the
2510 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
2511 the use of getpriority() to get the new priority.
2513 If we are of the nice family that returns the new priority, we
2514 need to clear errno before the call, and check if errno is filled
2515 before calling posix_error() on a returnvalue of -1, because the
2516 -1 may be the actual new priority! */
2518 errno = 0;
2519 value = nice(increment);
2520 #if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
2521 if (value == 0)
2522 value = getpriority(PRIO_PROCESS, 0);
2523 #endif
2524 if (value == -1 && errno != 0)
2525 /* either nice() or getpriority() returned an error */
2526 return posix_error();
2527 return PyInt_FromLong((long) value);
2529 #endif /* HAVE_NICE */
2531 PyDoc_STRVAR(posix_rename__doc__,
2532 "rename(old, new)\n\n\
2533 Rename a file or directory.");
2535 static PyObject *
2536 posix_rename(PyObject *self, PyObject *args)
2538 #ifdef MS_WINDOWS
2539 PyObject *o1, *o2;
2540 char *p1, *p2;
2541 BOOL result;
2542 if (unicode_file_names()) {
2543 if (!PyArg_ParseTuple(args, "O&O&:rename",
2544 convert_to_unicode, &o1,
2545 convert_to_unicode, &o2))
2546 PyErr_Clear();
2547 else {
2548 Py_BEGIN_ALLOW_THREADS
2549 result = MoveFileW(PyUnicode_AsUnicode(o1),
2550 PyUnicode_AsUnicode(o2));
2551 Py_END_ALLOW_THREADS
2552 Py_DECREF(o1);
2553 Py_DECREF(o2);
2554 if (!result)
2555 return win32_error("rename", NULL);
2556 Py_INCREF(Py_None);
2557 return Py_None;
2560 if (!PyArg_ParseTuple(args, "ss:rename", &p1, &p2))
2561 return NULL;
2562 Py_BEGIN_ALLOW_THREADS
2563 result = MoveFileA(p1, p2);
2564 Py_END_ALLOW_THREADS
2565 if (!result)
2566 return win32_error("rename", NULL);
2567 Py_INCREF(Py_None);
2568 return Py_None;
2569 #else
2570 return posix_2str(args, "etet:rename", rename);
2571 #endif
2575 PyDoc_STRVAR(posix_rmdir__doc__,
2576 "rmdir(path)\n\n\
2577 Remove a directory.");
2579 static PyObject *
2580 posix_rmdir(PyObject *self, PyObject *args)
2582 #ifdef MS_WINDOWS
2583 return win32_1str(args, "rmdir", "s:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
2584 #else
2585 return posix_1str(args, "et:rmdir", rmdir);
2586 #endif
2590 PyDoc_STRVAR(posix_stat__doc__,
2591 "stat(path) -> stat result\n\n\
2592 Perform a stat system call on the given path.");
2594 static PyObject *
2595 posix_stat(PyObject *self, PyObject *args)
2597 #ifdef MS_WINDOWS
2598 return posix_do_stat(self, args, "et:stat", STAT, "U:stat", win32_wstat);
2599 #else
2600 return posix_do_stat(self, args, "et:stat", STAT, NULL, NULL);
2601 #endif
2605 #ifdef HAVE_SYSTEM
2606 PyDoc_STRVAR(posix_system__doc__,
2607 "system(command) -> exit_status\n\n\
2608 Execute the command (a string) in a subshell.");
2610 static PyObject *
2611 posix_system(PyObject *self, PyObject *args)
2613 char *command;
2614 long sts;
2615 if (!PyArg_ParseTuple(args, "s:system", &command))
2616 return NULL;
2617 Py_BEGIN_ALLOW_THREADS
2618 sts = system(command);
2619 Py_END_ALLOW_THREADS
2620 return PyInt_FromLong(sts);
2622 #endif
2625 PyDoc_STRVAR(posix_umask__doc__,
2626 "umask(new_mask) -> old_mask\n\n\
2627 Set the current numeric umask and return the previous umask.");
2629 static PyObject *
2630 posix_umask(PyObject *self, PyObject *args)
2632 int i;
2633 if (!PyArg_ParseTuple(args, "i:umask", &i))
2634 return NULL;
2635 i = (int)umask(i);
2636 if (i < 0)
2637 return posix_error();
2638 return PyInt_FromLong((long)i);
2642 PyDoc_STRVAR(posix_unlink__doc__,
2643 "unlink(path)\n\n\
2644 Remove a file (same as remove(path)).");
2646 PyDoc_STRVAR(posix_remove__doc__,
2647 "remove(path)\n\n\
2648 Remove a file (same as unlink(path)).");
2650 static PyObject *
2651 posix_unlink(PyObject *self, PyObject *args)
2653 #ifdef MS_WINDOWS
2654 return win32_1str(args, "remove", "s:remove", DeleteFileA, "U:remove", DeleteFileW);
2655 #else
2656 return posix_1str(args, "et:remove", unlink);
2657 #endif
2661 #ifdef HAVE_UNAME
2662 PyDoc_STRVAR(posix_uname__doc__,
2663 "uname() -> (sysname, nodename, release, version, machine)\n\n\
2664 Return a tuple identifying the current operating system.");
2666 static PyObject *
2667 posix_uname(PyObject *self, PyObject *noargs)
2669 struct utsname u;
2670 int res;
2672 Py_BEGIN_ALLOW_THREADS
2673 res = uname(&u);
2674 Py_END_ALLOW_THREADS
2675 if (res < 0)
2676 return posix_error();
2677 return Py_BuildValue("(sssss)",
2678 u.sysname,
2679 u.nodename,
2680 u.release,
2681 u.version,
2682 u.machine);
2684 #endif /* HAVE_UNAME */
2686 static int
2687 extract_time(PyObject *t, long* sec, long* usec)
2689 long intval;
2690 if (PyFloat_Check(t)) {
2691 double tval = PyFloat_AsDouble(t);
2692 PyObject *intobj = Py_TYPE(t)->tp_as_number->nb_int(t);
2693 if (!intobj)
2694 return -1;
2695 intval = PyInt_AsLong(intobj);
2696 Py_DECREF(intobj);
2697 if (intval == -1 && PyErr_Occurred())
2698 return -1;
2699 *sec = intval;
2700 *usec = (long)((tval - intval) * 1e6); /* can't exceed 1000000 */
2701 if (*usec < 0)
2702 /* If rounding gave us a negative number,
2703 truncate. */
2704 *usec = 0;
2705 return 0;
2707 intval = PyInt_AsLong(t);
2708 if (intval == -1 && PyErr_Occurred())
2709 return -1;
2710 *sec = intval;
2711 *usec = 0;
2712 return 0;
2715 PyDoc_STRVAR(posix_utime__doc__,
2716 "utime(path, (atime, mtime))\n\
2717 utime(path, None)\n\n\
2718 Set the access and modified time of the file to the given values. If the\n\
2719 second form is used, set the access and modified times to the current time.");
2721 static PyObject *
2722 posix_utime(PyObject *self, PyObject *args)
2724 #ifdef Py_WIN_WIDE_FILENAMES
2725 PyObject *arg;
2726 PyUnicodeObject *obwpath;
2727 wchar_t *wpath = NULL;
2728 char *apath = NULL;
2729 HANDLE hFile;
2730 long atimesec, mtimesec, ausec, musec;
2731 FILETIME atime, mtime;
2732 PyObject *result = NULL;
2734 if (unicode_file_names()) {
2735 if (PyArg_ParseTuple(args, "UO|:utime", &obwpath, &arg)) {
2736 wpath = PyUnicode_AS_UNICODE(obwpath);
2737 Py_BEGIN_ALLOW_THREADS
2738 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
2739 NULL, OPEN_EXISTING,
2740 FILE_FLAG_BACKUP_SEMANTICS, NULL);
2741 Py_END_ALLOW_THREADS
2742 if (hFile == INVALID_HANDLE_VALUE)
2743 return win32_error_unicode("utime", wpath);
2744 } else
2745 /* Drop the argument parsing error as narrow strings
2746 are also valid. */
2747 PyErr_Clear();
2749 if (!wpath) {
2750 if (!PyArg_ParseTuple(args, "etO:utime",
2751 Py_FileSystemDefaultEncoding, &apath, &arg))
2752 return NULL;
2753 Py_BEGIN_ALLOW_THREADS
2754 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
2755 NULL, OPEN_EXISTING,
2756 FILE_FLAG_BACKUP_SEMANTICS, NULL);
2757 Py_END_ALLOW_THREADS
2758 if (hFile == INVALID_HANDLE_VALUE) {
2759 win32_error("utime", apath);
2760 PyMem_Free(apath);
2761 return NULL;
2763 PyMem_Free(apath);
2766 if (arg == Py_None) {
2767 SYSTEMTIME now;
2768 GetSystemTime(&now);
2769 if (!SystemTimeToFileTime(&now, &mtime) ||
2770 !SystemTimeToFileTime(&now, &atime)) {
2771 win32_error("utime", NULL);
2772 goto done;
2775 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
2776 PyErr_SetString(PyExc_TypeError,
2777 "utime() arg 2 must be a tuple (atime, mtime)");
2778 goto done;
2780 else {
2781 if (extract_time(PyTuple_GET_ITEM(arg, 0),
2782 &atimesec, &ausec) == -1)
2783 goto done;
2784 time_t_to_FILE_TIME(atimesec, 1000*ausec, &atime);
2785 if (extract_time(PyTuple_GET_ITEM(arg, 1),
2786 &mtimesec, &musec) == -1)
2787 goto done;
2788 time_t_to_FILE_TIME(mtimesec, 1000*musec, &mtime);
2790 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
2791 /* Avoid putting the file name into the error here,
2792 as that may confuse the user into believing that
2793 something is wrong with the file, when it also
2794 could be the time stamp that gives a problem. */
2795 win32_error("utime", NULL);
2797 Py_INCREF(Py_None);
2798 result = Py_None;
2799 done:
2800 CloseHandle(hFile);
2801 return result;
2802 #else /* Py_WIN_WIDE_FILENAMES */
2804 char *path = NULL;
2805 long atime, mtime, ausec, musec;
2806 int res;
2807 PyObject* arg;
2809 #if defined(HAVE_UTIMES)
2810 struct timeval buf[2];
2811 #define ATIME buf[0].tv_sec
2812 #define MTIME buf[1].tv_sec
2813 #elif defined(HAVE_UTIME_H)
2814 /* XXX should define struct utimbuf instead, above */
2815 struct utimbuf buf;
2816 #define ATIME buf.actime
2817 #define MTIME buf.modtime
2818 #define UTIME_ARG &buf
2819 #else /* HAVE_UTIMES */
2820 time_t buf[2];
2821 #define ATIME buf[0]
2822 #define MTIME buf[1]
2823 #define UTIME_ARG buf
2824 #endif /* HAVE_UTIMES */
2827 if (!PyArg_ParseTuple(args, "etO:utime",
2828 Py_FileSystemDefaultEncoding, &path, &arg))
2829 return NULL;
2830 if (arg == Py_None) {
2831 /* optional time values not given */
2832 Py_BEGIN_ALLOW_THREADS
2833 res = utime(path, NULL);
2834 Py_END_ALLOW_THREADS
2836 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
2837 PyErr_SetString(PyExc_TypeError,
2838 "utime() arg 2 must be a tuple (atime, mtime)");
2839 PyMem_Free(path);
2840 return NULL;
2842 else {
2843 if (extract_time(PyTuple_GET_ITEM(arg, 0),
2844 &atime, &ausec) == -1) {
2845 PyMem_Free(path);
2846 return NULL;
2848 if (extract_time(PyTuple_GET_ITEM(arg, 1),
2849 &mtime, &musec) == -1) {
2850 PyMem_Free(path);
2851 return NULL;
2853 ATIME = atime;
2854 MTIME = mtime;
2855 #ifdef HAVE_UTIMES
2856 buf[0].tv_usec = ausec;
2857 buf[1].tv_usec = musec;
2858 Py_BEGIN_ALLOW_THREADS
2859 res = utimes(path, buf);
2860 Py_END_ALLOW_THREADS
2861 #else
2862 Py_BEGIN_ALLOW_THREADS
2863 res = utime(path, UTIME_ARG);
2864 Py_END_ALLOW_THREADS
2865 #endif /* HAVE_UTIMES */
2867 if (res < 0) {
2868 return posix_error_with_allocated_filename(path);
2870 PyMem_Free(path);
2871 Py_INCREF(Py_None);
2872 return Py_None;
2873 #undef UTIME_ARG
2874 #undef ATIME
2875 #undef MTIME
2876 #endif /* Py_WIN_WIDE_FILENAMES */
2880 /* Process operations */
2882 PyDoc_STRVAR(posix__exit__doc__,
2883 "_exit(status)\n\n\
2884 Exit to the system with specified status, without normal exit processing.");
2886 static PyObject *
2887 posix__exit(PyObject *self, PyObject *args)
2889 int sts;
2890 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
2891 return NULL;
2892 _exit(sts);
2893 return NULL; /* Make gcc -Wall happy */
2896 #if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
2897 static void
2898 free_string_array(char **array, Py_ssize_t count)
2900 Py_ssize_t i;
2901 for (i = 0; i < count; i++)
2902 PyMem_Free(array[i]);
2903 PyMem_DEL(array);
2905 #endif
2908 #ifdef HAVE_EXECV
2909 PyDoc_STRVAR(posix_execv__doc__,
2910 "execv(path, args)\n\n\
2911 Execute an executable path with arguments, replacing current process.\n\
2913 path: path of executable file\n\
2914 args: tuple or list of strings");
2916 static PyObject *
2917 posix_execv(PyObject *self, PyObject *args)
2919 char *path;
2920 PyObject *argv;
2921 char **argvlist;
2922 Py_ssize_t i, argc;
2923 PyObject *(*getitem)(PyObject *, Py_ssize_t);
2925 /* execv has two arguments: (path, argv), where
2926 argv is a list or tuple of strings. */
2928 if (!PyArg_ParseTuple(args, "etO:execv",
2929 Py_FileSystemDefaultEncoding,
2930 &path, &argv))
2931 return NULL;
2932 if (PyList_Check(argv)) {
2933 argc = PyList_Size(argv);
2934 getitem = PyList_GetItem;
2936 else if (PyTuple_Check(argv)) {
2937 argc = PyTuple_Size(argv);
2938 getitem = PyTuple_GetItem;
2940 else {
2941 PyErr_SetString(PyExc_TypeError, "execv() arg 2 must be a tuple or list");
2942 PyMem_Free(path);
2943 return NULL;
2946 argvlist = PyMem_NEW(char *, argc+1);
2947 if (argvlist == NULL) {
2948 PyMem_Free(path);
2949 return PyErr_NoMemory();
2951 for (i = 0; i < argc; i++) {
2952 if (!PyArg_Parse((*getitem)(argv, i), "et",
2953 Py_FileSystemDefaultEncoding,
2954 &argvlist[i])) {
2955 free_string_array(argvlist, i);
2956 PyErr_SetString(PyExc_TypeError,
2957 "execv() arg 2 must contain only strings");
2958 PyMem_Free(path);
2959 return NULL;
2963 argvlist[argc] = NULL;
2965 execv(path, argvlist);
2967 /* If we get here it's definitely an error */
2969 free_string_array(argvlist, argc);
2970 PyMem_Free(path);
2971 return posix_error();
2975 PyDoc_STRVAR(posix_execve__doc__,
2976 "execve(path, args, env)\n\n\
2977 Execute a path with arguments and environment, replacing current process.\n\
2979 path: path of executable file\n\
2980 args: tuple or list of arguments\n\
2981 env: dictionary of strings mapping to strings");
2983 static PyObject *
2984 posix_execve(PyObject *self, PyObject *args)
2986 char *path;
2987 PyObject *argv, *env;
2988 char **argvlist;
2989 char **envlist;
2990 PyObject *key, *val, *keys=NULL, *vals=NULL;
2991 Py_ssize_t i, pos, argc, envc;
2992 PyObject *(*getitem)(PyObject *, Py_ssize_t);
2993 Py_ssize_t lastarg = 0;
2995 /* execve has three arguments: (path, argv, env), where
2996 argv is a list or tuple of strings and env is a dictionary
2997 like posix.environ. */
2999 if (!PyArg_ParseTuple(args, "etOO:execve",
3000 Py_FileSystemDefaultEncoding,
3001 &path, &argv, &env))
3002 return NULL;
3003 if (PyList_Check(argv)) {
3004 argc = PyList_Size(argv);
3005 getitem = PyList_GetItem;
3007 else if (PyTuple_Check(argv)) {
3008 argc = PyTuple_Size(argv);
3009 getitem = PyTuple_GetItem;
3011 else {
3012 PyErr_SetString(PyExc_TypeError,
3013 "execve() arg 2 must be a tuple or list");
3014 goto fail_0;
3016 if (!PyMapping_Check(env)) {
3017 PyErr_SetString(PyExc_TypeError,
3018 "execve() arg 3 must be a mapping object");
3019 goto fail_0;
3022 argvlist = PyMem_NEW(char *, argc+1);
3023 if (argvlist == NULL) {
3024 PyErr_NoMemory();
3025 goto fail_0;
3027 for (i = 0; i < argc; i++) {
3028 if (!PyArg_Parse((*getitem)(argv, i),
3029 "et;execve() arg 2 must contain only strings",
3030 Py_FileSystemDefaultEncoding,
3031 &argvlist[i]))
3033 lastarg = i;
3034 goto fail_1;
3037 lastarg = argc;
3038 argvlist[argc] = NULL;
3040 i = PyMapping_Size(env);
3041 if (i < 0)
3042 goto fail_1;
3043 envlist = PyMem_NEW(char *, i + 1);
3044 if (envlist == NULL) {
3045 PyErr_NoMemory();
3046 goto fail_1;
3048 envc = 0;
3049 keys = PyMapping_Keys(env);
3050 vals = PyMapping_Values(env);
3051 if (!keys || !vals)
3052 goto fail_2;
3053 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3054 PyErr_SetString(PyExc_TypeError,
3055 "execve(): env.keys() or env.values() is not a list");
3056 goto fail_2;
3059 for (pos = 0; pos < i; pos++) {
3060 char *p, *k, *v;
3061 size_t len;
3063 key = PyList_GetItem(keys, pos);
3064 val = PyList_GetItem(vals, pos);
3065 if (!key || !val)
3066 goto fail_2;
3068 if (!PyArg_Parse(
3069 key,
3070 "s;execve() arg 3 contains a non-string key",
3071 &k) ||
3072 !PyArg_Parse(
3073 val,
3074 "s;execve() arg 3 contains a non-string value",
3075 &v))
3077 goto fail_2;
3080 #if defined(PYOS_OS2)
3081 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
3082 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
3083 #endif
3084 len = PyString_Size(key) + PyString_Size(val) + 2;
3085 p = PyMem_NEW(char, len);
3086 if (p == NULL) {
3087 PyErr_NoMemory();
3088 goto fail_2;
3090 PyOS_snprintf(p, len, "%s=%s", k, v);
3091 envlist[envc++] = p;
3092 #if defined(PYOS_OS2)
3094 #endif
3096 envlist[envc] = 0;
3098 execve(path, argvlist, envlist);
3100 /* If we get here it's definitely an error */
3102 (void) posix_error();
3104 fail_2:
3105 while (--envc >= 0)
3106 PyMem_DEL(envlist[envc]);
3107 PyMem_DEL(envlist);
3108 fail_1:
3109 free_string_array(argvlist, lastarg);
3110 Py_XDECREF(vals);
3111 Py_XDECREF(keys);
3112 fail_0:
3113 PyMem_Free(path);
3114 return NULL;
3116 #endif /* HAVE_EXECV */
3119 #ifdef HAVE_SPAWNV
3120 PyDoc_STRVAR(posix_spawnv__doc__,
3121 "spawnv(mode, path, args)\n\n\
3122 Execute the program 'path' in a new process.\n\
3124 mode: mode of process creation\n\
3125 path: path of executable file\n\
3126 args: tuple or list of strings");
3128 static PyObject *
3129 posix_spawnv(PyObject *self, PyObject *args)
3131 char *path;
3132 PyObject *argv;
3133 char **argvlist;
3134 int mode, i;
3135 Py_ssize_t argc;
3136 Py_intptr_t spawnval;
3137 PyObject *(*getitem)(PyObject *, Py_ssize_t);
3139 /* spawnv has three arguments: (mode, path, argv), where
3140 argv is a list or tuple of strings. */
3142 if (!PyArg_ParseTuple(args, "ietO:spawnv", &mode,
3143 Py_FileSystemDefaultEncoding,
3144 &path, &argv))
3145 return NULL;
3146 if (PyList_Check(argv)) {
3147 argc = PyList_Size(argv);
3148 getitem = PyList_GetItem;
3150 else if (PyTuple_Check(argv)) {
3151 argc = PyTuple_Size(argv);
3152 getitem = PyTuple_GetItem;
3154 else {
3155 PyErr_SetString(PyExc_TypeError,
3156 "spawnv() arg 2 must be a tuple or list");
3157 PyMem_Free(path);
3158 return NULL;
3161 argvlist = PyMem_NEW(char *, argc+1);
3162 if (argvlist == NULL) {
3163 PyMem_Free(path);
3164 return PyErr_NoMemory();
3166 for (i = 0; i < argc; i++) {
3167 if (!PyArg_Parse((*getitem)(argv, i), "et",
3168 Py_FileSystemDefaultEncoding,
3169 &argvlist[i])) {
3170 free_string_array(argvlist, i);
3171 PyErr_SetString(
3172 PyExc_TypeError,
3173 "spawnv() arg 2 must contain only strings");
3174 PyMem_Free(path);
3175 return NULL;
3178 argvlist[argc] = NULL;
3180 #if defined(PYOS_OS2) && defined(PYCC_GCC)
3181 Py_BEGIN_ALLOW_THREADS
3182 spawnval = spawnv(mode, path, argvlist);
3183 Py_END_ALLOW_THREADS
3184 #else
3185 if (mode == _OLD_P_OVERLAY)
3186 mode = _P_OVERLAY;
3188 Py_BEGIN_ALLOW_THREADS
3189 spawnval = _spawnv(mode, path, argvlist);
3190 Py_END_ALLOW_THREADS
3191 #endif
3193 free_string_array(argvlist, argc);
3194 PyMem_Free(path);
3196 if (spawnval == -1)
3197 return posix_error();
3198 else
3199 #if SIZEOF_LONG == SIZEOF_VOID_P
3200 return Py_BuildValue("l", (long) spawnval);
3201 #else
3202 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
3203 #endif
3207 PyDoc_STRVAR(posix_spawnve__doc__,
3208 "spawnve(mode, path, args, env)\n\n\
3209 Execute the program 'path' in a new process.\n\
3211 mode: mode of process creation\n\
3212 path: path of executable file\n\
3213 args: tuple or list of arguments\n\
3214 env: dictionary of strings mapping to strings");
3216 static PyObject *
3217 posix_spawnve(PyObject *self, PyObject *args)
3219 char *path;
3220 PyObject *argv, *env;
3221 char **argvlist;
3222 char **envlist;
3223 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
3224 int mode, pos, envc;
3225 Py_ssize_t argc, i;
3226 Py_intptr_t spawnval;
3227 PyObject *(*getitem)(PyObject *, Py_ssize_t);
3228 Py_ssize_t lastarg = 0;
3230 /* spawnve has four arguments: (mode, path, argv, env), where
3231 argv is a list or tuple of strings and env is a dictionary
3232 like posix.environ. */
3234 if (!PyArg_ParseTuple(args, "ietOO:spawnve", &mode,
3235 Py_FileSystemDefaultEncoding,
3236 &path, &argv, &env))
3237 return NULL;
3238 if (PyList_Check(argv)) {
3239 argc = PyList_Size(argv);
3240 getitem = PyList_GetItem;
3242 else if (PyTuple_Check(argv)) {
3243 argc = PyTuple_Size(argv);
3244 getitem = PyTuple_GetItem;
3246 else {
3247 PyErr_SetString(PyExc_TypeError,
3248 "spawnve() arg 2 must be a tuple or list");
3249 goto fail_0;
3251 if (!PyMapping_Check(env)) {
3252 PyErr_SetString(PyExc_TypeError,
3253 "spawnve() arg 3 must be a mapping object");
3254 goto fail_0;
3257 argvlist = PyMem_NEW(char *, argc+1);
3258 if (argvlist == NULL) {
3259 PyErr_NoMemory();
3260 goto fail_0;
3262 for (i = 0; i < argc; i++) {
3263 if (!PyArg_Parse((*getitem)(argv, i),
3264 "et;spawnve() arg 2 must contain only strings",
3265 Py_FileSystemDefaultEncoding,
3266 &argvlist[i]))
3268 lastarg = i;
3269 goto fail_1;
3272 lastarg = argc;
3273 argvlist[argc] = NULL;
3275 i = PyMapping_Size(env);
3276 if (i < 0)
3277 goto fail_1;
3278 envlist = PyMem_NEW(char *, i + 1);
3279 if (envlist == NULL) {
3280 PyErr_NoMemory();
3281 goto fail_1;
3283 envc = 0;
3284 keys = PyMapping_Keys(env);
3285 vals = PyMapping_Values(env);
3286 if (!keys || !vals)
3287 goto fail_2;
3288 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3289 PyErr_SetString(PyExc_TypeError,
3290 "spawnve(): env.keys() or env.values() is not a list");
3291 goto fail_2;
3294 for (pos = 0; pos < i; pos++) {
3295 char *p, *k, *v;
3296 size_t len;
3298 key = PyList_GetItem(keys, pos);
3299 val = PyList_GetItem(vals, pos);
3300 if (!key || !val)
3301 goto fail_2;
3303 if (!PyArg_Parse(
3304 key,
3305 "s;spawnve() arg 3 contains a non-string key",
3306 &k) ||
3307 !PyArg_Parse(
3308 val,
3309 "s;spawnve() arg 3 contains a non-string value",
3310 &v))
3312 goto fail_2;
3314 len = PyString_Size(key) + PyString_Size(val) + 2;
3315 p = PyMem_NEW(char, len);
3316 if (p == NULL) {
3317 PyErr_NoMemory();
3318 goto fail_2;
3320 PyOS_snprintf(p, len, "%s=%s", k, v);
3321 envlist[envc++] = p;
3323 envlist[envc] = 0;
3325 #if defined(PYOS_OS2) && defined(PYCC_GCC)
3326 Py_BEGIN_ALLOW_THREADS
3327 spawnval = spawnve(mode, path, argvlist, envlist);
3328 Py_END_ALLOW_THREADS
3329 #else
3330 if (mode == _OLD_P_OVERLAY)
3331 mode = _P_OVERLAY;
3333 Py_BEGIN_ALLOW_THREADS
3334 spawnval = _spawnve(mode, path, argvlist, envlist);
3335 Py_END_ALLOW_THREADS
3336 #endif
3338 if (spawnval == -1)
3339 (void) posix_error();
3340 else
3341 #if SIZEOF_LONG == SIZEOF_VOID_P
3342 res = Py_BuildValue("l", (long) spawnval);
3343 #else
3344 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
3345 #endif
3347 fail_2:
3348 while (--envc >= 0)
3349 PyMem_DEL(envlist[envc]);
3350 PyMem_DEL(envlist);
3351 fail_1:
3352 free_string_array(argvlist, lastarg);
3353 Py_XDECREF(vals);
3354 Py_XDECREF(keys);
3355 fail_0:
3356 PyMem_Free(path);
3357 return res;
3360 /* OS/2 supports spawnvp & spawnvpe natively */
3361 #if defined(PYOS_OS2)
3362 PyDoc_STRVAR(posix_spawnvp__doc__,
3363 "spawnvp(mode, file, args)\n\n\
3364 Execute the program 'file' in a new process, using the environment\n\
3365 search path to find the file.\n\
3367 mode: mode of process creation\n\
3368 file: executable file name\n\
3369 args: tuple or list of strings");
3371 static PyObject *
3372 posix_spawnvp(PyObject *self, PyObject *args)
3374 char *path;
3375 PyObject *argv;
3376 char **argvlist;
3377 int mode, i, argc;
3378 Py_intptr_t spawnval;
3379 PyObject *(*getitem)(PyObject *, Py_ssize_t);
3381 /* spawnvp has three arguments: (mode, path, argv), where
3382 argv is a list or tuple of strings. */
3384 if (!PyArg_ParseTuple(args, "ietO:spawnvp", &mode,
3385 Py_FileSystemDefaultEncoding,
3386 &path, &argv))
3387 return NULL;
3388 if (PyList_Check(argv)) {
3389 argc = PyList_Size(argv);
3390 getitem = PyList_GetItem;
3392 else if (PyTuple_Check(argv)) {
3393 argc = PyTuple_Size(argv);
3394 getitem = PyTuple_GetItem;
3396 else {
3397 PyErr_SetString(PyExc_TypeError,
3398 "spawnvp() arg 2 must be a tuple or list");
3399 PyMem_Free(path);
3400 return NULL;
3403 argvlist = PyMem_NEW(char *, argc+1);
3404 if (argvlist == NULL) {
3405 PyMem_Free(path);
3406 return PyErr_NoMemory();
3408 for (i = 0; i < argc; i++) {
3409 if (!PyArg_Parse((*getitem)(argv, i), "et",
3410 Py_FileSystemDefaultEncoding,
3411 &argvlist[i])) {
3412 free_string_array(argvlist, i);
3413 PyErr_SetString(
3414 PyExc_TypeError,
3415 "spawnvp() arg 2 must contain only strings");
3416 PyMem_Free(path);
3417 return NULL;
3420 argvlist[argc] = NULL;
3422 Py_BEGIN_ALLOW_THREADS
3423 #if defined(PYCC_GCC)
3424 spawnval = spawnvp(mode, path, argvlist);
3425 #else
3426 spawnval = _spawnvp(mode, path, argvlist);
3427 #endif
3428 Py_END_ALLOW_THREADS
3430 free_string_array(argvlist, argc);
3431 PyMem_Free(path);
3433 if (spawnval == -1)
3434 return posix_error();
3435 else
3436 return Py_BuildValue("l", (long) spawnval);
3440 PyDoc_STRVAR(posix_spawnvpe__doc__,
3441 "spawnvpe(mode, file, args, env)\n\n\
3442 Execute the program 'file' in a new process, using the environment\n\
3443 search path to find the file.\n\
3445 mode: mode of process creation\n\
3446 file: executable file name\n\
3447 args: tuple or list of arguments\n\
3448 env: dictionary of strings mapping to strings");
3450 static PyObject *
3451 posix_spawnvpe(PyObject *self, PyObject *args)
3453 char *path;
3454 PyObject *argv, *env;
3455 char **argvlist;
3456 char **envlist;
3457 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
3458 int mode, i, pos, argc, envc;
3459 Py_intptr_t spawnval;
3460 PyObject *(*getitem)(PyObject *, Py_ssize_t);
3461 int lastarg = 0;
3463 /* spawnvpe has four arguments: (mode, path, argv, env), where
3464 argv is a list or tuple of strings and env is a dictionary
3465 like posix.environ. */
3467 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
3468 Py_FileSystemDefaultEncoding,
3469 &path, &argv, &env))
3470 return NULL;
3471 if (PyList_Check(argv)) {
3472 argc = PyList_Size(argv);
3473 getitem = PyList_GetItem;
3475 else if (PyTuple_Check(argv)) {
3476 argc = PyTuple_Size(argv);
3477 getitem = PyTuple_GetItem;
3479 else {
3480 PyErr_SetString(PyExc_TypeError,
3481 "spawnvpe() arg 2 must be a tuple or list");
3482 goto fail_0;
3484 if (!PyMapping_Check(env)) {
3485 PyErr_SetString(PyExc_TypeError,
3486 "spawnvpe() arg 3 must be a mapping object");
3487 goto fail_0;
3490 argvlist = PyMem_NEW(char *, argc+1);
3491 if (argvlist == NULL) {
3492 PyErr_NoMemory();
3493 goto fail_0;
3495 for (i = 0; i < argc; i++) {
3496 if (!PyArg_Parse((*getitem)(argv, i),
3497 "et;spawnvpe() arg 2 must contain only strings",
3498 Py_FileSystemDefaultEncoding,
3499 &argvlist[i]))
3501 lastarg = i;
3502 goto fail_1;
3505 lastarg = argc;
3506 argvlist[argc] = NULL;
3508 i = PyMapping_Size(env);
3509 if (i < 0)
3510 goto fail_1;
3511 envlist = PyMem_NEW(char *, i + 1);
3512 if (envlist == NULL) {
3513 PyErr_NoMemory();
3514 goto fail_1;
3516 envc = 0;
3517 keys = PyMapping_Keys(env);
3518 vals = PyMapping_Values(env);
3519 if (!keys || !vals)
3520 goto fail_2;
3521 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3522 PyErr_SetString(PyExc_TypeError,
3523 "spawnvpe(): env.keys() or env.values() is not a list");
3524 goto fail_2;
3527 for (pos = 0; pos < i; pos++) {
3528 char *p, *k, *v;
3529 size_t len;
3531 key = PyList_GetItem(keys, pos);
3532 val = PyList_GetItem(vals, pos);
3533 if (!key || !val)
3534 goto fail_2;
3536 if (!PyArg_Parse(
3537 key,
3538 "s;spawnvpe() arg 3 contains a non-string key",
3539 &k) ||
3540 !PyArg_Parse(
3541 val,
3542 "s;spawnvpe() arg 3 contains a non-string value",
3543 &v))
3545 goto fail_2;
3547 len = PyString_Size(key) + PyString_Size(val) + 2;
3548 p = PyMem_NEW(char, len);
3549 if (p == NULL) {
3550 PyErr_NoMemory();
3551 goto fail_2;
3553 PyOS_snprintf(p, len, "%s=%s", k, v);
3554 envlist[envc++] = p;
3556 envlist[envc] = 0;
3558 Py_BEGIN_ALLOW_THREADS
3559 #if defined(PYCC_GCC)
3560 spawnval = spawnvpe(mode, path, argvlist, envlist);
3561 #else
3562 spawnval = _spawnvpe(mode, path, argvlist, envlist);
3563 #endif
3564 Py_END_ALLOW_THREADS
3566 if (spawnval == -1)
3567 (void) posix_error();
3568 else
3569 res = Py_BuildValue("l", (long) spawnval);
3571 fail_2:
3572 while (--envc >= 0)
3573 PyMem_DEL(envlist[envc]);
3574 PyMem_DEL(envlist);
3575 fail_1:
3576 free_string_array(argvlist, lastarg);
3577 Py_XDECREF(vals);
3578 Py_XDECREF(keys);
3579 fail_0:
3580 PyMem_Free(path);
3581 return res;
3583 #endif /* PYOS_OS2 */
3584 #endif /* HAVE_SPAWNV */
3587 #ifdef HAVE_FORK1
3588 PyDoc_STRVAR(posix_fork1__doc__,
3589 "fork1() -> pid\n\n\
3590 Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
3592 Return 0 to child process and PID of child to parent process.");
3594 static PyObject *
3595 posix_fork1(PyObject *self, PyObject *noargs)
3597 pid_t pid = fork1();
3598 if (pid == -1)
3599 return posix_error();
3600 PyOS_AfterFork();
3601 return PyInt_FromLong(pid);
3603 #endif
3606 #ifdef HAVE_FORK
3607 PyDoc_STRVAR(posix_fork__doc__,
3608 "fork() -> pid\n\n\
3609 Fork a child process.\n\
3610 Return 0 to child process and PID of child to parent process.");
3612 static PyObject *
3613 posix_fork(PyObject *self, PyObject *noargs)
3615 pid_t pid = fork();
3616 if (pid == -1)
3617 return posix_error();
3618 if (pid == 0)
3619 PyOS_AfterFork();
3620 return PyInt_FromLong(pid);
3622 #endif
3624 /* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
3625 /* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
3626 #if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
3627 #define DEV_PTY_FILE "/dev/ptc"
3628 #define HAVE_DEV_PTMX
3629 #else
3630 #define DEV_PTY_FILE "/dev/ptmx"
3631 #endif
3633 #if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
3634 #ifdef HAVE_PTY_H
3635 #include <pty.h>
3636 #else
3637 #ifdef HAVE_LIBUTIL_H
3638 #include <libutil.h>
3639 #endif /* HAVE_LIBUTIL_H */
3640 #endif /* HAVE_PTY_H */
3641 #ifdef HAVE_STROPTS_H
3642 #include <stropts.h>
3643 #endif
3644 #endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
3646 #if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
3647 PyDoc_STRVAR(posix_openpty__doc__,
3648 "openpty() -> (master_fd, slave_fd)\n\n\
3649 Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
3651 static PyObject *
3652 posix_openpty(PyObject *self, PyObject *noargs)
3654 int master_fd, slave_fd;
3655 #ifndef HAVE_OPENPTY
3656 char * slave_name;
3657 #endif
3658 #if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
3659 PyOS_sighandler_t sig_saved;
3660 #ifdef sun
3661 extern char *ptsname(int fildes);
3662 #endif
3663 #endif
3665 #ifdef HAVE_OPENPTY
3666 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
3667 return posix_error();
3668 #elif defined(HAVE__GETPTY)
3669 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
3670 if (slave_name == NULL)
3671 return posix_error();
3673 slave_fd = open(slave_name, O_RDWR);
3674 if (slave_fd < 0)
3675 return posix_error();
3676 #else
3677 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
3678 if (master_fd < 0)
3679 return posix_error();
3680 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
3681 /* change permission of slave */
3682 if (grantpt(master_fd) < 0) {
3683 PyOS_setsig(SIGCHLD, sig_saved);
3684 return posix_error();
3686 /* unlock slave */
3687 if (unlockpt(master_fd) < 0) {
3688 PyOS_setsig(SIGCHLD, sig_saved);
3689 return posix_error();
3691 PyOS_setsig(SIGCHLD, sig_saved);
3692 slave_name = ptsname(master_fd); /* get name of slave */
3693 if (slave_name == NULL)
3694 return posix_error();
3695 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
3696 if (slave_fd < 0)
3697 return posix_error();
3698 #if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
3699 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
3700 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
3701 #ifndef __hpux
3702 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
3703 #endif /* __hpux */
3704 #endif /* HAVE_CYGWIN */
3705 #endif /* HAVE_OPENPTY */
3707 return Py_BuildValue("(ii)", master_fd, slave_fd);
3710 #endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
3712 #ifdef HAVE_FORKPTY
3713 PyDoc_STRVAR(posix_forkpty__doc__,
3714 "forkpty() -> (pid, master_fd)\n\n\
3715 Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
3716 Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
3717 To both, return fd of newly opened pseudo-terminal.\n");
3719 static PyObject *
3720 posix_forkpty(PyObject *self, PyObject *noargs)
3722 int master_fd = -1;
3723 pid_t pid;
3725 pid = forkpty(&master_fd, NULL, NULL, NULL);
3726 if (pid == -1)
3727 return posix_error();
3728 if (pid == 0)
3729 PyOS_AfterFork();
3730 return Py_BuildValue("(li)", pid, master_fd);
3732 #endif
3734 #ifdef HAVE_GETEGID
3735 PyDoc_STRVAR(posix_getegid__doc__,
3736 "getegid() -> egid\n\n\
3737 Return the current process's effective group id.");
3739 static PyObject *
3740 posix_getegid(PyObject *self, PyObject *noargs)
3742 return PyInt_FromLong((long)getegid());
3744 #endif
3747 #ifdef HAVE_GETEUID
3748 PyDoc_STRVAR(posix_geteuid__doc__,
3749 "geteuid() -> euid\n\n\
3750 Return the current process's effective user id.");
3752 static PyObject *
3753 posix_geteuid(PyObject *self, PyObject *noargs)
3755 return PyInt_FromLong((long)geteuid());
3757 #endif
3760 #ifdef HAVE_GETGID
3761 PyDoc_STRVAR(posix_getgid__doc__,
3762 "getgid() -> gid\n\n\
3763 Return the current process's group id.");
3765 static PyObject *
3766 posix_getgid(PyObject *self, PyObject *noargs)
3768 return PyInt_FromLong((long)getgid());
3770 #endif
3773 PyDoc_STRVAR(posix_getpid__doc__,
3774 "getpid() -> pid\n\n\
3775 Return the current process id");
3777 static PyObject *
3778 posix_getpid(PyObject *self, PyObject *noargs)
3780 return PyInt_FromLong((long)getpid());
3784 #ifdef HAVE_GETGROUPS
3785 PyDoc_STRVAR(posix_getgroups__doc__,
3786 "getgroups() -> list of group IDs\n\n\
3787 Return list of supplemental group IDs for the process.");
3789 static PyObject *
3790 posix_getgroups(PyObject *self, PyObject *noargs)
3792 PyObject *result = NULL;
3794 #ifdef NGROUPS_MAX
3795 #define MAX_GROUPS NGROUPS_MAX
3796 #else
3797 /* defined to be 16 on Solaris7, so this should be a small number */
3798 #define MAX_GROUPS 64
3799 #endif
3800 gid_t grouplist[MAX_GROUPS];
3801 int n;
3803 n = getgroups(MAX_GROUPS, grouplist);
3804 if (n < 0)
3805 posix_error();
3806 else {
3807 result = PyList_New(n);
3808 if (result != NULL) {
3809 int i;
3810 for (i = 0; i < n; ++i) {
3811 PyObject *o = PyInt_FromLong((long)grouplist[i]);
3812 if (o == NULL) {
3813 Py_DECREF(result);
3814 result = NULL;
3815 break;
3817 PyList_SET_ITEM(result, i, o);
3822 return result;
3824 #endif
3826 #ifdef HAVE_GETPGID
3827 PyDoc_STRVAR(posix_getpgid__doc__,
3828 "getpgid(pid) -> pgid\n\n\
3829 Call the system call getpgid().");
3831 static PyObject *
3832 posix_getpgid(PyObject *self, PyObject *args)
3834 int pid, pgid;
3835 if (!PyArg_ParseTuple(args, "i:getpgid", &pid))
3836 return NULL;
3837 pgid = getpgid(pid);
3838 if (pgid < 0)
3839 return posix_error();
3840 return PyInt_FromLong((long)pgid);
3842 #endif /* HAVE_GETPGID */
3845 #ifdef HAVE_GETPGRP
3846 PyDoc_STRVAR(posix_getpgrp__doc__,
3847 "getpgrp() -> pgrp\n\n\
3848 Return the current process group id.");
3850 static PyObject *
3851 posix_getpgrp(PyObject *self, PyObject *noargs)
3853 #ifdef GETPGRP_HAVE_ARG
3854 return PyInt_FromLong((long)getpgrp(0));
3855 #else /* GETPGRP_HAVE_ARG */
3856 return PyInt_FromLong((long)getpgrp());
3857 #endif /* GETPGRP_HAVE_ARG */
3859 #endif /* HAVE_GETPGRP */
3862 #ifdef HAVE_SETPGRP
3863 PyDoc_STRVAR(posix_setpgrp__doc__,
3864 "setpgrp()\n\n\
3865 Make this process a session leader.");
3867 static PyObject *
3868 posix_setpgrp(PyObject *self, PyObject *noargs)
3870 #ifdef SETPGRP_HAVE_ARG
3871 if (setpgrp(0, 0) < 0)
3872 #else /* SETPGRP_HAVE_ARG */
3873 if (setpgrp() < 0)
3874 #endif /* SETPGRP_HAVE_ARG */
3875 return posix_error();
3876 Py_INCREF(Py_None);
3877 return Py_None;
3880 #endif /* HAVE_SETPGRP */
3882 #ifdef HAVE_GETPPID
3883 PyDoc_STRVAR(posix_getppid__doc__,
3884 "getppid() -> ppid\n\n\
3885 Return the parent's process id.");
3887 static PyObject *
3888 posix_getppid(PyObject *self, PyObject *noargs)
3890 return PyInt_FromLong(getppid());
3892 #endif
3895 #ifdef HAVE_GETLOGIN
3896 PyDoc_STRVAR(posix_getlogin__doc__,
3897 "getlogin() -> string\n\n\
3898 Return the actual login name.");
3900 static PyObject *
3901 posix_getlogin(PyObject *self, PyObject *noargs)
3903 PyObject *result = NULL;
3904 char *name;
3905 int old_errno = errno;
3907 errno = 0;
3908 name = getlogin();
3909 if (name == NULL) {
3910 if (errno)
3911 posix_error();
3912 else
3913 PyErr_SetString(PyExc_OSError,
3914 "unable to determine login name");
3916 else
3917 result = PyString_FromString(name);
3918 errno = old_errno;
3920 return result;
3922 #endif
3924 #ifdef HAVE_GETUID
3925 PyDoc_STRVAR(posix_getuid__doc__,
3926 "getuid() -> uid\n\n\
3927 Return the current process's user id.");
3929 static PyObject *
3930 posix_getuid(PyObject *self, PyObject *noargs)
3932 return PyInt_FromLong((long)getuid());
3934 #endif
3937 #ifdef HAVE_KILL
3938 PyDoc_STRVAR(posix_kill__doc__,
3939 "kill(pid, sig)\n\n\
3940 Kill a process with a signal.");
3942 static PyObject *
3943 posix_kill(PyObject *self, PyObject *args)
3945 pid_t pid;
3946 int sig;
3947 if (!PyArg_ParseTuple(args, "ii:kill", &pid, &sig))
3948 return NULL;
3949 #if defined(PYOS_OS2) && !defined(PYCC_GCC)
3950 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
3951 APIRET rc;
3952 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
3953 return os2_error(rc);
3955 } else if (sig == XCPT_SIGNAL_KILLPROC) {
3956 APIRET rc;
3957 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
3958 return os2_error(rc);
3960 } else
3961 return NULL; /* Unrecognized Signal Requested */
3962 #else
3963 if (kill(pid, sig) == -1)
3964 return posix_error();
3965 #endif
3966 Py_INCREF(Py_None);
3967 return Py_None;
3969 #endif
3971 #ifdef HAVE_KILLPG
3972 PyDoc_STRVAR(posix_killpg__doc__,
3973 "killpg(pgid, sig)\n\n\
3974 Kill a process group with a signal.");
3976 static PyObject *
3977 posix_killpg(PyObject *self, PyObject *args)
3979 int pgid, sig;
3980 if (!PyArg_ParseTuple(args, "ii:killpg", &pgid, &sig))
3981 return NULL;
3982 if (killpg(pgid, sig) == -1)
3983 return posix_error();
3984 Py_INCREF(Py_None);
3985 return Py_None;
3987 #endif
3989 #ifdef HAVE_PLOCK
3991 #ifdef HAVE_SYS_LOCK_H
3992 #include <sys/lock.h>
3993 #endif
3995 PyDoc_STRVAR(posix_plock__doc__,
3996 "plock(op)\n\n\
3997 Lock program segments into memory.");
3999 static PyObject *
4000 posix_plock(PyObject *self, PyObject *args)
4002 int op;
4003 if (!PyArg_ParseTuple(args, "i:plock", &op))
4004 return NULL;
4005 if (plock(op) == -1)
4006 return posix_error();
4007 Py_INCREF(Py_None);
4008 return Py_None;
4010 #endif
4013 #ifdef HAVE_POPEN
4014 PyDoc_STRVAR(posix_popen__doc__,
4015 "popen(command [, mode='r' [, bufsize]]) -> pipe\n\n\
4016 Open a pipe to/from a command returning a file object.");
4018 #if defined(PYOS_OS2)
4019 #if defined(PYCC_VACPP)
4020 static int
4021 async_system(const char *command)
4023 char errormsg[256], args[1024];
4024 RESULTCODES rcodes;
4025 APIRET rc;
4027 char *shell = getenv("COMSPEC");
4028 if (!shell)
4029 shell = "cmd";
4031 /* avoid overflowing the argument buffer */
4032 if (strlen(shell) + 3 + strlen(command) >= 1024)
4033 return ERROR_NOT_ENOUGH_MEMORY
4035 args[0] = '\0';
4036 strcat(args, shell);
4037 strcat(args, "/c ");
4038 strcat(args, command);
4040 /* execute asynchronously, inheriting the environment */
4041 rc = DosExecPgm(errormsg,
4042 sizeof(errormsg),
4043 EXEC_ASYNC,
4044 args,
4045 NULL,
4046 &rcodes,
4047 shell);
4048 return rc;
4051 static FILE *
4052 popen(const char *command, const char *mode, int pipesize, int *err)
4054 int oldfd, tgtfd;
4055 HFILE pipeh[2];
4056 APIRET rc;
4058 /* mode determines which of stdin or stdout is reconnected to
4059 * the pipe to the child
4061 if (strchr(mode, 'r') != NULL) {
4062 tgt_fd = 1; /* stdout */
4063 } else if (strchr(mode, 'w')) {
4064 tgt_fd = 0; /* stdin */
4065 } else {
4066 *err = ERROR_INVALID_ACCESS;
4067 return NULL;
4070 /* setup the pipe */
4071 if ((rc = DosCreatePipe(&pipeh[0], &pipeh[1], pipesize)) != NO_ERROR) {
4072 *err = rc;
4073 return NULL;
4076 /* prevent other threads accessing stdio */
4077 DosEnterCritSec();
4079 /* reconnect stdio and execute child */
4080 oldfd = dup(tgtfd);
4081 close(tgtfd);
4082 if (dup2(pipeh[tgtfd], tgtfd) == 0) {
4083 DosClose(pipeh[tgtfd]);
4084 rc = async_system(command);
4087 /* restore stdio */
4088 dup2(oldfd, tgtfd);
4089 close(oldfd);
4091 /* allow other threads access to stdio */
4092 DosExitCritSec();
4094 /* if execution of child was successful return file stream */
4095 if (rc == NO_ERROR)
4096 return fdopen(pipeh[1 - tgtfd], mode);
4097 else {
4098 DosClose(pipeh[1 - tgtfd]);
4099 *err = rc;
4100 return NULL;
4104 static PyObject *
4105 posix_popen(PyObject *self, PyObject *args)
4107 char *name;
4108 char *mode = "r";
4109 int err, bufsize = -1;
4110 FILE *fp;
4111 PyObject *f;
4112 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
4113 return NULL;
4114 Py_BEGIN_ALLOW_THREADS
4115 fp = popen(name, mode, (bufsize > 0) ? bufsize : 4096, &err);
4116 Py_END_ALLOW_THREADS
4117 if (fp == NULL)
4118 return os2_error(err);
4120 f = PyFile_FromFile(fp, name, mode, fclose);
4121 if (f != NULL)
4122 PyFile_SetBufSize(f, bufsize);
4123 return f;
4126 #elif defined(PYCC_GCC)
4128 /* standard posix version of popen() support */
4129 static PyObject *
4130 posix_popen(PyObject *self, PyObject *args)
4132 char *name;
4133 char *mode = "r";
4134 int bufsize = -1;
4135 FILE *fp;
4136 PyObject *f;
4137 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
4138 return NULL;
4139 Py_BEGIN_ALLOW_THREADS
4140 fp = popen(name, mode);
4141 Py_END_ALLOW_THREADS
4142 if (fp == NULL)
4143 return posix_error();
4144 f = PyFile_FromFile(fp, name, mode, pclose);
4145 if (f != NULL)
4146 PyFile_SetBufSize(f, bufsize);
4147 return f;
4150 /* fork() under OS/2 has lots'o'warts
4151 * EMX supports pipe() and spawn*() so we can synthesize popen[234]()
4152 * most of this code is a ripoff of the win32 code, but using the
4153 * capabilities of EMX's C library routines
4156 /* These tell _PyPopen() whether to return 1, 2, or 3 file objects. */
4157 #define POPEN_1 1
4158 #define POPEN_2 2
4159 #define POPEN_3 3
4160 #define POPEN_4 4
4162 static PyObject *_PyPopen(char *, int, int, int);
4163 static int _PyPclose(FILE *file);
4166 * Internal dictionary mapping popen* file pointers to process handles,
4167 * for use when retrieving the process exit code. See _PyPclose() below
4168 * for more information on this dictionary's use.
4170 static PyObject *_PyPopenProcs = NULL;
4172 /* os2emx version of popen2()
4174 * The result of this function is a pipe (file) connected to the
4175 * process's stdin, and a pipe connected to the process's stdout.
4178 static PyObject *
4179 os2emx_popen2(PyObject *self, PyObject *args)
4181 PyObject *f;
4182 int tm=0;
4184 char *cmdstring;
4185 char *mode = "t";
4186 int bufsize = -1;
4187 if (!PyArg_ParseTuple(args, "s|si:popen2", &cmdstring, &mode, &bufsize))
4188 return NULL;
4190 if (*mode == 't')
4191 tm = O_TEXT;
4192 else if (*mode != 'b') {
4193 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
4194 return NULL;
4195 } else
4196 tm = O_BINARY;
4198 f = _PyPopen(cmdstring, tm, POPEN_2, bufsize);
4200 return f;
4204 * Variation on os2emx.popen2
4206 * The result of this function is 3 pipes - the process's stdin,
4207 * stdout and stderr
4210 static PyObject *
4211 os2emx_popen3(PyObject *self, PyObject *args)
4213 PyObject *f;
4214 int tm = 0;
4216 char *cmdstring;
4217 char *mode = "t";
4218 int bufsize = -1;
4219 if (!PyArg_ParseTuple(args, "s|si:popen3", &cmdstring, &mode, &bufsize))
4220 return NULL;
4222 if (*mode == 't')
4223 tm = O_TEXT;
4224 else if (*mode != 'b') {
4225 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
4226 return NULL;
4227 } else
4228 tm = O_BINARY;
4230 f = _PyPopen(cmdstring, tm, POPEN_3, bufsize);
4232 return f;
4236 * Variation on os2emx.popen2
4238 * The result of this function is 2 pipes - the processes stdin,
4239 * and stdout+stderr combined as a single pipe.
4242 static PyObject *
4243 os2emx_popen4(PyObject *self, PyObject *args)
4245 PyObject *f;
4246 int tm = 0;
4248 char *cmdstring;
4249 char *mode = "t";
4250 int bufsize = -1;
4251 if (!PyArg_ParseTuple(args, "s|si:popen4", &cmdstring, &mode, &bufsize))
4252 return NULL;
4254 if (*mode == 't')
4255 tm = O_TEXT;
4256 else if (*mode != 'b') {
4257 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
4258 return NULL;
4259 } else
4260 tm = O_BINARY;
4262 f = _PyPopen(cmdstring, tm, POPEN_4, bufsize);
4264 return f;
4267 /* a couple of structures for convenient handling of multiple
4268 * file handles and pipes
4270 struct file_ref
4272 int handle;
4273 int flags;
4276 struct pipe_ref
4278 int rd;
4279 int wr;
4282 /* The following code is derived from the win32 code */
4284 static PyObject *
4285 _PyPopen(char *cmdstring, int mode, int n, int bufsize)
4287 struct file_ref stdio[3];
4288 struct pipe_ref p_fd[3];
4289 FILE *p_s[3];
4290 int file_count, i, pipe_err;
4291 pid_t pipe_pid;
4292 char *shell, *sh_name, *opt, *rd_mode, *wr_mode;
4293 PyObject *f, *p_f[3];
4295 /* file modes for subsequent fdopen's on pipe handles */
4296 if (mode == O_TEXT)
4298 rd_mode = "rt";
4299 wr_mode = "wt";
4301 else
4303 rd_mode = "rb";
4304 wr_mode = "wb";
4307 /* prepare shell references */
4308 if ((shell = getenv("EMXSHELL")) == NULL)
4309 if ((shell = getenv("COMSPEC")) == NULL)
4311 errno = ENOENT;
4312 return posix_error();
4315 sh_name = _getname(shell);
4316 if (stricmp(sh_name, "cmd.exe") == 0 || stricmp(sh_name, "4os2.exe") == 0)
4317 opt = "/c";
4318 else
4319 opt = "-c";
4321 /* save current stdio fds + their flags, and set not inheritable */
4322 i = pipe_err = 0;
4323 while (pipe_err >= 0 && i < 3)
4325 pipe_err = stdio[i].handle = dup(i);
4326 stdio[i].flags = fcntl(i, F_GETFD, 0);
4327 fcntl(stdio[i].handle, F_SETFD, stdio[i].flags | FD_CLOEXEC);
4328 i++;
4330 if (pipe_err < 0)
4332 /* didn't get them all saved - clean up and bail out */
4333 int saved_err = errno;
4334 while (i-- > 0)
4336 close(stdio[i].handle);
4338 errno = saved_err;
4339 return posix_error();
4342 /* create pipe ends */
4343 file_count = 2;
4344 if (n == POPEN_3)
4345 file_count = 3;
4346 i = pipe_err = 0;
4347 while ((pipe_err == 0) && (i < file_count))
4348 pipe_err = pipe((int *)&p_fd[i++]);
4349 if (pipe_err < 0)
4351 /* didn't get them all made - clean up and bail out */
4352 while (i-- > 0)
4354 close(p_fd[i].wr);
4355 close(p_fd[i].rd);
4357 errno = EPIPE;
4358 return posix_error();
4361 /* change the actual standard IO streams over temporarily,
4362 * making the retained pipe ends non-inheritable
4364 pipe_err = 0;
4366 /* - stdin */
4367 if (dup2(p_fd[0].rd, 0) == 0)
4369 close(p_fd[0].rd);
4370 i = fcntl(p_fd[0].wr, F_GETFD, 0);
4371 fcntl(p_fd[0].wr, F_SETFD, i | FD_CLOEXEC);
4372 if ((p_s[0] = fdopen(p_fd[0].wr, wr_mode)) == NULL)
4374 close(p_fd[0].wr);
4375 pipe_err = -1;
4378 else
4380 pipe_err = -1;
4383 /* - stdout */
4384 if (pipe_err == 0)
4386 if (dup2(p_fd[1].wr, 1) == 1)
4388 close(p_fd[1].wr);
4389 i = fcntl(p_fd[1].rd, F_GETFD, 0);
4390 fcntl(p_fd[1].rd, F_SETFD, i | FD_CLOEXEC);
4391 if ((p_s[1] = fdopen(p_fd[1].rd, rd_mode)) == NULL)
4393 close(p_fd[1].rd);
4394 pipe_err = -1;
4397 else
4399 pipe_err = -1;
4403 /* - stderr, as required */
4404 if (pipe_err == 0)
4405 switch (n)
4407 case POPEN_3:
4409 if (dup2(p_fd[2].wr, 2) == 2)
4411 close(p_fd[2].wr);
4412 i = fcntl(p_fd[2].rd, F_GETFD, 0);
4413 fcntl(p_fd[2].rd, F_SETFD, i | FD_CLOEXEC);
4414 if ((p_s[2] = fdopen(p_fd[2].rd, rd_mode)) == NULL)
4416 close(p_fd[2].rd);
4417 pipe_err = -1;
4420 else
4422 pipe_err = -1;
4424 break;
4427 case POPEN_4:
4429 if (dup2(1, 2) != 2)
4431 pipe_err = -1;
4433 break;
4437 /* spawn the child process */
4438 if (pipe_err == 0)
4440 pipe_pid = spawnlp(P_NOWAIT, shell, shell, opt, cmdstring, (char *)0);
4441 if (pipe_pid == -1)
4443 pipe_err = -1;
4445 else
4447 /* save the PID into the FILE structure
4448 * NOTE: this implementation doesn't actually
4449 * take advantage of this, but do it for
4450 * completeness - AIM Apr01
4452 for (i = 0; i < file_count; i++)
4453 p_s[i]->_pid = pipe_pid;
4457 /* reset standard IO to normal */
4458 for (i = 0; i < 3; i++)
4460 dup2(stdio[i].handle, i);
4461 fcntl(i, F_SETFD, stdio[i].flags);
4462 close(stdio[i].handle);
4465 /* if any remnant problems, clean up and bail out */
4466 if (pipe_err < 0)
4468 for (i = 0; i < 3; i++)
4470 close(p_fd[i].rd);
4471 close(p_fd[i].wr);
4473 errno = EPIPE;
4474 return posix_error_with_filename(cmdstring);
4477 /* build tuple of file objects to return */
4478 if ((p_f[0] = PyFile_FromFile(p_s[0], cmdstring, wr_mode, _PyPclose)) != NULL)
4479 PyFile_SetBufSize(p_f[0], bufsize);
4480 if ((p_f[1] = PyFile_FromFile(p_s[1], cmdstring, rd_mode, _PyPclose)) != NULL)
4481 PyFile_SetBufSize(p_f[1], bufsize);
4482 if (n == POPEN_3)
4484 if ((p_f[2] = PyFile_FromFile(p_s[2], cmdstring, rd_mode, _PyPclose)) != NULL)
4485 PyFile_SetBufSize(p_f[0], bufsize);
4486 f = PyTuple_Pack(3, p_f[0], p_f[1], p_f[2]);
4488 else
4489 f = PyTuple_Pack(2, p_f[0], p_f[1]);
4492 * Insert the files we've created into the process dictionary
4493 * all referencing the list with the process handle and the
4494 * initial number of files (see description below in _PyPclose).
4495 * Since if _PyPclose later tried to wait on a process when all
4496 * handles weren't closed, it could create a deadlock with the
4497 * child, we spend some energy here to try to ensure that we
4498 * either insert all file handles into the dictionary or none
4499 * at all. It's a little clumsy with the various popen modes
4500 * and variable number of files involved.
4502 if (!_PyPopenProcs)
4504 _PyPopenProcs = PyDict_New();
4507 if (_PyPopenProcs)
4509 PyObject *procObj, *pidObj, *intObj, *fileObj[3];
4510 int ins_rc[3];
4512 fileObj[0] = fileObj[1] = fileObj[2] = NULL;
4513 ins_rc[0] = ins_rc[1] = ins_rc[2] = 0;
4515 procObj = PyList_New(2);
4516 pidObj = PyInt_FromLong((long) pipe_pid);
4517 intObj = PyInt_FromLong((long) file_count);
4519 if (procObj && pidObj && intObj)
4521 PyList_SetItem(procObj, 0, pidObj);
4522 PyList_SetItem(procObj, 1, intObj);
4524 fileObj[0] = PyLong_FromVoidPtr(p_s[0]);
4525 if (fileObj[0])
4527 ins_rc[0] = PyDict_SetItem(_PyPopenProcs,
4528 fileObj[0],
4529 procObj);
4531 fileObj[1] = PyLong_FromVoidPtr(p_s[1]);
4532 if (fileObj[1])
4534 ins_rc[1] = PyDict_SetItem(_PyPopenProcs,
4535 fileObj[1],
4536 procObj);
4538 if (file_count >= 3)
4540 fileObj[2] = PyLong_FromVoidPtr(p_s[2]);
4541 if (fileObj[2])
4543 ins_rc[2] = PyDict_SetItem(_PyPopenProcs,
4544 fileObj[2],
4545 procObj);
4549 if (ins_rc[0] < 0 || !fileObj[0] ||
4550 ins_rc[1] < 0 || (file_count > 1 && !fileObj[1]) ||
4551 ins_rc[2] < 0 || (file_count > 2 && !fileObj[2]))
4553 /* Something failed - remove any dictionary
4554 * entries that did make it.
4556 if (!ins_rc[0] && fileObj[0])
4558 PyDict_DelItem(_PyPopenProcs,
4559 fileObj[0]);
4561 if (!ins_rc[1] && fileObj[1])
4563 PyDict_DelItem(_PyPopenProcs,
4564 fileObj[1]);
4566 if (!ins_rc[2] && fileObj[2])
4568 PyDict_DelItem(_PyPopenProcs,
4569 fileObj[2]);
4575 * Clean up our localized references for the dictionary keys
4576 * and value since PyDict_SetItem will Py_INCREF any copies
4577 * that got placed in the dictionary.
4579 Py_XDECREF(procObj);
4580 Py_XDECREF(fileObj[0]);
4581 Py_XDECREF(fileObj[1]);
4582 Py_XDECREF(fileObj[2]);
4585 /* Child is launched. */
4586 return f;
4590 * Wrapper for fclose() to use for popen* files, so we can retrieve the
4591 * exit code for the child process and return as a result of the close.
4593 * This function uses the _PyPopenProcs dictionary in order to map the
4594 * input file pointer to information about the process that was
4595 * originally created by the popen* call that created the file pointer.
4596 * The dictionary uses the file pointer as a key (with one entry
4597 * inserted for each file returned by the original popen* call) and a
4598 * single list object as the value for all files from a single call.
4599 * The list object contains the Win32 process handle at [0], and a file
4600 * count at [1], which is initialized to the total number of file
4601 * handles using that list.
4603 * This function closes whichever handle it is passed, and decrements
4604 * the file count in the dictionary for the process handle pointed to
4605 * by this file. On the last close (when the file count reaches zero),
4606 * this function will wait for the child process and then return its
4607 * exit code as the result of the close() operation. This permits the
4608 * files to be closed in any order - it is always the close() of the
4609 * final handle that will return the exit code.
4611 * NOTE: This function is currently called with the GIL released.
4612 * hence we use the GILState API to manage our state.
4615 static int _PyPclose(FILE *file)
4617 int result;
4618 int exit_code;
4619 pid_t pipe_pid;
4620 PyObject *procObj, *pidObj, *intObj, *fileObj;
4621 int file_count;
4622 #ifdef WITH_THREAD
4623 PyGILState_STATE state;
4624 #endif
4626 /* Close the file handle first, to ensure it can't block the
4627 * child from exiting if it's the last handle.
4629 result = fclose(file);
4631 #ifdef WITH_THREAD
4632 state = PyGILState_Ensure();
4633 #endif
4634 if (_PyPopenProcs)
4636 if ((fileObj = PyLong_FromVoidPtr(file)) != NULL &&
4637 (procObj = PyDict_GetItem(_PyPopenProcs,
4638 fileObj)) != NULL &&
4639 (pidObj = PyList_GetItem(procObj,0)) != NULL &&
4640 (intObj = PyList_GetItem(procObj,1)) != NULL)
4642 pipe_pid = (int) PyInt_AsLong(pidObj);
4643 file_count = (int) PyInt_AsLong(intObj);
4645 if (file_count > 1)
4647 /* Still other files referencing process */
4648 file_count--;
4649 PyList_SetItem(procObj,1,
4650 PyInt_FromLong((long) file_count));
4652 else
4654 /* Last file for this process */
4655 if (result != EOF &&
4656 waitpid(pipe_pid, &exit_code, 0) == pipe_pid)
4658 /* extract exit status */
4659 if (WIFEXITED(exit_code))
4661 result = WEXITSTATUS(exit_code);
4663 else
4665 errno = EPIPE;
4666 result = -1;
4669 else
4671 /* Indicate failure - this will cause the file object
4672 * to raise an I/O error and translate the last
4673 * error code from errno. We do have a problem with
4674 * last errors that overlap the normal errno table,
4675 * but that's a consistent problem with the file object.
4677 result = -1;
4681 /* Remove this file pointer from dictionary */
4682 PyDict_DelItem(_PyPopenProcs, fileObj);
4684 if (PyDict_Size(_PyPopenProcs) == 0)
4686 Py_DECREF(_PyPopenProcs);
4687 _PyPopenProcs = NULL;
4690 } /* if object retrieval ok */
4692 Py_XDECREF(fileObj);
4693 } /* if _PyPopenProcs */
4695 #ifdef WITH_THREAD
4696 PyGILState_Release(state);
4697 #endif
4698 return result;
4701 #endif /* PYCC_??? */
4703 #elif defined(MS_WINDOWS)
4706 * Portable 'popen' replacement for Win32.
4708 * Written by Bill Tutt <billtut@microsoft.com>. Minor tweaks
4709 * and 2.0 integration by Fredrik Lundh <fredrik@pythonware.com>
4710 * Return code handling by David Bolen <db3l@fitlinxx.com>.
4713 #include <malloc.h>
4714 #include <io.h>
4715 #include <fcntl.h>
4717 /* These tell _PyPopen() wether to return 1, 2, or 3 file objects. */
4718 #define POPEN_1 1
4719 #define POPEN_2 2
4720 #define POPEN_3 3
4721 #define POPEN_4 4
4723 static PyObject *_PyPopen(char *, int, int);
4724 static int _PyPclose(FILE *file);
4727 * Internal dictionary mapping popen* file pointers to process handles,
4728 * for use when retrieving the process exit code. See _PyPclose() below
4729 * for more information on this dictionary's use.
4731 static PyObject *_PyPopenProcs = NULL;
4734 /* popen that works from a GUI.
4736 * The result of this function is a pipe (file) connected to the
4737 * processes stdin or stdout, depending on the requested mode.
4740 static PyObject *
4741 posix_popen(PyObject *self, PyObject *args)
4743 PyObject *f;
4744 int tm = 0;
4746 char *cmdstring;
4747 char *mode = "r";
4748 int bufsize = -1;
4749 if (!PyArg_ParseTuple(args, "s|si:popen", &cmdstring, &mode, &bufsize))
4750 return NULL;
4752 if (*mode == 'r')
4753 tm = _O_RDONLY;
4754 else if (*mode != 'w') {
4755 PyErr_SetString(PyExc_ValueError, "popen() arg 2 must be 'r' or 'w'");
4756 return NULL;
4757 } else
4758 tm = _O_WRONLY;
4760 if (bufsize != -1) {
4761 PyErr_SetString(PyExc_ValueError, "popen() arg 3 must be -1");
4762 return NULL;
4765 if (*(mode+1) == 't')
4766 f = _PyPopen(cmdstring, tm | _O_TEXT, POPEN_1);
4767 else if (*(mode+1) == 'b')
4768 f = _PyPopen(cmdstring, tm | _O_BINARY, POPEN_1);
4769 else
4770 f = _PyPopen(cmdstring, tm | _O_TEXT, POPEN_1);
4772 return f;
4775 /* Variation on win32pipe.popen
4777 * The result of this function is a pipe (file) connected to the
4778 * process's stdin, and a pipe connected to the process's stdout.
4781 static PyObject *
4782 win32_popen2(PyObject *self, PyObject *args)
4784 PyObject *f;
4785 int tm=0;
4787 char *cmdstring;
4788 char *mode = "t";
4789 int bufsize = -1;
4790 if (!PyArg_ParseTuple(args, "s|si:popen2", &cmdstring, &mode, &bufsize))
4791 return NULL;
4793 if (*mode == 't')
4794 tm = _O_TEXT;
4795 else if (*mode != 'b') {
4796 PyErr_SetString(PyExc_ValueError, "popen2() arg 2 must be 't' or 'b'");
4797 return NULL;
4798 } else
4799 tm = _O_BINARY;
4801 if (bufsize != -1) {
4802 PyErr_SetString(PyExc_ValueError, "popen2() arg 3 must be -1");
4803 return NULL;
4806 f = _PyPopen(cmdstring, tm, POPEN_2);
4808 return f;
4812 * Variation on <om win32pipe.popen>
4814 * The result of this function is 3 pipes - the process's stdin,
4815 * stdout and stderr
4818 static PyObject *
4819 win32_popen3(PyObject *self, PyObject *args)
4821 PyObject *f;
4822 int tm = 0;
4824 char *cmdstring;
4825 char *mode = "t";
4826 int bufsize = -1;
4827 if (!PyArg_ParseTuple(args, "s|si:popen3", &cmdstring, &mode, &bufsize))
4828 return NULL;
4830 if (*mode == 't')
4831 tm = _O_TEXT;
4832 else if (*mode != 'b') {
4833 PyErr_SetString(PyExc_ValueError, "popen3() arg 2 must be 't' or 'b'");
4834 return NULL;
4835 } else
4836 tm = _O_BINARY;
4838 if (bufsize != -1) {
4839 PyErr_SetString(PyExc_ValueError, "popen3() arg 3 must be -1");
4840 return NULL;
4843 f = _PyPopen(cmdstring, tm, POPEN_3);
4845 return f;
4849 * Variation on win32pipe.popen
4851 * The result of this function is 2 pipes - the processes stdin,
4852 * and stdout+stderr combined as a single pipe.
4855 static PyObject *
4856 win32_popen4(PyObject *self, PyObject *args)
4858 PyObject *f;
4859 int tm = 0;
4861 char *cmdstring;
4862 char *mode = "t";
4863 int bufsize = -1;
4864 if (!PyArg_ParseTuple(args, "s|si:popen4", &cmdstring, &mode, &bufsize))
4865 return NULL;
4867 if (*mode == 't')
4868 tm = _O_TEXT;
4869 else if (*mode != 'b') {
4870 PyErr_SetString(PyExc_ValueError, "popen4() arg 2 must be 't' or 'b'");
4871 return NULL;
4872 } else
4873 tm = _O_BINARY;
4875 if (bufsize != -1) {
4876 PyErr_SetString(PyExc_ValueError, "popen4() arg 3 must be -1");
4877 return NULL;
4880 f = _PyPopen(cmdstring, tm, POPEN_4);
4882 return f;
4885 static BOOL
4886 _PyPopenCreateProcess(char *cmdstring,
4887 HANDLE hStdin,
4888 HANDLE hStdout,
4889 HANDLE hStderr,
4890 HANDLE *hProcess)
4892 PROCESS_INFORMATION piProcInfo;
4893 STARTUPINFO siStartInfo;
4894 DWORD dwProcessFlags = 0; /* no NEW_CONSOLE by default for Ctrl+C handling */
4895 char *s1,*s2, *s3 = " /c ";
4896 const char *szConsoleSpawn = "w9xpopen.exe";
4897 int i;
4898 Py_ssize_t x;
4900 if (i = GetEnvironmentVariable("COMSPEC",NULL,0)) {
4901 char *comshell;
4903 s1 = (char *)alloca(i);
4904 if (!(x = GetEnvironmentVariable("COMSPEC", s1, i)))
4905 /* x < i, so x fits into an integer */
4906 return (int)x;
4908 /* Explicitly check if we are using COMMAND.COM. If we are
4909 * then use the w9xpopen hack.
4911 comshell = s1 + x;
4912 while (comshell >= s1 && *comshell != '\\')
4913 --comshell;
4914 ++comshell;
4916 if (GetVersion() < 0x80000000 &&
4917 _stricmp(comshell, "command.com") != 0) {
4918 /* NT/2000 and not using command.com. */
4919 x = i + strlen(s3) + strlen(cmdstring) + 1;
4920 s2 = (char *)alloca(x);
4921 ZeroMemory(s2, x);
4922 PyOS_snprintf(s2, x, "%s%s%s", s1, s3, cmdstring);
4924 else {
4926 * Oh gag, we're on Win9x or using COMMAND.COM. Use
4927 * the workaround listed in KB: Q150956
4929 char modulepath[_MAX_PATH];
4930 struct stat statinfo;
4931 GetModuleFileName(NULL, modulepath, sizeof(modulepath));
4932 for (x = i = 0; modulepath[i]; i++)
4933 if (modulepath[i] == SEP)
4934 x = i+1;
4935 modulepath[x] = '\0';
4936 /* Create the full-name to w9xpopen, so we can test it exists */
4937 strncat(modulepath,
4938 szConsoleSpawn,
4939 (sizeof(modulepath)/sizeof(modulepath[0]))
4940 -strlen(modulepath));
4941 if (stat(modulepath, &statinfo) != 0) {
4942 size_t mplen = sizeof(modulepath)/sizeof(modulepath[0]);
4943 /* Eeek - file-not-found - possibly an embedding
4944 situation - see if we can locate it in sys.prefix
4946 strncpy(modulepath,
4947 Py_GetExecPrefix(),
4948 mplen);
4949 modulepath[mplen-1] = '\0';
4950 if (modulepath[strlen(modulepath)-1] != '\\')
4951 strcat(modulepath, "\\");
4952 strncat(modulepath,
4953 szConsoleSpawn,
4954 mplen-strlen(modulepath));
4955 /* No where else to look - raise an easily identifiable
4956 error, rather than leaving Windows to report
4957 "file not found" - as the user is probably blissfully
4958 unaware this shim EXE is used, and it will confuse them.
4959 (well, it confused me for a while ;-)
4961 if (stat(modulepath, &statinfo) != 0) {
4962 PyErr_Format(PyExc_RuntimeError,
4963 "Can not locate '%s' which is needed "
4964 "for popen to work with your shell "
4965 "or platform.",
4966 szConsoleSpawn);
4967 return FALSE;
4970 x = i + strlen(s3) + strlen(cmdstring) + 1 +
4971 strlen(modulepath) +
4972 strlen(szConsoleSpawn) + 1;
4974 s2 = (char *)alloca(x);
4975 ZeroMemory(s2, x);
4976 /* To maintain correct argument passing semantics,
4977 we pass the command-line as it stands, and allow
4978 quoting to be applied. w9xpopen.exe will then
4979 use its argv vector, and re-quote the necessary
4980 args for the ultimate child process.
4982 PyOS_snprintf(
4983 s2, x,
4984 "\"%s\" %s%s%s",
4985 modulepath,
4988 cmdstring);
4989 /* Not passing CREATE_NEW_CONSOLE has been known to
4990 cause random failures on win9x. Specifically a
4991 dialog:
4992 "Your program accessed mem currently in use at xxx"
4993 and a hopeful warning about the stability of your
4994 system.
4995 Cost is Ctrl+C wont kill children, but anyone
4996 who cares can have a go!
4998 dwProcessFlags |= CREATE_NEW_CONSOLE;
5002 /* Could be an else here to try cmd.exe / command.com in the path
5003 Now we'll just error out.. */
5004 else {
5005 PyErr_SetString(PyExc_RuntimeError,
5006 "Cannot locate a COMSPEC environment variable to "
5007 "use as the shell");
5008 return FALSE;
5011 ZeroMemory(&siStartInfo, sizeof(STARTUPINFO));
5012 siStartInfo.cb = sizeof(STARTUPINFO);
5013 siStartInfo.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
5014 siStartInfo.hStdInput = hStdin;
5015 siStartInfo.hStdOutput = hStdout;
5016 siStartInfo.hStdError = hStderr;
5017 siStartInfo.wShowWindow = SW_HIDE;
5019 if (CreateProcess(NULL,
5021 NULL,
5022 NULL,
5023 TRUE,
5024 dwProcessFlags,
5025 NULL,
5026 NULL,
5027 &siStartInfo,
5028 &piProcInfo) ) {
5029 /* Close the handles now so anyone waiting is woken. */
5030 CloseHandle(piProcInfo.hThread);
5032 /* Return process handle */
5033 *hProcess = piProcInfo.hProcess;
5034 return TRUE;
5036 win32_error("CreateProcess", s2);
5037 return FALSE;
5040 /* The following code is based off of KB: Q190351 */
5042 static PyObject *
5043 _PyPopen(char *cmdstring, int mode, int n)
5045 HANDLE hChildStdinRd, hChildStdinWr, hChildStdoutRd, hChildStdoutWr,
5046 hChildStderrRd, hChildStderrWr, hChildStdinWrDup, hChildStdoutRdDup,
5047 hChildStderrRdDup, hProcess; /* hChildStdoutWrDup; */
5049 SECURITY_ATTRIBUTES saAttr;
5050 BOOL fSuccess;
5051 int fd1, fd2, fd3;
5052 FILE *f1, *f2, *f3;
5053 long file_count;
5054 PyObject *f;
5056 saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
5057 saAttr.bInheritHandle = TRUE;
5058 saAttr.lpSecurityDescriptor = NULL;
5060 if (!CreatePipe(&hChildStdinRd, &hChildStdinWr, &saAttr, 0))
5061 return win32_error("CreatePipe", NULL);
5063 /* Create new output read handle and the input write handle. Set
5064 * the inheritance properties to FALSE. Otherwise, the child inherits
5065 * these handles; resulting in non-closeable handles to the pipes
5066 * being created. */
5067 fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdinWr,
5068 GetCurrentProcess(), &hChildStdinWrDup, 0,
5069 FALSE,
5070 DUPLICATE_SAME_ACCESS);
5071 if (!fSuccess)
5072 return win32_error("DuplicateHandle", NULL);
5074 /* Close the inheritable version of ChildStdin
5075 that we're using. */
5076 CloseHandle(hChildStdinWr);
5078 if (!CreatePipe(&hChildStdoutRd, &hChildStdoutWr, &saAttr, 0))
5079 return win32_error("CreatePipe", NULL);
5081 fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdoutRd,
5082 GetCurrentProcess(), &hChildStdoutRdDup, 0,
5083 FALSE, DUPLICATE_SAME_ACCESS);
5084 if (!fSuccess)
5085 return win32_error("DuplicateHandle", NULL);
5087 /* Close the inheritable version of ChildStdout
5088 that we're using. */
5089 CloseHandle(hChildStdoutRd);
5091 if (n != POPEN_4) {
5092 if (!CreatePipe(&hChildStderrRd, &hChildStderrWr, &saAttr, 0))
5093 return win32_error("CreatePipe", NULL);
5094 fSuccess = DuplicateHandle(GetCurrentProcess(),
5095 hChildStderrRd,
5096 GetCurrentProcess(),
5097 &hChildStderrRdDup, 0,
5098 FALSE, DUPLICATE_SAME_ACCESS);
5099 if (!fSuccess)
5100 return win32_error("DuplicateHandle", NULL);
5101 /* Close the inheritable version of ChildStdErr that we're using. */
5102 CloseHandle(hChildStderrRd);
5105 switch (n) {
5106 case POPEN_1:
5107 switch (mode & (_O_RDONLY | _O_TEXT | _O_BINARY | _O_WRONLY)) {
5108 case _O_WRONLY | _O_TEXT:
5109 /* Case for writing to child Stdin in text mode. */
5110 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
5111 f1 = _fdopen(fd1, "w");
5112 f = PyFile_FromFile(f1, cmdstring, "w", _PyPclose);
5113 PyFile_SetBufSize(f, 0);
5114 /* We don't care about these pipes anymore, so close them. */
5115 CloseHandle(hChildStdoutRdDup);
5116 CloseHandle(hChildStderrRdDup);
5117 break;
5119 case _O_RDONLY | _O_TEXT:
5120 /* Case for reading from child Stdout in text mode. */
5121 fd1 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
5122 f1 = _fdopen(fd1, "r");
5123 f = PyFile_FromFile(f1, cmdstring, "r", _PyPclose);
5124 PyFile_SetBufSize(f, 0);
5125 /* We don't care about these pipes anymore, so close them. */
5126 CloseHandle(hChildStdinWrDup);
5127 CloseHandle(hChildStderrRdDup);
5128 break;
5130 case _O_RDONLY | _O_BINARY:
5131 /* Case for readinig from child Stdout in binary mode. */
5132 fd1 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
5133 f1 = _fdopen(fd1, "rb");
5134 f = PyFile_FromFile(f1, cmdstring, "rb", _PyPclose);
5135 PyFile_SetBufSize(f, 0);
5136 /* We don't care about these pipes anymore, so close them. */
5137 CloseHandle(hChildStdinWrDup);
5138 CloseHandle(hChildStderrRdDup);
5139 break;
5141 case _O_WRONLY | _O_BINARY:
5142 /* Case for writing to child Stdin in binary mode. */
5143 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
5144 f1 = _fdopen(fd1, "wb");
5145 f = PyFile_FromFile(f1, cmdstring, "wb", _PyPclose);
5146 PyFile_SetBufSize(f, 0);
5147 /* We don't care about these pipes anymore, so close them. */
5148 CloseHandle(hChildStdoutRdDup);
5149 CloseHandle(hChildStderrRdDup);
5150 break;
5152 file_count = 1;
5153 break;
5155 case POPEN_2:
5156 case POPEN_4:
5158 char *m1, *m2;
5159 PyObject *p1, *p2;
5161 if (mode & _O_TEXT) {
5162 m1 = "r";
5163 m2 = "w";
5164 } else {
5165 m1 = "rb";
5166 m2 = "wb";
5169 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
5170 f1 = _fdopen(fd1, m2);
5171 fd2 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
5172 f2 = _fdopen(fd2, m1);
5173 p1 = PyFile_FromFile(f1, cmdstring, m2, _PyPclose);
5174 PyFile_SetBufSize(p1, 0);
5175 p2 = PyFile_FromFile(f2, cmdstring, m1, _PyPclose);
5176 PyFile_SetBufSize(p2, 0);
5178 if (n != 4)
5179 CloseHandle(hChildStderrRdDup);
5181 f = PyTuple_Pack(2,p1,p2);
5182 Py_XDECREF(p1);
5183 Py_XDECREF(p2);
5184 file_count = 2;
5185 break;
5188 case POPEN_3:
5190 char *m1, *m2;
5191 PyObject *p1, *p2, *p3;
5193 if (mode & _O_TEXT) {
5194 m1 = "r";
5195 m2 = "w";
5196 } else {
5197 m1 = "rb";
5198 m2 = "wb";
5201 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
5202 f1 = _fdopen(fd1, m2);
5203 fd2 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
5204 f2 = _fdopen(fd2, m1);
5205 fd3 = _open_osfhandle((Py_intptr_t)hChildStderrRdDup, mode);
5206 f3 = _fdopen(fd3, m1);
5207 p1 = PyFile_FromFile(f1, cmdstring, m2, _PyPclose);
5208 p2 = PyFile_FromFile(f2, cmdstring, m1, _PyPclose);
5209 p3 = PyFile_FromFile(f3, cmdstring, m1, _PyPclose);
5210 PyFile_SetBufSize(p1, 0);
5211 PyFile_SetBufSize(p2, 0);
5212 PyFile_SetBufSize(p3, 0);
5213 f = PyTuple_Pack(3,p1,p2,p3);
5214 Py_XDECREF(p1);
5215 Py_XDECREF(p2);
5216 Py_XDECREF(p3);
5217 file_count = 3;
5218 break;
5222 if (n == POPEN_4) {
5223 if (!_PyPopenCreateProcess(cmdstring,
5224 hChildStdinRd,
5225 hChildStdoutWr,
5226 hChildStdoutWr,
5227 &hProcess))
5228 return NULL;
5230 else {
5231 if (!_PyPopenCreateProcess(cmdstring,
5232 hChildStdinRd,
5233 hChildStdoutWr,
5234 hChildStderrWr,
5235 &hProcess))
5236 return NULL;
5240 * Insert the files we've created into the process dictionary
5241 * all referencing the list with the process handle and the
5242 * initial number of files (see description below in _PyPclose).
5243 * Since if _PyPclose later tried to wait on a process when all
5244 * handles weren't closed, it could create a deadlock with the
5245 * child, we spend some energy here to try to ensure that we
5246 * either insert all file handles into the dictionary or none
5247 * at all. It's a little clumsy with the various popen modes
5248 * and variable number of files involved.
5250 if (!_PyPopenProcs) {
5251 _PyPopenProcs = PyDict_New();
5254 if (_PyPopenProcs) {
5255 PyObject *procObj, *hProcessObj, *intObj, *fileObj[3];
5256 int ins_rc[3];
5258 fileObj[0] = fileObj[1] = fileObj[2] = NULL;
5259 ins_rc[0] = ins_rc[1] = ins_rc[2] = 0;
5261 procObj = PyList_New(2);
5262 hProcessObj = PyLong_FromVoidPtr(hProcess);
5263 intObj = PyInt_FromLong(file_count);
5265 if (procObj && hProcessObj && intObj) {
5266 PyList_SetItem(procObj,0,hProcessObj);
5267 PyList_SetItem(procObj,1,intObj);
5269 fileObj[0] = PyLong_FromVoidPtr(f1);
5270 if (fileObj[0]) {
5271 ins_rc[0] = PyDict_SetItem(_PyPopenProcs,
5272 fileObj[0],
5273 procObj);
5275 if (file_count >= 2) {
5276 fileObj[1] = PyLong_FromVoidPtr(f2);
5277 if (fileObj[1]) {
5278 ins_rc[1] = PyDict_SetItem(_PyPopenProcs,
5279 fileObj[1],
5280 procObj);
5283 if (file_count >= 3) {
5284 fileObj[2] = PyLong_FromVoidPtr(f3);
5285 if (fileObj[2]) {
5286 ins_rc[2] = PyDict_SetItem(_PyPopenProcs,
5287 fileObj[2],
5288 procObj);
5292 if (ins_rc[0] < 0 || !fileObj[0] ||
5293 ins_rc[1] < 0 || (file_count > 1 && !fileObj[1]) ||
5294 ins_rc[2] < 0 || (file_count > 2 && !fileObj[2])) {
5295 /* Something failed - remove any dictionary
5296 * entries that did make it.
5298 if (!ins_rc[0] && fileObj[0]) {
5299 PyDict_DelItem(_PyPopenProcs,
5300 fileObj[0]);
5302 if (!ins_rc[1] && fileObj[1]) {
5303 PyDict_DelItem(_PyPopenProcs,
5304 fileObj[1]);
5306 if (!ins_rc[2] && fileObj[2]) {
5307 PyDict_DelItem(_PyPopenProcs,
5308 fileObj[2]);
5314 * Clean up our localized references for the dictionary keys
5315 * and value since PyDict_SetItem will Py_INCREF any copies
5316 * that got placed in the dictionary.
5318 Py_XDECREF(procObj);
5319 Py_XDECREF(fileObj[0]);
5320 Py_XDECREF(fileObj[1]);
5321 Py_XDECREF(fileObj[2]);
5324 /* Child is launched. Close the parents copy of those pipe
5325 * handles that only the child should have open. You need to
5326 * make sure that no handles to the write end of the output pipe
5327 * are maintained in this process or else the pipe will not close
5328 * when the child process exits and the ReadFile will hang. */
5330 if (!CloseHandle(hChildStdinRd))
5331 return win32_error("CloseHandle", NULL);
5333 if (!CloseHandle(hChildStdoutWr))
5334 return win32_error("CloseHandle", NULL);
5336 if ((n != 4) && (!CloseHandle(hChildStderrWr)))
5337 return win32_error("CloseHandle", NULL);
5339 return f;
5343 * Wrapper for fclose() to use for popen* files, so we can retrieve the
5344 * exit code for the child process and return as a result of the close.
5346 * This function uses the _PyPopenProcs dictionary in order to map the
5347 * input file pointer to information about the process that was
5348 * originally created by the popen* call that created the file pointer.
5349 * The dictionary uses the file pointer as a key (with one entry
5350 * inserted for each file returned by the original popen* call) and a
5351 * single list object as the value for all files from a single call.
5352 * The list object contains the Win32 process handle at [0], and a file
5353 * count at [1], which is initialized to the total number of file
5354 * handles using that list.
5356 * This function closes whichever handle it is passed, and decrements
5357 * the file count in the dictionary for the process handle pointed to
5358 * by this file. On the last close (when the file count reaches zero),
5359 * this function will wait for the child process and then return its
5360 * exit code as the result of the close() operation. This permits the
5361 * files to be closed in any order - it is always the close() of the
5362 * final handle that will return the exit code.
5364 * NOTE: This function is currently called with the GIL released.
5365 * hence we use the GILState API to manage our state.
5368 static int _PyPclose(FILE *file)
5370 int result;
5371 DWORD exit_code;
5372 HANDLE hProcess;
5373 PyObject *procObj, *hProcessObj, *intObj, *fileObj;
5374 long file_count;
5375 #ifdef WITH_THREAD
5376 PyGILState_STATE state;
5377 #endif
5379 /* Close the file handle first, to ensure it can't block the
5380 * child from exiting if it's the last handle.
5382 result = fclose(file);
5383 #ifdef WITH_THREAD
5384 state = PyGILState_Ensure();
5385 #endif
5386 if (_PyPopenProcs) {
5387 if ((fileObj = PyLong_FromVoidPtr(file)) != NULL &&
5388 (procObj = PyDict_GetItem(_PyPopenProcs,
5389 fileObj)) != NULL &&
5390 (hProcessObj = PyList_GetItem(procObj,0)) != NULL &&
5391 (intObj = PyList_GetItem(procObj,1)) != NULL) {
5393 hProcess = PyLong_AsVoidPtr(hProcessObj);
5394 file_count = PyInt_AsLong(intObj);
5396 if (file_count > 1) {
5397 /* Still other files referencing process */
5398 file_count--;
5399 PyList_SetItem(procObj,1,
5400 PyInt_FromLong(file_count));
5401 } else {
5402 /* Last file for this process */
5403 if (result != EOF &&
5404 WaitForSingleObject(hProcess, INFINITE) != WAIT_FAILED &&
5405 GetExitCodeProcess(hProcess, &exit_code)) {
5406 /* Possible truncation here in 16-bit environments, but
5407 * real exit codes are just the lower byte in any event.
5409 result = exit_code;
5410 } else {
5411 /* Indicate failure - this will cause the file object
5412 * to raise an I/O error and translate the last Win32
5413 * error code from errno. We do have a problem with
5414 * last errors that overlap the normal errno table,
5415 * but that's a consistent problem with the file object.
5417 if (result != EOF) {
5418 /* If the error wasn't from the fclose(), then
5419 * set errno for the file object error handling.
5421 errno = GetLastError();
5423 result = -1;
5426 /* Free up the native handle at this point */
5427 CloseHandle(hProcess);
5430 /* Remove this file pointer from dictionary */
5431 PyDict_DelItem(_PyPopenProcs, fileObj);
5433 if (PyDict_Size(_PyPopenProcs) == 0) {
5434 Py_DECREF(_PyPopenProcs);
5435 _PyPopenProcs = NULL;
5438 } /* if object retrieval ok */
5440 Py_XDECREF(fileObj);
5441 } /* if _PyPopenProcs */
5443 #ifdef WITH_THREAD
5444 PyGILState_Release(state);
5445 #endif
5446 return result;
5449 #else /* which OS? */
5450 static PyObject *
5451 posix_popen(PyObject *self, PyObject *args)
5453 char *name;
5454 char *mode = "r";
5455 int bufsize = -1;
5456 FILE *fp;
5457 PyObject *f;
5458 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
5459 return NULL;
5460 /* Strip mode of binary or text modifiers */
5461 if (strcmp(mode, "rb") == 0 || strcmp(mode, "rt") == 0)
5462 mode = "r";
5463 else if (strcmp(mode, "wb") == 0 || strcmp(mode, "wt") == 0)
5464 mode = "w";
5465 Py_BEGIN_ALLOW_THREADS
5466 fp = popen(name, mode);
5467 Py_END_ALLOW_THREADS
5468 if (fp == NULL)
5469 return posix_error();
5470 f = PyFile_FromFile(fp, name, mode, pclose);
5471 if (f != NULL)
5472 PyFile_SetBufSize(f, bufsize);
5473 return f;
5476 #endif /* PYOS_??? */
5477 #endif /* HAVE_POPEN */
5480 #ifdef HAVE_SETUID
5481 PyDoc_STRVAR(posix_setuid__doc__,
5482 "setuid(uid)\n\n\
5483 Set the current process's user id.");
5485 static PyObject *
5486 posix_setuid(PyObject *self, PyObject *args)
5488 int uid;
5489 if (!PyArg_ParseTuple(args, "i:setuid", &uid))
5490 return NULL;
5491 if (setuid(uid) < 0)
5492 return posix_error();
5493 Py_INCREF(Py_None);
5494 return Py_None;
5496 #endif /* HAVE_SETUID */
5499 #ifdef HAVE_SETEUID
5500 PyDoc_STRVAR(posix_seteuid__doc__,
5501 "seteuid(uid)\n\n\
5502 Set the current process's effective user id.");
5504 static PyObject *
5505 posix_seteuid (PyObject *self, PyObject *args)
5507 int euid;
5508 if (!PyArg_ParseTuple(args, "i", &euid)) {
5509 return NULL;
5510 } else if (seteuid(euid) < 0) {
5511 return posix_error();
5512 } else {
5513 Py_INCREF(Py_None);
5514 return Py_None;
5517 #endif /* HAVE_SETEUID */
5519 #ifdef HAVE_SETEGID
5520 PyDoc_STRVAR(posix_setegid__doc__,
5521 "setegid(gid)\n\n\
5522 Set the current process's effective group id.");
5524 static PyObject *
5525 posix_setegid (PyObject *self, PyObject *args)
5527 int egid;
5528 if (!PyArg_ParseTuple(args, "i", &egid)) {
5529 return NULL;
5530 } else if (setegid(egid) < 0) {
5531 return posix_error();
5532 } else {
5533 Py_INCREF(Py_None);
5534 return Py_None;
5537 #endif /* HAVE_SETEGID */
5539 #ifdef HAVE_SETREUID
5540 PyDoc_STRVAR(posix_setreuid__doc__,
5541 "setreuid(ruid, euid)\n\n\
5542 Set the current process's real and effective user ids.");
5544 static PyObject *
5545 posix_setreuid (PyObject *self, PyObject *args)
5547 int ruid, euid;
5548 if (!PyArg_ParseTuple(args, "ii", &ruid, &euid)) {
5549 return NULL;
5550 } else if (setreuid(ruid, euid) < 0) {
5551 return posix_error();
5552 } else {
5553 Py_INCREF(Py_None);
5554 return Py_None;
5557 #endif /* HAVE_SETREUID */
5559 #ifdef HAVE_SETREGID
5560 PyDoc_STRVAR(posix_setregid__doc__,
5561 "setregid(rgid, egid)\n\n\
5562 Set the current process's real and effective group ids.");
5564 static PyObject *
5565 posix_setregid (PyObject *self, PyObject *args)
5567 int rgid, egid;
5568 if (!PyArg_ParseTuple(args, "ii", &rgid, &egid)) {
5569 return NULL;
5570 } else if (setregid(rgid, egid) < 0) {
5571 return posix_error();
5572 } else {
5573 Py_INCREF(Py_None);
5574 return Py_None;
5577 #endif /* HAVE_SETREGID */
5579 #ifdef HAVE_SETGID
5580 PyDoc_STRVAR(posix_setgid__doc__,
5581 "setgid(gid)\n\n\
5582 Set the current process's group id.");
5584 static PyObject *
5585 posix_setgid(PyObject *self, PyObject *args)
5587 int gid;
5588 if (!PyArg_ParseTuple(args, "i:setgid", &gid))
5589 return NULL;
5590 if (setgid(gid) < 0)
5591 return posix_error();
5592 Py_INCREF(Py_None);
5593 return Py_None;
5595 #endif /* HAVE_SETGID */
5597 #ifdef HAVE_SETGROUPS
5598 PyDoc_STRVAR(posix_setgroups__doc__,
5599 "setgroups(list)\n\n\
5600 Set the groups of the current process to list.");
5602 static PyObject *
5603 posix_setgroups(PyObject *self, PyObject *groups)
5605 int i, len;
5606 gid_t grouplist[MAX_GROUPS];
5608 if (!PySequence_Check(groups)) {
5609 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
5610 return NULL;
5612 len = PySequence_Size(groups);
5613 if (len > MAX_GROUPS) {
5614 PyErr_SetString(PyExc_ValueError, "too many groups");
5615 return NULL;
5617 for(i = 0; i < len; i++) {
5618 PyObject *elem;
5619 elem = PySequence_GetItem(groups, i);
5620 if (!elem)
5621 return NULL;
5622 if (!PyInt_Check(elem)) {
5623 if (!PyLong_Check(elem)) {
5624 PyErr_SetString(PyExc_TypeError,
5625 "groups must be integers");
5626 Py_DECREF(elem);
5627 return NULL;
5628 } else {
5629 unsigned long x = PyLong_AsUnsignedLong(elem);
5630 if (PyErr_Occurred()) {
5631 PyErr_SetString(PyExc_TypeError,
5632 "group id too big");
5633 Py_DECREF(elem);
5634 return NULL;
5636 grouplist[i] = x;
5637 /* read back the value to see if it fitted in gid_t */
5638 if (grouplist[i] != x) {
5639 PyErr_SetString(PyExc_TypeError,
5640 "group id too big");
5641 Py_DECREF(elem);
5642 return NULL;
5645 } else {
5646 long x = PyInt_AsLong(elem);
5647 grouplist[i] = x;
5648 if (grouplist[i] != x) {
5649 PyErr_SetString(PyExc_TypeError,
5650 "group id too big");
5651 Py_DECREF(elem);
5652 return NULL;
5655 Py_DECREF(elem);
5658 if (setgroups(len, grouplist) < 0)
5659 return posix_error();
5660 Py_INCREF(Py_None);
5661 return Py_None;
5663 #endif /* HAVE_SETGROUPS */
5665 #if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
5666 static PyObject *
5667 wait_helper(pid_t pid, int status, struct rusage *ru)
5669 PyObject *result;
5670 static PyObject *struct_rusage;
5672 if (pid == -1)
5673 return posix_error();
5675 if (struct_rusage == NULL) {
5676 PyObject *m = PyImport_ImportModuleNoBlock("resource");
5677 if (m == NULL)
5678 return NULL;
5679 struct_rusage = PyObject_GetAttrString(m, "struct_rusage");
5680 Py_DECREF(m);
5681 if (struct_rusage == NULL)
5682 return NULL;
5685 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
5686 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
5687 if (!result)
5688 return NULL;
5690 #ifndef doubletime
5691 #define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
5692 #endif
5694 PyStructSequence_SET_ITEM(result, 0,
5695 PyFloat_FromDouble(doubletime(ru->ru_utime)));
5696 PyStructSequence_SET_ITEM(result, 1,
5697 PyFloat_FromDouble(doubletime(ru->ru_stime)));
5698 #define SET_INT(result, index, value)\
5699 PyStructSequence_SET_ITEM(result, index, PyInt_FromLong(value))
5700 SET_INT(result, 2, ru->ru_maxrss);
5701 SET_INT(result, 3, ru->ru_ixrss);
5702 SET_INT(result, 4, ru->ru_idrss);
5703 SET_INT(result, 5, ru->ru_isrss);
5704 SET_INT(result, 6, ru->ru_minflt);
5705 SET_INT(result, 7, ru->ru_majflt);
5706 SET_INT(result, 8, ru->ru_nswap);
5707 SET_INT(result, 9, ru->ru_inblock);
5708 SET_INT(result, 10, ru->ru_oublock);
5709 SET_INT(result, 11, ru->ru_msgsnd);
5710 SET_INT(result, 12, ru->ru_msgrcv);
5711 SET_INT(result, 13, ru->ru_nsignals);
5712 SET_INT(result, 14, ru->ru_nvcsw);
5713 SET_INT(result, 15, ru->ru_nivcsw);
5714 #undef SET_INT
5716 if (PyErr_Occurred()) {
5717 Py_DECREF(result);
5718 return NULL;
5721 return Py_BuildValue("iiN", pid, status, result);
5723 #endif /* HAVE_WAIT3 || HAVE_WAIT4 */
5725 #ifdef HAVE_WAIT3
5726 PyDoc_STRVAR(posix_wait3__doc__,
5727 "wait3(options) -> (pid, status, rusage)\n\n\
5728 Wait for completion of a child process.");
5730 static PyObject *
5731 posix_wait3(PyObject *self, PyObject *args)
5733 pid_t pid;
5734 int options;
5735 struct rusage ru;
5736 WAIT_TYPE status;
5737 WAIT_STATUS_INT(status) = 0;
5739 if (!PyArg_ParseTuple(args, "i:wait3", &options))
5740 return NULL;
5742 Py_BEGIN_ALLOW_THREADS
5743 pid = wait3(&status, options, &ru);
5744 Py_END_ALLOW_THREADS
5746 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
5748 #endif /* HAVE_WAIT3 */
5750 #ifdef HAVE_WAIT4
5751 PyDoc_STRVAR(posix_wait4__doc__,
5752 "wait4(pid, options) -> (pid, status, rusage)\n\n\
5753 Wait for completion of a given child process.");
5755 static PyObject *
5756 posix_wait4(PyObject *self, PyObject *args)
5758 pid_t pid;
5759 int options;
5760 struct rusage ru;
5761 WAIT_TYPE status;
5762 WAIT_STATUS_INT(status) = 0;
5764 if (!PyArg_ParseTuple(args, "ii:wait4", &pid, &options))
5765 return NULL;
5767 Py_BEGIN_ALLOW_THREADS
5768 pid = wait4(pid, &status, options, &ru);
5769 Py_END_ALLOW_THREADS
5771 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
5773 #endif /* HAVE_WAIT4 */
5775 #ifdef HAVE_WAITPID
5776 PyDoc_STRVAR(posix_waitpid__doc__,
5777 "waitpid(pid, options) -> (pid, status)\n\n\
5778 Wait for completion of a given child process.");
5780 static PyObject *
5781 posix_waitpid(PyObject *self, PyObject *args)
5783 pid_t pid;
5784 int options;
5785 WAIT_TYPE status;
5786 WAIT_STATUS_INT(status) = 0;
5788 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
5789 return NULL;
5790 Py_BEGIN_ALLOW_THREADS
5791 pid = waitpid(pid, &status, options);
5792 Py_END_ALLOW_THREADS
5793 if (pid == -1)
5794 return posix_error();
5796 return Py_BuildValue("ii", pid, WAIT_STATUS_INT(status));
5799 #elif defined(HAVE_CWAIT)
5801 /* MS C has a variant of waitpid() that's usable for most purposes. */
5802 PyDoc_STRVAR(posix_waitpid__doc__,
5803 "waitpid(pid, options) -> (pid, status << 8)\n\n"
5804 "Wait for completion of a given process. options is ignored on Windows.");
5806 static PyObject *
5807 posix_waitpid(PyObject *self, PyObject *args)
5809 Py_intptr_t pid;
5810 int status, options;
5812 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
5813 return NULL;
5814 Py_BEGIN_ALLOW_THREADS
5815 pid = _cwait(&status, pid, options);
5816 Py_END_ALLOW_THREADS
5817 if (pid == -1)
5818 return posix_error();
5820 /* shift the status left a byte so this is more like the POSIX waitpid */
5821 return Py_BuildValue("ii", pid, status << 8);
5823 #endif /* HAVE_WAITPID || HAVE_CWAIT */
5825 #ifdef HAVE_WAIT
5826 PyDoc_STRVAR(posix_wait__doc__,
5827 "wait() -> (pid, status)\n\n\
5828 Wait for completion of a child process.");
5830 static PyObject *
5831 posix_wait(PyObject *self, PyObject *noargs)
5833 pid_t pid;
5834 WAIT_TYPE status;
5835 WAIT_STATUS_INT(status) = 0;
5837 Py_BEGIN_ALLOW_THREADS
5838 pid = wait(&status);
5839 Py_END_ALLOW_THREADS
5840 if (pid == -1)
5841 return posix_error();
5843 return Py_BuildValue("ii", pid, WAIT_STATUS_INT(status));
5845 #endif
5848 PyDoc_STRVAR(posix_lstat__doc__,
5849 "lstat(path) -> stat result\n\n\
5850 Like stat(path), but do not follow symbolic links.");
5852 static PyObject *
5853 posix_lstat(PyObject *self, PyObject *args)
5855 #ifdef HAVE_LSTAT
5856 return posix_do_stat(self, args, "et:lstat", lstat, NULL, NULL);
5857 #else /* !HAVE_LSTAT */
5858 #ifdef MS_WINDOWS
5859 return posix_do_stat(self, args, "et:lstat", STAT, "U:lstat", win32_wstat);
5860 #else
5861 return posix_do_stat(self, args, "et:lstat", STAT, NULL, NULL);
5862 #endif
5863 #endif /* !HAVE_LSTAT */
5867 #ifdef HAVE_READLINK
5868 PyDoc_STRVAR(posix_readlink__doc__,
5869 "readlink(path) -> path\n\n\
5870 Return a string representing the path to which the symbolic link points.");
5872 static PyObject *
5873 posix_readlink(PyObject *self, PyObject *args)
5875 PyObject* v;
5876 char buf[MAXPATHLEN];
5877 char *path;
5878 int n;
5879 #ifdef Py_USING_UNICODE
5880 int arg_is_unicode = 0;
5881 #endif
5883 if (!PyArg_ParseTuple(args, "et:readlink",
5884 Py_FileSystemDefaultEncoding, &path))
5885 return NULL;
5886 #ifdef Py_USING_UNICODE
5887 v = PySequence_GetItem(args, 0);
5888 if (v == NULL) {
5889 PyMem_Free(path);
5890 return NULL;
5893 if (PyUnicode_Check(v)) {
5894 arg_is_unicode = 1;
5896 Py_DECREF(v);
5897 #endif
5899 Py_BEGIN_ALLOW_THREADS
5900 n = readlink(path, buf, (int) sizeof buf);
5901 Py_END_ALLOW_THREADS
5902 if (n < 0)
5903 return posix_error_with_allocated_filename(path);
5905 PyMem_Free(path);
5906 v = PyString_FromStringAndSize(buf, n);
5907 #ifdef Py_USING_UNICODE
5908 if (arg_is_unicode) {
5909 PyObject *w;
5911 w = PyUnicode_FromEncodedObject(v,
5912 Py_FileSystemDefaultEncoding,
5913 "strict");
5914 if (w != NULL) {
5915 Py_DECREF(v);
5916 v = w;
5918 else {
5919 /* fall back to the original byte string, as
5920 discussed in patch #683592 */
5921 PyErr_Clear();
5924 #endif
5925 return v;
5927 #endif /* HAVE_READLINK */
5930 #ifdef HAVE_SYMLINK
5931 PyDoc_STRVAR(posix_symlink__doc__,
5932 "symlink(src, dst)\n\n\
5933 Create a symbolic link pointing to src named dst.");
5935 static PyObject *
5936 posix_symlink(PyObject *self, PyObject *args)
5938 return posix_2str(args, "etet:symlink", symlink);
5940 #endif /* HAVE_SYMLINK */
5943 #ifdef HAVE_TIMES
5944 #ifndef HZ
5945 #define HZ 60 /* Universal constant :-) */
5946 #endif /* HZ */
5948 #if defined(PYCC_VACPP) && defined(PYOS_OS2)
5949 static long
5950 system_uptime(void)
5952 ULONG value = 0;
5954 Py_BEGIN_ALLOW_THREADS
5955 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
5956 Py_END_ALLOW_THREADS
5958 return value;
5961 static PyObject *
5962 posix_times(PyObject *self, PyObject *noargs)
5964 /* Currently Only Uptime is Provided -- Others Later */
5965 return Py_BuildValue("ddddd",
5966 (double)0 /* t.tms_utime / HZ */,
5967 (double)0 /* t.tms_stime / HZ */,
5968 (double)0 /* t.tms_cutime / HZ */,
5969 (double)0 /* t.tms_cstime / HZ */,
5970 (double)system_uptime() / 1000);
5972 #else /* not OS2 */
5973 static PyObject *
5974 posix_times(PyObject *self, PyObject *noargs)
5976 struct tms t;
5977 clock_t c;
5978 errno = 0;
5979 c = times(&t);
5980 if (c == (clock_t) -1)
5981 return posix_error();
5982 return Py_BuildValue("ddddd",
5983 (double)t.tms_utime / HZ,
5984 (double)t.tms_stime / HZ,
5985 (double)t.tms_cutime / HZ,
5986 (double)t.tms_cstime / HZ,
5987 (double)c / HZ);
5989 #endif /* not OS2 */
5990 #endif /* HAVE_TIMES */
5993 #ifdef MS_WINDOWS
5994 #define HAVE_TIMES /* so the method table will pick it up */
5995 static PyObject *
5996 posix_times(PyObject *self, PyObject *noargs)
5998 FILETIME create, exit, kernel, user;
5999 HANDLE hProc;
6000 hProc = GetCurrentProcess();
6001 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
6002 /* The fields of a FILETIME structure are the hi and lo part
6003 of a 64-bit value expressed in 100 nanosecond units.
6004 1e7 is one second in such units; 1e-7 the inverse.
6005 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
6007 return Py_BuildValue(
6008 "ddddd",
6009 (double)(user.dwHighDateTime*429.4967296 +
6010 user.dwLowDateTime*1e-7),
6011 (double)(kernel.dwHighDateTime*429.4967296 +
6012 kernel.dwLowDateTime*1e-7),
6013 (double)0,
6014 (double)0,
6015 (double)0);
6017 #endif /* MS_WINDOWS */
6019 #ifdef HAVE_TIMES
6020 PyDoc_STRVAR(posix_times__doc__,
6021 "times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
6022 Return a tuple of floating point numbers indicating process times.");
6023 #endif
6026 #ifdef HAVE_GETSID
6027 PyDoc_STRVAR(posix_getsid__doc__,
6028 "getsid(pid) -> sid\n\n\
6029 Call the system call getsid().");
6031 static PyObject *
6032 posix_getsid(PyObject *self, PyObject *args)
6034 pid_t pid;
6035 int sid;
6036 if (!PyArg_ParseTuple(args, "i:getsid", &pid))
6037 return NULL;
6038 sid = getsid(pid);
6039 if (sid < 0)
6040 return posix_error();
6041 return PyInt_FromLong((long)sid);
6043 #endif /* HAVE_GETSID */
6046 #ifdef HAVE_SETSID
6047 PyDoc_STRVAR(posix_setsid__doc__,
6048 "setsid()\n\n\
6049 Call the system call setsid().");
6051 static PyObject *
6052 posix_setsid(PyObject *self, PyObject *noargs)
6054 if (setsid() < 0)
6055 return posix_error();
6056 Py_INCREF(Py_None);
6057 return Py_None;
6059 #endif /* HAVE_SETSID */
6061 #ifdef HAVE_SETPGID
6062 PyDoc_STRVAR(posix_setpgid__doc__,
6063 "setpgid(pid, pgrp)\n\n\
6064 Call the system call setpgid().");
6066 static PyObject *
6067 posix_setpgid(PyObject *self, PyObject *args)
6069 pid_t pid;
6070 int pgrp;
6071 if (!PyArg_ParseTuple(args, "ii:setpgid", &pid, &pgrp))
6072 return NULL;
6073 if (setpgid(pid, pgrp) < 0)
6074 return posix_error();
6075 Py_INCREF(Py_None);
6076 return Py_None;
6078 #endif /* HAVE_SETPGID */
6081 #ifdef HAVE_TCGETPGRP
6082 PyDoc_STRVAR(posix_tcgetpgrp__doc__,
6083 "tcgetpgrp(fd) -> pgid\n\n\
6084 Return the process group associated with the terminal given by a fd.");
6086 static PyObject *
6087 posix_tcgetpgrp(PyObject *self, PyObject *args)
6089 int fd;
6090 pid_t pgid;
6091 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
6092 return NULL;
6093 pgid = tcgetpgrp(fd);
6094 if (pgid < 0)
6095 return posix_error();
6096 return PyInt_FromLong((long)pgid);
6098 #endif /* HAVE_TCGETPGRP */
6101 #ifdef HAVE_TCSETPGRP
6102 PyDoc_STRVAR(posix_tcsetpgrp__doc__,
6103 "tcsetpgrp(fd, pgid)\n\n\
6104 Set the process group associated with the terminal given by a fd.");
6106 static PyObject *
6107 posix_tcsetpgrp(PyObject *self, PyObject *args)
6109 int fd, pgid;
6110 if (!PyArg_ParseTuple(args, "ii:tcsetpgrp", &fd, &pgid))
6111 return NULL;
6112 if (tcsetpgrp(fd, pgid) < 0)
6113 return posix_error();
6114 Py_INCREF(Py_None);
6115 return Py_None;
6117 #endif /* HAVE_TCSETPGRP */
6119 /* Functions acting on file descriptors */
6121 PyDoc_STRVAR(posix_open__doc__,
6122 "open(filename, flag [, mode=0777]) -> fd\n\n\
6123 Open a file (for low level IO).");
6125 static PyObject *
6126 posix_open(PyObject *self, PyObject *args)
6128 char *file = NULL;
6129 int flag;
6130 int mode = 0777;
6131 int fd;
6133 #ifdef MS_WINDOWS
6134 if (unicode_file_names()) {
6135 PyUnicodeObject *po;
6136 if (PyArg_ParseTuple(args, "Ui|i:mkdir", &po, &flag, &mode)) {
6137 Py_BEGIN_ALLOW_THREADS
6138 /* PyUnicode_AS_UNICODE OK without thread
6139 lock as it is a simple dereference. */
6140 fd = _wopen(PyUnicode_AS_UNICODE(po), flag, mode);
6141 Py_END_ALLOW_THREADS
6142 if (fd < 0)
6143 return posix_error();
6144 return PyInt_FromLong((long)fd);
6146 /* Drop the argument parsing error as narrow strings
6147 are also valid. */
6148 PyErr_Clear();
6150 #endif
6152 if (!PyArg_ParseTuple(args, "eti|i",
6153 Py_FileSystemDefaultEncoding, &file,
6154 &flag, &mode))
6155 return NULL;
6157 Py_BEGIN_ALLOW_THREADS
6158 fd = open(file, flag, mode);
6159 Py_END_ALLOW_THREADS
6160 if (fd < 0)
6161 return posix_error_with_allocated_filename(file);
6162 PyMem_Free(file);
6163 return PyInt_FromLong((long)fd);
6167 PyDoc_STRVAR(posix_close__doc__,
6168 "close(fd)\n\n\
6169 Close a file descriptor (for low level IO).");
6171 static PyObject *
6172 posix_close(PyObject *self, PyObject *args)
6174 int fd, res;
6175 if (!PyArg_ParseTuple(args, "i:close", &fd))
6176 return NULL;
6177 Py_BEGIN_ALLOW_THREADS
6178 res = close(fd);
6179 Py_END_ALLOW_THREADS
6180 if (res < 0)
6181 return posix_error();
6182 Py_INCREF(Py_None);
6183 return Py_None;
6187 PyDoc_STRVAR(posix_closerange__doc__,
6188 "closerange(fd_low, fd_high)\n\n\
6189 Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
6191 static PyObject *
6192 posix_closerange(PyObject *self, PyObject *args)
6194 int fd_from, fd_to, i;
6195 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
6196 return NULL;
6197 Py_BEGIN_ALLOW_THREADS
6198 for (i = fd_from; i < fd_to; i++)
6199 close(i);
6200 Py_END_ALLOW_THREADS
6201 Py_RETURN_NONE;
6205 PyDoc_STRVAR(posix_dup__doc__,
6206 "dup(fd) -> fd2\n\n\
6207 Return a duplicate of a file descriptor.");
6209 static PyObject *
6210 posix_dup(PyObject *self, PyObject *args)
6212 int fd;
6213 if (!PyArg_ParseTuple(args, "i:dup", &fd))
6214 return NULL;
6215 Py_BEGIN_ALLOW_THREADS
6216 fd = dup(fd);
6217 Py_END_ALLOW_THREADS
6218 if (fd < 0)
6219 return posix_error();
6220 return PyInt_FromLong((long)fd);
6224 PyDoc_STRVAR(posix_dup2__doc__,
6225 "dup2(old_fd, new_fd)\n\n\
6226 Duplicate file descriptor.");
6228 static PyObject *
6229 posix_dup2(PyObject *self, PyObject *args)
6231 int fd, fd2, res;
6232 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
6233 return NULL;
6234 Py_BEGIN_ALLOW_THREADS
6235 res = dup2(fd, fd2);
6236 Py_END_ALLOW_THREADS
6237 if (res < 0)
6238 return posix_error();
6239 Py_INCREF(Py_None);
6240 return Py_None;
6244 PyDoc_STRVAR(posix_lseek__doc__,
6245 "lseek(fd, pos, how) -> newpos\n\n\
6246 Set the current position of a file descriptor.");
6248 static PyObject *
6249 posix_lseek(PyObject *self, PyObject *args)
6251 int fd, how;
6252 #if defined(MS_WIN64) || defined(MS_WINDOWS)
6253 PY_LONG_LONG pos, res;
6254 #else
6255 off_t pos, res;
6256 #endif
6257 PyObject *posobj;
6258 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
6259 return NULL;
6260 #ifdef SEEK_SET
6261 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
6262 switch (how) {
6263 case 0: how = SEEK_SET; break;
6264 case 1: how = SEEK_CUR; break;
6265 case 2: how = SEEK_END; break;
6267 #endif /* SEEK_END */
6269 #if !defined(HAVE_LARGEFILE_SUPPORT)
6270 pos = PyInt_AsLong(posobj);
6271 #else
6272 pos = PyLong_Check(posobj) ?
6273 PyLong_AsLongLong(posobj) : PyInt_AsLong(posobj);
6274 #endif
6275 if (PyErr_Occurred())
6276 return NULL;
6278 Py_BEGIN_ALLOW_THREADS
6279 #if defined(MS_WIN64) || defined(MS_WINDOWS)
6280 res = _lseeki64(fd, pos, how);
6281 #else
6282 res = lseek(fd, pos, how);
6283 #endif
6284 Py_END_ALLOW_THREADS
6285 if (res < 0)
6286 return posix_error();
6288 #if !defined(HAVE_LARGEFILE_SUPPORT)
6289 return PyInt_FromLong(res);
6290 #else
6291 return PyLong_FromLongLong(res);
6292 #endif
6296 PyDoc_STRVAR(posix_read__doc__,
6297 "read(fd, buffersize) -> string\n\n\
6298 Read a file descriptor.");
6300 static PyObject *
6301 posix_read(PyObject *self, PyObject *args)
6303 int fd, size, n;
6304 PyObject *buffer;
6305 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
6306 return NULL;
6307 if (size < 0) {
6308 errno = EINVAL;
6309 return posix_error();
6311 buffer = PyString_FromStringAndSize((char *)NULL, size);
6312 if (buffer == NULL)
6313 return NULL;
6314 Py_BEGIN_ALLOW_THREADS
6315 n = read(fd, PyString_AsString(buffer), size);
6316 Py_END_ALLOW_THREADS
6317 if (n < 0) {
6318 Py_DECREF(buffer);
6319 return posix_error();
6321 if (n != size)
6322 _PyString_Resize(&buffer, n);
6323 return buffer;
6327 PyDoc_STRVAR(posix_write__doc__,
6328 "write(fd, string) -> byteswritten\n\n\
6329 Write a string to a file descriptor.");
6331 static PyObject *
6332 posix_write(PyObject *self, PyObject *args)
6334 int fd;
6335 Py_ssize_t size;
6336 char *buffer;
6338 if (!PyArg_ParseTuple(args, "is#:write", &fd, &buffer, &size))
6339 return NULL;
6340 Py_BEGIN_ALLOW_THREADS
6341 size = write(fd, buffer, (size_t)size);
6342 Py_END_ALLOW_THREADS
6343 if (size < 0)
6344 return posix_error();
6345 return PyInt_FromSsize_t(size);
6349 PyDoc_STRVAR(posix_fstat__doc__,
6350 "fstat(fd) -> stat result\n\n\
6351 Like stat(), but for an open file descriptor.");
6353 static PyObject *
6354 posix_fstat(PyObject *self, PyObject *args)
6356 int fd;
6357 STRUCT_STAT st;
6358 int res;
6359 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
6360 return NULL;
6361 #ifdef __VMS
6362 /* on OpenVMS we must ensure that all bytes are written to the file */
6363 fsync(fd);
6364 #endif
6365 Py_BEGIN_ALLOW_THREADS
6366 res = FSTAT(fd, &st);
6367 Py_END_ALLOW_THREADS
6368 if (res != 0) {
6369 #ifdef MS_WINDOWS
6370 return win32_error("fstat", NULL);
6371 #else
6372 return posix_error();
6373 #endif
6376 return _pystat_fromstructstat(&st);
6380 PyDoc_STRVAR(posix_fdopen__doc__,
6381 "fdopen(fd [, mode='r' [, bufsize]]) -> file_object\n\n\
6382 Return an open file object connected to a file descriptor.");
6384 static PyObject *
6385 posix_fdopen(PyObject *self, PyObject *args)
6387 int fd;
6388 char *orgmode = "r";
6389 int bufsize = -1;
6390 FILE *fp;
6391 PyObject *f;
6392 char *mode;
6393 if (!PyArg_ParseTuple(args, "i|si", &fd, &orgmode, &bufsize))
6394 return NULL;
6396 /* Sanitize mode. See fileobject.c */
6397 mode = PyMem_MALLOC(strlen(orgmode)+3);
6398 if (!mode) {
6399 PyErr_NoMemory();
6400 return NULL;
6402 strcpy(mode, orgmode);
6403 if (_PyFile_SanitizeMode(mode)) {
6404 PyMem_FREE(mode);
6405 return NULL;
6407 Py_BEGIN_ALLOW_THREADS
6408 #if !defined(MS_WINDOWS) && defined(HAVE_FCNTL_H)
6409 if (mode[0] == 'a') {
6410 /* try to make sure the O_APPEND flag is set */
6411 int flags;
6412 flags = fcntl(fd, F_GETFL);
6413 if (flags != -1)
6414 fcntl(fd, F_SETFL, flags | O_APPEND);
6415 fp = fdopen(fd, mode);
6416 if (fp == NULL && flags != -1)
6417 /* restore old mode if fdopen failed */
6418 fcntl(fd, F_SETFL, flags);
6419 } else {
6420 fp = fdopen(fd, mode);
6422 #else
6423 fp = fdopen(fd, mode);
6424 #endif
6425 Py_END_ALLOW_THREADS
6426 PyMem_FREE(mode);
6427 if (fp == NULL)
6428 return posix_error();
6429 f = PyFile_FromFile(fp, "<fdopen>", orgmode, fclose);
6430 if (f != NULL)
6431 PyFile_SetBufSize(f, bufsize);
6432 return f;
6435 PyDoc_STRVAR(posix_isatty__doc__,
6436 "isatty(fd) -> bool\n\n\
6437 Return True if the file descriptor 'fd' is an open file descriptor\n\
6438 connected to the slave end of a terminal.");
6440 static PyObject *
6441 posix_isatty(PyObject *self, PyObject *args)
6443 int fd;
6444 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
6445 return NULL;
6446 return PyBool_FromLong(isatty(fd));
6449 #ifdef HAVE_PIPE
6450 PyDoc_STRVAR(posix_pipe__doc__,
6451 "pipe() -> (read_end, write_end)\n\n\
6452 Create a pipe.");
6454 static PyObject *
6455 posix_pipe(PyObject *self, PyObject *noargs)
6457 #if defined(PYOS_OS2)
6458 HFILE read, write;
6459 APIRET rc;
6461 Py_BEGIN_ALLOW_THREADS
6462 rc = DosCreatePipe( &read, &write, 4096);
6463 Py_END_ALLOW_THREADS
6464 if (rc != NO_ERROR)
6465 return os2_error(rc);
6467 return Py_BuildValue("(ii)", read, write);
6468 #else
6469 #if !defined(MS_WINDOWS)
6470 int fds[2];
6471 int res;
6472 Py_BEGIN_ALLOW_THREADS
6473 res = pipe(fds);
6474 Py_END_ALLOW_THREADS
6475 if (res != 0)
6476 return posix_error();
6477 return Py_BuildValue("(ii)", fds[0], fds[1]);
6478 #else /* MS_WINDOWS */
6479 HANDLE read, write;
6480 int read_fd, write_fd;
6481 BOOL ok;
6482 Py_BEGIN_ALLOW_THREADS
6483 ok = CreatePipe(&read, &write, NULL, 0);
6484 Py_END_ALLOW_THREADS
6485 if (!ok)
6486 return win32_error("CreatePipe", NULL);
6487 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
6488 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
6489 return Py_BuildValue("(ii)", read_fd, write_fd);
6490 #endif /* MS_WINDOWS */
6491 #endif
6493 #endif /* HAVE_PIPE */
6496 #ifdef HAVE_MKFIFO
6497 PyDoc_STRVAR(posix_mkfifo__doc__,
6498 "mkfifo(filename [, mode=0666])\n\n\
6499 Create a FIFO (a POSIX named pipe).");
6501 static PyObject *
6502 posix_mkfifo(PyObject *self, PyObject *args)
6504 char *filename;
6505 int mode = 0666;
6506 int res;
6507 if (!PyArg_ParseTuple(args, "s|i:mkfifo", &filename, &mode))
6508 return NULL;
6509 Py_BEGIN_ALLOW_THREADS
6510 res = mkfifo(filename, mode);
6511 Py_END_ALLOW_THREADS
6512 if (res < 0)
6513 return posix_error();
6514 Py_INCREF(Py_None);
6515 return Py_None;
6517 #endif
6520 #if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
6521 PyDoc_STRVAR(posix_mknod__doc__,
6522 "mknod(filename [, mode=0600, device])\n\n\
6523 Create a filesystem node (file, device special file or named pipe)\n\
6524 named filename. mode specifies both the permissions to use and the\n\
6525 type of node to be created, being combined (bitwise OR) with one of\n\
6526 S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
6527 device defines the newly created device special file (probably using\n\
6528 os.makedev()), otherwise it is ignored.");
6531 static PyObject *
6532 posix_mknod(PyObject *self, PyObject *args)
6534 char *filename;
6535 int mode = 0600;
6536 int device = 0;
6537 int res;
6538 if (!PyArg_ParseTuple(args, "s|ii:mknod", &filename, &mode, &device))
6539 return NULL;
6540 Py_BEGIN_ALLOW_THREADS
6541 res = mknod(filename, mode, device);
6542 Py_END_ALLOW_THREADS
6543 if (res < 0)
6544 return posix_error();
6545 Py_INCREF(Py_None);
6546 return Py_None;
6548 #endif
6550 #ifdef HAVE_DEVICE_MACROS
6551 PyDoc_STRVAR(posix_major__doc__,
6552 "major(device) -> major number\n\
6553 Extracts a device major number from a raw device number.");
6555 static PyObject *
6556 posix_major(PyObject *self, PyObject *args)
6558 int device;
6559 if (!PyArg_ParseTuple(args, "i:major", &device))
6560 return NULL;
6561 return PyInt_FromLong((long)major(device));
6564 PyDoc_STRVAR(posix_minor__doc__,
6565 "minor(device) -> minor number\n\
6566 Extracts a device minor number from a raw device number.");
6568 static PyObject *
6569 posix_minor(PyObject *self, PyObject *args)
6571 int device;
6572 if (!PyArg_ParseTuple(args, "i:minor", &device))
6573 return NULL;
6574 return PyInt_FromLong((long)minor(device));
6577 PyDoc_STRVAR(posix_makedev__doc__,
6578 "makedev(major, minor) -> device number\n\
6579 Composes a raw device number from the major and minor device numbers.");
6581 static PyObject *
6582 posix_makedev(PyObject *self, PyObject *args)
6584 int major, minor;
6585 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
6586 return NULL;
6587 return PyInt_FromLong((long)makedev(major, minor));
6589 #endif /* device macros */
6592 #ifdef HAVE_FTRUNCATE
6593 PyDoc_STRVAR(posix_ftruncate__doc__,
6594 "ftruncate(fd, length)\n\n\
6595 Truncate a file to a specified length.");
6597 static PyObject *
6598 posix_ftruncate(PyObject *self, PyObject *args)
6600 int fd;
6601 off_t length;
6602 int res;
6603 PyObject *lenobj;
6605 if (!PyArg_ParseTuple(args, "iO:ftruncate", &fd, &lenobj))
6606 return NULL;
6608 #if !defined(HAVE_LARGEFILE_SUPPORT)
6609 length = PyInt_AsLong(lenobj);
6610 #else
6611 length = PyLong_Check(lenobj) ?
6612 PyLong_AsLongLong(lenobj) : PyInt_AsLong(lenobj);
6613 #endif
6614 if (PyErr_Occurred())
6615 return NULL;
6617 Py_BEGIN_ALLOW_THREADS
6618 res = ftruncate(fd, length);
6619 Py_END_ALLOW_THREADS
6620 if (res < 0) {
6621 PyErr_SetFromErrno(PyExc_IOError);
6622 return NULL;
6624 Py_INCREF(Py_None);
6625 return Py_None;
6627 #endif
6629 #ifdef HAVE_PUTENV
6630 PyDoc_STRVAR(posix_putenv__doc__,
6631 "putenv(key, value)\n\n\
6632 Change or add an environment variable.");
6634 /* Save putenv() parameters as values here, so we can collect them when they
6635 * get re-set with another call for the same key. */
6636 static PyObject *posix_putenv_garbage;
6638 static PyObject *
6639 posix_putenv(PyObject *self, PyObject *args)
6641 char *s1, *s2;
6642 char *newenv;
6643 PyObject *newstr;
6644 size_t len;
6646 if (!PyArg_ParseTuple(args, "ss:putenv", &s1, &s2))
6647 return NULL;
6649 #if defined(PYOS_OS2)
6650 if (stricmp(s1, "BEGINLIBPATH") == 0) {
6651 APIRET rc;
6653 rc = DosSetExtLIBPATH(s2, BEGIN_LIBPATH);
6654 if (rc != NO_ERROR)
6655 return os2_error(rc);
6657 } else if (stricmp(s1, "ENDLIBPATH") == 0) {
6658 APIRET rc;
6660 rc = DosSetExtLIBPATH(s2, END_LIBPATH);
6661 if (rc != NO_ERROR)
6662 return os2_error(rc);
6663 } else {
6664 #endif
6666 /* XXX This can leak memory -- not easy to fix :-( */
6667 len = strlen(s1) + strlen(s2) + 2;
6668 /* len includes space for a trailing \0; the size arg to
6669 PyString_FromStringAndSize does not count that */
6670 newstr = PyString_FromStringAndSize(NULL, (int)len - 1);
6671 if (newstr == NULL)
6672 return PyErr_NoMemory();
6673 newenv = PyString_AS_STRING(newstr);
6674 PyOS_snprintf(newenv, len, "%s=%s", s1, s2);
6675 if (putenv(newenv)) {
6676 Py_DECREF(newstr);
6677 posix_error();
6678 return NULL;
6680 /* Install the first arg and newstr in posix_putenv_garbage;
6681 * this will cause previous value to be collected. This has to
6682 * happen after the real putenv() call because the old value
6683 * was still accessible until then. */
6684 if (PyDict_SetItem(posix_putenv_garbage,
6685 PyTuple_GET_ITEM(args, 0), newstr)) {
6686 /* really not much we can do; just leak */
6687 PyErr_Clear();
6689 else {
6690 Py_DECREF(newstr);
6693 #if defined(PYOS_OS2)
6695 #endif
6696 Py_INCREF(Py_None);
6697 return Py_None;
6699 #endif /* putenv */
6701 #ifdef HAVE_UNSETENV
6702 PyDoc_STRVAR(posix_unsetenv__doc__,
6703 "unsetenv(key)\n\n\
6704 Delete an environment variable.");
6706 static PyObject *
6707 posix_unsetenv(PyObject *self, PyObject *args)
6709 char *s1;
6711 if (!PyArg_ParseTuple(args, "s:unsetenv", &s1))
6712 return NULL;
6714 unsetenv(s1);
6716 /* Remove the key from posix_putenv_garbage;
6717 * this will cause it to be collected. This has to
6718 * happen after the real unsetenv() call because the
6719 * old value was still accessible until then.
6721 if (PyDict_DelItem(posix_putenv_garbage,
6722 PyTuple_GET_ITEM(args, 0))) {
6723 /* really not much we can do; just leak */
6724 PyErr_Clear();
6727 Py_INCREF(Py_None);
6728 return Py_None;
6730 #endif /* unsetenv */
6732 PyDoc_STRVAR(posix_strerror__doc__,
6733 "strerror(code) -> string\n\n\
6734 Translate an error code to a message string.");
6736 static PyObject *
6737 posix_strerror(PyObject *self, PyObject *args)
6739 int code;
6740 char *message;
6741 if (!PyArg_ParseTuple(args, "i:strerror", &code))
6742 return NULL;
6743 message = strerror(code);
6744 if (message == NULL) {
6745 PyErr_SetString(PyExc_ValueError,
6746 "strerror() argument out of range");
6747 return NULL;
6749 return PyString_FromString(message);
6753 #ifdef HAVE_SYS_WAIT_H
6755 #ifdef WCOREDUMP
6756 PyDoc_STRVAR(posix_WCOREDUMP__doc__,
6757 "WCOREDUMP(status) -> bool\n\n\
6758 Return True if the process returning 'status' was dumped to a core file.");
6760 static PyObject *
6761 posix_WCOREDUMP(PyObject *self, PyObject *args)
6763 WAIT_TYPE status;
6764 WAIT_STATUS_INT(status) = 0;
6766 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
6767 return NULL;
6769 return PyBool_FromLong(WCOREDUMP(status));
6771 #endif /* WCOREDUMP */
6773 #ifdef WIFCONTINUED
6774 PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
6775 "WIFCONTINUED(status) -> bool\n\n\
6776 Return True if the process returning 'status' was continued from a\n\
6777 job control stop.");
6779 static PyObject *
6780 posix_WIFCONTINUED(PyObject *self, PyObject *args)
6782 WAIT_TYPE status;
6783 WAIT_STATUS_INT(status) = 0;
6785 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
6786 return NULL;
6788 return PyBool_FromLong(WIFCONTINUED(status));
6790 #endif /* WIFCONTINUED */
6792 #ifdef WIFSTOPPED
6793 PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
6794 "WIFSTOPPED(status) -> bool\n\n\
6795 Return True if the process returning 'status' was stopped.");
6797 static PyObject *
6798 posix_WIFSTOPPED(PyObject *self, PyObject *args)
6800 WAIT_TYPE status;
6801 WAIT_STATUS_INT(status) = 0;
6803 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
6804 return NULL;
6806 return PyBool_FromLong(WIFSTOPPED(status));
6808 #endif /* WIFSTOPPED */
6810 #ifdef WIFSIGNALED
6811 PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
6812 "WIFSIGNALED(status) -> bool\n\n\
6813 Return True if the process returning 'status' was terminated by a signal.");
6815 static PyObject *
6816 posix_WIFSIGNALED(PyObject *self, PyObject *args)
6818 WAIT_TYPE status;
6819 WAIT_STATUS_INT(status) = 0;
6821 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
6822 return NULL;
6824 return PyBool_FromLong(WIFSIGNALED(status));
6826 #endif /* WIFSIGNALED */
6828 #ifdef WIFEXITED
6829 PyDoc_STRVAR(posix_WIFEXITED__doc__,
6830 "WIFEXITED(status) -> bool\n\n\
6831 Return true if the process returning 'status' exited using the exit()\n\
6832 system call.");
6834 static PyObject *
6835 posix_WIFEXITED(PyObject *self, PyObject *args)
6837 WAIT_TYPE status;
6838 WAIT_STATUS_INT(status) = 0;
6840 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
6841 return NULL;
6843 return PyBool_FromLong(WIFEXITED(status));
6845 #endif /* WIFEXITED */
6847 #ifdef WEXITSTATUS
6848 PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
6849 "WEXITSTATUS(status) -> integer\n\n\
6850 Return the process return code from 'status'.");
6852 static PyObject *
6853 posix_WEXITSTATUS(PyObject *self, PyObject *args)
6855 WAIT_TYPE status;
6856 WAIT_STATUS_INT(status) = 0;
6858 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
6859 return NULL;
6861 return Py_BuildValue("i", WEXITSTATUS(status));
6863 #endif /* WEXITSTATUS */
6865 #ifdef WTERMSIG
6866 PyDoc_STRVAR(posix_WTERMSIG__doc__,
6867 "WTERMSIG(status) -> integer\n\n\
6868 Return the signal that terminated the process that provided the 'status'\n\
6869 value.");
6871 static PyObject *
6872 posix_WTERMSIG(PyObject *self, PyObject *args)
6874 WAIT_TYPE status;
6875 WAIT_STATUS_INT(status) = 0;
6877 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
6878 return NULL;
6880 return Py_BuildValue("i", WTERMSIG(status));
6882 #endif /* WTERMSIG */
6884 #ifdef WSTOPSIG
6885 PyDoc_STRVAR(posix_WSTOPSIG__doc__,
6886 "WSTOPSIG(status) -> integer\n\n\
6887 Return the signal that stopped the process that provided\n\
6888 the 'status' value.");
6890 static PyObject *
6891 posix_WSTOPSIG(PyObject *self, PyObject *args)
6893 WAIT_TYPE status;
6894 WAIT_STATUS_INT(status) = 0;
6896 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &WAIT_STATUS_INT(status)))
6897 return NULL;
6899 return Py_BuildValue("i", WSTOPSIG(status));
6901 #endif /* WSTOPSIG */
6903 #endif /* HAVE_SYS_WAIT_H */
6906 #if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
6907 #ifdef _SCO_DS
6908 /* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
6909 needed definitions in sys/statvfs.h */
6910 #define _SVID3
6911 #endif
6912 #include <sys/statvfs.h>
6914 static PyObject*
6915 _pystatvfs_fromstructstatvfs(struct statvfs st) {
6916 PyObject *v = PyStructSequence_New(&StatVFSResultType);
6917 if (v == NULL)
6918 return NULL;
6920 #if !defined(HAVE_LARGEFILE_SUPPORT)
6921 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long) st.f_bsize));
6922 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long) st.f_frsize));
6923 PyStructSequence_SET_ITEM(v, 2, PyInt_FromLong((long) st.f_blocks));
6924 PyStructSequence_SET_ITEM(v, 3, PyInt_FromLong((long) st.f_bfree));
6925 PyStructSequence_SET_ITEM(v, 4, PyInt_FromLong((long) st.f_bavail));
6926 PyStructSequence_SET_ITEM(v, 5, PyInt_FromLong((long) st.f_files));
6927 PyStructSequence_SET_ITEM(v, 6, PyInt_FromLong((long) st.f_ffree));
6928 PyStructSequence_SET_ITEM(v, 7, PyInt_FromLong((long) st.f_favail));
6929 PyStructSequence_SET_ITEM(v, 8, PyInt_FromLong((long) st.f_flag));
6930 PyStructSequence_SET_ITEM(v, 9, PyInt_FromLong((long) st.f_namemax));
6931 #else
6932 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long) st.f_bsize));
6933 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long) st.f_frsize));
6934 PyStructSequence_SET_ITEM(v, 2,
6935 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
6936 PyStructSequence_SET_ITEM(v, 3,
6937 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
6938 PyStructSequence_SET_ITEM(v, 4,
6939 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
6940 PyStructSequence_SET_ITEM(v, 5,
6941 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
6942 PyStructSequence_SET_ITEM(v, 6,
6943 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
6944 PyStructSequence_SET_ITEM(v, 7,
6945 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
6946 PyStructSequence_SET_ITEM(v, 8, PyInt_FromLong((long) st.f_flag));
6947 PyStructSequence_SET_ITEM(v, 9, PyInt_FromLong((long) st.f_namemax));
6948 #endif
6950 return v;
6953 PyDoc_STRVAR(posix_fstatvfs__doc__,
6954 "fstatvfs(fd) -> statvfs result\n\n\
6955 Perform an fstatvfs system call on the given fd.");
6957 static PyObject *
6958 posix_fstatvfs(PyObject *self, PyObject *args)
6960 int fd, res;
6961 struct statvfs st;
6963 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
6964 return NULL;
6965 Py_BEGIN_ALLOW_THREADS
6966 res = fstatvfs(fd, &st);
6967 Py_END_ALLOW_THREADS
6968 if (res != 0)
6969 return posix_error();
6971 return _pystatvfs_fromstructstatvfs(st);
6973 #endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
6976 #if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
6977 #include <sys/statvfs.h>
6979 PyDoc_STRVAR(posix_statvfs__doc__,
6980 "statvfs(path) -> statvfs result\n\n\
6981 Perform a statvfs system call on the given path.");
6983 static PyObject *
6984 posix_statvfs(PyObject *self, PyObject *args)
6986 char *path;
6987 int res;
6988 struct statvfs st;
6989 if (!PyArg_ParseTuple(args, "s:statvfs", &path))
6990 return NULL;
6991 Py_BEGIN_ALLOW_THREADS
6992 res = statvfs(path, &st);
6993 Py_END_ALLOW_THREADS
6994 if (res != 0)
6995 return posix_error_with_filename(path);
6997 return _pystatvfs_fromstructstatvfs(st);
6999 #endif /* HAVE_STATVFS */
7002 #ifdef HAVE_TEMPNAM
7003 PyDoc_STRVAR(posix_tempnam__doc__,
7004 "tempnam([dir[, prefix]]) -> string\n\n\
7005 Return a unique name for a temporary file.\n\
7006 The directory and a prefix may be specified as strings; they may be omitted\n\
7007 or None if not needed.");
7009 static PyObject *
7010 posix_tempnam(PyObject *self, PyObject *args)
7012 PyObject *result = NULL;
7013 char *dir = NULL;
7014 char *pfx = NULL;
7015 char *name;
7017 if (!PyArg_ParseTuple(args, "|zz:tempnam", &dir, &pfx))
7018 return NULL;
7020 if (PyErr_Warn(PyExc_RuntimeWarning,
7021 "tempnam is a potential security risk to your program") < 0)
7022 return NULL;
7024 #ifdef MS_WINDOWS
7025 name = _tempnam(dir, pfx);
7026 #else
7027 name = tempnam(dir, pfx);
7028 #endif
7029 if (name == NULL)
7030 return PyErr_NoMemory();
7031 result = PyString_FromString(name);
7032 free(name);
7033 return result;
7035 #endif
7038 #ifdef HAVE_TMPFILE
7039 PyDoc_STRVAR(posix_tmpfile__doc__,
7040 "tmpfile() -> file object\n\n\
7041 Create a temporary file with no directory entries.");
7043 static PyObject *
7044 posix_tmpfile(PyObject *self, PyObject *noargs)
7046 FILE *fp;
7048 fp = tmpfile();
7049 if (fp == NULL)
7050 return posix_error();
7051 return PyFile_FromFile(fp, "<tmpfile>", "w+b", fclose);
7053 #endif
7056 #ifdef HAVE_TMPNAM
7057 PyDoc_STRVAR(posix_tmpnam__doc__,
7058 "tmpnam() -> string\n\n\
7059 Return a unique name for a temporary file.");
7061 static PyObject *
7062 posix_tmpnam(PyObject *self, PyObject *noargs)
7064 char buffer[L_tmpnam];
7065 char *name;
7067 if (PyErr_Warn(PyExc_RuntimeWarning,
7068 "tmpnam is a potential security risk to your program") < 0)
7069 return NULL;
7071 #ifdef USE_TMPNAM_R
7072 name = tmpnam_r(buffer);
7073 #else
7074 name = tmpnam(buffer);
7075 #endif
7076 if (name == NULL) {
7077 PyObject *err = Py_BuildValue("is", 0,
7078 #ifdef USE_TMPNAM_R
7079 "unexpected NULL from tmpnam_r"
7080 #else
7081 "unexpected NULL from tmpnam"
7082 #endif
7084 PyErr_SetObject(PyExc_OSError, err);
7085 Py_XDECREF(err);
7086 return NULL;
7088 return PyString_FromString(buffer);
7090 #endif
7093 /* This is used for fpathconf(), pathconf(), confstr() and sysconf().
7094 * It maps strings representing configuration variable names to
7095 * integer values, allowing those functions to be called with the
7096 * magic names instead of polluting the module's namespace with tons of
7097 * rarely-used constants. There are three separate tables that use
7098 * these definitions.
7100 * This code is always included, even if none of the interfaces that
7101 * need it are included. The #if hackery needed to avoid it would be
7102 * sufficiently pervasive that it's not worth the loss of readability.
7104 struct constdef {
7105 char *name;
7106 long value;
7109 static int
7110 conv_confname(PyObject *arg, int *valuep, struct constdef *table,
7111 size_t tablesize)
7113 if (PyInt_Check(arg)) {
7114 *valuep = PyInt_AS_LONG(arg);
7115 return 1;
7117 if (PyString_Check(arg)) {
7118 /* look up the value in the table using a binary search */
7119 size_t lo = 0;
7120 size_t mid;
7121 size_t hi = tablesize;
7122 int cmp;
7123 char *confname = PyString_AS_STRING(arg);
7124 while (lo < hi) {
7125 mid = (lo + hi) / 2;
7126 cmp = strcmp(confname, table[mid].name);
7127 if (cmp < 0)
7128 hi = mid;
7129 else if (cmp > 0)
7130 lo = mid + 1;
7131 else {
7132 *valuep = table[mid].value;
7133 return 1;
7136 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
7138 else
7139 PyErr_SetString(PyExc_TypeError,
7140 "configuration names must be strings or integers");
7141 return 0;
7145 #if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
7146 static struct constdef posix_constants_pathconf[] = {
7147 #ifdef _PC_ABI_AIO_XFER_MAX
7148 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
7149 #endif
7150 #ifdef _PC_ABI_ASYNC_IO
7151 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
7152 #endif
7153 #ifdef _PC_ASYNC_IO
7154 {"PC_ASYNC_IO", _PC_ASYNC_IO},
7155 #endif
7156 #ifdef _PC_CHOWN_RESTRICTED
7157 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
7158 #endif
7159 #ifdef _PC_FILESIZEBITS
7160 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
7161 #endif
7162 #ifdef _PC_LAST
7163 {"PC_LAST", _PC_LAST},
7164 #endif
7165 #ifdef _PC_LINK_MAX
7166 {"PC_LINK_MAX", _PC_LINK_MAX},
7167 #endif
7168 #ifdef _PC_MAX_CANON
7169 {"PC_MAX_CANON", _PC_MAX_CANON},
7170 #endif
7171 #ifdef _PC_MAX_INPUT
7172 {"PC_MAX_INPUT", _PC_MAX_INPUT},
7173 #endif
7174 #ifdef _PC_NAME_MAX
7175 {"PC_NAME_MAX", _PC_NAME_MAX},
7176 #endif
7177 #ifdef _PC_NO_TRUNC
7178 {"PC_NO_TRUNC", _PC_NO_TRUNC},
7179 #endif
7180 #ifdef _PC_PATH_MAX
7181 {"PC_PATH_MAX", _PC_PATH_MAX},
7182 #endif
7183 #ifdef _PC_PIPE_BUF
7184 {"PC_PIPE_BUF", _PC_PIPE_BUF},
7185 #endif
7186 #ifdef _PC_PRIO_IO
7187 {"PC_PRIO_IO", _PC_PRIO_IO},
7188 #endif
7189 #ifdef _PC_SOCK_MAXBUF
7190 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
7191 #endif
7192 #ifdef _PC_SYNC_IO
7193 {"PC_SYNC_IO", _PC_SYNC_IO},
7194 #endif
7195 #ifdef _PC_VDISABLE
7196 {"PC_VDISABLE", _PC_VDISABLE},
7197 #endif
7200 static int
7201 conv_path_confname(PyObject *arg, int *valuep)
7203 return conv_confname(arg, valuep, posix_constants_pathconf,
7204 sizeof(posix_constants_pathconf)
7205 / sizeof(struct constdef));
7207 #endif
7209 #ifdef HAVE_FPATHCONF
7210 PyDoc_STRVAR(posix_fpathconf__doc__,
7211 "fpathconf(fd, name) -> integer\n\n\
7212 Return the configuration limit name for the file descriptor fd.\n\
7213 If there is no limit, return -1.");
7215 static PyObject *
7216 posix_fpathconf(PyObject *self, PyObject *args)
7218 PyObject *result = NULL;
7219 int name, fd;
7221 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
7222 conv_path_confname, &name)) {
7223 long limit;
7225 errno = 0;
7226 limit = fpathconf(fd, name);
7227 if (limit == -1 && errno != 0)
7228 posix_error();
7229 else
7230 result = PyInt_FromLong(limit);
7232 return result;
7234 #endif
7237 #ifdef HAVE_PATHCONF
7238 PyDoc_STRVAR(posix_pathconf__doc__,
7239 "pathconf(path, name) -> integer\n\n\
7240 Return the configuration limit name for the file or directory path.\n\
7241 If there is no limit, return -1.");
7243 static PyObject *
7244 posix_pathconf(PyObject *self, PyObject *args)
7246 PyObject *result = NULL;
7247 int name;
7248 char *path;
7250 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
7251 conv_path_confname, &name)) {
7252 long limit;
7254 errno = 0;
7255 limit = pathconf(path, name);
7256 if (limit == -1 && errno != 0) {
7257 if (errno == EINVAL)
7258 /* could be a path or name problem */
7259 posix_error();
7260 else
7261 posix_error_with_filename(path);
7263 else
7264 result = PyInt_FromLong(limit);
7266 return result;
7268 #endif
7270 #ifdef HAVE_CONFSTR
7271 static struct constdef posix_constants_confstr[] = {
7272 #ifdef _CS_ARCHITECTURE
7273 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
7274 #endif
7275 #ifdef _CS_HOSTNAME
7276 {"CS_HOSTNAME", _CS_HOSTNAME},
7277 #endif
7278 #ifdef _CS_HW_PROVIDER
7279 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
7280 #endif
7281 #ifdef _CS_HW_SERIAL
7282 {"CS_HW_SERIAL", _CS_HW_SERIAL},
7283 #endif
7284 #ifdef _CS_INITTAB_NAME
7285 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
7286 #endif
7287 #ifdef _CS_LFS64_CFLAGS
7288 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
7289 #endif
7290 #ifdef _CS_LFS64_LDFLAGS
7291 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
7292 #endif
7293 #ifdef _CS_LFS64_LIBS
7294 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
7295 #endif
7296 #ifdef _CS_LFS64_LINTFLAGS
7297 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
7298 #endif
7299 #ifdef _CS_LFS_CFLAGS
7300 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
7301 #endif
7302 #ifdef _CS_LFS_LDFLAGS
7303 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
7304 #endif
7305 #ifdef _CS_LFS_LIBS
7306 {"CS_LFS_LIBS", _CS_LFS_LIBS},
7307 #endif
7308 #ifdef _CS_LFS_LINTFLAGS
7309 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
7310 #endif
7311 #ifdef _CS_MACHINE
7312 {"CS_MACHINE", _CS_MACHINE},
7313 #endif
7314 #ifdef _CS_PATH
7315 {"CS_PATH", _CS_PATH},
7316 #endif
7317 #ifdef _CS_RELEASE
7318 {"CS_RELEASE", _CS_RELEASE},
7319 #endif
7320 #ifdef _CS_SRPC_DOMAIN
7321 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
7322 #endif
7323 #ifdef _CS_SYSNAME
7324 {"CS_SYSNAME", _CS_SYSNAME},
7325 #endif
7326 #ifdef _CS_VERSION
7327 {"CS_VERSION", _CS_VERSION},
7328 #endif
7329 #ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
7330 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
7331 #endif
7332 #ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
7333 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
7334 #endif
7335 #ifdef _CS_XBS5_ILP32_OFF32_LIBS
7336 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
7337 #endif
7338 #ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
7339 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
7340 #endif
7341 #ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
7342 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
7343 #endif
7344 #ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
7345 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
7346 #endif
7347 #ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
7348 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
7349 #endif
7350 #ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
7351 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
7352 #endif
7353 #ifdef _CS_XBS5_LP64_OFF64_CFLAGS
7354 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
7355 #endif
7356 #ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
7357 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
7358 #endif
7359 #ifdef _CS_XBS5_LP64_OFF64_LIBS
7360 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
7361 #endif
7362 #ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
7363 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
7364 #endif
7365 #ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
7366 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
7367 #endif
7368 #ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
7369 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
7370 #endif
7371 #ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
7372 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
7373 #endif
7374 #ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
7375 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
7376 #endif
7377 #ifdef _MIPS_CS_AVAIL_PROCESSORS
7378 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
7379 #endif
7380 #ifdef _MIPS_CS_BASE
7381 {"MIPS_CS_BASE", _MIPS_CS_BASE},
7382 #endif
7383 #ifdef _MIPS_CS_HOSTID
7384 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
7385 #endif
7386 #ifdef _MIPS_CS_HW_NAME
7387 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
7388 #endif
7389 #ifdef _MIPS_CS_NUM_PROCESSORS
7390 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
7391 #endif
7392 #ifdef _MIPS_CS_OSREL_MAJ
7393 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
7394 #endif
7395 #ifdef _MIPS_CS_OSREL_MIN
7396 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
7397 #endif
7398 #ifdef _MIPS_CS_OSREL_PATCH
7399 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
7400 #endif
7401 #ifdef _MIPS_CS_OS_NAME
7402 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
7403 #endif
7404 #ifdef _MIPS_CS_OS_PROVIDER
7405 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
7406 #endif
7407 #ifdef _MIPS_CS_PROCESSORS
7408 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
7409 #endif
7410 #ifdef _MIPS_CS_SERIAL
7411 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
7412 #endif
7413 #ifdef _MIPS_CS_VENDOR
7414 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
7415 #endif
7418 static int
7419 conv_confstr_confname(PyObject *arg, int *valuep)
7421 return conv_confname(arg, valuep, posix_constants_confstr,
7422 sizeof(posix_constants_confstr)
7423 / sizeof(struct constdef));
7426 PyDoc_STRVAR(posix_confstr__doc__,
7427 "confstr(name) -> string\n\n\
7428 Return a string-valued system configuration variable.");
7430 static PyObject *
7431 posix_confstr(PyObject *self, PyObject *args)
7433 PyObject *result = NULL;
7434 int name;
7435 char buffer[256];
7437 if (PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name)) {
7438 int len;
7440 errno = 0;
7441 len = confstr(name, buffer, sizeof(buffer));
7442 if (len == 0) {
7443 if (errno) {
7444 posix_error();
7446 else {
7447 result = Py_None;
7448 Py_INCREF(Py_None);
7451 else {
7452 if ((unsigned int)len >= sizeof(buffer)) {
7453 result = PyString_FromStringAndSize(NULL, len-1);
7454 if (result != NULL)
7455 confstr(name, PyString_AS_STRING(result), len);
7457 else
7458 result = PyString_FromStringAndSize(buffer, len-1);
7461 return result;
7463 #endif
7466 #ifdef HAVE_SYSCONF
7467 static struct constdef posix_constants_sysconf[] = {
7468 #ifdef _SC_2_CHAR_TERM
7469 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
7470 #endif
7471 #ifdef _SC_2_C_BIND
7472 {"SC_2_C_BIND", _SC_2_C_BIND},
7473 #endif
7474 #ifdef _SC_2_C_DEV
7475 {"SC_2_C_DEV", _SC_2_C_DEV},
7476 #endif
7477 #ifdef _SC_2_C_VERSION
7478 {"SC_2_C_VERSION", _SC_2_C_VERSION},
7479 #endif
7480 #ifdef _SC_2_FORT_DEV
7481 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
7482 #endif
7483 #ifdef _SC_2_FORT_RUN
7484 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
7485 #endif
7486 #ifdef _SC_2_LOCALEDEF
7487 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
7488 #endif
7489 #ifdef _SC_2_SW_DEV
7490 {"SC_2_SW_DEV", _SC_2_SW_DEV},
7491 #endif
7492 #ifdef _SC_2_UPE
7493 {"SC_2_UPE", _SC_2_UPE},
7494 #endif
7495 #ifdef _SC_2_VERSION
7496 {"SC_2_VERSION", _SC_2_VERSION},
7497 #endif
7498 #ifdef _SC_ABI_ASYNCHRONOUS_IO
7499 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
7500 #endif
7501 #ifdef _SC_ACL
7502 {"SC_ACL", _SC_ACL},
7503 #endif
7504 #ifdef _SC_AIO_LISTIO_MAX
7505 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
7506 #endif
7507 #ifdef _SC_AIO_MAX
7508 {"SC_AIO_MAX", _SC_AIO_MAX},
7509 #endif
7510 #ifdef _SC_AIO_PRIO_DELTA_MAX
7511 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
7512 #endif
7513 #ifdef _SC_ARG_MAX
7514 {"SC_ARG_MAX", _SC_ARG_MAX},
7515 #endif
7516 #ifdef _SC_ASYNCHRONOUS_IO
7517 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
7518 #endif
7519 #ifdef _SC_ATEXIT_MAX
7520 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
7521 #endif
7522 #ifdef _SC_AUDIT
7523 {"SC_AUDIT", _SC_AUDIT},
7524 #endif
7525 #ifdef _SC_AVPHYS_PAGES
7526 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
7527 #endif
7528 #ifdef _SC_BC_BASE_MAX
7529 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
7530 #endif
7531 #ifdef _SC_BC_DIM_MAX
7532 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
7533 #endif
7534 #ifdef _SC_BC_SCALE_MAX
7535 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
7536 #endif
7537 #ifdef _SC_BC_STRING_MAX
7538 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
7539 #endif
7540 #ifdef _SC_CAP
7541 {"SC_CAP", _SC_CAP},
7542 #endif
7543 #ifdef _SC_CHARCLASS_NAME_MAX
7544 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
7545 #endif
7546 #ifdef _SC_CHAR_BIT
7547 {"SC_CHAR_BIT", _SC_CHAR_BIT},
7548 #endif
7549 #ifdef _SC_CHAR_MAX
7550 {"SC_CHAR_MAX", _SC_CHAR_MAX},
7551 #endif
7552 #ifdef _SC_CHAR_MIN
7553 {"SC_CHAR_MIN", _SC_CHAR_MIN},
7554 #endif
7555 #ifdef _SC_CHILD_MAX
7556 {"SC_CHILD_MAX", _SC_CHILD_MAX},
7557 #endif
7558 #ifdef _SC_CLK_TCK
7559 {"SC_CLK_TCK", _SC_CLK_TCK},
7560 #endif
7561 #ifdef _SC_COHER_BLKSZ
7562 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
7563 #endif
7564 #ifdef _SC_COLL_WEIGHTS_MAX
7565 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
7566 #endif
7567 #ifdef _SC_DCACHE_ASSOC
7568 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
7569 #endif
7570 #ifdef _SC_DCACHE_BLKSZ
7571 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
7572 #endif
7573 #ifdef _SC_DCACHE_LINESZ
7574 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
7575 #endif
7576 #ifdef _SC_DCACHE_SZ
7577 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
7578 #endif
7579 #ifdef _SC_DCACHE_TBLKSZ
7580 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
7581 #endif
7582 #ifdef _SC_DELAYTIMER_MAX
7583 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
7584 #endif
7585 #ifdef _SC_EQUIV_CLASS_MAX
7586 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
7587 #endif
7588 #ifdef _SC_EXPR_NEST_MAX
7589 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
7590 #endif
7591 #ifdef _SC_FSYNC
7592 {"SC_FSYNC", _SC_FSYNC},
7593 #endif
7594 #ifdef _SC_GETGR_R_SIZE_MAX
7595 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
7596 #endif
7597 #ifdef _SC_GETPW_R_SIZE_MAX
7598 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
7599 #endif
7600 #ifdef _SC_ICACHE_ASSOC
7601 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
7602 #endif
7603 #ifdef _SC_ICACHE_BLKSZ
7604 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
7605 #endif
7606 #ifdef _SC_ICACHE_LINESZ
7607 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
7608 #endif
7609 #ifdef _SC_ICACHE_SZ
7610 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
7611 #endif
7612 #ifdef _SC_INF
7613 {"SC_INF", _SC_INF},
7614 #endif
7615 #ifdef _SC_INT_MAX
7616 {"SC_INT_MAX", _SC_INT_MAX},
7617 #endif
7618 #ifdef _SC_INT_MIN
7619 {"SC_INT_MIN", _SC_INT_MIN},
7620 #endif
7621 #ifdef _SC_IOV_MAX
7622 {"SC_IOV_MAX", _SC_IOV_MAX},
7623 #endif
7624 #ifdef _SC_IP_SECOPTS
7625 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
7626 #endif
7627 #ifdef _SC_JOB_CONTROL
7628 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
7629 #endif
7630 #ifdef _SC_KERN_POINTERS
7631 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
7632 #endif
7633 #ifdef _SC_KERN_SIM
7634 {"SC_KERN_SIM", _SC_KERN_SIM},
7635 #endif
7636 #ifdef _SC_LINE_MAX
7637 {"SC_LINE_MAX", _SC_LINE_MAX},
7638 #endif
7639 #ifdef _SC_LOGIN_NAME_MAX
7640 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
7641 #endif
7642 #ifdef _SC_LOGNAME_MAX
7643 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
7644 #endif
7645 #ifdef _SC_LONG_BIT
7646 {"SC_LONG_BIT", _SC_LONG_BIT},
7647 #endif
7648 #ifdef _SC_MAC
7649 {"SC_MAC", _SC_MAC},
7650 #endif
7651 #ifdef _SC_MAPPED_FILES
7652 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
7653 #endif
7654 #ifdef _SC_MAXPID
7655 {"SC_MAXPID", _SC_MAXPID},
7656 #endif
7657 #ifdef _SC_MB_LEN_MAX
7658 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
7659 #endif
7660 #ifdef _SC_MEMLOCK
7661 {"SC_MEMLOCK", _SC_MEMLOCK},
7662 #endif
7663 #ifdef _SC_MEMLOCK_RANGE
7664 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
7665 #endif
7666 #ifdef _SC_MEMORY_PROTECTION
7667 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
7668 #endif
7669 #ifdef _SC_MESSAGE_PASSING
7670 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
7671 #endif
7672 #ifdef _SC_MMAP_FIXED_ALIGNMENT
7673 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
7674 #endif
7675 #ifdef _SC_MQ_OPEN_MAX
7676 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
7677 #endif
7678 #ifdef _SC_MQ_PRIO_MAX
7679 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
7680 #endif
7681 #ifdef _SC_NACLS_MAX
7682 {"SC_NACLS_MAX", _SC_NACLS_MAX},
7683 #endif
7684 #ifdef _SC_NGROUPS_MAX
7685 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
7686 #endif
7687 #ifdef _SC_NL_ARGMAX
7688 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
7689 #endif
7690 #ifdef _SC_NL_LANGMAX
7691 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
7692 #endif
7693 #ifdef _SC_NL_MSGMAX
7694 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
7695 #endif
7696 #ifdef _SC_NL_NMAX
7697 {"SC_NL_NMAX", _SC_NL_NMAX},
7698 #endif
7699 #ifdef _SC_NL_SETMAX
7700 {"SC_NL_SETMAX", _SC_NL_SETMAX},
7701 #endif
7702 #ifdef _SC_NL_TEXTMAX
7703 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
7704 #endif
7705 #ifdef _SC_NPROCESSORS_CONF
7706 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
7707 #endif
7708 #ifdef _SC_NPROCESSORS_ONLN
7709 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
7710 #endif
7711 #ifdef _SC_NPROC_CONF
7712 {"SC_NPROC_CONF", _SC_NPROC_CONF},
7713 #endif
7714 #ifdef _SC_NPROC_ONLN
7715 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
7716 #endif
7717 #ifdef _SC_NZERO
7718 {"SC_NZERO", _SC_NZERO},
7719 #endif
7720 #ifdef _SC_OPEN_MAX
7721 {"SC_OPEN_MAX", _SC_OPEN_MAX},
7722 #endif
7723 #ifdef _SC_PAGESIZE
7724 {"SC_PAGESIZE", _SC_PAGESIZE},
7725 #endif
7726 #ifdef _SC_PAGE_SIZE
7727 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
7728 #endif
7729 #ifdef _SC_PASS_MAX
7730 {"SC_PASS_MAX", _SC_PASS_MAX},
7731 #endif
7732 #ifdef _SC_PHYS_PAGES
7733 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
7734 #endif
7735 #ifdef _SC_PII
7736 {"SC_PII", _SC_PII},
7737 #endif
7738 #ifdef _SC_PII_INTERNET
7739 {"SC_PII_INTERNET", _SC_PII_INTERNET},
7740 #endif
7741 #ifdef _SC_PII_INTERNET_DGRAM
7742 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
7743 #endif
7744 #ifdef _SC_PII_INTERNET_STREAM
7745 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
7746 #endif
7747 #ifdef _SC_PII_OSI
7748 {"SC_PII_OSI", _SC_PII_OSI},
7749 #endif
7750 #ifdef _SC_PII_OSI_CLTS
7751 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
7752 #endif
7753 #ifdef _SC_PII_OSI_COTS
7754 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
7755 #endif
7756 #ifdef _SC_PII_OSI_M
7757 {"SC_PII_OSI_M", _SC_PII_OSI_M},
7758 #endif
7759 #ifdef _SC_PII_SOCKET
7760 {"SC_PII_SOCKET", _SC_PII_SOCKET},
7761 #endif
7762 #ifdef _SC_PII_XTI
7763 {"SC_PII_XTI", _SC_PII_XTI},
7764 #endif
7765 #ifdef _SC_POLL
7766 {"SC_POLL", _SC_POLL},
7767 #endif
7768 #ifdef _SC_PRIORITIZED_IO
7769 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
7770 #endif
7771 #ifdef _SC_PRIORITY_SCHEDULING
7772 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
7773 #endif
7774 #ifdef _SC_REALTIME_SIGNALS
7775 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
7776 #endif
7777 #ifdef _SC_RE_DUP_MAX
7778 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
7779 #endif
7780 #ifdef _SC_RTSIG_MAX
7781 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
7782 #endif
7783 #ifdef _SC_SAVED_IDS
7784 {"SC_SAVED_IDS", _SC_SAVED_IDS},
7785 #endif
7786 #ifdef _SC_SCHAR_MAX
7787 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
7788 #endif
7789 #ifdef _SC_SCHAR_MIN
7790 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
7791 #endif
7792 #ifdef _SC_SELECT
7793 {"SC_SELECT", _SC_SELECT},
7794 #endif
7795 #ifdef _SC_SEMAPHORES
7796 {"SC_SEMAPHORES", _SC_SEMAPHORES},
7797 #endif
7798 #ifdef _SC_SEM_NSEMS_MAX
7799 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
7800 #endif
7801 #ifdef _SC_SEM_VALUE_MAX
7802 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
7803 #endif
7804 #ifdef _SC_SHARED_MEMORY_OBJECTS
7805 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
7806 #endif
7807 #ifdef _SC_SHRT_MAX
7808 {"SC_SHRT_MAX", _SC_SHRT_MAX},
7809 #endif
7810 #ifdef _SC_SHRT_MIN
7811 {"SC_SHRT_MIN", _SC_SHRT_MIN},
7812 #endif
7813 #ifdef _SC_SIGQUEUE_MAX
7814 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
7815 #endif
7816 #ifdef _SC_SIGRT_MAX
7817 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
7818 #endif
7819 #ifdef _SC_SIGRT_MIN
7820 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
7821 #endif
7822 #ifdef _SC_SOFTPOWER
7823 {"SC_SOFTPOWER", _SC_SOFTPOWER},
7824 #endif
7825 #ifdef _SC_SPLIT_CACHE
7826 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
7827 #endif
7828 #ifdef _SC_SSIZE_MAX
7829 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
7830 #endif
7831 #ifdef _SC_STACK_PROT
7832 {"SC_STACK_PROT", _SC_STACK_PROT},
7833 #endif
7834 #ifdef _SC_STREAM_MAX
7835 {"SC_STREAM_MAX", _SC_STREAM_MAX},
7836 #endif
7837 #ifdef _SC_SYNCHRONIZED_IO
7838 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
7839 #endif
7840 #ifdef _SC_THREADS
7841 {"SC_THREADS", _SC_THREADS},
7842 #endif
7843 #ifdef _SC_THREAD_ATTR_STACKADDR
7844 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
7845 #endif
7846 #ifdef _SC_THREAD_ATTR_STACKSIZE
7847 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
7848 #endif
7849 #ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
7850 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
7851 #endif
7852 #ifdef _SC_THREAD_KEYS_MAX
7853 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
7854 #endif
7855 #ifdef _SC_THREAD_PRIORITY_SCHEDULING
7856 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
7857 #endif
7858 #ifdef _SC_THREAD_PRIO_INHERIT
7859 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
7860 #endif
7861 #ifdef _SC_THREAD_PRIO_PROTECT
7862 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
7863 #endif
7864 #ifdef _SC_THREAD_PROCESS_SHARED
7865 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
7866 #endif
7867 #ifdef _SC_THREAD_SAFE_FUNCTIONS
7868 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
7869 #endif
7870 #ifdef _SC_THREAD_STACK_MIN
7871 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
7872 #endif
7873 #ifdef _SC_THREAD_THREADS_MAX
7874 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
7875 #endif
7876 #ifdef _SC_TIMERS
7877 {"SC_TIMERS", _SC_TIMERS},
7878 #endif
7879 #ifdef _SC_TIMER_MAX
7880 {"SC_TIMER_MAX", _SC_TIMER_MAX},
7881 #endif
7882 #ifdef _SC_TTY_NAME_MAX
7883 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
7884 #endif
7885 #ifdef _SC_TZNAME_MAX
7886 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
7887 #endif
7888 #ifdef _SC_T_IOV_MAX
7889 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
7890 #endif
7891 #ifdef _SC_UCHAR_MAX
7892 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
7893 #endif
7894 #ifdef _SC_UINT_MAX
7895 {"SC_UINT_MAX", _SC_UINT_MAX},
7896 #endif
7897 #ifdef _SC_UIO_MAXIOV
7898 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
7899 #endif
7900 #ifdef _SC_ULONG_MAX
7901 {"SC_ULONG_MAX", _SC_ULONG_MAX},
7902 #endif
7903 #ifdef _SC_USHRT_MAX
7904 {"SC_USHRT_MAX", _SC_USHRT_MAX},
7905 #endif
7906 #ifdef _SC_VERSION
7907 {"SC_VERSION", _SC_VERSION},
7908 #endif
7909 #ifdef _SC_WORD_BIT
7910 {"SC_WORD_BIT", _SC_WORD_BIT},
7911 #endif
7912 #ifdef _SC_XBS5_ILP32_OFF32
7913 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
7914 #endif
7915 #ifdef _SC_XBS5_ILP32_OFFBIG
7916 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
7917 #endif
7918 #ifdef _SC_XBS5_LP64_OFF64
7919 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
7920 #endif
7921 #ifdef _SC_XBS5_LPBIG_OFFBIG
7922 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
7923 #endif
7924 #ifdef _SC_XOPEN_CRYPT
7925 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
7926 #endif
7927 #ifdef _SC_XOPEN_ENH_I18N
7928 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
7929 #endif
7930 #ifdef _SC_XOPEN_LEGACY
7931 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
7932 #endif
7933 #ifdef _SC_XOPEN_REALTIME
7934 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
7935 #endif
7936 #ifdef _SC_XOPEN_REALTIME_THREADS
7937 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
7938 #endif
7939 #ifdef _SC_XOPEN_SHM
7940 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
7941 #endif
7942 #ifdef _SC_XOPEN_UNIX
7943 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
7944 #endif
7945 #ifdef _SC_XOPEN_VERSION
7946 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
7947 #endif
7948 #ifdef _SC_XOPEN_XCU_VERSION
7949 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
7950 #endif
7951 #ifdef _SC_XOPEN_XPG2
7952 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
7953 #endif
7954 #ifdef _SC_XOPEN_XPG3
7955 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
7956 #endif
7957 #ifdef _SC_XOPEN_XPG4
7958 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
7959 #endif
7962 static int
7963 conv_sysconf_confname(PyObject *arg, int *valuep)
7965 return conv_confname(arg, valuep, posix_constants_sysconf,
7966 sizeof(posix_constants_sysconf)
7967 / sizeof(struct constdef));
7970 PyDoc_STRVAR(posix_sysconf__doc__,
7971 "sysconf(name) -> integer\n\n\
7972 Return an integer-valued system configuration variable.");
7974 static PyObject *
7975 posix_sysconf(PyObject *self, PyObject *args)
7977 PyObject *result = NULL;
7978 int name;
7980 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
7981 int value;
7983 errno = 0;
7984 value = sysconf(name);
7985 if (value == -1 && errno != 0)
7986 posix_error();
7987 else
7988 result = PyInt_FromLong(value);
7990 return result;
7992 #endif
7995 /* This code is used to ensure that the tables of configuration value names
7996 * are in sorted order as required by conv_confname(), and also to build the
7997 * the exported dictionaries that are used to publish information about the
7998 * names available on the host platform.
8000 * Sorting the table at runtime ensures that the table is properly ordered
8001 * when used, even for platforms we're not able to test on. It also makes
8002 * it easier to add additional entries to the tables.
8005 static int
8006 cmp_constdefs(const void *v1, const void *v2)
8008 const struct constdef *c1 =
8009 (const struct constdef *) v1;
8010 const struct constdef *c2 =
8011 (const struct constdef *) v2;
8013 return strcmp(c1->name, c2->name);
8016 static int
8017 setup_confname_table(struct constdef *table, size_t tablesize,
8018 char *tablename, PyObject *module)
8020 PyObject *d = NULL;
8021 size_t i;
8023 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
8024 d = PyDict_New();
8025 if (d == NULL)
8026 return -1;
8028 for (i=0; i < tablesize; ++i) {
8029 PyObject *o = PyInt_FromLong(table[i].value);
8030 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
8031 Py_XDECREF(o);
8032 Py_DECREF(d);
8033 return -1;
8035 Py_DECREF(o);
8037 return PyModule_AddObject(module, tablename, d);
8040 /* Return -1 on failure, 0 on success. */
8041 static int
8042 setup_confname_tables(PyObject *module)
8044 #if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
8045 if (setup_confname_table(posix_constants_pathconf,
8046 sizeof(posix_constants_pathconf)
8047 / sizeof(struct constdef),
8048 "pathconf_names", module))
8049 return -1;
8050 #endif
8051 #ifdef HAVE_CONFSTR
8052 if (setup_confname_table(posix_constants_confstr,
8053 sizeof(posix_constants_confstr)
8054 / sizeof(struct constdef),
8055 "confstr_names", module))
8056 return -1;
8057 #endif
8058 #ifdef HAVE_SYSCONF
8059 if (setup_confname_table(posix_constants_sysconf,
8060 sizeof(posix_constants_sysconf)
8061 / sizeof(struct constdef),
8062 "sysconf_names", module))
8063 return -1;
8064 #endif
8065 return 0;
8069 PyDoc_STRVAR(posix_abort__doc__,
8070 "abort() -> does not return!\n\n\
8071 Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
8072 in the hardest way possible on the hosting operating system.");
8074 static PyObject *
8075 posix_abort(PyObject *self, PyObject *noargs)
8077 abort();
8078 /*NOTREACHED*/
8079 Py_FatalError("abort() called from Python code didn't abort!");
8080 return NULL;
8083 #ifdef MS_WINDOWS
8084 PyDoc_STRVAR(win32_startfile__doc__,
8085 "startfile(filepath [, operation]) - Start a file with its associated\n\
8086 application.\n\
8088 When \"operation\" is not specified or \"open\", this acts like\n\
8089 double-clicking the file in Explorer, or giving the file name as an\n\
8090 argument to the DOS \"start\" command: the file is opened with whatever\n\
8091 application (if any) its extension is associated.\n\
8092 When another \"operation\" is given, it specifies what should be done with\n\
8093 the file. A typical operation is \"print\".\n\
8095 startfile returns as soon as the associated application is launched.\n\
8096 There is no option to wait for the application to close, and no way\n\
8097 to retrieve the application's exit status.\n\
8099 The filepath is relative to the current directory. If you want to use\n\
8100 an absolute path, make sure the first character is not a slash (\"/\");\n\
8101 the underlying Win32 ShellExecute function doesn't work if it is.");
8103 static PyObject *
8104 win32_startfile(PyObject *self, PyObject *args)
8106 char *filepath;
8107 char *operation = NULL;
8108 HINSTANCE rc;
8109 #ifdef Py_WIN_WIDE_FILENAMES
8110 if (unicode_file_names()) {
8111 PyObject *unipath, *woperation = NULL;
8112 if (!PyArg_ParseTuple(args, "U|s:startfile",
8113 &unipath, &operation)) {
8114 PyErr_Clear();
8115 goto normal;
8119 if (operation) {
8120 woperation = PyUnicode_DecodeASCII(operation,
8121 strlen(operation), NULL);
8122 if (!woperation) {
8123 PyErr_Clear();
8124 operation = NULL;
8125 goto normal;
8129 Py_BEGIN_ALLOW_THREADS
8130 rc = ShellExecuteW((HWND)0, woperation ? PyUnicode_AS_UNICODE(woperation) : 0,
8131 PyUnicode_AS_UNICODE(unipath),
8132 NULL, NULL, SW_SHOWNORMAL);
8133 Py_END_ALLOW_THREADS
8135 Py_XDECREF(woperation);
8136 if (rc <= (HINSTANCE)32) {
8137 PyObject *errval = win32_error_unicode("startfile",
8138 PyUnicode_AS_UNICODE(unipath));
8139 return errval;
8141 Py_INCREF(Py_None);
8142 return Py_None;
8144 #endif
8146 normal:
8147 if (!PyArg_ParseTuple(args, "et|s:startfile",
8148 Py_FileSystemDefaultEncoding, &filepath,
8149 &operation))
8150 return NULL;
8151 Py_BEGIN_ALLOW_THREADS
8152 rc = ShellExecute((HWND)0, operation, filepath,
8153 NULL, NULL, SW_SHOWNORMAL);
8154 Py_END_ALLOW_THREADS
8155 if (rc <= (HINSTANCE)32) {
8156 PyObject *errval = win32_error("startfile", filepath);
8157 PyMem_Free(filepath);
8158 return errval;
8160 PyMem_Free(filepath);
8161 Py_INCREF(Py_None);
8162 return Py_None;
8164 #endif
8166 #ifdef HAVE_GETLOADAVG
8167 PyDoc_STRVAR(posix_getloadavg__doc__,
8168 "getloadavg() -> (float, float, float)\n\n\
8169 Return the number of processes in the system run queue averaged over\n\
8170 the last 1, 5, and 15 minutes or raises OSError if the load average\n\
8171 was unobtainable");
8173 static PyObject *
8174 posix_getloadavg(PyObject *self, PyObject *noargs)
8176 double loadavg[3];
8177 if (getloadavg(loadavg, 3)!=3) {
8178 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
8179 return NULL;
8180 } else
8181 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
8183 #endif
8185 #ifdef MS_WINDOWS
8187 PyDoc_STRVAR(win32_urandom__doc__,
8188 "urandom(n) -> str\n\n\
8189 Return a string of n random bytes suitable for cryptographic use.");
8191 typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
8192 LPCSTR pszContainer, LPCSTR pszProvider, DWORD dwProvType,\
8193 DWORD dwFlags );
8194 typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV hProv, DWORD dwLen,\
8195 BYTE *pbBuffer );
8197 static CRYPTGENRANDOM pCryptGenRandom = NULL;
8198 /* This handle is never explicitly released. Instead, the operating
8199 system will release it when the process terminates. */
8200 static HCRYPTPROV hCryptProv = 0;
8202 static PyObject*
8203 win32_urandom(PyObject *self, PyObject *args)
8205 int howMany;
8206 PyObject* result;
8208 /* Read arguments */
8209 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
8210 return NULL;
8211 if (howMany < 0)
8212 return PyErr_Format(PyExc_ValueError,
8213 "negative argument not allowed");
8215 if (hCryptProv == 0) {
8216 HINSTANCE hAdvAPI32 = NULL;
8217 CRYPTACQUIRECONTEXTA pCryptAcquireContext = NULL;
8219 /* Obtain handle to the DLL containing CryptoAPI
8220 This should not fail */
8221 hAdvAPI32 = GetModuleHandle("advapi32.dll");
8222 if(hAdvAPI32 == NULL)
8223 return win32_error("GetModuleHandle", NULL);
8225 /* Obtain pointers to the CryptoAPI functions
8226 This will fail on some early versions of Win95 */
8227 pCryptAcquireContext = (CRYPTACQUIRECONTEXTA)GetProcAddress(
8228 hAdvAPI32,
8229 "CryptAcquireContextA");
8230 if (pCryptAcquireContext == NULL)
8231 return PyErr_Format(PyExc_NotImplementedError,
8232 "CryptAcquireContextA not found");
8234 pCryptGenRandom = (CRYPTGENRANDOM)GetProcAddress(
8235 hAdvAPI32, "CryptGenRandom");
8236 if (pCryptGenRandom == NULL)
8237 return PyErr_Format(PyExc_NotImplementedError,
8238 "CryptGenRandom not found");
8240 /* Acquire context */
8241 if (! pCryptAcquireContext(&hCryptProv, NULL, NULL,
8242 PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
8243 return win32_error("CryptAcquireContext", NULL);
8246 /* Allocate bytes */
8247 result = PyString_FromStringAndSize(NULL, howMany);
8248 if (result != NULL) {
8249 /* Get random data */
8250 if (! pCryptGenRandom(hCryptProv, howMany, (unsigned char*)
8251 PyString_AS_STRING(result))) {
8252 Py_DECREF(result);
8253 return win32_error("CryptGenRandom", NULL);
8256 return result;
8258 #endif
8260 #ifdef __VMS
8261 /* Use openssl random routine */
8262 #include <openssl/rand.h>
8263 PyDoc_STRVAR(vms_urandom__doc__,
8264 "urandom(n) -> str\n\n\
8265 Return a string of n random bytes suitable for cryptographic use.");
8267 static PyObject*
8268 vms_urandom(PyObject *self, PyObject *args)
8270 int howMany;
8271 PyObject* result;
8273 /* Read arguments */
8274 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
8275 return NULL;
8276 if (howMany < 0)
8277 return PyErr_Format(PyExc_ValueError,
8278 "negative argument not allowed");
8280 /* Allocate bytes */
8281 result = PyString_FromStringAndSize(NULL, howMany);
8282 if (result != NULL) {
8283 /* Get random data */
8284 if (RAND_pseudo_bytes((unsigned char*)
8285 PyString_AS_STRING(result),
8286 howMany) < 0) {
8287 Py_DECREF(result);
8288 return PyErr_Format(PyExc_ValueError,
8289 "RAND_pseudo_bytes");
8292 return result;
8294 #endif
8296 static PyMethodDef posix_methods[] = {
8297 {"access", posix_access, METH_VARARGS, posix_access__doc__},
8298 #ifdef HAVE_TTYNAME
8299 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
8300 #endif
8301 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
8302 #ifdef HAVE_CHFLAGS
8303 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
8304 #endif /* HAVE_CHFLAGS */
8305 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
8306 #ifdef HAVE_FCHMOD
8307 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
8308 #endif /* HAVE_FCHMOD */
8309 #ifdef HAVE_CHOWN
8310 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
8311 #endif /* HAVE_CHOWN */
8312 #ifdef HAVE_LCHMOD
8313 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
8314 #endif /* HAVE_LCHMOD */
8315 #ifdef HAVE_FCHOWN
8316 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
8317 #endif /* HAVE_FCHOWN */
8318 #ifdef HAVE_LCHFLAGS
8319 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
8320 #endif /* HAVE_LCHFLAGS */
8321 #ifdef HAVE_LCHOWN
8322 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
8323 #endif /* HAVE_LCHOWN */
8324 #ifdef HAVE_CHROOT
8325 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
8326 #endif
8327 #ifdef HAVE_CTERMID
8328 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
8329 #endif
8330 #ifdef HAVE_GETCWD
8331 {"getcwd", posix_getcwd, METH_NOARGS, posix_getcwd__doc__},
8332 #ifdef Py_USING_UNICODE
8333 {"getcwdu", posix_getcwdu, METH_NOARGS, posix_getcwdu__doc__},
8334 #endif
8335 #endif
8336 #ifdef HAVE_LINK
8337 {"link", posix_link, METH_VARARGS, posix_link__doc__},
8338 #endif /* HAVE_LINK */
8339 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
8340 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
8341 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
8342 #ifdef HAVE_NICE
8343 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
8344 #endif /* HAVE_NICE */
8345 #ifdef HAVE_READLINK
8346 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
8347 #endif /* HAVE_READLINK */
8348 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
8349 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
8350 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
8351 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
8352 #ifdef HAVE_SYMLINK
8353 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
8354 #endif /* HAVE_SYMLINK */
8355 #ifdef HAVE_SYSTEM
8356 {"system", posix_system, METH_VARARGS, posix_system__doc__},
8357 #endif
8358 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
8359 #ifdef HAVE_UNAME
8360 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
8361 #endif /* HAVE_UNAME */
8362 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
8363 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
8364 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
8365 #ifdef HAVE_TIMES
8366 {"times", posix_times, METH_NOARGS, posix_times__doc__},
8367 #endif /* HAVE_TIMES */
8368 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
8369 #ifdef HAVE_EXECV
8370 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
8371 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
8372 #endif /* HAVE_EXECV */
8373 #ifdef HAVE_SPAWNV
8374 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
8375 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
8376 #if defined(PYOS_OS2)
8377 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
8378 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
8379 #endif /* PYOS_OS2 */
8380 #endif /* HAVE_SPAWNV */
8381 #ifdef HAVE_FORK1
8382 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
8383 #endif /* HAVE_FORK1 */
8384 #ifdef HAVE_FORK
8385 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
8386 #endif /* HAVE_FORK */
8387 #if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
8388 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
8389 #endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
8390 #ifdef HAVE_FORKPTY
8391 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
8392 #endif /* HAVE_FORKPTY */
8393 #ifdef HAVE_GETEGID
8394 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
8395 #endif /* HAVE_GETEGID */
8396 #ifdef HAVE_GETEUID
8397 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
8398 #endif /* HAVE_GETEUID */
8399 #ifdef HAVE_GETGID
8400 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
8401 #endif /* HAVE_GETGID */
8402 #ifdef HAVE_GETGROUPS
8403 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
8404 #endif
8405 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
8406 #ifdef HAVE_GETPGRP
8407 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
8408 #endif /* HAVE_GETPGRP */
8409 #ifdef HAVE_GETPPID
8410 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
8411 #endif /* HAVE_GETPPID */
8412 #ifdef HAVE_GETUID
8413 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
8414 #endif /* HAVE_GETUID */
8415 #ifdef HAVE_GETLOGIN
8416 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
8417 #endif
8418 #ifdef HAVE_KILL
8419 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
8420 #endif /* HAVE_KILL */
8421 #ifdef HAVE_KILLPG
8422 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
8423 #endif /* HAVE_KILLPG */
8424 #ifdef HAVE_PLOCK
8425 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
8426 #endif /* HAVE_PLOCK */
8427 #ifdef HAVE_POPEN
8428 {"popen", posix_popen, METH_VARARGS, posix_popen__doc__},
8429 #ifdef MS_WINDOWS
8430 {"popen2", win32_popen2, METH_VARARGS},
8431 {"popen3", win32_popen3, METH_VARARGS},
8432 {"popen4", win32_popen4, METH_VARARGS},
8433 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
8434 #else
8435 #if defined(PYOS_OS2) && defined(PYCC_GCC)
8436 {"popen2", os2emx_popen2, METH_VARARGS},
8437 {"popen3", os2emx_popen3, METH_VARARGS},
8438 {"popen4", os2emx_popen4, METH_VARARGS},
8439 #endif
8440 #endif
8441 #endif /* HAVE_POPEN */
8442 #ifdef HAVE_SETUID
8443 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
8444 #endif /* HAVE_SETUID */
8445 #ifdef HAVE_SETEUID
8446 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
8447 #endif /* HAVE_SETEUID */
8448 #ifdef HAVE_SETEGID
8449 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
8450 #endif /* HAVE_SETEGID */
8451 #ifdef HAVE_SETREUID
8452 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
8453 #endif /* HAVE_SETREUID */
8454 #ifdef HAVE_SETREGID
8455 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
8456 #endif /* HAVE_SETREGID */
8457 #ifdef HAVE_SETGID
8458 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
8459 #endif /* HAVE_SETGID */
8460 #ifdef HAVE_SETGROUPS
8461 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
8462 #endif /* HAVE_SETGROUPS */
8463 #ifdef HAVE_GETPGID
8464 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
8465 #endif /* HAVE_GETPGID */
8466 #ifdef HAVE_SETPGRP
8467 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
8468 #endif /* HAVE_SETPGRP */
8469 #ifdef HAVE_WAIT
8470 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
8471 #endif /* HAVE_WAIT */
8472 #ifdef HAVE_WAIT3
8473 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
8474 #endif /* HAVE_WAIT3 */
8475 #ifdef HAVE_WAIT4
8476 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
8477 #endif /* HAVE_WAIT4 */
8478 #if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
8479 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
8480 #endif /* HAVE_WAITPID */
8481 #ifdef HAVE_GETSID
8482 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
8483 #endif /* HAVE_GETSID */
8484 #ifdef HAVE_SETSID
8485 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
8486 #endif /* HAVE_SETSID */
8487 #ifdef HAVE_SETPGID
8488 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
8489 #endif /* HAVE_SETPGID */
8490 #ifdef HAVE_TCGETPGRP
8491 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
8492 #endif /* HAVE_TCGETPGRP */
8493 #ifdef HAVE_TCSETPGRP
8494 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
8495 #endif /* HAVE_TCSETPGRP */
8496 {"open", posix_open, METH_VARARGS, posix_open__doc__},
8497 {"close", posix_close, METH_VARARGS, posix_close__doc__},
8498 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
8499 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
8500 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
8501 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
8502 {"read", posix_read, METH_VARARGS, posix_read__doc__},
8503 {"write", posix_write, METH_VARARGS, posix_write__doc__},
8504 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
8505 {"fdopen", posix_fdopen, METH_VARARGS, posix_fdopen__doc__},
8506 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
8507 #ifdef HAVE_PIPE
8508 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
8509 #endif
8510 #ifdef HAVE_MKFIFO
8511 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
8512 #endif
8513 #if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
8514 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
8515 #endif
8516 #ifdef HAVE_DEVICE_MACROS
8517 {"major", posix_major, METH_VARARGS, posix_major__doc__},
8518 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
8519 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
8520 #endif
8521 #ifdef HAVE_FTRUNCATE
8522 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
8523 #endif
8524 #ifdef HAVE_PUTENV
8525 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
8526 #endif
8527 #ifdef HAVE_UNSETENV
8528 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
8529 #endif
8530 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
8531 #ifdef HAVE_FCHDIR
8532 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
8533 #endif
8534 #ifdef HAVE_FSYNC
8535 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
8536 #endif
8537 #ifdef HAVE_FDATASYNC
8538 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
8539 #endif
8540 #ifdef HAVE_SYS_WAIT_H
8541 #ifdef WCOREDUMP
8542 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
8543 #endif /* WCOREDUMP */
8544 #ifdef WIFCONTINUED
8545 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
8546 #endif /* WIFCONTINUED */
8547 #ifdef WIFSTOPPED
8548 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
8549 #endif /* WIFSTOPPED */
8550 #ifdef WIFSIGNALED
8551 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
8552 #endif /* WIFSIGNALED */
8553 #ifdef WIFEXITED
8554 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
8555 #endif /* WIFEXITED */
8556 #ifdef WEXITSTATUS
8557 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
8558 #endif /* WEXITSTATUS */
8559 #ifdef WTERMSIG
8560 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
8561 #endif /* WTERMSIG */
8562 #ifdef WSTOPSIG
8563 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
8564 #endif /* WSTOPSIG */
8565 #endif /* HAVE_SYS_WAIT_H */
8566 #if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
8567 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
8568 #endif
8569 #if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
8570 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
8571 #endif
8572 #ifdef HAVE_TMPFILE
8573 {"tmpfile", posix_tmpfile, METH_NOARGS, posix_tmpfile__doc__},
8574 #endif
8575 #ifdef HAVE_TEMPNAM
8576 {"tempnam", posix_tempnam, METH_VARARGS, posix_tempnam__doc__},
8577 #endif
8578 #ifdef HAVE_TMPNAM
8579 {"tmpnam", posix_tmpnam, METH_NOARGS, posix_tmpnam__doc__},
8580 #endif
8581 #ifdef HAVE_CONFSTR
8582 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
8583 #endif
8584 #ifdef HAVE_SYSCONF
8585 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
8586 #endif
8587 #ifdef HAVE_FPATHCONF
8588 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
8589 #endif
8590 #ifdef HAVE_PATHCONF
8591 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
8592 #endif
8593 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
8594 #ifdef MS_WINDOWS
8595 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
8596 #endif
8597 #ifdef HAVE_GETLOADAVG
8598 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
8599 #endif
8600 #ifdef MS_WINDOWS
8601 {"urandom", win32_urandom, METH_VARARGS, win32_urandom__doc__},
8602 #endif
8603 #ifdef __VMS
8604 {"urandom", vms_urandom, METH_VARARGS, vms_urandom__doc__},
8605 #endif
8606 {NULL, NULL} /* Sentinel */
8610 static int
8611 ins(PyObject *module, char *symbol, long value)
8613 return PyModule_AddIntConstant(module, symbol, value);
8616 #if defined(PYOS_OS2)
8617 /* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
8618 static int insertvalues(PyObject *module)
8620 APIRET rc;
8621 ULONG values[QSV_MAX+1];
8622 PyObject *v;
8623 char *ver, tmp[50];
8625 Py_BEGIN_ALLOW_THREADS
8626 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
8627 Py_END_ALLOW_THREADS
8629 if (rc != NO_ERROR) {
8630 os2_error(rc);
8631 return -1;
8634 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
8635 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
8636 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
8637 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
8638 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
8639 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
8640 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
8642 switch (values[QSV_VERSION_MINOR]) {
8643 case 0: ver = "2.00"; break;
8644 case 10: ver = "2.10"; break;
8645 case 11: ver = "2.11"; break;
8646 case 30: ver = "3.00"; break;
8647 case 40: ver = "4.00"; break;
8648 case 50: ver = "5.00"; break;
8649 default:
8650 PyOS_snprintf(tmp, sizeof(tmp),
8651 "%d-%d", values[QSV_VERSION_MAJOR],
8652 values[QSV_VERSION_MINOR]);
8653 ver = &tmp[0];
8656 /* Add Indicator of the Version of the Operating System */
8657 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
8658 return -1;
8660 /* Add Indicator of Which Drive was Used to Boot the System */
8661 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
8662 tmp[1] = ':';
8663 tmp[2] = '\0';
8665 return PyModule_AddStringConstant(module, "bootdrive", tmp);
8667 #endif
8669 static int
8670 all_ins(PyObject *d)
8672 #ifdef F_OK
8673 if (ins(d, "F_OK", (long)F_OK)) return -1;
8674 #endif
8675 #ifdef R_OK
8676 if (ins(d, "R_OK", (long)R_OK)) return -1;
8677 #endif
8678 #ifdef W_OK
8679 if (ins(d, "W_OK", (long)W_OK)) return -1;
8680 #endif
8681 #ifdef X_OK
8682 if (ins(d, "X_OK", (long)X_OK)) return -1;
8683 #endif
8684 #ifdef NGROUPS_MAX
8685 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
8686 #endif
8687 #ifdef TMP_MAX
8688 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
8689 #endif
8690 #ifdef WCONTINUED
8691 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
8692 #endif
8693 #ifdef WNOHANG
8694 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
8695 #endif
8696 #ifdef WUNTRACED
8697 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
8698 #endif
8699 #ifdef O_RDONLY
8700 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
8701 #endif
8702 #ifdef O_WRONLY
8703 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
8704 #endif
8705 #ifdef O_RDWR
8706 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
8707 #endif
8708 #ifdef O_NDELAY
8709 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
8710 #endif
8711 #ifdef O_NONBLOCK
8712 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
8713 #endif
8714 #ifdef O_APPEND
8715 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
8716 #endif
8717 #ifdef O_DSYNC
8718 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
8719 #endif
8720 #ifdef O_RSYNC
8721 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
8722 #endif
8723 #ifdef O_SYNC
8724 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
8725 #endif
8726 #ifdef O_NOCTTY
8727 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
8728 #endif
8729 #ifdef O_CREAT
8730 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
8731 #endif
8732 #ifdef O_EXCL
8733 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
8734 #endif
8735 #ifdef O_TRUNC
8736 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
8737 #endif
8738 #ifdef O_BINARY
8739 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
8740 #endif
8741 #ifdef O_TEXT
8742 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
8743 #endif
8744 #ifdef O_LARGEFILE
8745 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
8746 #endif
8747 #ifdef O_SHLOCK
8748 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
8749 #endif
8750 #ifdef O_EXLOCK
8751 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
8752 #endif
8754 /* MS Windows */
8755 #ifdef O_NOINHERIT
8756 /* Don't inherit in child processes. */
8757 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
8758 #endif
8759 #ifdef _O_SHORT_LIVED
8760 /* Optimize for short life (keep in memory). */
8761 /* MS forgot to define this one with a non-underscore form too. */
8762 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
8763 #endif
8764 #ifdef O_TEMPORARY
8765 /* Automatically delete when last handle is closed. */
8766 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
8767 #endif
8768 #ifdef O_RANDOM
8769 /* Optimize for random access. */
8770 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
8771 #endif
8772 #ifdef O_SEQUENTIAL
8773 /* Optimize for sequential access. */
8774 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
8775 #endif
8777 /* GNU extensions. */
8778 #ifdef O_ASYNC
8779 /* Send a SIGIO signal whenever input or output
8780 becomes available on file descriptor */
8781 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
8782 #endif
8783 #ifdef O_DIRECT
8784 /* Direct disk access. */
8785 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
8786 #endif
8787 #ifdef O_DIRECTORY
8788 /* Must be a directory. */
8789 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
8790 #endif
8791 #ifdef O_NOFOLLOW
8792 /* Do not follow links. */
8793 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
8794 #endif
8795 #ifdef O_NOATIME
8796 /* Do not update the access time. */
8797 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
8798 #endif
8800 /* These come from sysexits.h */
8801 #ifdef EX_OK
8802 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
8803 #endif /* EX_OK */
8804 #ifdef EX_USAGE
8805 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
8806 #endif /* EX_USAGE */
8807 #ifdef EX_DATAERR
8808 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
8809 #endif /* EX_DATAERR */
8810 #ifdef EX_NOINPUT
8811 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
8812 #endif /* EX_NOINPUT */
8813 #ifdef EX_NOUSER
8814 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
8815 #endif /* EX_NOUSER */
8816 #ifdef EX_NOHOST
8817 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
8818 #endif /* EX_NOHOST */
8819 #ifdef EX_UNAVAILABLE
8820 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
8821 #endif /* EX_UNAVAILABLE */
8822 #ifdef EX_SOFTWARE
8823 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
8824 #endif /* EX_SOFTWARE */
8825 #ifdef EX_OSERR
8826 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
8827 #endif /* EX_OSERR */
8828 #ifdef EX_OSFILE
8829 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
8830 #endif /* EX_OSFILE */
8831 #ifdef EX_CANTCREAT
8832 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
8833 #endif /* EX_CANTCREAT */
8834 #ifdef EX_IOERR
8835 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
8836 #endif /* EX_IOERR */
8837 #ifdef EX_TEMPFAIL
8838 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
8839 #endif /* EX_TEMPFAIL */
8840 #ifdef EX_PROTOCOL
8841 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
8842 #endif /* EX_PROTOCOL */
8843 #ifdef EX_NOPERM
8844 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
8845 #endif /* EX_NOPERM */
8846 #ifdef EX_CONFIG
8847 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
8848 #endif /* EX_CONFIG */
8849 #ifdef EX_NOTFOUND
8850 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
8851 #endif /* EX_NOTFOUND */
8853 #ifdef HAVE_SPAWNV
8854 #if defined(PYOS_OS2) && defined(PYCC_GCC)
8855 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
8856 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
8857 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
8858 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
8859 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
8860 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
8861 if (ins(d, "P_PM", (long)P_PM)) return -1;
8862 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
8863 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
8864 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
8865 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
8866 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
8867 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
8868 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
8869 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
8870 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
8871 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
8872 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
8873 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
8874 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
8875 #else
8876 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
8877 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
8878 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
8879 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
8880 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
8881 #endif
8882 #endif
8884 #if defined(PYOS_OS2)
8885 if (insertvalues(d)) return -1;
8886 #endif
8887 return 0;
8891 #if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
8892 #define INITFUNC initnt
8893 #define MODNAME "nt"
8895 #elif defined(PYOS_OS2)
8896 #define INITFUNC initos2
8897 #define MODNAME "os2"
8899 #else
8900 #define INITFUNC initposix
8901 #define MODNAME "posix"
8902 #endif
8904 PyMODINIT_FUNC
8905 INITFUNC(void)
8907 PyObject *m, *v;
8909 m = Py_InitModule3(MODNAME,
8910 posix_methods,
8911 posix__doc__);
8912 if (m == NULL)
8913 return;
8915 /* Initialize environ dictionary */
8916 v = convertenviron();
8917 Py_XINCREF(v);
8918 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
8919 return;
8920 Py_DECREF(v);
8922 if (all_ins(m))
8923 return;
8925 if (setup_confname_tables(m))
8926 return;
8928 Py_INCREF(PyExc_OSError);
8929 PyModule_AddObject(m, "error", PyExc_OSError);
8931 #ifdef HAVE_PUTENV
8932 if (posix_putenv_garbage == NULL)
8933 posix_putenv_garbage = PyDict_New();
8934 #endif
8936 if (!initialized) {
8937 stat_result_desc.name = MODNAME ".stat_result";
8938 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
8939 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
8940 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
8941 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
8942 structseq_new = StatResultType.tp_new;
8943 StatResultType.tp_new = statresult_new;
8945 statvfs_result_desc.name = MODNAME ".statvfs_result";
8946 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
8948 Py_INCREF((PyObject*) &StatResultType);
8949 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
8950 Py_INCREF((PyObject*) &StatVFSResultType);
8951 PyModule_AddObject(m, "statvfs_result",
8952 (PyObject*) &StatVFSResultType);
8953 initialized = 1;
8955 #ifdef __APPLE__
8957 * Step 2 of weak-linking support on Mac OS X.
8959 * The code below removes functions that are not available on the
8960 * currently active platform.
8962 * This block allow one to use a python binary that was build on
8963 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
8964 * OSX 10.4.
8966 #ifdef HAVE_FSTATVFS
8967 if (fstatvfs == NULL) {
8968 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
8969 return;
8972 #endif /* HAVE_FSTATVFS */
8974 #ifdef HAVE_STATVFS
8975 if (statvfs == NULL) {
8976 if (PyObject_DelAttrString(m, "statvfs") == -1) {
8977 return;
8980 #endif /* HAVE_STATVFS */
8982 # ifdef HAVE_LCHOWN
8983 if (lchown == NULL) {
8984 if (PyObject_DelAttrString(m, "lchown") == -1) {
8985 return;
8988 #endif /* HAVE_LCHOWN */
8991 #endif /* __APPLE__ */
8995 #ifdef __cplusplus
8997 #endif