blob: ff79a0e8faa70a99252f557b5bdba4c5c7f4228e [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* POSIX module implementation */
3
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004/* 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
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007 assumes that for Windows NT, the macro 'MS_WINDOWS' is defined independent
Guido van Rossumad0ee831995-03-01 10:34:45 +00008 of the compiler used. Different compilers define their own feature
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00009 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. */
Guido van Rossumad0ee831995-03-01 10:34:45 +000013
Guido van Rossuma4916fa1996-05-23 22:58:55 +000014/* See also ../Dos/dosmodule.c */
Guido van Rossumad0ee831995-03-01 10:34:45 +000015
Ronald Oussorend06b6f22006-04-23 11:59:25 +000016#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
Thomas Wouters68bc4f92006-03-01 01:05:10 +000028#define PY_SSIZE_T_CLEAN
29
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000030#include "Python.h"
31#include "structseq.h"
32
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000033#if defined(__VMS)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000034# include <unixio.h>
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000035#endif /* defined(__VMS) */
36
Anthony Baxterac6bd462006-04-13 02:06:09 +000037#ifdef __cplusplus
38extern "C" {
39#endif
40
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000041PyDoc_STRVAR(posix__doc__,
Guido van Rossumec4f4ac1997-06-02 22:20:51 +000042"This module provides access to operating system functionality that is\n\
43standardized by the C Standard and the POSIX standard (a thinly\n\
44disguised Unix interface). Refer to the library manual and\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000045corresponding Unix manual entries for more information on calls.");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000046
Martin v. Löwis0073f2e2002-11-21 23:52:35 +000047#ifndef Py_USING_UNICODE
48/* This is used in signatures of functions. */
49#define Py_UNICODE void
50#endif
51
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000052#if defined(PYOS_OS2)
53#define INCL_DOS
54#define INCL_DOSERRORS
55#define INCL_DOSPROCESS
56#define INCL_NOPMAPI
57#include <os2.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000058#if defined(PYCC_GCC)
59#include <ctype.h>
60#include <io.h>
61#include <stdio.h>
62#include <process.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000063#endif
Andrew MacIntyreda4d6cb2004-03-29 11:53:38 +000064#include "osdefs.h"
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000065#endif
66
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000067#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000068#include <sys/types.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000069#endif /* HAVE_SYS_TYPES_H */
70
71#ifdef HAVE_SYS_STAT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000072#include <sys/stat.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000073#endif /* HAVE_SYS_STAT_H */
Guido van Rossuma6535fd2001-10-18 19:44:10 +000074
Guido van Rossum36bc6801995-06-14 22:54:23 +000075#ifdef HAVE_SYS_WAIT_H
76#include <sys/wait.h> /* For WNOHANG */
77#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +000078
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000079#ifdef HAVE_SIGNAL_H
Guido van Rossuma376cc51996-12-05 23:43:35 +000080#include <signal.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000081#endif
Guido van Rossuma376cc51996-12-05 23:43:35 +000082
Guido van Rossumb6775db1994-08-01 11:34:53 +000083#ifdef HAVE_FCNTL_H
84#include <fcntl.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000085#endif /* HAVE_FCNTL_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +000086
Guido van Rossuma6535fd2001-10-18 19:44:10 +000087#ifdef HAVE_GRP_H
88#include <grp.h>
89#endif
90
Barry Warsaw5676bd12003-01-07 20:57:09 +000091#ifdef HAVE_SYSEXITS_H
92#include <sysexits.h>
93#endif /* HAVE_SYSEXITS_H */
94
Anthony Baxter8a560de2004-10-13 15:30:56 +000095#ifdef HAVE_SYS_LOADAVG_H
96#include <sys/loadavg.h>
97#endif
98
Guido van Rossuma4916fa1996-05-23 22:58:55 +000099/* Various compilers have only certain posix functions */
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +0000100/* XXX Gosh I wish these were all moved into pyconfig.h */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000101#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000102#include <process.h>
103#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000104#if defined(__WATCOMC__) && !defined(__QNX__) /* Watcom compiler */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000105#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
Guido van Rossumad0ee831995-03-01 10:34:45 +0000111#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000112#include <process.h>
113#else
114#ifdef __BORLANDC__ /* Borland compiler */
115#define HAVE_EXECV 1
116#define HAVE_GETCWD 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000117#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 */
Guido van Rossum8d665e61996-06-26 18:22:49 +0000124#define HAVE_GETCWD 1
Guido van Rossuma1065681999-01-25 23:20:23 +0000125#define HAVE_SPAWNV 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000126#define HAVE_EXECV 1
127#define HAVE_PIPE 1
128#define HAVE_POPEN 1
129#define HAVE_SYSTEM 1
Tim Petersab034fa2002-02-01 11:27:43 +0000130#define HAVE_CWAIT 1
Tim Peters11b23062003-04-23 02:39:17 +0000131#define HAVE_FSYNC 1
132#define fsync _commit
Andrew MacIntyre6c73af22002-03-03 03:07:07 +0000133#else
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000134#if defined(PYOS_OS2) && defined(PYCC_GCC) || defined(__VMS)
135/* Everything needed is defined in PC/os2emx/pyconfig.h or vms/pyconfig.h */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000136#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
Guido van Rossum2242f2f2001-04-11 20:58:20 +0000140#if defined(__USLC__) && defined(__SCO_VERSION__) /* SCO UDK Compiler */
141#define HAVE_FORK1 1
142#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000143#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
Martin v. Löwis212ede62003-09-20 11:20:30 +0000152#ifndef __rtems__
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000153#define HAVE_POPEN 1
Martin v. Löwis212ede62003-09-20 11:20:30 +0000154#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000155#define HAVE_SYSTEM 1
156#define HAVE_WAIT 1
Guido van Rossumd371ff11999-01-25 16:12:23 +0000157#define HAVE_TTYNAME 1
Martin v. Löwis7a924e62003-03-05 14:15:21 +0000158#endif /* PYOS_OS2 && PYCC_GCC && __VMS */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000159#endif /* _MSC_VER */
160#endif /* __BORLANDC__ */
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000161#endif /* ! __WATCOMC__ || __QNX__ */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000162#endif /* ! __IBMC__ */
Guido van Rossumad0ee831995-03-01 10:34:45 +0000163
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000164#ifndef _MSC_VER
Guido van Rossum36bc6801995-06-14 22:54:23 +0000165
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000166#if defined(__sgi)&&_COMPILER_VERSION>=700
167/* declare ctermid_r if compiling with MIPSPro 7.x in ANSI C mode
168 (default) */
169extern char *ctermid_r(char *);
170#endif
171
Thomas Wouters1e0c2f42000-07-24 16:06:23 +0000172#ifndef HAVE_UNISTD_H
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000173#if defined(PYCC_VACPP)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000174extern int mkdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000175#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000176#if ( defined(__WATCOMC__) || defined(_MSC_VER) ) && !defined(__QNX__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000177extern int mkdir(const char *);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000178#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000179extern int mkdir(const char *, mode_t);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000180#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000181#endif
182#if defined(__IBMC__) || defined(__IBMCPP__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000183extern int chdir(char *);
184extern int rmdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000185#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000186extern int chdir(const char *);
187extern int rmdir(const char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000188#endif
Tim Peters58e0a8c2001-05-14 22:32:33 +0000189#ifdef __BORLANDC__
190extern int chmod(const char *, int);
191#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000192extern int chmod(const char *, mode_t);
Tim Peters58e0a8c2001-05-14 22:32:33 +0000193#endif
Christian Heimes36281872007-11-30 21:11:28 +0000194/*#ifdef HAVE_FCHMOD
195extern int fchmod(int, mode_t);
196#endif*/
197/*#ifdef HAVE_LCHMOD
198extern int lchmod(const char *, mode_t);
199#endif*/
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000200extern int chown(const char *, uid_t, gid_t);
201extern char *getcwd(char *, int);
202extern char *strerror(int);
203extern int link(const char *, const char *);
204extern int rename(const char *, const char *);
205extern int stat(const char *, struct stat *);
206extern int unlink(const char *);
207extern int pclose(FILE *);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000208#ifdef HAVE_SYMLINK
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000209extern int symlink(const char *, const char *);
Guido van Rossuma38a5031995-02-17 15:11:36 +0000210#endif /* HAVE_SYMLINK */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000211#ifdef HAVE_LSTAT
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000212extern int lstat(const char *, struct stat *);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000213#endif /* HAVE_LSTAT */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000214#endif /* !HAVE_UNISTD_H */
Guido van Rossum36bc6801995-06-14 22:54:23 +0000215
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000216#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000217
Guido van Rossumb6775db1994-08-01 11:34:53 +0000218#ifdef HAVE_UTIME_H
219#include <utime.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000220#endif /* HAVE_UTIME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000221
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000222#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
Guido van Rossumb6775db1994-08-01 11:34:53 +0000227#ifdef HAVE_SYS_TIMES_H
228#include <sys/times.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000229#endif /* HAVE_SYS_TIMES_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000230
231#ifdef HAVE_SYS_PARAM_H
232#include <sys/param.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000233#endif /* HAVE_SYS_PARAM_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000234
235#ifdef HAVE_SYS_UTSNAME_H
236#include <sys/utsname.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000237#endif /* HAVE_SYS_UTSNAME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000238
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000239#ifdef HAVE_DIRENT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000240#include <dirent.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000241#define NAMLEN(dirent) strlen((dirent)->d_name)
242#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000243#if defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000244#include <direct.h>
245#define NAMLEN(dirent) strlen((dirent)->d_name)
246#else
Guido van Rossumb6775db1994-08-01 11:34:53 +0000247#define dirent direct
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000248#define NAMLEN(dirent) (dirent)->d_namlen
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000249#endif
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000250#ifdef HAVE_SYS_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000251#include <sys/ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000252#endif
253#ifdef HAVE_SYS_DIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000254#include <sys/dir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000255#endif
256#ifdef HAVE_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000257#include <ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000258#endif
259#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000260
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000261#ifdef _MSC_VER
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +0000262#ifdef HAVE_DIRECT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000263#include <direct.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +0000264#endif
265#ifdef HAVE_IO_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000266#include <io.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +0000267#endif
268#ifdef HAVE_PROCESS_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000269#include <process.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +0000270#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000271#include "osdefs.h"
Guido van Rossumb6775db1994-08-01 11:34:53 +0000272#include <windows.h>
Tim Petersee66d0c2002-07-15 16:10:55 +0000273#include <shellapi.h> /* for ShellExecute() */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000274#define popen _popen
Guido van Rossum794d8131994-08-23 13:48:48 +0000275#define pclose _pclose
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000276#endif /* _MSC_VER */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000277
Guido van Rossumd48f2521997-12-05 22:19:34 +0000278#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000279#include <io.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000280#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000281
Tim Petersbc2e10e2002-03-03 23:17:02 +0000282#ifndef MAXPATHLEN
Thomas Wouters1ddba602006-04-25 15:29:46 +0000283#if defined(PATH_MAX) && PATH_MAX > 1024
284#define MAXPATHLEN PATH_MAX
285#else
Tim Petersbc2e10e2002-03-03 23:17:02 +0000286#define MAXPATHLEN 1024
Thomas Wouters1ddba602006-04-25 15:29:46 +0000287#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000288#endif /* MAXPATHLEN */
289
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000290#ifdef UNION_WAIT
291/* Emulate some macros on systems that have a union instead of macros */
292
293#ifndef WIFEXITED
294#define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
295#endif
296
297#ifndef WEXITSTATUS
298#define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
299#endif
300
301#ifndef WTERMSIG
302#define WTERMSIG(u_wait) ((u_wait).w_termsig)
303#endif
304
Neal Norwitzd5a37542006-03-20 06:48:34 +0000305#define WAIT_TYPE union wait
306#define WAIT_STATUS_INT(s) (s.w_status)
307
308#else /* !UNION_WAIT */
309#define WAIT_TYPE int
310#define WAIT_STATUS_INT(s) (s)
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000311#endif /* UNION_WAIT */
312
Greg Wardb48bc172000-03-01 21:51:56 +0000313/* Don't use the "_r" form if we don't need it (also, won't have a
314 prototype for it, at least on Solaris -- maybe others as well?). */
315#if defined(HAVE_CTERMID_R) && defined(WITH_THREAD)
316#define USE_CTERMID_R
317#endif
318
319#if defined(HAVE_TMPNAM_R) && defined(WITH_THREAD)
320#define USE_TMPNAM_R
321#endif
322
Fred Drake699f3522000-06-29 21:12:41 +0000323/* choose the appropriate stat and fstat functions and return structs */
Guido van Rossum64529cd2000-06-30 22:45:12 +0000324#undef STAT
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000325#if defined(MS_WIN64) || defined(MS_WINDOWS)
Martin v. Löwis14694662006-02-03 12:54:16 +0000326# define STAT win32_stat
327# define FSTAT win32_fstat
328# define STRUCT_STAT struct win32_stat
Fred Drake699f3522000-06-29 21:12:41 +0000329#else
330# define STAT stat
331# define FSTAT fstat
332# define STRUCT_STAT struct stat
333#endif
334
Tim Peters11b23062003-04-23 02:39:17 +0000335#if defined(MAJOR_IN_MKDEV)
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000336#include <sys/mkdev.h>
337#else
338#if defined(MAJOR_IN_SYSMACROS)
339#include <sys/sysmacros.h>
340#endif
Neal Norwitz3d949422002-04-20 13:46:43 +0000341#if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
342#include <sys/mkdev.h>
343#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000344#endif
Fred Drake699f3522000-06-29 21:12:41 +0000345
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000346/* Return a dictionary corresponding to the POSIX environment table */
Jack Jansenea0c3822002-08-01 21:57:49 +0000347#ifdef WITH_NEXT_FRAMEWORK
348/* On Darwin/MacOSX a shared library or framework has no access to
349** environ directly, we must obtain it with _NSGetEnviron().
350*/
351#include <crt_externs.h>
352static char **environ;
353#elif !defined(_MSC_VER) && ( !defined(__WATCOMC__) || defined(__QNX__) )
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000354extern char **environ;
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000355#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000356
Barry Warsaw53699e91996-12-10 23:23:01 +0000357static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000358convertenviron(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000359{
Barry Warsaw53699e91996-12-10 23:23:01 +0000360 PyObject *d;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000361 char **e;
Barry Warsaw53699e91996-12-10 23:23:01 +0000362 d = PyDict_New();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000363 if (d == NULL)
364 return NULL;
Jack Jansenea0c3822002-08-01 21:57:49 +0000365#ifdef WITH_NEXT_FRAMEWORK
366 if (environ == NULL)
367 environ = *_NSGetEnviron();
368#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000369 if (environ == NULL)
370 return d;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000371 /* This part ignores errors */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000372 for (e = environ; *e != NULL; e++) {
Guido van Rossum6a619f41999-08-03 19:41:10 +0000373 PyObject *k;
Barry Warsaw53699e91996-12-10 23:23:01 +0000374 PyObject *v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000375 char *p = strchr(*e, '=');
376 if (p == NULL)
377 continue;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000378 k = PyString_FromStringAndSize(*e, (int)(p-*e));
379 if (k == NULL) {
380 PyErr_Clear();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000381 continue;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000382 }
383 v = PyString_FromString(p+1);
384 if (v == NULL) {
385 PyErr_Clear();
386 Py_DECREF(k);
387 continue;
388 }
389 if (PyDict_GetItem(d, k) == NULL) {
390 if (PyDict_SetItem(d, k, v) != 0)
391 PyErr_Clear();
392 }
393 Py_DECREF(k);
Barry Warsaw53699e91996-12-10 23:23:01 +0000394 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000395 }
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +0000396#if defined(PYOS_OS2)
Guido van Rossumd48f2521997-12-05 22:19:34 +0000397 {
398 APIRET rc;
399 char buffer[1024]; /* OS/2 Provides a Documented Max of 1024 Chars */
400
401 rc = DosQueryExtLIBPATH(buffer, BEGIN_LIBPATH);
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000402 if (rc == NO_ERROR) { /* (not a type, envname is NOT 'BEGIN_LIBPATH') */
Guido van Rossumd48f2521997-12-05 22:19:34 +0000403 PyObject *v = PyString_FromString(buffer);
404 PyDict_SetItemString(d, "BEGINLIBPATH", v);
405 Py_DECREF(v);
406 }
407 rc = DosQueryExtLIBPATH(buffer, END_LIBPATH);
408 if (rc == NO_ERROR) { /* (not a typo, envname is NOT 'END_LIBPATH') */
409 PyObject *v = PyString_FromString(buffer);
410 PyDict_SetItemString(d, "ENDLIBPATH", v);
411 Py_DECREF(v);
412 }
413 }
414#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000415 return d;
416}
417
418
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000419/* Set a POSIX-specific error from errno, and return NULL */
420
Barry Warsawd58d7641998-07-23 16:14:40 +0000421static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000422posix_error(void)
Guido van Rossumad0ee831995-03-01 10:34:45 +0000423{
Barry Warsawca74da41999-02-09 19:31:45 +0000424 return PyErr_SetFromErrno(PyExc_OSError);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000425}
Barry Warsawd58d7641998-07-23 16:14:40 +0000426static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000427posix_error_with_filename(char* name)
Barry Warsawd58d7641998-07-23 16:14:40 +0000428{
Barry Warsawca74da41999-02-09 19:31:45 +0000429 return PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
Barry Warsawd58d7641998-07-23 16:14:40 +0000430}
431
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000432#ifdef Py_WIN_WIDE_FILENAMES
433static PyObject *
434posix_error_with_unicode_filename(Py_UNICODE* name)
435{
436 return PyErr_SetFromErrnoWithUnicodeFilename(PyExc_OSError, name);
437}
438#endif /* Py_WIN_WIDE_FILENAMES */
439
440
Mark Hammondef8b6542001-05-13 08:04:26 +0000441static PyObject *
442posix_error_with_allocated_filename(char* name)
443{
444 PyObject *rc = PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
445 PyMem_Free(name);
446 return rc;
447}
448
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000449#ifdef MS_WINDOWS
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000450static PyObject *
451win32_error(char* function, char* filename)
452{
Mark Hammond33a6da92000-08-15 00:46:38 +0000453 /* XXX We should pass the function name along in the future.
454 (_winreg.c also wants to pass the function name.)
Tim Peters5aa91602002-01-30 05:46:57 +0000455 This would however require an additional param to the
Mark Hammond33a6da92000-08-15 00:46:38 +0000456 Windows error object, which is non-trivial.
457 */
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000458 errno = GetLastError();
459 if (filename)
Mark Hammond33a6da92000-08-15 00:46:38 +0000460 return PyErr_SetFromWindowsErrWithFilename(errno, filename);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000461 else
Mark Hammond33a6da92000-08-15 00:46:38 +0000462 return PyErr_SetFromWindowsErr(errno);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000463}
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000464
465#ifdef Py_WIN_WIDE_FILENAMES
466static PyObject *
467win32_error_unicode(char* function, Py_UNICODE* filename)
468{
469 /* XXX - see win32_error for comments on 'function' */
470 errno = GetLastError();
471 if (filename)
472 return PyErr_SetFromWindowsErrWithUnicodeFilename(errno, filename);
473 else
474 return PyErr_SetFromWindowsErr(errno);
475}
476
477static PyObject *_PyUnicode_FromFileSystemEncodedObject(register PyObject *obj)
478{
Martin v. Löwis8e0d4942006-05-04 10:08:42 +0000479}
480
481/* Function suitable for O& conversion */
482static int
483convert_to_unicode(PyObject *arg, void* _param)
484{
485 PyObject **param = (PyObject**)_param;
486 if (PyUnicode_CheckExact(arg)) {
487 Py_INCREF(arg);
488 *param = arg;
489 }
490 else if (PyUnicode_Check(arg)) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000491 /* For a Unicode subtype that's not a Unicode object,
492 return a true Unicode object with the same data. */
Martin v. Löwis8e0d4942006-05-04 10:08:42 +0000493 *param = PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(arg),
494 PyUnicode_GET_SIZE(arg));
495 return *param != NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000496 }
Martin v. Löwis8e0d4942006-05-04 10:08:42 +0000497 else
498 *param = PyUnicode_FromEncodedObject(arg,
499 Py_FileSystemDefaultEncoding,
500 "strict");
501 return (*param) != NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000502}
503
504#endif /* Py_WIN_WIDE_FILENAMES */
505
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000506#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000507
Guido van Rossumd48f2521997-12-05 22:19:34 +0000508#if defined(PYOS_OS2)
509/**********************************************************************
510 * Helper Function to Trim and Format OS/2 Messages
511 **********************************************************************/
512 static void
513os2_formatmsg(char *msgbuf, int msglen, char *reason)
514{
515 msgbuf[msglen] = '\0'; /* OS/2 Doesn't Guarantee a Terminator */
516
517 if (strlen(msgbuf) > 0) { /* If Non-Empty Msg, Trim CRLF */
518 char *lastc = &msgbuf[ strlen(msgbuf)-1 ];
519
Neal Norwitz30b5c5d2005-12-19 06:05:18 +0000520 while (lastc > msgbuf && isspace(Py_CHARMASK(*lastc)))
Guido van Rossumd48f2521997-12-05 22:19:34 +0000521 *lastc-- = '\0'; /* Trim Trailing Whitespace (CRLF) */
522 }
523
524 /* Add Optional Reason Text */
525 if (reason) {
526 strcat(msgbuf, " : ");
527 strcat(msgbuf, reason);
528 }
529}
530
531/**********************************************************************
532 * Decode an OS/2 Operating System Error Code
533 *
534 * A convenience function to lookup an OS/2 error code and return a
535 * text message we can use to raise a Python exception.
536 *
537 * Notes:
538 * The messages for errors returned from the OS/2 kernel reside in
539 * the file OSO001.MSG in the \OS2 directory hierarchy.
540 *
541 **********************************************************************/
542 static char *
543os2_strerror(char *msgbuf, int msgbuflen, int errorcode, char *reason)
544{
545 APIRET rc;
546 ULONG msglen;
547
548 /* Retrieve Kernel-Related Error Message from OSO001.MSG File */
549 Py_BEGIN_ALLOW_THREADS
550 rc = DosGetMessage(NULL, 0, msgbuf, msgbuflen,
551 errorcode, "oso001.msg", &msglen);
552 Py_END_ALLOW_THREADS
553
554 if (rc == NO_ERROR)
555 os2_formatmsg(msgbuf, msglen, reason);
556 else
Tim Peters1ceb5fb2001-11-28 20:32:57 +0000557 PyOS_snprintf(msgbuf, msgbuflen,
Tim Peters885d4572001-11-28 20:27:42 +0000558 "unknown OS error #%d", errorcode);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000559
560 return msgbuf;
561}
562
563/* Set an OS/2-specific error and return NULL. OS/2 kernel
564 errors are not in a global variable e.g. 'errno' nor are
565 they congruent with posix error numbers. */
566
567static PyObject * os2_error(int code)
568{
569 char text[1024];
570 PyObject *v;
571
572 os2_strerror(text, sizeof(text), code, "");
573
574 v = Py_BuildValue("(is)", code, text);
575 if (v != NULL) {
Barry Warsawca74da41999-02-09 19:31:45 +0000576 PyErr_SetObject(PyExc_OSError, v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000577 Py_DECREF(v);
578 }
579 return NULL; /* Signal to Python that an Exception is Pending */
580}
581
582#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000583
584/* POSIX generic methods */
585
Barry Warsaw53699e91996-12-10 23:23:01 +0000586static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +0000587posix_fildes(PyObject *fdobj, int (*func)(int))
588{
589 int fd;
590 int res;
591 fd = PyObject_AsFileDescriptor(fdobj);
592 if (fd < 0)
593 return NULL;
594 Py_BEGIN_ALLOW_THREADS
595 res = (*func)(fd);
596 Py_END_ALLOW_THREADS
597 if (res < 0)
598 return posix_error();
599 Py_INCREF(Py_None);
600 return Py_None;
601}
Guido van Rossum21142a01999-01-08 21:05:37 +0000602
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000603#ifdef Py_WIN_WIDE_FILENAMES
Tim Peters11b23062003-04-23 02:39:17 +0000604static int
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000605unicode_file_names(void)
606{
607 static int canusewide = -1;
608 if (canusewide == -1) {
Tim Peters11b23062003-04-23 02:39:17 +0000609 /* As per doc for ::GetVersion(), this is the correct test for
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000610 the Windows NT family. */
611 canusewide = (GetVersion() < 0x80000000) ? 1 : 0;
612 }
613 return canusewide;
614}
615#endif
Tim Peters11b23062003-04-23 02:39:17 +0000616
Guido van Rossum21142a01999-01-08 21:05:37 +0000617static PyObject *
Martin v. Löwis8e0d4942006-05-04 10:08:42 +0000618posix_1str(PyObject *args, char *format, int (*func)(const char*))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000619{
Mark Hammondef8b6542001-05-13 08:04:26 +0000620 char *path1 = NULL;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000621 int res;
Tim Peters5aa91602002-01-30 05:46:57 +0000622 if (!PyArg_ParseTuple(args, format,
Mark Hammondef8b6542001-05-13 08:04:26 +0000623 Py_FileSystemDefaultEncoding, &path1))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000624 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +0000625 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000626 res = (*func)(path1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000627 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000628 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +0000629 return posix_error_with_allocated_filename(path1);
630 PyMem_Free(path1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000631 Py_INCREF(Py_None);
632 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000633}
634
Barry Warsaw53699e91996-12-10 23:23:01 +0000635static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +0000636posix_2str(PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000637 char *format,
Martin v. Löwis8e0d4942006-05-04 10:08:42 +0000638 int (*func)(const char *, const char *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000639{
Mark Hammondef8b6542001-05-13 08:04:26 +0000640 char *path1 = NULL, *path2 = NULL;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000641 int res;
Mark Hammondef8b6542001-05-13 08:04:26 +0000642 if (!PyArg_ParseTuple(args, format,
Tim Peters5aa91602002-01-30 05:46:57 +0000643 Py_FileSystemDefaultEncoding, &path1,
Mark Hammondef8b6542001-05-13 08:04:26 +0000644 Py_FileSystemDefaultEncoding, &path2))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000645 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +0000646 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000647 res = (*func)(path1, path2);
Barry Warsaw53699e91996-12-10 23:23:01 +0000648 Py_END_ALLOW_THREADS
Mark Hammondef8b6542001-05-13 08:04:26 +0000649 PyMem_Free(path1);
650 PyMem_Free(path2);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000651 if (res != 0)
Barry Warsawd58d7641998-07-23 16:14:40 +0000652 /* XXX how to report both path1 and path2??? */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000653 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +0000654 Py_INCREF(Py_None);
655 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000656}
657
Martin v. Löwis8e0d4942006-05-04 10:08:42 +0000658#ifdef Py_WIN_WIDE_FILENAMES
659static PyObject*
660win32_1str(PyObject* args, char* func,
661 char* format, BOOL (__stdcall *funcA)(LPCSTR),
662 char* wformat, BOOL (__stdcall *funcW)(LPWSTR))
663{
664 PyObject *uni;
665 char *ansi;
666 BOOL result;
667 if (unicode_file_names()) {
668 if (!PyArg_ParseTuple(args, wformat, &uni))
669 PyErr_Clear();
670 else {
671 Py_BEGIN_ALLOW_THREADS
672 result = funcW(PyUnicode_AsUnicode(uni));
673 Py_END_ALLOW_THREADS
674 if (!result)
675 return win32_error_unicode(func, PyUnicode_AsUnicode(uni));
676 Py_INCREF(Py_None);
677 return Py_None;
678 }
679 }
680 if (!PyArg_ParseTuple(args, format, &ansi))
681 return NULL;
682 Py_BEGIN_ALLOW_THREADS
683 result = funcA(ansi);
684 Py_END_ALLOW_THREADS
685 if (!result)
686 return win32_error(func, ansi);
687 Py_INCREF(Py_None);
688 return Py_None;
689
690}
691
692/* This is a reimplementation of the C library's chdir function,
693 but one that produces Win32 errors instead of DOS error codes.
694 chdir is essentially a wrapper around SetCurrentDirectory; however,
695 it also needs to set "magic" environment variables indicating
696 the per-drive current directory, which are of the form =<drive>: */
697BOOL __stdcall
698win32_chdir(LPCSTR path)
699{
700 char new_path[MAX_PATH+1];
701 int result;
702 char env[4] = "=x:";
703
704 if(!SetCurrentDirectoryA(path))
705 return FALSE;
706 result = GetCurrentDirectoryA(MAX_PATH+1, new_path);
707 if (!result)
708 return FALSE;
709 /* In the ANSI API, there should not be any paths longer
710 than MAX_PATH. */
711 assert(result <= MAX_PATH+1);
712 if (strncmp(new_path, "\\\\", 2) == 0 ||
713 strncmp(new_path, "//", 2) == 0)
714 /* UNC path, nothing to do. */
715 return TRUE;
716 env[1] = new_path[0];
717 return SetEnvironmentVariableA(env, new_path);
718}
719
720/* The Unicode version differs from the ANSI version
721 since the current directory might exceed MAX_PATH characters */
722BOOL __stdcall
723win32_wchdir(LPCWSTR path)
724{
725 wchar_t _new_path[MAX_PATH+1], *new_path = _new_path;
726 int result;
727 wchar_t env[4] = L"=x:";
728
729 if(!SetCurrentDirectoryW(path))
730 return FALSE;
731 result = GetCurrentDirectoryW(MAX_PATH+1, new_path);
732 if (!result)
733 return FALSE;
734 if (result > MAX_PATH+1) {
735 new_path = malloc(result);
736 if (!new_path) {
737 SetLastError(ERROR_OUTOFMEMORY);
738 return FALSE;
739 }
740 }
741 if (wcsncmp(new_path, L"\\\\", 2) == 0 ||
742 wcsncmp(new_path, L"//", 2) == 0)
743 /* UNC path, nothing to do. */
744 return TRUE;
745 env[1] = new_path[0];
746 result = SetEnvironmentVariableW(env, new_path);
747 if (new_path != _new_path)
748 free(new_path);
749 return result;
750}
751#endif
752
Martin v. Löwis14694662006-02-03 12:54:16 +0000753#ifdef MS_WINDOWS
754/* The CRT of Windows has a number of flaws wrt. its stat() implementation:
755 - time stamps are restricted to second resolution
756 - file modification times suffer from forth-and-back conversions between
757 UTC and local time
758 Therefore, we implement our own stat, based on the Win32 API directly.
759*/
760#define HAVE_STAT_NSEC 1
761
762struct win32_stat{
763 int st_dev;
764 __int64 st_ino;
765 unsigned short st_mode;
766 int st_nlink;
767 int st_uid;
768 int st_gid;
769 int st_rdev;
770 __int64 st_size;
771 int st_atime;
772 int st_atime_nsec;
773 int st_mtime;
774 int st_mtime_nsec;
775 int st_ctime;
776 int st_ctime_nsec;
777};
778
779static __int64 secs_between_epochs = 11644473600; /* Seconds between 1.1.1601 and 1.1.1970 */
780
781static void
782FILE_TIME_to_time_t_nsec(FILETIME *in_ptr, int *time_out, int* nsec_out)
783{
Martin v. Löwis77c176d2006-05-12 17:22:04 +0000784 /* XXX endianness. Shouldn't matter, as all Windows implementations are little-endian */
785 /* Cannot simply cast and dereference in_ptr,
786 since it might not be aligned properly */
787 __int64 in;
788 memcpy(&in, in_ptr, sizeof(in));
Martin v. Löwis14694662006-02-03 12:54:16 +0000789 *nsec_out = (int)(in % 10000000) * 100; /* FILETIME is in units of 100 nsec. */
790 /* XXX Win32 supports time stamps past 2038; we currently don't */
791 *time_out = Py_SAFE_DOWNCAST((in / 10000000) - secs_between_epochs, __int64, int);
792}
793
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +0000794static void
795time_t_to_FILE_TIME(int time_in, int nsec_in, FILETIME *out_ptr)
796{
797 /* XXX endianness */
798 __int64 out;
799 out = time_in + secs_between_epochs;
Martin v. Löwisf43893a2006-10-09 20:44:25 +0000800 out = out * 10000000 + nsec_in / 100;
Martin v. Löwis77c176d2006-05-12 17:22:04 +0000801 memcpy(out_ptr, &out, sizeof(out));
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +0000802}
803
Martin v. Löwis14694662006-02-03 12:54:16 +0000804/* Below, we *know* that ugo+r is 0444 */
805#if _S_IREAD != 0400
806#error Unsupported C library
807#endif
808static int
809attributes_to_mode(DWORD attr)
810{
811 int m = 0;
812 if (attr & FILE_ATTRIBUTE_DIRECTORY)
813 m |= _S_IFDIR | 0111; /* IFEXEC for user,group,other */
814 else
815 m |= _S_IFREG;
816 if (attr & FILE_ATTRIBUTE_READONLY)
817 m |= 0444;
818 else
819 m |= 0666;
820 return m;
821}
822
823static int
824attribute_data_to_stat(WIN32_FILE_ATTRIBUTE_DATA *info, struct win32_stat *result)
825{
826 memset(result, 0, sizeof(*result));
827 result->st_mode = attributes_to_mode(info->dwFileAttributes);
828 result->st_size = (((__int64)info->nFileSizeHigh)<<32) + info->nFileSizeLow;
829 FILE_TIME_to_time_t_nsec(&info->ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
830 FILE_TIME_to_time_t_nsec(&info->ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
831 FILE_TIME_to_time_t_nsec(&info->ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
832
833 return 0;
834}
835
Martin v. Löwis012bc722006-10-15 09:43:39 +0000836/* Emulate GetFileAttributesEx[AW] on Windows 95 */
837static int checked = 0;
838static BOOL (CALLBACK *gfaxa)(LPCSTR, GET_FILEEX_INFO_LEVELS, LPVOID);
839static BOOL (CALLBACK *gfaxw)(LPCWSTR, GET_FILEEX_INFO_LEVELS, LPVOID);
840static void
841check_gfax()
842{
843 HINSTANCE hKernel32;
844 if (checked)
845 return;
846 checked = 1;
847 hKernel32 = GetModuleHandle("KERNEL32");
848 *(FARPROC*)&gfaxa = GetProcAddress(hKernel32, "GetFileAttributesExA");
849 *(FARPROC*)&gfaxw = GetProcAddress(hKernel32, "GetFileAttributesExW");
850}
851
Martin v. Löwis3bf573f2007-04-04 18:30:36 +0000852static BOOL
853attributes_from_dir(LPCSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
854{
855 HANDLE hFindFile;
856 WIN32_FIND_DATAA FileData;
857 hFindFile = FindFirstFileA(pszFile, &FileData);
858 if (hFindFile == INVALID_HANDLE_VALUE)
859 return FALSE;
860 FindClose(hFindFile);
861 pfad->dwFileAttributes = FileData.dwFileAttributes;
862 pfad->ftCreationTime = FileData.ftCreationTime;
863 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
864 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
865 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
866 pfad->nFileSizeLow = FileData.nFileSizeLow;
867 return TRUE;
868}
869
870static BOOL
871attributes_from_dir_w(LPCWSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
872{
873 HANDLE hFindFile;
874 WIN32_FIND_DATAW FileData;
875 hFindFile = FindFirstFileW(pszFile, &FileData);
876 if (hFindFile == INVALID_HANDLE_VALUE)
877 return FALSE;
878 FindClose(hFindFile);
879 pfad->dwFileAttributes = FileData.dwFileAttributes;
880 pfad->ftCreationTime = FileData.ftCreationTime;
881 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
882 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
883 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
884 pfad->nFileSizeLow = FileData.nFileSizeLow;
885 return TRUE;
886}
887
Martin v. Löwis012bc722006-10-15 09:43:39 +0000888static BOOL WINAPI
889Py_GetFileAttributesExA(LPCSTR pszFile,
890 GET_FILEEX_INFO_LEVELS level,
891 LPVOID pv)
892{
893 BOOL result;
Martin v. Löwis012bc722006-10-15 09:43:39 +0000894 LPWIN32_FILE_ATTRIBUTE_DATA pfad = pv;
895 /* First try to use the system's implementation, if that is
896 available and either succeeds to gives an error other than
897 that it isn't implemented. */
898 check_gfax();
899 if (gfaxa) {
900 result = gfaxa(pszFile, level, pv);
901 if (result || GetLastError() != ERROR_CALL_NOT_IMPLEMENTED)
902 return result;
903 }
904 /* It's either not present, or not implemented.
905 Emulate using FindFirstFile. */
906 if (level != GetFileExInfoStandard) {
907 SetLastError(ERROR_INVALID_PARAMETER);
908 return FALSE;
909 }
910 /* Use GetFileAttributes to validate that the file name
911 does not contain wildcards (which FindFirstFile would
912 accept). */
913 if (GetFileAttributesA(pszFile) == 0xFFFFFFFF)
914 return FALSE;
Martin v. Löwis3bf573f2007-04-04 18:30:36 +0000915 return attributes_from_dir(pszFile, pfad);
Martin v. Löwis012bc722006-10-15 09:43:39 +0000916}
917
918static BOOL WINAPI
919Py_GetFileAttributesExW(LPCWSTR pszFile,
920 GET_FILEEX_INFO_LEVELS level,
921 LPVOID pv)
922{
923 BOOL result;
Martin v. Löwis012bc722006-10-15 09:43:39 +0000924 LPWIN32_FILE_ATTRIBUTE_DATA pfad = pv;
925 /* First try to use the system's implementation, if that is
926 available and either succeeds to gives an error other than
927 that it isn't implemented. */
928 check_gfax();
929 if (gfaxa) {
930 result = gfaxw(pszFile, level, pv);
931 if (result || GetLastError() != ERROR_CALL_NOT_IMPLEMENTED)
932 return result;
933 }
934 /* It's either not present, or not implemented.
935 Emulate using FindFirstFile. */
936 if (level != GetFileExInfoStandard) {
937 SetLastError(ERROR_INVALID_PARAMETER);
938 return FALSE;
939 }
940 /* Use GetFileAttributes to validate that the file name
941 does not contain wildcards (which FindFirstFile would
942 accept). */
943 if (GetFileAttributesW(pszFile) == 0xFFFFFFFF)
944 return FALSE;
Martin v. Löwis3bf573f2007-04-04 18:30:36 +0000945 return attributes_from_dir_w(pszFile, pfad);
Martin v. Löwis012bc722006-10-15 09:43:39 +0000946}
947
Martin v. Löwis14694662006-02-03 12:54:16 +0000948static int
949win32_stat(const char* path, struct win32_stat *result)
950{
951 WIN32_FILE_ATTRIBUTE_DATA info;
952 int code;
953 char *dot;
954 /* XXX not supported on Win95 and NT 3.x */
Martin v. Löwis012bc722006-10-15 09:43:39 +0000955 if (!Py_GetFileAttributesExA(path, GetFileExInfoStandard, &info)) {
Martin v. Löwis3bf573f2007-04-04 18:30:36 +0000956 if (GetLastError() != ERROR_SHARING_VIOLATION) {
957 /* Protocol violation: we explicitly clear errno, instead of
958 setting it to a POSIX error. Callers should use GetLastError. */
959 errno = 0;
960 return -1;
961 } else {
962 /* Could not get attributes on open file. Fall back to
963 reading the directory. */
964 if (!attributes_from_dir(path, &info)) {
965 /* Very strange. This should not fail now */
966 errno = 0;
967 return -1;
968 }
969 }
Martin v. Löwis14694662006-02-03 12:54:16 +0000970 }
971 code = attribute_data_to_stat(&info, result);
972 if (code != 0)
973 return code;
974 /* Set S_IFEXEC if it is an .exe, .bat, ... */
975 dot = strrchr(path, '.');
976 if (dot) {
977 if (stricmp(dot, ".bat") == 0 ||
978 stricmp(dot, ".cmd") == 0 ||
979 stricmp(dot, ".exe") == 0 ||
980 stricmp(dot, ".com") == 0)
981 result->st_mode |= 0111;
982 }
983 return code;
984}
985
986static int
987win32_wstat(const wchar_t* path, struct win32_stat *result)
988{
989 int code;
990 const wchar_t *dot;
991 WIN32_FILE_ATTRIBUTE_DATA info;
992 /* XXX not supported on Win95 and NT 3.x */
Martin v. Löwis012bc722006-10-15 09:43:39 +0000993 if (!Py_GetFileAttributesExW(path, GetFileExInfoStandard, &info)) {
Martin v. Löwis3bf573f2007-04-04 18:30:36 +0000994 if (GetLastError() != ERROR_SHARING_VIOLATION) {
995 /* Protocol violation: we explicitly clear errno, instead of
996 setting it to a POSIX error. Callers should use GetLastError. */
997 errno = 0;
998 return -1;
999 } else {
1000 /* Could not get attributes on open file. Fall back to
1001 reading the directory. */
1002 if (!attributes_from_dir_w(path, &info)) {
1003 /* Very strange. This should not fail now */
1004 errno = 0;
1005 return -1;
1006 }
1007 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001008 }
1009 code = attribute_data_to_stat(&info, result);
1010 if (code < 0)
1011 return code;
1012 /* Set IFEXEC if it is an .exe, .bat, ... */
1013 dot = wcsrchr(path, '.');
1014 if (dot) {
1015 if (_wcsicmp(dot, L".bat") == 0 ||
1016 _wcsicmp(dot, L".cmd") == 0 ||
1017 _wcsicmp(dot, L".exe") == 0 ||
1018 _wcsicmp(dot, L".com") == 0)
1019 result->st_mode |= 0111;
1020 }
1021 return code;
1022}
1023
1024static int
1025win32_fstat(int file_number, struct win32_stat *result)
1026{
1027 BY_HANDLE_FILE_INFORMATION info;
1028 HANDLE h;
1029 int type;
1030
1031 h = (HANDLE)_get_osfhandle(file_number);
1032
1033 /* Protocol violation: we explicitly clear errno, instead of
1034 setting it to a POSIX error. Callers should use GetLastError. */
1035 errno = 0;
1036
1037 if (h == INVALID_HANDLE_VALUE) {
1038 /* This is really a C library error (invalid file handle).
1039 We set the Win32 error to the closes one matching. */
1040 SetLastError(ERROR_INVALID_HANDLE);
1041 return -1;
1042 }
1043 memset(result, 0, sizeof(*result));
1044
1045 type = GetFileType(h);
1046 if (type == FILE_TYPE_UNKNOWN) {
1047 DWORD error = GetLastError();
1048 if (error != 0) {
1049 return -1;
1050 }
1051 /* else: valid but unknown file */
1052 }
1053
1054 if (type != FILE_TYPE_DISK) {
1055 if (type == FILE_TYPE_CHAR)
1056 result->st_mode = _S_IFCHR;
1057 else if (type == FILE_TYPE_PIPE)
1058 result->st_mode = _S_IFIFO;
1059 return 0;
1060 }
1061
1062 if (!GetFileInformationByHandle(h, &info)) {
1063 return -1;
1064 }
1065
1066 /* similar to stat() */
1067 result->st_mode = attributes_to_mode(info.dwFileAttributes);
1068 result->st_size = (((__int64)info.nFileSizeHigh)<<32) + info.nFileSizeLow;
1069 FILE_TIME_to_time_t_nsec(&info.ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1070 FILE_TIME_to_time_t_nsec(&info.ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1071 FILE_TIME_to_time_t_nsec(&info.ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
1072 /* specific to fstat() */
1073 result->st_nlink = info.nNumberOfLinks;
1074 result->st_ino = (((__int64)info.nFileIndexHigh)<<32) + info.nFileIndexLow;
1075 return 0;
1076}
1077
1078#endif /* MS_WINDOWS */
1079
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001080PyDoc_STRVAR(stat_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001081"stat_result: Result from stat or lstat.\n\n\
1082This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001083 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001084or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
1085\n\
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001086Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
1087or st_flags, they are available as attributes only.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001088\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001089See os.stat for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001090
1091static PyStructSequence_Field stat_result_fields[] = {
1092 {"st_mode", "protection bits"},
1093 {"st_ino", "inode"},
1094 {"st_dev", "device"},
1095 {"st_nlink", "number of hard links"},
1096 {"st_uid", "user ID of owner"},
1097 {"st_gid", "group ID of owner"},
1098 {"st_size", "total size, in bytes"},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001099 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
1100 {NULL, "integer time of last access"},
1101 {NULL, "integer time of last modification"},
1102 {NULL, "integer time of last change"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001103 {"st_atime", "time of last access"},
1104 {"st_mtime", "time of last modification"},
1105 {"st_ctime", "time of last change"},
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001106#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001107 {"st_blksize", "blocksize for filesystem I/O"},
1108#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001109#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001110 {"st_blocks", "number of blocks allocated"},
1111#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001112#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001113 {"st_rdev", "device type (if inode device)"},
1114#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001115#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1116 {"st_flags", "user defined flags for file"},
1117#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001118#ifdef HAVE_STRUCT_STAT_ST_GEN
1119 {"st_gen", "generation number"},
1120#endif
1121#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1122 {"st_birthtime", "time of creation"},
1123#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001124 {0}
1125};
1126
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001127#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001128#define ST_BLKSIZE_IDX 13
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001129#else
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001130#define ST_BLKSIZE_IDX 12
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001131#endif
1132
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001133#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001134#define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
1135#else
1136#define ST_BLOCKS_IDX ST_BLKSIZE_IDX
1137#endif
1138
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001139#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001140#define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
1141#else
1142#define ST_RDEV_IDX ST_BLOCKS_IDX
1143#endif
1144
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001145#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1146#define ST_FLAGS_IDX (ST_RDEV_IDX+1)
1147#else
1148#define ST_FLAGS_IDX ST_RDEV_IDX
1149#endif
1150
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001151#ifdef HAVE_STRUCT_STAT_ST_GEN
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001152#define ST_GEN_IDX (ST_FLAGS_IDX+1)
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001153#else
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001154#define ST_GEN_IDX ST_FLAGS_IDX
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001155#endif
1156
1157#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1158#define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
1159#else
1160#define ST_BIRTHTIME_IDX ST_GEN_IDX
1161#endif
1162
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001163static PyStructSequence_Desc stat_result_desc = {
1164 "stat_result", /* name */
1165 stat_result__doc__, /* doc */
1166 stat_result_fields,
1167 10
1168};
1169
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001170PyDoc_STRVAR(statvfs_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001171"statvfs_result: Result from statvfs or fstatvfs.\n\n\
1172This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001173 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
Guido van Rossuma4dc73e2001-10-18 20:53:15 +00001174or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001175\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001176See os.statvfs for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001177
1178static PyStructSequence_Field statvfs_result_fields[] = {
1179 {"f_bsize", },
1180 {"f_frsize", },
1181 {"f_blocks", },
1182 {"f_bfree", },
1183 {"f_bavail", },
1184 {"f_files", },
1185 {"f_ffree", },
1186 {"f_favail", },
1187 {"f_flag", },
1188 {"f_namemax",},
1189 {0}
1190};
1191
1192static PyStructSequence_Desc statvfs_result_desc = {
1193 "statvfs_result", /* name */
1194 statvfs_result__doc__, /* doc */
1195 statvfs_result_fields,
1196 10
1197};
1198
Martin v. Löwis19ab6c92006-04-16 18:55:50 +00001199static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001200static PyTypeObject StatResultType;
1201static PyTypeObject StatVFSResultType;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001202static newfunc structseq_new;
1203
1204static PyObject *
1205statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1206{
1207 PyStructSequence *result;
1208 int i;
1209
1210 result = (PyStructSequence*)structseq_new(type, args, kwds);
1211 if (!result)
1212 return NULL;
1213 /* If we have been initialized from a tuple,
1214 st_?time might be set to None. Initialize it
1215 from the int slots. */
1216 for (i = 7; i <= 9; i++) {
1217 if (result->ob_item[i+3] == Py_None) {
1218 Py_DECREF(Py_None);
1219 Py_INCREF(result->ob_item[i]);
1220 result->ob_item[i+3] = result->ob_item[i];
1221 }
1222 }
1223 return (PyObject*)result;
1224}
1225
1226
1227
1228/* If true, st_?time is float. */
Martin v. Löwisfe33d0b2005-01-16 08:57:39 +00001229static int _stat_float_times = 1;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001230
1231PyDoc_STRVAR(stat_float_times__doc__,
1232"stat_float_times([newval]) -> oldval\n\n\
1233Determine whether os.[lf]stat represents time stamps as float objects.\n\
1234If newval is True, future calls to stat() return floats, if it is False,\n\
1235future calls return ints. \n\
1236If newval is omitted, return the current setting.\n");
1237
1238static PyObject*
1239stat_float_times(PyObject* self, PyObject *args)
1240{
1241 int newval = -1;
1242 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
1243 return NULL;
1244 if (newval == -1)
1245 /* Return old value */
1246 return PyBool_FromLong(_stat_float_times);
1247 _stat_float_times = newval;
1248 Py_INCREF(Py_None);
1249 return Py_None;
1250}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001251
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001252static void
1253fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
1254{
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001255 PyObject *fval,*ival;
1256#if SIZEOF_TIME_T > SIZEOF_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00001257 ival = PyLong_FromLongLong((PY_LONG_LONG)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001258#else
1259 ival = PyInt_FromLong((long)sec);
1260#endif
Neal Norwitze0a81af2006-08-12 01:50:38 +00001261 if (!ival)
1262 return;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001263 if (_stat_float_times) {
1264 fval = PyFloat_FromDouble(sec + 1e-9*nsec);
1265 } else {
1266 fval = ival;
1267 Py_INCREF(fval);
1268 }
1269 PyStructSequence_SET_ITEM(v, index, ival);
1270 PyStructSequence_SET_ITEM(v, index+3, fval);
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001271}
1272
Tim Peters5aa91602002-01-30 05:46:57 +00001273/* pack a system stat C structure into the Python stat tuple
Fred Drake699f3522000-06-29 21:12:41 +00001274 (used by posix_stat() and posix_fstat()) */
1275static PyObject*
Martin v. Löwis14694662006-02-03 12:54:16 +00001276_pystat_fromstructstat(STRUCT_STAT *st)
Fred Drake699f3522000-06-29 21:12:41 +00001277{
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001278 unsigned long ansec, mnsec, cnsec;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001279 PyObject *v = PyStructSequence_New(&StatResultType);
Fred Drake699f3522000-06-29 21:12:41 +00001280 if (v == NULL)
1281 return NULL;
1282
Martin v. Löwis14694662006-02-03 12:54:16 +00001283 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long)st->st_mode));
Fred Drake699f3522000-06-29 21:12:41 +00001284#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +00001285 PyStructSequence_SET_ITEM(v, 1,
Martin v. Löwis14694662006-02-03 12:54:16 +00001286 PyLong_FromLongLong((PY_LONG_LONG)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001287#else
Martin v. Löwis14694662006-02-03 12:54:16 +00001288 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001289#endif
1290#if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
Tim Peters5aa91602002-01-30 05:46:57 +00001291 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwis14694662006-02-03 12:54:16 +00001292 PyLong_FromLongLong((PY_LONG_LONG)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001293#else
Martin v. Löwis14694662006-02-03 12:54:16 +00001294 PyStructSequence_SET_ITEM(v, 2, PyInt_FromLong((long)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001295#endif
Martin v. Löwis14694662006-02-03 12:54:16 +00001296 PyStructSequence_SET_ITEM(v, 3, PyInt_FromLong((long)st->st_nlink));
1297 PyStructSequence_SET_ITEM(v, 4, PyInt_FromLong((long)st->st_uid));
1298 PyStructSequence_SET_ITEM(v, 5, PyInt_FromLong((long)st->st_gid));
Fred Drake699f3522000-06-29 21:12:41 +00001299#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +00001300 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwis14694662006-02-03 12:54:16 +00001301 PyLong_FromLongLong((PY_LONG_LONG)st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001302#else
Martin v. Löwis14694662006-02-03 12:54:16 +00001303 PyStructSequence_SET_ITEM(v, 6, PyInt_FromLong(st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001304#endif
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001305
Martin v. Löwis14694662006-02-03 12:54:16 +00001306#if defined(HAVE_STAT_TV_NSEC)
1307 ansec = st->st_atim.tv_nsec;
1308 mnsec = st->st_mtim.tv_nsec;
1309 cnsec = st->st_ctim.tv_nsec;
1310#elif defined(HAVE_STAT_TV_NSEC2)
1311 ansec = st->st_atimespec.tv_nsec;
1312 mnsec = st->st_mtimespec.tv_nsec;
1313 cnsec = st->st_ctimespec.tv_nsec;
1314#elif defined(HAVE_STAT_NSEC)
1315 ansec = st->st_atime_nsec;
1316 mnsec = st->st_mtime_nsec;
1317 cnsec = st->st_ctime_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001318#else
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001319 ansec = mnsec = cnsec = 0;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001320#endif
Martin v. Löwis14694662006-02-03 12:54:16 +00001321 fill_time(v, 7, st->st_atime, ansec);
1322 fill_time(v, 8, st->st_mtime, mnsec);
1323 fill_time(v, 9, st->st_ctime, cnsec);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001324
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001325#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Tim Peters5aa91602002-01-30 05:46:57 +00001326 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
Martin v. Löwis14694662006-02-03 12:54:16 +00001327 PyInt_FromLong((long)st->st_blksize));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001328#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001329#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Tim Peters5aa91602002-01-30 05:46:57 +00001330 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
Martin v. Löwis14694662006-02-03 12:54:16 +00001331 PyInt_FromLong((long)st->st_blocks));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001332#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001333#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001334 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
Martin v. Löwis14694662006-02-03 12:54:16 +00001335 PyInt_FromLong((long)st->st_rdev));
Fred Drake699f3522000-06-29 21:12:41 +00001336#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001337#ifdef HAVE_STRUCT_STAT_ST_GEN
1338 PyStructSequence_SET_ITEM(v, ST_GEN_IDX,
Martin v. Löwis14694662006-02-03 12:54:16 +00001339 PyInt_FromLong((long)st->st_gen));
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001340#endif
1341#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1342 {
1343 PyObject *val;
1344 unsigned long bsec,bnsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001345 bsec = (long)st->st_birthtime;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001346#ifdef HAVE_STAT_TV_NSEC2
Hye-Shik Changd69e0342006-02-19 16:22:22 +00001347 bnsec = st->st_birthtimespec.tv_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001348#else
1349 bnsec = 0;
1350#endif
1351 if (_stat_float_times) {
1352 val = PyFloat_FromDouble(bsec + 1e-9*bnsec);
1353 } else {
1354 val = PyInt_FromLong((long)bsec);
1355 }
1356 PyStructSequence_SET_ITEM(v, ST_BIRTHTIME_IDX,
1357 val);
1358 }
1359#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001360#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1361 PyStructSequence_SET_ITEM(v, ST_FLAGS_IDX,
Martin v. Löwis14694662006-02-03 12:54:16 +00001362 PyInt_FromLong((long)st->st_flags));
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001363#endif
Fred Drake699f3522000-06-29 21:12:41 +00001364
1365 if (PyErr_Occurred()) {
1366 Py_DECREF(v);
1367 return NULL;
1368 }
1369
1370 return v;
1371}
1372
Martin v. Löwisd8948722004-06-02 09:57:56 +00001373#ifdef MS_WINDOWS
1374
1375/* IsUNCRoot -- test whether the supplied path is of the form \\SERVER\SHARE\,
1376 where / can be used in place of \ and the trailing slash is optional.
1377 Both SERVER and SHARE must have at least one character.
1378*/
1379
1380#define ISSLASHA(c) ((c) == '\\' || (c) == '/')
1381#define ISSLASHW(c) ((c) == L'\\' || (c) == L'/')
Kristján Valur Jónssondbeaa692006-06-09 16:28:01 +00001382#ifndef ARRAYSIZE
Martin v. Löwisd8948722004-06-02 09:57:56 +00001383#define ARRAYSIZE(a) (sizeof(a) / sizeof(a[0]))
Kristján Valur Jónssondbeaa692006-06-09 16:28:01 +00001384#endif
Martin v. Löwisd8948722004-06-02 09:57:56 +00001385
Tim Peters4ad82172004-08-30 17:02:04 +00001386static BOOL
Martin v. Löwisd8948722004-06-02 09:57:56 +00001387IsUNCRootA(char *path, int pathlen)
1388{
1389 #define ISSLASH ISSLASHA
1390
1391 int i, share;
1392
1393 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1394 /* minimum UNCRoot is \\x\y */
1395 return FALSE;
1396 for (i = 2; i < pathlen ; i++)
1397 if (ISSLASH(path[i])) break;
1398 if (i == 2 || i == pathlen)
1399 /* do not allow \\\SHARE or \\SERVER */
1400 return FALSE;
1401 share = i+1;
1402 for (i = share; i < pathlen; i++)
1403 if (ISSLASH(path[i])) break;
1404 return (i != share && (i == pathlen || i == pathlen-1));
1405
1406 #undef ISSLASH
1407}
1408
1409#ifdef Py_WIN_WIDE_FILENAMES
Tim Peters4ad82172004-08-30 17:02:04 +00001410static BOOL
Martin v. Löwisd8948722004-06-02 09:57:56 +00001411IsUNCRootW(Py_UNICODE *path, int pathlen)
1412{
1413 #define ISSLASH ISSLASHW
1414
1415 int i, share;
1416
1417 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1418 /* minimum UNCRoot is \\x\y */
1419 return FALSE;
1420 for (i = 2; i < pathlen ; i++)
1421 if (ISSLASH(path[i])) break;
1422 if (i == 2 || i == pathlen)
1423 /* do not allow \\\SHARE or \\SERVER */
1424 return FALSE;
1425 share = i+1;
1426 for (i = share; i < pathlen; i++)
1427 if (ISSLASH(path[i])) break;
1428 return (i != share && (i == pathlen || i == pathlen-1));
1429
1430 #undef ISSLASH
1431}
1432#endif /* Py_WIN_WIDE_FILENAMES */
1433#endif /* MS_WINDOWS */
1434
Barry Warsaw53699e91996-12-10 23:23:01 +00001435static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +00001436posix_do_stat(PyObject *self, PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001437 char *format,
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001438#ifdef __VMS
1439 int (*statfunc)(const char *, STRUCT_STAT *, ...),
1440#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001441 int (*statfunc)(const char *, STRUCT_STAT *),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001442#endif
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001443 char *wformat,
1444 int (*wstatfunc)(const Py_UNICODE *, STRUCT_STAT *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001445{
Fred Drake699f3522000-06-29 21:12:41 +00001446 STRUCT_STAT st;
Tim Peters500bd032001-12-19 19:05:01 +00001447 char *path = NULL; /* pass this to stat; do not free() it */
1448 char *pathfree = NULL; /* this memory must be free'd */
Guido van Rossumff4949e1992-08-05 19:58:53 +00001449 int res;
Martin v. Löwis14694662006-02-03 12:54:16 +00001450 PyObject *result;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001451
1452#ifdef Py_WIN_WIDE_FILENAMES
1453 /* If on wide-character-capable OS see if argument
1454 is Unicode and if so use wide API. */
1455 if (unicode_file_names()) {
1456 PyUnicodeObject *po;
1457 if (PyArg_ParseTuple(args, wformat, &po)) {
Martin v. Löwis14694662006-02-03 12:54:16 +00001458 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
1459
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001460 Py_BEGIN_ALLOW_THREADS
1461 /* PyUnicode_AS_UNICODE result OK without
1462 thread lock as it is a simple dereference. */
1463 res = wstatfunc(wpath, &st);
1464 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001465
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001466 if (res != 0)
Martin v. Löwis14694662006-02-03 12:54:16 +00001467 return win32_error_unicode("stat", wpath);
1468 return _pystat_fromstructstat(&st);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001469 }
1470 /* Drop the argument parsing error as narrow strings
1471 are also valid. */
1472 PyErr_Clear();
1473 }
1474#endif
1475
Tim Peters5aa91602002-01-30 05:46:57 +00001476 if (!PyArg_ParseTuple(args, format,
Mark Hammondef8b6542001-05-13 08:04:26 +00001477 Py_FileSystemDefaultEncoding, &path))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001478 return NULL;
Tim Peters500bd032001-12-19 19:05:01 +00001479 pathfree = path;
Guido van Rossumace88ae2000-04-21 18:54:45 +00001480
Barry Warsaw53699e91996-12-10 23:23:01 +00001481 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001482 res = (*statfunc)(path, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00001483 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001484
1485 if (res != 0) {
1486#ifdef MS_WINDOWS
1487 result = win32_error("stat", pathfree);
1488#else
1489 result = posix_error_with_filename(pathfree);
1490#endif
1491 }
1492 else
1493 result = _pystat_fromstructstat(&st);
Fred Drake699f3522000-06-29 21:12:41 +00001494
Tim Peters500bd032001-12-19 19:05:01 +00001495 PyMem_Free(pathfree);
Martin v. Löwis14694662006-02-03 12:54:16 +00001496 return result;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001497}
1498
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001499/* POSIX methods */
1500
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001501PyDoc_STRVAR(posix_access__doc__,
Georg Brandl7a284472007-01-27 19:38:50 +00001502"access(path, mode) -> True if granted, False otherwise\n\n\
Guido van Rossuma0b90752002-06-18 16:22:43 +00001503Use the real uid/gid to test for access to a path. Note that most\n\
1504operations will use the effective uid/gid, therefore this routine can\n\
1505be used in a suid/sgid environment to test if the invoking user has the\n\
1506specified access to the path. The mode argument can be F_OK to test\n\
1507existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001508
1509static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001510posix_access(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001511{
Guido van Rossum015f22a1999-01-06 22:52:38 +00001512 char *path;
1513 int mode;
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001514
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001515#ifdef Py_WIN_WIDE_FILENAMES
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001516 DWORD attr;
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001517 if (unicode_file_names()) {
1518 PyUnicodeObject *po;
1519 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
1520 Py_BEGIN_ALLOW_THREADS
1521 /* PyUnicode_AS_UNICODE OK without thread lock as
1522 it is a simple dereference. */
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001523 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001524 Py_END_ALLOW_THREADS
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001525 goto finish;
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001526 }
1527 /* Drop the argument parsing error as narrow strings
1528 are also valid. */
1529 PyErr_Clear();
1530 }
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001531 if (!PyArg_ParseTuple(args, "eti:access",
1532 Py_FileSystemDefaultEncoding, &path, &mode))
1533 return 0;
1534 Py_BEGIN_ALLOW_THREADS
1535 attr = GetFileAttributesA(path);
1536 Py_END_ALLOW_THREADS
1537 PyMem_Free(path);
1538finish:
1539 if (attr == 0xFFFFFFFF)
1540 /* File does not exist, or cannot read attributes */
1541 return PyBool_FromLong(0);
1542 /* Access is possible if either write access wasn't requested, or
1543 the file isn't read-only. */
Martin v. Löwisee1e06d2006-07-02 18:44:00 +00001544 return PyBool_FromLong(!(mode & 2) || !(attr & FILE_ATTRIBUTE_READONLY));
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001545#else
1546 int res;
Martin v. Löwisb60ae992005-03-08 09:10:29 +00001547 if (!PyArg_ParseTuple(args, "eti:access",
1548 Py_FileSystemDefaultEncoding, &path, &mode))
Guido van Rossum015f22a1999-01-06 22:52:38 +00001549 return NULL;
1550 Py_BEGIN_ALLOW_THREADS
1551 res = access(path, mode);
1552 Py_END_ALLOW_THREADS
Neal Norwitz24b3c222005-09-19 06:43:44 +00001553 PyMem_Free(path);
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001554 return PyBool_FromLong(res == 0);
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001555#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001556}
1557
Guido van Rossumd371ff11999-01-25 16:12:23 +00001558#ifndef F_OK
1559#define F_OK 0
1560#endif
1561#ifndef R_OK
1562#define R_OK 4
1563#endif
1564#ifndef W_OK
1565#define W_OK 2
1566#endif
1567#ifndef X_OK
1568#define X_OK 1
1569#endif
1570
1571#ifdef HAVE_TTYNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001572PyDoc_STRVAR(posix_ttyname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001573"ttyname(fd) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001574Return the name of the terminal device connected to 'fd'.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001575
1576static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001577posix_ttyname(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001578{
Guido van Rossum94f6f721999-01-06 18:42:14 +00001579 int id;
1580 char *ret;
1581
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001582 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
Guido van Rossum94f6f721999-01-06 18:42:14 +00001583 return NULL;
1584
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001585#if defined(__VMS)
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001586 /* file descriptor 0 only, the default input device (stdin) */
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001587 if (id == 0) {
1588 ret = ttyname();
1589 }
1590 else {
1591 ret = NULL;
1592 }
1593#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00001594 ret = ttyname(id);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001595#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001596 if (ret == NULL)
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001597 return posix_error();
1598 return PyString_FromString(ret);
Guido van Rossum94f6f721999-01-06 18:42:14 +00001599}
Guido van Rossumd371ff11999-01-25 16:12:23 +00001600#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001601
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001602#ifdef HAVE_CTERMID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001603PyDoc_STRVAR(posix_ctermid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001604"ctermid() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001605Return the name of the controlling terminal for this process.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001606
1607static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001608posix_ctermid(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001609{
1610 char *ret;
1611 char buffer[L_ctermid];
1612
Greg Wardb48bc172000-03-01 21:51:56 +00001613#ifdef USE_CTERMID_R
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001614 ret = ctermid_r(buffer);
1615#else
1616 ret = ctermid(buffer);
1617#endif
1618 if (ret == NULL)
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001619 return posix_error();
1620 return PyString_FromString(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001621}
1622#endif
1623
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001624PyDoc_STRVAR(posix_chdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001625"chdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001626Change the current working directory to the specified path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001627
Barry Warsaw53699e91996-12-10 23:23:01 +00001628static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001629posix_chdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001630{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001631#ifdef MS_WINDOWS
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00001632 return win32_1str(args, "chdir", "s:chdir", win32_chdir, "U:chdir", win32_wchdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001633#elif defined(PYOS_OS2) && defined(PYCC_GCC)
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00001634 return posix_1str(args, "et:chdir", _chdir2);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00001635#elif defined(__VMS)
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00001636 return posix_1str(args, "et:chdir", (int (*)(const char *))chdir);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001637#else
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00001638 return posix_1str(args, "et:chdir", chdir);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001639#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001640}
1641
Fred Drake4d1e64b2002-04-15 19:40:07 +00001642#ifdef HAVE_FCHDIR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001643PyDoc_STRVAR(posix_fchdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001644"fchdir(fildes)\n\n\
Fred Drake4d1e64b2002-04-15 19:40:07 +00001645Change to the directory of the given file descriptor. fildes must be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001646opened on a directory, not a file.");
Fred Drake4d1e64b2002-04-15 19:40:07 +00001647
1648static PyObject *
1649posix_fchdir(PyObject *self, PyObject *fdobj)
1650{
1651 return posix_fildes(fdobj, fchdir);
1652}
1653#endif /* HAVE_FCHDIR */
1654
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001655
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001656PyDoc_STRVAR(posix_chmod__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001657"chmod(path, mode)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001658Change the access permissions of a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001659
Barry Warsaw53699e91996-12-10 23:23:01 +00001660static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001661posix_chmod(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001662{
Mark Hammondef8b6542001-05-13 08:04:26 +00001663 char *path = NULL;
Guido van Rossumffd15f52000-03-31 00:47:28 +00001664 int i;
1665 int res;
Mark Hammond817c9292003-12-03 01:22:38 +00001666#ifdef Py_WIN_WIDE_FILENAMES
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001667 DWORD attr;
Mark Hammond817c9292003-12-03 01:22:38 +00001668 if (unicode_file_names()) {
1669 PyUnicodeObject *po;
1670 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
1671 Py_BEGIN_ALLOW_THREADS
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001672 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1673 if (attr != 0xFFFFFFFF) {
1674 if (i & _S_IWRITE)
1675 attr &= ~FILE_ATTRIBUTE_READONLY;
1676 else
1677 attr |= FILE_ATTRIBUTE_READONLY;
1678 res = SetFileAttributesW(PyUnicode_AS_UNICODE(po), attr);
1679 }
1680 else
1681 res = 0;
Mark Hammond817c9292003-12-03 01:22:38 +00001682 Py_END_ALLOW_THREADS
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001683 if (!res)
1684 return win32_error_unicode("chmod",
Mark Hammond817c9292003-12-03 01:22:38 +00001685 PyUnicode_AS_UNICODE(po));
1686 Py_INCREF(Py_None);
1687 return Py_None;
1688 }
1689 /* Drop the argument parsing error as narrow strings
1690 are also valid. */
1691 PyErr_Clear();
1692 }
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001693 if (!PyArg_ParseTuple(args, "eti:chmod", Py_FileSystemDefaultEncoding,
1694 &path, &i))
1695 return NULL;
1696 Py_BEGIN_ALLOW_THREADS
1697 attr = GetFileAttributesA(path);
1698 if (attr != 0xFFFFFFFF) {
1699 if (i & _S_IWRITE)
1700 attr &= ~FILE_ATTRIBUTE_READONLY;
1701 else
1702 attr |= FILE_ATTRIBUTE_READONLY;
1703 res = SetFileAttributesA(path, attr);
1704 }
1705 else
1706 res = 0;
1707 Py_END_ALLOW_THREADS
1708 if (!res) {
1709 win32_error("chmod", path);
1710 PyMem_Free(path);
1711 return NULL;
1712 }
Martin v. Löwis9f485bc2006-05-08 05:25:56 +00001713 PyMem_Free(path);
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001714 Py_INCREF(Py_None);
1715 return Py_None;
1716#else /* Py_WIN_WIDE_FILENAMES */
Mark Hammond817c9292003-12-03 01:22:38 +00001717 if (!PyArg_ParseTuple(args, "eti:chmod", Py_FileSystemDefaultEncoding,
Mark Hammondef8b6542001-05-13 08:04:26 +00001718 &path, &i))
Guido van Rossumffd15f52000-03-31 00:47:28 +00001719 return NULL;
1720 Py_BEGIN_ALLOW_THREADS
Guido van Rossumef40e772000-03-31 01:26:23 +00001721 res = chmod(path, i);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001722 Py_END_ALLOW_THREADS
1723 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00001724 return posix_error_with_allocated_filename(path);
1725 PyMem_Free(path);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001726 Py_INCREF(Py_None);
1727 return Py_None;
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001728#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001729}
1730
Christian Heimes36281872007-11-30 21:11:28 +00001731#ifdef HAVE_FCHMOD
1732PyDoc_STRVAR(posix_fchmod__doc__,
1733"fchmod(fd, mode)\n\n\
1734Change the access permissions of the file given by file\n\
1735descriptor fd.");
1736
1737static PyObject *
1738posix_fchmod(PyObject *self, PyObject *args)
1739{
1740 int fd, mode, res;
1741 if (!PyArg_ParseTuple(args, "ii:fchmod", &fd, &mode))
1742 return NULL;
1743 Py_BEGIN_ALLOW_THREADS
1744 res = fchmod(fd, mode);
1745 Py_END_ALLOW_THREADS
1746 if (res < 0)
1747 return posix_error();
1748 Py_RETURN_NONE;
1749}
1750#endif /* HAVE_FCHMOD */
1751
1752#ifdef HAVE_LCHMOD
1753PyDoc_STRVAR(posix_lchmod__doc__,
1754"lchmod(path, mode)\n\n\
1755Change the access permissions of a file. If path is a symlink, this\n\
1756affects the link itself rather than the target.");
1757
1758static PyObject *
1759posix_lchmod(PyObject *self, PyObject *args)
1760{
1761 char *path = NULL;
1762 int i;
1763 int res;
1764 if (!PyArg_ParseTuple(args, "eti:lchmod", Py_FileSystemDefaultEncoding,
1765 &path, &i))
1766 return NULL;
1767 Py_BEGIN_ALLOW_THREADS
1768 res = lchmod(path, i);
1769 Py_END_ALLOW_THREADS
1770 if (res < 0)
1771 return posix_error_with_allocated_filename(path);
1772 PyMem_Free(path);
1773 Py_RETURN_NONE;
1774}
1775#endif /* HAVE_LCHMOD */
1776
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001777
Martin v. Löwis382abef2007-02-19 10:55:19 +00001778#ifdef HAVE_CHFLAGS
1779PyDoc_STRVAR(posix_chflags__doc__,
1780"chflags(path, flags)\n\n\
1781Set file flags.");
1782
1783static PyObject *
1784posix_chflags(PyObject *self, PyObject *args)
1785{
1786 char *path;
1787 unsigned long flags;
1788 int res;
1789 if (!PyArg_ParseTuple(args, "etk:chflags",
1790 Py_FileSystemDefaultEncoding, &path, &flags))
1791 return NULL;
1792 Py_BEGIN_ALLOW_THREADS
1793 res = chflags(path, flags);
1794 Py_END_ALLOW_THREADS
1795 if (res < 0)
1796 return posix_error_with_allocated_filename(path);
1797 PyMem_Free(path);
1798 Py_INCREF(Py_None);
1799 return Py_None;
1800}
1801#endif /* HAVE_CHFLAGS */
1802
1803#ifdef HAVE_LCHFLAGS
1804PyDoc_STRVAR(posix_lchflags__doc__,
1805"lchflags(path, flags)\n\n\
1806Set file flags.\n\
1807This function will not follow symbolic links.");
1808
1809static PyObject *
1810posix_lchflags(PyObject *self, PyObject *args)
1811{
1812 char *path;
1813 unsigned long flags;
1814 int res;
1815 if (!PyArg_ParseTuple(args, "etk:lchflags",
1816 Py_FileSystemDefaultEncoding, &path, &flags))
1817 return NULL;
1818 Py_BEGIN_ALLOW_THREADS
1819 res = lchflags(path, flags);
1820 Py_END_ALLOW_THREADS
1821 if (res < 0)
1822 return posix_error_with_allocated_filename(path);
1823 PyMem_Free(path);
1824 Py_INCREF(Py_None);
1825 return Py_None;
1826}
1827#endif /* HAVE_LCHFLAGS */
1828
Martin v. Löwis244edc82001-10-04 22:44:26 +00001829#ifdef HAVE_CHROOT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001830PyDoc_STRVAR(posix_chroot__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001831"chroot(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001832Change root directory to path.");
Martin v. Löwis244edc82001-10-04 22:44:26 +00001833
1834static PyObject *
1835posix_chroot(PyObject *self, PyObject *args)
1836{
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00001837 return posix_1str(args, "et:chroot", chroot);
Martin v. Löwis244edc82001-10-04 22:44:26 +00001838}
1839#endif
1840
Guido van Rossum21142a01999-01-08 21:05:37 +00001841#ifdef HAVE_FSYNC
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001842PyDoc_STRVAR(posix_fsync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001843"fsync(fildes)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001844force write of file with filedescriptor to disk.");
Guido van Rossum21142a01999-01-08 21:05:37 +00001845
1846static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00001847posix_fsync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00001848{
Fred Drake4d1e64b2002-04-15 19:40:07 +00001849 return posix_fildes(fdobj, fsync);
Guido van Rossum21142a01999-01-08 21:05:37 +00001850}
1851#endif /* HAVE_FSYNC */
1852
1853#ifdef HAVE_FDATASYNC
Guido van Rossumecc23b02000-09-22 16:01:05 +00001854
Guido van Rossum7f58e2e2000-09-22 17:26:14 +00001855#ifdef __hpux
Guido van Rossumecc23b02000-09-22 16:01:05 +00001856extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
1857#endif
1858
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001859PyDoc_STRVAR(posix_fdatasync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001860"fdatasync(fildes)\n\n\
Guido van Rossum21142a01999-01-08 21:05:37 +00001861force write of file with filedescriptor to disk.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001862 does not force update of metadata.");
Guido van Rossum21142a01999-01-08 21:05:37 +00001863
1864static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00001865posix_fdatasync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00001866{
Fred Drake4d1e64b2002-04-15 19:40:07 +00001867 return posix_fildes(fdobj, fdatasync);
Guido van Rossum21142a01999-01-08 21:05:37 +00001868}
1869#endif /* HAVE_FDATASYNC */
1870
1871
Fredrik Lundh10723342000-07-10 16:38:09 +00001872#ifdef HAVE_CHOWN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001873PyDoc_STRVAR(posix_chown__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001874"chown(path, uid, gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001875Change the owner and group id of path to the numeric uid and gid.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001876
Barry Warsaw53699e91996-12-10 23:23:01 +00001877static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001878posix_chown(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001879{
Mark Hammondef8b6542001-05-13 08:04:26 +00001880 char *path = NULL;
Fredrik Lundh44328e62000-07-10 15:59:30 +00001881 int uid, gid;
1882 int res;
Tim Peters5aa91602002-01-30 05:46:57 +00001883 if (!PyArg_ParseTuple(args, "etii:chown",
1884 Py_FileSystemDefaultEncoding, &path,
Mark Hammondef8b6542001-05-13 08:04:26 +00001885 &uid, &gid))
Fredrik Lundh44328e62000-07-10 15:59:30 +00001886 return NULL;
1887 Py_BEGIN_ALLOW_THREADS
1888 res = chown(path, (uid_t) uid, (gid_t) gid);
1889 Py_END_ALLOW_THREADS
1890 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00001891 return posix_error_with_allocated_filename(path);
1892 PyMem_Free(path);
Fredrik Lundh44328e62000-07-10 15:59:30 +00001893 Py_INCREF(Py_None);
1894 return Py_None;
Guido van Rossumb6775db1994-08-01 11:34:53 +00001895}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001896#endif /* HAVE_CHOWN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00001897
Christian Heimes36281872007-11-30 21:11:28 +00001898#ifdef HAVE_FCHOWN
1899PyDoc_STRVAR(posix_fchown__doc__,
1900"fchown(fd, uid, gid)\n\n\
1901Change the owner and group id of the file given by file descriptor\n\
1902fd to the numeric uid and gid.");
1903
1904static PyObject *
1905posix_fchown(PyObject *self, PyObject *args)
1906{
1907 int fd, uid, gid;
1908 int res;
1909 if (!PyArg_ParseTuple(args, "iii:chown", &fd, &uid, &gid))
1910 return NULL;
1911 Py_BEGIN_ALLOW_THREADS
1912 res = fchown(fd, (uid_t) uid, (gid_t) gid);
1913 Py_END_ALLOW_THREADS
1914 if (res < 0)
1915 return posix_error();
1916 Py_RETURN_NONE;
1917}
1918#endif /* HAVE_FCHOWN */
1919
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00001920#ifdef HAVE_LCHOWN
1921PyDoc_STRVAR(posix_lchown__doc__,
1922"lchown(path, uid, gid)\n\n\
1923Change the owner and group id of path to the numeric uid and gid.\n\
1924This function will not follow symbolic links.");
1925
1926static PyObject *
1927posix_lchown(PyObject *self, PyObject *args)
1928{
1929 char *path = NULL;
1930 int uid, gid;
1931 int res;
1932 if (!PyArg_ParseTuple(args, "etii:lchown",
1933 Py_FileSystemDefaultEncoding, &path,
1934 &uid, &gid))
1935 return NULL;
1936 Py_BEGIN_ALLOW_THREADS
1937 res = lchown(path, (uid_t) uid, (gid_t) gid);
1938 Py_END_ALLOW_THREADS
Tim Peters11b23062003-04-23 02:39:17 +00001939 if (res < 0)
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00001940 return posix_error_with_allocated_filename(path);
1941 PyMem_Free(path);
1942 Py_INCREF(Py_None);
1943 return Py_None;
1944}
1945#endif /* HAVE_LCHOWN */
1946
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001947
Guido van Rossum36bc6801995-06-14 22:54:23 +00001948#ifdef HAVE_GETCWD
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001949PyDoc_STRVAR(posix_getcwd__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001950"getcwd() -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001951Return a string representing the current working directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001952
Barry Warsaw53699e91996-12-10 23:23:01 +00001953static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001954posix_getcwd(PyObject *self, PyObject *noargs)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001955{
1956 char buf[1026];
Guido van Rossumff4949e1992-08-05 19:58:53 +00001957 char *res;
Neal Norwitze241ce82003-02-17 18:17:05 +00001958
Barry Warsaw53699e91996-12-10 23:23:01 +00001959 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001960#if defined(PYOS_OS2) && defined(PYCC_GCC)
1961 res = _getcwd2(buf, sizeof buf);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001962#else
Guido van Rossumff4949e1992-08-05 19:58:53 +00001963 res = getcwd(buf, sizeof buf);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001964#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00001965 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001966 if (res == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001967 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00001968 return PyString_FromString(buf);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001969}
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001970
Walter Dörwald3b918c32002-11-21 20:18:46 +00001971#ifdef Py_USING_UNICODE
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001972PyDoc_STRVAR(posix_getcwdu__doc__,
1973"getcwdu() -> path\n\n\
1974Return a unicode string representing the current working directory.");
1975
1976static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001977posix_getcwdu(PyObject *self, PyObject *noargs)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001978{
1979 char buf[1026];
1980 char *res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001981
1982#ifdef Py_WIN_WIDE_FILENAMES
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001983 DWORD len;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001984 if (unicode_file_names()) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001985 wchar_t wbuf[1026];
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001986 wchar_t *wbuf2 = wbuf;
1987 PyObject *resobj;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001988 Py_BEGIN_ALLOW_THREADS
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001989 len = GetCurrentDirectoryW(sizeof wbuf/ sizeof wbuf[0], wbuf);
1990 /* If the buffer is large enough, len does not include the
1991 terminating \0. If the buffer is too small, len includes
1992 the space needed for the terminator. */
1993 if (len >= sizeof wbuf/ sizeof wbuf[0]) {
1994 wbuf2 = malloc(len * sizeof(wchar_t));
1995 if (wbuf2)
1996 len = GetCurrentDirectoryW(len, wbuf2);
1997 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001998 Py_END_ALLOW_THREADS
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00001999 if (!wbuf2) {
2000 PyErr_NoMemory();
2001 return NULL;
2002 }
2003 if (!len) {
2004 if (wbuf2 != wbuf) free(wbuf2);
2005 return win32_error("getcwdu", NULL);
2006 }
2007 resobj = PyUnicode_FromWideChar(wbuf2, len);
2008 if (wbuf2 != wbuf) free(wbuf2);
2009 return resobj;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002010 }
2011#endif
2012
2013 Py_BEGIN_ALLOW_THREADS
2014#if defined(PYOS_OS2) && defined(PYCC_GCC)
2015 res = _getcwd2(buf, sizeof buf);
2016#else
2017 res = getcwd(buf, sizeof buf);
2018#endif
2019 Py_END_ALLOW_THREADS
2020 if (res == NULL)
2021 return posix_error();
2022 return PyUnicode_Decode(buf, strlen(buf), Py_FileSystemDefaultEncoding,"strict");
2023}
Guido van Rossum36bc6801995-06-14 22:54:23 +00002024#endif
Walter Dörwald3b918c32002-11-21 20:18:46 +00002025#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002026
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002027
Guido van Rossumb6775db1994-08-01 11:34:53 +00002028#ifdef HAVE_LINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002029PyDoc_STRVAR(posix_link__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002030"link(src, dst)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002031Create a hard link to a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002032
Barry Warsaw53699e91996-12-10 23:23:01 +00002033static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002034posix_link(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002035{
Martin v. Löwis4fc2bda2006-05-04 12:04:27 +00002036 return posix_2str(args, "etet:link", link);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002037}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002038#endif /* HAVE_LINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002039
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002040
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002041PyDoc_STRVAR(posix_listdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002042"listdir(path) -> list_of_strings\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002043Return a list containing the names of the entries in the directory.\n\
2044\n\
2045 path: path of directory to list\n\
2046\n\
2047The list is in arbitrary order. It does not include the special\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002048entries '.' and '..' even if they are present in the directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002049
Barry Warsaw53699e91996-12-10 23:23:01 +00002050static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002051posix_listdir(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002052{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002053 /* XXX Should redo this putting the (now four) versions of opendir
Guido van Rossum6d8841c1997-08-14 19:57:39 +00002054 in separate files instead of having them all here... */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002055#if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002056
Barry Warsaw53699e91996-12-10 23:23:01 +00002057 PyObject *d, *v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002058 HANDLE hFindFile;
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002059 BOOL result;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002060 WIN32_FIND_DATA FileData;
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002061 char namebuf[MAX_PATH+5]; /* Overallocate for \\*.*\0 */
Mark Hammondef8b6542001-05-13 08:04:26 +00002062 char *bufptr = namebuf;
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002063 Py_ssize_t len = sizeof(namebuf)-5; /* only claim to have space for MAX_PATH */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002064
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002065#ifdef Py_WIN_WIDE_FILENAMES
2066 /* If on wide-character-capable OS see if argument
2067 is Unicode and if so use wide API. */
2068 if (unicode_file_names()) {
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002069 PyObject *po;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002070 if (PyArg_ParseTuple(args, "U:listdir", &po)) {
2071 WIN32_FIND_DATAW wFileData;
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002072 Py_UNICODE *wnamebuf;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002073 Py_UNICODE wch;
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002074 /* Overallocate for \\*.*\0 */
2075 len = PyUnicode_GET_SIZE(po);
2076 wnamebuf = malloc((len + 5) * sizeof(wchar_t));
2077 if (!wnamebuf) {
2078 PyErr_NoMemory();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002079 return NULL;
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002080 }
2081 wcscpy(wnamebuf, PyUnicode_AS_UNICODE(po));
2082 wch = len > 0 ? wnamebuf[len-1] : '\0';
2083 if (wch != L'/' && wch != L'\\' && wch != L':')
2084 wnamebuf[len++] = L'\\';
2085 wcscpy(wnamebuf + len, L"*.*");
2086 if ((d = PyList_New(0)) == NULL) {
2087 free(wnamebuf);
2088 return NULL;
2089 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002090 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
2091 if (hFindFile == INVALID_HANDLE_VALUE) {
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002092 int error = GetLastError();
2093 if (error == ERROR_FILE_NOT_FOUND) {
2094 free(wnamebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002095 return d;
2096 }
2097 Py_DECREF(d);
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002098 win32_error_unicode("FindFirstFileW", wnamebuf);
2099 free(wnamebuf);
2100 return NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002101 }
2102 do {
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002103 /* Skip over . and .. */
2104 if (wcscmp(wFileData.cFileName, L".") != 0 &&
2105 wcscmp(wFileData.cFileName, L"..") != 0) {
2106 v = PyUnicode_FromUnicode(wFileData.cFileName, wcslen(wFileData.cFileName));
2107 if (v == NULL) {
2108 Py_DECREF(d);
2109 d = NULL;
2110 break;
2111 }
2112 if (PyList_Append(d, v) != 0) {
2113 Py_DECREF(v);
2114 Py_DECREF(d);
2115 d = NULL;
2116 break;
2117 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002118 Py_DECREF(v);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002119 }
Georg Brandl622927b2006-03-07 12:48:03 +00002120 Py_BEGIN_ALLOW_THREADS
2121 result = FindNextFileW(hFindFile, &wFileData);
2122 Py_END_ALLOW_THREADS
Martin v. Löwis982e9fe2006-07-24 12:54:17 +00002123 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2124 it got to the end of the directory. */
2125 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2126 Py_DECREF(d);
2127 win32_error_unicode("FindNextFileW", wnamebuf);
2128 FindClose(hFindFile);
2129 free(wnamebuf);
2130 return NULL;
2131 }
Georg Brandl622927b2006-03-07 12:48:03 +00002132 } while (result == TRUE);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002133
2134 if (FindClose(hFindFile) == FALSE) {
2135 Py_DECREF(d);
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002136 win32_error_unicode("FindClose", wnamebuf);
2137 free(wnamebuf);
2138 return NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002139 }
Martin v. Löwise3edaea2006-05-15 05:51:36 +00002140 free(wnamebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002141 return d;
2142 }
2143 /* Drop the argument parsing error as narrow strings
2144 are also valid. */
2145 PyErr_Clear();
2146 }
2147#endif
2148
Tim Peters5aa91602002-01-30 05:46:57 +00002149 if (!PyArg_ParseTuple(args, "et#:listdir",
Mark Hammondef8b6542001-05-13 08:04:26 +00002150 Py_FileSystemDefaultEncoding, &bufptr, &len))
Guido van Rossumb6775db1994-08-01 11:34:53 +00002151 return NULL;
Neil Schemenauer94b866a2002-03-22 20:51:58 +00002152 if (len > 0) {
2153 char ch = namebuf[len-1];
2154 if (ch != SEP && ch != ALTSEP && ch != ':')
2155 namebuf[len++] = '/';
2156 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002157 strcpy(namebuf + len, "*.*");
2158
Barry Warsaw53699e91996-12-10 23:23:01 +00002159 if ((d = PyList_New(0)) == NULL)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002160 return NULL;
2161
2162 hFindFile = FindFirstFile(namebuf, &FileData);
2163 if (hFindFile == INVALID_HANDLE_VALUE) {
Martin v. Löwis682b1bb2006-05-12 12:27:28 +00002164 int error = GetLastError();
2165 if (error == ERROR_FILE_NOT_FOUND)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002166 return d;
2167 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00002168 return win32_error("FindFirstFile", namebuf);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002169 }
2170 do {
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002171 /* Skip over . and .. */
2172 if (strcmp(FileData.cFileName, ".") != 0 &&
2173 strcmp(FileData.cFileName, "..") != 0) {
2174 v = PyString_FromString(FileData.cFileName);
2175 if (v == NULL) {
2176 Py_DECREF(d);
2177 d = NULL;
2178 break;
2179 }
2180 if (PyList_Append(d, v) != 0) {
2181 Py_DECREF(v);
2182 Py_DECREF(d);
2183 d = NULL;
2184 break;
2185 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002186 Py_DECREF(v);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002187 }
Georg Brandl622927b2006-03-07 12:48:03 +00002188 Py_BEGIN_ALLOW_THREADS
2189 result = FindNextFile(hFindFile, &FileData);
2190 Py_END_ALLOW_THREADS
Martin v. Löwis982e9fe2006-07-24 12:54:17 +00002191 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2192 it got to the end of the directory. */
2193 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2194 Py_DECREF(d);
2195 win32_error("FindNextFile", namebuf);
2196 FindClose(hFindFile);
2197 return NULL;
2198 }
Georg Brandl622927b2006-03-07 12:48:03 +00002199 } while (result == TRUE);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002200
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002201 if (FindClose(hFindFile) == FALSE) {
2202 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00002203 return win32_error("FindClose", namebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002204 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002205
2206 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002207
Tim Peters0bb44a42000-09-15 07:44:49 +00002208#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002209
2210#ifndef MAX_PATH
2211#define MAX_PATH CCHMAXPATH
2212#endif
2213 char *name, *pt;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00002214 Py_ssize_t len;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002215 PyObject *d, *v;
2216 char namebuf[MAX_PATH+5];
2217 HDIR hdir = 1;
2218 ULONG srchcnt = 1;
2219 FILEFINDBUF3 ep;
2220 APIRET rc;
2221
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002222 if (!PyArg_ParseTuple(args, "t#:listdir", &name, &len))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002223 return NULL;
2224 if (len >= MAX_PATH) {
2225 PyErr_SetString(PyExc_ValueError, "path too long");
2226 return NULL;
2227 }
2228 strcpy(namebuf, name);
2229 for (pt = namebuf; *pt; pt++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002230 if (*pt == ALTSEP)
2231 *pt = SEP;
2232 if (namebuf[len-1] != SEP)
2233 namebuf[len++] = SEP;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002234 strcpy(namebuf + len, "*.*");
2235
2236 if ((d = PyList_New(0)) == NULL)
2237 return NULL;
2238
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002239 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
2240 &hdir, /* Handle to Use While Search Directory */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002241 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002242 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
2243 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
2244 FIL_STANDARD); /* Format of Entry (EAs or Not) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002245
2246 if (rc != NO_ERROR) {
2247 errno = ENOENT;
Barry Warsawf63b8cc1999-05-27 23:13:21 +00002248 return posix_error_with_filename(name);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002249 }
2250
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002251 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002252 do {
2253 if (ep.achName[0] == '.'
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002254 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002255 continue; /* Skip Over "." and ".." Names */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002256
2257 strcpy(namebuf, ep.achName);
2258
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002259 /* Leave Case of Name Alone -- In Native Form */
2260 /* (Removed Forced Lowercasing Code) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002261
2262 v = PyString_FromString(namebuf);
2263 if (v == NULL) {
2264 Py_DECREF(d);
2265 d = NULL;
2266 break;
2267 }
2268 if (PyList_Append(d, v) != 0) {
2269 Py_DECREF(v);
2270 Py_DECREF(d);
2271 d = NULL;
2272 break;
2273 }
2274 Py_DECREF(v);
2275 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
2276 }
2277
2278 return d;
2279#else
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002280
Just van Rossum2fe07fd2003-03-03 19:07:13 +00002281 char *name = NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002282 PyObject *d, *v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002283 DIR *dirp;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002284 struct dirent *ep;
Just van Rossum96b1c902003-03-03 17:32:15 +00002285 int arg_is_unicode = 1;
2286
Georg Brandl05e89b82006-04-11 07:04:06 +00002287 errno = 0;
Just van Rossum96b1c902003-03-03 17:32:15 +00002288 if (!PyArg_ParseTuple(args, "U:listdir", &v)) {
2289 arg_is_unicode = 0;
2290 PyErr_Clear();
2291 }
2292 if (!PyArg_ParseTuple(args, "et:listdir", Py_FileSystemDefaultEncoding, &name))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002293 return NULL;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002294 if ((dirp = opendir(name)) == NULL) {
Just van Rossum2fe07fd2003-03-03 19:07:13 +00002295 return posix_error_with_allocated_filename(name);
Guido van Rossumff4949e1992-08-05 19:58:53 +00002296 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002297 if ((d = PyList_New(0)) == NULL) {
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002298 closedir(dirp);
Just van Rossum2fe07fd2003-03-03 19:07:13 +00002299 PyMem_Free(name);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002300 return NULL;
2301 }
Georg Brandl622927b2006-03-07 12:48:03 +00002302 for (;;) {
2303 Py_BEGIN_ALLOW_THREADS
2304 ep = readdir(dirp);
2305 Py_END_ALLOW_THREADS
2306 if (ep == NULL)
2307 break;
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002308 if (ep->d_name[0] == '.' &&
2309 (NAMLEN(ep) == 1 ||
Guido van Rossuma376cc51996-12-05 23:43:35 +00002310 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002311 continue;
Barry Warsaw53699e91996-12-10 23:23:01 +00002312 v = PyString_FromStringAndSize(ep->d_name, NAMLEN(ep));
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002313 if (v == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00002314 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002315 d = NULL;
2316 break;
2317 }
Just van Rossum46c97842003-02-25 21:42:15 +00002318#ifdef Py_USING_UNICODE
Just van Rossum96b1c902003-03-03 17:32:15 +00002319 if (arg_is_unicode) {
Just van Rossum46c97842003-02-25 21:42:15 +00002320 PyObject *w;
2321
2322 w = PyUnicode_FromEncodedObject(v,
Tim Peters11b23062003-04-23 02:39:17 +00002323 Py_FileSystemDefaultEncoding,
Just van Rossum46c97842003-02-25 21:42:15 +00002324 "strict");
Just van Rossum6a421832003-03-04 19:30:44 +00002325 if (w != NULL) {
2326 Py_DECREF(v);
2327 v = w;
2328 }
2329 else {
2330 /* fall back to the original byte string, as
2331 discussed in patch #683592 */
2332 PyErr_Clear();
Just van Rossum46c97842003-02-25 21:42:15 +00002333 }
Just van Rossum46c97842003-02-25 21:42:15 +00002334 }
2335#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002336 if (PyList_Append(d, v) != 0) {
2337 Py_DECREF(v);
2338 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002339 d = NULL;
2340 break;
2341 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002342 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002343 }
Georg Brandlbbfe4fa2006-04-11 06:47:43 +00002344 if (errno != 0 && d != NULL) {
2345 /* readdir() returned NULL and set errno */
2346 closedir(dirp);
2347 Py_DECREF(d);
2348 return posix_error_with_allocated_filename(name);
2349 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002350 closedir(dirp);
Just van Rossum2fe07fd2003-03-03 19:07:13 +00002351 PyMem_Free(name);
Guido van Rossum0ee42cd1991-04-08 21:01:03 +00002352
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002353 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002354
Tim Peters0bb44a42000-09-15 07:44:49 +00002355#endif /* which OS */
2356} /* end of posix_listdir */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002357
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002358#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00002359/* A helper function for abspath on win32 */
2360static PyObject *
2361posix__getfullpathname(PyObject *self, PyObject *args)
2362{
2363 /* assume encoded strings wont more than double no of chars */
2364 char inbuf[MAX_PATH*2];
2365 char *inbufp = inbuf;
Tim Peters67d70eb2006-03-01 04:35:45 +00002366 Py_ssize_t insize = sizeof(inbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00002367 char outbuf[MAX_PATH*2];
2368 char *temp;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002369#ifdef Py_WIN_WIDE_FILENAMES
2370 if (unicode_file_names()) {
2371 PyUnicodeObject *po;
2372 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po)) {
2373 Py_UNICODE woutbuf[MAX_PATH*2];
2374 Py_UNICODE *wtemp;
Tim Peters11b23062003-04-23 02:39:17 +00002375 if (!GetFullPathNameW(PyUnicode_AS_UNICODE(po),
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002376 sizeof(woutbuf)/sizeof(woutbuf[0]),
2377 woutbuf, &wtemp))
2378 return win32_error("GetFullPathName", "");
2379 return PyUnicode_FromUnicode(woutbuf, wcslen(woutbuf));
2380 }
2381 /* Drop the argument parsing error as narrow strings
2382 are also valid. */
2383 PyErr_Clear();
2384 }
2385#endif
Tim Peters5aa91602002-01-30 05:46:57 +00002386 if (!PyArg_ParseTuple (args, "et#:_getfullpathname",
2387 Py_FileSystemDefaultEncoding, &inbufp,
Mark Hammondef8b6542001-05-13 08:04:26 +00002388 &insize))
2389 return NULL;
2390 if (!GetFullPathName(inbuf, sizeof(outbuf)/sizeof(outbuf[0]),
2391 outbuf, &temp))
2392 return win32_error("GetFullPathName", inbuf);
Martin v. Löwis969297f2004-06-15 18:49:58 +00002393 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
2394 return PyUnicode_Decode(outbuf, strlen(outbuf),
2395 Py_FileSystemDefaultEncoding, NULL);
2396 }
Mark Hammondef8b6542001-05-13 08:04:26 +00002397 return PyString_FromString(outbuf);
2398} /* end of posix__getfullpathname */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002399#endif /* MS_WINDOWS */
Mark Hammondef8b6542001-05-13 08:04:26 +00002400
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002401PyDoc_STRVAR(posix_mkdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002402"mkdir(path [, mode=0777])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002403Create a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002404
Barry Warsaw53699e91996-12-10 23:23:01 +00002405static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002406posix_mkdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002407{
Guido van Rossumb0824db1996-02-25 04:50:32 +00002408 int res;
Mark Hammondef8b6542001-05-13 08:04:26 +00002409 char *path = NULL;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002410 int mode = 0777;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002411
2412#ifdef Py_WIN_WIDE_FILENAMES
2413 if (unicode_file_names()) {
2414 PyUnicodeObject *po;
Mark Hammond05107b62003-02-19 04:08:27 +00002415 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode)) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002416 Py_BEGIN_ALLOW_THREADS
2417 /* PyUnicode_AS_UNICODE OK without thread lock as
2418 it is a simple dereference. */
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002419 res = CreateDirectoryW(PyUnicode_AS_UNICODE(po), NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002420 Py_END_ALLOW_THREADS
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002421 if (!res)
2422 return win32_error_unicode("mkdir", PyUnicode_AS_UNICODE(po));
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002423 Py_INCREF(Py_None);
2424 return Py_None;
2425 }
2426 /* Drop the argument parsing error as narrow strings
2427 are also valid. */
2428 PyErr_Clear();
2429 }
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002430 if (!PyArg_ParseTuple(args, "et|i:mkdir",
2431 Py_FileSystemDefaultEncoding, &path, &mode))
2432 return NULL;
2433 Py_BEGIN_ALLOW_THREADS
2434 /* PyUnicode_AS_UNICODE OK without thread lock as
2435 it is a simple dereference. */
2436 res = CreateDirectoryA(path, NULL);
2437 Py_END_ALLOW_THREADS
2438 if (!res) {
2439 win32_error("mkdir", path);
2440 PyMem_Free(path);
2441 return NULL;
2442 }
2443 PyMem_Free(path);
2444 Py_INCREF(Py_None);
2445 return Py_None;
2446#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002447
Tim Peters5aa91602002-01-30 05:46:57 +00002448 if (!PyArg_ParseTuple(args, "et|i:mkdir",
Mark Hammondef8b6542001-05-13 08:04:26 +00002449 Py_FileSystemDefaultEncoding, &path, &mode))
Guido van Rossumb0824db1996-02-25 04:50:32 +00002450 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002451 Py_BEGIN_ALLOW_THREADS
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002452#if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002453 res = mkdir(path);
2454#else
Guido van Rossumb0824db1996-02-25 04:50:32 +00002455 res = mkdir(path, mode);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002456#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002457 Py_END_ALLOW_THREADS
Guido van Rossumb0824db1996-02-25 04:50:32 +00002458 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00002459 return posix_error_with_allocated_filename(path);
2460 PyMem_Free(path);
Barry Warsaw53699e91996-12-10 23:23:01 +00002461 Py_INCREF(Py_None);
2462 return Py_None;
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002463#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002464}
2465
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002466
Neal Norwitz1818ed72006-03-26 00:29:48 +00002467/* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
2468#if defined(HAVE_SYS_RESOURCE_H)
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002469#include <sys/resource.h>
2470#endif
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002471
Neal Norwitz1818ed72006-03-26 00:29:48 +00002472
2473#ifdef HAVE_NICE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002474PyDoc_STRVAR(posix_nice__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002475"nice(inc) -> new_priority\n\n\
2476Decrease the priority of process by inc and return the new priority.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002477
Barry Warsaw53699e91996-12-10 23:23:01 +00002478static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002479posix_nice(PyObject *self, PyObject *args)
Guido van Rossum775f4da1993-01-09 17:18:52 +00002480{
2481 int increment, value;
2482
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002483 if (!PyArg_ParseTuple(args, "i:nice", &increment))
Guido van Rossum775f4da1993-01-09 17:18:52 +00002484 return NULL;
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002485
2486 /* There are two flavours of 'nice': one that returns the new
2487 priority (as required by almost all standards out there) and the
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002488 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
2489 the use of getpriority() to get the new priority.
Tim Peters5aa91602002-01-30 05:46:57 +00002490
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002491 If we are of the nice family that returns the new priority, we
2492 need to clear errno before the call, and check if errno is filled
2493 before calling posix_error() on a returnvalue of -1, because the
2494 -1 may be the actual new priority! */
2495
2496 errno = 0;
Guido van Rossum775f4da1993-01-09 17:18:52 +00002497 value = nice(increment);
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002498#if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002499 if (value == 0)
2500 value = getpriority(PRIO_PROCESS, 0);
2501#endif
2502 if (value == -1 && errno != 0)
2503 /* either nice() or getpriority() returned an error */
Guido van Rossum775f4da1993-01-09 17:18:52 +00002504 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00002505 return PyInt_FromLong((long) value);
Guido van Rossum775f4da1993-01-09 17:18:52 +00002506}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002507#endif /* HAVE_NICE */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002508
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002509PyDoc_STRVAR(posix_rename__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002510"rename(old, new)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002511Rename a file or directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002512
Barry Warsaw53699e91996-12-10 23:23:01 +00002513static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002514posix_rename(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002515{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002516#ifdef MS_WINDOWS
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00002517 PyObject *o1, *o2;
2518 char *p1, *p2;
2519 BOOL result;
2520 if (unicode_file_names()) {
2521 if (!PyArg_ParseTuple(args, "O&O&:rename",
2522 convert_to_unicode, &o1,
2523 convert_to_unicode, &o2))
2524 PyErr_Clear();
2525 else {
2526 Py_BEGIN_ALLOW_THREADS
2527 result = MoveFileW(PyUnicode_AsUnicode(o1),
2528 PyUnicode_AsUnicode(o2));
2529 Py_END_ALLOW_THREADS
2530 Py_DECREF(o1);
2531 Py_DECREF(o2);
2532 if (!result)
2533 return win32_error("rename", NULL);
2534 Py_INCREF(Py_None);
2535 return Py_None;
2536 }
2537 }
2538 if (!PyArg_ParseTuple(args, "ss:rename", &p1, &p2))
2539 return NULL;
2540 Py_BEGIN_ALLOW_THREADS
2541 result = MoveFileA(p1, p2);
2542 Py_END_ALLOW_THREADS
2543 if (!result)
2544 return win32_error("rename", NULL);
2545 Py_INCREF(Py_None);
2546 return Py_None;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002547#else
Martin v. Löwis4fc2bda2006-05-04 12:04:27 +00002548 return posix_2str(args, "etet:rename", rename);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002549#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002550}
2551
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002552
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002553PyDoc_STRVAR(posix_rmdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002554"rmdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002555Remove a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002556
Barry Warsaw53699e91996-12-10 23:23:01 +00002557static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002558posix_rmdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002559{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002560#ifdef MS_WINDOWS
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00002561 return win32_1str(args, "rmdir", "s:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002562#else
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00002563 return posix_1str(args, "et:rmdir", rmdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002564#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002565}
2566
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002567
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002568PyDoc_STRVAR(posix_stat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002569"stat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002570Perform a stat system call on the given path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002571
Barry Warsaw53699e91996-12-10 23:23:01 +00002572static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002573posix_stat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002574{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002575#ifdef MS_WINDOWS
Martin v. Löwis14694662006-02-03 12:54:16 +00002576 return posix_do_stat(self, args, "et:stat", STAT, "U:stat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002577#else
2578 return posix_do_stat(self, args, "et:stat", STAT, NULL, NULL);
2579#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002580}
2581
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002582
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002583#ifdef HAVE_SYSTEM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002584PyDoc_STRVAR(posix_system__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002585"system(command) -> exit_status\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002586Execute the command (a string) in a subshell.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002587
Barry Warsaw53699e91996-12-10 23:23:01 +00002588static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002589posix_system(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002590{
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002591 char *command;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002592 long sts;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002593 if (!PyArg_ParseTuple(args, "s:system", &command))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002594 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002595 Py_BEGIN_ALLOW_THREADS
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002596 sts = system(command);
Barry Warsaw53699e91996-12-10 23:23:01 +00002597 Py_END_ALLOW_THREADS
2598 return PyInt_FromLong(sts);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002599}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002600#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002601
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002602
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002603PyDoc_STRVAR(posix_umask__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002604"umask(new_mask) -> old_mask\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002605Set the current numeric umask and return the previous umask.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002606
Barry Warsaw53699e91996-12-10 23:23:01 +00002607static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002608posix_umask(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002609{
2610 int i;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002611 if (!PyArg_ParseTuple(args, "i:umask", &i))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002612 return NULL;
Fred Drake0368bc42001-07-19 20:48:32 +00002613 i = (int)umask(i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002614 if (i < 0)
2615 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00002616 return PyInt_FromLong((long)i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002617}
2618
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002619
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002620PyDoc_STRVAR(posix_unlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002621"unlink(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002622Remove a file (same as remove(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002623
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002624PyDoc_STRVAR(posix_remove__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002625"remove(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002626Remove a file (same as unlink(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002627
Barry Warsaw53699e91996-12-10 23:23:01 +00002628static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002629posix_unlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002630{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002631#ifdef MS_WINDOWS
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00002632 return win32_1str(args, "remove", "s:remove", DeleteFileA, "U:remove", DeleteFileW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002633#else
Martin v. Löwis8e0d4942006-05-04 10:08:42 +00002634 return posix_1str(args, "et:remove", unlink);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002635#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002636}
2637
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002638
Guido van Rossumb6775db1994-08-01 11:34:53 +00002639#ifdef HAVE_UNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002640PyDoc_STRVAR(posix_uname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002641"uname() -> (sysname, nodename, release, version, machine)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002642Return a tuple identifying the current operating system.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002643
Barry Warsaw53699e91996-12-10 23:23:01 +00002644static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002645posix_uname(PyObject *self, PyObject *noargs)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002646{
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002647 struct utsname u;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002648 int res;
Neal Norwitze241ce82003-02-17 18:17:05 +00002649
Barry Warsaw53699e91996-12-10 23:23:01 +00002650 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002651 res = uname(&u);
Barry Warsaw53699e91996-12-10 23:23:01 +00002652 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002653 if (res < 0)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002654 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00002655 return Py_BuildValue("(sssss)",
Barry Warsaw43d68b81996-12-19 22:10:44 +00002656 u.sysname,
2657 u.nodename,
2658 u.release,
2659 u.version,
2660 u.machine);
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002661}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002662#endif /* HAVE_UNAME */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002663
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002664static int
2665extract_time(PyObject *t, long* sec, long* usec)
2666{
2667 long intval;
2668 if (PyFloat_Check(t)) {
2669 double tval = PyFloat_AsDouble(t);
Martin v. Löwis68192102007-07-21 06:55:02 +00002670 PyObject *intobj = Py_Type(t)->tp_as_number->nb_int(t);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002671 if (!intobj)
2672 return -1;
2673 intval = PyInt_AsLong(intobj);
2674 Py_DECREF(intobj);
Michael W. Hudsonb8963812005-07-05 15:21:58 +00002675 if (intval == -1 && PyErr_Occurred())
2676 return -1;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002677 *sec = intval;
Tim Peters96940cf2002-09-10 15:37:28 +00002678 *usec = (long)((tval - intval) * 1e6); /* can't exceed 1000000 */
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002679 if (*usec < 0)
2680 /* If rounding gave us a negative number,
2681 truncate. */
2682 *usec = 0;
2683 return 0;
2684 }
2685 intval = PyInt_AsLong(t);
2686 if (intval == -1 && PyErr_Occurred())
2687 return -1;
2688 *sec = intval;
2689 *usec = 0;
Martin v. Löwis076b2092002-09-10 15:04:41 +00002690 return 0;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002691}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002692
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002693PyDoc_STRVAR(posix_utime__doc__,
Georg Brandl9e5b5e42006-05-17 14:18:20 +00002694"utime(path, (atime, mtime))\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00002695utime(path, None)\n\n\
Barry Warsaw3cef8562000-05-01 16:17:24 +00002696Set the access and modified time of the file to the given values. If the\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002697second form is used, set the access and modified times to the current time.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002698
Barry Warsaw53699e91996-12-10 23:23:01 +00002699static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002700posix_utime(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002701{
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002702#ifdef Py_WIN_WIDE_FILENAMES
2703 PyObject *arg;
2704 PyUnicodeObject *obwpath;
2705 wchar_t *wpath = NULL;
2706 char *apath = NULL;
2707 HANDLE hFile;
2708 long atimesec, mtimesec, ausec, musec;
2709 FILETIME atime, mtime;
2710 PyObject *result = NULL;
2711
2712 if (unicode_file_names()) {
2713 if (PyArg_ParseTuple(args, "UO|:utime", &obwpath, &arg)) {
2714 wpath = PyUnicode_AS_UNICODE(obwpath);
2715 Py_BEGIN_ALLOW_THREADS
2716 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
Martin v. Löwis18aaa562006-10-15 08:43:33 +00002717 NULL, OPEN_EXISTING,
2718 FILE_FLAG_BACKUP_SEMANTICS, NULL);
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002719 Py_END_ALLOW_THREADS
2720 if (hFile == INVALID_HANDLE_VALUE)
2721 return win32_error_unicode("utime", wpath);
2722 } else
2723 /* Drop the argument parsing error as narrow strings
2724 are also valid. */
2725 PyErr_Clear();
2726 }
2727 if (!wpath) {
2728 if (!PyArg_ParseTuple(args, "etO:utime",
2729 Py_FileSystemDefaultEncoding, &apath, &arg))
2730 return NULL;
2731 Py_BEGIN_ALLOW_THREADS
2732 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
Martin v. Löwis18aaa562006-10-15 08:43:33 +00002733 NULL, OPEN_EXISTING,
2734 FILE_FLAG_BACKUP_SEMANTICS, NULL);
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002735 Py_END_ALLOW_THREADS
2736 if (hFile == INVALID_HANDLE_VALUE) {
2737 win32_error("utime", apath);
2738 PyMem_Free(apath);
2739 return NULL;
2740 }
2741 PyMem_Free(apath);
2742 }
2743
2744 if (arg == Py_None) {
2745 SYSTEMTIME now;
2746 GetSystemTime(&now);
2747 if (!SystemTimeToFileTime(&now, &mtime) ||
2748 !SystemTimeToFileTime(&now, &atime)) {
2749 win32_error("utime", NULL);
2750 goto done;
2751 }
2752 }
2753 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
2754 PyErr_SetString(PyExc_TypeError,
2755 "utime() arg 2 must be a tuple (atime, mtime)");
2756 goto done;
2757 }
2758 else {
2759 if (extract_time(PyTuple_GET_ITEM(arg, 0),
2760 &atimesec, &ausec) == -1)
2761 goto done;
Martin v. Löwisf43893a2006-10-09 20:44:25 +00002762 time_t_to_FILE_TIME(atimesec, 1000*ausec, &atime);
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002763 if (extract_time(PyTuple_GET_ITEM(arg, 1),
2764 &mtimesec, &musec) == -1)
2765 goto done;
Martin v. Löwisf43893a2006-10-09 20:44:25 +00002766 time_t_to_FILE_TIME(mtimesec, 1000*musec, &mtime);
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002767 }
2768 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
2769 /* Avoid putting the file name into the error here,
2770 as that may confuse the user into believing that
2771 something is wrong with the file, when it also
2772 could be the time stamp that gives a problem. */
2773 win32_error("utime", NULL);
2774 }
2775 Py_INCREF(Py_None);
2776 result = Py_None;
2777done:
2778 CloseHandle(hFile);
2779 return result;
2780#else /* Py_WIN_WIDE_FILENAMES */
2781
Neal Norwitz2adf2102004-06-09 01:46:02 +00002782 char *path = NULL;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002783 long atime, mtime, ausec, musec;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002784 int res;
Barry Warsaw3cef8562000-05-01 16:17:24 +00002785 PyObject* arg;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002786
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002787#if defined(HAVE_UTIMES)
2788 struct timeval buf[2];
2789#define ATIME buf[0].tv_sec
2790#define MTIME buf[1].tv_sec
2791#elif defined(HAVE_UTIME_H)
Guido van Rossum6d8841c1997-08-14 19:57:39 +00002792/* XXX should define struct utimbuf instead, above */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002793 struct utimbuf buf;
2794#define ATIME buf.actime
2795#define MTIME buf.modtime
2796#define UTIME_ARG &buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002797#else /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002798 time_t buf[2];
2799#define ATIME buf[0]
2800#define MTIME buf[1]
2801#define UTIME_ARG buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002802#endif /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002803
Mark Hammond817c9292003-12-03 01:22:38 +00002804
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002805 if (!PyArg_ParseTuple(args, "etO:utime",
Hye-Shik Chang2b2c9732004-01-04 13:54:25 +00002806 Py_FileSystemDefaultEncoding, &path, &arg))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002807 return NULL;
Barry Warsaw3cef8562000-05-01 16:17:24 +00002808 if (arg == Py_None) {
2809 /* optional time values not given */
2810 Py_BEGIN_ALLOW_THREADS
2811 res = utime(path, NULL);
2812 Py_END_ALLOW_THREADS
2813 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002814 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
Barry Warsaw3cef8562000-05-01 16:17:24 +00002815 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00002816 "utime() arg 2 must be a tuple (atime, mtime)");
Neal Norwitz96652712004-06-06 20:40:27 +00002817 PyMem_Free(path);
Barry Warsaw3cef8562000-05-01 16:17:24 +00002818 return NULL;
2819 }
2820 else {
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002821 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Neal Norwitz96652712004-06-06 20:40:27 +00002822 &atime, &ausec) == -1) {
2823 PyMem_Free(path);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002824 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00002825 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002826 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Neal Norwitz96652712004-06-06 20:40:27 +00002827 &mtime, &musec) == -1) {
2828 PyMem_Free(path);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002829 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00002830 }
Barry Warsaw3cef8562000-05-01 16:17:24 +00002831 ATIME = atime;
2832 MTIME = mtime;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002833#ifdef HAVE_UTIMES
2834 buf[0].tv_usec = ausec;
2835 buf[1].tv_usec = musec;
2836 Py_BEGIN_ALLOW_THREADS
2837 res = utimes(path, buf);
2838 Py_END_ALLOW_THREADS
2839#else
Barry Warsaw3cef8562000-05-01 16:17:24 +00002840 Py_BEGIN_ALLOW_THREADS
2841 res = utime(path, UTIME_ARG);
2842 Py_END_ALLOW_THREADS
Mark Hammond817c9292003-12-03 01:22:38 +00002843#endif /* HAVE_UTIMES */
Barry Warsaw3cef8562000-05-01 16:17:24 +00002844 }
Mark Hammond2d5914b2004-05-04 08:10:37 +00002845 if (res < 0) {
Neal Norwitz96652712004-06-06 20:40:27 +00002846 return posix_error_with_allocated_filename(path);
Mark Hammond2d5914b2004-05-04 08:10:37 +00002847 }
Neal Norwitz96652712004-06-06 20:40:27 +00002848 PyMem_Free(path);
Barry Warsaw53699e91996-12-10 23:23:01 +00002849 Py_INCREF(Py_None);
2850 return Py_None;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002851#undef UTIME_ARG
2852#undef ATIME
2853#undef MTIME
Martin v. Löwisd4e3bb32006-05-06 16:32:54 +00002854#endif /* Py_WIN_WIDE_FILENAMES */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002855}
2856
Guido van Rossum85e3b011991-06-03 12:42:10 +00002857
Guido van Rossum3b066191991-06-04 19:40:25 +00002858/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002859
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002860PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002861"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002862Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002863
Barry Warsaw53699e91996-12-10 23:23:01 +00002864static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002865posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002866{
2867 int sts;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002868 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
Guido van Rossum85e3b011991-06-03 12:42:10 +00002869 return NULL;
2870 _exit(sts);
Guido van Rossuma376cc51996-12-05 23:43:35 +00002871 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002872}
2873
Martin v. Löwis114619e2002-10-07 06:44:21 +00002874#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
2875static void
Martin v. Löwis725507b2006-03-07 12:08:51 +00002876free_string_array(char **array, Py_ssize_t count)
Martin v. Löwis114619e2002-10-07 06:44:21 +00002877{
Martin v. Löwis725507b2006-03-07 12:08:51 +00002878 Py_ssize_t i;
Martin v. Löwis114619e2002-10-07 06:44:21 +00002879 for (i = 0; i < count; i++)
2880 PyMem_Free(array[i]);
2881 PyMem_DEL(array);
2882}
2883#endif
2884
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002885
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002886#ifdef HAVE_EXECV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002887PyDoc_STRVAR(posix_execv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002888"execv(path, args)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002889Execute an executable path with arguments, replacing current process.\n\
2890\n\
2891 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002892 args: tuple or list of strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002893
Barry Warsaw53699e91996-12-10 23:23:01 +00002894static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002895posix_execv(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002896{
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002897 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00002898 PyObject *argv;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002899 char **argvlist;
Martin v. Löwis725507b2006-03-07 12:08:51 +00002900 Py_ssize_t i, argc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002901 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossum85e3b011991-06-03 12:42:10 +00002902
Guido van Rossum89b33251993-10-22 14:26:06 +00002903 /* execv has two arguments: (path, argv), where
Guido van Rossum85e3b011991-06-03 12:42:10 +00002904 argv is a list or tuple of strings. */
2905
Martin v. Löwis114619e2002-10-07 06:44:21 +00002906 if (!PyArg_ParseTuple(args, "etO:execv",
2907 Py_FileSystemDefaultEncoding,
2908 &path, &argv))
Guido van Rossum85e3b011991-06-03 12:42:10 +00002909 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002910 if (PyList_Check(argv)) {
2911 argc = PyList_Size(argv);
2912 getitem = PyList_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002913 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002914 else if (PyTuple_Check(argv)) {
2915 argc = PyTuple_Size(argv);
2916 getitem = PyTuple_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002917 }
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002918 else {
Fred Drake661ea262000-10-24 19:57:45 +00002919 PyErr_SetString(PyExc_TypeError, "execv() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002920 PyMem_Free(path);
Guido van Rossum50422b42000-04-26 20:34:28 +00002921 return NULL;
2922 }
2923
Barry Warsaw53699e91996-12-10 23:23:01 +00002924 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00002925 if (argvlist == NULL) {
2926 PyMem_Free(path);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00002927 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00002928 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00002929 for (i = 0; i < argc; i++) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00002930 if (!PyArg_Parse((*getitem)(argv, i), "et",
2931 Py_FileSystemDefaultEncoding,
2932 &argvlist[i])) {
2933 free_string_array(argvlist, i);
Tim Peters5aa91602002-01-30 05:46:57 +00002934 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00002935 "execv() arg 2 must contain only strings");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002936 PyMem_Free(path);
Guido van Rossum50422b42000-04-26 20:34:28 +00002937 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00002938
Guido van Rossum85e3b011991-06-03 12:42:10 +00002939 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00002940 }
2941 argvlist[argc] = NULL;
2942
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002943 execv(path, argvlist);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002944
Guido van Rossum85e3b011991-06-03 12:42:10 +00002945 /* If we get here it's definitely an error */
2946
Martin v. Löwis114619e2002-10-07 06:44:21 +00002947 free_string_array(argvlist, argc);
2948 PyMem_Free(path);
Guido van Rossum85e3b011991-06-03 12:42:10 +00002949 return posix_error();
2950}
2951
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002952
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002953PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002954"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002955Execute a path with arguments and environment, replacing current process.\n\
2956\n\
2957 path: path of executable file\n\
2958 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002959 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002960
Barry Warsaw53699e91996-12-10 23:23:01 +00002961static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002962posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002963{
2964 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00002965 PyObject *argv, *env;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002966 char **argvlist;
2967 char **envlist;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00002968 PyObject *key, *val, *keys=NULL, *vals=NULL;
Martin v. Löwis725507b2006-03-07 12:08:51 +00002969 Py_ssize_t i, pos, argc, envc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002970 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Martin v. Löwis26fd9602006-04-22 11:15:41 +00002971 Py_ssize_t lastarg = 0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002972
2973 /* execve has three arguments: (path, argv, env), where
2974 argv is a list or tuple of strings and env is a dictionary
2975 like posix.environ. */
2976
Martin v. Löwis114619e2002-10-07 06:44:21 +00002977 if (!PyArg_ParseTuple(args, "etOO:execve",
2978 Py_FileSystemDefaultEncoding,
2979 &path, &argv, &env))
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002980 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002981 if (PyList_Check(argv)) {
2982 argc = PyList_Size(argv);
2983 getitem = PyList_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002984 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002985 else if (PyTuple_Check(argv)) {
2986 argc = PyTuple_Size(argv);
2987 getitem = PyTuple_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002988 }
2989 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002990 PyErr_SetString(PyExc_TypeError,
2991 "execve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002992 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002993 }
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00002994 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002995 PyErr_SetString(PyExc_TypeError,
2996 "execve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002997 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002998 }
2999
Barry Warsaw53699e91996-12-10 23:23:01 +00003000 argvlist = PyMem_NEW(char *, argc+1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003001 if (argvlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003002 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003003 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003004 }
3005 for (i = 0; i < argc; i++) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003006 if (!PyArg_Parse((*getitem)(argv, i),
Martin v. Löwis114619e2002-10-07 06:44:21 +00003007 "et;execve() arg 2 must contain only strings",
Guido van Rossum1e700d22002-10-16 16:52:11 +00003008 Py_FileSystemDefaultEncoding,
Barry Warsaw43d68b81996-12-19 22:10:44 +00003009 &argvlist[i]))
3010 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003011 lastarg = i;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003012 goto fail_1;
3013 }
3014 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003015 lastarg = argc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003016 argvlist[argc] = NULL;
3017
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003018 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003019 if (i < 0)
3020 goto fail_1;
Barry Warsaw53699e91996-12-10 23:23:01 +00003021 envlist = PyMem_NEW(char *, i + 1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003022 if (envlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003023 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003024 goto fail_1;
3025 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003026 envc = 0;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003027 keys = PyMapping_Keys(env);
3028 vals = PyMapping_Values(env);
3029 if (!keys || !vals)
3030 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003031 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3032 PyErr_SetString(PyExc_TypeError,
3033 "execve(): env.keys() or env.values() is not a list");
3034 goto fail_2;
3035 }
Tim Peters5aa91602002-01-30 05:46:57 +00003036
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003037 for (pos = 0; pos < i; pos++) {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003038 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003039 size_t len;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003040
3041 key = PyList_GetItem(keys, pos);
3042 val = PyList_GetItem(vals, pos);
3043 if (!key || !val)
3044 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003045
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003046 if (!PyArg_Parse(
3047 key,
3048 "s;execve() arg 3 contains a non-string key",
3049 &k) ||
3050 !PyArg_Parse(
3051 val,
3052 "s;execve() arg 3 contains a non-string value",
3053 &v))
Barry Warsaw43d68b81996-12-19 22:10:44 +00003054 {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003055 goto fail_2;
3056 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00003057
3058#if defined(PYOS_OS2)
3059 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
3060 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
3061#endif
Tim Petersc8996f52001-12-03 20:41:00 +00003062 len = PyString_Size(key) + PyString_Size(val) + 2;
3063 p = PyMem_NEW(char, len);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003064 if (p == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003065 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003066 goto fail_2;
3067 }
Tim Petersc8996f52001-12-03 20:41:00 +00003068 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003069 envlist[envc++] = p;
Guido van Rossumd48f2521997-12-05 22:19:34 +00003070#if defined(PYOS_OS2)
3071 }
3072#endif
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003073 }
3074 envlist[envc] = 0;
3075
3076 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00003077
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003078 /* If we get here it's definitely an error */
3079
3080 (void) posix_error();
3081
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003082 fail_2:
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003083 while (--envc >= 0)
Barry Warsaw53699e91996-12-10 23:23:01 +00003084 PyMem_DEL(envlist[envc]);
3085 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003086 fail_1:
3087 free_string_array(argvlist, lastarg);
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003088 Py_XDECREF(vals);
3089 Py_XDECREF(keys);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003090 fail_0:
Martin v. Löwis114619e2002-10-07 06:44:21 +00003091 PyMem_Free(path);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003092 return NULL;
3093}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003094#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003095
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003096
Guido van Rossuma1065681999-01-25 23:20:23 +00003097#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003098PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003099"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003100Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003101\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003102 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003103 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003104 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003105
3106static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003107posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003108{
3109 char *path;
3110 PyObject *argv;
3111 char **argvlist;
Martin v. Löwis26fd9602006-04-22 11:15:41 +00003112 int mode, i;
3113 Py_ssize_t argc;
Tim Peters79248aa2001-08-29 21:37:10 +00003114 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003115 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossuma1065681999-01-25 23:20:23 +00003116
3117 /* spawnv has three arguments: (mode, path, argv), where
3118 argv is a list or tuple of strings. */
3119
Martin v. Löwis114619e2002-10-07 06:44:21 +00003120 if (!PyArg_ParseTuple(args, "ietO:spawnv", &mode,
3121 Py_FileSystemDefaultEncoding,
3122 &path, &argv))
Guido van Rossuma1065681999-01-25 23:20:23 +00003123 return NULL;
3124 if (PyList_Check(argv)) {
3125 argc = PyList_Size(argv);
3126 getitem = PyList_GetItem;
3127 }
3128 else if (PyTuple_Check(argv)) {
3129 argc = PyTuple_Size(argv);
3130 getitem = PyTuple_GetItem;
3131 }
3132 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003133 PyErr_SetString(PyExc_TypeError,
3134 "spawnv() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003135 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00003136 return NULL;
3137 }
3138
3139 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003140 if (argvlist == NULL) {
3141 PyMem_Free(path);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00003142 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003143 }
Guido van Rossuma1065681999-01-25 23:20:23 +00003144 for (i = 0; i < argc; i++) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003145 if (!PyArg_Parse((*getitem)(argv, i), "et",
3146 Py_FileSystemDefaultEncoding,
3147 &argvlist[i])) {
3148 free_string_array(argvlist, i);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003149 PyErr_SetString(
3150 PyExc_TypeError,
3151 "spawnv() arg 2 must contain only strings");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003152 PyMem_Free(path);
Fred Drake137507e2000-06-01 02:02:46 +00003153 return NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00003154 }
3155 }
3156 argvlist[argc] = NULL;
3157
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003158#if defined(PYOS_OS2) && defined(PYCC_GCC)
3159 Py_BEGIN_ALLOW_THREADS
3160 spawnval = spawnv(mode, path, argvlist);
3161 Py_END_ALLOW_THREADS
3162#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003163 if (mode == _OLD_P_OVERLAY)
3164 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00003165
Tim Peters25059d32001-12-07 20:35:43 +00003166 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003167 spawnval = _spawnv(mode, path, argvlist);
Tim Peters25059d32001-12-07 20:35:43 +00003168 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003169#endif
Tim Peters5aa91602002-01-30 05:46:57 +00003170
Martin v. Löwis114619e2002-10-07 06:44:21 +00003171 free_string_array(argvlist, argc);
3172 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00003173
Fred Drake699f3522000-06-29 21:12:41 +00003174 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003175 return posix_error();
3176 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003177#if SIZEOF_LONG == SIZEOF_VOID_P
3178 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003179#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003180 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003181#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003182}
3183
3184
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003185PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003186"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003187Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003188\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003189 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003190 path: path of executable file\n\
3191 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003192 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003193
3194static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003195posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003196{
3197 char *path;
3198 PyObject *argv, *env;
3199 char **argvlist;
3200 char **envlist;
3201 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
Martin v. Löwis26fd9602006-04-22 11:15:41 +00003202 int mode, pos, envc;
3203 Py_ssize_t argc, i;
Tim Peters79248aa2001-08-29 21:37:10 +00003204 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003205 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Martin v. Löwis26fd9602006-04-22 11:15:41 +00003206 Py_ssize_t lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003207
3208 /* spawnve has four arguments: (mode, path, argv, env), where
3209 argv is a list or tuple of strings and env is a dictionary
3210 like posix.environ. */
3211
Martin v. Löwis114619e2002-10-07 06:44:21 +00003212 if (!PyArg_ParseTuple(args, "ietOO:spawnve", &mode,
3213 Py_FileSystemDefaultEncoding,
3214 &path, &argv, &env))
Guido van Rossuma1065681999-01-25 23:20:23 +00003215 return NULL;
3216 if (PyList_Check(argv)) {
3217 argc = PyList_Size(argv);
3218 getitem = PyList_GetItem;
3219 }
3220 else if (PyTuple_Check(argv)) {
3221 argc = PyTuple_Size(argv);
3222 getitem = PyTuple_GetItem;
3223 }
3224 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003225 PyErr_SetString(PyExc_TypeError,
3226 "spawnve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003227 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003228 }
3229 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003230 PyErr_SetString(PyExc_TypeError,
3231 "spawnve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003232 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003233 }
3234
3235 argvlist = PyMem_NEW(char *, argc+1);
3236 if (argvlist == NULL) {
3237 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003238 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003239 }
3240 for (i = 0; i < argc; i++) {
3241 if (!PyArg_Parse((*getitem)(argv, i),
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003242 "et;spawnve() arg 2 must contain only strings",
Martin v. Löwis114619e2002-10-07 06:44:21 +00003243 Py_FileSystemDefaultEncoding,
Guido van Rossuma1065681999-01-25 23:20:23 +00003244 &argvlist[i]))
3245 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003246 lastarg = i;
Guido van Rossuma1065681999-01-25 23:20:23 +00003247 goto fail_1;
3248 }
3249 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003250 lastarg = argc;
Guido van Rossuma1065681999-01-25 23:20:23 +00003251 argvlist[argc] = NULL;
3252
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003253 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003254 if (i < 0)
3255 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00003256 envlist = PyMem_NEW(char *, i + 1);
3257 if (envlist == NULL) {
3258 PyErr_NoMemory();
3259 goto fail_1;
3260 }
3261 envc = 0;
3262 keys = PyMapping_Keys(env);
3263 vals = PyMapping_Values(env);
3264 if (!keys || !vals)
3265 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003266 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3267 PyErr_SetString(PyExc_TypeError,
3268 "spawnve(): env.keys() or env.values() is not a list");
3269 goto fail_2;
3270 }
Tim Peters5aa91602002-01-30 05:46:57 +00003271
Guido van Rossuma1065681999-01-25 23:20:23 +00003272 for (pos = 0; pos < i; pos++) {
3273 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003274 size_t len;
Guido van Rossuma1065681999-01-25 23:20:23 +00003275
3276 key = PyList_GetItem(keys, pos);
3277 val = PyList_GetItem(vals, pos);
3278 if (!key || !val)
3279 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003280
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003281 if (!PyArg_Parse(
3282 key,
3283 "s;spawnve() arg 3 contains a non-string key",
3284 &k) ||
3285 !PyArg_Parse(
3286 val,
3287 "s;spawnve() arg 3 contains a non-string value",
3288 &v))
Guido van Rossuma1065681999-01-25 23:20:23 +00003289 {
3290 goto fail_2;
3291 }
Tim Petersc8996f52001-12-03 20:41:00 +00003292 len = PyString_Size(key) + PyString_Size(val) + 2;
3293 p = PyMem_NEW(char, len);
Guido van Rossuma1065681999-01-25 23:20:23 +00003294 if (p == NULL) {
3295 PyErr_NoMemory();
3296 goto fail_2;
3297 }
Tim Petersc8996f52001-12-03 20:41:00 +00003298 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossuma1065681999-01-25 23:20:23 +00003299 envlist[envc++] = p;
3300 }
3301 envlist[envc] = 0;
3302
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003303#if defined(PYOS_OS2) && defined(PYCC_GCC)
3304 Py_BEGIN_ALLOW_THREADS
3305 spawnval = spawnve(mode, path, argvlist, envlist);
3306 Py_END_ALLOW_THREADS
3307#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003308 if (mode == _OLD_P_OVERLAY)
3309 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00003310
3311 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003312 spawnval = _spawnve(mode, path, argvlist, envlist);
Tim Peters25059d32001-12-07 20:35:43 +00003313 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003314#endif
Tim Peters25059d32001-12-07 20:35:43 +00003315
Fred Drake699f3522000-06-29 21:12:41 +00003316 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003317 (void) posix_error();
3318 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003319#if SIZEOF_LONG == SIZEOF_VOID_P
3320 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003321#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003322 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003323#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003324
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003325 fail_2:
Guido van Rossuma1065681999-01-25 23:20:23 +00003326 while (--envc >= 0)
3327 PyMem_DEL(envlist[envc]);
3328 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003329 fail_1:
Martin v. Löwis114619e2002-10-07 06:44:21 +00003330 free_string_array(argvlist, lastarg);
Guido van Rossuma1065681999-01-25 23:20:23 +00003331 Py_XDECREF(vals);
3332 Py_XDECREF(keys);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003333 fail_0:
3334 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00003335 return res;
3336}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003337
3338/* OS/2 supports spawnvp & spawnvpe natively */
3339#if defined(PYOS_OS2)
3340PyDoc_STRVAR(posix_spawnvp__doc__,
3341"spawnvp(mode, file, args)\n\n\
3342Execute the program 'file' in a new process, using the environment\n\
3343search path to find the file.\n\
3344\n\
3345 mode: mode of process creation\n\
3346 file: executable file name\n\
3347 args: tuple or list of strings");
3348
3349static PyObject *
3350posix_spawnvp(PyObject *self, PyObject *args)
3351{
3352 char *path;
3353 PyObject *argv;
3354 char **argvlist;
3355 int mode, i, argc;
3356 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003357 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003358
3359 /* spawnvp has three arguments: (mode, path, argv), where
3360 argv is a list or tuple of strings. */
3361
3362 if (!PyArg_ParseTuple(args, "ietO:spawnvp", &mode,
3363 Py_FileSystemDefaultEncoding,
3364 &path, &argv))
3365 return NULL;
3366 if (PyList_Check(argv)) {
3367 argc = PyList_Size(argv);
3368 getitem = PyList_GetItem;
3369 }
3370 else if (PyTuple_Check(argv)) {
3371 argc = PyTuple_Size(argv);
3372 getitem = PyTuple_GetItem;
3373 }
3374 else {
3375 PyErr_SetString(PyExc_TypeError,
3376 "spawnvp() arg 2 must be a tuple or list");
3377 PyMem_Free(path);
3378 return NULL;
3379 }
3380
3381 argvlist = PyMem_NEW(char *, argc+1);
3382 if (argvlist == NULL) {
3383 PyMem_Free(path);
3384 return PyErr_NoMemory();
3385 }
3386 for (i = 0; i < argc; i++) {
3387 if (!PyArg_Parse((*getitem)(argv, i), "et",
3388 Py_FileSystemDefaultEncoding,
3389 &argvlist[i])) {
3390 free_string_array(argvlist, i);
3391 PyErr_SetString(
3392 PyExc_TypeError,
3393 "spawnvp() arg 2 must contain only strings");
3394 PyMem_Free(path);
3395 return NULL;
3396 }
3397 }
3398 argvlist[argc] = NULL;
3399
3400 Py_BEGIN_ALLOW_THREADS
3401#if defined(PYCC_GCC)
3402 spawnval = spawnvp(mode, path, argvlist);
3403#else
3404 spawnval = _spawnvp(mode, path, argvlist);
3405#endif
3406 Py_END_ALLOW_THREADS
3407
3408 free_string_array(argvlist, argc);
3409 PyMem_Free(path);
3410
3411 if (spawnval == -1)
3412 return posix_error();
3413 else
3414 return Py_BuildValue("l", (long) spawnval);
3415}
3416
3417
3418PyDoc_STRVAR(posix_spawnvpe__doc__,
3419"spawnvpe(mode, file, args, env)\n\n\
3420Execute the program 'file' in a new process, using the environment\n\
3421search path to find the file.\n\
3422\n\
3423 mode: mode of process creation\n\
3424 file: executable file name\n\
3425 args: tuple or list of arguments\n\
3426 env: dictionary of strings mapping to strings");
3427
3428static PyObject *
3429posix_spawnvpe(PyObject *self, PyObject *args)
3430{
3431 char *path;
3432 PyObject *argv, *env;
3433 char **argvlist;
3434 char **envlist;
3435 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
3436 int mode, i, pos, argc, envc;
3437 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003438 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003439 int lastarg = 0;
3440
3441 /* spawnvpe has four arguments: (mode, path, argv, env), where
3442 argv is a list or tuple of strings and env is a dictionary
3443 like posix.environ. */
3444
3445 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
3446 Py_FileSystemDefaultEncoding,
3447 &path, &argv, &env))
3448 return NULL;
3449 if (PyList_Check(argv)) {
3450 argc = PyList_Size(argv);
3451 getitem = PyList_GetItem;
3452 }
3453 else if (PyTuple_Check(argv)) {
3454 argc = PyTuple_Size(argv);
3455 getitem = PyTuple_GetItem;
3456 }
3457 else {
3458 PyErr_SetString(PyExc_TypeError,
3459 "spawnvpe() arg 2 must be a tuple or list");
3460 goto fail_0;
3461 }
3462 if (!PyMapping_Check(env)) {
3463 PyErr_SetString(PyExc_TypeError,
3464 "spawnvpe() arg 3 must be a mapping object");
3465 goto fail_0;
3466 }
3467
3468 argvlist = PyMem_NEW(char *, argc+1);
3469 if (argvlist == NULL) {
3470 PyErr_NoMemory();
3471 goto fail_0;
3472 }
3473 for (i = 0; i < argc; i++) {
3474 if (!PyArg_Parse((*getitem)(argv, i),
3475 "et;spawnvpe() arg 2 must contain only strings",
3476 Py_FileSystemDefaultEncoding,
3477 &argvlist[i]))
3478 {
3479 lastarg = i;
3480 goto fail_1;
3481 }
3482 }
3483 lastarg = argc;
3484 argvlist[argc] = NULL;
3485
3486 i = PyMapping_Size(env);
3487 if (i < 0)
3488 goto fail_1;
3489 envlist = PyMem_NEW(char *, i + 1);
3490 if (envlist == NULL) {
3491 PyErr_NoMemory();
3492 goto fail_1;
3493 }
3494 envc = 0;
3495 keys = PyMapping_Keys(env);
3496 vals = PyMapping_Values(env);
3497 if (!keys || !vals)
3498 goto fail_2;
3499 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3500 PyErr_SetString(PyExc_TypeError,
3501 "spawnvpe(): env.keys() or env.values() is not a list");
3502 goto fail_2;
3503 }
3504
3505 for (pos = 0; pos < i; pos++) {
3506 char *p, *k, *v;
3507 size_t len;
3508
3509 key = PyList_GetItem(keys, pos);
3510 val = PyList_GetItem(vals, pos);
3511 if (!key || !val)
3512 goto fail_2;
3513
3514 if (!PyArg_Parse(
3515 key,
3516 "s;spawnvpe() arg 3 contains a non-string key",
3517 &k) ||
3518 !PyArg_Parse(
3519 val,
3520 "s;spawnvpe() arg 3 contains a non-string value",
3521 &v))
3522 {
3523 goto fail_2;
3524 }
3525 len = PyString_Size(key) + PyString_Size(val) + 2;
3526 p = PyMem_NEW(char, len);
3527 if (p == NULL) {
3528 PyErr_NoMemory();
3529 goto fail_2;
3530 }
3531 PyOS_snprintf(p, len, "%s=%s", k, v);
3532 envlist[envc++] = p;
3533 }
3534 envlist[envc] = 0;
3535
3536 Py_BEGIN_ALLOW_THREADS
3537#if defined(PYCC_GCC)
3538 spawnval = spawnve(mode, path, argvlist, envlist);
3539#else
3540 spawnval = _spawnve(mode, path, argvlist, envlist);
3541#endif
3542 Py_END_ALLOW_THREADS
3543
3544 if (spawnval == -1)
3545 (void) posix_error();
3546 else
3547 res = Py_BuildValue("l", (long) spawnval);
3548
3549 fail_2:
3550 while (--envc >= 0)
3551 PyMem_DEL(envlist[envc]);
3552 PyMem_DEL(envlist);
3553 fail_1:
3554 free_string_array(argvlist, lastarg);
3555 Py_XDECREF(vals);
3556 Py_XDECREF(keys);
3557 fail_0:
3558 PyMem_Free(path);
3559 return res;
3560}
3561#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00003562#endif /* HAVE_SPAWNV */
3563
3564
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003565#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003566PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003567"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003568Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
3569\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003570Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003571
3572static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003573posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003574{
Neal Norwitze241ce82003-02-17 18:17:05 +00003575 int pid = fork1();
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003576 if (pid == -1)
3577 return posix_error();
3578 PyOS_AfterFork();
3579 return PyInt_FromLong((long)pid);
3580}
3581#endif
3582
3583
Guido van Rossumad0ee831995-03-01 10:34:45 +00003584#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003585PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003586"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003587Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003588Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003589
Barry Warsaw53699e91996-12-10 23:23:01 +00003590static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003591posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003592{
Neal Norwitze241ce82003-02-17 18:17:05 +00003593 int pid = fork();
Guido van Rossum85e3b011991-06-03 12:42:10 +00003594 if (pid == -1)
3595 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003596 if (pid == 0)
3597 PyOS_AfterFork();
Barry Warsaw53699e91996-12-10 23:23:01 +00003598 return PyInt_FromLong((long)pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003599}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003600#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003601
Neal Norwitzb59798b2003-03-21 01:43:31 +00003602/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00003603/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
3604#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00003605#define DEV_PTY_FILE "/dev/ptc"
3606#define HAVE_DEV_PTMX
3607#else
3608#define DEV_PTY_FILE "/dev/ptmx"
3609#endif
3610
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003611#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003612#ifdef HAVE_PTY_H
3613#include <pty.h>
3614#else
3615#ifdef HAVE_LIBUTIL_H
3616#include <libutil.h>
Fred Drake8cef4cf2000-06-28 16:40:38 +00003617#endif /* HAVE_LIBUTIL_H */
3618#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00003619#ifdef HAVE_STROPTS_H
3620#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003621#endif
3622#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003623
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003624#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003625PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003626"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003627Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003628
3629static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003630posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003631{
3632 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003633#ifndef HAVE_OPENPTY
3634 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003635#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003636#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003637 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003638#ifdef sun
Neal Norwitz84a98e02006-04-10 07:44:23 +00003639 extern char *ptsname(int fildes);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003640#endif
3641#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00003642
Thomas Wouters70c21a12000-07-14 14:28:33 +00003643#ifdef HAVE_OPENPTY
Fred Drake8cef4cf2000-06-28 16:40:38 +00003644 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
3645 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003646#elif defined(HAVE__GETPTY)
Thomas Wouters70c21a12000-07-14 14:28:33 +00003647 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
3648 if (slave_name == NULL)
3649 return posix_error();
3650
3651 slave_fd = open(slave_name, O_RDWR);
3652 if (slave_fd < 0)
3653 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003654#else
Neal Norwitzb59798b2003-03-21 01:43:31 +00003655 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003656 if (master_fd < 0)
3657 return posix_error();
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003658 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003659 /* change permission of slave */
3660 if (grantpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003661 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003662 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003663 }
3664 /* unlock slave */
3665 if (unlockpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003666 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003667 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003668 }
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003669 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003670 slave_name = ptsname(master_fd); /* get name of slave */
3671 if (slave_name == NULL)
3672 return posix_error();
3673 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
3674 if (slave_fd < 0)
3675 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003676#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003677 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
3678 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00003679#ifndef __hpux
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003680 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00003681#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003682#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00003683#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00003684
Fred Drake8cef4cf2000-06-28 16:40:38 +00003685 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00003686
Fred Drake8cef4cf2000-06-28 16:40:38 +00003687}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003688#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003689
3690#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003691PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003692"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00003693Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
3694Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003695To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003696
3697static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003698posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003699{
Neal Norwitz24b3c222005-09-19 06:43:44 +00003700 int master_fd = -1, pid;
Tim Peters5aa91602002-01-30 05:46:57 +00003701
Fred Drake8cef4cf2000-06-28 16:40:38 +00003702 pid = forkpty(&master_fd, NULL, NULL, NULL);
3703 if (pid == -1)
3704 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003705 if (pid == 0)
3706 PyOS_AfterFork();
Fred Drake8cef4cf2000-06-28 16:40:38 +00003707 return Py_BuildValue("(ii)", pid, master_fd);
3708}
3709#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003710
Guido van Rossumad0ee831995-03-01 10:34:45 +00003711#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003712PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003713"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003714Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003715
Barry Warsaw53699e91996-12-10 23:23:01 +00003716static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003717posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003718{
Barry Warsaw53699e91996-12-10 23:23:01 +00003719 return PyInt_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003720}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003721#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003722
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003723
Guido van Rossumad0ee831995-03-01 10:34:45 +00003724#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003725PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003726"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003727Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003728
Barry Warsaw53699e91996-12-10 23:23:01 +00003729static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003730posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003731{
Barry Warsaw53699e91996-12-10 23:23:01 +00003732 return PyInt_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003733}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003734#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003735
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003736
Guido van Rossumad0ee831995-03-01 10:34:45 +00003737#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003738PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003739"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003740Return the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003741
Barry Warsaw53699e91996-12-10 23:23:01 +00003742static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003743posix_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003744{
Barry Warsaw53699e91996-12-10 23:23:01 +00003745 return PyInt_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003746}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003747#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003748
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003749
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003750PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003751"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003752Return the current process id");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003753
Barry Warsaw53699e91996-12-10 23:23:01 +00003754static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003755posix_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003756{
Barry Warsaw53699e91996-12-10 23:23:01 +00003757 return PyInt_FromLong((long)getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00003758}
3759
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003760
Fred Drakec9680921999-12-13 16:37:25 +00003761#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003762PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003763"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003764Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00003765
3766static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003767posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00003768{
3769 PyObject *result = NULL;
3770
Fred Drakec9680921999-12-13 16:37:25 +00003771#ifdef NGROUPS_MAX
3772#define MAX_GROUPS NGROUPS_MAX
3773#else
3774 /* defined to be 16 on Solaris7, so this should be a small number */
3775#define MAX_GROUPS 64
3776#endif
3777 gid_t grouplist[MAX_GROUPS];
3778 int n;
3779
3780 n = getgroups(MAX_GROUPS, grouplist);
3781 if (n < 0)
3782 posix_error();
3783 else {
3784 result = PyList_New(n);
3785 if (result != NULL) {
Fred Drakec9680921999-12-13 16:37:25 +00003786 int i;
3787 for (i = 0; i < n; ++i) {
Neal Norwitze241ce82003-02-17 18:17:05 +00003788 PyObject *o = PyInt_FromLong((long)grouplist[i]);
Fred Drakec9680921999-12-13 16:37:25 +00003789 if (o == NULL) {
3790 Py_DECREF(result);
3791 result = NULL;
3792 break;
3793 }
3794 PyList_SET_ITEM(result, i, o);
3795 }
3796 }
3797 }
Neal Norwitze241ce82003-02-17 18:17:05 +00003798
Fred Drakec9680921999-12-13 16:37:25 +00003799 return result;
3800}
3801#endif
3802
Martin v. Löwis606edc12002-06-13 21:09:11 +00003803#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00003804PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003805"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00003806Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00003807
3808static PyObject *
3809posix_getpgid(PyObject *self, PyObject *args)
3810{
3811 int pid, pgid;
3812 if (!PyArg_ParseTuple(args, "i:getpgid", &pid))
3813 return NULL;
3814 pgid = getpgid(pid);
3815 if (pgid < 0)
3816 return posix_error();
3817 return PyInt_FromLong((long)pgid);
3818}
3819#endif /* HAVE_GETPGID */
3820
3821
Guido van Rossumb6775db1994-08-01 11:34:53 +00003822#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003823PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003824"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003825Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003826
Barry Warsaw53699e91996-12-10 23:23:01 +00003827static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003828posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00003829{
Guido van Rossumb6775db1994-08-01 11:34:53 +00003830#ifdef GETPGRP_HAVE_ARG
Barry Warsaw53699e91996-12-10 23:23:01 +00003831 return PyInt_FromLong((long)getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003832#else /* GETPGRP_HAVE_ARG */
Barry Warsaw53699e91996-12-10 23:23:01 +00003833 return PyInt_FromLong((long)getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003834#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00003835}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003836#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00003837
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003838
Guido van Rossumb6775db1994-08-01 11:34:53 +00003839#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003840PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003841"setpgrp()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003842Make this process a session leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003843
Barry Warsaw53699e91996-12-10 23:23:01 +00003844static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003845posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00003846{
Guido van Rossum64933891994-10-20 21:56:42 +00003847#ifdef SETPGRP_HAVE_ARG
Guido van Rossumc2670a01992-09-13 20:07:29 +00003848 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00003849#else /* SETPGRP_HAVE_ARG */
Guido van Rossumb6775db1994-08-01 11:34:53 +00003850 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00003851#endif /* SETPGRP_HAVE_ARG */
Guido van Rossum687dd131993-05-17 08:34:16 +00003852 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00003853 Py_INCREF(Py_None);
3854 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00003855}
3856
Guido van Rossumb6775db1994-08-01 11:34:53 +00003857#endif /* HAVE_SETPGRP */
3858
Guido van Rossumad0ee831995-03-01 10:34:45 +00003859#ifdef HAVE_GETPPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003860PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003861"getppid() -> ppid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003862Return the parent's process id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003863
Barry Warsaw53699e91996-12-10 23:23:01 +00003864static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003865posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003866{
Barry Warsaw53699e91996-12-10 23:23:01 +00003867 return PyInt_FromLong((long)getppid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00003868}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003869#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003870
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003871
Fred Drake12c6e2d1999-12-14 21:25:03 +00003872#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003873PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003874"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003875Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00003876
3877static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003878posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00003879{
Neal Norwitze241ce82003-02-17 18:17:05 +00003880 PyObject *result = NULL;
Fred Drakea30680b2000-12-06 21:24:28 +00003881 char *name;
3882 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00003883
Fred Drakea30680b2000-12-06 21:24:28 +00003884 errno = 0;
3885 name = getlogin();
3886 if (name == NULL) {
3887 if (errno)
3888 posix_error();
3889 else
3890 PyErr_SetString(PyExc_OSError,
Fred Drakee63544f2000-12-06 21:45:33 +00003891 "unable to determine login name");
Fred Drakea30680b2000-12-06 21:24:28 +00003892 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00003893 else
3894 result = PyString_FromString(name);
Fred Drakea30680b2000-12-06 21:24:28 +00003895 errno = old_errno;
Neal Norwitze241ce82003-02-17 18:17:05 +00003896
Fred Drake12c6e2d1999-12-14 21:25:03 +00003897 return result;
3898}
3899#endif
3900
Guido van Rossumad0ee831995-03-01 10:34:45 +00003901#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003902PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003903"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003904Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003905
Barry Warsaw53699e91996-12-10 23:23:01 +00003906static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003907posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003908{
Barry Warsaw53699e91996-12-10 23:23:01 +00003909 return PyInt_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003910}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003911#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003912
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003913
Guido van Rossumad0ee831995-03-01 10:34:45 +00003914#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003915PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003916"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003917Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003918
Barry Warsaw53699e91996-12-10 23:23:01 +00003919static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003920posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003921{
3922 int pid, sig;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00003923 if (!PyArg_ParseTuple(args, "ii:kill", &pid, &sig))
Guido van Rossum85e3b011991-06-03 12:42:10 +00003924 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003925#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003926 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
3927 APIRET rc;
3928 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00003929 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003930
3931 } else if (sig == XCPT_SIGNAL_KILLPROC) {
3932 APIRET rc;
3933 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00003934 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003935
3936 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003937 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003938#else
Guido van Rossum85e3b011991-06-03 12:42:10 +00003939 if (kill(pid, sig) == -1)
3940 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003941#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00003942 Py_INCREF(Py_None);
3943 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00003944}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003945#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003946
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00003947#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003948PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003949"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003950Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00003951
3952static PyObject *
3953posix_killpg(PyObject *self, PyObject *args)
3954{
3955 int pgid, sig;
3956 if (!PyArg_ParseTuple(args, "ii:killpg", &pgid, &sig))
3957 return NULL;
3958 if (killpg(pgid, sig) == -1)
3959 return posix_error();
3960 Py_INCREF(Py_None);
3961 return Py_None;
3962}
3963#endif
3964
Guido van Rossumc0125471996-06-28 18:55:32 +00003965#ifdef HAVE_PLOCK
3966
3967#ifdef HAVE_SYS_LOCK_H
3968#include <sys/lock.h>
3969#endif
3970
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003971PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003972"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003973Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003974
Barry Warsaw53699e91996-12-10 23:23:01 +00003975static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003976posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00003977{
3978 int op;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00003979 if (!PyArg_ParseTuple(args, "i:plock", &op))
Guido van Rossumc0125471996-06-28 18:55:32 +00003980 return NULL;
3981 if (plock(op) == -1)
3982 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00003983 Py_INCREF(Py_None);
3984 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00003985}
3986#endif
3987
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003988
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003989#ifdef HAVE_POPEN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003990PyDoc_STRVAR(posix_popen__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003991"popen(command [, mode='r' [, bufsize]]) -> pipe\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003992Open a pipe to/from a command returning a file object.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003993
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003994#if defined(PYOS_OS2)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003995#if defined(PYCC_VACPP)
Guido van Rossumd48f2521997-12-05 22:19:34 +00003996static int
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003997async_system(const char *command)
3998{
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00003999 char errormsg[256], args[1024];
4000 RESULTCODES rcodes;
4001 APIRET rc;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004002
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004003 char *shell = getenv("COMSPEC");
4004 if (!shell)
4005 shell = "cmd";
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004006
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004007 /* avoid overflowing the argument buffer */
4008 if (strlen(shell) + 3 + strlen(command) >= 1024)
4009 return ERROR_NOT_ENOUGH_MEMORY
4010
4011 args[0] = '\0';
4012 strcat(args, shell);
4013 strcat(args, "/c ");
4014 strcat(args, command);
4015
4016 /* execute asynchronously, inheriting the environment */
4017 rc = DosExecPgm(errormsg,
4018 sizeof(errormsg),
4019 EXEC_ASYNC,
4020 args,
4021 NULL,
4022 &rcodes,
4023 shell);
4024 return rc;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004025}
4026
Guido van Rossumd48f2521997-12-05 22:19:34 +00004027static FILE *
4028popen(const char *command, const char *mode, int pipesize, int *err)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004029{
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004030 int oldfd, tgtfd;
4031 HFILE pipeh[2];
4032 APIRET rc;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004033
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004034 /* mode determines which of stdin or stdout is reconnected to
4035 * the pipe to the child
4036 */
4037 if (strchr(mode, 'r') != NULL) {
4038 tgt_fd = 1; /* stdout */
4039 } else if (strchr(mode, 'w')) {
4040 tgt_fd = 0; /* stdin */
4041 } else {
4042 *err = ERROR_INVALID_ACCESS;
4043 return NULL;
4044 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004045
Andrew MacIntyrea3be2582004-12-18 09:51:05 +00004046 /* setup the pipe */
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004047 if ((rc = DosCreatePipe(&pipeh[0], &pipeh[1], pipesize)) != NO_ERROR) {
4048 *err = rc;
4049 return NULL;
4050 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004051
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004052 /* prevent other threads accessing stdio */
4053 DosEnterCritSec();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004054
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004055 /* reconnect stdio and execute child */
4056 oldfd = dup(tgtfd);
4057 close(tgtfd);
4058 if (dup2(pipeh[tgtfd], tgtfd) == 0) {
4059 DosClose(pipeh[tgtfd]);
4060 rc = async_system(command);
4061 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004062
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004063 /* restore stdio */
4064 dup2(oldfd, tgtfd);
4065 close(oldfd);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004066
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004067 /* allow other threads access to stdio */
4068 DosExitCritSec();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004069
Andrew MacIntyrea4a8afb2004-12-12 08:30:51 +00004070 /* if execution of child was successful return file stream */
4071 if (rc == NO_ERROR)
4072 return fdopen(pipeh[1 - tgtfd], mode);
4073 else {
4074 DosClose(pipeh[1 - tgtfd]);
4075 *err = rc;
4076 return NULL;
4077 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004078}
4079
4080static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004081posix_popen(PyObject *self, PyObject *args)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004082{
4083 char *name;
4084 char *mode = "r";
Guido van Rossumd48f2521997-12-05 22:19:34 +00004085 int err, bufsize = -1;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004086 FILE *fp;
4087 PyObject *f;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004088 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004089 return NULL;
4090 Py_BEGIN_ALLOW_THREADS
Guido van Rossumd48f2521997-12-05 22:19:34 +00004091 fp = popen(name, mode, (bufsize > 0) ? bufsize : 4096, &err);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004092 Py_END_ALLOW_THREADS
4093 if (fp == NULL)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004094 return os2_error(err);
4095
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004096 f = PyFile_FromFile(fp, name, mode, fclose);
4097 if (f != NULL)
4098 PyFile_SetBufSize(f, bufsize);
4099 return f;
4100}
4101
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004102#elif defined(PYCC_GCC)
4103
4104/* standard posix version of popen() support */
4105static PyObject *
4106posix_popen(PyObject *self, PyObject *args)
4107{
4108 char *name;
4109 char *mode = "r";
4110 int bufsize = -1;
4111 FILE *fp;
4112 PyObject *f;
4113 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
4114 return NULL;
4115 Py_BEGIN_ALLOW_THREADS
4116 fp = popen(name, mode);
4117 Py_END_ALLOW_THREADS
4118 if (fp == NULL)
4119 return posix_error();
4120 f = PyFile_FromFile(fp, name, mode, pclose);
4121 if (f != NULL)
4122 PyFile_SetBufSize(f, bufsize);
4123 return f;
4124}
4125
4126/* fork() under OS/2 has lots'o'warts
4127 * EMX supports pipe() and spawn*() so we can synthesize popen[234]()
4128 * most of this code is a ripoff of the win32 code, but using the
4129 * capabilities of EMX's C library routines
4130 */
4131
4132/* These tell _PyPopen() whether to return 1, 2, or 3 file objects. */
4133#define POPEN_1 1
4134#define POPEN_2 2
4135#define POPEN_3 3
4136#define POPEN_4 4
4137
4138static PyObject *_PyPopen(char *, int, int, int);
4139static int _PyPclose(FILE *file);
4140
4141/*
4142 * Internal dictionary mapping popen* file pointers to process handles,
4143 * for use when retrieving the process exit code. See _PyPclose() below
4144 * for more information on this dictionary's use.
4145 */
4146static PyObject *_PyPopenProcs = NULL;
4147
4148/* os2emx version of popen2()
4149 *
4150 * The result of this function is a pipe (file) connected to the
4151 * process's stdin, and a pipe connected to the process's stdout.
4152 */
4153
4154static PyObject *
4155os2emx_popen2(PyObject *self, PyObject *args)
4156{
4157 PyObject *f;
4158 int tm=0;
4159
4160 char *cmdstring;
4161 char *mode = "t";
4162 int bufsize = -1;
4163 if (!PyArg_ParseTuple(args, "s|si:popen2", &cmdstring, &mode, &bufsize))
4164 return NULL;
4165
4166 if (*mode == 't')
4167 tm = O_TEXT;
4168 else if (*mode != 'b') {
4169 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
4170 return NULL;
4171 } else
4172 tm = O_BINARY;
4173
4174 f = _PyPopen(cmdstring, tm, POPEN_2, bufsize);
4175
4176 return f;
4177}
4178
4179/*
4180 * Variation on os2emx.popen2
4181 *
4182 * The result of this function is 3 pipes - the process's stdin,
4183 * stdout and stderr
4184 */
4185
4186static PyObject *
4187os2emx_popen3(PyObject *self, PyObject *args)
4188{
4189 PyObject *f;
4190 int tm = 0;
4191
4192 char *cmdstring;
4193 char *mode = "t";
4194 int bufsize = -1;
4195 if (!PyArg_ParseTuple(args, "s|si:popen3", &cmdstring, &mode, &bufsize))
4196 return NULL;
4197
4198 if (*mode == 't')
4199 tm = O_TEXT;
4200 else if (*mode != 'b') {
4201 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
4202 return NULL;
4203 } else
4204 tm = O_BINARY;
4205
4206 f = _PyPopen(cmdstring, tm, POPEN_3, bufsize);
4207
4208 return f;
4209}
4210
4211/*
4212 * Variation on os2emx.popen2
4213 *
Tim Peters11b23062003-04-23 02:39:17 +00004214 * The result of this function is 2 pipes - the processes stdin,
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004215 * and stdout+stderr combined as a single pipe.
4216 */
4217
4218static PyObject *
4219os2emx_popen4(PyObject *self, PyObject *args)
4220{
4221 PyObject *f;
4222 int tm = 0;
4223
4224 char *cmdstring;
4225 char *mode = "t";
4226 int bufsize = -1;
4227 if (!PyArg_ParseTuple(args, "s|si:popen4", &cmdstring, &mode, &bufsize))
4228 return NULL;
4229
4230 if (*mode == 't')
4231 tm = O_TEXT;
4232 else if (*mode != 'b') {
4233 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
4234 return NULL;
4235 } else
4236 tm = O_BINARY;
4237
4238 f = _PyPopen(cmdstring, tm, POPEN_4, bufsize);
4239
4240 return f;
4241}
4242
4243/* a couple of structures for convenient handling of multiple
4244 * file handles and pipes
4245 */
4246struct file_ref
4247{
4248 int handle;
4249 int flags;
4250};
4251
4252struct pipe_ref
4253{
4254 int rd;
4255 int wr;
4256};
4257
4258/* The following code is derived from the win32 code */
4259
4260static PyObject *
4261_PyPopen(char *cmdstring, int mode, int n, int bufsize)
4262{
4263 struct file_ref stdio[3];
4264 struct pipe_ref p_fd[3];
4265 FILE *p_s[3];
4266 int file_count, i, pipe_err, pipe_pid;
4267 char *shell, *sh_name, *opt, *rd_mode, *wr_mode;
4268 PyObject *f, *p_f[3];
4269
4270 /* file modes for subsequent fdopen's on pipe handles */
4271 if (mode == O_TEXT)
4272 {
4273 rd_mode = "rt";
4274 wr_mode = "wt";
4275 }
4276 else
4277 {
4278 rd_mode = "rb";
4279 wr_mode = "wb";
4280 }
4281
4282 /* prepare shell references */
4283 if ((shell = getenv("EMXSHELL")) == NULL)
4284 if ((shell = getenv("COMSPEC")) == NULL)
4285 {
4286 errno = ENOENT;
4287 return posix_error();
4288 }
4289
4290 sh_name = _getname(shell);
4291 if (stricmp(sh_name, "cmd.exe") == 0 || stricmp(sh_name, "4os2.exe") == 0)
4292 opt = "/c";
4293 else
4294 opt = "-c";
4295
4296 /* save current stdio fds + their flags, and set not inheritable */
4297 i = pipe_err = 0;
4298 while (pipe_err >= 0 && i < 3)
4299 {
4300 pipe_err = stdio[i].handle = dup(i);
4301 stdio[i].flags = fcntl(i, F_GETFD, 0);
4302 fcntl(stdio[i].handle, F_SETFD, stdio[i].flags | FD_CLOEXEC);
4303 i++;
4304 }
4305 if (pipe_err < 0)
4306 {
4307 /* didn't get them all saved - clean up and bail out */
4308 int saved_err = errno;
4309 while (i-- > 0)
4310 {
4311 close(stdio[i].handle);
4312 }
4313 errno = saved_err;
4314 return posix_error();
4315 }
4316
4317 /* create pipe ends */
4318 file_count = 2;
4319 if (n == POPEN_3)
4320 file_count = 3;
4321 i = pipe_err = 0;
4322 while ((pipe_err == 0) && (i < file_count))
4323 pipe_err = pipe((int *)&p_fd[i++]);
4324 if (pipe_err < 0)
4325 {
4326 /* didn't get them all made - clean up and bail out */
4327 while (i-- > 0)
4328 {
4329 close(p_fd[i].wr);
4330 close(p_fd[i].rd);
4331 }
4332 errno = EPIPE;
4333 return posix_error();
4334 }
4335
4336 /* change the actual standard IO streams over temporarily,
4337 * making the retained pipe ends non-inheritable
4338 */
4339 pipe_err = 0;
4340
4341 /* - stdin */
4342 if (dup2(p_fd[0].rd, 0) == 0)
4343 {
4344 close(p_fd[0].rd);
4345 i = fcntl(p_fd[0].wr, F_GETFD, 0);
4346 fcntl(p_fd[0].wr, F_SETFD, i | FD_CLOEXEC);
4347 if ((p_s[0] = fdopen(p_fd[0].wr, wr_mode)) == NULL)
4348 {
4349 close(p_fd[0].wr);
4350 pipe_err = -1;
4351 }
4352 }
4353 else
4354 {
4355 pipe_err = -1;
4356 }
4357
4358 /* - stdout */
4359 if (pipe_err == 0)
4360 {
4361 if (dup2(p_fd[1].wr, 1) == 1)
4362 {
4363 close(p_fd[1].wr);
4364 i = fcntl(p_fd[1].rd, F_GETFD, 0);
4365 fcntl(p_fd[1].rd, F_SETFD, i | FD_CLOEXEC);
4366 if ((p_s[1] = fdopen(p_fd[1].rd, rd_mode)) == NULL)
4367 {
4368 close(p_fd[1].rd);
4369 pipe_err = -1;
4370 }
4371 }
4372 else
4373 {
4374 pipe_err = -1;
4375 }
4376 }
4377
4378 /* - stderr, as required */
4379 if (pipe_err == 0)
4380 switch (n)
4381 {
4382 case POPEN_3:
4383 {
4384 if (dup2(p_fd[2].wr, 2) == 2)
4385 {
4386 close(p_fd[2].wr);
4387 i = fcntl(p_fd[2].rd, F_GETFD, 0);
4388 fcntl(p_fd[2].rd, F_SETFD, i | FD_CLOEXEC);
4389 if ((p_s[2] = fdopen(p_fd[2].rd, rd_mode)) == NULL)
4390 {
4391 close(p_fd[2].rd);
4392 pipe_err = -1;
4393 }
4394 }
4395 else
4396 {
4397 pipe_err = -1;
4398 }
4399 break;
4400 }
4401
4402 case POPEN_4:
4403 {
4404 if (dup2(1, 2) != 2)
4405 {
4406 pipe_err = -1;
4407 }
4408 break;
4409 }
4410 }
4411
4412 /* spawn the child process */
4413 if (pipe_err == 0)
4414 {
4415 pipe_pid = spawnlp(P_NOWAIT, shell, shell, opt, cmdstring, (char *)0);
4416 if (pipe_pid == -1)
4417 {
4418 pipe_err = -1;
4419 }
4420 else
4421 {
4422 /* save the PID into the FILE structure
4423 * NOTE: this implementation doesn't actually
4424 * take advantage of this, but do it for
4425 * completeness - AIM Apr01
4426 */
4427 for (i = 0; i < file_count; i++)
4428 p_s[i]->_pid = pipe_pid;
4429 }
4430 }
4431
4432 /* reset standard IO to normal */
4433 for (i = 0; i < 3; i++)
4434 {
4435 dup2(stdio[i].handle, i);
4436 fcntl(i, F_SETFD, stdio[i].flags);
4437 close(stdio[i].handle);
4438 }
4439
4440 /* if any remnant problems, clean up and bail out */
4441 if (pipe_err < 0)
4442 {
4443 for (i = 0; i < 3; i++)
4444 {
4445 close(p_fd[i].rd);
4446 close(p_fd[i].wr);
4447 }
4448 errno = EPIPE;
4449 return posix_error_with_filename(cmdstring);
4450 }
4451
4452 /* build tuple of file objects to return */
4453 if ((p_f[0] = PyFile_FromFile(p_s[0], cmdstring, wr_mode, _PyPclose)) != NULL)
4454 PyFile_SetBufSize(p_f[0], bufsize);
4455 if ((p_f[1] = PyFile_FromFile(p_s[1], cmdstring, rd_mode, _PyPclose)) != NULL)
4456 PyFile_SetBufSize(p_f[1], bufsize);
4457 if (n == POPEN_3)
4458 {
4459 if ((p_f[2] = PyFile_FromFile(p_s[2], cmdstring, rd_mode, _PyPclose)) != NULL)
4460 PyFile_SetBufSize(p_f[0], bufsize);
Raymond Hettinger8ae46892003-10-12 19:09:37 +00004461 f = PyTuple_Pack(3, p_f[0], p_f[1], p_f[2]);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004462 }
4463 else
Raymond Hettinger8ae46892003-10-12 19:09:37 +00004464 f = PyTuple_Pack(2, p_f[0], p_f[1]);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004465
4466 /*
4467 * Insert the files we've created into the process dictionary
4468 * all referencing the list with the process handle and the
4469 * initial number of files (see description below in _PyPclose).
4470 * Since if _PyPclose later tried to wait on a process when all
4471 * handles weren't closed, it could create a deadlock with the
4472 * child, we spend some energy here to try to ensure that we
4473 * either insert all file handles into the dictionary or none
4474 * at all. It's a little clumsy with the various popen modes
4475 * and variable number of files involved.
4476 */
4477 if (!_PyPopenProcs)
4478 {
4479 _PyPopenProcs = PyDict_New();
4480 }
4481
4482 if (_PyPopenProcs)
4483 {
4484 PyObject *procObj, *pidObj, *intObj, *fileObj[3];
4485 int ins_rc[3];
4486
4487 fileObj[0] = fileObj[1] = fileObj[2] = NULL;
4488 ins_rc[0] = ins_rc[1] = ins_rc[2] = 0;
4489
4490 procObj = PyList_New(2);
4491 pidObj = PyInt_FromLong((long) pipe_pid);
4492 intObj = PyInt_FromLong((long) file_count);
4493
4494 if (procObj && pidObj && intObj)
4495 {
4496 PyList_SetItem(procObj, 0, pidObj);
4497 PyList_SetItem(procObj, 1, intObj);
4498
4499 fileObj[0] = PyLong_FromVoidPtr(p_s[0]);
4500 if (fileObj[0])
4501 {
4502 ins_rc[0] = PyDict_SetItem(_PyPopenProcs,
4503 fileObj[0],
4504 procObj);
4505 }
4506 fileObj[1] = PyLong_FromVoidPtr(p_s[1]);
4507 if (fileObj[1])
4508 {
4509 ins_rc[1] = PyDict_SetItem(_PyPopenProcs,
4510 fileObj[1],
4511 procObj);
4512 }
4513 if (file_count >= 3)
4514 {
4515 fileObj[2] = PyLong_FromVoidPtr(p_s[2]);
4516 if (fileObj[2])
4517 {
4518 ins_rc[2] = PyDict_SetItem(_PyPopenProcs,
4519 fileObj[2],
4520 procObj);
4521 }
4522 }
4523
4524 if (ins_rc[0] < 0 || !fileObj[0] ||
4525 ins_rc[1] < 0 || (file_count > 1 && !fileObj[1]) ||
4526 ins_rc[2] < 0 || (file_count > 2 && !fileObj[2]))
4527 {
4528 /* Something failed - remove any dictionary
4529 * entries that did make it.
4530 */
4531 if (!ins_rc[0] && fileObj[0])
4532 {
4533 PyDict_DelItem(_PyPopenProcs,
4534 fileObj[0]);
4535 }
4536 if (!ins_rc[1] && fileObj[1])
4537 {
4538 PyDict_DelItem(_PyPopenProcs,
4539 fileObj[1]);
4540 }
4541 if (!ins_rc[2] && fileObj[2])
4542 {
4543 PyDict_DelItem(_PyPopenProcs,
4544 fileObj[2]);
4545 }
4546 }
4547 }
Tim Peters11b23062003-04-23 02:39:17 +00004548
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004549 /*
4550 * Clean up our localized references for the dictionary keys
4551 * and value since PyDict_SetItem will Py_INCREF any copies
4552 * that got placed in the dictionary.
4553 */
4554 Py_XDECREF(procObj);
4555 Py_XDECREF(fileObj[0]);
4556 Py_XDECREF(fileObj[1]);
4557 Py_XDECREF(fileObj[2]);
4558 }
4559
4560 /* Child is launched. */
4561 return f;
4562}
4563
4564/*
4565 * Wrapper for fclose() to use for popen* files, so we can retrieve the
4566 * exit code for the child process and return as a result of the close.
4567 *
4568 * This function uses the _PyPopenProcs dictionary in order to map the
4569 * input file pointer to information about the process that was
4570 * originally created by the popen* call that created the file pointer.
4571 * The dictionary uses the file pointer as a key (with one entry
4572 * inserted for each file returned by the original popen* call) and a
4573 * single list object as the value for all files from a single call.
4574 * The list object contains the Win32 process handle at [0], and a file
4575 * count at [1], which is initialized to the total number of file
4576 * handles using that list.
4577 *
4578 * This function closes whichever handle it is passed, and decrements
4579 * the file count in the dictionary for the process handle pointed to
4580 * by this file. On the last close (when the file count reaches zero),
4581 * this function will wait for the child process and then return its
4582 * exit code as the result of the close() operation. This permits the
4583 * files to be closed in any order - it is always the close() of the
4584 * final handle that will return the exit code.
Andrew MacIntyrebaf25b02003-04-21 14:22:36 +00004585 *
4586 * NOTE: This function is currently called with the GIL released.
4587 * hence we use the GILState API to manage our state.
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004588 */
4589
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004590static int _PyPclose(FILE *file)
4591{
4592 int result;
4593 int exit_code;
4594 int pipe_pid;
4595 PyObject *procObj, *pidObj, *intObj, *fileObj;
4596 int file_count;
4597#ifdef WITH_THREAD
Andrew MacIntyrebaf25b02003-04-21 14:22:36 +00004598 PyGILState_STATE state;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004599#endif
4600
4601 /* Close the file handle first, to ensure it can't block the
4602 * child from exiting if it's the last handle.
4603 */
4604 result = fclose(file);
4605
4606#ifdef WITH_THREAD
Andrew MacIntyrebaf25b02003-04-21 14:22:36 +00004607 state = PyGILState_Ensure();
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004608#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004609 if (_PyPopenProcs)
4610 {
4611 if ((fileObj = PyLong_FromVoidPtr(file)) != NULL &&
4612 (procObj = PyDict_GetItem(_PyPopenProcs,
4613 fileObj)) != NULL &&
4614 (pidObj = PyList_GetItem(procObj,0)) != NULL &&
4615 (intObj = PyList_GetItem(procObj,1)) != NULL)
4616 {
4617 pipe_pid = (int) PyInt_AsLong(pidObj);
4618 file_count = (int) PyInt_AsLong(intObj);
4619
4620 if (file_count > 1)
4621 {
4622 /* Still other files referencing process */
4623 file_count--;
4624 PyList_SetItem(procObj,1,
4625 PyInt_FromLong((long) file_count));
4626 }
4627 else
4628 {
4629 /* Last file for this process */
4630 if (result != EOF &&
4631 waitpid(pipe_pid, &exit_code, 0) == pipe_pid)
4632 {
4633 /* extract exit status */
4634 if (WIFEXITED(exit_code))
4635 {
4636 result = WEXITSTATUS(exit_code);
4637 }
4638 else
4639 {
4640 errno = EPIPE;
4641 result = -1;
4642 }
4643 }
4644 else
4645 {
4646 /* Indicate failure - this will cause the file object
4647 * to raise an I/O error and translate the last
4648 * error code from errno. We do have a problem with
4649 * last errors that overlap the normal errno table,
4650 * but that's a consistent problem with the file object.
4651 */
4652 result = -1;
4653 }
4654 }
4655
4656 /* Remove this file pointer from dictionary */
4657 PyDict_DelItem(_PyPopenProcs, fileObj);
4658
4659 if (PyDict_Size(_PyPopenProcs) == 0)
4660 {
4661 Py_DECREF(_PyPopenProcs);
4662 _PyPopenProcs = NULL;
4663 }
4664
4665 } /* if object retrieval ok */
4666
4667 Py_XDECREF(fileObj);
4668 } /* if _PyPopenProcs */
4669
4670#ifdef WITH_THREAD
Andrew MacIntyrebaf25b02003-04-21 14:22:36 +00004671 PyGILState_Release(state);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004672#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004673 return result;
4674}
4675
4676#endif /* PYCC_??? */
4677
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004678#elif defined(MS_WINDOWS)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004679
4680/*
4681 * Portable 'popen' replacement for Win32.
4682 *
4683 * Written by Bill Tutt <billtut@microsoft.com>. Minor tweaks
4684 * and 2.0 integration by Fredrik Lundh <fredrik@pythonware.com>
Mark Hammondb37a3732000-08-14 04:47:33 +00004685 * Return code handling by David Bolen <db3l@fitlinxx.com>.
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004686 */
4687
4688#include <malloc.h>
4689#include <io.h>
4690#include <fcntl.h>
4691
4692/* These tell _PyPopen() wether to return 1, 2, or 3 file objects. */
4693#define POPEN_1 1
4694#define POPEN_2 2
4695#define POPEN_3 3
4696#define POPEN_4 4
4697
4698static PyObject *_PyPopen(char *, int, int);
Fredrik Lundh56055a42000-07-23 19:47:12 +00004699static int _PyPclose(FILE *file);
4700
4701/*
4702 * Internal dictionary mapping popen* file pointers to process handles,
Mark Hammondb37a3732000-08-14 04:47:33 +00004703 * for use when retrieving the process exit code. See _PyPclose() below
4704 * for more information on this dictionary's use.
Fredrik Lundh56055a42000-07-23 19:47:12 +00004705 */
4706static PyObject *_PyPopenProcs = NULL;
4707
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004708
4709/* popen that works from a GUI.
4710 *
4711 * The result of this function is a pipe (file) connected to the
4712 * processes stdin or stdout, depending on the requested mode.
4713 */
4714
4715static PyObject *
4716posix_popen(PyObject *self, PyObject *args)
4717{
Raymond Hettingerb5cb6652003-09-01 22:25:41 +00004718 PyObject *f;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004719 int tm = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00004720
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004721 char *cmdstring;
4722 char *mode = "r";
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00004723 int bufsize = -1;
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004724 if (!PyArg_ParseTuple(args, "s|si:popen", &cmdstring, &mode, &bufsize))
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004725 return NULL;
4726
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004727 if (*mode == 'r')
4728 tm = _O_RDONLY;
4729 else if (*mode != 'w') {
Fred Drake661ea262000-10-24 19:57:45 +00004730 PyErr_SetString(PyExc_ValueError, "popen() arg 2 must be 'r' or 'w'");
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004731 return NULL;
4732 } else
4733 tm = _O_WRONLY;
Tim Peters5aa91602002-01-30 05:46:57 +00004734
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004735 if (bufsize != -1) {
Fred Drake661ea262000-10-24 19:57:45 +00004736 PyErr_SetString(PyExc_ValueError, "popen() arg 3 must be -1");
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004737 return NULL;
4738 }
4739
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004740 if (*(mode+1) == 't')
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00004741 f = _PyPopen(cmdstring, tm | _O_TEXT, POPEN_1);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004742 else if (*(mode+1) == 'b')
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00004743 f = _PyPopen(cmdstring, tm | _O_BINARY, POPEN_1);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004744 else
4745 f = _PyPopen(cmdstring, tm | _O_TEXT, POPEN_1);
4746
4747 return f;
4748}
4749
4750/* Variation on win32pipe.popen
4751 *
4752 * The result of this function is a pipe (file) connected to the
4753 * process's stdin, and a pipe connected to the process's stdout.
4754 */
4755
4756static PyObject *
4757win32_popen2(PyObject *self, PyObject *args)
4758{
4759 PyObject *f;
4760 int tm=0;
Tim Peters5aa91602002-01-30 05:46:57 +00004761
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004762 char *cmdstring;
4763 char *mode = "t";
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004764 int bufsize = -1;
4765 if (!PyArg_ParseTuple(args, "s|si:popen2", &cmdstring, &mode, &bufsize))
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004766 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00004767
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004768 if (*mode == 't')
4769 tm = _O_TEXT;
4770 else if (*mode != 'b') {
Fred Drake661ea262000-10-24 19:57:45 +00004771 PyErr_SetString(PyExc_ValueError, "popen2() arg 2 must be 't' or 'b'");
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004772 return NULL;
4773 } else
4774 tm = _O_BINARY;
Tim Peters5aa91602002-01-30 05:46:57 +00004775
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004776 if (bufsize != -1) {
Fred Drake661ea262000-10-24 19:57:45 +00004777 PyErr_SetString(PyExc_ValueError, "popen2() arg 3 must be -1");
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004778 return NULL;
4779 }
4780
4781 f = _PyPopen(cmdstring, tm, POPEN_2);
Tim Peters5aa91602002-01-30 05:46:57 +00004782
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004783 return f;
4784}
4785
4786/*
4787 * Variation on <om win32pipe.popen>
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00004788 *
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004789 * The result of this function is 3 pipes - the process's stdin,
4790 * stdout and stderr
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004791 */
4792
4793static PyObject *
4794win32_popen3(PyObject *self, PyObject *args)
4795{
4796 PyObject *f;
4797 int tm = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00004798
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004799 char *cmdstring;
4800 char *mode = "t";
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004801 int bufsize = -1;
4802 if (!PyArg_ParseTuple(args, "s|si:popen3", &cmdstring, &mode, &bufsize))
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004803 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00004804
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004805 if (*mode == 't')
4806 tm = _O_TEXT;
4807 else if (*mode != 'b') {
Fred Drake661ea262000-10-24 19:57:45 +00004808 PyErr_SetString(PyExc_ValueError, "popen3() arg 2 must be 't' or 'b'");
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004809 return NULL;
4810 } else
4811 tm = _O_BINARY;
Tim Peters5aa91602002-01-30 05:46:57 +00004812
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004813 if (bufsize != -1) {
Fred Drake661ea262000-10-24 19:57:45 +00004814 PyErr_SetString(PyExc_ValueError, "popen3() arg 3 must be -1");
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004815 return NULL;
4816 }
4817
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004818 f = _PyPopen(cmdstring, tm, POPEN_3);
Tim Peters5aa91602002-01-30 05:46:57 +00004819
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004820 return f;
4821}
4822
4823/*
4824 * Variation on win32pipe.popen
4825 *
Tim Peters5aa91602002-01-30 05:46:57 +00004826 * The result of this function is 2 pipes - the processes stdin,
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004827 * and stdout+stderr combined as a single pipe.
4828 */
4829
4830static PyObject *
4831win32_popen4(PyObject *self, PyObject *args)
4832{
4833 PyObject *f;
4834 int tm = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00004835
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004836 char *cmdstring;
4837 char *mode = "t";
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004838 int bufsize = -1;
4839 if (!PyArg_ParseTuple(args, "s|si:popen4", &cmdstring, &mode, &bufsize))
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004840 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00004841
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004842 if (*mode == 't')
4843 tm = _O_TEXT;
4844 else if (*mode != 'b') {
Fred Drake661ea262000-10-24 19:57:45 +00004845 PyErr_SetString(PyExc_ValueError, "popen4() arg 2 must be 't' or 'b'");
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004846 return NULL;
4847 } else
4848 tm = _O_BINARY;
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004849
4850 if (bufsize != -1) {
Fred Drake661ea262000-10-24 19:57:45 +00004851 PyErr_SetString(PyExc_ValueError, "popen4() arg 3 must be -1");
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004852 return NULL;
4853 }
4854
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00004855 f = _PyPopen(cmdstring, tm, POPEN_4);
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004856
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004857 return f;
4858}
4859
Mark Hammond08501372001-01-31 07:30:29 +00004860static BOOL
Mark Hammondb37a3732000-08-14 04:47:33 +00004861_PyPopenCreateProcess(char *cmdstring,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004862 HANDLE hStdin,
4863 HANDLE hStdout,
Mark Hammondb37a3732000-08-14 04:47:33 +00004864 HANDLE hStderr,
4865 HANDLE *hProcess)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004866{
4867 PROCESS_INFORMATION piProcInfo;
4868 STARTUPINFO siStartInfo;
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004869 DWORD dwProcessFlags = 0; /* no NEW_CONSOLE by default for Ctrl+C handling */
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004870 char *s1,*s2, *s3 = " /c ";
Mark Hammond08501372001-01-31 07:30:29 +00004871 const char *szConsoleSpawn = "w9xpopen.exe";
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004872 int i;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004873 Py_ssize_t x;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004874
4875 if (i = GetEnvironmentVariable("COMSPEC",NULL,0)) {
Tim Peters402d5982001-08-27 06:37:48 +00004876 char *comshell;
4877
Tim Peters92e4dd82002-10-05 01:47:34 +00004878 s1 = (char *)alloca(i);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004879 if (!(x = GetEnvironmentVariable("COMSPEC", s1, i)))
Martin v. Löwis18e16552006-02-15 17:27:45 +00004880 /* x < i, so x fits into an integer */
4881 return (int)x;
Tim Peters402d5982001-08-27 06:37:48 +00004882
4883 /* Explicitly check if we are using COMMAND.COM. If we are
4884 * then use the w9xpopen hack.
4885 */
4886 comshell = s1 + x;
4887 while (comshell >= s1 && *comshell != '\\')
4888 --comshell;
4889 ++comshell;
4890
4891 if (GetVersion() < 0x80000000 &&
4892 _stricmp(comshell, "command.com") != 0) {
4893 /* NT/2000 and not using command.com. */
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004894 x = i + strlen(s3) + strlen(cmdstring) + 1;
Tim Peters92e4dd82002-10-05 01:47:34 +00004895 s2 = (char *)alloca(x);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004896 ZeroMemory(s2, x);
Tim Peters75cdad52001-11-28 22:07:30 +00004897 PyOS_snprintf(s2, x, "%s%s%s", s1, s3, cmdstring);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004898 }
4899 else {
4900 /*
Tim Peters402d5982001-08-27 06:37:48 +00004901 * Oh gag, we're on Win9x or using COMMAND.COM. Use
4902 * the workaround listed in KB: Q150956
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004903 */
Mark Hammond08501372001-01-31 07:30:29 +00004904 char modulepath[_MAX_PATH];
4905 struct stat statinfo;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004906 GetModuleFileName(NULL, modulepath, sizeof(modulepath));
Martin v. Löwis26fd9602006-04-22 11:15:41 +00004907 for (x = i = 0; modulepath[i]; i++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004908 if (modulepath[i] == SEP)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004909 x = i+1;
4910 modulepath[x] = '\0';
Mark Hammond08501372001-01-31 07:30:29 +00004911 /* Create the full-name to w9xpopen, so we can test it exists */
Tim Peters5aa91602002-01-30 05:46:57 +00004912 strncat(modulepath,
4913 szConsoleSpawn,
Mark Hammond08501372001-01-31 07:30:29 +00004914 (sizeof(modulepath)/sizeof(modulepath[0]))
4915 -strlen(modulepath));
4916 if (stat(modulepath, &statinfo) != 0) {
Kristján Valur Jónsson17b8e972007-04-25 00:10:50 +00004917 size_t mplen = sizeof(modulepath)/sizeof(modulepath[0]);
Tim Peters5aa91602002-01-30 05:46:57 +00004918 /* Eeek - file-not-found - possibly an embedding
4919 situation - see if we can locate it in sys.prefix
Mark Hammond08501372001-01-31 07:30:29 +00004920 */
Tim Peters5aa91602002-01-30 05:46:57 +00004921 strncpy(modulepath,
4922 Py_GetExecPrefix(),
Kristján Valur Jónsson17b8e972007-04-25 00:10:50 +00004923 mplen);
4924 modulepath[mplen-1] = '\0';
Mark Hammond08501372001-01-31 07:30:29 +00004925 if (modulepath[strlen(modulepath)-1] != '\\')
4926 strcat(modulepath, "\\");
Tim Peters5aa91602002-01-30 05:46:57 +00004927 strncat(modulepath,
4928 szConsoleSpawn,
Kristján Valur Jónsson17b8e972007-04-25 00:10:50 +00004929 mplen-strlen(modulepath));
Mark Hammond08501372001-01-31 07:30:29 +00004930 /* No where else to look - raise an easily identifiable
4931 error, rather than leaving Windows to report
4932 "file not found" - as the user is probably blissfully
4933 unaware this shim EXE is used, and it will confuse them.
4934 (well, it confused me for a while ;-)
4935 */
4936 if (stat(modulepath, &statinfo) != 0) {
Tim Peters5aa91602002-01-30 05:46:57 +00004937 PyErr_Format(PyExc_RuntimeError,
Mark Hammond08501372001-01-31 07:30:29 +00004938 "Can not locate '%s' which is needed "
Tim Peters402d5982001-08-27 06:37:48 +00004939 "for popen to work with your shell "
4940 "or platform.",
Mark Hammond08501372001-01-31 07:30:29 +00004941 szConsoleSpawn);
4942 return FALSE;
4943 }
4944 }
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004945 x = i + strlen(s3) + strlen(cmdstring) + 1 +
Tim Peters5aa91602002-01-30 05:46:57 +00004946 strlen(modulepath) +
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004947 strlen(szConsoleSpawn) + 1;
Mark Hammond08501372001-01-31 07:30:29 +00004948
Tim Peters92e4dd82002-10-05 01:47:34 +00004949 s2 = (char *)alloca(x);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004950 ZeroMemory(s2, x);
Mark Hammonde7fefbf2002-04-03 01:47:00 +00004951 /* To maintain correct argument passing semantics,
4952 we pass the command-line as it stands, and allow
4953 quoting to be applied. w9xpopen.exe will then
4954 use its argv vector, and re-quote the necessary
4955 args for the ultimate child process.
4956 */
Tim Peters75cdad52001-11-28 22:07:30 +00004957 PyOS_snprintf(
4958 s2, x,
Mark Hammonde7fefbf2002-04-03 01:47:00 +00004959 "\"%s\" %s%s%s",
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004960 modulepath,
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004961 s1,
4962 s3,
4963 cmdstring);
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004964 /* Not passing CREATE_NEW_CONSOLE has been known to
Tim Peters11b23062003-04-23 02:39:17 +00004965 cause random failures on win9x. Specifically a
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004966 dialog:
4967 "Your program accessed mem currently in use at xxx"
4968 and a hopeful warning about the stability of your
4969 system.
4970 Cost is Ctrl+C wont kill children, but anyone
4971 who cares can have a go!
4972 */
4973 dwProcessFlags |= CREATE_NEW_CONSOLE;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004974 }
4975 }
4976
4977 /* Could be an else here to try cmd.exe / command.com in the path
4978 Now we'll just error out.. */
Mark Hammond08501372001-01-31 07:30:29 +00004979 else {
Tim Peters402d5982001-08-27 06:37:48 +00004980 PyErr_SetString(PyExc_RuntimeError,
4981 "Cannot locate a COMSPEC environment variable to "
4982 "use as the shell");
Mark Hammond08501372001-01-31 07:30:29 +00004983 return FALSE;
4984 }
Tim Peters5aa91602002-01-30 05:46:57 +00004985
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004986 ZeroMemory(&siStartInfo, sizeof(STARTUPINFO));
4987 siStartInfo.cb = sizeof(STARTUPINFO);
4988 siStartInfo.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
4989 siStartInfo.hStdInput = hStdin;
4990 siStartInfo.hStdOutput = hStdout;
4991 siStartInfo.hStdError = hStderr;
4992 siStartInfo.wShowWindow = SW_HIDE;
4993
4994 if (CreateProcess(NULL,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004995 s2,
4996 NULL,
4997 NULL,
4998 TRUE,
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004999 dwProcessFlags,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00005000 NULL,
5001 NULL,
5002 &siStartInfo,
5003 &piProcInfo) ) {
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005004 /* Close the handles now so anyone waiting is woken. */
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005005 CloseHandle(piProcInfo.hThread);
Fredrik Lundh56055a42000-07-23 19:47:12 +00005006
Mark Hammondb37a3732000-08-14 04:47:33 +00005007 /* Return process handle */
5008 *hProcess = piProcInfo.hProcess;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005009 return TRUE;
5010 }
Tim Peters402d5982001-08-27 06:37:48 +00005011 win32_error("CreateProcess", s2);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005012 return FALSE;
5013}
5014
5015/* The following code is based off of KB: Q190351 */
5016
5017static PyObject *
5018_PyPopen(char *cmdstring, int mode, int n)
5019{
5020 HANDLE hChildStdinRd, hChildStdinWr, hChildStdoutRd, hChildStdoutWr,
5021 hChildStderrRd, hChildStderrWr, hChildStdinWrDup, hChildStdoutRdDup,
Mark Hammondb37a3732000-08-14 04:47:33 +00005022 hChildStderrRdDup, hProcess; /* hChildStdoutWrDup; */
Tim Peters5aa91602002-01-30 05:46:57 +00005023
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005024 SECURITY_ATTRIBUTES saAttr;
5025 BOOL fSuccess;
5026 int fd1, fd2, fd3;
5027 FILE *f1, *f2, *f3;
Mark Hammondb37a3732000-08-14 04:47:33 +00005028 long file_count;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005029 PyObject *f;
5030
5031 saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
5032 saAttr.bInheritHandle = TRUE;
5033 saAttr.lpSecurityDescriptor = NULL;
5034
5035 if (!CreatePipe(&hChildStdinRd, &hChildStdinWr, &saAttr, 0))
5036 return win32_error("CreatePipe", NULL);
5037
5038 /* Create new output read handle and the input write handle. Set
5039 * the inheritance properties to FALSE. Otherwise, the child inherits
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00005040 * these handles; resulting in non-closeable handles to the pipes
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005041 * being created. */
5042 fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdinWr,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00005043 GetCurrentProcess(), &hChildStdinWrDup, 0,
5044 FALSE,
5045 DUPLICATE_SAME_ACCESS);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005046 if (!fSuccess)
5047 return win32_error("DuplicateHandle", NULL);
5048
5049 /* Close the inheritable version of ChildStdin
5050 that we're using. */
5051 CloseHandle(hChildStdinWr);
5052
5053 if (!CreatePipe(&hChildStdoutRd, &hChildStdoutWr, &saAttr, 0))
5054 return win32_error("CreatePipe", NULL);
5055
5056 fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdoutRd,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00005057 GetCurrentProcess(), &hChildStdoutRdDup, 0,
5058 FALSE, DUPLICATE_SAME_ACCESS);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005059 if (!fSuccess)
5060 return win32_error("DuplicateHandle", NULL);
5061
5062 /* Close the inheritable version of ChildStdout
5063 that we're using. */
5064 CloseHandle(hChildStdoutRd);
5065
5066 if (n != POPEN_4) {
5067 if (!CreatePipe(&hChildStderrRd, &hChildStderrWr, &saAttr, 0))
5068 return win32_error("CreatePipe", NULL);
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00005069 fSuccess = DuplicateHandle(GetCurrentProcess(),
5070 hChildStderrRd,
5071 GetCurrentProcess(),
5072 &hChildStderrRdDup, 0,
5073 FALSE, DUPLICATE_SAME_ACCESS);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005074 if (!fSuccess)
5075 return win32_error("DuplicateHandle", NULL);
5076 /* Close the inheritable version of ChildStdErr that we're using. */
5077 CloseHandle(hChildStderrRd);
5078 }
Tim Peters5aa91602002-01-30 05:46:57 +00005079
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005080 switch (n) {
5081 case POPEN_1:
5082 switch (mode & (_O_RDONLY | _O_TEXT | _O_BINARY | _O_WRONLY)) {
5083 case _O_WRONLY | _O_TEXT:
5084 /* Case for writing to child Stdin in text mode. */
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005085 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005086 f1 = _fdopen(fd1, "w");
Fredrik Lundh56055a42000-07-23 19:47:12 +00005087 f = PyFile_FromFile(f1, cmdstring, "w", _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005088 PyFile_SetBufSize(f, 0);
5089 /* We don't care about these pipes anymore, so close them. */
5090 CloseHandle(hChildStdoutRdDup);
5091 CloseHandle(hChildStderrRdDup);
5092 break;
5093
5094 case _O_RDONLY | _O_TEXT:
5095 /* Case for reading from child Stdout in text mode. */
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005096 fd1 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005097 f1 = _fdopen(fd1, "r");
Fredrik Lundh56055a42000-07-23 19:47:12 +00005098 f = PyFile_FromFile(f1, cmdstring, "r", _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005099 PyFile_SetBufSize(f, 0);
5100 /* We don't care about these pipes anymore, so close them. */
5101 CloseHandle(hChildStdinWrDup);
5102 CloseHandle(hChildStderrRdDup);
5103 break;
5104
5105 case _O_RDONLY | _O_BINARY:
5106 /* Case for readinig from child Stdout in binary mode. */
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005107 fd1 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005108 f1 = _fdopen(fd1, "rb");
Fredrik Lundh56055a42000-07-23 19:47:12 +00005109 f = PyFile_FromFile(f1, cmdstring, "rb", _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005110 PyFile_SetBufSize(f, 0);
5111 /* We don't care about these pipes anymore, so close them. */
5112 CloseHandle(hChildStdinWrDup);
5113 CloseHandle(hChildStderrRdDup);
5114 break;
5115
5116 case _O_WRONLY | _O_BINARY:
5117 /* Case for writing to child Stdin in binary mode. */
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005118 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005119 f1 = _fdopen(fd1, "wb");
Fredrik Lundh56055a42000-07-23 19:47:12 +00005120 f = PyFile_FromFile(f1, cmdstring, "wb", _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005121 PyFile_SetBufSize(f, 0);
5122 /* We don't care about these pipes anymore, so close them. */
5123 CloseHandle(hChildStdoutRdDup);
5124 CloseHandle(hChildStderrRdDup);
5125 break;
5126 }
Mark Hammondb37a3732000-08-14 04:47:33 +00005127 file_count = 1;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005128 break;
Tim Peters5aa91602002-01-30 05:46:57 +00005129
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005130 case POPEN_2:
5131 case POPEN_4:
5132 {
5133 char *m1, *m2;
5134 PyObject *p1, *p2;
Tim Peters5aa91602002-01-30 05:46:57 +00005135
Tim Peters7dca21e2002-08-19 00:42:29 +00005136 if (mode & _O_TEXT) {
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005137 m1 = "r";
5138 m2 = "w";
5139 } else {
5140 m1 = "rb";
5141 m2 = "wb";
5142 }
5143
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005144 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005145 f1 = _fdopen(fd1, m2);
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005146 fd2 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005147 f2 = _fdopen(fd2, m1);
Fredrik Lundh56055a42000-07-23 19:47:12 +00005148 p1 = PyFile_FromFile(f1, cmdstring, m2, _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005149 PyFile_SetBufSize(p1, 0);
Mark Hammondb37a3732000-08-14 04:47:33 +00005150 p2 = PyFile_FromFile(f2, cmdstring, m1, _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005151 PyFile_SetBufSize(p2, 0);
5152
5153 if (n != 4)
5154 CloseHandle(hChildStderrRdDup);
5155
Raymond Hettinger8ae46892003-10-12 19:09:37 +00005156 f = PyTuple_Pack(2,p1,p2);
Mark Hammond64aae662001-01-31 05:38:47 +00005157 Py_XDECREF(p1);
5158 Py_XDECREF(p2);
Mark Hammondb37a3732000-08-14 04:47:33 +00005159 file_count = 2;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005160 break;
5161 }
Tim Peters5aa91602002-01-30 05:46:57 +00005162
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005163 case POPEN_3:
5164 {
5165 char *m1, *m2;
5166 PyObject *p1, *p2, *p3;
Tim Peters5aa91602002-01-30 05:46:57 +00005167
Tim Peters7dca21e2002-08-19 00:42:29 +00005168 if (mode & _O_TEXT) {
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005169 m1 = "r";
5170 m2 = "w";
5171 } else {
5172 m1 = "rb";
5173 m2 = "wb";
5174 }
5175
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005176 fd1 = _open_osfhandle((Py_intptr_t)hChildStdinWrDup, mode);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005177 f1 = _fdopen(fd1, m2);
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005178 fd2 = _open_osfhandle((Py_intptr_t)hChildStdoutRdDup, mode);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005179 f2 = _fdopen(fd2, m1);
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005180 fd3 = _open_osfhandle((Py_intptr_t)hChildStderrRdDup, mode);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005181 f3 = _fdopen(fd3, m1);
Fredrik Lundh56055a42000-07-23 19:47:12 +00005182 p1 = PyFile_FromFile(f1, cmdstring, m2, _PyPclose);
Mark Hammondb37a3732000-08-14 04:47:33 +00005183 p2 = PyFile_FromFile(f2, cmdstring, m1, _PyPclose);
5184 p3 = PyFile_FromFile(f3, cmdstring, m1, _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005185 PyFile_SetBufSize(p1, 0);
5186 PyFile_SetBufSize(p2, 0);
5187 PyFile_SetBufSize(p3, 0);
Raymond Hettinger8ae46892003-10-12 19:09:37 +00005188 f = PyTuple_Pack(3,p1,p2,p3);
Mark Hammond64aae662001-01-31 05:38:47 +00005189 Py_XDECREF(p1);
5190 Py_XDECREF(p2);
5191 Py_XDECREF(p3);
Mark Hammondb37a3732000-08-14 04:47:33 +00005192 file_count = 3;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005193 break;
5194 }
5195 }
5196
5197 if (n == POPEN_4) {
5198 if (!_PyPopenCreateProcess(cmdstring,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00005199 hChildStdinRd,
5200 hChildStdoutWr,
Mark Hammondb37a3732000-08-14 04:47:33 +00005201 hChildStdoutWr,
5202 &hProcess))
Mark Hammond08501372001-01-31 07:30:29 +00005203 return NULL;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005204 }
5205 else {
5206 if (!_PyPopenCreateProcess(cmdstring,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00005207 hChildStdinRd,
5208 hChildStdoutWr,
Mark Hammondb37a3732000-08-14 04:47:33 +00005209 hChildStderrWr,
5210 &hProcess))
Mark Hammond08501372001-01-31 07:30:29 +00005211 return NULL;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005212 }
5213
Mark Hammondb37a3732000-08-14 04:47:33 +00005214 /*
5215 * Insert the files we've created into the process dictionary
5216 * all referencing the list with the process handle and the
5217 * initial number of files (see description below in _PyPclose).
5218 * Since if _PyPclose later tried to wait on a process when all
5219 * handles weren't closed, it could create a deadlock with the
5220 * child, we spend some energy here to try to ensure that we
5221 * either insert all file handles into the dictionary or none
5222 * at all. It's a little clumsy with the various popen modes
5223 * and variable number of files involved.
5224 */
5225 if (!_PyPopenProcs) {
5226 _PyPopenProcs = PyDict_New();
5227 }
5228
5229 if (_PyPopenProcs) {
5230 PyObject *procObj, *hProcessObj, *intObj, *fileObj[3];
5231 int ins_rc[3];
5232
5233 fileObj[0] = fileObj[1] = fileObj[2] = NULL;
5234 ins_rc[0] = ins_rc[1] = ins_rc[2] = 0;
5235
5236 procObj = PyList_New(2);
5237 hProcessObj = PyLong_FromVoidPtr(hProcess);
5238 intObj = PyInt_FromLong(file_count);
5239
5240 if (procObj && hProcessObj && intObj) {
5241 PyList_SetItem(procObj,0,hProcessObj);
5242 PyList_SetItem(procObj,1,intObj);
5243
5244 fileObj[0] = PyLong_FromVoidPtr(f1);
5245 if (fileObj[0]) {
5246 ins_rc[0] = PyDict_SetItem(_PyPopenProcs,
5247 fileObj[0],
5248 procObj);
5249 }
5250 if (file_count >= 2) {
5251 fileObj[1] = PyLong_FromVoidPtr(f2);
5252 if (fileObj[1]) {
5253 ins_rc[1] = PyDict_SetItem(_PyPopenProcs,
5254 fileObj[1],
5255 procObj);
5256 }
5257 }
5258 if (file_count >= 3) {
5259 fileObj[2] = PyLong_FromVoidPtr(f3);
5260 if (fileObj[2]) {
5261 ins_rc[2] = PyDict_SetItem(_PyPopenProcs,
5262 fileObj[2],
5263 procObj);
5264 }
5265 }
5266
5267 if (ins_rc[0] < 0 || !fileObj[0] ||
5268 ins_rc[1] < 0 || (file_count > 1 && !fileObj[1]) ||
5269 ins_rc[2] < 0 || (file_count > 2 && !fileObj[2])) {
5270 /* Something failed - remove any dictionary
5271 * entries that did make it.
5272 */
5273 if (!ins_rc[0] && fileObj[0]) {
5274 PyDict_DelItem(_PyPopenProcs,
5275 fileObj[0]);
5276 }
5277 if (!ins_rc[1] && fileObj[1]) {
5278 PyDict_DelItem(_PyPopenProcs,
5279 fileObj[1]);
5280 }
5281 if (!ins_rc[2] && fileObj[2]) {
5282 PyDict_DelItem(_PyPopenProcs,
5283 fileObj[2]);
5284 }
5285 }
5286 }
Tim Peters5aa91602002-01-30 05:46:57 +00005287
Mark Hammondb37a3732000-08-14 04:47:33 +00005288 /*
5289 * Clean up our localized references for the dictionary keys
5290 * and value since PyDict_SetItem will Py_INCREF any copies
5291 * that got placed in the dictionary.
5292 */
5293 Py_XDECREF(procObj);
5294 Py_XDECREF(fileObj[0]);
5295 Py_XDECREF(fileObj[1]);
5296 Py_XDECREF(fileObj[2]);
5297 }
5298
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005299 /* Child is launched. Close the parents copy of those pipe
5300 * handles that only the child should have open. You need to
5301 * make sure that no handles to the write end of the output pipe
5302 * are maintained in this process or else the pipe will not close
5303 * when the child process exits and the ReadFile will hang. */
5304
5305 if (!CloseHandle(hChildStdinRd))
5306 return win32_error("CloseHandle", NULL);
Tim Peters5aa91602002-01-30 05:46:57 +00005307
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005308 if (!CloseHandle(hChildStdoutWr))
5309 return win32_error("CloseHandle", NULL);
Tim Peters5aa91602002-01-30 05:46:57 +00005310
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005311 if ((n != 4) && (!CloseHandle(hChildStderrWr)))
5312 return win32_error("CloseHandle", NULL);
5313
5314 return f;
5315}
Fredrik Lundh56055a42000-07-23 19:47:12 +00005316
5317/*
5318 * Wrapper for fclose() to use for popen* files, so we can retrieve the
5319 * exit code for the child process and return as a result of the close.
Mark Hammondb37a3732000-08-14 04:47:33 +00005320 *
5321 * This function uses the _PyPopenProcs dictionary in order to map the
5322 * input file pointer to information about the process that was
5323 * originally created by the popen* call that created the file pointer.
5324 * The dictionary uses the file pointer as a key (with one entry
5325 * inserted for each file returned by the original popen* call) and a
5326 * single list object as the value for all files from a single call.
5327 * The list object contains the Win32 process handle at [0], and a file
5328 * count at [1], which is initialized to the total number of file
5329 * handles using that list.
5330 *
5331 * This function closes whichever handle it is passed, and decrements
5332 * the file count in the dictionary for the process handle pointed to
5333 * by this file. On the last close (when the file count reaches zero),
5334 * this function will wait for the child process and then return its
5335 * exit code as the result of the close() operation. This permits the
5336 * files to be closed in any order - it is always the close() of the
5337 * final handle that will return the exit code.
Mark Hammond8d98d2c2003-04-19 15:41:53 +00005338 *
5339 * NOTE: This function is currently called with the GIL released.
5340 * hence we use the GILState API to manage our state.
Fredrik Lundh56055a42000-07-23 19:47:12 +00005341 */
Tim Peters736aa322000-09-01 06:51:24 +00005342
Fredrik Lundh56055a42000-07-23 19:47:12 +00005343static int _PyPclose(FILE *file)
5344{
Fredrik Lundh20318932000-07-26 17:29:12 +00005345 int result;
Fredrik Lundh56055a42000-07-23 19:47:12 +00005346 DWORD exit_code;
5347 HANDLE hProcess;
Mark Hammondb37a3732000-08-14 04:47:33 +00005348 PyObject *procObj, *hProcessObj, *intObj, *fileObj;
5349 long file_count;
Tim Peters736aa322000-09-01 06:51:24 +00005350#ifdef WITH_THREAD
Mark Hammond8d98d2c2003-04-19 15:41:53 +00005351 PyGILState_STATE state;
Tim Peters736aa322000-09-01 06:51:24 +00005352#endif
5353
Fredrik Lundh20318932000-07-26 17:29:12 +00005354 /* Close the file handle first, to ensure it can't block the
Mark Hammondb37a3732000-08-14 04:47:33 +00005355 * child from exiting if it's the last handle.
Fredrik Lundh20318932000-07-26 17:29:12 +00005356 */
5357 result = fclose(file);
Tim Peters736aa322000-09-01 06:51:24 +00005358#ifdef WITH_THREAD
Mark Hammond8d98d2c2003-04-19 15:41:53 +00005359 state = PyGILState_Ensure();
Tim Peters736aa322000-09-01 06:51:24 +00005360#endif
Fredrik Lundh56055a42000-07-23 19:47:12 +00005361 if (_PyPopenProcs) {
Mark Hammondb37a3732000-08-14 04:47:33 +00005362 if ((fileObj = PyLong_FromVoidPtr(file)) != NULL &&
5363 (procObj = PyDict_GetItem(_PyPopenProcs,
5364 fileObj)) != NULL &&
5365 (hProcessObj = PyList_GetItem(procObj,0)) != NULL &&
5366 (intObj = PyList_GetItem(procObj,1)) != NULL) {
5367
5368 hProcess = PyLong_AsVoidPtr(hProcessObj);
5369 file_count = PyInt_AsLong(intObj);
5370
5371 if (file_count > 1) {
5372 /* Still other files referencing process */
5373 file_count--;
5374 PyList_SetItem(procObj,1,
5375 PyInt_FromLong(file_count));
5376 } else {
5377 /* Last file for this process */
Fredrik Lundh20318932000-07-26 17:29:12 +00005378 if (result != EOF &&
5379 WaitForSingleObject(hProcess, INFINITE) != WAIT_FAILED &&
5380 GetExitCodeProcess(hProcess, &exit_code)) {
Fredrik Lundh56055a42000-07-23 19:47:12 +00005381 /* Possible truncation here in 16-bit environments, but
5382 * real exit codes are just the lower byte in any event.
5383 */
5384 result = exit_code;
Fredrik Lundh56055a42000-07-23 19:47:12 +00005385 } else {
Fredrik Lundh20318932000-07-26 17:29:12 +00005386 /* Indicate failure - this will cause the file object
5387 * to raise an I/O error and translate the last Win32
5388 * error code from errno. We do have a problem with
5389 * last errors that overlap the normal errno table,
5390 * but that's a consistent problem with the file object.
Fredrik Lundh56055a42000-07-23 19:47:12 +00005391 */
Fredrik Lundh20318932000-07-26 17:29:12 +00005392 if (result != EOF) {
5393 /* If the error wasn't from the fclose(), then
5394 * set errno for the file object error handling.
5395 */
5396 errno = GetLastError();
5397 }
5398 result = -1;
Fredrik Lundh56055a42000-07-23 19:47:12 +00005399 }
5400
5401 /* Free up the native handle at this point */
5402 CloseHandle(hProcess);
Mark Hammondb37a3732000-08-14 04:47:33 +00005403 }
Fredrik Lundh56055a42000-07-23 19:47:12 +00005404
Mark Hammondb37a3732000-08-14 04:47:33 +00005405 /* Remove this file pointer from dictionary */
5406 PyDict_DelItem(_PyPopenProcs, fileObj);
5407
5408 if (PyDict_Size(_PyPopenProcs) == 0) {
5409 Py_DECREF(_PyPopenProcs);
5410 _PyPopenProcs = NULL;
5411 }
5412
5413 } /* if object retrieval ok */
5414
5415 Py_XDECREF(fileObj);
Fredrik Lundh56055a42000-07-23 19:47:12 +00005416 } /* if _PyPopenProcs */
5417
Tim Peters736aa322000-09-01 06:51:24 +00005418#ifdef WITH_THREAD
Mark Hammond8d98d2c2003-04-19 15:41:53 +00005419 PyGILState_Release(state);
Tim Peters736aa322000-09-01 06:51:24 +00005420#endif
Fredrik Lundh56055a42000-07-23 19:47:12 +00005421 return result;
5422}
Tim Peters9acdd3a2000-09-01 19:26:36 +00005423
5424#else /* which OS? */
Barry Warsaw53699e91996-12-10 23:23:01 +00005425static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005426posix_popen(PyObject *self, PyObject *args)
Guido van Rossum3b066191991-06-04 19:40:25 +00005427{
Guido van Rossuma6a1e531995-01-10 15:36:38 +00005428 char *name;
5429 char *mode = "r";
5430 int bufsize = -1;
Guido van Rossum3b066191991-06-04 19:40:25 +00005431 FILE *fp;
Barry Warsaw53699e91996-12-10 23:23:01 +00005432 PyObject *f;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005433 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
Guido van Rossum3b066191991-06-04 19:40:25 +00005434 return NULL;
Martin v. Löwis9ad853b2003-10-31 10:01:53 +00005435 /* Strip mode of binary or text modifiers */
5436 if (strcmp(mode, "rb") == 0 || strcmp(mode, "rt") == 0)
5437 mode = "r";
5438 else if (strcmp(mode, "wb") == 0 || strcmp(mode, "wt") == 0)
5439 mode = "w";
Barry Warsaw53699e91996-12-10 23:23:01 +00005440 Py_BEGIN_ALLOW_THREADS
Guido van Rossumef0a00e1992-01-27 16:51:30 +00005441 fp = popen(name, mode);
Barry Warsaw53699e91996-12-10 23:23:01 +00005442 Py_END_ALLOW_THREADS
Guido van Rossum3b066191991-06-04 19:40:25 +00005443 if (fp == NULL)
5444 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005445 f = PyFile_FromFile(fp, name, mode, pclose);
Guido van Rossuma6a1e531995-01-10 15:36:38 +00005446 if (f != NULL)
Barry Warsaw53699e91996-12-10 23:23:01 +00005447 PyFile_SetBufSize(f, bufsize);
Guido van Rossuma6a1e531995-01-10 15:36:38 +00005448 return f;
Guido van Rossum3b066191991-06-04 19:40:25 +00005449}
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005450
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00005451#endif /* PYOS_??? */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005452#endif /* HAVE_POPEN */
Guido van Rossum3b066191991-06-04 19:40:25 +00005453
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005454
Guido van Rossumb6775db1994-08-01 11:34:53 +00005455#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005456PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005457"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005458Set the current process's user id.");
5459
Barry Warsaw53699e91996-12-10 23:23:01 +00005460static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005461posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005462{
5463 int uid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005464 if (!PyArg_ParseTuple(args, "i:setuid", &uid))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005465 return NULL;
5466 if (setuid(uid) < 0)
5467 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005468 Py_INCREF(Py_None);
5469 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005470}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005471#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005472
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005473
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005474#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005475PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005476"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005477Set the current process's effective user id.");
5478
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005479static PyObject *
5480posix_seteuid (PyObject *self, PyObject *args)
5481{
5482 int euid;
5483 if (!PyArg_ParseTuple(args, "i", &euid)) {
5484 return NULL;
5485 } else if (seteuid(euid) < 0) {
5486 return posix_error();
5487 } else {
5488 Py_INCREF(Py_None);
5489 return Py_None;
5490 }
5491}
5492#endif /* HAVE_SETEUID */
5493
5494#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005495PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005496"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005497Set the current process's effective group id.");
5498
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005499static PyObject *
5500posix_setegid (PyObject *self, PyObject *args)
5501{
5502 int egid;
5503 if (!PyArg_ParseTuple(args, "i", &egid)) {
5504 return NULL;
5505 } else if (setegid(egid) < 0) {
5506 return posix_error();
5507 } else {
5508 Py_INCREF(Py_None);
5509 return Py_None;
5510 }
5511}
5512#endif /* HAVE_SETEGID */
5513
5514#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005515PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00005516"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005517Set the current process's real and effective user ids.");
5518
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005519static PyObject *
5520posix_setreuid (PyObject *self, PyObject *args)
5521{
5522 int ruid, euid;
5523 if (!PyArg_ParseTuple(args, "ii", &ruid, &euid)) {
5524 return NULL;
5525 } else if (setreuid(ruid, euid) < 0) {
5526 return posix_error();
5527 } else {
5528 Py_INCREF(Py_None);
5529 return Py_None;
5530 }
5531}
5532#endif /* HAVE_SETREUID */
5533
5534#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005535PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00005536"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005537Set the current process's real and effective group ids.");
5538
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005539static PyObject *
5540posix_setregid (PyObject *self, PyObject *args)
5541{
5542 int rgid, egid;
5543 if (!PyArg_ParseTuple(args, "ii", &rgid, &egid)) {
5544 return NULL;
5545 } else if (setregid(rgid, egid) < 0) {
5546 return posix_error();
5547 } else {
5548 Py_INCREF(Py_None);
5549 return Py_None;
5550 }
5551}
5552#endif /* HAVE_SETREGID */
5553
Guido van Rossumb6775db1994-08-01 11:34:53 +00005554#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005555PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005556"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005557Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005558
Barry Warsaw53699e91996-12-10 23:23:01 +00005559static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005560posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005561{
5562 int gid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005563 if (!PyArg_ParseTuple(args, "i:setgid", &gid))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005564 return NULL;
5565 if (setgid(gid) < 0)
5566 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005567 Py_INCREF(Py_None);
5568 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005569}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005570#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005571
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005572#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005573PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005574"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005575Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005576
5577static PyObject *
Georg Brandl96a8c392006-05-29 21:04:52 +00005578posix_setgroups(PyObject *self, PyObject *groups)
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005579{
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005580 int i, len;
5581 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00005582
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005583 if (!PySequence_Check(groups)) {
5584 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
5585 return NULL;
5586 }
5587 len = PySequence_Size(groups);
5588 if (len > MAX_GROUPS) {
5589 PyErr_SetString(PyExc_ValueError, "too many groups");
5590 return NULL;
5591 }
5592 for(i = 0; i < len; i++) {
5593 PyObject *elem;
5594 elem = PySequence_GetItem(groups, i);
5595 if (!elem)
5596 return NULL;
5597 if (!PyInt_Check(elem)) {
Georg Brandla13c2442005-11-22 19:30:31 +00005598 if (!PyLong_Check(elem)) {
5599 PyErr_SetString(PyExc_TypeError,
5600 "groups must be integers");
5601 Py_DECREF(elem);
5602 return NULL;
5603 } else {
5604 unsigned long x = PyLong_AsUnsignedLong(elem);
5605 if (PyErr_Occurred()) {
5606 PyErr_SetString(PyExc_TypeError,
5607 "group id too big");
5608 Py_DECREF(elem);
5609 return NULL;
5610 }
5611 grouplist[i] = x;
5612 /* read back the value to see if it fitted in gid_t */
5613 if (grouplist[i] != x) {
5614 PyErr_SetString(PyExc_TypeError,
5615 "group id too big");
5616 Py_DECREF(elem);
5617 return NULL;
5618 }
5619 }
5620 } else {
5621 long x = PyInt_AsLong(elem);
5622 grouplist[i] = x;
5623 if (grouplist[i] != x) {
5624 PyErr_SetString(PyExc_TypeError,
5625 "group id too big");
5626 Py_DECREF(elem);
5627 return NULL;
5628 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005629 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005630 Py_DECREF(elem);
5631 }
5632
5633 if (setgroups(len, grouplist) < 0)
5634 return posix_error();
5635 Py_INCREF(Py_None);
5636 return Py_None;
5637}
5638#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005639
Neal Norwitz6c2f9132006-03-20 07:25:26 +00005640#if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
Neal Norwitz05a45592006-03-20 06:30:08 +00005641static PyObject *
5642wait_helper(int pid, int status, struct rusage *ru)
5643{
5644 PyObject *result;
5645 static PyObject *struct_rusage;
5646
5647 if (pid == -1)
5648 return posix_error();
5649
5650 if (struct_rusage == NULL) {
5651 PyObject *m = PyImport_ImportModule("resource");
5652 if (m == NULL)
5653 return NULL;
5654 struct_rusage = PyObject_GetAttrString(m, "struct_rusage");
5655 Py_DECREF(m);
5656 if (struct_rusage == NULL)
5657 return NULL;
5658 }
5659
5660 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
5661 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
5662 if (!result)
5663 return NULL;
5664
5665#ifndef doubletime
5666#define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
5667#endif
5668
5669 PyStructSequence_SET_ITEM(result, 0,
5670 PyFloat_FromDouble(doubletime(ru->ru_utime)));
5671 PyStructSequence_SET_ITEM(result, 1,
5672 PyFloat_FromDouble(doubletime(ru->ru_stime)));
5673#define SET_INT(result, index, value)\
5674 PyStructSequence_SET_ITEM(result, index, PyInt_FromLong(value))
5675 SET_INT(result, 2, ru->ru_maxrss);
5676 SET_INT(result, 3, ru->ru_ixrss);
5677 SET_INT(result, 4, ru->ru_idrss);
5678 SET_INT(result, 5, ru->ru_isrss);
5679 SET_INT(result, 6, ru->ru_minflt);
5680 SET_INT(result, 7, ru->ru_majflt);
5681 SET_INT(result, 8, ru->ru_nswap);
5682 SET_INT(result, 9, ru->ru_inblock);
5683 SET_INT(result, 10, ru->ru_oublock);
5684 SET_INT(result, 11, ru->ru_msgsnd);
5685 SET_INT(result, 12, ru->ru_msgrcv);
5686 SET_INT(result, 13, ru->ru_nsignals);
5687 SET_INT(result, 14, ru->ru_nvcsw);
5688 SET_INT(result, 15, ru->ru_nivcsw);
5689#undef SET_INT
5690
5691 if (PyErr_Occurred()) {
5692 Py_DECREF(result);
5693 return NULL;
5694 }
5695
Neal Norwitz9b00a562006-03-20 08:47:12 +00005696 return Py_BuildValue("iiN", pid, status, result);
Neal Norwitz05a45592006-03-20 06:30:08 +00005697}
Neal Norwitz6c2f9132006-03-20 07:25:26 +00005698#endif /* HAVE_WAIT3 || HAVE_WAIT4 */
Neal Norwitz05a45592006-03-20 06:30:08 +00005699
5700#ifdef HAVE_WAIT3
5701PyDoc_STRVAR(posix_wait3__doc__,
5702"wait3(options) -> (pid, status, rusage)\n\n\
5703Wait for completion of a child process.");
5704
5705static PyObject *
5706posix_wait3(PyObject *self, PyObject *args)
5707{
5708 int pid, options;
5709 struct rusage ru;
Neal Norwitzd5a37542006-03-20 06:48:34 +00005710 WAIT_TYPE status;
5711 WAIT_STATUS_INT(status) = 0;
Neal Norwitz05a45592006-03-20 06:30:08 +00005712
5713 if (!PyArg_ParseTuple(args, "i:wait3", &options))
5714 return NULL;
5715
5716 Py_BEGIN_ALLOW_THREADS
5717 pid = wait3(&status, options, &ru);
5718 Py_END_ALLOW_THREADS
5719
Neal Norwitzd5a37542006-03-20 06:48:34 +00005720 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Neal Norwitz05a45592006-03-20 06:30:08 +00005721}
5722#endif /* HAVE_WAIT3 */
5723
5724#ifdef HAVE_WAIT4
5725PyDoc_STRVAR(posix_wait4__doc__,
5726"wait4(pid, options) -> (pid, status, rusage)\n\n\
5727Wait for completion of a given child process.");
5728
5729static PyObject *
5730posix_wait4(PyObject *self, PyObject *args)
5731{
5732 int pid, options;
5733 struct rusage ru;
Neal Norwitzd5a37542006-03-20 06:48:34 +00005734 WAIT_TYPE status;
5735 WAIT_STATUS_INT(status) = 0;
Neal Norwitz05a45592006-03-20 06:30:08 +00005736
5737 if (!PyArg_ParseTuple(args, "ii:wait4", &pid, &options))
5738 return NULL;
5739
5740 Py_BEGIN_ALLOW_THREADS
5741 pid = wait4(pid, &status, options, &ru);
5742 Py_END_ALLOW_THREADS
5743
Neal Norwitzd5a37542006-03-20 06:48:34 +00005744 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Neal Norwitz05a45592006-03-20 06:30:08 +00005745}
5746#endif /* HAVE_WAIT4 */
5747
Guido van Rossumb6775db1994-08-01 11:34:53 +00005748#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005749PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005750"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005751Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005752
Barry Warsaw53699e91996-12-10 23:23:01 +00005753static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005754posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005755{
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005756 int pid, options;
Neal Norwitzd5a37542006-03-20 06:48:34 +00005757 WAIT_TYPE status;
5758 WAIT_STATUS_INT(status) = 0;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005759
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005760 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
Guido van Rossum21803b81992-08-09 12:55:27 +00005761 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005762 Py_BEGIN_ALLOW_THREADS
Guido van Rossume6a3aa61999-02-01 16:15:30 +00005763 pid = waitpid(pid, &status, options);
Barry Warsaw53699e91996-12-10 23:23:01 +00005764 Py_END_ALLOW_THREADS
Guido van Rossum85e3b011991-06-03 12:42:10 +00005765 if (pid == -1)
5766 return posix_error();
Neal Norwitzd5a37542006-03-20 06:48:34 +00005767
5768 return Py_BuildValue("ii", pid, WAIT_STATUS_INT(status));
Guido van Rossum21803b81992-08-09 12:55:27 +00005769}
5770
Tim Petersab034fa2002-02-01 11:27:43 +00005771#elif defined(HAVE_CWAIT)
5772
5773/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005774PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005775"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005776"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00005777
5778static PyObject *
5779posix_waitpid(PyObject *self, PyObject *args)
5780{
Martin v. Löwisa43190b2006-05-22 09:15:18 +00005781 Py_intptr_t pid;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005782 int status, options;
Tim Petersab034fa2002-02-01 11:27:43 +00005783
5784 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
5785 return NULL;
5786 Py_BEGIN_ALLOW_THREADS
5787 pid = _cwait(&status, pid, options);
5788 Py_END_ALLOW_THREADS
5789 if (pid == -1)
5790 return posix_error();
Neal Norwitzd5a37542006-03-20 06:48:34 +00005791
5792 /* shift the status left a byte so this is more like the POSIX waitpid */
5793 return Py_BuildValue("ii", pid, status << 8);
Tim Petersab034fa2002-02-01 11:27:43 +00005794}
5795#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005796
Guido van Rossumad0ee831995-03-01 10:34:45 +00005797#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005798PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005799"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005800Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005801
Barry Warsaw53699e91996-12-10 23:23:01 +00005802static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005803posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00005804{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005805 int pid;
Neal Norwitzd5a37542006-03-20 06:48:34 +00005806 WAIT_TYPE status;
5807 WAIT_STATUS_INT(status) = 0;
Neal Norwitze241ce82003-02-17 18:17:05 +00005808
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005809 Py_BEGIN_ALLOW_THREADS
5810 pid = wait(&status);
Barry Warsaw53699e91996-12-10 23:23:01 +00005811 Py_END_ALLOW_THREADS
Guido van Rossum21803b81992-08-09 12:55:27 +00005812 if (pid == -1)
5813 return posix_error();
Neal Norwitzd5a37542006-03-20 06:48:34 +00005814
5815 return Py_BuildValue("ii", pid, WAIT_STATUS_INT(status));
Guido van Rossum85e3b011991-06-03 12:42:10 +00005816}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005817#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00005818
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005819
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005820PyDoc_STRVAR(posix_lstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005821"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005822Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005823
Barry Warsaw53699e91996-12-10 23:23:01 +00005824static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005825posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005826{
Guido van Rossumb6775db1994-08-01 11:34:53 +00005827#ifdef HAVE_LSTAT
Mark Hammondc2e85bd2002-10-03 05:10:39 +00005828 return posix_do_stat(self, args, "et:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00005829#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00005830#ifdef MS_WINDOWS
Martin v. Löwis14694662006-02-03 12:54:16 +00005831 return posix_do_stat(self, args, "et:lstat", STAT, "U:lstat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00005832#else
5833 return posix_do_stat(self, args, "et:lstat", STAT, NULL, NULL);
5834#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00005835#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005836}
5837
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005838
Guido van Rossumb6775db1994-08-01 11:34:53 +00005839#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005840PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005841"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005842Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005843
Barry Warsaw53699e91996-12-10 23:23:01 +00005844static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005845posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005846{
Ronald Oussoren10168f22006-10-22 10:45:18 +00005847 PyObject* v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00005848 char buf[MAXPATHLEN];
Guido van Rossumef0a00e1992-01-27 16:51:30 +00005849 char *path;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005850 int n;
Ronald Oussoren10168f22006-10-22 10:45:18 +00005851#ifdef Py_USING_UNICODE
5852 int arg_is_unicode = 0;
5853#endif
5854
5855 if (!PyArg_ParseTuple(args, "et:readlink",
5856 Py_FileSystemDefaultEncoding, &path))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005857 return NULL;
Ronald Oussoren10168f22006-10-22 10:45:18 +00005858#ifdef Py_USING_UNICODE
5859 v = PySequence_GetItem(args, 0);
Neal Norwitz91a57212007-08-12 17:11:13 +00005860 if (v == NULL) {
5861 PyMem_Free(path);
5862 return NULL;
5863 }
Ronald Oussoren10168f22006-10-22 10:45:18 +00005864
5865 if (PyUnicode_Check(v)) {
5866 arg_is_unicode = 1;
5867 }
5868 Py_DECREF(v);
5869#endif
5870
Barry Warsaw53699e91996-12-10 23:23:01 +00005871 Py_BEGIN_ALLOW_THREADS
Guido van Rossum50e61dc1992-03-27 17:22:31 +00005872 n = readlink(path, buf, (int) sizeof buf);
Barry Warsaw53699e91996-12-10 23:23:01 +00005873 Py_END_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005874 if (n < 0)
Neal Norwitz91a57212007-08-12 17:11:13 +00005875 return posix_error_with_allocated_filename(path);
Ronald Oussoren10168f22006-10-22 10:45:18 +00005876
Neal Norwitz91a57212007-08-12 17:11:13 +00005877 PyMem_Free(path);
Ronald Oussoren10168f22006-10-22 10:45:18 +00005878 v = PyString_FromStringAndSize(buf, n);
5879#ifdef Py_USING_UNICODE
5880 if (arg_is_unicode) {
5881 PyObject *w;
5882
5883 w = PyUnicode_FromEncodedObject(v,
5884 Py_FileSystemDefaultEncoding,
5885 "strict");
5886 if (w != NULL) {
5887 Py_DECREF(v);
5888 v = w;
5889 }
5890 else {
5891 /* fall back to the original byte string, as
5892 discussed in patch #683592 */
5893 PyErr_Clear();
5894 }
5895 }
5896#endif
5897 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005898}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005899#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005900
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005901
Guido van Rossumb6775db1994-08-01 11:34:53 +00005902#ifdef HAVE_SYMLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005903PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005904"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00005905Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005906
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005907static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005908posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005909{
Martin v. Löwis4fc2bda2006-05-04 12:04:27 +00005910 return posix_2str(args, "etet:symlink", symlink);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005911}
5912#endif /* HAVE_SYMLINK */
5913
5914
5915#ifdef HAVE_TIMES
5916#ifndef HZ
5917#define HZ 60 /* Universal constant :-) */
5918#endif /* HZ */
Tim Peters5aa91602002-01-30 05:46:57 +00005919
Guido van Rossumd48f2521997-12-05 22:19:34 +00005920#if defined(PYCC_VACPP) && defined(PYOS_OS2)
5921static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00005922system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005923{
5924 ULONG value = 0;
5925
5926 Py_BEGIN_ALLOW_THREADS
5927 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
5928 Py_END_ALLOW_THREADS
5929
5930 return value;
5931}
5932
5933static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005934posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005935{
Guido van Rossumd48f2521997-12-05 22:19:34 +00005936 /* Currently Only Uptime is Provided -- Others Later */
5937 return Py_BuildValue("ddddd",
5938 (double)0 /* t.tms_utime / HZ */,
5939 (double)0 /* t.tms_stime / HZ */,
5940 (double)0 /* t.tms_cutime / HZ */,
5941 (double)0 /* t.tms_cstime / HZ */,
5942 (double)system_uptime() / 1000);
5943}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005944#else /* not OS2 */
Barry Warsaw53699e91996-12-10 23:23:01 +00005945static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005946posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00005947{
5948 struct tms t;
5949 clock_t c;
Guido van Rossum22db57e1992-04-05 14:25:30 +00005950 errno = 0;
5951 c = times(&t);
Guido van Rossum687dd131993-05-17 08:34:16 +00005952 if (c == (clock_t) -1)
5953 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005954 return Py_BuildValue("ddddd",
Barry Warsaw43d68b81996-12-19 22:10:44 +00005955 (double)t.tms_utime / HZ,
5956 (double)t.tms_stime / HZ,
5957 (double)t.tms_cutime / HZ,
5958 (double)t.tms_cstime / HZ,
5959 (double)c / HZ);
Guido van Rossum22db57e1992-04-05 14:25:30 +00005960}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005961#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00005962#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005963
5964
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005965#ifdef MS_WINDOWS
Guido van Rossum14ed0b21994-09-29 09:50:09 +00005966#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00005967static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005968posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00005969{
5970 FILETIME create, exit, kernel, user;
5971 HANDLE hProc;
Guido van Rossum14ed0b21994-09-29 09:50:09 +00005972 hProc = GetCurrentProcess();
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00005973 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
5974 /* The fields of a FILETIME structure are the hi and lo part
5975 of a 64-bit value expressed in 100 nanosecond units.
5976 1e7 is one second in such units; 1e-7 the inverse.
5977 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
5978 */
Barry Warsaw53699e91996-12-10 23:23:01 +00005979 return Py_BuildValue(
5980 "ddddd",
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00005981 (double)(kernel.dwHighDateTime*429.4967296 +
5982 kernel.dwLowDateTime*1e-7),
5983 (double)(user.dwHighDateTime*429.4967296 +
5984 user.dwLowDateTime*1e-7),
Barry Warsaw53699e91996-12-10 23:23:01 +00005985 (double)0,
5986 (double)0,
5987 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00005988}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005989#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005990
5991#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005992PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005993"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005994Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005995#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00005996
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005997
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00005998#ifdef HAVE_GETSID
5999PyDoc_STRVAR(posix_getsid__doc__,
6000"getsid(pid) -> sid\n\n\
6001Call the system call getsid().");
6002
6003static PyObject *
6004posix_getsid(PyObject *self, PyObject *args)
6005{
6006 int pid, sid;
6007 if (!PyArg_ParseTuple(args, "i:getsid", &pid))
6008 return NULL;
6009 sid = getsid(pid);
6010 if (sid < 0)
6011 return posix_error();
6012 return PyInt_FromLong((long)sid);
6013}
6014#endif /* HAVE_GETSID */
6015
6016
Guido van Rossumb6775db1994-08-01 11:34:53 +00006017#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006018PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006019"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006020Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006021
Barry Warsaw53699e91996-12-10 23:23:01 +00006022static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006023posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00006024{
Guido van Rossum687dd131993-05-17 08:34:16 +00006025 if (setsid() < 0)
6026 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00006027 Py_INCREF(Py_None);
6028 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00006029}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006030#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00006031
Guido van Rossumb6775db1994-08-01 11:34:53 +00006032#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006033PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006034"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006035Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006036
Barry Warsaw53699e91996-12-10 23:23:01 +00006037static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006038posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00006039{
6040 int pid, pgrp;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006041 if (!PyArg_ParseTuple(args, "ii:setpgid", &pid, &pgrp))
Guido van Rossumc2670a01992-09-13 20:07:29 +00006042 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00006043 if (setpgid(pid, pgrp) < 0)
6044 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00006045 Py_INCREF(Py_None);
6046 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00006047}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006048#endif /* HAVE_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00006049
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006050
Guido van Rossumb6775db1994-08-01 11:34:53 +00006051#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006052PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006053"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006054Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006055
Barry Warsaw53699e91996-12-10 23:23:01 +00006056static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006057posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00006058{
6059 int fd, pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006060 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
Guido van Rossum7066dd71992-09-17 17:54:56 +00006061 return NULL;
6062 pgid = tcgetpgrp(fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006063 if (pgid < 0)
6064 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00006065 return PyInt_FromLong((long)pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00006066}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006067#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00006068
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006069
Guido van Rossumb6775db1994-08-01 11:34:53 +00006070#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006071PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006072"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006073Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006074
Barry Warsaw53699e91996-12-10 23:23:01 +00006075static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006076posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00006077{
6078 int fd, pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006079 if (!PyArg_ParseTuple(args, "ii:tcsetpgrp", &fd, &pgid))
Guido van Rossum7066dd71992-09-17 17:54:56 +00006080 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00006081 if (tcsetpgrp(fd, pgid) < 0)
6082 return posix_error();
Barry Warsaw43d68b81996-12-19 22:10:44 +00006083 Py_INCREF(Py_None);
Barry Warsaw53699e91996-12-10 23:23:01 +00006084 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00006085}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006086#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00006087
Guido van Rossum687dd131993-05-17 08:34:16 +00006088/* Functions acting on file descriptors */
6089
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006090PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006091"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006092Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006093
Barry Warsaw53699e91996-12-10 23:23:01 +00006094static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006095posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006096{
Mark Hammondef8b6542001-05-13 08:04:26 +00006097 char *file = NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00006098 int flag;
6099 int mode = 0777;
6100 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006101
6102#ifdef MS_WINDOWS
6103 if (unicode_file_names()) {
6104 PyUnicodeObject *po;
6105 if (PyArg_ParseTuple(args, "Ui|i:mkdir", &po, &flag, &mode)) {
6106 Py_BEGIN_ALLOW_THREADS
6107 /* PyUnicode_AS_UNICODE OK without thread
6108 lock as it is a simple dereference. */
6109 fd = _wopen(PyUnicode_AS_UNICODE(po), flag, mode);
6110 Py_END_ALLOW_THREADS
6111 if (fd < 0)
6112 return posix_error();
6113 return PyInt_FromLong((long)fd);
6114 }
6115 /* Drop the argument parsing error as narrow strings
6116 are also valid. */
6117 PyErr_Clear();
6118 }
6119#endif
6120
Tim Peters5aa91602002-01-30 05:46:57 +00006121 if (!PyArg_ParseTuple(args, "eti|i",
Mark Hammondef8b6542001-05-13 08:04:26 +00006122 Py_FileSystemDefaultEncoding, &file,
6123 &flag, &mode))
Barry Warsaw43d68b81996-12-19 22:10:44 +00006124 return NULL;
6125
Barry Warsaw53699e91996-12-10 23:23:01 +00006126 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006127 fd = open(file, flag, mode);
Barry Warsaw53699e91996-12-10 23:23:01 +00006128 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006129 if (fd < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00006130 return posix_error_with_allocated_filename(file);
6131 PyMem_Free(file);
Barry Warsaw53699e91996-12-10 23:23:01 +00006132 return PyInt_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006133}
6134
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006135
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006136PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006137"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006138Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006139
Barry Warsaw53699e91996-12-10 23:23:01 +00006140static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006141posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006142{
6143 int fd, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006144 if (!PyArg_ParseTuple(args, "i:close", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00006145 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00006146 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006147 res = close(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00006148 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006149 if (res < 0)
6150 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00006151 Py_INCREF(Py_None);
6152 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006153}
6154
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006155
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006156PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006157"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006158Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006159
Barry Warsaw53699e91996-12-10 23:23:01 +00006160static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006161posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006162{
6163 int fd;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006164 if (!PyArg_ParseTuple(args, "i:dup", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00006165 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00006166 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006167 fd = dup(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00006168 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006169 if (fd < 0)
6170 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00006171 return PyInt_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006172}
6173
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006174
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006175PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00006176"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006177Duplicate file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006178
Barry Warsaw53699e91996-12-10 23:23:01 +00006179static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006180posix_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006181{
6182 int fd, fd2, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006183 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
Guido van Rossum687dd131993-05-17 08:34:16 +00006184 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00006185 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006186 res = dup2(fd, fd2);
Barry Warsaw53699e91996-12-10 23:23:01 +00006187 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006188 if (res < 0)
6189 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00006190 Py_INCREF(Py_None);
6191 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006192}
6193
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006194
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006195PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006196"lseek(fd, pos, how) -> newpos\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006197Set the current position of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006198
Barry Warsaw53699e91996-12-10 23:23:01 +00006199static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006200posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006201{
6202 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006203#if defined(MS_WIN64) || defined(MS_WINDOWS)
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00006204 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006205#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00006206 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006207#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00006208 PyObject *posobj;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006209 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Guido van Rossum687dd131993-05-17 08:34:16 +00006210 return NULL;
6211#ifdef SEEK_SET
6212 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
6213 switch (how) {
6214 case 0: how = SEEK_SET; break;
6215 case 1: how = SEEK_CUR; break;
6216 case 2: how = SEEK_END; break;
6217 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006218#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00006219
6220#if !defined(HAVE_LARGEFILE_SUPPORT)
6221 pos = PyInt_AsLong(posobj);
6222#else
6223 pos = PyLong_Check(posobj) ?
6224 PyLong_AsLongLong(posobj) : PyInt_AsLong(posobj);
6225#endif
6226 if (PyErr_Occurred())
6227 return NULL;
6228
Barry Warsaw53699e91996-12-10 23:23:01 +00006229 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006230#if defined(MS_WIN64) || defined(MS_WINDOWS)
Fred Drake699f3522000-06-29 21:12:41 +00006231 res = _lseeki64(fd, pos, how);
6232#else
Guido van Rossum687dd131993-05-17 08:34:16 +00006233 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00006234#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00006235 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006236 if (res < 0)
6237 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00006238
6239#if !defined(HAVE_LARGEFILE_SUPPORT)
Barry Warsaw53699e91996-12-10 23:23:01 +00006240 return PyInt_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006241#else
6242 return PyLong_FromLongLong(res);
6243#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00006244}
6245
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006246
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006247PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006248"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006249Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006250
Barry Warsaw53699e91996-12-10 23:23:01 +00006251static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006252posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006253{
Guido van Rossum8bac5461996-06-11 18:38:48 +00006254 int fd, size, n;
Barry Warsaw53699e91996-12-10 23:23:01 +00006255 PyObject *buffer;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006256 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
Guido van Rossum687dd131993-05-17 08:34:16 +00006257 return NULL;
Michael W. Hudson9867ced2005-01-31 17:01:59 +00006258 if (size < 0) {
6259 errno = EINVAL;
6260 return posix_error();
6261 }
Barry Warsaw53699e91996-12-10 23:23:01 +00006262 buffer = PyString_FromStringAndSize((char *)NULL, size);
Guido van Rossum687dd131993-05-17 08:34:16 +00006263 if (buffer == NULL)
6264 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00006265 Py_BEGIN_ALLOW_THREADS
6266 n = read(fd, PyString_AsString(buffer), size);
6267 Py_END_ALLOW_THREADS
Guido van Rossum8bac5461996-06-11 18:38:48 +00006268 if (n < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00006269 Py_DECREF(buffer);
Guido van Rossum687dd131993-05-17 08:34:16 +00006270 return posix_error();
6271 }
Guido van Rossum8bac5461996-06-11 18:38:48 +00006272 if (n != size)
Barry Warsaw53699e91996-12-10 23:23:01 +00006273 _PyString_Resize(&buffer, n);
Guido van Rossum687dd131993-05-17 08:34:16 +00006274 return buffer;
6275}
6276
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006277
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006278PyDoc_STRVAR(posix_write__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006279"write(fd, string) -> byteswritten\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006280Write a string to a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006281
Barry Warsaw53699e91996-12-10 23:23:01 +00006282static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006283posix_write(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006284{
Thomas Wouters68bc4f92006-03-01 01:05:10 +00006285 int fd;
6286 Py_ssize_t size;
Guido van Rossum687dd131993-05-17 08:34:16 +00006287 char *buffer;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00006288
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006289 if (!PyArg_ParseTuple(args, "is#:write", &fd, &buffer, &size))
Guido van Rossum687dd131993-05-17 08:34:16 +00006290 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00006291 Py_BEGIN_ALLOW_THREADS
Thomas Wouters68bc4f92006-03-01 01:05:10 +00006292 size = write(fd, buffer, (size_t)size);
Barry Warsaw53699e91996-12-10 23:23:01 +00006293 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006294 if (size < 0)
6295 return posix_error();
Thomas Wouters68bc4f92006-03-01 01:05:10 +00006296 return PyInt_FromSsize_t(size);
Guido van Rossum687dd131993-05-17 08:34:16 +00006297}
6298
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006299
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006300PyDoc_STRVAR(posix_fstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006301"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006302Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006303
Barry Warsaw53699e91996-12-10 23:23:01 +00006304static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006305posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006306{
6307 int fd;
Fred Drake699f3522000-06-29 21:12:41 +00006308 STRUCT_STAT st;
Guido van Rossum687dd131993-05-17 08:34:16 +00006309 int res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006310 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00006311 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00006312#ifdef __VMS
6313 /* on OpenVMS we must ensure that all bytes are written to the file */
6314 fsync(fd);
6315#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00006316 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00006317 res = FSTAT(fd, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00006318 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00006319 if (res != 0) {
6320#ifdef MS_WINDOWS
6321 return win32_error("fstat", NULL);
6322#else
Guido van Rossum687dd131993-05-17 08:34:16 +00006323 return posix_error();
Martin v. Löwis14694662006-02-03 12:54:16 +00006324#endif
6325 }
Tim Peters5aa91602002-01-30 05:46:57 +00006326
Martin v. Löwis14694662006-02-03 12:54:16 +00006327 return _pystat_fromstructstat(&st);
Guido van Rossum687dd131993-05-17 08:34:16 +00006328}
6329
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006330
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006331PyDoc_STRVAR(posix_fdopen__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00006332"fdopen(fd [, mode='r' [, bufsize]]) -> file_object\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006333Return an open file object connected to a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006334
Barry Warsaw53699e91996-12-10 23:23:01 +00006335static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006336posix_fdopen(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006337{
Guido van Rossum687dd131993-05-17 08:34:16 +00006338 int fd;
Kristján Valur Jónsson0a440d42007-04-26 09:15:08 +00006339 char *orgmode = "r";
Guido van Rossuma6a1e531995-01-10 15:36:38 +00006340 int bufsize = -1;
Guido van Rossum687dd131993-05-17 08:34:16 +00006341 FILE *fp;
Barry Warsaw53699e91996-12-10 23:23:01 +00006342 PyObject *f;
Kristján Valur Jónsson0a440d42007-04-26 09:15:08 +00006343 char *mode;
6344 if (!PyArg_ParseTuple(args, "i|si", &fd, &orgmode, &bufsize))
Guido van Rossum687dd131993-05-17 08:34:16 +00006345 return NULL;
Barry Warsaw43d68b81996-12-19 22:10:44 +00006346
Kristján Valur Jónsson0a440d42007-04-26 09:15:08 +00006347 /* Sanitize mode. See fileobject.c */
6348 mode = PyMem_MALLOC(strlen(orgmode)+3);
6349 if (!mode) {
6350 PyErr_NoMemory();
6351 return NULL;
6352 }
6353 strcpy(mode, orgmode);
6354 if (_PyFile_SanitizeMode(mode)) {
6355 PyMem_FREE(mode);
Thomas Heller1f043e22002-11-07 16:00:59 +00006356 return NULL;
6357 }
Barry Warsaw53699e91996-12-10 23:23:01 +00006358 Py_BEGIN_ALLOW_THREADS
Georg Brandl644b1e72006-03-31 20:27:22 +00006359#if !defined(MS_WINDOWS) && defined(HAVE_FCNTL_H)
Georg Brandl54a188a2006-03-31 20:00:11 +00006360 if (mode[0] == 'a') {
6361 /* try to make sure the O_APPEND flag is set */
6362 int flags;
6363 flags = fcntl(fd, F_GETFL);
6364 if (flags != -1)
6365 fcntl(fd, F_SETFL, flags | O_APPEND);
6366 fp = fdopen(fd, mode);
Thomas Wouters2a9a6b02006-03-31 22:38:19 +00006367 if (fp == NULL && flags != -1)
Georg Brandl54a188a2006-03-31 20:00:11 +00006368 /* restore old mode if fdopen failed */
6369 fcntl(fd, F_SETFL, flags);
6370 } else {
6371 fp = fdopen(fd, mode);
6372 }
Georg Brandl644b1e72006-03-31 20:27:22 +00006373#else
6374 fp = fdopen(fd, mode);
6375#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00006376 Py_END_ALLOW_THREADS
Neal Norwitzd83eb312007-05-02 04:47:55 +00006377 PyMem_FREE(mode);
Guido van Rossum687dd131993-05-17 08:34:16 +00006378 if (fp == NULL)
6379 return posix_error();
Kristján Valur Jónsson0a440d42007-04-26 09:15:08 +00006380 f = PyFile_FromFile(fp, "<fdopen>", orgmode, fclose);
Guido van Rossuma6a1e531995-01-10 15:36:38 +00006381 if (f != NULL)
Barry Warsaw53699e91996-12-10 23:23:01 +00006382 PyFile_SetBufSize(f, bufsize);
Guido van Rossuma6a1e531995-01-10 15:36:38 +00006383 return f;
Guido van Rossum687dd131993-05-17 08:34:16 +00006384}
6385
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006386PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006387"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00006388Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006389connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00006390
6391static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00006392posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00006393{
6394 int fd;
6395 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
6396 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00006397 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00006398}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006399
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006400#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006401PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006402"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006403Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006404
Barry Warsaw53699e91996-12-10 23:23:01 +00006405static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006406posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00006407{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00006408#if defined(PYOS_OS2)
6409 HFILE read, write;
6410 APIRET rc;
6411
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00006412 Py_BEGIN_ALLOW_THREADS
6413 rc = DosCreatePipe( &read, &write, 4096);
6414 Py_END_ALLOW_THREADS
6415 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00006416 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00006417
6418 return Py_BuildValue("(ii)", read, write);
6419#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006420#if !defined(MS_WINDOWS)
Guido van Rossum687dd131993-05-17 08:34:16 +00006421 int fds[2];
6422 int res;
Barry Warsaw53699e91996-12-10 23:23:01 +00006423 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006424 res = pipe(fds);
Barry Warsaw53699e91996-12-10 23:23:01 +00006425 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00006426 if (res != 0)
6427 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00006428 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006429#else /* MS_WINDOWS */
Guido van Rossum794d8131994-08-23 13:48:48 +00006430 HANDLE read, write;
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00006431 int read_fd, write_fd;
Guido van Rossum794d8131994-08-23 13:48:48 +00006432 BOOL ok;
Barry Warsaw53699e91996-12-10 23:23:01 +00006433 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00006434 ok = CreatePipe(&read, &write, NULL, 0);
Barry Warsaw53699e91996-12-10 23:23:01 +00006435 Py_END_ALLOW_THREADS
Guido van Rossum794d8131994-08-23 13:48:48 +00006436 if (!ok)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00006437 return win32_error("CreatePipe", NULL);
Tim Peters79248aa2001-08-29 21:37:10 +00006438 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
6439 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00006440 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006441#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00006442#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00006443}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006444#endif /* HAVE_PIPE */
6445
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006446
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006447#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006448PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00006449"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006450Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006451
Barry Warsaw53699e91996-12-10 23:23:01 +00006452static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006453posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006454{
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006455 char *filename;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006456 int mode = 0666;
6457 int res;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006458 if (!PyArg_ParseTuple(args, "s|i:mkfifo", &filename, &mode))
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006459 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00006460 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006461 res = mkfifo(filename, mode);
6462 Py_END_ALLOW_THREADS
6463 if (res < 0)
6464 return posix_error();
6465 Py_INCREF(Py_None);
6466 return Py_None;
6467}
6468#endif
6469
6470
Neal Norwitz11690112002-07-30 01:08:28 +00006471#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006472PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00006473"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006474Create a filesystem node (file, device special file or named pipe)\n\
6475named filename. mode specifies both the permissions to use and the\n\
6476type of node to be created, being combined (bitwise OR) with one of\n\
6477S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006478device defines the newly created device special file (probably using\n\
6479os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006480
6481
6482static PyObject *
6483posix_mknod(PyObject *self, PyObject *args)
6484{
6485 char *filename;
6486 int mode = 0600;
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006487 int device = 0;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006488 int res;
Martin v. Löwisd631ebe2002-11-02 17:42:33 +00006489 if (!PyArg_ParseTuple(args, "s|ii:mknod", &filename, &mode, &device))
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006490 return NULL;
6491 Py_BEGIN_ALLOW_THREADS
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006492 res = mknod(filename, mode, device);
Barry Warsaw53699e91996-12-10 23:23:01 +00006493 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006494 if (res < 0)
6495 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00006496 Py_INCREF(Py_None);
6497 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006498}
6499#endif
6500
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006501#ifdef HAVE_DEVICE_MACROS
6502PyDoc_STRVAR(posix_major__doc__,
6503"major(device) -> major number\n\
6504Extracts a device major number from a raw device number.");
6505
6506static PyObject *
6507posix_major(PyObject *self, PyObject *args)
6508{
6509 int device;
6510 if (!PyArg_ParseTuple(args, "i:major", &device))
6511 return NULL;
6512 return PyInt_FromLong((long)major(device));
6513}
6514
6515PyDoc_STRVAR(posix_minor__doc__,
6516"minor(device) -> minor number\n\
6517Extracts a device minor number from a raw device number.");
6518
6519static PyObject *
6520posix_minor(PyObject *self, PyObject *args)
6521{
6522 int device;
6523 if (!PyArg_ParseTuple(args, "i:minor", &device))
6524 return NULL;
6525 return PyInt_FromLong((long)minor(device));
6526}
6527
6528PyDoc_STRVAR(posix_makedev__doc__,
6529"makedev(major, minor) -> device number\n\
6530Composes a raw device number from the major and minor device numbers.");
6531
6532static PyObject *
6533posix_makedev(PyObject *self, PyObject *args)
6534{
6535 int major, minor;
6536 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
6537 return NULL;
6538 return PyInt_FromLong((long)makedev(major, minor));
6539}
6540#endif /* device macros */
6541
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006542
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006543#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006544PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006545"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006546Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006547
Barry Warsaw53699e91996-12-10 23:23:01 +00006548static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006549posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006550{
6551 int fd;
Guido van Rossum94f6f721999-01-06 18:42:14 +00006552 off_t length;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006553 int res;
Guido van Rossum94f6f721999-01-06 18:42:14 +00006554 PyObject *lenobj;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006555
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006556 if (!PyArg_ParseTuple(args, "iO:ftruncate", &fd, &lenobj))
Guido van Rossum94f6f721999-01-06 18:42:14 +00006557 return NULL;
6558
6559#if !defined(HAVE_LARGEFILE_SUPPORT)
6560 length = PyInt_AsLong(lenobj);
6561#else
6562 length = PyLong_Check(lenobj) ?
6563 PyLong_AsLongLong(lenobj) : PyInt_AsLong(lenobj);
6564#endif
6565 if (PyErr_Occurred())
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006566 return NULL;
6567
Barry Warsaw53699e91996-12-10 23:23:01 +00006568 Py_BEGIN_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006569 res = ftruncate(fd, length);
Barry Warsaw53699e91996-12-10 23:23:01 +00006570 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006571 if (res < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00006572 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006573 return NULL;
6574 }
Barry Warsaw53699e91996-12-10 23:23:01 +00006575 Py_INCREF(Py_None);
6576 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006577}
6578#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00006579
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006580#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006581PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006582"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006583Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006584
Fred Drake762e2061999-08-26 17:23:54 +00006585/* Save putenv() parameters as values here, so we can collect them when they
6586 * get re-set with another call for the same key. */
6587static PyObject *posix_putenv_garbage;
6588
Tim Peters5aa91602002-01-30 05:46:57 +00006589static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006590posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006591{
6592 char *s1, *s2;
Anthony Baxter64182fe2006-04-11 12:14:09 +00006593 char *newenv;
Fred Drake762e2061999-08-26 17:23:54 +00006594 PyObject *newstr;
Tim Petersc8996f52001-12-03 20:41:00 +00006595 size_t len;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006596
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006597 if (!PyArg_ParseTuple(args, "ss:putenv", &s1, &s2))
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006598 return NULL;
Guido van Rossumd48f2521997-12-05 22:19:34 +00006599
6600#if defined(PYOS_OS2)
6601 if (stricmp(s1, "BEGINLIBPATH") == 0) {
6602 APIRET rc;
6603
Guido van Rossumd48f2521997-12-05 22:19:34 +00006604 rc = DosSetExtLIBPATH(s2, BEGIN_LIBPATH);
6605 if (rc != NO_ERROR)
6606 return os2_error(rc);
6607
6608 } else if (stricmp(s1, "ENDLIBPATH") == 0) {
6609 APIRET rc;
6610
Guido van Rossumd48f2521997-12-05 22:19:34 +00006611 rc = DosSetExtLIBPATH(s2, END_LIBPATH);
6612 if (rc != NO_ERROR)
6613 return os2_error(rc);
6614 } else {
6615#endif
6616
Fred Drake762e2061999-08-26 17:23:54 +00006617 /* XXX This can leak memory -- not easy to fix :-( */
Tim Petersc8996f52001-12-03 20:41:00 +00006618 len = strlen(s1) + strlen(s2) + 2;
6619 /* len includes space for a trailing \0; the size arg to
6620 PyString_FromStringAndSize does not count that */
6621 newstr = PyString_FromStringAndSize(NULL, (int)len - 1);
Fred Drake762e2061999-08-26 17:23:54 +00006622 if (newstr == NULL)
6623 return PyErr_NoMemory();
Anthony Baxter64182fe2006-04-11 12:14:09 +00006624 newenv = PyString_AS_STRING(newstr);
6625 PyOS_snprintf(newenv, len, "%s=%s", s1, s2);
6626 if (putenv(newenv)) {
Neal Norwitz4adc9ab2003-02-10 03:10:43 +00006627 Py_DECREF(newstr);
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006628 posix_error();
6629 return NULL;
6630 }
Fred Drake762e2061999-08-26 17:23:54 +00006631 /* Install the first arg and newstr in posix_putenv_garbage;
6632 * this will cause previous value to be collected. This has to
6633 * happen after the real putenv() call because the old value
6634 * was still accessible until then. */
6635 if (PyDict_SetItem(posix_putenv_garbage,
6636 PyTuple_GET_ITEM(args, 0), newstr)) {
6637 /* really not much we can do; just leak */
6638 PyErr_Clear();
6639 }
6640 else {
6641 Py_DECREF(newstr);
6642 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00006643
6644#if defined(PYOS_OS2)
6645 }
6646#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00006647 Py_INCREF(Py_None);
6648 return Py_None;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006649}
Guido van Rossumb6a47161997-09-15 22:54:34 +00006650#endif /* putenv */
6651
Guido van Rossumc524d952001-10-19 01:31:59 +00006652#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006653PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006654"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006655Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00006656
6657static PyObject *
6658posix_unsetenv(PyObject *self, PyObject *args)
6659{
6660 char *s1;
6661
6662 if (!PyArg_ParseTuple(args, "s:unsetenv", &s1))
6663 return NULL;
6664
6665 unsetenv(s1);
6666
6667 /* Remove the key from posix_putenv_garbage;
6668 * this will cause it to be collected. This has to
Tim Peters5aa91602002-01-30 05:46:57 +00006669 * happen after the real unsetenv() call because the
Guido van Rossumc524d952001-10-19 01:31:59 +00006670 * old value was still accessible until then.
6671 */
6672 if (PyDict_DelItem(posix_putenv_garbage,
6673 PyTuple_GET_ITEM(args, 0))) {
6674 /* really not much we can do; just leak */
6675 PyErr_Clear();
6676 }
6677
6678 Py_INCREF(Py_None);
6679 return Py_None;
6680}
6681#endif /* unsetenv */
6682
Guido van Rossumb6a47161997-09-15 22:54:34 +00006683#ifdef HAVE_STRERROR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006684PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006685"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006686Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00006687
Guido van Rossumf68d8e52001-04-14 17:55:09 +00006688static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006689posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00006690{
6691 int code;
6692 char *message;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006693 if (!PyArg_ParseTuple(args, "i:strerror", &code))
Guido van Rossumb6a47161997-09-15 22:54:34 +00006694 return NULL;
6695 message = strerror(code);
6696 if (message == NULL) {
6697 PyErr_SetString(PyExc_ValueError,
Fred Drake661ea262000-10-24 19:57:45 +00006698 "strerror() argument out of range");
Guido van Rossumb6a47161997-09-15 22:54:34 +00006699 return NULL;
6700 }
6701 return PyString_FromString(message);
6702}
6703#endif /* strerror */
6704
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006705
Guido van Rossumc9641791998-08-04 15:26:23 +00006706#ifdef HAVE_SYS_WAIT_H
6707
Fred Drake106c1a02002-04-23 15:58:02 +00006708#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006709PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006710"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006711Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00006712
6713static PyObject *
6714posix_WCOREDUMP(PyObject *self, PyObject *args)
6715{
Neal Norwitzd5a37542006-03-20 06:48:34 +00006716 WAIT_TYPE status;
6717 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00006718
Neal Norwitzd5a37542006-03-20 06:48:34 +00006719 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00006720 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00006721
6722 return PyBool_FromLong(WCOREDUMP(status));
Fred Drake106c1a02002-04-23 15:58:02 +00006723}
6724#endif /* WCOREDUMP */
6725
6726#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006727PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006728"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00006729Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006730job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00006731
6732static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00006733posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00006734{
Neal Norwitzd5a37542006-03-20 06:48:34 +00006735 WAIT_TYPE status;
6736 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00006737
Neal Norwitzd5a37542006-03-20 06:48:34 +00006738 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00006739 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00006740
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00006741 return PyBool_FromLong(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00006742}
6743#endif /* WIFCONTINUED */
6744
Guido van Rossumc9641791998-08-04 15:26:23 +00006745#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006746PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006747"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006748Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00006749
6750static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006751posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00006752{
Neal Norwitzd5a37542006-03-20 06:48:34 +00006753 WAIT_TYPE status;
6754 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00006755
Neal Norwitzd5a37542006-03-20 06:48:34 +00006756 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00006757 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00006758
Fred Drake106c1a02002-04-23 15:58:02 +00006759 return PyBool_FromLong(WIFSTOPPED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00006760}
6761#endif /* WIFSTOPPED */
6762
6763#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006764PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006765"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006766Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00006767
6768static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006769posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00006770{
Neal Norwitzd5a37542006-03-20 06:48:34 +00006771 WAIT_TYPE status;
6772 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00006773
Neal Norwitzd5a37542006-03-20 06:48:34 +00006774 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00006775 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00006776
Fred Drake106c1a02002-04-23 15:58:02 +00006777 return PyBool_FromLong(WIFSIGNALED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00006778}
6779#endif /* WIFSIGNALED */
6780
6781#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006782PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006783"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00006784Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006785system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00006786
6787static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006788posix_WIFEXITED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00006789{
Neal Norwitzd5a37542006-03-20 06:48:34 +00006790 WAIT_TYPE status;
6791 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00006792
Neal Norwitzd5a37542006-03-20 06:48:34 +00006793 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00006794 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00006795
Fred Drake106c1a02002-04-23 15:58:02 +00006796 return PyBool_FromLong(WIFEXITED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00006797}
6798#endif /* WIFEXITED */
6799
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00006800#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006801PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006802"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006803Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00006804
6805static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006806posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00006807{
Neal Norwitzd5a37542006-03-20 06:48:34 +00006808 WAIT_TYPE status;
6809 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00006810
Neal Norwitzd5a37542006-03-20 06:48:34 +00006811 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00006812 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00006813
Guido van Rossumc9641791998-08-04 15:26:23 +00006814 return Py_BuildValue("i", WEXITSTATUS(status));
6815}
6816#endif /* WEXITSTATUS */
6817
6818#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006819PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006820"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00006821Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006822value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00006823
6824static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006825posix_WTERMSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00006826{
Neal Norwitzd5a37542006-03-20 06:48:34 +00006827 WAIT_TYPE status;
6828 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00006829
Neal Norwitzd5a37542006-03-20 06:48:34 +00006830 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00006831 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00006832
Guido van Rossumc9641791998-08-04 15:26:23 +00006833 return Py_BuildValue("i", WTERMSIG(status));
6834}
6835#endif /* WTERMSIG */
6836
6837#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006838PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006839"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006840Return the signal that stopped the process that provided\n\
6841the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00006842
6843static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006844posix_WSTOPSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00006845{
Neal Norwitzd5a37542006-03-20 06:48:34 +00006846 WAIT_TYPE status;
6847 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00006848
Neal Norwitzd5a37542006-03-20 06:48:34 +00006849 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00006850 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00006851
Guido van Rossumc9641791998-08-04 15:26:23 +00006852 return Py_BuildValue("i", WSTOPSIG(status));
6853}
6854#endif /* WSTOPSIG */
6855
6856#endif /* HAVE_SYS_WAIT_H */
6857
6858
Martin v. Löwis5f5d99c2006-05-16 07:05:37 +00006859#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossumd5753e11999-10-19 13:29:23 +00006860#ifdef _SCO_DS
6861/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
6862 needed definitions in sys/statvfs.h */
6863#define _SVID3
6864#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00006865#include <sys/statvfs.h>
6866
Guido van Rossum98bf58f2001-10-18 20:34:25 +00006867static PyObject*
6868_pystatvfs_fromstructstatvfs(struct statvfs st) {
6869 PyObject *v = PyStructSequence_New(&StatVFSResultType);
6870 if (v == NULL)
6871 return NULL;
6872
6873#if !defined(HAVE_LARGEFILE_SUPPORT)
6874 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long) st.f_bsize));
6875 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long) st.f_frsize));
6876 PyStructSequence_SET_ITEM(v, 2, PyInt_FromLong((long) st.f_blocks));
6877 PyStructSequence_SET_ITEM(v, 3, PyInt_FromLong((long) st.f_bfree));
6878 PyStructSequence_SET_ITEM(v, 4, PyInt_FromLong((long) st.f_bavail));
6879 PyStructSequence_SET_ITEM(v, 5, PyInt_FromLong((long) st.f_files));
6880 PyStructSequence_SET_ITEM(v, 6, PyInt_FromLong((long) st.f_ffree));
6881 PyStructSequence_SET_ITEM(v, 7, PyInt_FromLong((long) st.f_favail));
6882 PyStructSequence_SET_ITEM(v, 8, PyInt_FromLong((long) st.f_flag));
6883 PyStructSequence_SET_ITEM(v, 9, PyInt_FromLong((long) st.f_namemax));
6884#else
6885 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long) st.f_bsize));
6886 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long) st.f_frsize));
Tim Peters5aa91602002-01-30 05:46:57 +00006887 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00006888 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
Tim Peters5aa91602002-01-30 05:46:57 +00006889 PyStructSequence_SET_ITEM(v, 3,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00006890 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00006891 PyStructSequence_SET_ITEM(v, 4,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00006892 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
Tim Peters5aa91602002-01-30 05:46:57 +00006893 PyStructSequence_SET_ITEM(v, 5,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00006894 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
Tim Peters5aa91602002-01-30 05:46:57 +00006895 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00006896 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
Tim Peters5aa91602002-01-30 05:46:57 +00006897 PyStructSequence_SET_ITEM(v, 7,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00006898 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00006899 PyStructSequence_SET_ITEM(v, 8, PyInt_FromLong((long) st.f_flag));
6900 PyStructSequence_SET_ITEM(v, 9, PyInt_FromLong((long) st.f_namemax));
6901#endif
6902
6903 return v;
6904}
6905
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006906PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006907"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006908Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00006909
6910static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006911posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00006912{
6913 int fd, res;
6914 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00006915
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006916 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
Guido van Rossum94f6f721999-01-06 18:42:14 +00006917 return NULL;
6918 Py_BEGIN_ALLOW_THREADS
6919 res = fstatvfs(fd, &st);
6920 Py_END_ALLOW_THREADS
6921 if (res != 0)
6922 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00006923
6924 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006925}
Martin v. Löwis5f5d99c2006-05-16 07:05:37 +00006926#endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00006927
6928
Martin v. Löwis5f5d99c2006-05-16 07:05:37 +00006929#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossum94f6f721999-01-06 18:42:14 +00006930#include <sys/statvfs.h>
6931
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006932PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006933"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006934Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00006935
6936static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006937posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00006938{
6939 char *path;
6940 int res;
6941 struct statvfs st;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006942 if (!PyArg_ParseTuple(args, "s:statvfs", &path))
Guido van Rossum94f6f721999-01-06 18:42:14 +00006943 return NULL;
6944 Py_BEGIN_ALLOW_THREADS
6945 res = statvfs(path, &st);
6946 Py_END_ALLOW_THREADS
6947 if (res != 0)
6948 return posix_error_with_filename(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00006949
6950 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006951}
6952#endif /* HAVE_STATVFS */
6953
6954
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006955#ifdef HAVE_TEMPNAM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006956PyDoc_STRVAR(posix_tempnam__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006957"tempnam([dir[, prefix]]) -> string\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006958Return a unique name for a temporary file.\n\
Neal Norwitz50d5d4f2002-07-30 01:17:43 +00006959The directory and a prefix may be specified as strings; they may be omitted\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006960or None if not needed.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006961
6962static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006963posix_tempnam(PyObject *self, PyObject *args)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006964{
6965 PyObject *result = NULL;
6966 char *dir = NULL;
6967 char *pfx = NULL;
6968 char *name;
6969
6970 if (!PyArg_ParseTuple(args, "|zz:tempnam", &dir, &pfx))
6971 return NULL;
Skip Montanaro95618b52001-08-18 18:52:10 +00006972
6973 if (PyErr_Warn(PyExc_RuntimeWarning,
6974 "tempnam is a potential security risk to your program") < 0)
6975 return NULL;
6976
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006977#ifdef MS_WINDOWS
Fred Drake78b71c22001-07-17 20:37:36 +00006978 name = _tempnam(dir, pfx);
6979#else
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006980 name = tempnam(dir, pfx);
Fred Drake78b71c22001-07-17 20:37:36 +00006981#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006982 if (name == NULL)
6983 return PyErr_NoMemory();
6984 result = PyString_FromString(name);
6985 free(name);
6986 return result;
6987}
Guido van Rossumd371ff11999-01-25 16:12:23 +00006988#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006989
6990
6991#ifdef HAVE_TMPFILE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006992PyDoc_STRVAR(posix_tmpfile__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006993"tmpfile() -> file object\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006994Create a temporary file with no directory entries.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006995
6996static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006997posix_tmpfile(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006998{
6999 FILE *fp;
7000
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007001 fp = tmpfile();
7002 if (fp == NULL)
7003 return posix_error();
Guido van Rossumdb9198a2002-06-10 19:23:22 +00007004 return PyFile_FromFile(fp, "<tmpfile>", "w+b", fclose);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007005}
7006#endif
7007
7008
7009#ifdef HAVE_TMPNAM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007010PyDoc_STRVAR(posix_tmpnam__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007011"tmpnam() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007012Return a unique name for a temporary file.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007013
7014static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00007015posix_tmpnam(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007016{
7017 char buffer[L_tmpnam];
7018 char *name;
7019
Skip Montanaro95618b52001-08-18 18:52:10 +00007020 if (PyErr_Warn(PyExc_RuntimeWarning,
7021 "tmpnam is a potential security risk to your program") < 0)
7022 return NULL;
7023
Greg Wardb48bc172000-03-01 21:51:56 +00007024#ifdef USE_TMPNAM_R
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007025 name = tmpnam_r(buffer);
7026#else
7027 name = tmpnam(buffer);
7028#endif
7029 if (name == NULL) {
Neal Norwitzd1e0ef62006-03-20 04:08:12 +00007030 PyObject *err = Py_BuildValue("is", 0,
Greg Wardb48bc172000-03-01 21:51:56 +00007031#ifdef USE_TMPNAM_R
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007032 "unexpected NULL from tmpnam_r"
7033#else
7034 "unexpected NULL from tmpnam"
7035#endif
Neal Norwitzd1e0ef62006-03-20 04:08:12 +00007036 );
7037 PyErr_SetObject(PyExc_OSError, err);
7038 Py_XDECREF(err);
7039 return NULL;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007040 }
7041 return PyString_FromString(buffer);
7042}
7043#endif
7044
7045
Fred Drakec9680921999-12-13 16:37:25 +00007046/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
7047 * It maps strings representing configuration variable names to
7048 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00007049 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00007050 * rarely-used constants. There are three separate tables that use
7051 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00007052 *
7053 * This code is always included, even if none of the interfaces that
7054 * need it are included. The #if hackery needed to avoid it would be
7055 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00007056 */
7057struct constdef {
7058 char *name;
7059 long value;
7060};
7061
Fred Drake12c6e2d1999-12-14 21:25:03 +00007062static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007063conv_confname(PyObject *arg, int *valuep, struct constdef *table,
7064 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00007065{
7066 if (PyInt_Check(arg)) {
7067 *valuep = PyInt_AS_LONG(arg);
7068 return 1;
7069 }
7070 if (PyString_Check(arg)) {
7071 /* look up the value in the table using a binary search */
Fred Drake699f3522000-06-29 21:12:41 +00007072 size_t lo = 0;
7073 size_t mid;
7074 size_t hi = tablesize;
7075 int cmp;
Fred Drake12c6e2d1999-12-14 21:25:03 +00007076 char *confname = PyString_AS_STRING(arg);
7077 while (lo < hi) {
7078 mid = (lo + hi) / 2;
7079 cmp = strcmp(confname, table[mid].name);
7080 if (cmp < 0)
7081 hi = mid;
7082 else if (cmp > 0)
7083 lo = mid + 1;
7084 else {
7085 *valuep = table[mid].value;
7086 return 1;
7087 }
7088 }
7089 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
7090 }
7091 else
7092 PyErr_SetString(PyExc_TypeError,
7093 "configuration names must be strings or integers");
7094 return 0;
7095}
7096
7097
7098#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
7099static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00007100#ifdef _PC_ABI_AIO_XFER_MAX
7101 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
7102#endif
7103#ifdef _PC_ABI_ASYNC_IO
7104 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
7105#endif
Fred Drakec9680921999-12-13 16:37:25 +00007106#ifdef _PC_ASYNC_IO
7107 {"PC_ASYNC_IO", _PC_ASYNC_IO},
7108#endif
7109#ifdef _PC_CHOWN_RESTRICTED
7110 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
7111#endif
7112#ifdef _PC_FILESIZEBITS
7113 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
7114#endif
7115#ifdef _PC_LAST
7116 {"PC_LAST", _PC_LAST},
7117#endif
7118#ifdef _PC_LINK_MAX
7119 {"PC_LINK_MAX", _PC_LINK_MAX},
7120#endif
7121#ifdef _PC_MAX_CANON
7122 {"PC_MAX_CANON", _PC_MAX_CANON},
7123#endif
7124#ifdef _PC_MAX_INPUT
7125 {"PC_MAX_INPUT", _PC_MAX_INPUT},
7126#endif
7127#ifdef _PC_NAME_MAX
7128 {"PC_NAME_MAX", _PC_NAME_MAX},
7129#endif
7130#ifdef _PC_NO_TRUNC
7131 {"PC_NO_TRUNC", _PC_NO_TRUNC},
7132#endif
7133#ifdef _PC_PATH_MAX
7134 {"PC_PATH_MAX", _PC_PATH_MAX},
7135#endif
7136#ifdef _PC_PIPE_BUF
7137 {"PC_PIPE_BUF", _PC_PIPE_BUF},
7138#endif
7139#ifdef _PC_PRIO_IO
7140 {"PC_PRIO_IO", _PC_PRIO_IO},
7141#endif
7142#ifdef _PC_SOCK_MAXBUF
7143 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
7144#endif
7145#ifdef _PC_SYNC_IO
7146 {"PC_SYNC_IO", _PC_SYNC_IO},
7147#endif
7148#ifdef _PC_VDISABLE
7149 {"PC_VDISABLE", _PC_VDISABLE},
7150#endif
7151};
7152
Fred Drakec9680921999-12-13 16:37:25 +00007153static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007154conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00007155{
7156 return conv_confname(arg, valuep, posix_constants_pathconf,
7157 sizeof(posix_constants_pathconf)
7158 / sizeof(struct constdef));
7159}
7160#endif
7161
7162#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007163PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007164"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00007165Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007166If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00007167
7168static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007169posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00007170{
7171 PyObject *result = NULL;
7172 int name, fd;
7173
Fred Drake12c6e2d1999-12-14 21:25:03 +00007174 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
7175 conv_path_confname, &name)) {
Fred Drakec9680921999-12-13 16:37:25 +00007176 long limit;
7177
7178 errno = 0;
7179 limit = fpathconf(fd, name);
7180 if (limit == -1 && errno != 0)
7181 posix_error();
7182 else
7183 result = PyInt_FromLong(limit);
7184 }
7185 return result;
7186}
7187#endif
7188
7189
7190#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007191PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007192"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00007193Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007194If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00007195
7196static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007197posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00007198{
7199 PyObject *result = NULL;
7200 int name;
7201 char *path;
7202
7203 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
7204 conv_path_confname, &name)) {
7205 long limit;
7206
7207 errno = 0;
7208 limit = pathconf(path, name);
Fred Drake12c6e2d1999-12-14 21:25:03 +00007209 if (limit == -1 && errno != 0) {
Fred Drakec9680921999-12-13 16:37:25 +00007210 if (errno == EINVAL)
7211 /* could be a path or name problem */
7212 posix_error();
7213 else
7214 posix_error_with_filename(path);
Fred Drake12c6e2d1999-12-14 21:25:03 +00007215 }
Fred Drakec9680921999-12-13 16:37:25 +00007216 else
7217 result = PyInt_FromLong(limit);
7218 }
7219 return result;
7220}
7221#endif
7222
7223#ifdef HAVE_CONFSTR
7224static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00007225#ifdef _CS_ARCHITECTURE
7226 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
7227#endif
7228#ifdef _CS_HOSTNAME
7229 {"CS_HOSTNAME", _CS_HOSTNAME},
7230#endif
7231#ifdef _CS_HW_PROVIDER
7232 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
7233#endif
7234#ifdef _CS_HW_SERIAL
7235 {"CS_HW_SERIAL", _CS_HW_SERIAL},
7236#endif
7237#ifdef _CS_INITTAB_NAME
7238 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
7239#endif
Fred Drakec9680921999-12-13 16:37:25 +00007240#ifdef _CS_LFS64_CFLAGS
7241 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
7242#endif
7243#ifdef _CS_LFS64_LDFLAGS
7244 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
7245#endif
7246#ifdef _CS_LFS64_LIBS
7247 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
7248#endif
7249#ifdef _CS_LFS64_LINTFLAGS
7250 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
7251#endif
7252#ifdef _CS_LFS_CFLAGS
7253 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
7254#endif
7255#ifdef _CS_LFS_LDFLAGS
7256 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
7257#endif
7258#ifdef _CS_LFS_LIBS
7259 {"CS_LFS_LIBS", _CS_LFS_LIBS},
7260#endif
7261#ifdef _CS_LFS_LINTFLAGS
7262 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
7263#endif
Fred Draked86ed291999-12-15 15:34:33 +00007264#ifdef _CS_MACHINE
7265 {"CS_MACHINE", _CS_MACHINE},
7266#endif
Fred Drakec9680921999-12-13 16:37:25 +00007267#ifdef _CS_PATH
7268 {"CS_PATH", _CS_PATH},
7269#endif
Fred Draked86ed291999-12-15 15:34:33 +00007270#ifdef _CS_RELEASE
7271 {"CS_RELEASE", _CS_RELEASE},
7272#endif
7273#ifdef _CS_SRPC_DOMAIN
7274 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
7275#endif
7276#ifdef _CS_SYSNAME
7277 {"CS_SYSNAME", _CS_SYSNAME},
7278#endif
7279#ifdef _CS_VERSION
7280 {"CS_VERSION", _CS_VERSION},
7281#endif
Fred Drakec9680921999-12-13 16:37:25 +00007282#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
7283 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
7284#endif
7285#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
7286 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
7287#endif
7288#ifdef _CS_XBS5_ILP32_OFF32_LIBS
7289 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
7290#endif
7291#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
7292 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
7293#endif
7294#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
7295 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
7296#endif
7297#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
7298 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
7299#endif
7300#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
7301 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
7302#endif
7303#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
7304 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
7305#endif
7306#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
7307 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
7308#endif
7309#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
7310 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
7311#endif
7312#ifdef _CS_XBS5_LP64_OFF64_LIBS
7313 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
7314#endif
7315#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
7316 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
7317#endif
7318#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
7319 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
7320#endif
7321#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
7322 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
7323#endif
7324#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
7325 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
7326#endif
7327#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
7328 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
7329#endif
Fred Draked86ed291999-12-15 15:34:33 +00007330#ifdef _MIPS_CS_AVAIL_PROCESSORS
7331 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
7332#endif
7333#ifdef _MIPS_CS_BASE
7334 {"MIPS_CS_BASE", _MIPS_CS_BASE},
7335#endif
7336#ifdef _MIPS_CS_HOSTID
7337 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
7338#endif
7339#ifdef _MIPS_CS_HW_NAME
7340 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
7341#endif
7342#ifdef _MIPS_CS_NUM_PROCESSORS
7343 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
7344#endif
7345#ifdef _MIPS_CS_OSREL_MAJ
7346 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
7347#endif
7348#ifdef _MIPS_CS_OSREL_MIN
7349 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
7350#endif
7351#ifdef _MIPS_CS_OSREL_PATCH
7352 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
7353#endif
7354#ifdef _MIPS_CS_OS_NAME
7355 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
7356#endif
7357#ifdef _MIPS_CS_OS_PROVIDER
7358 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
7359#endif
7360#ifdef _MIPS_CS_PROCESSORS
7361 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
7362#endif
7363#ifdef _MIPS_CS_SERIAL
7364 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
7365#endif
7366#ifdef _MIPS_CS_VENDOR
7367 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
7368#endif
Fred Drakec9680921999-12-13 16:37:25 +00007369};
7370
7371static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007372conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00007373{
7374 return conv_confname(arg, valuep, posix_constants_confstr,
7375 sizeof(posix_constants_confstr)
7376 / sizeof(struct constdef));
7377}
7378
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007379PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007380"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007381Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00007382
7383static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007384posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00007385{
7386 PyObject *result = NULL;
7387 int name;
Neal Norwitz449b24e2006-04-20 06:56:05 +00007388 char buffer[256];
Fred Drakec9680921999-12-13 16:37:25 +00007389
7390 if (PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name)) {
Skip Montanarodd527fc2006-04-18 00:49:49 +00007391 int len;
Fred Drakec9680921999-12-13 16:37:25 +00007392
Fred Drakec9680921999-12-13 16:37:25 +00007393 errno = 0;
Skip Montanarodd527fc2006-04-18 00:49:49 +00007394 len = confstr(name, buffer, sizeof(buffer));
Skip Montanaro94785ef2006-04-20 01:29:48 +00007395 if (len == 0) {
7396 if (errno) {
7397 posix_error();
7398 }
7399 else {
7400 result = Py_None;
7401 Py_INCREF(Py_None);
7402 }
Fred Drakec9680921999-12-13 16:37:25 +00007403 }
7404 else {
Neal Norwitz0d21b1e2006-04-20 06:44:42 +00007405 if ((unsigned int)len >= sizeof(buffer)) {
Neal Norwitz449b24e2006-04-20 06:56:05 +00007406 result = PyString_FromStringAndSize(NULL, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00007407 if (result != NULL)
Neal Norwitz449b24e2006-04-20 06:56:05 +00007408 confstr(name, PyString_AS_STRING(result), len);
Fred Drakec9680921999-12-13 16:37:25 +00007409 }
7410 else
Neal Norwitz449b24e2006-04-20 06:56:05 +00007411 result = PyString_FromStringAndSize(buffer, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00007412 }
7413 }
7414 return result;
7415}
7416#endif
7417
7418
7419#ifdef HAVE_SYSCONF
7420static struct constdef posix_constants_sysconf[] = {
7421#ifdef _SC_2_CHAR_TERM
7422 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
7423#endif
7424#ifdef _SC_2_C_BIND
7425 {"SC_2_C_BIND", _SC_2_C_BIND},
7426#endif
7427#ifdef _SC_2_C_DEV
7428 {"SC_2_C_DEV", _SC_2_C_DEV},
7429#endif
7430#ifdef _SC_2_C_VERSION
7431 {"SC_2_C_VERSION", _SC_2_C_VERSION},
7432#endif
7433#ifdef _SC_2_FORT_DEV
7434 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
7435#endif
7436#ifdef _SC_2_FORT_RUN
7437 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
7438#endif
7439#ifdef _SC_2_LOCALEDEF
7440 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
7441#endif
7442#ifdef _SC_2_SW_DEV
7443 {"SC_2_SW_DEV", _SC_2_SW_DEV},
7444#endif
7445#ifdef _SC_2_UPE
7446 {"SC_2_UPE", _SC_2_UPE},
7447#endif
7448#ifdef _SC_2_VERSION
7449 {"SC_2_VERSION", _SC_2_VERSION},
7450#endif
Fred Draked86ed291999-12-15 15:34:33 +00007451#ifdef _SC_ABI_ASYNCHRONOUS_IO
7452 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
7453#endif
7454#ifdef _SC_ACL
7455 {"SC_ACL", _SC_ACL},
7456#endif
Fred Drakec9680921999-12-13 16:37:25 +00007457#ifdef _SC_AIO_LISTIO_MAX
7458 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
7459#endif
Fred Drakec9680921999-12-13 16:37:25 +00007460#ifdef _SC_AIO_MAX
7461 {"SC_AIO_MAX", _SC_AIO_MAX},
7462#endif
7463#ifdef _SC_AIO_PRIO_DELTA_MAX
7464 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
7465#endif
7466#ifdef _SC_ARG_MAX
7467 {"SC_ARG_MAX", _SC_ARG_MAX},
7468#endif
7469#ifdef _SC_ASYNCHRONOUS_IO
7470 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
7471#endif
7472#ifdef _SC_ATEXIT_MAX
7473 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
7474#endif
Fred Draked86ed291999-12-15 15:34:33 +00007475#ifdef _SC_AUDIT
7476 {"SC_AUDIT", _SC_AUDIT},
7477#endif
Fred Drakec9680921999-12-13 16:37:25 +00007478#ifdef _SC_AVPHYS_PAGES
7479 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
7480#endif
7481#ifdef _SC_BC_BASE_MAX
7482 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
7483#endif
7484#ifdef _SC_BC_DIM_MAX
7485 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
7486#endif
7487#ifdef _SC_BC_SCALE_MAX
7488 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
7489#endif
7490#ifdef _SC_BC_STRING_MAX
7491 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
7492#endif
Fred Draked86ed291999-12-15 15:34:33 +00007493#ifdef _SC_CAP
7494 {"SC_CAP", _SC_CAP},
7495#endif
Fred Drakec9680921999-12-13 16:37:25 +00007496#ifdef _SC_CHARCLASS_NAME_MAX
7497 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
7498#endif
7499#ifdef _SC_CHAR_BIT
7500 {"SC_CHAR_BIT", _SC_CHAR_BIT},
7501#endif
7502#ifdef _SC_CHAR_MAX
7503 {"SC_CHAR_MAX", _SC_CHAR_MAX},
7504#endif
7505#ifdef _SC_CHAR_MIN
7506 {"SC_CHAR_MIN", _SC_CHAR_MIN},
7507#endif
7508#ifdef _SC_CHILD_MAX
7509 {"SC_CHILD_MAX", _SC_CHILD_MAX},
7510#endif
7511#ifdef _SC_CLK_TCK
7512 {"SC_CLK_TCK", _SC_CLK_TCK},
7513#endif
7514#ifdef _SC_COHER_BLKSZ
7515 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
7516#endif
7517#ifdef _SC_COLL_WEIGHTS_MAX
7518 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
7519#endif
7520#ifdef _SC_DCACHE_ASSOC
7521 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
7522#endif
7523#ifdef _SC_DCACHE_BLKSZ
7524 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
7525#endif
7526#ifdef _SC_DCACHE_LINESZ
7527 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
7528#endif
7529#ifdef _SC_DCACHE_SZ
7530 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
7531#endif
7532#ifdef _SC_DCACHE_TBLKSZ
7533 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
7534#endif
7535#ifdef _SC_DELAYTIMER_MAX
7536 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
7537#endif
7538#ifdef _SC_EQUIV_CLASS_MAX
7539 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
7540#endif
7541#ifdef _SC_EXPR_NEST_MAX
7542 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
7543#endif
7544#ifdef _SC_FSYNC
7545 {"SC_FSYNC", _SC_FSYNC},
7546#endif
7547#ifdef _SC_GETGR_R_SIZE_MAX
7548 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
7549#endif
7550#ifdef _SC_GETPW_R_SIZE_MAX
7551 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
7552#endif
7553#ifdef _SC_ICACHE_ASSOC
7554 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
7555#endif
7556#ifdef _SC_ICACHE_BLKSZ
7557 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
7558#endif
7559#ifdef _SC_ICACHE_LINESZ
7560 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
7561#endif
7562#ifdef _SC_ICACHE_SZ
7563 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
7564#endif
Fred Draked86ed291999-12-15 15:34:33 +00007565#ifdef _SC_INF
7566 {"SC_INF", _SC_INF},
7567#endif
Fred Drakec9680921999-12-13 16:37:25 +00007568#ifdef _SC_INT_MAX
7569 {"SC_INT_MAX", _SC_INT_MAX},
7570#endif
7571#ifdef _SC_INT_MIN
7572 {"SC_INT_MIN", _SC_INT_MIN},
7573#endif
7574#ifdef _SC_IOV_MAX
7575 {"SC_IOV_MAX", _SC_IOV_MAX},
7576#endif
Fred Draked86ed291999-12-15 15:34:33 +00007577#ifdef _SC_IP_SECOPTS
7578 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
7579#endif
Fred Drakec9680921999-12-13 16:37:25 +00007580#ifdef _SC_JOB_CONTROL
7581 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
7582#endif
Fred Draked86ed291999-12-15 15:34:33 +00007583#ifdef _SC_KERN_POINTERS
7584 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
7585#endif
7586#ifdef _SC_KERN_SIM
7587 {"SC_KERN_SIM", _SC_KERN_SIM},
7588#endif
Fred Drakec9680921999-12-13 16:37:25 +00007589#ifdef _SC_LINE_MAX
7590 {"SC_LINE_MAX", _SC_LINE_MAX},
7591#endif
7592#ifdef _SC_LOGIN_NAME_MAX
7593 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
7594#endif
7595#ifdef _SC_LOGNAME_MAX
7596 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
7597#endif
7598#ifdef _SC_LONG_BIT
7599 {"SC_LONG_BIT", _SC_LONG_BIT},
7600#endif
Fred Draked86ed291999-12-15 15:34:33 +00007601#ifdef _SC_MAC
7602 {"SC_MAC", _SC_MAC},
7603#endif
Fred Drakec9680921999-12-13 16:37:25 +00007604#ifdef _SC_MAPPED_FILES
7605 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
7606#endif
7607#ifdef _SC_MAXPID
7608 {"SC_MAXPID", _SC_MAXPID},
7609#endif
7610#ifdef _SC_MB_LEN_MAX
7611 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
7612#endif
7613#ifdef _SC_MEMLOCK
7614 {"SC_MEMLOCK", _SC_MEMLOCK},
7615#endif
7616#ifdef _SC_MEMLOCK_RANGE
7617 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
7618#endif
7619#ifdef _SC_MEMORY_PROTECTION
7620 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
7621#endif
7622#ifdef _SC_MESSAGE_PASSING
7623 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
7624#endif
Fred Draked86ed291999-12-15 15:34:33 +00007625#ifdef _SC_MMAP_FIXED_ALIGNMENT
7626 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
7627#endif
Fred Drakec9680921999-12-13 16:37:25 +00007628#ifdef _SC_MQ_OPEN_MAX
7629 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
7630#endif
7631#ifdef _SC_MQ_PRIO_MAX
7632 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
7633#endif
Fred Draked86ed291999-12-15 15:34:33 +00007634#ifdef _SC_NACLS_MAX
7635 {"SC_NACLS_MAX", _SC_NACLS_MAX},
7636#endif
Fred Drakec9680921999-12-13 16:37:25 +00007637#ifdef _SC_NGROUPS_MAX
7638 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
7639#endif
7640#ifdef _SC_NL_ARGMAX
7641 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
7642#endif
7643#ifdef _SC_NL_LANGMAX
7644 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
7645#endif
7646#ifdef _SC_NL_MSGMAX
7647 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
7648#endif
7649#ifdef _SC_NL_NMAX
7650 {"SC_NL_NMAX", _SC_NL_NMAX},
7651#endif
7652#ifdef _SC_NL_SETMAX
7653 {"SC_NL_SETMAX", _SC_NL_SETMAX},
7654#endif
7655#ifdef _SC_NL_TEXTMAX
7656 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
7657#endif
7658#ifdef _SC_NPROCESSORS_CONF
7659 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
7660#endif
7661#ifdef _SC_NPROCESSORS_ONLN
7662 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
7663#endif
Fred Draked86ed291999-12-15 15:34:33 +00007664#ifdef _SC_NPROC_CONF
7665 {"SC_NPROC_CONF", _SC_NPROC_CONF},
7666#endif
7667#ifdef _SC_NPROC_ONLN
7668 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
7669#endif
Fred Drakec9680921999-12-13 16:37:25 +00007670#ifdef _SC_NZERO
7671 {"SC_NZERO", _SC_NZERO},
7672#endif
7673#ifdef _SC_OPEN_MAX
7674 {"SC_OPEN_MAX", _SC_OPEN_MAX},
7675#endif
7676#ifdef _SC_PAGESIZE
7677 {"SC_PAGESIZE", _SC_PAGESIZE},
7678#endif
7679#ifdef _SC_PAGE_SIZE
7680 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
7681#endif
7682#ifdef _SC_PASS_MAX
7683 {"SC_PASS_MAX", _SC_PASS_MAX},
7684#endif
7685#ifdef _SC_PHYS_PAGES
7686 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
7687#endif
7688#ifdef _SC_PII
7689 {"SC_PII", _SC_PII},
7690#endif
7691#ifdef _SC_PII_INTERNET
7692 {"SC_PII_INTERNET", _SC_PII_INTERNET},
7693#endif
7694#ifdef _SC_PII_INTERNET_DGRAM
7695 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
7696#endif
7697#ifdef _SC_PII_INTERNET_STREAM
7698 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
7699#endif
7700#ifdef _SC_PII_OSI
7701 {"SC_PII_OSI", _SC_PII_OSI},
7702#endif
7703#ifdef _SC_PII_OSI_CLTS
7704 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
7705#endif
7706#ifdef _SC_PII_OSI_COTS
7707 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
7708#endif
7709#ifdef _SC_PII_OSI_M
7710 {"SC_PII_OSI_M", _SC_PII_OSI_M},
7711#endif
7712#ifdef _SC_PII_SOCKET
7713 {"SC_PII_SOCKET", _SC_PII_SOCKET},
7714#endif
7715#ifdef _SC_PII_XTI
7716 {"SC_PII_XTI", _SC_PII_XTI},
7717#endif
7718#ifdef _SC_POLL
7719 {"SC_POLL", _SC_POLL},
7720#endif
7721#ifdef _SC_PRIORITIZED_IO
7722 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
7723#endif
7724#ifdef _SC_PRIORITY_SCHEDULING
7725 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
7726#endif
7727#ifdef _SC_REALTIME_SIGNALS
7728 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
7729#endif
7730#ifdef _SC_RE_DUP_MAX
7731 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
7732#endif
7733#ifdef _SC_RTSIG_MAX
7734 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
7735#endif
7736#ifdef _SC_SAVED_IDS
7737 {"SC_SAVED_IDS", _SC_SAVED_IDS},
7738#endif
7739#ifdef _SC_SCHAR_MAX
7740 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
7741#endif
7742#ifdef _SC_SCHAR_MIN
7743 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
7744#endif
7745#ifdef _SC_SELECT
7746 {"SC_SELECT", _SC_SELECT},
7747#endif
7748#ifdef _SC_SEMAPHORES
7749 {"SC_SEMAPHORES", _SC_SEMAPHORES},
7750#endif
7751#ifdef _SC_SEM_NSEMS_MAX
7752 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
7753#endif
7754#ifdef _SC_SEM_VALUE_MAX
7755 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
7756#endif
7757#ifdef _SC_SHARED_MEMORY_OBJECTS
7758 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
7759#endif
7760#ifdef _SC_SHRT_MAX
7761 {"SC_SHRT_MAX", _SC_SHRT_MAX},
7762#endif
7763#ifdef _SC_SHRT_MIN
7764 {"SC_SHRT_MIN", _SC_SHRT_MIN},
7765#endif
7766#ifdef _SC_SIGQUEUE_MAX
7767 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
7768#endif
7769#ifdef _SC_SIGRT_MAX
7770 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
7771#endif
7772#ifdef _SC_SIGRT_MIN
7773 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
7774#endif
Fred Draked86ed291999-12-15 15:34:33 +00007775#ifdef _SC_SOFTPOWER
7776 {"SC_SOFTPOWER", _SC_SOFTPOWER},
7777#endif
Fred Drakec9680921999-12-13 16:37:25 +00007778#ifdef _SC_SPLIT_CACHE
7779 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
7780#endif
7781#ifdef _SC_SSIZE_MAX
7782 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
7783#endif
7784#ifdef _SC_STACK_PROT
7785 {"SC_STACK_PROT", _SC_STACK_PROT},
7786#endif
7787#ifdef _SC_STREAM_MAX
7788 {"SC_STREAM_MAX", _SC_STREAM_MAX},
7789#endif
7790#ifdef _SC_SYNCHRONIZED_IO
7791 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
7792#endif
7793#ifdef _SC_THREADS
7794 {"SC_THREADS", _SC_THREADS},
7795#endif
7796#ifdef _SC_THREAD_ATTR_STACKADDR
7797 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
7798#endif
7799#ifdef _SC_THREAD_ATTR_STACKSIZE
7800 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
7801#endif
7802#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
7803 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
7804#endif
7805#ifdef _SC_THREAD_KEYS_MAX
7806 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
7807#endif
7808#ifdef _SC_THREAD_PRIORITY_SCHEDULING
7809 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
7810#endif
7811#ifdef _SC_THREAD_PRIO_INHERIT
7812 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
7813#endif
7814#ifdef _SC_THREAD_PRIO_PROTECT
7815 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
7816#endif
7817#ifdef _SC_THREAD_PROCESS_SHARED
7818 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
7819#endif
7820#ifdef _SC_THREAD_SAFE_FUNCTIONS
7821 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
7822#endif
7823#ifdef _SC_THREAD_STACK_MIN
7824 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
7825#endif
7826#ifdef _SC_THREAD_THREADS_MAX
7827 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
7828#endif
7829#ifdef _SC_TIMERS
7830 {"SC_TIMERS", _SC_TIMERS},
7831#endif
7832#ifdef _SC_TIMER_MAX
7833 {"SC_TIMER_MAX", _SC_TIMER_MAX},
7834#endif
7835#ifdef _SC_TTY_NAME_MAX
7836 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
7837#endif
7838#ifdef _SC_TZNAME_MAX
7839 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
7840#endif
7841#ifdef _SC_T_IOV_MAX
7842 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
7843#endif
7844#ifdef _SC_UCHAR_MAX
7845 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
7846#endif
7847#ifdef _SC_UINT_MAX
7848 {"SC_UINT_MAX", _SC_UINT_MAX},
7849#endif
7850#ifdef _SC_UIO_MAXIOV
7851 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
7852#endif
7853#ifdef _SC_ULONG_MAX
7854 {"SC_ULONG_MAX", _SC_ULONG_MAX},
7855#endif
7856#ifdef _SC_USHRT_MAX
7857 {"SC_USHRT_MAX", _SC_USHRT_MAX},
7858#endif
7859#ifdef _SC_VERSION
7860 {"SC_VERSION", _SC_VERSION},
7861#endif
7862#ifdef _SC_WORD_BIT
7863 {"SC_WORD_BIT", _SC_WORD_BIT},
7864#endif
7865#ifdef _SC_XBS5_ILP32_OFF32
7866 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
7867#endif
7868#ifdef _SC_XBS5_ILP32_OFFBIG
7869 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
7870#endif
7871#ifdef _SC_XBS5_LP64_OFF64
7872 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
7873#endif
7874#ifdef _SC_XBS5_LPBIG_OFFBIG
7875 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
7876#endif
7877#ifdef _SC_XOPEN_CRYPT
7878 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
7879#endif
7880#ifdef _SC_XOPEN_ENH_I18N
7881 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
7882#endif
7883#ifdef _SC_XOPEN_LEGACY
7884 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
7885#endif
7886#ifdef _SC_XOPEN_REALTIME
7887 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
7888#endif
7889#ifdef _SC_XOPEN_REALTIME_THREADS
7890 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
7891#endif
7892#ifdef _SC_XOPEN_SHM
7893 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
7894#endif
7895#ifdef _SC_XOPEN_UNIX
7896 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
7897#endif
7898#ifdef _SC_XOPEN_VERSION
7899 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
7900#endif
7901#ifdef _SC_XOPEN_XCU_VERSION
7902 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
7903#endif
7904#ifdef _SC_XOPEN_XPG2
7905 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
7906#endif
7907#ifdef _SC_XOPEN_XPG3
7908 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
7909#endif
7910#ifdef _SC_XOPEN_XPG4
7911 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
7912#endif
7913};
7914
7915static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007916conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00007917{
7918 return conv_confname(arg, valuep, posix_constants_sysconf,
7919 sizeof(posix_constants_sysconf)
7920 / sizeof(struct constdef));
7921}
7922
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007923PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007924"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007925Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00007926
7927static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007928posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00007929{
7930 PyObject *result = NULL;
7931 int name;
7932
7933 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
7934 int value;
7935
7936 errno = 0;
7937 value = sysconf(name);
7938 if (value == -1 && errno != 0)
7939 posix_error();
7940 else
7941 result = PyInt_FromLong(value);
7942 }
7943 return result;
7944}
7945#endif
7946
7947
Fred Drakebec628d1999-12-15 18:31:10 +00007948/* This code is used to ensure that the tables of configuration value names
7949 * are in sorted order as required by conv_confname(), and also to build the
7950 * the exported dictionaries that are used to publish information about the
7951 * names available on the host platform.
7952 *
7953 * Sorting the table at runtime ensures that the table is properly ordered
7954 * when used, even for platforms we're not able to test on. It also makes
7955 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00007956 */
Fred Drakebec628d1999-12-15 18:31:10 +00007957
7958static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007959cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00007960{
7961 const struct constdef *c1 =
7962 (const struct constdef *) v1;
7963 const struct constdef *c2 =
7964 (const struct constdef *) v2;
7965
7966 return strcmp(c1->name, c2->name);
7967}
7968
7969static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007970setup_confname_table(struct constdef *table, size_t tablesize,
Fred Drake4d1e64b2002-04-15 19:40:07 +00007971 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00007972{
Fred Drakebec628d1999-12-15 18:31:10 +00007973 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00007974 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00007975
7976 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
7977 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00007978 if (d == NULL)
7979 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00007980
Barry Warsaw3155db32000-04-13 15:20:40 +00007981 for (i=0; i < tablesize; ++i) {
7982 PyObject *o = PyInt_FromLong(table[i].value);
7983 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
7984 Py_XDECREF(o);
7985 Py_DECREF(d);
7986 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00007987 }
Barry Warsaw3155db32000-04-13 15:20:40 +00007988 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00007989 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00007990 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00007991}
7992
Fred Drakebec628d1999-12-15 18:31:10 +00007993/* Return -1 on failure, 0 on success. */
7994static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00007995setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00007996{
7997#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00007998 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00007999 sizeof(posix_constants_pathconf)
8000 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00008001 "pathconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00008002 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00008003#endif
8004#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00008005 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00008006 sizeof(posix_constants_confstr)
8007 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00008008 "confstr_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00008009 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00008010#endif
8011#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00008012 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00008013 sizeof(posix_constants_sysconf)
8014 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00008015 "sysconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00008016 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00008017#endif
Fred Drakebec628d1999-12-15 18:31:10 +00008018 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00008019}
Fred Draked86ed291999-12-15 15:34:33 +00008020
8021
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008022PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008023"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008024Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008025in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008026
8027static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00008028posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008029{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008030 abort();
8031 /*NOTREACHED*/
8032 Py_FatalError("abort() called from Python code didn't abort!");
8033 return NULL;
8034}
Fred Drakebec628d1999-12-15 18:31:10 +00008035
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00008036#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008037PyDoc_STRVAR(win32_startfile__doc__,
Georg Brandlf4f44152006-02-18 22:29:33 +00008038"startfile(filepath [, operation]) - Start a file with its associated\n\
8039application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00008040\n\
Georg Brandlf4f44152006-02-18 22:29:33 +00008041When \"operation\" is not specified or \"open\", this acts like\n\
8042double-clicking the file in Explorer, or giving the file name as an\n\
8043argument to the DOS \"start\" command: the file is opened with whatever\n\
8044application (if any) its extension is associated.\n\
8045When another \"operation\" is given, it specifies what should be done with\n\
8046the file. A typical operation is \"print\".\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00008047\n\
8048startfile returns as soon as the associated application is launched.\n\
8049There is no option to wait for the application to close, and no way\n\
8050to retrieve the application's exit status.\n\
8051\n\
8052The filepath is relative to the current directory. If you want to use\n\
8053an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008054the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00008055
8056static PyObject *
8057win32_startfile(PyObject *self, PyObject *args)
8058{
8059 char *filepath;
Georg Brandlf4f44152006-02-18 22:29:33 +00008060 char *operation = NULL;
Tim Petersf58a7aa2000-09-22 10:05:54 +00008061 HINSTANCE rc;
Georg Brandlad89dc82006-04-03 12:26:26 +00008062#ifdef Py_WIN_WIDE_FILENAMES
8063 if (unicode_file_names()) {
Martin v. Löwis5fe715f2006-04-03 23:01:24 +00008064 PyObject *unipath, *woperation = NULL;
Georg Brandlad89dc82006-04-03 12:26:26 +00008065 if (!PyArg_ParseTuple(args, "U|s:startfile",
8066 &unipath, &operation)) {
8067 PyErr_Clear();
8068 goto normal;
8069 }
8070
Martin v. Löwis5fe715f2006-04-03 23:01:24 +00008071
8072 if (operation) {
8073 woperation = PyUnicode_DecodeASCII(operation,
8074 strlen(operation), NULL);
8075 if (!woperation) {
8076 PyErr_Clear();
8077 operation = NULL;
8078 goto normal;
8079 }
Georg Brandlad89dc82006-04-03 12:26:26 +00008080 }
8081
8082 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis5fe715f2006-04-03 23:01:24 +00008083 rc = ShellExecuteW((HWND)0, woperation ? PyUnicode_AS_UNICODE(woperation) : 0,
Georg Brandlad89dc82006-04-03 12:26:26 +00008084 PyUnicode_AS_UNICODE(unipath),
Georg Brandlad89dc82006-04-03 12:26:26 +00008085 NULL, NULL, SW_SHOWNORMAL);
8086 Py_END_ALLOW_THREADS
8087
Martin v. Löwis5fe715f2006-04-03 23:01:24 +00008088 Py_XDECREF(woperation);
Georg Brandlad89dc82006-04-03 12:26:26 +00008089 if (rc <= (HINSTANCE)32) {
8090 PyObject *errval = win32_error_unicode("startfile",
8091 PyUnicode_AS_UNICODE(unipath));
8092 return errval;
8093 }
8094 Py_INCREF(Py_None);
8095 return Py_None;
8096 }
8097#endif
8098
8099normal:
Georg Brandlf4f44152006-02-18 22:29:33 +00008100 if (!PyArg_ParseTuple(args, "et|s:startfile",
8101 Py_FileSystemDefaultEncoding, &filepath,
8102 &operation))
Tim Petersf58a7aa2000-09-22 10:05:54 +00008103 return NULL;
8104 Py_BEGIN_ALLOW_THREADS
Georg Brandlf4f44152006-02-18 22:29:33 +00008105 rc = ShellExecute((HWND)0, operation, filepath,
8106 NULL, NULL, SW_SHOWNORMAL);
Tim Petersf58a7aa2000-09-22 10:05:54 +00008107 Py_END_ALLOW_THREADS
Georg Brandle9f8ec92005-09-25 06:16:40 +00008108 if (rc <= (HINSTANCE)32) {
8109 PyObject *errval = win32_error("startfile", filepath);
8110 PyMem_Free(filepath);
8111 return errval;
8112 }
8113 PyMem_Free(filepath);
Tim Petersf58a7aa2000-09-22 10:05:54 +00008114 Py_INCREF(Py_None);
8115 return Py_None;
8116}
8117#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008118
Martin v. Löwis438b5342002-12-27 10:16:42 +00008119#ifdef HAVE_GETLOADAVG
8120PyDoc_STRVAR(posix_getloadavg__doc__,
8121"getloadavg() -> (float, float, float)\n\n\
8122Return the number of processes in the system run queue averaged over\n\
8123the last 1, 5, and 15 minutes or raises OSError if the load average\n\
8124was unobtainable");
8125
8126static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00008127posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00008128{
8129 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00008130 if (getloadavg(loadavg, 3)!=3) {
8131 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
8132 return NULL;
8133 } else
8134 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
8135}
8136#endif
8137
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008138#ifdef MS_WINDOWS
8139
8140PyDoc_STRVAR(win32_urandom__doc__,
8141"urandom(n) -> str\n\n\
8142Return a string of n random bytes suitable for cryptographic use.");
8143
8144typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
8145 LPCSTR pszContainer, LPCSTR pszProvider, DWORD dwProvType,\
8146 DWORD dwFlags );
8147typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV hProv, DWORD dwLen,\
8148 BYTE *pbBuffer );
8149
8150static CRYPTGENRANDOM pCryptGenRandom = NULL;
Martin v. Löwisaf699dd2007-09-04 09:51:57 +00008151/* This handle is never explicitly released. Instead, the operating
8152 system will release it when the process terminates. */
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008153static HCRYPTPROV hCryptProv = 0;
8154
Tim Peters4ad82172004-08-30 17:02:04 +00008155static PyObject*
8156win32_urandom(PyObject *self, PyObject *args)
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008157{
Tim Petersd3115382004-08-30 17:36:46 +00008158 int howMany;
8159 PyObject* result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008160
Tim Peters4ad82172004-08-30 17:02:04 +00008161 /* Read arguments */
Tim Peters9b279a82004-08-30 17:10:53 +00008162 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
Tim Peters4ad82172004-08-30 17:02:04 +00008163 return NULL;
Tim Peters51eba612004-08-30 17:08:02 +00008164 if (howMany < 0)
8165 return PyErr_Format(PyExc_ValueError,
8166 "negative argument not allowed");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008167
Tim Peters4ad82172004-08-30 17:02:04 +00008168 if (hCryptProv == 0) {
8169 HINSTANCE hAdvAPI32 = NULL;
8170 CRYPTACQUIRECONTEXTA pCryptAcquireContext = NULL;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008171
Tim Peters4ad82172004-08-30 17:02:04 +00008172 /* Obtain handle to the DLL containing CryptoAPI
8173 This should not fail */
8174 hAdvAPI32 = GetModuleHandle("advapi32.dll");
8175 if(hAdvAPI32 == NULL)
8176 return win32_error("GetModuleHandle", NULL);
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008177
Tim Peters4ad82172004-08-30 17:02:04 +00008178 /* Obtain pointers to the CryptoAPI functions
8179 This will fail on some early versions of Win95 */
8180 pCryptAcquireContext = (CRYPTACQUIRECONTEXTA)GetProcAddress(
8181 hAdvAPI32,
8182 "CryptAcquireContextA");
8183 if (pCryptAcquireContext == NULL)
8184 return PyErr_Format(PyExc_NotImplementedError,
8185 "CryptAcquireContextA not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008186
Tim Peters4ad82172004-08-30 17:02:04 +00008187 pCryptGenRandom = (CRYPTGENRANDOM)GetProcAddress(
8188 hAdvAPI32, "CryptGenRandom");
Georg Brandl74bb7832006-09-06 06:03:59 +00008189 if (pCryptGenRandom == NULL)
Tim Peters4ad82172004-08-30 17:02:04 +00008190 return PyErr_Format(PyExc_NotImplementedError,
8191 "CryptGenRandom not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008192
Tim Peters4ad82172004-08-30 17:02:04 +00008193 /* Acquire context */
8194 if (! pCryptAcquireContext(&hCryptProv, NULL, NULL,
8195 PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
8196 return win32_error("CryptAcquireContext", NULL);
8197 }
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008198
Tim Peters4ad82172004-08-30 17:02:04 +00008199 /* Allocate bytes */
Tim Petersd3115382004-08-30 17:36:46 +00008200 result = PyString_FromStringAndSize(NULL, howMany);
8201 if (result != NULL) {
8202 /* Get random data */
8203 if (! pCryptGenRandom(hCryptProv, howMany, (unsigned char*)
8204 PyString_AS_STRING(result))) {
8205 Py_DECREF(result);
8206 return win32_error("CryptGenRandom", NULL);
8207 }
Tim Peters4ad82172004-08-30 17:02:04 +00008208 }
Tim Petersd3115382004-08-30 17:36:46 +00008209 return result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008210}
8211#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00008212
Neal Norwitz2a30cd02006-07-10 01:18:57 +00008213#ifdef __VMS
8214/* Use openssl random routine */
8215#include <openssl/rand.h>
8216PyDoc_STRVAR(vms_urandom__doc__,
8217"urandom(n) -> str\n\n\
8218Return a string of n random bytes suitable for cryptographic use.");
8219
8220static PyObject*
8221vms_urandom(PyObject *self, PyObject *args)
8222{
8223 int howMany;
8224 PyObject* result;
8225
8226 /* Read arguments */
8227 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
8228 return NULL;
8229 if (howMany < 0)
8230 return PyErr_Format(PyExc_ValueError,
8231 "negative argument not allowed");
8232
8233 /* Allocate bytes */
8234 result = PyString_FromStringAndSize(NULL, howMany);
8235 if (result != NULL) {
8236 /* Get random data */
8237 if (RAND_pseudo_bytes((unsigned char*)
8238 PyString_AS_STRING(result),
8239 howMany) < 0) {
8240 Py_DECREF(result);
8241 return PyErr_Format(PyExc_ValueError,
8242 "RAND_pseudo_bytes");
8243 }
8244 }
8245 return result;
8246}
8247#endif
8248
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008249static PyMethodDef posix_methods[] = {
8250 {"access", posix_access, METH_VARARGS, posix_access__doc__},
8251#ifdef HAVE_TTYNAME
8252 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
8253#endif
8254 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
Martin v. Löwis382abef2007-02-19 10:55:19 +00008255#ifdef HAVE_CHFLAGS
8256 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
8257#endif /* HAVE_CHFLAGS */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008258 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Christian Heimes36281872007-11-30 21:11:28 +00008259#ifdef HAVE_FCHMOD
8260 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
8261#endif /* HAVE_FCHMOD */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008262#ifdef HAVE_CHOWN
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008263 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008264#endif /* HAVE_CHOWN */
Christian Heimes36281872007-11-30 21:11:28 +00008265#ifdef HAVE_LCHMOD
8266 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
8267#endif /* HAVE_LCHMOD */
8268#ifdef HAVE_FCHOWN
8269 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
8270#endif /* HAVE_FCHOWN */
Martin v. Löwis382abef2007-02-19 10:55:19 +00008271#ifdef HAVE_LCHFLAGS
8272 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
8273#endif /* HAVE_LCHFLAGS */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00008274#ifdef HAVE_LCHOWN
8275 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
8276#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +00008277#ifdef HAVE_CHROOT
8278 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
8279#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008280#ifdef HAVE_CTERMID
Neal Norwitze241ce82003-02-17 18:17:05 +00008281 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008282#endif
Guido van Rossum36bc6801995-06-14 22:54:23 +00008283#ifdef HAVE_GETCWD
Neal Norwitze241ce82003-02-17 18:17:05 +00008284 {"getcwd", posix_getcwd, METH_NOARGS, posix_getcwd__doc__},
Walter Dörwald3b918c32002-11-21 20:18:46 +00008285#ifdef Py_USING_UNICODE
Neal Norwitze241ce82003-02-17 18:17:05 +00008286 {"getcwdu", posix_getcwdu, METH_NOARGS, posix_getcwdu__doc__},
Guido van Rossum36bc6801995-06-14 22:54:23 +00008287#endif
Walter Dörwald3b918c32002-11-21 20:18:46 +00008288#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00008289#ifdef HAVE_LINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008290 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008291#endif /* HAVE_LINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008292 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
8293 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
8294 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00008295#ifdef HAVE_NICE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008296 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008297#endif /* HAVE_NICE */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008298#ifdef HAVE_READLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008299 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008300#endif /* HAVE_READLINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008301 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
8302 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
8303 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00008304 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00008305#ifdef HAVE_SYMLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008306 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008307#endif /* HAVE_SYMLINK */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008308#ifdef HAVE_SYSTEM
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008309 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008310#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008311 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00008312#ifdef HAVE_UNAME
Neal Norwitze241ce82003-02-17 18:17:05 +00008313 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008314#endif /* HAVE_UNAME */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008315 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
8316 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
8317 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00008318#ifdef HAVE_TIMES
Neal Norwitze241ce82003-02-17 18:17:05 +00008319 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008320#endif /* HAVE_TIMES */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008321 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008322#ifdef HAVE_EXECV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008323 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
8324 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008325#endif /* HAVE_EXECV */
Guido van Rossuma1065681999-01-25 23:20:23 +00008326#ifdef HAVE_SPAWNV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008327 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
8328 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00008329#if defined(PYOS_OS2)
8330 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
8331 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
8332#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00008333#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +00008334#ifdef HAVE_FORK1
Neal Norwitze241ce82003-02-17 18:17:05 +00008335 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +00008336#endif /* HAVE_FORK1 */
Guido van Rossumad0ee831995-03-01 10:34:45 +00008337#ifdef HAVE_FORK
Neal Norwitze241ce82003-02-17 18:17:05 +00008338 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00008339#endif /* HAVE_FORK */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00008340#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Neal Norwitze241ce82003-02-17 18:17:05 +00008341 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +00008342#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00008343#ifdef HAVE_FORKPTY
Neal Norwitze241ce82003-02-17 18:17:05 +00008344 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +00008345#endif /* HAVE_FORKPTY */
Guido van Rossumad0ee831995-03-01 10:34:45 +00008346#ifdef HAVE_GETEGID
Neal Norwitze241ce82003-02-17 18:17:05 +00008347 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00008348#endif /* HAVE_GETEGID */
8349#ifdef HAVE_GETEUID
Neal Norwitze241ce82003-02-17 18:17:05 +00008350 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00008351#endif /* HAVE_GETEUID */
8352#ifdef HAVE_GETGID
Neal Norwitze241ce82003-02-17 18:17:05 +00008353 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00008354#endif /* HAVE_GETGID */
Fred Drakec9680921999-12-13 16:37:25 +00008355#ifdef HAVE_GETGROUPS
Neal Norwitze241ce82003-02-17 18:17:05 +00008356 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00008357#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00008358 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00008359#ifdef HAVE_GETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00008360 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008361#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00008362#ifdef HAVE_GETPPID
Neal Norwitze241ce82003-02-17 18:17:05 +00008363 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00008364#endif /* HAVE_GETPPID */
8365#ifdef HAVE_GETUID
Neal Norwitze241ce82003-02-17 18:17:05 +00008366 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00008367#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +00008368#ifdef HAVE_GETLOGIN
Neal Norwitze241ce82003-02-17 18:17:05 +00008369 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +00008370#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +00008371#ifdef HAVE_KILL
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008372 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00008373#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00008374#ifdef HAVE_KILLPG
8375 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
8376#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +00008377#ifdef HAVE_PLOCK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008378 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +00008379#endif /* HAVE_PLOCK */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008380#ifdef HAVE_POPEN
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008381 {"popen", posix_popen, METH_VARARGS, posix_popen__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00008382#ifdef MS_WINDOWS
Fredrik Lundhffb9c772000-07-09 14:49:51 +00008383 {"popen2", win32_popen2, METH_VARARGS},
8384 {"popen3", win32_popen3, METH_VARARGS},
8385 {"popen4", win32_popen4, METH_VARARGS},
Tim Petersf58a7aa2000-09-22 10:05:54 +00008386 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00008387#else
8388#if defined(PYOS_OS2) && defined(PYCC_GCC)
8389 {"popen2", os2emx_popen2, METH_VARARGS},
8390 {"popen3", os2emx_popen3, METH_VARARGS},
8391 {"popen4", os2emx_popen4, METH_VARARGS},
8392#endif
Fredrik Lundhffb9c772000-07-09 14:49:51 +00008393#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008394#endif /* HAVE_POPEN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008395#ifdef HAVE_SETUID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008396 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008397#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00008398#ifdef HAVE_SETEUID
8399 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
8400#endif /* HAVE_SETEUID */
8401#ifdef HAVE_SETEGID
8402 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
8403#endif /* HAVE_SETEGID */
8404#ifdef HAVE_SETREUID
8405 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
8406#endif /* HAVE_SETREUID */
8407#ifdef HAVE_SETREGID
8408 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
8409#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008410#ifdef HAVE_SETGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008411 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008412#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00008413#ifdef HAVE_SETGROUPS
Georg Brandl96a8c392006-05-29 21:04:52 +00008414 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00008415#endif /* HAVE_SETGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +00008416#ifdef HAVE_GETPGID
8417 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
8418#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008419#ifdef HAVE_SETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00008420 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008421#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00008422#ifdef HAVE_WAIT
Neal Norwitze241ce82003-02-17 18:17:05 +00008423 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00008424#endif /* HAVE_WAIT */
Neal Norwitz05a45592006-03-20 06:30:08 +00008425#ifdef HAVE_WAIT3
8426 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
8427#endif /* HAVE_WAIT3 */
8428#ifdef HAVE_WAIT4
8429 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
8430#endif /* HAVE_WAIT4 */
Tim Petersab034fa2002-02-01 11:27:43 +00008431#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008432 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008433#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00008434#ifdef HAVE_GETSID
8435 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
8436#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008437#ifdef HAVE_SETSID
Neal Norwitze241ce82003-02-17 18:17:05 +00008438 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008439#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008440#ifdef HAVE_SETPGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008441 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008442#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008443#ifdef HAVE_TCGETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008444 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008445#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008446#ifdef HAVE_TCSETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008447 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00008448#endif /* HAVE_TCSETPGRP */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008449 {"open", posix_open, METH_VARARGS, posix_open__doc__},
8450 {"close", posix_close, METH_VARARGS, posix_close__doc__},
8451 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
8452 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
8453 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
8454 {"read", posix_read, METH_VARARGS, posix_read__doc__},
8455 {"write", posix_write, METH_VARARGS, posix_write__doc__},
8456 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
8457 {"fdopen", posix_fdopen, METH_VARARGS, posix_fdopen__doc__},
Skip Montanaro1517d842000-07-19 14:34:14 +00008458 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008459#ifdef HAVE_PIPE
Neal Norwitze241ce82003-02-17 18:17:05 +00008460 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008461#endif
8462#ifdef HAVE_MKFIFO
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008463 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008464#endif
Neal Norwitz11690112002-07-30 01:08:28 +00008465#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis06a83e92002-04-14 10:19:44 +00008466 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
8467#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00008468#ifdef HAVE_DEVICE_MACROS
8469 {"major", posix_major, METH_VARARGS, posix_major__doc__},
8470 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
8471 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
8472#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008473#ifdef HAVE_FTRUNCATE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008474 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00008475#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00008476#ifdef HAVE_PUTENV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008477 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00008478#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00008479#ifdef HAVE_UNSETENV
8480 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
8481#endif
Guido van Rossumb6a47161997-09-15 22:54:34 +00008482#ifdef HAVE_STRERROR
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008483 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Guido van Rossumb6a47161997-09-15 22:54:34 +00008484#endif
Fred Drake4d1e64b2002-04-15 19:40:07 +00008485#ifdef HAVE_FCHDIR
8486 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
8487#endif
Guido van Rossum21142a01999-01-08 21:05:37 +00008488#ifdef HAVE_FSYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00008489 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00008490#endif
8491#ifdef HAVE_FDATASYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00008492 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00008493#endif
Guido van Rossumc9641791998-08-04 15:26:23 +00008494#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +00008495#ifdef WCOREDUMP
8496 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
8497#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00008498#ifdef WIFCONTINUED
8499 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
8500#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +00008501#ifdef WIFSTOPPED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008502 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00008503#endif /* WIFSTOPPED */
8504#ifdef WIFSIGNALED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008505 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00008506#endif /* WIFSIGNALED */
8507#ifdef WIFEXITED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008508 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00008509#endif /* WIFEXITED */
8510#ifdef WEXITSTATUS
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008511 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00008512#endif /* WEXITSTATUS */
8513#ifdef WTERMSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008514 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00008515#endif /* WTERMSIG */
8516#ifdef WSTOPSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008517 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00008518#endif /* WSTOPSIG */
8519#endif /* HAVE_SYS_WAIT_H */
Martin v. Löwis5f5d99c2006-05-16 07:05:37 +00008520#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008521 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00008522#endif
Martin v. Löwis5f5d99c2006-05-16 07:05:37 +00008523#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008524 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00008525#endif
Guido van Rossume2ad6332001-01-08 17:51:55 +00008526#ifdef HAVE_TMPFILE
Neal Norwitze241ce82003-02-17 18:17:05 +00008527 {"tmpfile", posix_tmpfile, METH_NOARGS, posix_tmpfile__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008528#endif
8529#ifdef HAVE_TEMPNAM
8530 {"tempnam", posix_tempnam, METH_VARARGS, posix_tempnam__doc__},
8531#endif
8532#ifdef HAVE_TMPNAM
Neal Norwitze241ce82003-02-17 18:17:05 +00008533 {"tmpnam", posix_tmpnam, METH_NOARGS, posix_tmpnam__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008534#endif
Fred Drakec9680921999-12-13 16:37:25 +00008535#ifdef HAVE_CONFSTR
8536 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
8537#endif
8538#ifdef HAVE_SYSCONF
8539 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
8540#endif
8541#ifdef HAVE_FPATHCONF
8542 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
8543#endif
8544#ifdef HAVE_PATHCONF
8545 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
8546#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00008547 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00008548#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00008549 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
8550#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00008551#ifdef HAVE_GETLOADAVG
Neal Norwitze241ce82003-02-17 18:17:05 +00008552 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +00008553#endif
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008554 #ifdef MS_WINDOWS
8555 {"urandom", win32_urandom, METH_VARARGS, win32_urandom__doc__},
8556 #endif
Neal Norwitz2a30cd02006-07-10 01:18:57 +00008557 #ifdef __VMS
8558 {"urandom", vms_urandom, METH_VARARGS, vms_urandom__doc__},
8559 #endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00008560 {NULL, NULL} /* Sentinel */
8561};
8562
8563
Barry Warsaw4a342091996-12-19 23:50:02 +00008564static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00008565ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +00008566{
Fred Drake4d1e64b2002-04-15 19:40:07 +00008567 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +00008568}
8569
Guido van Rossumd48f2521997-12-05 22:19:34 +00008570#if defined(PYOS_OS2)
8571/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +00008572static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +00008573{
8574 APIRET rc;
8575 ULONG values[QSV_MAX+1];
8576 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +00008577 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +00008578
8579 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +00008580 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +00008581 Py_END_ALLOW_THREADS
8582
8583 if (rc != NO_ERROR) {
8584 os2_error(rc);
8585 return -1;
8586 }
8587
Fred Drake4d1e64b2002-04-15 19:40:07 +00008588 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
8589 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
8590 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
8591 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
8592 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
8593 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
8594 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00008595
8596 switch (values[QSV_VERSION_MINOR]) {
8597 case 0: ver = "2.00"; break;
8598 case 10: ver = "2.10"; break;
8599 case 11: ver = "2.11"; break;
8600 case 30: ver = "3.00"; break;
8601 case 40: ver = "4.00"; break;
8602 case 50: ver = "5.00"; break;
8603 default:
Tim Peters885d4572001-11-28 20:27:42 +00008604 PyOS_snprintf(tmp, sizeof(tmp),
8605 "%d-%d", values[QSV_VERSION_MAJOR],
8606 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +00008607 ver = &tmp[0];
8608 }
8609
8610 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +00008611 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +00008612 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00008613
8614 /* Add Indicator of Which Drive was Used to Boot the System */
8615 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
8616 tmp[1] = ':';
8617 tmp[2] = '\0';
8618
Fred Drake4d1e64b2002-04-15 19:40:07 +00008619 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +00008620}
8621#endif
8622
Barry Warsaw4a342091996-12-19 23:50:02 +00008623static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00008624all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +00008625{
Guido van Rossum94f6f721999-01-06 18:42:14 +00008626#ifdef F_OK
8627 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00008628#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00008629#ifdef R_OK
8630 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00008631#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00008632#ifdef W_OK
8633 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00008634#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00008635#ifdef X_OK
8636 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00008637#endif
Fred Drakec9680921999-12-13 16:37:25 +00008638#ifdef NGROUPS_MAX
8639 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
8640#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008641#ifdef TMP_MAX
8642 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
8643#endif
Fred Drake106c1a02002-04-23 15:58:02 +00008644#ifdef WCONTINUED
8645 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
8646#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00008647#ifdef WNOHANG
8648 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00008649#endif
Fred Drake106c1a02002-04-23 15:58:02 +00008650#ifdef WUNTRACED
8651 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
8652#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00008653#ifdef O_RDONLY
8654 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
8655#endif
8656#ifdef O_WRONLY
8657 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
8658#endif
8659#ifdef O_RDWR
8660 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
8661#endif
8662#ifdef O_NDELAY
8663 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
8664#endif
8665#ifdef O_NONBLOCK
8666 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
8667#endif
8668#ifdef O_APPEND
8669 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
8670#endif
8671#ifdef O_DSYNC
8672 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
8673#endif
8674#ifdef O_RSYNC
8675 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
8676#endif
8677#ifdef O_SYNC
8678 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
8679#endif
8680#ifdef O_NOCTTY
8681 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
8682#endif
8683#ifdef O_CREAT
8684 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
8685#endif
8686#ifdef O_EXCL
8687 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
8688#endif
8689#ifdef O_TRUNC
8690 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
8691#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +00008692#ifdef O_BINARY
8693 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
8694#endif
8695#ifdef O_TEXT
8696 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
8697#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00008698#ifdef O_LARGEFILE
8699 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
8700#endif
Skip Montanaro5ff14922005-05-16 02:42:22 +00008701#ifdef O_SHLOCK
8702 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
8703#endif
8704#ifdef O_EXLOCK
8705 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
8706#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00008707
Tim Peters5aa91602002-01-30 05:46:57 +00008708/* MS Windows */
8709#ifdef O_NOINHERIT
8710 /* Don't inherit in child processes. */
8711 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
8712#endif
8713#ifdef _O_SHORT_LIVED
8714 /* Optimize for short life (keep in memory). */
8715 /* MS forgot to define this one with a non-underscore form too. */
8716 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
8717#endif
8718#ifdef O_TEMPORARY
8719 /* Automatically delete when last handle is closed. */
8720 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
8721#endif
8722#ifdef O_RANDOM
8723 /* Optimize for random access. */
8724 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
8725#endif
8726#ifdef O_SEQUENTIAL
8727 /* Optimize for sequential access. */
8728 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
8729#endif
8730
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00008731/* GNU extensions. */
8732#ifdef O_DIRECT
8733 /* Direct disk access. */
8734 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
8735#endif
8736#ifdef O_DIRECTORY
8737 /* Must be a directory. */
8738 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
8739#endif
8740#ifdef O_NOFOLLOW
8741 /* Do not follow links. */
8742 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
8743#endif
Georg Brandlb67da6e2007-11-24 13:56:09 +00008744#ifdef O_NOATIME
8745 /* Do not update the access time. */
8746 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
8747#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00008748
Barry Warsaw5676bd12003-01-07 20:57:09 +00008749 /* These come from sysexits.h */
8750#ifdef EX_OK
8751 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008752#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008753#ifdef EX_USAGE
8754 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008755#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008756#ifdef EX_DATAERR
8757 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008758#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008759#ifdef EX_NOINPUT
8760 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008761#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008762#ifdef EX_NOUSER
8763 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008764#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008765#ifdef EX_NOHOST
8766 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008767#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008768#ifdef EX_UNAVAILABLE
8769 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008770#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008771#ifdef EX_SOFTWARE
8772 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008773#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008774#ifdef EX_OSERR
8775 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008776#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008777#ifdef EX_OSFILE
8778 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008779#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008780#ifdef EX_CANTCREAT
8781 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008782#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008783#ifdef EX_IOERR
8784 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008785#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008786#ifdef EX_TEMPFAIL
8787 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008788#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008789#ifdef EX_PROTOCOL
8790 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008791#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008792#ifdef EX_NOPERM
8793 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008794#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008795#ifdef EX_CONFIG
8796 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008797#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008798#ifdef EX_NOTFOUND
8799 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00008800#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +00008801
Guido van Rossum246bc171999-02-01 23:54:31 +00008802#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00008803#if defined(PYOS_OS2) && defined(PYCC_GCC)
8804 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
8805 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
8806 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
8807 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
8808 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
8809 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
8810 if (ins(d, "P_PM", (long)P_PM)) return -1;
8811 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
8812 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
8813 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
8814 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
8815 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
8816 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
8817 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
8818 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
8819 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
8820 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
8821 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
8822 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
8823 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
8824#else
Guido van Rossum7d385291999-02-16 19:38:04 +00008825 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
8826 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
8827 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
8828 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
8829 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +00008830#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00008831#endif
Guido van Rossum246bc171999-02-01 23:54:31 +00008832
Guido van Rossumd48f2521997-12-05 22:19:34 +00008833#if defined(PYOS_OS2)
8834 if (insertvalues(d)) return -1;
8835#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00008836 return 0;
8837}
8838
8839
Tim Peters5aa91602002-01-30 05:46:57 +00008840#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Guido van Rossum0cb96de1997-10-01 04:29:29 +00008841#define INITFUNC initnt
8842#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +00008843
8844#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00008845#define INITFUNC initos2
8846#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +00008847
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00008848#else
Guido van Rossum0cb96de1997-10-01 04:29:29 +00008849#define INITFUNC initposix
8850#define MODNAME "posix"
8851#endif
8852
Mark Hammondfe51c6d2002-08-02 02:27:13 +00008853PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00008854INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +00008855{
Fred Drake4d1e64b2002-04-15 19:40:07 +00008856 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +00008857
Fred Drake4d1e64b2002-04-15 19:40:07 +00008858 m = Py_InitModule3(MODNAME,
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00008859 posix_methods,
Fred Drake4d1e64b2002-04-15 19:40:07 +00008860 posix__doc__);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00008861 if (m == NULL)
8862 return;
Tim Peters5aa91602002-01-30 05:46:57 +00008863
Guido van Rossum0cb96de1997-10-01 04:29:29 +00008864 /* Initialize environ dictionary */
Guido van Rossumb6775db1994-08-01 11:34:53 +00008865 v = convertenviron();
Fred Drake4d1e64b2002-04-15 19:40:07 +00008866 Py_XINCREF(v);
8867 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
Guido van Rossum0cb96de1997-10-01 04:29:29 +00008868 return;
Barry Warsaw53699e91996-12-10 23:23:01 +00008869 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +00008870
Fred Drake4d1e64b2002-04-15 19:40:07 +00008871 if (all_ins(m))
Barry Warsaw4a342091996-12-19 23:50:02 +00008872 return;
8873
Fred Drake4d1e64b2002-04-15 19:40:07 +00008874 if (setup_confname_tables(m))
Fred Drakebec628d1999-12-15 18:31:10 +00008875 return;
8876
Fred Drake4d1e64b2002-04-15 19:40:07 +00008877 Py_INCREF(PyExc_OSError);
8878 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +00008879
Guido van Rossumb3d39562000-01-31 18:41:26 +00008880#ifdef HAVE_PUTENV
Neil Schemenauer19030a02001-01-16 04:27:47 +00008881 if (posix_putenv_garbage == NULL)
8882 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +00008883#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008884
Martin v. Löwis19ab6c92006-04-16 18:55:50 +00008885 if (!initialized) {
8886 stat_result_desc.name = MODNAME ".stat_result";
8887 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
8888 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
8889 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
8890 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
8891 structseq_new = StatResultType.tp_new;
8892 StatResultType.tp_new = statresult_new;
8893
8894 statvfs_result_desc.name = MODNAME ".statvfs_result";
8895 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
8896 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00008897 Py_INCREF((PyObject*) &StatResultType);
8898 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
Fred Drake4d1e64b2002-04-15 19:40:07 +00008899 Py_INCREF((PyObject*) &StatVFSResultType);
8900 PyModule_AddObject(m, "statvfs_result",
8901 (PyObject*) &StatVFSResultType);
Martin v. Löwis19ab6c92006-04-16 18:55:50 +00008902 initialized = 1;
Ronald Oussorend06b6f22006-04-23 11:59:25 +00008903
8904#ifdef __APPLE__
8905 /*
8906 * Step 2 of weak-linking support on Mac OS X.
8907 *
8908 * The code below removes functions that are not available on the
8909 * currently active platform.
8910 *
8911 * This block allow one to use a python binary that was build on
8912 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
8913 * OSX 10.4.
8914 */
8915#ifdef HAVE_FSTATVFS
8916 if (fstatvfs == NULL) {
8917 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
8918 return;
8919 }
8920 }
8921#endif /* HAVE_FSTATVFS */
8922
8923#ifdef HAVE_STATVFS
8924 if (statvfs == NULL) {
8925 if (PyObject_DelAttrString(m, "statvfs") == -1) {
8926 return;
8927 }
8928 }
8929#endif /* HAVE_STATVFS */
8930
8931# ifdef HAVE_LCHOWN
8932 if (lchown == NULL) {
8933 if (PyObject_DelAttrString(m, "lchown") == -1) {
8934 return;
8935 }
8936 }
8937#endif /* HAVE_LCHOWN */
8938
8939
8940#endif /* __APPLE__ */
8941
Guido van Rossumb6775db1994-08-01 11:34:53 +00008942}
Anthony Baxterac6bd462006-04-13 02:06:09 +00008943
8944#ifdef __cplusplus
8945}
8946#endif
8947
Martin v. Löwis18aaa562006-10-15 08:43:33 +00008948