blob: aec9f745f051cea2fceb2f442a94a55e38ffc505 [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
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000016#include "Python.h"
17#include "structseq.h"
18
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000019#if defined(__VMS)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000020# include <unixio.h>
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000021#endif /* defined(__VMS) */
22
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000023PyDoc_STRVAR(posix__doc__,
Guido van Rossumec4f4ac1997-06-02 22:20:51 +000024"This module provides access to operating system functionality that is\n\
25standardized by the C Standard and the POSIX standard (a thinly\n\
26disguised Unix interface). Refer to the library manual and\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000027corresponding Unix manual entries for more information on calls.");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000028
Martin v. Löwis0073f2e2002-11-21 23:52:35 +000029#ifndef Py_USING_UNICODE
30/* This is used in signatures of functions. */
31#define Py_UNICODE void
32#endif
33
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000034#if defined(PYOS_OS2)
35#define INCL_DOS
36#define INCL_DOSERRORS
37#define INCL_DOSPROCESS
38#define INCL_NOPMAPI
39#include <os2.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000040#if defined(PYCC_GCC)
41#include <ctype.h>
42#include <io.h>
43#include <stdio.h>
44#include <process.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000045#endif
Andrew MacIntyreda4d6cb2004-03-29 11:53:38 +000046#include "osdefs.h"
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000047#endif
48
Guido van Rossumb6775db1994-08-01 11:34:53 +000049#include <sys/types.h>
50#include <sys/stat.h>
Guido van Rossuma6535fd2001-10-18 19:44:10 +000051
Guido van Rossum36bc6801995-06-14 22:54:23 +000052#ifdef HAVE_SYS_WAIT_H
53#include <sys/wait.h> /* For WNOHANG */
54#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +000055
Guido van Rossuma376cc51996-12-05 23:43:35 +000056#include <signal.h>
Guido van Rossuma376cc51996-12-05 23:43:35 +000057
Guido van Rossumb6775db1994-08-01 11:34:53 +000058#ifdef HAVE_FCNTL_H
59#include <fcntl.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000060#endif /* HAVE_FCNTL_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +000061
Guido van Rossuma6535fd2001-10-18 19:44:10 +000062#ifdef HAVE_GRP_H
63#include <grp.h>
64#endif
65
Barry Warsaw5676bd12003-01-07 20:57:09 +000066#ifdef HAVE_SYSEXITS_H
67#include <sysexits.h>
68#endif /* HAVE_SYSEXITS_H */
69
Guido van Rossuma4916fa1996-05-23 22:58:55 +000070/* Various compilers have only certain posix functions */
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +000071/* XXX Gosh I wish these were all moved into pyconfig.h */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000072#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000073#include <process.h>
74#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +000075#if defined(__WATCOMC__) && !defined(__QNX__) /* Watcom compiler */
Guido van Rossuma4916fa1996-05-23 22:58:55 +000076#define HAVE_GETCWD 1
77#define HAVE_OPENDIR 1
78#define HAVE_SYSTEM 1
79#if defined(__OS2__)
80#define HAVE_EXECV 1
81#define HAVE_WAIT 1
Guido van Rossumad0ee831995-03-01 10:34:45 +000082#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +000083#include <process.h>
84#else
85#ifdef __BORLANDC__ /* Borland compiler */
86#define HAVE_EXECV 1
87#define HAVE_GETCWD 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +000088#define HAVE_OPENDIR 1
89#define HAVE_PIPE 1
90#define HAVE_POPEN 1
91#define HAVE_SYSTEM 1
92#define HAVE_WAIT 1
93#else
94#ifdef _MSC_VER /* Microsoft compiler */
Guido van Rossum8d665e61996-06-26 18:22:49 +000095#define HAVE_GETCWD 1
Guido van Rossuma1065681999-01-25 23:20:23 +000096#define HAVE_SPAWNV 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +000097#define HAVE_EXECV 1
98#define HAVE_PIPE 1
99#define HAVE_POPEN 1
100#define HAVE_SYSTEM 1
Tim Petersab034fa2002-02-01 11:27:43 +0000101#define HAVE_CWAIT 1
Tim Peters11b23062003-04-23 02:39:17 +0000102#define HAVE_FSYNC 1
103#define fsync _commit
Andrew MacIntyre6c73af22002-03-03 03:07:07 +0000104#else
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000105#if defined(PYOS_OS2) && defined(PYCC_GCC) || defined(__VMS)
106/* Everything needed is defined in PC/os2emx/pyconfig.h or vms/pyconfig.h */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000107#else /* all other compilers */
108/* Unix functions that the configure script doesn't check for */
109#define HAVE_EXECV 1
110#define HAVE_FORK 1
Guido van Rossum2242f2f2001-04-11 20:58:20 +0000111#if defined(__USLC__) && defined(__SCO_VERSION__) /* SCO UDK Compiler */
112#define HAVE_FORK1 1
113#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000114#define HAVE_GETCWD 1
115#define HAVE_GETEGID 1
116#define HAVE_GETEUID 1
117#define HAVE_GETGID 1
118#define HAVE_GETPPID 1
119#define HAVE_GETUID 1
120#define HAVE_KILL 1
121#define HAVE_OPENDIR 1
122#define HAVE_PIPE 1
Martin v. Löwis212ede62003-09-20 11:20:30 +0000123#ifndef __rtems__
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000124#define HAVE_POPEN 1
Martin v. Löwis212ede62003-09-20 11:20:30 +0000125#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000126#define HAVE_SYSTEM 1
127#define HAVE_WAIT 1
Guido van Rossumd371ff11999-01-25 16:12:23 +0000128#define HAVE_TTYNAME 1
Martin v. Löwis7a924e62003-03-05 14:15:21 +0000129#endif /* PYOS_OS2 && PYCC_GCC && __VMS */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000130#endif /* _MSC_VER */
131#endif /* __BORLANDC__ */
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000132#endif /* ! __WATCOMC__ || __QNX__ */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000133#endif /* ! __IBMC__ */
Guido van Rossumad0ee831995-03-01 10:34:45 +0000134
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000135#ifndef _MSC_VER
Guido van Rossum36bc6801995-06-14 22:54:23 +0000136
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000137#if defined(__sgi)&&_COMPILER_VERSION>=700
138/* declare ctermid_r if compiling with MIPSPro 7.x in ANSI C mode
139 (default) */
140extern char *ctermid_r(char *);
141#endif
142
Thomas Wouters1e0c2f42000-07-24 16:06:23 +0000143#ifndef HAVE_UNISTD_H
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000144#if defined(PYCC_VACPP)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000145extern int mkdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000146#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000147#if ( defined(__WATCOMC__) || defined(_MSC_VER) ) && !defined(__QNX__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000148extern int mkdir(const char *);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000149#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000150extern int mkdir(const char *, mode_t);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000151#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000152#endif
153#if defined(__IBMC__) || defined(__IBMCPP__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000154extern int chdir(char *);
155extern int rmdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000156#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000157extern int chdir(const char *);
158extern int rmdir(const char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000159#endif
Tim Peters58e0a8c2001-05-14 22:32:33 +0000160#ifdef __BORLANDC__
161extern int chmod(const char *, int);
162#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000163extern int chmod(const char *, mode_t);
Tim Peters58e0a8c2001-05-14 22:32:33 +0000164#endif
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000165extern int chown(const char *, uid_t, gid_t);
166extern char *getcwd(char *, int);
167extern char *strerror(int);
168extern int link(const char *, const char *);
169extern int rename(const char *, const char *);
170extern int stat(const char *, struct stat *);
171extern int unlink(const char *);
172extern int pclose(FILE *);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000173#ifdef HAVE_SYMLINK
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000174extern int symlink(const char *, const char *);
Guido van Rossuma38a5031995-02-17 15:11:36 +0000175#endif /* HAVE_SYMLINK */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000176#ifdef HAVE_LSTAT
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000177extern int lstat(const char *, struct stat *);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000178#endif /* HAVE_LSTAT */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000179#endif /* !HAVE_UNISTD_H */
Guido van Rossum36bc6801995-06-14 22:54:23 +0000180
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000181#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000182
Guido van Rossumb6775db1994-08-01 11:34:53 +0000183#ifdef HAVE_UTIME_H
184#include <utime.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000185#endif /* HAVE_UTIME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000186
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000187#ifdef HAVE_SYS_UTIME_H
188#include <sys/utime.h>
189#define HAVE_UTIME_H /* pretend we do for the rest of this file */
190#endif /* HAVE_SYS_UTIME_H */
191
Guido van Rossumb6775db1994-08-01 11:34:53 +0000192#ifdef HAVE_SYS_TIMES_H
193#include <sys/times.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000194#endif /* HAVE_SYS_TIMES_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000195
196#ifdef HAVE_SYS_PARAM_H
197#include <sys/param.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000198#endif /* HAVE_SYS_PARAM_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000199
200#ifdef HAVE_SYS_UTSNAME_H
201#include <sys/utsname.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000202#endif /* HAVE_SYS_UTSNAME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000203
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000204#ifdef HAVE_DIRENT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000205#include <dirent.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000206#define NAMLEN(dirent) strlen((dirent)->d_name)
207#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000208#if defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000209#include <direct.h>
210#define NAMLEN(dirent) strlen((dirent)->d_name)
211#else
Guido van Rossumb6775db1994-08-01 11:34:53 +0000212#define dirent direct
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000213#define NAMLEN(dirent) (dirent)->d_namlen
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000214#endif
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000215#ifdef HAVE_SYS_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000216#include <sys/ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000217#endif
218#ifdef HAVE_SYS_DIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000219#include <sys/dir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000220#endif
221#ifdef HAVE_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000222#include <ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000223#endif
224#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000225
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000226#ifdef _MSC_VER
Guido van Rossumb6775db1994-08-01 11:34:53 +0000227#include <direct.h>
228#include <io.h>
229#include <process.h>
Tim Petersbc2e10e2002-03-03 23:17:02 +0000230#include "osdefs.h"
Tim Petersee66d0c2002-07-15 16:10:55 +0000231#define WIN32_LEAN_AND_MEAN
Guido van Rossumb6775db1994-08-01 11:34:53 +0000232#include <windows.h>
Tim Petersee66d0c2002-07-15 16:10:55 +0000233#include <shellapi.h> /* for ShellExecute() */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000234#define popen _popen
Guido van Rossum794d8131994-08-23 13:48:48 +0000235#define pclose _pclose
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000236#endif /* _MSC_VER */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000237
Guido van Rossumd48f2521997-12-05 22:19:34 +0000238#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000239#include <io.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000240#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000241
Tim Petersbc2e10e2002-03-03 23:17:02 +0000242#ifndef MAXPATHLEN
243#define MAXPATHLEN 1024
244#endif /* MAXPATHLEN */
245
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000246#ifdef UNION_WAIT
247/* Emulate some macros on systems that have a union instead of macros */
248
249#ifndef WIFEXITED
250#define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
251#endif
252
253#ifndef WEXITSTATUS
254#define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
255#endif
256
257#ifndef WTERMSIG
258#define WTERMSIG(u_wait) ((u_wait).w_termsig)
259#endif
260
261#endif /* UNION_WAIT */
262
Greg Wardb48bc172000-03-01 21:51:56 +0000263/* Don't use the "_r" form if we don't need it (also, won't have a
264 prototype for it, at least on Solaris -- maybe others as well?). */
265#if defined(HAVE_CTERMID_R) && defined(WITH_THREAD)
266#define USE_CTERMID_R
267#endif
268
269#if defined(HAVE_TMPNAM_R) && defined(WITH_THREAD)
270#define USE_TMPNAM_R
271#endif
272
Fred Drake699f3522000-06-29 21:12:41 +0000273/* choose the appropriate stat and fstat functions and return structs */
Guido van Rossum64529cd2000-06-30 22:45:12 +0000274#undef STAT
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000275#if defined(MS_WIN64) || defined(MS_WINDOWS)
Fred Drake699f3522000-06-29 21:12:41 +0000276# define STAT _stati64
277# define FSTAT _fstati64
278# define STRUCT_STAT struct _stati64
279#else
280# define STAT stat
281# define FSTAT fstat
282# define STRUCT_STAT struct stat
283#endif
284
Tim Peters11b23062003-04-23 02:39:17 +0000285#if defined(MAJOR_IN_MKDEV)
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000286#include <sys/mkdev.h>
287#else
288#if defined(MAJOR_IN_SYSMACROS)
289#include <sys/sysmacros.h>
290#endif
Neal Norwitz3d949422002-04-20 13:46:43 +0000291#if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
292#include <sys/mkdev.h>
293#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000294#endif
Fred Drake699f3522000-06-29 21:12:41 +0000295
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000296/* Return a dictionary corresponding to the POSIX environment table */
Jack Jansenea0c3822002-08-01 21:57:49 +0000297#ifdef WITH_NEXT_FRAMEWORK
298/* On Darwin/MacOSX a shared library or framework has no access to
299** environ directly, we must obtain it with _NSGetEnviron().
300*/
301#include <crt_externs.h>
302static char **environ;
303#elif !defined(_MSC_VER) && ( !defined(__WATCOMC__) || defined(__QNX__) )
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000304extern char **environ;
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000305#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000306
Barry Warsaw53699e91996-12-10 23:23:01 +0000307static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000308convertenviron(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000309{
Barry Warsaw53699e91996-12-10 23:23:01 +0000310 PyObject *d;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000311 char **e;
Barry Warsaw53699e91996-12-10 23:23:01 +0000312 d = PyDict_New();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000313 if (d == NULL)
314 return NULL;
Jack Jansenea0c3822002-08-01 21:57:49 +0000315#ifdef WITH_NEXT_FRAMEWORK
316 if (environ == NULL)
317 environ = *_NSGetEnviron();
318#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000319 if (environ == NULL)
320 return d;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000321 /* This part ignores errors */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000322 for (e = environ; *e != NULL; e++) {
Guido van Rossum6a619f41999-08-03 19:41:10 +0000323 PyObject *k;
Barry Warsaw53699e91996-12-10 23:23:01 +0000324 PyObject *v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000325 char *p = strchr(*e, '=');
326 if (p == NULL)
327 continue;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000328 k = PyString_FromStringAndSize(*e, (int)(p-*e));
329 if (k == NULL) {
330 PyErr_Clear();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000331 continue;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000332 }
333 v = PyString_FromString(p+1);
334 if (v == NULL) {
335 PyErr_Clear();
336 Py_DECREF(k);
337 continue;
338 }
339 if (PyDict_GetItem(d, k) == NULL) {
340 if (PyDict_SetItem(d, k, v) != 0)
341 PyErr_Clear();
342 }
343 Py_DECREF(k);
Barry Warsaw53699e91996-12-10 23:23:01 +0000344 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000345 }
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +0000346#if defined(PYOS_OS2)
Guido van Rossumd48f2521997-12-05 22:19:34 +0000347 {
348 APIRET rc;
349 char buffer[1024]; /* OS/2 Provides a Documented Max of 1024 Chars */
350
351 rc = DosQueryExtLIBPATH(buffer, BEGIN_LIBPATH);
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000352 if (rc == NO_ERROR) { /* (not a type, envname is NOT 'BEGIN_LIBPATH') */
Guido van Rossumd48f2521997-12-05 22:19:34 +0000353 PyObject *v = PyString_FromString(buffer);
354 PyDict_SetItemString(d, "BEGINLIBPATH", v);
355 Py_DECREF(v);
356 }
357 rc = DosQueryExtLIBPATH(buffer, END_LIBPATH);
358 if (rc == NO_ERROR) { /* (not a typo, envname is NOT 'END_LIBPATH') */
359 PyObject *v = PyString_FromString(buffer);
360 PyDict_SetItemString(d, "ENDLIBPATH", v);
361 Py_DECREF(v);
362 }
363 }
364#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000365 return d;
366}
367
368
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000369/* Set a POSIX-specific error from errno, and return NULL */
370
Barry Warsawd58d7641998-07-23 16:14:40 +0000371static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000372posix_error(void)
Guido van Rossumad0ee831995-03-01 10:34:45 +0000373{
Barry Warsawca74da41999-02-09 19:31:45 +0000374 return PyErr_SetFromErrno(PyExc_OSError);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000375}
Barry Warsawd58d7641998-07-23 16:14:40 +0000376static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000377posix_error_with_filename(char* name)
Barry Warsawd58d7641998-07-23 16:14:40 +0000378{
Barry Warsawca74da41999-02-09 19:31:45 +0000379 return PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
Barry Warsawd58d7641998-07-23 16:14:40 +0000380}
381
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000382#ifdef Py_WIN_WIDE_FILENAMES
383static PyObject *
384posix_error_with_unicode_filename(Py_UNICODE* name)
385{
386 return PyErr_SetFromErrnoWithUnicodeFilename(PyExc_OSError, name);
387}
388#endif /* Py_WIN_WIDE_FILENAMES */
389
390
Mark Hammondef8b6542001-05-13 08:04:26 +0000391static PyObject *
392posix_error_with_allocated_filename(char* name)
393{
394 PyObject *rc = PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
395 PyMem_Free(name);
396 return rc;
397}
398
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000399#ifdef MS_WINDOWS
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000400static PyObject *
401win32_error(char* function, char* filename)
402{
Mark Hammond33a6da92000-08-15 00:46:38 +0000403 /* XXX We should pass the function name along in the future.
404 (_winreg.c also wants to pass the function name.)
Tim Peters5aa91602002-01-30 05:46:57 +0000405 This would however require an additional param to the
Mark Hammond33a6da92000-08-15 00:46:38 +0000406 Windows error object, which is non-trivial.
407 */
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000408 errno = GetLastError();
409 if (filename)
Mark Hammond33a6da92000-08-15 00:46:38 +0000410 return PyErr_SetFromWindowsErrWithFilename(errno, filename);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000411 else
Mark Hammond33a6da92000-08-15 00:46:38 +0000412 return PyErr_SetFromWindowsErr(errno);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000413}
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000414
415#ifdef Py_WIN_WIDE_FILENAMES
416static PyObject *
417win32_error_unicode(char* function, Py_UNICODE* filename)
418{
419 /* XXX - see win32_error for comments on 'function' */
420 errno = GetLastError();
421 if (filename)
422 return PyErr_SetFromWindowsErrWithUnicodeFilename(errno, filename);
423 else
424 return PyErr_SetFromWindowsErr(errno);
425}
426
427static PyObject *_PyUnicode_FromFileSystemEncodedObject(register PyObject *obj)
428{
429 /* XXX Perhaps we should make this API an alias of
430 PyObject_Unicode() instead ?! */
431 if (PyUnicode_CheckExact(obj)) {
432 Py_INCREF(obj);
433 return obj;
434 }
435 if (PyUnicode_Check(obj)) {
436 /* For a Unicode subtype that's not a Unicode object,
437 return a true Unicode object with the same data. */
438 return PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(obj),
439 PyUnicode_GET_SIZE(obj));
440 }
Tim Peters11b23062003-04-23 02:39:17 +0000441 return PyUnicode_FromEncodedObject(obj,
442 Py_FileSystemDefaultEncoding,
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000443 "strict");
444}
445
446#endif /* Py_WIN_WIDE_FILENAMES */
447
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000448#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000449
Guido van Rossumd48f2521997-12-05 22:19:34 +0000450#if defined(PYOS_OS2)
451/**********************************************************************
452 * Helper Function to Trim and Format OS/2 Messages
453 **********************************************************************/
454 static void
455os2_formatmsg(char *msgbuf, int msglen, char *reason)
456{
457 msgbuf[msglen] = '\0'; /* OS/2 Doesn't Guarantee a Terminator */
458
459 if (strlen(msgbuf) > 0) { /* If Non-Empty Msg, Trim CRLF */
460 char *lastc = &msgbuf[ strlen(msgbuf)-1 ];
461
462 while (lastc > msgbuf && isspace(*lastc))
463 *lastc-- = '\0'; /* Trim Trailing Whitespace (CRLF) */
464 }
465
466 /* Add Optional Reason Text */
467 if (reason) {
468 strcat(msgbuf, " : ");
469 strcat(msgbuf, reason);
470 }
471}
472
473/**********************************************************************
474 * Decode an OS/2 Operating System Error Code
475 *
476 * A convenience function to lookup an OS/2 error code and return a
477 * text message we can use to raise a Python exception.
478 *
479 * Notes:
480 * The messages for errors returned from the OS/2 kernel reside in
481 * the file OSO001.MSG in the \OS2 directory hierarchy.
482 *
483 **********************************************************************/
484 static char *
485os2_strerror(char *msgbuf, int msgbuflen, int errorcode, char *reason)
486{
487 APIRET rc;
488 ULONG msglen;
489
490 /* Retrieve Kernel-Related Error Message from OSO001.MSG File */
491 Py_BEGIN_ALLOW_THREADS
492 rc = DosGetMessage(NULL, 0, msgbuf, msgbuflen,
493 errorcode, "oso001.msg", &msglen);
494 Py_END_ALLOW_THREADS
495
496 if (rc == NO_ERROR)
497 os2_formatmsg(msgbuf, msglen, reason);
498 else
Tim Peters1ceb5fb2001-11-28 20:32:57 +0000499 PyOS_snprintf(msgbuf, msgbuflen,
Tim Peters885d4572001-11-28 20:27:42 +0000500 "unknown OS error #%d", errorcode);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000501
502 return msgbuf;
503}
504
505/* Set an OS/2-specific error and return NULL. OS/2 kernel
506 errors are not in a global variable e.g. 'errno' nor are
507 they congruent with posix error numbers. */
508
509static PyObject * os2_error(int code)
510{
511 char text[1024];
512 PyObject *v;
513
514 os2_strerror(text, sizeof(text), code, "");
515
516 v = Py_BuildValue("(is)", code, text);
517 if (v != NULL) {
Barry Warsawca74da41999-02-09 19:31:45 +0000518 PyErr_SetObject(PyExc_OSError, v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000519 Py_DECREF(v);
520 }
521 return NULL; /* Signal to Python that an Exception is Pending */
522}
523
524#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000525
526/* POSIX generic methods */
527
Barry Warsaw53699e91996-12-10 23:23:01 +0000528static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +0000529posix_fildes(PyObject *fdobj, int (*func)(int))
530{
531 int fd;
532 int res;
533 fd = PyObject_AsFileDescriptor(fdobj);
534 if (fd < 0)
535 return NULL;
536 Py_BEGIN_ALLOW_THREADS
537 res = (*func)(fd);
538 Py_END_ALLOW_THREADS
539 if (res < 0)
540 return posix_error();
541 Py_INCREF(Py_None);
542 return Py_None;
543}
Guido van Rossum21142a01999-01-08 21:05:37 +0000544
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000545#ifdef Py_WIN_WIDE_FILENAMES
Tim Peters11b23062003-04-23 02:39:17 +0000546static int
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000547unicode_file_names(void)
548{
549 static int canusewide = -1;
550 if (canusewide == -1) {
Tim Peters11b23062003-04-23 02:39:17 +0000551 /* As per doc for ::GetVersion(), this is the correct test for
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000552 the Windows NT family. */
553 canusewide = (GetVersion() < 0x80000000) ? 1 : 0;
554 }
555 return canusewide;
556}
557#endif
Tim Peters11b23062003-04-23 02:39:17 +0000558
Guido van Rossum21142a01999-01-08 21:05:37 +0000559static PyObject *
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000560posix_1str(PyObject *args, char *format, int (*func)(const char*),
561 char *wformat, int (*wfunc)(const Py_UNICODE*))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000562{
Mark Hammondef8b6542001-05-13 08:04:26 +0000563 char *path1 = NULL;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000564 int res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000565#ifdef Py_WIN_WIDE_FILENAMES
566 if (unicode_file_names()) {
567 PyUnicodeObject *po;
568 if (PyArg_ParseTuple(args, wformat, &po)) {
569 Py_BEGIN_ALLOW_THREADS
570 /* PyUnicode_AS_UNICODE OK without thread
571 lock as it is a simple dereference. */
572 res = (*wfunc)(PyUnicode_AS_UNICODE(po));
573 Py_END_ALLOW_THREADS
574 if (res < 0)
Mark Hammondd3890362002-10-03 07:24:48 +0000575 return posix_error_with_unicode_filename(PyUnicode_AS_UNICODE(po));
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000576 Py_INCREF(Py_None);
577 return Py_None;
578 }
579 /* Drop the argument parsing error as narrow
580 strings are also valid. */
581 PyErr_Clear();
582 }
583#else
584 /* Platforms that don't support Unicode filenames
585 shouldn't be passing these extra params */
586 assert(wformat==NULL && wfunc == NULL);
587#endif
588
Tim Peters5aa91602002-01-30 05:46:57 +0000589 if (!PyArg_ParseTuple(args, format,
Mark Hammondef8b6542001-05-13 08:04:26 +0000590 Py_FileSystemDefaultEncoding, &path1))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000591 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +0000592 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000593 res = (*func)(path1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000594 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000595 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +0000596 return posix_error_with_allocated_filename(path1);
597 PyMem_Free(path1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000598 Py_INCREF(Py_None);
599 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000600}
601
Barry Warsaw53699e91996-12-10 23:23:01 +0000602static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +0000603posix_2str(PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000604 char *format,
605 int (*func)(const char *, const char *),
606 char *wformat,
607 int (*wfunc)(const Py_UNICODE *, const Py_UNICODE *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000608{
Mark Hammondef8b6542001-05-13 08:04:26 +0000609 char *path1 = NULL, *path2 = NULL;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000610 int res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000611#ifdef Py_WIN_WIDE_FILENAMES
612 if (unicode_file_names()) {
613 PyObject *po1;
614 PyObject *po2;
615 if (PyArg_ParseTuple(args, wformat, &po1, &po2)) {
616 if (PyUnicode_Check(po1) || PyUnicode_Check(po2)) {
617 PyObject *wpath1;
618 PyObject *wpath2;
619 wpath1 = _PyUnicode_FromFileSystemEncodedObject(po1);
620 wpath2 = _PyUnicode_FromFileSystemEncodedObject(po2);
621 if (!wpath1 || !wpath2) {
622 Py_XDECREF(wpath1);
623 Py_XDECREF(wpath2);
624 return NULL;
625 }
626 Py_BEGIN_ALLOW_THREADS
627 /* PyUnicode_AS_UNICODE OK without thread
628 lock as it is a simple dereference. */
629 res = (*wfunc)(PyUnicode_AS_UNICODE(wpath1),
630 PyUnicode_AS_UNICODE(wpath2));
631 Py_END_ALLOW_THREADS
632 Py_XDECREF(wpath1);
633 Py_XDECREF(wpath2);
634 if (res != 0)
635 return posix_error();
636 Py_INCREF(Py_None);
637 return Py_None;
638 }
639 /* Else flow through as neither is Unicode. */
640 }
641 /* Drop the argument parsing error as narrow
642 strings are also valid. */
643 PyErr_Clear();
644 }
645#else
646 /* Platforms that don't support Unicode filenames
647 shouldn't be passing these extra params */
648 assert(wformat==NULL && wfunc == NULL);
649#endif
650
Mark Hammondef8b6542001-05-13 08:04:26 +0000651 if (!PyArg_ParseTuple(args, format,
Tim Peters5aa91602002-01-30 05:46:57 +0000652 Py_FileSystemDefaultEncoding, &path1,
Mark Hammondef8b6542001-05-13 08:04:26 +0000653 Py_FileSystemDefaultEncoding, &path2))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000654 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +0000655 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000656 res = (*func)(path1, path2);
Barry Warsaw53699e91996-12-10 23:23:01 +0000657 Py_END_ALLOW_THREADS
Mark Hammondef8b6542001-05-13 08:04:26 +0000658 PyMem_Free(path1);
659 PyMem_Free(path2);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000660 if (res != 0)
Barry Warsawd58d7641998-07-23 16:14:40 +0000661 /* XXX how to report both path1 and path2??? */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000662 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +0000663 Py_INCREF(Py_None);
664 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000665}
666
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000667PyDoc_STRVAR(stat_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000668"stat_result: Result from stat or lstat.\n\n\
669This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +0000670 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000671or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
672\n\
Guido van Rossuma4dc73e2001-10-18 20:53:15 +0000673Posix/windows: If your platform supports st_blksize, st_blocks, or st_rdev,\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000674they are available as attributes only.\n\
675\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000676See os.stat for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000677
678static PyStructSequence_Field stat_result_fields[] = {
679 {"st_mode", "protection bits"},
680 {"st_ino", "inode"},
681 {"st_dev", "device"},
682 {"st_nlink", "number of hard links"},
683 {"st_uid", "user ID of owner"},
684 {"st_gid", "group ID of owner"},
685 {"st_size", "total size, in bytes"},
Martin v. Löwisf607bda2002-10-16 18:27:39 +0000686 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
687 {NULL, "integer time of last access"},
688 {NULL, "integer time of last modification"},
689 {NULL, "integer time of last change"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000690 {"st_atime", "time of last access"},
691 {"st_mtime", "time of last modification"},
692 {"st_ctime", "time of last change"},
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000693#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000694 {"st_blksize", "blocksize for filesystem I/O"},
695#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000696#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000697 {"st_blocks", "number of blocks allocated"},
698#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000699#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000700 {"st_rdev", "device type (if inode device)"},
701#endif
702 {0}
703};
704
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000705#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Martin v. Löwisf607bda2002-10-16 18:27:39 +0000706#define ST_BLKSIZE_IDX 13
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000707#else
Martin v. Löwisf607bda2002-10-16 18:27:39 +0000708#define ST_BLKSIZE_IDX 12
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000709#endif
710
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000711#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000712#define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
713#else
714#define ST_BLOCKS_IDX ST_BLKSIZE_IDX
715#endif
716
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000717#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000718#define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
719#else
720#define ST_RDEV_IDX ST_BLOCKS_IDX
721#endif
722
723static PyStructSequence_Desc stat_result_desc = {
724 "stat_result", /* name */
725 stat_result__doc__, /* doc */
726 stat_result_fields,
727 10
728};
729
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000730PyDoc_STRVAR(statvfs_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000731"statvfs_result: Result from statvfs or fstatvfs.\n\n\
732This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +0000733 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
Guido van Rossuma4dc73e2001-10-18 20:53:15 +0000734or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000735\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000736See os.statvfs for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000737
738static PyStructSequence_Field statvfs_result_fields[] = {
739 {"f_bsize", },
740 {"f_frsize", },
741 {"f_blocks", },
742 {"f_bfree", },
743 {"f_bavail", },
744 {"f_files", },
745 {"f_ffree", },
746 {"f_favail", },
747 {"f_flag", },
748 {"f_namemax",},
749 {0}
750};
751
752static PyStructSequence_Desc statvfs_result_desc = {
753 "statvfs_result", /* name */
754 statvfs_result__doc__, /* doc */
755 statvfs_result_fields,
756 10
757};
758
759static PyTypeObject StatResultType;
760static PyTypeObject StatVFSResultType;
Martin v. Löwisf607bda2002-10-16 18:27:39 +0000761static newfunc structseq_new;
762
763static PyObject *
764statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
765{
766 PyStructSequence *result;
767 int i;
768
769 result = (PyStructSequence*)structseq_new(type, args, kwds);
770 if (!result)
771 return NULL;
772 /* If we have been initialized from a tuple,
773 st_?time might be set to None. Initialize it
774 from the int slots. */
775 for (i = 7; i <= 9; i++) {
776 if (result->ob_item[i+3] == Py_None) {
777 Py_DECREF(Py_None);
778 Py_INCREF(result->ob_item[i]);
779 result->ob_item[i+3] = result->ob_item[i];
780 }
781 }
782 return (PyObject*)result;
783}
784
785
786
787/* If true, st_?time is float. */
788static int _stat_float_times = 0;
789
790PyDoc_STRVAR(stat_float_times__doc__,
791"stat_float_times([newval]) -> oldval\n\n\
792Determine whether os.[lf]stat represents time stamps as float objects.\n\
793If newval is True, future calls to stat() return floats, if it is False,\n\
794future calls return ints. \n\
795If newval is omitted, return the current setting.\n");
796
797static PyObject*
798stat_float_times(PyObject* self, PyObject *args)
799{
800 int newval = -1;
801 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
802 return NULL;
803 if (newval == -1)
804 /* Return old value */
805 return PyBool_FromLong(_stat_float_times);
806 _stat_float_times = newval;
807 Py_INCREF(Py_None);
808 return Py_None;
809}
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000810
Martin v. Löwis94717ed2002-09-09 14:24:16 +0000811static void
812fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
813{
Martin v. Löwisf607bda2002-10-16 18:27:39 +0000814 PyObject *fval,*ival;
815#if SIZEOF_TIME_T > SIZEOF_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000816 ival = PyLong_FromLongLong((PY_LONG_LONG)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +0000817#else
818 ival = PyInt_FromLong((long)sec);
819#endif
820 if (_stat_float_times) {
821 fval = PyFloat_FromDouble(sec + 1e-9*nsec);
822 } else {
823 fval = ival;
824 Py_INCREF(fval);
825 }
826 PyStructSequence_SET_ITEM(v, index, ival);
827 PyStructSequence_SET_ITEM(v, index+3, fval);
Martin v. Löwis94717ed2002-09-09 14:24:16 +0000828}
829
Tim Peters5aa91602002-01-30 05:46:57 +0000830/* pack a system stat C structure into the Python stat tuple
Fred Drake699f3522000-06-29 21:12:41 +0000831 (used by posix_stat() and posix_fstat()) */
832static PyObject*
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000833_pystat_fromstructstat(STRUCT_STAT st)
Fred Drake699f3522000-06-29 21:12:41 +0000834{
Martin v. Löwis94717ed2002-09-09 14:24:16 +0000835 unsigned long ansec, mnsec, cnsec;
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000836 PyObject *v = PyStructSequence_New(&StatResultType);
Fred Drake699f3522000-06-29 21:12:41 +0000837 if (v == NULL)
838 return NULL;
839
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000840 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long)st.st_mode));
Fred Drake699f3522000-06-29 21:12:41 +0000841#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +0000842 PyStructSequence_SET_ITEM(v, 1,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000843 PyLong_FromLongLong((PY_LONG_LONG)st.st_ino));
Fred Drake699f3522000-06-29 21:12:41 +0000844#else
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000845 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long)st.st_ino));
Fred Drake699f3522000-06-29 21:12:41 +0000846#endif
847#if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
Tim Peters5aa91602002-01-30 05:46:57 +0000848 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000849 PyLong_FromLongLong((PY_LONG_LONG)st.st_dev));
Fred Drake699f3522000-06-29 21:12:41 +0000850#else
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000851 PyStructSequence_SET_ITEM(v, 2, PyInt_FromLong((long)st.st_dev));
Fred Drake699f3522000-06-29 21:12:41 +0000852#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000853 PyStructSequence_SET_ITEM(v, 3, PyInt_FromLong((long)st.st_nlink));
854 PyStructSequence_SET_ITEM(v, 4, PyInt_FromLong((long)st.st_uid));
855 PyStructSequence_SET_ITEM(v, 5, PyInt_FromLong((long)st.st_gid));
Fred Drake699f3522000-06-29 21:12:41 +0000856#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +0000857 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000858 PyLong_FromLongLong((PY_LONG_LONG)st.st_size));
Fred Drake699f3522000-06-29 21:12:41 +0000859#else
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000860 PyStructSequence_SET_ITEM(v, 6, PyInt_FromLong(st.st_size));
Fred Drake699f3522000-06-29 21:12:41 +0000861#endif
Martin v. Löwis94717ed2002-09-09 14:24:16 +0000862
863#ifdef HAVE_STAT_TV_NSEC
864 ansec = st.st_atim.tv_nsec;
865 mnsec = st.st_mtim.tv_nsec;
866 cnsec = st.st_ctim.tv_nsec;
Fred Drake699f3522000-06-29 21:12:41 +0000867#else
Martin v. Löwis94717ed2002-09-09 14:24:16 +0000868 ansec = mnsec = cnsec = 0;
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000869#endif
Martin v. Löwis94717ed2002-09-09 14:24:16 +0000870 fill_time(v, 7, st.st_atime, ansec);
871 fill_time(v, 8, st.st_mtime, mnsec);
872 fill_time(v, 9, st.st_ctime, cnsec);
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000873
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000874#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Tim Peters5aa91602002-01-30 05:46:57 +0000875 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000876 PyInt_FromLong((long)st.st_blksize));
877#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000878#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Tim Peters5aa91602002-01-30 05:46:57 +0000879 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000880 PyInt_FromLong((long)st.st_blocks));
881#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000882#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000883 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
884 PyInt_FromLong((long)st.st_rdev));
Fred Drake699f3522000-06-29 21:12:41 +0000885#endif
886
887 if (PyErr_Occurred()) {
888 Py_DECREF(v);
889 return NULL;
890 }
891
892 return v;
893}
894
Barry Warsaw53699e91996-12-10 23:23:01 +0000895static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +0000896posix_do_stat(PyObject *self, PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000897 char *format,
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000898#ifdef __VMS
899 int (*statfunc)(const char *, STRUCT_STAT *, ...),
900#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000901 int (*statfunc)(const char *, STRUCT_STAT *),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000902#endif
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000903 char *wformat,
904 int (*wstatfunc)(const Py_UNICODE *, STRUCT_STAT *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000905{
Fred Drake699f3522000-06-29 21:12:41 +0000906 STRUCT_STAT st;
Tim Peters500bd032001-12-19 19:05:01 +0000907 char *path = NULL; /* pass this to stat; do not free() it */
908 char *pathfree = NULL; /* this memory must be free'd */
Guido van Rossumff4949e1992-08-05 19:58:53 +0000909 int res;
Guido van Rossumace88ae2000-04-21 18:54:45 +0000910
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000911#ifdef MS_WINDOWS
Tim Peters500bd032001-12-19 19:05:01 +0000912 int pathlen;
913 char pathcopy[MAX_PATH];
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000914#endif /* MS_WINDOWS */
Guido van Rossumace88ae2000-04-21 18:54:45 +0000915
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000916
917#ifdef Py_WIN_WIDE_FILENAMES
918 /* If on wide-character-capable OS see if argument
919 is Unicode and if so use wide API. */
920 if (unicode_file_names()) {
921 PyUnicodeObject *po;
922 if (PyArg_ParseTuple(args, wformat, &po)) {
923 Py_UNICODE wpath[MAX_PATH+1];
924 pathlen = wcslen(PyUnicode_AS_UNICODE(po));
925 /* the library call can blow up if the file name is too long! */
926 if (pathlen > MAX_PATH) {
927 errno = ENAMETOOLONG;
928 return posix_error();
929 }
930 wcscpy(wpath, PyUnicode_AS_UNICODE(po));
931 /* Remove trailing slash or backslash, unless it's the current
932 drive root (/ or \) or a specific drive's root (like c:\ or c:/).
933 */
934 if (pathlen > 0 &&
935 (wpath[pathlen-1]== L'\\' || wpath[pathlen-1] == L'/')) {
936 /* It does end with a slash -- exempt the root drive cases. */
937 /* XXX UNC root drives should also be exempted? */
938 if (pathlen == 1 || (pathlen == 3 && wpath[1] == L':'))
939 /* leave it alone */;
940 else {
941 /* nuke the trailing backslash */
942 wpath[pathlen-1] = L'\0';
943 }
944 }
945 Py_BEGIN_ALLOW_THREADS
946 /* PyUnicode_AS_UNICODE result OK without
947 thread lock as it is a simple dereference. */
948 res = wstatfunc(wpath, &st);
949 Py_END_ALLOW_THREADS
950 if (res != 0)
951 return posix_error_with_unicode_filename(wpath);
952 return _pystat_fromstructstat(st);
953 }
954 /* Drop the argument parsing error as narrow strings
955 are also valid. */
956 PyErr_Clear();
957 }
958#endif
959
Tim Peters5aa91602002-01-30 05:46:57 +0000960 if (!PyArg_ParseTuple(args, format,
Mark Hammondef8b6542001-05-13 08:04:26 +0000961 Py_FileSystemDefaultEncoding, &path))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000962 return NULL;
Tim Peters500bd032001-12-19 19:05:01 +0000963 pathfree = path;
Guido van Rossumace88ae2000-04-21 18:54:45 +0000964
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000965#ifdef MS_WINDOWS
Guido van Rossumace88ae2000-04-21 18:54:45 +0000966 pathlen = strlen(path);
967 /* the library call can blow up if the file name is too long! */
968 if (pathlen > MAX_PATH) {
Tim Peters500bd032001-12-19 19:05:01 +0000969 PyMem_Free(pathfree);
Guido van Rossumace88ae2000-04-21 18:54:45 +0000970 errno = ENAMETOOLONG;
971 return posix_error();
972 }
973
Tim Peters500bd032001-12-19 19:05:01 +0000974 /* Remove trailing slash or backslash, unless it's the current
975 drive root (/ or \) or a specific drive's root (like c:\ or c:/).
976 */
977 if (pathlen > 0 &&
978 (path[pathlen-1]== '\\' || path[pathlen-1] == '/')) {
979 /* It does end with a slash -- exempt the root drive cases. */
980 /* XXX UNC root drives should also be exempted? */
981 if (pathlen == 1 || (pathlen == 3 && path[1] == ':'))
982 /* leave it alone */;
983 else {
984 /* nuke the trailing backslash */
Guido van Rossumace88ae2000-04-21 18:54:45 +0000985 strncpy(pathcopy, path, pathlen);
Tim Peters500bd032001-12-19 19:05:01 +0000986 pathcopy[pathlen-1] = '\0';
Guido van Rossumace88ae2000-04-21 18:54:45 +0000987 path = pathcopy;
988 }
989 }
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000990#endif /* MS_WINDOWS */
Guido van Rossumace88ae2000-04-21 18:54:45 +0000991
Barry Warsaw53699e91996-12-10 23:23:01 +0000992 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000993 res = (*statfunc)(path, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +0000994 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000995 if (res != 0)
Tim Peters500bd032001-12-19 19:05:01 +0000996 return posix_error_with_allocated_filename(pathfree);
Fred Drake699f3522000-06-29 21:12:41 +0000997
Tim Peters500bd032001-12-19 19:05:01 +0000998 PyMem_Free(pathfree);
Fred Drake699f3522000-06-29 21:12:41 +0000999 return _pystat_fromstructstat(st);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001000}
1001
1002
1003/* POSIX methods */
1004
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001005PyDoc_STRVAR(posix_access__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001006"access(path, mode) -> 1 if granted, 0 otherwise\n\n\
Guido van Rossuma0b90752002-06-18 16:22:43 +00001007Use the real uid/gid to test for access to a path. Note that most\n\
1008operations will use the effective uid/gid, therefore this routine can\n\
1009be used in a suid/sgid environment to test if the invoking user has the\n\
1010specified access to the path. The mode argument can be F_OK to test\n\
1011existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001012
1013static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001014posix_access(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001015{
Guido van Rossum015f22a1999-01-06 22:52:38 +00001016 char *path;
1017 int mode;
1018 int res;
1019
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001020#ifdef Py_WIN_WIDE_FILENAMES
1021 if (unicode_file_names()) {
1022 PyUnicodeObject *po;
1023 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
1024 Py_BEGIN_ALLOW_THREADS
1025 /* PyUnicode_AS_UNICODE OK without thread lock as
1026 it is a simple dereference. */
1027 res = _waccess(PyUnicode_AS_UNICODE(po), mode);
1028 Py_END_ALLOW_THREADS
1029 return(PyBool_FromLong(res == 0));
1030 }
1031 /* Drop the argument parsing error as narrow strings
1032 are also valid. */
1033 PyErr_Clear();
1034 }
1035#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001036 if (!PyArg_ParseTuple(args, "si:access", &path, &mode))
Guido van Rossum015f22a1999-01-06 22:52:38 +00001037 return NULL;
1038 Py_BEGIN_ALLOW_THREADS
1039 res = access(path, mode);
1040 Py_END_ALLOW_THREADS
Guido van Rossum674deb22002-09-01 15:06:28 +00001041 return(PyBool_FromLong(res == 0));
Guido van Rossum94f6f721999-01-06 18:42:14 +00001042}
1043
Guido van Rossumd371ff11999-01-25 16:12:23 +00001044#ifndef F_OK
1045#define F_OK 0
1046#endif
1047#ifndef R_OK
1048#define R_OK 4
1049#endif
1050#ifndef W_OK
1051#define W_OK 2
1052#endif
1053#ifndef X_OK
1054#define X_OK 1
1055#endif
1056
1057#ifdef HAVE_TTYNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001058PyDoc_STRVAR(posix_ttyname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001059"ttyname(fd) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001060Return the name of the terminal device connected to 'fd'.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001061
1062static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001063posix_ttyname(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001064{
Guido van Rossum94f6f721999-01-06 18:42:14 +00001065 int id;
1066 char *ret;
1067
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001068 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
Guido van Rossum94f6f721999-01-06 18:42:14 +00001069 return NULL;
1070
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001071#if defined(__VMS)
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001072 /* file descriptor 0 only, the default input device (stdin) */
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001073 if (id == 0) {
1074 ret = ttyname();
1075 }
1076 else {
1077 ret = NULL;
1078 }
1079#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00001080 ret = ttyname(id);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001081#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001082 if (ret == NULL)
1083 return(posix_error());
1084 return(PyString_FromString(ret));
1085}
Guido van Rossumd371ff11999-01-25 16:12:23 +00001086#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001087
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001088#ifdef HAVE_CTERMID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001089PyDoc_STRVAR(posix_ctermid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001090"ctermid() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001091Return the name of the controlling terminal for this process.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001092
1093static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001094posix_ctermid(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001095{
1096 char *ret;
1097 char buffer[L_ctermid];
1098
Greg Wardb48bc172000-03-01 21:51:56 +00001099#ifdef USE_CTERMID_R
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001100 ret = ctermid_r(buffer);
1101#else
1102 ret = ctermid(buffer);
1103#endif
1104 if (ret == NULL)
1105 return(posix_error());
1106 return(PyString_FromString(buffer));
1107}
1108#endif
1109
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001110PyDoc_STRVAR(posix_chdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001111"chdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001112Change the current working directory to the specified path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001113
Barry Warsaw53699e91996-12-10 23:23:01 +00001114static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001115posix_chdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001116{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001117#ifdef MS_WINDOWS
1118 return posix_1str(args, "et:chdir", chdir, "U:chdir", _wchdir);
1119#elif defined(PYOS_OS2) && defined(PYCC_GCC)
1120 return posix_1str(args, "et:chdir", _chdir2, NULL, NULL);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00001121#elif defined(__VMS)
Tim Peters11b23062003-04-23 02:39:17 +00001122 return posix_1str(args, "et:chdir", (int (*)(const char *))chdir,
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001123 NULL, NULL);
1124#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001125 return posix_1str(args, "et:chdir", chdir, NULL, NULL);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001126#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001127}
1128
Fred Drake4d1e64b2002-04-15 19:40:07 +00001129#ifdef HAVE_FCHDIR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001130PyDoc_STRVAR(posix_fchdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001131"fchdir(fildes)\n\n\
Fred Drake4d1e64b2002-04-15 19:40:07 +00001132Change to the directory of the given file descriptor. fildes must be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001133opened on a directory, not a file.");
Fred Drake4d1e64b2002-04-15 19:40:07 +00001134
1135static PyObject *
1136posix_fchdir(PyObject *self, PyObject *fdobj)
1137{
1138 return posix_fildes(fdobj, fchdir);
1139}
1140#endif /* HAVE_FCHDIR */
1141
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001142
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001143PyDoc_STRVAR(posix_chmod__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001144"chmod(path, mode)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001145Change the access permissions of a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001146
Barry Warsaw53699e91996-12-10 23:23:01 +00001147static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001148posix_chmod(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001149{
Mark Hammondef8b6542001-05-13 08:04:26 +00001150 char *path = NULL;
Guido van Rossumffd15f52000-03-31 00:47:28 +00001151 int i;
1152 int res;
Mark Hammond817c9292003-12-03 01:22:38 +00001153#ifdef Py_WIN_WIDE_FILENAMES
1154 if (unicode_file_names()) {
1155 PyUnicodeObject *po;
1156 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
1157 Py_BEGIN_ALLOW_THREADS
1158 res = _wchmod(PyUnicode_AS_UNICODE(po), i);
1159 Py_END_ALLOW_THREADS
1160 if (res < 0)
1161 return posix_error_with_unicode_filename(
1162 PyUnicode_AS_UNICODE(po));
1163 Py_INCREF(Py_None);
1164 return Py_None;
1165 }
1166 /* Drop the argument parsing error as narrow strings
1167 are also valid. */
1168 PyErr_Clear();
1169 }
1170#endif /* Py_WIN_WIDE_FILENAMES */
1171 if (!PyArg_ParseTuple(args, "eti:chmod", Py_FileSystemDefaultEncoding,
Mark Hammondef8b6542001-05-13 08:04:26 +00001172 &path, &i))
Guido van Rossumffd15f52000-03-31 00:47:28 +00001173 return NULL;
1174 Py_BEGIN_ALLOW_THREADS
Guido van Rossumef40e772000-03-31 01:26:23 +00001175 res = chmod(path, i);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001176 Py_END_ALLOW_THREADS
1177 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00001178 return posix_error_with_allocated_filename(path);
1179 PyMem_Free(path);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001180 Py_INCREF(Py_None);
1181 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001182}
1183
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001184
Martin v. Löwis244edc82001-10-04 22:44:26 +00001185#ifdef HAVE_CHROOT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001186PyDoc_STRVAR(posix_chroot__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001187"chroot(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001188Change root directory to path.");
Martin v. Löwis244edc82001-10-04 22:44:26 +00001189
1190static PyObject *
1191posix_chroot(PyObject *self, PyObject *args)
1192{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001193 return posix_1str(args, "et:chroot", chroot, NULL, NULL);
Martin v. Löwis244edc82001-10-04 22:44:26 +00001194}
1195#endif
1196
Guido van Rossum21142a01999-01-08 21:05:37 +00001197#ifdef HAVE_FSYNC
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001198PyDoc_STRVAR(posix_fsync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001199"fsync(fildes)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001200force write of file with filedescriptor to disk.");
Guido van Rossum21142a01999-01-08 21:05:37 +00001201
1202static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00001203posix_fsync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00001204{
Fred Drake4d1e64b2002-04-15 19:40:07 +00001205 return posix_fildes(fdobj, fsync);
Guido van Rossum21142a01999-01-08 21:05:37 +00001206}
1207#endif /* HAVE_FSYNC */
1208
1209#ifdef HAVE_FDATASYNC
Guido van Rossumecc23b02000-09-22 16:01:05 +00001210
Guido van Rossum7f58e2e2000-09-22 17:26:14 +00001211#ifdef __hpux
Guido van Rossumecc23b02000-09-22 16:01:05 +00001212extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
1213#endif
1214
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001215PyDoc_STRVAR(posix_fdatasync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001216"fdatasync(fildes)\n\n\
Guido van Rossum21142a01999-01-08 21:05:37 +00001217force write of file with filedescriptor to disk.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001218 does not force update of metadata.");
Guido van Rossum21142a01999-01-08 21:05:37 +00001219
1220static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00001221posix_fdatasync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00001222{
Fred Drake4d1e64b2002-04-15 19:40:07 +00001223 return posix_fildes(fdobj, fdatasync);
Guido van Rossum21142a01999-01-08 21:05:37 +00001224}
1225#endif /* HAVE_FDATASYNC */
1226
1227
Fredrik Lundh10723342000-07-10 16:38:09 +00001228#ifdef HAVE_CHOWN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001229PyDoc_STRVAR(posix_chown__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001230"chown(path, uid, gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001231Change the owner and group id of path to the numeric uid and gid.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001232
Barry Warsaw53699e91996-12-10 23:23:01 +00001233static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001234posix_chown(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001235{
Mark Hammondef8b6542001-05-13 08:04:26 +00001236 char *path = NULL;
Fredrik Lundh44328e62000-07-10 15:59:30 +00001237 int uid, gid;
1238 int res;
Tim Peters5aa91602002-01-30 05:46:57 +00001239 if (!PyArg_ParseTuple(args, "etii:chown",
1240 Py_FileSystemDefaultEncoding, &path,
Mark Hammondef8b6542001-05-13 08:04:26 +00001241 &uid, &gid))
Fredrik Lundh44328e62000-07-10 15:59:30 +00001242 return NULL;
1243 Py_BEGIN_ALLOW_THREADS
1244 res = chown(path, (uid_t) uid, (gid_t) gid);
1245 Py_END_ALLOW_THREADS
1246 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00001247 return posix_error_with_allocated_filename(path);
1248 PyMem_Free(path);
Fredrik Lundh44328e62000-07-10 15:59:30 +00001249 Py_INCREF(Py_None);
1250 return Py_None;
Guido van Rossumb6775db1994-08-01 11:34:53 +00001251}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001252#endif /* HAVE_CHOWN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00001253
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00001254#ifdef HAVE_LCHOWN
1255PyDoc_STRVAR(posix_lchown__doc__,
1256"lchown(path, uid, gid)\n\n\
1257Change the owner and group id of path to the numeric uid and gid.\n\
1258This function will not follow symbolic links.");
1259
1260static PyObject *
1261posix_lchown(PyObject *self, PyObject *args)
1262{
1263 char *path = NULL;
1264 int uid, gid;
1265 int res;
1266 if (!PyArg_ParseTuple(args, "etii:lchown",
1267 Py_FileSystemDefaultEncoding, &path,
1268 &uid, &gid))
1269 return NULL;
1270 Py_BEGIN_ALLOW_THREADS
1271 res = lchown(path, (uid_t) uid, (gid_t) gid);
1272 Py_END_ALLOW_THREADS
Tim Peters11b23062003-04-23 02:39:17 +00001273 if (res < 0)
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00001274 return posix_error_with_allocated_filename(path);
1275 PyMem_Free(path);
1276 Py_INCREF(Py_None);
1277 return Py_None;
1278}
1279#endif /* HAVE_LCHOWN */
1280
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001281
Guido van Rossum36bc6801995-06-14 22:54:23 +00001282#ifdef HAVE_GETCWD
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001283PyDoc_STRVAR(posix_getcwd__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001284"getcwd() -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001285Return a string representing the current working directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001286
Barry Warsaw53699e91996-12-10 23:23:01 +00001287static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001288posix_getcwd(PyObject *self, PyObject *noargs)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001289{
1290 char buf[1026];
Guido van Rossumff4949e1992-08-05 19:58:53 +00001291 char *res;
Neal Norwitze241ce82003-02-17 18:17:05 +00001292
Barry Warsaw53699e91996-12-10 23:23:01 +00001293 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001294#if defined(PYOS_OS2) && defined(PYCC_GCC)
1295 res = _getcwd2(buf, sizeof buf);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001296#else
Guido van Rossumff4949e1992-08-05 19:58:53 +00001297 res = getcwd(buf, sizeof buf);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001298#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00001299 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001300 if (res == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001301 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00001302 return PyString_FromString(buf);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001303}
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001304
Walter Dörwald3b918c32002-11-21 20:18:46 +00001305#ifdef Py_USING_UNICODE
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001306PyDoc_STRVAR(posix_getcwdu__doc__,
1307"getcwdu() -> path\n\n\
1308Return a unicode string representing the current working directory.");
1309
1310static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001311posix_getcwdu(PyObject *self, PyObject *noargs)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001312{
1313 char buf[1026];
1314 char *res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001315
1316#ifdef Py_WIN_WIDE_FILENAMES
1317 if (unicode_file_names()) {
1318 wchar_t *wres;
1319 wchar_t wbuf[1026];
1320 Py_BEGIN_ALLOW_THREADS
1321 wres = _wgetcwd(wbuf, sizeof wbuf/ sizeof wbuf[0]);
1322 Py_END_ALLOW_THREADS
1323 if (wres == NULL)
1324 return posix_error();
1325 return PyUnicode_FromWideChar(wbuf, wcslen(wbuf));
1326 }
1327#endif
1328
1329 Py_BEGIN_ALLOW_THREADS
1330#if defined(PYOS_OS2) && defined(PYCC_GCC)
1331 res = _getcwd2(buf, sizeof buf);
1332#else
1333 res = getcwd(buf, sizeof buf);
1334#endif
1335 Py_END_ALLOW_THREADS
1336 if (res == NULL)
1337 return posix_error();
1338 return PyUnicode_Decode(buf, strlen(buf), Py_FileSystemDefaultEncoding,"strict");
1339}
Guido van Rossum36bc6801995-06-14 22:54:23 +00001340#endif
Walter Dörwald3b918c32002-11-21 20:18:46 +00001341#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001342
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001343
Guido van Rossumb6775db1994-08-01 11:34:53 +00001344#ifdef HAVE_LINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001345PyDoc_STRVAR(posix_link__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001346"link(src, dst)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001347Create a hard link to a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001348
Barry Warsaw53699e91996-12-10 23:23:01 +00001349static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001350posix_link(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001351{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001352 return posix_2str(args, "etet:link", link, NULL, NULL);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001353}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001354#endif /* HAVE_LINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001355
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001356
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001357PyDoc_STRVAR(posix_listdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001358"listdir(path) -> list_of_strings\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001359Return a list containing the names of the entries in the directory.\n\
1360\n\
1361 path: path of directory to list\n\
1362\n\
1363The list is in arbitrary order. It does not include the special\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001364entries '.' and '..' even if they are present in the directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001365
Barry Warsaw53699e91996-12-10 23:23:01 +00001366static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001367posix_listdir(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001368{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001369 /* XXX Should redo this putting the (now four) versions of opendir
Guido van Rossum6d8841c1997-08-14 19:57:39 +00001370 in separate files instead of having them all here... */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001371#if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001372
Barry Warsaw53699e91996-12-10 23:23:01 +00001373 PyObject *d, *v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00001374 HANDLE hFindFile;
1375 WIN32_FIND_DATA FileData;
Mark Hammondef8b6542001-05-13 08:04:26 +00001376 /* MAX_PATH characters could mean a bigger encoded string */
1377 char namebuf[MAX_PATH*2+5];
1378 char *bufptr = namebuf;
1379 int len = sizeof(namebuf)/sizeof(namebuf[0]);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001380
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001381#ifdef Py_WIN_WIDE_FILENAMES
1382 /* If on wide-character-capable OS see if argument
1383 is Unicode and if so use wide API. */
1384 if (unicode_file_names()) {
1385 PyUnicodeObject *po;
1386 if (PyArg_ParseTuple(args, "U:listdir", &po)) {
1387 WIN32_FIND_DATAW wFileData;
1388 Py_UNICODE wnamebuf[MAX_PATH*2+5];
1389 Py_UNICODE wch;
1390 wcsncpy(wnamebuf, PyUnicode_AS_UNICODE(po), MAX_PATH);
1391 wnamebuf[MAX_PATH] = L'\0';
1392 len = wcslen(wnamebuf);
1393 wch = (len > 0) ? wnamebuf[len-1] : L'\0';
1394 if (wch != L'/' && wch != L'\\' && wch != L':')
1395 wnamebuf[len++] = L'/';
1396 wcscpy(wnamebuf + len, L"*.*");
1397 if ((d = PyList_New(0)) == NULL)
1398 return NULL;
1399 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
1400 if (hFindFile == INVALID_HANDLE_VALUE) {
1401 errno = GetLastError();
1402 if (errno == ERROR_FILE_NOT_FOUND) {
1403 return d;
1404 }
1405 Py_DECREF(d);
1406 return win32_error_unicode("FindFirstFileW", wnamebuf);
1407 }
1408 do {
1409 if (wFileData.cFileName[0] == L'.' &&
1410 (wFileData.cFileName[1] == L'\0' ||
1411 wFileData.cFileName[1] == L'.' &&
1412 wFileData.cFileName[2] == L'\0'))
1413 continue;
1414 v = PyUnicode_FromUnicode(wFileData.cFileName, wcslen(wFileData.cFileName));
1415 if (v == NULL) {
1416 Py_DECREF(d);
1417 d = NULL;
1418 break;
1419 }
1420 if (PyList_Append(d, v) != 0) {
1421 Py_DECREF(v);
1422 Py_DECREF(d);
1423 d = NULL;
1424 break;
1425 }
1426 Py_DECREF(v);
1427 } while (FindNextFileW(hFindFile, &wFileData) == TRUE);
1428
1429 if (FindClose(hFindFile) == FALSE) {
1430 Py_DECREF(d);
1431 return win32_error_unicode("FindClose", wnamebuf);
1432 }
1433 return d;
1434 }
1435 /* Drop the argument parsing error as narrow strings
1436 are also valid. */
1437 PyErr_Clear();
1438 }
1439#endif
1440
Tim Peters5aa91602002-01-30 05:46:57 +00001441 if (!PyArg_ParseTuple(args, "et#:listdir",
Mark Hammondef8b6542001-05-13 08:04:26 +00001442 Py_FileSystemDefaultEncoding, &bufptr, &len))
Guido van Rossumb6775db1994-08-01 11:34:53 +00001443 return NULL;
Neil Schemenauer94b866a2002-03-22 20:51:58 +00001444 if (len > 0) {
1445 char ch = namebuf[len-1];
1446 if (ch != SEP && ch != ALTSEP && ch != ':')
1447 namebuf[len++] = '/';
1448 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001449 strcpy(namebuf + len, "*.*");
1450
Barry Warsaw53699e91996-12-10 23:23:01 +00001451 if ((d = PyList_New(0)) == NULL)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001452 return NULL;
1453
1454 hFindFile = FindFirstFile(namebuf, &FileData);
1455 if (hFindFile == INVALID_HANDLE_VALUE) {
1456 errno = GetLastError();
Guido van Rossum617bc191998-08-06 03:23:32 +00001457 if (errno == ERROR_FILE_NOT_FOUND)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001458 return d;
1459 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00001460 return win32_error("FindFirstFile", namebuf);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001461 }
1462 do {
Guido van Rossum24f42ac1995-07-18 18:16:52 +00001463 if (FileData.cFileName[0] == '.' &&
1464 (FileData.cFileName[1] == '\0' ||
1465 FileData.cFileName[1] == '.' &&
1466 FileData.cFileName[2] == '\0'))
1467 continue;
Barry Warsaw53699e91996-12-10 23:23:01 +00001468 v = PyString_FromString(FileData.cFileName);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001469 if (v == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00001470 Py_DECREF(d);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001471 d = NULL;
1472 break;
1473 }
Barry Warsaw53699e91996-12-10 23:23:01 +00001474 if (PyList_Append(d, v) != 0) {
1475 Py_DECREF(v);
1476 Py_DECREF(d);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001477 d = NULL;
1478 break;
1479 }
Barry Warsaw53699e91996-12-10 23:23:01 +00001480 Py_DECREF(v);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001481 } while (FindNextFile(hFindFile, &FileData) == TRUE);
1482
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001483 if (FindClose(hFindFile) == FALSE) {
1484 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00001485 return win32_error("FindClose", namebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001486 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001487
1488 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001489
Tim Peters0bb44a42000-09-15 07:44:49 +00001490#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001491
1492#ifndef MAX_PATH
1493#define MAX_PATH CCHMAXPATH
1494#endif
1495 char *name, *pt;
1496 int len;
1497 PyObject *d, *v;
1498 char namebuf[MAX_PATH+5];
1499 HDIR hdir = 1;
1500 ULONG srchcnt = 1;
1501 FILEFINDBUF3 ep;
1502 APIRET rc;
1503
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001504 if (!PyArg_ParseTuple(args, "t#:listdir", &name, &len))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001505 return NULL;
1506 if (len >= MAX_PATH) {
1507 PyErr_SetString(PyExc_ValueError, "path too long");
1508 return NULL;
1509 }
1510 strcpy(namebuf, name);
1511 for (pt = namebuf; *pt; pt++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001512 if (*pt == ALTSEP)
1513 *pt = SEP;
1514 if (namebuf[len-1] != SEP)
1515 namebuf[len++] = SEP;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001516 strcpy(namebuf + len, "*.*");
1517
1518 if ((d = PyList_New(0)) == NULL)
1519 return NULL;
1520
Guido van Rossumc5a0f531997-12-02 20:36:02 +00001521 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
1522 &hdir, /* Handle to Use While Search Directory */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001523 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
Guido van Rossumc5a0f531997-12-02 20:36:02 +00001524 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
1525 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
1526 FIL_STANDARD); /* Format of Entry (EAs or Not) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001527
1528 if (rc != NO_ERROR) {
1529 errno = ENOENT;
Barry Warsawf63b8cc1999-05-27 23:13:21 +00001530 return posix_error_with_filename(name);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001531 }
1532
Guido van Rossumc5a0f531997-12-02 20:36:02 +00001533 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001534 do {
1535 if (ep.achName[0] == '.'
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001536 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
Guido van Rossumc5a0f531997-12-02 20:36:02 +00001537 continue; /* Skip Over "." and ".." Names */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001538
1539 strcpy(namebuf, ep.achName);
1540
Guido van Rossumc5a0f531997-12-02 20:36:02 +00001541 /* Leave Case of Name Alone -- In Native Form */
1542 /* (Removed Forced Lowercasing Code) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001543
1544 v = PyString_FromString(namebuf);
1545 if (v == NULL) {
1546 Py_DECREF(d);
1547 d = NULL;
1548 break;
1549 }
1550 if (PyList_Append(d, v) != 0) {
1551 Py_DECREF(v);
1552 Py_DECREF(d);
1553 d = NULL;
1554 break;
1555 }
1556 Py_DECREF(v);
1557 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
1558 }
1559
1560 return d;
1561#else
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001562
Just van Rossum2fe07fd2003-03-03 19:07:13 +00001563 char *name = NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00001564 PyObject *d, *v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001565 DIR *dirp;
Guido van Rossumb6775db1994-08-01 11:34:53 +00001566 struct dirent *ep;
Just van Rossum96b1c902003-03-03 17:32:15 +00001567 int arg_is_unicode = 1;
1568
1569 if (!PyArg_ParseTuple(args, "U:listdir", &v)) {
1570 arg_is_unicode = 0;
1571 PyErr_Clear();
1572 }
1573 if (!PyArg_ParseTuple(args, "et:listdir", Py_FileSystemDefaultEncoding, &name))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001574 return NULL;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001575 if ((dirp = opendir(name)) == NULL) {
Just van Rossum2fe07fd2003-03-03 19:07:13 +00001576 return posix_error_with_allocated_filename(name);
Guido van Rossumff4949e1992-08-05 19:58:53 +00001577 }
Barry Warsaw53699e91996-12-10 23:23:01 +00001578 if ((d = PyList_New(0)) == NULL) {
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001579 closedir(dirp);
Just van Rossum2fe07fd2003-03-03 19:07:13 +00001580 PyMem_Free(name);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001581 return NULL;
1582 }
1583 while ((ep = readdir(dirp)) != NULL) {
Guido van Rossum24f42ac1995-07-18 18:16:52 +00001584 if (ep->d_name[0] == '.' &&
1585 (NAMLEN(ep) == 1 ||
Guido van Rossuma376cc51996-12-05 23:43:35 +00001586 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
Guido van Rossum24f42ac1995-07-18 18:16:52 +00001587 continue;
Barry Warsaw53699e91996-12-10 23:23:01 +00001588 v = PyString_FromStringAndSize(ep->d_name, NAMLEN(ep));
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001589 if (v == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00001590 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001591 d = NULL;
1592 break;
1593 }
Just van Rossum46c97842003-02-25 21:42:15 +00001594#ifdef Py_USING_UNICODE
Just van Rossum96b1c902003-03-03 17:32:15 +00001595 if (arg_is_unicode) {
Just van Rossum46c97842003-02-25 21:42:15 +00001596 PyObject *w;
1597
1598 w = PyUnicode_FromEncodedObject(v,
Tim Peters11b23062003-04-23 02:39:17 +00001599 Py_FileSystemDefaultEncoding,
Just van Rossum46c97842003-02-25 21:42:15 +00001600 "strict");
Just van Rossum6a421832003-03-04 19:30:44 +00001601 if (w != NULL) {
1602 Py_DECREF(v);
1603 v = w;
1604 }
1605 else {
1606 /* fall back to the original byte string, as
1607 discussed in patch #683592 */
1608 PyErr_Clear();
Just van Rossum46c97842003-02-25 21:42:15 +00001609 }
Just van Rossum46c97842003-02-25 21:42:15 +00001610 }
1611#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00001612 if (PyList_Append(d, v) != 0) {
1613 Py_DECREF(v);
1614 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001615 d = NULL;
1616 break;
1617 }
Barry Warsaw53699e91996-12-10 23:23:01 +00001618 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001619 }
1620 closedir(dirp);
Just van Rossum2fe07fd2003-03-03 19:07:13 +00001621 PyMem_Free(name);
Guido van Rossum0ee42cd1991-04-08 21:01:03 +00001622
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001623 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001624
Tim Peters0bb44a42000-09-15 07:44:49 +00001625#endif /* which OS */
1626} /* end of posix_listdir */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001627
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001628#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00001629/* A helper function for abspath on win32 */
1630static PyObject *
1631posix__getfullpathname(PyObject *self, PyObject *args)
1632{
1633 /* assume encoded strings wont more than double no of chars */
1634 char inbuf[MAX_PATH*2];
1635 char *inbufp = inbuf;
1636 int insize = sizeof(inbuf)/sizeof(inbuf[0]);
1637 char outbuf[MAX_PATH*2];
1638 char *temp;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001639#ifdef Py_WIN_WIDE_FILENAMES
1640 if (unicode_file_names()) {
1641 PyUnicodeObject *po;
1642 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po)) {
1643 Py_UNICODE woutbuf[MAX_PATH*2];
1644 Py_UNICODE *wtemp;
Tim Peters11b23062003-04-23 02:39:17 +00001645 if (!GetFullPathNameW(PyUnicode_AS_UNICODE(po),
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001646 sizeof(woutbuf)/sizeof(woutbuf[0]),
1647 woutbuf, &wtemp))
1648 return win32_error("GetFullPathName", "");
1649 return PyUnicode_FromUnicode(woutbuf, wcslen(woutbuf));
1650 }
1651 /* Drop the argument parsing error as narrow strings
1652 are also valid. */
1653 PyErr_Clear();
1654 }
1655#endif
Tim Peters5aa91602002-01-30 05:46:57 +00001656 if (!PyArg_ParseTuple (args, "et#:_getfullpathname",
1657 Py_FileSystemDefaultEncoding, &inbufp,
Mark Hammondef8b6542001-05-13 08:04:26 +00001658 &insize))
1659 return NULL;
1660 if (!GetFullPathName(inbuf, sizeof(outbuf)/sizeof(outbuf[0]),
1661 outbuf, &temp))
1662 return win32_error("GetFullPathName", inbuf);
1663 return PyString_FromString(outbuf);
1664} /* end of posix__getfullpathname */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001665#endif /* MS_WINDOWS */
Mark Hammondef8b6542001-05-13 08:04:26 +00001666
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001667PyDoc_STRVAR(posix_mkdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001668"mkdir(path [, mode=0777])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001669Create a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001670
Barry Warsaw53699e91996-12-10 23:23:01 +00001671static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001672posix_mkdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001673{
Guido van Rossumb0824db1996-02-25 04:50:32 +00001674 int res;
Mark Hammondef8b6542001-05-13 08:04:26 +00001675 char *path = NULL;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00001676 int mode = 0777;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001677
1678#ifdef Py_WIN_WIDE_FILENAMES
1679 if (unicode_file_names()) {
1680 PyUnicodeObject *po;
Mark Hammond05107b62003-02-19 04:08:27 +00001681 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode)) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001682 Py_BEGIN_ALLOW_THREADS
1683 /* PyUnicode_AS_UNICODE OK without thread lock as
1684 it is a simple dereference. */
1685 res = _wmkdir(PyUnicode_AS_UNICODE(po));
1686 Py_END_ALLOW_THREADS
1687 if (res < 0)
1688 return posix_error();
1689 Py_INCREF(Py_None);
1690 return Py_None;
1691 }
1692 /* Drop the argument parsing error as narrow strings
1693 are also valid. */
1694 PyErr_Clear();
1695 }
1696#endif
1697
Tim Peters5aa91602002-01-30 05:46:57 +00001698 if (!PyArg_ParseTuple(args, "et|i:mkdir",
Mark Hammondef8b6542001-05-13 08:04:26 +00001699 Py_FileSystemDefaultEncoding, &path, &mode))
Guido van Rossumb0824db1996-02-25 04:50:32 +00001700 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00001701 Py_BEGIN_ALLOW_THREADS
Guido van Rossumc5a0f531997-12-02 20:36:02 +00001702#if ( defined(__WATCOMC__) || defined(_MSC_VER) || defined(PYCC_VACPP) ) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00001703 res = mkdir(path);
1704#else
Guido van Rossumb0824db1996-02-25 04:50:32 +00001705 res = mkdir(path, mode);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00001706#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00001707 Py_END_ALLOW_THREADS
Guido van Rossumb0824db1996-02-25 04:50:32 +00001708 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00001709 return posix_error_with_allocated_filename(path);
1710 PyMem_Free(path);
Barry Warsaw53699e91996-12-10 23:23:01 +00001711 Py_INCREF(Py_None);
1712 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001713}
1714
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001715
Guido van Rossumb6775db1994-08-01 11:34:53 +00001716#ifdef HAVE_NICE
Thomas Wouterse38b2f12001-07-11 22:35:31 +00001717#if defined(HAVE_BROKEN_NICE) && defined(HAVE_SYS_RESOURCE_H)
1718#if defined(HAVE_GETPRIORITY) && !defined(PRIO_PROCESS)
1719#include <sys/resource.h>
1720#endif
1721#endif
1722
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001723PyDoc_STRVAR(posix_nice__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001724"nice(inc) -> new_priority\n\n\
1725Decrease the priority of process by inc and return the new priority.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001726
Barry Warsaw53699e91996-12-10 23:23:01 +00001727static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001728posix_nice(PyObject *self, PyObject *args)
Guido van Rossum775f4da1993-01-09 17:18:52 +00001729{
1730 int increment, value;
1731
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001732 if (!PyArg_ParseTuple(args, "i:nice", &increment))
Guido van Rossum775f4da1993-01-09 17:18:52 +00001733 return NULL;
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00001734
1735 /* There are two flavours of 'nice': one that returns the new
1736 priority (as required by almost all standards out there) and the
Thomas Wouterse38b2f12001-07-11 22:35:31 +00001737 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
1738 the use of getpriority() to get the new priority.
Tim Peters5aa91602002-01-30 05:46:57 +00001739
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00001740 If we are of the nice family that returns the new priority, we
1741 need to clear errno before the call, and check if errno is filled
1742 before calling posix_error() on a returnvalue of -1, because the
1743 -1 may be the actual new priority! */
1744
1745 errno = 0;
Guido van Rossum775f4da1993-01-09 17:18:52 +00001746 value = nice(increment);
Thomas Wouterse38b2f12001-07-11 22:35:31 +00001747#if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00001748 if (value == 0)
1749 value = getpriority(PRIO_PROCESS, 0);
1750#endif
1751 if (value == -1 && errno != 0)
1752 /* either nice() or getpriority() returned an error */
Guido van Rossum775f4da1993-01-09 17:18:52 +00001753 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00001754 return PyInt_FromLong((long) value);
Guido van Rossum775f4da1993-01-09 17:18:52 +00001755}
Guido van Rossumb6775db1994-08-01 11:34:53 +00001756#endif /* HAVE_NICE */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00001757
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001758
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001759PyDoc_STRVAR(posix_rename__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001760"rename(old, new)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001761Rename a file or directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001762
Barry Warsaw53699e91996-12-10 23:23:01 +00001763static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001764posix_rename(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001765{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001766#ifdef MS_WINDOWS
1767 return posix_2str(args, "etet:rename", rename, "OO:rename", _wrename);
1768#else
1769 return posix_2str(args, "etet:rename", rename, NULL, NULL);
1770#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001771}
1772
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001773
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001774PyDoc_STRVAR(posix_rmdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001775"rmdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001776Remove a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001777
Barry Warsaw53699e91996-12-10 23:23:01 +00001778static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001779posix_rmdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001780{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001781#ifdef MS_WINDOWS
1782 return posix_1str(args, "et:rmdir", rmdir, "U:rmdir", _wrmdir);
1783#else
1784 return posix_1str(args, "et:rmdir", rmdir, NULL, NULL);
1785#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001786}
1787
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001788
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001789PyDoc_STRVAR(posix_stat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001790"stat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001791Perform a stat system call on the given path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001792
Barry Warsaw53699e91996-12-10 23:23:01 +00001793static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001794posix_stat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001795{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001796#ifdef MS_WINDOWS
1797 return posix_do_stat(self, args, "et:stat", STAT, "U:stat", _wstati64);
1798#else
1799 return posix_do_stat(self, args, "et:stat", STAT, NULL, NULL);
1800#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001801}
1802
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001803
Guido van Rossuma4916fa1996-05-23 22:58:55 +00001804#ifdef HAVE_SYSTEM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001805PyDoc_STRVAR(posix_system__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001806"system(command) -> exit_status\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001807Execute the command (a string) in a subshell.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001808
Barry Warsaw53699e91996-12-10 23:23:01 +00001809static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001810posix_system(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001811{
Guido van Rossumef0a00e1992-01-27 16:51:30 +00001812 char *command;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001813 long sts;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001814 if (!PyArg_ParseTuple(args, "s:system", &command))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001815 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00001816 Py_BEGIN_ALLOW_THREADS
Guido van Rossumef0a00e1992-01-27 16:51:30 +00001817 sts = system(command);
Barry Warsaw53699e91996-12-10 23:23:01 +00001818 Py_END_ALLOW_THREADS
1819 return PyInt_FromLong(sts);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001820}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00001821#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001822
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001823
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001824PyDoc_STRVAR(posix_umask__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001825"umask(new_mask) -> old_mask\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001826Set the current numeric umask and return the previous umask.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001827
Barry Warsaw53699e91996-12-10 23:23:01 +00001828static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001829posix_umask(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001830{
1831 int i;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001832 if (!PyArg_ParseTuple(args, "i:umask", &i))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001833 return NULL;
Fred Drake0368bc42001-07-19 20:48:32 +00001834 i = (int)umask(i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001835 if (i < 0)
1836 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00001837 return PyInt_FromLong((long)i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001838}
1839
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001840
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001841PyDoc_STRVAR(posix_unlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001842"unlink(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001843Remove a file (same as remove(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001844
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001845PyDoc_STRVAR(posix_remove__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001846"remove(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001847Remove a file (same as unlink(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001848
Barry Warsaw53699e91996-12-10 23:23:01 +00001849static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001850posix_unlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001851{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001852#ifdef MS_WINDOWS
1853 return posix_1str(args, "et:remove", unlink, "U:remove", _wunlink);
1854#else
1855 return posix_1str(args, "et:remove", unlink, NULL, NULL);
1856#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001857}
1858
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001859
Guido van Rossumb6775db1994-08-01 11:34:53 +00001860#ifdef HAVE_UNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001861PyDoc_STRVAR(posix_uname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001862"uname() -> (sysname, nodename, release, version, machine)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001863Return a tuple identifying the current operating system.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001864
Barry Warsaw53699e91996-12-10 23:23:01 +00001865static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001866posix_uname(PyObject *self, PyObject *noargs)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00001867{
Guido van Rossumc39de5f1992-02-05 11:15:54 +00001868 struct utsname u;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001869 int res;
Neal Norwitze241ce82003-02-17 18:17:05 +00001870
Barry Warsaw53699e91996-12-10 23:23:01 +00001871 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001872 res = uname(&u);
Barry Warsaw53699e91996-12-10 23:23:01 +00001873 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001874 if (res < 0)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00001875 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00001876 return Py_BuildValue("(sssss)",
Barry Warsaw43d68b81996-12-19 22:10:44 +00001877 u.sysname,
1878 u.nodename,
1879 u.release,
1880 u.version,
1881 u.machine);
Guido van Rossumc39de5f1992-02-05 11:15:54 +00001882}
Guido van Rossumb6775db1994-08-01 11:34:53 +00001883#endif /* HAVE_UNAME */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00001884
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001885static int
1886extract_time(PyObject *t, long* sec, long* usec)
1887{
1888 long intval;
1889 if (PyFloat_Check(t)) {
1890 double tval = PyFloat_AsDouble(t);
1891 PyObject *intobj = t->ob_type->tp_as_number->nb_int(t);
1892 if (!intobj)
1893 return -1;
1894 intval = PyInt_AsLong(intobj);
1895 Py_DECREF(intobj);
1896 *sec = intval;
Tim Peters96940cf2002-09-10 15:37:28 +00001897 *usec = (long)((tval - intval) * 1e6); /* can't exceed 1000000 */
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001898 if (*usec < 0)
1899 /* If rounding gave us a negative number,
1900 truncate. */
1901 *usec = 0;
1902 return 0;
1903 }
1904 intval = PyInt_AsLong(t);
1905 if (intval == -1 && PyErr_Occurred())
1906 return -1;
1907 *sec = intval;
1908 *usec = 0;
Martin v. Löwis076b2092002-09-10 15:04:41 +00001909 return 0;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001910}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001911
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001912PyDoc_STRVAR(posix_utime__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001913"utime(path, (atime, utime))\n\
1914utime(path, None)\n\n\
Barry Warsaw3cef8562000-05-01 16:17:24 +00001915Set the access and modified time of the file to the given values. If the\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001916second form is used, set the access and modified times to the current time.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001917
Barry Warsaw53699e91996-12-10 23:23:01 +00001918static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001919posix_utime(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001920{
Guido van Rossumef0a00e1992-01-27 16:51:30 +00001921 char *path;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001922 long atime, mtime, ausec, musec;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001923 int res;
Barry Warsaw3cef8562000-05-01 16:17:24 +00001924 PyObject* arg;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00001925
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001926#if defined(HAVE_UTIMES)
1927 struct timeval buf[2];
1928#define ATIME buf[0].tv_sec
1929#define MTIME buf[1].tv_sec
1930#elif defined(HAVE_UTIME_H)
Guido van Rossum6d8841c1997-08-14 19:57:39 +00001931/* XXX should define struct utimbuf instead, above */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00001932 struct utimbuf buf;
1933#define ATIME buf.actime
1934#define MTIME buf.modtime
1935#define UTIME_ARG &buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001936#else /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00001937 time_t buf[2];
1938#define ATIME buf[0]
1939#define MTIME buf[1]
1940#define UTIME_ARG buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001941#endif /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00001942
Mark Hammond817c9292003-12-03 01:22:38 +00001943 int have_unicode_filename = 0;
1944#ifdef Py_WIN_WIDE_FILENAMES
1945 PyUnicodeObject *obwpath;
1946 wchar_t *wpath;
1947 if (unicode_file_names()) {
1948 if (PyArg_ParseTuple(args, "UO|:utime", &obwpath, &arg)) {
1949 wpath = PyUnicode_AS_UNICODE(obwpath);
1950 have_unicode_filename = 1;
1951 } else
1952 /* Drop the argument parsing error as narrow strings
1953 are also valid. */
1954 PyErr_Clear();
1955 }
1956#endif /* Py_WIN_WIDE_FILENAMES */
1957
1958 if (!have_unicode_filename && \
Hye-Shik Chang2b2c9732004-01-04 13:54:25 +00001959 !PyArg_ParseTuple(args, "etO:utime",
1960 Py_FileSystemDefaultEncoding, &path, &arg))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001961 return NULL;
Barry Warsaw3cef8562000-05-01 16:17:24 +00001962 if (arg == Py_None) {
1963 /* optional time values not given */
1964 Py_BEGIN_ALLOW_THREADS
Mark Hammond817c9292003-12-03 01:22:38 +00001965#ifdef Py_WIN_WIDE_FILENAMES
1966 if (have_unicode_filename)
1967 res = _wutime(wpath, NULL);
1968 else
1969#endif /* Py_WIN_WIDE_FILENAMES */
Barry Warsaw3cef8562000-05-01 16:17:24 +00001970 res = utime(path, NULL);
1971 Py_END_ALLOW_THREADS
1972 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001973 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
Barry Warsaw3cef8562000-05-01 16:17:24 +00001974 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00001975 "utime() arg 2 must be a tuple (atime, mtime)");
Barry Warsaw3cef8562000-05-01 16:17:24 +00001976 return NULL;
1977 }
1978 else {
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001979 if (extract_time(PyTuple_GET_ITEM(arg, 0),
1980 &atime, &ausec) == -1)
1981 return NULL;
1982 if (extract_time(PyTuple_GET_ITEM(arg, 1),
1983 &mtime, &musec) == -1)
1984 return NULL;
Barry Warsaw3cef8562000-05-01 16:17:24 +00001985 ATIME = atime;
1986 MTIME = mtime;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00001987#ifdef HAVE_UTIMES
1988 buf[0].tv_usec = ausec;
1989 buf[1].tv_usec = musec;
1990 Py_BEGIN_ALLOW_THREADS
1991 res = utimes(path, buf);
1992 Py_END_ALLOW_THREADS
1993#else
Barry Warsaw3cef8562000-05-01 16:17:24 +00001994 Py_BEGIN_ALLOW_THREADS
Mark Hammond817c9292003-12-03 01:22:38 +00001995#ifdef Py_WIN_WIDE_FILENAMES
1996 if (have_unicode_filename)
1997 /* utime is OK with utimbuf, but _wutime insists
1998 on _utimbuf (the msvc headers assert the
1999 underscore version is ansi) */
2000 res = _wutime(wpath, (struct _utimbuf *)UTIME_ARG);
2001 else
2002#endif /* Py_WIN_WIDE_FILENAMES */
Barry Warsaw3cef8562000-05-01 16:17:24 +00002003 res = utime(path, UTIME_ARG);
2004 Py_END_ALLOW_THREADS
Mark Hammond817c9292003-12-03 01:22:38 +00002005#endif /* HAVE_UTIMES */
Barry Warsaw3cef8562000-05-01 16:17:24 +00002006 }
Guido van Rossumff4949e1992-08-05 19:58:53 +00002007 if (res < 0)
Barry Warsawd58d7641998-07-23 16:14:40 +00002008 return posix_error_with_filename(path);
Barry Warsaw53699e91996-12-10 23:23:01 +00002009 Py_INCREF(Py_None);
2010 return Py_None;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002011#undef UTIME_ARG
2012#undef ATIME
2013#undef MTIME
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002014}
2015
Guido van Rossum85e3b011991-06-03 12:42:10 +00002016
Guido van Rossum3b066191991-06-04 19:40:25 +00002017/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002018
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002019PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002020"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002021Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002022
Barry Warsaw53699e91996-12-10 23:23:01 +00002023static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002024posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002025{
2026 int sts;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002027 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
Guido van Rossum85e3b011991-06-03 12:42:10 +00002028 return NULL;
2029 _exit(sts);
Guido van Rossuma376cc51996-12-05 23:43:35 +00002030 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002031}
2032
Martin v. Löwis114619e2002-10-07 06:44:21 +00002033#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
2034static void
2035free_string_array(char **array, int count)
2036{
2037 int i;
2038 for (i = 0; i < count; i++)
2039 PyMem_Free(array[i]);
2040 PyMem_DEL(array);
2041}
2042#endif
2043
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002044
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002045#ifdef HAVE_EXECV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002046PyDoc_STRVAR(posix_execv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002047"execv(path, args)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002048Execute an executable path with arguments, replacing current process.\n\
2049\n\
2050 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002051 args: tuple or list of strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002052
Barry Warsaw53699e91996-12-10 23:23:01 +00002053static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002054posix_execv(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002055{
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002056 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00002057 PyObject *argv;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002058 char **argvlist;
2059 int i, argc;
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002060 PyObject *(*getitem)(PyObject *, int);
Guido van Rossum85e3b011991-06-03 12:42:10 +00002061
Guido van Rossum89b33251993-10-22 14:26:06 +00002062 /* execv has two arguments: (path, argv), where
Guido van Rossum85e3b011991-06-03 12:42:10 +00002063 argv is a list or tuple of strings. */
2064
Martin v. Löwis114619e2002-10-07 06:44:21 +00002065 if (!PyArg_ParseTuple(args, "etO:execv",
2066 Py_FileSystemDefaultEncoding,
2067 &path, &argv))
Guido van Rossum85e3b011991-06-03 12:42:10 +00002068 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002069 if (PyList_Check(argv)) {
2070 argc = PyList_Size(argv);
2071 getitem = PyList_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002072 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002073 else if (PyTuple_Check(argv)) {
2074 argc = PyTuple_Size(argv);
2075 getitem = PyTuple_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002076 }
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002077 else {
Fred Drake661ea262000-10-24 19:57:45 +00002078 PyErr_SetString(PyExc_TypeError, "execv() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002079 PyMem_Free(path);
Guido van Rossum50422b42000-04-26 20:34:28 +00002080 return NULL;
2081 }
2082
2083 if (argc == 0) {
Fred Drake661ea262000-10-24 19:57:45 +00002084 PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002085 PyMem_Free(path);
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002086 return NULL;
2087 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00002088
Barry Warsaw53699e91996-12-10 23:23:01 +00002089 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00002090 if (argvlist == NULL) {
2091 PyMem_Free(path);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00002092 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00002093 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00002094 for (i = 0; i < argc; i++) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00002095 if (!PyArg_Parse((*getitem)(argv, i), "et",
2096 Py_FileSystemDefaultEncoding,
2097 &argvlist[i])) {
2098 free_string_array(argvlist, i);
Tim Peters5aa91602002-01-30 05:46:57 +00002099 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00002100 "execv() arg 2 must contain only strings");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002101 PyMem_Free(path);
Guido van Rossum50422b42000-04-26 20:34:28 +00002102 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00002103
Guido van Rossum85e3b011991-06-03 12:42:10 +00002104 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00002105 }
2106 argvlist[argc] = NULL;
2107
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002108 execv(path, argvlist);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002109
Guido van Rossum85e3b011991-06-03 12:42:10 +00002110 /* If we get here it's definitely an error */
2111
Martin v. Löwis114619e2002-10-07 06:44:21 +00002112 free_string_array(argvlist, argc);
2113 PyMem_Free(path);
Guido van Rossum85e3b011991-06-03 12:42:10 +00002114 return posix_error();
2115}
2116
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002117
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002118PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002119"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002120Execute a path with arguments and environment, replacing current process.\n\
2121\n\
2122 path: path of executable file\n\
2123 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002124 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002125
Barry Warsaw53699e91996-12-10 23:23:01 +00002126static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002127posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002128{
2129 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00002130 PyObject *argv, *env;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002131 char **argvlist;
2132 char **envlist;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00002133 PyObject *key, *val, *keys=NULL, *vals=NULL;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002134 int i, pos, argc, envc;
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002135 PyObject *(*getitem)(PyObject *, int);
Martin v. Löwis114619e2002-10-07 06:44:21 +00002136 int lastarg = 0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002137
2138 /* execve has three arguments: (path, argv, env), where
2139 argv is a list or tuple of strings and env is a dictionary
2140 like posix.environ. */
2141
Martin v. Löwis114619e2002-10-07 06:44:21 +00002142 if (!PyArg_ParseTuple(args, "etOO:execve",
2143 Py_FileSystemDefaultEncoding,
2144 &path, &argv, &env))
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002145 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002146 if (PyList_Check(argv)) {
2147 argc = PyList_Size(argv);
2148 getitem = PyList_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002149 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002150 else if (PyTuple_Check(argv)) {
2151 argc = PyTuple_Size(argv);
2152 getitem = PyTuple_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002153 }
2154 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002155 PyErr_SetString(PyExc_TypeError,
2156 "execve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002157 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002158 }
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00002159 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002160 PyErr_SetString(PyExc_TypeError,
2161 "execve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002162 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002163 }
2164
Guido van Rossum50422b42000-04-26 20:34:28 +00002165 if (argc == 0) {
Tim Peters5aa91602002-01-30 05:46:57 +00002166 PyErr_SetString(PyExc_ValueError,
Fred Drake661ea262000-10-24 19:57:45 +00002167 "execve() arg 2 must not be empty");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002168 goto fail_0;
Guido van Rossum50422b42000-04-26 20:34:28 +00002169 }
2170
Barry Warsaw53699e91996-12-10 23:23:01 +00002171 argvlist = PyMem_NEW(char *, argc+1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002172 if (argvlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00002173 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00002174 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002175 }
2176 for (i = 0; i < argc; i++) {
Barry Warsaw53699e91996-12-10 23:23:01 +00002177 if (!PyArg_Parse((*getitem)(argv, i),
Martin v. Löwis114619e2002-10-07 06:44:21 +00002178 "et;execve() arg 2 must contain only strings",
Guido van Rossum1e700d22002-10-16 16:52:11 +00002179 Py_FileSystemDefaultEncoding,
Barry Warsaw43d68b81996-12-19 22:10:44 +00002180 &argvlist[i]))
2181 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00002182 lastarg = i;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002183 goto fail_1;
2184 }
2185 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00002186 lastarg = argc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002187 argvlist[argc] = NULL;
2188
Jeremy Hylton03657cf2000-07-12 13:05:33 +00002189 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002190 if (i < 0)
2191 goto fail_1;
Barry Warsaw53699e91996-12-10 23:23:01 +00002192 envlist = PyMem_NEW(char *, i + 1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002193 if (envlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00002194 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002195 goto fail_1;
2196 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002197 envc = 0;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00002198 keys = PyMapping_Keys(env);
2199 vals = PyMapping_Values(env);
2200 if (!keys || !vals)
2201 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002202 if (!PyList_Check(keys) || !PyList_Check(vals)) {
2203 PyErr_SetString(PyExc_TypeError,
2204 "execve(): env.keys() or env.values() is not a list");
2205 goto fail_2;
2206 }
Tim Peters5aa91602002-01-30 05:46:57 +00002207
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00002208 for (pos = 0; pos < i; pos++) {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002209 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00002210 size_t len;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00002211
2212 key = PyList_GetItem(keys, pos);
2213 val = PyList_GetItem(vals, pos);
2214 if (!key || !val)
2215 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00002216
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002217 if (!PyArg_Parse(
2218 key,
2219 "s;execve() arg 3 contains a non-string key",
2220 &k) ||
2221 !PyArg_Parse(
2222 val,
2223 "s;execve() arg 3 contains a non-string value",
2224 &v))
Barry Warsaw43d68b81996-12-19 22:10:44 +00002225 {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002226 goto fail_2;
2227 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00002228
2229#if defined(PYOS_OS2)
2230 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
2231 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
2232#endif
Tim Petersc8996f52001-12-03 20:41:00 +00002233 len = PyString_Size(key) + PyString_Size(val) + 2;
2234 p = PyMem_NEW(char, len);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002235 if (p == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00002236 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002237 goto fail_2;
2238 }
Tim Petersc8996f52001-12-03 20:41:00 +00002239 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002240 envlist[envc++] = p;
Guido van Rossumd48f2521997-12-05 22:19:34 +00002241#if defined(PYOS_OS2)
2242 }
2243#endif
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002244 }
2245 envlist[envc] = 0;
2246
2247 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00002248
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002249 /* If we get here it's definitely an error */
2250
2251 (void) posix_error();
2252
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002253 fail_2:
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002254 while (--envc >= 0)
Barry Warsaw53699e91996-12-10 23:23:01 +00002255 PyMem_DEL(envlist[envc]);
2256 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002257 fail_1:
2258 free_string_array(argvlist, lastarg);
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00002259 Py_XDECREF(vals);
2260 Py_XDECREF(keys);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002261 fail_0:
Martin v. Löwis114619e2002-10-07 06:44:21 +00002262 PyMem_Free(path);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002263 return NULL;
2264}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002265#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00002266
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002267
Guido van Rossuma1065681999-01-25 23:20:23 +00002268#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002269PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002270"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00002271Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00002272\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00002273 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00002274 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002275 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00002276
2277static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002278posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00002279{
2280 char *path;
2281 PyObject *argv;
2282 char **argvlist;
2283 int mode, i, argc;
Tim Peters79248aa2001-08-29 21:37:10 +00002284 Py_intptr_t spawnval;
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002285 PyObject *(*getitem)(PyObject *, int);
Guido van Rossuma1065681999-01-25 23:20:23 +00002286
2287 /* spawnv has three arguments: (mode, path, argv), where
2288 argv is a list or tuple of strings. */
2289
Martin v. Löwis114619e2002-10-07 06:44:21 +00002290 if (!PyArg_ParseTuple(args, "ietO:spawnv", &mode,
2291 Py_FileSystemDefaultEncoding,
2292 &path, &argv))
Guido van Rossuma1065681999-01-25 23:20:23 +00002293 return NULL;
2294 if (PyList_Check(argv)) {
2295 argc = PyList_Size(argv);
2296 getitem = PyList_GetItem;
2297 }
2298 else if (PyTuple_Check(argv)) {
2299 argc = PyTuple_Size(argv);
2300 getitem = PyTuple_GetItem;
2301 }
2302 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002303 PyErr_SetString(PyExc_TypeError,
2304 "spawnv() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002305 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00002306 return NULL;
2307 }
2308
2309 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00002310 if (argvlist == NULL) {
2311 PyMem_Free(path);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00002312 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00002313 }
Guido van Rossuma1065681999-01-25 23:20:23 +00002314 for (i = 0; i < argc; i++) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00002315 if (!PyArg_Parse((*getitem)(argv, i), "et",
2316 Py_FileSystemDefaultEncoding,
2317 &argvlist[i])) {
2318 free_string_array(argvlist, i);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002319 PyErr_SetString(
2320 PyExc_TypeError,
2321 "spawnv() arg 2 must contain only strings");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002322 PyMem_Free(path);
Fred Drake137507e2000-06-01 02:02:46 +00002323 return NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00002324 }
2325 }
2326 argvlist[argc] = NULL;
2327
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002328#if defined(PYOS_OS2) && defined(PYCC_GCC)
2329 Py_BEGIN_ALLOW_THREADS
2330 spawnval = spawnv(mode, path, argvlist);
2331 Py_END_ALLOW_THREADS
2332#else
Guido van Rossum246bc171999-02-01 23:54:31 +00002333 if (mode == _OLD_P_OVERLAY)
2334 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00002335
Tim Peters25059d32001-12-07 20:35:43 +00002336 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00002337 spawnval = _spawnv(mode, path, argvlist);
Tim Peters25059d32001-12-07 20:35:43 +00002338 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002339#endif
Tim Peters5aa91602002-01-30 05:46:57 +00002340
Martin v. Löwis114619e2002-10-07 06:44:21 +00002341 free_string_array(argvlist, argc);
2342 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00002343
Fred Drake699f3522000-06-29 21:12:41 +00002344 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00002345 return posix_error();
2346 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00002347#if SIZEOF_LONG == SIZEOF_VOID_P
2348 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00002349#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00002350 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00002351#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00002352}
2353
2354
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002355PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002356"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00002357Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00002358\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00002359 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00002360 path: path of executable file\n\
2361 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002362 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00002363
2364static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002365posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00002366{
2367 char *path;
2368 PyObject *argv, *env;
2369 char **argvlist;
2370 char **envlist;
2371 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
2372 int mode, i, pos, argc, envc;
Tim Peters79248aa2001-08-29 21:37:10 +00002373 Py_intptr_t spawnval;
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002374 PyObject *(*getitem)(PyObject *, int);
Martin v. Löwis114619e2002-10-07 06:44:21 +00002375 int lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00002376
2377 /* spawnve has four arguments: (mode, path, argv, env), where
2378 argv is a list or tuple of strings and env is a dictionary
2379 like posix.environ. */
2380
Martin v. Löwis114619e2002-10-07 06:44:21 +00002381 if (!PyArg_ParseTuple(args, "ietOO:spawnve", &mode,
2382 Py_FileSystemDefaultEncoding,
2383 &path, &argv, &env))
Guido van Rossuma1065681999-01-25 23:20:23 +00002384 return NULL;
2385 if (PyList_Check(argv)) {
2386 argc = PyList_Size(argv);
2387 getitem = PyList_GetItem;
2388 }
2389 else if (PyTuple_Check(argv)) {
2390 argc = PyTuple_Size(argv);
2391 getitem = PyTuple_GetItem;
2392 }
2393 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002394 PyErr_SetString(PyExc_TypeError,
2395 "spawnve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002396 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00002397 }
2398 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002399 PyErr_SetString(PyExc_TypeError,
2400 "spawnve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002401 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00002402 }
2403
2404 argvlist = PyMem_NEW(char *, argc+1);
2405 if (argvlist == NULL) {
2406 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00002407 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00002408 }
2409 for (i = 0; i < argc; i++) {
2410 if (!PyArg_Parse((*getitem)(argv, i),
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002411 "et;spawnve() arg 2 must contain only strings",
Martin v. Löwis114619e2002-10-07 06:44:21 +00002412 Py_FileSystemDefaultEncoding,
Guido van Rossuma1065681999-01-25 23:20:23 +00002413 &argvlist[i]))
2414 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00002415 lastarg = i;
Guido van Rossuma1065681999-01-25 23:20:23 +00002416 goto fail_1;
2417 }
2418 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00002419 lastarg = argc;
Guido van Rossuma1065681999-01-25 23:20:23 +00002420 argvlist[argc] = NULL;
2421
Jeremy Hylton03657cf2000-07-12 13:05:33 +00002422 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002423 if (i < 0)
2424 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00002425 envlist = PyMem_NEW(char *, i + 1);
2426 if (envlist == NULL) {
2427 PyErr_NoMemory();
2428 goto fail_1;
2429 }
2430 envc = 0;
2431 keys = PyMapping_Keys(env);
2432 vals = PyMapping_Values(env);
2433 if (!keys || !vals)
2434 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002435 if (!PyList_Check(keys) || !PyList_Check(vals)) {
2436 PyErr_SetString(PyExc_TypeError,
2437 "spawnve(): env.keys() or env.values() is not a list");
2438 goto fail_2;
2439 }
Tim Peters5aa91602002-01-30 05:46:57 +00002440
Guido van Rossuma1065681999-01-25 23:20:23 +00002441 for (pos = 0; pos < i; pos++) {
2442 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00002443 size_t len;
Guido van Rossuma1065681999-01-25 23:20:23 +00002444
2445 key = PyList_GetItem(keys, pos);
2446 val = PyList_GetItem(vals, pos);
2447 if (!key || !val)
2448 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00002449
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002450 if (!PyArg_Parse(
2451 key,
2452 "s;spawnve() arg 3 contains a non-string key",
2453 &k) ||
2454 !PyArg_Parse(
2455 val,
2456 "s;spawnve() arg 3 contains a non-string value",
2457 &v))
Guido van Rossuma1065681999-01-25 23:20:23 +00002458 {
2459 goto fail_2;
2460 }
Tim Petersc8996f52001-12-03 20:41:00 +00002461 len = PyString_Size(key) + PyString_Size(val) + 2;
2462 p = PyMem_NEW(char, len);
Guido van Rossuma1065681999-01-25 23:20:23 +00002463 if (p == NULL) {
2464 PyErr_NoMemory();
2465 goto fail_2;
2466 }
Tim Petersc8996f52001-12-03 20:41:00 +00002467 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossuma1065681999-01-25 23:20:23 +00002468 envlist[envc++] = p;
2469 }
2470 envlist[envc] = 0;
2471
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002472#if defined(PYOS_OS2) && defined(PYCC_GCC)
2473 Py_BEGIN_ALLOW_THREADS
2474 spawnval = spawnve(mode, path, argvlist, envlist);
2475 Py_END_ALLOW_THREADS
2476#else
Guido van Rossum246bc171999-02-01 23:54:31 +00002477 if (mode == _OLD_P_OVERLAY)
2478 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00002479
2480 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00002481 spawnval = _spawnve(mode, path, argvlist, envlist);
Tim Peters25059d32001-12-07 20:35:43 +00002482 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002483#endif
Tim Peters25059d32001-12-07 20:35:43 +00002484
Fred Drake699f3522000-06-29 21:12:41 +00002485 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00002486 (void) posix_error();
2487 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00002488#if SIZEOF_LONG == SIZEOF_VOID_P
2489 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00002490#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00002491 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00002492#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00002493
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002494 fail_2:
Guido van Rossuma1065681999-01-25 23:20:23 +00002495 while (--envc >= 0)
2496 PyMem_DEL(envlist[envc]);
2497 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00002498 fail_1:
Martin v. Löwis114619e2002-10-07 06:44:21 +00002499 free_string_array(argvlist, lastarg);
Guido van Rossuma1065681999-01-25 23:20:23 +00002500 Py_XDECREF(vals);
2501 Py_XDECREF(keys);
Martin v. Löwis114619e2002-10-07 06:44:21 +00002502 fail_0:
2503 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00002504 return res;
2505}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00002506
2507/* OS/2 supports spawnvp & spawnvpe natively */
2508#if defined(PYOS_OS2)
2509PyDoc_STRVAR(posix_spawnvp__doc__,
2510"spawnvp(mode, file, args)\n\n\
2511Execute the program 'file' in a new process, using the environment\n\
2512search path to find the file.\n\
2513\n\
2514 mode: mode of process creation\n\
2515 file: executable file name\n\
2516 args: tuple or list of strings");
2517
2518static PyObject *
2519posix_spawnvp(PyObject *self, PyObject *args)
2520{
2521 char *path;
2522 PyObject *argv;
2523 char **argvlist;
2524 int mode, i, argc;
2525 Py_intptr_t spawnval;
2526 PyObject *(*getitem)(PyObject *, int);
2527
2528 /* spawnvp has three arguments: (mode, path, argv), where
2529 argv is a list or tuple of strings. */
2530
2531 if (!PyArg_ParseTuple(args, "ietO:spawnvp", &mode,
2532 Py_FileSystemDefaultEncoding,
2533 &path, &argv))
2534 return NULL;
2535 if (PyList_Check(argv)) {
2536 argc = PyList_Size(argv);
2537 getitem = PyList_GetItem;
2538 }
2539 else if (PyTuple_Check(argv)) {
2540 argc = PyTuple_Size(argv);
2541 getitem = PyTuple_GetItem;
2542 }
2543 else {
2544 PyErr_SetString(PyExc_TypeError,
2545 "spawnvp() arg 2 must be a tuple or list");
2546 PyMem_Free(path);
2547 return NULL;
2548 }
2549
2550 argvlist = PyMem_NEW(char *, argc+1);
2551 if (argvlist == NULL) {
2552 PyMem_Free(path);
2553 return PyErr_NoMemory();
2554 }
2555 for (i = 0; i < argc; i++) {
2556 if (!PyArg_Parse((*getitem)(argv, i), "et",
2557 Py_FileSystemDefaultEncoding,
2558 &argvlist[i])) {
2559 free_string_array(argvlist, i);
2560 PyErr_SetString(
2561 PyExc_TypeError,
2562 "spawnvp() arg 2 must contain only strings");
2563 PyMem_Free(path);
2564 return NULL;
2565 }
2566 }
2567 argvlist[argc] = NULL;
2568
2569 Py_BEGIN_ALLOW_THREADS
2570#if defined(PYCC_GCC)
2571 spawnval = spawnvp(mode, path, argvlist);
2572#else
2573 spawnval = _spawnvp(mode, path, argvlist);
2574#endif
2575 Py_END_ALLOW_THREADS
2576
2577 free_string_array(argvlist, argc);
2578 PyMem_Free(path);
2579
2580 if (spawnval == -1)
2581 return posix_error();
2582 else
2583 return Py_BuildValue("l", (long) spawnval);
2584}
2585
2586
2587PyDoc_STRVAR(posix_spawnvpe__doc__,
2588"spawnvpe(mode, file, args, env)\n\n\
2589Execute the program 'file' in a new process, using the environment\n\
2590search path to find the file.\n\
2591\n\
2592 mode: mode of process creation\n\
2593 file: executable file name\n\
2594 args: tuple or list of arguments\n\
2595 env: dictionary of strings mapping to strings");
2596
2597static PyObject *
2598posix_spawnvpe(PyObject *self, PyObject *args)
2599{
2600 char *path;
2601 PyObject *argv, *env;
2602 char **argvlist;
2603 char **envlist;
2604 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
2605 int mode, i, pos, argc, envc;
2606 Py_intptr_t spawnval;
2607 PyObject *(*getitem)(PyObject *, int);
2608 int lastarg = 0;
2609
2610 /* spawnvpe has four arguments: (mode, path, argv, env), where
2611 argv is a list or tuple of strings and env is a dictionary
2612 like posix.environ. */
2613
2614 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
2615 Py_FileSystemDefaultEncoding,
2616 &path, &argv, &env))
2617 return NULL;
2618 if (PyList_Check(argv)) {
2619 argc = PyList_Size(argv);
2620 getitem = PyList_GetItem;
2621 }
2622 else if (PyTuple_Check(argv)) {
2623 argc = PyTuple_Size(argv);
2624 getitem = PyTuple_GetItem;
2625 }
2626 else {
2627 PyErr_SetString(PyExc_TypeError,
2628 "spawnvpe() arg 2 must be a tuple or list");
2629 goto fail_0;
2630 }
2631 if (!PyMapping_Check(env)) {
2632 PyErr_SetString(PyExc_TypeError,
2633 "spawnvpe() arg 3 must be a mapping object");
2634 goto fail_0;
2635 }
2636
2637 argvlist = PyMem_NEW(char *, argc+1);
2638 if (argvlist == NULL) {
2639 PyErr_NoMemory();
2640 goto fail_0;
2641 }
2642 for (i = 0; i < argc; i++) {
2643 if (!PyArg_Parse((*getitem)(argv, i),
2644 "et;spawnvpe() arg 2 must contain only strings",
2645 Py_FileSystemDefaultEncoding,
2646 &argvlist[i]))
2647 {
2648 lastarg = i;
2649 goto fail_1;
2650 }
2651 }
2652 lastarg = argc;
2653 argvlist[argc] = NULL;
2654
2655 i = PyMapping_Size(env);
2656 if (i < 0)
2657 goto fail_1;
2658 envlist = PyMem_NEW(char *, i + 1);
2659 if (envlist == NULL) {
2660 PyErr_NoMemory();
2661 goto fail_1;
2662 }
2663 envc = 0;
2664 keys = PyMapping_Keys(env);
2665 vals = PyMapping_Values(env);
2666 if (!keys || !vals)
2667 goto fail_2;
2668 if (!PyList_Check(keys) || !PyList_Check(vals)) {
2669 PyErr_SetString(PyExc_TypeError,
2670 "spawnvpe(): env.keys() or env.values() is not a list");
2671 goto fail_2;
2672 }
2673
2674 for (pos = 0; pos < i; pos++) {
2675 char *p, *k, *v;
2676 size_t len;
2677
2678 key = PyList_GetItem(keys, pos);
2679 val = PyList_GetItem(vals, pos);
2680 if (!key || !val)
2681 goto fail_2;
2682
2683 if (!PyArg_Parse(
2684 key,
2685 "s;spawnvpe() arg 3 contains a non-string key",
2686 &k) ||
2687 !PyArg_Parse(
2688 val,
2689 "s;spawnvpe() arg 3 contains a non-string value",
2690 &v))
2691 {
2692 goto fail_2;
2693 }
2694 len = PyString_Size(key) + PyString_Size(val) + 2;
2695 p = PyMem_NEW(char, len);
2696 if (p == NULL) {
2697 PyErr_NoMemory();
2698 goto fail_2;
2699 }
2700 PyOS_snprintf(p, len, "%s=%s", k, v);
2701 envlist[envc++] = p;
2702 }
2703 envlist[envc] = 0;
2704
2705 Py_BEGIN_ALLOW_THREADS
2706#if defined(PYCC_GCC)
2707 spawnval = spawnve(mode, path, argvlist, envlist);
2708#else
2709 spawnval = _spawnve(mode, path, argvlist, envlist);
2710#endif
2711 Py_END_ALLOW_THREADS
2712
2713 if (spawnval == -1)
2714 (void) posix_error();
2715 else
2716 res = Py_BuildValue("l", (long) spawnval);
2717
2718 fail_2:
2719 while (--envc >= 0)
2720 PyMem_DEL(envlist[envc]);
2721 PyMem_DEL(envlist);
2722 fail_1:
2723 free_string_array(argvlist, lastarg);
2724 Py_XDECREF(vals);
2725 Py_XDECREF(keys);
2726 fail_0:
2727 PyMem_Free(path);
2728 return res;
2729}
2730#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00002731#endif /* HAVE_SPAWNV */
2732
2733
Guido van Rossum2242f2f2001-04-11 20:58:20 +00002734#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002735PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002736"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00002737Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
2738\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002739Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00002740
2741static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002742posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00002743{
Neal Norwitze241ce82003-02-17 18:17:05 +00002744 int pid = fork1();
Guido van Rossum2242f2f2001-04-11 20:58:20 +00002745 if (pid == -1)
2746 return posix_error();
2747 PyOS_AfterFork();
2748 return PyInt_FromLong((long)pid);
2749}
2750#endif
2751
2752
Guido van Rossumad0ee831995-03-01 10:34:45 +00002753#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002754PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002755"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002756Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002757Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002758
Barry Warsaw53699e91996-12-10 23:23:01 +00002759static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002760posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002761{
Neal Norwitze241ce82003-02-17 18:17:05 +00002762 int pid = fork();
Guido van Rossum85e3b011991-06-03 12:42:10 +00002763 if (pid == -1)
2764 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00002765 if (pid == 0)
2766 PyOS_AfterFork();
Barry Warsaw53699e91996-12-10 23:23:01 +00002767 return PyInt_FromLong((long)pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00002768}
Guido van Rossumad0ee831995-03-01 10:34:45 +00002769#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00002770
Neal Norwitzb59798b2003-03-21 01:43:31 +00002771/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00002772/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
2773#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00002774#define DEV_PTY_FILE "/dev/ptc"
2775#define HAVE_DEV_PTMX
2776#else
2777#define DEV_PTY_FILE "/dev/ptmx"
2778#endif
2779
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002780#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00002781#ifdef HAVE_PTY_H
2782#include <pty.h>
2783#else
2784#ifdef HAVE_LIBUTIL_H
2785#include <libutil.h>
Fred Drake8cef4cf2000-06-28 16:40:38 +00002786#endif /* HAVE_LIBUTIL_H */
2787#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00002788#ifdef HAVE_STROPTS_H
2789#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002790#endif
2791#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00002792
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002793#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002794PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002795"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002796Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00002797
2798static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002799posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00002800{
2801 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00002802#ifndef HAVE_OPENPTY
2803 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00002804#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002805#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00002806 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002807#ifdef sun
2808 extern char *ptsname();
2809#endif
2810#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00002811
Thomas Wouters70c21a12000-07-14 14:28:33 +00002812#ifdef HAVE_OPENPTY
Fred Drake8cef4cf2000-06-28 16:40:38 +00002813 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
2814 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00002815#elif defined(HAVE__GETPTY)
Thomas Wouters70c21a12000-07-14 14:28:33 +00002816 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
2817 if (slave_name == NULL)
2818 return posix_error();
2819
2820 slave_fd = open(slave_name, O_RDWR);
2821 if (slave_fd < 0)
2822 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002823#else
Neal Norwitzb59798b2003-03-21 01:43:31 +00002824 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002825 if (master_fd < 0)
2826 return posix_error();
2827 sig_saved = signal(SIGCHLD, SIG_DFL);
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00002828 /* change permission of slave */
2829 if (grantpt(master_fd) < 0) {
2830 signal(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002831 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00002832 }
2833 /* unlock slave */
2834 if (unlockpt(master_fd) < 0) {
2835 signal(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002836 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00002837 }
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002838 signal(SIGCHLD, sig_saved);
2839 slave_name = ptsname(master_fd); /* get name of slave */
2840 if (slave_name == NULL)
2841 return posix_error();
2842 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
2843 if (slave_fd < 0)
2844 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00002845#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002846 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
2847 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00002848#ifndef __hpux
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002849 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00002850#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002851#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00002852#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00002853
Fred Drake8cef4cf2000-06-28 16:40:38 +00002854 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00002855
Fred Drake8cef4cf2000-06-28 16:40:38 +00002856}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00002857#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00002858
2859#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002860PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002861"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00002862Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
2863Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002864To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00002865
2866static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002867posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00002868{
2869 int master_fd, pid;
Tim Peters5aa91602002-01-30 05:46:57 +00002870
Fred Drake8cef4cf2000-06-28 16:40:38 +00002871 pid = forkpty(&master_fd, NULL, NULL, NULL);
2872 if (pid == -1)
2873 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00002874 if (pid == 0)
2875 PyOS_AfterFork();
Fred Drake8cef4cf2000-06-28 16:40:38 +00002876 return Py_BuildValue("(ii)", pid, master_fd);
2877}
2878#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002879
Guido van Rossumad0ee831995-03-01 10:34:45 +00002880#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002881PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002882"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002883Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002884
Barry Warsaw53699e91996-12-10 23:23:01 +00002885static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002886posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00002887{
Barry Warsaw53699e91996-12-10 23:23:01 +00002888 return PyInt_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00002889}
Guido van Rossumad0ee831995-03-01 10:34:45 +00002890#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00002891
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002892
Guido van Rossumad0ee831995-03-01 10:34:45 +00002893#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002894PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002895"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002896Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002897
Barry Warsaw53699e91996-12-10 23:23:01 +00002898static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002899posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00002900{
Barry Warsaw53699e91996-12-10 23:23:01 +00002901 return PyInt_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00002902}
Guido van Rossumad0ee831995-03-01 10:34:45 +00002903#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00002904
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002905
Guido van Rossumad0ee831995-03-01 10:34:45 +00002906#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002907PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002908"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002909Return the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002910
Barry Warsaw53699e91996-12-10 23:23:01 +00002911static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002912posix_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00002913{
Barry Warsaw53699e91996-12-10 23:23:01 +00002914 return PyInt_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00002915}
Guido van Rossumad0ee831995-03-01 10:34:45 +00002916#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00002917
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002918
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002919PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002920"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002921Return the current process id");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002922
Barry Warsaw53699e91996-12-10 23:23:01 +00002923static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002924posix_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002925{
Barry Warsaw53699e91996-12-10 23:23:01 +00002926 return PyInt_FromLong((long)getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00002927}
2928
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002929
Fred Drakec9680921999-12-13 16:37:25 +00002930#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002931PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002932"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002933Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00002934
2935static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002936posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00002937{
2938 PyObject *result = NULL;
2939
Fred Drakec9680921999-12-13 16:37:25 +00002940#ifdef NGROUPS_MAX
2941#define MAX_GROUPS NGROUPS_MAX
2942#else
2943 /* defined to be 16 on Solaris7, so this should be a small number */
2944#define MAX_GROUPS 64
2945#endif
2946 gid_t grouplist[MAX_GROUPS];
2947 int n;
2948
2949 n = getgroups(MAX_GROUPS, grouplist);
2950 if (n < 0)
2951 posix_error();
2952 else {
2953 result = PyList_New(n);
2954 if (result != NULL) {
Fred Drakec9680921999-12-13 16:37:25 +00002955 int i;
2956 for (i = 0; i < n; ++i) {
Neal Norwitze241ce82003-02-17 18:17:05 +00002957 PyObject *o = PyInt_FromLong((long)grouplist[i]);
Fred Drakec9680921999-12-13 16:37:25 +00002958 if (o == NULL) {
2959 Py_DECREF(result);
2960 result = NULL;
2961 break;
2962 }
2963 PyList_SET_ITEM(result, i, o);
2964 }
2965 }
2966 }
Neal Norwitze241ce82003-02-17 18:17:05 +00002967
Fred Drakec9680921999-12-13 16:37:25 +00002968 return result;
2969}
2970#endif
2971
Martin v. Löwis606edc12002-06-13 21:09:11 +00002972#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00002973PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002974"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00002975Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00002976
2977static PyObject *
2978posix_getpgid(PyObject *self, PyObject *args)
2979{
2980 int pid, pgid;
2981 if (!PyArg_ParseTuple(args, "i:getpgid", &pid))
2982 return NULL;
2983 pgid = getpgid(pid);
2984 if (pgid < 0)
2985 return posix_error();
2986 return PyInt_FromLong((long)pgid);
2987}
2988#endif /* HAVE_GETPGID */
2989
2990
Guido van Rossumb6775db1994-08-01 11:34:53 +00002991#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002992PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002993"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002994Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002995
Barry Warsaw53699e91996-12-10 23:23:01 +00002996static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002997posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00002998{
Guido van Rossumb6775db1994-08-01 11:34:53 +00002999#ifdef GETPGRP_HAVE_ARG
Barry Warsaw53699e91996-12-10 23:23:01 +00003000 return PyInt_FromLong((long)getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003001#else /* GETPGRP_HAVE_ARG */
Barry Warsaw53699e91996-12-10 23:23:01 +00003002 return PyInt_FromLong((long)getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003003#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00003004}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003005#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00003006
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003007
Guido van Rossumb6775db1994-08-01 11:34:53 +00003008#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003009PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003010"setpgrp()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003011Make this process a session leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003012
Barry Warsaw53699e91996-12-10 23:23:01 +00003013static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003014posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00003015{
Guido van Rossum64933891994-10-20 21:56:42 +00003016#ifdef SETPGRP_HAVE_ARG
Guido van Rossumc2670a01992-09-13 20:07:29 +00003017 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00003018#else /* SETPGRP_HAVE_ARG */
Guido van Rossumb6775db1994-08-01 11:34:53 +00003019 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00003020#endif /* SETPGRP_HAVE_ARG */
Guido van Rossum687dd131993-05-17 08:34:16 +00003021 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00003022 Py_INCREF(Py_None);
3023 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00003024}
3025
Guido van Rossumb6775db1994-08-01 11:34:53 +00003026#endif /* HAVE_SETPGRP */
3027
Guido van Rossumad0ee831995-03-01 10:34:45 +00003028#ifdef HAVE_GETPPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003029PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003030"getppid() -> ppid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003031Return the parent's process id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003032
Barry Warsaw53699e91996-12-10 23:23:01 +00003033static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003034posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003035{
Barry Warsaw53699e91996-12-10 23:23:01 +00003036 return PyInt_FromLong((long)getppid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00003037}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003038#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003039
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003040
Fred Drake12c6e2d1999-12-14 21:25:03 +00003041#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003042PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003043"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003044Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00003045
3046static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003047posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00003048{
Neal Norwitze241ce82003-02-17 18:17:05 +00003049 PyObject *result = NULL;
Fred Drakea30680b2000-12-06 21:24:28 +00003050 char *name;
3051 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00003052
Fred Drakea30680b2000-12-06 21:24:28 +00003053 errno = 0;
3054 name = getlogin();
3055 if (name == NULL) {
3056 if (errno)
3057 posix_error();
3058 else
3059 PyErr_SetString(PyExc_OSError,
Fred Drakee63544f2000-12-06 21:45:33 +00003060 "unable to determine login name");
Fred Drakea30680b2000-12-06 21:24:28 +00003061 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00003062 else
3063 result = PyString_FromString(name);
Fred Drakea30680b2000-12-06 21:24:28 +00003064 errno = old_errno;
Neal Norwitze241ce82003-02-17 18:17:05 +00003065
Fred Drake12c6e2d1999-12-14 21:25:03 +00003066 return result;
3067}
3068#endif
3069
Guido van Rossumad0ee831995-03-01 10:34:45 +00003070#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003071PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003072"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003073Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003074
Barry Warsaw53699e91996-12-10 23:23:01 +00003075static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003076posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003077{
Barry Warsaw53699e91996-12-10 23:23:01 +00003078 return PyInt_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003079}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003080#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003081
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003082
Guido van Rossumad0ee831995-03-01 10:34:45 +00003083#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003084PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003085"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003086Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003087
Barry Warsaw53699e91996-12-10 23:23:01 +00003088static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003089posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003090{
3091 int pid, sig;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00003092 if (!PyArg_ParseTuple(args, "ii:kill", &pid, &sig))
Guido van Rossum85e3b011991-06-03 12:42:10 +00003093 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003094#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003095 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
3096 APIRET rc;
3097 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00003098 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003099
3100 } else if (sig == XCPT_SIGNAL_KILLPROC) {
3101 APIRET rc;
3102 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00003103 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003104
3105 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003106 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003107#else
Guido van Rossum85e3b011991-06-03 12:42:10 +00003108 if (kill(pid, sig) == -1)
3109 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003110#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00003111 Py_INCREF(Py_None);
3112 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00003113}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003114#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003115
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00003116#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003117PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003118"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003119Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00003120
3121static PyObject *
3122posix_killpg(PyObject *self, PyObject *args)
3123{
3124 int pgid, sig;
3125 if (!PyArg_ParseTuple(args, "ii:killpg", &pgid, &sig))
3126 return NULL;
3127 if (killpg(pgid, sig) == -1)
3128 return posix_error();
3129 Py_INCREF(Py_None);
3130 return Py_None;
3131}
3132#endif
3133
Guido van Rossumc0125471996-06-28 18:55:32 +00003134#ifdef HAVE_PLOCK
3135
3136#ifdef HAVE_SYS_LOCK_H
3137#include <sys/lock.h>
3138#endif
3139
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003140PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003141"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003142Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003143
Barry Warsaw53699e91996-12-10 23:23:01 +00003144static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003145posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00003146{
3147 int op;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00003148 if (!PyArg_ParseTuple(args, "i:plock", &op))
Guido van Rossumc0125471996-06-28 18:55:32 +00003149 return NULL;
3150 if (plock(op) == -1)
3151 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00003152 Py_INCREF(Py_None);
3153 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00003154}
3155#endif
3156
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003157
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003158#ifdef HAVE_POPEN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003159PyDoc_STRVAR(posix_popen__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003160"popen(command [, mode='r' [, bufsize]]) -> pipe\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003161Open a pipe to/from a command returning a file object.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003162
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003163#if defined(PYOS_OS2)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003164#if defined(PYCC_VACPP)
Guido van Rossumd48f2521997-12-05 22:19:34 +00003165static int
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003166async_system(const char *command)
3167{
3168 char *p, errormsg[256], args[1024];
3169 RESULTCODES rcodes;
3170 APIRET rc;
3171 char *shell = getenv("COMSPEC");
3172 if (!shell)
3173 shell = "cmd";
3174
3175 strcpy(args, shell);
3176 p = &args[ strlen(args)+1 ];
3177 strcpy(p, "/c ");
3178 strcat(p, command);
3179 p += strlen(p) + 1;
3180 *p = '\0';
3181
3182 rc = DosExecPgm(errormsg, sizeof(errormsg),
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003183 EXEC_ASYNC, /* Execute Async w/o Wait for Results */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003184 args,
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003185 NULL, /* Inherit Parent's Environment */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003186 &rcodes, shell);
3187 return rc;
3188}
3189
Guido van Rossumd48f2521997-12-05 22:19:34 +00003190static FILE *
3191popen(const char *command, const char *mode, int pipesize, int *err)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003192{
3193 HFILE rhan, whan;
3194 FILE *retfd = NULL;
3195 APIRET rc = DosCreatePipe(&rhan, &whan, pipesize);
3196
Guido van Rossumd48f2521997-12-05 22:19:34 +00003197 if (rc != NO_ERROR) {
3198 *err = rc;
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003199 return NULL; /* ERROR - Unable to Create Anon Pipe */
Guido van Rossumd48f2521997-12-05 22:19:34 +00003200 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003201
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003202 if (strchr(mode, 'r') != NULL) { /* Treat Command as a Data Source */
3203 int oldfd = dup(1); /* Save STDOUT Handle in Another Handle */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003204
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003205 DosEnterCritSec(); /* Stop Other Threads While Changing Handles */
3206 close(1); /* Make STDOUT Available for Reallocation */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003207
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003208 if (dup2(whan, 1) == 0) { /* Connect STDOUT to Pipe Write Side */
3209 DosClose(whan); /* Close Now-Unused Pipe Write Handle */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003210
Martin v. Löwisdedbe252001-11-02 23:59:11 +00003211 rc = async_system(command);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003212 }
3213
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003214 dup2(oldfd, 1); /* Reconnect STDOUT to Original Handle */
3215 DosExitCritSec(); /* Now Allow Other Threads to Run */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003216
Martin v. Löwisdedbe252001-11-02 23:59:11 +00003217 if (rc == NO_ERROR)
3218 retfd = fdopen(rhan, mode); /* And Return Pipe Read Handle */
3219
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003220 close(oldfd); /* And Close Saved STDOUT Handle */
3221 return retfd; /* Return fd of Pipe or NULL if Error */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003222
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003223 } else if (strchr(mode, 'w')) { /* Treat Command as a Data Sink */
3224 int oldfd = dup(0); /* Save STDIN Handle in Another Handle */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003225
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003226 DosEnterCritSec(); /* Stop Other Threads While Changing Handles */
3227 close(0); /* Make STDIN Available for Reallocation */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003228
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003229 if (dup2(rhan, 0) == 0) { /* Connect STDIN to Pipe Read Side */
3230 DosClose(rhan); /* Close Now-Unused Pipe Read Handle */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003231
Martin v. Löwisdedbe252001-11-02 23:59:11 +00003232 rc = async_system(command);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003233 }
3234
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003235 dup2(oldfd, 0); /* Reconnect STDIN to Original Handle */
3236 DosExitCritSec(); /* Now Allow Other Threads to Run */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003237
Martin v. Löwisdedbe252001-11-02 23:59:11 +00003238 if (rc == NO_ERROR)
3239 retfd = fdopen(whan, mode); /* And Return Pipe Write Handle */
3240
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003241 close(oldfd); /* And Close Saved STDIN Handle */
3242 return retfd; /* Return fd of Pipe or NULL if Error */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003243
Guido van Rossumd48f2521997-12-05 22:19:34 +00003244 } else {
3245 *err = ERROR_INVALID_ACCESS;
Guido van Rossumc5a0f531997-12-02 20:36:02 +00003246 return NULL; /* ERROR - Invalid Mode (Neither Read nor Write) */
Guido van Rossumd48f2521997-12-05 22:19:34 +00003247 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003248}
3249
3250static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003251posix_popen(PyObject *self, PyObject *args)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003252{
3253 char *name;
3254 char *mode = "r";
Guido van Rossumd48f2521997-12-05 22:19:34 +00003255 int err, bufsize = -1;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003256 FILE *fp;
3257 PyObject *f;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00003258 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003259 return NULL;
3260 Py_BEGIN_ALLOW_THREADS
Guido van Rossumd48f2521997-12-05 22:19:34 +00003261 fp = popen(name, mode, (bufsize > 0) ? bufsize : 4096, &err);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003262 Py_END_ALLOW_THREADS
3263 if (fp == NULL)
Guido van Rossumd48f2521997-12-05 22:19:34 +00003264 return os2_error(err);
3265
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003266 f = PyFile_FromFile(fp, name, mode, fclose);
3267 if (f != NULL)
3268 PyFile_SetBufSize(f, bufsize);
3269 return f;
3270}
3271
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003272#elif defined(PYCC_GCC)
3273
3274/* standard posix version of popen() support */
3275static PyObject *
3276posix_popen(PyObject *self, PyObject *args)
3277{
3278 char *name;
3279 char *mode = "r";
3280 int bufsize = -1;
3281 FILE *fp;
3282 PyObject *f;
3283 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
3284 return NULL;
3285 Py_BEGIN_ALLOW_THREADS
3286 fp = popen(name, mode);
3287 Py_END_ALLOW_THREADS
3288 if (fp == NULL)
3289 return posix_error();
3290 f = PyFile_FromFile(fp, name, mode, pclose);
3291 if (f != NULL)
3292 PyFile_SetBufSize(f, bufsize);
3293 return f;
3294}
3295
3296/* fork() under OS/2 has lots'o'warts
3297 * EMX supports pipe() and spawn*() so we can synthesize popen[234]()
3298 * most of this code is a ripoff of the win32 code, but using the
3299 * capabilities of EMX's C library routines
3300 */
3301
3302/* These tell _PyPopen() whether to return 1, 2, or 3 file objects. */
3303#define POPEN_1 1
3304#define POPEN_2 2
3305#define POPEN_3 3
3306#define POPEN_4 4
3307
3308static PyObject *_PyPopen(char *, int, int, int);
3309static int _PyPclose(FILE *file);
3310
3311/*
3312 * Internal dictionary mapping popen* file pointers to process handles,
3313 * for use when retrieving the process exit code. See _PyPclose() below
3314 * for more information on this dictionary's use.
3315 */
3316static PyObject *_PyPopenProcs = NULL;
3317
3318/* os2emx version of popen2()
3319 *
3320 * The result of this function is a pipe (file) connected to the
3321 * process's stdin, and a pipe connected to the process's stdout.
3322 */
3323
3324static PyObject *
3325os2emx_popen2(PyObject *self, PyObject *args)
3326{
3327 PyObject *f;
3328 int tm=0;
3329
3330 char *cmdstring;
3331 char *mode = "t";
3332 int bufsize = -1;
3333 if (!PyArg_ParseTuple(args, "s|si:popen2", &cmdstring, &mode, &bufsize))
3334 return NULL;
3335
3336 if (*mode == 't')
3337 tm = O_TEXT;
3338 else if (*mode != 'b') {
3339 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
3340 return NULL;
3341 } else
3342 tm = O_BINARY;
3343
3344 f = _PyPopen(cmdstring, tm, POPEN_2, bufsize);
3345
3346 return f;
3347}
3348
3349/*
3350 * Variation on os2emx.popen2
3351 *
3352 * The result of this function is 3 pipes - the process's stdin,
3353 * stdout and stderr
3354 */
3355
3356static PyObject *
3357os2emx_popen3(PyObject *self, PyObject *args)
3358{
3359 PyObject *f;
3360 int tm = 0;
3361
3362 char *cmdstring;
3363 char *mode = "t";
3364 int bufsize = -1;
3365 if (!PyArg_ParseTuple(args, "s|si:popen3", &cmdstring, &mode, &bufsize))
3366 return NULL;
3367
3368 if (*mode == 't')
3369 tm = O_TEXT;
3370 else if (*mode != 'b') {
3371 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
3372 return NULL;
3373 } else
3374 tm = O_BINARY;
3375
3376 f = _PyPopen(cmdstring, tm, POPEN_3, bufsize);
3377
3378 return f;
3379}
3380
3381/*
3382 * Variation on os2emx.popen2
3383 *
Tim Peters11b23062003-04-23 02:39:17 +00003384 * The result of this function is 2 pipes - the processes stdin,
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003385 * and stdout+stderr combined as a single pipe.
3386 */
3387
3388static PyObject *
3389os2emx_popen4(PyObject *self, PyObject *args)
3390{
3391 PyObject *f;
3392 int tm = 0;
3393
3394 char *cmdstring;
3395 char *mode = "t";
3396 int bufsize = -1;
3397 if (!PyArg_ParseTuple(args, "s|si:popen4", &cmdstring, &mode, &bufsize))
3398 return NULL;
3399
3400 if (*mode == 't')
3401 tm = O_TEXT;
3402 else if (*mode != 'b') {
3403 PyErr_SetString(PyExc_ValueError, "mode must be 't' or 'b'");
3404 return NULL;
3405 } else
3406 tm = O_BINARY;
3407
3408 f = _PyPopen(cmdstring, tm, POPEN_4, bufsize);
3409
3410 return f;
3411}
3412
3413/* a couple of structures for convenient handling of multiple
3414 * file handles and pipes
3415 */
3416struct file_ref
3417{
3418 int handle;
3419 int flags;
3420};
3421
3422struct pipe_ref
3423{
3424 int rd;
3425 int wr;
3426};
3427
3428/* The following code is derived from the win32 code */
3429
3430static PyObject *
3431_PyPopen(char *cmdstring, int mode, int n, int bufsize)
3432{
3433 struct file_ref stdio[3];
3434 struct pipe_ref p_fd[3];
3435 FILE *p_s[3];
3436 int file_count, i, pipe_err, pipe_pid;
3437 char *shell, *sh_name, *opt, *rd_mode, *wr_mode;
3438 PyObject *f, *p_f[3];
3439
3440 /* file modes for subsequent fdopen's on pipe handles */
3441 if (mode == O_TEXT)
3442 {
3443 rd_mode = "rt";
3444 wr_mode = "wt";
3445 }
3446 else
3447 {
3448 rd_mode = "rb";
3449 wr_mode = "wb";
3450 }
3451
3452 /* prepare shell references */
3453 if ((shell = getenv("EMXSHELL")) == NULL)
3454 if ((shell = getenv("COMSPEC")) == NULL)
3455 {
3456 errno = ENOENT;
3457 return posix_error();
3458 }
3459
3460 sh_name = _getname(shell);
3461 if (stricmp(sh_name, "cmd.exe") == 0 || stricmp(sh_name, "4os2.exe") == 0)
3462 opt = "/c";
3463 else
3464 opt = "-c";
3465
3466 /* save current stdio fds + their flags, and set not inheritable */
3467 i = pipe_err = 0;
3468 while (pipe_err >= 0 && i < 3)
3469 {
3470 pipe_err = stdio[i].handle = dup(i);
3471 stdio[i].flags = fcntl(i, F_GETFD, 0);
3472 fcntl(stdio[i].handle, F_SETFD, stdio[i].flags | FD_CLOEXEC);
3473 i++;
3474 }
3475 if (pipe_err < 0)
3476 {
3477 /* didn't get them all saved - clean up and bail out */
3478 int saved_err = errno;
3479 while (i-- > 0)
3480 {
3481 close(stdio[i].handle);
3482 }
3483 errno = saved_err;
3484 return posix_error();
3485 }
3486
3487 /* create pipe ends */
3488 file_count = 2;
3489 if (n == POPEN_3)
3490 file_count = 3;
3491 i = pipe_err = 0;
3492 while ((pipe_err == 0) && (i < file_count))
3493 pipe_err = pipe((int *)&p_fd[i++]);
3494 if (pipe_err < 0)
3495 {
3496 /* didn't get them all made - clean up and bail out */
3497 while (i-- > 0)
3498 {
3499 close(p_fd[i].wr);
3500 close(p_fd[i].rd);
3501 }
3502 errno = EPIPE;
3503 return posix_error();
3504 }
3505
3506 /* change the actual standard IO streams over temporarily,
3507 * making the retained pipe ends non-inheritable
3508 */
3509 pipe_err = 0;
3510
3511 /* - stdin */
3512 if (dup2(p_fd[0].rd, 0) == 0)
3513 {
3514 close(p_fd[0].rd);
3515 i = fcntl(p_fd[0].wr, F_GETFD, 0);
3516 fcntl(p_fd[0].wr, F_SETFD, i | FD_CLOEXEC);
3517 if ((p_s[0] = fdopen(p_fd[0].wr, wr_mode)) == NULL)
3518 {
3519 close(p_fd[0].wr);
3520 pipe_err = -1;
3521 }
3522 }
3523 else
3524 {
3525 pipe_err = -1;
3526 }
3527
3528 /* - stdout */
3529 if (pipe_err == 0)
3530 {
3531 if (dup2(p_fd[1].wr, 1) == 1)
3532 {
3533 close(p_fd[1].wr);
3534 i = fcntl(p_fd[1].rd, F_GETFD, 0);
3535 fcntl(p_fd[1].rd, F_SETFD, i | FD_CLOEXEC);
3536 if ((p_s[1] = fdopen(p_fd[1].rd, rd_mode)) == NULL)
3537 {
3538 close(p_fd[1].rd);
3539 pipe_err = -1;
3540 }
3541 }
3542 else
3543 {
3544 pipe_err = -1;
3545 }
3546 }
3547
3548 /* - stderr, as required */
3549 if (pipe_err == 0)
3550 switch (n)
3551 {
3552 case POPEN_3:
3553 {
3554 if (dup2(p_fd[2].wr, 2) == 2)
3555 {
3556 close(p_fd[2].wr);
3557 i = fcntl(p_fd[2].rd, F_GETFD, 0);
3558 fcntl(p_fd[2].rd, F_SETFD, i | FD_CLOEXEC);
3559 if ((p_s[2] = fdopen(p_fd[2].rd, rd_mode)) == NULL)
3560 {
3561 close(p_fd[2].rd);
3562 pipe_err = -1;
3563 }
3564 }
3565 else
3566 {
3567 pipe_err = -1;
3568 }
3569 break;
3570 }
3571
3572 case POPEN_4:
3573 {
3574 if (dup2(1, 2) != 2)
3575 {
3576 pipe_err = -1;
3577 }
3578 break;
3579 }
3580 }
3581
3582 /* spawn the child process */
3583 if (pipe_err == 0)
3584 {
3585 pipe_pid = spawnlp(P_NOWAIT, shell, shell, opt, cmdstring, (char *)0);
3586 if (pipe_pid == -1)
3587 {
3588 pipe_err = -1;
3589 }
3590 else
3591 {
3592 /* save the PID into the FILE structure
3593 * NOTE: this implementation doesn't actually
3594 * take advantage of this, but do it for
3595 * completeness - AIM Apr01
3596 */
3597 for (i = 0; i < file_count; i++)
3598 p_s[i]->_pid = pipe_pid;
3599 }
3600 }
3601
3602 /* reset standard IO to normal */
3603 for (i = 0; i < 3; i++)
3604 {
3605 dup2(stdio[i].handle, i);
3606 fcntl(i, F_SETFD, stdio[i].flags);
3607 close(stdio[i].handle);
3608 }
3609
3610 /* if any remnant problems, clean up and bail out */
3611 if (pipe_err < 0)
3612 {
3613 for (i = 0; i < 3; i++)
3614 {
3615 close(p_fd[i].rd);
3616 close(p_fd[i].wr);
3617 }
3618 errno = EPIPE;
3619 return posix_error_with_filename(cmdstring);
3620 }
3621
3622 /* build tuple of file objects to return */
3623 if ((p_f[0] = PyFile_FromFile(p_s[0], cmdstring, wr_mode, _PyPclose)) != NULL)
3624 PyFile_SetBufSize(p_f[0], bufsize);
3625 if ((p_f[1] = PyFile_FromFile(p_s[1], cmdstring, rd_mode, _PyPclose)) != NULL)
3626 PyFile_SetBufSize(p_f[1], bufsize);
3627 if (n == POPEN_3)
3628 {
3629 if ((p_f[2] = PyFile_FromFile(p_s[2], cmdstring, rd_mode, _PyPclose)) != NULL)
3630 PyFile_SetBufSize(p_f[0], bufsize);
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003631 f = PyTuple_Pack(3, p_f[0], p_f[1], p_f[2]);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003632 }
3633 else
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003634 f = PyTuple_Pack(2, p_f[0], p_f[1]);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003635
3636 /*
3637 * Insert the files we've created into the process dictionary
3638 * all referencing the list with the process handle and the
3639 * initial number of files (see description below in _PyPclose).
3640 * Since if _PyPclose later tried to wait on a process when all
3641 * handles weren't closed, it could create a deadlock with the
3642 * child, we spend some energy here to try to ensure that we
3643 * either insert all file handles into the dictionary or none
3644 * at all. It's a little clumsy with the various popen modes
3645 * and variable number of files involved.
3646 */
3647 if (!_PyPopenProcs)
3648 {
3649 _PyPopenProcs = PyDict_New();
3650 }
3651
3652 if (_PyPopenProcs)
3653 {
3654 PyObject *procObj, *pidObj, *intObj, *fileObj[3];
3655 int ins_rc[3];
3656
3657 fileObj[0] = fileObj[1] = fileObj[2] = NULL;
3658 ins_rc[0] = ins_rc[1] = ins_rc[2] = 0;
3659
3660 procObj = PyList_New(2);
3661 pidObj = PyInt_FromLong((long) pipe_pid);
3662 intObj = PyInt_FromLong((long) file_count);
3663
3664 if (procObj && pidObj && intObj)
3665 {
3666 PyList_SetItem(procObj, 0, pidObj);
3667 PyList_SetItem(procObj, 1, intObj);
3668
3669 fileObj[0] = PyLong_FromVoidPtr(p_s[0]);
3670 if (fileObj[0])
3671 {
3672 ins_rc[0] = PyDict_SetItem(_PyPopenProcs,
3673 fileObj[0],
3674 procObj);
3675 }
3676 fileObj[1] = PyLong_FromVoidPtr(p_s[1]);
3677 if (fileObj[1])
3678 {
3679 ins_rc[1] = PyDict_SetItem(_PyPopenProcs,
3680 fileObj[1],
3681 procObj);
3682 }
3683 if (file_count >= 3)
3684 {
3685 fileObj[2] = PyLong_FromVoidPtr(p_s[2]);
3686 if (fileObj[2])
3687 {
3688 ins_rc[2] = PyDict_SetItem(_PyPopenProcs,
3689 fileObj[2],
3690 procObj);
3691 }
3692 }
3693
3694 if (ins_rc[0] < 0 || !fileObj[0] ||
3695 ins_rc[1] < 0 || (file_count > 1 && !fileObj[1]) ||
3696 ins_rc[2] < 0 || (file_count > 2 && !fileObj[2]))
3697 {
3698 /* Something failed - remove any dictionary
3699 * entries that did make it.
3700 */
3701 if (!ins_rc[0] && fileObj[0])
3702 {
3703 PyDict_DelItem(_PyPopenProcs,
3704 fileObj[0]);
3705 }
3706 if (!ins_rc[1] && fileObj[1])
3707 {
3708 PyDict_DelItem(_PyPopenProcs,
3709 fileObj[1]);
3710 }
3711 if (!ins_rc[2] && fileObj[2])
3712 {
3713 PyDict_DelItem(_PyPopenProcs,
3714 fileObj[2]);
3715 }
3716 }
3717 }
Tim Peters11b23062003-04-23 02:39:17 +00003718
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003719 /*
3720 * Clean up our localized references for the dictionary keys
3721 * and value since PyDict_SetItem will Py_INCREF any copies
3722 * that got placed in the dictionary.
3723 */
3724 Py_XDECREF(procObj);
3725 Py_XDECREF(fileObj[0]);
3726 Py_XDECREF(fileObj[1]);
3727 Py_XDECREF(fileObj[2]);
3728 }
3729
3730 /* Child is launched. */
3731 return f;
3732}
3733
3734/*
3735 * Wrapper for fclose() to use for popen* files, so we can retrieve the
3736 * exit code for the child process and return as a result of the close.
3737 *
3738 * This function uses the _PyPopenProcs dictionary in order to map the
3739 * input file pointer to information about the process that was
3740 * originally created by the popen* call that created the file pointer.
3741 * The dictionary uses the file pointer as a key (with one entry
3742 * inserted for each file returned by the original popen* call) and a
3743 * single list object as the value for all files from a single call.
3744 * The list object contains the Win32 process handle at [0], and a file
3745 * count at [1], which is initialized to the total number of file
3746 * handles using that list.
3747 *
3748 * This function closes whichever handle it is passed, and decrements
3749 * the file count in the dictionary for the process handle pointed to
3750 * by this file. On the last close (when the file count reaches zero),
3751 * this function will wait for the child process and then return its
3752 * exit code as the result of the close() operation. This permits the
3753 * files to be closed in any order - it is always the close() of the
3754 * final handle that will return the exit code.
Andrew MacIntyrebaf25b02003-04-21 14:22:36 +00003755 *
3756 * NOTE: This function is currently called with the GIL released.
3757 * hence we use the GILState API to manage our state.
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003758 */
3759
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003760static int _PyPclose(FILE *file)
3761{
3762 int result;
3763 int exit_code;
3764 int pipe_pid;
3765 PyObject *procObj, *pidObj, *intObj, *fileObj;
3766 int file_count;
3767#ifdef WITH_THREAD
Andrew MacIntyrebaf25b02003-04-21 14:22:36 +00003768 PyGILState_STATE state;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003769#endif
3770
3771 /* Close the file handle first, to ensure it can't block the
3772 * child from exiting if it's the last handle.
3773 */
3774 result = fclose(file);
3775
3776#ifdef WITH_THREAD
Andrew MacIntyrebaf25b02003-04-21 14:22:36 +00003777 state = PyGILState_Ensure();
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003778#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003779 if (_PyPopenProcs)
3780 {
3781 if ((fileObj = PyLong_FromVoidPtr(file)) != NULL &&
3782 (procObj = PyDict_GetItem(_PyPopenProcs,
3783 fileObj)) != NULL &&
3784 (pidObj = PyList_GetItem(procObj,0)) != NULL &&
3785 (intObj = PyList_GetItem(procObj,1)) != NULL)
3786 {
3787 pipe_pid = (int) PyInt_AsLong(pidObj);
3788 file_count = (int) PyInt_AsLong(intObj);
3789
3790 if (file_count > 1)
3791 {
3792 /* Still other files referencing process */
3793 file_count--;
3794 PyList_SetItem(procObj,1,
3795 PyInt_FromLong((long) file_count));
3796 }
3797 else
3798 {
3799 /* Last file for this process */
3800 if (result != EOF &&
3801 waitpid(pipe_pid, &exit_code, 0) == pipe_pid)
3802 {
3803 /* extract exit status */
3804 if (WIFEXITED(exit_code))
3805 {
3806 result = WEXITSTATUS(exit_code);
3807 }
3808 else
3809 {
3810 errno = EPIPE;
3811 result = -1;
3812 }
3813 }
3814 else
3815 {
3816 /* Indicate failure - this will cause the file object
3817 * to raise an I/O error and translate the last
3818 * error code from errno. We do have a problem with
3819 * last errors that overlap the normal errno table,
3820 * but that's a consistent problem with the file object.
3821 */
3822 result = -1;
3823 }
3824 }
3825
3826 /* Remove this file pointer from dictionary */
3827 PyDict_DelItem(_PyPopenProcs, fileObj);
3828
3829 if (PyDict_Size(_PyPopenProcs) == 0)
3830 {
3831 Py_DECREF(_PyPopenProcs);
3832 _PyPopenProcs = NULL;
3833 }
3834
3835 } /* if object retrieval ok */
3836
3837 Py_XDECREF(fileObj);
3838 } /* if _PyPopenProcs */
3839
3840#ifdef WITH_THREAD
Andrew MacIntyrebaf25b02003-04-21 14:22:36 +00003841 PyGILState_Release(state);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003842#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003843 return result;
3844}
3845
3846#endif /* PYCC_??? */
3847
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00003848#elif defined(MS_WINDOWS)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003849
3850/*
3851 * Portable 'popen' replacement for Win32.
3852 *
3853 * Written by Bill Tutt <billtut@microsoft.com>. Minor tweaks
3854 * and 2.0 integration by Fredrik Lundh <fredrik@pythonware.com>
Mark Hammondb37a3732000-08-14 04:47:33 +00003855 * Return code handling by David Bolen <db3l@fitlinxx.com>.
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003856 */
3857
3858#include <malloc.h>
3859#include <io.h>
3860#include <fcntl.h>
3861
3862/* These tell _PyPopen() wether to return 1, 2, or 3 file objects. */
3863#define POPEN_1 1
3864#define POPEN_2 2
3865#define POPEN_3 3
3866#define POPEN_4 4
3867
3868static PyObject *_PyPopen(char *, int, int);
Fredrik Lundh56055a42000-07-23 19:47:12 +00003869static int _PyPclose(FILE *file);
3870
3871/*
3872 * Internal dictionary mapping popen* file pointers to process handles,
Mark Hammondb37a3732000-08-14 04:47:33 +00003873 * for use when retrieving the process exit code. See _PyPclose() below
3874 * for more information on this dictionary's use.
Fredrik Lundh56055a42000-07-23 19:47:12 +00003875 */
3876static PyObject *_PyPopenProcs = NULL;
3877
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003878
3879/* popen that works from a GUI.
3880 *
3881 * The result of this function is a pipe (file) connected to the
3882 * processes stdin or stdout, depending on the requested mode.
3883 */
3884
3885static PyObject *
3886posix_popen(PyObject *self, PyObject *args)
3887{
Raymond Hettingerb5cb6652003-09-01 22:25:41 +00003888 PyObject *f;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003889 int tm = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00003890
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003891 char *cmdstring;
3892 char *mode = "r";
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00003893 int bufsize = -1;
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00003894 if (!PyArg_ParseTuple(args, "s|si:popen", &cmdstring, &mode, &bufsize))
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003895 return NULL;
3896
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003897 if (*mode == 'r')
3898 tm = _O_RDONLY;
3899 else if (*mode != 'w') {
Fred Drake661ea262000-10-24 19:57:45 +00003900 PyErr_SetString(PyExc_ValueError, "popen() arg 2 must be 'r' or 'w'");
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003901 return NULL;
3902 } else
3903 tm = _O_WRONLY;
Tim Peters5aa91602002-01-30 05:46:57 +00003904
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00003905 if (bufsize != -1) {
Fred Drake661ea262000-10-24 19:57:45 +00003906 PyErr_SetString(PyExc_ValueError, "popen() arg 3 must be -1");
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00003907 return NULL;
3908 }
3909
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003910 if (*(mode+1) == 't')
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00003911 f = _PyPopen(cmdstring, tm | _O_TEXT, POPEN_1);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003912 else if (*(mode+1) == 'b')
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00003913 f = _PyPopen(cmdstring, tm | _O_BINARY, POPEN_1);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003914 else
3915 f = _PyPopen(cmdstring, tm | _O_TEXT, POPEN_1);
3916
3917 return f;
3918}
3919
3920/* Variation on win32pipe.popen
3921 *
3922 * The result of this function is a pipe (file) connected to the
3923 * process's stdin, and a pipe connected to the process's stdout.
3924 */
3925
3926static PyObject *
3927win32_popen2(PyObject *self, PyObject *args)
3928{
3929 PyObject *f;
3930 int tm=0;
Tim Peters5aa91602002-01-30 05:46:57 +00003931
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003932 char *cmdstring;
3933 char *mode = "t";
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00003934 int bufsize = -1;
3935 if (!PyArg_ParseTuple(args, "s|si:popen2", &cmdstring, &mode, &bufsize))
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003936 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00003937
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003938 if (*mode == 't')
3939 tm = _O_TEXT;
3940 else if (*mode != 'b') {
Fred Drake661ea262000-10-24 19:57:45 +00003941 PyErr_SetString(PyExc_ValueError, "popen2() arg 2 must be 't' or 'b'");
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003942 return NULL;
3943 } else
3944 tm = _O_BINARY;
Tim Peters5aa91602002-01-30 05:46:57 +00003945
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00003946 if (bufsize != -1) {
Fred Drake661ea262000-10-24 19:57:45 +00003947 PyErr_SetString(PyExc_ValueError, "popen2() arg 3 must be -1");
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00003948 return NULL;
3949 }
3950
3951 f = _PyPopen(cmdstring, tm, POPEN_2);
Tim Peters5aa91602002-01-30 05:46:57 +00003952
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003953 return f;
3954}
3955
3956/*
3957 * Variation on <om win32pipe.popen>
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00003958 *
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003959 * The result of this function is 3 pipes - the process's stdin,
3960 * stdout and stderr
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003961 */
3962
3963static PyObject *
3964win32_popen3(PyObject *self, PyObject *args)
3965{
3966 PyObject *f;
3967 int tm = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00003968
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003969 char *cmdstring;
3970 char *mode = "t";
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00003971 int bufsize = -1;
3972 if (!PyArg_ParseTuple(args, "s|si:popen3", &cmdstring, &mode, &bufsize))
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003973 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00003974
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003975 if (*mode == 't')
3976 tm = _O_TEXT;
3977 else if (*mode != 'b') {
Fred Drake661ea262000-10-24 19:57:45 +00003978 PyErr_SetString(PyExc_ValueError, "popen3() arg 2 must be 't' or 'b'");
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003979 return NULL;
3980 } else
3981 tm = _O_BINARY;
Tim Peters5aa91602002-01-30 05:46:57 +00003982
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00003983 if (bufsize != -1) {
Fred Drake661ea262000-10-24 19:57:45 +00003984 PyErr_SetString(PyExc_ValueError, "popen3() arg 3 must be -1");
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00003985 return NULL;
3986 }
3987
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003988 f = _PyPopen(cmdstring, tm, POPEN_3);
Tim Peters5aa91602002-01-30 05:46:57 +00003989
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003990 return f;
3991}
3992
3993/*
3994 * Variation on win32pipe.popen
3995 *
Tim Peters5aa91602002-01-30 05:46:57 +00003996 * The result of this function is 2 pipes - the processes stdin,
Fredrik Lundhffb9c772000-07-09 14:49:51 +00003997 * and stdout+stderr combined as a single pipe.
3998 */
3999
4000static PyObject *
4001win32_popen4(PyObject *self, PyObject *args)
4002{
4003 PyObject *f;
4004 int tm = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00004005
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004006 char *cmdstring;
4007 char *mode = "t";
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004008 int bufsize = -1;
4009 if (!PyArg_ParseTuple(args, "s|si:popen4", &cmdstring, &mode, &bufsize))
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004010 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00004011
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004012 if (*mode == 't')
4013 tm = _O_TEXT;
4014 else if (*mode != 'b') {
Fred Drake661ea262000-10-24 19:57:45 +00004015 PyErr_SetString(PyExc_ValueError, "popen4() arg 2 must be 't' or 'b'");
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004016 return NULL;
4017 } else
4018 tm = _O_BINARY;
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004019
4020 if (bufsize != -1) {
Fred Drake661ea262000-10-24 19:57:45 +00004021 PyErr_SetString(PyExc_ValueError, "popen4() arg 3 must be -1");
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004022 return NULL;
4023 }
4024
Fredrik Lundhbb7eeff2000-07-09 17:59:32 +00004025 f = _PyPopen(cmdstring, tm, POPEN_4);
Fredrik Lundh766ccdc2000-07-09 17:41:01 +00004026
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004027 return f;
4028}
4029
Mark Hammond08501372001-01-31 07:30:29 +00004030static BOOL
Mark Hammondb37a3732000-08-14 04:47:33 +00004031_PyPopenCreateProcess(char *cmdstring,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004032 HANDLE hStdin,
4033 HANDLE hStdout,
Mark Hammondb37a3732000-08-14 04:47:33 +00004034 HANDLE hStderr,
4035 HANDLE *hProcess)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004036{
4037 PROCESS_INFORMATION piProcInfo;
4038 STARTUPINFO siStartInfo;
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004039 DWORD dwProcessFlags = 0; /* no NEW_CONSOLE by default for Ctrl+C handling */
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004040 char *s1,*s2, *s3 = " /c ";
Mark Hammond08501372001-01-31 07:30:29 +00004041 const char *szConsoleSpawn = "w9xpopen.exe";
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004042 int i;
4043 int x;
4044
4045 if (i = GetEnvironmentVariable("COMSPEC",NULL,0)) {
Tim Peters402d5982001-08-27 06:37:48 +00004046 char *comshell;
4047
Tim Peters92e4dd82002-10-05 01:47:34 +00004048 s1 = (char *)alloca(i);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004049 if (!(x = GetEnvironmentVariable("COMSPEC", s1, i)))
4050 return x;
Tim Peters402d5982001-08-27 06:37:48 +00004051
4052 /* Explicitly check if we are using COMMAND.COM. If we are
4053 * then use the w9xpopen hack.
4054 */
4055 comshell = s1 + x;
4056 while (comshell >= s1 && *comshell != '\\')
4057 --comshell;
4058 ++comshell;
4059
4060 if (GetVersion() < 0x80000000 &&
4061 _stricmp(comshell, "command.com") != 0) {
4062 /* NT/2000 and not using command.com. */
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004063 x = i + strlen(s3) + strlen(cmdstring) + 1;
Tim Peters92e4dd82002-10-05 01:47:34 +00004064 s2 = (char *)alloca(x);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004065 ZeroMemory(s2, x);
Tim Peters75cdad52001-11-28 22:07:30 +00004066 PyOS_snprintf(s2, x, "%s%s%s", s1, s3, cmdstring);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004067 }
4068 else {
4069 /*
Tim Peters402d5982001-08-27 06:37:48 +00004070 * Oh gag, we're on Win9x or using COMMAND.COM. Use
4071 * the workaround listed in KB: Q150956
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004072 */
Mark Hammond08501372001-01-31 07:30:29 +00004073 char modulepath[_MAX_PATH];
4074 struct stat statinfo;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004075 GetModuleFileName(NULL, modulepath, sizeof(modulepath));
4076 for (i = x = 0; modulepath[i]; i++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004077 if (modulepath[i] == SEP)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004078 x = i+1;
4079 modulepath[x] = '\0';
Mark Hammond08501372001-01-31 07:30:29 +00004080 /* Create the full-name to w9xpopen, so we can test it exists */
Tim Peters5aa91602002-01-30 05:46:57 +00004081 strncat(modulepath,
4082 szConsoleSpawn,
Mark Hammond08501372001-01-31 07:30:29 +00004083 (sizeof(modulepath)/sizeof(modulepath[0]))
4084 -strlen(modulepath));
4085 if (stat(modulepath, &statinfo) != 0) {
Tim Peters5aa91602002-01-30 05:46:57 +00004086 /* Eeek - file-not-found - possibly an embedding
4087 situation - see if we can locate it in sys.prefix
Mark Hammond08501372001-01-31 07:30:29 +00004088 */
Tim Peters5aa91602002-01-30 05:46:57 +00004089 strncpy(modulepath,
4090 Py_GetExecPrefix(),
Mark Hammond08501372001-01-31 07:30:29 +00004091 sizeof(modulepath)/sizeof(modulepath[0]));
4092 if (modulepath[strlen(modulepath)-1] != '\\')
4093 strcat(modulepath, "\\");
Tim Peters5aa91602002-01-30 05:46:57 +00004094 strncat(modulepath,
4095 szConsoleSpawn,
Mark Hammond08501372001-01-31 07:30:29 +00004096 (sizeof(modulepath)/sizeof(modulepath[0]))
4097 -strlen(modulepath));
4098 /* No where else to look - raise an easily identifiable
4099 error, rather than leaving Windows to report
4100 "file not found" - as the user is probably blissfully
4101 unaware this shim EXE is used, and it will confuse them.
4102 (well, it confused me for a while ;-)
4103 */
4104 if (stat(modulepath, &statinfo) != 0) {
Tim Peters5aa91602002-01-30 05:46:57 +00004105 PyErr_Format(PyExc_RuntimeError,
Mark Hammond08501372001-01-31 07:30:29 +00004106 "Can not locate '%s' which is needed "
Tim Peters402d5982001-08-27 06:37:48 +00004107 "for popen to work with your shell "
4108 "or platform.",
Mark Hammond08501372001-01-31 07:30:29 +00004109 szConsoleSpawn);
4110 return FALSE;
4111 }
4112 }
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004113 x = i + strlen(s3) + strlen(cmdstring) + 1 +
Tim Peters5aa91602002-01-30 05:46:57 +00004114 strlen(modulepath) +
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004115 strlen(szConsoleSpawn) + 1;
Mark Hammond08501372001-01-31 07:30:29 +00004116
Tim Peters92e4dd82002-10-05 01:47:34 +00004117 s2 = (char *)alloca(x);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004118 ZeroMemory(s2, x);
Mark Hammonde7fefbf2002-04-03 01:47:00 +00004119 /* To maintain correct argument passing semantics,
4120 we pass the command-line as it stands, and allow
4121 quoting to be applied. w9xpopen.exe will then
4122 use its argv vector, and re-quote the necessary
4123 args for the ultimate child process.
4124 */
Tim Peters75cdad52001-11-28 22:07:30 +00004125 PyOS_snprintf(
4126 s2, x,
Mark Hammonde7fefbf2002-04-03 01:47:00 +00004127 "\"%s\" %s%s%s",
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004128 modulepath,
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004129 s1,
4130 s3,
4131 cmdstring);
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004132 /* Not passing CREATE_NEW_CONSOLE has been known to
Tim Peters11b23062003-04-23 02:39:17 +00004133 cause random failures on win9x. Specifically a
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004134 dialog:
4135 "Your program accessed mem currently in use at xxx"
4136 and a hopeful warning about the stability of your
4137 system.
4138 Cost is Ctrl+C wont kill children, but anyone
4139 who cares can have a go!
4140 */
4141 dwProcessFlags |= CREATE_NEW_CONSOLE;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004142 }
4143 }
4144
4145 /* Could be an else here to try cmd.exe / command.com in the path
4146 Now we'll just error out.. */
Mark Hammond08501372001-01-31 07:30:29 +00004147 else {
Tim Peters402d5982001-08-27 06:37:48 +00004148 PyErr_SetString(PyExc_RuntimeError,
4149 "Cannot locate a COMSPEC environment variable to "
4150 "use as the shell");
Mark Hammond08501372001-01-31 07:30:29 +00004151 return FALSE;
4152 }
Tim Peters5aa91602002-01-30 05:46:57 +00004153
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004154 ZeroMemory(&siStartInfo, sizeof(STARTUPINFO));
4155 siStartInfo.cb = sizeof(STARTUPINFO);
4156 siStartInfo.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
4157 siStartInfo.hStdInput = hStdin;
4158 siStartInfo.hStdOutput = hStdout;
4159 siStartInfo.hStdError = hStderr;
4160 siStartInfo.wShowWindow = SW_HIDE;
4161
4162 if (CreateProcess(NULL,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004163 s2,
4164 NULL,
4165 NULL,
4166 TRUE,
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004167 dwProcessFlags,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004168 NULL,
4169 NULL,
4170 &siStartInfo,
4171 &piProcInfo) ) {
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004172 /* Close the handles now so anyone waiting is woken. */
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004173 CloseHandle(piProcInfo.hThread);
Fredrik Lundh56055a42000-07-23 19:47:12 +00004174
Mark Hammondb37a3732000-08-14 04:47:33 +00004175 /* Return process handle */
4176 *hProcess = piProcInfo.hProcess;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004177 return TRUE;
4178 }
Tim Peters402d5982001-08-27 06:37:48 +00004179 win32_error("CreateProcess", s2);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004180 return FALSE;
4181}
4182
4183/* The following code is based off of KB: Q190351 */
4184
4185static PyObject *
4186_PyPopen(char *cmdstring, int mode, int n)
4187{
4188 HANDLE hChildStdinRd, hChildStdinWr, hChildStdoutRd, hChildStdoutWr,
4189 hChildStderrRd, hChildStderrWr, hChildStdinWrDup, hChildStdoutRdDup,
Mark Hammondb37a3732000-08-14 04:47:33 +00004190 hChildStderrRdDup, hProcess; /* hChildStdoutWrDup; */
Tim Peters5aa91602002-01-30 05:46:57 +00004191
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004192 SECURITY_ATTRIBUTES saAttr;
4193 BOOL fSuccess;
4194 int fd1, fd2, fd3;
4195 FILE *f1, *f2, *f3;
Mark Hammondb37a3732000-08-14 04:47:33 +00004196 long file_count;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004197 PyObject *f;
4198
4199 saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
4200 saAttr.bInheritHandle = TRUE;
4201 saAttr.lpSecurityDescriptor = NULL;
4202
4203 if (!CreatePipe(&hChildStdinRd, &hChildStdinWr, &saAttr, 0))
4204 return win32_error("CreatePipe", NULL);
4205
4206 /* Create new output read handle and the input write handle. Set
4207 * the inheritance properties to FALSE. Otherwise, the child inherits
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00004208 * these handles; resulting in non-closeable handles to the pipes
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004209 * being created. */
4210 fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdinWr,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004211 GetCurrentProcess(), &hChildStdinWrDup, 0,
4212 FALSE,
4213 DUPLICATE_SAME_ACCESS);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004214 if (!fSuccess)
4215 return win32_error("DuplicateHandle", NULL);
4216
4217 /* Close the inheritable version of ChildStdin
4218 that we're using. */
4219 CloseHandle(hChildStdinWr);
4220
4221 if (!CreatePipe(&hChildStdoutRd, &hChildStdoutWr, &saAttr, 0))
4222 return win32_error("CreatePipe", NULL);
4223
4224 fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdoutRd,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004225 GetCurrentProcess(), &hChildStdoutRdDup, 0,
4226 FALSE, DUPLICATE_SAME_ACCESS);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004227 if (!fSuccess)
4228 return win32_error("DuplicateHandle", NULL);
4229
4230 /* Close the inheritable version of ChildStdout
4231 that we're using. */
4232 CloseHandle(hChildStdoutRd);
4233
4234 if (n != POPEN_4) {
4235 if (!CreatePipe(&hChildStderrRd, &hChildStderrWr, &saAttr, 0))
4236 return win32_error("CreatePipe", NULL);
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004237 fSuccess = DuplicateHandle(GetCurrentProcess(),
4238 hChildStderrRd,
4239 GetCurrentProcess(),
4240 &hChildStderrRdDup, 0,
4241 FALSE, DUPLICATE_SAME_ACCESS);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004242 if (!fSuccess)
4243 return win32_error("DuplicateHandle", NULL);
4244 /* Close the inheritable version of ChildStdErr that we're using. */
4245 CloseHandle(hChildStderrRd);
4246 }
Tim Peters5aa91602002-01-30 05:46:57 +00004247
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004248 switch (n) {
4249 case POPEN_1:
4250 switch (mode & (_O_RDONLY | _O_TEXT | _O_BINARY | _O_WRONLY)) {
4251 case _O_WRONLY | _O_TEXT:
4252 /* Case for writing to child Stdin in text mode. */
4253 fd1 = _open_osfhandle((long)hChildStdinWrDup, mode);
4254 f1 = _fdopen(fd1, "w");
Fredrik Lundh56055a42000-07-23 19:47:12 +00004255 f = PyFile_FromFile(f1, cmdstring, "w", _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004256 PyFile_SetBufSize(f, 0);
4257 /* We don't care about these pipes anymore, so close them. */
4258 CloseHandle(hChildStdoutRdDup);
4259 CloseHandle(hChildStderrRdDup);
4260 break;
4261
4262 case _O_RDONLY | _O_TEXT:
4263 /* Case for reading from child Stdout in text mode. */
4264 fd1 = _open_osfhandle((long)hChildStdoutRdDup, mode);
4265 f1 = _fdopen(fd1, "r");
Fredrik Lundh56055a42000-07-23 19:47:12 +00004266 f = PyFile_FromFile(f1, cmdstring, "r", _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004267 PyFile_SetBufSize(f, 0);
4268 /* We don't care about these pipes anymore, so close them. */
4269 CloseHandle(hChildStdinWrDup);
4270 CloseHandle(hChildStderrRdDup);
4271 break;
4272
4273 case _O_RDONLY | _O_BINARY:
4274 /* Case for readinig from child Stdout in binary mode. */
4275 fd1 = _open_osfhandle((long)hChildStdoutRdDup, mode);
4276 f1 = _fdopen(fd1, "rb");
Fredrik Lundh56055a42000-07-23 19:47:12 +00004277 f = PyFile_FromFile(f1, cmdstring, "rb", _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004278 PyFile_SetBufSize(f, 0);
4279 /* We don't care about these pipes anymore, so close them. */
4280 CloseHandle(hChildStdinWrDup);
4281 CloseHandle(hChildStderrRdDup);
4282 break;
4283
4284 case _O_WRONLY | _O_BINARY:
4285 /* Case for writing to child Stdin in binary mode. */
4286 fd1 = _open_osfhandle((long)hChildStdinWrDup, mode);
4287 f1 = _fdopen(fd1, "wb");
Fredrik Lundh56055a42000-07-23 19:47:12 +00004288 f = PyFile_FromFile(f1, cmdstring, "wb", _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004289 PyFile_SetBufSize(f, 0);
4290 /* We don't care about these pipes anymore, so close them. */
4291 CloseHandle(hChildStdoutRdDup);
4292 CloseHandle(hChildStderrRdDup);
4293 break;
4294 }
Mark Hammondb37a3732000-08-14 04:47:33 +00004295 file_count = 1;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004296 break;
Tim Peters5aa91602002-01-30 05:46:57 +00004297
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004298 case POPEN_2:
4299 case POPEN_4:
4300 {
4301 char *m1, *m2;
4302 PyObject *p1, *p2;
Tim Peters5aa91602002-01-30 05:46:57 +00004303
Tim Peters7dca21e2002-08-19 00:42:29 +00004304 if (mode & _O_TEXT) {
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004305 m1 = "r";
4306 m2 = "w";
4307 } else {
4308 m1 = "rb";
4309 m2 = "wb";
4310 }
4311
4312 fd1 = _open_osfhandle((long)hChildStdinWrDup, mode);
4313 f1 = _fdopen(fd1, m2);
4314 fd2 = _open_osfhandle((long)hChildStdoutRdDup, mode);
4315 f2 = _fdopen(fd2, m1);
Fredrik Lundh56055a42000-07-23 19:47:12 +00004316 p1 = PyFile_FromFile(f1, cmdstring, m2, _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004317 PyFile_SetBufSize(p1, 0);
Mark Hammondb37a3732000-08-14 04:47:33 +00004318 p2 = PyFile_FromFile(f2, cmdstring, m1, _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004319 PyFile_SetBufSize(p2, 0);
4320
4321 if (n != 4)
4322 CloseHandle(hChildStderrRdDup);
4323
Raymond Hettinger8ae46892003-10-12 19:09:37 +00004324 f = PyTuple_Pack(2,p1,p2);
Mark Hammond64aae662001-01-31 05:38:47 +00004325 Py_XDECREF(p1);
4326 Py_XDECREF(p2);
Mark Hammondb37a3732000-08-14 04:47:33 +00004327 file_count = 2;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004328 break;
4329 }
Tim Peters5aa91602002-01-30 05:46:57 +00004330
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004331 case POPEN_3:
4332 {
4333 char *m1, *m2;
4334 PyObject *p1, *p2, *p3;
Tim Peters5aa91602002-01-30 05:46:57 +00004335
Tim Peters7dca21e2002-08-19 00:42:29 +00004336 if (mode & _O_TEXT) {
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004337 m1 = "r";
4338 m2 = "w";
4339 } else {
4340 m1 = "rb";
4341 m2 = "wb";
4342 }
4343
4344 fd1 = _open_osfhandle((long)hChildStdinWrDup, mode);
4345 f1 = _fdopen(fd1, m2);
4346 fd2 = _open_osfhandle((long)hChildStdoutRdDup, mode);
4347 f2 = _fdopen(fd2, m1);
4348 fd3 = _open_osfhandle((long)hChildStderrRdDup, mode);
4349 f3 = _fdopen(fd3, m1);
Fredrik Lundh56055a42000-07-23 19:47:12 +00004350 p1 = PyFile_FromFile(f1, cmdstring, m2, _PyPclose);
Mark Hammondb37a3732000-08-14 04:47:33 +00004351 p2 = PyFile_FromFile(f2, cmdstring, m1, _PyPclose);
4352 p3 = PyFile_FromFile(f3, cmdstring, m1, _PyPclose);
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004353 PyFile_SetBufSize(p1, 0);
4354 PyFile_SetBufSize(p2, 0);
4355 PyFile_SetBufSize(p3, 0);
Raymond Hettinger8ae46892003-10-12 19:09:37 +00004356 f = PyTuple_Pack(3,p1,p2,p3);
Mark Hammond64aae662001-01-31 05:38:47 +00004357 Py_XDECREF(p1);
4358 Py_XDECREF(p2);
4359 Py_XDECREF(p3);
Mark Hammondb37a3732000-08-14 04:47:33 +00004360 file_count = 3;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004361 break;
4362 }
4363 }
4364
4365 if (n == POPEN_4) {
4366 if (!_PyPopenCreateProcess(cmdstring,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004367 hChildStdinRd,
4368 hChildStdoutWr,
Mark Hammondb37a3732000-08-14 04:47:33 +00004369 hChildStdoutWr,
4370 &hProcess))
Mark Hammond08501372001-01-31 07:30:29 +00004371 return NULL;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004372 }
4373 else {
4374 if (!_PyPopenCreateProcess(cmdstring,
Fredrik Lundh9ac81f62000-07-09 23:35:24 +00004375 hChildStdinRd,
4376 hChildStdoutWr,
Mark Hammondb37a3732000-08-14 04:47:33 +00004377 hChildStderrWr,
4378 &hProcess))
Mark Hammond08501372001-01-31 07:30:29 +00004379 return NULL;
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004380 }
4381
Mark Hammondb37a3732000-08-14 04:47:33 +00004382 /*
4383 * Insert the files we've created into the process dictionary
4384 * all referencing the list with the process handle and the
4385 * initial number of files (see description below in _PyPclose).
4386 * Since if _PyPclose later tried to wait on a process when all
4387 * handles weren't closed, it could create a deadlock with the
4388 * child, we spend some energy here to try to ensure that we
4389 * either insert all file handles into the dictionary or none
4390 * at all. It's a little clumsy with the various popen modes
4391 * and variable number of files involved.
4392 */
4393 if (!_PyPopenProcs) {
4394 _PyPopenProcs = PyDict_New();
4395 }
4396
4397 if (_PyPopenProcs) {
4398 PyObject *procObj, *hProcessObj, *intObj, *fileObj[3];
4399 int ins_rc[3];
4400
4401 fileObj[0] = fileObj[1] = fileObj[2] = NULL;
4402 ins_rc[0] = ins_rc[1] = ins_rc[2] = 0;
4403
4404 procObj = PyList_New(2);
4405 hProcessObj = PyLong_FromVoidPtr(hProcess);
4406 intObj = PyInt_FromLong(file_count);
4407
4408 if (procObj && hProcessObj && intObj) {
4409 PyList_SetItem(procObj,0,hProcessObj);
4410 PyList_SetItem(procObj,1,intObj);
4411
4412 fileObj[0] = PyLong_FromVoidPtr(f1);
4413 if (fileObj[0]) {
4414 ins_rc[0] = PyDict_SetItem(_PyPopenProcs,
4415 fileObj[0],
4416 procObj);
4417 }
4418 if (file_count >= 2) {
4419 fileObj[1] = PyLong_FromVoidPtr(f2);
4420 if (fileObj[1]) {
4421 ins_rc[1] = PyDict_SetItem(_PyPopenProcs,
4422 fileObj[1],
4423 procObj);
4424 }
4425 }
4426 if (file_count >= 3) {
4427 fileObj[2] = PyLong_FromVoidPtr(f3);
4428 if (fileObj[2]) {
4429 ins_rc[2] = PyDict_SetItem(_PyPopenProcs,
4430 fileObj[2],
4431 procObj);
4432 }
4433 }
4434
4435 if (ins_rc[0] < 0 || !fileObj[0] ||
4436 ins_rc[1] < 0 || (file_count > 1 && !fileObj[1]) ||
4437 ins_rc[2] < 0 || (file_count > 2 && !fileObj[2])) {
4438 /* Something failed - remove any dictionary
4439 * entries that did make it.
4440 */
4441 if (!ins_rc[0] && fileObj[0]) {
4442 PyDict_DelItem(_PyPopenProcs,
4443 fileObj[0]);
4444 }
4445 if (!ins_rc[1] && fileObj[1]) {
4446 PyDict_DelItem(_PyPopenProcs,
4447 fileObj[1]);
4448 }
4449 if (!ins_rc[2] && fileObj[2]) {
4450 PyDict_DelItem(_PyPopenProcs,
4451 fileObj[2]);
4452 }
4453 }
4454 }
Tim Peters5aa91602002-01-30 05:46:57 +00004455
Mark Hammondb37a3732000-08-14 04:47:33 +00004456 /*
4457 * Clean up our localized references for the dictionary keys
4458 * and value since PyDict_SetItem will Py_INCREF any copies
4459 * that got placed in the dictionary.
4460 */
4461 Py_XDECREF(procObj);
4462 Py_XDECREF(fileObj[0]);
4463 Py_XDECREF(fileObj[1]);
4464 Py_XDECREF(fileObj[2]);
4465 }
4466
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004467 /* Child is launched. Close the parents copy of those pipe
4468 * handles that only the child should have open. You need to
4469 * make sure that no handles to the write end of the output pipe
4470 * are maintained in this process or else the pipe will not close
4471 * when the child process exits and the ReadFile will hang. */
4472
4473 if (!CloseHandle(hChildStdinRd))
4474 return win32_error("CloseHandle", NULL);
Tim Peters5aa91602002-01-30 05:46:57 +00004475
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004476 if (!CloseHandle(hChildStdoutWr))
4477 return win32_error("CloseHandle", NULL);
Tim Peters5aa91602002-01-30 05:46:57 +00004478
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004479 if ((n != 4) && (!CloseHandle(hChildStderrWr)))
4480 return win32_error("CloseHandle", NULL);
4481
4482 return f;
4483}
Fredrik Lundh56055a42000-07-23 19:47:12 +00004484
4485/*
4486 * Wrapper for fclose() to use for popen* files, so we can retrieve the
4487 * exit code for the child process and return as a result of the close.
Mark Hammondb37a3732000-08-14 04:47:33 +00004488 *
4489 * This function uses the _PyPopenProcs dictionary in order to map the
4490 * input file pointer to information about the process that was
4491 * originally created by the popen* call that created the file pointer.
4492 * The dictionary uses the file pointer as a key (with one entry
4493 * inserted for each file returned by the original popen* call) and a
4494 * single list object as the value for all files from a single call.
4495 * The list object contains the Win32 process handle at [0], and a file
4496 * count at [1], which is initialized to the total number of file
4497 * handles using that list.
4498 *
4499 * This function closes whichever handle it is passed, and decrements
4500 * the file count in the dictionary for the process handle pointed to
4501 * by this file. On the last close (when the file count reaches zero),
4502 * this function will wait for the child process and then return its
4503 * exit code as the result of the close() operation. This permits the
4504 * files to be closed in any order - it is always the close() of the
4505 * final handle that will return the exit code.
Mark Hammond8d98d2c2003-04-19 15:41:53 +00004506 *
4507 * NOTE: This function is currently called with the GIL released.
4508 * hence we use the GILState API to manage our state.
Fredrik Lundh56055a42000-07-23 19:47:12 +00004509 */
Tim Peters736aa322000-09-01 06:51:24 +00004510
Fredrik Lundh56055a42000-07-23 19:47:12 +00004511static int _PyPclose(FILE *file)
4512{
Fredrik Lundh20318932000-07-26 17:29:12 +00004513 int result;
Fredrik Lundh56055a42000-07-23 19:47:12 +00004514 DWORD exit_code;
4515 HANDLE hProcess;
Mark Hammondb37a3732000-08-14 04:47:33 +00004516 PyObject *procObj, *hProcessObj, *intObj, *fileObj;
4517 long file_count;
Tim Peters736aa322000-09-01 06:51:24 +00004518#ifdef WITH_THREAD
Mark Hammond8d98d2c2003-04-19 15:41:53 +00004519 PyGILState_STATE state;
Tim Peters736aa322000-09-01 06:51:24 +00004520#endif
4521
Fredrik Lundh20318932000-07-26 17:29:12 +00004522 /* Close the file handle first, to ensure it can't block the
Mark Hammondb37a3732000-08-14 04:47:33 +00004523 * child from exiting if it's the last handle.
Fredrik Lundh20318932000-07-26 17:29:12 +00004524 */
4525 result = fclose(file);
Tim Peters736aa322000-09-01 06:51:24 +00004526#ifdef WITH_THREAD
Mark Hammond8d98d2c2003-04-19 15:41:53 +00004527 state = PyGILState_Ensure();
Tim Peters736aa322000-09-01 06:51:24 +00004528#endif
Fredrik Lundh56055a42000-07-23 19:47:12 +00004529 if (_PyPopenProcs) {
Mark Hammondb37a3732000-08-14 04:47:33 +00004530 if ((fileObj = PyLong_FromVoidPtr(file)) != NULL &&
4531 (procObj = PyDict_GetItem(_PyPopenProcs,
4532 fileObj)) != NULL &&
4533 (hProcessObj = PyList_GetItem(procObj,0)) != NULL &&
4534 (intObj = PyList_GetItem(procObj,1)) != NULL) {
4535
4536 hProcess = PyLong_AsVoidPtr(hProcessObj);
4537 file_count = PyInt_AsLong(intObj);
4538
4539 if (file_count > 1) {
4540 /* Still other files referencing process */
4541 file_count--;
4542 PyList_SetItem(procObj,1,
4543 PyInt_FromLong(file_count));
4544 } else {
4545 /* Last file for this process */
Fredrik Lundh20318932000-07-26 17:29:12 +00004546 if (result != EOF &&
4547 WaitForSingleObject(hProcess, INFINITE) != WAIT_FAILED &&
4548 GetExitCodeProcess(hProcess, &exit_code)) {
Fredrik Lundh56055a42000-07-23 19:47:12 +00004549 /* Possible truncation here in 16-bit environments, but
4550 * real exit codes are just the lower byte in any event.
4551 */
4552 result = exit_code;
Fredrik Lundh56055a42000-07-23 19:47:12 +00004553 } else {
Fredrik Lundh20318932000-07-26 17:29:12 +00004554 /* Indicate failure - this will cause the file object
4555 * to raise an I/O error and translate the last Win32
4556 * error code from errno. We do have a problem with
4557 * last errors that overlap the normal errno table,
4558 * but that's a consistent problem with the file object.
Fredrik Lundh56055a42000-07-23 19:47:12 +00004559 */
Fredrik Lundh20318932000-07-26 17:29:12 +00004560 if (result != EOF) {
4561 /* If the error wasn't from the fclose(), then
4562 * set errno for the file object error handling.
4563 */
4564 errno = GetLastError();
4565 }
4566 result = -1;
Fredrik Lundh56055a42000-07-23 19:47:12 +00004567 }
4568
4569 /* Free up the native handle at this point */
4570 CloseHandle(hProcess);
Mark Hammondb37a3732000-08-14 04:47:33 +00004571 }
Fredrik Lundh56055a42000-07-23 19:47:12 +00004572
Mark Hammondb37a3732000-08-14 04:47:33 +00004573 /* Remove this file pointer from dictionary */
4574 PyDict_DelItem(_PyPopenProcs, fileObj);
4575
4576 if (PyDict_Size(_PyPopenProcs) == 0) {
4577 Py_DECREF(_PyPopenProcs);
4578 _PyPopenProcs = NULL;
4579 }
4580
4581 } /* if object retrieval ok */
4582
4583 Py_XDECREF(fileObj);
Fredrik Lundh56055a42000-07-23 19:47:12 +00004584 } /* if _PyPopenProcs */
4585
Tim Peters736aa322000-09-01 06:51:24 +00004586#ifdef WITH_THREAD
Mark Hammond8d98d2c2003-04-19 15:41:53 +00004587 PyGILState_Release(state);
Tim Peters736aa322000-09-01 06:51:24 +00004588#endif
Fredrik Lundh56055a42000-07-23 19:47:12 +00004589 return result;
4590}
Tim Peters9acdd3a2000-09-01 19:26:36 +00004591
4592#else /* which OS? */
Barry Warsaw53699e91996-12-10 23:23:01 +00004593static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004594posix_popen(PyObject *self, PyObject *args)
Guido van Rossum3b066191991-06-04 19:40:25 +00004595{
Guido van Rossuma6a1e531995-01-10 15:36:38 +00004596 char *name;
4597 char *mode = "r";
4598 int bufsize = -1;
Guido van Rossum3b066191991-06-04 19:40:25 +00004599 FILE *fp;
Barry Warsaw53699e91996-12-10 23:23:01 +00004600 PyObject *f;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004601 if (!PyArg_ParseTuple(args, "s|si:popen", &name, &mode, &bufsize))
Guido van Rossum3b066191991-06-04 19:40:25 +00004602 return NULL;
Martin v. Löwis9ad853b2003-10-31 10:01:53 +00004603 /* Strip mode of binary or text modifiers */
4604 if (strcmp(mode, "rb") == 0 || strcmp(mode, "rt") == 0)
4605 mode = "r";
4606 else if (strcmp(mode, "wb") == 0 || strcmp(mode, "wt") == 0)
4607 mode = "w";
Barry Warsaw53699e91996-12-10 23:23:01 +00004608 Py_BEGIN_ALLOW_THREADS
Guido van Rossumef0a00e1992-01-27 16:51:30 +00004609 fp = popen(name, mode);
Barry Warsaw53699e91996-12-10 23:23:01 +00004610 Py_END_ALLOW_THREADS
Guido van Rossum3b066191991-06-04 19:40:25 +00004611 if (fp == NULL)
4612 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004613 f = PyFile_FromFile(fp, name, mode, pclose);
Guido van Rossuma6a1e531995-01-10 15:36:38 +00004614 if (f != NULL)
Barry Warsaw53699e91996-12-10 23:23:01 +00004615 PyFile_SetBufSize(f, bufsize);
Guido van Rossuma6a1e531995-01-10 15:36:38 +00004616 return f;
Guido van Rossum3b066191991-06-04 19:40:25 +00004617}
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004618
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004619#endif /* PYOS_??? */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00004620#endif /* HAVE_POPEN */
Guido van Rossum3b066191991-06-04 19:40:25 +00004621
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004622
Guido van Rossumb6775db1994-08-01 11:34:53 +00004623#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004624PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004625"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004626Set the current process's user id.");
4627
Barry Warsaw53699e91996-12-10 23:23:01 +00004628static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004629posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004630{
4631 int uid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004632 if (!PyArg_ParseTuple(args, "i:setuid", &uid))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004633 return NULL;
4634 if (setuid(uid) < 0)
4635 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004636 Py_INCREF(Py_None);
4637 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004638}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004639#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004640
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004641
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004642#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004643PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004644"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004645Set the current process's effective user id.");
4646
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004647static PyObject *
4648posix_seteuid (PyObject *self, PyObject *args)
4649{
4650 int euid;
4651 if (!PyArg_ParseTuple(args, "i", &euid)) {
4652 return NULL;
4653 } else if (seteuid(euid) < 0) {
4654 return posix_error();
4655 } else {
4656 Py_INCREF(Py_None);
4657 return Py_None;
4658 }
4659}
4660#endif /* HAVE_SETEUID */
4661
4662#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004663PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004664"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004665Set the current process's effective group id.");
4666
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004667static PyObject *
4668posix_setegid (PyObject *self, PyObject *args)
4669{
4670 int egid;
4671 if (!PyArg_ParseTuple(args, "i", &egid)) {
4672 return NULL;
4673 } else if (setegid(egid) < 0) {
4674 return posix_error();
4675 } else {
4676 Py_INCREF(Py_None);
4677 return Py_None;
4678 }
4679}
4680#endif /* HAVE_SETEGID */
4681
4682#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004683PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004684"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004685Set the current process's real and effective user ids.");
4686
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004687static PyObject *
4688posix_setreuid (PyObject *self, PyObject *args)
4689{
4690 int ruid, euid;
4691 if (!PyArg_ParseTuple(args, "ii", &ruid, &euid)) {
4692 return NULL;
4693 } else if (setreuid(ruid, euid) < 0) {
4694 return posix_error();
4695 } else {
4696 Py_INCREF(Py_None);
4697 return Py_None;
4698 }
4699}
4700#endif /* HAVE_SETREUID */
4701
4702#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004703PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004704"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004705Set the current process's real and effective group ids.");
4706
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004707static PyObject *
4708posix_setregid (PyObject *self, PyObject *args)
4709{
4710 int rgid, egid;
4711 if (!PyArg_ParseTuple(args, "ii", &rgid, &egid)) {
4712 return NULL;
4713 } else if (setregid(rgid, egid) < 0) {
4714 return posix_error();
4715 } else {
4716 Py_INCREF(Py_None);
4717 return Py_None;
4718 }
4719}
4720#endif /* HAVE_SETREGID */
4721
Guido van Rossumb6775db1994-08-01 11:34:53 +00004722#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004723PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004724"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004725Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004726
Barry Warsaw53699e91996-12-10 23:23:01 +00004727static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004728posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004729{
4730 int gid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004731 if (!PyArg_ParseTuple(args, "i:setgid", &gid))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004732 return NULL;
4733 if (setgid(gid) < 0)
4734 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004735 Py_INCREF(Py_None);
4736 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004737}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004738#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004739
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004740#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004741PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004742"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004743Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004744
4745static PyObject *
4746posix_setgroups(PyObject *self, PyObject *args)
4747{
4748 PyObject *groups;
4749 int i, len;
4750 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00004751
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004752 if (!PyArg_ParseTuple(args, "O:setgid", &groups))
4753 return NULL;
4754 if (!PySequence_Check(groups)) {
4755 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
4756 return NULL;
4757 }
4758 len = PySequence_Size(groups);
4759 if (len > MAX_GROUPS) {
4760 PyErr_SetString(PyExc_ValueError, "too many groups");
4761 return NULL;
4762 }
4763 for(i = 0; i < len; i++) {
4764 PyObject *elem;
4765 elem = PySequence_GetItem(groups, i);
4766 if (!elem)
4767 return NULL;
4768 if (!PyInt_Check(elem)) {
4769 PyErr_SetString(PyExc_TypeError,
4770 "groups must be integers");
4771 Py_DECREF(elem);
4772 return NULL;
4773 }
4774 /* XXX: check that value fits into gid_t. */
4775 grouplist[i] = PyInt_AsLong(elem);
4776 Py_DECREF(elem);
4777 }
4778
4779 if (setgroups(len, grouplist) < 0)
4780 return posix_error();
4781 Py_INCREF(Py_None);
4782 return Py_None;
4783}
4784#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004785
Guido van Rossumb6775db1994-08-01 11:34:53 +00004786#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004787PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004788"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004789Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004790
Barry Warsaw53699e91996-12-10 23:23:01 +00004791static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004792posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004793{
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004794 int pid, options;
4795#ifdef UNION_WAIT
4796 union wait status;
4797#define status_i (status.w_status)
4798#else
4799 int status;
4800#define status_i status
4801#endif
4802 status_i = 0;
4803
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004804 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
Guido van Rossum21803b81992-08-09 12:55:27 +00004805 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004806 Py_BEGIN_ALLOW_THREADS
Guido van Rossume6a3aa61999-02-01 16:15:30 +00004807 pid = waitpid(pid, &status, options);
Barry Warsaw53699e91996-12-10 23:23:01 +00004808 Py_END_ALLOW_THREADS
Guido van Rossum85e3b011991-06-03 12:42:10 +00004809 if (pid == -1)
4810 return posix_error();
Guido van Rossum21803b81992-08-09 12:55:27 +00004811 else
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004812 return Py_BuildValue("ii", pid, status_i);
Guido van Rossum21803b81992-08-09 12:55:27 +00004813}
4814
Tim Petersab034fa2002-02-01 11:27:43 +00004815#elif defined(HAVE_CWAIT)
4816
4817/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004818PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004819"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004820"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00004821
4822static PyObject *
4823posix_waitpid(PyObject *self, PyObject *args)
4824{
4825 int pid, options;
4826 int status;
4827
4828 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
4829 return NULL;
4830 Py_BEGIN_ALLOW_THREADS
4831 pid = _cwait(&status, pid, options);
4832 Py_END_ALLOW_THREADS
4833 if (pid == -1)
4834 return posix_error();
4835 else
4836 /* shift the status left a byte so this is more like the
4837 POSIX waitpid */
4838 return Py_BuildValue("ii", pid, status << 8);
4839}
4840#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004841
Guido van Rossumad0ee831995-03-01 10:34:45 +00004842#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004843PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004844"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004845Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004846
Barry Warsaw53699e91996-12-10 23:23:01 +00004847static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004848posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00004849{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004850 int pid;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004851#ifdef UNION_WAIT
4852 union wait status;
4853#define status_i (status.w_status)
Guido van Rossumb9f866c1997-05-22 15:12:39 +00004854#else
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004855 int status;
4856#define status_i status
Guido van Rossumb9f866c1997-05-22 15:12:39 +00004857#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00004858
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004859 status_i = 0;
4860 Py_BEGIN_ALLOW_THREADS
4861 pid = wait(&status);
Barry Warsaw53699e91996-12-10 23:23:01 +00004862 Py_END_ALLOW_THREADS
Guido van Rossum21803b81992-08-09 12:55:27 +00004863 if (pid == -1)
4864 return posix_error();
4865 else
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004866 return Py_BuildValue("ii", pid, status_i);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004867#undef status_i
Guido van Rossum85e3b011991-06-03 12:42:10 +00004868}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004869#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004870
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004871
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004872PyDoc_STRVAR(posix_lstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004873"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004874Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004875
Barry Warsaw53699e91996-12-10 23:23:01 +00004876static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004877posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004878{
Guido van Rossumb6775db1994-08-01 11:34:53 +00004879#ifdef HAVE_LSTAT
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004880 return posix_do_stat(self, args, "et:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004881#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004882#ifdef MS_WINDOWS
Mark Hammond7edd0a92003-08-06 02:46:58 +00004883 return posix_do_stat(self, args, "et:lstat", STAT, "U:lstat", _wstati64);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004884#else
4885 return posix_do_stat(self, args, "et:lstat", STAT, NULL, NULL);
4886#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00004887#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004888}
4889
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004890
Guido van Rossumb6775db1994-08-01 11:34:53 +00004891#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004892PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004893"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004894Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004895
Barry Warsaw53699e91996-12-10 23:23:01 +00004896static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004897posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004898{
Guido van Rossumb6775db1994-08-01 11:34:53 +00004899 char buf[MAXPATHLEN];
Guido van Rossumef0a00e1992-01-27 16:51:30 +00004900 char *path;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004901 int n;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004902 if (!PyArg_ParseTuple(args, "s:readlink", &path))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004903 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004904 Py_BEGIN_ALLOW_THREADS
Guido van Rossum50e61dc1992-03-27 17:22:31 +00004905 n = readlink(path, buf, (int) sizeof buf);
Barry Warsaw53699e91996-12-10 23:23:01 +00004906 Py_END_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004907 if (n < 0)
Barry Warsawd58d7641998-07-23 16:14:40 +00004908 return posix_error_with_filename(path);
Barry Warsaw53699e91996-12-10 23:23:01 +00004909 return PyString_FromStringAndSize(buf, n);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004910}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004911#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004912
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004913
Guido van Rossumb6775db1994-08-01 11:34:53 +00004914#ifdef HAVE_SYMLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004915PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004916"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00004917Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004918
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004919static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004920posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004921{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004922 return posix_2str(args, "etet:symlink", symlink, NULL, NULL);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004923}
4924#endif /* HAVE_SYMLINK */
4925
4926
4927#ifdef HAVE_TIMES
4928#ifndef HZ
4929#define HZ 60 /* Universal constant :-) */
4930#endif /* HZ */
Tim Peters5aa91602002-01-30 05:46:57 +00004931
Guido van Rossumd48f2521997-12-05 22:19:34 +00004932#if defined(PYCC_VACPP) && defined(PYOS_OS2)
4933static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00004934system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004935{
4936 ULONG value = 0;
4937
4938 Py_BEGIN_ALLOW_THREADS
4939 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
4940 Py_END_ALLOW_THREADS
4941
4942 return value;
4943}
4944
4945static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004946posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004947{
Guido van Rossumd48f2521997-12-05 22:19:34 +00004948 /* Currently Only Uptime is Provided -- Others Later */
4949 return Py_BuildValue("ddddd",
4950 (double)0 /* t.tms_utime / HZ */,
4951 (double)0 /* t.tms_stime / HZ */,
4952 (double)0 /* t.tms_cutime / HZ */,
4953 (double)0 /* t.tms_cstime / HZ */,
4954 (double)system_uptime() / 1000);
4955}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004956#else /* not OS2 */
Barry Warsaw53699e91996-12-10 23:23:01 +00004957static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004958posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00004959{
4960 struct tms t;
4961 clock_t c;
Guido van Rossum22db57e1992-04-05 14:25:30 +00004962 errno = 0;
4963 c = times(&t);
Guido van Rossum687dd131993-05-17 08:34:16 +00004964 if (c == (clock_t) -1)
4965 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004966 return Py_BuildValue("ddddd",
Barry Warsaw43d68b81996-12-19 22:10:44 +00004967 (double)t.tms_utime / HZ,
4968 (double)t.tms_stime / HZ,
4969 (double)t.tms_cutime / HZ,
4970 (double)t.tms_cstime / HZ,
4971 (double)c / HZ);
Guido van Rossum22db57e1992-04-05 14:25:30 +00004972}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004973#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00004974#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004975
4976
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004977#ifdef MS_WINDOWS
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004978#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00004979static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004980posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004981{
4982 FILETIME create, exit, kernel, user;
4983 HANDLE hProc;
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004984 hProc = GetCurrentProcess();
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004985 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
4986 /* The fields of a FILETIME structure are the hi and lo part
4987 of a 64-bit value expressed in 100 nanosecond units.
4988 1e7 is one second in such units; 1e-7 the inverse.
4989 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
4990 */
Barry Warsaw53699e91996-12-10 23:23:01 +00004991 return Py_BuildValue(
4992 "ddddd",
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004993 (double)(kernel.dwHighDateTime*429.4967296 +
4994 kernel.dwLowDateTime*1e-7),
4995 (double)(user.dwHighDateTime*429.4967296 +
4996 user.dwLowDateTime*1e-7),
Barry Warsaw53699e91996-12-10 23:23:01 +00004997 (double)0,
4998 (double)0,
4999 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00005000}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005001#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005002
5003#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005004PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005005"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005006Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005007#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00005008
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005009
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00005010#ifdef HAVE_GETSID
5011PyDoc_STRVAR(posix_getsid__doc__,
5012"getsid(pid) -> sid\n\n\
5013Call the system call getsid().");
5014
5015static PyObject *
5016posix_getsid(PyObject *self, PyObject *args)
5017{
5018 int pid, sid;
5019 if (!PyArg_ParseTuple(args, "i:getsid", &pid))
5020 return NULL;
5021 sid = getsid(pid);
5022 if (sid < 0)
5023 return posix_error();
5024 return PyInt_FromLong((long)sid);
5025}
5026#endif /* HAVE_GETSID */
5027
5028
Guido van Rossumb6775db1994-08-01 11:34:53 +00005029#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005030PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005031"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005032Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005033
Barry Warsaw53699e91996-12-10 23:23:01 +00005034static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005035posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00005036{
Guido van Rossum687dd131993-05-17 08:34:16 +00005037 if (setsid() < 0)
5038 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005039 Py_INCREF(Py_None);
5040 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00005041}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005042#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00005043
Guido van Rossumb6775db1994-08-01 11:34:53 +00005044#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005045PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005046"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005047Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005048
Barry Warsaw53699e91996-12-10 23:23:01 +00005049static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005050posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00005051{
5052 int pid, pgrp;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005053 if (!PyArg_ParseTuple(args, "ii:setpgid", &pid, &pgrp))
Guido van Rossumc2670a01992-09-13 20:07:29 +00005054 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00005055 if (setpgid(pid, pgrp) < 0)
5056 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005057 Py_INCREF(Py_None);
5058 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00005059}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005060#endif /* HAVE_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00005061
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005062
Guido van Rossumb6775db1994-08-01 11:34:53 +00005063#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005064PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005065"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005066Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005067
Barry Warsaw53699e91996-12-10 23:23:01 +00005068static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005069posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00005070{
5071 int fd, pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005072 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
Guido van Rossum7066dd71992-09-17 17:54:56 +00005073 return NULL;
5074 pgid = tcgetpgrp(fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00005075 if (pgid < 0)
5076 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005077 return PyInt_FromLong((long)pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00005078}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005079#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00005080
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005081
Guido van Rossumb6775db1994-08-01 11:34:53 +00005082#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005083PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005084"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005085Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005086
Barry Warsaw53699e91996-12-10 23:23:01 +00005087static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005088posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00005089{
5090 int fd, pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005091 if (!PyArg_ParseTuple(args, "ii:tcsetpgrp", &fd, &pgid))
Guido van Rossum7066dd71992-09-17 17:54:56 +00005092 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00005093 if (tcsetpgrp(fd, pgid) < 0)
5094 return posix_error();
Barry Warsaw43d68b81996-12-19 22:10:44 +00005095 Py_INCREF(Py_None);
Barry Warsaw53699e91996-12-10 23:23:01 +00005096 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00005097}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005098#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00005099
Guido van Rossum687dd131993-05-17 08:34:16 +00005100/* Functions acting on file descriptors */
5101
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005102PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005103"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005104Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005105
Barry Warsaw53699e91996-12-10 23:23:01 +00005106static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005107posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005108{
Mark Hammondef8b6542001-05-13 08:04:26 +00005109 char *file = NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00005110 int flag;
5111 int mode = 0777;
5112 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00005113
5114#ifdef MS_WINDOWS
5115 if (unicode_file_names()) {
5116 PyUnicodeObject *po;
5117 if (PyArg_ParseTuple(args, "Ui|i:mkdir", &po, &flag, &mode)) {
5118 Py_BEGIN_ALLOW_THREADS
5119 /* PyUnicode_AS_UNICODE OK without thread
5120 lock as it is a simple dereference. */
5121 fd = _wopen(PyUnicode_AS_UNICODE(po), flag, mode);
5122 Py_END_ALLOW_THREADS
5123 if (fd < 0)
5124 return posix_error();
5125 return PyInt_FromLong((long)fd);
5126 }
5127 /* Drop the argument parsing error as narrow strings
5128 are also valid. */
5129 PyErr_Clear();
5130 }
5131#endif
5132
Tim Peters5aa91602002-01-30 05:46:57 +00005133 if (!PyArg_ParseTuple(args, "eti|i",
Mark Hammondef8b6542001-05-13 08:04:26 +00005134 Py_FileSystemDefaultEncoding, &file,
5135 &flag, &mode))
Barry Warsaw43d68b81996-12-19 22:10:44 +00005136 return NULL;
5137
Barry Warsaw53699e91996-12-10 23:23:01 +00005138 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005139 fd = open(file, flag, mode);
Barry Warsaw53699e91996-12-10 23:23:01 +00005140 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005141 if (fd < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00005142 return posix_error_with_allocated_filename(file);
5143 PyMem_Free(file);
Barry Warsaw53699e91996-12-10 23:23:01 +00005144 return PyInt_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00005145}
5146
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005147
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005148PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005149"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005150Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005151
Barry Warsaw53699e91996-12-10 23:23:01 +00005152static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005153posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005154{
5155 int fd, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005156 if (!PyArg_ParseTuple(args, "i:close", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00005157 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005158 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005159 res = close(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00005160 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005161 if (res < 0)
5162 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005163 Py_INCREF(Py_None);
5164 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00005165}
5166
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005167
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005168PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005169"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005170Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005171
Barry Warsaw53699e91996-12-10 23:23:01 +00005172static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005173posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005174{
5175 int fd;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005176 if (!PyArg_ParseTuple(args, "i:dup", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00005177 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005178 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005179 fd = dup(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00005180 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005181 if (fd < 0)
5182 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005183 return PyInt_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00005184}
5185
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005186
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005187PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00005188"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005189Duplicate file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005190
Barry Warsaw53699e91996-12-10 23:23:01 +00005191static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005192posix_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005193{
5194 int fd, fd2, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005195 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
Guido van Rossum687dd131993-05-17 08:34:16 +00005196 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005197 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005198 res = dup2(fd, fd2);
Barry Warsaw53699e91996-12-10 23:23:01 +00005199 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005200 if (res < 0)
5201 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005202 Py_INCREF(Py_None);
5203 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00005204}
5205
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005206
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005207PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005208"lseek(fd, pos, how) -> newpos\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005209Set the current position of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005210
Barry Warsaw53699e91996-12-10 23:23:01 +00005211static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005212posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005213{
5214 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005215#if defined(MS_WIN64) || defined(MS_WINDOWS)
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005216 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00005217#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00005218 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00005219#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00005220 PyObject *posobj;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005221 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Guido van Rossum687dd131993-05-17 08:34:16 +00005222 return NULL;
5223#ifdef SEEK_SET
5224 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
5225 switch (how) {
5226 case 0: how = SEEK_SET; break;
5227 case 1: how = SEEK_CUR; break;
5228 case 2: how = SEEK_END; break;
5229 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005230#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00005231
5232#if !defined(HAVE_LARGEFILE_SUPPORT)
5233 pos = PyInt_AsLong(posobj);
5234#else
5235 pos = PyLong_Check(posobj) ?
5236 PyLong_AsLongLong(posobj) : PyInt_AsLong(posobj);
5237#endif
5238 if (PyErr_Occurred())
5239 return NULL;
5240
Barry Warsaw53699e91996-12-10 23:23:01 +00005241 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005242#if defined(MS_WIN64) || defined(MS_WINDOWS)
Fred Drake699f3522000-06-29 21:12:41 +00005243 res = _lseeki64(fd, pos, how);
5244#else
Guido van Rossum687dd131993-05-17 08:34:16 +00005245 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00005246#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00005247 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005248 if (res < 0)
5249 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00005250
5251#if !defined(HAVE_LARGEFILE_SUPPORT)
Barry Warsaw53699e91996-12-10 23:23:01 +00005252 return PyInt_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005253#else
5254 return PyLong_FromLongLong(res);
5255#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00005256}
5257
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005258
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005259PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005260"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005261Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005262
Barry Warsaw53699e91996-12-10 23:23:01 +00005263static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005264posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005265{
Guido van Rossum8bac5461996-06-11 18:38:48 +00005266 int fd, size, n;
Barry Warsaw53699e91996-12-10 23:23:01 +00005267 PyObject *buffer;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005268 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
Guido van Rossum687dd131993-05-17 08:34:16 +00005269 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005270 buffer = PyString_FromStringAndSize((char *)NULL, size);
Guido van Rossum687dd131993-05-17 08:34:16 +00005271 if (buffer == NULL)
5272 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005273 Py_BEGIN_ALLOW_THREADS
5274 n = read(fd, PyString_AsString(buffer), size);
5275 Py_END_ALLOW_THREADS
Guido van Rossum8bac5461996-06-11 18:38:48 +00005276 if (n < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00005277 Py_DECREF(buffer);
Guido van Rossum687dd131993-05-17 08:34:16 +00005278 return posix_error();
5279 }
Guido van Rossum8bac5461996-06-11 18:38:48 +00005280 if (n != size)
Barry Warsaw53699e91996-12-10 23:23:01 +00005281 _PyString_Resize(&buffer, n);
Guido van Rossum687dd131993-05-17 08:34:16 +00005282 return buffer;
5283}
5284
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005285
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005286PyDoc_STRVAR(posix_write__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005287"write(fd, string) -> byteswritten\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005288Write a string to a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005289
Barry Warsaw53699e91996-12-10 23:23:01 +00005290static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005291posix_write(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005292{
5293 int fd, size;
5294 char *buffer;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005295 if (!PyArg_ParseTuple(args, "is#:write", &fd, &buffer, &size))
Guido van Rossum687dd131993-05-17 08:34:16 +00005296 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005297 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005298 size = write(fd, buffer, size);
Barry Warsaw53699e91996-12-10 23:23:01 +00005299 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005300 if (size < 0)
5301 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005302 return PyInt_FromLong((long)size);
Guido van Rossum687dd131993-05-17 08:34:16 +00005303}
5304
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005305
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005306PyDoc_STRVAR(posix_fstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005307"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005308Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005309
Barry Warsaw53699e91996-12-10 23:23:01 +00005310static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005311posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005312{
5313 int fd;
Fred Drake699f3522000-06-29 21:12:41 +00005314 STRUCT_STAT st;
Guido van Rossum687dd131993-05-17 08:34:16 +00005315 int res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005316 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00005317 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00005318#ifdef __VMS
5319 /* on OpenVMS we must ensure that all bytes are written to the file */
5320 fsync(fd);
5321#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00005322 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00005323 res = FSTAT(fd, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00005324 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005325 if (res != 0)
5326 return posix_error();
Tim Peters5aa91602002-01-30 05:46:57 +00005327
Fred Drake699f3522000-06-29 21:12:41 +00005328 return _pystat_fromstructstat(st);
Guido van Rossum687dd131993-05-17 08:34:16 +00005329}
5330
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005331
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005332PyDoc_STRVAR(posix_fdopen__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005333"fdopen(fd [, mode='r' [, bufsize]]) -> file_object\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005334Return an open file object connected to a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005335
Barry Warsaw53699e91996-12-10 23:23:01 +00005336static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005337posix_fdopen(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005338{
Guido van Rossum687dd131993-05-17 08:34:16 +00005339 int fd;
Guido van Rossuma6a1e531995-01-10 15:36:38 +00005340 char *mode = "r";
5341 int bufsize = -1;
Guido van Rossum687dd131993-05-17 08:34:16 +00005342 FILE *fp;
Barry Warsaw53699e91996-12-10 23:23:01 +00005343 PyObject *f;
5344 if (!PyArg_ParseTuple(args, "i|si", &fd, &mode, &bufsize))
Guido van Rossum687dd131993-05-17 08:34:16 +00005345 return NULL;
Barry Warsaw43d68b81996-12-19 22:10:44 +00005346
Thomas Heller1f043e22002-11-07 16:00:59 +00005347 if (mode[0] != 'r' && mode[0] != 'w' && mode[0] != 'a') {
5348 PyErr_Format(PyExc_ValueError,
5349 "invalid file mode '%s'", mode);
5350 return NULL;
5351 }
5352
Barry Warsaw53699e91996-12-10 23:23:01 +00005353 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005354 fp = fdopen(fd, mode);
Barry Warsaw53699e91996-12-10 23:23:01 +00005355 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005356 if (fp == NULL)
5357 return posix_error();
Guido van Rossumbd6be7a2002-09-15 18:45:46 +00005358 f = PyFile_FromFile(fp, "<fdopen>", mode, fclose);
Guido van Rossuma6a1e531995-01-10 15:36:38 +00005359 if (f != NULL)
Barry Warsaw53699e91996-12-10 23:23:01 +00005360 PyFile_SetBufSize(f, bufsize);
Guido van Rossuma6a1e531995-01-10 15:36:38 +00005361 return f;
Guido van Rossum687dd131993-05-17 08:34:16 +00005362}
5363
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005364PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005365"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00005366Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005367connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00005368
5369static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00005370posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00005371{
5372 int fd;
5373 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
5374 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00005375 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00005376}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005377
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005378#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005379PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005380"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005381Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005382
Barry Warsaw53699e91996-12-10 23:23:01 +00005383static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005384posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00005385{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005386#if defined(PYOS_OS2)
5387 HFILE read, write;
5388 APIRET rc;
5389
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005390 Py_BEGIN_ALLOW_THREADS
5391 rc = DosCreatePipe( &read, &write, 4096);
5392 Py_END_ALLOW_THREADS
5393 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005394 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005395
5396 return Py_BuildValue("(ii)", read, write);
5397#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005398#if !defined(MS_WINDOWS)
Guido van Rossum687dd131993-05-17 08:34:16 +00005399 int fds[2];
5400 int res;
Barry Warsaw53699e91996-12-10 23:23:01 +00005401 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005402 res = pipe(fds);
Barry Warsaw53699e91996-12-10 23:23:01 +00005403 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005404 if (res != 0)
5405 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005406 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005407#else /* MS_WINDOWS */
Guido van Rossum794d8131994-08-23 13:48:48 +00005408 HANDLE read, write;
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00005409 int read_fd, write_fd;
Guido van Rossum794d8131994-08-23 13:48:48 +00005410 BOOL ok;
Barry Warsaw53699e91996-12-10 23:23:01 +00005411 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00005412 ok = CreatePipe(&read, &write, NULL, 0);
Barry Warsaw53699e91996-12-10 23:23:01 +00005413 Py_END_ALLOW_THREADS
Guido van Rossum794d8131994-08-23 13:48:48 +00005414 if (!ok)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005415 return win32_error("CreatePipe", NULL);
Tim Peters79248aa2001-08-29 21:37:10 +00005416 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
5417 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00005418 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005419#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005420#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00005421}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005422#endif /* HAVE_PIPE */
5423
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005424
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005425#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005426PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005427"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005428Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005429
Barry Warsaw53699e91996-12-10 23:23:01 +00005430static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005431posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005432{
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005433 char *filename;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005434 int mode = 0666;
5435 int res;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005436 if (!PyArg_ParseTuple(args, "s|i:mkfifo", &filename, &mode))
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005437 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005438 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005439 res = mkfifo(filename, mode);
5440 Py_END_ALLOW_THREADS
5441 if (res < 0)
5442 return posix_error();
5443 Py_INCREF(Py_None);
5444 return Py_None;
5445}
5446#endif
5447
5448
Neal Norwitz11690112002-07-30 01:08:28 +00005449#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005450PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005451"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005452Create a filesystem node (file, device special file or named pipe)\n\
5453named filename. mode specifies both the permissions to use and the\n\
5454type of node to be created, being combined (bitwise OR) with one of\n\
5455S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005456device defines the newly created device special file (probably using\n\
5457os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005458
5459
5460static PyObject *
5461posix_mknod(PyObject *self, PyObject *args)
5462{
5463 char *filename;
5464 int mode = 0600;
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005465 int device = 0;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005466 int res;
Martin v. Löwisd631ebe2002-11-02 17:42:33 +00005467 if (!PyArg_ParseTuple(args, "s|ii:mknod", &filename, &mode, &device))
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005468 return NULL;
5469 Py_BEGIN_ALLOW_THREADS
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005470 res = mknod(filename, mode, device);
Barry Warsaw53699e91996-12-10 23:23:01 +00005471 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005472 if (res < 0)
5473 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005474 Py_INCREF(Py_None);
5475 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005476}
5477#endif
5478
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005479#ifdef HAVE_DEVICE_MACROS
5480PyDoc_STRVAR(posix_major__doc__,
5481"major(device) -> major number\n\
5482Extracts a device major number from a raw device number.");
5483
5484static PyObject *
5485posix_major(PyObject *self, PyObject *args)
5486{
5487 int device;
5488 if (!PyArg_ParseTuple(args, "i:major", &device))
5489 return NULL;
5490 return PyInt_FromLong((long)major(device));
5491}
5492
5493PyDoc_STRVAR(posix_minor__doc__,
5494"minor(device) -> minor number\n\
5495Extracts a device minor number from a raw device number.");
5496
5497static PyObject *
5498posix_minor(PyObject *self, PyObject *args)
5499{
5500 int device;
5501 if (!PyArg_ParseTuple(args, "i:minor", &device))
5502 return NULL;
5503 return PyInt_FromLong((long)minor(device));
5504}
5505
5506PyDoc_STRVAR(posix_makedev__doc__,
5507"makedev(major, minor) -> device number\n\
5508Composes a raw device number from the major and minor device numbers.");
5509
5510static PyObject *
5511posix_makedev(PyObject *self, PyObject *args)
5512{
5513 int major, minor;
5514 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
5515 return NULL;
5516 return PyInt_FromLong((long)makedev(major, minor));
5517}
5518#endif /* device macros */
5519
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005520
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005521#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005522PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005523"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005524Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005525
Barry Warsaw53699e91996-12-10 23:23:01 +00005526static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005527posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005528{
5529 int fd;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005530 off_t length;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005531 int res;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005532 PyObject *lenobj;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005533
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005534 if (!PyArg_ParseTuple(args, "iO:ftruncate", &fd, &lenobj))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005535 return NULL;
5536
5537#if !defined(HAVE_LARGEFILE_SUPPORT)
5538 length = PyInt_AsLong(lenobj);
5539#else
5540 length = PyLong_Check(lenobj) ?
5541 PyLong_AsLongLong(lenobj) : PyInt_AsLong(lenobj);
5542#endif
5543 if (PyErr_Occurred())
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005544 return NULL;
5545
Barry Warsaw53699e91996-12-10 23:23:01 +00005546 Py_BEGIN_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005547 res = ftruncate(fd, length);
Barry Warsaw53699e91996-12-10 23:23:01 +00005548 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005549 if (res < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00005550 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005551 return NULL;
5552 }
Barry Warsaw53699e91996-12-10 23:23:01 +00005553 Py_INCREF(Py_None);
5554 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005555}
5556#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00005557
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005558#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005559PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005560"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005561Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005562
Fred Drake762e2061999-08-26 17:23:54 +00005563/* Save putenv() parameters as values here, so we can collect them when they
5564 * get re-set with another call for the same key. */
5565static PyObject *posix_putenv_garbage;
5566
Tim Peters5aa91602002-01-30 05:46:57 +00005567static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005568posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005569{
5570 char *s1, *s2;
5571 char *new;
Fred Drake762e2061999-08-26 17:23:54 +00005572 PyObject *newstr;
Tim Petersc8996f52001-12-03 20:41:00 +00005573 size_t len;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005574
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005575 if (!PyArg_ParseTuple(args, "ss:putenv", &s1, &s2))
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005576 return NULL;
Guido van Rossumd48f2521997-12-05 22:19:34 +00005577
5578#if defined(PYOS_OS2)
5579 if (stricmp(s1, "BEGINLIBPATH") == 0) {
5580 APIRET rc;
5581
Guido van Rossumd48f2521997-12-05 22:19:34 +00005582 rc = DosSetExtLIBPATH(s2, BEGIN_LIBPATH);
5583 if (rc != NO_ERROR)
5584 return os2_error(rc);
5585
5586 } else if (stricmp(s1, "ENDLIBPATH") == 0) {
5587 APIRET rc;
5588
Guido van Rossumd48f2521997-12-05 22:19:34 +00005589 rc = DosSetExtLIBPATH(s2, END_LIBPATH);
5590 if (rc != NO_ERROR)
5591 return os2_error(rc);
5592 } else {
5593#endif
5594
Fred Drake762e2061999-08-26 17:23:54 +00005595 /* XXX This can leak memory -- not easy to fix :-( */
Tim Petersc8996f52001-12-03 20:41:00 +00005596 len = strlen(s1) + strlen(s2) + 2;
5597 /* len includes space for a trailing \0; the size arg to
5598 PyString_FromStringAndSize does not count that */
5599 newstr = PyString_FromStringAndSize(NULL, (int)len - 1);
Fred Drake762e2061999-08-26 17:23:54 +00005600 if (newstr == NULL)
5601 return PyErr_NoMemory();
5602 new = PyString_AS_STRING(newstr);
Tim Petersc8996f52001-12-03 20:41:00 +00005603 PyOS_snprintf(new, len, "%s=%s", s1, s2);
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005604 if (putenv(new)) {
Neal Norwitz4adc9ab2003-02-10 03:10:43 +00005605 Py_DECREF(newstr);
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005606 posix_error();
5607 return NULL;
5608 }
Fred Drake762e2061999-08-26 17:23:54 +00005609 /* Install the first arg and newstr in posix_putenv_garbage;
5610 * this will cause previous value to be collected. This has to
5611 * happen after the real putenv() call because the old value
5612 * was still accessible until then. */
5613 if (PyDict_SetItem(posix_putenv_garbage,
5614 PyTuple_GET_ITEM(args, 0), newstr)) {
5615 /* really not much we can do; just leak */
5616 PyErr_Clear();
5617 }
5618 else {
5619 Py_DECREF(newstr);
5620 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00005621
5622#if defined(PYOS_OS2)
5623 }
5624#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00005625 Py_INCREF(Py_None);
5626 return Py_None;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005627}
Guido van Rossumb6a47161997-09-15 22:54:34 +00005628#endif /* putenv */
5629
Guido van Rossumc524d952001-10-19 01:31:59 +00005630#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005631PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005632"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005633Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00005634
5635static PyObject *
5636posix_unsetenv(PyObject *self, PyObject *args)
5637{
5638 char *s1;
5639
5640 if (!PyArg_ParseTuple(args, "s:unsetenv", &s1))
5641 return NULL;
5642
5643 unsetenv(s1);
5644
5645 /* Remove the key from posix_putenv_garbage;
5646 * this will cause it to be collected. This has to
Tim Peters5aa91602002-01-30 05:46:57 +00005647 * happen after the real unsetenv() call because the
Guido van Rossumc524d952001-10-19 01:31:59 +00005648 * old value was still accessible until then.
5649 */
5650 if (PyDict_DelItem(posix_putenv_garbage,
5651 PyTuple_GET_ITEM(args, 0))) {
5652 /* really not much we can do; just leak */
5653 PyErr_Clear();
5654 }
5655
5656 Py_INCREF(Py_None);
5657 return Py_None;
5658}
5659#endif /* unsetenv */
5660
Guido van Rossumb6a47161997-09-15 22:54:34 +00005661#ifdef HAVE_STRERROR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005662PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005663"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005664Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005665
Guido van Rossumf68d8e52001-04-14 17:55:09 +00005666static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005667posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00005668{
5669 int code;
5670 char *message;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005671 if (!PyArg_ParseTuple(args, "i:strerror", &code))
Guido van Rossumb6a47161997-09-15 22:54:34 +00005672 return NULL;
5673 message = strerror(code);
5674 if (message == NULL) {
5675 PyErr_SetString(PyExc_ValueError,
Fred Drake661ea262000-10-24 19:57:45 +00005676 "strerror() argument out of range");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005677 return NULL;
5678 }
5679 return PyString_FromString(message);
5680}
5681#endif /* strerror */
5682
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005683
Guido van Rossumc9641791998-08-04 15:26:23 +00005684#ifdef HAVE_SYS_WAIT_H
5685
Fred Drake106c1a02002-04-23 15:58:02 +00005686#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005687PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005688"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005689Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00005690
5691static PyObject *
5692posix_WCOREDUMP(PyObject *self, PyObject *args)
5693{
5694#ifdef UNION_WAIT
5695 union wait status;
5696#define status_i (status.w_status)
5697#else
5698 int status;
5699#define status_i status
5700#endif
5701 status_i = 0;
5702
5703 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &status_i))
5704 {
5705 return NULL;
5706 }
5707
5708 return PyBool_FromLong(WCOREDUMP(status));
5709#undef status_i
5710}
5711#endif /* WCOREDUMP */
5712
5713#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005714PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005715"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00005716Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005717job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00005718
5719static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005720posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00005721{
5722#ifdef UNION_WAIT
5723 union wait status;
5724#define status_i (status.w_status)
5725#else
5726 int status;
5727#define status_i status
5728#endif
5729 status_i = 0;
5730
5731 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &status_i))
5732 {
5733 return NULL;
5734 }
5735
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005736 return PyBool_FromLong(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00005737#undef status_i
5738}
5739#endif /* WIFCONTINUED */
5740
Guido van Rossumc9641791998-08-04 15:26:23 +00005741#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005742PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005743"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005744Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005745
5746static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005747posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005748{
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005749#ifdef UNION_WAIT
5750 union wait status;
5751#define status_i (status.w_status)
5752#else
5753 int status;
5754#define status_i status
5755#endif
5756 status_i = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005757
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005758 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &status_i))
Guido van Rossumc9641791998-08-04 15:26:23 +00005759 {
5760 return NULL;
5761 }
Tim Peters5aa91602002-01-30 05:46:57 +00005762
Fred Drake106c1a02002-04-23 15:58:02 +00005763 return PyBool_FromLong(WIFSTOPPED(status));
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005764#undef status_i
Guido van Rossumc9641791998-08-04 15:26:23 +00005765}
5766#endif /* WIFSTOPPED */
5767
5768#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005769PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005770"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005771Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005772
5773static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005774posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005775{
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005776#ifdef UNION_WAIT
5777 union wait status;
5778#define status_i (status.w_status)
5779#else
5780 int status;
5781#define status_i status
5782#endif
5783 status_i = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005784
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005785 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &status_i))
Guido van Rossumc9641791998-08-04 15:26:23 +00005786 {
5787 return NULL;
5788 }
Tim Peters5aa91602002-01-30 05:46:57 +00005789
Fred Drake106c1a02002-04-23 15:58:02 +00005790 return PyBool_FromLong(WIFSIGNALED(status));
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005791#undef status_i
Guido van Rossumc9641791998-08-04 15:26:23 +00005792}
5793#endif /* WIFSIGNALED */
5794
5795#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005796PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005797"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005798Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005799system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005800
5801static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005802posix_WIFEXITED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005803{
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005804#ifdef UNION_WAIT
5805 union wait status;
5806#define status_i (status.w_status)
5807#else
5808 int status;
5809#define status_i status
5810#endif
5811 status_i = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005812
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005813 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &status_i))
Guido van Rossumc9641791998-08-04 15:26:23 +00005814 {
5815 return NULL;
5816 }
Tim Peters5aa91602002-01-30 05:46:57 +00005817
Fred Drake106c1a02002-04-23 15:58:02 +00005818 return PyBool_FromLong(WIFEXITED(status));
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005819#undef status_i
Guido van Rossumc9641791998-08-04 15:26:23 +00005820}
5821#endif /* WIFEXITED */
5822
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005823#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005824PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005825"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005826Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005827
5828static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005829posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005830{
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005831#ifdef UNION_WAIT
5832 union wait status;
5833#define status_i (status.w_status)
5834#else
5835 int status;
5836#define status_i status
5837#endif
5838 status_i = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005839
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005840 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &status_i))
Guido van Rossumc9641791998-08-04 15:26:23 +00005841 {
5842 return NULL;
5843 }
Tim Peters5aa91602002-01-30 05:46:57 +00005844
Guido van Rossumc9641791998-08-04 15:26:23 +00005845 return Py_BuildValue("i", WEXITSTATUS(status));
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005846#undef status_i
Guido van Rossumc9641791998-08-04 15:26:23 +00005847}
5848#endif /* WEXITSTATUS */
5849
5850#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005851PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005852"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005853Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005854value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005855
5856static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005857posix_WTERMSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005858{
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005859#ifdef UNION_WAIT
5860 union wait status;
5861#define status_i (status.w_status)
5862#else
5863 int status;
5864#define status_i status
5865#endif
5866 status_i = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005867
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005868 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &status_i))
Guido van Rossumc9641791998-08-04 15:26:23 +00005869 {
5870 return NULL;
5871 }
Tim Peters5aa91602002-01-30 05:46:57 +00005872
Guido van Rossumc9641791998-08-04 15:26:23 +00005873 return Py_BuildValue("i", WTERMSIG(status));
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005874#undef status_i
Guido van Rossumc9641791998-08-04 15:26:23 +00005875}
5876#endif /* WTERMSIG */
5877
5878#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005879PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005880"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005881Return the signal that stopped the process that provided\n\
5882the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005883
5884static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005885posix_WSTOPSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005886{
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005887#ifdef UNION_WAIT
5888 union wait status;
5889#define status_i (status.w_status)
5890#else
5891 int status;
5892#define status_i status
5893#endif
5894 status_i = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005895
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005896 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &status_i))
Guido van Rossumc9641791998-08-04 15:26:23 +00005897 {
5898 return NULL;
5899 }
Tim Peters5aa91602002-01-30 05:46:57 +00005900
Guido van Rossumc9641791998-08-04 15:26:23 +00005901 return Py_BuildValue("i", WSTOPSIG(status));
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005902#undef status_i
Guido van Rossumc9641791998-08-04 15:26:23 +00005903}
5904#endif /* WSTOPSIG */
5905
5906#endif /* HAVE_SYS_WAIT_H */
5907
5908
Guido van Rossum94f6f721999-01-06 18:42:14 +00005909#if defined(HAVE_FSTATVFS)
Guido van Rossumd5753e11999-10-19 13:29:23 +00005910#ifdef _SCO_DS
5911/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
5912 needed definitions in sys/statvfs.h */
5913#define _SVID3
5914#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00005915#include <sys/statvfs.h>
5916
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005917static PyObject*
5918_pystatvfs_fromstructstatvfs(struct statvfs st) {
5919 PyObject *v = PyStructSequence_New(&StatVFSResultType);
5920 if (v == NULL)
5921 return NULL;
5922
5923#if !defined(HAVE_LARGEFILE_SUPPORT)
5924 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long) st.f_bsize));
5925 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long) st.f_frsize));
5926 PyStructSequence_SET_ITEM(v, 2, PyInt_FromLong((long) st.f_blocks));
5927 PyStructSequence_SET_ITEM(v, 3, PyInt_FromLong((long) st.f_bfree));
5928 PyStructSequence_SET_ITEM(v, 4, PyInt_FromLong((long) st.f_bavail));
5929 PyStructSequence_SET_ITEM(v, 5, PyInt_FromLong((long) st.f_files));
5930 PyStructSequence_SET_ITEM(v, 6, PyInt_FromLong((long) st.f_ffree));
5931 PyStructSequence_SET_ITEM(v, 7, PyInt_FromLong((long) st.f_favail));
5932 PyStructSequence_SET_ITEM(v, 8, PyInt_FromLong((long) st.f_flag));
5933 PyStructSequence_SET_ITEM(v, 9, PyInt_FromLong((long) st.f_namemax));
5934#else
5935 PyStructSequence_SET_ITEM(v, 0, PyInt_FromLong((long) st.f_bsize));
5936 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long) st.f_frsize));
Tim Peters5aa91602002-01-30 05:46:57 +00005937 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005938 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
Tim Peters5aa91602002-01-30 05:46:57 +00005939 PyStructSequence_SET_ITEM(v, 3,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005940 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005941 PyStructSequence_SET_ITEM(v, 4,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005942 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
Tim Peters5aa91602002-01-30 05:46:57 +00005943 PyStructSequence_SET_ITEM(v, 5,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005944 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
Tim Peters5aa91602002-01-30 05:46:57 +00005945 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005946 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
Tim Peters5aa91602002-01-30 05:46:57 +00005947 PyStructSequence_SET_ITEM(v, 7,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005948 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005949 PyStructSequence_SET_ITEM(v, 8, PyInt_FromLong((long) st.f_flag));
5950 PyStructSequence_SET_ITEM(v, 9, PyInt_FromLong((long) st.f_namemax));
5951#endif
5952
5953 return v;
5954}
5955
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005956PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005957"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005958Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005959
5960static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005961posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005962{
5963 int fd, res;
5964 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005965
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005966 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005967 return NULL;
5968 Py_BEGIN_ALLOW_THREADS
5969 res = fstatvfs(fd, &st);
5970 Py_END_ALLOW_THREADS
5971 if (res != 0)
5972 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005973
5974 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005975}
5976#endif /* HAVE_FSTATVFS */
5977
5978
5979#if defined(HAVE_STATVFS)
5980#include <sys/statvfs.h>
5981
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005982PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005983"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005984Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005985
5986static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005987posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005988{
5989 char *path;
5990 int res;
5991 struct statvfs st;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005992 if (!PyArg_ParseTuple(args, "s:statvfs", &path))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005993 return NULL;
5994 Py_BEGIN_ALLOW_THREADS
5995 res = statvfs(path, &st);
5996 Py_END_ALLOW_THREADS
5997 if (res != 0)
5998 return posix_error_with_filename(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005999
6000 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006001}
6002#endif /* HAVE_STATVFS */
6003
6004
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006005#ifdef HAVE_TEMPNAM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006006PyDoc_STRVAR(posix_tempnam__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006007"tempnam([dir[, prefix]]) -> string\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006008Return a unique name for a temporary file.\n\
Neal Norwitz50d5d4f2002-07-30 01:17:43 +00006009The directory and a prefix may be specified as strings; they may be omitted\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006010or None if not needed.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006011
6012static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006013posix_tempnam(PyObject *self, PyObject *args)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006014{
6015 PyObject *result = NULL;
6016 char *dir = NULL;
6017 char *pfx = NULL;
6018 char *name;
6019
6020 if (!PyArg_ParseTuple(args, "|zz:tempnam", &dir, &pfx))
6021 return NULL;
Skip Montanaro95618b52001-08-18 18:52:10 +00006022
6023 if (PyErr_Warn(PyExc_RuntimeWarning,
6024 "tempnam is a potential security risk to your program") < 0)
6025 return NULL;
6026
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006027#ifdef MS_WINDOWS
Fred Drake78b71c22001-07-17 20:37:36 +00006028 name = _tempnam(dir, pfx);
6029#else
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006030 name = tempnam(dir, pfx);
Fred Drake78b71c22001-07-17 20:37:36 +00006031#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006032 if (name == NULL)
6033 return PyErr_NoMemory();
6034 result = PyString_FromString(name);
6035 free(name);
6036 return result;
6037}
Guido van Rossumd371ff11999-01-25 16:12:23 +00006038#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006039
6040
6041#ifdef HAVE_TMPFILE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006042PyDoc_STRVAR(posix_tmpfile__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006043"tmpfile() -> file object\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006044Create a temporary file with no directory entries.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006045
6046static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006047posix_tmpfile(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006048{
6049 FILE *fp;
6050
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006051 fp = tmpfile();
6052 if (fp == NULL)
6053 return posix_error();
Guido van Rossumdb9198a2002-06-10 19:23:22 +00006054 return PyFile_FromFile(fp, "<tmpfile>", "w+b", fclose);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006055}
6056#endif
6057
6058
6059#ifdef HAVE_TMPNAM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006060PyDoc_STRVAR(posix_tmpnam__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006061"tmpnam() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006062Return a unique name for a temporary file.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006063
6064static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006065posix_tmpnam(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006066{
6067 char buffer[L_tmpnam];
6068 char *name;
6069
Skip Montanaro95618b52001-08-18 18:52:10 +00006070 if (PyErr_Warn(PyExc_RuntimeWarning,
6071 "tmpnam is a potential security risk to your program") < 0)
6072 return NULL;
6073
Greg Wardb48bc172000-03-01 21:51:56 +00006074#ifdef USE_TMPNAM_R
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006075 name = tmpnam_r(buffer);
6076#else
6077 name = tmpnam(buffer);
6078#endif
6079 if (name == NULL) {
6080 PyErr_SetObject(PyExc_OSError,
6081 Py_BuildValue("is", 0,
Greg Wardb48bc172000-03-01 21:51:56 +00006082#ifdef USE_TMPNAM_R
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006083 "unexpected NULL from tmpnam_r"
6084#else
6085 "unexpected NULL from tmpnam"
6086#endif
6087 ));
6088 return NULL;
6089 }
6090 return PyString_FromString(buffer);
6091}
6092#endif
6093
6094
Fred Drakec9680921999-12-13 16:37:25 +00006095/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
6096 * It maps strings representing configuration variable names to
6097 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00006098 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00006099 * rarely-used constants. There are three separate tables that use
6100 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00006101 *
6102 * This code is always included, even if none of the interfaces that
6103 * need it are included. The #if hackery needed to avoid it would be
6104 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00006105 */
6106struct constdef {
6107 char *name;
6108 long value;
6109};
6110
Fred Drake12c6e2d1999-12-14 21:25:03 +00006111static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006112conv_confname(PyObject *arg, int *valuep, struct constdef *table,
6113 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00006114{
6115 if (PyInt_Check(arg)) {
6116 *valuep = PyInt_AS_LONG(arg);
6117 return 1;
6118 }
6119 if (PyString_Check(arg)) {
6120 /* look up the value in the table using a binary search */
Fred Drake699f3522000-06-29 21:12:41 +00006121 size_t lo = 0;
6122 size_t mid;
6123 size_t hi = tablesize;
6124 int cmp;
Fred Drake12c6e2d1999-12-14 21:25:03 +00006125 char *confname = PyString_AS_STRING(arg);
6126 while (lo < hi) {
6127 mid = (lo + hi) / 2;
6128 cmp = strcmp(confname, table[mid].name);
6129 if (cmp < 0)
6130 hi = mid;
6131 else if (cmp > 0)
6132 lo = mid + 1;
6133 else {
6134 *valuep = table[mid].value;
6135 return 1;
6136 }
6137 }
6138 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
6139 }
6140 else
6141 PyErr_SetString(PyExc_TypeError,
6142 "configuration names must be strings or integers");
6143 return 0;
6144}
6145
6146
6147#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
6148static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00006149#ifdef _PC_ABI_AIO_XFER_MAX
6150 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
6151#endif
6152#ifdef _PC_ABI_ASYNC_IO
6153 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
6154#endif
Fred Drakec9680921999-12-13 16:37:25 +00006155#ifdef _PC_ASYNC_IO
6156 {"PC_ASYNC_IO", _PC_ASYNC_IO},
6157#endif
6158#ifdef _PC_CHOWN_RESTRICTED
6159 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
6160#endif
6161#ifdef _PC_FILESIZEBITS
6162 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
6163#endif
6164#ifdef _PC_LAST
6165 {"PC_LAST", _PC_LAST},
6166#endif
6167#ifdef _PC_LINK_MAX
6168 {"PC_LINK_MAX", _PC_LINK_MAX},
6169#endif
6170#ifdef _PC_MAX_CANON
6171 {"PC_MAX_CANON", _PC_MAX_CANON},
6172#endif
6173#ifdef _PC_MAX_INPUT
6174 {"PC_MAX_INPUT", _PC_MAX_INPUT},
6175#endif
6176#ifdef _PC_NAME_MAX
6177 {"PC_NAME_MAX", _PC_NAME_MAX},
6178#endif
6179#ifdef _PC_NO_TRUNC
6180 {"PC_NO_TRUNC", _PC_NO_TRUNC},
6181#endif
6182#ifdef _PC_PATH_MAX
6183 {"PC_PATH_MAX", _PC_PATH_MAX},
6184#endif
6185#ifdef _PC_PIPE_BUF
6186 {"PC_PIPE_BUF", _PC_PIPE_BUF},
6187#endif
6188#ifdef _PC_PRIO_IO
6189 {"PC_PRIO_IO", _PC_PRIO_IO},
6190#endif
6191#ifdef _PC_SOCK_MAXBUF
6192 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
6193#endif
6194#ifdef _PC_SYNC_IO
6195 {"PC_SYNC_IO", _PC_SYNC_IO},
6196#endif
6197#ifdef _PC_VDISABLE
6198 {"PC_VDISABLE", _PC_VDISABLE},
6199#endif
6200};
6201
Fred Drakec9680921999-12-13 16:37:25 +00006202static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006203conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00006204{
6205 return conv_confname(arg, valuep, posix_constants_pathconf,
6206 sizeof(posix_constants_pathconf)
6207 / sizeof(struct constdef));
6208}
6209#endif
6210
6211#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006212PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006213"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00006214Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006215If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00006216
6217static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006218posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006219{
6220 PyObject *result = NULL;
6221 int name, fd;
6222
Fred Drake12c6e2d1999-12-14 21:25:03 +00006223 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
6224 conv_path_confname, &name)) {
Fred Drakec9680921999-12-13 16:37:25 +00006225 long limit;
6226
6227 errno = 0;
6228 limit = fpathconf(fd, name);
6229 if (limit == -1 && errno != 0)
6230 posix_error();
6231 else
6232 result = PyInt_FromLong(limit);
6233 }
6234 return result;
6235}
6236#endif
6237
6238
6239#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006240PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006241"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00006242Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006243If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00006244
6245static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006246posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006247{
6248 PyObject *result = NULL;
6249 int name;
6250 char *path;
6251
6252 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
6253 conv_path_confname, &name)) {
6254 long limit;
6255
6256 errno = 0;
6257 limit = pathconf(path, name);
Fred Drake12c6e2d1999-12-14 21:25:03 +00006258 if (limit == -1 && errno != 0) {
Fred Drakec9680921999-12-13 16:37:25 +00006259 if (errno == EINVAL)
6260 /* could be a path or name problem */
6261 posix_error();
6262 else
6263 posix_error_with_filename(path);
Fred Drake12c6e2d1999-12-14 21:25:03 +00006264 }
Fred Drakec9680921999-12-13 16:37:25 +00006265 else
6266 result = PyInt_FromLong(limit);
6267 }
6268 return result;
6269}
6270#endif
6271
6272#ifdef HAVE_CONFSTR
6273static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00006274#ifdef _CS_ARCHITECTURE
6275 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
6276#endif
6277#ifdef _CS_HOSTNAME
6278 {"CS_HOSTNAME", _CS_HOSTNAME},
6279#endif
6280#ifdef _CS_HW_PROVIDER
6281 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
6282#endif
6283#ifdef _CS_HW_SERIAL
6284 {"CS_HW_SERIAL", _CS_HW_SERIAL},
6285#endif
6286#ifdef _CS_INITTAB_NAME
6287 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
6288#endif
Fred Drakec9680921999-12-13 16:37:25 +00006289#ifdef _CS_LFS64_CFLAGS
6290 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
6291#endif
6292#ifdef _CS_LFS64_LDFLAGS
6293 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
6294#endif
6295#ifdef _CS_LFS64_LIBS
6296 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
6297#endif
6298#ifdef _CS_LFS64_LINTFLAGS
6299 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
6300#endif
6301#ifdef _CS_LFS_CFLAGS
6302 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
6303#endif
6304#ifdef _CS_LFS_LDFLAGS
6305 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
6306#endif
6307#ifdef _CS_LFS_LIBS
6308 {"CS_LFS_LIBS", _CS_LFS_LIBS},
6309#endif
6310#ifdef _CS_LFS_LINTFLAGS
6311 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
6312#endif
Fred Draked86ed291999-12-15 15:34:33 +00006313#ifdef _CS_MACHINE
6314 {"CS_MACHINE", _CS_MACHINE},
6315#endif
Fred Drakec9680921999-12-13 16:37:25 +00006316#ifdef _CS_PATH
6317 {"CS_PATH", _CS_PATH},
6318#endif
Fred Draked86ed291999-12-15 15:34:33 +00006319#ifdef _CS_RELEASE
6320 {"CS_RELEASE", _CS_RELEASE},
6321#endif
6322#ifdef _CS_SRPC_DOMAIN
6323 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
6324#endif
6325#ifdef _CS_SYSNAME
6326 {"CS_SYSNAME", _CS_SYSNAME},
6327#endif
6328#ifdef _CS_VERSION
6329 {"CS_VERSION", _CS_VERSION},
6330#endif
Fred Drakec9680921999-12-13 16:37:25 +00006331#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
6332 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
6333#endif
6334#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
6335 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
6336#endif
6337#ifdef _CS_XBS5_ILP32_OFF32_LIBS
6338 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
6339#endif
6340#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
6341 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
6342#endif
6343#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
6344 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
6345#endif
6346#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
6347 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
6348#endif
6349#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
6350 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
6351#endif
6352#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
6353 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
6354#endif
6355#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
6356 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
6357#endif
6358#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
6359 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
6360#endif
6361#ifdef _CS_XBS5_LP64_OFF64_LIBS
6362 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
6363#endif
6364#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
6365 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
6366#endif
6367#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
6368 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
6369#endif
6370#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
6371 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
6372#endif
6373#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
6374 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
6375#endif
6376#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
6377 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
6378#endif
Fred Draked86ed291999-12-15 15:34:33 +00006379#ifdef _MIPS_CS_AVAIL_PROCESSORS
6380 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
6381#endif
6382#ifdef _MIPS_CS_BASE
6383 {"MIPS_CS_BASE", _MIPS_CS_BASE},
6384#endif
6385#ifdef _MIPS_CS_HOSTID
6386 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
6387#endif
6388#ifdef _MIPS_CS_HW_NAME
6389 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
6390#endif
6391#ifdef _MIPS_CS_NUM_PROCESSORS
6392 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
6393#endif
6394#ifdef _MIPS_CS_OSREL_MAJ
6395 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
6396#endif
6397#ifdef _MIPS_CS_OSREL_MIN
6398 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
6399#endif
6400#ifdef _MIPS_CS_OSREL_PATCH
6401 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
6402#endif
6403#ifdef _MIPS_CS_OS_NAME
6404 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
6405#endif
6406#ifdef _MIPS_CS_OS_PROVIDER
6407 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
6408#endif
6409#ifdef _MIPS_CS_PROCESSORS
6410 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
6411#endif
6412#ifdef _MIPS_CS_SERIAL
6413 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
6414#endif
6415#ifdef _MIPS_CS_VENDOR
6416 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
6417#endif
Fred Drakec9680921999-12-13 16:37:25 +00006418};
6419
6420static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006421conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00006422{
6423 return conv_confname(arg, valuep, posix_constants_confstr,
6424 sizeof(posix_constants_confstr)
6425 / sizeof(struct constdef));
6426}
6427
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006428PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006429"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006430Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00006431
6432static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006433posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006434{
6435 PyObject *result = NULL;
6436 int name;
6437 char buffer[64];
6438
6439 if (PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name)) {
6440 int len = confstr(name, buffer, sizeof(buffer));
6441
Fred Drakec9680921999-12-13 16:37:25 +00006442 errno = 0;
6443 if (len == 0) {
6444 if (errno != 0)
6445 posix_error();
6446 else
6447 result = PyString_FromString("");
6448 }
6449 else {
6450 if (len >= sizeof(buffer)) {
6451 result = PyString_FromStringAndSize(NULL, len);
6452 if (result != NULL)
6453 confstr(name, PyString_AS_STRING(result), len+1);
6454 }
6455 else
6456 result = PyString_FromString(buffer);
6457 }
6458 }
6459 return result;
6460}
6461#endif
6462
6463
6464#ifdef HAVE_SYSCONF
6465static struct constdef posix_constants_sysconf[] = {
6466#ifdef _SC_2_CHAR_TERM
6467 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
6468#endif
6469#ifdef _SC_2_C_BIND
6470 {"SC_2_C_BIND", _SC_2_C_BIND},
6471#endif
6472#ifdef _SC_2_C_DEV
6473 {"SC_2_C_DEV", _SC_2_C_DEV},
6474#endif
6475#ifdef _SC_2_C_VERSION
6476 {"SC_2_C_VERSION", _SC_2_C_VERSION},
6477#endif
6478#ifdef _SC_2_FORT_DEV
6479 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
6480#endif
6481#ifdef _SC_2_FORT_RUN
6482 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
6483#endif
6484#ifdef _SC_2_LOCALEDEF
6485 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
6486#endif
6487#ifdef _SC_2_SW_DEV
6488 {"SC_2_SW_DEV", _SC_2_SW_DEV},
6489#endif
6490#ifdef _SC_2_UPE
6491 {"SC_2_UPE", _SC_2_UPE},
6492#endif
6493#ifdef _SC_2_VERSION
6494 {"SC_2_VERSION", _SC_2_VERSION},
6495#endif
Fred Draked86ed291999-12-15 15:34:33 +00006496#ifdef _SC_ABI_ASYNCHRONOUS_IO
6497 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
6498#endif
6499#ifdef _SC_ACL
6500 {"SC_ACL", _SC_ACL},
6501#endif
Fred Drakec9680921999-12-13 16:37:25 +00006502#ifdef _SC_AIO_LISTIO_MAX
6503 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
6504#endif
Fred Drakec9680921999-12-13 16:37:25 +00006505#ifdef _SC_AIO_MAX
6506 {"SC_AIO_MAX", _SC_AIO_MAX},
6507#endif
6508#ifdef _SC_AIO_PRIO_DELTA_MAX
6509 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
6510#endif
6511#ifdef _SC_ARG_MAX
6512 {"SC_ARG_MAX", _SC_ARG_MAX},
6513#endif
6514#ifdef _SC_ASYNCHRONOUS_IO
6515 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
6516#endif
6517#ifdef _SC_ATEXIT_MAX
6518 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
6519#endif
Fred Draked86ed291999-12-15 15:34:33 +00006520#ifdef _SC_AUDIT
6521 {"SC_AUDIT", _SC_AUDIT},
6522#endif
Fred Drakec9680921999-12-13 16:37:25 +00006523#ifdef _SC_AVPHYS_PAGES
6524 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
6525#endif
6526#ifdef _SC_BC_BASE_MAX
6527 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
6528#endif
6529#ifdef _SC_BC_DIM_MAX
6530 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
6531#endif
6532#ifdef _SC_BC_SCALE_MAX
6533 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
6534#endif
6535#ifdef _SC_BC_STRING_MAX
6536 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
6537#endif
Fred Draked86ed291999-12-15 15:34:33 +00006538#ifdef _SC_CAP
6539 {"SC_CAP", _SC_CAP},
6540#endif
Fred Drakec9680921999-12-13 16:37:25 +00006541#ifdef _SC_CHARCLASS_NAME_MAX
6542 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
6543#endif
6544#ifdef _SC_CHAR_BIT
6545 {"SC_CHAR_BIT", _SC_CHAR_BIT},
6546#endif
6547#ifdef _SC_CHAR_MAX
6548 {"SC_CHAR_MAX", _SC_CHAR_MAX},
6549#endif
6550#ifdef _SC_CHAR_MIN
6551 {"SC_CHAR_MIN", _SC_CHAR_MIN},
6552#endif
6553#ifdef _SC_CHILD_MAX
6554 {"SC_CHILD_MAX", _SC_CHILD_MAX},
6555#endif
6556#ifdef _SC_CLK_TCK
6557 {"SC_CLK_TCK", _SC_CLK_TCK},
6558#endif
6559#ifdef _SC_COHER_BLKSZ
6560 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
6561#endif
6562#ifdef _SC_COLL_WEIGHTS_MAX
6563 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
6564#endif
6565#ifdef _SC_DCACHE_ASSOC
6566 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
6567#endif
6568#ifdef _SC_DCACHE_BLKSZ
6569 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
6570#endif
6571#ifdef _SC_DCACHE_LINESZ
6572 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
6573#endif
6574#ifdef _SC_DCACHE_SZ
6575 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
6576#endif
6577#ifdef _SC_DCACHE_TBLKSZ
6578 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
6579#endif
6580#ifdef _SC_DELAYTIMER_MAX
6581 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
6582#endif
6583#ifdef _SC_EQUIV_CLASS_MAX
6584 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
6585#endif
6586#ifdef _SC_EXPR_NEST_MAX
6587 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
6588#endif
6589#ifdef _SC_FSYNC
6590 {"SC_FSYNC", _SC_FSYNC},
6591#endif
6592#ifdef _SC_GETGR_R_SIZE_MAX
6593 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
6594#endif
6595#ifdef _SC_GETPW_R_SIZE_MAX
6596 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
6597#endif
6598#ifdef _SC_ICACHE_ASSOC
6599 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
6600#endif
6601#ifdef _SC_ICACHE_BLKSZ
6602 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
6603#endif
6604#ifdef _SC_ICACHE_LINESZ
6605 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
6606#endif
6607#ifdef _SC_ICACHE_SZ
6608 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
6609#endif
Fred Draked86ed291999-12-15 15:34:33 +00006610#ifdef _SC_INF
6611 {"SC_INF", _SC_INF},
6612#endif
Fred Drakec9680921999-12-13 16:37:25 +00006613#ifdef _SC_INT_MAX
6614 {"SC_INT_MAX", _SC_INT_MAX},
6615#endif
6616#ifdef _SC_INT_MIN
6617 {"SC_INT_MIN", _SC_INT_MIN},
6618#endif
6619#ifdef _SC_IOV_MAX
6620 {"SC_IOV_MAX", _SC_IOV_MAX},
6621#endif
Fred Draked86ed291999-12-15 15:34:33 +00006622#ifdef _SC_IP_SECOPTS
6623 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
6624#endif
Fred Drakec9680921999-12-13 16:37:25 +00006625#ifdef _SC_JOB_CONTROL
6626 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
6627#endif
Fred Draked86ed291999-12-15 15:34:33 +00006628#ifdef _SC_KERN_POINTERS
6629 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
6630#endif
6631#ifdef _SC_KERN_SIM
6632 {"SC_KERN_SIM", _SC_KERN_SIM},
6633#endif
Fred Drakec9680921999-12-13 16:37:25 +00006634#ifdef _SC_LINE_MAX
6635 {"SC_LINE_MAX", _SC_LINE_MAX},
6636#endif
6637#ifdef _SC_LOGIN_NAME_MAX
6638 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
6639#endif
6640#ifdef _SC_LOGNAME_MAX
6641 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
6642#endif
6643#ifdef _SC_LONG_BIT
6644 {"SC_LONG_BIT", _SC_LONG_BIT},
6645#endif
Fred Draked86ed291999-12-15 15:34:33 +00006646#ifdef _SC_MAC
6647 {"SC_MAC", _SC_MAC},
6648#endif
Fred Drakec9680921999-12-13 16:37:25 +00006649#ifdef _SC_MAPPED_FILES
6650 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
6651#endif
6652#ifdef _SC_MAXPID
6653 {"SC_MAXPID", _SC_MAXPID},
6654#endif
6655#ifdef _SC_MB_LEN_MAX
6656 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
6657#endif
6658#ifdef _SC_MEMLOCK
6659 {"SC_MEMLOCK", _SC_MEMLOCK},
6660#endif
6661#ifdef _SC_MEMLOCK_RANGE
6662 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
6663#endif
6664#ifdef _SC_MEMORY_PROTECTION
6665 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
6666#endif
6667#ifdef _SC_MESSAGE_PASSING
6668 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
6669#endif
Fred Draked86ed291999-12-15 15:34:33 +00006670#ifdef _SC_MMAP_FIXED_ALIGNMENT
6671 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
6672#endif
Fred Drakec9680921999-12-13 16:37:25 +00006673#ifdef _SC_MQ_OPEN_MAX
6674 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
6675#endif
6676#ifdef _SC_MQ_PRIO_MAX
6677 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
6678#endif
Fred Draked86ed291999-12-15 15:34:33 +00006679#ifdef _SC_NACLS_MAX
6680 {"SC_NACLS_MAX", _SC_NACLS_MAX},
6681#endif
Fred Drakec9680921999-12-13 16:37:25 +00006682#ifdef _SC_NGROUPS_MAX
6683 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
6684#endif
6685#ifdef _SC_NL_ARGMAX
6686 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
6687#endif
6688#ifdef _SC_NL_LANGMAX
6689 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
6690#endif
6691#ifdef _SC_NL_MSGMAX
6692 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
6693#endif
6694#ifdef _SC_NL_NMAX
6695 {"SC_NL_NMAX", _SC_NL_NMAX},
6696#endif
6697#ifdef _SC_NL_SETMAX
6698 {"SC_NL_SETMAX", _SC_NL_SETMAX},
6699#endif
6700#ifdef _SC_NL_TEXTMAX
6701 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
6702#endif
6703#ifdef _SC_NPROCESSORS_CONF
6704 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
6705#endif
6706#ifdef _SC_NPROCESSORS_ONLN
6707 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
6708#endif
Fred Draked86ed291999-12-15 15:34:33 +00006709#ifdef _SC_NPROC_CONF
6710 {"SC_NPROC_CONF", _SC_NPROC_CONF},
6711#endif
6712#ifdef _SC_NPROC_ONLN
6713 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
6714#endif
Fred Drakec9680921999-12-13 16:37:25 +00006715#ifdef _SC_NZERO
6716 {"SC_NZERO", _SC_NZERO},
6717#endif
6718#ifdef _SC_OPEN_MAX
6719 {"SC_OPEN_MAX", _SC_OPEN_MAX},
6720#endif
6721#ifdef _SC_PAGESIZE
6722 {"SC_PAGESIZE", _SC_PAGESIZE},
6723#endif
6724#ifdef _SC_PAGE_SIZE
6725 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
6726#endif
6727#ifdef _SC_PASS_MAX
6728 {"SC_PASS_MAX", _SC_PASS_MAX},
6729#endif
6730#ifdef _SC_PHYS_PAGES
6731 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
6732#endif
6733#ifdef _SC_PII
6734 {"SC_PII", _SC_PII},
6735#endif
6736#ifdef _SC_PII_INTERNET
6737 {"SC_PII_INTERNET", _SC_PII_INTERNET},
6738#endif
6739#ifdef _SC_PII_INTERNET_DGRAM
6740 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
6741#endif
6742#ifdef _SC_PII_INTERNET_STREAM
6743 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
6744#endif
6745#ifdef _SC_PII_OSI
6746 {"SC_PII_OSI", _SC_PII_OSI},
6747#endif
6748#ifdef _SC_PII_OSI_CLTS
6749 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
6750#endif
6751#ifdef _SC_PII_OSI_COTS
6752 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
6753#endif
6754#ifdef _SC_PII_OSI_M
6755 {"SC_PII_OSI_M", _SC_PII_OSI_M},
6756#endif
6757#ifdef _SC_PII_SOCKET
6758 {"SC_PII_SOCKET", _SC_PII_SOCKET},
6759#endif
6760#ifdef _SC_PII_XTI
6761 {"SC_PII_XTI", _SC_PII_XTI},
6762#endif
6763#ifdef _SC_POLL
6764 {"SC_POLL", _SC_POLL},
6765#endif
6766#ifdef _SC_PRIORITIZED_IO
6767 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
6768#endif
6769#ifdef _SC_PRIORITY_SCHEDULING
6770 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
6771#endif
6772#ifdef _SC_REALTIME_SIGNALS
6773 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
6774#endif
6775#ifdef _SC_RE_DUP_MAX
6776 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
6777#endif
6778#ifdef _SC_RTSIG_MAX
6779 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
6780#endif
6781#ifdef _SC_SAVED_IDS
6782 {"SC_SAVED_IDS", _SC_SAVED_IDS},
6783#endif
6784#ifdef _SC_SCHAR_MAX
6785 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
6786#endif
6787#ifdef _SC_SCHAR_MIN
6788 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
6789#endif
6790#ifdef _SC_SELECT
6791 {"SC_SELECT", _SC_SELECT},
6792#endif
6793#ifdef _SC_SEMAPHORES
6794 {"SC_SEMAPHORES", _SC_SEMAPHORES},
6795#endif
6796#ifdef _SC_SEM_NSEMS_MAX
6797 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
6798#endif
6799#ifdef _SC_SEM_VALUE_MAX
6800 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
6801#endif
6802#ifdef _SC_SHARED_MEMORY_OBJECTS
6803 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
6804#endif
6805#ifdef _SC_SHRT_MAX
6806 {"SC_SHRT_MAX", _SC_SHRT_MAX},
6807#endif
6808#ifdef _SC_SHRT_MIN
6809 {"SC_SHRT_MIN", _SC_SHRT_MIN},
6810#endif
6811#ifdef _SC_SIGQUEUE_MAX
6812 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
6813#endif
6814#ifdef _SC_SIGRT_MAX
6815 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
6816#endif
6817#ifdef _SC_SIGRT_MIN
6818 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
6819#endif
Fred Draked86ed291999-12-15 15:34:33 +00006820#ifdef _SC_SOFTPOWER
6821 {"SC_SOFTPOWER", _SC_SOFTPOWER},
6822#endif
Fred Drakec9680921999-12-13 16:37:25 +00006823#ifdef _SC_SPLIT_CACHE
6824 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
6825#endif
6826#ifdef _SC_SSIZE_MAX
6827 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
6828#endif
6829#ifdef _SC_STACK_PROT
6830 {"SC_STACK_PROT", _SC_STACK_PROT},
6831#endif
6832#ifdef _SC_STREAM_MAX
6833 {"SC_STREAM_MAX", _SC_STREAM_MAX},
6834#endif
6835#ifdef _SC_SYNCHRONIZED_IO
6836 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
6837#endif
6838#ifdef _SC_THREADS
6839 {"SC_THREADS", _SC_THREADS},
6840#endif
6841#ifdef _SC_THREAD_ATTR_STACKADDR
6842 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
6843#endif
6844#ifdef _SC_THREAD_ATTR_STACKSIZE
6845 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
6846#endif
6847#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
6848 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
6849#endif
6850#ifdef _SC_THREAD_KEYS_MAX
6851 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
6852#endif
6853#ifdef _SC_THREAD_PRIORITY_SCHEDULING
6854 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
6855#endif
6856#ifdef _SC_THREAD_PRIO_INHERIT
6857 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
6858#endif
6859#ifdef _SC_THREAD_PRIO_PROTECT
6860 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
6861#endif
6862#ifdef _SC_THREAD_PROCESS_SHARED
6863 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
6864#endif
6865#ifdef _SC_THREAD_SAFE_FUNCTIONS
6866 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
6867#endif
6868#ifdef _SC_THREAD_STACK_MIN
6869 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
6870#endif
6871#ifdef _SC_THREAD_THREADS_MAX
6872 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
6873#endif
6874#ifdef _SC_TIMERS
6875 {"SC_TIMERS", _SC_TIMERS},
6876#endif
6877#ifdef _SC_TIMER_MAX
6878 {"SC_TIMER_MAX", _SC_TIMER_MAX},
6879#endif
6880#ifdef _SC_TTY_NAME_MAX
6881 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
6882#endif
6883#ifdef _SC_TZNAME_MAX
6884 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
6885#endif
6886#ifdef _SC_T_IOV_MAX
6887 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
6888#endif
6889#ifdef _SC_UCHAR_MAX
6890 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
6891#endif
6892#ifdef _SC_UINT_MAX
6893 {"SC_UINT_MAX", _SC_UINT_MAX},
6894#endif
6895#ifdef _SC_UIO_MAXIOV
6896 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
6897#endif
6898#ifdef _SC_ULONG_MAX
6899 {"SC_ULONG_MAX", _SC_ULONG_MAX},
6900#endif
6901#ifdef _SC_USHRT_MAX
6902 {"SC_USHRT_MAX", _SC_USHRT_MAX},
6903#endif
6904#ifdef _SC_VERSION
6905 {"SC_VERSION", _SC_VERSION},
6906#endif
6907#ifdef _SC_WORD_BIT
6908 {"SC_WORD_BIT", _SC_WORD_BIT},
6909#endif
6910#ifdef _SC_XBS5_ILP32_OFF32
6911 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
6912#endif
6913#ifdef _SC_XBS5_ILP32_OFFBIG
6914 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
6915#endif
6916#ifdef _SC_XBS5_LP64_OFF64
6917 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
6918#endif
6919#ifdef _SC_XBS5_LPBIG_OFFBIG
6920 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
6921#endif
6922#ifdef _SC_XOPEN_CRYPT
6923 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
6924#endif
6925#ifdef _SC_XOPEN_ENH_I18N
6926 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
6927#endif
6928#ifdef _SC_XOPEN_LEGACY
6929 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
6930#endif
6931#ifdef _SC_XOPEN_REALTIME
6932 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
6933#endif
6934#ifdef _SC_XOPEN_REALTIME_THREADS
6935 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
6936#endif
6937#ifdef _SC_XOPEN_SHM
6938 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
6939#endif
6940#ifdef _SC_XOPEN_UNIX
6941 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
6942#endif
6943#ifdef _SC_XOPEN_VERSION
6944 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
6945#endif
6946#ifdef _SC_XOPEN_XCU_VERSION
6947 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
6948#endif
6949#ifdef _SC_XOPEN_XPG2
6950 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
6951#endif
6952#ifdef _SC_XOPEN_XPG3
6953 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
6954#endif
6955#ifdef _SC_XOPEN_XPG4
6956 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
6957#endif
6958};
6959
6960static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006961conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00006962{
6963 return conv_confname(arg, valuep, posix_constants_sysconf,
6964 sizeof(posix_constants_sysconf)
6965 / sizeof(struct constdef));
6966}
6967
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006968PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006969"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006970Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00006971
6972static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006973posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006974{
6975 PyObject *result = NULL;
6976 int name;
6977
6978 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
6979 int value;
6980
6981 errno = 0;
6982 value = sysconf(name);
6983 if (value == -1 && errno != 0)
6984 posix_error();
6985 else
6986 result = PyInt_FromLong(value);
6987 }
6988 return result;
6989}
6990#endif
6991
6992
Fred Drakebec628d1999-12-15 18:31:10 +00006993/* This code is used to ensure that the tables of configuration value names
6994 * are in sorted order as required by conv_confname(), and also to build the
6995 * the exported dictionaries that are used to publish information about the
6996 * names available on the host platform.
6997 *
6998 * Sorting the table at runtime ensures that the table is properly ordered
6999 * when used, even for platforms we're not able to test on. It also makes
7000 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00007001 */
Fred Drakebec628d1999-12-15 18:31:10 +00007002
7003static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007004cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00007005{
7006 const struct constdef *c1 =
7007 (const struct constdef *) v1;
7008 const struct constdef *c2 =
7009 (const struct constdef *) v2;
7010
7011 return strcmp(c1->name, c2->name);
7012}
7013
7014static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007015setup_confname_table(struct constdef *table, size_t tablesize,
Fred Drake4d1e64b2002-04-15 19:40:07 +00007016 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00007017{
Fred Drakebec628d1999-12-15 18:31:10 +00007018 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00007019 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00007020
7021 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
7022 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00007023 if (d == NULL)
7024 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00007025
Barry Warsaw3155db32000-04-13 15:20:40 +00007026 for (i=0; i < tablesize; ++i) {
7027 PyObject *o = PyInt_FromLong(table[i].value);
7028 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
7029 Py_XDECREF(o);
7030 Py_DECREF(d);
7031 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00007032 }
Barry Warsaw3155db32000-04-13 15:20:40 +00007033 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00007034 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00007035 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00007036}
7037
Fred Drakebec628d1999-12-15 18:31:10 +00007038/* Return -1 on failure, 0 on success. */
7039static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00007040setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00007041{
7042#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00007043 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00007044 sizeof(posix_constants_pathconf)
7045 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00007046 "pathconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00007047 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00007048#endif
7049#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00007050 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00007051 sizeof(posix_constants_confstr)
7052 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00007053 "confstr_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00007054 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00007055#endif
7056#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00007057 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00007058 sizeof(posix_constants_sysconf)
7059 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00007060 "sysconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00007061 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00007062#endif
Fred Drakebec628d1999-12-15 18:31:10 +00007063 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00007064}
Fred Draked86ed291999-12-15 15:34:33 +00007065
7066
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007067PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007068"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007069Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007070in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007071
7072static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00007073posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007074{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007075 abort();
7076 /*NOTREACHED*/
7077 Py_FatalError("abort() called from Python code didn't abort!");
7078 return NULL;
7079}
Fred Drakebec628d1999-12-15 18:31:10 +00007080
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007081#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007082PyDoc_STRVAR(win32_startfile__doc__,
7083"startfile(filepath) - Start a file with its associated application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00007084\n\
7085This acts like double-clicking the file in Explorer, or giving the file\n\
7086name as an argument to the DOS \"start\" command: the file is opened\n\
7087with whatever application (if any) its extension is associated.\n\
7088\n\
7089startfile returns as soon as the associated application is launched.\n\
7090There is no option to wait for the application to close, and no way\n\
7091to retrieve the application's exit status.\n\
7092\n\
7093The filepath is relative to the current directory. If you want to use\n\
7094an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007095the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00007096
7097static PyObject *
7098win32_startfile(PyObject *self, PyObject *args)
7099{
7100 char *filepath;
7101 HINSTANCE rc;
7102 if (!PyArg_ParseTuple(args, "s:startfile", &filepath))
7103 return NULL;
7104 Py_BEGIN_ALLOW_THREADS
7105 rc = ShellExecute((HWND)0, NULL, filepath, NULL, NULL, SW_SHOWNORMAL);
7106 Py_END_ALLOW_THREADS
7107 if (rc <= (HINSTANCE)32)
7108 return win32_error("startfile", filepath);
7109 Py_INCREF(Py_None);
7110 return Py_None;
7111}
7112#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007113
Martin v. Löwis438b5342002-12-27 10:16:42 +00007114#ifdef HAVE_GETLOADAVG
7115PyDoc_STRVAR(posix_getloadavg__doc__,
7116"getloadavg() -> (float, float, float)\n\n\
7117Return the number of processes in the system run queue averaged over\n\
7118the last 1, 5, and 15 minutes or raises OSError if the load average\n\
7119was unobtainable");
7120
7121static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00007122posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00007123{
7124 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00007125 if (getloadavg(loadavg, 3)!=3) {
7126 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
7127 return NULL;
7128 } else
7129 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
7130}
7131#endif
7132
7133
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007134static PyMethodDef posix_methods[] = {
7135 {"access", posix_access, METH_VARARGS, posix_access__doc__},
7136#ifdef HAVE_TTYNAME
7137 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
7138#endif
7139 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
7140 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007141#ifdef HAVE_CHOWN
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007142 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007143#endif /* HAVE_CHOWN */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00007144#ifdef HAVE_LCHOWN
7145 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
7146#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +00007147#ifdef HAVE_CHROOT
7148 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
7149#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007150#ifdef HAVE_CTERMID
Neal Norwitze241ce82003-02-17 18:17:05 +00007151 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007152#endif
Guido van Rossum36bc6801995-06-14 22:54:23 +00007153#ifdef HAVE_GETCWD
Neal Norwitze241ce82003-02-17 18:17:05 +00007154 {"getcwd", posix_getcwd, METH_NOARGS, posix_getcwd__doc__},
Walter Dörwald3b918c32002-11-21 20:18:46 +00007155#ifdef Py_USING_UNICODE
Neal Norwitze241ce82003-02-17 18:17:05 +00007156 {"getcwdu", posix_getcwdu, METH_NOARGS, posix_getcwdu__doc__},
Guido van Rossum36bc6801995-06-14 22:54:23 +00007157#endif
Walter Dörwald3b918c32002-11-21 20:18:46 +00007158#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00007159#ifdef HAVE_LINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007160 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007161#endif /* HAVE_LINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007162 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
7163 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
7164 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007165#ifdef HAVE_NICE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007166 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007167#endif /* HAVE_NICE */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007168#ifdef HAVE_READLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007169 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007170#endif /* HAVE_READLINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007171 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
7172 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
7173 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00007174 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007175#ifdef HAVE_SYMLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007176 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007177#endif /* HAVE_SYMLINK */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007178#ifdef HAVE_SYSTEM
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007179 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007180#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007181 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007182#ifdef HAVE_UNAME
Neal Norwitze241ce82003-02-17 18:17:05 +00007183 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007184#endif /* HAVE_UNAME */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007185 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
7186 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
7187 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007188#ifdef HAVE_TIMES
Neal Norwitze241ce82003-02-17 18:17:05 +00007189 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007190#endif /* HAVE_TIMES */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007191 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007192#ifdef HAVE_EXECV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007193 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
7194 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007195#endif /* HAVE_EXECV */
Guido van Rossuma1065681999-01-25 23:20:23 +00007196#ifdef HAVE_SPAWNV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007197 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
7198 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00007199#if defined(PYOS_OS2)
7200 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
7201 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
7202#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00007203#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +00007204#ifdef HAVE_FORK1
Neal Norwitze241ce82003-02-17 18:17:05 +00007205 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +00007206#endif /* HAVE_FORK1 */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007207#ifdef HAVE_FORK
Neal Norwitze241ce82003-02-17 18:17:05 +00007208 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007209#endif /* HAVE_FORK */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00007210#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Neal Norwitze241ce82003-02-17 18:17:05 +00007211 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +00007212#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00007213#ifdef HAVE_FORKPTY
Neal Norwitze241ce82003-02-17 18:17:05 +00007214 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +00007215#endif /* HAVE_FORKPTY */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007216#ifdef HAVE_GETEGID
Neal Norwitze241ce82003-02-17 18:17:05 +00007217 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007218#endif /* HAVE_GETEGID */
7219#ifdef HAVE_GETEUID
Neal Norwitze241ce82003-02-17 18:17:05 +00007220 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007221#endif /* HAVE_GETEUID */
7222#ifdef HAVE_GETGID
Neal Norwitze241ce82003-02-17 18:17:05 +00007223 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007224#endif /* HAVE_GETGID */
Fred Drakec9680921999-12-13 16:37:25 +00007225#ifdef HAVE_GETGROUPS
Neal Norwitze241ce82003-02-17 18:17:05 +00007226 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00007227#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00007228 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007229#ifdef HAVE_GETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00007230 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007231#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007232#ifdef HAVE_GETPPID
Neal Norwitze241ce82003-02-17 18:17:05 +00007233 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007234#endif /* HAVE_GETPPID */
7235#ifdef HAVE_GETUID
Neal Norwitze241ce82003-02-17 18:17:05 +00007236 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007237#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +00007238#ifdef HAVE_GETLOGIN
Neal Norwitze241ce82003-02-17 18:17:05 +00007239 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +00007240#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +00007241#ifdef HAVE_KILL
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007242 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007243#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00007244#ifdef HAVE_KILLPG
7245 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
7246#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +00007247#ifdef HAVE_PLOCK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007248 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +00007249#endif /* HAVE_PLOCK */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007250#ifdef HAVE_POPEN
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007251 {"popen", posix_popen, METH_VARARGS, posix_popen__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007252#ifdef MS_WINDOWS
Fredrik Lundhffb9c772000-07-09 14:49:51 +00007253 {"popen2", win32_popen2, METH_VARARGS},
7254 {"popen3", win32_popen3, METH_VARARGS},
7255 {"popen4", win32_popen4, METH_VARARGS},
Tim Petersf58a7aa2000-09-22 10:05:54 +00007256 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007257#else
7258#if defined(PYOS_OS2) && defined(PYCC_GCC)
7259 {"popen2", os2emx_popen2, METH_VARARGS},
7260 {"popen3", os2emx_popen3, METH_VARARGS},
7261 {"popen4", os2emx_popen4, METH_VARARGS},
7262#endif
Fredrik Lundhffb9c772000-07-09 14:49:51 +00007263#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007264#endif /* HAVE_POPEN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007265#ifdef HAVE_SETUID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007266 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007267#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00007268#ifdef HAVE_SETEUID
7269 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
7270#endif /* HAVE_SETEUID */
7271#ifdef HAVE_SETEGID
7272 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
7273#endif /* HAVE_SETEGID */
7274#ifdef HAVE_SETREUID
7275 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
7276#endif /* HAVE_SETREUID */
7277#ifdef HAVE_SETREGID
7278 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
7279#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007280#ifdef HAVE_SETGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007281 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007282#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00007283#ifdef HAVE_SETGROUPS
7284 {"setgroups", posix_setgroups, METH_VARARGS, posix_setgroups__doc__},
7285#endif /* HAVE_SETGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +00007286#ifdef HAVE_GETPGID
7287 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
7288#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007289#ifdef HAVE_SETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00007290 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007291#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007292#ifdef HAVE_WAIT
Neal Norwitze241ce82003-02-17 18:17:05 +00007293 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007294#endif /* HAVE_WAIT */
Tim Petersab034fa2002-02-01 11:27:43 +00007295#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007296 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007297#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00007298#ifdef HAVE_GETSID
7299 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
7300#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007301#ifdef HAVE_SETSID
Neal Norwitze241ce82003-02-17 18:17:05 +00007302 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007303#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007304#ifdef HAVE_SETPGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007305 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007306#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007307#ifdef HAVE_TCGETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007308 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007309#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007310#ifdef HAVE_TCSETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007311 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007312#endif /* HAVE_TCSETPGRP */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007313 {"open", posix_open, METH_VARARGS, posix_open__doc__},
7314 {"close", posix_close, METH_VARARGS, posix_close__doc__},
7315 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
7316 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
7317 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
7318 {"read", posix_read, METH_VARARGS, posix_read__doc__},
7319 {"write", posix_write, METH_VARARGS, posix_write__doc__},
7320 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
7321 {"fdopen", posix_fdopen, METH_VARARGS, posix_fdopen__doc__},
Skip Montanaro1517d842000-07-19 14:34:14 +00007322 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007323#ifdef HAVE_PIPE
Neal Norwitze241ce82003-02-17 18:17:05 +00007324 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007325#endif
7326#ifdef HAVE_MKFIFO
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007327 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007328#endif
Neal Norwitz11690112002-07-30 01:08:28 +00007329#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007330 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
7331#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007332#ifdef HAVE_DEVICE_MACROS
7333 {"major", posix_major, METH_VARARGS, posix_major__doc__},
7334 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
7335 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
7336#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007337#ifdef HAVE_FTRUNCATE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007338 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007339#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007340#ifdef HAVE_PUTENV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007341 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007342#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00007343#ifdef HAVE_UNSETENV
7344 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
7345#endif
Guido van Rossumb6a47161997-09-15 22:54:34 +00007346#ifdef HAVE_STRERROR
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007347 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Guido van Rossumb6a47161997-09-15 22:54:34 +00007348#endif
Fred Drake4d1e64b2002-04-15 19:40:07 +00007349#ifdef HAVE_FCHDIR
7350 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
7351#endif
Guido van Rossum21142a01999-01-08 21:05:37 +00007352#ifdef HAVE_FSYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007353 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007354#endif
7355#ifdef HAVE_FDATASYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007356 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007357#endif
Guido van Rossumc9641791998-08-04 15:26:23 +00007358#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +00007359#ifdef WCOREDUMP
7360 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
7361#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00007362#ifdef WIFCONTINUED
7363 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
7364#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +00007365#ifdef WIFSTOPPED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007366 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007367#endif /* WIFSTOPPED */
7368#ifdef WIFSIGNALED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007369 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007370#endif /* WIFSIGNALED */
7371#ifdef WIFEXITED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007372 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007373#endif /* WIFEXITED */
7374#ifdef WEXITSTATUS
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007375 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007376#endif /* WEXITSTATUS */
7377#ifdef WTERMSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007378 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007379#endif /* WTERMSIG */
7380#ifdef WSTOPSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007381 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007382#endif /* WSTOPSIG */
7383#endif /* HAVE_SYS_WAIT_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00007384#ifdef HAVE_FSTATVFS
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007385 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007386#endif
7387#ifdef HAVE_STATVFS
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007388 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007389#endif
Guido van Rossume2ad6332001-01-08 17:51:55 +00007390#ifdef HAVE_TMPFILE
Neal Norwitze241ce82003-02-17 18:17:05 +00007391 {"tmpfile", posix_tmpfile, METH_NOARGS, posix_tmpfile__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007392#endif
7393#ifdef HAVE_TEMPNAM
7394 {"tempnam", posix_tempnam, METH_VARARGS, posix_tempnam__doc__},
7395#endif
7396#ifdef HAVE_TMPNAM
Neal Norwitze241ce82003-02-17 18:17:05 +00007397 {"tmpnam", posix_tmpnam, METH_NOARGS, posix_tmpnam__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007398#endif
Fred Drakec9680921999-12-13 16:37:25 +00007399#ifdef HAVE_CONFSTR
7400 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
7401#endif
7402#ifdef HAVE_SYSCONF
7403 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
7404#endif
7405#ifdef HAVE_FPATHCONF
7406 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
7407#endif
7408#ifdef HAVE_PATHCONF
7409 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
7410#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00007411 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007412#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00007413 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
7414#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00007415#ifdef HAVE_GETLOADAVG
Neal Norwitze241ce82003-02-17 18:17:05 +00007416 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +00007417#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007418 {NULL, NULL} /* Sentinel */
7419};
7420
7421
Barry Warsaw4a342091996-12-19 23:50:02 +00007422static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00007423ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +00007424{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007425 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +00007426}
7427
Guido van Rossumd48f2521997-12-05 22:19:34 +00007428#if defined(PYOS_OS2)
7429/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007430static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007431{
7432 APIRET rc;
7433 ULONG values[QSV_MAX+1];
7434 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +00007435 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +00007436
7437 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +00007438 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007439 Py_END_ALLOW_THREADS
7440
7441 if (rc != NO_ERROR) {
7442 os2_error(rc);
7443 return -1;
7444 }
7445
Fred Drake4d1e64b2002-04-15 19:40:07 +00007446 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
7447 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
7448 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
7449 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
7450 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
7451 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
7452 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007453
7454 switch (values[QSV_VERSION_MINOR]) {
7455 case 0: ver = "2.00"; break;
7456 case 10: ver = "2.10"; break;
7457 case 11: ver = "2.11"; break;
7458 case 30: ver = "3.00"; break;
7459 case 40: ver = "4.00"; break;
7460 case 50: ver = "5.00"; break;
7461 default:
Tim Peters885d4572001-11-28 20:27:42 +00007462 PyOS_snprintf(tmp, sizeof(tmp),
7463 "%d-%d", values[QSV_VERSION_MAJOR],
7464 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007465 ver = &tmp[0];
7466 }
7467
7468 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007469 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007470 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007471
7472 /* Add Indicator of Which Drive was Used to Boot the System */
7473 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
7474 tmp[1] = ':';
7475 tmp[2] = '\0';
7476
Fred Drake4d1e64b2002-04-15 19:40:07 +00007477 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007478}
7479#endif
7480
Barry Warsaw4a342091996-12-19 23:50:02 +00007481static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007482all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +00007483{
Guido van Rossum94f6f721999-01-06 18:42:14 +00007484#ifdef F_OK
7485 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007486#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007487#ifdef R_OK
7488 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007489#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007490#ifdef W_OK
7491 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007492#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007493#ifdef X_OK
7494 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007495#endif
Fred Drakec9680921999-12-13 16:37:25 +00007496#ifdef NGROUPS_MAX
7497 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
7498#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007499#ifdef TMP_MAX
7500 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
7501#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007502#ifdef WCONTINUED
7503 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
7504#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007505#ifdef WNOHANG
7506 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007507#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007508#ifdef WUNTRACED
7509 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
7510#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007511#ifdef O_RDONLY
7512 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
7513#endif
7514#ifdef O_WRONLY
7515 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
7516#endif
7517#ifdef O_RDWR
7518 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
7519#endif
7520#ifdef O_NDELAY
7521 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
7522#endif
7523#ifdef O_NONBLOCK
7524 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
7525#endif
7526#ifdef O_APPEND
7527 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
7528#endif
7529#ifdef O_DSYNC
7530 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
7531#endif
7532#ifdef O_RSYNC
7533 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
7534#endif
7535#ifdef O_SYNC
7536 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
7537#endif
7538#ifdef O_NOCTTY
7539 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
7540#endif
7541#ifdef O_CREAT
7542 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
7543#endif
7544#ifdef O_EXCL
7545 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
7546#endif
7547#ifdef O_TRUNC
7548 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
7549#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +00007550#ifdef O_BINARY
7551 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
7552#endif
7553#ifdef O_TEXT
7554 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
7555#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007556#ifdef O_LARGEFILE
7557 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
7558#endif
7559
Tim Peters5aa91602002-01-30 05:46:57 +00007560/* MS Windows */
7561#ifdef O_NOINHERIT
7562 /* Don't inherit in child processes. */
7563 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
7564#endif
7565#ifdef _O_SHORT_LIVED
7566 /* Optimize for short life (keep in memory). */
7567 /* MS forgot to define this one with a non-underscore form too. */
7568 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
7569#endif
7570#ifdef O_TEMPORARY
7571 /* Automatically delete when last handle is closed. */
7572 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
7573#endif
7574#ifdef O_RANDOM
7575 /* Optimize for random access. */
7576 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
7577#endif
7578#ifdef O_SEQUENTIAL
7579 /* Optimize for sequential access. */
7580 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
7581#endif
7582
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007583/* GNU extensions. */
7584#ifdef O_DIRECT
7585 /* Direct disk access. */
7586 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
7587#endif
7588#ifdef O_DIRECTORY
7589 /* Must be a directory. */
7590 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
7591#endif
7592#ifdef O_NOFOLLOW
7593 /* Do not follow links. */
7594 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
7595#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00007596
Barry Warsaw5676bd12003-01-07 20:57:09 +00007597 /* These come from sysexits.h */
7598#ifdef EX_OK
7599 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007600#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007601#ifdef EX_USAGE
7602 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007603#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007604#ifdef EX_DATAERR
7605 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007606#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007607#ifdef EX_NOINPUT
7608 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007609#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007610#ifdef EX_NOUSER
7611 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007612#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007613#ifdef EX_NOHOST
7614 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007615#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007616#ifdef EX_UNAVAILABLE
7617 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007618#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007619#ifdef EX_SOFTWARE
7620 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007621#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007622#ifdef EX_OSERR
7623 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007624#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007625#ifdef EX_OSFILE
7626 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007627#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007628#ifdef EX_CANTCREAT
7629 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007630#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007631#ifdef EX_IOERR
7632 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007633#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007634#ifdef EX_TEMPFAIL
7635 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007636#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007637#ifdef EX_PROTOCOL
7638 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007639#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007640#ifdef EX_NOPERM
7641 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007642#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007643#ifdef EX_CONFIG
7644 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007645#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007646#ifdef EX_NOTFOUND
7647 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007648#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007649
Guido van Rossum246bc171999-02-01 23:54:31 +00007650#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007651#if defined(PYOS_OS2) && defined(PYCC_GCC)
7652 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
7653 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
7654 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
7655 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
7656 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
7657 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
7658 if (ins(d, "P_PM", (long)P_PM)) return -1;
7659 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
7660 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
7661 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
7662 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
7663 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
7664 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
7665 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
7666 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
7667 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
7668 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
7669 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
7670 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
7671 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
7672#else
Guido van Rossum7d385291999-02-16 19:38:04 +00007673 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
7674 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
7675 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
7676 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
7677 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +00007678#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007679#endif
Guido van Rossum246bc171999-02-01 23:54:31 +00007680
Guido van Rossumd48f2521997-12-05 22:19:34 +00007681#if defined(PYOS_OS2)
7682 if (insertvalues(d)) return -1;
7683#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007684 return 0;
7685}
7686
7687
Tim Peters5aa91602002-01-30 05:46:57 +00007688#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007689#define INITFUNC initnt
7690#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007691
7692#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007693#define INITFUNC initos2
7694#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007695
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007696#else
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007697#define INITFUNC initposix
7698#define MODNAME "posix"
7699#endif
7700
Mark Hammondfe51c6d2002-08-02 02:27:13 +00007701PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007702INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +00007703{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007704 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +00007705
Fred Drake4d1e64b2002-04-15 19:40:07 +00007706 m = Py_InitModule3(MODNAME,
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007707 posix_methods,
Fred Drake4d1e64b2002-04-15 19:40:07 +00007708 posix__doc__);
Tim Peters5aa91602002-01-30 05:46:57 +00007709
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007710 /* Initialize environ dictionary */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007711 v = convertenviron();
Fred Drake4d1e64b2002-04-15 19:40:07 +00007712 Py_XINCREF(v);
7713 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007714 return;
Barry Warsaw53699e91996-12-10 23:23:01 +00007715 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +00007716
Fred Drake4d1e64b2002-04-15 19:40:07 +00007717 if (all_ins(m))
Barry Warsaw4a342091996-12-19 23:50:02 +00007718 return;
7719
Fred Drake4d1e64b2002-04-15 19:40:07 +00007720 if (setup_confname_tables(m))
Fred Drakebec628d1999-12-15 18:31:10 +00007721 return;
7722
Fred Drake4d1e64b2002-04-15 19:40:07 +00007723 Py_INCREF(PyExc_OSError);
7724 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +00007725
Guido van Rossumb3d39562000-01-31 18:41:26 +00007726#ifdef HAVE_PUTENV
Neil Schemenauer19030a02001-01-16 04:27:47 +00007727 if (posix_putenv_garbage == NULL)
7728 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +00007729#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007730
Guido van Rossum14648392001-12-08 18:02:58 +00007731 stat_result_desc.name = MODNAME ".stat_result";
Martin v. Löwisf607bda2002-10-16 18:27:39 +00007732 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
7733 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
7734 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007735 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00007736 structseq_new = StatResultType.tp_new;
7737 StatResultType.tp_new = statresult_new;
Fred Drake4d1e64b2002-04-15 19:40:07 +00007738 Py_INCREF((PyObject*) &StatResultType);
7739 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007740
Guido van Rossum14648392001-12-08 18:02:58 +00007741 statvfs_result_desc.name = MODNAME ".statvfs_result";
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007742 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
Fred Drake4d1e64b2002-04-15 19:40:07 +00007743 Py_INCREF((PyObject*) &StatVFSResultType);
7744 PyModule_AddObject(m, "statvfs_result",
7745 (PyObject*) &StatVFSResultType);
Guido van Rossumb6775db1994-08-01 11:34:53 +00007746}