blob: d25034f14d7c993205ec5a400d0ca703956806aa [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
Thomas Wouters477c8d52006-05-27 19:21:47 +000016#ifdef __APPLE__
17 /*
18 * Step 1 of support for weak-linking a number of symbols existing on
19 * OSX 10.4 and later, see the comment in the #ifdef __APPLE__ block
20 * at the end of this file for more information.
21 */
22# pragma weak lchown
23# pragma weak statvfs
24# pragma weak fstatvfs
25
26#endif /* __APPLE__ */
27
Thomas Wouters68bc4f92006-03-01 01:05:10 +000028#define PY_SSIZE_T_CLEAN
29
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000030#include "Python.h"
31#include "structseq.h"
32
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000033#if defined(__VMS)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000034# include <unixio.h>
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000035#endif /* defined(__VMS) */
36
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000037#ifdef __cplusplus
38extern "C" {
39#endif
40
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000041PyDoc_STRVAR(posix__doc__,
Guido van Rossumec4f4ac1997-06-02 22:20:51 +000042"This module provides access to operating system functionality that is\n\
43standardized by the C Standard and the POSIX standard (a thinly\n\
44disguised Unix interface). Refer to the library manual and\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000045corresponding Unix manual entries for more information on calls.");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000046
Martin v. Löwis0073f2e2002-11-21 23:52:35 +000047
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000048#if defined(PYOS_OS2)
49#define INCL_DOS
50#define INCL_DOSERRORS
51#define INCL_DOSPROCESS
52#define INCL_NOPMAPI
53#include <os2.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000054#if defined(PYCC_GCC)
55#include <ctype.h>
56#include <io.h>
57#include <stdio.h>
58#include <process.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000059#endif
Andrew MacIntyreda4d6cb2004-03-29 11:53:38 +000060#include "osdefs.h"
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000061#endif
62
Thomas Wouters0e3f5912006-08-11 14:57:12 +000063#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000064#include <sys/types.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000065#endif /* HAVE_SYS_TYPES_H */
66
67#ifdef HAVE_SYS_STAT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000068#include <sys/stat.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000069#endif /* HAVE_SYS_STAT_H */
Guido van Rossuma6535fd2001-10-18 19:44:10 +000070
Guido van Rossum36bc6801995-06-14 22:54:23 +000071#ifdef HAVE_SYS_WAIT_H
72#include <sys/wait.h> /* For WNOHANG */
73#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +000074
Thomas Wouters0e3f5912006-08-11 14:57:12 +000075#ifdef HAVE_SIGNAL_H
Guido van Rossuma376cc51996-12-05 23:43:35 +000076#include <signal.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000077#endif
Guido van Rossuma376cc51996-12-05 23:43:35 +000078
Guido van Rossumb6775db1994-08-01 11:34:53 +000079#ifdef HAVE_FCNTL_H
80#include <fcntl.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000081#endif /* HAVE_FCNTL_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +000082
Guido van Rossuma6535fd2001-10-18 19:44:10 +000083#ifdef HAVE_GRP_H
84#include <grp.h>
85#endif
86
Barry Warsaw5676bd12003-01-07 20:57:09 +000087#ifdef HAVE_SYSEXITS_H
88#include <sysexits.h>
89#endif /* HAVE_SYSEXITS_H */
90
Anthony Baxter8a560de2004-10-13 15:30:56 +000091#ifdef HAVE_SYS_LOADAVG_H
92#include <sys/loadavg.h>
93#endif
94
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +000095#ifdef HAVE_LANGINFO_H
96#include <langinfo.h>
97#endif
98
Guido van Rossuma4916fa1996-05-23 22:58:55 +000099/* Various compilers have only certain posix functions */
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +0000100/* XXX Gosh I wish these were all moved into pyconfig.h */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000101#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000102#include <process.h>
103#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000104#if defined(__WATCOMC__) && !defined(__QNX__) /* Watcom compiler */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000105#define HAVE_GETCWD 1
106#define HAVE_OPENDIR 1
107#define HAVE_SYSTEM 1
108#if defined(__OS2__)
109#define HAVE_EXECV 1
110#define HAVE_WAIT 1
Guido van Rossumad0ee831995-03-01 10:34:45 +0000111#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000112#include <process.h>
113#else
114#ifdef __BORLANDC__ /* Borland compiler */
115#define HAVE_EXECV 1
116#define HAVE_GETCWD 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000117#define HAVE_OPENDIR 1
118#define HAVE_PIPE 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000119#define HAVE_SYSTEM 1
120#define HAVE_WAIT 1
121#else
122#ifdef _MSC_VER /* Microsoft compiler */
Guido van Rossum8d665e61996-06-26 18:22:49 +0000123#define HAVE_GETCWD 1
Guido van Rossuma1065681999-01-25 23:20:23 +0000124#define HAVE_SPAWNV 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000125#define HAVE_EXECV 1
126#define HAVE_PIPE 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000127#define HAVE_SYSTEM 1
Tim Petersab034fa2002-02-01 11:27:43 +0000128#define HAVE_CWAIT 1
Tim Peters11b23062003-04-23 02:39:17 +0000129#define HAVE_FSYNC 1
130#define fsync _commit
Andrew MacIntyre6c73af22002-03-03 03:07:07 +0000131#else
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000132#if defined(PYOS_OS2) && defined(PYCC_GCC) || defined(__VMS)
133/* Everything needed is defined in PC/os2emx/pyconfig.h or vms/pyconfig.h */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000134#else /* all other compilers */
135/* Unix functions that the configure script doesn't check for */
136#define HAVE_EXECV 1
137#define HAVE_FORK 1
Guido van Rossum2242f2f2001-04-11 20:58:20 +0000138#if defined(__USLC__) && defined(__SCO_VERSION__) /* SCO UDK Compiler */
139#define HAVE_FORK1 1
140#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000141#define HAVE_GETCWD 1
142#define HAVE_GETEGID 1
143#define HAVE_GETEUID 1
144#define HAVE_GETGID 1
145#define HAVE_GETPPID 1
146#define HAVE_GETUID 1
147#define HAVE_KILL 1
148#define HAVE_OPENDIR 1
149#define HAVE_PIPE 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000150#define HAVE_SYSTEM 1
151#define HAVE_WAIT 1
Guido van Rossumd371ff11999-01-25 16:12:23 +0000152#define HAVE_TTYNAME 1
Martin v. Löwis7a924e62003-03-05 14:15:21 +0000153#endif /* PYOS_OS2 && PYCC_GCC && __VMS */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000154#endif /* _MSC_VER */
155#endif /* __BORLANDC__ */
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000156#endif /* ! __WATCOMC__ || __QNX__ */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000157#endif /* ! __IBMC__ */
Guido van Rossumad0ee831995-03-01 10:34:45 +0000158
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000159#ifndef _MSC_VER
Guido van Rossum36bc6801995-06-14 22:54:23 +0000160
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000161#if defined(__sgi)&&_COMPILER_VERSION>=700
162/* declare ctermid_r if compiling with MIPSPro 7.x in ANSI C mode
163 (default) */
164extern char *ctermid_r(char *);
165#endif
166
Thomas Wouters1e0c2f42000-07-24 16:06:23 +0000167#ifndef HAVE_UNISTD_H
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000168#if defined(PYCC_VACPP)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000169extern int mkdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000170#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000171#if ( defined(__WATCOMC__) || defined(_MSC_VER) ) && !defined(__QNX__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000172extern int mkdir(const char *);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000173#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000174extern int mkdir(const char *, mode_t);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000175#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000176#endif
177#if defined(__IBMC__) || defined(__IBMCPP__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000178extern int chdir(char *);
179extern int rmdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000180#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000181extern int chdir(const char *);
182extern int rmdir(const char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000183#endif
Tim Peters58e0a8c2001-05-14 22:32:33 +0000184#ifdef __BORLANDC__
185extern int chmod(const char *, int);
186#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000187extern int chmod(const char *, mode_t);
Tim Peters58e0a8c2001-05-14 22:32:33 +0000188#endif
Christian Heimes4e30a842007-11-30 22:12:06 +0000189/*#ifdef HAVE_FCHMOD
190extern int fchmod(int, mode_t);
191#endif*/
192/*#ifdef HAVE_LCHMOD
193extern int lchmod(const char *, mode_t);
194#endif*/
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000195extern int chown(const char *, uid_t, gid_t);
196extern char *getcwd(char *, int);
197extern char *strerror(int);
198extern int link(const char *, const char *);
199extern int rename(const char *, const char *);
200extern int stat(const char *, struct stat *);
201extern int unlink(const char *);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000202#ifdef HAVE_SYMLINK
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000203extern int symlink(const char *, const char *);
Guido van Rossuma38a5031995-02-17 15:11:36 +0000204#endif /* HAVE_SYMLINK */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000205#ifdef HAVE_LSTAT
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000206extern int lstat(const char *, struct stat *);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000207#endif /* HAVE_LSTAT */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000208#endif /* !HAVE_UNISTD_H */
Guido van Rossum36bc6801995-06-14 22:54:23 +0000209
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000210#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000211
Guido van Rossumb6775db1994-08-01 11:34:53 +0000212#ifdef HAVE_UTIME_H
213#include <utime.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000214#endif /* HAVE_UTIME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000215
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000216#ifdef HAVE_SYS_UTIME_H
217#include <sys/utime.h>
218#define HAVE_UTIME_H /* pretend we do for the rest of this file */
219#endif /* HAVE_SYS_UTIME_H */
220
Guido van Rossumb6775db1994-08-01 11:34:53 +0000221#ifdef HAVE_SYS_TIMES_H
222#include <sys/times.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000223#endif /* HAVE_SYS_TIMES_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000224
225#ifdef HAVE_SYS_PARAM_H
226#include <sys/param.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000227#endif /* HAVE_SYS_PARAM_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000228
229#ifdef HAVE_SYS_UTSNAME_H
230#include <sys/utsname.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000231#endif /* HAVE_SYS_UTSNAME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000232
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000233#ifdef HAVE_DIRENT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000234#include <dirent.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000235#define NAMLEN(dirent) strlen((dirent)->d_name)
236#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000237#if defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000238#include <direct.h>
239#define NAMLEN(dirent) strlen((dirent)->d_name)
240#else
Guido van Rossumb6775db1994-08-01 11:34:53 +0000241#define dirent direct
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000242#define NAMLEN(dirent) (dirent)->d_namlen
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000243#endif
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000244#ifdef HAVE_SYS_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000245#include <sys/ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000246#endif
247#ifdef HAVE_SYS_DIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000248#include <sys/dir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000249#endif
250#ifdef HAVE_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000251#include <ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000252#endif
253#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000254
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000255#ifdef _MSC_VER
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000256#ifdef HAVE_DIRECT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000257#include <direct.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000258#endif
259#ifdef HAVE_IO_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000260#include <io.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000261#endif
262#ifdef HAVE_PROCESS_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000263#include <process.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000264#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000265#include "osdefs.h"
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000266#include <malloc.h>
Guido van Rossumb6775db1994-08-01 11:34:53 +0000267#include <windows.h>
Tim Petersee66d0c2002-07-15 16:10:55 +0000268#include <shellapi.h> /* for ShellExecute() */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000269#endif /* _MSC_VER */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000270
Guido van Rossumd48f2521997-12-05 22:19:34 +0000271#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000272#include <io.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000273#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000274
Tim Petersbc2e10e2002-03-03 23:17:02 +0000275#ifndef MAXPATHLEN
Thomas Wouters477c8d52006-05-27 19:21:47 +0000276#if defined(PATH_MAX) && PATH_MAX > 1024
277#define MAXPATHLEN PATH_MAX
278#else
Tim Petersbc2e10e2002-03-03 23:17:02 +0000279#define MAXPATHLEN 1024
Thomas Wouters477c8d52006-05-27 19:21:47 +0000280#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000281#endif /* MAXPATHLEN */
282
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000283#ifdef UNION_WAIT
284/* Emulate some macros on systems that have a union instead of macros */
285
286#ifndef WIFEXITED
287#define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
288#endif
289
290#ifndef WEXITSTATUS
291#define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
292#endif
293
294#ifndef WTERMSIG
295#define WTERMSIG(u_wait) ((u_wait).w_termsig)
296#endif
297
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000298#define WAIT_TYPE union wait
299#define WAIT_STATUS_INT(s) (s.w_status)
300
301#else /* !UNION_WAIT */
302#define WAIT_TYPE int
303#define WAIT_STATUS_INT(s) (s)
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000304#endif /* UNION_WAIT */
305
Antoine Pitrouc3ee1662009-05-23 16:02:33 +0000306/* Issue #1983: pid_t can be longer than a C long on some systems */
Antoine Pitrou7852c422009-05-24 11:58:35 +0000307#if !defined(SIZEOF_PID_T) || SIZEOF_PID_T == SIZEOF_INT
Antoine Pitrouc3ee1662009-05-23 16:02:33 +0000308#define PARSE_PID "i"
309#define PyLong_FromPid PyLong_FromLong
310#define PyLong_AsPid PyLong_AsLong
311#elif SIZEOF_PID_T == SIZEOF_LONG
312#define PARSE_PID "l"
313#define PyLong_FromPid PyLong_FromLong
314#define PyLong_AsPid PyLong_AsLong
315#elif defined(SIZEOF_LONG_LONG) && SIZEOF_PID_T == SIZEOF_LONG_LONG
316#define PARSE_PID "L"
317#define PyLong_FromPid PyLong_FromLongLong
318#define PyLong_AsPid PyLong_AsLongLong
319#else
320#error "sizeof(pid_t) is neither sizeof(int), sizeof(long) or sizeof(long long)"
Antoine Pitrouc3ee1662009-05-23 16:02:33 +0000321#endif /* SIZEOF_PID_T */
322
Greg Wardb48bc172000-03-01 21:51:56 +0000323/* Don't use the "_r" form if we don't need it (also, won't have a
324 prototype for it, at least on Solaris -- maybe others as well?). */
325#if defined(HAVE_CTERMID_R) && defined(WITH_THREAD)
326#define USE_CTERMID_R
327#endif
328
Fred Drake699f3522000-06-29 21:12:41 +0000329/* choose the appropriate stat and fstat functions and return structs */
Guido van Rossum64529cd2000-06-30 22:45:12 +0000330#undef STAT
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000331#if defined(MS_WIN64) || defined(MS_WINDOWS)
Martin v. Löwis14694662006-02-03 12:54:16 +0000332# define STAT win32_stat
333# define FSTAT win32_fstat
334# define STRUCT_STAT struct win32_stat
Fred Drake699f3522000-06-29 21:12:41 +0000335#else
336# define STAT stat
337# define FSTAT fstat
338# define STRUCT_STAT struct stat
339#endif
340
Tim Peters11b23062003-04-23 02:39:17 +0000341#if defined(MAJOR_IN_MKDEV)
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000342#include <sys/mkdev.h>
343#else
344#if defined(MAJOR_IN_SYSMACROS)
345#include <sys/sysmacros.h>
346#endif
Neal Norwitz3d949422002-04-20 13:46:43 +0000347#if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
348#include <sys/mkdev.h>
349#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000350#endif
Fred Drake699f3522000-06-29 21:12:41 +0000351
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000352#if defined _MSC_VER && _MSC_VER >= 1400
353/* Microsoft CRT in VS2005 and higher will verify that a filehandle is
354 * valid and throw an assertion if it isn't.
355 * Normally, an invalid fd is likely to be a C program error and therefore
356 * an assertion can be useful, but it does contradict the POSIX standard
357 * which for write(2) states:
358 * "Otherwise, -1 shall be returned and errno set to indicate the error."
359 * "[EBADF] The fildes argument is not a valid file descriptor open for
360 * writing."
361 * Furthermore, python allows the user to enter any old integer
362 * as a fd and should merely raise a python exception on error.
363 * The Microsoft CRT doesn't provide an official way to check for the
364 * validity of a file descriptor, but we can emulate its internal behaviour
365 * by using the exported __pinfo data member and knowledge of the
366 * internal structures involved.
367 * The structures below must be updated for each version of visual studio
368 * according to the file internal.h in the CRT source, until MS comes
369 * up with a less hacky way to do this.
370 * (all of this is to avoid globally modifying the CRT behaviour using
371 * _set_invalid_parameter_handler() and _CrtSetReportMode())
372 */
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000373/* The actual size of the structure is determined at runtime.
374 * Only the first items must be present.
375 */
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000376typedef struct {
377 intptr_t osfhnd;
378 char osfile;
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000379} my_ioinfo;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000380
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000381extern __declspec(dllimport) char * __pioinfo[];
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000382#define IOINFO_L2E 5
383#define IOINFO_ARRAY_ELTS (1 << IOINFO_L2E)
384#define IOINFO_ARRAYS 64
385#define _NHANDLE_ (IOINFO_ARRAYS * IOINFO_ARRAY_ELTS)
386#define FOPEN 0x01
387#define _NO_CONSOLE_FILENO (intptr_t)-2
388
389/* This function emulates what the windows CRT does to validate file handles */
390int
391_PyVerify_fd(int fd)
392{
393 const int i1 = fd >> IOINFO_L2E;
394 const int i2 = fd & ((1 << IOINFO_L2E) - 1);
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000395
396 static int sizeof_ioinfo = 0;
397
398 /* Determine the actual size of the ioinfo structure,
399 * as used by the CRT loaded in memory
400 */
401 if (sizeof_ioinfo == 0 && __pioinfo[0] != NULL) {
402 sizeof_ioinfo = _msize(__pioinfo[0]) / IOINFO_ARRAY_ELTS;
403 }
404 if (sizeof_ioinfo == 0) {
405 /* This should not happen... */
406 goto fail;
407 }
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000408
409 /* See that it isn't a special CLEAR fileno */
410 if (fd != _NO_CONSOLE_FILENO) {
411 /* Microsoft CRT would check that 0<=fd<_nhandle but we can't do that. Instead
412 * we check pointer validity and other info
413 */
414 if (0 <= i1 && i1 < IOINFO_ARRAYS && __pioinfo[i1] != NULL) {
415 /* finally, check that the file is open */
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000416 my_ioinfo* info = (my_ioinfo*)(__pioinfo[i1] + i2 * sizeof_ioinfo);
417 if (info->osfile & FOPEN) {
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000418 return 1;
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000419 }
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000420 }
421 }
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000422 fail:
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000423 errno = EBADF;
424 return 0;
425}
426
427/* the special case of checking dup2. The target fd must be in a sensible range */
428static int
429_PyVerify_fd_dup2(int fd1, int fd2)
430{
431 if (!_PyVerify_fd(fd1))
432 return 0;
433 if (fd2 == _NO_CONSOLE_FILENO)
434 return 0;
435 if ((unsigned)fd2 < _NHANDLE_)
436 return 1;
437 else
438 return 0;
439}
440#else
441/* dummy version. _PyVerify_fd() is already defined in fileobject.h */
442#define _PyVerify_fd_dup2(A, B) (1)
443#endif
444
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000445/* Return a dictionary corresponding to the POSIX environment table */
Jack Jansenea0c3822002-08-01 21:57:49 +0000446#ifdef WITH_NEXT_FRAMEWORK
447/* On Darwin/MacOSX a shared library or framework has no access to
448** environ directly, we must obtain it with _NSGetEnviron().
449*/
450#include <crt_externs.h>
451static char **environ;
452#elif !defined(_MSC_VER) && ( !defined(__WATCOMC__) || defined(__QNX__) )
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000453extern char **environ;
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000454#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000455
Barry Warsaw53699e91996-12-10 23:23:01 +0000456static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000457convertenviron(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000458{
Barry Warsaw53699e91996-12-10 23:23:01 +0000459 PyObject *d;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000460#ifdef MS_WINDOWS
461 wchar_t **e;
462#else
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000463 char **e;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000464#endif
Barry Warsaw53699e91996-12-10 23:23:01 +0000465 d = PyDict_New();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000466 if (d == NULL)
467 return NULL;
Jack Jansenea0c3822002-08-01 21:57:49 +0000468#ifdef WITH_NEXT_FRAMEWORK
469 if (environ == NULL)
470 environ = *_NSGetEnviron();
471#endif
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000472#ifdef MS_WINDOWS
473 /* _wenviron must be initialized in this way if the program is started
474 through main() instead of wmain(). */
475 _wgetenv(L"");
476 if (_wenviron == NULL)
477 return d;
478 /* This part ignores errors */
479 for (e = _wenviron; *e != NULL; e++) {
480 PyObject *k;
481 PyObject *v;
482 wchar_t *p = wcschr(*e, L'=');
483 if (p == NULL)
484 continue;
485 k = PyUnicode_FromWideChar(*e, (Py_ssize_t)(p-*e));
486 if (k == NULL) {
487 PyErr_Clear();
488 continue;
489 }
490 v = PyUnicode_FromWideChar(p+1, wcslen(p+1));
491 if (v == NULL) {
492 PyErr_Clear();
493 Py_DECREF(k);
494 continue;
495 }
496 if (PyDict_GetItem(d, k) == NULL) {
497 if (PyDict_SetItem(d, k, v) != 0)
498 PyErr_Clear();
499 }
500 Py_DECREF(k);
501 Py_DECREF(v);
502 }
503#else
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000504 if (environ == NULL)
505 return d;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000506 /* This part ignores errors */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000507 for (e = environ; *e != NULL; e++) {
Guido van Rossum6a619f41999-08-03 19:41:10 +0000508 PyObject *k;
Barry Warsaw53699e91996-12-10 23:23:01 +0000509 PyObject *v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000510 char *p = strchr(*e, '=');
511 if (p == NULL)
512 continue;
Martin v. Löwis011e8422009-05-05 04:43:17 +0000513 k = PyUnicode_Decode(*e, (int)(p-*e),
Martin v. Löwis43c57782009-05-10 08:15:24 +0000514 Py_FileSystemDefaultEncoding, "surrogateescape");
Guido van Rossum6a619f41999-08-03 19:41:10 +0000515 if (k == NULL) {
516 PyErr_Clear();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000517 continue;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000518 }
Martin v. Löwis011e8422009-05-05 04:43:17 +0000519 v = PyUnicode_Decode(p+1, strlen(p+1),
Martin v. Löwis43c57782009-05-10 08:15:24 +0000520 Py_FileSystemDefaultEncoding, "surrogateescape");
Guido van Rossum6a619f41999-08-03 19:41:10 +0000521 if (v == NULL) {
522 PyErr_Clear();
523 Py_DECREF(k);
524 continue;
525 }
526 if (PyDict_GetItem(d, k) == NULL) {
527 if (PyDict_SetItem(d, k, v) != 0)
528 PyErr_Clear();
529 }
530 Py_DECREF(k);
Barry Warsaw53699e91996-12-10 23:23:01 +0000531 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000532 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000533#endif
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +0000534#if defined(PYOS_OS2)
Guido van Rossumd48f2521997-12-05 22:19:34 +0000535 {
536 APIRET rc;
537 char buffer[1024]; /* OS/2 Provides a Documented Max of 1024 Chars */
538
539 rc = DosQueryExtLIBPATH(buffer, BEGIN_LIBPATH);
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000540 if (rc == NO_ERROR) { /* (not a type, envname is NOT 'BEGIN_LIBPATH') */
Christian Heimes72b710a2008-05-26 13:28:38 +0000541 PyObject *v = PyBytes_FromString(buffer);
Neal Norwitz93c56822007-08-26 07:10:06 +0000542 PyDict_SetItemString(d, "BEGINLIBPATH", v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000543 Py_DECREF(v);
544 }
545 rc = DosQueryExtLIBPATH(buffer, END_LIBPATH);
546 if (rc == NO_ERROR) { /* (not a typo, envname is NOT 'END_LIBPATH') */
Christian Heimes72b710a2008-05-26 13:28:38 +0000547 PyObject *v = PyBytes_FromString(buffer);
Neal Norwitz93c56822007-08-26 07:10:06 +0000548 PyDict_SetItemString(d, "ENDLIBPATH", v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000549 Py_DECREF(v);
550 }
551 }
552#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000553 return d;
554}
555
Martin v. Löwis011e8422009-05-05 04:43:17 +0000556/* Convert a bytes object to a char*. Optionally lock the buffer if it is a
557 bytes array. */
558
559static char*
560bytes2str(PyObject* o, int lock)
561{
562 if(PyBytes_Check(o))
563 return PyBytes_AsString(o);
564 else if(PyByteArray_Check(o)) {
565 if (lock && PyObject_GetBuffer(o, NULL, 0) < 0)
566 /* On a bytearray, this should not fail. */
567 PyErr_BadInternalCall();
568 return PyByteArray_AsString(o);
569 } else {
570 /* The FS converter should have verified that this
571 is either bytes or bytearray. */
572 Py_FatalError("bad object passed to bytes2str");
573 /* not reached. */
574 return "";
575 }
576}
577
578/* Release the lock, decref the object. */
579static void
580release_bytes(PyObject* o)
581{
582 if (PyByteArray_Check(o))
583 o->ob_type->tp_as_buffer->bf_releasebuffer(NULL, 0);
584 Py_DECREF(o);
585}
586
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000587
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000588/* Set a POSIX-specific error from errno, and return NULL */
589
Barry Warsawd58d7641998-07-23 16:14:40 +0000590static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000591posix_error(void)
Guido van Rossumad0ee831995-03-01 10:34:45 +0000592{
Barry Warsawca74da41999-02-09 19:31:45 +0000593 return PyErr_SetFromErrno(PyExc_OSError);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000594}
Barry Warsawd58d7641998-07-23 16:14:40 +0000595static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000596posix_error_with_filename(char* name)
Barry Warsawd58d7641998-07-23 16:14:40 +0000597{
Barry Warsawca74da41999-02-09 19:31:45 +0000598 return PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
Barry Warsawd58d7641998-07-23 16:14:40 +0000599}
600
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000601#ifdef MS_WINDOWS
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000602static PyObject *
603posix_error_with_unicode_filename(Py_UNICODE* name)
604{
605 return PyErr_SetFromErrnoWithUnicodeFilename(PyExc_OSError, name);
606}
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000607#endif /* MS_WINDOWS */
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000608
609
Mark Hammondef8b6542001-05-13 08:04:26 +0000610static PyObject *
Martin v. Löwis011e8422009-05-05 04:43:17 +0000611posix_error_with_allocated_filename(PyObject* name)
Mark Hammondef8b6542001-05-13 08:04:26 +0000612{
Martin v. Löwis011e8422009-05-05 04:43:17 +0000613 PyObject *rc = PyErr_SetFromErrnoWithFilename(PyExc_OSError,
614 bytes2str(name, 0));
615 release_bytes(name);
Mark Hammondef8b6542001-05-13 08:04:26 +0000616 return rc;
617}
618
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000619#ifdef MS_WINDOWS
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000620static PyObject *
621win32_error(char* function, char* filename)
622{
Mark Hammond33a6da92000-08-15 00:46:38 +0000623 /* XXX We should pass the function name along in the future.
Georg Brandl38feaf02008-05-25 07:45:51 +0000624 (winreg.c also wants to pass the function name.)
Tim Peters5aa91602002-01-30 05:46:57 +0000625 This would however require an additional param to the
Mark Hammond33a6da92000-08-15 00:46:38 +0000626 Windows error object, which is non-trivial.
627 */
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000628 errno = GetLastError();
629 if (filename)
Mark Hammond33a6da92000-08-15 00:46:38 +0000630 return PyErr_SetFromWindowsErrWithFilename(errno, filename);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000631 else
Mark Hammond33a6da92000-08-15 00:46:38 +0000632 return PyErr_SetFromWindowsErr(errno);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000633}
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000634
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000635static PyObject *
636win32_error_unicode(char* function, Py_UNICODE* filename)
637{
638 /* XXX - see win32_error for comments on 'function' */
639 errno = GetLastError();
640 if (filename)
641 return PyErr_SetFromWindowsErrWithUnicodeFilename(errno, filename);
642 else
643 return PyErr_SetFromWindowsErr(errno);
644}
645
Thomas Wouters477c8d52006-05-27 19:21:47 +0000646static int
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +0000647convert_to_unicode(PyObject **param)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000648{
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +0000649 if (PyUnicode_CheckExact(*param))
650 Py_INCREF(*param);
651 else if (PyUnicode_Check(*param))
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000652 /* For a Unicode subtype that's not a Unicode object,
653 return a true Unicode object with the same data. */
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +0000654 *param = PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(*param),
655 PyUnicode_GET_SIZE(*param));
Thomas Wouters477c8d52006-05-27 19:21:47 +0000656 else
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +0000657 *param = PyUnicode_FromEncodedObject(*param,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000658 Py_FileSystemDefaultEncoding,
659 "strict");
660 return (*param) != NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000661}
662
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000663#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000664
Guido van Rossumd48f2521997-12-05 22:19:34 +0000665#if defined(PYOS_OS2)
666/**********************************************************************
667 * Helper Function to Trim and Format OS/2 Messages
668 **********************************************************************/
669 static void
670os2_formatmsg(char *msgbuf, int msglen, char *reason)
671{
672 msgbuf[msglen] = '\0'; /* OS/2 Doesn't Guarantee a Terminator */
673
674 if (strlen(msgbuf) > 0) { /* If Non-Empty Msg, Trim CRLF */
675 char *lastc = &msgbuf[ strlen(msgbuf)-1 ];
676
Neal Norwitz30b5c5d2005-12-19 06:05:18 +0000677 while (lastc > msgbuf && isspace(Py_CHARMASK(*lastc)))
Guido van Rossumd48f2521997-12-05 22:19:34 +0000678 *lastc-- = '\0'; /* Trim Trailing Whitespace (CRLF) */
679 }
680
681 /* Add Optional Reason Text */
682 if (reason) {
683 strcat(msgbuf, " : ");
684 strcat(msgbuf, reason);
685 }
686}
687
688/**********************************************************************
689 * Decode an OS/2 Operating System Error Code
690 *
691 * A convenience function to lookup an OS/2 error code and return a
692 * text message we can use to raise a Python exception.
693 *
694 * Notes:
695 * The messages for errors returned from the OS/2 kernel reside in
696 * the file OSO001.MSG in the \OS2 directory hierarchy.
697 *
698 **********************************************************************/
699 static char *
700os2_strerror(char *msgbuf, int msgbuflen, int errorcode, char *reason)
701{
702 APIRET rc;
703 ULONG msglen;
704
705 /* Retrieve Kernel-Related Error Message from OSO001.MSG File */
706 Py_BEGIN_ALLOW_THREADS
707 rc = DosGetMessage(NULL, 0, msgbuf, msgbuflen,
708 errorcode, "oso001.msg", &msglen);
709 Py_END_ALLOW_THREADS
710
711 if (rc == NO_ERROR)
712 os2_formatmsg(msgbuf, msglen, reason);
713 else
Tim Peters1ceb5fb2001-11-28 20:32:57 +0000714 PyOS_snprintf(msgbuf, msgbuflen,
Tim Peters885d4572001-11-28 20:27:42 +0000715 "unknown OS error #%d", errorcode);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000716
717 return msgbuf;
718}
719
720/* Set an OS/2-specific error and return NULL. OS/2 kernel
721 errors are not in a global variable e.g. 'errno' nor are
722 they congruent with posix error numbers. */
723
724static PyObject * os2_error(int code)
725{
726 char text[1024];
727 PyObject *v;
728
729 os2_strerror(text, sizeof(text), code, "");
730
731 v = Py_BuildValue("(is)", code, text);
732 if (v != NULL) {
Barry Warsawca74da41999-02-09 19:31:45 +0000733 PyErr_SetObject(PyExc_OSError, v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000734 Py_DECREF(v);
735 }
736 return NULL; /* Signal to Python that an Exception is Pending */
737}
738
739#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000740
741/* POSIX generic methods */
742
Barry Warsaw53699e91996-12-10 23:23:01 +0000743static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +0000744posix_fildes(PyObject *fdobj, int (*func)(int))
745{
746 int fd;
747 int res;
748 fd = PyObject_AsFileDescriptor(fdobj);
749 if (fd < 0)
750 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000751 if (!_PyVerify_fd(fd))
752 return posix_error();
Fred Drake4d1e64b2002-04-15 19:40:07 +0000753 Py_BEGIN_ALLOW_THREADS
754 res = (*func)(fd);
755 Py_END_ALLOW_THREADS
756 if (res < 0)
757 return posix_error();
758 Py_INCREF(Py_None);
759 return Py_None;
760}
Guido van Rossum21142a01999-01-08 21:05:37 +0000761
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000762#ifdef MS_WINDOWS
Tim Peters11b23062003-04-23 02:39:17 +0000763static int
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000764unicode_file_names(void)
765{
766 static int canusewide = -1;
767 if (canusewide == -1) {
Tim Peters11b23062003-04-23 02:39:17 +0000768 /* As per doc for ::GetVersion(), this is the correct test for
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000769 the Windows NT family. */
770 canusewide = (GetVersion() < 0x80000000) ? 1 : 0;
771 }
772 return canusewide;
773}
774#endif
Tim Peters11b23062003-04-23 02:39:17 +0000775
Guido van Rossum21142a01999-01-08 21:05:37 +0000776static PyObject *
Thomas Wouters477c8d52006-05-27 19:21:47 +0000777posix_1str(PyObject *args, char *format, int (*func)(const char*))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000778{
Martin v. Löwis011e8422009-05-05 04:43:17 +0000779 PyObject *opath1 = NULL;
780 char *path1;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000781 int res;
Tim Peters5aa91602002-01-30 05:46:57 +0000782 if (!PyArg_ParseTuple(args, format,
Martin v. Löwis011e8422009-05-05 04:43:17 +0000783 PyUnicode_FSConverter, &opath1))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000784 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +0000785 path1 = bytes2str(opath1, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000786 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000787 res = (*func)(path1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000788 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000789 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +0000790 return posix_error_with_allocated_filename(opath1);
791 release_bytes(opath1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000792 Py_INCREF(Py_None);
793 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000794}
795
Barry Warsaw53699e91996-12-10 23:23:01 +0000796static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +0000797posix_2str(PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000798 char *format,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000799 int (*func)(const char *, const char *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000800{
Antoine Pitroua6bb9842009-05-10 22:27:00 +0000801 PyObject *opath1 = NULL, *opath2 = NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +0000802 char *path1, *path2;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000803 int res;
Mark Hammondef8b6542001-05-13 08:04:26 +0000804 if (!PyArg_ParseTuple(args, format,
Martin v. Löwis011e8422009-05-05 04:43:17 +0000805 PyUnicode_FSConverter, &opath1,
Antoine Pitroua6bb9842009-05-10 22:27:00 +0000806 PyUnicode_FSConverter, &opath2)) {
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000807 return NULL;
Antoine Pitroua6bb9842009-05-10 22:27:00 +0000808 }
Martin v. Löwis011e8422009-05-05 04:43:17 +0000809 path1 = bytes2str(opath1, 1);
810 path2 = bytes2str(opath2, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000811 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000812 res = (*func)(path1, path2);
Barry Warsaw53699e91996-12-10 23:23:01 +0000813 Py_END_ALLOW_THREADS
Martin v. Löwis011e8422009-05-05 04:43:17 +0000814 release_bytes(opath1);
815 release_bytes(opath2);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000816 if (res != 0)
Barry Warsawd58d7641998-07-23 16:14:40 +0000817 /* XXX how to report both path1 and path2??? */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000818 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +0000819 Py_INCREF(Py_None);
820 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000821}
822
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000823#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +0000824static PyObject*
825win32_1str(PyObject* args, char* func,
826 char* format, BOOL (__stdcall *funcA)(LPCSTR),
827 char* wformat, BOOL (__stdcall *funcW)(LPWSTR))
828{
829 PyObject *uni;
830 char *ansi;
831 BOOL result;
832 if (unicode_file_names()) {
833 if (!PyArg_ParseTuple(args, wformat, &uni))
834 PyErr_Clear();
835 else {
836 Py_BEGIN_ALLOW_THREADS
837 result = funcW(PyUnicode_AsUnicode(uni));
838 Py_END_ALLOW_THREADS
839 if (!result)
840 return win32_error_unicode(func, PyUnicode_AsUnicode(uni));
841 Py_INCREF(Py_None);
842 return Py_None;
843 }
844 }
845 if (!PyArg_ParseTuple(args, format, &ansi))
846 return NULL;
847 Py_BEGIN_ALLOW_THREADS
848 result = funcA(ansi);
849 Py_END_ALLOW_THREADS
850 if (!result)
851 return win32_error(func, ansi);
852 Py_INCREF(Py_None);
853 return Py_None;
854
855}
856
857/* This is a reimplementation of the C library's chdir function,
858 but one that produces Win32 errors instead of DOS error codes.
859 chdir is essentially a wrapper around SetCurrentDirectory; however,
860 it also needs to set "magic" environment variables indicating
861 the per-drive current directory, which are of the form =<drive>: */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000862static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000863win32_chdir(LPCSTR path)
864{
865 char new_path[MAX_PATH+1];
866 int result;
867 char env[4] = "=x:";
868
869 if(!SetCurrentDirectoryA(path))
870 return FALSE;
871 result = GetCurrentDirectoryA(MAX_PATH+1, new_path);
872 if (!result)
873 return FALSE;
874 /* In the ANSI API, there should not be any paths longer
875 than MAX_PATH. */
876 assert(result <= MAX_PATH+1);
877 if (strncmp(new_path, "\\\\", 2) == 0 ||
878 strncmp(new_path, "//", 2) == 0)
879 /* UNC path, nothing to do. */
880 return TRUE;
881 env[1] = new_path[0];
882 return SetEnvironmentVariableA(env, new_path);
883}
884
885/* The Unicode version differs from the ANSI version
886 since the current directory might exceed MAX_PATH characters */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000887static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000888win32_wchdir(LPCWSTR path)
889{
890 wchar_t _new_path[MAX_PATH+1], *new_path = _new_path;
891 int result;
892 wchar_t env[4] = L"=x:";
893
894 if(!SetCurrentDirectoryW(path))
895 return FALSE;
896 result = GetCurrentDirectoryW(MAX_PATH+1, new_path);
897 if (!result)
898 return FALSE;
899 if (result > MAX_PATH+1) {
Benjamin Petersonf10a79a2008-10-11 00:49:57 +0000900 new_path = malloc(result * sizeof(wchar_t));
Thomas Wouters477c8d52006-05-27 19:21:47 +0000901 if (!new_path) {
902 SetLastError(ERROR_OUTOFMEMORY);
903 return FALSE;
904 }
Benjamin Petersonf10a79a2008-10-11 00:49:57 +0000905 result = GetCurrentDirectoryW(result, new_path);
906 if (!result) {
907 free(new_path);
908 return FALSE;
909 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000910 }
911 if (wcsncmp(new_path, L"\\\\", 2) == 0 ||
912 wcsncmp(new_path, L"//", 2) == 0)
913 /* UNC path, nothing to do. */
914 return TRUE;
915 env[1] = new_path[0];
916 result = SetEnvironmentVariableW(env, new_path);
917 if (new_path != _new_path)
918 free(new_path);
919 return result;
920}
921#endif
922
Martin v. Löwis14694662006-02-03 12:54:16 +0000923#ifdef MS_WINDOWS
924/* The CRT of Windows has a number of flaws wrt. its stat() implementation:
925 - time stamps are restricted to second resolution
926 - file modification times suffer from forth-and-back conversions between
927 UTC and local time
928 Therefore, we implement our own stat, based on the Win32 API directly.
929*/
930#define HAVE_STAT_NSEC 1
931
932struct win32_stat{
933 int st_dev;
934 __int64 st_ino;
935 unsigned short st_mode;
936 int st_nlink;
937 int st_uid;
938 int st_gid;
939 int st_rdev;
940 __int64 st_size;
941 int st_atime;
942 int st_atime_nsec;
943 int st_mtime;
944 int st_mtime_nsec;
945 int st_ctime;
946 int st_ctime_nsec;
947};
948
949static __int64 secs_between_epochs = 11644473600; /* Seconds between 1.1.1601 and 1.1.1970 */
950
951static void
952FILE_TIME_to_time_t_nsec(FILETIME *in_ptr, int *time_out, int* nsec_out)
953{
Thomas Wouters477c8d52006-05-27 19:21:47 +0000954 /* XXX endianness. Shouldn't matter, as all Windows implementations are little-endian */
955 /* Cannot simply cast and dereference in_ptr,
956 since it might not be aligned properly */
957 __int64 in;
958 memcpy(&in, in_ptr, sizeof(in));
Martin v. Löwis14694662006-02-03 12:54:16 +0000959 *nsec_out = (int)(in % 10000000) * 100; /* FILETIME is in units of 100 nsec. */
960 /* XXX Win32 supports time stamps past 2038; we currently don't */
961 *time_out = Py_SAFE_DOWNCAST((in / 10000000) - secs_between_epochs, __int64, int);
962}
963
Thomas Wouters477c8d52006-05-27 19:21:47 +0000964static void
965time_t_to_FILE_TIME(int time_in, int nsec_in, FILETIME *out_ptr)
966{
967 /* XXX endianness */
968 __int64 out;
969 out = time_in + secs_between_epochs;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000970 out = out * 10000000 + nsec_in / 100;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000971 memcpy(out_ptr, &out, sizeof(out));
972}
973
Martin v. Löwis14694662006-02-03 12:54:16 +0000974/* Below, we *know* that ugo+r is 0444 */
975#if _S_IREAD != 0400
976#error Unsupported C library
977#endif
978static int
979attributes_to_mode(DWORD attr)
980{
981 int m = 0;
982 if (attr & FILE_ATTRIBUTE_DIRECTORY)
983 m |= _S_IFDIR | 0111; /* IFEXEC for user,group,other */
984 else
985 m |= _S_IFREG;
986 if (attr & FILE_ATTRIBUTE_READONLY)
987 m |= 0444;
988 else
989 m |= 0666;
990 return m;
991}
992
993static int
994attribute_data_to_stat(WIN32_FILE_ATTRIBUTE_DATA *info, struct win32_stat *result)
995{
996 memset(result, 0, sizeof(*result));
997 result->st_mode = attributes_to_mode(info->dwFileAttributes);
998 result->st_size = (((__int64)info->nFileSizeHigh)<<32) + info->nFileSizeLow;
999 FILE_TIME_to_time_t_nsec(&info->ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1000 FILE_TIME_to_time_t_nsec(&info->ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1001 FILE_TIME_to_time_t_nsec(&info->ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
1002
1003 return 0;
1004}
1005
Thomas Wouters89f507f2006-12-13 04:49:30 +00001006/* Emulate GetFileAttributesEx[AW] on Windows 95 */
1007static int checked = 0;
1008static BOOL (CALLBACK *gfaxa)(LPCSTR, GET_FILEEX_INFO_LEVELS, LPVOID);
1009static BOOL (CALLBACK *gfaxw)(LPCWSTR, GET_FILEEX_INFO_LEVELS, LPVOID);
1010static void
1011check_gfax()
1012{
1013 HINSTANCE hKernel32;
1014 if (checked)
1015 return;
1016 checked = 1;
1017 hKernel32 = GetModuleHandle("KERNEL32");
1018 *(FARPROC*)&gfaxa = GetProcAddress(hKernel32, "GetFileAttributesExA");
1019 *(FARPROC*)&gfaxw = GetProcAddress(hKernel32, "GetFileAttributesExW");
1020}
1021
Guido van Rossumd8faa362007-04-27 19:54:29 +00001022static BOOL
1023attributes_from_dir(LPCSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
1024{
1025 HANDLE hFindFile;
1026 WIN32_FIND_DATAA FileData;
1027 hFindFile = FindFirstFileA(pszFile, &FileData);
1028 if (hFindFile == INVALID_HANDLE_VALUE)
1029 return FALSE;
1030 FindClose(hFindFile);
1031 pfad->dwFileAttributes = FileData.dwFileAttributes;
1032 pfad->ftCreationTime = FileData.ftCreationTime;
1033 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
1034 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
1035 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
1036 pfad->nFileSizeLow = FileData.nFileSizeLow;
1037 return TRUE;
1038}
1039
1040static BOOL
1041attributes_from_dir_w(LPCWSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
1042{
1043 HANDLE hFindFile;
1044 WIN32_FIND_DATAW FileData;
1045 hFindFile = FindFirstFileW(pszFile, &FileData);
1046 if (hFindFile == INVALID_HANDLE_VALUE)
1047 return FALSE;
1048 FindClose(hFindFile);
1049 pfad->dwFileAttributes = FileData.dwFileAttributes;
1050 pfad->ftCreationTime = FileData.ftCreationTime;
1051 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
1052 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
1053 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
1054 pfad->nFileSizeLow = FileData.nFileSizeLow;
1055 return TRUE;
1056}
1057
Thomas Wouters89f507f2006-12-13 04:49:30 +00001058static BOOL WINAPI
1059Py_GetFileAttributesExA(LPCSTR pszFile,
1060 GET_FILEEX_INFO_LEVELS level,
1061 LPVOID pv)
1062{
1063 BOOL result;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001064 LPWIN32_FILE_ATTRIBUTE_DATA pfad = pv;
1065 /* First try to use the system's implementation, if that is
1066 available and either succeeds to gives an error other than
1067 that it isn't implemented. */
1068 check_gfax();
1069 if (gfaxa) {
1070 result = gfaxa(pszFile, level, pv);
1071 if (result || GetLastError() != ERROR_CALL_NOT_IMPLEMENTED)
1072 return result;
1073 }
1074 /* It's either not present, or not implemented.
1075 Emulate using FindFirstFile. */
1076 if (level != GetFileExInfoStandard) {
1077 SetLastError(ERROR_INVALID_PARAMETER);
1078 return FALSE;
1079 }
1080 /* Use GetFileAttributes to validate that the file name
1081 does not contain wildcards (which FindFirstFile would
1082 accept). */
1083 if (GetFileAttributesA(pszFile) == 0xFFFFFFFF)
1084 return FALSE;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001085 return attributes_from_dir(pszFile, pfad);
Thomas Wouters89f507f2006-12-13 04:49:30 +00001086}
1087
1088static BOOL WINAPI
1089Py_GetFileAttributesExW(LPCWSTR pszFile,
1090 GET_FILEEX_INFO_LEVELS level,
1091 LPVOID pv)
1092{
1093 BOOL result;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001094 LPWIN32_FILE_ATTRIBUTE_DATA pfad = pv;
1095 /* First try to use the system's implementation, if that is
1096 available and either succeeds to gives an error other than
1097 that it isn't implemented. */
1098 check_gfax();
1099 if (gfaxa) {
1100 result = gfaxw(pszFile, level, pv);
1101 if (result || GetLastError() != ERROR_CALL_NOT_IMPLEMENTED)
1102 return result;
1103 }
1104 /* It's either not present, or not implemented.
1105 Emulate using FindFirstFile. */
1106 if (level != GetFileExInfoStandard) {
1107 SetLastError(ERROR_INVALID_PARAMETER);
1108 return FALSE;
1109 }
1110 /* Use GetFileAttributes to validate that the file name
1111 does not contain wildcards (which FindFirstFile would
1112 accept). */
1113 if (GetFileAttributesW(pszFile) == 0xFFFFFFFF)
1114 return FALSE;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001115 return attributes_from_dir_w(pszFile, pfad);
Thomas Wouters89f507f2006-12-13 04:49:30 +00001116}
1117
Martin v. Löwis14694662006-02-03 12:54:16 +00001118static int
1119win32_stat(const char* path, struct win32_stat *result)
1120{
1121 WIN32_FILE_ATTRIBUTE_DATA info;
1122 int code;
1123 char *dot;
1124 /* XXX not supported on Win95 and NT 3.x */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001125 if (!Py_GetFileAttributesExA(path, GetFileExInfoStandard, &info)) {
Guido van Rossumd8faa362007-04-27 19:54:29 +00001126 if (GetLastError() != ERROR_SHARING_VIOLATION) {
1127 /* Protocol violation: we explicitly clear errno, instead of
1128 setting it to a POSIX error. Callers should use GetLastError. */
1129 errno = 0;
1130 return -1;
1131 } else {
1132 /* Could not get attributes on open file. Fall back to
1133 reading the directory. */
1134 if (!attributes_from_dir(path, &info)) {
1135 /* Very strange. This should not fail now */
1136 errno = 0;
1137 return -1;
1138 }
1139 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001140 }
1141 code = attribute_data_to_stat(&info, result);
1142 if (code != 0)
1143 return code;
1144 /* Set S_IFEXEC if it is an .exe, .bat, ... */
1145 dot = strrchr(path, '.');
1146 if (dot) {
1147 if (stricmp(dot, ".bat") == 0 ||
1148 stricmp(dot, ".cmd") == 0 ||
1149 stricmp(dot, ".exe") == 0 ||
1150 stricmp(dot, ".com") == 0)
1151 result->st_mode |= 0111;
1152 }
1153 return code;
1154}
1155
1156static int
1157win32_wstat(const wchar_t* path, struct win32_stat *result)
1158{
1159 int code;
1160 const wchar_t *dot;
1161 WIN32_FILE_ATTRIBUTE_DATA info;
1162 /* XXX not supported on Win95 and NT 3.x */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001163 if (!Py_GetFileAttributesExW(path, GetFileExInfoStandard, &info)) {
Guido van Rossumd8faa362007-04-27 19:54:29 +00001164 if (GetLastError() != ERROR_SHARING_VIOLATION) {
1165 /* Protocol violation: we explicitly clear errno, instead of
1166 setting it to a POSIX error. Callers should use GetLastError. */
1167 errno = 0;
1168 return -1;
1169 } else {
1170 /* Could not get attributes on open file. Fall back to
1171 reading the directory. */
1172 if (!attributes_from_dir_w(path, &info)) {
1173 /* Very strange. This should not fail now */
1174 errno = 0;
1175 return -1;
1176 }
1177 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001178 }
1179 code = attribute_data_to_stat(&info, result);
1180 if (code < 0)
1181 return code;
1182 /* Set IFEXEC if it is an .exe, .bat, ... */
1183 dot = wcsrchr(path, '.');
1184 if (dot) {
1185 if (_wcsicmp(dot, L".bat") == 0 ||
1186 _wcsicmp(dot, L".cmd") == 0 ||
1187 _wcsicmp(dot, L".exe") == 0 ||
1188 _wcsicmp(dot, L".com") == 0)
1189 result->st_mode |= 0111;
1190 }
1191 return code;
1192}
1193
1194static int
1195win32_fstat(int file_number, struct win32_stat *result)
1196{
1197 BY_HANDLE_FILE_INFORMATION info;
1198 HANDLE h;
1199 int type;
1200
1201 h = (HANDLE)_get_osfhandle(file_number);
1202
1203 /* Protocol violation: we explicitly clear errno, instead of
1204 setting it to a POSIX error. Callers should use GetLastError. */
1205 errno = 0;
1206
1207 if (h == INVALID_HANDLE_VALUE) {
1208 /* This is really a C library error (invalid file handle).
1209 We set the Win32 error to the closes one matching. */
1210 SetLastError(ERROR_INVALID_HANDLE);
1211 return -1;
1212 }
1213 memset(result, 0, sizeof(*result));
1214
1215 type = GetFileType(h);
1216 if (type == FILE_TYPE_UNKNOWN) {
1217 DWORD error = GetLastError();
1218 if (error != 0) {
1219 return -1;
1220 }
1221 /* else: valid but unknown file */
1222 }
1223
1224 if (type != FILE_TYPE_DISK) {
1225 if (type == FILE_TYPE_CHAR)
1226 result->st_mode = _S_IFCHR;
1227 else if (type == FILE_TYPE_PIPE)
1228 result->st_mode = _S_IFIFO;
1229 return 0;
1230 }
1231
1232 if (!GetFileInformationByHandle(h, &info)) {
1233 return -1;
1234 }
1235
1236 /* similar to stat() */
1237 result->st_mode = attributes_to_mode(info.dwFileAttributes);
1238 result->st_size = (((__int64)info.nFileSizeHigh)<<32) + info.nFileSizeLow;
1239 FILE_TIME_to_time_t_nsec(&info.ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1240 FILE_TIME_to_time_t_nsec(&info.ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1241 FILE_TIME_to_time_t_nsec(&info.ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
1242 /* specific to fstat() */
1243 result->st_nlink = info.nNumberOfLinks;
1244 result->st_ino = (((__int64)info.nFileIndexHigh)<<32) + info.nFileIndexLow;
1245 return 0;
1246}
1247
1248#endif /* MS_WINDOWS */
1249
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001250PyDoc_STRVAR(stat_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001251"stat_result: Result from stat or lstat.\n\n\
1252This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001253 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001254or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
1255\n\
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001256Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
1257or st_flags, they are available as attributes only.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001258\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001259See os.stat for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001260
1261static PyStructSequence_Field stat_result_fields[] = {
1262 {"st_mode", "protection bits"},
1263 {"st_ino", "inode"},
1264 {"st_dev", "device"},
1265 {"st_nlink", "number of hard links"},
1266 {"st_uid", "user ID of owner"},
1267 {"st_gid", "group ID of owner"},
1268 {"st_size", "total size, in bytes"},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001269 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
1270 {NULL, "integer time of last access"},
1271 {NULL, "integer time of last modification"},
1272 {NULL, "integer time of last change"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001273 {"st_atime", "time of last access"},
1274 {"st_mtime", "time of last modification"},
1275 {"st_ctime", "time of last change"},
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001276#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001277 {"st_blksize", "blocksize for filesystem I/O"},
1278#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001279#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001280 {"st_blocks", "number of blocks allocated"},
1281#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001282#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001283 {"st_rdev", "device type (if inode device)"},
1284#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001285#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1286 {"st_flags", "user defined flags for file"},
1287#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001288#ifdef HAVE_STRUCT_STAT_ST_GEN
1289 {"st_gen", "generation number"},
1290#endif
1291#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1292 {"st_birthtime", "time of creation"},
1293#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001294 {0}
1295};
1296
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001297#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001298#define ST_BLKSIZE_IDX 13
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001299#else
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001300#define ST_BLKSIZE_IDX 12
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001301#endif
1302
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001303#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001304#define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
1305#else
1306#define ST_BLOCKS_IDX ST_BLKSIZE_IDX
1307#endif
1308
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001309#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001310#define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
1311#else
1312#define ST_RDEV_IDX ST_BLOCKS_IDX
1313#endif
1314
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001315#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1316#define ST_FLAGS_IDX (ST_RDEV_IDX+1)
1317#else
1318#define ST_FLAGS_IDX ST_RDEV_IDX
1319#endif
1320
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001321#ifdef HAVE_STRUCT_STAT_ST_GEN
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001322#define ST_GEN_IDX (ST_FLAGS_IDX+1)
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001323#else
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001324#define ST_GEN_IDX ST_FLAGS_IDX
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001325#endif
1326
1327#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1328#define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
1329#else
1330#define ST_BIRTHTIME_IDX ST_GEN_IDX
1331#endif
1332
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001333static PyStructSequence_Desc stat_result_desc = {
1334 "stat_result", /* name */
1335 stat_result__doc__, /* doc */
1336 stat_result_fields,
1337 10
1338};
1339
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001340PyDoc_STRVAR(statvfs_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001341"statvfs_result: Result from statvfs or fstatvfs.\n\n\
1342This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001343 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
Guido van Rossuma4dc73e2001-10-18 20:53:15 +00001344or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001345\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001346See os.statvfs for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001347
1348static PyStructSequence_Field statvfs_result_fields[] = {
1349 {"f_bsize", },
1350 {"f_frsize", },
1351 {"f_blocks", },
1352 {"f_bfree", },
1353 {"f_bavail", },
1354 {"f_files", },
1355 {"f_ffree", },
1356 {"f_favail", },
1357 {"f_flag", },
1358 {"f_namemax",},
1359 {0}
1360};
1361
1362static PyStructSequence_Desc statvfs_result_desc = {
1363 "statvfs_result", /* name */
1364 statvfs_result__doc__, /* doc */
1365 statvfs_result_fields,
1366 10
1367};
1368
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001369static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001370static PyTypeObject StatResultType;
1371static PyTypeObject StatVFSResultType;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001372static newfunc structseq_new;
1373
1374static PyObject *
1375statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1376{
1377 PyStructSequence *result;
1378 int i;
1379
1380 result = (PyStructSequence*)structseq_new(type, args, kwds);
1381 if (!result)
1382 return NULL;
1383 /* If we have been initialized from a tuple,
1384 st_?time might be set to None. Initialize it
1385 from the int slots. */
1386 for (i = 7; i <= 9; i++) {
1387 if (result->ob_item[i+3] == Py_None) {
1388 Py_DECREF(Py_None);
1389 Py_INCREF(result->ob_item[i]);
1390 result->ob_item[i+3] = result->ob_item[i];
1391 }
1392 }
1393 return (PyObject*)result;
1394}
1395
1396
1397
1398/* If true, st_?time is float. */
Martin v. Löwisfe33d0b2005-01-16 08:57:39 +00001399static int _stat_float_times = 1;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001400
1401PyDoc_STRVAR(stat_float_times__doc__,
1402"stat_float_times([newval]) -> oldval\n\n\
1403Determine whether os.[lf]stat represents time stamps as float objects.\n\
1404If newval is True, future calls to stat() return floats, if it is False,\n\
1405future calls return ints. \n\
1406If newval is omitted, return the current setting.\n");
1407
1408static PyObject*
1409stat_float_times(PyObject* self, PyObject *args)
1410{
1411 int newval = -1;
1412 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
1413 return NULL;
1414 if (newval == -1)
1415 /* Return old value */
1416 return PyBool_FromLong(_stat_float_times);
1417 _stat_float_times = newval;
1418 Py_INCREF(Py_None);
1419 return Py_None;
1420}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001421
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001422static void
1423fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
1424{
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001425 PyObject *fval,*ival;
1426#if SIZEOF_TIME_T > SIZEOF_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00001427 ival = PyLong_FromLongLong((PY_LONG_LONG)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001428#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001429 ival = PyLong_FromLong((long)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001430#endif
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001431 if (!ival)
1432 return;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001433 if (_stat_float_times) {
1434 fval = PyFloat_FromDouble(sec + 1e-9*nsec);
1435 } else {
1436 fval = ival;
1437 Py_INCREF(fval);
1438 }
1439 PyStructSequence_SET_ITEM(v, index, ival);
1440 PyStructSequence_SET_ITEM(v, index+3, fval);
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001441}
1442
Tim Peters5aa91602002-01-30 05:46:57 +00001443/* pack a system stat C structure into the Python stat tuple
Fred Drake699f3522000-06-29 21:12:41 +00001444 (used by posix_stat() and posix_fstat()) */
1445static PyObject*
Martin v. Löwis14694662006-02-03 12:54:16 +00001446_pystat_fromstructstat(STRUCT_STAT *st)
Fred Drake699f3522000-06-29 21:12:41 +00001447{
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001448 unsigned long ansec, mnsec, cnsec;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001449 PyObject *v = PyStructSequence_New(&StatResultType);
Fred Drake699f3522000-06-29 21:12:41 +00001450 if (v == NULL)
1451 return NULL;
1452
Christian Heimes217cfd12007-12-02 14:31:20 +00001453 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long)st->st_mode));
Fred Drake699f3522000-06-29 21:12:41 +00001454#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +00001455 PyStructSequence_SET_ITEM(v, 1,
Martin v. Löwis14694662006-02-03 12:54:16 +00001456 PyLong_FromLongLong((PY_LONG_LONG)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001457#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001458 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001459#endif
1460#if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
Tim Peters5aa91602002-01-30 05:46:57 +00001461 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwis14694662006-02-03 12:54:16 +00001462 PyLong_FromLongLong((PY_LONG_LONG)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001463#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001464 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001465#endif
Christian Heimes217cfd12007-12-02 14:31:20 +00001466 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long)st->st_nlink));
1467 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long)st->st_uid));
1468 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long)st->st_gid));
Fred Drake699f3522000-06-29 21:12:41 +00001469#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +00001470 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwis14694662006-02-03 12:54:16 +00001471 PyLong_FromLongLong((PY_LONG_LONG)st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001472#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001473 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong(st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001474#endif
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001475
Martin v. Löwis14694662006-02-03 12:54:16 +00001476#if defined(HAVE_STAT_TV_NSEC)
1477 ansec = st->st_atim.tv_nsec;
1478 mnsec = st->st_mtim.tv_nsec;
1479 cnsec = st->st_ctim.tv_nsec;
1480#elif defined(HAVE_STAT_TV_NSEC2)
1481 ansec = st->st_atimespec.tv_nsec;
1482 mnsec = st->st_mtimespec.tv_nsec;
1483 cnsec = st->st_ctimespec.tv_nsec;
1484#elif defined(HAVE_STAT_NSEC)
1485 ansec = st->st_atime_nsec;
1486 mnsec = st->st_mtime_nsec;
1487 cnsec = st->st_ctime_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001488#else
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001489 ansec = mnsec = cnsec = 0;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001490#endif
Martin v. Löwis14694662006-02-03 12:54:16 +00001491 fill_time(v, 7, st->st_atime, ansec);
1492 fill_time(v, 8, st->st_mtime, mnsec);
1493 fill_time(v, 9, st->st_ctime, cnsec);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001494
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001495#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Tim Peters5aa91602002-01-30 05:46:57 +00001496 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001497 PyLong_FromLong((long)st->st_blksize));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001498#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001499#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Tim Peters5aa91602002-01-30 05:46:57 +00001500 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001501 PyLong_FromLong((long)st->st_blocks));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001502#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001503#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001504 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001505 PyLong_FromLong((long)st->st_rdev));
Fred Drake699f3522000-06-29 21:12:41 +00001506#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001507#ifdef HAVE_STRUCT_STAT_ST_GEN
1508 PyStructSequence_SET_ITEM(v, ST_GEN_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001509 PyLong_FromLong((long)st->st_gen));
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001510#endif
1511#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1512 {
1513 PyObject *val;
1514 unsigned long bsec,bnsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001515 bsec = (long)st->st_birthtime;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001516#ifdef HAVE_STAT_TV_NSEC2
Hye-Shik Changd69e0342006-02-19 16:22:22 +00001517 bnsec = st->st_birthtimespec.tv_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001518#else
1519 bnsec = 0;
1520#endif
1521 if (_stat_float_times) {
1522 val = PyFloat_FromDouble(bsec + 1e-9*bnsec);
1523 } else {
Christian Heimes217cfd12007-12-02 14:31:20 +00001524 val = PyLong_FromLong((long)bsec);
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001525 }
1526 PyStructSequence_SET_ITEM(v, ST_BIRTHTIME_IDX,
1527 val);
1528 }
1529#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001530#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1531 PyStructSequence_SET_ITEM(v, ST_FLAGS_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001532 PyLong_FromLong((long)st->st_flags));
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001533#endif
Fred Drake699f3522000-06-29 21:12:41 +00001534
1535 if (PyErr_Occurred()) {
1536 Py_DECREF(v);
1537 return NULL;
1538 }
1539
1540 return v;
1541}
1542
Martin v. Löwisd8948722004-06-02 09:57:56 +00001543#ifdef MS_WINDOWS
1544
1545/* IsUNCRoot -- test whether the supplied path is of the form \\SERVER\SHARE\,
1546 where / can be used in place of \ and the trailing slash is optional.
1547 Both SERVER and SHARE must have at least one character.
1548*/
1549
1550#define ISSLASHA(c) ((c) == '\\' || (c) == '/')
1551#define ISSLASHW(c) ((c) == L'\\' || (c) == L'/')
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001552#ifndef ARRAYSIZE
Martin v. Löwisd8948722004-06-02 09:57:56 +00001553#define ARRAYSIZE(a) (sizeof(a) / sizeof(a[0]))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001554#endif
Martin v. Löwisd8948722004-06-02 09:57:56 +00001555
Tim Peters4ad82172004-08-30 17:02:04 +00001556static BOOL
Martin v. Löwisd8948722004-06-02 09:57:56 +00001557IsUNCRootA(char *path, int pathlen)
1558{
1559 #define ISSLASH ISSLASHA
1560
1561 int i, share;
1562
1563 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1564 /* minimum UNCRoot is \\x\y */
1565 return FALSE;
1566 for (i = 2; i < pathlen ; i++)
1567 if (ISSLASH(path[i])) break;
1568 if (i == 2 || i == pathlen)
1569 /* do not allow \\\SHARE or \\SERVER */
1570 return FALSE;
1571 share = i+1;
1572 for (i = share; i < pathlen; i++)
1573 if (ISSLASH(path[i])) break;
1574 return (i != share && (i == pathlen || i == pathlen-1));
1575
1576 #undef ISSLASH
1577}
1578
Tim Peters4ad82172004-08-30 17:02:04 +00001579static BOOL
Martin v. Löwisd8948722004-06-02 09:57:56 +00001580IsUNCRootW(Py_UNICODE *path, int pathlen)
1581{
1582 #define ISSLASH ISSLASHW
1583
1584 int i, share;
1585
1586 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1587 /* minimum UNCRoot is \\x\y */
1588 return FALSE;
1589 for (i = 2; i < pathlen ; i++)
1590 if (ISSLASH(path[i])) break;
1591 if (i == 2 || i == pathlen)
1592 /* do not allow \\\SHARE or \\SERVER */
1593 return FALSE;
1594 share = i+1;
1595 for (i = share; i < pathlen; i++)
1596 if (ISSLASH(path[i])) break;
1597 return (i != share && (i == pathlen || i == pathlen-1));
1598
1599 #undef ISSLASH
1600}
Martin v. Löwisd8948722004-06-02 09:57:56 +00001601#endif /* MS_WINDOWS */
1602
Barry Warsaw53699e91996-12-10 23:23:01 +00001603static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +00001604posix_do_stat(PyObject *self, PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001605 char *format,
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001606#ifdef __VMS
1607 int (*statfunc)(const char *, STRUCT_STAT *, ...),
1608#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001609 int (*statfunc)(const char *, STRUCT_STAT *),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001610#endif
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001611 char *wformat,
1612 int (*wstatfunc)(const Py_UNICODE *, STRUCT_STAT *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001613{
Fred Drake699f3522000-06-29 21:12:41 +00001614 STRUCT_STAT st;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001615 PyObject *opath;
1616 char *path;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001617 int res;
Martin v. Löwis14694662006-02-03 12:54:16 +00001618 PyObject *result;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001619
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001620#ifdef MS_WINDOWS
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001621 /* If on wide-character-capable OS see if argument
1622 is Unicode and if so use wide API. */
1623 if (unicode_file_names()) {
1624 PyUnicodeObject *po;
1625 if (PyArg_ParseTuple(args, wformat, &po)) {
Martin v. Löwis14694662006-02-03 12:54:16 +00001626 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
1627
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001628 Py_BEGIN_ALLOW_THREADS
1629 /* PyUnicode_AS_UNICODE result OK without
1630 thread lock as it is a simple dereference. */
1631 res = wstatfunc(wpath, &st);
1632 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001633
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001634 if (res != 0)
Martin v. Löwis14694662006-02-03 12:54:16 +00001635 return win32_error_unicode("stat", wpath);
1636 return _pystat_fromstructstat(&st);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001637 }
1638 /* Drop the argument parsing error as narrow strings
1639 are also valid. */
1640 PyErr_Clear();
1641 }
1642#endif
1643
Tim Peters5aa91602002-01-30 05:46:57 +00001644 if (!PyArg_ParseTuple(args, format,
Martin v. Löwis011e8422009-05-05 04:43:17 +00001645 PyUnicode_FSConverter, &opath))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001646 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001647 path = bytes2str(opath, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00001648 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001649 res = (*statfunc)(path, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00001650 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001651
1652 if (res != 0) {
1653#ifdef MS_WINDOWS
Martin v. Löwis011e8422009-05-05 04:43:17 +00001654 result = win32_error("stat", path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001655#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00001656 result = posix_error_with_filename(path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001657#endif
1658 }
1659 else
1660 result = _pystat_fromstructstat(&st);
Fred Drake699f3522000-06-29 21:12:41 +00001661
Martin v. Löwis011e8422009-05-05 04:43:17 +00001662 release_bytes(opath);
Martin v. Löwis14694662006-02-03 12:54:16 +00001663 return result;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001664}
1665
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001666/* POSIX methods */
1667
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001668PyDoc_STRVAR(posix_access__doc__,
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001669"access(path, mode) -> True if granted, False otherwise\n\n\
Guido van Rossuma0b90752002-06-18 16:22:43 +00001670Use the real uid/gid to test for access to a path. Note that most\n\
1671operations will use the effective uid/gid, therefore this routine can\n\
1672be used in a suid/sgid environment to test if the invoking user has the\n\
1673specified access to the path. The mode argument can be F_OK to test\n\
1674existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001675
1676static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001677posix_access(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001678{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001679 PyObject *opath;
Guido van Rossum015f22a1999-01-06 22:52:38 +00001680 char *path;
1681 int mode;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001682
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001683#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001684 DWORD attr;
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001685 if (unicode_file_names()) {
1686 PyUnicodeObject *po;
1687 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
1688 Py_BEGIN_ALLOW_THREADS
1689 /* PyUnicode_AS_UNICODE OK without thread lock as
1690 it is a simple dereference. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001691 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001692 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001693 goto finish;
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001694 }
1695 /* Drop the argument parsing error as narrow strings
1696 are also valid. */
1697 PyErr_Clear();
1698 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00001699 if (!PyArg_ParseTuple(args, "O&i:access",
1700 PyUnicode_FSConverter, &opath, &mode))
Thomas Wouters477c8d52006-05-27 19:21:47 +00001701 return 0;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001702 path = bytes2str(opath, 1);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001703 Py_BEGIN_ALLOW_THREADS
1704 attr = GetFileAttributesA(path);
1705 Py_END_ALLOW_THREADS
Martin v. Löwis011e8422009-05-05 04:43:17 +00001706 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001707finish:
1708 if (attr == 0xFFFFFFFF)
1709 /* File does not exist, or cannot read attributes */
1710 return PyBool_FromLong(0);
1711 /* Access is possible if either write access wasn't requested, or
Guido van Rossumb00324f2007-12-04 01:13:14 +00001712 the file isn't read-only, or if it's a directory, as there are
1713 no read-only directories on Windows. */
1714 return PyBool_FromLong(!(mode & 2)
1715 || !(attr & FILE_ATTRIBUTE_READONLY)
1716 || (attr & FILE_ATTRIBUTE_DIRECTORY));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001717#else
1718 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001719 if (!PyArg_ParseTuple(args, "O&i:access",
1720 PyUnicode_FSConverter, &opath, &mode))
Guido van Rossum015f22a1999-01-06 22:52:38 +00001721 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001722 path = bytes2str(opath, 1);
Guido van Rossum015f22a1999-01-06 22:52:38 +00001723 Py_BEGIN_ALLOW_THREADS
1724 res = access(path, mode);
1725 Py_END_ALLOW_THREADS
Martin v. Löwis011e8422009-05-05 04:43:17 +00001726 release_bytes(opath);
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001727 return PyBool_FromLong(res == 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001728#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001729}
1730
Guido van Rossumd371ff11999-01-25 16:12:23 +00001731#ifndef F_OK
1732#define F_OK 0
1733#endif
1734#ifndef R_OK
1735#define R_OK 4
1736#endif
1737#ifndef W_OK
1738#define W_OK 2
1739#endif
1740#ifndef X_OK
1741#define X_OK 1
1742#endif
1743
1744#ifdef HAVE_TTYNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001745PyDoc_STRVAR(posix_ttyname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001746"ttyname(fd) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001747Return the name of the terminal device connected to 'fd'.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001748
1749static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001750posix_ttyname(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001751{
Guido van Rossum94f6f721999-01-06 18:42:14 +00001752 int id;
1753 char *ret;
1754
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001755 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
Guido van Rossum94f6f721999-01-06 18:42:14 +00001756 return NULL;
1757
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001758#if defined(__VMS)
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001759 /* file descriptor 0 only, the default input device (stdin) */
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001760 if (id == 0) {
1761 ret = ttyname();
1762 }
1763 else {
1764 ret = NULL;
1765 }
1766#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00001767 ret = ttyname(id);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001768#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001769 if (ret == NULL)
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001770 return posix_error();
Neal Norwitz93c56822007-08-26 07:10:06 +00001771 return PyUnicode_FromString(ret);
Guido van Rossum94f6f721999-01-06 18:42:14 +00001772}
Guido van Rossumd371ff11999-01-25 16:12:23 +00001773#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001774
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001775#ifdef HAVE_CTERMID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001776PyDoc_STRVAR(posix_ctermid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001777"ctermid() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001778Return the name of the controlling terminal for this process.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001779
1780static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001781posix_ctermid(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001782{
1783 char *ret;
1784 char buffer[L_ctermid];
1785
Greg Wardb48bc172000-03-01 21:51:56 +00001786#ifdef USE_CTERMID_R
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001787 ret = ctermid_r(buffer);
1788#else
1789 ret = ctermid(buffer);
1790#endif
1791 if (ret == NULL)
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001792 return posix_error();
Neal Norwitz93c56822007-08-26 07:10:06 +00001793 return PyUnicode_FromString(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001794}
1795#endif
1796
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001797PyDoc_STRVAR(posix_chdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001798"chdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001799Change the current working directory to the specified path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001800
Barry Warsaw53699e91996-12-10 23:23:01 +00001801static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001802posix_chdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001803{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001804#ifdef MS_WINDOWS
Amaury Forgeot d'Arcc72ef8b2008-10-03 18:38:26 +00001805 return win32_1str(args, "chdir", "y:chdir", win32_chdir, "U:chdir", win32_wchdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001806#elif defined(PYOS_OS2) && defined(PYCC_GCC)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001807 return posix_1str(args, "O&:chdir", _chdir2);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00001808#elif defined(__VMS)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001809 return posix_1str(args, "O&:chdir", (int (*)(const char *))chdir);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001810#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00001811 return posix_1str(args, "O&:chdir", chdir);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001812#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001813}
1814
Fred Drake4d1e64b2002-04-15 19:40:07 +00001815#ifdef HAVE_FCHDIR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001816PyDoc_STRVAR(posix_fchdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001817"fchdir(fildes)\n\n\
Fred Drake4d1e64b2002-04-15 19:40:07 +00001818Change to the directory of the given file descriptor. fildes must be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001819opened on a directory, not a file.");
Fred Drake4d1e64b2002-04-15 19:40:07 +00001820
1821static PyObject *
1822posix_fchdir(PyObject *self, PyObject *fdobj)
1823{
1824 return posix_fildes(fdobj, fchdir);
1825}
1826#endif /* HAVE_FCHDIR */
1827
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001828
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001829PyDoc_STRVAR(posix_chmod__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001830"chmod(path, mode)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001831Change the access permissions of a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001832
Barry Warsaw53699e91996-12-10 23:23:01 +00001833static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001834posix_chmod(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001835{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001836 PyObject *opath = NULL;
Mark Hammondef8b6542001-05-13 08:04:26 +00001837 char *path = NULL;
Guido van Rossumffd15f52000-03-31 00:47:28 +00001838 int i;
1839 int res;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001840#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001841 DWORD attr;
Mark Hammond817c9292003-12-03 01:22:38 +00001842 if (unicode_file_names()) {
1843 PyUnicodeObject *po;
1844 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
1845 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001846 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1847 if (attr != 0xFFFFFFFF) {
1848 if (i & _S_IWRITE)
1849 attr &= ~FILE_ATTRIBUTE_READONLY;
1850 else
1851 attr |= FILE_ATTRIBUTE_READONLY;
1852 res = SetFileAttributesW(PyUnicode_AS_UNICODE(po), attr);
1853 }
1854 else
1855 res = 0;
Mark Hammond817c9292003-12-03 01:22:38 +00001856 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001857 if (!res)
1858 return win32_error_unicode("chmod",
Mark Hammond817c9292003-12-03 01:22:38 +00001859 PyUnicode_AS_UNICODE(po));
1860 Py_INCREF(Py_None);
1861 return Py_None;
1862 }
1863 /* Drop the argument parsing error as narrow strings
1864 are also valid. */
1865 PyErr_Clear();
1866 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00001867 if (!PyArg_ParseTuple(args, "O&i:chmod", PyUnicode_FSConverter,
1868 &opath, &i))
Thomas Wouters477c8d52006-05-27 19:21:47 +00001869 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001870 path = bytes2str(opath, 1);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001871 Py_BEGIN_ALLOW_THREADS
1872 attr = GetFileAttributesA(path);
1873 if (attr != 0xFFFFFFFF) {
1874 if (i & _S_IWRITE)
1875 attr &= ~FILE_ATTRIBUTE_READONLY;
1876 else
1877 attr |= FILE_ATTRIBUTE_READONLY;
1878 res = SetFileAttributesA(path, attr);
1879 }
1880 else
1881 res = 0;
1882 Py_END_ALLOW_THREADS
1883 if (!res) {
1884 win32_error("chmod", path);
Martin v. Löwis011e8422009-05-05 04:43:17 +00001885 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001886 return NULL;
1887 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00001888 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001889 Py_INCREF(Py_None);
1890 return Py_None;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001891#else /* MS_WINDOWS */
Martin v. Löwis011e8422009-05-05 04:43:17 +00001892 if (!PyArg_ParseTuple(args, "O&i:chmod", PyUnicode_FSConverter,
1893 &opath, &i))
Guido van Rossumffd15f52000-03-31 00:47:28 +00001894 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001895 path = bytes2str(opath, 1);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001896 Py_BEGIN_ALLOW_THREADS
Guido van Rossumef40e772000-03-31 01:26:23 +00001897 res = chmod(path, i);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001898 Py_END_ALLOW_THREADS
1899 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001900 return posix_error_with_allocated_filename(opath);
1901 release_bytes(opath);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001902 Py_INCREF(Py_None);
1903 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001904#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001905}
1906
Christian Heimes4e30a842007-11-30 22:12:06 +00001907#ifdef HAVE_FCHMOD
1908PyDoc_STRVAR(posix_fchmod__doc__,
1909"fchmod(fd, mode)\n\n\
1910Change the access permissions of the file given by file\n\
1911descriptor fd.");
1912
1913static PyObject *
1914posix_fchmod(PyObject *self, PyObject *args)
1915{
1916 int fd, mode, res;
1917 if (!PyArg_ParseTuple(args, "ii:fchmod", &fd, &mode))
1918 return NULL;
1919 Py_BEGIN_ALLOW_THREADS
1920 res = fchmod(fd, mode);
1921 Py_END_ALLOW_THREADS
1922 if (res < 0)
1923 return posix_error();
1924 Py_RETURN_NONE;
1925}
1926#endif /* HAVE_FCHMOD */
1927
1928#ifdef HAVE_LCHMOD
1929PyDoc_STRVAR(posix_lchmod__doc__,
1930"lchmod(path, mode)\n\n\
1931Change the access permissions of a file. If path is a symlink, this\n\
1932affects the link itself rather than the target.");
1933
1934static PyObject *
1935posix_lchmod(PyObject *self, PyObject *args)
1936{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001937 PyObject *opath;
1938 char *path;
Christian Heimes4e30a842007-11-30 22:12:06 +00001939 int i;
1940 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001941 if (!PyArg_ParseTuple(args, "O&i:lchmod", PyUnicode_FSConverter,
1942 &opath, &i))
Christian Heimes4e30a842007-11-30 22:12:06 +00001943 return NULL;
Eric Smith86a05ec2009-05-05 13:07:30 +00001944 path = bytes2str(opath, 1);
Christian Heimes4e30a842007-11-30 22:12:06 +00001945 Py_BEGIN_ALLOW_THREADS
1946 res = lchmod(path, i);
1947 Py_END_ALLOW_THREADS
1948 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001949 return posix_error_with_allocated_filename(opath);
1950 release_bytes(opath);
Christian Heimes4e30a842007-11-30 22:12:06 +00001951 Py_RETURN_NONE;
1952}
1953#endif /* HAVE_LCHMOD */
1954
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001955
Thomas Wouterscf297e42007-02-23 15:07:44 +00001956#ifdef HAVE_CHFLAGS
1957PyDoc_STRVAR(posix_chflags__doc__,
1958"chflags(path, flags)\n\n\
1959Set file flags.");
1960
1961static PyObject *
1962posix_chflags(PyObject *self, PyObject *args)
1963{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001964 PyObject *opath;
Thomas Wouterscf297e42007-02-23 15:07:44 +00001965 char *path;
1966 unsigned long flags;
1967 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001968 if (!PyArg_ParseTuple(args, "O&k:chflags",
1969 PyUnicode_FSConverter, &opath, &flags))
Thomas Wouterscf297e42007-02-23 15:07:44 +00001970 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001971 path = bytes2str(opath, 1);
Thomas Wouterscf297e42007-02-23 15:07:44 +00001972 Py_BEGIN_ALLOW_THREADS
1973 res = chflags(path, flags);
1974 Py_END_ALLOW_THREADS
1975 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001976 return posix_error_with_allocated_filename(opath);
1977 release_bytes(opath);
Thomas Wouterscf297e42007-02-23 15:07:44 +00001978 Py_INCREF(Py_None);
1979 return Py_None;
1980}
1981#endif /* HAVE_CHFLAGS */
1982
1983#ifdef HAVE_LCHFLAGS
1984PyDoc_STRVAR(posix_lchflags__doc__,
1985"lchflags(path, flags)\n\n\
1986Set file flags.\n\
1987This function will not follow symbolic links.");
1988
1989static PyObject *
1990posix_lchflags(PyObject *self, PyObject *args)
1991{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001992 PyObject *opath;
Thomas Wouterscf297e42007-02-23 15:07:44 +00001993 char *path;
1994 unsigned long flags;
1995 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001996 if (!PyArg_ParseTuple(args, "O&k:lchflags",
Martin v. Löwis4adbc342009-05-05 17:17:55 +00001997 PyUnicode_FSConverter, &opath, &flags))
Thomas Wouterscf297e42007-02-23 15:07:44 +00001998 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001999 path = bytes2str(opath, 1);
Thomas Wouterscf297e42007-02-23 15:07:44 +00002000 Py_BEGIN_ALLOW_THREADS
2001 res = lchflags(path, flags);
2002 Py_END_ALLOW_THREADS
2003 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00002004 return posix_error_with_allocated_filename(opath);
2005 release_bytes(opath);
Thomas Wouterscf297e42007-02-23 15:07:44 +00002006 Py_INCREF(Py_None);
2007 return Py_None;
2008}
2009#endif /* HAVE_LCHFLAGS */
2010
Martin v. Löwis244edc82001-10-04 22:44:26 +00002011#ifdef HAVE_CHROOT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002012PyDoc_STRVAR(posix_chroot__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002013"chroot(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002014Change root directory to path.");
Martin v. Löwis244edc82001-10-04 22:44:26 +00002015
2016static PyObject *
2017posix_chroot(PyObject *self, PyObject *args)
2018{
Martin v. Löwis011e8422009-05-05 04:43:17 +00002019 return posix_1str(args, "O&:chroot", chroot);
Martin v. Löwis244edc82001-10-04 22:44:26 +00002020}
2021#endif
2022
Guido van Rossum21142a01999-01-08 21:05:37 +00002023#ifdef HAVE_FSYNC
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002024PyDoc_STRVAR(posix_fsync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002025"fsync(fildes)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002026force write of file with filedescriptor to disk.");
Guido van Rossum21142a01999-01-08 21:05:37 +00002027
2028static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00002029posix_fsync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00002030{
Fred Drake4d1e64b2002-04-15 19:40:07 +00002031 return posix_fildes(fdobj, fsync);
Guido van Rossum21142a01999-01-08 21:05:37 +00002032}
2033#endif /* HAVE_FSYNC */
2034
2035#ifdef HAVE_FDATASYNC
Guido van Rossumecc23b02000-09-22 16:01:05 +00002036
Guido van Rossum7f58e2e2000-09-22 17:26:14 +00002037#ifdef __hpux
Guido van Rossumecc23b02000-09-22 16:01:05 +00002038extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
2039#endif
2040
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002041PyDoc_STRVAR(posix_fdatasync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002042"fdatasync(fildes)\n\n\
Guido van Rossum21142a01999-01-08 21:05:37 +00002043force write of file with filedescriptor to disk.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002044 does not force update of metadata.");
Guido van Rossum21142a01999-01-08 21:05:37 +00002045
2046static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00002047posix_fdatasync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00002048{
Fred Drake4d1e64b2002-04-15 19:40:07 +00002049 return posix_fildes(fdobj, fdatasync);
Guido van Rossum21142a01999-01-08 21:05:37 +00002050}
2051#endif /* HAVE_FDATASYNC */
2052
2053
Fredrik Lundh10723342000-07-10 16:38:09 +00002054#ifdef HAVE_CHOWN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002055PyDoc_STRVAR(posix_chown__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002056"chown(path, uid, gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002057Change the owner and group id of path to the numeric uid and gid.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002058
Barry Warsaw53699e91996-12-10 23:23:01 +00002059static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002060posix_chown(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002061{
Martin v. Löwis011e8422009-05-05 04:43:17 +00002062 PyObject *opath;
2063 char *path;
Christian Heimesd5e2b6f2008-03-19 21:50:51 +00002064 long uid, gid;
Fredrik Lundh44328e62000-07-10 15:59:30 +00002065 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002066 if (!PyArg_ParseTuple(args, "O&ll:chown",
2067 PyUnicode_FSConverter, &opath,
Mark Hammondef8b6542001-05-13 08:04:26 +00002068 &uid, &gid))
Fredrik Lundh44328e62000-07-10 15:59:30 +00002069 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002070 path = bytes2str(opath, 1);
Fredrik Lundh44328e62000-07-10 15:59:30 +00002071 Py_BEGIN_ALLOW_THREADS
2072 res = chown(path, (uid_t) uid, (gid_t) gid);
2073 Py_END_ALLOW_THREADS
2074 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00002075 return posix_error_with_allocated_filename(opath);
2076 release_bytes(opath);
Fredrik Lundh44328e62000-07-10 15:59:30 +00002077 Py_INCREF(Py_None);
2078 return Py_None;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002079}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002080#endif /* HAVE_CHOWN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002081
Christian Heimes4e30a842007-11-30 22:12:06 +00002082#ifdef HAVE_FCHOWN
2083PyDoc_STRVAR(posix_fchown__doc__,
2084"fchown(fd, uid, gid)\n\n\
2085Change the owner and group id of the file given by file descriptor\n\
2086fd to the numeric uid and gid.");
2087
2088static PyObject *
2089posix_fchown(PyObject *self, PyObject *args)
2090{
2091 int fd, uid, gid;
2092 int res;
2093 if (!PyArg_ParseTuple(args, "iii:chown", &fd, &uid, &gid))
2094 return NULL;
2095 Py_BEGIN_ALLOW_THREADS
2096 res = fchown(fd, (uid_t) uid, (gid_t) gid);
2097 Py_END_ALLOW_THREADS
2098 if (res < 0)
2099 return posix_error();
2100 Py_RETURN_NONE;
2101}
2102#endif /* HAVE_FCHOWN */
2103
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002104#ifdef HAVE_LCHOWN
2105PyDoc_STRVAR(posix_lchown__doc__,
2106"lchown(path, uid, gid)\n\n\
2107Change the owner and group id of path to the numeric uid and gid.\n\
2108This function will not follow symbolic links.");
2109
2110static PyObject *
2111posix_lchown(PyObject *self, PyObject *args)
2112{
Martin v. Löwis011e8422009-05-05 04:43:17 +00002113 PyObject *opath;
2114 char *path;
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002115 int uid, gid;
2116 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002117 if (!PyArg_ParseTuple(args, "O&ii:lchown",
2118 PyUnicode_FSConverter, &opath,
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002119 &uid, &gid))
2120 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002121 path = bytes2str(opath, 1);
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002122 Py_BEGIN_ALLOW_THREADS
2123 res = lchown(path, (uid_t) uid, (gid_t) gid);
2124 Py_END_ALLOW_THREADS
Tim Peters11b23062003-04-23 02:39:17 +00002125 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00002126 return posix_error_with_allocated_filename(opath);
2127 release_bytes(opath);
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002128 Py_INCREF(Py_None);
2129 return Py_None;
2130}
2131#endif /* HAVE_LCHOWN */
2132
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002133
Guido van Rossum36bc6801995-06-14 22:54:23 +00002134#ifdef HAVE_GETCWD
Barry Warsaw53699e91996-12-10 23:23:01 +00002135static PyObject *
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002136posix_getcwd(int use_bytes)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002137{
2138 char buf[1026];
2139 char *res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002140
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002141#ifdef MS_WINDOWS
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002142 if (!use_bytes && unicode_file_names()) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002143 wchar_t wbuf[1026];
Thomas Wouters477c8d52006-05-27 19:21:47 +00002144 wchar_t *wbuf2 = wbuf;
2145 PyObject *resobj;
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002146 DWORD len;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002147 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002148 len = GetCurrentDirectoryW(sizeof wbuf/ sizeof wbuf[0], wbuf);
2149 /* If the buffer is large enough, len does not include the
2150 terminating \0. If the buffer is too small, len includes
2151 the space needed for the terminator. */
2152 if (len >= sizeof wbuf/ sizeof wbuf[0]) {
2153 wbuf2 = malloc(len * sizeof(wchar_t));
2154 if (wbuf2)
2155 len = GetCurrentDirectoryW(len, wbuf2);
2156 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002157 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002158 if (!wbuf2) {
2159 PyErr_NoMemory();
2160 return NULL;
2161 }
2162 if (!len) {
2163 if (wbuf2 != wbuf) free(wbuf2);
2164 return win32_error("getcwdu", NULL);
2165 }
2166 resobj = PyUnicode_FromWideChar(wbuf2, len);
2167 if (wbuf2 != wbuf) free(wbuf2);
2168 return resobj;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002169 }
2170#endif
2171
2172 Py_BEGIN_ALLOW_THREADS
2173#if defined(PYOS_OS2) && defined(PYCC_GCC)
2174 res = _getcwd2(buf, sizeof buf);
2175#else
2176 res = getcwd(buf, sizeof buf);
2177#endif
2178 Py_END_ALLOW_THREADS
2179 if (res == NULL)
2180 return posix_error();
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002181 if (use_bytes)
2182 return PyBytes_FromStringAndSize(buf, strlen(buf));
Martin v. Löwis43c57782009-05-10 08:15:24 +00002183 return PyUnicode_Decode(buf, strlen(buf), Py_FileSystemDefaultEncoding,"surrogateescape");
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002184}
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002185
2186PyDoc_STRVAR(posix_getcwd__doc__,
2187"getcwd() -> path\n\n\
2188Return a unicode string representing the current working directory.");
2189
2190static PyObject *
2191posix_getcwd_unicode(PyObject *self)
2192{
2193 return posix_getcwd(0);
2194}
2195
2196PyDoc_STRVAR(posix_getcwdb__doc__,
2197"getcwdb() -> path\n\n\
2198Return a bytes string representing the current working directory.");
2199
2200static PyObject *
2201posix_getcwd_bytes(PyObject *self)
2202{
2203 return posix_getcwd(1);
2204}
Guido van Rossum36bc6801995-06-14 22:54:23 +00002205#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002206
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002207
Guido van Rossumb6775db1994-08-01 11:34:53 +00002208#ifdef HAVE_LINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002209PyDoc_STRVAR(posix_link__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002210"link(src, dst)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002211Create a hard link to a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002212
Barry Warsaw53699e91996-12-10 23:23:01 +00002213static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002214posix_link(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002215{
Martin v. Löwis011e8422009-05-05 04:43:17 +00002216 return posix_2str(args, "O&O&:link", link);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002217}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002218#endif /* HAVE_LINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002219
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002220
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002221PyDoc_STRVAR(posix_listdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002222"listdir(path) -> list_of_strings\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002223Return a list containing the names of the entries in the directory.\n\
2224\n\
2225 path: path of directory to list\n\
2226\n\
2227The list is in arbitrary order. It does not include the special\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002228entries '.' and '..' even if they are present in the directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002229
Barry Warsaw53699e91996-12-10 23:23:01 +00002230static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002231posix_listdir(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002232{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002233 /* XXX Should redo this putting the (now four) versions of opendir
Guido van Rossum6d8841c1997-08-14 19:57:39 +00002234 in separate files instead of having them all here... */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002235#if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002236
Barry Warsaw53699e91996-12-10 23:23:01 +00002237 PyObject *d, *v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002238 HANDLE hFindFile;
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002239 BOOL result;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002240 WIN32_FIND_DATA FileData;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002241 PyObject *opath;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002242 char namebuf[MAX_PATH+5]; /* Overallocate for \\*.*\0 */
Mark Hammondef8b6542001-05-13 08:04:26 +00002243 char *bufptr = namebuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002244 Py_ssize_t len = sizeof(namebuf)-5; /* only claim to have space for MAX_PATH */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002245
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002246 /* If on wide-character-capable OS see if argument
2247 is Unicode and if so use wide API. */
2248 if (unicode_file_names()) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002249 PyObject *po;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002250 if (PyArg_ParseTuple(args, "U:listdir", &po)) {
2251 WIN32_FIND_DATAW wFileData;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002252 Py_UNICODE *wnamebuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002253 /* Overallocate for \\*.*\0 */
2254 len = PyUnicode_GET_SIZE(po);
2255 wnamebuf = malloc((len + 5) * sizeof(wchar_t));
2256 if (!wnamebuf) {
2257 PyErr_NoMemory();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002258 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002259 }
2260 wcscpy(wnamebuf, PyUnicode_AS_UNICODE(po));
Hirokazu Yamamotobbb9be72009-05-04 05:56:46 +00002261 if (len > 0) {
2262 Py_UNICODE wch = wnamebuf[len-1];
2263 if (wch != L'/' && wch != L'\\' && wch != L':')
2264 wnamebuf[len++] = L'\\';
2265 wcscpy(wnamebuf + len, L"*.*");
2266 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002267 if ((d = PyList_New(0)) == NULL) {
2268 free(wnamebuf);
2269 return NULL;
2270 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002271 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
2272 if (hFindFile == INVALID_HANDLE_VALUE) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002273 int error = GetLastError();
2274 if (error == ERROR_FILE_NOT_FOUND) {
2275 free(wnamebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002276 return d;
2277 }
2278 Py_DECREF(d);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002279 win32_error_unicode("FindFirstFileW", wnamebuf);
2280 free(wnamebuf);
2281 return NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002282 }
2283 do {
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002284 /* Skip over . and .. */
2285 if (wcscmp(wFileData.cFileName, L".") != 0 &&
2286 wcscmp(wFileData.cFileName, L"..") != 0) {
2287 v = PyUnicode_FromUnicode(wFileData.cFileName, wcslen(wFileData.cFileName));
2288 if (v == NULL) {
2289 Py_DECREF(d);
2290 d = NULL;
2291 break;
2292 }
2293 if (PyList_Append(d, v) != 0) {
2294 Py_DECREF(v);
2295 Py_DECREF(d);
2296 d = NULL;
2297 break;
2298 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002299 Py_DECREF(v);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002300 }
Georg Brandl622927b2006-03-07 12:48:03 +00002301 Py_BEGIN_ALLOW_THREADS
2302 result = FindNextFileW(hFindFile, &wFileData);
2303 Py_END_ALLOW_THREADS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002304 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2305 it got to the end of the directory. */
2306 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2307 Py_DECREF(d);
2308 win32_error_unicode("FindNextFileW", wnamebuf);
2309 FindClose(hFindFile);
2310 free(wnamebuf);
2311 return NULL;
2312 }
Georg Brandl622927b2006-03-07 12:48:03 +00002313 } while (result == TRUE);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002314
2315 if (FindClose(hFindFile) == FALSE) {
2316 Py_DECREF(d);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002317 win32_error_unicode("FindClose", wnamebuf);
2318 free(wnamebuf);
2319 return NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002320 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002321 free(wnamebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002322 return d;
2323 }
2324 /* Drop the argument parsing error as narrow strings
2325 are also valid. */
2326 PyErr_Clear();
2327 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002328
Martin v. Löwis011e8422009-05-05 04:43:17 +00002329 if (!PyArg_ParseTuple(args, "O&:listdir",
2330 PyUnicode_FSConverter, &opath))
Guido van Rossumb6775db1994-08-01 11:34:53 +00002331 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002332 if (PyObject_Size(opath)+1 > MAX_PATH) {
2333 PyErr_SetString(PyExc_ValueError, "path too long");
2334 Py_DECREF(opath);
2335 return NULL;
2336 }
2337 strcpy(namebuf, bytes2str(opath, 0));
2338 len = PyObject_Size(opath);
Neil Schemenauer94b866a2002-03-22 20:51:58 +00002339 if (len > 0) {
2340 char ch = namebuf[len-1];
2341 if (ch != SEP && ch != ALTSEP && ch != ':')
2342 namebuf[len++] = '/';
Hirokazu Yamamotobbb9be72009-05-04 05:56:46 +00002343 strcpy(namebuf + len, "*.*");
Neil Schemenauer94b866a2002-03-22 20:51:58 +00002344 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002345
Barry Warsaw53699e91996-12-10 23:23:01 +00002346 if ((d = PyList_New(0)) == NULL)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002347 return NULL;
2348
2349 hFindFile = FindFirstFile(namebuf, &FileData);
2350 if (hFindFile == INVALID_HANDLE_VALUE) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002351 int error = GetLastError();
2352 if (error == ERROR_FILE_NOT_FOUND)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002353 return d;
2354 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00002355 return win32_error("FindFirstFile", namebuf);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002356 }
2357 do {
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002358 /* Skip over . and .. */
2359 if (strcmp(FileData.cFileName, ".") != 0 &&
2360 strcmp(FileData.cFileName, "..") != 0) {
Christian Heimes72b710a2008-05-26 13:28:38 +00002361 v = PyBytes_FromString(FileData.cFileName);
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002362 if (v == NULL) {
2363 Py_DECREF(d);
2364 d = NULL;
2365 break;
2366 }
2367 if (PyList_Append(d, v) != 0) {
2368 Py_DECREF(v);
2369 Py_DECREF(d);
2370 d = NULL;
2371 break;
2372 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002373 Py_DECREF(v);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002374 }
Georg Brandl622927b2006-03-07 12:48:03 +00002375 Py_BEGIN_ALLOW_THREADS
2376 result = FindNextFile(hFindFile, &FileData);
2377 Py_END_ALLOW_THREADS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002378 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2379 it got to the end of the directory. */
2380 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2381 Py_DECREF(d);
2382 win32_error("FindNextFile", namebuf);
2383 FindClose(hFindFile);
2384 return NULL;
2385 }
Georg Brandl622927b2006-03-07 12:48:03 +00002386 } while (result == TRUE);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002387
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002388 if (FindClose(hFindFile) == FALSE) {
2389 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00002390 return win32_error("FindClose", namebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002391 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002392
2393 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002394
Tim Peters0bb44a42000-09-15 07:44:49 +00002395#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002396
2397#ifndef MAX_PATH
2398#define MAX_PATH CCHMAXPATH
2399#endif
Martin v. Löwis011e8422009-05-05 04:43:17 +00002400 PyObject *oname;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002401 char *name, *pt;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00002402 Py_ssize_t len;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002403 PyObject *d, *v;
2404 char namebuf[MAX_PATH+5];
2405 HDIR hdir = 1;
2406 ULONG srchcnt = 1;
2407 FILEFINDBUF3 ep;
2408 APIRET rc;
2409
Martin v. Löwis011e8422009-05-05 04:43:17 +00002410 if (!PyArg_ParseTuple(args, "O&:listdir",
2411 PyUnicode_FSConverter, &oname))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002412 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002413 name = bytes2str(oname);
2414 len = PyObject_Size(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002415 if (len >= MAX_PATH) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002416 release_bytes(oname);
Neal Norwitz6c913782007-10-14 03:23:09 +00002417 PyErr_SetString(PyExc_ValueError, "path too long");
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002418 return NULL;
2419 }
2420 strcpy(namebuf, name);
2421 for (pt = namebuf; *pt; pt++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002422 if (*pt == ALTSEP)
2423 *pt = SEP;
2424 if (namebuf[len-1] != SEP)
2425 namebuf[len++] = SEP;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002426 strcpy(namebuf + len, "*.*");
2427
Neal Norwitz6c913782007-10-14 03:23:09 +00002428 if ((d = PyList_New(0)) == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002429 release_bytes(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002430 return NULL;
Alexandre Vassalotti4167ebc2007-10-14 02:54:41 +00002431 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002432
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002433 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
2434 &hdir, /* Handle to Use While Search Directory */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002435 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002436 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
2437 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
2438 FIL_STANDARD); /* Format of Entry (EAs or Not) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002439
2440 if (rc != NO_ERROR) {
2441 errno = ENOENT;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002442 return posix_error_with_allocated_filename(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002443 }
2444
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002445 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002446 do {
2447 if (ep.achName[0] == '.'
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002448 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002449 continue; /* Skip Over "." and ".." Names */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002450
2451 strcpy(namebuf, ep.achName);
2452
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002453 /* Leave Case of Name Alone -- In Native Form */
2454 /* (Removed Forced Lowercasing Code) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002455
Christian Heimes72b710a2008-05-26 13:28:38 +00002456 v = PyBytes_FromString(namebuf);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002457 if (v == NULL) {
2458 Py_DECREF(d);
2459 d = NULL;
2460 break;
2461 }
2462 if (PyList_Append(d, v) != 0) {
2463 Py_DECREF(v);
2464 Py_DECREF(d);
2465 d = NULL;
2466 break;
2467 }
2468 Py_DECREF(v);
2469 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
2470 }
2471
Martin v. Löwis011e8422009-05-05 04:43:17 +00002472 release_bytes(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002473 return d;
2474#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002475 PyObject *oname;
2476 char *name;
Barry Warsaw53699e91996-12-10 23:23:01 +00002477 PyObject *d, *v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002478 DIR *dirp;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002479 struct dirent *ep;
Just van Rossum96b1c902003-03-03 17:32:15 +00002480 int arg_is_unicode = 1;
2481
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002482 errno = 0;
Just van Rossum96b1c902003-03-03 17:32:15 +00002483 if (!PyArg_ParseTuple(args, "U:listdir", &v)) {
2484 arg_is_unicode = 0;
2485 PyErr_Clear();
2486 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00002487 if (!PyArg_ParseTuple(args, "O&:listdir", PyUnicode_FSConverter, &oname))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002488 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002489 name = bytes2str(oname, 1);
Guido van Rossumff4949e1992-08-05 19:58:53 +00002490 if ((dirp = opendir(name)) == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002491 return posix_error_with_allocated_filename(oname);
Guido van Rossumff4949e1992-08-05 19:58:53 +00002492 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002493 if ((d = PyList_New(0)) == NULL) {
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002494 closedir(dirp);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002495 release_bytes(oname);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002496 return NULL;
2497 }
Georg Brandl622927b2006-03-07 12:48:03 +00002498 for (;;) {
Georg Brandl3dbca812008-07-23 16:10:53 +00002499 errno = 0;
Georg Brandl622927b2006-03-07 12:48:03 +00002500 Py_BEGIN_ALLOW_THREADS
2501 ep = readdir(dirp);
2502 Py_END_ALLOW_THREADS
Georg Brandl3dbca812008-07-23 16:10:53 +00002503 if (ep == NULL) {
2504 if (errno == 0) {
2505 break;
2506 } else {
2507 closedir(dirp);
2508 Py_DECREF(d);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002509 return posix_error_with_allocated_filename(oname);
Georg Brandl3dbca812008-07-23 16:10:53 +00002510 }
2511 }
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002512 if (ep->d_name[0] == '.' &&
2513 (NAMLEN(ep) == 1 ||
Guido van Rossuma376cc51996-12-05 23:43:35 +00002514 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002515 continue;
Christian Heimes72b710a2008-05-26 13:28:38 +00002516 v = PyBytes_FromStringAndSize(ep->d_name, NAMLEN(ep));
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002517 if (v == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00002518 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002519 d = NULL;
2520 break;
2521 }
Just van Rossum96b1c902003-03-03 17:32:15 +00002522 if (arg_is_unicode) {
Just van Rossum46c97842003-02-25 21:42:15 +00002523 PyObject *w;
2524
2525 w = PyUnicode_FromEncodedObject(v,
Tim Peters11b23062003-04-23 02:39:17 +00002526 Py_FileSystemDefaultEncoding,
Martin v. Löwis43c57782009-05-10 08:15:24 +00002527 "surrogateescape");
Martin v. Löwis011e8422009-05-05 04:43:17 +00002528 Py_DECREF(v);
2529 if (w != NULL)
Just van Rossum6a421832003-03-04 19:30:44 +00002530 v = w;
Just van Rossum6a421832003-03-04 19:30:44 +00002531 else {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002532 /* Encoding failed to decode ASCII bytes.
2533 Raise exception. */
2534 Py_DECREF(d);
2535 d = NULL;
2536 break;
Just van Rossum46c97842003-02-25 21:42:15 +00002537 }
Just van Rossum46c97842003-02-25 21:42:15 +00002538 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002539 if (PyList_Append(d, v) != 0) {
2540 Py_DECREF(v);
2541 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002542 d = NULL;
2543 break;
2544 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002545 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002546 }
2547 closedir(dirp);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002548 release_bytes(oname);
Guido van Rossum0ee42cd1991-04-08 21:01:03 +00002549
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002550 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002551
Tim Peters0bb44a42000-09-15 07:44:49 +00002552#endif /* which OS */
2553} /* end of posix_listdir */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002554
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002555#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00002556/* A helper function for abspath on win32 */
2557static PyObject *
2558posix__getfullpathname(PyObject *self, PyObject *args)
2559{
Martin v. Löwis011e8422009-05-05 04:43:17 +00002560 PyObject *opath;
2561 char *path;
Mark Hammondef8b6542001-05-13 08:04:26 +00002562 char outbuf[MAX_PATH*2];
2563 char *temp;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002564#ifdef MS_WINDOWS
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002565 if (unicode_file_names()) {
2566 PyUnicodeObject *po;
2567 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po)) {
Benjamin Petersonf608c612008-11-16 18:33:53 +00002568 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
2569 Py_UNICODE woutbuf[MAX_PATH*2], *woutbufp = woutbuf;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002570 Py_UNICODE *wtemp;
Benjamin Petersonf608c612008-11-16 18:33:53 +00002571 DWORD result;
2572 PyObject *v;
2573 result = GetFullPathNameW(wpath,
2574 sizeof(woutbuf)/sizeof(woutbuf[0]),
2575 woutbuf, &wtemp);
2576 if (result > sizeof(woutbuf)/sizeof(woutbuf[0])) {
2577 woutbufp = malloc(result * sizeof(Py_UNICODE));
2578 if (!woutbufp)
2579 return PyErr_NoMemory();
2580 result = GetFullPathNameW(wpath, result, woutbufp, &wtemp);
2581 }
2582 if (result)
2583 v = PyUnicode_FromUnicode(woutbufp, wcslen(woutbufp));
2584 else
2585 v = win32_error_unicode("GetFullPathNameW", wpath);
2586 if (woutbufp != woutbuf)
2587 free(woutbufp);
2588 return v;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002589 }
2590 /* Drop the argument parsing error as narrow strings
2591 are also valid. */
2592 PyErr_Clear();
2593 }
2594#endif
Martin v. Löwis011e8422009-05-05 04:43:17 +00002595 if (!PyArg_ParseTuple (args, "O&:_getfullpathname",
2596 PyUnicode_FSConverter, &opath))
Mark Hammondef8b6542001-05-13 08:04:26 +00002597 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002598 path = bytes2str(opath, 1);
2599 if (!GetFullPathName(path, sizeof(outbuf)/sizeof(outbuf[0]),
2600 outbuf, &temp)) {
2601 win32_error("GetFullPathName", path);
2602 release_bytes(opath);
2603 return NULL;
2604 }
2605 release_bytes(opath);
Martin v. Löwis969297f2004-06-15 18:49:58 +00002606 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
2607 return PyUnicode_Decode(outbuf, strlen(outbuf),
2608 Py_FileSystemDefaultEncoding, NULL);
2609 }
Christian Heimes72b710a2008-05-26 13:28:38 +00002610 return PyBytes_FromString(outbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00002611} /* end of posix__getfullpathname */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002612#endif /* MS_WINDOWS */
Mark Hammondef8b6542001-05-13 08:04:26 +00002613
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002614PyDoc_STRVAR(posix_mkdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002615"mkdir(path [, mode=0777])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002616Create a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002617
Barry Warsaw53699e91996-12-10 23:23:01 +00002618static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002619posix_mkdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002620{
Guido van Rossumb0824db1996-02-25 04:50:32 +00002621 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002622 PyObject *opath;
2623 char *path;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002624 int mode = 0777;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002625
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002626#ifdef MS_WINDOWS
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002627 if (unicode_file_names()) {
2628 PyUnicodeObject *po;
Mark Hammond05107b62003-02-19 04:08:27 +00002629 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode)) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002630 Py_BEGIN_ALLOW_THREADS
2631 /* PyUnicode_AS_UNICODE OK without thread lock as
2632 it is a simple dereference. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002633 res = CreateDirectoryW(PyUnicode_AS_UNICODE(po), NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002634 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002635 if (!res)
2636 return win32_error_unicode("mkdir", PyUnicode_AS_UNICODE(po));
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002637 Py_INCREF(Py_None);
2638 return Py_None;
2639 }
2640 /* Drop the argument parsing error as narrow strings
2641 are also valid. */
2642 PyErr_Clear();
2643 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00002644 if (!PyArg_ParseTuple(args, "O&|i:mkdir",
2645 PyUnicode_FSConverter, &opath, &mode))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002646 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002647 path = bytes2str(opath, 1);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002648 Py_BEGIN_ALLOW_THREADS
2649 /* PyUnicode_AS_UNICODE OK without thread lock as
2650 it is a simple dereference. */
2651 res = CreateDirectoryA(path, NULL);
2652 Py_END_ALLOW_THREADS
2653 if (!res) {
2654 win32_error("mkdir", path);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002655 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002656 return NULL;
2657 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00002658 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002659 Py_INCREF(Py_None);
2660 return Py_None;
2661#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002662
Martin v. Löwis011e8422009-05-05 04:43:17 +00002663 if (!PyArg_ParseTuple(args, "O&|i:mkdir",
2664 PyUnicode_FSConverter, &opath, &mode))
Guido van Rossumb0824db1996-02-25 04:50:32 +00002665 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002666 path = bytes2str(opath, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00002667 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002668#if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002669 res = mkdir(path);
2670#else
Guido van Rossumb0824db1996-02-25 04:50:32 +00002671 res = mkdir(path, mode);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002672#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002673 Py_END_ALLOW_THREADS
Guido van Rossumb0824db1996-02-25 04:50:32 +00002674 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00002675 return posix_error_with_allocated_filename(opath);
2676 release_bytes(opath);
Barry Warsaw53699e91996-12-10 23:23:01 +00002677 Py_INCREF(Py_None);
2678 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002679#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002680}
2681
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002682
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002683/* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
2684#if defined(HAVE_SYS_RESOURCE_H)
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002685#include <sys/resource.h>
2686#endif
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002687
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002688
2689#ifdef HAVE_NICE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002690PyDoc_STRVAR(posix_nice__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002691"nice(inc) -> new_priority\n\n\
2692Decrease the priority of process by inc and return the new priority.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002693
Barry Warsaw53699e91996-12-10 23:23:01 +00002694static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002695posix_nice(PyObject *self, PyObject *args)
Guido van Rossum775f4da1993-01-09 17:18:52 +00002696{
2697 int increment, value;
2698
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002699 if (!PyArg_ParseTuple(args, "i:nice", &increment))
Guido van Rossum775f4da1993-01-09 17:18:52 +00002700 return NULL;
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002701
2702 /* There are two flavours of 'nice': one that returns the new
2703 priority (as required by almost all standards out there) and the
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002704 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
2705 the use of getpriority() to get the new priority.
Tim Peters5aa91602002-01-30 05:46:57 +00002706
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002707 If we are of the nice family that returns the new priority, we
2708 need to clear errno before the call, and check if errno is filled
2709 before calling posix_error() on a returnvalue of -1, because the
2710 -1 may be the actual new priority! */
2711
2712 errno = 0;
Guido van Rossum775f4da1993-01-09 17:18:52 +00002713 value = nice(increment);
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002714#if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002715 if (value == 0)
2716 value = getpriority(PRIO_PROCESS, 0);
2717#endif
2718 if (value == -1 && errno != 0)
2719 /* either nice() or getpriority() returned an error */
Guido van Rossum775f4da1993-01-09 17:18:52 +00002720 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00002721 return PyLong_FromLong((long) value);
Guido van Rossum775f4da1993-01-09 17:18:52 +00002722}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002723#endif /* HAVE_NICE */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002724
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002725PyDoc_STRVAR(posix_rename__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002726"rename(old, new)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002727Rename a file or directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002728
Barry Warsaw53699e91996-12-10 23:23:01 +00002729static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002730posix_rename(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002731{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002732#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002733 PyObject *o1, *o2;
2734 char *p1, *p2;
2735 BOOL result;
2736 if (unicode_file_names()) {
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +00002737 if (!PyArg_ParseTuple(args, "OO:rename", &o1, &o2))
2738 goto error;
2739 if (!convert_to_unicode(&o1))
2740 goto error;
2741 if (!convert_to_unicode(&o2)) {
2742 Py_DECREF(o1);
2743 goto error;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002744 }
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +00002745 Py_BEGIN_ALLOW_THREADS
2746 result = MoveFileW(PyUnicode_AsUnicode(o1),
2747 PyUnicode_AsUnicode(o2));
2748 Py_END_ALLOW_THREADS
2749 Py_DECREF(o1);
2750 Py_DECREF(o2);
2751 if (!result)
2752 return win32_error("rename", NULL);
2753 Py_INCREF(Py_None);
2754 return Py_None;
2755error:
2756 PyErr_Clear();
Thomas Wouters477c8d52006-05-27 19:21:47 +00002757 }
2758 if (!PyArg_ParseTuple(args, "ss:rename", &p1, &p2))
2759 return NULL;
2760 Py_BEGIN_ALLOW_THREADS
2761 result = MoveFileA(p1, p2);
2762 Py_END_ALLOW_THREADS
2763 if (!result)
2764 return win32_error("rename", NULL);
2765 Py_INCREF(Py_None);
2766 return Py_None;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002767#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002768 return posix_2str(args, "O&O&:rename", rename);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002769#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002770}
2771
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002772
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002773PyDoc_STRVAR(posix_rmdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002774"rmdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002775Remove a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002776
Barry Warsaw53699e91996-12-10 23:23:01 +00002777static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002778posix_rmdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002779{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002780#ifdef MS_WINDOWS
Amaury Forgeot d'Arcc72ef8b2008-10-03 18:38:26 +00002781 return win32_1str(args, "rmdir", "y:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002782#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002783 return posix_1str(args, "O&:rmdir", rmdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002784#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002785}
2786
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002787
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002788PyDoc_STRVAR(posix_stat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002789"stat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002790Perform a stat system call on the given path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002791
Barry Warsaw53699e91996-12-10 23:23:01 +00002792static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002793posix_stat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002794{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002795#ifdef MS_WINDOWS
Martin v. Löwis011e8422009-05-05 04:43:17 +00002796 return posix_do_stat(self, args, "O&:stat", STAT, "U:stat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002797#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002798 return posix_do_stat(self, args, "O&:stat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002799#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002800}
2801
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002802
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002803#ifdef HAVE_SYSTEM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002804PyDoc_STRVAR(posix_system__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002805"system(command) -> exit_status\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002806Execute the command (a string) in a subshell.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002807
Barry Warsaw53699e91996-12-10 23:23:01 +00002808static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002809posix_system(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002810{
Guido van Rossumff4949e1992-08-05 19:58:53 +00002811 long sts;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002812#ifdef MS_WINDOWS
2813 wchar_t *command;
2814 if (!PyArg_ParseTuple(args, "u:system", &command))
2815 return NULL;
2816#else
2817 char *command;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002818 if (!PyArg_ParseTuple(args, "s:system", &command))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002819 return NULL;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002820#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002821 Py_BEGIN_ALLOW_THREADS
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002822#ifdef MS_WINDOWS
2823 sts = _wsystem(command);
2824#else
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002825 sts = system(command);
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002826#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002827 Py_END_ALLOW_THREADS
Christian Heimes217cfd12007-12-02 14:31:20 +00002828 return PyLong_FromLong(sts);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002829}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002830#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002831
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002832
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002833PyDoc_STRVAR(posix_umask__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002834"umask(new_mask) -> old_mask\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002835Set the current numeric umask and return the previous umask.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002836
Barry Warsaw53699e91996-12-10 23:23:01 +00002837static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002838posix_umask(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002839{
2840 int i;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002841 if (!PyArg_ParseTuple(args, "i:umask", &i))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002842 return NULL;
Fred Drake0368bc42001-07-19 20:48:32 +00002843 i = (int)umask(i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002844 if (i < 0)
2845 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00002846 return PyLong_FromLong((long)i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002847}
2848
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002849
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002850PyDoc_STRVAR(posix_unlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002851"unlink(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002852Remove a file (same as remove(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002853
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002854PyDoc_STRVAR(posix_remove__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002855"remove(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002856Remove a file (same as unlink(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002857
Barry Warsaw53699e91996-12-10 23:23:01 +00002858static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002859posix_unlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002860{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002861#ifdef MS_WINDOWS
Amaury Forgeot d'Arcc72ef8b2008-10-03 18:38:26 +00002862 return win32_1str(args, "remove", "y:remove", DeleteFileA, "U:remove", DeleteFileW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002863#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002864 return posix_1str(args, "O&:remove", unlink);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002865#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002866}
2867
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002868
Guido van Rossumb6775db1994-08-01 11:34:53 +00002869#ifdef HAVE_UNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002870PyDoc_STRVAR(posix_uname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002871"uname() -> (sysname, nodename, release, version, machine)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002872Return a tuple identifying the current operating system.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002873
Barry Warsaw53699e91996-12-10 23:23:01 +00002874static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002875posix_uname(PyObject *self, PyObject *noargs)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002876{
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002877 struct utsname u;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002878 int res;
Neal Norwitze241ce82003-02-17 18:17:05 +00002879
Barry Warsaw53699e91996-12-10 23:23:01 +00002880 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002881 res = uname(&u);
Barry Warsaw53699e91996-12-10 23:23:01 +00002882 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002883 if (res < 0)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002884 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00002885 return Py_BuildValue("(sssss)",
Barry Warsaw43d68b81996-12-19 22:10:44 +00002886 u.sysname,
2887 u.nodename,
2888 u.release,
2889 u.version,
2890 u.machine);
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002891}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002892#endif /* HAVE_UNAME */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002893
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002894static int
2895extract_time(PyObject *t, long* sec, long* usec)
2896{
2897 long intval;
2898 if (PyFloat_Check(t)) {
2899 double tval = PyFloat_AsDouble(t);
Christian Heimes90aa7642007-12-19 02:45:37 +00002900 PyObject *intobj = Py_TYPE(t)->tp_as_number->nb_int(t);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002901 if (!intobj)
2902 return -1;
Christian Heimes217cfd12007-12-02 14:31:20 +00002903 intval = PyLong_AsLong(intobj);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002904 Py_DECREF(intobj);
Michael W. Hudsonb8963812005-07-05 15:21:58 +00002905 if (intval == -1 && PyErr_Occurred())
2906 return -1;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002907 *sec = intval;
Tim Peters96940cf2002-09-10 15:37:28 +00002908 *usec = (long)((tval - intval) * 1e6); /* can't exceed 1000000 */
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002909 if (*usec < 0)
2910 /* If rounding gave us a negative number,
2911 truncate. */
2912 *usec = 0;
2913 return 0;
2914 }
Christian Heimes217cfd12007-12-02 14:31:20 +00002915 intval = PyLong_AsLong(t);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002916 if (intval == -1 && PyErr_Occurred())
2917 return -1;
2918 *sec = intval;
2919 *usec = 0;
Martin v. Löwis076b2092002-09-10 15:04:41 +00002920 return 0;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002921}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002922
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002923PyDoc_STRVAR(posix_utime__doc__,
Thomas Wouters477c8d52006-05-27 19:21:47 +00002924"utime(path, (atime, mtime))\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00002925utime(path, None)\n\n\
Barry Warsaw3cef8562000-05-01 16:17:24 +00002926Set the access and modified time of the file to the given values. If the\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002927second form is used, set the access and modified times to the current time.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002928
Barry Warsaw53699e91996-12-10 23:23:01 +00002929static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002930posix_utime(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002931{
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002932#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002933 PyObject *arg;
2934 PyUnicodeObject *obwpath;
2935 wchar_t *wpath = NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002936 PyObject *oapath;
2937 char *apath;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002938 HANDLE hFile;
2939 long atimesec, mtimesec, ausec, musec;
2940 FILETIME atime, mtime;
2941 PyObject *result = NULL;
2942
2943 if (unicode_file_names()) {
2944 if (PyArg_ParseTuple(args, "UO|:utime", &obwpath, &arg)) {
2945 wpath = PyUnicode_AS_UNICODE(obwpath);
2946 Py_BEGIN_ALLOW_THREADS
2947 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002948 NULL, OPEN_EXISTING,
2949 FILE_FLAG_BACKUP_SEMANTICS, NULL);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002950 Py_END_ALLOW_THREADS
2951 if (hFile == INVALID_HANDLE_VALUE)
2952 return win32_error_unicode("utime", wpath);
2953 } else
2954 /* Drop the argument parsing error as narrow strings
2955 are also valid. */
2956 PyErr_Clear();
2957 }
2958 if (!wpath) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002959 if (!PyArg_ParseTuple(args, "O&O:utime",
2960 PyUnicode_FSConverter, &oapath, &arg))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002961 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002962 apath = bytes2str(oapath, 1);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002963 Py_BEGIN_ALLOW_THREADS
2964 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002965 NULL, OPEN_EXISTING,
2966 FILE_FLAG_BACKUP_SEMANTICS, NULL);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002967 Py_END_ALLOW_THREADS
2968 if (hFile == INVALID_HANDLE_VALUE) {
2969 win32_error("utime", apath);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002970 release_bytes(oapath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002971 return NULL;
2972 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00002973 release_bytes(oapath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002974 }
2975
2976 if (arg == Py_None) {
2977 SYSTEMTIME now;
2978 GetSystemTime(&now);
2979 if (!SystemTimeToFileTime(&now, &mtime) ||
2980 !SystemTimeToFileTime(&now, &atime)) {
2981 win32_error("utime", NULL);
2982 goto done;
2983 }
2984 }
2985 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
2986 PyErr_SetString(PyExc_TypeError,
2987 "utime() arg 2 must be a tuple (atime, mtime)");
2988 goto done;
2989 }
2990 else {
2991 if (extract_time(PyTuple_GET_ITEM(arg, 0),
2992 &atimesec, &ausec) == -1)
2993 goto done;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002994 time_t_to_FILE_TIME(atimesec, 1000*ausec, &atime);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002995 if (extract_time(PyTuple_GET_ITEM(arg, 1),
2996 &mtimesec, &musec) == -1)
2997 goto done;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002998 time_t_to_FILE_TIME(mtimesec, 1000*musec, &mtime);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002999 }
3000 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
3001 /* Avoid putting the file name into the error here,
3002 as that may confuse the user into believing that
3003 something is wrong with the file, when it also
3004 could be the time stamp that gives a problem. */
3005 win32_error("utime", NULL);
3006 }
3007 Py_INCREF(Py_None);
3008 result = Py_None;
3009done:
3010 CloseHandle(hFile);
3011 return result;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003012#else /* MS_WINDOWS */
Thomas Wouters477c8d52006-05-27 19:21:47 +00003013
Martin v. Löwis011e8422009-05-05 04:43:17 +00003014 PyObject *opath;
3015 char *path;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003016 long atime, mtime, ausec, musec;
Guido van Rossumff4949e1992-08-05 19:58:53 +00003017 int res;
Barry Warsaw3cef8562000-05-01 16:17:24 +00003018 PyObject* arg;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003019
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003020#if defined(HAVE_UTIMES)
3021 struct timeval buf[2];
3022#define ATIME buf[0].tv_sec
3023#define MTIME buf[1].tv_sec
3024#elif defined(HAVE_UTIME_H)
Guido van Rossum6d8841c1997-08-14 19:57:39 +00003025/* XXX should define struct utimbuf instead, above */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003026 struct utimbuf buf;
3027#define ATIME buf.actime
3028#define MTIME buf.modtime
3029#define UTIME_ARG &buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003030#else /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003031 time_t buf[2];
3032#define ATIME buf[0]
3033#define MTIME buf[1]
3034#define UTIME_ARG buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003035#endif /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003036
Mark Hammond817c9292003-12-03 01:22:38 +00003037
Martin v. Löwis011e8422009-05-05 04:43:17 +00003038 if (!PyArg_ParseTuple(args, "O&O:utime",
3039 PyUnicode_FSConverter, &opath, &arg))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003040 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003041 path = bytes2str(opath, 1);
Barry Warsaw3cef8562000-05-01 16:17:24 +00003042 if (arg == Py_None) {
3043 /* optional time values not given */
3044 Py_BEGIN_ALLOW_THREADS
3045 res = utime(path, NULL);
3046 Py_END_ALLOW_THREADS
3047 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003048 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
Barry Warsaw3cef8562000-05-01 16:17:24 +00003049 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003050 "utime() arg 2 must be a tuple (atime, mtime)");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003051 release_bytes(opath);
Barry Warsaw3cef8562000-05-01 16:17:24 +00003052 return NULL;
3053 }
3054 else {
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003055 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Neal Norwitz96652712004-06-06 20:40:27 +00003056 &atime, &ausec) == -1) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003057 release_bytes(opath);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003058 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00003059 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003060 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Neal Norwitz96652712004-06-06 20:40:27 +00003061 &mtime, &musec) == -1) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003062 release_bytes(opath);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003063 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00003064 }
Barry Warsaw3cef8562000-05-01 16:17:24 +00003065 ATIME = atime;
3066 MTIME = mtime;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003067#ifdef HAVE_UTIMES
3068 buf[0].tv_usec = ausec;
3069 buf[1].tv_usec = musec;
3070 Py_BEGIN_ALLOW_THREADS
3071 res = utimes(path, buf);
3072 Py_END_ALLOW_THREADS
3073#else
Barry Warsaw3cef8562000-05-01 16:17:24 +00003074 Py_BEGIN_ALLOW_THREADS
3075 res = utime(path, UTIME_ARG);
3076 Py_END_ALLOW_THREADS
Mark Hammond817c9292003-12-03 01:22:38 +00003077#endif /* HAVE_UTIMES */
Barry Warsaw3cef8562000-05-01 16:17:24 +00003078 }
Mark Hammond2d5914b2004-05-04 08:10:37 +00003079 if (res < 0) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003080 return posix_error_with_allocated_filename(opath);
Mark Hammond2d5914b2004-05-04 08:10:37 +00003081 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00003082 release_bytes(opath);
Barry Warsaw53699e91996-12-10 23:23:01 +00003083 Py_INCREF(Py_None);
3084 return Py_None;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003085#undef UTIME_ARG
3086#undef ATIME
3087#undef MTIME
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003088#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003089}
3090
Guido van Rossum85e3b011991-06-03 12:42:10 +00003091
Guido van Rossum3b066191991-06-04 19:40:25 +00003092/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003093
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003094PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003095"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003096Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003097
Barry Warsaw53699e91996-12-10 23:23:01 +00003098static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003099posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003100{
3101 int sts;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00003102 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
Guido van Rossum85e3b011991-06-03 12:42:10 +00003103 return NULL;
3104 _exit(sts);
Guido van Rossuma376cc51996-12-05 23:43:35 +00003105 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003106}
3107
Martin v. Löwis114619e2002-10-07 06:44:21 +00003108#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
3109static void
Martin v. Löwis725507b2006-03-07 12:08:51 +00003110free_string_array(char **array, Py_ssize_t count)
Martin v. Löwis114619e2002-10-07 06:44:21 +00003111{
Martin v. Löwis725507b2006-03-07 12:08:51 +00003112 Py_ssize_t i;
Martin v. Löwis114619e2002-10-07 06:44:21 +00003113 for (i = 0; i < count; i++)
3114 PyMem_Free(array[i]);
3115 PyMem_DEL(array);
3116}
Martin v. Löwis011e8422009-05-05 04:43:17 +00003117
Antoine Pitrou69f71142009-05-24 21:25:49 +00003118static
Martin v. Löwis011e8422009-05-05 04:43:17 +00003119int fsconvert_strdup(PyObject *o, char**out)
3120{
3121 PyObject *bytes;
3122 Py_ssize_t size;
3123 if (!PyUnicode_FSConverter(o, &bytes))
3124 return 0;
3125 size = PyObject_Size(bytes);
3126 *out = PyMem_Malloc(size+1);
3127 if (!*out)
3128 return 0;
3129 /* Don't lock bytes, as we hold the GIL */
3130 memcpy(*out, bytes2str(bytes, 0), size+1);
3131 Py_DECREF(bytes);
3132 return 1;
3133}
Martin v. Löwis114619e2002-10-07 06:44:21 +00003134#endif
3135
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003136
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003137#ifdef HAVE_EXECV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003138PyDoc_STRVAR(posix_execv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003139"execv(path, args)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003140Execute an executable path with arguments, replacing current process.\n\
3141\n\
3142 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003143 args: tuple or list of strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003144
Barry Warsaw53699e91996-12-10 23:23:01 +00003145static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003146posix_execv(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003147{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003148 PyObject *opath;
Guido van Rossumef0a00e1992-01-27 16:51:30 +00003149 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00003150 PyObject *argv;
Guido van Rossum85e3b011991-06-03 12:42:10 +00003151 char **argvlist;
Martin v. Löwis725507b2006-03-07 12:08:51 +00003152 Py_ssize_t i, argc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003153 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003154
Guido van Rossum89b33251993-10-22 14:26:06 +00003155 /* execv has two arguments: (path, argv), where
Guido van Rossum85e3b011991-06-03 12:42:10 +00003156 argv is a list or tuple of strings. */
3157
Martin v. Löwis011e8422009-05-05 04:43:17 +00003158 if (!PyArg_ParseTuple(args, "O&O:execv",
3159 PyUnicode_FSConverter,
3160 &opath, &argv))
Guido van Rossum85e3b011991-06-03 12:42:10 +00003161 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003162 path = bytes2str(opath, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00003163 if (PyList_Check(argv)) {
3164 argc = PyList_Size(argv);
3165 getitem = PyList_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00003166 }
Barry Warsaw53699e91996-12-10 23:23:01 +00003167 else if (PyTuple_Check(argv)) {
3168 argc = PyTuple_Size(argv);
3169 getitem = PyTuple_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00003170 }
Guido van Rossumef0a00e1992-01-27 16:51:30 +00003171 else {
Fred Drake661ea262000-10-24 19:57:45 +00003172 PyErr_SetString(PyExc_TypeError, "execv() arg 2 must be a tuple or list");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003173 release_bytes(opath);
Guido van Rossum50422b42000-04-26 20:34:28 +00003174 return NULL;
3175 }
Thomas Heller6790d602007-08-30 17:15:14 +00003176 if (argc < 1) {
3177 PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003178 release_bytes(opath);
Thomas Heller6790d602007-08-30 17:15:14 +00003179 return NULL;
3180 }
Guido van Rossum50422b42000-04-26 20:34:28 +00003181
Barry Warsaw53699e91996-12-10 23:23:01 +00003182 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003183 if (argvlist == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003184 release_bytes(opath);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00003185 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003186 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00003187 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003188 if (!fsconvert_strdup((*getitem)(argv, i),
3189 &argvlist[i])) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003190 free_string_array(argvlist, i);
Tim Peters5aa91602002-01-30 05:46:57 +00003191 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003192 "execv() arg 2 must contain only strings");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003193 release_bytes(opath);
Guido van Rossum50422b42000-04-26 20:34:28 +00003194 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00003195
Guido van Rossum85e3b011991-06-03 12:42:10 +00003196 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00003197 }
3198 argvlist[argc] = NULL;
3199
Guido van Rossumef0a00e1992-01-27 16:51:30 +00003200 execv(path, argvlist);
Guido van Rossumb6775db1994-08-01 11:34:53 +00003201
Guido van Rossum85e3b011991-06-03 12:42:10 +00003202 /* If we get here it's definitely an error */
3203
Martin v. Löwis114619e2002-10-07 06:44:21 +00003204 free_string_array(argvlist, argc);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003205 release_bytes(opath);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003206 return posix_error();
3207}
3208
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003209
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003210PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003211"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003212Execute a path with arguments and environment, replacing current process.\n\
3213\n\
3214 path: path of executable file\n\
3215 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003216 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003217
Barry Warsaw53699e91996-12-10 23:23:01 +00003218static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003219posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003220{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003221 PyObject *opath;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003222 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00003223 PyObject *argv, *env;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003224 char **argvlist;
3225 char **envlist;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003226 PyObject *key, *val, *keys=NULL, *vals=NULL;
Martin v. Löwis725507b2006-03-07 12:08:51 +00003227 Py_ssize_t i, pos, argc, envc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003228 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Thomas Wouters477c8d52006-05-27 19:21:47 +00003229 Py_ssize_t lastarg = 0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003230
3231 /* execve has three arguments: (path, argv, env), where
3232 argv is a list or tuple of strings and env is a dictionary
3233 like posix.environ. */
3234
Martin v. Löwis011e8422009-05-05 04:43:17 +00003235 if (!PyArg_ParseTuple(args, "O&OO:execve",
3236 PyUnicode_FSConverter,
3237 &opath, &argv, &env))
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003238 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003239 path = bytes2str(opath, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00003240 if (PyList_Check(argv)) {
3241 argc = PyList_Size(argv);
3242 getitem = PyList_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003243 }
Barry Warsaw53699e91996-12-10 23:23:01 +00003244 else if (PyTuple_Check(argv)) {
3245 argc = PyTuple_Size(argv);
3246 getitem = PyTuple_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003247 }
3248 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003249 PyErr_SetString(PyExc_TypeError,
3250 "execve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003251 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003252 }
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003253 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003254 PyErr_SetString(PyExc_TypeError,
3255 "execve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003256 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003257 }
3258
Barry Warsaw53699e91996-12-10 23:23:01 +00003259 argvlist = PyMem_NEW(char *, argc+1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003260 if (argvlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003261 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003262 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003263 }
3264 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003265 if (!fsconvert_strdup((*getitem)(argv, i),
3266 &argvlist[i]))
Barry Warsaw43d68b81996-12-19 22:10:44 +00003267 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003268 lastarg = i;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003269 goto fail_1;
3270 }
3271 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003272 lastarg = argc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003273 argvlist[argc] = NULL;
3274
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003275 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003276 if (i < 0)
3277 goto fail_1;
Barry Warsaw53699e91996-12-10 23:23:01 +00003278 envlist = PyMem_NEW(char *, i + 1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003279 if (envlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003280 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003281 goto fail_1;
3282 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003283 envc = 0;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003284 keys = PyMapping_Keys(env);
3285 vals = PyMapping_Values(env);
3286 if (!keys || !vals)
3287 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003288 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3289 PyErr_SetString(PyExc_TypeError,
3290 "execve(): env.keys() or env.values() is not a list");
3291 goto fail_2;
3292 }
Tim Peters5aa91602002-01-30 05:46:57 +00003293
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003294 for (pos = 0; pos < i; pos++) {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003295 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003296 size_t len;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003297
3298 key = PyList_GetItem(keys, pos);
3299 val = PyList_GetItem(vals, pos);
3300 if (!key || !val)
3301 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003302
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003303 if (!PyArg_Parse(
3304 key,
3305 "s;execve() arg 3 contains a non-string key",
3306 &k) ||
3307 !PyArg_Parse(
3308 val,
3309 "s;execve() arg 3 contains a non-string value",
3310 &v))
Barry Warsaw43d68b81996-12-19 22:10:44 +00003311 {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003312 goto fail_2;
3313 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00003314
3315#if defined(PYOS_OS2)
3316 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
3317 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
3318#endif
Christian Heimes830a4bc2007-11-22 07:43:40 +00003319 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Tim Petersc8996f52001-12-03 20:41:00 +00003320 p = PyMem_NEW(char, len);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003321 if (p == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003322 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003323 goto fail_2;
3324 }
Tim Petersc8996f52001-12-03 20:41:00 +00003325 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003326 envlist[envc++] = p;
Guido van Rossumd48f2521997-12-05 22:19:34 +00003327#if defined(PYOS_OS2)
3328 }
3329#endif
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003330 }
3331 envlist[envc] = 0;
3332
3333 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00003334
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003335 /* If we get here it's definitely an error */
3336
3337 (void) posix_error();
3338
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003339 fail_2:
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003340 while (--envc >= 0)
Barry Warsaw53699e91996-12-10 23:23:01 +00003341 PyMem_DEL(envlist[envc]);
3342 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003343 fail_1:
3344 free_string_array(argvlist, lastarg);
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003345 Py_XDECREF(vals);
3346 Py_XDECREF(keys);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003347 fail_0:
Martin v. Löwis011e8422009-05-05 04:43:17 +00003348 release_bytes(opath);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003349 return NULL;
3350}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003351#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003352
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003353
Guido van Rossuma1065681999-01-25 23:20:23 +00003354#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003355PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003356"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003357Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003358\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003359 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003360 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003361 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003362
3363static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003364posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003365{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003366 PyObject *opath;
Guido van Rossuma1065681999-01-25 23:20:23 +00003367 char *path;
3368 PyObject *argv;
3369 char **argvlist;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003370 int mode, i;
3371 Py_ssize_t argc;
Tim Peters79248aa2001-08-29 21:37:10 +00003372 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003373 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossuma1065681999-01-25 23:20:23 +00003374
3375 /* spawnv has three arguments: (mode, path, argv), where
3376 argv is a list or tuple of strings. */
3377
Martin v. Löwis011e8422009-05-05 04:43:17 +00003378 if (!PyArg_ParseTuple(args, "iO&O:spawnv", &mode,
3379 PyUnicode_FSConverter,
3380 &opath, &argv))
Guido van Rossuma1065681999-01-25 23:20:23 +00003381 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003382 path = bytes2str(opath, 1);
Guido van Rossuma1065681999-01-25 23:20:23 +00003383 if (PyList_Check(argv)) {
3384 argc = PyList_Size(argv);
3385 getitem = PyList_GetItem;
3386 }
3387 else if (PyTuple_Check(argv)) {
3388 argc = PyTuple_Size(argv);
3389 getitem = PyTuple_GetItem;
3390 }
3391 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003392 PyErr_SetString(PyExc_TypeError,
3393 "spawnv() arg 2 must be a tuple or list");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003394 release_bytes(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00003395 return NULL;
3396 }
3397
3398 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003399 if (argvlist == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003400 release_bytes(opath);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00003401 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003402 }
Guido van Rossuma1065681999-01-25 23:20:23 +00003403 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003404 if (!fsconvert_strdup((*getitem)(argv, i),
3405 &argvlist[i])) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003406 free_string_array(argvlist, i);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003407 PyErr_SetString(
3408 PyExc_TypeError,
3409 "spawnv() arg 2 must contain only strings");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003410 release_bytes(opath);
Fred Drake137507e2000-06-01 02:02:46 +00003411 return NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00003412 }
3413 }
3414 argvlist[argc] = NULL;
3415
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003416#if defined(PYOS_OS2) && defined(PYCC_GCC)
3417 Py_BEGIN_ALLOW_THREADS
3418 spawnval = spawnv(mode, path, argvlist);
3419 Py_END_ALLOW_THREADS
3420#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003421 if (mode == _OLD_P_OVERLAY)
3422 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00003423
Tim Peters25059d32001-12-07 20:35:43 +00003424 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003425 spawnval = _spawnv(mode, path, argvlist);
Tim Peters25059d32001-12-07 20:35:43 +00003426 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003427#endif
Tim Peters5aa91602002-01-30 05:46:57 +00003428
Martin v. Löwis114619e2002-10-07 06:44:21 +00003429 free_string_array(argvlist, argc);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003430 release_bytes(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00003431
Fred Drake699f3522000-06-29 21:12:41 +00003432 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003433 return posix_error();
3434 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003435#if SIZEOF_LONG == SIZEOF_VOID_P
3436 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003437#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003438 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003439#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003440}
3441
3442
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003443PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003444"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003445Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003446\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003447 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003448 path: path of executable file\n\
3449 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003450 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003451
3452static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003453posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003454{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003455 PyObject *opath;
Guido van Rossuma1065681999-01-25 23:20:23 +00003456 char *path;
3457 PyObject *argv, *env;
3458 char **argvlist;
3459 char **envlist;
3460 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003461 int mode, pos, envc;
3462 Py_ssize_t argc, i;
Tim Peters79248aa2001-08-29 21:37:10 +00003463 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003464 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Thomas Wouters477c8d52006-05-27 19:21:47 +00003465 Py_ssize_t lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003466
3467 /* spawnve has four arguments: (mode, path, argv, env), where
3468 argv is a list or tuple of strings and env is a dictionary
3469 like posix.environ. */
3470
Martin v. Löwis011e8422009-05-05 04:43:17 +00003471 if (!PyArg_ParseTuple(args, "iO&OO:spawnve", &mode,
3472 PyUnicode_FSConverter,
3473 &opath, &argv, &env))
Guido van Rossuma1065681999-01-25 23:20:23 +00003474 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003475 path = bytes2str(opath, 1);
Guido van Rossuma1065681999-01-25 23:20:23 +00003476 if (PyList_Check(argv)) {
3477 argc = PyList_Size(argv);
3478 getitem = PyList_GetItem;
3479 }
3480 else if (PyTuple_Check(argv)) {
3481 argc = PyTuple_Size(argv);
3482 getitem = PyTuple_GetItem;
3483 }
3484 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003485 PyErr_SetString(PyExc_TypeError,
3486 "spawnve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003487 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003488 }
3489 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003490 PyErr_SetString(PyExc_TypeError,
3491 "spawnve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003492 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003493 }
3494
3495 argvlist = PyMem_NEW(char *, argc+1);
3496 if (argvlist == NULL) {
3497 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003498 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003499 }
3500 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003501 if (!fsconvert_strdup((*getitem)(argv, i),
3502 &argvlist[i]))
Guido van Rossuma1065681999-01-25 23:20:23 +00003503 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003504 lastarg = i;
Guido van Rossuma1065681999-01-25 23:20:23 +00003505 goto fail_1;
3506 }
3507 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003508 lastarg = argc;
Guido van Rossuma1065681999-01-25 23:20:23 +00003509 argvlist[argc] = NULL;
3510
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003511 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003512 if (i < 0)
3513 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00003514 envlist = PyMem_NEW(char *, i + 1);
3515 if (envlist == NULL) {
3516 PyErr_NoMemory();
3517 goto fail_1;
3518 }
3519 envc = 0;
3520 keys = PyMapping_Keys(env);
3521 vals = PyMapping_Values(env);
3522 if (!keys || !vals)
3523 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003524 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3525 PyErr_SetString(PyExc_TypeError,
3526 "spawnve(): env.keys() or env.values() is not a list");
3527 goto fail_2;
3528 }
Tim Peters5aa91602002-01-30 05:46:57 +00003529
Guido van Rossuma1065681999-01-25 23:20:23 +00003530 for (pos = 0; pos < i; pos++) {
3531 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003532 size_t len;
Guido van Rossuma1065681999-01-25 23:20:23 +00003533
3534 key = PyList_GetItem(keys, pos);
3535 val = PyList_GetItem(vals, pos);
3536 if (!key || !val)
3537 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003538
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003539 if (!PyArg_Parse(
3540 key,
3541 "s;spawnve() arg 3 contains a non-string key",
3542 &k) ||
3543 !PyArg_Parse(
3544 val,
3545 "s;spawnve() arg 3 contains a non-string value",
3546 &v))
Guido van Rossuma1065681999-01-25 23:20:23 +00003547 {
3548 goto fail_2;
3549 }
Christian Heimes830a4bc2007-11-22 07:43:40 +00003550 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Tim Petersc8996f52001-12-03 20:41:00 +00003551 p = PyMem_NEW(char, len);
Guido van Rossuma1065681999-01-25 23:20:23 +00003552 if (p == NULL) {
3553 PyErr_NoMemory();
3554 goto fail_2;
3555 }
Tim Petersc8996f52001-12-03 20:41:00 +00003556 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossuma1065681999-01-25 23:20:23 +00003557 envlist[envc++] = p;
3558 }
3559 envlist[envc] = 0;
3560
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003561#if defined(PYOS_OS2) && defined(PYCC_GCC)
3562 Py_BEGIN_ALLOW_THREADS
3563 spawnval = spawnve(mode, path, argvlist, envlist);
3564 Py_END_ALLOW_THREADS
3565#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003566 if (mode == _OLD_P_OVERLAY)
3567 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00003568
3569 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003570 spawnval = _spawnve(mode, path, argvlist, envlist);
Tim Peters25059d32001-12-07 20:35:43 +00003571 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003572#endif
Tim Peters25059d32001-12-07 20:35:43 +00003573
Fred Drake699f3522000-06-29 21:12:41 +00003574 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003575 (void) posix_error();
3576 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003577#if SIZEOF_LONG == SIZEOF_VOID_P
3578 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003579#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003580 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003581#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003582
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003583 fail_2:
Guido van Rossuma1065681999-01-25 23:20:23 +00003584 while (--envc >= 0)
3585 PyMem_DEL(envlist[envc]);
3586 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003587 fail_1:
Martin v. Löwis114619e2002-10-07 06:44:21 +00003588 free_string_array(argvlist, lastarg);
Guido van Rossuma1065681999-01-25 23:20:23 +00003589 Py_XDECREF(vals);
3590 Py_XDECREF(keys);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003591 fail_0:
Martin v. Löwis011e8422009-05-05 04:43:17 +00003592 release_bytes(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00003593 return res;
3594}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003595
3596/* OS/2 supports spawnvp & spawnvpe natively */
3597#if defined(PYOS_OS2)
3598PyDoc_STRVAR(posix_spawnvp__doc__,
3599"spawnvp(mode, file, args)\n\n\
3600Execute the program 'file' in a new process, using the environment\n\
3601search path to find the file.\n\
3602\n\
3603 mode: mode of process creation\n\
3604 file: executable file name\n\
3605 args: tuple or list of strings");
3606
3607static PyObject *
3608posix_spawnvp(PyObject *self, PyObject *args)
3609{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003610 PyObject *opath;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003611 char *path;
3612 PyObject *argv;
3613 char **argvlist;
3614 int mode, i, argc;
3615 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003616 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003617
3618 /* spawnvp has three arguments: (mode, path, argv), where
3619 argv is a list or tuple of strings. */
3620
Martin v. Löwis011e8422009-05-05 04:43:17 +00003621 if (!PyArg_ParseTuple(args, "iO&O:spawnvp", &mode,
3622 PyUnicode_FSConverter,
3623 &opath, &argv))
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003624 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003625 path = bytes2str(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003626 if (PyList_Check(argv)) {
3627 argc = PyList_Size(argv);
3628 getitem = PyList_GetItem;
3629 }
3630 else if (PyTuple_Check(argv)) {
3631 argc = PyTuple_Size(argv);
3632 getitem = PyTuple_GetItem;
3633 }
3634 else {
3635 PyErr_SetString(PyExc_TypeError,
3636 "spawnvp() arg 2 must be a tuple or list");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003637 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003638 return NULL;
3639 }
3640
3641 argvlist = PyMem_NEW(char *, argc+1);
3642 if (argvlist == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003643 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003644 return PyErr_NoMemory();
3645 }
3646 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003647 if (!fsconvert_strdup((*getitem)(argv, i),
3648 &argvlist[i])) {
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003649 free_string_array(argvlist, i);
3650 PyErr_SetString(
3651 PyExc_TypeError,
3652 "spawnvp() arg 2 must contain only strings");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003653 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003654 return NULL;
3655 }
3656 }
3657 argvlist[argc] = NULL;
3658
3659 Py_BEGIN_ALLOW_THREADS
3660#if defined(PYCC_GCC)
3661 spawnval = spawnvp(mode, path, argvlist);
3662#else
3663 spawnval = _spawnvp(mode, path, argvlist);
3664#endif
3665 Py_END_ALLOW_THREADS
3666
3667 free_string_array(argvlist, argc);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003668 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003669
3670 if (spawnval == -1)
3671 return posix_error();
3672 else
3673 return Py_BuildValue("l", (long) spawnval);
3674}
3675
3676
3677PyDoc_STRVAR(posix_spawnvpe__doc__,
3678"spawnvpe(mode, file, args, env)\n\n\
3679Execute the program 'file' in a new process, using the environment\n\
3680search path to find the file.\n\
3681\n\
3682 mode: mode of process creation\n\
3683 file: executable file name\n\
3684 args: tuple or list of arguments\n\
3685 env: dictionary of strings mapping to strings");
3686
3687static PyObject *
3688posix_spawnvpe(PyObject *self, PyObject *args)
3689{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003690 PyObject *opath
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003691 char *path;
3692 PyObject *argv, *env;
3693 char **argvlist;
3694 char **envlist;
3695 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
3696 int mode, i, pos, argc, envc;
3697 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003698 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003699 int lastarg = 0;
3700
3701 /* spawnvpe has four arguments: (mode, path, argv, env), where
3702 argv is a list or tuple of strings and env is a dictionary
3703 like posix.environ. */
3704
3705 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
Martin v. Löwis011e8422009-05-05 04:43:17 +00003706 PyUnicode_FSConverter,
3707 &opath, &argv, &env))
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003708 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003709 path = bytes2str(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003710 if (PyList_Check(argv)) {
3711 argc = PyList_Size(argv);
3712 getitem = PyList_GetItem;
3713 }
3714 else if (PyTuple_Check(argv)) {
3715 argc = PyTuple_Size(argv);
3716 getitem = PyTuple_GetItem;
3717 }
3718 else {
3719 PyErr_SetString(PyExc_TypeError,
3720 "spawnvpe() arg 2 must be a tuple or list");
3721 goto fail_0;
3722 }
3723 if (!PyMapping_Check(env)) {
3724 PyErr_SetString(PyExc_TypeError,
3725 "spawnvpe() arg 3 must be a mapping object");
3726 goto fail_0;
3727 }
3728
3729 argvlist = PyMem_NEW(char *, argc+1);
3730 if (argvlist == NULL) {
3731 PyErr_NoMemory();
3732 goto fail_0;
3733 }
3734 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003735 if (!fsconvert_strdup((*getitem)(argv, i),
3736 &argvlist[i]))
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003737 {
3738 lastarg = i;
3739 goto fail_1;
3740 }
3741 }
3742 lastarg = argc;
3743 argvlist[argc] = NULL;
3744
3745 i = PyMapping_Size(env);
3746 if (i < 0)
3747 goto fail_1;
3748 envlist = PyMem_NEW(char *, i + 1);
3749 if (envlist == NULL) {
3750 PyErr_NoMemory();
3751 goto fail_1;
3752 }
3753 envc = 0;
3754 keys = PyMapping_Keys(env);
3755 vals = PyMapping_Values(env);
3756 if (!keys || !vals)
3757 goto fail_2;
3758 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3759 PyErr_SetString(PyExc_TypeError,
3760 "spawnvpe(): env.keys() or env.values() is not a list");
3761 goto fail_2;
3762 }
3763
3764 for (pos = 0; pos < i; pos++) {
3765 char *p, *k, *v;
3766 size_t len;
3767
3768 key = PyList_GetItem(keys, pos);
3769 val = PyList_GetItem(vals, pos);
3770 if (!key || !val)
3771 goto fail_2;
3772
3773 if (!PyArg_Parse(
3774 key,
3775 "s;spawnvpe() arg 3 contains a non-string key",
3776 &k) ||
3777 !PyArg_Parse(
3778 val,
3779 "s;spawnvpe() arg 3 contains a non-string value",
3780 &v))
3781 {
3782 goto fail_2;
3783 }
Christian Heimes830a4bc2007-11-22 07:43:40 +00003784 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003785 p = PyMem_NEW(char, len);
3786 if (p == NULL) {
3787 PyErr_NoMemory();
3788 goto fail_2;
3789 }
3790 PyOS_snprintf(p, len, "%s=%s", k, v);
3791 envlist[envc++] = p;
3792 }
3793 envlist[envc] = 0;
3794
3795 Py_BEGIN_ALLOW_THREADS
3796#if defined(PYCC_GCC)
Christian Heimes292d3512008-02-03 16:51:08 +00003797 spawnval = spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003798#else
Christian Heimes292d3512008-02-03 16:51:08 +00003799 spawnval = _spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003800#endif
3801 Py_END_ALLOW_THREADS
3802
3803 if (spawnval == -1)
3804 (void) posix_error();
3805 else
3806 res = Py_BuildValue("l", (long) spawnval);
3807
3808 fail_2:
3809 while (--envc >= 0)
3810 PyMem_DEL(envlist[envc]);
3811 PyMem_DEL(envlist);
3812 fail_1:
3813 free_string_array(argvlist, lastarg);
3814 Py_XDECREF(vals);
3815 Py_XDECREF(keys);
3816 fail_0:
Martin v. Löwis011e8422009-05-05 04:43:17 +00003817 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003818 return res;
3819}
3820#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00003821#endif /* HAVE_SPAWNV */
3822
3823
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003824#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003825PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003826"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003827Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
3828\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003829Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003830
3831static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003832posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003833{
Christian Heimes400adb02008-02-01 08:12:03 +00003834 pid_t pid = fork1();
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003835 if (pid == -1)
3836 return posix_error();
Georg Brandl2ee470f2008-07-16 12:55:28 +00003837 if (pid == 0)
3838 PyOS_AfterFork();
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00003839 return PyLong_FromPid(pid);
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003840}
3841#endif
3842
3843
Guido van Rossumad0ee831995-03-01 10:34:45 +00003844#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003845PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003846"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003847Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003848Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003849
Barry Warsaw53699e91996-12-10 23:23:01 +00003850static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003851posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003852{
Christian Heimes400adb02008-02-01 08:12:03 +00003853 pid_t pid = fork();
Guido van Rossum85e3b011991-06-03 12:42:10 +00003854 if (pid == -1)
3855 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003856 if (pid == 0)
3857 PyOS_AfterFork();
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00003858 return PyLong_FromPid(pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003859}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003860#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003861
Neal Norwitzb59798b2003-03-21 01:43:31 +00003862/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00003863/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
3864#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00003865#define DEV_PTY_FILE "/dev/ptc"
3866#define HAVE_DEV_PTMX
3867#else
3868#define DEV_PTY_FILE "/dev/ptmx"
3869#endif
3870
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003871#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003872#ifdef HAVE_PTY_H
3873#include <pty.h>
3874#else
3875#ifdef HAVE_LIBUTIL_H
3876#include <libutil.h>
Fred Drake8cef4cf2000-06-28 16:40:38 +00003877#endif /* HAVE_LIBUTIL_H */
3878#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00003879#ifdef HAVE_STROPTS_H
3880#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003881#endif
3882#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003883
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003884#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003885PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003886"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003887Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003888
3889static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003890posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003891{
3892 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003893#ifndef HAVE_OPENPTY
3894 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003895#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003896#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003897 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003898#ifdef sun
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003899 extern char *ptsname(int fildes);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003900#endif
3901#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00003902
Thomas Wouters70c21a12000-07-14 14:28:33 +00003903#ifdef HAVE_OPENPTY
Fred Drake8cef4cf2000-06-28 16:40:38 +00003904 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
3905 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003906#elif defined(HAVE__GETPTY)
Thomas Wouters70c21a12000-07-14 14:28:33 +00003907 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
3908 if (slave_name == NULL)
3909 return posix_error();
3910
3911 slave_fd = open(slave_name, O_RDWR);
3912 if (slave_fd < 0)
3913 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003914#else
Neal Norwitzb59798b2003-03-21 01:43:31 +00003915 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003916 if (master_fd < 0)
3917 return posix_error();
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003918 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003919 /* change permission of slave */
3920 if (grantpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003921 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003922 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003923 }
3924 /* unlock slave */
3925 if (unlockpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003926 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003927 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003928 }
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003929 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003930 slave_name = ptsname(master_fd); /* get name of slave */
3931 if (slave_name == NULL)
3932 return posix_error();
3933 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
3934 if (slave_fd < 0)
3935 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003936#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003937 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
3938 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00003939#ifndef __hpux
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003940 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00003941#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003942#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00003943#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00003944
Fred Drake8cef4cf2000-06-28 16:40:38 +00003945 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00003946
Fred Drake8cef4cf2000-06-28 16:40:38 +00003947}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003948#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003949
3950#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003951PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003952"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00003953Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
3954Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003955To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003956
3957static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003958posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003959{
Christian Heimes400adb02008-02-01 08:12:03 +00003960 int master_fd = -1;
3961 pid_t pid;
Tim Peters5aa91602002-01-30 05:46:57 +00003962
Fred Drake8cef4cf2000-06-28 16:40:38 +00003963 pid = forkpty(&master_fd, NULL, NULL, NULL);
3964 if (pid == -1)
3965 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003966 if (pid == 0)
3967 PyOS_AfterFork();
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00003968 return Py_BuildValue("(Ni)", PyLong_FromPid(pid), master_fd);
Fred Drake8cef4cf2000-06-28 16:40:38 +00003969}
3970#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003971
Guido van Rossumad0ee831995-03-01 10:34:45 +00003972#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003973PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003974"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003975Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003976
Barry Warsaw53699e91996-12-10 23:23:01 +00003977static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003978posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003979{
Christian Heimes217cfd12007-12-02 14:31:20 +00003980 return PyLong_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003981}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003982#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003983
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003984
Guido van Rossumad0ee831995-03-01 10:34:45 +00003985#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003986PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003987"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003988Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003989
Barry Warsaw53699e91996-12-10 23:23:01 +00003990static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003991posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003992{
Christian Heimes217cfd12007-12-02 14:31:20 +00003993 return PyLong_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003994}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003995#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003996
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003997
Guido van Rossumad0ee831995-03-01 10:34:45 +00003998#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003999PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004000"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004001Return the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004002
Barry Warsaw53699e91996-12-10 23:23:01 +00004003static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004004posix_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00004005{
Christian Heimes217cfd12007-12-02 14:31:20 +00004006 return PyLong_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00004007}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004008#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00004009
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004010
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004011PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004012"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004013Return the current process id");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004014
Barry Warsaw53699e91996-12-10 23:23:01 +00004015static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004016posix_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004017{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004018 return PyLong_FromPid(getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00004019}
4020
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004021
Fred Drakec9680921999-12-13 16:37:25 +00004022#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004023PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004024"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004025Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00004026
4027static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004028posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00004029{
4030 PyObject *result = NULL;
4031
Fred Drakec9680921999-12-13 16:37:25 +00004032#ifdef NGROUPS_MAX
4033#define MAX_GROUPS NGROUPS_MAX
4034#else
4035 /* defined to be 16 on Solaris7, so this should be a small number */
4036#define MAX_GROUPS 64
4037#endif
4038 gid_t grouplist[MAX_GROUPS];
4039 int n;
4040
4041 n = getgroups(MAX_GROUPS, grouplist);
4042 if (n < 0)
4043 posix_error();
4044 else {
4045 result = PyList_New(n);
4046 if (result != NULL) {
Fred Drakec9680921999-12-13 16:37:25 +00004047 int i;
4048 for (i = 0; i < n; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +00004049 PyObject *o = PyLong_FromLong((long)grouplist[i]);
Fred Drakec9680921999-12-13 16:37:25 +00004050 if (o == NULL) {
4051 Py_DECREF(result);
4052 result = NULL;
4053 break;
4054 }
4055 PyList_SET_ITEM(result, i, o);
4056 }
4057 }
4058 }
Neal Norwitze241ce82003-02-17 18:17:05 +00004059
Fred Drakec9680921999-12-13 16:37:25 +00004060 return result;
4061}
4062#endif
4063
Martin v. Löwis606edc12002-06-13 21:09:11 +00004064#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00004065PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004066"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00004067Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00004068
4069static PyObject *
4070posix_getpgid(PyObject *self, PyObject *args)
4071{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004072 pid_t pid, pgid;
4073 if (!PyArg_ParseTuple(args, PARSE_PID ":getpgid", &pid))
Martin v. Löwis606edc12002-06-13 21:09:11 +00004074 return NULL;
4075 pgid = getpgid(pid);
4076 if (pgid < 0)
4077 return posix_error();
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004078 return PyLong_FromPid(pgid);
Martin v. Löwis606edc12002-06-13 21:09:11 +00004079}
4080#endif /* HAVE_GETPGID */
4081
4082
Guido van Rossumb6775db1994-08-01 11:34:53 +00004083#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004084PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004085"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004086Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004087
Barry Warsaw53699e91996-12-10 23:23:01 +00004088static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004089posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00004090{
Guido van Rossumb6775db1994-08-01 11:34:53 +00004091#ifdef GETPGRP_HAVE_ARG
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004092 return PyLong_FromPid(getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004093#else /* GETPGRP_HAVE_ARG */
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004094 return PyLong_FromPid(getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004095#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00004096}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004097#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00004098
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004099
Guido van Rossumb6775db1994-08-01 11:34:53 +00004100#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004101PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004102"setpgrp()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004103Make this process a session leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004104
Barry Warsaw53699e91996-12-10 23:23:01 +00004105static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004106posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004107{
Guido van Rossum64933891994-10-20 21:56:42 +00004108#ifdef SETPGRP_HAVE_ARG
Guido van Rossumc2670a01992-09-13 20:07:29 +00004109 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00004110#else /* SETPGRP_HAVE_ARG */
Guido van Rossumb6775db1994-08-01 11:34:53 +00004111 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00004112#endif /* SETPGRP_HAVE_ARG */
Guido van Rossum687dd131993-05-17 08:34:16 +00004113 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004114 Py_INCREF(Py_None);
4115 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004116}
4117
Guido van Rossumb6775db1994-08-01 11:34:53 +00004118#endif /* HAVE_SETPGRP */
4119
Guido van Rossumad0ee831995-03-01 10:34:45 +00004120#ifdef HAVE_GETPPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004121PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004122"getppid() -> ppid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004123Return the parent's process id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004124
Barry Warsaw53699e91996-12-10 23:23:01 +00004125static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004126posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004127{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004128 return PyLong_FromPid(getppid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00004129}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004130#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004131
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004132
Fred Drake12c6e2d1999-12-14 21:25:03 +00004133#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004134PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004135"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004136Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00004137
4138static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004139posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00004140{
Neal Norwitze241ce82003-02-17 18:17:05 +00004141 PyObject *result = NULL;
Fred Drakea30680b2000-12-06 21:24:28 +00004142 char *name;
4143 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00004144
Fred Drakea30680b2000-12-06 21:24:28 +00004145 errno = 0;
4146 name = getlogin();
4147 if (name == NULL) {
4148 if (errno)
4149 posix_error();
4150 else
4151 PyErr_SetString(PyExc_OSError,
Fred Drakee63544f2000-12-06 21:45:33 +00004152 "unable to determine login name");
Fred Drakea30680b2000-12-06 21:24:28 +00004153 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00004154 else
Neal Norwitz93c56822007-08-26 07:10:06 +00004155 result = PyUnicode_FromString(name);
Fred Drakea30680b2000-12-06 21:24:28 +00004156 errno = old_errno;
Neal Norwitze241ce82003-02-17 18:17:05 +00004157
Fred Drake12c6e2d1999-12-14 21:25:03 +00004158 return result;
4159}
4160#endif
4161
Guido van Rossumad0ee831995-03-01 10:34:45 +00004162#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004163PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004164"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004165Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004166
Barry Warsaw53699e91996-12-10 23:23:01 +00004167static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004168posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00004169{
Christian Heimes217cfd12007-12-02 14:31:20 +00004170 return PyLong_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00004171}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004172#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00004173
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004174
Guido van Rossumad0ee831995-03-01 10:34:45 +00004175#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004176PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004177"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004178Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004179
Barry Warsaw53699e91996-12-10 23:23:01 +00004180static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004181posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004182{
Christian Heimes292d3512008-02-03 16:51:08 +00004183 pid_t pid;
4184 int sig;
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004185 if (!PyArg_ParseTuple(args, PARSE_PID "i:kill", &pid, &sig))
Guido van Rossum85e3b011991-06-03 12:42:10 +00004186 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004187#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004188 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
4189 APIRET rc;
4190 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004191 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004192
4193 } else if (sig == XCPT_SIGNAL_KILLPROC) {
4194 APIRET rc;
4195 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004196 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004197
4198 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00004199 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004200#else
Guido van Rossum85e3b011991-06-03 12:42:10 +00004201 if (kill(pid, sig) == -1)
4202 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004203#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00004204 Py_INCREF(Py_None);
4205 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00004206}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004207#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004208
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004209#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004210PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004211"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004212Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004213
4214static PyObject *
4215posix_killpg(PyObject *self, PyObject *args)
4216{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004217 int sig;
4218 pid_t pgid;
4219 /* XXX some man pages make the `pgid` parameter an int, others
4220 a pid_t. Since getpgrp() returns a pid_t, we assume killpg should
4221 take the same type. Moreover, pid_t is always at least as wide as
4222 int (else compilation of this module fails), which is safe. */
4223 if (!PyArg_ParseTuple(args, PARSE_PID "i:killpg", &pgid, &sig))
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004224 return NULL;
4225 if (killpg(pgid, sig) == -1)
4226 return posix_error();
4227 Py_INCREF(Py_None);
4228 return Py_None;
4229}
4230#endif
4231
Guido van Rossumc0125471996-06-28 18:55:32 +00004232#ifdef HAVE_PLOCK
4233
4234#ifdef HAVE_SYS_LOCK_H
4235#include <sys/lock.h>
4236#endif
4237
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004238PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004239"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004240Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004241
Barry Warsaw53699e91996-12-10 23:23:01 +00004242static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004243posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00004244{
4245 int op;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004246 if (!PyArg_ParseTuple(args, "i:plock", &op))
Guido van Rossumc0125471996-06-28 18:55:32 +00004247 return NULL;
4248 if (plock(op) == -1)
4249 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004250 Py_INCREF(Py_None);
4251 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00004252}
4253#endif
4254
Guido van Rossumb6775db1994-08-01 11:34:53 +00004255#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004256PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004257"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004258Set the current process's user id.");
4259
Barry Warsaw53699e91996-12-10 23:23:01 +00004260static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004261posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004262{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004263 long uid_arg;
4264 uid_t uid;
4265 if (!PyArg_ParseTuple(args, "l:setuid", &uid_arg))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004266 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004267 uid = uid_arg;
4268 if (uid != uid_arg) {
4269 PyErr_SetString(PyExc_OverflowError, "user id too big");
4270 return NULL;
4271 }
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004272 if (setuid(uid) < 0)
4273 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004274 Py_INCREF(Py_None);
4275 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004276}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004277#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004278
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004279
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004280#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004281PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004282"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004283Set the current process's effective user id.");
4284
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004285static PyObject *
4286posix_seteuid (PyObject *self, PyObject *args)
4287{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004288 long euid_arg;
4289 uid_t euid;
4290 if (!PyArg_ParseTuple(args, "l", &euid_arg))
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004291 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004292 euid = euid_arg;
4293 if (euid != euid_arg) {
4294 PyErr_SetString(PyExc_OverflowError, "user id too big");
4295 return NULL;
4296 }
4297 if (seteuid(euid) < 0) {
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004298 return posix_error();
4299 } else {
4300 Py_INCREF(Py_None);
4301 return Py_None;
4302 }
4303}
4304#endif /* HAVE_SETEUID */
4305
4306#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004307PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004308"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004309Set the current process's effective group id.");
4310
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004311static PyObject *
4312posix_setegid (PyObject *self, PyObject *args)
4313{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004314 long egid_arg;
4315 gid_t egid;
4316 if (!PyArg_ParseTuple(args, "l", &egid_arg))
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004317 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004318 egid = egid_arg;
4319 if (egid != egid_arg) {
4320 PyErr_SetString(PyExc_OverflowError, "group id too big");
4321 return NULL;
4322 }
4323 if (setegid(egid) < 0) {
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004324 return posix_error();
4325 } else {
4326 Py_INCREF(Py_None);
4327 return Py_None;
4328 }
4329}
4330#endif /* HAVE_SETEGID */
4331
4332#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004333PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004334"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004335Set the current process's real and effective user ids.");
4336
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004337static PyObject *
4338posix_setreuid (PyObject *self, PyObject *args)
4339{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004340 long ruid_arg, euid_arg;
4341 uid_t ruid, euid;
4342 if (!PyArg_ParseTuple(args, "ll", &ruid_arg, &euid_arg))
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004343 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004344 ruid = ruid_arg;
4345 euid = euid_arg;
4346 if (euid != euid_arg || ruid != ruid_arg) {
4347 PyErr_SetString(PyExc_OverflowError, "user id too big");
4348 return NULL;
4349 }
4350 if (setreuid(ruid, euid) < 0) {
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004351 return posix_error();
4352 } else {
4353 Py_INCREF(Py_None);
4354 return Py_None;
4355 }
4356}
4357#endif /* HAVE_SETREUID */
4358
4359#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004360PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004361"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004362Set the current process's real and effective group ids.");
4363
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004364static PyObject *
4365posix_setregid (PyObject *self, PyObject *args)
4366{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004367 long rgid_arg, egid_arg;
4368 gid_t rgid, egid;
4369 if (!PyArg_ParseTuple(args, "ll", &rgid_arg, &egid_arg))
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004370 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004371 rgid = rgid_arg;
4372 egid = egid_arg;
4373 if (egid != egid_arg || rgid != rgid_arg) {
4374 PyErr_SetString(PyExc_OverflowError, "group id too big");
4375 return NULL;
4376 }
4377 if (setregid(rgid, egid) < 0) {
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004378 return posix_error();
4379 } else {
4380 Py_INCREF(Py_None);
4381 return Py_None;
4382 }
4383}
4384#endif /* HAVE_SETREGID */
4385
Guido van Rossumb6775db1994-08-01 11:34:53 +00004386#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004387PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004388"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004389Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004390
Barry Warsaw53699e91996-12-10 23:23:01 +00004391static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004392posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004393{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004394 long gid_arg;
4395 gid_t gid;
4396 if (!PyArg_ParseTuple(args, "l:setgid", &gid_arg))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004397 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004398 gid = gid_arg;
4399 if (gid != gid_arg) {
4400 PyErr_SetString(PyExc_OverflowError, "group id too big");
4401 return NULL;
4402 }
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004403 if (setgid(gid) < 0)
4404 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004405 Py_INCREF(Py_None);
4406 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004407}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004408#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004409
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004410#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004411PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004412"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004413Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004414
4415static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00004416posix_setgroups(PyObject *self, PyObject *groups)
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004417{
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004418 int i, len;
4419 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00004420
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004421 if (!PySequence_Check(groups)) {
4422 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
4423 return NULL;
4424 }
4425 len = PySequence_Size(groups);
4426 if (len > MAX_GROUPS) {
4427 PyErr_SetString(PyExc_ValueError, "too many groups");
4428 return NULL;
4429 }
4430 for(i = 0; i < len; i++) {
4431 PyObject *elem;
4432 elem = PySequence_GetItem(groups, i);
4433 if (!elem)
4434 return NULL;
Guido van Rossumddefaf32007-01-14 03:31:43 +00004435 if (!PyLong_Check(elem)) {
4436 PyErr_SetString(PyExc_TypeError,
4437 "groups must be integers");
4438 Py_DECREF(elem);
4439 return NULL;
4440 } else {
4441 unsigned long x = PyLong_AsUnsignedLong(elem);
4442 if (PyErr_Occurred()) {
4443 PyErr_SetString(PyExc_TypeError,
4444 "group id too big");
Georg Brandla13c2442005-11-22 19:30:31 +00004445 Py_DECREF(elem);
4446 return NULL;
Georg Brandla13c2442005-11-22 19:30:31 +00004447 }
Georg Brandla13c2442005-11-22 19:30:31 +00004448 grouplist[i] = x;
Guido van Rossumddefaf32007-01-14 03:31:43 +00004449 /* read back the value to see if it fitted in gid_t */
Georg Brandla13c2442005-11-22 19:30:31 +00004450 if (grouplist[i] != x) {
4451 PyErr_SetString(PyExc_TypeError,
4452 "group id too big");
4453 Py_DECREF(elem);
4454 return NULL;
4455 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004456 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004457 Py_DECREF(elem);
4458 }
4459
4460 if (setgroups(len, grouplist) < 0)
4461 return posix_error();
4462 Py_INCREF(Py_None);
4463 return Py_None;
4464}
4465#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004466
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004467#if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
4468static PyObject *
Christian Heimes292d3512008-02-03 16:51:08 +00004469wait_helper(pid_t pid, int status, struct rusage *ru)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004470{
4471 PyObject *result;
4472 static PyObject *struct_rusage;
4473
4474 if (pid == -1)
4475 return posix_error();
4476
4477 if (struct_rusage == NULL) {
Christian Heimes072c0f12008-01-03 23:01:04 +00004478 PyObject *m = PyImport_ImportModuleNoBlock("resource");
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004479 if (m == NULL)
4480 return NULL;
4481 struct_rusage = PyObject_GetAttrString(m, "struct_rusage");
4482 Py_DECREF(m);
4483 if (struct_rusage == NULL)
4484 return NULL;
4485 }
4486
4487 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
4488 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
4489 if (!result)
4490 return NULL;
4491
4492#ifndef doubletime
4493#define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
4494#endif
4495
4496 PyStructSequence_SET_ITEM(result, 0,
4497 PyFloat_FromDouble(doubletime(ru->ru_utime)));
4498 PyStructSequence_SET_ITEM(result, 1,
4499 PyFloat_FromDouble(doubletime(ru->ru_stime)));
4500#define SET_INT(result, index, value)\
Christian Heimes217cfd12007-12-02 14:31:20 +00004501 PyStructSequence_SET_ITEM(result, index, PyLong_FromLong(value))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004502 SET_INT(result, 2, ru->ru_maxrss);
4503 SET_INT(result, 3, ru->ru_ixrss);
4504 SET_INT(result, 4, ru->ru_idrss);
4505 SET_INT(result, 5, ru->ru_isrss);
4506 SET_INT(result, 6, ru->ru_minflt);
4507 SET_INT(result, 7, ru->ru_majflt);
4508 SET_INT(result, 8, ru->ru_nswap);
4509 SET_INT(result, 9, ru->ru_inblock);
4510 SET_INT(result, 10, ru->ru_oublock);
4511 SET_INT(result, 11, ru->ru_msgsnd);
4512 SET_INT(result, 12, ru->ru_msgrcv);
4513 SET_INT(result, 13, ru->ru_nsignals);
4514 SET_INT(result, 14, ru->ru_nvcsw);
4515 SET_INT(result, 15, ru->ru_nivcsw);
4516#undef SET_INT
4517
4518 if (PyErr_Occurred()) {
4519 Py_DECREF(result);
4520 return NULL;
4521 }
4522
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004523 return Py_BuildValue("NiN", PyLong_FromPid(pid), status, result);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004524}
4525#endif /* HAVE_WAIT3 || HAVE_WAIT4 */
4526
4527#ifdef HAVE_WAIT3
4528PyDoc_STRVAR(posix_wait3__doc__,
4529"wait3(options) -> (pid, status, rusage)\n\n\
4530Wait for completion of a child process.");
4531
4532static PyObject *
4533posix_wait3(PyObject *self, PyObject *args)
4534{
Christian Heimes292d3512008-02-03 16:51:08 +00004535 pid_t pid;
4536 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004537 struct rusage ru;
4538 WAIT_TYPE status;
4539 WAIT_STATUS_INT(status) = 0;
4540
4541 if (!PyArg_ParseTuple(args, "i:wait3", &options))
4542 return NULL;
4543
4544 Py_BEGIN_ALLOW_THREADS
4545 pid = wait3(&status, options, &ru);
4546 Py_END_ALLOW_THREADS
4547
4548 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
4549}
4550#endif /* HAVE_WAIT3 */
4551
4552#ifdef HAVE_WAIT4
4553PyDoc_STRVAR(posix_wait4__doc__,
4554"wait4(pid, options) -> (pid, status, rusage)\n\n\
4555Wait for completion of a given child process.");
4556
4557static PyObject *
4558posix_wait4(PyObject *self, PyObject *args)
4559{
Christian Heimes292d3512008-02-03 16:51:08 +00004560 pid_t pid;
4561 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004562 struct rusage ru;
4563 WAIT_TYPE status;
4564 WAIT_STATUS_INT(status) = 0;
4565
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004566 if (!PyArg_ParseTuple(args, PARSE_PID "i:wait4", &pid, &options))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004567 return NULL;
4568
4569 Py_BEGIN_ALLOW_THREADS
4570 pid = wait4(pid, &status, options, &ru);
4571 Py_END_ALLOW_THREADS
4572
4573 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
4574}
4575#endif /* HAVE_WAIT4 */
4576
Guido van Rossumb6775db1994-08-01 11:34:53 +00004577#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004578PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004579"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004580Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004581
Barry Warsaw53699e91996-12-10 23:23:01 +00004582static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004583posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004584{
Christian Heimes292d3512008-02-03 16:51:08 +00004585 pid_t pid;
4586 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004587 WAIT_TYPE status;
4588 WAIT_STATUS_INT(status) = 0;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004589
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004590 if (!PyArg_ParseTuple(args, PARSE_PID "i:waitpid", &pid, &options))
Guido van Rossum21803b81992-08-09 12:55:27 +00004591 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004592 Py_BEGIN_ALLOW_THREADS
Guido van Rossume6a3aa61999-02-01 16:15:30 +00004593 pid = waitpid(pid, &status, options);
Barry Warsaw53699e91996-12-10 23:23:01 +00004594 Py_END_ALLOW_THREADS
Guido van Rossum85e3b011991-06-03 12:42:10 +00004595 if (pid == -1)
4596 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004597
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004598 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum21803b81992-08-09 12:55:27 +00004599}
4600
Tim Petersab034fa2002-02-01 11:27:43 +00004601#elif defined(HAVE_CWAIT)
4602
4603/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004604PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004605"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004606"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00004607
4608static PyObject *
4609posix_waitpid(PyObject *self, PyObject *args)
4610{
Thomas Wouters477c8d52006-05-27 19:21:47 +00004611 Py_intptr_t pid;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004612 int status, options;
Tim Petersab034fa2002-02-01 11:27:43 +00004613
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004614 if (!PyArg_ParseTuple(args, PARSE_PID "i:waitpid", &pid, &options))
Tim Petersab034fa2002-02-01 11:27:43 +00004615 return NULL;
4616 Py_BEGIN_ALLOW_THREADS
4617 pid = _cwait(&status, pid, options);
4618 Py_END_ALLOW_THREADS
4619 if (pid == -1)
4620 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004621
4622 /* shift the status left a byte so this is more like the POSIX waitpid */
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004623 return Py_BuildValue("Ni", PyLong_FromPid(pid), status << 8);
Tim Petersab034fa2002-02-01 11:27:43 +00004624}
4625#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004626
Guido van Rossumad0ee831995-03-01 10:34:45 +00004627#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004628PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004629"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004630Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004631
Barry Warsaw53699e91996-12-10 23:23:01 +00004632static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004633posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00004634{
Christian Heimes292d3512008-02-03 16:51:08 +00004635 pid_t pid;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004636 WAIT_TYPE status;
4637 WAIT_STATUS_INT(status) = 0;
Neal Norwitze241ce82003-02-17 18:17:05 +00004638
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004639 Py_BEGIN_ALLOW_THREADS
4640 pid = wait(&status);
Barry Warsaw53699e91996-12-10 23:23:01 +00004641 Py_END_ALLOW_THREADS
Guido van Rossum21803b81992-08-09 12:55:27 +00004642 if (pid == -1)
4643 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004644
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004645 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum85e3b011991-06-03 12:42:10 +00004646}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004647#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004648
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004649
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004650PyDoc_STRVAR(posix_lstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004651"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004652Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004653
Barry Warsaw53699e91996-12-10 23:23:01 +00004654static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004655posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004656{
Guido van Rossumb6775db1994-08-01 11:34:53 +00004657#ifdef HAVE_LSTAT
Martin v. Löwis011e8422009-05-05 04:43:17 +00004658 return posix_do_stat(self, args, "O&:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004659#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004660#ifdef MS_WINDOWS
Martin v. Löwis011e8422009-05-05 04:43:17 +00004661 return posix_do_stat(self, args, "O&:lstat", STAT, "U:lstat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004662#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00004663 return posix_do_stat(self, args, "O&:lstat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004664#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00004665#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004666}
4667
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004668
Guido van Rossumb6775db1994-08-01 11:34:53 +00004669#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004670PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004671"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004672Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004673
Barry Warsaw53699e91996-12-10 23:23:01 +00004674static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004675posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004676{
Thomas Wouters89f507f2006-12-13 04:49:30 +00004677 PyObject* v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00004678 char buf[MAXPATHLEN];
Martin v. Löwis011e8422009-05-05 04:43:17 +00004679 PyObject *opath;
Guido van Rossumef0a00e1992-01-27 16:51:30 +00004680 char *path;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004681 int n;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004682 int arg_is_unicode = 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004683
Martin v. Löwis011e8422009-05-05 04:43:17 +00004684 if (!PyArg_ParseTuple(args, "O&:readlink",
4685 PyUnicode_FSConverter, &opath))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004686 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00004687 path = bytes2str(opath, 1);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004688 v = PySequence_GetItem(args, 0);
Neal Norwitzcda5c062007-08-12 17:09:36 +00004689 if (v == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00004690 release_bytes(opath);
Neal Norwitzcda5c062007-08-12 17:09:36 +00004691 return NULL;
4692 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004693
4694 if (PyUnicode_Check(v)) {
4695 arg_is_unicode = 1;
4696 }
4697 Py_DECREF(v);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004698
Barry Warsaw53699e91996-12-10 23:23:01 +00004699 Py_BEGIN_ALLOW_THREADS
Guido van Rossum50e61dc1992-03-27 17:22:31 +00004700 n = readlink(path, buf, (int) sizeof buf);
Barry Warsaw53699e91996-12-10 23:23:01 +00004701 Py_END_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004702 if (n < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00004703 return posix_error_with_allocated_filename(opath);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004704
Martin v. Löwis011e8422009-05-05 04:43:17 +00004705 release_bytes(opath);
Christian Heimes72b710a2008-05-26 13:28:38 +00004706 v = PyBytes_FromStringAndSize(buf, n);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004707 if (arg_is_unicode) {
4708 PyObject *w;
4709
4710 w = PyUnicode_FromEncodedObject(v,
4711 Py_FileSystemDefaultEncoding,
Martin v. Löwis43c57782009-05-10 08:15:24 +00004712 "surrogateescape");
Thomas Wouters89f507f2006-12-13 04:49:30 +00004713 if (w != NULL) {
4714 Py_DECREF(v);
4715 v = w;
4716 }
4717 else {
Guido van Rossumf0af3e32008-10-02 18:55:37 +00004718 v = NULL;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004719 }
4720 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004721 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004722}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004723#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004724
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004725
Guido van Rossumb6775db1994-08-01 11:34:53 +00004726#ifdef HAVE_SYMLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004727PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004728"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00004729Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004730
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004731static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004732posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004733{
Martin v. Löwis011e8422009-05-05 04:43:17 +00004734 return posix_2str(args, "O&O&:symlink", symlink);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004735}
4736#endif /* HAVE_SYMLINK */
4737
4738
4739#ifdef HAVE_TIMES
Guido van Rossumd48f2521997-12-05 22:19:34 +00004740#if defined(PYCC_VACPP) && defined(PYOS_OS2)
4741static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00004742system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004743{
4744 ULONG value = 0;
4745
4746 Py_BEGIN_ALLOW_THREADS
4747 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
4748 Py_END_ALLOW_THREADS
4749
4750 return value;
4751}
4752
4753static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004754posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004755{
Guido van Rossumd48f2521997-12-05 22:19:34 +00004756 /* Currently Only Uptime is Provided -- Others Later */
4757 return Py_BuildValue("ddddd",
4758 (double)0 /* t.tms_utime / HZ */,
4759 (double)0 /* t.tms_stime / HZ */,
4760 (double)0 /* t.tms_cutime / HZ */,
4761 (double)0 /* t.tms_cstime / HZ */,
4762 (double)system_uptime() / 1000);
4763}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004764#else /* not OS2 */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00004765#define NEED_TICKS_PER_SECOND
4766static long ticks_per_second = -1;
Barry Warsaw53699e91996-12-10 23:23:01 +00004767static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004768posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00004769{
4770 struct tms t;
4771 clock_t c;
Guido van Rossum22db57e1992-04-05 14:25:30 +00004772 errno = 0;
4773 c = times(&t);
Guido van Rossum687dd131993-05-17 08:34:16 +00004774 if (c == (clock_t) -1)
4775 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004776 return Py_BuildValue("ddddd",
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00004777 (double)t.tms_utime / ticks_per_second,
4778 (double)t.tms_stime / ticks_per_second,
4779 (double)t.tms_cutime / ticks_per_second,
4780 (double)t.tms_cstime / ticks_per_second,
4781 (double)c / ticks_per_second);
Guido van Rossum22db57e1992-04-05 14:25:30 +00004782}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004783#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00004784#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004785
4786
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004787#ifdef MS_WINDOWS
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004788#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00004789static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004790posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004791{
4792 FILETIME create, exit, kernel, user;
4793 HANDLE hProc;
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004794 hProc = GetCurrentProcess();
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004795 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
4796 /* The fields of a FILETIME structure are the hi and lo part
4797 of a 64-bit value expressed in 100 nanosecond units.
4798 1e7 is one second in such units; 1e-7 the inverse.
4799 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
4800 */
Barry Warsaw53699e91996-12-10 23:23:01 +00004801 return Py_BuildValue(
4802 "ddddd",
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004803 (double)(user.dwHighDateTime*429.4967296 +
4804 user.dwLowDateTime*1e-7),
Christian Heimes68f5fbe2008-02-14 08:27:37 +00004805 (double)(kernel.dwHighDateTime*429.4967296 +
4806 kernel.dwLowDateTime*1e-7),
Barry Warsaw53699e91996-12-10 23:23:01 +00004807 (double)0,
4808 (double)0,
4809 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004810}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004811#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004812
4813#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004814PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004815"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004816Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004817#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00004818
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004819
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004820#ifdef HAVE_GETSID
4821PyDoc_STRVAR(posix_getsid__doc__,
4822"getsid(pid) -> sid\n\n\
4823Call the system call getsid().");
4824
4825static PyObject *
4826posix_getsid(PyObject *self, PyObject *args)
4827{
Christian Heimes292d3512008-02-03 16:51:08 +00004828 pid_t pid;
4829 int sid;
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004830 if (!PyArg_ParseTuple(args, PARSE_PID ":getsid", &pid))
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004831 return NULL;
4832 sid = getsid(pid);
4833 if (sid < 0)
4834 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004835 return PyLong_FromLong((long)sid);
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004836}
4837#endif /* HAVE_GETSID */
4838
4839
Guido van Rossumb6775db1994-08-01 11:34:53 +00004840#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004841PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004842"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004843Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004844
Barry Warsaw53699e91996-12-10 23:23:01 +00004845static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004846posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004847{
Guido van Rossum687dd131993-05-17 08:34:16 +00004848 if (setsid() < 0)
4849 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004850 Py_INCREF(Py_None);
4851 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004852}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004853#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00004854
Guido van Rossumb6775db1994-08-01 11:34:53 +00004855#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004856PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004857"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004858Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004859
Barry Warsaw53699e91996-12-10 23:23:01 +00004860static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004861posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004862{
Christian Heimes292d3512008-02-03 16:51:08 +00004863 pid_t pid;
4864 int pgrp;
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004865 if (!PyArg_ParseTuple(args, PARSE_PID "i:setpgid", &pid, &pgrp))
Guido van Rossumc2670a01992-09-13 20:07:29 +00004866 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004867 if (setpgid(pid, pgrp) < 0)
4868 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004869 Py_INCREF(Py_None);
4870 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004871}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004872#endif /* HAVE_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00004873
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004874
Guido van Rossumb6775db1994-08-01 11:34:53 +00004875#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004876PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004877"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004878Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004879
Barry Warsaw53699e91996-12-10 23:23:01 +00004880static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004881posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00004882{
Christian Heimes15ebc882008-02-04 18:48:49 +00004883 int fd;
4884 pid_t pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004885 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
Guido van Rossum7066dd71992-09-17 17:54:56 +00004886 return NULL;
4887 pgid = tcgetpgrp(fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004888 if (pgid < 0)
4889 return posix_error();
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004890 return PyLong_FromPid(pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00004891}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004892#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00004893
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004894
Guido van Rossumb6775db1994-08-01 11:34:53 +00004895#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004896PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004897"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004898Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004899
Barry Warsaw53699e91996-12-10 23:23:01 +00004900static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004901posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00004902{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004903 int fd;
4904 pid_t pgid;
4905 if (!PyArg_ParseTuple(args, "i" PARSE_PID ":tcsetpgrp", &fd, &pgid))
Guido van Rossum7066dd71992-09-17 17:54:56 +00004906 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004907 if (tcsetpgrp(fd, pgid) < 0)
4908 return posix_error();
Barry Warsaw43d68b81996-12-19 22:10:44 +00004909 Py_INCREF(Py_None);
Barry Warsaw53699e91996-12-10 23:23:01 +00004910 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00004911}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004912#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00004913
Guido van Rossum687dd131993-05-17 08:34:16 +00004914/* Functions acting on file descriptors */
4915
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004916PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004917"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004918Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004919
Barry Warsaw53699e91996-12-10 23:23:01 +00004920static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004921posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004922{
Martin v. Löwis011e8422009-05-05 04:43:17 +00004923 PyObject *ofile;
4924 char *file;
Guido van Rossum687dd131993-05-17 08:34:16 +00004925 int flag;
4926 int mode = 0777;
4927 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004928
4929#ifdef MS_WINDOWS
4930 if (unicode_file_names()) {
4931 PyUnicodeObject *po;
4932 if (PyArg_ParseTuple(args, "Ui|i:mkdir", &po, &flag, &mode)) {
4933 Py_BEGIN_ALLOW_THREADS
4934 /* PyUnicode_AS_UNICODE OK without thread
4935 lock as it is a simple dereference. */
4936 fd = _wopen(PyUnicode_AS_UNICODE(po), flag, mode);
4937 Py_END_ALLOW_THREADS
4938 if (fd < 0)
4939 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004940 return PyLong_FromLong((long)fd);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004941 }
4942 /* Drop the argument parsing error as narrow strings
4943 are also valid. */
4944 PyErr_Clear();
4945 }
4946#endif
4947
Martin v. Löwis011e8422009-05-05 04:43:17 +00004948 if (!PyArg_ParseTuple(args, "O&i|i",
4949 PyUnicode_FSConverter, &ofile,
Mark Hammondef8b6542001-05-13 08:04:26 +00004950 &flag, &mode))
Barry Warsaw43d68b81996-12-19 22:10:44 +00004951 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00004952 file = bytes2str(ofile, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00004953 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004954 fd = open(file, flag, mode);
Barry Warsaw53699e91996-12-10 23:23:01 +00004955 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004956 if (fd < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00004957 return posix_error_with_allocated_filename(ofile);
4958 release_bytes(ofile);
Christian Heimes217cfd12007-12-02 14:31:20 +00004959 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004960}
4961
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004962
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004963PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004964"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004965Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004966
Barry Warsaw53699e91996-12-10 23:23:01 +00004967static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004968posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004969{
4970 int fd, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004971 if (!PyArg_ParseTuple(args, "i:close", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00004972 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00004973 if (!_PyVerify_fd(fd))
4974 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004975 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004976 res = close(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00004977 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004978 if (res < 0)
4979 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004980 Py_INCREF(Py_None);
4981 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00004982}
4983
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004984
Christian Heimesfdab48e2008-01-20 09:06:41 +00004985PyDoc_STRVAR(posix_closerange__doc__,
4986"closerange(fd_low, fd_high)\n\n\
4987Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
4988
4989static PyObject *
4990posix_closerange(PyObject *self, PyObject *args)
4991{
4992 int fd_from, fd_to, i;
4993 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
4994 return NULL;
4995 Py_BEGIN_ALLOW_THREADS
4996 for (i = fd_from; i < fd_to; i++)
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00004997 if (_PyVerify_fd(i))
4998 close(i);
Christian Heimesfdab48e2008-01-20 09:06:41 +00004999 Py_END_ALLOW_THREADS
5000 Py_RETURN_NONE;
5001}
5002
5003
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005004PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005005"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005006Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005007
Barry Warsaw53699e91996-12-10 23:23:01 +00005008static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005009posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005010{
5011 int fd;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005012 if (!PyArg_ParseTuple(args, "i:dup", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00005013 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005014 if (!_PyVerify_fd(fd))
5015 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005016 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005017 fd = dup(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00005018 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005019 if (fd < 0)
5020 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00005021 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00005022}
5023
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005024
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005025PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00005026"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005027Duplicate file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005028
Barry Warsaw53699e91996-12-10 23:23:01 +00005029static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005030posix_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005031{
5032 int fd, fd2, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005033 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
Guido van Rossum687dd131993-05-17 08:34:16 +00005034 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005035 if (!_PyVerify_fd_dup2(fd, fd2))
5036 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005037 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005038 res = dup2(fd, fd2);
Barry Warsaw53699e91996-12-10 23:23:01 +00005039 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005040 if (res < 0)
5041 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005042 Py_INCREF(Py_None);
5043 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00005044}
5045
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005046
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005047PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005048"lseek(fd, pos, how) -> newpos\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005049Set the current position of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005050
Barry Warsaw53699e91996-12-10 23:23:01 +00005051static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005052posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005053{
5054 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005055#if defined(MS_WIN64) || defined(MS_WINDOWS)
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005056 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00005057#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00005058 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00005059#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00005060 PyObject *posobj;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005061 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Guido van Rossum687dd131993-05-17 08:34:16 +00005062 return NULL;
5063#ifdef SEEK_SET
5064 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
5065 switch (how) {
5066 case 0: how = SEEK_SET; break;
5067 case 1: how = SEEK_CUR; break;
5068 case 2: how = SEEK_END; break;
5069 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005070#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00005071
5072#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005073 pos = PyLong_AsLong(posobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005074#else
5075 pos = PyLong_Check(posobj) ?
Christian Heimes217cfd12007-12-02 14:31:20 +00005076 PyLong_AsLongLong(posobj) : PyLong_AsLong(posobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005077#endif
5078 if (PyErr_Occurred())
5079 return NULL;
5080
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005081 if (!_PyVerify_fd(fd))
5082 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005083 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005084#if defined(MS_WIN64) || defined(MS_WINDOWS)
Fred Drake699f3522000-06-29 21:12:41 +00005085 res = _lseeki64(fd, pos, how);
5086#else
Guido van Rossum687dd131993-05-17 08:34:16 +00005087 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00005088#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00005089 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005090 if (res < 0)
5091 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00005092
5093#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005094 return PyLong_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005095#else
5096 return PyLong_FromLongLong(res);
5097#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00005098}
5099
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005100
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005101PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005102"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005103Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005104
Barry Warsaw53699e91996-12-10 23:23:01 +00005105static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005106posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005107{
Guido van Rossum572dbf82007-04-27 23:53:51 +00005108 int fd, size;
5109 Py_ssize_t n;
Barry Warsaw53699e91996-12-10 23:23:01 +00005110 PyObject *buffer;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005111 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
Guido van Rossum687dd131993-05-17 08:34:16 +00005112 return NULL;
Michael W. Hudson9867ced2005-01-31 17:01:59 +00005113 if (size < 0) {
5114 errno = EINVAL;
5115 return posix_error();
5116 }
Christian Heimes72b710a2008-05-26 13:28:38 +00005117 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
Guido van Rossum687dd131993-05-17 08:34:16 +00005118 if (buffer == NULL)
5119 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005120 if (!_PyVerify_fd(fd))
5121 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005122 Py_BEGIN_ALLOW_THREADS
Christian Heimes72b710a2008-05-26 13:28:38 +00005123 n = read(fd, PyBytes_AS_STRING(buffer), size);
Barry Warsaw53699e91996-12-10 23:23:01 +00005124 Py_END_ALLOW_THREADS
Guido van Rossum8bac5461996-06-11 18:38:48 +00005125 if (n < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00005126 Py_DECREF(buffer);
Guido van Rossum687dd131993-05-17 08:34:16 +00005127 return posix_error();
5128 }
Guido van Rossum8bac5461996-06-11 18:38:48 +00005129 if (n != size)
Christian Heimes72b710a2008-05-26 13:28:38 +00005130 _PyBytes_Resize(&buffer, n);
Guido van Rossum687dd131993-05-17 08:34:16 +00005131 return buffer;
5132}
5133
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005134
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005135PyDoc_STRVAR(posix_write__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005136"write(fd, string) -> byteswritten\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005137Write a string to a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005138
Barry Warsaw53699e91996-12-10 23:23:01 +00005139static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005140posix_write(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005141{
Martin v. Löwis423be952008-08-13 15:53:07 +00005142 Py_buffer pbuf;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00005143 int fd;
5144 Py_ssize_t size;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00005145
Antoine Pitrou9cadb1b2008-09-15 23:02:56 +00005146 if (!PyArg_ParseTuple(args, "iy*:write", &fd, &pbuf))
Guido van Rossum687dd131993-05-17 08:34:16 +00005147 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005148 if (!_PyVerify_fd(fd))
5149 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005150 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis423be952008-08-13 15:53:07 +00005151 size = write(fd, pbuf.buf, (size_t)pbuf.len);
Barry Warsaw53699e91996-12-10 23:23:01 +00005152 Py_END_ALLOW_THREADS
Martin v. Löwis423be952008-08-13 15:53:07 +00005153 PyBuffer_Release(&pbuf);
Guido van Rossum687dd131993-05-17 08:34:16 +00005154 if (size < 0)
5155 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00005156 return PyLong_FromSsize_t(size);
Guido van Rossum687dd131993-05-17 08:34:16 +00005157}
5158
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005159
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005160PyDoc_STRVAR(posix_fstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005161"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005162Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005163
Barry Warsaw53699e91996-12-10 23:23:01 +00005164static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005165posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005166{
5167 int fd;
Fred Drake699f3522000-06-29 21:12:41 +00005168 STRUCT_STAT st;
Guido van Rossum687dd131993-05-17 08:34:16 +00005169 int res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005170 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00005171 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00005172#ifdef __VMS
5173 /* on OpenVMS we must ensure that all bytes are written to the file */
5174 fsync(fd);
5175#endif
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005176 if (!_PyVerify_fd(fd))
5177 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005178 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00005179 res = FSTAT(fd, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00005180 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00005181 if (res != 0) {
5182#ifdef MS_WINDOWS
5183 return win32_error("fstat", NULL);
5184#else
Guido van Rossum687dd131993-05-17 08:34:16 +00005185 return posix_error();
Martin v. Löwis14694662006-02-03 12:54:16 +00005186#endif
5187 }
Tim Peters5aa91602002-01-30 05:46:57 +00005188
Martin v. Löwis14694662006-02-03 12:54:16 +00005189 return _pystat_fromstructstat(&st);
Guido van Rossum687dd131993-05-17 08:34:16 +00005190}
5191
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005192PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005193"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00005194Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005195connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00005196
5197static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00005198posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00005199{
5200 int fd;
5201 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
5202 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005203 if (!_PyVerify_fd(fd))
5204 return PyBool_FromLong(0);
Fred Drake106c1a02002-04-23 15:58:02 +00005205 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00005206}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005207
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005208#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005209PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005210"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005211Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005212
Barry Warsaw53699e91996-12-10 23:23:01 +00005213static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005214posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00005215{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005216#if defined(PYOS_OS2)
5217 HFILE read, write;
5218 APIRET rc;
5219
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005220 Py_BEGIN_ALLOW_THREADS
5221 rc = DosCreatePipe( &read, &write, 4096);
5222 Py_END_ALLOW_THREADS
5223 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005224 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005225
5226 return Py_BuildValue("(ii)", read, write);
5227#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005228#if !defined(MS_WINDOWS)
Guido van Rossum687dd131993-05-17 08:34:16 +00005229 int fds[2];
5230 int res;
Barry Warsaw53699e91996-12-10 23:23:01 +00005231 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005232 res = pipe(fds);
Barry Warsaw53699e91996-12-10 23:23:01 +00005233 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005234 if (res != 0)
5235 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005236 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005237#else /* MS_WINDOWS */
Guido van Rossum794d8131994-08-23 13:48:48 +00005238 HANDLE read, write;
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00005239 int read_fd, write_fd;
Guido van Rossum794d8131994-08-23 13:48:48 +00005240 BOOL ok;
Barry Warsaw53699e91996-12-10 23:23:01 +00005241 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00005242 ok = CreatePipe(&read, &write, NULL, 0);
Barry Warsaw53699e91996-12-10 23:23:01 +00005243 Py_END_ALLOW_THREADS
Guido van Rossum794d8131994-08-23 13:48:48 +00005244 if (!ok)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005245 return win32_error("CreatePipe", NULL);
Tim Peters79248aa2001-08-29 21:37:10 +00005246 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
5247 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00005248 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005249#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005250#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00005251}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005252#endif /* HAVE_PIPE */
5253
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005254
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005255#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005256PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005257"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005258Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005259
Barry Warsaw53699e91996-12-10 23:23:01 +00005260static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005261posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005262{
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005263 char *filename;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005264 int mode = 0666;
5265 int res;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005266 if (!PyArg_ParseTuple(args, "s|i:mkfifo", &filename, &mode))
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005267 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005268 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005269 res = mkfifo(filename, mode);
5270 Py_END_ALLOW_THREADS
5271 if (res < 0)
5272 return posix_error();
5273 Py_INCREF(Py_None);
5274 return Py_None;
5275}
5276#endif
5277
5278
Neal Norwitz11690112002-07-30 01:08:28 +00005279#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005280PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005281"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005282Create a filesystem node (file, device special file or named pipe)\n\
5283named filename. mode specifies both the permissions to use and the\n\
5284type of node to be created, being combined (bitwise OR) with one of\n\
5285S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005286device defines the newly created device special file (probably using\n\
5287os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005288
5289
5290static PyObject *
5291posix_mknod(PyObject *self, PyObject *args)
5292{
5293 char *filename;
5294 int mode = 0600;
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005295 int device = 0;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005296 int res;
Martin v. Löwisd631ebe2002-11-02 17:42:33 +00005297 if (!PyArg_ParseTuple(args, "s|ii:mknod", &filename, &mode, &device))
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005298 return NULL;
5299 Py_BEGIN_ALLOW_THREADS
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005300 res = mknod(filename, mode, device);
Barry Warsaw53699e91996-12-10 23:23:01 +00005301 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005302 if (res < 0)
5303 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005304 Py_INCREF(Py_None);
5305 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005306}
5307#endif
5308
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005309#ifdef HAVE_DEVICE_MACROS
5310PyDoc_STRVAR(posix_major__doc__,
5311"major(device) -> major number\n\
5312Extracts a device major number from a raw device number.");
5313
5314static PyObject *
5315posix_major(PyObject *self, PyObject *args)
5316{
5317 int device;
5318 if (!PyArg_ParseTuple(args, "i:major", &device))
5319 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005320 return PyLong_FromLong((long)major(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005321}
5322
5323PyDoc_STRVAR(posix_minor__doc__,
5324"minor(device) -> minor number\n\
5325Extracts a device minor number from a raw device number.");
5326
5327static PyObject *
5328posix_minor(PyObject *self, PyObject *args)
5329{
5330 int device;
5331 if (!PyArg_ParseTuple(args, "i:minor", &device))
5332 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005333 return PyLong_FromLong((long)minor(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005334}
5335
5336PyDoc_STRVAR(posix_makedev__doc__,
5337"makedev(major, minor) -> device number\n\
5338Composes a raw device number from the major and minor device numbers.");
5339
5340static PyObject *
5341posix_makedev(PyObject *self, PyObject *args)
5342{
5343 int major, minor;
5344 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
5345 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005346 return PyLong_FromLong((long)makedev(major, minor));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005347}
5348#endif /* device macros */
5349
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005350
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005351#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005352PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005353"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005354Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005355
Barry Warsaw53699e91996-12-10 23:23:01 +00005356static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005357posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005358{
5359 int fd;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005360 off_t length;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005361 int res;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005362 PyObject *lenobj;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005363
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005364 if (!PyArg_ParseTuple(args, "iO:ftruncate", &fd, &lenobj))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005365 return NULL;
5366
5367#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005368 length = PyLong_AsLong(lenobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005369#else
5370 length = PyLong_Check(lenobj) ?
Christian Heimes217cfd12007-12-02 14:31:20 +00005371 PyLong_AsLongLong(lenobj) : PyLong_AsLong(lenobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005372#endif
5373 if (PyErr_Occurred())
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005374 return NULL;
5375
Barry Warsaw53699e91996-12-10 23:23:01 +00005376 Py_BEGIN_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005377 res = ftruncate(fd, length);
Barry Warsaw53699e91996-12-10 23:23:01 +00005378 Py_END_ALLOW_THREADS
Benjamin Peterson9053d752009-01-19 17:53:36 +00005379 if (res < 0)
5380 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005381 Py_INCREF(Py_None);
5382 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005383}
5384#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00005385
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005386#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005387PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005388"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005389Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005390
Fred Drake762e2061999-08-26 17:23:54 +00005391/* Save putenv() parameters as values here, so we can collect them when they
5392 * get re-set with another call for the same key. */
5393static PyObject *posix_putenv_garbage;
5394
Tim Peters5aa91602002-01-30 05:46:57 +00005395static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005396posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005397{
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005398#ifdef MS_WINDOWS
5399 wchar_t *s1, *s2;
5400 wchar_t *newenv;
5401#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00005402 PyObject *os1, *os2;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005403 char *s1, *s2;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005404 char *newenv;
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005405#endif
Fred Drake762e2061999-08-26 17:23:54 +00005406 PyObject *newstr;
Tim Petersc8996f52001-12-03 20:41:00 +00005407 size_t len;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005408
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005409#ifdef MS_WINDOWS
Martin v. Löwis011e8422009-05-05 04:43:17 +00005410 if (!PyArg_ParseTuple(args,
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005411 "uu:putenv",
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005412 &s1, &s2))
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005413 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00005414#else
5415 if (!PyArg_ParseTuple(args,
5416 "O&O&:putenv",
5417 PyUnicode_FSConverter, &os1,
5418 PyUnicode_FSConverter, &os2))
5419 return NULL;
5420 s1 = bytes2str(os1, 1);
5421 s2 = bytes2str(os2, 1);
5422#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00005423
5424#if defined(PYOS_OS2)
5425 if (stricmp(s1, "BEGINLIBPATH") == 0) {
5426 APIRET rc;
5427
Guido van Rossumd48f2521997-12-05 22:19:34 +00005428 rc = DosSetExtLIBPATH(s2, BEGIN_LIBPATH);
5429 if (rc != NO_ERROR)
5430 return os2_error(rc);
5431
5432 } else if (stricmp(s1, "ENDLIBPATH") == 0) {
5433 APIRET rc;
5434
Guido van Rossumd48f2521997-12-05 22:19:34 +00005435 rc = DosSetExtLIBPATH(s2, END_LIBPATH);
5436 if (rc != NO_ERROR)
5437 return os2_error(rc);
5438 } else {
5439#endif
Fred Drake762e2061999-08-26 17:23:54 +00005440 /* XXX This can leak memory -- not easy to fix :-( */
Tim Petersc8996f52001-12-03 20:41:00 +00005441 /* len includes space for a trailing \0; the size arg to
Christian Heimes72b710a2008-05-26 13:28:38 +00005442 PyBytes_FromStringAndSize does not count that */
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005443#ifdef MS_WINDOWS
5444 len = wcslen(s1) + wcslen(s2) + 2;
5445 newstr = PyUnicode_FromUnicode(NULL, (int)len - 1);
5446#else
5447 len = strlen(s1) + strlen(s2) + 2;
Christian Heimes72b710a2008-05-26 13:28:38 +00005448 newstr = PyBytes_FromStringAndSize(NULL, (int)len - 1);
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005449#endif
Fred Drake762e2061999-08-26 17:23:54 +00005450 if (newstr == NULL)
5451 return PyErr_NoMemory();
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005452#ifdef MS_WINDOWS
5453 newenv = PyUnicode_AsUnicode(newstr);
5454 _snwprintf(newenv, len, L"%s=%s", s1, s2);
5455 if (_wputenv(newenv)) {
5456 Py_DECREF(newstr);
5457 posix_error();
5458 return NULL;
5459 }
5460#else
Christian Heimes72b710a2008-05-26 13:28:38 +00005461 newenv = PyBytes_AS_STRING(newstr);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005462 PyOS_snprintf(newenv, len, "%s=%s", s1, s2);
5463 if (putenv(newenv)) {
Neal Norwitz4adc9ab2003-02-10 03:10:43 +00005464 Py_DECREF(newstr);
Martin v. Löwis011e8422009-05-05 04:43:17 +00005465 release_bytes(os1);
5466 release_bytes(os2);
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005467 posix_error();
5468 return NULL;
5469 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005470#endif
Fred Drake762e2061999-08-26 17:23:54 +00005471 /* Install the first arg and newstr in posix_putenv_garbage;
5472 * this will cause previous value to be collected. This has to
5473 * happen after the real putenv() call because the old value
5474 * was still accessible until then. */
5475 if (PyDict_SetItem(posix_putenv_garbage,
5476 PyTuple_GET_ITEM(args, 0), newstr)) {
5477 /* really not much we can do; just leak */
5478 PyErr_Clear();
5479 }
5480 else {
5481 Py_DECREF(newstr);
5482 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00005483
5484#if defined(PYOS_OS2)
5485 }
5486#endif
Martin v. Löwis011e8422009-05-05 04:43:17 +00005487#ifndef MS_WINDOWS
5488 release_bytes(os1);
5489 release_bytes(os2);
5490#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00005491 Py_INCREF(Py_None);
5492 return Py_None;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005493}
Guido van Rossumb6a47161997-09-15 22:54:34 +00005494#endif /* putenv */
5495
Guido van Rossumc524d952001-10-19 01:31:59 +00005496#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005497PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005498"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005499Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00005500
5501static PyObject *
5502posix_unsetenv(PyObject *self, PyObject *args)
5503{
5504 char *s1;
5505
5506 if (!PyArg_ParseTuple(args, "s:unsetenv", &s1))
5507 return NULL;
5508
5509 unsetenv(s1);
5510
5511 /* Remove the key from posix_putenv_garbage;
5512 * this will cause it to be collected. This has to
Tim Peters5aa91602002-01-30 05:46:57 +00005513 * happen after the real unsetenv() call because the
Guido van Rossumc524d952001-10-19 01:31:59 +00005514 * old value was still accessible until then.
5515 */
5516 if (PyDict_DelItem(posix_putenv_garbage,
5517 PyTuple_GET_ITEM(args, 0))) {
5518 /* really not much we can do; just leak */
5519 PyErr_Clear();
5520 }
5521
5522 Py_INCREF(Py_None);
5523 return Py_None;
5524}
5525#endif /* unsetenv */
5526
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005527PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005528"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005529Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005530
Guido van Rossumf68d8e52001-04-14 17:55:09 +00005531static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005532posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00005533{
5534 int code;
5535 char *message;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005536 if (!PyArg_ParseTuple(args, "i:strerror", &code))
Guido van Rossumb6a47161997-09-15 22:54:34 +00005537 return NULL;
5538 message = strerror(code);
5539 if (message == NULL) {
5540 PyErr_SetString(PyExc_ValueError,
Fred Drake661ea262000-10-24 19:57:45 +00005541 "strerror() argument out of range");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005542 return NULL;
5543 }
Neal Norwitz93c56822007-08-26 07:10:06 +00005544 return PyUnicode_FromString(message);
Guido van Rossumb6a47161997-09-15 22:54:34 +00005545}
Guido van Rossumb6a47161997-09-15 22:54:34 +00005546
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005547
Guido van Rossumc9641791998-08-04 15:26:23 +00005548#ifdef HAVE_SYS_WAIT_H
5549
Fred Drake106c1a02002-04-23 15:58:02 +00005550#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005551PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005552"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005553Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00005554
5555static PyObject *
5556posix_WCOREDUMP(PyObject *self, PyObject *args)
5557{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005558 WAIT_TYPE status;
5559 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00005560
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005561 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00005562 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00005563
5564 return PyBool_FromLong(WCOREDUMP(status));
Fred Drake106c1a02002-04-23 15:58:02 +00005565}
5566#endif /* WCOREDUMP */
5567
5568#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005569PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005570"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00005571Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005572job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00005573
5574static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005575posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00005576{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005577 WAIT_TYPE status;
5578 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00005579
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005580 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00005581 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00005582
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005583 return PyBool_FromLong(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00005584}
5585#endif /* WIFCONTINUED */
5586
Guido van Rossumc9641791998-08-04 15:26:23 +00005587#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005588PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005589"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005590Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005591
5592static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005593posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005594{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005595 WAIT_TYPE status;
5596 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005597
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005598 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005599 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005600
Fred Drake106c1a02002-04-23 15:58:02 +00005601 return PyBool_FromLong(WIFSTOPPED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005602}
5603#endif /* WIFSTOPPED */
5604
5605#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005606PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005607"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005608Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005609
5610static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005611posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005612{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005613 WAIT_TYPE status;
5614 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005615
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005616 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005617 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005618
Fred Drake106c1a02002-04-23 15:58:02 +00005619 return PyBool_FromLong(WIFSIGNALED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005620}
5621#endif /* WIFSIGNALED */
5622
5623#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005624PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005625"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005626Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005627system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005628
5629static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005630posix_WIFEXITED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005631{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005632 WAIT_TYPE status;
5633 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005634
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005635 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005636 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005637
Fred Drake106c1a02002-04-23 15:58:02 +00005638 return PyBool_FromLong(WIFEXITED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005639}
5640#endif /* WIFEXITED */
5641
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005642#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005643PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005644"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005645Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005646
5647static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005648posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005649{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005650 WAIT_TYPE status;
5651 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005652
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005653 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005654 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005655
Guido van Rossumc9641791998-08-04 15:26:23 +00005656 return Py_BuildValue("i", WEXITSTATUS(status));
5657}
5658#endif /* WEXITSTATUS */
5659
5660#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005661PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005662"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005663Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005664value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005665
5666static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005667posix_WTERMSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005668{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005669 WAIT_TYPE status;
5670 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005671
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005672 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005673 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005674
Guido van Rossumc9641791998-08-04 15:26:23 +00005675 return Py_BuildValue("i", WTERMSIG(status));
5676}
5677#endif /* WTERMSIG */
5678
5679#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005680PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005681"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005682Return the signal that stopped the process that provided\n\
5683the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005684
5685static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005686posix_WSTOPSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005687{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005688 WAIT_TYPE status;
5689 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005690
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005691 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005692 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005693
Guido van Rossumc9641791998-08-04 15:26:23 +00005694 return Py_BuildValue("i", WSTOPSIG(status));
5695}
5696#endif /* WSTOPSIG */
5697
5698#endif /* HAVE_SYS_WAIT_H */
5699
5700
Thomas Wouters477c8d52006-05-27 19:21:47 +00005701#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossumd5753e11999-10-19 13:29:23 +00005702#ifdef _SCO_DS
5703/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
5704 needed definitions in sys/statvfs.h */
5705#define _SVID3
5706#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00005707#include <sys/statvfs.h>
5708
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005709static PyObject*
5710_pystatvfs_fromstructstatvfs(struct statvfs st) {
5711 PyObject *v = PyStructSequence_New(&StatVFSResultType);
5712 if (v == NULL)
5713 return NULL;
5714
5715#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005716 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
5717 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
5718 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long) st.f_blocks));
5719 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long) st.f_bfree));
5720 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long) st.f_bavail));
5721 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long) st.f_files));
5722 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong((long) st.f_ffree));
5723 PyStructSequence_SET_ITEM(v, 7, PyLong_FromLong((long) st.f_favail));
5724 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
5725 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005726#else
Christian Heimes217cfd12007-12-02 14:31:20 +00005727 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
5728 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
Tim Peters5aa91602002-01-30 05:46:57 +00005729 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005730 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
Tim Peters5aa91602002-01-30 05:46:57 +00005731 PyStructSequence_SET_ITEM(v, 3,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005732 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005733 PyStructSequence_SET_ITEM(v, 4,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005734 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
Tim Peters5aa91602002-01-30 05:46:57 +00005735 PyStructSequence_SET_ITEM(v, 5,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005736 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
Tim Peters5aa91602002-01-30 05:46:57 +00005737 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005738 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
Tim Peters5aa91602002-01-30 05:46:57 +00005739 PyStructSequence_SET_ITEM(v, 7,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005740 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
Christian Heimes217cfd12007-12-02 14:31:20 +00005741 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
5742 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005743#endif
5744
5745 return v;
5746}
5747
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005748PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005749"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005750Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005751
5752static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005753posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005754{
5755 int fd, res;
5756 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005757
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005758 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005759 return NULL;
5760 Py_BEGIN_ALLOW_THREADS
5761 res = fstatvfs(fd, &st);
5762 Py_END_ALLOW_THREADS
5763 if (res != 0)
5764 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005765
5766 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005767}
Thomas Wouters477c8d52006-05-27 19:21:47 +00005768#endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00005769
5770
Thomas Wouters477c8d52006-05-27 19:21:47 +00005771#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005772#include <sys/statvfs.h>
5773
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005774PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005775"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005776Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005777
5778static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005779posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005780{
5781 char *path;
5782 int res;
5783 struct statvfs st;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005784 if (!PyArg_ParseTuple(args, "s:statvfs", &path))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005785 return NULL;
5786 Py_BEGIN_ALLOW_THREADS
5787 res = statvfs(path, &st);
5788 Py_END_ALLOW_THREADS
5789 if (res != 0)
5790 return posix_error_with_filename(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005791
5792 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005793}
5794#endif /* HAVE_STATVFS */
5795
Fred Drakec9680921999-12-13 16:37:25 +00005796/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
5797 * It maps strings representing configuration variable names to
5798 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00005799 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00005800 * rarely-used constants. There are three separate tables that use
5801 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00005802 *
5803 * This code is always included, even if none of the interfaces that
5804 * need it are included. The #if hackery needed to avoid it would be
5805 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00005806 */
5807struct constdef {
5808 char *name;
5809 long value;
5810};
5811
Fred Drake12c6e2d1999-12-14 21:25:03 +00005812static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005813conv_confname(PyObject *arg, int *valuep, struct constdef *table,
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005814 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00005815{
Christian Heimes217cfd12007-12-02 14:31:20 +00005816 if (PyLong_Check(arg)) {
5817 *valuep = PyLong_AS_LONG(arg);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005818 return 1;
5819 }
Guido van Rossumbce56a62007-05-10 18:04:33 +00005820 else {
Fred Drake12c6e2d1999-12-14 21:25:03 +00005821 /* look up the value in the table using a binary search */
Fred Drake699f3522000-06-29 21:12:41 +00005822 size_t lo = 0;
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005823 size_t mid;
Fred Drake699f3522000-06-29 21:12:41 +00005824 size_t hi = tablesize;
5825 int cmp;
Guido van Rossumbce56a62007-05-10 18:04:33 +00005826 const char *confname;
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005827 if (!PyUnicode_Check(arg)) {
Guido van Rossumbce56a62007-05-10 18:04:33 +00005828 PyErr_SetString(PyExc_TypeError,
5829 "configuration names must be strings or integers");
5830 return 0;
5831 }
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00005832 confname = _PyUnicode_AsString(arg);
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005833 if (confname == NULL)
5834 return 0;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005835 while (lo < hi) {
5836 mid = (lo + hi) / 2;
5837 cmp = strcmp(confname, table[mid].name);
5838 if (cmp < 0)
5839 hi = mid;
5840 else if (cmp > 0)
5841 lo = mid + 1;
5842 else {
5843 *valuep = table[mid].value;
5844 return 1;
5845 }
5846 }
5847 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
Guido van Rossumbce56a62007-05-10 18:04:33 +00005848 return 0;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005849 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00005850}
5851
5852
5853#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
5854static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00005855#ifdef _PC_ABI_AIO_XFER_MAX
5856 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
5857#endif
5858#ifdef _PC_ABI_ASYNC_IO
5859 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
5860#endif
Fred Drakec9680921999-12-13 16:37:25 +00005861#ifdef _PC_ASYNC_IO
5862 {"PC_ASYNC_IO", _PC_ASYNC_IO},
5863#endif
5864#ifdef _PC_CHOWN_RESTRICTED
5865 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
5866#endif
5867#ifdef _PC_FILESIZEBITS
5868 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
5869#endif
5870#ifdef _PC_LAST
5871 {"PC_LAST", _PC_LAST},
5872#endif
5873#ifdef _PC_LINK_MAX
5874 {"PC_LINK_MAX", _PC_LINK_MAX},
5875#endif
5876#ifdef _PC_MAX_CANON
5877 {"PC_MAX_CANON", _PC_MAX_CANON},
5878#endif
5879#ifdef _PC_MAX_INPUT
5880 {"PC_MAX_INPUT", _PC_MAX_INPUT},
5881#endif
5882#ifdef _PC_NAME_MAX
5883 {"PC_NAME_MAX", _PC_NAME_MAX},
5884#endif
5885#ifdef _PC_NO_TRUNC
5886 {"PC_NO_TRUNC", _PC_NO_TRUNC},
5887#endif
5888#ifdef _PC_PATH_MAX
5889 {"PC_PATH_MAX", _PC_PATH_MAX},
5890#endif
5891#ifdef _PC_PIPE_BUF
5892 {"PC_PIPE_BUF", _PC_PIPE_BUF},
5893#endif
5894#ifdef _PC_PRIO_IO
5895 {"PC_PRIO_IO", _PC_PRIO_IO},
5896#endif
5897#ifdef _PC_SOCK_MAXBUF
5898 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
5899#endif
5900#ifdef _PC_SYNC_IO
5901 {"PC_SYNC_IO", _PC_SYNC_IO},
5902#endif
5903#ifdef _PC_VDISABLE
5904 {"PC_VDISABLE", _PC_VDISABLE},
5905#endif
5906};
5907
Fred Drakec9680921999-12-13 16:37:25 +00005908static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005909conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00005910{
5911 return conv_confname(arg, valuep, posix_constants_pathconf,
5912 sizeof(posix_constants_pathconf)
5913 / sizeof(struct constdef));
5914}
5915#endif
5916
5917#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005918PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005919"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00005920Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005921If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00005922
5923static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005924posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005925{
5926 PyObject *result = NULL;
5927 int name, fd;
5928
Fred Drake12c6e2d1999-12-14 21:25:03 +00005929 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
5930 conv_path_confname, &name)) {
Fred Drakec9680921999-12-13 16:37:25 +00005931 long limit;
5932
5933 errno = 0;
5934 limit = fpathconf(fd, name);
5935 if (limit == -1 && errno != 0)
5936 posix_error();
5937 else
Christian Heimes217cfd12007-12-02 14:31:20 +00005938 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00005939 }
5940 return result;
5941}
5942#endif
5943
5944
5945#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005946PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005947"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00005948Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005949If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00005950
5951static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005952posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005953{
5954 PyObject *result = NULL;
5955 int name;
5956 char *path;
5957
5958 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
5959 conv_path_confname, &name)) {
5960 long limit;
5961
5962 errno = 0;
5963 limit = pathconf(path, name);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005964 if (limit == -1 && errno != 0) {
Fred Drakec9680921999-12-13 16:37:25 +00005965 if (errno == EINVAL)
5966 /* could be a path or name problem */
5967 posix_error();
5968 else
5969 posix_error_with_filename(path);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005970 }
Fred Drakec9680921999-12-13 16:37:25 +00005971 else
Christian Heimes217cfd12007-12-02 14:31:20 +00005972 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00005973 }
5974 return result;
5975}
5976#endif
5977
5978#ifdef HAVE_CONFSTR
5979static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00005980#ifdef _CS_ARCHITECTURE
5981 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
5982#endif
5983#ifdef _CS_HOSTNAME
5984 {"CS_HOSTNAME", _CS_HOSTNAME},
5985#endif
5986#ifdef _CS_HW_PROVIDER
5987 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
5988#endif
5989#ifdef _CS_HW_SERIAL
5990 {"CS_HW_SERIAL", _CS_HW_SERIAL},
5991#endif
5992#ifdef _CS_INITTAB_NAME
5993 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
5994#endif
Fred Drakec9680921999-12-13 16:37:25 +00005995#ifdef _CS_LFS64_CFLAGS
5996 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
5997#endif
5998#ifdef _CS_LFS64_LDFLAGS
5999 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
6000#endif
6001#ifdef _CS_LFS64_LIBS
6002 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
6003#endif
6004#ifdef _CS_LFS64_LINTFLAGS
6005 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
6006#endif
6007#ifdef _CS_LFS_CFLAGS
6008 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
6009#endif
6010#ifdef _CS_LFS_LDFLAGS
6011 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
6012#endif
6013#ifdef _CS_LFS_LIBS
6014 {"CS_LFS_LIBS", _CS_LFS_LIBS},
6015#endif
6016#ifdef _CS_LFS_LINTFLAGS
6017 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
6018#endif
Fred Draked86ed291999-12-15 15:34:33 +00006019#ifdef _CS_MACHINE
6020 {"CS_MACHINE", _CS_MACHINE},
6021#endif
Fred Drakec9680921999-12-13 16:37:25 +00006022#ifdef _CS_PATH
6023 {"CS_PATH", _CS_PATH},
6024#endif
Fred Draked86ed291999-12-15 15:34:33 +00006025#ifdef _CS_RELEASE
6026 {"CS_RELEASE", _CS_RELEASE},
6027#endif
6028#ifdef _CS_SRPC_DOMAIN
6029 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
6030#endif
6031#ifdef _CS_SYSNAME
6032 {"CS_SYSNAME", _CS_SYSNAME},
6033#endif
6034#ifdef _CS_VERSION
6035 {"CS_VERSION", _CS_VERSION},
6036#endif
Fred Drakec9680921999-12-13 16:37:25 +00006037#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
6038 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
6039#endif
6040#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
6041 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
6042#endif
6043#ifdef _CS_XBS5_ILP32_OFF32_LIBS
6044 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
6045#endif
6046#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
6047 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
6048#endif
6049#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
6050 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
6051#endif
6052#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
6053 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
6054#endif
6055#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
6056 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
6057#endif
6058#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
6059 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
6060#endif
6061#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
6062 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
6063#endif
6064#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
6065 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
6066#endif
6067#ifdef _CS_XBS5_LP64_OFF64_LIBS
6068 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
6069#endif
6070#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
6071 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
6072#endif
6073#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
6074 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
6075#endif
6076#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
6077 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
6078#endif
6079#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
6080 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
6081#endif
6082#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
6083 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
6084#endif
Fred Draked86ed291999-12-15 15:34:33 +00006085#ifdef _MIPS_CS_AVAIL_PROCESSORS
6086 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
6087#endif
6088#ifdef _MIPS_CS_BASE
6089 {"MIPS_CS_BASE", _MIPS_CS_BASE},
6090#endif
6091#ifdef _MIPS_CS_HOSTID
6092 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
6093#endif
6094#ifdef _MIPS_CS_HW_NAME
6095 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
6096#endif
6097#ifdef _MIPS_CS_NUM_PROCESSORS
6098 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
6099#endif
6100#ifdef _MIPS_CS_OSREL_MAJ
6101 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
6102#endif
6103#ifdef _MIPS_CS_OSREL_MIN
6104 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
6105#endif
6106#ifdef _MIPS_CS_OSREL_PATCH
6107 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
6108#endif
6109#ifdef _MIPS_CS_OS_NAME
6110 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
6111#endif
6112#ifdef _MIPS_CS_OS_PROVIDER
6113 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
6114#endif
6115#ifdef _MIPS_CS_PROCESSORS
6116 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
6117#endif
6118#ifdef _MIPS_CS_SERIAL
6119 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
6120#endif
6121#ifdef _MIPS_CS_VENDOR
6122 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
6123#endif
Fred Drakec9680921999-12-13 16:37:25 +00006124};
6125
6126static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006127conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00006128{
6129 return conv_confname(arg, valuep, posix_constants_confstr,
6130 sizeof(posix_constants_confstr)
6131 / sizeof(struct constdef));
6132}
6133
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006134PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006135"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006136Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00006137
6138static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006139posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006140{
6141 PyObject *result = NULL;
6142 int name;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006143 char buffer[256];
Fred Drakec9680921999-12-13 16:37:25 +00006144
6145 if (PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name)) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006146 int len;
Fred Drakec9680921999-12-13 16:37:25 +00006147
Fred Drakec9680921999-12-13 16:37:25 +00006148 errno = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006149 len = confstr(name, buffer, sizeof(buffer));
6150 if (len == 0) {
6151 if (errno) {
6152 posix_error();
6153 }
6154 else {
6155 result = Py_None;
6156 Py_INCREF(Py_None);
6157 }
Fred Drakec9680921999-12-13 16:37:25 +00006158 }
6159 else {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006160 if ((unsigned int)len >= sizeof(buffer)) {
Neal Norwitz93c56822007-08-26 07:10:06 +00006161 result = PyUnicode_FromStringAndSize(NULL, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00006162 if (result != NULL)
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00006163 confstr(name, _PyUnicode_AsString(result), len);
Fred Drakec9680921999-12-13 16:37:25 +00006164 }
6165 else
Neal Norwitz93c56822007-08-26 07:10:06 +00006166 result = PyUnicode_FromStringAndSize(buffer, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00006167 }
6168 }
6169 return result;
6170}
6171#endif
6172
6173
6174#ifdef HAVE_SYSCONF
6175static struct constdef posix_constants_sysconf[] = {
6176#ifdef _SC_2_CHAR_TERM
6177 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
6178#endif
6179#ifdef _SC_2_C_BIND
6180 {"SC_2_C_BIND", _SC_2_C_BIND},
6181#endif
6182#ifdef _SC_2_C_DEV
6183 {"SC_2_C_DEV", _SC_2_C_DEV},
6184#endif
6185#ifdef _SC_2_C_VERSION
6186 {"SC_2_C_VERSION", _SC_2_C_VERSION},
6187#endif
6188#ifdef _SC_2_FORT_DEV
6189 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
6190#endif
6191#ifdef _SC_2_FORT_RUN
6192 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
6193#endif
6194#ifdef _SC_2_LOCALEDEF
6195 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
6196#endif
6197#ifdef _SC_2_SW_DEV
6198 {"SC_2_SW_DEV", _SC_2_SW_DEV},
6199#endif
6200#ifdef _SC_2_UPE
6201 {"SC_2_UPE", _SC_2_UPE},
6202#endif
6203#ifdef _SC_2_VERSION
6204 {"SC_2_VERSION", _SC_2_VERSION},
6205#endif
Fred Draked86ed291999-12-15 15:34:33 +00006206#ifdef _SC_ABI_ASYNCHRONOUS_IO
6207 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
6208#endif
6209#ifdef _SC_ACL
6210 {"SC_ACL", _SC_ACL},
6211#endif
Fred Drakec9680921999-12-13 16:37:25 +00006212#ifdef _SC_AIO_LISTIO_MAX
6213 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
6214#endif
Fred Drakec9680921999-12-13 16:37:25 +00006215#ifdef _SC_AIO_MAX
6216 {"SC_AIO_MAX", _SC_AIO_MAX},
6217#endif
6218#ifdef _SC_AIO_PRIO_DELTA_MAX
6219 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
6220#endif
6221#ifdef _SC_ARG_MAX
6222 {"SC_ARG_MAX", _SC_ARG_MAX},
6223#endif
6224#ifdef _SC_ASYNCHRONOUS_IO
6225 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
6226#endif
6227#ifdef _SC_ATEXIT_MAX
6228 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
6229#endif
Fred Draked86ed291999-12-15 15:34:33 +00006230#ifdef _SC_AUDIT
6231 {"SC_AUDIT", _SC_AUDIT},
6232#endif
Fred Drakec9680921999-12-13 16:37:25 +00006233#ifdef _SC_AVPHYS_PAGES
6234 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
6235#endif
6236#ifdef _SC_BC_BASE_MAX
6237 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
6238#endif
6239#ifdef _SC_BC_DIM_MAX
6240 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
6241#endif
6242#ifdef _SC_BC_SCALE_MAX
6243 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
6244#endif
6245#ifdef _SC_BC_STRING_MAX
6246 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
6247#endif
Fred Draked86ed291999-12-15 15:34:33 +00006248#ifdef _SC_CAP
6249 {"SC_CAP", _SC_CAP},
6250#endif
Fred Drakec9680921999-12-13 16:37:25 +00006251#ifdef _SC_CHARCLASS_NAME_MAX
6252 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
6253#endif
6254#ifdef _SC_CHAR_BIT
6255 {"SC_CHAR_BIT", _SC_CHAR_BIT},
6256#endif
6257#ifdef _SC_CHAR_MAX
6258 {"SC_CHAR_MAX", _SC_CHAR_MAX},
6259#endif
6260#ifdef _SC_CHAR_MIN
6261 {"SC_CHAR_MIN", _SC_CHAR_MIN},
6262#endif
6263#ifdef _SC_CHILD_MAX
6264 {"SC_CHILD_MAX", _SC_CHILD_MAX},
6265#endif
6266#ifdef _SC_CLK_TCK
6267 {"SC_CLK_TCK", _SC_CLK_TCK},
6268#endif
6269#ifdef _SC_COHER_BLKSZ
6270 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
6271#endif
6272#ifdef _SC_COLL_WEIGHTS_MAX
6273 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
6274#endif
6275#ifdef _SC_DCACHE_ASSOC
6276 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
6277#endif
6278#ifdef _SC_DCACHE_BLKSZ
6279 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
6280#endif
6281#ifdef _SC_DCACHE_LINESZ
6282 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
6283#endif
6284#ifdef _SC_DCACHE_SZ
6285 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
6286#endif
6287#ifdef _SC_DCACHE_TBLKSZ
6288 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
6289#endif
6290#ifdef _SC_DELAYTIMER_MAX
6291 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
6292#endif
6293#ifdef _SC_EQUIV_CLASS_MAX
6294 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
6295#endif
6296#ifdef _SC_EXPR_NEST_MAX
6297 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
6298#endif
6299#ifdef _SC_FSYNC
6300 {"SC_FSYNC", _SC_FSYNC},
6301#endif
6302#ifdef _SC_GETGR_R_SIZE_MAX
6303 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
6304#endif
6305#ifdef _SC_GETPW_R_SIZE_MAX
6306 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
6307#endif
6308#ifdef _SC_ICACHE_ASSOC
6309 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
6310#endif
6311#ifdef _SC_ICACHE_BLKSZ
6312 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
6313#endif
6314#ifdef _SC_ICACHE_LINESZ
6315 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
6316#endif
6317#ifdef _SC_ICACHE_SZ
6318 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
6319#endif
Fred Draked86ed291999-12-15 15:34:33 +00006320#ifdef _SC_INF
6321 {"SC_INF", _SC_INF},
6322#endif
Fred Drakec9680921999-12-13 16:37:25 +00006323#ifdef _SC_INT_MAX
6324 {"SC_INT_MAX", _SC_INT_MAX},
6325#endif
6326#ifdef _SC_INT_MIN
6327 {"SC_INT_MIN", _SC_INT_MIN},
6328#endif
6329#ifdef _SC_IOV_MAX
6330 {"SC_IOV_MAX", _SC_IOV_MAX},
6331#endif
Fred Draked86ed291999-12-15 15:34:33 +00006332#ifdef _SC_IP_SECOPTS
6333 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
6334#endif
Fred Drakec9680921999-12-13 16:37:25 +00006335#ifdef _SC_JOB_CONTROL
6336 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
6337#endif
Fred Draked86ed291999-12-15 15:34:33 +00006338#ifdef _SC_KERN_POINTERS
6339 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
6340#endif
6341#ifdef _SC_KERN_SIM
6342 {"SC_KERN_SIM", _SC_KERN_SIM},
6343#endif
Fred Drakec9680921999-12-13 16:37:25 +00006344#ifdef _SC_LINE_MAX
6345 {"SC_LINE_MAX", _SC_LINE_MAX},
6346#endif
6347#ifdef _SC_LOGIN_NAME_MAX
6348 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
6349#endif
6350#ifdef _SC_LOGNAME_MAX
6351 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
6352#endif
6353#ifdef _SC_LONG_BIT
6354 {"SC_LONG_BIT", _SC_LONG_BIT},
6355#endif
Fred Draked86ed291999-12-15 15:34:33 +00006356#ifdef _SC_MAC
6357 {"SC_MAC", _SC_MAC},
6358#endif
Fred Drakec9680921999-12-13 16:37:25 +00006359#ifdef _SC_MAPPED_FILES
6360 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
6361#endif
6362#ifdef _SC_MAXPID
6363 {"SC_MAXPID", _SC_MAXPID},
6364#endif
6365#ifdef _SC_MB_LEN_MAX
6366 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
6367#endif
6368#ifdef _SC_MEMLOCK
6369 {"SC_MEMLOCK", _SC_MEMLOCK},
6370#endif
6371#ifdef _SC_MEMLOCK_RANGE
6372 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
6373#endif
6374#ifdef _SC_MEMORY_PROTECTION
6375 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
6376#endif
6377#ifdef _SC_MESSAGE_PASSING
6378 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
6379#endif
Fred Draked86ed291999-12-15 15:34:33 +00006380#ifdef _SC_MMAP_FIXED_ALIGNMENT
6381 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
6382#endif
Fred Drakec9680921999-12-13 16:37:25 +00006383#ifdef _SC_MQ_OPEN_MAX
6384 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
6385#endif
6386#ifdef _SC_MQ_PRIO_MAX
6387 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
6388#endif
Fred Draked86ed291999-12-15 15:34:33 +00006389#ifdef _SC_NACLS_MAX
6390 {"SC_NACLS_MAX", _SC_NACLS_MAX},
6391#endif
Fred Drakec9680921999-12-13 16:37:25 +00006392#ifdef _SC_NGROUPS_MAX
6393 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
6394#endif
6395#ifdef _SC_NL_ARGMAX
6396 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
6397#endif
6398#ifdef _SC_NL_LANGMAX
6399 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
6400#endif
6401#ifdef _SC_NL_MSGMAX
6402 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
6403#endif
6404#ifdef _SC_NL_NMAX
6405 {"SC_NL_NMAX", _SC_NL_NMAX},
6406#endif
6407#ifdef _SC_NL_SETMAX
6408 {"SC_NL_SETMAX", _SC_NL_SETMAX},
6409#endif
6410#ifdef _SC_NL_TEXTMAX
6411 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
6412#endif
6413#ifdef _SC_NPROCESSORS_CONF
6414 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
6415#endif
6416#ifdef _SC_NPROCESSORS_ONLN
6417 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
6418#endif
Fred Draked86ed291999-12-15 15:34:33 +00006419#ifdef _SC_NPROC_CONF
6420 {"SC_NPROC_CONF", _SC_NPROC_CONF},
6421#endif
6422#ifdef _SC_NPROC_ONLN
6423 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
6424#endif
Fred Drakec9680921999-12-13 16:37:25 +00006425#ifdef _SC_NZERO
6426 {"SC_NZERO", _SC_NZERO},
6427#endif
6428#ifdef _SC_OPEN_MAX
6429 {"SC_OPEN_MAX", _SC_OPEN_MAX},
6430#endif
6431#ifdef _SC_PAGESIZE
6432 {"SC_PAGESIZE", _SC_PAGESIZE},
6433#endif
6434#ifdef _SC_PAGE_SIZE
6435 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
6436#endif
6437#ifdef _SC_PASS_MAX
6438 {"SC_PASS_MAX", _SC_PASS_MAX},
6439#endif
6440#ifdef _SC_PHYS_PAGES
6441 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
6442#endif
6443#ifdef _SC_PII
6444 {"SC_PII", _SC_PII},
6445#endif
6446#ifdef _SC_PII_INTERNET
6447 {"SC_PII_INTERNET", _SC_PII_INTERNET},
6448#endif
6449#ifdef _SC_PII_INTERNET_DGRAM
6450 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
6451#endif
6452#ifdef _SC_PII_INTERNET_STREAM
6453 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
6454#endif
6455#ifdef _SC_PII_OSI
6456 {"SC_PII_OSI", _SC_PII_OSI},
6457#endif
6458#ifdef _SC_PII_OSI_CLTS
6459 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
6460#endif
6461#ifdef _SC_PII_OSI_COTS
6462 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
6463#endif
6464#ifdef _SC_PII_OSI_M
6465 {"SC_PII_OSI_M", _SC_PII_OSI_M},
6466#endif
6467#ifdef _SC_PII_SOCKET
6468 {"SC_PII_SOCKET", _SC_PII_SOCKET},
6469#endif
6470#ifdef _SC_PII_XTI
6471 {"SC_PII_XTI", _SC_PII_XTI},
6472#endif
6473#ifdef _SC_POLL
6474 {"SC_POLL", _SC_POLL},
6475#endif
6476#ifdef _SC_PRIORITIZED_IO
6477 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
6478#endif
6479#ifdef _SC_PRIORITY_SCHEDULING
6480 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
6481#endif
6482#ifdef _SC_REALTIME_SIGNALS
6483 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
6484#endif
6485#ifdef _SC_RE_DUP_MAX
6486 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
6487#endif
6488#ifdef _SC_RTSIG_MAX
6489 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
6490#endif
6491#ifdef _SC_SAVED_IDS
6492 {"SC_SAVED_IDS", _SC_SAVED_IDS},
6493#endif
6494#ifdef _SC_SCHAR_MAX
6495 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
6496#endif
6497#ifdef _SC_SCHAR_MIN
6498 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
6499#endif
6500#ifdef _SC_SELECT
6501 {"SC_SELECT", _SC_SELECT},
6502#endif
6503#ifdef _SC_SEMAPHORES
6504 {"SC_SEMAPHORES", _SC_SEMAPHORES},
6505#endif
6506#ifdef _SC_SEM_NSEMS_MAX
6507 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
6508#endif
6509#ifdef _SC_SEM_VALUE_MAX
6510 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
6511#endif
6512#ifdef _SC_SHARED_MEMORY_OBJECTS
6513 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
6514#endif
6515#ifdef _SC_SHRT_MAX
6516 {"SC_SHRT_MAX", _SC_SHRT_MAX},
6517#endif
6518#ifdef _SC_SHRT_MIN
6519 {"SC_SHRT_MIN", _SC_SHRT_MIN},
6520#endif
6521#ifdef _SC_SIGQUEUE_MAX
6522 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
6523#endif
6524#ifdef _SC_SIGRT_MAX
6525 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
6526#endif
6527#ifdef _SC_SIGRT_MIN
6528 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
6529#endif
Fred Draked86ed291999-12-15 15:34:33 +00006530#ifdef _SC_SOFTPOWER
6531 {"SC_SOFTPOWER", _SC_SOFTPOWER},
6532#endif
Fred Drakec9680921999-12-13 16:37:25 +00006533#ifdef _SC_SPLIT_CACHE
6534 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
6535#endif
6536#ifdef _SC_SSIZE_MAX
6537 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
6538#endif
6539#ifdef _SC_STACK_PROT
6540 {"SC_STACK_PROT", _SC_STACK_PROT},
6541#endif
6542#ifdef _SC_STREAM_MAX
6543 {"SC_STREAM_MAX", _SC_STREAM_MAX},
6544#endif
6545#ifdef _SC_SYNCHRONIZED_IO
6546 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
6547#endif
6548#ifdef _SC_THREADS
6549 {"SC_THREADS", _SC_THREADS},
6550#endif
6551#ifdef _SC_THREAD_ATTR_STACKADDR
6552 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
6553#endif
6554#ifdef _SC_THREAD_ATTR_STACKSIZE
6555 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
6556#endif
6557#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
6558 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
6559#endif
6560#ifdef _SC_THREAD_KEYS_MAX
6561 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
6562#endif
6563#ifdef _SC_THREAD_PRIORITY_SCHEDULING
6564 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
6565#endif
6566#ifdef _SC_THREAD_PRIO_INHERIT
6567 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
6568#endif
6569#ifdef _SC_THREAD_PRIO_PROTECT
6570 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
6571#endif
6572#ifdef _SC_THREAD_PROCESS_SHARED
6573 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
6574#endif
6575#ifdef _SC_THREAD_SAFE_FUNCTIONS
6576 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
6577#endif
6578#ifdef _SC_THREAD_STACK_MIN
6579 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
6580#endif
6581#ifdef _SC_THREAD_THREADS_MAX
6582 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
6583#endif
6584#ifdef _SC_TIMERS
6585 {"SC_TIMERS", _SC_TIMERS},
6586#endif
6587#ifdef _SC_TIMER_MAX
6588 {"SC_TIMER_MAX", _SC_TIMER_MAX},
6589#endif
6590#ifdef _SC_TTY_NAME_MAX
6591 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
6592#endif
6593#ifdef _SC_TZNAME_MAX
6594 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
6595#endif
6596#ifdef _SC_T_IOV_MAX
6597 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
6598#endif
6599#ifdef _SC_UCHAR_MAX
6600 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
6601#endif
6602#ifdef _SC_UINT_MAX
6603 {"SC_UINT_MAX", _SC_UINT_MAX},
6604#endif
6605#ifdef _SC_UIO_MAXIOV
6606 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
6607#endif
6608#ifdef _SC_ULONG_MAX
6609 {"SC_ULONG_MAX", _SC_ULONG_MAX},
6610#endif
6611#ifdef _SC_USHRT_MAX
6612 {"SC_USHRT_MAX", _SC_USHRT_MAX},
6613#endif
6614#ifdef _SC_VERSION
6615 {"SC_VERSION", _SC_VERSION},
6616#endif
6617#ifdef _SC_WORD_BIT
6618 {"SC_WORD_BIT", _SC_WORD_BIT},
6619#endif
6620#ifdef _SC_XBS5_ILP32_OFF32
6621 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
6622#endif
6623#ifdef _SC_XBS5_ILP32_OFFBIG
6624 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
6625#endif
6626#ifdef _SC_XBS5_LP64_OFF64
6627 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
6628#endif
6629#ifdef _SC_XBS5_LPBIG_OFFBIG
6630 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
6631#endif
6632#ifdef _SC_XOPEN_CRYPT
6633 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
6634#endif
6635#ifdef _SC_XOPEN_ENH_I18N
6636 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
6637#endif
6638#ifdef _SC_XOPEN_LEGACY
6639 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
6640#endif
6641#ifdef _SC_XOPEN_REALTIME
6642 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
6643#endif
6644#ifdef _SC_XOPEN_REALTIME_THREADS
6645 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
6646#endif
6647#ifdef _SC_XOPEN_SHM
6648 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
6649#endif
6650#ifdef _SC_XOPEN_UNIX
6651 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
6652#endif
6653#ifdef _SC_XOPEN_VERSION
6654 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
6655#endif
6656#ifdef _SC_XOPEN_XCU_VERSION
6657 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
6658#endif
6659#ifdef _SC_XOPEN_XPG2
6660 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
6661#endif
6662#ifdef _SC_XOPEN_XPG3
6663 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
6664#endif
6665#ifdef _SC_XOPEN_XPG4
6666 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
6667#endif
6668};
6669
6670static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006671conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00006672{
6673 return conv_confname(arg, valuep, posix_constants_sysconf,
6674 sizeof(posix_constants_sysconf)
6675 / sizeof(struct constdef));
6676}
6677
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006678PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006679"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006680Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00006681
6682static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006683posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006684{
6685 PyObject *result = NULL;
6686 int name;
6687
6688 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
6689 int value;
6690
6691 errno = 0;
6692 value = sysconf(name);
6693 if (value == -1 && errno != 0)
6694 posix_error();
6695 else
Christian Heimes217cfd12007-12-02 14:31:20 +00006696 result = PyLong_FromLong(value);
Fred Drakec9680921999-12-13 16:37:25 +00006697 }
6698 return result;
6699}
6700#endif
6701
6702
Fred Drakebec628d1999-12-15 18:31:10 +00006703/* This code is used to ensure that the tables of configuration value names
6704 * are in sorted order as required by conv_confname(), and also to build the
6705 * the exported dictionaries that are used to publish information about the
6706 * names available on the host platform.
6707 *
6708 * Sorting the table at runtime ensures that the table is properly ordered
6709 * when used, even for platforms we're not able to test on. It also makes
6710 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00006711 */
Fred Drakebec628d1999-12-15 18:31:10 +00006712
6713static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006714cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00006715{
6716 const struct constdef *c1 =
6717 (const struct constdef *) v1;
6718 const struct constdef *c2 =
6719 (const struct constdef *) v2;
6720
6721 return strcmp(c1->name, c2->name);
6722}
6723
6724static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006725setup_confname_table(struct constdef *table, size_t tablesize,
Fred Drake4d1e64b2002-04-15 19:40:07 +00006726 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00006727{
Fred Drakebec628d1999-12-15 18:31:10 +00006728 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00006729 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00006730
6731 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
6732 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00006733 if (d == NULL)
6734 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006735
Barry Warsaw3155db32000-04-13 15:20:40 +00006736 for (i=0; i < tablesize; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +00006737 PyObject *o = PyLong_FromLong(table[i].value);
Barry Warsaw3155db32000-04-13 15:20:40 +00006738 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
6739 Py_XDECREF(o);
6740 Py_DECREF(d);
6741 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006742 }
Barry Warsaw3155db32000-04-13 15:20:40 +00006743 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00006744 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00006745 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00006746}
6747
Fred Drakebec628d1999-12-15 18:31:10 +00006748/* Return -1 on failure, 0 on success. */
6749static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00006750setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00006751{
6752#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00006753 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00006754 sizeof(posix_constants_pathconf)
6755 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006756 "pathconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006757 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006758#endif
6759#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00006760 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00006761 sizeof(posix_constants_confstr)
6762 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006763 "confstr_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006764 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006765#endif
6766#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00006767 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00006768 sizeof(posix_constants_sysconf)
6769 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006770 "sysconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006771 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006772#endif
Fred Drakebec628d1999-12-15 18:31:10 +00006773 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00006774}
Fred Draked86ed291999-12-15 15:34:33 +00006775
6776
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006777PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006778"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006779Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006780in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006781
6782static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006783posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006784{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006785 abort();
6786 /*NOTREACHED*/
6787 Py_FatalError("abort() called from Python code didn't abort!");
6788 return NULL;
6789}
Fred Drakebec628d1999-12-15 18:31:10 +00006790
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006791#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006792PyDoc_STRVAR(win32_startfile__doc__,
Georg Brandlf4f44152006-02-18 22:29:33 +00006793"startfile(filepath [, operation]) - Start a file with its associated\n\
6794application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00006795\n\
Georg Brandlf4f44152006-02-18 22:29:33 +00006796When \"operation\" is not specified or \"open\", this acts like\n\
6797double-clicking the file in Explorer, or giving the file name as an\n\
6798argument to the DOS \"start\" command: the file is opened with whatever\n\
6799application (if any) its extension is associated.\n\
6800When another \"operation\" is given, it specifies what should be done with\n\
6801the file. A typical operation is \"print\".\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00006802\n\
6803startfile returns as soon as the associated application is launched.\n\
6804There is no option to wait for the application to close, and no way\n\
6805to retrieve the application's exit status.\n\
6806\n\
6807The filepath is relative to the current directory. If you want to use\n\
6808an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006809the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00006810
6811static PyObject *
6812win32_startfile(PyObject *self, PyObject *args)
6813{
Martin v. Löwis011e8422009-05-05 04:43:17 +00006814 PyObject *ofilepath;
Tim Petersf58a7aa2000-09-22 10:05:54 +00006815 char *filepath;
Georg Brandlf4f44152006-02-18 22:29:33 +00006816 char *operation = NULL;
Tim Petersf58a7aa2000-09-22 10:05:54 +00006817 HINSTANCE rc;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00006818
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006819 if (unicode_file_names()) {
6820 PyObject *unipath, *woperation = NULL;
6821 if (!PyArg_ParseTuple(args, "U|s:startfile",
6822 &unipath, &operation)) {
6823 PyErr_Clear();
6824 goto normal;
6825 }
6826
6827
6828 if (operation) {
6829 woperation = PyUnicode_DecodeASCII(operation,
6830 strlen(operation), NULL);
6831 if (!woperation) {
6832 PyErr_Clear();
6833 operation = NULL;
6834 goto normal;
6835 }
6836 }
6837
6838 Py_BEGIN_ALLOW_THREADS
6839 rc = ShellExecuteW((HWND)0, woperation ? PyUnicode_AS_UNICODE(woperation) : 0,
6840 PyUnicode_AS_UNICODE(unipath),
6841 NULL, NULL, SW_SHOWNORMAL);
6842 Py_END_ALLOW_THREADS
6843
6844 Py_XDECREF(woperation);
6845 if (rc <= (HINSTANCE)32) {
6846 PyObject *errval = win32_error_unicode("startfile",
6847 PyUnicode_AS_UNICODE(unipath));
6848 return errval;
6849 }
6850 Py_INCREF(Py_None);
6851 return Py_None;
6852 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006853
6854normal:
Martin v. Löwis011e8422009-05-05 04:43:17 +00006855 if (!PyArg_ParseTuple(args, "O&|s:startfile",
6856 PyUnicode_FSConverter, &ofilepath,
Georg Brandlf4f44152006-02-18 22:29:33 +00006857 &operation))
Tim Petersf58a7aa2000-09-22 10:05:54 +00006858 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00006859 filepath = bytes2str(ofilepath, 1);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006860 Py_BEGIN_ALLOW_THREADS
Georg Brandlf4f44152006-02-18 22:29:33 +00006861 rc = ShellExecute((HWND)0, operation, filepath,
6862 NULL, NULL, SW_SHOWNORMAL);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006863 Py_END_ALLOW_THREADS
Georg Brandle9f8ec92005-09-25 06:16:40 +00006864 if (rc <= (HINSTANCE)32) {
6865 PyObject *errval = win32_error("startfile", filepath);
Martin v. Löwis011e8422009-05-05 04:43:17 +00006866 release_bytes(ofilepath);
Georg Brandle9f8ec92005-09-25 06:16:40 +00006867 return errval;
6868 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00006869 release_bytes(ofilepath);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006870 Py_INCREF(Py_None);
6871 return Py_None;
6872}
6873#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006874
Martin v. Löwis438b5342002-12-27 10:16:42 +00006875#ifdef HAVE_GETLOADAVG
6876PyDoc_STRVAR(posix_getloadavg__doc__,
6877"getloadavg() -> (float, float, float)\n\n\
6878Return the number of processes in the system run queue averaged over\n\
6879the last 1, 5, and 15 minutes or raises OSError if the load average\n\
6880was unobtainable");
6881
6882static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006883posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00006884{
6885 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00006886 if (getloadavg(loadavg, 3)!=3) {
6887 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
6888 return NULL;
6889 } else
6890 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
6891}
6892#endif
6893
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006894#ifdef MS_WINDOWS
6895
6896PyDoc_STRVAR(win32_urandom__doc__,
6897"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00006898Return n random bytes suitable for cryptographic use.");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006899
6900typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
6901 LPCSTR pszContainer, LPCSTR pszProvider, DWORD dwProvType,\
6902 DWORD dwFlags );
6903typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV hProv, DWORD dwLen,\
6904 BYTE *pbBuffer );
6905
6906static CRYPTGENRANDOM pCryptGenRandom = NULL;
Thomas Wouters89d996e2007-09-08 17:39:28 +00006907/* This handle is never explicitly released. Instead, the operating
6908 system will release it when the process terminates. */
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006909static HCRYPTPROV hCryptProv = 0;
6910
Tim Peters4ad82172004-08-30 17:02:04 +00006911static PyObject*
6912win32_urandom(PyObject *self, PyObject *args)
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006913{
Tim Petersd3115382004-08-30 17:36:46 +00006914 int howMany;
6915 PyObject* result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006916
Tim Peters4ad82172004-08-30 17:02:04 +00006917 /* Read arguments */
Tim Peters9b279a82004-08-30 17:10:53 +00006918 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
Tim Peters4ad82172004-08-30 17:02:04 +00006919 return NULL;
Tim Peters51eba612004-08-30 17:08:02 +00006920 if (howMany < 0)
6921 return PyErr_Format(PyExc_ValueError,
6922 "negative argument not allowed");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006923
Tim Peters4ad82172004-08-30 17:02:04 +00006924 if (hCryptProv == 0) {
6925 HINSTANCE hAdvAPI32 = NULL;
6926 CRYPTACQUIRECONTEXTA pCryptAcquireContext = NULL;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006927
Tim Peters4ad82172004-08-30 17:02:04 +00006928 /* Obtain handle to the DLL containing CryptoAPI
6929 This should not fail */
6930 hAdvAPI32 = GetModuleHandle("advapi32.dll");
6931 if(hAdvAPI32 == NULL)
6932 return win32_error("GetModuleHandle", NULL);
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006933
Tim Peters4ad82172004-08-30 17:02:04 +00006934 /* Obtain pointers to the CryptoAPI functions
6935 This will fail on some early versions of Win95 */
6936 pCryptAcquireContext = (CRYPTACQUIRECONTEXTA)GetProcAddress(
6937 hAdvAPI32,
6938 "CryptAcquireContextA");
6939 if (pCryptAcquireContext == NULL)
6940 return PyErr_Format(PyExc_NotImplementedError,
6941 "CryptAcquireContextA not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006942
Tim Peters4ad82172004-08-30 17:02:04 +00006943 pCryptGenRandom = (CRYPTGENRANDOM)GetProcAddress(
6944 hAdvAPI32, "CryptGenRandom");
Thomas Wouters89f507f2006-12-13 04:49:30 +00006945 if (pCryptGenRandom == NULL)
Tim Peters4ad82172004-08-30 17:02:04 +00006946 return PyErr_Format(PyExc_NotImplementedError,
6947 "CryptGenRandom not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006948
Tim Peters4ad82172004-08-30 17:02:04 +00006949 /* Acquire context */
6950 if (! pCryptAcquireContext(&hCryptProv, NULL, NULL,
6951 PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
6952 return win32_error("CryptAcquireContext", NULL);
6953 }
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006954
Tim Peters4ad82172004-08-30 17:02:04 +00006955 /* Allocate bytes */
Christian Heimes72b710a2008-05-26 13:28:38 +00006956 result = PyBytes_FromStringAndSize(NULL, howMany);
Tim Petersd3115382004-08-30 17:36:46 +00006957 if (result != NULL) {
6958 /* Get random data */
Amaury Forgeot d'Arca05ada32008-07-21 21:13:14 +00006959 memset(PyBytes_AS_STRING(result), 0, howMany); /* zero seed */
Tim Petersd3115382004-08-30 17:36:46 +00006960 if (! pCryptGenRandom(hCryptProv, howMany, (unsigned char*)
Christian Heimes72b710a2008-05-26 13:28:38 +00006961 PyBytes_AS_STRING(result))) {
Tim Petersd3115382004-08-30 17:36:46 +00006962 Py_DECREF(result);
6963 return win32_error("CryptGenRandom", NULL);
6964 }
Tim Peters4ad82172004-08-30 17:02:04 +00006965 }
Tim Petersd3115382004-08-30 17:36:46 +00006966 return result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006967}
6968#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00006969
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00006970PyDoc_STRVAR(device_encoding__doc__,
6971"device_encoding(fd) -> str\n\n\
6972Return a string describing the encoding of the device\n\
6973if the output is a terminal; else return None.");
6974
6975static PyObject *
6976device_encoding(PyObject *self, PyObject *args)
6977{
6978 int fd;
6979 if (!PyArg_ParseTuple(args, "i:device_encoding", &fd))
6980 return NULL;
Kristján Valur Jónsson649170b2009-03-24 14:15:49 +00006981 if (!_PyVerify_fd(fd) || !isatty(fd)) {
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00006982 Py_INCREF(Py_None);
6983 return Py_None;
6984 }
6985#if defined(MS_WINDOWS) || defined(MS_WIN64)
6986 if (fd == 0) {
6987 char buf[100];
6988 sprintf(buf, "cp%d", GetConsoleCP());
6989 return PyUnicode_FromString(buf);
6990 }
6991 if (fd == 1 || fd == 2) {
6992 char buf[100];
6993 sprintf(buf, "cp%d", GetConsoleOutputCP());
6994 return PyUnicode_FromString(buf);
6995 }
6996#elif defined(CODESET)
6997 {
6998 char *codeset = nl_langinfo(CODESET);
Mark Dickinsonda2706b2008-12-11 18:03:03 +00006999 if (codeset != NULL && codeset[0] != 0)
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00007000 return PyUnicode_FromString(codeset);
7001 }
7002#endif
7003 Py_INCREF(Py_None);
7004 return Py_None;
7005}
7006
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007007#ifdef __VMS
7008/* Use openssl random routine */
7009#include <openssl/rand.h>
7010PyDoc_STRVAR(vms_urandom__doc__,
7011"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00007012Return n random bytes suitable for cryptographic use.");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007013
7014static PyObject*
7015vms_urandom(PyObject *self, PyObject *args)
7016{
7017 int howMany;
7018 PyObject* result;
7019
7020 /* Read arguments */
7021 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
7022 return NULL;
7023 if (howMany < 0)
7024 return PyErr_Format(PyExc_ValueError,
7025 "negative argument not allowed");
7026
7027 /* Allocate bytes */
Christian Heimes72b710a2008-05-26 13:28:38 +00007028 result = PyBytes_FromStringAndSize(NULL, howMany);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007029 if (result != NULL) {
7030 /* Get random data */
7031 if (RAND_pseudo_bytes((unsigned char*)
Christian Heimes72b710a2008-05-26 13:28:38 +00007032 PyBytes_AS_STRING(result),
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007033 howMany) < 0) {
7034 Py_DECREF(result);
7035 return PyErr_Format(PyExc_ValueError,
7036 "RAND_pseudo_bytes");
7037 }
7038 }
7039 return result;
7040}
7041#endif
7042
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007043static PyMethodDef posix_methods[] = {
7044 {"access", posix_access, METH_VARARGS, posix_access__doc__},
7045#ifdef HAVE_TTYNAME
7046 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
7047#endif
7048 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +00007049#ifdef HAVE_CHFLAGS
7050 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
7051#endif /* HAVE_CHFLAGS */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007052 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +00007053#ifdef HAVE_FCHMOD
7054 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
7055#endif /* HAVE_FCHMOD */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007056#ifdef HAVE_CHOWN
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007057 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007058#endif /* HAVE_CHOWN */
Christian Heimes4e30a842007-11-30 22:12:06 +00007059#ifdef HAVE_LCHMOD
7060 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
7061#endif /* HAVE_LCHMOD */
7062#ifdef HAVE_FCHOWN
7063 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
7064#endif /* HAVE_FCHOWN */
Thomas Wouterscf297e42007-02-23 15:07:44 +00007065#ifdef HAVE_LCHFLAGS
7066 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
7067#endif /* HAVE_LCHFLAGS */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00007068#ifdef HAVE_LCHOWN
7069 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
7070#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +00007071#ifdef HAVE_CHROOT
7072 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
7073#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007074#ifdef HAVE_CTERMID
Neal Norwitze241ce82003-02-17 18:17:05 +00007075 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007076#endif
Guido van Rossum36bc6801995-06-14 22:54:23 +00007077#ifdef HAVE_GETCWD
Guido van Rossumf0af3e32008-10-02 18:55:37 +00007078 {"getcwd", (PyCFunction)posix_getcwd_unicode,
7079 METH_NOARGS, posix_getcwd__doc__},
7080 {"getcwdb", (PyCFunction)posix_getcwd_bytes,
7081 METH_NOARGS, posix_getcwdb__doc__},
Guido van Rossum36bc6801995-06-14 22:54:23 +00007082#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00007083#ifdef HAVE_LINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007084 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007085#endif /* HAVE_LINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007086 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
7087 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
7088 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007089#ifdef HAVE_NICE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007090 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007091#endif /* HAVE_NICE */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007092#ifdef HAVE_READLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007093 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007094#endif /* HAVE_READLINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007095 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
7096 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
7097 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00007098 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007099#ifdef HAVE_SYMLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007100 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007101#endif /* HAVE_SYMLINK */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007102#ifdef HAVE_SYSTEM
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007103 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007104#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007105 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007106#ifdef HAVE_UNAME
Neal Norwitze241ce82003-02-17 18:17:05 +00007107 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007108#endif /* HAVE_UNAME */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007109 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
7110 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
7111 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007112#ifdef HAVE_TIMES
Neal Norwitze241ce82003-02-17 18:17:05 +00007113 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007114#endif /* HAVE_TIMES */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007115 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007116#ifdef HAVE_EXECV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007117 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
7118 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007119#endif /* HAVE_EXECV */
Guido van Rossuma1065681999-01-25 23:20:23 +00007120#ifdef HAVE_SPAWNV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007121 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
7122 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00007123#if defined(PYOS_OS2)
7124 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
7125 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
7126#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00007127#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +00007128#ifdef HAVE_FORK1
Neal Norwitze241ce82003-02-17 18:17:05 +00007129 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +00007130#endif /* HAVE_FORK1 */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007131#ifdef HAVE_FORK
Neal Norwitze241ce82003-02-17 18:17:05 +00007132 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007133#endif /* HAVE_FORK */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00007134#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Neal Norwitze241ce82003-02-17 18:17:05 +00007135 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +00007136#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00007137#ifdef HAVE_FORKPTY
Neal Norwitze241ce82003-02-17 18:17:05 +00007138 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +00007139#endif /* HAVE_FORKPTY */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007140#ifdef HAVE_GETEGID
Neal Norwitze241ce82003-02-17 18:17:05 +00007141 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007142#endif /* HAVE_GETEGID */
7143#ifdef HAVE_GETEUID
Neal Norwitze241ce82003-02-17 18:17:05 +00007144 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007145#endif /* HAVE_GETEUID */
7146#ifdef HAVE_GETGID
Neal Norwitze241ce82003-02-17 18:17:05 +00007147 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007148#endif /* HAVE_GETGID */
Fred Drakec9680921999-12-13 16:37:25 +00007149#ifdef HAVE_GETGROUPS
Neal Norwitze241ce82003-02-17 18:17:05 +00007150 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00007151#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00007152 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007153#ifdef HAVE_GETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00007154 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007155#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007156#ifdef HAVE_GETPPID
Neal Norwitze241ce82003-02-17 18:17:05 +00007157 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007158#endif /* HAVE_GETPPID */
7159#ifdef HAVE_GETUID
Neal Norwitze241ce82003-02-17 18:17:05 +00007160 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007161#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +00007162#ifdef HAVE_GETLOGIN
Neal Norwitze241ce82003-02-17 18:17:05 +00007163 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +00007164#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +00007165#ifdef HAVE_KILL
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007166 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007167#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00007168#ifdef HAVE_KILLPG
7169 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
7170#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +00007171#ifdef HAVE_PLOCK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007172 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +00007173#endif /* HAVE_PLOCK */
Thomas Heller8b7a9572007-08-31 06:44:36 +00007174#ifdef MS_WINDOWS
7175 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
7176#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00007177#ifdef HAVE_SETUID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007178 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007179#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00007180#ifdef HAVE_SETEUID
7181 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
7182#endif /* HAVE_SETEUID */
7183#ifdef HAVE_SETEGID
7184 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
7185#endif /* HAVE_SETEGID */
7186#ifdef HAVE_SETREUID
7187 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
7188#endif /* HAVE_SETREUID */
7189#ifdef HAVE_SETREGID
7190 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
7191#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007192#ifdef HAVE_SETGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007193 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007194#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00007195#ifdef HAVE_SETGROUPS
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00007196 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00007197#endif /* HAVE_SETGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +00007198#ifdef HAVE_GETPGID
7199 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
7200#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007201#ifdef HAVE_SETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00007202 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007203#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007204#ifdef HAVE_WAIT
Neal Norwitze241ce82003-02-17 18:17:05 +00007205 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007206#endif /* HAVE_WAIT */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007207#ifdef HAVE_WAIT3
7208 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
7209#endif /* HAVE_WAIT3 */
7210#ifdef HAVE_WAIT4
7211 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
7212#endif /* HAVE_WAIT4 */
Tim Petersab034fa2002-02-01 11:27:43 +00007213#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007214 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007215#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00007216#ifdef HAVE_GETSID
7217 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
7218#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007219#ifdef HAVE_SETSID
Neal Norwitze241ce82003-02-17 18:17:05 +00007220 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007221#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007222#ifdef HAVE_SETPGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007223 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007224#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007225#ifdef HAVE_TCGETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007226 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007227#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007228#ifdef HAVE_TCSETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007229 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007230#endif /* HAVE_TCSETPGRP */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007231 {"open", posix_open, METH_VARARGS, posix_open__doc__},
7232 {"close", posix_close, METH_VARARGS, posix_close__doc__},
Christian Heimesfdab48e2008-01-20 09:06:41 +00007233 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00007234 {"device_encoding", device_encoding, METH_VARARGS, device_encoding__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007235 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
7236 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
7237 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
7238 {"read", posix_read, METH_VARARGS, posix_read__doc__},
7239 {"write", posix_write, METH_VARARGS, posix_write__doc__},
7240 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
Skip Montanaro1517d842000-07-19 14:34:14 +00007241 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007242#ifdef HAVE_PIPE
Neal Norwitze241ce82003-02-17 18:17:05 +00007243 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007244#endif
7245#ifdef HAVE_MKFIFO
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007246 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007247#endif
Neal Norwitz11690112002-07-30 01:08:28 +00007248#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007249 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
7250#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007251#ifdef HAVE_DEVICE_MACROS
7252 {"major", posix_major, METH_VARARGS, posix_major__doc__},
7253 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
7254 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
7255#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007256#ifdef HAVE_FTRUNCATE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007257 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007258#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007259#ifdef HAVE_PUTENV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007260 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007261#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00007262#ifdef HAVE_UNSETENV
7263 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
7264#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007265 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +00007266#ifdef HAVE_FCHDIR
7267 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
7268#endif
Guido van Rossum21142a01999-01-08 21:05:37 +00007269#ifdef HAVE_FSYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007270 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007271#endif
7272#ifdef HAVE_FDATASYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007273 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007274#endif
Guido van Rossumc9641791998-08-04 15:26:23 +00007275#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +00007276#ifdef WCOREDUMP
7277 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
7278#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00007279#ifdef WIFCONTINUED
7280 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
7281#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +00007282#ifdef WIFSTOPPED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007283 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007284#endif /* WIFSTOPPED */
7285#ifdef WIFSIGNALED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007286 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007287#endif /* WIFSIGNALED */
7288#ifdef WIFEXITED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007289 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007290#endif /* WIFEXITED */
7291#ifdef WEXITSTATUS
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007292 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007293#endif /* WEXITSTATUS */
7294#ifdef WTERMSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007295 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007296#endif /* WTERMSIG */
7297#ifdef WSTOPSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007298 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007299#endif /* WSTOPSIG */
7300#endif /* HAVE_SYS_WAIT_H */
Thomas Wouters477c8d52006-05-27 19:21:47 +00007301#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007302 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007303#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00007304#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007305 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007306#endif
Fred Drakec9680921999-12-13 16:37:25 +00007307#ifdef HAVE_CONFSTR
7308 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
7309#endif
7310#ifdef HAVE_SYSCONF
7311 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
7312#endif
7313#ifdef HAVE_FPATHCONF
7314 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
7315#endif
7316#ifdef HAVE_PATHCONF
7317 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
7318#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00007319 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007320#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00007321 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
7322#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00007323#ifdef HAVE_GETLOADAVG
Neal Norwitze241ce82003-02-17 18:17:05 +00007324 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +00007325#endif
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00007326 #ifdef MS_WINDOWS
7327 {"urandom", win32_urandom, METH_VARARGS, win32_urandom__doc__},
7328 #endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007329 #ifdef __VMS
7330 {"urandom", vms_urandom, METH_VARARGS, vms_urandom__doc__},
7331 #endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007332 {NULL, NULL} /* Sentinel */
7333};
7334
7335
Barry Warsaw4a342091996-12-19 23:50:02 +00007336static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00007337ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +00007338{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007339 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +00007340}
7341
Guido van Rossumd48f2521997-12-05 22:19:34 +00007342#if defined(PYOS_OS2)
7343/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007344static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007345{
7346 APIRET rc;
7347 ULONG values[QSV_MAX+1];
7348 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +00007349 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +00007350
7351 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +00007352 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007353 Py_END_ALLOW_THREADS
7354
7355 if (rc != NO_ERROR) {
7356 os2_error(rc);
7357 return -1;
7358 }
7359
Fred Drake4d1e64b2002-04-15 19:40:07 +00007360 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
7361 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
7362 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
7363 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
7364 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
7365 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
7366 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007367
7368 switch (values[QSV_VERSION_MINOR]) {
7369 case 0: ver = "2.00"; break;
7370 case 10: ver = "2.10"; break;
7371 case 11: ver = "2.11"; break;
7372 case 30: ver = "3.00"; break;
7373 case 40: ver = "4.00"; break;
7374 case 50: ver = "5.00"; break;
7375 default:
Tim Peters885d4572001-11-28 20:27:42 +00007376 PyOS_snprintf(tmp, sizeof(tmp),
7377 "%d-%d", values[QSV_VERSION_MAJOR],
7378 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007379 ver = &tmp[0];
7380 }
7381
7382 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007383 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007384 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007385
7386 /* Add Indicator of Which Drive was Used to Boot the System */
7387 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
7388 tmp[1] = ':';
7389 tmp[2] = '\0';
7390
Fred Drake4d1e64b2002-04-15 19:40:07 +00007391 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007392}
7393#endif
7394
Barry Warsaw4a342091996-12-19 23:50:02 +00007395static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007396all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +00007397{
Guido van Rossum94f6f721999-01-06 18:42:14 +00007398#ifdef F_OK
7399 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007400#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007401#ifdef R_OK
7402 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007403#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007404#ifdef W_OK
7405 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007406#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007407#ifdef X_OK
7408 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007409#endif
Fred Drakec9680921999-12-13 16:37:25 +00007410#ifdef NGROUPS_MAX
7411 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
7412#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007413#ifdef TMP_MAX
7414 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
7415#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007416#ifdef WCONTINUED
7417 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
7418#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007419#ifdef WNOHANG
7420 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007421#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007422#ifdef WUNTRACED
7423 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
7424#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007425#ifdef O_RDONLY
7426 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
7427#endif
7428#ifdef O_WRONLY
7429 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
7430#endif
7431#ifdef O_RDWR
7432 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
7433#endif
7434#ifdef O_NDELAY
7435 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
7436#endif
7437#ifdef O_NONBLOCK
7438 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
7439#endif
7440#ifdef O_APPEND
7441 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
7442#endif
7443#ifdef O_DSYNC
7444 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
7445#endif
7446#ifdef O_RSYNC
7447 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
7448#endif
7449#ifdef O_SYNC
7450 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
7451#endif
7452#ifdef O_NOCTTY
7453 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
7454#endif
7455#ifdef O_CREAT
7456 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
7457#endif
7458#ifdef O_EXCL
7459 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
7460#endif
7461#ifdef O_TRUNC
7462 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
7463#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +00007464#ifdef O_BINARY
7465 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
7466#endif
7467#ifdef O_TEXT
7468 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
7469#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007470#ifdef O_LARGEFILE
7471 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
7472#endif
Skip Montanaro5ff14922005-05-16 02:42:22 +00007473#ifdef O_SHLOCK
7474 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
7475#endif
7476#ifdef O_EXLOCK
7477 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
7478#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007479
Tim Peters5aa91602002-01-30 05:46:57 +00007480/* MS Windows */
7481#ifdef O_NOINHERIT
7482 /* Don't inherit in child processes. */
7483 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
7484#endif
7485#ifdef _O_SHORT_LIVED
7486 /* Optimize for short life (keep in memory). */
7487 /* MS forgot to define this one with a non-underscore form too. */
7488 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
7489#endif
7490#ifdef O_TEMPORARY
7491 /* Automatically delete when last handle is closed. */
7492 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
7493#endif
7494#ifdef O_RANDOM
7495 /* Optimize for random access. */
7496 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
7497#endif
7498#ifdef O_SEQUENTIAL
7499 /* Optimize for sequential access. */
7500 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
7501#endif
7502
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007503/* GNU extensions. */
Alexandre Vassalottibee32532008-05-16 18:15:12 +00007504#ifdef O_ASYNC
7505 /* Send a SIGIO signal whenever input or output
7506 becomes available on file descriptor */
7507 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
7508#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007509#ifdef O_DIRECT
7510 /* Direct disk access. */
7511 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
7512#endif
7513#ifdef O_DIRECTORY
7514 /* Must be a directory. */
7515 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
7516#endif
7517#ifdef O_NOFOLLOW
7518 /* Do not follow links. */
7519 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
7520#endif
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +00007521#ifdef O_NOATIME
7522 /* Do not update the access time. */
7523 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
7524#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00007525
Barry Warsaw5676bd12003-01-07 20:57:09 +00007526 /* These come from sysexits.h */
7527#ifdef EX_OK
7528 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007529#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007530#ifdef EX_USAGE
7531 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007532#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007533#ifdef EX_DATAERR
7534 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007535#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007536#ifdef EX_NOINPUT
7537 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007538#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007539#ifdef EX_NOUSER
7540 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007541#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007542#ifdef EX_NOHOST
7543 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007544#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007545#ifdef EX_UNAVAILABLE
7546 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007547#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007548#ifdef EX_SOFTWARE
7549 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007550#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007551#ifdef EX_OSERR
7552 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007553#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007554#ifdef EX_OSFILE
7555 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007556#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007557#ifdef EX_CANTCREAT
7558 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007559#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007560#ifdef EX_IOERR
7561 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007562#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007563#ifdef EX_TEMPFAIL
7564 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007565#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007566#ifdef EX_PROTOCOL
7567 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007568#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007569#ifdef EX_NOPERM
7570 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007571#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007572#ifdef EX_CONFIG
7573 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007574#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007575#ifdef EX_NOTFOUND
7576 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007577#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007578
Guido van Rossum246bc171999-02-01 23:54:31 +00007579#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007580#if defined(PYOS_OS2) && defined(PYCC_GCC)
7581 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
7582 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
7583 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
7584 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
7585 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
7586 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
7587 if (ins(d, "P_PM", (long)P_PM)) return -1;
7588 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
7589 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
7590 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
7591 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
7592 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
7593 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
7594 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
7595 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
7596 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
7597 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
7598 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
7599 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
7600 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
7601#else
Guido van Rossum7d385291999-02-16 19:38:04 +00007602 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
7603 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
7604 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
7605 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
7606 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +00007607#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007608#endif
Guido van Rossum246bc171999-02-01 23:54:31 +00007609
Guido van Rossumd48f2521997-12-05 22:19:34 +00007610#if defined(PYOS_OS2)
7611 if (insertvalues(d)) return -1;
7612#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007613 return 0;
7614}
7615
7616
Tim Peters5aa91602002-01-30 05:46:57 +00007617#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007618#define INITFUNC PyInit_nt
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007619#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007620
7621#elif defined(PYOS_OS2)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007622#define INITFUNC PyInit_os2
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007623#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007624
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007625#else
Martin v. Löwis1a214512008-06-11 05:26:20 +00007626#define INITFUNC PyInit_posix
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007627#define MODNAME "posix"
7628#endif
7629
Martin v. Löwis1a214512008-06-11 05:26:20 +00007630static struct PyModuleDef posixmodule = {
7631 PyModuleDef_HEAD_INIT,
7632 MODNAME,
7633 posix__doc__,
7634 -1,
7635 posix_methods,
7636 NULL,
7637 NULL,
7638 NULL,
7639 NULL
7640};
7641
7642
Mark Hammondfe51c6d2002-08-02 02:27:13 +00007643PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007644INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +00007645{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007646 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +00007647
Martin v. Löwis1a214512008-06-11 05:26:20 +00007648 m = PyModule_Create(&posixmodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00007649 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007650 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007651
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007652 /* Initialize environ dictionary */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007653 v = convertenviron();
Fred Drake4d1e64b2002-04-15 19:40:07 +00007654 Py_XINCREF(v);
7655 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007656 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00007657 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +00007658
Fred Drake4d1e64b2002-04-15 19:40:07 +00007659 if (all_ins(m))
Martin v. Löwis1a214512008-06-11 05:26:20 +00007660 return NULL;
Barry Warsaw4a342091996-12-19 23:50:02 +00007661
Fred Drake4d1e64b2002-04-15 19:40:07 +00007662 if (setup_confname_tables(m))
Martin v. Löwis1a214512008-06-11 05:26:20 +00007663 return NULL;
Fred Drakebec628d1999-12-15 18:31:10 +00007664
Fred Drake4d1e64b2002-04-15 19:40:07 +00007665 Py_INCREF(PyExc_OSError);
7666 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +00007667
Guido van Rossumb3d39562000-01-31 18:41:26 +00007668#ifdef HAVE_PUTENV
Neil Schemenauer19030a02001-01-16 04:27:47 +00007669 if (posix_putenv_garbage == NULL)
7670 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +00007671#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007672
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007673 if (!initialized) {
7674 stat_result_desc.name = MODNAME ".stat_result";
7675 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
7676 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
7677 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
7678 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
7679 structseq_new = StatResultType.tp_new;
7680 StatResultType.tp_new = statresult_new;
7681
7682 statvfs_result_desc.name = MODNAME ".statvfs_result";
7683 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00007684#ifdef NEED_TICKS_PER_SECOND
7685# if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
7686 ticks_per_second = sysconf(_SC_CLK_TCK);
7687# elif defined(HZ)
7688 ticks_per_second = HZ;
7689# else
7690 ticks_per_second = 60; /* magic fallback value; may be bogus */
7691# endif
7692#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007693 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00007694 Py_INCREF((PyObject*) &StatResultType);
7695 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
Fred Drake4d1e64b2002-04-15 19:40:07 +00007696 Py_INCREF((PyObject*) &StatVFSResultType);
7697 PyModule_AddObject(m, "statvfs_result",
7698 (PyObject*) &StatVFSResultType);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007699 initialized = 1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007700
7701#ifdef __APPLE__
7702 /*
7703 * Step 2 of weak-linking support on Mac OS X.
7704 *
7705 * The code below removes functions that are not available on the
7706 * currently active platform.
7707 *
7708 * This block allow one to use a python binary that was build on
7709 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
7710 * OSX 10.4.
7711 */
7712#ifdef HAVE_FSTATVFS
7713 if (fstatvfs == NULL) {
7714 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007715 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007716 }
7717 }
7718#endif /* HAVE_FSTATVFS */
7719
7720#ifdef HAVE_STATVFS
7721 if (statvfs == NULL) {
7722 if (PyObject_DelAttrString(m, "statvfs") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007723 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007724 }
7725 }
7726#endif /* HAVE_STATVFS */
7727
7728# ifdef HAVE_LCHOWN
7729 if (lchown == NULL) {
7730 if (PyObject_DelAttrString(m, "lchown") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007731 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007732 }
7733 }
7734#endif /* HAVE_LCHOWN */
7735
7736
7737#endif /* __APPLE__ */
Martin v. Löwis1a214512008-06-11 05:26:20 +00007738 return m;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007739
Guido van Rossumb6775db1994-08-01 11:34:53 +00007740}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007741
7742#ifdef __cplusplus
7743}
7744#endif