blob: 92a627745851b7240ddbc65454a4409006ba080a [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
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003575PyDoc_STRVAR(posix_utime__doc__,
Brian Curtinc1b65d12011-11-07 14:18:54 -06003576"utime(path[, (atime, mtime)])\n\
3577Set the access and modified time of the file to the given values.\n\
3578If no second argument is used, set the access and modified times to\n\
3579the current time.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003580
Barry Warsaw53699e91996-12-10 23:23:01 +00003581static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003582posix_utime(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003583{
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003584#ifdef MS_WINDOWS
Brian Curtinb0d5b5d2011-11-07 10:51:18 -06003585 PyObject *arg = Py_None;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003586 PyObject *obwpath;
Victor Stinner8c62be82010-05-06 00:08:46 +00003587 wchar_t *wpath = NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003588 const char *apath;
Victor Stinner8c62be82010-05-06 00:08:46 +00003589 HANDLE hFile;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003590 time_t atimesec, mtimesec;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003591 long ansec, mnsec;
Victor Stinner8c62be82010-05-06 00:08:46 +00003592 FILETIME atime, mtime;
3593 PyObject *result = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003594
Brian Curtin52fbea12011-11-06 13:41:17 -06003595 if (PyArg_ParseTuple(args, "U|O:utime", &obwpath, &arg)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02003596 wpath = PyUnicode_AsUnicode(obwpath);
3597 if (wpath == NULL)
3598 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00003599 Py_BEGIN_ALLOW_THREADS
3600 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
3601 NULL, OPEN_EXISTING,
3602 FILE_FLAG_BACKUP_SEMANTICS, NULL);
3603 Py_END_ALLOW_THREADS
3604 if (hFile == INVALID_HANDLE_VALUE)
Victor Stinnereb5657a2011-09-30 01:44:27 +02003605 return win32_error_object("utime", obwpath);
3606 }
3607 else {
Victor Stinner8c62be82010-05-06 00:08:46 +00003608 /* Drop the argument parsing error as narrow strings
3609 are also valid. */
3610 PyErr_Clear();
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00003611
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003612 if (!PyArg_ParseTuple(args, "y|O:utime", &apath, &arg))
3613 return NULL;
3614 if (win32_warn_bytes_api())
Victor Stinner8c62be82010-05-06 00:08:46 +00003615 return NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003616
Victor Stinner8c62be82010-05-06 00:08:46 +00003617 Py_BEGIN_ALLOW_THREADS
3618 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
3619 NULL, OPEN_EXISTING,
3620 FILE_FLAG_BACKUP_SEMANTICS, NULL);
3621 Py_END_ALLOW_THREADS
3622 if (hFile == INVALID_HANDLE_VALUE) {
3623 win32_error("utime", apath);
Victor Stinner8c62be82010-05-06 00:08:46 +00003624 return NULL;
3625 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003626 }
3627
Brian Curtinb0d5b5d2011-11-07 10:51:18 -06003628 if (arg == Py_None) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003629 SYSTEMTIME now;
3630 GetSystemTime(&now);
3631 if (!SystemTimeToFileTime(&now, &mtime) ||
3632 !SystemTimeToFileTime(&now, &atime)) {
3633 win32_error("utime", NULL);
3634 goto done;
Stefan Krah0e803b32010-11-26 16:16:47 +00003635 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003636 }
3637 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3638 PyErr_SetString(PyExc_TypeError,
3639 "utime() arg 2 must be a tuple (atime, mtime)");
3640 goto done;
3641 }
3642 else {
Victor Stinner5d272cc2012-03-13 13:35:55 +01003643 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 0),
3644 &atimesec, &ansec) == -1)
Victor Stinner8c62be82010-05-06 00:08:46 +00003645 goto done;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003646 time_t_to_FILE_TIME(atimesec, ansec, &atime);
Victor Stinner5d272cc2012-03-13 13:35:55 +01003647 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 1),
3648 &mtimesec, &mnsec) == -1)
Victor Stinner8c62be82010-05-06 00:08:46 +00003649 goto done;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003650 time_t_to_FILE_TIME(mtimesec, mnsec, &mtime);
Victor Stinner8c62be82010-05-06 00:08:46 +00003651 }
3652 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
3653 /* Avoid putting the file name into the error here,
3654 as that may confuse the user into believing that
3655 something is wrong with the file, when it also
3656 could be the time stamp that gives a problem. */
3657 win32_error("utime", NULL);
Antoine Pitroue85da7a2011-01-06 18:25:55 +00003658 goto done;
Victor Stinner8c62be82010-05-06 00:08:46 +00003659 }
3660 Py_INCREF(Py_None);
3661 result = Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003662done:
Victor Stinner8c62be82010-05-06 00:08:46 +00003663 CloseHandle(hFile);
3664 return result;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003665#else /* MS_WINDOWS */
Thomas Wouters477c8d52006-05-27 19:21:47 +00003666
Victor Stinner8c62be82010-05-06 00:08:46 +00003667 PyObject *opath;
3668 char *path;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003669 time_t atime, mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003670 long ansec, mnsec;
Victor Stinner8c62be82010-05-06 00:08:46 +00003671 int res;
Brian Curtinb0d5b5d2011-11-07 10:51:18 -06003672 PyObject* arg = Py_None;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003673
Brian Curtin52fbea12011-11-06 13:41:17 -06003674 if (!PyArg_ParseTuple(args, "O&|O:utime",
Victor Stinner8c62be82010-05-06 00:08:46 +00003675 PyUnicode_FSConverter, &opath, &arg))
3676 return NULL;
3677 path = PyBytes_AsString(opath);
Brian Curtinb0d5b5d2011-11-07 10:51:18 -06003678 if (arg == Py_None) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003679 /* optional time values not given */
3680 Py_BEGIN_ALLOW_THREADS
3681 res = utime(path, NULL);
3682 Py_END_ALLOW_THREADS
3683 }
3684 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3685 PyErr_SetString(PyExc_TypeError,
3686 "utime() arg 2 must be a tuple (atime, mtime)");
3687 Py_DECREF(opath);
3688 return NULL;
3689 }
3690 else {
Victor Stinner5d272cc2012-03-13 13:35:55 +01003691 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 0),
3692 &atime, &ansec) == -1) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003693 Py_DECREF(opath);
3694 return NULL;
3695 }
Victor Stinner5d272cc2012-03-13 13:35:55 +01003696 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 1),
3697 &mtime, &mnsec) == -1) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003698 Py_DECREF(opath);
3699 return NULL;
3700 }
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003701
3702 Py_BEGIN_ALLOW_THREADS
3703 {
3704#ifdef HAVE_UTIMENSAT
3705 struct timespec buf[2];
3706 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003707 buf[0].tv_nsec = ansec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003708 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003709 buf[1].tv_nsec = mnsec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003710 res = utimensat(AT_FDCWD, path, buf, 0);
3711#elif defined(HAVE_UTIMES)
3712 struct timeval buf[2];
3713 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003714 buf[0].tv_usec = ansec / 1000;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003715 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003716 buf[1].tv_usec = mnsec / 1000;
Victor Stinner8c62be82010-05-06 00:08:46 +00003717 res = utimes(path, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003718#elif defined(HAVE_UTIME_H)
3719 /* XXX should define struct utimbuf instead, above */
3720 struct utimbuf buf;
3721 buf.actime = atime;
3722 buf.modtime = mtime;
3723 res = utime(path, &buf);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003724#else
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003725 time_t buf[2];
3726 buf[0] = atime;
3727 buf[1] = mtime;
3728 res = utime(path, buf);
3729#endif
3730 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003731 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00003732 }
3733 if (res < 0) {
3734 return posix_error_with_allocated_filename(opath);
3735 }
3736 Py_DECREF(opath);
3737 Py_INCREF(Py_None);
3738 return Py_None;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003739#undef UTIME_EXTRACT
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003740#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003741}
3742
Ross Lagerwall7807c352011-03-17 20:20:30 +02003743#ifdef HAVE_FUTIMES
3744PyDoc_STRVAR(posix_futimes__doc__,
Brian Curtinc1b65d12011-11-07 14:18:54 -06003745"futimes(fd[, (atime, mtime)])\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003746Set the access and modified time of the file specified by the file\n\
Brian Curtinc1b65d12011-11-07 14:18:54 -06003747descriptor fd to the given values. If no second argument is used, set the\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003748access and modified times to the current time.");
3749
3750static PyObject *
3751posix_futimes(PyObject *self, PyObject *args)
3752{
3753 int res, fd;
Brian Curtinc1b65d12011-11-07 14:18:54 -06003754 PyObject* arg = Py_None;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003755 time_t atime, mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003756 long ansec, mnsec;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003757
Brian Curtinc1b65d12011-11-07 14:18:54 -06003758 if (!PyArg_ParseTuple(args, "i|O:futimes", &fd, &arg))
Ross Lagerwall7807c352011-03-17 20:20:30 +02003759 return NULL;
3760
3761 if (arg == Py_None) {
3762 /* optional time values not given */
3763 Py_BEGIN_ALLOW_THREADS
3764 res = futimes(fd, NULL);
3765 Py_END_ALLOW_THREADS
3766 }
3767 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3768 PyErr_SetString(PyExc_TypeError,
3769 "futimes() arg 2 must be a tuple (atime, mtime)");
3770 return NULL;
3771 }
3772 else {
Victor Stinner5d272cc2012-03-13 13:35:55 +01003773 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 0),
3774 &atime, &ansec) == -1) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003775 return NULL;
3776 }
Victor Stinner5d272cc2012-03-13 13:35:55 +01003777 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 1),
3778 &mtime, &mnsec) == -1) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003779 return NULL;
3780 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003781 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003782 {
3783#ifdef HAVE_FUTIMENS
3784 struct timespec buf[2];
3785 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003786 buf[0].tv_nsec = ansec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003787 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003788 buf[1].tv_nsec = mnsec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003789 res = futimens(fd, buf);
3790#else
3791 struct timeval buf[2];
3792 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003793 buf[0].tv_usec = ansec / 1000;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003794 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003795 buf[1].tv_usec = mnsec / 1000;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003796 res = futimes(fd, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003797#endif
3798 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003799 Py_END_ALLOW_THREADS
3800 }
3801 if (res < 0)
3802 return posix_error();
3803 Py_RETURN_NONE;
3804}
3805#endif
3806
3807#ifdef HAVE_LUTIMES
3808PyDoc_STRVAR(posix_lutimes__doc__,
Brian Curtinc1b65d12011-11-07 14:18:54 -06003809"lutimes(path[, (atime, mtime)])\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003810Like utime(), but if path is a symbolic link, it is not dereferenced.");
3811
3812static PyObject *
3813posix_lutimes(PyObject *self, PyObject *args)
3814{
Brian Curtinc1b65d12011-11-07 14:18:54 -06003815 PyObject *opath;
3816 PyObject *arg = Py_None;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003817 const char *path;
3818 int res;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003819 time_t atime, mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003820 long ansec, mnsec;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003821
Brian Curtinc1b65d12011-11-07 14:18:54 -06003822 if (!PyArg_ParseTuple(args, "O&|O:lutimes",
Ross Lagerwall7807c352011-03-17 20:20:30 +02003823 PyUnicode_FSConverter, &opath, &arg))
3824 return NULL;
3825 path = PyBytes_AsString(opath);
3826 if (arg == Py_None) {
3827 /* optional time values not given */
3828 Py_BEGIN_ALLOW_THREADS
3829 res = lutimes(path, NULL);
3830 Py_END_ALLOW_THREADS
3831 }
3832 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3833 PyErr_SetString(PyExc_TypeError,
3834 "lutimes() arg 2 must be a tuple (atime, mtime)");
3835 Py_DECREF(opath);
3836 return NULL;
3837 }
3838 else {
Victor Stinner5d272cc2012-03-13 13:35:55 +01003839 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 0),
3840 &atime, &ansec) == -1) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003841 Py_DECREF(opath);
3842 return NULL;
3843 }
Victor Stinner5d272cc2012-03-13 13:35:55 +01003844 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 1),
3845 &mtime, &mnsec) == -1) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003846 Py_DECREF(opath);
3847 return NULL;
3848 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003849 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003850 {
3851#ifdef HAVE_UTIMENSAT
3852 struct timespec buf[2];
3853 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003854 buf[0].tv_nsec = ansec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003855 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003856 buf[1].tv_nsec = mnsec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003857 res = utimensat(AT_FDCWD, path, buf, AT_SYMLINK_NOFOLLOW);
3858#else
3859 struct timeval buf[2];
3860 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003861 buf[0].tv_usec = ansec / 1000;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003862 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003863 buf[1].tv_usec = mnsec / 1000;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003864 res = lutimes(path, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003865#endif
3866 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003867 Py_END_ALLOW_THREADS
3868 }
3869 Py_DECREF(opath);
3870 if (res < 0)
3871 return posix_error();
3872 Py_RETURN_NONE;
3873}
3874#endif
3875
3876#ifdef HAVE_FUTIMENS
3877PyDoc_STRVAR(posix_futimens__doc__,
Brian Curtinc1b65d12011-11-07 14:18:54 -06003878"futimens(fd[, (atime_sec, atime_nsec), (mtime_sec, mtime_nsec)])\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003879Updates the timestamps of a file specified by the file descriptor fd, with\n\
3880nanosecond precision.\n\
Brian Curtinc1b65d12011-11-07 14:18:54 -06003881If no second argument is given, set atime and mtime to the current time.\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003882If *_nsec is specified as UTIME_NOW, the timestamp is updated to the\n\
3883current time.\n\
3884If *_nsec is specified as UTIME_OMIT, the timestamp is not updated.");
3885
3886static PyObject *
3887posix_futimens(PyObject *self, PyObject *args)
3888{
3889 int res, fd;
Brian Curtinc1b65d12011-11-07 14:18:54 -06003890 PyObject *atime = Py_None;
3891 PyObject *mtime = Py_None;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003892 struct timespec buf[2];
3893
Brian Curtinc1b65d12011-11-07 14:18:54 -06003894 if (!PyArg_ParseTuple(args, "i|OO:futimens",
Ross Lagerwall7807c352011-03-17 20:20:30 +02003895 &fd, &atime, &mtime))
3896 return NULL;
3897 if (atime == Py_None && mtime == Py_None) {
3898 /* optional time values not given */
3899 Py_BEGIN_ALLOW_THREADS
3900 res = futimens(fd, NULL);
3901 Py_END_ALLOW_THREADS
3902 }
3903 else if (!PyTuple_Check(atime) || PyTuple_Size(atime) != 2) {
3904 PyErr_SetString(PyExc_TypeError,
3905 "futimens() arg 2 must be a tuple (atime_sec, atime_nsec)");
3906 return NULL;
3907 }
3908 else if (!PyTuple_Check(mtime) || PyTuple_Size(mtime) != 2) {
3909 PyErr_SetString(PyExc_TypeError,
3910 "futimens() arg 3 must be a tuple (mtime_sec, mtime_nsec)");
3911 return NULL;
3912 }
3913 else {
3914 if (!PyArg_ParseTuple(atime, "ll:futimens",
3915 &(buf[0].tv_sec), &(buf[0].tv_nsec))) {
3916 return NULL;
3917 }
3918 if (!PyArg_ParseTuple(mtime, "ll:futimens",
3919 &(buf[1].tv_sec), &(buf[1].tv_nsec))) {
3920 return NULL;
3921 }
3922 Py_BEGIN_ALLOW_THREADS
3923 res = futimens(fd, buf);
3924 Py_END_ALLOW_THREADS
3925 }
3926 if (res < 0)
3927 return posix_error();
3928 Py_RETURN_NONE;
3929}
3930#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003931
Guido van Rossum3b066191991-06-04 19:40:25 +00003932/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003933
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003934PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003935"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003936Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003937
Barry Warsaw53699e91996-12-10 23:23:01 +00003938static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003939posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003940{
Victor Stinner8c62be82010-05-06 00:08:46 +00003941 int sts;
3942 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
3943 return NULL;
3944 _exit(sts);
3945 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003946}
3947
Martin v. Löwis114619e2002-10-07 06:44:21 +00003948#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
3949static void
Martin v. Löwis725507b2006-03-07 12:08:51 +00003950free_string_array(char **array, Py_ssize_t count)
Martin v. Löwis114619e2002-10-07 06:44:21 +00003951{
Victor Stinner8c62be82010-05-06 00:08:46 +00003952 Py_ssize_t i;
3953 for (i = 0; i < count; i++)
3954 PyMem_Free(array[i]);
3955 PyMem_DEL(array);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003956}
Martin v. Löwis011e8422009-05-05 04:43:17 +00003957
Antoine Pitrou69f71142009-05-24 21:25:49 +00003958static
Martin v. Löwis011e8422009-05-05 04:43:17 +00003959int fsconvert_strdup(PyObject *o, char**out)
3960{
Victor Stinner8c62be82010-05-06 00:08:46 +00003961 PyObject *bytes;
3962 Py_ssize_t size;
3963 if (!PyUnicode_FSConverter(o, &bytes))
3964 return 0;
3965 size = PyBytes_GET_SIZE(bytes);
3966 *out = PyMem_Malloc(size+1);
3967 if (!*out)
3968 return 0;
3969 memcpy(*out, PyBytes_AsString(bytes), size+1);
3970 Py_DECREF(bytes);
3971 return 1;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003972}
Martin v. Löwis114619e2002-10-07 06:44:21 +00003973#endif
3974
Ross Lagerwall7807c352011-03-17 20:20:30 +02003975#if defined(HAVE_EXECV) || defined (HAVE_FEXECVE)
Victor Stinner13bb71c2010-04-23 21:41:56 +00003976static char**
3977parse_envlist(PyObject* env, Py_ssize_t *envc_ptr)
3978{
Victor Stinner8c62be82010-05-06 00:08:46 +00003979 char **envlist;
3980 Py_ssize_t i, pos, envc;
3981 PyObject *keys=NULL, *vals=NULL;
3982 PyObject *key, *val, *key2, *val2;
3983 char *p, *k, *v;
3984 size_t len;
Victor Stinner13bb71c2010-04-23 21:41:56 +00003985
Victor Stinner8c62be82010-05-06 00:08:46 +00003986 i = PyMapping_Size(env);
3987 if (i < 0)
3988 return NULL;
3989 envlist = PyMem_NEW(char *, i + 1);
3990 if (envlist == NULL) {
3991 PyErr_NoMemory();
3992 return NULL;
3993 }
3994 envc = 0;
3995 keys = PyMapping_Keys(env);
3996 vals = PyMapping_Values(env);
3997 if (!keys || !vals)
3998 goto error;
3999 if (!PyList_Check(keys) || !PyList_Check(vals)) {
4000 PyErr_Format(PyExc_TypeError,
4001 "env.keys() or env.values() is not a list");
4002 goto error;
4003 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00004004
Victor Stinner8c62be82010-05-06 00:08:46 +00004005 for (pos = 0; pos < i; pos++) {
4006 key = PyList_GetItem(keys, pos);
4007 val = PyList_GetItem(vals, pos);
4008 if (!key || !val)
4009 goto error;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004010
Victor Stinner8c62be82010-05-06 00:08:46 +00004011 if (PyUnicode_FSConverter(key, &key2) == 0)
4012 goto error;
4013 if (PyUnicode_FSConverter(val, &val2) == 0) {
4014 Py_DECREF(key2);
4015 goto error;
4016 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00004017
4018#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +00004019 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
4020 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
Victor Stinner13bb71c2010-04-23 21:41:56 +00004021#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004022 k = PyBytes_AsString(key2);
4023 v = PyBytes_AsString(val2);
4024 len = PyBytes_GET_SIZE(key2) + PyBytes_GET_SIZE(val2) + 2;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004025
Victor Stinner8c62be82010-05-06 00:08:46 +00004026 p = PyMem_NEW(char, len);
4027 if (p == NULL) {
4028 PyErr_NoMemory();
4029 Py_DECREF(key2);
4030 Py_DECREF(val2);
4031 goto error;
4032 }
4033 PyOS_snprintf(p, len, "%s=%s", k, v);
4034 envlist[envc++] = p;
4035 Py_DECREF(key2);
4036 Py_DECREF(val2);
Victor Stinner13bb71c2010-04-23 21:41:56 +00004037#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +00004038 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00004039#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004040 }
4041 Py_DECREF(vals);
4042 Py_DECREF(keys);
Victor Stinner13bb71c2010-04-23 21:41:56 +00004043
Victor Stinner8c62be82010-05-06 00:08:46 +00004044 envlist[envc] = 0;
4045 *envc_ptr = envc;
4046 return envlist;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004047
4048error:
Victor Stinner8c62be82010-05-06 00:08:46 +00004049 Py_XDECREF(keys);
4050 Py_XDECREF(vals);
4051 while (--envc >= 0)
4052 PyMem_DEL(envlist[envc]);
4053 PyMem_DEL(envlist);
4054 return NULL;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004055}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004056
Ross Lagerwall7807c352011-03-17 20:20:30 +02004057static char**
4058parse_arglist(PyObject* argv, Py_ssize_t *argc)
4059{
4060 int i;
4061 char **argvlist = PyMem_NEW(char *, *argc+1);
4062 if (argvlist == NULL) {
4063 PyErr_NoMemory();
4064 return NULL;
4065 }
4066 for (i = 0; i < *argc; i++) {
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02004067 PyObject* item = PySequence_ITEM(argv, i);
4068 if (item == NULL)
4069 goto fail;
4070 if (!fsconvert_strdup(item, &argvlist[i])) {
4071 Py_DECREF(item);
Ross Lagerwall7807c352011-03-17 20:20:30 +02004072 goto fail;
4073 }
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02004074 Py_DECREF(item);
Ross Lagerwall7807c352011-03-17 20:20:30 +02004075 }
4076 argvlist[*argc] = NULL;
4077 return argvlist;
4078fail:
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02004079 *argc = i;
Ross Lagerwall7807c352011-03-17 20:20:30 +02004080 free_string_array(argvlist, *argc);
4081 return NULL;
4082}
4083#endif
4084
4085#ifdef HAVE_EXECV
4086PyDoc_STRVAR(posix_execv__doc__,
4087"execv(path, args)\n\n\
4088Execute an executable path with arguments, replacing current process.\n\
4089\n\
4090 path: path of executable file\n\
4091 args: tuple or list of strings");
4092
4093static PyObject *
4094posix_execv(PyObject *self, PyObject *args)
4095{
4096 PyObject *opath;
4097 char *path;
4098 PyObject *argv;
4099 char **argvlist;
4100 Py_ssize_t argc;
4101
4102 /* execv has two arguments: (path, argv), where
4103 argv is a list or tuple of strings. */
4104
4105 if (!PyArg_ParseTuple(args, "O&O:execv",
4106 PyUnicode_FSConverter,
4107 &opath, &argv))
4108 return NULL;
4109 path = PyBytes_AsString(opath);
4110 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
4111 PyErr_SetString(PyExc_TypeError,
4112 "execv() arg 2 must be a tuple or list");
4113 Py_DECREF(opath);
4114 return NULL;
4115 }
4116 argc = PySequence_Size(argv);
4117 if (argc < 1) {
4118 PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
4119 Py_DECREF(opath);
4120 return NULL;
4121 }
4122
4123 argvlist = parse_arglist(argv, &argc);
4124 if (argvlist == NULL) {
4125 Py_DECREF(opath);
4126 return NULL;
4127 }
4128
4129 execv(path, argvlist);
4130
4131 /* If we get here it's definitely an error */
4132
4133 free_string_array(argvlist, argc);
4134 Py_DECREF(opath);
4135 return posix_error();
4136}
4137
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004138PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004139"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004140Execute a path with arguments and environment, replacing current process.\n\
4141\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004142 path: path of executable file\n\
4143 args: tuple or list of arguments\n\
4144 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004145
Barry Warsaw53699e91996-12-10 23:23:01 +00004146static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004147posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004148{
Victor Stinner8c62be82010-05-06 00:08:46 +00004149 PyObject *opath;
4150 char *path;
4151 PyObject *argv, *env;
4152 char **argvlist;
4153 char **envlist;
Ross Lagerwall7807c352011-03-17 20:20:30 +02004154 Py_ssize_t argc, envc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004155
Victor Stinner8c62be82010-05-06 00:08:46 +00004156 /* execve has three arguments: (path, argv, env), where
4157 argv is a list or tuple of strings and env is a dictionary
4158 like posix.environ. */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004159
Victor Stinner8c62be82010-05-06 00:08:46 +00004160 if (!PyArg_ParseTuple(args, "O&OO:execve",
4161 PyUnicode_FSConverter,
4162 &opath, &argv, &env))
4163 return NULL;
4164 path = PyBytes_AsString(opath);
Ross Lagerwall7807c352011-03-17 20:20:30 +02004165 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00004166 PyErr_SetString(PyExc_TypeError,
4167 "execve() arg 2 must be a tuple or list");
4168 goto fail_0;
4169 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02004170 argc = PySequence_Size(argv);
Victor Stinner8c62be82010-05-06 00:08:46 +00004171 if (!PyMapping_Check(env)) {
4172 PyErr_SetString(PyExc_TypeError,
4173 "execve() arg 3 must be a mapping object");
4174 goto fail_0;
4175 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004176
Ross Lagerwall7807c352011-03-17 20:20:30 +02004177 argvlist = parse_arglist(argv, &argc);
Victor Stinner8c62be82010-05-06 00:08:46 +00004178 if (argvlist == NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00004179 goto fail_0;
4180 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004181
Victor Stinner8c62be82010-05-06 00:08:46 +00004182 envlist = parse_envlist(env, &envc);
4183 if (envlist == NULL)
4184 goto fail_1;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004185
Victor Stinner8c62be82010-05-06 00:08:46 +00004186 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00004187
Victor Stinner8c62be82010-05-06 00:08:46 +00004188 /* If we get here it's definitely an error */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004189
Victor Stinner8c62be82010-05-06 00:08:46 +00004190 (void) posix_error();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004191
Victor Stinner8c62be82010-05-06 00:08:46 +00004192 while (--envc >= 0)
4193 PyMem_DEL(envlist[envc]);
4194 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004195 fail_1:
Ross Lagerwall7807c352011-03-17 20:20:30 +02004196 free_string_array(argvlist, argc);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004197 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004198 Py_DECREF(opath);
4199 return NULL;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004200}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00004201#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004202
Ross Lagerwall7807c352011-03-17 20:20:30 +02004203#ifdef HAVE_FEXECVE
4204PyDoc_STRVAR(posix_fexecve__doc__,
4205"fexecve(fd, args, env)\n\n\
4206Execute the program specified by a file descriptor with arguments and\n\
4207environment, replacing the current process.\n\
4208\n\
4209 fd: file descriptor of executable\n\
4210 args: tuple or list of arguments\n\
4211 env: dictionary of strings mapping to strings");
4212
4213static PyObject *
4214posix_fexecve(PyObject *self, PyObject *args)
4215{
4216 int fd;
4217 PyObject *argv, *env;
4218 char **argvlist;
4219 char **envlist;
4220 Py_ssize_t argc, envc;
4221
4222 if (!PyArg_ParseTuple(args, "iOO:fexecve",
4223 &fd, &argv, &env))
4224 return NULL;
4225 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
4226 PyErr_SetString(PyExc_TypeError,
4227 "fexecve() arg 2 must be a tuple or list");
4228 return NULL;
4229 }
4230 argc = PySequence_Size(argv);
4231 if (!PyMapping_Check(env)) {
4232 PyErr_SetString(PyExc_TypeError,
4233 "fexecve() arg 3 must be a mapping object");
4234 return NULL;
4235 }
4236
4237 argvlist = parse_arglist(argv, &argc);
4238 if (argvlist == NULL)
4239 return NULL;
4240
4241 envlist = parse_envlist(env, &envc);
4242 if (envlist == NULL)
4243 goto fail;
4244
4245 fexecve(fd, argvlist, envlist);
4246
4247 /* If we get here it's definitely an error */
4248
4249 (void) posix_error();
4250
4251 while (--envc >= 0)
4252 PyMem_DEL(envlist[envc]);
4253 PyMem_DEL(envlist);
4254 fail:
4255 free_string_array(argvlist, argc);
4256 return NULL;
4257}
4258#endif /* HAVE_FEXECVE */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004259
Guido van Rossuma1065681999-01-25 23:20:23 +00004260#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004261PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004262"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00004263Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00004264\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004265 mode: mode of process creation\n\
4266 path: path of executable file\n\
4267 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00004268
4269static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004270posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00004271{
Victor Stinner8c62be82010-05-06 00:08:46 +00004272 PyObject *opath;
4273 char *path;
4274 PyObject *argv;
4275 char **argvlist;
4276 int mode, i;
4277 Py_ssize_t argc;
4278 Py_intptr_t spawnval;
4279 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossuma1065681999-01-25 23:20:23 +00004280
Victor Stinner8c62be82010-05-06 00:08:46 +00004281 /* spawnv has three arguments: (mode, path, argv), where
4282 argv is a list or tuple of strings. */
Guido van Rossuma1065681999-01-25 23:20:23 +00004283
Victor Stinner8c62be82010-05-06 00:08:46 +00004284 if (!PyArg_ParseTuple(args, "iO&O:spawnv", &mode,
4285 PyUnicode_FSConverter,
4286 &opath, &argv))
4287 return NULL;
4288 path = PyBytes_AsString(opath);
4289 if (PyList_Check(argv)) {
4290 argc = PyList_Size(argv);
4291 getitem = PyList_GetItem;
4292 }
4293 else if (PyTuple_Check(argv)) {
4294 argc = PyTuple_Size(argv);
4295 getitem = PyTuple_GetItem;
4296 }
4297 else {
4298 PyErr_SetString(PyExc_TypeError,
4299 "spawnv() arg 2 must be a tuple or list");
4300 Py_DECREF(opath);
4301 return NULL;
4302 }
Guido van Rossuma1065681999-01-25 23:20:23 +00004303
Victor Stinner8c62be82010-05-06 00:08:46 +00004304 argvlist = PyMem_NEW(char *, argc+1);
4305 if (argvlist == NULL) {
4306 Py_DECREF(opath);
4307 return PyErr_NoMemory();
4308 }
4309 for (i = 0; i < argc; i++) {
4310 if (!fsconvert_strdup((*getitem)(argv, i),
4311 &argvlist[i])) {
4312 free_string_array(argvlist, i);
4313 PyErr_SetString(
4314 PyExc_TypeError,
4315 "spawnv() arg 2 must contain only strings");
4316 Py_DECREF(opath);
4317 return NULL;
4318 }
4319 }
4320 argvlist[argc] = NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00004321
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004322#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004323 Py_BEGIN_ALLOW_THREADS
4324 spawnval = spawnv(mode, path, argvlist);
4325 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004326#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004327 if (mode == _OLD_P_OVERLAY)
4328 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00004329
Victor Stinner8c62be82010-05-06 00:08:46 +00004330 Py_BEGIN_ALLOW_THREADS
4331 spawnval = _spawnv(mode, path, argvlist);
4332 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004333#endif
Tim Peters5aa91602002-01-30 05:46:57 +00004334
Victor Stinner8c62be82010-05-06 00:08:46 +00004335 free_string_array(argvlist, argc);
4336 Py_DECREF(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00004337
Victor Stinner8c62be82010-05-06 00:08:46 +00004338 if (spawnval == -1)
4339 return posix_error();
4340 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00004341#if SIZEOF_LONG == SIZEOF_VOID_P
Victor Stinner8c62be82010-05-06 00:08:46 +00004342 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004343#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004344 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004345#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00004346}
4347
4348
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004349PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004350"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00004351Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00004352\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004353 mode: mode of process creation\n\
4354 path: path of executable file\n\
4355 args: tuple or list of arguments\n\
4356 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00004357
4358static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004359posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00004360{
Victor Stinner8c62be82010-05-06 00:08:46 +00004361 PyObject *opath;
4362 char *path;
4363 PyObject *argv, *env;
4364 char **argvlist;
4365 char **envlist;
4366 PyObject *res = NULL;
Antoine Pitrou22e41552010-08-15 18:07:50 +00004367 int mode;
4368 Py_ssize_t argc, i, envc;
Victor Stinner8c62be82010-05-06 00:08:46 +00004369 Py_intptr_t spawnval;
4370 PyObject *(*getitem)(PyObject *, Py_ssize_t);
4371 Py_ssize_t lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00004372
Victor Stinner8c62be82010-05-06 00:08:46 +00004373 /* spawnve has four arguments: (mode, path, argv, env), where
4374 argv is a list or tuple of strings and env is a dictionary
4375 like posix.environ. */
Guido van Rossuma1065681999-01-25 23:20:23 +00004376
Victor Stinner8c62be82010-05-06 00:08:46 +00004377 if (!PyArg_ParseTuple(args, "iO&OO:spawnve", &mode,
4378 PyUnicode_FSConverter,
4379 &opath, &argv, &env))
4380 return NULL;
4381 path = PyBytes_AsString(opath);
4382 if (PyList_Check(argv)) {
4383 argc = PyList_Size(argv);
4384 getitem = PyList_GetItem;
4385 }
4386 else if (PyTuple_Check(argv)) {
4387 argc = PyTuple_Size(argv);
4388 getitem = PyTuple_GetItem;
4389 }
4390 else {
4391 PyErr_SetString(PyExc_TypeError,
4392 "spawnve() arg 2 must be a tuple or list");
4393 goto fail_0;
4394 }
4395 if (!PyMapping_Check(env)) {
4396 PyErr_SetString(PyExc_TypeError,
4397 "spawnve() arg 3 must be a mapping object");
4398 goto fail_0;
4399 }
Guido van Rossuma1065681999-01-25 23:20:23 +00004400
Victor Stinner8c62be82010-05-06 00:08:46 +00004401 argvlist = PyMem_NEW(char *, argc+1);
4402 if (argvlist == NULL) {
4403 PyErr_NoMemory();
4404 goto fail_0;
4405 }
4406 for (i = 0; i < argc; i++) {
4407 if (!fsconvert_strdup((*getitem)(argv, i),
4408 &argvlist[i]))
4409 {
4410 lastarg = i;
4411 goto fail_1;
4412 }
4413 }
4414 lastarg = argc;
4415 argvlist[argc] = NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00004416
Victor Stinner8c62be82010-05-06 00:08:46 +00004417 envlist = parse_envlist(env, &envc);
4418 if (envlist == NULL)
4419 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00004420
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004421#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004422 Py_BEGIN_ALLOW_THREADS
4423 spawnval = spawnve(mode, path, argvlist, envlist);
4424 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004425#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004426 if (mode == _OLD_P_OVERLAY)
4427 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00004428
Victor Stinner8c62be82010-05-06 00:08:46 +00004429 Py_BEGIN_ALLOW_THREADS
4430 spawnval = _spawnve(mode, path, argvlist, envlist);
4431 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004432#endif
Tim Peters25059d32001-12-07 20:35:43 +00004433
Victor Stinner8c62be82010-05-06 00:08:46 +00004434 if (spawnval == -1)
4435 (void) posix_error();
4436 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00004437#if SIZEOF_LONG == SIZEOF_VOID_P
Victor Stinner8c62be82010-05-06 00:08:46 +00004438 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004439#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004440 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004441#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00004442
Victor Stinner8c62be82010-05-06 00:08:46 +00004443 while (--envc >= 0)
4444 PyMem_DEL(envlist[envc]);
4445 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004446 fail_1:
Victor Stinner8c62be82010-05-06 00:08:46 +00004447 free_string_array(argvlist, lastarg);
Martin v. Löwis114619e2002-10-07 06:44:21 +00004448 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004449 Py_DECREF(opath);
4450 return res;
Guido van Rossuma1065681999-01-25 23:20:23 +00004451}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004452
4453/* OS/2 supports spawnvp & spawnvpe natively */
4454#if defined(PYOS_OS2)
4455PyDoc_STRVAR(posix_spawnvp__doc__,
4456"spawnvp(mode, file, args)\n\n\
4457Execute the program 'file' in a new process, using the environment\n\
4458search path to find the file.\n\
4459\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004460 mode: mode of process creation\n\
4461 file: executable file name\n\
4462 args: tuple or list of strings");
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004463
4464static PyObject *
4465posix_spawnvp(PyObject *self, PyObject *args)
4466{
Victor Stinner8c62be82010-05-06 00:08:46 +00004467 PyObject *opath;
4468 char *path;
4469 PyObject *argv;
4470 char **argvlist;
4471 int mode, i, argc;
4472 Py_intptr_t spawnval;
4473 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004474
Victor Stinner8c62be82010-05-06 00:08:46 +00004475 /* spawnvp has three arguments: (mode, path, argv), where
4476 argv is a list or tuple of strings. */
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004477
Victor Stinner8c62be82010-05-06 00:08:46 +00004478 if (!PyArg_ParseTuple(args, "iO&O:spawnvp", &mode,
4479 PyUnicode_FSConverter,
4480 &opath, &argv))
4481 return NULL;
4482 path = PyBytes_AsString(opath);
4483 if (PyList_Check(argv)) {
4484 argc = PyList_Size(argv);
4485 getitem = PyList_GetItem;
4486 }
4487 else if (PyTuple_Check(argv)) {
4488 argc = PyTuple_Size(argv);
4489 getitem = PyTuple_GetItem;
4490 }
4491 else {
4492 PyErr_SetString(PyExc_TypeError,
4493 "spawnvp() arg 2 must be a tuple or list");
4494 Py_DECREF(opath);
4495 return NULL;
4496 }
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004497
Victor Stinner8c62be82010-05-06 00:08:46 +00004498 argvlist = PyMem_NEW(char *, argc+1);
4499 if (argvlist == NULL) {
4500 Py_DECREF(opath);
4501 return PyErr_NoMemory();
4502 }
4503 for (i = 0; i < argc; i++) {
4504 if (!fsconvert_strdup((*getitem)(argv, i),
4505 &argvlist[i])) {
4506 free_string_array(argvlist, i);
4507 PyErr_SetString(
4508 PyExc_TypeError,
4509 "spawnvp() arg 2 must contain only strings");
4510 Py_DECREF(opath);
4511 return NULL;
4512 }
4513 }
4514 argvlist[argc] = NULL;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004515
Victor Stinner8c62be82010-05-06 00:08:46 +00004516 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004517#if defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004518 spawnval = spawnvp(mode, path, argvlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004519#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004520 spawnval = _spawnvp(mode, path, argvlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004521#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004522 Py_END_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004523
Victor Stinner8c62be82010-05-06 00:08:46 +00004524 free_string_array(argvlist, argc);
4525 Py_DECREF(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004526
Victor Stinner8c62be82010-05-06 00:08:46 +00004527 if (spawnval == -1)
4528 return posix_error();
4529 else
4530 return Py_BuildValue("l", (long) spawnval);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004531}
4532
4533
4534PyDoc_STRVAR(posix_spawnvpe__doc__,
4535"spawnvpe(mode, file, args, env)\n\n\
4536Execute the program 'file' in a new process, using the environment\n\
4537search path to find the file.\n\
4538\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004539 mode: mode of process creation\n\
4540 file: executable file name\n\
4541 args: tuple or list of arguments\n\
4542 env: dictionary of strings mapping to strings");
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004543
4544static PyObject *
4545posix_spawnvpe(PyObject *self, PyObject *args)
4546{
Ross Lagerwalldcfde5a2011-11-04 07:09:14 +02004547 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00004548 char *path;
4549 PyObject *argv, *env;
4550 char **argvlist;
4551 char **envlist;
4552 PyObject *res=NULL;
Antoine Pitrou22e41552010-08-15 18:07:50 +00004553 int mode;
4554 Py_ssize_t argc, i, envc;
Victor Stinner8c62be82010-05-06 00:08:46 +00004555 Py_intptr_t spawnval;
4556 PyObject *(*getitem)(PyObject *, Py_ssize_t);
4557 int lastarg = 0;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004558
Victor Stinner8c62be82010-05-06 00:08:46 +00004559 /* spawnvpe has four arguments: (mode, path, argv, env), where
4560 argv is a list or tuple of strings and env is a dictionary
4561 like posix.environ. */
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004562
Victor Stinner8c62be82010-05-06 00:08:46 +00004563 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
4564 PyUnicode_FSConverter,
4565 &opath, &argv, &env))
4566 return NULL;
4567 path = PyBytes_AsString(opath);
4568 if (PyList_Check(argv)) {
4569 argc = PyList_Size(argv);
4570 getitem = PyList_GetItem;
4571 }
4572 else if (PyTuple_Check(argv)) {
4573 argc = PyTuple_Size(argv);
4574 getitem = PyTuple_GetItem;
4575 }
4576 else {
4577 PyErr_SetString(PyExc_TypeError,
4578 "spawnvpe() arg 2 must be a tuple or list");
4579 goto fail_0;
4580 }
4581 if (!PyMapping_Check(env)) {
4582 PyErr_SetString(PyExc_TypeError,
4583 "spawnvpe() arg 3 must be a mapping object");
4584 goto fail_0;
4585 }
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004586
Victor Stinner8c62be82010-05-06 00:08:46 +00004587 argvlist = PyMem_NEW(char *, argc+1);
4588 if (argvlist == NULL) {
4589 PyErr_NoMemory();
4590 goto fail_0;
4591 }
4592 for (i = 0; i < argc; i++) {
4593 if (!fsconvert_strdup((*getitem)(argv, i),
4594 &argvlist[i]))
4595 {
4596 lastarg = i;
4597 goto fail_1;
4598 }
4599 }
4600 lastarg = argc;
4601 argvlist[argc] = NULL;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004602
Victor Stinner8c62be82010-05-06 00:08:46 +00004603 envlist = parse_envlist(env, &envc);
4604 if (envlist == NULL)
4605 goto fail_1;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004606
Victor Stinner8c62be82010-05-06 00:08:46 +00004607 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004608#if defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004609 spawnval = spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004610#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004611 spawnval = _spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004612#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004613 Py_END_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004614
Victor Stinner8c62be82010-05-06 00:08:46 +00004615 if (spawnval == -1)
4616 (void) posix_error();
4617 else
4618 res = Py_BuildValue("l", (long) spawnval);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004619
Victor Stinner8c62be82010-05-06 00:08:46 +00004620 while (--envc >= 0)
4621 PyMem_DEL(envlist[envc]);
4622 PyMem_DEL(envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004623 fail_1:
Victor Stinner8c62be82010-05-06 00:08:46 +00004624 free_string_array(argvlist, lastarg);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004625 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004626 Py_DECREF(opath);
4627 return res;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004628}
4629#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00004630#endif /* HAVE_SPAWNV */
4631
4632
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004633#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004634PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004635"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004636Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
4637\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004638Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004639
4640static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004641posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004642{
Victor Stinner8c62be82010-05-06 00:08:46 +00004643 pid_t pid;
4644 int result = 0;
4645 _PyImport_AcquireLock();
4646 pid = fork1();
4647 if (pid == 0) {
4648 /* child: this clobbers and resets the import lock. */
4649 PyOS_AfterFork();
4650 } else {
4651 /* parent: release the import lock. */
4652 result = _PyImport_ReleaseLock();
4653 }
4654 if (pid == -1)
4655 return posix_error();
4656 if (result < 0) {
4657 /* Don't clobber the OSError if the fork failed. */
4658 PyErr_SetString(PyExc_RuntimeError,
4659 "not holding the import lock");
4660 return NULL;
4661 }
4662 return PyLong_FromPid(pid);
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004663}
4664#endif
4665
4666
Guido van Rossumad0ee831995-03-01 10:34:45 +00004667#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004668PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004669"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004670Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004671Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004672
Barry Warsaw53699e91996-12-10 23:23:01 +00004673static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004674posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004675{
Victor Stinner8c62be82010-05-06 00:08:46 +00004676 pid_t pid;
4677 int result = 0;
4678 _PyImport_AcquireLock();
4679 pid = fork();
4680 if (pid == 0) {
4681 /* child: this clobbers and resets the import lock. */
4682 PyOS_AfterFork();
4683 } else {
4684 /* parent: release the import lock. */
4685 result = _PyImport_ReleaseLock();
4686 }
4687 if (pid == -1)
4688 return posix_error();
4689 if (result < 0) {
4690 /* Don't clobber the OSError if the fork failed. */
4691 PyErr_SetString(PyExc_RuntimeError,
4692 "not holding the import lock");
4693 return NULL;
4694 }
4695 return PyLong_FromPid(pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00004696}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004697#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004698
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004699#ifdef HAVE_SCHED_H
4700
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +02004701#ifdef HAVE_SCHED_GET_PRIORITY_MAX
4702
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004703PyDoc_STRVAR(posix_sched_get_priority_max__doc__,
4704"sched_get_priority_max(policy)\n\n\
4705Get the maximum scheduling priority for *policy*.");
4706
4707static PyObject *
4708posix_sched_get_priority_max(PyObject *self, PyObject *args)
4709{
4710 int policy, max;
4711
4712 if (!PyArg_ParseTuple(args, "i:sched_get_priority_max", &policy))
4713 return NULL;
4714 max = sched_get_priority_max(policy);
4715 if (max < 0)
4716 return posix_error();
4717 return PyLong_FromLong(max);
4718}
4719
4720PyDoc_STRVAR(posix_sched_get_priority_min__doc__,
4721"sched_get_priority_min(policy)\n\n\
4722Get the minimum scheduling priority for *policy*.");
4723
4724static PyObject *
4725posix_sched_get_priority_min(PyObject *self, PyObject *args)
4726{
4727 int policy, min;
4728
4729 if (!PyArg_ParseTuple(args, "i:sched_get_priority_min", &policy))
4730 return NULL;
4731 min = sched_get_priority_min(policy);
4732 if (min < 0)
4733 return posix_error();
4734 return PyLong_FromLong(min);
4735}
4736
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +02004737#endif /* HAVE_SCHED_GET_PRIORITY_MAX */
4738
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004739#ifdef HAVE_SCHED_SETSCHEDULER
4740
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004741PyDoc_STRVAR(posix_sched_getscheduler__doc__,
4742"sched_getscheduler(pid)\n\n\
4743Get the scheduling policy for the process with a PID of *pid*.\n\
4744Passing a PID of 0 returns the scheduling policy for the calling process.");
4745
4746static PyObject *
4747posix_sched_getscheduler(PyObject *self, PyObject *args)
4748{
4749 pid_t pid;
4750 int policy;
4751
4752 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":sched_getscheduler", &pid))
4753 return NULL;
4754 policy = sched_getscheduler(pid);
4755 if (policy < 0)
4756 return posix_error();
4757 return PyLong_FromLong(policy);
4758}
4759
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004760#endif
4761
4762#if defined(HAVE_SCHED_SETSCHEDULER) || defined(HAVE_SCHED_SETPARAM)
4763
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004764static PyObject *
4765sched_param_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
4766{
4767 PyObject *res, *priority;
Benjamin Peterson4e36d5a2011-08-02 19:56:11 -05004768 static char *kwlist[] = {"sched_priority", NULL};
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004769
4770 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O:sched_param", kwlist, &priority))
4771 return NULL;
4772 res = PyStructSequence_New(type);
4773 if (!res)
4774 return NULL;
4775 Py_INCREF(priority);
4776 PyStructSequence_SET_ITEM(res, 0, priority);
4777 return res;
4778}
4779
4780PyDoc_STRVAR(sched_param__doc__,
4781"sched_param(sched_priority): A scheduling parameter.\n\n\
4782Current has only one field: sched_priority");
4783
4784static PyStructSequence_Field sched_param_fields[] = {
4785 {"sched_priority", "the scheduling priority"},
4786 {0}
4787};
4788
4789static PyStructSequence_Desc sched_param_desc = {
4790 "sched_param", /* name */
4791 sched_param__doc__, /* doc */
4792 sched_param_fields,
4793 1
4794};
4795
4796static int
4797convert_sched_param(PyObject *param, struct sched_param *res)
4798{
4799 long priority;
4800
4801 if (Py_TYPE(param) != &SchedParamType) {
4802 PyErr_SetString(PyExc_TypeError, "must have a sched_param object");
4803 return 0;
4804 }
4805 priority = PyLong_AsLong(PyStructSequence_GET_ITEM(param, 0));
4806 if (priority == -1 && PyErr_Occurred())
4807 return 0;
4808 if (priority > INT_MAX || priority < INT_MIN) {
4809 PyErr_SetString(PyExc_OverflowError, "sched_priority out of range");
4810 return 0;
4811 }
4812 res->sched_priority = Py_SAFE_DOWNCAST(priority, long, int);
4813 return 1;
4814}
4815
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004816#endif
4817
4818#ifdef HAVE_SCHED_SETSCHEDULER
4819
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004820PyDoc_STRVAR(posix_sched_setscheduler__doc__,
4821"sched_setscheduler(pid, policy, param)\n\n\
4822Set the scheduling policy, *policy*, for *pid*.\n\
4823If *pid* is 0, the calling process is changed.\n\
4824*param* is an instance of sched_param.");
4825
4826static PyObject *
4827posix_sched_setscheduler(PyObject *self, PyObject *args)
4828{
4829 pid_t pid;
4830 int policy;
4831 struct sched_param param;
4832
4833 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "iO&:sched_setscheduler",
4834 &pid, &policy, &convert_sched_param, &param))
4835 return NULL;
Jesus Cea9c822272011-09-10 01:40:52 +02004836
4837 /*
Jesus Cea54b01492011-09-10 01:53:19 +02004838 ** sched_setscheduler() returns 0 in Linux, but the previous
4839 ** scheduling policy under Solaris/Illumos, and others.
4840 ** On error, -1 is returned in all Operating Systems.
Jesus Cea9c822272011-09-10 01:40:52 +02004841 */
4842 if (sched_setscheduler(pid, policy, &param) == -1)
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004843 return posix_error();
4844 Py_RETURN_NONE;
4845}
4846
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004847#endif
4848
4849#ifdef HAVE_SCHED_SETPARAM
4850
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004851PyDoc_STRVAR(posix_sched_getparam__doc__,
4852"sched_getparam(pid) -> sched_param\n\n\
4853Returns scheduling parameters for the process with *pid* as an instance of the\n\
4854sched_param class. A PID of 0 means the calling process.");
4855
4856static PyObject *
4857posix_sched_getparam(PyObject *self, PyObject *args)
4858{
4859 pid_t pid;
4860 struct sched_param param;
4861 PyObject *res, *priority;
4862
4863 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":sched_getparam", &pid))
4864 return NULL;
4865 if (sched_getparam(pid, &param))
4866 return posix_error();
4867 res = PyStructSequence_New(&SchedParamType);
4868 if (!res)
4869 return NULL;
4870 priority = PyLong_FromLong(param.sched_priority);
4871 if (!priority) {
4872 Py_DECREF(res);
4873 return NULL;
4874 }
4875 PyStructSequence_SET_ITEM(res, 0, priority);
4876 return res;
4877}
4878
4879PyDoc_STRVAR(posix_sched_setparam__doc__,
4880"sched_setparam(pid, param)\n\n\
4881Set scheduling parameters for a process with PID *pid*.\n\
4882A PID of 0 means the calling process.");
4883
4884static PyObject *
4885posix_sched_setparam(PyObject *self, PyObject *args)
4886{
4887 pid_t pid;
4888 struct sched_param param;
4889
4890 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "O&:sched_setparam",
4891 &pid, &convert_sched_param, &param))
4892 return NULL;
4893 if (sched_setparam(pid, &param))
4894 return posix_error();
4895 Py_RETURN_NONE;
4896}
4897
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004898#endif
4899
4900#ifdef HAVE_SCHED_RR_GET_INTERVAL
4901
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004902PyDoc_STRVAR(posix_sched_rr_get_interval__doc__,
4903"sched_rr_get_interval(pid) -> float\n\n\
4904Return the round-robin quantum for the process with PID *pid* in seconds.");
4905
4906static PyObject *
4907posix_sched_rr_get_interval(PyObject *self, PyObject *args)
4908{
4909 pid_t pid;
4910 struct timespec interval;
4911
4912 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":sched_rr_get_interval", &pid))
4913 return NULL;
4914 if (sched_rr_get_interval(pid, &interval))
4915 return posix_error();
4916 return PyFloat_FromDouble((double)interval.tv_sec + 1e-9*interval.tv_nsec);
4917}
4918
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004919#endif
4920
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004921PyDoc_STRVAR(posix_sched_yield__doc__,
4922"sched_yield()\n\n\
4923Voluntarily relinquish the CPU.");
4924
4925static PyObject *
4926posix_sched_yield(PyObject *self, PyObject *noargs)
4927{
4928 if (sched_yield())
4929 return posix_error();
4930 Py_RETURN_NONE;
4931}
4932
Benjamin Peterson2740af82011-08-02 17:41:34 -05004933#ifdef HAVE_SCHED_SETAFFINITY
4934
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004935typedef struct {
4936 PyObject_HEAD;
4937 Py_ssize_t size;
4938 int ncpus;
4939 cpu_set_t *set;
4940} Py_cpu_set;
4941
4942static PyTypeObject cpu_set_type;
4943
4944static void
4945cpu_set_dealloc(Py_cpu_set *set)
4946{
4947 assert(set->set);
4948 CPU_FREE(set->set);
4949 Py_TYPE(set)->tp_free(set);
4950}
4951
4952static Py_cpu_set *
4953make_new_cpu_set(PyTypeObject *type, Py_ssize_t size)
4954{
4955 Py_cpu_set *set;
4956
4957 if (size < 0) {
4958 PyErr_SetString(PyExc_ValueError, "negative size");
4959 return NULL;
4960 }
4961 set = (Py_cpu_set *)type->tp_alloc(type, 0);
4962 if (!set)
4963 return NULL;
4964 set->ncpus = size;
4965 set->size = CPU_ALLOC_SIZE(size);
4966 set->set = CPU_ALLOC(size);
4967 if (!set->set) {
4968 type->tp_free(set);
4969 PyErr_NoMemory();
4970 return NULL;
4971 }
4972 CPU_ZERO_S(set->size, set->set);
4973 return set;
4974}
4975
4976static PyObject *
4977cpu_set_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
4978{
4979 int size;
4980
4981 if (!_PyArg_NoKeywords("cpu_set()", kwargs) ||
4982 !PyArg_ParseTuple(args, "i:cpu_set", &size))
4983 return NULL;
4984 return (PyObject *)make_new_cpu_set(type, size);
4985}
4986
4987static PyObject *
4988cpu_set_repr(Py_cpu_set *set)
4989{
4990 return PyUnicode_FromFormat("<cpu_set with %li entries>", set->ncpus);
Victor Stinner63941882011-09-29 00:42:28 +02004991}
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004992
4993static Py_ssize_t
4994cpu_set_len(Py_cpu_set *set)
4995{
4996 return set->ncpus;
4997}
4998
4999static int
5000_get_cpu(Py_cpu_set *set, const char *requester, PyObject *args)
5001{
5002 int cpu;
5003 if (!PyArg_ParseTuple(args, requester, &cpu))
5004 return -1;
5005 if (cpu < 0) {
5006 PyErr_SetString(PyExc_ValueError, "cpu < 0 not valid");
5007 return -1;
5008 }
5009 if (cpu >= set->ncpus) {
5010 PyErr_SetString(PyExc_ValueError, "cpu too large for set");
5011 return -1;
5012 }
5013 return cpu;
5014}
5015
5016PyDoc_STRVAR(cpu_set_set_doc,
5017"cpu_set.set(i)\n\n\
5018Add CPU *i* to the set.");
5019
5020static PyObject *
5021cpu_set_set(Py_cpu_set *set, PyObject *args)
5022{
5023 int cpu = _get_cpu(set, "i|set", args);
5024 if (cpu == -1)
5025 return NULL;
5026 CPU_SET_S(cpu, set->size, set->set);
5027 Py_RETURN_NONE;
5028}
5029
5030PyDoc_STRVAR(cpu_set_count_doc,
5031"cpu_set.count() -> int\n\n\
5032Return the number of CPUs active in the set.");
5033
5034static PyObject *
5035cpu_set_count(Py_cpu_set *set, PyObject *noargs)
5036{
5037 return PyLong_FromLong(CPU_COUNT_S(set->size, set->set));
5038}
5039
5040PyDoc_STRVAR(cpu_set_clear_doc,
5041"cpu_set.clear(i)\n\n\
5042Remove CPU *i* from the set.");
5043
5044static PyObject *
5045cpu_set_clear(Py_cpu_set *set, PyObject *args)
5046{
5047 int cpu = _get_cpu(set, "i|clear", args);
5048 if (cpu == -1)
5049 return NULL;
5050 CPU_CLR_S(cpu, set->size, set->set);
5051 Py_RETURN_NONE;
5052}
5053
5054PyDoc_STRVAR(cpu_set_isset_doc,
5055"cpu_set.isset(i) -> bool\n\n\
5056Test if CPU *i* is in the set.");
5057
5058static PyObject *
5059cpu_set_isset(Py_cpu_set *set, PyObject *args)
5060{
5061 int cpu = _get_cpu(set, "i|isset", args);
5062 if (cpu == -1)
5063 return NULL;
5064 if (CPU_ISSET_S(cpu, set->size, set->set))
5065 Py_RETURN_TRUE;
5066 Py_RETURN_FALSE;
5067}
5068
5069PyDoc_STRVAR(cpu_set_zero_doc,
5070"cpu_set.zero()\n\n\
5071Clear the cpu_set.");
5072
5073static PyObject *
5074cpu_set_zero(Py_cpu_set *set, PyObject *noargs)
5075{
5076 CPU_ZERO_S(set->size, set->set);
5077 Py_RETURN_NONE;
5078}
5079
5080static PyObject *
5081cpu_set_richcompare(Py_cpu_set *set, Py_cpu_set *other, int op)
5082{
5083 int eq;
5084
Brian Curtindfc80e32011-08-10 20:28:54 -05005085 if ((op != Py_EQ && op != Py_NE) || Py_TYPE(other) != &cpu_set_type)
5086 Py_RETURN_NOTIMPLEMENTED;
5087
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005088 eq = set->ncpus == other->ncpus && CPU_EQUAL_S(set->size, set->set, other->set);
5089 if ((op == Py_EQ) ? eq : !eq)
5090 Py_RETURN_TRUE;
5091 else
5092 Py_RETURN_FALSE;
5093}
5094
5095#define CPU_SET_BINOP(name, op) \
5096 static PyObject * \
5097 do_cpu_set_##name(Py_cpu_set *left, Py_cpu_set *right, Py_cpu_set *res) { \
5098 if (res) { \
5099 Py_INCREF(res); \
5100 } \
5101 else { \
Benjamin Petersone870fe62011-08-02 17:44:26 -05005102 res = make_new_cpu_set(&cpu_set_type, left->ncpus); \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005103 if (!res) \
5104 return NULL; \
5105 } \
Benjamin Peterson9b374bf2011-08-02 18:22:30 -05005106 if (Py_TYPE(right) != &cpu_set_type || left->ncpus != right->ncpus) { \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005107 Py_DECREF(res); \
Brian Curtindfc80e32011-08-10 20:28:54 -05005108 Py_RETURN_NOTIMPLEMENTED; \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005109 } \
Benjamin Peterson8f7bdd32011-08-02 18:34:30 -05005110 assert(left->size == right->size && right->size == res->size); \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005111 op(res->size, res->set, left->set, right->set); \
5112 return (PyObject *)res; \
5113 } \
5114 static PyObject * \
5115 cpu_set_##name(Py_cpu_set *left, Py_cpu_set *right) { \
5116 return do_cpu_set_##name(left, right, NULL); \
5117 } \
5118 static PyObject * \
5119 cpu_set_i##name(Py_cpu_set *left, Py_cpu_set *right) { \
5120 return do_cpu_set_##name(left, right, left); \
5121 } \
5122
5123CPU_SET_BINOP(and, CPU_AND_S)
5124CPU_SET_BINOP(or, CPU_OR_S)
5125CPU_SET_BINOP(xor, CPU_XOR_S)
5126#undef CPU_SET_BINOP
5127
5128PyDoc_STRVAR(cpu_set_doc,
5129"cpu_set(size)\n\n\
5130Create an empty mask of CPUs.");
5131
5132static PyNumberMethods cpu_set_as_number = {
5133 0, /*nb_add*/
5134 0, /*nb_subtract*/
5135 0, /*nb_multiply*/
5136 0, /*nb_remainder*/
5137 0, /*nb_divmod*/
5138 0, /*nb_power*/
5139 0, /*nb_negative*/
5140 0, /*nb_positive*/
5141 0, /*nb_absolute*/
5142 0, /*nb_bool*/
5143 0, /*nb_invert*/
5144 0, /*nb_lshift*/
5145 0, /*nb_rshift*/
5146 (binaryfunc)cpu_set_and, /*nb_and*/
5147 (binaryfunc)cpu_set_xor, /*nb_xor*/
5148 (binaryfunc)cpu_set_or, /*nb_or*/
5149 0, /*nb_int*/
5150 0, /*nb_reserved*/
5151 0, /*nb_float*/
5152 0, /*nb_inplace_add*/
5153 0, /*nb_inplace_subtract*/
5154 0, /*nb_inplace_multiply*/
5155 0, /*nb_inplace_remainder*/
5156 0, /*nb_inplace_power*/
5157 0, /*nb_inplace_lshift*/
5158 0, /*nb_inplace_rshift*/
5159 (binaryfunc)cpu_set_iand, /*nb_inplace_and*/
5160 (binaryfunc)cpu_set_ixor, /*nb_inplace_xor*/
5161 (binaryfunc)cpu_set_ior, /*nb_inplace_or*/
5162};
5163
5164static PySequenceMethods cpu_set_as_sequence = {
5165 (lenfunc)cpu_set_len, /* sq_length */
5166};
5167
5168static PyMethodDef cpu_set_methods[] = {
5169 {"clear", (PyCFunction)cpu_set_clear, METH_VARARGS, cpu_set_clear_doc},
5170 {"count", (PyCFunction)cpu_set_count, METH_NOARGS, cpu_set_count_doc},
5171 {"isset", (PyCFunction)cpu_set_isset, METH_VARARGS, cpu_set_isset_doc},
5172 {"set", (PyCFunction)cpu_set_set, METH_VARARGS, cpu_set_set_doc},
5173 {"zero", (PyCFunction)cpu_set_zero, METH_NOARGS, cpu_set_zero_doc},
5174 {NULL, NULL} /* sentinel */
5175};
5176
5177static PyTypeObject cpu_set_type = {
5178 PyVarObject_HEAD_INIT(&PyType_Type, 0)
5179 "posix.cpu_set", /* tp_name */
5180 sizeof(Py_cpu_set), /* tp_basicsize */
5181 0, /* tp_itemsize */
5182 /* methods */
5183 (destructor)cpu_set_dealloc, /* tp_dealloc */
5184 0, /* tp_print */
5185 0, /* tp_getattr */
5186 0, /* tp_setattr */
5187 0, /* tp_reserved */
5188 (reprfunc)cpu_set_repr, /* tp_repr */
5189 &cpu_set_as_number, /* tp_as_number */
5190 &cpu_set_as_sequence, /* tp_as_sequence */
5191 0, /* tp_as_mapping */
5192 PyObject_HashNotImplemented, /* tp_hash */
5193 0, /* tp_call */
5194 0, /* tp_str */
5195 PyObject_GenericGetAttr, /* tp_getattro */
5196 0, /* tp_setattro */
5197 0, /* tp_as_buffer */
5198 Py_TPFLAGS_DEFAULT, /* tp_flags */
5199 cpu_set_doc, /* tp_doc */
5200 0, /* tp_traverse */
5201 0, /* tp_clear */
5202 (richcmpfunc)cpu_set_richcompare, /* tp_richcompare */
5203 0, /* tp_weaklistoffset */
5204 0, /* tp_iter */
5205 0, /* tp_iternext */
5206 cpu_set_methods, /* tp_methods */
5207 0, /* tp_members */
5208 0, /* tp_getset */
5209 0, /* tp_base */
5210 0, /* tp_dict */
5211 0, /* tp_descr_get */
5212 0, /* tp_descr_set */
5213 0, /* tp_dictoffset */
5214 0, /* tp_init */
5215 PyType_GenericAlloc, /* tp_alloc */
5216 cpu_set_new, /* tp_new */
5217 PyObject_Del, /* tp_free */
5218};
5219
5220PyDoc_STRVAR(posix_sched_setaffinity__doc__,
5221"sched_setaffinity(pid, cpu_set)\n\n\
5222Set the affinity of the process with PID *pid* to *cpu_set*.");
5223
5224static PyObject *
5225posix_sched_setaffinity(PyObject *self, PyObject *args)
5226{
5227 pid_t pid;
5228 Py_cpu_set *cpu_set;
5229
Benjamin Petersona17a5d62011-08-09 16:49:13 -05005230 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "O!:sched_setaffinity",
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005231 &pid, &cpu_set_type, &cpu_set))
5232 return NULL;
5233 if (sched_setaffinity(pid, cpu_set->size, cpu_set->set))
5234 return posix_error();
5235 Py_RETURN_NONE;
5236}
5237
5238PyDoc_STRVAR(posix_sched_getaffinity__doc__,
5239"sched_getaffinity(pid, ncpus) -> cpu_set\n\n\
5240Return the affinity of the process with PID *pid*.\n\
5241The returned cpu_set will be of size *ncpus*.");
5242
5243static PyObject *
5244posix_sched_getaffinity(PyObject *self, PyObject *args)
5245{
5246 pid_t pid;
5247 int ncpus;
5248 Py_cpu_set *res;
5249
Benjamin Peterson7ac92142011-08-03 08:54:26 -05005250 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:sched_getaffinity",
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005251 &pid, &ncpus))
5252 return NULL;
5253 res = make_new_cpu_set(&cpu_set_type, ncpus);
5254 if (!res)
5255 return NULL;
5256 if (sched_getaffinity(pid, res->size, res->set)) {
5257 Py_DECREF(res);
5258 return posix_error();
5259 }
5260 return (PyObject *)res;
5261}
5262
Benjamin Peterson2740af82011-08-02 17:41:34 -05005263#endif /* HAVE_SCHED_SETAFFINITY */
5264
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005265#endif /* HAVE_SCHED_H */
5266
Neal Norwitzb59798b2003-03-21 01:43:31 +00005267/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00005268/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
5269#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00005270#define DEV_PTY_FILE "/dev/ptc"
5271#define HAVE_DEV_PTMX
5272#else
5273#define DEV_PTY_FILE "/dev/ptmx"
5274#endif
5275
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005276#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00005277#ifdef HAVE_PTY_H
5278#include <pty.h>
5279#else
5280#ifdef HAVE_LIBUTIL_H
5281#include <libutil.h>
Ronald Oussoren755740f2010-02-07 19:56:39 +00005282#else
5283#ifdef HAVE_UTIL_H
5284#include <util.h>
5285#endif /* HAVE_UTIL_H */
Fred Drake8cef4cf2000-06-28 16:40:38 +00005286#endif /* HAVE_LIBUTIL_H */
5287#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00005288#ifdef HAVE_STROPTS_H
5289#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005290#endif
5291#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00005292
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005293#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005294PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005295"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005296Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00005297
5298static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005299posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00005300{
Victor Stinner8c62be82010-05-06 00:08:46 +00005301 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00005302#ifndef HAVE_OPENPTY
Victor Stinner8c62be82010-05-06 00:08:46 +00005303 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00005304#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005305#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Victor Stinner8c62be82010-05-06 00:08:46 +00005306 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005307#ifdef sun
Victor Stinner8c62be82010-05-06 00:08:46 +00005308 extern char *ptsname(int fildes);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005309#endif
5310#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00005311
Thomas Wouters70c21a12000-07-14 14:28:33 +00005312#ifdef HAVE_OPENPTY
Victor Stinner8c62be82010-05-06 00:08:46 +00005313 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
5314 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00005315#elif defined(HAVE__GETPTY)
Victor Stinner8c62be82010-05-06 00:08:46 +00005316 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
5317 if (slave_name == NULL)
5318 return posix_error();
Thomas Wouters70c21a12000-07-14 14:28:33 +00005319
Victor Stinner8c62be82010-05-06 00:08:46 +00005320 slave_fd = open(slave_name, O_RDWR);
5321 if (slave_fd < 0)
5322 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005323#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005324 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
5325 if (master_fd < 0)
5326 return posix_error();
5327 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
5328 /* change permission of slave */
5329 if (grantpt(master_fd) < 0) {
5330 PyOS_setsig(SIGCHLD, sig_saved);
5331 return posix_error();
5332 }
5333 /* unlock slave */
5334 if (unlockpt(master_fd) < 0) {
5335 PyOS_setsig(SIGCHLD, sig_saved);
5336 return posix_error();
5337 }
5338 PyOS_setsig(SIGCHLD, sig_saved);
5339 slave_name = ptsname(master_fd); /* get name of slave */
5340 if (slave_name == NULL)
5341 return posix_error();
5342 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
5343 if (slave_fd < 0)
5344 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00005345#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Victor Stinner8c62be82010-05-06 00:08:46 +00005346 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
5347 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00005348#ifndef __hpux
Victor Stinner8c62be82010-05-06 00:08:46 +00005349 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00005350#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005351#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00005352#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00005353
Victor Stinner8c62be82010-05-06 00:08:46 +00005354 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00005355
Fred Drake8cef4cf2000-06-28 16:40:38 +00005356}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005357#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00005358
5359#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005360PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005361"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00005362Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
5363Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005364To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00005365
5366static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005367posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00005368{
Victor Stinner8c62be82010-05-06 00:08:46 +00005369 int master_fd = -1, result = 0;
5370 pid_t pid;
Tim Peters5aa91602002-01-30 05:46:57 +00005371
Victor Stinner8c62be82010-05-06 00:08:46 +00005372 _PyImport_AcquireLock();
5373 pid = forkpty(&master_fd, NULL, NULL, NULL);
5374 if (pid == 0) {
5375 /* child: this clobbers and resets the import lock. */
5376 PyOS_AfterFork();
5377 } else {
5378 /* parent: release the import lock. */
5379 result = _PyImport_ReleaseLock();
5380 }
5381 if (pid == -1)
5382 return posix_error();
5383 if (result < 0) {
5384 /* Don't clobber the OSError if the fork failed. */
5385 PyErr_SetString(PyExc_RuntimeError,
5386 "not holding the import lock");
5387 return NULL;
5388 }
5389 return Py_BuildValue("(Ni)", PyLong_FromPid(pid), master_fd);
Fred Drake8cef4cf2000-06-28 16:40:38 +00005390}
5391#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005392
Ross Lagerwall7807c352011-03-17 20:20:30 +02005393
Guido van Rossumad0ee831995-03-01 10:34:45 +00005394#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005395PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005396"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005397Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005398
Barry Warsaw53699e91996-12-10 23:23:01 +00005399static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005400posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005401{
Victor Stinner8c62be82010-05-06 00:08:46 +00005402 return PyLong_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005403}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005404#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005405
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005406
Guido van Rossumad0ee831995-03-01 10:34:45 +00005407#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005408PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005409"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005410Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005411
Barry Warsaw53699e91996-12-10 23:23:01 +00005412static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005413posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005414{
Victor Stinner8c62be82010-05-06 00:08:46 +00005415 return PyLong_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005416}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005417#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005418
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005419
Guido van Rossumad0ee831995-03-01 10:34:45 +00005420#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005421PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005422"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005423Return the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005424
Barry Warsaw53699e91996-12-10 23:23:01 +00005425static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005426posix_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005427{
Victor Stinner8c62be82010-05-06 00:08:46 +00005428 return PyLong_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005429}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005430#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005431
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005432
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005433PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005434"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005435Return the current process 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_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005439{
Victor Stinner8c62be82010-05-06 00:08:46 +00005440 return PyLong_FromPid(getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00005441}
5442
Ross Lagerwallb0ae53d2011-06-10 07:30:30 +02005443#ifdef HAVE_GETGROUPLIST
5444PyDoc_STRVAR(posix_getgrouplist__doc__,
5445"getgrouplist(user, group) -> list of groups to which a user belongs\n\n\
5446Returns a list of groups to which a user belongs.\n\n\
5447 user: username to lookup\n\
5448 group: base group id of the user");
5449
5450static PyObject *
5451posix_getgrouplist(PyObject *self, PyObject *args)
5452{
5453#ifdef NGROUPS_MAX
5454#define MAX_GROUPS NGROUPS_MAX
5455#else
5456 /* defined to be 16 on Solaris7, so this should be a small number */
5457#define MAX_GROUPS 64
5458#endif
5459
5460 const char *user;
5461 int i, ngroups;
5462 PyObject *list;
5463#ifdef __APPLE__
5464 int *groups, basegid;
5465#else
5466 gid_t *groups, basegid;
5467#endif
5468 ngroups = MAX_GROUPS;
5469
5470 if (!PyArg_ParseTuple(args, "si", &user, &basegid))
5471 return NULL;
5472
5473#ifdef __APPLE__
5474 groups = PyMem_Malloc(ngroups * sizeof(int));
5475#else
5476 groups = PyMem_Malloc(ngroups * sizeof(gid_t));
5477#endif
5478 if (groups == NULL)
5479 return PyErr_NoMemory();
5480
5481 if (getgrouplist(user, basegid, groups, &ngroups) == -1) {
5482 PyMem_Del(groups);
5483 return posix_error();
5484 }
5485
5486 list = PyList_New(ngroups);
5487 if (list == NULL) {
5488 PyMem_Del(groups);
5489 return NULL;
5490 }
5491
5492 for (i = 0; i < ngroups; i++) {
5493 PyObject *o = PyLong_FromUnsignedLong((unsigned long)groups[i]);
5494 if (o == NULL) {
5495 Py_DECREF(list);
5496 PyMem_Del(groups);
5497 return NULL;
5498 }
5499 PyList_SET_ITEM(list, i, o);
5500 }
5501
5502 PyMem_Del(groups);
5503
5504 return list;
5505}
5506#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005507
Fred Drakec9680921999-12-13 16:37:25 +00005508#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005509PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005510"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005511Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00005512
5513static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005514posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00005515{
5516 PyObject *result = NULL;
5517
Fred Drakec9680921999-12-13 16:37:25 +00005518#ifdef NGROUPS_MAX
5519#define MAX_GROUPS NGROUPS_MAX
5520#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005521 /* defined to be 16 on Solaris7, so this should be a small number */
Fred Drakec9680921999-12-13 16:37:25 +00005522#define MAX_GROUPS 64
5523#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00005524 gid_t grouplist[MAX_GROUPS];
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005525
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005526 /* On MacOSX getgroups(2) can return more than MAX_GROUPS results
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005527 * This is a helper variable to store the intermediate result when
5528 * that happens.
5529 *
5530 * To keep the code readable the OSX behaviour is unconditional,
5531 * according to the POSIX spec this should be safe on all unix-y
5532 * systems.
5533 */
5534 gid_t* alt_grouplist = grouplist;
Victor Stinner8c62be82010-05-06 00:08:46 +00005535 int n;
Fred Drakec9680921999-12-13 16:37:25 +00005536
Victor Stinner8c62be82010-05-06 00:08:46 +00005537 n = getgroups(MAX_GROUPS, grouplist);
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005538 if (n < 0) {
5539 if (errno == EINVAL) {
5540 n = getgroups(0, NULL);
5541 if (n == -1) {
5542 return posix_error();
5543 }
5544 if (n == 0) {
5545 /* Avoid malloc(0) */
5546 alt_grouplist = grouplist;
5547 } else {
5548 alt_grouplist = PyMem_Malloc(n * sizeof(gid_t));
5549 if (alt_grouplist == NULL) {
5550 errno = EINVAL;
5551 return posix_error();
5552 }
5553 n = getgroups(n, alt_grouplist);
5554 if (n == -1) {
5555 PyMem_Free(alt_grouplist);
5556 return posix_error();
5557 }
5558 }
5559 } else {
5560 return posix_error();
5561 }
5562 }
5563 result = PyList_New(n);
5564 if (result != NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00005565 int i;
5566 for (i = 0; i < n; ++i) {
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005567 PyObject *o = PyLong_FromLong((long)alt_grouplist[i]);
Victor Stinner8c62be82010-05-06 00:08:46 +00005568 if (o == NULL) {
Stefan Krah0e803b32010-11-26 16:16:47 +00005569 Py_DECREF(result);
5570 result = NULL;
5571 break;
Fred Drakec9680921999-12-13 16:37:25 +00005572 }
Victor Stinner8c62be82010-05-06 00:08:46 +00005573 PyList_SET_ITEM(result, i, o);
Fred Drakec9680921999-12-13 16:37:25 +00005574 }
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005575 }
5576
5577 if (alt_grouplist != grouplist) {
5578 PyMem_Free(alt_grouplist);
Victor Stinner8c62be82010-05-06 00:08:46 +00005579 }
Neal Norwitze241ce82003-02-17 18:17:05 +00005580
Fred Drakec9680921999-12-13 16:37:25 +00005581 return result;
5582}
5583#endif
5584
Antoine Pitroub7572f02009-12-02 20:46:48 +00005585#ifdef HAVE_INITGROUPS
5586PyDoc_STRVAR(posix_initgroups__doc__,
5587"initgroups(username, gid) -> None\n\n\
5588Call the system initgroups() to initialize the group access list with all of\n\
5589the groups of which the specified username is a member, plus the specified\n\
5590group id.");
5591
5592static PyObject *
5593posix_initgroups(PyObject *self, PyObject *args)
5594{
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005595 PyObject *oname;
Victor Stinner8c62be82010-05-06 00:08:46 +00005596 char *username;
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005597 int res;
Victor Stinner8c62be82010-05-06 00:08:46 +00005598 long gid;
Antoine Pitroub7572f02009-12-02 20:46:48 +00005599
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005600 if (!PyArg_ParseTuple(args, "O&l:initgroups",
5601 PyUnicode_FSConverter, &oname, &gid))
Victor Stinner8c62be82010-05-06 00:08:46 +00005602 return NULL;
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005603 username = PyBytes_AS_STRING(oname);
Antoine Pitroub7572f02009-12-02 20:46:48 +00005604
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005605 res = initgroups(username, (gid_t) gid);
5606 Py_DECREF(oname);
5607 if (res == -1)
Victor Stinner8c62be82010-05-06 00:08:46 +00005608 return PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitroub7572f02009-12-02 20:46:48 +00005609
Victor Stinner8c62be82010-05-06 00:08:46 +00005610 Py_INCREF(Py_None);
5611 return Py_None;
Antoine Pitroub7572f02009-12-02 20:46:48 +00005612}
5613#endif
5614
Martin v. Löwis606edc12002-06-13 21:09:11 +00005615#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00005616PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005617"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00005618Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00005619
5620static PyObject *
5621posix_getpgid(PyObject *self, PyObject *args)
5622{
Victor Stinner8c62be82010-05-06 00:08:46 +00005623 pid_t pid, pgid;
5624 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":getpgid", &pid))
5625 return NULL;
5626 pgid = getpgid(pid);
5627 if (pgid < 0)
5628 return posix_error();
5629 return PyLong_FromPid(pgid);
Martin v. Löwis606edc12002-06-13 21:09:11 +00005630}
5631#endif /* HAVE_GETPGID */
5632
5633
Guido van Rossumb6775db1994-08-01 11:34:53 +00005634#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005635PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005636"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005637Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005638
Barry Warsaw53699e91996-12-10 23:23:01 +00005639static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005640posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00005641{
Guido van Rossumb6775db1994-08-01 11:34:53 +00005642#ifdef GETPGRP_HAVE_ARG
Victor Stinner8c62be82010-05-06 00:08:46 +00005643 return PyLong_FromPid(getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005644#else /* GETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00005645 return PyLong_FromPid(getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005646#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00005647}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005648#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00005649
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005650
Guido van Rossumb6775db1994-08-01 11:34:53 +00005651#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005652PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005653"setpgrp()\n\n\
Senthil Kumaran684760a2010-06-17 16:48:06 +00005654Make this process the process group leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005655
Barry Warsaw53699e91996-12-10 23:23:01 +00005656static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005657posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00005658{
Guido van Rossum64933891994-10-20 21:56:42 +00005659#ifdef SETPGRP_HAVE_ARG
Victor Stinner8c62be82010-05-06 00:08:46 +00005660 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00005661#else /* SETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00005662 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00005663#endif /* SETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00005664 return posix_error();
5665 Py_INCREF(Py_None);
5666 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00005667}
5668
Guido van Rossumb6775db1994-08-01 11:34:53 +00005669#endif /* HAVE_SETPGRP */
5670
Guido van Rossumad0ee831995-03-01 10:34:45 +00005671#ifdef HAVE_GETPPID
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005672
5673#ifdef MS_WINDOWS
5674#include <tlhelp32.h>
5675
5676static PyObject*
5677win32_getppid()
5678{
5679 HANDLE snapshot;
5680 pid_t mypid;
5681 PyObject* result = NULL;
5682 BOOL have_record;
5683 PROCESSENTRY32 pe;
5684
5685 mypid = getpid(); /* This function never fails */
5686
5687 snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
5688 if (snapshot == INVALID_HANDLE_VALUE)
5689 return PyErr_SetFromWindowsErr(GetLastError());
5690
5691 pe.dwSize = sizeof(pe);
5692 have_record = Process32First(snapshot, &pe);
5693 while (have_record) {
5694 if (mypid == (pid_t)pe.th32ProcessID) {
5695 /* We could cache the ulong value in a static variable. */
5696 result = PyLong_FromPid((pid_t)pe.th32ParentProcessID);
5697 break;
5698 }
5699
5700 have_record = Process32Next(snapshot, &pe);
5701 }
5702
5703 /* If our loop exits and our pid was not found (result will be NULL)
5704 * then GetLastError will return ERROR_NO_MORE_FILES. This is an
5705 * error anyway, so let's raise it. */
5706 if (!result)
5707 result = PyErr_SetFromWindowsErr(GetLastError());
5708
5709 CloseHandle(snapshot);
5710
5711 return result;
5712}
5713#endif /*MS_WINDOWS*/
5714
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005715PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005716"getppid() -> ppid\n\n\
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005717Return the parent's process id. If the parent process has already exited,\n\
5718Windows machines will still return its id; others systems will return the id\n\
5719of the 'init' process (1).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005720
Barry Warsaw53699e91996-12-10 23:23:01 +00005721static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005722posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005723{
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005724#ifdef MS_WINDOWS
5725 return win32_getppid();
5726#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005727 return PyLong_FromPid(getppid());
Guido van Rossumad0ee831995-03-01 10:34:45 +00005728#endif
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005729}
5730#endif /* HAVE_GETPPID */
Guido van Rossum85e3b011991-06-03 12:42:10 +00005731
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005732
Fred Drake12c6e2d1999-12-14 21:25:03 +00005733#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005734PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005735"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005736Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00005737
5738static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005739posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00005740{
Victor Stinner8c62be82010-05-06 00:08:46 +00005741 PyObject *result = NULL;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005742#ifdef MS_WINDOWS
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005743 wchar_t user_name[UNLEN + 1];
Victor Stinner63941882011-09-29 00:42:28 +02005744 DWORD num_chars = Py_ARRAY_LENGTH(user_name);
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005745
5746 if (GetUserNameW(user_name, &num_chars)) {
5747 /* num_chars is the number of unicode chars plus null terminator */
5748 result = PyUnicode_FromWideChar(user_name, num_chars - 1);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005749 }
5750 else
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005751 result = PyErr_SetFromWindowsErr(GetLastError());
5752#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005753 char *name;
5754 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005755
Victor Stinner8c62be82010-05-06 00:08:46 +00005756 errno = 0;
5757 name = getlogin();
5758 if (name == NULL) {
5759 if (errno)
Victor Stinnere039ffe2010-08-15 09:33:08 +00005760 posix_error();
Fred Drake12c6e2d1999-12-14 21:25:03 +00005761 else
Victor Stinnere039ffe2010-08-15 09:33:08 +00005762 PyErr_SetString(PyExc_OSError, "unable to determine login name");
Victor Stinner8c62be82010-05-06 00:08:46 +00005763 }
5764 else
Victor Stinnere039ffe2010-08-15 09:33:08 +00005765 result = PyUnicode_DecodeFSDefault(name);
Victor Stinner8c62be82010-05-06 00:08:46 +00005766 errno = old_errno;
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005767#endif
Fred Drake12c6e2d1999-12-14 21:25:03 +00005768 return result;
5769}
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005770#endif /* HAVE_GETLOGIN */
Fred Drake12c6e2d1999-12-14 21:25:03 +00005771
Guido van Rossumad0ee831995-03-01 10:34:45 +00005772#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005773PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005774"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005775Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005776
Barry Warsaw53699e91996-12-10 23:23:01 +00005777static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005778posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005779{
Victor Stinner8c62be82010-05-06 00:08:46 +00005780 return PyLong_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005781}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005782#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005783
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005784
Guido van Rossumad0ee831995-03-01 10:34:45 +00005785#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005786PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005787"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005788Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005789
Barry Warsaw53699e91996-12-10 23:23:01 +00005790static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005791posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005792{
Victor Stinner8c62be82010-05-06 00:08:46 +00005793 pid_t pid;
5794 int sig;
5795 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:kill", &pid, &sig))
5796 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00005797#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005798 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
5799 APIRET rc;
5800 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005801 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005802
5803 } else if (sig == XCPT_SIGNAL_KILLPROC) {
5804 APIRET rc;
5805 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005806 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005807
5808 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00005809 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005810#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005811 if (kill(pid, sig) == -1)
5812 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005813#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00005814 Py_INCREF(Py_None);
5815 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00005816}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005817#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00005818
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005819#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005820PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005821"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005822Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005823
5824static PyObject *
5825posix_killpg(PyObject *self, PyObject *args)
5826{
Victor Stinner8c62be82010-05-06 00:08:46 +00005827 int sig;
5828 pid_t pgid;
5829 /* XXX some man pages make the `pgid` parameter an int, others
5830 a pid_t. Since getpgrp() returns a pid_t, we assume killpg should
5831 take the same type. Moreover, pid_t is always at least as wide as
5832 int (else compilation of this module fails), which is safe. */
5833 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:killpg", &pgid, &sig))
5834 return NULL;
5835 if (killpg(pgid, sig) == -1)
5836 return posix_error();
5837 Py_INCREF(Py_None);
5838 return Py_None;
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005839}
5840#endif
5841
Brian Curtineb24d742010-04-12 17:16:38 +00005842#ifdef MS_WINDOWS
5843PyDoc_STRVAR(win32_kill__doc__,
5844"kill(pid, sig)\n\n\
5845Kill a process with a signal.");
5846
5847static PyObject *
5848win32_kill(PyObject *self, PyObject *args)
5849{
Amaury Forgeot d'Arc0a589c92010-05-15 20:35:12 +00005850 PyObject *result;
Victor Stinner8c62be82010-05-06 00:08:46 +00005851 DWORD pid, sig, err;
5852 HANDLE handle;
Brian Curtineb24d742010-04-12 17:16:38 +00005853
Victor Stinner8c62be82010-05-06 00:08:46 +00005854 if (!PyArg_ParseTuple(args, "kk:kill", &pid, &sig))
5855 return NULL;
Brian Curtineb24d742010-04-12 17:16:38 +00005856
Victor Stinner8c62be82010-05-06 00:08:46 +00005857 /* Console processes which share a common console can be sent CTRL+C or
5858 CTRL+BREAK events, provided they handle said events. */
5859 if (sig == CTRL_C_EVENT || sig == CTRL_BREAK_EVENT) {
5860 if (GenerateConsoleCtrlEvent(sig, pid) == 0) {
5861 err = GetLastError();
5862 PyErr_SetFromWindowsErr(err);
5863 }
5864 else
5865 Py_RETURN_NONE;
5866 }
Brian Curtineb24d742010-04-12 17:16:38 +00005867
Victor Stinner8c62be82010-05-06 00:08:46 +00005868 /* If the signal is outside of what GenerateConsoleCtrlEvent can use,
5869 attempt to open and terminate the process. */
5870 handle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, pid);
5871 if (handle == NULL) {
5872 err = GetLastError();
5873 return PyErr_SetFromWindowsErr(err);
5874 }
Brian Curtineb24d742010-04-12 17:16:38 +00005875
Victor Stinner8c62be82010-05-06 00:08:46 +00005876 if (TerminateProcess(handle, sig) == 0) {
5877 err = GetLastError();
5878 result = PyErr_SetFromWindowsErr(err);
5879 } else {
5880 Py_INCREF(Py_None);
5881 result = Py_None;
5882 }
Brian Curtineb24d742010-04-12 17:16:38 +00005883
Victor Stinner8c62be82010-05-06 00:08:46 +00005884 CloseHandle(handle);
5885 return result;
Brian Curtineb24d742010-04-12 17:16:38 +00005886}
5887#endif /* MS_WINDOWS */
5888
Guido van Rossumc0125471996-06-28 18:55:32 +00005889#ifdef HAVE_PLOCK
5890
5891#ifdef HAVE_SYS_LOCK_H
5892#include <sys/lock.h>
5893#endif
5894
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005895PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005896"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005897Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005898
Barry Warsaw53699e91996-12-10 23:23:01 +00005899static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005900posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00005901{
Victor Stinner8c62be82010-05-06 00:08:46 +00005902 int op;
5903 if (!PyArg_ParseTuple(args, "i:plock", &op))
5904 return NULL;
5905 if (plock(op) == -1)
5906 return posix_error();
5907 Py_INCREF(Py_None);
5908 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00005909}
5910#endif
5911
Guido van Rossumb6775db1994-08-01 11:34:53 +00005912#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005913PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005914"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005915Set the current process's user id.");
5916
Barry Warsaw53699e91996-12-10 23:23:01 +00005917static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005918posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005919{
Victor Stinner8c62be82010-05-06 00:08:46 +00005920 long uid_arg;
5921 uid_t uid;
5922 if (!PyArg_ParseTuple(args, "l:setuid", &uid_arg))
5923 return NULL;
5924 uid = uid_arg;
5925 if (uid != uid_arg) {
5926 PyErr_SetString(PyExc_OverflowError, "user id too big");
5927 return NULL;
5928 }
5929 if (setuid(uid) < 0)
5930 return posix_error();
5931 Py_INCREF(Py_None);
5932 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005933}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005934#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005935
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005936
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005937#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005938PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005939"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005940Set the current process's effective user id.");
5941
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005942static PyObject *
5943posix_seteuid (PyObject *self, PyObject *args)
5944{
Victor Stinner8c62be82010-05-06 00:08:46 +00005945 long euid_arg;
5946 uid_t euid;
5947 if (!PyArg_ParseTuple(args, "l", &euid_arg))
5948 return NULL;
5949 euid = euid_arg;
5950 if (euid != euid_arg) {
5951 PyErr_SetString(PyExc_OverflowError, "user id too big");
5952 return NULL;
5953 }
5954 if (seteuid(euid) < 0) {
5955 return posix_error();
5956 } else {
5957 Py_INCREF(Py_None);
5958 return Py_None;
5959 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005960}
5961#endif /* HAVE_SETEUID */
5962
5963#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005964PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005965"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005966Set the current process's effective group id.");
5967
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005968static PyObject *
5969posix_setegid (PyObject *self, PyObject *args)
5970{
Victor Stinner8c62be82010-05-06 00:08:46 +00005971 long egid_arg;
5972 gid_t egid;
5973 if (!PyArg_ParseTuple(args, "l", &egid_arg))
5974 return NULL;
5975 egid = egid_arg;
5976 if (egid != egid_arg) {
5977 PyErr_SetString(PyExc_OverflowError, "group id too big");
5978 return NULL;
5979 }
5980 if (setegid(egid) < 0) {
5981 return posix_error();
5982 } else {
5983 Py_INCREF(Py_None);
5984 return Py_None;
5985 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005986}
5987#endif /* HAVE_SETEGID */
5988
5989#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005990PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00005991"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005992Set the current process's real and effective user ids.");
5993
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005994static PyObject *
5995posix_setreuid (PyObject *self, PyObject *args)
5996{
Victor Stinner8c62be82010-05-06 00:08:46 +00005997 long ruid_arg, euid_arg;
5998 uid_t ruid, euid;
5999 if (!PyArg_ParseTuple(args, "ll", &ruid_arg, &euid_arg))
6000 return NULL;
6001 if (ruid_arg == -1)
6002 ruid = (uid_t)-1; /* let the compiler choose how -1 fits */
6003 else
6004 ruid = ruid_arg; /* otherwise, assign from our long */
6005 if (euid_arg == -1)
6006 euid = (uid_t)-1;
6007 else
6008 euid = euid_arg;
6009 if ((euid_arg != -1 && euid != euid_arg) ||
6010 (ruid_arg != -1 && ruid != ruid_arg)) {
6011 PyErr_SetString(PyExc_OverflowError, "user id too big");
6012 return NULL;
6013 }
6014 if (setreuid(ruid, euid) < 0) {
6015 return posix_error();
6016 } else {
6017 Py_INCREF(Py_None);
6018 return Py_None;
6019 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006020}
6021#endif /* HAVE_SETREUID */
6022
6023#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006024PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00006025"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006026Set the current process's real and effective group ids.");
6027
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006028static PyObject *
6029posix_setregid (PyObject *self, PyObject *args)
6030{
Victor Stinner8c62be82010-05-06 00:08:46 +00006031 long rgid_arg, egid_arg;
6032 gid_t rgid, egid;
6033 if (!PyArg_ParseTuple(args, "ll", &rgid_arg, &egid_arg))
6034 return NULL;
6035 if (rgid_arg == -1)
6036 rgid = (gid_t)-1; /* let the compiler choose how -1 fits */
6037 else
6038 rgid = rgid_arg; /* otherwise, assign from our long */
6039 if (egid_arg == -1)
6040 egid = (gid_t)-1;
6041 else
6042 egid = egid_arg;
6043 if ((egid_arg != -1 && egid != egid_arg) ||
6044 (rgid_arg != -1 && rgid != rgid_arg)) {
6045 PyErr_SetString(PyExc_OverflowError, "group id too big");
6046 return NULL;
6047 }
6048 if (setregid(rgid, egid) < 0) {
6049 return posix_error();
6050 } else {
6051 Py_INCREF(Py_None);
6052 return Py_None;
6053 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006054}
6055#endif /* HAVE_SETREGID */
6056
Guido van Rossumb6775db1994-08-01 11:34:53 +00006057#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006058PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006059"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006060Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006061
Barry Warsaw53699e91996-12-10 23:23:01 +00006062static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006063posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00006064{
Victor Stinner8c62be82010-05-06 00:08:46 +00006065 long gid_arg;
6066 gid_t gid;
6067 if (!PyArg_ParseTuple(args, "l:setgid", &gid_arg))
6068 return NULL;
6069 gid = gid_arg;
6070 if (gid != gid_arg) {
6071 PyErr_SetString(PyExc_OverflowError, "group id too big");
6072 return NULL;
6073 }
6074 if (setgid(gid) < 0)
6075 return posix_error();
6076 Py_INCREF(Py_None);
6077 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00006078}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006079#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00006080
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006081#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006082PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006083"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006084Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006085
6086static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00006087posix_setgroups(PyObject *self, PyObject *groups)
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006088{
Victor Stinner8c62be82010-05-06 00:08:46 +00006089 int i, len;
6090 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00006091
Victor Stinner8c62be82010-05-06 00:08:46 +00006092 if (!PySequence_Check(groups)) {
6093 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
6094 return NULL;
6095 }
6096 len = PySequence_Size(groups);
6097 if (len > MAX_GROUPS) {
6098 PyErr_SetString(PyExc_ValueError, "too many groups");
6099 return NULL;
6100 }
6101 for(i = 0; i < len; i++) {
6102 PyObject *elem;
6103 elem = PySequence_GetItem(groups, i);
6104 if (!elem)
6105 return NULL;
6106 if (!PyLong_Check(elem)) {
6107 PyErr_SetString(PyExc_TypeError,
6108 "groups must be integers");
6109 Py_DECREF(elem);
6110 return NULL;
6111 } else {
6112 unsigned long x = PyLong_AsUnsignedLong(elem);
6113 if (PyErr_Occurred()) {
6114 PyErr_SetString(PyExc_TypeError,
6115 "group id too big");
6116 Py_DECREF(elem);
6117 return NULL;
6118 }
6119 grouplist[i] = x;
6120 /* read back the value to see if it fitted in gid_t */
6121 if (grouplist[i] != x) {
6122 PyErr_SetString(PyExc_TypeError,
6123 "group id too big");
6124 Py_DECREF(elem);
6125 return NULL;
6126 }
6127 }
6128 Py_DECREF(elem);
6129 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006130
Victor Stinner8c62be82010-05-06 00:08:46 +00006131 if (setgroups(len, grouplist) < 0)
6132 return posix_error();
6133 Py_INCREF(Py_None);
6134 return Py_None;
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006135}
6136#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006137
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006138#if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
6139static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006140wait_helper(pid_t pid, int status, struct rusage *ru)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006141{
Victor Stinner8c62be82010-05-06 00:08:46 +00006142 PyObject *result;
6143 static PyObject *struct_rusage;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02006144 _Py_IDENTIFIER(struct_rusage);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006145
Victor Stinner8c62be82010-05-06 00:08:46 +00006146 if (pid == -1)
6147 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006148
Victor Stinner8c62be82010-05-06 00:08:46 +00006149 if (struct_rusage == NULL) {
6150 PyObject *m = PyImport_ImportModuleNoBlock("resource");
6151 if (m == NULL)
6152 return NULL;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006153 struct_rusage = _PyObject_GetAttrId(m, &PyId_struct_rusage);
Victor Stinner8c62be82010-05-06 00:08:46 +00006154 Py_DECREF(m);
6155 if (struct_rusage == NULL)
6156 return NULL;
6157 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006158
Victor Stinner8c62be82010-05-06 00:08:46 +00006159 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
6160 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
6161 if (!result)
6162 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006163
6164#ifndef doubletime
6165#define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
6166#endif
6167
Victor Stinner8c62be82010-05-06 00:08:46 +00006168 PyStructSequence_SET_ITEM(result, 0,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006169 PyFloat_FromDouble(doubletime(ru->ru_utime)));
Victor Stinner8c62be82010-05-06 00:08:46 +00006170 PyStructSequence_SET_ITEM(result, 1,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006171 PyFloat_FromDouble(doubletime(ru->ru_stime)));
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006172#define SET_INT(result, index, value)\
Victor Stinner8c62be82010-05-06 00:08:46 +00006173 PyStructSequence_SET_ITEM(result, index, PyLong_FromLong(value))
6174 SET_INT(result, 2, ru->ru_maxrss);
6175 SET_INT(result, 3, ru->ru_ixrss);
6176 SET_INT(result, 4, ru->ru_idrss);
6177 SET_INT(result, 5, ru->ru_isrss);
6178 SET_INT(result, 6, ru->ru_minflt);
6179 SET_INT(result, 7, ru->ru_majflt);
6180 SET_INT(result, 8, ru->ru_nswap);
6181 SET_INT(result, 9, ru->ru_inblock);
6182 SET_INT(result, 10, ru->ru_oublock);
6183 SET_INT(result, 11, ru->ru_msgsnd);
6184 SET_INT(result, 12, ru->ru_msgrcv);
6185 SET_INT(result, 13, ru->ru_nsignals);
6186 SET_INT(result, 14, ru->ru_nvcsw);
6187 SET_INT(result, 15, ru->ru_nivcsw);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006188#undef SET_INT
6189
Victor Stinner8c62be82010-05-06 00:08:46 +00006190 if (PyErr_Occurred()) {
6191 Py_DECREF(result);
6192 return NULL;
6193 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006194
Victor Stinner8c62be82010-05-06 00:08:46 +00006195 return Py_BuildValue("NiN", PyLong_FromPid(pid), status, result);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006196}
6197#endif /* HAVE_WAIT3 || HAVE_WAIT4 */
6198
6199#ifdef HAVE_WAIT3
6200PyDoc_STRVAR(posix_wait3__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006201"wait3(options) -> (pid, status, rusage)\n\n\
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006202Wait for completion of a child process.");
6203
6204static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006205posix_wait3(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006206{
Victor Stinner8c62be82010-05-06 00:08:46 +00006207 pid_t pid;
6208 int options;
6209 struct rusage ru;
6210 WAIT_TYPE status;
6211 WAIT_STATUS_INT(status) = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006212
Victor Stinner4195b5c2012-02-08 23:03:19 +01006213 if (!PyArg_ParseTuple(args, "i:wait3", &options))
Victor Stinner8c62be82010-05-06 00:08:46 +00006214 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006215
Victor Stinner8c62be82010-05-06 00:08:46 +00006216 Py_BEGIN_ALLOW_THREADS
6217 pid = wait3(&status, options, &ru);
6218 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006219
Victor Stinner4195b5c2012-02-08 23:03:19 +01006220 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006221}
6222#endif /* HAVE_WAIT3 */
6223
6224#ifdef HAVE_WAIT4
6225PyDoc_STRVAR(posix_wait4__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006226"wait4(pid, options) -> (pid, status, rusage)\n\n\
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006227Wait for completion of a given child process.");
6228
6229static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006230posix_wait4(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006231{
Victor Stinner8c62be82010-05-06 00:08:46 +00006232 pid_t pid;
6233 int options;
6234 struct rusage ru;
6235 WAIT_TYPE status;
6236 WAIT_STATUS_INT(status) = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006237
Victor Stinner4195b5c2012-02-08 23:03:19 +01006238 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:wait4", &pid, &options))
Victor Stinner8c62be82010-05-06 00:08:46 +00006239 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006240
Victor Stinner8c62be82010-05-06 00:08:46 +00006241 Py_BEGIN_ALLOW_THREADS
6242 pid = wait4(pid, &status, options, &ru);
6243 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006244
Victor Stinner4195b5c2012-02-08 23:03:19 +01006245 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006246}
6247#endif /* HAVE_WAIT4 */
6248
Ross Lagerwall7807c352011-03-17 20:20:30 +02006249#if defined(HAVE_WAITID) && !defined(__APPLE__)
6250PyDoc_STRVAR(posix_waitid__doc__,
6251"waitid(idtype, id, options) -> waitid_result\n\n\
6252Wait for the completion of one or more child processes.\n\n\
6253idtype can be P_PID, P_PGID or P_ALL.\n\
6254id specifies the pid to wait on.\n\
6255options is constructed from the ORing of one or more of WEXITED, WSTOPPED\n\
6256or WCONTINUED and additionally may be ORed with WNOHANG or WNOWAIT.\n\
6257Returns either waitid_result or None if WNOHANG is specified and there are\n\
6258no children in a waitable state.");
6259
6260static PyObject *
6261posix_waitid(PyObject *self, PyObject *args)
6262{
6263 PyObject *result;
6264 idtype_t idtype;
6265 id_t id;
6266 int options, res;
6267 siginfo_t si;
6268 si.si_pid = 0;
6269 if (!PyArg_ParseTuple(args, "i" _Py_PARSE_PID "i:waitid", &idtype, &id, &options))
6270 return NULL;
6271 Py_BEGIN_ALLOW_THREADS
6272 res = waitid(idtype, id, &si, options);
6273 Py_END_ALLOW_THREADS
6274 if (res == -1)
6275 return posix_error();
6276
6277 if (si.si_pid == 0)
6278 Py_RETURN_NONE;
6279
6280 result = PyStructSequence_New(&WaitidResultType);
6281 if (!result)
6282 return NULL;
6283
6284 PyStructSequence_SET_ITEM(result, 0, PyLong_FromPid(si.si_pid));
6285 PyStructSequence_SET_ITEM(result, 1, PyLong_FromPid(si.si_uid));
6286 PyStructSequence_SET_ITEM(result, 2, PyLong_FromLong((long)(si.si_signo)));
6287 PyStructSequence_SET_ITEM(result, 3, PyLong_FromLong((long)(si.si_status)));
6288 PyStructSequence_SET_ITEM(result, 4, PyLong_FromLong((long)(si.si_code)));
6289 if (PyErr_Occurred()) {
6290 Py_DECREF(result);
6291 return NULL;
6292 }
6293
6294 return result;
6295}
6296#endif
6297
Guido van Rossumb6775db1994-08-01 11:34:53 +00006298#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006299PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006300"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006301Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006302
Barry Warsaw53699e91996-12-10 23:23:01 +00006303static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006304posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00006305{
Victor Stinner8c62be82010-05-06 00:08:46 +00006306 pid_t pid;
6307 int options;
6308 WAIT_TYPE status;
6309 WAIT_STATUS_INT(status) = 0;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00006310
Victor Stinner8c62be82010-05-06 00:08:46 +00006311 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:waitpid", &pid, &options))
6312 return NULL;
6313 Py_BEGIN_ALLOW_THREADS
6314 pid = waitpid(pid, &status, options);
6315 Py_END_ALLOW_THREADS
6316 if (pid == -1)
6317 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006318
Victor Stinner8c62be82010-05-06 00:08:46 +00006319 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum21803b81992-08-09 12:55:27 +00006320}
6321
Tim Petersab034fa2002-02-01 11:27:43 +00006322#elif defined(HAVE_CWAIT)
6323
6324/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006325PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006326"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006327"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00006328
6329static PyObject *
6330posix_waitpid(PyObject *self, PyObject *args)
6331{
Victor Stinner8c62be82010-05-06 00:08:46 +00006332 Py_intptr_t pid;
6333 int status, options;
Tim Petersab034fa2002-02-01 11:27:43 +00006334
Victor Stinner8c62be82010-05-06 00:08:46 +00006335 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:waitpid", &pid, &options))
6336 return NULL;
6337 Py_BEGIN_ALLOW_THREADS
6338 pid = _cwait(&status, pid, options);
6339 Py_END_ALLOW_THREADS
6340 if (pid == -1)
6341 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006342
Victor Stinner8c62be82010-05-06 00:08:46 +00006343 /* shift the status left a byte so this is more like the POSIX waitpid */
6344 return Py_BuildValue("Ni", PyLong_FromPid(pid), status << 8);
Tim Petersab034fa2002-02-01 11:27:43 +00006345}
6346#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006347
Guido van Rossumad0ee831995-03-01 10:34:45 +00006348#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006349PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006350"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006351Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006352
Barry Warsaw53699e91996-12-10 23:23:01 +00006353static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006354posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00006355{
Victor Stinner8c62be82010-05-06 00:08:46 +00006356 pid_t pid;
6357 WAIT_TYPE status;
6358 WAIT_STATUS_INT(status) = 0;
Neal Norwitze241ce82003-02-17 18:17:05 +00006359
Victor Stinner8c62be82010-05-06 00:08:46 +00006360 Py_BEGIN_ALLOW_THREADS
6361 pid = wait(&status);
6362 Py_END_ALLOW_THREADS
6363 if (pid == -1)
6364 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006365
Victor Stinner8c62be82010-05-06 00:08:46 +00006366 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum85e3b011991-06-03 12:42:10 +00006367}
Guido van Rossumad0ee831995-03-01 10:34:45 +00006368#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00006369
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006370
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006371PyDoc_STRVAR(posix_lstat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006372"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006373Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006374
Barry Warsaw53699e91996-12-10 23:23:01 +00006375static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006376posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006377{
Guido van Rossumb6775db1994-08-01 11:34:53 +00006378#ifdef HAVE_LSTAT
Victor Stinner4195b5c2012-02-08 23:03:19 +01006379 return posix_do_stat(self, args, "O&:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00006380#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006381#ifdef MS_WINDOWS
Victor Stinner4195b5c2012-02-08 23:03:19 +01006382 return posix_do_stat(self, args, "O&:lstat", win32_lstat, "U:lstat",
Brian Curtind40e6f72010-07-08 21:39:08 +00006383 win32_lstat_w);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006384#else
Victor Stinner4195b5c2012-02-08 23:03:19 +01006385 return posix_do_stat(self, args, "O&:lstat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006386#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00006387#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006388}
6389
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006390
Guido van Rossumb6775db1994-08-01 11:34:53 +00006391#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006392PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006393"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006394Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006395
Barry Warsaw53699e91996-12-10 23:23:01 +00006396static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006397posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006398{
Victor Stinner8c62be82010-05-06 00:08:46 +00006399 PyObject* v;
6400 char buf[MAXPATHLEN];
6401 PyObject *opath;
6402 char *path;
6403 int n;
6404 int arg_is_unicode = 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00006405
Victor Stinner8c62be82010-05-06 00:08:46 +00006406 if (!PyArg_ParseTuple(args, "O&:readlink",
6407 PyUnicode_FSConverter, &opath))
6408 return NULL;
6409 path = PyBytes_AsString(opath);
6410 v = PySequence_GetItem(args, 0);
6411 if (v == NULL) {
6412 Py_DECREF(opath);
6413 return NULL;
6414 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00006415
Victor Stinner8c62be82010-05-06 00:08:46 +00006416 if (PyUnicode_Check(v)) {
6417 arg_is_unicode = 1;
6418 }
6419 Py_DECREF(v);
Thomas Wouters89f507f2006-12-13 04:49:30 +00006420
Victor Stinner8c62be82010-05-06 00:08:46 +00006421 Py_BEGIN_ALLOW_THREADS
6422 n = readlink(path, buf, (int) sizeof buf);
6423 Py_END_ALLOW_THREADS
6424 if (n < 0)
6425 return posix_error_with_allocated_filename(opath);
Thomas Wouters89f507f2006-12-13 04:49:30 +00006426
Victor Stinner8c62be82010-05-06 00:08:46 +00006427 Py_DECREF(opath);
Victor Stinnera45598a2010-05-14 16:35:39 +00006428 if (arg_is_unicode)
6429 return PyUnicode_DecodeFSDefaultAndSize(buf, n);
6430 else
6431 return PyBytes_FromStringAndSize(buf, n);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006432}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006433#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006434
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006435
Brian Curtin52173d42010-12-02 18:29:18 +00006436#if defined(HAVE_SYMLINK) && !defined(MS_WINDOWS)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006437PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006438"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00006439Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006440
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006441static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006442posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006443{
Victor Stinner8c62be82010-05-06 00:08:46 +00006444 return posix_2str(args, "O&O&:symlink", symlink);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006445}
6446#endif /* HAVE_SYMLINK */
6447
Brian Curtind40e6f72010-07-08 21:39:08 +00006448#if !defined(HAVE_READLINK) && defined(MS_WINDOWS)
6449
6450PyDoc_STRVAR(win_readlink__doc__,
6451"readlink(path) -> path\n\n\
6452Return a string representing the path to which the symbolic link points.");
6453
Brian Curtind40e6f72010-07-08 21:39:08 +00006454/* Windows readlink implementation */
6455static PyObject *
6456win_readlink(PyObject *self, PyObject *args)
6457{
6458 wchar_t *path;
6459 DWORD n_bytes_returned;
6460 DWORD io_result;
Victor Stinnereb5657a2011-09-30 01:44:27 +02006461 PyObject *po, *result;
Brian Curtind40e6f72010-07-08 21:39:08 +00006462 HANDLE reparse_point_handle;
6463
6464 char target_buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
6465 REPARSE_DATA_BUFFER *rdb = (REPARSE_DATA_BUFFER *)target_buffer;
6466 wchar_t *print_name;
6467
6468 if (!PyArg_ParseTuple(args,
Victor Stinnereb5657a2011-09-30 01:44:27 +02006469 "U:readlink",
6470 &po))
6471 return NULL;
6472 path = PyUnicode_AsUnicode(po);
6473 if (path == NULL)
Brian Curtind40e6f72010-07-08 21:39:08 +00006474 return NULL;
6475
6476 /* First get a handle to the reparse point */
6477 Py_BEGIN_ALLOW_THREADS
6478 reparse_point_handle = CreateFileW(
6479 path,
6480 0,
6481 0,
6482 0,
6483 OPEN_EXISTING,
6484 FILE_FLAG_OPEN_REPARSE_POINT|FILE_FLAG_BACKUP_SEMANTICS,
6485 0);
6486 Py_END_ALLOW_THREADS
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006487
Brian Curtind40e6f72010-07-08 21:39:08 +00006488 if (reparse_point_handle==INVALID_HANDLE_VALUE)
Victor Stinnereb5657a2011-09-30 01:44:27 +02006489 return win32_error_object("readlink", po);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006490
Brian Curtind40e6f72010-07-08 21:39:08 +00006491 Py_BEGIN_ALLOW_THREADS
6492 /* New call DeviceIoControl to read the reparse point */
6493 io_result = DeviceIoControl(
6494 reparse_point_handle,
6495 FSCTL_GET_REPARSE_POINT,
6496 0, 0, /* in buffer */
6497 target_buffer, sizeof(target_buffer),
6498 &n_bytes_returned,
6499 0 /* we're not using OVERLAPPED_IO */
6500 );
6501 CloseHandle(reparse_point_handle);
6502 Py_END_ALLOW_THREADS
6503
6504 if (io_result==0)
Victor Stinnereb5657a2011-09-30 01:44:27 +02006505 return win32_error_object("readlink", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00006506
6507 if (rdb->ReparseTag != IO_REPARSE_TAG_SYMLINK)
6508 {
6509 PyErr_SetString(PyExc_ValueError,
6510 "not a symbolic link");
6511 return NULL;
6512 }
Brian Curtin74e45612010-07-09 15:58:59 +00006513 print_name = rdb->SymbolicLinkReparseBuffer.PathBuffer +
6514 rdb->SymbolicLinkReparseBuffer.PrintNameOffset;
6515
6516 result = PyUnicode_FromWideChar(print_name,
6517 rdb->SymbolicLinkReparseBuffer.PrintNameLength/2);
Brian Curtind40e6f72010-07-08 21:39:08 +00006518 return result;
6519}
6520
6521#endif /* !defined(HAVE_READLINK) && defined(MS_WINDOWS) */
6522
Brian Curtin52173d42010-12-02 18:29:18 +00006523#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtind40e6f72010-07-08 21:39:08 +00006524
6525/* Grab CreateSymbolicLinkW dynamically from kernel32 */
6526static int has_CreateSymbolicLinkW = 0;
6527static DWORD (CALLBACK *Py_CreateSymbolicLinkW)(LPWSTR, LPWSTR, DWORD);
6528static int
6529check_CreateSymbolicLinkW()
6530{
6531 HINSTANCE hKernel32;
6532 /* only recheck */
6533 if (has_CreateSymbolicLinkW)
6534 return has_CreateSymbolicLinkW;
Martin v. Löwis50590f12012-01-14 17:54:09 +01006535 hKernel32 = GetModuleHandleW(L"KERNEL32");
Brian Curtin74e45612010-07-09 15:58:59 +00006536 *(FARPROC*)&Py_CreateSymbolicLinkW = GetProcAddress(hKernel32,
6537 "CreateSymbolicLinkW");
Brian Curtind40e6f72010-07-08 21:39:08 +00006538 if (Py_CreateSymbolicLinkW)
6539 has_CreateSymbolicLinkW = 1;
6540 return has_CreateSymbolicLinkW;
6541}
6542
6543PyDoc_STRVAR(win_symlink__doc__,
6544"symlink(src, dst, target_is_directory=False)\n\n\
6545Create a symbolic link pointing to src named dst.\n\
6546target_is_directory is required if the target is to be interpreted as\n\
6547a directory.\n\
6548This function requires Windows 6.0 or greater, and raises a\n\
6549NotImplementedError otherwise.");
6550
6551static PyObject *
6552win_symlink(PyObject *self, PyObject *args, PyObject *kwargs)
6553{
6554 static char *kwlist[] = {"src", "dest", "target_is_directory", NULL};
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006555 PyObject *osrc, *odest;
6556 PyObject *usrc = NULL, *udest = NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02006557 wchar_t *wsrc, *wdest;
Brian Curtind40e6f72010-07-08 21:39:08 +00006558 int target_is_directory = 0;
6559 DWORD res;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006560
Brian Curtind40e6f72010-07-08 21:39:08 +00006561 if (!check_CreateSymbolicLinkW())
6562 {
6563 /* raise NotImplementedError */
6564 return PyErr_Format(PyExc_NotImplementedError,
6565 "CreateSymbolicLinkW not found");
6566 }
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006567 if (!PyArg_ParseTupleAndKeywords(
6568 args, kwargs, "OO|i:symlink", kwlist,
6569 &osrc, &odest, &target_is_directory))
Brian Curtind40e6f72010-07-08 21:39:08 +00006570 return NULL;
Brian Curtin3b4499c2010-12-28 14:31:47 +00006571
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006572 usrc = win32_decode_filename(osrc);
6573 if (!usrc)
6574 return NULL;
6575 udest = win32_decode_filename(odest);
6576 if (!udest)
6577 goto error;
6578
Brian Curtin3b4499c2010-12-28 14:31:47 +00006579 if (win32_can_symlink == 0)
6580 return PyErr_Format(PyExc_OSError, "symbolic link privilege not held");
6581
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006582 wsrc = PyUnicode_AsUnicode(usrc);
Victor Stinnereb5657a2011-09-30 01:44:27 +02006583 if (wsrc == NULL)
6584 goto error;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006585 wdest = PyUnicode_AsUnicode(udest);
Victor Stinnereb5657a2011-09-30 01:44:27 +02006586 if (wsrc == NULL)
6587 goto error;
6588
Brian Curtind40e6f72010-07-08 21:39:08 +00006589 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006590 res = Py_CreateSymbolicLinkW(wdest, wsrc, target_is_directory);
Brian Curtind40e6f72010-07-08 21:39:08 +00006591 Py_END_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006592
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006593 Py_DECREF(usrc);
6594 Py_DECREF(udest);
Brian Curtind40e6f72010-07-08 21:39:08 +00006595 if (!res)
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006596 return win32_error_object("symlink", osrc);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006597
Brian Curtind40e6f72010-07-08 21:39:08 +00006598 Py_INCREF(Py_None);
6599 return Py_None;
Victor Stinnereb5657a2011-09-30 01:44:27 +02006600
6601error:
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006602 Py_XDECREF(usrc);
6603 Py_XDECREF(udest);
Victor Stinnereb5657a2011-09-30 01:44:27 +02006604 return NULL;
Brian Curtind40e6f72010-07-08 21:39:08 +00006605}
Brian Curtin52173d42010-12-02 18:29:18 +00006606#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006607
6608#ifdef HAVE_TIMES
Guido van Rossumd48f2521997-12-05 22:19:34 +00006609#if defined(PYCC_VACPP) && defined(PYOS_OS2)
6610static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00006611system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00006612{
6613 ULONG value = 0;
6614
6615 Py_BEGIN_ALLOW_THREADS
6616 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
6617 Py_END_ALLOW_THREADS
6618
6619 return value;
6620}
6621
6622static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006623posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00006624{
Guido van Rossumd48f2521997-12-05 22:19:34 +00006625 /* Currently Only Uptime is Provided -- Others Later */
Victor Stinner8c62be82010-05-06 00:08:46 +00006626 return Py_BuildValue("ddddd",
6627 (double)0 /* t.tms_utime / HZ */,
6628 (double)0 /* t.tms_stime / HZ */,
6629 (double)0 /* t.tms_cutime / HZ */,
6630 (double)0 /* t.tms_cstime / HZ */,
6631 (double)system_uptime() / 1000);
Guido van Rossumd48f2521997-12-05 22:19:34 +00006632}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006633#else /* not OS2 */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00006634#define NEED_TICKS_PER_SECOND
6635static long ticks_per_second = -1;
Barry Warsaw53699e91996-12-10 23:23:01 +00006636static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006637posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00006638{
Victor Stinner8c62be82010-05-06 00:08:46 +00006639 struct tms t;
6640 clock_t c;
6641 errno = 0;
6642 c = times(&t);
6643 if (c == (clock_t) -1)
6644 return posix_error();
6645 return Py_BuildValue("ddddd",
6646 (double)t.tms_utime / ticks_per_second,
6647 (double)t.tms_stime / ticks_per_second,
6648 (double)t.tms_cutime / ticks_per_second,
6649 (double)t.tms_cstime / ticks_per_second,
6650 (double)c / ticks_per_second);
Guido van Rossum22db57e1992-04-05 14:25:30 +00006651}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006652#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006653#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006654
6655
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006656#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00006657#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00006658static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006659posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00006660{
Victor Stinner8c62be82010-05-06 00:08:46 +00006661 FILETIME create, exit, kernel, user;
6662 HANDLE hProc;
6663 hProc = GetCurrentProcess();
6664 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
6665 /* The fields of a FILETIME structure are the hi and lo part
6666 of a 64-bit value expressed in 100 nanosecond units.
6667 1e7 is one second in such units; 1e-7 the inverse.
6668 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
6669 */
6670 return Py_BuildValue(
6671 "ddddd",
6672 (double)(user.dwHighDateTime*429.4967296 +
6673 user.dwLowDateTime*1e-7),
6674 (double)(kernel.dwHighDateTime*429.4967296 +
6675 kernel.dwLowDateTime*1e-7),
6676 (double)0,
6677 (double)0,
6678 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00006679}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006680#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006681
6682#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006683PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006684"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006685Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006686#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00006687
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006688
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00006689#ifdef HAVE_GETSID
6690PyDoc_STRVAR(posix_getsid__doc__,
6691"getsid(pid) -> sid\n\n\
6692Call the system call getsid().");
6693
6694static PyObject *
6695posix_getsid(PyObject *self, PyObject *args)
6696{
Victor Stinner8c62be82010-05-06 00:08:46 +00006697 pid_t pid;
6698 int sid;
6699 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":getsid", &pid))
6700 return NULL;
6701 sid = getsid(pid);
6702 if (sid < 0)
6703 return posix_error();
6704 return PyLong_FromLong((long)sid);
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00006705}
6706#endif /* HAVE_GETSID */
6707
6708
Guido van Rossumb6775db1994-08-01 11:34:53 +00006709#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006710PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006711"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006712Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006713
Barry Warsaw53699e91996-12-10 23:23:01 +00006714static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006715posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00006716{
Victor Stinner8c62be82010-05-06 00:08:46 +00006717 if (setsid() < 0)
6718 return posix_error();
6719 Py_INCREF(Py_None);
6720 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00006721}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006722#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00006723
Guido van Rossumb6775db1994-08-01 11:34:53 +00006724#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006725PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006726"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006727Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006728
Barry Warsaw53699e91996-12-10 23:23:01 +00006729static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006730posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00006731{
Victor Stinner8c62be82010-05-06 00:08:46 +00006732 pid_t pid;
6733 int pgrp;
6734 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:setpgid", &pid, &pgrp))
6735 return NULL;
6736 if (setpgid(pid, pgrp) < 0)
6737 return posix_error();
6738 Py_INCREF(Py_None);
6739 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00006740}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006741#endif /* HAVE_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00006742
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006743
Guido van Rossumb6775db1994-08-01 11:34:53 +00006744#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006745PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006746"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006747Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006748
Barry Warsaw53699e91996-12-10 23:23:01 +00006749static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006750posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00006751{
Victor Stinner8c62be82010-05-06 00:08:46 +00006752 int fd;
6753 pid_t pgid;
6754 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
6755 return NULL;
6756 pgid = tcgetpgrp(fd);
6757 if (pgid < 0)
6758 return posix_error();
6759 return PyLong_FromPid(pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00006760}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006761#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00006762
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006763
Guido van Rossumb6775db1994-08-01 11:34:53 +00006764#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006765PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006766"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006767Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006768
Barry Warsaw53699e91996-12-10 23:23:01 +00006769static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006770posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00006771{
Victor Stinner8c62be82010-05-06 00:08:46 +00006772 int fd;
6773 pid_t pgid;
6774 if (!PyArg_ParseTuple(args, "i" _Py_PARSE_PID ":tcsetpgrp", &fd, &pgid))
6775 return NULL;
6776 if (tcsetpgrp(fd, pgid) < 0)
6777 return posix_error();
6778 Py_INCREF(Py_None);
6779 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00006780}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006781#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00006782
Guido van Rossum687dd131993-05-17 08:34:16 +00006783/* Functions acting on file descriptors */
6784
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006785PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006786"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006787Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006788
Barry Warsaw53699e91996-12-10 23:23:01 +00006789static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006790posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006791{
Victor Stinner8c62be82010-05-06 00:08:46 +00006792 PyObject *ofile;
6793 char *file;
6794 int flag;
6795 int mode = 0777;
6796 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006797
6798#ifdef MS_WINDOWS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006799 PyObject *po;
Victor Stinner26de69d2011-06-17 15:15:38 +02006800 if (PyArg_ParseTuple(args, "Ui|i:open", &po, &flag, &mode)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02006801 wchar_t *wpath = PyUnicode_AsUnicode(po);
6802 if (wpath == NULL)
6803 return NULL;
6804
Victor Stinner8c62be82010-05-06 00:08:46 +00006805 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006806 fd = _wopen(wpath, flag, mode);
Victor Stinner8c62be82010-05-06 00:08:46 +00006807 Py_END_ALLOW_THREADS
6808 if (fd < 0)
6809 return posix_error();
6810 return PyLong_FromLong((long)fd);
6811 }
6812 /* Drop the argument parsing error as narrow strings
6813 are also valid. */
6814 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006815#endif
6816
Victor Stinner26de69d2011-06-17 15:15:38 +02006817 if (!PyArg_ParseTuple(args, "O&i|i:open",
Victor Stinner8c62be82010-05-06 00:08:46 +00006818 PyUnicode_FSConverter, &ofile,
6819 &flag, &mode))
6820 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006821#ifdef MS_WINDOWS
6822 if (win32_warn_bytes_api()) {
6823 Py_DECREF(ofile);
6824 return NULL;
6825 }
6826#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006827 file = PyBytes_AsString(ofile);
6828 Py_BEGIN_ALLOW_THREADS
6829 fd = open(file, flag, mode);
6830 Py_END_ALLOW_THREADS
6831 if (fd < 0)
6832 return posix_error_with_allocated_filename(ofile);
6833 Py_DECREF(ofile);
6834 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006835}
6836
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006837
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006838PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006839"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006840Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006841
Barry Warsaw53699e91996-12-10 23:23:01 +00006842static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006843posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006844{
Victor Stinner8c62be82010-05-06 00:08:46 +00006845 int fd, res;
6846 if (!PyArg_ParseTuple(args, "i:close", &fd))
6847 return NULL;
6848 if (!_PyVerify_fd(fd))
6849 return posix_error();
6850 Py_BEGIN_ALLOW_THREADS
6851 res = close(fd);
6852 Py_END_ALLOW_THREADS
6853 if (res < 0)
6854 return posix_error();
6855 Py_INCREF(Py_None);
6856 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006857}
6858
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006859
Victor Stinner8c62be82010-05-06 00:08:46 +00006860PyDoc_STRVAR(posix_closerange__doc__,
Christian Heimesfdab48e2008-01-20 09:06:41 +00006861"closerange(fd_low, fd_high)\n\n\
6862Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
6863
6864static PyObject *
6865posix_closerange(PyObject *self, PyObject *args)
6866{
Victor Stinner8c62be82010-05-06 00:08:46 +00006867 int fd_from, fd_to, i;
6868 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
6869 return NULL;
6870 Py_BEGIN_ALLOW_THREADS
6871 for (i = fd_from; i < fd_to; i++)
6872 if (_PyVerify_fd(i))
6873 close(i);
6874 Py_END_ALLOW_THREADS
6875 Py_RETURN_NONE;
Christian Heimesfdab48e2008-01-20 09:06:41 +00006876}
6877
6878
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006879PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006880"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006881Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006882
Barry Warsaw53699e91996-12-10 23:23:01 +00006883static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006884posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006885{
Victor Stinner8c62be82010-05-06 00:08:46 +00006886 int fd;
6887 if (!PyArg_ParseTuple(args, "i:dup", &fd))
6888 return NULL;
6889 if (!_PyVerify_fd(fd))
6890 return posix_error();
Victor Stinner8c62be82010-05-06 00:08:46 +00006891 fd = dup(fd);
Victor Stinner8c62be82010-05-06 00:08:46 +00006892 if (fd < 0)
6893 return posix_error();
6894 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006895}
6896
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006897
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006898PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00006899"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006900Duplicate file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006901
Barry Warsaw53699e91996-12-10 23:23:01 +00006902static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006903posix_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006904{
Victor Stinner8c62be82010-05-06 00:08:46 +00006905 int fd, fd2, res;
6906 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
6907 return NULL;
6908 if (!_PyVerify_fd_dup2(fd, fd2))
6909 return posix_error();
Victor Stinner8c62be82010-05-06 00:08:46 +00006910 res = dup2(fd, fd2);
Victor Stinner8c62be82010-05-06 00:08:46 +00006911 if (res < 0)
6912 return posix_error();
6913 Py_INCREF(Py_None);
6914 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006915}
6916
Ross Lagerwall7807c352011-03-17 20:20:30 +02006917#ifdef HAVE_LOCKF
6918PyDoc_STRVAR(posix_lockf__doc__,
6919"lockf(fd, cmd, len)\n\n\
6920Apply, test or remove a POSIX lock on an open file descriptor.\n\n\
6921fd is an open file descriptor.\n\
6922cmd specifies the command to use - one of F_LOCK, F_TLOCK, F_ULOCK or\n\
6923F_TEST.\n\
6924len specifies the section of the file to lock.");
6925
6926static PyObject *
6927posix_lockf(PyObject *self, PyObject *args)
6928{
6929 int fd, cmd, res;
6930 off_t len;
6931 if (!PyArg_ParseTuple(args, "iiO&:lockf",
6932 &fd, &cmd, _parse_off_t, &len))
6933 return NULL;
6934
6935 Py_BEGIN_ALLOW_THREADS
6936 res = lockf(fd, cmd, len);
6937 Py_END_ALLOW_THREADS
6938
6939 if (res < 0)
6940 return posix_error();
6941
6942 Py_RETURN_NONE;
6943}
6944#endif
6945
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006946
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006947PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006948"lseek(fd, pos, how) -> newpos\n\n\
Victor Stinnere83f8992011-12-17 23:15:09 +01006949Set the current position of a file descriptor.\n\
6950Return the new cursor position in bytes, starting from the beginning.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006951
Barry Warsaw53699e91996-12-10 23:23:01 +00006952static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006953posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006954{
Victor Stinner8c62be82010-05-06 00:08:46 +00006955 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006956#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00006957 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006958#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006959 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006960#endif
Ross Lagerwall8e749672011-03-17 21:54:07 +02006961 PyObject *posobj;
6962 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Victor Stinner8c62be82010-05-06 00:08:46 +00006963 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00006964#ifdef SEEK_SET
Victor Stinner8c62be82010-05-06 00:08:46 +00006965 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
6966 switch (how) {
6967 case 0: how = SEEK_SET; break;
6968 case 1: how = SEEK_CUR; break;
6969 case 2: how = SEEK_END; break;
6970 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006971#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00006972
Ross Lagerwall8e749672011-03-17 21:54:07 +02006973#if !defined(HAVE_LARGEFILE_SUPPORT)
6974 pos = PyLong_AsLong(posobj);
6975#else
6976 pos = PyLong_AsLongLong(posobj);
6977#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006978 if (PyErr_Occurred())
6979 return NULL;
Guido van Rossum94f6f721999-01-06 18:42:14 +00006980
Victor Stinner8c62be82010-05-06 00:08:46 +00006981 if (!_PyVerify_fd(fd))
6982 return posix_error();
6983 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006984#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00006985 res = _lseeki64(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00006986#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006987 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00006988#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006989 Py_END_ALLOW_THREADS
6990 if (res < 0)
6991 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00006992
6993#if !defined(HAVE_LARGEFILE_SUPPORT)
Victor Stinner8c62be82010-05-06 00:08:46 +00006994 return PyLong_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006995#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006996 return PyLong_FromLongLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006997#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00006998}
6999
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007000
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007001PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007002"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007003Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007004
Barry Warsaw53699e91996-12-10 23:23:01 +00007005static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007006posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00007007{
Victor Stinner8c62be82010-05-06 00:08:46 +00007008 int fd, size;
7009 Py_ssize_t n;
7010 PyObject *buffer;
7011 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
7012 return NULL;
7013 if (size < 0) {
7014 errno = EINVAL;
7015 return posix_error();
7016 }
7017 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
7018 if (buffer == NULL)
7019 return NULL;
Stefan Krah99439262010-11-26 12:58:05 +00007020 if (!_PyVerify_fd(fd)) {
7021 Py_DECREF(buffer);
Victor Stinner8c62be82010-05-06 00:08:46 +00007022 return posix_error();
Stefan Krah99439262010-11-26 12:58:05 +00007023 }
Victor Stinner8c62be82010-05-06 00:08:46 +00007024 Py_BEGIN_ALLOW_THREADS
7025 n = read(fd, PyBytes_AS_STRING(buffer), size);
7026 Py_END_ALLOW_THREADS
7027 if (n < 0) {
7028 Py_DECREF(buffer);
7029 return posix_error();
7030 }
7031 if (n != size)
7032 _PyBytes_Resize(&buffer, n);
7033 return buffer;
Guido van Rossum687dd131993-05-17 08:34:16 +00007034}
7035
Ross Lagerwall7807c352011-03-17 20:20:30 +02007036#if (defined(HAVE_SENDFILE) && (defined(__FreeBSD__) || defined(__DragonFly__) \
7037 || defined(__APPLE__))) || defined(HAVE_READV) || defined(HAVE_WRITEV)
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007038static Py_ssize_t
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007039iov_setup(struct iovec **iov, Py_buffer **buf, PyObject *seq, int cnt, int type)
7040{
7041 int i, j;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007042 Py_ssize_t blen, total = 0;
7043
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007044 *iov = PyMem_New(struct iovec, cnt);
7045 if (*iov == NULL) {
7046 PyErr_NoMemory();
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007047 return total;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007048 }
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007049
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007050 *buf = PyMem_New(Py_buffer, cnt);
7051 if (*buf == NULL) {
7052 PyMem_Del(*iov);
7053 PyErr_NoMemory();
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007054 return total;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007055 }
7056
7057 for (i = 0; i < cnt; i++) {
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02007058 PyObject *item = PySequence_GetItem(seq, i);
7059 if (item == NULL)
7060 goto fail;
7061 if (PyObject_GetBuffer(item, &(*buf)[i], type) == -1) {
7062 Py_DECREF(item);
7063 goto fail;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007064 }
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02007065 Py_DECREF(item);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007066 (*iov)[i].iov_base = (*buf)[i].buf;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007067 blen = (*buf)[i].len;
7068 (*iov)[i].iov_len = blen;
7069 total += blen;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007070 }
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007071 return total;
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02007072
7073fail:
7074 PyMem_Del(*iov);
7075 for (j = 0; j < i; j++) {
7076 PyBuffer_Release(&(*buf)[j]);
7077 }
7078 PyMem_Del(*buf);
7079 return 0;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007080}
7081
7082static void
7083iov_cleanup(struct iovec *iov, Py_buffer *buf, int cnt)
7084{
7085 int i;
7086 PyMem_Del(iov);
7087 for (i = 0; i < cnt; i++) {
7088 PyBuffer_Release(&buf[i]);
7089 }
7090 PyMem_Del(buf);
7091}
7092#endif
7093
Ross Lagerwall7807c352011-03-17 20:20:30 +02007094#ifdef HAVE_READV
7095PyDoc_STRVAR(posix_readv__doc__,
7096"readv(fd, buffers) -> bytesread\n\n\
7097Read from a file descriptor into a number of writable buffers. buffers\n\
7098is an arbitrary sequence of writable buffers.\n\
7099Returns the total number of bytes read.");
7100
7101static PyObject *
7102posix_readv(PyObject *self, PyObject *args)
7103{
7104 int fd, cnt;
7105 Py_ssize_t n;
7106 PyObject *seq;
7107 struct iovec *iov;
7108 Py_buffer *buf;
7109
7110 if (!PyArg_ParseTuple(args, "iO:readv", &fd, &seq))
7111 return NULL;
7112 if (!PySequence_Check(seq)) {
7113 PyErr_SetString(PyExc_TypeError,
7114 "readv() arg 2 must be a sequence");
7115 return NULL;
7116 }
7117 cnt = PySequence_Size(seq);
7118
7119 if (!iov_setup(&iov, &buf, seq, cnt, PyBUF_WRITABLE))
7120 return NULL;
7121
7122 Py_BEGIN_ALLOW_THREADS
7123 n = readv(fd, iov, cnt);
7124 Py_END_ALLOW_THREADS
7125
7126 iov_cleanup(iov, buf, cnt);
7127 return PyLong_FromSsize_t(n);
7128}
7129#endif
7130
7131#ifdef HAVE_PREAD
7132PyDoc_STRVAR(posix_pread__doc__,
7133"pread(fd, buffersize, offset) -> string\n\n\
7134Read from a file descriptor, fd, at a position of offset. It will read up\n\
7135to buffersize number of bytes. The file offset remains unchanged.");
7136
7137static PyObject *
7138posix_pread(PyObject *self, PyObject *args)
7139{
7140 int fd, size;
7141 off_t offset;
7142 Py_ssize_t n;
7143 PyObject *buffer;
7144 if (!PyArg_ParseTuple(args, "iiO&:pread", &fd, &size, _parse_off_t, &offset))
7145 return NULL;
7146
7147 if (size < 0) {
7148 errno = EINVAL;
7149 return posix_error();
7150 }
7151 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
7152 if (buffer == NULL)
7153 return NULL;
7154 if (!_PyVerify_fd(fd)) {
7155 Py_DECREF(buffer);
7156 return posix_error();
7157 }
7158 Py_BEGIN_ALLOW_THREADS
7159 n = pread(fd, PyBytes_AS_STRING(buffer), size, offset);
7160 Py_END_ALLOW_THREADS
7161 if (n < 0) {
7162 Py_DECREF(buffer);
7163 return posix_error();
7164 }
7165 if (n != size)
7166 _PyBytes_Resize(&buffer, n);
7167 return buffer;
7168}
7169#endif
7170
7171PyDoc_STRVAR(posix_write__doc__,
7172"write(fd, string) -> byteswritten\n\n\
7173Write a string to a file descriptor.");
7174
7175static PyObject *
7176posix_write(PyObject *self, PyObject *args)
7177{
7178 Py_buffer pbuf;
7179 int fd;
7180 Py_ssize_t size, len;
7181
7182 if (!PyArg_ParseTuple(args, "iy*:write", &fd, &pbuf))
7183 return NULL;
7184 if (!_PyVerify_fd(fd)) {
7185 PyBuffer_Release(&pbuf);
7186 return posix_error();
7187 }
7188 len = pbuf.len;
7189 Py_BEGIN_ALLOW_THREADS
7190#if defined(MS_WIN64) || defined(MS_WINDOWS)
7191 if (len > INT_MAX)
7192 len = INT_MAX;
7193 size = write(fd, pbuf.buf, (int)len);
7194#else
7195 size = write(fd, pbuf.buf, len);
7196#endif
7197 Py_END_ALLOW_THREADS
7198 PyBuffer_Release(&pbuf);
7199 if (size < 0)
7200 return posix_error();
7201 return PyLong_FromSsize_t(size);
7202}
7203
7204#ifdef HAVE_SENDFILE
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007205PyDoc_STRVAR(posix_sendfile__doc__,
7206"sendfile(out, in, offset, nbytes) -> byteswritten\n\
7207sendfile(out, in, offset, nbytes, headers=None, trailers=None, flags=0)\n\
7208 -> byteswritten\n\
7209Copy nbytes bytes from file descriptor in to file descriptor out.");
7210
7211static PyObject *
7212posix_sendfile(PyObject *self, PyObject *args, PyObject *kwdict)
7213{
7214 int in, out;
7215 Py_ssize_t ret;
7216 off_t offset;
7217
7218#if defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__)
7219#ifndef __APPLE__
7220 Py_ssize_t len;
7221#endif
7222 PyObject *headers = NULL, *trailers = NULL;
7223 Py_buffer *hbuf, *tbuf;
7224 off_t sbytes;
7225 struct sf_hdtr sf;
7226 int flags = 0;
7227 sf.headers = NULL;
7228 sf.trailers = NULL;
Benjamin Petersond8a43b42011-02-26 21:35:16 +00007229 static char *keywords[] = {"out", "in",
7230 "offset", "count",
7231 "headers", "trailers", "flags", NULL};
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007232
7233#ifdef __APPLE__
7234 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiO&O&|OOi:sendfile",
Georg Brandla391b112011-02-25 15:23:18 +00007235 keywords, &out, &in, _parse_off_t, &offset, _parse_off_t, &sbytes,
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007236#else
7237 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiO&n|OOi:sendfile",
Georg Brandla391b112011-02-25 15:23:18 +00007238 keywords, &out, &in, _parse_off_t, &offset, &len,
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007239#endif
7240 &headers, &trailers, &flags))
7241 return NULL;
7242 if (headers != NULL) {
7243 if (!PySequence_Check(headers)) {
7244 PyErr_SetString(PyExc_TypeError,
7245 "sendfile() headers must be a sequence or None");
7246 return NULL;
7247 } else {
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007248 Py_ssize_t i = 0; /* Avoid uninitialized warning */
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007249 sf.hdr_cnt = PySequence_Size(headers);
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007250 if (sf.hdr_cnt > 0 &&
7251 !(i = iov_setup(&(sf.headers), &hbuf,
7252 headers, sf.hdr_cnt, PyBUF_SIMPLE)))
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007253 return NULL;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007254#ifdef __APPLE__
7255 sbytes += i;
7256#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007257 }
7258 }
7259 if (trailers != NULL) {
7260 if (!PySequence_Check(trailers)) {
7261 PyErr_SetString(PyExc_TypeError,
7262 "sendfile() trailers must be a sequence or None");
7263 return NULL;
7264 } else {
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007265 Py_ssize_t i = 0; /* Avoid uninitialized warning */
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007266 sf.trl_cnt = PySequence_Size(trailers);
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007267 if (sf.trl_cnt > 0 &&
7268 !(i = iov_setup(&(sf.trailers), &tbuf,
7269 trailers, sf.trl_cnt, PyBUF_SIMPLE)))
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007270 return NULL;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007271#ifdef __APPLE__
7272 sbytes += i;
7273#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007274 }
7275 }
7276
7277 Py_BEGIN_ALLOW_THREADS
7278#ifdef __APPLE__
7279 ret = sendfile(in, out, offset, &sbytes, &sf, flags);
7280#else
7281 ret = sendfile(in, out, offset, len, &sf, &sbytes, flags);
7282#endif
7283 Py_END_ALLOW_THREADS
7284
7285 if (sf.headers != NULL)
7286 iov_cleanup(sf.headers, hbuf, sf.hdr_cnt);
7287 if (sf.trailers != NULL)
7288 iov_cleanup(sf.trailers, tbuf, sf.trl_cnt);
7289
7290 if (ret < 0) {
7291 if ((errno == EAGAIN) || (errno == EBUSY)) {
7292 if (sbytes != 0) {
7293 // some data has been sent
7294 goto done;
7295 }
7296 else {
7297 // no data has been sent; upper application is supposed
7298 // to retry on EAGAIN or EBUSY
7299 return posix_error();
7300 }
7301 }
7302 return posix_error();
7303 }
7304 goto done;
7305
7306done:
7307 #if !defined(HAVE_LARGEFILE_SUPPORT)
7308 return Py_BuildValue("l", sbytes);
7309 #else
7310 return Py_BuildValue("L", sbytes);
7311 #endif
7312
7313#else
7314 Py_ssize_t count;
7315 PyObject *offobj;
7316 static char *keywords[] = {"out", "in",
7317 "offset", "count", NULL};
7318 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiOn:sendfile",
7319 keywords, &out, &in, &offobj, &count))
7320 return NULL;
7321#ifdef linux
7322 if (offobj == Py_None) {
7323 Py_BEGIN_ALLOW_THREADS
7324 ret = sendfile(out, in, NULL, count);
7325 Py_END_ALLOW_THREADS
7326 if (ret < 0)
7327 return posix_error();
Giampaolo Rodola'ff1a7352011-04-19 09:47:16 +02007328 return Py_BuildValue("n", ret);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007329 }
7330#endif
Antoine Pitroudcc20b82011-02-26 13:38:35 +00007331 if (!_parse_off_t(offobj, &offset))
7332 return NULL;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007333 Py_BEGIN_ALLOW_THREADS
7334 ret = sendfile(out, in, &offset, count);
7335 Py_END_ALLOW_THREADS
7336 if (ret < 0)
7337 return posix_error();
7338 return Py_BuildValue("n", ret);
7339#endif
7340}
7341#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007342
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007343PyDoc_STRVAR(posix_fstat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01007344"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007345Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007346
Barry Warsaw53699e91996-12-10 23:23:01 +00007347static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01007348posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00007349{
Victor Stinner8c62be82010-05-06 00:08:46 +00007350 int fd;
7351 STRUCT_STAT st;
7352 int res;
Victor Stinner4195b5c2012-02-08 23:03:19 +01007353 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
Victor Stinner8c62be82010-05-06 00:08:46 +00007354 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00007355#ifdef __VMS
Victor Stinner8c62be82010-05-06 00:08:46 +00007356 /* on OpenVMS we must ensure that all bytes are written to the file */
7357 fsync(fd);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00007358#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00007359 if (!_PyVerify_fd(fd))
7360 return posix_error();
7361 Py_BEGIN_ALLOW_THREADS
7362 res = FSTAT(fd, &st);
7363 Py_END_ALLOW_THREADS
7364 if (res != 0) {
Martin v. Löwis14694662006-02-03 12:54:16 +00007365#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00007366 return win32_error("fstat", NULL);
Martin v. Löwis14694662006-02-03 12:54:16 +00007367#else
Victor Stinner8c62be82010-05-06 00:08:46 +00007368 return posix_error();
Martin v. Löwis14694662006-02-03 12:54:16 +00007369#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00007370 }
Tim Peters5aa91602002-01-30 05:46:57 +00007371
Victor Stinner4195b5c2012-02-08 23:03:19 +01007372 return _pystat_fromstructstat(&st);
Guido van Rossum687dd131993-05-17 08:34:16 +00007373}
7374
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007375PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007376"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00007377Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007378connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00007379
7380static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00007381posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00007382{
Victor Stinner8c62be82010-05-06 00:08:46 +00007383 int fd;
7384 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
7385 return NULL;
7386 if (!_PyVerify_fd(fd))
7387 return PyBool_FromLong(0);
7388 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00007389}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007390
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007391#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007392PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007393"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007394Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007395
Barry Warsaw53699e91996-12-10 23:23:01 +00007396static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00007397posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00007398{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007399#if defined(PYOS_OS2)
7400 HFILE read, write;
7401 APIRET rc;
7402
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007403 rc = DosCreatePipe( &read, &write, 4096);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007404 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007405 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007406
7407 return Py_BuildValue("(ii)", read, write);
7408#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007409#if !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00007410 int fds[2];
7411 int res;
Victor Stinner8c62be82010-05-06 00:08:46 +00007412 res = pipe(fds);
Victor Stinner8c62be82010-05-06 00:08:46 +00007413 if (res != 0)
7414 return posix_error();
7415 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007416#else /* MS_WINDOWS */
Victor Stinner8c62be82010-05-06 00:08:46 +00007417 HANDLE read, write;
7418 int read_fd, write_fd;
7419 BOOL ok;
Victor Stinner8c62be82010-05-06 00:08:46 +00007420 ok = CreatePipe(&read, &write, NULL, 0);
Victor Stinner8c62be82010-05-06 00:08:46 +00007421 if (!ok)
7422 return win32_error("CreatePipe", NULL);
7423 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
7424 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
7425 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007426#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007427#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00007428}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007429#endif /* HAVE_PIPE */
7430
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007431#ifdef HAVE_PIPE2
7432PyDoc_STRVAR(posix_pipe2__doc__,
Charles-François Natali368f34b2011-06-06 19:49:47 +02007433"pipe2(flags) -> (read_end, write_end)\n\n\
7434Create a pipe with flags set atomically.\n\
7435flags can be constructed by ORing together one or more of these values:\n\
7436O_NONBLOCK, O_CLOEXEC.\n\
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007437");
7438
7439static PyObject *
Charles-François Natali368f34b2011-06-06 19:49:47 +02007440posix_pipe2(PyObject *self, PyObject *arg)
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007441{
Charles-François Natali368f34b2011-06-06 19:49:47 +02007442 int flags;
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007443 int fds[2];
7444 int res;
7445
Charles-François Natali368f34b2011-06-06 19:49:47 +02007446 flags = PyLong_AsLong(arg);
7447 if (flags == -1 && PyErr_Occurred())
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007448 return NULL;
7449
7450 res = pipe2(fds, flags);
7451 if (res != 0)
7452 return posix_error();
7453 return Py_BuildValue("(ii)", fds[0], fds[1]);
7454}
7455#endif /* HAVE_PIPE2 */
7456
Ross Lagerwall7807c352011-03-17 20:20:30 +02007457#ifdef HAVE_WRITEV
7458PyDoc_STRVAR(posix_writev__doc__,
7459"writev(fd, buffers) -> byteswritten\n\n\
7460Write the contents of buffers to a file descriptor, where buffers is an\n\
7461arbitrary sequence of buffers.\n\
7462Returns the total bytes written.");
7463
7464static PyObject *
7465posix_writev(PyObject *self, PyObject *args)
7466{
7467 int fd, cnt;
7468 Py_ssize_t res;
7469 PyObject *seq;
7470 struct iovec *iov;
7471 Py_buffer *buf;
7472 if (!PyArg_ParseTuple(args, "iO:writev", &fd, &seq))
7473 return NULL;
7474 if (!PySequence_Check(seq)) {
7475 PyErr_SetString(PyExc_TypeError,
7476 "writev() arg 2 must be a sequence");
7477 return NULL;
7478 }
7479 cnt = PySequence_Size(seq);
7480
7481 if (!iov_setup(&iov, &buf, seq, cnt, PyBUF_SIMPLE)) {
7482 return NULL;
7483 }
7484
7485 Py_BEGIN_ALLOW_THREADS
7486 res = writev(fd, iov, cnt);
7487 Py_END_ALLOW_THREADS
7488
7489 iov_cleanup(iov, buf, cnt);
7490 return PyLong_FromSsize_t(res);
7491}
7492#endif
7493
7494#ifdef HAVE_PWRITE
7495PyDoc_STRVAR(posix_pwrite__doc__,
7496"pwrite(fd, string, offset) -> byteswritten\n\n\
7497Write string to a file descriptor, fd, from offset, leaving the file\n\
7498offset unchanged.");
7499
7500static PyObject *
7501posix_pwrite(PyObject *self, PyObject *args)
7502{
7503 Py_buffer pbuf;
7504 int fd;
7505 off_t offset;
7506 Py_ssize_t size;
7507
7508 if (!PyArg_ParseTuple(args, "iy*O&:pwrite", &fd, &pbuf, _parse_off_t, &offset))
7509 return NULL;
7510
7511 if (!_PyVerify_fd(fd)) {
7512 PyBuffer_Release(&pbuf);
7513 return posix_error();
7514 }
7515 Py_BEGIN_ALLOW_THREADS
7516 size = pwrite(fd, pbuf.buf, (size_t)pbuf.len, offset);
7517 Py_END_ALLOW_THREADS
7518 PyBuffer_Release(&pbuf);
7519 if (size < 0)
7520 return posix_error();
7521 return PyLong_FromSsize_t(size);
7522}
7523#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007524
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007525#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007526PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00007527"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007528Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007529
Barry Warsaw53699e91996-12-10 23:23:01 +00007530static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007531posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007532{
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007533 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00007534 char *filename;
7535 int mode = 0666;
7536 int res;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007537 if (!PyArg_ParseTuple(args, "O&|i:mkfifo", PyUnicode_FSConverter, &opath,
7538 &mode))
Victor Stinner8c62be82010-05-06 00:08:46 +00007539 return NULL;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007540 filename = PyBytes_AS_STRING(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007541 Py_BEGIN_ALLOW_THREADS
7542 res = mkfifo(filename, mode);
7543 Py_END_ALLOW_THREADS
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007544 Py_DECREF(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007545 if (res < 0)
7546 return posix_error();
7547 Py_INCREF(Py_None);
7548 return Py_None;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007549}
7550#endif
7551
7552
Neal Norwitz11690112002-07-30 01:08:28 +00007553#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007554PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00007555"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007556Create a filesystem node (file, device special file or named pipe)\n\
7557named filename. mode specifies both the permissions to use and the\n\
7558type of node to be created, being combined (bitwise OR) with one of\n\
7559S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007560device defines the newly created device special file (probably using\n\
7561os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007562
7563
7564static PyObject *
7565posix_mknod(PyObject *self, PyObject *args)
7566{
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007567 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00007568 char *filename;
7569 int mode = 0600;
7570 int device = 0;
7571 int res;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007572 if (!PyArg_ParseTuple(args, "O&|ii:mknod", PyUnicode_FSConverter, &opath,
7573 &mode, &device))
Victor Stinner8c62be82010-05-06 00:08:46 +00007574 return NULL;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007575 filename = PyBytes_AS_STRING(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007576 Py_BEGIN_ALLOW_THREADS
7577 res = mknod(filename, mode, device);
7578 Py_END_ALLOW_THREADS
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007579 Py_DECREF(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007580 if (res < 0)
7581 return posix_error();
7582 Py_INCREF(Py_None);
7583 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007584}
7585#endif
7586
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007587#ifdef HAVE_DEVICE_MACROS
7588PyDoc_STRVAR(posix_major__doc__,
7589"major(device) -> major number\n\
7590Extracts a device major number from a raw device number.");
7591
7592static PyObject *
7593posix_major(PyObject *self, PyObject *args)
7594{
Victor Stinner8c62be82010-05-06 00:08:46 +00007595 int device;
7596 if (!PyArg_ParseTuple(args, "i:major", &device))
7597 return NULL;
7598 return PyLong_FromLong((long)major(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007599}
7600
7601PyDoc_STRVAR(posix_minor__doc__,
7602"minor(device) -> minor number\n\
7603Extracts a device minor number from a raw device number.");
7604
7605static PyObject *
7606posix_minor(PyObject *self, PyObject *args)
7607{
Victor Stinner8c62be82010-05-06 00:08:46 +00007608 int device;
7609 if (!PyArg_ParseTuple(args, "i:minor", &device))
7610 return NULL;
7611 return PyLong_FromLong((long)minor(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007612}
7613
7614PyDoc_STRVAR(posix_makedev__doc__,
7615"makedev(major, minor) -> device number\n\
7616Composes a raw device number from the major and minor device numbers.");
7617
7618static PyObject *
7619posix_makedev(PyObject *self, PyObject *args)
7620{
Victor Stinner8c62be82010-05-06 00:08:46 +00007621 int major, minor;
7622 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
7623 return NULL;
7624 return PyLong_FromLong((long)makedev(major, minor));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007625}
7626#endif /* device macros */
7627
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007628
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007629#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007630PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007631"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007632Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007633
Barry Warsaw53699e91996-12-10 23:23:01 +00007634static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007635posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007636{
Victor Stinner8c62be82010-05-06 00:08:46 +00007637 int fd;
7638 off_t length;
7639 int res;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007640
Ross Lagerwall7807c352011-03-17 20:20:30 +02007641 if (!PyArg_ParseTuple(args, "iO&:ftruncate", &fd, _parse_off_t, &length))
Victor Stinner8c62be82010-05-06 00:08:46 +00007642 return NULL;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007643
Victor Stinner8c62be82010-05-06 00:08:46 +00007644 Py_BEGIN_ALLOW_THREADS
7645 res = ftruncate(fd, length);
7646 Py_END_ALLOW_THREADS
7647 if (res < 0)
7648 return posix_error();
7649 Py_INCREF(Py_None);
7650 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007651}
7652#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00007653
Ross Lagerwall7807c352011-03-17 20:20:30 +02007654#ifdef HAVE_TRUNCATE
7655PyDoc_STRVAR(posix_truncate__doc__,
7656"truncate(path, length)\n\n\
7657Truncate the file given by path to length bytes.");
7658
7659static PyObject *
7660posix_truncate(PyObject *self, PyObject *args)
7661{
7662 PyObject *opath;
7663 const char *path;
7664 off_t length;
7665 int res;
7666
7667 if (!PyArg_ParseTuple(args, "O&O&:truncate",
7668 PyUnicode_FSConverter, &opath, _parse_off_t, &length))
7669 return NULL;
7670 path = PyBytes_AsString(opath);
7671
7672 Py_BEGIN_ALLOW_THREADS
7673 res = truncate(path, length);
7674 Py_END_ALLOW_THREADS
7675 Py_DECREF(opath);
7676 if (res < 0)
7677 return posix_error();
7678 Py_RETURN_NONE;
7679}
7680#endif
7681
7682#ifdef HAVE_POSIX_FALLOCATE
7683PyDoc_STRVAR(posix_posix_fallocate__doc__,
7684"posix_fallocate(fd, offset, len)\n\n\
7685Ensures that enough disk space is allocated for the file specified by fd\n\
7686starting from offset and continuing for len bytes.");
7687
7688static PyObject *
7689posix_posix_fallocate(PyObject *self, PyObject *args)
7690{
7691 off_t len, offset;
7692 int res, fd;
7693
7694 if (!PyArg_ParseTuple(args, "iO&O&:posix_fallocate",
7695 &fd, _parse_off_t, &offset, _parse_off_t, &len))
7696 return NULL;
7697
7698 Py_BEGIN_ALLOW_THREADS
7699 res = posix_fallocate(fd, offset, len);
7700 Py_END_ALLOW_THREADS
7701 if (res != 0) {
7702 errno = res;
7703 return posix_error();
7704 }
7705 Py_RETURN_NONE;
7706}
7707#endif
7708
7709#ifdef HAVE_POSIX_FADVISE
7710PyDoc_STRVAR(posix_posix_fadvise__doc__,
7711"posix_fadvise(fd, offset, len, advice)\n\n\
7712Announces an intention to access data in a specific pattern thus allowing\n\
7713the kernel to make optimizations.\n\
7714The advice applies to the region of the file specified by fd starting at\n\
7715offset and continuing for len bytes.\n\
7716advice is one of POSIX_FADV_NORMAL, POSIX_FADV_SEQUENTIAL,\n\
7717POSIX_FADV_RANDOM, POSIX_FADV_NOREUSE, POSIX_FADV_WILLNEED or\n\
7718POSIX_FADV_DONTNEED.");
7719
7720static PyObject *
7721posix_posix_fadvise(PyObject *self, PyObject *args)
7722{
7723 off_t len, offset;
7724 int res, fd, advice;
7725
7726 if (!PyArg_ParseTuple(args, "iO&O&i:posix_fadvise",
7727 &fd, _parse_off_t, &offset, _parse_off_t, &len, &advice))
7728 return NULL;
7729
7730 Py_BEGIN_ALLOW_THREADS
7731 res = posix_fadvise(fd, offset, len, advice);
7732 Py_END_ALLOW_THREADS
7733 if (res != 0) {
7734 errno = res;
7735 return posix_error();
7736 }
7737 Py_RETURN_NONE;
7738}
7739#endif
7740
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007741#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007742PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007743"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007744Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007745
Fred Drake762e2061999-08-26 17:23:54 +00007746/* Save putenv() parameters as values here, so we can collect them when they
7747 * get re-set with another call for the same key. */
7748static PyObject *posix_putenv_garbage;
7749
Tim Peters5aa91602002-01-30 05:46:57 +00007750static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007751posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007752{
Victor Stinner84ae1182010-05-06 22:05:07 +00007753 PyObject *newstr = NULL;
Thomas Hellerf78f12a2007-11-08 19:33:05 +00007754#ifdef MS_WINDOWS
Victor Stinner65170952011-11-22 22:16:17 +01007755 PyObject *os1, *os2;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007756 wchar_t *newenv;
Victor Stinner65170952011-11-22 22:16:17 +01007757
Victor Stinner8c62be82010-05-06 00:08:46 +00007758 if (!PyArg_ParseTuple(args,
Victor Stinner9d3b93b2011-11-22 02:27:30 +01007759 "UU:putenv",
Victor Stinner65170952011-11-22 22:16:17 +01007760 &os1, &os2))
Victor Stinner8c62be82010-05-06 00:08:46 +00007761 return NULL;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007762
Victor Stinner65170952011-11-22 22:16:17 +01007763 newstr = PyUnicode_FromFormat("%U=%U", os1, os2);
Victor Stinner84ae1182010-05-06 22:05:07 +00007764 if (newstr == NULL) {
7765 PyErr_NoMemory();
7766 goto error;
7767 }
Victor Stinner65170952011-11-22 22:16:17 +01007768 if (_MAX_ENV < PyUnicode_GET_LENGTH(newstr)) {
7769 PyErr_Format(PyExc_ValueError,
7770 "the environment variable is longer than %u characters",
7771 _MAX_ENV);
7772 goto error;
7773 }
7774
Victor Stinner8c62be82010-05-06 00:08:46 +00007775 newenv = PyUnicode_AsUnicode(newstr);
Victor Stinnereb5657a2011-09-30 01:44:27 +02007776 if (newenv == NULL)
7777 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00007778 if (_wputenv(newenv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007779 posix_error();
Victor Stinner84ae1182010-05-06 22:05:07 +00007780 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00007781 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00007782#else
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007783 PyObject *os1, *os2;
7784 char *s1, *s2;
7785 char *newenv;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007786
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007787 if (!PyArg_ParseTuple(args,
7788 "O&O&:putenv",
7789 PyUnicode_FSConverter, &os1,
7790 PyUnicode_FSConverter, &os2))
7791 return NULL;
7792 s1 = PyBytes_AsString(os1);
7793 s2 = PyBytes_AsString(os2);
Guido van Rossumad0ee831995-03-01 10:34:45 +00007794
Victor Stinner65170952011-11-22 22:16:17 +01007795 newstr = PyBytes_FromFormat("%s=%s", s1, s2);
Victor Stinner9d3b93b2011-11-22 02:27:30 +01007796 if (newstr == NULL) {
7797 PyErr_NoMemory();
7798 goto error;
7799 }
7800
Victor Stinner8c62be82010-05-06 00:08:46 +00007801 newenv = PyBytes_AS_STRING(newstr);
Victor Stinner8c62be82010-05-06 00:08:46 +00007802 if (putenv(newenv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007803 posix_error();
Victor Stinner84ae1182010-05-06 22:05:07 +00007804 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00007805 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00007806#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007807
Victor Stinner8c62be82010-05-06 00:08:46 +00007808 /* Install the first arg and newstr in posix_putenv_garbage;
7809 * this will cause previous value to be collected. This has to
7810 * happen after the real putenv() call because the old value
7811 * was still accessible until then. */
Victor Stinner65170952011-11-22 22:16:17 +01007812 if (PyDict_SetItem(posix_putenv_garbage, os1, newstr)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007813 /* really not much we can do; just leak */
7814 PyErr_Clear();
7815 }
7816 else {
7817 Py_DECREF(newstr);
7818 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00007819
Martin v. Löwis011e8422009-05-05 04:43:17 +00007820#ifndef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00007821 Py_DECREF(os1);
7822 Py_DECREF(os2);
Martin v. Löwis011e8422009-05-05 04:43:17 +00007823#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007824 Py_RETURN_NONE;
7825
7826error:
7827#ifndef MS_WINDOWS
7828 Py_DECREF(os1);
7829 Py_DECREF(os2);
7830#endif
7831 Py_XDECREF(newstr);
7832 return NULL;
Guido van Rossumb6775db1994-08-01 11:34:53 +00007833}
Guido van Rossumb6a47161997-09-15 22:54:34 +00007834#endif /* putenv */
7835
Guido van Rossumc524d952001-10-19 01:31:59 +00007836#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007837PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007838"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007839Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00007840
7841static PyObject *
7842posix_unsetenv(PyObject *self, PyObject *args)
7843{
Victor Stinner65170952011-11-22 22:16:17 +01007844 PyObject *name;
Victor Stinner984890f2011-11-24 13:53:38 +01007845#ifndef HAVE_BROKEN_UNSETENV
Victor Stinner60b385e2011-11-22 22:01:28 +01007846 int err;
Victor Stinner984890f2011-11-24 13:53:38 +01007847#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007848
7849 if (!PyArg_ParseTuple(args, "O&:unsetenv",
Guido van Rossumc524d952001-10-19 01:31:59 +00007850
Victor Stinner65170952011-11-22 22:16:17 +01007851 PyUnicode_FSConverter, &name))
Guido van Rossumc524d952001-10-19 01:31:59 +00007852 return NULL;
Guido van Rossumc524d952001-10-19 01:31:59 +00007853
Victor Stinner984890f2011-11-24 13:53:38 +01007854#ifdef HAVE_BROKEN_UNSETENV
7855 unsetenv(PyBytes_AS_STRING(name));
7856#else
Victor Stinner65170952011-11-22 22:16:17 +01007857 err = unsetenv(PyBytes_AS_STRING(name));
Benjamin Peterson4bb867d2011-11-22 23:12:49 -06007858 if (err) {
Benjamin Petersone8eb0e82011-11-22 23:14:47 -06007859 Py_DECREF(name);
Victor Stinner60b385e2011-11-22 22:01:28 +01007860 return posix_error();
Benjamin Peterson4bb867d2011-11-22 23:12:49 -06007861 }
Victor Stinner984890f2011-11-24 13:53:38 +01007862#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00007863
Victor Stinner8c62be82010-05-06 00:08:46 +00007864 /* Remove the key from posix_putenv_garbage;
7865 * this will cause it to be collected. This has to
7866 * happen after the real unsetenv() call because the
7867 * old value was still accessible until then.
7868 */
Victor Stinner65170952011-11-22 22:16:17 +01007869 if (PyDict_DelItem(posix_putenv_garbage, name)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007870 /* really not much we can do; just leak */
7871 PyErr_Clear();
7872 }
Victor Stinner65170952011-11-22 22:16:17 +01007873 Py_DECREF(name);
Victor Stinner84ae1182010-05-06 22:05:07 +00007874 Py_RETURN_NONE;
Guido van Rossumc524d952001-10-19 01:31:59 +00007875}
7876#endif /* unsetenv */
7877
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007878PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007879"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007880Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00007881
Guido van Rossumf68d8e52001-04-14 17:55:09 +00007882static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007883posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00007884{
Victor Stinner8c62be82010-05-06 00:08:46 +00007885 int code;
7886 char *message;
7887 if (!PyArg_ParseTuple(args, "i:strerror", &code))
7888 return NULL;
7889 message = strerror(code);
7890 if (message == NULL) {
7891 PyErr_SetString(PyExc_ValueError,
7892 "strerror() argument out of range");
7893 return NULL;
7894 }
Victor Stinner1b579672011-12-17 05:47:23 +01007895 return PyUnicode_DecodeLocale(message, "surrogateescape");
Guido van Rossumb6a47161997-09-15 22:54:34 +00007896}
Guido van Rossumb6a47161997-09-15 22:54:34 +00007897
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007898
Guido van Rossumc9641791998-08-04 15:26:23 +00007899#ifdef HAVE_SYS_WAIT_H
7900
Fred Drake106c1a02002-04-23 15:58:02 +00007901#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007902PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007903"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007904Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00007905
7906static PyObject *
7907posix_WCOREDUMP(PyObject *self, PyObject *args)
7908{
Victor Stinner8c62be82010-05-06 00:08:46 +00007909 WAIT_TYPE status;
7910 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00007911
Victor Stinner8c62be82010-05-06 00:08:46 +00007912 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
7913 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00007914
Victor Stinner8c62be82010-05-06 00:08:46 +00007915 return PyBool_FromLong(WCOREDUMP(status));
Fred Drake106c1a02002-04-23 15:58:02 +00007916}
7917#endif /* WCOREDUMP */
7918
7919#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007920PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007921"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00007922Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007923job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00007924
7925static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00007926posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00007927{
Victor Stinner8c62be82010-05-06 00:08:46 +00007928 WAIT_TYPE status;
7929 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00007930
Victor Stinner8c62be82010-05-06 00:08:46 +00007931 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
7932 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00007933
Victor Stinner8c62be82010-05-06 00:08:46 +00007934 return PyBool_FromLong(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00007935}
7936#endif /* WIFCONTINUED */
7937
Guido van Rossumc9641791998-08-04 15:26:23 +00007938#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007939PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007940"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007941Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007942
7943static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007944posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007945{
Victor Stinner8c62be82010-05-06 00:08:46 +00007946 WAIT_TYPE status;
7947 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007948
Victor Stinner8c62be82010-05-06 00:08:46 +00007949 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
7950 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007951
Victor Stinner8c62be82010-05-06 00:08:46 +00007952 return PyBool_FromLong(WIFSTOPPED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007953}
7954#endif /* WIFSTOPPED */
7955
7956#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007957PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007958"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007959Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007960
7961static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007962posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007963{
Victor Stinner8c62be82010-05-06 00:08:46 +00007964 WAIT_TYPE status;
7965 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007966
Victor Stinner8c62be82010-05-06 00:08:46 +00007967 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
7968 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007969
Victor Stinner8c62be82010-05-06 00:08:46 +00007970 return PyBool_FromLong(WIFSIGNALED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007971}
7972#endif /* WIFSIGNALED */
7973
7974#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007975PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007976"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00007977Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007978system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007979
7980static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007981posix_WIFEXITED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007982{
Victor Stinner8c62be82010-05-06 00:08:46 +00007983 WAIT_TYPE status;
7984 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007985
Victor Stinner8c62be82010-05-06 00:08:46 +00007986 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
7987 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007988
Victor Stinner8c62be82010-05-06 00:08:46 +00007989 return PyBool_FromLong(WIFEXITED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007990}
7991#endif /* WIFEXITED */
7992
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00007993#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007994PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007995"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007996Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007997
7998static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007999posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00008000{
Victor Stinner8c62be82010-05-06 00:08:46 +00008001 WAIT_TYPE status;
8002 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00008003
Victor Stinner8c62be82010-05-06 00:08:46 +00008004 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
8005 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00008006
Victor Stinner8c62be82010-05-06 00:08:46 +00008007 return Py_BuildValue("i", WEXITSTATUS(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008008}
8009#endif /* WEXITSTATUS */
8010
8011#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008012PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008013"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00008014Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008015value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00008016
8017static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008018posix_WTERMSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00008019{
Victor Stinner8c62be82010-05-06 00:08:46 +00008020 WAIT_TYPE status;
8021 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00008022
Victor Stinner8c62be82010-05-06 00:08:46 +00008023 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
8024 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00008025
Victor Stinner8c62be82010-05-06 00:08:46 +00008026 return Py_BuildValue("i", WTERMSIG(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008027}
8028#endif /* WTERMSIG */
8029
8030#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008031PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008032"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008033Return the signal that stopped the process that provided\n\
8034the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00008035
8036static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008037posix_WSTOPSIG(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:WSTOPSIG", &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", WSTOPSIG(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008046}
8047#endif /* WSTOPSIG */
8048
8049#endif /* HAVE_SYS_WAIT_H */
8050
8051
Thomas Wouters477c8d52006-05-27 19:21:47 +00008052#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossumd5753e11999-10-19 13:29:23 +00008053#ifdef _SCO_DS
8054/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
8055 needed definitions in sys/statvfs.h */
8056#define _SVID3
8057#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00008058#include <sys/statvfs.h>
8059
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008060static PyObject*
8061_pystatvfs_fromstructstatvfs(struct statvfs st) {
Victor Stinner8c62be82010-05-06 00:08:46 +00008062 PyObject *v = PyStructSequence_New(&StatVFSResultType);
8063 if (v == NULL)
8064 return NULL;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008065
8066#if !defined(HAVE_LARGEFILE_SUPPORT)
Victor Stinner8c62be82010-05-06 00:08:46 +00008067 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
8068 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
8069 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long) st.f_blocks));
8070 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long) st.f_bfree));
8071 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long) st.f_bavail));
8072 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long) st.f_files));
8073 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong((long) st.f_ffree));
8074 PyStructSequence_SET_ITEM(v, 7, PyLong_FromLong((long) st.f_favail));
8075 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
8076 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008077#else
Victor Stinner8c62be82010-05-06 00:08:46 +00008078 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
8079 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
8080 PyStructSequence_SET_ITEM(v, 2,
8081 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
8082 PyStructSequence_SET_ITEM(v, 3,
8083 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
8084 PyStructSequence_SET_ITEM(v, 4,
8085 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
8086 PyStructSequence_SET_ITEM(v, 5,
8087 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
8088 PyStructSequence_SET_ITEM(v, 6,
8089 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
8090 PyStructSequence_SET_ITEM(v, 7,
8091 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
8092 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
8093 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008094#endif
8095
Victor Stinner8c62be82010-05-06 00:08:46 +00008096 return v;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008097}
8098
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008099PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008100"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008101Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00008102
8103static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008104posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00008105{
Victor Stinner8c62be82010-05-06 00:08:46 +00008106 int fd, res;
8107 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008108
Victor Stinner8c62be82010-05-06 00:08:46 +00008109 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
8110 return NULL;
8111 Py_BEGIN_ALLOW_THREADS
8112 res = fstatvfs(fd, &st);
8113 Py_END_ALLOW_THREADS
8114 if (res != 0)
8115 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008116
Victor Stinner8c62be82010-05-06 00:08:46 +00008117 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00008118}
Thomas Wouters477c8d52006-05-27 19:21:47 +00008119#endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00008120
8121
Thomas Wouters477c8d52006-05-27 19:21:47 +00008122#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossum94f6f721999-01-06 18:42:14 +00008123#include <sys/statvfs.h>
8124
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008125PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008126"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008127Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00008128
8129static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008130posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00008131{
Victor Stinner6fa67772011-09-20 04:04:33 +02008132 PyObject *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00008133 int res;
8134 struct statvfs st;
Victor Stinner6fa67772011-09-20 04:04:33 +02008135 if (!PyArg_ParseTuple(args, "O&:statvfs", PyUnicode_FSConverter, &path))
Victor Stinner8c62be82010-05-06 00:08:46 +00008136 return NULL;
8137 Py_BEGIN_ALLOW_THREADS
Victor Stinner6fa67772011-09-20 04:04:33 +02008138 res = statvfs(PyBytes_AS_STRING(path), &st);
Victor Stinner8c62be82010-05-06 00:08:46 +00008139 Py_END_ALLOW_THREADS
Victor Stinner6fa67772011-09-20 04:04:33 +02008140 if (res != 0) {
8141 posix_error_with_filename(PyBytes_AS_STRING(path));
8142 Py_DECREF(path);
8143 return NULL;
8144 }
8145 Py_DECREF(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008146
Victor Stinner8c62be82010-05-06 00:08:46 +00008147 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00008148}
8149#endif /* HAVE_STATVFS */
8150
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008151#ifdef MS_WINDOWS
8152PyDoc_STRVAR(win32__getdiskusage__doc__,
8153"_getdiskusage(path) -> (total, free)\n\n\
8154Return disk usage statistics about the given path as (total, free) tuple.");
8155
8156static PyObject *
8157win32__getdiskusage(PyObject *self, PyObject *args)
8158{
8159 BOOL retval;
8160 ULARGE_INTEGER _, total, free;
Victor Stinner6139c1b2011-11-09 22:14:14 +01008161 const wchar_t *path;
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008162
Victor Stinner6139c1b2011-11-09 22:14:14 +01008163 if (! PyArg_ParseTuple(args, "u", &path))
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008164 return NULL;
8165
8166 Py_BEGIN_ALLOW_THREADS
Victor Stinner6139c1b2011-11-09 22:14:14 +01008167 retval = GetDiskFreeSpaceExW(path, &_, &total, &free);
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008168 Py_END_ALLOW_THREADS
8169 if (retval == 0)
8170 return PyErr_SetFromWindowsErr(0);
8171
8172 return Py_BuildValue("(LL)", total.QuadPart, free.QuadPart);
8173}
8174#endif
8175
8176
Fred Drakec9680921999-12-13 16:37:25 +00008177/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
8178 * It maps strings representing configuration variable names to
8179 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00008180 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00008181 * rarely-used constants. There are three separate tables that use
8182 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00008183 *
8184 * This code is always included, even if none of the interfaces that
8185 * need it are included. The #if hackery needed to avoid it would be
8186 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00008187 */
8188struct constdef {
8189 char *name;
8190 long value;
8191};
8192
Fred Drake12c6e2d1999-12-14 21:25:03 +00008193static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008194conv_confname(PyObject *arg, int *valuep, struct constdef *table,
Guido van Rossum7d5baac2007-08-27 23:24:46 +00008195 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00008196{
Christian Heimes217cfd12007-12-02 14:31:20 +00008197 if (PyLong_Check(arg)) {
Stefan Krah0e803b32010-11-26 16:16:47 +00008198 *valuep = PyLong_AS_LONG(arg);
8199 return 1;
Fred Drake12c6e2d1999-12-14 21:25:03 +00008200 }
Guido van Rossumbce56a62007-05-10 18:04:33 +00008201 else {
Stefan Krah0e803b32010-11-26 16:16:47 +00008202 /* look up the value in the table using a binary search */
8203 size_t lo = 0;
8204 size_t mid;
8205 size_t hi = tablesize;
8206 int cmp;
8207 const char *confname;
8208 if (!PyUnicode_Check(arg)) {
8209 PyErr_SetString(PyExc_TypeError,
8210 "configuration names must be strings or integers");
8211 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00008212 }
Stefan Krah0e803b32010-11-26 16:16:47 +00008213 confname = _PyUnicode_AsString(arg);
8214 if (confname == NULL)
8215 return 0;
8216 while (lo < hi) {
8217 mid = (lo + hi) / 2;
8218 cmp = strcmp(confname, table[mid].name);
8219 if (cmp < 0)
8220 hi = mid;
8221 else if (cmp > 0)
8222 lo = mid + 1;
8223 else {
8224 *valuep = table[mid].value;
8225 return 1;
8226 }
8227 }
8228 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
8229 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00008230 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00008231}
8232
8233
8234#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
8235static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00008236#ifdef _PC_ABI_AIO_XFER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008237 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
Fred Draked86ed291999-12-15 15:34:33 +00008238#endif
8239#ifdef _PC_ABI_ASYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008240 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
Fred Draked86ed291999-12-15 15:34:33 +00008241#endif
Fred Drakec9680921999-12-13 16:37:25 +00008242#ifdef _PC_ASYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008243 {"PC_ASYNC_IO", _PC_ASYNC_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008244#endif
8245#ifdef _PC_CHOWN_RESTRICTED
Victor Stinner8c62be82010-05-06 00:08:46 +00008246 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
Fred Drakec9680921999-12-13 16:37:25 +00008247#endif
8248#ifdef _PC_FILESIZEBITS
Victor Stinner8c62be82010-05-06 00:08:46 +00008249 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
Fred Drakec9680921999-12-13 16:37:25 +00008250#endif
8251#ifdef _PC_LAST
Victor Stinner8c62be82010-05-06 00:08:46 +00008252 {"PC_LAST", _PC_LAST},
Fred Drakec9680921999-12-13 16:37:25 +00008253#endif
8254#ifdef _PC_LINK_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008255 {"PC_LINK_MAX", _PC_LINK_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008256#endif
8257#ifdef _PC_MAX_CANON
Victor Stinner8c62be82010-05-06 00:08:46 +00008258 {"PC_MAX_CANON", _PC_MAX_CANON},
Fred Drakec9680921999-12-13 16:37:25 +00008259#endif
8260#ifdef _PC_MAX_INPUT
Victor Stinner8c62be82010-05-06 00:08:46 +00008261 {"PC_MAX_INPUT", _PC_MAX_INPUT},
Fred Drakec9680921999-12-13 16:37:25 +00008262#endif
8263#ifdef _PC_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008264 {"PC_NAME_MAX", _PC_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008265#endif
8266#ifdef _PC_NO_TRUNC
Victor Stinner8c62be82010-05-06 00:08:46 +00008267 {"PC_NO_TRUNC", _PC_NO_TRUNC},
Fred Drakec9680921999-12-13 16:37:25 +00008268#endif
8269#ifdef _PC_PATH_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008270 {"PC_PATH_MAX", _PC_PATH_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008271#endif
8272#ifdef _PC_PIPE_BUF
Victor Stinner8c62be82010-05-06 00:08:46 +00008273 {"PC_PIPE_BUF", _PC_PIPE_BUF},
Fred Drakec9680921999-12-13 16:37:25 +00008274#endif
8275#ifdef _PC_PRIO_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008276 {"PC_PRIO_IO", _PC_PRIO_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008277#endif
8278#ifdef _PC_SOCK_MAXBUF
Victor Stinner8c62be82010-05-06 00:08:46 +00008279 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
Fred Drakec9680921999-12-13 16:37:25 +00008280#endif
8281#ifdef _PC_SYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008282 {"PC_SYNC_IO", _PC_SYNC_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008283#endif
8284#ifdef _PC_VDISABLE
Victor Stinner8c62be82010-05-06 00:08:46 +00008285 {"PC_VDISABLE", _PC_VDISABLE},
Fred Drakec9680921999-12-13 16:37:25 +00008286#endif
Jesus Cea7e9065c2010-10-25 13:02:04 +00008287#ifdef _PC_ACL_ENABLED
8288 {"PC_ACL_ENABLED", _PC_ACL_ENABLED},
8289#endif
8290#ifdef _PC_MIN_HOLE_SIZE
8291 {"PC_MIN_HOLE_SIZE", _PC_MIN_HOLE_SIZE},
8292#endif
8293#ifdef _PC_ALLOC_SIZE_MIN
8294 {"PC_ALLOC_SIZE_MIN", _PC_ALLOC_SIZE_MIN},
8295#endif
8296#ifdef _PC_REC_INCR_XFER_SIZE
8297 {"PC_REC_INCR_XFER_SIZE", _PC_REC_INCR_XFER_SIZE},
8298#endif
8299#ifdef _PC_REC_MAX_XFER_SIZE
8300 {"PC_REC_MAX_XFER_SIZE", _PC_REC_MAX_XFER_SIZE},
8301#endif
8302#ifdef _PC_REC_MIN_XFER_SIZE
8303 {"PC_REC_MIN_XFER_SIZE", _PC_REC_MIN_XFER_SIZE},
8304#endif
8305#ifdef _PC_REC_XFER_ALIGN
8306 {"PC_REC_XFER_ALIGN", _PC_REC_XFER_ALIGN},
8307#endif
8308#ifdef _PC_SYMLINK_MAX
8309 {"PC_SYMLINK_MAX", _PC_SYMLINK_MAX},
8310#endif
8311#ifdef _PC_XATTR_ENABLED
8312 {"PC_XATTR_ENABLED", _PC_XATTR_ENABLED},
8313#endif
8314#ifdef _PC_XATTR_EXISTS
8315 {"PC_XATTR_EXISTS", _PC_XATTR_EXISTS},
8316#endif
8317#ifdef _PC_TIMESTAMP_RESOLUTION
8318 {"PC_TIMESTAMP_RESOLUTION", _PC_TIMESTAMP_RESOLUTION},
8319#endif
Fred Drakec9680921999-12-13 16:37:25 +00008320};
8321
Fred Drakec9680921999-12-13 16:37:25 +00008322static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008323conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00008324{
8325 return conv_confname(arg, valuep, posix_constants_pathconf,
8326 sizeof(posix_constants_pathconf)
8327 / sizeof(struct constdef));
8328}
8329#endif
8330
8331#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008332PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008333"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00008334Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008335If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00008336
8337static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008338posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008339{
8340 PyObject *result = NULL;
8341 int name, fd;
8342
Fred Drake12c6e2d1999-12-14 21:25:03 +00008343 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
8344 conv_path_confname, &name)) {
Stefan Krah0e803b32010-11-26 16:16:47 +00008345 long limit;
Fred Drakec9680921999-12-13 16:37:25 +00008346
Stefan Krah0e803b32010-11-26 16:16:47 +00008347 errno = 0;
8348 limit = fpathconf(fd, name);
8349 if (limit == -1 && errno != 0)
8350 posix_error();
8351 else
8352 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00008353 }
8354 return result;
8355}
8356#endif
8357
8358
8359#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008360PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008361"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00008362Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008363If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00008364
8365static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008366posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008367{
8368 PyObject *result = NULL;
8369 int name;
8370 char *path;
8371
8372 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
8373 conv_path_confname, &name)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00008374 long limit;
Fred Drakec9680921999-12-13 16:37:25 +00008375
Victor Stinner8c62be82010-05-06 00:08:46 +00008376 errno = 0;
8377 limit = pathconf(path, name);
8378 if (limit == -1 && errno != 0) {
8379 if (errno == EINVAL)
Stefan Krah99439262010-11-26 12:58:05 +00008380 /* could be a path or name problem */
8381 posix_error();
Fred Drakec9680921999-12-13 16:37:25 +00008382 else
Stefan Krah99439262010-11-26 12:58:05 +00008383 posix_error_with_filename(path);
Victor Stinner8c62be82010-05-06 00:08:46 +00008384 }
8385 else
8386 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00008387 }
8388 return result;
8389}
8390#endif
8391
8392#ifdef HAVE_CONFSTR
8393static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00008394#ifdef _CS_ARCHITECTURE
Victor Stinner8c62be82010-05-06 00:08:46 +00008395 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
Fred Draked86ed291999-12-15 15:34:33 +00008396#endif
Mark Dickinson876d7c82010-04-16 12:47:52 +00008397#ifdef _CS_GNU_LIBC_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008398 {"CS_GNU_LIBC_VERSION", _CS_GNU_LIBC_VERSION},
Mark Dickinson876d7c82010-04-16 12:47:52 +00008399#endif
8400#ifdef _CS_GNU_LIBPTHREAD_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008401 {"CS_GNU_LIBPTHREAD_VERSION", _CS_GNU_LIBPTHREAD_VERSION},
Mark Dickinson876d7c82010-04-16 12:47:52 +00008402#endif
Fred Draked86ed291999-12-15 15:34:33 +00008403#ifdef _CS_HOSTNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008404 {"CS_HOSTNAME", _CS_HOSTNAME},
Fred Draked86ed291999-12-15 15:34:33 +00008405#endif
8406#ifdef _CS_HW_PROVIDER
Victor Stinner8c62be82010-05-06 00:08:46 +00008407 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
Fred Draked86ed291999-12-15 15:34:33 +00008408#endif
8409#ifdef _CS_HW_SERIAL
Victor Stinner8c62be82010-05-06 00:08:46 +00008410 {"CS_HW_SERIAL", _CS_HW_SERIAL},
Fred Draked86ed291999-12-15 15:34:33 +00008411#endif
8412#ifdef _CS_INITTAB_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008413 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00008414#endif
Fred Drakec9680921999-12-13 16:37:25 +00008415#ifdef _CS_LFS64_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008416 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008417#endif
8418#ifdef _CS_LFS64_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008419 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008420#endif
8421#ifdef _CS_LFS64_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008422 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008423#endif
8424#ifdef _CS_LFS64_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008425 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008426#endif
8427#ifdef _CS_LFS_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008428 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008429#endif
8430#ifdef _CS_LFS_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008431 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008432#endif
8433#ifdef _CS_LFS_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008434 {"CS_LFS_LIBS", _CS_LFS_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008435#endif
8436#ifdef _CS_LFS_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008437 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008438#endif
Fred Draked86ed291999-12-15 15:34:33 +00008439#ifdef _CS_MACHINE
Victor Stinner8c62be82010-05-06 00:08:46 +00008440 {"CS_MACHINE", _CS_MACHINE},
Fred Draked86ed291999-12-15 15:34:33 +00008441#endif
Fred Drakec9680921999-12-13 16:37:25 +00008442#ifdef _CS_PATH
Victor Stinner8c62be82010-05-06 00:08:46 +00008443 {"CS_PATH", _CS_PATH},
Fred Drakec9680921999-12-13 16:37:25 +00008444#endif
Fred Draked86ed291999-12-15 15:34:33 +00008445#ifdef _CS_RELEASE
Victor Stinner8c62be82010-05-06 00:08:46 +00008446 {"CS_RELEASE", _CS_RELEASE},
Fred Draked86ed291999-12-15 15:34:33 +00008447#endif
8448#ifdef _CS_SRPC_DOMAIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008449 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
Fred Draked86ed291999-12-15 15:34:33 +00008450#endif
8451#ifdef _CS_SYSNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008452 {"CS_SYSNAME", _CS_SYSNAME},
Fred Draked86ed291999-12-15 15:34:33 +00008453#endif
8454#ifdef _CS_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008455 {"CS_VERSION", _CS_VERSION},
Fred Draked86ed291999-12-15 15:34:33 +00008456#endif
Fred Drakec9680921999-12-13 16:37:25 +00008457#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008458 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008459#endif
8460#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008461 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008462#endif
8463#ifdef _CS_XBS5_ILP32_OFF32_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008464 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008465#endif
8466#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008467 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008468#endif
8469#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008470 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008471#endif
8472#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008473 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008474#endif
8475#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008476 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008477#endif
8478#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008479 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008480#endif
8481#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008482 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008483#endif
8484#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008485 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008486#endif
8487#ifdef _CS_XBS5_LP64_OFF64_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008488 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008489#endif
8490#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008491 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008492#endif
8493#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008494 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008495#endif
8496#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008497 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008498#endif
8499#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008500 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008501#endif
8502#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008503 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008504#endif
Fred Draked86ed291999-12-15 15:34:33 +00008505#ifdef _MIPS_CS_AVAIL_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00008506 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00008507#endif
8508#ifdef _MIPS_CS_BASE
Victor Stinner8c62be82010-05-06 00:08:46 +00008509 {"MIPS_CS_BASE", _MIPS_CS_BASE},
Fred Draked86ed291999-12-15 15:34:33 +00008510#endif
8511#ifdef _MIPS_CS_HOSTID
Victor Stinner8c62be82010-05-06 00:08:46 +00008512 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
Fred Draked86ed291999-12-15 15:34:33 +00008513#endif
8514#ifdef _MIPS_CS_HW_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008515 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00008516#endif
8517#ifdef _MIPS_CS_NUM_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00008518 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00008519#endif
8520#ifdef _MIPS_CS_OSREL_MAJ
Victor Stinner8c62be82010-05-06 00:08:46 +00008521 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
Fred Draked86ed291999-12-15 15:34:33 +00008522#endif
8523#ifdef _MIPS_CS_OSREL_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008524 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
Fred Draked86ed291999-12-15 15:34:33 +00008525#endif
8526#ifdef _MIPS_CS_OSREL_PATCH
Victor Stinner8c62be82010-05-06 00:08:46 +00008527 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
Fred Draked86ed291999-12-15 15:34:33 +00008528#endif
8529#ifdef _MIPS_CS_OS_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008530 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00008531#endif
8532#ifdef _MIPS_CS_OS_PROVIDER
Victor Stinner8c62be82010-05-06 00:08:46 +00008533 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
Fred Draked86ed291999-12-15 15:34:33 +00008534#endif
8535#ifdef _MIPS_CS_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00008536 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00008537#endif
8538#ifdef _MIPS_CS_SERIAL
Victor Stinner8c62be82010-05-06 00:08:46 +00008539 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
Fred Draked86ed291999-12-15 15:34:33 +00008540#endif
8541#ifdef _MIPS_CS_VENDOR
Victor Stinner8c62be82010-05-06 00:08:46 +00008542 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
Fred Draked86ed291999-12-15 15:34:33 +00008543#endif
Fred Drakec9680921999-12-13 16:37:25 +00008544};
8545
8546static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008547conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00008548{
8549 return conv_confname(arg, valuep, posix_constants_confstr,
8550 sizeof(posix_constants_confstr)
8551 / sizeof(struct constdef));
8552}
8553
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008554PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008555"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008556Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00008557
8558static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008559posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008560{
8561 PyObject *result = NULL;
8562 int name;
Victor Stinnercb043522010-09-10 23:49:04 +00008563 char buffer[255];
Stefan Krah99439262010-11-26 12:58:05 +00008564 int len;
Fred Drakec9680921999-12-13 16:37:25 +00008565
Victor Stinnercb043522010-09-10 23:49:04 +00008566 if (!PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name))
8567 return NULL;
8568
8569 errno = 0;
8570 len = confstr(name, buffer, sizeof(buffer));
8571 if (len == 0) {
8572 if (errno) {
8573 posix_error();
8574 return NULL;
Fred Drakec9680921999-12-13 16:37:25 +00008575 }
8576 else {
Victor Stinnercb043522010-09-10 23:49:04 +00008577 Py_RETURN_NONE;
Fred Drakec9680921999-12-13 16:37:25 +00008578 }
8579 }
Victor Stinnercb043522010-09-10 23:49:04 +00008580
8581 if ((unsigned int)len >= sizeof(buffer)) {
8582 char *buf = PyMem_Malloc(len);
8583 if (buf == NULL)
8584 return PyErr_NoMemory();
8585 confstr(name, buf, len);
8586 result = PyUnicode_DecodeFSDefaultAndSize(buf, len-1);
8587 PyMem_Free(buf);
8588 }
8589 else
8590 result = PyUnicode_DecodeFSDefaultAndSize(buffer, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00008591 return result;
8592}
8593#endif
8594
8595
8596#ifdef HAVE_SYSCONF
8597static struct constdef posix_constants_sysconf[] = {
8598#ifdef _SC_2_CHAR_TERM
Victor Stinner8c62be82010-05-06 00:08:46 +00008599 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
Fred Drakec9680921999-12-13 16:37:25 +00008600#endif
8601#ifdef _SC_2_C_BIND
Victor Stinner8c62be82010-05-06 00:08:46 +00008602 {"SC_2_C_BIND", _SC_2_C_BIND},
Fred Drakec9680921999-12-13 16:37:25 +00008603#endif
8604#ifdef _SC_2_C_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00008605 {"SC_2_C_DEV", _SC_2_C_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00008606#endif
8607#ifdef _SC_2_C_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008608 {"SC_2_C_VERSION", _SC_2_C_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00008609#endif
8610#ifdef _SC_2_FORT_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00008611 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00008612#endif
8613#ifdef _SC_2_FORT_RUN
Victor Stinner8c62be82010-05-06 00:08:46 +00008614 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
Fred Drakec9680921999-12-13 16:37:25 +00008615#endif
8616#ifdef _SC_2_LOCALEDEF
Victor Stinner8c62be82010-05-06 00:08:46 +00008617 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
Fred Drakec9680921999-12-13 16:37:25 +00008618#endif
8619#ifdef _SC_2_SW_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00008620 {"SC_2_SW_DEV", _SC_2_SW_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00008621#endif
8622#ifdef _SC_2_UPE
Victor Stinner8c62be82010-05-06 00:08:46 +00008623 {"SC_2_UPE", _SC_2_UPE},
Fred Drakec9680921999-12-13 16:37:25 +00008624#endif
8625#ifdef _SC_2_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008626 {"SC_2_VERSION", _SC_2_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00008627#endif
Fred Draked86ed291999-12-15 15:34:33 +00008628#ifdef _SC_ABI_ASYNCHRONOUS_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008629 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
Fred Draked86ed291999-12-15 15:34:33 +00008630#endif
8631#ifdef _SC_ACL
Victor Stinner8c62be82010-05-06 00:08:46 +00008632 {"SC_ACL", _SC_ACL},
Fred Draked86ed291999-12-15 15:34:33 +00008633#endif
Fred Drakec9680921999-12-13 16:37:25 +00008634#ifdef _SC_AIO_LISTIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008635 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008636#endif
Fred Drakec9680921999-12-13 16:37:25 +00008637#ifdef _SC_AIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008638 {"SC_AIO_MAX", _SC_AIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008639#endif
8640#ifdef _SC_AIO_PRIO_DELTA_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008641 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008642#endif
8643#ifdef _SC_ARG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008644 {"SC_ARG_MAX", _SC_ARG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008645#endif
8646#ifdef _SC_ASYNCHRONOUS_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008647 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008648#endif
8649#ifdef _SC_ATEXIT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008650 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008651#endif
Fred Draked86ed291999-12-15 15:34:33 +00008652#ifdef _SC_AUDIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008653 {"SC_AUDIT", _SC_AUDIT},
Fred Draked86ed291999-12-15 15:34:33 +00008654#endif
Fred Drakec9680921999-12-13 16:37:25 +00008655#ifdef _SC_AVPHYS_PAGES
Victor Stinner8c62be82010-05-06 00:08:46 +00008656 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
Fred Drakec9680921999-12-13 16:37:25 +00008657#endif
8658#ifdef _SC_BC_BASE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008659 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008660#endif
8661#ifdef _SC_BC_DIM_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008662 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008663#endif
8664#ifdef _SC_BC_SCALE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008665 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008666#endif
8667#ifdef _SC_BC_STRING_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008668 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008669#endif
Fred Draked86ed291999-12-15 15:34:33 +00008670#ifdef _SC_CAP
Victor Stinner8c62be82010-05-06 00:08:46 +00008671 {"SC_CAP", _SC_CAP},
Fred Draked86ed291999-12-15 15:34:33 +00008672#endif
Fred Drakec9680921999-12-13 16:37:25 +00008673#ifdef _SC_CHARCLASS_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008674 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008675#endif
8676#ifdef _SC_CHAR_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008677 {"SC_CHAR_BIT", _SC_CHAR_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00008678#endif
8679#ifdef _SC_CHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008680 {"SC_CHAR_MAX", _SC_CHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008681#endif
8682#ifdef _SC_CHAR_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008683 {"SC_CHAR_MIN", _SC_CHAR_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008684#endif
8685#ifdef _SC_CHILD_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008686 {"SC_CHILD_MAX", _SC_CHILD_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008687#endif
8688#ifdef _SC_CLK_TCK
Victor Stinner8c62be82010-05-06 00:08:46 +00008689 {"SC_CLK_TCK", _SC_CLK_TCK},
Fred Drakec9680921999-12-13 16:37:25 +00008690#endif
8691#ifdef _SC_COHER_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008692 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008693#endif
8694#ifdef _SC_COLL_WEIGHTS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008695 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008696#endif
8697#ifdef _SC_DCACHE_ASSOC
Victor Stinner8c62be82010-05-06 00:08:46 +00008698 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
Fred Drakec9680921999-12-13 16:37:25 +00008699#endif
8700#ifdef _SC_DCACHE_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008701 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008702#endif
8703#ifdef _SC_DCACHE_LINESZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008704 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
Fred Drakec9680921999-12-13 16:37:25 +00008705#endif
8706#ifdef _SC_DCACHE_SZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008707 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
Fred Drakec9680921999-12-13 16:37:25 +00008708#endif
8709#ifdef _SC_DCACHE_TBLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008710 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008711#endif
8712#ifdef _SC_DELAYTIMER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008713 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008714#endif
8715#ifdef _SC_EQUIV_CLASS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008716 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008717#endif
8718#ifdef _SC_EXPR_NEST_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008719 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008720#endif
8721#ifdef _SC_FSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00008722 {"SC_FSYNC", _SC_FSYNC},
Fred Drakec9680921999-12-13 16:37:25 +00008723#endif
8724#ifdef _SC_GETGR_R_SIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008725 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008726#endif
8727#ifdef _SC_GETPW_R_SIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008728 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008729#endif
8730#ifdef _SC_ICACHE_ASSOC
Victor Stinner8c62be82010-05-06 00:08:46 +00008731 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
Fred Drakec9680921999-12-13 16:37:25 +00008732#endif
8733#ifdef _SC_ICACHE_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008734 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008735#endif
8736#ifdef _SC_ICACHE_LINESZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008737 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
Fred Drakec9680921999-12-13 16:37:25 +00008738#endif
8739#ifdef _SC_ICACHE_SZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008740 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
Fred Drakec9680921999-12-13 16:37:25 +00008741#endif
Fred Draked86ed291999-12-15 15:34:33 +00008742#ifdef _SC_INF
Victor Stinner8c62be82010-05-06 00:08:46 +00008743 {"SC_INF", _SC_INF},
Fred Draked86ed291999-12-15 15:34:33 +00008744#endif
Fred Drakec9680921999-12-13 16:37:25 +00008745#ifdef _SC_INT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008746 {"SC_INT_MAX", _SC_INT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008747#endif
8748#ifdef _SC_INT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008749 {"SC_INT_MIN", _SC_INT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008750#endif
8751#ifdef _SC_IOV_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008752 {"SC_IOV_MAX", _SC_IOV_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008753#endif
Fred Draked86ed291999-12-15 15:34:33 +00008754#ifdef _SC_IP_SECOPTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008755 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
Fred Draked86ed291999-12-15 15:34:33 +00008756#endif
Fred Drakec9680921999-12-13 16:37:25 +00008757#ifdef _SC_JOB_CONTROL
Victor Stinner8c62be82010-05-06 00:08:46 +00008758 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
Fred Drakec9680921999-12-13 16:37:25 +00008759#endif
Fred Draked86ed291999-12-15 15:34:33 +00008760#ifdef _SC_KERN_POINTERS
Victor Stinner8c62be82010-05-06 00:08:46 +00008761 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
Fred Draked86ed291999-12-15 15:34:33 +00008762#endif
8763#ifdef _SC_KERN_SIM
Victor Stinner8c62be82010-05-06 00:08:46 +00008764 {"SC_KERN_SIM", _SC_KERN_SIM},
Fred Draked86ed291999-12-15 15:34:33 +00008765#endif
Fred Drakec9680921999-12-13 16:37:25 +00008766#ifdef _SC_LINE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008767 {"SC_LINE_MAX", _SC_LINE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008768#endif
8769#ifdef _SC_LOGIN_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008770 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008771#endif
8772#ifdef _SC_LOGNAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008773 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008774#endif
8775#ifdef _SC_LONG_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008776 {"SC_LONG_BIT", _SC_LONG_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00008777#endif
Fred Draked86ed291999-12-15 15:34:33 +00008778#ifdef _SC_MAC
Victor Stinner8c62be82010-05-06 00:08:46 +00008779 {"SC_MAC", _SC_MAC},
Fred Draked86ed291999-12-15 15:34:33 +00008780#endif
Fred Drakec9680921999-12-13 16:37:25 +00008781#ifdef _SC_MAPPED_FILES
Victor Stinner8c62be82010-05-06 00:08:46 +00008782 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
Fred Drakec9680921999-12-13 16:37:25 +00008783#endif
8784#ifdef _SC_MAXPID
Victor Stinner8c62be82010-05-06 00:08:46 +00008785 {"SC_MAXPID", _SC_MAXPID},
Fred Drakec9680921999-12-13 16:37:25 +00008786#endif
8787#ifdef _SC_MB_LEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008788 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008789#endif
8790#ifdef _SC_MEMLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +00008791 {"SC_MEMLOCK", _SC_MEMLOCK},
Fred Drakec9680921999-12-13 16:37:25 +00008792#endif
8793#ifdef _SC_MEMLOCK_RANGE
Victor Stinner8c62be82010-05-06 00:08:46 +00008794 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
Fred Drakec9680921999-12-13 16:37:25 +00008795#endif
8796#ifdef _SC_MEMORY_PROTECTION
Victor Stinner8c62be82010-05-06 00:08:46 +00008797 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
Fred Drakec9680921999-12-13 16:37:25 +00008798#endif
8799#ifdef _SC_MESSAGE_PASSING
Victor Stinner8c62be82010-05-06 00:08:46 +00008800 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
Fred Drakec9680921999-12-13 16:37:25 +00008801#endif
Fred Draked86ed291999-12-15 15:34:33 +00008802#ifdef _SC_MMAP_FIXED_ALIGNMENT
Victor Stinner8c62be82010-05-06 00:08:46 +00008803 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
Fred Draked86ed291999-12-15 15:34:33 +00008804#endif
Fred Drakec9680921999-12-13 16:37:25 +00008805#ifdef _SC_MQ_OPEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008806 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008807#endif
8808#ifdef _SC_MQ_PRIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008809 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008810#endif
Fred Draked86ed291999-12-15 15:34:33 +00008811#ifdef _SC_NACLS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008812 {"SC_NACLS_MAX", _SC_NACLS_MAX},
Fred Draked86ed291999-12-15 15:34:33 +00008813#endif
Fred Drakec9680921999-12-13 16:37:25 +00008814#ifdef _SC_NGROUPS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008815 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008816#endif
8817#ifdef _SC_NL_ARGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008818 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008819#endif
8820#ifdef _SC_NL_LANGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008821 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008822#endif
8823#ifdef _SC_NL_MSGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008824 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008825#endif
8826#ifdef _SC_NL_NMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008827 {"SC_NL_NMAX", _SC_NL_NMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008828#endif
8829#ifdef _SC_NL_SETMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008830 {"SC_NL_SETMAX", _SC_NL_SETMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008831#endif
8832#ifdef _SC_NL_TEXTMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008833 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008834#endif
8835#ifdef _SC_NPROCESSORS_CONF
Victor Stinner8c62be82010-05-06 00:08:46 +00008836 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
Fred Drakec9680921999-12-13 16:37:25 +00008837#endif
8838#ifdef _SC_NPROCESSORS_ONLN
Victor Stinner8c62be82010-05-06 00:08:46 +00008839 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
Fred Drakec9680921999-12-13 16:37:25 +00008840#endif
Fred Draked86ed291999-12-15 15:34:33 +00008841#ifdef _SC_NPROC_CONF
Victor Stinner8c62be82010-05-06 00:08:46 +00008842 {"SC_NPROC_CONF", _SC_NPROC_CONF},
Fred Draked86ed291999-12-15 15:34:33 +00008843#endif
8844#ifdef _SC_NPROC_ONLN
Victor Stinner8c62be82010-05-06 00:08:46 +00008845 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
Fred Draked86ed291999-12-15 15:34:33 +00008846#endif
Fred Drakec9680921999-12-13 16:37:25 +00008847#ifdef _SC_NZERO
Victor Stinner8c62be82010-05-06 00:08:46 +00008848 {"SC_NZERO", _SC_NZERO},
Fred Drakec9680921999-12-13 16:37:25 +00008849#endif
8850#ifdef _SC_OPEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008851 {"SC_OPEN_MAX", _SC_OPEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008852#endif
8853#ifdef _SC_PAGESIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008854 {"SC_PAGESIZE", _SC_PAGESIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008855#endif
8856#ifdef _SC_PAGE_SIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008857 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008858#endif
8859#ifdef _SC_PASS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008860 {"SC_PASS_MAX", _SC_PASS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008861#endif
8862#ifdef _SC_PHYS_PAGES
Victor Stinner8c62be82010-05-06 00:08:46 +00008863 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
Fred Drakec9680921999-12-13 16:37:25 +00008864#endif
8865#ifdef _SC_PII
Victor Stinner8c62be82010-05-06 00:08:46 +00008866 {"SC_PII", _SC_PII},
Fred Drakec9680921999-12-13 16:37:25 +00008867#endif
8868#ifdef _SC_PII_INTERNET
Victor Stinner8c62be82010-05-06 00:08:46 +00008869 {"SC_PII_INTERNET", _SC_PII_INTERNET},
Fred Drakec9680921999-12-13 16:37:25 +00008870#endif
8871#ifdef _SC_PII_INTERNET_DGRAM
Victor Stinner8c62be82010-05-06 00:08:46 +00008872 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
Fred Drakec9680921999-12-13 16:37:25 +00008873#endif
8874#ifdef _SC_PII_INTERNET_STREAM
Victor Stinner8c62be82010-05-06 00:08:46 +00008875 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
Fred Drakec9680921999-12-13 16:37:25 +00008876#endif
8877#ifdef _SC_PII_OSI
Victor Stinner8c62be82010-05-06 00:08:46 +00008878 {"SC_PII_OSI", _SC_PII_OSI},
Fred Drakec9680921999-12-13 16:37:25 +00008879#endif
8880#ifdef _SC_PII_OSI_CLTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008881 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
Fred Drakec9680921999-12-13 16:37:25 +00008882#endif
8883#ifdef _SC_PII_OSI_COTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008884 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
Fred Drakec9680921999-12-13 16:37:25 +00008885#endif
8886#ifdef _SC_PII_OSI_M
Victor Stinner8c62be82010-05-06 00:08:46 +00008887 {"SC_PII_OSI_M", _SC_PII_OSI_M},
Fred Drakec9680921999-12-13 16:37:25 +00008888#endif
8889#ifdef _SC_PII_SOCKET
Victor Stinner8c62be82010-05-06 00:08:46 +00008890 {"SC_PII_SOCKET", _SC_PII_SOCKET},
Fred Drakec9680921999-12-13 16:37:25 +00008891#endif
8892#ifdef _SC_PII_XTI
Victor Stinner8c62be82010-05-06 00:08:46 +00008893 {"SC_PII_XTI", _SC_PII_XTI},
Fred Drakec9680921999-12-13 16:37:25 +00008894#endif
8895#ifdef _SC_POLL
Victor Stinner8c62be82010-05-06 00:08:46 +00008896 {"SC_POLL", _SC_POLL},
Fred Drakec9680921999-12-13 16:37:25 +00008897#endif
8898#ifdef _SC_PRIORITIZED_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008899 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008900#endif
8901#ifdef _SC_PRIORITY_SCHEDULING
Victor Stinner8c62be82010-05-06 00:08:46 +00008902 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
Fred Drakec9680921999-12-13 16:37:25 +00008903#endif
8904#ifdef _SC_REALTIME_SIGNALS
Victor Stinner8c62be82010-05-06 00:08:46 +00008905 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
Fred Drakec9680921999-12-13 16:37:25 +00008906#endif
8907#ifdef _SC_RE_DUP_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008908 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008909#endif
8910#ifdef _SC_RTSIG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008911 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008912#endif
8913#ifdef _SC_SAVED_IDS
Victor Stinner8c62be82010-05-06 00:08:46 +00008914 {"SC_SAVED_IDS", _SC_SAVED_IDS},
Fred Drakec9680921999-12-13 16:37:25 +00008915#endif
8916#ifdef _SC_SCHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008917 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008918#endif
8919#ifdef _SC_SCHAR_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008920 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008921#endif
8922#ifdef _SC_SELECT
Victor Stinner8c62be82010-05-06 00:08:46 +00008923 {"SC_SELECT", _SC_SELECT},
Fred Drakec9680921999-12-13 16:37:25 +00008924#endif
8925#ifdef _SC_SEMAPHORES
Victor Stinner8c62be82010-05-06 00:08:46 +00008926 {"SC_SEMAPHORES", _SC_SEMAPHORES},
Fred Drakec9680921999-12-13 16:37:25 +00008927#endif
8928#ifdef _SC_SEM_NSEMS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008929 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008930#endif
8931#ifdef _SC_SEM_VALUE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008932 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008933#endif
8934#ifdef _SC_SHARED_MEMORY_OBJECTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008935 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
Fred Drakec9680921999-12-13 16:37:25 +00008936#endif
8937#ifdef _SC_SHRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008938 {"SC_SHRT_MAX", _SC_SHRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008939#endif
8940#ifdef _SC_SHRT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008941 {"SC_SHRT_MIN", _SC_SHRT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008942#endif
8943#ifdef _SC_SIGQUEUE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008944 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008945#endif
8946#ifdef _SC_SIGRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008947 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008948#endif
8949#ifdef _SC_SIGRT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008950 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008951#endif
Fred Draked86ed291999-12-15 15:34:33 +00008952#ifdef _SC_SOFTPOWER
Victor Stinner8c62be82010-05-06 00:08:46 +00008953 {"SC_SOFTPOWER", _SC_SOFTPOWER},
Fred Draked86ed291999-12-15 15:34:33 +00008954#endif
Fred Drakec9680921999-12-13 16:37:25 +00008955#ifdef _SC_SPLIT_CACHE
Victor Stinner8c62be82010-05-06 00:08:46 +00008956 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
Fred Drakec9680921999-12-13 16:37:25 +00008957#endif
8958#ifdef _SC_SSIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008959 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008960#endif
8961#ifdef _SC_STACK_PROT
Victor Stinner8c62be82010-05-06 00:08:46 +00008962 {"SC_STACK_PROT", _SC_STACK_PROT},
Fred Drakec9680921999-12-13 16:37:25 +00008963#endif
8964#ifdef _SC_STREAM_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008965 {"SC_STREAM_MAX", _SC_STREAM_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008966#endif
8967#ifdef _SC_SYNCHRONIZED_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008968 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008969#endif
8970#ifdef _SC_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00008971 {"SC_THREADS", _SC_THREADS},
Fred Drakec9680921999-12-13 16:37:25 +00008972#endif
8973#ifdef _SC_THREAD_ATTR_STACKADDR
Victor Stinner8c62be82010-05-06 00:08:46 +00008974 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
Fred Drakec9680921999-12-13 16:37:25 +00008975#endif
8976#ifdef _SC_THREAD_ATTR_STACKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008977 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008978#endif
8979#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
Victor Stinner8c62be82010-05-06 00:08:46 +00008980 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
Fred Drakec9680921999-12-13 16:37:25 +00008981#endif
8982#ifdef _SC_THREAD_KEYS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008983 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008984#endif
8985#ifdef _SC_THREAD_PRIORITY_SCHEDULING
Victor Stinner8c62be82010-05-06 00:08:46 +00008986 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
Fred Drakec9680921999-12-13 16:37:25 +00008987#endif
8988#ifdef _SC_THREAD_PRIO_INHERIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008989 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
Fred Drakec9680921999-12-13 16:37:25 +00008990#endif
8991#ifdef _SC_THREAD_PRIO_PROTECT
Victor Stinner8c62be82010-05-06 00:08:46 +00008992 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
Fred Drakec9680921999-12-13 16:37:25 +00008993#endif
8994#ifdef _SC_THREAD_PROCESS_SHARED
Victor Stinner8c62be82010-05-06 00:08:46 +00008995 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
Fred Drakec9680921999-12-13 16:37:25 +00008996#endif
8997#ifdef _SC_THREAD_SAFE_FUNCTIONS
Victor Stinner8c62be82010-05-06 00:08:46 +00008998 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
Fred Drakec9680921999-12-13 16:37:25 +00008999#endif
9000#ifdef _SC_THREAD_STACK_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00009001 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00009002#endif
9003#ifdef _SC_THREAD_THREADS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009004 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009005#endif
9006#ifdef _SC_TIMERS
Victor Stinner8c62be82010-05-06 00:08:46 +00009007 {"SC_TIMERS", _SC_TIMERS},
Fred Drakec9680921999-12-13 16:37:25 +00009008#endif
9009#ifdef _SC_TIMER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009010 {"SC_TIMER_MAX", _SC_TIMER_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009011#endif
9012#ifdef _SC_TTY_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009013 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009014#endif
9015#ifdef _SC_TZNAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009016 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009017#endif
9018#ifdef _SC_T_IOV_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009019 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009020#endif
9021#ifdef _SC_UCHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009022 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009023#endif
9024#ifdef _SC_UINT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009025 {"SC_UINT_MAX", _SC_UINT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009026#endif
9027#ifdef _SC_UIO_MAXIOV
Victor Stinner8c62be82010-05-06 00:08:46 +00009028 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
Fred Drakec9680921999-12-13 16:37:25 +00009029#endif
9030#ifdef _SC_ULONG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009031 {"SC_ULONG_MAX", _SC_ULONG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009032#endif
9033#ifdef _SC_USHRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009034 {"SC_USHRT_MAX", _SC_USHRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009035#endif
9036#ifdef _SC_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00009037 {"SC_VERSION", _SC_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00009038#endif
9039#ifdef _SC_WORD_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00009040 {"SC_WORD_BIT", _SC_WORD_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00009041#endif
9042#ifdef _SC_XBS5_ILP32_OFF32
Victor Stinner8c62be82010-05-06 00:08:46 +00009043 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
Fred Drakec9680921999-12-13 16:37:25 +00009044#endif
9045#ifdef _SC_XBS5_ILP32_OFFBIG
Victor Stinner8c62be82010-05-06 00:08:46 +00009046 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
Fred Drakec9680921999-12-13 16:37:25 +00009047#endif
9048#ifdef _SC_XBS5_LP64_OFF64
Victor Stinner8c62be82010-05-06 00:08:46 +00009049 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
Fred Drakec9680921999-12-13 16:37:25 +00009050#endif
9051#ifdef _SC_XBS5_LPBIG_OFFBIG
Victor Stinner8c62be82010-05-06 00:08:46 +00009052 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
Fred Drakec9680921999-12-13 16:37:25 +00009053#endif
9054#ifdef _SC_XOPEN_CRYPT
Victor Stinner8c62be82010-05-06 00:08:46 +00009055 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
Fred Drakec9680921999-12-13 16:37:25 +00009056#endif
9057#ifdef _SC_XOPEN_ENH_I18N
Victor Stinner8c62be82010-05-06 00:08:46 +00009058 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
Fred Drakec9680921999-12-13 16:37:25 +00009059#endif
9060#ifdef _SC_XOPEN_LEGACY
Victor Stinner8c62be82010-05-06 00:08:46 +00009061 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
Fred Drakec9680921999-12-13 16:37:25 +00009062#endif
9063#ifdef _SC_XOPEN_REALTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00009064 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
Fred Drakec9680921999-12-13 16:37:25 +00009065#endif
9066#ifdef _SC_XOPEN_REALTIME_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00009067 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
Fred Drakec9680921999-12-13 16:37:25 +00009068#endif
9069#ifdef _SC_XOPEN_SHM
Victor Stinner8c62be82010-05-06 00:08:46 +00009070 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
Fred Drakec9680921999-12-13 16:37:25 +00009071#endif
9072#ifdef _SC_XOPEN_UNIX
Victor Stinner8c62be82010-05-06 00:08:46 +00009073 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
Fred Drakec9680921999-12-13 16:37:25 +00009074#endif
9075#ifdef _SC_XOPEN_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00009076 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00009077#endif
9078#ifdef _SC_XOPEN_XCU_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00009079 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00009080#endif
9081#ifdef _SC_XOPEN_XPG2
Victor Stinner8c62be82010-05-06 00:08:46 +00009082 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
Fred Drakec9680921999-12-13 16:37:25 +00009083#endif
9084#ifdef _SC_XOPEN_XPG3
Victor Stinner8c62be82010-05-06 00:08:46 +00009085 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
Fred Drakec9680921999-12-13 16:37:25 +00009086#endif
9087#ifdef _SC_XOPEN_XPG4
Victor Stinner8c62be82010-05-06 00:08:46 +00009088 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
Fred Drakec9680921999-12-13 16:37:25 +00009089#endif
9090};
9091
9092static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009093conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00009094{
9095 return conv_confname(arg, valuep, posix_constants_sysconf,
9096 sizeof(posix_constants_sysconf)
9097 / sizeof(struct constdef));
9098}
9099
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009100PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00009101"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009102Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00009103
9104static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009105posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00009106{
9107 PyObject *result = NULL;
9108 int name;
9109
9110 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
9111 int value;
9112
9113 errno = 0;
9114 value = sysconf(name);
9115 if (value == -1 && errno != 0)
9116 posix_error();
9117 else
Christian Heimes217cfd12007-12-02 14:31:20 +00009118 result = PyLong_FromLong(value);
Fred Drakec9680921999-12-13 16:37:25 +00009119 }
9120 return result;
9121}
9122#endif
9123
9124
Fred Drakebec628d1999-12-15 18:31:10 +00009125/* This code is used to ensure that the tables of configuration value names
9126 * are in sorted order as required by conv_confname(), and also to build the
9127 * the exported dictionaries that are used to publish information about the
9128 * names available on the host platform.
9129 *
9130 * Sorting the table at runtime ensures that the table is properly ordered
9131 * when used, even for platforms we're not able to test on. It also makes
9132 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00009133 */
Fred Drakebec628d1999-12-15 18:31:10 +00009134
9135static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009136cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00009137{
9138 const struct constdef *c1 =
Victor Stinner8c62be82010-05-06 00:08:46 +00009139 (const struct constdef *) v1;
Fred Drakebec628d1999-12-15 18:31:10 +00009140 const struct constdef *c2 =
Victor Stinner8c62be82010-05-06 00:08:46 +00009141 (const struct constdef *) v2;
Fred Drakebec628d1999-12-15 18:31:10 +00009142
9143 return strcmp(c1->name, c2->name);
9144}
9145
9146static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009147setup_confname_table(struct constdef *table, size_t tablesize,
Victor Stinner8c62be82010-05-06 00:08:46 +00009148 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00009149{
Fred Drakebec628d1999-12-15 18:31:10 +00009150 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00009151 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00009152
9153 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
9154 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00009155 if (d == NULL)
Victor Stinner8c62be82010-05-06 00:08:46 +00009156 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009157
Barry Warsaw3155db32000-04-13 15:20:40 +00009158 for (i=0; i < tablesize; ++i) {
Victor Stinner8c62be82010-05-06 00:08:46 +00009159 PyObject *o = PyLong_FromLong(table[i].value);
9160 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
9161 Py_XDECREF(o);
9162 Py_DECREF(d);
9163 return -1;
9164 }
9165 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00009166 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00009167 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00009168}
9169
Fred Drakebec628d1999-12-15 18:31:10 +00009170/* Return -1 on failure, 0 on success. */
9171static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00009172setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00009173{
9174#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00009175 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00009176 sizeof(posix_constants_pathconf)
9177 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00009178 "pathconf_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00009179 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009180#endif
9181#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00009182 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00009183 sizeof(posix_constants_confstr)
9184 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00009185 "confstr_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00009186 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009187#endif
9188#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00009189 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00009190 sizeof(posix_constants_sysconf)
9191 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00009192 "sysconf_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00009193 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009194#endif
Fred Drakebec628d1999-12-15 18:31:10 +00009195 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00009196}
Fred Draked86ed291999-12-15 15:34:33 +00009197
9198
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009199PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00009200"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009201Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009202in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009203
9204static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00009205posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009206{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009207 abort();
9208 /*NOTREACHED*/
9209 Py_FatalError("abort() called from Python code didn't abort!");
9210 return NULL;
9211}
Fred Drakebec628d1999-12-15 18:31:10 +00009212
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00009213#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009214PyDoc_STRVAR(win32_startfile__doc__,
Georg Brandlf4f44152006-02-18 22:29:33 +00009215"startfile(filepath [, operation]) - Start a file with its associated\n\
9216application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00009217\n\
Georg Brandlf4f44152006-02-18 22:29:33 +00009218When \"operation\" is not specified or \"open\", this acts like\n\
9219double-clicking the file in Explorer, or giving the file name as an\n\
9220argument to the DOS \"start\" command: the file is opened with whatever\n\
9221application (if any) its extension is associated.\n\
9222When another \"operation\" is given, it specifies what should be done with\n\
9223the file. A typical operation is \"print\".\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00009224\n\
9225startfile returns as soon as the associated application is launched.\n\
9226There is no option to wait for the application to close, and no way\n\
9227to retrieve the application's exit status.\n\
9228\n\
9229The filepath is relative to the current directory. If you want to use\n\
9230an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009231the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00009232
9233static PyObject *
9234win32_startfile(PyObject *self, PyObject *args)
9235{
Victor Stinner8c62be82010-05-06 00:08:46 +00009236 PyObject *ofilepath;
9237 char *filepath;
9238 char *operation = NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02009239 wchar_t *wpath, *woperation;
Victor Stinner8c62be82010-05-06 00:08:46 +00009240 HINSTANCE rc;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00009241
Victor Stinnereb5657a2011-09-30 01:44:27 +02009242 PyObject *unipath, *uoperation = NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00009243 if (!PyArg_ParseTuple(args, "U|s:startfile",
9244 &unipath, &operation)) {
9245 PyErr_Clear();
9246 goto normal;
9247 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00009248
Victor Stinner8c62be82010-05-06 00:08:46 +00009249 if (operation) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02009250 uoperation = PyUnicode_DecodeASCII(operation,
Victor Stinner8c62be82010-05-06 00:08:46 +00009251 strlen(operation), NULL);
Victor Stinnereb5657a2011-09-30 01:44:27 +02009252 if (!uoperation) {
Victor Stinner8c62be82010-05-06 00:08:46 +00009253 PyErr_Clear();
9254 operation = NULL;
9255 goto normal;
9256 }
9257 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00009258
Victor Stinnereb5657a2011-09-30 01:44:27 +02009259 wpath = PyUnicode_AsUnicode(unipath);
9260 if (wpath == NULL)
9261 goto normal;
9262 if (uoperation) {
9263 woperation = PyUnicode_AsUnicode(uoperation);
9264 if (woperation == NULL)
9265 goto normal;
9266 }
9267 else
9268 woperation = NULL;
9269
Victor Stinner8c62be82010-05-06 00:08:46 +00009270 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02009271 rc = ShellExecuteW((HWND)0, woperation, wpath,
9272 NULL, NULL, SW_SHOWNORMAL);
Victor Stinner8c62be82010-05-06 00:08:46 +00009273 Py_END_ALLOW_THREADS
9274
Victor Stinnereb5657a2011-09-30 01:44:27 +02009275 Py_XDECREF(uoperation);
Victor Stinner8c62be82010-05-06 00:08:46 +00009276 if (rc <= (HINSTANCE)32) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02009277 win32_error_object("startfile", unipath);
9278 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00009279 }
9280 Py_INCREF(Py_None);
9281 return Py_None;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009282
9283normal:
Victor Stinner8c62be82010-05-06 00:08:46 +00009284 if (!PyArg_ParseTuple(args, "O&|s:startfile",
9285 PyUnicode_FSConverter, &ofilepath,
9286 &operation))
9287 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01009288 if (win32_warn_bytes_api()) {
9289 Py_DECREF(ofilepath);
9290 return NULL;
9291 }
Victor Stinner8c62be82010-05-06 00:08:46 +00009292 filepath = PyBytes_AsString(ofilepath);
9293 Py_BEGIN_ALLOW_THREADS
9294 rc = ShellExecute((HWND)0, operation, filepath,
9295 NULL, NULL, SW_SHOWNORMAL);
9296 Py_END_ALLOW_THREADS
9297 if (rc <= (HINSTANCE)32) {
9298 PyObject *errval = win32_error("startfile", filepath);
9299 Py_DECREF(ofilepath);
9300 return errval;
9301 }
9302 Py_DECREF(ofilepath);
9303 Py_INCREF(Py_None);
9304 return Py_None;
Tim Petersf58a7aa2000-09-22 10:05:54 +00009305}
9306#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009307
Martin v. Löwis438b5342002-12-27 10:16:42 +00009308#ifdef HAVE_GETLOADAVG
9309PyDoc_STRVAR(posix_getloadavg__doc__,
9310"getloadavg() -> (float, float, float)\n\n\
9311Return the number of processes in the system run queue averaged over\n\
9312the last 1, 5, and 15 minutes or raises OSError if the load average\n\
9313was unobtainable");
9314
9315static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00009316posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00009317{
9318 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00009319 if (getloadavg(loadavg, 3)!=3) {
Stefan Krah0e803b32010-11-26 16:16:47 +00009320 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
9321 return NULL;
Martin v. Löwis438b5342002-12-27 10:16:42 +00009322 } else
Stefan Krah0e803b32010-11-26 16:16:47 +00009323 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
Martin v. Löwis438b5342002-12-27 10:16:42 +00009324}
9325#endif
9326
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00009327PyDoc_STRVAR(device_encoding__doc__,
9328"device_encoding(fd) -> str\n\n\
9329Return a string describing the encoding of the device\n\
9330if the output is a terminal; else return None.");
9331
9332static PyObject *
9333device_encoding(PyObject *self, PyObject *args)
9334{
Victor Stinner8c62be82010-05-06 00:08:46 +00009335 int fd;
Brett Cannonefb00c02012-02-29 18:31:31 -05009336
Victor Stinner8c62be82010-05-06 00:08:46 +00009337 if (!PyArg_ParseTuple(args, "i:device_encoding", &fd))
9338 return NULL;
Brett Cannonefb00c02012-02-29 18:31:31 -05009339
9340 return _Py_device_encoding(fd);
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00009341}
9342
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009343#ifdef __VMS
9344/* Use openssl random routine */
9345#include <openssl/rand.h>
9346PyDoc_STRVAR(vms_urandom__doc__,
9347"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00009348Return n random bytes suitable for cryptographic use.");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009349
9350static PyObject*
9351vms_urandom(PyObject *self, PyObject *args)
9352{
Victor Stinner8c62be82010-05-06 00:08:46 +00009353 int howMany;
9354 PyObject* result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009355
Victor Stinner8c62be82010-05-06 00:08:46 +00009356 /* Read arguments */
9357 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
9358 return NULL;
9359 if (howMany < 0)
9360 return PyErr_Format(PyExc_ValueError,
9361 "negative argument not allowed");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009362
Victor Stinner8c62be82010-05-06 00:08:46 +00009363 /* Allocate bytes */
9364 result = PyBytes_FromStringAndSize(NULL, howMany);
9365 if (result != NULL) {
9366 /* Get random data */
9367 if (RAND_pseudo_bytes((unsigned char*)
9368 PyBytes_AS_STRING(result),
9369 howMany) < 0) {
9370 Py_DECREF(result);
9371 return PyErr_Format(PyExc_ValueError,
9372 "RAND_pseudo_bytes");
9373 }
9374 }
9375 return result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009376}
9377#endif
9378
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009379#ifdef HAVE_SETRESUID
9380PyDoc_STRVAR(posix_setresuid__doc__,
9381"setresuid(ruid, euid, suid)\n\n\
9382Set the current process's real, effective, and saved user ids.");
9383
9384static PyObject*
9385posix_setresuid (PyObject *self, PyObject *args)
9386{
Victor Stinner8c62be82010-05-06 00:08:46 +00009387 /* We assume uid_t is no larger than a long. */
9388 long ruid, euid, suid;
9389 if (!PyArg_ParseTuple(args, "lll", &ruid, &euid, &suid))
9390 return NULL;
9391 if (setresuid(ruid, euid, suid) < 0)
9392 return posix_error();
9393 Py_RETURN_NONE;
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009394}
9395#endif
9396
9397#ifdef HAVE_SETRESGID
9398PyDoc_STRVAR(posix_setresgid__doc__,
9399"setresgid(rgid, egid, sgid)\n\n\
9400Set the current process's real, effective, and saved group ids.");
9401
9402static PyObject*
9403posix_setresgid (PyObject *self, PyObject *args)
9404{
Victor Stinner8c62be82010-05-06 00:08:46 +00009405 /* We assume uid_t is no larger than a long. */
9406 long rgid, egid, sgid;
9407 if (!PyArg_ParseTuple(args, "lll", &rgid, &egid, &sgid))
9408 return NULL;
9409 if (setresgid(rgid, egid, sgid) < 0)
9410 return posix_error();
9411 Py_RETURN_NONE;
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009412}
9413#endif
9414
9415#ifdef HAVE_GETRESUID
9416PyDoc_STRVAR(posix_getresuid__doc__,
9417"getresuid() -> (ruid, euid, suid)\n\n\
9418Get tuple of the current process's real, effective, and saved user ids.");
9419
9420static PyObject*
9421posix_getresuid (PyObject *self, PyObject *noargs)
9422{
Victor Stinner8c62be82010-05-06 00:08:46 +00009423 uid_t ruid, euid, suid;
9424 long l_ruid, l_euid, l_suid;
9425 if (getresuid(&ruid, &euid, &suid) < 0)
9426 return posix_error();
9427 /* Force the values into long's as we don't know the size of uid_t. */
9428 l_ruid = ruid;
9429 l_euid = euid;
9430 l_suid = suid;
9431 return Py_BuildValue("(lll)", l_ruid, l_euid, l_suid);
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009432}
9433#endif
9434
9435#ifdef HAVE_GETRESGID
9436PyDoc_STRVAR(posix_getresgid__doc__,
9437"getresgid() -> (rgid, egid, sgid)\n\n\
Georg Brandla9b51d22010-09-05 17:07:12 +00009438Get tuple of the current process's real, effective, and saved group ids.");
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009439
9440static PyObject*
9441posix_getresgid (PyObject *self, PyObject *noargs)
9442{
Victor Stinner8c62be82010-05-06 00:08:46 +00009443 uid_t rgid, egid, sgid;
9444 long l_rgid, l_egid, l_sgid;
9445 if (getresgid(&rgid, &egid, &sgid) < 0)
9446 return posix_error();
9447 /* Force the values into long's as we don't know the size of uid_t. */
9448 l_rgid = rgid;
9449 l_egid = egid;
9450 l_sgid = sgid;
9451 return Py_BuildValue("(lll)", l_rgid, l_egid, l_sgid);
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009452}
9453#endif
9454
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009455/* Posix *at family of functions:
9456 faccessat, fchmodat, fchownat, fstatat, futimesat,
9457 linkat, mkdirat, mknodat, openat, readlinkat, renameat, symlinkat,
9458 unlinkat, utimensat, mkfifoat */
9459
9460#ifdef HAVE_FACCESSAT
9461PyDoc_STRVAR(posix_faccessat__doc__,
9462"faccessat(dirfd, path, mode, flags=0) -> True if granted, False otherwise\n\n\
9463Like access() but if path is relative, it is taken as relative to dirfd.\n\
9464flags is optional and can be constructed by ORing together zero or more\n\
9465of these values: AT_SYMLINK_NOFOLLOW, AT_EACCESS.\n\
9466If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9467is interpreted relative to the current working directory.");
9468
9469static PyObject *
9470posix_faccessat(PyObject *self, PyObject *args)
9471{
9472 PyObject *opath;
9473 char *path;
9474 int mode;
9475 int res;
9476 int dirfd, flags = 0;
9477 if (!PyArg_ParseTuple(args, "iO&i|i:faccessat",
9478 &dirfd, PyUnicode_FSConverter, &opath, &mode, &flags))
9479 return NULL;
9480 path = PyBytes_AsString(opath);
9481 Py_BEGIN_ALLOW_THREADS
9482 res = faccessat(dirfd, path, mode, flags);
9483 Py_END_ALLOW_THREADS
9484 Py_DECREF(opath);
9485 return PyBool_FromLong(res == 0);
9486}
9487#endif
9488
9489#ifdef HAVE_FCHMODAT
9490PyDoc_STRVAR(posix_fchmodat__doc__,
9491"fchmodat(dirfd, path, mode, flags=0)\n\n\
9492Like chmod() but if path is relative, it is taken as relative to dirfd.\n\
9493flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9494If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9495is interpreted relative to the current working directory.");
9496
9497static PyObject *
9498posix_fchmodat(PyObject *self, PyObject *args)
9499{
9500 int dirfd, mode, res;
9501 int flags = 0;
9502 PyObject *opath;
9503 char *path;
9504
9505 if (!PyArg_ParseTuple(args, "iO&i|i:fchmodat",
9506 &dirfd, PyUnicode_FSConverter, &opath, &mode, &flags))
9507 return NULL;
9508
9509 path = PyBytes_AsString(opath);
9510
9511 Py_BEGIN_ALLOW_THREADS
9512 res = fchmodat(dirfd, path, mode, flags);
9513 Py_END_ALLOW_THREADS
9514 Py_DECREF(opath);
9515 if (res < 0)
9516 return posix_error();
9517 Py_RETURN_NONE;
9518}
9519#endif /* HAVE_FCHMODAT */
9520
9521#ifdef HAVE_FCHOWNAT
9522PyDoc_STRVAR(posix_fchownat__doc__,
9523"fchownat(dirfd, path, uid, gid, flags=0)\n\n\
9524Like chown() but if path is relative, it is taken as relative to dirfd.\n\
9525flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9526If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9527is interpreted relative to the current working directory.");
9528
9529static PyObject *
9530posix_fchownat(PyObject *self, PyObject *args)
9531{
9532 PyObject *opath;
9533 int dirfd, res;
9534 long uid, gid;
9535 int flags = 0;
9536 char *path;
Giampaolo Rodola'ff1a7352011-04-19 09:47:16 +02009537
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009538 if (!PyArg_ParseTuple(args, "iO&ll|i:fchownat",
9539 &dirfd, PyUnicode_FSConverter, &opath, &uid, &gid, &flags))
9540 return NULL;
9541
9542 path = PyBytes_AsString(opath);
9543
9544 Py_BEGIN_ALLOW_THREADS
9545 res = fchownat(dirfd, path, (uid_t) uid, (gid_t) gid, flags);
9546 Py_END_ALLOW_THREADS
9547 Py_DECREF(opath);
9548 if (res < 0)
9549 return posix_error();
9550 Py_RETURN_NONE;
9551}
9552#endif /* HAVE_FCHOWNAT */
9553
9554#ifdef HAVE_FSTATAT
9555PyDoc_STRVAR(posix_fstatat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01009556"fstatat(dirfd, path, flags=0) -> stat result\n\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009557Like stat() but if path is relative, it is taken as relative to dirfd.\n\
9558flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9559If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9560is interpreted relative to the current working directory.");
9561
9562static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01009563posix_fstatat(PyObject *self, PyObject *args)
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009564{
9565 PyObject *opath;
9566 char *path;
9567 STRUCT_STAT st;
9568 int dirfd, res, flags = 0;
9569
Victor Stinner4195b5c2012-02-08 23:03:19 +01009570 if (!PyArg_ParseTuple(args, "iO&|i:fstatat",
9571 &dirfd, PyUnicode_FSConverter, &opath, &flags))
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009572 return NULL;
9573 path = PyBytes_AsString(opath);
9574
9575 Py_BEGIN_ALLOW_THREADS
9576 res = fstatat(dirfd, path, &st, flags);
9577 Py_END_ALLOW_THREADS
9578 Py_DECREF(opath);
9579 if (res != 0)
9580 return posix_error();
9581
Victor Stinner4195b5c2012-02-08 23:03:19 +01009582 return _pystat_fromstructstat(&st);
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009583}
9584#endif
9585
9586#ifdef HAVE_FUTIMESAT
9587PyDoc_STRVAR(posix_futimesat__doc__,
Brian Curtin7ef53ef2011-11-07 14:38:24 -06009588"futimesat(dirfd, path[, (atime, mtime)])\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009589Like utime() but if path is relative, it is taken as relative to dirfd.\n\
9590If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9591is interpreted relative to the current working directory.");
9592
9593static PyObject *
9594posix_futimesat(PyObject *self, PyObject *args)
9595{
9596 PyObject *opath;
9597 char *path;
9598 int res, dirfd;
Brian Curtin7ef53ef2011-11-07 14:38:24 -06009599 PyObject* arg = Py_None;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009600 time_t atime, mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009601 long ansec, mnsec;
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009602
Brian Curtin7ef53ef2011-11-07 14:38:24 -06009603 if (!PyArg_ParseTuple(args, "iO&|O:futimesat",
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009604 &dirfd, PyUnicode_FSConverter, &opath, &arg))
9605 return NULL;
9606 path = PyBytes_AsString(opath);
9607 if (arg == Py_None) {
9608 /* optional time values not given */
9609 Py_BEGIN_ALLOW_THREADS
9610 res = futimesat(dirfd, path, NULL);
9611 Py_END_ALLOW_THREADS
9612 }
9613 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
9614 PyErr_SetString(PyExc_TypeError,
9615 "futimesat() arg 3 must be a tuple (atime, mtime)");
9616 Py_DECREF(opath);
9617 return NULL;
9618 }
9619 else {
Victor Stinner5d272cc2012-03-13 13:35:55 +01009620 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 0),
9621 &atime, &ansec) == -1) {
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009622 Py_DECREF(opath);
9623 return NULL;
9624 }
Victor Stinner5d272cc2012-03-13 13:35:55 +01009625 if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(arg, 1),
9626 &mtime, &mnsec) == -1) {
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009627 Py_DECREF(opath);
9628 return NULL;
9629 }
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009630
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009631 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009632 {
9633#ifdef HAVE_UTIMENSAT
9634 struct timespec buf[2];
9635 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009636 buf[0].tv_nsec = ansec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009637 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009638 buf[1].tv_nsec = mnsec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009639 res = utimensat(dirfd, path, buf, 0);
9640#else
9641 struct timeval buf[2];
9642 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009643 buf[0].tv_usec = ansec / 1000;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009644 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009645 buf[1].tv_usec = mnsec / 1000;
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009646 res = futimesat(dirfd, path, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009647#endif
9648 }
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009649 Py_END_ALLOW_THREADS
9650 }
9651 Py_DECREF(opath);
9652 if (res < 0) {
9653 return posix_error();
9654 }
9655 Py_RETURN_NONE;
9656}
9657#endif
9658
9659#ifdef HAVE_LINKAT
9660PyDoc_STRVAR(posix_linkat__doc__,
9661"linkat(srcfd, srcpath, dstfd, dstpath, flags=0)\n\n\
9662Like link() but if srcpath is relative, it is taken as relative to srcfd\n\
9663and if dstpath is relative, it is taken as relative to dstfd.\n\
9664flags is optional and may be 0 or AT_SYMLINK_FOLLOW.\n\
9665If srcpath is relative and srcfd is the special value AT_FDCWD, then\n\
9666srcpath is interpreted relative to the current working directory. This\n\
9667also applies for dstpath.");
9668
9669static PyObject *
9670posix_linkat(PyObject *self, PyObject *args)
9671{
9672 PyObject *osrc, *odst;
9673 char *src, *dst;
9674 int res, srcfd, dstfd;
9675 int flags = 0;
9676
9677 if (!PyArg_ParseTuple(args, "iO&iO&|i:linkat",
9678 &srcfd, PyUnicode_FSConverter, &osrc, &dstfd, PyUnicode_FSConverter, &odst, &flags))
9679 return NULL;
9680 src = PyBytes_AsString(osrc);
9681 dst = PyBytes_AsString(odst);
9682 Py_BEGIN_ALLOW_THREADS
9683 res = linkat(srcfd, src, dstfd, dst, flags);
9684 Py_END_ALLOW_THREADS
9685 Py_DECREF(osrc);
9686 Py_DECREF(odst);
9687 if (res < 0)
9688 return posix_error();
9689 Py_RETURN_NONE;
9690}
9691#endif /* HAVE_LINKAT */
9692
9693#ifdef HAVE_MKDIRAT
9694PyDoc_STRVAR(posix_mkdirat__doc__,
9695"mkdirat(dirfd, path, mode=0o777)\n\n\
9696Like mkdir() but if path is relative, it is taken as relative to dirfd.\n\
9697If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9698is interpreted relative to the current working directory.");
9699
9700static PyObject *
9701posix_mkdirat(PyObject *self, PyObject *args)
9702{
9703 int res, dirfd;
9704 PyObject *opath;
9705 char *path;
9706 int mode = 0777;
9707
9708 if (!PyArg_ParseTuple(args, "iO&|i:mkdirat",
9709 &dirfd, PyUnicode_FSConverter, &opath, &mode))
9710 return NULL;
9711 path = PyBytes_AsString(opath);
9712 Py_BEGIN_ALLOW_THREADS
9713 res = mkdirat(dirfd, path, mode);
9714 Py_END_ALLOW_THREADS
9715 Py_DECREF(opath);
9716 if (res < 0)
9717 return posix_error();
9718 Py_RETURN_NONE;
9719}
9720#endif
9721
9722#if defined(HAVE_MKNODAT) && defined(HAVE_MAKEDEV)
9723PyDoc_STRVAR(posix_mknodat__doc__,
9724"mknodat(dirfd, path, mode=0o600, device=0)\n\n\
9725Like mknod() but if path is relative, it is taken as relative to dirfd.\n\
9726If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9727is interpreted relative to the current working directory.");
9728
9729static PyObject *
9730posix_mknodat(PyObject *self, PyObject *args)
9731{
9732 PyObject *opath;
9733 char *filename;
9734 int mode = 0600;
9735 int device = 0;
9736 int res, dirfd;
9737 if (!PyArg_ParseTuple(args, "iO&|ii:mknodat", &dirfd,
9738 PyUnicode_FSConverter, &opath, &mode, &device))
9739 return NULL;
9740 filename = PyBytes_AS_STRING(opath);
9741 Py_BEGIN_ALLOW_THREADS
9742 res = mknodat(dirfd, filename, mode, device);
9743 Py_END_ALLOW_THREADS
9744 Py_DECREF(opath);
9745 if (res < 0)
9746 return posix_error();
9747 Py_RETURN_NONE;
9748}
9749#endif
9750
9751#ifdef HAVE_OPENAT
9752PyDoc_STRVAR(posix_openat__doc__,
9753"openat(dirfd, path, flag, mode=0o777) -> fd\n\n\
9754Like open() but if path is relative, it is taken as relative to dirfd.\n\
9755If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9756is interpreted relative to the current working directory.");
9757
9758static PyObject *
9759posix_openat(PyObject *self, PyObject *args)
9760{
9761 PyObject *ofile;
9762 char *file;
9763 int flag, dirfd, fd;
9764 int mode = 0777;
9765
9766 if (!PyArg_ParseTuple(args, "iO&i|i:openat",
9767 &dirfd, PyUnicode_FSConverter, &ofile,
9768 &flag, &mode))
9769 return NULL;
9770 file = PyBytes_AsString(ofile);
9771 Py_BEGIN_ALLOW_THREADS
9772 fd = openat(dirfd, file, flag, mode);
9773 Py_END_ALLOW_THREADS
9774 Py_DECREF(ofile);
9775 if (fd < 0)
9776 return posix_error();
9777 return PyLong_FromLong((long)fd);
9778}
9779#endif
9780
9781#ifdef HAVE_READLINKAT
9782PyDoc_STRVAR(posix_readlinkat__doc__,
9783"readlinkat(dirfd, path) -> path\n\n\
9784Like readlink() but if path is relative, it is taken as relative to dirfd.\n\
9785If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9786is interpreted relative to the current working directory.");
9787
9788static PyObject *
9789posix_readlinkat(PyObject *self, PyObject *args)
9790{
9791 PyObject *v, *opath;
9792 char buf[MAXPATHLEN];
9793 char *path;
9794 int n, dirfd;
9795 int arg_is_unicode = 0;
9796
9797 if (!PyArg_ParseTuple(args, "iO&:readlinkat",
9798 &dirfd, PyUnicode_FSConverter, &opath))
9799 return NULL;
9800 path = PyBytes_AsString(opath);
9801 v = PySequence_GetItem(args, 1);
9802 if (v == NULL) {
9803 Py_DECREF(opath);
9804 return NULL;
9805 }
9806
9807 if (PyUnicode_Check(v)) {
9808 arg_is_unicode = 1;
9809 }
9810 Py_DECREF(v);
9811
9812 Py_BEGIN_ALLOW_THREADS
9813 n = readlinkat(dirfd, path, buf, (int) sizeof buf);
9814 Py_END_ALLOW_THREADS
9815 Py_DECREF(opath);
9816 if (n < 0)
9817 return posix_error();
9818
9819 if (arg_is_unicode)
9820 return PyUnicode_DecodeFSDefaultAndSize(buf, n);
9821 else
9822 return PyBytes_FromStringAndSize(buf, n);
9823}
9824#endif /* HAVE_READLINKAT */
9825
9826#ifdef HAVE_RENAMEAT
9827PyDoc_STRVAR(posix_renameat__doc__,
9828"renameat(olddirfd, oldpath, newdirfd, newpath)\n\n\
9829Like rename() but if oldpath is relative, it is taken as relative to\n\
9830olddirfd and if newpath is relative, it is taken as relative to newdirfd.\n\
9831If oldpath is relative and olddirfd is the special value AT_FDCWD, then\n\
9832oldpath is interpreted relative to the current working directory. This\n\
9833also applies for newpath.");
9834
9835static PyObject *
9836posix_renameat(PyObject *self, PyObject *args)
9837{
9838 int res;
9839 PyObject *opathold, *opathnew;
9840 char *opath, *npath;
9841 int oldfd, newfd;
9842
9843 if (!PyArg_ParseTuple(args, "iO&iO&:renameat",
9844 &oldfd, PyUnicode_FSConverter, &opathold, &newfd, PyUnicode_FSConverter, &opathnew))
9845 return NULL;
9846 opath = PyBytes_AsString(opathold);
9847 npath = PyBytes_AsString(opathnew);
9848 Py_BEGIN_ALLOW_THREADS
9849 res = renameat(oldfd, opath, newfd, npath);
9850 Py_END_ALLOW_THREADS
9851 Py_DECREF(opathold);
9852 Py_DECREF(opathnew);
9853 if (res < 0)
9854 return posix_error();
9855 Py_RETURN_NONE;
9856}
9857#endif
9858
9859#if HAVE_SYMLINKAT
9860PyDoc_STRVAR(posix_symlinkat__doc__,
9861"symlinkat(src, dstfd, dst)\n\n\
9862Like symlink() but if dst is relative, it is taken as relative to dstfd.\n\
9863If dst is relative and dstfd is the special value AT_FDCWD, then dst\n\
9864is interpreted relative to the current working directory.");
9865
9866static PyObject *
9867posix_symlinkat(PyObject *self, PyObject *args)
9868{
9869 int res, dstfd;
9870 PyObject *osrc, *odst;
9871 char *src, *dst;
9872
9873 if (!PyArg_ParseTuple(args, "O&iO&:symlinkat",
9874 PyUnicode_FSConverter, &osrc, &dstfd, PyUnicode_FSConverter, &odst))
9875 return NULL;
9876 src = PyBytes_AsString(osrc);
9877 dst = PyBytes_AsString(odst);
9878 Py_BEGIN_ALLOW_THREADS
9879 res = symlinkat(src, dstfd, dst);
9880 Py_END_ALLOW_THREADS
9881 Py_DECREF(osrc);
9882 Py_DECREF(odst);
9883 if (res < 0)
9884 return posix_error();
9885 Py_RETURN_NONE;
9886}
9887#endif /* HAVE_SYMLINKAT */
9888
9889#ifdef HAVE_UNLINKAT
9890PyDoc_STRVAR(posix_unlinkat__doc__,
9891"unlinkat(dirfd, path, flags=0)\n\n\
9892Like unlink() but if path is relative, it is taken as relative to dirfd.\n\
9893flags is optional and may be 0 or AT_REMOVEDIR. If AT_REMOVEDIR is\n\
9894specified, unlinkat() behaves like rmdir().\n\
9895If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9896is interpreted relative to the current working directory.");
9897
9898static PyObject *
9899posix_unlinkat(PyObject *self, PyObject *args)
9900{
9901 int dirfd, res, flags = 0;
9902 PyObject *opath;
9903 char *path;
9904
9905 if (!PyArg_ParseTuple(args, "iO&|i:unlinkat",
9906 &dirfd, PyUnicode_FSConverter, &opath, &flags))
9907 return NULL;
9908 path = PyBytes_AsString(opath);
9909 Py_BEGIN_ALLOW_THREADS
9910 res = unlinkat(dirfd, path, flags);
9911 Py_END_ALLOW_THREADS
9912 Py_DECREF(opath);
9913 if (res < 0)
9914 return posix_error();
9915 Py_RETURN_NONE;
9916}
9917#endif
9918
9919#ifdef HAVE_UTIMENSAT
9920PyDoc_STRVAR(posix_utimensat__doc__,
Brian Curtin569b4942011-11-07 16:09:20 -06009921"utimensat(dirfd, path[, atime=(atime_sec, atime_nsec),\n\
9922 mtime=(mtime_sec, mtime_nsec), flags=0])\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009923utimensat(dirfd, path, None, None, flags)\n\n\
9924Updates the timestamps of a file with nanosecond precision. If path is\n\
9925relative, it is taken as relative to dirfd.\n\
Brian Curtin569b4942011-11-07 16:09:20 -06009926If atime and mtime are both None, which is the default, set atime and\n\
9927mtime to the current time.\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009928flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9929If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9930is interpreted relative to the current working directory.\n\
9931If *_nsec is specified as UTIME_NOW, the timestamp is updated to the\n\
9932current time.\n\
9933If *_nsec is specified as UTIME_OMIT, the timestamp is not updated.");
9934
9935static PyObject *
Brian Curtin569b4942011-11-07 16:09:20 -06009936posix_utimensat(PyObject *self, PyObject *args, PyObject *kwargs)
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009937{
9938 PyObject *opath;
9939 char *path;
9940 int res, dirfd, flags = 0;
Brian Curtin569b4942011-11-07 16:09:20 -06009941 PyObject *atime = Py_None;
9942 PyObject *mtime = Py_None;
9943
9944 static char *kwlist[] = {"dirfd", "path", "atime", "mtime", "flags", NULL};
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009945
9946 struct timespec buf[2];
9947
Brian Curtin569b4942011-11-07 16:09:20 -06009948 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "iO&|OOi:utimensat", kwlist,
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009949 &dirfd, PyUnicode_FSConverter, &opath, &atime, &mtime, &flags))
9950 return NULL;
9951 path = PyBytes_AsString(opath);
9952 if (atime == Py_None && mtime == Py_None) {
9953 /* optional time values not given */
9954 Py_BEGIN_ALLOW_THREADS
9955 res = utimensat(dirfd, path, NULL, flags);
9956 Py_END_ALLOW_THREADS
9957 }
9958 else if (!PyTuple_Check(atime) || PyTuple_Size(atime) != 2) {
9959 PyErr_SetString(PyExc_TypeError,
9960 "utimensat() arg 3 must be a tuple (atime_sec, atime_nsec)");
9961 Py_DECREF(opath);
9962 return NULL;
9963 }
9964 else if (!PyTuple_Check(mtime) || PyTuple_Size(mtime) != 2) {
9965 PyErr_SetString(PyExc_TypeError,
9966 "utimensat() arg 4 must be a tuple (mtime_sec, mtime_nsec)");
9967 Py_DECREF(opath);
9968 return NULL;
9969 }
9970 else {
9971 if (!PyArg_ParseTuple(atime, "ll:utimensat",
9972 &(buf[0].tv_sec), &(buf[0].tv_nsec))) {
9973 Py_DECREF(opath);
9974 return NULL;
9975 }
9976 if (!PyArg_ParseTuple(mtime, "ll:utimensat",
9977 &(buf[1].tv_sec), &(buf[1].tv_nsec))) {
9978 Py_DECREF(opath);
9979 return NULL;
9980 }
9981 Py_BEGIN_ALLOW_THREADS
9982 res = utimensat(dirfd, path, buf, flags);
9983 Py_END_ALLOW_THREADS
9984 }
9985 Py_DECREF(opath);
9986 if (res < 0) {
9987 return posix_error();
9988 }
9989 Py_RETURN_NONE;
9990}
9991#endif
9992
9993#ifdef HAVE_MKFIFOAT
9994PyDoc_STRVAR(posix_mkfifoat__doc__,
9995"mkfifoat(dirfd, path, mode=0o666)\n\n\
9996Like mkfifo() but if path is relative, it is taken as relative to dirfd.\n\
9997If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9998is interpreted relative to the current working directory.");
9999
10000static PyObject *
10001posix_mkfifoat(PyObject *self, PyObject *args)
10002{
10003 PyObject *opath;
10004 char *filename;
10005 int mode = 0666;
10006 int res, dirfd;
10007 if (!PyArg_ParseTuple(args, "iO&|i:mkfifoat",
10008 &dirfd, PyUnicode_FSConverter, &opath, &mode))
10009 return NULL;
10010 filename = PyBytes_AS_STRING(opath);
10011 Py_BEGIN_ALLOW_THREADS
10012 res = mkfifoat(dirfd, filename, mode);
10013 Py_END_ALLOW_THREADS
10014 Py_DECREF(opath);
10015 if (res < 0)
10016 return posix_error();
10017 Py_RETURN_NONE;
10018}
10019#endif
10020
Benjamin Peterson9428d532011-09-14 11:45:52 -040010021#ifdef USE_XATTRS
Benjamin Peterson799bd802011-08-31 22:15:17 -040010022
10023static int
10024try_getxattr(const char *path, const char *name,
10025 ssize_t (*get)(const char *, const char *, void *, size_t),
10026 Py_ssize_t buf_size, PyObject **res)
10027{
10028 PyObject *value;
10029 Py_ssize_t len;
10030
10031 assert(buf_size <= XATTR_SIZE_MAX);
10032 value = PyBytes_FromStringAndSize(NULL, buf_size);
10033 if (!value)
10034 return 0;
10035 Py_BEGIN_ALLOW_THREADS;
10036 len = get(path, name, PyBytes_AS_STRING(value), buf_size);
10037 Py_END_ALLOW_THREADS;
10038 if (len < 0) {
10039 Py_DECREF(value);
10040 if (errno == ERANGE) {
10041 value = NULL;
10042 }
10043 else {
10044 posix_error();
10045 return 0;
10046 }
10047 }
10048 else if (len != buf_size) {
10049 /* Can only shrink. */
10050 _PyBytes_Resize(&value, len);
10051 }
10052 *res = value;
10053 return 1;
10054}
10055
10056static PyObject *
10057getxattr_common(const char *path, PyObject *name_obj,
10058 ssize_t (*get)(const char *, const char *, void *, size_t))
10059{
10060 PyObject *value;
10061 const char *name = PyBytes_AS_STRING(name_obj);
10062
10063 /* Try a small value first. */
10064 if (!try_getxattr(path, name, get, 128, &value))
10065 return NULL;
10066 if (value)
10067 return value;
10068 /* Now the maximum possible one. */
10069 if (!try_getxattr(path, name, get, XATTR_SIZE_MAX, &value))
10070 return NULL;
10071 assert(value);
10072 return value;
10073}
10074
10075PyDoc_STRVAR(posix_getxattr__doc__,
10076"getxattr(path, attr) -> value\n\n\
10077Return the value of extended attribute *name* on *path*.");
10078
10079static PyObject *
10080posix_getxattr(PyObject *self, PyObject *args)
10081{
10082 PyObject *path, *res, *name;
10083
10084 if (!PyArg_ParseTuple(args, "O&O&:getxattr", PyUnicode_FSConverter, &path,
10085 PyUnicode_FSConverter, &name))
10086 return NULL;
10087 res = getxattr_common(PyBytes_AS_STRING(path), name, getxattr);
10088 Py_DECREF(path);
10089 Py_DECREF(name);
10090 return res;
10091}
10092
10093PyDoc_STRVAR(posix_lgetxattr__doc__,
10094"lgetxattr(path, attr) -> value\n\n\
10095Like getxattr but don't follow symlinks.");
10096
10097static PyObject *
10098posix_lgetxattr(PyObject *self, PyObject *args)
10099{
10100 PyObject *path, *res, *name;
10101
10102 if (!PyArg_ParseTuple(args, "O&O&:lgetxattr", PyUnicode_FSConverter, &path,
10103 PyUnicode_FSConverter, &name))
10104 return NULL;
10105 res = getxattr_common(PyBytes_AS_STRING(path), name, lgetxattr);
10106 Py_DECREF(path);
10107 Py_DECREF(name);
10108 return res;
10109}
10110
10111static ssize_t
10112wrap_fgetxattr(const char *path, const char *name, void *value, size_t size)
10113{
10114 /* Hack to share code. */
10115 return fgetxattr((int)(Py_uintptr_t)path, name, value, size);
10116}
10117
10118PyDoc_STRVAR(posix_fgetxattr__doc__,
10119"fgetxattr(fd, attr) -> value\n\n\
10120Like getxattr but operate on a fd instead of a path.");
10121
10122static PyObject *
10123posix_fgetxattr(PyObject *self, PyObject *args)
10124{
10125 PyObject *res, *name;
10126 int fd;
10127
10128 if (!PyArg_ParseTuple(args, "iO&:fgetxattr", &fd, PyUnicode_FSConverter, &name))
10129 return NULL;
10130 res = getxattr_common((const char *)(Py_uintptr_t)fd, name, wrap_fgetxattr);
10131 Py_DECREF(name);
10132 return res;
10133}
10134
10135PyDoc_STRVAR(posix_setxattr__doc__,
10136"setxattr(path, attr, value, flags=0)\n\n\
10137Set extended attribute *attr* on *path* to *value*.");
10138
10139static PyObject *
10140posix_setxattr(PyObject *self, PyObject *args)
10141{
10142 PyObject *path, *name;
10143 Py_buffer data;
10144 int flags = 0, err;
10145
10146 if (!PyArg_ParseTuple(args, "O&O&y*|i:setxattr", PyUnicode_FSConverter,
10147 &path, PyUnicode_FSConverter, &name, &data, &flags))
10148 return NULL;
10149 Py_BEGIN_ALLOW_THREADS;
10150 err = setxattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name),
10151 data.buf, data.len, flags);
10152 Py_END_ALLOW_THREADS;
10153 Py_DECREF(path);
10154 Py_DECREF(name);
10155 PyBuffer_Release(&data);
10156 if (err)
10157 return posix_error();
10158 Py_RETURN_NONE;
10159}
10160
10161PyDoc_STRVAR(posix_lsetxattr__doc__,
10162"lsetxattr(path, attr, value, flags=0)\n\n\
10163Like setxattr but don't follow symlinks.");
10164
10165static PyObject *
10166posix_lsetxattr(PyObject *self, PyObject *args)
10167{
10168 PyObject *path, *name;
10169 Py_buffer data;
10170 int flags = 0, err;
10171
10172 if (!PyArg_ParseTuple(args, "O&O&y*|i:lsetxattr", PyUnicode_FSConverter,
10173 &path, PyUnicode_FSConverter, &name, &data, &flags))
10174 return NULL;
10175 Py_BEGIN_ALLOW_THREADS;
10176 err = lsetxattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name),
10177 data.buf, data.len, flags);
10178 Py_END_ALLOW_THREADS;
10179 Py_DECREF(path);
10180 Py_DECREF(name);
10181 PyBuffer_Release(&data);
10182 if (err)
10183 return posix_error();
10184 Py_RETURN_NONE;
10185}
10186
10187PyDoc_STRVAR(posix_fsetxattr__doc__,
10188"fsetxattr(fd, attr, value, flags=0)\n\n\
10189Like setxattr but operates on *fd* instead of a path.");
10190
10191static PyObject *
10192posix_fsetxattr(PyObject *self, PyObject *args)
10193{
10194 Py_buffer data;
10195 const char *name;
10196 int fd, flags = 0, err;
10197
10198 if (!PyArg_ParseTuple(args, "iO&y*|i:fsetxattr", &fd, PyUnicode_FSConverter,
10199 &name, &data, &flags))
10200 return NULL;
10201 Py_BEGIN_ALLOW_THREADS;
10202 err = fsetxattr(fd, PyBytes_AS_STRING(name), data.buf, data.len, flags);
10203 Py_END_ALLOW_THREADS;
10204 Py_DECREF(name);
10205 PyBuffer_Release(&data);
10206 if (err)
10207 return posix_error();
10208 Py_RETURN_NONE;
10209}
10210
10211PyDoc_STRVAR(posix_removexattr__doc__,
10212"removexattr(path, attr)\n\n\
10213Remove extended attribute *attr* on *path*.");
10214
10215static PyObject *
10216posix_removexattr(PyObject *self, PyObject *args)
10217{
10218 PyObject *path, *name;
10219 int err;
10220
10221 if (!PyArg_ParseTuple(args, "O&O&:removexattr", PyUnicode_FSConverter, &path,
10222 PyUnicode_FSConverter, &name))
10223 return NULL;
10224 Py_BEGIN_ALLOW_THREADS;
10225 err = removexattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name));
10226 Py_END_ALLOW_THREADS;
10227 Py_DECREF(path);
10228 Py_DECREF(name);
10229 if (err)
10230 return posix_error();
10231 Py_RETURN_NONE;
10232}
10233
10234PyDoc_STRVAR(posix_lremovexattr__doc__,
10235"lremovexattr(path, attr)\n\n\
10236Like removexattr but don't follow symlinks.");
10237
10238static PyObject *
10239posix_lremovexattr(PyObject *self, PyObject *args)
10240{
10241 PyObject *path, *name;
10242 int err;
10243
10244 if (!PyArg_ParseTuple(args, "O&O&:lremovexattr", PyUnicode_FSConverter, &path,
10245 PyUnicode_FSConverter, &name))
10246 return NULL;
10247 Py_BEGIN_ALLOW_THREADS;
10248 err = lremovexattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name));
10249 Py_END_ALLOW_THREADS;
10250 Py_DECREF(path);
10251 Py_DECREF(name);
10252 if (err)
10253 return posix_error();
10254 Py_RETURN_NONE;
10255}
10256
10257PyDoc_STRVAR(posix_fremovexattr__doc__,
10258"fremovexattr(fd, attr)\n\n\
10259Like removexattr but operates on a file descriptor.");
10260
10261static PyObject *
10262posix_fremovexattr(PyObject *self, PyObject *args)
10263{
10264 PyObject *name;
10265 int fd, err;
10266
10267 if (!PyArg_ParseTuple(args, "iO&:fremovexattr", &fd,
10268 PyUnicode_FSConverter, &name))
10269 return NULL;
10270 Py_BEGIN_ALLOW_THREADS;
10271 err = fremovexattr(fd, PyBytes_AS_STRING(name));
10272 Py_END_ALLOW_THREADS;
10273 Py_DECREF(name);
10274 if (err)
10275 return posix_error();
10276 Py_RETURN_NONE;
10277}
10278
10279static Py_ssize_t
10280try_listxattr(const char *path, ssize_t (*list)(const char *, char *, size_t),
10281 Py_ssize_t buf_size, char **buf)
10282{
10283 Py_ssize_t len;
10284
10285 *buf = PyMem_MALLOC(buf_size);
10286 if (!*buf) {
10287 PyErr_NoMemory();
10288 return -1;
10289 }
10290 Py_BEGIN_ALLOW_THREADS;
10291 len = list(path, *buf, buf_size);
10292 Py_END_ALLOW_THREADS;
10293 if (len < 0) {
10294 PyMem_FREE(*buf);
10295 if (errno != ERANGE)
10296 posix_error();
10297 return -1;
10298 }
10299 return len;
10300}
10301
10302static PyObject *
10303listxattr_common(const char *path, ssize_t (*list)(const char *, char *, size_t))
10304{
10305 PyObject *res, *attr;
10306 Py_ssize_t len, err, start, i;
10307 char *buf;
10308
10309 len = try_listxattr(path, list, 256, &buf);
10310 if (len < 0) {
10311 if (PyErr_Occurred())
10312 return NULL;
10313 len = try_listxattr(path, list, XATTR_LIST_MAX, &buf);
10314 if (len < 0)
10315 return NULL;
10316 }
10317 res = PyList_New(0);
10318 if (!res) {
10319 PyMem_FREE(buf);
10320 return NULL;
10321 }
10322 for (start = i = 0; i < len; i++) {
10323 if (!buf[i]) {
10324 attr = PyUnicode_DecodeFSDefaultAndSize(&buf[start], i - start);
10325 if (!attr) {
10326 Py_DECREF(res);
10327 PyMem_FREE(buf);
10328 return NULL;
10329 }
10330 err = PyList_Append(res, attr);
10331 Py_DECREF(attr);
10332 if (err) {
10333 Py_DECREF(res);
10334 PyMem_FREE(buf);
10335 return NULL;
10336 }
10337 start = i + 1;
10338 }
10339 }
10340 PyMem_FREE(buf);
10341 return res;
10342}
10343
10344PyDoc_STRVAR(posix_listxattr__doc__,
10345"listxattr(path)\n\n\
10346Return a list of extended attributes on *path*.");
10347
10348static PyObject *
10349posix_listxattr(PyObject *self, PyObject *args)
10350{
10351 PyObject *path, *res;
10352
10353 if (!PyArg_ParseTuple(args, "O&:listxattr", PyUnicode_FSConverter, &path))
10354 return NULL;
10355 res = listxattr_common(PyBytes_AS_STRING(path), listxattr);
10356 Py_DECREF(path);
10357 return res;
10358}
10359
10360PyDoc_STRVAR(posix_llistxattr__doc__,
10361"llistxattr(path)\n\n\
10362Like listxattr but don't follow symlinks..");
10363
10364static PyObject *
10365posix_llistxattr(PyObject *self, PyObject *args)
10366{
10367 PyObject *path, *res;
10368
10369 if (!PyArg_ParseTuple(args, "O&:llistxattr", PyUnicode_FSConverter, &path))
10370 return NULL;
10371 res = listxattr_common(PyBytes_AS_STRING(path), llistxattr);
10372 Py_DECREF(path);
10373 return res;
10374}
10375
10376static ssize_t
10377wrap_flistxattr(const char *path, char *buf, size_t len)
10378{
10379 /* Hack to share code. */
10380 return flistxattr((int)(Py_uintptr_t)path, buf, len);
10381}
10382
10383PyDoc_STRVAR(posix_flistxattr__doc__,
10384"flistxattr(path)\n\n\
10385Like flistxattr but operates on a file descriptor.");
10386
10387static PyObject *
10388posix_flistxattr(PyObject *self, PyObject *args)
10389{
10390 long fd;
10391
10392 if (!PyArg_ParseTuple(args, "i:flistxattr", &fd))
10393 return NULL;
10394 return listxattr_common((const char *)(Py_uintptr_t)fd, wrap_flistxattr);
10395}
10396
Benjamin Peterson9428d532011-09-14 11:45:52 -040010397#endif /* USE_XATTRS */
Benjamin Peterson799bd802011-08-31 22:15:17 -040010398
Antoine Pitroubcf2b592012-02-08 23:28:36 +010010399
Georg Brandl2fb477c2012-02-21 00:33:36 +010010400PyDoc_STRVAR(posix_urandom__doc__,
10401"urandom(n) -> str\n\n\
10402Return n random bytes suitable for cryptographic use.");
10403
10404static PyObject *
10405posix_urandom(PyObject *self, PyObject *args)
10406{
10407 Py_ssize_t size;
10408 PyObject *result;
10409 int ret;
10410
10411 /* Read arguments */
10412 if (!PyArg_ParseTuple(args, "n:urandom", &size))
10413 return NULL;
10414 if (size < 0)
10415 return PyErr_Format(PyExc_ValueError,
10416 "negative argument not allowed");
10417 result = PyBytes_FromStringAndSize(NULL, size);
10418 if (result == NULL)
10419 return NULL;
10420
10421 ret = _PyOS_URandom(PyBytes_AS_STRING(result),
10422 PyBytes_GET_SIZE(result));
10423 if (ret == -1) {
10424 Py_DECREF(result);
10425 return NULL;
10426 }
10427 return result;
10428}
10429
Antoine Pitroubcf2b592012-02-08 23:28:36 +010010430/* Terminal size querying */
10431
10432static PyTypeObject TerminalSizeType;
10433
10434PyDoc_STRVAR(TerminalSize_docstring,
10435 "A tuple of (columns, lines) for holding terminal window size");
10436
10437static PyStructSequence_Field TerminalSize_fields[] = {
10438 {"columns", "width of the terminal window in characters"},
10439 {"lines", "height of the terminal window in characters"},
10440 {NULL, NULL}
10441};
10442
10443static PyStructSequence_Desc TerminalSize_desc = {
10444 "os.terminal_size",
10445 TerminalSize_docstring,
10446 TerminalSize_fields,
10447 2,
10448};
10449
10450#if defined(TERMSIZE_USE_CONIO) || defined(TERMSIZE_USE_IOCTL)
10451PyDoc_STRVAR(termsize__doc__,
10452 "Return the size of the terminal window as (columns, lines).\n" \
10453 "\n" \
10454 "The optional argument fd (default standard output) specifies\n" \
10455 "which file descriptor should be queried.\n" \
10456 "\n" \
10457 "If the file descriptor is not connected to a terminal, an OSError\n" \
10458 "is thrown.\n" \
10459 "\n" \
10460 "This function will only be defined if an implementation is\n" \
10461 "available for this system.\n" \
10462 "\n" \
10463 "shutil.get_terminal_size is the high-level function which should \n" \
10464 "normally be used, os.get_terminal_size is the low-level implementation.");
10465
10466static PyObject*
10467get_terminal_size(PyObject *self, PyObject *args)
10468{
10469 int columns, lines;
10470 PyObject *termsize;
10471
10472 int fd = fileno(stdout);
10473 /* Under some conditions stdout may not be connected and
10474 * fileno(stdout) may point to an invalid file descriptor. For example
10475 * GUI apps don't have valid standard streams by default.
10476 *
10477 * If this happens, and the optional fd argument is not present,
10478 * the ioctl below will fail returning EBADF. This is what we want.
10479 */
10480
10481 if (!PyArg_ParseTuple(args, "|i", &fd))
10482 return NULL;
10483
10484#ifdef TERMSIZE_USE_IOCTL
10485 {
10486 struct winsize w;
10487 if (ioctl(fd, TIOCGWINSZ, &w))
10488 return PyErr_SetFromErrno(PyExc_OSError);
10489 columns = w.ws_col;
10490 lines = w.ws_row;
10491 }
10492#endif /* TERMSIZE_USE_IOCTL */
10493
10494#ifdef TERMSIZE_USE_CONIO
10495 {
10496 DWORD nhandle;
10497 HANDLE handle;
10498 CONSOLE_SCREEN_BUFFER_INFO csbi;
10499 switch (fd) {
10500 case 0: nhandle = STD_INPUT_HANDLE;
10501 break;
10502 case 1: nhandle = STD_OUTPUT_HANDLE;
10503 break;
10504 case 2: nhandle = STD_ERROR_HANDLE;
10505 break;
10506 default:
10507 return PyErr_Format(PyExc_ValueError, "bad file descriptor");
10508 }
10509 handle = GetStdHandle(nhandle);
10510 if (handle == NULL)
10511 return PyErr_Format(PyExc_OSError, "handle cannot be retrieved");
10512 if (handle == INVALID_HANDLE_VALUE)
10513 return PyErr_SetFromWindowsErr(0);
10514
10515 if (!GetConsoleScreenBufferInfo(handle, &csbi))
10516 return PyErr_SetFromWindowsErr(0);
10517
10518 columns = csbi.srWindow.Right - csbi.srWindow.Left + 1;
10519 lines = csbi.srWindow.Bottom - csbi.srWindow.Top + 1;
10520 }
10521#endif /* TERMSIZE_USE_CONIO */
10522
10523 termsize = PyStructSequence_New(&TerminalSizeType);
10524 if (termsize == NULL)
10525 return NULL;
10526 PyStructSequence_SET_ITEM(termsize, 0, PyLong_FromLong(columns));
10527 PyStructSequence_SET_ITEM(termsize, 1, PyLong_FromLong(lines));
10528 if (PyErr_Occurred()) {
10529 Py_DECREF(termsize);
10530 return NULL;
10531 }
10532 return termsize;
10533}
10534#endif /* defined(TERMSIZE_USE_CONIO) || defined(TERMSIZE_USE_IOCTL) */
10535
10536
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010537static PyMethodDef posix_methods[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +000010538 {"access", posix_access, METH_VARARGS, posix_access__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010539#ifdef HAVE_TTYNAME
Victor Stinner8c62be82010-05-06 00:08:46 +000010540 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010541#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010542 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +000010543#ifdef HAVE_CHFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +000010544 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +000010545#endif /* HAVE_CHFLAGS */
Victor Stinner8c62be82010-05-06 00:08:46 +000010546 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010547#ifdef HAVE_FCHMOD
Victor Stinner8c62be82010-05-06 00:08:46 +000010548 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010549#endif /* HAVE_FCHMOD */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010550#ifdef HAVE_CHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010551 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010552#endif /* HAVE_CHOWN */
Christian Heimes4e30a842007-11-30 22:12:06 +000010553#ifdef HAVE_LCHMOD
Victor Stinner8c62be82010-05-06 00:08:46 +000010554 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010555#endif /* HAVE_LCHMOD */
10556#ifdef HAVE_FCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010557 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010558#endif /* HAVE_FCHOWN */
Thomas Wouterscf297e42007-02-23 15:07:44 +000010559#ifdef HAVE_LCHFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +000010560 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +000010561#endif /* HAVE_LCHFLAGS */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +000010562#ifdef HAVE_LCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010563 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +000010564#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +000010565#ifdef HAVE_CHROOT
Victor Stinner8c62be82010-05-06 00:08:46 +000010566 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
Martin v. Löwis244edc82001-10-04 22:44:26 +000010567#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010568#ifdef HAVE_CTERMID
Victor Stinner8c62be82010-05-06 00:08:46 +000010569 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010570#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010571#ifdef HAVE_GETCWD
Victor Stinner8c62be82010-05-06 00:08:46 +000010572 {"getcwd", (PyCFunction)posix_getcwd_unicode,
10573 METH_NOARGS, posix_getcwd__doc__},
10574 {"getcwdb", (PyCFunction)posix_getcwd_bytes,
10575 METH_NOARGS, posix_getcwdb__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010576#endif
10577#ifdef HAVE_LINK
Victor Stinner8c62be82010-05-06 00:08:46 +000010578 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010579#endif /* HAVE_LINK */
Victor Stinner8c62be82010-05-06 00:08:46 +000010580 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
Antoine Pitrou8250e232011-02-25 23:41:16 +000010581#ifdef HAVE_FDOPENDIR
Charles-François Natali77940902012-02-06 19:54:48 +010010582 {"flistdir", posix_flistdir, METH_VARARGS, posix_flistdir__doc__},
Antoine Pitrou8250e232011-02-25 23:41:16 +000010583#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +010010584 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +000010585 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010586#ifdef HAVE_NICE
Victor Stinner8c62be82010-05-06 00:08:46 +000010587 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum775f4da1993-01-09 17:18:52 +000010588#endif /* HAVE_NICE */
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +000010589#ifdef HAVE_GETPRIORITY
10590 {"getpriority", posix_getpriority, METH_VARARGS, posix_getpriority__doc__},
10591#endif /* HAVE_GETPRIORITY */
10592#ifdef HAVE_SETPRIORITY
10593 {"setpriority", posix_setpriority, METH_VARARGS, posix_setpriority__doc__},
10594#endif /* HAVE_SETPRIORITY */
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010595#ifdef HAVE_READLINK
Victor Stinner8c62be82010-05-06 00:08:46 +000010596 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010597#endif /* HAVE_READLINK */
Brian Curtind40e6f72010-07-08 21:39:08 +000010598#if !defined(HAVE_READLINK) && defined(MS_WINDOWS)
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +000010599 {"readlink", win_readlink, METH_VARARGS, win_readlink__doc__},
Brian Curtind40e6f72010-07-08 21:39:08 +000010600#endif /* !defined(HAVE_READLINK) && defined(MS_WINDOWS) */
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +000010601 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
Antoine Pitrouf3b2d882012-01-30 22:08:52 +010010602 {"replace", posix_replace, METH_VARARGS, posix_replace__doc__},
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +000010603 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
Victor Stinner4195b5c2012-02-08 23:03:19 +010010604 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +000010605 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Brian Curtin52173d42010-12-02 18:29:18 +000010606#if defined(HAVE_SYMLINK) && !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +000010607 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010608#endif /* HAVE_SYMLINK */
Brian Curtin52173d42010-12-02 18:29:18 +000010609#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000010610 {"symlink", (PyCFunction)win_symlink, METH_VARARGS | METH_KEYWORDS,
Brian Curtin52173d42010-12-02 18:29:18 +000010611 win_symlink__doc__},
10612#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
Guido van Rossum85e3b011991-06-03 12:42:10 +000010613#ifdef HAVE_SYSTEM
Victor Stinner8c62be82010-05-06 00:08:46 +000010614 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010615#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010616 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010617#ifdef HAVE_UNAME
Victor Stinner8c62be82010-05-06 00:08:46 +000010618 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010619#endif /* HAVE_UNAME */
Victor Stinner8c62be82010-05-06 00:08:46 +000010620 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
10621 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
10622 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010623#ifdef HAVE_FUTIMES
10624 {"futimes", posix_futimes, METH_VARARGS, posix_futimes__doc__},
10625#endif
10626#ifdef HAVE_LUTIMES
10627 {"lutimes", posix_lutimes, METH_VARARGS, posix_lutimes__doc__},
10628#endif
10629#ifdef HAVE_FUTIMENS
10630 {"futimens", posix_futimens, METH_VARARGS, posix_futimens__doc__},
10631#endif
Guido van Rossum0ee42cd1991-04-08 21:01:03 +000010632#ifdef HAVE_TIMES
Victor Stinner8c62be82010-05-06 00:08:46 +000010633 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum3b066191991-06-04 19:40:25 +000010634#endif /* HAVE_TIMES */
Victor Stinner8c62be82010-05-06 00:08:46 +000010635 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossum3b066191991-06-04 19:40:25 +000010636#ifdef HAVE_EXECV
Victor Stinner8c62be82010-05-06 00:08:46 +000010637 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
10638 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossum3b066191991-06-04 19:40:25 +000010639#endif /* HAVE_EXECV */
Ross Lagerwall7807c352011-03-17 20:20:30 +020010640#ifdef HAVE_FEXECVE
10641 {"fexecve", posix_fexecve, METH_VARARGS, posix_fexecve__doc__},
10642#endif
Guido van Rossuma1065681999-01-25 23:20:23 +000010643#ifdef HAVE_SPAWNV
Victor Stinner8c62be82010-05-06 00:08:46 +000010644 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
10645 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +000010646#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +000010647 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
10648 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +000010649#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +000010650#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +000010651#ifdef HAVE_FORK1
Victor Stinner8c62be82010-05-06 00:08:46 +000010652 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +000010653#endif /* HAVE_FORK1 */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010654#ifdef HAVE_FORK
Victor Stinner8c62be82010-05-06 00:08:46 +000010655 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010656#endif /* HAVE_FORK */
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010657#ifdef HAVE_SCHED_H
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +020010658#ifdef HAVE_SCHED_GET_PRIORITY_MAX
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010659 {"sched_get_priority_max", posix_sched_get_priority_max, METH_VARARGS, posix_sched_get_priority_max__doc__},
10660 {"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 +020010661#endif
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010662#ifdef HAVE_SCHED_SETPARAM
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010663 {"sched_getparam", posix_sched_getparam, METH_VARARGS, posix_sched_getparam__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010664#endif
10665#ifdef HAVE_SCHED_SETSCHEDULER
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010666 {"sched_getscheduler", posix_sched_getscheduler, METH_VARARGS, posix_sched_getscheduler__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010667#endif
10668#ifdef HAVE_SCHED_RR_GET_INTERVAL
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010669 {"sched_rr_get_interval", posix_sched_rr_get_interval, METH_VARARGS, posix_sched_rr_get_interval__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010670#endif
10671#ifdef HAVE_SCHED_SETPARAM
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010672 {"sched_setparam", posix_sched_setparam, METH_VARARGS, posix_sched_setparam__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010673#endif
10674#ifdef HAVE_SCHED_SETSCHEDULER
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010675 {"sched_setscheduler", posix_sched_setscheduler, METH_VARARGS, posix_sched_setscheduler__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010676#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010677 {"sched_yield", posix_sched_yield, METH_NOARGS, posix_sched_yield__doc__},
Benjamin Peterson2740af82011-08-02 17:41:34 -050010678#ifdef HAVE_SCHED_SETAFFINITY
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010679 {"sched_setaffinity", posix_sched_setaffinity, METH_VARARGS, posix_sched_setaffinity__doc__},
10680 {"sched_getaffinity", posix_sched_getaffinity, METH_VARARGS, posix_sched_getaffinity__doc__},
10681#endif
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +020010682#endif /* HAVE_SCHED_H */
Martin v. Löwis24a880b2002-12-31 12:55:15 +000010683#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Victor Stinner8c62be82010-05-06 00:08:46 +000010684 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +000010685#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +000010686#ifdef HAVE_FORKPTY
Victor Stinner8c62be82010-05-06 00:08:46 +000010687 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +000010688#endif /* HAVE_FORKPTY */
Guido van Rossum3b066191991-06-04 19:40:25 +000010689#ifdef HAVE_GETEGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010690 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +000010691#endif /* HAVE_GETEGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010692#ifdef HAVE_GETEUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010693 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossum46003ff1992-05-15 11:05:24 +000010694#endif /* HAVE_GETEUID */
10695#ifdef HAVE_GETGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010696 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010697#endif /* HAVE_GETGID */
Ross Lagerwallb0ae53d2011-06-10 07:30:30 +020010698#ifdef HAVE_GETGROUPLIST
10699 {"getgrouplist", posix_getgrouplist, METH_VARARGS, posix_getgrouplist__doc__},
10700#endif
Fred Drakec9680921999-12-13 16:37:25 +000010701#ifdef HAVE_GETGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +000010702 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010703#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010704 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +000010705#ifdef HAVE_GETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010706 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010707#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +000010708#ifdef HAVE_GETPPID
Victor Stinner8c62be82010-05-06 00:08:46 +000010709 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010710#endif /* HAVE_GETPPID */
10711#ifdef HAVE_GETUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010712 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010713#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +000010714#ifdef HAVE_GETLOGIN
Victor Stinner8c62be82010-05-06 00:08:46 +000010715 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +000010716#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +000010717#ifdef HAVE_KILL
Victor Stinner8c62be82010-05-06 00:08:46 +000010718 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010719#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +000010720#ifdef HAVE_KILLPG
Victor Stinner8c62be82010-05-06 00:08:46 +000010721 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
Martin v. Löwisb2c92f42002-02-16 23:35:41 +000010722#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +000010723#ifdef HAVE_PLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000010724 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +000010725#endif /* HAVE_PLOCK */
Thomas Heller8b7a9572007-08-31 06:44:36 +000010726#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +000010727 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
10728 {"kill", win32_kill, METH_VARARGS, win32_kill__doc__},
Brian Curtin1b9df392010-11-24 20:24:31 +000010729 {"link", win32_link, METH_VARARGS, win32_link__doc__},
Thomas Heller8b7a9572007-08-31 06:44:36 +000010730#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +000010731#ifdef HAVE_SETUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010732 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010733#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010734#ifdef HAVE_SETEUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010735 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010736#endif /* HAVE_SETEUID */
10737#ifdef HAVE_SETEGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010738 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010739#endif /* HAVE_SETEGID */
10740#ifdef HAVE_SETREUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010741 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010742#endif /* HAVE_SETREUID */
10743#ifdef HAVE_SETREGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010744 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010745#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010746#ifdef HAVE_SETGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010747 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010748#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +000010749#ifdef HAVE_SETGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +000010750 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
Martin v. Löwis61c5edf2001-10-18 04:06:00 +000010751#endif /* HAVE_SETGROUPS */
Antoine Pitroub7572f02009-12-02 20:46:48 +000010752#ifdef HAVE_INITGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +000010753 {"initgroups", posix_initgroups, METH_VARARGS, posix_initgroups__doc__},
Antoine Pitroub7572f02009-12-02 20:46:48 +000010754#endif /* HAVE_INITGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +000010755#ifdef HAVE_GETPGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010756 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
Martin v. Löwis606edc12002-06-13 21:09:11 +000010757#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010758#ifdef HAVE_SETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010759 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010760#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +000010761#ifdef HAVE_WAIT
Victor Stinner8c62be82010-05-06 00:08:46 +000010762 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010763#endif /* HAVE_WAIT */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010764#ifdef HAVE_WAIT3
Victor Stinner4195b5c2012-02-08 23:03:19 +010010765 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010766#endif /* HAVE_WAIT3 */
10767#ifdef HAVE_WAIT4
Victor Stinner4195b5c2012-02-08 23:03:19 +010010768 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010769#endif /* HAVE_WAIT4 */
Ross Lagerwall7807c352011-03-17 20:20:30 +020010770#if defined(HAVE_WAITID) && !defined(__APPLE__)
10771 {"waitid", posix_waitid, METH_VARARGS, posix_waitid__doc__},
10772#endif
Tim Petersab034fa2002-02-01 11:27:43 +000010773#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Victor Stinner8c62be82010-05-06 00:08:46 +000010774 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010775#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +000010776#ifdef HAVE_GETSID
Victor Stinner8c62be82010-05-06 00:08:46 +000010777 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
Martin v. Löwis49ee14d2003-11-10 06:35:36 +000010778#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010779#ifdef HAVE_SETSID
Victor Stinner8c62be82010-05-06 00:08:46 +000010780 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010781#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010782#ifdef HAVE_SETPGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010783 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010784#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010785#ifdef HAVE_TCGETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010786 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010787#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010788#ifdef HAVE_TCSETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010789 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010790#endif /* HAVE_TCSETPGRP */
Victor Stinner8c62be82010-05-06 00:08:46 +000010791 {"open", posix_open, METH_VARARGS, posix_open__doc__},
10792 {"close", posix_close, METH_VARARGS, posix_close__doc__},
10793 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
10794 {"device_encoding", device_encoding, METH_VARARGS, device_encoding__doc__},
10795 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
10796 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010797#ifdef HAVE_LOCKF
10798 {"lockf", posix_lockf, METH_VARARGS, posix_lockf__doc__},
10799#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010800 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
10801 {"read", posix_read, METH_VARARGS, posix_read__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010802#ifdef HAVE_READV
10803 {"readv", posix_readv, METH_VARARGS, posix_readv__doc__},
10804#endif
10805#ifdef HAVE_PREAD
10806 {"pread", posix_pread, METH_VARARGS, posix_pread__doc__},
10807#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010808 {"write", posix_write, METH_VARARGS, posix_write__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010809#ifdef HAVE_WRITEV
10810 {"writev", posix_writev, METH_VARARGS, posix_writev__doc__},
10811#endif
10812#ifdef HAVE_PWRITE
10813 {"pwrite", posix_pwrite, METH_VARARGS, posix_pwrite__doc__},
10814#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +000010815#ifdef HAVE_SENDFILE
10816 {"sendfile", (PyCFunction)posix_sendfile, METH_VARARGS | METH_KEYWORDS,
10817 posix_sendfile__doc__},
10818#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +010010819 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +000010820 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010821#ifdef HAVE_PIPE
Victor Stinner8c62be82010-05-06 00:08:46 +000010822 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010823#endif
Charles-François Natalidaafdd52011-05-29 20:07:40 +020010824#ifdef HAVE_PIPE2
Charles-François Natali368f34b2011-06-06 19:49:47 +020010825 {"pipe2", posix_pipe2, METH_O, posix_pipe2__doc__},
Charles-François Natalidaafdd52011-05-29 20:07:40 +020010826#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010827#ifdef HAVE_MKFIFO
Victor Stinner8c62be82010-05-06 00:08:46 +000010828 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010829#endif
Neal Norwitz11690112002-07-30 01:08:28 +000010830#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Victor Stinner8c62be82010-05-06 00:08:46 +000010831 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
Martin v. Löwis06a83e92002-04-14 10:19:44 +000010832#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +000010833#ifdef HAVE_DEVICE_MACROS
Victor Stinner8c62be82010-05-06 00:08:46 +000010834 {"major", posix_major, METH_VARARGS, posix_major__doc__},
10835 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
10836 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
Martin v. Löwisdbe3f762002-10-10 14:27:30 +000010837#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010838#ifdef HAVE_FTRUNCATE
Victor Stinner8c62be82010-05-06 00:08:46 +000010839 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010840#endif
Ross Lagerwall7807c352011-03-17 20:20:30 +020010841#ifdef HAVE_TRUNCATE
10842 {"truncate", posix_truncate, METH_VARARGS, posix_truncate__doc__},
10843#endif
10844#ifdef HAVE_POSIX_FALLOCATE
10845 {"posix_fallocate", posix_posix_fallocate, METH_VARARGS, posix_posix_fallocate__doc__},
10846#endif
10847#ifdef HAVE_POSIX_FADVISE
10848 {"posix_fadvise", posix_posix_fadvise, METH_VARARGS, posix_posix_fadvise__doc__},
10849#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +000010850#ifdef HAVE_PUTENV
Victor Stinner8c62be82010-05-06 00:08:46 +000010851 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +000010852#endif
Guido van Rossumc524d952001-10-19 01:31:59 +000010853#ifdef HAVE_UNSETENV
Victor Stinner8c62be82010-05-06 00:08:46 +000010854 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
Guido van Rossumc524d952001-10-19 01:31:59 +000010855#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010856 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +000010857#ifdef HAVE_FCHDIR
Victor Stinner8c62be82010-05-06 00:08:46 +000010858 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +000010859#endif
Guido van Rossum21142a01999-01-08 21:05:37 +000010860#ifdef HAVE_FSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000010861 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +000010862#endif
Ross Lagerwall7807c352011-03-17 20:20:30 +020010863#ifdef HAVE_SYNC
10864 {"sync", posix_sync, METH_NOARGS, posix_sync__doc__},
10865#endif
Guido van Rossum21142a01999-01-08 21:05:37 +000010866#ifdef HAVE_FDATASYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000010867 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +000010868#endif
Guido van Rossumc9641791998-08-04 15:26:23 +000010869#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +000010870#ifdef WCOREDUMP
Victor Stinner8c62be82010-05-06 00:08:46 +000010871 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
Fred Drake106c1a02002-04-23 15:58:02 +000010872#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +000010873#ifdef WIFCONTINUED
Victor Stinner8c62be82010-05-06 00:08:46 +000010874 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +000010875#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +000010876#ifdef WIFSTOPPED
Victor Stinner8c62be82010-05-06 00:08:46 +000010877 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010878#endif /* WIFSTOPPED */
10879#ifdef WIFSIGNALED
Victor Stinner8c62be82010-05-06 00:08:46 +000010880 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010881#endif /* WIFSIGNALED */
10882#ifdef WIFEXITED
Victor Stinner8c62be82010-05-06 00:08:46 +000010883 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010884#endif /* WIFEXITED */
10885#ifdef WEXITSTATUS
Victor Stinner8c62be82010-05-06 00:08:46 +000010886 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010887#endif /* WEXITSTATUS */
10888#ifdef WTERMSIG
Victor Stinner8c62be82010-05-06 00:08:46 +000010889 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010890#endif /* WTERMSIG */
10891#ifdef WSTOPSIG
Victor Stinner8c62be82010-05-06 00:08:46 +000010892 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010893#endif /* WSTOPSIG */
10894#endif /* HAVE_SYS_WAIT_H */
Thomas Wouters477c8d52006-05-27 19:21:47 +000010895#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Victor Stinner8c62be82010-05-06 00:08:46 +000010896 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +000010897#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +000010898#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Victor Stinner8c62be82010-05-06 00:08:46 +000010899 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +000010900#endif
Fred Drakec9680921999-12-13 16:37:25 +000010901#ifdef HAVE_CONFSTR
Victor Stinner8c62be82010-05-06 00:08:46 +000010902 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010903#endif
10904#ifdef HAVE_SYSCONF
Victor Stinner8c62be82010-05-06 00:08:46 +000010905 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010906#endif
10907#ifdef HAVE_FPATHCONF
Victor Stinner8c62be82010-05-06 00:08:46 +000010908 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010909#endif
10910#ifdef HAVE_PATHCONF
Victor Stinner8c62be82010-05-06 00:08:46 +000010911 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010912#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010913 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000010914#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +000010915 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
Brian Curtind40e6f72010-07-08 21:39:08 +000010916 {"_getfinalpathname", posix__getfinalpathname, METH_VARARGS, NULL},
Brian Curtin62857742010-09-06 17:07:27 +000010917 {"_getfileinformation", posix__getfileinformation, METH_VARARGS, NULL},
Brian Curtin95d028f2011-06-09 09:10:38 -050010918 {"_isdir", posix__isdir, METH_VARARGS, posix__isdir__doc__},
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +020010919 {"_getdiskusage", win32__getdiskusage, METH_VARARGS, win32__getdiskusage__doc__},
Mark Hammondef8b6542001-05-13 08:04:26 +000010920#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +000010921#ifdef HAVE_GETLOADAVG
Victor Stinner8c62be82010-05-06 00:08:46 +000010922 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +000010923#endif
Georg Brandl2daf6ae2012-02-20 19:54:16 +010010924 {"urandom", posix_urandom, METH_VARARGS, posix_urandom__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010925#ifdef HAVE_SETRESUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010926 {"setresuid", posix_setresuid, METH_VARARGS, posix_setresuid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010927#endif
10928#ifdef HAVE_SETRESGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010929 {"setresgid", posix_setresgid, METH_VARARGS, posix_setresgid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010930#endif
10931#ifdef HAVE_GETRESUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010932 {"getresuid", posix_getresuid, METH_NOARGS, posix_getresuid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010933#endif
10934#ifdef HAVE_GETRESGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010935 {"getresgid", posix_getresgid, METH_NOARGS, posix_getresgid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010936#endif
10937
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010938/* posix *at family of functions */
10939#ifdef HAVE_FACCESSAT
10940 {"faccessat", posix_faccessat, METH_VARARGS, posix_faccessat__doc__},
10941#endif
10942#ifdef HAVE_FCHMODAT
10943 {"fchmodat", posix_fchmodat, METH_VARARGS, posix_fchmodat__doc__},
10944#endif /* HAVE_FCHMODAT */
10945#ifdef HAVE_FCHOWNAT
10946 {"fchownat", posix_fchownat, METH_VARARGS, posix_fchownat__doc__},
10947#endif /* HAVE_FCHOWNAT */
10948#ifdef HAVE_FSTATAT
Victor Stinner4195b5c2012-02-08 23:03:19 +010010949 {"fstatat", posix_fstatat, METH_VARARGS, posix_fstatat__doc__},
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010950#endif
10951#ifdef HAVE_FUTIMESAT
10952 {"futimesat", posix_futimesat, METH_VARARGS, posix_futimesat__doc__},
10953#endif
10954#ifdef HAVE_LINKAT
10955 {"linkat", posix_linkat, METH_VARARGS, posix_linkat__doc__},
10956#endif /* HAVE_LINKAT */
10957#ifdef HAVE_MKDIRAT
10958 {"mkdirat", posix_mkdirat, METH_VARARGS, posix_mkdirat__doc__},
10959#endif
10960#if defined(HAVE_MKNODAT) && defined(HAVE_MAKEDEV)
10961 {"mknodat", posix_mknodat, METH_VARARGS, posix_mknodat__doc__},
10962#endif
10963#ifdef HAVE_OPENAT
10964 {"openat", posix_openat, METH_VARARGS, posix_openat__doc__},
10965#endif
10966#ifdef HAVE_READLINKAT
10967 {"readlinkat", posix_readlinkat, METH_VARARGS, posix_readlinkat__doc__},
10968#endif /* HAVE_READLINKAT */
10969#ifdef HAVE_RENAMEAT
10970 {"renameat", posix_renameat, METH_VARARGS, posix_renameat__doc__},
10971#endif
10972#if HAVE_SYMLINKAT
10973 {"symlinkat", posix_symlinkat, METH_VARARGS, posix_symlinkat__doc__},
10974#endif /* HAVE_SYMLINKAT */
10975#ifdef HAVE_UNLINKAT
10976 {"unlinkat", posix_unlinkat, METH_VARARGS, posix_unlinkat__doc__},
10977#endif
10978#ifdef HAVE_UTIMENSAT
Jesus Cead03a4912011-11-08 17:28:04 +010010979 {"utimensat", (PyCFunction)posix_utimensat,
10980 METH_VARARGS | METH_KEYWORDS,
Brian Curtin569b4942011-11-07 16:09:20 -060010981 posix_utimensat__doc__},
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010982#endif
10983#ifdef HAVE_MKFIFOAT
10984 {"mkfifoat", posix_mkfifoat, METH_VARARGS, posix_mkfifoat__doc__},
10985#endif
Benjamin Peterson9428d532011-09-14 11:45:52 -040010986#ifdef USE_XATTRS
Benjamin Peterson799bd802011-08-31 22:15:17 -040010987 {"setxattr", posix_setxattr, METH_VARARGS, posix_setxattr__doc__},
10988 {"lsetxattr", posix_lsetxattr, METH_VARARGS, posix_lsetxattr__doc__},
10989 {"fsetxattr", posix_fsetxattr, METH_VARARGS, posix_fsetxattr__doc__},
10990 {"getxattr", posix_getxattr, METH_VARARGS, posix_getxattr__doc__},
10991 {"lgetxattr", posix_lgetxattr, METH_VARARGS, posix_lgetxattr__doc__},
10992 {"fgetxattr", posix_fgetxattr, METH_VARARGS, posix_fgetxattr__doc__},
10993 {"removexattr", posix_removexattr, METH_VARARGS, posix_removexattr__doc__},
10994 {"lremovexattr", posix_lremovexattr, METH_VARARGS, posix_lremovexattr__doc__},
10995 {"fremovexattr", posix_fremovexattr, METH_VARARGS, posix_fremovexattr__doc__},
10996 {"listxattr", posix_listxattr, METH_VARARGS, posix_listxattr__doc__},
10997 {"llistxattr", posix_llistxattr, METH_VARARGS, posix_llistxattr__doc__},
10998 {"flistxattr", posix_flistxattr, METH_VARARGS, posix_flistxattr__doc__},
10999#endif
Antoine Pitroubcf2b592012-02-08 23:28:36 +010011000#if defined(TERMSIZE_USE_CONIO) || defined(TERMSIZE_USE_IOCTL)
11001 {"get_terminal_size", get_terminal_size, METH_VARARGS, termsize__doc__},
11002#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011003 {NULL, NULL} /* Sentinel */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000011004};
11005
11006
Barry Warsaw4a342091996-12-19 23:50:02 +000011007static int
Fred Drake4d1e64b2002-04-15 19:40:07 +000011008ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +000011009{
Victor Stinner8c62be82010-05-06 00:08:46 +000011010 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +000011011}
11012
Guido van Rossumd48f2521997-12-05 22:19:34 +000011013#if defined(PYOS_OS2)
11014/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +000011015static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +000011016{
11017 APIRET rc;
11018 ULONG values[QSV_MAX+1];
11019 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +000011020 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +000011021
11022 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +000011023 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +000011024 Py_END_ALLOW_THREADS
11025
11026 if (rc != NO_ERROR) {
11027 os2_error(rc);
11028 return -1;
11029 }
11030
Fred Drake4d1e64b2002-04-15 19:40:07 +000011031 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
11032 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
11033 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
11034 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
11035 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
11036 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
11037 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000011038
11039 switch (values[QSV_VERSION_MINOR]) {
11040 case 0: ver = "2.00"; break;
11041 case 10: ver = "2.10"; break;
11042 case 11: ver = "2.11"; break;
11043 case 30: ver = "3.00"; break;
11044 case 40: ver = "4.00"; break;
11045 case 50: ver = "5.00"; break;
11046 default:
Tim Peters885d4572001-11-28 20:27:42 +000011047 PyOS_snprintf(tmp, sizeof(tmp),
Victor Stinner8c62be82010-05-06 00:08:46 +000011048 "%d-%d", values[QSV_VERSION_MAJOR],
Tim Peters885d4572001-11-28 20:27:42 +000011049 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +000011050 ver = &tmp[0];
11051 }
11052
11053 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +000011054 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +000011055 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000011056
11057 /* Add Indicator of Which Drive was Used to Boot the System */
11058 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
11059 tmp[1] = ':';
11060 tmp[2] = '\0';
11061
Fred Drake4d1e64b2002-04-15 19:40:07 +000011062 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +000011063}
11064#endif
11065
Brian Curtin52173d42010-12-02 18:29:18 +000011066#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000011067static int
Brian Curtin52173d42010-12-02 18:29:18 +000011068enable_symlink()
11069{
11070 HANDLE tok;
11071 TOKEN_PRIVILEGES tok_priv;
11072 LUID luid;
11073 int meth_idx = 0;
11074
11075 if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &tok))
Brian Curtin3b4499c2010-12-28 14:31:47 +000011076 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +000011077
11078 if (!LookupPrivilegeValue(NULL, SE_CREATE_SYMBOLIC_LINK_NAME, &luid))
Brian Curtin3b4499c2010-12-28 14:31:47 +000011079 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +000011080
11081 tok_priv.PrivilegeCount = 1;
11082 tok_priv.Privileges[0].Luid = luid;
11083 tok_priv.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
11084
11085 if (!AdjustTokenPrivileges(tok, FALSE, &tok_priv,
11086 sizeof(TOKEN_PRIVILEGES),
11087 (PTOKEN_PRIVILEGES) NULL, (PDWORD) NULL))
Brian Curtin3b4499c2010-12-28 14:31:47 +000011088 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +000011089
Brian Curtin3b4499c2010-12-28 14:31:47 +000011090 /* ERROR_NOT_ALL_ASSIGNED returned when the privilege can't be assigned. */
11091 return GetLastError() == ERROR_NOT_ALL_ASSIGNED ? 0 : 1;
Brian Curtin52173d42010-12-02 18:29:18 +000011092}
11093#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
11094
Barry Warsaw4a342091996-12-19 23:50:02 +000011095static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +000011096all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +000011097{
Guido van Rossum94f6f721999-01-06 18:42:14 +000011098#ifdef F_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011099 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011100#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +000011101#ifdef R_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011102 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011103#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +000011104#ifdef W_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011105 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011106#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +000011107#ifdef X_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011108 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011109#endif
Fred Drakec9680921999-12-13 16:37:25 +000011110#ifdef NGROUPS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +000011111 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
Fred Drakec9680921999-12-13 16:37:25 +000011112#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +000011113#ifdef TMP_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +000011114 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
Fred Drake5ab8eaf1999-12-09 21:13:07 +000011115#endif
Fred Drake106c1a02002-04-23 15:58:02 +000011116#ifdef WCONTINUED
Victor Stinner8c62be82010-05-06 00:08:46 +000011117 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
Fred Drake106c1a02002-04-23 15:58:02 +000011118#endif
Barry Warsaw4a342091996-12-19 23:50:02 +000011119#ifdef WNOHANG
Victor Stinner8c62be82010-05-06 00:08:46 +000011120 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011121#endif
Fred Drake106c1a02002-04-23 15:58:02 +000011122#ifdef WUNTRACED
Victor Stinner8c62be82010-05-06 00:08:46 +000011123 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
Fred Drake106c1a02002-04-23 15:58:02 +000011124#endif
Barry Warsaw4a342091996-12-19 23:50:02 +000011125#ifdef O_RDONLY
Victor Stinner8c62be82010-05-06 00:08:46 +000011126 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011127#endif
11128#ifdef O_WRONLY
Victor Stinner8c62be82010-05-06 00:08:46 +000011129 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011130#endif
11131#ifdef O_RDWR
Victor Stinner8c62be82010-05-06 00:08:46 +000011132 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011133#endif
11134#ifdef O_NDELAY
Victor Stinner8c62be82010-05-06 00:08:46 +000011135 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011136#endif
11137#ifdef O_NONBLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000011138 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011139#endif
11140#ifdef O_APPEND
Victor Stinner8c62be82010-05-06 00:08:46 +000011141 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011142#endif
11143#ifdef O_DSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011144 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011145#endif
11146#ifdef O_RSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011147 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011148#endif
11149#ifdef O_SYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011150 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011151#endif
11152#ifdef O_NOCTTY
Victor Stinner8c62be82010-05-06 00:08:46 +000011153 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011154#endif
11155#ifdef O_CREAT
Victor Stinner8c62be82010-05-06 00:08:46 +000011156 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011157#endif
11158#ifdef O_EXCL
Victor Stinner8c62be82010-05-06 00:08:46 +000011159 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011160#endif
11161#ifdef O_TRUNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011162 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011163#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +000011164#ifdef O_BINARY
Victor Stinner8c62be82010-05-06 00:08:46 +000011165 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
Guido van Rossum98d9d091997-08-08 21:48:51 +000011166#endif
11167#ifdef O_TEXT
Victor Stinner8c62be82010-05-06 00:08:46 +000011168 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
Guido van Rossum98d9d091997-08-08 21:48:51 +000011169#endif
Jesus Cea1d642d22012-04-24 20:59:17 +020011170#ifdef O_XATTR
11171 if (ins(d, "O_XATTR", (long)O_XATTR)) return -1;
11172#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011173#ifdef O_LARGEFILE
Victor Stinner8c62be82010-05-06 00:08:46 +000011174 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011175#endif
Skip Montanaro5ff14922005-05-16 02:42:22 +000011176#ifdef O_SHLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000011177 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
Skip Montanaro5ff14922005-05-16 02:42:22 +000011178#endif
11179#ifdef O_EXLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000011180 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
Skip Montanaro5ff14922005-05-16 02:42:22 +000011181#endif
Jesus Ceacf381202012-04-24 20:44:40 +020011182#ifdef O_EXEC
11183 if (ins(d, "O_EXEC", (long)O_EXEC)) return -1;
11184#endif
11185#ifdef O_SEARCH
11186 if (ins(d, "O_SEARCH", (long)O_SEARCH)) return -1;
11187#endif
11188#ifdef O_TTY_INIT
11189 if (ins(d, "O_TTY_INIT", (long)O_TTY_INIT)) return -1;
11190#endif
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +000011191#ifdef PRIO_PROCESS
11192 if (ins(d, "PRIO_PROCESS", (long)PRIO_PROCESS)) return -1;
11193#endif
11194#ifdef PRIO_PGRP
11195 if (ins(d, "PRIO_PGRP", (long)PRIO_PGRP)) return -1;
11196#endif
11197#ifdef PRIO_USER
11198 if (ins(d, "PRIO_USER", (long)PRIO_USER)) return -1;
11199#endif
Charles-François Natali1e045b12011-05-22 20:42:32 +020011200#ifdef O_CLOEXEC
11201 if (ins(d, "O_CLOEXEC", (long)O_CLOEXEC)) return -1;
11202#endif
Jesus Cea1d642d22012-04-24 20:59:17 +020011203#ifdef O_ACCMODE
11204 if (ins(d, "O_ACCMODE", (long)O_ACCMODE)) return -1;
11205#endif
Antoine Pitrouf65132d2011-02-25 23:25:17 +000011206/* posix - constants for *at functions */
11207#ifdef AT_SYMLINK_NOFOLLOW
11208 if (ins(d, "AT_SYMLINK_NOFOLLOW", (long)AT_SYMLINK_NOFOLLOW)) return -1;
11209#endif
11210#ifdef AT_EACCESS
11211 if (ins(d, "AT_EACCESS", (long)AT_EACCESS)) return -1;
11212#endif
11213#ifdef AT_FDCWD
11214 if (ins(d, "AT_FDCWD", (long)AT_FDCWD)) return -1;
11215#endif
11216#ifdef AT_REMOVEDIR
11217 if (ins(d, "AT_REMOVEDIR", (long)AT_REMOVEDIR)) return -1;
11218#endif
11219#ifdef AT_SYMLINK_FOLLOW
11220 if (ins(d, "AT_SYMLINK_FOLLOW", (long)AT_SYMLINK_FOLLOW)) return -1;
11221#endif
11222#ifdef UTIME_NOW
11223 if (ins(d, "UTIME_NOW", (long)UTIME_NOW)) return -1;
11224#endif
11225#ifdef UTIME_OMIT
11226 if (ins(d, "UTIME_OMIT", (long)UTIME_OMIT)) return -1;
11227#endif
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +000011228
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011229
Tim Peters5aa91602002-01-30 05:46:57 +000011230/* MS Windows */
11231#ifdef O_NOINHERIT
Victor Stinner8c62be82010-05-06 00:08:46 +000011232 /* Don't inherit in child processes. */
11233 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011234#endif
11235#ifdef _O_SHORT_LIVED
Victor Stinner8c62be82010-05-06 00:08:46 +000011236 /* Optimize for short life (keep in memory). */
11237 /* MS forgot to define this one with a non-underscore form too. */
11238 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011239#endif
11240#ifdef O_TEMPORARY
Victor Stinner8c62be82010-05-06 00:08:46 +000011241 /* Automatically delete when last handle is closed. */
11242 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011243#endif
11244#ifdef O_RANDOM
Victor Stinner8c62be82010-05-06 00:08:46 +000011245 /* Optimize for random access. */
11246 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011247#endif
11248#ifdef O_SEQUENTIAL
Victor Stinner8c62be82010-05-06 00:08:46 +000011249 /* Optimize for sequential access. */
11250 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011251#endif
11252
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011253/* GNU extensions. */
Alexandre Vassalottibee32532008-05-16 18:15:12 +000011254#ifdef O_ASYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011255 /* Send a SIGIO signal whenever input or output
11256 becomes available on file descriptor */
11257 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
Alexandre Vassalottibee32532008-05-16 18:15:12 +000011258#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011259#ifdef O_DIRECT
Victor Stinner8c62be82010-05-06 00:08:46 +000011260 /* Direct disk access. */
11261 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011262#endif
11263#ifdef O_DIRECTORY
Victor Stinner8c62be82010-05-06 00:08:46 +000011264 /* Must be a directory. */
11265 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011266#endif
11267#ifdef O_NOFOLLOW
Victor Stinner8c62be82010-05-06 00:08:46 +000011268 /* Do not follow links. */
11269 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011270#endif
Jesus Cea1d642d22012-04-24 20:59:17 +020011271#ifdef O_NOLINKS
11272 /* Fails if link count of the named file is greater than 1 */
11273 if (ins(d, "O_NOLINKS", (long)O_NOLINKS)) return -1;
11274#endif
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +000011275#ifdef O_NOATIME
Victor Stinner8c62be82010-05-06 00:08:46 +000011276 /* Do not update the access time. */
11277 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +000011278#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +000011279
Victor Stinner8c62be82010-05-06 00:08:46 +000011280 /* These come from sysexits.h */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011281#ifdef EX_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011282 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011283#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011284#ifdef EX_USAGE
Victor Stinner8c62be82010-05-06 00:08:46 +000011285 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011286#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011287#ifdef EX_DATAERR
Victor Stinner8c62be82010-05-06 00:08:46 +000011288 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011289#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011290#ifdef EX_NOINPUT
Victor Stinner8c62be82010-05-06 00:08:46 +000011291 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011292#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011293#ifdef EX_NOUSER
Victor Stinner8c62be82010-05-06 00:08:46 +000011294 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011295#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011296#ifdef EX_NOHOST
Victor Stinner8c62be82010-05-06 00:08:46 +000011297 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011298#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011299#ifdef EX_UNAVAILABLE
Victor Stinner8c62be82010-05-06 00:08:46 +000011300 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011301#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011302#ifdef EX_SOFTWARE
Victor Stinner8c62be82010-05-06 00:08:46 +000011303 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011304#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011305#ifdef EX_OSERR
Victor Stinner8c62be82010-05-06 00:08:46 +000011306 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011307#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011308#ifdef EX_OSFILE
Victor Stinner8c62be82010-05-06 00:08:46 +000011309 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011310#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011311#ifdef EX_CANTCREAT
Victor Stinner8c62be82010-05-06 00:08:46 +000011312 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011313#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011314#ifdef EX_IOERR
Victor Stinner8c62be82010-05-06 00:08:46 +000011315 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011316#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011317#ifdef EX_TEMPFAIL
Victor Stinner8c62be82010-05-06 00:08:46 +000011318 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011319#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011320#ifdef EX_PROTOCOL
Victor Stinner8c62be82010-05-06 00:08:46 +000011321 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011322#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011323#ifdef EX_NOPERM
Victor Stinner8c62be82010-05-06 00:08:46 +000011324 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011325#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011326#ifdef EX_CONFIG
Victor Stinner8c62be82010-05-06 00:08:46 +000011327 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011328#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011329#ifdef EX_NOTFOUND
Victor Stinner8c62be82010-05-06 00:08:46 +000011330 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011331#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011332
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +000011333 /* statvfs */
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +000011334#ifdef ST_RDONLY
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +000011335 if (ins(d, "ST_RDONLY", (long)ST_RDONLY)) return -1;
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +000011336#endif /* ST_RDONLY */
11337#ifdef ST_NOSUID
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +000011338 if (ins(d, "ST_NOSUID", (long)ST_NOSUID)) return -1;
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +000011339#endif /* ST_NOSUID */
11340
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +000011341 /* FreeBSD sendfile() constants */
11342#ifdef SF_NODISKIO
11343 if (ins(d, "SF_NODISKIO", (long)SF_NODISKIO)) return -1;
11344#endif
11345#ifdef SF_MNOWAIT
11346 if (ins(d, "SF_MNOWAIT", (long)SF_MNOWAIT)) return -1;
11347#endif
11348#ifdef SF_SYNC
11349 if (ins(d, "SF_SYNC", (long)SF_SYNC)) return -1;
11350#endif
11351
Ross Lagerwall7807c352011-03-17 20:20:30 +020011352 /* constants for posix_fadvise */
11353#ifdef POSIX_FADV_NORMAL
11354 if (ins(d, "POSIX_FADV_NORMAL", (long)POSIX_FADV_NORMAL)) return -1;
11355#endif
11356#ifdef POSIX_FADV_SEQUENTIAL
11357 if (ins(d, "POSIX_FADV_SEQUENTIAL", (long)POSIX_FADV_SEQUENTIAL)) return -1;
11358#endif
11359#ifdef POSIX_FADV_RANDOM
11360 if (ins(d, "POSIX_FADV_RANDOM", (long)POSIX_FADV_RANDOM)) return -1;
11361#endif
11362#ifdef POSIX_FADV_NOREUSE
11363 if (ins(d, "POSIX_FADV_NOREUSE", (long)POSIX_FADV_NOREUSE)) return -1;
11364#endif
11365#ifdef POSIX_FADV_WILLNEED
11366 if (ins(d, "POSIX_FADV_WILLNEED", (long)POSIX_FADV_WILLNEED)) return -1;
11367#endif
11368#ifdef POSIX_FADV_DONTNEED
11369 if (ins(d, "POSIX_FADV_DONTNEED", (long)POSIX_FADV_DONTNEED)) return -1;
11370#endif
11371
11372 /* constants for waitid */
11373#if defined(HAVE_SYS_WAIT_H) && defined(HAVE_WAITID)
11374 if (ins(d, "P_PID", (long)P_PID)) return -1;
11375 if (ins(d, "P_PGID", (long)P_PGID)) return -1;
11376 if (ins(d, "P_ALL", (long)P_ALL)) return -1;
11377#endif
11378#ifdef WEXITED
11379 if (ins(d, "WEXITED", (long)WEXITED)) return -1;
11380#endif
11381#ifdef WNOWAIT
11382 if (ins(d, "WNOWAIT", (long)WNOWAIT)) return -1;
11383#endif
11384#ifdef WSTOPPED
11385 if (ins(d, "WSTOPPED", (long)WSTOPPED)) return -1;
11386#endif
11387#ifdef CLD_EXITED
11388 if (ins(d, "CLD_EXITED", (long)CLD_EXITED)) return -1;
11389#endif
11390#ifdef CLD_DUMPED
11391 if (ins(d, "CLD_DUMPED", (long)CLD_DUMPED)) return -1;
11392#endif
11393#ifdef CLD_TRAPPED
11394 if (ins(d, "CLD_TRAPPED", (long)CLD_TRAPPED)) return -1;
11395#endif
11396#ifdef CLD_CONTINUED
11397 if (ins(d, "CLD_CONTINUED", (long)CLD_CONTINUED)) return -1;
11398#endif
11399
11400 /* constants for lockf */
11401#ifdef F_LOCK
11402 if (ins(d, "F_LOCK", (long)F_LOCK)) return -1;
11403#endif
11404#ifdef F_TLOCK
11405 if (ins(d, "F_TLOCK", (long)F_TLOCK)) return -1;
11406#endif
11407#ifdef F_ULOCK
11408 if (ins(d, "F_ULOCK", (long)F_ULOCK)) return -1;
11409#endif
11410#ifdef F_TEST
11411 if (ins(d, "F_TEST", (long)F_TEST)) return -1;
11412#endif
11413
11414 /* constants for futimens */
11415#ifdef UTIME_NOW
11416 if (ins(d, "UTIME_NOW", (long)UTIME_NOW)) return -1;
11417#endif
11418#ifdef UTIME_OMIT
11419 if (ins(d, "UTIME_OMIT", (long)UTIME_OMIT)) return -1;
11420#endif
11421
Guido van Rossum246bc171999-02-01 23:54:31 +000011422#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000011423#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +000011424 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
11425 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
11426 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
11427 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
11428 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
11429 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
11430 if (ins(d, "P_PM", (long)P_PM)) return -1;
11431 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
11432 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
11433 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
11434 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
11435 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
11436 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
11437 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
11438 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
11439 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
11440 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
11441 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
11442 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
11443 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000011444#else
Victor Stinner8c62be82010-05-06 00:08:46 +000011445 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
11446 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
11447 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
11448 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
11449 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +000011450#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000011451#endif
Guido van Rossum246bc171999-02-01 23:54:31 +000011452
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011453#ifdef HAVE_SCHED_H
Jesus Ceaf2cb4e82011-09-09 23:55:42 +020011454 if (ins(d, "SCHED_OTHER", (long)SCHED_OTHER)) return -1;
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011455 if (ins(d, "SCHED_FIFO", (long)SCHED_FIFO)) return -1;
11456 if (ins(d, "SCHED_RR", (long)SCHED_RR)) return -1;
11457#ifdef SCHED_SPORADIC
11458 if (ins(d, "SCHED_SPORADIC", (long)SCHED_SPORADIC) return -1;
11459#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011460#ifdef SCHED_BATCH
11461 if (ins(d, "SCHED_BATCH", (long)SCHED_BATCH)) return -1;
11462#endif
11463#ifdef SCHED_IDLE
11464 if (ins(d, "SCHED_IDLE", (long)SCHED_IDLE)) return -1;
11465#endif
11466#ifdef SCHED_RESET_ON_FORK
11467 if (ins(d, "SCHED_RESET_ON_FORK", (long)SCHED_RESET_ON_FORK)) return -1;
11468#endif
Jesus Ceaf2cb4e82011-09-09 23:55:42 +020011469#ifdef SCHED_SYS
11470 if (ins(d, "SCHED_SYS", (long)SCHED_SYS)) return -1;
11471#endif
11472#ifdef SCHED_IA
11473 if (ins(d, "SCHED_IA", (long)SCHED_IA)) return -1;
11474#endif
11475#ifdef SCHED_FSS
11476 if (ins(d, "SCHED_FSS", (long)SCHED_FSS)) return -1;
11477#endif
11478#ifdef SCHED_FX
11479 if (ins(d, "SCHED_FX", (long)SCHED_FSS)) return -1;
11480#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011481#endif
11482
Benjamin Peterson9428d532011-09-14 11:45:52 -040011483#ifdef USE_XATTRS
Benjamin Peterson799bd802011-08-31 22:15:17 -040011484 if (ins(d, "XATTR_CREATE", (long)XATTR_CREATE)) return -1;
11485 if (ins(d, "XATTR_REPLACE", (long)XATTR_REPLACE)) return -1;
11486 if (ins(d, "XATTR_SIZE_MAX", (long)XATTR_SIZE_MAX)) return -1;
11487#endif
11488
Victor Stinner8b905bd2011-10-25 13:34:04 +020011489#ifdef RTLD_LAZY
11490 if (PyModule_AddIntMacro(d, RTLD_LAZY)) return -1;
11491#endif
11492#ifdef RTLD_NOW
11493 if (PyModule_AddIntMacro(d, RTLD_NOW)) return -1;
11494#endif
11495#ifdef RTLD_GLOBAL
11496 if (PyModule_AddIntMacro(d, RTLD_GLOBAL)) return -1;
11497#endif
11498#ifdef RTLD_LOCAL
11499 if (PyModule_AddIntMacro(d, RTLD_LOCAL)) return -1;
11500#endif
11501#ifdef RTLD_NODELETE
11502 if (PyModule_AddIntMacro(d, RTLD_NODELETE)) return -1;
11503#endif
11504#ifdef RTLD_NOLOAD
11505 if (PyModule_AddIntMacro(d, RTLD_NOLOAD)) return -1;
11506#endif
11507#ifdef RTLD_DEEPBIND
11508 if (PyModule_AddIntMacro(d, RTLD_DEEPBIND)) return -1;
11509#endif
11510
Guido van Rossumd48f2521997-12-05 22:19:34 +000011511#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +000011512 if (insertvalues(d)) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000011513#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011514 return 0;
Barry Warsaw4a342091996-12-19 23:50:02 +000011515}
11516
11517
Tim Peters5aa91602002-01-30 05:46:57 +000011518#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Martin v. Löwis1a214512008-06-11 05:26:20 +000011519#define INITFUNC PyInit_nt
Guido van Rossum0cb96de1997-10-01 04:29:29 +000011520#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +000011521
11522#elif defined(PYOS_OS2)
Martin v. Löwis1a214512008-06-11 05:26:20 +000011523#define INITFUNC PyInit_os2
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000011524#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +000011525
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000011526#else
Martin v. Löwis1a214512008-06-11 05:26:20 +000011527#define INITFUNC PyInit_posix
Guido van Rossum0cb96de1997-10-01 04:29:29 +000011528#define MODNAME "posix"
11529#endif
11530
Martin v. Löwis1a214512008-06-11 05:26:20 +000011531static struct PyModuleDef posixmodule = {
Victor Stinner8c62be82010-05-06 00:08:46 +000011532 PyModuleDef_HEAD_INIT,
11533 MODNAME,
11534 posix__doc__,
11535 -1,
11536 posix_methods,
11537 NULL,
11538 NULL,
11539 NULL,
11540 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +000011541};
11542
11543
Mark Hammondfe51c6d2002-08-02 02:27:13 +000011544PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +000011545INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +000011546{
Victor Stinner8c62be82010-05-06 00:08:46 +000011547 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +000011548
Brian Curtin52173d42010-12-02 18:29:18 +000011549#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000011550 win32_can_symlink = enable_symlink();
Brian Curtin52173d42010-12-02 18:29:18 +000011551#endif
11552
Victor Stinner8c62be82010-05-06 00:08:46 +000011553 m = PyModule_Create(&posixmodule);
11554 if (m == NULL)
11555 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +000011556
Victor Stinner8c62be82010-05-06 00:08:46 +000011557 /* Initialize environ dictionary */
11558 v = convertenviron();
11559 Py_XINCREF(v);
11560 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
11561 return NULL;
11562 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +000011563
Victor Stinner8c62be82010-05-06 00:08:46 +000011564 if (all_ins(m))
11565 return NULL;
Barry Warsaw4a342091996-12-19 23:50:02 +000011566
Victor Stinner8c62be82010-05-06 00:08:46 +000011567 if (setup_confname_tables(m))
11568 return NULL;
Fred Drakebec628d1999-12-15 18:31:10 +000011569
Victor Stinner8c62be82010-05-06 00:08:46 +000011570 Py_INCREF(PyExc_OSError);
11571 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +000011572
Benjamin Peterson2740af82011-08-02 17:41:34 -050011573#ifdef HAVE_SCHED_SETAFFINITY
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011574 if (PyType_Ready(&cpu_set_type) < 0)
11575 return NULL;
11576 Py_INCREF(&cpu_set_type);
11577 PyModule_AddObject(m, "cpu_set", (PyObject *)&cpu_set_type);
Benjamin Peterson2740af82011-08-02 17:41:34 -050011578#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011579
Guido van Rossumb3d39562000-01-31 18:41:26 +000011580#ifdef HAVE_PUTENV
Victor Stinner8c62be82010-05-06 00:08:46 +000011581 if (posix_putenv_garbage == NULL)
11582 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +000011583#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +000011584
Victor Stinner8c62be82010-05-06 00:08:46 +000011585 if (!initialized) {
Ross Lagerwall7807c352011-03-17 20:20:30 +020011586#if defined(HAVE_WAITID) && !defined(__APPLE__)
11587 waitid_result_desc.name = MODNAME ".waitid_result";
11588 PyStructSequence_InitType(&WaitidResultType, &waitid_result_desc);
11589#endif
11590
Victor Stinner8c62be82010-05-06 00:08:46 +000011591 stat_result_desc.name = MODNAME ".stat_result";
11592 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
11593 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
11594 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
11595 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
11596 structseq_new = StatResultType.tp_new;
11597 StatResultType.tp_new = statresult_new;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011598
Victor Stinner8c62be82010-05-06 00:08:46 +000011599 statvfs_result_desc.name = MODNAME ".statvfs_result";
11600 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011601#ifdef NEED_TICKS_PER_SECOND
11602# if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
Victor Stinner8c62be82010-05-06 00:08:46 +000011603 ticks_per_second = sysconf(_SC_CLK_TCK);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011604# elif defined(HZ)
Victor Stinner8c62be82010-05-06 00:08:46 +000011605 ticks_per_second = HZ;
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011606# else
Victor Stinner8c62be82010-05-06 00:08:46 +000011607 ticks_per_second = 60; /* magic fallback value; may be bogus */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011608# endif
11609#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011610
Benjamin Peterson0163c9a2011-08-02 18:11:38 -050011611#if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER)
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011612 sched_param_desc.name = MODNAME ".sched_param";
11613 PyStructSequence_InitType(&SchedParamType, &sched_param_desc);
11614 SchedParamType.tp_new = sched_param_new;
11615#endif
Antoine Pitroubcf2b592012-02-08 23:28:36 +010011616
11617 /* initialize TerminalSize_info */
11618 PyStructSequence_InitType(&TerminalSizeType, &TerminalSize_desc);
11619 Py_INCREF(&TerminalSizeType);
Victor Stinner8c62be82010-05-06 00:08:46 +000011620 }
Ross Lagerwall7807c352011-03-17 20:20:30 +020011621#if defined(HAVE_WAITID) && !defined(__APPLE__)
11622 Py_INCREF((PyObject*) &WaitidResultType);
11623 PyModule_AddObject(m, "waitid_result", (PyObject*) &WaitidResultType);
11624#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011625 Py_INCREF((PyObject*) &StatResultType);
11626 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
11627 Py_INCREF((PyObject*) &StatVFSResultType);
11628 PyModule_AddObject(m, "statvfs_result",
11629 (PyObject*) &StatVFSResultType);
Benjamin Petersone3298dd2011-08-02 18:40:46 -050011630
11631#if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER)
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011632 Py_INCREF(&SchedParamType);
11633 PyModule_AddObject(m, "sched_param", (PyObject *)&SchedParamType);
Benjamin Petersone3298dd2011-08-02 18:40:46 -050011634#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011635 initialized = 1;
Thomas Wouters477c8d52006-05-27 19:21:47 +000011636
11637#ifdef __APPLE__
Victor Stinner8c62be82010-05-06 00:08:46 +000011638 /*
11639 * Step 2 of weak-linking support on Mac OS X.
11640 *
11641 * The code below removes functions that are not available on the
11642 * currently active platform.
11643 *
11644 * This block allow one to use a python binary that was build on
11645 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
11646 * OSX 10.4.
11647 */
Thomas Wouters477c8d52006-05-27 19:21:47 +000011648#ifdef HAVE_FSTATVFS
Victor Stinner8c62be82010-05-06 00:08:46 +000011649 if (fstatvfs == NULL) {
11650 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
11651 return NULL;
11652 }
11653 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011654#endif /* HAVE_FSTATVFS */
11655
11656#ifdef HAVE_STATVFS
Victor Stinner8c62be82010-05-06 00:08:46 +000011657 if (statvfs == NULL) {
11658 if (PyObject_DelAttrString(m, "statvfs") == -1) {
11659 return NULL;
11660 }
11661 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011662#endif /* HAVE_STATVFS */
11663
11664# ifdef HAVE_LCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000011665 if (lchown == NULL) {
11666 if (PyObject_DelAttrString(m, "lchown") == -1) {
11667 return NULL;
11668 }
11669 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011670#endif /* HAVE_LCHOWN */
11671
11672
11673#endif /* __APPLE__ */
Antoine Pitroubcf2b592012-02-08 23:28:36 +010011674
11675 PyModule_AddObject(m, "terminal_size", (PyObject*) &TerminalSizeType);
11676
Larry Hastings6fe20b32012-04-19 15:07:49 -070011677 billion = PyLong_FromLong(1000000000);
11678 if (!billion)
11679 return NULL;
11680
Victor Stinner8c62be82010-05-06 00:08:46 +000011681 return m;
Thomas Wouters477c8d52006-05-27 19:21:47 +000011682
Guido van Rossumb6775db1994-08-01 11:34:53 +000011683}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011684
11685#ifdef __cplusplus
11686}
11687#endif