blob: b499b47817737a8608959843d1791cd892e0894d [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 (;;) {
Georg Brandl3dbca812008-07-23 16:10:53 +00002353 errno = 0;
Georg Brandl622927b2006-03-07 12:48:03 +00002354 Py_BEGIN_ALLOW_THREADS
2355 ep = readdir(dirp);
2356 Py_END_ALLOW_THREADS
Georg Brandl3dbca812008-07-23 16:10:53 +00002357 if (ep == NULL) {
2358 if (errno == 0) {
2359 break;
2360 } else {
2361 closedir(dirp);
2362 Py_DECREF(d);
2363 return posix_error_with_allocated_filename(name);
2364 }
2365 }
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002366 if (ep->d_name[0] == '.' &&
2367 (NAMLEN(ep) == 1 ||
Guido van Rossuma376cc51996-12-05 23:43:35 +00002368 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
Guido van Rossum24f42ac1995-07-18 18:16:52 +00002369 continue;
Christian Heimes72b710a2008-05-26 13:28:38 +00002370 v = PyBytes_FromStringAndSize(ep->d_name, NAMLEN(ep));
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002371 if (v == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00002372 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002373 d = NULL;
2374 break;
2375 }
Just van Rossum96b1c902003-03-03 17:32:15 +00002376 if (arg_is_unicode) {
Just van Rossum46c97842003-02-25 21:42:15 +00002377 PyObject *w;
2378
2379 w = PyUnicode_FromEncodedObject(v,
Tim Peters11b23062003-04-23 02:39:17 +00002380 Py_FileSystemDefaultEncoding,
Just van Rossum46c97842003-02-25 21:42:15 +00002381 "strict");
Just van Rossum6a421832003-03-04 19:30:44 +00002382 if (w != NULL) {
2383 Py_DECREF(v);
2384 v = w;
2385 }
2386 else {
2387 /* fall back to the original byte string, as
2388 discussed in patch #683592 */
2389 PyErr_Clear();
Just van Rossum46c97842003-02-25 21:42:15 +00002390 }
Just van Rossum46c97842003-02-25 21:42:15 +00002391 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002392 if (PyList_Append(d, v) != 0) {
2393 Py_DECREF(v);
2394 Py_DECREF(d);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002395 d = NULL;
2396 break;
2397 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002398 Py_DECREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002399 }
2400 closedir(dirp);
Just van Rossum2fe07fd2003-03-03 19:07:13 +00002401 PyMem_Free(name);
Guido van Rossum0ee42cd1991-04-08 21:01:03 +00002402
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002403 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002404
Tim Peters0bb44a42000-09-15 07:44:49 +00002405#endif /* which OS */
2406} /* end of posix_listdir */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002407
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002408#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00002409/* A helper function for abspath on win32 */
2410static PyObject *
2411posix__getfullpathname(PyObject *self, PyObject *args)
2412{
2413 /* assume encoded strings wont more than double no of chars */
2414 char inbuf[MAX_PATH*2];
2415 char *inbufp = inbuf;
Tim Peters67d70eb2006-03-01 04:35:45 +00002416 Py_ssize_t insize = sizeof(inbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00002417 char outbuf[MAX_PATH*2];
2418 char *temp;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002419#ifdef Py_WIN_WIDE_FILENAMES
2420 if (unicode_file_names()) {
2421 PyUnicodeObject *po;
2422 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po)) {
2423 Py_UNICODE woutbuf[MAX_PATH*2];
2424 Py_UNICODE *wtemp;
Tim Peters11b23062003-04-23 02:39:17 +00002425 if (!GetFullPathNameW(PyUnicode_AS_UNICODE(po),
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002426 sizeof(woutbuf)/sizeof(woutbuf[0]),
2427 woutbuf, &wtemp))
2428 return win32_error("GetFullPathName", "");
2429 return PyUnicode_FromUnicode(woutbuf, wcslen(woutbuf));
2430 }
2431 /* Drop the argument parsing error as narrow strings
2432 are also valid. */
2433 PyErr_Clear();
2434 }
2435#endif
Tim Peters5aa91602002-01-30 05:46:57 +00002436 if (!PyArg_ParseTuple (args, "et#:_getfullpathname",
2437 Py_FileSystemDefaultEncoding, &inbufp,
Mark Hammondef8b6542001-05-13 08:04:26 +00002438 &insize))
2439 return NULL;
2440 if (!GetFullPathName(inbuf, sizeof(outbuf)/sizeof(outbuf[0]),
2441 outbuf, &temp))
2442 return win32_error("GetFullPathName", inbuf);
Martin v. Löwis969297f2004-06-15 18:49:58 +00002443 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
2444 return PyUnicode_Decode(outbuf, strlen(outbuf),
2445 Py_FileSystemDefaultEncoding, NULL);
2446 }
Christian Heimes72b710a2008-05-26 13:28:38 +00002447 return PyBytes_FromString(outbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00002448} /* end of posix__getfullpathname */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002449#endif /* MS_WINDOWS */
Mark Hammondef8b6542001-05-13 08:04:26 +00002450
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002451PyDoc_STRVAR(posix_mkdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002452"mkdir(path [, mode=0777])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002453Create a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002454
Barry Warsaw53699e91996-12-10 23:23:01 +00002455static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002456posix_mkdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002457{
Guido van Rossumb0824db1996-02-25 04:50:32 +00002458 int res;
Mark Hammondef8b6542001-05-13 08:04:26 +00002459 char *path = NULL;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002460 int mode = 0777;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002461
2462#ifdef Py_WIN_WIDE_FILENAMES
2463 if (unicode_file_names()) {
2464 PyUnicodeObject *po;
Mark Hammond05107b62003-02-19 04:08:27 +00002465 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode)) {
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002466 Py_BEGIN_ALLOW_THREADS
2467 /* PyUnicode_AS_UNICODE OK without thread lock as
2468 it is a simple dereference. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002469 res = CreateDirectoryW(PyUnicode_AS_UNICODE(po), NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002470 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002471 if (!res)
2472 return win32_error_unicode("mkdir", PyUnicode_AS_UNICODE(po));
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002473 Py_INCREF(Py_None);
2474 return Py_None;
2475 }
2476 /* Drop the argument parsing error as narrow strings
2477 are also valid. */
2478 PyErr_Clear();
2479 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002480 if (!PyArg_ParseTuple(args, "et|i:mkdir",
2481 Py_FileSystemDefaultEncoding, &path, &mode))
2482 return NULL;
2483 Py_BEGIN_ALLOW_THREADS
2484 /* PyUnicode_AS_UNICODE OK without thread lock as
2485 it is a simple dereference. */
2486 res = CreateDirectoryA(path, NULL);
2487 Py_END_ALLOW_THREADS
2488 if (!res) {
2489 win32_error("mkdir", path);
2490 PyMem_Free(path);
2491 return NULL;
2492 }
2493 PyMem_Free(path);
2494 Py_INCREF(Py_None);
2495 return Py_None;
2496#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002497
Tim Peters5aa91602002-01-30 05:46:57 +00002498 if (!PyArg_ParseTuple(args, "et|i:mkdir",
Mark Hammondef8b6542001-05-13 08:04:26 +00002499 Py_FileSystemDefaultEncoding, &path, &mode))
Guido van Rossumb0824db1996-02-25 04:50:32 +00002500 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002501 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002502#if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002503 res = mkdir(path);
2504#else
Guido van Rossumb0824db1996-02-25 04:50:32 +00002505 res = mkdir(path, mode);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002506#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002507 Py_END_ALLOW_THREADS
Guido van Rossumb0824db1996-02-25 04:50:32 +00002508 if (res < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00002509 return posix_error_with_allocated_filename(path);
2510 PyMem_Free(path);
Barry Warsaw53699e91996-12-10 23:23:01 +00002511 Py_INCREF(Py_None);
2512 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002513#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002514}
2515
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002516
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002517/* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
2518#if defined(HAVE_SYS_RESOURCE_H)
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002519#include <sys/resource.h>
2520#endif
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002521
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002522
2523#ifdef HAVE_NICE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002524PyDoc_STRVAR(posix_nice__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002525"nice(inc) -> new_priority\n\n\
2526Decrease the priority of process by inc and return the new priority.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002527
Barry Warsaw53699e91996-12-10 23:23:01 +00002528static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002529posix_nice(PyObject *self, PyObject *args)
Guido van Rossum775f4da1993-01-09 17:18:52 +00002530{
2531 int increment, value;
2532
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002533 if (!PyArg_ParseTuple(args, "i:nice", &increment))
Guido van Rossum775f4da1993-01-09 17:18:52 +00002534 return NULL;
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002535
2536 /* There are two flavours of 'nice': one that returns the new
2537 priority (as required by almost all standards out there) and the
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002538 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
2539 the use of getpriority() to get the new priority.
Tim Peters5aa91602002-01-30 05:46:57 +00002540
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002541 If we are of the nice family that returns the new priority, we
2542 need to clear errno before the call, and check if errno is filled
2543 before calling posix_error() on a returnvalue of -1, because the
2544 -1 may be the actual new priority! */
2545
2546 errno = 0;
Guido van Rossum775f4da1993-01-09 17:18:52 +00002547 value = nice(increment);
Thomas Wouterse38b2f12001-07-11 22:35:31 +00002548#if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00002549 if (value == 0)
2550 value = getpriority(PRIO_PROCESS, 0);
2551#endif
2552 if (value == -1 && errno != 0)
2553 /* either nice() or getpriority() returned an error */
Guido van Rossum775f4da1993-01-09 17:18:52 +00002554 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00002555 return PyLong_FromLong((long) value);
Guido van Rossum775f4da1993-01-09 17:18:52 +00002556}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002557#endif /* HAVE_NICE */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002558
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002559PyDoc_STRVAR(posix_rename__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002560"rename(old, new)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002561Rename a file or directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002562
Barry Warsaw53699e91996-12-10 23:23:01 +00002563static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002564posix_rename(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002565{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002566#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002567 PyObject *o1, *o2;
2568 char *p1, *p2;
2569 BOOL result;
2570 if (unicode_file_names()) {
2571 if (!PyArg_ParseTuple(args, "O&O&:rename",
2572 convert_to_unicode, &o1,
2573 convert_to_unicode, &o2))
2574 PyErr_Clear();
2575 else {
2576 Py_BEGIN_ALLOW_THREADS
2577 result = MoveFileW(PyUnicode_AsUnicode(o1),
2578 PyUnicode_AsUnicode(o2));
2579 Py_END_ALLOW_THREADS
2580 Py_DECREF(o1);
2581 Py_DECREF(o2);
2582 if (!result)
2583 return win32_error("rename", NULL);
2584 Py_INCREF(Py_None);
2585 return Py_None;
2586 }
2587 }
2588 if (!PyArg_ParseTuple(args, "ss:rename", &p1, &p2))
2589 return NULL;
2590 Py_BEGIN_ALLOW_THREADS
2591 result = MoveFileA(p1, p2);
2592 Py_END_ALLOW_THREADS
2593 if (!result)
2594 return win32_error("rename", NULL);
2595 Py_INCREF(Py_None);
2596 return Py_None;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002597#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002598 return posix_2str(args, "etet:rename", rename);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002599#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002600}
2601
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002602
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002603PyDoc_STRVAR(posix_rmdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002604"rmdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002605Remove a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002606
Barry Warsaw53699e91996-12-10 23:23:01 +00002607static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002608posix_rmdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002609{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002610#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002611 return win32_1str(args, "rmdir", "s:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002612#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002613 return posix_1str(args, "et:rmdir", rmdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002614#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002615}
2616
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002617
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002618PyDoc_STRVAR(posix_stat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002619"stat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002620Perform a stat system call on the given path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002621
Barry Warsaw53699e91996-12-10 23:23:01 +00002622static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002623posix_stat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002624{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002625#ifdef MS_WINDOWS
Martin v. Löwis14694662006-02-03 12:54:16 +00002626 return posix_do_stat(self, args, "et:stat", STAT, "U:stat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002627#else
2628 return posix_do_stat(self, args, "et:stat", STAT, NULL, NULL);
2629#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002630}
2631
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002632
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002633#ifdef HAVE_SYSTEM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002634PyDoc_STRVAR(posix_system__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002635"system(command) -> exit_status\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002636Execute the command (a string) in a subshell.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002637
Barry Warsaw53699e91996-12-10 23:23:01 +00002638static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002639posix_system(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002640{
Guido van Rossumff4949e1992-08-05 19:58:53 +00002641 long sts;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002642#ifdef MS_WINDOWS
2643 wchar_t *command;
2644 if (!PyArg_ParseTuple(args, "u:system", &command))
2645 return NULL;
2646#else
2647 char *command;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002648 if (!PyArg_ParseTuple(args, "s:system", &command))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002649 return NULL;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002650#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002651 Py_BEGIN_ALLOW_THREADS
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002652#ifdef MS_WINDOWS
2653 sts = _wsystem(command);
2654#else
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002655 sts = system(command);
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00002656#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00002657 Py_END_ALLOW_THREADS
Christian Heimes217cfd12007-12-02 14:31:20 +00002658 return PyLong_FromLong(sts);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002659}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002660#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002661
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002662
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002663PyDoc_STRVAR(posix_umask__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002664"umask(new_mask) -> old_mask\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002665Set the current numeric umask and return the previous umask.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002666
Barry Warsaw53699e91996-12-10 23:23:01 +00002667static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002668posix_umask(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002669{
2670 int i;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002671 if (!PyArg_ParseTuple(args, "i:umask", &i))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002672 return NULL;
Fred Drake0368bc42001-07-19 20:48:32 +00002673 i = (int)umask(i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002674 if (i < 0)
2675 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00002676 return PyLong_FromLong((long)i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002677}
2678
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002679
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002680PyDoc_STRVAR(posix_unlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002681"unlink(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002682Remove a file (same as remove(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002683
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002684PyDoc_STRVAR(posix_remove__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002685"remove(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002686Remove a file (same as unlink(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002687
Barry Warsaw53699e91996-12-10 23:23:01 +00002688static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002689posix_unlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002690{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002691#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002692 return win32_1str(args, "remove", "s:remove", DeleteFileA, "U:remove", DeleteFileW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002693#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002694 return posix_1str(args, "et:remove", unlink);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002695#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002696}
2697
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002698
Guido van Rossumb6775db1994-08-01 11:34:53 +00002699#ifdef HAVE_UNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002700PyDoc_STRVAR(posix_uname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002701"uname() -> (sysname, nodename, release, version, machine)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002702Return a tuple identifying the current operating system.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002703
Barry Warsaw53699e91996-12-10 23:23:01 +00002704static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002705posix_uname(PyObject *self, PyObject *noargs)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002706{
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002707 struct utsname u;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002708 int res;
Neal Norwitze241ce82003-02-17 18:17:05 +00002709
Barry Warsaw53699e91996-12-10 23:23:01 +00002710 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002711 res = uname(&u);
Barry Warsaw53699e91996-12-10 23:23:01 +00002712 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002713 if (res < 0)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002714 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00002715 return Py_BuildValue("(sssss)",
Barry Warsaw43d68b81996-12-19 22:10:44 +00002716 u.sysname,
2717 u.nodename,
2718 u.release,
2719 u.version,
2720 u.machine);
Guido van Rossumc39de5f1992-02-05 11:15:54 +00002721}
Guido van Rossumb6775db1994-08-01 11:34:53 +00002722#endif /* HAVE_UNAME */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002723
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002724static int
2725extract_time(PyObject *t, long* sec, long* usec)
2726{
2727 long intval;
2728 if (PyFloat_Check(t)) {
2729 double tval = PyFloat_AsDouble(t);
Christian Heimes90aa7642007-12-19 02:45:37 +00002730 PyObject *intobj = Py_TYPE(t)->tp_as_number->nb_int(t);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002731 if (!intobj)
2732 return -1;
Christian Heimes217cfd12007-12-02 14:31:20 +00002733 intval = PyLong_AsLong(intobj);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002734 Py_DECREF(intobj);
Michael W. Hudsonb8963812005-07-05 15:21:58 +00002735 if (intval == -1 && PyErr_Occurred())
2736 return -1;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002737 *sec = intval;
Tim Peters96940cf2002-09-10 15:37:28 +00002738 *usec = (long)((tval - intval) * 1e6); /* can't exceed 1000000 */
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002739 if (*usec < 0)
2740 /* If rounding gave us a negative number,
2741 truncate. */
2742 *usec = 0;
2743 return 0;
2744 }
Christian Heimes217cfd12007-12-02 14:31:20 +00002745 intval = PyLong_AsLong(t);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002746 if (intval == -1 && PyErr_Occurred())
2747 return -1;
2748 *sec = intval;
2749 *usec = 0;
Martin v. Löwis076b2092002-09-10 15:04:41 +00002750 return 0;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002751}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002752
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002753PyDoc_STRVAR(posix_utime__doc__,
Thomas Wouters477c8d52006-05-27 19:21:47 +00002754"utime(path, (atime, mtime))\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00002755utime(path, None)\n\n\
Barry Warsaw3cef8562000-05-01 16:17:24 +00002756Set the access and modified time of the file to the given values. If the\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002757second form is used, set the access and modified times to the current time.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002758
Barry Warsaw53699e91996-12-10 23:23:01 +00002759static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002760posix_utime(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002761{
Thomas Wouters477c8d52006-05-27 19:21:47 +00002762#ifdef Py_WIN_WIDE_FILENAMES
2763 PyObject *arg;
2764 PyUnicodeObject *obwpath;
2765 wchar_t *wpath = NULL;
2766 char *apath = NULL;
2767 HANDLE hFile;
2768 long atimesec, mtimesec, ausec, musec;
2769 FILETIME atime, mtime;
2770 PyObject *result = NULL;
2771
2772 if (unicode_file_names()) {
2773 if (PyArg_ParseTuple(args, "UO|:utime", &obwpath, &arg)) {
2774 wpath = PyUnicode_AS_UNICODE(obwpath);
2775 Py_BEGIN_ALLOW_THREADS
2776 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002777 NULL, OPEN_EXISTING,
2778 FILE_FLAG_BACKUP_SEMANTICS, NULL);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002779 Py_END_ALLOW_THREADS
2780 if (hFile == INVALID_HANDLE_VALUE)
2781 return win32_error_unicode("utime", wpath);
2782 } else
2783 /* Drop the argument parsing error as narrow strings
2784 are also valid. */
2785 PyErr_Clear();
2786 }
2787 if (!wpath) {
2788 if (!PyArg_ParseTuple(args, "etO:utime",
2789 Py_FileSystemDefaultEncoding, &apath, &arg))
2790 return NULL;
2791 Py_BEGIN_ALLOW_THREADS
2792 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002793 NULL, OPEN_EXISTING,
2794 FILE_FLAG_BACKUP_SEMANTICS, NULL);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002795 Py_END_ALLOW_THREADS
2796 if (hFile == INVALID_HANDLE_VALUE) {
2797 win32_error("utime", apath);
2798 PyMem_Free(apath);
2799 return NULL;
2800 }
2801 PyMem_Free(apath);
2802 }
2803
2804 if (arg == Py_None) {
2805 SYSTEMTIME now;
2806 GetSystemTime(&now);
2807 if (!SystemTimeToFileTime(&now, &mtime) ||
2808 !SystemTimeToFileTime(&now, &atime)) {
2809 win32_error("utime", NULL);
2810 goto done;
2811 }
2812 }
2813 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
2814 PyErr_SetString(PyExc_TypeError,
2815 "utime() arg 2 must be a tuple (atime, mtime)");
2816 goto done;
2817 }
2818 else {
2819 if (extract_time(PyTuple_GET_ITEM(arg, 0),
2820 &atimesec, &ausec) == -1)
2821 goto done;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002822 time_t_to_FILE_TIME(atimesec, 1000*ausec, &atime);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002823 if (extract_time(PyTuple_GET_ITEM(arg, 1),
2824 &mtimesec, &musec) == -1)
2825 goto done;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002826 time_t_to_FILE_TIME(mtimesec, 1000*musec, &mtime);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002827 }
2828 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
2829 /* Avoid putting the file name into the error here,
2830 as that may confuse the user into believing that
2831 something is wrong with the file, when it also
2832 could be the time stamp that gives a problem. */
2833 win32_error("utime", NULL);
2834 }
2835 Py_INCREF(Py_None);
2836 result = Py_None;
2837done:
2838 CloseHandle(hFile);
2839 return result;
2840#else /* Py_WIN_WIDE_FILENAMES */
2841
Neal Norwitz2adf2102004-06-09 01:46:02 +00002842 char *path = NULL;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002843 long atime, mtime, ausec, musec;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002844 int res;
Barry Warsaw3cef8562000-05-01 16:17:24 +00002845 PyObject* arg;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002846
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002847#if defined(HAVE_UTIMES)
2848 struct timeval buf[2];
2849#define ATIME buf[0].tv_sec
2850#define MTIME buf[1].tv_sec
2851#elif defined(HAVE_UTIME_H)
Guido van Rossum6d8841c1997-08-14 19:57:39 +00002852/* XXX should define struct utimbuf instead, above */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002853 struct utimbuf buf;
2854#define ATIME buf.actime
2855#define MTIME buf.modtime
2856#define UTIME_ARG &buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002857#else /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002858 time_t buf[2];
2859#define ATIME buf[0]
2860#define MTIME buf[1]
2861#define UTIME_ARG buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002862#endif /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002863
Mark Hammond817c9292003-12-03 01:22:38 +00002864
Thomas Wouters477c8d52006-05-27 19:21:47 +00002865 if (!PyArg_ParseTuple(args, "etO:utime",
Hye-Shik Chang2b2c9732004-01-04 13:54:25 +00002866 Py_FileSystemDefaultEncoding, &path, &arg))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002867 return NULL;
Barry Warsaw3cef8562000-05-01 16:17:24 +00002868 if (arg == Py_None) {
2869 /* optional time values not given */
2870 Py_BEGIN_ALLOW_THREADS
2871 res = utime(path, NULL);
2872 Py_END_ALLOW_THREADS
2873 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002874 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
Barry Warsaw3cef8562000-05-01 16:17:24 +00002875 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00002876 "utime() arg 2 must be a tuple (atime, mtime)");
Neal Norwitz96652712004-06-06 20:40:27 +00002877 PyMem_Free(path);
Barry Warsaw3cef8562000-05-01 16:17:24 +00002878 return NULL;
2879 }
2880 else {
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002881 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Neal Norwitz96652712004-06-06 20:40:27 +00002882 &atime, &ausec) == -1) {
2883 PyMem_Free(path);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002884 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00002885 }
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002886 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Neal Norwitz96652712004-06-06 20:40:27 +00002887 &mtime, &musec) == -1) {
2888 PyMem_Free(path);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002889 return NULL;
Neal Norwitz96652712004-06-06 20:40:27 +00002890 }
Barry Warsaw3cef8562000-05-01 16:17:24 +00002891 ATIME = atime;
2892 MTIME = mtime;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00002893#ifdef HAVE_UTIMES
2894 buf[0].tv_usec = ausec;
2895 buf[1].tv_usec = musec;
2896 Py_BEGIN_ALLOW_THREADS
2897 res = utimes(path, buf);
2898 Py_END_ALLOW_THREADS
2899#else
Barry Warsaw3cef8562000-05-01 16:17:24 +00002900 Py_BEGIN_ALLOW_THREADS
2901 res = utime(path, UTIME_ARG);
2902 Py_END_ALLOW_THREADS
Mark Hammond817c9292003-12-03 01:22:38 +00002903#endif /* HAVE_UTIMES */
Barry Warsaw3cef8562000-05-01 16:17:24 +00002904 }
Mark Hammond2d5914b2004-05-04 08:10:37 +00002905 if (res < 0) {
Neal Norwitz96652712004-06-06 20:40:27 +00002906 return posix_error_with_allocated_filename(path);
Mark Hammond2d5914b2004-05-04 08:10:37 +00002907 }
Neal Norwitz96652712004-06-06 20:40:27 +00002908 PyMem_Free(path);
Barry Warsaw53699e91996-12-10 23:23:01 +00002909 Py_INCREF(Py_None);
2910 return Py_None;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00002911#undef UTIME_ARG
2912#undef ATIME
2913#undef MTIME
Thomas Wouters477c8d52006-05-27 19:21:47 +00002914#endif /* Py_WIN_WIDE_FILENAMES */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002915}
2916
Guido van Rossum85e3b011991-06-03 12:42:10 +00002917
Guido van Rossum3b066191991-06-04 19:40:25 +00002918/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002919
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002920PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002921"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002922Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002923
Barry Warsaw53699e91996-12-10 23:23:01 +00002924static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002925posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002926{
2927 int sts;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002928 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
Guido van Rossum85e3b011991-06-03 12:42:10 +00002929 return NULL;
2930 _exit(sts);
Guido van Rossuma376cc51996-12-05 23:43:35 +00002931 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00002932}
2933
Martin v. Löwis114619e2002-10-07 06:44:21 +00002934#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
2935static void
Martin v. Löwis725507b2006-03-07 12:08:51 +00002936free_string_array(char **array, Py_ssize_t count)
Martin v. Löwis114619e2002-10-07 06:44:21 +00002937{
Martin v. Löwis725507b2006-03-07 12:08:51 +00002938 Py_ssize_t i;
Martin v. Löwis114619e2002-10-07 06:44:21 +00002939 for (i = 0; i < count; i++)
2940 PyMem_Free(array[i]);
2941 PyMem_DEL(array);
2942}
2943#endif
2944
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002945
Guido van Rossuma4916fa1996-05-23 22:58:55 +00002946#ifdef HAVE_EXECV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002947PyDoc_STRVAR(posix_execv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002948"execv(path, args)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002949Execute an executable path with arguments, replacing current process.\n\
2950\n\
2951 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002952 args: tuple or list of strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002953
Barry Warsaw53699e91996-12-10 23:23:01 +00002954static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002955posix_execv(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00002956{
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002957 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00002958 PyObject *argv;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002959 char **argvlist;
Martin v. Löwis725507b2006-03-07 12:08:51 +00002960 Py_ssize_t i, argc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002961 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossum85e3b011991-06-03 12:42:10 +00002962
Guido van Rossum89b33251993-10-22 14:26:06 +00002963 /* execv has two arguments: (path, argv), where
Guido van Rossum85e3b011991-06-03 12:42:10 +00002964 argv is a list or tuple of strings. */
2965
Martin v. Löwis114619e2002-10-07 06:44:21 +00002966 if (!PyArg_ParseTuple(args, "etO:execv",
2967 Py_FileSystemDefaultEncoding,
2968 &path, &argv))
Guido van Rossum85e3b011991-06-03 12:42:10 +00002969 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00002970 if (PyList_Check(argv)) {
2971 argc = PyList_Size(argv);
2972 getitem = PyList_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002973 }
Barry Warsaw53699e91996-12-10 23:23:01 +00002974 else if (PyTuple_Check(argv)) {
2975 argc = PyTuple_Size(argv);
2976 getitem = PyTuple_GetItem;
Guido van Rossum85e3b011991-06-03 12:42:10 +00002977 }
Guido van Rossumef0a00e1992-01-27 16:51:30 +00002978 else {
Fred Drake661ea262000-10-24 19:57:45 +00002979 PyErr_SetString(PyExc_TypeError, "execv() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00002980 PyMem_Free(path);
Guido van Rossum50422b42000-04-26 20:34:28 +00002981 return NULL;
2982 }
Thomas Heller6790d602007-08-30 17:15:14 +00002983 if (argc < 1) {
2984 PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
2985 PyMem_Free(path);
2986 return NULL;
2987 }
Guido van Rossum50422b42000-04-26 20:34:28 +00002988
Barry Warsaw53699e91996-12-10 23:23:01 +00002989 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00002990 if (argvlist == NULL) {
2991 PyMem_Free(path);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00002992 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00002993 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00002994 for (i = 0; i < argc; i++) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00002995 if (!PyArg_Parse((*getitem)(argv, i), "et",
2996 Py_FileSystemDefaultEncoding,
2997 &argvlist[i])) {
2998 free_string_array(argvlist, i);
Tim Peters5aa91602002-01-30 05:46:57 +00002999 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003000 "execv() arg 2 must contain only strings");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003001 PyMem_Free(path);
Guido van Rossum50422b42000-04-26 20:34:28 +00003002 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00003003
Guido van Rossum85e3b011991-06-03 12:42:10 +00003004 }
Guido van Rossum85e3b011991-06-03 12:42:10 +00003005 }
3006 argvlist[argc] = NULL;
3007
Guido van Rossumef0a00e1992-01-27 16:51:30 +00003008 execv(path, argvlist);
Guido van Rossumb6775db1994-08-01 11:34:53 +00003009
Guido van Rossum85e3b011991-06-03 12:42:10 +00003010 /* If we get here it's definitely an error */
3011
Martin v. Löwis114619e2002-10-07 06:44:21 +00003012 free_string_array(argvlist, argc);
3013 PyMem_Free(path);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003014 return posix_error();
3015}
3016
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003017
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003018PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003019"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003020Execute a path with arguments and environment, replacing current process.\n\
3021\n\
3022 path: path of executable file\n\
3023 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003024 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003025
Barry Warsaw53699e91996-12-10 23:23:01 +00003026static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003027posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003028{
3029 char *path;
Barry Warsaw53699e91996-12-10 23:23:01 +00003030 PyObject *argv, *env;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003031 char **argvlist;
3032 char **envlist;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003033 PyObject *key, *val, *keys=NULL, *vals=NULL;
Martin v. Löwis725507b2006-03-07 12:08:51 +00003034 Py_ssize_t i, pos, argc, envc;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003035 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Thomas Wouters477c8d52006-05-27 19:21:47 +00003036 Py_ssize_t lastarg = 0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003037
3038 /* execve has three arguments: (path, argv, env), where
3039 argv is a list or tuple of strings and env is a dictionary
3040 like posix.environ. */
3041
Martin v. Löwis114619e2002-10-07 06:44:21 +00003042 if (!PyArg_ParseTuple(args, "etOO:execve",
3043 Py_FileSystemDefaultEncoding,
3044 &path, &argv, &env))
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003045 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00003046 if (PyList_Check(argv)) {
3047 argc = PyList_Size(argv);
3048 getitem = PyList_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003049 }
Barry Warsaw53699e91996-12-10 23:23:01 +00003050 else if (PyTuple_Check(argv)) {
3051 argc = PyTuple_Size(argv);
3052 getitem = PyTuple_GetItem;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003053 }
3054 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003055 PyErr_SetString(PyExc_TypeError,
3056 "execve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003057 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003058 }
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003059 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003060 PyErr_SetString(PyExc_TypeError,
3061 "execve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003062 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003063 }
3064
Barry Warsaw53699e91996-12-10 23:23:01 +00003065 argvlist = PyMem_NEW(char *, argc+1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003066 if (argvlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003067 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003068 goto fail_0;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003069 }
3070 for (i = 0; i < argc; i++) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003071 if (!PyArg_Parse((*getitem)(argv, i),
Martin v. Löwis114619e2002-10-07 06:44:21 +00003072 "et;execve() arg 2 must contain only strings",
Guido van Rossum1e700d22002-10-16 16:52:11 +00003073 Py_FileSystemDefaultEncoding,
Barry Warsaw43d68b81996-12-19 22:10:44 +00003074 &argvlist[i]))
3075 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003076 lastarg = i;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003077 goto fail_1;
3078 }
3079 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003080 lastarg = argc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003081 argvlist[argc] = NULL;
3082
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003083 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003084 if (i < 0)
3085 goto fail_1;
Barry Warsaw53699e91996-12-10 23:23:01 +00003086 envlist = PyMem_NEW(char *, i + 1);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003087 if (envlist == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003088 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003089 goto fail_1;
3090 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003091 envc = 0;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003092 keys = PyMapping_Keys(env);
3093 vals = PyMapping_Values(env);
3094 if (!keys || !vals)
3095 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003096 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3097 PyErr_SetString(PyExc_TypeError,
3098 "execve(): env.keys() or env.values() is not a list");
3099 goto fail_2;
3100 }
Tim Peters5aa91602002-01-30 05:46:57 +00003101
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003102 for (pos = 0; pos < i; pos++) {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003103 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003104 size_t len;
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003105
3106 key = PyList_GetItem(keys, pos);
3107 val = PyList_GetItem(vals, pos);
3108 if (!key || !val)
3109 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003110
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003111 if (!PyArg_Parse(
3112 key,
3113 "s;execve() arg 3 contains a non-string key",
3114 &k) ||
3115 !PyArg_Parse(
3116 val,
3117 "s;execve() arg 3 contains a non-string value",
3118 &v))
Barry Warsaw43d68b81996-12-19 22:10:44 +00003119 {
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003120 goto fail_2;
3121 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00003122
3123#if defined(PYOS_OS2)
3124 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
3125 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
3126#endif
Christian Heimes830a4bc2007-11-22 07:43:40 +00003127 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Tim Petersc8996f52001-12-03 20:41:00 +00003128 p = PyMem_NEW(char, len);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003129 if (p == NULL) {
Barry Warsaw53699e91996-12-10 23:23:01 +00003130 PyErr_NoMemory();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003131 goto fail_2;
3132 }
Tim Petersc8996f52001-12-03 20:41:00 +00003133 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003134 envlist[envc++] = p;
Guido van Rossumd48f2521997-12-05 22:19:34 +00003135#if defined(PYOS_OS2)
3136 }
3137#endif
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003138 }
3139 envlist[envc] = 0;
3140
3141 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00003142
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003143 /* If we get here it's definitely an error */
3144
3145 (void) posix_error();
3146
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003147 fail_2:
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003148 while (--envc >= 0)
Barry Warsaw53699e91996-12-10 23:23:01 +00003149 PyMem_DEL(envlist[envc]);
3150 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003151 fail_1:
3152 free_string_array(argvlist, lastarg);
Barry Warsaw5ed19dc1997-01-29 15:08:24 +00003153 Py_XDECREF(vals);
3154 Py_XDECREF(keys);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003155 fail_0:
Martin v. Löwis114619e2002-10-07 06:44:21 +00003156 PyMem_Free(path);
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003157 return NULL;
3158}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003159#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003160
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003161
Guido van Rossuma1065681999-01-25 23:20:23 +00003162#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003163PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003164"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003165Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003166\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003167 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003168 path: path of executable file\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003169 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003170
3171static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003172posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003173{
3174 char *path;
3175 PyObject *argv;
3176 char **argvlist;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003177 int mode, i;
3178 Py_ssize_t argc;
Tim Peters79248aa2001-08-29 21:37:10 +00003179 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003180 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossuma1065681999-01-25 23:20:23 +00003181
3182 /* spawnv has three arguments: (mode, path, argv), where
3183 argv is a list or tuple of strings. */
3184
Martin v. Löwis114619e2002-10-07 06:44:21 +00003185 if (!PyArg_ParseTuple(args, "ietO:spawnv", &mode,
3186 Py_FileSystemDefaultEncoding,
3187 &path, &argv))
Guido van Rossuma1065681999-01-25 23:20:23 +00003188 return NULL;
3189 if (PyList_Check(argv)) {
3190 argc = PyList_Size(argv);
3191 getitem = PyList_GetItem;
3192 }
3193 else if (PyTuple_Check(argv)) {
3194 argc = PyTuple_Size(argv);
3195 getitem = PyTuple_GetItem;
3196 }
3197 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003198 PyErr_SetString(PyExc_TypeError,
3199 "spawnv() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003200 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00003201 return NULL;
3202 }
3203
3204 argvlist = PyMem_NEW(char *, argc+1);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003205 if (argvlist == NULL) {
3206 PyMem_Free(path);
Neal Norwitzec74f2f2003-02-11 23:05:40 +00003207 return PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003208 }
Guido van Rossuma1065681999-01-25 23:20:23 +00003209 for (i = 0; i < argc; i++) {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003210 if (!PyArg_Parse((*getitem)(argv, i), "et",
3211 Py_FileSystemDefaultEncoding,
3212 &argvlist[i])) {
3213 free_string_array(argvlist, i);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003214 PyErr_SetString(
3215 PyExc_TypeError,
3216 "spawnv() arg 2 must contain only strings");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003217 PyMem_Free(path);
Fred Drake137507e2000-06-01 02:02:46 +00003218 return NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00003219 }
3220 }
3221 argvlist[argc] = NULL;
3222
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003223#if defined(PYOS_OS2) && defined(PYCC_GCC)
3224 Py_BEGIN_ALLOW_THREADS
3225 spawnval = spawnv(mode, path, argvlist);
3226 Py_END_ALLOW_THREADS
3227#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003228 if (mode == _OLD_P_OVERLAY)
3229 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00003230
Tim Peters25059d32001-12-07 20:35:43 +00003231 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003232 spawnval = _spawnv(mode, path, argvlist);
Tim Peters25059d32001-12-07 20:35:43 +00003233 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003234#endif
Tim Peters5aa91602002-01-30 05:46:57 +00003235
Martin v. Löwis114619e2002-10-07 06:44:21 +00003236 free_string_array(argvlist, argc);
3237 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00003238
Fred Drake699f3522000-06-29 21:12:41 +00003239 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003240 return posix_error();
3241 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003242#if SIZEOF_LONG == SIZEOF_VOID_P
3243 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003244#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003245 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003246#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003247}
3248
3249
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003250PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003251"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00003252Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003253\n\
Fred Drakea6dff3e1999-02-01 22:24:40 +00003254 mode: mode of process creation\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00003255 path: path of executable file\n\
3256 args: tuple or list of arguments\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003257 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00003258
3259static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003260posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00003261{
3262 char *path;
3263 PyObject *argv, *env;
3264 char **argvlist;
3265 char **envlist;
3266 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003267 int mode, pos, envc;
3268 Py_ssize_t argc, i;
Tim Peters79248aa2001-08-29 21:37:10 +00003269 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003270 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Thomas Wouters477c8d52006-05-27 19:21:47 +00003271 Py_ssize_t lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003272
3273 /* spawnve has four arguments: (mode, path, argv, env), where
3274 argv is a list or tuple of strings and env is a dictionary
3275 like posix.environ. */
3276
Martin v. Löwis114619e2002-10-07 06:44:21 +00003277 if (!PyArg_ParseTuple(args, "ietOO:spawnve", &mode,
3278 Py_FileSystemDefaultEncoding,
3279 &path, &argv, &env))
Guido van Rossuma1065681999-01-25 23:20:23 +00003280 return NULL;
3281 if (PyList_Check(argv)) {
3282 argc = PyList_Size(argv);
3283 getitem = PyList_GetItem;
3284 }
3285 else if (PyTuple_Check(argv)) {
3286 argc = PyTuple_Size(argv);
3287 getitem = PyTuple_GetItem;
3288 }
3289 else {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003290 PyErr_SetString(PyExc_TypeError,
3291 "spawnve() arg 2 must be a tuple or list");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003292 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003293 }
3294 if (!PyMapping_Check(env)) {
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003295 PyErr_SetString(PyExc_TypeError,
3296 "spawnve() arg 3 must be a mapping object");
Martin v. Löwis114619e2002-10-07 06:44:21 +00003297 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003298 }
3299
3300 argvlist = PyMem_NEW(char *, argc+1);
3301 if (argvlist == NULL) {
3302 PyErr_NoMemory();
Martin v. Löwis114619e2002-10-07 06:44:21 +00003303 goto fail_0;
Guido van Rossuma1065681999-01-25 23:20:23 +00003304 }
3305 for (i = 0; i < argc; i++) {
3306 if (!PyArg_Parse((*getitem)(argv, i),
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003307 "et;spawnve() arg 2 must contain only strings",
Martin v. Löwis114619e2002-10-07 06:44:21 +00003308 Py_FileSystemDefaultEncoding,
Guido van Rossuma1065681999-01-25 23:20:23 +00003309 &argvlist[i]))
3310 {
Martin v. Löwis114619e2002-10-07 06:44:21 +00003311 lastarg = i;
Guido van Rossuma1065681999-01-25 23:20:23 +00003312 goto fail_1;
3313 }
3314 }
Martin v. Löwis114619e2002-10-07 06:44:21 +00003315 lastarg = argc;
Guido van Rossuma1065681999-01-25 23:20:23 +00003316 argvlist[argc] = NULL;
3317
Jeremy Hylton03657cf2000-07-12 13:05:33 +00003318 i = PyMapping_Size(env);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003319 if (i < 0)
3320 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00003321 envlist = PyMem_NEW(char *, i + 1);
3322 if (envlist == NULL) {
3323 PyErr_NoMemory();
3324 goto fail_1;
3325 }
3326 envc = 0;
3327 keys = PyMapping_Keys(env);
3328 vals = PyMapping_Values(env);
3329 if (!keys || !vals)
3330 goto fail_2;
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003331 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3332 PyErr_SetString(PyExc_TypeError,
3333 "spawnve(): env.keys() or env.values() is not a list");
3334 goto fail_2;
3335 }
Tim Peters5aa91602002-01-30 05:46:57 +00003336
Guido van Rossuma1065681999-01-25 23:20:23 +00003337 for (pos = 0; pos < i; pos++) {
3338 char *p, *k, *v;
Tim Petersc8996f52001-12-03 20:41:00 +00003339 size_t len;
Guido van Rossuma1065681999-01-25 23:20:23 +00003340
3341 key = PyList_GetItem(keys, pos);
3342 val = PyList_GetItem(vals, pos);
3343 if (!key || !val)
3344 goto fail_2;
Tim Peters5aa91602002-01-30 05:46:57 +00003345
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003346 if (!PyArg_Parse(
3347 key,
3348 "s;spawnve() arg 3 contains a non-string key",
3349 &k) ||
3350 !PyArg_Parse(
3351 val,
3352 "s;spawnve() arg 3 contains a non-string value",
3353 &v))
Guido van Rossuma1065681999-01-25 23:20:23 +00003354 {
3355 goto fail_2;
3356 }
Christian Heimes830a4bc2007-11-22 07:43:40 +00003357 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Tim Petersc8996f52001-12-03 20:41:00 +00003358 p = PyMem_NEW(char, len);
Guido van Rossuma1065681999-01-25 23:20:23 +00003359 if (p == NULL) {
3360 PyErr_NoMemory();
3361 goto fail_2;
3362 }
Tim Petersc8996f52001-12-03 20:41:00 +00003363 PyOS_snprintf(p, len, "%s=%s", k, v);
Guido van Rossuma1065681999-01-25 23:20:23 +00003364 envlist[envc++] = p;
3365 }
3366 envlist[envc] = 0;
3367
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003368#if defined(PYOS_OS2) && defined(PYCC_GCC)
3369 Py_BEGIN_ALLOW_THREADS
3370 spawnval = spawnve(mode, path, argvlist, envlist);
3371 Py_END_ALLOW_THREADS
3372#else
Guido van Rossum246bc171999-02-01 23:54:31 +00003373 if (mode == _OLD_P_OVERLAY)
3374 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00003375
3376 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00003377 spawnval = _spawnve(mode, path, argvlist, envlist);
Tim Peters25059d32001-12-07 20:35:43 +00003378 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003379#endif
Tim Peters25059d32001-12-07 20:35:43 +00003380
Fred Drake699f3522000-06-29 21:12:41 +00003381 if (spawnval == -1)
Guido van Rossuma1065681999-01-25 23:20:23 +00003382 (void) posix_error();
3383 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00003384#if SIZEOF_LONG == SIZEOF_VOID_P
3385 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003386#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00003387 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00003388#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00003389
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003390 fail_2:
Guido van Rossuma1065681999-01-25 23:20:23 +00003391 while (--envc >= 0)
3392 PyMem_DEL(envlist[envc]);
3393 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003394 fail_1:
Martin v. Löwis114619e2002-10-07 06:44:21 +00003395 free_string_array(argvlist, lastarg);
Guido van Rossuma1065681999-01-25 23:20:23 +00003396 Py_XDECREF(vals);
3397 Py_XDECREF(keys);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003398 fail_0:
3399 PyMem_Free(path);
Guido van Rossuma1065681999-01-25 23:20:23 +00003400 return res;
3401}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003402
3403/* OS/2 supports spawnvp & spawnvpe natively */
3404#if defined(PYOS_OS2)
3405PyDoc_STRVAR(posix_spawnvp__doc__,
3406"spawnvp(mode, file, args)\n\n\
3407Execute the program 'file' in a new process, using the environment\n\
3408search path to find the file.\n\
3409\n\
3410 mode: mode of process creation\n\
3411 file: executable file name\n\
3412 args: tuple or list of strings");
3413
3414static PyObject *
3415posix_spawnvp(PyObject *self, PyObject *args)
3416{
3417 char *path;
3418 PyObject *argv;
3419 char **argvlist;
3420 int mode, i, argc;
3421 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003422 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003423
3424 /* spawnvp has three arguments: (mode, path, argv), where
3425 argv is a list or tuple of strings. */
3426
3427 if (!PyArg_ParseTuple(args, "ietO:spawnvp", &mode,
3428 Py_FileSystemDefaultEncoding,
3429 &path, &argv))
3430 return NULL;
3431 if (PyList_Check(argv)) {
3432 argc = PyList_Size(argv);
3433 getitem = PyList_GetItem;
3434 }
3435 else if (PyTuple_Check(argv)) {
3436 argc = PyTuple_Size(argv);
3437 getitem = PyTuple_GetItem;
3438 }
3439 else {
3440 PyErr_SetString(PyExc_TypeError,
3441 "spawnvp() arg 2 must be a tuple or list");
3442 PyMem_Free(path);
3443 return NULL;
3444 }
3445
3446 argvlist = PyMem_NEW(char *, argc+1);
3447 if (argvlist == NULL) {
3448 PyMem_Free(path);
3449 return PyErr_NoMemory();
3450 }
3451 for (i = 0; i < argc; i++) {
3452 if (!PyArg_Parse((*getitem)(argv, i), "et",
3453 Py_FileSystemDefaultEncoding,
3454 &argvlist[i])) {
3455 free_string_array(argvlist, i);
3456 PyErr_SetString(
3457 PyExc_TypeError,
3458 "spawnvp() arg 2 must contain only strings");
3459 PyMem_Free(path);
3460 return NULL;
3461 }
3462 }
3463 argvlist[argc] = NULL;
3464
3465 Py_BEGIN_ALLOW_THREADS
3466#if defined(PYCC_GCC)
3467 spawnval = spawnvp(mode, path, argvlist);
3468#else
3469 spawnval = _spawnvp(mode, path, argvlist);
3470#endif
3471 Py_END_ALLOW_THREADS
3472
3473 free_string_array(argvlist, argc);
3474 PyMem_Free(path);
3475
3476 if (spawnval == -1)
3477 return posix_error();
3478 else
3479 return Py_BuildValue("l", (long) spawnval);
3480}
3481
3482
3483PyDoc_STRVAR(posix_spawnvpe__doc__,
3484"spawnvpe(mode, file, args, env)\n\n\
3485Execute the program 'file' in a new process, using the environment\n\
3486search path to find the file.\n\
3487\n\
3488 mode: mode of process creation\n\
3489 file: executable file name\n\
3490 args: tuple or list of arguments\n\
3491 env: dictionary of strings mapping to strings");
3492
3493static PyObject *
3494posix_spawnvpe(PyObject *self, PyObject *args)
3495{
3496 char *path;
3497 PyObject *argv, *env;
3498 char **argvlist;
3499 char **envlist;
3500 PyObject *key, *val, *keys=NULL, *vals=NULL, *res=NULL;
3501 int mode, i, pos, argc, envc;
3502 Py_intptr_t spawnval;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003503 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003504 int lastarg = 0;
3505
3506 /* spawnvpe has four arguments: (mode, path, argv, env), where
3507 argv is a list or tuple of strings and env is a dictionary
3508 like posix.environ. */
3509
3510 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
3511 Py_FileSystemDefaultEncoding,
3512 &path, &argv, &env))
3513 return NULL;
3514 if (PyList_Check(argv)) {
3515 argc = PyList_Size(argv);
3516 getitem = PyList_GetItem;
3517 }
3518 else if (PyTuple_Check(argv)) {
3519 argc = PyTuple_Size(argv);
3520 getitem = PyTuple_GetItem;
3521 }
3522 else {
3523 PyErr_SetString(PyExc_TypeError,
3524 "spawnvpe() arg 2 must be a tuple or list");
3525 goto fail_0;
3526 }
3527 if (!PyMapping_Check(env)) {
3528 PyErr_SetString(PyExc_TypeError,
3529 "spawnvpe() arg 3 must be a mapping object");
3530 goto fail_0;
3531 }
3532
3533 argvlist = PyMem_NEW(char *, argc+1);
3534 if (argvlist == NULL) {
3535 PyErr_NoMemory();
3536 goto fail_0;
3537 }
3538 for (i = 0; i < argc; i++) {
3539 if (!PyArg_Parse((*getitem)(argv, i),
3540 "et;spawnvpe() arg 2 must contain only strings",
3541 Py_FileSystemDefaultEncoding,
3542 &argvlist[i]))
3543 {
3544 lastarg = i;
3545 goto fail_1;
3546 }
3547 }
3548 lastarg = argc;
3549 argvlist[argc] = NULL;
3550
3551 i = PyMapping_Size(env);
3552 if (i < 0)
3553 goto fail_1;
3554 envlist = PyMem_NEW(char *, i + 1);
3555 if (envlist == NULL) {
3556 PyErr_NoMemory();
3557 goto fail_1;
3558 }
3559 envc = 0;
3560 keys = PyMapping_Keys(env);
3561 vals = PyMapping_Values(env);
3562 if (!keys || !vals)
3563 goto fail_2;
3564 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3565 PyErr_SetString(PyExc_TypeError,
3566 "spawnvpe(): env.keys() or env.values() is not a list");
3567 goto fail_2;
3568 }
3569
3570 for (pos = 0; pos < i; pos++) {
3571 char *p, *k, *v;
3572 size_t len;
3573
3574 key = PyList_GetItem(keys, pos);
3575 val = PyList_GetItem(vals, pos);
3576 if (!key || !val)
3577 goto fail_2;
3578
3579 if (!PyArg_Parse(
3580 key,
3581 "s;spawnvpe() arg 3 contains a non-string key",
3582 &k) ||
3583 !PyArg_Parse(
3584 val,
3585 "s;spawnvpe() arg 3 contains a non-string value",
3586 &v))
3587 {
3588 goto fail_2;
3589 }
Christian Heimes830a4bc2007-11-22 07:43:40 +00003590 len = PyUnicode_GetSize(key) + PyUnicode_GetSize(val) + 2;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003591 p = PyMem_NEW(char, len);
3592 if (p == NULL) {
3593 PyErr_NoMemory();
3594 goto fail_2;
3595 }
3596 PyOS_snprintf(p, len, "%s=%s", k, v);
3597 envlist[envc++] = p;
3598 }
3599 envlist[envc] = 0;
3600
3601 Py_BEGIN_ALLOW_THREADS
3602#if defined(PYCC_GCC)
Christian Heimes292d3512008-02-03 16:51:08 +00003603 spawnval = spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003604#else
Christian Heimes292d3512008-02-03 16:51:08 +00003605 spawnval = _spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00003606#endif
3607 Py_END_ALLOW_THREADS
3608
3609 if (spawnval == -1)
3610 (void) posix_error();
3611 else
3612 res = Py_BuildValue("l", (long) spawnval);
3613
3614 fail_2:
3615 while (--envc >= 0)
3616 PyMem_DEL(envlist[envc]);
3617 PyMem_DEL(envlist);
3618 fail_1:
3619 free_string_array(argvlist, lastarg);
3620 Py_XDECREF(vals);
3621 Py_XDECREF(keys);
3622 fail_0:
3623 PyMem_Free(path);
3624 return res;
3625}
3626#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00003627#endif /* HAVE_SPAWNV */
3628
3629
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003630#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003631PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003632"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003633Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
3634\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003635Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003636
3637static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003638posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003639{
Christian Heimes400adb02008-02-01 08:12:03 +00003640 pid_t pid = fork1();
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003641 if (pid == -1)
3642 return posix_error();
Georg Brandl2ee470f2008-07-16 12:55:28 +00003643 if (pid == 0)
3644 PyOS_AfterFork();
Christian Heimes400adb02008-02-01 08:12:03 +00003645 return PyLong_FromLong(pid);
Guido van Rossum2242f2f2001-04-11 20:58:20 +00003646}
3647#endif
3648
3649
Guido van Rossumad0ee831995-03-01 10:34:45 +00003650#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003651PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003652"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003653Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003654Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003655
Barry Warsaw53699e91996-12-10 23:23:01 +00003656static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003657posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003658{
Christian Heimes400adb02008-02-01 08:12:03 +00003659 pid_t pid = fork();
Guido van Rossum85e3b011991-06-03 12:42:10 +00003660 if (pid == -1)
3661 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003662 if (pid == 0)
3663 PyOS_AfterFork();
Christian Heimes400adb02008-02-01 08:12:03 +00003664 return PyLong_FromLong(pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00003665}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003666#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003667
Neal Norwitzb59798b2003-03-21 01:43:31 +00003668/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00003669/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
3670#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00003671#define DEV_PTY_FILE "/dev/ptc"
3672#define HAVE_DEV_PTMX
3673#else
3674#define DEV_PTY_FILE "/dev/ptmx"
3675#endif
3676
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003677#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003678#ifdef HAVE_PTY_H
3679#include <pty.h>
3680#else
3681#ifdef HAVE_LIBUTIL_H
3682#include <libutil.h>
Fred Drake8cef4cf2000-06-28 16:40:38 +00003683#endif /* HAVE_LIBUTIL_H */
3684#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00003685#ifdef HAVE_STROPTS_H
3686#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003687#endif
3688#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003689
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003690#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003691PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003692"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003693Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003694
3695static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003696posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003697{
3698 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003699#ifndef HAVE_OPENPTY
3700 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00003701#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003702#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003703 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003704#ifdef sun
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003705 extern char *ptsname(int fildes);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003706#endif
3707#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00003708
Thomas Wouters70c21a12000-07-14 14:28:33 +00003709#ifdef HAVE_OPENPTY
Fred Drake8cef4cf2000-06-28 16:40:38 +00003710 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
3711 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003712#elif defined(HAVE__GETPTY)
Thomas Wouters70c21a12000-07-14 14:28:33 +00003713 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
3714 if (slave_name == NULL)
3715 return posix_error();
3716
3717 slave_fd = open(slave_name, O_RDWR);
3718 if (slave_fd < 0)
3719 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003720#else
Neal Norwitzb59798b2003-03-21 01:43:31 +00003721 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003722 if (master_fd < 0)
3723 return posix_error();
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003724 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003725 /* change permission of slave */
3726 if (grantpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003727 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003728 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003729 }
3730 /* unlock slave */
3731 if (unlockpt(master_fd) < 0) {
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003732 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003733 return posix_error();
Martin v. Löwisc8b2e772002-12-31 14:30:26 +00003734 }
Anthony Baxter9ceaa722004-10-13 14:48:50 +00003735 PyOS_setsig(SIGCHLD, sig_saved);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003736 slave_name = ptsname(master_fd); /* get name of slave */
3737 if (slave_name == NULL)
3738 return posix_error();
3739 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
3740 if (slave_fd < 0)
3741 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00003742#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003743 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
3744 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00003745#ifndef __hpux
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003746 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00003747#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003748#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00003749#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00003750
Fred Drake8cef4cf2000-06-28 16:40:38 +00003751 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00003752
Fred Drake8cef4cf2000-06-28 16:40:38 +00003753}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00003754#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00003755
3756#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003757PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003758"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00003759Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
3760Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003761To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00003762
3763static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003764posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00003765{
Christian Heimes400adb02008-02-01 08:12:03 +00003766 int master_fd = -1;
3767 pid_t pid;
Tim Peters5aa91602002-01-30 05:46:57 +00003768
Fred Drake8cef4cf2000-06-28 16:40:38 +00003769 pid = forkpty(&master_fd, NULL, NULL, NULL);
3770 if (pid == -1)
3771 return posix_error();
Fred Drake49b0c3b2000-07-06 19:42:19 +00003772 if (pid == 0)
3773 PyOS_AfterFork();
Christian Heimes400adb02008-02-01 08:12:03 +00003774 return Py_BuildValue("(li)", pid, master_fd);
Fred Drake8cef4cf2000-06-28 16:40:38 +00003775}
3776#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003777
Guido van Rossumad0ee831995-03-01 10:34:45 +00003778#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003779PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003780"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003781Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003782
Barry Warsaw53699e91996-12-10 23:23:01 +00003783static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003784posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003785{
Christian Heimes217cfd12007-12-02 14:31:20 +00003786 return PyLong_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003787}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003788#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003789
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003790
Guido van Rossumad0ee831995-03-01 10:34:45 +00003791#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003792PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003793"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003794Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003795
Barry Warsaw53699e91996-12-10 23:23:01 +00003796static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003797posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003798{
Christian Heimes217cfd12007-12-02 14:31:20 +00003799 return PyLong_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003800}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003801#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003802
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003803
Guido van Rossumad0ee831995-03-01 10:34:45 +00003804#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003805PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003806"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003807Return the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003808
Barry Warsaw53699e91996-12-10 23:23:01 +00003809static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003810posix_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003811{
Christian Heimes217cfd12007-12-02 14:31:20 +00003812 return PyLong_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003813}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003814#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003815
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003816
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003817PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003818"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003819Return the current process id");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003820
Barry Warsaw53699e91996-12-10 23:23:01 +00003821static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003822posix_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003823{
Christian Heimes217cfd12007-12-02 14:31:20 +00003824 return PyLong_FromLong((long)getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00003825}
3826
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003827
Fred Drakec9680921999-12-13 16:37:25 +00003828#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003829PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003830"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003831Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00003832
3833static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003834posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00003835{
3836 PyObject *result = NULL;
3837
Fred Drakec9680921999-12-13 16:37:25 +00003838#ifdef NGROUPS_MAX
3839#define MAX_GROUPS NGROUPS_MAX
3840#else
3841 /* defined to be 16 on Solaris7, so this should be a small number */
3842#define MAX_GROUPS 64
3843#endif
3844 gid_t grouplist[MAX_GROUPS];
3845 int n;
3846
3847 n = getgroups(MAX_GROUPS, grouplist);
3848 if (n < 0)
3849 posix_error();
3850 else {
3851 result = PyList_New(n);
3852 if (result != NULL) {
Fred Drakec9680921999-12-13 16:37:25 +00003853 int i;
3854 for (i = 0; i < n; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +00003855 PyObject *o = PyLong_FromLong((long)grouplist[i]);
Fred Drakec9680921999-12-13 16:37:25 +00003856 if (o == NULL) {
3857 Py_DECREF(result);
3858 result = NULL;
3859 break;
3860 }
3861 PyList_SET_ITEM(result, i, o);
3862 }
3863 }
3864 }
Neal Norwitze241ce82003-02-17 18:17:05 +00003865
Fred Drakec9680921999-12-13 16:37:25 +00003866 return result;
3867}
3868#endif
3869
Martin v. Löwis606edc12002-06-13 21:09:11 +00003870#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00003871PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003872"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00003873Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00003874
3875static PyObject *
3876posix_getpgid(PyObject *self, PyObject *args)
3877{
3878 int pid, pgid;
3879 if (!PyArg_ParseTuple(args, "i:getpgid", &pid))
3880 return NULL;
3881 pgid = getpgid(pid);
3882 if (pgid < 0)
3883 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00003884 return PyLong_FromLong((long)pgid);
Martin v. Löwis606edc12002-06-13 21:09:11 +00003885}
3886#endif /* HAVE_GETPGID */
3887
3888
Guido van Rossumb6775db1994-08-01 11:34:53 +00003889#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003890PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003891"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003892Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003893
Barry Warsaw53699e91996-12-10 23:23:01 +00003894static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003895posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00003896{
Guido van Rossumb6775db1994-08-01 11:34:53 +00003897#ifdef GETPGRP_HAVE_ARG
Christian Heimes217cfd12007-12-02 14:31:20 +00003898 return PyLong_FromLong((long)getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003899#else /* GETPGRP_HAVE_ARG */
Christian Heimes217cfd12007-12-02 14:31:20 +00003900 return PyLong_FromLong((long)getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003901#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00003902}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003903#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00003904
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003905
Guido van Rossumb6775db1994-08-01 11:34:53 +00003906#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003907PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003908"setpgrp()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003909Make this process a session leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003910
Barry Warsaw53699e91996-12-10 23:23:01 +00003911static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003912posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00003913{
Guido van Rossum64933891994-10-20 21:56:42 +00003914#ifdef SETPGRP_HAVE_ARG
Guido van Rossumc2670a01992-09-13 20:07:29 +00003915 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00003916#else /* SETPGRP_HAVE_ARG */
Guido van Rossumb6775db1994-08-01 11:34:53 +00003917 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00003918#endif /* SETPGRP_HAVE_ARG */
Guido van Rossum687dd131993-05-17 08:34:16 +00003919 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00003920 Py_INCREF(Py_None);
3921 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00003922}
3923
Guido van Rossumb6775db1994-08-01 11:34:53 +00003924#endif /* HAVE_SETPGRP */
3925
Guido van Rossumad0ee831995-03-01 10:34:45 +00003926#ifdef HAVE_GETPPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003927PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003928"getppid() -> ppid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003929Return the parent's process id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003930
Barry Warsaw53699e91996-12-10 23:23:01 +00003931static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003932posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003933{
Christian Heimes217cfd12007-12-02 14:31:20 +00003934 return PyLong_FromLong((long)getppid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00003935}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003936#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003937
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003938
Fred Drake12c6e2d1999-12-14 21:25:03 +00003939#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003940PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003941"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003942Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00003943
3944static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003945posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00003946{
Neal Norwitze241ce82003-02-17 18:17:05 +00003947 PyObject *result = NULL;
Fred Drakea30680b2000-12-06 21:24:28 +00003948 char *name;
3949 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00003950
Fred Drakea30680b2000-12-06 21:24:28 +00003951 errno = 0;
3952 name = getlogin();
3953 if (name == NULL) {
3954 if (errno)
3955 posix_error();
3956 else
3957 PyErr_SetString(PyExc_OSError,
Fred Drakee63544f2000-12-06 21:45:33 +00003958 "unable to determine login name");
Fred Drakea30680b2000-12-06 21:24:28 +00003959 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00003960 else
Neal Norwitz93c56822007-08-26 07:10:06 +00003961 result = PyUnicode_FromString(name);
Fred Drakea30680b2000-12-06 21:24:28 +00003962 errno = old_errno;
Neal Norwitze241ce82003-02-17 18:17:05 +00003963
Fred Drake12c6e2d1999-12-14 21:25:03 +00003964 return result;
3965}
3966#endif
3967
Guido van Rossumad0ee831995-03-01 10:34:45 +00003968#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003969PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003970"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003971Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003972
Barry Warsaw53699e91996-12-10 23:23:01 +00003973static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003974posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00003975{
Christian Heimes217cfd12007-12-02 14:31:20 +00003976 return PyLong_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00003977}
Guido van Rossumad0ee831995-03-01 10:34:45 +00003978#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00003979
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003980
Guido van Rossumad0ee831995-03-01 10:34:45 +00003981#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003982PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003983"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003984Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003985
Barry Warsaw53699e91996-12-10 23:23:01 +00003986static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003987posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003988{
Christian Heimes292d3512008-02-03 16:51:08 +00003989 pid_t pid;
3990 int sig;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00003991 if (!PyArg_ParseTuple(args, "ii:kill", &pid, &sig))
Guido van Rossum85e3b011991-06-03 12:42:10 +00003992 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00003993#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003994 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
3995 APIRET rc;
3996 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00003997 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003998
3999 } else if (sig == XCPT_SIGNAL_KILLPROC) {
4000 APIRET rc;
4001 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004002 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004003
4004 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00004005 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004006#else
Guido van Rossum85e3b011991-06-03 12:42:10 +00004007 if (kill(pid, sig) == -1)
4008 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004009#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00004010 Py_INCREF(Py_None);
4011 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00004012}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004013#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004014
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004015#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004016PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004017"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004018Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004019
4020static PyObject *
4021posix_killpg(PyObject *self, PyObject *args)
4022{
4023 int pgid, sig;
4024 if (!PyArg_ParseTuple(args, "ii:killpg", &pgid, &sig))
4025 return NULL;
4026 if (killpg(pgid, sig) == -1)
4027 return posix_error();
4028 Py_INCREF(Py_None);
4029 return Py_None;
4030}
4031#endif
4032
Guido van Rossumc0125471996-06-28 18:55:32 +00004033#ifdef HAVE_PLOCK
4034
4035#ifdef HAVE_SYS_LOCK_H
4036#include <sys/lock.h>
4037#endif
4038
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004039PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004040"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004041Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004042
Barry Warsaw53699e91996-12-10 23:23:01 +00004043static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004044posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00004045{
4046 int op;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004047 if (!PyArg_ParseTuple(args, "i:plock", &op))
Guido van Rossumc0125471996-06-28 18:55:32 +00004048 return NULL;
4049 if (plock(op) == -1)
4050 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004051 Py_INCREF(Py_None);
4052 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00004053}
4054#endif
4055
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004056
Guido van Rossum3b066191991-06-04 19:40:25 +00004057
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004058
Guido van Rossumb6775db1994-08-01 11:34:53 +00004059#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004060PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004061"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004062Set the current process's user id.");
4063
Barry Warsaw53699e91996-12-10 23:23:01 +00004064static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004065posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004066{
4067 int uid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004068 if (!PyArg_ParseTuple(args, "i:setuid", &uid))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004069 return NULL;
4070 if (setuid(uid) < 0)
4071 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004072 Py_INCREF(Py_None);
4073 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004074}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004075#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004076
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004077
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004078#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004079PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004080"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004081Set the current process's effective user id.");
4082
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004083static PyObject *
4084posix_seteuid (PyObject *self, PyObject *args)
4085{
4086 int euid;
4087 if (!PyArg_ParseTuple(args, "i", &euid)) {
4088 return NULL;
4089 } else if (seteuid(euid) < 0) {
4090 return posix_error();
4091 } else {
4092 Py_INCREF(Py_None);
4093 return Py_None;
4094 }
4095}
4096#endif /* HAVE_SETEUID */
4097
4098#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004099PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004100"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004101Set the current process's effective group id.");
4102
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004103static PyObject *
4104posix_setegid (PyObject *self, PyObject *args)
4105{
4106 int egid;
4107 if (!PyArg_ParseTuple(args, "i", &egid)) {
4108 return NULL;
4109 } else if (setegid(egid) < 0) {
4110 return posix_error();
4111 } else {
4112 Py_INCREF(Py_None);
4113 return Py_None;
4114 }
4115}
4116#endif /* HAVE_SETEGID */
4117
4118#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004119PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004120"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004121Set the current process's real and effective user ids.");
4122
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004123static PyObject *
4124posix_setreuid (PyObject *self, PyObject *args)
4125{
4126 int ruid, euid;
4127 if (!PyArg_ParseTuple(args, "ii", &ruid, &euid)) {
4128 return NULL;
4129 } else if (setreuid(ruid, euid) < 0) {
4130 return posix_error();
4131 } else {
4132 Py_INCREF(Py_None);
4133 return Py_None;
4134 }
4135}
4136#endif /* HAVE_SETREUID */
4137
4138#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004139PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00004140"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004141Set the current process's real and effective group ids.");
4142
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00004143static PyObject *
4144posix_setregid (PyObject *self, PyObject *args)
4145{
4146 int rgid, egid;
4147 if (!PyArg_ParseTuple(args, "ii", &rgid, &egid)) {
4148 return NULL;
4149 } else if (setregid(rgid, egid) < 0) {
4150 return posix_error();
4151 } else {
4152 Py_INCREF(Py_None);
4153 return Py_None;
4154 }
4155}
4156#endif /* HAVE_SETREGID */
4157
Guido van Rossumb6775db1994-08-01 11:34:53 +00004158#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004159PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004160"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004161Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004162
Barry Warsaw53699e91996-12-10 23:23:01 +00004163static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004164posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004165{
4166 int gid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004167 if (!PyArg_ParseTuple(args, "i:setgid", &gid))
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004168 return NULL;
4169 if (setgid(gid) < 0)
4170 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004171 Py_INCREF(Py_None);
4172 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004173}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004174#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00004175
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004176#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004177PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004178"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004179Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004180
4181static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00004182posix_setgroups(PyObject *self, PyObject *groups)
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004183{
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004184 int i, len;
4185 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00004186
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004187 if (!PySequence_Check(groups)) {
4188 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
4189 return NULL;
4190 }
4191 len = PySequence_Size(groups);
4192 if (len > MAX_GROUPS) {
4193 PyErr_SetString(PyExc_ValueError, "too many groups");
4194 return NULL;
4195 }
4196 for(i = 0; i < len; i++) {
4197 PyObject *elem;
4198 elem = PySequence_GetItem(groups, i);
4199 if (!elem)
4200 return NULL;
Guido van Rossumddefaf32007-01-14 03:31:43 +00004201 if (!PyLong_Check(elem)) {
4202 PyErr_SetString(PyExc_TypeError,
4203 "groups must be integers");
4204 Py_DECREF(elem);
4205 return NULL;
4206 } else {
4207 unsigned long x = PyLong_AsUnsignedLong(elem);
4208 if (PyErr_Occurred()) {
4209 PyErr_SetString(PyExc_TypeError,
4210 "group id too big");
Georg Brandla13c2442005-11-22 19:30:31 +00004211 Py_DECREF(elem);
4212 return NULL;
Georg Brandla13c2442005-11-22 19:30:31 +00004213 }
Georg Brandla13c2442005-11-22 19:30:31 +00004214 grouplist[i] = x;
Guido van Rossumddefaf32007-01-14 03:31:43 +00004215 /* read back the value to see if it fitted in gid_t */
Georg Brandla13c2442005-11-22 19:30:31 +00004216 if (grouplist[i] != x) {
4217 PyErr_SetString(PyExc_TypeError,
4218 "group id too big");
4219 Py_DECREF(elem);
4220 return NULL;
4221 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004222 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00004223 Py_DECREF(elem);
4224 }
4225
4226 if (setgroups(len, grouplist) < 0)
4227 return posix_error();
4228 Py_INCREF(Py_None);
4229 return Py_None;
4230}
4231#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004232
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004233#if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
4234static PyObject *
Christian Heimes292d3512008-02-03 16:51:08 +00004235wait_helper(pid_t pid, int status, struct rusage *ru)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004236{
4237 PyObject *result;
4238 static PyObject *struct_rusage;
4239
4240 if (pid == -1)
4241 return posix_error();
4242
4243 if (struct_rusage == NULL) {
Christian Heimes072c0f12008-01-03 23:01:04 +00004244 PyObject *m = PyImport_ImportModuleNoBlock("resource");
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004245 if (m == NULL)
4246 return NULL;
4247 struct_rusage = PyObject_GetAttrString(m, "struct_rusage");
4248 Py_DECREF(m);
4249 if (struct_rusage == NULL)
4250 return NULL;
4251 }
4252
4253 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
4254 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
4255 if (!result)
4256 return NULL;
4257
4258#ifndef doubletime
4259#define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
4260#endif
4261
4262 PyStructSequence_SET_ITEM(result, 0,
4263 PyFloat_FromDouble(doubletime(ru->ru_utime)));
4264 PyStructSequence_SET_ITEM(result, 1,
4265 PyFloat_FromDouble(doubletime(ru->ru_stime)));
4266#define SET_INT(result, index, value)\
Christian Heimes217cfd12007-12-02 14:31:20 +00004267 PyStructSequence_SET_ITEM(result, index, PyLong_FromLong(value))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004268 SET_INT(result, 2, ru->ru_maxrss);
4269 SET_INT(result, 3, ru->ru_ixrss);
4270 SET_INT(result, 4, ru->ru_idrss);
4271 SET_INT(result, 5, ru->ru_isrss);
4272 SET_INT(result, 6, ru->ru_minflt);
4273 SET_INT(result, 7, ru->ru_majflt);
4274 SET_INT(result, 8, ru->ru_nswap);
4275 SET_INT(result, 9, ru->ru_inblock);
4276 SET_INT(result, 10, ru->ru_oublock);
4277 SET_INT(result, 11, ru->ru_msgsnd);
4278 SET_INT(result, 12, ru->ru_msgrcv);
4279 SET_INT(result, 13, ru->ru_nsignals);
4280 SET_INT(result, 14, ru->ru_nvcsw);
4281 SET_INT(result, 15, ru->ru_nivcsw);
4282#undef SET_INT
4283
4284 if (PyErr_Occurred()) {
4285 Py_DECREF(result);
4286 return NULL;
4287 }
4288
4289 return Py_BuildValue("iiN", pid, status, result);
4290}
4291#endif /* HAVE_WAIT3 || HAVE_WAIT4 */
4292
4293#ifdef HAVE_WAIT3
4294PyDoc_STRVAR(posix_wait3__doc__,
4295"wait3(options) -> (pid, status, rusage)\n\n\
4296Wait for completion of a child process.");
4297
4298static PyObject *
4299posix_wait3(PyObject *self, PyObject *args)
4300{
Christian Heimes292d3512008-02-03 16:51:08 +00004301 pid_t pid;
4302 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004303 struct rusage ru;
4304 WAIT_TYPE status;
4305 WAIT_STATUS_INT(status) = 0;
4306
4307 if (!PyArg_ParseTuple(args, "i:wait3", &options))
4308 return NULL;
4309
4310 Py_BEGIN_ALLOW_THREADS
4311 pid = wait3(&status, options, &ru);
4312 Py_END_ALLOW_THREADS
4313
4314 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
4315}
4316#endif /* HAVE_WAIT3 */
4317
4318#ifdef HAVE_WAIT4
4319PyDoc_STRVAR(posix_wait4__doc__,
4320"wait4(pid, options) -> (pid, status, rusage)\n\n\
4321Wait for completion of a given child process.");
4322
4323static PyObject *
4324posix_wait4(PyObject *self, PyObject *args)
4325{
Christian Heimes292d3512008-02-03 16:51:08 +00004326 pid_t pid;
4327 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004328 struct rusage ru;
4329 WAIT_TYPE status;
4330 WAIT_STATUS_INT(status) = 0;
4331
4332 if (!PyArg_ParseTuple(args, "ii:wait4", &pid, &options))
4333 return NULL;
4334
4335 Py_BEGIN_ALLOW_THREADS
4336 pid = wait4(pid, &status, options, &ru);
4337 Py_END_ALLOW_THREADS
4338
4339 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
4340}
4341#endif /* HAVE_WAIT4 */
4342
Guido van Rossumb6775db1994-08-01 11:34:53 +00004343#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004344PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004345"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004346Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004347
Barry Warsaw53699e91996-12-10 23:23:01 +00004348static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004349posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004350{
Christian Heimes292d3512008-02-03 16:51:08 +00004351 pid_t pid;
4352 int options;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004353 WAIT_TYPE status;
4354 WAIT_STATUS_INT(status) = 0;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004355
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004356 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
Guido van Rossum21803b81992-08-09 12:55:27 +00004357 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004358 Py_BEGIN_ALLOW_THREADS
Guido van Rossume6a3aa61999-02-01 16:15:30 +00004359 pid = waitpid(pid, &status, options);
Barry Warsaw53699e91996-12-10 23:23:01 +00004360 Py_END_ALLOW_THREADS
Guido van Rossum85e3b011991-06-03 12:42:10 +00004361 if (pid == -1)
4362 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004363
4364 return Py_BuildValue("ii", pid, WAIT_STATUS_INT(status));
Guido van Rossum21803b81992-08-09 12:55:27 +00004365}
4366
Tim Petersab034fa2002-02-01 11:27:43 +00004367#elif defined(HAVE_CWAIT)
4368
4369/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004370PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004371"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004372"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00004373
4374static PyObject *
4375posix_waitpid(PyObject *self, PyObject *args)
4376{
Thomas Wouters477c8d52006-05-27 19:21:47 +00004377 Py_intptr_t pid;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004378 int status, options;
Tim Petersab034fa2002-02-01 11:27:43 +00004379
4380 if (!PyArg_ParseTuple(args, "ii:waitpid", &pid, &options))
4381 return NULL;
4382 Py_BEGIN_ALLOW_THREADS
4383 pid = _cwait(&status, pid, options);
4384 Py_END_ALLOW_THREADS
4385 if (pid == -1)
4386 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004387
4388 /* shift the status left a byte so this is more like the POSIX waitpid */
4389 return Py_BuildValue("ii", pid, status << 8);
Tim Petersab034fa2002-02-01 11:27:43 +00004390}
4391#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004392
Guido van Rossumad0ee831995-03-01 10:34:45 +00004393#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004394PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004395"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004396Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004397
Barry Warsaw53699e91996-12-10 23:23:01 +00004398static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004399posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00004400{
Christian Heimes292d3512008-02-03 16:51:08 +00004401 pid_t pid;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004402 WAIT_TYPE status;
4403 WAIT_STATUS_INT(status) = 0;
Neal Norwitze241ce82003-02-17 18:17:05 +00004404
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00004405 Py_BEGIN_ALLOW_THREADS
4406 pid = wait(&status);
Barry Warsaw53699e91996-12-10 23:23:01 +00004407 Py_END_ALLOW_THREADS
Guido van Rossum21803b81992-08-09 12:55:27 +00004408 if (pid == -1)
4409 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004410
4411 return Py_BuildValue("ii", pid, WAIT_STATUS_INT(status));
Guido van Rossum85e3b011991-06-03 12:42:10 +00004412}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004413#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004414
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004415
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004416PyDoc_STRVAR(posix_lstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004417"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004418Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004419
Barry Warsaw53699e91996-12-10 23:23:01 +00004420static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004421posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004422{
Guido van Rossumb6775db1994-08-01 11:34:53 +00004423#ifdef HAVE_LSTAT
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004424 return posix_do_stat(self, args, "et:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004425#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004426#ifdef MS_WINDOWS
Martin v. Löwis14694662006-02-03 12:54:16 +00004427 return posix_do_stat(self, args, "et:lstat", STAT, "U:lstat", win32_wstat);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004428#else
4429 return posix_do_stat(self, args, "et:lstat", STAT, NULL, NULL);
4430#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00004431#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004432}
4433
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004434
Guido van Rossumb6775db1994-08-01 11:34:53 +00004435#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004436PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004437"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004438Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004439
Barry Warsaw53699e91996-12-10 23:23:01 +00004440static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004441posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004442{
Thomas Wouters89f507f2006-12-13 04:49:30 +00004443 PyObject* v;
Guido van Rossumb6775db1994-08-01 11:34:53 +00004444 char buf[MAXPATHLEN];
Guido van Rossumef0a00e1992-01-27 16:51:30 +00004445 char *path;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004446 int n;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004447 int arg_is_unicode = 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004448
4449 if (!PyArg_ParseTuple(args, "et:readlink",
4450 Py_FileSystemDefaultEncoding, &path))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004451 return NULL;
Thomas Wouters89f507f2006-12-13 04:49:30 +00004452 v = PySequence_GetItem(args, 0);
Neal Norwitzcda5c062007-08-12 17:09:36 +00004453 if (v == NULL) {
4454 PyMem_Free(path);
4455 return NULL;
4456 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004457
4458 if (PyUnicode_Check(v)) {
4459 arg_is_unicode = 1;
4460 }
4461 Py_DECREF(v);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004462
Barry Warsaw53699e91996-12-10 23:23:01 +00004463 Py_BEGIN_ALLOW_THREADS
Guido van Rossum50e61dc1992-03-27 17:22:31 +00004464 n = readlink(path, buf, (int) sizeof buf);
Barry Warsaw53699e91996-12-10 23:23:01 +00004465 Py_END_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004466 if (n < 0)
Neal Norwitzfca70052007-08-12 16:56:02 +00004467 return posix_error_with_allocated_filename(path);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004468
Neal Norwitzfca70052007-08-12 16:56:02 +00004469 PyMem_Free(path);
Christian Heimes72b710a2008-05-26 13:28:38 +00004470 v = PyBytes_FromStringAndSize(buf, n);
Thomas Wouters89f507f2006-12-13 04:49:30 +00004471 if (arg_is_unicode) {
4472 PyObject *w;
4473
4474 w = PyUnicode_FromEncodedObject(v,
4475 Py_FileSystemDefaultEncoding,
4476 "strict");
4477 if (w != NULL) {
4478 Py_DECREF(v);
4479 v = w;
4480 }
4481 else {
4482 /* fall back to the original byte string, as
4483 discussed in patch #683592 */
4484 PyErr_Clear();
4485 }
4486 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004487 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004488}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004489#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00004490
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004491
Guido van Rossumb6775db1994-08-01 11:34:53 +00004492#ifdef HAVE_SYMLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004493PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004494"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00004495Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004496
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004497static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004498posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004499{
Thomas Wouters477c8d52006-05-27 19:21:47 +00004500 return posix_2str(args, "etet:symlink", symlink);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004501}
4502#endif /* HAVE_SYMLINK */
4503
4504
4505#ifdef HAVE_TIMES
4506#ifndef HZ
4507#define HZ 60 /* Universal constant :-) */
4508#endif /* HZ */
Tim Peters5aa91602002-01-30 05:46:57 +00004509
Guido van Rossumd48f2521997-12-05 22:19:34 +00004510#if defined(PYCC_VACPP) && defined(PYOS_OS2)
4511static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00004512system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004513{
4514 ULONG value = 0;
4515
4516 Py_BEGIN_ALLOW_THREADS
4517 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
4518 Py_END_ALLOW_THREADS
4519
4520 return value;
4521}
4522
4523static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004524posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004525{
Guido van Rossumd48f2521997-12-05 22:19:34 +00004526 /* Currently Only Uptime is Provided -- Others Later */
4527 return Py_BuildValue("ddddd",
4528 (double)0 /* t.tms_utime / HZ */,
4529 (double)0 /* t.tms_stime / HZ */,
4530 (double)0 /* t.tms_cutime / HZ */,
4531 (double)0 /* t.tms_cstime / HZ */,
4532 (double)system_uptime() / 1000);
4533}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004534#else /* not OS2 */
Barry Warsaw53699e91996-12-10 23:23:01 +00004535static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004536posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00004537{
4538 struct tms t;
4539 clock_t c;
Guido van Rossum22db57e1992-04-05 14:25:30 +00004540 errno = 0;
4541 c = times(&t);
Guido van Rossum687dd131993-05-17 08:34:16 +00004542 if (c == (clock_t) -1)
4543 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004544 return Py_BuildValue("ddddd",
Barry Warsaw43d68b81996-12-19 22:10:44 +00004545 (double)t.tms_utime / HZ,
4546 (double)t.tms_stime / HZ,
4547 (double)t.tms_cutime / HZ,
4548 (double)t.tms_cstime / HZ,
4549 (double)c / HZ);
Guido van Rossum22db57e1992-04-05 14:25:30 +00004550}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004551#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00004552#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004553
4554
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004555#ifdef MS_WINDOWS
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004556#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00004557static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004558posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004559{
4560 FILETIME create, exit, kernel, user;
4561 HANDLE hProc;
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004562 hProc = GetCurrentProcess();
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004563 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
4564 /* The fields of a FILETIME structure are the hi and lo part
4565 of a 64-bit value expressed in 100 nanosecond units.
4566 1e7 is one second in such units; 1e-7 the inverse.
4567 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
4568 */
Barry Warsaw53699e91996-12-10 23:23:01 +00004569 return Py_BuildValue(
4570 "ddddd",
Guido van Rossumb8c3cbd1999-02-16 14:37:28 +00004571 (double)(user.dwHighDateTime*429.4967296 +
4572 user.dwLowDateTime*1e-7),
Christian Heimes68f5fbe2008-02-14 08:27:37 +00004573 (double)(kernel.dwHighDateTime*429.4967296 +
4574 kernel.dwLowDateTime*1e-7),
Barry Warsaw53699e91996-12-10 23:23:01 +00004575 (double)0,
4576 (double)0,
4577 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00004578}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004579#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004580
4581#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004582PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004583"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004584Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00004585#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00004586
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004587
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004588#ifdef HAVE_GETSID
4589PyDoc_STRVAR(posix_getsid__doc__,
4590"getsid(pid) -> sid\n\n\
4591Call the system call getsid().");
4592
4593static PyObject *
4594posix_getsid(PyObject *self, PyObject *args)
4595{
Christian Heimes292d3512008-02-03 16:51:08 +00004596 pid_t pid;
4597 int sid;
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004598 if (!PyArg_ParseTuple(args, "i:getsid", &pid))
4599 return NULL;
4600 sid = getsid(pid);
4601 if (sid < 0)
4602 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004603 return PyLong_FromLong((long)sid);
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00004604}
4605#endif /* HAVE_GETSID */
4606
4607
Guido van Rossumb6775db1994-08-01 11:34:53 +00004608#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004609PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004610"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004611Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004612
Barry Warsaw53699e91996-12-10 23:23:01 +00004613static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004614posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004615{
Guido van Rossum687dd131993-05-17 08:34:16 +00004616 if (setsid() < 0)
4617 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004618 Py_INCREF(Py_None);
4619 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004620}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004621#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00004622
Guido van Rossumb6775db1994-08-01 11:34:53 +00004623#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004624PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004625"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004626Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004627
Barry Warsaw53699e91996-12-10 23:23:01 +00004628static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004629posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004630{
Christian Heimes292d3512008-02-03 16:51:08 +00004631 pid_t pid;
4632 int pgrp;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004633 if (!PyArg_ParseTuple(args, "ii:setpgid", &pid, &pgrp))
Guido van Rossumc2670a01992-09-13 20:07:29 +00004634 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004635 if (setpgid(pid, pgrp) < 0)
4636 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004637 Py_INCREF(Py_None);
4638 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004639}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004640#endif /* HAVE_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00004641
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004642
Guido van Rossumb6775db1994-08-01 11:34:53 +00004643#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004644PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004645"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004646Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004647
Barry Warsaw53699e91996-12-10 23:23:01 +00004648static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004649posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00004650{
Christian Heimes15ebc882008-02-04 18:48:49 +00004651 int fd;
4652 pid_t pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004653 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
Guido van Rossum7066dd71992-09-17 17:54:56 +00004654 return NULL;
4655 pgid = tcgetpgrp(fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004656 if (pgid < 0)
4657 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004658 return PyLong_FromLong((long)pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00004659}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004660#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00004661
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004662
Guido van Rossumb6775db1994-08-01 11:34:53 +00004663#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004664PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004665"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004666Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004667
Barry Warsaw53699e91996-12-10 23:23:01 +00004668static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004669posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00004670{
4671 int fd, pgid;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004672 if (!PyArg_ParseTuple(args, "ii:tcsetpgrp", &fd, &pgid))
Guido van Rossum7066dd71992-09-17 17:54:56 +00004673 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004674 if (tcsetpgrp(fd, pgid) < 0)
4675 return posix_error();
Barry Warsaw43d68b81996-12-19 22:10:44 +00004676 Py_INCREF(Py_None);
Barry Warsaw53699e91996-12-10 23:23:01 +00004677 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00004678}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004679#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00004680
Guido van Rossum687dd131993-05-17 08:34:16 +00004681/* Functions acting on file descriptors */
4682
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004683PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004684"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004685Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004686
Barry Warsaw53699e91996-12-10 23:23:01 +00004687static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004688posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004689{
Mark Hammondef8b6542001-05-13 08:04:26 +00004690 char *file = NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00004691 int flag;
4692 int mode = 0777;
4693 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004694
4695#ifdef MS_WINDOWS
4696 if (unicode_file_names()) {
4697 PyUnicodeObject *po;
4698 if (PyArg_ParseTuple(args, "Ui|i:mkdir", &po, &flag, &mode)) {
4699 Py_BEGIN_ALLOW_THREADS
4700 /* PyUnicode_AS_UNICODE OK without thread
4701 lock as it is a simple dereference. */
4702 fd = _wopen(PyUnicode_AS_UNICODE(po), flag, mode);
4703 Py_END_ALLOW_THREADS
4704 if (fd < 0)
4705 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004706 return PyLong_FromLong((long)fd);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00004707 }
4708 /* Drop the argument parsing error as narrow strings
4709 are also valid. */
4710 PyErr_Clear();
4711 }
4712#endif
4713
Tim Peters5aa91602002-01-30 05:46:57 +00004714 if (!PyArg_ParseTuple(args, "eti|i",
Mark Hammondef8b6542001-05-13 08:04:26 +00004715 Py_FileSystemDefaultEncoding, &file,
4716 &flag, &mode))
Barry Warsaw43d68b81996-12-19 22:10:44 +00004717 return NULL;
4718
Barry Warsaw53699e91996-12-10 23:23:01 +00004719 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004720 fd = open(file, flag, mode);
Barry Warsaw53699e91996-12-10 23:23:01 +00004721 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004722 if (fd < 0)
Mark Hammondef8b6542001-05-13 08:04:26 +00004723 return posix_error_with_allocated_filename(file);
4724 PyMem_Free(file);
Christian Heimes217cfd12007-12-02 14:31:20 +00004725 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004726}
4727
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004728
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004729PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004730"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004731Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004732
Barry Warsaw53699e91996-12-10 23:23:01 +00004733static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004734posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004735{
4736 int fd, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004737 if (!PyArg_ParseTuple(args, "i:close", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00004738 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004739 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004740 res = close(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00004741 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004742 if (res < 0)
4743 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004744 Py_INCREF(Py_None);
4745 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00004746}
4747
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004748
Christian Heimesfdab48e2008-01-20 09:06:41 +00004749PyDoc_STRVAR(posix_closerange__doc__,
4750"closerange(fd_low, fd_high)\n\n\
4751Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
4752
4753static PyObject *
4754posix_closerange(PyObject *self, PyObject *args)
4755{
4756 int fd_from, fd_to, i;
4757 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
4758 return NULL;
4759 Py_BEGIN_ALLOW_THREADS
4760 for (i = fd_from; i < fd_to; i++)
4761 close(i);
4762 Py_END_ALLOW_THREADS
4763 Py_RETURN_NONE;
4764}
4765
4766
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004767PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004768"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004769Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004770
Barry Warsaw53699e91996-12-10 23:23:01 +00004771static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004772posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004773{
4774 int fd;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004775 if (!PyArg_ParseTuple(args, "i:dup", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00004776 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004777 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004778 fd = dup(fd);
Barry Warsaw53699e91996-12-10 23:23:01 +00004779 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004780 if (fd < 0)
4781 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004782 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00004783}
4784
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004785
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004786PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00004787"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004788Duplicate file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004789
Barry Warsaw53699e91996-12-10 23:23:01 +00004790static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004791posix_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004792{
4793 int fd, fd2, res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004794 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
Guido van Rossum687dd131993-05-17 08:34:16 +00004795 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004796 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004797 res = dup2(fd, fd2);
Barry Warsaw53699e91996-12-10 23:23:01 +00004798 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004799 if (res < 0)
4800 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004801 Py_INCREF(Py_None);
4802 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00004803}
4804
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004805
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004806PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004807"lseek(fd, pos, how) -> newpos\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004808Set the current position of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004809
Barry Warsaw53699e91996-12-10 23:23:01 +00004810static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004811posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004812{
4813 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004814#if defined(MS_WIN64) || defined(MS_WINDOWS)
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00004815 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00004816#else
Guido van Rossum94f6f721999-01-06 18:42:14 +00004817 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00004818#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00004819 PyObject *posobj;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004820 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Guido van Rossum687dd131993-05-17 08:34:16 +00004821 return NULL;
4822#ifdef SEEK_SET
4823 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
4824 switch (how) {
4825 case 0: how = SEEK_SET; break;
4826 case 1: how = SEEK_CUR; break;
4827 case 2: how = SEEK_END; break;
4828 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004829#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00004830
4831#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00004832 pos = PyLong_AsLong(posobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00004833#else
4834 pos = PyLong_Check(posobj) ?
Christian Heimes217cfd12007-12-02 14:31:20 +00004835 PyLong_AsLongLong(posobj) : PyLong_AsLong(posobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00004836#endif
4837 if (PyErr_Occurred())
4838 return NULL;
4839
Barry Warsaw53699e91996-12-10 23:23:01 +00004840 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004841#if defined(MS_WIN64) || defined(MS_WINDOWS)
Fred Drake699f3522000-06-29 21:12:41 +00004842 res = _lseeki64(fd, pos, how);
4843#else
Guido van Rossum687dd131993-05-17 08:34:16 +00004844 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00004845#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00004846 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004847 if (res < 0)
4848 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00004849
4850#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00004851 return PyLong_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00004852#else
4853 return PyLong_FromLongLong(res);
4854#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00004855}
4856
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004857
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004858PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004859"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004860Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004861
Barry Warsaw53699e91996-12-10 23:23:01 +00004862static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004863posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004864{
Guido van Rossum572dbf82007-04-27 23:53:51 +00004865 int fd, size;
4866 Py_ssize_t n;
Barry Warsaw53699e91996-12-10 23:23:01 +00004867 PyObject *buffer;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004868 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
Guido van Rossum687dd131993-05-17 08:34:16 +00004869 return NULL;
Michael W. Hudson9867ced2005-01-31 17:01:59 +00004870 if (size < 0) {
4871 errno = EINVAL;
4872 return posix_error();
4873 }
Christian Heimes72b710a2008-05-26 13:28:38 +00004874 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
Guido van Rossum687dd131993-05-17 08:34:16 +00004875 if (buffer == NULL)
4876 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004877 Py_BEGIN_ALLOW_THREADS
Christian Heimes72b710a2008-05-26 13:28:38 +00004878 n = read(fd, PyBytes_AS_STRING(buffer), size);
Barry Warsaw53699e91996-12-10 23:23:01 +00004879 Py_END_ALLOW_THREADS
Guido van Rossum8bac5461996-06-11 18:38:48 +00004880 if (n < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00004881 Py_DECREF(buffer);
Guido van Rossum687dd131993-05-17 08:34:16 +00004882 return posix_error();
4883 }
Guido van Rossum8bac5461996-06-11 18:38:48 +00004884 if (n != size)
Christian Heimes72b710a2008-05-26 13:28:38 +00004885 _PyBytes_Resize(&buffer, n);
Guido van Rossum687dd131993-05-17 08:34:16 +00004886 return buffer;
4887}
4888
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004889
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004890PyDoc_STRVAR(posix_write__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004891"write(fd, string) -> byteswritten\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004892Write a string to a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004893
Barry Warsaw53699e91996-12-10 23:23:01 +00004894static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004895posix_write(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004896{
Martin v. Löwis423be952008-08-13 15:53:07 +00004897 Py_buffer pbuf;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00004898 int fd;
4899 Py_ssize_t size;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00004900
Martin v. Löwis423be952008-08-13 15:53:07 +00004901 if (!PyArg_ParseTuple(args, "is*:write", &fd, &pbuf))
Guido van Rossum687dd131993-05-17 08:34:16 +00004902 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00004903 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis423be952008-08-13 15:53:07 +00004904 size = write(fd, pbuf.buf, (size_t)pbuf.len);
Barry Warsaw53699e91996-12-10 23:23:01 +00004905 Py_END_ALLOW_THREADS
Martin v. Löwis423be952008-08-13 15:53:07 +00004906 PyBuffer_Release(&pbuf);
Guido van Rossum687dd131993-05-17 08:34:16 +00004907 if (size < 0)
4908 return posix_error();
Christian Heimes217cfd12007-12-02 14:31:20 +00004909 return PyLong_FromSsize_t(size);
Guido van Rossum687dd131993-05-17 08:34:16 +00004910}
4911
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004912
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004913PyDoc_STRVAR(posix_fstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004914"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004915Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004916
Barry Warsaw53699e91996-12-10 23:23:01 +00004917static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004918posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00004919{
4920 int fd;
Fred Drake699f3522000-06-29 21:12:41 +00004921 STRUCT_STAT st;
Guido van Rossum687dd131993-05-17 08:34:16 +00004922 int res;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00004923 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
Guido van Rossum687dd131993-05-17 08:34:16 +00004924 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00004925#ifdef __VMS
4926 /* on OpenVMS we must ensure that all bytes are written to the file */
4927 fsync(fd);
4928#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00004929 Py_BEGIN_ALLOW_THREADS
Fred Drake699f3522000-06-29 21:12:41 +00004930 res = FSTAT(fd, &st);
Barry Warsaw53699e91996-12-10 23:23:01 +00004931 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00004932 if (res != 0) {
4933#ifdef MS_WINDOWS
4934 return win32_error("fstat", NULL);
4935#else
Guido van Rossum687dd131993-05-17 08:34:16 +00004936 return posix_error();
Martin v. Löwis14694662006-02-03 12:54:16 +00004937#endif
4938 }
Tim Peters5aa91602002-01-30 05:46:57 +00004939
Martin v. Löwis14694662006-02-03 12:54:16 +00004940 return _pystat_fromstructstat(&st);
Guido van Rossum687dd131993-05-17 08:34:16 +00004941}
4942
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004943PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004944"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00004945Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004946connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00004947
4948static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00004949posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00004950{
4951 int fd;
4952 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
4953 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00004954 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00004955}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004956
Guido van Rossuma4916fa1996-05-23 22:58:55 +00004957#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004958PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004959"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004960Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004961
Barry Warsaw53699e91996-12-10 23:23:01 +00004962static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004963posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00004964{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004965#if defined(PYOS_OS2)
4966 HFILE read, write;
4967 APIRET rc;
4968
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004969 Py_BEGIN_ALLOW_THREADS
4970 rc = DosCreatePipe( &read, &write, 4096);
4971 Py_END_ALLOW_THREADS
4972 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004973 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004974
4975 return Py_BuildValue("(ii)", read, write);
4976#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004977#if !defined(MS_WINDOWS)
Guido van Rossum687dd131993-05-17 08:34:16 +00004978 int fds[2];
4979 int res;
Barry Warsaw53699e91996-12-10 23:23:01 +00004980 Py_BEGIN_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004981 res = pipe(fds);
Barry Warsaw53699e91996-12-10 23:23:01 +00004982 Py_END_ALLOW_THREADS
Guido van Rossum687dd131993-05-17 08:34:16 +00004983 if (res != 0)
4984 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00004985 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004986#else /* MS_WINDOWS */
Guido van Rossum794d8131994-08-23 13:48:48 +00004987 HANDLE read, write;
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00004988 int read_fd, write_fd;
Guido van Rossum794d8131994-08-23 13:48:48 +00004989 BOOL ok;
Barry Warsaw53699e91996-12-10 23:23:01 +00004990 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00004991 ok = CreatePipe(&read, &write, NULL, 0);
Barry Warsaw53699e91996-12-10 23:23:01 +00004992 Py_END_ALLOW_THREADS
Guido van Rossum794d8131994-08-23 13:48:48 +00004993 if (!ok)
Fredrik Lundhffb9c772000-07-09 14:49:51 +00004994 return win32_error("CreatePipe", NULL);
Tim Peters79248aa2001-08-29 21:37:10 +00004995 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
4996 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
Guido van Rossumb3f9f4b1998-06-12 15:05:15 +00004997 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004998#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004999#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00005000}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005001#endif /* HAVE_PIPE */
5002
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005003
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005004#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005005PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005006"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005007Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005008
Barry Warsaw53699e91996-12-10 23:23:01 +00005009static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005010posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005011{
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005012 char *filename;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005013 int mode = 0666;
5014 int res;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005015 if (!PyArg_ParseTuple(args, "s|i:mkfifo", &filename, &mode))
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005016 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00005017 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005018 res = mkfifo(filename, mode);
5019 Py_END_ALLOW_THREADS
5020 if (res < 0)
5021 return posix_error();
5022 Py_INCREF(Py_None);
5023 return Py_None;
5024}
5025#endif
5026
5027
Neal Norwitz11690112002-07-30 01:08:28 +00005028#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005029PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00005030"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005031Create a filesystem node (file, device special file or named pipe)\n\
5032named filename. mode specifies both the permissions to use and the\n\
5033type of node to be created, being combined (bitwise OR) with one of\n\
5034S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005035device defines the newly created device special file (probably using\n\
5036os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005037
5038
5039static PyObject *
5040posix_mknod(PyObject *self, PyObject *args)
5041{
5042 char *filename;
5043 int mode = 0600;
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005044 int device = 0;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005045 int res;
Martin v. Löwisd631ebe2002-11-02 17:42:33 +00005046 if (!PyArg_ParseTuple(args, "s|ii:mknod", &filename, &mode, &device))
Martin v. Löwis06a83e92002-04-14 10:19:44 +00005047 return NULL;
5048 Py_BEGIN_ALLOW_THREADS
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005049 res = mknod(filename, mode, device);
Barry Warsaw53699e91996-12-10 23:23:01 +00005050 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005051 if (res < 0)
5052 return posix_error();
Barry Warsaw53699e91996-12-10 23:23:01 +00005053 Py_INCREF(Py_None);
5054 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005055}
5056#endif
5057
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005058#ifdef HAVE_DEVICE_MACROS
5059PyDoc_STRVAR(posix_major__doc__,
5060"major(device) -> major number\n\
5061Extracts a device major number from a raw device number.");
5062
5063static PyObject *
5064posix_major(PyObject *self, PyObject *args)
5065{
5066 int device;
5067 if (!PyArg_ParseTuple(args, "i:major", &device))
5068 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005069 return PyLong_FromLong((long)major(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005070}
5071
5072PyDoc_STRVAR(posix_minor__doc__,
5073"minor(device) -> minor number\n\
5074Extracts a device minor number from a raw device number.");
5075
5076static PyObject *
5077posix_minor(PyObject *self, PyObject *args)
5078{
5079 int device;
5080 if (!PyArg_ParseTuple(args, "i:minor", &device))
5081 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005082 return PyLong_FromLong((long)minor(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005083}
5084
5085PyDoc_STRVAR(posix_makedev__doc__,
5086"makedev(major, minor) -> device number\n\
5087Composes a raw device number from the major and minor device numbers.");
5088
5089static PyObject *
5090posix_makedev(PyObject *self, PyObject *args)
5091{
5092 int major, minor;
5093 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
5094 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005095 return PyLong_FromLong((long)makedev(major, minor));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00005096}
5097#endif /* device macros */
5098
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005099
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005100#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005101PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005102"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005103Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005104
Barry Warsaw53699e91996-12-10 23:23:01 +00005105static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005106posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005107{
5108 int fd;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005109 off_t length;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005110 int res;
Guido van Rossum94f6f721999-01-06 18:42:14 +00005111 PyObject *lenobj;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005112
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005113 if (!PyArg_ParseTuple(args, "iO:ftruncate", &fd, &lenobj))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005114 return NULL;
5115
5116#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005117 length = PyLong_AsLong(lenobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005118#else
5119 length = PyLong_Check(lenobj) ?
Christian Heimes217cfd12007-12-02 14:31:20 +00005120 PyLong_AsLongLong(lenobj) : PyLong_AsLong(lenobj);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005121#endif
5122 if (PyErr_Occurred())
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005123 return NULL;
5124
Barry Warsaw53699e91996-12-10 23:23:01 +00005125 Py_BEGIN_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005126 res = ftruncate(fd, length);
Barry Warsaw53699e91996-12-10 23:23:01 +00005127 Py_END_ALLOW_THREADS
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005128 if (res < 0) {
Barry Warsaw53699e91996-12-10 23:23:01 +00005129 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005130 return NULL;
5131 }
Barry Warsaw53699e91996-12-10 23:23:01 +00005132 Py_INCREF(Py_None);
5133 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00005134}
5135#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00005136
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005137#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005138PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005139"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005140Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005141
Fred Drake762e2061999-08-26 17:23:54 +00005142/* Save putenv() parameters as values here, so we can collect them when they
5143 * get re-set with another call for the same key. */
5144static PyObject *posix_putenv_garbage;
5145
Tim Peters5aa91602002-01-30 05:46:57 +00005146static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005147posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005148{
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005149#ifdef MS_WINDOWS
5150 wchar_t *s1, *s2;
5151 wchar_t *newenv;
5152#else
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005153 char *s1, *s2;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005154 char *newenv;
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005155#endif
Fred Drake762e2061999-08-26 17:23:54 +00005156 PyObject *newstr;
Tim Petersc8996f52001-12-03 20:41:00 +00005157 size_t len;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005158
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005159 if (!PyArg_ParseTuple(args,
5160#ifdef MS_WINDOWS
5161 "uu:putenv",
5162#else
5163 "ss:putenv",
5164#endif
5165 &s1, &s2))
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005166 return NULL;
Guido van Rossumd48f2521997-12-05 22:19:34 +00005167
5168#if defined(PYOS_OS2)
5169 if (stricmp(s1, "BEGINLIBPATH") == 0) {
5170 APIRET rc;
5171
Guido van Rossumd48f2521997-12-05 22:19:34 +00005172 rc = DosSetExtLIBPATH(s2, BEGIN_LIBPATH);
5173 if (rc != NO_ERROR)
5174 return os2_error(rc);
5175
5176 } else if (stricmp(s1, "ENDLIBPATH") == 0) {
5177 APIRET rc;
5178
Guido van Rossumd48f2521997-12-05 22:19:34 +00005179 rc = DosSetExtLIBPATH(s2, END_LIBPATH);
5180 if (rc != NO_ERROR)
5181 return os2_error(rc);
5182 } else {
5183#endif
Fred Drake762e2061999-08-26 17:23:54 +00005184 /* XXX This can leak memory -- not easy to fix :-( */
Tim Petersc8996f52001-12-03 20:41:00 +00005185 /* len includes space for a trailing \0; the size arg to
Christian Heimes72b710a2008-05-26 13:28:38 +00005186 PyBytes_FromStringAndSize does not count that */
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005187#ifdef MS_WINDOWS
5188 len = wcslen(s1) + wcslen(s2) + 2;
5189 newstr = PyUnicode_FromUnicode(NULL, (int)len - 1);
5190#else
5191 len = strlen(s1) + strlen(s2) + 2;
Christian Heimes72b710a2008-05-26 13:28:38 +00005192 newstr = PyBytes_FromStringAndSize(NULL, (int)len - 1);
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005193#endif
Fred Drake762e2061999-08-26 17:23:54 +00005194 if (newstr == NULL)
5195 return PyErr_NoMemory();
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005196#ifdef MS_WINDOWS
5197 newenv = PyUnicode_AsUnicode(newstr);
5198 _snwprintf(newenv, len, L"%s=%s", s1, s2);
5199 if (_wputenv(newenv)) {
5200 Py_DECREF(newstr);
5201 posix_error();
5202 return NULL;
5203 }
5204#else
Christian Heimes72b710a2008-05-26 13:28:38 +00005205 newenv = PyBytes_AS_STRING(newstr);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005206 PyOS_snprintf(newenv, len, "%s=%s", s1, s2);
5207 if (putenv(newenv)) {
Neal Norwitz4adc9ab2003-02-10 03:10:43 +00005208 Py_DECREF(newstr);
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005209 posix_error();
5210 return NULL;
5211 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00005212#endif
Fred Drake762e2061999-08-26 17:23:54 +00005213 /* Install the first arg and newstr in posix_putenv_garbage;
5214 * this will cause previous value to be collected. This has to
5215 * happen after the real putenv() call because the old value
5216 * was still accessible until then. */
5217 if (PyDict_SetItem(posix_putenv_garbage,
5218 PyTuple_GET_ITEM(args, 0), newstr)) {
5219 /* really not much we can do; just leak */
5220 PyErr_Clear();
5221 }
5222 else {
5223 Py_DECREF(newstr);
5224 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00005225
5226#if defined(PYOS_OS2)
5227 }
5228#endif
Barry Warsaw53699e91996-12-10 23:23:01 +00005229 Py_INCREF(Py_None);
5230 return Py_None;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005231}
Guido van Rossumb6a47161997-09-15 22:54:34 +00005232#endif /* putenv */
5233
Guido van Rossumc524d952001-10-19 01:31:59 +00005234#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005235PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005236"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005237Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00005238
5239static PyObject *
5240posix_unsetenv(PyObject *self, PyObject *args)
5241{
5242 char *s1;
5243
5244 if (!PyArg_ParseTuple(args, "s:unsetenv", &s1))
5245 return NULL;
5246
5247 unsetenv(s1);
5248
5249 /* Remove the key from posix_putenv_garbage;
5250 * this will cause it to be collected. This has to
Tim Peters5aa91602002-01-30 05:46:57 +00005251 * happen after the real unsetenv() call because the
Guido van Rossumc524d952001-10-19 01:31:59 +00005252 * old value was still accessible until then.
5253 */
5254 if (PyDict_DelItem(posix_putenv_garbage,
5255 PyTuple_GET_ITEM(args, 0))) {
5256 /* really not much we can do; just leak */
5257 PyErr_Clear();
5258 }
5259
5260 Py_INCREF(Py_None);
5261 return Py_None;
5262}
5263#endif /* unsetenv */
5264
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005265PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005266"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005267Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005268
Guido van Rossumf68d8e52001-04-14 17:55:09 +00005269static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005270posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00005271{
5272 int code;
5273 char *message;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005274 if (!PyArg_ParseTuple(args, "i:strerror", &code))
Guido van Rossumb6a47161997-09-15 22:54:34 +00005275 return NULL;
5276 message = strerror(code);
5277 if (message == NULL) {
5278 PyErr_SetString(PyExc_ValueError,
Fred Drake661ea262000-10-24 19:57:45 +00005279 "strerror() argument out of range");
Guido van Rossumb6a47161997-09-15 22:54:34 +00005280 return NULL;
5281 }
Neal Norwitz93c56822007-08-26 07:10:06 +00005282 return PyUnicode_FromString(message);
Guido van Rossumb6a47161997-09-15 22:54:34 +00005283}
Guido van Rossumb6a47161997-09-15 22:54:34 +00005284
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00005285
Guido van Rossumc9641791998-08-04 15:26:23 +00005286#ifdef HAVE_SYS_WAIT_H
5287
Fred Drake106c1a02002-04-23 15:58:02 +00005288#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005289PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005290"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005291Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00005292
5293static PyObject *
5294posix_WCOREDUMP(PyObject *self, PyObject *args)
5295{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005296 WAIT_TYPE status;
5297 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00005298
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005299 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00005300 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00005301
5302 return PyBool_FromLong(WCOREDUMP(status));
Fred Drake106c1a02002-04-23 15:58:02 +00005303}
5304#endif /* WCOREDUMP */
5305
5306#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005307PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005308"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00005309Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005310job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00005311
5312static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005313posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00005314{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005315 WAIT_TYPE status;
5316 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00005317
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005318 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
Fred Drake106c1a02002-04-23 15:58:02 +00005319 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00005320
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00005321 return PyBool_FromLong(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00005322}
5323#endif /* WIFCONTINUED */
5324
Guido van Rossumc9641791998-08-04 15:26:23 +00005325#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005326PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005327"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005328Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005329
5330static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005331posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005332{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005333 WAIT_TYPE status;
5334 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005335
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005336 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005337 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005338
Fred Drake106c1a02002-04-23 15:58:02 +00005339 return PyBool_FromLong(WIFSTOPPED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005340}
5341#endif /* WIFSTOPPED */
5342
5343#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005344PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005345"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005346Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005347
5348static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005349posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005350{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005351 WAIT_TYPE status;
5352 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005353
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005354 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005355 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005356
Fred Drake106c1a02002-04-23 15:58:02 +00005357 return PyBool_FromLong(WIFSIGNALED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005358}
5359#endif /* WIFSIGNALED */
5360
5361#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005362PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005363"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005364Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005365system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005366
5367static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005368posix_WIFEXITED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005369{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005370 WAIT_TYPE status;
5371 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005372
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005373 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005374 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005375
Fred Drake106c1a02002-04-23 15:58:02 +00005376 return PyBool_FromLong(WIFEXITED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00005377}
5378#endif /* WIFEXITED */
5379
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005380#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005381PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005382"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005383Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005384
5385static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005386posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005387{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005388 WAIT_TYPE status;
5389 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005390
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005391 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005392 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005393
Guido van Rossumc9641791998-08-04 15:26:23 +00005394 return Py_BuildValue("i", WEXITSTATUS(status));
5395}
5396#endif /* WEXITSTATUS */
5397
5398#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005399PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005400"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00005401Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005402value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005403
5404static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005405posix_WTERMSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005406{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005407 WAIT_TYPE status;
5408 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005409
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005410 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005411 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005412
Guido van Rossumc9641791998-08-04 15:26:23 +00005413 return Py_BuildValue("i", WTERMSIG(status));
5414}
5415#endif /* WTERMSIG */
5416
5417#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005418PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005419"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005420Return the signal that stopped the process that provided\n\
5421the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00005422
5423static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005424posix_WSTOPSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00005425{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005426 WAIT_TYPE status;
5427 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00005428
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005429 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &WAIT_STATUS_INT(status)))
Guido van Rossumc9641791998-08-04 15:26:23 +00005430 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00005431
Guido van Rossumc9641791998-08-04 15:26:23 +00005432 return Py_BuildValue("i", WSTOPSIG(status));
5433}
5434#endif /* WSTOPSIG */
5435
5436#endif /* HAVE_SYS_WAIT_H */
5437
5438
Thomas Wouters477c8d52006-05-27 19:21:47 +00005439#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossumd5753e11999-10-19 13:29:23 +00005440#ifdef _SCO_DS
5441/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
5442 needed definitions in sys/statvfs.h */
5443#define _SVID3
5444#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00005445#include <sys/statvfs.h>
5446
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005447static PyObject*
5448_pystatvfs_fromstructstatvfs(struct statvfs st) {
5449 PyObject *v = PyStructSequence_New(&StatVFSResultType);
5450 if (v == NULL)
5451 return NULL;
5452
5453#if !defined(HAVE_LARGEFILE_SUPPORT)
Christian Heimes217cfd12007-12-02 14:31:20 +00005454 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
5455 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
5456 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long) st.f_blocks));
5457 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long) st.f_bfree));
5458 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long) st.f_bavail));
5459 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long) st.f_files));
5460 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong((long) st.f_ffree));
5461 PyStructSequence_SET_ITEM(v, 7, PyLong_FromLong((long) st.f_favail));
5462 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
5463 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005464#else
Christian Heimes217cfd12007-12-02 14:31:20 +00005465 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
5466 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
Tim Peters5aa91602002-01-30 05:46:57 +00005467 PyStructSequence_SET_ITEM(v, 2,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005468 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
Tim Peters5aa91602002-01-30 05:46:57 +00005469 PyStructSequence_SET_ITEM(v, 3,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005470 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005471 PyStructSequence_SET_ITEM(v, 4,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005472 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
Tim Peters5aa91602002-01-30 05:46:57 +00005473 PyStructSequence_SET_ITEM(v, 5,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005474 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
Tim Peters5aa91602002-01-30 05:46:57 +00005475 PyStructSequence_SET_ITEM(v, 6,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005476 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
Tim Peters5aa91602002-01-30 05:46:57 +00005477 PyStructSequence_SET_ITEM(v, 7,
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00005478 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
Christian Heimes217cfd12007-12-02 14:31:20 +00005479 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
5480 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005481#endif
5482
5483 return v;
5484}
5485
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005486PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005487"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005488Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005489
5490static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005491posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005492{
5493 int fd, res;
5494 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005495
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005496 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005497 return NULL;
5498 Py_BEGIN_ALLOW_THREADS
5499 res = fstatvfs(fd, &st);
5500 Py_END_ALLOW_THREADS
5501 if (res != 0)
5502 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005503
5504 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005505}
Thomas Wouters477c8d52006-05-27 19:21:47 +00005506#endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00005507
5508
Thomas Wouters477c8d52006-05-27 19:21:47 +00005509#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005510#include <sys/statvfs.h>
5511
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005512PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005513"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005514Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00005515
5516static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005517posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00005518{
5519 char *path;
5520 int res;
5521 struct statvfs st;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00005522 if (!PyArg_ParseTuple(args, "s:statvfs", &path))
Guido van Rossum94f6f721999-01-06 18:42:14 +00005523 return NULL;
5524 Py_BEGIN_ALLOW_THREADS
5525 res = statvfs(path, &st);
5526 Py_END_ALLOW_THREADS
5527 if (res != 0)
5528 return posix_error_with_filename(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005529
5530 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00005531}
5532#endif /* HAVE_STATVFS */
5533
Fred Drakec9680921999-12-13 16:37:25 +00005534/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
5535 * It maps strings representing configuration variable names to
5536 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00005537 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00005538 * rarely-used constants. There are three separate tables that use
5539 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00005540 *
5541 * This code is always included, even if none of the interfaces that
5542 * need it are included. The #if hackery needed to avoid it would be
5543 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00005544 */
5545struct constdef {
5546 char *name;
5547 long value;
5548};
5549
Fred Drake12c6e2d1999-12-14 21:25:03 +00005550static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005551conv_confname(PyObject *arg, int *valuep, struct constdef *table,
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005552 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00005553{
Christian Heimes217cfd12007-12-02 14:31:20 +00005554 if (PyLong_Check(arg)) {
5555 *valuep = PyLong_AS_LONG(arg);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005556 return 1;
5557 }
Guido van Rossumbce56a62007-05-10 18:04:33 +00005558 else {
Fred Drake12c6e2d1999-12-14 21:25:03 +00005559 /* look up the value in the table using a binary search */
Fred Drake699f3522000-06-29 21:12:41 +00005560 size_t lo = 0;
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005561 size_t mid;
Fred Drake699f3522000-06-29 21:12:41 +00005562 size_t hi = tablesize;
5563 int cmp;
Guido van Rossumbce56a62007-05-10 18:04:33 +00005564 const char *confname;
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005565 if (!PyUnicode_Check(arg)) {
Guido van Rossumbce56a62007-05-10 18:04:33 +00005566 PyErr_SetString(PyExc_TypeError,
5567 "configuration names must be strings or integers");
5568 return 0;
5569 }
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00005570 confname = _PyUnicode_AsString(arg);
Guido van Rossum7d5baac2007-08-27 23:24:46 +00005571 if (confname == NULL)
5572 return 0;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005573 while (lo < hi) {
5574 mid = (lo + hi) / 2;
5575 cmp = strcmp(confname, table[mid].name);
5576 if (cmp < 0)
5577 hi = mid;
5578 else if (cmp > 0)
5579 lo = mid + 1;
5580 else {
5581 *valuep = table[mid].value;
5582 return 1;
5583 }
5584 }
5585 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
Guido van Rossumbce56a62007-05-10 18:04:33 +00005586 return 0;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005587 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00005588}
5589
5590
5591#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
5592static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00005593#ifdef _PC_ABI_AIO_XFER_MAX
5594 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
5595#endif
5596#ifdef _PC_ABI_ASYNC_IO
5597 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
5598#endif
Fred Drakec9680921999-12-13 16:37:25 +00005599#ifdef _PC_ASYNC_IO
5600 {"PC_ASYNC_IO", _PC_ASYNC_IO},
5601#endif
5602#ifdef _PC_CHOWN_RESTRICTED
5603 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
5604#endif
5605#ifdef _PC_FILESIZEBITS
5606 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
5607#endif
5608#ifdef _PC_LAST
5609 {"PC_LAST", _PC_LAST},
5610#endif
5611#ifdef _PC_LINK_MAX
5612 {"PC_LINK_MAX", _PC_LINK_MAX},
5613#endif
5614#ifdef _PC_MAX_CANON
5615 {"PC_MAX_CANON", _PC_MAX_CANON},
5616#endif
5617#ifdef _PC_MAX_INPUT
5618 {"PC_MAX_INPUT", _PC_MAX_INPUT},
5619#endif
5620#ifdef _PC_NAME_MAX
5621 {"PC_NAME_MAX", _PC_NAME_MAX},
5622#endif
5623#ifdef _PC_NO_TRUNC
5624 {"PC_NO_TRUNC", _PC_NO_TRUNC},
5625#endif
5626#ifdef _PC_PATH_MAX
5627 {"PC_PATH_MAX", _PC_PATH_MAX},
5628#endif
5629#ifdef _PC_PIPE_BUF
5630 {"PC_PIPE_BUF", _PC_PIPE_BUF},
5631#endif
5632#ifdef _PC_PRIO_IO
5633 {"PC_PRIO_IO", _PC_PRIO_IO},
5634#endif
5635#ifdef _PC_SOCK_MAXBUF
5636 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
5637#endif
5638#ifdef _PC_SYNC_IO
5639 {"PC_SYNC_IO", _PC_SYNC_IO},
5640#endif
5641#ifdef _PC_VDISABLE
5642 {"PC_VDISABLE", _PC_VDISABLE},
5643#endif
5644};
5645
Fred Drakec9680921999-12-13 16:37:25 +00005646static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005647conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00005648{
5649 return conv_confname(arg, valuep, posix_constants_pathconf,
5650 sizeof(posix_constants_pathconf)
5651 / sizeof(struct constdef));
5652}
5653#endif
5654
5655#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005656PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005657"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00005658Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005659If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00005660
5661static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005662posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005663{
5664 PyObject *result = NULL;
5665 int name, fd;
5666
Fred Drake12c6e2d1999-12-14 21:25:03 +00005667 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
5668 conv_path_confname, &name)) {
Fred Drakec9680921999-12-13 16:37:25 +00005669 long limit;
5670
5671 errno = 0;
5672 limit = fpathconf(fd, name);
5673 if (limit == -1 && errno != 0)
5674 posix_error();
5675 else
Christian Heimes217cfd12007-12-02 14:31:20 +00005676 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00005677 }
5678 return result;
5679}
5680#endif
5681
5682
5683#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005684PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005685"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00005686Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005687If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00005688
5689static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005690posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005691{
5692 PyObject *result = NULL;
5693 int name;
5694 char *path;
5695
5696 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
5697 conv_path_confname, &name)) {
5698 long limit;
5699
5700 errno = 0;
5701 limit = pathconf(path, name);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005702 if (limit == -1 && errno != 0) {
Fred Drakec9680921999-12-13 16:37:25 +00005703 if (errno == EINVAL)
5704 /* could be a path or name problem */
5705 posix_error();
5706 else
5707 posix_error_with_filename(path);
Fred Drake12c6e2d1999-12-14 21:25:03 +00005708 }
Fred Drakec9680921999-12-13 16:37:25 +00005709 else
Christian Heimes217cfd12007-12-02 14:31:20 +00005710 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00005711 }
5712 return result;
5713}
5714#endif
5715
5716#ifdef HAVE_CONFSTR
5717static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00005718#ifdef _CS_ARCHITECTURE
5719 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
5720#endif
5721#ifdef _CS_HOSTNAME
5722 {"CS_HOSTNAME", _CS_HOSTNAME},
5723#endif
5724#ifdef _CS_HW_PROVIDER
5725 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
5726#endif
5727#ifdef _CS_HW_SERIAL
5728 {"CS_HW_SERIAL", _CS_HW_SERIAL},
5729#endif
5730#ifdef _CS_INITTAB_NAME
5731 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
5732#endif
Fred Drakec9680921999-12-13 16:37:25 +00005733#ifdef _CS_LFS64_CFLAGS
5734 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
5735#endif
5736#ifdef _CS_LFS64_LDFLAGS
5737 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
5738#endif
5739#ifdef _CS_LFS64_LIBS
5740 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
5741#endif
5742#ifdef _CS_LFS64_LINTFLAGS
5743 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
5744#endif
5745#ifdef _CS_LFS_CFLAGS
5746 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
5747#endif
5748#ifdef _CS_LFS_LDFLAGS
5749 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
5750#endif
5751#ifdef _CS_LFS_LIBS
5752 {"CS_LFS_LIBS", _CS_LFS_LIBS},
5753#endif
5754#ifdef _CS_LFS_LINTFLAGS
5755 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
5756#endif
Fred Draked86ed291999-12-15 15:34:33 +00005757#ifdef _CS_MACHINE
5758 {"CS_MACHINE", _CS_MACHINE},
5759#endif
Fred Drakec9680921999-12-13 16:37:25 +00005760#ifdef _CS_PATH
5761 {"CS_PATH", _CS_PATH},
5762#endif
Fred Draked86ed291999-12-15 15:34:33 +00005763#ifdef _CS_RELEASE
5764 {"CS_RELEASE", _CS_RELEASE},
5765#endif
5766#ifdef _CS_SRPC_DOMAIN
5767 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
5768#endif
5769#ifdef _CS_SYSNAME
5770 {"CS_SYSNAME", _CS_SYSNAME},
5771#endif
5772#ifdef _CS_VERSION
5773 {"CS_VERSION", _CS_VERSION},
5774#endif
Fred Drakec9680921999-12-13 16:37:25 +00005775#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
5776 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
5777#endif
5778#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
5779 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
5780#endif
5781#ifdef _CS_XBS5_ILP32_OFF32_LIBS
5782 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
5783#endif
5784#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
5785 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
5786#endif
5787#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
5788 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
5789#endif
5790#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
5791 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
5792#endif
5793#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
5794 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
5795#endif
5796#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
5797 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
5798#endif
5799#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
5800 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
5801#endif
5802#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
5803 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
5804#endif
5805#ifdef _CS_XBS5_LP64_OFF64_LIBS
5806 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
5807#endif
5808#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
5809 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
5810#endif
5811#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
5812 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
5813#endif
5814#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
5815 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
5816#endif
5817#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
5818 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
5819#endif
5820#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
5821 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
5822#endif
Fred Draked86ed291999-12-15 15:34:33 +00005823#ifdef _MIPS_CS_AVAIL_PROCESSORS
5824 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
5825#endif
5826#ifdef _MIPS_CS_BASE
5827 {"MIPS_CS_BASE", _MIPS_CS_BASE},
5828#endif
5829#ifdef _MIPS_CS_HOSTID
5830 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
5831#endif
5832#ifdef _MIPS_CS_HW_NAME
5833 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
5834#endif
5835#ifdef _MIPS_CS_NUM_PROCESSORS
5836 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
5837#endif
5838#ifdef _MIPS_CS_OSREL_MAJ
5839 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
5840#endif
5841#ifdef _MIPS_CS_OSREL_MIN
5842 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
5843#endif
5844#ifdef _MIPS_CS_OSREL_PATCH
5845 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
5846#endif
5847#ifdef _MIPS_CS_OS_NAME
5848 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
5849#endif
5850#ifdef _MIPS_CS_OS_PROVIDER
5851 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
5852#endif
5853#ifdef _MIPS_CS_PROCESSORS
5854 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
5855#endif
5856#ifdef _MIPS_CS_SERIAL
5857 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
5858#endif
5859#ifdef _MIPS_CS_VENDOR
5860 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
5861#endif
Fred Drakec9680921999-12-13 16:37:25 +00005862};
5863
5864static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005865conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00005866{
5867 return conv_confname(arg, valuep, posix_constants_confstr,
5868 sizeof(posix_constants_confstr)
5869 / sizeof(struct constdef));
5870}
5871
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005872PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005873"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005874Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00005875
5876static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005877posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00005878{
5879 PyObject *result = NULL;
5880 int name;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005881 char buffer[256];
Fred Drakec9680921999-12-13 16:37:25 +00005882
5883 if (PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name)) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005884 int len;
Fred Drakec9680921999-12-13 16:37:25 +00005885
Fred Drakec9680921999-12-13 16:37:25 +00005886 errno = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005887 len = confstr(name, buffer, sizeof(buffer));
5888 if (len == 0) {
5889 if (errno) {
5890 posix_error();
5891 }
5892 else {
5893 result = Py_None;
5894 Py_INCREF(Py_None);
5895 }
Fred Drakec9680921999-12-13 16:37:25 +00005896 }
5897 else {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005898 if ((unsigned int)len >= sizeof(buffer)) {
Neal Norwitz93c56822007-08-26 07:10:06 +00005899 result = PyUnicode_FromStringAndSize(NULL, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00005900 if (result != NULL)
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00005901 confstr(name, _PyUnicode_AsString(result), len);
Fred Drakec9680921999-12-13 16:37:25 +00005902 }
5903 else
Neal Norwitz93c56822007-08-26 07:10:06 +00005904 result = PyUnicode_FromStringAndSize(buffer, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00005905 }
5906 }
5907 return result;
5908}
5909#endif
5910
5911
5912#ifdef HAVE_SYSCONF
5913static struct constdef posix_constants_sysconf[] = {
5914#ifdef _SC_2_CHAR_TERM
5915 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
5916#endif
5917#ifdef _SC_2_C_BIND
5918 {"SC_2_C_BIND", _SC_2_C_BIND},
5919#endif
5920#ifdef _SC_2_C_DEV
5921 {"SC_2_C_DEV", _SC_2_C_DEV},
5922#endif
5923#ifdef _SC_2_C_VERSION
5924 {"SC_2_C_VERSION", _SC_2_C_VERSION},
5925#endif
5926#ifdef _SC_2_FORT_DEV
5927 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
5928#endif
5929#ifdef _SC_2_FORT_RUN
5930 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
5931#endif
5932#ifdef _SC_2_LOCALEDEF
5933 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
5934#endif
5935#ifdef _SC_2_SW_DEV
5936 {"SC_2_SW_DEV", _SC_2_SW_DEV},
5937#endif
5938#ifdef _SC_2_UPE
5939 {"SC_2_UPE", _SC_2_UPE},
5940#endif
5941#ifdef _SC_2_VERSION
5942 {"SC_2_VERSION", _SC_2_VERSION},
5943#endif
Fred Draked86ed291999-12-15 15:34:33 +00005944#ifdef _SC_ABI_ASYNCHRONOUS_IO
5945 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
5946#endif
5947#ifdef _SC_ACL
5948 {"SC_ACL", _SC_ACL},
5949#endif
Fred Drakec9680921999-12-13 16:37:25 +00005950#ifdef _SC_AIO_LISTIO_MAX
5951 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
5952#endif
Fred Drakec9680921999-12-13 16:37:25 +00005953#ifdef _SC_AIO_MAX
5954 {"SC_AIO_MAX", _SC_AIO_MAX},
5955#endif
5956#ifdef _SC_AIO_PRIO_DELTA_MAX
5957 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
5958#endif
5959#ifdef _SC_ARG_MAX
5960 {"SC_ARG_MAX", _SC_ARG_MAX},
5961#endif
5962#ifdef _SC_ASYNCHRONOUS_IO
5963 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
5964#endif
5965#ifdef _SC_ATEXIT_MAX
5966 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
5967#endif
Fred Draked86ed291999-12-15 15:34:33 +00005968#ifdef _SC_AUDIT
5969 {"SC_AUDIT", _SC_AUDIT},
5970#endif
Fred Drakec9680921999-12-13 16:37:25 +00005971#ifdef _SC_AVPHYS_PAGES
5972 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
5973#endif
5974#ifdef _SC_BC_BASE_MAX
5975 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
5976#endif
5977#ifdef _SC_BC_DIM_MAX
5978 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
5979#endif
5980#ifdef _SC_BC_SCALE_MAX
5981 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
5982#endif
5983#ifdef _SC_BC_STRING_MAX
5984 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
5985#endif
Fred Draked86ed291999-12-15 15:34:33 +00005986#ifdef _SC_CAP
5987 {"SC_CAP", _SC_CAP},
5988#endif
Fred Drakec9680921999-12-13 16:37:25 +00005989#ifdef _SC_CHARCLASS_NAME_MAX
5990 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
5991#endif
5992#ifdef _SC_CHAR_BIT
5993 {"SC_CHAR_BIT", _SC_CHAR_BIT},
5994#endif
5995#ifdef _SC_CHAR_MAX
5996 {"SC_CHAR_MAX", _SC_CHAR_MAX},
5997#endif
5998#ifdef _SC_CHAR_MIN
5999 {"SC_CHAR_MIN", _SC_CHAR_MIN},
6000#endif
6001#ifdef _SC_CHILD_MAX
6002 {"SC_CHILD_MAX", _SC_CHILD_MAX},
6003#endif
6004#ifdef _SC_CLK_TCK
6005 {"SC_CLK_TCK", _SC_CLK_TCK},
6006#endif
6007#ifdef _SC_COHER_BLKSZ
6008 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
6009#endif
6010#ifdef _SC_COLL_WEIGHTS_MAX
6011 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
6012#endif
6013#ifdef _SC_DCACHE_ASSOC
6014 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
6015#endif
6016#ifdef _SC_DCACHE_BLKSZ
6017 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
6018#endif
6019#ifdef _SC_DCACHE_LINESZ
6020 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
6021#endif
6022#ifdef _SC_DCACHE_SZ
6023 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
6024#endif
6025#ifdef _SC_DCACHE_TBLKSZ
6026 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
6027#endif
6028#ifdef _SC_DELAYTIMER_MAX
6029 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
6030#endif
6031#ifdef _SC_EQUIV_CLASS_MAX
6032 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
6033#endif
6034#ifdef _SC_EXPR_NEST_MAX
6035 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
6036#endif
6037#ifdef _SC_FSYNC
6038 {"SC_FSYNC", _SC_FSYNC},
6039#endif
6040#ifdef _SC_GETGR_R_SIZE_MAX
6041 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
6042#endif
6043#ifdef _SC_GETPW_R_SIZE_MAX
6044 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
6045#endif
6046#ifdef _SC_ICACHE_ASSOC
6047 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
6048#endif
6049#ifdef _SC_ICACHE_BLKSZ
6050 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
6051#endif
6052#ifdef _SC_ICACHE_LINESZ
6053 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
6054#endif
6055#ifdef _SC_ICACHE_SZ
6056 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
6057#endif
Fred Draked86ed291999-12-15 15:34:33 +00006058#ifdef _SC_INF
6059 {"SC_INF", _SC_INF},
6060#endif
Fred Drakec9680921999-12-13 16:37:25 +00006061#ifdef _SC_INT_MAX
6062 {"SC_INT_MAX", _SC_INT_MAX},
6063#endif
6064#ifdef _SC_INT_MIN
6065 {"SC_INT_MIN", _SC_INT_MIN},
6066#endif
6067#ifdef _SC_IOV_MAX
6068 {"SC_IOV_MAX", _SC_IOV_MAX},
6069#endif
Fred Draked86ed291999-12-15 15:34:33 +00006070#ifdef _SC_IP_SECOPTS
6071 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
6072#endif
Fred Drakec9680921999-12-13 16:37:25 +00006073#ifdef _SC_JOB_CONTROL
6074 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
6075#endif
Fred Draked86ed291999-12-15 15:34:33 +00006076#ifdef _SC_KERN_POINTERS
6077 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
6078#endif
6079#ifdef _SC_KERN_SIM
6080 {"SC_KERN_SIM", _SC_KERN_SIM},
6081#endif
Fred Drakec9680921999-12-13 16:37:25 +00006082#ifdef _SC_LINE_MAX
6083 {"SC_LINE_MAX", _SC_LINE_MAX},
6084#endif
6085#ifdef _SC_LOGIN_NAME_MAX
6086 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
6087#endif
6088#ifdef _SC_LOGNAME_MAX
6089 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
6090#endif
6091#ifdef _SC_LONG_BIT
6092 {"SC_LONG_BIT", _SC_LONG_BIT},
6093#endif
Fred Draked86ed291999-12-15 15:34:33 +00006094#ifdef _SC_MAC
6095 {"SC_MAC", _SC_MAC},
6096#endif
Fred Drakec9680921999-12-13 16:37:25 +00006097#ifdef _SC_MAPPED_FILES
6098 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
6099#endif
6100#ifdef _SC_MAXPID
6101 {"SC_MAXPID", _SC_MAXPID},
6102#endif
6103#ifdef _SC_MB_LEN_MAX
6104 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
6105#endif
6106#ifdef _SC_MEMLOCK
6107 {"SC_MEMLOCK", _SC_MEMLOCK},
6108#endif
6109#ifdef _SC_MEMLOCK_RANGE
6110 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
6111#endif
6112#ifdef _SC_MEMORY_PROTECTION
6113 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
6114#endif
6115#ifdef _SC_MESSAGE_PASSING
6116 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
6117#endif
Fred Draked86ed291999-12-15 15:34:33 +00006118#ifdef _SC_MMAP_FIXED_ALIGNMENT
6119 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
6120#endif
Fred Drakec9680921999-12-13 16:37:25 +00006121#ifdef _SC_MQ_OPEN_MAX
6122 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
6123#endif
6124#ifdef _SC_MQ_PRIO_MAX
6125 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
6126#endif
Fred Draked86ed291999-12-15 15:34:33 +00006127#ifdef _SC_NACLS_MAX
6128 {"SC_NACLS_MAX", _SC_NACLS_MAX},
6129#endif
Fred Drakec9680921999-12-13 16:37:25 +00006130#ifdef _SC_NGROUPS_MAX
6131 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
6132#endif
6133#ifdef _SC_NL_ARGMAX
6134 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
6135#endif
6136#ifdef _SC_NL_LANGMAX
6137 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
6138#endif
6139#ifdef _SC_NL_MSGMAX
6140 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
6141#endif
6142#ifdef _SC_NL_NMAX
6143 {"SC_NL_NMAX", _SC_NL_NMAX},
6144#endif
6145#ifdef _SC_NL_SETMAX
6146 {"SC_NL_SETMAX", _SC_NL_SETMAX},
6147#endif
6148#ifdef _SC_NL_TEXTMAX
6149 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
6150#endif
6151#ifdef _SC_NPROCESSORS_CONF
6152 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
6153#endif
6154#ifdef _SC_NPROCESSORS_ONLN
6155 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
6156#endif
Fred Draked86ed291999-12-15 15:34:33 +00006157#ifdef _SC_NPROC_CONF
6158 {"SC_NPROC_CONF", _SC_NPROC_CONF},
6159#endif
6160#ifdef _SC_NPROC_ONLN
6161 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
6162#endif
Fred Drakec9680921999-12-13 16:37:25 +00006163#ifdef _SC_NZERO
6164 {"SC_NZERO", _SC_NZERO},
6165#endif
6166#ifdef _SC_OPEN_MAX
6167 {"SC_OPEN_MAX", _SC_OPEN_MAX},
6168#endif
6169#ifdef _SC_PAGESIZE
6170 {"SC_PAGESIZE", _SC_PAGESIZE},
6171#endif
6172#ifdef _SC_PAGE_SIZE
6173 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
6174#endif
6175#ifdef _SC_PASS_MAX
6176 {"SC_PASS_MAX", _SC_PASS_MAX},
6177#endif
6178#ifdef _SC_PHYS_PAGES
6179 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
6180#endif
6181#ifdef _SC_PII
6182 {"SC_PII", _SC_PII},
6183#endif
6184#ifdef _SC_PII_INTERNET
6185 {"SC_PII_INTERNET", _SC_PII_INTERNET},
6186#endif
6187#ifdef _SC_PII_INTERNET_DGRAM
6188 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
6189#endif
6190#ifdef _SC_PII_INTERNET_STREAM
6191 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
6192#endif
6193#ifdef _SC_PII_OSI
6194 {"SC_PII_OSI", _SC_PII_OSI},
6195#endif
6196#ifdef _SC_PII_OSI_CLTS
6197 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
6198#endif
6199#ifdef _SC_PII_OSI_COTS
6200 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
6201#endif
6202#ifdef _SC_PII_OSI_M
6203 {"SC_PII_OSI_M", _SC_PII_OSI_M},
6204#endif
6205#ifdef _SC_PII_SOCKET
6206 {"SC_PII_SOCKET", _SC_PII_SOCKET},
6207#endif
6208#ifdef _SC_PII_XTI
6209 {"SC_PII_XTI", _SC_PII_XTI},
6210#endif
6211#ifdef _SC_POLL
6212 {"SC_POLL", _SC_POLL},
6213#endif
6214#ifdef _SC_PRIORITIZED_IO
6215 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
6216#endif
6217#ifdef _SC_PRIORITY_SCHEDULING
6218 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
6219#endif
6220#ifdef _SC_REALTIME_SIGNALS
6221 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
6222#endif
6223#ifdef _SC_RE_DUP_MAX
6224 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
6225#endif
6226#ifdef _SC_RTSIG_MAX
6227 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
6228#endif
6229#ifdef _SC_SAVED_IDS
6230 {"SC_SAVED_IDS", _SC_SAVED_IDS},
6231#endif
6232#ifdef _SC_SCHAR_MAX
6233 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
6234#endif
6235#ifdef _SC_SCHAR_MIN
6236 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
6237#endif
6238#ifdef _SC_SELECT
6239 {"SC_SELECT", _SC_SELECT},
6240#endif
6241#ifdef _SC_SEMAPHORES
6242 {"SC_SEMAPHORES", _SC_SEMAPHORES},
6243#endif
6244#ifdef _SC_SEM_NSEMS_MAX
6245 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
6246#endif
6247#ifdef _SC_SEM_VALUE_MAX
6248 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
6249#endif
6250#ifdef _SC_SHARED_MEMORY_OBJECTS
6251 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
6252#endif
6253#ifdef _SC_SHRT_MAX
6254 {"SC_SHRT_MAX", _SC_SHRT_MAX},
6255#endif
6256#ifdef _SC_SHRT_MIN
6257 {"SC_SHRT_MIN", _SC_SHRT_MIN},
6258#endif
6259#ifdef _SC_SIGQUEUE_MAX
6260 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
6261#endif
6262#ifdef _SC_SIGRT_MAX
6263 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
6264#endif
6265#ifdef _SC_SIGRT_MIN
6266 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
6267#endif
Fred Draked86ed291999-12-15 15:34:33 +00006268#ifdef _SC_SOFTPOWER
6269 {"SC_SOFTPOWER", _SC_SOFTPOWER},
6270#endif
Fred Drakec9680921999-12-13 16:37:25 +00006271#ifdef _SC_SPLIT_CACHE
6272 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
6273#endif
6274#ifdef _SC_SSIZE_MAX
6275 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
6276#endif
6277#ifdef _SC_STACK_PROT
6278 {"SC_STACK_PROT", _SC_STACK_PROT},
6279#endif
6280#ifdef _SC_STREAM_MAX
6281 {"SC_STREAM_MAX", _SC_STREAM_MAX},
6282#endif
6283#ifdef _SC_SYNCHRONIZED_IO
6284 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
6285#endif
6286#ifdef _SC_THREADS
6287 {"SC_THREADS", _SC_THREADS},
6288#endif
6289#ifdef _SC_THREAD_ATTR_STACKADDR
6290 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
6291#endif
6292#ifdef _SC_THREAD_ATTR_STACKSIZE
6293 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
6294#endif
6295#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
6296 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
6297#endif
6298#ifdef _SC_THREAD_KEYS_MAX
6299 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
6300#endif
6301#ifdef _SC_THREAD_PRIORITY_SCHEDULING
6302 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
6303#endif
6304#ifdef _SC_THREAD_PRIO_INHERIT
6305 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
6306#endif
6307#ifdef _SC_THREAD_PRIO_PROTECT
6308 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
6309#endif
6310#ifdef _SC_THREAD_PROCESS_SHARED
6311 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
6312#endif
6313#ifdef _SC_THREAD_SAFE_FUNCTIONS
6314 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
6315#endif
6316#ifdef _SC_THREAD_STACK_MIN
6317 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
6318#endif
6319#ifdef _SC_THREAD_THREADS_MAX
6320 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
6321#endif
6322#ifdef _SC_TIMERS
6323 {"SC_TIMERS", _SC_TIMERS},
6324#endif
6325#ifdef _SC_TIMER_MAX
6326 {"SC_TIMER_MAX", _SC_TIMER_MAX},
6327#endif
6328#ifdef _SC_TTY_NAME_MAX
6329 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
6330#endif
6331#ifdef _SC_TZNAME_MAX
6332 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
6333#endif
6334#ifdef _SC_T_IOV_MAX
6335 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
6336#endif
6337#ifdef _SC_UCHAR_MAX
6338 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
6339#endif
6340#ifdef _SC_UINT_MAX
6341 {"SC_UINT_MAX", _SC_UINT_MAX},
6342#endif
6343#ifdef _SC_UIO_MAXIOV
6344 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
6345#endif
6346#ifdef _SC_ULONG_MAX
6347 {"SC_ULONG_MAX", _SC_ULONG_MAX},
6348#endif
6349#ifdef _SC_USHRT_MAX
6350 {"SC_USHRT_MAX", _SC_USHRT_MAX},
6351#endif
6352#ifdef _SC_VERSION
6353 {"SC_VERSION", _SC_VERSION},
6354#endif
6355#ifdef _SC_WORD_BIT
6356 {"SC_WORD_BIT", _SC_WORD_BIT},
6357#endif
6358#ifdef _SC_XBS5_ILP32_OFF32
6359 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
6360#endif
6361#ifdef _SC_XBS5_ILP32_OFFBIG
6362 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
6363#endif
6364#ifdef _SC_XBS5_LP64_OFF64
6365 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
6366#endif
6367#ifdef _SC_XBS5_LPBIG_OFFBIG
6368 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
6369#endif
6370#ifdef _SC_XOPEN_CRYPT
6371 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
6372#endif
6373#ifdef _SC_XOPEN_ENH_I18N
6374 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
6375#endif
6376#ifdef _SC_XOPEN_LEGACY
6377 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
6378#endif
6379#ifdef _SC_XOPEN_REALTIME
6380 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
6381#endif
6382#ifdef _SC_XOPEN_REALTIME_THREADS
6383 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
6384#endif
6385#ifdef _SC_XOPEN_SHM
6386 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
6387#endif
6388#ifdef _SC_XOPEN_UNIX
6389 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
6390#endif
6391#ifdef _SC_XOPEN_VERSION
6392 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
6393#endif
6394#ifdef _SC_XOPEN_XCU_VERSION
6395 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
6396#endif
6397#ifdef _SC_XOPEN_XPG2
6398 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
6399#endif
6400#ifdef _SC_XOPEN_XPG3
6401 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
6402#endif
6403#ifdef _SC_XOPEN_XPG4
6404 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
6405#endif
6406};
6407
6408static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006409conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00006410{
6411 return conv_confname(arg, valuep, posix_constants_sysconf,
6412 sizeof(posix_constants_sysconf)
6413 / sizeof(struct constdef));
6414}
6415
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006416PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006417"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006418Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00006419
6420static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006421posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00006422{
6423 PyObject *result = NULL;
6424 int name;
6425
6426 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
6427 int value;
6428
6429 errno = 0;
6430 value = sysconf(name);
6431 if (value == -1 && errno != 0)
6432 posix_error();
6433 else
Christian Heimes217cfd12007-12-02 14:31:20 +00006434 result = PyLong_FromLong(value);
Fred Drakec9680921999-12-13 16:37:25 +00006435 }
6436 return result;
6437}
6438#endif
6439
6440
Fred Drakebec628d1999-12-15 18:31:10 +00006441/* This code is used to ensure that the tables of configuration value names
6442 * are in sorted order as required by conv_confname(), and also to build the
6443 * the exported dictionaries that are used to publish information about the
6444 * names available on the host platform.
6445 *
6446 * Sorting the table at runtime ensures that the table is properly ordered
6447 * when used, even for platforms we're not able to test on. It also makes
6448 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00006449 */
Fred Drakebec628d1999-12-15 18:31:10 +00006450
6451static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006452cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00006453{
6454 const struct constdef *c1 =
6455 (const struct constdef *) v1;
6456 const struct constdef *c2 =
6457 (const struct constdef *) v2;
6458
6459 return strcmp(c1->name, c2->name);
6460}
6461
6462static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006463setup_confname_table(struct constdef *table, size_t tablesize,
Fred Drake4d1e64b2002-04-15 19:40:07 +00006464 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00006465{
Fred Drakebec628d1999-12-15 18:31:10 +00006466 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00006467 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00006468
6469 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
6470 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00006471 if (d == NULL)
6472 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006473
Barry Warsaw3155db32000-04-13 15:20:40 +00006474 for (i=0; i < tablesize; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +00006475 PyObject *o = PyLong_FromLong(table[i].value);
Barry Warsaw3155db32000-04-13 15:20:40 +00006476 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
6477 Py_XDECREF(o);
6478 Py_DECREF(d);
6479 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006480 }
Barry Warsaw3155db32000-04-13 15:20:40 +00006481 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00006482 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00006483 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00006484}
6485
Fred Drakebec628d1999-12-15 18:31:10 +00006486/* Return -1 on failure, 0 on success. */
6487static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00006488setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00006489{
6490#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00006491 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00006492 sizeof(posix_constants_pathconf)
6493 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006494 "pathconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006495 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006496#endif
6497#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00006498 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00006499 sizeof(posix_constants_confstr)
6500 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006501 "confstr_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006502 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006503#endif
6504#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00006505 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00006506 sizeof(posix_constants_sysconf)
6507 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00006508 "sysconf_names", module))
Fred Drakebec628d1999-12-15 18:31:10 +00006509 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00006510#endif
Fred Drakebec628d1999-12-15 18:31:10 +00006511 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00006512}
Fred Draked86ed291999-12-15 15:34:33 +00006513
6514
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006515PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006516"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006517Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006518in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006519
6520static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006521posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006522{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006523 abort();
6524 /*NOTREACHED*/
6525 Py_FatalError("abort() called from Python code didn't abort!");
6526 return NULL;
6527}
Fred Drakebec628d1999-12-15 18:31:10 +00006528
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006529#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006530PyDoc_STRVAR(win32_startfile__doc__,
Georg Brandlf4f44152006-02-18 22:29:33 +00006531"startfile(filepath [, operation]) - Start a file with its associated\n\
6532application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00006533\n\
Georg Brandlf4f44152006-02-18 22:29:33 +00006534When \"operation\" is not specified or \"open\", this acts like\n\
6535double-clicking the file in Explorer, or giving the file name as an\n\
6536argument to the DOS \"start\" command: the file is opened with whatever\n\
6537application (if any) its extension is associated.\n\
6538When another \"operation\" is given, it specifies what should be done with\n\
6539the file. A typical operation is \"print\".\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00006540\n\
6541startfile returns as soon as the associated application is launched.\n\
6542There is no option to wait for the application to close, and no way\n\
6543to retrieve the application's exit status.\n\
6544\n\
6545The filepath is relative to the current directory. If you want to use\n\
6546an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006547the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00006548
6549static PyObject *
6550win32_startfile(PyObject *self, PyObject *args)
6551{
6552 char *filepath;
Georg Brandlf4f44152006-02-18 22:29:33 +00006553 char *operation = NULL;
Tim Petersf58a7aa2000-09-22 10:05:54 +00006554 HINSTANCE rc;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006555#ifdef Py_WIN_WIDE_FILENAMES
6556 if (unicode_file_names()) {
6557 PyObject *unipath, *woperation = NULL;
6558 if (!PyArg_ParseTuple(args, "U|s:startfile",
6559 &unipath, &operation)) {
6560 PyErr_Clear();
6561 goto normal;
6562 }
6563
6564
6565 if (operation) {
6566 woperation = PyUnicode_DecodeASCII(operation,
6567 strlen(operation), NULL);
6568 if (!woperation) {
6569 PyErr_Clear();
6570 operation = NULL;
6571 goto normal;
6572 }
6573 }
6574
6575 Py_BEGIN_ALLOW_THREADS
6576 rc = ShellExecuteW((HWND)0, woperation ? PyUnicode_AS_UNICODE(woperation) : 0,
6577 PyUnicode_AS_UNICODE(unipath),
6578 NULL, NULL, SW_SHOWNORMAL);
6579 Py_END_ALLOW_THREADS
6580
6581 Py_XDECREF(woperation);
6582 if (rc <= (HINSTANCE)32) {
6583 PyObject *errval = win32_error_unicode("startfile",
6584 PyUnicode_AS_UNICODE(unipath));
6585 return errval;
6586 }
6587 Py_INCREF(Py_None);
6588 return Py_None;
6589 }
6590#endif
6591
6592normal:
Georg Brandlf4f44152006-02-18 22:29:33 +00006593 if (!PyArg_ParseTuple(args, "et|s:startfile",
6594 Py_FileSystemDefaultEncoding, &filepath,
6595 &operation))
Tim Petersf58a7aa2000-09-22 10:05:54 +00006596 return NULL;
6597 Py_BEGIN_ALLOW_THREADS
Georg Brandlf4f44152006-02-18 22:29:33 +00006598 rc = ShellExecute((HWND)0, operation, filepath,
6599 NULL, NULL, SW_SHOWNORMAL);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006600 Py_END_ALLOW_THREADS
Georg Brandle9f8ec92005-09-25 06:16:40 +00006601 if (rc <= (HINSTANCE)32) {
6602 PyObject *errval = win32_error("startfile", filepath);
6603 PyMem_Free(filepath);
6604 return errval;
6605 }
6606 PyMem_Free(filepath);
Tim Petersf58a7aa2000-09-22 10:05:54 +00006607 Py_INCREF(Py_None);
6608 return Py_None;
6609}
6610#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006611
Martin v. Löwis438b5342002-12-27 10:16:42 +00006612#ifdef HAVE_GETLOADAVG
6613PyDoc_STRVAR(posix_getloadavg__doc__,
6614"getloadavg() -> (float, float, float)\n\n\
6615Return the number of processes in the system run queue averaged over\n\
6616the last 1, 5, and 15 minutes or raises OSError if the load average\n\
6617was unobtainable");
6618
6619static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006620posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00006621{
6622 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00006623 if (getloadavg(loadavg, 3)!=3) {
6624 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
6625 return NULL;
6626 } else
6627 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
6628}
6629#endif
6630
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006631#ifdef MS_WINDOWS
6632
6633PyDoc_STRVAR(win32_urandom__doc__,
6634"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00006635Return n random bytes suitable for cryptographic use.");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006636
6637typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
6638 LPCSTR pszContainer, LPCSTR pszProvider, DWORD dwProvType,\
6639 DWORD dwFlags );
6640typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV hProv, DWORD dwLen,\
6641 BYTE *pbBuffer );
6642
6643static CRYPTGENRANDOM pCryptGenRandom = NULL;
Thomas Wouters89d996e2007-09-08 17:39:28 +00006644/* This handle is never explicitly released. Instead, the operating
6645 system will release it when the process terminates. */
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006646static HCRYPTPROV hCryptProv = 0;
6647
Tim Peters4ad82172004-08-30 17:02:04 +00006648static PyObject*
6649win32_urandom(PyObject *self, PyObject *args)
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006650{
Tim Petersd3115382004-08-30 17:36:46 +00006651 int howMany;
6652 PyObject* result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006653
Tim Peters4ad82172004-08-30 17:02:04 +00006654 /* Read arguments */
Tim Peters9b279a82004-08-30 17:10:53 +00006655 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
Tim Peters4ad82172004-08-30 17:02:04 +00006656 return NULL;
Tim Peters51eba612004-08-30 17:08:02 +00006657 if (howMany < 0)
6658 return PyErr_Format(PyExc_ValueError,
6659 "negative argument not allowed");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006660
Tim Peters4ad82172004-08-30 17:02:04 +00006661 if (hCryptProv == 0) {
6662 HINSTANCE hAdvAPI32 = NULL;
6663 CRYPTACQUIRECONTEXTA pCryptAcquireContext = NULL;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006664
Tim Peters4ad82172004-08-30 17:02:04 +00006665 /* Obtain handle to the DLL containing CryptoAPI
6666 This should not fail */
6667 hAdvAPI32 = GetModuleHandle("advapi32.dll");
6668 if(hAdvAPI32 == NULL)
6669 return win32_error("GetModuleHandle", NULL);
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006670
Tim Peters4ad82172004-08-30 17:02:04 +00006671 /* Obtain pointers to the CryptoAPI functions
6672 This will fail on some early versions of Win95 */
6673 pCryptAcquireContext = (CRYPTACQUIRECONTEXTA)GetProcAddress(
6674 hAdvAPI32,
6675 "CryptAcquireContextA");
6676 if (pCryptAcquireContext == NULL)
6677 return PyErr_Format(PyExc_NotImplementedError,
6678 "CryptAcquireContextA not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006679
Tim Peters4ad82172004-08-30 17:02:04 +00006680 pCryptGenRandom = (CRYPTGENRANDOM)GetProcAddress(
6681 hAdvAPI32, "CryptGenRandom");
Thomas Wouters89f507f2006-12-13 04:49:30 +00006682 if (pCryptGenRandom == NULL)
Tim Peters4ad82172004-08-30 17:02:04 +00006683 return PyErr_Format(PyExc_NotImplementedError,
6684 "CryptGenRandom not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006685
Tim Peters4ad82172004-08-30 17:02:04 +00006686 /* Acquire context */
6687 if (! pCryptAcquireContext(&hCryptProv, NULL, NULL,
6688 PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
6689 return win32_error("CryptAcquireContext", NULL);
6690 }
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006691
Tim Peters4ad82172004-08-30 17:02:04 +00006692 /* Allocate bytes */
Christian Heimes72b710a2008-05-26 13:28:38 +00006693 result = PyBytes_FromStringAndSize(NULL, howMany);
Tim Petersd3115382004-08-30 17:36:46 +00006694 if (result != NULL) {
6695 /* Get random data */
Amaury Forgeot d'Arca05ada32008-07-21 21:13:14 +00006696 memset(PyBytes_AS_STRING(result), 0, howMany); /* zero seed */
Tim Petersd3115382004-08-30 17:36:46 +00006697 if (! pCryptGenRandom(hCryptProv, howMany, (unsigned char*)
Christian Heimes72b710a2008-05-26 13:28:38 +00006698 PyBytes_AS_STRING(result))) {
Tim Petersd3115382004-08-30 17:36:46 +00006699 Py_DECREF(result);
6700 return win32_error("CryptGenRandom", NULL);
6701 }
Tim Peters4ad82172004-08-30 17:02:04 +00006702 }
Tim Petersd3115382004-08-30 17:36:46 +00006703 return result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00006704}
6705#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00006706
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00006707PyDoc_STRVAR(device_encoding__doc__,
6708"device_encoding(fd) -> str\n\n\
6709Return a string describing the encoding of the device\n\
6710if the output is a terminal; else return None.");
6711
6712static PyObject *
6713device_encoding(PyObject *self, PyObject *args)
6714{
6715 int fd;
6716 if (!PyArg_ParseTuple(args, "i:device_encoding", &fd))
6717 return NULL;
6718 if (!isatty(fd)) {
6719 Py_INCREF(Py_None);
6720 return Py_None;
6721 }
6722#if defined(MS_WINDOWS) || defined(MS_WIN64)
6723 if (fd == 0) {
6724 char buf[100];
6725 sprintf(buf, "cp%d", GetConsoleCP());
6726 return PyUnicode_FromString(buf);
6727 }
6728 if (fd == 1 || fd == 2) {
6729 char buf[100];
6730 sprintf(buf, "cp%d", GetConsoleOutputCP());
6731 return PyUnicode_FromString(buf);
6732 }
6733#elif defined(CODESET)
6734 {
6735 char *codeset = nl_langinfo(CODESET);
6736 if (codeset)
6737 return PyUnicode_FromString(codeset);
6738 }
6739#endif
6740 Py_INCREF(Py_None);
6741 return Py_None;
6742}
6743
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006744#ifdef __VMS
6745/* Use openssl random routine */
6746#include <openssl/rand.h>
6747PyDoc_STRVAR(vms_urandom__doc__,
6748"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00006749Return n random bytes suitable for cryptographic use.");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006750
6751static PyObject*
6752vms_urandom(PyObject *self, PyObject *args)
6753{
6754 int howMany;
6755 PyObject* result;
6756
6757 /* Read arguments */
6758 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
6759 return NULL;
6760 if (howMany < 0)
6761 return PyErr_Format(PyExc_ValueError,
6762 "negative argument not allowed");
6763
6764 /* Allocate bytes */
Christian Heimes72b710a2008-05-26 13:28:38 +00006765 result = PyBytes_FromStringAndSize(NULL, howMany);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006766 if (result != NULL) {
6767 /* Get random data */
6768 if (RAND_pseudo_bytes((unsigned char*)
Christian Heimes72b710a2008-05-26 13:28:38 +00006769 PyBytes_AS_STRING(result),
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006770 howMany) < 0) {
6771 Py_DECREF(result);
6772 return PyErr_Format(PyExc_ValueError,
6773 "RAND_pseudo_bytes");
6774 }
6775 }
6776 return result;
6777}
6778#endif
6779
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006780static PyMethodDef posix_methods[] = {
6781 {"access", posix_access, METH_VARARGS, posix_access__doc__},
6782#ifdef HAVE_TTYNAME
6783 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
6784#endif
6785 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +00006786#ifdef HAVE_CHFLAGS
6787 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
6788#endif /* HAVE_CHFLAGS */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006789 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +00006790#ifdef HAVE_FCHMOD
6791 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
6792#endif /* HAVE_FCHMOD */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006793#ifdef HAVE_CHOWN
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006794 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006795#endif /* HAVE_CHOWN */
Christian Heimes4e30a842007-11-30 22:12:06 +00006796#ifdef HAVE_LCHMOD
6797 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
6798#endif /* HAVE_LCHMOD */
6799#ifdef HAVE_FCHOWN
6800 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
6801#endif /* HAVE_FCHOWN */
Thomas Wouterscf297e42007-02-23 15:07:44 +00006802#ifdef HAVE_LCHFLAGS
6803 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
6804#endif /* HAVE_LCHFLAGS */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00006805#ifdef HAVE_LCHOWN
6806 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
6807#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +00006808#ifdef HAVE_CHROOT
6809 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
6810#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006811#ifdef HAVE_CTERMID
Neal Norwitze241ce82003-02-17 18:17:05 +00006812 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006813#endif
Guido van Rossum36bc6801995-06-14 22:54:23 +00006814#ifdef HAVE_GETCWD
Neal Norwitze241ce82003-02-17 18:17:05 +00006815 {"getcwd", posix_getcwd, METH_NOARGS, posix_getcwd__doc__},
Neal Norwitze241ce82003-02-17 18:17:05 +00006816 {"getcwdu", posix_getcwdu, METH_NOARGS, posix_getcwdu__doc__},
Guido van Rossum36bc6801995-06-14 22:54:23 +00006817#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00006818#ifdef HAVE_LINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006819 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006820#endif /* HAVE_LINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006821 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
6822 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
6823 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006824#ifdef HAVE_NICE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006825 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006826#endif /* HAVE_NICE */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006827#ifdef HAVE_READLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006828 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006829#endif /* HAVE_READLINK */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006830 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
6831 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
6832 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Martin v. Löwisf607bda2002-10-16 18:27:39 +00006833 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006834#ifdef HAVE_SYMLINK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006835 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006836#endif /* HAVE_SYMLINK */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006837#ifdef HAVE_SYSTEM
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006838 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006839#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006840 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006841#ifdef HAVE_UNAME
Neal Norwitze241ce82003-02-17 18:17:05 +00006842 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006843#endif /* HAVE_UNAME */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006844 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
6845 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
6846 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006847#ifdef HAVE_TIMES
Neal Norwitze241ce82003-02-17 18:17:05 +00006848 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006849#endif /* HAVE_TIMES */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006850 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006851#ifdef HAVE_EXECV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006852 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
6853 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006854#endif /* HAVE_EXECV */
Guido van Rossuma1065681999-01-25 23:20:23 +00006855#ifdef HAVE_SPAWNV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006856 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
6857 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00006858#if defined(PYOS_OS2)
6859 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
6860 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
6861#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00006862#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +00006863#ifdef HAVE_FORK1
Neal Norwitze241ce82003-02-17 18:17:05 +00006864 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +00006865#endif /* HAVE_FORK1 */
Guido van Rossumad0ee831995-03-01 10:34:45 +00006866#ifdef HAVE_FORK
Neal Norwitze241ce82003-02-17 18:17:05 +00006867 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006868#endif /* HAVE_FORK */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00006869#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Neal Norwitze241ce82003-02-17 18:17:05 +00006870 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +00006871#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00006872#ifdef HAVE_FORKPTY
Neal Norwitze241ce82003-02-17 18:17:05 +00006873 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +00006874#endif /* HAVE_FORKPTY */
Guido van Rossumad0ee831995-03-01 10:34:45 +00006875#ifdef HAVE_GETEGID
Neal Norwitze241ce82003-02-17 18:17:05 +00006876 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006877#endif /* HAVE_GETEGID */
6878#ifdef HAVE_GETEUID
Neal Norwitze241ce82003-02-17 18:17:05 +00006879 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006880#endif /* HAVE_GETEUID */
6881#ifdef HAVE_GETGID
Neal Norwitze241ce82003-02-17 18:17:05 +00006882 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006883#endif /* HAVE_GETGID */
Fred Drakec9680921999-12-13 16:37:25 +00006884#ifdef HAVE_GETGROUPS
Neal Norwitze241ce82003-02-17 18:17:05 +00006885 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00006886#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00006887 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00006888#ifdef HAVE_GETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00006889 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006890#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00006891#ifdef HAVE_GETPPID
Neal Norwitze241ce82003-02-17 18:17:05 +00006892 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006893#endif /* HAVE_GETPPID */
6894#ifdef HAVE_GETUID
Neal Norwitze241ce82003-02-17 18:17:05 +00006895 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006896#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +00006897#ifdef HAVE_GETLOGIN
Neal Norwitze241ce82003-02-17 18:17:05 +00006898 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +00006899#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +00006900#ifdef HAVE_KILL
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006901 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006902#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00006903#ifdef HAVE_KILLPG
6904 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
6905#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +00006906#ifdef HAVE_PLOCK
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006907 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +00006908#endif /* HAVE_PLOCK */
Thomas Heller8b7a9572007-08-31 06:44:36 +00006909#ifdef MS_WINDOWS
6910 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
6911#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00006912#ifdef HAVE_SETUID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006913 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006914#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006915#ifdef HAVE_SETEUID
6916 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
6917#endif /* HAVE_SETEUID */
6918#ifdef HAVE_SETEGID
6919 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
6920#endif /* HAVE_SETEGID */
6921#ifdef HAVE_SETREUID
6922 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
6923#endif /* HAVE_SETREUID */
6924#ifdef HAVE_SETREGID
6925 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
6926#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006927#ifdef HAVE_SETGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006928 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006929#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006930#ifdef HAVE_SETGROUPS
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00006931 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006932#endif /* HAVE_SETGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +00006933#ifdef HAVE_GETPGID
6934 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
6935#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006936#ifdef HAVE_SETPGRP
Neal Norwitze241ce82003-02-17 18:17:05 +00006937 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006938#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00006939#ifdef HAVE_WAIT
Neal Norwitze241ce82003-02-17 18:17:05 +00006940 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00006941#endif /* HAVE_WAIT */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006942#ifdef HAVE_WAIT3
6943 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
6944#endif /* HAVE_WAIT3 */
6945#ifdef HAVE_WAIT4
6946 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
6947#endif /* HAVE_WAIT4 */
Tim Petersab034fa2002-02-01 11:27:43 +00006948#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006949 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006950#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00006951#ifdef HAVE_GETSID
6952 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
6953#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006954#ifdef HAVE_SETSID
Neal Norwitze241ce82003-02-17 18:17:05 +00006955 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006956#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006957#ifdef HAVE_SETPGID
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006958 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006959#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006960#ifdef HAVE_TCGETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006961 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006962#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006963#ifdef HAVE_TCSETPGRP
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006964 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006965#endif /* HAVE_TCSETPGRP */
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006966 {"open", posix_open, METH_VARARGS, posix_open__doc__},
6967 {"close", posix_close, METH_VARARGS, posix_close__doc__},
Christian Heimesfdab48e2008-01-20 09:06:41 +00006968 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00006969 {"device_encoding", device_encoding, METH_VARARGS, device_encoding__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006970 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
6971 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
6972 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
6973 {"read", posix_read, METH_VARARGS, posix_read__doc__},
6974 {"write", posix_write, METH_VARARGS, posix_write__doc__},
6975 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
Skip Montanaro1517d842000-07-19 14:34:14 +00006976 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006977#ifdef HAVE_PIPE
Neal Norwitze241ce82003-02-17 18:17:05 +00006978 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006979#endif
6980#ifdef HAVE_MKFIFO
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006981 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006982#endif
Neal Norwitz11690112002-07-30 01:08:28 +00006983#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006984 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
6985#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006986#ifdef HAVE_DEVICE_MACROS
6987 {"major", posix_major, METH_VARARGS, posix_major__doc__},
6988 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
6989 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
6990#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006991#ifdef HAVE_FTRUNCATE
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006992 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006993#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006994#ifdef HAVE_PUTENV
Fred Drake5ab8eaf1999-12-09 21:13:07 +00006995 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006996#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00006997#ifdef HAVE_UNSETENV
6998 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
6999#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007000 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +00007001#ifdef HAVE_FCHDIR
7002 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
7003#endif
Guido van Rossum21142a01999-01-08 21:05:37 +00007004#ifdef HAVE_FSYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007005 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007006#endif
7007#ifdef HAVE_FDATASYNC
Fred Drake4d1e64b2002-04-15 19:40:07 +00007008 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00007009#endif
Guido van Rossumc9641791998-08-04 15:26:23 +00007010#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +00007011#ifdef WCOREDUMP
7012 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
7013#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00007014#ifdef WIFCONTINUED
7015 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
7016#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +00007017#ifdef WIFSTOPPED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007018 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007019#endif /* WIFSTOPPED */
7020#ifdef WIFSIGNALED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007021 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007022#endif /* WIFSIGNALED */
7023#ifdef WIFEXITED
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007024 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007025#endif /* WIFEXITED */
7026#ifdef WEXITSTATUS
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007027 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007028#endif /* WEXITSTATUS */
7029#ifdef WTERMSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007030 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007031#endif /* WTERMSIG */
7032#ifdef WSTOPSIG
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007033 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00007034#endif /* WSTOPSIG */
7035#endif /* HAVE_SYS_WAIT_H */
Thomas Wouters477c8d52006-05-27 19:21:47 +00007036#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007037 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007038#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00007039#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007040 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00007041#endif
Fred Drakec9680921999-12-13 16:37:25 +00007042#ifdef HAVE_CONFSTR
7043 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
7044#endif
7045#ifdef HAVE_SYSCONF
7046 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
7047#endif
7048#ifdef HAVE_FPATHCONF
7049 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
7050#endif
7051#ifdef HAVE_PATHCONF
7052 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
7053#endif
Neal Norwitze241ce82003-02-17 18:17:05 +00007054 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007055#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00007056 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
7057#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00007058#ifdef HAVE_GETLOADAVG
Neal Norwitze241ce82003-02-17 18:17:05 +00007059 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +00007060#endif
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00007061 #ifdef MS_WINDOWS
7062 {"urandom", win32_urandom, METH_VARARGS, win32_urandom__doc__},
7063 #endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007064 #ifdef __VMS
7065 {"urandom", vms_urandom, METH_VARARGS, vms_urandom__doc__},
7066 #endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007067 {NULL, NULL} /* Sentinel */
7068};
7069
7070
Barry Warsaw4a342091996-12-19 23:50:02 +00007071static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00007072ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +00007073{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007074 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +00007075}
7076
Guido van Rossumd48f2521997-12-05 22:19:34 +00007077#if defined(PYOS_OS2)
7078/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007079static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007080{
7081 APIRET rc;
7082 ULONG values[QSV_MAX+1];
7083 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +00007084 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +00007085
7086 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +00007087 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007088 Py_END_ALLOW_THREADS
7089
7090 if (rc != NO_ERROR) {
7091 os2_error(rc);
7092 return -1;
7093 }
7094
Fred Drake4d1e64b2002-04-15 19:40:07 +00007095 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
7096 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
7097 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
7098 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
7099 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
7100 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
7101 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007102
7103 switch (values[QSV_VERSION_MINOR]) {
7104 case 0: ver = "2.00"; break;
7105 case 10: ver = "2.10"; break;
7106 case 11: ver = "2.11"; break;
7107 case 30: ver = "3.00"; break;
7108 case 40: ver = "4.00"; break;
7109 case 50: ver = "5.00"; break;
7110 default:
Tim Peters885d4572001-11-28 20:27:42 +00007111 PyOS_snprintf(tmp, sizeof(tmp),
7112 "%d-%d", values[QSV_VERSION_MAJOR],
7113 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007114 ver = &tmp[0];
7115 }
7116
7117 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +00007118 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007119 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007120
7121 /* Add Indicator of Which Drive was Used to Boot the System */
7122 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
7123 tmp[1] = ':';
7124 tmp[2] = '\0';
7125
Fred Drake4d1e64b2002-04-15 19:40:07 +00007126 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +00007127}
7128#endif
7129
Barry Warsaw4a342091996-12-19 23:50:02 +00007130static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007131all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +00007132{
Guido van Rossum94f6f721999-01-06 18:42:14 +00007133#ifdef F_OK
7134 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007135#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007136#ifdef R_OK
7137 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007138#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007139#ifdef W_OK
7140 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007141#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007142#ifdef X_OK
7143 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007144#endif
Fred Drakec9680921999-12-13 16:37:25 +00007145#ifdef NGROUPS_MAX
7146 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
7147#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00007148#ifdef TMP_MAX
7149 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
7150#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007151#ifdef WCONTINUED
7152 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
7153#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007154#ifdef WNOHANG
7155 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00007156#endif
Fred Drake106c1a02002-04-23 15:58:02 +00007157#ifdef WUNTRACED
7158 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
7159#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007160#ifdef O_RDONLY
7161 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
7162#endif
7163#ifdef O_WRONLY
7164 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
7165#endif
7166#ifdef O_RDWR
7167 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
7168#endif
7169#ifdef O_NDELAY
7170 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
7171#endif
7172#ifdef O_NONBLOCK
7173 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
7174#endif
7175#ifdef O_APPEND
7176 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
7177#endif
7178#ifdef O_DSYNC
7179 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
7180#endif
7181#ifdef O_RSYNC
7182 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
7183#endif
7184#ifdef O_SYNC
7185 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
7186#endif
7187#ifdef O_NOCTTY
7188 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
7189#endif
7190#ifdef O_CREAT
7191 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
7192#endif
7193#ifdef O_EXCL
7194 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
7195#endif
7196#ifdef O_TRUNC
7197 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
7198#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +00007199#ifdef O_BINARY
7200 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
7201#endif
7202#ifdef O_TEXT
7203 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
7204#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007205#ifdef O_LARGEFILE
7206 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
7207#endif
Skip Montanaro5ff14922005-05-16 02:42:22 +00007208#ifdef O_SHLOCK
7209 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
7210#endif
7211#ifdef O_EXLOCK
7212 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
7213#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007214
Tim Peters5aa91602002-01-30 05:46:57 +00007215/* MS Windows */
7216#ifdef O_NOINHERIT
7217 /* Don't inherit in child processes. */
7218 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
7219#endif
7220#ifdef _O_SHORT_LIVED
7221 /* Optimize for short life (keep in memory). */
7222 /* MS forgot to define this one with a non-underscore form too. */
7223 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
7224#endif
7225#ifdef O_TEMPORARY
7226 /* Automatically delete when last handle is closed. */
7227 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
7228#endif
7229#ifdef O_RANDOM
7230 /* Optimize for random access. */
7231 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
7232#endif
7233#ifdef O_SEQUENTIAL
7234 /* Optimize for sequential access. */
7235 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
7236#endif
7237
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007238/* GNU extensions. */
Alexandre Vassalottibee32532008-05-16 18:15:12 +00007239#ifdef O_ASYNC
7240 /* Send a SIGIO signal whenever input or output
7241 becomes available on file descriptor */
7242 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
7243#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00007244#ifdef O_DIRECT
7245 /* Direct disk access. */
7246 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
7247#endif
7248#ifdef O_DIRECTORY
7249 /* Must be a directory. */
7250 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
7251#endif
7252#ifdef O_NOFOLLOW
7253 /* Do not follow links. */
7254 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
7255#endif
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +00007256#ifdef O_NOATIME
7257 /* Do not update the access time. */
7258 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
7259#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00007260
Barry Warsaw5676bd12003-01-07 20:57:09 +00007261 /* These come from sysexits.h */
7262#ifdef EX_OK
7263 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007264#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007265#ifdef EX_USAGE
7266 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007267#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007268#ifdef EX_DATAERR
7269 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007270#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007271#ifdef EX_NOINPUT
7272 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007273#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007274#ifdef EX_NOUSER
7275 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007276#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007277#ifdef EX_NOHOST
7278 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007279#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007280#ifdef EX_UNAVAILABLE
7281 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007282#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007283#ifdef EX_SOFTWARE
7284 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007285#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007286#ifdef EX_OSERR
7287 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007288#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007289#ifdef EX_OSFILE
7290 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007291#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007292#ifdef EX_CANTCREAT
7293 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007294#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007295#ifdef EX_IOERR
7296 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007297#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007298#ifdef EX_TEMPFAIL
7299 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007300#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007301#ifdef EX_PROTOCOL
7302 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007303#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007304#ifdef EX_NOPERM
7305 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007306#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007307#ifdef EX_CONFIG
7308 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007309#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007310#ifdef EX_NOTFOUND
7311 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00007312#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +00007313
Guido van Rossum246bc171999-02-01 23:54:31 +00007314#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007315#if defined(PYOS_OS2) && defined(PYCC_GCC)
7316 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
7317 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
7318 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
7319 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
7320 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
7321 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
7322 if (ins(d, "P_PM", (long)P_PM)) return -1;
7323 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
7324 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
7325 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
7326 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
7327 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
7328 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
7329 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
7330 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
7331 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
7332 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
7333 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
7334 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
7335 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
7336#else
Guido van Rossum7d385291999-02-16 19:38:04 +00007337 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
7338 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
7339 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
7340 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
7341 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +00007342#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00007343#endif
Guido van Rossum246bc171999-02-01 23:54:31 +00007344
Guido van Rossumd48f2521997-12-05 22:19:34 +00007345#if defined(PYOS_OS2)
7346 if (insertvalues(d)) return -1;
7347#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00007348 return 0;
7349}
7350
7351
Tim Peters5aa91602002-01-30 05:46:57 +00007352#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007353#define INITFUNC PyInit_nt
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007354#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007355
7356#elif defined(PYOS_OS2)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007357#define INITFUNC PyInit_os2
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007358#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +00007359
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007360#else
Martin v. Löwis1a214512008-06-11 05:26:20 +00007361#define INITFUNC PyInit_posix
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007362#define MODNAME "posix"
7363#endif
7364
Martin v. Löwis1a214512008-06-11 05:26:20 +00007365static struct PyModuleDef posixmodule = {
7366 PyModuleDef_HEAD_INIT,
7367 MODNAME,
7368 posix__doc__,
7369 -1,
7370 posix_methods,
7371 NULL,
7372 NULL,
7373 NULL,
7374 NULL
7375};
7376
7377
Mark Hammondfe51c6d2002-08-02 02:27:13 +00007378PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00007379INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +00007380{
Fred Drake4d1e64b2002-04-15 19:40:07 +00007381 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +00007382
Martin v. Löwis1a214512008-06-11 05:26:20 +00007383 m = PyModule_Create(&posixmodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00007384 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007385 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007386
Guido van Rossum0cb96de1997-10-01 04:29:29 +00007387 /* Initialize environ dictionary */
Guido van Rossumb6775db1994-08-01 11:34:53 +00007388 v = convertenviron();
Fred Drake4d1e64b2002-04-15 19:40:07 +00007389 Py_XINCREF(v);
7390 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007391 return NULL;
Barry Warsaw53699e91996-12-10 23:23:01 +00007392 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +00007393
Fred Drake4d1e64b2002-04-15 19:40:07 +00007394 if (all_ins(m))
Martin v. Löwis1a214512008-06-11 05:26:20 +00007395 return NULL;
Barry Warsaw4a342091996-12-19 23:50:02 +00007396
Fred Drake4d1e64b2002-04-15 19:40:07 +00007397 if (setup_confname_tables(m))
Martin v. Löwis1a214512008-06-11 05:26:20 +00007398 return NULL;
Fred Drakebec628d1999-12-15 18:31:10 +00007399
Fred Drake4d1e64b2002-04-15 19:40:07 +00007400 Py_INCREF(PyExc_OSError);
7401 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +00007402
Guido van Rossumb3d39562000-01-31 18:41:26 +00007403#ifdef HAVE_PUTENV
Neil Schemenauer19030a02001-01-16 04:27:47 +00007404 if (posix_putenv_garbage == NULL)
7405 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +00007406#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007407
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007408 if (!initialized) {
7409 stat_result_desc.name = MODNAME ".stat_result";
7410 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
7411 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
7412 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
7413 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
7414 structseq_new = StatResultType.tp_new;
7415 StatResultType.tp_new = statresult_new;
7416
7417 statvfs_result_desc.name = MODNAME ".statvfs_result";
7418 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
7419 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00007420 Py_INCREF((PyObject*) &StatResultType);
7421 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
Fred Drake4d1e64b2002-04-15 19:40:07 +00007422 Py_INCREF((PyObject*) &StatVFSResultType);
7423 PyModule_AddObject(m, "statvfs_result",
7424 (PyObject*) &StatVFSResultType);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007425 initialized = 1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007426
7427#ifdef __APPLE__
7428 /*
7429 * Step 2 of weak-linking support on Mac OS X.
7430 *
7431 * The code below removes functions that are not available on the
7432 * currently active platform.
7433 *
7434 * This block allow one to use a python binary that was build on
7435 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
7436 * OSX 10.4.
7437 */
7438#ifdef HAVE_FSTATVFS
7439 if (fstatvfs == NULL) {
7440 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007441 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007442 }
7443 }
7444#endif /* HAVE_FSTATVFS */
7445
7446#ifdef HAVE_STATVFS
7447 if (statvfs == NULL) {
7448 if (PyObject_DelAttrString(m, "statvfs") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007449 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007450 }
7451 }
7452#endif /* HAVE_STATVFS */
7453
7454# ifdef HAVE_LCHOWN
7455 if (lchown == NULL) {
7456 if (PyObject_DelAttrString(m, "lchown") == -1) {
Martin v. Löwis1a214512008-06-11 05:26:20 +00007457 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007458 }
7459 }
7460#endif /* HAVE_LCHOWN */
7461
7462
7463#endif /* __APPLE__ */
Martin v. Löwis1a214512008-06-11 05:26:20 +00007464 return m;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007465
Guido van Rossumb6775db1994-08-01 11:34:53 +00007466}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007467
7468#ifdef __cplusplus
7469}
7470#endif