blob: bc17b6b049be0527277029d30e72e9dbb90a65b4 [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"
Guido van Rossumb6775db1994-08-01 11:34:53 +0000266#include <windows.h>
Tim Petersee66d0c2002-07-15 16:10:55 +0000267#include <shellapi.h> /* for ShellExecute() */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000268#endif /* _MSC_VER */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000269
Guido van Rossumd48f2521997-12-05 22:19:34 +0000270#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000271#include <io.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000272#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000273
Tim Petersbc2e10e2002-03-03 23:17:02 +0000274#ifndef MAXPATHLEN
Thomas Wouters477c8d52006-05-27 19:21:47 +0000275#if defined(PATH_MAX) && PATH_MAX > 1024
276#define MAXPATHLEN PATH_MAX
277#else
Tim Petersbc2e10e2002-03-03 23:17:02 +0000278#define MAXPATHLEN 1024
Thomas Wouters477c8d52006-05-27 19:21:47 +0000279#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000280#endif /* MAXPATHLEN */
281
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000282#ifdef UNION_WAIT
283/* Emulate some macros on systems that have a union instead of macros */
284
285#ifndef WIFEXITED
286#define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
287#endif
288
289#ifndef WEXITSTATUS
290#define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
291#endif
292
293#ifndef WTERMSIG
294#define WTERMSIG(u_wait) ((u_wait).w_termsig)
295#endif
296
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000297#define WAIT_TYPE union wait
298#define WAIT_STATUS_INT(s) (s.w_status)
299
300#else /* !UNION_WAIT */
301#define WAIT_TYPE int
302#define WAIT_STATUS_INT(s) (s)
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000303#endif /* UNION_WAIT */
304
Greg Wardb48bc172000-03-01 21:51:56 +0000305/* Don't use the "_r" form if we don't need it (also, won't have a
306 prototype for it, at least on Solaris -- maybe others as well?). */
307#if defined(HAVE_CTERMID_R) && defined(WITH_THREAD)
308#define USE_CTERMID_R
309#endif
310
Fred Drake699f3522000-06-29 21:12:41 +0000311/* choose the appropriate stat and fstat functions and return structs */
Guido van Rossum64529cd2000-06-30 22:45:12 +0000312#undef STAT
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000313#if defined(MS_WIN64) || defined(MS_WINDOWS)
Martin v. Löwis14694662006-02-03 12:54:16 +0000314# define STAT win32_stat
315# define FSTAT win32_fstat
316# define STRUCT_STAT struct win32_stat
Fred Drake699f3522000-06-29 21:12:41 +0000317#else
318# define STAT stat
319# define FSTAT fstat
320# define STRUCT_STAT struct stat
321#endif
322
Tim Peters11b23062003-04-23 02:39:17 +0000323#if defined(MAJOR_IN_MKDEV)
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000324#include <sys/mkdev.h>
325#else
326#if defined(MAJOR_IN_SYSMACROS)
327#include <sys/sysmacros.h>
328#endif
Neal Norwitz3d949422002-04-20 13:46:43 +0000329#if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
330#include <sys/mkdev.h>
331#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000332#endif
Fred Drake699f3522000-06-29 21:12:41 +0000333
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000334/* Return a dictionary corresponding to the POSIX environment table */
Jack Jansenea0c3822002-08-01 21:57:49 +0000335#ifdef WITH_NEXT_FRAMEWORK
336/* On Darwin/MacOSX a shared library or framework has no access to
337** environ directly, we must obtain it with _NSGetEnviron().
338*/
339#include <crt_externs.h>
340static char **environ;
341#elif !defined(_MSC_VER) && ( !defined(__WATCOMC__) || defined(__QNX__) )
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000342extern char **environ;
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000343#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000344
Barry Warsaw53699e91996-12-10 23:23:01 +0000345static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000346convertenviron(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000347{
Barry Warsaw53699e91996-12-10 23:23:01 +0000348 PyObject *d;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000349#ifdef MS_WINDOWS
350 wchar_t **e;
351#else
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000352 char **e;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000353#endif
Barry Warsaw53699e91996-12-10 23:23:01 +0000354 d = PyDict_New();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000355 if (d == NULL)
356 return NULL;
Jack Jansenea0c3822002-08-01 21:57:49 +0000357#ifdef WITH_NEXT_FRAMEWORK
358 if (environ == NULL)
359 environ = *_NSGetEnviron();
360#endif
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000361#ifdef MS_WINDOWS
362 /* _wenviron must be initialized in this way if the program is started
363 through main() instead of wmain(). */
364 _wgetenv(L"");
365 if (_wenviron == NULL)
366 return d;
367 /* This part ignores errors */
368 for (e = _wenviron; *e != NULL; e++) {
369 PyObject *k;
370 PyObject *v;
371 wchar_t *p = wcschr(*e, L'=');
372 if (p == NULL)
373 continue;
374 k = PyUnicode_FromWideChar(*e, (Py_ssize_t)(p-*e));
375 if (k == NULL) {
376 PyErr_Clear();
377 continue;
378 }
379 v = PyUnicode_FromWideChar(p+1, wcslen(p+1));
380 if (v == NULL) {
381 PyErr_Clear();
382 Py_DECREF(k);
383 continue;
384 }
385 if (PyDict_GetItem(d, k) == NULL) {
386 if (PyDict_SetItem(d, k, v) != 0)
387 PyErr_Clear();
388 }
389 Py_DECREF(k);
390 Py_DECREF(v);
391 }
392#else
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000393 if (environ == NULL)
394 return d;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000395 /* This part ignores errors */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000396 for (e = environ; *e != NULL; e++) {
Guido van Rossum6a619f41999-08-03 19:41:10 +0000397 PyObject *k;
Barry Warsaw53699e91996-12-10 23:23:01 +0000398 PyObject *v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000399 char *p = strchr(*e, '=');
400 if (p == NULL)
401 continue;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000402 k = PyUnicode_FromStringAndSize(*e, (int)(p-*e));
Guido van Rossum6a619f41999-08-03 19:41:10 +0000403 if (k == NULL) {
404 PyErr_Clear();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000405 continue;
Guido van Rossum6a619f41999-08-03 19:41:10 +0000406 }
Guido van Rossum98297ee2007-11-06 21:34:58 +0000407 v = PyUnicode_FromString(p+1);
Guido van Rossum6a619f41999-08-03 19:41:10 +0000408 if (v == NULL) {
409 PyErr_Clear();
410 Py_DECREF(k);
411 continue;
412 }
413 if (PyDict_GetItem(d, k) == NULL) {
414 if (PyDict_SetItem(d, k, v) != 0)
415 PyErr_Clear();
416 }
417 Py_DECREF(k);
Barry Warsaw53699e91996-12-10 23:23:01 +0000418 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000419 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000420#endif
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +0000421#if defined(PYOS_OS2)
Guido van Rossumd48f2521997-12-05 22:19:34 +0000422 {
423 APIRET rc;
424 char buffer[1024]; /* OS/2 Provides a Documented Max of 1024 Chars */
425
426 rc = DosQueryExtLIBPATH(buffer, BEGIN_LIBPATH);
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000427 if (rc == NO_ERROR) { /* (not a type, envname is NOT 'BEGIN_LIBPATH') */
Christian Heimes72b710a2008-05-26 13:28:38 +0000428 PyObject *v = PyBytes_FromString(buffer);
Neal Norwitz93c56822007-08-26 07:10:06 +0000429 PyDict_SetItemString(d, "BEGINLIBPATH", v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000430 Py_DECREF(v);
431 }
432 rc = DosQueryExtLIBPATH(buffer, END_LIBPATH);
433 if (rc == NO_ERROR) { /* (not a typo, envname is NOT 'END_LIBPATH') */
Christian Heimes72b710a2008-05-26 13:28:38 +0000434 PyObject *v = PyBytes_FromString(buffer);
Neal Norwitz93c56822007-08-26 07:10:06 +0000435 PyDict_SetItemString(d, "ENDLIBPATH", v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000436 Py_DECREF(v);
437 }
438 }
439#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000440 return d;
441}
442
443
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000444/* Set a POSIX-specific error from errno, and return NULL */
445
Barry Warsawd58d7641998-07-23 16:14:40 +0000446static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000447posix_error(void)
Guido van Rossumad0ee831995-03-01 10:34:45 +0000448{
Barry Warsawca74da41999-02-09 19:31:45 +0000449 return PyErr_SetFromErrno(PyExc_OSError);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000450}
Barry Warsawd58d7641998-07-23 16:14:40 +0000451static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000452posix_error_with_filename(char* name)
Barry Warsawd58d7641998-07-23 16:14:40 +0000453{
Barry Warsawca74da41999-02-09 19:31:45 +0000454 return PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
Barry Warsawd58d7641998-07-23 16:14:40 +0000455}
456
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000457#ifdef Py_WIN_WIDE_FILENAMES
458static PyObject *
459posix_error_with_unicode_filename(Py_UNICODE* name)
460{
461 return PyErr_SetFromErrnoWithUnicodeFilename(PyExc_OSError, name);
462}
463#endif /* Py_WIN_WIDE_FILENAMES */
464
465
Mark Hammondef8b6542001-05-13 08:04:26 +0000466static PyObject *
467posix_error_with_allocated_filename(char* name)
468{
469 PyObject *rc = PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
470 PyMem_Free(name);
471 return rc;
472}
473
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000474#ifdef MS_WINDOWS
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000475static PyObject *
476win32_error(char* function, char* filename)
477{
Mark Hammond33a6da92000-08-15 00:46:38 +0000478 /* XXX We should pass the function name along in the future.
Georg Brandl38feaf02008-05-25 07:45:51 +0000479 (winreg.c also wants to pass the function name.)
Tim Peters5aa91602002-01-30 05:46:57 +0000480 This would however require an additional param to the
Mark Hammond33a6da92000-08-15 00:46:38 +0000481 Windows error object, which is non-trivial.
482 */
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000483 errno = GetLastError();
484 if (filename)
Mark Hammond33a6da92000-08-15 00:46:38 +0000485 return PyErr_SetFromWindowsErrWithFilename(errno, filename);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000486 else
Mark Hammond33a6da92000-08-15 00:46:38 +0000487 return PyErr_SetFromWindowsErr(errno);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000488}
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000489
490#ifdef Py_WIN_WIDE_FILENAMES
491static PyObject *
492win32_error_unicode(char* function, Py_UNICODE* filename)
493{
494 /* XXX - see win32_error for comments on 'function' */
495 errno = GetLastError();
496 if (filename)
497 return PyErr_SetFromWindowsErrWithUnicodeFilename(errno, filename);
498 else
499 return PyErr_SetFromWindowsErr(errno);
500}
501
502static PyObject *_PyUnicode_FromFileSystemEncodedObject(register PyObject *obj)
503{
Thomas Wouters477c8d52006-05-27 19:21:47 +0000504}
505
506/* Function suitable for O& conversion */
507static int
508convert_to_unicode(PyObject *arg, void* _param)
509{
510 PyObject **param = (PyObject**)_param;
511 if (PyUnicode_CheckExact(arg)) {
512 Py_INCREF(arg);
513 *param = arg;
514 }
515 else if (PyUnicode_Check(arg)) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000516 /* For a Unicode subtype that's not a Unicode object,
517 return a true Unicode object with the same data. */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000518 *param = PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(arg),
519 PyUnicode_GET_SIZE(arg));
520 return *param != NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000521 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000522 else
523 *param = PyUnicode_FromEncodedObject(arg,
524 Py_FileSystemDefaultEncoding,
525 "strict");
526 return (*param) != NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000527}
528
529#endif /* Py_WIN_WIDE_FILENAMES */
530
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000531#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000532
Guido van Rossumd48f2521997-12-05 22:19:34 +0000533#if defined(PYOS_OS2)
534/**********************************************************************
535 * Helper Function to Trim and Format OS/2 Messages
536 **********************************************************************/
537 static void
538os2_formatmsg(char *msgbuf, int msglen, char *reason)
539{
540 msgbuf[msglen] = '\0'; /* OS/2 Doesn't Guarantee a Terminator */
541
542 if (strlen(msgbuf) > 0) { /* If Non-Empty Msg, Trim CRLF */
543 char *lastc = &msgbuf[ strlen(msgbuf)-1 ];
544
Neal Norwitz30b5c5d2005-12-19 06:05:18 +0000545 while (lastc > msgbuf && isspace(Py_CHARMASK(*lastc)))
Guido van Rossumd48f2521997-12-05 22:19:34 +0000546 *lastc-- = '\0'; /* Trim Trailing Whitespace (CRLF) */
547 }
548
549 /* Add Optional Reason Text */
550 if (reason) {
551 strcat(msgbuf, " : ");
552 strcat(msgbuf, reason);
553 }
554}
555
556/**********************************************************************
557 * Decode an OS/2 Operating System Error Code
558 *
559 * A convenience function to lookup an OS/2 error code and return a
560 * text message we can use to raise a Python exception.
561 *
562 * Notes:
563 * The messages for errors returned from the OS/2 kernel reside in
564 * the file OSO001.MSG in the \OS2 directory hierarchy.
565 *
566 **********************************************************************/
567 static char *
568os2_strerror(char *msgbuf, int msgbuflen, int errorcode, char *reason)
569{
570 APIRET rc;
571 ULONG msglen;
572
573 /* Retrieve Kernel-Related Error Message from OSO001.MSG File */
574 Py_BEGIN_ALLOW_THREADS
575 rc = DosGetMessage(NULL, 0, msgbuf, msgbuflen,
576 errorcode, "oso001.msg", &msglen);
577 Py_END_ALLOW_THREADS
578
579 if (rc == NO_ERROR)
580 os2_formatmsg(msgbuf, msglen, reason);
581 else
Tim Peters1ceb5fb2001-11-28 20:32:57 +0000582 PyOS_snprintf(msgbuf, msgbuflen,
Tim Peters885d4572001-11-28 20:27:42 +0000583 "unknown OS error #%d", errorcode);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000584
585 return msgbuf;
586}
587
588/* Set an OS/2-specific error and return NULL. OS/2 kernel
589 errors are not in a global variable e.g. 'errno' nor are
590 they congruent with posix error numbers. */
591
592static PyObject * os2_error(int code)
593{
594 char text[1024];
595 PyObject *v;
596
597 os2_strerror(text, sizeof(text), code, "");
598
599 v = Py_BuildValue("(is)", code, text);
600 if (v != NULL) {
Barry Warsawca74da41999-02-09 19:31:45 +0000601 PyErr_SetObject(PyExc_OSError, v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000602 Py_DECREF(v);
603 }
604 return NULL; /* Signal to Python that an Exception is Pending */
605}
606
607#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000608
609/* POSIX generic methods */
610
Barry Warsaw53699e91996-12-10 23:23:01 +0000611static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +0000612posix_fildes(PyObject *fdobj, int (*func)(int))
613{
614 int fd;
615 int res;
616 fd = PyObject_AsFileDescriptor(fdobj);
617 if (fd < 0)
618 return NULL;
619 Py_BEGIN_ALLOW_THREADS
620 res = (*func)(fd);
621 Py_END_ALLOW_THREADS
622 if (res < 0)
623 return posix_error();
624 Py_INCREF(Py_None);
625 return Py_None;
626}
Guido van Rossum21142a01999-01-08 21:05:37 +0000627
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000628#ifdef Py_WIN_WIDE_FILENAMES
Tim Peters11b23062003-04-23 02:39:17 +0000629static int
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000630unicode_file_names(void)
631{
632 static int canusewide = -1;
633 if (canusewide == -1) {
Tim Peters11b23062003-04-23 02:39:17 +0000634 /* As per doc for ::GetVersion(), this is the correct test for
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000635 the Windows NT family. */
636 canusewide = (GetVersion() < 0x80000000) ? 1 : 0;
637 }
638 return canusewide;
639}
640#endif
Tim Peters11b23062003-04-23 02:39:17 +0000641
Guido van Rossum21142a01999-01-08 21:05:37 +0000642static PyObject *
Thomas Wouters477c8d52006-05-27 19:21:47 +0000643posix_1str(PyObject *args, char *format, int (*func)(const char*))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000644{
Mark Hammondef8b6542001-05-13 08:04:26 +0000645 char *path1 = NULL;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000646 int res;
Tim Peters5aa91602002-01-30 05:46:57 +0000647 if (!PyArg_ParseTuple(args, format,
Mark Hammondef8b6542001-05-13 08:04:26 +0000648 Py_FileSystemDefaultEncoding, &path1))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000649 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +0000650 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000651 res = (*func)(path1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000652 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000653 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +0000654 return posix_error_with_allocated_filename(path1);
655 PyMem_Free(path1);
Barry Warsaw53699e91996-12-10 23:23:01 +0000656 Py_INCREF(Py_None);
657 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000658}
659
Barry Warsaw53699e91996-12-10 23:23:01 +0000660static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +0000661posix_2str(PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000662 char *format,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000663 int (*func)(const char *, const char *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000664{
Mark Hammondef8b6542001-05-13 08:04:26 +0000665 char *path1 = NULL, *path2 = NULL;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000666 int res;
Mark Hammondef8b6542001-05-13 08:04:26 +0000667 if (!PyArg_ParseTuple(args, format,
Tim Peters5aa91602002-01-30 05:46:57 +0000668 Py_FileSystemDefaultEncoding, &path1,
Mark Hammondef8b6542001-05-13 08:04:26 +0000669 Py_FileSystemDefaultEncoding, &path2))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000670 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +0000671 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000672 res = (*func)(path1, path2);
Barry Warsaw53699e91996-12-10 23:23:01 +0000673 Py_END_ALLOW_THREADS
Mark Hammondef8b6542001-05-13 08:04:26 +0000674 PyMem_Free(path1);
675 PyMem_Free(path2);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000676 if (res != 0)
Barry Warsawd58d7641998-07-23 16:14:40 +0000677 /* XXX how to report both path1 and path2??? */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000678 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +0000679 Py_INCREF(Py_None);
680 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000681}
682
Thomas Wouters477c8d52006-05-27 19:21:47 +0000683#ifdef Py_WIN_WIDE_FILENAMES
684static PyObject*
685win32_1str(PyObject* args, char* func,
686 char* format, BOOL (__stdcall *funcA)(LPCSTR),
687 char* wformat, BOOL (__stdcall *funcW)(LPWSTR))
688{
689 PyObject *uni;
690 char *ansi;
691 BOOL result;
692 if (unicode_file_names()) {
693 if (!PyArg_ParseTuple(args, wformat, &uni))
694 PyErr_Clear();
695 else {
696 Py_BEGIN_ALLOW_THREADS
697 result = funcW(PyUnicode_AsUnicode(uni));
698 Py_END_ALLOW_THREADS
699 if (!result)
700 return win32_error_unicode(func, PyUnicode_AsUnicode(uni));
701 Py_INCREF(Py_None);
702 return Py_None;
703 }
704 }
705 if (!PyArg_ParseTuple(args, format, &ansi))
706 return NULL;
707 Py_BEGIN_ALLOW_THREADS
708 result = funcA(ansi);
709 Py_END_ALLOW_THREADS
710 if (!result)
711 return win32_error(func, ansi);
712 Py_INCREF(Py_None);
713 return Py_None;
714
715}
716
717/* This is a reimplementation of the C library's chdir function,
718 but one that produces Win32 errors instead of DOS error codes.
719 chdir is essentially a wrapper around SetCurrentDirectory; however,
720 it also needs to set "magic" environment variables indicating
721 the per-drive current directory, which are of the form =<drive>: */
722BOOL __stdcall
723win32_chdir(LPCSTR path)
724{
725 char new_path[MAX_PATH+1];
726 int result;
727 char env[4] = "=x:";
728
729 if(!SetCurrentDirectoryA(path))
730 return FALSE;
731 result = GetCurrentDirectoryA(MAX_PATH+1, new_path);
732 if (!result)
733 return FALSE;
734 /* In the ANSI API, there should not be any paths longer
735 than MAX_PATH. */
736 assert(result <= MAX_PATH+1);
737 if (strncmp(new_path, "\\\\", 2) == 0 ||
738 strncmp(new_path, "//", 2) == 0)
739 /* UNC path, nothing to do. */
740 return TRUE;
741 env[1] = new_path[0];
742 return SetEnvironmentVariableA(env, new_path);
743}
744
745/* The Unicode version differs from the ANSI version
746 since the current directory might exceed MAX_PATH characters */
747BOOL __stdcall
748win32_wchdir(LPCWSTR path)
749{
750 wchar_t _new_path[MAX_PATH+1], *new_path = _new_path;
751 int result;
752 wchar_t env[4] = L"=x:";
753
754 if(!SetCurrentDirectoryW(path))
755 return FALSE;
756 result = GetCurrentDirectoryW(MAX_PATH+1, new_path);
757 if (!result)
758 return FALSE;
759 if (result > MAX_PATH+1) {
760 new_path = malloc(result);
761 if (!new_path) {
762 SetLastError(ERROR_OUTOFMEMORY);
763 return FALSE;
764 }
765 }
766 if (wcsncmp(new_path, L"\\\\", 2) == 0 ||
767 wcsncmp(new_path, L"//", 2) == 0)
768 /* UNC path, nothing to do. */
769 return TRUE;
770 env[1] = new_path[0];
771 result = SetEnvironmentVariableW(env, new_path);
772 if (new_path != _new_path)
773 free(new_path);
774 return result;
775}
776#endif
777
Martin v. Löwis14694662006-02-03 12:54:16 +0000778#ifdef MS_WINDOWS
779/* The CRT of Windows has a number of flaws wrt. its stat() implementation:
780 - time stamps are restricted to second resolution
781 - file modification times suffer from forth-and-back conversions between
782 UTC and local time
783 Therefore, we implement our own stat, based on the Win32 API directly.
784*/
785#define HAVE_STAT_NSEC 1
786
787struct win32_stat{
788 int st_dev;
789 __int64 st_ino;
790 unsigned short st_mode;
791 int st_nlink;
792 int st_uid;
793 int st_gid;
794 int st_rdev;
795 __int64 st_size;
796 int st_atime;
797 int st_atime_nsec;
798 int st_mtime;
799 int st_mtime_nsec;
800 int st_ctime;
801 int st_ctime_nsec;
802};
803
804static __int64 secs_between_epochs = 11644473600; /* Seconds between 1.1.1601 and 1.1.1970 */
805
806static void
807FILE_TIME_to_time_t_nsec(FILETIME *in_ptr, int *time_out, int* nsec_out)
808{
Thomas Wouters477c8d52006-05-27 19:21:47 +0000809 /* XXX endianness. Shouldn't matter, as all Windows implementations are little-endian */
810 /* Cannot simply cast and dereference in_ptr,
811 since it might not be aligned properly */
812 __int64 in;
813 memcpy(&in, in_ptr, sizeof(in));
Martin v. Löwis14694662006-02-03 12:54:16 +0000814 *nsec_out = (int)(in % 10000000) * 100; /* FILETIME is in units of 100 nsec. */
815 /* XXX Win32 supports time stamps past 2038; we currently don't */
816 *time_out = Py_SAFE_DOWNCAST((in / 10000000) - secs_between_epochs, __int64, int);
817}
818
Thomas Wouters477c8d52006-05-27 19:21:47 +0000819static void
820time_t_to_FILE_TIME(int time_in, int nsec_in, FILETIME *out_ptr)
821{
822 /* XXX endianness */
823 __int64 out;
824 out = time_in + secs_between_epochs;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000825 out = out * 10000000 + nsec_in / 100;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000826 memcpy(out_ptr, &out, sizeof(out));
827}
828
Martin v. Löwis14694662006-02-03 12:54:16 +0000829/* Below, we *know* that ugo+r is 0444 */
830#if _S_IREAD != 0400
831#error Unsupported C library
832#endif
833static int
834attributes_to_mode(DWORD attr)
835{
836 int m = 0;
837 if (attr & FILE_ATTRIBUTE_DIRECTORY)
838 m |= _S_IFDIR | 0111; /* IFEXEC for user,group,other */
839 else
840 m |= _S_IFREG;
841 if (attr & FILE_ATTRIBUTE_READONLY)
842 m |= 0444;
843 else
844 m |= 0666;
845 return m;
846}
847
848static int
849attribute_data_to_stat(WIN32_FILE_ATTRIBUTE_DATA *info, struct win32_stat *result)
850{
851 memset(result, 0, sizeof(*result));
852 result->st_mode = attributes_to_mode(info->dwFileAttributes);
853 result->st_size = (((__int64)info->nFileSizeHigh)<<32) + info->nFileSizeLow;
854 FILE_TIME_to_time_t_nsec(&info->ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
855 FILE_TIME_to_time_t_nsec(&info->ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
856 FILE_TIME_to_time_t_nsec(&info->ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
857
858 return 0;
859}
860
Thomas Wouters89f507f2006-12-13 04:49:30 +0000861/* Emulate GetFileAttributesEx[AW] on Windows 95 */
862static int checked = 0;
863static BOOL (CALLBACK *gfaxa)(LPCSTR, GET_FILEEX_INFO_LEVELS, LPVOID);
864static BOOL (CALLBACK *gfaxw)(LPCWSTR, GET_FILEEX_INFO_LEVELS, LPVOID);
865static void
866check_gfax()
867{
868 HINSTANCE hKernel32;
869 if (checked)
870 return;
871 checked = 1;
872 hKernel32 = GetModuleHandle("KERNEL32");
873 *(FARPROC*)&gfaxa = GetProcAddress(hKernel32, "GetFileAttributesExA");
874 *(FARPROC*)&gfaxw = GetProcAddress(hKernel32, "GetFileAttributesExW");
875}
876
Guido van Rossumd8faa362007-04-27 19:54:29 +0000877static BOOL
878attributes_from_dir(LPCSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
879{
880 HANDLE hFindFile;
881 WIN32_FIND_DATAA FileData;
882 hFindFile = FindFirstFileA(pszFile, &FileData);
883 if (hFindFile == INVALID_HANDLE_VALUE)
884 return FALSE;
885 FindClose(hFindFile);
886 pfad->dwFileAttributes = FileData.dwFileAttributes;
887 pfad->ftCreationTime = FileData.ftCreationTime;
888 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
889 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
890 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
891 pfad->nFileSizeLow = FileData.nFileSizeLow;
892 return TRUE;
893}
894
895static BOOL
896attributes_from_dir_w(LPCWSTR pszFile, LPWIN32_FILE_ATTRIBUTE_DATA pfad)
897{
898 HANDLE hFindFile;
899 WIN32_FIND_DATAW FileData;
900 hFindFile = FindFirstFileW(pszFile, &FileData);
901 if (hFindFile == INVALID_HANDLE_VALUE)
902 return FALSE;
903 FindClose(hFindFile);
904 pfad->dwFileAttributes = FileData.dwFileAttributes;
905 pfad->ftCreationTime = FileData.ftCreationTime;
906 pfad->ftLastAccessTime = FileData.ftLastAccessTime;
907 pfad->ftLastWriteTime = FileData.ftLastWriteTime;
908 pfad->nFileSizeHigh = FileData.nFileSizeHigh;
909 pfad->nFileSizeLow = FileData.nFileSizeLow;
910 return TRUE;
911}
912
Thomas Wouters89f507f2006-12-13 04:49:30 +0000913static BOOL WINAPI
914Py_GetFileAttributesExA(LPCSTR pszFile,
915 GET_FILEEX_INFO_LEVELS level,
916 LPVOID pv)
917{
918 BOOL result;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000919 LPWIN32_FILE_ATTRIBUTE_DATA pfad = pv;
920 /* First try to use the system's implementation, if that is
921 available and either succeeds to gives an error other than
922 that it isn't implemented. */
923 check_gfax();
924 if (gfaxa) {
925 result = gfaxa(pszFile, level, pv);
926 if (result || GetLastError() != ERROR_CALL_NOT_IMPLEMENTED)
927 return result;
928 }
929 /* It's either not present, or not implemented.
930 Emulate using FindFirstFile. */
931 if (level != GetFileExInfoStandard) {
932 SetLastError(ERROR_INVALID_PARAMETER);
933 return FALSE;
934 }
935 /* Use GetFileAttributes to validate that the file name
936 does not contain wildcards (which FindFirstFile would
937 accept). */
938 if (GetFileAttributesA(pszFile) == 0xFFFFFFFF)
939 return FALSE;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000940 return attributes_from_dir(pszFile, pfad);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000941}
942
943static BOOL WINAPI
944Py_GetFileAttributesExW(LPCWSTR pszFile,
945 GET_FILEEX_INFO_LEVELS level,
946 LPVOID pv)
947{
948 BOOL result;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000949 LPWIN32_FILE_ATTRIBUTE_DATA pfad = pv;
950 /* First try to use the system's implementation, if that is
951 available and either succeeds to gives an error other than
952 that it isn't implemented. */
953 check_gfax();
954 if (gfaxa) {
955 result = gfaxw(pszFile, level, pv);
956 if (result || GetLastError() != ERROR_CALL_NOT_IMPLEMENTED)
957 return result;
958 }
959 /* It's either not present, or not implemented.
960 Emulate using FindFirstFile. */
961 if (level != GetFileExInfoStandard) {
962 SetLastError(ERROR_INVALID_PARAMETER);
963 return FALSE;
964 }
965 /* Use GetFileAttributes to validate that the file name
966 does not contain wildcards (which FindFirstFile would
967 accept). */
968 if (GetFileAttributesW(pszFile) == 0xFFFFFFFF)
969 return FALSE;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000970 return attributes_from_dir_w(pszFile, pfad);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000971}
972
Martin v. Löwis14694662006-02-03 12:54:16 +0000973static int
974win32_stat(const char* path, struct win32_stat *result)
975{
976 WIN32_FILE_ATTRIBUTE_DATA info;
977 int code;
978 char *dot;
979 /* XXX not supported on Win95 and NT 3.x */
Thomas Wouters89f507f2006-12-13 04:49:30 +0000980 if (!Py_GetFileAttributesExA(path, GetFileExInfoStandard, &info)) {
Guido van Rossumd8faa362007-04-27 19:54:29 +0000981 if (GetLastError() != ERROR_SHARING_VIOLATION) {
982 /* Protocol violation: we explicitly clear errno, instead of
983 setting it to a POSIX error. Callers should use GetLastError. */
984 errno = 0;
985 return -1;
986 } else {
987 /* Could not get attributes on open file. Fall back to
988 reading the directory. */
989 if (!attributes_from_dir(path, &info)) {
990 /* Very strange. This should not fail now */
991 errno = 0;
992 return -1;
993 }
994 }
Martin v. Löwis14694662006-02-03 12:54:16 +0000995 }
996 code = attribute_data_to_stat(&info, result);
997 if (code != 0)
998 return code;
999 /* Set S_IFEXEC if it is an .exe, .bat, ... */
1000 dot = strrchr(path, '.');
1001 if (dot) {
1002 if (stricmp(dot, ".bat") == 0 ||
1003 stricmp(dot, ".cmd") == 0 ||
1004 stricmp(dot, ".exe") == 0 ||
1005 stricmp(dot, ".com") == 0)
1006 result->st_mode |= 0111;
1007 }
1008 return code;
1009}
1010
1011static int
1012win32_wstat(const wchar_t* path, struct win32_stat *result)
1013{
1014 int code;
1015 const wchar_t *dot;
1016 WIN32_FILE_ATTRIBUTE_DATA info;
1017 /* XXX not supported on Win95 and NT 3.x */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001018 if (!Py_GetFileAttributesExW(path, GetFileExInfoStandard, &info)) {
Guido van Rossumd8faa362007-04-27 19:54:29 +00001019 if (GetLastError() != ERROR_SHARING_VIOLATION) {
1020 /* Protocol violation: we explicitly clear errno, instead of
1021 setting it to a POSIX error. Callers should use GetLastError. */
1022 errno = 0;
1023 return -1;
1024 } else {
1025 /* Could not get attributes on open file. Fall back to
1026 reading the directory. */
1027 if (!attributes_from_dir_w(path, &info)) {
1028 /* Very strange. This should not fail now */
1029 errno = 0;
1030 return -1;
1031 }
1032 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001033 }
1034 code = attribute_data_to_stat(&info, result);
1035 if (code < 0)
1036 return code;
1037 /* Set IFEXEC if it is an .exe, .bat, ... */
1038 dot = wcsrchr(path, '.');
1039 if (dot) {
1040 if (_wcsicmp(dot, L".bat") == 0 ||
1041 _wcsicmp(dot, L".cmd") == 0 ||
1042 _wcsicmp(dot, L".exe") == 0 ||
1043 _wcsicmp(dot, L".com") == 0)
1044 result->st_mode |= 0111;
1045 }
1046 return code;
1047}
1048
1049static int
1050win32_fstat(int file_number, struct win32_stat *result)
1051{
1052 BY_HANDLE_FILE_INFORMATION info;
1053 HANDLE h;
1054 int type;
1055
1056 h = (HANDLE)_get_osfhandle(file_number);
1057
1058 /* Protocol violation: we explicitly clear errno, instead of
1059 setting it to a POSIX error. Callers should use GetLastError. */
1060 errno = 0;
1061
1062 if (h == INVALID_HANDLE_VALUE) {
1063 /* This is really a C library error (invalid file handle).
1064 We set the Win32 error to the closes one matching. */
1065 SetLastError(ERROR_INVALID_HANDLE);
1066 return -1;
1067 }
1068 memset(result, 0, sizeof(*result));
1069
1070 type = GetFileType(h);
1071 if (type == FILE_TYPE_UNKNOWN) {
1072 DWORD error = GetLastError();
1073 if (error != 0) {
1074 return -1;
1075 }
1076 /* else: valid but unknown file */
1077 }
1078
1079 if (type != FILE_TYPE_DISK) {
1080 if (type == FILE_TYPE_CHAR)
1081 result->st_mode = _S_IFCHR;
1082 else if (type == FILE_TYPE_PIPE)
1083 result->st_mode = _S_IFIFO;
1084 return 0;
1085 }
1086
1087 if (!GetFileInformationByHandle(h, &info)) {
1088 return -1;
1089 }
1090
1091 /* similar to stat() */
1092 result->st_mode = attributes_to_mode(info.dwFileAttributes);
1093 result->st_size = (((__int64)info.nFileSizeHigh)<<32) + info.nFileSizeLow;
1094 FILE_TIME_to_time_t_nsec(&info.ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1095 FILE_TIME_to_time_t_nsec(&info.ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1096 FILE_TIME_to_time_t_nsec(&info.ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
1097 /* specific to fstat() */
1098 result->st_nlink = info.nNumberOfLinks;
1099 result->st_ino = (((__int64)info.nFileIndexHigh)<<32) + info.nFileIndexLow;
1100 return 0;
1101}
1102
1103#endif /* MS_WINDOWS */
1104
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001105PyDoc_STRVAR(stat_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001106"stat_result: Result from stat or lstat.\n\n\
1107This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001108 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001109or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
1110\n\
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001111Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
1112or st_flags, they are available as attributes only.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001113\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001114See os.stat for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001115
1116static PyStructSequence_Field stat_result_fields[] = {
1117 {"st_mode", "protection bits"},
1118 {"st_ino", "inode"},
1119 {"st_dev", "device"},
1120 {"st_nlink", "number of hard links"},
1121 {"st_uid", "user ID of owner"},
1122 {"st_gid", "group ID of owner"},
1123 {"st_size", "total size, in bytes"},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001124 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
1125 {NULL, "integer time of last access"},
1126 {NULL, "integer time of last modification"},
1127 {NULL, "integer time of last change"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001128 {"st_atime", "time of last access"},
1129 {"st_mtime", "time of last modification"},
1130 {"st_ctime", "time of last change"},
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001131#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001132 {"st_blksize", "blocksize for filesystem I/O"},
1133#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001134#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001135 {"st_blocks", "number of blocks allocated"},
1136#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001137#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001138 {"st_rdev", "device type (if inode device)"},
1139#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001140#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1141 {"st_flags", "user defined flags for file"},
1142#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001143#ifdef HAVE_STRUCT_STAT_ST_GEN
1144 {"st_gen", "generation number"},
1145#endif
1146#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1147 {"st_birthtime", "time of creation"},
1148#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001149 {0}
1150};
1151
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001152#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001153#define ST_BLKSIZE_IDX 13
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001154#else
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001155#define ST_BLKSIZE_IDX 12
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001156#endif
1157
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001158#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001159#define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
1160#else
1161#define ST_BLOCKS_IDX ST_BLKSIZE_IDX
1162#endif
1163
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001164#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001165#define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
1166#else
1167#define ST_RDEV_IDX ST_BLOCKS_IDX
1168#endif
1169
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001170#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1171#define ST_FLAGS_IDX (ST_RDEV_IDX+1)
1172#else
1173#define ST_FLAGS_IDX ST_RDEV_IDX
1174#endif
1175
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001176#ifdef HAVE_STRUCT_STAT_ST_GEN
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001177#define ST_GEN_IDX (ST_FLAGS_IDX+1)
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001178#else
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001179#define ST_GEN_IDX ST_FLAGS_IDX
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001180#endif
1181
1182#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1183#define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
1184#else
1185#define ST_BIRTHTIME_IDX ST_GEN_IDX
1186#endif
1187
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001188static PyStructSequence_Desc stat_result_desc = {
1189 "stat_result", /* name */
1190 stat_result__doc__, /* doc */
1191 stat_result_fields,
1192 10
1193};
1194
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001195PyDoc_STRVAR(statvfs_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001196"statvfs_result: Result from statvfs or fstatvfs.\n\n\
1197This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001198 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
Guido van Rossuma4dc73e2001-10-18 20:53:15 +00001199or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001200\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001201See os.statvfs for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001202
1203static PyStructSequence_Field statvfs_result_fields[] = {
1204 {"f_bsize", },
1205 {"f_frsize", },
1206 {"f_blocks", },
1207 {"f_bfree", },
1208 {"f_bavail", },
1209 {"f_files", },
1210 {"f_ffree", },
1211 {"f_favail", },
1212 {"f_flag", },
1213 {"f_namemax",},
1214 {0}
1215};
1216
1217static PyStructSequence_Desc statvfs_result_desc = {
1218 "statvfs_result", /* name */
1219 statvfs_result__doc__, /* doc */
1220 statvfs_result_fields,
1221 10
1222};
1223
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001224static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001225static PyTypeObject StatResultType;
1226static PyTypeObject StatVFSResultType;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001227static newfunc structseq_new;
1228
1229static PyObject *
1230statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1231{
1232 PyStructSequence *result;
1233 int i;
1234
1235 result = (PyStructSequence*)structseq_new(type, args, kwds);
1236 if (!result)
1237 return NULL;
1238 /* If we have been initialized from a tuple,
1239 st_?time might be set to None. Initialize it
1240 from the int slots. */
1241 for (i = 7; i <= 9; i++) {
1242 if (result->ob_item[i+3] == Py_None) {
1243 Py_DECREF(Py_None);
1244 Py_INCREF(result->ob_item[i]);
1245 result->ob_item[i+3] = result->ob_item[i];
1246 }
1247 }
1248 return (PyObject*)result;
1249}
1250
1251
1252
1253/* If true, st_?time is float. */
Martin v. Löwisfe33d0b2005-01-16 08:57:39 +00001254static int _stat_float_times = 1;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001255
1256PyDoc_STRVAR(stat_float_times__doc__,
1257"stat_float_times([newval]) -> oldval\n\n\
1258Determine whether os.[lf]stat represents time stamps as float objects.\n\
1259If newval is True, future calls to stat() return floats, if it is False,\n\
1260future calls return ints. \n\
1261If newval is omitted, return the current setting.\n");
1262
1263static PyObject*
1264stat_float_times(PyObject* self, PyObject *args)
1265{
1266 int newval = -1;
1267 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
1268 return NULL;
1269 if (newval == -1)
1270 /* Return old value */
1271 return PyBool_FromLong(_stat_float_times);
1272 _stat_float_times = newval;
1273 Py_INCREF(Py_None);
1274 return Py_None;
1275}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001276
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001277static void
1278fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
1279{
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001280 PyObject *fval,*ival;
1281#if SIZEOF_TIME_T > SIZEOF_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00001282 ival = PyLong_FromLongLong((PY_LONG_LONG)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001283#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001284 ival = PyLong_FromLong((long)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001285#endif
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001286 if (!ival)
1287 return;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001288 if (_stat_float_times) {
1289 fval = PyFloat_FromDouble(sec + 1e-9*nsec);
1290 } else {
1291 fval = ival;
1292 Py_INCREF(fval);
1293 }
1294 PyStructSequence_SET_ITEM(v, index, ival);
1295 PyStructSequence_SET_ITEM(v, index+3, fval);
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001296}
1297
Tim Peters5aa91602002-01-30 05:46:57 +00001298/* pack a system stat C structure into the Python stat tuple
Fred Drake699f3522000-06-29 21:12:41 +00001299 (used by posix_stat() and posix_fstat()) */
1300static PyObject*
Martin v. Löwis14694662006-02-03 12:54:16 +00001301_pystat_fromstructstat(STRUCT_STAT *st)
Fred Drake699f3522000-06-29 21:12:41 +00001302{
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001303 unsigned long ansec, mnsec, cnsec;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001304 PyObject *v = PyStructSequence_New(&StatResultType);
Fred Drake699f3522000-06-29 21:12:41 +00001305 if (v == NULL)
1306 return NULL;
1307
Christian Heimes217cfd12007-12-02 14:31:20 +00001308 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long)st->st_mode));
Fred Drake699f3522000-06-29 21:12:41 +00001309#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +00001310 PyStructSequence_SET_ITEM(v, 1,
Martin v. Löwis14694662006-02-03 12:54:16 +00001311 PyLong_FromLongLong((PY_LONG_LONG)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001312#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001313 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001314#endif
1315#if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
Tim Peters5aa91602002-01-30 05:46:57 +00001316 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwis14694662006-02-03 12:54:16 +00001317 PyLong_FromLongLong((PY_LONG_LONG)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001318#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001319 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001320#endif
Christian Heimes217cfd12007-12-02 14:31:20 +00001321 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long)st->st_nlink));
1322 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long)st->st_uid));
1323 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long)st->st_gid));
Fred Drake699f3522000-06-29 21:12:41 +00001324#ifdef HAVE_LARGEFILE_SUPPORT
Tim Peters5aa91602002-01-30 05:46:57 +00001325 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwis14694662006-02-03 12:54:16 +00001326 PyLong_FromLongLong((PY_LONG_LONG)st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001327#else
Christian Heimes217cfd12007-12-02 14:31:20 +00001328 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong(st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001329#endif
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001330
Martin v. Löwis14694662006-02-03 12:54:16 +00001331#if defined(HAVE_STAT_TV_NSEC)
1332 ansec = st->st_atim.tv_nsec;
1333 mnsec = st->st_mtim.tv_nsec;
1334 cnsec = st->st_ctim.tv_nsec;
1335#elif defined(HAVE_STAT_TV_NSEC2)
1336 ansec = st->st_atimespec.tv_nsec;
1337 mnsec = st->st_mtimespec.tv_nsec;
1338 cnsec = st->st_ctimespec.tv_nsec;
1339#elif defined(HAVE_STAT_NSEC)
1340 ansec = st->st_atime_nsec;
1341 mnsec = st->st_mtime_nsec;
1342 cnsec = st->st_ctime_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001343#else
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001344 ansec = mnsec = cnsec = 0;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001345#endif
Martin v. Löwis14694662006-02-03 12:54:16 +00001346 fill_time(v, 7, st->st_atime, ansec);
1347 fill_time(v, 8, st->st_mtime, mnsec);
1348 fill_time(v, 9, st->st_ctime, cnsec);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001349
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001350#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Tim Peters5aa91602002-01-30 05:46:57 +00001351 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001352 PyLong_FromLong((long)st->st_blksize));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001353#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001354#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Tim Peters5aa91602002-01-30 05:46:57 +00001355 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001356 PyLong_FromLong((long)st->st_blocks));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001357#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001358#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001359 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001360 PyLong_FromLong((long)st->st_rdev));
Fred Drake699f3522000-06-29 21:12:41 +00001361#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001362#ifdef HAVE_STRUCT_STAT_ST_GEN
1363 PyStructSequence_SET_ITEM(v, ST_GEN_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001364 PyLong_FromLong((long)st->st_gen));
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001365#endif
1366#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1367 {
1368 PyObject *val;
1369 unsigned long bsec,bnsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001370 bsec = (long)st->st_birthtime;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001371#ifdef HAVE_STAT_TV_NSEC2
Hye-Shik Changd69e0342006-02-19 16:22:22 +00001372 bnsec = st->st_birthtimespec.tv_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001373#else
1374 bnsec = 0;
1375#endif
1376 if (_stat_float_times) {
1377 val = PyFloat_FromDouble(bsec + 1e-9*bnsec);
1378 } else {
Christian Heimes217cfd12007-12-02 14:31:20 +00001379 val = PyLong_FromLong((long)bsec);
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001380 }
1381 PyStructSequence_SET_ITEM(v, ST_BIRTHTIME_IDX,
1382 val);
1383 }
1384#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001385#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1386 PyStructSequence_SET_ITEM(v, ST_FLAGS_IDX,
Christian Heimes217cfd12007-12-02 14:31:20 +00001387 PyLong_FromLong((long)st->st_flags));
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001388#endif
Fred Drake699f3522000-06-29 21:12:41 +00001389
1390 if (PyErr_Occurred()) {
1391 Py_DECREF(v);
1392 return NULL;
1393 }
1394
1395 return v;
1396}
1397
Martin v. Löwisd8948722004-06-02 09:57:56 +00001398#ifdef MS_WINDOWS
1399
1400/* IsUNCRoot -- test whether the supplied path is of the form \\SERVER\SHARE\,
1401 where / can be used in place of \ and the trailing slash is optional.
1402 Both SERVER and SHARE must have at least one character.
1403*/
1404
1405#define ISSLASHA(c) ((c) == '\\' || (c) == '/')
1406#define ISSLASHW(c) ((c) == L'\\' || (c) == L'/')
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001407#ifndef ARRAYSIZE
Martin v. Löwisd8948722004-06-02 09:57:56 +00001408#define ARRAYSIZE(a) (sizeof(a) / sizeof(a[0]))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001409#endif
Martin v. Löwisd8948722004-06-02 09:57:56 +00001410
Tim Peters4ad82172004-08-30 17:02:04 +00001411static BOOL
Martin v. Löwisd8948722004-06-02 09:57:56 +00001412IsUNCRootA(char *path, int pathlen)
1413{
1414 #define ISSLASH ISSLASHA
1415
1416 int i, share;
1417
1418 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1419 /* minimum UNCRoot is \\x\y */
1420 return FALSE;
1421 for (i = 2; i < pathlen ; i++)
1422 if (ISSLASH(path[i])) break;
1423 if (i == 2 || i == pathlen)
1424 /* do not allow \\\SHARE or \\SERVER */
1425 return FALSE;
1426 share = i+1;
1427 for (i = share; i < pathlen; i++)
1428 if (ISSLASH(path[i])) break;
1429 return (i != share && (i == pathlen || i == pathlen-1));
1430
1431 #undef ISSLASH
1432}
1433
1434#ifdef Py_WIN_WIDE_FILENAMES
Tim Peters4ad82172004-08-30 17:02:04 +00001435static BOOL
Martin v. Löwisd8948722004-06-02 09:57:56 +00001436IsUNCRootW(Py_UNICODE *path, int pathlen)
1437{
1438 #define ISSLASH ISSLASHW
1439
1440 int i, share;
1441
1442 if (pathlen < 5 || !ISSLASH(path[0]) || !ISSLASH(path[1]))
1443 /* minimum UNCRoot is \\x\y */
1444 return FALSE;
1445 for (i = 2; i < pathlen ; i++)
1446 if (ISSLASH(path[i])) break;
1447 if (i == 2 || i == pathlen)
1448 /* do not allow \\\SHARE or \\SERVER */
1449 return FALSE;
1450 share = i+1;
1451 for (i = share; i < pathlen; i++)
1452 if (ISSLASH(path[i])) break;
1453 return (i != share && (i == pathlen || i == pathlen-1));
1454
1455 #undef ISSLASH
1456}
1457#endif /* Py_WIN_WIDE_FILENAMES */
1458#endif /* MS_WINDOWS */
1459
Barry Warsaw53699e91996-12-10 23:23:01 +00001460static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +00001461posix_do_stat(PyObject *self, PyObject *args,
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001462 char *format,
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001463#ifdef __VMS
1464 int (*statfunc)(const char *, STRUCT_STAT *, ...),
1465#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001466 int (*statfunc)(const char *, STRUCT_STAT *),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001467#endif
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001468 char *wformat,
1469 int (*wstatfunc)(const Py_UNICODE *, STRUCT_STAT *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001470{
Fred Drake699f3522000-06-29 21:12:41 +00001471 STRUCT_STAT st;
Tim Peters500bd032001-12-19 19:05:01 +00001472 char *path = NULL; /* pass this to stat; do not free() it */
1473 char *pathfree = NULL; /* this memory must be free'd */
Guido van Rossumff4949e1992-08-05 19:58:53 +00001474 int res;
Martin v. Löwis14694662006-02-03 12:54:16 +00001475 PyObject *result;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001476
1477#ifdef Py_WIN_WIDE_FILENAMES
1478 /* If on wide-character-capable OS see if argument
1479 is Unicode and if so use wide API. */
1480 if (unicode_file_names()) {
1481 PyUnicodeObject *po;
1482 if (PyArg_ParseTuple(args, wformat, &po)) {
Martin v. Löwis14694662006-02-03 12:54:16 +00001483 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
1484
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001485 Py_BEGIN_ALLOW_THREADS
1486 /* PyUnicode_AS_UNICODE result OK without
1487 thread lock as it is a simple dereference. */
1488 res = wstatfunc(wpath, &st);
1489 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001490
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001491 if (res != 0)
Martin v. Löwis14694662006-02-03 12:54:16 +00001492 return win32_error_unicode("stat", wpath);
1493 return _pystat_fromstructstat(&st);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001494 }
1495 /* Drop the argument parsing error as narrow strings
1496 are also valid. */
1497 PyErr_Clear();
1498 }
1499#endif
1500
Tim Peters5aa91602002-01-30 05:46:57 +00001501 if (!PyArg_ParseTuple(args, format,
Mark Hammondef8b6542001-05-13 08:04:26 +00001502 Py_FileSystemDefaultEncoding, &path))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001503 return NULL;
Tim Peters500bd032001-12-19 19:05:01 +00001504 pathfree = path;
Guido van Rossumace88ae2000-04-21 18:54:45 +00001505
Barry Warsaw53699e91996-12-10 23:23:01 +00001506 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001507 res = (*statfunc)(path, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00001508 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001509
1510 if (res != 0) {
1511#ifdef MS_WINDOWS
1512 result = win32_error("stat", pathfree);
1513#else
1514 result = posix_error_with_filename(pathfree);
1515#endif
1516 }
1517 else
1518 result = _pystat_fromstructstat(&st);
Fred Drake699f3522000-06-29 21:12:41 +00001519
Tim Peters500bd032001-12-19 19:05:01 +00001520 PyMem_Free(pathfree);
Martin v. Löwis14694662006-02-03 12:54:16 +00001521 return result;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001522}
1523
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001524/* POSIX methods */
1525
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001526PyDoc_STRVAR(posix_access__doc__,
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001527"access(path, mode) -> True if granted, False otherwise\n\n\
Guido van Rossuma0b90752002-06-18 16:22:43 +00001528Use the real uid/gid to test for access to a path. Note that most\n\
1529operations will use the effective uid/gid, therefore this routine can\n\
1530be used in a suid/sgid environment to test if the invoking user has the\n\
1531specified access to the path. The mode argument can be F_OK to test\n\
1532existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001533
1534static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001535posix_access(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001536{
Guido van Rossum015f22a1999-01-06 22:52:38 +00001537 char *path;
1538 int mode;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001539
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001540#ifdef Py_WIN_WIDE_FILENAMES
Thomas Wouters477c8d52006-05-27 19:21:47 +00001541 DWORD attr;
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001542 if (unicode_file_names()) {
1543 PyUnicodeObject *po;
1544 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
1545 Py_BEGIN_ALLOW_THREADS
1546 /* PyUnicode_AS_UNICODE OK without thread lock as
1547 it is a simple dereference. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001548 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001549 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001550 goto finish;
Martin v. Löwis1b699a52003-09-12 16:25:38 +00001551 }
1552 /* Drop the argument parsing error as narrow strings
1553 are also valid. */
1554 PyErr_Clear();
1555 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001556 if (!PyArg_ParseTuple(args, "eti:access",
1557 Py_FileSystemDefaultEncoding, &path, &mode))
1558 return 0;
1559 Py_BEGIN_ALLOW_THREADS
1560 attr = GetFileAttributesA(path);
1561 Py_END_ALLOW_THREADS
1562 PyMem_Free(path);
1563finish:
1564 if (attr == 0xFFFFFFFF)
1565 /* File does not exist, or cannot read attributes */
1566 return PyBool_FromLong(0);
1567 /* Access is possible if either write access wasn't requested, or
Guido van Rossumb00324f2007-12-04 01:13:14 +00001568 the file isn't read-only, or if it's a directory, as there are
1569 no read-only directories on Windows. */
1570 return PyBool_FromLong(!(mode & 2)
1571 || !(attr & FILE_ATTRIBUTE_READONLY)
1572 || (attr & FILE_ATTRIBUTE_DIRECTORY));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001573#else
1574 int res;
Martin v. Löwisb60ae992005-03-08 09:10:29 +00001575 if (!PyArg_ParseTuple(args, "eti:access",
1576 Py_FileSystemDefaultEncoding, &path, &mode))
Guido van Rossum015f22a1999-01-06 22:52:38 +00001577 return NULL;
1578 Py_BEGIN_ALLOW_THREADS
1579 res = access(path, mode);
1580 Py_END_ALLOW_THREADS
Neal Norwitz24b3c222005-09-19 06:43:44 +00001581 PyMem_Free(path);
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001582 return PyBool_FromLong(res == 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001583#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001584}
1585
Guido van Rossumd371ff11999-01-25 16:12:23 +00001586#ifndef F_OK
1587#define F_OK 0
1588#endif
1589#ifndef R_OK
1590#define R_OK 4
1591#endif
1592#ifndef W_OK
1593#define W_OK 2
1594#endif
1595#ifndef X_OK
1596#define X_OK 1
1597#endif
1598
1599#ifdef HAVE_TTYNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001600PyDoc_STRVAR(posix_ttyname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001601"ttyname(fd) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001602Return the name of the terminal device connected to 'fd'.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001603
1604static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001605posix_ttyname(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001606{
Guido van Rossum94f6f721999-01-06 18:42:14 +00001607 int id;
1608 char *ret;
1609
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001610 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
Guido van Rossum94f6f721999-01-06 18:42:14 +00001611 return NULL;
1612
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001613#if defined(__VMS)
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001614 /* file descriptor 0 only, the default input device (stdin) */
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001615 if (id == 0) {
1616 ret = ttyname();
1617 }
1618 else {
1619 ret = NULL;
1620 }
1621#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00001622 ret = ttyname(id);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001623#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001624 if (ret == NULL)
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001625 return posix_error();
Neal Norwitz93c56822007-08-26 07:10:06 +00001626 return PyUnicode_FromString(ret);
Guido van Rossum94f6f721999-01-06 18:42:14 +00001627}
Guido van Rossumd371ff11999-01-25 16:12:23 +00001628#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001629
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001630#ifdef HAVE_CTERMID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001631PyDoc_STRVAR(posix_ctermid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001632"ctermid() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001633Return the name of the controlling terminal for this process.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001634
1635static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001636posix_ctermid(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001637{
1638 char *ret;
1639 char buffer[L_ctermid];
1640
Greg Wardb48bc172000-03-01 21:51:56 +00001641#ifdef USE_CTERMID_R
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001642 ret = ctermid_r(buffer);
1643#else
1644 ret = ctermid(buffer);
1645#endif
1646 if (ret == NULL)
Neal Norwitz3efd0a12005-09-19 06:45:53 +00001647 return posix_error();
Neal Norwitz93c56822007-08-26 07:10:06 +00001648 return PyUnicode_FromString(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001649}
1650#endif
1651
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001652PyDoc_STRVAR(posix_chdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001653"chdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001654Change the current working directory to the specified path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001655
Barry Warsaw53699e91996-12-10 23:23:01 +00001656static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001657posix_chdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001658{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001659#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001660 return win32_1str(args, "chdir", "s:chdir", win32_chdir, "U:chdir", win32_wchdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001661#elif defined(PYOS_OS2) && defined(PYCC_GCC)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001662 return posix_1str(args, "et:chdir", _chdir2);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00001663#elif defined(__VMS)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001664 return posix_1str(args, "et:chdir", (int (*)(const char *))chdir);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001665#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00001666 return posix_1str(args, "et:chdir", chdir);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001667#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001668}
1669
Fred Drake4d1e64b2002-04-15 19:40:07 +00001670#ifdef HAVE_FCHDIR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001671PyDoc_STRVAR(posix_fchdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001672"fchdir(fildes)\n\n\
Fred Drake4d1e64b2002-04-15 19:40:07 +00001673Change to the directory of the given file descriptor. fildes must be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001674opened on a directory, not a file.");
Fred Drake4d1e64b2002-04-15 19:40:07 +00001675
1676static PyObject *
1677posix_fchdir(PyObject *self, PyObject *fdobj)
1678{
1679 return posix_fildes(fdobj, fchdir);
1680}
1681#endif /* HAVE_FCHDIR */
1682
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001683
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001684PyDoc_STRVAR(posix_chmod__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001685"chmod(path, mode)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001686Change the access permissions of a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001687
Barry Warsaw53699e91996-12-10 23:23:01 +00001688static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001689posix_chmod(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001690{
Mark Hammondef8b6542001-05-13 08:04:26 +00001691 char *path = NULL;
Guido van Rossumffd15f52000-03-31 00:47:28 +00001692 int i;
1693 int res;
Mark Hammond817c9292003-12-03 01:22:38 +00001694#ifdef Py_WIN_WIDE_FILENAMES
Thomas Wouters477c8d52006-05-27 19:21:47 +00001695 DWORD attr;
Mark Hammond817c9292003-12-03 01:22:38 +00001696 if (unicode_file_names()) {
1697 PyUnicodeObject *po;
1698 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
1699 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001700 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1701 if (attr != 0xFFFFFFFF) {
1702 if (i & _S_IWRITE)
1703 attr &= ~FILE_ATTRIBUTE_READONLY;
1704 else
1705 attr |= FILE_ATTRIBUTE_READONLY;
1706 res = SetFileAttributesW(PyUnicode_AS_UNICODE(po), attr);
1707 }
1708 else
1709 res = 0;
Mark Hammond817c9292003-12-03 01:22:38 +00001710 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001711 if (!res)
1712 return win32_error_unicode("chmod",
Mark Hammond817c9292003-12-03 01:22:38 +00001713 PyUnicode_AS_UNICODE(po));
1714 Py_INCREF(Py_None);
1715 return Py_None;
1716 }
1717 /* Drop the argument parsing error as narrow strings
1718 are also valid. */
1719 PyErr_Clear();
1720 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001721 if (!PyArg_ParseTuple(args, "eti:chmod", Py_FileSystemDefaultEncoding,
1722 &path, &i))
1723 return NULL;
1724 Py_BEGIN_ALLOW_THREADS
1725 attr = GetFileAttributesA(path);
1726 if (attr != 0xFFFFFFFF) {
1727 if (i & _S_IWRITE)
1728 attr &= ~FILE_ATTRIBUTE_READONLY;
1729 else
1730 attr |= FILE_ATTRIBUTE_READONLY;
1731 res = SetFileAttributesA(path, attr);
1732 }
1733 else
1734 res = 0;
1735 Py_END_ALLOW_THREADS
1736 if (!res) {
1737 win32_error("chmod", path);
1738 PyMem_Free(path);
1739 return NULL;
1740 }
1741 PyMem_Free(path);
1742 Py_INCREF(Py_None);
1743 return Py_None;
1744#else /* Py_WIN_WIDE_FILENAMES */
Mark Hammond817c9292003-12-03 01:22:38 +00001745 if (!PyArg_ParseTuple(args, "eti:chmod", Py_FileSystemDefaultEncoding,
Mark Hammondef8b6542001-05-13 08:04:26 +00001746 &path, &i))
Guido van Rossumffd15f52000-03-31 00:47:28 +00001747 return NULL;
1748 Py_BEGIN_ALLOW_THREADS
Guido van Rossumef40e772000-03-31 01:26:23 +00001749 res = chmod(path, i);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001750 Py_END_ALLOW_THREADS
1751 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00001752 return posix_error_with_allocated_filename(path);
1753 PyMem_Free(path);
Guido van Rossumffd15f52000-03-31 00:47:28 +00001754 Py_INCREF(Py_None);
1755 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001756#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001757}
1758
Christian Heimes4e30a842007-11-30 22:12:06 +00001759#ifdef HAVE_FCHMOD
1760PyDoc_STRVAR(posix_fchmod__doc__,
1761"fchmod(fd, mode)\n\n\
1762Change the access permissions of the file given by file\n\
1763descriptor fd.");
1764
1765static PyObject *
1766posix_fchmod(PyObject *self, PyObject *args)
1767{
1768 int fd, mode, res;
1769 if (!PyArg_ParseTuple(args, "ii:fchmod", &fd, &mode))
1770 return NULL;
1771 Py_BEGIN_ALLOW_THREADS
1772 res = fchmod(fd, mode);
1773 Py_END_ALLOW_THREADS
1774 if (res < 0)
1775 return posix_error();
1776 Py_RETURN_NONE;
1777}
1778#endif /* HAVE_FCHMOD */
1779
1780#ifdef HAVE_LCHMOD
1781PyDoc_STRVAR(posix_lchmod__doc__,
1782"lchmod(path, mode)\n\n\
1783Change the access permissions of a file. If path is a symlink, this\n\
1784affects the link itself rather than the target.");
1785
1786static PyObject *
1787posix_lchmod(PyObject *self, PyObject *args)
1788{
1789 char *path = NULL;
1790 int i;
1791 int res;
1792 if (!PyArg_ParseTuple(args, "eti:lchmod", Py_FileSystemDefaultEncoding,
1793 &path, &i))
1794 return NULL;
1795 Py_BEGIN_ALLOW_THREADS
1796 res = lchmod(path, i);
1797 Py_END_ALLOW_THREADS
1798 if (res < 0)
1799 return posix_error_with_allocated_filename(path);
1800 PyMem_Free(path);
1801 Py_RETURN_NONE;
1802}
1803#endif /* HAVE_LCHMOD */
1804
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001805
Thomas Wouterscf297e42007-02-23 15:07:44 +00001806#ifdef HAVE_CHFLAGS
1807PyDoc_STRVAR(posix_chflags__doc__,
1808"chflags(path, flags)\n\n\
1809Set file flags.");
1810
1811static PyObject *
1812posix_chflags(PyObject *self, PyObject *args)
1813{
1814 char *path;
1815 unsigned long flags;
1816 int res;
1817 if (!PyArg_ParseTuple(args, "etk:chflags",
1818 Py_FileSystemDefaultEncoding, &path, &flags))
1819 return NULL;
1820 Py_BEGIN_ALLOW_THREADS
1821 res = chflags(path, flags);
1822 Py_END_ALLOW_THREADS
1823 if (res < 0)
1824 return posix_error_with_allocated_filename(path);
1825 PyMem_Free(path);
1826 Py_INCREF(Py_None);
1827 return Py_None;
1828}
1829#endif /* HAVE_CHFLAGS */
1830
1831#ifdef HAVE_LCHFLAGS
1832PyDoc_STRVAR(posix_lchflags__doc__,
1833"lchflags(path, flags)\n\n\
1834Set file flags.\n\
1835This function will not follow symbolic links.");
1836
1837static PyObject *
1838posix_lchflags(PyObject *self, PyObject *args)
1839{
1840 char *path;
1841 unsigned long flags;
1842 int res;
1843 if (!PyArg_ParseTuple(args, "etk:lchflags",
1844 Py_FileSystemDefaultEncoding, &path, &flags))
1845 return NULL;
1846 Py_BEGIN_ALLOW_THREADS
1847 res = lchflags(path, flags);
1848 Py_END_ALLOW_THREADS
1849 if (res < 0)
1850 return posix_error_with_allocated_filename(path);
1851 PyMem_Free(path);
1852 Py_INCREF(Py_None);
1853 return Py_None;
1854}
1855#endif /* HAVE_LCHFLAGS */
1856
Martin v. Löwis244edc82001-10-04 22:44:26 +00001857#ifdef HAVE_CHROOT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001858PyDoc_STRVAR(posix_chroot__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001859"chroot(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001860Change root directory to path.");
Martin v. Löwis244edc82001-10-04 22:44:26 +00001861
1862static PyObject *
1863posix_chroot(PyObject *self, PyObject *args)
1864{
Thomas Wouters477c8d52006-05-27 19:21:47 +00001865 return posix_1str(args, "et:chroot", chroot);
Martin v. Löwis244edc82001-10-04 22:44:26 +00001866}
1867#endif
1868
Guido van Rossum21142a01999-01-08 21:05:37 +00001869#ifdef HAVE_FSYNC
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001870PyDoc_STRVAR(posix_fsync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001871"fsync(fildes)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001872force write of file with filedescriptor to disk.");
Guido van Rossum21142a01999-01-08 21:05:37 +00001873
1874static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00001875posix_fsync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00001876{
Fred Drake4d1e64b2002-04-15 19:40:07 +00001877 return posix_fildes(fdobj, fsync);
Guido van Rossum21142a01999-01-08 21:05:37 +00001878}
1879#endif /* HAVE_FSYNC */
1880
1881#ifdef HAVE_FDATASYNC
Guido van Rossumecc23b02000-09-22 16:01:05 +00001882
Guido van Rossum7f58e2e2000-09-22 17:26:14 +00001883#ifdef __hpux
Guido van Rossumecc23b02000-09-22 16:01:05 +00001884extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
1885#endif
1886
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001887PyDoc_STRVAR(posix_fdatasync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001888"fdatasync(fildes)\n\n\
Guido van Rossum21142a01999-01-08 21:05:37 +00001889force write of file with filedescriptor to disk.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001890 does not force update of metadata.");
Guido van Rossum21142a01999-01-08 21:05:37 +00001891
1892static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00001893posix_fdatasync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00001894{
Fred Drake4d1e64b2002-04-15 19:40:07 +00001895 return posix_fildes(fdobj, fdatasync);
Guido van Rossum21142a01999-01-08 21:05:37 +00001896}
1897#endif /* HAVE_FDATASYNC */
1898
1899
Fredrik Lundh10723342000-07-10 16:38:09 +00001900#ifdef HAVE_CHOWN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001901PyDoc_STRVAR(posix_chown__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001902"chown(path, uid, gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001903Change the owner and group id of path to the numeric uid and gid.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001904
Barry Warsaw53699e91996-12-10 23:23:01 +00001905static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001906posix_chown(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001907{
Mark Hammondef8b6542001-05-13 08:04:26 +00001908 char *path = NULL;
Christian Heimesd5e2b6f2008-03-19 21:50:51 +00001909 long uid, gid;
Fredrik Lundh44328e62000-07-10 15:59:30 +00001910 int res;
Christian Heimesd5e2b6f2008-03-19 21:50:51 +00001911 if (!PyArg_ParseTuple(args, "etll:chown",
Tim Peters5aa91602002-01-30 05:46:57 +00001912 Py_FileSystemDefaultEncoding, &path,
Mark Hammondef8b6542001-05-13 08:04:26 +00001913 &uid, &gid))
Fredrik Lundh44328e62000-07-10 15:59:30 +00001914 return NULL;
1915 Py_BEGIN_ALLOW_THREADS
1916 res = chown(path, (uid_t) uid, (gid_t) gid);
1917 Py_END_ALLOW_THREADS
1918 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00001919 return posix_error_with_allocated_filename(path);
1920 PyMem_Free(path);
Fredrik Lundh44328e62000-07-10 15:59:30 +00001921 Py_INCREF(Py_None);
1922 return Py_None;
Guido van Rossumb6775db1994-08-01 11:34:53 +00001923}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001924#endif /* HAVE_CHOWN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00001925
Christian Heimes4e30a842007-11-30 22:12:06 +00001926#ifdef HAVE_FCHOWN
1927PyDoc_STRVAR(posix_fchown__doc__,
1928"fchown(fd, uid, gid)\n\n\
1929Change the owner and group id of the file given by file descriptor\n\
1930fd to the numeric uid and gid.");
1931
1932static PyObject *
1933posix_fchown(PyObject *self, PyObject *args)
1934{
1935 int fd, uid, gid;
1936 int res;
1937 if (!PyArg_ParseTuple(args, "iii:chown", &fd, &uid, &gid))
1938 return NULL;
1939 Py_BEGIN_ALLOW_THREADS
1940 res = fchown(fd, (uid_t) uid, (gid_t) gid);
1941 Py_END_ALLOW_THREADS
1942 if (res < 0)
1943 return posix_error();
1944 Py_RETURN_NONE;
1945}
1946#endif /* HAVE_FCHOWN */
1947
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00001948#ifdef HAVE_LCHOWN
1949PyDoc_STRVAR(posix_lchown__doc__,
1950"lchown(path, uid, gid)\n\n\
1951Change the owner and group id of path to the numeric uid and gid.\n\
1952This function will not follow symbolic links.");
1953
1954static PyObject *
1955posix_lchown(PyObject *self, PyObject *args)
1956{
1957 char *path = NULL;
1958 int uid, gid;
1959 int res;
1960 if (!PyArg_ParseTuple(args, "etii:lchown",
1961 Py_FileSystemDefaultEncoding, &path,
1962 &uid, &gid))
1963 return NULL;
1964 Py_BEGIN_ALLOW_THREADS
1965 res = lchown(path, (uid_t) uid, (gid_t) gid);
1966 Py_END_ALLOW_THREADS
Tim Peters11b23062003-04-23 02:39:17 +00001967 if (res < 0)
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00001968 return posix_error_with_allocated_filename(path);
1969 PyMem_Free(path);
1970 Py_INCREF(Py_None);
1971 return Py_None;
1972}
1973#endif /* HAVE_LCHOWN */
1974
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001975
Guido van Rossum36bc6801995-06-14 22:54:23 +00001976#ifdef HAVE_GETCWD
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001977PyDoc_STRVAR(posix_getcwd__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001978"getcwd() -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001979Return a string representing the current working directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001980
Barry Warsaw53699e91996-12-10 23:23:01 +00001981static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001982posix_getcwd(PyObject *self, PyObject *noargs)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001983{
Benjamin Petersondcf97b92008-07-02 17:30:14 +00001984 int bufsize_incr = 1024;
1985 int bufsize = 0;
1986 char *tmpbuf = NULL;
1987 char *res = NULL;
1988 PyObject *dynamic_return;
Neal Norwitze241ce82003-02-17 18:17:05 +00001989
Barry Warsaw53699e91996-12-10 23:23:01 +00001990 Py_BEGIN_ALLOW_THREADS
Benjamin Petersondcf97b92008-07-02 17:30:14 +00001991 do {
1992 bufsize = bufsize + bufsize_incr;
1993 tmpbuf = malloc(bufsize);
1994 if (tmpbuf == NULL) {
1995 break;
1996 }
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001997#if defined(PYOS_OS2) && defined(PYCC_GCC)
Benjamin Petersondcf97b92008-07-02 17:30:14 +00001998 res = _getcwd2(tmpbuf, bufsize);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001999#else
Benjamin Petersondcf97b92008-07-02 17:30:14 +00002000 res = getcwd(tmpbuf, bufsize);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002001#endif
Benjamin Petersondcf97b92008-07-02 17:30:14 +00002002
2003 if (res == NULL) {
2004 free(tmpbuf);
2005 }
2006 } while ((res == NULL) && (errno == ERANGE));
Barry Warsaw53699e91996-12-10 23:23:01 +00002007 Py_END_ALLOW_THREADS
Benjamin Petersondcf97b92008-07-02 17:30:14 +00002008
Guido van Rossumff4949e1992-08-05 19:58:53 +00002009 if (res == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002010 return posix_error();
Benjamin Petersondcf97b92008-07-02 17:30:14 +00002011
2012 dynamic_return = PyUnicode_FromString(tmpbuf);
2013 free(tmpbuf);
2014
2015 return dynamic_return;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002016}
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002017
2018PyDoc_STRVAR(posix_getcwdu__doc__,
2019"getcwdu() -> path\n\n\
2020Return a unicode string representing the current working directory.");
2021
2022static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002023posix_getcwdu(PyObject *self, PyObject *noargs)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002024{
2025 char buf[1026];
2026 char *res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002027
2028#ifdef Py_WIN_WIDE_FILENAMES
Thomas Wouters477c8d52006-05-27 19:21:47 +00002029 DWORD len;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002030 if (unicode_file_names()) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002031 wchar_t wbuf[1026];
Thomas Wouters477c8d52006-05-27 19:21:47 +00002032 wchar_t *wbuf2 = wbuf;
2033 PyObject *resobj;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002034 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002035 len = GetCurrentDirectoryW(sizeof wbuf/ sizeof wbuf[0], wbuf);
2036 /* If the buffer is large enough, len does not include the
2037 terminating \0. If the buffer is too small, len includes
2038 the space needed for the terminator. */
2039 if (len >= sizeof wbuf/ sizeof wbuf[0]) {
2040 wbuf2 = malloc(len * sizeof(wchar_t));
2041 if (wbuf2)
2042 len = GetCurrentDirectoryW(len, wbuf2);
2043 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002044 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002045 if (!wbuf2) {
2046 PyErr_NoMemory();
2047 return NULL;
2048 }
2049 if (!len) {
2050 if (wbuf2 != wbuf) free(wbuf2);
2051 return win32_error("getcwdu", NULL);
2052 }
2053 resobj = PyUnicode_FromWideChar(wbuf2, len);
2054 if (wbuf2 != wbuf) free(wbuf2);
2055 return resobj;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002056 }
2057#endif
2058
2059 Py_BEGIN_ALLOW_THREADS
2060#if defined(PYOS_OS2) && defined(PYCC_GCC)
2061 res = _getcwd2(buf, sizeof buf);
2062#else
2063 res = getcwd(buf, sizeof buf);
2064#endif
2065 Py_END_ALLOW_THREADS
2066 if (res == NULL)
2067 return posix_error();
2068 return PyUnicode_Decode(buf, strlen(buf), Py_FileSystemDefaultEncoding,"strict");
2069}
Guido van Rossum36bc6801995-06-14 22:54:23 +00002070#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002071
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002072
Guido van Rossumb6775db1994-08-01 11:34:53 +00002073#ifdef HAVE_LINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002074PyDoc_STRVAR(posix_link__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002075"link(src, dst)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002076Create a hard link to a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002077
Barry Warsaw53699e91996-12-10 23:23:01 +00002078static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002079posix_link(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002080{
Thomas Wouters477c8d52006-05-27 19:21:47 +00002081 return posix_2str(args, "etet:link", link);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002082}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002083#endif /* HAVE_LINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002084
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002085
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002086PyDoc_STRVAR(posix_listdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002087"listdir(path) -> list_of_strings\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002088Return a list containing the names of the entries in the directory.\n\
2089\n\
2090 path: path of directory to list\n\
2091\n\
2092The list is in arbitrary order. It does not include the special\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002093entries '.' and '..' even if they are present in the directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002094
Barry Warsaw53699e91996-12-10 23:23:01 +00002095static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002096posix_listdir(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002097{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002098 /* XXX Should redo this putting the (now four) versions of opendir
Guido van Rossum6d8841c1997-08-14 19:57:39 +00002099 in separate files instead of having them all here... */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002100#if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002101
Barry Warsaw53699e91996-12-10 23:23:01 +00002102 PyObject *d, *v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002103 HANDLE hFindFile;
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002104 BOOL result;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002105 WIN32_FIND_DATA FileData;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002106 char namebuf[MAX_PATH+5]; /* Overallocate for \\*.*\0 */
Mark Hammondef8b6542001-05-13 08:04:26 +00002107 char *bufptr = namebuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002108 Py_ssize_t len = sizeof(namebuf)-5; /* only claim to have space for MAX_PATH */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002109
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002110#ifdef Py_WIN_WIDE_FILENAMES
2111 /* If on wide-character-capable OS see if argument
2112 is Unicode and if so use wide API. */
2113 if (unicode_file_names()) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002114 PyObject *po;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002115 if (PyArg_ParseTuple(args, "U:listdir", &po)) {
2116 WIN32_FIND_DATAW wFileData;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002117 Py_UNICODE *wnamebuf;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002118 Py_UNICODE wch;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002119 /* Overallocate for \\*.*\0 */
2120 len = PyUnicode_GET_SIZE(po);
2121 wnamebuf = malloc((len + 5) * sizeof(wchar_t));
2122 if (!wnamebuf) {
2123 PyErr_NoMemory();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002124 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002125 }
2126 wcscpy(wnamebuf, PyUnicode_AS_UNICODE(po));
2127 wch = len > 0 ? wnamebuf[len-1] : '\0';
2128 if (wch != L'/' && wch != L'\\' && wch != L':')
2129 wnamebuf[len++] = L'\\';
2130 wcscpy(wnamebuf + len, L"*.*");
2131 if ((d = PyList_New(0)) == NULL) {
2132 free(wnamebuf);
2133 return NULL;
2134 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002135 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
2136 if (hFindFile == INVALID_HANDLE_VALUE) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002137 int error = GetLastError();
2138 if (error == ERROR_FILE_NOT_FOUND) {
2139 free(wnamebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002140 return d;
2141 }
2142 Py_DECREF(d);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002143 win32_error_unicode("FindFirstFileW", wnamebuf);
2144 free(wnamebuf);
2145 return NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002146 }
2147 do {
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002148 /* Skip over . and .. */
2149 if (wcscmp(wFileData.cFileName, L".") != 0 &&
2150 wcscmp(wFileData.cFileName, L"..") != 0) {
2151 v = PyUnicode_FromUnicode(wFileData.cFileName, wcslen(wFileData.cFileName));
2152 if (v == NULL) {
2153 Py_DECREF(d);
2154 d = NULL;
2155 break;
2156 }
2157 if (PyList_Append(d, v) != 0) {
2158 Py_DECREF(v);
2159 Py_DECREF(d);
2160 d = NULL;
2161 break;
2162 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002163 Py_DECREF(v);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002164 }
Georg Brandl622927b2006-03-07 12:48:03 +00002165 Py_BEGIN_ALLOW_THREADS
2166 result = FindNextFileW(hFindFile, &wFileData);
2167 Py_END_ALLOW_THREADS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002168 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2169 it got to the end of the directory. */
2170 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2171 Py_DECREF(d);
2172 win32_error_unicode("FindNextFileW", wnamebuf);
2173 FindClose(hFindFile);
2174 free(wnamebuf);
2175 return NULL;
2176 }
Georg Brandl622927b2006-03-07 12:48:03 +00002177 } while (result == TRUE);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002178
2179 if (FindClose(hFindFile) == FALSE) {
2180 Py_DECREF(d);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002181 win32_error_unicode("FindClose", wnamebuf);
2182 free(wnamebuf);
2183 return NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002184 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002185 free(wnamebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002186 return d;
2187 }
2188 /* Drop the argument parsing error as narrow strings
2189 are also valid. */
2190 PyErr_Clear();
2191 }
2192#endif
2193
Tim Peters5aa91602002-01-30 05:46:57 +00002194 if (!PyArg_ParseTuple(args, "et#:listdir",
Mark Hammondef8b6542001-05-13 08:04:26 +00002195 Py_FileSystemDefaultEncoding, &bufptr, &len))
Guido van Rossumb6775db1994-08-01 11:34:53 +00002196 return NULL;
Neil Schemenauer94b866a2002-03-22 20:51:58 +00002197 if (len > 0) {
2198 char ch = namebuf[len-1];
2199 if (ch != SEP && ch != ALTSEP && ch != ':')
2200 namebuf[len++] = '/';
2201 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002202 strcpy(namebuf + len, "*.*");
2203
Barry Warsaw53699e91996-12-10 23:23:01 +00002204 if ((d = PyList_New(0)) == NULL)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002205 return NULL;
2206
2207 hFindFile = FindFirstFile(namebuf, &FileData);
2208 if (hFindFile == INVALID_HANDLE_VALUE) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002209 int error = GetLastError();
2210 if (error == ERROR_FILE_NOT_FOUND)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002211 return d;
2212 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00002213 return win32_error("FindFirstFile", namebuf);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002214 }
2215 do {
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002216 /* Skip over . and .. */
2217 if (strcmp(FileData.cFileName, ".") != 0 &&
2218 strcmp(FileData.cFileName, "..") != 0) {
Christian Heimes72b710a2008-05-26 13:28:38 +00002219 v = PyBytes_FromString(FileData.cFileName);
Martin v. Löwise920f0d2006-03-07 23:59:33 +00002220 if (v == NULL) {
2221 Py_DECREF(d);
2222 d = NULL;
2223 break;
2224 }
2225 if (PyList_Append(d, v) != 0) {
2226 Py_DECREF(v);
2227 Py_DECREF(d);
2228 d = NULL;
2229 break;
2230 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002231 Py_DECREF(v);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002232 }
Georg Brandl622927b2006-03-07 12:48:03 +00002233 Py_BEGIN_ALLOW_THREADS
2234 result = FindNextFile(hFindFile, &FileData);
2235 Py_END_ALLOW_THREADS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002236 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2237 it got to the end of the directory. */
2238 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2239 Py_DECREF(d);
2240 win32_error("FindNextFile", namebuf);
2241 FindClose(hFindFile);
2242 return NULL;
2243 }
Georg Brandl622927b2006-03-07 12:48:03 +00002244 } while (result == TRUE);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002245
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002246 if (FindClose(hFindFile) == FALSE) {
2247 Py_DECREF(d);
Mark Hammondef8b6542001-05-13 08:04:26 +00002248 return win32_error("FindClose", namebuf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002249 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002250
2251 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002252
Tim Peters0bb44a42000-09-15 07:44:49 +00002253#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002254
2255#ifndef MAX_PATH
2256#define MAX_PATH CCHMAXPATH
2257#endif
2258 char *name, *pt;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00002259 Py_ssize_t len;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002260 PyObject *d, *v;
2261 char namebuf[MAX_PATH+5];
2262 HDIR hdir = 1;
2263 ULONG srchcnt = 1;
2264 FILEFINDBUF3 ep;
2265 APIRET rc;
2266
Alexandre Vassalotti70a23712007-10-14 02:05:51 +00002267 if (!PyArg_ParseTuple(args, "et#:listdir",
2268 Py_FileSystemDefaultEncoding, &name, &len))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002269 return NULL;
2270 if (len >= MAX_PATH) {
Neal Norwitz6c913782007-10-14 03:23:09 +00002271 PyMem_Free(name);
2272 PyErr_SetString(PyExc_ValueError, "path too long");
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002273 return NULL;
2274 }
2275 strcpy(namebuf, name);
2276 for (pt = namebuf; *pt; pt++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002277 if (*pt == ALTSEP)
2278 *pt = SEP;
2279 if (namebuf[len-1] != SEP)
2280 namebuf[len++] = SEP;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002281 strcpy(namebuf + len, "*.*");
2282
Neal Norwitz6c913782007-10-14 03:23:09 +00002283 if ((d = PyList_New(0)) == NULL) {
Alexandre Vassalotti4167ebc2007-10-14 02:54:41 +00002284 PyMem_Free(name);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002285 return NULL;
Alexandre Vassalotti4167ebc2007-10-14 02:54:41 +00002286 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002287
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002288 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
2289 &hdir, /* Handle to Use While Search Directory */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002290 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002291 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
2292 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
2293 FIL_STANDARD); /* Format of Entry (EAs or Not) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002294
2295 if (rc != NO_ERROR) {
2296 errno = ENOENT;
Neal Norwitz6c913782007-10-14 03:23:09 +00002297 return posix_error_with_allocated_filename(name);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002298 }
2299
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002300 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002301 do {
2302 if (ep.achName[0] == '.'
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002303 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002304 continue; /* Skip Over "." and ".." Names */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002305
2306 strcpy(namebuf, ep.achName);
2307
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002308 /* Leave Case of Name Alone -- In Native Form */
2309 /* (Removed Forced Lowercasing Code) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002310
Christian Heimes72b710a2008-05-26 13:28:38 +00002311 v = PyBytes_FromString(namebuf);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002312 if (v == NULL) {
2313 Py_DECREF(d);
2314 d = NULL;
2315 break;
2316 }
2317 if (PyList_Append(d, v) != 0) {
2318 Py_DECREF(v);
2319 Py_DECREF(d);
2320 d = NULL;
2321 break;
2322 }
2323 Py_DECREF(v);
2324 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
2325 }
2326
Alexandre Vassalotti4167ebc2007-10-14 02:54:41 +00002327 PyMem_Free(name);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002328 return d;
2329#else
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002330
Just van Rossum2fe07fd2003-03-03 19:07:13 +00002331 char *name = NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002332 PyObject *d, *v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002333 DIR *dirp;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002334 struct dirent *ep;
Just van Rossum96b1c902003-03-03 17:32:15 +00002335 int arg_is_unicode = 1;
2336
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002337 errno = 0;
Just van Rossum96b1c902003-03-03 17:32:15 +00002338 if (!PyArg_ParseTuple(args, "U:listdir", &v)) {
2339 arg_is_unicode = 0;
2340 PyErr_Clear();
2341 }
2342 if (!PyArg_ParseTuple(args, "et:listdir", Py_FileSystemDefaultEncoding, &name))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002343 return NULL;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002344 if ((dirp = opendir(name)) == NULL) {
Just van Rossum2fe07fd2003-03-03 19:07:13 +00002345 return posix_error_with_allocated_filename(name);
Guido van Rossumff4949e1992-08-05 19:58:53 +00002346 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002347 if ((d = PyList_New(0)) == NULL) {
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002348 closedir(dirp);
Just van Rossum2fe07fd2003-03-03 19:07:13 +00002349 PyMem_Free(name);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002350 return NULL;
2351 }
Georg Brandl622927b2006-03-07 12:48:03 +00002352 for (;;) {
2353 Py_BEGIN_ALLOW_THREADS
2354 ep = readdir(dirp);
2355 Py_END_ALLOW_THREADS
2356 if (ep == NULL)
2357 break;
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002358 if (ep->d_name[0] == '.' &&
2359 (NAMLEN(ep) == 1 ||
Guido van Rossuma376cc51996-12-05 23:43:35 +00002360 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002361 continue;
Christian Heimes72b710a2008-05-26 13:28:38 +00002362 v = PyBytes_FromStringAndSize(ep->d_name, NAMLEN(ep));
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002363 if (v == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00002364 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002365 d = NULL;
2366 break;
2367 }
Just van Rossum96b1c902003-03-03 17:32:15 +00002368 if (arg_is_unicode) {
Just van Rossum46c97842003-02-25 21:42:15 +00002369 PyObject *w;
2370
2371 w = PyUnicode_FromEncodedObject(v,
Tim Peters11b23062003-04-23 02:39:17 +00002372 Py_FileSystemDefaultEncoding,
Just van Rossum46c97842003-02-25 21:42:15 +00002373 "strict");
Just van Rossum6a421832003-03-04 19:30:44 +00002374 if (w != NULL) {
2375 Py_DECREF(v);
2376 v = w;
2377 }
2378 else {
2379 /* fall back to the original byte string, as
2380 discussed in patch #683592 */
2381 PyErr_Clear();
Just van Rossum46c97842003-02-25 21:42:15 +00002382 }
Just van Rossum46c97842003-02-25 21:42:15 +00002383 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002384 if (PyList_Append(d, v) != 0) {
2385 Py_DECREF(v);
2386 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002387 d = NULL;
2388 break;
2389 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002390 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002391 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002392 if (errno != 0 && d != NULL) {
2393 /* readdir() returned NULL and set errno */
2394 closedir(dirp);
2395 Py_DECREF(d);
2396 return posix_error_with_allocated_filename(name);
2397 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002398 closedir(dirp);
Just van Rossum2fe07fd2003-03-03 19:07:13 +00002399 PyMem_Free(name);
Guido van Rossum0ee42cd1991-04-08 21:01:03 +00002400
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002401 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002402
Tim Peters0bb44a42000-09-15 07:44:49 +00002403#endif /* which OS */
2404} /* end of posix_listdir */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002405
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002406#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00002407/* A helper function for abspath on win32 */
2408static PyObject *
2409posix__getfullpathname(PyObject *self, PyObject *args)
2410{
2411 /* assume encoded strings wont more than double no of chars */
2412 char inbuf[MAX_PATH*2];
2413 char *inbufp = inbuf;
Tim Peters67d70eb2006-03-01 04:35:45 +00002414 Py_ssize_t insize = sizeof(inbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00002415 char outbuf[MAX_PATH*2];
2416 char *temp;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002417#ifdef Py_WIN_WIDE_FILENAMES
2418 if (unicode_file_names()) {
2419 PyUnicodeObject *po;
2420 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po)) {
2421 Py_UNICODE woutbuf[MAX_PATH*2];
2422 Py_UNICODE *wtemp;
Tim Peters11b23062003-04-23 02:39:17 +00002423 if (!GetFullPathNameW(PyUnicode_AS_UNICODE(po),
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002424 sizeof(woutbuf)/sizeof(woutbuf[0]),
2425 woutbuf, &wtemp))
2426 return win32_error("GetFullPathName", "");
2427 return PyUnicode_FromUnicode(woutbuf, wcslen(woutbuf));
2428 }
2429 /* Drop the argument parsing error as narrow strings
2430 are also valid. */
2431 PyErr_Clear();
2432 }
2433#endif
Tim Peters5aa91602002-01-30 05:46:57 +00002434 if (!PyArg_ParseTuple (args, "et#:_getfullpathname",
2435 Py_FileSystemDefaultEncoding, &inbufp,
Mark Hammondef8b6542001-05-13 08:04:26 +00002436 &insize))
2437 return NULL;
2438 if (!GetFullPathName(inbuf, sizeof(outbuf)/sizeof(outbuf[0]),
2439 outbuf, &temp))
2440 return win32_error("GetFullPathName", inbuf);
Martin v. Löwis969297f2004-06-15 18:49:58 +00002441 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
2442 return PyUnicode_Decode(outbuf, strlen(outbuf),
2443 Py_FileSystemDefaultEncoding, NULL);
2444 }
Christian Heimes72b710a2008-05-26 13:28:38 +00002445 return PyBytes_FromString(outbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00002446} /* end of posix__getfullpathname */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002447#endif /* MS_WINDOWS */
Mark Hammondef8b6542001-05-13 08:04:26 +00002448
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002449PyDoc_STRVAR(posix_mkdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002450"mkdir(path [, mode=0777])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002451Create a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002452
Barry Warsaw53699e91996-12-10 23:23:01 +00002453static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002454posix_mkdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002455{
Guido van Rossumb0824db1996-02-25 04:50:32 +00002456 int res;
Mark Hammondef8b6542001-05-13 08:04:26 +00002457 char *path = NULL;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002458 int mode = 0777;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002459
2460#ifdef Py_WIN_WIDE_FILENAMES
2461 if (unicode_file_names()) {
2462 PyUnicodeObject *po;
Mark Hammond05107b62003-02-19 04:08:27 +00002463 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode)) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002464 Py_BEGIN_ALLOW_THREADS
2465 /* PyUnicode_AS_UNICODE OK without thread lock as
2466 it is a simple dereference. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002467 res = CreateDirectoryW(PyUnicode_AS_UNICODE(po), NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002468 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002469 if (!res)
2470 return win32_error_unicode("mkdir", PyUnicode_AS_UNICODE(po));
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002471 Py_INCREF(Py_None);
2472 return Py_None;
2473 }
2474 /* Drop the argument parsing error as narrow strings
2475 are also valid. */
2476 PyErr_Clear();
2477 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002478 if (!PyArg_ParseTuple(args, "et|i:mkdir",
2479 Py_FileSystemDefaultEncoding, &path, &mode))
2480 return NULL;
2481 Py_BEGIN_ALLOW_THREADS
2482 /* PyUnicode_AS_UNICODE OK without thread lock as
2483 it is a simple dereference. */
2484 res = CreateDirectoryA(path, NULL);
2485 Py_END_ALLOW_THREADS
2486 if (!res) {
2487 win32_error("mkdir", path);
2488 PyMem_Free(path);
2489 return NULL;
2490 }
2491 PyMem_Free(path);
2492 Py_INCREF(Py_None);
2493 return Py_None;
2494#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002495
Tim Peters5aa91602002-01-30 05:46:57 +00002496 if (!PyArg_ParseTuple(args, "et|i:mkdir",
Mark Hammondef8b6542001-05-13 08:04:26 +00002497 Py_FileSystemDefaultEncoding, &path, &mode))
Guido van Rossumb0824db1996-02-25 04:50:32 +00002498 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002499 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002500#if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002501 res = mkdir(path);
2502#else
Guido van Rossumb0824db1996-02-25 04:50:32 +00002503 res = mkdir(path, mode);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002504#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002505 Py_END_ALLOW_THREADS
Guido van Rossumb0824db1996-02-25 04:50:32 +00002506 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00002507 return posix_error_with_allocated_filename(path);
2508 PyMem_Free(path);
Barry Warsaw53699e91996-12-10 23:23:01 +00002509 Py_INCREF(Py_None);
2510 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002511#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002512}
2513
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002514
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002515/* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
2516#if defined(HAVE_SYS_RESOURCE_H)
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002517#include <sys/resource.h>
2518#endif
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002519
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002520
2521#ifdef HAVE_NICE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002522PyDoc_STRVAR(posix_nice__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002523"nice(inc) -> new_priority\n\n\
2524Decrease the priority of process by inc and return the new priority.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002525
Barry Warsaw53699e91996-12-10 23:23:01 +00002526static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002527posix_nice(PyObject *self, PyObject *args)
Guido van Rossum775f4da1993-01-09 17:18:52 +00002528{
2529 int increment, value;
2530
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002531 if (!PyArg_ParseTuple(args, "i:nice", &increment))
Guido van Rossum775f4da1993-01-09 17:18:52 +00002532 return NULL;
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002533
2534 /* There are two flavours of 'nice': one that returns the new
2535 priority (as required by almost all standards out there) and the
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002536 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
2537 the use of getpriority() to get the new priority.
Tim Peters5aa91602002-01-30 05:46:57 +00002538
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002539 If we are of the nice family that returns the new priority, we
2540 need to clear errno before the call, and check if errno is filled
2541 before calling posix_error() on a returnvalue of -1, because the
2542 -1 may be the actual new priority! */
2543
2544 errno = 0;
Guido van Rossum775f4da1993-01-09 17:18:52 +00002545 value = nice(increment);
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002546#if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002547 if (value == 0)
2548 value = getpriority(PRIO_PROCESS, 0);
2549#endif
2550 if (value == -1 && errno != 0)
2551 /* either nice() or getpriority() returned an error */
Guido van Rossum775f4da1993-01-09 17:18:52 +00002552 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00002553 return PyLong_FromLong((long) value);
Guido van Rossum775f4da1993-01-09 17:18:52 +00002554}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002555#endif /* HAVE_NICE */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002556
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002557PyDoc_STRVAR(posix_rename__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002558"rename(old, new)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002559Rename a file or directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002560
Barry Warsaw53699e91996-12-10 23:23:01 +00002561static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002562posix_rename(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002563{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002564#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002565 PyObject *o1, *o2;
2566 char *p1, *p2;
2567 BOOL result;
2568 if (unicode_file_names()) {
2569 if (!PyArg_ParseTuple(args, "O&O&:rename",
2570 convert_to_unicode, &o1,
2571 convert_to_unicode, &o2))
2572 PyErr_Clear();
2573 else {
2574 Py_BEGIN_ALLOW_THREADS
2575 result = MoveFileW(PyUnicode_AsUnicode(o1),
2576 PyUnicode_AsUnicode(o2));
2577 Py_END_ALLOW_THREADS
2578 Py_DECREF(o1);
2579 Py_DECREF(o2);
2580 if (!result)
2581 return win32_error("rename", NULL);
2582 Py_INCREF(Py_None);
2583 return Py_None;
2584 }
2585 }
2586 if (!PyArg_ParseTuple(args, "ss:rename", &p1, &p2))
2587 return NULL;
2588 Py_BEGIN_ALLOW_THREADS
2589 result = MoveFileA(p1, p2);
2590 Py_END_ALLOW_THREADS
2591 if (!result)
2592 return win32_error("rename", NULL);
2593 Py_INCREF(Py_None);
2594 return Py_None;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002595#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002596 return posix_2str(args, "etet:rename", rename);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002597#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002598}
2599
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002600
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002601PyDoc_STRVAR(posix_rmdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002602"rmdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002603Remove a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002604
Barry Warsaw53699e91996-12-10 23:23:01 +00002605static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002606posix_rmdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002607{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002608#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002609 return win32_1str(args, "rmdir", "s:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002610#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002611 return posix_1str(args, "et:rmdir", rmdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002612#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002613}
2614
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002615
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002616PyDoc_STRVAR(posix_stat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002617"stat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002618Perform a stat system call on the given path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002619
Barry Warsaw53699e91996-12-10 23:23:01 +00002620static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002621posix_stat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002622{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002623#ifdef MS_WINDOWS
Martin v. Löwis14694662006-02-03 12:54:16 +00002624 return posix_do_stat(self, args, "et:stat", STAT, "U:stat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002625#else
2626 return posix_do_stat(self, args, "et:stat", STAT, NULL, NULL);
2627#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002628}
2629
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002630
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002631#ifdef HAVE_SYSTEM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002632PyDoc_STRVAR(posix_system__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002633"system(command) -> exit_status\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002634Execute the command (a string) in a subshell.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002635
Barry Warsaw53699e91996-12-10 23:23:01 +00002636static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002637posix_system(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002638{
Guido van Rossumff4949e1992-08-05 19:58:53 +00002639 long sts;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002640#ifdef MS_WINDOWS
2641 wchar_t *command;
2642 if (!PyArg_ParseTuple(args, "u:system", &command))
2643 return NULL;
2644#else
2645 char *command;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002646 if (!PyArg_ParseTuple(args, "s:system", &command))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002647 return NULL;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002648#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002649 Py_BEGIN_ALLOW_THREADS
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002650#ifdef MS_WINDOWS
2651 sts = _wsystem(command);
2652#else
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002653 sts = system(command);
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002654#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002655 Py_END_ALLOW_THREADS
Christian Heimes217cfd12007-12-02 14:31:20 +00002656 return PyLong_FromLong(sts);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002657}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002658#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002659
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002660
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002661PyDoc_STRVAR(posix_umask__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002662"umask(new_mask) -> old_mask\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002663Set the current numeric umask and return the previous umask.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002664
Barry Warsaw53699e91996-12-10 23:23:01 +00002665static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002666posix_umask(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002667{
2668 int i;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002669 if (!PyArg_ParseTuple(args, "i:umask", &i))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002670 return NULL;
Fred Drake0368bc42001-07-19 20:48:32 +00002671 i = (int)umask(i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002672 if (i < 0)
2673 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00002674 return PyLong_FromLong((long)i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002675}
2676
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002677
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002678PyDoc_STRVAR(posix_unlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002679"unlink(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002680Remove a file (same as remove(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002681
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002682PyDoc_STRVAR(posix_remove__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002683"remove(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002684Remove a file (same as unlink(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002685
Barry Warsaw53699e91996-12-10 23:23:01 +00002686static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002687posix_unlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002688{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002689#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002690 return win32_1str(args, "remove", "s:remove", DeleteFileA, "U:remove", DeleteFileW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002691#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002692 return posix_1str(args, "et:remove", unlink);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002693#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002694}
2695
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002696
Guido van Rossumb6775db1994-08-01 11:34:53 +00002697#ifdef HAVE_UNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002698PyDoc_STRVAR(posix_uname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002699"uname() -> (sysname, nodename, release, version, machine)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002700Return a tuple identifying the current operating system.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002701
Barry Warsaw53699e91996-12-10 23:23:01 +00002702static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002703posix_uname(PyObject *self, PyObject *noargs)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002704{
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002705 struct utsname u;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002706 int res;
Neal Norwitze241ce82003-02-17 18:17:05 +00002707
Barry Warsaw53699e91996-12-10 23:23:01 +00002708 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002709 res = uname(&u);
Barry Warsaw53699e91996-12-10 23:23:01 +00002710 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002711 if (res < 0)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002712 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00002713 return Py_BuildValue("(sssss)",
Barry Warsaw43d68b81996-12-19 22:10:44 +00002714 u.sysname,
2715 u.nodename,
2716 u.release,
2717 u.version,
2718 u.machine);
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002719}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002720#endif /* HAVE_UNAME */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002721
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002722static int
2723extract_time(PyObject *t, long* sec, long* usec)
2724{
2725 long intval;
2726 if (PyFloat_Check(t)) {
2727 double tval = PyFloat_AsDouble(t);
Christian Heimes90aa7642007-12-19 02:45:37 +00002728 PyObject *intobj = Py_TYPE(t)->tp_as_number->nb_int(t);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002729 if (!intobj)
2730 return -1;
Christian Heimes217cfd12007-12-02 14:31:20 +00002731 intval = PyLong_AsLong(intobj);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002732 Py_DECREF(intobj);
Michael W. Hudsonb8963812005-07-05 15:21:58 +00002733 if (intval == -1 && PyErr_Occurred())
2734 return -1;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002735 *sec = intval;
Tim Peters96940cf2002-09-10 15:37:28 +00002736 *usec = (long)((tval - intval) * 1e6); /* can't exceed 1000000 */
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002737 if (*usec < 0)
2738 /* If rounding gave us a negative number,
2739 truncate. */
2740 *usec = 0;
2741 return 0;
2742 }
Christian Heimes217cfd12007-12-02 14:31:20 +00002743 intval = PyLong_AsLong(t);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002744 if (intval == -1 && PyErr_Occurred())
2745 return -1;
2746 *sec = intval;
2747 *usec = 0;
Martin v. Löwis076b2092002-09-10 15:04:41 +00002748 return 0;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002749}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002750
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002751PyDoc_STRVAR(posix_utime__doc__,
Thomas Wouters477c8d52006-05-27 19:21:47 +00002752"utime(path, (atime, mtime))\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00002753utime(path, None)\n\n\
Barry Warsaw3cef8562000-05-01 16:17:24 +00002754Set the access and modified time of the file to the given values. If the\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002755second form is used, set the access and modified times to the current time.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002756
Barry Warsaw53699e91996-12-10 23:23:01 +00002757static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002758posix_utime(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002759{
Thomas Wouters477c8d52006-05-27 19:21:47 +00002760#ifdef Py_WIN_WIDE_FILENAMES
2761 PyObject *arg;
2762 PyUnicodeObject *obwpath;
2763 wchar_t *wpath = NULL;
2764 char *apath = NULL;
2765 HANDLE hFile;
2766 long atimesec, mtimesec, ausec, musec;
2767 FILETIME atime, mtime;
2768 PyObject *result = NULL;
2769
2770 if (unicode_file_names()) {
2771 if (PyArg_ParseTuple(args, "UO|:utime", &obwpath, &arg)) {
2772 wpath = PyUnicode_AS_UNICODE(obwpath);
2773 Py_BEGIN_ALLOW_THREADS
2774 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002775 NULL, OPEN_EXISTING,
2776 FILE_FLAG_BACKUP_SEMANTICS, NULL);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002777 Py_END_ALLOW_THREADS
2778 if (hFile == INVALID_HANDLE_VALUE)
2779 return win32_error_unicode("utime", wpath);
2780 } else
2781 /* Drop the argument parsing error as narrow strings
2782 are also valid. */
2783 PyErr_Clear();
2784 }
2785 if (!wpath) {
2786 if (!PyArg_ParseTuple(args, "etO:utime",
2787 Py_FileSystemDefaultEncoding, &apath, &arg))
2788 return NULL;
2789 Py_BEGIN_ALLOW_THREADS
2790 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002791 NULL, OPEN_EXISTING,
2792 FILE_FLAG_BACKUP_SEMANTICS, NULL);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002793 Py_END_ALLOW_THREADS
2794 if (hFile == INVALID_HANDLE_VALUE) {
2795 win32_error("utime", apath);
2796 PyMem_Free(apath);
2797 return NULL;
2798 }
2799 PyMem_Free(apath);
2800 }
2801
2802 if (arg == Py_None) {
2803 SYSTEMTIME now;
2804 GetSystemTime(&now);
2805 if (!SystemTimeToFileTime(&now, &mtime) ||
2806 !SystemTimeToFileTime(&now, &atime)) {
2807 win32_error("utime", NULL);
2808 goto done;
2809 }
2810 }
2811 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
2812 PyErr_SetString(PyExc_TypeError,
2813 "utime() arg 2 must be a tuple (atime, mtime)");
2814 goto done;
2815 }
2816 else {
2817 if (extract_time(PyTuple_GET_ITEM(arg, 0),
2818 &atimesec, &ausec) == -1)
2819 goto done;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002820 time_t_to_FILE_TIME(atimesec, 1000*ausec, &atime);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002821 if (extract_time(PyTuple_GET_ITEM(arg, 1),
2822 &mtimesec, &musec) == -1)
2823 goto done;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002824 time_t_to_FILE_TIME(mtimesec, 1000*musec, &mtime);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002825 }
2826 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
2827 /* Avoid putting the file name into the error here,
2828 as that may confuse the user into believing that
2829 something is wrong with the file, when it also
2830 could be the time stamp that gives a problem. */
2831 win32_error("utime", NULL);
2832 }
2833 Py_INCREF(Py_None);
2834 result = Py_None;
2835done:
2836 CloseHandle(hFile);
2837 return result;
2838#else /* Py_WIN_WIDE_FILENAMES */
2839
Neal Norwitz2adf2102004-06-09 01:46:02 +00002840 char *path = NULL;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002841 long atime, mtime, ausec, musec;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002842 int res;
Barry Warsaw3cef8562000-05-01 16:17:24 +00002843 PyObject* arg;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002844
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002845#if defined(HAVE_UTIMES)
2846 struct timeval buf[2];
2847#define ATIME buf[0].tv_sec
2848#define MTIME buf[1].tv_sec
2849#elif defined(HAVE_UTIME_H)
Guido van Rossum6d8841c1997-08-14 19:57:39 +00002850/* XXX should define struct utimbuf instead, above */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002851 struct utimbuf buf;
2852#define ATIME buf.actime
2853#define MTIME buf.modtime
2854#define UTIME_ARG &buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002855#else /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002856 time_t buf[2];
2857#define ATIME buf[0]
2858#define MTIME buf[1]
2859#define UTIME_ARG buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002860#endif /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002861
Mark Hammond817c9292003-12-03 01:22:38 +00002862
Thomas Wouters477c8d52006-05-27 19:21:47 +00002863 if (!PyArg_ParseTuple(args, "etO:utime",
Hye-Shik Chang2b2c9732004-01-04 13:54:25 +00002864 Py_FileSystemDefaultEncoding, &path, &arg))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002865 return NULL;
Barry Warsaw3cef8562000-05-01 16:17:24 +00002866 if (arg == Py_None) {
2867 /* optional time values not given */
2868 Py_BEGIN_ALLOW_THREADS
2869 res = utime(path, NULL);
2870 Py_END_ALLOW_THREADS
2871 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002872 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
Barry Warsaw3cef8562000-05-01 16:17:24 +00002873 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00002874 "utime() arg 2 must be a tuple (atime, mtime)");
Neal Norwitz96652712004-06-06 20:40:27 +00002875 PyMem_Free(path);
Barry Warsaw3cef8562000-05-01 16:17:24 +00002876 return NULL;
2877 }
2878 else {
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002879 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Neal Norwitz96652712004-06-06 20:40:27 +00002880 &atime, &ausec) == -1) {
2881 PyMem_Free(path);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002882 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00002883 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002884 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Neal Norwitz96652712004-06-06 20:40:27 +00002885 &mtime, &musec) == -1) {
2886 PyMem_Free(path);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002887 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00002888 }
Barry Warsaw3cef8562000-05-01 16:17:24 +00002889 ATIME = atime;
2890 MTIME = mtime;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002891#ifdef HAVE_UTIMES
2892 buf[0].tv_usec = ausec;
2893 buf[1].tv_usec = musec;
2894 Py_BEGIN_ALLOW_THREADS
2895 res = utimes(path, buf);
2896 Py_END_ALLOW_THREADS
2897#else
Barry Warsaw3cef8562000-05-01 16:17:24 +00002898 Py_BEGIN_ALLOW_THREADS
2899 res = utime(path, UTIME_ARG);
2900 Py_END_ALLOW_THREADS
Mark Hammond817c9292003-12-03 01:22:38 +00002901#endif /* HAVE_UTIMES */
Barry Warsaw3cef8562000-05-01 16:17:24 +00002902 }
Mark Hammond2d5914b2004-05-04 08:10:37 +00002903 if (res < 0) {
Neal Norwitz96652712004-06-06 20:40:27 +00002904 return posix_error_with_allocated_filename(path);
Mark Hammond2d5914b2004-05-04 08:10:37 +00002905 }
Neal Norwitz96652712004-06-06 20:40:27 +00002906 PyMem_Free(path);
Barry Warsaw53699e91996-12-10 23:23:01 +00002907 Py_INCREF(Py_None);
2908 return Py_None;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002909#undef UTIME_ARG
2910#undef ATIME
2911#undef MTIME
Thomas Wouters477c8d52006-05-27 19:21:47 +00002912#endif /* Py_WIN_WIDE_FILENAMES */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002913}
2914
Guido van Rossum85e3b011991-06-03 12:42:10 +00002915
Guido van Rossum3b066191991-06-04 19:40:25 +00002916/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002917
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002918PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002919"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002920Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002921
Barry Warsaw53699e91996-12-10 23:23:01 +00002922static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002923posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002924{
2925 int sts;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002926 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
Guido van Rossum85e3b011991-06-03 12:42:10 +00002927 return NULL;
2928 _exit(sts);
Guido van Rossuma376cc51996-12-05 23:43:35 +00002929 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002930}
2931
Martin v. Löwis114619e2002-10-07 06:44:21 +00002932#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
2933static void
Martin v. Löwis725507b2006-03-07 12:08:51 +00002934free_string_array(char **array, Py_ssize_t count)
Martin v. Löwis114619e2002-10-07 06:44:21 +00002935{
Martin v. Löwis725507b2006-03-07 12:08:51 +00002936 Py_ssize_t i;
Martin v. Löwis114619e2002-10-07 06:44:21 +00002937 for (i = 0; i < count; i++)
2938 PyMem_Free(array[i]);
2939 PyMem_DEL(array);
2940}
2941#endif
2942
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002943
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002944#ifdef HAVE_EXECV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002945PyDoc_STRVAR(posix_execv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002946"execv(path, args)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002947Execute an executable path with arguments, replacing current process.\n\
2948\n\
2949 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002950 args: tuple or list of strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002951
Barry Warsaw53699e91996-12-10 23:23:01 +00002952static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002953posix_execv(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002954{
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002955 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00002956 PyObject *argv;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002957 char **argvlist;
Martin v. Löwis725507b2006-03-07 12:08:51 +00002958 Py_ssize_t i, argc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002959 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossum85e3b011991-06-03 12:42:10 +00002960
Guido van Rossum89b33251993-10-22 14:26:06 +00002961 /* execv has two arguments: (path, argv), where
Guido van Rossum85e3b011991-06-03 12:42:10 +00002962 argv is a list or tuple of strings. */
2963
Martin v. Löwis114619e2002-10-07 06:44:21 +00002964 if (!PyArg_ParseTuple(args, "etO:execv",
2965 Py_FileSystemDefaultEncoding,
2966 &path, &argv))
Guido van Rossum85e3b011991-06-03 12:42:10 +00002967 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002968 if (PyList_Check(argv)) {
2969 argc = PyList_Size(argv);
2970 getitem = PyList_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002971 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002972 else if (PyTuple_Check(argv)) {
2973 argc = PyTuple_Size(argv);
2974 getitem = PyTuple_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002975 }
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002976 else {
Fred Drake661ea262000-10-24 19:57:45 +00002977 PyErr_SetString(PyExc_TypeError, "execv() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002978 PyMem_Free(path);
Guido van Rossum50422b42000-04-26 20:34:28 +00002979 return NULL;
2980 }
Thomas Heller6790d602007-08-30 17:15:14 +00002981 if (argc < 1) {
2982 PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
2983 PyMem_Free(path);
2984 return NULL;
2985 }
Guido van Rossum50422b42000-04-26 20:34:28 +00002986
Barry Warsaw53699e91996-12-10 23:23:01 +00002987 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00002988 if (argvlist == NULL) {
2989 PyMem_Free(path);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00002990 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00002991 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00002992 for (i = 0; i < argc; i++) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00002993 if (!PyArg_Parse((*getitem)(argv, i), "et",
2994 Py_FileSystemDefaultEncoding,
2995 &argvlist[i])) {
2996 free_string_array(argvlist, i);
Tim Peters5aa91602002-01-30 05:46:57 +00002997 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00002998 "execv() arg 2 must contain only strings");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002999 PyMem_Free(path);
Guido van Rossum50422b42000-04-26 20:34:28 +00003000 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00003001
Guido van Rossum85e3b011991-06-03 12:42:10 +00003002 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00003003 }
3004 argvlist[argc] = NULL;
3005
Guido van Rossumef0a00e1992-01-27 16:51:30 +00003006 execv(path, argvlist);
Guido van Rossumb6775db1994-08-01 11:34:53 +00003007
Guido van Rossum85e3b011991-06-03 12:42:10 +00003008 /* If we get here it's definitely an error */
3009
Martin v. Löwis114619e2002-10-07 06:44:21 +00003010 free_string_array(argvlist, argc);
3011 PyMem_Free(path);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003012 return posix_error();
3013}
3014
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003015
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003016PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003017"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003018Execute a path with arguments and environment, replacing current process.\n\
3019\n\
3020 path: path of executable file\n\
3021 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003022 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003023
Barry Warsaw53699e91996-12-10 23:23:01 +00003024static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003025posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003026{
3027 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00003028 PyObject *argv, *env;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003029 char **argvlist;
3030 char **envlist;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003031 PyObject *key, *val, *keys=NULL, *vals=NULL;
Martin v. Löwis725507b2006-03-07 12:08:51 +00003032 Py_ssize_t i, pos, argc, envc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003033 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Thomas Wouters477c8d52006-05-27 19:21:47 +00003034 Py_ssize_t lastarg = 0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003035
3036 /* execve has three arguments: (path, argv, env), where
3037 argv is a list or tuple of strings and env is a dictionary
3038 like posix.environ. */
3039
Martin v. Löwis114619e2002-10-07 06:44:21 +00003040 if (!PyArg_ParseTuple(args, "etOO:execve",
3041 Py_FileSystemDefaultEncoding,
3042 &path, &argv, &env))
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003043 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00003044 if (PyList_Check(argv)) {
3045 argc = PyList_Size(argv);
3046 getitem = PyList_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003047 }
Barry Warsaw53699e91996-12-10 23:23:01 +00003048 else if (PyTuple_Check(argv)) {
3049 argc = PyTuple_Size(argv);
3050 getitem = PyTuple_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003051 }
3052 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003053 PyErr_SetString(PyExc_TypeError,
3054 "execve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003055 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003056 }
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003057 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003058 PyErr_SetString(PyExc_TypeError,
3059 "execve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003060 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003061 }
3062
Barry Warsaw53699e91996-12-10 23:23:01 +00003063 argvlist = PyMem_NEW(char *, argc+1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003064 if (argvlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003065 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003066 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003067 }
3068 for (i = 0; i < argc; i++) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003069 if (!PyArg_Parse((*getitem)(argv, i),
Martin v. Löwis114619e2002-10-07 06:44:21 +00003070 "et;execve() arg 2 must contain only strings",
Guido van Rossum1e700d22002-10-16 16:52:11 +00003071 Py_FileSystemDefaultEncoding,
Barry Warsaw43d68b81996-12-19 22:10:44 +00003072 &argvlist[i]))
3073 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003074 lastarg = i;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003075 goto fail_1;
3076 }
3077 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003078 lastarg = argc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003079 argvlist[argc] = NULL;
3080
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003081 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003082 if (i < 0)
3083 goto fail_1;
Barry Warsaw53699e91996-12-10 23:23:01 +00003084 envlist = PyMem_NEW(char *, i + 1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003085 if (envlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003086 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003087 goto fail_1;
3088 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003089 envc = 0;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003090 keys = PyMapping_Keys(env);
3091 vals = PyMapping_Values(env);
3092 if (!keys || !vals)
3093 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003094 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3095 PyErr_SetString(PyExc_TypeError,
3096 "execve(): env.keys() or env.values() is not a list");
3097 goto fail_2;
3098 }
Tim Peters5aa91602002-01-30 05:46:57 +00003099
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003100 for (pos = 0; pos < i; pos++) {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003101 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003102 size_t len;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003103
3104 key = PyList_GetItem(keys, pos);
3105 val = PyList_GetItem(vals, pos);
3106 if (!key || !val)
3107 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003108
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003109 if (!PyArg_Parse(
3110 key,
3111 "s;execve() arg 3 contains a non-string key",
3112 &k) ||
3113 !PyArg_Parse(
3114 val,
3115 "s;execve() arg 3 contains a non-string value",
3116 &v))
Barry Warsaw43d68b81996-12-19 22:10:44 +00003117 {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003118 goto fail_2;
3119 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00003120
3121#if defined(PYOS_OS2)
3122 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
3123 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
3124#endif
Christian Heimes830a4bc2007-11-22 07:43:40 +00003125 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Tim Petersc8996f52001-12-03 20:41:00 +00003126 p = PyMem_NEW(char, len);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003127 if (p == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003128 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003129 goto fail_2;
3130 }
Tim Petersc8996f52001-12-03 20:41:00 +00003131 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003132 envlist[envc++] = p;
Guido van Rossumd48f2521997-12-05 22:19:34 +00003133#if defined(PYOS_OS2)
3134 }
3135#endif
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003136 }
3137 envlist[envc] = 0;
3138
3139 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00003140
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003141 /* If we get here it's definitely an error */
3142
3143 (void) posix_error();
3144
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003145 fail_2:
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003146 while (--envc >= 0)
Barry Warsaw53699e91996-12-10 23:23:01 +00003147 PyMem_DEL(envlist[envc]);
3148 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003149 fail_1:
3150 free_string_array(argvlist, lastarg);
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003151 Py_XDECREF(vals);
3152 Py_XDECREF(keys);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003153 fail_0:
Martin v. Löwis114619e2002-10-07 06:44:21 +00003154 PyMem_Free(path);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003155 return NULL;
3156}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003157#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003158
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003159
Guido van Rossuma1065681999-01-25 23:20:23 +00003160#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003161PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003162"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003163Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003164\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003165 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003166 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003167 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003168
3169static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003170posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003171{
3172 char *path;
3173 PyObject *argv;
3174 char **argvlist;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003175 int mode, i;
3176 Py_ssize_t argc;
Tim Peters79248aa2001-08-29 21:37:10 +00003177 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003178 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossuma1065681999-01-25 23:20:23 +00003179
3180 /* spawnv has three arguments: (mode, path, argv), where
3181 argv is a list or tuple of strings. */
3182
Martin v. Löwis114619e2002-10-07 06:44:21 +00003183 if (!PyArg_ParseTuple(args, "ietO:spawnv", &mode,
3184 Py_FileSystemDefaultEncoding,
3185 &path, &argv))
Guido van Rossuma1065681999-01-25 23:20:23 +00003186 return NULL;
3187 if (PyList_Check(argv)) {
3188 argc = PyList_Size(argv);
3189 getitem = PyList_GetItem;
3190 }
3191 else if (PyTuple_Check(argv)) {
3192 argc = PyTuple_Size(argv);
3193 getitem = PyTuple_GetItem;
3194 }
3195 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003196 PyErr_SetString(PyExc_TypeError,
3197 "spawnv() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003198 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00003199 return NULL;
3200 }
3201
3202 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003203 if (argvlist == NULL) {
3204 PyMem_Free(path);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00003205 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003206 }
Guido van Rossuma1065681999-01-25 23:20:23 +00003207 for (i = 0; i < argc; i++) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003208 if (!PyArg_Parse((*getitem)(argv, i), "et",
3209 Py_FileSystemDefaultEncoding,
3210 &argvlist[i])) {
3211 free_string_array(argvlist, i);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003212 PyErr_SetString(
3213 PyExc_TypeError,
3214 "spawnv() arg 2 must contain only strings");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003215 PyMem_Free(path);
Fred Drake137507e2000-06-01 02:02:46 +00003216 return NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00003217 }
3218 }
3219 argvlist[argc] = NULL;
3220
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003221#if defined(PYOS_OS2) && defined(PYCC_GCC)
3222 Py_BEGIN_ALLOW_THREADS
3223 spawnval = spawnv(mode, path, argvlist);
3224 Py_END_ALLOW_THREADS
3225#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003226 if (mode == _OLD_P_OVERLAY)
3227 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00003228
Tim Peters25059d32001-12-07 20:35:43 +00003229 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003230 spawnval = _spawnv(mode, path, argvlist);
Tim Peters25059d32001-12-07 20:35:43 +00003231 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003232#endif
Tim Peters5aa91602002-01-30 05:46:57 +00003233
Martin v. Löwis114619e2002-10-07 06:44:21 +00003234 free_string_array(argvlist, argc);
3235 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00003236
Fred Drake699f3522000-06-29 21:12:41 +00003237 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003238 return posix_error();
3239 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003240#if SIZEOF_LONG == SIZEOF_VOID_P
3241 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003242#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003243 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003244#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003245}
3246
3247
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003248PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003249"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003250Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003251\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003252 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003253 path: path of executable file\n\
3254 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003255 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003256
3257static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003258posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003259{
3260 char *path;
3261 PyObject *argv, *env;
3262 char **argvlist;
3263 char **envlist;
3264 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003265 int mode, pos, envc;
3266 Py_ssize_t argc, i;
Tim Peters79248aa2001-08-29 21:37:10 +00003267 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003268 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Thomas Wouters477c8d52006-05-27 19:21:47 +00003269 Py_ssize_t lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003270
3271 /* spawnve has four arguments: (mode, path, argv, env), where
3272 argv is a list or tuple of strings and env is a dictionary
3273 like posix.environ. */
3274
Martin v. Löwis114619e2002-10-07 06:44:21 +00003275 if (!PyArg_ParseTuple(args, "ietOO:spawnve", &mode,
3276 Py_FileSystemDefaultEncoding,
3277 &path, &argv, &env))
Guido van Rossuma1065681999-01-25 23:20:23 +00003278 return NULL;
3279 if (PyList_Check(argv)) {
3280 argc = PyList_Size(argv);
3281 getitem = PyList_GetItem;
3282 }
3283 else if (PyTuple_Check(argv)) {
3284 argc = PyTuple_Size(argv);
3285 getitem = PyTuple_GetItem;
3286 }
3287 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003288 PyErr_SetString(PyExc_TypeError,
3289 "spawnve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003290 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003291 }
3292 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003293 PyErr_SetString(PyExc_TypeError,
3294 "spawnve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003295 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003296 }
3297
3298 argvlist = PyMem_NEW(char *, argc+1);
3299 if (argvlist == NULL) {
3300 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003301 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003302 }
3303 for (i = 0; i < argc; i++) {
3304 if (!PyArg_Parse((*getitem)(argv, i),
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003305 "et;spawnve() arg 2 must contain only strings",
Martin v. Löwis114619e2002-10-07 06:44:21 +00003306 Py_FileSystemDefaultEncoding,
Guido van Rossuma1065681999-01-25 23:20:23 +00003307 &argvlist[i]))
3308 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003309 lastarg = i;
Guido van Rossuma1065681999-01-25 23:20:23 +00003310 goto fail_1;
3311 }
3312 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003313 lastarg = argc;
Guido van Rossuma1065681999-01-25 23:20:23 +00003314 argvlist[argc] = NULL;
3315
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003316 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003317 if (i < 0)
3318 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00003319 envlist = PyMem_NEW(char *, i + 1);
3320 if (envlist == NULL) {
3321 PyErr_NoMemory();
3322 goto fail_1;
3323 }
3324 envc = 0;
3325 keys = PyMapping_Keys(env);
3326 vals = PyMapping_Values(env);
3327 if (!keys || !vals)
3328 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003329 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3330 PyErr_SetString(PyExc_TypeError,
3331 "spawnve(): env.keys() or env.values() is not a list");
3332 goto fail_2;
3333 }
Tim Peters5aa91602002-01-30 05:46:57 +00003334
Guido van Rossuma1065681999-01-25 23:20:23 +00003335 for (pos = 0; pos < i; pos++) {
3336 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003337 size_t len;
Guido van Rossuma1065681999-01-25 23:20:23 +00003338
3339 key = PyList_GetItem(keys, pos);
3340 val = PyList_GetItem(vals, pos);
3341 if (!key || !val)
3342 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003343
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003344 if (!PyArg_Parse(
3345 key,
3346 "s;spawnve() arg 3 contains a non-string key",
3347 &k) ||
3348 !PyArg_Parse(
3349 val,
3350 "s;spawnve() arg 3 contains a non-string value",
3351 &v))
Guido van Rossuma1065681999-01-25 23:20:23 +00003352 {
3353 goto fail_2;
3354 }
Christian Heimes830a4bc2007-11-22 07:43:40 +00003355 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Tim Petersc8996f52001-12-03 20:41:00 +00003356 p = PyMem_NEW(char, len);
Guido van Rossuma1065681999-01-25 23:20:23 +00003357 if (p == NULL) {
3358 PyErr_NoMemory();
3359 goto fail_2;
3360 }
Tim Petersc8996f52001-12-03 20:41:00 +00003361 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossuma1065681999-01-25 23:20:23 +00003362 envlist[envc++] = p;
3363 }
3364 envlist[envc] = 0;
3365
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003366#if defined(PYOS_OS2) && defined(PYCC_GCC)
3367 Py_BEGIN_ALLOW_THREADS
3368 spawnval = spawnve(mode, path, argvlist, envlist);
3369 Py_END_ALLOW_THREADS
3370#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003371 if (mode == _OLD_P_OVERLAY)
3372 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00003373
3374 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003375 spawnval = _spawnve(mode, path, argvlist, envlist);
Tim Peters25059d32001-12-07 20:35:43 +00003376 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003377#endif
Tim Peters25059d32001-12-07 20:35:43 +00003378
Fred Drake699f3522000-06-29 21:12:41 +00003379 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003380 (void) posix_error();
3381 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003382#if SIZEOF_LONG == SIZEOF_VOID_P
3383 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003384#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003385 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003386#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003387
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003388 fail_2:
Guido van Rossuma1065681999-01-25 23:20:23 +00003389 while (--envc >= 0)
3390 PyMem_DEL(envlist[envc]);
3391 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003392 fail_1:
Martin v. Löwis114619e2002-10-07 06:44:21 +00003393 free_string_array(argvlist, lastarg);
Guido van Rossuma1065681999-01-25 23:20:23 +00003394 Py_XDECREF(vals);
3395 Py_XDECREF(keys);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003396 fail_0:
3397 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00003398 return res;
3399}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003400
3401/* OS/2 supports spawnvp & spawnvpe natively */
3402#if defined(PYOS_OS2)
3403PyDoc_STRVAR(posix_spawnvp__doc__,
3404"spawnvp(mode, file, args)\n\n\
3405Execute the program 'file' in a new process, using the environment\n\
3406search path to find the file.\n\
3407\n\
3408 mode: mode of process creation\n\
3409 file: executable file name\n\
3410 args: tuple or list of strings");
3411
3412static PyObject *
3413posix_spawnvp(PyObject *self, PyObject *args)
3414{
3415 char *path;
3416 PyObject *argv;
3417 char **argvlist;
3418 int mode, i, argc;
3419 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003420 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003421
3422 /* spawnvp has three arguments: (mode, path, argv), where
3423 argv is a list or tuple of strings. */
3424
3425 if (!PyArg_ParseTuple(args, "ietO:spawnvp", &mode,
3426 Py_FileSystemDefaultEncoding,
3427 &path, &argv))
3428 return NULL;
3429 if (PyList_Check(argv)) {
3430 argc = PyList_Size(argv);
3431 getitem = PyList_GetItem;
3432 }
3433 else if (PyTuple_Check(argv)) {
3434 argc = PyTuple_Size(argv);
3435 getitem = PyTuple_GetItem;
3436 }
3437 else {
3438 PyErr_SetString(PyExc_TypeError,
3439 "spawnvp() arg 2 must be a tuple or list");
3440 PyMem_Free(path);
3441 return NULL;
3442 }
3443
3444 argvlist = PyMem_NEW(char *, argc+1);
3445 if (argvlist == NULL) {
3446 PyMem_Free(path);
3447 return PyErr_NoMemory();
3448 }
3449 for (i = 0; i < argc; i++) {
3450 if (!PyArg_Parse((*getitem)(argv, i), "et",
3451 Py_FileSystemDefaultEncoding,
3452 &argvlist[i])) {
3453 free_string_array(argvlist, i);
3454 PyErr_SetString(
3455 PyExc_TypeError,
3456 "spawnvp() arg 2 must contain only strings");
3457 PyMem_Free(path);
3458 return NULL;
3459 }
3460 }
3461 argvlist[argc] = NULL;
3462
3463 Py_BEGIN_ALLOW_THREADS
3464#if defined(PYCC_GCC)
3465 spawnval = spawnvp(mode, path, argvlist);
3466#else
3467 spawnval = _spawnvp(mode, path, argvlist);
3468#endif
3469 Py_END_ALLOW_THREADS
3470
3471 free_string_array(argvlist, argc);
3472 PyMem_Free(path);
3473
3474 if (spawnval == -1)
3475 return posix_error();
3476 else
3477 return Py_BuildValue("l", (long) spawnval);
3478}
3479
3480
3481PyDoc_STRVAR(posix_spawnvpe__doc__,
3482"spawnvpe(mode, file, args, env)\n\n\
3483Execute the program 'file' in a new process, using the environment\n\
3484search path to find the file.\n\
3485\n\
3486 mode: mode of process creation\n\
3487 file: executable file name\n\
3488 args: tuple or list of arguments\n\
3489 env: dictionary of strings mapping to strings");
3490
3491static PyObject *
3492posix_spawnvpe(PyObject *self, PyObject *args)
3493{
3494 char *path;
3495 PyObject *argv, *env;
3496 char **argvlist;
3497 char **envlist;
3498 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
3499 int mode, i, pos, argc, envc;
3500 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003501 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003502 int lastarg = 0;
3503
3504 /* spawnvpe has four arguments: (mode, path, argv, env), where
3505 argv is a list or tuple of strings and env is a dictionary
3506 like posix.environ. */
3507
3508 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
3509 Py_FileSystemDefaultEncoding,
3510 &path, &argv, &env))
3511 return NULL;
3512 if (PyList_Check(argv)) {
3513 argc = PyList_Size(argv);
3514 getitem = PyList_GetItem;
3515 }
3516 else if (PyTuple_Check(argv)) {
3517 argc = PyTuple_Size(argv);
3518 getitem = PyTuple_GetItem;
3519 }
3520 else {
3521 PyErr_SetString(PyExc_TypeError,
3522 "spawnvpe() arg 2 must be a tuple or list");
3523 goto fail_0;
3524 }
3525 if (!PyMapping_Check(env)) {
3526 PyErr_SetString(PyExc_TypeError,
3527 "spawnvpe() arg 3 must be a mapping object");
3528 goto fail_0;
3529 }
3530
3531 argvlist = PyMem_NEW(char *, argc+1);
3532 if (argvlist == NULL) {
3533 PyErr_NoMemory();
3534 goto fail_0;
3535 }
3536 for (i = 0; i < argc; i++) {
3537 if (!PyArg_Parse((*getitem)(argv, i),
3538 "et;spawnvpe() arg 2 must contain only strings",
3539 Py_FileSystemDefaultEncoding,
3540 &argvlist[i]))
3541 {
3542 lastarg = i;
3543 goto fail_1;
3544 }
3545 }
3546 lastarg = argc;
3547 argvlist[argc] = NULL;
3548
3549 i = PyMapping_Size(env);
3550 if (i < 0)
3551 goto fail_1;
3552 envlist = PyMem_NEW(char *, i + 1);
3553 if (envlist == NULL) {
3554 PyErr_NoMemory();
3555 goto fail_1;
3556 }
3557 envc = 0;
3558 keys = PyMapping_Keys(env);
3559 vals = PyMapping_Values(env);
3560 if (!keys || !vals)
3561 goto fail_2;
3562 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3563 PyErr_SetString(PyExc_TypeError,
3564 "spawnvpe(): env.keys() or env.values() is not a list");
3565 goto fail_2;
3566 }
3567
3568 for (pos = 0; pos < i; pos++) {
3569 char *p, *k, *v;
3570 size_t len;
3571
3572 key = PyList_GetItem(keys, pos);
3573 val = PyList_GetItem(vals, pos);
3574 if (!key || !val)
3575 goto fail_2;
3576
3577 if (!PyArg_Parse(
3578 key,
3579 "s;spawnvpe() arg 3 contains a non-string key",
3580 &k) ||
3581 !PyArg_Parse(
3582 val,
3583 "s;spawnvpe() arg 3 contains a non-string value",
3584 &v))
3585 {
3586 goto fail_2;
3587 }
Christian Heimes830a4bc2007-11-22 07:43:40 +00003588 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003589 p = PyMem_NEW(char, len);
3590 if (p == NULL) {
3591 PyErr_NoMemory();
3592 goto fail_2;
3593 }
3594 PyOS_snprintf(p, len, "%s=%s", k, v);
3595 envlist[envc++] = p;
3596 }
3597 envlist[envc] = 0;
3598
3599 Py_BEGIN_ALLOW_THREADS
3600#if defined(PYCC_GCC)
Christian Heimes292d3512008-02-03 16:51:08 +00003601 spawnval = spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003602#else
Christian Heimes292d3512008-02-03 16:51:08 +00003603 spawnval = _spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003604#endif
3605 Py_END_ALLOW_THREADS
3606
3607 if (spawnval == -1)
3608 (void) posix_error();
3609 else
3610 res = Py_BuildValue("l", (long) spawnval);
3611
3612 fail_2:
3613 while (--envc >= 0)
3614 PyMem_DEL(envlist[envc]);
3615 PyMem_DEL(envlist);
3616 fail_1:
3617 free_string_array(argvlist, lastarg);
3618 Py_XDECREF(vals);
3619 Py_XDECREF(keys);
3620 fail_0:
3621 PyMem_Free(path);
3622 return res;
3623}
3624#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00003625#endif /* HAVE_SPAWNV */
3626
3627
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003628#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003629PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003630"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003631Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
3632\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003633Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003634
3635static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003636posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003637{
Christian Heimes400adb02008-02-01 08:12:03 +00003638 pid_t pid = fork1();
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003639 if (pid == -1)
3640 return posix_error();
Georg Brandl2ee470f2008-07-16 12:55:28 +00003641 if (pid == 0)
3642 PyOS_AfterFork();
Christian Heimes400adb02008-02-01 08:12:03 +00003643 return PyLong_FromLong(pid);
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003644}
3645#endif
3646
3647
Guido van Rossumad0ee831995-03-01 10:34:45 +00003648#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003649PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003650"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003651Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003652Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003653
Barry Warsaw53699e91996-12-10 23:23:01 +00003654static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003655posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003656{
Christian Heimes400adb02008-02-01 08:12:03 +00003657 pid_t pid = fork();
Guido van Rossum85e3b011991-06-03 12:42:10 +00003658 if (pid == -1)
3659 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003660 if (pid == 0)
3661 PyOS_AfterFork();
Christian Heimes400adb02008-02-01 08:12:03 +00003662 return PyLong_FromLong(pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003663}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003664#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003665
Neal Norwitzb59798b2003-03-21 01:43:31 +00003666/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00003667/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
3668#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00003669#define DEV_PTY_FILE "/dev/ptc"
3670#define HAVE_DEV_PTMX
3671#else
3672#define DEV_PTY_FILE "/dev/ptmx"
3673#endif
3674
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003675#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003676#ifdef HAVE_PTY_H
3677#include <pty.h>
3678#else
3679#ifdef HAVE_LIBUTIL_H
3680#include <libutil.h>
Fred Drake8cef4cf2000-06-28 16:40:38 +00003681#endif /* HAVE_LIBUTIL_H */
3682#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00003683#ifdef HAVE_STROPTS_H
3684#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003685#endif
3686#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003687
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003688#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003689PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003690"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003691Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003692
3693static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003694posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003695{
3696 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003697#ifndef HAVE_OPENPTY
3698 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003699#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003700#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003701 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003702#ifdef sun
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003703 extern char *ptsname(int fildes);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003704#endif
3705#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00003706
Thomas Wouters70c21a12000-07-14 14:28:33 +00003707#ifdef HAVE_OPENPTY
Fred Drake8cef4cf2000-06-28 16:40:38 +00003708 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
3709 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003710#elif defined(HAVE__GETPTY)
Thomas Wouters70c21a12000-07-14 14:28:33 +00003711 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
3712 if (slave_name == NULL)
3713 return posix_error();
3714
3715 slave_fd = open(slave_name, O_RDWR);
3716 if (slave_fd < 0)
3717 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003718#else
Neal Norwitzb59798b2003-03-21 01:43:31 +00003719 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003720 if (master_fd < 0)
3721 return posix_error();
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003722 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003723 /* change permission of slave */
3724 if (grantpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003725 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003726 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003727 }
3728 /* unlock slave */
3729 if (unlockpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003730 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003731 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003732 }
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003733 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003734 slave_name = ptsname(master_fd); /* get name of slave */
3735 if (slave_name == NULL)
3736 return posix_error();
3737 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
3738 if (slave_fd < 0)
3739 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003740#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003741 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
3742 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00003743#ifndef __hpux
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003744 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00003745#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003746#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00003747#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00003748
Fred Drake8cef4cf2000-06-28 16:40:38 +00003749 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00003750
Fred Drake8cef4cf2000-06-28 16:40:38 +00003751}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003752#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003753
3754#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003755PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003756"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00003757Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
3758Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003759To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003760
3761static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003762posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003763{
Christian Heimes400adb02008-02-01 08:12:03 +00003764 int master_fd = -1;
3765 pid_t pid;
Tim Peters5aa91602002-01-30 05:46:57 +00003766
Fred Drake8cef4cf2000-06-28 16:40:38 +00003767 pid = forkpty(&master_fd, NULL, NULL, NULL);
3768 if (pid == -1)
3769 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003770 if (pid == 0)
3771 PyOS_AfterFork();
Christian Heimes400adb02008-02-01 08:12:03 +00003772 return Py_BuildValue("(li)", pid, master_fd);
Fred Drake8cef4cf2000-06-28 16:40:38 +00003773}
3774#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003775
Guido van Rossumad0ee831995-03-01 10:34:45 +00003776#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003777PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003778"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003779Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003780
Barry Warsaw53699e91996-12-10 23:23:01 +00003781static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003782posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003783{
Christian Heimes217cfd12007-12-02 14:31:20 +00003784 return PyLong_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003785}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003786#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003787
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003788
Guido van Rossumad0ee831995-03-01 10:34:45 +00003789#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003790PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003791"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003792Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003793
Barry Warsaw53699e91996-12-10 23:23:01 +00003794static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003795posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003796{
Christian Heimes217cfd12007-12-02 14:31:20 +00003797 return PyLong_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003798}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003799#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003800
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003801
Guido van Rossumad0ee831995-03-01 10:34:45 +00003802#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003803PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003804"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003805Return the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003806
Barry Warsaw53699e91996-12-10 23:23:01 +00003807static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003808posix_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003809{
Christian Heimes217cfd12007-12-02 14:31:20 +00003810 return PyLong_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003811}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003812#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003813
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003814
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003815PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003816"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003817Return the current process id");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003818
Barry Warsaw53699e91996-12-10 23:23:01 +00003819static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003820posix_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003821{
Christian Heimes217cfd12007-12-02 14:31:20 +00003822 return PyLong_FromLong((long)getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00003823}
3824
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003825
Fred Drakec9680921999-12-13 16:37:25 +00003826#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003827PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003828"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003829Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00003830
3831static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003832posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00003833{
3834 PyObject *result = NULL;
3835
Fred Drakec9680921999-12-13 16:37:25 +00003836#ifdef NGROUPS_MAX
3837#define MAX_GROUPS NGROUPS_MAX
3838#else
3839 /* defined to be 16 on Solaris7, so this should be a small number */
3840#define MAX_GROUPS 64
3841#endif
3842 gid_t grouplist[MAX_GROUPS];
3843 int n;
3844
3845 n = getgroups(MAX_GROUPS, grouplist);
3846 if (n < 0)
3847 posix_error();
3848 else {
3849 result = PyList_New(n);
3850 if (result != NULL) {
Fred Drakec9680921999-12-13 16:37:25 +00003851 int i;
3852 for (i = 0; i < n; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +00003853 PyObject *o = PyLong_FromLong((long)grouplist[i]);
Fred Drakec9680921999-12-13 16:37:25 +00003854 if (o == NULL) {
3855 Py_DECREF(result);
3856 result = NULL;
3857 break;
3858 }
3859 PyList_SET_ITEM(result, i, o);
3860 }
3861 }
3862 }
Neal Norwitze241ce82003-02-17 18:17:05 +00003863
Fred Drakec9680921999-12-13 16:37:25 +00003864 return result;
3865}
3866#endif
3867
Martin v. Löwis606edc12002-06-13 21:09:11 +00003868#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00003869PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003870"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00003871Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00003872
3873static PyObject *
3874posix_getpgid(PyObject *self, PyObject *args)
3875{
3876 int pid, pgid;
3877 if (!PyArg_ParseTuple(args, "i:getpgid", &pid))
3878 return NULL;
3879 pgid = getpgid(pid);
3880 if (pgid < 0)
3881 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00003882 return PyLong_FromLong((long)pgid);
Martin v. Löwis606edc12002-06-13 21:09:11 +00003883}
3884#endif /* HAVE_GETPGID */
3885
3886
Guido van Rossumb6775db1994-08-01 11:34:53 +00003887#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003888PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003889"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003890Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003891
Barry Warsaw53699e91996-12-10 23:23:01 +00003892static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003893posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00003894{
Guido van Rossumb6775db1994-08-01 11:34:53 +00003895#ifdef GETPGRP_HAVE_ARG
Christian Heimes217cfd12007-12-02 14:31:20 +00003896 return PyLong_FromLong((long)getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003897#else /* GETPGRP_HAVE_ARG */
Christian Heimes217cfd12007-12-02 14:31:20 +00003898 return PyLong_FromLong((long)getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003899#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00003900}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003901#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00003902
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003903
Guido van Rossumb6775db1994-08-01 11:34:53 +00003904#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003905PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003906"setpgrp()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003907Make this process a session leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003908
Barry Warsaw53699e91996-12-10 23:23:01 +00003909static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003910posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00003911{
Guido van Rossum64933891994-10-20 21:56:42 +00003912#ifdef SETPGRP_HAVE_ARG
Guido van Rossumc2670a01992-09-13 20:07:29 +00003913 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00003914#else /* SETPGRP_HAVE_ARG */
Guido van Rossumb6775db1994-08-01 11:34:53 +00003915 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00003916#endif /* SETPGRP_HAVE_ARG */
Guido van Rossum687dd131993-05-17 08:34:16 +00003917 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00003918 Py_INCREF(Py_None);
3919 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00003920}
3921
Guido van Rossumb6775db1994-08-01 11:34:53 +00003922#endif /* HAVE_SETPGRP */
3923
Guido van Rossumad0ee831995-03-01 10:34:45 +00003924#ifdef HAVE_GETPPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003925PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003926"getppid() -> ppid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003927Return the parent's process id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003928
Barry Warsaw53699e91996-12-10 23:23:01 +00003929static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003930posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003931{
Christian Heimes217cfd12007-12-02 14:31:20 +00003932 return PyLong_FromLong((long)getppid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00003933}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003934#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003935
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003936
Fred Drake12c6e2d1999-12-14 21:25:03 +00003937#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003938PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003939"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003940Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00003941
3942static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003943posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00003944{
Neal Norwitze241ce82003-02-17 18:17:05 +00003945 PyObject *result = NULL;
Fred Drakea30680b2000-12-06 21:24:28 +00003946 char *name;
3947 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00003948
Fred Drakea30680b2000-12-06 21:24:28 +00003949 errno = 0;
3950 name = getlogin();
3951 if (name == NULL) {
3952 if (errno)
3953 posix_error();
3954 else
3955 PyErr_SetString(PyExc_OSError,
Fred Drakee63544f2000-12-06 21:45:33 +00003956 "unable to determine login name");
Fred Drakea30680b2000-12-06 21:24:28 +00003957 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00003958 else
Neal Norwitz93c56822007-08-26 07:10:06 +00003959 result = PyUnicode_FromString(name);
Fred Drakea30680b2000-12-06 21:24:28 +00003960 errno = old_errno;
Neal Norwitze241ce82003-02-17 18:17:05 +00003961
Fred Drake12c6e2d1999-12-14 21:25:03 +00003962 return result;
3963}
3964#endif
3965
Guido van Rossumad0ee831995-03-01 10:34:45 +00003966#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003967PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003968"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003969Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003970
Barry Warsaw53699e91996-12-10 23:23:01 +00003971static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003972posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003973{
Christian Heimes217cfd12007-12-02 14:31:20 +00003974 return PyLong_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003975}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003976#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003977
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003978
Guido van Rossumad0ee831995-03-01 10:34:45 +00003979#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003980PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003981"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003982Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003983
Barry Warsaw53699e91996-12-10 23:23:01 +00003984static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003985posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003986{
Christian Heimes292d3512008-02-03 16:51:08 +00003987 pid_t pid;
3988 int sig;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00003989 if (!PyArg_ParseTuple(args, "ii:kill", &pid, &sig))
Guido van Rossum85e3b011991-06-03 12:42:10 +00003990 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003991#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003992 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
3993 APIRET rc;
3994 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00003995 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003996
3997 } else if (sig == XCPT_SIGNAL_KILLPROC) {
3998 APIRET rc;
3999 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004000 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004001
4002 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00004003 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004004#else
Guido van Rossum85e3b011991-06-03 12:42:10 +00004005 if (kill(pid, sig) == -1)
4006 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004007#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00004008 Py_INCREF(Py_None);
4009 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00004010}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004011#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004012
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004013#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004014PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004015"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004016Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004017
4018static PyObject *
4019posix_killpg(PyObject *self, PyObject *args)
4020{
4021 int pgid, sig;
4022 if (!PyArg_ParseTuple(args, "ii:killpg", &pgid, &sig))
4023 return NULL;
4024 if (killpg(pgid, sig) == -1)
4025 return posix_error();
4026 Py_INCREF(Py_None);
4027 return Py_None;
4028}
4029#endif
4030
Guido van Rossumc0125471996-06-28 18:55:32 +00004031#ifdef HAVE_PLOCK
4032
4033#ifdef HAVE_SYS_LOCK_H
4034#include <sys/lock.h>
4035#endif
4036
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004037PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004038"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004039Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004040
Barry Warsaw53699e91996-12-10 23:23:01 +00004041static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004042posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00004043{
4044 int op;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004045 if (!PyArg_ParseTuple(args, "i:plock", &op))
Guido van Rossumc0125471996-06-28 18:55:32 +00004046 return NULL;
4047 if (plock(op) == -1)
4048 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004049 Py_INCREF(Py_None);
4050 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00004051}
4052#endif
4053
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004054
Guido van Rossum3b066191991-06-04 19:40:25 +00004055
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004056
Guido van Rossumb6775db1994-08-01 11:34:53 +00004057#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004058PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004059"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004060Set the current process's user id.");
4061
Barry Warsaw53699e91996-12-10 23:23:01 +00004062static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004063posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004064{
4065 int uid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004066 if (!PyArg_ParseTuple(args, "i:setuid", &uid))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004067 return NULL;
4068 if (setuid(uid) < 0)
4069 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004070 Py_INCREF(Py_None);
4071 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004072}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004073#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004074
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004075
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004076#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004077PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004078"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004079Set the current process's effective user id.");
4080
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004081static PyObject *
4082posix_seteuid (PyObject *self, PyObject *args)
4083{
4084 int euid;
4085 if (!PyArg_ParseTuple(args, "i", &euid)) {
4086 return NULL;
4087 } else if (seteuid(euid) < 0) {
4088 return posix_error();
4089 } else {
4090 Py_INCREF(Py_None);
4091 return Py_None;
4092 }
4093}
4094#endif /* HAVE_SETEUID */
4095
4096#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004097PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004098"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004099Set the current process's effective group id.");
4100
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004101static PyObject *
4102posix_setegid (PyObject *self, PyObject *args)
4103{
4104 int egid;
4105 if (!PyArg_ParseTuple(args, "i", &egid)) {
4106 return NULL;
4107 } else if (setegid(egid) < 0) {
4108 return posix_error();
4109 } else {
4110 Py_INCREF(Py_None);
4111 return Py_None;
4112 }
4113}
4114#endif /* HAVE_SETEGID */
4115
4116#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004117PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004118"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004119Set the current process's real and effective user ids.");
4120
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004121static PyObject *
4122posix_setreuid (PyObject *self, PyObject *args)
4123{
4124 int ruid, euid;
4125 if (!PyArg_ParseTuple(args, "ii", &ruid, &euid)) {
4126 return NULL;
4127 } else if (setreuid(ruid, euid) < 0) {
4128 return posix_error();
4129 } else {
4130 Py_INCREF(Py_None);
4131 return Py_None;
4132 }
4133}
4134#endif /* HAVE_SETREUID */
4135
4136#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004137PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004138"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004139Set the current process's real and effective group ids.");
4140
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004141static PyObject *
4142posix_setregid (PyObject *self, PyObject *args)
4143{
4144 int rgid, egid;
4145 if (!PyArg_ParseTuple(args, "ii", &rgid, &egid)) {
4146 return NULL;
4147 } else if (setregid(rgid, egid) < 0) {
4148 return posix_error();
4149 } else {
4150 Py_INCREF(Py_None);
4151 return Py_None;
4152 }
4153}
4154#endif /* HAVE_SETREGID */
4155
Guido van Rossumb6775db1994-08-01 11:34:53 +00004156#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004157PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004158"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004159Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004160
Barry Warsaw53699e91996-12-10 23:23:01 +00004161static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004162posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004163{
4164 int gid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004165 if (!PyArg_ParseTuple(args, "i:setgid", &gid))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004166 return NULL;
4167 if (setgid(gid) < 0)
4168 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004169 Py_INCREF(Py_None);
4170 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004171}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004172#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004173
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004174#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004175PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004176"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004177Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004178
4179static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00004180posix_setgroups(PyObject *self, PyObject *groups)
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004181{
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004182 int i, len;
4183 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00004184
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004185 if (!PySequence_Check(groups)) {
4186 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
4187 return NULL;
4188 }
4189 len = PySequence_Size(groups);
4190 if (len > MAX_GROUPS) {
4191 PyErr_SetString(PyExc_ValueError, "too many groups");
4192 return NULL;
4193 }
4194 for(i = 0; i < len; i++) {
4195 PyObject *elem;
4196 elem = PySequence_GetItem(groups, i);
4197 if (!elem)
4198 return NULL;
Guido van Rossumddefaf32007-01-14 03:31:43 +00004199 if (!PyLong_Check(elem)) {
4200 PyErr_SetString(PyExc_TypeError,
4201 "groups must be integers");
4202 Py_DECREF(elem);
4203 return NULL;
4204 } else {
4205 unsigned long x = PyLong_AsUnsignedLong(elem);
4206 if (PyErr_Occurred()) {
4207 PyErr_SetString(PyExc_TypeError,
4208 "group id too big");
Georg Brandla13c2442005-11-22 19:30:31 +00004209 Py_DECREF(elem);
4210 return NULL;
Georg Brandla13c2442005-11-22 19:30:31 +00004211 }
Georg Brandla13c2442005-11-22 19:30:31 +00004212 grouplist[i] = x;
Guido van Rossumddefaf32007-01-14 03:31:43 +00004213 /* read back the value to see if it fitted in gid_t */
Georg Brandla13c2442005-11-22 19:30:31 +00004214 if (grouplist[i] != x) {
4215 PyErr_SetString(PyExc_TypeError,
4216 "group id too big");
4217 Py_DECREF(elem);
4218 return NULL;
4219 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004220 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004221 Py_DECREF(elem);
4222 }
4223
4224 if (setgroups(len, grouplist) < 0)
4225 return posix_error();
4226 Py_INCREF(Py_None);
4227 return Py_None;
4228}
4229#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004230
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004231#if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
4232static PyObject *
Christian Heimes292d3512008-02-03 16:51:08 +00004233wait_helper(pid_t pid, int status, struct rusage *ru)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004234{
4235 PyObject *result;
4236 static PyObject *struct_rusage;
4237
4238 if (pid == -1)
4239 return posix_error();
4240
4241 if (struct_rusage == NULL) {
Christian Heimes072c0f12008-01-03 23:01:04 +00004242 PyObject *m = PyImport_ImportModuleNoBlock("resource");
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004243 if (m == NULL)
4244 return NULL;
4245 struct_rusage = PyObject_GetAttrString(m, "struct_rusage");
4246 Py_DECREF(m);
4247 if (struct_rusage == NULL)
4248 return NULL;
4249 }
4250
4251 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
4252 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
4253 if (!result)
4254 return NULL;
4255
4256#ifndef doubletime
4257#define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
4258#endif
4259
4260 PyStructSequence_SET_ITEM(result, 0,
4261 PyFloat_FromDouble(doubletime(ru->ru_utime)));
4262 PyStructSequence_SET_ITEM(result, 1,
4263 PyFloat_FromDouble(doubletime(ru->ru_stime)));
4264#define SET_INT(result, index, value)\
Christian Heimes217cfd12007-12-02 14:31:20 +00004265 PyStructSequence_SET_ITEM(result, index, PyLong_FromLong(value))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004266 SET_INT(result, 2, ru->ru_maxrss);
4267 SET_INT(result, 3, ru->ru_ixrss);
4268 SET_INT(result, 4, ru->ru_idrss);
4269 SET_INT(result, 5, ru->ru_isrss);
4270 SET_INT(result, 6, ru->ru_minflt);
4271 SET_INT(result, 7, ru->ru_majflt);
4272 SET_INT(result, 8, ru->ru_nswap);
4273 SET_INT(result, 9, ru->ru_inblock);
4274 SET_INT(result, 10, ru->ru_oublock);
4275 SET_INT(result, 11, ru->ru_msgsnd);
4276 SET_INT(result, 12, ru->ru_msgrcv);
4277 SET_INT(result, 13, ru->ru_nsignals);
4278 SET_INT(result, 14, ru->ru_nvcsw);
4279 SET_INT(result, 15, ru->ru_nivcsw);
4280#undef SET_INT
4281
4282 if (PyErr_Occurred()) {
4283 Py_DECREF(result);
4284 return NULL;
4285 }
4286
4287 return Py_BuildValue("iiN", pid, status, result);
4288}
4289#endif /* HAVE_WAIT3 || HAVE_WAIT4 */
4290
4291#ifdef HAVE_WAIT3
4292PyDoc_STRVAR(posix_wait3__doc__,
4293"wait3(options) -> (pid, status, rusage)\n\n\
4294Wait for completion of a child process.");
4295
4296static PyObject *
4297posix_wait3(PyObject *self, PyObject *args)
4298{
Christian Heimes292d3512008-02-03 16:51:08 +00004299 pid_t pid;
4300 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004301 struct rusage ru;
4302 WAIT_TYPE status;
4303 WAIT_STATUS_INT(status) = 0;
4304
4305 if (!PyArg_ParseTuple(args, "i:wait3", &options))
4306 return NULL;
4307
4308 Py_BEGIN_ALLOW_THREADS
4309 pid = wait3(&status, options, &ru);
4310 Py_END_ALLOW_THREADS
4311
4312 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
4313}
4314#endif /* HAVE_WAIT3 */
4315
4316#ifdef HAVE_WAIT4
4317PyDoc_STRVAR(posix_wait4__doc__,
4318"wait4(pid, options) -> (pid, status, rusage)\n\n\
4319Wait for completion of a given child process.");
4320
4321static PyObject *
4322posix_wait4(PyObject *self, PyObject *args)
4323{
Christian Heimes292d3512008-02-03 16:51:08 +00004324 pid_t pid;
4325 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004326 struct rusage ru;
4327 WAIT_TYPE status;
4328 WAIT_STATUS_INT(status) = 0;
4329
4330 if (!PyArg_ParseTuple(args, "ii:wait4", &pid, &options))
4331 return NULL;
4332
4333 Py_BEGIN_ALLOW_THREADS
4334 pid = wait4(pid, &status, options, &ru);
4335 Py_END_ALLOW_THREADS
4336
4337 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
4338}
4339#endif /* HAVE_WAIT4 */
4340
Guido van Rossumb6775db1994-08-01 11:34:53 +00004341#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004342PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004343"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004344Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004345
Barry Warsaw53699e91996-12-10 23:23:01 +00004346static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004347posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004348{
Christian Heimes292d3512008-02-03 16:51:08 +00004349 pid_t pid;
4350 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004351 WAIT_TYPE status;
4352 WAIT_STATUS_INT(status) = 0;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004353
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004354 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
Guido van Rossum21803b81992-08-09 12:55:27 +00004355 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004356 Py_BEGIN_ALLOW_THREADS
Guido van Rossume6a3aa61999-02-01 16:15:30 +00004357 pid = waitpid(pid, &status, options);
Barry Warsaw53699e91996-12-10 23:23:01 +00004358 Py_END_ALLOW_THREADS
Guido van Rossum85e3b011991-06-03 12:42:10 +00004359 if (pid == -1)
4360 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004361
4362 return Py_BuildValue("ii", pid, WAIT_STATUS_INT(status));
Guido van Rossum21803b81992-08-09 12:55:27 +00004363}
4364
Tim Petersab034fa2002-02-01 11:27:43 +00004365#elif defined(HAVE_CWAIT)
4366
4367/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004368PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004369"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004370"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00004371
4372static PyObject *
4373posix_waitpid(PyObject *self, PyObject *args)
4374{
Thomas Wouters477c8d52006-05-27 19:21:47 +00004375 Py_intptr_t pid;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004376 int status, options;
Tim Petersab034fa2002-02-01 11:27:43 +00004377
4378 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
4379 return NULL;
4380 Py_BEGIN_ALLOW_THREADS
4381 pid = _cwait(&status, pid, options);
4382 Py_END_ALLOW_THREADS
4383 if (pid == -1)
4384 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004385
4386 /* shift the status left a byte so this is more like the POSIX waitpid */
4387 return Py_BuildValue("ii", pid, status << 8);
Tim Petersab034fa2002-02-01 11:27:43 +00004388}
4389#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004390
Guido van Rossumad0ee831995-03-01 10:34:45 +00004391#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004392PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004393"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004394Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004395
Barry Warsaw53699e91996-12-10 23:23:01 +00004396static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004397posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00004398{
Christian Heimes292d3512008-02-03 16:51:08 +00004399 pid_t pid;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004400 WAIT_TYPE status;
4401 WAIT_STATUS_INT(status) = 0;
Neal Norwitze241ce82003-02-17 18:17:05 +00004402
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004403 Py_BEGIN_ALLOW_THREADS
4404 pid = wait(&status);
Barry Warsaw53699e91996-12-10 23:23:01 +00004405 Py_END_ALLOW_THREADS
Guido van Rossum21803b81992-08-09 12:55:27 +00004406 if (pid == -1)
4407 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004408
4409 return Py_BuildValue("ii", pid, WAIT_STATUS_INT(status));
Guido van Rossum85e3b011991-06-03 12:42:10 +00004410}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004411#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004412
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004413
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004414PyDoc_STRVAR(posix_lstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004415"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004416Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004417
Barry Warsaw53699e91996-12-10 23:23:01 +00004418static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004419posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004420{
Guido van Rossumb6775db1994-08-01 11:34:53 +00004421#ifdef HAVE_LSTAT
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004422 return posix_do_stat(self, args, "et:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004423#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004424#ifdef MS_WINDOWS
Martin v. Löwis14694662006-02-03 12:54:16 +00004425 return posix_do_stat(self, args, "et:lstat", STAT, "U:lstat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004426#else
4427 return posix_do_stat(self, args, "et:lstat", STAT, NULL, NULL);
4428#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00004429#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004430}
4431
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004432
Guido van Rossumb6775db1994-08-01 11:34:53 +00004433#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004434PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004435"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004436Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004437
Barry Warsaw53699e91996-12-10 23:23:01 +00004438static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004439posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004440{
Thomas Wouters89f507f2006-12-13 04:49:30 +00004441 PyObject* v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00004442 char buf[MAXPATHLEN];
Guido van Rossumef0a00e1992-01-27 16:51:30 +00004443 char *path;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004444 int n;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004445 int arg_is_unicode = 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004446
4447 if (!PyArg_ParseTuple(args, "et:readlink",
4448 Py_FileSystemDefaultEncoding, &path))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004449 return NULL;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004450 v = PySequence_GetItem(args, 0);
Neal Norwitzcda5c062007-08-12 17:09:36 +00004451 if (v == NULL) {
4452 PyMem_Free(path);
4453 return NULL;
4454 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004455
4456 if (PyUnicode_Check(v)) {
4457 arg_is_unicode = 1;
4458 }
4459 Py_DECREF(v);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004460
Barry Warsaw53699e91996-12-10 23:23:01 +00004461 Py_BEGIN_ALLOW_THREADS
Guido van Rossum50e61dc1992-03-27 17:22:31 +00004462 n = readlink(path, buf, (int) sizeof buf);
Barry Warsaw53699e91996-12-10 23:23:01 +00004463 Py_END_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004464 if (n < 0)
Neal Norwitzfca70052007-08-12 16:56:02 +00004465 return posix_error_with_allocated_filename(path);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004466
Neal Norwitzfca70052007-08-12 16:56:02 +00004467 PyMem_Free(path);
Christian Heimes72b710a2008-05-26 13:28:38 +00004468 v = PyBytes_FromStringAndSize(buf, n);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004469 if (arg_is_unicode) {
4470 PyObject *w;
4471
4472 w = PyUnicode_FromEncodedObject(v,
4473 Py_FileSystemDefaultEncoding,
4474 "strict");
4475 if (w != NULL) {
4476 Py_DECREF(v);
4477 v = w;
4478 }
4479 else {
4480 /* fall back to the original byte string, as
4481 discussed in patch #683592 */
4482 PyErr_Clear();
4483 }
4484 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004485 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004486}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004487#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004488
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004489
Guido van Rossumb6775db1994-08-01 11:34:53 +00004490#ifdef HAVE_SYMLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004491PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004492"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00004493Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004494
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004495static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004496posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004497{
Thomas Wouters477c8d52006-05-27 19:21:47 +00004498 return posix_2str(args, "etet:symlink", symlink);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004499}
4500#endif /* HAVE_SYMLINK */
4501
4502
4503#ifdef HAVE_TIMES
4504#ifndef HZ
4505#define HZ 60 /* Universal constant :-) */
4506#endif /* HZ */
Tim Peters5aa91602002-01-30 05:46:57 +00004507
Guido van Rossumd48f2521997-12-05 22:19:34 +00004508#if defined(PYCC_VACPP) && defined(PYOS_OS2)
4509static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00004510system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004511{
4512 ULONG value = 0;
4513
4514 Py_BEGIN_ALLOW_THREADS
4515 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
4516 Py_END_ALLOW_THREADS
4517
4518 return value;
4519}
4520
4521static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004522posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004523{
Guido van Rossumd48f2521997-12-05 22:19:34 +00004524 /* Currently Only Uptime is Provided -- Others Later */
4525 return Py_BuildValue("ddddd",
4526 (double)0 /* t.tms_utime / HZ */,
4527 (double)0 /* t.tms_stime / HZ */,
4528 (double)0 /* t.tms_cutime / HZ */,
4529 (double)0 /* t.tms_cstime / HZ */,
4530 (double)system_uptime() / 1000);
4531}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004532#else /* not OS2 */
Barry Warsaw53699e91996-12-10 23:23:01 +00004533static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004534posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00004535{
4536 struct tms t;
4537 clock_t c;
Guido van Rossum22db57e1992-04-05 14:25:30 +00004538 errno = 0;
4539 c = times(&t);
Guido van Rossum687dd131993-05-17 08:34:16 +00004540 if (c == (clock_t) -1)
4541 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004542 return Py_BuildValue("ddddd",
Barry Warsaw43d68b81996-12-19 22:10:44 +00004543 (double)t.tms_utime / HZ,
4544 (double)t.tms_stime / HZ,
4545 (double)t.tms_cutime / HZ,
4546 (double)t.tms_cstime / HZ,
4547 (double)c / HZ);
Guido van Rossum22db57e1992-04-05 14:25:30 +00004548}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004549#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00004550#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004551
4552
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004553#ifdef MS_WINDOWS
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004554#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00004555static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004556posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004557{
4558 FILETIME create, exit, kernel, user;
4559 HANDLE hProc;
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004560 hProc = GetCurrentProcess();
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004561 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
4562 /* The fields of a FILETIME structure are the hi and lo part
4563 of a 64-bit value expressed in 100 nanosecond units.
4564 1e7 is one second in such units; 1e-7 the inverse.
4565 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
4566 */
Barry Warsaw53699e91996-12-10 23:23:01 +00004567 return Py_BuildValue(
4568 "ddddd",
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004569 (double)(user.dwHighDateTime*429.4967296 +
4570 user.dwLowDateTime*1e-7),
Christian Heimes68f5fbe2008-02-14 08:27:37 +00004571 (double)(kernel.dwHighDateTime*429.4967296 +
4572 kernel.dwLowDateTime*1e-7),
Barry Warsaw53699e91996-12-10 23:23:01 +00004573 (double)0,
4574 (double)0,
4575 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004576}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004577#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004578
4579#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004580PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004581"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004582Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004583#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00004584
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004585
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004586#ifdef HAVE_GETSID
4587PyDoc_STRVAR(posix_getsid__doc__,
4588"getsid(pid) -> sid\n\n\
4589Call the system call getsid().");
4590
4591static PyObject *
4592posix_getsid(PyObject *self, PyObject *args)
4593{
Christian Heimes292d3512008-02-03 16:51:08 +00004594 pid_t pid;
4595 int sid;
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004596 if (!PyArg_ParseTuple(args, "i:getsid", &pid))
4597 return NULL;
4598 sid = getsid(pid);
4599 if (sid < 0)
4600 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004601 return PyLong_FromLong((long)sid);
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004602}
4603#endif /* HAVE_GETSID */
4604
4605
Guido van Rossumb6775db1994-08-01 11:34:53 +00004606#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004607PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004608"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004609Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004610
Barry Warsaw53699e91996-12-10 23:23:01 +00004611static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004612posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004613{
Guido van Rossum687dd131993-05-17 08:34:16 +00004614 if (setsid() < 0)
4615 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004616 Py_INCREF(Py_None);
4617 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004618}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004619#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00004620
Guido van Rossumb6775db1994-08-01 11:34:53 +00004621#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004622PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004623"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004624Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004625
Barry Warsaw53699e91996-12-10 23:23:01 +00004626static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004627posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004628{
Christian Heimes292d3512008-02-03 16:51:08 +00004629 pid_t pid;
4630 int pgrp;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004631 if (!PyArg_ParseTuple(args, "ii:setpgid", &pid, &pgrp))
Guido van Rossumc2670a01992-09-13 20:07:29 +00004632 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004633 if (setpgid(pid, pgrp) < 0)
4634 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004635 Py_INCREF(Py_None);
4636 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004637}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004638#endif /* HAVE_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00004639
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004640
Guido van Rossumb6775db1994-08-01 11:34:53 +00004641#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004642PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004643"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004644Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004645
Barry Warsaw53699e91996-12-10 23:23:01 +00004646static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004647posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00004648{
Christian Heimes15ebc882008-02-04 18:48:49 +00004649 int fd;
4650 pid_t pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004651 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
Guido van Rossum7066dd71992-09-17 17:54:56 +00004652 return NULL;
4653 pgid = tcgetpgrp(fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004654 if (pgid < 0)
4655 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004656 return PyLong_FromLong((long)pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00004657}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004658#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00004659
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004660
Guido van Rossumb6775db1994-08-01 11:34:53 +00004661#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004662PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004663"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004664Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004665
Barry Warsaw53699e91996-12-10 23:23:01 +00004666static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004667posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00004668{
4669 int fd, pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004670 if (!PyArg_ParseTuple(args, "ii:tcsetpgrp", &fd, &pgid))
Guido van Rossum7066dd71992-09-17 17:54:56 +00004671 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004672 if (tcsetpgrp(fd, pgid) < 0)
4673 return posix_error();
Barry Warsaw43d68b81996-12-19 22:10:44 +00004674 Py_INCREF(Py_None);
Barry Warsaw53699e91996-12-10 23:23:01 +00004675 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00004676}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004677#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00004678
Guido van Rossum687dd131993-05-17 08:34:16 +00004679/* Functions acting on file descriptors */
4680
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004681PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004682"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004683Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004684
Barry Warsaw53699e91996-12-10 23:23:01 +00004685static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004686posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004687{
Mark Hammondef8b6542001-05-13 08:04:26 +00004688 char *file = NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004689 int flag;
4690 int mode = 0777;
4691 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004692
4693#ifdef MS_WINDOWS
4694 if (unicode_file_names()) {
4695 PyUnicodeObject *po;
4696 if (PyArg_ParseTuple(args, "Ui|i:mkdir", &po, &flag, &mode)) {
4697 Py_BEGIN_ALLOW_THREADS
4698 /* PyUnicode_AS_UNICODE OK without thread
4699 lock as it is a simple dereference. */
4700 fd = _wopen(PyUnicode_AS_UNICODE(po), flag, mode);
4701 Py_END_ALLOW_THREADS
4702 if (fd < 0)
4703 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004704 return PyLong_FromLong((long)fd);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004705 }
4706 /* Drop the argument parsing error as narrow strings
4707 are also valid. */
4708 PyErr_Clear();
4709 }
4710#endif
4711
Tim Peters5aa91602002-01-30 05:46:57 +00004712 if (!PyArg_ParseTuple(args, "eti|i",
Mark Hammondef8b6542001-05-13 08:04:26 +00004713 Py_FileSystemDefaultEncoding, &file,
4714 &flag, &mode))
Barry Warsaw43d68b81996-12-19 22:10:44 +00004715 return NULL;
4716
Barry Warsaw53699e91996-12-10 23:23:01 +00004717 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004718 fd = open(file, flag, mode);
Barry Warsaw53699e91996-12-10 23:23:01 +00004719 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004720 if (fd < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00004721 return posix_error_with_allocated_filename(file);
4722 PyMem_Free(file);
Christian Heimes217cfd12007-12-02 14:31:20 +00004723 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004724}
4725
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004726
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004727PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004728"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004729Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004730
Barry Warsaw53699e91996-12-10 23:23:01 +00004731static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004732posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004733{
4734 int fd, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004735 if (!PyArg_ParseTuple(args, "i:close", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00004736 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004737 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004738 res = close(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00004739 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004740 if (res < 0)
4741 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004742 Py_INCREF(Py_None);
4743 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00004744}
4745
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004746
Christian Heimesfdab48e2008-01-20 09:06:41 +00004747PyDoc_STRVAR(posix_closerange__doc__,
4748"closerange(fd_low, fd_high)\n\n\
4749Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
4750
4751static PyObject *
4752posix_closerange(PyObject *self, PyObject *args)
4753{
4754 int fd_from, fd_to, i;
4755 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
4756 return NULL;
4757 Py_BEGIN_ALLOW_THREADS
4758 for (i = fd_from; i < fd_to; i++)
4759 close(i);
4760 Py_END_ALLOW_THREADS
4761 Py_RETURN_NONE;
4762}
4763
4764
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004765PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004766"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004767Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004768
Barry Warsaw53699e91996-12-10 23:23:01 +00004769static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004770posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004771{
4772 int fd;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004773 if (!PyArg_ParseTuple(args, "i:dup", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00004774 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004775 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004776 fd = dup(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00004777 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004778 if (fd < 0)
4779 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004780 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004781}
4782
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004783
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004784PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00004785"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004786Duplicate file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004787
Barry Warsaw53699e91996-12-10 23:23:01 +00004788static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004789posix_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004790{
4791 int fd, fd2, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004792 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
Guido van Rossum687dd131993-05-17 08:34:16 +00004793 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004794 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004795 res = dup2(fd, fd2);
Barry Warsaw53699e91996-12-10 23:23:01 +00004796 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004797 if (res < 0)
4798 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004799 Py_INCREF(Py_None);
4800 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00004801}
4802
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004803
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004804PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004805"lseek(fd, pos, how) -> newpos\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004806Set the current position of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004807
Barry Warsaw53699e91996-12-10 23:23:01 +00004808static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004809posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004810{
4811 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004812#if defined(MS_WIN64) || defined(MS_WINDOWS)
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00004813 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00004814#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00004815 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00004816#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00004817 PyObject *posobj;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004818 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Guido van Rossum687dd131993-05-17 08:34:16 +00004819 return NULL;
4820#ifdef SEEK_SET
4821 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
4822 switch (how) {
4823 case 0: how = SEEK_SET; break;
4824 case 1: how = SEEK_CUR; break;
4825 case 2: how = SEEK_END; break;
4826 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004827#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00004828
4829#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00004830 pos = PyLong_AsLong(posobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00004831#else
4832 pos = PyLong_Check(posobj) ?
Christian Heimes217cfd12007-12-02 14:31:20 +00004833 PyLong_AsLongLong(posobj) : PyLong_AsLong(posobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00004834#endif
4835 if (PyErr_Occurred())
4836 return NULL;
4837
Barry Warsaw53699e91996-12-10 23:23:01 +00004838 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004839#if defined(MS_WIN64) || defined(MS_WINDOWS)
Fred Drake699f3522000-06-29 21:12:41 +00004840 res = _lseeki64(fd, pos, how);
4841#else
Guido van Rossum687dd131993-05-17 08:34:16 +00004842 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00004843#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00004844 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004845 if (res < 0)
4846 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00004847
4848#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00004849 return PyLong_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00004850#else
4851 return PyLong_FromLongLong(res);
4852#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00004853}
4854
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004855
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004856PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004857"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004858Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004859
Barry Warsaw53699e91996-12-10 23:23:01 +00004860static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004861posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004862{
Guido van Rossum572dbf82007-04-27 23:53:51 +00004863 int fd, size;
4864 Py_ssize_t n;
Barry Warsaw53699e91996-12-10 23:23:01 +00004865 PyObject *buffer;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004866 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
Guido van Rossum687dd131993-05-17 08:34:16 +00004867 return NULL;
Michael W. Hudson9867ced2005-01-31 17:01:59 +00004868 if (size < 0) {
4869 errno = EINVAL;
4870 return posix_error();
4871 }
Christian Heimes72b710a2008-05-26 13:28:38 +00004872 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
Guido van Rossum687dd131993-05-17 08:34:16 +00004873 if (buffer == NULL)
4874 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004875 Py_BEGIN_ALLOW_THREADS
Christian Heimes72b710a2008-05-26 13:28:38 +00004876 n = read(fd, PyBytes_AS_STRING(buffer), size);
Barry Warsaw53699e91996-12-10 23:23:01 +00004877 Py_END_ALLOW_THREADS
Guido van Rossum8bac5461996-06-11 18:38:48 +00004878 if (n < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00004879 Py_DECREF(buffer);
Guido van Rossum687dd131993-05-17 08:34:16 +00004880 return posix_error();
4881 }
Guido van Rossum8bac5461996-06-11 18:38:48 +00004882 if (n != size)
Christian Heimes72b710a2008-05-26 13:28:38 +00004883 _PyBytes_Resize(&buffer, n);
Guido van Rossum687dd131993-05-17 08:34:16 +00004884 return buffer;
4885}
4886
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004887
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004888PyDoc_STRVAR(posix_write__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004889"write(fd, string) -> byteswritten\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004890Write a string to a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004891
Barry Warsaw53699e91996-12-10 23:23:01 +00004892static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004893posix_write(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004894{
Thomas Wouters68bc4f92006-03-01 01:05:10 +00004895 int fd;
4896 Py_ssize_t size;
Guido van Rossum687dd131993-05-17 08:34:16 +00004897 char *buffer;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00004898
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004899 if (!PyArg_ParseTuple(args, "is#:write", &fd, &buffer, &size))
Guido van Rossum687dd131993-05-17 08:34:16 +00004900 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004901 Py_BEGIN_ALLOW_THREADS
Thomas Wouters68bc4f92006-03-01 01:05:10 +00004902 size = write(fd, buffer, (size_t)size);
Barry Warsaw53699e91996-12-10 23:23:01 +00004903 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004904 if (size < 0)
4905 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004906 return PyLong_FromSsize_t(size);
Guido van Rossum687dd131993-05-17 08:34:16 +00004907}
4908
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004909
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004910PyDoc_STRVAR(posix_fstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004911"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004912Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004913
Barry Warsaw53699e91996-12-10 23:23:01 +00004914static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004915posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004916{
4917 int fd;
Fred Drake699f3522000-06-29 21:12:41 +00004918 STRUCT_STAT st;
Guido van Rossum687dd131993-05-17 08:34:16 +00004919 int res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004920 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00004921 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00004922#ifdef __VMS
4923 /* on OpenVMS we must ensure that all bytes are written to the file */
4924 fsync(fd);
4925#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00004926 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00004927 res = FSTAT(fd, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00004928 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00004929 if (res != 0) {
4930#ifdef MS_WINDOWS
4931 return win32_error("fstat", NULL);
4932#else
Guido van Rossum687dd131993-05-17 08:34:16 +00004933 return posix_error();
Martin v. Löwis14694662006-02-03 12:54:16 +00004934#endif
4935 }
Tim Peters5aa91602002-01-30 05:46:57 +00004936
Martin v. Löwis14694662006-02-03 12:54:16 +00004937 return _pystat_fromstructstat(&st);
Guido van Rossum687dd131993-05-17 08:34:16 +00004938}
4939
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004940PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004941"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00004942Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004943connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00004944
4945static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00004946posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00004947{
4948 int fd;
4949 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
4950 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00004951 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00004952}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004953
Guido van Rossuma4916fa1996-05-23 22:58:55 +00004954#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004955PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004956"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004957Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004958
Barry Warsaw53699e91996-12-10 23:23:01 +00004959static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004960posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00004961{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004962#if defined(PYOS_OS2)
4963 HFILE read, write;
4964 APIRET rc;
4965
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004966 Py_BEGIN_ALLOW_THREADS
4967 rc = DosCreatePipe( &read, &write, 4096);
4968 Py_END_ALLOW_THREADS
4969 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004970 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004971
4972 return Py_BuildValue("(ii)", read, write);
4973#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004974#if !defined(MS_WINDOWS)
Guido van Rossum687dd131993-05-17 08:34:16 +00004975 int fds[2];
4976 int res;
Barry Warsaw53699e91996-12-10 23:23:01 +00004977 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004978 res = pipe(fds);
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 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004983#else /* MS_WINDOWS */
Guido van Rossum794d8131994-08-23 13:48:48 +00004984 HANDLE read, write;
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00004985 int read_fd, write_fd;
Guido van Rossum794d8131994-08-23 13:48:48 +00004986 BOOL ok;
Barry Warsaw53699e91996-12-10 23:23:01 +00004987 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00004988 ok = CreatePipe(&read, &write, NULL, 0);
Barry Warsaw53699e91996-12-10 23:23:01 +00004989 Py_END_ALLOW_THREADS
Guido van Rossum794d8131994-08-23 13:48:48 +00004990 if (!ok)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004991 return win32_error("CreatePipe", NULL);
Tim Peters79248aa2001-08-29 21:37:10 +00004992 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
4993 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00004994 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004995#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004996#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00004997}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00004998#endif /* HAVE_PIPE */
4999
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005000
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005001#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005002PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005003"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005004Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005005
Barry Warsaw53699e91996-12-10 23:23:01 +00005006static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005007posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005008{
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005009 char *filename;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005010 int mode = 0666;
5011 int res;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005012 if (!PyArg_ParseTuple(args, "s|i:mkfifo", &filename, &mode))
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005013 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005014 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005015 res = mkfifo(filename, mode);
5016 Py_END_ALLOW_THREADS
5017 if (res < 0)
5018 return posix_error();
5019 Py_INCREF(Py_None);
5020 return Py_None;
5021}
5022#endif
5023
5024
Neal Norwitz11690112002-07-30 01:08:28 +00005025#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005026PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005027"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005028Create a filesystem node (file, device special file or named pipe)\n\
5029named filename. mode specifies both the permissions to use and the\n\
5030type of node to be created, being combined (bitwise OR) with one of\n\
5031S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005032device defines the newly created device special file (probably using\n\
5033os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005034
5035
5036static PyObject *
5037posix_mknod(PyObject *self, PyObject *args)
5038{
5039 char *filename;
5040 int mode = 0600;
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005041 int device = 0;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005042 int res;
Martin v. Löwisd631ebe2002-11-02 17:42:33 +00005043 if (!PyArg_ParseTuple(args, "s|ii:mknod", &filename, &mode, &device))
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005044 return NULL;
5045 Py_BEGIN_ALLOW_THREADS
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005046 res = mknod(filename, mode, device);
Barry Warsaw53699e91996-12-10 23:23:01 +00005047 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005048 if (res < 0)
5049 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005050 Py_INCREF(Py_None);
5051 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005052}
5053#endif
5054
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005055#ifdef HAVE_DEVICE_MACROS
5056PyDoc_STRVAR(posix_major__doc__,
5057"major(device) -> major number\n\
5058Extracts a device major number from a raw device number.");
5059
5060static PyObject *
5061posix_major(PyObject *self, PyObject *args)
5062{
5063 int device;
5064 if (!PyArg_ParseTuple(args, "i:major", &device))
5065 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005066 return PyLong_FromLong((long)major(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005067}
5068
5069PyDoc_STRVAR(posix_minor__doc__,
5070"minor(device) -> minor number\n\
5071Extracts a device minor number from a raw device number.");
5072
5073static PyObject *
5074posix_minor(PyObject *self, PyObject *args)
5075{
5076 int device;
5077 if (!PyArg_ParseTuple(args, "i:minor", &device))
5078 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005079 return PyLong_FromLong((long)minor(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005080}
5081
5082PyDoc_STRVAR(posix_makedev__doc__,
5083"makedev(major, minor) -> device number\n\
5084Composes a raw device number from the major and minor device numbers.");
5085
5086static PyObject *
5087posix_makedev(PyObject *self, PyObject *args)
5088{
5089 int major, minor;
5090 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
5091 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005092 return PyLong_FromLong((long)makedev(major, minor));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005093}
5094#endif /* device macros */
5095
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005096
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005097#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005098PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005099"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005100Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005101
Barry Warsaw53699e91996-12-10 23:23:01 +00005102static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005103posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005104{
5105 int fd;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005106 off_t length;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005107 int res;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005108 PyObject *lenobj;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005109
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005110 if (!PyArg_ParseTuple(args, "iO:ftruncate", &fd, &lenobj))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005111 return NULL;
5112
5113#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005114 length = PyLong_AsLong(lenobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005115#else
5116 length = PyLong_Check(lenobj) ?
Christian Heimes217cfd12007-12-02 14:31:20 +00005117 PyLong_AsLongLong(lenobj) : PyLong_AsLong(lenobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005118#endif
5119 if (PyErr_Occurred())
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005120 return NULL;
5121
Barry Warsaw53699e91996-12-10 23:23:01 +00005122 Py_BEGIN_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005123 res = ftruncate(fd, length);
Barry Warsaw53699e91996-12-10 23:23:01 +00005124 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005125 if (res < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00005126 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005127 return NULL;
5128 }
Barry Warsaw53699e91996-12-10 23:23:01 +00005129 Py_INCREF(Py_None);
5130 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005131}
5132#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00005133
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005134#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005135PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005136"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005137Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005138
Fred Drake762e2061999-08-26 17:23:54 +00005139/* Save putenv() parameters as values here, so we can collect them when they
5140 * get re-set with another call for the same key. */
5141static PyObject *posix_putenv_garbage;
5142
Tim Peters5aa91602002-01-30 05:46:57 +00005143static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005144posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005145{
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005146#ifdef MS_WINDOWS
5147 wchar_t *s1, *s2;
5148 wchar_t *newenv;
5149#else
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005150 char *s1, *s2;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005151 char *newenv;
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005152#endif
Fred Drake762e2061999-08-26 17:23:54 +00005153 PyObject *newstr;
Tim Petersc8996f52001-12-03 20:41:00 +00005154 size_t len;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005155
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005156 if (!PyArg_ParseTuple(args,
5157#ifdef MS_WINDOWS
5158 "uu:putenv",
5159#else
5160 "ss:putenv",
5161#endif
5162 &s1, &s2))
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005163 return NULL;
Guido van Rossumd48f2521997-12-05 22:19:34 +00005164
5165#if defined(PYOS_OS2)
5166 if (stricmp(s1, "BEGINLIBPATH") == 0) {
5167 APIRET rc;
5168
Guido van Rossumd48f2521997-12-05 22:19:34 +00005169 rc = DosSetExtLIBPATH(s2, BEGIN_LIBPATH);
5170 if (rc != NO_ERROR)
5171 return os2_error(rc);
5172
5173 } else if (stricmp(s1, "ENDLIBPATH") == 0) {
5174 APIRET rc;
5175
Guido van Rossumd48f2521997-12-05 22:19:34 +00005176 rc = DosSetExtLIBPATH(s2, END_LIBPATH);
5177 if (rc != NO_ERROR)
5178 return os2_error(rc);
5179 } else {
5180#endif
Fred Drake762e2061999-08-26 17:23:54 +00005181 /* XXX This can leak memory -- not easy to fix :-( */
Tim Petersc8996f52001-12-03 20:41:00 +00005182 /* len includes space for a trailing \0; the size arg to
Christian Heimes72b710a2008-05-26 13:28:38 +00005183 PyBytes_FromStringAndSize does not count that */
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005184#ifdef MS_WINDOWS
5185 len = wcslen(s1) + wcslen(s2) + 2;
5186 newstr = PyUnicode_FromUnicode(NULL, (int)len - 1);
5187#else
5188 len = strlen(s1) + strlen(s2) + 2;
Christian Heimes72b710a2008-05-26 13:28:38 +00005189 newstr = PyBytes_FromStringAndSize(NULL, (int)len - 1);
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005190#endif
Fred Drake762e2061999-08-26 17:23:54 +00005191 if (newstr == NULL)
5192 return PyErr_NoMemory();
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005193#ifdef MS_WINDOWS
5194 newenv = PyUnicode_AsUnicode(newstr);
5195 _snwprintf(newenv, len, L"%s=%s", s1, s2);
5196 if (_wputenv(newenv)) {
5197 Py_DECREF(newstr);
5198 posix_error();
5199 return NULL;
5200 }
5201#else
Christian Heimes72b710a2008-05-26 13:28:38 +00005202 newenv = PyBytes_AS_STRING(newstr);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005203 PyOS_snprintf(newenv, len, "%s=%s", s1, s2);
5204 if (putenv(newenv)) {
Neal Norwitz4adc9ab2003-02-10 03:10:43 +00005205 Py_DECREF(newstr);
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005206 posix_error();
5207 return NULL;
5208 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005209#endif
Fred Drake762e2061999-08-26 17:23:54 +00005210 /* Install the first arg and newstr in posix_putenv_garbage;
5211 * this will cause previous value to be collected. This has to
5212 * happen after the real putenv() call because the old value
5213 * was still accessible until then. */
5214 if (PyDict_SetItem(posix_putenv_garbage,
5215 PyTuple_GET_ITEM(args, 0), newstr)) {
5216 /* really not much we can do; just leak */
5217 PyErr_Clear();
5218 }
5219 else {
5220 Py_DECREF(newstr);
5221 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00005222
5223#if defined(PYOS_OS2)
5224 }
5225#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00005226 Py_INCREF(Py_None);
5227 return Py_None;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005228}
Guido van Rossumb6a47161997-09-15 22:54:34 +00005229#endif /* putenv */
5230
Guido van Rossumc524d952001-10-19 01:31:59 +00005231#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005232PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005233"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005234Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00005235
5236static PyObject *
5237posix_unsetenv(PyObject *self, PyObject *args)
5238{
5239 char *s1;
5240
5241 if (!PyArg_ParseTuple(args, "s:unsetenv", &s1))
5242 return NULL;
5243
5244 unsetenv(s1);
5245
5246 /* Remove the key from posix_putenv_garbage;
5247 * this will cause it to be collected. This has to
Tim Peters5aa91602002-01-30 05:46:57 +00005248 * happen after the real unsetenv() call because the
Guido van Rossumc524d952001-10-19 01:31:59 +00005249 * old value was still accessible until then.
5250 */
5251 if (PyDict_DelItem(posix_putenv_garbage,
5252 PyTuple_GET_ITEM(args, 0))) {
5253 /* really not much we can do; just leak */
5254 PyErr_Clear();
5255 }
5256
5257 Py_INCREF(Py_None);
5258 return Py_None;
5259}
5260#endif /* unsetenv */
5261
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005262PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005263"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005264Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005265
Guido van Rossumf68d8e52001-04-14 17:55:09 +00005266static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005267posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00005268{
5269 int code;
5270 char *message;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005271 if (!PyArg_ParseTuple(args, "i:strerror", &code))
Guido van Rossumb6a47161997-09-15 22:54:34 +00005272 return NULL;
5273 message = strerror(code);
5274 if (message == NULL) {
5275 PyErr_SetString(PyExc_ValueError,
Fred Drake661ea262000-10-24 19:57:45 +00005276 "strerror() argument out of range");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005277 return NULL;
5278 }
Neal Norwitz93c56822007-08-26 07:10:06 +00005279 return PyUnicode_FromString(message);
Guido van Rossumb6a47161997-09-15 22:54:34 +00005280}
Guido van Rossumb6a47161997-09-15 22:54:34 +00005281
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005282
Guido van Rossumc9641791998-08-04 15:26:23 +00005283#ifdef HAVE_SYS_WAIT_H
5284
Fred Drake106c1a02002-04-23 15:58:02 +00005285#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005286PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005287"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005288Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00005289
5290static PyObject *
5291posix_WCOREDUMP(PyObject *self, PyObject *args)
5292{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005293 WAIT_TYPE status;
5294 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00005295
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005296 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00005297 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00005298
5299 return PyBool_FromLong(WCOREDUMP(status));
Fred Drake106c1a02002-04-23 15:58:02 +00005300}
5301#endif /* WCOREDUMP */
5302
5303#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005304PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005305"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00005306Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005307job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00005308
5309static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005310posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00005311{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005312 WAIT_TYPE status;
5313 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00005314
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005315 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00005316 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00005317
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005318 return PyBool_FromLong(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00005319}
5320#endif /* WIFCONTINUED */
5321
Guido van Rossumc9641791998-08-04 15:26:23 +00005322#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005323PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005324"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005325Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005326
5327static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005328posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005329{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005330 WAIT_TYPE status;
5331 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005332
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005333 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005334 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005335
Fred Drake106c1a02002-04-23 15:58:02 +00005336 return PyBool_FromLong(WIFSTOPPED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005337}
5338#endif /* WIFSTOPPED */
5339
5340#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005341PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005342"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005343Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005344
5345static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005346posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005347{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005348 WAIT_TYPE status;
5349 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005350
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005351 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005352 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005353
Fred Drake106c1a02002-04-23 15:58:02 +00005354 return PyBool_FromLong(WIFSIGNALED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005355}
5356#endif /* WIFSIGNALED */
5357
5358#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005359PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005360"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005361Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005362system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005363
5364static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005365posix_WIFEXITED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005366{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005367 WAIT_TYPE status;
5368 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005369
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005370 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005371 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005372
Fred Drake106c1a02002-04-23 15:58:02 +00005373 return PyBool_FromLong(WIFEXITED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005374}
5375#endif /* WIFEXITED */
5376
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005377#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005378PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005379"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005380Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005381
5382static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005383posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005384{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005385 WAIT_TYPE status;
5386 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005387
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005388 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005389 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005390
Guido van Rossumc9641791998-08-04 15:26:23 +00005391 return Py_BuildValue("i", WEXITSTATUS(status));
5392}
5393#endif /* WEXITSTATUS */
5394
5395#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005396PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005397"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005398Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005399value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005400
5401static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005402posix_WTERMSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005403{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005404 WAIT_TYPE status;
5405 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005406
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005407 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005408 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005409
Guido van Rossumc9641791998-08-04 15:26:23 +00005410 return Py_BuildValue("i", WTERMSIG(status));
5411}
5412#endif /* WTERMSIG */
5413
5414#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005415PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005416"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005417Return the signal that stopped the process that provided\n\
5418the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005419
5420static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005421posix_WSTOPSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005422{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005423 WAIT_TYPE status;
5424 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005425
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005426 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005427 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005428
Guido van Rossumc9641791998-08-04 15:26:23 +00005429 return Py_BuildValue("i", WSTOPSIG(status));
5430}
5431#endif /* WSTOPSIG */
5432
5433#endif /* HAVE_SYS_WAIT_H */
5434
5435
Thomas Wouters477c8d52006-05-27 19:21:47 +00005436#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossumd5753e11999-10-19 13:29:23 +00005437#ifdef _SCO_DS
5438/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
5439 needed definitions in sys/statvfs.h */
5440#define _SVID3
5441#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00005442#include <sys/statvfs.h>
5443
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005444static PyObject*
5445_pystatvfs_fromstructstatvfs(struct statvfs st) {
5446 PyObject *v = PyStructSequence_New(&StatVFSResultType);
5447 if (v == NULL)
5448 return NULL;
5449
5450#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005451 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
5452 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
5453 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long) st.f_blocks));
5454 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long) st.f_bfree));
5455 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long) st.f_bavail));
5456 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long) st.f_files));
5457 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong((long) st.f_ffree));
5458 PyStructSequence_SET_ITEM(v, 7, PyLong_FromLong((long) st.f_favail));
5459 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
5460 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005461#else
Christian Heimes217cfd12007-12-02 14:31:20 +00005462 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
5463 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
Tim Peters5aa91602002-01-30 05:46:57 +00005464 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005465 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
Tim Peters5aa91602002-01-30 05:46:57 +00005466 PyStructSequence_SET_ITEM(v, 3,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005467 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005468 PyStructSequence_SET_ITEM(v, 4,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005469 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
Tim Peters5aa91602002-01-30 05:46:57 +00005470 PyStructSequence_SET_ITEM(v, 5,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005471 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
Tim Peters5aa91602002-01-30 05:46:57 +00005472 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005473 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
Tim Peters5aa91602002-01-30 05:46:57 +00005474 PyStructSequence_SET_ITEM(v, 7,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005475 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
Christian Heimes217cfd12007-12-02 14:31:20 +00005476 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
5477 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005478#endif
5479
5480 return v;
5481}
5482
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005483PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005484"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005485Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005486
5487static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005488posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005489{
5490 int fd, res;
5491 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005492
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005493 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005494 return NULL;
5495 Py_BEGIN_ALLOW_THREADS
5496 res = fstatvfs(fd, &st);
5497 Py_END_ALLOW_THREADS
5498 if (res != 0)
5499 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005500
5501 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005502}
Thomas Wouters477c8d52006-05-27 19:21:47 +00005503#endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00005504
5505
Thomas Wouters477c8d52006-05-27 19:21:47 +00005506#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005507#include <sys/statvfs.h>
5508
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005509PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005510"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005511Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005512
5513static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005514posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005515{
5516 char *path;
5517 int res;
5518 struct statvfs st;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005519 if (!PyArg_ParseTuple(args, "s:statvfs", &path))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005520 return NULL;
5521 Py_BEGIN_ALLOW_THREADS
5522 res = statvfs(path, &st);
5523 Py_END_ALLOW_THREADS
5524 if (res != 0)
5525 return posix_error_with_filename(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005526
5527 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005528}
5529#endif /* HAVE_STATVFS */
5530
Fred Drakec9680921999-12-13 16:37:25 +00005531/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
5532 * It maps strings representing configuration variable names to
5533 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00005534 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00005535 * rarely-used constants. There are three separate tables that use
5536 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00005537 *
5538 * This code is always included, even if none of the interfaces that
5539 * need it are included. The #if hackery needed to avoid it would be
5540 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00005541 */
5542struct constdef {
5543 char *name;
5544 long value;
5545};
5546
Fred Drake12c6e2d1999-12-14 21:25:03 +00005547static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005548conv_confname(PyObject *arg, int *valuep, struct constdef *table,
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005549 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00005550{
Christian Heimes217cfd12007-12-02 14:31:20 +00005551 if (PyLong_Check(arg)) {
5552 *valuep = PyLong_AS_LONG(arg);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005553 return 1;
5554 }
Guido van Rossumbce56a62007-05-10 18:04:33 +00005555 else {
Fred Drake12c6e2d1999-12-14 21:25:03 +00005556 /* look up the value in the table using a binary search */
Fred Drake699f3522000-06-29 21:12:41 +00005557 size_t lo = 0;
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005558 size_t mid;
Fred Drake699f3522000-06-29 21:12:41 +00005559 size_t hi = tablesize;
5560 int cmp;
Guido van Rossumbce56a62007-05-10 18:04:33 +00005561 const char *confname;
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005562 if (!PyUnicode_Check(arg)) {
Guido van Rossumbce56a62007-05-10 18:04:33 +00005563 PyErr_SetString(PyExc_TypeError,
5564 "configuration names must be strings or integers");
5565 return 0;
5566 }
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005567 confname = PyUnicode_AsString(arg);
5568 if (confname == NULL)
5569 return 0;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005570 while (lo < hi) {
5571 mid = (lo + hi) / 2;
5572 cmp = strcmp(confname, table[mid].name);
5573 if (cmp < 0)
5574 hi = mid;
5575 else if (cmp > 0)
5576 lo = mid + 1;
5577 else {
5578 *valuep = table[mid].value;
5579 return 1;
5580 }
5581 }
5582 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
Guido van Rossumbce56a62007-05-10 18:04:33 +00005583 return 0;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005584 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00005585}
5586
5587
5588#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
5589static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00005590#ifdef _PC_ABI_AIO_XFER_MAX
5591 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
5592#endif
5593#ifdef _PC_ABI_ASYNC_IO
5594 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
5595#endif
Fred Drakec9680921999-12-13 16:37:25 +00005596#ifdef _PC_ASYNC_IO
5597 {"PC_ASYNC_IO", _PC_ASYNC_IO},
5598#endif
5599#ifdef _PC_CHOWN_RESTRICTED
5600 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
5601#endif
5602#ifdef _PC_FILESIZEBITS
5603 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
5604#endif
5605#ifdef _PC_LAST
5606 {"PC_LAST", _PC_LAST},
5607#endif
5608#ifdef _PC_LINK_MAX
5609 {"PC_LINK_MAX", _PC_LINK_MAX},
5610#endif
5611#ifdef _PC_MAX_CANON
5612 {"PC_MAX_CANON", _PC_MAX_CANON},
5613#endif
5614#ifdef _PC_MAX_INPUT
5615 {"PC_MAX_INPUT", _PC_MAX_INPUT},
5616#endif
5617#ifdef _PC_NAME_MAX
5618 {"PC_NAME_MAX", _PC_NAME_MAX},
5619#endif
5620#ifdef _PC_NO_TRUNC
5621 {"PC_NO_TRUNC", _PC_NO_TRUNC},
5622#endif
5623#ifdef _PC_PATH_MAX
5624 {"PC_PATH_MAX", _PC_PATH_MAX},
5625#endif
5626#ifdef _PC_PIPE_BUF
5627 {"PC_PIPE_BUF", _PC_PIPE_BUF},
5628#endif
5629#ifdef _PC_PRIO_IO
5630 {"PC_PRIO_IO", _PC_PRIO_IO},
5631#endif
5632#ifdef _PC_SOCK_MAXBUF
5633 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
5634#endif
5635#ifdef _PC_SYNC_IO
5636 {"PC_SYNC_IO", _PC_SYNC_IO},
5637#endif
5638#ifdef _PC_VDISABLE
5639 {"PC_VDISABLE", _PC_VDISABLE},
5640#endif
5641};
5642
Fred Drakec9680921999-12-13 16:37:25 +00005643static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005644conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00005645{
5646 return conv_confname(arg, valuep, posix_constants_pathconf,
5647 sizeof(posix_constants_pathconf)
5648 / sizeof(struct constdef));
5649}
5650#endif
5651
5652#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005653PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005654"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00005655Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005656If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00005657
5658static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005659posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005660{
5661 PyObject *result = NULL;
5662 int name, fd;
5663
Fred Drake12c6e2d1999-12-14 21:25:03 +00005664 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
5665 conv_path_confname, &name)) {
Fred Drakec9680921999-12-13 16:37:25 +00005666 long limit;
5667
5668 errno = 0;
5669 limit = fpathconf(fd, name);
5670 if (limit == -1 && errno != 0)
5671 posix_error();
5672 else
Christian Heimes217cfd12007-12-02 14:31:20 +00005673 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00005674 }
5675 return result;
5676}
5677#endif
5678
5679
5680#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005681PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005682"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00005683Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005684If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00005685
5686static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005687posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005688{
5689 PyObject *result = NULL;
5690 int name;
5691 char *path;
5692
5693 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
5694 conv_path_confname, &name)) {
5695 long limit;
5696
5697 errno = 0;
5698 limit = pathconf(path, name);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005699 if (limit == -1 && errno != 0) {
Fred Drakec9680921999-12-13 16:37:25 +00005700 if (errno == EINVAL)
5701 /* could be a path or name problem */
5702 posix_error();
5703 else
5704 posix_error_with_filename(path);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005705 }
Fred Drakec9680921999-12-13 16:37:25 +00005706 else
Christian Heimes217cfd12007-12-02 14:31:20 +00005707 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00005708 }
5709 return result;
5710}
5711#endif
5712
5713#ifdef HAVE_CONFSTR
5714static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00005715#ifdef _CS_ARCHITECTURE
5716 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
5717#endif
5718#ifdef _CS_HOSTNAME
5719 {"CS_HOSTNAME", _CS_HOSTNAME},
5720#endif
5721#ifdef _CS_HW_PROVIDER
5722 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
5723#endif
5724#ifdef _CS_HW_SERIAL
5725 {"CS_HW_SERIAL", _CS_HW_SERIAL},
5726#endif
5727#ifdef _CS_INITTAB_NAME
5728 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
5729#endif
Fred Drakec9680921999-12-13 16:37:25 +00005730#ifdef _CS_LFS64_CFLAGS
5731 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
5732#endif
5733#ifdef _CS_LFS64_LDFLAGS
5734 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
5735#endif
5736#ifdef _CS_LFS64_LIBS
5737 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
5738#endif
5739#ifdef _CS_LFS64_LINTFLAGS
5740 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
5741#endif
5742#ifdef _CS_LFS_CFLAGS
5743 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
5744#endif
5745#ifdef _CS_LFS_LDFLAGS
5746 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
5747#endif
5748#ifdef _CS_LFS_LIBS
5749 {"CS_LFS_LIBS", _CS_LFS_LIBS},
5750#endif
5751#ifdef _CS_LFS_LINTFLAGS
5752 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
5753#endif
Fred Draked86ed291999-12-15 15:34:33 +00005754#ifdef _CS_MACHINE
5755 {"CS_MACHINE", _CS_MACHINE},
5756#endif
Fred Drakec9680921999-12-13 16:37:25 +00005757#ifdef _CS_PATH
5758 {"CS_PATH", _CS_PATH},
5759#endif
Fred Draked86ed291999-12-15 15:34:33 +00005760#ifdef _CS_RELEASE
5761 {"CS_RELEASE", _CS_RELEASE},
5762#endif
5763#ifdef _CS_SRPC_DOMAIN
5764 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
5765#endif
5766#ifdef _CS_SYSNAME
5767 {"CS_SYSNAME", _CS_SYSNAME},
5768#endif
5769#ifdef _CS_VERSION
5770 {"CS_VERSION", _CS_VERSION},
5771#endif
Fred Drakec9680921999-12-13 16:37:25 +00005772#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
5773 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
5774#endif
5775#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
5776 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
5777#endif
5778#ifdef _CS_XBS5_ILP32_OFF32_LIBS
5779 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
5780#endif
5781#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
5782 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
5783#endif
5784#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
5785 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
5786#endif
5787#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
5788 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
5789#endif
5790#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
5791 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
5792#endif
5793#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
5794 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
5795#endif
5796#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
5797 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
5798#endif
5799#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
5800 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
5801#endif
5802#ifdef _CS_XBS5_LP64_OFF64_LIBS
5803 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
5804#endif
5805#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
5806 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
5807#endif
5808#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
5809 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
5810#endif
5811#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
5812 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
5813#endif
5814#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
5815 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
5816#endif
5817#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
5818 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
5819#endif
Fred Draked86ed291999-12-15 15:34:33 +00005820#ifdef _MIPS_CS_AVAIL_PROCESSORS
5821 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
5822#endif
5823#ifdef _MIPS_CS_BASE
5824 {"MIPS_CS_BASE", _MIPS_CS_BASE},
5825#endif
5826#ifdef _MIPS_CS_HOSTID
5827 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
5828#endif
5829#ifdef _MIPS_CS_HW_NAME
5830 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
5831#endif
5832#ifdef _MIPS_CS_NUM_PROCESSORS
5833 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
5834#endif
5835#ifdef _MIPS_CS_OSREL_MAJ
5836 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
5837#endif
5838#ifdef _MIPS_CS_OSREL_MIN
5839 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
5840#endif
5841#ifdef _MIPS_CS_OSREL_PATCH
5842 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
5843#endif
5844#ifdef _MIPS_CS_OS_NAME
5845 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
5846#endif
5847#ifdef _MIPS_CS_OS_PROVIDER
5848 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
5849#endif
5850#ifdef _MIPS_CS_PROCESSORS
5851 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
5852#endif
5853#ifdef _MIPS_CS_SERIAL
5854 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
5855#endif
5856#ifdef _MIPS_CS_VENDOR
5857 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
5858#endif
Fred Drakec9680921999-12-13 16:37:25 +00005859};
5860
5861static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005862conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00005863{
5864 return conv_confname(arg, valuep, posix_constants_confstr,
5865 sizeof(posix_constants_confstr)
5866 / sizeof(struct constdef));
5867}
5868
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005869PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005870"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005871Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00005872
5873static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005874posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005875{
5876 PyObject *result = NULL;
5877 int name;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005878 char buffer[256];
Fred Drakec9680921999-12-13 16:37:25 +00005879
5880 if (PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name)) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005881 int len;
Fred Drakec9680921999-12-13 16:37:25 +00005882
Fred Drakec9680921999-12-13 16:37:25 +00005883 errno = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005884 len = confstr(name, buffer, sizeof(buffer));
5885 if (len == 0) {
5886 if (errno) {
5887 posix_error();
5888 }
5889 else {
5890 result = Py_None;
5891 Py_INCREF(Py_None);
5892 }
Fred Drakec9680921999-12-13 16:37:25 +00005893 }
5894 else {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005895 if ((unsigned int)len >= sizeof(buffer)) {
Neal Norwitz93c56822007-08-26 07:10:06 +00005896 result = PyUnicode_FromStringAndSize(NULL, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00005897 if (result != NULL)
Neal Norwitz93c56822007-08-26 07:10:06 +00005898 confstr(name, PyUnicode_AsString(result), len);
Fred Drakec9680921999-12-13 16:37:25 +00005899 }
5900 else
Neal Norwitz93c56822007-08-26 07:10:06 +00005901 result = PyUnicode_FromStringAndSize(buffer, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00005902 }
5903 }
5904 return result;
5905}
5906#endif
5907
5908
5909#ifdef HAVE_SYSCONF
5910static struct constdef posix_constants_sysconf[] = {
5911#ifdef _SC_2_CHAR_TERM
5912 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
5913#endif
5914#ifdef _SC_2_C_BIND
5915 {"SC_2_C_BIND", _SC_2_C_BIND},
5916#endif
5917#ifdef _SC_2_C_DEV
5918 {"SC_2_C_DEV", _SC_2_C_DEV},
5919#endif
5920#ifdef _SC_2_C_VERSION
5921 {"SC_2_C_VERSION", _SC_2_C_VERSION},
5922#endif
5923#ifdef _SC_2_FORT_DEV
5924 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
5925#endif
5926#ifdef _SC_2_FORT_RUN
5927 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
5928#endif
5929#ifdef _SC_2_LOCALEDEF
5930 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
5931#endif
5932#ifdef _SC_2_SW_DEV
5933 {"SC_2_SW_DEV", _SC_2_SW_DEV},
5934#endif
5935#ifdef _SC_2_UPE
5936 {"SC_2_UPE", _SC_2_UPE},
5937#endif
5938#ifdef _SC_2_VERSION
5939 {"SC_2_VERSION", _SC_2_VERSION},
5940#endif
Fred Draked86ed291999-12-15 15:34:33 +00005941#ifdef _SC_ABI_ASYNCHRONOUS_IO
5942 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
5943#endif
5944#ifdef _SC_ACL
5945 {"SC_ACL", _SC_ACL},
5946#endif
Fred Drakec9680921999-12-13 16:37:25 +00005947#ifdef _SC_AIO_LISTIO_MAX
5948 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
5949#endif
Fred Drakec9680921999-12-13 16:37:25 +00005950#ifdef _SC_AIO_MAX
5951 {"SC_AIO_MAX", _SC_AIO_MAX},
5952#endif
5953#ifdef _SC_AIO_PRIO_DELTA_MAX
5954 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
5955#endif
5956#ifdef _SC_ARG_MAX
5957 {"SC_ARG_MAX", _SC_ARG_MAX},
5958#endif
5959#ifdef _SC_ASYNCHRONOUS_IO
5960 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
5961#endif
5962#ifdef _SC_ATEXIT_MAX
5963 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
5964#endif
Fred Draked86ed291999-12-15 15:34:33 +00005965#ifdef _SC_AUDIT
5966 {"SC_AUDIT", _SC_AUDIT},
5967#endif
Fred Drakec9680921999-12-13 16:37:25 +00005968#ifdef _SC_AVPHYS_PAGES
5969 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
5970#endif
5971#ifdef _SC_BC_BASE_MAX
5972 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
5973#endif
5974#ifdef _SC_BC_DIM_MAX
5975 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
5976#endif
5977#ifdef _SC_BC_SCALE_MAX
5978 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
5979#endif
5980#ifdef _SC_BC_STRING_MAX
5981 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
5982#endif
Fred Draked86ed291999-12-15 15:34:33 +00005983#ifdef _SC_CAP
5984 {"SC_CAP", _SC_CAP},
5985#endif
Fred Drakec9680921999-12-13 16:37:25 +00005986#ifdef _SC_CHARCLASS_NAME_MAX
5987 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
5988#endif
5989#ifdef _SC_CHAR_BIT
5990 {"SC_CHAR_BIT", _SC_CHAR_BIT},
5991#endif
5992#ifdef _SC_CHAR_MAX
5993 {"SC_CHAR_MAX", _SC_CHAR_MAX},
5994#endif
5995#ifdef _SC_CHAR_MIN
5996 {"SC_CHAR_MIN", _SC_CHAR_MIN},
5997#endif
5998#ifdef _SC_CHILD_MAX
5999 {"SC_CHILD_MAX", _SC_CHILD_MAX},
6000#endif
6001#ifdef _SC_CLK_TCK
6002 {"SC_CLK_TCK", _SC_CLK_TCK},
6003#endif
6004#ifdef _SC_COHER_BLKSZ
6005 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
6006#endif
6007#ifdef _SC_COLL_WEIGHTS_MAX
6008 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
6009#endif
6010#ifdef _SC_DCACHE_ASSOC
6011 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
6012#endif
6013#ifdef _SC_DCACHE_BLKSZ
6014 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
6015#endif
6016#ifdef _SC_DCACHE_LINESZ
6017 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
6018#endif
6019#ifdef _SC_DCACHE_SZ
6020 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
6021#endif
6022#ifdef _SC_DCACHE_TBLKSZ
6023 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
6024#endif
6025#ifdef _SC_DELAYTIMER_MAX
6026 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
6027#endif
6028#ifdef _SC_EQUIV_CLASS_MAX
6029 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
6030#endif
6031#ifdef _SC_EXPR_NEST_MAX
6032 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
6033#endif
6034#ifdef _SC_FSYNC
6035 {"SC_FSYNC", _SC_FSYNC},
6036#endif
6037#ifdef _SC_GETGR_R_SIZE_MAX
6038 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
6039#endif
6040#ifdef _SC_GETPW_R_SIZE_MAX
6041 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
6042#endif
6043#ifdef _SC_ICACHE_ASSOC
6044 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
6045#endif
6046#ifdef _SC_ICACHE_BLKSZ
6047 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
6048#endif
6049#ifdef _SC_ICACHE_LINESZ
6050 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
6051#endif
6052#ifdef _SC_ICACHE_SZ
6053 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
6054#endif
Fred Draked86ed291999-12-15 15:34:33 +00006055#ifdef _SC_INF
6056 {"SC_INF", _SC_INF},
6057#endif
Fred Drakec9680921999-12-13 16:37:25 +00006058#ifdef _SC_INT_MAX
6059 {"SC_INT_MAX", _SC_INT_MAX},
6060#endif
6061#ifdef _SC_INT_MIN
6062 {"SC_INT_MIN", _SC_INT_MIN},
6063#endif
6064#ifdef _SC_IOV_MAX
6065 {"SC_IOV_MAX", _SC_IOV_MAX},
6066#endif
Fred Draked86ed291999-12-15 15:34:33 +00006067#ifdef _SC_IP_SECOPTS
6068 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
6069#endif
Fred Drakec9680921999-12-13 16:37:25 +00006070#ifdef _SC_JOB_CONTROL
6071 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
6072#endif
Fred Draked86ed291999-12-15 15:34:33 +00006073#ifdef _SC_KERN_POINTERS
6074 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
6075#endif
6076#ifdef _SC_KERN_SIM
6077 {"SC_KERN_SIM", _SC_KERN_SIM},
6078#endif
Fred Drakec9680921999-12-13 16:37:25 +00006079#ifdef _SC_LINE_MAX
6080 {"SC_LINE_MAX", _SC_LINE_MAX},
6081#endif
6082#ifdef _SC_LOGIN_NAME_MAX
6083 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
6084#endif
6085#ifdef _SC_LOGNAME_MAX
6086 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
6087#endif
6088#ifdef _SC_LONG_BIT
6089 {"SC_LONG_BIT", _SC_LONG_BIT},
6090#endif
Fred Draked86ed291999-12-15 15:34:33 +00006091#ifdef _SC_MAC
6092 {"SC_MAC", _SC_MAC},
6093#endif
Fred Drakec9680921999-12-13 16:37:25 +00006094#ifdef _SC_MAPPED_FILES
6095 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
6096#endif
6097#ifdef _SC_MAXPID
6098 {"SC_MAXPID", _SC_MAXPID},
6099#endif
6100#ifdef _SC_MB_LEN_MAX
6101 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
6102#endif
6103#ifdef _SC_MEMLOCK
6104 {"SC_MEMLOCK", _SC_MEMLOCK},
6105#endif
6106#ifdef _SC_MEMLOCK_RANGE
6107 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
6108#endif
6109#ifdef _SC_MEMORY_PROTECTION
6110 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
6111#endif
6112#ifdef _SC_MESSAGE_PASSING
6113 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
6114#endif
Fred Draked86ed291999-12-15 15:34:33 +00006115#ifdef _SC_MMAP_FIXED_ALIGNMENT
6116 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
6117#endif
Fred Drakec9680921999-12-13 16:37:25 +00006118#ifdef _SC_MQ_OPEN_MAX
6119 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
6120#endif
6121#ifdef _SC_MQ_PRIO_MAX
6122 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
6123#endif
Fred Draked86ed291999-12-15 15:34:33 +00006124#ifdef _SC_NACLS_MAX
6125 {"SC_NACLS_MAX", _SC_NACLS_MAX},
6126#endif
Fred Drakec9680921999-12-13 16:37:25 +00006127#ifdef _SC_NGROUPS_MAX
6128 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
6129#endif
6130#ifdef _SC_NL_ARGMAX
6131 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
6132#endif
6133#ifdef _SC_NL_LANGMAX
6134 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
6135#endif
6136#ifdef _SC_NL_MSGMAX
6137 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
6138#endif
6139#ifdef _SC_NL_NMAX
6140 {"SC_NL_NMAX", _SC_NL_NMAX},
6141#endif
6142#ifdef _SC_NL_SETMAX
6143 {"SC_NL_SETMAX", _SC_NL_SETMAX},
6144#endif
6145#ifdef _SC_NL_TEXTMAX
6146 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
6147#endif
6148#ifdef _SC_NPROCESSORS_CONF
6149 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
6150#endif
6151#ifdef _SC_NPROCESSORS_ONLN
6152 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
6153#endif
Fred Draked86ed291999-12-15 15:34:33 +00006154#ifdef _SC_NPROC_CONF
6155 {"SC_NPROC_CONF", _SC_NPROC_CONF},
6156#endif
6157#ifdef _SC_NPROC_ONLN
6158 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
6159#endif
Fred Drakec9680921999-12-13 16:37:25 +00006160#ifdef _SC_NZERO
6161 {"SC_NZERO", _SC_NZERO},
6162#endif
6163#ifdef _SC_OPEN_MAX
6164 {"SC_OPEN_MAX", _SC_OPEN_MAX},
6165#endif
6166#ifdef _SC_PAGESIZE
6167 {"SC_PAGESIZE", _SC_PAGESIZE},
6168#endif
6169#ifdef _SC_PAGE_SIZE
6170 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
6171#endif
6172#ifdef _SC_PASS_MAX
6173 {"SC_PASS_MAX", _SC_PASS_MAX},
6174#endif
6175#ifdef _SC_PHYS_PAGES
6176 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
6177#endif
6178#ifdef _SC_PII
6179 {"SC_PII", _SC_PII},
6180#endif
6181#ifdef _SC_PII_INTERNET
6182 {"SC_PII_INTERNET", _SC_PII_INTERNET},
6183#endif
6184#ifdef _SC_PII_INTERNET_DGRAM
6185 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
6186#endif
6187#ifdef _SC_PII_INTERNET_STREAM
6188 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
6189#endif
6190#ifdef _SC_PII_OSI
6191 {"SC_PII_OSI", _SC_PII_OSI},
6192#endif
6193#ifdef _SC_PII_OSI_CLTS
6194 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
6195#endif
6196#ifdef _SC_PII_OSI_COTS
6197 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
6198#endif
6199#ifdef _SC_PII_OSI_M
6200 {"SC_PII_OSI_M", _SC_PII_OSI_M},
6201#endif
6202#ifdef _SC_PII_SOCKET
6203 {"SC_PII_SOCKET", _SC_PII_SOCKET},
6204#endif
6205#ifdef _SC_PII_XTI
6206 {"SC_PII_XTI", _SC_PII_XTI},
6207#endif
6208#ifdef _SC_POLL
6209 {"SC_POLL", _SC_POLL},
6210#endif
6211#ifdef _SC_PRIORITIZED_IO
6212 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
6213#endif
6214#ifdef _SC_PRIORITY_SCHEDULING
6215 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
6216#endif
6217#ifdef _SC_REALTIME_SIGNALS
6218 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
6219#endif
6220#ifdef _SC_RE_DUP_MAX
6221 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
6222#endif
6223#ifdef _SC_RTSIG_MAX
6224 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
6225#endif
6226#ifdef _SC_SAVED_IDS
6227 {"SC_SAVED_IDS", _SC_SAVED_IDS},
6228#endif
6229#ifdef _SC_SCHAR_MAX
6230 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
6231#endif
6232#ifdef _SC_SCHAR_MIN
6233 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
6234#endif
6235#ifdef _SC_SELECT
6236 {"SC_SELECT", _SC_SELECT},
6237#endif
6238#ifdef _SC_SEMAPHORES
6239 {"SC_SEMAPHORES", _SC_SEMAPHORES},
6240#endif
6241#ifdef _SC_SEM_NSEMS_MAX
6242 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
6243#endif
6244#ifdef _SC_SEM_VALUE_MAX
6245 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
6246#endif
6247#ifdef _SC_SHARED_MEMORY_OBJECTS
6248 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
6249#endif
6250#ifdef _SC_SHRT_MAX
6251 {"SC_SHRT_MAX", _SC_SHRT_MAX},
6252#endif
6253#ifdef _SC_SHRT_MIN
6254 {"SC_SHRT_MIN", _SC_SHRT_MIN},
6255#endif
6256#ifdef _SC_SIGQUEUE_MAX
6257 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
6258#endif
6259#ifdef _SC_SIGRT_MAX
6260 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
6261#endif
6262#ifdef _SC_SIGRT_MIN
6263 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
6264#endif
Fred Draked86ed291999-12-15 15:34:33 +00006265#ifdef _SC_SOFTPOWER
6266 {"SC_SOFTPOWER", _SC_SOFTPOWER},
6267#endif
Fred Drakec9680921999-12-13 16:37:25 +00006268#ifdef _SC_SPLIT_CACHE
6269 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
6270#endif
6271#ifdef _SC_SSIZE_MAX
6272 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
6273#endif
6274#ifdef _SC_STACK_PROT
6275 {"SC_STACK_PROT", _SC_STACK_PROT},
6276#endif
6277#ifdef _SC_STREAM_MAX
6278 {"SC_STREAM_MAX", _SC_STREAM_MAX},
6279#endif
6280#ifdef _SC_SYNCHRONIZED_IO
6281 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
6282#endif
6283#ifdef _SC_THREADS
6284 {"SC_THREADS", _SC_THREADS},
6285#endif
6286#ifdef _SC_THREAD_ATTR_STACKADDR
6287 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
6288#endif
6289#ifdef _SC_THREAD_ATTR_STACKSIZE
6290 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
6291#endif
6292#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
6293 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
6294#endif
6295#ifdef _SC_THREAD_KEYS_MAX
6296 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
6297#endif
6298#ifdef _SC_THREAD_PRIORITY_SCHEDULING
6299 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
6300#endif
6301#ifdef _SC_THREAD_PRIO_INHERIT
6302 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
6303#endif
6304#ifdef _SC_THREAD_PRIO_PROTECT
6305 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
6306#endif
6307#ifdef _SC_THREAD_PROCESS_SHARED
6308 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
6309#endif
6310#ifdef _SC_THREAD_SAFE_FUNCTIONS
6311 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
6312#endif
6313#ifdef _SC_THREAD_STACK_MIN
6314 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
6315#endif
6316#ifdef _SC_THREAD_THREADS_MAX
6317 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
6318#endif
6319#ifdef _SC_TIMERS
6320 {"SC_TIMERS", _SC_TIMERS},
6321#endif
6322#ifdef _SC_TIMER_MAX
6323 {"SC_TIMER_MAX", _SC_TIMER_MAX},
6324#endif
6325#ifdef _SC_TTY_NAME_MAX
6326 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
6327#endif
6328#ifdef _SC_TZNAME_MAX
6329 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
6330#endif
6331#ifdef _SC_T_IOV_MAX
6332 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
6333#endif
6334#ifdef _SC_UCHAR_MAX
6335 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
6336#endif
6337#ifdef _SC_UINT_MAX
6338 {"SC_UINT_MAX", _SC_UINT_MAX},
6339#endif
6340#ifdef _SC_UIO_MAXIOV
6341 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
6342#endif
6343#ifdef _SC_ULONG_MAX
6344 {"SC_ULONG_MAX", _SC_ULONG_MAX},
6345#endif
6346#ifdef _SC_USHRT_MAX
6347 {"SC_USHRT_MAX", _SC_USHRT_MAX},
6348#endif
6349#ifdef _SC_VERSION
6350 {"SC_VERSION", _SC_VERSION},
6351#endif
6352#ifdef _SC_WORD_BIT
6353 {"SC_WORD_BIT", _SC_WORD_BIT},
6354#endif
6355#ifdef _SC_XBS5_ILP32_OFF32
6356 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
6357#endif
6358#ifdef _SC_XBS5_ILP32_OFFBIG
6359 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
6360#endif
6361#ifdef _SC_XBS5_LP64_OFF64
6362 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
6363#endif
6364#ifdef _SC_XBS5_LPBIG_OFFBIG
6365 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
6366#endif
6367#ifdef _SC_XOPEN_CRYPT
6368 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
6369#endif
6370#ifdef _SC_XOPEN_ENH_I18N
6371 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
6372#endif
6373#ifdef _SC_XOPEN_LEGACY
6374 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
6375#endif
6376#ifdef _SC_XOPEN_REALTIME
6377 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
6378#endif
6379#ifdef _SC_XOPEN_REALTIME_THREADS
6380 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
6381#endif
6382#ifdef _SC_XOPEN_SHM
6383 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
6384#endif
6385#ifdef _SC_XOPEN_UNIX
6386 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
6387#endif
6388#ifdef _SC_XOPEN_VERSION
6389 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
6390#endif
6391#ifdef _SC_XOPEN_XCU_VERSION
6392 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
6393#endif
6394#ifdef _SC_XOPEN_XPG2
6395 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
6396#endif
6397#ifdef _SC_XOPEN_XPG3
6398 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
6399#endif
6400#ifdef _SC_XOPEN_XPG4
6401 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
6402#endif
6403};
6404
6405static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006406conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00006407{
6408 return conv_confname(arg, valuep, posix_constants_sysconf,
6409 sizeof(posix_constants_sysconf)
6410 / sizeof(struct constdef));
6411}
6412
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006413PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006414"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006415Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00006416
6417static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006418posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006419{
6420 PyObject *result = NULL;
6421 int name;
6422
6423 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
6424 int value;
6425
6426 errno = 0;
6427 value = sysconf(name);
6428 if (value == -1 && errno != 0)
6429 posix_error();
6430 else
Christian Heimes217cfd12007-12-02 14:31:20 +00006431 result = PyLong_FromLong(value);
Fred Drakec9680921999-12-13 16:37:25 +00006432 }
6433 return result;
6434}
6435#endif
6436
6437
Fred Drakebec628d1999-12-15 18:31:10 +00006438/* This code is used to ensure that the tables of configuration value names
6439 * are in sorted order as required by conv_confname(), and also to build the
6440 * the exported dictionaries that are used to publish information about the
6441 * names available on the host platform.
6442 *
6443 * Sorting the table at runtime ensures that the table is properly ordered
6444 * when used, even for platforms we're not able to test on. It also makes
6445 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00006446 */
Fred Drakebec628d1999-12-15 18:31:10 +00006447
6448static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006449cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00006450{
6451 const struct constdef *c1 =
6452 (const struct constdef *) v1;
6453 const struct constdef *c2 =
6454 (const struct constdef *) v2;
6455
6456 return strcmp(c1->name, c2->name);
6457}
6458
6459static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006460setup_confname_table(struct constdef *table, size_t tablesize,
Fred Drake4d1e64b2002-04-15 19:40:07 +00006461 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00006462{
Fred Drakebec628d1999-12-15 18:31:10 +00006463 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00006464 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00006465
6466 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
6467 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00006468 if (d == NULL)
6469 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006470
Barry Warsaw3155db32000-04-13 15:20:40 +00006471 for (i=0; i < tablesize; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +00006472 PyObject *o = PyLong_FromLong(table[i].value);
Barry Warsaw3155db32000-04-13 15:20:40 +00006473 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
6474 Py_XDECREF(o);
6475 Py_DECREF(d);
6476 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006477 }
Barry Warsaw3155db32000-04-13 15:20:40 +00006478 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00006479 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00006480 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00006481}
6482
Fred Drakebec628d1999-12-15 18:31:10 +00006483/* Return -1 on failure, 0 on success. */
6484static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00006485setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00006486{
6487#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00006488 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00006489 sizeof(posix_constants_pathconf)
6490 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006491 "pathconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006492 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006493#endif
6494#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00006495 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00006496 sizeof(posix_constants_confstr)
6497 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006498 "confstr_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006499 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006500#endif
6501#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00006502 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00006503 sizeof(posix_constants_sysconf)
6504 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006505 "sysconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006506 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006507#endif
Fred Drakebec628d1999-12-15 18:31:10 +00006508 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00006509}
Fred Draked86ed291999-12-15 15:34:33 +00006510
6511
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006512PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006513"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006514Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006515in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006516
6517static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006518posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006519{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006520 abort();
6521 /*NOTREACHED*/
6522 Py_FatalError("abort() called from Python code didn't abort!");
6523 return NULL;
6524}
Fred Drakebec628d1999-12-15 18:31:10 +00006525
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006526#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006527PyDoc_STRVAR(win32_startfile__doc__,
Georg Brandlf4f44152006-02-18 22:29:33 +00006528"startfile(filepath [, operation]) - Start a file with its associated\n\
6529application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00006530\n\
Georg Brandlf4f44152006-02-18 22:29:33 +00006531When \"operation\" is not specified or \"open\", this acts like\n\
6532double-clicking the file in Explorer, or giving the file name as an\n\
6533argument to the DOS \"start\" command: the file is opened with whatever\n\
6534application (if any) its extension is associated.\n\
6535When another \"operation\" is given, it specifies what should be done with\n\
6536the file. A typical operation is \"print\".\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00006537\n\
6538startfile returns as soon as the associated application is launched.\n\
6539There is no option to wait for the application to close, and no way\n\
6540to retrieve the application's exit status.\n\
6541\n\
6542The filepath is relative to the current directory. If you want to use\n\
6543an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006544the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00006545
6546static PyObject *
6547win32_startfile(PyObject *self, PyObject *args)
6548{
6549 char *filepath;
Georg Brandlf4f44152006-02-18 22:29:33 +00006550 char *operation = NULL;
Tim Petersf58a7aa2000-09-22 10:05:54 +00006551 HINSTANCE rc;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006552#ifdef Py_WIN_WIDE_FILENAMES
6553 if (unicode_file_names()) {
6554 PyObject *unipath, *woperation = NULL;
6555 if (!PyArg_ParseTuple(args, "U|s:startfile",
6556 &unipath, &operation)) {
6557 PyErr_Clear();
6558 goto normal;
6559 }
6560
6561
6562 if (operation) {
6563 woperation = PyUnicode_DecodeASCII(operation,
6564 strlen(operation), NULL);
6565 if (!woperation) {
6566 PyErr_Clear();
6567 operation = NULL;
6568 goto normal;
6569 }
6570 }
6571
6572 Py_BEGIN_ALLOW_THREADS
6573 rc = ShellExecuteW((HWND)0, woperation ? PyUnicode_AS_UNICODE(woperation) : 0,
6574 PyUnicode_AS_UNICODE(unipath),
6575 NULL, NULL, SW_SHOWNORMAL);
6576 Py_END_ALLOW_THREADS
6577
6578 Py_XDECREF(woperation);
6579 if (rc <= (HINSTANCE)32) {
6580 PyObject *errval = win32_error_unicode("startfile",
6581 PyUnicode_AS_UNICODE(unipath));
6582 return errval;
6583 }
6584 Py_INCREF(Py_None);
6585 return Py_None;
6586 }
6587#endif
6588
6589normal:
Georg Brandlf4f44152006-02-18 22:29:33 +00006590 if (!PyArg_ParseTuple(args, "et|s:startfile",
6591 Py_FileSystemDefaultEncoding, &filepath,
6592 &operation))
Tim Petersf58a7aa2000-09-22 10:05:54 +00006593 return NULL;
6594 Py_BEGIN_ALLOW_THREADS
Georg Brandlf4f44152006-02-18 22:29:33 +00006595 rc = ShellExecute((HWND)0, operation, filepath,
6596 NULL, NULL, SW_SHOWNORMAL);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006597 Py_END_ALLOW_THREADS
Georg Brandle9f8ec92005-09-25 06:16:40 +00006598 if (rc <= (HINSTANCE)32) {
6599 PyObject *errval = win32_error("startfile", filepath);
6600 PyMem_Free(filepath);
6601 return errval;
6602 }
6603 PyMem_Free(filepath);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006604 Py_INCREF(Py_None);
6605 return Py_None;
6606}
6607#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006608
Martin v. Löwis438b5342002-12-27 10:16:42 +00006609#ifdef HAVE_GETLOADAVG
6610PyDoc_STRVAR(posix_getloadavg__doc__,
6611"getloadavg() -> (float, float, float)\n\n\
6612Return the number of processes in the system run queue averaged over\n\
6613the last 1, 5, and 15 minutes or raises OSError if the load average\n\
6614was unobtainable");
6615
6616static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006617posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00006618{
6619 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00006620 if (getloadavg(loadavg, 3)!=3) {
6621 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
6622 return NULL;
6623 } else
6624 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
6625}
6626#endif
6627
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006628#ifdef MS_WINDOWS
6629
6630PyDoc_STRVAR(win32_urandom__doc__,
6631"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00006632Return n random bytes suitable for cryptographic use.");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006633
6634typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
6635 LPCSTR pszContainer, LPCSTR pszProvider, DWORD dwProvType,\
6636 DWORD dwFlags );
6637typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV hProv, DWORD dwLen,\
6638 BYTE *pbBuffer );
6639
6640static CRYPTGENRANDOM pCryptGenRandom = NULL;
Thomas Wouters89d996e2007-09-08 17:39:28 +00006641/* This handle is never explicitly released. Instead, the operating
6642 system will release it when the process terminates. */
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006643static HCRYPTPROV hCryptProv = 0;
6644
Tim Peters4ad82172004-08-30 17:02:04 +00006645static PyObject*
6646win32_urandom(PyObject *self, PyObject *args)
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006647{
Tim Petersd3115382004-08-30 17:36:46 +00006648 int howMany;
6649 PyObject* result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006650
Tim Peters4ad82172004-08-30 17:02:04 +00006651 /* Read arguments */
Tim Peters9b279a82004-08-30 17:10:53 +00006652 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
Tim Peters4ad82172004-08-30 17:02:04 +00006653 return NULL;
Tim Peters51eba612004-08-30 17:08:02 +00006654 if (howMany < 0)
6655 return PyErr_Format(PyExc_ValueError,
6656 "negative argument not allowed");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006657
Tim Peters4ad82172004-08-30 17:02:04 +00006658 if (hCryptProv == 0) {
6659 HINSTANCE hAdvAPI32 = NULL;
6660 CRYPTACQUIRECONTEXTA pCryptAcquireContext = NULL;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006661
Tim Peters4ad82172004-08-30 17:02:04 +00006662 /* Obtain handle to the DLL containing CryptoAPI
6663 This should not fail */
6664 hAdvAPI32 = GetModuleHandle("advapi32.dll");
6665 if(hAdvAPI32 == NULL)
6666 return win32_error("GetModuleHandle", NULL);
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006667
Tim Peters4ad82172004-08-30 17:02:04 +00006668 /* Obtain pointers to the CryptoAPI functions
6669 This will fail on some early versions of Win95 */
6670 pCryptAcquireContext = (CRYPTACQUIRECONTEXTA)GetProcAddress(
6671 hAdvAPI32,
6672 "CryptAcquireContextA");
6673 if (pCryptAcquireContext == NULL)
6674 return PyErr_Format(PyExc_NotImplementedError,
6675 "CryptAcquireContextA not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006676
Tim Peters4ad82172004-08-30 17:02:04 +00006677 pCryptGenRandom = (CRYPTGENRANDOM)GetProcAddress(
6678 hAdvAPI32, "CryptGenRandom");
Thomas Wouters89f507f2006-12-13 04:49:30 +00006679 if (pCryptGenRandom == NULL)
Tim Peters4ad82172004-08-30 17:02:04 +00006680 return PyErr_Format(PyExc_NotImplementedError,
6681 "CryptGenRandom not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006682
Tim Peters4ad82172004-08-30 17:02:04 +00006683 /* Acquire context */
6684 if (! pCryptAcquireContext(&hCryptProv, NULL, NULL,
6685 PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
6686 return win32_error("CryptAcquireContext", NULL);
6687 }
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006688
Tim Peters4ad82172004-08-30 17:02:04 +00006689 /* Allocate bytes */
Christian Heimes72b710a2008-05-26 13:28:38 +00006690 result = PyBytes_FromStringAndSize(NULL, howMany);
Tim Petersd3115382004-08-30 17:36:46 +00006691 if (result != NULL) {
6692 /* Get random data */
Amaury Forgeot d'Arca05ada32008-07-21 21:13:14 +00006693 memset(PyBytes_AS_STRING(result), 0, howMany); /* zero seed */
Tim Petersd3115382004-08-30 17:36:46 +00006694 if (! pCryptGenRandom(hCryptProv, howMany, (unsigned char*)
Christian Heimes72b710a2008-05-26 13:28:38 +00006695 PyBytes_AS_STRING(result))) {
Tim Petersd3115382004-08-30 17:36:46 +00006696 Py_DECREF(result);
6697 return win32_error("CryptGenRandom", NULL);
6698 }
Tim Peters4ad82172004-08-30 17:02:04 +00006699 }
Tim Petersd3115382004-08-30 17:36:46 +00006700 return result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006701}
6702#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00006703
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00006704PyDoc_STRVAR(device_encoding__doc__,
6705"device_encoding(fd) -> str\n\n\
6706Return a string describing the encoding of the device\n\
6707if the output is a terminal; else return None.");
6708
6709static PyObject *
6710device_encoding(PyObject *self, PyObject *args)
6711{
6712 int fd;
6713 if (!PyArg_ParseTuple(args, "i:device_encoding", &fd))
6714 return NULL;
6715 if (!isatty(fd)) {
6716 Py_INCREF(Py_None);
6717 return Py_None;
6718 }
6719#if defined(MS_WINDOWS) || defined(MS_WIN64)
6720 if (fd == 0) {
6721 char buf[100];
6722 sprintf(buf, "cp%d", GetConsoleCP());
6723 return PyUnicode_FromString(buf);
6724 }
6725 if (fd == 1 || fd == 2) {
6726 char buf[100];
6727 sprintf(buf, "cp%d", GetConsoleOutputCP());
6728 return PyUnicode_FromString(buf);
6729 }
6730#elif defined(CODESET)
6731 {
6732 char *codeset = nl_langinfo(CODESET);
6733 if (codeset)
6734 return PyUnicode_FromString(codeset);
6735 }
6736#endif
6737 Py_INCREF(Py_None);
6738 return Py_None;
6739}
6740
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006741#ifdef __VMS
6742/* Use openssl random routine */
6743#include <openssl/rand.h>
6744PyDoc_STRVAR(vms_urandom__doc__,
6745"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00006746Return n random bytes suitable for cryptographic use.");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006747
6748static PyObject*
6749vms_urandom(PyObject *self, PyObject *args)
6750{
6751 int howMany;
6752 PyObject* result;
6753
6754 /* Read arguments */
6755 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
6756 return NULL;
6757 if (howMany < 0)
6758 return PyErr_Format(PyExc_ValueError,
6759 "negative argument not allowed");
6760
6761 /* Allocate bytes */
Christian Heimes72b710a2008-05-26 13:28:38 +00006762 result = PyBytes_FromStringAndSize(NULL, howMany);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006763 if (result != NULL) {
6764 /* Get random data */
6765 if (RAND_pseudo_bytes((unsigned char*)
Christian Heimes72b710a2008-05-26 13:28:38 +00006766 PyBytes_AS_STRING(result),
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006767 howMany) < 0) {
6768 Py_DECREF(result);
6769 return PyErr_Format(PyExc_ValueError,
6770 "RAND_pseudo_bytes");
6771 }
6772 }
6773 return result;
6774}
6775#endif
6776
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006777static PyMethodDef posix_methods[] = {
6778 {"access", posix_access, METH_VARARGS, posix_access__doc__},
6779#ifdef HAVE_TTYNAME
6780 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
6781#endif
6782 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +00006783#ifdef HAVE_CHFLAGS
6784 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
6785#endif /* HAVE_CHFLAGS */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006786 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +00006787#ifdef HAVE_FCHMOD
6788 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
6789#endif /* HAVE_FCHMOD */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006790#ifdef HAVE_CHOWN
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006791 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006792#endif /* HAVE_CHOWN */
Christian Heimes4e30a842007-11-30 22:12:06 +00006793#ifdef HAVE_LCHMOD
6794 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
6795#endif /* HAVE_LCHMOD */
6796#ifdef HAVE_FCHOWN
6797 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
6798#endif /* HAVE_FCHOWN */
Thomas Wouterscf297e42007-02-23 15:07:44 +00006799#ifdef HAVE_LCHFLAGS
6800 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
6801#endif /* HAVE_LCHFLAGS */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00006802#ifdef HAVE_LCHOWN
6803 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
6804#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +00006805#ifdef HAVE_CHROOT
6806 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
6807#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006808#ifdef HAVE_CTERMID
Neal Norwitze241ce82003-02-17 18:17:05 +00006809 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006810#endif
Guido van Rossum36bc6801995-06-14 22:54:23 +00006811#ifdef HAVE_GETCWD
Neal Norwitze241ce82003-02-17 18:17:05 +00006812 {"getcwd", posix_getcwd, METH_NOARGS, posix_getcwd__doc__},
Neal Norwitze241ce82003-02-17 18:17:05 +00006813 {"getcwdu", posix_getcwdu, METH_NOARGS, posix_getcwdu__doc__},
Guido van Rossum36bc6801995-06-14 22:54:23 +00006814#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00006815#ifdef HAVE_LINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006816 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006817#endif /* HAVE_LINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006818 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
6819 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
6820 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006821#ifdef HAVE_NICE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006822 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006823#endif /* HAVE_NICE */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006824#ifdef HAVE_READLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006825 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006826#endif /* HAVE_READLINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006827 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
6828 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
6829 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00006830 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006831#ifdef HAVE_SYMLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006832 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006833#endif /* HAVE_SYMLINK */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006834#ifdef HAVE_SYSTEM
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006835 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006836#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006837 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006838#ifdef HAVE_UNAME
Neal Norwitze241ce82003-02-17 18:17:05 +00006839 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006840#endif /* HAVE_UNAME */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006841 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
6842 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
6843 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006844#ifdef HAVE_TIMES
Neal Norwitze241ce82003-02-17 18:17:05 +00006845 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006846#endif /* HAVE_TIMES */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006847 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006848#ifdef HAVE_EXECV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006849 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
6850 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006851#endif /* HAVE_EXECV */
Guido van Rossuma1065681999-01-25 23:20:23 +00006852#ifdef HAVE_SPAWNV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006853 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
6854 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00006855#if defined(PYOS_OS2)
6856 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
6857 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
6858#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00006859#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +00006860#ifdef HAVE_FORK1
Neal Norwitze241ce82003-02-17 18:17:05 +00006861 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +00006862#endif /* HAVE_FORK1 */
Guido van Rossumad0ee831995-03-01 10:34:45 +00006863#ifdef HAVE_FORK
Neal Norwitze241ce82003-02-17 18:17:05 +00006864 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006865#endif /* HAVE_FORK */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00006866#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Neal Norwitze241ce82003-02-17 18:17:05 +00006867 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +00006868#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00006869#ifdef HAVE_FORKPTY
Neal Norwitze241ce82003-02-17 18:17:05 +00006870 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +00006871#endif /* HAVE_FORKPTY */
Guido van Rossumad0ee831995-03-01 10:34:45 +00006872#ifdef HAVE_GETEGID
Neal Norwitze241ce82003-02-17 18:17:05 +00006873 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006874#endif /* HAVE_GETEGID */
6875#ifdef HAVE_GETEUID
Neal Norwitze241ce82003-02-17 18:17:05 +00006876 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006877#endif /* HAVE_GETEUID */
6878#ifdef HAVE_GETGID
Neal Norwitze241ce82003-02-17 18:17:05 +00006879 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006880#endif /* HAVE_GETGID */
Fred Drakec9680921999-12-13 16:37:25 +00006881#ifdef HAVE_GETGROUPS
Neal Norwitze241ce82003-02-17 18:17:05 +00006882 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00006883#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00006884 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006885#ifdef HAVE_GETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00006886 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006887#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00006888#ifdef HAVE_GETPPID
Neal Norwitze241ce82003-02-17 18:17:05 +00006889 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006890#endif /* HAVE_GETPPID */
6891#ifdef HAVE_GETUID
Neal Norwitze241ce82003-02-17 18:17:05 +00006892 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006893#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +00006894#ifdef HAVE_GETLOGIN
Neal Norwitze241ce82003-02-17 18:17:05 +00006895 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +00006896#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +00006897#ifdef HAVE_KILL
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006898 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006899#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00006900#ifdef HAVE_KILLPG
6901 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
6902#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +00006903#ifdef HAVE_PLOCK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006904 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +00006905#endif /* HAVE_PLOCK */
Thomas Heller8b7a9572007-08-31 06:44:36 +00006906#ifdef MS_WINDOWS
6907 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
6908#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00006909#ifdef HAVE_SETUID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006910 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006911#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006912#ifdef HAVE_SETEUID
6913 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
6914#endif /* HAVE_SETEUID */
6915#ifdef HAVE_SETEGID
6916 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
6917#endif /* HAVE_SETEGID */
6918#ifdef HAVE_SETREUID
6919 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
6920#endif /* HAVE_SETREUID */
6921#ifdef HAVE_SETREGID
6922 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
6923#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006924#ifdef HAVE_SETGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006925 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006926#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006927#ifdef HAVE_SETGROUPS
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00006928 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006929#endif /* HAVE_SETGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +00006930#ifdef HAVE_GETPGID
6931 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
6932#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006933#ifdef HAVE_SETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00006934 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006935#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00006936#ifdef HAVE_WAIT
Neal Norwitze241ce82003-02-17 18:17:05 +00006937 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006938#endif /* HAVE_WAIT */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006939#ifdef HAVE_WAIT3
6940 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
6941#endif /* HAVE_WAIT3 */
6942#ifdef HAVE_WAIT4
6943 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
6944#endif /* HAVE_WAIT4 */
Tim Petersab034fa2002-02-01 11:27:43 +00006945#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006946 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006947#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00006948#ifdef HAVE_GETSID
6949 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
6950#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006951#ifdef HAVE_SETSID
Neal Norwitze241ce82003-02-17 18:17:05 +00006952 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006953#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006954#ifdef HAVE_SETPGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006955 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006956#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006957#ifdef HAVE_TCGETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006958 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006959#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006960#ifdef HAVE_TCSETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006961 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006962#endif /* HAVE_TCSETPGRP */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006963 {"open", posix_open, METH_VARARGS, posix_open__doc__},
6964 {"close", posix_close, METH_VARARGS, posix_close__doc__},
Christian Heimesfdab48e2008-01-20 09:06:41 +00006965 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00006966 {"device_encoding", device_encoding, METH_VARARGS, device_encoding__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006967 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
6968 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
6969 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
6970 {"read", posix_read, METH_VARARGS, posix_read__doc__},
6971 {"write", posix_write, METH_VARARGS, posix_write__doc__},
6972 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
Skip Montanaro1517d842000-07-19 14:34:14 +00006973 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006974#ifdef HAVE_PIPE
Neal Norwitze241ce82003-02-17 18:17:05 +00006975 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006976#endif
6977#ifdef HAVE_MKFIFO
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006978 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006979#endif
Neal Norwitz11690112002-07-30 01:08:28 +00006980#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006981 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
6982#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006983#ifdef HAVE_DEVICE_MACROS
6984 {"major", posix_major, METH_VARARGS, posix_major__doc__},
6985 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
6986 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
6987#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006988#ifdef HAVE_FTRUNCATE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006989 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006990#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006991#ifdef HAVE_PUTENV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006992 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006993#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00006994#ifdef HAVE_UNSETENV
6995 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
6996#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006997 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +00006998#ifdef HAVE_FCHDIR
6999 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
7000#endif
Guido van Rossum21142a01999-01-08 21:05:37 +00007001#ifdef HAVE_FSYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007002 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007003#endif
7004#ifdef HAVE_FDATASYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007005 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007006#endif
Guido van Rossumc9641791998-08-04 15:26:23 +00007007#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +00007008#ifdef WCOREDUMP
7009 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
7010#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00007011#ifdef WIFCONTINUED
7012 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
7013#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +00007014#ifdef WIFSTOPPED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007015 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007016#endif /* WIFSTOPPED */
7017#ifdef WIFSIGNALED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007018 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007019#endif /* WIFSIGNALED */
7020#ifdef WIFEXITED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007021 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007022#endif /* WIFEXITED */
7023#ifdef WEXITSTATUS
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007024 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007025#endif /* WEXITSTATUS */
7026#ifdef WTERMSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007027 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007028#endif /* WTERMSIG */
7029#ifdef WSTOPSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007030 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007031#endif /* WSTOPSIG */
7032#endif /* HAVE_SYS_WAIT_H */
Thomas Wouters477c8d52006-05-27 19:21:47 +00007033#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007034 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007035#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00007036#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007037 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007038#endif
Fred Drakec9680921999-12-13 16:37:25 +00007039#ifdef HAVE_CONFSTR
7040 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
7041#endif
7042#ifdef HAVE_SYSCONF
7043 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
7044#endif
7045#ifdef HAVE_FPATHCONF
7046 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
7047#endif
7048#ifdef HAVE_PATHCONF
7049 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
7050#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00007051 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007052#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00007053 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
7054#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00007055#ifdef HAVE_GETLOADAVG
Neal Norwitze241ce82003-02-17 18:17:05 +00007056 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +00007057#endif
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00007058 #ifdef MS_WINDOWS
7059 {"urandom", win32_urandom, METH_VARARGS, win32_urandom__doc__},
7060 #endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007061 #ifdef __VMS
7062 {"urandom", vms_urandom, METH_VARARGS, vms_urandom__doc__},
7063 #endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007064 {NULL, NULL} /* Sentinel */
7065};
7066
7067
Barry Warsaw4a342091996-12-19 23:50:02 +00007068static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00007069ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +00007070{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007071 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +00007072}
7073
Guido van Rossumd48f2521997-12-05 22:19:34 +00007074#if defined(PYOS_OS2)
7075/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007076static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007077{
7078 APIRET rc;
7079 ULONG values[QSV_MAX+1];
7080 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +00007081 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +00007082
7083 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +00007084 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007085 Py_END_ALLOW_THREADS
7086
7087 if (rc != NO_ERROR) {
7088 os2_error(rc);
7089 return -1;
7090 }
7091
Fred Drake4d1e64b2002-04-15 19:40:07 +00007092 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
7093 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
7094 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
7095 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
7096 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
7097 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
7098 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007099
7100 switch (values[QSV_VERSION_MINOR]) {
7101 case 0: ver = "2.00"; break;
7102 case 10: ver = "2.10"; break;
7103 case 11: ver = "2.11"; break;
7104 case 30: ver = "3.00"; break;
7105 case 40: ver = "4.00"; break;
7106 case 50: ver = "5.00"; break;
7107 default:
Tim Peters885d4572001-11-28 20:27:42 +00007108 PyOS_snprintf(tmp, sizeof(tmp),
7109 "%d-%d", values[QSV_VERSION_MAJOR],
7110 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007111 ver = &tmp[0];
7112 }
7113
7114 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007115 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007116 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007117
7118 /* Add Indicator of Which Drive was Used to Boot the System */
7119 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
7120 tmp[1] = ':';
7121 tmp[2] = '\0';
7122
Fred Drake4d1e64b2002-04-15 19:40:07 +00007123 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007124}
7125#endif
7126
Barry Warsaw4a342091996-12-19 23:50:02 +00007127static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007128all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +00007129{
Guido van Rossum94f6f721999-01-06 18:42:14 +00007130#ifdef F_OK
7131 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007132#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007133#ifdef R_OK
7134 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007135#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007136#ifdef W_OK
7137 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007138#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007139#ifdef X_OK
7140 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007141#endif
Fred Drakec9680921999-12-13 16:37:25 +00007142#ifdef NGROUPS_MAX
7143 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
7144#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007145#ifdef TMP_MAX
7146 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
7147#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007148#ifdef WCONTINUED
7149 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
7150#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007151#ifdef WNOHANG
7152 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007153#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007154#ifdef WUNTRACED
7155 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
7156#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007157#ifdef O_RDONLY
7158 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
7159#endif
7160#ifdef O_WRONLY
7161 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
7162#endif
7163#ifdef O_RDWR
7164 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
7165#endif
7166#ifdef O_NDELAY
7167 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
7168#endif
7169#ifdef O_NONBLOCK
7170 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
7171#endif
7172#ifdef O_APPEND
7173 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
7174#endif
7175#ifdef O_DSYNC
7176 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
7177#endif
7178#ifdef O_RSYNC
7179 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
7180#endif
7181#ifdef O_SYNC
7182 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
7183#endif
7184#ifdef O_NOCTTY
7185 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
7186#endif
7187#ifdef O_CREAT
7188 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
7189#endif
7190#ifdef O_EXCL
7191 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
7192#endif
7193#ifdef O_TRUNC
7194 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
7195#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +00007196#ifdef O_BINARY
7197 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
7198#endif
7199#ifdef O_TEXT
7200 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
7201#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007202#ifdef O_LARGEFILE
7203 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
7204#endif
Skip Montanaro5ff14922005-05-16 02:42:22 +00007205#ifdef O_SHLOCK
7206 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
7207#endif
7208#ifdef O_EXLOCK
7209 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
7210#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007211
Tim Peters5aa91602002-01-30 05:46:57 +00007212/* MS Windows */
7213#ifdef O_NOINHERIT
7214 /* Don't inherit in child processes. */
7215 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
7216#endif
7217#ifdef _O_SHORT_LIVED
7218 /* Optimize for short life (keep in memory). */
7219 /* MS forgot to define this one with a non-underscore form too. */
7220 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
7221#endif
7222#ifdef O_TEMPORARY
7223 /* Automatically delete when last handle is closed. */
7224 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
7225#endif
7226#ifdef O_RANDOM
7227 /* Optimize for random access. */
7228 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
7229#endif
7230#ifdef O_SEQUENTIAL
7231 /* Optimize for sequential access. */
7232 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
7233#endif
7234
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007235/* GNU extensions. */
Alexandre Vassalottibee32532008-05-16 18:15:12 +00007236#ifdef O_ASYNC
7237 /* Send a SIGIO signal whenever input or output
7238 becomes available on file descriptor */
7239 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
7240#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007241#ifdef O_DIRECT
7242 /* Direct disk access. */
7243 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
7244#endif
7245#ifdef O_DIRECTORY
7246 /* Must be a directory. */
7247 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
7248#endif
7249#ifdef O_NOFOLLOW
7250 /* Do not follow links. */
7251 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
7252#endif
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +00007253#ifdef O_NOATIME
7254 /* Do not update the access time. */
7255 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
7256#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00007257
Barry Warsaw5676bd12003-01-07 20:57:09 +00007258 /* These come from sysexits.h */
7259#ifdef EX_OK
7260 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007261#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007262#ifdef EX_USAGE
7263 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007264#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007265#ifdef EX_DATAERR
7266 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007267#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007268#ifdef EX_NOINPUT
7269 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007270#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007271#ifdef EX_NOUSER
7272 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007273#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007274#ifdef EX_NOHOST
7275 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007276#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007277#ifdef EX_UNAVAILABLE
7278 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007279#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007280#ifdef EX_SOFTWARE
7281 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007282#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007283#ifdef EX_OSERR
7284 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007285#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007286#ifdef EX_OSFILE
7287 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007288#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007289#ifdef EX_CANTCREAT
7290 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007291#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007292#ifdef EX_IOERR
7293 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007294#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007295#ifdef EX_TEMPFAIL
7296 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007297#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007298#ifdef EX_PROTOCOL
7299 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007300#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007301#ifdef EX_NOPERM
7302 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007303#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007304#ifdef EX_CONFIG
7305 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007306#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007307#ifdef EX_NOTFOUND
7308 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007309#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007310
Guido van Rossum246bc171999-02-01 23:54:31 +00007311#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007312#if defined(PYOS_OS2) && defined(PYCC_GCC)
7313 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
7314 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
7315 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
7316 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
7317 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
7318 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
7319 if (ins(d, "P_PM", (long)P_PM)) return -1;
7320 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
7321 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
7322 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
7323 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
7324 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
7325 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
7326 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
7327 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
7328 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
7329 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
7330 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
7331 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
7332 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
7333#else
Guido van Rossum7d385291999-02-16 19:38:04 +00007334 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
7335 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
7336 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
7337 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
7338 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +00007339#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007340#endif
Guido van Rossum246bc171999-02-01 23:54:31 +00007341
Guido van Rossumd48f2521997-12-05 22:19:34 +00007342#if defined(PYOS_OS2)
7343 if (insertvalues(d)) return -1;
7344#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007345 return 0;
7346}
7347
7348
Tim Peters5aa91602002-01-30 05:46:57 +00007349#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007350#define INITFUNC PyInit_nt
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007351#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007352
7353#elif defined(PYOS_OS2)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007354#define INITFUNC PyInit_os2
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007355#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007356
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007357#else
Martin v. Löwis1a214512008-06-11 05:26:20 +00007358#define INITFUNC PyInit_posix
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007359#define MODNAME "posix"
7360#endif
7361
Martin v. Löwis1a214512008-06-11 05:26:20 +00007362static struct PyModuleDef posixmodule = {
7363 PyModuleDef_HEAD_INIT,
7364 MODNAME,
7365 posix__doc__,
7366 -1,
7367 posix_methods,
7368 NULL,
7369 NULL,
7370 NULL,
7371 NULL
7372};
7373
7374
Mark Hammondfe51c6d2002-08-02 02:27:13 +00007375PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007376INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +00007377{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007378 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +00007379
Martin v. Löwis1a214512008-06-11 05:26:20 +00007380 m = PyModule_Create(&posixmodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00007381 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007382 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007383
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007384 /* Initialize environ dictionary */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007385 v = convertenviron();
Fred Drake4d1e64b2002-04-15 19:40:07 +00007386 Py_XINCREF(v);
7387 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007388 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00007389 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +00007390
Fred Drake4d1e64b2002-04-15 19:40:07 +00007391 if (all_ins(m))
Martin v. Löwis1a214512008-06-11 05:26:20 +00007392 return NULL;
Barry Warsaw4a342091996-12-19 23:50:02 +00007393
Fred Drake4d1e64b2002-04-15 19:40:07 +00007394 if (setup_confname_tables(m))
Martin v. Löwis1a214512008-06-11 05:26:20 +00007395 return NULL;
Fred Drakebec628d1999-12-15 18:31:10 +00007396
Fred Drake4d1e64b2002-04-15 19:40:07 +00007397 Py_INCREF(PyExc_OSError);
7398 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +00007399
Guido van Rossumb3d39562000-01-31 18:41:26 +00007400#ifdef HAVE_PUTENV
Neil Schemenauer19030a02001-01-16 04:27:47 +00007401 if (posix_putenv_garbage == NULL)
7402 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +00007403#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007404
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007405 if (!initialized) {
7406 stat_result_desc.name = MODNAME ".stat_result";
7407 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
7408 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
7409 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
7410 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
7411 structseq_new = StatResultType.tp_new;
7412 StatResultType.tp_new = statresult_new;
7413
7414 statvfs_result_desc.name = MODNAME ".statvfs_result";
7415 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
7416 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00007417 Py_INCREF((PyObject*) &StatResultType);
7418 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
Fred Drake4d1e64b2002-04-15 19:40:07 +00007419 Py_INCREF((PyObject*) &StatVFSResultType);
7420 PyModule_AddObject(m, "statvfs_result",
7421 (PyObject*) &StatVFSResultType);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007422 initialized = 1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007423
7424#ifdef __APPLE__
7425 /*
7426 * Step 2 of weak-linking support on Mac OS X.
7427 *
7428 * The code below removes functions that are not available on the
7429 * currently active platform.
7430 *
7431 * This block allow one to use a python binary that was build on
7432 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
7433 * OSX 10.4.
7434 */
7435#ifdef HAVE_FSTATVFS
7436 if (fstatvfs == NULL) {
7437 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007438 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007439 }
7440 }
7441#endif /* HAVE_FSTATVFS */
7442
7443#ifdef HAVE_STATVFS
7444 if (statvfs == NULL) {
7445 if (PyObject_DelAttrString(m, "statvfs") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007446 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007447 }
7448 }
7449#endif /* HAVE_STATVFS */
7450
7451# ifdef HAVE_LCHOWN
7452 if (lchown == NULL) {
7453 if (PyObject_DelAttrString(m, "lchown") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007454 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007455 }
7456 }
7457#endif /* HAVE_LCHOWN */
7458
7459
7460#endif /* __APPLE__ */
Martin v. Löwis1a214512008-06-11 05:26:20 +00007461 return m;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007462
Guido van Rossumb6775db1994-08-01 11:34:53 +00007463}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007464
7465#ifdef __cplusplus
7466}
7467#endif