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