blob: 73f62f63efaf45cb49822f8ee9ca9a46a754b02d [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
Thomas Wouters477c8d52006-05-27 19:21:47 +000014#ifdef __APPLE__
15 /*
Victor Stinner8c62be82010-05-06 00:08:46 +000016 * Step 1 of support for weak-linking a number of symbols existing on
Thomas Wouters477c8d52006-05-27 19:21:47 +000017 * OSX 10.4 and later, see the comment in the #ifdef __APPLE__ block
18 * at the end of this file for more information.
19 */
20# pragma weak lchown
21# pragma weak statvfs
22# pragma weak fstatvfs
23
24#endif /* __APPLE__ */
25
Thomas Wouters68bc4f92006-03-01 01:05:10 +000026#define PY_SSIZE_T_CLEAN
27
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000028#include "Python.h"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000029
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000030#if defined(__VMS)
Victor Stinnerb90db4c2011-04-26 22:48:24 +020031# error "PEP 11: VMS is now unsupported, code will be removed in Python 3.4"
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000032# include <unixio.h>
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000033#endif /* defined(__VMS) */
34
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000035#ifdef __cplusplus
36extern "C" {
37#endif
38
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000039PyDoc_STRVAR(posix__doc__,
Guido van Rossumec4f4ac1997-06-02 22:20:51 +000040"This module provides access to operating system functionality that is\n\
41standardized by the C Standard and the POSIX standard (a thinly\n\
42disguised Unix interface). Refer to the library manual and\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000043corresponding Unix manual entries for more information on calls.");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000044
Martin v. Löwis0073f2e2002-11-21 23:52:35 +000045
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000046#if defined(PYOS_OS2)
Victor Stinnerb90db4c2011-04-26 22:48:24 +020047#error "PEP 11: OS/2 is now unsupported, code will be removed in Python 3.4"
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000048#define INCL_DOS
49#define INCL_DOSERRORS
50#define INCL_DOSPROCESS
51#define INCL_NOPMAPI
52#include <os2.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000053#if defined(PYCC_GCC)
54#include <ctype.h>
55#include <io.h>
56#include <stdio.h>
57#include <process.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000058#endif
Andrew MacIntyreda4d6cb2004-03-29 11:53:38 +000059#include "osdefs.h"
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000060#endif
61
Ross Lagerwall4d076da2011-03-18 06:56:53 +020062#ifdef HAVE_SYS_UIO_H
63#include <sys/uio.h>
64#endif
65
Thomas Wouters0e3f5912006-08-11 14:57:12 +000066#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000067#include <sys/types.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000068#endif /* HAVE_SYS_TYPES_H */
69
70#ifdef HAVE_SYS_STAT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000071#include <sys/stat.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000072#endif /* HAVE_SYS_STAT_H */
Guido van Rossuma6535fd2001-10-18 19:44:10 +000073
Guido van Rossum36bc6801995-06-14 22:54:23 +000074#ifdef HAVE_SYS_WAIT_H
Victor Stinner8c62be82010-05-06 00:08:46 +000075#include <sys/wait.h> /* For WNOHANG */
Guido van Rossum36bc6801995-06-14 22:54:23 +000076#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +000077
Thomas Wouters0e3f5912006-08-11 14:57:12 +000078#ifdef HAVE_SIGNAL_H
Guido van Rossuma376cc51996-12-05 23:43:35 +000079#include <signal.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000080#endif
Guido van Rossuma376cc51996-12-05 23:43:35 +000081
Guido van Rossumb6775db1994-08-01 11:34:53 +000082#ifdef HAVE_FCNTL_H
83#include <fcntl.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000084#endif /* HAVE_FCNTL_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +000085
Guido van Rossuma6535fd2001-10-18 19:44:10 +000086#ifdef HAVE_GRP_H
87#include <grp.h>
88#endif
89
Barry Warsaw5676bd12003-01-07 20:57:09 +000090#ifdef HAVE_SYSEXITS_H
91#include <sysexits.h>
92#endif /* HAVE_SYSEXITS_H */
93
Anthony Baxter8a560de2004-10-13 15:30:56 +000094#ifdef HAVE_SYS_LOADAVG_H
95#include <sys/loadavg.h>
96#endif
97
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +000098#ifdef HAVE_LANGINFO_H
99#include <langinfo.h>
100#endif
101
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000102#ifdef HAVE_SYS_SENDFILE_H
103#include <sys/sendfile.h>
104#endif
105
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500106#ifdef HAVE_SCHED_H
107#include <sched.h>
108#endif
109
Benjamin Peterson2dbda072012-03-16 10:12:55 -0500110#if !defined(CPU_ALLOC) && defined(HAVE_SCHED_SETAFFINITY)
Benjamin Peterson7b51b8d2012-03-14 22:28:25 -0500111#undef HAVE_SCHED_SETAFFINITY
112#endif
113
Benjamin Peterson9428d532011-09-14 11:45:52 -0400114#if defined(HAVE_SYS_XATTR_H) && defined(__GLIBC__)
115#define USE_XATTRS
116#endif
117
118#ifdef USE_XATTRS
Benjamin Petersonb77fe172011-09-13 17:20:47 -0400119#include <sys/xattr.h>
Benjamin Peterson799bd802011-08-31 22:15:17 -0400120#endif
121
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000122#if defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__)
123#ifdef HAVE_SYS_SOCKET_H
124#include <sys/socket.h>
125#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000126#endif
127
Victor Stinner8b905bd2011-10-25 13:34:04 +0200128#ifdef HAVE_DLFCN_H
129#include <dlfcn.h>
130#endif
131
Antoine Pitroubcf2b592012-02-08 23:28:36 +0100132#if defined(MS_WINDOWS)
133# define TERMSIZE_USE_CONIO
134#elif defined(HAVE_SYS_IOCTL_H)
135# include <sys/ioctl.h>
136# if defined(HAVE_TERMIOS_H)
137# include <termios.h>
138# endif
139# if defined(TIOCGWINSZ)
140# define TERMSIZE_USE_IOCTL
141# endif
142#endif /* MS_WINDOWS */
143
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000144/* Various compilers have only certain posix functions */
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +0000145/* XXX Gosh I wish these were all moved into pyconfig.h */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000146#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000147#include <process.h>
148#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000149#if defined(__WATCOMC__) && !defined(__QNX__) /* Watcom compiler */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000150#define HAVE_GETCWD 1
151#define HAVE_OPENDIR 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000152#define HAVE_SYSTEM 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000153#if defined(__OS2__)
154#define HAVE_EXECV 1
155#define HAVE_WAIT 1
Guido van Rossumad0ee831995-03-01 10:34:45 +0000156#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000157#include <process.h>
158#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000159#ifdef __BORLANDC__ /* Borland compiler */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000160#define HAVE_EXECV 1
161#define HAVE_GETCWD 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000162#define HAVE_OPENDIR 1
163#define HAVE_PIPE 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000164#define HAVE_SYSTEM 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000165#define HAVE_WAIT 1
166#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000167#ifdef _MSC_VER /* Microsoft compiler */
Guido van Rossum8d665e61996-06-26 18:22:49 +0000168#define HAVE_GETCWD 1
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +0000169#define HAVE_GETPPID 1
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000170#define HAVE_GETLOGIN 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000171#define HAVE_SPAWNV 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000172#define HAVE_EXECV 1
173#define HAVE_PIPE 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000174#define HAVE_SYSTEM 1
175#define HAVE_CWAIT 1
176#define HAVE_FSYNC 1
Tim Peters11b23062003-04-23 02:39:17 +0000177#define fsync _commit
Andrew MacIntyre6c73af22002-03-03 03:07:07 +0000178#else
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000179#if defined(PYOS_OS2) && defined(PYCC_GCC) || defined(__VMS)
180/* Everything needed is defined in PC/os2emx/pyconfig.h or vms/pyconfig.h */
Victor Stinner8c62be82010-05-06 00:08:46 +0000181#else /* all other compilers */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000182/* Unix functions that the configure script doesn't check for */
183#define HAVE_EXECV 1
184#define HAVE_FORK 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000185#if defined(__USLC__) && defined(__SCO_VERSION__) /* SCO UDK Compiler */
Guido van Rossum2242f2f2001-04-11 20:58:20 +0000186#define HAVE_FORK1 1
187#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000188#define HAVE_GETCWD 1
189#define HAVE_GETEGID 1
190#define HAVE_GETEUID 1
191#define HAVE_GETGID 1
192#define HAVE_GETPPID 1
193#define HAVE_GETUID 1
194#define HAVE_KILL 1
195#define HAVE_OPENDIR 1
196#define HAVE_PIPE 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000197#define HAVE_SYSTEM 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000198#define HAVE_WAIT 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000199#define HAVE_TTYNAME 1
Martin v. Löwis7a924e62003-03-05 14:15:21 +0000200#endif /* PYOS_OS2 && PYCC_GCC && __VMS */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000201#endif /* _MSC_VER */
202#endif /* __BORLANDC__ */
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000203#endif /* ! __WATCOMC__ || __QNX__ */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000204#endif /* ! __IBMC__ */
Guido van Rossumad0ee831995-03-01 10:34:45 +0000205
Victor Stinnera2f7c002012-02-08 03:36:25 +0100206
207
208
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000209#ifndef _MSC_VER
Guido van Rossum36bc6801995-06-14 22:54:23 +0000210
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000211#if defined(__sgi)&&_COMPILER_VERSION>=700
212/* declare ctermid_r if compiling with MIPSPro 7.x in ANSI C mode
213 (default) */
214extern char *ctermid_r(char *);
215#endif
216
Thomas Wouters1e0c2f42000-07-24 16:06:23 +0000217#ifndef HAVE_UNISTD_H
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000218#if defined(PYCC_VACPP)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000219extern int mkdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000220#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000221#if ( defined(__WATCOMC__) || defined(_MSC_VER) ) && !defined(__QNX__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000222extern int mkdir(const char *);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000223#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000224extern int mkdir(const char *, mode_t);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000225#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000226#endif
227#if defined(__IBMC__) || defined(__IBMCPP__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000228extern int chdir(char *);
229extern int rmdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000230#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000231extern int chdir(const char *);
232extern int rmdir(const char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000233#endif
Tim Peters58e0a8c2001-05-14 22:32:33 +0000234#ifdef __BORLANDC__
235extern int chmod(const char *, int);
236#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000237extern int chmod(const char *, mode_t);
Tim Peters58e0a8c2001-05-14 22:32:33 +0000238#endif
Christian Heimes4e30a842007-11-30 22:12:06 +0000239/*#ifdef HAVE_FCHMOD
240extern int fchmod(int, mode_t);
241#endif*/
242/*#ifdef HAVE_LCHMOD
243extern int lchmod(const char *, mode_t);
244#endif*/
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000245extern int chown(const char *, uid_t, gid_t);
246extern char *getcwd(char *, int);
247extern char *strerror(int);
248extern int link(const char *, const char *);
249extern int rename(const char *, const char *);
250extern int stat(const char *, struct stat *);
251extern int unlink(const char *);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000252#ifdef HAVE_SYMLINK
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000253extern int symlink(const char *, const char *);
Guido van Rossuma38a5031995-02-17 15:11:36 +0000254#endif /* HAVE_SYMLINK */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000255#ifdef HAVE_LSTAT
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000256extern int lstat(const char *, struct stat *);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000257#endif /* HAVE_LSTAT */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000258#endif /* !HAVE_UNISTD_H */
Guido van Rossum36bc6801995-06-14 22:54:23 +0000259
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000260#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000261
Guido van Rossumb6775db1994-08-01 11:34:53 +0000262#ifdef HAVE_UTIME_H
263#include <utime.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000264#endif /* HAVE_UTIME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000265
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000266#ifdef HAVE_SYS_UTIME_H
267#include <sys/utime.h>
268#define HAVE_UTIME_H /* pretend we do for the rest of this file */
269#endif /* HAVE_SYS_UTIME_H */
270
Guido van Rossumb6775db1994-08-01 11:34:53 +0000271#ifdef HAVE_SYS_TIMES_H
272#include <sys/times.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000273#endif /* HAVE_SYS_TIMES_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000274
275#ifdef HAVE_SYS_PARAM_H
276#include <sys/param.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000277#endif /* HAVE_SYS_PARAM_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000278
279#ifdef HAVE_SYS_UTSNAME_H
280#include <sys/utsname.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000281#endif /* HAVE_SYS_UTSNAME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000282
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000283#ifdef HAVE_DIRENT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000284#include <dirent.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000285#define NAMLEN(dirent) strlen((dirent)->d_name)
286#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000287#if defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000288#include <direct.h>
289#define NAMLEN(dirent) strlen((dirent)->d_name)
290#else
Guido van Rossumb6775db1994-08-01 11:34:53 +0000291#define dirent direct
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000292#define NAMLEN(dirent) (dirent)->d_namlen
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000293#endif
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000294#ifdef HAVE_SYS_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000295#include <sys/ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000296#endif
297#ifdef HAVE_SYS_DIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000298#include <sys/dir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000299#endif
300#ifdef HAVE_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000301#include <ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000302#endif
303#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000304
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000305#ifdef _MSC_VER
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000306#ifdef HAVE_DIRECT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000307#include <direct.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000308#endif
309#ifdef HAVE_IO_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000310#include <io.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000311#endif
312#ifdef HAVE_PROCESS_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000313#include <process.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000314#endif
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000315#ifndef VOLUME_NAME_DOS
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +0000316#define VOLUME_NAME_DOS 0x0
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000317#endif
318#ifndef VOLUME_NAME_NT
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +0000319#define VOLUME_NAME_NT 0x2
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000320#endif
321#ifndef IO_REPARSE_TAG_SYMLINK
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +0000322#define IO_REPARSE_TAG_SYMLINK (0xA000000CL)
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000323#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000324#include "osdefs.h"
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000325#include <malloc.h>
Guido van Rossumb6775db1994-08-01 11:34:53 +0000326#include <windows.h>
Victor Stinner8c62be82010-05-06 00:08:46 +0000327#include <shellapi.h> /* for ShellExecute() */
Brian Curtine8e4b3b2010-09-23 20:04:14 +0000328#include <lmcons.h> /* for UNLEN */
Brian Curtin52173d42010-12-02 18:29:18 +0000329#ifdef SE_CREATE_SYMBOLIC_LINK_NAME /* Available starting with Vista */
330#define HAVE_SYMLINK
Brian Curtin3b4499c2010-12-28 14:31:47 +0000331static int win32_can_symlink = 0;
Brian Curtin52173d42010-12-02 18:29:18 +0000332#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000333#endif /* _MSC_VER */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000334
Guido van Rossumd48f2521997-12-05 22:19:34 +0000335#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000336#include <io.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000337#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000338
Tim Petersbc2e10e2002-03-03 23:17:02 +0000339#ifndef MAXPATHLEN
Thomas Wouters477c8d52006-05-27 19:21:47 +0000340#if defined(PATH_MAX) && PATH_MAX > 1024
341#define MAXPATHLEN PATH_MAX
342#else
Tim Petersbc2e10e2002-03-03 23:17:02 +0000343#define MAXPATHLEN 1024
Thomas Wouters477c8d52006-05-27 19:21:47 +0000344#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000345#endif /* MAXPATHLEN */
346
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000347#ifdef UNION_WAIT
348/* Emulate some macros on systems that have a union instead of macros */
349
350#ifndef WIFEXITED
351#define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
352#endif
353
354#ifndef WEXITSTATUS
355#define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
356#endif
357
358#ifndef WTERMSIG
359#define WTERMSIG(u_wait) ((u_wait).w_termsig)
360#endif
361
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000362#define WAIT_TYPE union wait
363#define WAIT_STATUS_INT(s) (s.w_status)
364
365#else /* !UNION_WAIT */
366#define WAIT_TYPE int
367#define WAIT_STATUS_INT(s) (s)
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000368#endif /* UNION_WAIT */
369
Greg Wardb48bc172000-03-01 21:51:56 +0000370/* Don't use the "_r" form if we don't need it (also, won't have a
371 prototype for it, at least on Solaris -- maybe others as well?). */
372#if defined(HAVE_CTERMID_R) && defined(WITH_THREAD)
373#define USE_CTERMID_R
374#endif
375
Fred Drake699f3522000-06-29 21:12:41 +0000376/* choose the appropriate stat and fstat functions and return structs */
Guido van Rossum64529cd2000-06-30 22:45:12 +0000377#undef STAT
Antoine Pitroue47e0932011-01-19 15:21:35 +0000378#undef FSTAT
379#undef STRUCT_STAT
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000380#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +0000381# define STAT win32_stat
382# define FSTAT win32_fstat
383# define STRUCT_STAT struct win32_stat
Fred Drake699f3522000-06-29 21:12:41 +0000384#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000385# define STAT stat
386# define FSTAT fstat
387# define STRUCT_STAT struct stat
Fred Drake699f3522000-06-29 21:12:41 +0000388#endif
389
Tim Peters11b23062003-04-23 02:39:17 +0000390#if defined(MAJOR_IN_MKDEV)
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000391#include <sys/mkdev.h>
392#else
393#if defined(MAJOR_IN_SYSMACROS)
394#include <sys/sysmacros.h>
395#endif
Neal Norwitz3d949422002-04-20 13:46:43 +0000396#if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
397#include <sys/mkdev.h>
398#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000399#endif
Fred Drake699f3522000-06-29 21:12:41 +0000400
Ross Lagerwallb1e5d592011-09-19 08:30:43 +0200401/* A helper used by a number of POSIX-only functions */
402#ifndef MS_WINDOWS
Georg Brandla391b112011-02-25 15:23:18 +0000403static int
404_parse_off_t(PyObject* arg, void* addr)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000405{
406#if !defined(HAVE_LARGEFILE_SUPPORT)
407 *((off_t*)addr) = PyLong_AsLong(arg);
408#else
Antoine Pitroudcc20b82011-02-26 13:38:35 +0000409 *((off_t*)addr) = PyLong_AsLongLong(arg);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000410#endif
411 if (PyErr_Occurred())
412 return 0;
413 return 1;
414}
Ross Lagerwallb1e5d592011-09-19 08:30:43 +0200415#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000416
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000417#if defined _MSC_VER && _MSC_VER >= 1400
418/* Microsoft CRT in VS2005 and higher will verify that a filehandle is
419 * valid and throw an assertion if it isn't.
420 * Normally, an invalid fd is likely to be a C program error and therefore
421 * an assertion can be useful, but it does contradict the POSIX standard
422 * which for write(2) states:
423 * "Otherwise, -1 shall be returned and errno set to indicate the error."
424 * "[EBADF] The fildes argument is not a valid file descriptor open for
425 * writing."
426 * Furthermore, python allows the user to enter any old integer
427 * as a fd and should merely raise a python exception on error.
428 * The Microsoft CRT doesn't provide an official way to check for the
429 * validity of a file descriptor, but we can emulate its internal behaviour
Victor Stinner8c62be82010-05-06 00:08:46 +0000430 * by using the exported __pinfo data member and knowledge of the
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000431 * internal structures involved.
432 * The structures below must be updated for each version of visual studio
433 * according to the file internal.h in the CRT source, until MS comes
434 * up with a less hacky way to do this.
435 * (all of this is to avoid globally modifying the CRT behaviour using
436 * _set_invalid_parameter_handler() and _CrtSetReportMode())
437 */
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000438/* The actual size of the structure is determined at runtime.
439 * Only the first items must be present.
440 */
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000441typedef struct {
Victor Stinner8c62be82010-05-06 00:08:46 +0000442 intptr_t osfhnd;
443 char osfile;
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000444} my_ioinfo;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000445
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000446extern __declspec(dllimport) char * __pioinfo[];
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000447#define IOINFO_L2E 5
448#define IOINFO_ARRAY_ELTS (1 << IOINFO_L2E)
449#define IOINFO_ARRAYS 64
450#define _NHANDLE_ (IOINFO_ARRAYS * IOINFO_ARRAY_ELTS)
451#define FOPEN 0x01
452#define _NO_CONSOLE_FILENO (intptr_t)-2
453
454/* This function emulates what the windows CRT does to validate file handles */
455int
456_PyVerify_fd(int fd)
457{
Victor Stinner8c62be82010-05-06 00:08:46 +0000458 const int i1 = fd >> IOINFO_L2E;
459 const int i2 = fd & ((1 << IOINFO_L2E) - 1);
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000460
Antoine Pitrou22e41552010-08-15 18:07:50 +0000461 static size_t sizeof_ioinfo = 0;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000462
Victor Stinner8c62be82010-05-06 00:08:46 +0000463 /* Determine the actual size of the ioinfo structure,
464 * as used by the CRT loaded in memory
465 */
466 if (sizeof_ioinfo == 0 && __pioinfo[0] != NULL) {
467 sizeof_ioinfo = _msize(__pioinfo[0]) / IOINFO_ARRAY_ELTS;
468 }
469 if (sizeof_ioinfo == 0) {
470 /* This should not happen... */
471 goto fail;
472 }
473
474 /* See that it isn't a special CLEAR fileno */
475 if (fd != _NO_CONSOLE_FILENO) {
476 /* Microsoft CRT would check that 0<=fd<_nhandle but we can't do that. Instead
477 * we check pointer validity and other info
478 */
479 if (0 <= i1 && i1 < IOINFO_ARRAYS && __pioinfo[i1] != NULL) {
480 /* finally, check that the file is open */
481 my_ioinfo* info = (my_ioinfo*)(__pioinfo[i1] + i2 * sizeof_ioinfo);
482 if (info->osfile & FOPEN) {
483 return 1;
484 }
485 }
486 }
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000487 fail:
Victor Stinner8c62be82010-05-06 00:08:46 +0000488 errno = EBADF;
489 return 0;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000490}
491
492/* the special case of checking dup2. The target fd must be in a sensible range */
493static int
494_PyVerify_fd_dup2(int fd1, int fd2)
495{
Victor Stinner8c62be82010-05-06 00:08:46 +0000496 if (!_PyVerify_fd(fd1))
497 return 0;
498 if (fd2 == _NO_CONSOLE_FILENO)
499 return 0;
500 if ((unsigned)fd2 < _NHANDLE_)
501 return 1;
502 else
503 return 0;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000504}
505#else
506/* dummy version. _PyVerify_fd() is already defined in fileobject.h */
507#define _PyVerify_fd_dup2(A, B) (1)
508#endif
509
Brian Curtinfc1be6d2010-11-24 13:23:18 +0000510#ifdef MS_WINDOWS
Brian Curtinf5e76d02010-11-24 13:14:05 +0000511/* The following structure was copied from
512 http://msdn.microsoft.com/en-us/library/ms791514.aspx as the required
513 include doesn't seem to be present in the Windows SDK (at least as included
514 with Visual Studio Express). */
515typedef struct _REPARSE_DATA_BUFFER {
516 ULONG ReparseTag;
517 USHORT ReparseDataLength;
518 USHORT Reserved;
519 union {
520 struct {
521 USHORT SubstituteNameOffset;
522 USHORT SubstituteNameLength;
523 USHORT PrintNameOffset;
524 USHORT PrintNameLength;
525 ULONG Flags;
526 WCHAR PathBuffer[1];
527 } SymbolicLinkReparseBuffer;
528
529 struct {
530 USHORT SubstituteNameOffset;
531 USHORT SubstituteNameLength;
532 USHORT PrintNameOffset;
533 USHORT PrintNameLength;
534 WCHAR PathBuffer[1];
535 } MountPointReparseBuffer;
536
537 struct {
538 UCHAR DataBuffer[1];
539 } GenericReparseBuffer;
540 };
541} REPARSE_DATA_BUFFER, *PREPARSE_DATA_BUFFER;
542
543#define REPARSE_DATA_BUFFER_HEADER_SIZE FIELD_OFFSET(REPARSE_DATA_BUFFER,\
544 GenericReparseBuffer)
545#define MAXIMUM_REPARSE_DATA_BUFFER_SIZE ( 16 * 1024 )
546
547static int
Brian Curtind25aef52011-06-13 15:16:04 -0500548win32_get_reparse_tag(HANDLE reparse_point_handle, ULONG *reparse_tag)
Brian Curtinf5e76d02010-11-24 13:14:05 +0000549{
550 char target_buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
551 REPARSE_DATA_BUFFER *rdb = (REPARSE_DATA_BUFFER *)target_buffer;
552 DWORD n_bytes_returned;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000553
554 if (0 == DeviceIoControl(
555 reparse_point_handle,
556 FSCTL_GET_REPARSE_POINT,
557 NULL, 0, /* in buffer */
558 target_buffer, sizeof(target_buffer),
559 &n_bytes_returned,
560 NULL)) /* we're not using OVERLAPPED_IO */
Brian Curtind25aef52011-06-13 15:16:04 -0500561 return FALSE;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000562
563 if (reparse_tag)
564 *reparse_tag = rdb->ReparseTag;
565
Brian Curtind25aef52011-06-13 15:16:04 -0500566 return TRUE;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000567}
Victor Stinner1ab6c2d2011-11-15 22:27:41 +0100568
569static int
570win32_warn_bytes_api()
571{
572 return PyErr_WarnEx(PyExc_DeprecationWarning,
573 "The Windows bytes API has been deprecated, "
574 "use Unicode filenames instead",
575 1);
576}
577
578static PyObject*
579win32_decode_filename(PyObject *obj)
580{
581 PyObject *unicode;
582 if (PyUnicode_Check(obj)) {
583 if (PyUnicode_READY(obj))
584 return NULL;
585 Py_INCREF(obj);
586 return obj;
587 }
588 if (!PyUnicode_FSDecoder(obj, &unicode))
589 return NULL;
590 if (win32_warn_bytes_api()) {
591 Py_DECREF(unicode);
592 return NULL;
593 }
594 return unicode;
595}
Brian Curtinfc1be6d2010-11-24 13:23:18 +0000596#endif /* MS_WINDOWS */
Brian Curtinf5e76d02010-11-24 13:14:05 +0000597
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000598/* Return a dictionary corresponding to the POSIX environment table */
Jack Jansenea0c3822002-08-01 21:57:49 +0000599#ifdef WITH_NEXT_FRAMEWORK
600/* On Darwin/MacOSX a shared library or framework has no access to
601** environ directly, we must obtain it with _NSGetEnviron().
602*/
603#include <crt_externs.h>
604static char **environ;
605#elif !defined(_MSC_VER) && ( !defined(__WATCOMC__) || defined(__QNX__) )
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000606extern char **environ;
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000607#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000608
Barry Warsaw53699e91996-12-10 23:23:01 +0000609static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000610convertenviron(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000611{
Victor Stinner8c62be82010-05-06 00:08:46 +0000612 PyObject *d;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000613#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +0000614 wchar_t **e;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000615#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000616 char **e;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000617#endif
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +0000618#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +0000619 APIRET rc;
620 char buffer[1024]; /* OS/2 Provides a Documented Max of 1024 Chars */
621#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +0000622
Victor Stinner8c62be82010-05-06 00:08:46 +0000623 d = PyDict_New();
624 if (d == NULL)
625 return NULL;
626#ifdef WITH_NEXT_FRAMEWORK
627 if (environ == NULL)
628 environ = *_NSGetEnviron();
629#endif
630#ifdef MS_WINDOWS
631 /* _wenviron must be initialized in this way if the program is started
632 through main() instead of wmain(). */
633 _wgetenv(L"");
634 if (_wenviron == NULL)
635 return d;
636 /* This part ignores errors */
637 for (e = _wenviron; *e != NULL; e++) {
638 PyObject *k;
639 PyObject *v;
640 wchar_t *p = wcschr(*e, L'=');
641 if (p == NULL)
642 continue;
643 k = PyUnicode_FromWideChar(*e, (Py_ssize_t)(p-*e));
644 if (k == NULL) {
645 PyErr_Clear();
646 continue;
Guido van Rossumd48f2521997-12-05 22:19:34 +0000647 }
Victor Stinner8c62be82010-05-06 00:08:46 +0000648 v = PyUnicode_FromWideChar(p+1, wcslen(p+1));
649 if (v == NULL) {
650 PyErr_Clear();
651 Py_DECREF(k);
652 continue;
Guido van Rossumd48f2521997-12-05 22:19:34 +0000653 }
Victor Stinner8c62be82010-05-06 00:08:46 +0000654 if (PyDict_GetItem(d, k) == NULL) {
655 if (PyDict_SetItem(d, k, v) != 0)
656 PyErr_Clear();
657 }
658 Py_DECREF(k);
659 Py_DECREF(v);
660 }
661#else
662 if (environ == NULL)
663 return d;
664 /* This part ignores errors */
665 for (e = environ; *e != NULL; e++) {
666 PyObject *k;
667 PyObject *v;
668 char *p = strchr(*e, '=');
669 if (p == NULL)
670 continue;
Victor Stinner84ae1182010-05-06 22:05:07 +0000671 k = PyBytes_FromStringAndSize(*e, (int)(p-*e));
Victor Stinner8c62be82010-05-06 00:08:46 +0000672 if (k == NULL) {
673 PyErr_Clear();
674 continue;
675 }
Victor Stinner84ae1182010-05-06 22:05:07 +0000676 v = PyBytes_FromStringAndSize(p+1, strlen(p+1));
Victor Stinner8c62be82010-05-06 00:08:46 +0000677 if (v == NULL) {
678 PyErr_Clear();
679 Py_DECREF(k);
680 continue;
681 }
682 if (PyDict_GetItem(d, k) == NULL) {
683 if (PyDict_SetItem(d, k, v) != 0)
684 PyErr_Clear();
685 }
686 Py_DECREF(k);
687 Py_DECREF(v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000688 }
689#endif
Victor Stinner8c62be82010-05-06 00:08:46 +0000690#if defined(PYOS_OS2)
691 rc = DosQueryExtLIBPATH(buffer, BEGIN_LIBPATH);
692 if (rc == NO_ERROR) { /* (not a type, envname is NOT 'BEGIN_LIBPATH') */
693 PyObject *v = PyBytes_FromString(buffer);
694 PyDict_SetItemString(d, "BEGINLIBPATH", v);
695 Py_DECREF(v);
696 }
697 rc = DosQueryExtLIBPATH(buffer, END_LIBPATH);
698 if (rc == NO_ERROR) { /* (not a typo, envname is NOT 'END_LIBPATH') */
699 PyObject *v = PyBytes_FromString(buffer);
700 PyDict_SetItemString(d, "ENDLIBPATH", v);
701 Py_DECREF(v);
702 }
703#endif
704 return d;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000705}
706
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000707/* Set a POSIX-specific error from errno, and return NULL */
708
Barry Warsawd58d7641998-07-23 16:14:40 +0000709static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000710posix_error(void)
Guido van Rossumad0ee831995-03-01 10:34:45 +0000711{
Victor Stinner8c62be82010-05-06 00:08:46 +0000712 return PyErr_SetFromErrno(PyExc_OSError);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000713}
Barry Warsawd58d7641998-07-23 16:14:40 +0000714static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000715posix_error_with_filename(char* name)
Barry Warsawd58d7641998-07-23 16:14:40 +0000716{
Victor Stinner8c62be82010-05-06 00:08:46 +0000717 return PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
Barry Warsawd58d7641998-07-23 16:14:40 +0000718}
719
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000720
Mark Hammondef8b6542001-05-13 08:04:26 +0000721static PyObject *
Martin v. Löwis011e8422009-05-05 04:43:17 +0000722posix_error_with_allocated_filename(PyObject* name)
Mark Hammondef8b6542001-05-13 08:04:26 +0000723{
Victor Stinner77ccd6d2010-05-08 00:36:42 +0000724 PyObject *name_str, *rc;
725 name_str = PyUnicode_DecodeFSDefaultAndSize(PyBytes_AsString(name),
726 PyBytes_GET_SIZE(name));
Victor Stinner8c62be82010-05-06 00:08:46 +0000727 Py_DECREF(name);
Victor Stinner77ccd6d2010-05-08 00:36:42 +0000728 rc = PyErr_SetFromErrnoWithFilenameObject(PyExc_OSError,
729 name_str);
730 Py_XDECREF(name_str);
Victor Stinner8c62be82010-05-06 00:08:46 +0000731 return rc;
Mark Hammondef8b6542001-05-13 08:04:26 +0000732}
733
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000734#ifdef MS_WINDOWS
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000735static PyObject *
Brian Curtind40e6f72010-07-08 21:39:08 +0000736win32_error(char* function, const char* filename)
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000737{
Victor Stinner8c62be82010-05-06 00:08:46 +0000738 /* XXX We should pass the function name along in the future.
739 (winreg.c also wants to pass the function name.)
740 This would however require an additional param to the
741 Windows error object, which is non-trivial.
742 */
743 errno = GetLastError();
744 if (filename)
745 return PyErr_SetFromWindowsErrWithFilename(errno, filename);
746 else
747 return PyErr_SetFromWindowsErr(errno);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000748}
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000749
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000750static PyObject *
Victor Stinnereb5657a2011-09-30 01:44:27 +0200751win32_error_unicode(char* function, wchar_t* filename)
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000752{
Victor Stinner8c62be82010-05-06 00:08:46 +0000753 /* XXX - see win32_error for comments on 'function' */
754 errno = GetLastError();
755 if (filename)
756 return PyErr_SetFromWindowsErrWithUnicodeFilename(errno, filename);
757 else
758 return PyErr_SetFromWindowsErr(errno);
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000759}
760
Victor Stinnereb5657a2011-09-30 01:44:27 +0200761static PyObject *
762win32_error_object(char* function, PyObject* filename)
763{
764 /* XXX - see win32_error for comments on 'function' */
765 errno = GetLastError();
766 if (filename)
767 return PyErr_SetExcFromWindowsErrWithFilenameObject(
768 PyExc_WindowsError,
769 errno,
770 filename);
771 else
772 return PyErr_SetFromWindowsErr(errno);
773}
774
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000775#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000776
Guido van Rossumd48f2521997-12-05 22:19:34 +0000777#if defined(PYOS_OS2)
778/**********************************************************************
779 * Helper Function to Trim and Format OS/2 Messages
780 **********************************************************************/
Victor Stinner8c62be82010-05-06 00:08:46 +0000781static void
Guido van Rossumd48f2521997-12-05 22:19:34 +0000782os2_formatmsg(char *msgbuf, int msglen, char *reason)
783{
784 msgbuf[msglen] = '\0'; /* OS/2 Doesn't Guarantee a Terminator */
785
786 if (strlen(msgbuf) > 0) { /* If Non-Empty Msg, Trim CRLF */
787 char *lastc = &msgbuf[ strlen(msgbuf)-1 ];
788
Neal Norwitz30b5c5d2005-12-19 06:05:18 +0000789 while (lastc > msgbuf && isspace(Py_CHARMASK(*lastc)))
Guido van Rossumd48f2521997-12-05 22:19:34 +0000790 *lastc-- = '\0'; /* Trim Trailing Whitespace (CRLF) */
791 }
792
793 /* Add Optional Reason Text */
794 if (reason) {
795 strcat(msgbuf, " : ");
796 strcat(msgbuf, reason);
797 }
798}
799
800/**********************************************************************
801 * Decode an OS/2 Operating System Error Code
802 *
803 * A convenience function to lookup an OS/2 error code and return a
804 * text message we can use to raise a Python exception.
805 *
806 * Notes:
807 * The messages for errors returned from the OS/2 kernel reside in
808 * the file OSO001.MSG in the \OS2 directory hierarchy.
809 *
810 **********************************************************************/
Victor Stinner8c62be82010-05-06 00:08:46 +0000811static char *
Guido van Rossumd48f2521997-12-05 22:19:34 +0000812os2_strerror(char *msgbuf, int msgbuflen, int errorcode, char *reason)
813{
814 APIRET rc;
815 ULONG msglen;
816
817 /* Retrieve Kernel-Related Error Message from OSO001.MSG File */
818 Py_BEGIN_ALLOW_THREADS
819 rc = DosGetMessage(NULL, 0, msgbuf, msgbuflen,
820 errorcode, "oso001.msg", &msglen);
821 Py_END_ALLOW_THREADS
822
823 if (rc == NO_ERROR)
824 os2_formatmsg(msgbuf, msglen, reason);
825 else
Tim Peters1ceb5fb2001-11-28 20:32:57 +0000826 PyOS_snprintf(msgbuf, msgbuflen,
Victor Stinner8c62be82010-05-06 00:08:46 +0000827 "unknown OS error #%d", errorcode);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000828
829 return msgbuf;
830}
831
832/* Set an OS/2-specific error and return NULL. OS/2 kernel
833 errors are not in a global variable e.g. 'errno' nor are
834 they congruent with posix error numbers. */
835
Victor Stinner8c62be82010-05-06 00:08:46 +0000836static PyObject *
837os2_error(int code)
Guido van Rossumd48f2521997-12-05 22:19:34 +0000838{
839 char text[1024];
840 PyObject *v;
841
842 os2_strerror(text, sizeof(text), code, "");
843
844 v = Py_BuildValue("(is)", code, text);
845 if (v != NULL) {
Barry Warsawca74da41999-02-09 19:31:45 +0000846 PyErr_SetObject(PyExc_OSError, v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000847 Py_DECREF(v);
848 }
849 return NULL; /* Signal to Python that an Exception is Pending */
850}
851
852#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000853
854/* POSIX generic methods */
855
Barry Warsaw53699e91996-12-10 23:23:01 +0000856static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +0000857posix_fildes(PyObject *fdobj, int (*func)(int))
858{
Victor Stinner8c62be82010-05-06 00:08:46 +0000859 int fd;
860 int res;
861 fd = PyObject_AsFileDescriptor(fdobj);
862 if (fd < 0)
863 return NULL;
864 if (!_PyVerify_fd(fd))
865 return posix_error();
866 Py_BEGIN_ALLOW_THREADS
867 res = (*func)(fd);
868 Py_END_ALLOW_THREADS
869 if (res < 0)
870 return posix_error();
871 Py_INCREF(Py_None);
872 return Py_None;
Fred Drake4d1e64b2002-04-15 19:40:07 +0000873}
Guido van Rossum21142a01999-01-08 21:05:37 +0000874
875static PyObject *
Thomas Wouters477c8d52006-05-27 19:21:47 +0000876posix_1str(PyObject *args, char *format, int (*func)(const char*))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000877{
Victor Stinner8c62be82010-05-06 00:08:46 +0000878 PyObject *opath1 = NULL;
879 char *path1;
880 int res;
881 if (!PyArg_ParseTuple(args, format,
882 PyUnicode_FSConverter, &opath1))
883 return NULL;
884 path1 = PyBytes_AsString(opath1);
885 Py_BEGIN_ALLOW_THREADS
886 res = (*func)(path1);
887 Py_END_ALLOW_THREADS
888 if (res < 0)
889 return posix_error_with_allocated_filename(opath1);
890 Py_DECREF(opath1);
891 Py_INCREF(Py_None);
892 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000893}
894
Barry Warsaw53699e91996-12-10 23:23:01 +0000895static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +0000896posix_2str(PyObject *args,
Victor Stinner8c62be82010-05-06 00:08:46 +0000897 char *format,
898 int (*func)(const char *, const char *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000899{
Victor Stinner8c62be82010-05-06 00:08:46 +0000900 PyObject *opath1 = NULL, *opath2 = NULL;
901 char *path1, *path2;
902 int res;
903 if (!PyArg_ParseTuple(args, format,
904 PyUnicode_FSConverter, &opath1,
905 PyUnicode_FSConverter, &opath2)) {
906 return NULL;
907 }
908 path1 = PyBytes_AsString(opath1);
909 path2 = PyBytes_AsString(opath2);
910 Py_BEGIN_ALLOW_THREADS
911 res = (*func)(path1, path2);
912 Py_END_ALLOW_THREADS
913 Py_DECREF(opath1);
914 Py_DECREF(opath2);
915 if (res != 0)
916 /* XXX how to report both path1 and path2??? */
917 return posix_error();
918 Py_INCREF(Py_None);
919 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000920}
921
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000922#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +0000923static PyObject*
Victor Stinner8c62be82010-05-06 00:08:46 +0000924win32_1str(PyObject* args, char* func,
925 char* format, BOOL (__stdcall *funcA)(LPCSTR),
926 char* wformat, BOOL (__stdcall *funcW)(LPWSTR))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000927{
Victor Stinner8c62be82010-05-06 00:08:46 +0000928 PyObject *uni;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +0100929 const char *ansi;
Victor Stinner8c62be82010-05-06 00:08:46 +0000930 BOOL result;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +0000931
Victor Stinnereb5657a2011-09-30 01:44:27 +0200932 if (PyArg_ParseTuple(args, wformat, &uni))
933 {
934 wchar_t *wstr = PyUnicode_AsUnicode(uni);
935 if (wstr == NULL)
936 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +0000937 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +0200938 result = funcW(wstr);
Victor Stinner8c62be82010-05-06 00:08:46 +0000939 Py_END_ALLOW_THREADS
940 if (!result)
Victor Stinnereb5657a2011-09-30 01:44:27 +0200941 return win32_error_object(func, uni);
Victor Stinner8c62be82010-05-06 00:08:46 +0000942 Py_INCREF(Py_None);
943 return Py_None;
944 }
Victor Stinnereb5657a2011-09-30 01:44:27 +0200945 PyErr_Clear();
946
Victor Stinner8c62be82010-05-06 00:08:46 +0000947 if (!PyArg_ParseTuple(args, format, &ansi))
948 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +0100949 if (win32_warn_bytes_api())
950 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +0000951 Py_BEGIN_ALLOW_THREADS
952 result = funcA(ansi);
953 Py_END_ALLOW_THREADS
954 if (!result)
955 return win32_error(func, ansi);
956 Py_INCREF(Py_None);
957 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000958
959}
960
961/* This is a reimplementation of the C library's chdir function,
962 but one that produces Win32 errors instead of DOS error codes.
963 chdir is essentially a wrapper around SetCurrentDirectory; however,
964 it also needs to set "magic" environment variables indicating
965 the per-drive current directory, which are of the form =<drive>: */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000966static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000967win32_chdir(LPCSTR path)
968{
Victor Stinner8c62be82010-05-06 00:08:46 +0000969 char new_path[MAX_PATH+1];
970 int result;
971 char env[4] = "=x:";
Thomas Wouters477c8d52006-05-27 19:21:47 +0000972
Victor Stinner8c62be82010-05-06 00:08:46 +0000973 if(!SetCurrentDirectoryA(path))
974 return FALSE;
975 result = GetCurrentDirectoryA(MAX_PATH+1, new_path);
976 if (!result)
977 return FALSE;
978 /* In the ANSI API, there should not be any paths longer
979 than MAX_PATH. */
980 assert(result <= MAX_PATH+1);
981 if (strncmp(new_path, "\\\\", 2) == 0 ||
982 strncmp(new_path, "//", 2) == 0)
983 /* UNC path, nothing to do. */
984 return TRUE;
985 env[1] = new_path[0];
986 return SetEnvironmentVariableA(env, new_path);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000987}
988
989/* The Unicode version differs from the ANSI version
990 since the current directory might exceed MAX_PATH characters */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000991static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000992win32_wchdir(LPCWSTR path)
993{
Victor Stinner8c62be82010-05-06 00:08:46 +0000994 wchar_t _new_path[MAX_PATH+1], *new_path = _new_path;
995 int result;
996 wchar_t env[4] = L"=x:";
Thomas Wouters477c8d52006-05-27 19:21:47 +0000997
Victor Stinner8c62be82010-05-06 00:08:46 +0000998 if(!SetCurrentDirectoryW(path))
999 return FALSE;
1000 result = GetCurrentDirectoryW(MAX_PATH+1, new_path);
1001 if (!result)
1002 return FALSE;
1003 if (result > MAX_PATH+1) {
1004 new_path = malloc(result * sizeof(wchar_t));
1005 if (!new_path) {
1006 SetLastError(ERROR_OUTOFMEMORY);
1007 return FALSE;
1008 }
1009 result = GetCurrentDirectoryW(result, new_path);
1010 if (!result) {
1011 free(new_path);
1012 return FALSE;
1013 }
1014 }
1015 if (wcsncmp(new_path, L"\\\\", 2) == 0 ||
1016 wcsncmp(new_path, L"//", 2) == 0)
1017 /* UNC path, nothing to do. */
1018 return TRUE;
1019 env[1] = new_path[0];
1020 result = SetEnvironmentVariableW(env, new_path);
1021 if (new_path != _new_path)
1022 free(new_path);
1023 return result;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001024}
1025#endif
1026
Martin v. Löwis14694662006-02-03 12:54:16 +00001027#ifdef MS_WINDOWS
1028/* The CRT of Windows has a number of flaws wrt. its stat() implementation:
1029 - time stamps are restricted to second resolution
1030 - file modification times suffer from forth-and-back conversions between
1031 UTC and local time
1032 Therefore, we implement our own stat, based on the Win32 API directly.
1033*/
Victor Stinner8c62be82010-05-06 00:08:46 +00001034#define HAVE_STAT_NSEC 1
Martin v. Löwis14694662006-02-03 12:54:16 +00001035
1036struct win32_stat{
1037 int st_dev;
1038 __int64 st_ino;
1039 unsigned short st_mode;
1040 int st_nlink;
1041 int st_uid;
1042 int st_gid;
1043 int st_rdev;
1044 __int64 st_size;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001045 time_t st_atime;
Martin v. Löwis14694662006-02-03 12:54:16 +00001046 int st_atime_nsec;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001047 time_t st_mtime;
Martin v. Löwis14694662006-02-03 12:54:16 +00001048 int st_mtime_nsec;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001049 time_t st_ctime;
Martin v. Löwis14694662006-02-03 12:54:16 +00001050 int st_ctime_nsec;
1051};
1052
1053static __int64 secs_between_epochs = 11644473600; /* Seconds between 1.1.1601 and 1.1.1970 */
1054
1055static void
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001056FILE_TIME_to_time_t_nsec(FILETIME *in_ptr, time_t *time_out, int* nsec_out)
Martin v. Löwis14694662006-02-03 12:54:16 +00001057{
Victor Stinner8c62be82010-05-06 00:08:46 +00001058 /* XXX endianness. Shouldn't matter, as all Windows implementations are little-endian */
1059 /* Cannot simply cast and dereference in_ptr,
1060 since it might not be aligned properly */
1061 __int64 in;
1062 memcpy(&in, in_ptr, sizeof(in));
1063 *nsec_out = (int)(in % 10000000) * 100; /* FILETIME is in units of 100 nsec. */
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001064 *time_out = Py_SAFE_DOWNCAST((in / 10000000) - secs_between_epochs, __int64, time_t);
Martin v. Löwis14694662006-02-03 12:54:16 +00001065}
1066
Thomas Wouters477c8d52006-05-27 19:21:47 +00001067static void
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001068time_t_to_FILE_TIME(time_t time_in, int nsec_in, FILETIME *out_ptr)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001069{
Victor Stinner8c62be82010-05-06 00:08:46 +00001070 /* XXX endianness */
1071 __int64 out;
1072 out = time_in + secs_between_epochs;
1073 out = out * 10000000 + nsec_in / 100;
1074 memcpy(out_ptr, &out, sizeof(out));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001075}
1076
Martin v. Löwis14694662006-02-03 12:54:16 +00001077/* Below, we *know* that ugo+r is 0444 */
1078#if _S_IREAD != 0400
1079#error Unsupported C library
1080#endif
1081static int
1082attributes_to_mode(DWORD attr)
1083{
Victor Stinner8c62be82010-05-06 00:08:46 +00001084 int m = 0;
1085 if (attr & FILE_ATTRIBUTE_DIRECTORY)
1086 m |= _S_IFDIR | 0111; /* IFEXEC for user,group,other */
1087 else
1088 m |= _S_IFREG;
1089 if (attr & FILE_ATTRIBUTE_READONLY)
1090 m |= 0444;
1091 else
1092 m |= 0666;
1093 return m;
Martin v. Löwis14694662006-02-03 12:54:16 +00001094}
1095
1096static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001097attribute_data_to_stat(BY_HANDLE_FILE_INFORMATION *info, ULONG reparse_tag, struct win32_stat *result)
Martin v. Löwis14694662006-02-03 12:54:16 +00001098{
Victor Stinner8c62be82010-05-06 00:08:46 +00001099 memset(result, 0, sizeof(*result));
1100 result->st_mode = attributes_to_mode(info->dwFileAttributes);
1101 result->st_size = (((__int64)info->nFileSizeHigh)<<32) + info->nFileSizeLow;
1102 FILE_TIME_to_time_t_nsec(&info->ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1103 FILE_TIME_to_time_t_nsec(&info->ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1104 FILE_TIME_to_time_t_nsec(&info->ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001105 result->st_nlink = info->nNumberOfLinks;
Brian Curtin1b9df392010-11-24 20:24:31 +00001106 result->st_ino = (((__int64)info->nFileIndexHigh)<<32) + info->nFileIndexLow;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001107 if (reparse_tag == IO_REPARSE_TAG_SYMLINK) {
1108 /* first clear the S_IFMT bits */
1109 result->st_mode ^= (result->st_mode & 0170000);
1110 /* now set the bits that make this a symlink */
1111 result->st_mode |= 0120000;
1112 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001113
Victor Stinner8c62be82010-05-06 00:08:46 +00001114 return 0;
Martin v. Löwis14694662006-02-03 12:54:16 +00001115}
1116
Guido van Rossumd8faa362007-04-27 19:54:29 +00001117static BOOL
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001118attributes_from_dir(LPCSTR pszFile, BY_HANDLE_FILE_INFORMATION *info, ULONG *reparse_tag)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001119{
Victor Stinner8c62be82010-05-06 00:08:46 +00001120 HANDLE hFindFile;
1121 WIN32_FIND_DATAA FileData;
1122 hFindFile = FindFirstFileA(pszFile, &FileData);
1123 if (hFindFile == INVALID_HANDLE_VALUE)
1124 return FALSE;
1125 FindClose(hFindFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001126 memset(info, 0, sizeof(*info));
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001127 *reparse_tag = 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001128 info->dwFileAttributes = FileData.dwFileAttributes;
1129 info->ftCreationTime = FileData.ftCreationTime;
1130 info->ftLastAccessTime = FileData.ftLastAccessTime;
1131 info->ftLastWriteTime = FileData.ftLastWriteTime;
1132 info->nFileSizeHigh = FileData.nFileSizeHigh;
1133 info->nFileSizeLow = FileData.nFileSizeLow;
1134/* info->nNumberOfLinks = 1; */
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001135 if (FileData.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
1136 *reparse_tag = FileData.dwReserved0;
Victor Stinner8c62be82010-05-06 00:08:46 +00001137 return TRUE;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001138}
1139
1140static BOOL
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001141attributes_from_dir_w(LPCWSTR pszFile, BY_HANDLE_FILE_INFORMATION *info, ULONG *reparse_tag)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001142{
Victor Stinner8c62be82010-05-06 00:08:46 +00001143 HANDLE hFindFile;
1144 WIN32_FIND_DATAW FileData;
1145 hFindFile = FindFirstFileW(pszFile, &FileData);
1146 if (hFindFile == INVALID_HANDLE_VALUE)
1147 return FALSE;
1148 FindClose(hFindFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001149 memset(info, 0, sizeof(*info));
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001150 *reparse_tag = 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001151 info->dwFileAttributes = FileData.dwFileAttributes;
1152 info->ftCreationTime = FileData.ftCreationTime;
1153 info->ftLastAccessTime = FileData.ftLastAccessTime;
1154 info->ftLastWriteTime = FileData.ftLastWriteTime;
1155 info->nFileSizeHigh = FileData.nFileSizeHigh;
1156 info->nFileSizeLow = FileData.nFileSizeLow;
1157/* info->nNumberOfLinks = 1; */
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001158 if (FileData.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
1159 *reparse_tag = FileData.dwReserved0;
Victor Stinner8c62be82010-05-06 00:08:46 +00001160 return TRUE;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001161}
1162
Brian Curtind25aef52011-06-13 15:16:04 -05001163/* Grab GetFinalPathNameByHandle dynamically from kernel32 */
1164static int has_GetFinalPathNameByHandle = 0;
Brian Curtind25aef52011-06-13 15:16:04 -05001165static DWORD (CALLBACK *Py_GetFinalPathNameByHandleW)(HANDLE, LPWSTR, DWORD,
1166 DWORD);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001167static int
Brian Curtind25aef52011-06-13 15:16:04 -05001168check_GetFinalPathNameByHandle()
Brian Curtinf5e76d02010-11-24 13:14:05 +00001169{
Brian Curtind25aef52011-06-13 15:16:04 -05001170 HINSTANCE hKernel32;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001171 DWORD (CALLBACK *Py_GetFinalPathNameByHandleA)(HANDLE, LPSTR, DWORD,
1172 DWORD);
1173
Brian Curtind25aef52011-06-13 15:16:04 -05001174 /* only recheck */
1175 if (!has_GetFinalPathNameByHandle)
1176 {
Martin v. Löwis50590f12012-01-14 17:54:09 +01001177 hKernel32 = GetModuleHandleW(L"KERNEL32");
Brian Curtind25aef52011-06-13 15:16:04 -05001178 *(FARPROC*)&Py_GetFinalPathNameByHandleA = GetProcAddress(hKernel32,
1179 "GetFinalPathNameByHandleA");
1180 *(FARPROC*)&Py_GetFinalPathNameByHandleW = GetProcAddress(hKernel32,
1181 "GetFinalPathNameByHandleW");
1182 has_GetFinalPathNameByHandle = Py_GetFinalPathNameByHandleA &&
1183 Py_GetFinalPathNameByHandleW;
1184 }
1185 return has_GetFinalPathNameByHandle;
1186}
1187
1188static BOOL
1189get_target_path(HANDLE hdl, wchar_t **target_path)
1190{
1191 int buf_size, result_length;
1192 wchar_t *buf;
1193
1194 /* We have a good handle to the target, use it to determine
1195 the target path name (then we'll call lstat on it). */
1196 buf_size = Py_GetFinalPathNameByHandleW(hdl, 0, 0,
1197 VOLUME_NAME_DOS);
1198 if(!buf_size)
1199 return FALSE;
1200
1201 buf = (wchar_t *)malloc((buf_size+1)*sizeof(wchar_t));
Brian Curtinc8be8402011-06-14 09:52:50 -05001202 if (!buf) {
1203 SetLastError(ERROR_OUTOFMEMORY);
1204 return FALSE;
1205 }
1206
Brian Curtind25aef52011-06-13 15:16:04 -05001207 result_length = Py_GetFinalPathNameByHandleW(hdl,
1208 buf, buf_size, VOLUME_NAME_DOS);
1209
1210 if(!result_length) {
1211 free(buf);
1212 return FALSE;
1213 }
1214
1215 if(!CloseHandle(hdl)) {
1216 free(buf);
1217 return FALSE;
1218 }
1219
1220 buf[result_length] = 0;
1221
1222 *target_path = buf;
1223 return TRUE;
1224}
1225
1226static int
1227win32_xstat_impl_w(const wchar_t *path, struct win32_stat *result,
1228 BOOL traverse);
1229static int
1230win32_xstat_impl(const char *path, struct win32_stat *result,
1231 BOOL traverse)
1232{
Victor Stinner26de69d2011-06-17 15:15:38 +02001233 int code;
Brian Curtind25aef52011-06-13 15:16:04 -05001234 HANDLE hFile, hFile2;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001235 BY_HANDLE_FILE_INFORMATION info;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001236 ULONG reparse_tag = 0;
Victor Stinner26de69d2011-06-17 15:15:38 +02001237 wchar_t *target_path;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001238 const char *dot;
1239
Brian Curtind25aef52011-06-13 15:16:04 -05001240 if(!check_GetFinalPathNameByHandle()) {
Brian Curtinc8be8402011-06-14 09:52:50 -05001241 /* If the OS doesn't have GetFinalPathNameByHandle, don't
1242 traverse reparse point. */
1243 traverse = FALSE;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001244 }
1245
Brian Curtinf5e76d02010-11-24 13:14:05 +00001246 hFile = CreateFileA(
1247 path,
Brian Curtind25aef52011-06-13 15:16:04 -05001248 FILE_READ_ATTRIBUTES, /* desired access */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001249 0, /* share mode */
1250 NULL, /* security attributes */
1251 OPEN_EXISTING,
1252 /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
Brian Curtind25aef52011-06-13 15:16:04 -05001253 /* FILE_FLAG_OPEN_REPARSE_POINT does not follow the symlink.
1254 Because of this, calls like GetFinalPathNameByHandle will return
1255 the symlink path agin and not the actual final path. */
1256 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS|
1257 FILE_FLAG_OPEN_REPARSE_POINT,
Brian Curtinf5e76d02010-11-24 13:14:05 +00001258 NULL);
1259
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001260 if (hFile == INVALID_HANDLE_VALUE) {
Brian Curtinf5e76d02010-11-24 13:14:05 +00001261 /* Either the target doesn't exist, or we don't have access to
1262 get a handle to it. If the former, we need to return an error.
1263 If the latter, we can use attributes_from_dir. */
1264 if (GetLastError() != ERROR_SHARING_VIOLATION)
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001265 return -1;
1266 /* Could not get attributes on open file. Fall back to
1267 reading the directory. */
1268 if (!attributes_from_dir(path, &info, &reparse_tag))
1269 /* Very strange. This should not fail now */
1270 return -1;
1271 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
1272 if (traverse) {
1273 /* Should traverse, but could not open reparse point handle */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001274 SetLastError(ERROR_SHARING_VIOLATION);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001275 return -1;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001276 }
Brian Curtinf5e76d02010-11-24 13:14:05 +00001277 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001278 } else {
1279 if (!GetFileInformationByHandle(hFile, &info)) {
1280 CloseHandle(hFile);
Brian Curtind25aef52011-06-13 15:16:04 -05001281 return -1;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001282 }
1283 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
Brian Curtind25aef52011-06-13 15:16:04 -05001284 if (!win32_get_reparse_tag(hFile, &reparse_tag))
1285 return -1;
1286
1287 /* Close the outer open file handle now that we're about to
1288 reopen it with different flags. */
1289 if (!CloseHandle(hFile))
1290 return -1;
1291
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001292 if (traverse) {
Brian Curtind25aef52011-06-13 15:16:04 -05001293 /* In order to call GetFinalPathNameByHandle we need to open
1294 the file without the reparse handling flag set. */
1295 hFile2 = CreateFileA(
1296 path, FILE_READ_ATTRIBUTES, FILE_SHARE_READ,
1297 NULL, OPEN_EXISTING,
1298 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS,
1299 NULL);
1300 if (hFile2 == INVALID_HANDLE_VALUE)
1301 return -1;
1302
1303 if (!get_target_path(hFile2, &target_path))
1304 return -1;
1305
1306 code = win32_xstat_impl_w(target_path, result, FALSE);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001307 free(target_path);
1308 return code;
1309 }
Hirokazu Yamamoto7ed117a2010-12-07 10:24:37 +00001310 } else
1311 CloseHandle(hFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001312 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001313 attribute_data_to_stat(&info, reparse_tag, result);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001314
1315 /* Set S_IEXEC if it is an .exe, .bat, ... */
1316 dot = strrchr(path, '.');
1317 if (dot) {
1318 if (stricmp(dot, ".bat") == 0 || stricmp(dot, ".cmd") == 0 ||
1319 stricmp(dot, ".exe") == 0 || stricmp(dot, ".com") == 0)
1320 result->st_mode |= 0111;
1321 }
1322 return 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001323}
1324
1325static int
Brian Curtind25aef52011-06-13 15:16:04 -05001326win32_xstat_impl_w(const wchar_t *path, struct win32_stat *result,
1327 BOOL traverse)
Brian Curtinf5e76d02010-11-24 13:14:05 +00001328{
1329 int code;
Brian Curtind25aef52011-06-13 15:16:04 -05001330 HANDLE hFile, hFile2;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001331 BY_HANDLE_FILE_INFORMATION info;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001332 ULONG reparse_tag = 0;
Victor Stinner26de69d2011-06-17 15:15:38 +02001333 wchar_t *target_path;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001334 const wchar_t *dot;
1335
Brian Curtind25aef52011-06-13 15:16:04 -05001336 if(!check_GetFinalPathNameByHandle()) {
Brian Curtinc8be8402011-06-14 09:52:50 -05001337 /* If the OS doesn't have GetFinalPathNameByHandle, don't
1338 traverse reparse point. */
1339 traverse = FALSE;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001340 }
1341
Brian Curtinf5e76d02010-11-24 13:14:05 +00001342 hFile = CreateFileW(
1343 path,
Brian Curtind25aef52011-06-13 15:16:04 -05001344 FILE_READ_ATTRIBUTES, /* desired access */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001345 0, /* share mode */
1346 NULL, /* security attributes */
1347 OPEN_EXISTING,
1348 /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
Brian Curtind25aef52011-06-13 15:16:04 -05001349 /* FILE_FLAG_OPEN_REPARSE_POINT does not follow the symlink.
1350 Because of this, calls like GetFinalPathNameByHandle will return
1351 the symlink path agin and not the actual final path. */
Victor Stinner26de69d2011-06-17 15:15:38 +02001352 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS|
Brian Curtind25aef52011-06-13 15:16:04 -05001353 FILE_FLAG_OPEN_REPARSE_POINT,
Brian Curtinf5e76d02010-11-24 13:14:05 +00001354 NULL);
1355
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001356 if (hFile == INVALID_HANDLE_VALUE) {
Brian Curtinf5e76d02010-11-24 13:14:05 +00001357 /* Either the target doesn't exist, or we don't have access to
1358 get a handle to it. If the former, we need to return an error.
1359 If the latter, we can use attributes_from_dir. */
1360 if (GetLastError() != ERROR_SHARING_VIOLATION)
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001361 return -1;
1362 /* Could not get attributes on open file. Fall back to
1363 reading the directory. */
1364 if (!attributes_from_dir_w(path, &info, &reparse_tag))
1365 /* Very strange. This should not fail now */
1366 return -1;
1367 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
1368 if (traverse) {
1369 /* Should traverse, but could not open reparse point handle */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001370 SetLastError(ERROR_SHARING_VIOLATION);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001371 return -1;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001372 }
Brian Curtinf5e76d02010-11-24 13:14:05 +00001373 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001374 } else {
1375 if (!GetFileInformationByHandle(hFile, &info)) {
1376 CloseHandle(hFile);
Brian Curtind25aef52011-06-13 15:16:04 -05001377 return -1;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001378 }
1379 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
Brian Curtind25aef52011-06-13 15:16:04 -05001380 if (!win32_get_reparse_tag(hFile, &reparse_tag))
1381 return -1;
1382
1383 /* Close the outer open file handle now that we're about to
1384 reopen it with different flags. */
1385 if (!CloseHandle(hFile))
1386 return -1;
1387
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001388 if (traverse) {
Brian Curtind25aef52011-06-13 15:16:04 -05001389 /* In order to call GetFinalPathNameByHandle we need to open
1390 the file without the reparse handling flag set. */
1391 hFile2 = CreateFileW(
1392 path, FILE_READ_ATTRIBUTES, FILE_SHARE_READ,
1393 NULL, OPEN_EXISTING,
1394 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS,
1395 NULL);
1396 if (hFile2 == INVALID_HANDLE_VALUE)
1397 return -1;
1398
1399 if (!get_target_path(hFile2, &target_path))
1400 return -1;
1401
1402 code = win32_xstat_impl_w(target_path, result, FALSE);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001403 free(target_path);
1404 return code;
1405 }
Hirokazu Yamamoto7ed117a2010-12-07 10:24:37 +00001406 } else
1407 CloseHandle(hFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001408 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001409 attribute_data_to_stat(&info, reparse_tag, result);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001410
1411 /* Set S_IEXEC if it is an .exe, .bat, ... */
1412 dot = wcsrchr(path, '.');
1413 if (dot) {
1414 if (_wcsicmp(dot, L".bat") == 0 || _wcsicmp(dot, L".cmd") == 0 ||
1415 _wcsicmp(dot, L".exe") == 0 || _wcsicmp(dot, L".com") == 0)
1416 result->st_mode |= 0111;
1417 }
1418 return 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001419}
1420
1421static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001422win32_xstat(const char *path, struct win32_stat *result, BOOL traverse)
Brian Curtinf5e76d02010-11-24 13:14:05 +00001423{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001424 /* Protocol violation: we explicitly clear errno, instead of
1425 setting it to a POSIX error. Callers should use GetLastError. */
Brian Curtind25aef52011-06-13 15:16:04 -05001426 int code = win32_xstat_impl(path, result, traverse);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001427 errno = 0;
1428 return code;
1429}
Brian Curtinf5e76d02010-11-24 13:14:05 +00001430
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001431static int
1432win32_xstat_w(const wchar_t *path, struct win32_stat *result, BOOL traverse)
1433{
1434 /* Protocol violation: we explicitly clear errno, instead of
1435 setting it to a POSIX error. Callers should use GetLastError. */
Brian Curtind25aef52011-06-13 15:16:04 -05001436 int code = win32_xstat_impl_w(path, result, traverse);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001437 errno = 0;
1438 return code;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001439}
Brian Curtind25aef52011-06-13 15:16:04 -05001440/* About the following functions: win32_lstat_w, win32_stat, win32_stat_w
Brian Curtind40e6f72010-07-08 21:39:08 +00001441
1442 In Posix, stat automatically traverses symlinks and returns the stat
1443 structure for the target. In Windows, the equivalent GetFileAttributes by
1444 default does not traverse symlinks and instead returns attributes for
1445 the symlink.
1446
1447 Therefore, win32_lstat will get the attributes traditionally, and
1448 win32_stat will first explicitly resolve the symlink target and then will
1449 call win32_lstat on that result.
1450
Ezio Melotti4969f702011-03-15 05:59:46 +02001451 The _w represent Unicode equivalents of the aforementioned ANSI functions. */
Brian Curtind40e6f72010-07-08 21:39:08 +00001452
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001453static int
Brian Curtind40e6f72010-07-08 21:39:08 +00001454win32_lstat(const char* path, struct win32_stat *result)
Martin v. Löwis14694662006-02-03 12:54:16 +00001455{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001456 return win32_xstat(path, result, FALSE);
Martin v. Löwis14694662006-02-03 12:54:16 +00001457}
1458
Victor Stinner8c62be82010-05-06 00:08:46 +00001459static int
Brian Curtind40e6f72010-07-08 21:39:08 +00001460win32_lstat_w(const wchar_t* path, struct win32_stat *result)
Martin v. Löwis14694662006-02-03 12:54:16 +00001461{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001462 return win32_xstat_w(path, result, FALSE);
Brian Curtind40e6f72010-07-08 21:39:08 +00001463}
1464
1465static int
1466win32_stat(const char* path, struct win32_stat *result)
1467{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001468 return win32_xstat(path, result, TRUE);
Brian Curtind40e6f72010-07-08 21:39:08 +00001469}
1470
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001471static int
Brian Curtind40e6f72010-07-08 21:39:08 +00001472win32_stat_w(const wchar_t* path, struct win32_stat *result)
1473{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001474 return win32_xstat_w(path, result, TRUE);
Martin v. Löwis14694662006-02-03 12:54:16 +00001475}
1476
1477static int
1478win32_fstat(int file_number, struct win32_stat *result)
1479{
Victor Stinner8c62be82010-05-06 00:08:46 +00001480 BY_HANDLE_FILE_INFORMATION info;
1481 HANDLE h;
1482 int type;
Martin v. Löwis14694662006-02-03 12:54:16 +00001483
Victor Stinner8c62be82010-05-06 00:08:46 +00001484 h = (HANDLE)_get_osfhandle(file_number);
Martin v. Löwis14694662006-02-03 12:54:16 +00001485
Victor Stinner8c62be82010-05-06 00:08:46 +00001486 /* Protocol violation: we explicitly clear errno, instead of
1487 setting it to a POSIX error. Callers should use GetLastError. */
1488 errno = 0;
Martin v. Löwis14694662006-02-03 12:54:16 +00001489
Victor Stinner8c62be82010-05-06 00:08:46 +00001490 if (h == INVALID_HANDLE_VALUE) {
1491 /* This is really a C library error (invalid file handle).
1492 We set the Win32 error to the closes one matching. */
1493 SetLastError(ERROR_INVALID_HANDLE);
1494 return -1;
1495 }
1496 memset(result, 0, sizeof(*result));
Martin v. Löwis14694662006-02-03 12:54:16 +00001497
Victor Stinner8c62be82010-05-06 00:08:46 +00001498 type = GetFileType(h);
1499 if (type == FILE_TYPE_UNKNOWN) {
1500 DWORD error = GetLastError();
1501 if (error != 0) {
Brian Curtinf5e76d02010-11-24 13:14:05 +00001502 return -1;
Victor Stinner8c62be82010-05-06 00:08:46 +00001503 }
1504 /* else: valid but unknown file */
1505 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001506
Victor Stinner8c62be82010-05-06 00:08:46 +00001507 if (type != FILE_TYPE_DISK) {
1508 if (type == FILE_TYPE_CHAR)
1509 result->st_mode = _S_IFCHR;
1510 else if (type == FILE_TYPE_PIPE)
1511 result->st_mode = _S_IFIFO;
1512 return 0;
1513 }
1514
1515 if (!GetFileInformationByHandle(h, &info)) {
1516 return -1;
1517 }
1518
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001519 attribute_data_to_stat(&info, 0, result);
Victor Stinner8c62be82010-05-06 00:08:46 +00001520 /* specific to fstat() */
Victor Stinner8c62be82010-05-06 00:08:46 +00001521 result->st_ino = (((__int64)info.nFileIndexHigh)<<32) + info.nFileIndexLow;
1522 return 0;
Martin v. Löwis14694662006-02-03 12:54:16 +00001523}
1524
1525#endif /* MS_WINDOWS */
1526
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001527PyDoc_STRVAR(stat_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001528"stat_result: Result from stat or lstat.\n\n\
1529This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001530 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001531or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
1532\n\
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001533Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
1534or st_flags, they are available as attributes only.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001535\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001536See os.stat for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001537
1538static PyStructSequence_Field stat_result_fields[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001539 {"st_mode", "protection bits"},
1540 {"st_ino", "inode"},
1541 {"st_dev", "device"},
1542 {"st_nlink", "number of hard links"},
1543 {"st_uid", "user ID of owner"},
1544 {"st_gid", "group ID of owner"},
1545 {"st_size", "total size, in bytes"},
1546 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
1547 {NULL, "integer time of last access"},
1548 {NULL, "integer time of last modification"},
1549 {NULL, "integer time of last change"},
1550 {"st_atime", "time of last access"},
1551 {"st_mtime", "time of last modification"},
1552 {"st_ctime", "time of last change"},
Larry Hastings6fe20b32012-04-19 15:07:49 -07001553 {"st_atime_ns", "time of last access in nanoseconds"},
1554 {"st_mtime_ns", "time of last modification in nanoseconds"},
1555 {"st_ctime_ns", "time of last change in nanoseconds"},
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001556#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00001557 {"st_blksize", "blocksize for filesystem I/O"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001558#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001559#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Victor Stinner8c62be82010-05-06 00:08:46 +00001560 {"st_blocks", "number of blocks allocated"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001561#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001562#ifdef HAVE_STRUCT_STAT_ST_RDEV
Victor Stinner8c62be82010-05-06 00:08:46 +00001563 {"st_rdev", "device type (if inode device)"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001564#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001565#ifdef HAVE_STRUCT_STAT_ST_FLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00001566 {"st_flags", "user defined flags for file"},
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001567#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001568#ifdef HAVE_STRUCT_STAT_ST_GEN
Victor Stinner8c62be82010-05-06 00:08:46 +00001569 {"st_gen", "generation number"},
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001570#endif
1571#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00001572 {"st_birthtime", "time of creation"},
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001573#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001574 {0}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001575};
1576
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001577#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Larry Hastings6fe20b32012-04-19 15:07:49 -07001578#define ST_BLKSIZE_IDX 16
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001579#else
Larry Hastings6fe20b32012-04-19 15:07:49 -07001580#define ST_BLKSIZE_IDX 15
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001581#endif
1582
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001583#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001584#define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
1585#else
1586#define ST_BLOCKS_IDX ST_BLKSIZE_IDX
1587#endif
1588
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001589#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001590#define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
1591#else
1592#define ST_RDEV_IDX ST_BLOCKS_IDX
1593#endif
1594
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001595#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1596#define ST_FLAGS_IDX (ST_RDEV_IDX+1)
1597#else
1598#define ST_FLAGS_IDX ST_RDEV_IDX
1599#endif
1600
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001601#ifdef HAVE_STRUCT_STAT_ST_GEN
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001602#define ST_GEN_IDX (ST_FLAGS_IDX+1)
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001603#else
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001604#define ST_GEN_IDX ST_FLAGS_IDX
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001605#endif
1606
1607#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1608#define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
1609#else
1610#define ST_BIRTHTIME_IDX ST_GEN_IDX
1611#endif
1612
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001613static PyStructSequence_Desc stat_result_desc = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001614 "stat_result", /* name */
1615 stat_result__doc__, /* doc */
1616 stat_result_fields,
1617 10
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001618};
1619
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001620PyDoc_STRVAR(statvfs_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001621"statvfs_result: Result from statvfs or fstatvfs.\n\n\
1622This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001623 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
Guido van Rossuma4dc73e2001-10-18 20:53:15 +00001624or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001625\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001626See os.statvfs for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001627
1628static PyStructSequence_Field statvfs_result_fields[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001629 {"f_bsize", },
1630 {"f_frsize", },
1631 {"f_blocks", },
1632 {"f_bfree", },
1633 {"f_bavail", },
1634 {"f_files", },
1635 {"f_ffree", },
1636 {"f_favail", },
1637 {"f_flag", },
1638 {"f_namemax",},
1639 {0}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001640};
1641
1642static PyStructSequence_Desc statvfs_result_desc = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001643 "statvfs_result", /* name */
1644 statvfs_result__doc__, /* doc */
1645 statvfs_result_fields,
1646 10
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001647};
1648
Ross Lagerwall7807c352011-03-17 20:20:30 +02001649#if defined(HAVE_WAITID) && !defined(__APPLE__)
1650PyDoc_STRVAR(waitid_result__doc__,
1651"waitid_result: Result from waitid.\n\n\
1652This object may be accessed either as a tuple of\n\
1653 (si_pid, si_uid, si_signo, si_status, si_code),\n\
1654or via the attributes si_pid, si_uid, and so on.\n\
1655\n\
1656See os.waitid for more information.");
1657
1658static PyStructSequence_Field waitid_result_fields[] = {
1659 {"si_pid", },
1660 {"si_uid", },
1661 {"si_signo", },
1662 {"si_status", },
1663 {"si_code", },
1664 {0}
1665};
1666
1667static PyStructSequence_Desc waitid_result_desc = {
1668 "waitid_result", /* name */
1669 waitid_result__doc__, /* doc */
1670 waitid_result_fields,
1671 5
1672};
1673static PyTypeObject WaitidResultType;
1674#endif
1675
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001676static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001677static PyTypeObject StatResultType;
1678static PyTypeObject StatVFSResultType;
Benjamin Petersonbad9c2f2011-08-02 18:42:14 -05001679#if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER)
Benjamin Peterson94b580d2011-08-02 17:30:04 -05001680static PyTypeObject SchedParamType;
Benjamin Petersonbad9c2f2011-08-02 18:42:14 -05001681#endif
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001682static newfunc structseq_new;
1683
1684static PyObject *
1685statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1686{
Victor Stinner8c62be82010-05-06 00:08:46 +00001687 PyStructSequence *result;
1688 int i;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001689
Victor Stinner8c62be82010-05-06 00:08:46 +00001690 result = (PyStructSequence*)structseq_new(type, args, kwds);
1691 if (!result)
1692 return NULL;
1693 /* If we have been initialized from a tuple,
1694 st_?time might be set to None. Initialize it
1695 from the int slots. */
1696 for (i = 7; i <= 9; i++) {
1697 if (result->ob_item[i+3] == Py_None) {
1698 Py_DECREF(Py_None);
1699 Py_INCREF(result->ob_item[i]);
1700 result->ob_item[i+3] = result->ob_item[i];
1701 }
1702 }
1703 return (PyObject*)result;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001704}
1705
1706
1707
1708/* If true, st_?time is float. */
Martin v. Löwisfe33d0b2005-01-16 08:57:39 +00001709static int _stat_float_times = 1;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001710
1711PyDoc_STRVAR(stat_float_times__doc__,
1712"stat_float_times([newval]) -> oldval\n\n\
1713Determine whether os.[lf]stat represents time stamps as float objects.\n\
1714If newval is True, future calls to stat() return floats, if it is False,\n\
1715future calls return ints. \n\
1716If newval is omitted, return the current setting.\n");
1717
1718static PyObject*
1719stat_float_times(PyObject* self, PyObject *args)
1720{
Victor Stinner8c62be82010-05-06 00:08:46 +00001721 int newval = -1;
1722 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
1723 return NULL;
1724 if (newval == -1)
1725 /* Return old value */
1726 return PyBool_FromLong(_stat_float_times);
1727 _stat_float_times = newval;
1728 Py_INCREF(Py_None);
1729 return Py_None;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001730}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001731
Larry Hastings6fe20b32012-04-19 15:07:49 -07001732static PyObject *billion = NULL;
1733
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001734static void
Victor Stinner4195b5c2012-02-08 23:03:19 +01001735fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001736{
Larry Hastings6fe20b32012-04-19 15:07:49 -07001737 PyObject *s = _PyLong_FromTime_t(sec);
1738 PyObject *ns_fractional = PyLong_FromUnsignedLong(nsec);
1739 PyObject *s_in_ns = NULL;
1740 PyObject *ns_total = NULL;
1741 PyObject *float_s = NULL;
1742
1743 if (!(s && ns_fractional))
1744 goto exit;
1745
1746 s_in_ns = PyNumber_Multiply(s, billion);
1747 if (!s_in_ns)
1748 goto exit;
1749
1750 ns_total = PyNumber_Add(s_in_ns, ns_fractional);
1751 if (!ns_total)
1752 goto exit;
1753
Victor Stinner4195b5c2012-02-08 23:03:19 +01001754 if (_stat_float_times) {
Larry Hastings6fe20b32012-04-19 15:07:49 -07001755 float_s = PyFloat_FromDouble(sec + 1e-9*nsec);
1756 if (!float_s)
1757 goto exit;
Victor Stinner8c62be82010-05-06 00:08:46 +00001758 }
Larry Hastings6fe20b32012-04-19 15:07:49 -07001759 else {
1760 float_s = s;
1761 Py_INCREF(float_s);
1762 }
1763
1764 PyStructSequence_SET_ITEM(v, index, s);
1765 PyStructSequence_SET_ITEM(v, index+3, float_s);
1766 PyStructSequence_SET_ITEM(v, index+6, ns_total);
1767 s = NULL;
1768 float_s = NULL;
1769 ns_total = NULL;
1770exit:
1771 Py_XDECREF(s);
1772 Py_XDECREF(ns_fractional);
1773 Py_XDECREF(s_in_ns);
1774 Py_XDECREF(ns_total);
1775 Py_XDECREF(float_s);
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001776}
1777
Tim Peters5aa91602002-01-30 05:46:57 +00001778/* pack a system stat C structure into the Python stat tuple
Fred Drake699f3522000-06-29 21:12:41 +00001779 (used by posix_stat() and posix_fstat()) */
1780static PyObject*
Victor Stinner4195b5c2012-02-08 23:03:19 +01001781_pystat_fromstructstat(STRUCT_STAT *st)
Fred Drake699f3522000-06-29 21:12:41 +00001782{
Victor Stinner8c62be82010-05-06 00:08:46 +00001783 unsigned long ansec, mnsec, cnsec;
1784 PyObject *v = PyStructSequence_New(&StatResultType);
1785 if (v == NULL)
1786 return NULL;
Fred Drake699f3522000-06-29 21:12:41 +00001787
Victor Stinner8c62be82010-05-06 00:08:46 +00001788 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long)st->st_mode));
Fred Drake699f3522000-06-29 21:12:41 +00001789#ifdef HAVE_LARGEFILE_SUPPORT
Victor Stinner8c62be82010-05-06 00:08:46 +00001790 PyStructSequence_SET_ITEM(v, 1,
1791 PyLong_FromLongLong((PY_LONG_LONG)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001792#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001793 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001794#endif
1795#if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00001796 PyStructSequence_SET_ITEM(v, 2,
1797 PyLong_FromLongLong((PY_LONG_LONG)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001798#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001799 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001800#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001801 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long)st->st_nlink));
1802 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long)st->st_uid));
1803 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long)st->st_gid));
Fred Drake699f3522000-06-29 21:12:41 +00001804#ifdef HAVE_LARGEFILE_SUPPORT
Victor Stinner8c62be82010-05-06 00:08:46 +00001805 PyStructSequence_SET_ITEM(v, 6,
1806 PyLong_FromLongLong((PY_LONG_LONG)st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001807#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001808 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong(st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001809#endif
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001810
Martin v. Löwis14694662006-02-03 12:54:16 +00001811#if defined(HAVE_STAT_TV_NSEC)
Victor Stinner8c62be82010-05-06 00:08:46 +00001812 ansec = st->st_atim.tv_nsec;
1813 mnsec = st->st_mtim.tv_nsec;
1814 cnsec = st->st_ctim.tv_nsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001815#elif defined(HAVE_STAT_TV_NSEC2)
Victor Stinner8c62be82010-05-06 00:08:46 +00001816 ansec = st->st_atimespec.tv_nsec;
1817 mnsec = st->st_mtimespec.tv_nsec;
1818 cnsec = st->st_ctimespec.tv_nsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001819#elif defined(HAVE_STAT_NSEC)
Victor Stinner8c62be82010-05-06 00:08:46 +00001820 ansec = st->st_atime_nsec;
1821 mnsec = st->st_mtime_nsec;
1822 cnsec = st->st_ctime_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001823#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001824 ansec = mnsec = cnsec = 0;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001825#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +01001826 fill_time(v, 7, st->st_atime, ansec);
1827 fill_time(v, 8, st->st_mtime, mnsec);
1828 fill_time(v, 9, st->st_ctime, cnsec);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001829
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001830#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00001831 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
1832 PyLong_FromLong((long)st->st_blksize));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001833#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001834#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Victor Stinner8c62be82010-05-06 00:08:46 +00001835 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
1836 PyLong_FromLong((long)st->st_blocks));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001837#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001838#ifdef HAVE_STRUCT_STAT_ST_RDEV
Victor Stinner8c62be82010-05-06 00:08:46 +00001839 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
1840 PyLong_FromLong((long)st->st_rdev));
Fred Drake699f3522000-06-29 21:12:41 +00001841#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001842#ifdef HAVE_STRUCT_STAT_ST_GEN
Victor Stinner8c62be82010-05-06 00:08:46 +00001843 PyStructSequence_SET_ITEM(v, ST_GEN_IDX,
1844 PyLong_FromLong((long)st->st_gen));
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001845#endif
1846#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00001847 {
Victor Stinner4195b5c2012-02-08 23:03:19 +01001848 PyObject *val;
1849 unsigned long bsec,bnsec;
1850 bsec = (long)st->st_birthtime;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001851#ifdef HAVE_STAT_TV_NSEC2
Victor Stinner4195b5c2012-02-08 23:03:19 +01001852 bnsec = st->st_birthtimespec.tv_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001853#else
Victor Stinner4195b5c2012-02-08 23:03:19 +01001854 bnsec = 0;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001855#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +01001856 if (_stat_float_times) {
1857 val = PyFloat_FromDouble(bsec + 1e-9*bnsec);
1858 } else {
1859 val = PyLong_FromLong((long)bsec);
1860 }
1861 PyStructSequence_SET_ITEM(v, ST_BIRTHTIME_IDX,
1862 val);
Victor Stinner8c62be82010-05-06 00:08:46 +00001863 }
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001864#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001865#ifdef HAVE_STRUCT_STAT_ST_FLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00001866 PyStructSequence_SET_ITEM(v, ST_FLAGS_IDX,
1867 PyLong_FromLong((long)st->st_flags));
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001868#endif
Fred Drake699f3522000-06-29 21:12:41 +00001869
Victor Stinner8c62be82010-05-06 00:08:46 +00001870 if (PyErr_Occurred()) {
1871 Py_DECREF(v);
1872 return NULL;
1873 }
Fred Drake699f3522000-06-29 21:12:41 +00001874
Victor Stinner8c62be82010-05-06 00:08:46 +00001875 return v;
Fred Drake699f3522000-06-29 21:12:41 +00001876}
1877
Barry Warsaw53699e91996-12-10 23:23:01 +00001878static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01001879posix_do_stat(PyObject *self, PyObject *args,
Victor Stinner8c62be82010-05-06 00:08:46 +00001880 char *format,
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001881#ifdef __VMS
Victor Stinner8c62be82010-05-06 00:08:46 +00001882 int (*statfunc)(const char *, STRUCT_STAT *, ...),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001883#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001884 int (*statfunc)(const char *, STRUCT_STAT *),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001885#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001886 char *wformat,
Victor Stinnereb5657a2011-09-30 01:44:27 +02001887 int (*wstatfunc)(const wchar_t *, STRUCT_STAT *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001888{
Victor Stinner8c62be82010-05-06 00:08:46 +00001889 STRUCT_STAT st;
1890 PyObject *opath;
1891 char *path;
1892 int res;
1893 PyObject *result;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001894
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001895#ifdef MS_WINDOWS
Victor Stinnereb5657a2011-09-30 01:44:27 +02001896 PyObject *po;
Victor Stinner4195b5c2012-02-08 23:03:19 +01001897 if (PyArg_ParseTuple(args, wformat, &po)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02001898 wchar_t *wpath = PyUnicode_AsUnicode(po);
1899 if (wpath == NULL)
1900 return NULL;
Martin v. Löwis14694662006-02-03 12:54:16 +00001901
Victor Stinner8c62be82010-05-06 00:08:46 +00001902 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00001903 res = wstatfunc(wpath, &st);
1904 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001905
Victor Stinner8c62be82010-05-06 00:08:46 +00001906 if (res != 0)
Victor Stinnereb5657a2011-09-30 01:44:27 +02001907 return win32_error_object("stat", po);
Victor Stinner4195b5c2012-02-08 23:03:19 +01001908 return _pystat_fromstructstat(&st);
Victor Stinner8c62be82010-05-06 00:08:46 +00001909 }
1910 /* Drop the argument parsing error as narrow strings
1911 are also valid. */
1912 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001913#endif
1914
Victor Stinner4195b5c2012-02-08 23:03:19 +01001915 if (!PyArg_ParseTuple(args, format,
1916 PyUnicode_FSConverter, &opath))
Victor Stinner8c62be82010-05-06 00:08:46 +00001917 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001918#ifdef MS_WINDOWS
1919 if (win32_warn_bytes_api()) {
1920 Py_DECREF(opath);
1921 return NULL;
1922 }
1923#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001924 path = PyBytes_AsString(opath);
1925 Py_BEGIN_ALLOW_THREADS
1926 res = (*statfunc)(path, &st);
1927 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001928
Victor Stinner8c62be82010-05-06 00:08:46 +00001929 if (res != 0) {
Martin v. Löwis14694662006-02-03 12:54:16 +00001930#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00001931 result = win32_error("stat", path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001932#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001933 result = posix_error_with_filename(path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001934#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001935 }
1936 else
Victor Stinner4195b5c2012-02-08 23:03:19 +01001937 result = _pystat_fromstructstat(&st);
Fred Drake699f3522000-06-29 21:12:41 +00001938
Victor Stinner8c62be82010-05-06 00:08:46 +00001939 Py_DECREF(opath);
1940 return result;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001941}
1942
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001943/* POSIX methods */
1944
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001945PyDoc_STRVAR(posix_access__doc__,
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001946"access(path, mode) -> True if granted, False otherwise\n\n\
Guido van Rossuma0b90752002-06-18 16:22:43 +00001947Use the real uid/gid to test for access to a path. Note that most\n\
1948operations will use the effective uid/gid, therefore this routine can\n\
1949be used in a suid/sgid environment to test if the invoking user has the\n\
1950specified access to the path. The mode argument can be F_OK to test\n\
1951existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001952
1953static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001954posix_access(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001955{
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001956 const char *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00001957 int mode;
1958
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001959#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00001960 DWORD attr;
Victor Stinnereb5657a2011-09-30 01:44:27 +02001961 PyObject *po;
Victor Stinner8c62be82010-05-06 00:08:46 +00001962 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02001963 wchar_t* wpath = PyUnicode_AsUnicode(po);
1964 if (wpath == NULL)
1965 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00001966 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02001967 attr = GetFileAttributesW(wpath);
Victor Stinner8c62be82010-05-06 00:08:46 +00001968 Py_END_ALLOW_THREADS
1969 goto finish;
1970 }
1971 /* Drop the argument parsing error as narrow strings
1972 are also valid. */
1973 PyErr_Clear();
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001974 if (!PyArg_ParseTuple(args, "yi:access", &path, &mode))
Victor Stinner8c62be82010-05-06 00:08:46 +00001975 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001976 if (win32_warn_bytes_api())
1977 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00001978 Py_BEGIN_ALLOW_THREADS
1979 attr = GetFileAttributesA(path);
1980 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001981finish:
Victor Stinner8c62be82010-05-06 00:08:46 +00001982 if (attr == 0xFFFFFFFF)
1983 /* File does not exist, or cannot read attributes */
1984 return PyBool_FromLong(0);
1985 /* Access is possible if either write access wasn't requested, or
1986 the file isn't read-only, or if it's a directory, as there are
1987 no read-only directories on Windows. */
1988 return PyBool_FromLong(!(mode & 2)
1989 || !(attr & FILE_ATTRIBUTE_READONLY)
1990 || (attr & FILE_ATTRIBUTE_DIRECTORY));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001991#else
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001992 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00001993 int res;
1994 if (!PyArg_ParseTuple(args, "O&i:access",
1995 PyUnicode_FSConverter, &opath, &mode))
1996 return NULL;
1997 path = PyBytes_AsString(opath);
1998 Py_BEGIN_ALLOW_THREADS
1999 res = access(path, mode);
2000 Py_END_ALLOW_THREADS
2001 Py_DECREF(opath);
2002 return PyBool_FromLong(res == 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002003#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00002004}
2005
Guido van Rossumd371ff11999-01-25 16:12:23 +00002006#ifndef F_OK
2007#define F_OK 0
2008#endif
2009#ifndef R_OK
2010#define R_OK 4
2011#endif
2012#ifndef W_OK
2013#define W_OK 2
2014#endif
2015#ifndef X_OK
2016#define X_OK 1
2017#endif
2018
2019#ifdef HAVE_TTYNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002020PyDoc_STRVAR(posix_ttyname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002021"ttyname(fd) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002022Return the name of the terminal device connected to 'fd'.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00002023
2024static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002025posix_ttyname(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00002026{
Victor Stinner8c62be82010-05-06 00:08:46 +00002027 int id;
2028 char *ret;
Guido van Rossum94f6f721999-01-06 18:42:14 +00002029
Victor Stinner8c62be82010-05-06 00:08:46 +00002030 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
2031 return NULL;
Guido van Rossum94f6f721999-01-06 18:42:14 +00002032
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002033#if defined(__VMS)
Victor Stinner8c62be82010-05-06 00:08:46 +00002034 /* file descriptor 0 only, the default input device (stdin) */
2035 if (id == 0) {
2036 ret = ttyname();
2037 }
2038 else {
2039 ret = NULL;
2040 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002041#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002042 ret = ttyname(id);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002043#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00002044 if (ret == NULL)
2045 return posix_error();
Victor Stinner5fe6de82010-08-15 09:12:51 +00002046 return PyUnicode_DecodeFSDefault(ret);
Guido van Rossum94f6f721999-01-06 18:42:14 +00002047}
Guido van Rossumd371ff11999-01-25 16:12:23 +00002048#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00002049
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002050#ifdef HAVE_CTERMID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002051PyDoc_STRVAR(posix_ctermid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002052"ctermid() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002053Return the name of the controlling terminal for this process.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002054
2055static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002056posix_ctermid(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002057{
Victor Stinner8c62be82010-05-06 00:08:46 +00002058 char *ret;
2059 char buffer[L_ctermid];
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002060
Greg Wardb48bc172000-03-01 21:51:56 +00002061#ifdef USE_CTERMID_R
Victor Stinner8c62be82010-05-06 00:08:46 +00002062 ret = ctermid_r(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002063#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002064 ret = ctermid(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002065#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00002066 if (ret == NULL)
2067 return posix_error();
Victor Stinner5fe6de82010-08-15 09:12:51 +00002068 return PyUnicode_DecodeFSDefault(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002069}
2070#endif
2071
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002072PyDoc_STRVAR(posix_chdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002073"chdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002074Change the current working directory to the specified path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002075
Barry Warsaw53699e91996-12-10 23:23:01 +00002076static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002077posix_chdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002078{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002079#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00002080 return win32_1str(args, "chdir", "y:chdir", win32_chdir, "U:chdir", win32_wchdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002081#elif defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00002082 return posix_1str(args, "O&:chdir", _chdir2);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00002083#elif defined(__VMS)
Victor Stinner8c62be82010-05-06 00:08:46 +00002084 return posix_1str(args, "O&:chdir", (int (*)(const char *))chdir);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002085#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002086 return posix_1str(args, "O&:chdir", chdir);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002087#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002088}
2089
Fred Drake4d1e64b2002-04-15 19:40:07 +00002090#ifdef HAVE_FCHDIR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002091PyDoc_STRVAR(posix_fchdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002092"fchdir(fildes)\n\n\
Fred Drake4d1e64b2002-04-15 19:40:07 +00002093Change to the directory of the given file descriptor. fildes must be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002094opened on a directory, not a file.");
Fred Drake4d1e64b2002-04-15 19:40:07 +00002095
2096static PyObject *
2097posix_fchdir(PyObject *self, PyObject *fdobj)
2098{
Victor Stinner8c62be82010-05-06 00:08:46 +00002099 return posix_fildes(fdobj, fchdir);
Fred Drake4d1e64b2002-04-15 19:40:07 +00002100}
2101#endif /* HAVE_FCHDIR */
2102
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002103
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002104PyDoc_STRVAR(posix_chmod__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002105"chmod(path, mode)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002106Change the access permissions of a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002107
Barry Warsaw53699e91996-12-10 23:23:01 +00002108static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002109posix_chmod(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002110{
Victor Stinner8c62be82010-05-06 00:08:46 +00002111 PyObject *opath = NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002112 const char *path = NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00002113 int i;
2114 int res;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002115#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00002116 DWORD attr;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002117 PyObject *po;
Victor Stinner8c62be82010-05-06 00:08:46 +00002118 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02002119 wchar_t *wpath = PyUnicode_AsUnicode(po);
2120 if (wpath == NULL)
2121 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00002122 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02002123 attr = GetFileAttributesW(wpath);
Victor Stinner8c62be82010-05-06 00:08:46 +00002124 if (attr != 0xFFFFFFFF) {
2125 if (i & _S_IWRITE)
2126 attr &= ~FILE_ATTRIBUTE_READONLY;
2127 else
2128 attr |= FILE_ATTRIBUTE_READONLY;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002129 res = SetFileAttributesW(wpath, attr);
Victor Stinner8c62be82010-05-06 00:08:46 +00002130 }
2131 else
2132 res = 0;
2133 Py_END_ALLOW_THREADS
2134 if (!res)
Victor Stinnereb5657a2011-09-30 01:44:27 +02002135 return win32_error_object("chmod", po);
Victor Stinner8c62be82010-05-06 00:08:46 +00002136 Py_INCREF(Py_None);
2137 return Py_None;
2138 }
2139 /* Drop the argument parsing error as narrow strings
2140 are also valid. */
2141 PyErr_Clear();
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002142
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002143 if (!PyArg_ParseTuple(args, "yi:chmod", &path, &i))
Victor Stinner8c62be82010-05-06 00:08:46 +00002144 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002145 if (win32_warn_bytes_api())
2146 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00002147 Py_BEGIN_ALLOW_THREADS
2148 attr = GetFileAttributesA(path);
2149 if (attr != 0xFFFFFFFF) {
2150 if (i & _S_IWRITE)
2151 attr &= ~FILE_ATTRIBUTE_READONLY;
2152 else
2153 attr |= FILE_ATTRIBUTE_READONLY;
2154 res = SetFileAttributesA(path, attr);
2155 }
2156 else
2157 res = 0;
2158 Py_END_ALLOW_THREADS
2159 if (!res) {
2160 win32_error("chmod", path);
Victor Stinner8c62be82010-05-06 00:08:46 +00002161 return NULL;
2162 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002163 Py_INCREF(Py_None);
2164 return Py_None;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002165#else /* MS_WINDOWS */
Victor Stinner8c62be82010-05-06 00:08:46 +00002166 if (!PyArg_ParseTuple(args, "O&i:chmod", PyUnicode_FSConverter,
2167 &opath, &i))
2168 return NULL;
2169 path = PyBytes_AsString(opath);
2170 Py_BEGIN_ALLOW_THREADS
2171 res = chmod(path, i);
2172 Py_END_ALLOW_THREADS
2173 if (res < 0)
2174 return posix_error_with_allocated_filename(opath);
2175 Py_DECREF(opath);
2176 Py_INCREF(Py_None);
2177 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002178#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002179}
2180
Christian Heimes4e30a842007-11-30 22:12:06 +00002181#ifdef HAVE_FCHMOD
2182PyDoc_STRVAR(posix_fchmod__doc__,
2183"fchmod(fd, mode)\n\n\
2184Change the access permissions of the file given by file\n\
2185descriptor fd.");
2186
2187static PyObject *
2188posix_fchmod(PyObject *self, PyObject *args)
2189{
Victor Stinner8c62be82010-05-06 00:08:46 +00002190 int fd, mode, res;
2191 if (!PyArg_ParseTuple(args, "ii:fchmod", &fd, &mode))
2192 return NULL;
2193 Py_BEGIN_ALLOW_THREADS
2194 res = fchmod(fd, mode);
2195 Py_END_ALLOW_THREADS
2196 if (res < 0)
2197 return posix_error();
2198 Py_RETURN_NONE;
Christian Heimes4e30a842007-11-30 22:12:06 +00002199}
2200#endif /* HAVE_FCHMOD */
2201
2202#ifdef HAVE_LCHMOD
2203PyDoc_STRVAR(posix_lchmod__doc__,
2204"lchmod(path, mode)\n\n\
2205Change the access permissions of a file. If path is a symlink, this\n\
2206affects the link itself rather than the target.");
2207
2208static PyObject *
2209posix_lchmod(PyObject *self, PyObject *args)
2210{
Victor Stinner8c62be82010-05-06 00:08:46 +00002211 PyObject *opath;
2212 char *path;
2213 int i;
2214 int res;
2215 if (!PyArg_ParseTuple(args, "O&i:lchmod", PyUnicode_FSConverter,
2216 &opath, &i))
2217 return NULL;
2218 path = PyBytes_AsString(opath);
2219 Py_BEGIN_ALLOW_THREADS
2220 res = lchmod(path, i);
2221 Py_END_ALLOW_THREADS
2222 if (res < 0)
2223 return posix_error_with_allocated_filename(opath);
2224 Py_DECREF(opath);
2225 Py_RETURN_NONE;
Christian Heimes4e30a842007-11-30 22:12:06 +00002226}
2227#endif /* HAVE_LCHMOD */
2228
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002229
Thomas Wouterscf297e42007-02-23 15:07:44 +00002230#ifdef HAVE_CHFLAGS
2231PyDoc_STRVAR(posix_chflags__doc__,
2232"chflags(path, flags)\n\n\
2233Set file flags.");
2234
2235static PyObject *
2236posix_chflags(PyObject *self, PyObject *args)
2237{
Victor Stinner8c62be82010-05-06 00:08:46 +00002238 PyObject *opath;
2239 char *path;
2240 unsigned long flags;
2241 int res;
2242 if (!PyArg_ParseTuple(args, "O&k:chflags",
2243 PyUnicode_FSConverter, &opath, &flags))
2244 return NULL;
2245 path = PyBytes_AsString(opath);
2246 Py_BEGIN_ALLOW_THREADS
2247 res = chflags(path, flags);
2248 Py_END_ALLOW_THREADS
2249 if (res < 0)
2250 return posix_error_with_allocated_filename(opath);
2251 Py_DECREF(opath);
2252 Py_INCREF(Py_None);
2253 return Py_None;
Thomas Wouterscf297e42007-02-23 15:07:44 +00002254}
2255#endif /* HAVE_CHFLAGS */
2256
2257#ifdef HAVE_LCHFLAGS
2258PyDoc_STRVAR(posix_lchflags__doc__,
2259"lchflags(path, flags)\n\n\
2260Set file flags.\n\
2261This function will not follow symbolic links.");
2262
2263static PyObject *
2264posix_lchflags(PyObject *self, PyObject *args)
2265{
Victor Stinner8c62be82010-05-06 00:08:46 +00002266 PyObject *opath;
2267 char *path;
2268 unsigned long flags;
2269 int res;
2270 if (!PyArg_ParseTuple(args, "O&k:lchflags",
2271 PyUnicode_FSConverter, &opath, &flags))
2272 return NULL;
2273 path = PyBytes_AsString(opath);
2274 Py_BEGIN_ALLOW_THREADS
2275 res = lchflags(path, flags);
2276 Py_END_ALLOW_THREADS
2277 if (res < 0)
2278 return posix_error_with_allocated_filename(opath);
2279 Py_DECREF(opath);
2280 Py_INCREF(Py_None);
2281 return Py_None;
Thomas Wouterscf297e42007-02-23 15:07:44 +00002282}
2283#endif /* HAVE_LCHFLAGS */
2284
Martin v. Löwis244edc82001-10-04 22:44:26 +00002285#ifdef HAVE_CHROOT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002286PyDoc_STRVAR(posix_chroot__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002287"chroot(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002288Change root directory to path.");
Martin v. Löwis244edc82001-10-04 22:44:26 +00002289
2290static PyObject *
2291posix_chroot(PyObject *self, PyObject *args)
2292{
Victor Stinner8c62be82010-05-06 00:08:46 +00002293 return posix_1str(args, "O&:chroot", chroot);
Martin v. Löwis244edc82001-10-04 22:44:26 +00002294}
2295#endif
2296
Guido van Rossum21142a01999-01-08 21:05:37 +00002297#ifdef HAVE_FSYNC
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002298PyDoc_STRVAR(posix_fsync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002299"fsync(fildes)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002300force write of file with filedescriptor to disk.");
Guido van Rossum21142a01999-01-08 21:05:37 +00002301
2302static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00002303posix_fsync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00002304{
Stefan Krah0e803b32010-11-26 16:16:47 +00002305 return posix_fildes(fdobj, fsync);
Guido van Rossum21142a01999-01-08 21:05:37 +00002306}
2307#endif /* HAVE_FSYNC */
2308
Ross Lagerwall7807c352011-03-17 20:20:30 +02002309#ifdef HAVE_SYNC
2310PyDoc_STRVAR(posix_sync__doc__,
2311"sync()\n\n\
2312Force write of everything to disk.");
2313
2314static PyObject *
2315posix_sync(PyObject *self, PyObject *noargs)
2316{
2317 Py_BEGIN_ALLOW_THREADS
2318 sync();
2319 Py_END_ALLOW_THREADS
2320 Py_RETURN_NONE;
2321}
2322#endif
2323
Guido van Rossum21142a01999-01-08 21:05:37 +00002324#ifdef HAVE_FDATASYNC
Guido van Rossumecc23b02000-09-22 16:01:05 +00002325
Guido van Rossum7f58e2e2000-09-22 17:26:14 +00002326#ifdef __hpux
Guido van Rossumecc23b02000-09-22 16:01:05 +00002327extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
2328#endif
2329
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002330PyDoc_STRVAR(posix_fdatasync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002331"fdatasync(fildes)\n\n\
Guido van Rossum21142a01999-01-08 21:05:37 +00002332force write of file with filedescriptor to disk.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002333 does not force update of metadata.");
Guido van Rossum21142a01999-01-08 21:05:37 +00002334
2335static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00002336posix_fdatasync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00002337{
Stefan Krah0e803b32010-11-26 16:16:47 +00002338 return posix_fildes(fdobj, fdatasync);
Guido van Rossum21142a01999-01-08 21:05:37 +00002339}
2340#endif /* HAVE_FDATASYNC */
2341
2342
Fredrik Lundh10723342000-07-10 16:38:09 +00002343#ifdef HAVE_CHOWN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002344PyDoc_STRVAR(posix_chown__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002345"chown(path, uid, gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002346Change the owner and group id of path to the numeric uid and gid.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002347
Barry Warsaw53699e91996-12-10 23:23:01 +00002348static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002349posix_chown(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002350{
Victor Stinner8c62be82010-05-06 00:08:46 +00002351 PyObject *opath;
2352 char *path;
2353 long uid, gid;
2354 int res;
2355 if (!PyArg_ParseTuple(args, "O&ll:chown",
2356 PyUnicode_FSConverter, &opath,
2357 &uid, &gid))
2358 return NULL;
2359 path = PyBytes_AsString(opath);
2360 Py_BEGIN_ALLOW_THREADS
2361 res = chown(path, (uid_t) uid, (gid_t) gid);
2362 Py_END_ALLOW_THREADS
2363 if (res < 0)
2364 return posix_error_with_allocated_filename(opath);
2365 Py_DECREF(opath);
2366 Py_INCREF(Py_None);
2367 return Py_None;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002368}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002369#endif /* HAVE_CHOWN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002370
Christian Heimes4e30a842007-11-30 22:12:06 +00002371#ifdef HAVE_FCHOWN
2372PyDoc_STRVAR(posix_fchown__doc__,
2373"fchown(fd, uid, gid)\n\n\
2374Change the owner and group id of the file given by file descriptor\n\
2375fd to the numeric uid and gid.");
2376
2377static PyObject *
2378posix_fchown(PyObject *self, PyObject *args)
2379{
Victor Stinner8c62be82010-05-06 00:08:46 +00002380 int fd;
2381 long uid, gid;
2382 int res;
Victor Stinner26de69d2011-06-17 15:15:38 +02002383 if (!PyArg_ParseTuple(args, "ill:fchown", &fd, &uid, &gid))
Victor Stinner8c62be82010-05-06 00:08:46 +00002384 return NULL;
2385 Py_BEGIN_ALLOW_THREADS
2386 res = fchown(fd, (uid_t) uid, (gid_t) gid);
2387 Py_END_ALLOW_THREADS
2388 if (res < 0)
2389 return posix_error();
2390 Py_RETURN_NONE;
Christian Heimes4e30a842007-11-30 22:12:06 +00002391}
2392#endif /* HAVE_FCHOWN */
2393
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002394#ifdef HAVE_LCHOWN
2395PyDoc_STRVAR(posix_lchown__doc__,
2396"lchown(path, uid, gid)\n\n\
2397Change the owner and group id of path to the numeric uid and gid.\n\
2398This function will not follow symbolic links.");
2399
2400static PyObject *
2401posix_lchown(PyObject *self, PyObject *args)
2402{
Victor Stinner8c62be82010-05-06 00:08:46 +00002403 PyObject *opath;
2404 char *path;
2405 long uid, gid;
2406 int res;
2407 if (!PyArg_ParseTuple(args, "O&ll:lchown",
2408 PyUnicode_FSConverter, &opath,
2409 &uid, &gid))
2410 return NULL;
2411 path = PyBytes_AsString(opath);
2412 Py_BEGIN_ALLOW_THREADS
2413 res = lchown(path, (uid_t) uid, (gid_t) gid);
2414 Py_END_ALLOW_THREADS
2415 if (res < 0)
2416 return posix_error_with_allocated_filename(opath);
2417 Py_DECREF(opath);
2418 Py_INCREF(Py_None);
2419 return Py_None;
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002420}
2421#endif /* HAVE_LCHOWN */
2422
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002423
Guido van Rossum36bc6801995-06-14 22:54:23 +00002424#ifdef HAVE_GETCWD
Barry Warsaw53699e91996-12-10 23:23:01 +00002425static PyObject *
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002426posix_getcwd(int use_bytes)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002427{
Victor Stinner8c62be82010-05-06 00:08:46 +00002428 char buf[1026];
2429 char *res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002430
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002431#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00002432 if (!use_bytes) {
2433 wchar_t wbuf[1026];
2434 wchar_t *wbuf2 = wbuf;
2435 PyObject *resobj;
2436 DWORD len;
2437 Py_BEGIN_ALLOW_THREADS
2438 len = GetCurrentDirectoryW(sizeof wbuf/ sizeof wbuf[0], wbuf);
2439 /* If the buffer is large enough, len does not include the
2440 terminating \0. If the buffer is too small, len includes
2441 the space needed for the terminator. */
2442 if (len >= sizeof wbuf/ sizeof wbuf[0]) {
2443 wbuf2 = malloc(len * sizeof(wchar_t));
2444 if (wbuf2)
2445 len = GetCurrentDirectoryW(len, wbuf2);
2446 }
2447 Py_END_ALLOW_THREADS
2448 if (!wbuf2) {
2449 PyErr_NoMemory();
2450 return NULL;
2451 }
2452 if (!len) {
2453 if (wbuf2 != wbuf) free(wbuf2);
2454 return win32_error("getcwdu", NULL);
2455 }
2456 resobj = PyUnicode_FromWideChar(wbuf2, len);
2457 if (wbuf2 != wbuf) free(wbuf2);
2458 return resobj;
2459 }
Victor Stinnerf7c5ae22011-11-16 23:43:07 +01002460
2461 if (win32_warn_bytes_api())
2462 return NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002463#endif
2464
Victor Stinner8c62be82010-05-06 00:08:46 +00002465 Py_BEGIN_ALLOW_THREADS
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002466#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00002467 res = _getcwd2(buf, sizeof buf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002468#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002469 res = getcwd(buf, sizeof buf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002470#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00002471 Py_END_ALLOW_THREADS
2472 if (res == NULL)
2473 return posix_error();
2474 if (use_bytes)
2475 return PyBytes_FromStringAndSize(buf, strlen(buf));
Victor Stinner97c18ab2010-05-07 16:34:53 +00002476 return PyUnicode_DecodeFSDefault(buf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002477}
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002478
2479PyDoc_STRVAR(posix_getcwd__doc__,
2480"getcwd() -> path\n\n\
2481Return a unicode string representing the current working directory.");
2482
2483static PyObject *
2484posix_getcwd_unicode(PyObject *self)
2485{
2486 return posix_getcwd(0);
2487}
2488
2489PyDoc_STRVAR(posix_getcwdb__doc__,
2490"getcwdb() -> path\n\n\
2491Return a bytes string representing the current working directory.");
2492
2493static PyObject *
2494posix_getcwd_bytes(PyObject *self)
2495{
2496 return posix_getcwd(1);
2497}
Guido van Rossum36bc6801995-06-14 22:54:23 +00002498#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002499
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002500
Guido van Rossumb6775db1994-08-01 11:34:53 +00002501#ifdef HAVE_LINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002502PyDoc_STRVAR(posix_link__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002503"link(src, dst)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002504Create a hard link to a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002505
Barry Warsaw53699e91996-12-10 23:23:01 +00002506static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002507posix_link(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002508{
Victor Stinner8c62be82010-05-06 00:08:46 +00002509 return posix_2str(args, "O&O&:link", link);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002510}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002511#endif /* HAVE_LINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002512
Brian Curtin1b9df392010-11-24 20:24:31 +00002513#ifdef MS_WINDOWS
2514PyDoc_STRVAR(win32_link__doc__,
2515"link(src, dst)\n\n\
2516Create a hard link to a file.");
2517
2518static PyObject *
2519win32_link(PyObject *self, PyObject *args)
2520{
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002521 PyObject *src, *dst;
2522 BOOL ok;
Brian Curtin1b9df392010-11-24 20:24:31 +00002523
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002524 if (PyArg_ParseTuple(args, "UU:link", &src, &dst))
Victor Stinnereb5657a2011-09-30 01:44:27 +02002525 {
2526 wchar_t *wsrc, *wdst;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002527
2528 wsrc = PyUnicode_AsUnicode(src);
Victor Stinnereb5657a2011-09-30 01:44:27 +02002529 if (wsrc == NULL)
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002530 goto error;
2531 wdst = PyUnicode_AsUnicode(dst);
Victor Stinnereb5657a2011-09-30 01:44:27 +02002532 if (wdst == NULL)
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002533 goto error;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002534
Brian Curtinfc889c42010-11-28 23:59:46 +00002535 Py_BEGIN_ALLOW_THREADS
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002536 ok = CreateHardLinkW(wdst, wsrc, NULL);
Brian Curtinfc889c42010-11-28 23:59:46 +00002537 Py_END_ALLOW_THREADS
2538
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002539 if (!ok)
Brian Curtinfc889c42010-11-28 23:59:46 +00002540 return win32_error("link", NULL);
Brian Curtinfc889c42010-11-28 23:59:46 +00002541 Py_RETURN_NONE;
2542 }
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002543 else {
2544 PyErr_Clear();
2545 if (!PyArg_ParseTuple(args, "O&O&:link",
2546 PyUnicode_FSConverter, &src,
2547 PyUnicode_FSConverter, &dst))
2548 return NULL;
Brian Curtinfc889c42010-11-28 23:59:46 +00002549
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002550 if (win32_warn_bytes_api())
2551 goto error;
Brian Curtinfc889c42010-11-28 23:59:46 +00002552
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002553 Py_BEGIN_ALLOW_THREADS
2554 ok = CreateHardLinkA(PyBytes_AS_STRING(dst),
2555 PyBytes_AS_STRING(src),
2556 NULL);
2557 Py_END_ALLOW_THREADS
2558
2559 Py_XDECREF(src);
2560 Py_XDECREF(dst);
2561
2562 if (!ok)
2563 return win32_error("link", NULL);
2564 Py_RETURN_NONE;
2565
2566 error:
2567 Py_XDECREF(src);
2568 Py_XDECREF(dst);
Brian Curtin1b9df392010-11-24 20:24:31 +00002569 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002570 }
Brian Curtin1b9df392010-11-24 20:24:31 +00002571}
2572#endif /* MS_WINDOWS */
2573
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002574
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002575PyDoc_STRVAR(posix_listdir__doc__,
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002576"listdir([path]) -> list_of_strings\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002577Return a list containing the names of the entries in the directory.\n\
2578\n\
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002579 path: path of directory to list (default: '.')\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002580\n\
2581The list is in arbitrary order. It does not include the special\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002582entries '.' and '..' even if they are present in the directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002583
Barry Warsaw53699e91996-12-10 23:23:01 +00002584static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002585posix_listdir(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002586{
Victor Stinner8c62be82010-05-06 00:08:46 +00002587 /* XXX Should redo this putting the (now four) versions of opendir
2588 in separate files instead of having them all here... */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002589#if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002590
Victor Stinner8c62be82010-05-06 00:08:46 +00002591 PyObject *d, *v;
2592 HANDLE hFindFile;
2593 BOOL result;
2594 WIN32_FIND_DATA FileData;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002595 const char *path;
2596 Py_ssize_t pathlen;
Victor Stinner8c62be82010-05-06 00:08:46 +00002597 char namebuf[MAX_PATH+5]; /* Overallocate for \\*.*\0 */
2598 char *bufptr = namebuf;
2599 Py_ssize_t len = sizeof(namebuf)-5; /* only claim to have space for MAX_PATH */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002600
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002601 PyObject *po = NULL;
2602 if (PyArg_ParseTuple(args, "|U:listdir", &po)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002603 WIN32_FIND_DATAW wFileData;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002604 wchar_t *wnamebuf, *po_wchars;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00002605
Antoine Pitrou3d330ad2010-09-14 10:08:08 +00002606 if (po == NULL) { /* Default arg: "." */
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002607 po_wchars = L".";
2608 len = 1;
2609 } else {
Victor Stinnerbeac78b2011-10-11 21:55:01 +02002610 po_wchars = PyUnicode_AsUnicodeAndSize(po, &len);
Victor Stinnereb5657a2011-09-30 01:44:27 +02002611 if (po_wchars == NULL)
2612 return NULL;
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002613 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002614 /* Overallocate for \\*.*\0 */
Victor Stinner8c62be82010-05-06 00:08:46 +00002615 wnamebuf = malloc((len + 5) * sizeof(wchar_t));
2616 if (!wnamebuf) {
2617 PyErr_NoMemory();
2618 return NULL;
2619 }
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002620 wcscpy(wnamebuf, po_wchars);
Victor Stinner8c62be82010-05-06 00:08:46 +00002621 if (len > 0) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02002622 wchar_t wch = wnamebuf[len-1];
Victor Stinner8c62be82010-05-06 00:08:46 +00002623 if (wch != L'/' && wch != L'\\' && wch != L':')
2624 wnamebuf[len++] = L'\\';
2625 wcscpy(wnamebuf + len, L"*.*");
2626 }
2627 if ((d = PyList_New(0)) == NULL) {
2628 free(wnamebuf);
2629 return NULL;
2630 }
Antoine Pitroub73caab2010-08-09 23:39:31 +00002631 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002632 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
Antoine Pitroub73caab2010-08-09 23:39:31 +00002633 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002634 if (hFindFile == INVALID_HANDLE_VALUE) {
2635 int error = GetLastError();
2636 if (error == ERROR_FILE_NOT_FOUND) {
2637 free(wnamebuf);
2638 return d;
2639 }
2640 Py_DECREF(d);
2641 win32_error_unicode("FindFirstFileW", wnamebuf);
2642 free(wnamebuf);
2643 return NULL;
2644 }
2645 do {
2646 /* Skip over . and .. */
2647 if (wcscmp(wFileData.cFileName, L".") != 0 &&
2648 wcscmp(wFileData.cFileName, L"..") != 0) {
Victor Stinner9d3b93b2011-11-22 02:27:30 +01002649 v = PyUnicode_FromWideChar(wFileData.cFileName, wcslen(wFileData.cFileName));
Victor Stinner8c62be82010-05-06 00:08:46 +00002650 if (v == NULL) {
2651 Py_DECREF(d);
2652 d = NULL;
2653 break;
2654 }
2655 if (PyList_Append(d, v) != 0) {
2656 Py_DECREF(v);
2657 Py_DECREF(d);
2658 d = NULL;
2659 break;
2660 }
2661 Py_DECREF(v);
2662 }
2663 Py_BEGIN_ALLOW_THREADS
2664 result = FindNextFileW(hFindFile, &wFileData);
2665 Py_END_ALLOW_THREADS
2666 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2667 it got to the end of the directory. */
2668 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2669 Py_DECREF(d);
2670 win32_error_unicode("FindNextFileW", wnamebuf);
2671 FindClose(hFindFile);
2672 free(wnamebuf);
2673 return NULL;
2674 }
2675 } while (result == TRUE);
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002676
Victor Stinner8c62be82010-05-06 00:08:46 +00002677 if (FindClose(hFindFile) == FALSE) {
2678 Py_DECREF(d);
2679 win32_error_unicode("FindClose", wnamebuf);
2680 free(wnamebuf);
2681 return NULL;
2682 }
2683 free(wnamebuf);
2684 return d;
2685 }
2686 /* Drop the argument parsing error as narrow strings
2687 are also valid. */
2688 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002689
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002690 if (!PyArg_ParseTuple(args, "y#:listdir", &path, &pathlen))
Victor Stinner8c62be82010-05-06 00:08:46 +00002691 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002692 if (win32_warn_bytes_api())
2693 return NULL;
2694 if (pathlen+1 > MAX_PATH) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002695 PyErr_SetString(PyExc_ValueError, "path too long");
Victor Stinner8c62be82010-05-06 00:08:46 +00002696 return NULL;
2697 }
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002698 strcpy(namebuf, path);
2699 len = pathlen;
Victor Stinner8c62be82010-05-06 00:08:46 +00002700 if (len > 0) {
2701 char ch = namebuf[len-1];
2702 if (ch != SEP && ch != ALTSEP && ch != ':')
2703 namebuf[len++] = '/';
2704 strcpy(namebuf + len, "*.*");
2705 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002706
Victor Stinner8c62be82010-05-06 00:08:46 +00002707 if ((d = PyList_New(0)) == NULL)
2708 return NULL;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002709
Antoine Pitroub73caab2010-08-09 23:39:31 +00002710 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002711 hFindFile = FindFirstFile(namebuf, &FileData);
Antoine Pitroub73caab2010-08-09 23:39:31 +00002712 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002713 if (hFindFile == INVALID_HANDLE_VALUE) {
2714 int error = GetLastError();
2715 if (error == ERROR_FILE_NOT_FOUND)
2716 return d;
2717 Py_DECREF(d);
2718 return win32_error("FindFirstFile", namebuf);
2719 }
2720 do {
2721 /* Skip over . and .. */
2722 if (strcmp(FileData.cFileName, ".") != 0 &&
2723 strcmp(FileData.cFileName, "..") != 0) {
2724 v = PyBytes_FromString(FileData.cFileName);
2725 if (v == NULL) {
2726 Py_DECREF(d);
2727 d = NULL;
2728 break;
2729 }
2730 if (PyList_Append(d, v) != 0) {
2731 Py_DECREF(v);
2732 Py_DECREF(d);
2733 d = NULL;
2734 break;
2735 }
2736 Py_DECREF(v);
2737 }
2738 Py_BEGIN_ALLOW_THREADS
2739 result = FindNextFile(hFindFile, &FileData);
2740 Py_END_ALLOW_THREADS
2741 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2742 it got to the end of the directory. */
2743 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2744 Py_DECREF(d);
2745 win32_error("FindNextFile", namebuf);
2746 FindClose(hFindFile);
2747 return NULL;
2748 }
2749 } while (result == TRUE);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002750
Victor Stinner8c62be82010-05-06 00:08:46 +00002751 if (FindClose(hFindFile) == FALSE) {
2752 Py_DECREF(d);
2753 return win32_error("FindClose", namebuf);
2754 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002755
Victor Stinner8c62be82010-05-06 00:08:46 +00002756 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002757
Tim Peters0bb44a42000-09-15 07:44:49 +00002758#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002759
2760#ifndef MAX_PATH
2761#define MAX_PATH CCHMAXPATH
2762#endif
Martin v. Löwis011e8422009-05-05 04:43:17 +00002763 PyObject *oname;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002764 char *name, *pt;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00002765 Py_ssize_t len;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002766 PyObject *d, *v;
2767 char namebuf[MAX_PATH+5];
2768 HDIR hdir = 1;
2769 ULONG srchcnt = 1;
2770 FILEFINDBUF3 ep;
2771 APIRET rc;
2772
Victor Stinner8c62be82010-05-06 00:08:46 +00002773 if (!PyArg_ParseTuple(args, "O&:listdir",
Martin v. Löwis011e8422009-05-05 04:43:17 +00002774 PyUnicode_FSConverter, &oname))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002775 return NULL;
Victor Stinnerdcb24032010-04-22 12:08:36 +00002776 name = PyBytes_AsString(oname);
2777 len = PyBytes_GET_SIZE(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002778 if (len >= MAX_PATH) {
Victor Stinnerdcb24032010-04-22 12:08:36 +00002779 Py_DECREF(oname);
Neal Norwitz6c913782007-10-14 03:23:09 +00002780 PyErr_SetString(PyExc_ValueError, "path too long");
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002781 return NULL;
2782 }
2783 strcpy(namebuf, name);
2784 for (pt = namebuf; *pt; pt++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002785 if (*pt == ALTSEP)
2786 *pt = SEP;
2787 if (namebuf[len-1] != SEP)
2788 namebuf[len++] = SEP;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002789 strcpy(namebuf + len, "*.*");
2790
Neal Norwitz6c913782007-10-14 03:23:09 +00002791 if ((d = PyList_New(0)) == NULL) {
Victor Stinnerdcb24032010-04-22 12:08:36 +00002792 Py_DECREF(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002793 return NULL;
Alexandre Vassalotti4167ebc2007-10-14 02:54:41 +00002794 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002795
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002796 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
2797 &hdir, /* Handle to Use While Search Directory */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002798 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002799 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
2800 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
2801 FIL_STANDARD); /* Format of Entry (EAs or Not) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002802
2803 if (rc != NO_ERROR) {
2804 errno = ENOENT;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002805 return posix_error_with_allocated_filename(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002806 }
2807
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002808 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002809 do {
2810 if (ep.achName[0] == '.'
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002811 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002812 continue; /* Skip Over "." and ".." Names */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002813
2814 strcpy(namebuf, ep.achName);
2815
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002816 /* Leave Case of Name Alone -- In Native Form */
2817 /* (Removed Forced Lowercasing Code) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002818
Christian Heimes72b710a2008-05-26 13:28:38 +00002819 v = PyBytes_FromString(namebuf);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002820 if (v == NULL) {
2821 Py_DECREF(d);
2822 d = NULL;
2823 break;
2824 }
2825 if (PyList_Append(d, v) != 0) {
2826 Py_DECREF(v);
2827 Py_DECREF(d);
2828 d = NULL;
2829 break;
2830 }
2831 Py_DECREF(v);
2832 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
2833 }
2834
Victor Stinnerdcb24032010-04-22 12:08:36 +00002835 Py_DECREF(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002836 return d;
2837#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002838 PyObject *oname;
2839 char *name;
2840 PyObject *d, *v;
2841 DIR *dirp;
2842 struct dirent *ep;
2843 int arg_is_unicode = 1;
Just van Rossum96b1c902003-03-03 17:32:15 +00002844
Victor Stinner8c62be82010-05-06 00:08:46 +00002845 errno = 0;
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002846 /* v is never read, so it does not need to be initialized yet. */
2847 if (!PyArg_ParseTuple(args, "|U:listdir", &v)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002848 arg_is_unicode = 0;
2849 PyErr_Clear();
2850 }
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002851 oname = NULL;
2852 if (!PyArg_ParseTuple(args, "|O&:listdir", PyUnicode_FSConverter, &oname))
Victor Stinner8c62be82010-05-06 00:08:46 +00002853 return NULL;
Antoine Pitrou3d330ad2010-09-14 10:08:08 +00002854 if (oname == NULL) { /* Default arg: "." */
Stefan Krah0e803b32010-11-26 16:16:47 +00002855 oname = PyBytes_FromString(".");
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002856 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002857 name = PyBytes_AsString(oname);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002858 Py_BEGIN_ALLOW_THREADS
2859 dirp = opendir(name);
2860 Py_END_ALLOW_THREADS
2861 if (dirp == NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002862 return posix_error_with_allocated_filename(oname);
2863 }
2864 if ((d = PyList_New(0)) == NULL) {
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002865 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002866 closedir(dirp);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002867 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002868 Py_DECREF(oname);
2869 return NULL;
2870 }
2871 for (;;) {
2872 errno = 0;
2873 Py_BEGIN_ALLOW_THREADS
2874 ep = readdir(dirp);
2875 Py_END_ALLOW_THREADS
2876 if (ep == NULL) {
2877 if (errno == 0) {
2878 break;
2879 } else {
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002880 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002881 closedir(dirp);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002882 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002883 Py_DECREF(d);
2884 return posix_error_with_allocated_filename(oname);
2885 }
2886 }
2887 if (ep->d_name[0] == '.' &&
2888 (NAMLEN(ep) == 1 ||
2889 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
2890 continue;
Victor Stinnera45598a2010-05-14 16:35:39 +00002891 if (arg_is_unicode)
2892 v = PyUnicode_DecodeFSDefaultAndSize(ep->d_name, NAMLEN(ep));
2893 else
2894 v = PyBytes_FromStringAndSize(ep->d_name, NAMLEN(ep));
Victor Stinner8c62be82010-05-06 00:08:46 +00002895 if (v == NULL) {
Victor Stinnera45598a2010-05-14 16:35:39 +00002896 Py_CLEAR(d);
Victor Stinner8c62be82010-05-06 00:08:46 +00002897 break;
2898 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002899 if (PyList_Append(d, v) != 0) {
2900 Py_DECREF(v);
Victor Stinnera45598a2010-05-14 16:35:39 +00002901 Py_CLEAR(d);
Victor Stinner8c62be82010-05-06 00:08:46 +00002902 break;
2903 }
2904 Py_DECREF(v);
2905 }
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002906 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002907 closedir(dirp);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002908 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002909 Py_DECREF(oname);
Guido van Rossum0ee42cd1991-04-08 21:01:03 +00002910
Victor Stinner8c62be82010-05-06 00:08:46 +00002911 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002912
Tim Peters0bb44a42000-09-15 07:44:49 +00002913#endif /* which OS */
2914} /* end of posix_listdir */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002915
Antoine Pitrou8250e232011-02-25 23:41:16 +00002916#ifdef HAVE_FDOPENDIR
Charles-François Natali77940902012-02-06 19:54:48 +01002917PyDoc_STRVAR(posix_flistdir__doc__,
2918"flistdir(fd) -> list_of_strings\n\n\
Charles-François Natali76961fa2012-01-10 20:25:09 +01002919Like listdir(), but uses a file descriptor instead.");
Antoine Pitrou8250e232011-02-25 23:41:16 +00002920
2921static PyObject *
Charles-François Natali77940902012-02-06 19:54:48 +01002922posix_flistdir(PyObject *self, PyObject *args)
Antoine Pitrou8250e232011-02-25 23:41:16 +00002923{
2924 PyObject *d, *v;
2925 DIR *dirp;
2926 struct dirent *ep;
2927 int fd;
2928
2929 errno = 0;
Charles-François Natali77940902012-02-06 19:54:48 +01002930 if (!PyArg_ParseTuple(args, "i:flistdir", &fd))
Antoine Pitrou8250e232011-02-25 23:41:16 +00002931 return NULL;
Charles-François Natali76961fa2012-01-10 20:25:09 +01002932 /* closedir() closes the FD, so we duplicate it */
2933 fd = dup(fd);
2934 if (fd < 0)
2935 return posix_error();
Antoine Pitrou8250e232011-02-25 23:41:16 +00002936 Py_BEGIN_ALLOW_THREADS
2937 dirp = fdopendir(fd);
2938 Py_END_ALLOW_THREADS
2939 if (dirp == NULL) {
2940 close(fd);
2941 return posix_error();
2942 }
2943 if ((d = PyList_New(0)) == NULL) {
2944 Py_BEGIN_ALLOW_THREADS
2945 closedir(dirp);
2946 Py_END_ALLOW_THREADS
2947 return NULL;
2948 }
2949 for (;;) {
2950 errno = 0;
2951 Py_BEGIN_ALLOW_THREADS
2952 ep = readdir(dirp);
2953 Py_END_ALLOW_THREADS
2954 if (ep == NULL) {
2955 if (errno == 0) {
2956 break;
2957 } else {
2958 Py_BEGIN_ALLOW_THREADS
Charles-François Natalif2840a82012-01-08 20:30:47 +01002959 rewinddir(dirp);
Antoine Pitrou8250e232011-02-25 23:41:16 +00002960 closedir(dirp);
2961 Py_END_ALLOW_THREADS
2962 Py_DECREF(d);
2963 return posix_error();
2964 }
2965 }
2966 if (ep->d_name[0] == '.' &&
2967 (NAMLEN(ep) == 1 ||
2968 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
2969 continue;
2970 v = PyUnicode_DecodeFSDefaultAndSize(ep->d_name, NAMLEN(ep));
2971 if (v == NULL) {
2972 Py_CLEAR(d);
2973 break;
2974 }
2975 if (PyList_Append(d, v) != 0) {
2976 Py_DECREF(v);
2977 Py_CLEAR(d);
2978 break;
2979 }
2980 Py_DECREF(v);
2981 }
2982 Py_BEGIN_ALLOW_THREADS
Charles-François Natalif2840a82012-01-08 20:30:47 +01002983 rewinddir(dirp);
Antoine Pitrou8250e232011-02-25 23:41:16 +00002984 closedir(dirp);
2985 Py_END_ALLOW_THREADS
2986
2987 return d;
2988}
2989#endif
2990
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002991#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00002992/* A helper function for abspath on win32 */
2993static PyObject *
2994posix__getfullpathname(PyObject *self, PyObject *args)
2995{
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002996 const char *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00002997 char outbuf[MAX_PATH*2];
2998 char *temp;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002999 PyObject *po;
3000
3001 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po))
3002 {
3003 wchar_t *wpath;
3004 wchar_t woutbuf[MAX_PATH*2], *woutbufp = woutbuf;
3005 wchar_t *wtemp;
Victor Stinner8c62be82010-05-06 00:08:46 +00003006 DWORD result;
3007 PyObject *v;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003008
3009 wpath = PyUnicode_AsUnicode(po);
3010 if (wpath == NULL)
3011 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00003012 result = GetFullPathNameW(wpath,
Victor Stinner63941882011-09-29 00:42:28 +02003013 Py_ARRAY_LENGTH(woutbuf),
Victor Stinner8c62be82010-05-06 00:08:46 +00003014 woutbuf, &wtemp);
Victor Stinner63941882011-09-29 00:42:28 +02003015 if (result > Py_ARRAY_LENGTH(woutbuf)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02003016 woutbufp = malloc(result * sizeof(wchar_t));
Victor Stinner8c62be82010-05-06 00:08:46 +00003017 if (!woutbufp)
3018 return PyErr_NoMemory();
3019 result = GetFullPathNameW(wpath, result, woutbufp, &wtemp);
3020 }
3021 if (result)
Victor Stinner9d3b93b2011-11-22 02:27:30 +01003022 v = PyUnicode_FromWideChar(woutbufp, wcslen(woutbufp));
Victor Stinner8c62be82010-05-06 00:08:46 +00003023 else
Victor Stinnereb5657a2011-09-30 01:44:27 +02003024 v = win32_error_object("GetFullPathNameW", po);
Victor Stinner8c62be82010-05-06 00:08:46 +00003025 if (woutbufp != woutbuf)
3026 free(woutbufp);
3027 return v;
3028 }
3029 /* Drop the argument parsing error as narrow strings
3030 are also valid. */
3031 PyErr_Clear();
Victor Stinnereb5657a2011-09-30 01:44:27 +02003032
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003033 if (!PyArg_ParseTuple (args, "y:_getfullpathname",
3034 &path))
Victor Stinner8c62be82010-05-06 00:08:46 +00003035 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003036 if (win32_warn_bytes_api())
3037 return NULL;
Victor Stinner63941882011-09-29 00:42:28 +02003038 if (!GetFullPathName(path, Py_ARRAY_LENGTH(outbuf),
Victor Stinner8c62be82010-05-06 00:08:46 +00003039 outbuf, &temp)) {
3040 win32_error("GetFullPathName", path);
Victor Stinner8c62be82010-05-06 00:08:46 +00003041 return NULL;
3042 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003043 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
3044 return PyUnicode_Decode(outbuf, strlen(outbuf),
3045 Py_FileSystemDefaultEncoding, NULL);
3046 }
3047 return PyBytes_FromString(outbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00003048} /* end of posix__getfullpathname */
Brian Curtind40e6f72010-07-08 21:39:08 +00003049
Brian Curtind25aef52011-06-13 15:16:04 -05003050
Brian Curtinf5e76d02010-11-24 13:14:05 +00003051
Brian Curtind40e6f72010-07-08 21:39:08 +00003052/* A helper function for samepath on windows */
3053static PyObject *
3054posix__getfinalpathname(PyObject *self, PyObject *args)
3055{
3056 HANDLE hFile;
3057 int buf_size;
3058 wchar_t *target_path;
3059 int result_length;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003060 PyObject *po, *result;
Brian Curtind40e6f72010-07-08 21:39:08 +00003061 wchar_t *path;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00003062
Victor Stinnereb5657a2011-09-30 01:44:27 +02003063 if (!PyArg_ParseTuple(args, "U|:_getfinalpathname", &po))
Brian Curtind40e6f72010-07-08 21:39:08 +00003064 return NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003065 path = PyUnicode_AsUnicode(po);
3066 if (path == NULL)
3067 return NULL;
Brian Curtind40e6f72010-07-08 21:39:08 +00003068
3069 if(!check_GetFinalPathNameByHandle()) {
3070 /* If the OS doesn't have GetFinalPathNameByHandle, return a
3071 NotImplementedError. */
3072 return PyErr_Format(PyExc_NotImplementedError,
3073 "GetFinalPathNameByHandle not available on this platform");
3074 }
3075
3076 hFile = CreateFileW(
3077 path,
3078 0, /* desired access */
3079 0, /* share mode */
3080 NULL, /* security attributes */
3081 OPEN_EXISTING,
3082 /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
3083 FILE_FLAG_BACKUP_SEMANTICS,
3084 NULL);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00003085
Victor Stinnereb5657a2011-09-30 01:44:27 +02003086 if(hFile == INVALID_HANDLE_VALUE)
3087 return win32_error_object("CreateFileW", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00003088
3089 /* We have a good handle to the target, use it to determine the
3090 target path name. */
3091 buf_size = Py_GetFinalPathNameByHandleW(hFile, 0, 0, VOLUME_NAME_NT);
3092
3093 if(!buf_size)
Victor Stinnereb5657a2011-09-30 01:44:27 +02003094 return win32_error_object("GetFinalPathNameByHandle", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00003095
3096 target_path = (wchar_t *)malloc((buf_size+1)*sizeof(wchar_t));
3097 if(!target_path)
3098 return PyErr_NoMemory();
3099
3100 result_length = Py_GetFinalPathNameByHandleW(hFile, target_path,
3101 buf_size, VOLUME_NAME_DOS);
3102 if(!result_length)
Victor Stinnereb5657a2011-09-30 01:44:27 +02003103 return win32_error_object("GetFinalPathNamyByHandle", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00003104
3105 if(!CloseHandle(hFile))
Victor Stinnereb5657a2011-09-30 01:44:27 +02003106 return win32_error_object("CloseHandle", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00003107
3108 target_path[result_length] = 0;
Victor Stinner9d3b93b2011-11-22 02:27:30 +01003109 result = PyUnicode_FromWideChar(target_path, result_length);
Brian Curtind40e6f72010-07-08 21:39:08 +00003110 free(target_path);
3111 return result;
3112
3113} /* end of posix__getfinalpathname */
Brian Curtin62857742010-09-06 17:07:27 +00003114
3115static PyObject *
3116posix__getfileinformation(PyObject *self, PyObject *args)
3117{
3118 HANDLE hFile;
3119 BY_HANDLE_FILE_INFORMATION info;
3120 int fd;
3121
3122 if (!PyArg_ParseTuple(args, "i:_getfileinformation", &fd))
3123 return NULL;
3124
Hirokazu Yamamoto26253bb2010-12-05 04:16:47 +00003125 if (!_PyVerify_fd(fd))
3126 return posix_error();
Brian Curtin62857742010-09-06 17:07:27 +00003127
3128 hFile = (HANDLE)_get_osfhandle(fd);
3129 if (hFile == INVALID_HANDLE_VALUE)
Hirokazu Yamamoto26253bb2010-12-05 04:16:47 +00003130 return posix_error();
Brian Curtin62857742010-09-06 17:07:27 +00003131
3132 if (!GetFileInformationByHandle(hFile, &info))
3133 return win32_error("_getfileinformation", NULL);
3134
3135 return Py_BuildValue("iii", info.dwVolumeSerialNumber,
3136 info.nFileIndexHigh,
3137 info.nFileIndexLow);
3138}
Brian Curtin9c669cc2011-06-08 18:17:18 -05003139
Brian Curtin95d028f2011-06-09 09:10:38 -05003140PyDoc_STRVAR(posix__isdir__doc__,
3141"Return true if the pathname refers to an existing directory.");
3142
Brian Curtin9c669cc2011-06-08 18:17:18 -05003143static PyObject *
3144posix__isdir(PyObject *self, PyObject *args)
3145{
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003146 const char *path;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003147 PyObject *po;
Brian Curtin9c669cc2011-06-08 18:17:18 -05003148 DWORD attributes;
3149
3150 if (PyArg_ParseTuple(args, "U|:_isdir", &po)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02003151 wchar_t *wpath = PyUnicode_AsUnicode(po);
3152 if (wpath == NULL)
3153 return NULL;
Brian Curtin9c669cc2011-06-08 18:17:18 -05003154
3155 attributes = GetFileAttributesW(wpath);
3156 if (attributes == INVALID_FILE_ATTRIBUTES)
3157 Py_RETURN_FALSE;
3158 goto check;
3159 }
3160 /* Drop the argument parsing error as narrow strings
3161 are also valid. */
3162 PyErr_Clear();
3163
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003164 if (!PyArg_ParseTuple(args, "y:_isdir", &path))
Brian Curtin9c669cc2011-06-08 18:17:18 -05003165 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003166 if (win32_warn_bytes_api())
3167 return NULL;
Brian Curtin9c669cc2011-06-08 18:17:18 -05003168 attributes = GetFileAttributesA(path);
3169 if (attributes == INVALID_FILE_ATTRIBUTES)
3170 Py_RETURN_FALSE;
3171
3172check:
3173 if (attributes & FILE_ATTRIBUTE_DIRECTORY)
3174 Py_RETURN_TRUE;
3175 else
3176 Py_RETURN_FALSE;
3177}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00003178#endif /* MS_WINDOWS */
Mark Hammondef8b6542001-05-13 08:04:26 +00003179
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003180PyDoc_STRVAR(posix_mkdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003181"mkdir(path [, mode=0777])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003182Create a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003183
Barry Warsaw53699e91996-12-10 23:23:01 +00003184static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003185posix_mkdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003186{
Victor Stinner8c62be82010-05-06 00:08:46 +00003187 int res;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003188 const char *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00003189 int mode = 0777;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003190
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003191#ifdef MS_WINDOWS
Victor Stinnereb5657a2011-09-30 01:44:27 +02003192 PyObject *po;
3193 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode))
3194 {
3195 wchar_t *wpath = PyUnicode_AsUnicode(po);
3196 if (wpath == NULL)
3197 return NULL;
3198
Victor Stinner8c62be82010-05-06 00:08:46 +00003199 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02003200 res = CreateDirectoryW(wpath, NULL);
Victor Stinner8c62be82010-05-06 00:08:46 +00003201 Py_END_ALLOW_THREADS
3202 if (!res)
Victor Stinnereb5657a2011-09-30 01:44:27 +02003203 return win32_error_object("mkdir", po);
Victor Stinner8c62be82010-05-06 00:08:46 +00003204 Py_INCREF(Py_None);
3205 return Py_None;
3206 }
3207 /* Drop the argument parsing error as narrow strings
3208 are also valid. */
3209 PyErr_Clear();
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003210 if (!PyArg_ParseTuple(args, "y|i:mkdir", &path, &mode))
Victor Stinner8c62be82010-05-06 00:08:46 +00003211 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003212 if (win32_warn_bytes_api())
3213 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00003214 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00003215 res = CreateDirectoryA(path, NULL);
3216 Py_END_ALLOW_THREADS
3217 if (!res) {
3218 win32_error("mkdir", path);
Victor Stinner8c62be82010-05-06 00:08:46 +00003219 return NULL;
3220 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003221 Py_INCREF(Py_None);
3222 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003223#else
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003224 PyObject *opath;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003225
Victor Stinner8c62be82010-05-06 00:08:46 +00003226 if (!PyArg_ParseTuple(args, "O&|i:mkdir",
3227 PyUnicode_FSConverter, &opath, &mode))
3228 return NULL;
3229 path = PyBytes_AsString(opath);
3230 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00003231#if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
Victor Stinner8c62be82010-05-06 00:08:46 +00003232 res = mkdir(path);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003233#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003234 res = mkdir(path, mode);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003235#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003236 Py_END_ALLOW_THREADS
3237 if (res < 0)
3238 return posix_error_with_allocated_filename(opath);
3239 Py_DECREF(opath);
3240 Py_INCREF(Py_None);
3241 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003242#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003243}
3244
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003245
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003246/* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
3247#if defined(HAVE_SYS_RESOURCE_H)
Thomas Wouterse38b2f12001-07-11 22:35:31 +00003248#include <sys/resource.h>
3249#endif
Thomas Wouterse38b2f12001-07-11 22:35:31 +00003250
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003251
3252#ifdef HAVE_NICE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003253PyDoc_STRVAR(posix_nice__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003254"nice(inc) -> new_priority\n\n\
3255Decrease the priority of process by inc and return the new priority.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003256
Barry Warsaw53699e91996-12-10 23:23:01 +00003257static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003258posix_nice(PyObject *self, PyObject *args)
Guido van Rossum775f4da1993-01-09 17:18:52 +00003259{
Victor Stinner8c62be82010-05-06 00:08:46 +00003260 int increment, value;
Guido van Rossum775f4da1993-01-09 17:18:52 +00003261
Victor Stinner8c62be82010-05-06 00:08:46 +00003262 if (!PyArg_ParseTuple(args, "i:nice", &increment))
3263 return NULL;
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00003264
Victor Stinner8c62be82010-05-06 00:08:46 +00003265 /* There are two flavours of 'nice': one that returns the new
3266 priority (as required by almost all standards out there) and the
3267 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
3268 the use of getpriority() to get the new priority.
Tim Peters5aa91602002-01-30 05:46:57 +00003269
Victor Stinner8c62be82010-05-06 00:08:46 +00003270 If we are of the nice family that returns the new priority, we
3271 need to clear errno before the call, and check if errno is filled
3272 before calling posix_error() on a returnvalue of -1, because the
3273 -1 may be the actual new priority! */
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00003274
Victor Stinner8c62be82010-05-06 00:08:46 +00003275 errno = 0;
3276 value = nice(increment);
Thomas Wouterse38b2f12001-07-11 22:35:31 +00003277#if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
Victor Stinner8c62be82010-05-06 00:08:46 +00003278 if (value == 0)
3279 value = getpriority(PRIO_PROCESS, 0);
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00003280#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003281 if (value == -1 && errno != 0)
3282 /* either nice() or getpriority() returned an error */
3283 return posix_error();
3284 return PyLong_FromLong((long) value);
Guido van Rossum775f4da1993-01-09 17:18:52 +00003285}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003286#endif /* HAVE_NICE */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003287
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003288
3289#ifdef HAVE_GETPRIORITY
3290PyDoc_STRVAR(posix_getpriority__doc__,
3291"getpriority(which, who) -> current_priority\n\n\
3292Get program scheduling priority.");
3293
3294static PyObject *
3295posix_getpriority(PyObject *self, PyObject *args)
3296{
3297 int which, who, retval;
3298
3299 if (!PyArg_ParseTuple(args, "ii", &which, &who))
3300 return NULL;
3301 errno = 0;
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003302 retval = getpriority(which, who);
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003303 if (errno != 0)
3304 return posix_error();
3305 return PyLong_FromLong((long)retval);
3306}
3307#endif /* HAVE_GETPRIORITY */
3308
3309
3310#ifdef HAVE_SETPRIORITY
3311PyDoc_STRVAR(posix_setpriority__doc__,
3312"setpriority(which, who, prio) -> None\n\n\
3313Set program scheduling priority.");
3314
3315static PyObject *
3316posix_setpriority(PyObject *self, PyObject *args)
3317{
3318 int which, who, prio, retval;
3319
3320 if (!PyArg_ParseTuple(args, "iii", &which, &who, &prio))
3321 return NULL;
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003322 retval = setpriority(which, who, prio);
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003323 if (retval == -1)
3324 return posix_error();
3325 Py_RETURN_NONE;
3326}
3327#endif /* HAVE_SETPRIORITY */
3328
3329
Barry Warsaw53699e91996-12-10 23:23:01 +00003330static PyObject *
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003331internal_rename(PyObject *self, PyObject *args, int is_replace)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003332{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003333#ifdef MS_WINDOWS
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003334 PyObject *src, *dst;
Victor Stinner8c62be82010-05-06 00:08:46 +00003335 BOOL result;
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003336 int flags = is_replace ? MOVEFILE_REPLACE_EXISTING : 0;
3337 if (PyArg_ParseTuple(args,
3338 is_replace ? "UU:replace" : "UU:rename",
3339 &src, &dst))
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003340 {
3341 wchar_t *wsrc, *wdst;
3342
3343 wsrc = PyUnicode_AsUnicode(src);
3344 if (wsrc == NULL)
3345 return NULL;
3346 wdst = PyUnicode_AsUnicode(dst);
3347 if (wdst == NULL)
3348 return NULL;
3349 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003350 result = MoveFileExW(wsrc, wdst, flags);
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003351 Py_END_ALLOW_THREADS
3352 if (!result)
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003353 return win32_error(is_replace ? "replace" : "rename", NULL);
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003354 Py_INCREF(Py_None);
3355 return Py_None;
Victor Stinner8c62be82010-05-06 00:08:46 +00003356 }
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003357 else {
3358 PyErr_Clear();
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003359 if (!PyArg_ParseTuple(args,
3360 is_replace ? "O&O&:replace" : "O&O&:rename",
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003361 PyUnicode_FSConverter, &src,
3362 PyUnicode_FSConverter, &dst))
3363 return NULL;
3364
3365 if (win32_warn_bytes_api())
3366 goto error;
3367
3368 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003369 result = MoveFileExA(PyBytes_AS_STRING(src),
3370 PyBytes_AS_STRING(dst), flags);
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003371 Py_END_ALLOW_THREADS
3372
3373 Py_XDECREF(src);
3374 Py_XDECREF(dst);
3375
3376 if (!result)
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003377 return win32_error(is_replace ? "replace" : "rename", NULL);
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003378 Py_INCREF(Py_None);
3379 return Py_None;
3380
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00003381error:
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003382 Py_XDECREF(src);
3383 Py_XDECREF(dst);
Victor Stinner8c62be82010-05-06 00:08:46 +00003384 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003385 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003386#else
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003387 return posix_2str(args,
3388 is_replace ? "O&O&:replace" : "O&O&:rename", rename);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003389#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003390}
3391
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003392PyDoc_STRVAR(posix_rename__doc__,
3393"rename(old, new)\n\n\
3394Rename a file or directory.");
3395
3396static PyObject *
3397posix_rename(PyObject *self, PyObject *args)
3398{
3399 return internal_rename(self, args, 0);
3400}
3401
3402PyDoc_STRVAR(posix_replace__doc__,
3403"replace(old, new)\n\n\
3404Rename a file or directory, overwriting the destination.");
3405
3406static PyObject *
3407posix_replace(PyObject *self, PyObject *args)
3408{
3409 return internal_rename(self, args, 1);
3410}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003411
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003412PyDoc_STRVAR(posix_rmdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003413"rmdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003414Remove a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003415
Barry Warsaw53699e91996-12-10 23:23:01 +00003416static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003417posix_rmdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003418{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003419#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00003420 return win32_1str(args, "rmdir", "y:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003421#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003422 return posix_1str(args, "O&:rmdir", rmdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003423#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003424}
3425
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003426
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003427PyDoc_STRVAR(posix_stat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01003428"stat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003429Perform a stat system call on the given path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003430
Barry Warsaw53699e91996-12-10 23:23:01 +00003431static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01003432posix_stat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003433{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003434#ifdef MS_WINDOWS
Victor Stinner4195b5c2012-02-08 23:03:19 +01003435 return posix_do_stat(self, args, "O&:stat", STAT, "U:stat", win32_stat_w);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003436#else
Victor Stinner4195b5c2012-02-08 23:03:19 +01003437 return posix_do_stat(self, args, "O&:stat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003438#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003439}
3440
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003441
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003442#ifdef HAVE_SYSTEM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003443PyDoc_STRVAR(posix_system__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003444"system(command) -> exit_status\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003445Execute the command (a string) in a subshell.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003446
Barry Warsaw53699e91996-12-10 23:23:01 +00003447static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003448posix_system(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003449{
Victor Stinner8c62be82010-05-06 00:08:46 +00003450 long sts;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00003451#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00003452 wchar_t *command;
3453 if (!PyArg_ParseTuple(args, "u:system", &command))
3454 return NULL;
Victor Stinnercfa72782010-04-16 11:45:13 +00003455
Victor Stinner8c62be82010-05-06 00:08:46 +00003456 Py_BEGIN_ALLOW_THREADS
3457 sts = _wsystem(command);
3458 Py_END_ALLOW_THREADS
Victor Stinnercfa72782010-04-16 11:45:13 +00003459#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003460 PyObject *command_obj;
3461 char *command;
3462 if (!PyArg_ParseTuple(args, "O&:system",
3463 PyUnicode_FSConverter, &command_obj))
3464 return NULL;
Victor Stinnercfa72782010-04-16 11:45:13 +00003465
Victor Stinner8c62be82010-05-06 00:08:46 +00003466 command = PyBytes_AsString(command_obj);
3467 Py_BEGIN_ALLOW_THREADS
3468 sts = system(command);
3469 Py_END_ALLOW_THREADS
3470 Py_DECREF(command_obj);
Victor Stinnercfa72782010-04-16 11:45:13 +00003471#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003472 return PyLong_FromLong(sts);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003473}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003474#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003475
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003476
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003477PyDoc_STRVAR(posix_umask__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003478"umask(new_mask) -> old_mask\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003479Set the current numeric umask and return the previous umask.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003480
Barry Warsaw53699e91996-12-10 23:23:01 +00003481static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003482posix_umask(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003483{
Victor Stinner8c62be82010-05-06 00:08:46 +00003484 int i;
3485 if (!PyArg_ParseTuple(args, "i:umask", &i))
3486 return NULL;
3487 i = (int)umask(i);
3488 if (i < 0)
3489 return posix_error();
3490 return PyLong_FromLong((long)i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003491}
3492
Brian Curtind40e6f72010-07-08 21:39:08 +00003493#ifdef MS_WINDOWS
3494
3495/* override the default DeleteFileW behavior so that directory
3496symlinks can be removed with this function, the same as with
3497Unix symlinks */
3498BOOL WINAPI Py_DeleteFileW(LPCWSTR lpFileName)
3499{
3500 WIN32_FILE_ATTRIBUTE_DATA info;
3501 WIN32_FIND_DATAW find_data;
3502 HANDLE find_data_handle;
3503 int is_directory = 0;
3504 int is_link = 0;
3505
3506 if (GetFileAttributesExW(lpFileName, GetFileExInfoStandard, &info)) {
3507 is_directory = info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00003508
Brian Curtind40e6f72010-07-08 21:39:08 +00003509 /* Get WIN32_FIND_DATA structure for the path to determine if
3510 it is a symlink */
3511 if(is_directory &&
3512 info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
3513 find_data_handle = FindFirstFileW(lpFileName, &find_data);
3514
3515 if(find_data_handle != INVALID_HANDLE_VALUE) {
3516 is_link = find_data.dwReserved0 == IO_REPARSE_TAG_SYMLINK;
3517 FindClose(find_data_handle);
3518 }
3519 }
3520 }
3521
3522 if (is_directory && is_link)
3523 return RemoveDirectoryW(lpFileName);
3524
3525 return DeleteFileW(lpFileName);
3526}
3527#endif /* MS_WINDOWS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003528
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003529PyDoc_STRVAR(posix_unlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003530"unlink(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003531Remove a file (same as remove(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003532
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003533PyDoc_STRVAR(posix_remove__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003534"remove(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003535Remove a file (same as unlink(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003536
Barry Warsaw53699e91996-12-10 23:23:01 +00003537static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003538posix_unlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003539{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003540#ifdef MS_WINDOWS
Brian Curtin74e45612010-07-09 15:58:59 +00003541 return win32_1str(args, "remove", "y:remove", DeleteFileA,
3542 "U:remove", Py_DeleteFileW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003543#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003544 return posix_1str(args, "O&:remove", unlink);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003545#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003546}
3547
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003548
Guido van Rossumb6775db1994-08-01 11:34:53 +00003549#ifdef HAVE_UNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003550PyDoc_STRVAR(posix_uname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003551"uname() -> (sysname, nodename, release, version, machine)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003552Return a tuple identifying the current operating system.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003553
Barry Warsaw53699e91996-12-10 23:23:01 +00003554static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003555posix_uname(PyObject *self, PyObject *noargs)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00003556{
Victor Stinner8c62be82010-05-06 00:08:46 +00003557 struct utsname u;
3558 int res;
Neal Norwitze241ce82003-02-17 18:17:05 +00003559
Victor Stinner8c62be82010-05-06 00:08:46 +00003560 Py_BEGIN_ALLOW_THREADS
3561 res = uname(&u);
3562 Py_END_ALLOW_THREADS
3563 if (res < 0)
3564 return posix_error();
3565 return Py_BuildValue("(sssss)",
3566 u.sysname,
3567 u.nodename,
3568 u.release,
3569 u.version,
3570 u.machine);
Guido van Rossumc39de5f1992-02-05 11:15:54 +00003571}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003572#endif /* HAVE_UNAME */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003573
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003574
Larry Hastings76ad59b2012-05-03 00:30:07 -07003575static int
3576split_py_long_to_s_and_ns(PyObject *py_long, time_t *s, long *ns)
3577{
3578 int result = 0;
Benjamin Peterson3e2e3682012-05-04 01:14:03 -04003579 PyObject *divmod = NULL;
Larry Hastings76ad59b2012-05-03 00:30:07 -07003580 divmod = PyNumber_Divmod(py_long, billion);
3581 if (!divmod)
3582 goto exit;
3583 *s = _PyLong_AsTime_t(PyTuple_GET_ITEM(divmod, 0));
3584 if ((*s == -1) && PyErr_Occurred())
3585 goto exit;
3586 *ns = PyLong_AsLong(PyTuple_GET_ITEM(divmod, 1));
Benjamin Peterson35a8f0d2012-05-04 01:10:59 -04003587 if ((*ns == -1) && PyErr_Occurred())
Larry Hastings76ad59b2012-05-03 00:30:07 -07003588 goto exit;
3589
3590 result = 1;
3591exit:
3592 Py_XDECREF(divmod);
3593 return result;
3594}
3595
3596
3597typedef int (*parameter_converter_t)(PyObject *, void *);
3598
3599typedef struct {
3600 /* input only */
3601 char path_format;
3602 parameter_converter_t converter;
3603 char *function_name;
3604 char *first_argument_name;
3605 PyObject *args;
3606 PyObject *kwargs;
3607
3608 /* input/output */
3609 PyObject **path;
3610
3611 /* output only */
3612 int now;
3613 time_t atime_s;
3614 long atime_ns;
3615 time_t mtime_s;
3616 long mtime_ns;
3617} utime_arguments;
3618
3619#define DECLARE_UA(ua, fname) \
3620 utime_arguments ua; \
3621 memset(&ua, 0, sizeof(ua)); \
3622 ua.function_name = fname; \
3623 ua.args = args; \
3624 ua.kwargs = kwargs; \
3625 ua.first_argument_name = "path"; \
3626
3627/* UA_TO_FILETIME doesn't declare atime and mtime for you */
3628#define UA_TO_FILETIME(ua, atime, mtime) \
3629 time_t_to_FILE_TIME(ua.atime_s, ua.atime_ns, &atime); \
3630 time_t_to_FILE_TIME(ua.mtime_s, ua.mtime_ns, &mtime)
3631
3632/* the rest of these macros declare the output variable for you */
3633#define UA_TO_TIMESPEC(ua, ts) \
3634 struct timespec ts[2]; \
3635 ts[0].tv_sec = ua.atime_s; \
3636 ts[0].tv_nsec = ua.atime_ns; \
3637 ts[1].tv_sec = ua.mtime_s; \
3638 ts[1].tv_nsec = ua.mtime_ns
3639
3640#define UA_TO_TIMEVAL(ua, tv) \
3641 struct timeval tv[2]; \
3642 tv[0].tv_sec = ua.atime_s; \
3643 tv[0].tv_usec = ua.atime_ns / 1000; \
3644 tv[1].tv_sec = ua.mtime_s; \
3645 tv[1].tv_usec = ua.mtime_ns / 1000
3646
3647#define UA_TO_UTIMBUF(ua, u) \
3648 struct utimbuf u; \
3649 utimbuf.actime = ua.atime_s; \
3650 utimbuf.modtime = ua.mtime_s
3651
3652#define UA_TO_TIME_T(ua, timet) \
3653 time_t timet[2]; \
3654 timet[0] = ua.atime_s; \
3655 timet[1] = ua.mtime_s
3656
3657
3658/*
3659 * utime_read_time_arguments() processes arguments for the utime
3660 * family of functions.
3661 * returns zero on failure.
3662 */
3663static int
3664utime_read_time_arguments(utime_arguments *ua)
3665{
3666 PyObject *times = NULL;
3667 PyObject *ns = NULL;
3668 char format[24];
3669 char *kwlist[4];
3670 char **kw = kwlist;
3671 int return_value;
3672
3673 *kw++ = ua->first_argument_name;
3674 *kw++ = "times";
3675 *kw++ = "ns";
3676 *kw = NULL;
3677
3678 sprintf(format, "%c%s|O$O:%s",
3679 ua->path_format,
3680 ua->converter ? "&" : "",
3681 ua->function_name);
3682
3683 if (ua->converter)
3684 return_value = PyArg_ParseTupleAndKeywords(ua->args, ua->kwargs,
3685 format, kwlist, ua->converter, ua->path, &times, &ns);
3686 else
3687 return_value = PyArg_ParseTupleAndKeywords(ua->args, ua->kwargs,
3688 format, kwlist, ua->path, &times, &ns);
3689
3690 if (!return_value)
3691 return 0;
3692
3693 if (times && ns) {
3694 PyErr_Format(PyExc_RuntimeError,
3695 "%s: you may specify either 'times'"
3696 " or 'ns' but not both",
3697 ua->function_name);
Benjamin Petersonb399ab22012-05-04 01:31:13 -04003698 goto fail;
Larry Hastings76ad59b2012-05-03 00:30:07 -07003699 }
3700
3701 if (times && (times != Py_None)) {
3702 if (!PyTuple_CheckExact(times) || (PyTuple_Size(times) != 2)) {
3703 PyErr_Format(PyExc_TypeError,
3704 "%s: 'time' must be either"
3705 " a valid tuple of two ints or None",
3706 ua->function_name);
Benjamin Petersonb399ab22012-05-04 01:31:13 -04003707 goto fail;
Larry Hastings76ad59b2012-05-03 00:30:07 -07003708 }
3709 ua->now = 0;
Benjamin Petersonb399ab22012-05-04 01:31:13 -04003710 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(times, 0),
3711 &ua->atime_s, &ua->atime_ns) == -1 ||
3712 _PyTime_ObjectToTimespec(PyTuple_GET_ITEM(times, 1),
3713 &ua->mtime_s, &ua->mtime_ns) == -1)
3714 goto fail;
3715 return 1;
Larry Hastings76ad59b2012-05-03 00:30:07 -07003716 }
3717
3718 if (ns) {
3719 if (!PyTuple_CheckExact(ns) || (PyTuple_Size(ns) != 2)) {
3720 PyErr_Format(PyExc_TypeError,
3721 "%s: 'ns' must be a valid tuple of two ints",
3722 ua->function_name);
Benjamin Petersonb399ab22012-05-04 01:31:13 -04003723 goto fail;
Larry Hastings76ad59b2012-05-03 00:30:07 -07003724 }
3725 ua->now = 0;
Benjamin Petersonb399ab22012-05-04 01:31:13 -04003726 if (!split_py_long_to_s_and_ns(PyTuple_GET_ITEM(ns, 0),
3727 &ua->atime_s, &ua->atime_ns) ||
3728 !split_py_long_to_s_and_ns(PyTuple_GET_ITEM(ns, 1),
3729 &ua->mtime_s, &ua->mtime_ns))
3730 goto fail;
3731 return 1;
Larry Hastings76ad59b2012-05-03 00:30:07 -07003732 }
3733
3734 /* either times=None, or neither times nor ns was specified. use "now". */
3735 ua->now = 1;
3736 return 1;
Benjamin Petersonb399ab22012-05-04 01:31:13 -04003737
3738 fail:
3739 if (ua->converter)
3740 Py_DECREF(ua->path);
3741 return 0;
Larry Hastings76ad59b2012-05-03 00:30:07 -07003742}
3743
3744
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003745PyDoc_STRVAR(posix_utime__doc__,
Larry Hastings76ad59b2012-05-03 00:30:07 -07003746"utime(path[, times=(atime, mtime), *, ns=(atime_ns, mtime_ns)])\n\
3747Set the access and modified time of the file.\n\
3748If the second argument ('times') is specified,\n\
3749 the values should be expressed as float seconds since the epoch.\n\
3750If the keyword argument 'ns' is specified,\n\
3751 the values should be expressed as integer nanoseconds since the epoch.\n\
3752If neither the second nor the 'ns' argument is specified,\n\
3753 utime uses the current time.\n\
3754Specifying both 'times' and 'ns' is an error.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003755
Barry Warsaw53699e91996-12-10 23:23:01 +00003756static PyObject *
Larry Hastings76ad59b2012-05-03 00:30:07 -07003757posix_utime(PyObject *self, PyObject *args, PyObject *kwargs)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003758{
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003759#ifdef MS_WINDOWS
Larry Hastings76ad59b2012-05-03 00:30:07 -07003760 PyObject *upath;
Victor Stinner8c62be82010-05-06 00:08:46 +00003761 HANDLE hFile;
Victor Stinner8c62be82010-05-06 00:08:46 +00003762 PyObject *result = NULL;
Larry Hastings76ad59b2012-05-03 00:30:07 -07003763 FILETIME atime, mtime;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003764
Larry Hastings76ad59b2012-05-03 00:30:07 -07003765 DECLARE_UA(ua, "utime");
3766
3767 ua.path_format = 'U';
3768 ua.path = &upath;
3769
3770 if (!utime_read_time_arguments(&ua)) {
3771 wchar_t *wpath = PyUnicode_AsUnicode(upath);
Victor Stinnereb5657a2011-09-30 01:44:27 +02003772 if (wpath == NULL)
3773 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00003774 Py_BEGIN_ALLOW_THREADS
3775 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
3776 NULL, OPEN_EXISTING,
3777 FILE_FLAG_BACKUP_SEMANTICS, NULL);
3778 Py_END_ALLOW_THREADS
3779 if (hFile == INVALID_HANDLE_VALUE)
Larry Hastings76ad59b2012-05-03 00:30:07 -07003780 return win32_error_object("utime", upath);
Victor Stinnereb5657a2011-09-30 01:44:27 +02003781 }
3782 else {
Larry Hastings76ad59b2012-05-03 00:30:07 -07003783 const char *apath;
Victor Stinner8c62be82010-05-06 00:08:46 +00003784 /* Drop the argument parsing error as narrow strings
3785 are also valid. */
3786 PyErr_Clear();
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00003787
Larry Hastings76ad59b2012-05-03 00:30:07 -07003788 ua.path_format = 'y';
3789 ua.path = (PyObject **)&apath;
3790 if (!utime_read_time_arguments(&ua))
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003791 return NULL;
3792 if (win32_warn_bytes_api())
Victor Stinner8c62be82010-05-06 00:08:46 +00003793 return NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003794
Victor Stinner8c62be82010-05-06 00:08:46 +00003795 Py_BEGIN_ALLOW_THREADS
3796 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
3797 NULL, OPEN_EXISTING,
3798 FILE_FLAG_BACKUP_SEMANTICS, NULL);
3799 Py_END_ALLOW_THREADS
3800 if (hFile == INVALID_HANDLE_VALUE) {
3801 win32_error("utime", apath);
Victor Stinner8c62be82010-05-06 00:08:46 +00003802 return NULL;
3803 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003804 }
3805
Larry Hastings76ad59b2012-05-03 00:30:07 -07003806 if (ua.now) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003807 SYSTEMTIME now;
3808 GetSystemTime(&now);
3809 if (!SystemTimeToFileTime(&now, &mtime) ||
3810 !SystemTimeToFileTime(&now, &atime)) {
3811 win32_error("utime", NULL);
3812 goto done;
Stefan Krah0e803b32010-11-26 16:16:47 +00003813 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003814 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003815 else {
Larry Hastings76ad59b2012-05-03 00:30:07 -07003816 UA_TO_FILETIME(ua, atime, mtime);
Victor Stinner8c62be82010-05-06 00:08:46 +00003817 }
3818 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
3819 /* Avoid putting the file name into the error here,
3820 as that may confuse the user into believing that
3821 something is wrong with the file, when it also
3822 could be the time stamp that gives a problem. */
3823 win32_error("utime", NULL);
Antoine Pitroue85da7a2011-01-06 18:25:55 +00003824 goto done;
Victor Stinner8c62be82010-05-06 00:08:46 +00003825 }
3826 Py_INCREF(Py_None);
3827 result = Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003828done:
Victor Stinner8c62be82010-05-06 00:08:46 +00003829 CloseHandle(hFile);
3830 return result;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003831#else /* MS_WINDOWS */
Victor Stinner8c62be82010-05-06 00:08:46 +00003832 PyObject *opath;
3833 char *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00003834 int res;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003835
Larry Hastings76ad59b2012-05-03 00:30:07 -07003836 DECLARE_UA(ua, "utime");
3837
3838 ua.path_format = 'O';
3839 ua.path = &opath;
3840 ua.converter = PyUnicode_FSConverter;
3841
3842 if (!utime_read_time_arguments(&ua))
Victor Stinner8c62be82010-05-06 00:08:46 +00003843 return NULL;
3844 path = PyBytes_AsString(opath);
Larry Hastings76ad59b2012-05-03 00:30:07 -07003845 if (ua.now) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003846 Py_BEGIN_ALLOW_THREADS
3847 res = utime(path, NULL);
3848 Py_END_ALLOW_THREADS
3849 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003850 else {
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003851 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003852#ifdef HAVE_UTIMENSAT
Larry Hastings76ad59b2012-05-03 00:30:07 -07003853 UA_TO_TIMESPEC(ua, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003854 res = utimensat(AT_FDCWD, path, buf, 0);
3855#elif defined(HAVE_UTIMES)
Larry Hastings76ad59b2012-05-03 00:30:07 -07003856 UA_TO_TIMEVAL(ua, buf);
Victor Stinner8c62be82010-05-06 00:08:46 +00003857 res = utimes(path, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003858#elif defined(HAVE_UTIME_H)
3859 /* XXX should define struct utimbuf instead, above */
Larry Hastings76ad59b2012-05-03 00:30:07 -07003860 UA_TO_UTIMBUF(ua, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003861 res = utime(path, &buf);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003862#else
Larry Hastings76ad59b2012-05-03 00:30:07 -07003863 UA_TO_TIME_T(ua, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003864 res = utime(path, buf);
3865#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003866 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00003867 }
Larry Hastings76ad59b2012-05-03 00:30:07 -07003868
Victor Stinner8c62be82010-05-06 00:08:46 +00003869 if (res < 0) {
3870 return posix_error_with_allocated_filename(opath);
3871 }
3872 Py_DECREF(opath);
Larry Hastings76ad59b2012-05-03 00:30:07 -07003873 Py_RETURN_NONE;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003874#undef UTIME_EXTRACT
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003875#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003876}
3877
Ross Lagerwall7807c352011-03-17 20:20:30 +02003878#ifdef HAVE_FUTIMES
3879PyDoc_STRVAR(posix_futimes__doc__,
Larry Hastings76ad59b2012-05-03 00:30:07 -07003880"futimes(fd[, times=(atime, mtime), *, ns=(atime_ns, mtime_ns)])\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003881Set the access and modified time of the file specified by the file\n\
Larry Hastings76ad59b2012-05-03 00:30:07 -07003882descriptor fd. See utime for the semantics of the times and ns parameters.");
Ross Lagerwall7807c352011-03-17 20:20:30 +02003883
3884static PyObject *
Larry Hastings76ad59b2012-05-03 00:30:07 -07003885posix_futimes(PyObject *self, PyObject *args, PyObject *kwargs)
Ross Lagerwall7807c352011-03-17 20:20:30 +02003886{
3887 int res, fd;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003888
Larry Hastings76ad59b2012-05-03 00:30:07 -07003889 DECLARE_UA(ua, "futimes");
3890
3891 ua.path_format = 'i';
3892 ua.path = (PyObject **)&fd;
3893 ua.first_argument_name = "fd";
3894
3895 if (!utime_read_time_arguments(&ua))
Ross Lagerwall7807c352011-03-17 20:20:30 +02003896 return NULL;
3897
Larry Hastings76ad59b2012-05-03 00:30:07 -07003898 if (ua.now) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003899 Py_BEGIN_ALLOW_THREADS
3900 res = futimes(fd, NULL);
3901 Py_END_ALLOW_THREADS
3902 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003903 else {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003904 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003905 {
3906#ifdef HAVE_FUTIMENS
Larry Hastings76ad59b2012-05-03 00:30:07 -07003907 UA_TO_TIMESPEC(ua, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003908 res = futimens(fd, buf);
3909#else
Larry Hastings76ad59b2012-05-03 00:30:07 -07003910 UA_TO_TIMEVAL(ua, buf);
Ross Lagerwall7807c352011-03-17 20:20:30 +02003911 res = futimes(fd, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003912#endif
3913 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003914 Py_END_ALLOW_THREADS
3915 }
3916 if (res < 0)
3917 return posix_error();
3918 Py_RETURN_NONE;
3919}
3920#endif
3921
3922#ifdef HAVE_LUTIMES
3923PyDoc_STRVAR(posix_lutimes__doc__,
Larry Hastings76ad59b2012-05-03 00:30:07 -07003924"lutimes(path[, times=(atime, mtime), *, ns=(atime_ns, mtime_ns)])\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003925Like utime(), but if path is a symbolic link, it is not dereferenced.");
3926
3927static PyObject *
Larry Hastings76ad59b2012-05-03 00:30:07 -07003928posix_lutimes(PyObject *self, PyObject *args, PyObject *kwargs)
Ross Lagerwall7807c352011-03-17 20:20:30 +02003929{
Brian Curtinc1b65d12011-11-07 14:18:54 -06003930 PyObject *opath;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003931 const char *path;
3932 int res;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003933
Larry Hastings76ad59b2012-05-03 00:30:07 -07003934 DECLARE_UA(ua, "lutimes");
3935
3936 ua.path_format = 'O';
3937 ua.path = &opath;
3938 ua.converter = PyUnicode_FSConverter;
3939
3940 if (!utime_read_time_arguments(&ua))
Ross Lagerwall7807c352011-03-17 20:20:30 +02003941 return NULL;
3942 path = PyBytes_AsString(opath);
Larry Hastings76ad59b2012-05-03 00:30:07 -07003943
3944 if (ua.now) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003945 /* optional time values not given */
3946 Py_BEGIN_ALLOW_THREADS
3947 res = lutimes(path, NULL);
3948 Py_END_ALLOW_THREADS
3949 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003950 else {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003951 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003952 {
3953#ifdef HAVE_UTIMENSAT
Larry Hastings76ad59b2012-05-03 00:30:07 -07003954 UA_TO_TIMESPEC(ua, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003955 res = utimensat(AT_FDCWD, path, buf, AT_SYMLINK_NOFOLLOW);
3956#else
Larry Hastings76ad59b2012-05-03 00:30:07 -07003957 UA_TO_TIMEVAL(ua, buf);
Ross Lagerwall7807c352011-03-17 20:20:30 +02003958 res = lutimes(path, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003959#endif
3960 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003961 Py_END_ALLOW_THREADS
3962 }
3963 Py_DECREF(opath);
3964 if (res < 0)
3965 return posix_error();
3966 Py_RETURN_NONE;
3967}
3968#endif
3969
Guido van Rossum3b066191991-06-04 19:40:25 +00003970/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003971
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003972PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003973"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003974Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003975
Barry Warsaw53699e91996-12-10 23:23:01 +00003976static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003977posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003978{
Victor Stinner8c62be82010-05-06 00:08:46 +00003979 int sts;
3980 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
3981 return NULL;
3982 _exit(sts);
3983 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003984}
3985
Martin v. Löwis114619e2002-10-07 06:44:21 +00003986#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
3987static void
Martin v. Löwis725507b2006-03-07 12:08:51 +00003988free_string_array(char **array, Py_ssize_t count)
Martin v. Löwis114619e2002-10-07 06:44:21 +00003989{
Victor Stinner8c62be82010-05-06 00:08:46 +00003990 Py_ssize_t i;
3991 for (i = 0; i < count; i++)
3992 PyMem_Free(array[i]);
3993 PyMem_DEL(array);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003994}
Martin v. Löwis011e8422009-05-05 04:43:17 +00003995
Antoine Pitrou69f71142009-05-24 21:25:49 +00003996static
Martin v. Löwis011e8422009-05-05 04:43:17 +00003997int fsconvert_strdup(PyObject *o, char**out)
3998{
Victor Stinner8c62be82010-05-06 00:08:46 +00003999 PyObject *bytes;
4000 Py_ssize_t size;
4001 if (!PyUnicode_FSConverter(o, &bytes))
4002 return 0;
4003 size = PyBytes_GET_SIZE(bytes);
4004 *out = PyMem_Malloc(size+1);
4005 if (!*out)
4006 return 0;
4007 memcpy(*out, PyBytes_AsString(bytes), size+1);
4008 Py_DECREF(bytes);
4009 return 1;
Martin v. Löwis011e8422009-05-05 04:43:17 +00004010}
Martin v. Löwis114619e2002-10-07 06:44:21 +00004011#endif
4012
Ross Lagerwall7807c352011-03-17 20:20:30 +02004013#if defined(HAVE_EXECV) || defined (HAVE_FEXECVE)
Victor Stinner13bb71c2010-04-23 21:41:56 +00004014static char**
4015parse_envlist(PyObject* env, Py_ssize_t *envc_ptr)
4016{
Victor Stinner8c62be82010-05-06 00:08:46 +00004017 char **envlist;
4018 Py_ssize_t i, pos, envc;
4019 PyObject *keys=NULL, *vals=NULL;
4020 PyObject *key, *val, *key2, *val2;
4021 char *p, *k, *v;
4022 size_t len;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004023
Victor Stinner8c62be82010-05-06 00:08:46 +00004024 i = PyMapping_Size(env);
4025 if (i < 0)
4026 return NULL;
4027 envlist = PyMem_NEW(char *, i + 1);
4028 if (envlist == NULL) {
4029 PyErr_NoMemory();
4030 return NULL;
4031 }
4032 envc = 0;
4033 keys = PyMapping_Keys(env);
4034 vals = PyMapping_Values(env);
4035 if (!keys || !vals)
4036 goto error;
4037 if (!PyList_Check(keys) || !PyList_Check(vals)) {
4038 PyErr_Format(PyExc_TypeError,
4039 "env.keys() or env.values() is not a list");
4040 goto error;
4041 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00004042
Victor Stinner8c62be82010-05-06 00:08:46 +00004043 for (pos = 0; pos < i; pos++) {
4044 key = PyList_GetItem(keys, pos);
4045 val = PyList_GetItem(vals, pos);
4046 if (!key || !val)
4047 goto error;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004048
Victor Stinner8c62be82010-05-06 00:08:46 +00004049 if (PyUnicode_FSConverter(key, &key2) == 0)
4050 goto error;
4051 if (PyUnicode_FSConverter(val, &val2) == 0) {
4052 Py_DECREF(key2);
4053 goto error;
4054 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00004055
4056#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +00004057 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
4058 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
Victor Stinner13bb71c2010-04-23 21:41:56 +00004059#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004060 k = PyBytes_AsString(key2);
4061 v = PyBytes_AsString(val2);
4062 len = PyBytes_GET_SIZE(key2) + PyBytes_GET_SIZE(val2) + 2;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004063
Victor Stinner8c62be82010-05-06 00:08:46 +00004064 p = PyMem_NEW(char, len);
4065 if (p == NULL) {
4066 PyErr_NoMemory();
4067 Py_DECREF(key2);
4068 Py_DECREF(val2);
4069 goto error;
4070 }
4071 PyOS_snprintf(p, len, "%s=%s", k, v);
4072 envlist[envc++] = p;
4073 Py_DECREF(key2);
4074 Py_DECREF(val2);
Victor Stinner13bb71c2010-04-23 21:41:56 +00004075#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +00004076 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00004077#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004078 }
4079 Py_DECREF(vals);
4080 Py_DECREF(keys);
Victor Stinner13bb71c2010-04-23 21:41:56 +00004081
Victor Stinner8c62be82010-05-06 00:08:46 +00004082 envlist[envc] = 0;
4083 *envc_ptr = envc;
4084 return envlist;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004085
4086error:
Victor Stinner8c62be82010-05-06 00:08:46 +00004087 Py_XDECREF(keys);
4088 Py_XDECREF(vals);
4089 while (--envc >= 0)
4090 PyMem_DEL(envlist[envc]);
4091 PyMem_DEL(envlist);
4092 return NULL;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004093}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004094
Ross Lagerwall7807c352011-03-17 20:20:30 +02004095static char**
4096parse_arglist(PyObject* argv, Py_ssize_t *argc)
4097{
4098 int i;
4099 char **argvlist = PyMem_NEW(char *, *argc+1);
4100 if (argvlist == NULL) {
4101 PyErr_NoMemory();
4102 return NULL;
4103 }
4104 for (i = 0; i < *argc; i++) {
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02004105 PyObject* item = PySequence_ITEM(argv, i);
4106 if (item == NULL)
4107 goto fail;
4108 if (!fsconvert_strdup(item, &argvlist[i])) {
4109 Py_DECREF(item);
Ross Lagerwall7807c352011-03-17 20:20:30 +02004110 goto fail;
4111 }
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02004112 Py_DECREF(item);
Ross Lagerwall7807c352011-03-17 20:20:30 +02004113 }
4114 argvlist[*argc] = NULL;
4115 return argvlist;
4116fail:
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02004117 *argc = i;
Ross Lagerwall7807c352011-03-17 20:20:30 +02004118 free_string_array(argvlist, *argc);
4119 return NULL;
4120}
4121#endif
4122
4123#ifdef HAVE_EXECV
4124PyDoc_STRVAR(posix_execv__doc__,
4125"execv(path, args)\n\n\
4126Execute an executable path with arguments, replacing current process.\n\
4127\n\
4128 path: path of executable file\n\
4129 args: tuple or list of strings");
4130
4131static PyObject *
4132posix_execv(PyObject *self, PyObject *args)
4133{
4134 PyObject *opath;
4135 char *path;
4136 PyObject *argv;
4137 char **argvlist;
4138 Py_ssize_t argc;
4139
4140 /* execv has two arguments: (path, argv), where
4141 argv is a list or tuple of strings. */
4142
4143 if (!PyArg_ParseTuple(args, "O&O:execv",
4144 PyUnicode_FSConverter,
4145 &opath, &argv))
4146 return NULL;
4147 path = PyBytes_AsString(opath);
4148 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
4149 PyErr_SetString(PyExc_TypeError,
4150 "execv() arg 2 must be a tuple or list");
4151 Py_DECREF(opath);
4152 return NULL;
4153 }
4154 argc = PySequence_Size(argv);
4155 if (argc < 1) {
4156 PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
4157 Py_DECREF(opath);
4158 return NULL;
4159 }
4160
4161 argvlist = parse_arglist(argv, &argc);
4162 if (argvlist == NULL) {
4163 Py_DECREF(opath);
4164 return NULL;
4165 }
4166
4167 execv(path, argvlist);
4168
4169 /* If we get here it's definitely an error */
4170
4171 free_string_array(argvlist, argc);
4172 Py_DECREF(opath);
4173 return posix_error();
4174}
4175
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004176PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004177"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004178Execute a path with arguments and environment, replacing current process.\n\
4179\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004180 path: path of executable file\n\
4181 args: tuple or list of arguments\n\
4182 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004183
Barry Warsaw53699e91996-12-10 23:23:01 +00004184static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004185posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004186{
Victor Stinner8c62be82010-05-06 00:08:46 +00004187 PyObject *opath;
4188 char *path;
4189 PyObject *argv, *env;
4190 char **argvlist;
4191 char **envlist;
Ross Lagerwall7807c352011-03-17 20:20:30 +02004192 Py_ssize_t argc, envc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004193
Victor Stinner8c62be82010-05-06 00:08:46 +00004194 /* execve has three arguments: (path, argv, env), where
4195 argv is a list or tuple of strings and env is a dictionary
4196 like posix.environ. */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004197
Victor Stinner8c62be82010-05-06 00:08:46 +00004198 if (!PyArg_ParseTuple(args, "O&OO:execve",
4199 PyUnicode_FSConverter,
4200 &opath, &argv, &env))
4201 return NULL;
4202 path = PyBytes_AsString(opath);
Ross Lagerwall7807c352011-03-17 20:20:30 +02004203 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00004204 PyErr_SetString(PyExc_TypeError,
4205 "execve() arg 2 must be a tuple or list");
4206 goto fail_0;
4207 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02004208 argc = PySequence_Size(argv);
Victor Stinner8c62be82010-05-06 00:08:46 +00004209 if (!PyMapping_Check(env)) {
4210 PyErr_SetString(PyExc_TypeError,
4211 "execve() arg 3 must be a mapping object");
4212 goto fail_0;
4213 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004214
Ross Lagerwall7807c352011-03-17 20:20:30 +02004215 argvlist = parse_arglist(argv, &argc);
Victor Stinner8c62be82010-05-06 00:08:46 +00004216 if (argvlist == NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00004217 goto fail_0;
4218 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004219
Victor Stinner8c62be82010-05-06 00:08:46 +00004220 envlist = parse_envlist(env, &envc);
4221 if (envlist == NULL)
4222 goto fail_1;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004223
Victor Stinner8c62be82010-05-06 00:08:46 +00004224 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00004225
Victor Stinner8c62be82010-05-06 00:08:46 +00004226 /* If we get here it's definitely an error */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004227
Victor Stinner8c62be82010-05-06 00:08:46 +00004228 (void) posix_error();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004229
Victor Stinner8c62be82010-05-06 00:08:46 +00004230 while (--envc >= 0)
4231 PyMem_DEL(envlist[envc]);
4232 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004233 fail_1:
Ross Lagerwall7807c352011-03-17 20:20:30 +02004234 free_string_array(argvlist, argc);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004235 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004236 Py_DECREF(opath);
4237 return NULL;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004238}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00004239#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004240
Ross Lagerwall7807c352011-03-17 20:20:30 +02004241#ifdef HAVE_FEXECVE
4242PyDoc_STRVAR(posix_fexecve__doc__,
4243"fexecve(fd, args, env)\n\n\
4244Execute the program specified by a file descriptor with arguments and\n\
4245environment, replacing the current process.\n\
4246\n\
4247 fd: file descriptor of executable\n\
4248 args: tuple or list of arguments\n\
4249 env: dictionary of strings mapping to strings");
4250
4251static PyObject *
4252posix_fexecve(PyObject *self, PyObject *args)
4253{
4254 int fd;
4255 PyObject *argv, *env;
4256 char **argvlist;
4257 char **envlist;
4258 Py_ssize_t argc, envc;
4259
4260 if (!PyArg_ParseTuple(args, "iOO:fexecve",
4261 &fd, &argv, &env))
4262 return NULL;
4263 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
4264 PyErr_SetString(PyExc_TypeError,
4265 "fexecve() arg 2 must be a tuple or list");
4266 return NULL;
4267 }
4268 argc = PySequence_Size(argv);
4269 if (!PyMapping_Check(env)) {
4270 PyErr_SetString(PyExc_TypeError,
4271 "fexecve() arg 3 must be a mapping object");
4272 return NULL;
4273 }
4274
4275 argvlist = parse_arglist(argv, &argc);
4276 if (argvlist == NULL)
4277 return NULL;
4278
4279 envlist = parse_envlist(env, &envc);
4280 if (envlist == NULL)
4281 goto fail;
4282
4283 fexecve(fd, argvlist, envlist);
4284
4285 /* If we get here it's definitely an error */
4286
4287 (void) posix_error();
4288
4289 while (--envc >= 0)
4290 PyMem_DEL(envlist[envc]);
4291 PyMem_DEL(envlist);
4292 fail:
4293 free_string_array(argvlist, argc);
4294 return NULL;
4295}
4296#endif /* HAVE_FEXECVE */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004297
Guido van Rossuma1065681999-01-25 23:20:23 +00004298#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004299PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004300"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00004301Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00004302\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004303 mode: mode of process creation\n\
4304 path: path of executable file\n\
4305 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00004306
4307static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004308posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00004309{
Victor Stinner8c62be82010-05-06 00:08:46 +00004310 PyObject *opath;
4311 char *path;
4312 PyObject *argv;
4313 char **argvlist;
4314 int mode, i;
4315 Py_ssize_t argc;
4316 Py_intptr_t spawnval;
4317 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossuma1065681999-01-25 23:20:23 +00004318
Victor Stinner8c62be82010-05-06 00:08:46 +00004319 /* spawnv has three arguments: (mode, path, argv), where
4320 argv is a list or tuple of strings. */
Guido van Rossuma1065681999-01-25 23:20:23 +00004321
Victor Stinner8c62be82010-05-06 00:08:46 +00004322 if (!PyArg_ParseTuple(args, "iO&O:spawnv", &mode,
4323 PyUnicode_FSConverter,
4324 &opath, &argv))
4325 return NULL;
4326 path = PyBytes_AsString(opath);
4327 if (PyList_Check(argv)) {
4328 argc = PyList_Size(argv);
4329 getitem = PyList_GetItem;
4330 }
4331 else if (PyTuple_Check(argv)) {
4332 argc = PyTuple_Size(argv);
4333 getitem = PyTuple_GetItem;
4334 }
4335 else {
4336 PyErr_SetString(PyExc_TypeError,
4337 "spawnv() arg 2 must be a tuple or list");
4338 Py_DECREF(opath);
4339 return NULL;
4340 }
Guido van Rossuma1065681999-01-25 23:20:23 +00004341
Victor Stinner8c62be82010-05-06 00:08:46 +00004342 argvlist = PyMem_NEW(char *, argc+1);
4343 if (argvlist == NULL) {
4344 Py_DECREF(opath);
4345 return PyErr_NoMemory();
4346 }
4347 for (i = 0; i < argc; i++) {
4348 if (!fsconvert_strdup((*getitem)(argv, i),
4349 &argvlist[i])) {
4350 free_string_array(argvlist, i);
4351 PyErr_SetString(
4352 PyExc_TypeError,
4353 "spawnv() arg 2 must contain only strings");
4354 Py_DECREF(opath);
4355 return NULL;
4356 }
4357 }
4358 argvlist[argc] = NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00004359
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004360#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004361 Py_BEGIN_ALLOW_THREADS
4362 spawnval = spawnv(mode, path, argvlist);
4363 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004364#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004365 if (mode == _OLD_P_OVERLAY)
4366 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00004367
Victor Stinner8c62be82010-05-06 00:08:46 +00004368 Py_BEGIN_ALLOW_THREADS
4369 spawnval = _spawnv(mode, path, argvlist);
4370 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004371#endif
Tim Peters5aa91602002-01-30 05:46:57 +00004372
Victor Stinner8c62be82010-05-06 00:08:46 +00004373 free_string_array(argvlist, argc);
4374 Py_DECREF(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00004375
Victor Stinner8c62be82010-05-06 00:08:46 +00004376 if (spawnval == -1)
4377 return posix_error();
4378 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00004379#if SIZEOF_LONG == SIZEOF_VOID_P
Victor Stinner8c62be82010-05-06 00:08:46 +00004380 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004381#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004382 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004383#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00004384}
4385
4386
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004387PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004388"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00004389Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00004390\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004391 mode: mode of process creation\n\
4392 path: path of executable file\n\
4393 args: tuple or list of arguments\n\
4394 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00004395
4396static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004397posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00004398{
Victor Stinner8c62be82010-05-06 00:08:46 +00004399 PyObject *opath;
4400 char *path;
4401 PyObject *argv, *env;
4402 char **argvlist;
4403 char **envlist;
4404 PyObject *res = NULL;
Antoine Pitrou22e41552010-08-15 18:07:50 +00004405 int mode;
4406 Py_ssize_t argc, i, envc;
Victor Stinner8c62be82010-05-06 00:08:46 +00004407 Py_intptr_t spawnval;
4408 PyObject *(*getitem)(PyObject *, Py_ssize_t);
4409 Py_ssize_t lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00004410
Victor Stinner8c62be82010-05-06 00:08:46 +00004411 /* spawnve has four arguments: (mode, path, argv, env), where
4412 argv is a list or tuple of strings and env is a dictionary
4413 like posix.environ. */
Guido van Rossuma1065681999-01-25 23:20:23 +00004414
Victor Stinner8c62be82010-05-06 00:08:46 +00004415 if (!PyArg_ParseTuple(args, "iO&OO:spawnve", &mode,
4416 PyUnicode_FSConverter,
4417 &opath, &argv, &env))
4418 return NULL;
4419 path = PyBytes_AsString(opath);
4420 if (PyList_Check(argv)) {
4421 argc = PyList_Size(argv);
4422 getitem = PyList_GetItem;
4423 }
4424 else if (PyTuple_Check(argv)) {
4425 argc = PyTuple_Size(argv);
4426 getitem = PyTuple_GetItem;
4427 }
4428 else {
4429 PyErr_SetString(PyExc_TypeError,
4430 "spawnve() arg 2 must be a tuple or list");
4431 goto fail_0;
4432 }
4433 if (!PyMapping_Check(env)) {
4434 PyErr_SetString(PyExc_TypeError,
4435 "spawnve() arg 3 must be a mapping object");
4436 goto fail_0;
4437 }
Guido van Rossuma1065681999-01-25 23:20:23 +00004438
Victor Stinner8c62be82010-05-06 00:08:46 +00004439 argvlist = PyMem_NEW(char *, argc+1);
4440 if (argvlist == NULL) {
4441 PyErr_NoMemory();
4442 goto fail_0;
4443 }
4444 for (i = 0; i < argc; i++) {
4445 if (!fsconvert_strdup((*getitem)(argv, i),
4446 &argvlist[i]))
4447 {
4448 lastarg = i;
4449 goto fail_1;
4450 }
4451 }
4452 lastarg = argc;
4453 argvlist[argc] = NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00004454
Victor Stinner8c62be82010-05-06 00:08:46 +00004455 envlist = parse_envlist(env, &envc);
4456 if (envlist == NULL)
4457 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00004458
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004459#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004460 Py_BEGIN_ALLOW_THREADS
4461 spawnval = spawnve(mode, path, argvlist, envlist);
4462 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004463#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004464 if (mode == _OLD_P_OVERLAY)
4465 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00004466
Victor Stinner8c62be82010-05-06 00:08:46 +00004467 Py_BEGIN_ALLOW_THREADS
4468 spawnval = _spawnve(mode, path, argvlist, envlist);
4469 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004470#endif
Tim Peters25059d32001-12-07 20:35:43 +00004471
Victor Stinner8c62be82010-05-06 00:08:46 +00004472 if (spawnval == -1)
4473 (void) posix_error();
4474 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00004475#if SIZEOF_LONG == SIZEOF_VOID_P
Victor Stinner8c62be82010-05-06 00:08:46 +00004476 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004477#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004478 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004479#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00004480
Victor Stinner8c62be82010-05-06 00:08:46 +00004481 while (--envc >= 0)
4482 PyMem_DEL(envlist[envc]);
4483 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004484 fail_1:
Victor Stinner8c62be82010-05-06 00:08:46 +00004485 free_string_array(argvlist, lastarg);
Martin v. Löwis114619e2002-10-07 06:44:21 +00004486 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004487 Py_DECREF(opath);
4488 return res;
Guido van Rossuma1065681999-01-25 23:20:23 +00004489}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004490
4491/* OS/2 supports spawnvp & spawnvpe natively */
4492#if defined(PYOS_OS2)
4493PyDoc_STRVAR(posix_spawnvp__doc__,
4494"spawnvp(mode, file, args)\n\n\
4495Execute the program 'file' in a new process, using the environment\n\
4496search path to find the file.\n\
4497\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004498 mode: mode of process creation\n\
4499 file: executable file name\n\
4500 args: tuple or list of strings");
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004501
4502static PyObject *
4503posix_spawnvp(PyObject *self, PyObject *args)
4504{
Victor Stinner8c62be82010-05-06 00:08:46 +00004505 PyObject *opath;
4506 char *path;
4507 PyObject *argv;
4508 char **argvlist;
4509 int mode, i, argc;
4510 Py_intptr_t spawnval;
4511 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004512
Victor Stinner8c62be82010-05-06 00:08:46 +00004513 /* spawnvp has three arguments: (mode, path, argv), where
4514 argv is a list or tuple of strings. */
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004515
Victor Stinner8c62be82010-05-06 00:08:46 +00004516 if (!PyArg_ParseTuple(args, "iO&O:spawnvp", &mode,
4517 PyUnicode_FSConverter,
4518 &opath, &argv))
4519 return NULL;
4520 path = PyBytes_AsString(opath);
4521 if (PyList_Check(argv)) {
4522 argc = PyList_Size(argv);
4523 getitem = PyList_GetItem;
4524 }
4525 else if (PyTuple_Check(argv)) {
4526 argc = PyTuple_Size(argv);
4527 getitem = PyTuple_GetItem;
4528 }
4529 else {
4530 PyErr_SetString(PyExc_TypeError,
4531 "spawnvp() arg 2 must be a tuple or list");
4532 Py_DECREF(opath);
4533 return NULL;
4534 }
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004535
Victor Stinner8c62be82010-05-06 00:08:46 +00004536 argvlist = PyMem_NEW(char *, argc+1);
4537 if (argvlist == NULL) {
4538 Py_DECREF(opath);
4539 return PyErr_NoMemory();
4540 }
4541 for (i = 0; i < argc; i++) {
4542 if (!fsconvert_strdup((*getitem)(argv, i),
4543 &argvlist[i])) {
4544 free_string_array(argvlist, i);
4545 PyErr_SetString(
4546 PyExc_TypeError,
4547 "spawnvp() arg 2 must contain only strings");
4548 Py_DECREF(opath);
4549 return NULL;
4550 }
4551 }
4552 argvlist[argc] = NULL;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004553
Victor Stinner8c62be82010-05-06 00:08:46 +00004554 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004555#if defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004556 spawnval = spawnvp(mode, path, argvlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004557#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004558 spawnval = _spawnvp(mode, path, argvlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004559#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004560 Py_END_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004561
Victor Stinner8c62be82010-05-06 00:08:46 +00004562 free_string_array(argvlist, argc);
4563 Py_DECREF(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004564
Victor Stinner8c62be82010-05-06 00:08:46 +00004565 if (spawnval == -1)
4566 return posix_error();
4567 else
4568 return Py_BuildValue("l", (long) spawnval);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004569}
4570
4571
4572PyDoc_STRVAR(posix_spawnvpe__doc__,
4573"spawnvpe(mode, file, args, env)\n\n\
4574Execute the program 'file' in a new process, using the environment\n\
4575search path to find the file.\n\
4576\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004577 mode: mode of process creation\n\
4578 file: executable file name\n\
4579 args: tuple or list of arguments\n\
4580 env: dictionary of strings mapping to strings");
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004581
4582static PyObject *
4583posix_spawnvpe(PyObject *self, PyObject *args)
4584{
Ross Lagerwalldcfde5a2011-11-04 07:09:14 +02004585 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00004586 char *path;
4587 PyObject *argv, *env;
4588 char **argvlist;
4589 char **envlist;
4590 PyObject *res=NULL;
Antoine Pitrou22e41552010-08-15 18:07:50 +00004591 int mode;
4592 Py_ssize_t argc, i, envc;
Victor Stinner8c62be82010-05-06 00:08:46 +00004593 Py_intptr_t spawnval;
4594 PyObject *(*getitem)(PyObject *, Py_ssize_t);
4595 int lastarg = 0;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004596
Victor Stinner8c62be82010-05-06 00:08:46 +00004597 /* spawnvpe has four arguments: (mode, path, argv, env), where
4598 argv is a list or tuple of strings and env is a dictionary
4599 like posix.environ. */
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004600
Victor Stinner8c62be82010-05-06 00:08:46 +00004601 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
4602 PyUnicode_FSConverter,
4603 &opath, &argv, &env))
4604 return NULL;
4605 path = PyBytes_AsString(opath);
4606 if (PyList_Check(argv)) {
4607 argc = PyList_Size(argv);
4608 getitem = PyList_GetItem;
4609 }
4610 else if (PyTuple_Check(argv)) {
4611 argc = PyTuple_Size(argv);
4612 getitem = PyTuple_GetItem;
4613 }
4614 else {
4615 PyErr_SetString(PyExc_TypeError,
4616 "spawnvpe() arg 2 must be a tuple or list");
4617 goto fail_0;
4618 }
4619 if (!PyMapping_Check(env)) {
4620 PyErr_SetString(PyExc_TypeError,
4621 "spawnvpe() arg 3 must be a mapping object");
4622 goto fail_0;
4623 }
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004624
Victor Stinner8c62be82010-05-06 00:08:46 +00004625 argvlist = PyMem_NEW(char *, argc+1);
4626 if (argvlist == NULL) {
4627 PyErr_NoMemory();
4628 goto fail_0;
4629 }
4630 for (i = 0; i < argc; i++) {
4631 if (!fsconvert_strdup((*getitem)(argv, i),
4632 &argvlist[i]))
4633 {
4634 lastarg = i;
4635 goto fail_1;
4636 }
4637 }
4638 lastarg = argc;
4639 argvlist[argc] = NULL;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004640
Victor Stinner8c62be82010-05-06 00:08:46 +00004641 envlist = parse_envlist(env, &envc);
4642 if (envlist == NULL)
4643 goto fail_1;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004644
Victor Stinner8c62be82010-05-06 00:08:46 +00004645 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004646#if defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004647 spawnval = spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004648#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004649 spawnval = _spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004650#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004651 Py_END_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004652
Victor Stinner8c62be82010-05-06 00:08:46 +00004653 if (spawnval == -1)
4654 (void) posix_error();
4655 else
4656 res = Py_BuildValue("l", (long) spawnval);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004657
Victor Stinner8c62be82010-05-06 00:08:46 +00004658 while (--envc >= 0)
4659 PyMem_DEL(envlist[envc]);
4660 PyMem_DEL(envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004661 fail_1:
Victor Stinner8c62be82010-05-06 00:08:46 +00004662 free_string_array(argvlist, lastarg);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004663 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004664 Py_DECREF(opath);
4665 return res;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004666}
4667#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00004668#endif /* HAVE_SPAWNV */
4669
4670
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004671#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004672PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004673"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004674Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
4675\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004676Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004677
4678static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004679posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004680{
Victor Stinner8c62be82010-05-06 00:08:46 +00004681 pid_t pid;
4682 int result = 0;
4683 _PyImport_AcquireLock();
4684 pid = fork1();
4685 if (pid == 0) {
4686 /* child: this clobbers and resets the import lock. */
4687 PyOS_AfterFork();
4688 } else {
4689 /* parent: release the import lock. */
4690 result = _PyImport_ReleaseLock();
4691 }
4692 if (pid == -1)
4693 return posix_error();
4694 if (result < 0) {
4695 /* Don't clobber the OSError if the fork failed. */
4696 PyErr_SetString(PyExc_RuntimeError,
4697 "not holding the import lock");
4698 return NULL;
4699 }
4700 return PyLong_FromPid(pid);
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004701}
4702#endif
4703
4704
Guido van Rossumad0ee831995-03-01 10:34:45 +00004705#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004706PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004707"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004708Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004709Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004710
Barry Warsaw53699e91996-12-10 23:23:01 +00004711static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004712posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004713{
Victor Stinner8c62be82010-05-06 00:08:46 +00004714 pid_t pid;
4715 int result = 0;
4716 _PyImport_AcquireLock();
4717 pid = fork();
4718 if (pid == 0) {
4719 /* child: this clobbers and resets the import lock. */
4720 PyOS_AfterFork();
4721 } else {
4722 /* parent: release the import lock. */
4723 result = _PyImport_ReleaseLock();
4724 }
4725 if (pid == -1)
4726 return posix_error();
4727 if (result < 0) {
4728 /* Don't clobber the OSError if the fork failed. */
4729 PyErr_SetString(PyExc_RuntimeError,
4730 "not holding the import lock");
4731 return NULL;
4732 }
4733 return PyLong_FromPid(pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00004734}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004735#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004736
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004737#ifdef HAVE_SCHED_H
4738
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +02004739#ifdef HAVE_SCHED_GET_PRIORITY_MAX
4740
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004741PyDoc_STRVAR(posix_sched_get_priority_max__doc__,
4742"sched_get_priority_max(policy)\n\n\
4743Get the maximum scheduling priority for *policy*.");
4744
4745static PyObject *
4746posix_sched_get_priority_max(PyObject *self, PyObject *args)
4747{
4748 int policy, max;
4749
4750 if (!PyArg_ParseTuple(args, "i:sched_get_priority_max", &policy))
4751 return NULL;
4752 max = sched_get_priority_max(policy);
4753 if (max < 0)
4754 return posix_error();
4755 return PyLong_FromLong(max);
4756}
4757
4758PyDoc_STRVAR(posix_sched_get_priority_min__doc__,
4759"sched_get_priority_min(policy)\n\n\
4760Get the minimum scheduling priority for *policy*.");
4761
4762static PyObject *
4763posix_sched_get_priority_min(PyObject *self, PyObject *args)
4764{
4765 int policy, min;
4766
4767 if (!PyArg_ParseTuple(args, "i:sched_get_priority_min", &policy))
4768 return NULL;
4769 min = sched_get_priority_min(policy);
4770 if (min < 0)
4771 return posix_error();
4772 return PyLong_FromLong(min);
4773}
4774
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +02004775#endif /* HAVE_SCHED_GET_PRIORITY_MAX */
4776
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004777#ifdef HAVE_SCHED_SETSCHEDULER
4778
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004779PyDoc_STRVAR(posix_sched_getscheduler__doc__,
4780"sched_getscheduler(pid)\n\n\
4781Get the scheduling policy for the process with a PID of *pid*.\n\
4782Passing a PID of 0 returns the scheduling policy for the calling process.");
4783
4784static PyObject *
4785posix_sched_getscheduler(PyObject *self, PyObject *args)
4786{
4787 pid_t pid;
4788 int policy;
4789
4790 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":sched_getscheduler", &pid))
4791 return NULL;
4792 policy = sched_getscheduler(pid);
4793 if (policy < 0)
4794 return posix_error();
4795 return PyLong_FromLong(policy);
4796}
4797
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004798#endif
4799
4800#if defined(HAVE_SCHED_SETSCHEDULER) || defined(HAVE_SCHED_SETPARAM)
4801
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004802static PyObject *
4803sched_param_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
4804{
4805 PyObject *res, *priority;
Benjamin Peterson4e36d5a2011-08-02 19:56:11 -05004806 static char *kwlist[] = {"sched_priority", NULL};
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004807
4808 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O:sched_param", kwlist, &priority))
4809 return NULL;
4810 res = PyStructSequence_New(type);
4811 if (!res)
4812 return NULL;
4813 Py_INCREF(priority);
4814 PyStructSequence_SET_ITEM(res, 0, priority);
4815 return res;
4816}
4817
4818PyDoc_STRVAR(sched_param__doc__,
4819"sched_param(sched_priority): A scheduling parameter.\n\n\
4820Current has only one field: sched_priority");
4821
4822static PyStructSequence_Field sched_param_fields[] = {
4823 {"sched_priority", "the scheduling priority"},
4824 {0}
4825};
4826
4827static PyStructSequence_Desc sched_param_desc = {
4828 "sched_param", /* name */
4829 sched_param__doc__, /* doc */
4830 sched_param_fields,
4831 1
4832};
4833
4834static int
4835convert_sched_param(PyObject *param, struct sched_param *res)
4836{
4837 long priority;
4838
4839 if (Py_TYPE(param) != &SchedParamType) {
4840 PyErr_SetString(PyExc_TypeError, "must have a sched_param object");
4841 return 0;
4842 }
4843 priority = PyLong_AsLong(PyStructSequence_GET_ITEM(param, 0));
4844 if (priority == -1 && PyErr_Occurred())
4845 return 0;
4846 if (priority > INT_MAX || priority < INT_MIN) {
4847 PyErr_SetString(PyExc_OverflowError, "sched_priority out of range");
4848 return 0;
4849 }
4850 res->sched_priority = Py_SAFE_DOWNCAST(priority, long, int);
4851 return 1;
4852}
4853
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004854#endif
4855
4856#ifdef HAVE_SCHED_SETSCHEDULER
4857
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004858PyDoc_STRVAR(posix_sched_setscheduler__doc__,
4859"sched_setscheduler(pid, policy, param)\n\n\
4860Set the scheduling policy, *policy*, for *pid*.\n\
4861If *pid* is 0, the calling process is changed.\n\
4862*param* is an instance of sched_param.");
4863
4864static PyObject *
4865posix_sched_setscheduler(PyObject *self, PyObject *args)
4866{
4867 pid_t pid;
4868 int policy;
4869 struct sched_param param;
4870
4871 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "iO&:sched_setscheduler",
4872 &pid, &policy, &convert_sched_param, &param))
4873 return NULL;
Jesus Cea9c822272011-09-10 01:40:52 +02004874
4875 /*
Jesus Cea54b01492011-09-10 01:53:19 +02004876 ** sched_setscheduler() returns 0 in Linux, but the previous
4877 ** scheduling policy under Solaris/Illumos, and others.
4878 ** On error, -1 is returned in all Operating Systems.
Jesus Cea9c822272011-09-10 01:40:52 +02004879 */
4880 if (sched_setscheduler(pid, policy, &param) == -1)
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004881 return posix_error();
4882 Py_RETURN_NONE;
4883}
4884
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004885#endif
4886
4887#ifdef HAVE_SCHED_SETPARAM
4888
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004889PyDoc_STRVAR(posix_sched_getparam__doc__,
4890"sched_getparam(pid) -> sched_param\n\n\
4891Returns scheduling parameters for the process with *pid* as an instance of the\n\
4892sched_param class. A PID of 0 means the calling process.");
4893
4894static PyObject *
4895posix_sched_getparam(PyObject *self, PyObject *args)
4896{
4897 pid_t pid;
4898 struct sched_param param;
4899 PyObject *res, *priority;
4900
4901 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":sched_getparam", &pid))
4902 return NULL;
4903 if (sched_getparam(pid, &param))
4904 return posix_error();
4905 res = PyStructSequence_New(&SchedParamType);
4906 if (!res)
4907 return NULL;
4908 priority = PyLong_FromLong(param.sched_priority);
4909 if (!priority) {
4910 Py_DECREF(res);
4911 return NULL;
4912 }
4913 PyStructSequence_SET_ITEM(res, 0, priority);
4914 return res;
4915}
4916
4917PyDoc_STRVAR(posix_sched_setparam__doc__,
4918"sched_setparam(pid, param)\n\n\
4919Set scheduling parameters for a process with PID *pid*.\n\
4920A PID of 0 means the calling process.");
4921
4922static PyObject *
4923posix_sched_setparam(PyObject *self, PyObject *args)
4924{
4925 pid_t pid;
4926 struct sched_param param;
4927
4928 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "O&:sched_setparam",
4929 &pid, &convert_sched_param, &param))
4930 return NULL;
4931 if (sched_setparam(pid, &param))
4932 return posix_error();
4933 Py_RETURN_NONE;
4934}
4935
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004936#endif
4937
4938#ifdef HAVE_SCHED_RR_GET_INTERVAL
4939
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004940PyDoc_STRVAR(posix_sched_rr_get_interval__doc__,
4941"sched_rr_get_interval(pid) -> float\n\n\
4942Return the round-robin quantum for the process with PID *pid* in seconds.");
4943
4944static PyObject *
4945posix_sched_rr_get_interval(PyObject *self, PyObject *args)
4946{
4947 pid_t pid;
4948 struct timespec interval;
4949
4950 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":sched_rr_get_interval", &pid))
4951 return NULL;
4952 if (sched_rr_get_interval(pid, &interval))
4953 return posix_error();
4954 return PyFloat_FromDouble((double)interval.tv_sec + 1e-9*interval.tv_nsec);
4955}
4956
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004957#endif
4958
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004959PyDoc_STRVAR(posix_sched_yield__doc__,
4960"sched_yield()\n\n\
4961Voluntarily relinquish the CPU.");
4962
4963static PyObject *
4964posix_sched_yield(PyObject *self, PyObject *noargs)
4965{
4966 if (sched_yield())
4967 return posix_error();
4968 Py_RETURN_NONE;
4969}
4970
Benjamin Peterson2740af82011-08-02 17:41:34 -05004971#ifdef HAVE_SCHED_SETAFFINITY
4972
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004973typedef struct {
4974 PyObject_HEAD;
4975 Py_ssize_t size;
4976 int ncpus;
4977 cpu_set_t *set;
4978} Py_cpu_set;
4979
4980static PyTypeObject cpu_set_type;
4981
4982static void
4983cpu_set_dealloc(Py_cpu_set *set)
4984{
4985 assert(set->set);
4986 CPU_FREE(set->set);
4987 Py_TYPE(set)->tp_free(set);
4988}
4989
4990static Py_cpu_set *
4991make_new_cpu_set(PyTypeObject *type, Py_ssize_t size)
4992{
4993 Py_cpu_set *set;
4994
4995 if (size < 0) {
4996 PyErr_SetString(PyExc_ValueError, "negative size");
4997 return NULL;
4998 }
4999 set = (Py_cpu_set *)type->tp_alloc(type, 0);
5000 if (!set)
5001 return NULL;
5002 set->ncpus = size;
5003 set->size = CPU_ALLOC_SIZE(size);
5004 set->set = CPU_ALLOC(size);
5005 if (!set->set) {
5006 type->tp_free(set);
5007 PyErr_NoMemory();
5008 return NULL;
5009 }
5010 CPU_ZERO_S(set->size, set->set);
5011 return set;
5012}
5013
5014static PyObject *
5015cpu_set_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
5016{
5017 int size;
5018
5019 if (!_PyArg_NoKeywords("cpu_set()", kwargs) ||
5020 !PyArg_ParseTuple(args, "i:cpu_set", &size))
5021 return NULL;
5022 return (PyObject *)make_new_cpu_set(type, size);
5023}
5024
5025static PyObject *
5026cpu_set_repr(Py_cpu_set *set)
5027{
5028 return PyUnicode_FromFormat("<cpu_set with %li entries>", set->ncpus);
Victor Stinner63941882011-09-29 00:42:28 +02005029}
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005030
5031static Py_ssize_t
5032cpu_set_len(Py_cpu_set *set)
5033{
5034 return set->ncpus;
5035}
5036
5037static int
5038_get_cpu(Py_cpu_set *set, const char *requester, PyObject *args)
5039{
5040 int cpu;
5041 if (!PyArg_ParseTuple(args, requester, &cpu))
5042 return -1;
5043 if (cpu < 0) {
5044 PyErr_SetString(PyExc_ValueError, "cpu < 0 not valid");
5045 return -1;
5046 }
5047 if (cpu >= set->ncpus) {
5048 PyErr_SetString(PyExc_ValueError, "cpu too large for set");
5049 return -1;
5050 }
5051 return cpu;
5052}
5053
5054PyDoc_STRVAR(cpu_set_set_doc,
5055"cpu_set.set(i)\n\n\
5056Add CPU *i* to the set.");
5057
5058static PyObject *
5059cpu_set_set(Py_cpu_set *set, PyObject *args)
5060{
5061 int cpu = _get_cpu(set, "i|set", args);
5062 if (cpu == -1)
5063 return NULL;
5064 CPU_SET_S(cpu, set->size, set->set);
5065 Py_RETURN_NONE;
5066}
5067
5068PyDoc_STRVAR(cpu_set_count_doc,
5069"cpu_set.count() -> int\n\n\
5070Return the number of CPUs active in the set.");
5071
5072static PyObject *
5073cpu_set_count(Py_cpu_set *set, PyObject *noargs)
5074{
5075 return PyLong_FromLong(CPU_COUNT_S(set->size, set->set));
5076}
5077
5078PyDoc_STRVAR(cpu_set_clear_doc,
5079"cpu_set.clear(i)\n\n\
5080Remove CPU *i* from the set.");
5081
5082static PyObject *
5083cpu_set_clear(Py_cpu_set *set, PyObject *args)
5084{
5085 int cpu = _get_cpu(set, "i|clear", args);
5086 if (cpu == -1)
5087 return NULL;
5088 CPU_CLR_S(cpu, set->size, set->set);
5089 Py_RETURN_NONE;
5090}
5091
5092PyDoc_STRVAR(cpu_set_isset_doc,
5093"cpu_set.isset(i) -> bool\n\n\
5094Test if CPU *i* is in the set.");
5095
5096static PyObject *
5097cpu_set_isset(Py_cpu_set *set, PyObject *args)
5098{
5099 int cpu = _get_cpu(set, "i|isset", args);
5100 if (cpu == -1)
5101 return NULL;
5102 if (CPU_ISSET_S(cpu, set->size, set->set))
5103 Py_RETURN_TRUE;
5104 Py_RETURN_FALSE;
5105}
5106
5107PyDoc_STRVAR(cpu_set_zero_doc,
5108"cpu_set.zero()\n\n\
5109Clear the cpu_set.");
5110
5111static PyObject *
5112cpu_set_zero(Py_cpu_set *set, PyObject *noargs)
5113{
5114 CPU_ZERO_S(set->size, set->set);
5115 Py_RETURN_NONE;
5116}
5117
5118static PyObject *
5119cpu_set_richcompare(Py_cpu_set *set, Py_cpu_set *other, int op)
5120{
5121 int eq;
5122
Brian Curtindfc80e32011-08-10 20:28:54 -05005123 if ((op != Py_EQ && op != Py_NE) || Py_TYPE(other) != &cpu_set_type)
5124 Py_RETURN_NOTIMPLEMENTED;
5125
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005126 eq = set->ncpus == other->ncpus && CPU_EQUAL_S(set->size, set->set, other->set);
5127 if ((op == Py_EQ) ? eq : !eq)
5128 Py_RETURN_TRUE;
5129 else
5130 Py_RETURN_FALSE;
5131}
5132
5133#define CPU_SET_BINOP(name, op) \
5134 static PyObject * \
5135 do_cpu_set_##name(Py_cpu_set *left, Py_cpu_set *right, Py_cpu_set *res) { \
5136 if (res) { \
5137 Py_INCREF(res); \
5138 } \
5139 else { \
Benjamin Petersone870fe62011-08-02 17:44:26 -05005140 res = make_new_cpu_set(&cpu_set_type, left->ncpus); \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005141 if (!res) \
5142 return NULL; \
5143 } \
Benjamin Peterson9b374bf2011-08-02 18:22:30 -05005144 if (Py_TYPE(right) != &cpu_set_type || left->ncpus != right->ncpus) { \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005145 Py_DECREF(res); \
Brian Curtindfc80e32011-08-10 20:28:54 -05005146 Py_RETURN_NOTIMPLEMENTED; \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005147 } \
Benjamin Peterson8f7bdd32011-08-02 18:34:30 -05005148 assert(left->size == right->size && right->size == res->size); \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005149 op(res->size, res->set, left->set, right->set); \
5150 return (PyObject *)res; \
5151 } \
5152 static PyObject * \
5153 cpu_set_##name(Py_cpu_set *left, Py_cpu_set *right) { \
5154 return do_cpu_set_##name(left, right, NULL); \
5155 } \
5156 static PyObject * \
5157 cpu_set_i##name(Py_cpu_set *left, Py_cpu_set *right) { \
5158 return do_cpu_set_##name(left, right, left); \
5159 } \
5160
5161CPU_SET_BINOP(and, CPU_AND_S)
5162CPU_SET_BINOP(or, CPU_OR_S)
5163CPU_SET_BINOP(xor, CPU_XOR_S)
5164#undef CPU_SET_BINOP
5165
5166PyDoc_STRVAR(cpu_set_doc,
5167"cpu_set(size)\n\n\
5168Create an empty mask of CPUs.");
5169
5170static PyNumberMethods cpu_set_as_number = {
5171 0, /*nb_add*/
5172 0, /*nb_subtract*/
5173 0, /*nb_multiply*/
5174 0, /*nb_remainder*/
5175 0, /*nb_divmod*/
5176 0, /*nb_power*/
5177 0, /*nb_negative*/
5178 0, /*nb_positive*/
5179 0, /*nb_absolute*/
5180 0, /*nb_bool*/
5181 0, /*nb_invert*/
5182 0, /*nb_lshift*/
5183 0, /*nb_rshift*/
5184 (binaryfunc)cpu_set_and, /*nb_and*/
5185 (binaryfunc)cpu_set_xor, /*nb_xor*/
5186 (binaryfunc)cpu_set_or, /*nb_or*/
5187 0, /*nb_int*/
5188 0, /*nb_reserved*/
5189 0, /*nb_float*/
5190 0, /*nb_inplace_add*/
5191 0, /*nb_inplace_subtract*/
5192 0, /*nb_inplace_multiply*/
5193 0, /*nb_inplace_remainder*/
5194 0, /*nb_inplace_power*/
5195 0, /*nb_inplace_lshift*/
5196 0, /*nb_inplace_rshift*/
5197 (binaryfunc)cpu_set_iand, /*nb_inplace_and*/
5198 (binaryfunc)cpu_set_ixor, /*nb_inplace_xor*/
5199 (binaryfunc)cpu_set_ior, /*nb_inplace_or*/
5200};
5201
5202static PySequenceMethods cpu_set_as_sequence = {
5203 (lenfunc)cpu_set_len, /* sq_length */
5204};
5205
5206static PyMethodDef cpu_set_methods[] = {
5207 {"clear", (PyCFunction)cpu_set_clear, METH_VARARGS, cpu_set_clear_doc},
5208 {"count", (PyCFunction)cpu_set_count, METH_NOARGS, cpu_set_count_doc},
5209 {"isset", (PyCFunction)cpu_set_isset, METH_VARARGS, cpu_set_isset_doc},
5210 {"set", (PyCFunction)cpu_set_set, METH_VARARGS, cpu_set_set_doc},
5211 {"zero", (PyCFunction)cpu_set_zero, METH_NOARGS, cpu_set_zero_doc},
5212 {NULL, NULL} /* sentinel */
5213};
5214
5215static PyTypeObject cpu_set_type = {
5216 PyVarObject_HEAD_INIT(&PyType_Type, 0)
5217 "posix.cpu_set", /* tp_name */
5218 sizeof(Py_cpu_set), /* tp_basicsize */
5219 0, /* tp_itemsize */
5220 /* methods */
5221 (destructor)cpu_set_dealloc, /* tp_dealloc */
5222 0, /* tp_print */
5223 0, /* tp_getattr */
5224 0, /* tp_setattr */
5225 0, /* tp_reserved */
5226 (reprfunc)cpu_set_repr, /* tp_repr */
5227 &cpu_set_as_number, /* tp_as_number */
5228 &cpu_set_as_sequence, /* tp_as_sequence */
5229 0, /* tp_as_mapping */
5230 PyObject_HashNotImplemented, /* tp_hash */
5231 0, /* tp_call */
5232 0, /* tp_str */
5233 PyObject_GenericGetAttr, /* tp_getattro */
5234 0, /* tp_setattro */
5235 0, /* tp_as_buffer */
5236 Py_TPFLAGS_DEFAULT, /* tp_flags */
5237 cpu_set_doc, /* tp_doc */
5238 0, /* tp_traverse */
5239 0, /* tp_clear */
5240 (richcmpfunc)cpu_set_richcompare, /* tp_richcompare */
5241 0, /* tp_weaklistoffset */
5242 0, /* tp_iter */
5243 0, /* tp_iternext */
5244 cpu_set_methods, /* tp_methods */
5245 0, /* tp_members */
5246 0, /* tp_getset */
5247 0, /* tp_base */
5248 0, /* tp_dict */
5249 0, /* tp_descr_get */
5250 0, /* tp_descr_set */
5251 0, /* tp_dictoffset */
5252 0, /* tp_init */
5253 PyType_GenericAlloc, /* tp_alloc */
5254 cpu_set_new, /* tp_new */
5255 PyObject_Del, /* tp_free */
5256};
5257
5258PyDoc_STRVAR(posix_sched_setaffinity__doc__,
5259"sched_setaffinity(pid, cpu_set)\n\n\
5260Set the affinity of the process with PID *pid* to *cpu_set*.");
5261
5262static PyObject *
5263posix_sched_setaffinity(PyObject *self, PyObject *args)
5264{
5265 pid_t pid;
5266 Py_cpu_set *cpu_set;
5267
Benjamin Petersona17a5d62011-08-09 16:49:13 -05005268 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "O!:sched_setaffinity",
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005269 &pid, &cpu_set_type, &cpu_set))
5270 return NULL;
5271 if (sched_setaffinity(pid, cpu_set->size, cpu_set->set))
5272 return posix_error();
5273 Py_RETURN_NONE;
5274}
5275
5276PyDoc_STRVAR(posix_sched_getaffinity__doc__,
5277"sched_getaffinity(pid, ncpus) -> cpu_set\n\n\
5278Return the affinity of the process with PID *pid*.\n\
5279The returned cpu_set will be of size *ncpus*.");
5280
5281static PyObject *
5282posix_sched_getaffinity(PyObject *self, PyObject *args)
5283{
5284 pid_t pid;
5285 int ncpus;
5286 Py_cpu_set *res;
5287
Benjamin Peterson7ac92142011-08-03 08:54:26 -05005288 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:sched_getaffinity",
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005289 &pid, &ncpus))
5290 return NULL;
5291 res = make_new_cpu_set(&cpu_set_type, ncpus);
5292 if (!res)
5293 return NULL;
5294 if (sched_getaffinity(pid, res->size, res->set)) {
5295 Py_DECREF(res);
5296 return posix_error();
5297 }
5298 return (PyObject *)res;
5299}
5300
Benjamin Peterson2740af82011-08-02 17:41:34 -05005301#endif /* HAVE_SCHED_SETAFFINITY */
5302
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005303#endif /* HAVE_SCHED_H */
5304
Neal Norwitzb59798b2003-03-21 01:43:31 +00005305/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00005306/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
5307#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00005308#define DEV_PTY_FILE "/dev/ptc"
5309#define HAVE_DEV_PTMX
5310#else
5311#define DEV_PTY_FILE "/dev/ptmx"
5312#endif
5313
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005314#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00005315#ifdef HAVE_PTY_H
5316#include <pty.h>
5317#else
5318#ifdef HAVE_LIBUTIL_H
5319#include <libutil.h>
Ronald Oussoren755740f2010-02-07 19:56:39 +00005320#else
5321#ifdef HAVE_UTIL_H
5322#include <util.h>
5323#endif /* HAVE_UTIL_H */
Fred Drake8cef4cf2000-06-28 16:40:38 +00005324#endif /* HAVE_LIBUTIL_H */
5325#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00005326#ifdef HAVE_STROPTS_H
5327#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005328#endif
5329#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00005330
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005331#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005332PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005333"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005334Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00005335
5336static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005337posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00005338{
Victor Stinner8c62be82010-05-06 00:08:46 +00005339 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00005340#ifndef HAVE_OPENPTY
Victor Stinner8c62be82010-05-06 00:08:46 +00005341 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00005342#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005343#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Victor Stinner8c62be82010-05-06 00:08:46 +00005344 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005345#ifdef sun
Victor Stinner8c62be82010-05-06 00:08:46 +00005346 extern char *ptsname(int fildes);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005347#endif
5348#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00005349
Thomas Wouters70c21a12000-07-14 14:28:33 +00005350#ifdef HAVE_OPENPTY
Victor Stinner8c62be82010-05-06 00:08:46 +00005351 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
5352 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00005353#elif defined(HAVE__GETPTY)
Victor Stinner8c62be82010-05-06 00:08:46 +00005354 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
5355 if (slave_name == NULL)
5356 return posix_error();
Thomas Wouters70c21a12000-07-14 14:28:33 +00005357
Victor Stinner8c62be82010-05-06 00:08:46 +00005358 slave_fd = open(slave_name, O_RDWR);
5359 if (slave_fd < 0)
5360 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005361#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005362 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
5363 if (master_fd < 0)
5364 return posix_error();
5365 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
5366 /* change permission of slave */
5367 if (grantpt(master_fd) < 0) {
5368 PyOS_setsig(SIGCHLD, sig_saved);
5369 return posix_error();
5370 }
5371 /* unlock slave */
5372 if (unlockpt(master_fd) < 0) {
5373 PyOS_setsig(SIGCHLD, sig_saved);
5374 return posix_error();
5375 }
5376 PyOS_setsig(SIGCHLD, sig_saved);
5377 slave_name = ptsname(master_fd); /* get name of slave */
5378 if (slave_name == NULL)
5379 return posix_error();
5380 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
5381 if (slave_fd < 0)
5382 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00005383#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Victor Stinner8c62be82010-05-06 00:08:46 +00005384 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
5385 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00005386#ifndef __hpux
Victor Stinner8c62be82010-05-06 00:08:46 +00005387 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00005388#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005389#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00005390#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00005391
Victor Stinner8c62be82010-05-06 00:08:46 +00005392 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00005393
Fred Drake8cef4cf2000-06-28 16:40:38 +00005394}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005395#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00005396
5397#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005398PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005399"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00005400Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
5401Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005402To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00005403
5404static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005405posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00005406{
Victor Stinner8c62be82010-05-06 00:08:46 +00005407 int master_fd = -1, result = 0;
5408 pid_t pid;
Tim Peters5aa91602002-01-30 05:46:57 +00005409
Victor Stinner8c62be82010-05-06 00:08:46 +00005410 _PyImport_AcquireLock();
5411 pid = forkpty(&master_fd, NULL, NULL, NULL);
5412 if (pid == 0) {
5413 /* child: this clobbers and resets the import lock. */
5414 PyOS_AfterFork();
5415 } else {
5416 /* parent: release the import lock. */
5417 result = _PyImport_ReleaseLock();
5418 }
5419 if (pid == -1)
5420 return posix_error();
5421 if (result < 0) {
5422 /* Don't clobber the OSError if the fork failed. */
5423 PyErr_SetString(PyExc_RuntimeError,
5424 "not holding the import lock");
5425 return NULL;
5426 }
5427 return Py_BuildValue("(Ni)", PyLong_FromPid(pid), master_fd);
Fred Drake8cef4cf2000-06-28 16:40:38 +00005428}
5429#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005430
Ross Lagerwall7807c352011-03-17 20:20:30 +02005431
Guido van Rossumad0ee831995-03-01 10:34:45 +00005432#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005433PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005434"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005435Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005436
Barry Warsaw53699e91996-12-10 23:23:01 +00005437static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005438posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005439{
Victor Stinner8c62be82010-05-06 00:08:46 +00005440 return PyLong_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005441}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005442#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005443
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005444
Guido van Rossumad0ee831995-03-01 10:34:45 +00005445#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005446PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005447"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005448Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005449
Barry Warsaw53699e91996-12-10 23:23:01 +00005450static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005451posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005452{
Victor Stinner8c62be82010-05-06 00:08:46 +00005453 return PyLong_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005454}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005455#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005456
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005457
Guido van Rossumad0ee831995-03-01 10:34:45 +00005458#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005459PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005460"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005461Return the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005462
Barry Warsaw53699e91996-12-10 23:23:01 +00005463static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005464posix_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005465{
Victor Stinner8c62be82010-05-06 00:08:46 +00005466 return PyLong_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005467}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005468#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005469
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005470
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005471PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005472"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005473Return the current process id");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005474
Barry Warsaw53699e91996-12-10 23:23:01 +00005475static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005476posix_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005477{
Victor Stinner8c62be82010-05-06 00:08:46 +00005478 return PyLong_FromPid(getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00005479}
5480
Ross Lagerwallb0ae53d2011-06-10 07:30:30 +02005481#ifdef HAVE_GETGROUPLIST
5482PyDoc_STRVAR(posix_getgrouplist__doc__,
5483"getgrouplist(user, group) -> list of groups to which a user belongs\n\n\
5484Returns a list of groups to which a user belongs.\n\n\
5485 user: username to lookup\n\
5486 group: base group id of the user");
5487
5488static PyObject *
5489posix_getgrouplist(PyObject *self, PyObject *args)
5490{
5491#ifdef NGROUPS_MAX
5492#define MAX_GROUPS NGROUPS_MAX
5493#else
5494 /* defined to be 16 on Solaris7, so this should be a small number */
5495#define MAX_GROUPS 64
5496#endif
5497
5498 const char *user;
5499 int i, ngroups;
5500 PyObject *list;
5501#ifdef __APPLE__
5502 int *groups, basegid;
5503#else
5504 gid_t *groups, basegid;
5505#endif
5506 ngroups = MAX_GROUPS;
5507
5508 if (!PyArg_ParseTuple(args, "si", &user, &basegid))
5509 return NULL;
5510
5511#ifdef __APPLE__
5512 groups = PyMem_Malloc(ngroups * sizeof(int));
5513#else
5514 groups = PyMem_Malloc(ngroups * sizeof(gid_t));
5515#endif
5516 if (groups == NULL)
5517 return PyErr_NoMemory();
5518
5519 if (getgrouplist(user, basegid, groups, &ngroups) == -1) {
5520 PyMem_Del(groups);
5521 return posix_error();
5522 }
5523
5524 list = PyList_New(ngroups);
5525 if (list == NULL) {
5526 PyMem_Del(groups);
5527 return NULL;
5528 }
5529
5530 for (i = 0; i < ngroups; i++) {
5531 PyObject *o = PyLong_FromUnsignedLong((unsigned long)groups[i]);
5532 if (o == NULL) {
5533 Py_DECREF(list);
5534 PyMem_Del(groups);
5535 return NULL;
5536 }
5537 PyList_SET_ITEM(list, i, o);
5538 }
5539
5540 PyMem_Del(groups);
5541
5542 return list;
5543}
5544#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005545
Fred Drakec9680921999-12-13 16:37:25 +00005546#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005547PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005548"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005549Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00005550
5551static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005552posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00005553{
5554 PyObject *result = NULL;
5555
Fred Drakec9680921999-12-13 16:37:25 +00005556#ifdef NGROUPS_MAX
5557#define MAX_GROUPS NGROUPS_MAX
5558#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005559 /* defined to be 16 on Solaris7, so this should be a small number */
Fred Drakec9680921999-12-13 16:37:25 +00005560#define MAX_GROUPS 64
5561#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00005562 gid_t grouplist[MAX_GROUPS];
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005563
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005564 /* On MacOSX getgroups(2) can return more than MAX_GROUPS results
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005565 * This is a helper variable to store the intermediate result when
5566 * that happens.
5567 *
5568 * To keep the code readable the OSX behaviour is unconditional,
5569 * according to the POSIX spec this should be safe on all unix-y
5570 * systems.
5571 */
5572 gid_t* alt_grouplist = grouplist;
Victor Stinner8c62be82010-05-06 00:08:46 +00005573 int n;
Fred Drakec9680921999-12-13 16:37:25 +00005574
Victor Stinner8c62be82010-05-06 00:08:46 +00005575 n = getgroups(MAX_GROUPS, grouplist);
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005576 if (n < 0) {
5577 if (errno == EINVAL) {
5578 n = getgroups(0, NULL);
5579 if (n == -1) {
5580 return posix_error();
5581 }
5582 if (n == 0) {
5583 /* Avoid malloc(0) */
5584 alt_grouplist = grouplist;
5585 } else {
5586 alt_grouplist = PyMem_Malloc(n * sizeof(gid_t));
5587 if (alt_grouplist == NULL) {
5588 errno = EINVAL;
5589 return posix_error();
5590 }
5591 n = getgroups(n, alt_grouplist);
5592 if (n == -1) {
5593 PyMem_Free(alt_grouplist);
5594 return posix_error();
5595 }
5596 }
5597 } else {
5598 return posix_error();
5599 }
5600 }
5601 result = PyList_New(n);
5602 if (result != NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00005603 int i;
5604 for (i = 0; i < n; ++i) {
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005605 PyObject *o = PyLong_FromLong((long)alt_grouplist[i]);
Victor Stinner8c62be82010-05-06 00:08:46 +00005606 if (o == NULL) {
Stefan Krah0e803b32010-11-26 16:16:47 +00005607 Py_DECREF(result);
5608 result = NULL;
5609 break;
Fred Drakec9680921999-12-13 16:37:25 +00005610 }
Victor Stinner8c62be82010-05-06 00:08:46 +00005611 PyList_SET_ITEM(result, i, o);
Fred Drakec9680921999-12-13 16:37:25 +00005612 }
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005613 }
5614
5615 if (alt_grouplist != grouplist) {
5616 PyMem_Free(alt_grouplist);
Victor Stinner8c62be82010-05-06 00:08:46 +00005617 }
Neal Norwitze241ce82003-02-17 18:17:05 +00005618
Fred Drakec9680921999-12-13 16:37:25 +00005619 return result;
5620}
5621#endif
5622
Antoine Pitroub7572f02009-12-02 20:46:48 +00005623#ifdef HAVE_INITGROUPS
5624PyDoc_STRVAR(posix_initgroups__doc__,
5625"initgroups(username, gid) -> None\n\n\
5626Call the system initgroups() to initialize the group access list with all of\n\
5627the groups of which the specified username is a member, plus the specified\n\
5628group id.");
5629
5630static PyObject *
5631posix_initgroups(PyObject *self, PyObject *args)
5632{
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005633 PyObject *oname;
Victor Stinner8c62be82010-05-06 00:08:46 +00005634 char *username;
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005635 int res;
Victor Stinner8c62be82010-05-06 00:08:46 +00005636 long gid;
Antoine Pitroub7572f02009-12-02 20:46:48 +00005637
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005638 if (!PyArg_ParseTuple(args, "O&l:initgroups",
5639 PyUnicode_FSConverter, &oname, &gid))
Victor Stinner8c62be82010-05-06 00:08:46 +00005640 return NULL;
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005641 username = PyBytes_AS_STRING(oname);
Antoine Pitroub7572f02009-12-02 20:46:48 +00005642
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005643 res = initgroups(username, (gid_t) gid);
5644 Py_DECREF(oname);
5645 if (res == -1)
Victor Stinner8c62be82010-05-06 00:08:46 +00005646 return PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitroub7572f02009-12-02 20:46:48 +00005647
Victor Stinner8c62be82010-05-06 00:08:46 +00005648 Py_INCREF(Py_None);
5649 return Py_None;
Antoine Pitroub7572f02009-12-02 20:46:48 +00005650}
5651#endif
5652
Martin v. Löwis606edc12002-06-13 21:09:11 +00005653#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00005654PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005655"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00005656Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00005657
5658static PyObject *
5659posix_getpgid(PyObject *self, PyObject *args)
5660{
Victor Stinner8c62be82010-05-06 00:08:46 +00005661 pid_t pid, pgid;
5662 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":getpgid", &pid))
5663 return NULL;
5664 pgid = getpgid(pid);
5665 if (pgid < 0)
5666 return posix_error();
5667 return PyLong_FromPid(pgid);
Martin v. Löwis606edc12002-06-13 21:09:11 +00005668}
5669#endif /* HAVE_GETPGID */
5670
5671
Guido van Rossumb6775db1994-08-01 11:34:53 +00005672#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005673PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005674"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005675Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005676
Barry Warsaw53699e91996-12-10 23:23:01 +00005677static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005678posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00005679{
Guido van Rossumb6775db1994-08-01 11:34:53 +00005680#ifdef GETPGRP_HAVE_ARG
Victor Stinner8c62be82010-05-06 00:08:46 +00005681 return PyLong_FromPid(getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005682#else /* GETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00005683 return PyLong_FromPid(getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005684#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00005685}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005686#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00005687
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005688
Guido van Rossumb6775db1994-08-01 11:34:53 +00005689#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005690PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005691"setpgrp()\n\n\
Senthil Kumaran684760a2010-06-17 16:48:06 +00005692Make this process the process group leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005693
Barry Warsaw53699e91996-12-10 23:23:01 +00005694static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005695posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00005696{
Guido van Rossum64933891994-10-20 21:56:42 +00005697#ifdef SETPGRP_HAVE_ARG
Victor Stinner8c62be82010-05-06 00:08:46 +00005698 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00005699#else /* SETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00005700 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00005701#endif /* SETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00005702 return posix_error();
5703 Py_INCREF(Py_None);
5704 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00005705}
5706
Guido van Rossumb6775db1994-08-01 11:34:53 +00005707#endif /* HAVE_SETPGRP */
5708
Guido van Rossumad0ee831995-03-01 10:34:45 +00005709#ifdef HAVE_GETPPID
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005710
5711#ifdef MS_WINDOWS
5712#include <tlhelp32.h>
5713
5714static PyObject*
5715win32_getppid()
5716{
5717 HANDLE snapshot;
5718 pid_t mypid;
5719 PyObject* result = NULL;
5720 BOOL have_record;
5721 PROCESSENTRY32 pe;
5722
5723 mypid = getpid(); /* This function never fails */
5724
5725 snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
5726 if (snapshot == INVALID_HANDLE_VALUE)
5727 return PyErr_SetFromWindowsErr(GetLastError());
5728
5729 pe.dwSize = sizeof(pe);
5730 have_record = Process32First(snapshot, &pe);
5731 while (have_record) {
5732 if (mypid == (pid_t)pe.th32ProcessID) {
5733 /* We could cache the ulong value in a static variable. */
5734 result = PyLong_FromPid((pid_t)pe.th32ParentProcessID);
5735 break;
5736 }
5737
5738 have_record = Process32Next(snapshot, &pe);
5739 }
5740
5741 /* If our loop exits and our pid was not found (result will be NULL)
5742 * then GetLastError will return ERROR_NO_MORE_FILES. This is an
5743 * error anyway, so let's raise it. */
5744 if (!result)
5745 result = PyErr_SetFromWindowsErr(GetLastError());
5746
5747 CloseHandle(snapshot);
5748
5749 return result;
5750}
5751#endif /*MS_WINDOWS*/
5752
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005753PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005754"getppid() -> ppid\n\n\
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005755Return the parent's process id. If the parent process has already exited,\n\
5756Windows machines will still return its id; others systems will return the id\n\
5757of the 'init' process (1).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005758
Barry Warsaw53699e91996-12-10 23:23:01 +00005759static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005760posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005761{
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005762#ifdef MS_WINDOWS
5763 return win32_getppid();
5764#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005765 return PyLong_FromPid(getppid());
Guido van Rossumad0ee831995-03-01 10:34:45 +00005766#endif
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005767}
5768#endif /* HAVE_GETPPID */
Guido van Rossum85e3b011991-06-03 12:42:10 +00005769
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005770
Fred Drake12c6e2d1999-12-14 21:25:03 +00005771#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005772PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005773"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005774Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00005775
5776static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005777posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00005778{
Victor Stinner8c62be82010-05-06 00:08:46 +00005779 PyObject *result = NULL;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005780#ifdef MS_WINDOWS
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005781 wchar_t user_name[UNLEN + 1];
Victor Stinner63941882011-09-29 00:42:28 +02005782 DWORD num_chars = Py_ARRAY_LENGTH(user_name);
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005783
5784 if (GetUserNameW(user_name, &num_chars)) {
5785 /* num_chars is the number of unicode chars plus null terminator */
5786 result = PyUnicode_FromWideChar(user_name, num_chars - 1);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005787 }
5788 else
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005789 result = PyErr_SetFromWindowsErr(GetLastError());
5790#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005791 char *name;
5792 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005793
Victor Stinner8c62be82010-05-06 00:08:46 +00005794 errno = 0;
5795 name = getlogin();
5796 if (name == NULL) {
5797 if (errno)
Victor Stinnere039ffe2010-08-15 09:33:08 +00005798 posix_error();
Fred Drake12c6e2d1999-12-14 21:25:03 +00005799 else
Victor Stinnere039ffe2010-08-15 09:33:08 +00005800 PyErr_SetString(PyExc_OSError, "unable to determine login name");
Victor Stinner8c62be82010-05-06 00:08:46 +00005801 }
5802 else
Victor Stinnere039ffe2010-08-15 09:33:08 +00005803 result = PyUnicode_DecodeFSDefault(name);
Victor Stinner8c62be82010-05-06 00:08:46 +00005804 errno = old_errno;
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005805#endif
Fred Drake12c6e2d1999-12-14 21:25:03 +00005806 return result;
5807}
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005808#endif /* HAVE_GETLOGIN */
Fred Drake12c6e2d1999-12-14 21:25:03 +00005809
Guido van Rossumad0ee831995-03-01 10:34:45 +00005810#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005811PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005812"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005813Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005814
Barry Warsaw53699e91996-12-10 23:23:01 +00005815static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005816posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005817{
Victor Stinner8c62be82010-05-06 00:08:46 +00005818 return PyLong_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005819}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005820#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005821
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005822
Guido van Rossumad0ee831995-03-01 10:34:45 +00005823#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005824PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005825"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005826Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005827
Barry Warsaw53699e91996-12-10 23:23:01 +00005828static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005829posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005830{
Victor Stinner8c62be82010-05-06 00:08:46 +00005831 pid_t pid;
5832 int sig;
5833 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:kill", &pid, &sig))
5834 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00005835#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005836 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
5837 APIRET rc;
5838 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005839 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005840
5841 } else if (sig == XCPT_SIGNAL_KILLPROC) {
5842 APIRET rc;
5843 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005844 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005845
5846 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00005847 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005848#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005849 if (kill(pid, sig) == -1)
5850 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005851#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00005852 Py_INCREF(Py_None);
5853 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00005854}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005855#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00005856
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005857#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005858PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005859"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005860Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005861
5862static PyObject *
5863posix_killpg(PyObject *self, PyObject *args)
5864{
Victor Stinner8c62be82010-05-06 00:08:46 +00005865 int sig;
5866 pid_t pgid;
5867 /* XXX some man pages make the `pgid` parameter an int, others
5868 a pid_t. Since getpgrp() returns a pid_t, we assume killpg should
5869 take the same type. Moreover, pid_t is always at least as wide as
5870 int (else compilation of this module fails), which is safe. */
5871 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:killpg", &pgid, &sig))
5872 return NULL;
5873 if (killpg(pgid, sig) == -1)
5874 return posix_error();
5875 Py_INCREF(Py_None);
5876 return Py_None;
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005877}
5878#endif
5879
Brian Curtineb24d742010-04-12 17:16:38 +00005880#ifdef MS_WINDOWS
5881PyDoc_STRVAR(win32_kill__doc__,
5882"kill(pid, sig)\n\n\
5883Kill a process with a signal.");
5884
5885static PyObject *
5886win32_kill(PyObject *self, PyObject *args)
5887{
Amaury Forgeot d'Arc0a589c92010-05-15 20:35:12 +00005888 PyObject *result;
Victor Stinner8c62be82010-05-06 00:08:46 +00005889 DWORD pid, sig, err;
5890 HANDLE handle;
Brian Curtineb24d742010-04-12 17:16:38 +00005891
Victor Stinner8c62be82010-05-06 00:08:46 +00005892 if (!PyArg_ParseTuple(args, "kk:kill", &pid, &sig))
5893 return NULL;
Brian Curtineb24d742010-04-12 17:16:38 +00005894
Victor Stinner8c62be82010-05-06 00:08:46 +00005895 /* Console processes which share a common console can be sent CTRL+C or
5896 CTRL+BREAK events, provided they handle said events. */
5897 if (sig == CTRL_C_EVENT || sig == CTRL_BREAK_EVENT) {
5898 if (GenerateConsoleCtrlEvent(sig, pid) == 0) {
5899 err = GetLastError();
5900 PyErr_SetFromWindowsErr(err);
5901 }
5902 else
5903 Py_RETURN_NONE;
5904 }
Brian Curtineb24d742010-04-12 17:16:38 +00005905
Victor Stinner8c62be82010-05-06 00:08:46 +00005906 /* If the signal is outside of what GenerateConsoleCtrlEvent can use,
5907 attempt to open and terminate the process. */
5908 handle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, pid);
5909 if (handle == NULL) {
5910 err = GetLastError();
5911 return PyErr_SetFromWindowsErr(err);
5912 }
Brian Curtineb24d742010-04-12 17:16:38 +00005913
Victor Stinner8c62be82010-05-06 00:08:46 +00005914 if (TerminateProcess(handle, sig) == 0) {
5915 err = GetLastError();
5916 result = PyErr_SetFromWindowsErr(err);
5917 } else {
5918 Py_INCREF(Py_None);
5919 result = Py_None;
5920 }
Brian Curtineb24d742010-04-12 17:16:38 +00005921
Victor Stinner8c62be82010-05-06 00:08:46 +00005922 CloseHandle(handle);
5923 return result;
Brian Curtineb24d742010-04-12 17:16:38 +00005924}
5925#endif /* MS_WINDOWS */
5926
Guido van Rossumc0125471996-06-28 18:55:32 +00005927#ifdef HAVE_PLOCK
5928
5929#ifdef HAVE_SYS_LOCK_H
5930#include <sys/lock.h>
5931#endif
5932
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005933PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005934"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005935Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005936
Barry Warsaw53699e91996-12-10 23:23:01 +00005937static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005938posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00005939{
Victor Stinner8c62be82010-05-06 00:08:46 +00005940 int op;
5941 if (!PyArg_ParseTuple(args, "i:plock", &op))
5942 return NULL;
5943 if (plock(op) == -1)
5944 return posix_error();
5945 Py_INCREF(Py_None);
5946 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00005947}
5948#endif
5949
Guido van Rossumb6775db1994-08-01 11:34:53 +00005950#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005951PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005952"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005953Set the current process's user id.");
5954
Barry Warsaw53699e91996-12-10 23:23:01 +00005955static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005956posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005957{
Victor Stinner8c62be82010-05-06 00:08:46 +00005958 long uid_arg;
5959 uid_t uid;
5960 if (!PyArg_ParseTuple(args, "l:setuid", &uid_arg))
5961 return NULL;
5962 uid = uid_arg;
5963 if (uid != uid_arg) {
5964 PyErr_SetString(PyExc_OverflowError, "user id too big");
5965 return NULL;
5966 }
5967 if (setuid(uid) < 0)
5968 return posix_error();
5969 Py_INCREF(Py_None);
5970 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005971}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005972#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005973
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005974
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005975#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005976PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005977"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005978Set the current process's effective user id.");
5979
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005980static PyObject *
5981posix_seteuid (PyObject *self, PyObject *args)
5982{
Victor Stinner8c62be82010-05-06 00:08:46 +00005983 long euid_arg;
5984 uid_t euid;
5985 if (!PyArg_ParseTuple(args, "l", &euid_arg))
5986 return NULL;
5987 euid = euid_arg;
5988 if (euid != euid_arg) {
5989 PyErr_SetString(PyExc_OverflowError, "user id too big");
5990 return NULL;
5991 }
5992 if (seteuid(euid) < 0) {
5993 return posix_error();
5994 } else {
5995 Py_INCREF(Py_None);
5996 return Py_None;
5997 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005998}
5999#endif /* HAVE_SETEUID */
6000
6001#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006002PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006003"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006004Set the current process's effective group id.");
6005
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006006static PyObject *
6007posix_setegid (PyObject *self, PyObject *args)
6008{
Victor Stinner8c62be82010-05-06 00:08:46 +00006009 long egid_arg;
6010 gid_t egid;
6011 if (!PyArg_ParseTuple(args, "l", &egid_arg))
6012 return NULL;
6013 egid = egid_arg;
6014 if (egid != egid_arg) {
6015 PyErr_SetString(PyExc_OverflowError, "group id too big");
6016 return NULL;
6017 }
6018 if (setegid(egid) < 0) {
6019 return posix_error();
6020 } else {
6021 Py_INCREF(Py_None);
6022 return Py_None;
6023 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006024}
6025#endif /* HAVE_SETEGID */
6026
6027#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006028PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00006029"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006030Set the current process's real and effective user ids.");
6031
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006032static PyObject *
6033posix_setreuid (PyObject *self, PyObject *args)
6034{
Victor Stinner8c62be82010-05-06 00:08:46 +00006035 long ruid_arg, euid_arg;
6036 uid_t ruid, euid;
6037 if (!PyArg_ParseTuple(args, "ll", &ruid_arg, &euid_arg))
6038 return NULL;
6039 if (ruid_arg == -1)
6040 ruid = (uid_t)-1; /* let the compiler choose how -1 fits */
6041 else
6042 ruid = ruid_arg; /* otherwise, assign from our long */
6043 if (euid_arg == -1)
6044 euid = (uid_t)-1;
6045 else
6046 euid = euid_arg;
6047 if ((euid_arg != -1 && euid != euid_arg) ||
6048 (ruid_arg != -1 && ruid != ruid_arg)) {
6049 PyErr_SetString(PyExc_OverflowError, "user id too big");
6050 return NULL;
6051 }
6052 if (setreuid(ruid, euid) < 0) {
6053 return posix_error();
6054 } else {
6055 Py_INCREF(Py_None);
6056 return Py_None;
6057 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006058}
6059#endif /* HAVE_SETREUID */
6060
6061#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006062PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00006063"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006064Set the current process's real and effective group ids.");
6065
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006066static PyObject *
6067posix_setregid (PyObject *self, PyObject *args)
6068{
Victor Stinner8c62be82010-05-06 00:08:46 +00006069 long rgid_arg, egid_arg;
6070 gid_t rgid, egid;
6071 if (!PyArg_ParseTuple(args, "ll", &rgid_arg, &egid_arg))
6072 return NULL;
6073 if (rgid_arg == -1)
6074 rgid = (gid_t)-1; /* let the compiler choose how -1 fits */
6075 else
6076 rgid = rgid_arg; /* otherwise, assign from our long */
6077 if (egid_arg == -1)
6078 egid = (gid_t)-1;
6079 else
6080 egid = egid_arg;
6081 if ((egid_arg != -1 && egid != egid_arg) ||
6082 (rgid_arg != -1 && rgid != rgid_arg)) {
6083 PyErr_SetString(PyExc_OverflowError, "group id too big");
6084 return NULL;
6085 }
6086 if (setregid(rgid, egid) < 0) {
6087 return posix_error();
6088 } else {
6089 Py_INCREF(Py_None);
6090 return Py_None;
6091 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006092}
6093#endif /* HAVE_SETREGID */
6094
Guido van Rossumb6775db1994-08-01 11:34:53 +00006095#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006096PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006097"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006098Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006099
Barry Warsaw53699e91996-12-10 23:23:01 +00006100static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006101posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00006102{
Victor Stinner8c62be82010-05-06 00:08:46 +00006103 long gid_arg;
6104 gid_t gid;
6105 if (!PyArg_ParseTuple(args, "l:setgid", &gid_arg))
6106 return NULL;
6107 gid = gid_arg;
6108 if (gid != gid_arg) {
6109 PyErr_SetString(PyExc_OverflowError, "group id too big");
6110 return NULL;
6111 }
6112 if (setgid(gid) < 0)
6113 return posix_error();
6114 Py_INCREF(Py_None);
6115 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00006116}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006117#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00006118
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006119#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006120PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006121"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006122Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006123
6124static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00006125posix_setgroups(PyObject *self, PyObject *groups)
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006126{
Victor Stinner8c62be82010-05-06 00:08:46 +00006127 int i, len;
6128 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00006129
Victor Stinner8c62be82010-05-06 00:08:46 +00006130 if (!PySequence_Check(groups)) {
6131 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
6132 return NULL;
6133 }
6134 len = PySequence_Size(groups);
6135 if (len > MAX_GROUPS) {
6136 PyErr_SetString(PyExc_ValueError, "too many groups");
6137 return NULL;
6138 }
6139 for(i = 0; i < len; i++) {
6140 PyObject *elem;
6141 elem = PySequence_GetItem(groups, i);
6142 if (!elem)
6143 return NULL;
6144 if (!PyLong_Check(elem)) {
6145 PyErr_SetString(PyExc_TypeError,
6146 "groups must be integers");
6147 Py_DECREF(elem);
6148 return NULL;
6149 } else {
6150 unsigned long x = PyLong_AsUnsignedLong(elem);
6151 if (PyErr_Occurred()) {
6152 PyErr_SetString(PyExc_TypeError,
6153 "group id too big");
6154 Py_DECREF(elem);
6155 return NULL;
6156 }
6157 grouplist[i] = x;
6158 /* read back the value to see if it fitted in gid_t */
6159 if (grouplist[i] != x) {
6160 PyErr_SetString(PyExc_TypeError,
6161 "group id too big");
6162 Py_DECREF(elem);
6163 return NULL;
6164 }
6165 }
6166 Py_DECREF(elem);
6167 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006168
Victor Stinner8c62be82010-05-06 00:08:46 +00006169 if (setgroups(len, grouplist) < 0)
6170 return posix_error();
6171 Py_INCREF(Py_None);
6172 return Py_None;
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006173}
6174#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006175
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006176#if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
6177static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006178wait_helper(pid_t pid, int status, struct rusage *ru)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006179{
Victor Stinner8c62be82010-05-06 00:08:46 +00006180 PyObject *result;
6181 static PyObject *struct_rusage;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02006182 _Py_IDENTIFIER(struct_rusage);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006183
Victor Stinner8c62be82010-05-06 00:08:46 +00006184 if (pid == -1)
6185 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006186
Victor Stinner8c62be82010-05-06 00:08:46 +00006187 if (struct_rusage == NULL) {
6188 PyObject *m = PyImport_ImportModuleNoBlock("resource");
6189 if (m == NULL)
6190 return NULL;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006191 struct_rusage = _PyObject_GetAttrId(m, &PyId_struct_rusage);
Victor Stinner8c62be82010-05-06 00:08:46 +00006192 Py_DECREF(m);
6193 if (struct_rusage == NULL)
6194 return NULL;
6195 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006196
Victor Stinner8c62be82010-05-06 00:08:46 +00006197 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
6198 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
6199 if (!result)
6200 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006201
6202#ifndef doubletime
6203#define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
6204#endif
6205
Victor Stinner8c62be82010-05-06 00:08:46 +00006206 PyStructSequence_SET_ITEM(result, 0,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006207 PyFloat_FromDouble(doubletime(ru->ru_utime)));
Victor Stinner8c62be82010-05-06 00:08:46 +00006208 PyStructSequence_SET_ITEM(result, 1,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006209 PyFloat_FromDouble(doubletime(ru->ru_stime)));
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006210#define SET_INT(result, index, value)\
Victor Stinner8c62be82010-05-06 00:08:46 +00006211 PyStructSequence_SET_ITEM(result, index, PyLong_FromLong(value))
6212 SET_INT(result, 2, ru->ru_maxrss);
6213 SET_INT(result, 3, ru->ru_ixrss);
6214 SET_INT(result, 4, ru->ru_idrss);
6215 SET_INT(result, 5, ru->ru_isrss);
6216 SET_INT(result, 6, ru->ru_minflt);
6217 SET_INT(result, 7, ru->ru_majflt);
6218 SET_INT(result, 8, ru->ru_nswap);
6219 SET_INT(result, 9, ru->ru_inblock);
6220 SET_INT(result, 10, ru->ru_oublock);
6221 SET_INT(result, 11, ru->ru_msgsnd);
6222 SET_INT(result, 12, ru->ru_msgrcv);
6223 SET_INT(result, 13, ru->ru_nsignals);
6224 SET_INT(result, 14, ru->ru_nvcsw);
6225 SET_INT(result, 15, ru->ru_nivcsw);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006226#undef SET_INT
6227
Victor Stinner8c62be82010-05-06 00:08:46 +00006228 if (PyErr_Occurred()) {
6229 Py_DECREF(result);
6230 return NULL;
6231 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006232
Victor Stinner8c62be82010-05-06 00:08:46 +00006233 return Py_BuildValue("NiN", PyLong_FromPid(pid), status, result);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006234}
6235#endif /* HAVE_WAIT3 || HAVE_WAIT4 */
6236
6237#ifdef HAVE_WAIT3
6238PyDoc_STRVAR(posix_wait3__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006239"wait3(options) -> (pid, status, rusage)\n\n\
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006240Wait for completion of a child process.");
6241
6242static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006243posix_wait3(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006244{
Victor Stinner8c62be82010-05-06 00:08:46 +00006245 pid_t pid;
6246 int options;
6247 struct rusage ru;
6248 WAIT_TYPE status;
6249 WAIT_STATUS_INT(status) = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006250
Victor Stinner4195b5c2012-02-08 23:03:19 +01006251 if (!PyArg_ParseTuple(args, "i:wait3", &options))
Victor Stinner8c62be82010-05-06 00:08:46 +00006252 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006253
Victor Stinner8c62be82010-05-06 00:08:46 +00006254 Py_BEGIN_ALLOW_THREADS
6255 pid = wait3(&status, options, &ru);
6256 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006257
Victor Stinner4195b5c2012-02-08 23:03:19 +01006258 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006259}
6260#endif /* HAVE_WAIT3 */
6261
6262#ifdef HAVE_WAIT4
6263PyDoc_STRVAR(posix_wait4__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006264"wait4(pid, options) -> (pid, status, rusage)\n\n\
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006265Wait for completion of a given child process.");
6266
6267static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006268posix_wait4(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006269{
Victor Stinner8c62be82010-05-06 00:08:46 +00006270 pid_t pid;
6271 int options;
6272 struct rusage ru;
6273 WAIT_TYPE status;
6274 WAIT_STATUS_INT(status) = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006275
Victor Stinner4195b5c2012-02-08 23:03:19 +01006276 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:wait4", &pid, &options))
Victor Stinner8c62be82010-05-06 00:08:46 +00006277 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006278
Victor Stinner8c62be82010-05-06 00:08:46 +00006279 Py_BEGIN_ALLOW_THREADS
6280 pid = wait4(pid, &status, options, &ru);
6281 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006282
Victor Stinner4195b5c2012-02-08 23:03:19 +01006283 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006284}
6285#endif /* HAVE_WAIT4 */
6286
Ross Lagerwall7807c352011-03-17 20:20:30 +02006287#if defined(HAVE_WAITID) && !defined(__APPLE__)
6288PyDoc_STRVAR(posix_waitid__doc__,
6289"waitid(idtype, id, options) -> waitid_result\n\n\
6290Wait for the completion of one or more child processes.\n\n\
6291idtype can be P_PID, P_PGID or P_ALL.\n\
6292id specifies the pid to wait on.\n\
6293options is constructed from the ORing of one or more of WEXITED, WSTOPPED\n\
6294or WCONTINUED and additionally may be ORed with WNOHANG or WNOWAIT.\n\
6295Returns either waitid_result or None if WNOHANG is specified and there are\n\
6296no children in a waitable state.");
6297
6298static PyObject *
6299posix_waitid(PyObject *self, PyObject *args)
6300{
6301 PyObject *result;
6302 idtype_t idtype;
6303 id_t id;
6304 int options, res;
6305 siginfo_t si;
6306 si.si_pid = 0;
6307 if (!PyArg_ParseTuple(args, "i" _Py_PARSE_PID "i:waitid", &idtype, &id, &options))
6308 return NULL;
6309 Py_BEGIN_ALLOW_THREADS
6310 res = waitid(idtype, id, &si, options);
6311 Py_END_ALLOW_THREADS
6312 if (res == -1)
6313 return posix_error();
6314
6315 if (si.si_pid == 0)
6316 Py_RETURN_NONE;
6317
6318 result = PyStructSequence_New(&WaitidResultType);
6319 if (!result)
6320 return NULL;
6321
6322 PyStructSequence_SET_ITEM(result, 0, PyLong_FromPid(si.si_pid));
6323 PyStructSequence_SET_ITEM(result, 1, PyLong_FromPid(si.si_uid));
6324 PyStructSequence_SET_ITEM(result, 2, PyLong_FromLong((long)(si.si_signo)));
6325 PyStructSequence_SET_ITEM(result, 3, PyLong_FromLong((long)(si.si_status)));
6326 PyStructSequence_SET_ITEM(result, 4, PyLong_FromLong((long)(si.si_code)));
6327 if (PyErr_Occurred()) {
6328 Py_DECREF(result);
6329 return NULL;
6330 }
6331
6332 return result;
6333}
6334#endif
6335
Guido van Rossumb6775db1994-08-01 11:34:53 +00006336#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006337PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006338"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006339Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006340
Barry Warsaw53699e91996-12-10 23:23:01 +00006341static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006342posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00006343{
Victor Stinner8c62be82010-05-06 00:08:46 +00006344 pid_t pid;
6345 int options;
6346 WAIT_TYPE status;
6347 WAIT_STATUS_INT(status) = 0;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00006348
Victor Stinner8c62be82010-05-06 00:08:46 +00006349 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:waitpid", &pid, &options))
6350 return NULL;
6351 Py_BEGIN_ALLOW_THREADS
6352 pid = waitpid(pid, &status, options);
6353 Py_END_ALLOW_THREADS
6354 if (pid == -1)
6355 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006356
Victor Stinner8c62be82010-05-06 00:08:46 +00006357 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum21803b81992-08-09 12:55:27 +00006358}
6359
Tim Petersab034fa2002-02-01 11:27:43 +00006360#elif defined(HAVE_CWAIT)
6361
6362/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006363PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006364"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006365"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00006366
6367static PyObject *
6368posix_waitpid(PyObject *self, PyObject *args)
6369{
Victor Stinner8c62be82010-05-06 00:08:46 +00006370 Py_intptr_t pid;
6371 int status, options;
Tim Petersab034fa2002-02-01 11:27:43 +00006372
Victor Stinner8c62be82010-05-06 00:08:46 +00006373 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:waitpid", &pid, &options))
6374 return NULL;
6375 Py_BEGIN_ALLOW_THREADS
6376 pid = _cwait(&status, pid, options);
6377 Py_END_ALLOW_THREADS
6378 if (pid == -1)
6379 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006380
Victor Stinner8c62be82010-05-06 00:08:46 +00006381 /* shift the status left a byte so this is more like the POSIX waitpid */
6382 return Py_BuildValue("Ni", PyLong_FromPid(pid), status << 8);
Tim Petersab034fa2002-02-01 11:27:43 +00006383}
6384#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006385
Guido van Rossumad0ee831995-03-01 10:34:45 +00006386#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006387PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006388"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006389Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006390
Barry Warsaw53699e91996-12-10 23:23:01 +00006391static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006392posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00006393{
Victor Stinner8c62be82010-05-06 00:08:46 +00006394 pid_t pid;
6395 WAIT_TYPE status;
6396 WAIT_STATUS_INT(status) = 0;
Neal Norwitze241ce82003-02-17 18:17:05 +00006397
Victor Stinner8c62be82010-05-06 00:08:46 +00006398 Py_BEGIN_ALLOW_THREADS
6399 pid = wait(&status);
6400 Py_END_ALLOW_THREADS
6401 if (pid == -1)
6402 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006403
Victor Stinner8c62be82010-05-06 00:08:46 +00006404 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum85e3b011991-06-03 12:42:10 +00006405}
Guido van Rossumad0ee831995-03-01 10:34:45 +00006406#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00006407
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006408
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006409PyDoc_STRVAR(posix_lstat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006410"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006411Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006412
Barry Warsaw53699e91996-12-10 23:23:01 +00006413static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006414posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006415{
Guido van Rossumb6775db1994-08-01 11:34:53 +00006416#ifdef HAVE_LSTAT
Victor Stinner4195b5c2012-02-08 23:03:19 +01006417 return posix_do_stat(self, args, "O&:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00006418#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006419#ifdef MS_WINDOWS
Victor Stinner4195b5c2012-02-08 23:03:19 +01006420 return posix_do_stat(self, args, "O&:lstat", win32_lstat, "U:lstat",
Brian Curtind40e6f72010-07-08 21:39:08 +00006421 win32_lstat_w);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006422#else
Victor Stinner4195b5c2012-02-08 23:03:19 +01006423 return posix_do_stat(self, args, "O&:lstat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006424#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00006425#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006426}
6427
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006428
Guido van Rossumb6775db1994-08-01 11:34:53 +00006429#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006430PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006431"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006432Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006433
Barry Warsaw53699e91996-12-10 23:23:01 +00006434static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006435posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006436{
Victor Stinner8c62be82010-05-06 00:08:46 +00006437 PyObject* v;
6438 char buf[MAXPATHLEN];
6439 PyObject *opath;
6440 char *path;
6441 int n;
6442 int arg_is_unicode = 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00006443
Victor Stinner8c62be82010-05-06 00:08:46 +00006444 if (!PyArg_ParseTuple(args, "O&:readlink",
6445 PyUnicode_FSConverter, &opath))
6446 return NULL;
6447 path = PyBytes_AsString(opath);
6448 v = PySequence_GetItem(args, 0);
6449 if (v == NULL) {
6450 Py_DECREF(opath);
6451 return NULL;
6452 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00006453
Victor Stinner8c62be82010-05-06 00:08:46 +00006454 if (PyUnicode_Check(v)) {
6455 arg_is_unicode = 1;
6456 }
6457 Py_DECREF(v);
Thomas Wouters89f507f2006-12-13 04:49:30 +00006458
Victor Stinner8c62be82010-05-06 00:08:46 +00006459 Py_BEGIN_ALLOW_THREADS
6460 n = readlink(path, buf, (int) sizeof buf);
6461 Py_END_ALLOW_THREADS
6462 if (n < 0)
6463 return posix_error_with_allocated_filename(opath);
Thomas Wouters89f507f2006-12-13 04:49:30 +00006464
Victor Stinner8c62be82010-05-06 00:08:46 +00006465 Py_DECREF(opath);
Victor Stinnera45598a2010-05-14 16:35:39 +00006466 if (arg_is_unicode)
6467 return PyUnicode_DecodeFSDefaultAndSize(buf, n);
6468 else
6469 return PyBytes_FromStringAndSize(buf, n);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006470}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006471#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006472
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006473
Brian Curtin52173d42010-12-02 18:29:18 +00006474#if defined(HAVE_SYMLINK) && !defined(MS_WINDOWS)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006475PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006476"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00006477Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006478
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006479static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006480posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006481{
Victor Stinner8c62be82010-05-06 00:08:46 +00006482 return posix_2str(args, "O&O&:symlink", symlink);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006483}
6484#endif /* HAVE_SYMLINK */
6485
Brian Curtind40e6f72010-07-08 21:39:08 +00006486#if !defined(HAVE_READLINK) && defined(MS_WINDOWS)
6487
6488PyDoc_STRVAR(win_readlink__doc__,
6489"readlink(path) -> path\n\n\
6490Return a string representing the path to which the symbolic link points.");
6491
Brian Curtind40e6f72010-07-08 21:39:08 +00006492/* Windows readlink implementation */
6493static PyObject *
6494win_readlink(PyObject *self, PyObject *args)
6495{
6496 wchar_t *path;
6497 DWORD n_bytes_returned;
6498 DWORD io_result;
Victor Stinnereb5657a2011-09-30 01:44:27 +02006499 PyObject *po, *result;
Brian Curtind40e6f72010-07-08 21:39:08 +00006500 HANDLE reparse_point_handle;
6501
6502 char target_buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
6503 REPARSE_DATA_BUFFER *rdb = (REPARSE_DATA_BUFFER *)target_buffer;
6504 wchar_t *print_name;
6505
6506 if (!PyArg_ParseTuple(args,
Victor Stinnereb5657a2011-09-30 01:44:27 +02006507 "U:readlink",
6508 &po))
6509 return NULL;
6510 path = PyUnicode_AsUnicode(po);
6511 if (path == NULL)
Brian Curtind40e6f72010-07-08 21:39:08 +00006512 return NULL;
6513
6514 /* First get a handle to the reparse point */
6515 Py_BEGIN_ALLOW_THREADS
6516 reparse_point_handle = CreateFileW(
6517 path,
6518 0,
6519 0,
6520 0,
6521 OPEN_EXISTING,
6522 FILE_FLAG_OPEN_REPARSE_POINT|FILE_FLAG_BACKUP_SEMANTICS,
6523 0);
6524 Py_END_ALLOW_THREADS
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006525
Brian Curtind40e6f72010-07-08 21:39:08 +00006526 if (reparse_point_handle==INVALID_HANDLE_VALUE)
Victor Stinnereb5657a2011-09-30 01:44:27 +02006527 return win32_error_object("readlink", po);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006528
Brian Curtind40e6f72010-07-08 21:39:08 +00006529 Py_BEGIN_ALLOW_THREADS
6530 /* New call DeviceIoControl to read the reparse point */
6531 io_result = DeviceIoControl(
6532 reparse_point_handle,
6533 FSCTL_GET_REPARSE_POINT,
6534 0, 0, /* in buffer */
6535 target_buffer, sizeof(target_buffer),
6536 &n_bytes_returned,
6537 0 /* we're not using OVERLAPPED_IO */
6538 );
6539 CloseHandle(reparse_point_handle);
6540 Py_END_ALLOW_THREADS
6541
6542 if (io_result==0)
Victor Stinnereb5657a2011-09-30 01:44:27 +02006543 return win32_error_object("readlink", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00006544
6545 if (rdb->ReparseTag != IO_REPARSE_TAG_SYMLINK)
6546 {
6547 PyErr_SetString(PyExc_ValueError,
6548 "not a symbolic link");
6549 return NULL;
6550 }
Brian Curtin74e45612010-07-09 15:58:59 +00006551 print_name = rdb->SymbolicLinkReparseBuffer.PathBuffer +
6552 rdb->SymbolicLinkReparseBuffer.PrintNameOffset;
6553
6554 result = PyUnicode_FromWideChar(print_name,
6555 rdb->SymbolicLinkReparseBuffer.PrintNameLength/2);
Brian Curtind40e6f72010-07-08 21:39:08 +00006556 return result;
6557}
6558
6559#endif /* !defined(HAVE_READLINK) && defined(MS_WINDOWS) */
6560
Brian Curtin52173d42010-12-02 18:29:18 +00006561#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtind40e6f72010-07-08 21:39:08 +00006562
6563/* Grab CreateSymbolicLinkW dynamically from kernel32 */
6564static int has_CreateSymbolicLinkW = 0;
6565static DWORD (CALLBACK *Py_CreateSymbolicLinkW)(LPWSTR, LPWSTR, DWORD);
6566static int
6567check_CreateSymbolicLinkW()
6568{
6569 HINSTANCE hKernel32;
6570 /* only recheck */
6571 if (has_CreateSymbolicLinkW)
6572 return has_CreateSymbolicLinkW;
Martin v. Löwis50590f12012-01-14 17:54:09 +01006573 hKernel32 = GetModuleHandleW(L"KERNEL32");
Brian Curtin74e45612010-07-09 15:58:59 +00006574 *(FARPROC*)&Py_CreateSymbolicLinkW = GetProcAddress(hKernel32,
6575 "CreateSymbolicLinkW");
Brian Curtind40e6f72010-07-08 21:39:08 +00006576 if (Py_CreateSymbolicLinkW)
6577 has_CreateSymbolicLinkW = 1;
6578 return has_CreateSymbolicLinkW;
6579}
6580
6581PyDoc_STRVAR(win_symlink__doc__,
6582"symlink(src, dst, target_is_directory=False)\n\n\
6583Create a symbolic link pointing to src named dst.\n\
6584target_is_directory is required if the target is to be interpreted as\n\
6585a directory.\n\
6586This function requires Windows 6.0 or greater, and raises a\n\
6587NotImplementedError otherwise.");
6588
6589static PyObject *
6590win_symlink(PyObject *self, PyObject *args, PyObject *kwargs)
6591{
6592 static char *kwlist[] = {"src", "dest", "target_is_directory", NULL};
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006593 PyObject *osrc, *odest;
6594 PyObject *usrc = NULL, *udest = NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02006595 wchar_t *wsrc, *wdest;
Brian Curtind40e6f72010-07-08 21:39:08 +00006596 int target_is_directory = 0;
6597 DWORD res;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006598
Brian Curtind40e6f72010-07-08 21:39:08 +00006599 if (!check_CreateSymbolicLinkW())
6600 {
6601 /* raise NotImplementedError */
6602 return PyErr_Format(PyExc_NotImplementedError,
6603 "CreateSymbolicLinkW not found");
6604 }
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006605 if (!PyArg_ParseTupleAndKeywords(
6606 args, kwargs, "OO|i:symlink", kwlist,
6607 &osrc, &odest, &target_is_directory))
Brian Curtind40e6f72010-07-08 21:39:08 +00006608 return NULL;
Brian Curtin3b4499c2010-12-28 14:31:47 +00006609
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006610 usrc = win32_decode_filename(osrc);
6611 if (!usrc)
6612 return NULL;
6613 udest = win32_decode_filename(odest);
6614 if (!udest)
6615 goto error;
6616
Brian Curtin3b4499c2010-12-28 14:31:47 +00006617 if (win32_can_symlink == 0)
6618 return PyErr_Format(PyExc_OSError, "symbolic link privilege not held");
6619
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006620 wsrc = PyUnicode_AsUnicode(usrc);
Victor Stinnereb5657a2011-09-30 01:44:27 +02006621 if (wsrc == NULL)
6622 goto error;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006623 wdest = PyUnicode_AsUnicode(udest);
Victor Stinnereb5657a2011-09-30 01:44:27 +02006624 if (wsrc == NULL)
6625 goto error;
6626
Brian Curtind40e6f72010-07-08 21:39:08 +00006627 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006628 res = Py_CreateSymbolicLinkW(wdest, wsrc, target_is_directory);
Brian Curtind40e6f72010-07-08 21:39:08 +00006629 Py_END_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006630
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006631 Py_DECREF(usrc);
6632 Py_DECREF(udest);
Brian Curtind40e6f72010-07-08 21:39:08 +00006633 if (!res)
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006634 return win32_error_object("symlink", osrc);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006635
Brian Curtind40e6f72010-07-08 21:39:08 +00006636 Py_INCREF(Py_None);
6637 return Py_None;
Victor Stinnereb5657a2011-09-30 01:44:27 +02006638
6639error:
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006640 Py_XDECREF(usrc);
6641 Py_XDECREF(udest);
Victor Stinnereb5657a2011-09-30 01:44:27 +02006642 return NULL;
Brian Curtind40e6f72010-07-08 21:39:08 +00006643}
Brian Curtin52173d42010-12-02 18:29:18 +00006644#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006645
6646#ifdef HAVE_TIMES
Guido van Rossumd48f2521997-12-05 22:19:34 +00006647#if defined(PYCC_VACPP) && defined(PYOS_OS2)
6648static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00006649system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00006650{
6651 ULONG value = 0;
6652
6653 Py_BEGIN_ALLOW_THREADS
6654 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
6655 Py_END_ALLOW_THREADS
6656
6657 return value;
6658}
6659
6660static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006661posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00006662{
Guido van Rossumd48f2521997-12-05 22:19:34 +00006663 /* Currently Only Uptime is Provided -- Others Later */
Victor Stinner8c62be82010-05-06 00:08:46 +00006664 return Py_BuildValue("ddddd",
6665 (double)0 /* t.tms_utime / HZ */,
6666 (double)0 /* t.tms_stime / HZ */,
6667 (double)0 /* t.tms_cutime / HZ */,
6668 (double)0 /* t.tms_cstime / HZ */,
6669 (double)system_uptime() / 1000);
Guido van Rossumd48f2521997-12-05 22:19:34 +00006670}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006671#else /* not OS2 */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00006672#define NEED_TICKS_PER_SECOND
6673static long ticks_per_second = -1;
Barry Warsaw53699e91996-12-10 23:23:01 +00006674static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006675posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00006676{
Victor Stinner8c62be82010-05-06 00:08:46 +00006677 struct tms t;
6678 clock_t c;
6679 errno = 0;
6680 c = times(&t);
6681 if (c == (clock_t) -1)
6682 return posix_error();
6683 return Py_BuildValue("ddddd",
6684 (double)t.tms_utime / ticks_per_second,
6685 (double)t.tms_stime / ticks_per_second,
6686 (double)t.tms_cutime / ticks_per_second,
6687 (double)t.tms_cstime / ticks_per_second,
6688 (double)c / ticks_per_second);
Guido van Rossum22db57e1992-04-05 14:25:30 +00006689}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006690#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006691#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006692
6693
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006694#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00006695#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00006696static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006697posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00006698{
Victor Stinner8c62be82010-05-06 00:08:46 +00006699 FILETIME create, exit, kernel, user;
6700 HANDLE hProc;
6701 hProc = GetCurrentProcess();
6702 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
6703 /* The fields of a FILETIME structure are the hi and lo part
6704 of a 64-bit value expressed in 100 nanosecond units.
6705 1e7 is one second in such units; 1e-7 the inverse.
6706 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
6707 */
6708 return Py_BuildValue(
6709 "ddddd",
6710 (double)(user.dwHighDateTime*429.4967296 +
6711 user.dwLowDateTime*1e-7),
6712 (double)(kernel.dwHighDateTime*429.4967296 +
6713 kernel.dwLowDateTime*1e-7),
6714 (double)0,
6715 (double)0,
6716 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00006717}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006718#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006719
6720#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006721PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006722"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006723Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006724#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00006725
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006726
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00006727#ifdef HAVE_GETSID
6728PyDoc_STRVAR(posix_getsid__doc__,
6729"getsid(pid) -> sid\n\n\
6730Call the system call getsid().");
6731
6732static PyObject *
6733posix_getsid(PyObject *self, PyObject *args)
6734{
Victor Stinner8c62be82010-05-06 00:08:46 +00006735 pid_t pid;
6736 int sid;
6737 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":getsid", &pid))
6738 return NULL;
6739 sid = getsid(pid);
6740 if (sid < 0)
6741 return posix_error();
6742 return PyLong_FromLong((long)sid);
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00006743}
6744#endif /* HAVE_GETSID */
6745
6746
Guido van Rossumb6775db1994-08-01 11:34:53 +00006747#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006748PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006749"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006750Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006751
Barry Warsaw53699e91996-12-10 23:23:01 +00006752static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006753posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00006754{
Victor Stinner8c62be82010-05-06 00:08:46 +00006755 if (setsid() < 0)
6756 return posix_error();
6757 Py_INCREF(Py_None);
6758 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00006759}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006760#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00006761
Guido van Rossumb6775db1994-08-01 11:34:53 +00006762#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006763PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006764"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006765Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006766
Barry Warsaw53699e91996-12-10 23:23:01 +00006767static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006768posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00006769{
Victor Stinner8c62be82010-05-06 00:08:46 +00006770 pid_t pid;
6771 int pgrp;
6772 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:setpgid", &pid, &pgrp))
6773 return NULL;
6774 if (setpgid(pid, pgrp) < 0)
6775 return posix_error();
6776 Py_INCREF(Py_None);
6777 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00006778}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006779#endif /* HAVE_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00006780
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006781
Guido van Rossumb6775db1994-08-01 11:34:53 +00006782#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006783PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006784"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006785Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006786
Barry Warsaw53699e91996-12-10 23:23:01 +00006787static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006788posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00006789{
Victor Stinner8c62be82010-05-06 00:08:46 +00006790 int fd;
6791 pid_t pgid;
6792 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
6793 return NULL;
6794 pgid = tcgetpgrp(fd);
6795 if (pgid < 0)
6796 return posix_error();
6797 return PyLong_FromPid(pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00006798}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006799#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00006800
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006801
Guido van Rossumb6775db1994-08-01 11:34:53 +00006802#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006803PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006804"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006805Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006806
Barry Warsaw53699e91996-12-10 23:23:01 +00006807static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006808posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00006809{
Victor Stinner8c62be82010-05-06 00:08:46 +00006810 int fd;
6811 pid_t pgid;
6812 if (!PyArg_ParseTuple(args, "i" _Py_PARSE_PID ":tcsetpgrp", &fd, &pgid))
6813 return NULL;
6814 if (tcsetpgrp(fd, pgid) < 0)
6815 return posix_error();
6816 Py_INCREF(Py_None);
6817 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00006818}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006819#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00006820
Guido van Rossum687dd131993-05-17 08:34:16 +00006821/* Functions acting on file descriptors */
6822
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006823PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006824"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006825Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006826
Barry Warsaw53699e91996-12-10 23:23:01 +00006827static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006828posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006829{
Victor Stinner8c62be82010-05-06 00:08:46 +00006830 PyObject *ofile;
6831 char *file;
6832 int flag;
6833 int mode = 0777;
6834 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006835
6836#ifdef MS_WINDOWS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006837 PyObject *po;
Victor Stinner26de69d2011-06-17 15:15:38 +02006838 if (PyArg_ParseTuple(args, "Ui|i:open", &po, &flag, &mode)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02006839 wchar_t *wpath = PyUnicode_AsUnicode(po);
6840 if (wpath == NULL)
6841 return NULL;
6842
Victor Stinner8c62be82010-05-06 00:08:46 +00006843 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006844 fd = _wopen(wpath, flag, mode);
Victor Stinner8c62be82010-05-06 00:08:46 +00006845 Py_END_ALLOW_THREADS
6846 if (fd < 0)
6847 return posix_error();
6848 return PyLong_FromLong((long)fd);
6849 }
6850 /* Drop the argument parsing error as narrow strings
6851 are also valid. */
6852 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006853#endif
6854
Victor Stinner26de69d2011-06-17 15:15:38 +02006855 if (!PyArg_ParseTuple(args, "O&i|i:open",
Victor Stinner8c62be82010-05-06 00:08:46 +00006856 PyUnicode_FSConverter, &ofile,
6857 &flag, &mode))
6858 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006859#ifdef MS_WINDOWS
6860 if (win32_warn_bytes_api()) {
6861 Py_DECREF(ofile);
6862 return NULL;
6863 }
6864#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006865 file = PyBytes_AsString(ofile);
6866 Py_BEGIN_ALLOW_THREADS
6867 fd = open(file, flag, mode);
6868 Py_END_ALLOW_THREADS
6869 if (fd < 0)
6870 return posix_error_with_allocated_filename(ofile);
6871 Py_DECREF(ofile);
6872 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006873}
6874
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006875
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006876PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006877"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006878Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006879
Barry Warsaw53699e91996-12-10 23:23:01 +00006880static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006881posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006882{
Victor Stinner8c62be82010-05-06 00:08:46 +00006883 int fd, res;
6884 if (!PyArg_ParseTuple(args, "i:close", &fd))
6885 return NULL;
6886 if (!_PyVerify_fd(fd))
6887 return posix_error();
6888 Py_BEGIN_ALLOW_THREADS
6889 res = close(fd);
6890 Py_END_ALLOW_THREADS
6891 if (res < 0)
6892 return posix_error();
6893 Py_INCREF(Py_None);
6894 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006895}
6896
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006897
Victor Stinner8c62be82010-05-06 00:08:46 +00006898PyDoc_STRVAR(posix_closerange__doc__,
Christian Heimesfdab48e2008-01-20 09:06:41 +00006899"closerange(fd_low, fd_high)\n\n\
6900Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
6901
6902static PyObject *
6903posix_closerange(PyObject *self, PyObject *args)
6904{
Victor Stinner8c62be82010-05-06 00:08:46 +00006905 int fd_from, fd_to, i;
6906 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
6907 return NULL;
6908 Py_BEGIN_ALLOW_THREADS
6909 for (i = fd_from; i < fd_to; i++)
6910 if (_PyVerify_fd(i))
6911 close(i);
6912 Py_END_ALLOW_THREADS
6913 Py_RETURN_NONE;
Christian Heimesfdab48e2008-01-20 09:06:41 +00006914}
6915
6916
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006917PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006918"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006919Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006920
Barry Warsaw53699e91996-12-10 23:23:01 +00006921static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006922posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006923{
Victor Stinner8c62be82010-05-06 00:08:46 +00006924 int fd;
6925 if (!PyArg_ParseTuple(args, "i:dup", &fd))
6926 return NULL;
6927 if (!_PyVerify_fd(fd))
6928 return posix_error();
Victor Stinner8c62be82010-05-06 00:08:46 +00006929 fd = dup(fd);
Victor Stinner8c62be82010-05-06 00:08:46 +00006930 if (fd < 0)
6931 return posix_error();
6932 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006933}
6934
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006935
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006936PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00006937"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006938Duplicate file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006939
Barry Warsaw53699e91996-12-10 23:23:01 +00006940static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006941posix_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006942{
Victor Stinner8c62be82010-05-06 00:08:46 +00006943 int fd, fd2, res;
6944 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
6945 return NULL;
6946 if (!_PyVerify_fd_dup2(fd, fd2))
6947 return posix_error();
Victor Stinner8c62be82010-05-06 00:08:46 +00006948 res = dup2(fd, fd2);
Victor Stinner8c62be82010-05-06 00:08:46 +00006949 if (res < 0)
6950 return posix_error();
6951 Py_INCREF(Py_None);
6952 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006953}
6954
Ross Lagerwall7807c352011-03-17 20:20:30 +02006955#ifdef HAVE_LOCKF
6956PyDoc_STRVAR(posix_lockf__doc__,
6957"lockf(fd, cmd, len)\n\n\
6958Apply, test or remove a POSIX lock on an open file descriptor.\n\n\
6959fd is an open file descriptor.\n\
6960cmd specifies the command to use - one of F_LOCK, F_TLOCK, F_ULOCK or\n\
6961F_TEST.\n\
6962len specifies the section of the file to lock.");
6963
6964static PyObject *
6965posix_lockf(PyObject *self, PyObject *args)
6966{
6967 int fd, cmd, res;
6968 off_t len;
6969 if (!PyArg_ParseTuple(args, "iiO&:lockf",
6970 &fd, &cmd, _parse_off_t, &len))
6971 return NULL;
6972
6973 Py_BEGIN_ALLOW_THREADS
6974 res = lockf(fd, cmd, len);
6975 Py_END_ALLOW_THREADS
6976
6977 if (res < 0)
6978 return posix_error();
6979
6980 Py_RETURN_NONE;
6981}
6982#endif
6983
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006984
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006985PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006986"lseek(fd, pos, how) -> newpos\n\n\
Victor Stinnere83f8992011-12-17 23:15:09 +01006987Set the current position of a file descriptor.\n\
6988Return the new cursor position in bytes, starting from the beginning.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006989
Barry Warsaw53699e91996-12-10 23:23:01 +00006990static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006991posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006992{
Victor Stinner8c62be82010-05-06 00:08:46 +00006993 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006994#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00006995 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006996#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006997 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006998#endif
Ross Lagerwall8e749672011-03-17 21:54:07 +02006999 PyObject *posobj;
7000 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Victor Stinner8c62be82010-05-06 00:08:46 +00007001 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00007002#ifdef SEEK_SET
Victor Stinner8c62be82010-05-06 00:08:46 +00007003 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
7004 switch (how) {
7005 case 0: how = SEEK_SET; break;
7006 case 1: how = SEEK_CUR; break;
7007 case 2: how = SEEK_END; break;
7008 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007009#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00007010
Ross Lagerwall8e749672011-03-17 21:54:07 +02007011#if !defined(HAVE_LARGEFILE_SUPPORT)
7012 pos = PyLong_AsLong(posobj);
7013#else
7014 pos = PyLong_AsLongLong(posobj);
7015#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00007016 if (PyErr_Occurred())
7017 return NULL;
Guido van Rossum94f6f721999-01-06 18:42:14 +00007018
Victor Stinner8c62be82010-05-06 00:08:46 +00007019 if (!_PyVerify_fd(fd))
7020 return posix_error();
7021 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007022#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00007023 res = _lseeki64(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00007024#else
Victor Stinner8c62be82010-05-06 00:08:46 +00007025 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00007026#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00007027 Py_END_ALLOW_THREADS
7028 if (res < 0)
7029 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00007030
7031#if !defined(HAVE_LARGEFILE_SUPPORT)
Victor Stinner8c62be82010-05-06 00:08:46 +00007032 return PyLong_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00007033#else
Victor Stinner8c62be82010-05-06 00:08:46 +00007034 return PyLong_FromLongLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00007035#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00007036}
7037
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007038
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007039PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007040"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007041Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007042
Barry Warsaw53699e91996-12-10 23:23:01 +00007043static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007044posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00007045{
Victor Stinner8c62be82010-05-06 00:08:46 +00007046 int fd, size;
7047 Py_ssize_t n;
7048 PyObject *buffer;
7049 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
7050 return NULL;
7051 if (size < 0) {
7052 errno = EINVAL;
7053 return posix_error();
7054 }
7055 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
7056 if (buffer == NULL)
7057 return NULL;
Stefan Krah99439262010-11-26 12:58:05 +00007058 if (!_PyVerify_fd(fd)) {
7059 Py_DECREF(buffer);
Victor Stinner8c62be82010-05-06 00:08:46 +00007060 return posix_error();
Stefan Krah99439262010-11-26 12:58:05 +00007061 }
Victor Stinner8c62be82010-05-06 00:08:46 +00007062 Py_BEGIN_ALLOW_THREADS
7063 n = read(fd, PyBytes_AS_STRING(buffer), size);
7064 Py_END_ALLOW_THREADS
7065 if (n < 0) {
7066 Py_DECREF(buffer);
7067 return posix_error();
7068 }
7069 if (n != size)
7070 _PyBytes_Resize(&buffer, n);
7071 return buffer;
Guido van Rossum687dd131993-05-17 08:34:16 +00007072}
7073
Ross Lagerwall7807c352011-03-17 20:20:30 +02007074#if (defined(HAVE_SENDFILE) && (defined(__FreeBSD__) || defined(__DragonFly__) \
7075 || defined(__APPLE__))) || defined(HAVE_READV) || defined(HAVE_WRITEV)
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007076static Py_ssize_t
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007077iov_setup(struct iovec **iov, Py_buffer **buf, PyObject *seq, int cnt, int type)
7078{
7079 int i, j;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007080 Py_ssize_t blen, total = 0;
7081
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007082 *iov = PyMem_New(struct iovec, cnt);
7083 if (*iov == NULL) {
7084 PyErr_NoMemory();
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007085 return total;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007086 }
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007087
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007088 *buf = PyMem_New(Py_buffer, cnt);
7089 if (*buf == NULL) {
7090 PyMem_Del(*iov);
7091 PyErr_NoMemory();
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007092 return total;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007093 }
7094
7095 for (i = 0; i < cnt; i++) {
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02007096 PyObject *item = PySequence_GetItem(seq, i);
7097 if (item == NULL)
7098 goto fail;
7099 if (PyObject_GetBuffer(item, &(*buf)[i], type) == -1) {
7100 Py_DECREF(item);
7101 goto fail;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007102 }
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02007103 Py_DECREF(item);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007104 (*iov)[i].iov_base = (*buf)[i].buf;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007105 blen = (*buf)[i].len;
7106 (*iov)[i].iov_len = blen;
7107 total += blen;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007108 }
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007109 return total;
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02007110
7111fail:
7112 PyMem_Del(*iov);
7113 for (j = 0; j < i; j++) {
7114 PyBuffer_Release(&(*buf)[j]);
7115 }
7116 PyMem_Del(*buf);
7117 return 0;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007118}
7119
7120static void
7121iov_cleanup(struct iovec *iov, Py_buffer *buf, int cnt)
7122{
7123 int i;
7124 PyMem_Del(iov);
7125 for (i = 0; i < cnt; i++) {
7126 PyBuffer_Release(&buf[i]);
7127 }
7128 PyMem_Del(buf);
7129}
7130#endif
7131
Ross Lagerwall7807c352011-03-17 20:20:30 +02007132#ifdef HAVE_READV
7133PyDoc_STRVAR(posix_readv__doc__,
7134"readv(fd, buffers) -> bytesread\n\n\
7135Read from a file descriptor into a number of writable buffers. buffers\n\
7136is an arbitrary sequence of writable buffers.\n\
7137Returns the total number of bytes read.");
7138
7139static PyObject *
7140posix_readv(PyObject *self, PyObject *args)
7141{
7142 int fd, cnt;
7143 Py_ssize_t n;
7144 PyObject *seq;
7145 struct iovec *iov;
7146 Py_buffer *buf;
7147
7148 if (!PyArg_ParseTuple(args, "iO:readv", &fd, &seq))
7149 return NULL;
7150 if (!PySequence_Check(seq)) {
7151 PyErr_SetString(PyExc_TypeError,
7152 "readv() arg 2 must be a sequence");
7153 return NULL;
7154 }
7155 cnt = PySequence_Size(seq);
7156
7157 if (!iov_setup(&iov, &buf, seq, cnt, PyBUF_WRITABLE))
7158 return NULL;
7159
7160 Py_BEGIN_ALLOW_THREADS
7161 n = readv(fd, iov, cnt);
7162 Py_END_ALLOW_THREADS
7163
7164 iov_cleanup(iov, buf, cnt);
7165 return PyLong_FromSsize_t(n);
7166}
7167#endif
7168
7169#ifdef HAVE_PREAD
7170PyDoc_STRVAR(posix_pread__doc__,
7171"pread(fd, buffersize, offset) -> string\n\n\
7172Read from a file descriptor, fd, at a position of offset. It will read up\n\
7173to buffersize number of bytes. The file offset remains unchanged.");
7174
7175static PyObject *
7176posix_pread(PyObject *self, PyObject *args)
7177{
7178 int fd, size;
7179 off_t offset;
7180 Py_ssize_t n;
7181 PyObject *buffer;
7182 if (!PyArg_ParseTuple(args, "iiO&:pread", &fd, &size, _parse_off_t, &offset))
7183 return NULL;
7184
7185 if (size < 0) {
7186 errno = EINVAL;
7187 return posix_error();
7188 }
7189 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
7190 if (buffer == NULL)
7191 return NULL;
7192 if (!_PyVerify_fd(fd)) {
7193 Py_DECREF(buffer);
7194 return posix_error();
7195 }
7196 Py_BEGIN_ALLOW_THREADS
7197 n = pread(fd, PyBytes_AS_STRING(buffer), size, offset);
7198 Py_END_ALLOW_THREADS
7199 if (n < 0) {
7200 Py_DECREF(buffer);
7201 return posix_error();
7202 }
7203 if (n != size)
7204 _PyBytes_Resize(&buffer, n);
7205 return buffer;
7206}
7207#endif
7208
7209PyDoc_STRVAR(posix_write__doc__,
7210"write(fd, string) -> byteswritten\n\n\
7211Write a string to a file descriptor.");
7212
7213static PyObject *
7214posix_write(PyObject *self, PyObject *args)
7215{
7216 Py_buffer pbuf;
7217 int fd;
7218 Py_ssize_t size, len;
7219
7220 if (!PyArg_ParseTuple(args, "iy*:write", &fd, &pbuf))
7221 return NULL;
7222 if (!_PyVerify_fd(fd)) {
7223 PyBuffer_Release(&pbuf);
7224 return posix_error();
7225 }
7226 len = pbuf.len;
7227 Py_BEGIN_ALLOW_THREADS
7228#if defined(MS_WIN64) || defined(MS_WINDOWS)
7229 if (len > INT_MAX)
7230 len = INT_MAX;
7231 size = write(fd, pbuf.buf, (int)len);
7232#else
7233 size = write(fd, pbuf.buf, len);
7234#endif
7235 Py_END_ALLOW_THREADS
7236 PyBuffer_Release(&pbuf);
7237 if (size < 0)
7238 return posix_error();
7239 return PyLong_FromSsize_t(size);
7240}
7241
7242#ifdef HAVE_SENDFILE
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007243PyDoc_STRVAR(posix_sendfile__doc__,
7244"sendfile(out, in, offset, nbytes) -> byteswritten\n\
7245sendfile(out, in, offset, nbytes, headers=None, trailers=None, flags=0)\n\
7246 -> byteswritten\n\
7247Copy nbytes bytes from file descriptor in to file descriptor out.");
7248
7249static PyObject *
7250posix_sendfile(PyObject *self, PyObject *args, PyObject *kwdict)
7251{
7252 int in, out;
7253 Py_ssize_t ret;
7254 off_t offset;
7255
7256#if defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__)
7257#ifndef __APPLE__
7258 Py_ssize_t len;
7259#endif
7260 PyObject *headers = NULL, *trailers = NULL;
7261 Py_buffer *hbuf, *tbuf;
7262 off_t sbytes;
7263 struct sf_hdtr sf;
7264 int flags = 0;
7265 sf.headers = NULL;
7266 sf.trailers = NULL;
Benjamin Petersond8a43b42011-02-26 21:35:16 +00007267 static char *keywords[] = {"out", "in",
7268 "offset", "count",
7269 "headers", "trailers", "flags", NULL};
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007270
7271#ifdef __APPLE__
7272 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiO&O&|OOi:sendfile",
Georg Brandla391b112011-02-25 15:23:18 +00007273 keywords, &out, &in, _parse_off_t, &offset, _parse_off_t, &sbytes,
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007274#else
7275 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiO&n|OOi:sendfile",
Georg Brandla391b112011-02-25 15:23:18 +00007276 keywords, &out, &in, _parse_off_t, &offset, &len,
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007277#endif
7278 &headers, &trailers, &flags))
7279 return NULL;
7280 if (headers != NULL) {
7281 if (!PySequence_Check(headers)) {
7282 PyErr_SetString(PyExc_TypeError,
7283 "sendfile() headers must be a sequence or None");
7284 return NULL;
7285 } else {
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007286 Py_ssize_t i = 0; /* Avoid uninitialized warning */
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007287 sf.hdr_cnt = PySequence_Size(headers);
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007288 if (sf.hdr_cnt > 0 &&
7289 !(i = iov_setup(&(sf.headers), &hbuf,
7290 headers, sf.hdr_cnt, PyBUF_SIMPLE)))
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007291 return NULL;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007292#ifdef __APPLE__
7293 sbytes += i;
7294#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007295 }
7296 }
7297 if (trailers != NULL) {
7298 if (!PySequence_Check(trailers)) {
7299 PyErr_SetString(PyExc_TypeError,
7300 "sendfile() trailers must be a sequence or None");
7301 return NULL;
7302 } else {
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007303 Py_ssize_t i = 0; /* Avoid uninitialized warning */
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007304 sf.trl_cnt = PySequence_Size(trailers);
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007305 if (sf.trl_cnt > 0 &&
7306 !(i = iov_setup(&(sf.trailers), &tbuf,
7307 trailers, sf.trl_cnt, PyBUF_SIMPLE)))
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007308 return NULL;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007309#ifdef __APPLE__
7310 sbytes += i;
7311#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007312 }
7313 }
7314
7315 Py_BEGIN_ALLOW_THREADS
7316#ifdef __APPLE__
7317 ret = sendfile(in, out, offset, &sbytes, &sf, flags);
7318#else
7319 ret = sendfile(in, out, offset, len, &sf, &sbytes, flags);
7320#endif
7321 Py_END_ALLOW_THREADS
7322
7323 if (sf.headers != NULL)
7324 iov_cleanup(sf.headers, hbuf, sf.hdr_cnt);
7325 if (sf.trailers != NULL)
7326 iov_cleanup(sf.trailers, tbuf, sf.trl_cnt);
7327
7328 if (ret < 0) {
7329 if ((errno == EAGAIN) || (errno == EBUSY)) {
7330 if (sbytes != 0) {
7331 // some data has been sent
7332 goto done;
7333 }
7334 else {
7335 // no data has been sent; upper application is supposed
7336 // to retry on EAGAIN or EBUSY
7337 return posix_error();
7338 }
7339 }
7340 return posix_error();
7341 }
7342 goto done;
7343
7344done:
7345 #if !defined(HAVE_LARGEFILE_SUPPORT)
7346 return Py_BuildValue("l", sbytes);
7347 #else
7348 return Py_BuildValue("L", sbytes);
7349 #endif
7350
7351#else
7352 Py_ssize_t count;
7353 PyObject *offobj;
7354 static char *keywords[] = {"out", "in",
7355 "offset", "count", NULL};
7356 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiOn:sendfile",
7357 keywords, &out, &in, &offobj, &count))
7358 return NULL;
7359#ifdef linux
7360 if (offobj == Py_None) {
7361 Py_BEGIN_ALLOW_THREADS
7362 ret = sendfile(out, in, NULL, count);
7363 Py_END_ALLOW_THREADS
7364 if (ret < 0)
7365 return posix_error();
Giampaolo Rodola'ff1a7352011-04-19 09:47:16 +02007366 return Py_BuildValue("n", ret);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007367 }
7368#endif
Antoine Pitroudcc20b82011-02-26 13:38:35 +00007369 if (!_parse_off_t(offobj, &offset))
7370 return NULL;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007371 Py_BEGIN_ALLOW_THREADS
7372 ret = sendfile(out, in, &offset, count);
7373 Py_END_ALLOW_THREADS
7374 if (ret < 0)
7375 return posix_error();
7376 return Py_BuildValue("n", ret);
7377#endif
7378}
7379#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007380
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007381PyDoc_STRVAR(posix_fstat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01007382"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007383Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007384
Barry Warsaw53699e91996-12-10 23:23:01 +00007385static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01007386posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00007387{
Victor Stinner8c62be82010-05-06 00:08:46 +00007388 int fd;
7389 STRUCT_STAT st;
7390 int res;
Victor Stinner4195b5c2012-02-08 23:03:19 +01007391 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
Victor Stinner8c62be82010-05-06 00:08:46 +00007392 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00007393#ifdef __VMS
Victor Stinner8c62be82010-05-06 00:08:46 +00007394 /* on OpenVMS we must ensure that all bytes are written to the file */
7395 fsync(fd);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00007396#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00007397 if (!_PyVerify_fd(fd))
7398 return posix_error();
7399 Py_BEGIN_ALLOW_THREADS
7400 res = FSTAT(fd, &st);
7401 Py_END_ALLOW_THREADS
7402 if (res != 0) {
Martin v. Löwis14694662006-02-03 12:54:16 +00007403#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00007404 return win32_error("fstat", NULL);
Martin v. Löwis14694662006-02-03 12:54:16 +00007405#else
Victor Stinner8c62be82010-05-06 00:08:46 +00007406 return posix_error();
Martin v. Löwis14694662006-02-03 12:54:16 +00007407#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00007408 }
Tim Peters5aa91602002-01-30 05:46:57 +00007409
Victor Stinner4195b5c2012-02-08 23:03:19 +01007410 return _pystat_fromstructstat(&st);
Guido van Rossum687dd131993-05-17 08:34:16 +00007411}
7412
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007413PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007414"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00007415Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007416connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00007417
7418static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00007419posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00007420{
Victor Stinner8c62be82010-05-06 00:08:46 +00007421 int fd;
7422 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
7423 return NULL;
7424 if (!_PyVerify_fd(fd))
7425 return PyBool_FromLong(0);
7426 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00007427}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007428
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007429#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007430PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007431"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007432Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007433
Barry Warsaw53699e91996-12-10 23:23:01 +00007434static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00007435posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00007436{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007437#if defined(PYOS_OS2)
7438 HFILE read, write;
7439 APIRET rc;
7440
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007441 rc = DosCreatePipe( &read, &write, 4096);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007442 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007443 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007444
7445 return Py_BuildValue("(ii)", read, write);
7446#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007447#if !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00007448 int fds[2];
7449 int res;
Victor Stinner8c62be82010-05-06 00:08:46 +00007450 res = pipe(fds);
Victor Stinner8c62be82010-05-06 00:08:46 +00007451 if (res != 0)
7452 return posix_error();
7453 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007454#else /* MS_WINDOWS */
Victor Stinner8c62be82010-05-06 00:08:46 +00007455 HANDLE read, write;
7456 int read_fd, write_fd;
7457 BOOL ok;
Victor Stinner8c62be82010-05-06 00:08:46 +00007458 ok = CreatePipe(&read, &write, NULL, 0);
Victor Stinner8c62be82010-05-06 00:08:46 +00007459 if (!ok)
7460 return win32_error("CreatePipe", NULL);
7461 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
7462 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
7463 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007464#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007465#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00007466}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007467#endif /* HAVE_PIPE */
7468
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007469#ifdef HAVE_PIPE2
7470PyDoc_STRVAR(posix_pipe2__doc__,
Charles-François Natali368f34b2011-06-06 19:49:47 +02007471"pipe2(flags) -> (read_end, write_end)\n\n\
7472Create a pipe with flags set atomically.\n\
7473flags can be constructed by ORing together one or more of these values:\n\
7474O_NONBLOCK, O_CLOEXEC.\n\
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007475");
7476
7477static PyObject *
Charles-François Natali368f34b2011-06-06 19:49:47 +02007478posix_pipe2(PyObject *self, PyObject *arg)
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007479{
Charles-François Natali368f34b2011-06-06 19:49:47 +02007480 int flags;
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007481 int fds[2];
7482 int res;
7483
Charles-François Natali368f34b2011-06-06 19:49:47 +02007484 flags = PyLong_AsLong(arg);
7485 if (flags == -1 && PyErr_Occurred())
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007486 return NULL;
7487
7488 res = pipe2(fds, flags);
7489 if (res != 0)
7490 return posix_error();
7491 return Py_BuildValue("(ii)", fds[0], fds[1]);
7492}
7493#endif /* HAVE_PIPE2 */
7494
Ross Lagerwall7807c352011-03-17 20:20:30 +02007495#ifdef HAVE_WRITEV
7496PyDoc_STRVAR(posix_writev__doc__,
7497"writev(fd, buffers) -> byteswritten\n\n\
7498Write the contents of buffers to a file descriptor, where buffers is an\n\
7499arbitrary sequence of buffers.\n\
7500Returns the total bytes written.");
7501
7502static PyObject *
7503posix_writev(PyObject *self, PyObject *args)
7504{
7505 int fd, cnt;
7506 Py_ssize_t res;
7507 PyObject *seq;
7508 struct iovec *iov;
7509 Py_buffer *buf;
7510 if (!PyArg_ParseTuple(args, "iO:writev", &fd, &seq))
7511 return NULL;
7512 if (!PySequence_Check(seq)) {
7513 PyErr_SetString(PyExc_TypeError,
7514 "writev() arg 2 must be a sequence");
7515 return NULL;
7516 }
7517 cnt = PySequence_Size(seq);
7518
7519 if (!iov_setup(&iov, &buf, seq, cnt, PyBUF_SIMPLE)) {
7520 return NULL;
7521 }
7522
7523 Py_BEGIN_ALLOW_THREADS
7524 res = writev(fd, iov, cnt);
7525 Py_END_ALLOW_THREADS
7526
7527 iov_cleanup(iov, buf, cnt);
7528 return PyLong_FromSsize_t(res);
7529}
7530#endif
7531
7532#ifdef HAVE_PWRITE
7533PyDoc_STRVAR(posix_pwrite__doc__,
7534"pwrite(fd, string, offset) -> byteswritten\n\n\
7535Write string to a file descriptor, fd, from offset, leaving the file\n\
7536offset unchanged.");
7537
7538static PyObject *
7539posix_pwrite(PyObject *self, PyObject *args)
7540{
7541 Py_buffer pbuf;
7542 int fd;
7543 off_t offset;
7544 Py_ssize_t size;
7545
7546 if (!PyArg_ParseTuple(args, "iy*O&:pwrite", &fd, &pbuf, _parse_off_t, &offset))
7547 return NULL;
7548
7549 if (!_PyVerify_fd(fd)) {
7550 PyBuffer_Release(&pbuf);
7551 return posix_error();
7552 }
7553 Py_BEGIN_ALLOW_THREADS
7554 size = pwrite(fd, pbuf.buf, (size_t)pbuf.len, offset);
7555 Py_END_ALLOW_THREADS
7556 PyBuffer_Release(&pbuf);
7557 if (size < 0)
7558 return posix_error();
7559 return PyLong_FromSsize_t(size);
7560}
7561#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007562
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007563#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007564PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00007565"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007566Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007567
Barry Warsaw53699e91996-12-10 23:23:01 +00007568static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007569posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007570{
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007571 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00007572 char *filename;
7573 int mode = 0666;
7574 int res;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007575 if (!PyArg_ParseTuple(args, "O&|i:mkfifo", PyUnicode_FSConverter, &opath,
7576 &mode))
Victor Stinner8c62be82010-05-06 00:08:46 +00007577 return NULL;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007578 filename = PyBytes_AS_STRING(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007579 Py_BEGIN_ALLOW_THREADS
7580 res = mkfifo(filename, mode);
7581 Py_END_ALLOW_THREADS
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007582 Py_DECREF(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007583 if (res < 0)
7584 return posix_error();
7585 Py_INCREF(Py_None);
7586 return Py_None;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007587}
7588#endif
7589
7590
Neal Norwitz11690112002-07-30 01:08:28 +00007591#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007592PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00007593"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007594Create a filesystem node (file, device special file or named pipe)\n\
7595named filename. mode specifies both the permissions to use and the\n\
7596type of node to be created, being combined (bitwise OR) with one of\n\
7597S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007598device defines the newly created device special file (probably using\n\
7599os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007600
7601
7602static PyObject *
7603posix_mknod(PyObject *self, PyObject *args)
7604{
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007605 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00007606 char *filename;
7607 int mode = 0600;
7608 int device = 0;
7609 int res;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007610 if (!PyArg_ParseTuple(args, "O&|ii:mknod", PyUnicode_FSConverter, &opath,
7611 &mode, &device))
Victor Stinner8c62be82010-05-06 00:08:46 +00007612 return NULL;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007613 filename = PyBytes_AS_STRING(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007614 Py_BEGIN_ALLOW_THREADS
7615 res = mknod(filename, mode, device);
7616 Py_END_ALLOW_THREADS
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007617 Py_DECREF(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007618 if (res < 0)
7619 return posix_error();
7620 Py_INCREF(Py_None);
7621 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007622}
7623#endif
7624
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007625#ifdef HAVE_DEVICE_MACROS
7626PyDoc_STRVAR(posix_major__doc__,
7627"major(device) -> major number\n\
7628Extracts a device major number from a raw device number.");
7629
7630static PyObject *
7631posix_major(PyObject *self, PyObject *args)
7632{
Victor Stinner8c62be82010-05-06 00:08:46 +00007633 int device;
7634 if (!PyArg_ParseTuple(args, "i:major", &device))
7635 return NULL;
7636 return PyLong_FromLong((long)major(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007637}
7638
7639PyDoc_STRVAR(posix_minor__doc__,
7640"minor(device) -> minor number\n\
7641Extracts a device minor number from a raw device number.");
7642
7643static PyObject *
7644posix_minor(PyObject *self, PyObject *args)
7645{
Victor Stinner8c62be82010-05-06 00:08:46 +00007646 int device;
7647 if (!PyArg_ParseTuple(args, "i:minor", &device))
7648 return NULL;
7649 return PyLong_FromLong((long)minor(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007650}
7651
7652PyDoc_STRVAR(posix_makedev__doc__,
7653"makedev(major, minor) -> device number\n\
7654Composes a raw device number from the major and minor device numbers.");
7655
7656static PyObject *
7657posix_makedev(PyObject *self, PyObject *args)
7658{
Victor Stinner8c62be82010-05-06 00:08:46 +00007659 int major, minor;
7660 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
7661 return NULL;
7662 return PyLong_FromLong((long)makedev(major, minor));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007663}
7664#endif /* device macros */
7665
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007666
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007667#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007668PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007669"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007670Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007671
Barry Warsaw53699e91996-12-10 23:23:01 +00007672static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007673posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007674{
Victor Stinner8c62be82010-05-06 00:08:46 +00007675 int fd;
7676 off_t length;
7677 int res;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007678
Ross Lagerwall7807c352011-03-17 20:20:30 +02007679 if (!PyArg_ParseTuple(args, "iO&:ftruncate", &fd, _parse_off_t, &length))
Victor Stinner8c62be82010-05-06 00:08:46 +00007680 return NULL;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007681
Victor Stinner8c62be82010-05-06 00:08:46 +00007682 Py_BEGIN_ALLOW_THREADS
7683 res = ftruncate(fd, length);
7684 Py_END_ALLOW_THREADS
7685 if (res < 0)
7686 return posix_error();
7687 Py_INCREF(Py_None);
7688 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007689}
7690#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00007691
Ross Lagerwall7807c352011-03-17 20:20:30 +02007692#ifdef HAVE_TRUNCATE
7693PyDoc_STRVAR(posix_truncate__doc__,
7694"truncate(path, length)\n\n\
7695Truncate the file given by path to length bytes.");
7696
7697static PyObject *
7698posix_truncate(PyObject *self, PyObject *args)
7699{
7700 PyObject *opath;
7701 const char *path;
7702 off_t length;
7703 int res;
7704
7705 if (!PyArg_ParseTuple(args, "O&O&:truncate",
7706 PyUnicode_FSConverter, &opath, _parse_off_t, &length))
7707 return NULL;
7708 path = PyBytes_AsString(opath);
7709
7710 Py_BEGIN_ALLOW_THREADS
7711 res = truncate(path, length);
7712 Py_END_ALLOW_THREADS
7713 Py_DECREF(opath);
7714 if (res < 0)
7715 return posix_error();
7716 Py_RETURN_NONE;
7717}
7718#endif
7719
7720#ifdef HAVE_POSIX_FALLOCATE
7721PyDoc_STRVAR(posix_posix_fallocate__doc__,
7722"posix_fallocate(fd, offset, len)\n\n\
7723Ensures that enough disk space is allocated for the file specified by fd\n\
7724starting from offset and continuing for len bytes.");
7725
7726static PyObject *
7727posix_posix_fallocate(PyObject *self, PyObject *args)
7728{
7729 off_t len, offset;
7730 int res, fd;
7731
7732 if (!PyArg_ParseTuple(args, "iO&O&:posix_fallocate",
7733 &fd, _parse_off_t, &offset, _parse_off_t, &len))
7734 return NULL;
7735
7736 Py_BEGIN_ALLOW_THREADS
7737 res = posix_fallocate(fd, offset, len);
7738 Py_END_ALLOW_THREADS
7739 if (res != 0) {
7740 errno = res;
7741 return posix_error();
7742 }
7743 Py_RETURN_NONE;
7744}
7745#endif
7746
7747#ifdef HAVE_POSIX_FADVISE
7748PyDoc_STRVAR(posix_posix_fadvise__doc__,
7749"posix_fadvise(fd, offset, len, advice)\n\n\
7750Announces an intention to access data in a specific pattern thus allowing\n\
7751the kernel to make optimizations.\n\
7752The advice applies to the region of the file specified by fd starting at\n\
7753offset and continuing for len bytes.\n\
7754advice is one of POSIX_FADV_NORMAL, POSIX_FADV_SEQUENTIAL,\n\
7755POSIX_FADV_RANDOM, POSIX_FADV_NOREUSE, POSIX_FADV_WILLNEED or\n\
7756POSIX_FADV_DONTNEED.");
7757
7758static PyObject *
7759posix_posix_fadvise(PyObject *self, PyObject *args)
7760{
7761 off_t len, offset;
7762 int res, fd, advice;
7763
7764 if (!PyArg_ParseTuple(args, "iO&O&i:posix_fadvise",
7765 &fd, _parse_off_t, &offset, _parse_off_t, &len, &advice))
7766 return NULL;
7767
7768 Py_BEGIN_ALLOW_THREADS
7769 res = posix_fadvise(fd, offset, len, advice);
7770 Py_END_ALLOW_THREADS
7771 if (res != 0) {
7772 errno = res;
7773 return posix_error();
7774 }
7775 Py_RETURN_NONE;
7776}
7777#endif
7778
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007779#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007780PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007781"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007782Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007783
Fred Drake762e2061999-08-26 17:23:54 +00007784/* Save putenv() parameters as values here, so we can collect them when they
7785 * get re-set with another call for the same key. */
7786static PyObject *posix_putenv_garbage;
7787
Tim Peters5aa91602002-01-30 05:46:57 +00007788static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007789posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007790{
Victor Stinner84ae1182010-05-06 22:05:07 +00007791 PyObject *newstr = NULL;
Thomas Hellerf78f12a2007-11-08 19:33:05 +00007792#ifdef MS_WINDOWS
Victor Stinner65170952011-11-22 22:16:17 +01007793 PyObject *os1, *os2;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007794 wchar_t *newenv;
Victor Stinner65170952011-11-22 22:16:17 +01007795
Victor Stinner8c62be82010-05-06 00:08:46 +00007796 if (!PyArg_ParseTuple(args,
Victor Stinner9d3b93b2011-11-22 02:27:30 +01007797 "UU:putenv",
Victor Stinner65170952011-11-22 22:16:17 +01007798 &os1, &os2))
Victor Stinner8c62be82010-05-06 00:08:46 +00007799 return NULL;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007800
Victor Stinner65170952011-11-22 22:16:17 +01007801 newstr = PyUnicode_FromFormat("%U=%U", os1, os2);
Victor Stinner84ae1182010-05-06 22:05:07 +00007802 if (newstr == NULL) {
7803 PyErr_NoMemory();
7804 goto error;
7805 }
Victor Stinner65170952011-11-22 22:16:17 +01007806 if (_MAX_ENV < PyUnicode_GET_LENGTH(newstr)) {
7807 PyErr_Format(PyExc_ValueError,
7808 "the environment variable is longer than %u characters",
7809 _MAX_ENV);
7810 goto error;
7811 }
7812
Victor Stinner8c62be82010-05-06 00:08:46 +00007813 newenv = PyUnicode_AsUnicode(newstr);
Victor Stinnereb5657a2011-09-30 01:44:27 +02007814 if (newenv == NULL)
7815 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00007816 if (_wputenv(newenv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007817 posix_error();
Victor Stinner84ae1182010-05-06 22:05:07 +00007818 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00007819 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00007820#else
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007821 PyObject *os1, *os2;
7822 char *s1, *s2;
7823 char *newenv;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007824
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007825 if (!PyArg_ParseTuple(args,
7826 "O&O&:putenv",
7827 PyUnicode_FSConverter, &os1,
7828 PyUnicode_FSConverter, &os2))
7829 return NULL;
7830 s1 = PyBytes_AsString(os1);
7831 s2 = PyBytes_AsString(os2);
Guido van Rossumad0ee831995-03-01 10:34:45 +00007832
Victor Stinner65170952011-11-22 22:16:17 +01007833 newstr = PyBytes_FromFormat("%s=%s", s1, s2);
Victor Stinner9d3b93b2011-11-22 02:27:30 +01007834 if (newstr == NULL) {
7835 PyErr_NoMemory();
7836 goto error;
7837 }
7838
Victor Stinner8c62be82010-05-06 00:08:46 +00007839 newenv = PyBytes_AS_STRING(newstr);
Victor Stinner8c62be82010-05-06 00:08:46 +00007840 if (putenv(newenv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007841 posix_error();
Victor Stinner84ae1182010-05-06 22:05:07 +00007842 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00007843 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00007844#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007845
Victor Stinner8c62be82010-05-06 00:08:46 +00007846 /* Install the first arg and newstr in posix_putenv_garbage;
7847 * this will cause previous value to be collected. This has to
7848 * happen after the real putenv() call because the old value
7849 * was still accessible until then. */
Victor Stinner65170952011-11-22 22:16:17 +01007850 if (PyDict_SetItem(posix_putenv_garbage, os1, newstr)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007851 /* really not much we can do; just leak */
7852 PyErr_Clear();
7853 }
7854 else {
7855 Py_DECREF(newstr);
7856 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00007857
Martin v. Löwis011e8422009-05-05 04:43:17 +00007858#ifndef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00007859 Py_DECREF(os1);
7860 Py_DECREF(os2);
Martin v. Löwis011e8422009-05-05 04:43:17 +00007861#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007862 Py_RETURN_NONE;
7863
7864error:
7865#ifndef MS_WINDOWS
7866 Py_DECREF(os1);
7867 Py_DECREF(os2);
7868#endif
7869 Py_XDECREF(newstr);
7870 return NULL;
Guido van Rossumb6775db1994-08-01 11:34:53 +00007871}
Guido van Rossumb6a47161997-09-15 22:54:34 +00007872#endif /* putenv */
7873
Guido van Rossumc524d952001-10-19 01:31:59 +00007874#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007875PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007876"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007877Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00007878
7879static PyObject *
7880posix_unsetenv(PyObject *self, PyObject *args)
7881{
Victor Stinner65170952011-11-22 22:16:17 +01007882 PyObject *name;
Victor Stinner984890f2011-11-24 13:53:38 +01007883#ifndef HAVE_BROKEN_UNSETENV
Victor Stinner60b385e2011-11-22 22:01:28 +01007884 int err;
Victor Stinner984890f2011-11-24 13:53:38 +01007885#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007886
7887 if (!PyArg_ParseTuple(args, "O&:unsetenv",
Guido van Rossumc524d952001-10-19 01:31:59 +00007888
Victor Stinner65170952011-11-22 22:16:17 +01007889 PyUnicode_FSConverter, &name))
Guido van Rossumc524d952001-10-19 01:31:59 +00007890 return NULL;
Guido van Rossumc524d952001-10-19 01:31:59 +00007891
Victor Stinner984890f2011-11-24 13:53:38 +01007892#ifdef HAVE_BROKEN_UNSETENV
7893 unsetenv(PyBytes_AS_STRING(name));
7894#else
Victor Stinner65170952011-11-22 22:16:17 +01007895 err = unsetenv(PyBytes_AS_STRING(name));
Benjamin Peterson4bb867d2011-11-22 23:12:49 -06007896 if (err) {
Benjamin Petersone8eb0e82011-11-22 23:14:47 -06007897 Py_DECREF(name);
Victor Stinner60b385e2011-11-22 22:01:28 +01007898 return posix_error();
Benjamin Peterson4bb867d2011-11-22 23:12:49 -06007899 }
Victor Stinner984890f2011-11-24 13:53:38 +01007900#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00007901
Victor Stinner8c62be82010-05-06 00:08:46 +00007902 /* Remove the key from posix_putenv_garbage;
7903 * this will cause it to be collected. This has to
7904 * happen after the real unsetenv() call because the
7905 * old value was still accessible until then.
7906 */
Victor Stinner65170952011-11-22 22:16:17 +01007907 if (PyDict_DelItem(posix_putenv_garbage, name)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007908 /* really not much we can do; just leak */
7909 PyErr_Clear();
7910 }
Victor Stinner65170952011-11-22 22:16:17 +01007911 Py_DECREF(name);
Victor Stinner84ae1182010-05-06 22:05:07 +00007912 Py_RETURN_NONE;
Guido van Rossumc524d952001-10-19 01:31:59 +00007913}
7914#endif /* unsetenv */
7915
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007916PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007917"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007918Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00007919
Guido van Rossumf68d8e52001-04-14 17:55:09 +00007920static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007921posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00007922{
Victor Stinner8c62be82010-05-06 00:08:46 +00007923 int code;
7924 char *message;
7925 if (!PyArg_ParseTuple(args, "i:strerror", &code))
7926 return NULL;
7927 message = strerror(code);
7928 if (message == NULL) {
7929 PyErr_SetString(PyExc_ValueError,
7930 "strerror() argument out of range");
7931 return NULL;
7932 }
Victor Stinner1b579672011-12-17 05:47:23 +01007933 return PyUnicode_DecodeLocale(message, "surrogateescape");
Guido van Rossumb6a47161997-09-15 22:54:34 +00007934}
Guido van Rossumb6a47161997-09-15 22:54:34 +00007935
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007936
Guido van Rossumc9641791998-08-04 15:26:23 +00007937#ifdef HAVE_SYS_WAIT_H
7938
Fred Drake106c1a02002-04-23 15:58:02 +00007939#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007940PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007941"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007942Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00007943
7944static PyObject *
7945posix_WCOREDUMP(PyObject *self, PyObject *args)
7946{
Victor Stinner8c62be82010-05-06 00:08:46 +00007947 WAIT_TYPE status;
7948 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00007949
Victor Stinner8c62be82010-05-06 00:08:46 +00007950 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
7951 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00007952
Victor Stinner8c62be82010-05-06 00:08:46 +00007953 return PyBool_FromLong(WCOREDUMP(status));
Fred Drake106c1a02002-04-23 15:58:02 +00007954}
7955#endif /* WCOREDUMP */
7956
7957#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007958PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007959"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00007960Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007961job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00007962
7963static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00007964posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00007965{
Victor Stinner8c62be82010-05-06 00:08:46 +00007966 WAIT_TYPE status;
7967 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00007968
Victor Stinner8c62be82010-05-06 00:08:46 +00007969 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
7970 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00007971
Victor Stinner8c62be82010-05-06 00:08:46 +00007972 return PyBool_FromLong(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00007973}
7974#endif /* WIFCONTINUED */
7975
Guido van Rossumc9641791998-08-04 15:26:23 +00007976#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007977PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007978"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007979Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007980
7981static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007982posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007983{
Victor Stinner8c62be82010-05-06 00:08:46 +00007984 WAIT_TYPE status;
7985 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007986
Victor Stinner8c62be82010-05-06 00:08:46 +00007987 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
7988 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007989
Victor Stinner8c62be82010-05-06 00:08:46 +00007990 return PyBool_FromLong(WIFSTOPPED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007991}
7992#endif /* WIFSTOPPED */
7993
7994#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007995PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007996"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007997Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007998
7999static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008000posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00008001{
Victor Stinner8c62be82010-05-06 00:08:46 +00008002 WAIT_TYPE status;
8003 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00008004
Victor Stinner8c62be82010-05-06 00:08:46 +00008005 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
8006 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00008007
Victor Stinner8c62be82010-05-06 00:08:46 +00008008 return PyBool_FromLong(WIFSIGNALED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008009}
8010#endif /* WIFSIGNALED */
8011
8012#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008013PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008014"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00008015Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008016system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00008017
8018static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008019posix_WIFEXITED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00008020{
Victor Stinner8c62be82010-05-06 00:08:46 +00008021 WAIT_TYPE status;
8022 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00008023
Victor Stinner8c62be82010-05-06 00:08:46 +00008024 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
8025 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00008026
Victor Stinner8c62be82010-05-06 00:08:46 +00008027 return PyBool_FromLong(WIFEXITED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008028}
8029#endif /* WIFEXITED */
8030
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00008031#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008032PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008033"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008034Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00008035
8036static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008037posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00008038{
Victor Stinner8c62be82010-05-06 00:08:46 +00008039 WAIT_TYPE status;
8040 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00008041
Victor Stinner8c62be82010-05-06 00:08:46 +00008042 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
8043 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00008044
Victor Stinner8c62be82010-05-06 00:08:46 +00008045 return Py_BuildValue("i", WEXITSTATUS(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008046}
8047#endif /* WEXITSTATUS */
8048
8049#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008050PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008051"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00008052Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008053value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00008054
8055static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008056posix_WTERMSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00008057{
Victor Stinner8c62be82010-05-06 00:08:46 +00008058 WAIT_TYPE status;
8059 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00008060
Victor Stinner8c62be82010-05-06 00:08:46 +00008061 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
8062 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00008063
Victor Stinner8c62be82010-05-06 00:08:46 +00008064 return Py_BuildValue("i", WTERMSIG(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008065}
8066#endif /* WTERMSIG */
8067
8068#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008069PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008070"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008071Return the signal that stopped the process that provided\n\
8072the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00008073
8074static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008075posix_WSTOPSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00008076{
Victor Stinner8c62be82010-05-06 00:08:46 +00008077 WAIT_TYPE status;
8078 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00008079
Victor Stinner8c62be82010-05-06 00:08:46 +00008080 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &WAIT_STATUS_INT(status)))
8081 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00008082
Victor Stinner8c62be82010-05-06 00:08:46 +00008083 return Py_BuildValue("i", WSTOPSIG(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008084}
8085#endif /* WSTOPSIG */
8086
8087#endif /* HAVE_SYS_WAIT_H */
8088
8089
Thomas Wouters477c8d52006-05-27 19:21:47 +00008090#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossumd5753e11999-10-19 13:29:23 +00008091#ifdef _SCO_DS
8092/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
8093 needed definitions in sys/statvfs.h */
8094#define _SVID3
8095#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00008096#include <sys/statvfs.h>
8097
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008098static PyObject*
8099_pystatvfs_fromstructstatvfs(struct statvfs st) {
Victor Stinner8c62be82010-05-06 00:08:46 +00008100 PyObject *v = PyStructSequence_New(&StatVFSResultType);
8101 if (v == NULL)
8102 return NULL;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008103
8104#if !defined(HAVE_LARGEFILE_SUPPORT)
Victor Stinner8c62be82010-05-06 00:08:46 +00008105 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
8106 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
8107 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long) st.f_blocks));
8108 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long) st.f_bfree));
8109 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long) st.f_bavail));
8110 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long) st.f_files));
8111 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong((long) st.f_ffree));
8112 PyStructSequence_SET_ITEM(v, 7, PyLong_FromLong((long) st.f_favail));
8113 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
8114 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008115#else
Victor Stinner8c62be82010-05-06 00:08:46 +00008116 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
8117 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
8118 PyStructSequence_SET_ITEM(v, 2,
8119 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
8120 PyStructSequence_SET_ITEM(v, 3,
8121 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
8122 PyStructSequence_SET_ITEM(v, 4,
8123 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
8124 PyStructSequence_SET_ITEM(v, 5,
8125 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
8126 PyStructSequence_SET_ITEM(v, 6,
8127 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
8128 PyStructSequence_SET_ITEM(v, 7,
8129 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
8130 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
8131 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008132#endif
8133
Victor Stinner8c62be82010-05-06 00:08:46 +00008134 return v;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008135}
8136
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008137PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008138"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008139Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00008140
8141static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008142posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00008143{
Victor Stinner8c62be82010-05-06 00:08:46 +00008144 int fd, res;
8145 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008146
Victor Stinner8c62be82010-05-06 00:08:46 +00008147 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
8148 return NULL;
8149 Py_BEGIN_ALLOW_THREADS
8150 res = fstatvfs(fd, &st);
8151 Py_END_ALLOW_THREADS
8152 if (res != 0)
8153 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008154
Victor Stinner8c62be82010-05-06 00:08:46 +00008155 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00008156}
Thomas Wouters477c8d52006-05-27 19:21:47 +00008157#endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00008158
8159
Thomas Wouters477c8d52006-05-27 19:21:47 +00008160#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossum94f6f721999-01-06 18:42:14 +00008161#include <sys/statvfs.h>
8162
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008163PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008164"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008165Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00008166
8167static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008168posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00008169{
Victor Stinner6fa67772011-09-20 04:04:33 +02008170 PyObject *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00008171 int res;
8172 struct statvfs st;
Victor Stinner6fa67772011-09-20 04:04:33 +02008173 if (!PyArg_ParseTuple(args, "O&:statvfs", PyUnicode_FSConverter, &path))
Victor Stinner8c62be82010-05-06 00:08:46 +00008174 return NULL;
8175 Py_BEGIN_ALLOW_THREADS
Victor Stinner6fa67772011-09-20 04:04:33 +02008176 res = statvfs(PyBytes_AS_STRING(path), &st);
Victor Stinner8c62be82010-05-06 00:08:46 +00008177 Py_END_ALLOW_THREADS
Victor Stinner6fa67772011-09-20 04:04:33 +02008178 if (res != 0) {
8179 posix_error_with_filename(PyBytes_AS_STRING(path));
8180 Py_DECREF(path);
8181 return NULL;
8182 }
8183 Py_DECREF(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008184
Victor Stinner8c62be82010-05-06 00:08:46 +00008185 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00008186}
8187#endif /* HAVE_STATVFS */
8188
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008189#ifdef MS_WINDOWS
8190PyDoc_STRVAR(win32__getdiskusage__doc__,
8191"_getdiskusage(path) -> (total, free)\n\n\
8192Return disk usage statistics about the given path as (total, free) tuple.");
8193
8194static PyObject *
8195win32__getdiskusage(PyObject *self, PyObject *args)
8196{
8197 BOOL retval;
8198 ULARGE_INTEGER _, total, free;
Victor Stinner6139c1b2011-11-09 22:14:14 +01008199 const wchar_t *path;
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008200
Victor Stinner6139c1b2011-11-09 22:14:14 +01008201 if (! PyArg_ParseTuple(args, "u", &path))
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008202 return NULL;
8203
8204 Py_BEGIN_ALLOW_THREADS
Victor Stinner6139c1b2011-11-09 22:14:14 +01008205 retval = GetDiskFreeSpaceExW(path, &_, &total, &free);
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008206 Py_END_ALLOW_THREADS
8207 if (retval == 0)
8208 return PyErr_SetFromWindowsErr(0);
8209
8210 return Py_BuildValue("(LL)", total.QuadPart, free.QuadPart);
8211}
8212#endif
8213
8214
Fred Drakec9680921999-12-13 16:37:25 +00008215/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
8216 * It maps strings representing configuration variable names to
8217 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00008218 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00008219 * rarely-used constants. There are three separate tables that use
8220 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00008221 *
8222 * This code is always included, even if none of the interfaces that
8223 * need it are included. The #if hackery needed to avoid it would be
8224 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00008225 */
8226struct constdef {
8227 char *name;
8228 long value;
8229};
8230
Fred Drake12c6e2d1999-12-14 21:25:03 +00008231static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008232conv_confname(PyObject *arg, int *valuep, struct constdef *table,
Guido van Rossum7d5baac2007-08-27 23:24:46 +00008233 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00008234{
Christian Heimes217cfd12007-12-02 14:31:20 +00008235 if (PyLong_Check(arg)) {
Stefan Krah0e803b32010-11-26 16:16:47 +00008236 *valuep = PyLong_AS_LONG(arg);
8237 return 1;
Fred Drake12c6e2d1999-12-14 21:25:03 +00008238 }
Guido van Rossumbce56a62007-05-10 18:04:33 +00008239 else {
Stefan Krah0e803b32010-11-26 16:16:47 +00008240 /* look up the value in the table using a binary search */
8241 size_t lo = 0;
8242 size_t mid;
8243 size_t hi = tablesize;
8244 int cmp;
8245 const char *confname;
8246 if (!PyUnicode_Check(arg)) {
8247 PyErr_SetString(PyExc_TypeError,
8248 "configuration names must be strings or integers");
8249 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00008250 }
Stefan Krah0e803b32010-11-26 16:16:47 +00008251 confname = _PyUnicode_AsString(arg);
8252 if (confname == NULL)
8253 return 0;
8254 while (lo < hi) {
8255 mid = (lo + hi) / 2;
8256 cmp = strcmp(confname, table[mid].name);
8257 if (cmp < 0)
8258 hi = mid;
8259 else if (cmp > 0)
8260 lo = mid + 1;
8261 else {
8262 *valuep = table[mid].value;
8263 return 1;
8264 }
8265 }
8266 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
8267 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00008268 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00008269}
8270
8271
8272#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
8273static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00008274#ifdef _PC_ABI_AIO_XFER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008275 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
Fred Draked86ed291999-12-15 15:34:33 +00008276#endif
8277#ifdef _PC_ABI_ASYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008278 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
Fred Draked86ed291999-12-15 15:34:33 +00008279#endif
Fred Drakec9680921999-12-13 16:37:25 +00008280#ifdef _PC_ASYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008281 {"PC_ASYNC_IO", _PC_ASYNC_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008282#endif
8283#ifdef _PC_CHOWN_RESTRICTED
Victor Stinner8c62be82010-05-06 00:08:46 +00008284 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
Fred Drakec9680921999-12-13 16:37:25 +00008285#endif
8286#ifdef _PC_FILESIZEBITS
Victor Stinner8c62be82010-05-06 00:08:46 +00008287 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
Fred Drakec9680921999-12-13 16:37:25 +00008288#endif
8289#ifdef _PC_LAST
Victor Stinner8c62be82010-05-06 00:08:46 +00008290 {"PC_LAST", _PC_LAST},
Fred Drakec9680921999-12-13 16:37:25 +00008291#endif
8292#ifdef _PC_LINK_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008293 {"PC_LINK_MAX", _PC_LINK_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008294#endif
8295#ifdef _PC_MAX_CANON
Victor Stinner8c62be82010-05-06 00:08:46 +00008296 {"PC_MAX_CANON", _PC_MAX_CANON},
Fred Drakec9680921999-12-13 16:37:25 +00008297#endif
8298#ifdef _PC_MAX_INPUT
Victor Stinner8c62be82010-05-06 00:08:46 +00008299 {"PC_MAX_INPUT", _PC_MAX_INPUT},
Fred Drakec9680921999-12-13 16:37:25 +00008300#endif
8301#ifdef _PC_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008302 {"PC_NAME_MAX", _PC_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008303#endif
8304#ifdef _PC_NO_TRUNC
Victor Stinner8c62be82010-05-06 00:08:46 +00008305 {"PC_NO_TRUNC", _PC_NO_TRUNC},
Fred Drakec9680921999-12-13 16:37:25 +00008306#endif
8307#ifdef _PC_PATH_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008308 {"PC_PATH_MAX", _PC_PATH_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008309#endif
8310#ifdef _PC_PIPE_BUF
Victor Stinner8c62be82010-05-06 00:08:46 +00008311 {"PC_PIPE_BUF", _PC_PIPE_BUF},
Fred Drakec9680921999-12-13 16:37:25 +00008312#endif
8313#ifdef _PC_PRIO_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008314 {"PC_PRIO_IO", _PC_PRIO_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008315#endif
8316#ifdef _PC_SOCK_MAXBUF
Victor Stinner8c62be82010-05-06 00:08:46 +00008317 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
Fred Drakec9680921999-12-13 16:37:25 +00008318#endif
8319#ifdef _PC_SYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008320 {"PC_SYNC_IO", _PC_SYNC_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008321#endif
8322#ifdef _PC_VDISABLE
Victor Stinner8c62be82010-05-06 00:08:46 +00008323 {"PC_VDISABLE", _PC_VDISABLE},
Fred Drakec9680921999-12-13 16:37:25 +00008324#endif
Jesus Cea7e9065c2010-10-25 13:02:04 +00008325#ifdef _PC_ACL_ENABLED
8326 {"PC_ACL_ENABLED", _PC_ACL_ENABLED},
8327#endif
8328#ifdef _PC_MIN_HOLE_SIZE
8329 {"PC_MIN_HOLE_SIZE", _PC_MIN_HOLE_SIZE},
8330#endif
8331#ifdef _PC_ALLOC_SIZE_MIN
8332 {"PC_ALLOC_SIZE_MIN", _PC_ALLOC_SIZE_MIN},
8333#endif
8334#ifdef _PC_REC_INCR_XFER_SIZE
8335 {"PC_REC_INCR_XFER_SIZE", _PC_REC_INCR_XFER_SIZE},
8336#endif
8337#ifdef _PC_REC_MAX_XFER_SIZE
8338 {"PC_REC_MAX_XFER_SIZE", _PC_REC_MAX_XFER_SIZE},
8339#endif
8340#ifdef _PC_REC_MIN_XFER_SIZE
8341 {"PC_REC_MIN_XFER_SIZE", _PC_REC_MIN_XFER_SIZE},
8342#endif
8343#ifdef _PC_REC_XFER_ALIGN
8344 {"PC_REC_XFER_ALIGN", _PC_REC_XFER_ALIGN},
8345#endif
8346#ifdef _PC_SYMLINK_MAX
8347 {"PC_SYMLINK_MAX", _PC_SYMLINK_MAX},
8348#endif
8349#ifdef _PC_XATTR_ENABLED
8350 {"PC_XATTR_ENABLED", _PC_XATTR_ENABLED},
8351#endif
8352#ifdef _PC_XATTR_EXISTS
8353 {"PC_XATTR_EXISTS", _PC_XATTR_EXISTS},
8354#endif
8355#ifdef _PC_TIMESTAMP_RESOLUTION
8356 {"PC_TIMESTAMP_RESOLUTION", _PC_TIMESTAMP_RESOLUTION},
8357#endif
Fred Drakec9680921999-12-13 16:37:25 +00008358};
8359
Fred Drakec9680921999-12-13 16:37:25 +00008360static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008361conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00008362{
8363 return conv_confname(arg, valuep, posix_constants_pathconf,
8364 sizeof(posix_constants_pathconf)
8365 / sizeof(struct constdef));
8366}
8367#endif
8368
8369#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008370PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008371"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00008372Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008373If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00008374
8375static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008376posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008377{
8378 PyObject *result = NULL;
8379 int name, fd;
8380
Fred Drake12c6e2d1999-12-14 21:25:03 +00008381 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
8382 conv_path_confname, &name)) {
Stefan Krah0e803b32010-11-26 16:16:47 +00008383 long limit;
Fred Drakec9680921999-12-13 16:37:25 +00008384
Stefan Krah0e803b32010-11-26 16:16:47 +00008385 errno = 0;
8386 limit = fpathconf(fd, name);
8387 if (limit == -1 && errno != 0)
8388 posix_error();
8389 else
8390 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00008391 }
8392 return result;
8393}
8394#endif
8395
8396
8397#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008398PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008399"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00008400Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008401If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00008402
8403static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008404posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008405{
8406 PyObject *result = NULL;
8407 int name;
8408 char *path;
8409
8410 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
8411 conv_path_confname, &name)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00008412 long limit;
Fred Drakec9680921999-12-13 16:37:25 +00008413
Victor Stinner8c62be82010-05-06 00:08:46 +00008414 errno = 0;
8415 limit = pathconf(path, name);
8416 if (limit == -1 && errno != 0) {
8417 if (errno == EINVAL)
Stefan Krah99439262010-11-26 12:58:05 +00008418 /* could be a path or name problem */
8419 posix_error();
Fred Drakec9680921999-12-13 16:37:25 +00008420 else
Stefan Krah99439262010-11-26 12:58:05 +00008421 posix_error_with_filename(path);
Victor Stinner8c62be82010-05-06 00:08:46 +00008422 }
8423 else
8424 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00008425 }
8426 return result;
8427}
8428#endif
8429
8430#ifdef HAVE_CONFSTR
8431static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00008432#ifdef _CS_ARCHITECTURE
Victor Stinner8c62be82010-05-06 00:08:46 +00008433 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
Fred Draked86ed291999-12-15 15:34:33 +00008434#endif
Mark Dickinson876d7c82010-04-16 12:47:52 +00008435#ifdef _CS_GNU_LIBC_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008436 {"CS_GNU_LIBC_VERSION", _CS_GNU_LIBC_VERSION},
Mark Dickinson876d7c82010-04-16 12:47:52 +00008437#endif
8438#ifdef _CS_GNU_LIBPTHREAD_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008439 {"CS_GNU_LIBPTHREAD_VERSION", _CS_GNU_LIBPTHREAD_VERSION},
Mark Dickinson876d7c82010-04-16 12:47:52 +00008440#endif
Fred Draked86ed291999-12-15 15:34:33 +00008441#ifdef _CS_HOSTNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008442 {"CS_HOSTNAME", _CS_HOSTNAME},
Fred Draked86ed291999-12-15 15:34:33 +00008443#endif
8444#ifdef _CS_HW_PROVIDER
Victor Stinner8c62be82010-05-06 00:08:46 +00008445 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
Fred Draked86ed291999-12-15 15:34:33 +00008446#endif
8447#ifdef _CS_HW_SERIAL
Victor Stinner8c62be82010-05-06 00:08:46 +00008448 {"CS_HW_SERIAL", _CS_HW_SERIAL},
Fred Draked86ed291999-12-15 15:34:33 +00008449#endif
8450#ifdef _CS_INITTAB_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008451 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00008452#endif
Fred Drakec9680921999-12-13 16:37:25 +00008453#ifdef _CS_LFS64_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008454 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008455#endif
8456#ifdef _CS_LFS64_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008457 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008458#endif
8459#ifdef _CS_LFS64_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008460 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008461#endif
8462#ifdef _CS_LFS64_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008463 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008464#endif
8465#ifdef _CS_LFS_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008466 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008467#endif
8468#ifdef _CS_LFS_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008469 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008470#endif
8471#ifdef _CS_LFS_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008472 {"CS_LFS_LIBS", _CS_LFS_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008473#endif
8474#ifdef _CS_LFS_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008475 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008476#endif
Fred Draked86ed291999-12-15 15:34:33 +00008477#ifdef _CS_MACHINE
Victor Stinner8c62be82010-05-06 00:08:46 +00008478 {"CS_MACHINE", _CS_MACHINE},
Fred Draked86ed291999-12-15 15:34:33 +00008479#endif
Fred Drakec9680921999-12-13 16:37:25 +00008480#ifdef _CS_PATH
Victor Stinner8c62be82010-05-06 00:08:46 +00008481 {"CS_PATH", _CS_PATH},
Fred Drakec9680921999-12-13 16:37:25 +00008482#endif
Fred Draked86ed291999-12-15 15:34:33 +00008483#ifdef _CS_RELEASE
Victor Stinner8c62be82010-05-06 00:08:46 +00008484 {"CS_RELEASE", _CS_RELEASE},
Fred Draked86ed291999-12-15 15:34:33 +00008485#endif
8486#ifdef _CS_SRPC_DOMAIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008487 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
Fred Draked86ed291999-12-15 15:34:33 +00008488#endif
8489#ifdef _CS_SYSNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008490 {"CS_SYSNAME", _CS_SYSNAME},
Fred Draked86ed291999-12-15 15:34:33 +00008491#endif
8492#ifdef _CS_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008493 {"CS_VERSION", _CS_VERSION},
Fred Draked86ed291999-12-15 15:34:33 +00008494#endif
Fred Drakec9680921999-12-13 16:37:25 +00008495#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008496 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008497#endif
8498#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008499 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008500#endif
8501#ifdef _CS_XBS5_ILP32_OFF32_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008502 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008503#endif
8504#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008505 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008506#endif
8507#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008508 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008509#endif
8510#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008511 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008512#endif
8513#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008514 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008515#endif
8516#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008517 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008518#endif
8519#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008520 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008521#endif
8522#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008523 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008524#endif
8525#ifdef _CS_XBS5_LP64_OFF64_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008526 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008527#endif
8528#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008529 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008530#endif
8531#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008532 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008533#endif
8534#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008535 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008536#endif
8537#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008538 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008539#endif
8540#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008541 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008542#endif
Fred Draked86ed291999-12-15 15:34:33 +00008543#ifdef _MIPS_CS_AVAIL_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00008544 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00008545#endif
8546#ifdef _MIPS_CS_BASE
Victor Stinner8c62be82010-05-06 00:08:46 +00008547 {"MIPS_CS_BASE", _MIPS_CS_BASE},
Fred Draked86ed291999-12-15 15:34:33 +00008548#endif
8549#ifdef _MIPS_CS_HOSTID
Victor Stinner8c62be82010-05-06 00:08:46 +00008550 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
Fred Draked86ed291999-12-15 15:34:33 +00008551#endif
8552#ifdef _MIPS_CS_HW_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008553 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00008554#endif
8555#ifdef _MIPS_CS_NUM_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00008556 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00008557#endif
8558#ifdef _MIPS_CS_OSREL_MAJ
Victor Stinner8c62be82010-05-06 00:08:46 +00008559 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
Fred Draked86ed291999-12-15 15:34:33 +00008560#endif
8561#ifdef _MIPS_CS_OSREL_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008562 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
Fred Draked86ed291999-12-15 15:34:33 +00008563#endif
8564#ifdef _MIPS_CS_OSREL_PATCH
Victor Stinner8c62be82010-05-06 00:08:46 +00008565 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
Fred Draked86ed291999-12-15 15:34:33 +00008566#endif
8567#ifdef _MIPS_CS_OS_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008568 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00008569#endif
8570#ifdef _MIPS_CS_OS_PROVIDER
Victor Stinner8c62be82010-05-06 00:08:46 +00008571 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
Fred Draked86ed291999-12-15 15:34:33 +00008572#endif
8573#ifdef _MIPS_CS_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00008574 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00008575#endif
8576#ifdef _MIPS_CS_SERIAL
Victor Stinner8c62be82010-05-06 00:08:46 +00008577 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
Fred Draked86ed291999-12-15 15:34:33 +00008578#endif
8579#ifdef _MIPS_CS_VENDOR
Victor Stinner8c62be82010-05-06 00:08:46 +00008580 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
Fred Draked86ed291999-12-15 15:34:33 +00008581#endif
Fred Drakec9680921999-12-13 16:37:25 +00008582};
8583
8584static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008585conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00008586{
8587 return conv_confname(arg, valuep, posix_constants_confstr,
8588 sizeof(posix_constants_confstr)
8589 / sizeof(struct constdef));
8590}
8591
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008592PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008593"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008594Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00008595
8596static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008597posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008598{
8599 PyObject *result = NULL;
8600 int name;
Victor Stinnercb043522010-09-10 23:49:04 +00008601 char buffer[255];
Stefan Krah99439262010-11-26 12:58:05 +00008602 int len;
Fred Drakec9680921999-12-13 16:37:25 +00008603
Victor Stinnercb043522010-09-10 23:49:04 +00008604 if (!PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name))
8605 return NULL;
8606
8607 errno = 0;
8608 len = confstr(name, buffer, sizeof(buffer));
8609 if (len == 0) {
8610 if (errno) {
8611 posix_error();
8612 return NULL;
Fred Drakec9680921999-12-13 16:37:25 +00008613 }
8614 else {
Victor Stinnercb043522010-09-10 23:49:04 +00008615 Py_RETURN_NONE;
Fred Drakec9680921999-12-13 16:37:25 +00008616 }
8617 }
Victor Stinnercb043522010-09-10 23:49:04 +00008618
8619 if ((unsigned int)len >= sizeof(buffer)) {
8620 char *buf = PyMem_Malloc(len);
8621 if (buf == NULL)
8622 return PyErr_NoMemory();
8623 confstr(name, buf, len);
8624 result = PyUnicode_DecodeFSDefaultAndSize(buf, len-1);
8625 PyMem_Free(buf);
8626 }
8627 else
8628 result = PyUnicode_DecodeFSDefaultAndSize(buffer, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00008629 return result;
8630}
8631#endif
8632
8633
8634#ifdef HAVE_SYSCONF
8635static struct constdef posix_constants_sysconf[] = {
8636#ifdef _SC_2_CHAR_TERM
Victor Stinner8c62be82010-05-06 00:08:46 +00008637 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
Fred Drakec9680921999-12-13 16:37:25 +00008638#endif
8639#ifdef _SC_2_C_BIND
Victor Stinner8c62be82010-05-06 00:08:46 +00008640 {"SC_2_C_BIND", _SC_2_C_BIND},
Fred Drakec9680921999-12-13 16:37:25 +00008641#endif
8642#ifdef _SC_2_C_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00008643 {"SC_2_C_DEV", _SC_2_C_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00008644#endif
8645#ifdef _SC_2_C_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008646 {"SC_2_C_VERSION", _SC_2_C_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00008647#endif
8648#ifdef _SC_2_FORT_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00008649 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00008650#endif
8651#ifdef _SC_2_FORT_RUN
Victor Stinner8c62be82010-05-06 00:08:46 +00008652 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
Fred Drakec9680921999-12-13 16:37:25 +00008653#endif
8654#ifdef _SC_2_LOCALEDEF
Victor Stinner8c62be82010-05-06 00:08:46 +00008655 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
Fred Drakec9680921999-12-13 16:37:25 +00008656#endif
8657#ifdef _SC_2_SW_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00008658 {"SC_2_SW_DEV", _SC_2_SW_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00008659#endif
8660#ifdef _SC_2_UPE
Victor Stinner8c62be82010-05-06 00:08:46 +00008661 {"SC_2_UPE", _SC_2_UPE},
Fred Drakec9680921999-12-13 16:37:25 +00008662#endif
8663#ifdef _SC_2_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008664 {"SC_2_VERSION", _SC_2_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00008665#endif
Fred Draked86ed291999-12-15 15:34:33 +00008666#ifdef _SC_ABI_ASYNCHRONOUS_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008667 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
Fred Draked86ed291999-12-15 15:34:33 +00008668#endif
8669#ifdef _SC_ACL
Victor Stinner8c62be82010-05-06 00:08:46 +00008670 {"SC_ACL", _SC_ACL},
Fred Draked86ed291999-12-15 15:34:33 +00008671#endif
Fred Drakec9680921999-12-13 16:37:25 +00008672#ifdef _SC_AIO_LISTIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008673 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008674#endif
Fred Drakec9680921999-12-13 16:37:25 +00008675#ifdef _SC_AIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008676 {"SC_AIO_MAX", _SC_AIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008677#endif
8678#ifdef _SC_AIO_PRIO_DELTA_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008679 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008680#endif
8681#ifdef _SC_ARG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008682 {"SC_ARG_MAX", _SC_ARG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008683#endif
8684#ifdef _SC_ASYNCHRONOUS_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008685 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008686#endif
8687#ifdef _SC_ATEXIT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008688 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008689#endif
Fred Draked86ed291999-12-15 15:34:33 +00008690#ifdef _SC_AUDIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008691 {"SC_AUDIT", _SC_AUDIT},
Fred Draked86ed291999-12-15 15:34:33 +00008692#endif
Fred Drakec9680921999-12-13 16:37:25 +00008693#ifdef _SC_AVPHYS_PAGES
Victor Stinner8c62be82010-05-06 00:08:46 +00008694 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
Fred Drakec9680921999-12-13 16:37:25 +00008695#endif
8696#ifdef _SC_BC_BASE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008697 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008698#endif
8699#ifdef _SC_BC_DIM_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008700 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008701#endif
8702#ifdef _SC_BC_SCALE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008703 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008704#endif
8705#ifdef _SC_BC_STRING_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008706 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008707#endif
Fred Draked86ed291999-12-15 15:34:33 +00008708#ifdef _SC_CAP
Victor Stinner8c62be82010-05-06 00:08:46 +00008709 {"SC_CAP", _SC_CAP},
Fred Draked86ed291999-12-15 15:34:33 +00008710#endif
Fred Drakec9680921999-12-13 16:37:25 +00008711#ifdef _SC_CHARCLASS_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008712 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008713#endif
8714#ifdef _SC_CHAR_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008715 {"SC_CHAR_BIT", _SC_CHAR_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00008716#endif
8717#ifdef _SC_CHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008718 {"SC_CHAR_MAX", _SC_CHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008719#endif
8720#ifdef _SC_CHAR_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008721 {"SC_CHAR_MIN", _SC_CHAR_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008722#endif
8723#ifdef _SC_CHILD_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008724 {"SC_CHILD_MAX", _SC_CHILD_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008725#endif
8726#ifdef _SC_CLK_TCK
Victor Stinner8c62be82010-05-06 00:08:46 +00008727 {"SC_CLK_TCK", _SC_CLK_TCK},
Fred Drakec9680921999-12-13 16:37:25 +00008728#endif
8729#ifdef _SC_COHER_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008730 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008731#endif
8732#ifdef _SC_COLL_WEIGHTS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008733 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008734#endif
8735#ifdef _SC_DCACHE_ASSOC
Victor Stinner8c62be82010-05-06 00:08:46 +00008736 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
Fred Drakec9680921999-12-13 16:37:25 +00008737#endif
8738#ifdef _SC_DCACHE_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008739 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008740#endif
8741#ifdef _SC_DCACHE_LINESZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008742 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
Fred Drakec9680921999-12-13 16:37:25 +00008743#endif
8744#ifdef _SC_DCACHE_SZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008745 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
Fred Drakec9680921999-12-13 16:37:25 +00008746#endif
8747#ifdef _SC_DCACHE_TBLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008748 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008749#endif
8750#ifdef _SC_DELAYTIMER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008751 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008752#endif
8753#ifdef _SC_EQUIV_CLASS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008754 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008755#endif
8756#ifdef _SC_EXPR_NEST_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008757 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008758#endif
8759#ifdef _SC_FSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00008760 {"SC_FSYNC", _SC_FSYNC},
Fred Drakec9680921999-12-13 16:37:25 +00008761#endif
8762#ifdef _SC_GETGR_R_SIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008763 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008764#endif
8765#ifdef _SC_GETPW_R_SIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008766 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008767#endif
8768#ifdef _SC_ICACHE_ASSOC
Victor Stinner8c62be82010-05-06 00:08:46 +00008769 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
Fred Drakec9680921999-12-13 16:37:25 +00008770#endif
8771#ifdef _SC_ICACHE_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008772 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008773#endif
8774#ifdef _SC_ICACHE_LINESZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008775 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
Fred Drakec9680921999-12-13 16:37:25 +00008776#endif
8777#ifdef _SC_ICACHE_SZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008778 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
Fred Drakec9680921999-12-13 16:37:25 +00008779#endif
Fred Draked86ed291999-12-15 15:34:33 +00008780#ifdef _SC_INF
Victor Stinner8c62be82010-05-06 00:08:46 +00008781 {"SC_INF", _SC_INF},
Fred Draked86ed291999-12-15 15:34:33 +00008782#endif
Fred Drakec9680921999-12-13 16:37:25 +00008783#ifdef _SC_INT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008784 {"SC_INT_MAX", _SC_INT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008785#endif
8786#ifdef _SC_INT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008787 {"SC_INT_MIN", _SC_INT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008788#endif
8789#ifdef _SC_IOV_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008790 {"SC_IOV_MAX", _SC_IOV_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008791#endif
Fred Draked86ed291999-12-15 15:34:33 +00008792#ifdef _SC_IP_SECOPTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008793 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
Fred Draked86ed291999-12-15 15:34:33 +00008794#endif
Fred Drakec9680921999-12-13 16:37:25 +00008795#ifdef _SC_JOB_CONTROL
Victor Stinner8c62be82010-05-06 00:08:46 +00008796 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
Fred Drakec9680921999-12-13 16:37:25 +00008797#endif
Fred Draked86ed291999-12-15 15:34:33 +00008798#ifdef _SC_KERN_POINTERS
Victor Stinner8c62be82010-05-06 00:08:46 +00008799 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
Fred Draked86ed291999-12-15 15:34:33 +00008800#endif
8801#ifdef _SC_KERN_SIM
Victor Stinner8c62be82010-05-06 00:08:46 +00008802 {"SC_KERN_SIM", _SC_KERN_SIM},
Fred Draked86ed291999-12-15 15:34:33 +00008803#endif
Fred Drakec9680921999-12-13 16:37:25 +00008804#ifdef _SC_LINE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008805 {"SC_LINE_MAX", _SC_LINE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008806#endif
8807#ifdef _SC_LOGIN_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008808 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008809#endif
8810#ifdef _SC_LOGNAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008811 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008812#endif
8813#ifdef _SC_LONG_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008814 {"SC_LONG_BIT", _SC_LONG_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00008815#endif
Fred Draked86ed291999-12-15 15:34:33 +00008816#ifdef _SC_MAC
Victor Stinner8c62be82010-05-06 00:08:46 +00008817 {"SC_MAC", _SC_MAC},
Fred Draked86ed291999-12-15 15:34:33 +00008818#endif
Fred Drakec9680921999-12-13 16:37:25 +00008819#ifdef _SC_MAPPED_FILES
Victor Stinner8c62be82010-05-06 00:08:46 +00008820 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
Fred Drakec9680921999-12-13 16:37:25 +00008821#endif
8822#ifdef _SC_MAXPID
Victor Stinner8c62be82010-05-06 00:08:46 +00008823 {"SC_MAXPID", _SC_MAXPID},
Fred Drakec9680921999-12-13 16:37:25 +00008824#endif
8825#ifdef _SC_MB_LEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008826 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008827#endif
8828#ifdef _SC_MEMLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +00008829 {"SC_MEMLOCK", _SC_MEMLOCK},
Fred Drakec9680921999-12-13 16:37:25 +00008830#endif
8831#ifdef _SC_MEMLOCK_RANGE
Victor Stinner8c62be82010-05-06 00:08:46 +00008832 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
Fred Drakec9680921999-12-13 16:37:25 +00008833#endif
8834#ifdef _SC_MEMORY_PROTECTION
Victor Stinner8c62be82010-05-06 00:08:46 +00008835 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
Fred Drakec9680921999-12-13 16:37:25 +00008836#endif
8837#ifdef _SC_MESSAGE_PASSING
Victor Stinner8c62be82010-05-06 00:08:46 +00008838 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
Fred Drakec9680921999-12-13 16:37:25 +00008839#endif
Fred Draked86ed291999-12-15 15:34:33 +00008840#ifdef _SC_MMAP_FIXED_ALIGNMENT
Victor Stinner8c62be82010-05-06 00:08:46 +00008841 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
Fred Draked86ed291999-12-15 15:34:33 +00008842#endif
Fred Drakec9680921999-12-13 16:37:25 +00008843#ifdef _SC_MQ_OPEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008844 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008845#endif
8846#ifdef _SC_MQ_PRIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008847 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008848#endif
Fred Draked86ed291999-12-15 15:34:33 +00008849#ifdef _SC_NACLS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008850 {"SC_NACLS_MAX", _SC_NACLS_MAX},
Fred Draked86ed291999-12-15 15:34:33 +00008851#endif
Fred Drakec9680921999-12-13 16:37:25 +00008852#ifdef _SC_NGROUPS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008853 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008854#endif
8855#ifdef _SC_NL_ARGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008856 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008857#endif
8858#ifdef _SC_NL_LANGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008859 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008860#endif
8861#ifdef _SC_NL_MSGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008862 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008863#endif
8864#ifdef _SC_NL_NMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008865 {"SC_NL_NMAX", _SC_NL_NMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008866#endif
8867#ifdef _SC_NL_SETMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008868 {"SC_NL_SETMAX", _SC_NL_SETMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008869#endif
8870#ifdef _SC_NL_TEXTMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008871 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008872#endif
8873#ifdef _SC_NPROCESSORS_CONF
Victor Stinner8c62be82010-05-06 00:08:46 +00008874 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
Fred Drakec9680921999-12-13 16:37:25 +00008875#endif
8876#ifdef _SC_NPROCESSORS_ONLN
Victor Stinner8c62be82010-05-06 00:08:46 +00008877 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
Fred Drakec9680921999-12-13 16:37:25 +00008878#endif
Fred Draked86ed291999-12-15 15:34:33 +00008879#ifdef _SC_NPROC_CONF
Victor Stinner8c62be82010-05-06 00:08:46 +00008880 {"SC_NPROC_CONF", _SC_NPROC_CONF},
Fred Draked86ed291999-12-15 15:34:33 +00008881#endif
8882#ifdef _SC_NPROC_ONLN
Victor Stinner8c62be82010-05-06 00:08:46 +00008883 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
Fred Draked86ed291999-12-15 15:34:33 +00008884#endif
Fred Drakec9680921999-12-13 16:37:25 +00008885#ifdef _SC_NZERO
Victor Stinner8c62be82010-05-06 00:08:46 +00008886 {"SC_NZERO", _SC_NZERO},
Fred Drakec9680921999-12-13 16:37:25 +00008887#endif
8888#ifdef _SC_OPEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008889 {"SC_OPEN_MAX", _SC_OPEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008890#endif
8891#ifdef _SC_PAGESIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008892 {"SC_PAGESIZE", _SC_PAGESIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008893#endif
8894#ifdef _SC_PAGE_SIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008895 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008896#endif
8897#ifdef _SC_PASS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008898 {"SC_PASS_MAX", _SC_PASS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008899#endif
8900#ifdef _SC_PHYS_PAGES
Victor Stinner8c62be82010-05-06 00:08:46 +00008901 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
Fred Drakec9680921999-12-13 16:37:25 +00008902#endif
8903#ifdef _SC_PII
Victor Stinner8c62be82010-05-06 00:08:46 +00008904 {"SC_PII", _SC_PII},
Fred Drakec9680921999-12-13 16:37:25 +00008905#endif
8906#ifdef _SC_PII_INTERNET
Victor Stinner8c62be82010-05-06 00:08:46 +00008907 {"SC_PII_INTERNET", _SC_PII_INTERNET},
Fred Drakec9680921999-12-13 16:37:25 +00008908#endif
8909#ifdef _SC_PII_INTERNET_DGRAM
Victor Stinner8c62be82010-05-06 00:08:46 +00008910 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
Fred Drakec9680921999-12-13 16:37:25 +00008911#endif
8912#ifdef _SC_PII_INTERNET_STREAM
Victor Stinner8c62be82010-05-06 00:08:46 +00008913 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
Fred Drakec9680921999-12-13 16:37:25 +00008914#endif
8915#ifdef _SC_PII_OSI
Victor Stinner8c62be82010-05-06 00:08:46 +00008916 {"SC_PII_OSI", _SC_PII_OSI},
Fred Drakec9680921999-12-13 16:37:25 +00008917#endif
8918#ifdef _SC_PII_OSI_CLTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008919 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
Fred Drakec9680921999-12-13 16:37:25 +00008920#endif
8921#ifdef _SC_PII_OSI_COTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008922 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
Fred Drakec9680921999-12-13 16:37:25 +00008923#endif
8924#ifdef _SC_PII_OSI_M
Victor Stinner8c62be82010-05-06 00:08:46 +00008925 {"SC_PII_OSI_M", _SC_PII_OSI_M},
Fred Drakec9680921999-12-13 16:37:25 +00008926#endif
8927#ifdef _SC_PII_SOCKET
Victor Stinner8c62be82010-05-06 00:08:46 +00008928 {"SC_PII_SOCKET", _SC_PII_SOCKET},
Fred Drakec9680921999-12-13 16:37:25 +00008929#endif
8930#ifdef _SC_PII_XTI
Victor Stinner8c62be82010-05-06 00:08:46 +00008931 {"SC_PII_XTI", _SC_PII_XTI},
Fred Drakec9680921999-12-13 16:37:25 +00008932#endif
8933#ifdef _SC_POLL
Victor Stinner8c62be82010-05-06 00:08:46 +00008934 {"SC_POLL", _SC_POLL},
Fred Drakec9680921999-12-13 16:37:25 +00008935#endif
8936#ifdef _SC_PRIORITIZED_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008937 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008938#endif
8939#ifdef _SC_PRIORITY_SCHEDULING
Victor Stinner8c62be82010-05-06 00:08:46 +00008940 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
Fred Drakec9680921999-12-13 16:37:25 +00008941#endif
8942#ifdef _SC_REALTIME_SIGNALS
Victor Stinner8c62be82010-05-06 00:08:46 +00008943 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
Fred Drakec9680921999-12-13 16:37:25 +00008944#endif
8945#ifdef _SC_RE_DUP_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008946 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008947#endif
8948#ifdef _SC_RTSIG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008949 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008950#endif
8951#ifdef _SC_SAVED_IDS
Victor Stinner8c62be82010-05-06 00:08:46 +00008952 {"SC_SAVED_IDS", _SC_SAVED_IDS},
Fred Drakec9680921999-12-13 16:37:25 +00008953#endif
8954#ifdef _SC_SCHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008955 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008956#endif
8957#ifdef _SC_SCHAR_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008958 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008959#endif
8960#ifdef _SC_SELECT
Victor Stinner8c62be82010-05-06 00:08:46 +00008961 {"SC_SELECT", _SC_SELECT},
Fred Drakec9680921999-12-13 16:37:25 +00008962#endif
8963#ifdef _SC_SEMAPHORES
Victor Stinner8c62be82010-05-06 00:08:46 +00008964 {"SC_SEMAPHORES", _SC_SEMAPHORES},
Fred Drakec9680921999-12-13 16:37:25 +00008965#endif
8966#ifdef _SC_SEM_NSEMS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008967 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008968#endif
8969#ifdef _SC_SEM_VALUE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008970 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008971#endif
8972#ifdef _SC_SHARED_MEMORY_OBJECTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008973 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
Fred Drakec9680921999-12-13 16:37:25 +00008974#endif
8975#ifdef _SC_SHRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008976 {"SC_SHRT_MAX", _SC_SHRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008977#endif
8978#ifdef _SC_SHRT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008979 {"SC_SHRT_MIN", _SC_SHRT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008980#endif
8981#ifdef _SC_SIGQUEUE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008982 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008983#endif
8984#ifdef _SC_SIGRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008985 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008986#endif
8987#ifdef _SC_SIGRT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008988 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008989#endif
Fred Draked86ed291999-12-15 15:34:33 +00008990#ifdef _SC_SOFTPOWER
Victor Stinner8c62be82010-05-06 00:08:46 +00008991 {"SC_SOFTPOWER", _SC_SOFTPOWER},
Fred Draked86ed291999-12-15 15:34:33 +00008992#endif
Fred Drakec9680921999-12-13 16:37:25 +00008993#ifdef _SC_SPLIT_CACHE
Victor Stinner8c62be82010-05-06 00:08:46 +00008994 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
Fred Drakec9680921999-12-13 16:37:25 +00008995#endif
8996#ifdef _SC_SSIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008997 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008998#endif
8999#ifdef _SC_STACK_PROT
Victor Stinner8c62be82010-05-06 00:08:46 +00009000 {"SC_STACK_PROT", _SC_STACK_PROT},
Fred Drakec9680921999-12-13 16:37:25 +00009001#endif
9002#ifdef _SC_STREAM_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009003 {"SC_STREAM_MAX", _SC_STREAM_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009004#endif
9005#ifdef _SC_SYNCHRONIZED_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00009006 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
Fred Drakec9680921999-12-13 16:37:25 +00009007#endif
9008#ifdef _SC_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00009009 {"SC_THREADS", _SC_THREADS},
Fred Drakec9680921999-12-13 16:37:25 +00009010#endif
9011#ifdef _SC_THREAD_ATTR_STACKADDR
Victor Stinner8c62be82010-05-06 00:08:46 +00009012 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
Fred Drakec9680921999-12-13 16:37:25 +00009013#endif
9014#ifdef _SC_THREAD_ATTR_STACKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00009015 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
Fred Drakec9680921999-12-13 16:37:25 +00009016#endif
9017#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
Victor Stinner8c62be82010-05-06 00:08:46 +00009018 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
Fred Drakec9680921999-12-13 16:37:25 +00009019#endif
9020#ifdef _SC_THREAD_KEYS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009021 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009022#endif
9023#ifdef _SC_THREAD_PRIORITY_SCHEDULING
Victor Stinner8c62be82010-05-06 00:08:46 +00009024 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
Fred Drakec9680921999-12-13 16:37:25 +00009025#endif
9026#ifdef _SC_THREAD_PRIO_INHERIT
Victor Stinner8c62be82010-05-06 00:08:46 +00009027 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
Fred Drakec9680921999-12-13 16:37:25 +00009028#endif
9029#ifdef _SC_THREAD_PRIO_PROTECT
Victor Stinner8c62be82010-05-06 00:08:46 +00009030 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
Fred Drakec9680921999-12-13 16:37:25 +00009031#endif
9032#ifdef _SC_THREAD_PROCESS_SHARED
Victor Stinner8c62be82010-05-06 00:08:46 +00009033 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
Fred Drakec9680921999-12-13 16:37:25 +00009034#endif
9035#ifdef _SC_THREAD_SAFE_FUNCTIONS
Victor Stinner8c62be82010-05-06 00:08:46 +00009036 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
Fred Drakec9680921999-12-13 16:37:25 +00009037#endif
9038#ifdef _SC_THREAD_STACK_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00009039 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00009040#endif
9041#ifdef _SC_THREAD_THREADS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009042 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009043#endif
9044#ifdef _SC_TIMERS
Victor Stinner8c62be82010-05-06 00:08:46 +00009045 {"SC_TIMERS", _SC_TIMERS},
Fred Drakec9680921999-12-13 16:37:25 +00009046#endif
9047#ifdef _SC_TIMER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009048 {"SC_TIMER_MAX", _SC_TIMER_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009049#endif
9050#ifdef _SC_TTY_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009051 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009052#endif
9053#ifdef _SC_TZNAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009054 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009055#endif
9056#ifdef _SC_T_IOV_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009057 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009058#endif
9059#ifdef _SC_UCHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009060 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009061#endif
9062#ifdef _SC_UINT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009063 {"SC_UINT_MAX", _SC_UINT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009064#endif
9065#ifdef _SC_UIO_MAXIOV
Victor Stinner8c62be82010-05-06 00:08:46 +00009066 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
Fred Drakec9680921999-12-13 16:37:25 +00009067#endif
9068#ifdef _SC_ULONG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009069 {"SC_ULONG_MAX", _SC_ULONG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009070#endif
9071#ifdef _SC_USHRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009072 {"SC_USHRT_MAX", _SC_USHRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009073#endif
9074#ifdef _SC_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00009075 {"SC_VERSION", _SC_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00009076#endif
9077#ifdef _SC_WORD_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00009078 {"SC_WORD_BIT", _SC_WORD_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00009079#endif
9080#ifdef _SC_XBS5_ILP32_OFF32
Victor Stinner8c62be82010-05-06 00:08:46 +00009081 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
Fred Drakec9680921999-12-13 16:37:25 +00009082#endif
9083#ifdef _SC_XBS5_ILP32_OFFBIG
Victor Stinner8c62be82010-05-06 00:08:46 +00009084 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
Fred Drakec9680921999-12-13 16:37:25 +00009085#endif
9086#ifdef _SC_XBS5_LP64_OFF64
Victor Stinner8c62be82010-05-06 00:08:46 +00009087 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
Fred Drakec9680921999-12-13 16:37:25 +00009088#endif
9089#ifdef _SC_XBS5_LPBIG_OFFBIG
Victor Stinner8c62be82010-05-06 00:08:46 +00009090 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
Fred Drakec9680921999-12-13 16:37:25 +00009091#endif
9092#ifdef _SC_XOPEN_CRYPT
Victor Stinner8c62be82010-05-06 00:08:46 +00009093 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
Fred Drakec9680921999-12-13 16:37:25 +00009094#endif
9095#ifdef _SC_XOPEN_ENH_I18N
Victor Stinner8c62be82010-05-06 00:08:46 +00009096 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
Fred Drakec9680921999-12-13 16:37:25 +00009097#endif
9098#ifdef _SC_XOPEN_LEGACY
Victor Stinner8c62be82010-05-06 00:08:46 +00009099 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
Fred Drakec9680921999-12-13 16:37:25 +00009100#endif
9101#ifdef _SC_XOPEN_REALTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00009102 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
Fred Drakec9680921999-12-13 16:37:25 +00009103#endif
9104#ifdef _SC_XOPEN_REALTIME_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00009105 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
Fred Drakec9680921999-12-13 16:37:25 +00009106#endif
9107#ifdef _SC_XOPEN_SHM
Victor Stinner8c62be82010-05-06 00:08:46 +00009108 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
Fred Drakec9680921999-12-13 16:37:25 +00009109#endif
9110#ifdef _SC_XOPEN_UNIX
Victor Stinner8c62be82010-05-06 00:08:46 +00009111 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
Fred Drakec9680921999-12-13 16:37:25 +00009112#endif
9113#ifdef _SC_XOPEN_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00009114 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00009115#endif
9116#ifdef _SC_XOPEN_XCU_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00009117 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00009118#endif
9119#ifdef _SC_XOPEN_XPG2
Victor Stinner8c62be82010-05-06 00:08:46 +00009120 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
Fred Drakec9680921999-12-13 16:37:25 +00009121#endif
9122#ifdef _SC_XOPEN_XPG3
Victor Stinner8c62be82010-05-06 00:08:46 +00009123 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
Fred Drakec9680921999-12-13 16:37:25 +00009124#endif
9125#ifdef _SC_XOPEN_XPG4
Victor Stinner8c62be82010-05-06 00:08:46 +00009126 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
Fred Drakec9680921999-12-13 16:37:25 +00009127#endif
9128};
9129
9130static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009131conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00009132{
9133 return conv_confname(arg, valuep, posix_constants_sysconf,
9134 sizeof(posix_constants_sysconf)
9135 / sizeof(struct constdef));
9136}
9137
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009138PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00009139"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009140Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00009141
9142static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009143posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00009144{
9145 PyObject *result = NULL;
9146 int name;
9147
9148 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
9149 int value;
9150
9151 errno = 0;
9152 value = sysconf(name);
9153 if (value == -1 && errno != 0)
9154 posix_error();
9155 else
Christian Heimes217cfd12007-12-02 14:31:20 +00009156 result = PyLong_FromLong(value);
Fred Drakec9680921999-12-13 16:37:25 +00009157 }
9158 return result;
9159}
9160#endif
9161
9162
Fred Drakebec628d1999-12-15 18:31:10 +00009163/* This code is used to ensure that the tables of configuration value names
9164 * are in sorted order as required by conv_confname(), and also to build the
9165 * the exported dictionaries that are used to publish information about the
9166 * names available on the host platform.
9167 *
9168 * Sorting the table at runtime ensures that the table is properly ordered
9169 * when used, even for platforms we're not able to test on. It also makes
9170 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00009171 */
Fred Drakebec628d1999-12-15 18:31:10 +00009172
9173static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009174cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00009175{
9176 const struct constdef *c1 =
Victor Stinner8c62be82010-05-06 00:08:46 +00009177 (const struct constdef *) v1;
Fred Drakebec628d1999-12-15 18:31:10 +00009178 const struct constdef *c2 =
Victor Stinner8c62be82010-05-06 00:08:46 +00009179 (const struct constdef *) v2;
Fred Drakebec628d1999-12-15 18:31:10 +00009180
9181 return strcmp(c1->name, c2->name);
9182}
9183
9184static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009185setup_confname_table(struct constdef *table, size_t tablesize,
Victor Stinner8c62be82010-05-06 00:08:46 +00009186 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00009187{
Fred Drakebec628d1999-12-15 18:31:10 +00009188 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00009189 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00009190
9191 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
9192 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00009193 if (d == NULL)
Victor Stinner8c62be82010-05-06 00:08:46 +00009194 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009195
Barry Warsaw3155db32000-04-13 15:20:40 +00009196 for (i=0; i < tablesize; ++i) {
Victor Stinner8c62be82010-05-06 00:08:46 +00009197 PyObject *o = PyLong_FromLong(table[i].value);
9198 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
9199 Py_XDECREF(o);
9200 Py_DECREF(d);
9201 return -1;
9202 }
9203 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00009204 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00009205 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00009206}
9207
Fred Drakebec628d1999-12-15 18:31:10 +00009208/* Return -1 on failure, 0 on success. */
9209static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00009210setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00009211{
9212#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00009213 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00009214 sizeof(posix_constants_pathconf)
9215 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00009216 "pathconf_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00009217 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009218#endif
9219#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00009220 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00009221 sizeof(posix_constants_confstr)
9222 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00009223 "confstr_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00009224 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009225#endif
9226#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00009227 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00009228 sizeof(posix_constants_sysconf)
9229 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00009230 "sysconf_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00009231 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009232#endif
Fred Drakebec628d1999-12-15 18:31:10 +00009233 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00009234}
Fred Draked86ed291999-12-15 15:34:33 +00009235
9236
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009237PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00009238"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009239Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009240in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009241
9242static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00009243posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009244{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009245 abort();
9246 /*NOTREACHED*/
9247 Py_FatalError("abort() called from Python code didn't abort!");
9248 return NULL;
9249}
Fred Drakebec628d1999-12-15 18:31:10 +00009250
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00009251#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009252PyDoc_STRVAR(win32_startfile__doc__,
Georg Brandlf4f44152006-02-18 22:29:33 +00009253"startfile(filepath [, operation]) - Start a file with its associated\n\
9254application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00009255\n\
Georg Brandlf4f44152006-02-18 22:29:33 +00009256When \"operation\" is not specified or \"open\", this acts like\n\
9257double-clicking the file in Explorer, or giving the file name as an\n\
9258argument to the DOS \"start\" command: the file is opened with whatever\n\
9259application (if any) its extension is associated.\n\
9260When another \"operation\" is given, it specifies what should be done with\n\
9261the file. A typical operation is \"print\".\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00009262\n\
9263startfile returns as soon as the associated application is launched.\n\
9264There is no option to wait for the application to close, and no way\n\
9265to retrieve the application's exit status.\n\
9266\n\
9267The filepath is relative to the current directory. If you want to use\n\
9268an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009269the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00009270
9271static PyObject *
9272win32_startfile(PyObject *self, PyObject *args)
9273{
Victor Stinner8c62be82010-05-06 00:08:46 +00009274 PyObject *ofilepath;
9275 char *filepath;
9276 char *operation = NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02009277 wchar_t *wpath, *woperation;
Victor Stinner8c62be82010-05-06 00:08:46 +00009278 HINSTANCE rc;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00009279
Victor Stinnereb5657a2011-09-30 01:44:27 +02009280 PyObject *unipath, *uoperation = NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00009281 if (!PyArg_ParseTuple(args, "U|s:startfile",
9282 &unipath, &operation)) {
9283 PyErr_Clear();
9284 goto normal;
9285 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00009286
Victor Stinner8c62be82010-05-06 00:08:46 +00009287 if (operation) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02009288 uoperation = PyUnicode_DecodeASCII(operation,
Victor Stinner8c62be82010-05-06 00:08:46 +00009289 strlen(operation), NULL);
Victor Stinnereb5657a2011-09-30 01:44:27 +02009290 if (!uoperation) {
Victor Stinner8c62be82010-05-06 00:08:46 +00009291 PyErr_Clear();
9292 operation = NULL;
9293 goto normal;
9294 }
9295 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00009296
Victor Stinnereb5657a2011-09-30 01:44:27 +02009297 wpath = PyUnicode_AsUnicode(unipath);
9298 if (wpath == NULL)
9299 goto normal;
9300 if (uoperation) {
9301 woperation = PyUnicode_AsUnicode(uoperation);
9302 if (woperation == NULL)
9303 goto normal;
9304 }
9305 else
9306 woperation = NULL;
9307
Victor Stinner8c62be82010-05-06 00:08:46 +00009308 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02009309 rc = ShellExecuteW((HWND)0, woperation, wpath,
9310 NULL, NULL, SW_SHOWNORMAL);
Victor Stinner8c62be82010-05-06 00:08:46 +00009311 Py_END_ALLOW_THREADS
9312
Victor Stinnereb5657a2011-09-30 01:44:27 +02009313 Py_XDECREF(uoperation);
Victor Stinner8c62be82010-05-06 00:08:46 +00009314 if (rc <= (HINSTANCE)32) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02009315 win32_error_object("startfile", unipath);
9316 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00009317 }
9318 Py_INCREF(Py_None);
9319 return Py_None;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009320
9321normal:
Victor Stinner8c62be82010-05-06 00:08:46 +00009322 if (!PyArg_ParseTuple(args, "O&|s:startfile",
9323 PyUnicode_FSConverter, &ofilepath,
9324 &operation))
9325 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01009326 if (win32_warn_bytes_api()) {
9327 Py_DECREF(ofilepath);
9328 return NULL;
9329 }
Victor Stinner8c62be82010-05-06 00:08:46 +00009330 filepath = PyBytes_AsString(ofilepath);
9331 Py_BEGIN_ALLOW_THREADS
9332 rc = ShellExecute((HWND)0, operation, filepath,
9333 NULL, NULL, SW_SHOWNORMAL);
9334 Py_END_ALLOW_THREADS
9335 if (rc <= (HINSTANCE)32) {
9336 PyObject *errval = win32_error("startfile", filepath);
9337 Py_DECREF(ofilepath);
9338 return errval;
9339 }
9340 Py_DECREF(ofilepath);
9341 Py_INCREF(Py_None);
9342 return Py_None;
Tim Petersf58a7aa2000-09-22 10:05:54 +00009343}
9344#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009345
Martin v. Löwis438b5342002-12-27 10:16:42 +00009346#ifdef HAVE_GETLOADAVG
9347PyDoc_STRVAR(posix_getloadavg__doc__,
9348"getloadavg() -> (float, float, float)\n\n\
9349Return the number of processes in the system run queue averaged over\n\
9350the last 1, 5, and 15 minutes or raises OSError if the load average\n\
9351was unobtainable");
9352
9353static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00009354posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00009355{
9356 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00009357 if (getloadavg(loadavg, 3)!=3) {
Stefan Krah0e803b32010-11-26 16:16:47 +00009358 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
9359 return NULL;
Martin v. Löwis438b5342002-12-27 10:16:42 +00009360 } else
Stefan Krah0e803b32010-11-26 16:16:47 +00009361 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
Martin v. Löwis438b5342002-12-27 10:16:42 +00009362}
9363#endif
9364
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00009365PyDoc_STRVAR(device_encoding__doc__,
9366"device_encoding(fd) -> str\n\n\
9367Return a string describing the encoding of the device\n\
9368if the output is a terminal; else return None.");
9369
9370static PyObject *
9371device_encoding(PyObject *self, PyObject *args)
9372{
Victor Stinner8c62be82010-05-06 00:08:46 +00009373 int fd;
Brett Cannonefb00c02012-02-29 18:31:31 -05009374
Victor Stinner8c62be82010-05-06 00:08:46 +00009375 if (!PyArg_ParseTuple(args, "i:device_encoding", &fd))
9376 return NULL;
Brett Cannonefb00c02012-02-29 18:31:31 -05009377
9378 return _Py_device_encoding(fd);
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00009379}
9380
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009381#ifdef __VMS
9382/* Use openssl random routine */
9383#include <openssl/rand.h>
9384PyDoc_STRVAR(vms_urandom__doc__,
9385"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00009386Return n random bytes suitable for cryptographic use.");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009387
9388static PyObject*
9389vms_urandom(PyObject *self, PyObject *args)
9390{
Victor Stinner8c62be82010-05-06 00:08:46 +00009391 int howMany;
9392 PyObject* result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009393
Victor Stinner8c62be82010-05-06 00:08:46 +00009394 /* Read arguments */
9395 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
9396 return NULL;
9397 if (howMany < 0)
9398 return PyErr_Format(PyExc_ValueError,
9399 "negative argument not allowed");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009400
Victor Stinner8c62be82010-05-06 00:08:46 +00009401 /* Allocate bytes */
9402 result = PyBytes_FromStringAndSize(NULL, howMany);
9403 if (result != NULL) {
9404 /* Get random data */
9405 if (RAND_pseudo_bytes((unsigned char*)
9406 PyBytes_AS_STRING(result),
9407 howMany) < 0) {
9408 Py_DECREF(result);
9409 return PyErr_Format(PyExc_ValueError,
9410 "RAND_pseudo_bytes");
9411 }
9412 }
9413 return result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009414}
9415#endif
9416
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009417#ifdef HAVE_SETRESUID
9418PyDoc_STRVAR(posix_setresuid__doc__,
9419"setresuid(ruid, euid, suid)\n\n\
9420Set the current process's real, effective, and saved user ids.");
9421
9422static PyObject*
9423posix_setresuid (PyObject *self, PyObject *args)
9424{
Victor Stinner8c62be82010-05-06 00:08:46 +00009425 /* We assume uid_t is no larger than a long. */
9426 long ruid, euid, suid;
9427 if (!PyArg_ParseTuple(args, "lll", &ruid, &euid, &suid))
9428 return NULL;
9429 if (setresuid(ruid, euid, suid) < 0)
9430 return posix_error();
9431 Py_RETURN_NONE;
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009432}
9433#endif
9434
9435#ifdef HAVE_SETRESGID
9436PyDoc_STRVAR(posix_setresgid__doc__,
9437"setresgid(rgid, egid, sgid)\n\n\
9438Set the current process's real, effective, and saved group ids.");
9439
9440static PyObject*
9441posix_setresgid (PyObject *self, PyObject *args)
9442{
Victor Stinner8c62be82010-05-06 00:08:46 +00009443 /* We assume uid_t is no larger than a long. */
9444 long rgid, egid, sgid;
9445 if (!PyArg_ParseTuple(args, "lll", &rgid, &egid, &sgid))
9446 return NULL;
9447 if (setresgid(rgid, egid, sgid) < 0)
9448 return posix_error();
9449 Py_RETURN_NONE;
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009450}
9451#endif
9452
9453#ifdef HAVE_GETRESUID
9454PyDoc_STRVAR(posix_getresuid__doc__,
9455"getresuid() -> (ruid, euid, suid)\n\n\
9456Get tuple of the current process's real, effective, and saved user ids.");
9457
9458static PyObject*
9459posix_getresuid (PyObject *self, PyObject *noargs)
9460{
Victor Stinner8c62be82010-05-06 00:08:46 +00009461 uid_t ruid, euid, suid;
9462 long l_ruid, l_euid, l_suid;
9463 if (getresuid(&ruid, &euid, &suid) < 0)
9464 return posix_error();
9465 /* Force the values into long's as we don't know the size of uid_t. */
9466 l_ruid = ruid;
9467 l_euid = euid;
9468 l_suid = suid;
9469 return Py_BuildValue("(lll)", l_ruid, l_euid, l_suid);
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009470}
9471#endif
9472
9473#ifdef HAVE_GETRESGID
9474PyDoc_STRVAR(posix_getresgid__doc__,
9475"getresgid() -> (rgid, egid, sgid)\n\n\
Georg Brandla9b51d22010-09-05 17:07:12 +00009476Get tuple of the current process's real, effective, and saved group ids.");
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009477
9478static PyObject*
9479posix_getresgid (PyObject *self, PyObject *noargs)
9480{
Victor Stinner8c62be82010-05-06 00:08:46 +00009481 uid_t rgid, egid, sgid;
9482 long l_rgid, l_egid, l_sgid;
9483 if (getresgid(&rgid, &egid, &sgid) < 0)
9484 return posix_error();
9485 /* Force the values into long's as we don't know the size of uid_t. */
9486 l_rgid = rgid;
9487 l_egid = egid;
9488 l_sgid = sgid;
9489 return Py_BuildValue("(lll)", l_rgid, l_egid, l_sgid);
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009490}
9491#endif
9492
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009493/* Posix *at family of functions:
9494 faccessat, fchmodat, fchownat, fstatat, futimesat,
9495 linkat, mkdirat, mknodat, openat, readlinkat, renameat, symlinkat,
9496 unlinkat, utimensat, mkfifoat */
9497
9498#ifdef HAVE_FACCESSAT
9499PyDoc_STRVAR(posix_faccessat__doc__,
9500"faccessat(dirfd, path, mode, flags=0) -> True if granted, False otherwise\n\n\
9501Like access() but if path is relative, it is taken as relative to dirfd.\n\
9502flags is optional and can be constructed by ORing together zero or more\n\
9503of these values: AT_SYMLINK_NOFOLLOW, AT_EACCESS.\n\
9504If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9505is interpreted relative to the current working directory.");
9506
9507static PyObject *
9508posix_faccessat(PyObject *self, PyObject *args)
9509{
9510 PyObject *opath;
9511 char *path;
9512 int mode;
9513 int res;
9514 int dirfd, flags = 0;
9515 if (!PyArg_ParseTuple(args, "iO&i|i:faccessat",
9516 &dirfd, PyUnicode_FSConverter, &opath, &mode, &flags))
9517 return NULL;
9518 path = PyBytes_AsString(opath);
9519 Py_BEGIN_ALLOW_THREADS
9520 res = faccessat(dirfd, path, mode, flags);
9521 Py_END_ALLOW_THREADS
9522 Py_DECREF(opath);
9523 return PyBool_FromLong(res == 0);
9524}
9525#endif
9526
9527#ifdef HAVE_FCHMODAT
9528PyDoc_STRVAR(posix_fchmodat__doc__,
9529"fchmodat(dirfd, path, mode, flags=0)\n\n\
9530Like chmod() but if path is relative, it is taken as relative to dirfd.\n\
9531flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9532If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9533is interpreted relative to the current working directory.");
9534
9535static PyObject *
9536posix_fchmodat(PyObject *self, PyObject *args)
9537{
9538 int dirfd, mode, res;
9539 int flags = 0;
9540 PyObject *opath;
9541 char *path;
9542
9543 if (!PyArg_ParseTuple(args, "iO&i|i:fchmodat",
9544 &dirfd, PyUnicode_FSConverter, &opath, &mode, &flags))
9545 return NULL;
9546
9547 path = PyBytes_AsString(opath);
9548
9549 Py_BEGIN_ALLOW_THREADS
9550 res = fchmodat(dirfd, path, mode, flags);
9551 Py_END_ALLOW_THREADS
9552 Py_DECREF(opath);
9553 if (res < 0)
9554 return posix_error();
9555 Py_RETURN_NONE;
9556}
9557#endif /* HAVE_FCHMODAT */
9558
9559#ifdef HAVE_FCHOWNAT
9560PyDoc_STRVAR(posix_fchownat__doc__,
9561"fchownat(dirfd, path, uid, gid, flags=0)\n\n\
9562Like chown() but if path is relative, it is taken as relative to dirfd.\n\
9563flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9564If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9565is interpreted relative to the current working directory.");
9566
9567static PyObject *
9568posix_fchownat(PyObject *self, PyObject *args)
9569{
9570 PyObject *opath;
9571 int dirfd, res;
9572 long uid, gid;
9573 int flags = 0;
9574 char *path;
Giampaolo Rodola'ff1a7352011-04-19 09:47:16 +02009575
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009576 if (!PyArg_ParseTuple(args, "iO&ll|i:fchownat",
9577 &dirfd, PyUnicode_FSConverter, &opath, &uid, &gid, &flags))
9578 return NULL;
9579
9580 path = PyBytes_AsString(opath);
9581
9582 Py_BEGIN_ALLOW_THREADS
9583 res = fchownat(dirfd, path, (uid_t) uid, (gid_t) gid, flags);
9584 Py_END_ALLOW_THREADS
9585 Py_DECREF(opath);
9586 if (res < 0)
9587 return posix_error();
9588 Py_RETURN_NONE;
9589}
9590#endif /* HAVE_FCHOWNAT */
9591
9592#ifdef HAVE_FSTATAT
9593PyDoc_STRVAR(posix_fstatat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01009594"fstatat(dirfd, path, flags=0) -> stat result\n\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009595Like stat() but if path is relative, it is taken as relative to dirfd.\n\
9596flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9597If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9598is interpreted relative to the current working directory.");
9599
9600static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01009601posix_fstatat(PyObject *self, PyObject *args)
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009602{
9603 PyObject *opath;
9604 char *path;
9605 STRUCT_STAT st;
9606 int dirfd, res, flags = 0;
9607
Victor Stinner4195b5c2012-02-08 23:03:19 +01009608 if (!PyArg_ParseTuple(args, "iO&|i:fstatat",
9609 &dirfd, PyUnicode_FSConverter, &opath, &flags))
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009610 return NULL;
9611 path = PyBytes_AsString(opath);
9612
9613 Py_BEGIN_ALLOW_THREADS
9614 res = fstatat(dirfd, path, &st, flags);
9615 Py_END_ALLOW_THREADS
9616 Py_DECREF(opath);
9617 if (res != 0)
9618 return posix_error();
9619
Victor Stinner4195b5c2012-02-08 23:03:19 +01009620 return _pystat_fromstructstat(&st);
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009621}
9622#endif
9623
9624#ifdef HAVE_FUTIMESAT
9625PyDoc_STRVAR(posix_futimesat__doc__,
Brian Curtin7ef53ef2011-11-07 14:38:24 -06009626"futimesat(dirfd, path[, (atime, mtime)])\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009627Like utime() but if path is relative, it is taken as relative to dirfd.\n\
9628If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9629is interpreted relative to the current working directory.");
9630
9631static PyObject *
9632posix_futimesat(PyObject *self, PyObject *args)
9633{
9634 PyObject *opath;
9635 char *path;
9636 int res, dirfd;
Brian Curtin7ef53ef2011-11-07 14:38:24 -06009637 PyObject* arg = Py_None;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009638 time_t atime, mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009639 long ansec, mnsec;
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009640
Brian Curtin7ef53ef2011-11-07 14:38:24 -06009641 if (!PyArg_ParseTuple(args, "iO&|O:futimesat",
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009642 &dirfd, PyUnicode_FSConverter, &opath, &arg))
9643 return NULL;
9644 path = PyBytes_AsString(opath);
9645 if (arg == Py_None) {
9646 /* optional time values not given */
9647 Py_BEGIN_ALLOW_THREADS
9648 res = futimesat(dirfd, path, NULL);
9649 Py_END_ALLOW_THREADS
9650 }
9651 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
9652 PyErr_SetString(PyExc_TypeError,
9653 "futimesat() arg 3 must be a tuple (atime, mtime)");
9654 Py_DECREF(opath);
9655 return NULL;
9656 }
9657 else {
Victor Stinner5d272cc2012-03-13 13:35:55 +01009658 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 0),
9659 &atime, &ansec) == -1) {
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009660 Py_DECREF(opath);
9661 return NULL;
9662 }
Victor Stinner5d272cc2012-03-13 13:35:55 +01009663 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 1),
9664 &mtime, &mnsec) == -1) {
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009665 Py_DECREF(opath);
9666 return NULL;
9667 }
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009668
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009669 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009670 {
9671#ifdef HAVE_UTIMENSAT
9672 struct timespec buf[2];
9673 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009674 buf[0].tv_nsec = ansec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009675 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009676 buf[1].tv_nsec = mnsec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009677 res = utimensat(dirfd, path, buf, 0);
9678#else
9679 struct timeval buf[2];
9680 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009681 buf[0].tv_usec = ansec / 1000;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009682 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009683 buf[1].tv_usec = mnsec / 1000;
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009684 res = futimesat(dirfd, path, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009685#endif
9686 }
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009687 Py_END_ALLOW_THREADS
9688 }
9689 Py_DECREF(opath);
9690 if (res < 0) {
9691 return posix_error();
9692 }
9693 Py_RETURN_NONE;
9694}
9695#endif
9696
9697#ifdef HAVE_LINKAT
9698PyDoc_STRVAR(posix_linkat__doc__,
9699"linkat(srcfd, srcpath, dstfd, dstpath, flags=0)\n\n\
9700Like link() but if srcpath is relative, it is taken as relative to srcfd\n\
9701and if dstpath is relative, it is taken as relative to dstfd.\n\
9702flags is optional and may be 0 or AT_SYMLINK_FOLLOW.\n\
9703If srcpath is relative and srcfd is the special value AT_FDCWD, then\n\
9704srcpath is interpreted relative to the current working directory. This\n\
9705also applies for dstpath.");
9706
9707static PyObject *
9708posix_linkat(PyObject *self, PyObject *args)
9709{
9710 PyObject *osrc, *odst;
9711 char *src, *dst;
9712 int res, srcfd, dstfd;
9713 int flags = 0;
9714
9715 if (!PyArg_ParseTuple(args, "iO&iO&|i:linkat",
9716 &srcfd, PyUnicode_FSConverter, &osrc, &dstfd, PyUnicode_FSConverter, &odst, &flags))
9717 return NULL;
9718 src = PyBytes_AsString(osrc);
9719 dst = PyBytes_AsString(odst);
9720 Py_BEGIN_ALLOW_THREADS
9721 res = linkat(srcfd, src, dstfd, dst, flags);
9722 Py_END_ALLOW_THREADS
9723 Py_DECREF(osrc);
9724 Py_DECREF(odst);
9725 if (res < 0)
9726 return posix_error();
9727 Py_RETURN_NONE;
9728}
9729#endif /* HAVE_LINKAT */
9730
9731#ifdef HAVE_MKDIRAT
9732PyDoc_STRVAR(posix_mkdirat__doc__,
9733"mkdirat(dirfd, path, mode=0o777)\n\n\
9734Like mkdir() but if path is relative, it is taken as relative to dirfd.\n\
9735If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9736is interpreted relative to the current working directory.");
9737
9738static PyObject *
9739posix_mkdirat(PyObject *self, PyObject *args)
9740{
9741 int res, dirfd;
9742 PyObject *opath;
9743 char *path;
9744 int mode = 0777;
9745
9746 if (!PyArg_ParseTuple(args, "iO&|i:mkdirat",
9747 &dirfd, PyUnicode_FSConverter, &opath, &mode))
9748 return NULL;
9749 path = PyBytes_AsString(opath);
9750 Py_BEGIN_ALLOW_THREADS
9751 res = mkdirat(dirfd, path, mode);
9752 Py_END_ALLOW_THREADS
9753 Py_DECREF(opath);
9754 if (res < 0)
9755 return posix_error();
9756 Py_RETURN_NONE;
9757}
9758#endif
9759
9760#if defined(HAVE_MKNODAT) && defined(HAVE_MAKEDEV)
9761PyDoc_STRVAR(posix_mknodat__doc__,
9762"mknodat(dirfd, path, mode=0o600, device=0)\n\n\
9763Like mknod() but if path is relative, it is taken as relative to dirfd.\n\
9764If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9765is interpreted relative to the current working directory.");
9766
9767static PyObject *
9768posix_mknodat(PyObject *self, PyObject *args)
9769{
9770 PyObject *opath;
9771 char *filename;
9772 int mode = 0600;
9773 int device = 0;
9774 int res, dirfd;
9775 if (!PyArg_ParseTuple(args, "iO&|ii:mknodat", &dirfd,
9776 PyUnicode_FSConverter, &opath, &mode, &device))
9777 return NULL;
9778 filename = PyBytes_AS_STRING(opath);
9779 Py_BEGIN_ALLOW_THREADS
9780 res = mknodat(dirfd, filename, mode, device);
9781 Py_END_ALLOW_THREADS
9782 Py_DECREF(opath);
9783 if (res < 0)
9784 return posix_error();
9785 Py_RETURN_NONE;
9786}
9787#endif
9788
9789#ifdef HAVE_OPENAT
9790PyDoc_STRVAR(posix_openat__doc__,
9791"openat(dirfd, path, flag, mode=0o777) -> fd\n\n\
9792Like open() but if path is relative, it is taken as relative to dirfd.\n\
9793If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9794is interpreted relative to the current working directory.");
9795
9796static PyObject *
9797posix_openat(PyObject *self, PyObject *args)
9798{
9799 PyObject *ofile;
9800 char *file;
9801 int flag, dirfd, fd;
9802 int mode = 0777;
9803
9804 if (!PyArg_ParseTuple(args, "iO&i|i:openat",
9805 &dirfd, PyUnicode_FSConverter, &ofile,
9806 &flag, &mode))
9807 return NULL;
9808 file = PyBytes_AsString(ofile);
9809 Py_BEGIN_ALLOW_THREADS
9810 fd = openat(dirfd, file, flag, mode);
9811 Py_END_ALLOW_THREADS
9812 Py_DECREF(ofile);
9813 if (fd < 0)
9814 return posix_error();
9815 return PyLong_FromLong((long)fd);
9816}
9817#endif
9818
9819#ifdef HAVE_READLINKAT
9820PyDoc_STRVAR(posix_readlinkat__doc__,
9821"readlinkat(dirfd, path) -> path\n\n\
9822Like readlink() but if path is relative, it is taken as relative to dirfd.\n\
9823If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9824is interpreted relative to the current working directory.");
9825
9826static PyObject *
9827posix_readlinkat(PyObject *self, PyObject *args)
9828{
9829 PyObject *v, *opath;
9830 char buf[MAXPATHLEN];
9831 char *path;
9832 int n, dirfd;
9833 int arg_is_unicode = 0;
9834
9835 if (!PyArg_ParseTuple(args, "iO&:readlinkat",
9836 &dirfd, PyUnicode_FSConverter, &opath))
9837 return NULL;
9838 path = PyBytes_AsString(opath);
9839 v = PySequence_GetItem(args, 1);
9840 if (v == NULL) {
9841 Py_DECREF(opath);
9842 return NULL;
9843 }
9844
9845 if (PyUnicode_Check(v)) {
9846 arg_is_unicode = 1;
9847 }
9848 Py_DECREF(v);
9849
9850 Py_BEGIN_ALLOW_THREADS
9851 n = readlinkat(dirfd, path, buf, (int) sizeof buf);
9852 Py_END_ALLOW_THREADS
9853 Py_DECREF(opath);
9854 if (n < 0)
9855 return posix_error();
9856
9857 if (arg_is_unicode)
9858 return PyUnicode_DecodeFSDefaultAndSize(buf, n);
9859 else
9860 return PyBytes_FromStringAndSize(buf, n);
9861}
9862#endif /* HAVE_READLINKAT */
9863
9864#ifdef HAVE_RENAMEAT
9865PyDoc_STRVAR(posix_renameat__doc__,
9866"renameat(olddirfd, oldpath, newdirfd, newpath)\n\n\
9867Like rename() but if oldpath is relative, it is taken as relative to\n\
9868olddirfd and if newpath is relative, it is taken as relative to newdirfd.\n\
9869If oldpath is relative and olddirfd is the special value AT_FDCWD, then\n\
9870oldpath is interpreted relative to the current working directory. This\n\
9871also applies for newpath.");
9872
9873static PyObject *
9874posix_renameat(PyObject *self, PyObject *args)
9875{
9876 int res;
9877 PyObject *opathold, *opathnew;
9878 char *opath, *npath;
9879 int oldfd, newfd;
9880
9881 if (!PyArg_ParseTuple(args, "iO&iO&:renameat",
9882 &oldfd, PyUnicode_FSConverter, &opathold, &newfd, PyUnicode_FSConverter, &opathnew))
9883 return NULL;
9884 opath = PyBytes_AsString(opathold);
9885 npath = PyBytes_AsString(opathnew);
9886 Py_BEGIN_ALLOW_THREADS
9887 res = renameat(oldfd, opath, newfd, npath);
9888 Py_END_ALLOW_THREADS
9889 Py_DECREF(opathold);
9890 Py_DECREF(opathnew);
9891 if (res < 0)
9892 return posix_error();
9893 Py_RETURN_NONE;
9894}
9895#endif
9896
9897#if HAVE_SYMLINKAT
9898PyDoc_STRVAR(posix_symlinkat__doc__,
9899"symlinkat(src, dstfd, dst)\n\n\
9900Like symlink() but if dst is relative, it is taken as relative to dstfd.\n\
9901If dst is relative and dstfd is the special value AT_FDCWD, then dst\n\
9902is interpreted relative to the current working directory.");
9903
9904static PyObject *
9905posix_symlinkat(PyObject *self, PyObject *args)
9906{
9907 int res, dstfd;
9908 PyObject *osrc, *odst;
9909 char *src, *dst;
9910
9911 if (!PyArg_ParseTuple(args, "O&iO&:symlinkat",
9912 PyUnicode_FSConverter, &osrc, &dstfd, PyUnicode_FSConverter, &odst))
9913 return NULL;
9914 src = PyBytes_AsString(osrc);
9915 dst = PyBytes_AsString(odst);
9916 Py_BEGIN_ALLOW_THREADS
9917 res = symlinkat(src, dstfd, dst);
9918 Py_END_ALLOW_THREADS
9919 Py_DECREF(osrc);
9920 Py_DECREF(odst);
9921 if (res < 0)
9922 return posix_error();
9923 Py_RETURN_NONE;
9924}
9925#endif /* HAVE_SYMLINKAT */
9926
9927#ifdef HAVE_UNLINKAT
9928PyDoc_STRVAR(posix_unlinkat__doc__,
9929"unlinkat(dirfd, path, flags=0)\n\n\
9930Like unlink() but if path is relative, it is taken as relative to dirfd.\n\
9931flags is optional and may be 0 or AT_REMOVEDIR. If AT_REMOVEDIR is\n\
9932specified, unlinkat() behaves like rmdir().\n\
9933If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9934is interpreted relative to the current working directory.");
9935
9936static PyObject *
9937posix_unlinkat(PyObject *self, PyObject *args)
9938{
9939 int dirfd, res, flags = 0;
9940 PyObject *opath;
9941 char *path;
9942
9943 if (!PyArg_ParseTuple(args, "iO&|i:unlinkat",
9944 &dirfd, PyUnicode_FSConverter, &opath, &flags))
9945 return NULL;
9946 path = PyBytes_AsString(opath);
9947 Py_BEGIN_ALLOW_THREADS
9948 res = unlinkat(dirfd, path, flags);
9949 Py_END_ALLOW_THREADS
9950 Py_DECREF(opath);
9951 if (res < 0)
9952 return posix_error();
9953 Py_RETURN_NONE;
9954}
9955#endif
9956
9957#ifdef HAVE_UTIMENSAT
9958PyDoc_STRVAR(posix_utimensat__doc__,
Brian Curtin569b4942011-11-07 16:09:20 -06009959"utimensat(dirfd, path[, atime=(atime_sec, atime_nsec),\n\
9960 mtime=(mtime_sec, mtime_nsec), flags=0])\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009961utimensat(dirfd, path, None, None, flags)\n\n\
9962Updates the timestamps of a file with nanosecond precision. If path is\n\
9963relative, it is taken as relative to dirfd.\n\
Brian Curtin569b4942011-11-07 16:09:20 -06009964If atime and mtime are both None, which is the default, set atime and\n\
9965mtime to the current time.\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009966flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9967If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9968is interpreted relative to the current working directory.\n\
9969If *_nsec is specified as UTIME_NOW, the timestamp is updated to the\n\
9970current time.\n\
9971If *_nsec is specified as UTIME_OMIT, the timestamp is not updated.");
9972
9973static PyObject *
Brian Curtin569b4942011-11-07 16:09:20 -06009974posix_utimensat(PyObject *self, PyObject *args, PyObject *kwargs)
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009975{
9976 PyObject *opath;
9977 char *path;
9978 int res, dirfd, flags = 0;
Brian Curtin569b4942011-11-07 16:09:20 -06009979 PyObject *atime = Py_None;
9980 PyObject *mtime = Py_None;
9981
9982 static char *kwlist[] = {"dirfd", "path", "atime", "mtime", "flags", NULL};
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009983
9984 struct timespec buf[2];
9985
Brian Curtin569b4942011-11-07 16:09:20 -06009986 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "iO&|OOi:utimensat", kwlist,
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009987 &dirfd, PyUnicode_FSConverter, &opath, &atime, &mtime, &flags))
9988 return NULL;
9989 path = PyBytes_AsString(opath);
9990 if (atime == Py_None && mtime == Py_None) {
9991 /* optional time values not given */
9992 Py_BEGIN_ALLOW_THREADS
9993 res = utimensat(dirfd, path, NULL, flags);
9994 Py_END_ALLOW_THREADS
9995 }
9996 else if (!PyTuple_Check(atime) || PyTuple_Size(atime) != 2) {
9997 PyErr_SetString(PyExc_TypeError,
9998 "utimensat() arg 3 must be a tuple (atime_sec, atime_nsec)");
9999 Py_DECREF(opath);
10000 return NULL;
10001 }
10002 else if (!PyTuple_Check(mtime) || PyTuple_Size(mtime) != 2) {
10003 PyErr_SetString(PyExc_TypeError,
10004 "utimensat() arg 4 must be a tuple (mtime_sec, mtime_nsec)");
10005 Py_DECREF(opath);
10006 return NULL;
10007 }
10008 else {
10009 if (!PyArg_ParseTuple(atime, "ll:utimensat",
10010 &(buf[0].tv_sec), &(buf[0].tv_nsec))) {
10011 Py_DECREF(opath);
10012 return NULL;
10013 }
10014 if (!PyArg_ParseTuple(mtime, "ll:utimensat",
10015 &(buf[1].tv_sec), &(buf[1].tv_nsec))) {
10016 Py_DECREF(opath);
10017 return NULL;
10018 }
10019 Py_BEGIN_ALLOW_THREADS
10020 res = utimensat(dirfd, path, buf, flags);
10021 Py_END_ALLOW_THREADS
10022 }
10023 Py_DECREF(opath);
10024 if (res < 0) {
10025 return posix_error();
10026 }
10027 Py_RETURN_NONE;
10028}
10029#endif
10030
10031#ifdef HAVE_MKFIFOAT
10032PyDoc_STRVAR(posix_mkfifoat__doc__,
10033"mkfifoat(dirfd, path, mode=0o666)\n\n\
10034Like mkfifo() but if path is relative, it is taken as relative to dirfd.\n\
10035If path is relative and dirfd is the special value AT_FDCWD, then path\n\
10036is interpreted relative to the current working directory.");
10037
10038static PyObject *
10039posix_mkfifoat(PyObject *self, PyObject *args)
10040{
10041 PyObject *opath;
10042 char *filename;
10043 int mode = 0666;
10044 int res, dirfd;
10045 if (!PyArg_ParseTuple(args, "iO&|i:mkfifoat",
10046 &dirfd, PyUnicode_FSConverter, &opath, &mode))
10047 return NULL;
10048 filename = PyBytes_AS_STRING(opath);
10049 Py_BEGIN_ALLOW_THREADS
10050 res = mkfifoat(dirfd, filename, mode);
10051 Py_END_ALLOW_THREADS
10052 Py_DECREF(opath);
10053 if (res < 0)
10054 return posix_error();
10055 Py_RETURN_NONE;
10056}
10057#endif
10058
Benjamin Peterson9428d532011-09-14 11:45:52 -040010059#ifdef USE_XATTRS
Benjamin Peterson799bd802011-08-31 22:15:17 -040010060
10061static int
10062try_getxattr(const char *path, const char *name,
10063 ssize_t (*get)(const char *, const char *, void *, size_t),
10064 Py_ssize_t buf_size, PyObject **res)
10065{
10066 PyObject *value;
10067 Py_ssize_t len;
10068
10069 assert(buf_size <= XATTR_SIZE_MAX);
10070 value = PyBytes_FromStringAndSize(NULL, buf_size);
10071 if (!value)
10072 return 0;
10073 Py_BEGIN_ALLOW_THREADS;
10074 len = get(path, name, PyBytes_AS_STRING(value), buf_size);
10075 Py_END_ALLOW_THREADS;
10076 if (len < 0) {
10077 Py_DECREF(value);
10078 if (errno == ERANGE) {
10079 value = NULL;
10080 }
10081 else {
10082 posix_error();
10083 return 0;
10084 }
10085 }
10086 else if (len != buf_size) {
10087 /* Can only shrink. */
10088 _PyBytes_Resize(&value, len);
10089 }
10090 *res = value;
10091 return 1;
10092}
10093
10094static PyObject *
10095getxattr_common(const char *path, PyObject *name_obj,
10096 ssize_t (*get)(const char *, const char *, void *, size_t))
10097{
10098 PyObject *value;
10099 const char *name = PyBytes_AS_STRING(name_obj);
10100
10101 /* Try a small value first. */
10102 if (!try_getxattr(path, name, get, 128, &value))
10103 return NULL;
10104 if (value)
10105 return value;
10106 /* Now the maximum possible one. */
10107 if (!try_getxattr(path, name, get, XATTR_SIZE_MAX, &value))
10108 return NULL;
10109 assert(value);
10110 return value;
10111}
10112
10113PyDoc_STRVAR(posix_getxattr__doc__,
10114"getxattr(path, attr) -> value\n\n\
10115Return the value of extended attribute *name* on *path*.");
10116
10117static PyObject *
10118posix_getxattr(PyObject *self, PyObject *args)
10119{
10120 PyObject *path, *res, *name;
10121
10122 if (!PyArg_ParseTuple(args, "O&O&:getxattr", PyUnicode_FSConverter, &path,
10123 PyUnicode_FSConverter, &name))
10124 return NULL;
10125 res = getxattr_common(PyBytes_AS_STRING(path), name, getxattr);
10126 Py_DECREF(path);
10127 Py_DECREF(name);
10128 return res;
10129}
10130
10131PyDoc_STRVAR(posix_lgetxattr__doc__,
10132"lgetxattr(path, attr) -> value\n\n\
10133Like getxattr but don't follow symlinks.");
10134
10135static PyObject *
10136posix_lgetxattr(PyObject *self, PyObject *args)
10137{
10138 PyObject *path, *res, *name;
10139
10140 if (!PyArg_ParseTuple(args, "O&O&:lgetxattr", PyUnicode_FSConverter, &path,
10141 PyUnicode_FSConverter, &name))
10142 return NULL;
10143 res = getxattr_common(PyBytes_AS_STRING(path), name, lgetxattr);
10144 Py_DECREF(path);
10145 Py_DECREF(name);
10146 return res;
10147}
10148
10149static ssize_t
10150wrap_fgetxattr(const char *path, const char *name, void *value, size_t size)
10151{
10152 /* Hack to share code. */
10153 return fgetxattr((int)(Py_uintptr_t)path, name, value, size);
10154}
10155
10156PyDoc_STRVAR(posix_fgetxattr__doc__,
10157"fgetxattr(fd, attr) -> value\n\n\
10158Like getxattr but operate on a fd instead of a path.");
10159
10160static PyObject *
10161posix_fgetxattr(PyObject *self, PyObject *args)
10162{
10163 PyObject *res, *name;
10164 int fd;
10165
10166 if (!PyArg_ParseTuple(args, "iO&:fgetxattr", &fd, PyUnicode_FSConverter, &name))
10167 return NULL;
10168 res = getxattr_common((const char *)(Py_uintptr_t)fd, name, wrap_fgetxattr);
10169 Py_DECREF(name);
10170 return res;
10171}
10172
10173PyDoc_STRVAR(posix_setxattr__doc__,
10174"setxattr(path, attr, value, flags=0)\n\n\
10175Set extended attribute *attr* on *path* to *value*.");
10176
10177static PyObject *
10178posix_setxattr(PyObject *self, PyObject *args)
10179{
10180 PyObject *path, *name;
10181 Py_buffer data;
10182 int flags = 0, err;
10183
10184 if (!PyArg_ParseTuple(args, "O&O&y*|i:setxattr", PyUnicode_FSConverter,
10185 &path, PyUnicode_FSConverter, &name, &data, &flags))
10186 return NULL;
10187 Py_BEGIN_ALLOW_THREADS;
10188 err = setxattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name),
10189 data.buf, data.len, flags);
10190 Py_END_ALLOW_THREADS;
10191 Py_DECREF(path);
10192 Py_DECREF(name);
10193 PyBuffer_Release(&data);
10194 if (err)
10195 return posix_error();
10196 Py_RETURN_NONE;
10197}
10198
10199PyDoc_STRVAR(posix_lsetxattr__doc__,
10200"lsetxattr(path, attr, value, flags=0)\n\n\
10201Like setxattr but don't follow symlinks.");
10202
10203static PyObject *
10204posix_lsetxattr(PyObject *self, PyObject *args)
10205{
10206 PyObject *path, *name;
10207 Py_buffer data;
10208 int flags = 0, err;
10209
10210 if (!PyArg_ParseTuple(args, "O&O&y*|i:lsetxattr", PyUnicode_FSConverter,
10211 &path, PyUnicode_FSConverter, &name, &data, &flags))
10212 return NULL;
10213 Py_BEGIN_ALLOW_THREADS;
10214 err = lsetxattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name),
10215 data.buf, data.len, flags);
10216 Py_END_ALLOW_THREADS;
10217 Py_DECREF(path);
10218 Py_DECREF(name);
10219 PyBuffer_Release(&data);
10220 if (err)
10221 return posix_error();
10222 Py_RETURN_NONE;
10223}
10224
10225PyDoc_STRVAR(posix_fsetxattr__doc__,
10226"fsetxattr(fd, attr, value, flags=0)\n\n\
10227Like setxattr but operates on *fd* instead of a path.");
10228
10229static PyObject *
10230posix_fsetxattr(PyObject *self, PyObject *args)
10231{
10232 Py_buffer data;
10233 const char *name;
10234 int fd, flags = 0, err;
10235
10236 if (!PyArg_ParseTuple(args, "iO&y*|i:fsetxattr", &fd, PyUnicode_FSConverter,
10237 &name, &data, &flags))
10238 return NULL;
10239 Py_BEGIN_ALLOW_THREADS;
10240 err = fsetxattr(fd, PyBytes_AS_STRING(name), data.buf, data.len, flags);
10241 Py_END_ALLOW_THREADS;
10242 Py_DECREF(name);
10243 PyBuffer_Release(&data);
10244 if (err)
10245 return posix_error();
10246 Py_RETURN_NONE;
10247}
10248
10249PyDoc_STRVAR(posix_removexattr__doc__,
10250"removexattr(path, attr)\n\n\
10251Remove extended attribute *attr* on *path*.");
10252
10253static PyObject *
10254posix_removexattr(PyObject *self, PyObject *args)
10255{
10256 PyObject *path, *name;
10257 int err;
10258
10259 if (!PyArg_ParseTuple(args, "O&O&:removexattr", PyUnicode_FSConverter, &path,
10260 PyUnicode_FSConverter, &name))
10261 return NULL;
10262 Py_BEGIN_ALLOW_THREADS;
10263 err = removexattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name));
10264 Py_END_ALLOW_THREADS;
10265 Py_DECREF(path);
10266 Py_DECREF(name);
10267 if (err)
10268 return posix_error();
10269 Py_RETURN_NONE;
10270}
10271
10272PyDoc_STRVAR(posix_lremovexattr__doc__,
10273"lremovexattr(path, attr)\n\n\
10274Like removexattr but don't follow symlinks.");
10275
10276static PyObject *
10277posix_lremovexattr(PyObject *self, PyObject *args)
10278{
10279 PyObject *path, *name;
10280 int err;
10281
10282 if (!PyArg_ParseTuple(args, "O&O&:lremovexattr", PyUnicode_FSConverter, &path,
10283 PyUnicode_FSConverter, &name))
10284 return NULL;
10285 Py_BEGIN_ALLOW_THREADS;
10286 err = lremovexattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name));
10287 Py_END_ALLOW_THREADS;
10288 Py_DECREF(path);
10289 Py_DECREF(name);
10290 if (err)
10291 return posix_error();
10292 Py_RETURN_NONE;
10293}
10294
10295PyDoc_STRVAR(posix_fremovexattr__doc__,
10296"fremovexattr(fd, attr)\n\n\
10297Like removexattr but operates on a file descriptor.");
10298
10299static PyObject *
10300posix_fremovexattr(PyObject *self, PyObject *args)
10301{
10302 PyObject *name;
10303 int fd, err;
10304
10305 if (!PyArg_ParseTuple(args, "iO&:fremovexattr", &fd,
10306 PyUnicode_FSConverter, &name))
10307 return NULL;
10308 Py_BEGIN_ALLOW_THREADS;
10309 err = fremovexattr(fd, PyBytes_AS_STRING(name));
10310 Py_END_ALLOW_THREADS;
10311 Py_DECREF(name);
10312 if (err)
10313 return posix_error();
10314 Py_RETURN_NONE;
10315}
10316
10317static Py_ssize_t
10318try_listxattr(const char *path, ssize_t (*list)(const char *, char *, size_t),
10319 Py_ssize_t buf_size, char **buf)
10320{
10321 Py_ssize_t len;
10322
10323 *buf = PyMem_MALLOC(buf_size);
10324 if (!*buf) {
10325 PyErr_NoMemory();
10326 return -1;
10327 }
10328 Py_BEGIN_ALLOW_THREADS;
10329 len = list(path, *buf, buf_size);
10330 Py_END_ALLOW_THREADS;
10331 if (len < 0) {
10332 PyMem_FREE(*buf);
10333 if (errno != ERANGE)
10334 posix_error();
10335 return -1;
10336 }
10337 return len;
10338}
10339
10340static PyObject *
10341listxattr_common(const char *path, ssize_t (*list)(const char *, char *, size_t))
10342{
10343 PyObject *res, *attr;
10344 Py_ssize_t len, err, start, i;
10345 char *buf;
10346
10347 len = try_listxattr(path, list, 256, &buf);
10348 if (len < 0) {
10349 if (PyErr_Occurred())
10350 return NULL;
10351 len = try_listxattr(path, list, XATTR_LIST_MAX, &buf);
10352 if (len < 0)
10353 return NULL;
10354 }
10355 res = PyList_New(0);
10356 if (!res) {
10357 PyMem_FREE(buf);
10358 return NULL;
10359 }
10360 for (start = i = 0; i < len; i++) {
10361 if (!buf[i]) {
10362 attr = PyUnicode_DecodeFSDefaultAndSize(&buf[start], i - start);
10363 if (!attr) {
10364 Py_DECREF(res);
10365 PyMem_FREE(buf);
10366 return NULL;
10367 }
10368 err = PyList_Append(res, attr);
10369 Py_DECREF(attr);
10370 if (err) {
10371 Py_DECREF(res);
10372 PyMem_FREE(buf);
10373 return NULL;
10374 }
10375 start = i + 1;
10376 }
10377 }
10378 PyMem_FREE(buf);
10379 return res;
10380}
10381
10382PyDoc_STRVAR(posix_listxattr__doc__,
10383"listxattr(path)\n\n\
10384Return a list of extended attributes on *path*.");
10385
10386static PyObject *
10387posix_listxattr(PyObject *self, PyObject *args)
10388{
10389 PyObject *path, *res;
10390
10391 if (!PyArg_ParseTuple(args, "O&:listxattr", PyUnicode_FSConverter, &path))
10392 return NULL;
10393 res = listxattr_common(PyBytes_AS_STRING(path), listxattr);
10394 Py_DECREF(path);
10395 return res;
10396}
10397
10398PyDoc_STRVAR(posix_llistxattr__doc__,
10399"llistxattr(path)\n\n\
10400Like listxattr but don't follow symlinks..");
10401
10402static PyObject *
10403posix_llistxattr(PyObject *self, PyObject *args)
10404{
10405 PyObject *path, *res;
10406
10407 if (!PyArg_ParseTuple(args, "O&:llistxattr", PyUnicode_FSConverter, &path))
10408 return NULL;
10409 res = listxattr_common(PyBytes_AS_STRING(path), llistxattr);
10410 Py_DECREF(path);
10411 return res;
10412}
10413
10414static ssize_t
10415wrap_flistxattr(const char *path, char *buf, size_t len)
10416{
10417 /* Hack to share code. */
10418 return flistxattr((int)(Py_uintptr_t)path, buf, len);
10419}
10420
10421PyDoc_STRVAR(posix_flistxattr__doc__,
10422"flistxattr(path)\n\n\
10423Like flistxattr but operates on a file descriptor.");
10424
10425static PyObject *
10426posix_flistxattr(PyObject *self, PyObject *args)
10427{
10428 long fd;
10429
10430 if (!PyArg_ParseTuple(args, "i:flistxattr", &fd))
10431 return NULL;
10432 return listxattr_common((const char *)(Py_uintptr_t)fd, wrap_flistxattr);
10433}
10434
Benjamin Peterson9428d532011-09-14 11:45:52 -040010435#endif /* USE_XATTRS */
Benjamin Peterson799bd802011-08-31 22:15:17 -040010436
Antoine Pitroubcf2b592012-02-08 23:28:36 +010010437
Georg Brandl2fb477c2012-02-21 00:33:36 +010010438PyDoc_STRVAR(posix_urandom__doc__,
10439"urandom(n) -> str\n\n\
10440Return n random bytes suitable for cryptographic use.");
10441
10442static PyObject *
10443posix_urandom(PyObject *self, PyObject *args)
10444{
10445 Py_ssize_t size;
10446 PyObject *result;
10447 int ret;
10448
10449 /* Read arguments */
10450 if (!PyArg_ParseTuple(args, "n:urandom", &size))
10451 return NULL;
10452 if (size < 0)
10453 return PyErr_Format(PyExc_ValueError,
10454 "negative argument not allowed");
10455 result = PyBytes_FromStringAndSize(NULL, size);
10456 if (result == NULL)
10457 return NULL;
10458
10459 ret = _PyOS_URandom(PyBytes_AS_STRING(result),
10460 PyBytes_GET_SIZE(result));
10461 if (ret == -1) {
10462 Py_DECREF(result);
10463 return NULL;
10464 }
10465 return result;
10466}
10467
Antoine Pitroubcf2b592012-02-08 23:28:36 +010010468/* Terminal size querying */
10469
10470static PyTypeObject TerminalSizeType;
10471
10472PyDoc_STRVAR(TerminalSize_docstring,
10473 "A tuple of (columns, lines) for holding terminal window size");
10474
10475static PyStructSequence_Field TerminalSize_fields[] = {
10476 {"columns", "width of the terminal window in characters"},
10477 {"lines", "height of the terminal window in characters"},
10478 {NULL, NULL}
10479};
10480
10481static PyStructSequence_Desc TerminalSize_desc = {
10482 "os.terminal_size",
10483 TerminalSize_docstring,
10484 TerminalSize_fields,
10485 2,
10486};
10487
10488#if defined(TERMSIZE_USE_CONIO) || defined(TERMSIZE_USE_IOCTL)
10489PyDoc_STRVAR(termsize__doc__,
10490 "Return the size of the terminal window as (columns, lines).\n" \
10491 "\n" \
10492 "The optional argument fd (default standard output) specifies\n" \
10493 "which file descriptor should be queried.\n" \
10494 "\n" \
10495 "If the file descriptor is not connected to a terminal, an OSError\n" \
10496 "is thrown.\n" \
10497 "\n" \
10498 "This function will only be defined if an implementation is\n" \
10499 "available for this system.\n" \
10500 "\n" \
10501 "shutil.get_terminal_size is the high-level function which should \n" \
10502 "normally be used, os.get_terminal_size is the low-level implementation.");
10503
10504static PyObject*
10505get_terminal_size(PyObject *self, PyObject *args)
10506{
10507 int columns, lines;
10508 PyObject *termsize;
10509
10510 int fd = fileno(stdout);
10511 /* Under some conditions stdout may not be connected and
10512 * fileno(stdout) may point to an invalid file descriptor. For example
10513 * GUI apps don't have valid standard streams by default.
10514 *
10515 * If this happens, and the optional fd argument is not present,
10516 * the ioctl below will fail returning EBADF. This is what we want.
10517 */
10518
10519 if (!PyArg_ParseTuple(args, "|i", &fd))
10520 return NULL;
10521
10522#ifdef TERMSIZE_USE_IOCTL
10523 {
10524 struct winsize w;
10525 if (ioctl(fd, TIOCGWINSZ, &w))
10526 return PyErr_SetFromErrno(PyExc_OSError);
10527 columns = w.ws_col;
10528 lines = w.ws_row;
10529 }
10530#endif /* TERMSIZE_USE_IOCTL */
10531
10532#ifdef TERMSIZE_USE_CONIO
10533 {
10534 DWORD nhandle;
10535 HANDLE handle;
10536 CONSOLE_SCREEN_BUFFER_INFO csbi;
10537 switch (fd) {
10538 case 0: nhandle = STD_INPUT_HANDLE;
10539 break;
10540 case 1: nhandle = STD_OUTPUT_HANDLE;
10541 break;
10542 case 2: nhandle = STD_ERROR_HANDLE;
10543 break;
10544 default:
10545 return PyErr_Format(PyExc_ValueError, "bad file descriptor");
10546 }
10547 handle = GetStdHandle(nhandle);
10548 if (handle == NULL)
10549 return PyErr_Format(PyExc_OSError, "handle cannot be retrieved");
10550 if (handle == INVALID_HANDLE_VALUE)
10551 return PyErr_SetFromWindowsErr(0);
10552
10553 if (!GetConsoleScreenBufferInfo(handle, &csbi))
10554 return PyErr_SetFromWindowsErr(0);
10555
10556 columns = csbi.srWindow.Right - csbi.srWindow.Left + 1;
10557 lines = csbi.srWindow.Bottom - csbi.srWindow.Top + 1;
10558 }
10559#endif /* TERMSIZE_USE_CONIO */
10560
10561 termsize = PyStructSequence_New(&TerminalSizeType);
10562 if (termsize == NULL)
10563 return NULL;
10564 PyStructSequence_SET_ITEM(termsize, 0, PyLong_FromLong(columns));
10565 PyStructSequence_SET_ITEM(termsize, 1, PyLong_FromLong(lines));
10566 if (PyErr_Occurred()) {
10567 Py_DECREF(termsize);
10568 return NULL;
10569 }
10570 return termsize;
10571}
10572#endif /* defined(TERMSIZE_USE_CONIO) || defined(TERMSIZE_USE_IOCTL) */
10573
10574
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010575static PyMethodDef posix_methods[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +000010576 {"access", posix_access, METH_VARARGS, posix_access__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010577#ifdef HAVE_TTYNAME
Victor Stinner8c62be82010-05-06 00:08:46 +000010578 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010579#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010580 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +000010581#ifdef HAVE_CHFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +000010582 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +000010583#endif /* HAVE_CHFLAGS */
Victor Stinner8c62be82010-05-06 00:08:46 +000010584 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010585#ifdef HAVE_FCHMOD
Victor Stinner8c62be82010-05-06 00:08:46 +000010586 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010587#endif /* HAVE_FCHMOD */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010588#ifdef HAVE_CHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010589 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010590#endif /* HAVE_CHOWN */
Christian Heimes4e30a842007-11-30 22:12:06 +000010591#ifdef HAVE_LCHMOD
Victor Stinner8c62be82010-05-06 00:08:46 +000010592 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010593#endif /* HAVE_LCHMOD */
10594#ifdef HAVE_FCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010595 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010596#endif /* HAVE_FCHOWN */
Thomas Wouterscf297e42007-02-23 15:07:44 +000010597#ifdef HAVE_LCHFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +000010598 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +000010599#endif /* HAVE_LCHFLAGS */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +000010600#ifdef HAVE_LCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010601 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +000010602#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +000010603#ifdef HAVE_CHROOT
Victor Stinner8c62be82010-05-06 00:08:46 +000010604 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
Martin v. Löwis244edc82001-10-04 22:44:26 +000010605#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010606#ifdef HAVE_CTERMID
Victor Stinner8c62be82010-05-06 00:08:46 +000010607 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010608#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010609#ifdef HAVE_GETCWD
Victor Stinner8c62be82010-05-06 00:08:46 +000010610 {"getcwd", (PyCFunction)posix_getcwd_unicode,
10611 METH_NOARGS, posix_getcwd__doc__},
10612 {"getcwdb", (PyCFunction)posix_getcwd_bytes,
10613 METH_NOARGS, posix_getcwdb__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010614#endif
10615#ifdef HAVE_LINK
Victor Stinner8c62be82010-05-06 00:08:46 +000010616 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010617#endif /* HAVE_LINK */
Victor Stinner8c62be82010-05-06 00:08:46 +000010618 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
Antoine Pitrou8250e232011-02-25 23:41:16 +000010619#ifdef HAVE_FDOPENDIR
Charles-François Natali77940902012-02-06 19:54:48 +010010620 {"flistdir", posix_flistdir, METH_VARARGS, posix_flistdir__doc__},
Antoine Pitrou8250e232011-02-25 23:41:16 +000010621#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +010010622 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +000010623 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010624#ifdef HAVE_NICE
Victor Stinner8c62be82010-05-06 00:08:46 +000010625 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum775f4da1993-01-09 17:18:52 +000010626#endif /* HAVE_NICE */
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +000010627#ifdef HAVE_GETPRIORITY
10628 {"getpriority", posix_getpriority, METH_VARARGS, posix_getpriority__doc__},
10629#endif /* HAVE_GETPRIORITY */
10630#ifdef HAVE_SETPRIORITY
10631 {"setpriority", posix_setpriority, METH_VARARGS, posix_setpriority__doc__},
10632#endif /* HAVE_SETPRIORITY */
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010633#ifdef HAVE_READLINK
Victor Stinner8c62be82010-05-06 00:08:46 +000010634 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010635#endif /* HAVE_READLINK */
Brian Curtind40e6f72010-07-08 21:39:08 +000010636#if !defined(HAVE_READLINK) && defined(MS_WINDOWS)
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +000010637 {"readlink", win_readlink, METH_VARARGS, win_readlink__doc__},
Brian Curtind40e6f72010-07-08 21:39:08 +000010638#endif /* !defined(HAVE_READLINK) && defined(MS_WINDOWS) */
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +000010639 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
Antoine Pitrouf3b2d882012-01-30 22:08:52 +010010640 {"replace", posix_replace, METH_VARARGS, posix_replace__doc__},
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +000010641 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
Victor Stinner4195b5c2012-02-08 23:03:19 +010010642 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +000010643 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Brian Curtin52173d42010-12-02 18:29:18 +000010644#if defined(HAVE_SYMLINK) && !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +000010645 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010646#endif /* HAVE_SYMLINK */
Brian Curtin52173d42010-12-02 18:29:18 +000010647#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000010648 {"symlink", (PyCFunction)win_symlink, METH_VARARGS | METH_KEYWORDS,
Brian Curtin52173d42010-12-02 18:29:18 +000010649 win_symlink__doc__},
10650#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
Guido van Rossum85e3b011991-06-03 12:42:10 +000010651#ifdef HAVE_SYSTEM
Victor Stinner8c62be82010-05-06 00:08:46 +000010652 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010653#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010654 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010655#ifdef HAVE_UNAME
Victor Stinner8c62be82010-05-06 00:08:46 +000010656 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010657#endif /* HAVE_UNAME */
Victor Stinner8c62be82010-05-06 00:08:46 +000010658 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
10659 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
Larry Hastings76ad59b2012-05-03 00:30:07 -070010660 {"utime", (PyCFunction)posix_utime,
10661 METH_VARARGS | METH_KEYWORDS, posix_utime__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010662#ifdef HAVE_FUTIMES
Larry Hastings76ad59b2012-05-03 00:30:07 -070010663 {"futimes", (PyCFunction)posix_futimes,
10664 METH_VARARGS | METH_KEYWORDS, posix_futimes__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010665#endif
10666#ifdef HAVE_LUTIMES
Larry Hastings76ad59b2012-05-03 00:30:07 -070010667 {"lutimes", (PyCFunction)posix_lutimes,
10668 METH_VARARGS | METH_KEYWORDS, posix_lutimes__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010669#endif
Guido van Rossum0ee42cd1991-04-08 21:01:03 +000010670#ifdef HAVE_TIMES
Victor Stinner8c62be82010-05-06 00:08:46 +000010671 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum3b066191991-06-04 19:40:25 +000010672#endif /* HAVE_TIMES */
Victor Stinner8c62be82010-05-06 00:08:46 +000010673 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossum3b066191991-06-04 19:40:25 +000010674#ifdef HAVE_EXECV
Victor Stinner8c62be82010-05-06 00:08:46 +000010675 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
10676 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossum3b066191991-06-04 19:40:25 +000010677#endif /* HAVE_EXECV */
Ross Lagerwall7807c352011-03-17 20:20:30 +020010678#ifdef HAVE_FEXECVE
10679 {"fexecve", posix_fexecve, METH_VARARGS, posix_fexecve__doc__},
10680#endif
Guido van Rossuma1065681999-01-25 23:20:23 +000010681#ifdef HAVE_SPAWNV
Victor Stinner8c62be82010-05-06 00:08:46 +000010682 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
10683 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +000010684#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +000010685 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
10686 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +000010687#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +000010688#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +000010689#ifdef HAVE_FORK1
Victor Stinner8c62be82010-05-06 00:08:46 +000010690 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +000010691#endif /* HAVE_FORK1 */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010692#ifdef HAVE_FORK
Victor Stinner8c62be82010-05-06 00:08:46 +000010693 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010694#endif /* HAVE_FORK */
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010695#ifdef HAVE_SCHED_H
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +020010696#ifdef HAVE_SCHED_GET_PRIORITY_MAX
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010697 {"sched_get_priority_max", posix_sched_get_priority_max, METH_VARARGS, posix_sched_get_priority_max__doc__},
10698 {"sched_get_priority_min", posix_sched_get_priority_min, METH_VARARGS, posix_sched_get_priority_min__doc__},
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +020010699#endif
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010700#ifdef HAVE_SCHED_SETPARAM
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010701 {"sched_getparam", posix_sched_getparam, METH_VARARGS, posix_sched_getparam__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010702#endif
10703#ifdef HAVE_SCHED_SETSCHEDULER
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010704 {"sched_getscheduler", posix_sched_getscheduler, METH_VARARGS, posix_sched_getscheduler__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010705#endif
10706#ifdef HAVE_SCHED_RR_GET_INTERVAL
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010707 {"sched_rr_get_interval", posix_sched_rr_get_interval, METH_VARARGS, posix_sched_rr_get_interval__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010708#endif
10709#ifdef HAVE_SCHED_SETPARAM
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010710 {"sched_setparam", posix_sched_setparam, METH_VARARGS, posix_sched_setparam__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010711#endif
10712#ifdef HAVE_SCHED_SETSCHEDULER
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010713 {"sched_setscheduler", posix_sched_setscheduler, METH_VARARGS, posix_sched_setscheduler__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010714#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010715 {"sched_yield", posix_sched_yield, METH_NOARGS, posix_sched_yield__doc__},
Benjamin Peterson2740af82011-08-02 17:41:34 -050010716#ifdef HAVE_SCHED_SETAFFINITY
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010717 {"sched_setaffinity", posix_sched_setaffinity, METH_VARARGS, posix_sched_setaffinity__doc__},
10718 {"sched_getaffinity", posix_sched_getaffinity, METH_VARARGS, posix_sched_getaffinity__doc__},
10719#endif
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +020010720#endif /* HAVE_SCHED_H */
Martin v. Löwis24a880b2002-12-31 12:55:15 +000010721#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Victor Stinner8c62be82010-05-06 00:08:46 +000010722 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +000010723#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +000010724#ifdef HAVE_FORKPTY
Victor Stinner8c62be82010-05-06 00:08:46 +000010725 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +000010726#endif /* HAVE_FORKPTY */
Guido van Rossum3b066191991-06-04 19:40:25 +000010727#ifdef HAVE_GETEGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010728 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +000010729#endif /* HAVE_GETEGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010730#ifdef HAVE_GETEUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010731 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossum46003ff1992-05-15 11:05:24 +000010732#endif /* HAVE_GETEUID */
10733#ifdef HAVE_GETGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010734 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010735#endif /* HAVE_GETGID */
Ross Lagerwallb0ae53d2011-06-10 07:30:30 +020010736#ifdef HAVE_GETGROUPLIST
10737 {"getgrouplist", posix_getgrouplist, METH_VARARGS, posix_getgrouplist__doc__},
10738#endif
Fred Drakec9680921999-12-13 16:37:25 +000010739#ifdef HAVE_GETGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +000010740 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010741#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010742 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +000010743#ifdef HAVE_GETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010744 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010745#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +000010746#ifdef HAVE_GETPPID
Victor Stinner8c62be82010-05-06 00:08:46 +000010747 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010748#endif /* HAVE_GETPPID */
10749#ifdef HAVE_GETUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010750 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010751#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +000010752#ifdef HAVE_GETLOGIN
Victor Stinner8c62be82010-05-06 00:08:46 +000010753 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +000010754#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +000010755#ifdef HAVE_KILL
Victor Stinner8c62be82010-05-06 00:08:46 +000010756 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010757#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +000010758#ifdef HAVE_KILLPG
Victor Stinner8c62be82010-05-06 00:08:46 +000010759 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
Martin v. Löwisb2c92f42002-02-16 23:35:41 +000010760#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +000010761#ifdef HAVE_PLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000010762 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +000010763#endif /* HAVE_PLOCK */
Thomas Heller8b7a9572007-08-31 06:44:36 +000010764#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +000010765 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
10766 {"kill", win32_kill, METH_VARARGS, win32_kill__doc__},
Brian Curtin1b9df392010-11-24 20:24:31 +000010767 {"link", win32_link, METH_VARARGS, win32_link__doc__},
Thomas Heller8b7a9572007-08-31 06:44:36 +000010768#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +000010769#ifdef HAVE_SETUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010770 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010771#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010772#ifdef HAVE_SETEUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010773 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010774#endif /* HAVE_SETEUID */
10775#ifdef HAVE_SETEGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010776 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010777#endif /* HAVE_SETEGID */
10778#ifdef HAVE_SETREUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010779 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010780#endif /* HAVE_SETREUID */
10781#ifdef HAVE_SETREGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010782 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010783#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010784#ifdef HAVE_SETGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010785 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010786#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +000010787#ifdef HAVE_SETGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +000010788 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
Martin v. Löwis61c5edf2001-10-18 04:06:00 +000010789#endif /* HAVE_SETGROUPS */
Antoine Pitroub7572f02009-12-02 20:46:48 +000010790#ifdef HAVE_INITGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +000010791 {"initgroups", posix_initgroups, METH_VARARGS, posix_initgroups__doc__},
Antoine Pitroub7572f02009-12-02 20:46:48 +000010792#endif /* HAVE_INITGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +000010793#ifdef HAVE_GETPGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010794 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
Martin v. Löwis606edc12002-06-13 21:09:11 +000010795#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010796#ifdef HAVE_SETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010797 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010798#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +000010799#ifdef HAVE_WAIT
Victor Stinner8c62be82010-05-06 00:08:46 +000010800 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010801#endif /* HAVE_WAIT */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010802#ifdef HAVE_WAIT3
Victor Stinner4195b5c2012-02-08 23:03:19 +010010803 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010804#endif /* HAVE_WAIT3 */
10805#ifdef HAVE_WAIT4
Victor Stinner4195b5c2012-02-08 23:03:19 +010010806 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010807#endif /* HAVE_WAIT4 */
Ross Lagerwall7807c352011-03-17 20:20:30 +020010808#if defined(HAVE_WAITID) && !defined(__APPLE__)
10809 {"waitid", posix_waitid, METH_VARARGS, posix_waitid__doc__},
10810#endif
Tim Petersab034fa2002-02-01 11:27:43 +000010811#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Victor Stinner8c62be82010-05-06 00:08:46 +000010812 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010813#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +000010814#ifdef HAVE_GETSID
Victor Stinner8c62be82010-05-06 00:08:46 +000010815 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
Martin v. Löwis49ee14d2003-11-10 06:35:36 +000010816#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010817#ifdef HAVE_SETSID
Victor Stinner8c62be82010-05-06 00:08:46 +000010818 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010819#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010820#ifdef HAVE_SETPGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010821 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010822#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010823#ifdef HAVE_TCGETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010824 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010825#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010826#ifdef HAVE_TCSETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010827 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010828#endif /* HAVE_TCSETPGRP */
Victor Stinner8c62be82010-05-06 00:08:46 +000010829 {"open", posix_open, METH_VARARGS, posix_open__doc__},
10830 {"close", posix_close, METH_VARARGS, posix_close__doc__},
10831 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
10832 {"device_encoding", device_encoding, METH_VARARGS, device_encoding__doc__},
10833 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
10834 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010835#ifdef HAVE_LOCKF
10836 {"lockf", posix_lockf, METH_VARARGS, posix_lockf__doc__},
10837#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010838 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
10839 {"read", posix_read, METH_VARARGS, posix_read__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010840#ifdef HAVE_READV
10841 {"readv", posix_readv, METH_VARARGS, posix_readv__doc__},
10842#endif
10843#ifdef HAVE_PREAD
10844 {"pread", posix_pread, METH_VARARGS, posix_pread__doc__},
10845#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010846 {"write", posix_write, METH_VARARGS, posix_write__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010847#ifdef HAVE_WRITEV
10848 {"writev", posix_writev, METH_VARARGS, posix_writev__doc__},
10849#endif
10850#ifdef HAVE_PWRITE
10851 {"pwrite", posix_pwrite, METH_VARARGS, posix_pwrite__doc__},
10852#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +000010853#ifdef HAVE_SENDFILE
10854 {"sendfile", (PyCFunction)posix_sendfile, METH_VARARGS | METH_KEYWORDS,
10855 posix_sendfile__doc__},
10856#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +010010857 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +000010858 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010859#ifdef HAVE_PIPE
Victor Stinner8c62be82010-05-06 00:08:46 +000010860 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010861#endif
Charles-François Natalidaafdd52011-05-29 20:07:40 +020010862#ifdef HAVE_PIPE2
Charles-François Natali368f34b2011-06-06 19:49:47 +020010863 {"pipe2", posix_pipe2, METH_O, posix_pipe2__doc__},
Charles-François Natalidaafdd52011-05-29 20:07:40 +020010864#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010865#ifdef HAVE_MKFIFO
Victor Stinner8c62be82010-05-06 00:08:46 +000010866 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010867#endif
Neal Norwitz11690112002-07-30 01:08:28 +000010868#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Victor Stinner8c62be82010-05-06 00:08:46 +000010869 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
Martin v. Löwis06a83e92002-04-14 10:19:44 +000010870#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +000010871#ifdef HAVE_DEVICE_MACROS
Victor Stinner8c62be82010-05-06 00:08:46 +000010872 {"major", posix_major, METH_VARARGS, posix_major__doc__},
10873 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
10874 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
Martin v. Löwisdbe3f762002-10-10 14:27:30 +000010875#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010876#ifdef HAVE_FTRUNCATE
Victor Stinner8c62be82010-05-06 00:08:46 +000010877 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010878#endif
Ross Lagerwall7807c352011-03-17 20:20:30 +020010879#ifdef HAVE_TRUNCATE
10880 {"truncate", posix_truncate, METH_VARARGS, posix_truncate__doc__},
10881#endif
10882#ifdef HAVE_POSIX_FALLOCATE
10883 {"posix_fallocate", posix_posix_fallocate, METH_VARARGS, posix_posix_fallocate__doc__},
10884#endif
10885#ifdef HAVE_POSIX_FADVISE
10886 {"posix_fadvise", posix_posix_fadvise, METH_VARARGS, posix_posix_fadvise__doc__},
10887#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +000010888#ifdef HAVE_PUTENV
Victor Stinner8c62be82010-05-06 00:08:46 +000010889 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +000010890#endif
Guido van Rossumc524d952001-10-19 01:31:59 +000010891#ifdef HAVE_UNSETENV
Victor Stinner8c62be82010-05-06 00:08:46 +000010892 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
Guido van Rossumc524d952001-10-19 01:31:59 +000010893#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010894 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +000010895#ifdef HAVE_FCHDIR
Victor Stinner8c62be82010-05-06 00:08:46 +000010896 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +000010897#endif
Guido van Rossum21142a01999-01-08 21:05:37 +000010898#ifdef HAVE_FSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000010899 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +000010900#endif
Ross Lagerwall7807c352011-03-17 20:20:30 +020010901#ifdef HAVE_SYNC
10902 {"sync", posix_sync, METH_NOARGS, posix_sync__doc__},
10903#endif
Guido van Rossum21142a01999-01-08 21:05:37 +000010904#ifdef HAVE_FDATASYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000010905 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +000010906#endif
Guido van Rossumc9641791998-08-04 15:26:23 +000010907#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +000010908#ifdef WCOREDUMP
Victor Stinner8c62be82010-05-06 00:08:46 +000010909 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
Fred Drake106c1a02002-04-23 15:58:02 +000010910#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +000010911#ifdef WIFCONTINUED
Victor Stinner8c62be82010-05-06 00:08:46 +000010912 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +000010913#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +000010914#ifdef WIFSTOPPED
Victor Stinner8c62be82010-05-06 00:08:46 +000010915 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010916#endif /* WIFSTOPPED */
10917#ifdef WIFSIGNALED
Victor Stinner8c62be82010-05-06 00:08:46 +000010918 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010919#endif /* WIFSIGNALED */
10920#ifdef WIFEXITED
Victor Stinner8c62be82010-05-06 00:08:46 +000010921 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010922#endif /* WIFEXITED */
10923#ifdef WEXITSTATUS
Victor Stinner8c62be82010-05-06 00:08:46 +000010924 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010925#endif /* WEXITSTATUS */
10926#ifdef WTERMSIG
Victor Stinner8c62be82010-05-06 00:08:46 +000010927 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010928#endif /* WTERMSIG */
10929#ifdef WSTOPSIG
Victor Stinner8c62be82010-05-06 00:08:46 +000010930 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010931#endif /* WSTOPSIG */
10932#endif /* HAVE_SYS_WAIT_H */
Thomas Wouters477c8d52006-05-27 19:21:47 +000010933#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Victor Stinner8c62be82010-05-06 00:08:46 +000010934 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +000010935#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +000010936#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Victor Stinner8c62be82010-05-06 00:08:46 +000010937 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +000010938#endif
Fred Drakec9680921999-12-13 16:37:25 +000010939#ifdef HAVE_CONFSTR
Victor Stinner8c62be82010-05-06 00:08:46 +000010940 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010941#endif
10942#ifdef HAVE_SYSCONF
Victor Stinner8c62be82010-05-06 00:08:46 +000010943 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010944#endif
10945#ifdef HAVE_FPATHCONF
Victor Stinner8c62be82010-05-06 00:08:46 +000010946 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010947#endif
10948#ifdef HAVE_PATHCONF
Victor Stinner8c62be82010-05-06 00:08:46 +000010949 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010950#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010951 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000010952#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +000010953 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
Brian Curtind40e6f72010-07-08 21:39:08 +000010954 {"_getfinalpathname", posix__getfinalpathname, METH_VARARGS, NULL},
Brian Curtin62857742010-09-06 17:07:27 +000010955 {"_getfileinformation", posix__getfileinformation, METH_VARARGS, NULL},
Brian Curtin95d028f2011-06-09 09:10:38 -050010956 {"_isdir", posix__isdir, METH_VARARGS, posix__isdir__doc__},
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +020010957 {"_getdiskusage", win32__getdiskusage, METH_VARARGS, win32__getdiskusage__doc__},
Mark Hammondef8b6542001-05-13 08:04:26 +000010958#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +000010959#ifdef HAVE_GETLOADAVG
Victor Stinner8c62be82010-05-06 00:08:46 +000010960 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +000010961#endif
Georg Brandl2daf6ae2012-02-20 19:54:16 +010010962 {"urandom", posix_urandom, METH_VARARGS, posix_urandom__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010963#ifdef HAVE_SETRESUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010964 {"setresuid", posix_setresuid, METH_VARARGS, posix_setresuid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010965#endif
10966#ifdef HAVE_SETRESGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010967 {"setresgid", posix_setresgid, METH_VARARGS, posix_setresgid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010968#endif
10969#ifdef HAVE_GETRESUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010970 {"getresuid", posix_getresuid, METH_NOARGS, posix_getresuid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010971#endif
10972#ifdef HAVE_GETRESGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010973 {"getresgid", posix_getresgid, METH_NOARGS, posix_getresgid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010974#endif
10975
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010976/* posix *at family of functions */
10977#ifdef HAVE_FACCESSAT
10978 {"faccessat", posix_faccessat, METH_VARARGS, posix_faccessat__doc__},
10979#endif
10980#ifdef HAVE_FCHMODAT
10981 {"fchmodat", posix_fchmodat, METH_VARARGS, posix_fchmodat__doc__},
10982#endif /* HAVE_FCHMODAT */
10983#ifdef HAVE_FCHOWNAT
10984 {"fchownat", posix_fchownat, METH_VARARGS, posix_fchownat__doc__},
10985#endif /* HAVE_FCHOWNAT */
10986#ifdef HAVE_FSTATAT
Victor Stinner4195b5c2012-02-08 23:03:19 +010010987 {"fstatat", posix_fstatat, METH_VARARGS, posix_fstatat__doc__},
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010988#endif
10989#ifdef HAVE_FUTIMESAT
10990 {"futimesat", posix_futimesat, METH_VARARGS, posix_futimesat__doc__},
10991#endif
10992#ifdef HAVE_LINKAT
10993 {"linkat", posix_linkat, METH_VARARGS, posix_linkat__doc__},
10994#endif /* HAVE_LINKAT */
10995#ifdef HAVE_MKDIRAT
10996 {"mkdirat", posix_mkdirat, METH_VARARGS, posix_mkdirat__doc__},
10997#endif
10998#if defined(HAVE_MKNODAT) && defined(HAVE_MAKEDEV)
10999 {"mknodat", posix_mknodat, METH_VARARGS, posix_mknodat__doc__},
11000#endif
11001#ifdef HAVE_OPENAT
11002 {"openat", posix_openat, METH_VARARGS, posix_openat__doc__},
11003#endif
11004#ifdef HAVE_READLINKAT
11005 {"readlinkat", posix_readlinkat, METH_VARARGS, posix_readlinkat__doc__},
11006#endif /* HAVE_READLINKAT */
11007#ifdef HAVE_RENAMEAT
11008 {"renameat", posix_renameat, METH_VARARGS, posix_renameat__doc__},
11009#endif
11010#if HAVE_SYMLINKAT
11011 {"symlinkat", posix_symlinkat, METH_VARARGS, posix_symlinkat__doc__},
11012#endif /* HAVE_SYMLINKAT */
11013#ifdef HAVE_UNLINKAT
11014 {"unlinkat", posix_unlinkat, METH_VARARGS, posix_unlinkat__doc__},
11015#endif
11016#ifdef HAVE_UTIMENSAT
Jesus Cead03a4912011-11-08 17:28:04 +010011017 {"utimensat", (PyCFunction)posix_utimensat,
11018 METH_VARARGS | METH_KEYWORDS,
Brian Curtin569b4942011-11-07 16:09:20 -060011019 posix_utimensat__doc__},
Antoine Pitrouf65132d2011-02-25 23:25:17 +000011020#endif
11021#ifdef HAVE_MKFIFOAT
11022 {"mkfifoat", posix_mkfifoat, METH_VARARGS, posix_mkfifoat__doc__},
11023#endif
Benjamin Peterson9428d532011-09-14 11:45:52 -040011024#ifdef USE_XATTRS
Benjamin Peterson799bd802011-08-31 22:15:17 -040011025 {"setxattr", posix_setxattr, METH_VARARGS, posix_setxattr__doc__},
11026 {"lsetxattr", posix_lsetxattr, METH_VARARGS, posix_lsetxattr__doc__},
11027 {"fsetxattr", posix_fsetxattr, METH_VARARGS, posix_fsetxattr__doc__},
11028 {"getxattr", posix_getxattr, METH_VARARGS, posix_getxattr__doc__},
11029 {"lgetxattr", posix_lgetxattr, METH_VARARGS, posix_lgetxattr__doc__},
11030 {"fgetxattr", posix_fgetxattr, METH_VARARGS, posix_fgetxattr__doc__},
11031 {"removexattr", posix_removexattr, METH_VARARGS, posix_removexattr__doc__},
11032 {"lremovexattr", posix_lremovexattr, METH_VARARGS, posix_lremovexattr__doc__},
11033 {"fremovexattr", posix_fremovexattr, METH_VARARGS, posix_fremovexattr__doc__},
11034 {"listxattr", posix_listxattr, METH_VARARGS, posix_listxattr__doc__},
11035 {"llistxattr", posix_llistxattr, METH_VARARGS, posix_llistxattr__doc__},
11036 {"flistxattr", posix_flistxattr, METH_VARARGS, posix_flistxattr__doc__},
11037#endif
Antoine Pitroubcf2b592012-02-08 23:28:36 +010011038#if defined(TERMSIZE_USE_CONIO) || defined(TERMSIZE_USE_IOCTL)
11039 {"get_terminal_size", get_terminal_size, METH_VARARGS, termsize__doc__},
11040#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011041 {NULL, NULL} /* Sentinel */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000011042};
11043
11044
Barry Warsaw4a342091996-12-19 23:50:02 +000011045static int
Fred Drake4d1e64b2002-04-15 19:40:07 +000011046ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +000011047{
Victor Stinner8c62be82010-05-06 00:08:46 +000011048 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +000011049}
11050
Guido van Rossumd48f2521997-12-05 22:19:34 +000011051#if defined(PYOS_OS2)
11052/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +000011053static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +000011054{
11055 APIRET rc;
11056 ULONG values[QSV_MAX+1];
11057 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +000011058 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +000011059
11060 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +000011061 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +000011062 Py_END_ALLOW_THREADS
11063
11064 if (rc != NO_ERROR) {
11065 os2_error(rc);
11066 return -1;
11067 }
11068
Fred Drake4d1e64b2002-04-15 19:40:07 +000011069 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
11070 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
11071 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
11072 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
11073 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
11074 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
11075 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000011076
11077 switch (values[QSV_VERSION_MINOR]) {
11078 case 0: ver = "2.00"; break;
11079 case 10: ver = "2.10"; break;
11080 case 11: ver = "2.11"; break;
11081 case 30: ver = "3.00"; break;
11082 case 40: ver = "4.00"; break;
11083 case 50: ver = "5.00"; break;
11084 default:
Tim Peters885d4572001-11-28 20:27:42 +000011085 PyOS_snprintf(tmp, sizeof(tmp),
Victor Stinner8c62be82010-05-06 00:08:46 +000011086 "%d-%d", values[QSV_VERSION_MAJOR],
Tim Peters885d4572001-11-28 20:27:42 +000011087 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +000011088 ver = &tmp[0];
11089 }
11090
11091 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +000011092 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +000011093 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000011094
11095 /* Add Indicator of Which Drive was Used to Boot the System */
11096 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
11097 tmp[1] = ':';
11098 tmp[2] = '\0';
11099
Fred Drake4d1e64b2002-04-15 19:40:07 +000011100 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +000011101}
11102#endif
11103
Brian Curtin52173d42010-12-02 18:29:18 +000011104#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000011105static int
Brian Curtin52173d42010-12-02 18:29:18 +000011106enable_symlink()
11107{
11108 HANDLE tok;
11109 TOKEN_PRIVILEGES tok_priv;
11110 LUID luid;
11111 int meth_idx = 0;
11112
11113 if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &tok))
Brian Curtin3b4499c2010-12-28 14:31:47 +000011114 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +000011115
11116 if (!LookupPrivilegeValue(NULL, SE_CREATE_SYMBOLIC_LINK_NAME, &luid))
Brian Curtin3b4499c2010-12-28 14:31:47 +000011117 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +000011118
11119 tok_priv.PrivilegeCount = 1;
11120 tok_priv.Privileges[0].Luid = luid;
11121 tok_priv.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
11122
11123 if (!AdjustTokenPrivileges(tok, FALSE, &tok_priv,
11124 sizeof(TOKEN_PRIVILEGES),
11125 (PTOKEN_PRIVILEGES) NULL, (PDWORD) NULL))
Brian Curtin3b4499c2010-12-28 14:31:47 +000011126 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +000011127
Brian Curtin3b4499c2010-12-28 14:31:47 +000011128 /* ERROR_NOT_ALL_ASSIGNED returned when the privilege can't be assigned. */
11129 return GetLastError() == ERROR_NOT_ALL_ASSIGNED ? 0 : 1;
Brian Curtin52173d42010-12-02 18:29:18 +000011130}
11131#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
11132
Barry Warsaw4a342091996-12-19 23:50:02 +000011133static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +000011134all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +000011135{
Guido van Rossum94f6f721999-01-06 18:42:14 +000011136#ifdef F_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011137 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011138#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +000011139#ifdef R_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011140 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011141#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +000011142#ifdef W_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011143 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011144#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +000011145#ifdef X_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011146 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011147#endif
Fred Drakec9680921999-12-13 16:37:25 +000011148#ifdef NGROUPS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +000011149 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
Fred Drakec9680921999-12-13 16:37:25 +000011150#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +000011151#ifdef TMP_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +000011152 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
Fred Drake5ab8eaf1999-12-09 21:13:07 +000011153#endif
Fred Drake106c1a02002-04-23 15:58:02 +000011154#ifdef WCONTINUED
Victor Stinner8c62be82010-05-06 00:08:46 +000011155 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
Fred Drake106c1a02002-04-23 15:58:02 +000011156#endif
Barry Warsaw4a342091996-12-19 23:50:02 +000011157#ifdef WNOHANG
Victor Stinner8c62be82010-05-06 00:08:46 +000011158 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011159#endif
Fred Drake106c1a02002-04-23 15:58:02 +000011160#ifdef WUNTRACED
Victor Stinner8c62be82010-05-06 00:08:46 +000011161 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
Fred Drake106c1a02002-04-23 15:58:02 +000011162#endif
Barry Warsaw4a342091996-12-19 23:50:02 +000011163#ifdef O_RDONLY
Victor Stinner8c62be82010-05-06 00:08:46 +000011164 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011165#endif
11166#ifdef O_WRONLY
Victor Stinner8c62be82010-05-06 00:08:46 +000011167 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011168#endif
11169#ifdef O_RDWR
Victor Stinner8c62be82010-05-06 00:08:46 +000011170 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011171#endif
11172#ifdef O_NDELAY
Victor Stinner8c62be82010-05-06 00:08:46 +000011173 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011174#endif
11175#ifdef O_NONBLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000011176 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011177#endif
11178#ifdef O_APPEND
Victor Stinner8c62be82010-05-06 00:08:46 +000011179 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011180#endif
11181#ifdef O_DSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011182 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011183#endif
11184#ifdef O_RSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011185 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011186#endif
11187#ifdef O_SYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011188 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011189#endif
11190#ifdef O_NOCTTY
Victor Stinner8c62be82010-05-06 00:08:46 +000011191 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011192#endif
11193#ifdef O_CREAT
Victor Stinner8c62be82010-05-06 00:08:46 +000011194 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011195#endif
11196#ifdef O_EXCL
Victor Stinner8c62be82010-05-06 00:08:46 +000011197 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011198#endif
11199#ifdef O_TRUNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011200 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011201#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +000011202#ifdef O_BINARY
Victor Stinner8c62be82010-05-06 00:08:46 +000011203 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
Guido van Rossum98d9d091997-08-08 21:48:51 +000011204#endif
11205#ifdef O_TEXT
Victor Stinner8c62be82010-05-06 00:08:46 +000011206 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
Guido van Rossum98d9d091997-08-08 21:48:51 +000011207#endif
Jesus Cea1d642d22012-04-24 20:59:17 +020011208#ifdef O_XATTR
11209 if (ins(d, "O_XATTR", (long)O_XATTR)) return -1;
11210#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011211#ifdef O_LARGEFILE
Victor Stinner8c62be82010-05-06 00:08:46 +000011212 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011213#endif
Skip Montanaro5ff14922005-05-16 02:42:22 +000011214#ifdef O_SHLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000011215 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
Skip Montanaro5ff14922005-05-16 02:42:22 +000011216#endif
11217#ifdef O_EXLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000011218 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
Skip Montanaro5ff14922005-05-16 02:42:22 +000011219#endif
Jesus Ceacf381202012-04-24 20:44:40 +020011220#ifdef O_EXEC
11221 if (ins(d, "O_EXEC", (long)O_EXEC)) return -1;
11222#endif
11223#ifdef O_SEARCH
11224 if (ins(d, "O_SEARCH", (long)O_SEARCH)) return -1;
11225#endif
11226#ifdef O_TTY_INIT
11227 if (ins(d, "O_TTY_INIT", (long)O_TTY_INIT)) return -1;
11228#endif
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +000011229#ifdef PRIO_PROCESS
11230 if (ins(d, "PRIO_PROCESS", (long)PRIO_PROCESS)) return -1;
11231#endif
11232#ifdef PRIO_PGRP
11233 if (ins(d, "PRIO_PGRP", (long)PRIO_PGRP)) return -1;
11234#endif
11235#ifdef PRIO_USER
11236 if (ins(d, "PRIO_USER", (long)PRIO_USER)) return -1;
11237#endif
Charles-François Natali1e045b12011-05-22 20:42:32 +020011238#ifdef O_CLOEXEC
11239 if (ins(d, "O_CLOEXEC", (long)O_CLOEXEC)) return -1;
11240#endif
Jesus Cea1d642d22012-04-24 20:59:17 +020011241#ifdef O_ACCMODE
11242 if (ins(d, "O_ACCMODE", (long)O_ACCMODE)) return -1;
11243#endif
Antoine Pitrouf65132d2011-02-25 23:25:17 +000011244/* posix - constants for *at functions */
11245#ifdef AT_SYMLINK_NOFOLLOW
11246 if (ins(d, "AT_SYMLINK_NOFOLLOW", (long)AT_SYMLINK_NOFOLLOW)) return -1;
11247#endif
11248#ifdef AT_EACCESS
11249 if (ins(d, "AT_EACCESS", (long)AT_EACCESS)) return -1;
11250#endif
11251#ifdef AT_FDCWD
11252 if (ins(d, "AT_FDCWD", (long)AT_FDCWD)) return -1;
11253#endif
11254#ifdef AT_REMOVEDIR
11255 if (ins(d, "AT_REMOVEDIR", (long)AT_REMOVEDIR)) return -1;
11256#endif
11257#ifdef AT_SYMLINK_FOLLOW
11258 if (ins(d, "AT_SYMLINK_FOLLOW", (long)AT_SYMLINK_FOLLOW)) return -1;
11259#endif
11260#ifdef UTIME_NOW
11261 if (ins(d, "UTIME_NOW", (long)UTIME_NOW)) return -1;
11262#endif
11263#ifdef UTIME_OMIT
11264 if (ins(d, "UTIME_OMIT", (long)UTIME_OMIT)) return -1;
11265#endif
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +000011266
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011267
Tim Peters5aa91602002-01-30 05:46:57 +000011268/* MS Windows */
11269#ifdef O_NOINHERIT
Victor Stinner8c62be82010-05-06 00:08:46 +000011270 /* Don't inherit in child processes. */
11271 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011272#endif
11273#ifdef _O_SHORT_LIVED
Victor Stinner8c62be82010-05-06 00:08:46 +000011274 /* Optimize for short life (keep in memory). */
11275 /* MS forgot to define this one with a non-underscore form too. */
11276 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011277#endif
11278#ifdef O_TEMPORARY
Victor Stinner8c62be82010-05-06 00:08:46 +000011279 /* Automatically delete when last handle is closed. */
11280 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011281#endif
11282#ifdef O_RANDOM
Victor Stinner8c62be82010-05-06 00:08:46 +000011283 /* Optimize for random access. */
11284 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011285#endif
11286#ifdef O_SEQUENTIAL
Victor Stinner8c62be82010-05-06 00:08:46 +000011287 /* Optimize for sequential access. */
11288 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011289#endif
11290
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011291/* GNU extensions. */
Alexandre Vassalottibee32532008-05-16 18:15:12 +000011292#ifdef O_ASYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011293 /* Send a SIGIO signal whenever input or output
11294 becomes available on file descriptor */
11295 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
Alexandre Vassalottibee32532008-05-16 18:15:12 +000011296#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011297#ifdef O_DIRECT
Victor Stinner8c62be82010-05-06 00:08:46 +000011298 /* Direct disk access. */
11299 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011300#endif
11301#ifdef O_DIRECTORY
Victor Stinner8c62be82010-05-06 00:08:46 +000011302 /* Must be a directory. */
11303 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011304#endif
11305#ifdef O_NOFOLLOW
Victor Stinner8c62be82010-05-06 00:08:46 +000011306 /* Do not follow links. */
11307 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011308#endif
Jesus Cea1d642d22012-04-24 20:59:17 +020011309#ifdef O_NOLINKS
11310 /* Fails if link count of the named file is greater than 1 */
11311 if (ins(d, "O_NOLINKS", (long)O_NOLINKS)) return -1;
11312#endif
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +000011313#ifdef O_NOATIME
Victor Stinner8c62be82010-05-06 00:08:46 +000011314 /* Do not update the access time. */
11315 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +000011316#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +000011317
Victor Stinner8c62be82010-05-06 00:08:46 +000011318 /* These come from sysexits.h */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011319#ifdef EX_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011320 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011321#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011322#ifdef EX_USAGE
Victor Stinner8c62be82010-05-06 00:08:46 +000011323 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011324#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011325#ifdef EX_DATAERR
Victor Stinner8c62be82010-05-06 00:08:46 +000011326 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011327#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011328#ifdef EX_NOINPUT
Victor Stinner8c62be82010-05-06 00:08:46 +000011329 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011330#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011331#ifdef EX_NOUSER
Victor Stinner8c62be82010-05-06 00:08:46 +000011332 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011333#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011334#ifdef EX_NOHOST
Victor Stinner8c62be82010-05-06 00:08:46 +000011335 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011336#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011337#ifdef EX_UNAVAILABLE
Victor Stinner8c62be82010-05-06 00:08:46 +000011338 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011339#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011340#ifdef EX_SOFTWARE
Victor Stinner8c62be82010-05-06 00:08:46 +000011341 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011342#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011343#ifdef EX_OSERR
Victor Stinner8c62be82010-05-06 00:08:46 +000011344 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011345#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011346#ifdef EX_OSFILE
Victor Stinner8c62be82010-05-06 00:08:46 +000011347 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011348#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011349#ifdef EX_CANTCREAT
Victor Stinner8c62be82010-05-06 00:08:46 +000011350 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011351#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011352#ifdef EX_IOERR
Victor Stinner8c62be82010-05-06 00:08:46 +000011353 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011354#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011355#ifdef EX_TEMPFAIL
Victor Stinner8c62be82010-05-06 00:08:46 +000011356 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011357#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011358#ifdef EX_PROTOCOL
Victor Stinner8c62be82010-05-06 00:08:46 +000011359 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011360#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011361#ifdef EX_NOPERM
Victor Stinner8c62be82010-05-06 00:08:46 +000011362 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011363#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011364#ifdef EX_CONFIG
Victor Stinner8c62be82010-05-06 00:08:46 +000011365 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011366#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011367#ifdef EX_NOTFOUND
Victor Stinner8c62be82010-05-06 00:08:46 +000011368 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011369#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011370
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +000011371 /* statvfs */
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +000011372#ifdef ST_RDONLY
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +000011373 if (ins(d, "ST_RDONLY", (long)ST_RDONLY)) return -1;
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +000011374#endif /* ST_RDONLY */
11375#ifdef ST_NOSUID
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +000011376 if (ins(d, "ST_NOSUID", (long)ST_NOSUID)) return -1;
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +000011377#endif /* ST_NOSUID */
11378
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +000011379 /* FreeBSD sendfile() constants */
11380#ifdef SF_NODISKIO
11381 if (ins(d, "SF_NODISKIO", (long)SF_NODISKIO)) return -1;
11382#endif
11383#ifdef SF_MNOWAIT
11384 if (ins(d, "SF_MNOWAIT", (long)SF_MNOWAIT)) return -1;
11385#endif
11386#ifdef SF_SYNC
11387 if (ins(d, "SF_SYNC", (long)SF_SYNC)) return -1;
11388#endif
11389
Ross Lagerwall7807c352011-03-17 20:20:30 +020011390 /* constants for posix_fadvise */
11391#ifdef POSIX_FADV_NORMAL
11392 if (ins(d, "POSIX_FADV_NORMAL", (long)POSIX_FADV_NORMAL)) return -1;
11393#endif
11394#ifdef POSIX_FADV_SEQUENTIAL
11395 if (ins(d, "POSIX_FADV_SEQUENTIAL", (long)POSIX_FADV_SEQUENTIAL)) return -1;
11396#endif
11397#ifdef POSIX_FADV_RANDOM
11398 if (ins(d, "POSIX_FADV_RANDOM", (long)POSIX_FADV_RANDOM)) return -1;
11399#endif
11400#ifdef POSIX_FADV_NOREUSE
11401 if (ins(d, "POSIX_FADV_NOREUSE", (long)POSIX_FADV_NOREUSE)) return -1;
11402#endif
11403#ifdef POSIX_FADV_WILLNEED
11404 if (ins(d, "POSIX_FADV_WILLNEED", (long)POSIX_FADV_WILLNEED)) return -1;
11405#endif
11406#ifdef POSIX_FADV_DONTNEED
11407 if (ins(d, "POSIX_FADV_DONTNEED", (long)POSIX_FADV_DONTNEED)) return -1;
11408#endif
11409
11410 /* constants for waitid */
11411#if defined(HAVE_SYS_WAIT_H) && defined(HAVE_WAITID)
11412 if (ins(d, "P_PID", (long)P_PID)) return -1;
11413 if (ins(d, "P_PGID", (long)P_PGID)) return -1;
11414 if (ins(d, "P_ALL", (long)P_ALL)) return -1;
11415#endif
11416#ifdef WEXITED
11417 if (ins(d, "WEXITED", (long)WEXITED)) return -1;
11418#endif
11419#ifdef WNOWAIT
11420 if (ins(d, "WNOWAIT", (long)WNOWAIT)) return -1;
11421#endif
11422#ifdef WSTOPPED
11423 if (ins(d, "WSTOPPED", (long)WSTOPPED)) return -1;
11424#endif
11425#ifdef CLD_EXITED
11426 if (ins(d, "CLD_EXITED", (long)CLD_EXITED)) return -1;
11427#endif
11428#ifdef CLD_DUMPED
11429 if (ins(d, "CLD_DUMPED", (long)CLD_DUMPED)) return -1;
11430#endif
11431#ifdef CLD_TRAPPED
11432 if (ins(d, "CLD_TRAPPED", (long)CLD_TRAPPED)) return -1;
11433#endif
11434#ifdef CLD_CONTINUED
11435 if (ins(d, "CLD_CONTINUED", (long)CLD_CONTINUED)) return -1;
11436#endif
11437
11438 /* constants for lockf */
11439#ifdef F_LOCK
11440 if (ins(d, "F_LOCK", (long)F_LOCK)) return -1;
11441#endif
11442#ifdef F_TLOCK
11443 if (ins(d, "F_TLOCK", (long)F_TLOCK)) return -1;
11444#endif
11445#ifdef F_ULOCK
11446 if (ins(d, "F_ULOCK", (long)F_ULOCK)) return -1;
11447#endif
11448#ifdef F_TEST
11449 if (ins(d, "F_TEST", (long)F_TEST)) return -1;
11450#endif
11451
11452 /* constants for futimens */
11453#ifdef UTIME_NOW
11454 if (ins(d, "UTIME_NOW", (long)UTIME_NOW)) return -1;
11455#endif
11456#ifdef UTIME_OMIT
11457 if (ins(d, "UTIME_OMIT", (long)UTIME_OMIT)) return -1;
11458#endif
11459
Guido van Rossum246bc171999-02-01 23:54:31 +000011460#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000011461#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +000011462 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
11463 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
11464 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
11465 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
11466 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
11467 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
11468 if (ins(d, "P_PM", (long)P_PM)) return -1;
11469 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
11470 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
11471 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
11472 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
11473 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
11474 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
11475 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
11476 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
11477 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
11478 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
11479 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
11480 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
11481 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000011482#else
Victor Stinner8c62be82010-05-06 00:08:46 +000011483 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
11484 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
11485 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
11486 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
11487 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +000011488#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000011489#endif
Guido van Rossum246bc171999-02-01 23:54:31 +000011490
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011491#ifdef HAVE_SCHED_H
Jesus Ceaf2cb4e82011-09-09 23:55:42 +020011492 if (ins(d, "SCHED_OTHER", (long)SCHED_OTHER)) return -1;
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011493 if (ins(d, "SCHED_FIFO", (long)SCHED_FIFO)) return -1;
11494 if (ins(d, "SCHED_RR", (long)SCHED_RR)) return -1;
11495#ifdef SCHED_SPORADIC
11496 if (ins(d, "SCHED_SPORADIC", (long)SCHED_SPORADIC) return -1;
11497#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011498#ifdef SCHED_BATCH
11499 if (ins(d, "SCHED_BATCH", (long)SCHED_BATCH)) return -1;
11500#endif
11501#ifdef SCHED_IDLE
11502 if (ins(d, "SCHED_IDLE", (long)SCHED_IDLE)) return -1;
11503#endif
11504#ifdef SCHED_RESET_ON_FORK
11505 if (ins(d, "SCHED_RESET_ON_FORK", (long)SCHED_RESET_ON_FORK)) return -1;
11506#endif
Jesus Ceaf2cb4e82011-09-09 23:55:42 +020011507#ifdef SCHED_SYS
11508 if (ins(d, "SCHED_SYS", (long)SCHED_SYS)) return -1;
11509#endif
11510#ifdef SCHED_IA
11511 if (ins(d, "SCHED_IA", (long)SCHED_IA)) return -1;
11512#endif
11513#ifdef SCHED_FSS
11514 if (ins(d, "SCHED_FSS", (long)SCHED_FSS)) return -1;
11515#endif
11516#ifdef SCHED_FX
11517 if (ins(d, "SCHED_FX", (long)SCHED_FSS)) return -1;
11518#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011519#endif
11520
Benjamin Peterson9428d532011-09-14 11:45:52 -040011521#ifdef USE_XATTRS
Benjamin Peterson799bd802011-08-31 22:15:17 -040011522 if (ins(d, "XATTR_CREATE", (long)XATTR_CREATE)) return -1;
11523 if (ins(d, "XATTR_REPLACE", (long)XATTR_REPLACE)) return -1;
11524 if (ins(d, "XATTR_SIZE_MAX", (long)XATTR_SIZE_MAX)) return -1;
11525#endif
11526
Victor Stinner8b905bd2011-10-25 13:34:04 +020011527#ifdef RTLD_LAZY
11528 if (PyModule_AddIntMacro(d, RTLD_LAZY)) return -1;
11529#endif
11530#ifdef RTLD_NOW
11531 if (PyModule_AddIntMacro(d, RTLD_NOW)) return -1;
11532#endif
11533#ifdef RTLD_GLOBAL
11534 if (PyModule_AddIntMacro(d, RTLD_GLOBAL)) return -1;
11535#endif
11536#ifdef RTLD_LOCAL
11537 if (PyModule_AddIntMacro(d, RTLD_LOCAL)) return -1;
11538#endif
11539#ifdef RTLD_NODELETE
11540 if (PyModule_AddIntMacro(d, RTLD_NODELETE)) return -1;
11541#endif
11542#ifdef RTLD_NOLOAD
11543 if (PyModule_AddIntMacro(d, RTLD_NOLOAD)) return -1;
11544#endif
11545#ifdef RTLD_DEEPBIND
11546 if (PyModule_AddIntMacro(d, RTLD_DEEPBIND)) return -1;
11547#endif
11548
Guido van Rossumd48f2521997-12-05 22:19:34 +000011549#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +000011550 if (insertvalues(d)) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000011551#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011552 return 0;
Barry Warsaw4a342091996-12-19 23:50:02 +000011553}
11554
11555
Tim Peters5aa91602002-01-30 05:46:57 +000011556#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Martin v. Löwis1a214512008-06-11 05:26:20 +000011557#define INITFUNC PyInit_nt
Guido van Rossum0cb96de1997-10-01 04:29:29 +000011558#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +000011559
11560#elif defined(PYOS_OS2)
Martin v. Löwis1a214512008-06-11 05:26:20 +000011561#define INITFUNC PyInit_os2
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000011562#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +000011563
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000011564#else
Martin v. Löwis1a214512008-06-11 05:26:20 +000011565#define INITFUNC PyInit_posix
Guido van Rossum0cb96de1997-10-01 04:29:29 +000011566#define MODNAME "posix"
11567#endif
11568
Martin v. Löwis1a214512008-06-11 05:26:20 +000011569static struct PyModuleDef posixmodule = {
Victor Stinner8c62be82010-05-06 00:08:46 +000011570 PyModuleDef_HEAD_INIT,
11571 MODNAME,
11572 posix__doc__,
11573 -1,
11574 posix_methods,
11575 NULL,
11576 NULL,
11577 NULL,
11578 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +000011579};
11580
11581
Mark Hammondfe51c6d2002-08-02 02:27:13 +000011582PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +000011583INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +000011584{
Victor Stinner8c62be82010-05-06 00:08:46 +000011585 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +000011586
Brian Curtin52173d42010-12-02 18:29:18 +000011587#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000011588 win32_can_symlink = enable_symlink();
Brian Curtin52173d42010-12-02 18:29:18 +000011589#endif
11590
Victor Stinner8c62be82010-05-06 00:08:46 +000011591 m = PyModule_Create(&posixmodule);
11592 if (m == NULL)
11593 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +000011594
Victor Stinner8c62be82010-05-06 00:08:46 +000011595 /* Initialize environ dictionary */
11596 v = convertenviron();
11597 Py_XINCREF(v);
11598 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
11599 return NULL;
11600 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +000011601
Victor Stinner8c62be82010-05-06 00:08:46 +000011602 if (all_ins(m))
11603 return NULL;
Barry Warsaw4a342091996-12-19 23:50:02 +000011604
Victor Stinner8c62be82010-05-06 00:08:46 +000011605 if (setup_confname_tables(m))
11606 return NULL;
Fred Drakebec628d1999-12-15 18:31:10 +000011607
Victor Stinner8c62be82010-05-06 00:08:46 +000011608 Py_INCREF(PyExc_OSError);
11609 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +000011610
Benjamin Peterson2740af82011-08-02 17:41:34 -050011611#ifdef HAVE_SCHED_SETAFFINITY
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011612 if (PyType_Ready(&cpu_set_type) < 0)
11613 return NULL;
11614 Py_INCREF(&cpu_set_type);
11615 PyModule_AddObject(m, "cpu_set", (PyObject *)&cpu_set_type);
Benjamin Peterson2740af82011-08-02 17:41:34 -050011616#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011617
Guido van Rossumb3d39562000-01-31 18:41:26 +000011618#ifdef HAVE_PUTENV
Victor Stinner8c62be82010-05-06 00:08:46 +000011619 if (posix_putenv_garbage == NULL)
11620 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +000011621#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +000011622
Victor Stinner8c62be82010-05-06 00:08:46 +000011623 if (!initialized) {
Ross Lagerwall7807c352011-03-17 20:20:30 +020011624#if defined(HAVE_WAITID) && !defined(__APPLE__)
11625 waitid_result_desc.name = MODNAME ".waitid_result";
11626 PyStructSequence_InitType(&WaitidResultType, &waitid_result_desc);
11627#endif
11628
Victor Stinner8c62be82010-05-06 00:08:46 +000011629 stat_result_desc.name = MODNAME ".stat_result";
11630 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
11631 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
11632 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
11633 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
11634 structseq_new = StatResultType.tp_new;
11635 StatResultType.tp_new = statresult_new;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011636
Victor Stinner8c62be82010-05-06 00:08:46 +000011637 statvfs_result_desc.name = MODNAME ".statvfs_result";
11638 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011639#ifdef NEED_TICKS_PER_SECOND
11640# if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
Victor Stinner8c62be82010-05-06 00:08:46 +000011641 ticks_per_second = sysconf(_SC_CLK_TCK);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011642# elif defined(HZ)
Victor Stinner8c62be82010-05-06 00:08:46 +000011643 ticks_per_second = HZ;
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011644# else
Victor Stinner8c62be82010-05-06 00:08:46 +000011645 ticks_per_second = 60; /* magic fallback value; may be bogus */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011646# endif
11647#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011648
Benjamin Peterson0163c9a2011-08-02 18:11:38 -050011649#if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER)
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011650 sched_param_desc.name = MODNAME ".sched_param";
11651 PyStructSequence_InitType(&SchedParamType, &sched_param_desc);
11652 SchedParamType.tp_new = sched_param_new;
11653#endif
Antoine Pitroubcf2b592012-02-08 23:28:36 +010011654
11655 /* initialize TerminalSize_info */
11656 PyStructSequence_InitType(&TerminalSizeType, &TerminalSize_desc);
11657 Py_INCREF(&TerminalSizeType);
Victor Stinner8c62be82010-05-06 00:08:46 +000011658 }
Ross Lagerwall7807c352011-03-17 20:20:30 +020011659#if defined(HAVE_WAITID) && !defined(__APPLE__)
11660 Py_INCREF((PyObject*) &WaitidResultType);
11661 PyModule_AddObject(m, "waitid_result", (PyObject*) &WaitidResultType);
11662#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011663 Py_INCREF((PyObject*) &StatResultType);
11664 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
11665 Py_INCREF((PyObject*) &StatVFSResultType);
11666 PyModule_AddObject(m, "statvfs_result",
11667 (PyObject*) &StatVFSResultType);
Benjamin Petersone3298dd2011-08-02 18:40:46 -050011668
11669#if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER)
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011670 Py_INCREF(&SchedParamType);
11671 PyModule_AddObject(m, "sched_param", (PyObject *)&SchedParamType);
Benjamin Petersone3298dd2011-08-02 18:40:46 -050011672#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011673 initialized = 1;
Thomas Wouters477c8d52006-05-27 19:21:47 +000011674
11675#ifdef __APPLE__
Victor Stinner8c62be82010-05-06 00:08:46 +000011676 /*
11677 * Step 2 of weak-linking support on Mac OS X.
11678 *
11679 * The code below removes functions that are not available on the
11680 * currently active platform.
11681 *
11682 * This block allow one to use a python binary that was build on
11683 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
11684 * OSX 10.4.
11685 */
Thomas Wouters477c8d52006-05-27 19:21:47 +000011686#ifdef HAVE_FSTATVFS
Victor Stinner8c62be82010-05-06 00:08:46 +000011687 if (fstatvfs == NULL) {
11688 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
11689 return NULL;
11690 }
11691 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011692#endif /* HAVE_FSTATVFS */
11693
11694#ifdef HAVE_STATVFS
Victor Stinner8c62be82010-05-06 00:08:46 +000011695 if (statvfs == NULL) {
11696 if (PyObject_DelAttrString(m, "statvfs") == -1) {
11697 return NULL;
11698 }
11699 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011700#endif /* HAVE_STATVFS */
11701
11702# ifdef HAVE_LCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000011703 if (lchown == NULL) {
11704 if (PyObject_DelAttrString(m, "lchown") == -1) {
11705 return NULL;
11706 }
11707 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011708#endif /* HAVE_LCHOWN */
11709
11710
11711#endif /* __APPLE__ */
Antoine Pitroubcf2b592012-02-08 23:28:36 +010011712
11713 PyModule_AddObject(m, "terminal_size", (PyObject*) &TerminalSizeType);
11714
Larry Hastings6fe20b32012-04-19 15:07:49 -070011715 billion = PyLong_FromLong(1000000000);
11716 if (!billion)
11717 return NULL;
11718
Victor Stinner8c62be82010-05-06 00:08:46 +000011719 return m;
Thomas Wouters477c8d52006-05-27 19:21:47 +000011720
Guido van Rossumb6775db1994-08-01 11:34:53 +000011721}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011722
11723#ifdef __cplusplus
11724}
11725#endif