Issue #7701: Fix crash in binascii.b2a_uu() in debug mode when given a
[python.git] / Modules / posixmodule.c
blob76609eeae3e9bed27060f31d02c1cd986ddba47c
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 <malloc.h>
273 #include <windows.h>
274 #include <shellapi.h> /* for ShellExecute() */
275 #define popen _popen
276 #define pclose _pclose
277 #endif /* _MSC_VER */
279 #if defined(PYCC_VACPP) && defined(PYOS_OS2)
280 #include <io.h>
281 #endif /* OS2 */
283 #ifndef MAXPATHLEN
284 #if defined(PATH_MAX) && PATH_MAX > 1024
285 #define MAXPATHLEN PATH_MAX
286 #else
287 #define MAXPATHLEN 1024
288 #endif
289 #endif /* MAXPATHLEN */
291 #ifdef UNION_WAIT
292 /* Emulate some macros on systems that have a union instead of macros */
294 #ifndef WIFEXITED
295 #define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
296 #endif
298 #ifndef WEXITSTATUS
299 #define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
300 #endif
302 #ifndef WTERMSIG
303 #define WTERMSIG(u_wait) ((u_wait).w_termsig)
304 #endif
306 #define WAIT_TYPE union wait
307 #define WAIT_STATUS_INT(s) (s.w_status)
309 #else /* !UNION_WAIT */
310 #define WAIT_TYPE int
311 #define WAIT_STATUS_INT(s) (s)
312 #endif /* UNION_WAIT */
314 /* Issue #1983: pid_t can be longer than a C long on some systems */
315 #if !defined(SIZEOF_PID_T) || SIZEOF_PID_T == SIZEOF_INT
316 #define PARSE_PID "i"
317 #define PyLong_FromPid PyInt_FromLong
318 #define PyLong_AsPid PyInt_AsLong
319 #elif SIZEOF_PID_T == SIZEOF_LONG
320 #define PARSE_PID "l"
321 #define PyLong_FromPid PyInt_FromLong
322 #define PyLong_AsPid PyInt_AsLong
323 #elif defined(SIZEOF_LONG_LONG) && SIZEOF_PID_T == SIZEOF_LONG_LONG
324 #define PARSE_PID "L"
325 #define PyLong_FromPid PyLong_FromLongLong
326 #define PyLong_AsPid PyInt_AsLongLong
327 #else
328 #error "sizeof(pid_t) is neither sizeof(int), sizeof(long) or sizeof(long long)"
329 #endif /* SIZEOF_PID_T */
331 /* Don't use the "_r" form if we don't need it (also, won't have a
332 prototype for it, at least on Solaris -- maybe others as well?). */
333 #if defined(HAVE_CTERMID_R) && defined(WITH_THREAD)
334 #define USE_CTERMID_R
335 #endif
337 #if defined(HAVE_TMPNAM_R) && defined(WITH_THREAD)
338 #define USE_TMPNAM_R
339 #endif
341 /* choose the appropriate stat and fstat functions and return structs */
342 #undef STAT
343 #if defined(MS_WIN64) || defined(MS_WINDOWS)
344 # define STAT win32_stat
345 # define FSTAT win32_fstat
346 # define STRUCT_STAT struct win32_stat
347 #else
348 # define STAT stat
349 # define FSTAT fstat
350 # define STRUCT_STAT struct stat
351 #endif
353 #if defined(MAJOR_IN_MKDEV)
354 #include <sys/mkdev.h>
355 #else
356 #if defined(MAJOR_IN_SYSMACROS)
357 #include <sys/sysmacros.h>
358 #endif
359 #if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
360 #include <sys/mkdev.h>
361 #endif
362 #endif
364 #if defined _MSC_VER && _MSC_VER >= 1400
365 /* Microsoft CRT in VS2005 and higher will verify that a filehandle is
366 * valid and throw an assertion if it isn't.
367 * Normally, an invalid fd is likely to be a C program error and therefore
368 * an assertion can be useful, but it does contradict the POSIX standard
369 * which for write(2) states:
370 * "Otherwise, -1 shall be returned and errno set to indicate the error."
371 * "[EBADF] The fildes argument is not a valid file descriptor open for
372 * writing."
373 * Furthermore, python allows the user to enter any old integer
374 * as a fd and should merely raise a python exception on error.
375 * The Microsoft CRT doesn't provide an official way to check for the
376 * validity of a file descriptor, but we can emulate its internal behaviour
377 * by using the exported __pinfo data member and knowledge of the
378 * internal structures involved.
379 * The structures below must be updated for each version of visual studio
380 * according to the file internal.h in the CRT source, until MS comes
381 * up with a less hacky way to do this.
382 * (all of this is to avoid globally modifying the CRT behaviour using
383 * _set_invalid_parameter_handler() and _CrtSetReportMode())
385 /* The actual size of the structure is determined at runtime.
386 * Only the first items must be present.
388 typedef struct {
389 intptr_t osfhnd;
390 char osfile;
391 } my_ioinfo;
393 extern __declspec(dllimport) char * __pioinfo[];
394 #define IOINFO_L2E 5
395 #define IOINFO_ARRAY_ELTS (1 << IOINFO_L2E)
396 #define IOINFO_ARRAYS 64
397 #define _NHANDLE_ (IOINFO_ARRAYS * IOINFO_ARRAY_ELTS)
398 #define FOPEN 0x01
399 #define _NO_CONSOLE_FILENO (intptr_t)-2
401 /* This function emulates what the windows CRT does to validate file handles */
403 _PyVerify_fd(int fd)
405 const int i1 = fd >> IOINFO_L2E;
406 const int i2 = fd & ((1 << IOINFO_L2E) - 1);
408 static int sizeof_ioinfo = 0;
410 /* Determine the actual size of the ioinfo structure,
411 * as used by the CRT loaded in memory
413 if (sizeof_ioinfo == 0 && __pioinfo[0] != NULL) {
414 sizeof_ioinfo = _msize(__pioinfo[0]) / IOINFO_ARRAY_ELTS;
416 if (sizeof_ioinfo == 0) {
417 /* This should not happen... */
418 goto fail;
421 /* See that it isn't a special CLEAR fileno */
422 if (fd != _NO_CONSOLE_FILENO) {
423 /* Microsoft CRT would check that 0<=fd<_nhandle but we can't do that. Instead
424 * we check pointer validity and other info
426 if (0 <= i1 && i1 < IOINFO_ARRAYS && __pioinfo[i1] != NULL) {
427 /* finally, check that the file is open */
428 my_ioinfo* info = (my_ioinfo*)(__pioinfo[i1] + i2 * sizeof_ioinfo);
429 if (info->osfile & FOPEN) {
430 return 1;
434 fail:
435 errno = EBADF;
436 return 0;
439 /* the special case of checking dup2. The target fd must be in a sensible range */
440 static int
441 _PyVerify_fd_dup2(int fd1, int fd2)
443 if (!_PyVerify_fd(fd1))
444 return 0;
445 if (fd2 == _NO_CONSOLE_FILENO)
446 return 0;
447 if ((unsigned)fd2 < _NHANDLE_)
448 return 1;
449 else
450 return 0;
452 #else
453 /* dummy version. _PyVerify_fd() is already defined in fileobject.h */
454 #define _PyVerify_fd_dup2(A, B) (1)
455 #endif
457 /* Return a dictionary corresponding to the POSIX environment table */
458 #ifdef WITH_NEXT_FRAMEWORK
459 /* On Darwin/MacOSX a shared library or framework has no access to
460 ** environ directly, we must obtain it with _NSGetEnviron().
462 #include <crt_externs.h>
463 static char **environ;
464 #elif !defined(_MSC_VER) && ( !defined(__WATCOMC__) || defined(__QNX__) )
465 extern char **environ;
466 #endif /* !_MSC_VER */
468 static PyObject *
469 convertenviron(void)
471 PyObject *d;
472 char **e;
473 d = PyDict_New();
474 if (d == NULL)
475 return NULL;
476 #ifdef WITH_NEXT_FRAMEWORK
477 if (environ == NULL)
478 environ = *_NSGetEnviron();
479 #endif
480 if (environ == NULL)
481 return d;
482 /* This part ignores errors */
483 for (e = environ; *e != NULL; e++) {
484 PyObject *k;
485 PyObject *v;
486 char *p = strchr(*e, '=');
487 if (p == NULL)
488 continue;
489 k = PyString_FromStringAndSize(*e, (int)(p-*e));
490 if (k == NULL) {
491 PyErr_Clear();
492 continue;
494 v = PyString_FromString(p+1);
495 if (v == NULL) {
496 PyErr_Clear();
497 Py_DECREF(k);
498 continue;
500 if (PyDict_GetItem(d, k) == NULL) {
501 if (PyDict_SetItem(d, k, v) != 0)
502 PyErr_Clear();
504 Py_DECREF(k);
505 Py_DECREF(v);
507 #if defined(PYOS_OS2)
509 APIRET rc;
510 char buffer[1024]; /* OS/2 Provides a Documented Max of 1024 Chars */
512 rc = DosQueryExtLIBPATH(buffer, BEGIN_LIBPATH);
513 if (rc == NO_ERROR) { /* (not a type, envname is NOT 'BEGIN_LIBPATH') */
514 PyObject *v = PyString_FromString(buffer);
515 PyDict_SetItemString(d, "BEGINLIBPATH", v);
516 Py_DECREF(v);
518 rc = DosQueryExtLIBPATH(buffer, END_LIBPATH);
519 if (rc == NO_ERROR) { /* (not a typo, envname is NOT 'END_LIBPATH') */
520 PyObject *v = PyString_FromString(buffer);
521 PyDict_SetItemString(d, "ENDLIBPATH", v);
522 Py_DECREF(v);
525 #endif
526 return d;
530 /* Set a POSIX-specific error from errno, and return NULL */
532 static PyObject *
533 posix_error(void)
535 return PyErr_SetFromErrno(PyExc_OSError);
537 static PyObject *
538 posix_error_with_filename(char* name)
540 return PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
543 #ifdef MS_WINDOWS
544 static PyObject *
545 posix_error_with_unicode_filename(Py_UNICODE* name)
547 return PyErr_SetFromErrnoWithUnicodeFilename(PyExc_OSError, name);
549 #endif /* MS_WINDOWS */
552 static PyObject *
553 posix_error_with_allocated_filename(char* name)
555 PyObject *rc = PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
556 PyMem_Free(name);
557 return rc;
560 #ifdef MS_WINDOWS
561 static PyObject *
562 win32_error(char* function, char* filename)
564 /* XXX We should pass the function name along in the future.
565 (_winreg.c also wants to pass the function name.)
566 This would however require an additional param to the
567 Windows error object, which is non-trivial.
569 errno = GetLastError();
570 if (filename)
571 return PyErr_SetFromWindowsErrWithFilename(errno, filename);
572 else
573 return PyErr_SetFromWindowsErr(errno);
576 static PyObject *
577 win32_error_unicode(char* function, Py_UNICODE* filename)
579 /* XXX - see win32_error for comments on 'function' */
580 errno = GetLastError();
581 if (filename)
582 return PyErr_SetFromWindowsErrWithUnicodeFilename(errno, filename);
583 else
584 return PyErr_SetFromWindowsErr(errno);
587 static int
588 convert_to_unicode(PyObject **param)
590 if (PyUnicode_CheckExact(*param))
591 Py_INCREF(*param);
592 else if (PyUnicode_Check(*param))
593 /* For a Unicode subtype that's not a Unicode object,
594 return a true Unicode object with the same data. */
595 *param = PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(*param),
596 PyUnicode_GET_SIZE(*param));
597 else
598 *param = PyUnicode_FromEncodedObject(*param,
599 Py_FileSystemDefaultEncoding,
600 "strict");
601 return (*param) != NULL;
604 #endif /* MS_WINDOWS */
606 #if defined(PYOS_OS2)
607 /**********************************************************************
608 * Helper Function to Trim and Format OS/2 Messages
609 **********************************************************************/
610 static void
611 os2_formatmsg(char *msgbuf, int msglen, char *reason)
613 msgbuf[msglen] = '\0'; /* OS/2 Doesn't Guarantee a Terminator */
615 if (strlen(msgbuf) > 0) { /* If Non-Empty Msg, Trim CRLF */
616 char *lastc = &msgbuf[ strlen(msgbuf)-1 ];
618 while (lastc > msgbuf && isspace(Py_CHARMASK(*lastc)))
619 *lastc-- = '\0'; /* Trim Trailing Whitespace (CRLF) */
622 /* Add Optional Reason Text */
623 if (reason) {
624 strcat(msgbuf, " : ");
625 strcat(msgbuf, reason);
629 /**********************************************************************
630 * Decode an OS/2 Operating System Error Code
632 * A convenience function to lookup an OS/2 error code and return a
633 * text message we can use to raise a Python exception.
635 * Notes:
636 * The messages for errors returned from the OS/2 kernel reside in
637 * the file OSO001.MSG in the \OS2 directory hierarchy.
639 **********************************************************************/
640 static char *
641 os2_strerror(char *msgbuf, int msgbuflen, int errorcode, char *reason)
643 APIRET rc;
644 ULONG msglen;
646 /* Retrieve Kernel-Related Error Message from OSO001.MSG File */
647 Py_BEGIN_ALLOW_THREADS
648 rc = DosGetMessage(NULL, 0, msgbuf, msgbuflen,
649 errorcode, "oso001.msg", &msglen);
650 Py_END_ALLOW_THREADS
652 if (rc == NO_ERROR)
653 os2_formatmsg(msgbuf, msglen, reason);
654 else
655 PyOS_snprintf(msgbuf, msgbuflen,
656 "unknown OS error #%d", errorcode);
658 return msgbuf;
661 /* Set an OS/2-specific error and return NULL. OS/2 kernel
662 errors are not in a global variable e.g. 'errno' nor are
663 they congruent with posix error numbers. */
665 static PyObject * os2_error(int code)
667 char text[1024];
668 PyObject *v;
670 os2_strerror(text, sizeof(text), code, "");
672 v = Py_BuildValue("(is)", code, text);
673 if (v != NULL) {
674 PyErr_SetObject(PyExc_OSError, v);
675 Py_DECREF(v);
677 return NULL; /* Signal to Python that an Exception is Pending */
680 #endif /* OS2 */
682 /* POSIX generic methods */
684 static PyObject *
685 posix_fildes(PyObject *fdobj, int (*func)(int))
687 int fd;
688 int res;
689 fd = PyObject_AsFileDescriptor(fdobj);
690 if (fd < 0)
691 return NULL;
692 if (!_PyVerify_fd(fd))
693 return posix_error();
694 Py_BEGIN_ALLOW_THREADS
695 res = (*func)(fd);
696 Py_END_ALLOW_THREADS
697 if (res < 0)
698 return posix_error();
699 Py_INCREF(Py_None);
700 return Py_None;
703 static PyObject *
704 posix_1str(PyObject *args, char *format, int (*func)(const char*))
706 char *path1 = NULL;
707 int res;
708 if (!PyArg_ParseTuple(args, format,
709 Py_FileSystemDefaultEncoding, &path1))
710 return NULL;
711 Py_BEGIN_ALLOW_THREADS
712 res = (*func)(path1);
713 Py_END_ALLOW_THREADS
714 if (res < 0)
715 return posix_error_with_allocated_filename(path1);
716 PyMem_Free(path1);
717 Py_INCREF(Py_None);
718 return Py_None;
721 static PyObject *
722 posix_2str(PyObject *args,
723 char *format,
724 int (*func)(const char *, const char *))
726 char *path1 = NULL, *path2 = NULL;
727 int res;
728 if (!PyArg_ParseTuple(args, format,
729 Py_FileSystemDefaultEncoding, &path1,
730 Py_FileSystemDefaultEncoding, &path2))
731 return NULL;
732 Py_BEGIN_ALLOW_THREADS
733 res = (*func)(path1, path2);
734 Py_END_ALLOW_THREADS
735 PyMem_Free(path1);
736 PyMem_Free(path2);
737 if (res != 0)
738 /* XXX how to report both path1 and path2??? */
739 return posix_error();
740 Py_INCREF(Py_None);
741 return Py_None;
744 #ifdef MS_WINDOWS
745 static PyObject*
746 win32_1str(PyObject* args, char* func,
747 char* format, BOOL (__stdcall *funcA)(LPCSTR),
748 char* wformat, BOOL (__stdcall *funcW)(LPWSTR))
750 PyObject *uni;
751 char *ansi;
752 BOOL result;
754 if (!PyArg_ParseTuple(args, wformat, &uni))
755 PyErr_Clear();
756 else {
757 Py_BEGIN_ALLOW_THREADS
758 result = funcW(PyUnicode_AsUnicode(uni));
759 Py_END_ALLOW_THREADS
760 if (!result)
761 return win32_error_unicode(func, PyUnicode_AsUnicode(uni));
762 Py_INCREF(Py_None);
763 return Py_None;
765 if (!PyArg_ParseTuple(args, format, &ansi))
766 return NULL;
767 Py_BEGIN_ALLOW_THREADS
768 result = funcA(ansi);
769 Py_END_ALLOW_THREADS
770 if (!result)
771 return win32_error(func, ansi);
772 Py_INCREF(Py_None);
773 return Py_None;
777 /* This is a reimplementation of the C library's chdir function,
778 but one that produces Win32 errors instead of DOS error codes.
779 chdir is essentially a wrapper around SetCurrentDirectory; however,
780 it also needs to set "magic" environment variables indicating
781 the per-drive current directory, which are of the form =<drive>: */
782 static BOOL __stdcall
783 win32_chdir(LPCSTR path)
785 char new_path[MAX_PATH+1];
786 int result;
787 char env[4] = "=x:";
789 if(!SetCurrentDirectoryA(path))
790 return FALSE;
791 result = GetCurrentDirectoryA(MAX_PATH+1, new_path);
792 if (!result)
793 return FALSE;
794 /* In the ANSI API, there should not be any paths longer
795 than MAX_PATH. */
796 assert(result <= MAX_PATH+1);
797 if (strncmp(new_path, "\\\\", 2) == 0 ||
798 strncmp(new_path, "//", 2) == 0)
799 /* UNC path, nothing to do. */
800 return TRUE;
801 env[1] = new_path[0];
802 return SetEnvironmentVariableA(env, new_path);
805 /* The Unicode version differs from the ANSI version
806 since the current directory might exceed MAX_PATH characters */
807 static BOOL __stdcall
808 win32_wchdir(LPCWSTR path)
810 wchar_t _new_path[MAX_PATH+1], *new_path = _new_path;
811 int result;
812 wchar_t env[4] = L"=x:";
814 if(!SetCurrentDirectoryW(path))
815 return FALSE;
816 result = GetCurrentDirectoryW(MAX_PATH+1, new_path);
817 if (!result)
818 return FALSE;
819 if (result > MAX_PATH+1) {
820 new_path = malloc(result * sizeof(wchar_t));
821 if (!new_path) {
822 SetLastError(ERROR_OUTOFMEMORY);
823 return FALSE;
825 result = GetCurrentDirectoryW(result, new_path);
826 if (!result) {
827 free(new_path);
828 return FALSE;
831 if (wcsncmp(new_path, L"\\\\", 2) == 0 ||
832 wcsncmp(new_path, L"//", 2) == 0)
833 /* UNC path, nothing to do. */
834 return TRUE;
835 env[1] = new_path[0];
836 result = SetEnvironmentVariableW(env, new_path);
837 if (new_path != _new_path)
838 free(new_path);
839 return result;
841 #endif
843 #ifdef MS_WINDOWS
844 /* The CRT of Windows has a number of flaws wrt. its stat() implementation:
845 - time stamps are restricted to second resolution
846 - file modification times suffer from forth-and-back conversions between
847 UTC and local time
848 Therefore, we implement our own stat, based on the Win32 API directly.
850 #define HAVE_STAT_NSEC 1
852 struct win32_stat{
853 int st_dev;
854 __int64 st_ino;
855 unsigned short st_mode;
856 int st_nlink;
857 int st_uid;
858 int st_gid;
859 int st_rdev;
860 __int64 st_size;
861 int st_atime;
862 int st_atime_nsec;
863 int st_mtime;
864 int st_mtime_nsec;
865 int st_ctime;
866 int st_ctime_nsec;
869 static __int64 secs_between_epochs = 11644473600; /* Seconds between 1.1.1601 and 1.1.1970 */
871 static void
872 FILE_TIME_to_time_t_nsec(FILETIME *in_ptr, int *time_out, int* nsec_out)
874 /* XXX endianness. Shouldn't matter, as all Windows implementations are little-endian */
875 /* Cannot simply cast and dereference in_ptr,
876 since it might not be aligned properly */
877 __int64 in;
878 memcpy(&in, in_ptr, sizeof(in));
879 *nsec_out = (int)(in % 10000000) * 100; /* FILETIME is in units of 100 nsec. */
880 /* XXX Win32 supports time stamps past 2038; we currently don't */
881 *time_out = Py_SAFE_DOWNCAST((in / 10000000) - secs_between_epochs, __int64, int);
884 static void
885 time_t_to_FILE_TIME(int time_in, int nsec_in, FILETIME *out_ptr)
887 /* XXX endianness */
888 __int64 out;
889 out = time_in + secs_between_epochs;
890 out = out * 10000000 + nsec_in / 100;
891 memcpy(out_ptr, &out, sizeof(out));
894 /* Below, we *know* that ugo+r is 0444 */
895 #if _S_IREAD != 0400
896 #error Unsupported C library
897 #endif
898 static int
899 attributes_to_mode(DWORD attr)
901 int m = 0;
902 if (attr & FILE_ATTRIBUTE_DIRECTORY)
903 m |= _S_IFDIR | 0111; /* IFEXEC for user,group,other */
904 else
905 m |= _S_IFREG;
906 if (attr & FILE_ATTRIBUTE_READONLY)
907 m |= 0444;
908 else
909 m |= 0666;
910 return m;
913 static int
914 attribute_data_to_stat(WIN32_FILE_ATTRIBUTE_DATA *info, struct win32_stat *result)
916 memset(result, 0, sizeof(*result));
917 result->st_mode = attributes_to_mode(info->dwFileAttributes);
918 result->st_size = (((__int64)info->nFileSizeHigh)<<32) + info->nFileSizeLow;
919 FILE_TIME_to_time_t_nsec(&info->ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
920 FILE_TIME_to_time_t_nsec(&info->ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
921 FILE_TIME_to_time_t_nsec(&info->ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
923 return 0;
926 static BOOL
927 attributes_from_dir(LPCSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
929 HANDLE hFindFile;
930 WIN32_FIND_DATAA FileData;
931 hFindFile = FindFirstFileA(pszFile, &FileData);
932 if (hFindFile == INVALID_HANDLE_VALUE)
933 return FALSE;
934 FindClose(hFindFile);
935 pfad->dwFileAttributes = FileData.dwFileAttributes;
936 pfad->ftCreationTime = FileData.ftCreationTime;
937 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
938 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
939 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
940 pfad->nFileSizeLow = FileData.nFileSizeLow;
941 return TRUE;
944 static BOOL
945 attributes_from_dir_w(LPCWSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
947 HANDLE hFindFile;
948 WIN32_FIND_DATAW FileData;
949 hFindFile = FindFirstFileW(pszFile, &FileData);
950 if (hFindFile == INVALID_HANDLE_VALUE)
951 return FALSE;
952 FindClose(hFindFile);
953 pfad->dwFileAttributes = FileData.dwFileAttributes;
954 pfad->ftCreationTime = FileData.ftCreationTime;
955 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
956 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
957 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
958 pfad->nFileSizeLow = FileData.nFileSizeLow;
959 return TRUE;
962 static int
963 win32_stat(const char* path, struct win32_stat *result)
965 WIN32_FILE_ATTRIBUTE_DATA info;
966 int code;
967 char *dot;
968 if (!GetFileAttributesExA(path, GetFileExInfoStandard, &info)) {
969 if (GetLastError() != ERROR_SHARING_VIOLATION) {
970 /* Protocol violation: we explicitly clear errno, instead of
971 setting it to a POSIX error. Callers should use GetLastError. */
972 errno = 0;
973 return -1;
974 } else {
975 /* Could not get attributes on open file. Fall back to
976 reading the directory. */
977 if (!attributes_from_dir(path, &info)) {
978 /* Very strange. This should not fail now */
979 errno = 0;
980 return -1;
984 code = attribute_data_to_stat(&info, result);
985 if (code != 0)
986 return code;
987 /* Set S_IFEXEC if it is an .exe, .bat, ... */
988 dot = strrchr(path, '.');
989 if (dot) {
990 if (stricmp(dot, ".bat") == 0 ||
991 stricmp(dot, ".cmd") == 0 ||
992 stricmp(dot, ".exe") == 0 ||
993 stricmp(dot, ".com") == 0)
994 result->st_mode |= 0111;
996 return code;
999 static int
1000 win32_wstat(const wchar_t* path, struct win32_stat *result)
1002 int code;
1003 const wchar_t *dot;
1004 WIN32_FILE_ATTRIBUTE_DATA info;
1005 if (!GetFileAttributesExW(path, GetFileExInfoStandard, &info)) {
1006 if (GetLastError() != ERROR_SHARING_VIOLATION) {
1007 /* Protocol violation: we explicitly clear errno, instead of
1008 setting it to a POSIX error. Callers should use GetLastError. */
1009 errno = 0;
1010 return -1;
1011 } else {
1012 /* Could not get attributes on open file. Fall back to
1013 reading the directory. */
1014 if (!attributes_from_dir_w(path, &info)) {
1015 /* Very strange. This should not fail now */
1016 errno = 0;
1017 return -1;
1021 code = attribute_data_to_stat(&info, result);
1022 if (code < 0)
1023 return code;
1024 /* Set IFEXEC if it is an .exe, .bat, ... */
1025 dot = wcsrchr(path, '.');
1026 if (dot) {
1027 if (_wcsicmp(dot, L".bat") == 0 ||
1028 _wcsicmp(dot, L".cmd") == 0 ||
1029 _wcsicmp(dot, L".exe") == 0 ||
1030 _wcsicmp(dot, L".com") == 0)
1031 result->st_mode |= 0111;
1033 return code;
1036 static int
1037 win32_fstat(int file_number, struct win32_stat *result)
1039 BY_HANDLE_FILE_INFORMATION info;
1040 HANDLE h;
1041 int type;
1043 h = (HANDLE)_get_osfhandle(file_number);
1045 /* Protocol violation: we explicitly clear errno, instead of
1046 setting it to a POSIX error. Callers should use GetLastError. */
1047 errno = 0;
1049 if (h == INVALID_HANDLE_VALUE) {
1050 /* This is really a C library error (invalid file handle).
1051 We set the Win32 error to the closes one matching. */
1052 SetLastError(ERROR_INVALID_HANDLE);
1053 return -1;
1055 memset(result, 0, sizeof(*result));
1057 type = GetFileType(h);
1058 if (type == FILE_TYPE_UNKNOWN) {
1059 DWORD error = GetLastError();
1060 if (error != 0) {
1061 return -1;
1063 /* else: valid but unknown file */
1066 if (type != FILE_TYPE_DISK) {
1067 if (type == FILE_TYPE_CHAR)
1068 result->st_mode = _S_IFCHR;
1069 else if (type == FILE_TYPE_PIPE)
1070 result->st_mode = _S_IFIFO;
1071 return 0;
1074 if (!GetFileInformationByHandle(h, &info)) {
1075 return -1;
1078 /* similar to stat() */
1079 result->st_mode = attributes_to_mode(info.dwFileAttributes);
1080 result->st_size = (((__int64)info.nFileSizeHigh)<<32) + info.nFileSizeLow;
1081 FILE_TIME_to_time_t_nsec(&info.ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1082 FILE_TIME_to_time_t_nsec(&info.ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1083 FILE_TIME_to_time_t_nsec(&info.ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
1084 /* specific to fstat() */
1085 result->st_nlink = info.nNumberOfLinks;
1086 result->st_ino = (((__int64)info.nFileIndexHigh)<<32) + info.nFileIndexLow;
1087 return 0;
1090 #endif /* MS_WINDOWS */
1092 PyDoc_STRVAR(stat_result__doc__,
1093 "stat_result: Result from stat or lstat.\n\n\
1094 This object may be accessed either as a tuple of\n\
1095 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
1096 or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
1098 Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
1099 or st_flags, they are available as attributes only.\n\
1101 See os.stat for more information.");
1103 static PyStructSequence_Field stat_result_fields[] = {
1104 {"st_mode", "protection bits"},
1105 {"st_ino", "inode"},
1106 {"st_dev", "device"},
1107 {"st_nlink", "number of hard links"},
1108 {"st_uid", "user ID of owner"},
1109 {"st_gid", "group ID of owner"},
1110 {"st_size", "total size, in bytes"},
1111 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
1112 {NULL, "integer time of last access"},
1113 {NULL, "integer time of last modification"},
1114 {NULL, "integer time of last change"},
1115 {"st_atime", "time of last access"},
1116 {"st_mtime", "time of last modification"},
1117 {"st_ctime", "time of last change"},
1118 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
1119 {"st_blksize", "blocksize for filesystem I/O"},
1120 #endif
1121 #ifdef HAVE_STRUCT_STAT_ST_BLOCKS
1122 {"st_blocks", "number of blocks allocated"},
1123 #endif
1124 #ifdef HAVE_STRUCT_STAT_ST_RDEV
1125 {"st_rdev", "device type (if inode device)"},
1126 #endif
1127 #ifdef HAVE_STRUCT_STAT_ST_FLAGS
1128 {"st_flags", "user defined flags for file"},
1129 #endif
1130 #ifdef HAVE_STRUCT_STAT_ST_GEN
1131 {"st_gen", "generation number"},
1132 #endif
1133 #ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1134 {"st_birthtime", "time of creation"},
1135 #endif
1139 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
1140 #define ST_BLKSIZE_IDX 13
1141 #else
1142 #define ST_BLKSIZE_IDX 12
1143 #endif
1145 #ifdef HAVE_STRUCT_STAT_ST_BLOCKS
1146 #define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
1147 #else
1148 #define ST_BLOCKS_IDX ST_BLKSIZE_IDX
1149 #endif
1151 #ifdef HAVE_STRUCT_STAT_ST_RDEV
1152 #define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
1153 #else
1154 #define ST_RDEV_IDX ST_BLOCKS_IDX
1155 #endif
1157 #ifdef HAVE_STRUCT_STAT_ST_FLAGS
1158 #define ST_FLAGS_IDX (ST_RDEV_IDX+1)
1159 #else
1160 #define ST_FLAGS_IDX ST_RDEV_IDX
1161 #endif
1163 #ifdef HAVE_STRUCT_STAT_ST_GEN
1164 #define ST_GEN_IDX (ST_FLAGS_IDX+1)
1165 #else
1166 #define ST_GEN_IDX ST_FLAGS_IDX
1167 #endif
1169 #ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1170 #define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
1171 #else
1172 #define ST_BIRTHTIME_IDX ST_GEN_IDX
1173 #endif
1175 static PyStructSequence_Desc stat_result_desc = {
1176 "stat_result", /* name */
1177 stat_result__doc__, /* doc */
1178 stat_result_fields,
1182 PyDoc_STRVAR(statvfs_result__doc__,
1183 "statvfs_result: Result from statvfs or fstatvfs.\n\n\
1184 This object may be accessed either as a tuple of\n\
1185 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
1186 or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
1188 See os.statvfs for more information.");
1190 static PyStructSequence_Field statvfs_result_fields[] = {
1191 {"f_bsize", },
1192 {"f_frsize", },
1193 {"f_blocks", },
1194 {"f_bfree", },
1195 {"f_bavail", },
1196 {"f_files", },
1197 {"f_ffree", },
1198 {"f_favail", },
1199 {"f_flag", },
1200 {"f_namemax",},
1204 static PyStructSequence_Desc statvfs_result_desc = {
1205 "statvfs_result", /* name */
1206 statvfs_result__doc__, /* doc */
1207 statvfs_result_fields,
1211 static int initialized;
1212 static PyTypeObject StatResultType;
1213 static PyTypeObject StatVFSResultType;
1214 static newfunc structseq_new;
1216 static PyObject *
1217 statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1219 PyStructSequence *result;
1220 int i;
1222 result = (PyStructSequence*)structseq_new(type, args, kwds);
1223 if (!result)
1224 return NULL;
1225 /* If we have been initialized from a tuple,
1226 st_?time might be set to None. Initialize it
1227 from the int slots. */
1228 for (i = 7; i <= 9; i++) {
1229 if (result->ob_item[i+3] == Py_None) {
1230 Py_DECREF(Py_None);
1231 Py_INCREF(result->ob_item[i]);
1232 result->ob_item[i+3] = result->ob_item[i];
1235 return (PyObject*)result;
1240 /* If true, st_?time is float. */
1241 static int _stat_float_times = 1;
1243 PyDoc_STRVAR(stat_float_times__doc__,
1244 "stat_float_times([newval]) -> oldval\n\n\
1245 Determine whether os.[lf]stat represents time stamps as float objects.\n\
1246 If newval is True, future calls to stat() return floats, if it is False,\n\
1247 future calls return ints. \n\
1248 If newval is omitted, return the current setting.\n");
1250 static PyObject*
1251 stat_float_times(PyObject* self, PyObject *args)
1253 int newval = -1;
1254 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
1255 return NULL;
1256 if (newval == -1)
1257 /* Return old value */
1258 return PyBool_FromLong(_stat_float_times);
1259 _stat_float_times = newval;
1260 Py_INCREF(Py_None);
1261 return Py_None;
1264 static void
1265 fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
1267 PyObject *fval,*ival;
1268 #if SIZEOF_TIME_T > SIZEOF_LONG
1269 ival = PyLong_FromLongLong((PY_LONG_LONG)sec);
1270 #else
1271 ival = PyInt_FromLong((long)sec);
1272 #endif
1273 if (!ival)
1274 return;
1275 if (_stat_float_times) {
1276 fval = PyFloat_FromDouble(sec + 1e-9*nsec);
1277 } else {
1278 fval = ival;
1279 Py_INCREF(fval);
1281 PyStructSequence_SET_ITEM(v, index, ival);
1282 PyStructSequence_SET_ITEM(v, index+3, fval);
1285 /* pack a system stat C structure into the Python stat tuple
1286 (used by posix_stat() and posix_fstat()) */
1287 static PyObject*
1288 _pystat_fromstructstat(STRUCT_STAT *st)
1290 unsigned long ansec, mnsec, cnsec;
1291 PyObject *v = PyStructSequence_New(&StatResultType);
1292 if (v == NULL)
1293 return NULL;
1295 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long)st->st_mode));
1296 #ifdef HAVE_LARGEFILE_SUPPORT
1297 PyStructSequence_SET_ITEM(v, 1,
1298 PyLong_FromLongLong((PY_LONG_LONG)st->st_ino));
1299 #else
1300 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long)st->st_ino));
1301 #endif
1302 #if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
1303 PyStructSequence_SET_ITEM(v, 2,
1304 PyLong_FromLongLong((PY_LONG_LONG)st->st_dev));
1305 #else
1306 PyStructSequence_SET_ITEM(v, 2, PyInt_FromLong((long)st->st_dev));
1307 #endif
1308 PyStructSequence_SET_ITEM(v, 3, PyInt_FromLong((long)st->st_nlink));
1309 PyStructSequence_SET_ITEM(v, 4, PyInt_FromLong((long)st->st_uid));
1310 PyStructSequence_SET_ITEM(v, 5, PyInt_FromLong((long)st->st_gid));
1311 #ifdef HAVE_LARGEFILE_SUPPORT
1312 PyStructSequence_SET_ITEM(v, 6,
1313 PyLong_FromLongLong((PY_LONG_LONG)st->st_size));
1314 #else
1315 PyStructSequence_SET_ITEM(v, 6, PyInt_FromLong(st->st_size));
1316 #endif
1318 #if defined(HAVE_STAT_TV_NSEC)
1319 ansec = st->st_atim.tv_nsec;
1320 mnsec = st->st_mtim.tv_nsec;
1321 cnsec = st->st_ctim.tv_nsec;
1322 #elif defined(HAVE_STAT_TV_NSEC2)
1323 ansec = st->st_atimespec.tv_nsec;
1324 mnsec = st->st_mtimespec.tv_nsec;
1325 cnsec = st->st_ctimespec.tv_nsec;
1326 #elif defined(HAVE_STAT_NSEC)
1327 ansec = st->st_atime_nsec;
1328 mnsec = st->st_mtime_nsec;
1329 cnsec = st->st_ctime_nsec;
1330 #else
1331 ansec = mnsec = cnsec = 0;
1332 #endif
1333 fill_time(v, 7, st->st_atime, ansec);
1334 fill_time(v, 8, st->st_mtime, mnsec);
1335 fill_time(v, 9, st->st_ctime, cnsec);
1337 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
1338 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
1339 PyInt_FromLong((long)st->st_blksize));
1340 #endif
1341 #ifdef HAVE_STRUCT_STAT_ST_BLOCKS
1342 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
1343 PyInt_FromLong((long)st->st_blocks));
1344 #endif
1345 #ifdef HAVE_STRUCT_STAT_ST_RDEV
1346 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
1347 PyInt_FromLong((long)st->st_rdev));
1348 #endif
1349 #ifdef HAVE_STRUCT_STAT_ST_GEN
1350 PyStructSequence_SET_ITEM(v, ST_GEN_IDX,
1351 PyInt_FromLong((long)st->st_gen));
1352 #endif
1353 #ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1355 PyObject *val;
1356 unsigned long bsec,bnsec;
1357 bsec = (long)st->st_birthtime;
1358 #ifdef HAVE_STAT_TV_NSEC2
1359 bnsec = st->st_birthtimespec.tv_nsec;
1360 #else
1361 bnsec = 0;
1362 #endif
1363 if (_stat_float_times) {
1364 val = PyFloat_FromDouble(bsec + 1e-9*bnsec);
1365 } else {
1366 val = PyInt_FromLong((long)bsec);
1368 PyStructSequence_SET_ITEM(v, ST_BIRTHTIME_IDX,
1369 val);
1371 #endif
1372 #ifdef HAVE_STRUCT_STAT_ST_FLAGS
1373 PyStructSequence_SET_ITEM(v, ST_FLAGS_IDX,
1374 PyInt_FromLong((long)st->st_flags));
1375 #endif
1377 if (PyErr_Occurred()) {
1378 Py_DECREF(v);
1379 return NULL;
1382 return v;
1385 #ifdef MS_WINDOWS
1387 /* IsUNCRoot -- test whether the supplied path is of the form \\SERVER\SHARE\,
1388 where / can be used in place of \ and the trailing slash is optional.
1389 Both SERVER and SHARE must have at least one character.
1392 #define ISSLASHA(c) ((c) == '\\' || (c) == '/')
1393 #define ISSLASHW(c) ((c) == L'\\' || (c) == L'/')
1394 #ifndef ARRAYSIZE
1395 #define ARRAYSIZE(a) (sizeof(a) / sizeof(a[0]))
1396 #endif
1398 static BOOL
1399 IsUNCRootA(char *path, int pathlen)
1401 #define ISSLASH ISSLASHA
1403 int i, share;
1405 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1406 /* minimum UNCRoot is \\x\y */
1407 return FALSE;
1408 for (i = 2; i < pathlen ; i++)
1409 if (ISSLASH(path[i])) break;
1410 if (i == 2 || i == pathlen)
1411 /* do not allow \\\SHARE or \\SERVER */
1412 return FALSE;
1413 share = i+1;
1414 for (i = share; i < pathlen; i++)
1415 if (ISSLASH(path[i])) break;
1416 return (i != share && (i == pathlen || i == pathlen-1));
1418 #undef ISSLASH
1421 static BOOL
1422 IsUNCRootW(Py_UNICODE *path, int pathlen)
1424 #define ISSLASH ISSLASHW
1426 int i, share;
1428 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1429 /* minimum UNCRoot is \\x\y */
1430 return FALSE;
1431 for (i = 2; i < pathlen ; i++)
1432 if (ISSLASH(path[i])) break;
1433 if (i == 2 || i == pathlen)
1434 /* do not allow \\\SHARE or \\SERVER */
1435 return FALSE;
1436 share = i+1;
1437 for (i = share; i < pathlen; i++)
1438 if (ISSLASH(path[i])) break;
1439 return (i != share && (i == pathlen || i == pathlen-1));
1441 #undef ISSLASH
1443 #endif /* MS_WINDOWS */
1445 static PyObject *
1446 posix_do_stat(PyObject *self, PyObject *args,
1447 char *format,
1448 #ifdef __VMS
1449 int (*statfunc)(const char *, STRUCT_STAT *, ...),
1450 #else
1451 int (*statfunc)(const char *, STRUCT_STAT *),
1452 #endif
1453 char *wformat,
1454 int (*wstatfunc)(const Py_UNICODE *, STRUCT_STAT *))
1456 STRUCT_STAT st;
1457 char *path = NULL; /* pass this to stat; do not free() it */
1458 char *pathfree = NULL; /* this memory must be free'd */
1459 int res;
1460 PyObject *result;
1462 #ifdef MS_WINDOWS
1463 PyUnicodeObject *po;
1464 if (PyArg_ParseTuple(args, wformat, &po)) {
1465 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
1467 Py_BEGIN_ALLOW_THREADS
1468 /* PyUnicode_AS_UNICODE result OK without
1469 thread lock as it is a simple dereference. */
1470 res = wstatfunc(wpath, &st);
1471 Py_END_ALLOW_THREADS
1473 if (res != 0)
1474 return win32_error_unicode("stat", wpath);
1475 return _pystat_fromstructstat(&st);
1477 /* Drop the argument parsing error as narrow strings
1478 are also valid. */
1479 PyErr_Clear();
1480 #endif
1482 if (!PyArg_ParseTuple(args, format,
1483 Py_FileSystemDefaultEncoding, &path))
1484 return NULL;
1485 pathfree = path;
1487 Py_BEGIN_ALLOW_THREADS
1488 res = (*statfunc)(path, &st);
1489 Py_END_ALLOW_THREADS
1491 if (res != 0) {
1492 #ifdef MS_WINDOWS
1493 result = win32_error("stat", pathfree);
1494 #else
1495 result = posix_error_with_filename(pathfree);
1496 #endif
1498 else
1499 result = _pystat_fromstructstat(&st);
1501 PyMem_Free(pathfree);
1502 return result;
1505 /* POSIX methods */
1507 PyDoc_STRVAR(posix_access__doc__,
1508 "access(path, mode) -> True if granted, False otherwise\n\n\
1509 Use the real uid/gid to test for access to a path. Note that most\n\
1510 operations will use the effective uid/gid, therefore this routine can\n\
1511 be used in a suid/sgid environment to test if the invoking user has the\n\
1512 specified access to the path. The mode argument can be F_OK to test\n\
1513 existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
1515 static PyObject *
1516 posix_access(PyObject *self, PyObject *args)
1518 char *path;
1519 int mode;
1521 #ifdef MS_WINDOWS
1522 DWORD attr;
1523 PyUnicodeObject *po;
1524 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
1525 Py_BEGIN_ALLOW_THREADS
1526 /* PyUnicode_AS_UNICODE OK without thread lock as
1527 it is a simple dereference. */
1528 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1529 Py_END_ALLOW_THREADS
1530 goto finish;
1532 /* Drop the argument parsing error as narrow strings
1533 are also valid. */
1534 PyErr_Clear();
1535 if (!PyArg_ParseTuple(args, "eti:access",
1536 Py_FileSystemDefaultEncoding, &path, &mode))
1537 return NULL;
1538 Py_BEGIN_ALLOW_THREADS
1539 attr = GetFileAttributesA(path);
1540 Py_END_ALLOW_THREADS
1541 PyMem_Free(path);
1542 finish:
1543 if (attr == 0xFFFFFFFF)
1544 /* File does not exist, or cannot read attributes */
1545 return PyBool_FromLong(0);
1546 /* Access is possible if either write access wasn't requested, or
1547 the file isn't read-only, or if it's a directory, as there are
1548 no read-only directories on Windows. */
1549 return PyBool_FromLong(!(mode & 2)
1550 || !(attr & FILE_ATTRIBUTE_READONLY)
1551 || (attr & FILE_ATTRIBUTE_DIRECTORY));
1552 #else /* MS_WINDOWS */
1553 int res;
1554 if (!PyArg_ParseTuple(args, "eti:access",
1555 Py_FileSystemDefaultEncoding, &path, &mode))
1556 return NULL;
1557 Py_BEGIN_ALLOW_THREADS
1558 res = access(path, mode);
1559 Py_END_ALLOW_THREADS
1560 PyMem_Free(path);
1561 return PyBool_FromLong(res == 0);
1562 #endif /* MS_WINDOWS */
1565 #ifndef F_OK
1566 #define F_OK 0
1567 #endif
1568 #ifndef R_OK
1569 #define R_OK 4
1570 #endif
1571 #ifndef W_OK
1572 #define W_OK 2
1573 #endif
1574 #ifndef X_OK
1575 #define X_OK 1
1576 #endif
1578 #ifdef HAVE_TTYNAME
1579 PyDoc_STRVAR(posix_ttyname__doc__,
1580 "ttyname(fd) -> string\n\n\
1581 Return the name of the terminal device connected to 'fd'.");
1583 static PyObject *
1584 posix_ttyname(PyObject *self, PyObject *args)
1586 int id;
1587 char *ret;
1589 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
1590 return NULL;
1592 #if defined(__VMS)
1593 /* file descriptor 0 only, the default input device (stdin) */
1594 if (id == 0) {
1595 ret = ttyname();
1597 else {
1598 ret = NULL;
1600 #else
1601 ret = ttyname(id);
1602 #endif
1603 if (ret == NULL)
1604 return posix_error();
1605 return PyString_FromString(ret);
1607 #endif
1609 #ifdef HAVE_CTERMID
1610 PyDoc_STRVAR(posix_ctermid__doc__,
1611 "ctermid() -> string\n\n\
1612 Return the name of the controlling terminal for this process.");
1614 static PyObject *
1615 posix_ctermid(PyObject *self, PyObject *noargs)
1617 char *ret;
1618 char buffer[L_ctermid];
1620 #ifdef USE_CTERMID_R
1621 ret = ctermid_r(buffer);
1622 #else
1623 ret = ctermid(buffer);
1624 #endif
1625 if (ret == NULL)
1626 return posix_error();
1627 return PyString_FromString(buffer);
1629 #endif
1631 PyDoc_STRVAR(posix_chdir__doc__,
1632 "chdir(path)\n\n\
1633 Change the current working directory to the specified path.");
1635 static PyObject *
1636 posix_chdir(PyObject *self, PyObject *args)
1638 #ifdef MS_WINDOWS
1639 return win32_1str(args, "chdir", "s:chdir", win32_chdir, "U:chdir", win32_wchdir);
1640 #elif defined(PYOS_OS2) && defined(PYCC_GCC)
1641 return posix_1str(args, "et:chdir", _chdir2);
1642 #elif defined(__VMS)
1643 return posix_1str(args, "et:chdir", (int (*)(const char *))chdir);
1644 #else
1645 return posix_1str(args, "et:chdir", chdir);
1646 #endif
1649 #ifdef HAVE_FCHDIR
1650 PyDoc_STRVAR(posix_fchdir__doc__,
1651 "fchdir(fildes)\n\n\
1652 Change to the directory of the given file descriptor. fildes must be\n\
1653 opened on a directory, not a file.");
1655 static PyObject *
1656 posix_fchdir(PyObject *self, PyObject *fdobj)
1658 return posix_fildes(fdobj, fchdir);
1660 #endif /* HAVE_FCHDIR */
1663 PyDoc_STRVAR(posix_chmod__doc__,
1664 "chmod(path, mode)\n\n\
1665 Change the access permissions of a file.");
1667 static PyObject *
1668 posix_chmod(PyObject *self, PyObject *args)
1670 char *path = NULL;
1671 int i;
1672 int res;
1673 #ifdef MS_WINDOWS
1674 DWORD attr;
1675 PyUnicodeObject *po;
1676 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
1677 Py_BEGIN_ALLOW_THREADS
1678 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1679 if (attr != 0xFFFFFFFF) {
1680 if (i & _S_IWRITE)
1681 attr &= ~FILE_ATTRIBUTE_READONLY;
1682 else
1683 attr |= FILE_ATTRIBUTE_READONLY;
1684 res = SetFileAttributesW(PyUnicode_AS_UNICODE(po), attr);
1686 else
1687 res = 0;
1688 Py_END_ALLOW_THREADS
1689 if (!res)
1690 return win32_error_unicode("chmod",
1691 PyUnicode_AS_UNICODE(po));
1692 Py_INCREF(Py_None);
1693 return Py_None;
1695 /* Drop the argument parsing error as narrow strings
1696 are also valid. */
1697 PyErr_Clear();
1699 if (!PyArg_ParseTuple(args, "eti:chmod", Py_FileSystemDefaultEncoding,
1700 &path, &i))
1701 return NULL;
1702 Py_BEGIN_ALLOW_THREADS
1703 attr = GetFileAttributesA(path);
1704 if (attr != 0xFFFFFFFF) {
1705 if (i & _S_IWRITE)
1706 attr &= ~FILE_ATTRIBUTE_READONLY;
1707 else
1708 attr |= FILE_ATTRIBUTE_READONLY;
1709 res = SetFileAttributesA(path, attr);
1711 else
1712 res = 0;
1713 Py_END_ALLOW_THREADS
1714 if (!res) {
1715 win32_error("chmod", path);
1716 PyMem_Free(path);
1717 return NULL;
1719 PyMem_Free(path);
1720 Py_INCREF(Py_None);
1721 return Py_None;
1722 #else /* MS_WINDOWS */
1723 if (!PyArg_ParseTuple(args, "eti:chmod", Py_FileSystemDefaultEncoding,
1724 &path, &i))
1725 return NULL;
1726 Py_BEGIN_ALLOW_THREADS
1727 res = chmod(path, i);
1728 Py_END_ALLOW_THREADS
1729 if (res < 0)
1730 return posix_error_with_allocated_filename(path);
1731 PyMem_Free(path);
1732 Py_INCREF(Py_None);
1733 return Py_None;
1734 #endif
1737 #ifdef HAVE_FCHMOD
1738 PyDoc_STRVAR(posix_fchmod__doc__,
1739 "fchmod(fd, mode)\n\n\
1740 Change the access permissions of the file given by file\n\
1741 descriptor fd.");
1743 static PyObject *
1744 posix_fchmod(PyObject *self, PyObject *args)
1746 int fd, mode, res;
1747 if (!PyArg_ParseTuple(args, "ii:fchmod", &fd, &mode))
1748 return NULL;
1749 Py_BEGIN_ALLOW_THREADS
1750 res = fchmod(fd, mode);
1751 Py_END_ALLOW_THREADS
1752 if (res < 0)
1753 return posix_error();
1754 Py_RETURN_NONE;
1756 #endif /* HAVE_FCHMOD */
1758 #ifdef HAVE_LCHMOD
1759 PyDoc_STRVAR(posix_lchmod__doc__,
1760 "lchmod(path, mode)\n\n\
1761 Change the access permissions of a file. If path is a symlink, this\n\
1762 affects the link itself rather than the target.");
1764 static PyObject *
1765 posix_lchmod(PyObject *self, PyObject *args)
1767 char *path = NULL;
1768 int i;
1769 int res;
1770 if (!PyArg_ParseTuple(args, "eti:lchmod", Py_FileSystemDefaultEncoding,
1771 &path, &i))
1772 return NULL;
1773 Py_BEGIN_ALLOW_THREADS
1774 res = lchmod(path, i);
1775 Py_END_ALLOW_THREADS
1776 if (res < 0)
1777 return posix_error_with_allocated_filename(path);
1778 PyMem_Free(path);
1779 Py_RETURN_NONE;
1781 #endif /* HAVE_LCHMOD */
1784 #ifdef HAVE_CHFLAGS
1785 PyDoc_STRVAR(posix_chflags__doc__,
1786 "chflags(path, flags)\n\n\
1787 Set file flags.");
1789 static PyObject *
1790 posix_chflags(PyObject *self, PyObject *args)
1792 char *path;
1793 unsigned long flags;
1794 int res;
1795 if (!PyArg_ParseTuple(args, "etk:chflags",
1796 Py_FileSystemDefaultEncoding, &path, &flags))
1797 return NULL;
1798 Py_BEGIN_ALLOW_THREADS
1799 res = chflags(path, flags);
1800 Py_END_ALLOW_THREADS
1801 if (res < 0)
1802 return posix_error_with_allocated_filename(path);
1803 PyMem_Free(path);
1804 Py_INCREF(Py_None);
1805 return Py_None;
1807 #endif /* HAVE_CHFLAGS */
1809 #ifdef HAVE_LCHFLAGS
1810 PyDoc_STRVAR(posix_lchflags__doc__,
1811 "lchflags(path, flags)\n\n\
1812 Set file flags.\n\
1813 This function will not follow symbolic links.");
1815 static PyObject *
1816 posix_lchflags(PyObject *self, PyObject *args)
1818 char *path;
1819 unsigned long flags;
1820 int res;
1821 if (!PyArg_ParseTuple(args, "etk:lchflags",
1822 Py_FileSystemDefaultEncoding, &path, &flags))
1823 return NULL;
1824 Py_BEGIN_ALLOW_THREADS
1825 res = lchflags(path, flags);
1826 Py_END_ALLOW_THREADS
1827 if (res < 0)
1828 return posix_error_with_allocated_filename(path);
1829 PyMem_Free(path);
1830 Py_INCREF(Py_None);
1831 return Py_None;
1833 #endif /* HAVE_LCHFLAGS */
1835 #ifdef HAVE_CHROOT
1836 PyDoc_STRVAR(posix_chroot__doc__,
1837 "chroot(path)\n\n\
1838 Change root directory to path.");
1840 static PyObject *
1841 posix_chroot(PyObject *self, PyObject *args)
1843 return posix_1str(args, "et:chroot", chroot);
1845 #endif
1847 #ifdef HAVE_FSYNC
1848 PyDoc_STRVAR(posix_fsync__doc__,
1849 "fsync(fildes)\n\n\
1850 force write of file with filedescriptor to disk.");
1852 static PyObject *
1853 posix_fsync(PyObject *self, PyObject *fdobj)
1855 return posix_fildes(fdobj, fsync);
1857 #endif /* HAVE_FSYNC */
1859 #ifdef HAVE_FDATASYNC
1861 #ifdef __hpux
1862 extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
1863 #endif
1865 PyDoc_STRVAR(posix_fdatasync__doc__,
1866 "fdatasync(fildes)\n\n\
1867 force write of file with filedescriptor to disk.\n\
1868 does not force update of metadata.");
1870 static PyObject *
1871 posix_fdatasync(PyObject *self, PyObject *fdobj)
1873 return posix_fildes(fdobj, fdatasync);
1875 #endif /* HAVE_FDATASYNC */
1878 #ifdef HAVE_CHOWN
1879 PyDoc_STRVAR(posix_chown__doc__,
1880 "chown(path, uid, gid)\n\n\
1881 Change the owner and group id of path to the numeric uid and gid.");
1883 static PyObject *
1884 posix_chown(PyObject *self, PyObject *args)
1886 char *path = NULL;
1887 long uid, gid;
1888 int res;
1889 if (!PyArg_ParseTuple(args, "etll:chown",
1890 Py_FileSystemDefaultEncoding, &path,
1891 &uid, &gid))
1892 return NULL;
1893 Py_BEGIN_ALLOW_THREADS
1894 res = chown(path, (uid_t) uid, (gid_t) gid);
1895 Py_END_ALLOW_THREADS
1896 if (res < 0)
1897 return posix_error_with_allocated_filename(path);
1898 PyMem_Free(path);
1899 Py_INCREF(Py_None);
1900 return Py_None;
1902 #endif /* HAVE_CHOWN */
1904 #ifdef HAVE_FCHOWN
1905 PyDoc_STRVAR(posix_fchown__doc__,
1906 "fchown(fd, uid, gid)\n\n\
1907 Change the owner and group id of the file given by file descriptor\n\
1908 fd to the numeric uid and gid.");
1910 static PyObject *
1911 posix_fchown(PyObject *self, PyObject *args)
1913 int fd;
1914 long uid, gid;
1915 int res;
1916 if (!PyArg_ParseTuple(args, "ill:chown", &fd, &uid, &gid))
1917 return NULL;
1918 Py_BEGIN_ALLOW_THREADS
1919 res = fchown(fd, (uid_t) uid, (gid_t) gid);
1920 Py_END_ALLOW_THREADS
1921 if (res < 0)
1922 return posix_error();
1923 Py_RETURN_NONE;
1925 #endif /* HAVE_FCHOWN */
1927 #ifdef HAVE_LCHOWN
1928 PyDoc_STRVAR(posix_lchown__doc__,
1929 "lchown(path, uid, gid)\n\n\
1930 Change the owner and group id of path to the numeric uid and gid.\n\
1931 This function will not follow symbolic links.");
1933 static PyObject *
1934 posix_lchown(PyObject *self, PyObject *args)
1936 char *path = NULL;
1937 long uid, gid;
1938 int res;
1939 if (!PyArg_ParseTuple(args, "etll:lchown",
1940 Py_FileSystemDefaultEncoding, &path,
1941 &uid, &gid))
1942 return NULL;
1943 Py_BEGIN_ALLOW_THREADS
1944 res = lchown(path, (uid_t) uid, (gid_t) gid);
1945 Py_END_ALLOW_THREADS
1946 if (res < 0)
1947 return posix_error_with_allocated_filename(path);
1948 PyMem_Free(path);
1949 Py_INCREF(Py_None);
1950 return Py_None;
1952 #endif /* HAVE_LCHOWN */
1955 #ifdef HAVE_GETCWD
1956 PyDoc_STRVAR(posix_getcwd__doc__,
1957 "getcwd() -> path\n\n\
1958 Return a string representing the current working directory.");
1960 static PyObject *
1961 posix_getcwd(PyObject *self, PyObject *noargs)
1963 int bufsize_incr = 1024;
1964 int bufsize = 0;
1965 char *tmpbuf = NULL;
1966 char *res = NULL;
1967 PyObject *dynamic_return;
1969 Py_BEGIN_ALLOW_THREADS
1970 do {
1971 bufsize = bufsize + bufsize_incr;
1972 tmpbuf = malloc(bufsize);
1973 if (tmpbuf == NULL) {
1974 break;
1976 #if defined(PYOS_OS2) && defined(PYCC_GCC)
1977 res = _getcwd2(tmpbuf, bufsize);
1978 #else
1979 res = getcwd(tmpbuf, bufsize);
1980 #endif
1982 if (res == NULL) {
1983 free(tmpbuf);
1985 } while ((res == NULL) && (errno == ERANGE));
1986 Py_END_ALLOW_THREADS
1988 if (res == NULL)
1989 return posix_error();
1991 dynamic_return = PyString_FromString(tmpbuf);
1992 free(tmpbuf);
1994 return dynamic_return;
1997 #ifdef Py_USING_UNICODE
1998 PyDoc_STRVAR(posix_getcwdu__doc__,
1999 "getcwdu() -> path\n\n\
2000 Return a unicode string representing the current working directory.");
2002 static PyObject *
2003 posix_getcwdu(PyObject *self, PyObject *noargs)
2005 char buf[1026];
2006 char *res;
2008 #ifdef MS_WINDOWS
2009 DWORD len;
2010 wchar_t wbuf[1026];
2011 wchar_t *wbuf2 = wbuf;
2012 PyObject *resobj;
2013 Py_BEGIN_ALLOW_THREADS
2014 len = GetCurrentDirectoryW(sizeof wbuf/ sizeof wbuf[0], wbuf);
2015 /* If the buffer is large enough, len does not include the
2016 terminating \0. If the buffer is too small, len includes
2017 the space needed for the terminator. */
2018 if (len >= sizeof wbuf/ sizeof wbuf[0]) {
2019 wbuf2 = malloc(len * sizeof(wchar_t));
2020 if (wbuf2)
2021 len = GetCurrentDirectoryW(len, wbuf2);
2023 Py_END_ALLOW_THREADS
2024 if (!wbuf2) {
2025 PyErr_NoMemory();
2026 return NULL;
2028 if (!len) {
2029 if (wbuf2 != wbuf) free(wbuf2);
2030 return win32_error("getcwdu", NULL);
2032 resobj = PyUnicode_FromWideChar(wbuf2, len);
2033 if (wbuf2 != wbuf) free(wbuf2);
2034 return resobj;
2035 #endif /* MS_WINDOWS */
2037 Py_BEGIN_ALLOW_THREADS
2038 #if defined(PYOS_OS2) && defined(PYCC_GCC)
2039 res = _getcwd2(buf, sizeof buf);
2040 #else
2041 res = getcwd(buf, sizeof buf);
2042 #endif
2043 Py_END_ALLOW_THREADS
2044 if (res == NULL)
2045 return posix_error();
2046 return PyUnicode_Decode(buf, strlen(buf), Py_FileSystemDefaultEncoding,"strict");
2048 #endif /* Py_USING_UNICODE */
2049 #endif /* HAVE_GETCWD */
2052 #ifdef HAVE_LINK
2053 PyDoc_STRVAR(posix_link__doc__,
2054 "link(src, dst)\n\n\
2055 Create a hard link to a file.");
2057 static PyObject *
2058 posix_link(PyObject *self, PyObject *args)
2060 return posix_2str(args, "etet:link", link);
2062 #endif /* HAVE_LINK */
2065 PyDoc_STRVAR(posix_listdir__doc__,
2066 "listdir(path) -> list_of_strings\n\n\
2067 Return a list containing the names of the entries in the directory.\n\
2069 path: path of directory to list\n\
2071 The list is in arbitrary order. It does not include the special\n\
2072 entries '.' and '..' even if they are present in the directory.");
2074 static PyObject *
2075 posix_listdir(PyObject *self, PyObject *args)
2077 /* XXX Should redo this putting the (now four) versions of opendir
2078 in separate files instead of having them all here... */
2079 #if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
2081 PyObject *d, *v;
2082 HANDLE hFindFile;
2083 BOOL result;
2084 WIN32_FIND_DATA FileData;
2085 char namebuf[MAX_PATH+5]; /* Overallocate for \\*.*\0 */
2086 char *bufptr = namebuf;
2087 Py_ssize_t len = sizeof(namebuf)-5; /* only claim to have space for MAX_PATH */
2089 PyObject *po;
2090 if (PyArg_ParseTuple(args, "U:listdir", &po)) {
2091 WIN32_FIND_DATAW wFileData;
2092 Py_UNICODE *wnamebuf;
2093 /* Overallocate for \\*.*\0 */
2094 len = PyUnicode_GET_SIZE(po);
2095 wnamebuf = malloc((len + 5) * sizeof(wchar_t));
2096 if (!wnamebuf) {
2097 PyErr_NoMemory();
2098 return NULL;
2100 wcscpy(wnamebuf, PyUnicode_AS_UNICODE(po));
2101 if (len > 0) {
2102 Py_UNICODE wch = wnamebuf[len-1];
2103 if (wch != L'/' && wch != L'\\' && wch != L':')
2104 wnamebuf[len++] = L'\\';
2105 wcscpy(wnamebuf + len, L"*.*");
2107 if ((d = PyList_New(0)) == NULL) {
2108 free(wnamebuf);
2109 return NULL;
2111 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
2112 if (hFindFile == INVALID_HANDLE_VALUE) {
2113 int error = GetLastError();
2114 if (error == ERROR_FILE_NOT_FOUND) {
2115 free(wnamebuf);
2116 return d;
2118 Py_DECREF(d);
2119 win32_error_unicode("FindFirstFileW", wnamebuf);
2120 free(wnamebuf);
2121 return NULL;
2123 do {
2124 /* Skip over . and .. */
2125 if (wcscmp(wFileData.cFileName, L".") != 0 &&
2126 wcscmp(wFileData.cFileName, L"..") != 0) {
2127 v = PyUnicode_FromUnicode(wFileData.cFileName, wcslen(wFileData.cFileName));
2128 if (v == NULL) {
2129 Py_DECREF(d);
2130 d = NULL;
2131 break;
2133 if (PyList_Append(d, v) != 0) {
2134 Py_DECREF(v);
2135 Py_DECREF(d);
2136 d = NULL;
2137 break;
2139 Py_DECREF(v);
2141 Py_BEGIN_ALLOW_THREADS
2142 result = FindNextFileW(hFindFile, &wFileData);
2143 Py_END_ALLOW_THREADS
2144 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2145 it got to the end of the directory. */
2146 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2147 Py_DECREF(d);
2148 win32_error_unicode("FindNextFileW", wnamebuf);
2149 FindClose(hFindFile);
2150 free(wnamebuf);
2151 return NULL;
2153 } while (result == TRUE);
2155 if (FindClose(hFindFile) == FALSE) {
2156 Py_DECREF(d);
2157 win32_error_unicode("FindClose", wnamebuf);
2158 free(wnamebuf);
2159 return NULL;
2161 free(wnamebuf);
2162 return d;
2164 /* Drop the argument parsing error as narrow strings
2165 are also valid. */
2166 PyErr_Clear();
2168 if (!PyArg_ParseTuple(args, "et#:listdir",
2169 Py_FileSystemDefaultEncoding, &bufptr, &len))
2170 return NULL;
2171 if (len > 0) {
2172 char ch = namebuf[len-1];
2173 if (ch != SEP && ch != ALTSEP && ch != ':')
2174 namebuf[len++] = '/';
2175 strcpy(namebuf + len, "*.*");
2178 if ((d = PyList_New(0)) == NULL)
2179 return NULL;
2181 hFindFile = FindFirstFile(namebuf, &FileData);
2182 if (hFindFile == INVALID_HANDLE_VALUE) {
2183 int error = GetLastError();
2184 if (error == ERROR_FILE_NOT_FOUND)
2185 return d;
2186 Py_DECREF(d);
2187 return win32_error("FindFirstFile", namebuf);
2189 do {
2190 /* Skip over . and .. */
2191 if (strcmp(FileData.cFileName, ".") != 0 &&
2192 strcmp(FileData.cFileName, "..") != 0) {
2193 v = PyString_FromString(FileData.cFileName);
2194 if (v == NULL) {
2195 Py_DECREF(d);
2196 d = NULL;
2197 break;
2199 if (PyList_Append(d, v) != 0) {
2200 Py_DECREF(v);
2201 Py_DECREF(d);
2202 d = NULL;
2203 break;
2205 Py_DECREF(v);
2207 Py_BEGIN_ALLOW_THREADS
2208 result = FindNextFile(hFindFile, &FileData);
2209 Py_END_ALLOW_THREADS
2210 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2211 it got to the end of the directory. */
2212 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2213 Py_DECREF(d);
2214 win32_error("FindNextFile", namebuf);
2215 FindClose(hFindFile);
2216 return NULL;
2218 } while (result == TRUE);
2220 if (FindClose(hFindFile) == FALSE) {
2221 Py_DECREF(d);
2222 return win32_error("FindClose", namebuf);
2225 return d;
2227 #elif defined(PYOS_OS2)
2229 #ifndef MAX_PATH
2230 #define MAX_PATH CCHMAXPATH
2231 #endif
2232 char *name, *pt;
2233 Py_ssize_t len;
2234 PyObject *d, *v;
2235 char namebuf[MAX_PATH+5];
2236 HDIR hdir = 1;
2237 ULONG srchcnt = 1;
2238 FILEFINDBUF3 ep;
2239 APIRET rc;
2241 if (!PyArg_ParseTuple(args, "t#:listdir", &name, &len))
2242 return NULL;
2243 if (len >= MAX_PATH) {
2244 PyErr_SetString(PyExc_ValueError, "path too long");
2245 return NULL;
2247 strcpy(namebuf, name);
2248 for (pt = namebuf; *pt; pt++)
2249 if (*pt == ALTSEP)
2250 *pt = SEP;
2251 if (namebuf[len-1] != SEP)
2252 namebuf[len++] = SEP;
2253 strcpy(namebuf + len, "*.*");
2255 if ((d = PyList_New(0)) == NULL)
2256 return NULL;
2258 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
2259 &hdir, /* Handle to Use While Search Directory */
2260 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
2261 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
2262 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
2263 FIL_STANDARD); /* Format of Entry (EAs or Not) */
2265 if (rc != NO_ERROR) {
2266 errno = ENOENT;
2267 return posix_error_with_filename(name);
2270 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
2271 do {
2272 if (ep.achName[0] == '.'
2273 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
2274 continue; /* Skip Over "." and ".." Names */
2276 strcpy(namebuf, ep.achName);
2278 /* Leave Case of Name Alone -- In Native Form */
2279 /* (Removed Forced Lowercasing Code) */
2281 v = PyString_FromString(namebuf);
2282 if (v == NULL) {
2283 Py_DECREF(d);
2284 d = NULL;
2285 break;
2287 if (PyList_Append(d, v) != 0) {
2288 Py_DECREF(v);
2289 Py_DECREF(d);
2290 d = NULL;
2291 break;
2293 Py_DECREF(v);
2294 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
2297 return d;
2298 #else
2300 char *name = NULL;
2301 PyObject *d, *v;
2302 DIR *dirp;
2303 struct dirent *ep;
2304 int arg_is_unicode = 1;
2306 errno = 0;
2307 if (!PyArg_ParseTuple(args, "U:listdir", &v)) {
2308 arg_is_unicode = 0;
2309 PyErr_Clear();
2311 if (!PyArg_ParseTuple(args, "et:listdir", Py_FileSystemDefaultEncoding, &name))
2312 return NULL;
2313 if ((dirp = opendir(name)) == NULL) {
2314 return posix_error_with_allocated_filename(name);
2316 if ((d = PyList_New(0)) == NULL) {
2317 closedir(dirp);
2318 PyMem_Free(name);
2319 return NULL;
2321 for (;;) {
2322 errno = 0;
2323 Py_BEGIN_ALLOW_THREADS
2324 ep = readdir(dirp);
2325 Py_END_ALLOW_THREADS
2326 if (ep == NULL) {
2327 if (errno == 0) {
2328 break;
2329 } else {
2330 closedir(dirp);
2331 Py_DECREF(d);
2332 return posix_error_with_allocated_filename(name);
2335 if (ep->d_name[0] == '.' &&
2336 (NAMLEN(ep) == 1 ||
2337 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
2338 continue;
2339 v = PyString_FromStringAndSize(ep->d_name, NAMLEN(ep));
2340 if (v == NULL) {
2341 Py_DECREF(d);
2342 d = NULL;
2343 break;
2345 #ifdef Py_USING_UNICODE
2346 if (arg_is_unicode) {
2347 PyObject *w;
2349 w = PyUnicode_FromEncodedObject(v,
2350 Py_FileSystemDefaultEncoding,
2351 "strict");
2352 if (w != NULL) {
2353 Py_DECREF(v);
2354 v = w;
2356 else {
2357 /* fall back to the original byte string, as
2358 discussed in patch #683592 */
2359 PyErr_Clear();
2362 #endif
2363 if (PyList_Append(d, v) != 0) {
2364 Py_DECREF(v);
2365 Py_DECREF(d);
2366 d = NULL;
2367 break;
2369 Py_DECREF(v);
2371 closedir(dirp);
2372 PyMem_Free(name);
2374 return d;
2376 #endif /* which OS */
2377 } /* end of posix_listdir */
2379 #ifdef MS_WINDOWS
2380 /* A helper function for abspath on win32 */
2381 static PyObject *
2382 posix__getfullpathname(PyObject *self, PyObject *args)
2384 /* assume encoded strings won't more than double no of chars */
2385 char inbuf[MAX_PATH*2];
2386 char *inbufp = inbuf;
2387 Py_ssize_t insize = sizeof(inbuf);
2388 char outbuf[MAX_PATH*2];
2389 char *temp;
2391 PyUnicodeObject *po;
2392 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po)) {
2393 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
2394 Py_UNICODE woutbuf[MAX_PATH*2], *woutbufp = woutbuf;
2395 Py_UNICODE *wtemp;
2396 DWORD result;
2397 PyObject *v;
2398 result = GetFullPathNameW(wpath,
2399 sizeof(woutbuf)/sizeof(woutbuf[0]),
2400 woutbuf, &wtemp);
2401 if (result > sizeof(woutbuf)/sizeof(woutbuf[0])) {
2402 woutbufp = malloc(result * sizeof(Py_UNICODE));
2403 if (!woutbufp)
2404 return PyErr_NoMemory();
2405 result = GetFullPathNameW(wpath, result, woutbufp, &wtemp);
2407 if (result)
2408 v = PyUnicode_FromUnicode(woutbufp, wcslen(woutbufp));
2409 else
2410 v = win32_error_unicode("GetFullPathNameW", wpath);
2411 if (woutbufp != woutbuf)
2412 free(woutbufp);
2413 return v;
2415 /* Drop the argument parsing error as narrow strings
2416 are also valid. */
2417 PyErr_Clear();
2419 if (!PyArg_ParseTuple (args, "et#:_getfullpathname",
2420 Py_FileSystemDefaultEncoding, &inbufp,
2421 &insize))
2422 return NULL;
2423 if (!GetFullPathName(inbuf, sizeof(outbuf)/sizeof(outbuf[0]),
2424 outbuf, &temp))
2425 return win32_error("GetFullPathName", inbuf);
2426 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
2427 return PyUnicode_Decode(outbuf, strlen(outbuf),
2428 Py_FileSystemDefaultEncoding, NULL);
2430 return PyString_FromString(outbuf);
2431 } /* end of posix__getfullpathname */
2432 #endif /* MS_WINDOWS */
2434 PyDoc_STRVAR(posix_mkdir__doc__,
2435 "mkdir(path [, mode=0777])\n\n\
2436 Create a directory.");
2438 static PyObject *
2439 posix_mkdir(PyObject *self, PyObject *args)
2441 int res;
2442 char *path = NULL;
2443 int mode = 0777;
2445 #ifdef MS_WINDOWS
2446 PyUnicodeObject *po;
2447 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode)) {
2448 Py_BEGIN_ALLOW_THREADS
2449 /* PyUnicode_AS_UNICODE OK without thread lock as
2450 it is a simple dereference. */
2451 res = CreateDirectoryW(PyUnicode_AS_UNICODE(po), NULL);
2452 Py_END_ALLOW_THREADS
2453 if (!res)
2454 return win32_error_unicode("mkdir", PyUnicode_AS_UNICODE(po));
2455 Py_INCREF(Py_None);
2456 return Py_None;
2458 /* Drop the argument parsing error as narrow strings
2459 are also valid. */
2460 PyErr_Clear();
2461 if (!PyArg_ParseTuple(args, "et|i:mkdir",
2462 Py_FileSystemDefaultEncoding, &path, &mode))
2463 return NULL;
2464 Py_BEGIN_ALLOW_THREADS
2465 /* PyUnicode_AS_UNICODE OK without thread lock as
2466 it is a simple dereference. */
2467 res = CreateDirectoryA(path, NULL);
2468 Py_END_ALLOW_THREADS
2469 if (!res) {
2470 win32_error("mkdir", path);
2471 PyMem_Free(path);
2472 return NULL;
2474 PyMem_Free(path);
2475 Py_INCREF(Py_None);
2476 return Py_None;
2477 #else /* MS_WINDOWS */
2479 if (!PyArg_ParseTuple(args, "et|i:mkdir",
2480 Py_FileSystemDefaultEncoding, &path, &mode))
2481 return NULL;
2482 Py_BEGIN_ALLOW_THREADS
2483 #if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
2484 res = mkdir(path);
2485 #else
2486 res = mkdir(path, mode);
2487 #endif
2488 Py_END_ALLOW_THREADS
2489 if (res < 0)
2490 return posix_error_with_allocated_filename(path);
2491 PyMem_Free(path);
2492 Py_INCREF(Py_None);
2493 return Py_None;
2494 #endif /* MS_WINDOWS */
2498 /* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
2499 #if defined(HAVE_SYS_RESOURCE_H)
2500 #include <sys/resource.h>
2501 #endif
2504 #ifdef HAVE_NICE
2505 PyDoc_STRVAR(posix_nice__doc__,
2506 "nice(inc) -> new_priority\n\n\
2507 Decrease the priority of process by inc and return the new priority.");
2509 static PyObject *
2510 posix_nice(PyObject *self, PyObject *args)
2512 int increment, value;
2514 if (!PyArg_ParseTuple(args, "i:nice", &increment))
2515 return NULL;
2517 /* There are two flavours of 'nice': one that returns the new
2518 priority (as required by almost all standards out there) and the
2519 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
2520 the use of getpriority() to get the new priority.
2522 If we are of the nice family that returns the new priority, we
2523 need to clear errno before the call, and check if errno is filled
2524 before calling posix_error() on a returnvalue of -1, because the
2525 -1 may be the actual new priority! */
2527 errno = 0;
2528 value = nice(increment);
2529 #if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
2530 if (value == 0)
2531 value = getpriority(PRIO_PROCESS, 0);
2532 #endif
2533 if (value == -1 && errno != 0)
2534 /* either nice() or getpriority() returned an error */
2535 return posix_error();
2536 return PyInt_FromLong((long) value);
2538 #endif /* HAVE_NICE */
2540 PyDoc_STRVAR(posix_rename__doc__,
2541 "rename(old, new)\n\n\
2542 Rename a file or directory.");
2544 static PyObject *
2545 posix_rename(PyObject *self, PyObject *args)
2547 #ifdef MS_WINDOWS
2548 PyObject *o1, *o2;
2549 char *p1, *p2;
2550 BOOL result;
2551 if (!PyArg_ParseTuple(args, "OO:rename", &o1, &o2))
2552 goto error;
2553 if (!convert_to_unicode(&o1))
2554 goto error;
2555 if (!convert_to_unicode(&o2)) {
2556 Py_DECREF(o1);
2557 goto error;
2559 Py_BEGIN_ALLOW_THREADS
2560 result = MoveFileW(PyUnicode_AsUnicode(o1),
2561 PyUnicode_AsUnicode(o2));
2562 Py_END_ALLOW_THREADS
2563 Py_DECREF(o1);
2564 Py_DECREF(o2);
2565 if (!result)
2566 return win32_error("rename", NULL);
2567 Py_INCREF(Py_None);
2568 return Py_None;
2569 error:
2570 PyErr_Clear();
2571 if (!PyArg_ParseTuple(args, "ss:rename", &p1, &p2))
2572 return NULL;
2573 Py_BEGIN_ALLOW_THREADS
2574 result = MoveFileA(p1, p2);
2575 Py_END_ALLOW_THREADS
2576 if (!result)
2577 return win32_error("rename", NULL);
2578 Py_INCREF(Py_None);
2579 return Py_None;
2580 #else
2581 return posix_2str(args, "etet:rename", rename);
2582 #endif
2586 PyDoc_STRVAR(posix_rmdir__doc__,
2587 "rmdir(path)\n\n\
2588 Remove a directory.");
2590 static PyObject *
2591 posix_rmdir(PyObject *self, PyObject *args)
2593 #ifdef MS_WINDOWS
2594 return win32_1str(args, "rmdir", "s:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
2595 #else
2596 return posix_1str(args, "et:rmdir", rmdir);
2597 #endif
2601 PyDoc_STRVAR(posix_stat__doc__,
2602 "stat(path) -> stat result\n\n\
2603 Perform a stat system call on the given path.");
2605 static PyObject *
2606 posix_stat(PyObject *self, PyObject *args)
2608 #ifdef MS_WINDOWS
2609 return posix_do_stat(self, args, "et:stat", STAT, "U:stat", win32_wstat);
2610 #else
2611 return posix_do_stat(self, args, "et:stat", STAT, NULL, NULL);
2612 #endif
2616 #ifdef HAVE_SYSTEM
2617 PyDoc_STRVAR(posix_system__doc__,
2618 "system(command) -> exit_status\n\n\
2619 Execute the command (a string) in a subshell.");
2621 static PyObject *
2622 posix_system(PyObject *self, PyObject *args)
2624 char *command;
2625 long sts;
2626 if (!PyArg_ParseTuple(args, "s:system", &command))
2627 return NULL;
2628 Py_BEGIN_ALLOW_THREADS
2629 sts = system(command);
2630 Py_END_ALLOW_THREADS
2631 return PyInt_FromLong(sts);
2633 #endif
2636 PyDoc_STRVAR(posix_umask__doc__,
2637 "umask(new_mask) -> old_mask\n\n\
2638 Set the current numeric umask and return the previous umask.");
2640 static PyObject *
2641 posix_umask(PyObject *self, PyObject *args)
2643 int i;
2644 if (!PyArg_ParseTuple(args, "i:umask", &i))
2645 return NULL;
2646 i = (int)umask(i);
2647 if (i < 0)
2648 return posix_error();
2649 return PyInt_FromLong((long)i);
2653 PyDoc_STRVAR(posix_unlink__doc__,
2654 "unlink(path)\n\n\
2655 Remove a file (same as remove(path)).");
2657 PyDoc_STRVAR(posix_remove__doc__,
2658 "remove(path)\n\n\
2659 Remove a file (same as unlink(path)).");
2661 static PyObject *
2662 posix_unlink(PyObject *self, PyObject *args)
2664 #ifdef MS_WINDOWS
2665 return win32_1str(args, "remove", "s:remove", DeleteFileA, "U:remove", DeleteFileW);
2666 #else
2667 return posix_1str(args, "et:remove", unlink);
2668 #endif
2672 #ifdef HAVE_UNAME
2673 PyDoc_STRVAR(posix_uname__doc__,
2674 "uname() -> (sysname, nodename, release, version, machine)\n\n\
2675 Return a tuple identifying the current operating system.");
2677 static PyObject *
2678 posix_uname(PyObject *self, PyObject *noargs)
2680 struct utsname u;
2681 int res;
2683 Py_BEGIN_ALLOW_THREADS
2684 res = uname(&u);
2685 Py_END_ALLOW_THREADS
2686 if (res < 0)
2687 return posix_error();
2688 return Py_BuildValue("(sssss)",
2689 u.sysname,
2690 u.nodename,
2691 u.release,
2692 u.version,
2693 u.machine);
2695 #endif /* HAVE_UNAME */
2697 static int
2698 extract_time(PyObject *t, long* sec, long* usec)
2700 long intval;
2701 if (PyFloat_Check(t)) {
2702 double tval = PyFloat_AsDouble(t);
2703 PyObject *intobj = Py_TYPE(t)->tp_as_number->nb_int(t);
2704 if (!intobj)
2705 return -1;
2706 intval = PyInt_AsLong(intobj);
2707 Py_DECREF(intobj);
2708 if (intval == -1 && PyErr_Occurred())
2709 return -1;
2710 *sec = intval;
2711 *usec = (long)((tval - intval) * 1e6); /* can't exceed 1000000 */
2712 if (*usec < 0)
2713 /* If rounding gave us a negative number,
2714 truncate. */
2715 *usec = 0;
2716 return 0;
2718 intval = PyInt_AsLong(t);
2719 if (intval == -1 && PyErr_Occurred())
2720 return -1;
2721 *sec = intval;
2722 *usec = 0;
2723 return 0;
2726 PyDoc_STRVAR(posix_utime__doc__,
2727 "utime(path, (atime, mtime))\n\
2728 utime(path, None)\n\n\
2729 Set the access and modified time of the file to the given values. If the\n\
2730 second form is used, set the access and modified times to the current time.");
2732 static PyObject *
2733 posix_utime(PyObject *self, PyObject *args)
2735 #ifdef MS_WINDOWS
2736 PyObject *arg;
2737 PyUnicodeObject *obwpath;
2738 wchar_t *wpath = NULL;
2739 char *apath = NULL;
2740 HANDLE hFile;
2741 long atimesec, mtimesec, ausec, musec;
2742 FILETIME atime, mtime;
2743 PyObject *result = NULL;
2745 if (PyArg_ParseTuple(args, "UO|:utime", &obwpath, &arg)) {
2746 wpath = PyUnicode_AS_UNICODE(obwpath);
2747 Py_BEGIN_ALLOW_THREADS
2748 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
2749 NULL, OPEN_EXISTING,
2750 FILE_FLAG_BACKUP_SEMANTICS, NULL);
2751 Py_END_ALLOW_THREADS
2752 if (hFile == INVALID_HANDLE_VALUE)
2753 return win32_error_unicode("utime", wpath);
2754 } else
2755 /* Drop the argument parsing error as narrow strings
2756 are also valid. */
2757 PyErr_Clear();
2759 if (!wpath) {
2760 if (!PyArg_ParseTuple(args, "etO:utime",
2761 Py_FileSystemDefaultEncoding, &apath, &arg))
2762 return NULL;
2763 Py_BEGIN_ALLOW_THREADS
2764 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
2765 NULL, OPEN_EXISTING,
2766 FILE_FLAG_BACKUP_SEMANTICS, NULL);
2767 Py_END_ALLOW_THREADS
2768 if (hFile == INVALID_HANDLE_VALUE) {
2769 win32_error("utime", apath);
2770 PyMem_Free(apath);
2771 return NULL;
2773 PyMem_Free(apath);
2776 if (arg == Py_None) {
2777 SYSTEMTIME now;
2778 GetSystemTime(&now);
2779 if (!SystemTimeToFileTime(&now, &mtime) ||
2780 !SystemTimeToFileTime(&now, &atime)) {
2781 win32_error("utime", NULL);
2782 goto done;
2785 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
2786 PyErr_SetString(PyExc_TypeError,
2787 "utime() arg 2 must be a tuple (atime, mtime)");
2788 goto done;
2790 else {
2791 if (extract_time(PyTuple_GET_ITEM(arg, 0),
2792 &atimesec, &ausec) == -1)
2793 goto done;
2794 time_t_to_FILE_TIME(atimesec, 1000*ausec, &atime);
2795 if (extract_time(PyTuple_GET_ITEM(arg, 1),
2796 &mtimesec, &musec) == -1)
2797 goto done;
2798 time_t_to_FILE_TIME(mtimesec, 1000*musec, &mtime);
2800 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
2801 /* Avoid putting the file name into the error here,
2802 as that may confuse the user into believing that
2803 something is wrong with the file, when it also
2804 could be the time stamp that gives a problem. */
2805 win32_error("utime", NULL);
2807 Py_INCREF(Py_None);
2808 result = Py_None;
2809 done:
2810 CloseHandle(hFile);
2811 return result;
2812 #else /* MS_WINDOWS */
2814 char *path = NULL;
2815 long atime, mtime, ausec, musec;
2816 int res;
2817 PyObject* arg;
2819 #if defined(HAVE_UTIMES)
2820 struct timeval buf[2];
2821 #define ATIME buf[0].tv_sec
2822 #define MTIME buf[1].tv_sec
2823 #elif defined(HAVE_UTIME_H)
2824 /* XXX should define struct utimbuf instead, above */
2825 struct utimbuf buf;
2826 #define ATIME buf.actime
2827 #define MTIME buf.modtime
2828 #define UTIME_ARG &buf
2829 #else /* HAVE_UTIMES */
2830 time_t buf[2];
2831 #define ATIME buf[0]
2832 #define MTIME buf[1]
2833 #define UTIME_ARG buf
2834 #endif /* HAVE_UTIMES */
2837 if (!PyArg_ParseTuple(args, "etO:utime",
2838 Py_FileSystemDefaultEncoding, &path, &arg))
2839 return NULL;
2840 if (arg == Py_None) {
2841 /* optional time values not given */
2842 Py_BEGIN_ALLOW_THREADS
2843 res = utime(path, NULL);
2844 Py_END_ALLOW_THREADS
2846 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
2847 PyErr_SetString(PyExc_TypeError,
2848 "utime() arg 2 must be a tuple (atime, mtime)");
2849 PyMem_Free(path);
2850 return NULL;
2852 else {
2853 if (extract_time(PyTuple_GET_ITEM(arg, 0),
2854 &atime, &ausec) == -1) {
2855 PyMem_Free(path);
2856 return NULL;
2858 if (extract_time(PyTuple_GET_ITEM(arg, 1),
2859 &mtime, &musec) == -1) {
2860 PyMem_Free(path);
2861 return NULL;
2863 ATIME = atime;
2864 MTIME = mtime;
2865 #ifdef HAVE_UTIMES
2866 buf[0].tv_usec = ausec;
2867 buf[1].tv_usec = musec;
2868 Py_BEGIN_ALLOW_THREADS
2869 res = utimes(path, buf);
2870 Py_END_ALLOW_THREADS
2871 #else
2872 Py_BEGIN_ALLOW_THREADS
2873 res = utime(path, UTIME_ARG);
2874 Py_END_ALLOW_THREADS
2875 #endif /* HAVE_UTIMES */
2877 if (res < 0) {
2878 return posix_error_with_allocated_filename(path);
2880 PyMem_Free(path);
2881 Py_INCREF(Py_None);
2882 return Py_None;
2883 #undef UTIME_ARG
2884 #undef ATIME
2885 #undef MTIME
2886 #endif /* MS_WINDOWS */
2890 /* Process operations */
2892 PyDoc_STRVAR(posix__exit__doc__,
2893 "_exit(status)\n\n\
2894 Exit to the system with specified status, without normal exit processing.");
2896 static PyObject *
2897 posix__exit(PyObject *self, PyObject *args)
2899 int sts;
2900 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
2901 return NULL;
2902 _exit(sts);
2903 return NULL; /* Make gcc -Wall happy */
2906 #if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
2907 static void
2908 free_string_array(char **array, Py_ssize_t count)
2910 Py_ssize_t i;
2911 for (i = 0; i < count; i++)
2912 PyMem_Free(array[i]);
2913 PyMem_DEL(array);
2915 #endif
2918 #ifdef HAVE_EXECV
2919 PyDoc_STRVAR(posix_execv__doc__,
2920 "execv(path, args)\n\n\
2921 Execute an executable path with arguments, replacing current process.\n\
2923 path: path of executable file\n\
2924 args: tuple or list of strings");
2926 static PyObject *
2927 posix_execv(PyObject *self, PyObject *args)
2929 char *path;
2930 PyObject *argv;
2931 char **argvlist;
2932 Py_ssize_t i, argc;
2933 PyObject *(*getitem)(PyObject *, Py_ssize_t);
2935 /* execv has two arguments: (path, argv), where
2936 argv is a list or tuple of strings. */
2938 if (!PyArg_ParseTuple(args, "etO:execv",
2939 Py_FileSystemDefaultEncoding,
2940 &path, &argv))
2941 return NULL;
2942 if (PyList_Check(argv)) {
2943 argc = PyList_Size(argv);
2944 getitem = PyList_GetItem;
2946 else if (PyTuple_Check(argv)) {
2947 argc = PyTuple_Size(argv);
2948 getitem = PyTuple_GetItem;
2950 else {
2951 PyErr_SetString(PyExc_TypeError, "execv() arg 2 must be a tuple or list");
2952 PyMem_Free(path);
2953 return NULL;
2956 argvlist = PyMem_NEW(char *, argc+1);
2957 if (argvlist == NULL) {
2958 PyMem_Free(path);
2959 return PyErr_NoMemory();
2961 for (i = 0; i < argc; i++) {
2962 if (!PyArg_Parse((*getitem)(argv, i), "et",
2963 Py_FileSystemDefaultEncoding,
2964 &argvlist[i])) {
2965 free_string_array(argvlist, i);
2966 PyErr_SetString(PyExc_TypeError,
2967 "execv() arg 2 must contain only strings");
2968 PyMem_Free(path);
2969 return NULL;
2973 argvlist[argc] = NULL;
2975 execv(path, argvlist);
2977 /* If we get here it's definitely an error */
2979 free_string_array(argvlist, argc);
2980 PyMem_Free(path);
2981 return posix_error();
2985 PyDoc_STRVAR(posix_execve__doc__,
2986 "execve(path, args, env)\n\n\
2987 Execute a path with arguments and environment, replacing current process.\n\
2989 path: path of executable file\n\
2990 args: tuple or list of arguments\n\
2991 env: dictionary of strings mapping to strings");
2993 static PyObject *
2994 posix_execve(PyObject *self, PyObject *args)
2996 char *path;
2997 PyObject *argv, *env;
2998 char **argvlist;
2999 char **envlist;
3000 PyObject *key, *val, *keys=NULL, *vals=NULL;
3001 Py_ssize_t i, pos, argc, envc;
3002 PyObject *(*getitem)(PyObject *, Py_ssize_t);
3003 Py_ssize_t lastarg = 0;
3005 /* execve has three arguments: (path, argv, env), where
3006 argv is a list or tuple of strings and env is a dictionary
3007 like posix.environ. */
3009 if (!PyArg_ParseTuple(args, "etOO:execve",
3010 Py_FileSystemDefaultEncoding,
3011 &path, &argv, &env))
3012 return NULL;
3013 if (PyList_Check(argv)) {
3014 argc = PyList_Size(argv);
3015 getitem = PyList_GetItem;
3017 else if (PyTuple_Check(argv)) {
3018 argc = PyTuple_Size(argv);
3019 getitem = PyTuple_GetItem;
3021 else {
3022 PyErr_SetString(PyExc_TypeError,
3023 "execve() arg 2 must be a tuple or list");
3024 goto fail_0;
3026 if (!PyMapping_Check(env)) {
3027 PyErr_SetString(PyExc_TypeError,
3028 "execve() arg 3 must be a mapping object");
3029 goto fail_0;
3032 argvlist = PyMem_NEW(char *, argc+1);
3033 if (argvlist == NULL) {
3034 PyErr_NoMemory();
3035 goto fail_0;
3037 for (i = 0; i < argc; i++) {
3038 if (!PyArg_Parse((*getitem)(argv, i),
3039 "et;execve() arg 2 must contain only strings",
3040 Py_FileSystemDefaultEncoding,
3041 &argvlist[i]))
3043 lastarg = i;
3044 goto fail_1;
3047 lastarg = argc;
3048 argvlist[argc] = NULL;
3050 i = PyMapping_Size(env);
3051 if (i < 0)
3052 goto fail_1;
3053 envlist = PyMem_NEW(char *, i + 1);
3054 if (envlist == NULL) {
3055 PyErr_NoMemory();
3056 goto fail_1;
3058 envc = 0;
3059 keys = PyMapping_Keys(env);
3060 vals = PyMapping_Values(env);
3061 if (!keys || !vals)
3062 goto fail_2;
3063 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3064 PyErr_SetString(PyExc_TypeError,
3065 "execve(): env.keys() or env.values() is not a list");
3066 goto fail_2;
3069 for (pos = 0; pos < i; pos++) {
3070 char *p, *k, *v;
3071 size_t len;
3073 key = PyList_GetItem(keys, pos);
3074 val = PyList_GetItem(vals, pos);
3075 if (!key || !val)
3076 goto fail_2;
3078 if (!PyArg_Parse(
3079 key,
3080 "s;execve() arg 3 contains a non-string key",
3081 &k) ||
3082 !PyArg_Parse(
3083 val,
3084 "s;execve() arg 3 contains a non-string value",
3085 &v))
3087 goto fail_2;
3090 #if defined(PYOS_OS2)
3091 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
3092 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
3093 #endif
3094 len = PyString_Size(key) + PyString_Size(val) + 2;
3095 p = PyMem_NEW(char, len);
3096 if (p == NULL) {
3097 PyErr_NoMemory();
3098 goto fail_2;
3100 PyOS_snprintf(p, len, "%s=%s", k, v);
3101 envlist[envc++] = p;
3102 #if defined(PYOS_OS2)
3104 #endif
3106 envlist[envc] = 0;
3108 execve(path, argvlist, envlist);
3110 /* If we get here it's definitely an error */
3112 (void) posix_error();
3114 fail_2:
3115 while (--envc >= 0)
3116 PyMem_DEL(envlist[envc]);
3117 PyMem_DEL(envlist);
3118 fail_1:
3119 free_string_array(argvlist, lastarg);
3120 Py_XDECREF(vals);
3121 Py_XDECREF(keys);
3122 fail_0:
3123 PyMem_Free(path);
3124 return NULL;
3126 #endif /* HAVE_EXECV */
3129 #ifdef HAVE_SPAWNV
3130 PyDoc_STRVAR(posix_spawnv__doc__,
3131 "spawnv(mode, path, args)\n\n\
3132 Execute the program 'path' in a new process.\n\
3134 mode: mode of process creation\n\
3135 path: path of executable file\n\
3136 args: tuple or list of strings");
3138 static PyObject *
3139 posix_spawnv(PyObject *self, PyObject *args)
3141 char *path;
3142 PyObject *argv;
3143 char **argvlist;
3144 int mode, i;
3145 Py_ssize_t argc;
3146 Py_intptr_t spawnval;
3147 PyObject *(*getitem)(PyObject *, Py_ssize_t);
3149 /* spawnv has three arguments: (mode, path, argv), where
3150 argv is a list or tuple of strings. */
3152 if (!PyArg_ParseTuple(args, "ietO:spawnv", &mode,
3153 Py_FileSystemDefaultEncoding,
3154 &path, &argv))
3155 return NULL;
3156 if (PyList_Check(argv)) {
3157 argc = PyList_Size(argv);
3158 getitem = PyList_GetItem;
3160 else if (PyTuple_Check(argv)) {
3161 argc = PyTuple_Size(argv);
3162 getitem = PyTuple_GetItem;
3164 else {
3165 PyErr_SetString(PyExc_TypeError,
3166 "spawnv() arg 2 must be a tuple or list");
3167 PyMem_Free(path);
3168 return NULL;
3171 argvlist = PyMem_NEW(char *, argc+1);
3172 if (argvlist == NULL) {
3173 PyMem_Free(path);
3174 return PyErr_NoMemory();
3176 for (i = 0; i < argc; i++) {
3177 if (!PyArg_Parse((*getitem)(argv, i), "et",
3178 Py_FileSystemDefaultEncoding,
3179 &argvlist[i])) {
3180 free_string_array(argvlist, i);
3181 PyErr_SetString(
3182 PyExc_TypeError,
3183 "spawnv() arg 2 must contain only strings");
3184 PyMem_Free(path);
3185 return NULL;
3188 argvlist[argc] = NULL;
3190 #if defined(PYOS_OS2) && defined(PYCC_GCC)
3191 Py_BEGIN_ALLOW_THREADS
3192 spawnval = spawnv(mode, path, argvlist);
3193 Py_END_ALLOW_THREADS
3194 #else
3195 if (mode == _OLD_P_OVERLAY)
3196 mode = _P_OVERLAY;
3198 Py_BEGIN_ALLOW_THREADS
3199 spawnval = _spawnv(mode, path, argvlist);
3200 Py_END_ALLOW_THREADS
3201 #endif
3203 free_string_array(argvlist, argc);
3204 PyMem_Free(path);
3206 if (spawnval == -1)
3207 return posix_error();
3208 else
3209 #if SIZEOF_LONG == SIZEOF_VOID_P
3210 return Py_BuildValue("l", (long) spawnval);
3211 #else
3212 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
3213 #endif
3217 PyDoc_STRVAR(posix_spawnve__doc__,
3218 "spawnve(mode, path, args, env)\n\n\
3219 Execute the program 'path' in a new process.\n\
3221 mode: mode of process creation\n\
3222 path: path of executable file\n\
3223 args: tuple or list of arguments\n\
3224 env: dictionary of strings mapping to strings");
3226 static PyObject *
3227 posix_spawnve(PyObject *self, PyObject *args)
3229 char *path;
3230 PyObject *argv, *env;
3231 char **argvlist;
3232 char **envlist;
3233 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
3234 int mode, pos, envc;
3235 Py_ssize_t argc, i;
3236 Py_intptr_t spawnval;
3237 PyObject *(*getitem)(PyObject *, Py_ssize_t);
3238 Py_ssize_t lastarg = 0;
3240 /* spawnve has four arguments: (mode, path, argv, env), where
3241 argv is a list or tuple of strings and env is a dictionary
3242 like posix.environ. */
3244 if (!PyArg_ParseTuple(args, "ietOO:spawnve", &mode,
3245 Py_FileSystemDefaultEncoding,
3246 &path, &argv, &env))
3247 return NULL;
3248 if (PyList_Check(argv)) {
3249 argc = PyList_Size(argv);
3250 getitem = PyList_GetItem;
3252 else if (PyTuple_Check(argv)) {
3253 argc = PyTuple_Size(argv);
3254 getitem = PyTuple_GetItem;
3256 else {
3257 PyErr_SetString(PyExc_TypeError,
3258 "spawnve() arg 2 must be a tuple or list");
3259 goto fail_0;
3261 if (!PyMapping_Check(env)) {
3262 PyErr_SetString(PyExc_TypeError,
3263 "spawnve() arg 3 must be a mapping object");
3264 goto fail_0;
3267 argvlist = PyMem_NEW(char *, argc+1);
3268 if (argvlist == NULL) {
3269 PyErr_NoMemory();
3270 goto fail_0;
3272 for (i = 0; i < argc; i++) {
3273 if (!PyArg_Parse((*getitem)(argv, i),
3274 "et;spawnve() arg 2 must contain only strings",
3275 Py_FileSystemDefaultEncoding,
3276 &argvlist[i]))
3278 lastarg = i;
3279 goto fail_1;
3282 lastarg = argc;
3283 argvlist[argc] = NULL;
3285 i = PyMapping_Size(env);
3286 if (i < 0)
3287 goto fail_1;
3288 envlist = PyMem_NEW(char *, i + 1);
3289 if (envlist == NULL) {
3290 PyErr_NoMemory();
3291 goto fail_1;
3293 envc = 0;
3294 keys = PyMapping_Keys(env);
3295 vals = PyMapping_Values(env);
3296 if (!keys || !vals)
3297 goto fail_2;
3298 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3299 PyErr_SetString(PyExc_TypeError,
3300 "spawnve(): env.keys() or env.values() is not a list");
3301 goto fail_2;
3304 for (pos = 0; pos < i; pos++) {
3305 char *p, *k, *v;
3306 size_t len;
3308 key = PyList_GetItem(keys, pos);
3309 val = PyList_GetItem(vals, pos);
3310 if (!key || !val)
3311 goto fail_2;
3313 if (!PyArg_Parse(
3314 key,
3315 "s;spawnve() arg 3 contains a non-string key",
3316 &k) ||
3317 !PyArg_Parse(
3318 val,
3319 "s;spawnve() arg 3 contains a non-string value",
3320 &v))
3322 goto fail_2;
3324 len = PyString_Size(key) + PyString_Size(val) + 2;
3325 p = PyMem_NEW(char, len);
3326 if (p == NULL) {
3327 PyErr_NoMemory();
3328 goto fail_2;
3330 PyOS_snprintf(p, len, "%s=%s", k, v);
3331 envlist[envc++] = p;
3333 envlist[envc] = 0;
3335 #if defined(PYOS_OS2) && defined(PYCC_GCC)
3336 Py_BEGIN_ALLOW_THREADS
3337 spawnval = spawnve(mode, path, argvlist, envlist);
3338 Py_END_ALLOW_THREADS
3339 #else
3340 if (mode == _OLD_P_OVERLAY)
3341 mode = _P_OVERLAY;
3343 Py_BEGIN_ALLOW_THREADS
3344 spawnval = _spawnve(mode, path, argvlist, envlist);
3345 Py_END_ALLOW_THREADS
3346 #endif
3348 if (spawnval == -1)
3349 (void) posix_error();
3350 else
3351 #if SIZEOF_LONG == SIZEOF_VOID_P
3352 res = Py_BuildValue("l", (long) spawnval);
3353 #else
3354 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
3355 #endif
3357 fail_2:
3358 while (--envc >= 0)
3359 PyMem_DEL(envlist[envc]);
3360 PyMem_DEL(envlist);
3361 fail_1:
3362 free_string_array(argvlist, lastarg);
3363 Py_XDECREF(vals);
3364 Py_XDECREF(keys);
3365 fail_0:
3366 PyMem_Free(path);
3367 return res;
3370 /* OS/2 supports spawnvp & spawnvpe natively */
3371 #if defined(PYOS_OS2)
3372 PyDoc_STRVAR(posix_spawnvp__doc__,
3373 "spawnvp(mode, file, args)\n\n\
3374 Execute the program 'file' in a new process, using the environment\n\
3375 search path to find the file.\n\
3377 mode: mode of process creation\n\
3378 file: executable file name\n\
3379 args: tuple or list of strings");
3381 static PyObject *
3382 posix_spawnvp(PyObject *self, PyObject *args)
3384 char *path;
3385 PyObject *argv;
3386 char **argvlist;
3387 int mode, i, argc;
3388 Py_intptr_t spawnval;
3389 PyObject *(*getitem)(PyObject *, Py_ssize_t);
3391 /* spawnvp has three arguments: (mode, path, argv), where
3392 argv is a list or tuple of strings. */
3394 if (!PyArg_ParseTuple(args, "ietO:spawnvp", &mode,
3395 Py_FileSystemDefaultEncoding,
3396 &path, &argv))
3397 return NULL;
3398 if (PyList_Check(argv)) {
3399 argc = PyList_Size(argv);
3400 getitem = PyList_GetItem;
3402 else if (PyTuple_Check(argv)) {
3403 argc = PyTuple_Size(argv);
3404 getitem = PyTuple_GetItem;
3406 else {
3407 PyErr_SetString(PyExc_TypeError,
3408 "spawnvp() arg 2 must be a tuple or list");
3409 PyMem_Free(path);
3410 return NULL;
3413 argvlist = PyMem_NEW(char *, argc+1);
3414 if (argvlist == NULL) {
3415 PyMem_Free(path);
3416 return PyErr_NoMemory();
3418 for (i = 0; i < argc; i++) {
3419 if (!PyArg_Parse((*getitem)(argv, i), "et",
3420 Py_FileSystemDefaultEncoding,
3421 &argvlist[i])) {
3422 free_string_array(argvlist, i);
3423 PyErr_SetString(
3424 PyExc_TypeError,
3425 "spawnvp() arg 2 must contain only strings");
3426 PyMem_Free(path);
3427 return NULL;
3430 argvlist[argc] = NULL;
3432 Py_BEGIN_ALLOW_THREADS
3433 #if defined(PYCC_GCC)
3434 spawnval = spawnvp(mode, path, argvlist);
3435 #else
3436 spawnval = _spawnvp(mode, path, argvlist);
3437 #endif
3438 Py_END_ALLOW_THREADS
3440 free_string_array(argvlist, argc);
3441 PyMem_Free(path);
3443 if (spawnval == -1)
3444 return posix_error();
3445 else
3446 return Py_BuildValue("l", (long) spawnval);
3450 PyDoc_STRVAR(posix_spawnvpe__doc__,
3451 "spawnvpe(mode, file, args, env)\n\n\
3452 Execute the program 'file' in a new process, using the environment\n\
3453 search path to find the file.\n\
3455 mode: mode of process creation\n\
3456 file: executable file name\n\
3457 args: tuple or list of arguments\n\
3458 env: dictionary of strings mapping to strings");
3460 static PyObject *
3461 posix_spawnvpe(PyObject *self, PyObject *args)
3463 char *path;
3464 PyObject *argv, *env;
3465 char **argvlist;
3466 char **envlist;
3467 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
3468 int mode, i, pos, argc, envc;
3469 Py_intptr_t spawnval;
3470 PyObject *(*getitem)(PyObject *, Py_ssize_t);
3471 int lastarg = 0;
3473 /* spawnvpe has four arguments: (mode, path, argv, env), where
3474 argv is a list or tuple of strings and env is a dictionary
3475 like posix.environ. */
3477 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
3478 Py_FileSystemDefaultEncoding,
3479 &path, &argv, &env))
3480 return NULL;
3481 if (PyList_Check(argv)) {
3482 argc = PyList_Size(argv);
3483 getitem = PyList_GetItem;
3485 else if (PyTuple_Check(argv)) {
3486 argc = PyTuple_Size(argv);
3487 getitem = PyTuple_GetItem;
3489 else {
3490 PyErr_SetString(PyExc_TypeError,
3491 "spawnvpe() arg 2 must be a tuple or list");
3492 goto fail_0;
3494 if (!PyMapping_Check(env)) {
3495 PyErr_SetString(PyExc_TypeError,
3496 "spawnvpe() arg 3 must be a mapping object");
3497 goto fail_0;
3500 argvlist = PyMem_NEW(char *, argc+1);
3501 if (argvlist == NULL) {
3502 PyErr_NoMemory();
3503 goto fail_0;
3505 for (i = 0; i < argc; i++) {
3506 if (!PyArg_Parse((*getitem)(argv, i),
3507 "et;spawnvpe() arg 2 must contain only strings",
3508 Py_FileSystemDefaultEncoding,
3509 &argvlist[i]))
3511 lastarg = i;
3512 goto fail_1;
3515 lastarg = argc;
3516 argvlist[argc] = NULL;
3518 i = PyMapping_Size(env);
3519 if (i < 0)
3520 goto fail_1;
3521 envlist = PyMem_NEW(char *, i + 1);
3522 if (envlist == NULL) {
3523 PyErr_NoMemory();
3524 goto fail_1;
3526 envc = 0;
3527 keys = PyMapping_Keys(env);
3528 vals = PyMapping_Values(env);
3529 if (!keys || !vals)
3530 goto fail_2;
3531 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3532 PyErr_SetString(PyExc_TypeError,
3533 "spawnvpe(): env.keys() or env.values() is not a list");
3534 goto fail_2;
3537 for (pos = 0; pos < i; pos++) {
3538 char *p, *k, *v;
3539 size_t len;
3541 key = PyList_GetItem(keys, pos);
3542 val = PyList_GetItem(vals, pos);
3543 if (!key || !val)
3544 goto fail_2;
3546 if (!PyArg_Parse(
3547 key,
3548 "s;spawnvpe() arg 3 contains a non-string key",
3549 &k) ||
3550 !PyArg_Parse(
3551 val,
3552 "s;spawnvpe() arg 3 contains a non-string value",
3553 &v))
3555 goto fail_2;
3557 len = PyString_Size(key) + PyString_Size(val) + 2;
3558 p = PyMem_NEW(char, len);
3559 if (p == NULL) {
3560 PyErr_NoMemory();
3561 goto fail_2;
3563 PyOS_snprintf(p, len, "%s=%s", k, v);
3564 envlist[envc++] = p;
3566 envlist[envc] = 0;
3568 Py_BEGIN_ALLOW_THREADS
3569 #if defined(PYCC_GCC)
3570 spawnval = spawnvpe(mode, path, argvlist, envlist);
3571 #else
3572 spawnval = _spawnvpe(mode, path, argvlist, envlist);
3573 #endif
3574 Py_END_ALLOW_THREADS
3576 if (spawnval == -1)
3577 (void) posix_error();
3578 else
3579 res = Py_BuildValue("l", (long) spawnval);
3581 fail_2:
3582 while (--envc >= 0)
3583 PyMem_DEL(envlist[envc]);
3584 PyMem_DEL(envlist);
3585 fail_1:
3586 free_string_array(argvlist, lastarg);
3587 Py_XDECREF(vals);
3588 Py_XDECREF(keys);
3589 fail_0:
3590 PyMem_Free(path);
3591 return res;
3593 #endif /* PYOS_OS2 */
3594 #endif /* HAVE_SPAWNV */
3597 #ifdef HAVE_FORK1
3598 PyDoc_STRVAR(posix_fork1__doc__,
3599 "fork1() -> pid\n\n\
3600 Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
3602 Return 0 to child process and PID of child to parent process.");
3604 static PyObject *
3605 posix_fork1(PyObject *self, PyObject *noargs)
3607 pid_t pid;
3608 int result;
3609 _PyImport_AcquireLock();
3610 pid = fork1();
3611 result = _PyImport_ReleaseLock();
3612 if (pid == -1)
3613 return posix_error();
3614 if (pid == 0)
3615 PyOS_AfterFork();
3616 if (result < 0) {
3617 /* Don't clobber the OSError if the fork failed. */
3618 PyErr_SetString(PyExc_RuntimeError,
3619 "not holding the import lock");
3620 return NULL;
3622 return PyLong_FromPid(pid);
3624 #endif
3627 #ifdef HAVE_FORK
3628 PyDoc_STRVAR(posix_fork__doc__,
3629 "fork() -> pid\n\n\
3630 Fork a child process.\n\
3631 Return 0 to child process and PID of child to parent process.");
3633 static PyObject *
3634 posix_fork(PyObject *self, PyObject *noargs)
3636 pid_t pid;
3637 int result;
3638 _PyImport_AcquireLock();
3639 pid = fork();
3640 result = _PyImport_ReleaseLock();
3641 if (pid == -1)
3642 return posix_error();
3643 if (pid == 0)
3644 PyOS_AfterFork();
3645 if (result < 0) {
3646 /* Don't clobber the OSError if the fork failed. */
3647 PyErr_SetString(PyExc_RuntimeError,
3648 "not holding the import lock");
3649 return NULL;
3651 return PyLong_FromPid(pid);
3653 #endif
3655 /* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
3656 /* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
3657 #if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
3658 #define DEV_PTY_FILE "/dev/ptc"
3659 #define HAVE_DEV_PTMX
3660 #else
3661 #define DEV_PTY_FILE "/dev/ptmx"
3662 #endif
3664 #if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
3665 #ifdef HAVE_PTY_H
3666 #include <pty.h>
3667 #else
3668 #ifdef HAVE_LIBUTIL_H
3669 #include <libutil.h>
3670 #endif /* HAVE_LIBUTIL_H */
3671 #endif /* HAVE_PTY_H */
3672 #ifdef HAVE_STROPTS_H
3673 #include <stropts.h>
3674 #endif
3675 #endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
3677 #if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
3678 PyDoc_STRVAR(posix_openpty__doc__,
3679 "openpty() -> (master_fd, slave_fd)\n\n\
3680 Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
3682 static PyObject *
3683 posix_openpty(PyObject *self, PyObject *noargs)
3685 int master_fd, slave_fd;
3686 #ifndef HAVE_OPENPTY
3687 char * slave_name;
3688 #endif
3689 #if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
3690 PyOS_sighandler_t sig_saved;
3691 #ifdef sun
3692 extern char *ptsname(int fildes);
3693 #endif
3694 #endif
3696 #ifdef HAVE_OPENPTY
3697 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
3698 return posix_error();
3699 #elif defined(HAVE__GETPTY)
3700 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
3701 if (slave_name == NULL)
3702 return posix_error();
3704 slave_fd = open(slave_name, O_RDWR);
3705 if (slave_fd < 0)
3706 return posix_error();
3707 #else
3708 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
3709 if (master_fd < 0)
3710 return posix_error();
3711 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
3712 /* change permission of slave */
3713 if (grantpt(master_fd) < 0) {
3714 PyOS_setsig(SIGCHLD, sig_saved);
3715 return posix_error();
3717 /* unlock slave */
3718 if (unlockpt(master_fd) < 0) {
3719 PyOS_setsig(SIGCHLD, sig_saved);
3720 return posix_error();
3722 PyOS_setsig(SIGCHLD, sig_saved);
3723 slave_name = ptsname(master_fd); /* get name of slave */
3724 if (slave_name == NULL)
3725 return posix_error();
3726 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
3727 if (slave_fd < 0)
3728 return posix_error();
3729 #if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
3730 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
3731 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
3732 #ifndef __hpux
3733 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
3734 #endif /* __hpux */
3735 #endif /* HAVE_CYGWIN */
3736 #endif /* HAVE_OPENPTY */
3738 return Py_BuildValue("(ii)", master_fd, slave_fd);
3741 #endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
3743 #ifdef HAVE_FORKPTY
3744 PyDoc_STRVAR(posix_forkpty__doc__,
3745 "forkpty() -> (pid, master_fd)\n\n\
3746 Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
3747 Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
3748 To both, return fd of newly opened pseudo-terminal.\n");
3750 static PyObject *
3751 posix_forkpty(PyObject *self, PyObject *noargs)
3753 int master_fd = -1, result;
3754 pid_t pid;
3756 _PyImport_AcquireLock();
3757 pid = forkpty(&master_fd, NULL, NULL, NULL);
3758 result = _PyImport_ReleaseLock();
3759 if (pid == -1)
3760 return posix_error();
3761 if (pid == 0)
3762 PyOS_AfterFork();
3763 if (result < 0) {
3764 /* Don't clobber the OSError if the fork failed. */
3765 PyErr_SetString(PyExc_RuntimeError,
3766 "not holding the import lock");
3767 return NULL;
3769 return Py_BuildValue("(Ni)", PyLong_FromPid(pid), master_fd);
3771 #endif
3773 #ifdef HAVE_GETEGID
3774 PyDoc_STRVAR(posix_getegid__doc__,
3775 "getegid() -> egid\n\n\
3776 Return the current process's effective group id.");
3778 static PyObject *
3779 posix_getegid(PyObject *self, PyObject *noargs)
3781 return PyInt_FromLong((long)getegid());
3783 #endif
3786 #ifdef HAVE_GETEUID
3787 PyDoc_STRVAR(posix_geteuid__doc__,
3788 "geteuid() -> euid\n\n\
3789 Return the current process's effective user id.");
3791 static PyObject *
3792 posix_geteuid(PyObject *self, PyObject *noargs)
3794 return PyInt_FromLong((long)geteuid());
3796 #endif
3799 #ifdef HAVE_GETGID
3800 PyDoc_STRVAR(posix_getgid__doc__,
3801 "getgid() -> gid\n\n\
3802 Return the current process's group id.");
3804 static PyObject *
3805 posix_getgid(PyObject *self, PyObject *noargs)
3807 return PyInt_FromLong((long)getgid());
3809 #endif
3812 PyDoc_STRVAR(posix_getpid__doc__,
3813 "getpid() -> pid\n\n\
3814 Return the current process id");
3816 static PyObject *
3817 posix_getpid(PyObject *self, PyObject *noargs)
3819 return PyLong_FromPid(getpid());
3823 #ifdef HAVE_GETGROUPS
3824 PyDoc_STRVAR(posix_getgroups__doc__,
3825 "getgroups() -> list of group IDs\n\n\
3826 Return list of supplemental group IDs for the process.");
3828 static PyObject *
3829 posix_getgroups(PyObject *self, PyObject *noargs)
3831 PyObject *result = NULL;
3833 #ifdef NGROUPS_MAX
3834 #define MAX_GROUPS NGROUPS_MAX
3835 #else
3836 /* defined to be 16 on Solaris7, so this should be a small number */
3837 #define MAX_GROUPS 64
3838 #endif
3839 gid_t grouplist[MAX_GROUPS];
3840 int n;
3842 n = getgroups(MAX_GROUPS, grouplist);
3843 if (n < 0)
3844 posix_error();
3845 else {
3846 result = PyList_New(n);
3847 if (result != NULL) {
3848 int i;
3849 for (i = 0; i < n; ++i) {
3850 PyObject *o = PyInt_FromLong((long)grouplist[i]);
3851 if (o == NULL) {
3852 Py_DECREF(result);
3853 result = NULL;
3854 break;
3856 PyList_SET_ITEM(result, i, o);
3861 return result;
3863 #endif
3865 #ifdef HAVE_INITGROUPS
3866 PyDoc_STRVAR(posix_initgroups__doc__,
3867 "initgroups(username, gid) -> None\n\n\
3868 Call the system initgroups() to initialize the group access list with all of\n\
3869 the groups of which the specified username is a member, plus the specified\n\
3870 group id.");
3872 static PyObject *
3873 posix_initgroups(PyObject *self, PyObject *args)
3875 char *username;
3876 long gid;
3878 if (!PyArg_ParseTuple(args, "sl:initgroups", &username, &gid))
3879 return NULL;
3881 if (initgroups(username, (gid_t) gid) == -1)
3882 return PyErr_SetFromErrno(PyExc_OSError);
3884 Py_INCREF(Py_None);
3885 return Py_None;
3887 #endif
3889 #ifdef HAVE_GETPGID
3890 PyDoc_STRVAR(posix_getpgid__doc__,
3891 "getpgid(pid) -> pgid\n\n\
3892 Call the system call getpgid().");
3894 static PyObject *
3895 posix_getpgid(PyObject *self, PyObject *args)
3897 pid_t pid, pgid;
3898 if (!PyArg_ParseTuple(args, PARSE_PID ":getpgid", &pid))
3899 return NULL;
3900 pgid = getpgid(pid);
3901 if (pgid < 0)
3902 return posix_error();
3903 return PyLong_FromPid(pgid);
3905 #endif /* HAVE_GETPGID */
3908 #ifdef HAVE_GETPGRP
3909 PyDoc_STRVAR(posix_getpgrp__doc__,
3910 "getpgrp() -> pgrp\n\n\
3911 Return the current process group id.");
3913 static PyObject *
3914 posix_getpgrp(PyObject *self, PyObject *noargs)
3916 #ifdef GETPGRP_HAVE_ARG
3917 return PyLong_FromPid(getpgrp(0));
3918 #else /* GETPGRP_HAVE_ARG */
3919 return PyLong_FromPid(getpgrp());
3920 #endif /* GETPGRP_HAVE_ARG */
3922 #endif /* HAVE_GETPGRP */
3925 #ifdef HAVE_SETPGRP
3926 PyDoc_STRVAR(posix_setpgrp__doc__,
3927 "setpgrp()\n\n\
3928 Make this process a session leader.");
3930 static PyObject *
3931 posix_setpgrp(PyObject *self, PyObject *noargs)
3933 #ifdef SETPGRP_HAVE_ARG
3934 if (setpgrp(0, 0) < 0)
3935 #else /* SETPGRP_HAVE_ARG */
3936 if (setpgrp() < 0)
3937 #endif /* SETPGRP_HAVE_ARG */
3938 return posix_error();
3939 Py_INCREF(Py_None);
3940 return Py_None;
3943 #endif /* HAVE_SETPGRP */
3945 #ifdef HAVE_GETPPID
3946 PyDoc_STRVAR(posix_getppid__doc__,
3947 "getppid() -> ppid\n\n\
3948 Return the parent's process id.");
3950 static PyObject *
3951 posix_getppid(PyObject *self, PyObject *noargs)
3953 return PyLong_FromPid(getppid());
3955 #endif
3958 #ifdef HAVE_GETLOGIN
3959 PyDoc_STRVAR(posix_getlogin__doc__,
3960 "getlogin() -> string\n\n\
3961 Return the actual login name.");
3963 static PyObject *
3964 posix_getlogin(PyObject *self, PyObject *noargs)
3966 PyObject *result = NULL;
3967 char *name;
3968 int old_errno = errno;
3970 errno = 0;
3971 name = getlogin();
3972 if (name == NULL) {
3973 if (errno)
3974 posix_error();
3975 else
3976 PyErr_SetString(PyExc_OSError,
3977 "unable to determine login name");
3979 else
3980 result = PyString_FromString(name);
3981 errno = old_errno;
3983 return result;
3985 #endif
3987 #ifdef HAVE_GETUID
3988 PyDoc_STRVAR(posix_getuid__doc__,
3989 "getuid() -> uid\n\n\
3990 Return the current process's user id.");
3992 static PyObject *
3993 posix_getuid(PyObject *self, PyObject *noargs)
3995 return PyInt_FromLong((long)getuid());
3997 #endif
4000 #ifdef HAVE_KILL
4001 PyDoc_STRVAR(posix_kill__doc__,
4002 "kill(pid, sig)\n\n\
4003 Kill a process with a signal.");
4005 static PyObject *
4006 posix_kill(PyObject *self, PyObject *args)
4008 pid_t pid;
4009 int sig;
4010 if (!PyArg_ParseTuple(args, PARSE_PID "i:kill", &pid, &sig))
4011 return NULL;
4012 #if defined(PYOS_OS2) && !defined(PYCC_GCC)
4013 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
4014 APIRET rc;
4015 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
4016 return os2_error(rc);
4018 } else if (sig == XCPT_SIGNAL_KILLPROC) {
4019 APIRET rc;
4020 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
4021 return os2_error(rc);
4023 } else
4024 return NULL; /* Unrecognized Signal Requested */
4025 #else
4026 if (kill(pid, sig) == -1)
4027 return posix_error();
4028 #endif
4029 Py_INCREF(Py_None);
4030 return Py_None;
4032 #endif
4034 #ifdef HAVE_KILLPG
4035 PyDoc_STRVAR(posix_killpg__doc__,
4036 "killpg(pgid, sig)\n\n\
4037 Kill a process group with a signal.");
4039 static PyObject *
4040 posix_killpg(PyObject *self, PyObject *args)
4042 int sig;
4043 pid_t pgid;
4044 /* XXX some man pages make the `pgid` parameter an int, others
4045 a pid_t. Since getpgrp() returns a pid_t, we assume killpg should
4046 take the same type. Moreover, pid_t is always at least as wide as
4047 int (else compilation of this module fails), which is safe. */
4048 if (!PyArg_ParseTuple(args, PARSE_PID "i:killpg", &pgid, &sig))
4049 return NULL;
4050 if (killpg(pgid, sig) == -1)
4051 return posix_error();
4052 Py_INCREF(Py_None);
4053 return Py_None;
4055 #endif
4057 #ifdef HAVE_PLOCK
4059 #ifdef HAVE_SYS_LOCK_H
4060 #include <sys/lock.h>
4061 #endif
4063 PyDoc_STRVAR(posix_plock__doc__,
4064 "plock(op)\n\n\
4065 Lock program segments into memory.");
4067 static PyObject *
4068 posix_plock(PyObject *self, PyObject *args)
4070 int op;
4071 if (!PyArg_ParseTuple(args, "i:plock", &op))
4072 return NULL;
4073 if (plock(op) == -1)
4074 return posix_error();
4075 Py_INCREF(Py_None);
4076 return Py_None;
4078 #endif
4081 #ifdef HAVE_POPEN
4082 PyDoc_STRVAR(posix_popen__doc__,
4083 "popen(command [, mode='r' [, bufsize]]) -> pipe\n\n\
4084 Open a pipe to/from a command returning a file object.");
4086 #if defined(PYOS_OS2)
4087 #if defined(PYCC_VACPP)
4088 static int
4089 async_system(const char *command)
4091 char errormsg[256], args[1024];
4092 RESULTCODES rcodes;
4093 APIRET rc;
4095 char *shell = getenv("COMSPEC");
4096 if (!shell)
4097 shell = "cmd";
4099 /* avoid overflowing the argument buffer */
4100 if (strlen(shell) + 3 + strlen(command) >= 1024)
4101 return ERROR_NOT_ENOUGH_MEMORY
4103 args[0] = '\0';
4104 strcat(args, shell);
4105 strcat(args, "/c ");
4106 strcat(args, command);
4108 /* execute asynchronously, inheriting the environment */
4109 rc = DosExecPgm(errormsg,
4110 sizeof(errormsg),
4111 EXEC_ASYNC,
4112 args,
4113 NULL,
4114 &rcodes,
4115 shell);
4116 return rc;
4119 static FILE *
4120 popen(const char *command, const char *mode, int pipesize, int *err)
4122 int oldfd, tgtfd;
4123 HFILE pipeh[2];
4124 APIRET rc;
4126 /* mode determines which of stdin or stdout is reconnected to
4127 * the pipe to the child
4129 if (strchr(mode, 'r') != NULL) {
4130 tgt_fd = 1; /* stdout */
4131 } else if (strchr(mode, 'w')) {
4132 tgt_fd = 0; /* stdin */
4133 } else {
4134 *err = ERROR_INVALID_ACCESS;
4135 return NULL;
4138 /* setup the pipe */
4139 if ((rc = DosCreatePipe(&pipeh[0], &pipeh[1], pipesize)) != NO_ERROR) {
4140 *err = rc;
4141 return NULL;
4144 /* prevent other threads accessing stdio */
4145 DosEnterCritSec();
4147 /* reconnect stdio and execute child */
4148 oldfd = dup(tgtfd);
4149 close(tgtfd);
4150 if (dup2(pipeh[tgtfd], tgtfd) == 0) {
4151 DosClose(pipeh[tgtfd]);
4152 rc = async_system(command);
4155 /* restore stdio */
4156 dup2(oldfd, tgtfd);
4157 close(oldfd);
4159 /* allow other threads access to stdio */
4160 DosExitCritSec();
4162 /* if execution of child was successful return file stream */
4163 if (rc == NO_ERROR)
4164 return fdopen(pipeh[1 - tgtfd], mode);
4165 else {
4166 DosClose(pipeh[1 - tgtfd]);
4167 *err = rc;
4168 return NULL;
4172 static PyObject *
4173 posix_popen(PyObject *self, PyObject *args)
4175 char *name;
4176 char *mode = "r";
4177 int err, bufsize = -1;
4178 FILE *fp;
4179 PyObject *f;
4180 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
4181 return NULL;
4182 Py_BEGIN_ALLOW_THREADS
4183 fp = popen(name, mode, (bufsize > 0) ? bufsize : 4096, &err);
4184 Py_END_ALLOW_THREADS
4185 if (fp == NULL)
4186 return os2_error(err);
4188 f = PyFile_FromFile(fp, name, mode, fclose);
4189 if (f != NULL)
4190 PyFile_SetBufSize(f, bufsize);
4191 return f;
4194 #elif defined(PYCC_GCC)
4196 /* standard posix version of popen() support */
4197 static PyObject *
4198 posix_popen(PyObject *self, PyObject *args)
4200 char *name;
4201 char *mode = "r";
4202 int bufsize = -1;
4203 FILE *fp;
4204 PyObject *f;
4205 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
4206 return NULL;
4207 Py_BEGIN_ALLOW_THREADS
4208 fp = popen(name, mode);
4209 Py_END_ALLOW_THREADS
4210 if (fp == NULL)
4211 return posix_error();
4212 f = PyFile_FromFile(fp, name, mode, pclose);
4213 if (f != NULL)
4214 PyFile_SetBufSize(f, bufsize);
4215 return f;
4218 /* fork() under OS/2 has lots'o'warts
4219 * EMX supports pipe() and spawn*() so we can synthesize popen[234]()
4220 * most of this code is a ripoff of the win32 code, but using the
4221 * capabilities of EMX's C library routines
4224 /* These tell _PyPopen() whether to return 1, 2, or 3 file objects. */
4225 #define POPEN_1 1
4226 #define POPEN_2 2
4227 #define POPEN_3 3
4228 #define POPEN_4 4
4230 static PyObject *_PyPopen(char *, int, int, int);
4231 static int _PyPclose(FILE *file);
4234 * Internal dictionary mapping popen* file pointers to process handles,
4235 * for use when retrieving the process exit code. See _PyPclose() below
4236 * for more information on this dictionary's use.
4238 static PyObject *_PyPopenProcs = NULL;
4240 /* os2emx version of popen2()
4242 * The result of this function is a pipe (file) connected to the
4243 * process's stdin, and a pipe connected to the process's stdout.
4246 static PyObject *
4247 os2emx_popen2(PyObject *self, PyObject *args)
4249 PyObject *f;
4250 int tm=0;
4252 char *cmdstring;
4253 char *mode = "t";
4254 int bufsize = -1;
4255 if (!PyArg_ParseTuple(args, "s|si:popen2", &cmdstring, &mode, &bufsize))
4256 return NULL;
4258 if (*mode == 't')
4259 tm = O_TEXT;
4260 else if (*mode != 'b') {
4261 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
4262 return NULL;
4263 } else
4264 tm = O_BINARY;
4266 f = _PyPopen(cmdstring, tm, POPEN_2, bufsize);
4268 return f;
4272 * Variation on os2emx.popen2
4274 * The result of this function is 3 pipes - the process's stdin,
4275 * stdout and stderr
4278 static PyObject *
4279 os2emx_popen3(PyObject *self, PyObject *args)
4281 PyObject *f;
4282 int tm = 0;
4284 char *cmdstring;
4285 char *mode = "t";
4286 int bufsize = -1;
4287 if (!PyArg_ParseTuple(args, "s|si:popen3", &cmdstring, &mode, &bufsize))
4288 return NULL;
4290 if (*mode == 't')
4291 tm = O_TEXT;
4292 else if (*mode != 'b') {
4293 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
4294 return NULL;
4295 } else
4296 tm = O_BINARY;
4298 f = _PyPopen(cmdstring, tm, POPEN_3, bufsize);
4300 return f;
4304 * Variation on os2emx.popen2
4306 * The result of this function is 2 pipes - the processes stdin,
4307 * and stdout+stderr combined as a single pipe.
4310 static PyObject *
4311 os2emx_popen4(PyObject *self, PyObject *args)
4313 PyObject *f;
4314 int tm = 0;
4316 char *cmdstring;
4317 char *mode = "t";
4318 int bufsize = -1;
4319 if (!PyArg_ParseTuple(args, "s|si:popen4", &cmdstring, &mode, &bufsize))
4320 return NULL;
4322 if (*mode == 't')
4323 tm = O_TEXT;
4324 else if (*mode != 'b') {
4325 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
4326 return NULL;
4327 } else
4328 tm = O_BINARY;
4330 f = _PyPopen(cmdstring, tm, POPEN_4, bufsize);
4332 return f;
4335 /* a couple of structures for convenient handling of multiple
4336 * file handles and pipes
4338 struct file_ref
4340 int handle;
4341 int flags;
4344 struct pipe_ref
4346 int rd;
4347 int wr;
4350 /* The following code is derived from the win32 code */
4352 static PyObject *
4353 _PyPopen(char *cmdstring, int mode, int n, int bufsize)
4355 struct file_ref stdio[3];
4356 struct pipe_ref p_fd[3];
4357 FILE *p_s[3];
4358 int file_count, i, pipe_err;
4359 pid_t pipe_pid;
4360 char *shell, *sh_name, *opt, *rd_mode, *wr_mode;
4361 PyObject *f, *p_f[3];
4363 /* file modes for subsequent fdopen's on pipe handles */
4364 if (mode == O_TEXT)
4366 rd_mode = "rt";
4367 wr_mode = "wt";
4369 else
4371 rd_mode = "rb";
4372 wr_mode = "wb";
4375 /* prepare shell references */
4376 if ((shell = getenv("EMXSHELL")) == NULL)
4377 if ((shell = getenv("COMSPEC")) == NULL)
4379 errno = ENOENT;
4380 return posix_error();
4383 sh_name = _getname(shell);
4384 if (stricmp(sh_name, "cmd.exe") == 0 || stricmp(sh_name, "4os2.exe") == 0)
4385 opt = "/c";
4386 else
4387 opt = "-c";
4389 /* save current stdio fds + their flags, and set not inheritable */
4390 i = pipe_err = 0;
4391 while (pipe_err >= 0 && i < 3)
4393 pipe_err = stdio[i].handle = dup(i);
4394 stdio[i].flags = fcntl(i, F_GETFD, 0);
4395 fcntl(stdio[i].handle, F_SETFD, stdio[i].flags | FD_CLOEXEC);
4396 i++;
4398 if (pipe_err < 0)
4400 /* didn't get them all saved - clean up and bail out */
4401 int saved_err = errno;
4402 while (i-- > 0)
4404 close(stdio[i].handle);
4406 errno = saved_err;
4407 return posix_error();
4410 /* create pipe ends */
4411 file_count = 2;
4412 if (n == POPEN_3)
4413 file_count = 3;
4414 i = pipe_err = 0;
4415 while ((pipe_err == 0) && (i < file_count))
4416 pipe_err = pipe((int *)&p_fd[i++]);
4417 if (pipe_err < 0)
4419 /* didn't get them all made - clean up and bail out */
4420 while (i-- > 0)
4422 close(p_fd[i].wr);
4423 close(p_fd[i].rd);
4425 errno = EPIPE;
4426 return posix_error();
4429 /* change the actual standard IO streams over temporarily,
4430 * making the retained pipe ends non-inheritable
4432 pipe_err = 0;
4434 /* - stdin */
4435 if (dup2(p_fd[0].rd, 0) == 0)
4437 close(p_fd[0].rd);
4438 i = fcntl(p_fd[0].wr, F_GETFD, 0);
4439 fcntl(p_fd[0].wr, F_SETFD, i | FD_CLOEXEC);
4440 if ((p_s[0] = fdopen(p_fd[0].wr, wr_mode)) == NULL)
4442 close(p_fd[0].wr);
4443 pipe_err = -1;
4446 else
4448 pipe_err = -1;
4451 /* - stdout */
4452 if (pipe_err == 0)
4454 if (dup2(p_fd[1].wr, 1) == 1)
4456 close(p_fd[1].wr);
4457 i = fcntl(p_fd[1].rd, F_GETFD, 0);
4458 fcntl(p_fd[1].rd, F_SETFD, i | FD_CLOEXEC);
4459 if ((p_s[1] = fdopen(p_fd[1].rd, rd_mode)) == NULL)
4461 close(p_fd[1].rd);
4462 pipe_err = -1;
4465 else
4467 pipe_err = -1;
4471 /* - stderr, as required */
4472 if (pipe_err == 0)
4473 switch (n)
4475 case POPEN_3:
4477 if (dup2(p_fd[2].wr, 2) == 2)
4479 close(p_fd[2].wr);
4480 i = fcntl(p_fd[2].rd, F_GETFD, 0);
4481 fcntl(p_fd[2].rd, F_SETFD, i | FD_CLOEXEC);
4482 if ((p_s[2] = fdopen(p_fd[2].rd, rd_mode)) == NULL)
4484 close(p_fd[2].rd);
4485 pipe_err = -1;
4488 else
4490 pipe_err = -1;
4492 break;
4495 case POPEN_4:
4497 if (dup2(1, 2) != 2)
4499 pipe_err = -1;
4501 break;
4505 /* spawn the child process */
4506 if (pipe_err == 0)
4508 pipe_pid = spawnlp(P_NOWAIT, shell, shell, opt, cmdstring, (char *)0);
4509 if (pipe_pid == -1)
4511 pipe_err = -1;
4513 else
4515 /* save the PID into the FILE structure
4516 * NOTE: this implementation doesn't actually
4517 * take advantage of this, but do it for
4518 * completeness - AIM Apr01
4520 for (i = 0; i < file_count; i++)
4521 p_s[i]->_pid = pipe_pid;
4525 /* reset standard IO to normal */
4526 for (i = 0; i < 3; i++)
4528 dup2(stdio[i].handle, i);
4529 fcntl(i, F_SETFD, stdio[i].flags);
4530 close(stdio[i].handle);
4533 /* if any remnant problems, clean up and bail out */
4534 if (pipe_err < 0)
4536 for (i = 0; i < 3; i++)
4538 close(p_fd[i].rd);
4539 close(p_fd[i].wr);
4541 errno = EPIPE;
4542 return posix_error_with_filename(cmdstring);
4545 /* build tuple of file objects to return */
4546 if ((p_f[0] = PyFile_FromFile(p_s[0], cmdstring, wr_mode, _PyPclose)) != NULL)
4547 PyFile_SetBufSize(p_f[0], bufsize);
4548 if ((p_f[1] = PyFile_FromFile(p_s[1], cmdstring, rd_mode, _PyPclose)) != NULL)
4549 PyFile_SetBufSize(p_f[1], bufsize);
4550 if (n == POPEN_3)
4552 if ((p_f[2] = PyFile_FromFile(p_s[2], cmdstring, rd_mode, _PyPclose)) != NULL)
4553 PyFile_SetBufSize(p_f[0], bufsize);
4554 f = PyTuple_Pack(3, p_f[0], p_f[1], p_f[2]);
4556 else
4557 f = PyTuple_Pack(2, p_f[0], p_f[1]);
4560 * Insert the files we've created into the process dictionary
4561 * all referencing the list with the process handle and the
4562 * initial number of files (see description below in _PyPclose).
4563 * Since if _PyPclose later tried to wait on a process when all
4564 * handles weren't closed, it could create a deadlock with the
4565 * child, we spend some energy here to try to ensure that we
4566 * either insert all file handles into the dictionary or none
4567 * at all. It's a little clumsy with the various popen modes
4568 * and variable number of files involved.
4570 if (!_PyPopenProcs)
4572 _PyPopenProcs = PyDict_New();
4575 if (_PyPopenProcs)
4577 PyObject *procObj, *pidObj, *intObj, *fileObj[3];
4578 int ins_rc[3];
4580 fileObj[0] = fileObj[1] = fileObj[2] = NULL;
4581 ins_rc[0] = ins_rc[1] = ins_rc[2] = 0;
4583 procObj = PyList_New(2);
4584 pidObj = PyLong_FromPid(pipe_pid);
4585 intObj = PyInt_FromLong((long) file_count);
4587 if (procObj && pidObj && intObj)
4589 PyList_SetItem(procObj, 0, pidObj);
4590 PyList_SetItem(procObj, 1, intObj);
4592 fileObj[0] = PyLong_FromVoidPtr(p_s[0]);
4593 if (fileObj[0])
4595 ins_rc[0] = PyDict_SetItem(_PyPopenProcs,
4596 fileObj[0],
4597 procObj);
4599 fileObj[1] = PyLong_FromVoidPtr(p_s[1]);
4600 if (fileObj[1])
4602 ins_rc[1] = PyDict_SetItem(_PyPopenProcs,
4603 fileObj[1],
4604 procObj);
4606 if (file_count >= 3)
4608 fileObj[2] = PyLong_FromVoidPtr(p_s[2]);
4609 if (fileObj[2])
4611 ins_rc[2] = PyDict_SetItem(_PyPopenProcs,
4612 fileObj[2],
4613 procObj);
4617 if (ins_rc[0] < 0 || !fileObj[0] ||
4618 ins_rc[1] < 0 || (file_count > 1 && !fileObj[1]) ||
4619 ins_rc[2] < 0 || (file_count > 2 && !fileObj[2]))
4621 /* Something failed - remove any dictionary
4622 * entries that did make it.
4624 if (!ins_rc[0] && fileObj[0])
4626 PyDict_DelItem(_PyPopenProcs,
4627 fileObj[0]);
4629 if (!ins_rc[1] && fileObj[1])
4631 PyDict_DelItem(_PyPopenProcs,
4632 fileObj[1]);
4634 if (!ins_rc[2] && fileObj[2])
4636 PyDict_DelItem(_PyPopenProcs,
4637 fileObj[2]);
4643 * Clean up our localized references for the dictionary keys
4644 * and value since PyDict_SetItem will Py_INCREF any copies
4645 * that got placed in the dictionary.
4647 Py_XDECREF(procObj);
4648 Py_XDECREF(fileObj[0]);
4649 Py_XDECREF(fileObj[1]);
4650 Py_XDECREF(fileObj[2]);
4653 /* Child is launched. */
4654 return f;
4658 * Wrapper for fclose() to use for popen* files, so we can retrieve the
4659 * exit code for the child process and return as a result of the close.
4661 * This function uses the _PyPopenProcs dictionary in order to map the
4662 * input file pointer to information about the process that was
4663 * originally created by the popen* call that created the file pointer.
4664 * The dictionary uses the file pointer as a key (with one entry
4665 * inserted for each file returned by the original popen* call) and a
4666 * single list object as the value for all files from a single call.
4667 * The list object contains the Win32 process handle at [0], and a file
4668 * count at [1], which is initialized to the total number of file
4669 * handles using that list.
4671 * This function closes whichever handle it is passed, and decrements
4672 * the file count in the dictionary for the process handle pointed to
4673 * by this file. On the last close (when the file count reaches zero),
4674 * this function will wait for the child process and then return its
4675 * exit code as the result of the close() operation. This permits the
4676 * files to be closed in any order - it is always the close() of the
4677 * final handle that will return the exit code.
4679 * NOTE: This function is currently called with the GIL released.
4680 * hence we use the GILState API to manage our state.
4683 static int _PyPclose(FILE *file)
4685 int result;
4686 int exit_code;
4687 pid_t pipe_pid;
4688 PyObject *procObj, *pidObj, *intObj, *fileObj;
4689 int file_count;
4690 #ifdef WITH_THREAD
4691 PyGILState_STATE state;
4692 #endif
4694 /* Close the file handle first, to ensure it can't block the
4695 * child from exiting if it's the last handle.
4697 result = fclose(file);
4699 #ifdef WITH_THREAD
4700 state = PyGILState_Ensure();
4701 #endif
4702 if (_PyPopenProcs)
4704 if ((fileObj = PyLong_FromVoidPtr(file)) != NULL &&
4705 (procObj = PyDict_GetItem(_PyPopenProcs,
4706 fileObj)) != NULL &&
4707 (pidObj = PyList_GetItem(procObj,0)) != NULL &&
4708 (intObj = PyList_GetItem(procObj,1)) != NULL)
4710 pipe_pid = (pid_t) PyLong_AsPid(pidObj);
4711 file_count = (int) PyInt_AsLong(intObj);
4713 if (file_count > 1)
4715 /* Still other files referencing process */
4716 file_count--;
4717 PyList_SetItem(procObj,1,
4718 PyInt_FromLong((long) file_count));
4720 else
4722 /* Last file for this process */
4723 if (result != EOF &&
4724 waitpid(pipe_pid, &exit_code, 0) == pipe_pid)
4726 /* extract exit status */
4727 if (WIFEXITED(exit_code))
4729 result = WEXITSTATUS(exit_code);
4731 else
4733 errno = EPIPE;
4734 result = -1;
4737 else
4739 /* Indicate failure - this will cause the file object
4740 * to raise an I/O error and translate the last
4741 * error code from errno. We do have a problem with
4742 * last errors that overlap the normal errno table,
4743 * but that's a consistent problem with the file object.
4745 result = -1;
4749 /* Remove this file pointer from dictionary */
4750 PyDict_DelItem(_PyPopenProcs, fileObj);
4752 if (PyDict_Size(_PyPopenProcs) == 0)
4754 Py_DECREF(_PyPopenProcs);
4755 _PyPopenProcs = NULL;
4758 } /* if object retrieval ok */
4760 Py_XDECREF(fileObj);
4761 } /* if _PyPopenProcs */
4763 #ifdef WITH_THREAD
4764 PyGILState_Release(state);
4765 #endif
4766 return result;
4769 #endif /* PYCC_??? */
4771 #elif defined(MS_WINDOWS)
4774 * Portable 'popen' replacement for Win32.
4776 * Written by Bill Tutt <billtut@microsoft.com>. Minor tweaks
4777 * and 2.0 integration by Fredrik Lundh <fredrik@pythonware.com>
4778 * Return code handling by David Bolen <db3l@fitlinxx.com>.
4781 #include <malloc.h>
4782 #include <io.h>
4783 #include <fcntl.h>
4785 /* These tell _PyPopen() wether to return 1, 2, or 3 file objects. */
4786 #define POPEN_1 1
4787 #define POPEN_2 2
4788 #define POPEN_3 3
4789 #define POPEN_4 4
4791 static PyObject *_PyPopen(char *, int, int);
4792 static int _PyPclose(FILE *file);
4795 * Internal dictionary mapping popen* file pointers to process handles,
4796 * for use when retrieving the process exit code. See _PyPclose() below
4797 * for more information on this dictionary's use.
4799 static PyObject *_PyPopenProcs = NULL;
4802 /* popen that works from a GUI.
4804 * The result of this function is a pipe (file) connected to the
4805 * processes stdin or stdout, depending on the requested mode.
4808 static PyObject *
4809 posix_popen(PyObject *self, PyObject *args)
4811 PyObject *f;
4812 int tm = 0;
4814 char *cmdstring;
4815 char *mode = "r";
4816 int bufsize = -1;
4817 if (!PyArg_ParseTuple(args, "s|si:popen", &cmdstring, &mode, &bufsize))
4818 return NULL;
4820 if (*mode == 'r')
4821 tm = _O_RDONLY;
4822 else if (*mode != 'w') {
4823 PyErr_SetString(PyExc_ValueError, "popen() arg 2 must be 'r' or 'w'");
4824 return NULL;
4825 } else
4826 tm = _O_WRONLY;
4828 if (bufsize != -1) {
4829 PyErr_SetString(PyExc_ValueError, "popen() arg 3 must be -1");
4830 return NULL;
4833 if (*(mode+1) == 't')
4834 f = _PyPopen(cmdstring, tm | _O_TEXT, POPEN_1);
4835 else if (*(mode+1) == 'b')
4836 f = _PyPopen(cmdstring, tm | _O_BINARY, POPEN_1);
4837 else
4838 f = _PyPopen(cmdstring, tm | _O_TEXT, POPEN_1);
4840 return f;
4843 /* Variation on win32pipe.popen
4845 * The result of this function is a pipe (file) connected to the
4846 * process's stdin, and a pipe connected to the process's stdout.
4849 static PyObject *
4850 win32_popen2(PyObject *self, PyObject *args)
4852 PyObject *f;
4853 int tm=0;
4855 char *cmdstring;
4856 char *mode = "t";
4857 int bufsize = -1;
4858 if (!PyArg_ParseTuple(args, "s|si:popen2", &cmdstring, &mode, &bufsize))
4859 return NULL;
4861 if (*mode == 't')
4862 tm = _O_TEXT;
4863 else if (*mode != 'b') {
4864 PyErr_SetString(PyExc_ValueError, "popen2() arg 2 must be 't' or 'b'");
4865 return NULL;
4866 } else
4867 tm = _O_BINARY;
4869 if (bufsize != -1) {
4870 PyErr_SetString(PyExc_ValueError, "popen2() arg 3 must be -1");
4871 return NULL;
4874 f = _PyPopen(cmdstring, tm, POPEN_2);
4876 return f;
4880 * Variation on <om win32pipe.popen>
4882 * The result of this function is 3 pipes - the process's stdin,
4883 * stdout and stderr
4886 static PyObject *
4887 win32_popen3(PyObject *self, PyObject *args)
4889 PyObject *f;
4890 int tm = 0;
4892 char *cmdstring;
4893 char *mode = "t";
4894 int bufsize = -1;
4895 if (!PyArg_ParseTuple(args, "s|si:popen3", &cmdstring, &mode, &bufsize))
4896 return NULL;
4898 if (*mode == 't')
4899 tm = _O_TEXT;
4900 else if (*mode != 'b') {
4901 PyErr_SetString(PyExc_ValueError, "popen3() arg 2 must be 't' or 'b'");
4902 return NULL;
4903 } else
4904 tm = _O_BINARY;
4906 if (bufsize != -1) {
4907 PyErr_SetString(PyExc_ValueError, "popen3() arg 3 must be -1");
4908 return NULL;
4911 f = _PyPopen(cmdstring, tm, POPEN_3);
4913 return f;
4917 * Variation on win32pipe.popen
4919 * The result of this function is 2 pipes - the processes stdin,
4920 * and stdout+stderr combined as a single pipe.
4923 static PyObject *
4924 win32_popen4(PyObject *self, PyObject *args)
4926 PyObject *f;
4927 int tm = 0;
4929 char *cmdstring;
4930 char *mode = "t";
4931 int bufsize = -1;
4932 if (!PyArg_ParseTuple(args, "s|si:popen4", &cmdstring, &mode, &bufsize))
4933 return NULL;
4935 if (*mode == 't')
4936 tm = _O_TEXT;
4937 else if (*mode != 'b') {
4938 PyErr_SetString(PyExc_ValueError, "popen4() arg 2 must be 't' or 'b'");
4939 return NULL;
4940 } else
4941 tm = _O_BINARY;
4943 if (bufsize != -1) {
4944 PyErr_SetString(PyExc_ValueError, "popen4() arg 3 must be -1");
4945 return NULL;
4948 f = _PyPopen(cmdstring, tm, POPEN_4);
4950 return f;
4953 static BOOL
4954 _PyPopenCreateProcess(char *cmdstring,
4955 HANDLE hStdin,
4956 HANDLE hStdout,
4957 HANDLE hStderr,
4958 HANDLE *hProcess)
4960 PROCESS_INFORMATION piProcInfo;
4961 STARTUPINFO siStartInfo;
4962 DWORD dwProcessFlags = 0; /* no NEW_CONSOLE by default for Ctrl+C handling */
4963 char *s1,*s2, *s3 = " /c ";
4964 const char *szConsoleSpawn = "w9xpopen.exe";
4965 int i;
4966 Py_ssize_t x;
4968 if (i = GetEnvironmentVariable("COMSPEC",NULL,0)) {
4969 char *comshell;
4971 s1 = (char *)alloca(i);
4972 if (!(x = GetEnvironmentVariable("COMSPEC", s1, i)))
4973 /* x < i, so x fits into an integer */
4974 return (int)x;
4976 /* Explicitly check if we are using COMMAND.COM. If we are
4977 * then use the w9xpopen hack.
4979 comshell = s1 + x;
4980 while (comshell >= s1 && *comshell != '\\')
4981 --comshell;
4982 ++comshell;
4984 if (GetVersion() < 0x80000000 &&
4985 _stricmp(comshell, "command.com") != 0) {
4986 /* NT/2000 and not using command.com. */
4987 x = i + strlen(s3) + strlen(cmdstring) + 1;
4988 s2 = (char *)alloca(x);
4989 ZeroMemory(s2, x);
4990 PyOS_snprintf(s2, x, "%s%s%s", s1, s3, cmdstring);
4992 else {
4994 * Oh gag, we're on Win9x or using COMMAND.COM. Use
4995 * the workaround listed in KB: Q150956
4997 char modulepath[_MAX_PATH];
4998 struct stat statinfo;
4999 GetModuleFileName(NULL, modulepath, sizeof(modulepath));
5000 for (x = i = 0; modulepath[i]; i++)
5001 if (modulepath[i] == SEP)
5002 x = i+1;
5003 modulepath[x] = '\0';
5004 /* Create the full-name to w9xpopen, so we can test it exists */
5005 strncat(modulepath,
5006 szConsoleSpawn,
5007 (sizeof(modulepath)/sizeof(modulepath[0]))
5008 -strlen(modulepath));
5009 if (stat(modulepath, &statinfo) != 0) {
5010 size_t mplen = sizeof(modulepath)/sizeof(modulepath[0]);
5011 /* Eeek - file-not-found - possibly an embedding
5012 situation - see if we can locate it in sys.prefix
5014 strncpy(modulepath,
5015 Py_GetExecPrefix(),
5016 mplen);
5017 modulepath[mplen-1] = '\0';
5018 if (modulepath[strlen(modulepath)-1] != '\\')
5019 strcat(modulepath, "\\");
5020 strncat(modulepath,
5021 szConsoleSpawn,
5022 mplen-strlen(modulepath));
5023 /* No where else to look - raise an easily identifiable
5024 error, rather than leaving Windows to report
5025 "file not found" - as the user is probably blissfully
5026 unaware this shim EXE is used, and it will confuse them.
5027 (well, it confused me for a while ;-)
5029 if (stat(modulepath, &statinfo) != 0) {
5030 PyErr_Format(PyExc_RuntimeError,
5031 "Can not locate '%s' which is needed "
5032 "for popen to work with your shell "
5033 "or platform.",
5034 szConsoleSpawn);
5035 return FALSE;
5038 x = i + strlen(s3) + strlen(cmdstring) + 1 +
5039 strlen(modulepath) +
5040 strlen(szConsoleSpawn) + 1;
5042 s2 = (char *)alloca(x);
5043 ZeroMemory(s2, x);
5044 /* To maintain correct argument passing semantics,
5045 we pass the command-line as it stands, and allow
5046 quoting to be applied. w9xpopen.exe will then
5047 use its argv vector, and re-quote the necessary
5048 args for the ultimate child process.
5050 PyOS_snprintf(
5051 s2, x,
5052 "\"%s\" %s%s%s",
5053 modulepath,
5056 cmdstring);
5057 /* Not passing CREATE_NEW_CONSOLE has been known to
5058 cause random failures on win9x. Specifically a
5059 dialog:
5060 "Your program accessed mem currently in use at xxx"
5061 and a hopeful warning about the stability of your
5062 system.
5063 Cost is Ctrl+C won't kill children, but anyone
5064 who cares can have a go!
5066 dwProcessFlags |= CREATE_NEW_CONSOLE;
5070 /* Could be an else here to try cmd.exe / command.com in the path
5071 Now we'll just error out.. */
5072 else {
5073 PyErr_SetString(PyExc_RuntimeError,
5074 "Cannot locate a COMSPEC environment variable to "
5075 "use as the shell");
5076 return FALSE;
5079 ZeroMemory(&siStartInfo, sizeof(STARTUPINFO));
5080 siStartInfo.cb = sizeof(STARTUPINFO);
5081 siStartInfo.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
5082 siStartInfo.hStdInput = hStdin;
5083 siStartInfo.hStdOutput = hStdout;
5084 siStartInfo.hStdError = hStderr;
5085 siStartInfo.wShowWindow = SW_HIDE;
5087 if (CreateProcess(NULL,
5089 NULL,
5090 NULL,
5091 TRUE,
5092 dwProcessFlags,
5093 NULL,
5094 NULL,
5095 &siStartInfo,
5096 &piProcInfo) ) {
5097 /* Close the handles now so anyone waiting is woken. */
5098 CloseHandle(piProcInfo.hThread);
5100 /* Return process handle */
5101 *hProcess = piProcInfo.hProcess;
5102 return TRUE;
5104 win32_error("CreateProcess", s2);
5105 return FALSE;
5108 /* The following code is based off of KB: Q190351 */
5110 static PyObject *
5111 _PyPopen(char *cmdstring, int mode, int n)
5113 HANDLE hChildStdinRd, hChildStdinWr, hChildStdoutRd, hChildStdoutWr,
5114 hChildStderrRd, hChildStderrWr, hChildStdinWrDup, hChildStdoutRdDup,
5115 hChildStderrRdDup, hProcess; /* hChildStdoutWrDup; */
5117 SECURITY_ATTRIBUTES saAttr;
5118 BOOL fSuccess;
5119 int fd1, fd2, fd3;
5120 FILE *f1, *f2, *f3;
5121 long file_count;
5122 PyObject *f;
5124 saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
5125 saAttr.bInheritHandle = TRUE;
5126 saAttr.lpSecurityDescriptor = NULL;
5128 if (!CreatePipe(&hChildStdinRd, &hChildStdinWr, &saAttr, 0))
5129 return win32_error("CreatePipe", NULL);
5131 /* Create new output read handle and the input write handle. Set
5132 * the inheritance properties to FALSE. Otherwise, the child inherits
5133 * these handles; resulting in non-closeable handles to the pipes
5134 * being created. */
5135 fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdinWr,
5136 GetCurrentProcess(), &hChildStdinWrDup, 0,
5137 FALSE,
5138 DUPLICATE_SAME_ACCESS);
5139 if (!fSuccess)
5140 return win32_error("DuplicateHandle", NULL);
5142 /* Close the inheritable version of ChildStdin
5143 that we're using. */
5144 CloseHandle(hChildStdinWr);
5146 if (!CreatePipe(&hChildStdoutRd, &hChildStdoutWr, &saAttr, 0))
5147 return win32_error("CreatePipe", NULL);
5149 fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdoutRd,
5150 GetCurrentProcess(), &hChildStdoutRdDup, 0,
5151 FALSE, DUPLICATE_SAME_ACCESS);
5152 if (!fSuccess)
5153 return win32_error("DuplicateHandle", NULL);
5155 /* Close the inheritable version of ChildStdout
5156 that we're using. */
5157 CloseHandle(hChildStdoutRd);
5159 if (n != POPEN_4) {
5160 if (!CreatePipe(&hChildStderrRd, &hChildStderrWr, &saAttr, 0))
5161 return win32_error("CreatePipe", NULL);
5162 fSuccess = DuplicateHandle(GetCurrentProcess(),
5163 hChildStderrRd,
5164 GetCurrentProcess(),
5165 &hChildStderrRdDup, 0,
5166 FALSE, DUPLICATE_SAME_ACCESS);
5167 if (!fSuccess)
5168 return win32_error("DuplicateHandle", NULL);
5169 /* Close the inheritable version of ChildStdErr that we're using. */
5170 CloseHandle(hChildStderrRd);
5173 switch (n) {
5174 case POPEN_1:
5175 switch (mode & (_O_RDONLY | _O_TEXT | _O_BINARY | _O_WRONLY)) {
5176 case _O_WRONLY | _O_TEXT:
5177 /* Case for writing to child Stdin in text mode. */
5178 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
5179 f1 = _fdopen(fd1, "w");
5180 f = PyFile_FromFile(f1, cmdstring, "w", _PyPclose);
5181 PyFile_SetBufSize(f, 0);
5182 /* We don't care about these pipes anymore, so close them. */
5183 CloseHandle(hChildStdoutRdDup);
5184 CloseHandle(hChildStderrRdDup);
5185 break;
5187 case _O_RDONLY | _O_TEXT:
5188 /* Case for reading from child Stdout in text mode. */
5189 fd1 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
5190 f1 = _fdopen(fd1, "r");
5191 f = PyFile_FromFile(f1, cmdstring, "r", _PyPclose);
5192 PyFile_SetBufSize(f, 0);
5193 /* We don't care about these pipes anymore, so close them. */
5194 CloseHandle(hChildStdinWrDup);
5195 CloseHandle(hChildStderrRdDup);
5196 break;
5198 case _O_RDONLY | _O_BINARY:
5199 /* Case for readinig from child Stdout in binary mode. */
5200 fd1 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
5201 f1 = _fdopen(fd1, "rb");
5202 f = PyFile_FromFile(f1, cmdstring, "rb", _PyPclose);
5203 PyFile_SetBufSize(f, 0);
5204 /* We don't care about these pipes anymore, so close them. */
5205 CloseHandle(hChildStdinWrDup);
5206 CloseHandle(hChildStderrRdDup);
5207 break;
5209 case _O_WRONLY | _O_BINARY:
5210 /* Case for writing to child Stdin in binary mode. */
5211 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
5212 f1 = _fdopen(fd1, "wb");
5213 f = PyFile_FromFile(f1, cmdstring, "wb", _PyPclose);
5214 PyFile_SetBufSize(f, 0);
5215 /* We don't care about these pipes anymore, so close them. */
5216 CloseHandle(hChildStdoutRdDup);
5217 CloseHandle(hChildStderrRdDup);
5218 break;
5220 file_count = 1;
5221 break;
5223 case POPEN_2:
5224 case POPEN_4:
5226 char *m1, *m2;
5227 PyObject *p1, *p2;
5229 if (mode & _O_TEXT) {
5230 m1 = "r";
5231 m2 = "w";
5232 } else {
5233 m1 = "rb";
5234 m2 = "wb";
5237 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
5238 f1 = _fdopen(fd1, m2);
5239 fd2 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
5240 f2 = _fdopen(fd2, m1);
5241 p1 = PyFile_FromFile(f1, cmdstring, m2, _PyPclose);
5242 PyFile_SetBufSize(p1, 0);
5243 p2 = PyFile_FromFile(f2, cmdstring, m1, _PyPclose);
5244 PyFile_SetBufSize(p2, 0);
5246 if (n != 4)
5247 CloseHandle(hChildStderrRdDup);
5249 f = PyTuple_Pack(2,p1,p2);
5250 Py_XDECREF(p1);
5251 Py_XDECREF(p2);
5252 file_count = 2;
5253 break;
5256 case POPEN_3:
5258 char *m1, *m2;
5259 PyObject *p1, *p2, *p3;
5261 if (mode & _O_TEXT) {
5262 m1 = "r";
5263 m2 = "w";
5264 } else {
5265 m1 = "rb";
5266 m2 = "wb";
5269 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
5270 f1 = _fdopen(fd1, m2);
5271 fd2 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
5272 f2 = _fdopen(fd2, m1);
5273 fd3 = _open_osfhandle((Py_intptr_t)hChildStderrRdDup, mode);
5274 f3 = _fdopen(fd3, m1);
5275 p1 = PyFile_FromFile(f1, cmdstring, m2, _PyPclose);
5276 p2 = PyFile_FromFile(f2, cmdstring, m1, _PyPclose);
5277 p3 = PyFile_FromFile(f3, cmdstring, m1, _PyPclose);
5278 PyFile_SetBufSize(p1, 0);
5279 PyFile_SetBufSize(p2, 0);
5280 PyFile_SetBufSize(p3, 0);
5281 f = PyTuple_Pack(3,p1,p2,p3);
5282 Py_XDECREF(p1);
5283 Py_XDECREF(p2);
5284 Py_XDECREF(p3);
5285 file_count = 3;
5286 break;
5290 if (n == POPEN_4) {
5291 if (!_PyPopenCreateProcess(cmdstring,
5292 hChildStdinRd,
5293 hChildStdoutWr,
5294 hChildStdoutWr,
5295 &hProcess))
5296 return NULL;
5298 else {
5299 if (!_PyPopenCreateProcess(cmdstring,
5300 hChildStdinRd,
5301 hChildStdoutWr,
5302 hChildStderrWr,
5303 &hProcess))
5304 return NULL;
5308 * Insert the files we've created into the process dictionary
5309 * all referencing the list with the process handle and the
5310 * initial number of files (see description below in _PyPclose).
5311 * Since if _PyPclose later tried to wait on a process when all
5312 * handles weren't closed, it could create a deadlock with the
5313 * child, we spend some energy here to try to ensure that we
5314 * either insert all file handles into the dictionary or none
5315 * at all. It's a little clumsy with the various popen modes
5316 * and variable number of files involved.
5318 if (!_PyPopenProcs) {
5319 _PyPopenProcs = PyDict_New();
5322 if (_PyPopenProcs) {
5323 PyObject *procObj, *hProcessObj, *intObj, *fileObj[3];
5324 int ins_rc[3];
5326 fileObj[0] = fileObj[1] = fileObj[2] = NULL;
5327 ins_rc[0] = ins_rc[1] = ins_rc[2] = 0;
5329 procObj = PyList_New(2);
5330 hProcessObj = PyLong_FromVoidPtr(hProcess);
5331 intObj = PyInt_FromLong(file_count);
5333 if (procObj && hProcessObj && intObj) {
5334 PyList_SetItem(procObj,0,hProcessObj);
5335 PyList_SetItem(procObj,1,intObj);
5337 fileObj[0] = PyLong_FromVoidPtr(f1);
5338 if (fileObj[0]) {
5339 ins_rc[0] = PyDict_SetItem(_PyPopenProcs,
5340 fileObj[0],
5341 procObj);
5343 if (file_count >= 2) {
5344 fileObj[1] = PyLong_FromVoidPtr(f2);
5345 if (fileObj[1]) {
5346 ins_rc[1] = PyDict_SetItem(_PyPopenProcs,
5347 fileObj[1],
5348 procObj);
5351 if (file_count >= 3) {
5352 fileObj[2] = PyLong_FromVoidPtr(f3);
5353 if (fileObj[2]) {
5354 ins_rc[2] = PyDict_SetItem(_PyPopenProcs,
5355 fileObj[2],
5356 procObj);
5360 if (ins_rc[0] < 0 || !fileObj[0] ||
5361 ins_rc[1] < 0 || (file_count > 1 && !fileObj[1]) ||
5362 ins_rc[2] < 0 || (file_count > 2 && !fileObj[2])) {
5363 /* Something failed - remove any dictionary
5364 * entries that did make it.
5366 if (!ins_rc[0] && fileObj[0]) {
5367 PyDict_DelItem(_PyPopenProcs,
5368 fileObj[0]);
5370 if (!ins_rc[1] && fileObj[1]) {
5371 PyDict_DelItem(_PyPopenProcs,
5372 fileObj[1]);
5374 if (!ins_rc[2] && fileObj[2]) {
5375 PyDict_DelItem(_PyPopenProcs,
5376 fileObj[2]);
5382 * Clean up our localized references for the dictionary keys
5383 * and value since PyDict_SetItem will Py_INCREF any copies
5384 * that got placed in the dictionary.
5386 Py_XDECREF(procObj);
5387 Py_XDECREF(fileObj[0]);
5388 Py_XDECREF(fileObj[1]);
5389 Py_XDECREF(fileObj[2]);
5392 /* Child is launched. Close the parents copy of those pipe
5393 * handles that only the child should have open. You need to
5394 * make sure that no handles to the write end of the output pipe
5395 * are maintained in this process or else the pipe will not close
5396 * when the child process exits and the ReadFile will hang. */
5398 if (!CloseHandle(hChildStdinRd))
5399 return win32_error("CloseHandle", NULL);
5401 if (!CloseHandle(hChildStdoutWr))
5402 return win32_error("CloseHandle", NULL);
5404 if ((n != 4) && (!CloseHandle(hChildStderrWr)))
5405 return win32_error("CloseHandle", NULL);
5407 return f;
5411 * Wrapper for fclose() to use for popen* files, so we can retrieve the
5412 * exit code for the child process and return as a result of the close.
5414 * This function uses the _PyPopenProcs dictionary in order to map the
5415 * input file pointer to information about the process that was
5416 * originally created by the popen* call that created the file pointer.
5417 * The dictionary uses the file pointer as a key (with one entry
5418 * inserted for each file returned by the original popen* call) and a
5419 * single list object as the value for all files from a single call.
5420 * The list object contains the Win32 process handle at [0], and a file
5421 * count at [1], which is initialized to the total number of file
5422 * handles using that list.
5424 * This function closes whichever handle it is passed, and decrements
5425 * the file count in the dictionary for the process handle pointed to
5426 * by this file. On the last close (when the file count reaches zero),
5427 * this function will wait for the child process and then return its
5428 * exit code as the result of the close() operation. This permits the
5429 * files to be closed in any order - it is always the close() of the
5430 * final handle that will return the exit code.
5432 * NOTE: This function is currently called with the GIL released.
5433 * hence we use the GILState API to manage our state.
5436 static int _PyPclose(FILE *file)
5438 int result;
5439 DWORD exit_code;
5440 HANDLE hProcess;
5441 PyObject *procObj, *hProcessObj, *intObj, *fileObj;
5442 long file_count;
5443 #ifdef WITH_THREAD
5444 PyGILState_STATE state;
5445 #endif
5447 /* Close the file handle first, to ensure it can't block the
5448 * child from exiting if it's the last handle.
5450 result = fclose(file);
5451 #ifdef WITH_THREAD
5452 state = PyGILState_Ensure();
5453 #endif
5454 if (_PyPopenProcs) {
5455 if ((fileObj = PyLong_FromVoidPtr(file)) != NULL &&
5456 (procObj = PyDict_GetItem(_PyPopenProcs,
5457 fileObj)) != NULL &&
5458 (hProcessObj = PyList_GetItem(procObj,0)) != NULL &&
5459 (intObj = PyList_GetItem(procObj,1)) != NULL) {
5461 hProcess = PyLong_AsVoidPtr(hProcessObj);
5462 file_count = PyInt_AsLong(intObj);
5464 if (file_count > 1) {
5465 /* Still other files referencing process */
5466 file_count--;
5467 PyList_SetItem(procObj,1,
5468 PyInt_FromLong(file_count));
5469 } else {
5470 /* Last file for this process */
5471 if (result != EOF &&
5472 WaitForSingleObject(hProcess, INFINITE) != WAIT_FAILED &&
5473 GetExitCodeProcess(hProcess, &exit_code)) {
5474 /* Possible truncation here in 16-bit environments, but
5475 * real exit codes are just the lower byte in any event.
5477 result = exit_code;
5478 } else {
5479 /* Indicate failure - this will cause the file object
5480 * to raise an I/O error and translate the last Win32
5481 * error code from errno. We do have a problem with
5482 * last errors that overlap the normal errno table,
5483 * but that's a consistent problem with the file object.
5485 if (result != EOF) {
5486 /* If the error wasn't from the fclose(), then
5487 * set errno for the file object error handling.
5489 errno = GetLastError();
5491 result = -1;
5494 /* Free up the native handle at this point */
5495 CloseHandle(hProcess);
5498 /* Remove this file pointer from dictionary */
5499 PyDict_DelItem(_PyPopenProcs, fileObj);
5501 if (PyDict_Size(_PyPopenProcs) == 0) {
5502 Py_DECREF(_PyPopenProcs);
5503 _PyPopenProcs = NULL;
5506 } /* if object retrieval ok */
5508 Py_XDECREF(fileObj);
5509 } /* if _PyPopenProcs */
5511 #ifdef WITH_THREAD
5512 PyGILState_Release(state);
5513 #endif
5514 return result;
5517 #else /* which OS? */
5518 static PyObject *
5519 posix_popen(PyObject *self, PyObject *args)
5521 char *name;
5522 char *mode = "r";
5523 int bufsize = -1;
5524 FILE *fp;
5525 PyObject *f;
5526 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
5527 return NULL;
5528 /* Strip mode of binary or text modifiers */
5529 if (strcmp(mode, "rb") == 0 || strcmp(mode, "rt") == 0)
5530 mode = "r";
5531 else if (strcmp(mode, "wb") == 0 || strcmp(mode, "wt") == 0)
5532 mode = "w";
5533 Py_BEGIN_ALLOW_THREADS
5534 fp = popen(name, mode);
5535 Py_END_ALLOW_THREADS
5536 if (fp == NULL)
5537 return posix_error();
5538 f = PyFile_FromFile(fp, name, mode, pclose);
5539 if (f != NULL)
5540 PyFile_SetBufSize(f, bufsize);
5541 return f;
5544 #endif /* PYOS_??? */
5545 #endif /* HAVE_POPEN */
5548 #ifdef HAVE_SETUID
5549 PyDoc_STRVAR(posix_setuid__doc__,
5550 "setuid(uid)\n\n\
5551 Set the current process's user id.");
5553 static PyObject *
5554 posix_setuid(PyObject *self, PyObject *args)
5556 long uid_arg;
5557 uid_t uid;
5558 if (!PyArg_ParseTuple(args, "l:setuid", &uid_arg))
5559 return NULL;
5560 uid = uid_arg;
5561 if (uid != uid_arg) {
5562 PyErr_SetString(PyExc_OverflowError, "user id too big");
5563 return NULL;
5565 if (setuid(uid) < 0)
5566 return posix_error();
5567 Py_INCREF(Py_None);
5568 return Py_None;
5570 #endif /* HAVE_SETUID */
5573 #ifdef HAVE_SETEUID
5574 PyDoc_STRVAR(posix_seteuid__doc__,
5575 "seteuid(uid)\n\n\
5576 Set the current process's effective user id.");
5578 static PyObject *
5579 posix_seteuid (PyObject *self, PyObject *args)
5581 long euid_arg;
5582 uid_t euid;
5583 if (!PyArg_ParseTuple(args, "l", &euid_arg))
5584 return NULL;
5585 euid = euid_arg;
5586 if (euid != euid_arg) {
5587 PyErr_SetString(PyExc_OverflowError, "user id too big");
5588 return NULL;
5590 if (seteuid(euid) < 0) {
5591 return posix_error();
5592 } else {
5593 Py_INCREF(Py_None);
5594 return Py_None;
5597 #endif /* HAVE_SETEUID */
5599 #ifdef HAVE_SETEGID
5600 PyDoc_STRVAR(posix_setegid__doc__,
5601 "setegid(gid)\n\n\
5602 Set the current process's effective group id.");
5604 static PyObject *
5605 posix_setegid (PyObject *self, PyObject *args)
5607 long egid_arg;
5608 gid_t egid;
5609 if (!PyArg_ParseTuple(args, "l", &egid_arg))
5610 return NULL;
5611 egid = egid_arg;
5612 if (egid != egid_arg) {
5613 PyErr_SetString(PyExc_OverflowError, "group id too big");
5614 return NULL;
5616 if (setegid(egid) < 0) {
5617 return posix_error();
5618 } else {
5619 Py_INCREF(Py_None);
5620 return Py_None;
5623 #endif /* HAVE_SETEGID */
5625 #ifdef HAVE_SETREUID
5626 PyDoc_STRVAR(posix_setreuid__doc__,
5627 "setreuid(ruid, euid)\n\n\
5628 Set the current process's real and effective user ids.");
5630 static PyObject *
5631 posix_setreuid (PyObject *self, PyObject *args)
5633 long ruid_arg, euid_arg;
5634 uid_t ruid, euid;
5635 if (!PyArg_ParseTuple(args, "ll", &ruid_arg, &euid_arg))
5636 return NULL;
5637 ruid = ruid_arg;
5638 euid = euid_arg;
5639 if (euid != euid_arg || ruid != ruid_arg) {
5640 PyErr_SetString(PyExc_OverflowError, "user id too big");
5641 return NULL;
5643 if (setreuid(ruid, euid) < 0) {
5644 return posix_error();
5645 } else {
5646 Py_INCREF(Py_None);
5647 return Py_None;
5650 #endif /* HAVE_SETREUID */
5652 #ifdef HAVE_SETREGID
5653 PyDoc_STRVAR(posix_setregid__doc__,
5654 "setregid(rgid, egid)\n\n\
5655 Set the current process's real and effective group ids.");
5657 static PyObject *
5658 posix_setregid (PyObject *self, PyObject *args)
5660 long rgid_arg, egid_arg;
5661 gid_t rgid, egid;
5662 if (!PyArg_ParseTuple(args, "ll", &rgid_arg, &egid_arg))
5663 return NULL;
5664 rgid = rgid_arg;
5665 egid = egid_arg;
5666 if (egid != egid_arg || rgid != rgid_arg) {
5667 PyErr_SetString(PyExc_OverflowError, "group id too big");
5668 return NULL;
5670 if (setregid(rgid, egid) < 0) {
5671 return posix_error();
5672 } else {
5673 Py_INCREF(Py_None);
5674 return Py_None;
5677 #endif /* HAVE_SETREGID */
5679 #ifdef HAVE_SETGID
5680 PyDoc_STRVAR(posix_setgid__doc__,
5681 "setgid(gid)\n\n\
5682 Set the current process's group id.");
5684 static PyObject *
5685 posix_setgid(PyObject *self, PyObject *args)
5687 long gid_arg;
5688 gid_t gid;
5689 if (!PyArg_ParseTuple(args, "l:setgid", &gid_arg))
5690 return NULL;
5691 gid = gid_arg;
5692 if (gid != gid_arg) {
5693 PyErr_SetString(PyExc_OverflowError, "group id too big");
5694 return NULL;
5696 if (setgid(gid) < 0)
5697 return posix_error();
5698 Py_INCREF(Py_None);
5699 return Py_None;
5701 #endif /* HAVE_SETGID */
5703 #ifdef HAVE_SETGROUPS
5704 PyDoc_STRVAR(posix_setgroups__doc__,
5705 "setgroups(list)\n\n\
5706 Set the groups of the current process to list.");
5708 static PyObject *
5709 posix_setgroups(PyObject *self, PyObject *groups)
5711 int i, len;
5712 gid_t grouplist[MAX_GROUPS];
5714 if (!PySequence_Check(groups)) {
5715 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
5716 return NULL;
5718 len = PySequence_Size(groups);
5719 if (len > MAX_GROUPS) {
5720 PyErr_SetString(PyExc_ValueError, "too many groups");
5721 return NULL;
5723 for(i = 0; i < len; i++) {
5724 PyObject *elem;
5725 elem = PySequence_GetItem(groups, i);
5726 if (!elem)
5727 return NULL;
5728 if (!PyInt_Check(elem)) {
5729 if (!PyLong_Check(elem)) {
5730 PyErr_SetString(PyExc_TypeError,
5731 "groups must be integers");
5732 Py_DECREF(elem);
5733 return NULL;
5734 } else {
5735 unsigned long x = PyLong_AsUnsignedLong(elem);
5736 if (PyErr_Occurred()) {
5737 PyErr_SetString(PyExc_TypeError,
5738 "group id too big");
5739 Py_DECREF(elem);
5740 return NULL;
5742 grouplist[i] = x;
5743 /* read back to see if it fits in gid_t */
5744 if (grouplist[i] != x) {
5745 PyErr_SetString(PyExc_TypeError,
5746 "group id too big");
5747 Py_DECREF(elem);
5748 return NULL;
5751 } else {
5752 long x = PyInt_AsLong(elem);
5753 grouplist[i] = x;
5754 if (grouplist[i] != x) {
5755 PyErr_SetString(PyExc_TypeError,
5756 "group id too big");
5757 Py_DECREF(elem);
5758 return NULL;
5761 Py_DECREF(elem);
5764 if (setgroups(len, grouplist) < 0)
5765 return posix_error();
5766 Py_INCREF(Py_None);
5767 return Py_None;
5769 #endif /* HAVE_SETGROUPS */
5771 #if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
5772 static PyObject *
5773 wait_helper(pid_t pid, int status, struct rusage *ru)
5775 PyObject *result;
5776 static PyObject *struct_rusage;
5778 if (pid == -1)
5779 return posix_error();
5781 if (struct_rusage == NULL) {
5782 PyObject *m = PyImport_ImportModuleNoBlock("resource");
5783 if (m == NULL)
5784 return NULL;
5785 struct_rusage = PyObject_GetAttrString(m, "struct_rusage");
5786 Py_DECREF(m);
5787 if (struct_rusage == NULL)
5788 return NULL;
5791 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
5792 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
5793 if (!result)
5794 return NULL;
5796 #ifndef doubletime
5797 #define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
5798 #endif
5800 PyStructSequence_SET_ITEM(result, 0,
5801 PyFloat_FromDouble(doubletime(ru->ru_utime)));
5802 PyStructSequence_SET_ITEM(result, 1,
5803 PyFloat_FromDouble(doubletime(ru->ru_stime)));
5804 #define SET_INT(result, index, value)\
5805 PyStructSequence_SET_ITEM(result, index, PyInt_FromLong(value))
5806 SET_INT(result, 2, ru->ru_maxrss);
5807 SET_INT(result, 3, ru->ru_ixrss);
5808 SET_INT(result, 4, ru->ru_idrss);
5809 SET_INT(result, 5, ru->ru_isrss);
5810 SET_INT(result, 6, ru->ru_minflt);
5811 SET_INT(result, 7, ru->ru_majflt);
5812 SET_INT(result, 8, ru->ru_nswap);
5813 SET_INT(result, 9, ru->ru_inblock);
5814 SET_INT(result, 10, ru->ru_oublock);
5815 SET_INT(result, 11, ru->ru_msgsnd);
5816 SET_INT(result, 12, ru->ru_msgrcv);
5817 SET_INT(result, 13, ru->ru_nsignals);
5818 SET_INT(result, 14, ru->ru_nvcsw);
5819 SET_INT(result, 15, ru->ru_nivcsw);
5820 #undef SET_INT
5822 if (PyErr_Occurred()) {
5823 Py_DECREF(result);
5824 return NULL;
5827 return Py_BuildValue("NiN", PyLong_FromPid(pid), status, result);
5829 #endif /* HAVE_WAIT3 || HAVE_WAIT4 */
5831 #ifdef HAVE_WAIT3
5832 PyDoc_STRVAR(posix_wait3__doc__,
5833 "wait3(options) -> (pid, status, rusage)\n\n\
5834 Wait for completion of a child process.");
5836 static PyObject *
5837 posix_wait3(PyObject *self, PyObject *args)
5839 pid_t pid;
5840 int options;
5841 struct rusage ru;
5842 WAIT_TYPE status;
5843 WAIT_STATUS_INT(status) = 0;
5845 if (!PyArg_ParseTuple(args, "i:wait3", &options))
5846 return NULL;
5848 Py_BEGIN_ALLOW_THREADS
5849 pid = wait3(&status, options, &ru);
5850 Py_END_ALLOW_THREADS
5852 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
5854 #endif /* HAVE_WAIT3 */
5856 #ifdef HAVE_WAIT4
5857 PyDoc_STRVAR(posix_wait4__doc__,
5858 "wait4(pid, options) -> (pid, status, rusage)\n\n\
5859 Wait for completion of a given child process.");
5861 static PyObject *
5862 posix_wait4(PyObject *self, PyObject *args)
5864 pid_t pid;
5865 int options;
5866 struct rusage ru;
5867 WAIT_TYPE status;
5868 WAIT_STATUS_INT(status) = 0;
5870 if (!PyArg_ParseTuple(args, PARSE_PID "i:wait4", &pid, &options))
5871 return NULL;
5873 Py_BEGIN_ALLOW_THREADS
5874 pid = wait4(pid, &status, options, &ru);
5875 Py_END_ALLOW_THREADS
5877 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
5879 #endif /* HAVE_WAIT4 */
5881 #ifdef HAVE_WAITPID
5882 PyDoc_STRVAR(posix_waitpid__doc__,
5883 "waitpid(pid, options) -> (pid, status)\n\n\
5884 Wait for completion of a given child process.");
5886 static PyObject *
5887 posix_waitpid(PyObject *self, PyObject *args)
5889 pid_t pid;
5890 int options;
5891 WAIT_TYPE status;
5892 WAIT_STATUS_INT(status) = 0;
5894 if (!PyArg_ParseTuple(args, PARSE_PID "i:waitpid", &pid, &options))
5895 return NULL;
5896 Py_BEGIN_ALLOW_THREADS
5897 pid = waitpid(pid, &status, options);
5898 Py_END_ALLOW_THREADS
5899 if (pid == -1)
5900 return posix_error();
5902 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
5905 #elif defined(HAVE_CWAIT)
5907 /* MS C has a variant of waitpid() that's usable for most purposes. */
5908 PyDoc_STRVAR(posix_waitpid__doc__,
5909 "waitpid(pid, options) -> (pid, status << 8)\n\n"
5910 "Wait for completion of a given process. options is ignored on Windows.");
5912 static PyObject *
5913 posix_waitpid(PyObject *self, PyObject *args)
5915 Py_intptr_t pid;
5916 int status, options;
5918 if (!PyArg_ParseTuple(args, PARSE_PID "i:waitpid", &pid, &options))
5919 return NULL;
5920 Py_BEGIN_ALLOW_THREADS
5921 pid = _cwait(&status, pid, options);
5922 Py_END_ALLOW_THREADS
5923 if (pid == -1)
5924 return posix_error();
5926 /* shift the status left a byte so this is more like the POSIX waitpid */
5927 return Py_BuildValue("Ni", PyLong_FromPid(pid), status << 8);
5929 #endif /* HAVE_WAITPID || HAVE_CWAIT */
5931 #ifdef HAVE_WAIT
5932 PyDoc_STRVAR(posix_wait__doc__,
5933 "wait() -> (pid, status)\n\n\
5934 Wait for completion of a child process.");
5936 static PyObject *
5937 posix_wait(PyObject *self, PyObject *noargs)
5939 pid_t pid;
5940 WAIT_TYPE status;
5941 WAIT_STATUS_INT(status) = 0;
5943 Py_BEGIN_ALLOW_THREADS
5944 pid = wait(&status);
5945 Py_END_ALLOW_THREADS
5946 if (pid == -1)
5947 return posix_error();
5949 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
5951 #endif
5954 PyDoc_STRVAR(posix_lstat__doc__,
5955 "lstat(path) -> stat result\n\n\
5956 Like stat(path), but do not follow symbolic links.");
5958 static PyObject *
5959 posix_lstat(PyObject *self, PyObject *args)
5961 #ifdef HAVE_LSTAT
5962 return posix_do_stat(self, args, "et:lstat", lstat, NULL, NULL);
5963 #else /* !HAVE_LSTAT */
5964 #ifdef MS_WINDOWS
5965 return posix_do_stat(self, args, "et:lstat", STAT, "U:lstat", win32_wstat);
5966 #else
5967 return posix_do_stat(self, args, "et:lstat", STAT, NULL, NULL);
5968 #endif
5969 #endif /* !HAVE_LSTAT */
5973 #ifdef HAVE_READLINK
5974 PyDoc_STRVAR(posix_readlink__doc__,
5975 "readlink(path) -> path\n\n\
5976 Return a string representing the path to which the symbolic link points.");
5978 static PyObject *
5979 posix_readlink(PyObject *self, PyObject *args)
5981 PyObject* v;
5982 char buf[MAXPATHLEN];
5983 char *path;
5984 int n;
5985 #ifdef Py_USING_UNICODE
5986 int arg_is_unicode = 0;
5987 #endif
5989 if (!PyArg_ParseTuple(args, "et:readlink",
5990 Py_FileSystemDefaultEncoding, &path))
5991 return NULL;
5992 #ifdef Py_USING_UNICODE
5993 v = PySequence_GetItem(args, 0);
5994 if (v == NULL) {
5995 PyMem_Free(path);
5996 return NULL;
5999 if (PyUnicode_Check(v)) {
6000 arg_is_unicode = 1;
6002 Py_DECREF(v);
6003 #endif
6005 Py_BEGIN_ALLOW_THREADS
6006 n = readlink(path, buf, (int) sizeof buf);
6007 Py_END_ALLOW_THREADS
6008 if (n < 0)
6009 return posix_error_with_allocated_filename(path);
6011 PyMem_Free(path);
6012 v = PyString_FromStringAndSize(buf, n);
6013 #ifdef Py_USING_UNICODE
6014 if (arg_is_unicode) {
6015 PyObject *w;
6017 w = PyUnicode_FromEncodedObject(v,
6018 Py_FileSystemDefaultEncoding,
6019 "strict");
6020 if (w != NULL) {
6021 Py_DECREF(v);
6022 v = w;
6024 else {
6025 /* fall back to the original byte string, as
6026 discussed in patch #683592 */
6027 PyErr_Clear();
6030 #endif
6031 return v;
6033 #endif /* HAVE_READLINK */
6036 #ifdef HAVE_SYMLINK
6037 PyDoc_STRVAR(posix_symlink__doc__,
6038 "symlink(src, dst)\n\n\
6039 Create a symbolic link pointing to src named dst.");
6041 static PyObject *
6042 posix_symlink(PyObject *self, PyObject *args)
6044 return posix_2str(args, "etet:symlink", symlink);
6046 #endif /* HAVE_SYMLINK */
6049 #ifdef HAVE_TIMES
6050 #if defined(PYCC_VACPP) && defined(PYOS_OS2)
6051 static long
6052 system_uptime(void)
6054 ULONG value = 0;
6056 Py_BEGIN_ALLOW_THREADS
6057 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
6058 Py_END_ALLOW_THREADS
6060 return value;
6063 static PyObject *
6064 posix_times(PyObject *self, PyObject *noargs)
6066 /* Currently Only Uptime is Provided -- Others Later */
6067 return Py_BuildValue("ddddd",
6068 (double)0 /* t.tms_utime / HZ */,
6069 (double)0 /* t.tms_stime / HZ */,
6070 (double)0 /* t.tms_cutime / HZ */,
6071 (double)0 /* t.tms_cstime / HZ */,
6072 (double)system_uptime() / 1000);
6074 #else /* not OS2 */
6075 #define NEED_TICKS_PER_SECOND
6076 static long ticks_per_second = -1;
6077 static PyObject *
6078 posix_times(PyObject *self, PyObject *noargs)
6080 struct tms t;
6081 clock_t c;
6082 errno = 0;
6083 c = times(&t);
6084 if (c == (clock_t) -1)
6085 return posix_error();
6086 return Py_BuildValue("ddddd",
6087 (double)t.tms_utime / ticks_per_second,
6088 (double)t.tms_stime / ticks_per_second,
6089 (double)t.tms_cutime / ticks_per_second,
6090 (double)t.tms_cstime / ticks_per_second,
6091 (double)c / ticks_per_second);
6093 #endif /* not OS2 */
6094 #endif /* HAVE_TIMES */
6097 #ifdef MS_WINDOWS
6098 #define HAVE_TIMES /* so the method table will pick it up */
6099 static PyObject *
6100 posix_times(PyObject *self, PyObject *noargs)
6102 FILETIME create, exit, kernel, user;
6103 HANDLE hProc;
6104 hProc = GetCurrentProcess();
6105 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
6106 /* The fields of a FILETIME structure are the hi and lo part
6107 of a 64-bit value expressed in 100 nanosecond units.
6108 1e7 is one second in such units; 1e-7 the inverse.
6109 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
6111 return Py_BuildValue(
6112 "ddddd",
6113 (double)(user.dwHighDateTime*429.4967296 +
6114 user.dwLowDateTime*1e-7),
6115 (double)(kernel.dwHighDateTime*429.4967296 +
6116 kernel.dwLowDateTime*1e-7),
6117 (double)0,
6118 (double)0,
6119 (double)0);
6121 #endif /* MS_WINDOWS */
6123 #ifdef HAVE_TIMES
6124 PyDoc_STRVAR(posix_times__doc__,
6125 "times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
6126 Return a tuple of floating point numbers indicating process times.");
6127 #endif
6130 #ifdef HAVE_GETSID
6131 PyDoc_STRVAR(posix_getsid__doc__,
6132 "getsid(pid) -> sid\n\n\
6133 Call the system call getsid().");
6135 static PyObject *
6136 posix_getsid(PyObject *self, PyObject *args)
6138 pid_t pid;
6139 int sid;
6140 if (!PyArg_ParseTuple(args, PARSE_PID ":getsid", &pid))
6141 return NULL;
6142 sid = getsid(pid);
6143 if (sid < 0)
6144 return posix_error();
6145 return PyInt_FromLong((long)sid);
6147 #endif /* HAVE_GETSID */
6150 #ifdef HAVE_SETSID
6151 PyDoc_STRVAR(posix_setsid__doc__,
6152 "setsid()\n\n\
6153 Call the system call setsid().");
6155 static PyObject *
6156 posix_setsid(PyObject *self, PyObject *noargs)
6158 if (setsid() < 0)
6159 return posix_error();
6160 Py_INCREF(Py_None);
6161 return Py_None;
6163 #endif /* HAVE_SETSID */
6165 #ifdef HAVE_SETPGID
6166 PyDoc_STRVAR(posix_setpgid__doc__,
6167 "setpgid(pid, pgrp)\n\n\
6168 Call the system call setpgid().");
6170 static PyObject *
6171 posix_setpgid(PyObject *self, PyObject *args)
6173 pid_t pid;
6174 int pgrp;
6175 if (!PyArg_ParseTuple(args, PARSE_PID "i:setpgid", &pid, &pgrp))
6176 return NULL;
6177 if (setpgid(pid, pgrp) < 0)
6178 return posix_error();
6179 Py_INCREF(Py_None);
6180 return Py_None;
6182 #endif /* HAVE_SETPGID */
6185 #ifdef HAVE_TCGETPGRP
6186 PyDoc_STRVAR(posix_tcgetpgrp__doc__,
6187 "tcgetpgrp(fd) -> pgid\n\n\
6188 Return the process group associated with the terminal given by a fd.");
6190 static PyObject *
6191 posix_tcgetpgrp(PyObject *self, PyObject *args)
6193 int fd;
6194 pid_t pgid;
6195 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
6196 return NULL;
6197 pgid = tcgetpgrp(fd);
6198 if (pgid < 0)
6199 return posix_error();
6200 return PyLong_FromPid(pgid);
6202 #endif /* HAVE_TCGETPGRP */
6205 #ifdef HAVE_TCSETPGRP
6206 PyDoc_STRVAR(posix_tcsetpgrp__doc__,
6207 "tcsetpgrp(fd, pgid)\n\n\
6208 Set the process group associated with the terminal given by a fd.");
6210 static PyObject *
6211 posix_tcsetpgrp(PyObject *self, PyObject *args)
6213 int fd;
6214 pid_t pgid;
6215 if (!PyArg_ParseTuple(args, "i" PARSE_PID ":tcsetpgrp", &fd, &pgid))
6216 return NULL;
6217 if (tcsetpgrp(fd, pgid) < 0)
6218 return posix_error();
6219 Py_INCREF(Py_None);
6220 return Py_None;
6222 #endif /* HAVE_TCSETPGRP */
6224 /* Functions acting on file descriptors */
6226 PyDoc_STRVAR(posix_open__doc__,
6227 "open(filename, flag [, mode=0777]) -> fd\n\n\
6228 Open a file (for low level IO).");
6230 static PyObject *
6231 posix_open(PyObject *self, PyObject *args)
6233 char *file = NULL;
6234 int flag;
6235 int mode = 0777;
6236 int fd;
6238 #ifdef MS_WINDOWS
6239 PyUnicodeObject *po;
6240 if (PyArg_ParseTuple(args, "Ui|i:mkdir", &po, &flag, &mode)) {
6241 Py_BEGIN_ALLOW_THREADS
6242 /* PyUnicode_AS_UNICODE OK without thread
6243 lock as it is a simple dereference. */
6244 fd = _wopen(PyUnicode_AS_UNICODE(po), flag, mode);
6245 Py_END_ALLOW_THREADS
6246 if (fd < 0)
6247 return posix_error();
6248 return PyInt_FromLong((long)fd);
6250 /* Drop the argument parsing error as narrow strings
6251 are also valid. */
6252 PyErr_Clear();
6253 #endif
6255 if (!PyArg_ParseTuple(args, "eti|i",
6256 Py_FileSystemDefaultEncoding, &file,
6257 &flag, &mode))
6258 return NULL;
6260 Py_BEGIN_ALLOW_THREADS
6261 fd = open(file, flag, mode);
6262 Py_END_ALLOW_THREADS
6263 if (fd < 0)
6264 return posix_error_with_allocated_filename(file);
6265 PyMem_Free(file);
6266 return PyInt_FromLong((long)fd);
6270 PyDoc_STRVAR(posix_close__doc__,
6271 "close(fd)\n\n\
6272 Close a file descriptor (for low level IO).");
6274 static PyObject *
6275 posix_close(PyObject *self, PyObject *args)
6277 int fd, res;
6278 if (!PyArg_ParseTuple(args, "i:close", &fd))
6279 return NULL;
6280 if (!_PyVerify_fd(fd))
6281 return posix_error();
6282 Py_BEGIN_ALLOW_THREADS
6283 res = close(fd);
6284 Py_END_ALLOW_THREADS
6285 if (res < 0)
6286 return posix_error();
6287 Py_INCREF(Py_None);
6288 return Py_None;
6292 PyDoc_STRVAR(posix_closerange__doc__,
6293 "closerange(fd_low, fd_high)\n\n\
6294 Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
6296 static PyObject *
6297 posix_closerange(PyObject *self, PyObject *args)
6299 int fd_from, fd_to, i;
6300 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
6301 return NULL;
6302 Py_BEGIN_ALLOW_THREADS
6303 for (i = fd_from; i < fd_to; i++)
6304 if (_PyVerify_fd(i))
6305 close(i);
6306 Py_END_ALLOW_THREADS
6307 Py_RETURN_NONE;
6311 PyDoc_STRVAR(posix_dup__doc__,
6312 "dup(fd) -> fd2\n\n\
6313 Return a duplicate of a file descriptor.");
6315 static PyObject *
6316 posix_dup(PyObject *self, PyObject *args)
6318 int fd;
6319 if (!PyArg_ParseTuple(args, "i:dup", &fd))
6320 return NULL;
6321 if (!_PyVerify_fd(fd))
6322 return posix_error();
6323 Py_BEGIN_ALLOW_THREADS
6324 fd = dup(fd);
6325 Py_END_ALLOW_THREADS
6326 if (fd < 0)
6327 return posix_error();
6328 return PyInt_FromLong((long)fd);
6332 PyDoc_STRVAR(posix_dup2__doc__,
6333 "dup2(old_fd, new_fd)\n\n\
6334 Duplicate file descriptor.");
6336 static PyObject *
6337 posix_dup2(PyObject *self, PyObject *args)
6339 int fd, fd2, res;
6340 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
6341 return NULL;
6342 if (!_PyVerify_fd_dup2(fd, fd2))
6343 return posix_error();
6344 Py_BEGIN_ALLOW_THREADS
6345 res = dup2(fd, fd2);
6346 Py_END_ALLOW_THREADS
6347 if (res < 0)
6348 return posix_error();
6349 Py_INCREF(Py_None);
6350 return Py_None;
6354 PyDoc_STRVAR(posix_lseek__doc__,
6355 "lseek(fd, pos, how) -> newpos\n\n\
6356 Set the current position of a file descriptor.");
6358 static PyObject *
6359 posix_lseek(PyObject *self, PyObject *args)
6361 int fd, how;
6362 #if defined(MS_WIN64) || defined(MS_WINDOWS)
6363 PY_LONG_LONG pos, res;
6364 #else
6365 off_t pos, res;
6366 #endif
6367 PyObject *posobj;
6368 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
6369 return NULL;
6370 #ifdef SEEK_SET
6371 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
6372 switch (how) {
6373 case 0: how = SEEK_SET; break;
6374 case 1: how = SEEK_CUR; break;
6375 case 2: how = SEEK_END; break;
6377 #endif /* SEEK_END */
6379 #if !defined(HAVE_LARGEFILE_SUPPORT)
6380 pos = PyInt_AsLong(posobj);
6381 #else
6382 pos = PyLong_Check(posobj) ?
6383 PyLong_AsLongLong(posobj) : PyInt_AsLong(posobj);
6384 #endif
6385 if (PyErr_Occurred())
6386 return NULL;
6388 if (!_PyVerify_fd(fd))
6389 return posix_error();
6390 Py_BEGIN_ALLOW_THREADS
6391 #if defined(MS_WIN64) || defined(MS_WINDOWS)
6392 res = _lseeki64(fd, pos, how);
6393 #else
6394 res = lseek(fd, pos, how);
6395 #endif
6396 Py_END_ALLOW_THREADS
6397 if (res < 0)
6398 return posix_error();
6400 #if !defined(HAVE_LARGEFILE_SUPPORT)
6401 return PyInt_FromLong(res);
6402 #else
6403 return PyLong_FromLongLong(res);
6404 #endif
6408 PyDoc_STRVAR(posix_read__doc__,
6409 "read(fd, buffersize) -> string\n\n\
6410 Read a file descriptor.");
6412 static PyObject *
6413 posix_read(PyObject *self, PyObject *args)
6415 int fd, size, n;
6416 PyObject *buffer;
6417 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
6418 return NULL;
6419 if (size < 0) {
6420 errno = EINVAL;
6421 return posix_error();
6423 buffer = PyString_FromStringAndSize((char *)NULL, size);
6424 if (buffer == NULL)
6425 return NULL;
6426 if (!_PyVerify_fd(fd))
6427 return posix_error();
6428 Py_BEGIN_ALLOW_THREADS
6429 n = read(fd, PyString_AsString(buffer), size);
6430 Py_END_ALLOW_THREADS
6431 if (n < 0) {
6432 Py_DECREF(buffer);
6433 return posix_error();
6435 if (n != size)
6436 _PyString_Resize(&buffer, n);
6437 return buffer;
6441 PyDoc_STRVAR(posix_write__doc__,
6442 "write(fd, string) -> byteswritten\n\n\
6443 Write a string to a file descriptor.");
6445 static PyObject *
6446 posix_write(PyObject *self, PyObject *args)
6448 Py_buffer pbuf;
6449 int fd;
6450 Py_ssize_t size;
6452 if (!PyArg_ParseTuple(args, "is*:write", &fd, &pbuf))
6453 return NULL;
6454 if (!_PyVerify_fd(fd))
6455 return posix_error();
6456 Py_BEGIN_ALLOW_THREADS
6457 size = write(fd, pbuf.buf, (size_t)pbuf.len);
6458 Py_END_ALLOW_THREADS
6459 PyBuffer_Release(&pbuf);
6460 if (size < 0)
6461 return posix_error();
6462 return PyInt_FromSsize_t(size);
6466 PyDoc_STRVAR(posix_fstat__doc__,
6467 "fstat(fd) -> stat result\n\n\
6468 Like stat(), but for an open file descriptor.");
6470 static PyObject *
6471 posix_fstat(PyObject *self, PyObject *args)
6473 int fd;
6474 STRUCT_STAT st;
6475 int res;
6476 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
6477 return NULL;
6478 #ifdef __VMS
6479 /* on OpenVMS we must ensure that all bytes are written to the file */
6480 fsync(fd);
6481 #endif
6482 if (!_PyVerify_fd(fd))
6483 return posix_error();
6484 Py_BEGIN_ALLOW_THREADS
6485 res = FSTAT(fd, &st);
6486 Py_END_ALLOW_THREADS
6487 if (res != 0) {
6488 #ifdef MS_WINDOWS
6489 return win32_error("fstat", NULL);
6490 #else
6491 return posix_error();
6492 #endif
6495 return _pystat_fromstructstat(&st);
6499 PyDoc_STRVAR(posix_fdopen__doc__,
6500 "fdopen(fd [, mode='r' [, bufsize]]) -> file_object\n\n\
6501 Return an open file object connected to a file descriptor.");
6503 static PyObject *
6504 posix_fdopen(PyObject *self, PyObject *args)
6506 int fd;
6507 char *orgmode = "r";
6508 int bufsize = -1;
6509 FILE *fp;
6510 PyObject *f;
6511 char *mode;
6512 if (!PyArg_ParseTuple(args, "i|si", &fd, &orgmode, &bufsize))
6513 return NULL;
6515 /* Sanitize mode. See fileobject.c */
6516 mode = PyMem_MALLOC(strlen(orgmode)+3);
6517 if (!mode) {
6518 PyErr_NoMemory();
6519 return NULL;
6521 strcpy(mode, orgmode);
6522 if (_PyFile_SanitizeMode(mode)) {
6523 PyMem_FREE(mode);
6524 return NULL;
6526 if (!_PyVerify_fd(fd))
6527 return posix_error();
6528 Py_BEGIN_ALLOW_THREADS
6529 #if !defined(MS_WINDOWS) && defined(HAVE_FCNTL_H)
6530 if (mode[0] == 'a') {
6531 /* try to make sure the O_APPEND flag is set */
6532 int flags;
6533 flags = fcntl(fd, F_GETFL);
6534 if (flags != -1)
6535 fcntl(fd, F_SETFL, flags | O_APPEND);
6536 fp = fdopen(fd, mode);
6537 if (fp == NULL && flags != -1)
6538 /* restore old mode if fdopen failed */
6539 fcntl(fd, F_SETFL, flags);
6540 } else {
6541 fp = fdopen(fd, mode);
6543 #else
6544 fp = fdopen(fd, mode);
6545 #endif
6546 Py_END_ALLOW_THREADS
6547 PyMem_FREE(mode);
6548 if (fp == NULL)
6549 return posix_error();
6550 f = PyFile_FromFile(fp, "<fdopen>", orgmode, fclose);
6551 if (f != NULL)
6552 PyFile_SetBufSize(f, bufsize);
6553 return f;
6556 PyDoc_STRVAR(posix_isatty__doc__,
6557 "isatty(fd) -> bool\n\n\
6558 Return True if the file descriptor 'fd' is an open file descriptor\n\
6559 connected to the slave end of a terminal.");
6561 static PyObject *
6562 posix_isatty(PyObject *self, PyObject *args)
6564 int fd;
6565 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
6566 return NULL;
6567 if (!_PyVerify_fd(fd))
6568 return PyBool_FromLong(0);
6569 return PyBool_FromLong(isatty(fd));
6572 #ifdef HAVE_PIPE
6573 PyDoc_STRVAR(posix_pipe__doc__,
6574 "pipe() -> (read_end, write_end)\n\n\
6575 Create a pipe.");
6577 static PyObject *
6578 posix_pipe(PyObject *self, PyObject *noargs)
6580 #if defined(PYOS_OS2)
6581 HFILE read, write;
6582 APIRET rc;
6584 Py_BEGIN_ALLOW_THREADS
6585 rc = DosCreatePipe( &read, &write, 4096);
6586 Py_END_ALLOW_THREADS
6587 if (rc != NO_ERROR)
6588 return os2_error(rc);
6590 return Py_BuildValue("(ii)", read, write);
6591 #else
6592 #if !defined(MS_WINDOWS)
6593 int fds[2];
6594 int res;
6595 Py_BEGIN_ALLOW_THREADS
6596 res = pipe(fds);
6597 Py_END_ALLOW_THREADS
6598 if (res != 0)
6599 return posix_error();
6600 return Py_BuildValue("(ii)", fds[0], fds[1]);
6601 #else /* MS_WINDOWS */
6602 HANDLE read, write;
6603 int read_fd, write_fd;
6604 BOOL ok;
6605 Py_BEGIN_ALLOW_THREADS
6606 ok = CreatePipe(&read, &write, NULL, 0);
6607 Py_END_ALLOW_THREADS
6608 if (!ok)
6609 return win32_error("CreatePipe", NULL);
6610 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
6611 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
6612 return Py_BuildValue("(ii)", read_fd, write_fd);
6613 #endif /* MS_WINDOWS */
6614 #endif
6616 #endif /* HAVE_PIPE */
6619 #ifdef HAVE_MKFIFO
6620 PyDoc_STRVAR(posix_mkfifo__doc__,
6621 "mkfifo(filename [, mode=0666])\n\n\
6622 Create a FIFO (a POSIX named pipe).");
6624 static PyObject *
6625 posix_mkfifo(PyObject *self, PyObject *args)
6627 char *filename;
6628 int mode = 0666;
6629 int res;
6630 if (!PyArg_ParseTuple(args, "s|i:mkfifo", &filename, &mode))
6631 return NULL;
6632 Py_BEGIN_ALLOW_THREADS
6633 res = mkfifo(filename, mode);
6634 Py_END_ALLOW_THREADS
6635 if (res < 0)
6636 return posix_error();
6637 Py_INCREF(Py_None);
6638 return Py_None;
6640 #endif
6643 #if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
6644 PyDoc_STRVAR(posix_mknod__doc__,
6645 "mknod(filename [, mode=0600, device])\n\n\
6646 Create a filesystem node (file, device special file or named pipe)\n\
6647 named filename. mode specifies both the permissions to use and the\n\
6648 type of node to be created, being combined (bitwise OR) with one of\n\
6649 S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
6650 device defines the newly created device special file (probably using\n\
6651 os.makedev()), otherwise it is ignored.");
6654 static PyObject *
6655 posix_mknod(PyObject *self, PyObject *args)
6657 char *filename;
6658 int mode = 0600;
6659 int device = 0;
6660 int res;
6661 if (!PyArg_ParseTuple(args, "s|ii:mknod", &filename, &mode, &device))
6662 return NULL;
6663 Py_BEGIN_ALLOW_THREADS
6664 res = mknod(filename, mode, device);
6665 Py_END_ALLOW_THREADS
6666 if (res < 0)
6667 return posix_error();
6668 Py_INCREF(Py_None);
6669 return Py_None;
6671 #endif
6673 #ifdef HAVE_DEVICE_MACROS
6674 PyDoc_STRVAR(posix_major__doc__,
6675 "major(device) -> major number\n\
6676 Extracts a device major number from a raw device number.");
6678 static PyObject *
6679 posix_major(PyObject *self, PyObject *args)
6681 int device;
6682 if (!PyArg_ParseTuple(args, "i:major", &device))
6683 return NULL;
6684 return PyInt_FromLong((long)major(device));
6687 PyDoc_STRVAR(posix_minor__doc__,
6688 "minor(device) -> minor number\n\
6689 Extracts a device minor number from a raw device number.");
6691 static PyObject *
6692 posix_minor(PyObject *self, PyObject *args)
6694 int device;
6695 if (!PyArg_ParseTuple(args, "i:minor", &device))
6696 return NULL;
6697 return PyInt_FromLong((long)minor(device));
6700 PyDoc_STRVAR(posix_makedev__doc__,
6701 "makedev(major, minor) -> device number\n\
6702 Composes a raw device number from the major and minor device numbers.");
6704 static PyObject *
6705 posix_makedev(PyObject *self, PyObject *args)
6707 int major, minor;
6708 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
6709 return NULL;
6710 return PyInt_FromLong((long)makedev(major, minor));
6712 #endif /* device macros */
6715 #ifdef HAVE_FTRUNCATE
6716 PyDoc_STRVAR(posix_ftruncate__doc__,
6717 "ftruncate(fd, length)\n\n\
6718 Truncate a file to a specified length.");
6720 static PyObject *
6721 posix_ftruncate(PyObject *self, PyObject *args)
6723 int fd;
6724 off_t length;
6725 int res;
6726 PyObject *lenobj;
6728 if (!PyArg_ParseTuple(args, "iO:ftruncate", &fd, &lenobj))
6729 return NULL;
6731 #if !defined(HAVE_LARGEFILE_SUPPORT)
6732 length = PyInt_AsLong(lenobj);
6733 #else
6734 length = PyLong_Check(lenobj) ?
6735 PyLong_AsLongLong(lenobj) : PyInt_AsLong(lenobj);
6736 #endif
6737 if (PyErr_Occurred())
6738 return NULL;
6740 Py_BEGIN_ALLOW_THREADS
6741 res = ftruncate(fd, length);
6742 Py_END_ALLOW_THREADS
6743 if (res < 0)
6744 return posix_error();
6745 Py_INCREF(Py_None);
6746 return Py_None;
6748 #endif
6750 #ifdef HAVE_PUTENV
6751 PyDoc_STRVAR(posix_putenv__doc__,
6752 "putenv(key, value)\n\n\
6753 Change or add an environment variable.");
6755 /* Save putenv() parameters as values here, so we can collect them when they
6756 * get re-set with another call for the same key. */
6757 static PyObject *posix_putenv_garbage;
6759 static PyObject *
6760 posix_putenv(PyObject *self, PyObject *args)
6762 char *s1, *s2;
6763 char *newenv;
6764 PyObject *newstr;
6765 size_t len;
6767 if (!PyArg_ParseTuple(args, "ss:putenv", &s1, &s2))
6768 return NULL;
6770 #if defined(PYOS_OS2)
6771 if (stricmp(s1, "BEGINLIBPATH") == 0) {
6772 APIRET rc;
6774 rc = DosSetExtLIBPATH(s2, BEGIN_LIBPATH);
6775 if (rc != NO_ERROR)
6776 return os2_error(rc);
6778 } else if (stricmp(s1, "ENDLIBPATH") == 0) {
6779 APIRET rc;
6781 rc = DosSetExtLIBPATH(s2, END_LIBPATH);
6782 if (rc != NO_ERROR)
6783 return os2_error(rc);
6784 } else {
6785 #endif
6787 /* XXX This can leak memory -- not easy to fix :-( */
6788 len = strlen(s1) + strlen(s2) + 2;
6789 /* len includes space for a trailing \0; the size arg to
6790 PyString_FromStringAndSize does not count that */
6791 newstr = PyString_FromStringAndSize(NULL, (int)len - 1);
6792 if (newstr == NULL)
6793 return PyErr_NoMemory();
6794 newenv = PyString_AS_STRING(newstr);
6795 PyOS_snprintf(newenv, len, "%s=%s", s1, s2);
6796 if (putenv(newenv)) {
6797 Py_DECREF(newstr);
6798 posix_error();
6799 return NULL;
6801 /* Install the first arg and newstr in posix_putenv_garbage;
6802 * this will cause previous value to be collected. This has to
6803 * happen after the real putenv() call because the old value
6804 * was still accessible until then. */
6805 if (PyDict_SetItem(posix_putenv_garbage,
6806 PyTuple_GET_ITEM(args, 0), newstr)) {
6807 /* really not much we can do; just leak */
6808 PyErr_Clear();
6810 else {
6811 Py_DECREF(newstr);
6814 #if defined(PYOS_OS2)
6816 #endif
6817 Py_INCREF(Py_None);
6818 return Py_None;
6820 #endif /* putenv */
6822 #ifdef HAVE_UNSETENV
6823 PyDoc_STRVAR(posix_unsetenv__doc__,
6824 "unsetenv(key)\n\n\
6825 Delete an environment variable.");
6827 static PyObject *
6828 posix_unsetenv(PyObject *self, PyObject *args)
6830 char *s1;
6832 if (!PyArg_ParseTuple(args, "s:unsetenv", &s1))
6833 return NULL;
6835 unsetenv(s1);
6837 /* Remove the key from posix_putenv_garbage;
6838 * this will cause it to be collected. This has to
6839 * happen after the real unsetenv() call because the
6840 * old value was still accessible until then.
6842 if (PyDict_DelItem(posix_putenv_garbage,
6843 PyTuple_GET_ITEM(args, 0))) {
6844 /* really not much we can do; just leak */
6845 PyErr_Clear();
6848 Py_INCREF(Py_None);
6849 return Py_None;
6851 #endif /* unsetenv */
6853 PyDoc_STRVAR(posix_strerror__doc__,
6854 "strerror(code) -> string\n\n\
6855 Translate an error code to a message string.");
6857 static PyObject *
6858 posix_strerror(PyObject *self, PyObject *args)
6860 int code;
6861 char *message;
6862 if (!PyArg_ParseTuple(args, "i:strerror", &code))
6863 return NULL;
6864 message = strerror(code);
6865 if (message == NULL) {
6866 PyErr_SetString(PyExc_ValueError,
6867 "strerror() argument out of range");
6868 return NULL;
6870 return PyString_FromString(message);
6874 #ifdef HAVE_SYS_WAIT_H
6876 #ifdef WCOREDUMP
6877 PyDoc_STRVAR(posix_WCOREDUMP__doc__,
6878 "WCOREDUMP(status) -> bool\n\n\
6879 Return True if the process returning 'status' was dumped to a core file.");
6881 static PyObject *
6882 posix_WCOREDUMP(PyObject *self, PyObject *args)
6884 WAIT_TYPE status;
6885 WAIT_STATUS_INT(status) = 0;
6887 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
6888 return NULL;
6890 return PyBool_FromLong(WCOREDUMP(status));
6892 #endif /* WCOREDUMP */
6894 #ifdef WIFCONTINUED
6895 PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
6896 "WIFCONTINUED(status) -> bool\n\n\
6897 Return True if the process returning 'status' was continued from a\n\
6898 job control stop.");
6900 static PyObject *
6901 posix_WIFCONTINUED(PyObject *self, PyObject *args)
6903 WAIT_TYPE status;
6904 WAIT_STATUS_INT(status) = 0;
6906 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
6907 return NULL;
6909 return PyBool_FromLong(WIFCONTINUED(status));
6911 #endif /* WIFCONTINUED */
6913 #ifdef WIFSTOPPED
6914 PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
6915 "WIFSTOPPED(status) -> bool\n\n\
6916 Return True if the process returning 'status' was stopped.");
6918 static PyObject *
6919 posix_WIFSTOPPED(PyObject *self, PyObject *args)
6921 WAIT_TYPE status;
6922 WAIT_STATUS_INT(status) = 0;
6924 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
6925 return NULL;
6927 return PyBool_FromLong(WIFSTOPPED(status));
6929 #endif /* WIFSTOPPED */
6931 #ifdef WIFSIGNALED
6932 PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
6933 "WIFSIGNALED(status) -> bool\n\n\
6934 Return True if the process returning 'status' was terminated by a signal.");
6936 static PyObject *
6937 posix_WIFSIGNALED(PyObject *self, PyObject *args)
6939 WAIT_TYPE status;
6940 WAIT_STATUS_INT(status) = 0;
6942 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
6943 return NULL;
6945 return PyBool_FromLong(WIFSIGNALED(status));
6947 #endif /* WIFSIGNALED */
6949 #ifdef WIFEXITED
6950 PyDoc_STRVAR(posix_WIFEXITED__doc__,
6951 "WIFEXITED(status) -> bool\n\n\
6952 Return true if the process returning 'status' exited using the exit()\n\
6953 system call.");
6955 static PyObject *
6956 posix_WIFEXITED(PyObject *self, PyObject *args)
6958 WAIT_TYPE status;
6959 WAIT_STATUS_INT(status) = 0;
6961 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
6962 return NULL;
6964 return PyBool_FromLong(WIFEXITED(status));
6966 #endif /* WIFEXITED */
6968 #ifdef WEXITSTATUS
6969 PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
6970 "WEXITSTATUS(status) -> integer\n\n\
6971 Return the process return code from 'status'.");
6973 static PyObject *
6974 posix_WEXITSTATUS(PyObject *self, PyObject *args)
6976 WAIT_TYPE status;
6977 WAIT_STATUS_INT(status) = 0;
6979 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
6980 return NULL;
6982 return Py_BuildValue("i", WEXITSTATUS(status));
6984 #endif /* WEXITSTATUS */
6986 #ifdef WTERMSIG
6987 PyDoc_STRVAR(posix_WTERMSIG__doc__,
6988 "WTERMSIG(status) -> integer\n\n\
6989 Return the signal that terminated the process that provided the 'status'\n\
6990 value.");
6992 static PyObject *
6993 posix_WTERMSIG(PyObject *self, PyObject *args)
6995 WAIT_TYPE status;
6996 WAIT_STATUS_INT(status) = 0;
6998 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
6999 return NULL;
7001 return Py_BuildValue("i", WTERMSIG(status));
7003 #endif /* WTERMSIG */
7005 #ifdef WSTOPSIG
7006 PyDoc_STRVAR(posix_WSTOPSIG__doc__,
7007 "WSTOPSIG(status) -> integer\n\n\
7008 Return the signal that stopped the process that provided\n\
7009 the 'status' value.");
7011 static PyObject *
7012 posix_WSTOPSIG(PyObject *self, PyObject *args)
7014 WAIT_TYPE status;
7015 WAIT_STATUS_INT(status) = 0;
7017 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &WAIT_STATUS_INT(status)))
7018 return NULL;
7020 return Py_BuildValue("i", WSTOPSIG(status));
7022 #endif /* WSTOPSIG */
7024 #endif /* HAVE_SYS_WAIT_H */
7027 #if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
7028 #ifdef _SCO_DS
7029 /* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
7030 needed definitions in sys/statvfs.h */
7031 #define _SVID3
7032 #endif
7033 #include <sys/statvfs.h>
7035 static PyObject*
7036 _pystatvfs_fromstructstatvfs(struct statvfs st) {
7037 PyObject *v = PyStructSequence_New(&StatVFSResultType);
7038 if (v == NULL)
7039 return NULL;
7041 #if !defined(HAVE_LARGEFILE_SUPPORT)
7042 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long) st.f_bsize));
7043 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long) st.f_frsize));
7044 PyStructSequence_SET_ITEM(v, 2, PyInt_FromLong((long) st.f_blocks));
7045 PyStructSequence_SET_ITEM(v, 3, PyInt_FromLong((long) st.f_bfree));
7046 PyStructSequence_SET_ITEM(v, 4, PyInt_FromLong((long) st.f_bavail));
7047 PyStructSequence_SET_ITEM(v, 5, PyInt_FromLong((long) st.f_files));
7048 PyStructSequence_SET_ITEM(v, 6, PyInt_FromLong((long) st.f_ffree));
7049 PyStructSequence_SET_ITEM(v, 7, PyInt_FromLong((long) st.f_favail));
7050 PyStructSequence_SET_ITEM(v, 8, PyInt_FromLong((long) st.f_flag));
7051 PyStructSequence_SET_ITEM(v, 9, PyInt_FromLong((long) st.f_namemax));
7052 #else
7053 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long) st.f_bsize));
7054 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long) st.f_frsize));
7055 PyStructSequence_SET_ITEM(v, 2,
7056 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
7057 PyStructSequence_SET_ITEM(v, 3,
7058 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
7059 PyStructSequence_SET_ITEM(v, 4,
7060 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
7061 PyStructSequence_SET_ITEM(v, 5,
7062 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
7063 PyStructSequence_SET_ITEM(v, 6,
7064 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
7065 PyStructSequence_SET_ITEM(v, 7,
7066 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
7067 PyStructSequence_SET_ITEM(v, 8, PyInt_FromLong((long) st.f_flag));
7068 PyStructSequence_SET_ITEM(v, 9, PyInt_FromLong((long) st.f_namemax));
7069 #endif
7071 return v;
7074 PyDoc_STRVAR(posix_fstatvfs__doc__,
7075 "fstatvfs(fd) -> statvfs result\n\n\
7076 Perform an fstatvfs system call on the given fd.");
7078 static PyObject *
7079 posix_fstatvfs(PyObject *self, PyObject *args)
7081 int fd, res;
7082 struct statvfs st;
7084 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
7085 return NULL;
7086 Py_BEGIN_ALLOW_THREADS
7087 res = fstatvfs(fd, &st);
7088 Py_END_ALLOW_THREADS
7089 if (res != 0)
7090 return posix_error();
7092 return _pystatvfs_fromstructstatvfs(st);
7094 #endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
7097 #if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
7098 #include <sys/statvfs.h>
7100 PyDoc_STRVAR(posix_statvfs__doc__,
7101 "statvfs(path) -> statvfs result\n\n\
7102 Perform a statvfs system call on the given path.");
7104 static PyObject *
7105 posix_statvfs(PyObject *self, PyObject *args)
7107 char *path;
7108 int res;
7109 struct statvfs st;
7110 if (!PyArg_ParseTuple(args, "s:statvfs", &path))
7111 return NULL;
7112 Py_BEGIN_ALLOW_THREADS
7113 res = statvfs(path, &st);
7114 Py_END_ALLOW_THREADS
7115 if (res != 0)
7116 return posix_error_with_filename(path);
7118 return _pystatvfs_fromstructstatvfs(st);
7120 #endif /* HAVE_STATVFS */
7123 #ifdef HAVE_TEMPNAM
7124 PyDoc_STRVAR(posix_tempnam__doc__,
7125 "tempnam([dir[, prefix]]) -> string\n\n\
7126 Return a unique name for a temporary file.\n\
7127 The directory and a prefix may be specified as strings; they may be omitted\n\
7128 or None if not needed.");
7130 static PyObject *
7131 posix_tempnam(PyObject *self, PyObject *args)
7133 PyObject *result = NULL;
7134 char *dir = NULL;
7135 char *pfx = NULL;
7136 char *name;
7138 if (!PyArg_ParseTuple(args, "|zz:tempnam", &dir, &pfx))
7139 return NULL;
7141 if (PyErr_Warn(PyExc_RuntimeWarning,
7142 "tempnam is a potential security risk to your program") < 0)
7143 return NULL;
7145 #ifdef MS_WINDOWS
7146 name = _tempnam(dir, pfx);
7147 #else
7148 name = tempnam(dir, pfx);
7149 #endif
7150 if (name == NULL)
7151 return PyErr_NoMemory();
7152 result = PyString_FromString(name);
7153 free(name);
7154 return result;
7156 #endif
7159 #ifdef HAVE_TMPFILE
7160 PyDoc_STRVAR(posix_tmpfile__doc__,
7161 "tmpfile() -> file object\n\n\
7162 Create a temporary file with no directory entries.");
7164 static PyObject *
7165 posix_tmpfile(PyObject *self, PyObject *noargs)
7167 FILE *fp;
7169 fp = tmpfile();
7170 if (fp == NULL)
7171 return posix_error();
7172 return PyFile_FromFile(fp, "<tmpfile>", "w+b", fclose);
7174 #endif
7177 #ifdef HAVE_TMPNAM
7178 PyDoc_STRVAR(posix_tmpnam__doc__,
7179 "tmpnam() -> string\n\n\
7180 Return a unique name for a temporary file.");
7182 static PyObject *
7183 posix_tmpnam(PyObject *self, PyObject *noargs)
7185 char buffer[L_tmpnam];
7186 char *name;
7188 if (PyErr_Warn(PyExc_RuntimeWarning,
7189 "tmpnam is a potential security risk to your program") < 0)
7190 return NULL;
7192 #ifdef USE_TMPNAM_R
7193 name = tmpnam_r(buffer);
7194 #else
7195 name = tmpnam(buffer);
7196 #endif
7197 if (name == NULL) {
7198 PyObject *err = Py_BuildValue("is", 0,
7199 #ifdef USE_TMPNAM_R
7200 "unexpected NULL from tmpnam_r"
7201 #else
7202 "unexpected NULL from tmpnam"
7203 #endif
7205 PyErr_SetObject(PyExc_OSError, err);
7206 Py_XDECREF(err);
7207 return NULL;
7209 return PyString_FromString(buffer);
7211 #endif
7214 /* This is used for fpathconf(), pathconf(), confstr() and sysconf().
7215 * It maps strings representing configuration variable names to
7216 * integer values, allowing those functions to be called with the
7217 * magic names instead of polluting the module's namespace with tons of
7218 * rarely-used constants. There are three separate tables that use
7219 * these definitions.
7221 * This code is always included, even if none of the interfaces that
7222 * need it are included. The #if hackery needed to avoid it would be
7223 * sufficiently pervasive that it's not worth the loss of readability.
7225 struct constdef {
7226 char *name;
7227 long value;
7230 static int
7231 conv_confname(PyObject *arg, int *valuep, struct constdef *table,
7232 size_t tablesize)
7234 if (PyInt_Check(arg)) {
7235 *valuep = PyInt_AS_LONG(arg);
7236 return 1;
7238 if (PyString_Check(arg)) {
7239 /* look up the value in the table using a binary search */
7240 size_t lo = 0;
7241 size_t mid;
7242 size_t hi = tablesize;
7243 int cmp;
7244 char *confname = PyString_AS_STRING(arg);
7245 while (lo < hi) {
7246 mid = (lo + hi) / 2;
7247 cmp = strcmp(confname, table[mid].name);
7248 if (cmp < 0)
7249 hi = mid;
7250 else if (cmp > 0)
7251 lo = mid + 1;
7252 else {
7253 *valuep = table[mid].value;
7254 return 1;
7257 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
7259 else
7260 PyErr_SetString(PyExc_TypeError,
7261 "configuration names must be strings or integers");
7262 return 0;
7266 #if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
7267 static struct constdef posix_constants_pathconf[] = {
7268 #ifdef _PC_ABI_AIO_XFER_MAX
7269 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
7270 #endif
7271 #ifdef _PC_ABI_ASYNC_IO
7272 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
7273 #endif
7274 #ifdef _PC_ASYNC_IO
7275 {"PC_ASYNC_IO", _PC_ASYNC_IO},
7276 #endif
7277 #ifdef _PC_CHOWN_RESTRICTED
7278 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
7279 #endif
7280 #ifdef _PC_FILESIZEBITS
7281 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
7282 #endif
7283 #ifdef _PC_LAST
7284 {"PC_LAST", _PC_LAST},
7285 #endif
7286 #ifdef _PC_LINK_MAX
7287 {"PC_LINK_MAX", _PC_LINK_MAX},
7288 #endif
7289 #ifdef _PC_MAX_CANON
7290 {"PC_MAX_CANON", _PC_MAX_CANON},
7291 #endif
7292 #ifdef _PC_MAX_INPUT
7293 {"PC_MAX_INPUT", _PC_MAX_INPUT},
7294 #endif
7295 #ifdef _PC_NAME_MAX
7296 {"PC_NAME_MAX", _PC_NAME_MAX},
7297 #endif
7298 #ifdef _PC_NO_TRUNC
7299 {"PC_NO_TRUNC", _PC_NO_TRUNC},
7300 #endif
7301 #ifdef _PC_PATH_MAX
7302 {"PC_PATH_MAX", _PC_PATH_MAX},
7303 #endif
7304 #ifdef _PC_PIPE_BUF
7305 {"PC_PIPE_BUF", _PC_PIPE_BUF},
7306 #endif
7307 #ifdef _PC_PRIO_IO
7308 {"PC_PRIO_IO", _PC_PRIO_IO},
7309 #endif
7310 #ifdef _PC_SOCK_MAXBUF
7311 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
7312 #endif
7313 #ifdef _PC_SYNC_IO
7314 {"PC_SYNC_IO", _PC_SYNC_IO},
7315 #endif
7316 #ifdef _PC_VDISABLE
7317 {"PC_VDISABLE", _PC_VDISABLE},
7318 #endif
7321 static int
7322 conv_path_confname(PyObject *arg, int *valuep)
7324 return conv_confname(arg, valuep, posix_constants_pathconf,
7325 sizeof(posix_constants_pathconf)
7326 / sizeof(struct constdef));
7328 #endif
7330 #ifdef HAVE_FPATHCONF
7331 PyDoc_STRVAR(posix_fpathconf__doc__,
7332 "fpathconf(fd, name) -> integer\n\n\
7333 Return the configuration limit name for the file descriptor fd.\n\
7334 If there is no limit, return -1.");
7336 static PyObject *
7337 posix_fpathconf(PyObject *self, PyObject *args)
7339 PyObject *result = NULL;
7340 int name, fd;
7342 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
7343 conv_path_confname, &name)) {
7344 long limit;
7346 errno = 0;
7347 limit = fpathconf(fd, name);
7348 if (limit == -1 && errno != 0)
7349 posix_error();
7350 else
7351 result = PyInt_FromLong(limit);
7353 return result;
7355 #endif
7358 #ifdef HAVE_PATHCONF
7359 PyDoc_STRVAR(posix_pathconf__doc__,
7360 "pathconf(path, name) -> integer\n\n\
7361 Return the configuration limit name for the file or directory path.\n\
7362 If there is no limit, return -1.");
7364 static PyObject *
7365 posix_pathconf(PyObject *self, PyObject *args)
7367 PyObject *result = NULL;
7368 int name;
7369 char *path;
7371 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
7372 conv_path_confname, &name)) {
7373 long limit;
7375 errno = 0;
7376 limit = pathconf(path, name);
7377 if (limit == -1 && errno != 0) {
7378 if (errno == EINVAL)
7379 /* could be a path or name problem */
7380 posix_error();
7381 else
7382 posix_error_with_filename(path);
7384 else
7385 result = PyInt_FromLong(limit);
7387 return result;
7389 #endif
7391 #ifdef HAVE_CONFSTR
7392 static struct constdef posix_constants_confstr[] = {
7393 #ifdef _CS_ARCHITECTURE
7394 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
7395 #endif
7396 #ifdef _CS_HOSTNAME
7397 {"CS_HOSTNAME", _CS_HOSTNAME},
7398 #endif
7399 #ifdef _CS_HW_PROVIDER
7400 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
7401 #endif
7402 #ifdef _CS_HW_SERIAL
7403 {"CS_HW_SERIAL", _CS_HW_SERIAL},
7404 #endif
7405 #ifdef _CS_INITTAB_NAME
7406 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
7407 #endif
7408 #ifdef _CS_LFS64_CFLAGS
7409 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
7410 #endif
7411 #ifdef _CS_LFS64_LDFLAGS
7412 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
7413 #endif
7414 #ifdef _CS_LFS64_LIBS
7415 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
7416 #endif
7417 #ifdef _CS_LFS64_LINTFLAGS
7418 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
7419 #endif
7420 #ifdef _CS_LFS_CFLAGS
7421 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
7422 #endif
7423 #ifdef _CS_LFS_LDFLAGS
7424 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
7425 #endif
7426 #ifdef _CS_LFS_LIBS
7427 {"CS_LFS_LIBS", _CS_LFS_LIBS},
7428 #endif
7429 #ifdef _CS_LFS_LINTFLAGS
7430 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
7431 #endif
7432 #ifdef _CS_MACHINE
7433 {"CS_MACHINE", _CS_MACHINE},
7434 #endif
7435 #ifdef _CS_PATH
7436 {"CS_PATH", _CS_PATH},
7437 #endif
7438 #ifdef _CS_RELEASE
7439 {"CS_RELEASE", _CS_RELEASE},
7440 #endif
7441 #ifdef _CS_SRPC_DOMAIN
7442 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
7443 #endif
7444 #ifdef _CS_SYSNAME
7445 {"CS_SYSNAME", _CS_SYSNAME},
7446 #endif
7447 #ifdef _CS_VERSION
7448 {"CS_VERSION", _CS_VERSION},
7449 #endif
7450 #ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
7451 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
7452 #endif
7453 #ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
7454 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
7455 #endif
7456 #ifdef _CS_XBS5_ILP32_OFF32_LIBS
7457 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
7458 #endif
7459 #ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
7460 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
7461 #endif
7462 #ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
7463 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
7464 #endif
7465 #ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
7466 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
7467 #endif
7468 #ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
7469 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
7470 #endif
7471 #ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
7472 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
7473 #endif
7474 #ifdef _CS_XBS5_LP64_OFF64_CFLAGS
7475 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
7476 #endif
7477 #ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
7478 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
7479 #endif
7480 #ifdef _CS_XBS5_LP64_OFF64_LIBS
7481 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
7482 #endif
7483 #ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
7484 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
7485 #endif
7486 #ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
7487 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
7488 #endif
7489 #ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
7490 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
7491 #endif
7492 #ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
7493 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
7494 #endif
7495 #ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
7496 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
7497 #endif
7498 #ifdef _MIPS_CS_AVAIL_PROCESSORS
7499 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
7500 #endif
7501 #ifdef _MIPS_CS_BASE
7502 {"MIPS_CS_BASE", _MIPS_CS_BASE},
7503 #endif
7504 #ifdef _MIPS_CS_HOSTID
7505 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
7506 #endif
7507 #ifdef _MIPS_CS_HW_NAME
7508 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
7509 #endif
7510 #ifdef _MIPS_CS_NUM_PROCESSORS
7511 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
7512 #endif
7513 #ifdef _MIPS_CS_OSREL_MAJ
7514 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
7515 #endif
7516 #ifdef _MIPS_CS_OSREL_MIN
7517 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
7518 #endif
7519 #ifdef _MIPS_CS_OSREL_PATCH
7520 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
7521 #endif
7522 #ifdef _MIPS_CS_OS_NAME
7523 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
7524 #endif
7525 #ifdef _MIPS_CS_OS_PROVIDER
7526 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
7527 #endif
7528 #ifdef _MIPS_CS_PROCESSORS
7529 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
7530 #endif
7531 #ifdef _MIPS_CS_SERIAL
7532 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
7533 #endif
7534 #ifdef _MIPS_CS_VENDOR
7535 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
7536 #endif
7539 static int
7540 conv_confstr_confname(PyObject *arg, int *valuep)
7542 return conv_confname(arg, valuep, posix_constants_confstr,
7543 sizeof(posix_constants_confstr)
7544 / sizeof(struct constdef));
7547 PyDoc_STRVAR(posix_confstr__doc__,
7548 "confstr(name) -> string\n\n\
7549 Return a string-valued system configuration variable.");
7551 static PyObject *
7552 posix_confstr(PyObject *self, PyObject *args)
7554 PyObject *result = NULL;
7555 int name;
7556 char buffer[256];
7558 if (PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name)) {
7559 int len;
7561 errno = 0;
7562 len = confstr(name, buffer, sizeof(buffer));
7563 if (len == 0) {
7564 if (errno) {
7565 posix_error();
7567 else {
7568 result = Py_None;
7569 Py_INCREF(Py_None);
7572 else {
7573 if ((unsigned int)len >= sizeof(buffer)) {
7574 result = PyString_FromStringAndSize(NULL, len-1);
7575 if (result != NULL)
7576 confstr(name, PyString_AS_STRING(result), len);
7578 else
7579 result = PyString_FromStringAndSize(buffer, len-1);
7582 return result;
7584 #endif
7587 #ifdef HAVE_SYSCONF
7588 static struct constdef posix_constants_sysconf[] = {
7589 #ifdef _SC_2_CHAR_TERM
7590 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
7591 #endif
7592 #ifdef _SC_2_C_BIND
7593 {"SC_2_C_BIND", _SC_2_C_BIND},
7594 #endif
7595 #ifdef _SC_2_C_DEV
7596 {"SC_2_C_DEV", _SC_2_C_DEV},
7597 #endif
7598 #ifdef _SC_2_C_VERSION
7599 {"SC_2_C_VERSION", _SC_2_C_VERSION},
7600 #endif
7601 #ifdef _SC_2_FORT_DEV
7602 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
7603 #endif
7604 #ifdef _SC_2_FORT_RUN
7605 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
7606 #endif
7607 #ifdef _SC_2_LOCALEDEF
7608 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
7609 #endif
7610 #ifdef _SC_2_SW_DEV
7611 {"SC_2_SW_DEV", _SC_2_SW_DEV},
7612 #endif
7613 #ifdef _SC_2_UPE
7614 {"SC_2_UPE", _SC_2_UPE},
7615 #endif
7616 #ifdef _SC_2_VERSION
7617 {"SC_2_VERSION", _SC_2_VERSION},
7618 #endif
7619 #ifdef _SC_ABI_ASYNCHRONOUS_IO
7620 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
7621 #endif
7622 #ifdef _SC_ACL
7623 {"SC_ACL", _SC_ACL},
7624 #endif
7625 #ifdef _SC_AIO_LISTIO_MAX
7626 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
7627 #endif
7628 #ifdef _SC_AIO_MAX
7629 {"SC_AIO_MAX", _SC_AIO_MAX},
7630 #endif
7631 #ifdef _SC_AIO_PRIO_DELTA_MAX
7632 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
7633 #endif
7634 #ifdef _SC_ARG_MAX
7635 {"SC_ARG_MAX", _SC_ARG_MAX},
7636 #endif
7637 #ifdef _SC_ASYNCHRONOUS_IO
7638 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
7639 #endif
7640 #ifdef _SC_ATEXIT_MAX
7641 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
7642 #endif
7643 #ifdef _SC_AUDIT
7644 {"SC_AUDIT", _SC_AUDIT},
7645 #endif
7646 #ifdef _SC_AVPHYS_PAGES
7647 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
7648 #endif
7649 #ifdef _SC_BC_BASE_MAX
7650 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
7651 #endif
7652 #ifdef _SC_BC_DIM_MAX
7653 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
7654 #endif
7655 #ifdef _SC_BC_SCALE_MAX
7656 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
7657 #endif
7658 #ifdef _SC_BC_STRING_MAX
7659 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
7660 #endif
7661 #ifdef _SC_CAP
7662 {"SC_CAP", _SC_CAP},
7663 #endif
7664 #ifdef _SC_CHARCLASS_NAME_MAX
7665 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
7666 #endif
7667 #ifdef _SC_CHAR_BIT
7668 {"SC_CHAR_BIT", _SC_CHAR_BIT},
7669 #endif
7670 #ifdef _SC_CHAR_MAX
7671 {"SC_CHAR_MAX", _SC_CHAR_MAX},
7672 #endif
7673 #ifdef _SC_CHAR_MIN
7674 {"SC_CHAR_MIN", _SC_CHAR_MIN},
7675 #endif
7676 #ifdef _SC_CHILD_MAX
7677 {"SC_CHILD_MAX", _SC_CHILD_MAX},
7678 #endif
7679 #ifdef _SC_CLK_TCK
7680 {"SC_CLK_TCK", _SC_CLK_TCK},
7681 #endif
7682 #ifdef _SC_COHER_BLKSZ
7683 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
7684 #endif
7685 #ifdef _SC_COLL_WEIGHTS_MAX
7686 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
7687 #endif
7688 #ifdef _SC_DCACHE_ASSOC
7689 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
7690 #endif
7691 #ifdef _SC_DCACHE_BLKSZ
7692 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
7693 #endif
7694 #ifdef _SC_DCACHE_LINESZ
7695 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
7696 #endif
7697 #ifdef _SC_DCACHE_SZ
7698 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
7699 #endif
7700 #ifdef _SC_DCACHE_TBLKSZ
7701 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
7702 #endif
7703 #ifdef _SC_DELAYTIMER_MAX
7704 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
7705 #endif
7706 #ifdef _SC_EQUIV_CLASS_MAX
7707 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
7708 #endif
7709 #ifdef _SC_EXPR_NEST_MAX
7710 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
7711 #endif
7712 #ifdef _SC_FSYNC
7713 {"SC_FSYNC", _SC_FSYNC},
7714 #endif
7715 #ifdef _SC_GETGR_R_SIZE_MAX
7716 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
7717 #endif
7718 #ifdef _SC_GETPW_R_SIZE_MAX
7719 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
7720 #endif
7721 #ifdef _SC_ICACHE_ASSOC
7722 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
7723 #endif
7724 #ifdef _SC_ICACHE_BLKSZ
7725 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
7726 #endif
7727 #ifdef _SC_ICACHE_LINESZ
7728 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
7729 #endif
7730 #ifdef _SC_ICACHE_SZ
7731 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
7732 #endif
7733 #ifdef _SC_INF
7734 {"SC_INF", _SC_INF},
7735 #endif
7736 #ifdef _SC_INT_MAX
7737 {"SC_INT_MAX", _SC_INT_MAX},
7738 #endif
7739 #ifdef _SC_INT_MIN
7740 {"SC_INT_MIN", _SC_INT_MIN},
7741 #endif
7742 #ifdef _SC_IOV_MAX
7743 {"SC_IOV_MAX", _SC_IOV_MAX},
7744 #endif
7745 #ifdef _SC_IP_SECOPTS
7746 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
7747 #endif
7748 #ifdef _SC_JOB_CONTROL
7749 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
7750 #endif
7751 #ifdef _SC_KERN_POINTERS
7752 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
7753 #endif
7754 #ifdef _SC_KERN_SIM
7755 {"SC_KERN_SIM", _SC_KERN_SIM},
7756 #endif
7757 #ifdef _SC_LINE_MAX
7758 {"SC_LINE_MAX", _SC_LINE_MAX},
7759 #endif
7760 #ifdef _SC_LOGIN_NAME_MAX
7761 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
7762 #endif
7763 #ifdef _SC_LOGNAME_MAX
7764 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
7765 #endif
7766 #ifdef _SC_LONG_BIT
7767 {"SC_LONG_BIT", _SC_LONG_BIT},
7768 #endif
7769 #ifdef _SC_MAC
7770 {"SC_MAC", _SC_MAC},
7771 #endif
7772 #ifdef _SC_MAPPED_FILES
7773 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
7774 #endif
7775 #ifdef _SC_MAXPID
7776 {"SC_MAXPID", _SC_MAXPID},
7777 #endif
7778 #ifdef _SC_MB_LEN_MAX
7779 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
7780 #endif
7781 #ifdef _SC_MEMLOCK
7782 {"SC_MEMLOCK", _SC_MEMLOCK},
7783 #endif
7784 #ifdef _SC_MEMLOCK_RANGE
7785 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
7786 #endif
7787 #ifdef _SC_MEMORY_PROTECTION
7788 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
7789 #endif
7790 #ifdef _SC_MESSAGE_PASSING
7791 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
7792 #endif
7793 #ifdef _SC_MMAP_FIXED_ALIGNMENT
7794 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
7795 #endif
7796 #ifdef _SC_MQ_OPEN_MAX
7797 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
7798 #endif
7799 #ifdef _SC_MQ_PRIO_MAX
7800 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
7801 #endif
7802 #ifdef _SC_NACLS_MAX
7803 {"SC_NACLS_MAX", _SC_NACLS_MAX},
7804 #endif
7805 #ifdef _SC_NGROUPS_MAX
7806 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
7807 #endif
7808 #ifdef _SC_NL_ARGMAX
7809 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
7810 #endif
7811 #ifdef _SC_NL_LANGMAX
7812 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
7813 #endif
7814 #ifdef _SC_NL_MSGMAX
7815 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
7816 #endif
7817 #ifdef _SC_NL_NMAX
7818 {"SC_NL_NMAX", _SC_NL_NMAX},
7819 #endif
7820 #ifdef _SC_NL_SETMAX
7821 {"SC_NL_SETMAX", _SC_NL_SETMAX},
7822 #endif
7823 #ifdef _SC_NL_TEXTMAX
7824 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
7825 #endif
7826 #ifdef _SC_NPROCESSORS_CONF
7827 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
7828 #endif
7829 #ifdef _SC_NPROCESSORS_ONLN
7830 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
7831 #endif
7832 #ifdef _SC_NPROC_CONF
7833 {"SC_NPROC_CONF", _SC_NPROC_CONF},
7834 #endif
7835 #ifdef _SC_NPROC_ONLN
7836 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
7837 #endif
7838 #ifdef _SC_NZERO
7839 {"SC_NZERO", _SC_NZERO},
7840 #endif
7841 #ifdef _SC_OPEN_MAX
7842 {"SC_OPEN_MAX", _SC_OPEN_MAX},
7843 #endif
7844 #ifdef _SC_PAGESIZE
7845 {"SC_PAGESIZE", _SC_PAGESIZE},
7846 #endif
7847 #ifdef _SC_PAGE_SIZE
7848 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
7849 #endif
7850 #ifdef _SC_PASS_MAX
7851 {"SC_PASS_MAX", _SC_PASS_MAX},
7852 #endif
7853 #ifdef _SC_PHYS_PAGES
7854 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
7855 #endif
7856 #ifdef _SC_PII
7857 {"SC_PII", _SC_PII},
7858 #endif
7859 #ifdef _SC_PII_INTERNET
7860 {"SC_PII_INTERNET", _SC_PII_INTERNET},
7861 #endif
7862 #ifdef _SC_PII_INTERNET_DGRAM
7863 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
7864 #endif
7865 #ifdef _SC_PII_INTERNET_STREAM
7866 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
7867 #endif
7868 #ifdef _SC_PII_OSI
7869 {"SC_PII_OSI", _SC_PII_OSI},
7870 #endif
7871 #ifdef _SC_PII_OSI_CLTS
7872 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
7873 #endif
7874 #ifdef _SC_PII_OSI_COTS
7875 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
7876 #endif
7877 #ifdef _SC_PII_OSI_M
7878 {"SC_PII_OSI_M", _SC_PII_OSI_M},
7879 #endif
7880 #ifdef _SC_PII_SOCKET
7881 {"SC_PII_SOCKET", _SC_PII_SOCKET},
7882 #endif
7883 #ifdef _SC_PII_XTI
7884 {"SC_PII_XTI", _SC_PII_XTI},
7885 #endif
7886 #ifdef _SC_POLL
7887 {"SC_POLL", _SC_POLL},
7888 #endif
7889 #ifdef _SC_PRIORITIZED_IO
7890 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
7891 #endif
7892 #ifdef _SC_PRIORITY_SCHEDULING
7893 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
7894 #endif
7895 #ifdef _SC_REALTIME_SIGNALS
7896 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
7897 #endif
7898 #ifdef _SC_RE_DUP_MAX
7899 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
7900 #endif
7901 #ifdef _SC_RTSIG_MAX
7902 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
7903 #endif
7904 #ifdef _SC_SAVED_IDS
7905 {"SC_SAVED_IDS", _SC_SAVED_IDS},
7906 #endif
7907 #ifdef _SC_SCHAR_MAX
7908 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
7909 #endif
7910 #ifdef _SC_SCHAR_MIN
7911 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
7912 #endif
7913 #ifdef _SC_SELECT
7914 {"SC_SELECT", _SC_SELECT},
7915 #endif
7916 #ifdef _SC_SEMAPHORES
7917 {"SC_SEMAPHORES", _SC_SEMAPHORES},
7918 #endif
7919 #ifdef _SC_SEM_NSEMS_MAX
7920 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
7921 #endif
7922 #ifdef _SC_SEM_VALUE_MAX
7923 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
7924 #endif
7925 #ifdef _SC_SHARED_MEMORY_OBJECTS
7926 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
7927 #endif
7928 #ifdef _SC_SHRT_MAX
7929 {"SC_SHRT_MAX", _SC_SHRT_MAX},
7930 #endif
7931 #ifdef _SC_SHRT_MIN
7932 {"SC_SHRT_MIN", _SC_SHRT_MIN},
7933 #endif
7934 #ifdef _SC_SIGQUEUE_MAX
7935 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
7936 #endif
7937 #ifdef _SC_SIGRT_MAX
7938 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
7939 #endif
7940 #ifdef _SC_SIGRT_MIN
7941 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
7942 #endif
7943 #ifdef _SC_SOFTPOWER
7944 {"SC_SOFTPOWER", _SC_SOFTPOWER},
7945 #endif
7946 #ifdef _SC_SPLIT_CACHE
7947 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
7948 #endif
7949 #ifdef _SC_SSIZE_MAX
7950 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
7951 #endif
7952 #ifdef _SC_STACK_PROT
7953 {"SC_STACK_PROT", _SC_STACK_PROT},
7954 #endif
7955 #ifdef _SC_STREAM_MAX
7956 {"SC_STREAM_MAX", _SC_STREAM_MAX},
7957 #endif
7958 #ifdef _SC_SYNCHRONIZED_IO
7959 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
7960 #endif
7961 #ifdef _SC_THREADS
7962 {"SC_THREADS", _SC_THREADS},
7963 #endif
7964 #ifdef _SC_THREAD_ATTR_STACKADDR
7965 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
7966 #endif
7967 #ifdef _SC_THREAD_ATTR_STACKSIZE
7968 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
7969 #endif
7970 #ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
7971 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
7972 #endif
7973 #ifdef _SC_THREAD_KEYS_MAX
7974 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
7975 #endif
7976 #ifdef _SC_THREAD_PRIORITY_SCHEDULING
7977 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
7978 #endif
7979 #ifdef _SC_THREAD_PRIO_INHERIT
7980 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
7981 #endif
7982 #ifdef _SC_THREAD_PRIO_PROTECT
7983 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
7984 #endif
7985 #ifdef _SC_THREAD_PROCESS_SHARED
7986 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
7987 #endif
7988 #ifdef _SC_THREAD_SAFE_FUNCTIONS
7989 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
7990 #endif
7991 #ifdef _SC_THREAD_STACK_MIN
7992 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
7993 #endif
7994 #ifdef _SC_THREAD_THREADS_MAX
7995 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
7996 #endif
7997 #ifdef _SC_TIMERS
7998 {"SC_TIMERS", _SC_TIMERS},
7999 #endif
8000 #ifdef _SC_TIMER_MAX
8001 {"SC_TIMER_MAX", _SC_TIMER_MAX},
8002 #endif
8003 #ifdef _SC_TTY_NAME_MAX
8004 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
8005 #endif
8006 #ifdef _SC_TZNAME_MAX
8007 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
8008 #endif
8009 #ifdef _SC_T_IOV_MAX
8010 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
8011 #endif
8012 #ifdef _SC_UCHAR_MAX
8013 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
8014 #endif
8015 #ifdef _SC_UINT_MAX
8016 {"SC_UINT_MAX", _SC_UINT_MAX},
8017 #endif
8018 #ifdef _SC_UIO_MAXIOV
8019 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
8020 #endif
8021 #ifdef _SC_ULONG_MAX
8022 {"SC_ULONG_MAX", _SC_ULONG_MAX},
8023 #endif
8024 #ifdef _SC_USHRT_MAX
8025 {"SC_USHRT_MAX", _SC_USHRT_MAX},
8026 #endif
8027 #ifdef _SC_VERSION
8028 {"SC_VERSION", _SC_VERSION},
8029 #endif
8030 #ifdef _SC_WORD_BIT
8031 {"SC_WORD_BIT", _SC_WORD_BIT},
8032 #endif
8033 #ifdef _SC_XBS5_ILP32_OFF32
8034 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
8035 #endif
8036 #ifdef _SC_XBS5_ILP32_OFFBIG
8037 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
8038 #endif
8039 #ifdef _SC_XBS5_LP64_OFF64
8040 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
8041 #endif
8042 #ifdef _SC_XBS5_LPBIG_OFFBIG
8043 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
8044 #endif
8045 #ifdef _SC_XOPEN_CRYPT
8046 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
8047 #endif
8048 #ifdef _SC_XOPEN_ENH_I18N
8049 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
8050 #endif
8051 #ifdef _SC_XOPEN_LEGACY
8052 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
8053 #endif
8054 #ifdef _SC_XOPEN_REALTIME
8055 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
8056 #endif
8057 #ifdef _SC_XOPEN_REALTIME_THREADS
8058 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
8059 #endif
8060 #ifdef _SC_XOPEN_SHM
8061 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
8062 #endif
8063 #ifdef _SC_XOPEN_UNIX
8064 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
8065 #endif
8066 #ifdef _SC_XOPEN_VERSION
8067 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
8068 #endif
8069 #ifdef _SC_XOPEN_XCU_VERSION
8070 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
8071 #endif
8072 #ifdef _SC_XOPEN_XPG2
8073 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
8074 #endif
8075 #ifdef _SC_XOPEN_XPG3
8076 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
8077 #endif
8078 #ifdef _SC_XOPEN_XPG4
8079 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
8080 #endif
8083 static int
8084 conv_sysconf_confname(PyObject *arg, int *valuep)
8086 return conv_confname(arg, valuep, posix_constants_sysconf,
8087 sizeof(posix_constants_sysconf)
8088 / sizeof(struct constdef));
8091 PyDoc_STRVAR(posix_sysconf__doc__,
8092 "sysconf(name) -> integer\n\n\
8093 Return an integer-valued system configuration variable.");
8095 static PyObject *
8096 posix_sysconf(PyObject *self, PyObject *args)
8098 PyObject *result = NULL;
8099 int name;
8101 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
8102 int value;
8104 errno = 0;
8105 value = sysconf(name);
8106 if (value == -1 && errno != 0)
8107 posix_error();
8108 else
8109 result = PyInt_FromLong(value);
8111 return result;
8113 #endif
8116 /* This code is used to ensure that the tables of configuration value names
8117 * are in sorted order as required by conv_confname(), and also to build the
8118 * the exported dictionaries that are used to publish information about the
8119 * names available on the host platform.
8121 * Sorting the table at runtime ensures that the table is properly ordered
8122 * when used, even for platforms we're not able to test on. It also makes
8123 * it easier to add additional entries to the tables.
8126 static int
8127 cmp_constdefs(const void *v1, const void *v2)
8129 const struct constdef *c1 =
8130 (const struct constdef *) v1;
8131 const struct constdef *c2 =
8132 (const struct constdef *) v2;
8134 return strcmp(c1->name, c2->name);
8137 static int
8138 setup_confname_table(struct constdef *table, size_t tablesize,
8139 char *tablename, PyObject *module)
8141 PyObject *d = NULL;
8142 size_t i;
8144 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
8145 d = PyDict_New();
8146 if (d == NULL)
8147 return -1;
8149 for (i=0; i < tablesize; ++i) {
8150 PyObject *o = PyInt_FromLong(table[i].value);
8151 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
8152 Py_XDECREF(o);
8153 Py_DECREF(d);
8154 return -1;
8156 Py_DECREF(o);
8158 return PyModule_AddObject(module, tablename, d);
8161 /* Return -1 on failure, 0 on success. */
8162 static int
8163 setup_confname_tables(PyObject *module)
8165 #if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
8166 if (setup_confname_table(posix_constants_pathconf,
8167 sizeof(posix_constants_pathconf)
8168 / sizeof(struct constdef),
8169 "pathconf_names", module))
8170 return -1;
8171 #endif
8172 #ifdef HAVE_CONFSTR
8173 if (setup_confname_table(posix_constants_confstr,
8174 sizeof(posix_constants_confstr)
8175 / sizeof(struct constdef),
8176 "confstr_names", module))
8177 return -1;
8178 #endif
8179 #ifdef HAVE_SYSCONF
8180 if (setup_confname_table(posix_constants_sysconf,
8181 sizeof(posix_constants_sysconf)
8182 / sizeof(struct constdef),
8183 "sysconf_names", module))
8184 return -1;
8185 #endif
8186 return 0;
8190 PyDoc_STRVAR(posix_abort__doc__,
8191 "abort() -> does not return!\n\n\
8192 Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
8193 in the hardest way possible on the hosting operating system.");
8195 static PyObject *
8196 posix_abort(PyObject *self, PyObject *noargs)
8198 abort();
8199 /*NOTREACHED*/
8200 Py_FatalError("abort() called from Python code didn't abort!");
8201 return NULL;
8204 #ifdef MS_WINDOWS
8205 PyDoc_STRVAR(win32_startfile__doc__,
8206 "startfile(filepath [, operation]) - Start a file with its associated\n\
8207 application.\n\
8209 When \"operation\" is not specified or \"open\", this acts like\n\
8210 double-clicking the file in Explorer, or giving the file name as an\n\
8211 argument to the DOS \"start\" command: the file is opened with whatever\n\
8212 application (if any) its extension is associated.\n\
8213 When another \"operation\" is given, it specifies what should be done with\n\
8214 the file. A typical operation is \"print\".\n\
8216 startfile returns as soon as the associated application is launched.\n\
8217 There is no option to wait for the application to close, and no way\n\
8218 to retrieve the application's exit status.\n\
8220 The filepath is relative to the current directory. If you want to use\n\
8221 an absolute path, make sure the first character is not a slash (\"/\");\n\
8222 the underlying Win32 ShellExecute function doesn't work if it is.");
8224 static PyObject *
8225 win32_startfile(PyObject *self, PyObject *args)
8227 char *filepath;
8228 char *operation = NULL;
8229 HINSTANCE rc;
8231 PyObject *unipath, *woperation = NULL;
8232 if (!PyArg_ParseTuple(args, "U|s:startfile",
8233 &unipath, &operation)) {
8234 PyErr_Clear();
8235 goto normal;
8238 if (operation) {
8239 woperation = PyUnicode_DecodeASCII(operation,
8240 strlen(operation), NULL);
8241 if (!woperation) {
8242 PyErr_Clear();
8243 operation = NULL;
8244 goto normal;
8248 Py_BEGIN_ALLOW_THREADS
8249 rc = ShellExecuteW((HWND)0, woperation ? PyUnicode_AS_UNICODE(woperation) : 0,
8250 PyUnicode_AS_UNICODE(unipath),
8251 NULL, NULL, SW_SHOWNORMAL);
8252 Py_END_ALLOW_THREADS
8254 Py_XDECREF(woperation);
8255 if (rc <= (HINSTANCE)32) {
8256 PyObject *errval = win32_error_unicode("startfile",
8257 PyUnicode_AS_UNICODE(unipath));
8258 return errval;
8260 Py_INCREF(Py_None);
8261 return Py_None;
8263 normal:
8264 if (!PyArg_ParseTuple(args, "et|s:startfile",
8265 Py_FileSystemDefaultEncoding, &filepath,
8266 &operation))
8267 return NULL;
8268 Py_BEGIN_ALLOW_THREADS
8269 rc = ShellExecute((HWND)0, operation, filepath,
8270 NULL, NULL, SW_SHOWNORMAL);
8271 Py_END_ALLOW_THREADS
8272 if (rc <= (HINSTANCE)32) {
8273 PyObject *errval = win32_error("startfile", filepath);
8274 PyMem_Free(filepath);
8275 return errval;
8277 PyMem_Free(filepath);
8278 Py_INCREF(Py_None);
8279 return Py_None;
8281 #endif /* MS_WINDOWS */
8283 #ifdef HAVE_GETLOADAVG
8284 PyDoc_STRVAR(posix_getloadavg__doc__,
8285 "getloadavg() -> (float, float, float)\n\n\
8286 Return the number of processes in the system run queue averaged over\n\
8287 the last 1, 5, and 15 minutes or raises OSError if the load average\n\
8288 was unobtainable");
8290 static PyObject *
8291 posix_getloadavg(PyObject *self, PyObject *noargs)
8293 double loadavg[3];
8294 if (getloadavg(loadavg, 3)!=3) {
8295 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
8296 return NULL;
8297 } else
8298 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
8300 #endif
8302 #ifdef MS_WINDOWS
8304 PyDoc_STRVAR(win32_urandom__doc__,
8305 "urandom(n) -> str\n\n\
8306 Return a string of n random bytes suitable for cryptographic use.");
8308 typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
8309 LPCSTR pszContainer, LPCSTR pszProvider, DWORD dwProvType,\
8310 DWORD dwFlags );
8311 typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV hProv, DWORD dwLen,\
8312 BYTE *pbBuffer );
8314 static CRYPTGENRANDOM pCryptGenRandom = NULL;
8315 /* This handle is never explicitly released. Instead, the operating
8316 system will release it when the process terminates. */
8317 static HCRYPTPROV hCryptProv = 0;
8319 static PyObject*
8320 win32_urandom(PyObject *self, PyObject *args)
8322 int howMany;
8323 PyObject* result;
8325 /* Read arguments */
8326 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
8327 return NULL;
8328 if (howMany < 0)
8329 return PyErr_Format(PyExc_ValueError,
8330 "negative argument not allowed");
8332 if (hCryptProv == 0) {
8333 HINSTANCE hAdvAPI32 = NULL;
8334 CRYPTACQUIRECONTEXTA pCryptAcquireContext = NULL;
8336 /* Obtain handle to the DLL containing CryptoAPI
8337 This should not fail */
8338 hAdvAPI32 = GetModuleHandle("advapi32.dll");
8339 if(hAdvAPI32 == NULL)
8340 return win32_error("GetModuleHandle", NULL);
8342 /* Obtain pointers to the CryptoAPI functions
8343 This will fail on some early versions of Win95 */
8344 pCryptAcquireContext = (CRYPTACQUIRECONTEXTA)GetProcAddress(
8345 hAdvAPI32,
8346 "CryptAcquireContextA");
8347 if (pCryptAcquireContext == NULL)
8348 return PyErr_Format(PyExc_NotImplementedError,
8349 "CryptAcquireContextA not found");
8351 pCryptGenRandom = (CRYPTGENRANDOM)GetProcAddress(
8352 hAdvAPI32, "CryptGenRandom");
8353 if (pCryptGenRandom == NULL)
8354 return PyErr_Format(PyExc_NotImplementedError,
8355 "CryptGenRandom not found");
8357 /* Acquire context */
8358 if (! pCryptAcquireContext(&hCryptProv, NULL, NULL,
8359 PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
8360 return win32_error("CryptAcquireContext", NULL);
8363 /* Allocate bytes */
8364 result = PyString_FromStringAndSize(NULL, howMany);
8365 if (result != NULL) {
8366 /* Get random data */
8367 memset(PyString_AS_STRING(result), 0, howMany); /* zero seed */
8368 if (! pCryptGenRandom(hCryptProv, howMany, (unsigned char*)
8369 PyString_AS_STRING(result))) {
8370 Py_DECREF(result);
8371 return win32_error("CryptGenRandom", NULL);
8374 return result;
8376 #endif
8378 #ifdef __VMS
8379 /* Use openssl random routine */
8380 #include <openssl/rand.h>
8381 PyDoc_STRVAR(vms_urandom__doc__,
8382 "urandom(n) -> str\n\n\
8383 Return a string of n random bytes suitable for cryptographic use.");
8385 static PyObject*
8386 vms_urandom(PyObject *self, PyObject *args)
8388 int howMany;
8389 PyObject* result;
8391 /* Read arguments */
8392 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
8393 return NULL;
8394 if (howMany < 0)
8395 return PyErr_Format(PyExc_ValueError,
8396 "negative argument not allowed");
8398 /* Allocate bytes */
8399 result = PyString_FromStringAndSize(NULL, howMany);
8400 if (result != NULL) {
8401 /* Get random data */
8402 if (RAND_pseudo_bytes((unsigned char*)
8403 PyString_AS_STRING(result),
8404 howMany) < 0) {
8405 Py_DECREF(result);
8406 return PyErr_Format(PyExc_ValueError,
8407 "RAND_pseudo_bytes");
8410 return result;
8412 #endif
8414 #ifdef HAVE_SETRESUID
8415 PyDoc_STRVAR(posix_setresuid__doc__,
8416 "setresuid(ruid, euid, suid)\n\n\
8417 Set the current process's real, effective, and saved user ids.");
8419 static PyObject*
8420 posix_setresuid (PyObject *self, PyObject *args)
8422 /* We assume uid_t is no larger than a long. */
8423 long ruid, euid, suid;
8424 if (!PyArg_ParseTuple(args, "lll", &ruid, &euid, &suid))
8425 return NULL;
8426 if (setresuid(ruid, euid, suid) < 0)
8427 return posix_error();
8428 Py_RETURN_NONE;
8430 #endif
8432 #ifdef HAVE_SETRESGID
8433 PyDoc_STRVAR(posix_setresgid__doc__,
8434 "setresgid(rgid, egid, sgid)\n\n\
8435 Set the current process's real, effective, and saved group ids.");
8437 static PyObject*
8438 posix_setresgid (PyObject *self, PyObject *args)
8440 /* We assume uid_t is no larger than a long. */
8441 long rgid, egid, sgid;
8442 if (!PyArg_ParseTuple(args, "lll", &rgid, &egid, &sgid))
8443 return NULL;
8444 if (setresgid(rgid, egid, sgid) < 0)
8445 return posix_error();
8446 Py_RETURN_NONE;
8448 #endif
8450 #ifdef HAVE_GETRESUID
8451 PyDoc_STRVAR(posix_getresuid__doc__,
8452 "getresuid() -> (ruid, euid, suid)\n\n\
8453 Get tuple of the current process's real, effective, and saved user ids.");
8455 static PyObject*
8456 posix_getresuid (PyObject *self, PyObject *noargs)
8458 uid_t ruid, euid, suid;
8459 long l_ruid, l_euid, l_suid;
8460 if (getresuid(&ruid, &euid, &suid) < 0)
8461 return posix_error();
8462 /* Force the values into long's as we don't know the size of uid_t. */
8463 l_ruid = ruid;
8464 l_euid = euid;
8465 l_suid = suid;
8466 return Py_BuildValue("(lll)", l_ruid, l_euid, l_suid);
8468 #endif
8470 #ifdef HAVE_GETRESGID
8471 PyDoc_STRVAR(posix_getresgid__doc__,
8472 "getresgid() -> (rgid, egid, sgid)\n\n\
8473 Get tuple of the current process's real, effective, and saved user ids.");
8475 static PyObject*
8476 posix_getresgid (PyObject *self, PyObject *noargs)
8478 uid_t rgid, egid, sgid;
8479 long l_rgid, l_egid, l_sgid;
8480 if (getresgid(&rgid, &egid, &sgid) < 0)
8481 return posix_error();
8482 /* Force the values into long's as we don't know the size of uid_t. */
8483 l_rgid = rgid;
8484 l_egid = egid;
8485 l_sgid = sgid;
8486 return Py_BuildValue("(lll)", l_rgid, l_egid, l_sgid);
8488 #endif
8490 static PyMethodDef posix_methods[] = {
8491 {"access", posix_access, METH_VARARGS, posix_access__doc__},
8492 #ifdef HAVE_TTYNAME
8493 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
8494 #endif
8495 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
8496 #ifdef HAVE_CHFLAGS
8497 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
8498 #endif /* HAVE_CHFLAGS */
8499 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
8500 #ifdef HAVE_FCHMOD
8501 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
8502 #endif /* HAVE_FCHMOD */
8503 #ifdef HAVE_CHOWN
8504 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
8505 #endif /* HAVE_CHOWN */
8506 #ifdef HAVE_LCHMOD
8507 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
8508 #endif /* HAVE_LCHMOD */
8509 #ifdef HAVE_FCHOWN
8510 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
8511 #endif /* HAVE_FCHOWN */
8512 #ifdef HAVE_LCHFLAGS
8513 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
8514 #endif /* HAVE_LCHFLAGS */
8515 #ifdef HAVE_LCHOWN
8516 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
8517 #endif /* HAVE_LCHOWN */
8518 #ifdef HAVE_CHROOT
8519 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
8520 #endif
8521 #ifdef HAVE_CTERMID
8522 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
8523 #endif
8524 #ifdef HAVE_GETCWD
8525 {"getcwd", posix_getcwd, METH_NOARGS, posix_getcwd__doc__},
8526 #ifdef Py_USING_UNICODE
8527 {"getcwdu", posix_getcwdu, METH_NOARGS, posix_getcwdu__doc__},
8528 #endif
8529 #endif
8530 #ifdef HAVE_LINK
8531 {"link", posix_link, METH_VARARGS, posix_link__doc__},
8532 #endif /* HAVE_LINK */
8533 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
8534 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
8535 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
8536 #ifdef HAVE_NICE
8537 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
8538 #endif /* HAVE_NICE */
8539 #ifdef HAVE_READLINK
8540 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
8541 #endif /* HAVE_READLINK */
8542 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
8543 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
8544 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
8545 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
8546 #ifdef HAVE_SYMLINK
8547 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
8548 #endif /* HAVE_SYMLINK */
8549 #ifdef HAVE_SYSTEM
8550 {"system", posix_system, METH_VARARGS, posix_system__doc__},
8551 #endif
8552 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
8553 #ifdef HAVE_UNAME
8554 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
8555 #endif /* HAVE_UNAME */
8556 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
8557 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
8558 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
8559 #ifdef HAVE_TIMES
8560 {"times", posix_times, METH_NOARGS, posix_times__doc__},
8561 #endif /* HAVE_TIMES */
8562 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
8563 #ifdef HAVE_EXECV
8564 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
8565 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
8566 #endif /* HAVE_EXECV */
8567 #ifdef HAVE_SPAWNV
8568 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
8569 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
8570 #if defined(PYOS_OS2)
8571 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
8572 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
8573 #endif /* PYOS_OS2 */
8574 #endif /* HAVE_SPAWNV */
8575 #ifdef HAVE_FORK1
8576 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
8577 #endif /* HAVE_FORK1 */
8578 #ifdef HAVE_FORK
8579 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
8580 #endif /* HAVE_FORK */
8581 #if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
8582 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
8583 #endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
8584 #ifdef HAVE_FORKPTY
8585 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
8586 #endif /* HAVE_FORKPTY */
8587 #ifdef HAVE_GETEGID
8588 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
8589 #endif /* HAVE_GETEGID */
8590 #ifdef HAVE_GETEUID
8591 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
8592 #endif /* HAVE_GETEUID */
8593 #ifdef HAVE_GETGID
8594 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
8595 #endif /* HAVE_GETGID */
8596 #ifdef HAVE_GETGROUPS
8597 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
8598 #endif
8599 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
8600 #ifdef HAVE_GETPGRP
8601 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
8602 #endif /* HAVE_GETPGRP */
8603 #ifdef HAVE_GETPPID
8604 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
8605 #endif /* HAVE_GETPPID */
8606 #ifdef HAVE_GETUID
8607 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
8608 #endif /* HAVE_GETUID */
8609 #ifdef HAVE_GETLOGIN
8610 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
8611 #endif
8612 #ifdef HAVE_KILL
8613 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
8614 #endif /* HAVE_KILL */
8615 #ifdef HAVE_KILLPG
8616 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
8617 #endif /* HAVE_KILLPG */
8618 #ifdef HAVE_PLOCK
8619 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
8620 #endif /* HAVE_PLOCK */
8621 #ifdef HAVE_POPEN
8622 {"popen", posix_popen, METH_VARARGS, posix_popen__doc__},
8623 #ifdef MS_WINDOWS
8624 {"popen2", win32_popen2, METH_VARARGS},
8625 {"popen3", win32_popen3, METH_VARARGS},
8626 {"popen4", win32_popen4, METH_VARARGS},
8627 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
8628 #else
8629 #if defined(PYOS_OS2) && defined(PYCC_GCC)
8630 {"popen2", os2emx_popen2, METH_VARARGS},
8631 {"popen3", os2emx_popen3, METH_VARARGS},
8632 {"popen4", os2emx_popen4, METH_VARARGS},
8633 #endif
8634 #endif
8635 #endif /* HAVE_POPEN */
8636 #ifdef HAVE_SETUID
8637 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
8638 #endif /* HAVE_SETUID */
8639 #ifdef HAVE_SETEUID
8640 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
8641 #endif /* HAVE_SETEUID */
8642 #ifdef HAVE_SETEGID
8643 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
8644 #endif /* HAVE_SETEGID */
8645 #ifdef HAVE_SETREUID
8646 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
8647 #endif /* HAVE_SETREUID */
8648 #ifdef HAVE_SETREGID
8649 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
8650 #endif /* HAVE_SETREGID */
8651 #ifdef HAVE_SETGID
8652 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
8653 #endif /* HAVE_SETGID */
8654 #ifdef HAVE_SETGROUPS
8655 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
8656 #endif /* HAVE_SETGROUPS */
8657 #ifdef HAVE_INITGROUPS
8658 {"initgroups", posix_initgroups, METH_VARARGS, posix_initgroups__doc__},
8659 #endif /* HAVE_INITGROUPS */
8660 #ifdef HAVE_GETPGID
8661 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
8662 #endif /* HAVE_GETPGID */
8663 #ifdef HAVE_SETPGRP
8664 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
8665 #endif /* HAVE_SETPGRP */
8666 #ifdef HAVE_WAIT
8667 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
8668 #endif /* HAVE_WAIT */
8669 #ifdef HAVE_WAIT3
8670 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
8671 #endif /* HAVE_WAIT3 */
8672 #ifdef HAVE_WAIT4
8673 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
8674 #endif /* HAVE_WAIT4 */
8675 #if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
8676 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
8677 #endif /* HAVE_WAITPID */
8678 #ifdef HAVE_GETSID
8679 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
8680 #endif /* HAVE_GETSID */
8681 #ifdef HAVE_SETSID
8682 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
8683 #endif /* HAVE_SETSID */
8684 #ifdef HAVE_SETPGID
8685 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
8686 #endif /* HAVE_SETPGID */
8687 #ifdef HAVE_TCGETPGRP
8688 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
8689 #endif /* HAVE_TCGETPGRP */
8690 #ifdef HAVE_TCSETPGRP
8691 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
8692 #endif /* HAVE_TCSETPGRP */
8693 {"open", posix_open, METH_VARARGS, posix_open__doc__},
8694 {"close", posix_close, METH_VARARGS, posix_close__doc__},
8695 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
8696 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
8697 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
8698 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
8699 {"read", posix_read, METH_VARARGS, posix_read__doc__},
8700 {"write", posix_write, METH_VARARGS, posix_write__doc__},
8701 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
8702 {"fdopen", posix_fdopen, METH_VARARGS, posix_fdopen__doc__},
8703 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
8704 #ifdef HAVE_PIPE
8705 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
8706 #endif
8707 #ifdef HAVE_MKFIFO
8708 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
8709 #endif
8710 #if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
8711 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
8712 #endif
8713 #ifdef HAVE_DEVICE_MACROS
8714 {"major", posix_major, METH_VARARGS, posix_major__doc__},
8715 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
8716 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
8717 #endif
8718 #ifdef HAVE_FTRUNCATE
8719 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
8720 #endif
8721 #ifdef HAVE_PUTENV
8722 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
8723 #endif
8724 #ifdef HAVE_UNSETENV
8725 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
8726 #endif
8727 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
8728 #ifdef HAVE_FCHDIR
8729 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
8730 #endif
8731 #ifdef HAVE_FSYNC
8732 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
8733 #endif
8734 #ifdef HAVE_FDATASYNC
8735 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
8736 #endif
8737 #ifdef HAVE_SYS_WAIT_H
8738 #ifdef WCOREDUMP
8739 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
8740 #endif /* WCOREDUMP */
8741 #ifdef WIFCONTINUED
8742 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
8743 #endif /* WIFCONTINUED */
8744 #ifdef WIFSTOPPED
8745 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
8746 #endif /* WIFSTOPPED */
8747 #ifdef WIFSIGNALED
8748 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
8749 #endif /* WIFSIGNALED */
8750 #ifdef WIFEXITED
8751 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
8752 #endif /* WIFEXITED */
8753 #ifdef WEXITSTATUS
8754 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
8755 #endif /* WEXITSTATUS */
8756 #ifdef WTERMSIG
8757 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
8758 #endif /* WTERMSIG */
8759 #ifdef WSTOPSIG
8760 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
8761 #endif /* WSTOPSIG */
8762 #endif /* HAVE_SYS_WAIT_H */
8763 #if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
8764 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
8765 #endif
8766 #if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
8767 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
8768 #endif
8769 #ifdef HAVE_TMPFILE
8770 {"tmpfile", posix_tmpfile, METH_NOARGS, posix_tmpfile__doc__},
8771 #endif
8772 #ifdef HAVE_TEMPNAM
8773 {"tempnam", posix_tempnam, METH_VARARGS, posix_tempnam__doc__},
8774 #endif
8775 #ifdef HAVE_TMPNAM
8776 {"tmpnam", posix_tmpnam, METH_NOARGS, posix_tmpnam__doc__},
8777 #endif
8778 #ifdef HAVE_CONFSTR
8779 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
8780 #endif
8781 #ifdef HAVE_SYSCONF
8782 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
8783 #endif
8784 #ifdef HAVE_FPATHCONF
8785 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
8786 #endif
8787 #ifdef HAVE_PATHCONF
8788 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
8789 #endif
8790 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
8791 #ifdef MS_WINDOWS
8792 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
8793 #endif
8794 #ifdef HAVE_GETLOADAVG
8795 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
8796 #endif
8797 #ifdef MS_WINDOWS
8798 {"urandom", win32_urandom, METH_VARARGS, win32_urandom__doc__},
8799 #endif
8800 #ifdef __VMS
8801 {"urandom", vms_urandom, METH_VARARGS, vms_urandom__doc__},
8802 #endif
8803 #ifdef HAVE_SETRESUID
8804 {"setresuid", posix_setresuid, METH_VARARGS, posix_setresuid__doc__},
8805 #endif
8806 #ifdef HAVE_SETRESGID
8807 {"setresgid", posix_setresgid, METH_VARARGS, posix_setresgid__doc__},
8808 #endif
8809 #ifdef HAVE_GETRESUID
8810 {"getresuid", posix_getresuid, METH_NOARGS, posix_getresuid__doc__},
8811 #endif
8812 #ifdef HAVE_GETRESGID
8813 {"getresgid", posix_getresgid, METH_NOARGS, posix_getresgid__doc__},
8814 #endif
8816 {NULL, NULL} /* Sentinel */
8820 static int
8821 ins(PyObject *module, char *symbol, long value)
8823 return PyModule_AddIntConstant(module, symbol, value);
8826 #if defined(PYOS_OS2)
8827 /* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
8828 static int insertvalues(PyObject *module)
8830 APIRET rc;
8831 ULONG values[QSV_MAX+1];
8832 PyObject *v;
8833 char *ver, tmp[50];
8835 Py_BEGIN_ALLOW_THREADS
8836 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
8837 Py_END_ALLOW_THREADS
8839 if (rc != NO_ERROR) {
8840 os2_error(rc);
8841 return -1;
8844 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
8845 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
8846 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
8847 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
8848 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
8849 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
8850 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
8852 switch (values[QSV_VERSION_MINOR]) {
8853 case 0: ver = "2.00"; break;
8854 case 10: ver = "2.10"; break;
8855 case 11: ver = "2.11"; break;
8856 case 30: ver = "3.00"; break;
8857 case 40: ver = "4.00"; break;
8858 case 50: ver = "5.00"; break;
8859 default:
8860 PyOS_snprintf(tmp, sizeof(tmp),
8861 "%d-%d", values[QSV_VERSION_MAJOR],
8862 values[QSV_VERSION_MINOR]);
8863 ver = &tmp[0];
8866 /* Add Indicator of the Version of the Operating System */
8867 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
8868 return -1;
8870 /* Add Indicator of Which Drive was Used to Boot the System */
8871 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
8872 tmp[1] = ':';
8873 tmp[2] = '\0';
8875 return PyModule_AddStringConstant(module, "bootdrive", tmp);
8877 #endif
8879 static int
8880 all_ins(PyObject *d)
8882 #ifdef F_OK
8883 if (ins(d, "F_OK", (long)F_OK)) return -1;
8884 #endif
8885 #ifdef R_OK
8886 if (ins(d, "R_OK", (long)R_OK)) return -1;
8887 #endif
8888 #ifdef W_OK
8889 if (ins(d, "W_OK", (long)W_OK)) return -1;
8890 #endif
8891 #ifdef X_OK
8892 if (ins(d, "X_OK", (long)X_OK)) return -1;
8893 #endif
8894 #ifdef NGROUPS_MAX
8895 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
8896 #endif
8897 #ifdef TMP_MAX
8898 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
8899 #endif
8900 #ifdef WCONTINUED
8901 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
8902 #endif
8903 #ifdef WNOHANG
8904 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
8905 #endif
8906 #ifdef WUNTRACED
8907 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
8908 #endif
8909 #ifdef O_RDONLY
8910 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
8911 #endif
8912 #ifdef O_WRONLY
8913 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
8914 #endif
8915 #ifdef O_RDWR
8916 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
8917 #endif
8918 #ifdef O_NDELAY
8919 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
8920 #endif
8921 #ifdef O_NONBLOCK
8922 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
8923 #endif
8924 #ifdef O_APPEND
8925 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
8926 #endif
8927 #ifdef O_DSYNC
8928 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
8929 #endif
8930 #ifdef O_RSYNC
8931 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
8932 #endif
8933 #ifdef O_SYNC
8934 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
8935 #endif
8936 #ifdef O_NOCTTY
8937 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
8938 #endif
8939 #ifdef O_CREAT
8940 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
8941 #endif
8942 #ifdef O_EXCL
8943 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
8944 #endif
8945 #ifdef O_TRUNC
8946 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
8947 #endif
8948 #ifdef O_BINARY
8949 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
8950 #endif
8951 #ifdef O_TEXT
8952 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
8953 #endif
8954 #ifdef O_LARGEFILE
8955 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
8956 #endif
8957 #ifdef O_SHLOCK
8958 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
8959 #endif
8960 #ifdef O_EXLOCK
8961 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
8962 #endif
8964 /* MS Windows */
8965 #ifdef O_NOINHERIT
8966 /* Don't inherit in child processes. */
8967 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
8968 #endif
8969 #ifdef _O_SHORT_LIVED
8970 /* Optimize for short life (keep in memory). */
8971 /* MS forgot to define this one with a non-underscore form too. */
8972 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
8973 #endif
8974 #ifdef O_TEMPORARY
8975 /* Automatically delete when last handle is closed. */
8976 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
8977 #endif
8978 #ifdef O_RANDOM
8979 /* Optimize for random access. */
8980 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
8981 #endif
8982 #ifdef O_SEQUENTIAL
8983 /* Optimize for sequential access. */
8984 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
8985 #endif
8987 /* GNU extensions. */
8988 #ifdef O_ASYNC
8989 /* Send a SIGIO signal whenever input or output
8990 becomes available on file descriptor */
8991 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
8992 #endif
8993 #ifdef O_DIRECT
8994 /* Direct disk access. */
8995 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
8996 #endif
8997 #ifdef O_DIRECTORY
8998 /* Must be a directory. */
8999 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
9000 #endif
9001 #ifdef O_NOFOLLOW
9002 /* Do not follow links. */
9003 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
9004 #endif
9005 #ifdef O_NOATIME
9006 /* Do not update the access time. */
9007 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
9008 #endif
9010 /* These come from sysexits.h */
9011 #ifdef EX_OK
9012 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
9013 #endif /* EX_OK */
9014 #ifdef EX_USAGE
9015 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
9016 #endif /* EX_USAGE */
9017 #ifdef EX_DATAERR
9018 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
9019 #endif /* EX_DATAERR */
9020 #ifdef EX_NOINPUT
9021 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
9022 #endif /* EX_NOINPUT */
9023 #ifdef EX_NOUSER
9024 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
9025 #endif /* EX_NOUSER */
9026 #ifdef EX_NOHOST
9027 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
9028 #endif /* EX_NOHOST */
9029 #ifdef EX_UNAVAILABLE
9030 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
9031 #endif /* EX_UNAVAILABLE */
9032 #ifdef EX_SOFTWARE
9033 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
9034 #endif /* EX_SOFTWARE */
9035 #ifdef EX_OSERR
9036 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
9037 #endif /* EX_OSERR */
9038 #ifdef EX_OSFILE
9039 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
9040 #endif /* EX_OSFILE */
9041 #ifdef EX_CANTCREAT
9042 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
9043 #endif /* EX_CANTCREAT */
9044 #ifdef EX_IOERR
9045 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
9046 #endif /* EX_IOERR */
9047 #ifdef EX_TEMPFAIL
9048 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
9049 #endif /* EX_TEMPFAIL */
9050 #ifdef EX_PROTOCOL
9051 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
9052 #endif /* EX_PROTOCOL */
9053 #ifdef EX_NOPERM
9054 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
9055 #endif /* EX_NOPERM */
9056 #ifdef EX_CONFIG
9057 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
9058 #endif /* EX_CONFIG */
9059 #ifdef EX_NOTFOUND
9060 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
9061 #endif /* EX_NOTFOUND */
9063 #ifdef HAVE_SPAWNV
9064 #if defined(PYOS_OS2) && defined(PYCC_GCC)
9065 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
9066 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
9067 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
9068 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
9069 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
9070 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
9071 if (ins(d, "P_PM", (long)P_PM)) return -1;
9072 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
9073 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
9074 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
9075 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
9076 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
9077 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
9078 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
9079 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
9080 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
9081 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
9082 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
9083 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
9084 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
9085 #else
9086 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
9087 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
9088 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
9089 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
9090 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
9091 #endif
9092 #endif
9094 #if defined(PYOS_OS2)
9095 if (insertvalues(d)) return -1;
9096 #endif
9097 return 0;
9101 #if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
9102 #define INITFUNC initnt
9103 #define MODNAME "nt"
9105 #elif defined(PYOS_OS2)
9106 #define INITFUNC initos2
9107 #define MODNAME "os2"
9109 #else
9110 #define INITFUNC initposix
9111 #define MODNAME "posix"
9112 #endif
9114 PyMODINIT_FUNC
9115 INITFUNC(void)
9117 PyObject *m, *v;
9119 m = Py_InitModule3(MODNAME,
9120 posix_methods,
9121 posix__doc__);
9122 if (m == NULL)
9123 return;
9125 /* Initialize environ dictionary */
9126 v = convertenviron();
9127 Py_XINCREF(v);
9128 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
9129 return;
9130 Py_DECREF(v);
9132 if (all_ins(m))
9133 return;
9135 if (setup_confname_tables(m))
9136 return;
9138 Py_INCREF(PyExc_OSError);
9139 PyModule_AddObject(m, "error", PyExc_OSError);
9141 #ifdef HAVE_PUTENV
9142 if (posix_putenv_garbage == NULL)
9143 posix_putenv_garbage = PyDict_New();
9144 #endif
9146 if (!initialized) {
9147 stat_result_desc.name = MODNAME ".stat_result";
9148 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
9149 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
9150 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
9151 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
9152 structseq_new = StatResultType.tp_new;
9153 StatResultType.tp_new = statresult_new;
9155 statvfs_result_desc.name = MODNAME ".statvfs_result";
9156 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
9157 #ifdef NEED_TICKS_PER_SECOND
9158 # if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
9159 ticks_per_second = sysconf(_SC_CLK_TCK);
9160 # elif defined(HZ)
9161 ticks_per_second = HZ;
9162 # else
9163 ticks_per_second = 60; /* magic fallback value; may be bogus */
9164 # endif
9165 #endif
9167 Py_INCREF((PyObject*) &StatResultType);
9168 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
9169 Py_INCREF((PyObject*) &StatVFSResultType);
9170 PyModule_AddObject(m, "statvfs_result",
9171 (PyObject*) &StatVFSResultType);
9172 initialized = 1;
9174 #ifdef __APPLE__
9176 * Step 2 of weak-linking support on Mac OS X.
9178 * The code below removes functions that are not available on the
9179 * currently active platform.
9181 * This block allow one to use a python binary that was build on
9182 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
9183 * OSX 10.4.
9185 #ifdef HAVE_FSTATVFS
9186 if (fstatvfs == NULL) {
9187 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
9188 return;
9191 #endif /* HAVE_FSTATVFS */
9193 #ifdef HAVE_STATVFS
9194 if (statvfs == NULL) {
9195 if (PyObject_DelAttrString(m, "statvfs") == -1) {
9196 return;
9199 #endif /* HAVE_STATVFS */
9201 # ifdef HAVE_LCHOWN
9202 if (lchown == NULL) {
9203 if (PyObject_DelAttrString(m, "lchown") == -1) {
9204 return;
9207 #endif /* HAVE_LCHOWN */
9210 #endif /* __APPLE__ */
9214 #ifdef __cplusplus
9216 #endif