blob: 3ed8d5f71f0c4886ab4f736fb81e5339affc230e [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
762static PyObject *
Thomas Wouters477c8d52006-05-27 19:21:47 +0000763posix_1str(PyObject *args, char *format, int (*func)(const char*))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000764{
Martin v. Löwis011e8422009-05-05 04:43:17 +0000765 PyObject *opath1 = NULL;
766 char *path1;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000767 int res;
Tim Peters5aa91602002-01-30 05:46:57 +0000768 if (!PyArg_ParseTuple(args, format,
Martin v. Löwis011e8422009-05-05 04:43:17 +0000769 PyUnicode_FSConverter, &opath1))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000770 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +0000771 path1 = bytes2str(opath1, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000772 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000773 res = (*func)(path1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000774 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000775 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +0000776 return posix_error_with_allocated_filename(opath1);
777 release_bytes(opath1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000778 Py_INCREF(Py_None);
779 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000780}
781
Barry Warsaw53699e91996-12-10 23:23:01 +0000782static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +0000783posix_2str(PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000784 char *format,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000785 int (*func)(const char *, const char *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000786{
Antoine Pitroua6bb9842009-05-10 22:27:00 +0000787 PyObject *opath1 = NULL, *opath2 = NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +0000788 char *path1, *path2;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000789 int res;
Mark Hammondef8b6542001-05-13 08:04:26 +0000790 if (!PyArg_ParseTuple(args, format,
Martin v. Löwis011e8422009-05-05 04:43:17 +0000791 PyUnicode_FSConverter, &opath1,
Antoine Pitroua6bb9842009-05-10 22:27:00 +0000792 PyUnicode_FSConverter, &opath2)) {
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000793 return NULL;
Antoine Pitroua6bb9842009-05-10 22:27:00 +0000794 }
Martin v. Löwis011e8422009-05-05 04:43:17 +0000795 path1 = bytes2str(opath1, 1);
796 path2 = bytes2str(opath2, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000797 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000798 res = (*func)(path1, path2);
Barry Warsaw53699e91996-12-10 23:23:01 +0000799 Py_END_ALLOW_THREADS
Martin v. Löwis011e8422009-05-05 04:43:17 +0000800 release_bytes(opath1);
801 release_bytes(opath2);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000802 if (res != 0)
Barry Warsawd58d7641998-07-23 16:14:40 +0000803 /* XXX how to report both path1 and path2??? */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000804 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +0000805 Py_INCREF(Py_None);
806 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000807}
808
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000809#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +0000810static PyObject*
811win32_1str(PyObject* args, char* func,
812 char* format, BOOL (__stdcall *funcA)(LPCSTR),
813 char* wformat, BOOL (__stdcall *funcW)(LPWSTR))
814{
815 PyObject *uni;
816 char *ansi;
817 BOOL result;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +0000818
819 if (!PyArg_ParseTuple(args, wformat, &uni))
820 PyErr_Clear();
821 else {
822 Py_BEGIN_ALLOW_THREADS
823 result = funcW(PyUnicode_AsUnicode(uni));
824 Py_END_ALLOW_THREADS
825 if (!result)
826 return win32_error_unicode(func, PyUnicode_AsUnicode(uni));
827 Py_INCREF(Py_None);
828 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000829 }
830 if (!PyArg_ParseTuple(args, format, &ansi))
831 return NULL;
832 Py_BEGIN_ALLOW_THREADS
833 result = funcA(ansi);
834 Py_END_ALLOW_THREADS
835 if (!result)
836 return win32_error(func, ansi);
837 Py_INCREF(Py_None);
838 return Py_None;
839
840}
841
842/* This is a reimplementation of the C library's chdir function,
843 but one that produces Win32 errors instead of DOS error codes.
844 chdir is essentially a wrapper around SetCurrentDirectory; however,
845 it also needs to set "magic" environment variables indicating
846 the per-drive current directory, which are of the form =<drive>: */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000847static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000848win32_chdir(LPCSTR path)
849{
850 char new_path[MAX_PATH+1];
851 int result;
852 char env[4] = "=x:";
853
854 if(!SetCurrentDirectoryA(path))
855 return FALSE;
856 result = GetCurrentDirectoryA(MAX_PATH+1, new_path);
857 if (!result)
858 return FALSE;
859 /* In the ANSI API, there should not be any paths longer
860 than MAX_PATH. */
861 assert(result <= MAX_PATH+1);
862 if (strncmp(new_path, "\\\\", 2) == 0 ||
863 strncmp(new_path, "//", 2) == 0)
864 /* UNC path, nothing to do. */
865 return TRUE;
866 env[1] = new_path[0];
867 return SetEnvironmentVariableA(env, new_path);
868}
869
870/* The Unicode version differs from the ANSI version
871 since the current directory might exceed MAX_PATH characters */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000872static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000873win32_wchdir(LPCWSTR path)
874{
875 wchar_t _new_path[MAX_PATH+1], *new_path = _new_path;
876 int result;
877 wchar_t env[4] = L"=x:";
878
879 if(!SetCurrentDirectoryW(path))
880 return FALSE;
881 result = GetCurrentDirectoryW(MAX_PATH+1, new_path);
882 if (!result)
883 return FALSE;
884 if (result > MAX_PATH+1) {
Benjamin Petersonf10a79a2008-10-11 00:49:57 +0000885 new_path = malloc(result * sizeof(wchar_t));
Thomas Wouters477c8d52006-05-27 19:21:47 +0000886 if (!new_path) {
887 SetLastError(ERROR_OUTOFMEMORY);
888 return FALSE;
889 }
Benjamin Petersonf10a79a2008-10-11 00:49:57 +0000890 result = GetCurrentDirectoryW(result, new_path);
891 if (!result) {
892 free(new_path);
893 return FALSE;
894 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000895 }
896 if (wcsncmp(new_path, L"\\\\", 2) == 0 ||
897 wcsncmp(new_path, L"//", 2) == 0)
898 /* UNC path, nothing to do. */
899 return TRUE;
900 env[1] = new_path[0];
901 result = SetEnvironmentVariableW(env, new_path);
902 if (new_path != _new_path)
903 free(new_path);
904 return result;
905}
906#endif
907
Martin v. Löwis14694662006-02-03 12:54:16 +0000908#ifdef MS_WINDOWS
909/* The CRT of Windows has a number of flaws wrt. its stat() implementation:
910 - time stamps are restricted to second resolution
911 - file modification times suffer from forth-and-back conversions between
912 UTC and local time
913 Therefore, we implement our own stat, based on the Win32 API directly.
914*/
915#define HAVE_STAT_NSEC 1
916
917struct win32_stat{
918 int st_dev;
919 __int64 st_ino;
920 unsigned short st_mode;
921 int st_nlink;
922 int st_uid;
923 int st_gid;
924 int st_rdev;
925 __int64 st_size;
926 int st_atime;
927 int st_atime_nsec;
928 int st_mtime;
929 int st_mtime_nsec;
930 int st_ctime;
931 int st_ctime_nsec;
932};
933
934static __int64 secs_between_epochs = 11644473600; /* Seconds between 1.1.1601 and 1.1.1970 */
935
936static void
937FILE_TIME_to_time_t_nsec(FILETIME *in_ptr, int *time_out, int* nsec_out)
938{
Thomas Wouters477c8d52006-05-27 19:21:47 +0000939 /* XXX endianness. Shouldn't matter, as all Windows implementations are little-endian */
940 /* Cannot simply cast and dereference in_ptr,
941 since it might not be aligned properly */
942 __int64 in;
943 memcpy(&in, in_ptr, sizeof(in));
Martin v. Löwis14694662006-02-03 12:54:16 +0000944 *nsec_out = (int)(in % 10000000) * 100; /* FILETIME is in units of 100 nsec. */
945 /* XXX Win32 supports time stamps past 2038; we currently don't */
946 *time_out = Py_SAFE_DOWNCAST((in / 10000000) - secs_between_epochs, __int64, int);
947}
948
Thomas Wouters477c8d52006-05-27 19:21:47 +0000949static void
950time_t_to_FILE_TIME(int time_in, int nsec_in, FILETIME *out_ptr)
951{
952 /* XXX endianness */
953 __int64 out;
954 out = time_in + secs_between_epochs;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000955 out = out * 10000000 + nsec_in / 100;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000956 memcpy(out_ptr, &out, sizeof(out));
957}
958
Martin v. Löwis14694662006-02-03 12:54:16 +0000959/* Below, we *know* that ugo+r is 0444 */
960#if _S_IREAD != 0400
961#error Unsupported C library
962#endif
963static int
964attributes_to_mode(DWORD attr)
965{
966 int m = 0;
967 if (attr & FILE_ATTRIBUTE_DIRECTORY)
968 m |= _S_IFDIR | 0111; /* IFEXEC for user,group,other */
969 else
970 m |= _S_IFREG;
971 if (attr & FILE_ATTRIBUTE_READONLY)
972 m |= 0444;
973 else
974 m |= 0666;
975 return m;
976}
977
978static int
979attribute_data_to_stat(WIN32_FILE_ATTRIBUTE_DATA *info, struct win32_stat *result)
980{
981 memset(result, 0, sizeof(*result));
982 result->st_mode = attributes_to_mode(info->dwFileAttributes);
983 result->st_size = (((__int64)info->nFileSizeHigh)<<32) + info->nFileSizeLow;
984 FILE_TIME_to_time_t_nsec(&info->ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
985 FILE_TIME_to_time_t_nsec(&info->ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
986 FILE_TIME_to_time_t_nsec(&info->ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
987
988 return 0;
989}
990
Guido van Rossumd8faa362007-04-27 19:54:29 +0000991static BOOL
992attributes_from_dir(LPCSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
993{
994 HANDLE hFindFile;
995 WIN32_FIND_DATAA FileData;
996 hFindFile = FindFirstFileA(pszFile, &FileData);
997 if (hFindFile == INVALID_HANDLE_VALUE)
998 return FALSE;
999 FindClose(hFindFile);
1000 pfad->dwFileAttributes = FileData.dwFileAttributes;
1001 pfad->ftCreationTime = FileData.ftCreationTime;
1002 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
1003 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
1004 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
1005 pfad->nFileSizeLow = FileData.nFileSizeLow;
1006 return TRUE;
1007}
1008
1009static BOOL
1010attributes_from_dir_w(LPCWSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
1011{
1012 HANDLE hFindFile;
1013 WIN32_FIND_DATAW FileData;
1014 hFindFile = FindFirstFileW(pszFile, &FileData);
1015 if (hFindFile == INVALID_HANDLE_VALUE)
1016 return FALSE;
1017 FindClose(hFindFile);
1018 pfad->dwFileAttributes = FileData.dwFileAttributes;
1019 pfad->ftCreationTime = FileData.ftCreationTime;
1020 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
1021 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
1022 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
1023 pfad->nFileSizeLow = FileData.nFileSizeLow;
1024 return TRUE;
1025}
1026
Martin v. Löwis14694662006-02-03 12:54:16 +00001027static int
1028win32_stat(const char* path, struct win32_stat *result)
1029{
1030 WIN32_FILE_ATTRIBUTE_DATA info;
1031 int code;
1032 char *dot;
Hirokazu Yamamoto6fbdfda2009-06-29 11:37:19 +00001033 if (!GetFileAttributesExA(path, GetFileExInfoStandard, &info)) {
Guido van Rossumd8faa362007-04-27 19:54:29 +00001034 if (GetLastError() != ERROR_SHARING_VIOLATION) {
1035 /* Protocol violation: we explicitly clear errno, instead of
1036 setting it to a POSIX error. Callers should use GetLastError. */
1037 errno = 0;
1038 return -1;
1039 } else {
1040 /* Could not get attributes on open file. Fall back to
1041 reading the directory. */
1042 if (!attributes_from_dir(path, &info)) {
1043 /* Very strange. This should not fail now */
1044 errno = 0;
1045 return -1;
1046 }
1047 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001048 }
1049 code = attribute_data_to_stat(&info, result);
1050 if (code != 0)
1051 return code;
1052 /* Set S_IFEXEC if it is an .exe, .bat, ... */
1053 dot = strrchr(path, '.');
1054 if (dot) {
1055 if (stricmp(dot, ".bat") == 0 ||
1056 stricmp(dot, ".cmd") == 0 ||
1057 stricmp(dot, ".exe") == 0 ||
1058 stricmp(dot, ".com") == 0)
1059 result->st_mode |= 0111;
1060 }
1061 return code;
1062}
1063
1064static int
1065win32_wstat(const wchar_t* path, struct win32_stat *result)
1066{
1067 int code;
1068 const wchar_t *dot;
1069 WIN32_FILE_ATTRIBUTE_DATA info;
Hirokazu Yamamoto6fbdfda2009-06-29 11:37:19 +00001070 if (!GetFileAttributesExW(path, GetFileExInfoStandard, &info)) {
Guido van Rossumd8faa362007-04-27 19:54:29 +00001071 if (GetLastError() != ERROR_SHARING_VIOLATION) {
1072 /* Protocol violation: we explicitly clear errno, instead of
1073 setting it to a POSIX error. Callers should use GetLastError. */
1074 errno = 0;
1075 return -1;
1076 } else {
1077 /* Could not get attributes on open file. Fall back to
1078 reading the directory. */
1079 if (!attributes_from_dir_w(path, &info)) {
1080 /* Very strange. This should not fail now */
1081 errno = 0;
1082 return -1;
1083 }
1084 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001085 }
1086 code = attribute_data_to_stat(&info, result);
1087 if (code < 0)
1088 return code;
1089 /* Set IFEXEC if it is an .exe, .bat, ... */
1090 dot = wcsrchr(path, '.');
1091 if (dot) {
1092 if (_wcsicmp(dot, L".bat") == 0 ||
1093 _wcsicmp(dot, L".cmd") == 0 ||
1094 _wcsicmp(dot, L".exe") == 0 ||
1095 _wcsicmp(dot, L".com") == 0)
1096 result->st_mode |= 0111;
1097 }
1098 return code;
1099}
1100
1101static int
1102win32_fstat(int file_number, struct win32_stat *result)
1103{
1104 BY_HANDLE_FILE_INFORMATION info;
1105 HANDLE h;
1106 int type;
1107
1108 h = (HANDLE)_get_osfhandle(file_number);
1109
1110 /* Protocol violation: we explicitly clear errno, instead of
1111 setting it to a POSIX error. Callers should use GetLastError. */
1112 errno = 0;
1113
1114 if (h == INVALID_HANDLE_VALUE) {
1115 /* This is really a C library error (invalid file handle).
1116 We set the Win32 error to the closes one matching. */
1117 SetLastError(ERROR_INVALID_HANDLE);
1118 return -1;
1119 }
1120 memset(result, 0, sizeof(*result));
1121
1122 type = GetFileType(h);
1123 if (type == FILE_TYPE_UNKNOWN) {
1124 DWORD error = GetLastError();
1125 if (error != 0) {
1126 return -1;
1127 }
1128 /* else: valid but unknown file */
1129 }
1130
1131 if (type != FILE_TYPE_DISK) {
1132 if (type == FILE_TYPE_CHAR)
1133 result->st_mode = _S_IFCHR;
1134 else if (type == FILE_TYPE_PIPE)
1135 result->st_mode = _S_IFIFO;
1136 return 0;
1137 }
1138
1139 if (!GetFileInformationByHandle(h, &info)) {
1140 return -1;
1141 }
1142
1143 /* similar to stat() */
1144 result->st_mode = attributes_to_mode(info.dwFileAttributes);
1145 result->st_size = (((__int64)info.nFileSizeHigh)<<32) + info.nFileSizeLow;
1146 FILE_TIME_to_time_t_nsec(&info.ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1147 FILE_TIME_to_time_t_nsec(&info.ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1148 FILE_TIME_to_time_t_nsec(&info.ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
1149 /* specific to fstat() */
1150 result->st_nlink = info.nNumberOfLinks;
1151 result->st_ino = (((__int64)info.nFileIndexHigh)<<32) + info.nFileIndexLow;
1152 return 0;
1153}
1154
1155#endif /* MS_WINDOWS */
1156
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001157PyDoc_STRVAR(stat_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001158"stat_result: Result from stat or lstat.\n\n\
1159This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001160 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001161or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
1162\n\
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001163Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
1164or st_flags, they are available as attributes only.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001165\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001166See os.stat for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001167
1168static PyStructSequence_Field stat_result_fields[] = {
1169 {"st_mode", "protection bits"},
1170 {"st_ino", "inode"},
1171 {"st_dev", "device"},
1172 {"st_nlink", "number of hard links"},
1173 {"st_uid", "user ID of owner"},
1174 {"st_gid", "group ID of owner"},
1175 {"st_size", "total size, in bytes"},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001176 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
1177 {NULL, "integer time of last access"},
1178 {NULL, "integer time of last modification"},
1179 {NULL, "integer time of last change"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001180 {"st_atime", "time of last access"},
1181 {"st_mtime", "time of last modification"},
1182 {"st_ctime", "time of last change"},
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001183#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001184 {"st_blksize", "blocksize for filesystem I/O"},
1185#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001186#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001187 {"st_blocks", "number of blocks allocated"},
1188#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001189#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001190 {"st_rdev", "device type (if inode device)"},
1191#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001192#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1193 {"st_flags", "user defined flags for file"},
1194#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001195#ifdef HAVE_STRUCT_STAT_ST_GEN
1196 {"st_gen", "generation number"},
1197#endif
1198#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1199 {"st_birthtime", "time of creation"},
1200#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001201 {0}
1202};
1203
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001204#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001205#define ST_BLKSIZE_IDX 13
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001206#else
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001207#define ST_BLKSIZE_IDX 12
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001208#endif
1209
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001210#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001211#define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
1212#else
1213#define ST_BLOCKS_IDX ST_BLKSIZE_IDX
1214#endif
1215
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001216#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001217#define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
1218#else
1219#define ST_RDEV_IDX ST_BLOCKS_IDX
1220#endif
1221
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001222#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1223#define ST_FLAGS_IDX (ST_RDEV_IDX+1)
1224#else
1225#define ST_FLAGS_IDX ST_RDEV_IDX
1226#endif
1227
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001228#ifdef HAVE_STRUCT_STAT_ST_GEN
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001229#define ST_GEN_IDX (ST_FLAGS_IDX+1)
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001230#else
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001231#define ST_GEN_IDX ST_FLAGS_IDX
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001232#endif
1233
1234#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1235#define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
1236#else
1237#define ST_BIRTHTIME_IDX ST_GEN_IDX
1238#endif
1239
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001240static PyStructSequence_Desc stat_result_desc = {
1241 "stat_result", /* name */
1242 stat_result__doc__, /* doc */
1243 stat_result_fields,
1244 10
1245};
1246
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001247PyDoc_STRVAR(statvfs_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001248"statvfs_result: Result from statvfs or fstatvfs.\n\n\
1249This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001250 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
Guido van Rossuma4dc73e2001-10-18 20:53:15 +00001251or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001252\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001253See os.statvfs for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001254
1255static PyStructSequence_Field statvfs_result_fields[] = {
1256 {"f_bsize", },
1257 {"f_frsize", },
1258 {"f_blocks", },
1259 {"f_bfree", },
1260 {"f_bavail", },
1261 {"f_files", },
1262 {"f_ffree", },
1263 {"f_favail", },
1264 {"f_flag", },
1265 {"f_namemax",},
1266 {0}
1267};
1268
1269static PyStructSequence_Desc statvfs_result_desc = {
1270 "statvfs_result", /* name */
1271 statvfs_result__doc__, /* doc */
1272 statvfs_result_fields,
1273 10
1274};
1275
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001276static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001277static PyTypeObject StatResultType;
1278static PyTypeObject StatVFSResultType;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001279static newfunc structseq_new;
1280
1281static PyObject *
1282statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1283{
1284 PyStructSequence *result;
1285 int i;
1286
1287 result = (PyStructSequence*)structseq_new(type, args, kwds);
1288 if (!result)
1289 return NULL;
1290 /* If we have been initialized from a tuple,
1291 st_?time might be set to None. Initialize it
1292 from the int slots. */
1293 for (i = 7; i <= 9; i++) {
1294 if (result->ob_item[i+3] == Py_None) {
1295 Py_DECREF(Py_None);
1296 Py_INCREF(result->ob_item[i]);
1297 result->ob_item[i+3] = result->ob_item[i];
1298 }
1299 }
1300 return (PyObject*)result;
1301}
1302
1303
1304
1305/* If true, st_?time is float. */
Martin v. Löwisfe33d0b2005-01-16 08:57:39 +00001306static int _stat_float_times = 1;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001307
1308PyDoc_STRVAR(stat_float_times__doc__,
1309"stat_float_times([newval]) -> oldval\n\n\
1310Determine whether os.[lf]stat represents time stamps as float objects.\n\
1311If newval is True, future calls to stat() return floats, if it is False,\n\
1312future calls return ints. \n\
1313If newval is omitted, return the current setting.\n");
1314
1315static PyObject*
1316stat_float_times(PyObject* self, PyObject *args)
1317{
1318 int newval = -1;
1319 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
1320 return NULL;
1321 if (newval == -1)
1322 /* Return old value */
1323 return PyBool_FromLong(_stat_float_times);
1324 _stat_float_times = newval;
1325 Py_INCREF(Py_None);
1326 return Py_None;
1327}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001328
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001329static void
1330fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
1331{
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001332 PyObject *fval,*ival;
1333#if SIZEOF_TIME_T > SIZEOF_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00001334 ival = PyLong_FromLongLong((PY_LONG_LONG)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001335#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001336 ival = PyLong_FromLong((long)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001337#endif
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001338 if (!ival)
1339 return;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001340 if (_stat_float_times) {
1341 fval = PyFloat_FromDouble(sec + 1e-9*nsec);
1342 } else {
1343 fval = ival;
1344 Py_INCREF(fval);
1345 }
1346 PyStructSequence_SET_ITEM(v, index, ival);
1347 PyStructSequence_SET_ITEM(v, index+3, fval);
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001348}
1349
Tim Peters5aa91602002-01-30 05:46:57 +00001350/* pack a system stat C structure into the Python stat tuple
Fred Drake699f3522000-06-29 21:12:41 +00001351 (used by posix_stat() and posix_fstat()) */
1352static PyObject*
Martin v. Löwis14694662006-02-03 12:54:16 +00001353_pystat_fromstructstat(STRUCT_STAT *st)
Fred Drake699f3522000-06-29 21:12:41 +00001354{
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001355 unsigned long ansec, mnsec, cnsec;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001356 PyObject *v = PyStructSequence_New(&StatResultType);
Fred Drake699f3522000-06-29 21:12:41 +00001357 if (v == NULL)
1358 return NULL;
1359
Christian Heimes217cfd12007-12-02 14:31:20 +00001360 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long)st->st_mode));
Fred Drake699f3522000-06-29 21:12:41 +00001361#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +00001362 PyStructSequence_SET_ITEM(v, 1,
Martin v. Löwis14694662006-02-03 12:54:16 +00001363 PyLong_FromLongLong((PY_LONG_LONG)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001364#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001365 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001366#endif
1367#if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
Tim Peters5aa91602002-01-30 05:46:57 +00001368 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwis14694662006-02-03 12:54:16 +00001369 PyLong_FromLongLong((PY_LONG_LONG)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001370#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001371 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001372#endif
Christian Heimes217cfd12007-12-02 14:31:20 +00001373 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long)st->st_nlink));
1374 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long)st->st_uid));
1375 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long)st->st_gid));
Fred Drake699f3522000-06-29 21:12:41 +00001376#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +00001377 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwis14694662006-02-03 12:54:16 +00001378 PyLong_FromLongLong((PY_LONG_LONG)st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001379#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001380 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong(st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001381#endif
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001382
Martin v. Löwis14694662006-02-03 12:54:16 +00001383#if defined(HAVE_STAT_TV_NSEC)
1384 ansec = st->st_atim.tv_nsec;
1385 mnsec = st->st_mtim.tv_nsec;
1386 cnsec = st->st_ctim.tv_nsec;
1387#elif defined(HAVE_STAT_TV_NSEC2)
1388 ansec = st->st_atimespec.tv_nsec;
1389 mnsec = st->st_mtimespec.tv_nsec;
1390 cnsec = st->st_ctimespec.tv_nsec;
1391#elif defined(HAVE_STAT_NSEC)
1392 ansec = st->st_atime_nsec;
1393 mnsec = st->st_mtime_nsec;
1394 cnsec = st->st_ctime_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001395#else
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001396 ansec = mnsec = cnsec = 0;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001397#endif
Martin v. Löwis14694662006-02-03 12:54:16 +00001398 fill_time(v, 7, st->st_atime, ansec);
1399 fill_time(v, 8, st->st_mtime, mnsec);
1400 fill_time(v, 9, st->st_ctime, cnsec);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001401
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001402#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Tim Peters5aa91602002-01-30 05:46:57 +00001403 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001404 PyLong_FromLong((long)st->st_blksize));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001405#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001406#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Tim Peters5aa91602002-01-30 05:46:57 +00001407 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001408 PyLong_FromLong((long)st->st_blocks));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001409#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001410#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001411 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001412 PyLong_FromLong((long)st->st_rdev));
Fred Drake699f3522000-06-29 21:12:41 +00001413#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001414#ifdef HAVE_STRUCT_STAT_ST_GEN
1415 PyStructSequence_SET_ITEM(v, ST_GEN_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001416 PyLong_FromLong((long)st->st_gen));
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001417#endif
1418#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1419 {
1420 PyObject *val;
1421 unsigned long bsec,bnsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001422 bsec = (long)st->st_birthtime;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001423#ifdef HAVE_STAT_TV_NSEC2
Hye-Shik Changd69e0342006-02-19 16:22:22 +00001424 bnsec = st->st_birthtimespec.tv_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001425#else
1426 bnsec = 0;
1427#endif
1428 if (_stat_float_times) {
1429 val = PyFloat_FromDouble(bsec + 1e-9*bnsec);
1430 } else {
Christian Heimes217cfd12007-12-02 14:31:20 +00001431 val = PyLong_FromLong((long)bsec);
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001432 }
1433 PyStructSequence_SET_ITEM(v, ST_BIRTHTIME_IDX,
1434 val);
1435 }
1436#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001437#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1438 PyStructSequence_SET_ITEM(v, ST_FLAGS_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001439 PyLong_FromLong((long)st->st_flags));
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001440#endif
Fred Drake699f3522000-06-29 21:12:41 +00001441
1442 if (PyErr_Occurred()) {
1443 Py_DECREF(v);
1444 return NULL;
1445 }
1446
1447 return v;
1448}
1449
Martin v. Löwisd8948722004-06-02 09:57:56 +00001450#ifdef MS_WINDOWS
1451
1452/* IsUNCRoot -- test whether the supplied path is of the form \\SERVER\SHARE\,
1453 where / can be used in place of \ and the trailing slash is optional.
1454 Both SERVER and SHARE must have at least one character.
1455*/
1456
1457#define ISSLASHA(c) ((c) == '\\' || (c) == '/')
1458#define ISSLASHW(c) ((c) == L'\\' || (c) == L'/')
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001459#ifndef ARRAYSIZE
Martin v. Löwisd8948722004-06-02 09:57:56 +00001460#define ARRAYSIZE(a) (sizeof(a) / sizeof(a[0]))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001461#endif
Martin v. Löwisd8948722004-06-02 09:57:56 +00001462
Tim Peters4ad82172004-08-30 17:02:04 +00001463static BOOL
Martin v. Löwisd8948722004-06-02 09:57:56 +00001464IsUNCRootA(char *path, int pathlen)
1465{
1466 #define ISSLASH ISSLASHA
1467
1468 int i, share;
1469
1470 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1471 /* minimum UNCRoot is \\x\y */
1472 return FALSE;
1473 for (i = 2; i < pathlen ; i++)
1474 if (ISSLASH(path[i])) break;
1475 if (i == 2 || i == pathlen)
1476 /* do not allow \\\SHARE or \\SERVER */
1477 return FALSE;
1478 share = i+1;
1479 for (i = share; i < pathlen; i++)
1480 if (ISSLASH(path[i])) break;
1481 return (i != share && (i == pathlen || i == pathlen-1));
1482
1483 #undef ISSLASH
1484}
1485
Tim Peters4ad82172004-08-30 17:02:04 +00001486static BOOL
Martin v. Löwisd8948722004-06-02 09:57:56 +00001487IsUNCRootW(Py_UNICODE *path, int pathlen)
1488{
1489 #define ISSLASH ISSLASHW
1490
1491 int i, share;
1492
1493 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1494 /* minimum UNCRoot is \\x\y */
1495 return FALSE;
1496 for (i = 2; i < pathlen ; i++)
1497 if (ISSLASH(path[i])) break;
1498 if (i == 2 || i == pathlen)
1499 /* do not allow \\\SHARE or \\SERVER */
1500 return FALSE;
1501 share = i+1;
1502 for (i = share; i < pathlen; i++)
1503 if (ISSLASH(path[i])) break;
1504 return (i != share && (i == pathlen || i == pathlen-1));
1505
1506 #undef ISSLASH
1507}
Martin v. Löwisd8948722004-06-02 09:57:56 +00001508#endif /* MS_WINDOWS */
1509
Barry Warsaw53699e91996-12-10 23:23:01 +00001510static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +00001511posix_do_stat(PyObject *self, PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001512 char *format,
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001513#ifdef __VMS
1514 int (*statfunc)(const char *, STRUCT_STAT *, ...),
1515#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001516 int (*statfunc)(const char *, STRUCT_STAT *),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001517#endif
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001518 char *wformat,
1519 int (*wstatfunc)(const Py_UNICODE *, STRUCT_STAT *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001520{
Fred Drake699f3522000-06-29 21:12:41 +00001521 STRUCT_STAT st;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001522 PyObject *opath;
1523 char *path;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001524 int res;
Martin v. Löwis14694662006-02-03 12:54:16 +00001525 PyObject *result;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001526
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001527#ifdef MS_WINDOWS
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001528 PyUnicodeObject *po;
1529 if (PyArg_ParseTuple(args, wformat, &po)) {
1530 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
Martin v. Löwis14694662006-02-03 12:54:16 +00001531
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001532 Py_BEGIN_ALLOW_THREADS
1533 /* PyUnicode_AS_UNICODE result OK without
1534 thread lock as it is a simple dereference. */
1535 res = wstatfunc(wpath, &st);
1536 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001537
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001538 if (res != 0)
1539 return win32_error_unicode("stat", wpath);
1540 return _pystat_fromstructstat(&st);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001541 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001542 /* Drop the argument parsing error as narrow strings
1543 are also valid. */
1544 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001545#endif
1546
Tim Peters5aa91602002-01-30 05:46:57 +00001547 if (!PyArg_ParseTuple(args, format,
Martin v. Löwis011e8422009-05-05 04:43:17 +00001548 PyUnicode_FSConverter, &opath))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001549 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001550 path = bytes2str(opath, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00001551 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001552 res = (*statfunc)(path, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00001553 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001554
1555 if (res != 0) {
1556#ifdef MS_WINDOWS
Martin v. Löwis011e8422009-05-05 04:43:17 +00001557 result = win32_error("stat", path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001558#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00001559 result = posix_error_with_filename(path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001560#endif
1561 }
1562 else
1563 result = _pystat_fromstructstat(&st);
Fred Drake699f3522000-06-29 21:12:41 +00001564
Martin v. Löwis011e8422009-05-05 04:43:17 +00001565 release_bytes(opath);
Martin v. Löwis14694662006-02-03 12:54:16 +00001566 return result;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001567}
1568
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001569/* POSIX methods */
1570
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001571PyDoc_STRVAR(posix_access__doc__,
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001572"access(path, mode) -> True if granted, False otherwise\n\n\
Guido van Rossuma0b90752002-06-18 16:22:43 +00001573Use the real uid/gid to test for access to a path. Note that most\n\
1574operations will use the effective uid/gid, therefore this routine can\n\
1575be used in a suid/sgid environment to test if the invoking user has the\n\
1576specified access to the path. The mode argument can be F_OK to test\n\
1577existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001578
1579static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001580posix_access(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001581{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001582 PyObject *opath;
Guido van Rossum015f22a1999-01-06 22:52:38 +00001583 char *path;
1584 int mode;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001585
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001586#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001587 DWORD attr;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001588 PyUnicodeObject *po;
1589 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
1590 Py_BEGIN_ALLOW_THREADS
1591 /* PyUnicode_AS_UNICODE OK without thread lock as
1592 it is a simple dereference. */
1593 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1594 Py_END_ALLOW_THREADS
1595 goto finish;
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001596 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001597 /* Drop the argument parsing error as narrow strings
1598 are also valid. */
1599 PyErr_Clear();
Martin v. Löwis011e8422009-05-05 04:43:17 +00001600 if (!PyArg_ParseTuple(args, "O&i:access",
1601 PyUnicode_FSConverter, &opath, &mode))
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001602 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001603 path = bytes2str(opath, 1);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001604 Py_BEGIN_ALLOW_THREADS
1605 attr = GetFileAttributesA(path);
1606 Py_END_ALLOW_THREADS
Martin v. Löwis011e8422009-05-05 04:43:17 +00001607 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001608finish:
1609 if (attr == 0xFFFFFFFF)
1610 /* File does not exist, or cannot read attributes */
1611 return PyBool_FromLong(0);
1612 /* Access is possible if either write access wasn't requested, or
Guido van Rossumb00324f2007-12-04 01:13:14 +00001613 the file isn't read-only, or if it's a directory, as there are
1614 no read-only directories on Windows. */
1615 return PyBool_FromLong(!(mode & 2)
1616 || !(attr & FILE_ATTRIBUTE_READONLY)
1617 || (attr & FILE_ATTRIBUTE_DIRECTORY));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001618#else
1619 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001620 if (!PyArg_ParseTuple(args, "O&i:access",
1621 PyUnicode_FSConverter, &opath, &mode))
Guido van Rossum015f22a1999-01-06 22:52:38 +00001622 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001623 path = bytes2str(opath, 1);
Guido van Rossum015f22a1999-01-06 22:52:38 +00001624 Py_BEGIN_ALLOW_THREADS
1625 res = access(path, mode);
1626 Py_END_ALLOW_THREADS
Martin v. Löwis011e8422009-05-05 04:43:17 +00001627 release_bytes(opath);
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001628 return PyBool_FromLong(res == 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001629#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001630}
1631
Guido van Rossumd371ff11999-01-25 16:12:23 +00001632#ifndef F_OK
1633#define F_OK 0
1634#endif
1635#ifndef R_OK
1636#define R_OK 4
1637#endif
1638#ifndef W_OK
1639#define W_OK 2
1640#endif
1641#ifndef X_OK
1642#define X_OK 1
1643#endif
1644
1645#ifdef HAVE_TTYNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001646PyDoc_STRVAR(posix_ttyname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001647"ttyname(fd) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001648Return the name of the terminal device connected to 'fd'.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001649
1650static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001651posix_ttyname(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001652{
Guido van Rossum94f6f721999-01-06 18:42:14 +00001653 int id;
1654 char *ret;
1655
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001656 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
Guido van Rossum94f6f721999-01-06 18:42:14 +00001657 return NULL;
1658
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001659#if defined(__VMS)
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001660 /* file descriptor 0 only, the default input device (stdin) */
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001661 if (id == 0) {
1662 ret = ttyname();
1663 }
1664 else {
1665 ret = NULL;
1666 }
1667#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00001668 ret = ttyname(id);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001669#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001670 if (ret == NULL)
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001671 return posix_error();
Neal Norwitz93c56822007-08-26 07:10:06 +00001672 return PyUnicode_FromString(ret);
Guido van Rossum94f6f721999-01-06 18:42:14 +00001673}
Guido van Rossumd371ff11999-01-25 16:12:23 +00001674#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001675
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001676#ifdef HAVE_CTERMID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001677PyDoc_STRVAR(posix_ctermid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001678"ctermid() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001679Return the name of the controlling terminal for this process.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001680
1681static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001682posix_ctermid(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001683{
1684 char *ret;
1685 char buffer[L_ctermid];
1686
Greg Wardb48bc172000-03-01 21:51:56 +00001687#ifdef USE_CTERMID_R
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001688 ret = ctermid_r(buffer);
1689#else
1690 ret = ctermid(buffer);
1691#endif
1692 if (ret == NULL)
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001693 return posix_error();
Neal Norwitz93c56822007-08-26 07:10:06 +00001694 return PyUnicode_FromString(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001695}
1696#endif
1697
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001698PyDoc_STRVAR(posix_chdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001699"chdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001700Change the current working directory to the specified path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001701
Barry Warsaw53699e91996-12-10 23:23:01 +00001702static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001703posix_chdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001704{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001705#ifdef MS_WINDOWS
Amaury Forgeot d'Arcc72ef8b2008-10-03 18:38:26 +00001706 return win32_1str(args, "chdir", "y:chdir", win32_chdir, "U:chdir", win32_wchdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001707#elif defined(PYOS_OS2) && defined(PYCC_GCC)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001708 return posix_1str(args, "O&:chdir", _chdir2);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00001709#elif defined(__VMS)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001710 return posix_1str(args, "O&:chdir", (int (*)(const char *))chdir);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001711#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00001712 return posix_1str(args, "O&:chdir", chdir);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001713#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001714}
1715
Fred Drake4d1e64b2002-04-15 19:40:07 +00001716#ifdef HAVE_FCHDIR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001717PyDoc_STRVAR(posix_fchdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001718"fchdir(fildes)\n\n\
Fred Drake4d1e64b2002-04-15 19:40:07 +00001719Change to the directory of the given file descriptor. fildes must be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001720opened on a directory, not a file.");
Fred Drake4d1e64b2002-04-15 19:40:07 +00001721
1722static PyObject *
1723posix_fchdir(PyObject *self, PyObject *fdobj)
1724{
1725 return posix_fildes(fdobj, fchdir);
1726}
1727#endif /* HAVE_FCHDIR */
1728
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001729
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001730PyDoc_STRVAR(posix_chmod__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001731"chmod(path, mode)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001732Change the access permissions of a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001733
Barry Warsaw53699e91996-12-10 23:23:01 +00001734static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001735posix_chmod(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001736{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001737 PyObject *opath = NULL;
Mark Hammondef8b6542001-05-13 08:04:26 +00001738 char *path = NULL;
Guido van Rossumffd15f52000-03-31 00:47:28 +00001739 int i;
1740 int res;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001741#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001742 DWORD attr;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001743 PyUnicodeObject *po;
1744 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
1745 Py_BEGIN_ALLOW_THREADS
1746 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1747 if (attr != 0xFFFFFFFF) {
1748 if (i & _S_IWRITE)
1749 attr &= ~FILE_ATTRIBUTE_READONLY;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001750 else
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001751 attr |= FILE_ATTRIBUTE_READONLY;
1752 res = SetFileAttributesW(PyUnicode_AS_UNICODE(po), attr);
Mark Hammond817c9292003-12-03 01:22:38 +00001753 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001754 else
1755 res = 0;
1756 Py_END_ALLOW_THREADS
1757 if (!res)
1758 return win32_error_unicode("chmod",
1759 PyUnicode_AS_UNICODE(po));
1760 Py_INCREF(Py_None);
1761 return Py_None;
Mark Hammond817c9292003-12-03 01:22:38 +00001762 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001763 /* Drop the argument parsing error as narrow strings
1764 are also valid. */
1765 PyErr_Clear();
1766
Martin v. Löwis011e8422009-05-05 04:43:17 +00001767 if (!PyArg_ParseTuple(args, "O&i:chmod", PyUnicode_FSConverter,
1768 &opath, &i))
Thomas Wouters477c8d52006-05-27 19:21:47 +00001769 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001770 path = bytes2str(opath, 1);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001771 Py_BEGIN_ALLOW_THREADS
1772 attr = GetFileAttributesA(path);
1773 if (attr != 0xFFFFFFFF) {
1774 if (i & _S_IWRITE)
1775 attr &= ~FILE_ATTRIBUTE_READONLY;
1776 else
1777 attr |= FILE_ATTRIBUTE_READONLY;
1778 res = SetFileAttributesA(path, attr);
1779 }
1780 else
1781 res = 0;
1782 Py_END_ALLOW_THREADS
1783 if (!res) {
1784 win32_error("chmod", path);
Martin v. Löwis011e8422009-05-05 04:43:17 +00001785 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001786 return NULL;
1787 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00001788 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001789 Py_INCREF(Py_None);
1790 return Py_None;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001791#else /* MS_WINDOWS */
Martin v. Löwis011e8422009-05-05 04:43:17 +00001792 if (!PyArg_ParseTuple(args, "O&i:chmod", PyUnicode_FSConverter,
1793 &opath, &i))
Guido van Rossumffd15f52000-03-31 00:47:28 +00001794 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001795 path = bytes2str(opath, 1);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001796 Py_BEGIN_ALLOW_THREADS
Guido van Rossumef40e772000-03-31 01:26:23 +00001797 res = chmod(path, i);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001798 Py_END_ALLOW_THREADS
1799 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001800 return posix_error_with_allocated_filename(opath);
1801 release_bytes(opath);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001802 Py_INCREF(Py_None);
1803 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001804#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001805}
1806
Christian Heimes4e30a842007-11-30 22:12:06 +00001807#ifdef HAVE_FCHMOD
1808PyDoc_STRVAR(posix_fchmod__doc__,
1809"fchmod(fd, mode)\n\n\
1810Change the access permissions of the file given by file\n\
1811descriptor fd.");
1812
1813static PyObject *
1814posix_fchmod(PyObject *self, PyObject *args)
1815{
1816 int fd, mode, res;
1817 if (!PyArg_ParseTuple(args, "ii:fchmod", &fd, &mode))
1818 return NULL;
1819 Py_BEGIN_ALLOW_THREADS
1820 res = fchmod(fd, mode);
1821 Py_END_ALLOW_THREADS
1822 if (res < 0)
1823 return posix_error();
1824 Py_RETURN_NONE;
1825}
1826#endif /* HAVE_FCHMOD */
1827
1828#ifdef HAVE_LCHMOD
1829PyDoc_STRVAR(posix_lchmod__doc__,
1830"lchmod(path, mode)\n\n\
1831Change the access permissions of a file. If path is a symlink, this\n\
1832affects the link itself rather than the target.");
1833
1834static PyObject *
1835posix_lchmod(PyObject *self, PyObject *args)
1836{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001837 PyObject *opath;
1838 char *path;
Christian Heimes4e30a842007-11-30 22:12:06 +00001839 int i;
1840 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001841 if (!PyArg_ParseTuple(args, "O&i:lchmod", PyUnicode_FSConverter,
1842 &opath, &i))
Christian Heimes4e30a842007-11-30 22:12:06 +00001843 return NULL;
Eric Smith86a05ec2009-05-05 13:07:30 +00001844 path = bytes2str(opath, 1);
Christian Heimes4e30a842007-11-30 22:12:06 +00001845 Py_BEGIN_ALLOW_THREADS
1846 res = lchmod(path, i);
1847 Py_END_ALLOW_THREADS
1848 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001849 return posix_error_with_allocated_filename(opath);
1850 release_bytes(opath);
Christian Heimes4e30a842007-11-30 22:12:06 +00001851 Py_RETURN_NONE;
1852}
1853#endif /* HAVE_LCHMOD */
1854
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001855
Thomas Wouterscf297e42007-02-23 15:07:44 +00001856#ifdef HAVE_CHFLAGS
1857PyDoc_STRVAR(posix_chflags__doc__,
1858"chflags(path, flags)\n\n\
1859Set file flags.");
1860
1861static PyObject *
1862posix_chflags(PyObject *self, PyObject *args)
1863{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001864 PyObject *opath;
Thomas Wouterscf297e42007-02-23 15:07:44 +00001865 char *path;
1866 unsigned long flags;
1867 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001868 if (!PyArg_ParseTuple(args, "O&k:chflags",
1869 PyUnicode_FSConverter, &opath, &flags))
Thomas Wouterscf297e42007-02-23 15:07:44 +00001870 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001871 path = bytes2str(opath, 1);
Thomas Wouterscf297e42007-02-23 15:07:44 +00001872 Py_BEGIN_ALLOW_THREADS
1873 res = chflags(path, flags);
1874 Py_END_ALLOW_THREADS
1875 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001876 return posix_error_with_allocated_filename(opath);
1877 release_bytes(opath);
Thomas Wouterscf297e42007-02-23 15:07:44 +00001878 Py_INCREF(Py_None);
1879 return Py_None;
1880}
1881#endif /* HAVE_CHFLAGS */
1882
1883#ifdef HAVE_LCHFLAGS
1884PyDoc_STRVAR(posix_lchflags__doc__,
1885"lchflags(path, flags)\n\n\
1886Set file flags.\n\
1887This function will not follow symbolic links.");
1888
1889static PyObject *
1890posix_lchflags(PyObject *self, PyObject *args)
1891{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001892 PyObject *opath;
Thomas Wouterscf297e42007-02-23 15:07:44 +00001893 char *path;
1894 unsigned long flags;
1895 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001896 if (!PyArg_ParseTuple(args, "O&k:lchflags",
Martin v. Löwis4adbc342009-05-05 17:17:55 +00001897 PyUnicode_FSConverter, &opath, &flags))
Thomas Wouterscf297e42007-02-23 15:07:44 +00001898 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001899 path = bytes2str(opath, 1);
Thomas Wouterscf297e42007-02-23 15:07:44 +00001900 Py_BEGIN_ALLOW_THREADS
1901 res = lchflags(path, flags);
1902 Py_END_ALLOW_THREADS
1903 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001904 return posix_error_with_allocated_filename(opath);
1905 release_bytes(opath);
Thomas Wouterscf297e42007-02-23 15:07:44 +00001906 Py_INCREF(Py_None);
1907 return Py_None;
1908}
1909#endif /* HAVE_LCHFLAGS */
1910
Martin v. Löwis244edc82001-10-04 22:44:26 +00001911#ifdef HAVE_CHROOT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001912PyDoc_STRVAR(posix_chroot__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001913"chroot(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001914Change root directory to path.");
Martin v. Löwis244edc82001-10-04 22:44:26 +00001915
1916static PyObject *
1917posix_chroot(PyObject *self, PyObject *args)
1918{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001919 return posix_1str(args, "O&:chroot", chroot);
Martin v. Löwis244edc82001-10-04 22:44:26 +00001920}
1921#endif
1922
Guido van Rossum21142a01999-01-08 21:05:37 +00001923#ifdef HAVE_FSYNC
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001924PyDoc_STRVAR(posix_fsync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001925"fsync(fildes)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001926force write of file with filedescriptor to disk.");
Guido van Rossum21142a01999-01-08 21:05:37 +00001927
1928static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00001929posix_fsync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00001930{
Fred Drake4d1e64b2002-04-15 19:40:07 +00001931 return posix_fildes(fdobj, fsync);
Guido van Rossum21142a01999-01-08 21:05:37 +00001932}
1933#endif /* HAVE_FSYNC */
1934
1935#ifdef HAVE_FDATASYNC
Guido van Rossumecc23b02000-09-22 16:01:05 +00001936
Guido van Rossum7f58e2e2000-09-22 17:26:14 +00001937#ifdef __hpux
Guido van Rossumecc23b02000-09-22 16:01:05 +00001938extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
1939#endif
1940
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001941PyDoc_STRVAR(posix_fdatasync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001942"fdatasync(fildes)\n\n\
Guido van Rossum21142a01999-01-08 21:05:37 +00001943force write of file with filedescriptor to disk.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001944 does not force update of metadata.");
Guido van Rossum21142a01999-01-08 21:05:37 +00001945
1946static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00001947posix_fdatasync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00001948{
Fred Drake4d1e64b2002-04-15 19:40:07 +00001949 return posix_fildes(fdobj, fdatasync);
Guido van Rossum21142a01999-01-08 21:05:37 +00001950}
1951#endif /* HAVE_FDATASYNC */
1952
1953
Fredrik Lundh10723342000-07-10 16:38:09 +00001954#ifdef HAVE_CHOWN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001955PyDoc_STRVAR(posix_chown__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001956"chown(path, uid, gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001957Change the owner and group id of path to the numeric uid and gid.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001958
Barry Warsaw53699e91996-12-10 23:23:01 +00001959static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001960posix_chown(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001961{
Martin v. Löwis011e8422009-05-05 04:43:17 +00001962 PyObject *opath;
1963 char *path;
Christian Heimesd5e2b6f2008-03-19 21:50:51 +00001964 long uid, gid;
Fredrik Lundh44328e62000-07-10 15:59:30 +00001965 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001966 if (!PyArg_ParseTuple(args, "O&ll:chown",
1967 PyUnicode_FSConverter, &opath,
Mark Hammondef8b6542001-05-13 08:04:26 +00001968 &uid, &gid))
Fredrik Lundh44328e62000-07-10 15:59:30 +00001969 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001970 path = bytes2str(opath, 1);
Fredrik Lundh44328e62000-07-10 15:59:30 +00001971 Py_BEGIN_ALLOW_THREADS
1972 res = chown(path, (uid_t) uid, (gid_t) gid);
1973 Py_END_ALLOW_THREADS
1974 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001975 return posix_error_with_allocated_filename(opath);
1976 release_bytes(opath);
Fredrik Lundh44328e62000-07-10 15:59:30 +00001977 Py_INCREF(Py_None);
1978 return Py_None;
Guido van Rossumb6775db1994-08-01 11:34:53 +00001979}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001980#endif /* HAVE_CHOWN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00001981
Christian Heimes4e30a842007-11-30 22:12:06 +00001982#ifdef HAVE_FCHOWN
1983PyDoc_STRVAR(posix_fchown__doc__,
1984"fchown(fd, uid, gid)\n\n\
1985Change the owner and group id of the file given by file descriptor\n\
1986fd to the numeric uid and gid.");
1987
1988static PyObject *
1989posix_fchown(PyObject *self, PyObject *args)
1990{
1991 int fd, uid, gid;
1992 int res;
1993 if (!PyArg_ParseTuple(args, "iii:chown", &fd, &uid, &gid))
1994 return NULL;
1995 Py_BEGIN_ALLOW_THREADS
1996 res = fchown(fd, (uid_t) uid, (gid_t) gid);
1997 Py_END_ALLOW_THREADS
1998 if (res < 0)
1999 return posix_error();
2000 Py_RETURN_NONE;
2001}
2002#endif /* HAVE_FCHOWN */
2003
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002004#ifdef HAVE_LCHOWN
2005PyDoc_STRVAR(posix_lchown__doc__,
2006"lchown(path, uid, gid)\n\n\
2007Change the owner and group id of path to the numeric uid and gid.\n\
2008This function will not follow symbolic links.");
2009
2010static PyObject *
2011posix_lchown(PyObject *self, PyObject *args)
2012{
Martin v. Löwis011e8422009-05-05 04:43:17 +00002013 PyObject *opath;
2014 char *path;
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002015 int uid, gid;
2016 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002017 if (!PyArg_ParseTuple(args, "O&ii:lchown",
2018 PyUnicode_FSConverter, &opath,
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002019 &uid, &gid))
2020 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002021 path = bytes2str(opath, 1);
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002022 Py_BEGIN_ALLOW_THREADS
2023 res = lchown(path, (uid_t) uid, (gid_t) gid);
2024 Py_END_ALLOW_THREADS
Tim Peters11b23062003-04-23 02:39:17 +00002025 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00002026 return posix_error_with_allocated_filename(opath);
2027 release_bytes(opath);
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002028 Py_INCREF(Py_None);
2029 return Py_None;
2030}
2031#endif /* HAVE_LCHOWN */
2032
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002033
Guido van Rossum36bc6801995-06-14 22:54:23 +00002034#ifdef HAVE_GETCWD
Barry Warsaw53699e91996-12-10 23:23:01 +00002035static PyObject *
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002036posix_getcwd(int use_bytes)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002037{
2038 char buf[1026];
2039 char *res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002040
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002041#ifdef MS_WINDOWS
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002042 if (!use_bytes) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002043 wchar_t wbuf[1026];
Thomas Wouters477c8d52006-05-27 19:21:47 +00002044 wchar_t *wbuf2 = wbuf;
2045 PyObject *resobj;
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002046 DWORD len;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002047 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002048 len = GetCurrentDirectoryW(sizeof wbuf/ sizeof wbuf[0], wbuf);
2049 /* If the buffer is large enough, len does not include the
2050 terminating \0. If the buffer is too small, len includes
2051 the space needed for the terminator. */
2052 if (len >= sizeof wbuf/ sizeof wbuf[0]) {
2053 wbuf2 = malloc(len * sizeof(wchar_t));
2054 if (wbuf2)
2055 len = GetCurrentDirectoryW(len, wbuf2);
2056 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002057 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002058 if (!wbuf2) {
2059 PyErr_NoMemory();
2060 return NULL;
2061 }
2062 if (!len) {
2063 if (wbuf2 != wbuf) free(wbuf2);
2064 return win32_error("getcwdu", NULL);
2065 }
2066 resobj = PyUnicode_FromWideChar(wbuf2, len);
2067 if (wbuf2 != wbuf) free(wbuf2);
2068 return resobj;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002069 }
2070#endif
2071
2072 Py_BEGIN_ALLOW_THREADS
2073#if defined(PYOS_OS2) && defined(PYCC_GCC)
2074 res = _getcwd2(buf, sizeof buf);
2075#else
2076 res = getcwd(buf, sizeof buf);
2077#endif
2078 Py_END_ALLOW_THREADS
2079 if (res == NULL)
2080 return posix_error();
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002081 if (use_bytes)
2082 return PyBytes_FromStringAndSize(buf, strlen(buf));
Martin v. Löwis43c57782009-05-10 08:15:24 +00002083 return PyUnicode_Decode(buf, strlen(buf), Py_FileSystemDefaultEncoding,"surrogateescape");
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002084}
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002085
2086PyDoc_STRVAR(posix_getcwd__doc__,
2087"getcwd() -> path\n\n\
2088Return a unicode string representing the current working directory.");
2089
2090static PyObject *
2091posix_getcwd_unicode(PyObject *self)
2092{
2093 return posix_getcwd(0);
2094}
2095
2096PyDoc_STRVAR(posix_getcwdb__doc__,
2097"getcwdb() -> path\n\n\
2098Return a bytes string representing the current working directory.");
2099
2100static PyObject *
2101posix_getcwd_bytes(PyObject *self)
2102{
2103 return posix_getcwd(1);
2104}
Guido van Rossum36bc6801995-06-14 22:54:23 +00002105#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002106
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002107
Guido van Rossumb6775db1994-08-01 11:34:53 +00002108#ifdef HAVE_LINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002109PyDoc_STRVAR(posix_link__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002110"link(src, dst)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002111Create a hard link to a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002112
Barry Warsaw53699e91996-12-10 23:23:01 +00002113static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002114posix_link(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002115{
Martin v. Löwis011e8422009-05-05 04:43:17 +00002116 return posix_2str(args, "O&O&:link", link);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002117}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002118#endif /* HAVE_LINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002119
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002120
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002121PyDoc_STRVAR(posix_listdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002122"listdir(path) -> list_of_strings\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002123Return a list containing the names of the entries in the directory.\n\
2124\n\
2125 path: path of directory to list\n\
2126\n\
2127The list is in arbitrary order. It does not include the special\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002128entries '.' and '..' even if they are present in the directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002129
Barry Warsaw53699e91996-12-10 23:23:01 +00002130static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002131posix_listdir(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002132{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002133 /* XXX Should redo this putting the (now four) versions of opendir
Guido van Rossum6d8841c1997-08-14 19:57:39 +00002134 in separate files instead of having them all here... */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002135#if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002136
Barry Warsaw53699e91996-12-10 23:23:01 +00002137 PyObject *d, *v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002138 HANDLE hFindFile;
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002139 BOOL result;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002140 WIN32_FIND_DATA FileData;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002141 PyObject *opath;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002142 char namebuf[MAX_PATH+5]; /* Overallocate for \\*.*\0 */
Mark Hammondef8b6542001-05-13 08:04:26 +00002143 char *bufptr = namebuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002144 Py_ssize_t len = sizeof(namebuf)-5; /* only claim to have space for MAX_PATH */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002145
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002146 PyObject *po;
2147 if (PyArg_ParseTuple(args, "U:listdir", &po)) {
2148 WIN32_FIND_DATAW wFileData;
2149 Py_UNICODE *wnamebuf;
2150 /* Overallocate for \\*.*\0 */
2151 len = PyUnicode_GET_SIZE(po);
2152 wnamebuf = malloc((len + 5) * sizeof(wchar_t));
2153 if (!wnamebuf) {
2154 PyErr_NoMemory();
2155 return NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002156 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002157 wcscpy(wnamebuf, PyUnicode_AS_UNICODE(po));
2158 if (len > 0) {
2159 Py_UNICODE wch = wnamebuf[len-1];
2160 if (wch != L'/' && wch != L'\\' && wch != L':')
2161 wnamebuf[len++] = L'\\';
2162 wcscpy(wnamebuf + len, L"*.*");
2163 }
2164 if ((d = PyList_New(0)) == NULL) {
2165 free(wnamebuf);
2166 return NULL;
2167 }
2168 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
2169 if (hFindFile == INVALID_HANDLE_VALUE) {
2170 int error = GetLastError();
2171 if (error == ERROR_FILE_NOT_FOUND) {
2172 free(wnamebuf);
2173 return d;
2174 }
2175 Py_DECREF(d);
2176 win32_error_unicode("FindFirstFileW", wnamebuf);
2177 free(wnamebuf);
2178 return NULL;
2179 }
2180 do {
2181 /* Skip over . and .. */
2182 if (wcscmp(wFileData.cFileName, L".") != 0 &&
2183 wcscmp(wFileData.cFileName, L"..") != 0) {
2184 v = PyUnicode_FromUnicode(wFileData.cFileName, wcslen(wFileData.cFileName));
2185 if (v == NULL) {
2186 Py_DECREF(d);
2187 d = NULL;
2188 break;
2189 }
2190 if (PyList_Append(d, v) != 0) {
2191 Py_DECREF(v);
2192 Py_DECREF(d);
2193 d = NULL;
2194 break;
2195 }
2196 Py_DECREF(v);
2197 }
2198 Py_BEGIN_ALLOW_THREADS
2199 result = FindNextFileW(hFindFile, &wFileData);
2200 Py_END_ALLOW_THREADS
2201 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2202 it got to the end of the directory. */
2203 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2204 Py_DECREF(d);
2205 win32_error_unicode("FindNextFileW", wnamebuf);
2206 FindClose(hFindFile);
2207 free(wnamebuf);
2208 return NULL;
2209 }
2210 } while (result == TRUE);
2211
2212 if (FindClose(hFindFile) == FALSE) {
2213 Py_DECREF(d);
2214 win32_error_unicode("FindClose", wnamebuf);
2215 free(wnamebuf);
2216 return NULL;
2217 }
2218 free(wnamebuf);
2219 return d;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002220 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002221 /* Drop the argument parsing error as narrow strings
2222 are also valid. */
2223 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002224
Martin v. Löwis011e8422009-05-05 04:43:17 +00002225 if (!PyArg_ParseTuple(args, "O&:listdir",
2226 PyUnicode_FSConverter, &opath))
Guido van Rossumb6775db1994-08-01 11:34:53 +00002227 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002228 if (PyObject_Size(opath)+1 > MAX_PATH) {
2229 PyErr_SetString(PyExc_ValueError, "path too long");
2230 Py_DECREF(opath);
2231 return NULL;
2232 }
2233 strcpy(namebuf, bytes2str(opath, 0));
2234 len = PyObject_Size(opath);
Neil Schemenauer94b866a2002-03-22 20:51:58 +00002235 if (len > 0) {
2236 char ch = namebuf[len-1];
2237 if (ch != SEP && ch != ALTSEP && ch != ':')
2238 namebuf[len++] = '/';
Hirokazu Yamamotobbb9be72009-05-04 05:56:46 +00002239 strcpy(namebuf + len, "*.*");
Neil Schemenauer94b866a2002-03-22 20:51:58 +00002240 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002241
Barry Warsaw53699e91996-12-10 23:23:01 +00002242 if ((d = PyList_New(0)) == NULL)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002243 return NULL;
2244
2245 hFindFile = FindFirstFile(namebuf, &FileData);
2246 if (hFindFile == INVALID_HANDLE_VALUE) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002247 int error = GetLastError();
2248 if (error == ERROR_FILE_NOT_FOUND)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002249 return d;
2250 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00002251 return win32_error("FindFirstFile", namebuf);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002252 }
2253 do {
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002254 /* Skip over . and .. */
2255 if (strcmp(FileData.cFileName, ".") != 0 &&
2256 strcmp(FileData.cFileName, "..") != 0) {
Christian Heimes72b710a2008-05-26 13:28:38 +00002257 v = PyBytes_FromString(FileData.cFileName);
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002258 if (v == NULL) {
2259 Py_DECREF(d);
2260 d = NULL;
2261 break;
2262 }
2263 if (PyList_Append(d, v) != 0) {
2264 Py_DECREF(v);
2265 Py_DECREF(d);
2266 d = NULL;
2267 break;
2268 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002269 Py_DECREF(v);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002270 }
Georg Brandl622927b2006-03-07 12:48:03 +00002271 Py_BEGIN_ALLOW_THREADS
2272 result = FindNextFile(hFindFile, &FileData);
2273 Py_END_ALLOW_THREADS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002274 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2275 it got to the end of the directory. */
2276 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2277 Py_DECREF(d);
2278 win32_error("FindNextFile", namebuf);
2279 FindClose(hFindFile);
2280 return NULL;
2281 }
Georg Brandl622927b2006-03-07 12:48:03 +00002282 } while (result == TRUE);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002283
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002284 if (FindClose(hFindFile) == FALSE) {
2285 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00002286 return win32_error("FindClose", namebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002287 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002288
2289 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002290
Tim Peters0bb44a42000-09-15 07:44:49 +00002291#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002292
2293#ifndef MAX_PATH
2294#define MAX_PATH CCHMAXPATH
2295#endif
Martin v. Löwis011e8422009-05-05 04:43:17 +00002296 PyObject *oname;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002297 char *name, *pt;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00002298 Py_ssize_t len;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002299 PyObject *d, *v;
2300 char namebuf[MAX_PATH+5];
2301 HDIR hdir = 1;
2302 ULONG srchcnt = 1;
2303 FILEFINDBUF3 ep;
2304 APIRET rc;
2305
Martin v. Löwis011e8422009-05-05 04:43:17 +00002306 if (!PyArg_ParseTuple(args, "O&:listdir",
2307 PyUnicode_FSConverter, &oname))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002308 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002309 name = bytes2str(oname);
2310 len = PyObject_Size(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002311 if (len >= MAX_PATH) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002312 release_bytes(oname);
Neal Norwitz6c913782007-10-14 03:23:09 +00002313 PyErr_SetString(PyExc_ValueError, "path too long");
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002314 return NULL;
2315 }
2316 strcpy(namebuf, name);
2317 for (pt = namebuf; *pt; pt++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002318 if (*pt == ALTSEP)
2319 *pt = SEP;
2320 if (namebuf[len-1] != SEP)
2321 namebuf[len++] = SEP;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002322 strcpy(namebuf + len, "*.*");
2323
Neal Norwitz6c913782007-10-14 03:23:09 +00002324 if ((d = PyList_New(0)) == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002325 release_bytes(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002326 return NULL;
Alexandre Vassalotti4167ebc2007-10-14 02:54:41 +00002327 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002328
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002329 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
2330 &hdir, /* Handle to Use While Search Directory */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002331 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002332 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
2333 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
2334 FIL_STANDARD); /* Format of Entry (EAs or Not) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002335
2336 if (rc != NO_ERROR) {
2337 errno = ENOENT;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002338 return posix_error_with_allocated_filename(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002339 }
2340
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002341 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002342 do {
2343 if (ep.achName[0] == '.'
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002344 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002345 continue; /* Skip Over "." and ".." Names */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002346
2347 strcpy(namebuf, ep.achName);
2348
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002349 /* Leave Case of Name Alone -- In Native Form */
2350 /* (Removed Forced Lowercasing Code) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002351
Christian Heimes72b710a2008-05-26 13:28:38 +00002352 v = PyBytes_FromString(namebuf);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002353 if (v == NULL) {
2354 Py_DECREF(d);
2355 d = NULL;
2356 break;
2357 }
2358 if (PyList_Append(d, v) != 0) {
2359 Py_DECREF(v);
2360 Py_DECREF(d);
2361 d = NULL;
2362 break;
2363 }
2364 Py_DECREF(v);
2365 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
2366 }
2367
Martin v. Löwis011e8422009-05-05 04:43:17 +00002368 release_bytes(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002369 return d;
2370#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002371 PyObject *oname;
2372 char *name;
Barry Warsaw53699e91996-12-10 23:23:01 +00002373 PyObject *d, *v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002374 DIR *dirp;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002375 struct dirent *ep;
Just van Rossum96b1c902003-03-03 17:32:15 +00002376 int arg_is_unicode = 1;
2377
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002378 errno = 0;
Just van Rossum96b1c902003-03-03 17:32:15 +00002379 if (!PyArg_ParseTuple(args, "U:listdir", &v)) {
2380 arg_is_unicode = 0;
2381 PyErr_Clear();
2382 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00002383 if (!PyArg_ParseTuple(args, "O&:listdir", PyUnicode_FSConverter, &oname))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002384 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002385 name = bytes2str(oname, 1);
Guido van Rossumff4949e1992-08-05 19:58:53 +00002386 if ((dirp = opendir(name)) == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002387 return posix_error_with_allocated_filename(oname);
Guido van Rossumff4949e1992-08-05 19:58:53 +00002388 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002389 if ((d = PyList_New(0)) == NULL) {
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002390 closedir(dirp);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002391 release_bytes(oname);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002392 return NULL;
2393 }
Georg Brandl622927b2006-03-07 12:48:03 +00002394 for (;;) {
Georg Brandl3dbca812008-07-23 16:10:53 +00002395 errno = 0;
Georg Brandl622927b2006-03-07 12:48:03 +00002396 Py_BEGIN_ALLOW_THREADS
2397 ep = readdir(dirp);
2398 Py_END_ALLOW_THREADS
Georg Brandl3dbca812008-07-23 16:10:53 +00002399 if (ep == NULL) {
2400 if (errno == 0) {
2401 break;
2402 } else {
2403 closedir(dirp);
2404 Py_DECREF(d);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002405 return posix_error_with_allocated_filename(oname);
Georg Brandl3dbca812008-07-23 16:10:53 +00002406 }
2407 }
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002408 if (ep->d_name[0] == '.' &&
2409 (NAMLEN(ep) == 1 ||
Guido van Rossuma376cc51996-12-05 23:43:35 +00002410 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002411 continue;
Christian Heimes72b710a2008-05-26 13:28:38 +00002412 v = PyBytes_FromStringAndSize(ep->d_name, NAMLEN(ep));
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002413 if (v == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00002414 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002415 d = NULL;
2416 break;
2417 }
Just van Rossum96b1c902003-03-03 17:32:15 +00002418 if (arg_is_unicode) {
Just van Rossum46c97842003-02-25 21:42:15 +00002419 PyObject *w;
2420
2421 w = PyUnicode_FromEncodedObject(v,
Tim Peters11b23062003-04-23 02:39:17 +00002422 Py_FileSystemDefaultEncoding,
Martin v. Löwis43c57782009-05-10 08:15:24 +00002423 "surrogateescape");
Martin v. Löwis011e8422009-05-05 04:43:17 +00002424 Py_DECREF(v);
2425 if (w != NULL)
Just van Rossum6a421832003-03-04 19:30:44 +00002426 v = w;
Just van Rossum6a421832003-03-04 19:30:44 +00002427 else {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002428 /* Encoding failed to decode ASCII bytes.
2429 Raise exception. */
2430 Py_DECREF(d);
2431 d = NULL;
2432 break;
Just van Rossum46c97842003-02-25 21:42:15 +00002433 }
Just van Rossum46c97842003-02-25 21:42:15 +00002434 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002435 if (PyList_Append(d, v) != 0) {
2436 Py_DECREF(v);
2437 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002438 d = NULL;
2439 break;
2440 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002441 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002442 }
2443 closedir(dirp);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002444 release_bytes(oname);
Guido van Rossum0ee42cd1991-04-08 21:01:03 +00002445
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002446 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002447
Tim Peters0bb44a42000-09-15 07:44:49 +00002448#endif /* which OS */
2449} /* end of posix_listdir */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002450
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002451#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00002452/* A helper function for abspath on win32 */
2453static PyObject *
2454posix__getfullpathname(PyObject *self, PyObject *args)
2455{
Martin v. Löwis011e8422009-05-05 04:43:17 +00002456 PyObject *opath;
2457 char *path;
Mark Hammondef8b6542001-05-13 08:04:26 +00002458 char outbuf[MAX_PATH*2];
2459 char *temp;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002460#ifdef MS_WINDOWS
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002461 PyUnicodeObject *po;
2462 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po)) {
2463 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
2464 Py_UNICODE woutbuf[MAX_PATH*2], *woutbufp = woutbuf;
2465 Py_UNICODE *wtemp;
2466 DWORD result;
2467 PyObject *v;
2468 result = GetFullPathNameW(wpath,
2469 sizeof(woutbuf)/sizeof(woutbuf[0]),
2470 woutbuf, &wtemp);
2471 if (result > sizeof(woutbuf)/sizeof(woutbuf[0])) {
2472 woutbufp = malloc(result * sizeof(Py_UNICODE));
2473 if (!woutbufp)
2474 return PyErr_NoMemory();
2475 result = GetFullPathNameW(wpath, result, woutbufp, &wtemp);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002476 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002477 if (result)
2478 v = PyUnicode_FromUnicode(woutbufp, wcslen(woutbufp));
2479 else
2480 v = win32_error_unicode("GetFullPathNameW", wpath);
2481 if (woutbufp != woutbuf)
2482 free(woutbufp);
2483 return v;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002484 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002485 /* Drop the argument parsing error as narrow strings
2486 are also valid. */
2487 PyErr_Clear();
2488
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002489#endif
Martin v. Löwis011e8422009-05-05 04:43:17 +00002490 if (!PyArg_ParseTuple (args, "O&:_getfullpathname",
2491 PyUnicode_FSConverter, &opath))
Mark Hammondef8b6542001-05-13 08:04:26 +00002492 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002493 path = bytes2str(opath, 1);
2494 if (!GetFullPathName(path, sizeof(outbuf)/sizeof(outbuf[0]),
2495 outbuf, &temp)) {
2496 win32_error("GetFullPathName", path);
2497 release_bytes(opath);
2498 return NULL;
2499 }
2500 release_bytes(opath);
Martin v. Löwis969297f2004-06-15 18:49:58 +00002501 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
2502 return PyUnicode_Decode(outbuf, strlen(outbuf),
2503 Py_FileSystemDefaultEncoding, NULL);
2504 }
Christian Heimes72b710a2008-05-26 13:28:38 +00002505 return PyBytes_FromString(outbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00002506} /* end of posix__getfullpathname */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002507#endif /* MS_WINDOWS */
Mark Hammondef8b6542001-05-13 08:04:26 +00002508
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002509PyDoc_STRVAR(posix_mkdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002510"mkdir(path [, mode=0777])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002511Create a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002512
Barry Warsaw53699e91996-12-10 23:23:01 +00002513static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002514posix_mkdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002515{
Guido van Rossumb0824db1996-02-25 04:50:32 +00002516 int res;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002517 PyObject *opath;
2518 char *path;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002519 int mode = 0777;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002520
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002521#ifdef MS_WINDOWS
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002522 PyUnicodeObject *po;
2523 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode)) {
2524 Py_BEGIN_ALLOW_THREADS
2525 /* PyUnicode_AS_UNICODE OK without thread lock as
2526 it is a simple dereference. */
2527 res = CreateDirectoryW(PyUnicode_AS_UNICODE(po), NULL);
2528 Py_END_ALLOW_THREADS
2529 if (!res)
2530 return win32_error_unicode("mkdir", PyUnicode_AS_UNICODE(po));
2531 Py_INCREF(Py_None);
2532 return Py_None;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002533 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002534 /* Drop the argument parsing error as narrow strings
2535 are also valid. */
2536 PyErr_Clear();
Martin v. Löwis011e8422009-05-05 04:43:17 +00002537 if (!PyArg_ParseTuple(args, "O&|i:mkdir",
2538 PyUnicode_FSConverter, &opath, &mode))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002539 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002540 path = bytes2str(opath, 1);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002541 Py_BEGIN_ALLOW_THREADS
2542 /* PyUnicode_AS_UNICODE OK without thread lock as
2543 it is a simple dereference. */
2544 res = CreateDirectoryA(path, NULL);
2545 Py_END_ALLOW_THREADS
2546 if (!res) {
2547 win32_error("mkdir", path);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002548 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002549 return NULL;
2550 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00002551 release_bytes(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002552 Py_INCREF(Py_None);
2553 return Py_None;
2554#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002555
Martin v. Löwis011e8422009-05-05 04:43:17 +00002556 if (!PyArg_ParseTuple(args, "O&|i:mkdir",
2557 PyUnicode_FSConverter, &opath, &mode))
Guido van Rossumb0824db1996-02-25 04:50:32 +00002558 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002559 path = bytes2str(opath, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00002560 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002561#if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002562 res = mkdir(path);
2563#else
Guido van Rossumb0824db1996-02-25 04:50:32 +00002564 res = mkdir(path, mode);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002565#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002566 Py_END_ALLOW_THREADS
Guido van Rossumb0824db1996-02-25 04:50:32 +00002567 if (res < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00002568 return posix_error_with_allocated_filename(opath);
2569 release_bytes(opath);
Barry Warsaw53699e91996-12-10 23:23:01 +00002570 Py_INCREF(Py_None);
2571 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002572#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002573}
2574
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002575
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002576/* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
2577#if defined(HAVE_SYS_RESOURCE_H)
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002578#include <sys/resource.h>
2579#endif
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002580
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002581
2582#ifdef HAVE_NICE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002583PyDoc_STRVAR(posix_nice__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002584"nice(inc) -> new_priority\n\n\
2585Decrease the priority of process by inc and return the new priority.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002586
Barry Warsaw53699e91996-12-10 23:23:01 +00002587static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002588posix_nice(PyObject *self, PyObject *args)
Guido van Rossum775f4da1993-01-09 17:18:52 +00002589{
2590 int increment, value;
2591
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002592 if (!PyArg_ParseTuple(args, "i:nice", &increment))
Guido van Rossum775f4da1993-01-09 17:18:52 +00002593 return NULL;
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002594
2595 /* There are two flavours of 'nice': one that returns the new
2596 priority (as required by almost all standards out there) and the
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002597 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
2598 the use of getpriority() to get the new priority.
Tim Peters5aa91602002-01-30 05:46:57 +00002599
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002600 If we are of the nice family that returns the new priority, we
2601 need to clear errno before the call, and check if errno is filled
2602 before calling posix_error() on a returnvalue of -1, because the
2603 -1 may be the actual new priority! */
2604
2605 errno = 0;
Guido van Rossum775f4da1993-01-09 17:18:52 +00002606 value = nice(increment);
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002607#if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002608 if (value == 0)
2609 value = getpriority(PRIO_PROCESS, 0);
2610#endif
2611 if (value == -1 && errno != 0)
2612 /* either nice() or getpriority() returned an error */
Guido van Rossum775f4da1993-01-09 17:18:52 +00002613 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00002614 return PyLong_FromLong((long) value);
Guido van Rossum775f4da1993-01-09 17:18:52 +00002615}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002616#endif /* HAVE_NICE */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002617
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002618PyDoc_STRVAR(posix_rename__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002619"rename(old, new)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002620Rename a file or directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002621
Barry Warsaw53699e91996-12-10 23:23:01 +00002622static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002623posix_rename(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002624{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002625#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002626 PyObject *o1, *o2;
2627 char *p1, *p2;
2628 BOOL result;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002629 if (!PyArg_ParseTuple(args, "OO:rename", &o1, &o2))
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +00002630 goto error;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002631 if (!convert_to_unicode(&o1))
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +00002632 goto error;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002633 if (!convert_to_unicode(&o2)) {
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +00002634 Py_DECREF(o1);
2635 goto error;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002636 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002637 Py_BEGIN_ALLOW_THREADS
2638 result = MoveFileW(PyUnicode_AsUnicode(o1),
2639 PyUnicode_AsUnicode(o2));
2640 Py_END_ALLOW_THREADS
2641 Py_DECREF(o1);
2642 Py_DECREF(o2);
2643 if (!result)
2644 return win32_error("rename", NULL);
2645 Py_INCREF(Py_None);
2646 return Py_None;
2647error:
2648 PyErr_Clear();
Thomas Wouters477c8d52006-05-27 19:21:47 +00002649 if (!PyArg_ParseTuple(args, "ss:rename", &p1, &p2))
2650 return NULL;
2651 Py_BEGIN_ALLOW_THREADS
2652 result = MoveFileA(p1, p2);
2653 Py_END_ALLOW_THREADS
2654 if (!result)
2655 return win32_error("rename", NULL);
2656 Py_INCREF(Py_None);
2657 return Py_None;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002658#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002659 return posix_2str(args, "O&O&:rename", rename);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002660#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002661}
2662
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002663
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002664PyDoc_STRVAR(posix_rmdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002665"rmdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002666Remove a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002667
Barry Warsaw53699e91996-12-10 23:23:01 +00002668static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002669posix_rmdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002670{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002671#ifdef MS_WINDOWS
Amaury Forgeot d'Arcc72ef8b2008-10-03 18:38:26 +00002672 return win32_1str(args, "rmdir", "y:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002673#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002674 return posix_1str(args, "O&:rmdir", rmdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002675#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002676}
2677
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002678
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002679PyDoc_STRVAR(posix_stat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002680"stat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002681Perform a stat system call on the given path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002682
Barry Warsaw53699e91996-12-10 23:23:01 +00002683static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002684posix_stat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002685{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002686#ifdef MS_WINDOWS
Martin v. Löwis011e8422009-05-05 04:43:17 +00002687 return posix_do_stat(self, args, "O&:stat", STAT, "U:stat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002688#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002689 return posix_do_stat(self, args, "O&:stat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002690#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002691}
2692
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002693
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002694#ifdef HAVE_SYSTEM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002695PyDoc_STRVAR(posix_system__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002696"system(command) -> exit_status\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002697Execute the command (a string) in a subshell.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002698
Barry Warsaw53699e91996-12-10 23:23:01 +00002699static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002700posix_system(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002701{
Guido van Rossumff4949e1992-08-05 19:58:53 +00002702 long sts;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002703#ifdef MS_WINDOWS
2704 wchar_t *command;
2705 if (!PyArg_ParseTuple(args, "u:system", &command))
2706 return NULL;
2707#else
2708 char *command;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002709 if (!PyArg_ParseTuple(args, "s:system", &command))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002710 return NULL;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002711#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002712 Py_BEGIN_ALLOW_THREADS
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002713#ifdef MS_WINDOWS
2714 sts = _wsystem(command);
2715#else
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002716 sts = system(command);
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002717#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002718 Py_END_ALLOW_THREADS
Christian Heimes217cfd12007-12-02 14:31:20 +00002719 return PyLong_FromLong(sts);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002720}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002721#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002722
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002723
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002724PyDoc_STRVAR(posix_umask__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002725"umask(new_mask) -> old_mask\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002726Set the current numeric umask and return the previous umask.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002727
Barry Warsaw53699e91996-12-10 23:23:01 +00002728static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002729posix_umask(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002730{
2731 int i;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002732 if (!PyArg_ParseTuple(args, "i:umask", &i))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002733 return NULL;
Fred Drake0368bc42001-07-19 20:48:32 +00002734 i = (int)umask(i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002735 if (i < 0)
2736 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00002737 return PyLong_FromLong((long)i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002738}
2739
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002740
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002741PyDoc_STRVAR(posix_unlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002742"unlink(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002743Remove a file (same as remove(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002744
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002745PyDoc_STRVAR(posix_remove__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002746"remove(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002747Remove a file (same as unlink(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002748
Barry Warsaw53699e91996-12-10 23:23:01 +00002749static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002750posix_unlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002751{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002752#ifdef MS_WINDOWS
Amaury Forgeot d'Arcc72ef8b2008-10-03 18:38:26 +00002753 return win32_1str(args, "remove", "y:remove", DeleteFileA, "U:remove", DeleteFileW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002754#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00002755 return posix_1str(args, "O&:remove", unlink);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002756#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002757}
2758
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002759
Guido van Rossumb6775db1994-08-01 11:34:53 +00002760#ifdef HAVE_UNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002761PyDoc_STRVAR(posix_uname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002762"uname() -> (sysname, nodename, release, version, machine)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002763Return a tuple identifying the current operating system.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002764
Barry Warsaw53699e91996-12-10 23:23:01 +00002765static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002766posix_uname(PyObject *self, PyObject *noargs)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002767{
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002768 struct utsname u;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002769 int res;
Neal Norwitze241ce82003-02-17 18:17:05 +00002770
Barry Warsaw53699e91996-12-10 23:23:01 +00002771 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002772 res = uname(&u);
Barry Warsaw53699e91996-12-10 23:23:01 +00002773 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002774 if (res < 0)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002775 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00002776 return Py_BuildValue("(sssss)",
Barry Warsaw43d68b81996-12-19 22:10:44 +00002777 u.sysname,
2778 u.nodename,
2779 u.release,
2780 u.version,
2781 u.machine);
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002782}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002783#endif /* HAVE_UNAME */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002784
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002785static int
2786extract_time(PyObject *t, long* sec, long* usec)
2787{
2788 long intval;
2789 if (PyFloat_Check(t)) {
2790 double tval = PyFloat_AsDouble(t);
Christian Heimes90aa7642007-12-19 02:45:37 +00002791 PyObject *intobj = Py_TYPE(t)->tp_as_number->nb_int(t);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002792 if (!intobj)
2793 return -1;
Christian Heimes217cfd12007-12-02 14:31:20 +00002794 intval = PyLong_AsLong(intobj);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002795 Py_DECREF(intobj);
Michael W. Hudsonb8963812005-07-05 15:21:58 +00002796 if (intval == -1 && PyErr_Occurred())
2797 return -1;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002798 *sec = intval;
Tim Peters96940cf2002-09-10 15:37:28 +00002799 *usec = (long)((tval - intval) * 1e6); /* can't exceed 1000000 */
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002800 if (*usec < 0)
2801 /* If rounding gave us a negative number,
2802 truncate. */
2803 *usec = 0;
2804 return 0;
2805 }
Christian Heimes217cfd12007-12-02 14:31:20 +00002806 intval = PyLong_AsLong(t);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002807 if (intval == -1 && PyErr_Occurred())
2808 return -1;
2809 *sec = intval;
2810 *usec = 0;
Martin v. Löwis076b2092002-09-10 15:04:41 +00002811 return 0;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002812}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002813
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002814PyDoc_STRVAR(posix_utime__doc__,
Thomas Wouters477c8d52006-05-27 19:21:47 +00002815"utime(path, (atime, mtime))\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00002816utime(path, None)\n\n\
Barry Warsaw3cef8562000-05-01 16:17:24 +00002817Set the access and modified time of the file to the given values. If the\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002818second form is used, set the access and modified times to the current time.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002819
Barry Warsaw53699e91996-12-10 23:23:01 +00002820static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002821posix_utime(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002822{
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002823#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002824 PyObject *arg;
2825 PyUnicodeObject *obwpath;
2826 wchar_t *wpath = NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002827 PyObject *oapath;
2828 char *apath;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002829 HANDLE hFile;
2830 long atimesec, mtimesec, ausec, musec;
2831 FILETIME atime, mtime;
2832 PyObject *result = NULL;
2833
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002834 if (PyArg_ParseTuple(args, "UO|:utime", &obwpath, &arg)) {
2835 wpath = PyUnicode_AS_UNICODE(obwpath);
2836 Py_BEGIN_ALLOW_THREADS
2837 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
2838 NULL, OPEN_EXISTING,
2839 FILE_FLAG_BACKUP_SEMANTICS, NULL);
2840 Py_END_ALLOW_THREADS
2841 if (hFile == INVALID_HANDLE_VALUE)
2842 return win32_error_unicode("utime", wpath);
2843 } else
2844 /* Drop the argument parsing error as narrow strings
2845 are also valid. */
2846 PyErr_Clear();
2847
Thomas Wouters477c8d52006-05-27 19:21:47 +00002848 if (!wpath) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002849 if (!PyArg_ParseTuple(args, "O&O:utime",
2850 PyUnicode_FSConverter, &oapath, &arg))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002851 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002852 apath = bytes2str(oapath, 1);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002853 Py_BEGIN_ALLOW_THREADS
2854 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002855 NULL, OPEN_EXISTING,
2856 FILE_FLAG_BACKUP_SEMANTICS, NULL);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002857 Py_END_ALLOW_THREADS
2858 if (hFile == INVALID_HANDLE_VALUE) {
2859 win32_error("utime", apath);
Martin v. Löwis011e8422009-05-05 04:43:17 +00002860 release_bytes(oapath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002861 return NULL;
2862 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00002863 release_bytes(oapath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002864 }
2865
2866 if (arg == Py_None) {
2867 SYSTEMTIME now;
2868 GetSystemTime(&now);
2869 if (!SystemTimeToFileTime(&now, &mtime) ||
2870 !SystemTimeToFileTime(&now, &atime)) {
2871 win32_error("utime", NULL);
2872 goto done;
2873 }
2874 }
2875 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
2876 PyErr_SetString(PyExc_TypeError,
2877 "utime() arg 2 must be a tuple (atime, mtime)");
2878 goto done;
2879 }
2880 else {
2881 if (extract_time(PyTuple_GET_ITEM(arg, 0),
2882 &atimesec, &ausec) == -1)
2883 goto done;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002884 time_t_to_FILE_TIME(atimesec, 1000*ausec, &atime);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002885 if (extract_time(PyTuple_GET_ITEM(arg, 1),
2886 &mtimesec, &musec) == -1)
2887 goto done;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002888 time_t_to_FILE_TIME(mtimesec, 1000*musec, &mtime);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002889 }
2890 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
2891 /* Avoid putting the file name into the error here,
2892 as that may confuse the user into believing that
2893 something is wrong with the file, when it also
2894 could be the time stamp that gives a problem. */
2895 win32_error("utime", NULL);
2896 }
2897 Py_INCREF(Py_None);
2898 result = Py_None;
2899done:
2900 CloseHandle(hFile);
2901 return result;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002902#else /* MS_WINDOWS */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002903
Martin v. Löwis011e8422009-05-05 04:43:17 +00002904 PyObject *opath;
2905 char *path;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002906 long atime, mtime, ausec, musec;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002907 int res;
Barry Warsaw3cef8562000-05-01 16:17:24 +00002908 PyObject* arg;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002909
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002910#if defined(HAVE_UTIMES)
2911 struct timeval buf[2];
2912#define ATIME buf[0].tv_sec
2913#define MTIME buf[1].tv_sec
2914#elif defined(HAVE_UTIME_H)
Guido van Rossum6d8841c1997-08-14 19:57:39 +00002915/* XXX should define struct utimbuf instead, above */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002916 struct utimbuf buf;
2917#define ATIME buf.actime
2918#define MTIME buf.modtime
2919#define UTIME_ARG &buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002920#else /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002921 time_t buf[2];
2922#define ATIME buf[0]
2923#define MTIME buf[1]
2924#define UTIME_ARG buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002925#endif /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002926
Mark Hammond817c9292003-12-03 01:22:38 +00002927
Martin v. Löwis011e8422009-05-05 04:43:17 +00002928 if (!PyArg_ParseTuple(args, "O&O:utime",
2929 PyUnicode_FSConverter, &opath, &arg))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002930 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002931 path = bytes2str(opath, 1);
Barry Warsaw3cef8562000-05-01 16:17:24 +00002932 if (arg == Py_None) {
2933 /* optional time values not given */
2934 Py_BEGIN_ALLOW_THREADS
2935 res = utime(path, NULL);
2936 Py_END_ALLOW_THREADS
2937 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002938 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
Barry Warsaw3cef8562000-05-01 16:17:24 +00002939 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00002940 "utime() arg 2 must be a tuple (atime, mtime)");
Martin v. Löwis011e8422009-05-05 04:43:17 +00002941 release_bytes(opath);
Barry Warsaw3cef8562000-05-01 16:17:24 +00002942 return NULL;
2943 }
2944 else {
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002945 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Neal Norwitz96652712004-06-06 20:40:27 +00002946 &atime, &ausec) == -1) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002947 release_bytes(opath);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002948 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00002949 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002950 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Neal Norwitz96652712004-06-06 20:40:27 +00002951 &mtime, &musec) == -1) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002952 release_bytes(opath);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002953 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00002954 }
Barry Warsaw3cef8562000-05-01 16:17:24 +00002955 ATIME = atime;
2956 MTIME = mtime;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002957#ifdef HAVE_UTIMES
2958 buf[0].tv_usec = ausec;
2959 buf[1].tv_usec = musec;
2960 Py_BEGIN_ALLOW_THREADS
2961 res = utimes(path, buf);
2962 Py_END_ALLOW_THREADS
2963#else
Barry Warsaw3cef8562000-05-01 16:17:24 +00002964 Py_BEGIN_ALLOW_THREADS
2965 res = utime(path, UTIME_ARG);
2966 Py_END_ALLOW_THREADS
Mark Hammond817c9292003-12-03 01:22:38 +00002967#endif /* HAVE_UTIMES */
Barry Warsaw3cef8562000-05-01 16:17:24 +00002968 }
Mark Hammond2d5914b2004-05-04 08:10:37 +00002969 if (res < 0) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00002970 return posix_error_with_allocated_filename(opath);
Mark Hammond2d5914b2004-05-04 08:10:37 +00002971 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00002972 release_bytes(opath);
Barry Warsaw53699e91996-12-10 23:23:01 +00002973 Py_INCREF(Py_None);
2974 return Py_None;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002975#undef UTIME_ARG
2976#undef ATIME
2977#undef MTIME
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002978#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002979}
2980
Guido van Rossum85e3b011991-06-03 12:42:10 +00002981
Guido van Rossum3b066191991-06-04 19:40:25 +00002982/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002983
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002984PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002985"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002986Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002987
Barry Warsaw53699e91996-12-10 23:23:01 +00002988static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002989posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002990{
2991 int sts;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002992 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
Guido van Rossum85e3b011991-06-03 12:42:10 +00002993 return NULL;
2994 _exit(sts);
Guido van Rossuma376cc51996-12-05 23:43:35 +00002995 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002996}
2997
Martin v. Löwis114619e2002-10-07 06:44:21 +00002998#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
2999static void
Martin v. Löwis725507b2006-03-07 12:08:51 +00003000free_string_array(char **array, Py_ssize_t count)
Martin v. Löwis114619e2002-10-07 06:44:21 +00003001{
Martin v. Löwis725507b2006-03-07 12:08:51 +00003002 Py_ssize_t i;
Martin v. Löwis114619e2002-10-07 06:44:21 +00003003 for (i = 0; i < count; i++)
3004 PyMem_Free(array[i]);
3005 PyMem_DEL(array);
3006}
Martin v. Löwis011e8422009-05-05 04:43:17 +00003007
Antoine Pitrou69f71142009-05-24 21:25:49 +00003008static
Martin v. Löwis011e8422009-05-05 04:43:17 +00003009int fsconvert_strdup(PyObject *o, char**out)
3010{
3011 PyObject *bytes;
3012 Py_ssize_t size;
3013 if (!PyUnicode_FSConverter(o, &bytes))
3014 return 0;
3015 size = PyObject_Size(bytes);
3016 *out = PyMem_Malloc(size+1);
3017 if (!*out)
3018 return 0;
3019 /* Don't lock bytes, as we hold the GIL */
3020 memcpy(*out, bytes2str(bytes, 0), size+1);
3021 Py_DECREF(bytes);
3022 return 1;
3023}
Martin v. Löwis114619e2002-10-07 06:44:21 +00003024#endif
3025
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003026
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003027#ifdef HAVE_EXECV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003028PyDoc_STRVAR(posix_execv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003029"execv(path, args)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003030Execute an executable path with arguments, replacing current process.\n\
3031\n\
3032 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003033 args: tuple or list of strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003034
Barry Warsaw53699e91996-12-10 23:23:01 +00003035static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003036posix_execv(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003037{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003038 PyObject *opath;
Guido van Rossumef0a00e1992-01-27 16:51:30 +00003039 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00003040 PyObject *argv;
Guido van Rossum85e3b011991-06-03 12:42:10 +00003041 char **argvlist;
Martin v. Löwis725507b2006-03-07 12:08:51 +00003042 Py_ssize_t i, argc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003043 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003044
Guido van Rossum89b33251993-10-22 14:26:06 +00003045 /* execv has two arguments: (path, argv), where
Guido van Rossum85e3b011991-06-03 12:42:10 +00003046 argv is a list or tuple of strings. */
3047
Martin v. Löwis011e8422009-05-05 04:43:17 +00003048 if (!PyArg_ParseTuple(args, "O&O:execv",
3049 PyUnicode_FSConverter,
3050 &opath, &argv))
Guido van Rossum85e3b011991-06-03 12:42:10 +00003051 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003052 path = bytes2str(opath, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00003053 if (PyList_Check(argv)) {
3054 argc = PyList_Size(argv);
3055 getitem = PyList_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00003056 }
Barry Warsaw53699e91996-12-10 23:23:01 +00003057 else if (PyTuple_Check(argv)) {
3058 argc = PyTuple_Size(argv);
3059 getitem = PyTuple_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00003060 }
Guido van Rossumef0a00e1992-01-27 16:51:30 +00003061 else {
Fred Drake661ea262000-10-24 19:57:45 +00003062 PyErr_SetString(PyExc_TypeError, "execv() arg 2 must be a tuple or list");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003063 release_bytes(opath);
Guido van Rossum50422b42000-04-26 20:34:28 +00003064 return NULL;
3065 }
Thomas Heller6790d602007-08-30 17:15:14 +00003066 if (argc < 1) {
3067 PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003068 release_bytes(opath);
Thomas Heller6790d602007-08-30 17:15:14 +00003069 return NULL;
3070 }
Guido van Rossum50422b42000-04-26 20:34:28 +00003071
Barry Warsaw53699e91996-12-10 23:23:01 +00003072 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003073 if (argvlist == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003074 release_bytes(opath);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00003075 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003076 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00003077 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003078 if (!fsconvert_strdup((*getitem)(argv, i),
3079 &argvlist[i])) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003080 free_string_array(argvlist, i);
Tim Peters5aa91602002-01-30 05:46:57 +00003081 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003082 "execv() arg 2 must contain only strings");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003083 release_bytes(opath);
Guido van Rossum50422b42000-04-26 20:34:28 +00003084 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00003085
Guido van Rossum85e3b011991-06-03 12:42:10 +00003086 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00003087 }
3088 argvlist[argc] = NULL;
3089
Guido van Rossumef0a00e1992-01-27 16:51:30 +00003090 execv(path, argvlist);
Guido van Rossumb6775db1994-08-01 11:34:53 +00003091
Guido van Rossum85e3b011991-06-03 12:42:10 +00003092 /* If we get here it's definitely an error */
3093
Martin v. Löwis114619e2002-10-07 06:44:21 +00003094 free_string_array(argvlist, argc);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003095 release_bytes(opath);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003096 return posix_error();
3097}
3098
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003099
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003100PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003101"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003102Execute a path with arguments and environment, replacing current process.\n\
3103\n\
3104 path: path of executable file\n\
3105 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003106 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003107
Barry Warsaw53699e91996-12-10 23:23:01 +00003108static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003109posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003110{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003111 PyObject *opath;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003112 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00003113 PyObject *argv, *env;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003114 char **argvlist;
3115 char **envlist;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003116 PyObject *key, *val, *keys=NULL, *vals=NULL;
Martin v. Löwis725507b2006-03-07 12:08:51 +00003117 Py_ssize_t i, pos, argc, envc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003118 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Thomas Wouters477c8d52006-05-27 19:21:47 +00003119 Py_ssize_t lastarg = 0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003120
3121 /* execve has three arguments: (path, argv, env), where
3122 argv is a list or tuple of strings and env is a dictionary
3123 like posix.environ. */
3124
Martin v. Löwis011e8422009-05-05 04:43:17 +00003125 if (!PyArg_ParseTuple(args, "O&OO:execve",
3126 PyUnicode_FSConverter,
3127 &opath, &argv, &env))
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003128 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003129 path = bytes2str(opath, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00003130 if (PyList_Check(argv)) {
3131 argc = PyList_Size(argv);
3132 getitem = PyList_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003133 }
Barry Warsaw53699e91996-12-10 23:23:01 +00003134 else if (PyTuple_Check(argv)) {
3135 argc = PyTuple_Size(argv);
3136 getitem = PyTuple_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003137 }
3138 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003139 PyErr_SetString(PyExc_TypeError,
3140 "execve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003141 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003142 }
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003143 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003144 PyErr_SetString(PyExc_TypeError,
3145 "execve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003146 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003147 }
3148
Barry Warsaw53699e91996-12-10 23:23:01 +00003149 argvlist = PyMem_NEW(char *, argc+1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003150 if (argvlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003151 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003152 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003153 }
3154 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003155 if (!fsconvert_strdup((*getitem)(argv, i),
3156 &argvlist[i]))
Barry Warsaw43d68b81996-12-19 22:10:44 +00003157 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003158 lastarg = i;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003159 goto fail_1;
3160 }
3161 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003162 lastarg = argc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003163 argvlist[argc] = NULL;
3164
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003165 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003166 if (i < 0)
3167 goto fail_1;
Barry Warsaw53699e91996-12-10 23:23:01 +00003168 envlist = PyMem_NEW(char *, i + 1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003169 if (envlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003170 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003171 goto fail_1;
3172 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003173 envc = 0;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003174 keys = PyMapping_Keys(env);
3175 vals = PyMapping_Values(env);
3176 if (!keys || !vals)
3177 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003178 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3179 PyErr_SetString(PyExc_TypeError,
3180 "execve(): env.keys() or env.values() is not a list");
3181 goto fail_2;
3182 }
Tim Peters5aa91602002-01-30 05:46:57 +00003183
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003184 for (pos = 0; pos < i; pos++) {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003185 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003186 size_t len;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003187
3188 key = PyList_GetItem(keys, pos);
3189 val = PyList_GetItem(vals, pos);
3190 if (!key || !val)
3191 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003192
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003193 if (!PyArg_Parse(
3194 key,
3195 "s;execve() arg 3 contains a non-string key",
3196 &k) ||
3197 !PyArg_Parse(
3198 val,
3199 "s;execve() arg 3 contains a non-string value",
3200 &v))
Barry Warsaw43d68b81996-12-19 22:10:44 +00003201 {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003202 goto fail_2;
3203 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00003204
3205#if defined(PYOS_OS2)
3206 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
3207 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
3208#endif
Christian Heimes830a4bc2007-11-22 07:43:40 +00003209 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Tim Petersc8996f52001-12-03 20:41:00 +00003210 p = PyMem_NEW(char, len);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003211 if (p == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003212 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003213 goto fail_2;
3214 }
Tim Petersc8996f52001-12-03 20:41:00 +00003215 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003216 envlist[envc++] = p;
Guido van Rossumd48f2521997-12-05 22:19:34 +00003217#if defined(PYOS_OS2)
3218 }
3219#endif
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003220 }
3221 envlist[envc] = 0;
3222
3223 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00003224
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003225 /* If we get here it's definitely an error */
3226
3227 (void) posix_error();
3228
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003229 fail_2:
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003230 while (--envc >= 0)
Barry Warsaw53699e91996-12-10 23:23:01 +00003231 PyMem_DEL(envlist[envc]);
3232 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003233 fail_1:
3234 free_string_array(argvlist, lastarg);
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003235 Py_XDECREF(vals);
3236 Py_XDECREF(keys);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003237 fail_0:
Martin v. Löwis011e8422009-05-05 04:43:17 +00003238 release_bytes(opath);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003239 return NULL;
3240}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003241#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003242
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003243
Guido van Rossuma1065681999-01-25 23:20:23 +00003244#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003245PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003246"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003247Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003248\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003249 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003250 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003251 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003252
3253static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003254posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003255{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003256 PyObject *opath;
Guido van Rossuma1065681999-01-25 23:20:23 +00003257 char *path;
3258 PyObject *argv;
3259 char **argvlist;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003260 int mode, i;
3261 Py_ssize_t argc;
Tim Peters79248aa2001-08-29 21:37:10 +00003262 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003263 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossuma1065681999-01-25 23:20:23 +00003264
3265 /* spawnv has three arguments: (mode, path, argv), where
3266 argv is a list or tuple of strings. */
3267
Martin v. Löwis011e8422009-05-05 04:43:17 +00003268 if (!PyArg_ParseTuple(args, "iO&O:spawnv", &mode,
3269 PyUnicode_FSConverter,
3270 &opath, &argv))
Guido van Rossuma1065681999-01-25 23:20:23 +00003271 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003272 path = bytes2str(opath, 1);
Guido van Rossuma1065681999-01-25 23:20:23 +00003273 if (PyList_Check(argv)) {
3274 argc = PyList_Size(argv);
3275 getitem = PyList_GetItem;
3276 }
3277 else if (PyTuple_Check(argv)) {
3278 argc = PyTuple_Size(argv);
3279 getitem = PyTuple_GetItem;
3280 }
3281 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003282 PyErr_SetString(PyExc_TypeError,
3283 "spawnv() arg 2 must be a tuple or list");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003284 release_bytes(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00003285 return NULL;
3286 }
3287
3288 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003289 if (argvlist == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003290 release_bytes(opath);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00003291 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003292 }
Guido van Rossuma1065681999-01-25 23:20:23 +00003293 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003294 if (!fsconvert_strdup((*getitem)(argv, i),
3295 &argvlist[i])) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003296 free_string_array(argvlist, i);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003297 PyErr_SetString(
3298 PyExc_TypeError,
3299 "spawnv() arg 2 must contain only strings");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003300 release_bytes(opath);
Fred Drake137507e2000-06-01 02:02:46 +00003301 return NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00003302 }
3303 }
3304 argvlist[argc] = NULL;
3305
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003306#if defined(PYOS_OS2) && defined(PYCC_GCC)
3307 Py_BEGIN_ALLOW_THREADS
3308 spawnval = spawnv(mode, path, argvlist);
3309 Py_END_ALLOW_THREADS
3310#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003311 if (mode == _OLD_P_OVERLAY)
3312 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00003313
Tim Peters25059d32001-12-07 20:35:43 +00003314 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003315 spawnval = _spawnv(mode, path, argvlist);
Tim Peters25059d32001-12-07 20:35:43 +00003316 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003317#endif
Tim Peters5aa91602002-01-30 05:46:57 +00003318
Martin v. Löwis114619e2002-10-07 06:44:21 +00003319 free_string_array(argvlist, argc);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003320 release_bytes(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00003321
Fred Drake699f3522000-06-29 21:12:41 +00003322 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003323 return posix_error();
3324 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003325#if SIZEOF_LONG == SIZEOF_VOID_P
3326 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003327#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003328 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003329#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003330}
3331
3332
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003333PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003334"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003335Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003336\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003337 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003338 path: path of executable file\n\
3339 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003340 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003341
3342static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003343posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003344{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003345 PyObject *opath;
Guido van Rossuma1065681999-01-25 23:20:23 +00003346 char *path;
3347 PyObject *argv, *env;
3348 char **argvlist;
3349 char **envlist;
3350 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003351 int mode, pos, envc;
3352 Py_ssize_t argc, i;
Tim Peters79248aa2001-08-29 21:37:10 +00003353 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003354 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Thomas Wouters477c8d52006-05-27 19:21:47 +00003355 Py_ssize_t lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003356
3357 /* spawnve has four arguments: (mode, path, argv, env), where
3358 argv is a list or tuple of strings and env is a dictionary
3359 like posix.environ. */
3360
Martin v. Löwis011e8422009-05-05 04:43:17 +00003361 if (!PyArg_ParseTuple(args, "iO&OO:spawnve", &mode,
3362 PyUnicode_FSConverter,
3363 &opath, &argv, &env))
Guido van Rossuma1065681999-01-25 23:20:23 +00003364 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003365 path = bytes2str(opath, 1);
Guido van Rossuma1065681999-01-25 23:20:23 +00003366 if (PyList_Check(argv)) {
3367 argc = PyList_Size(argv);
3368 getitem = PyList_GetItem;
3369 }
3370 else if (PyTuple_Check(argv)) {
3371 argc = PyTuple_Size(argv);
3372 getitem = PyTuple_GetItem;
3373 }
3374 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003375 PyErr_SetString(PyExc_TypeError,
3376 "spawnve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003377 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003378 }
3379 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003380 PyErr_SetString(PyExc_TypeError,
3381 "spawnve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003382 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003383 }
3384
3385 argvlist = PyMem_NEW(char *, argc+1);
3386 if (argvlist == NULL) {
3387 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003388 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003389 }
3390 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003391 if (!fsconvert_strdup((*getitem)(argv, i),
3392 &argvlist[i]))
Guido van Rossuma1065681999-01-25 23:20:23 +00003393 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003394 lastarg = i;
Guido van Rossuma1065681999-01-25 23:20:23 +00003395 goto fail_1;
3396 }
3397 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003398 lastarg = argc;
Guido van Rossuma1065681999-01-25 23:20:23 +00003399 argvlist[argc] = NULL;
3400
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003401 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003402 if (i < 0)
3403 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00003404 envlist = PyMem_NEW(char *, i + 1);
3405 if (envlist == NULL) {
3406 PyErr_NoMemory();
3407 goto fail_1;
3408 }
3409 envc = 0;
3410 keys = PyMapping_Keys(env);
3411 vals = PyMapping_Values(env);
3412 if (!keys || !vals)
3413 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003414 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3415 PyErr_SetString(PyExc_TypeError,
3416 "spawnve(): env.keys() or env.values() is not a list");
3417 goto fail_2;
3418 }
Tim Peters5aa91602002-01-30 05:46:57 +00003419
Guido van Rossuma1065681999-01-25 23:20:23 +00003420 for (pos = 0; pos < i; pos++) {
3421 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003422 size_t len;
Guido van Rossuma1065681999-01-25 23:20:23 +00003423
3424 key = PyList_GetItem(keys, pos);
3425 val = PyList_GetItem(vals, pos);
3426 if (!key || !val)
3427 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003428
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003429 if (!PyArg_Parse(
3430 key,
3431 "s;spawnve() arg 3 contains a non-string key",
3432 &k) ||
3433 !PyArg_Parse(
3434 val,
3435 "s;spawnve() arg 3 contains a non-string value",
3436 &v))
Guido van Rossuma1065681999-01-25 23:20:23 +00003437 {
3438 goto fail_2;
3439 }
Christian Heimes830a4bc2007-11-22 07:43:40 +00003440 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Tim Petersc8996f52001-12-03 20:41:00 +00003441 p = PyMem_NEW(char, len);
Guido van Rossuma1065681999-01-25 23:20:23 +00003442 if (p == NULL) {
3443 PyErr_NoMemory();
3444 goto fail_2;
3445 }
Tim Petersc8996f52001-12-03 20:41:00 +00003446 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossuma1065681999-01-25 23:20:23 +00003447 envlist[envc++] = p;
3448 }
3449 envlist[envc] = 0;
3450
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003451#if defined(PYOS_OS2) && defined(PYCC_GCC)
3452 Py_BEGIN_ALLOW_THREADS
3453 spawnval = spawnve(mode, path, argvlist, envlist);
3454 Py_END_ALLOW_THREADS
3455#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003456 if (mode == _OLD_P_OVERLAY)
3457 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00003458
3459 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003460 spawnval = _spawnve(mode, path, argvlist, envlist);
Tim Peters25059d32001-12-07 20:35:43 +00003461 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003462#endif
Tim Peters25059d32001-12-07 20:35:43 +00003463
Fred Drake699f3522000-06-29 21:12:41 +00003464 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003465 (void) posix_error();
3466 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003467#if SIZEOF_LONG == SIZEOF_VOID_P
3468 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003469#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003470 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003471#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003472
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003473 fail_2:
Guido van Rossuma1065681999-01-25 23:20:23 +00003474 while (--envc >= 0)
3475 PyMem_DEL(envlist[envc]);
3476 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003477 fail_1:
Martin v. Löwis114619e2002-10-07 06:44:21 +00003478 free_string_array(argvlist, lastarg);
Guido van Rossuma1065681999-01-25 23:20:23 +00003479 Py_XDECREF(vals);
3480 Py_XDECREF(keys);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003481 fail_0:
Martin v. Löwis011e8422009-05-05 04:43:17 +00003482 release_bytes(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00003483 return res;
3484}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003485
3486/* OS/2 supports spawnvp & spawnvpe natively */
3487#if defined(PYOS_OS2)
3488PyDoc_STRVAR(posix_spawnvp__doc__,
3489"spawnvp(mode, file, args)\n\n\
3490Execute the program 'file' in a new process, using the environment\n\
3491search path to find the file.\n\
3492\n\
3493 mode: mode of process creation\n\
3494 file: executable file name\n\
3495 args: tuple or list of strings");
3496
3497static PyObject *
3498posix_spawnvp(PyObject *self, PyObject *args)
3499{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003500 PyObject *opath;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003501 char *path;
3502 PyObject *argv;
3503 char **argvlist;
3504 int mode, i, argc;
3505 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003506 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003507
3508 /* spawnvp has three arguments: (mode, path, argv), where
3509 argv is a list or tuple of strings. */
3510
Martin v. Löwis011e8422009-05-05 04:43:17 +00003511 if (!PyArg_ParseTuple(args, "iO&O:spawnvp", &mode,
3512 PyUnicode_FSConverter,
3513 &opath, &argv))
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003514 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003515 path = bytes2str(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003516 if (PyList_Check(argv)) {
3517 argc = PyList_Size(argv);
3518 getitem = PyList_GetItem;
3519 }
3520 else if (PyTuple_Check(argv)) {
3521 argc = PyTuple_Size(argv);
3522 getitem = PyTuple_GetItem;
3523 }
3524 else {
3525 PyErr_SetString(PyExc_TypeError,
3526 "spawnvp() arg 2 must be a tuple or list");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003527 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003528 return NULL;
3529 }
3530
3531 argvlist = PyMem_NEW(char *, argc+1);
3532 if (argvlist == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003533 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003534 return PyErr_NoMemory();
3535 }
3536 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003537 if (!fsconvert_strdup((*getitem)(argv, i),
3538 &argvlist[i])) {
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003539 free_string_array(argvlist, i);
3540 PyErr_SetString(
3541 PyExc_TypeError,
3542 "spawnvp() arg 2 must contain only strings");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003543 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003544 return NULL;
3545 }
3546 }
3547 argvlist[argc] = NULL;
3548
3549 Py_BEGIN_ALLOW_THREADS
3550#if defined(PYCC_GCC)
3551 spawnval = spawnvp(mode, path, argvlist);
3552#else
3553 spawnval = _spawnvp(mode, path, argvlist);
3554#endif
3555 Py_END_ALLOW_THREADS
3556
3557 free_string_array(argvlist, argc);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003558 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003559
3560 if (spawnval == -1)
3561 return posix_error();
3562 else
3563 return Py_BuildValue("l", (long) spawnval);
3564}
3565
3566
3567PyDoc_STRVAR(posix_spawnvpe__doc__,
3568"spawnvpe(mode, file, args, env)\n\n\
3569Execute the program 'file' in a new process, using the environment\n\
3570search path to find the file.\n\
3571\n\
3572 mode: mode of process creation\n\
3573 file: executable file name\n\
3574 args: tuple or list of arguments\n\
3575 env: dictionary of strings mapping to strings");
3576
3577static PyObject *
3578posix_spawnvpe(PyObject *self, PyObject *args)
3579{
Martin v. Löwis011e8422009-05-05 04:43:17 +00003580 PyObject *opath
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003581 char *path;
3582 PyObject *argv, *env;
3583 char **argvlist;
3584 char **envlist;
3585 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
3586 int mode, i, pos, argc, envc;
3587 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003588 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003589 int lastarg = 0;
3590
3591 /* spawnvpe has four arguments: (mode, path, argv, env), where
3592 argv is a list or tuple of strings and env is a dictionary
3593 like posix.environ. */
3594
3595 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
Martin v. Löwis011e8422009-05-05 04:43:17 +00003596 PyUnicode_FSConverter,
3597 &opath, &argv, &env))
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003598 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003599 path = bytes2str(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003600 if (PyList_Check(argv)) {
3601 argc = PyList_Size(argv);
3602 getitem = PyList_GetItem;
3603 }
3604 else if (PyTuple_Check(argv)) {
3605 argc = PyTuple_Size(argv);
3606 getitem = PyTuple_GetItem;
3607 }
3608 else {
3609 PyErr_SetString(PyExc_TypeError,
3610 "spawnvpe() arg 2 must be a tuple or list");
3611 goto fail_0;
3612 }
3613 if (!PyMapping_Check(env)) {
3614 PyErr_SetString(PyExc_TypeError,
3615 "spawnvpe() arg 3 must be a mapping object");
3616 goto fail_0;
3617 }
3618
3619 argvlist = PyMem_NEW(char *, argc+1);
3620 if (argvlist == NULL) {
3621 PyErr_NoMemory();
3622 goto fail_0;
3623 }
3624 for (i = 0; i < argc; i++) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003625 if (!fsconvert_strdup((*getitem)(argv, i),
3626 &argvlist[i]))
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003627 {
3628 lastarg = i;
3629 goto fail_1;
3630 }
3631 }
3632 lastarg = argc;
3633 argvlist[argc] = NULL;
3634
3635 i = PyMapping_Size(env);
3636 if (i < 0)
3637 goto fail_1;
3638 envlist = PyMem_NEW(char *, i + 1);
3639 if (envlist == NULL) {
3640 PyErr_NoMemory();
3641 goto fail_1;
3642 }
3643 envc = 0;
3644 keys = PyMapping_Keys(env);
3645 vals = PyMapping_Values(env);
3646 if (!keys || !vals)
3647 goto fail_2;
3648 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3649 PyErr_SetString(PyExc_TypeError,
3650 "spawnvpe(): env.keys() or env.values() is not a list");
3651 goto fail_2;
3652 }
3653
3654 for (pos = 0; pos < i; pos++) {
3655 char *p, *k, *v;
3656 size_t len;
3657
3658 key = PyList_GetItem(keys, pos);
3659 val = PyList_GetItem(vals, pos);
3660 if (!key || !val)
3661 goto fail_2;
3662
3663 if (!PyArg_Parse(
3664 key,
3665 "s;spawnvpe() arg 3 contains a non-string key",
3666 &k) ||
3667 !PyArg_Parse(
3668 val,
3669 "s;spawnvpe() arg 3 contains a non-string value",
3670 &v))
3671 {
3672 goto fail_2;
3673 }
Christian Heimes830a4bc2007-11-22 07:43:40 +00003674 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003675 p = PyMem_NEW(char, len);
3676 if (p == NULL) {
3677 PyErr_NoMemory();
3678 goto fail_2;
3679 }
3680 PyOS_snprintf(p, len, "%s=%s", k, v);
3681 envlist[envc++] = p;
3682 }
3683 envlist[envc] = 0;
3684
3685 Py_BEGIN_ALLOW_THREADS
3686#if defined(PYCC_GCC)
Christian Heimes292d3512008-02-03 16:51:08 +00003687 spawnval = spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003688#else
Christian Heimes292d3512008-02-03 16:51:08 +00003689 spawnval = _spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003690#endif
3691 Py_END_ALLOW_THREADS
3692
3693 if (spawnval == -1)
3694 (void) posix_error();
3695 else
3696 res = Py_BuildValue("l", (long) spawnval);
3697
3698 fail_2:
3699 while (--envc >= 0)
3700 PyMem_DEL(envlist[envc]);
3701 PyMem_DEL(envlist);
3702 fail_1:
3703 free_string_array(argvlist, lastarg);
3704 Py_XDECREF(vals);
3705 Py_XDECREF(keys);
3706 fail_0:
Martin v. Löwis011e8422009-05-05 04:43:17 +00003707 release_bytes(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003708 return res;
3709}
3710#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00003711#endif /* HAVE_SPAWNV */
3712
3713
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003714#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003715PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003716"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003717Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
3718\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003719Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003720
3721static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003722posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003723{
Benjamin Peterson0df35a92009-10-04 20:32:25 +00003724 pid_t pid;
3725 int result;
3726 _PyImport_AcquireLock();
3727 pid = fork1();
3728 result = _PyImport_ReleaseLock();
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003729 if (pid == -1)
3730 return posix_error();
Georg Brandl2ee470f2008-07-16 12:55:28 +00003731 if (pid == 0)
3732 PyOS_AfterFork();
Benjamin Peterson0df35a92009-10-04 20:32:25 +00003733 if (result < 0) {
3734 /* Don't clobber the OSError if the fork failed. */
3735 PyErr_SetString(PyExc_RuntimeError,
3736 "not holding the import lock");
3737 return NULL;
3738 }
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00003739 return PyLong_FromPid(pid);
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003740}
3741#endif
3742
3743
Guido van Rossumad0ee831995-03-01 10:34:45 +00003744#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003745PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003746"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003747Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003748Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003749
Barry Warsaw53699e91996-12-10 23:23:01 +00003750static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003751posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003752{
Benjamin Peterson0df35a92009-10-04 20:32:25 +00003753 pid_t pid;
3754 int result;
3755 _PyImport_AcquireLock();
3756 pid = fork();
3757 result = _PyImport_ReleaseLock();
Guido van Rossum85e3b011991-06-03 12:42:10 +00003758 if (pid == -1)
3759 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003760 if (pid == 0)
3761 PyOS_AfterFork();
Benjamin Peterson0df35a92009-10-04 20:32:25 +00003762 if (result < 0) {
3763 /* Don't clobber the OSError if the fork failed. */
3764 PyErr_SetString(PyExc_RuntimeError,
3765 "not holding the import lock");
3766 return NULL;
3767 }
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00003768 return PyLong_FromPid(pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003769}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003770#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003771
Neal Norwitzb59798b2003-03-21 01:43:31 +00003772/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00003773/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
3774#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00003775#define DEV_PTY_FILE "/dev/ptc"
3776#define HAVE_DEV_PTMX
3777#else
3778#define DEV_PTY_FILE "/dev/ptmx"
3779#endif
3780
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003781#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003782#ifdef HAVE_PTY_H
3783#include <pty.h>
3784#else
3785#ifdef HAVE_LIBUTIL_H
3786#include <libutil.h>
Fred Drake8cef4cf2000-06-28 16:40:38 +00003787#endif /* HAVE_LIBUTIL_H */
3788#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00003789#ifdef HAVE_STROPTS_H
3790#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003791#endif
3792#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003793
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003794#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003795PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003796"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003797Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003798
3799static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003800posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003801{
3802 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003803#ifndef HAVE_OPENPTY
3804 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003805#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003806#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003807 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003808#ifdef sun
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003809 extern char *ptsname(int fildes);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003810#endif
3811#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00003812
Thomas Wouters70c21a12000-07-14 14:28:33 +00003813#ifdef HAVE_OPENPTY
Fred Drake8cef4cf2000-06-28 16:40:38 +00003814 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
3815 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003816#elif defined(HAVE__GETPTY)
Thomas Wouters70c21a12000-07-14 14:28:33 +00003817 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
3818 if (slave_name == NULL)
3819 return posix_error();
3820
3821 slave_fd = open(slave_name, O_RDWR);
3822 if (slave_fd < 0)
3823 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003824#else
Neal Norwitzb59798b2003-03-21 01:43:31 +00003825 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003826 if (master_fd < 0)
3827 return posix_error();
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003828 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003829 /* change permission of slave */
3830 if (grantpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003831 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003832 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003833 }
3834 /* unlock slave */
3835 if (unlockpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003836 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003837 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003838 }
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003839 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003840 slave_name = ptsname(master_fd); /* get name of slave */
3841 if (slave_name == NULL)
3842 return posix_error();
3843 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
3844 if (slave_fd < 0)
3845 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003846#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003847 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
3848 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00003849#ifndef __hpux
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003850 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00003851#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003852#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00003853#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00003854
Fred Drake8cef4cf2000-06-28 16:40:38 +00003855 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00003856
Fred Drake8cef4cf2000-06-28 16:40:38 +00003857}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003858#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003859
3860#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003861PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003862"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00003863Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
3864Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003865To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003866
3867static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003868posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003869{
Benjamin Peterson0df35a92009-10-04 20:32:25 +00003870 int master_fd = -1, result;
Christian Heimes400adb02008-02-01 08:12:03 +00003871 pid_t pid;
Tim Peters5aa91602002-01-30 05:46:57 +00003872
Benjamin Peterson0df35a92009-10-04 20:32:25 +00003873 _PyImport_AcquireLock();
Fred Drake8cef4cf2000-06-28 16:40:38 +00003874 pid = forkpty(&master_fd, NULL, NULL, NULL);
Benjamin Peterson0df35a92009-10-04 20:32:25 +00003875 result = _PyImport_ReleaseLock();
Fred Drake8cef4cf2000-06-28 16:40:38 +00003876 if (pid == -1)
3877 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003878 if (pid == 0)
3879 PyOS_AfterFork();
Benjamin Peterson0df35a92009-10-04 20:32:25 +00003880 if (result < 0) {
3881 /* Don't clobber the OSError if the fork failed. */
3882 PyErr_SetString(PyExc_RuntimeError,
3883 "not holding the import lock");
3884 return NULL;
3885 }
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00003886 return Py_BuildValue("(Ni)", PyLong_FromPid(pid), master_fd);
Fred Drake8cef4cf2000-06-28 16:40:38 +00003887}
3888#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003889
Guido van Rossumad0ee831995-03-01 10:34:45 +00003890#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003891PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003892"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003893Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003894
Barry Warsaw53699e91996-12-10 23:23:01 +00003895static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003896posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003897{
Christian Heimes217cfd12007-12-02 14:31:20 +00003898 return PyLong_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003899}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003900#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003901
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003902
Guido van Rossumad0ee831995-03-01 10:34:45 +00003903#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003904PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003905"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003906Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003907
Barry Warsaw53699e91996-12-10 23:23:01 +00003908static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003909posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003910{
Christian Heimes217cfd12007-12-02 14:31:20 +00003911 return PyLong_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003912}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003913#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003914
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003915
Guido van Rossumad0ee831995-03-01 10:34:45 +00003916#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003917PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003918"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003919Return the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003920
Barry Warsaw53699e91996-12-10 23:23:01 +00003921static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003922posix_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003923{
Christian Heimes217cfd12007-12-02 14:31:20 +00003924 return PyLong_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003925}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003926#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003927
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003928
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003929PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003930"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003931Return the current process id");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003932
Barry Warsaw53699e91996-12-10 23:23:01 +00003933static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003934posix_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003935{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00003936 return PyLong_FromPid(getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00003937}
3938
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003939
Fred Drakec9680921999-12-13 16:37:25 +00003940#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003941PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003942"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003943Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00003944
3945static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003946posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00003947{
3948 PyObject *result = NULL;
3949
Fred Drakec9680921999-12-13 16:37:25 +00003950#ifdef NGROUPS_MAX
3951#define MAX_GROUPS NGROUPS_MAX
3952#else
3953 /* defined to be 16 on Solaris7, so this should be a small number */
3954#define MAX_GROUPS 64
3955#endif
3956 gid_t grouplist[MAX_GROUPS];
3957 int n;
3958
3959 n = getgroups(MAX_GROUPS, grouplist);
3960 if (n < 0)
3961 posix_error();
3962 else {
3963 result = PyList_New(n);
3964 if (result != NULL) {
Fred Drakec9680921999-12-13 16:37:25 +00003965 int i;
3966 for (i = 0; i < n; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +00003967 PyObject *o = PyLong_FromLong((long)grouplist[i]);
Fred Drakec9680921999-12-13 16:37:25 +00003968 if (o == NULL) {
3969 Py_DECREF(result);
3970 result = NULL;
3971 break;
3972 }
3973 PyList_SET_ITEM(result, i, o);
3974 }
3975 }
3976 }
Neal Norwitze241ce82003-02-17 18:17:05 +00003977
Fred Drakec9680921999-12-13 16:37:25 +00003978 return result;
3979}
3980#endif
3981
Antoine Pitroub7572f02009-12-02 20:46:48 +00003982#ifdef HAVE_INITGROUPS
3983PyDoc_STRVAR(posix_initgroups__doc__,
3984"initgroups(username, gid) -> None\n\n\
3985Call the system initgroups() to initialize the group access list with all of\n\
3986the groups of which the specified username is a member, plus the specified\n\
3987group id.");
3988
3989static PyObject *
3990posix_initgroups(PyObject *self, PyObject *args)
3991{
3992 char *username;
3993 long gid;
3994
3995 if (!PyArg_ParseTuple(args, "sl:initgroups", &username, &gid))
3996 return NULL;
3997
3998 if (initgroups(username, (gid_t) gid) == -1)
3999 return PyErr_SetFromErrno(PyExc_OSError);
4000
4001 Py_INCREF(Py_None);
4002 return Py_None;
4003}
4004#endif
4005
Martin v. Löwis606edc12002-06-13 21:09:11 +00004006#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00004007PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004008"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00004009Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00004010
4011static PyObject *
4012posix_getpgid(PyObject *self, PyObject *args)
4013{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004014 pid_t pid, pgid;
4015 if (!PyArg_ParseTuple(args, PARSE_PID ":getpgid", &pid))
Martin v. Löwis606edc12002-06-13 21:09:11 +00004016 return NULL;
4017 pgid = getpgid(pid);
4018 if (pgid < 0)
4019 return posix_error();
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004020 return PyLong_FromPid(pgid);
Martin v. Löwis606edc12002-06-13 21:09:11 +00004021}
4022#endif /* HAVE_GETPGID */
4023
4024
Guido van Rossumb6775db1994-08-01 11:34:53 +00004025#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004026PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004027"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004028Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004029
Barry Warsaw53699e91996-12-10 23:23:01 +00004030static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004031posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00004032{
Guido van Rossumb6775db1994-08-01 11:34:53 +00004033#ifdef GETPGRP_HAVE_ARG
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004034 return PyLong_FromPid(getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004035#else /* GETPGRP_HAVE_ARG */
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004036 return PyLong_FromPid(getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004037#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00004038}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004039#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00004040
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004041
Guido van Rossumb6775db1994-08-01 11:34:53 +00004042#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004043PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004044"setpgrp()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004045Make this process a session leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004046
Barry Warsaw53699e91996-12-10 23:23:01 +00004047static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004048posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004049{
Guido van Rossum64933891994-10-20 21:56:42 +00004050#ifdef SETPGRP_HAVE_ARG
Guido van Rossumc2670a01992-09-13 20:07:29 +00004051 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00004052#else /* SETPGRP_HAVE_ARG */
Guido van Rossumb6775db1994-08-01 11:34:53 +00004053 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00004054#endif /* SETPGRP_HAVE_ARG */
Guido van Rossum687dd131993-05-17 08:34:16 +00004055 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004056 Py_INCREF(Py_None);
4057 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004058}
4059
Guido van Rossumb6775db1994-08-01 11:34:53 +00004060#endif /* HAVE_SETPGRP */
4061
Guido van Rossumad0ee831995-03-01 10:34:45 +00004062#ifdef HAVE_GETPPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004063PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004064"getppid() -> ppid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004065Return the parent's process id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004066
Barry Warsaw53699e91996-12-10 23:23:01 +00004067static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004068posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004069{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004070 return PyLong_FromPid(getppid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00004071}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004072#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004073
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004074
Fred Drake12c6e2d1999-12-14 21:25:03 +00004075#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004076PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004077"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004078Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00004079
4080static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004081posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00004082{
Neal Norwitze241ce82003-02-17 18:17:05 +00004083 PyObject *result = NULL;
Fred Drakea30680b2000-12-06 21:24:28 +00004084 char *name;
4085 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00004086
Fred Drakea30680b2000-12-06 21:24:28 +00004087 errno = 0;
4088 name = getlogin();
4089 if (name == NULL) {
4090 if (errno)
4091 posix_error();
4092 else
4093 PyErr_SetString(PyExc_OSError,
Fred Drakee63544f2000-12-06 21:45:33 +00004094 "unable to determine login name");
Fred Drakea30680b2000-12-06 21:24:28 +00004095 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00004096 else
Neal Norwitz93c56822007-08-26 07:10:06 +00004097 result = PyUnicode_FromString(name);
Fred Drakea30680b2000-12-06 21:24:28 +00004098 errno = old_errno;
Neal Norwitze241ce82003-02-17 18:17:05 +00004099
Fred Drake12c6e2d1999-12-14 21:25:03 +00004100 return result;
4101}
4102#endif
4103
Guido van Rossumad0ee831995-03-01 10:34:45 +00004104#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004105PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004106"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004107Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004108
Barry Warsaw53699e91996-12-10 23:23:01 +00004109static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004110posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00004111{
Christian Heimes217cfd12007-12-02 14:31:20 +00004112 return PyLong_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00004113}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004114#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00004115
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004116
Guido van Rossumad0ee831995-03-01 10:34:45 +00004117#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004118PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004119"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004120Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004121
Barry Warsaw53699e91996-12-10 23:23:01 +00004122static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004123posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004124{
Christian Heimes292d3512008-02-03 16:51:08 +00004125 pid_t pid;
4126 int sig;
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004127 if (!PyArg_ParseTuple(args, PARSE_PID "i:kill", &pid, &sig))
Guido van Rossum85e3b011991-06-03 12:42:10 +00004128 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004129#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004130 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
4131 APIRET rc;
4132 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004133 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004134
4135 } else if (sig == XCPT_SIGNAL_KILLPROC) {
4136 APIRET rc;
4137 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004138 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004139
4140 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00004141 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004142#else
Guido van Rossum85e3b011991-06-03 12:42:10 +00004143 if (kill(pid, sig) == -1)
4144 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004145#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00004146 Py_INCREF(Py_None);
4147 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00004148}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004149#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004150
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004151#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004152PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004153"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004154Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004155
4156static PyObject *
4157posix_killpg(PyObject *self, PyObject *args)
4158{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004159 int sig;
4160 pid_t pgid;
4161 /* XXX some man pages make the `pgid` parameter an int, others
4162 a pid_t. Since getpgrp() returns a pid_t, we assume killpg should
4163 take the same type. Moreover, pid_t is always at least as wide as
4164 int (else compilation of this module fails), which is safe. */
4165 if (!PyArg_ParseTuple(args, PARSE_PID "i:killpg", &pgid, &sig))
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004166 return NULL;
4167 if (killpg(pgid, sig) == -1)
4168 return posix_error();
4169 Py_INCREF(Py_None);
4170 return Py_None;
4171}
4172#endif
4173
Guido van Rossumc0125471996-06-28 18:55:32 +00004174#ifdef HAVE_PLOCK
4175
4176#ifdef HAVE_SYS_LOCK_H
4177#include <sys/lock.h>
4178#endif
4179
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004180PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004181"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004182Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004183
Barry Warsaw53699e91996-12-10 23:23:01 +00004184static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004185posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00004186{
4187 int op;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004188 if (!PyArg_ParseTuple(args, "i:plock", &op))
Guido van Rossumc0125471996-06-28 18:55:32 +00004189 return NULL;
4190 if (plock(op) == -1)
4191 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004192 Py_INCREF(Py_None);
4193 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00004194}
4195#endif
4196
Guido van Rossumb6775db1994-08-01 11:34:53 +00004197#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004198PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004199"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004200Set the current process's user id.");
4201
Barry Warsaw53699e91996-12-10 23:23:01 +00004202static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004203posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004204{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004205 long uid_arg;
4206 uid_t uid;
4207 if (!PyArg_ParseTuple(args, "l:setuid", &uid_arg))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004208 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004209 uid = uid_arg;
4210 if (uid != uid_arg) {
4211 PyErr_SetString(PyExc_OverflowError, "user id too big");
4212 return NULL;
4213 }
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004214 if (setuid(uid) < 0)
4215 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004216 Py_INCREF(Py_None);
4217 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004218}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004219#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004220
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004221
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004222#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004223PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004224"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004225Set the current process's effective user id.");
4226
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004227static PyObject *
4228posix_seteuid (PyObject *self, PyObject *args)
4229{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004230 long euid_arg;
4231 uid_t euid;
4232 if (!PyArg_ParseTuple(args, "l", &euid_arg))
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004233 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004234 euid = euid_arg;
4235 if (euid != euid_arg) {
4236 PyErr_SetString(PyExc_OverflowError, "user id too big");
4237 return NULL;
4238 }
4239 if (seteuid(euid) < 0) {
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004240 return posix_error();
4241 } else {
4242 Py_INCREF(Py_None);
4243 return Py_None;
4244 }
4245}
4246#endif /* HAVE_SETEUID */
4247
4248#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004249PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004250"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004251Set the current process's effective group id.");
4252
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004253static PyObject *
4254posix_setegid (PyObject *self, PyObject *args)
4255{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004256 long egid_arg;
4257 gid_t egid;
4258 if (!PyArg_ParseTuple(args, "l", &egid_arg))
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004259 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004260 egid = egid_arg;
4261 if (egid != egid_arg) {
4262 PyErr_SetString(PyExc_OverflowError, "group id too big");
4263 return NULL;
4264 }
4265 if (setegid(egid) < 0) {
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004266 return posix_error();
4267 } else {
4268 Py_INCREF(Py_None);
4269 return Py_None;
4270 }
4271}
4272#endif /* HAVE_SETEGID */
4273
4274#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004275PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004276"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004277Set the current process's real and effective user ids.");
4278
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004279static PyObject *
4280posix_setreuid (PyObject *self, PyObject *args)
4281{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004282 long ruid_arg, euid_arg;
4283 uid_t ruid, euid;
4284 if (!PyArg_ParseTuple(args, "ll", &ruid_arg, &euid_arg))
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004285 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004286 ruid = ruid_arg;
4287 euid = euid_arg;
4288 if (euid != euid_arg || ruid != ruid_arg) {
4289 PyErr_SetString(PyExc_OverflowError, "user id too big");
4290 return NULL;
4291 }
4292 if (setreuid(ruid, euid) < 0) {
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004293 return posix_error();
4294 } else {
4295 Py_INCREF(Py_None);
4296 return Py_None;
4297 }
4298}
4299#endif /* HAVE_SETREUID */
4300
4301#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004302PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004303"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004304Set the current process's real and effective group ids.");
4305
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004306static PyObject *
4307posix_setregid (PyObject *self, PyObject *args)
4308{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004309 long rgid_arg, egid_arg;
4310 gid_t rgid, egid;
4311 if (!PyArg_ParseTuple(args, "ll", &rgid_arg, &egid_arg))
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004312 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004313 rgid = rgid_arg;
4314 egid = egid_arg;
4315 if (egid != egid_arg || rgid != rgid_arg) {
4316 PyErr_SetString(PyExc_OverflowError, "group id too big");
4317 return NULL;
4318 }
4319 if (setregid(rgid, egid) < 0) {
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004320 return posix_error();
4321 } else {
4322 Py_INCREF(Py_None);
4323 return Py_None;
4324 }
4325}
4326#endif /* HAVE_SETREGID */
4327
Guido van Rossumb6775db1994-08-01 11:34:53 +00004328#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004329PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004330"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004331Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004332
Barry Warsaw53699e91996-12-10 23:23:01 +00004333static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004334posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004335{
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004336 long gid_arg;
4337 gid_t gid;
4338 if (!PyArg_ParseTuple(args, "l:setgid", &gid_arg))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004339 return NULL;
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00004340 gid = gid_arg;
4341 if (gid != gid_arg) {
4342 PyErr_SetString(PyExc_OverflowError, "group id too big");
4343 return NULL;
4344 }
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004345 if (setgid(gid) < 0)
4346 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004347 Py_INCREF(Py_None);
4348 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004349}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004350#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004351
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004352#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004353PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004354"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004355Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004356
4357static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00004358posix_setgroups(PyObject *self, PyObject *groups)
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004359{
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004360 int i, len;
4361 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00004362
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004363 if (!PySequence_Check(groups)) {
4364 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
4365 return NULL;
4366 }
4367 len = PySequence_Size(groups);
4368 if (len > MAX_GROUPS) {
4369 PyErr_SetString(PyExc_ValueError, "too many groups");
4370 return NULL;
4371 }
4372 for(i = 0; i < len; i++) {
4373 PyObject *elem;
4374 elem = PySequence_GetItem(groups, i);
4375 if (!elem)
4376 return NULL;
Guido van Rossumddefaf32007-01-14 03:31:43 +00004377 if (!PyLong_Check(elem)) {
4378 PyErr_SetString(PyExc_TypeError,
4379 "groups must be integers");
4380 Py_DECREF(elem);
4381 return NULL;
4382 } else {
4383 unsigned long x = PyLong_AsUnsignedLong(elem);
4384 if (PyErr_Occurred()) {
4385 PyErr_SetString(PyExc_TypeError,
4386 "group id too big");
Georg Brandla13c2442005-11-22 19:30:31 +00004387 Py_DECREF(elem);
4388 return NULL;
Georg Brandla13c2442005-11-22 19:30:31 +00004389 }
Georg Brandla13c2442005-11-22 19:30:31 +00004390 grouplist[i] = x;
Guido van Rossumddefaf32007-01-14 03:31:43 +00004391 /* read back the value to see if it fitted in gid_t */
Georg Brandla13c2442005-11-22 19:30:31 +00004392 if (grouplist[i] != x) {
4393 PyErr_SetString(PyExc_TypeError,
4394 "group id too big");
4395 Py_DECREF(elem);
4396 return NULL;
4397 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004398 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004399 Py_DECREF(elem);
4400 }
4401
4402 if (setgroups(len, grouplist) < 0)
4403 return posix_error();
4404 Py_INCREF(Py_None);
4405 return Py_None;
4406}
4407#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004408
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004409#if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
4410static PyObject *
Christian Heimes292d3512008-02-03 16:51:08 +00004411wait_helper(pid_t pid, int status, struct rusage *ru)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004412{
4413 PyObject *result;
4414 static PyObject *struct_rusage;
4415
4416 if (pid == -1)
4417 return posix_error();
4418
4419 if (struct_rusage == NULL) {
Christian Heimes072c0f12008-01-03 23:01:04 +00004420 PyObject *m = PyImport_ImportModuleNoBlock("resource");
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004421 if (m == NULL)
4422 return NULL;
4423 struct_rusage = PyObject_GetAttrString(m, "struct_rusage");
4424 Py_DECREF(m);
4425 if (struct_rusage == NULL)
4426 return NULL;
4427 }
4428
4429 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
4430 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
4431 if (!result)
4432 return NULL;
4433
4434#ifndef doubletime
4435#define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
4436#endif
4437
4438 PyStructSequence_SET_ITEM(result, 0,
4439 PyFloat_FromDouble(doubletime(ru->ru_utime)));
4440 PyStructSequence_SET_ITEM(result, 1,
4441 PyFloat_FromDouble(doubletime(ru->ru_stime)));
4442#define SET_INT(result, index, value)\
Christian Heimes217cfd12007-12-02 14:31:20 +00004443 PyStructSequence_SET_ITEM(result, index, PyLong_FromLong(value))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004444 SET_INT(result, 2, ru->ru_maxrss);
4445 SET_INT(result, 3, ru->ru_ixrss);
4446 SET_INT(result, 4, ru->ru_idrss);
4447 SET_INT(result, 5, ru->ru_isrss);
4448 SET_INT(result, 6, ru->ru_minflt);
4449 SET_INT(result, 7, ru->ru_majflt);
4450 SET_INT(result, 8, ru->ru_nswap);
4451 SET_INT(result, 9, ru->ru_inblock);
4452 SET_INT(result, 10, ru->ru_oublock);
4453 SET_INT(result, 11, ru->ru_msgsnd);
4454 SET_INT(result, 12, ru->ru_msgrcv);
4455 SET_INT(result, 13, ru->ru_nsignals);
4456 SET_INT(result, 14, ru->ru_nvcsw);
4457 SET_INT(result, 15, ru->ru_nivcsw);
4458#undef SET_INT
4459
4460 if (PyErr_Occurred()) {
4461 Py_DECREF(result);
4462 return NULL;
4463 }
4464
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004465 return Py_BuildValue("NiN", PyLong_FromPid(pid), status, result);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004466}
4467#endif /* HAVE_WAIT3 || HAVE_WAIT4 */
4468
4469#ifdef HAVE_WAIT3
4470PyDoc_STRVAR(posix_wait3__doc__,
4471"wait3(options) -> (pid, status, rusage)\n\n\
4472Wait for completion of a child process.");
4473
4474static PyObject *
4475posix_wait3(PyObject *self, PyObject *args)
4476{
Christian Heimes292d3512008-02-03 16:51:08 +00004477 pid_t pid;
4478 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004479 struct rusage ru;
4480 WAIT_TYPE status;
4481 WAIT_STATUS_INT(status) = 0;
4482
4483 if (!PyArg_ParseTuple(args, "i:wait3", &options))
4484 return NULL;
4485
4486 Py_BEGIN_ALLOW_THREADS
4487 pid = wait3(&status, options, &ru);
4488 Py_END_ALLOW_THREADS
4489
4490 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
4491}
4492#endif /* HAVE_WAIT3 */
4493
4494#ifdef HAVE_WAIT4
4495PyDoc_STRVAR(posix_wait4__doc__,
4496"wait4(pid, options) -> (pid, status, rusage)\n\n\
4497Wait for completion of a given child process.");
4498
4499static PyObject *
4500posix_wait4(PyObject *self, PyObject *args)
4501{
Christian Heimes292d3512008-02-03 16:51:08 +00004502 pid_t pid;
4503 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004504 struct rusage ru;
4505 WAIT_TYPE status;
4506 WAIT_STATUS_INT(status) = 0;
4507
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004508 if (!PyArg_ParseTuple(args, PARSE_PID "i:wait4", &pid, &options))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004509 return NULL;
4510
4511 Py_BEGIN_ALLOW_THREADS
4512 pid = wait4(pid, &status, options, &ru);
4513 Py_END_ALLOW_THREADS
4514
4515 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
4516}
4517#endif /* HAVE_WAIT4 */
4518
Guido van Rossumb6775db1994-08-01 11:34:53 +00004519#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004520PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004521"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004522Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004523
Barry Warsaw53699e91996-12-10 23:23:01 +00004524static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004525posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004526{
Christian Heimes292d3512008-02-03 16:51:08 +00004527 pid_t pid;
4528 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004529 WAIT_TYPE status;
4530 WAIT_STATUS_INT(status) = 0;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004531
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004532 if (!PyArg_ParseTuple(args, PARSE_PID "i:waitpid", &pid, &options))
Guido van Rossum21803b81992-08-09 12:55:27 +00004533 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004534 Py_BEGIN_ALLOW_THREADS
Guido van Rossume6a3aa61999-02-01 16:15:30 +00004535 pid = waitpid(pid, &status, options);
Barry Warsaw53699e91996-12-10 23:23:01 +00004536 Py_END_ALLOW_THREADS
Guido van Rossum85e3b011991-06-03 12:42:10 +00004537 if (pid == -1)
4538 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004539
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004540 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum21803b81992-08-09 12:55:27 +00004541}
4542
Tim Petersab034fa2002-02-01 11:27:43 +00004543#elif defined(HAVE_CWAIT)
4544
4545/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004546PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004547"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004548"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00004549
4550static PyObject *
4551posix_waitpid(PyObject *self, PyObject *args)
4552{
Thomas Wouters477c8d52006-05-27 19:21:47 +00004553 Py_intptr_t pid;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004554 int status, options;
Tim Petersab034fa2002-02-01 11:27:43 +00004555
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004556 if (!PyArg_ParseTuple(args, PARSE_PID "i:waitpid", &pid, &options))
Tim Petersab034fa2002-02-01 11:27:43 +00004557 return NULL;
4558 Py_BEGIN_ALLOW_THREADS
4559 pid = _cwait(&status, pid, options);
4560 Py_END_ALLOW_THREADS
4561 if (pid == -1)
4562 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004563
4564 /* shift the status left a byte so this is more like the POSIX waitpid */
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004565 return Py_BuildValue("Ni", PyLong_FromPid(pid), status << 8);
Tim Petersab034fa2002-02-01 11:27:43 +00004566}
4567#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004568
Guido van Rossumad0ee831995-03-01 10:34:45 +00004569#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004570PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004571"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004572Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004573
Barry Warsaw53699e91996-12-10 23:23:01 +00004574static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004575posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00004576{
Christian Heimes292d3512008-02-03 16:51:08 +00004577 pid_t pid;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004578 WAIT_TYPE status;
4579 WAIT_STATUS_INT(status) = 0;
Neal Norwitze241ce82003-02-17 18:17:05 +00004580
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004581 Py_BEGIN_ALLOW_THREADS
4582 pid = wait(&status);
Barry Warsaw53699e91996-12-10 23:23:01 +00004583 Py_END_ALLOW_THREADS
Guido van Rossum21803b81992-08-09 12:55:27 +00004584 if (pid == -1)
4585 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004586
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004587 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum85e3b011991-06-03 12:42:10 +00004588}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004589#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004590
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004591
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004592PyDoc_STRVAR(posix_lstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004593"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004594Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004595
Barry Warsaw53699e91996-12-10 23:23:01 +00004596static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004597posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004598{
Guido van Rossumb6775db1994-08-01 11:34:53 +00004599#ifdef HAVE_LSTAT
Martin v. Löwis011e8422009-05-05 04:43:17 +00004600 return posix_do_stat(self, args, "O&:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004601#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004602#ifdef MS_WINDOWS
Martin v. Löwis011e8422009-05-05 04:43:17 +00004603 return posix_do_stat(self, args, "O&:lstat", STAT, "U:lstat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004604#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00004605 return posix_do_stat(self, args, "O&:lstat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004606#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00004607#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004608}
4609
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004610
Guido van Rossumb6775db1994-08-01 11:34:53 +00004611#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004612PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004613"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004614Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004615
Barry Warsaw53699e91996-12-10 23:23:01 +00004616static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004617posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004618{
Thomas Wouters89f507f2006-12-13 04:49:30 +00004619 PyObject* v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00004620 char buf[MAXPATHLEN];
Martin v. Löwis011e8422009-05-05 04:43:17 +00004621 PyObject *opath;
Guido van Rossumef0a00e1992-01-27 16:51:30 +00004622 char *path;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004623 int n;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004624 int arg_is_unicode = 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004625
Martin v. Löwis011e8422009-05-05 04:43:17 +00004626 if (!PyArg_ParseTuple(args, "O&:readlink",
4627 PyUnicode_FSConverter, &opath))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004628 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00004629 path = bytes2str(opath, 1);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004630 v = PySequence_GetItem(args, 0);
Neal Norwitzcda5c062007-08-12 17:09:36 +00004631 if (v == NULL) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00004632 release_bytes(opath);
Neal Norwitzcda5c062007-08-12 17:09:36 +00004633 return NULL;
4634 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004635
4636 if (PyUnicode_Check(v)) {
4637 arg_is_unicode = 1;
4638 }
4639 Py_DECREF(v);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004640
Barry Warsaw53699e91996-12-10 23:23:01 +00004641 Py_BEGIN_ALLOW_THREADS
Guido van Rossum50e61dc1992-03-27 17:22:31 +00004642 n = readlink(path, buf, (int) sizeof buf);
Barry Warsaw53699e91996-12-10 23:23:01 +00004643 Py_END_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004644 if (n < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00004645 return posix_error_with_allocated_filename(opath);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004646
Martin v. Löwis011e8422009-05-05 04:43:17 +00004647 release_bytes(opath);
Christian Heimes72b710a2008-05-26 13:28:38 +00004648 v = PyBytes_FromStringAndSize(buf, n);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004649 if (arg_is_unicode) {
4650 PyObject *w;
4651
4652 w = PyUnicode_FromEncodedObject(v,
4653 Py_FileSystemDefaultEncoding,
Martin v. Löwis43c57782009-05-10 08:15:24 +00004654 "surrogateescape");
Thomas Wouters89f507f2006-12-13 04:49:30 +00004655 if (w != NULL) {
4656 Py_DECREF(v);
4657 v = w;
4658 }
4659 else {
Guido van Rossumf0af3e32008-10-02 18:55:37 +00004660 v = NULL;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004661 }
4662 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004663 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004664}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004665#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004666
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004667
Guido van Rossumb6775db1994-08-01 11:34:53 +00004668#ifdef HAVE_SYMLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004669PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004670"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00004671Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004672
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004673static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004674posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004675{
Martin v. Löwis011e8422009-05-05 04:43:17 +00004676 return posix_2str(args, "O&O&:symlink", symlink);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004677}
4678#endif /* HAVE_SYMLINK */
4679
4680
4681#ifdef HAVE_TIMES
Guido van Rossumd48f2521997-12-05 22:19:34 +00004682#if defined(PYCC_VACPP) && defined(PYOS_OS2)
4683static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00004684system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004685{
4686 ULONG value = 0;
4687
4688 Py_BEGIN_ALLOW_THREADS
4689 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
4690 Py_END_ALLOW_THREADS
4691
4692 return value;
4693}
4694
4695static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004696posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004697{
Guido van Rossumd48f2521997-12-05 22:19:34 +00004698 /* Currently Only Uptime is Provided -- Others Later */
4699 return Py_BuildValue("ddddd",
4700 (double)0 /* t.tms_utime / HZ */,
4701 (double)0 /* t.tms_stime / HZ */,
4702 (double)0 /* t.tms_cutime / HZ */,
4703 (double)0 /* t.tms_cstime / HZ */,
4704 (double)system_uptime() / 1000);
4705}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004706#else /* not OS2 */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00004707#define NEED_TICKS_PER_SECOND
4708static long ticks_per_second = -1;
Barry Warsaw53699e91996-12-10 23:23:01 +00004709static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004710posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00004711{
4712 struct tms t;
4713 clock_t c;
Guido van Rossum22db57e1992-04-05 14:25:30 +00004714 errno = 0;
4715 c = times(&t);
Guido van Rossum687dd131993-05-17 08:34:16 +00004716 if (c == (clock_t) -1)
4717 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004718 return Py_BuildValue("ddddd",
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00004719 (double)t.tms_utime / ticks_per_second,
4720 (double)t.tms_stime / ticks_per_second,
4721 (double)t.tms_cutime / ticks_per_second,
4722 (double)t.tms_cstime / ticks_per_second,
4723 (double)c / ticks_per_second);
Guido van Rossum22db57e1992-04-05 14:25:30 +00004724}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004725#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00004726#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004727
4728
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004729#ifdef MS_WINDOWS
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004730#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00004731static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004732posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004733{
4734 FILETIME create, exit, kernel, user;
4735 HANDLE hProc;
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004736 hProc = GetCurrentProcess();
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004737 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
4738 /* The fields of a FILETIME structure are the hi and lo part
4739 of a 64-bit value expressed in 100 nanosecond units.
4740 1e7 is one second in such units; 1e-7 the inverse.
4741 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
4742 */
Barry Warsaw53699e91996-12-10 23:23:01 +00004743 return Py_BuildValue(
4744 "ddddd",
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004745 (double)(user.dwHighDateTime*429.4967296 +
4746 user.dwLowDateTime*1e-7),
Christian Heimes68f5fbe2008-02-14 08:27:37 +00004747 (double)(kernel.dwHighDateTime*429.4967296 +
4748 kernel.dwLowDateTime*1e-7),
Barry Warsaw53699e91996-12-10 23:23:01 +00004749 (double)0,
4750 (double)0,
4751 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004752}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004753#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004754
4755#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004756PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004757"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004758Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004759#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00004760
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004761
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004762#ifdef HAVE_GETSID
4763PyDoc_STRVAR(posix_getsid__doc__,
4764"getsid(pid) -> sid\n\n\
4765Call the system call getsid().");
4766
4767static PyObject *
4768posix_getsid(PyObject *self, PyObject *args)
4769{
Christian Heimes292d3512008-02-03 16:51:08 +00004770 pid_t pid;
4771 int sid;
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004772 if (!PyArg_ParseTuple(args, PARSE_PID ":getsid", &pid))
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004773 return NULL;
4774 sid = getsid(pid);
4775 if (sid < 0)
4776 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004777 return PyLong_FromLong((long)sid);
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004778}
4779#endif /* HAVE_GETSID */
4780
4781
Guido van Rossumb6775db1994-08-01 11:34:53 +00004782#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004783PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004784"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004785Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004786
Barry Warsaw53699e91996-12-10 23:23:01 +00004787static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004788posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004789{
Guido van Rossum687dd131993-05-17 08:34:16 +00004790 if (setsid() < 0)
4791 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004792 Py_INCREF(Py_None);
4793 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004794}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004795#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00004796
Guido van Rossumb6775db1994-08-01 11:34:53 +00004797#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004798PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004799"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004800Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004801
Barry Warsaw53699e91996-12-10 23:23:01 +00004802static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004803posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004804{
Christian Heimes292d3512008-02-03 16:51:08 +00004805 pid_t pid;
4806 int pgrp;
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004807 if (!PyArg_ParseTuple(args, PARSE_PID "i:setpgid", &pid, &pgrp))
Guido van Rossumc2670a01992-09-13 20:07:29 +00004808 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004809 if (setpgid(pid, pgrp) < 0)
4810 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004811 Py_INCREF(Py_None);
4812 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004813}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004814#endif /* HAVE_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00004815
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004816
Guido van Rossumb6775db1994-08-01 11:34:53 +00004817#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004818PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004819"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004820Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004821
Barry Warsaw53699e91996-12-10 23:23:01 +00004822static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004823posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00004824{
Christian Heimes15ebc882008-02-04 18:48:49 +00004825 int fd;
4826 pid_t pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004827 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
Guido van Rossum7066dd71992-09-17 17:54:56 +00004828 return NULL;
4829 pgid = tcgetpgrp(fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004830 if (pgid < 0)
4831 return posix_error();
Antoine Pitrouc3ee1662009-05-23 16:02:33 +00004832 return PyLong_FromPid(pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00004833}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004834#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00004835
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004836
Guido van Rossumb6775db1994-08-01 11:34:53 +00004837#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004838PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004839"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004840Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004841
Barry Warsaw53699e91996-12-10 23:23:01 +00004842static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004843posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00004844{
Antoine Pitrou971e51b2009-05-23 16:14:27 +00004845 int fd;
4846 pid_t pgid;
4847 if (!PyArg_ParseTuple(args, "i" PARSE_PID ":tcsetpgrp", &fd, &pgid))
Guido van Rossum7066dd71992-09-17 17:54:56 +00004848 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004849 if (tcsetpgrp(fd, pgid) < 0)
4850 return posix_error();
Barry Warsaw43d68b81996-12-19 22:10:44 +00004851 Py_INCREF(Py_None);
Barry Warsaw53699e91996-12-10 23:23:01 +00004852 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00004853}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004854#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00004855
Guido van Rossum687dd131993-05-17 08:34:16 +00004856/* Functions acting on file descriptors */
4857
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004858PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004859"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004860Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004861
Barry Warsaw53699e91996-12-10 23:23:01 +00004862static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004863posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004864{
Martin v. Löwis011e8422009-05-05 04:43:17 +00004865 PyObject *ofile;
4866 char *file;
Guido van Rossum687dd131993-05-17 08:34:16 +00004867 int flag;
4868 int mode = 0777;
4869 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004870
4871#ifdef MS_WINDOWS
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00004872 PyUnicodeObject *po;
4873 if (PyArg_ParseTuple(args, "Ui|i:mkdir", &po, &flag, &mode)) {
4874 Py_BEGIN_ALLOW_THREADS
4875 /* PyUnicode_AS_UNICODE OK without thread
4876 lock as it is a simple dereference. */
4877 fd = _wopen(PyUnicode_AS_UNICODE(po), flag, mode);
4878 Py_END_ALLOW_THREADS
4879 if (fd < 0)
4880 return posix_error();
4881 return PyLong_FromLong((long)fd);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004882 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00004883 /* Drop the argument parsing error as narrow strings
4884 are also valid. */
4885 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004886#endif
4887
Martin v. Löwis011e8422009-05-05 04:43:17 +00004888 if (!PyArg_ParseTuple(args, "O&i|i",
4889 PyUnicode_FSConverter, &ofile,
Mark Hammondef8b6542001-05-13 08:04:26 +00004890 &flag, &mode))
Barry Warsaw43d68b81996-12-19 22:10:44 +00004891 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00004892 file = bytes2str(ofile, 1);
Barry Warsaw53699e91996-12-10 23:23:01 +00004893 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004894 fd = open(file, flag, mode);
Barry Warsaw53699e91996-12-10 23:23:01 +00004895 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004896 if (fd < 0)
Martin v. Löwis011e8422009-05-05 04:43:17 +00004897 return posix_error_with_allocated_filename(ofile);
4898 release_bytes(ofile);
Christian Heimes217cfd12007-12-02 14:31:20 +00004899 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004900}
4901
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004902
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004903PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004904"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004905Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004906
Barry Warsaw53699e91996-12-10 23:23:01 +00004907static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004908posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004909{
4910 int fd, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004911 if (!PyArg_ParseTuple(args, "i:close", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00004912 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00004913 if (!_PyVerify_fd(fd))
4914 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004915 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004916 res = close(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00004917 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004918 if (res < 0)
4919 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004920 Py_INCREF(Py_None);
4921 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00004922}
4923
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004924
Christian Heimesfdab48e2008-01-20 09:06:41 +00004925PyDoc_STRVAR(posix_closerange__doc__,
4926"closerange(fd_low, fd_high)\n\n\
4927Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
4928
4929static PyObject *
4930posix_closerange(PyObject *self, PyObject *args)
4931{
4932 int fd_from, fd_to, i;
4933 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
4934 return NULL;
4935 Py_BEGIN_ALLOW_THREADS
4936 for (i = fd_from; i < fd_to; i++)
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00004937 if (_PyVerify_fd(i))
4938 close(i);
Christian Heimesfdab48e2008-01-20 09:06:41 +00004939 Py_END_ALLOW_THREADS
4940 Py_RETURN_NONE;
4941}
4942
4943
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004944PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004945"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004946Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004947
Barry Warsaw53699e91996-12-10 23:23:01 +00004948static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004949posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004950{
4951 int fd;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004952 if (!PyArg_ParseTuple(args, "i:dup", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00004953 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00004954 if (!_PyVerify_fd(fd))
4955 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004956 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004957 fd = dup(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00004958 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004959 if (fd < 0)
4960 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004961 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004962}
4963
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004964
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004965PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00004966"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004967Duplicate file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004968
Barry Warsaw53699e91996-12-10 23:23:01 +00004969static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004970posix_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004971{
4972 int fd, fd2, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004973 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
Guido van Rossum687dd131993-05-17 08:34:16 +00004974 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00004975 if (!_PyVerify_fd_dup2(fd, fd2))
4976 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004977 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004978 res = dup2(fd, fd2);
Barry Warsaw53699e91996-12-10 23:23:01 +00004979 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004980 if (res < 0)
4981 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004982 Py_INCREF(Py_None);
4983 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00004984}
4985
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004986
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004987PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004988"lseek(fd, pos, how) -> newpos\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004989Set the current position of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004990
Barry Warsaw53699e91996-12-10 23:23:01 +00004991static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004992posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004993{
4994 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004995#if defined(MS_WIN64) || defined(MS_WINDOWS)
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00004996 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00004997#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00004998 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00004999#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00005000 PyObject *posobj;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005001 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Guido van Rossum687dd131993-05-17 08:34:16 +00005002 return NULL;
5003#ifdef SEEK_SET
5004 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
5005 switch (how) {
5006 case 0: how = SEEK_SET; break;
5007 case 1: how = SEEK_CUR; break;
5008 case 2: how = SEEK_END; break;
5009 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005010#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00005011
5012#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005013 pos = PyLong_AsLong(posobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005014#else
5015 pos = PyLong_Check(posobj) ?
Christian Heimes217cfd12007-12-02 14:31:20 +00005016 PyLong_AsLongLong(posobj) : PyLong_AsLong(posobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005017#endif
5018 if (PyErr_Occurred())
5019 return NULL;
5020
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005021 if (!_PyVerify_fd(fd))
5022 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005023 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005024#if defined(MS_WIN64) || defined(MS_WINDOWS)
Fred Drake699f3522000-06-29 21:12:41 +00005025 res = _lseeki64(fd, pos, how);
5026#else
Guido van Rossum687dd131993-05-17 08:34:16 +00005027 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00005028#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00005029 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005030 if (res < 0)
5031 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00005032
5033#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005034 return PyLong_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005035#else
5036 return PyLong_FromLongLong(res);
5037#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00005038}
5039
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005040
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005041PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005042"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005043Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005044
Barry Warsaw53699e91996-12-10 23:23:01 +00005045static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005046posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005047{
Guido van Rossum572dbf82007-04-27 23:53:51 +00005048 int fd, size;
5049 Py_ssize_t n;
Barry Warsaw53699e91996-12-10 23:23:01 +00005050 PyObject *buffer;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005051 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
Guido van Rossum687dd131993-05-17 08:34:16 +00005052 return NULL;
Michael W. Hudson9867ced2005-01-31 17:01:59 +00005053 if (size < 0) {
5054 errno = EINVAL;
5055 return posix_error();
5056 }
Christian Heimes72b710a2008-05-26 13:28:38 +00005057 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
Guido van Rossum687dd131993-05-17 08:34:16 +00005058 if (buffer == NULL)
5059 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005060 if (!_PyVerify_fd(fd))
5061 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005062 Py_BEGIN_ALLOW_THREADS
Christian Heimes72b710a2008-05-26 13:28:38 +00005063 n = read(fd, PyBytes_AS_STRING(buffer), size);
Barry Warsaw53699e91996-12-10 23:23:01 +00005064 Py_END_ALLOW_THREADS
Guido van Rossum8bac5461996-06-11 18:38:48 +00005065 if (n < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00005066 Py_DECREF(buffer);
Guido van Rossum687dd131993-05-17 08:34:16 +00005067 return posix_error();
5068 }
Guido van Rossum8bac5461996-06-11 18:38:48 +00005069 if (n != size)
Christian Heimes72b710a2008-05-26 13:28:38 +00005070 _PyBytes_Resize(&buffer, n);
Guido van Rossum687dd131993-05-17 08:34:16 +00005071 return buffer;
5072}
5073
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005074
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005075PyDoc_STRVAR(posix_write__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005076"write(fd, string) -> byteswritten\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005077Write a string to a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005078
Barry Warsaw53699e91996-12-10 23:23:01 +00005079static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005080posix_write(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005081{
Martin v. Löwis423be952008-08-13 15:53:07 +00005082 Py_buffer pbuf;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00005083 int fd;
5084 Py_ssize_t size;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00005085
Antoine Pitrou9cadb1b2008-09-15 23:02:56 +00005086 if (!PyArg_ParseTuple(args, "iy*:write", &fd, &pbuf))
Guido van Rossum687dd131993-05-17 08:34:16 +00005087 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005088 if (!_PyVerify_fd(fd))
5089 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005090 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis423be952008-08-13 15:53:07 +00005091 size = write(fd, pbuf.buf, (size_t)pbuf.len);
Barry Warsaw53699e91996-12-10 23:23:01 +00005092 Py_END_ALLOW_THREADS
Martin v. Löwis423be952008-08-13 15:53:07 +00005093 PyBuffer_Release(&pbuf);
Guido van Rossum687dd131993-05-17 08:34:16 +00005094 if (size < 0)
5095 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00005096 return PyLong_FromSsize_t(size);
Guido van Rossum687dd131993-05-17 08:34:16 +00005097}
5098
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005099
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005100PyDoc_STRVAR(posix_fstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005101"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005102Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005103
Barry Warsaw53699e91996-12-10 23:23:01 +00005104static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005105posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005106{
5107 int fd;
Fred Drake699f3522000-06-29 21:12:41 +00005108 STRUCT_STAT st;
Guido van Rossum687dd131993-05-17 08:34:16 +00005109 int res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005110 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00005111 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00005112#ifdef __VMS
5113 /* on OpenVMS we must ensure that all bytes are written to the file */
5114 fsync(fd);
5115#endif
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005116 if (!_PyVerify_fd(fd))
5117 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005118 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00005119 res = FSTAT(fd, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00005120 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00005121 if (res != 0) {
5122#ifdef MS_WINDOWS
5123 return win32_error("fstat", NULL);
5124#else
Guido van Rossum687dd131993-05-17 08:34:16 +00005125 return posix_error();
Martin v. Löwis14694662006-02-03 12:54:16 +00005126#endif
5127 }
Tim Peters5aa91602002-01-30 05:46:57 +00005128
Martin v. Löwis14694662006-02-03 12:54:16 +00005129 return _pystat_fromstructstat(&st);
Guido van Rossum687dd131993-05-17 08:34:16 +00005130}
5131
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005132PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005133"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00005134Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005135connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00005136
5137static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00005138posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00005139{
5140 int fd;
5141 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
5142 return NULL;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00005143 if (!_PyVerify_fd(fd))
5144 return PyBool_FromLong(0);
Fred Drake106c1a02002-04-23 15:58:02 +00005145 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00005146}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005147
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005148#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005149PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005150"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005151Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005152
Barry Warsaw53699e91996-12-10 23:23:01 +00005153static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005154posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00005155{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005156#if defined(PYOS_OS2)
5157 HFILE read, write;
5158 APIRET rc;
5159
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005160 Py_BEGIN_ALLOW_THREADS
5161 rc = DosCreatePipe( &read, &write, 4096);
5162 Py_END_ALLOW_THREADS
5163 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005164 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005165
5166 return Py_BuildValue("(ii)", read, write);
5167#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005168#if !defined(MS_WINDOWS)
Guido van Rossum687dd131993-05-17 08:34:16 +00005169 int fds[2];
5170 int res;
Barry Warsaw53699e91996-12-10 23:23:01 +00005171 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005172 res = pipe(fds);
Barry Warsaw53699e91996-12-10 23:23:01 +00005173 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00005174 if (res != 0)
5175 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005176 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005177#else /* MS_WINDOWS */
Guido van Rossum794d8131994-08-23 13:48:48 +00005178 HANDLE read, write;
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00005179 int read_fd, write_fd;
Guido van Rossum794d8131994-08-23 13:48:48 +00005180 BOOL ok;
Barry Warsaw53699e91996-12-10 23:23:01 +00005181 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00005182 ok = CreatePipe(&read, &write, NULL, 0);
Barry Warsaw53699e91996-12-10 23:23:01 +00005183 Py_END_ALLOW_THREADS
Guido van Rossum794d8131994-08-23 13:48:48 +00005184 if (!ok)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00005185 return win32_error("CreatePipe", NULL);
Tim Peters79248aa2001-08-29 21:37:10 +00005186 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
5187 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00005188 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005189#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005190#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00005191}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005192#endif /* HAVE_PIPE */
5193
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005194
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005195#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005196PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005197"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005198Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005199
Barry Warsaw53699e91996-12-10 23:23:01 +00005200static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005201posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005202{
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005203 char *filename;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005204 int mode = 0666;
5205 int res;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005206 if (!PyArg_ParseTuple(args, "s|i:mkfifo", &filename, &mode))
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005207 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005208 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005209 res = mkfifo(filename, mode);
5210 Py_END_ALLOW_THREADS
5211 if (res < 0)
5212 return posix_error();
5213 Py_INCREF(Py_None);
5214 return Py_None;
5215}
5216#endif
5217
5218
Neal Norwitz11690112002-07-30 01:08:28 +00005219#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005220PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005221"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005222Create a filesystem node (file, device special file or named pipe)\n\
5223named filename. mode specifies both the permissions to use and the\n\
5224type of node to be created, being combined (bitwise OR) with one of\n\
5225S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005226device defines the newly created device special file (probably using\n\
5227os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005228
5229
5230static PyObject *
5231posix_mknod(PyObject *self, PyObject *args)
5232{
5233 char *filename;
5234 int mode = 0600;
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005235 int device = 0;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005236 int res;
Martin v. Löwisd631ebe2002-11-02 17:42:33 +00005237 if (!PyArg_ParseTuple(args, "s|ii:mknod", &filename, &mode, &device))
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005238 return NULL;
5239 Py_BEGIN_ALLOW_THREADS
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005240 res = mknod(filename, mode, device);
Barry Warsaw53699e91996-12-10 23:23:01 +00005241 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005242 if (res < 0)
5243 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005244 Py_INCREF(Py_None);
5245 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005246}
5247#endif
5248
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005249#ifdef HAVE_DEVICE_MACROS
5250PyDoc_STRVAR(posix_major__doc__,
5251"major(device) -> major number\n\
5252Extracts a device major number from a raw device number.");
5253
5254static PyObject *
5255posix_major(PyObject *self, PyObject *args)
5256{
5257 int device;
5258 if (!PyArg_ParseTuple(args, "i:major", &device))
5259 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005260 return PyLong_FromLong((long)major(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005261}
5262
5263PyDoc_STRVAR(posix_minor__doc__,
5264"minor(device) -> minor number\n\
5265Extracts a device minor number from a raw device number.");
5266
5267static PyObject *
5268posix_minor(PyObject *self, PyObject *args)
5269{
5270 int device;
5271 if (!PyArg_ParseTuple(args, "i:minor", &device))
5272 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005273 return PyLong_FromLong((long)minor(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005274}
5275
5276PyDoc_STRVAR(posix_makedev__doc__,
5277"makedev(major, minor) -> device number\n\
5278Composes a raw device number from the major and minor device numbers.");
5279
5280static PyObject *
5281posix_makedev(PyObject *self, PyObject *args)
5282{
5283 int major, minor;
5284 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
5285 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005286 return PyLong_FromLong((long)makedev(major, minor));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005287}
5288#endif /* device macros */
5289
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005290
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005291#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005292PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005293"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005294Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005295
Barry Warsaw53699e91996-12-10 23:23:01 +00005296static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005297posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005298{
5299 int fd;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005300 off_t length;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005301 int res;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005302 PyObject *lenobj;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005303
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005304 if (!PyArg_ParseTuple(args, "iO:ftruncate", &fd, &lenobj))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005305 return NULL;
5306
5307#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005308 length = PyLong_AsLong(lenobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005309#else
5310 length = PyLong_Check(lenobj) ?
Christian Heimes217cfd12007-12-02 14:31:20 +00005311 PyLong_AsLongLong(lenobj) : PyLong_AsLong(lenobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005312#endif
5313 if (PyErr_Occurred())
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005314 return NULL;
5315
Barry Warsaw53699e91996-12-10 23:23:01 +00005316 Py_BEGIN_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005317 res = ftruncate(fd, length);
Barry Warsaw53699e91996-12-10 23:23:01 +00005318 Py_END_ALLOW_THREADS
Benjamin Peterson9053d752009-01-19 17:53:36 +00005319 if (res < 0)
5320 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005321 Py_INCREF(Py_None);
5322 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005323}
5324#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00005325
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005326#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005327PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005328"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005329Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005330
Fred Drake762e2061999-08-26 17:23:54 +00005331/* Save putenv() parameters as values here, so we can collect them when they
5332 * get re-set with another call for the same key. */
5333static PyObject *posix_putenv_garbage;
5334
Tim Peters5aa91602002-01-30 05:46:57 +00005335static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005336posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005337{
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005338#ifdef MS_WINDOWS
5339 wchar_t *s1, *s2;
5340 wchar_t *newenv;
5341#else
Martin v. Löwis011e8422009-05-05 04:43:17 +00005342 PyObject *os1, *os2;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005343 char *s1, *s2;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005344 char *newenv;
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005345#endif
Fred Drake762e2061999-08-26 17:23:54 +00005346 PyObject *newstr;
Tim Petersc8996f52001-12-03 20:41:00 +00005347 size_t len;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005348
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005349#ifdef MS_WINDOWS
Martin v. Löwis011e8422009-05-05 04:43:17 +00005350 if (!PyArg_ParseTuple(args,
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005351 "uu:putenv",
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005352 &s1, &s2))
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005353 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00005354#else
5355 if (!PyArg_ParseTuple(args,
5356 "O&O&:putenv",
5357 PyUnicode_FSConverter, &os1,
5358 PyUnicode_FSConverter, &os2))
5359 return NULL;
5360 s1 = bytes2str(os1, 1);
5361 s2 = bytes2str(os2, 1);
5362#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00005363
5364#if defined(PYOS_OS2)
5365 if (stricmp(s1, "BEGINLIBPATH") == 0) {
5366 APIRET rc;
5367
Guido van Rossumd48f2521997-12-05 22:19:34 +00005368 rc = DosSetExtLIBPATH(s2, BEGIN_LIBPATH);
5369 if (rc != NO_ERROR)
5370 return os2_error(rc);
5371
5372 } else if (stricmp(s1, "ENDLIBPATH") == 0) {
5373 APIRET rc;
5374
Guido van Rossumd48f2521997-12-05 22:19:34 +00005375 rc = DosSetExtLIBPATH(s2, END_LIBPATH);
5376 if (rc != NO_ERROR)
5377 return os2_error(rc);
5378 } else {
5379#endif
Fred Drake762e2061999-08-26 17:23:54 +00005380 /* XXX This can leak memory -- not easy to fix :-( */
Tim Petersc8996f52001-12-03 20:41:00 +00005381 /* len includes space for a trailing \0; the size arg to
Christian Heimes72b710a2008-05-26 13:28:38 +00005382 PyBytes_FromStringAndSize does not count that */
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005383#ifdef MS_WINDOWS
5384 len = wcslen(s1) + wcslen(s2) + 2;
5385 newstr = PyUnicode_FromUnicode(NULL, (int)len - 1);
5386#else
5387 len = strlen(s1) + strlen(s2) + 2;
Christian Heimes72b710a2008-05-26 13:28:38 +00005388 newstr = PyBytes_FromStringAndSize(NULL, (int)len - 1);
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005389#endif
Fred Drake762e2061999-08-26 17:23:54 +00005390 if (newstr == NULL)
5391 return PyErr_NoMemory();
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005392#ifdef MS_WINDOWS
5393 newenv = PyUnicode_AsUnicode(newstr);
5394 _snwprintf(newenv, len, L"%s=%s", s1, s2);
5395 if (_wputenv(newenv)) {
5396 Py_DECREF(newstr);
5397 posix_error();
5398 return NULL;
5399 }
5400#else
Christian Heimes72b710a2008-05-26 13:28:38 +00005401 newenv = PyBytes_AS_STRING(newstr);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005402 PyOS_snprintf(newenv, len, "%s=%s", s1, s2);
5403 if (putenv(newenv)) {
Neal Norwitz4adc9ab2003-02-10 03:10:43 +00005404 Py_DECREF(newstr);
Martin v. Löwis011e8422009-05-05 04:43:17 +00005405 release_bytes(os1);
5406 release_bytes(os2);
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005407 posix_error();
5408 return NULL;
5409 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005410#endif
Fred Drake762e2061999-08-26 17:23:54 +00005411 /* Install the first arg and newstr in posix_putenv_garbage;
5412 * this will cause previous value to be collected. This has to
5413 * happen after the real putenv() call because the old value
5414 * was still accessible until then. */
5415 if (PyDict_SetItem(posix_putenv_garbage,
5416 PyTuple_GET_ITEM(args, 0), newstr)) {
5417 /* really not much we can do; just leak */
5418 PyErr_Clear();
5419 }
5420 else {
5421 Py_DECREF(newstr);
5422 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00005423
5424#if defined(PYOS_OS2)
5425 }
5426#endif
Martin v. Löwis011e8422009-05-05 04:43:17 +00005427#ifndef MS_WINDOWS
5428 release_bytes(os1);
5429 release_bytes(os2);
5430#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00005431 Py_INCREF(Py_None);
5432 return Py_None;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005433}
Guido van Rossumb6a47161997-09-15 22:54:34 +00005434#endif /* putenv */
5435
Guido van Rossumc524d952001-10-19 01:31:59 +00005436#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005437PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005438"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005439Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00005440
5441static PyObject *
5442posix_unsetenv(PyObject *self, PyObject *args)
5443{
5444 char *s1;
5445
5446 if (!PyArg_ParseTuple(args, "s:unsetenv", &s1))
5447 return NULL;
5448
5449 unsetenv(s1);
5450
5451 /* Remove the key from posix_putenv_garbage;
5452 * this will cause it to be collected. This has to
Tim Peters5aa91602002-01-30 05:46:57 +00005453 * happen after the real unsetenv() call because the
Guido van Rossumc524d952001-10-19 01:31:59 +00005454 * old value was still accessible until then.
5455 */
5456 if (PyDict_DelItem(posix_putenv_garbage,
5457 PyTuple_GET_ITEM(args, 0))) {
5458 /* really not much we can do; just leak */
5459 PyErr_Clear();
5460 }
5461
5462 Py_INCREF(Py_None);
5463 return Py_None;
5464}
5465#endif /* unsetenv */
5466
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005467PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005468"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005469Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005470
Guido van Rossumf68d8e52001-04-14 17:55:09 +00005471static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005472posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00005473{
5474 int code;
5475 char *message;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005476 if (!PyArg_ParseTuple(args, "i:strerror", &code))
Guido van Rossumb6a47161997-09-15 22:54:34 +00005477 return NULL;
5478 message = strerror(code);
5479 if (message == NULL) {
5480 PyErr_SetString(PyExc_ValueError,
Fred Drake661ea262000-10-24 19:57:45 +00005481 "strerror() argument out of range");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005482 return NULL;
5483 }
Neal Norwitz93c56822007-08-26 07:10:06 +00005484 return PyUnicode_FromString(message);
Guido van Rossumb6a47161997-09-15 22:54:34 +00005485}
Guido van Rossumb6a47161997-09-15 22:54:34 +00005486
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005487
Guido van Rossumc9641791998-08-04 15:26:23 +00005488#ifdef HAVE_SYS_WAIT_H
5489
Fred Drake106c1a02002-04-23 15:58:02 +00005490#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005491PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005492"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005493Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00005494
5495static PyObject *
5496posix_WCOREDUMP(PyObject *self, PyObject *args)
5497{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005498 WAIT_TYPE status;
5499 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00005500
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005501 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00005502 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00005503
5504 return PyBool_FromLong(WCOREDUMP(status));
Fred Drake106c1a02002-04-23 15:58:02 +00005505}
5506#endif /* WCOREDUMP */
5507
5508#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005509PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005510"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00005511Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005512job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00005513
5514static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005515posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00005516{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005517 WAIT_TYPE status;
5518 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00005519
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005520 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00005521 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00005522
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005523 return PyBool_FromLong(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00005524}
5525#endif /* WIFCONTINUED */
5526
Guido van Rossumc9641791998-08-04 15:26:23 +00005527#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005528PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005529"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005530Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005531
5532static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005533posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005534{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005535 WAIT_TYPE status;
5536 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005537
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005538 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005539 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005540
Fred Drake106c1a02002-04-23 15:58:02 +00005541 return PyBool_FromLong(WIFSTOPPED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005542}
5543#endif /* WIFSTOPPED */
5544
5545#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005546PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005547"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005548Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005549
5550static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005551posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005552{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005553 WAIT_TYPE status;
5554 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005555
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005556 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005557 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005558
Fred Drake106c1a02002-04-23 15:58:02 +00005559 return PyBool_FromLong(WIFSIGNALED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005560}
5561#endif /* WIFSIGNALED */
5562
5563#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005564PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005565"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005566Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005567system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005568
5569static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005570posix_WIFEXITED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005571{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005572 WAIT_TYPE status;
5573 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005574
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005575 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005576 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005577
Fred Drake106c1a02002-04-23 15:58:02 +00005578 return PyBool_FromLong(WIFEXITED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005579}
5580#endif /* WIFEXITED */
5581
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005582#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005583PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005584"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005585Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005586
5587static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005588posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005589{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005590 WAIT_TYPE status;
5591 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005592
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005593 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005594 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005595
Guido van Rossumc9641791998-08-04 15:26:23 +00005596 return Py_BuildValue("i", WEXITSTATUS(status));
5597}
5598#endif /* WEXITSTATUS */
5599
5600#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005601PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005602"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005603Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005604value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005605
5606static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005607posix_WTERMSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005608{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005609 WAIT_TYPE status;
5610 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005611
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005612 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005613 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005614
Guido van Rossumc9641791998-08-04 15:26:23 +00005615 return Py_BuildValue("i", WTERMSIG(status));
5616}
5617#endif /* WTERMSIG */
5618
5619#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005620PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005621"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005622Return the signal that stopped the process that provided\n\
5623the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005624
5625static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005626posix_WSTOPSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005627{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005628 WAIT_TYPE status;
5629 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005630
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005631 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005632 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005633
Guido van Rossumc9641791998-08-04 15:26:23 +00005634 return Py_BuildValue("i", WSTOPSIG(status));
5635}
5636#endif /* WSTOPSIG */
5637
5638#endif /* HAVE_SYS_WAIT_H */
5639
5640
Thomas Wouters477c8d52006-05-27 19:21:47 +00005641#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossumd5753e11999-10-19 13:29:23 +00005642#ifdef _SCO_DS
5643/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
5644 needed definitions in sys/statvfs.h */
5645#define _SVID3
5646#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00005647#include <sys/statvfs.h>
5648
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005649static PyObject*
5650_pystatvfs_fromstructstatvfs(struct statvfs st) {
5651 PyObject *v = PyStructSequence_New(&StatVFSResultType);
5652 if (v == NULL)
5653 return NULL;
5654
5655#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005656 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
5657 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
5658 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long) st.f_blocks));
5659 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long) st.f_bfree));
5660 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long) st.f_bavail));
5661 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long) st.f_files));
5662 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong((long) st.f_ffree));
5663 PyStructSequence_SET_ITEM(v, 7, PyLong_FromLong((long) st.f_favail));
5664 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
5665 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005666#else
Christian Heimes217cfd12007-12-02 14:31:20 +00005667 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
5668 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
Tim Peters5aa91602002-01-30 05:46:57 +00005669 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005670 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
Tim Peters5aa91602002-01-30 05:46:57 +00005671 PyStructSequence_SET_ITEM(v, 3,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005672 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005673 PyStructSequence_SET_ITEM(v, 4,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005674 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
Tim Peters5aa91602002-01-30 05:46:57 +00005675 PyStructSequence_SET_ITEM(v, 5,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005676 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
Tim Peters5aa91602002-01-30 05:46:57 +00005677 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005678 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
Tim Peters5aa91602002-01-30 05:46:57 +00005679 PyStructSequence_SET_ITEM(v, 7,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005680 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
Christian Heimes217cfd12007-12-02 14:31:20 +00005681 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
5682 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005683#endif
5684
5685 return v;
5686}
5687
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005688PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005689"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005690Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005691
5692static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005693posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005694{
5695 int fd, res;
5696 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005697
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005698 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005699 return NULL;
5700 Py_BEGIN_ALLOW_THREADS
5701 res = fstatvfs(fd, &st);
5702 Py_END_ALLOW_THREADS
5703 if (res != 0)
5704 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005705
5706 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005707}
Thomas Wouters477c8d52006-05-27 19:21:47 +00005708#endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00005709
5710
Thomas Wouters477c8d52006-05-27 19:21:47 +00005711#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005712#include <sys/statvfs.h>
5713
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005714PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005715"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005716Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005717
5718static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005719posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005720{
5721 char *path;
5722 int res;
5723 struct statvfs st;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005724 if (!PyArg_ParseTuple(args, "s:statvfs", &path))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005725 return NULL;
5726 Py_BEGIN_ALLOW_THREADS
5727 res = statvfs(path, &st);
5728 Py_END_ALLOW_THREADS
5729 if (res != 0)
5730 return posix_error_with_filename(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005731
5732 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005733}
5734#endif /* HAVE_STATVFS */
5735
Fred Drakec9680921999-12-13 16:37:25 +00005736/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
5737 * It maps strings representing configuration variable names to
5738 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00005739 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00005740 * rarely-used constants. There are three separate tables that use
5741 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00005742 *
5743 * This code is always included, even if none of the interfaces that
5744 * need it are included. The #if hackery needed to avoid it would be
5745 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00005746 */
5747struct constdef {
5748 char *name;
5749 long value;
5750};
5751
Fred Drake12c6e2d1999-12-14 21:25:03 +00005752static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005753conv_confname(PyObject *arg, int *valuep, struct constdef *table,
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005754 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00005755{
Christian Heimes217cfd12007-12-02 14:31:20 +00005756 if (PyLong_Check(arg)) {
5757 *valuep = PyLong_AS_LONG(arg);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005758 return 1;
5759 }
Guido van Rossumbce56a62007-05-10 18:04:33 +00005760 else {
Fred Drake12c6e2d1999-12-14 21:25:03 +00005761 /* look up the value in the table using a binary search */
Fred Drake699f3522000-06-29 21:12:41 +00005762 size_t lo = 0;
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005763 size_t mid;
Fred Drake699f3522000-06-29 21:12:41 +00005764 size_t hi = tablesize;
5765 int cmp;
Guido van Rossumbce56a62007-05-10 18:04:33 +00005766 const char *confname;
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005767 if (!PyUnicode_Check(arg)) {
Guido van Rossumbce56a62007-05-10 18:04:33 +00005768 PyErr_SetString(PyExc_TypeError,
5769 "configuration names must be strings or integers");
5770 return 0;
5771 }
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00005772 confname = _PyUnicode_AsString(arg);
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005773 if (confname == NULL)
5774 return 0;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005775 while (lo < hi) {
5776 mid = (lo + hi) / 2;
5777 cmp = strcmp(confname, table[mid].name);
5778 if (cmp < 0)
5779 hi = mid;
5780 else if (cmp > 0)
5781 lo = mid + 1;
5782 else {
5783 *valuep = table[mid].value;
5784 return 1;
5785 }
5786 }
5787 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
Guido van Rossumbce56a62007-05-10 18:04:33 +00005788 return 0;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005789 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00005790}
5791
5792
5793#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
5794static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00005795#ifdef _PC_ABI_AIO_XFER_MAX
5796 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
5797#endif
5798#ifdef _PC_ABI_ASYNC_IO
5799 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
5800#endif
Fred Drakec9680921999-12-13 16:37:25 +00005801#ifdef _PC_ASYNC_IO
5802 {"PC_ASYNC_IO", _PC_ASYNC_IO},
5803#endif
5804#ifdef _PC_CHOWN_RESTRICTED
5805 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
5806#endif
5807#ifdef _PC_FILESIZEBITS
5808 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
5809#endif
5810#ifdef _PC_LAST
5811 {"PC_LAST", _PC_LAST},
5812#endif
5813#ifdef _PC_LINK_MAX
5814 {"PC_LINK_MAX", _PC_LINK_MAX},
5815#endif
5816#ifdef _PC_MAX_CANON
5817 {"PC_MAX_CANON", _PC_MAX_CANON},
5818#endif
5819#ifdef _PC_MAX_INPUT
5820 {"PC_MAX_INPUT", _PC_MAX_INPUT},
5821#endif
5822#ifdef _PC_NAME_MAX
5823 {"PC_NAME_MAX", _PC_NAME_MAX},
5824#endif
5825#ifdef _PC_NO_TRUNC
5826 {"PC_NO_TRUNC", _PC_NO_TRUNC},
5827#endif
5828#ifdef _PC_PATH_MAX
5829 {"PC_PATH_MAX", _PC_PATH_MAX},
5830#endif
5831#ifdef _PC_PIPE_BUF
5832 {"PC_PIPE_BUF", _PC_PIPE_BUF},
5833#endif
5834#ifdef _PC_PRIO_IO
5835 {"PC_PRIO_IO", _PC_PRIO_IO},
5836#endif
5837#ifdef _PC_SOCK_MAXBUF
5838 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
5839#endif
5840#ifdef _PC_SYNC_IO
5841 {"PC_SYNC_IO", _PC_SYNC_IO},
5842#endif
5843#ifdef _PC_VDISABLE
5844 {"PC_VDISABLE", _PC_VDISABLE},
5845#endif
5846};
5847
Fred Drakec9680921999-12-13 16:37:25 +00005848static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005849conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00005850{
5851 return conv_confname(arg, valuep, posix_constants_pathconf,
5852 sizeof(posix_constants_pathconf)
5853 / sizeof(struct constdef));
5854}
5855#endif
5856
5857#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005858PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005859"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00005860Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005861If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00005862
5863static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005864posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005865{
5866 PyObject *result = NULL;
5867 int name, fd;
5868
Fred Drake12c6e2d1999-12-14 21:25:03 +00005869 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
5870 conv_path_confname, &name)) {
Fred Drakec9680921999-12-13 16:37:25 +00005871 long limit;
5872
5873 errno = 0;
5874 limit = fpathconf(fd, name);
5875 if (limit == -1 && errno != 0)
5876 posix_error();
5877 else
Christian Heimes217cfd12007-12-02 14:31:20 +00005878 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00005879 }
5880 return result;
5881}
5882#endif
5883
5884
5885#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005886PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005887"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00005888Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005889If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00005890
5891static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005892posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005893{
5894 PyObject *result = NULL;
5895 int name;
5896 char *path;
5897
5898 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
5899 conv_path_confname, &name)) {
5900 long limit;
5901
5902 errno = 0;
5903 limit = pathconf(path, name);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005904 if (limit == -1 && errno != 0) {
Fred Drakec9680921999-12-13 16:37:25 +00005905 if (errno == EINVAL)
5906 /* could be a path or name problem */
5907 posix_error();
5908 else
5909 posix_error_with_filename(path);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005910 }
Fred Drakec9680921999-12-13 16:37:25 +00005911 else
Christian Heimes217cfd12007-12-02 14:31:20 +00005912 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00005913 }
5914 return result;
5915}
5916#endif
5917
5918#ifdef HAVE_CONFSTR
5919static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00005920#ifdef _CS_ARCHITECTURE
5921 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
5922#endif
5923#ifdef _CS_HOSTNAME
5924 {"CS_HOSTNAME", _CS_HOSTNAME},
5925#endif
5926#ifdef _CS_HW_PROVIDER
5927 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
5928#endif
5929#ifdef _CS_HW_SERIAL
5930 {"CS_HW_SERIAL", _CS_HW_SERIAL},
5931#endif
5932#ifdef _CS_INITTAB_NAME
5933 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
5934#endif
Fred Drakec9680921999-12-13 16:37:25 +00005935#ifdef _CS_LFS64_CFLAGS
5936 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
5937#endif
5938#ifdef _CS_LFS64_LDFLAGS
5939 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
5940#endif
5941#ifdef _CS_LFS64_LIBS
5942 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
5943#endif
5944#ifdef _CS_LFS64_LINTFLAGS
5945 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
5946#endif
5947#ifdef _CS_LFS_CFLAGS
5948 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
5949#endif
5950#ifdef _CS_LFS_LDFLAGS
5951 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
5952#endif
5953#ifdef _CS_LFS_LIBS
5954 {"CS_LFS_LIBS", _CS_LFS_LIBS},
5955#endif
5956#ifdef _CS_LFS_LINTFLAGS
5957 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
5958#endif
Fred Draked86ed291999-12-15 15:34:33 +00005959#ifdef _CS_MACHINE
5960 {"CS_MACHINE", _CS_MACHINE},
5961#endif
Fred Drakec9680921999-12-13 16:37:25 +00005962#ifdef _CS_PATH
5963 {"CS_PATH", _CS_PATH},
5964#endif
Fred Draked86ed291999-12-15 15:34:33 +00005965#ifdef _CS_RELEASE
5966 {"CS_RELEASE", _CS_RELEASE},
5967#endif
5968#ifdef _CS_SRPC_DOMAIN
5969 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
5970#endif
5971#ifdef _CS_SYSNAME
5972 {"CS_SYSNAME", _CS_SYSNAME},
5973#endif
5974#ifdef _CS_VERSION
5975 {"CS_VERSION", _CS_VERSION},
5976#endif
Fred Drakec9680921999-12-13 16:37:25 +00005977#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
5978 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
5979#endif
5980#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
5981 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
5982#endif
5983#ifdef _CS_XBS5_ILP32_OFF32_LIBS
5984 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
5985#endif
5986#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
5987 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
5988#endif
5989#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
5990 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
5991#endif
5992#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
5993 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
5994#endif
5995#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
5996 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
5997#endif
5998#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
5999 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
6000#endif
6001#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
6002 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
6003#endif
6004#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
6005 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
6006#endif
6007#ifdef _CS_XBS5_LP64_OFF64_LIBS
6008 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
6009#endif
6010#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
6011 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
6012#endif
6013#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
6014 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
6015#endif
6016#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
6017 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
6018#endif
6019#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
6020 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
6021#endif
6022#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
6023 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
6024#endif
Fred Draked86ed291999-12-15 15:34:33 +00006025#ifdef _MIPS_CS_AVAIL_PROCESSORS
6026 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
6027#endif
6028#ifdef _MIPS_CS_BASE
6029 {"MIPS_CS_BASE", _MIPS_CS_BASE},
6030#endif
6031#ifdef _MIPS_CS_HOSTID
6032 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
6033#endif
6034#ifdef _MIPS_CS_HW_NAME
6035 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
6036#endif
6037#ifdef _MIPS_CS_NUM_PROCESSORS
6038 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
6039#endif
6040#ifdef _MIPS_CS_OSREL_MAJ
6041 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
6042#endif
6043#ifdef _MIPS_CS_OSREL_MIN
6044 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
6045#endif
6046#ifdef _MIPS_CS_OSREL_PATCH
6047 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
6048#endif
6049#ifdef _MIPS_CS_OS_NAME
6050 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
6051#endif
6052#ifdef _MIPS_CS_OS_PROVIDER
6053 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
6054#endif
6055#ifdef _MIPS_CS_PROCESSORS
6056 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
6057#endif
6058#ifdef _MIPS_CS_SERIAL
6059 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
6060#endif
6061#ifdef _MIPS_CS_VENDOR
6062 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
6063#endif
Fred Drakec9680921999-12-13 16:37:25 +00006064};
6065
6066static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006067conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00006068{
6069 return conv_confname(arg, valuep, posix_constants_confstr,
6070 sizeof(posix_constants_confstr)
6071 / sizeof(struct constdef));
6072}
6073
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006074PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006075"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006076Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00006077
6078static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006079posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006080{
6081 PyObject *result = NULL;
6082 int name;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006083 char buffer[256];
Fred Drakec9680921999-12-13 16:37:25 +00006084
6085 if (PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name)) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006086 int len;
Fred Drakec9680921999-12-13 16:37:25 +00006087
Fred Drakec9680921999-12-13 16:37:25 +00006088 errno = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006089 len = confstr(name, buffer, sizeof(buffer));
6090 if (len == 0) {
6091 if (errno) {
6092 posix_error();
6093 }
6094 else {
6095 result = Py_None;
6096 Py_INCREF(Py_None);
6097 }
Fred Drakec9680921999-12-13 16:37:25 +00006098 }
6099 else {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006100 if ((unsigned int)len >= sizeof(buffer)) {
Neal Norwitz93c56822007-08-26 07:10:06 +00006101 result = PyUnicode_FromStringAndSize(NULL, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00006102 if (result != NULL)
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00006103 confstr(name, _PyUnicode_AsString(result), len);
Fred Drakec9680921999-12-13 16:37:25 +00006104 }
6105 else
Neal Norwitz93c56822007-08-26 07:10:06 +00006106 result = PyUnicode_FromStringAndSize(buffer, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00006107 }
6108 }
6109 return result;
6110}
6111#endif
6112
6113
6114#ifdef HAVE_SYSCONF
6115static struct constdef posix_constants_sysconf[] = {
6116#ifdef _SC_2_CHAR_TERM
6117 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
6118#endif
6119#ifdef _SC_2_C_BIND
6120 {"SC_2_C_BIND", _SC_2_C_BIND},
6121#endif
6122#ifdef _SC_2_C_DEV
6123 {"SC_2_C_DEV", _SC_2_C_DEV},
6124#endif
6125#ifdef _SC_2_C_VERSION
6126 {"SC_2_C_VERSION", _SC_2_C_VERSION},
6127#endif
6128#ifdef _SC_2_FORT_DEV
6129 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
6130#endif
6131#ifdef _SC_2_FORT_RUN
6132 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
6133#endif
6134#ifdef _SC_2_LOCALEDEF
6135 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
6136#endif
6137#ifdef _SC_2_SW_DEV
6138 {"SC_2_SW_DEV", _SC_2_SW_DEV},
6139#endif
6140#ifdef _SC_2_UPE
6141 {"SC_2_UPE", _SC_2_UPE},
6142#endif
6143#ifdef _SC_2_VERSION
6144 {"SC_2_VERSION", _SC_2_VERSION},
6145#endif
Fred Draked86ed291999-12-15 15:34:33 +00006146#ifdef _SC_ABI_ASYNCHRONOUS_IO
6147 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
6148#endif
6149#ifdef _SC_ACL
6150 {"SC_ACL", _SC_ACL},
6151#endif
Fred Drakec9680921999-12-13 16:37:25 +00006152#ifdef _SC_AIO_LISTIO_MAX
6153 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
6154#endif
Fred Drakec9680921999-12-13 16:37:25 +00006155#ifdef _SC_AIO_MAX
6156 {"SC_AIO_MAX", _SC_AIO_MAX},
6157#endif
6158#ifdef _SC_AIO_PRIO_DELTA_MAX
6159 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
6160#endif
6161#ifdef _SC_ARG_MAX
6162 {"SC_ARG_MAX", _SC_ARG_MAX},
6163#endif
6164#ifdef _SC_ASYNCHRONOUS_IO
6165 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
6166#endif
6167#ifdef _SC_ATEXIT_MAX
6168 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
6169#endif
Fred Draked86ed291999-12-15 15:34:33 +00006170#ifdef _SC_AUDIT
6171 {"SC_AUDIT", _SC_AUDIT},
6172#endif
Fred Drakec9680921999-12-13 16:37:25 +00006173#ifdef _SC_AVPHYS_PAGES
6174 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
6175#endif
6176#ifdef _SC_BC_BASE_MAX
6177 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
6178#endif
6179#ifdef _SC_BC_DIM_MAX
6180 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
6181#endif
6182#ifdef _SC_BC_SCALE_MAX
6183 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
6184#endif
6185#ifdef _SC_BC_STRING_MAX
6186 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
6187#endif
Fred Draked86ed291999-12-15 15:34:33 +00006188#ifdef _SC_CAP
6189 {"SC_CAP", _SC_CAP},
6190#endif
Fred Drakec9680921999-12-13 16:37:25 +00006191#ifdef _SC_CHARCLASS_NAME_MAX
6192 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
6193#endif
6194#ifdef _SC_CHAR_BIT
6195 {"SC_CHAR_BIT", _SC_CHAR_BIT},
6196#endif
6197#ifdef _SC_CHAR_MAX
6198 {"SC_CHAR_MAX", _SC_CHAR_MAX},
6199#endif
6200#ifdef _SC_CHAR_MIN
6201 {"SC_CHAR_MIN", _SC_CHAR_MIN},
6202#endif
6203#ifdef _SC_CHILD_MAX
6204 {"SC_CHILD_MAX", _SC_CHILD_MAX},
6205#endif
6206#ifdef _SC_CLK_TCK
6207 {"SC_CLK_TCK", _SC_CLK_TCK},
6208#endif
6209#ifdef _SC_COHER_BLKSZ
6210 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
6211#endif
6212#ifdef _SC_COLL_WEIGHTS_MAX
6213 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
6214#endif
6215#ifdef _SC_DCACHE_ASSOC
6216 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
6217#endif
6218#ifdef _SC_DCACHE_BLKSZ
6219 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
6220#endif
6221#ifdef _SC_DCACHE_LINESZ
6222 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
6223#endif
6224#ifdef _SC_DCACHE_SZ
6225 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
6226#endif
6227#ifdef _SC_DCACHE_TBLKSZ
6228 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
6229#endif
6230#ifdef _SC_DELAYTIMER_MAX
6231 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
6232#endif
6233#ifdef _SC_EQUIV_CLASS_MAX
6234 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
6235#endif
6236#ifdef _SC_EXPR_NEST_MAX
6237 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
6238#endif
6239#ifdef _SC_FSYNC
6240 {"SC_FSYNC", _SC_FSYNC},
6241#endif
6242#ifdef _SC_GETGR_R_SIZE_MAX
6243 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
6244#endif
6245#ifdef _SC_GETPW_R_SIZE_MAX
6246 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
6247#endif
6248#ifdef _SC_ICACHE_ASSOC
6249 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
6250#endif
6251#ifdef _SC_ICACHE_BLKSZ
6252 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
6253#endif
6254#ifdef _SC_ICACHE_LINESZ
6255 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
6256#endif
6257#ifdef _SC_ICACHE_SZ
6258 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
6259#endif
Fred Draked86ed291999-12-15 15:34:33 +00006260#ifdef _SC_INF
6261 {"SC_INF", _SC_INF},
6262#endif
Fred Drakec9680921999-12-13 16:37:25 +00006263#ifdef _SC_INT_MAX
6264 {"SC_INT_MAX", _SC_INT_MAX},
6265#endif
6266#ifdef _SC_INT_MIN
6267 {"SC_INT_MIN", _SC_INT_MIN},
6268#endif
6269#ifdef _SC_IOV_MAX
6270 {"SC_IOV_MAX", _SC_IOV_MAX},
6271#endif
Fred Draked86ed291999-12-15 15:34:33 +00006272#ifdef _SC_IP_SECOPTS
6273 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
6274#endif
Fred Drakec9680921999-12-13 16:37:25 +00006275#ifdef _SC_JOB_CONTROL
6276 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
6277#endif
Fred Draked86ed291999-12-15 15:34:33 +00006278#ifdef _SC_KERN_POINTERS
6279 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
6280#endif
6281#ifdef _SC_KERN_SIM
6282 {"SC_KERN_SIM", _SC_KERN_SIM},
6283#endif
Fred Drakec9680921999-12-13 16:37:25 +00006284#ifdef _SC_LINE_MAX
6285 {"SC_LINE_MAX", _SC_LINE_MAX},
6286#endif
6287#ifdef _SC_LOGIN_NAME_MAX
6288 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
6289#endif
6290#ifdef _SC_LOGNAME_MAX
6291 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
6292#endif
6293#ifdef _SC_LONG_BIT
6294 {"SC_LONG_BIT", _SC_LONG_BIT},
6295#endif
Fred Draked86ed291999-12-15 15:34:33 +00006296#ifdef _SC_MAC
6297 {"SC_MAC", _SC_MAC},
6298#endif
Fred Drakec9680921999-12-13 16:37:25 +00006299#ifdef _SC_MAPPED_FILES
6300 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
6301#endif
6302#ifdef _SC_MAXPID
6303 {"SC_MAXPID", _SC_MAXPID},
6304#endif
6305#ifdef _SC_MB_LEN_MAX
6306 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
6307#endif
6308#ifdef _SC_MEMLOCK
6309 {"SC_MEMLOCK", _SC_MEMLOCK},
6310#endif
6311#ifdef _SC_MEMLOCK_RANGE
6312 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
6313#endif
6314#ifdef _SC_MEMORY_PROTECTION
6315 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
6316#endif
6317#ifdef _SC_MESSAGE_PASSING
6318 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
6319#endif
Fred Draked86ed291999-12-15 15:34:33 +00006320#ifdef _SC_MMAP_FIXED_ALIGNMENT
6321 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
6322#endif
Fred Drakec9680921999-12-13 16:37:25 +00006323#ifdef _SC_MQ_OPEN_MAX
6324 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
6325#endif
6326#ifdef _SC_MQ_PRIO_MAX
6327 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
6328#endif
Fred Draked86ed291999-12-15 15:34:33 +00006329#ifdef _SC_NACLS_MAX
6330 {"SC_NACLS_MAX", _SC_NACLS_MAX},
6331#endif
Fred Drakec9680921999-12-13 16:37:25 +00006332#ifdef _SC_NGROUPS_MAX
6333 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
6334#endif
6335#ifdef _SC_NL_ARGMAX
6336 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
6337#endif
6338#ifdef _SC_NL_LANGMAX
6339 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
6340#endif
6341#ifdef _SC_NL_MSGMAX
6342 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
6343#endif
6344#ifdef _SC_NL_NMAX
6345 {"SC_NL_NMAX", _SC_NL_NMAX},
6346#endif
6347#ifdef _SC_NL_SETMAX
6348 {"SC_NL_SETMAX", _SC_NL_SETMAX},
6349#endif
6350#ifdef _SC_NL_TEXTMAX
6351 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
6352#endif
6353#ifdef _SC_NPROCESSORS_CONF
6354 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
6355#endif
6356#ifdef _SC_NPROCESSORS_ONLN
6357 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
6358#endif
Fred Draked86ed291999-12-15 15:34:33 +00006359#ifdef _SC_NPROC_CONF
6360 {"SC_NPROC_CONF", _SC_NPROC_CONF},
6361#endif
6362#ifdef _SC_NPROC_ONLN
6363 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
6364#endif
Fred Drakec9680921999-12-13 16:37:25 +00006365#ifdef _SC_NZERO
6366 {"SC_NZERO", _SC_NZERO},
6367#endif
6368#ifdef _SC_OPEN_MAX
6369 {"SC_OPEN_MAX", _SC_OPEN_MAX},
6370#endif
6371#ifdef _SC_PAGESIZE
6372 {"SC_PAGESIZE", _SC_PAGESIZE},
6373#endif
6374#ifdef _SC_PAGE_SIZE
6375 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
6376#endif
6377#ifdef _SC_PASS_MAX
6378 {"SC_PASS_MAX", _SC_PASS_MAX},
6379#endif
6380#ifdef _SC_PHYS_PAGES
6381 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
6382#endif
6383#ifdef _SC_PII
6384 {"SC_PII", _SC_PII},
6385#endif
6386#ifdef _SC_PII_INTERNET
6387 {"SC_PII_INTERNET", _SC_PII_INTERNET},
6388#endif
6389#ifdef _SC_PII_INTERNET_DGRAM
6390 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
6391#endif
6392#ifdef _SC_PII_INTERNET_STREAM
6393 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
6394#endif
6395#ifdef _SC_PII_OSI
6396 {"SC_PII_OSI", _SC_PII_OSI},
6397#endif
6398#ifdef _SC_PII_OSI_CLTS
6399 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
6400#endif
6401#ifdef _SC_PII_OSI_COTS
6402 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
6403#endif
6404#ifdef _SC_PII_OSI_M
6405 {"SC_PII_OSI_M", _SC_PII_OSI_M},
6406#endif
6407#ifdef _SC_PII_SOCKET
6408 {"SC_PII_SOCKET", _SC_PII_SOCKET},
6409#endif
6410#ifdef _SC_PII_XTI
6411 {"SC_PII_XTI", _SC_PII_XTI},
6412#endif
6413#ifdef _SC_POLL
6414 {"SC_POLL", _SC_POLL},
6415#endif
6416#ifdef _SC_PRIORITIZED_IO
6417 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
6418#endif
6419#ifdef _SC_PRIORITY_SCHEDULING
6420 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
6421#endif
6422#ifdef _SC_REALTIME_SIGNALS
6423 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
6424#endif
6425#ifdef _SC_RE_DUP_MAX
6426 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
6427#endif
6428#ifdef _SC_RTSIG_MAX
6429 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
6430#endif
6431#ifdef _SC_SAVED_IDS
6432 {"SC_SAVED_IDS", _SC_SAVED_IDS},
6433#endif
6434#ifdef _SC_SCHAR_MAX
6435 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
6436#endif
6437#ifdef _SC_SCHAR_MIN
6438 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
6439#endif
6440#ifdef _SC_SELECT
6441 {"SC_SELECT", _SC_SELECT},
6442#endif
6443#ifdef _SC_SEMAPHORES
6444 {"SC_SEMAPHORES", _SC_SEMAPHORES},
6445#endif
6446#ifdef _SC_SEM_NSEMS_MAX
6447 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
6448#endif
6449#ifdef _SC_SEM_VALUE_MAX
6450 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
6451#endif
6452#ifdef _SC_SHARED_MEMORY_OBJECTS
6453 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
6454#endif
6455#ifdef _SC_SHRT_MAX
6456 {"SC_SHRT_MAX", _SC_SHRT_MAX},
6457#endif
6458#ifdef _SC_SHRT_MIN
6459 {"SC_SHRT_MIN", _SC_SHRT_MIN},
6460#endif
6461#ifdef _SC_SIGQUEUE_MAX
6462 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
6463#endif
6464#ifdef _SC_SIGRT_MAX
6465 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
6466#endif
6467#ifdef _SC_SIGRT_MIN
6468 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
6469#endif
Fred Draked86ed291999-12-15 15:34:33 +00006470#ifdef _SC_SOFTPOWER
6471 {"SC_SOFTPOWER", _SC_SOFTPOWER},
6472#endif
Fred Drakec9680921999-12-13 16:37:25 +00006473#ifdef _SC_SPLIT_CACHE
6474 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
6475#endif
6476#ifdef _SC_SSIZE_MAX
6477 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
6478#endif
6479#ifdef _SC_STACK_PROT
6480 {"SC_STACK_PROT", _SC_STACK_PROT},
6481#endif
6482#ifdef _SC_STREAM_MAX
6483 {"SC_STREAM_MAX", _SC_STREAM_MAX},
6484#endif
6485#ifdef _SC_SYNCHRONIZED_IO
6486 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
6487#endif
6488#ifdef _SC_THREADS
6489 {"SC_THREADS", _SC_THREADS},
6490#endif
6491#ifdef _SC_THREAD_ATTR_STACKADDR
6492 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
6493#endif
6494#ifdef _SC_THREAD_ATTR_STACKSIZE
6495 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
6496#endif
6497#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
6498 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
6499#endif
6500#ifdef _SC_THREAD_KEYS_MAX
6501 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
6502#endif
6503#ifdef _SC_THREAD_PRIORITY_SCHEDULING
6504 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
6505#endif
6506#ifdef _SC_THREAD_PRIO_INHERIT
6507 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
6508#endif
6509#ifdef _SC_THREAD_PRIO_PROTECT
6510 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
6511#endif
6512#ifdef _SC_THREAD_PROCESS_SHARED
6513 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
6514#endif
6515#ifdef _SC_THREAD_SAFE_FUNCTIONS
6516 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
6517#endif
6518#ifdef _SC_THREAD_STACK_MIN
6519 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
6520#endif
6521#ifdef _SC_THREAD_THREADS_MAX
6522 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
6523#endif
6524#ifdef _SC_TIMERS
6525 {"SC_TIMERS", _SC_TIMERS},
6526#endif
6527#ifdef _SC_TIMER_MAX
6528 {"SC_TIMER_MAX", _SC_TIMER_MAX},
6529#endif
6530#ifdef _SC_TTY_NAME_MAX
6531 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
6532#endif
6533#ifdef _SC_TZNAME_MAX
6534 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
6535#endif
6536#ifdef _SC_T_IOV_MAX
6537 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
6538#endif
6539#ifdef _SC_UCHAR_MAX
6540 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
6541#endif
6542#ifdef _SC_UINT_MAX
6543 {"SC_UINT_MAX", _SC_UINT_MAX},
6544#endif
6545#ifdef _SC_UIO_MAXIOV
6546 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
6547#endif
6548#ifdef _SC_ULONG_MAX
6549 {"SC_ULONG_MAX", _SC_ULONG_MAX},
6550#endif
6551#ifdef _SC_USHRT_MAX
6552 {"SC_USHRT_MAX", _SC_USHRT_MAX},
6553#endif
6554#ifdef _SC_VERSION
6555 {"SC_VERSION", _SC_VERSION},
6556#endif
6557#ifdef _SC_WORD_BIT
6558 {"SC_WORD_BIT", _SC_WORD_BIT},
6559#endif
6560#ifdef _SC_XBS5_ILP32_OFF32
6561 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
6562#endif
6563#ifdef _SC_XBS5_ILP32_OFFBIG
6564 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
6565#endif
6566#ifdef _SC_XBS5_LP64_OFF64
6567 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
6568#endif
6569#ifdef _SC_XBS5_LPBIG_OFFBIG
6570 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
6571#endif
6572#ifdef _SC_XOPEN_CRYPT
6573 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
6574#endif
6575#ifdef _SC_XOPEN_ENH_I18N
6576 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
6577#endif
6578#ifdef _SC_XOPEN_LEGACY
6579 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
6580#endif
6581#ifdef _SC_XOPEN_REALTIME
6582 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
6583#endif
6584#ifdef _SC_XOPEN_REALTIME_THREADS
6585 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
6586#endif
6587#ifdef _SC_XOPEN_SHM
6588 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
6589#endif
6590#ifdef _SC_XOPEN_UNIX
6591 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
6592#endif
6593#ifdef _SC_XOPEN_VERSION
6594 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
6595#endif
6596#ifdef _SC_XOPEN_XCU_VERSION
6597 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
6598#endif
6599#ifdef _SC_XOPEN_XPG2
6600 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
6601#endif
6602#ifdef _SC_XOPEN_XPG3
6603 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
6604#endif
6605#ifdef _SC_XOPEN_XPG4
6606 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
6607#endif
6608};
6609
6610static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006611conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00006612{
6613 return conv_confname(arg, valuep, posix_constants_sysconf,
6614 sizeof(posix_constants_sysconf)
6615 / sizeof(struct constdef));
6616}
6617
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006618PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006619"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006620Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00006621
6622static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006623posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006624{
6625 PyObject *result = NULL;
6626 int name;
6627
6628 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
6629 int value;
6630
6631 errno = 0;
6632 value = sysconf(name);
6633 if (value == -1 && errno != 0)
6634 posix_error();
6635 else
Christian Heimes217cfd12007-12-02 14:31:20 +00006636 result = PyLong_FromLong(value);
Fred Drakec9680921999-12-13 16:37:25 +00006637 }
6638 return result;
6639}
6640#endif
6641
6642
Fred Drakebec628d1999-12-15 18:31:10 +00006643/* This code is used to ensure that the tables of configuration value names
6644 * are in sorted order as required by conv_confname(), and also to build the
6645 * the exported dictionaries that are used to publish information about the
6646 * names available on the host platform.
6647 *
6648 * Sorting the table at runtime ensures that the table is properly ordered
6649 * when used, even for platforms we're not able to test on. It also makes
6650 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00006651 */
Fred Drakebec628d1999-12-15 18:31:10 +00006652
6653static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006654cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00006655{
6656 const struct constdef *c1 =
6657 (const struct constdef *) v1;
6658 const struct constdef *c2 =
6659 (const struct constdef *) v2;
6660
6661 return strcmp(c1->name, c2->name);
6662}
6663
6664static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006665setup_confname_table(struct constdef *table, size_t tablesize,
Fred Drake4d1e64b2002-04-15 19:40:07 +00006666 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00006667{
Fred Drakebec628d1999-12-15 18:31:10 +00006668 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00006669 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00006670
6671 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
6672 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00006673 if (d == NULL)
6674 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006675
Barry Warsaw3155db32000-04-13 15:20:40 +00006676 for (i=0; i < tablesize; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +00006677 PyObject *o = PyLong_FromLong(table[i].value);
Barry Warsaw3155db32000-04-13 15:20:40 +00006678 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
6679 Py_XDECREF(o);
6680 Py_DECREF(d);
6681 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006682 }
Barry Warsaw3155db32000-04-13 15:20:40 +00006683 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00006684 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00006685 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00006686}
6687
Fred Drakebec628d1999-12-15 18:31:10 +00006688/* Return -1 on failure, 0 on success. */
6689static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00006690setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00006691{
6692#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00006693 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00006694 sizeof(posix_constants_pathconf)
6695 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006696 "pathconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006697 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006698#endif
6699#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00006700 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00006701 sizeof(posix_constants_confstr)
6702 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006703 "confstr_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006704 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006705#endif
6706#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00006707 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00006708 sizeof(posix_constants_sysconf)
6709 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006710 "sysconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006711 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006712#endif
Fred Drakebec628d1999-12-15 18:31:10 +00006713 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00006714}
Fred Draked86ed291999-12-15 15:34:33 +00006715
6716
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006717PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006718"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006719Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006720in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006721
6722static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006723posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006724{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006725 abort();
6726 /*NOTREACHED*/
6727 Py_FatalError("abort() called from Python code didn't abort!");
6728 return NULL;
6729}
Fred Drakebec628d1999-12-15 18:31:10 +00006730
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006731#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006732PyDoc_STRVAR(win32_startfile__doc__,
Georg Brandlf4f44152006-02-18 22:29:33 +00006733"startfile(filepath [, operation]) - Start a file with its associated\n\
6734application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00006735\n\
Georg Brandlf4f44152006-02-18 22:29:33 +00006736When \"operation\" is not specified or \"open\", this acts like\n\
6737double-clicking the file in Explorer, or giving the file name as an\n\
6738argument to the DOS \"start\" command: the file is opened with whatever\n\
6739application (if any) its extension is associated.\n\
6740When another \"operation\" is given, it specifies what should be done with\n\
6741the file. A typical operation is \"print\".\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00006742\n\
6743startfile returns as soon as the associated application is launched.\n\
6744There is no option to wait for the application to close, and no way\n\
6745to retrieve the application's exit status.\n\
6746\n\
6747The filepath is relative to the current directory. If you want to use\n\
6748an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006749the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00006750
6751static PyObject *
6752win32_startfile(PyObject *self, PyObject *args)
6753{
Martin v. Löwis011e8422009-05-05 04:43:17 +00006754 PyObject *ofilepath;
Tim Petersf58a7aa2000-09-22 10:05:54 +00006755 char *filepath;
Georg Brandlf4f44152006-02-18 22:29:33 +00006756 char *operation = NULL;
Tim Petersf58a7aa2000-09-22 10:05:54 +00006757 HINSTANCE rc;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00006758
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00006759 PyObject *unipath, *woperation = NULL;
6760 if (!PyArg_ParseTuple(args, "U|s:startfile",
6761 &unipath, &operation)) {
6762 PyErr_Clear();
6763 goto normal;
6764 }
6765
6766 if (operation) {
6767 woperation = PyUnicode_DecodeASCII(operation,
6768 strlen(operation), NULL);
6769 if (!woperation) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006770 PyErr_Clear();
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00006771 operation = NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006772 goto normal;
6773 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006774 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00006775
6776 Py_BEGIN_ALLOW_THREADS
6777 rc = ShellExecuteW((HWND)0, woperation ? PyUnicode_AS_UNICODE(woperation) : 0,
6778 PyUnicode_AS_UNICODE(unipath),
6779 NULL, NULL, SW_SHOWNORMAL);
6780 Py_END_ALLOW_THREADS
6781
6782 Py_XDECREF(woperation);
6783 if (rc <= (HINSTANCE)32) {
6784 PyObject *errval = win32_error_unicode("startfile",
6785 PyUnicode_AS_UNICODE(unipath));
6786 return errval;
6787 }
6788 Py_INCREF(Py_None);
6789 return Py_None;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006790
6791normal:
Martin v. Löwis011e8422009-05-05 04:43:17 +00006792 if (!PyArg_ParseTuple(args, "O&|s:startfile",
6793 PyUnicode_FSConverter, &ofilepath,
Georg Brandlf4f44152006-02-18 22:29:33 +00006794 &operation))
Tim Petersf58a7aa2000-09-22 10:05:54 +00006795 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00006796 filepath = bytes2str(ofilepath, 1);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006797 Py_BEGIN_ALLOW_THREADS
Georg Brandlf4f44152006-02-18 22:29:33 +00006798 rc = ShellExecute((HWND)0, operation, filepath,
6799 NULL, NULL, SW_SHOWNORMAL);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006800 Py_END_ALLOW_THREADS
Georg Brandle9f8ec92005-09-25 06:16:40 +00006801 if (rc <= (HINSTANCE)32) {
6802 PyObject *errval = win32_error("startfile", filepath);
Martin v. Löwis011e8422009-05-05 04:43:17 +00006803 release_bytes(ofilepath);
Georg Brandle9f8ec92005-09-25 06:16:40 +00006804 return errval;
6805 }
Martin v. Löwis011e8422009-05-05 04:43:17 +00006806 release_bytes(ofilepath);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006807 Py_INCREF(Py_None);
6808 return Py_None;
6809}
6810#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006811
Martin v. Löwis438b5342002-12-27 10:16:42 +00006812#ifdef HAVE_GETLOADAVG
6813PyDoc_STRVAR(posix_getloadavg__doc__,
6814"getloadavg() -> (float, float, float)\n\n\
6815Return the number of processes in the system run queue averaged over\n\
6816the last 1, 5, and 15 minutes or raises OSError if the load average\n\
6817was unobtainable");
6818
6819static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006820posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00006821{
6822 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00006823 if (getloadavg(loadavg, 3)!=3) {
6824 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
6825 return NULL;
6826 } else
6827 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
6828}
6829#endif
6830
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006831#ifdef MS_WINDOWS
6832
6833PyDoc_STRVAR(win32_urandom__doc__,
6834"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00006835Return n random bytes suitable for cryptographic use.");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006836
6837typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
6838 LPCSTR pszContainer, LPCSTR pszProvider, DWORD dwProvType,\
6839 DWORD dwFlags );
6840typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV hProv, DWORD dwLen,\
6841 BYTE *pbBuffer );
6842
6843static CRYPTGENRANDOM pCryptGenRandom = NULL;
Thomas Wouters89d996e2007-09-08 17:39:28 +00006844/* This handle is never explicitly released. Instead, the operating
6845 system will release it when the process terminates. */
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006846static HCRYPTPROV hCryptProv = 0;
6847
Tim Peters4ad82172004-08-30 17:02:04 +00006848static PyObject*
6849win32_urandom(PyObject *self, PyObject *args)
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006850{
Tim Petersd3115382004-08-30 17:36:46 +00006851 int howMany;
6852 PyObject* result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006853
Tim Peters4ad82172004-08-30 17:02:04 +00006854 /* Read arguments */
Tim Peters9b279a82004-08-30 17:10:53 +00006855 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
Tim Peters4ad82172004-08-30 17:02:04 +00006856 return NULL;
Tim Peters51eba612004-08-30 17:08:02 +00006857 if (howMany < 0)
6858 return PyErr_Format(PyExc_ValueError,
6859 "negative argument not allowed");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006860
Tim Peters4ad82172004-08-30 17:02:04 +00006861 if (hCryptProv == 0) {
6862 HINSTANCE hAdvAPI32 = NULL;
6863 CRYPTACQUIRECONTEXTA pCryptAcquireContext = NULL;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006864
Tim Peters4ad82172004-08-30 17:02:04 +00006865 /* Obtain handle to the DLL containing CryptoAPI
6866 This should not fail */
6867 hAdvAPI32 = GetModuleHandle("advapi32.dll");
6868 if(hAdvAPI32 == NULL)
6869 return win32_error("GetModuleHandle", NULL);
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006870
Tim Peters4ad82172004-08-30 17:02:04 +00006871 /* Obtain pointers to the CryptoAPI functions
6872 This will fail on some early versions of Win95 */
6873 pCryptAcquireContext = (CRYPTACQUIRECONTEXTA)GetProcAddress(
6874 hAdvAPI32,
6875 "CryptAcquireContextA");
6876 if (pCryptAcquireContext == NULL)
6877 return PyErr_Format(PyExc_NotImplementedError,
6878 "CryptAcquireContextA not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006879
Tim Peters4ad82172004-08-30 17:02:04 +00006880 pCryptGenRandom = (CRYPTGENRANDOM)GetProcAddress(
6881 hAdvAPI32, "CryptGenRandom");
Thomas Wouters89f507f2006-12-13 04:49:30 +00006882 if (pCryptGenRandom == NULL)
Tim Peters4ad82172004-08-30 17:02:04 +00006883 return PyErr_Format(PyExc_NotImplementedError,
6884 "CryptGenRandom not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006885
Tim Peters4ad82172004-08-30 17:02:04 +00006886 /* Acquire context */
6887 if (! pCryptAcquireContext(&hCryptProv, NULL, NULL,
6888 PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
6889 return win32_error("CryptAcquireContext", NULL);
6890 }
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006891
Tim Peters4ad82172004-08-30 17:02:04 +00006892 /* Allocate bytes */
Christian Heimes72b710a2008-05-26 13:28:38 +00006893 result = PyBytes_FromStringAndSize(NULL, howMany);
Tim Petersd3115382004-08-30 17:36:46 +00006894 if (result != NULL) {
6895 /* Get random data */
Amaury Forgeot d'Arca05ada32008-07-21 21:13:14 +00006896 memset(PyBytes_AS_STRING(result), 0, howMany); /* zero seed */
Tim Petersd3115382004-08-30 17:36:46 +00006897 if (! pCryptGenRandom(hCryptProv, howMany, (unsigned char*)
Christian Heimes72b710a2008-05-26 13:28:38 +00006898 PyBytes_AS_STRING(result))) {
Tim Petersd3115382004-08-30 17:36:46 +00006899 Py_DECREF(result);
6900 return win32_error("CryptGenRandom", NULL);
6901 }
Tim Peters4ad82172004-08-30 17:02:04 +00006902 }
Tim Petersd3115382004-08-30 17:36:46 +00006903 return result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006904}
6905#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00006906
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00006907PyDoc_STRVAR(device_encoding__doc__,
6908"device_encoding(fd) -> str\n\n\
6909Return a string describing the encoding of the device\n\
6910if the output is a terminal; else return None.");
6911
6912static PyObject *
6913device_encoding(PyObject *self, PyObject *args)
6914{
6915 int fd;
6916 if (!PyArg_ParseTuple(args, "i:device_encoding", &fd))
6917 return NULL;
Kristján Valur Jónsson649170b2009-03-24 14:15:49 +00006918 if (!_PyVerify_fd(fd) || !isatty(fd)) {
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00006919 Py_INCREF(Py_None);
6920 return Py_None;
6921 }
6922#if defined(MS_WINDOWS) || defined(MS_WIN64)
6923 if (fd == 0) {
6924 char buf[100];
6925 sprintf(buf, "cp%d", GetConsoleCP());
6926 return PyUnicode_FromString(buf);
6927 }
6928 if (fd == 1 || fd == 2) {
6929 char buf[100];
6930 sprintf(buf, "cp%d", GetConsoleOutputCP());
6931 return PyUnicode_FromString(buf);
6932 }
6933#elif defined(CODESET)
6934 {
6935 char *codeset = nl_langinfo(CODESET);
Mark Dickinsonda2706b2008-12-11 18:03:03 +00006936 if (codeset != NULL && codeset[0] != 0)
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00006937 return PyUnicode_FromString(codeset);
6938 }
6939#endif
6940 Py_INCREF(Py_None);
6941 return Py_None;
6942}
6943
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006944#ifdef __VMS
6945/* Use openssl random routine */
6946#include <openssl/rand.h>
6947PyDoc_STRVAR(vms_urandom__doc__,
6948"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00006949Return n random bytes suitable for cryptographic use.");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006950
6951static PyObject*
6952vms_urandom(PyObject *self, PyObject *args)
6953{
6954 int howMany;
6955 PyObject* result;
6956
6957 /* Read arguments */
6958 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
6959 return NULL;
6960 if (howMany < 0)
6961 return PyErr_Format(PyExc_ValueError,
6962 "negative argument not allowed");
6963
6964 /* Allocate bytes */
Christian Heimes72b710a2008-05-26 13:28:38 +00006965 result = PyBytes_FromStringAndSize(NULL, howMany);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006966 if (result != NULL) {
6967 /* Get random data */
6968 if (RAND_pseudo_bytes((unsigned char*)
Christian Heimes72b710a2008-05-26 13:28:38 +00006969 PyBytes_AS_STRING(result),
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006970 howMany) < 0) {
6971 Py_DECREF(result);
6972 return PyErr_Format(PyExc_ValueError,
6973 "RAND_pseudo_bytes");
6974 }
6975 }
6976 return result;
6977}
6978#endif
6979
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00006980#ifdef HAVE_SETRESUID
6981PyDoc_STRVAR(posix_setresuid__doc__,
6982"setresuid(ruid, euid, suid)\n\n\
6983Set the current process's real, effective, and saved user ids.");
6984
6985static PyObject*
6986posix_setresuid (PyObject *self, PyObject *args)
6987{
6988 /* We assume uid_t is no larger than a long. */
6989 long ruid, euid, suid;
6990 if (!PyArg_ParseTuple(args, "lll", &ruid, &euid, &suid))
6991 return NULL;
6992 if (setresuid(ruid, euid, suid) < 0)
6993 return posix_error();
6994 Py_RETURN_NONE;
6995}
6996#endif
6997
6998#ifdef HAVE_SETRESGID
6999PyDoc_STRVAR(posix_setresgid__doc__,
7000"setresgid(rgid, egid, sgid)\n\n\
7001Set the current process's real, effective, and saved group ids.");
7002
7003static PyObject*
7004posix_setresgid (PyObject *self, PyObject *args)
7005{
7006 /* We assume uid_t is no larger than a long. */
7007 long rgid, egid, sgid;
7008 if (!PyArg_ParseTuple(args, "lll", &rgid, &egid, &sgid))
7009 return NULL;
7010 if (setresgid(rgid, egid, sgid) < 0)
7011 return posix_error();
7012 Py_RETURN_NONE;
7013}
7014#endif
7015
7016#ifdef HAVE_GETRESUID
7017PyDoc_STRVAR(posix_getresuid__doc__,
7018"getresuid() -> (ruid, euid, suid)\n\n\
7019Get tuple of the current process's real, effective, and saved user ids.");
7020
7021static PyObject*
7022posix_getresuid (PyObject *self, PyObject *noargs)
7023{
7024 uid_t ruid, euid, suid;
7025 long l_ruid, l_euid, l_suid;
7026 if (getresuid(&ruid, &euid, &suid) < 0)
7027 return posix_error();
7028 /* Force the values into long's as we don't know the size of uid_t. */
7029 l_ruid = ruid;
7030 l_euid = euid;
7031 l_suid = suid;
7032 return Py_BuildValue("(lll)", l_ruid, l_euid, l_suid);
7033}
7034#endif
7035
7036#ifdef HAVE_GETRESGID
7037PyDoc_STRVAR(posix_getresgid__doc__,
7038"getresgid() -> (rgid, egid, sgid)\n\n\
7039Get tuple of the current process's real, effective, and saved user ids.");
7040
7041static PyObject*
7042posix_getresgid (PyObject *self, PyObject *noargs)
7043{
7044 uid_t rgid, egid, sgid;
7045 long l_rgid, l_egid, l_sgid;
7046 if (getresgid(&rgid, &egid, &sgid) < 0)
7047 return posix_error();
7048 /* Force the values into long's as we don't know the size of uid_t. */
7049 l_rgid = rgid;
7050 l_egid = egid;
7051 l_sgid = sgid;
7052 return Py_BuildValue("(lll)", l_rgid, l_egid, l_sgid);
7053}
7054#endif
7055
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007056static PyMethodDef posix_methods[] = {
7057 {"access", posix_access, METH_VARARGS, posix_access__doc__},
7058#ifdef HAVE_TTYNAME
7059 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
7060#endif
7061 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +00007062#ifdef HAVE_CHFLAGS
7063 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
7064#endif /* HAVE_CHFLAGS */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007065 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +00007066#ifdef HAVE_FCHMOD
7067 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
7068#endif /* HAVE_FCHMOD */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007069#ifdef HAVE_CHOWN
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007070 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007071#endif /* HAVE_CHOWN */
Christian Heimes4e30a842007-11-30 22:12:06 +00007072#ifdef HAVE_LCHMOD
7073 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
7074#endif /* HAVE_LCHMOD */
7075#ifdef HAVE_FCHOWN
7076 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
7077#endif /* HAVE_FCHOWN */
Thomas Wouterscf297e42007-02-23 15:07:44 +00007078#ifdef HAVE_LCHFLAGS
7079 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
7080#endif /* HAVE_LCHFLAGS */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00007081#ifdef HAVE_LCHOWN
7082 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
7083#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +00007084#ifdef HAVE_CHROOT
7085 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
7086#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007087#ifdef HAVE_CTERMID
Neal Norwitze241ce82003-02-17 18:17:05 +00007088 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007089#endif
Guido van Rossum36bc6801995-06-14 22:54:23 +00007090#ifdef HAVE_GETCWD
Guido van Rossumf0af3e32008-10-02 18:55:37 +00007091 {"getcwd", (PyCFunction)posix_getcwd_unicode,
7092 METH_NOARGS, posix_getcwd__doc__},
7093 {"getcwdb", (PyCFunction)posix_getcwd_bytes,
7094 METH_NOARGS, posix_getcwdb__doc__},
Guido van Rossum36bc6801995-06-14 22:54:23 +00007095#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00007096#ifdef HAVE_LINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007097 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007098#endif /* HAVE_LINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007099 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
7100 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
7101 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007102#ifdef HAVE_NICE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007103 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007104#endif /* HAVE_NICE */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007105#ifdef HAVE_READLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007106 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007107#endif /* HAVE_READLINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007108 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
7109 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
7110 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00007111 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007112#ifdef HAVE_SYMLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007113 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007114#endif /* HAVE_SYMLINK */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007115#ifdef HAVE_SYSTEM
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007116 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007117#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007118 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007119#ifdef HAVE_UNAME
Neal Norwitze241ce82003-02-17 18:17:05 +00007120 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007121#endif /* HAVE_UNAME */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007122 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
7123 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
7124 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007125#ifdef HAVE_TIMES
Neal Norwitze241ce82003-02-17 18:17:05 +00007126 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007127#endif /* HAVE_TIMES */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007128 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007129#ifdef HAVE_EXECV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007130 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
7131 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007132#endif /* HAVE_EXECV */
Guido van Rossuma1065681999-01-25 23:20:23 +00007133#ifdef HAVE_SPAWNV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007134 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
7135 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00007136#if defined(PYOS_OS2)
7137 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
7138 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
7139#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00007140#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +00007141#ifdef HAVE_FORK1
Neal Norwitze241ce82003-02-17 18:17:05 +00007142 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +00007143#endif /* HAVE_FORK1 */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007144#ifdef HAVE_FORK
Neal Norwitze241ce82003-02-17 18:17:05 +00007145 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007146#endif /* HAVE_FORK */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00007147#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Neal Norwitze241ce82003-02-17 18:17:05 +00007148 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +00007149#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00007150#ifdef HAVE_FORKPTY
Neal Norwitze241ce82003-02-17 18:17:05 +00007151 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +00007152#endif /* HAVE_FORKPTY */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007153#ifdef HAVE_GETEGID
Neal Norwitze241ce82003-02-17 18:17:05 +00007154 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007155#endif /* HAVE_GETEGID */
7156#ifdef HAVE_GETEUID
Neal Norwitze241ce82003-02-17 18:17:05 +00007157 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007158#endif /* HAVE_GETEUID */
7159#ifdef HAVE_GETGID
Neal Norwitze241ce82003-02-17 18:17:05 +00007160 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007161#endif /* HAVE_GETGID */
Fred Drakec9680921999-12-13 16:37:25 +00007162#ifdef HAVE_GETGROUPS
Neal Norwitze241ce82003-02-17 18:17:05 +00007163 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00007164#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00007165 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00007166#ifdef HAVE_GETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00007167 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007168#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007169#ifdef HAVE_GETPPID
Neal Norwitze241ce82003-02-17 18:17:05 +00007170 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007171#endif /* HAVE_GETPPID */
7172#ifdef HAVE_GETUID
Neal Norwitze241ce82003-02-17 18:17:05 +00007173 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007174#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +00007175#ifdef HAVE_GETLOGIN
Neal Norwitze241ce82003-02-17 18:17:05 +00007176 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +00007177#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +00007178#ifdef HAVE_KILL
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007179 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007180#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00007181#ifdef HAVE_KILLPG
7182 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
7183#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +00007184#ifdef HAVE_PLOCK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007185 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +00007186#endif /* HAVE_PLOCK */
Thomas Heller8b7a9572007-08-31 06:44:36 +00007187#ifdef MS_WINDOWS
7188 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
7189#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00007190#ifdef HAVE_SETUID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007191 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007192#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00007193#ifdef HAVE_SETEUID
7194 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
7195#endif /* HAVE_SETEUID */
7196#ifdef HAVE_SETEGID
7197 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
7198#endif /* HAVE_SETEGID */
7199#ifdef HAVE_SETREUID
7200 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
7201#endif /* HAVE_SETREUID */
7202#ifdef HAVE_SETREGID
7203 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
7204#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007205#ifdef HAVE_SETGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007206 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007207#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00007208#ifdef HAVE_SETGROUPS
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00007209 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00007210#endif /* HAVE_SETGROUPS */
Antoine Pitroub7572f02009-12-02 20:46:48 +00007211#ifdef HAVE_INITGROUPS
7212 {"initgroups", posix_initgroups, METH_VARARGS, posix_initgroups__doc__},
7213#endif /* HAVE_INITGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +00007214#ifdef HAVE_GETPGID
7215 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
7216#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007217#ifdef HAVE_SETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00007218 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007219#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00007220#ifdef HAVE_WAIT
Neal Norwitze241ce82003-02-17 18:17:05 +00007221 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00007222#endif /* HAVE_WAIT */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007223#ifdef HAVE_WAIT3
7224 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
7225#endif /* HAVE_WAIT3 */
7226#ifdef HAVE_WAIT4
7227 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
7228#endif /* HAVE_WAIT4 */
Tim Petersab034fa2002-02-01 11:27:43 +00007229#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007230 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007231#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00007232#ifdef HAVE_GETSID
7233 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
7234#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007235#ifdef HAVE_SETSID
Neal Norwitze241ce82003-02-17 18:17:05 +00007236 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007237#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007238#ifdef HAVE_SETPGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007239 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007240#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007241#ifdef HAVE_TCGETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007242 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007243#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007244#ifdef HAVE_TCSETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007245 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007246#endif /* HAVE_TCSETPGRP */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007247 {"open", posix_open, METH_VARARGS, posix_open__doc__},
7248 {"close", posix_close, METH_VARARGS, posix_close__doc__},
Christian Heimesfdab48e2008-01-20 09:06:41 +00007249 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00007250 {"device_encoding", device_encoding, METH_VARARGS, device_encoding__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007251 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
7252 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
7253 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
7254 {"read", posix_read, METH_VARARGS, posix_read__doc__},
7255 {"write", posix_write, METH_VARARGS, posix_write__doc__},
7256 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
Skip Montanaro1517d842000-07-19 14:34:14 +00007257 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007258#ifdef HAVE_PIPE
Neal Norwitze241ce82003-02-17 18:17:05 +00007259 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007260#endif
7261#ifdef HAVE_MKFIFO
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007262 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007263#endif
Neal Norwitz11690112002-07-30 01:08:28 +00007264#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007265 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
7266#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007267#ifdef HAVE_DEVICE_MACROS
7268 {"major", posix_major, METH_VARARGS, posix_major__doc__},
7269 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
7270 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
7271#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007272#ifdef HAVE_FTRUNCATE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007273 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007274#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007275#ifdef HAVE_PUTENV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007276 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007277#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00007278#ifdef HAVE_UNSETENV
7279 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
7280#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007281 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +00007282#ifdef HAVE_FCHDIR
7283 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
7284#endif
Guido van Rossum21142a01999-01-08 21:05:37 +00007285#ifdef HAVE_FSYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007286 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007287#endif
7288#ifdef HAVE_FDATASYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007289 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007290#endif
Guido van Rossumc9641791998-08-04 15:26:23 +00007291#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +00007292#ifdef WCOREDUMP
7293 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
7294#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00007295#ifdef WIFCONTINUED
7296 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
7297#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +00007298#ifdef WIFSTOPPED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007299 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007300#endif /* WIFSTOPPED */
7301#ifdef WIFSIGNALED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007302 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007303#endif /* WIFSIGNALED */
7304#ifdef WIFEXITED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007305 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007306#endif /* WIFEXITED */
7307#ifdef WEXITSTATUS
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007308 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007309#endif /* WEXITSTATUS */
7310#ifdef WTERMSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007311 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007312#endif /* WTERMSIG */
7313#ifdef WSTOPSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007314 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007315#endif /* WSTOPSIG */
7316#endif /* HAVE_SYS_WAIT_H */
Thomas Wouters477c8d52006-05-27 19:21:47 +00007317#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007318 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007319#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00007320#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007321 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007322#endif
Fred Drakec9680921999-12-13 16:37:25 +00007323#ifdef HAVE_CONFSTR
7324 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
7325#endif
7326#ifdef HAVE_SYSCONF
7327 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
7328#endif
7329#ifdef HAVE_FPATHCONF
7330 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
7331#endif
7332#ifdef HAVE_PATHCONF
7333 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
7334#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00007335 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007336#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00007337 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
7338#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00007339#ifdef HAVE_GETLOADAVG
Neal Norwitze241ce82003-02-17 18:17:05 +00007340 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +00007341#endif
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00007342 #ifdef MS_WINDOWS
7343 {"urandom", win32_urandom, METH_VARARGS, win32_urandom__doc__},
7344 #endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007345 #ifdef __VMS
7346 {"urandom", vms_urandom, METH_VARARGS, vms_urandom__doc__},
7347 #endif
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00007348#ifdef HAVE_SETRESUID
7349 {"setresuid", posix_setresuid, METH_VARARGS, posix_setresuid__doc__},
7350#endif
7351#ifdef HAVE_SETRESGID
7352 {"setresgid", posix_setresgid, METH_VARARGS, posix_setresgid__doc__},
7353#endif
7354#ifdef HAVE_GETRESUID
7355 {"getresuid", posix_getresuid, METH_NOARGS, posix_getresuid__doc__},
7356#endif
7357#ifdef HAVE_GETRESGID
7358 {"getresgid", posix_getresgid, METH_NOARGS, posix_getresgid__doc__},
7359#endif
7360
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007361 {NULL, NULL} /* Sentinel */
7362};
7363
7364
Barry Warsaw4a342091996-12-19 23:50:02 +00007365static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00007366ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +00007367{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007368 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +00007369}
7370
Guido van Rossumd48f2521997-12-05 22:19:34 +00007371#if defined(PYOS_OS2)
7372/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007373static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007374{
7375 APIRET rc;
7376 ULONG values[QSV_MAX+1];
7377 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +00007378 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +00007379
7380 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +00007381 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007382 Py_END_ALLOW_THREADS
7383
7384 if (rc != NO_ERROR) {
7385 os2_error(rc);
7386 return -1;
7387 }
7388
Fred Drake4d1e64b2002-04-15 19:40:07 +00007389 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
7390 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
7391 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
7392 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
7393 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
7394 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
7395 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007396
7397 switch (values[QSV_VERSION_MINOR]) {
7398 case 0: ver = "2.00"; break;
7399 case 10: ver = "2.10"; break;
7400 case 11: ver = "2.11"; break;
7401 case 30: ver = "3.00"; break;
7402 case 40: ver = "4.00"; break;
7403 case 50: ver = "5.00"; break;
7404 default:
Tim Peters885d4572001-11-28 20:27:42 +00007405 PyOS_snprintf(tmp, sizeof(tmp),
7406 "%d-%d", values[QSV_VERSION_MAJOR],
7407 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007408 ver = &tmp[0];
7409 }
7410
7411 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007412 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007413 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007414
7415 /* Add Indicator of Which Drive was Used to Boot the System */
7416 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
7417 tmp[1] = ':';
7418 tmp[2] = '\0';
7419
Fred Drake4d1e64b2002-04-15 19:40:07 +00007420 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007421}
7422#endif
7423
Barry Warsaw4a342091996-12-19 23:50:02 +00007424static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007425all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +00007426{
Guido van Rossum94f6f721999-01-06 18:42:14 +00007427#ifdef F_OK
7428 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007429#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007430#ifdef R_OK
7431 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007432#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007433#ifdef W_OK
7434 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007435#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007436#ifdef X_OK
7437 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007438#endif
Fred Drakec9680921999-12-13 16:37:25 +00007439#ifdef NGROUPS_MAX
7440 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
7441#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007442#ifdef TMP_MAX
7443 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
7444#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007445#ifdef WCONTINUED
7446 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
7447#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007448#ifdef WNOHANG
7449 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007450#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007451#ifdef WUNTRACED
7452 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
7453#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007454#ifdef O_RDONLY
7455 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
7456#endif
7457#ifdef O_WRONLY
7458 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
7459#endif
7460#ifdef O_RDWR
7461 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
7462#endif
7463#ifdef O_NDELAY
7464 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
7465#endif
7466#ifdef O_NONBLOCK
7467 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
7468#endif
7469#ifdef O_APPEND
7470 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
7471#endif
7472#ifdef O_DSYNC
7473 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
7474#endif
7475#ifdef O_RSYNC
7476 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
7477#endif
7478#ifdef O_SYNC
7479 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
7480#endif
7481#ifdef O_NOCTTY
7482 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
7483#endif
7484#ifdef O_CREAT
7485 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
7486#endif
7487#ifdef O_EXCL
7488 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
7489#endif
7490#ifdef O_TRUNC
7491 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
7492#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +00007493#ifdef O_BINARY
7494 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
7495#endif
7496#ifdef O_TEXT
7497 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
7498#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007499#ifdef O_LARGEFILE
7500 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
7501#endif
Skip Montanaro5ff14922005-05-16 02:42:22 +00007502#ifdef O_SHLOCK
7503 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
7504#endif
7505#ifdef O_EXLOCK
7506 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
7507#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007508
Tim Peters5aa91602002-01-30 05:46:57 +00007509/* MS Windows */
7510#ifdef O_NOINHERIT
7511 /* Don't inherit in child processes. */
7512 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
7513#endif
7514#ifdef _O_SHORT_LIVED
7515 /* Optimize for short life (keep in memory). */
7516 /* MS forgot to define this one with a non-underscore form too. */
7517 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
7518#endif
7519#ifdef O_TEMPORARY
7520 /* Automatically delete when last handle is closed. */
7521 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
7522#endif
7523#ifdef O_RANDOM
7524 /* Optimize for random access. */
7525 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
7526#endif
7527#ifdef O_SEQUENTIAL
7528 /* Optimize for sequential access. */
7529 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
7530#endif
7531
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007532/* GNU extensions. */
Alexandre Vassalottibee32532008-05-16 18:15:12 +00007533#ifdef O_ASYNC
7534 /* Send a SIGIO signal whenever input or output
7535 becomes available on file descriptor */
7536 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
7537#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007538#ifdef O_DIRECT
7539 /* Direct disk access. */
7540 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
7541#endif
7542#ifdef O_DIRECTORY
7543 /* Must be a directory. */
7544 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
7545#endif
7546#ifdef O_NOFOLLOW
7547 /* Do not follow links. */
7548 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
7549#endif
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +00007550#ifdef O_NOATIME
7551 /* Do not update the access time. */
7552 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
7553#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00007554
Barry Warsaw5676bd12003-01-07 20:57:09 +00007555 /* These come from sysexits.h */
7556#ifdef EX_OK
7557 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007558#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007559#ifdef EX_USAGE
7560 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007561#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007562#ifdef EX_DATAERR
7563 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007564#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007565#ifdef EX_NOINPUT
7566 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007567#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007568#ifdef EX_NOUSER
7569 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007570#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007571#ifdef EX_NOHOST
7572 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007573#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007574#ifdef EX_UNAVAILABLE
7575 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007576#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007577#ifdef EX_SOFTWARE
7578 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007579#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007580#ifdef EX_OSERR
7581 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007582#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007583#ifdef EX_OSFILE
7584 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007585#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007586#ifdef EX_CANTCREAT
7587 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007588#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007589#ifdef EX_IOERR
7590 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007591#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007592#ifdef EX_TEMPFAIL
7593 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007594#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007595#ifdef EX_PROTOCOL
7596 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007597#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007598#ifdef EX_NOPERM
7599 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007600#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007601#ifdef EX_CONFIG
7602 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007603#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007604#ifdef EX_NOTFOUND
7605 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007606#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007607
Guido van Rossum246bc171999-02-01 23:54:31 +00007608#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007609#if defined(PYOS_OS2) && defined(PYCC_GCC)
7610 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
7611 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
7612 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
7613 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
7614 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
7615 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
7616 if (ins(d, "P_PM", (long)P_PM)) return -1;
7617 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
7618 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
7619 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
7620 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
7621 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
7622 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
7623 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
7624 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
7625 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
7626 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
7627 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
7628 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
7629 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
7630#else
Guido van Rossum7d385291999-02-16 19:38:04 +00007631 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
7632 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
7633 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
7634 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
7635 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +00007636#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007637#endif
Guido van Rossum246bc171999-02-01 23:54:31 +00007638
Guido van Rossumd48f2521997-12-05 22:19:34 +00007639#if defined(PYOS_OS2)
7640 if (insertvalues(d)) return -1;
7641#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007642 return 0;
7643}
7644
7645
Tim Peters5aa91602002-01-30 05:46:57 +00007646#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007647#define INITFUNC PyInit_nt
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007648#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007649
7650#elif defined(PYOS_OS2)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007651#define INITFUNC PyInit_os2
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007652#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007653
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007654#else
Martin v. Löwis1a214512008-06-11 05:26:20 +00007655#define INITFUNC PyInit_posix
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007656#define MODNAME "posix"
7657#endif
7658
Martin v. Löwis1a214512008-06-11 05:26:20 +00007659static struct PyModuleDef posixmodule = {
7660 PyModuleDef_HEAD_INIT,
7661 MODNAME,
7662 posix__doc__,
7663 -1,
7664 posix_methods,
7665 NULL,
7666 NULL,
7667 NULL,
7668 NULL
7669};
7670
7671
Mark Hammondfe51c6d2002-08-02 02:27:13 +00007672PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007673INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +00007674{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007675 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +00007676
Martin v. Löwis1a214512008-06-11 05:26:20 +00007677 m = PyModule_Create(&posixmodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00007678 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007679 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007680
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007681 /* Initialize environ dictionary */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007682 v = convertenviron();
Fred Drake4d1e64b2002-04-15 19:40:07 +00007683 Py_XINCREF(v);
7684 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007685 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00007686 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +00007687
Fred Drake4d1e64b2002-04-15 19:40:07 +00007688 if (all_ins(m))
Martin v. Löwis1a214512008-06-11 05:26:20 +00007689 return NULL;
Barry Warsaw4a342091996-12-19 23:50:02 +00007690
Fred Drake4d1e64b2002-04-15 19:40:07 +00007691 if (setup_confname_tables(m))
Martin v. Löwis1a214512008-06-11 05:26:20 +00007692 return NULL;
Fred Drakebec628d1999-12-15 18:31:10 +00007693
Fred Drake4d1e64b2002-04-15 19:40:07 +00007694 Py_INCREF(PyExc_OSError);
7695 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +00007696
Guido van Rossumb3d39562000-01-31 18:41:26 +00007697#ifdef HAVE_PUTENV
Neil Schemenauer19030a02001-01-16 04:27:47 +00007698 if (posix_putenv_garbage == NULL)
7699 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +00007700#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007701
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007702 if (!initialized) {
7703 stat_result_desc.name = MODNAME ".stat_result";
7704 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
7705 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
7706 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
7707 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
7708 structseq_new = StatResultType.tp_new;
7709 StatResultType.tp_new = statresult_new;
7710
7711 statvfs_result_desc.name = MODNAME ".statvfs_result";
7712 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00007713#ifdef NEED_TICKS_PER_SECOND
7714# if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
7715 ticks_per_second = sysconf(_SC_CLK_TCK);
7716# elif defined(HZ)
7717 ticks_per_second = HZ;
7718# else
7719 ticks_per_second = 60; /* magic fallback value; may be bogus */
7720# endif
7721#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007722 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00007723 Py_INCREF((PyObject*) &StatResultType);
7724 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
Fred Drake4d1e64b2002-04-15 19:40:07 +00007725 Py_INCREF((PyObject*) &StatVFSResultType);
7726 PyModule_AddObject(m, "statvfs_result",
7727 (PyObject*) &StatVFSResultType);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007728 initialized = 1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007729
7730#ifdef __APPLE__
7731 /*
7732 * Step 2 of weak-linking support on Mac OS X.
7733 *
7734 * The code below removes functions that are not available on the
7735 * currently active platform.
7736 *
7737 * This block allow one to use a python binary that was build on
7738 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
7739 * OSX 10.4.
7740 */
7741#ifdef HAVE_FSTATVFS
7742 if (fstatvfs == NULL) {
7743 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007744 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007745 }
7746 }
7747#endif /* HAVE_FSTATVFS */
7748
7749#ifdef HAVE_STATVFS
7750 if (statvfs == NULL) {
7751 if (PyObject_DelAttrString(m, "statvfs") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007752 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007753 }
7754 }
7755#endif /* HAVE_STATVFS */
7756
7757# ifdef HAVE_LCHOWN
7758 if (lchown == NULL) {
7759 if (PyObject_DelAttrString(m, "lchown") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007760 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007761 }
7762 }
7763#endif /* HAVE_LCHOWN */
7764
7765
7766#endif /* __APPLE__ */
Martin v. Löwis1a214512008-06-11 05:26:20 +00007767 return m;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007768
Guido van Rossumb6775db1994-08-01 11:34:53 +00007769}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007770
7771#ifdef __cplusplus
7772}
7773#endif