blob: 8b2b2117cb1dbd9c7e286dd36f95ed57d39f8a7e [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 Peterson9428d532011-09-14 11:45:52 -0400110#if defined(HAVE_SYS_XATTR_H) && defined(__GLIBC__)
111#define USE_XATTRS
112#endif
113
114#ifdef USE_XATTRS
Benjamin Petersonb77fe172011-09-13 17:20:47 -0400115#include <sys/xattr.h>
Benjamin Peterson799bd802011-08-31 22:15:17 -0400116#endif
117
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000118#if defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__)
119#ifdef HAVE_SYS_SOCKET_H
120#include <sys/socket.h>
121#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000122#endif
123
Victor Stinner8b905bd2011-10-25 13:34:04 +0200124#ifdef HAVE_DLFCN_H
125#include <dlfcn.h>
126#endif
127
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000128/* Various compilers have only certain posix functions */
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +0000129/* XXX Gosh I wish these were all moved into pyconfig.h */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000130#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000131#include <process.h>
132#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000133#if defined(__WATCOMC__) && !defined(__QNX__) /* Watcom compiler */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000134#define HAVE_GETCWD 1
135#define HAVE_OPENDIR 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000136#define HAVE_SYSTEM 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000137#if defined(__OS2__)
138#define HAVE_EXECV 1
139#define HAVE_WAIT 1
Guido van Rossumad0ee831995-03-01 10:34:45 +0000140#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000141#include <process.h>
142#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000143#ifdef __BORLANDC__ /* Borland compiler */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000144#define HAVE_EXECV 1
145#define HAVE_GETCWD 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000146#define HAVE_OPENDIR 1
147#define HAVE_PIPE 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000148#define HAVE_SYSTEM 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000149#define HAVE_WAIT 1
150#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000151#ifdef _MSC_VER /* Microsoft compiler */
Guido van Rossum8d665e61996-06-26 18:22:49 +0000152#define HAVE_GETCWD 1
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +0000153#define HAVE_GETPPID 1
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000154#define HAVE_GETLOGIN 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000155#define HAVE_SPAWNV 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000156#define HAVE_EXECV 1
157#define HAVE_PIPE 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000158#define HAVE_SYSTEM 1
159#define HAVE_CWAIT 1
160#define HAVE_FSYNC 1
Tim Peters11b23062003-04-23 02:39:17 +0000161#define fsync _commit
Andrew MacIntyre6c73af22002-03-03 03:07:07 +0000162#else
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000163#if defined(PYOS_OS2) && defined(PYCC_GCC) || defined(__VMS)
164/* Everything needed is defined in PC/os2emx/pyconfig.h or vms/pyconfig.h */
Victor Stinner8c62be82010-05-06 00:08:46 +0000165#else /* all other compilers */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000166/* Unix functions that the configure script doesn't check for */
167#define HAVE_EXECV 1
168#define HAVE_FORK 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000169#if defined(__USLC__) && defined(__SCO_VERSION__) /* SCO UDK Compiler */
Guido van Rossum2242f2f2001-04-11 20:58:20 +0000170#define HAVE_FORK1 1
171#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000172#define HAVE_GETCWD 1
173#define HAVE_GETEGID 1
174#define HAVE_GETEUID 1
175#define HAVE_GETGID 1
176#define HAVE_GETPPID 1
177#define HAVE_GETUID 1
178#define HAVE_KILL 1
179#define HAVE_OPENDIR 1
180#define HAVE_PIPE 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000181#define HAVE_SYSTEM 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000182#define HAVE_WAIT 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000183#define HAVE_TTYNAME 1
Martin v. Löwis7a924e62003-03-05 14:15:21 +0000184#endif /* PYOS_OS2 && PYCC_GCC && __VMS */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000185#endif /* _MSC_VER */
186#endif /* __BORLANDC__ */
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000187#endif /* ! __WATCOMC__ || __QNX__ */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000188#endif /* ! __IBMC__ */
Guido van Rossumad0ee831995-03-01 10:34:45 +0000189
Victor Stinnera2f7c002012-02-08 03:36:25 +0100190
191
192
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000193#ifndef _MSC_VER
Guido van Rossum36bc6801995-06-14 22:54:23 +0000194
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000195#if defined(__sgi)&&_COMPILER_VERSION>=700
196/* declare ctermid_r if compiling with MIPSPro 7.x in ANSI C mode
197 (default) */
198extern char *ctermid_r(char *);
199#endif
200
Thomas Wouters1e0c2f42000-07-24 16:06:23 +0000201#ifndef HAVE_UNISTD_H
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000202#if defined(PYCC_VACPP)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000203extern int mkdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000204#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000205#if ( defined(__WATCOMC__) || defined(_MSC_VER) ) && !defined(__QNX__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000206extern int mkdir(const char *);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000207#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000208extern int mkdir(const char *, mode_t);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000209#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000210#endif
211#if defined(__IBMC__) || defined(__IBMCPP__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000212extern int chdir(char *);
213extern int rmdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000214#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000215extern int chdir(const char *);
216extern int rmdir(const char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000217#endif
Tim Peters58e0a8c2001-05-14 22:32:33 +0000218#ifdef __BORLANDC__
219extern int chmod(const char *, int);
220#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000221extern int chmod(const char *, mode_t);
Tim Peters58e0a8c2001-05-14 22:32:33 +0000222#endif
Christian Heimes4e30a842007-11-30 22:12:06 +0000223/*#ifdef HAVE_FCHMOD
224extern int fchmod(int, mode_t);
225#endif*/
226/*#ifdef HAVE_LCHMOD
227extern int lchmod(const char *, mode_t);
228#endif*/
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000229extern int chown(const char *, uid_t, gid_t);
230extern char *getcwd(char *, int);
231extern char *strerror(int);
232extern int link(const char *, const char *);
233extern int rename(const char *, const char *);
234extern int stat(const char *, struct stat *);
235extern int unlink(const char *);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000236#ifdef HAVE_SYMLINK
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000237extern int symlink(const char *, const char *);
Guido van Rossuma38a5031995-02-17 15:11:36 +0000238#endif /* HAVE_SYMLINK */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000239#ifdef HAVE_LSTAT
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000240extern int lstat(const char *, struct stat *);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000241#endif /* HAVE_LSTAT */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000242#endif /* !HAVE_UNISTD_H */
Guido van Rossum36bc6801995-06-14 22:54:23 +0000243
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000244#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000245
Guido van Rossumb6775db1994-08-01 11:34:53 +0000246#ifdef HAVE_UTIME_H
247#include <utime.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000248#endif /* HAVE_UTIME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000249
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000250#ifdef HAVE_SYS_UTIME_H
251#include <sys/utime.h>
252#define HAVE_UTIME_H /* pretend we do for the rest of this file */
253#endif /* HAVE_SYS_UTIME_H */
254
Guido van Rossumb6775db1994-08-01 11:34:53 +0000255#ifdef HAVE_SYS_TIMES_H
256#include <sys/times.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000257#endif /* HAVE_SYS_TIMES_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000258
259#ifdef HAVE_SYS_PARAM_H
260#include <sys/param.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000261#endif /* HAVE_SYS_PARAM_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000262
263#ifdef HAVE_SYS_UTSNAME_H
264#include <sys/utsname.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000265#endif /* HAVE_SYS_UTSNAME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000266
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000267#ifdef HAVE_DIRENT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000268#include <dirent.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000269#define NAMLEN(dirent) strlen((dirent)->d_name)
270#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000271#if defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000272#include <direct.h>
273#define NAMLEN(dirent) strlen((dirent)->d_name)
274#else
Guido van Rossumb6775db1994-08-01 11:34:53 +0000275#define dirent direct
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000276#define NAMLEN(dirent) (dirent)->d_namlen
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000277#endif
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000278#ifdef HAVE_SYS_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000279#include <sys/ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000280#endif
281#ifdef HAVE_SYS_DIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000282#include <sys/dir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000283#endif
284#ifdef HAVE_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000285#include <ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000286#endif
287#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000288
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000289#ifdef _MSC_VER
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000290#ifdef HAVE_DIRECT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000291#include <direct.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000292#endif
293#ifdef HAVE_IO_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000294#include <io.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000295#endif
296#ifdef HAVE_PROCESS_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000297#include <process.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000298#endif
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000299#ifndef VOLUME_NAME_DOS
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +0000300#define VOLUME_NAME_DOS 0x0
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000301#endif
302#ifndef VOLUME_NAME_NT
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +0000303#define VOLUME_NAME_NT 0x2
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000304#endif
305#ifndef IO_REPARSE_TAG_SYMLINK
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +0000306#define IO_REPARSE_TAG_SYMLINK (0xA000000CL)
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000307#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000308#include "osdefs.h"
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000309#include <malloc.h>
Guido van Rossumb6775db1994-08-01 11:34:53 +0000310#include <windows.h>
Victor Stinner8c62be82010-05-06 00:08:46 +0000311#include <shellapi.h> /* for ShellExecute() */
Brian Curtine8e4b3b2010-09-23 20:04:14 +0000312#include <lmcons.h> /* for UNLEN */
Brian Curtin52173d42010-12-02 18:29:18 +0000313#ifdef SE_CREATE_SYMBOLIC_LINK_NAME /* Available starting with Vista */
314#define HAVE_SYMLINK
Brian Curtin3b4499c2010-12-28 14:31:47 +0000315static int win32_can_symlink = 0;
Brian Curtin52173d42010-12-02 18:29:18 +0000316#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000317#endif /* _MSC_VER */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000318
Guido van Rossumd48f2521997-12-05 22:19:34 +0000319#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000320#include <io.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000321#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000322
Tim Petersbc2e10e2002-03-03 23:17:02 +0000323#ifndef MAXPATHLEN
Thomas Wouters477c8d52006-05-27 19:21:47 +0000324#if defined(PATH_MAX) && PATH_MAX > 1024
325#define MAXPATHLEN PATH_MAX
326#else
Tim Petersbc2e10e2002-03-03 23:17:02 +0000327#define MAXPATHLEN 1024
Thomas Wouters477c8d52006-05-27 19:21:47 +0000328#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000329#endif /* MAXPATHLEN */
330
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000331#ifdef UNION_WAIT
332/* Emulate some macros on systems that have a union instead of macros */
333
334#ifndef WIFEXITED
335#define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
336#endif
337
338#ifndef WEXITSTATUS
339#define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
340#endif
341
342#ifndef WTERMSIG
343#define WTERMSIG(u_wait) ((u_wait).w_termsig)
344#endif
345
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000346#define WAIT_TYPE union wait
347#define WAIT_STATUS_INT(s) (s.w_status)
348
349#else /* !UNION_WAIT */
350#define WAIT_TYPE int
351#define WAIT_STATUS_INT(s) (s)
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000352#endif /* UNION_WAIT */
353
Greg Wardb48bc172000-03-01 21:51:56 +0000354/* Don't use the "_r" form if we don't need it (also, won't have a
355 prototype for it, at least on Solaris -- maybe others as well?). */
356#if defined(HAVE_CTERMID_R) && defined(WITH_THREAD)
357#define USE_CTERMID_R
358#endif
359
Fred Drake699f3522000-06-29 21:12:41 +0000360/* choose the appropriate stat and fstat functions and return structs */
Guido van Rossum64529cd2000-06-30 22:45:12 +0000361#undef STAT
Antoine Pitroue47e0932011-01-19 15:21:35 +0000362#undef FSTAT
363#undef STRUCT_STAT
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000364#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +0000365# define STAT win32_stat
366# define FSTAT win32_fstat
367# define STRUCT_STAT struct win32_stat
Fred Drake699f3522000-06-29 21:12:41 +0000368#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000369# define STAT stat
370# define FSTAT fstat
371# define STRUCT_STAT struct stat
Fred Drake699f3522000-06-29 21:12:41 +0000372#endif
373
Tim Peters11b23062003-04-23 02:39:17 +0000374#if defined(MAJOR_IN_MKDEV)
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000375#include <sys/mkdev.h>
376#else
377#if defined(MAJOR_IN_SYSMACROS)
378#include <sys/sysmacros.h>
379#endif
Neal Norwitz3d949422002-04-20 13:46:43 +0000380#if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
381#include <sys/mkdev.h>
382#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000383#endif
Fred Drake699f3522000-06-29 21:12:41 +0000384
Ross Lagerwallb1e5d592011-09-19 08:30:43 +0200385/* A helper used by a number of POSIX-only functions */
386#ifndef MS_WINDOWS
Georg Brandla391b112011-02-25 15:23:18 +0000387static int
388_parse_off_t(PyObject* arg, void* addr)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000389{
390#if !defined(HAVE_LARGEFILE_SUPPORT)
391 *((off_t*)addr) = PyLong_AsLong(arg);
392#else
Antoine Pitroudcc20b82011-02-26 13:38:35 +0000393 *((off_t*)addr) = PyLong_AsLongLong(arg);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000394#endif
395 if (PyErr_Occurred())
396 return 0;
397 return 1;
398}
Ross Lagerwallb1e5d592011-09-19 08:30:43 +0200399#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000400
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000401#if defined _MSC_VER && _MSC_VER >= 1400
402/* Microsoft CRT in VS2005 and higher will verify that a filehandle is
403 * valid and throw an assertion if it isn't.
404 * Normally, an invalid fd is likely to be a C program error and therefore
405 * an assertion can be useful, but it does contradict the POSIX standard
406 * which for write(2) states:
407 * "Otherwise, -1 shall be returned and errno set to indicate the error."
408 * "[EBADF] The fildes argument is not a valid file descriptor open for
409 * writing."
410 * Furthermore, python allows the user to enter any old integer
411 * as a fd and should merely raise a python exception on error.
412 * The Microsoft CRT doesn't provide an official way to check for the
413 * validity of a file descriptor, but we can emulate its internal behaviour
Victor Stinner8c62be82010-05-06 00:08:46 +0000414 * by using the exported __pinfo data member and knowledge of the
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000415 * internal structures involved.
416 * The structures below must be updated for each version of visual studio
417 * according to the file internal.h in the CRT source, until MS comes
418 * up with a less hacky way to do this.
419 * (all of this is to avoid globally modifying the CRT behaviour using
420 * _set_invalid_parameter_handler() and _CrtSetReportMode())
421 */
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000422/* The actual size of the structure is determined at runtime.
423 * Only the first items must be present.
424 */
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000425typedef struct {
Victor Stinner8c62be82010-05-06 00:08:46 +0000426 intptr_t osfhnd;
427 char osfile;
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000428} my_ioinfo;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000429
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000430extern __declspec(dllimport) char * __pioinfo[];
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000431#define IOINFO_L2E 5
432#define IOINFO_ARRAY_ELTS (1 << IOINFO_L2E)
433#define IOINFO_ARRAYS 64
434#define _NHANDLE_ (IOINFO_ARRAYS * IOINFO_ARRAY_ELTS)
435#define FOPEN 0x01
436#define _NO_CONSOLE_FILENO (intptr_t)-2
437
438/* This function emulates what the windows CRT does to validate file handles */
439int
440_PyVerify_fd(int fd)
441{
Victor Stinner8c62be82010-05-06 00:08:46 +0000442 const int i1 = fd >> IOINFO_L2E;
443 const int i2 = fd & ((1 << IOINFO_L2E) - 1);
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000444
Antoine Pitrou22e41552010-08-15 18:07:50 +0000445 static size_t sizeof_ioinfo = 0;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000446
Victor Stinner8c62be82010-05-06 00:08:46 +0000447 /* Determine the actual size of the ioinfo structure,
448 * as used by the CRT loaded in memory
449 */
450 if (sizeof_ioinfo == 0 && __pioinfo[0] != NULL) {
451 sizeof_ioinfo = _msize(__pioinfo[0]) / IOINFO_ARRAY_ELTS;
452 }
453 if (sizeof_ioinfo == 0) {
454 /* This should not happen... */
455 goto fail;
456 }
457
458 /* See that it isn't a special CLEAR fileno */
459 if (fd != _NO_CONSOLE_FILENO) {
460 /* Microsoft CRT would check that 0<=fd<_nhandle but we can't do that. Instead
461 * we check pointer validity and other info
462 */
463 if (0 <= i1 && i1 < IOINFO_ARRAYS && __pioinfo[i1] != NULL) {
464 /* finally, check that the file is open */
465 my_ioinfo* info = (my_ioinfo*)(__pioinfo[i1] + i2 * sizeof_ioinfo);
466 if (info->osfile & FOPEN) {
467 return 1;
468 }
469 }
470 }
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000471 fail:
Victor Stinner8c62be82010-05-06 00:08:46 +0000472 errno = EBADF;
473 return 0;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000474}
475
476/* the special case of checking dup2. The target fd must be in a sensible range */
477static int
478_PyVerify_fd_dup2(int fd1, int fd2)
479{
Victor Stinner8c62be82010-05-06 00:08:46 +0000480 if (!_PyVerify_fd(fd1))
481 return 0;
482 if (fd2 == _NO_CONSOLE_FILENO)
483 return 0;
484 if ((unsigned)fd2 < _NHANDLE_)
485 return 1;
486 else
487 return 0;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000488}
489#else
490/* dummy version. _PyVerify_fd() is already defined in fileobject.h */
491#define _PyVerify_fd_dup2(A, B) (1)
492#endif
493
Brian Curtinfc1be6d2010-11-24 13:23:18 +0000494#ifdef MS_WINDOWS
Brian Curtinf5e76d02010-11-24 13:14:05 +0000495/* The following structure was copied from
496 http://msdn.microsoft.com/en-us/library/ms791514.aspx as the required
497 include doesn't seem to be present in the Windows SDK (at least as included
498 with Visual Studio Express). */
499typedef struct _REPARSE_DATA_BUFFER {
500 ULONG ReparseTag;
501 USHORT ReparseDataLength;
502 USHORT Reserved;
503 union {
504 struct {
505 USHORT SubstituteNameOffset;
506 USHORT SubstituteNameLength;
507 USHORT PrintNameOffset;
508 USHORT PrintNameLength;
509 ULONG Flags;
510 WCHAR PathBuffer[1];
511 } SymbolicLinkReparseBuffer;
512
513 struct {
514 USHORT SubstituteNameOffset;
515 USHORT SubstituteNameLength;
516 USHORT PrintNameOffset;
517 USHORT PrintNameLength;
518 WCHAR PathBuffer[1];
519 } MountPointReparseBuffer;
520
521 struct {
522 UCHAR DataBuffer[1];
523 } GenericReparseBuffer;
524 };
525} REPARSE_DATA_BUFFER, *PREPARSE_DATA_BUFFER;
526
527#define REPARSE_DATA_BUFFER_HEADER_SIZE FIELD_OFFSET(REPARSE_DATA_BUFFER,\
528 GenericReparseBuffer)
529#define MAXIMUM_REPARSE_DATA_BUFFER_SIZE ( 16 * 1024 )
530
531static int
Brian Curtind25aef52011-06-13 15:16:04 -0500532win32_get_reparse_tag(HANDLE reparse_point_handle, ULONG *reparse_tag)
Brian Curtinf5e76d02010-11-24 13:14:05 +0000533{
534 char target_buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
535 REPARSE_DATA_BUFFER *rdb = (REPARSE_DATA_BUFFER *)target_buffer;
536 DWORD n_bytes_returned;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000537
538 if (0 == DeviceIoControl(
539 reparse_point_handle,
540 FSCTL_GET_REPARSE_POINT,
541 NULL, 0, /* in buffer */
542 target_buffer, sizeof(target_buffer),
543 &n_bytes_returned,
544 NULL)) /* we're not using OVERLAPPED_IO */
Brian Curtind25aef52011-06-13 15:16:04 -0500545 return FALSE;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000546
547 if (reparse_tag)
548 *reparse_tag = rdb->ReparseTag;
549
Brian Curtind25aef52011-06-13 15:16:04 -0500550 return TRUE;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000551}
Victor Stinner1ab6c2d2011-11-15 22:27:41 +0100552
553static int
554win32_warn_bytes_api()
555{
556 return PyErr_WarnEx(PyExc_DeprecationWarning,
557 "The Windows bytes API has been deprecated, "
558 "use Unicode filenames instead",
559 1);
560}
561
562static PyObject*
563win32_decode_filename(PyObject *obj)
564{
565 PyObject *unicode;
566 if (PyUnicode_Check(obj)) {
567 if (PyUnicode_READY(obj))
568 return NULL;
569 Py_INCREF(obj);
570 return obj;
571 }
572 if (!PyUnicode_FSDecoder(obj, &unicode))
573 return NULL;
574 if (win32_warn_bytes_api()) {
575 Py_DECREF(unicode);
576 return NULL;
577 }
578 return unicode;
579}
Brian Curtinfc1be6d2010-11-24 13:23:18 +0000580#endif /* MS_WINDOWS */
Brian Curtinf5e76d02010-11-24 13:14:05 +0000581
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000582/* Return a dictionary corresponding to the POSIX environment table */
Jack Jansenea0c3822002-08-01 21:57:49 +0000583#ifdef WITH_NEXT_FRAMEWORK
584/* On Darwin/MacOSX a shared library or framework has no access to
585** environ directly, we must obtain it with _NSGetEnviron().
586*/
587#include <crt_externs.h>
588static char **environ;
589#elif !defined(_MSC_VER) && ( !defined(__WATCOMC__) || defined(__QNX__) )
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000590extern char **environ;
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000591#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000592
Barry Warsaw53699e91996-12-10 23:23:01 +0000593static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000594convertenviron(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000595{
Victor Stinner8c62be82010-05-06 00:08:46 +0000596 PyObject *d;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000597#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +0000598 wchar_t **e;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000599#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000600 char **e;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000601#endif
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +0000602#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +0000603 APIRET rc;
604 char buffer[1024]; /* OS/2 Provides a Documented Max of 1024 Chars */
605#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +0000606
Victor Stinner8c62be82010-05-06 00:08:46 +0000607 d = PyDict_New();
608 if (d == NULL)
609 return NULL;
610#ifdef WITH_NEXT_FRAMEWORK
611 if (environ == NULL)
612 environ = *_NSGetEnviron();
613#endif
614#ifdef MS_WINDOWS
615 /* _wenviron must be initialized in this way if the program is started
616 through main() instead of wmain(). */
617 _wgetenv(L"");
618 if (_wenviron == NULL)
619 return d;
620 /* This part ignores errors */
621 for (e = _wenviron; *e != NULL; e++) {
622 PyObject *k;
623 PyObject *v;
624 wchar_t *p = wcschr(*e, L'=');
625 if (p == NULL)
626 continue;
627 k = PyUnicode_FromWideChar(*e, (Py_ssize_t)(p-*e));
628 if (k == NULL) {
629 PyErr_Clear();
630 continue;
Guido van Rossumd48f2521997-12-05 22:19:34 +0000631 }
Victor Stinner8c62be82010-05-06 00:08:46 +0000632 v = PyUnicode_FromWideChar(p+1, wcslen(p+1));
633 if (v == NULL) {
634 PyErr_Clear();
635 Py_DECREF(k);
636 continue;
Guido van Rossumd48f2521997-12-05 22:19:34 +0000637 }
Victor Stinner8c62be82010-05-06 00:08:46 +0000638 if (PyDict_GetItem(d, k) == NULL) {
639 if (PyDict_SetItem(d, k, v) != 0)
640 PyErr_Clear();
641 }
642 Py_DECREF(k);
643 Py_DECREF(v);
644 }
645#else
646 if (environ == NULL)
647 return d;
648 /* This part ignores errors */
649 for (e = environ; *e != NULL; e++) {
650 PyObject *k;
651 PyObject *v;
652 char *p = strchr(*e, '=');
653 if (p == NULL)
654 continue;
Victor Stinner84ae1182010-05-06 22:05:07 +0000655 k = PyBytes_FromStringAndSize(*e, (int)(p-*e));
Victor Stinner8c62be82010-05-06 00:08:46 +0000656 if (k == NULL) {
657 PyErr_Clear();
658 continue;
659 }
Victor Stinner84ae1182010-05-06 22:05:07 +0000660 v = PyBytes_FromStringAndSize(p+1, strlen(p+1));
Victor Stinner8c62be82010-05-06 00:08:46 +0000661 if (v == NULL) {
662 PyErr_Clear();
663 Py_DECREF(k);
664 continue;
665 }
666 if (PyDict_GetItem(d, k) == NULL) {
667 if (PyDict_SetItem(d, k, v) != 0)
668 PyErr_Clear();
669 }
670 Py_DECREF(k);
671 Py_DECREF(v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000672 }
673#endif
Victor Stinner8c62be82010-05-06 00:08:46 +0000674#if defined(PYOS_OS2)
675 rc = DosQueryExtLIBPATH(buffer, BEGIN_LIBPATH);
676 if (rc == NO_ERROR) { /* (not a type, envname is NOT 'BEGIN_LIBPATH') */
677 PyObject *v = PyBytes_FromString(buffer);
678 PyDict_SetItemString(d, "BEGINLIBPATH", v);
679 Py_DECREF(v);
680 }
681 rc = DosQueryExtLIBPATH(buffer, END_LIBPATH);
682 if (rc == NO_ERROR) { /* (not a typo, envname is NOT 'END_LIBPATH') */
683 PyObject *v = PyBytes_FromString(buffer);
684 PyDict_SetItemString(d, "ENDLIBPATH", v);
685 Py_DECREF(v);
686 }
687#endif
688 return d;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000689}
690
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000691/* Set a POSIX-specific error from errno, and return NULL */
692
Barry Warsawd58d7641998-07-23 16:14:40 +0000693static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000694posix_error(void)
Guido van Rossumad0ee831995-03-01 10:34:45 +0000695{
Victor Stinner8c62be82010-05-06 00:08:46 +0000696 return PyErr_SetFromErrno(PyExc_OSError);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000697}
Barry Warsawd58d7641998-07-23 16:14:40 +0000698static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000699posix_error_with_filename(char* name)
Barry Warsawd58d7641998-07-23 16:14:40 +0000700{
Victor Stinner8c62be82010-05-06 00:08:46 +0000701 return PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
Barry Warsawd58d7641998-07-23 16:14:40 +0000702}
703
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000704
Mark Hammondef8b6542001-05-13 08:04:26 +0000705static PyObject *
Martin v. Löwis011e8422009-05-05 04:43:17 +0000706posix_error_with_allocated_filename(PyObject* name)
Mark Hammondef8b6542001-05-13 08:04:26 +0000707{
Victor Stinner77ccd6d2010-05-08 00:36:42 +0000708 PyObject *name_str, *rc;
709 name_str = PyUnicode_DecodeFSDefaultAndSize(PyBytes_AsString(name),
710 PyBytes_GET_SIZE(name));
Victor Stinner8c62be82010-05-06 00:08:46 +0000711 Py_DECREF(name);
Victor Stinner77ccd6d2010-05-08 00:36:42 +0000712 rc = PyErr_SetFromErrnoWithFilenameObject(PyExc_OSError,
713 name_str);
714 Py_XDECREF(name_str);
Victor Stinner8c62be82010-05-06 00:08:46 +0000715 return rc;
Mark Hammondef8b6542001-05-13 08:04:26 +0000716}
717
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000718#ifdef MS_WINDOWS
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000719static PyObject *
Brian Curtind40e6f72010-07-08 21:39:08 +0000720win32_error(char* function, const char* filename)
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000721{
Victor Stinner8c62be82010-05-06 00:08:46 +0000722 /* XXX We should pass the function name along in the future.
723 (winreg.c also wants to pass the function name.)
724 This would however require an additional param to the
725 Windows error object, which is non-trivial.
726 */
727 errno = GetLastError();
728 if (filename)
729 return PyErr_SetFromWindowsErrWithFilename(errno, filename);
730 else
731 return PyErr_SetFromWindowsErr(errno);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000732}
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000733
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000734static PyObject *
Victor Stinnereb5657a2011-09-30 01:44:27 +0200735win32_error_unicode(char* function, wchar_t* filename)
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000736{
Victor Stinner8c62be82010-05-06 00:08:46 +0000737 /* XXX - see win32_error for comments on 'function' */
738 errno = GetLastError();
739 if (filename)
740 return PyErr_SetFromWindowsErrWithUnicodeFilename(errno, filename);
741 else
742 return PyErr_SetFromWindowsErr(errno);
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000743}
744
Victor Stinnereb5657a2011-09-30 01:44:27 +0200745static PyObject *
746win32_error_object(char* function, PyObject* filename)
747{
748 /* XXX - see win32_error for comments on 'function' */
749 errno = GetLastError();
750 if (filename)
751 return PyErr_SetExcFromWindowsErrWithFilenameObject(
752 PyExc_WindowsError,
753 errno,
754 filename);
755 else
756 return PyErr_SetFromWindowsErr(errno);
757}
758
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000759#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000760
Guido van Rossumd48f2521997-12-05 22:19:34 +0000761#if defined(PYOS_OS2)
762/**********************************************************************
763 * Helper Function to Trim and Format OS/2 Messages
764 **********************************************************************/
Victor Stinner8c62be82010-05-06 00:08:46 +0000765static void
Guido van Rossumd48f2521997-12-05 22:19:34 +0000766os2_formatmsg(char *msgbuf, int msglen, char *reason)
767{
768 msgbuf[msglen] = '\0'; /* OS/2 Doesn't Guarantee a Terminator */
769
770 if (strlen(msgbuf) > 0) { /* If Non-Empty Msg, Trim CRLF */
771 char *lastc = &msgbuf[ strlen(msgbuf)-1 ];
772
Neal Norwitz30b5c5d2005-12-19 06:05:18 +0000773 while (lastc > msgbuf && isspace(Py_CHARMASK(*lastc)))
Guido van Rossumd48f2521997-12-05 22:19:34 +0000774 *lastc-- = '\0'; /* Trim Trailing Whitespace (CRLF) */
775 }
776
777 /* Add Optional Reason Text */
778 if (reason) {
779 strcat(msgbuf, " : ");
780 strcat(msgbuf, reason);
781 }
782}
783
784/**********************************************************************
785 * Decode an OS/2 Operating System Error Code
786 *
787 * A convenience function to lookup an OS/2 error code and return a
788 * text message we can use to raise a Python exception.
789 *
790 * Notes:
791 * The messages for errors returned from the OS/2 kernel reside in
792 * the file OSO001.MSG in the \OS2 directory hierarchy.
793 *
794 **********************************************************************/
Victor Stinner8c62be82010-05-06 00:08:46 +0000795static char *
Guido van Rossumd48f2521997-12-05 22:19:34 +0000796os2_strerror(char *msgbuf, int msgbuflen, int errorcode, char *reason)
797{
798 APIRET rc;
799 ULONG msglen;
800
801 /* Retrieve Kernel-Related Error Message from OSO001.MSG File */
802 Py_BEGIN_ALLOW_THREADS
803 rc = DosGetMessage(NULL, 0, msgbuf, msgbuflen,
804 errorcode, "oso001.msg", &msglen);
805 Py_END_ALLOW_THREADS
806
807 if (rc == NO_ERROR)
808 os2_formatmsg(msgbuf, msglen, reason);
809 else
Tim Peters1ceb5fb2001-11-28 20:32:57 +0000810 PyOS_snprintf(msgbuf, msgbuflen,
Victor Stinner8c62be82010-05-06 00:08:46 +0000811 "unknown OS error #%d", errorcode);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000812
813 return msgbuf;
814}
815
816/* Set an OS/2-specific error and return NULL. OS/2 kernel
817 errors are not in a global variable e.g. 'errno' nor are
818 they congruent with posix error numbers. */
819
Victor Stinner8c62be82010-05-06 00:08:46 +0000820static PyObject *
821os2_error(int code)
Guido van Rossumd48f2521997-12-05 22:19:34 +0000822{
823 char text[1024];
824 PyObject *v;
825
826 os2_strerror(text, sizeof(text), code, "");
827
828 v = Py_BuildValue("(is)", code, text);
829 if (v != NULL) {
Barry Warsawca74da41999-02-09 19:31:45 +0000830 PyErr_SetObject(PyExc_OSError, v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000831 Py_DECREF(v);
832 }
833 return NULL; /* Signal to Python that an Exception is Pending */
834}
835
836#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000837
838/* POSIX generic methods */
839
Barry Warsaw53699e91996-12-10 23:23:01 +0000840static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +0000841posix_fildes(PyObject *fdobj, int (*func)(int))
842{
Victor Stinner8c62be82010-05-06 00:08:46 +0000843 int fd;
844 int res;
845 fd = PyObject_AsFileDescriptor(fdobj);
846 if (fd < 0)
847 return NULL;
848 if (!_PyVerify_fd(fd))
849 return posix_error();
850 Py_BEGIN_ALLOW_THREADS
851 res = (*func)(fd);
852 Py_END_ALLOW_THREADS
853 if (res < 0)
854 return posix_error();
855 Py_INCREF(Py_None);
856 return Py_None;
Fred Drake4d1e64b2002-04-15 19:40:07 +0000857}
Guido van Rossum21142a01999-01-08 21:05:37 +0000858
859static PyObject *
Thomas Wouters477c8d52006-05-27 19:21:47 +0000860posix_1str(PyObject *args, char *format, int (*func)(const char*))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000861{
Victor Stinner8c62be82010-05-06 00:08:46 +0000862 PyObject *opath1 = NULL;
863 char *path1;
864 int res;
865 if (!PyArg_ParseTuple(args, format,
866 PyUnicode_FSConverter, &opath1))
867 return NULL;
868 path1 = PyBytes_AsString(opath1);
869 Py_BEGIN_ALLOW_THREADS
870 res = (*func)(path1);
871 Py_END_ALLOW_THREADS
872 if (res < 0)
873 return posix_error_with_allocated_filename(opath1);
874 Py_DECREF(opath1);
875 Py_INCREF(Py_None);
876 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000877}
878
Barry Warsaw53699e91996-12-10 23:23:01 +0000879static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +0000880posix_2str(PyObject *args,
Victor Stinner8c62be82010-05-06 00:08:46 +0000881 char *format,
882 int (*func)(const char *, const char *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000883{
Victor Stinner8c62be82010-05-06 00:08:46 +0000884 PyObject *opath1 = NULL, *opath2 = NULL;
885 char *path1, *path2;
886 int res;
887 if (!PyArg_ParseTuple(args, format,
888 PyUnicode_FSConverter, &opath1,
889 PyUnicode_FSConverter, &opath2)) {
890 return NULL;
891 }
892 path1 = PyBytes_AsString(opath1);
893 path2 = PyBytes_AsString(opath2);
894 Py_BEGIN_ALLOW_THREADS
895 res = (*func)(path1, path2);
896 Py_END_ALLOW_THREADS
897 Py_DECREF(opath1);
898 Py_DECREF(opath2);
899 if (res != 0)
900 /* XXX how to report both path1 and path2??? */
901 return posix_error();
902 Py_INCREF(Py_None);
903 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000904}
905
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000906#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +0000907static PyObject*
Victor Stinner8c62be82010-05-06 00:08:46 +0000908win32_1str(PyObject* args, char* func,
909 char* format, BOOL (__stdcall *funcA)(LPCSTR),
910 char* wformat, BOOL (__stdcall *funcW)(LPWSTR))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000911{
Victor Stinner8c62be82010-05-06 00:08:46 +0000912 PyObject *uni;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +0100913 const char *ansi;
Victor Stinner8c62be82010-05-06 00:08:46 +0000914 BOOL result;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +0000915
Victor Stinnereb5657a2011-09-30 01:44:27 +0200916 if (PyArg_ParseTuple(args, wformat, &uni))
917 {
918 wchar_t *wstr = PyUnicode_AsUnicode(uni);
919 if (wstr == NULL)
920 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +0000921 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +0200922 result = funcW(wstr);
Victor Stinner8c62be82010-05-06 00:08:46 +0000923 Py_END_ALLOW_THREADS
924 if (!result)
Victor Stinnereb5657a2011-09-30 01:44:27 +0200925 return win32_error_object(func, uni);
Victor Stinner8c62be82010-05-06 00:08:46 +0000926 Py_INCREF(Py_None);
927 return Py_None;
928 }
Victor Stinnereb5657a2011-09-30 01:44:27 +0200929 PyErr_Clear();
930
Victor Stinner8c62be82010-05-06 00:08:46 +0000931 if (!PyArg_ParseTuple(args, format, &ansi))
932 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +0100933 if (win32_warn_bytes_api())
934 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +0000935 Py_BEGIN_ALLOW_THREADS
936 result = funcA(ansi);
937 Py_END_ALLOW_THREADS
938 if (!result)
939 return win32_error(func, ansi);
940 Py_INCREF(Py_None);
941 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000942
943}
944
945/* This is a reimplementation of the C library's chdir function,
946 but one that produces Win32 errors instead of DOS error codes.
947 chdir is essentially a wrapper around SetCurrentDirectory; however,
948 it also needs to set "magic" environment variables indicating
949 the per-drive current directory, which are of the form =<drive>: */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000950static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000951win32_chdir(LPCSTR path)
952{
Victor Stinner8c62be82010-05-06 00:08:46 +0000953 char new_path[MAX_PATH+1];
954 int result;
955 char env[4] = "=x:";
Thomas Wouters477c8d52006-05-27 19:21:47 +0000956
Victor Stinner8c62be82010-05-06 00:08:46 +0000957 if(!SetCurrentDirectoryA(path))
958 return FALSE;
959 result = GetCurrentDirectoryA(MAX_PATH+1, new_path);
960 if (!result)
961 return FALSE;
962 /* In the ANSI API, there should not be any paths longer
963 than MAX_PATH. */
964 assert(result <= MAX_PATH+1);
965 if (strncmp(new_path, "\\\\", 2) == 0 ||
966 strncmp(new_path, "//", 2) == 0)
967 /* UNC path, nothing to do. */
968 return TRUE;
969 env[1] = new_path[0];
970 return SetEnvironmentVariableA(env, new_path);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000971}
972
973/* The Unicode version differs from the ANSI version
974 since the current directory might exceed MAX_PATH characters */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000975static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000976win32_wchdir(LPCWSTR path)
977{
Victor Stinner8c62be82010-05-06 00:08:46 +0000978 wchar_t _new_path[MAX_PATH+1], *new_path = _new_path;
979 int result;
980 wchar_t env[4] = L"=x:";
Thomas Wouters477c8d52006-05-27 19:21:47 +0000981
Victor Stinner8c62be82010-05-06 00:08:46 +0000982 if(!SetCurrentDirectoryW(path))
983 return FALSE;
984 result = GetCurrentDirectoryW(MAX_PATH+1, new_path);
985 if (!result)
986 return FALSE;
987 if (result > MAX_PATH+1) {
988 new_path = malloc(result * sizeof(wchar_t));
989 if (!new_path) {
990 SetLastError(ERROR_OUTOFMEMORY);
991 return FALSE;
992 }
993 result = GetCurrentDirectoryW(result, new_path);
994 if (!result) {
995 free(new_path);
996 return FALSE;
997 }
998 }
999 if (wcsncmp(new_path, L"\\\\", 2) == 0 ||
1000 wcsncmp(new_path, L"//", 2) == 0)
1001 /* UNC path, nothing to do. */
1002 return TRUE;
1003 env[1] = new_path[0];
1004 result = SetEnvironmentVariableW(env, new_path);
1005 if (new_path != _new_path)
1006 free(new_path);
1007 return result;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001008}
1009#endif
1010
Martin v. Löwis14694662006-02-03 12:54:16 +00001011#ifdef MS_WINDOWS
1012/* The CRT of Windows has a number of flaws wrt. its stat() implementation:
1013 - time stamps are restricted to second resolution
1014 - file modification times suffer from forth-and-back conversions between
1015 UTC and local time
1016 Therefore, we implement our own stat, based on the Win32 API directly.
1017*/
Victor Stinner8c62be82010-05-06 00:08:46 +00001018#define HAVE_STAT_NSEC 1
Martin v. Löwis14694662006-02-03 12:54:16 +00001019
1020struct win32_stat{
1021 int st_dev;
1022 __int64 st_ino;
1023 unsigned short st_mode;
1024 int st_nlink;
1025 int st_uid;
1026 int st_gid;
1027 int st_rdev;
1028 __int64 st_size;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001029 time_t st_atime;
Martin v. Löwis14694662006-02-03 12:54:16 +00001030 int st_atime_nsec;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001031 time_t st_mtime;
Martin v. Löwis14694662006-02-03 12:54:16 +00001032 int st_mtime_nsec;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001033 time_t st_ctime;
Martin v. Löwis14694662006-02-03 12:54:16 +00001034 int st_ctime_nsec;
1035};
1036
1037static __int64 secs_between_epochs = 11644473600; /* Seconds between 1.1.1601 and 1.1.1970 */
1038
1039static void
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001040FILE_TIME_to_time_t_nsec(FILETIME *in_ptr, time_t *time_out, int* nsec_out)
Martin v. Löwis14694662006-02-03 12:54:16 +00001041{
Victor Stinner8c62be82010-05-06 00:08:46 +00001042 /* XXX endianness. Shouldn't matter, as all Windows implementations are little-endian */
1043 /* Cannot simply cast and dereference in_ptr,
1044 since it might not be aligned properly */
1045 __int64 in;
1046 memcpy(&in, in_ptr, sizeof(in));
1047 *nsec_out = (int)(in % 10000000) * 100; /* FILETIME is in units of 100 nsec. */
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001048 *time_out = Py_SAFE_DOWNCAST((in / 10000000) - secs_between_epochs, __int64, time_t);
Martin v. Löwis14694662006-02-03 12:54:16 +00001049}
1050
Thomas Wouters477c8d52006-05-27 19:21:47 +00001051static void
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001052time_t_to_FILE_TIME(time_t time_in, int nsec_in, FILETIME *out_ptr)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001053{
Victor Stinner8c62be82010-05-06 00:08:46 +00001054 /* XXX endianness */
1055 __int64 out;
1056 out = time_in + secs_between_epochs;
1057 out = out * 10000000 + nsec_in / 100;
1058 memcpy(out_ptr, &out, sizeof(out));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001059}
1060
Martin v. Löwis14694662006-02-03 12:54:16 +00001061/* Below, we *know* that ugo+r is 0444 */
1062#if _S_IREAD != 0400
1063#error Unsupported C library
1064#endif
1065static int
1066attributes_to_mode(DWORD attr)
1067{
Victor Stinner8c62be82010-05-06 00:08:46 +00001068 int m = 0;
1069 if (attr & FILE_ATTRIBUTE_DIRECTORY)
1070 m |= _S_IFDIR | 0111; /* IFEXEC for user,group,other */
1071 else
1072 m |= _S_IFREG;
1073 if (attr & FILE_ATTRIBUTE_READONLY)
1074 m |= 0444;
1075 else
1076 m |= 0666;
1077 return m;
Martin v. Löwis14694662006-02-03 12:54:16 +00001078}
1079
1080static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001081attribute_data_to_stat(BY_HANDLE_FILE_INFORMATION *info, ULONG reparse_tag, struct win32_stat *result)
Martin v. Löwis14694662006-02-03 12:54:16 +00001082{
Victor Stinner8c62be82010-05-06 00:08:46 +00001083 memset(result, 0, sizeof(*result));
1084 result->st_mode = attributes_to_mode(info->dwFileAttributes);
1085 result->st_size = (((__int64)info->nFileSizeHigh)<<32) + info->nFileSizeLow;
1086 FILE_TIME_to_time_t_nsec(&info->ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1087 FILE_TIME_to_time_t_nsec(&info->ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1088 FILE_TIME_to_time_t_nsec(&info->ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001089 result->st_nlink = info->nNumberOfLinks;
Brian Curtin1b9df392010-11-24 20:24:31 +00001090 result->st_ino = (((__int64)info->nFileIndexHigh)<<32) + info->nFileIndexLow;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001091 if (reparse_tag == IO_REPARSE_TAG_SYMLINK) {
1092 /* first clear the S_IFMT bits */
1093 result->st_mode ^= (result->st_mode & 0170000);
1094 /* now set the bits that make this a symlink */
1095 result->st_mode |= 0120000;
1096 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001097
Victor Stinner8c62be82010-05-06 00:08:46 +00001098 return 0;
Martin v. Löwis14694662006-02-03 12:54:16 +00001099}
1100
Guido van Rossumd8faa362007-04-27 19:54:29 +00001101static BOOL
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001102attributes_from_dir(LPCSTR pszFile, BY_HANDLE_FILE_INFORMATION *info, ULONG *reparse_tag)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001103{
Victor Stinner8c62be82010-05-06 00:08:46 +00001104 HANDLE hFindFile;
1105 WIN32_FIND_DATAA FileData;
1106 hFindFile = FindFirstFileA(pszFile, &FileData);
1107 if (hFindFile == INVALID_HANDLE_VALUE)
1108 return FALSE;
1109 FindClose(hFindFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001110 memset(info, 0, sizeof(*info));
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001111 *reparse_tag = 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001112 info->dwFileAttributes = FileData.dwFileAttributes;
1113 info->ftCreationTime = FileData.ftCreationTime;
1114 info->ftLastAccessTime = FileData.ftLastAccessTime;
1115 info->ftLastWriteTime = FileData.ftLastWriteTime;
1116 info->nFileSizeHigh = FileData.nFileSizeHigh;
1117 info->nFileSizeLow = FileData.nFileSizeLow;
1118/* info->nNumberOfLinks = 1; */
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001119 if (FileData.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
1120 *reparse_tag = FileData.dwReserved0;
Victor Stinner8c62be82010-05-06 00:08:46 +00001121 return TRUE;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001122}
1123
1124static BOOL
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001125attributes_from_dir_w(LPCWSTR pszFile, BY_HANDLE_FILE_INFORMATION *info, ULONG *reparse_tag)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001126{
Victor Stinner8c62be82010-05-06 00:08:46 +00001127 HANDLE hFindFile;
1128 WIN32_FIND_DATAW FileData;
1129 hFindFile = FindFirstFileW(pszFile, &FileData);
1130 if (hFindFile == INVALID_HANDLE_VALUE)
1131 return FALSE;
1132 FindClose(hFindFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001133 memset(info, 0, sizeof(*info));
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001134 *reparse_tag = 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001135 info->dwFileAttributes = FileData.dwFileAttributes;
1136 info->ftCreationTime = FileData.ftCreationTime;
1137 info->ftLastAccessTime = FileData.ftLastAccessTime;
1138 info->ftLastWriteTime = FileData.ftLastWriteTime;
1139 info->nFileSizeHigh = FileData.nFileSizeHigh;
1140 info->nFileSizeLow = FileData.nFileSizeLow;
1141/* info->nNumberOfLinks = 1; */
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001142 if (FileData.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
1143 *reparse_tag = FileData.dwReserved0;
Victor Stinner8c62be82010-05-06 00:08:46 +00001144 return TRUE;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001145}
1146
Brian Curtind25aef52011-06-13 15:16:04 -05001147/* Grab GetFinalPathNameByHandle dynamically from kernel32 */
1148static int has_GetFinalPathNameByHandle = 0;
Brian Curtind25aef52011-06-13 15:16:04 -05001149static DWORD (CALLBACK *Py_GetFinalPathNameByHandleW)(HANDLE, LPWSTR, DWORD,
1150 DWORD);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001151static int
Brian Curtind25aef52011-06-13 15:16:04 -05001152check_GetFinalPathNameByHandle()
Brian Curtinf5e76d02010-11-24 13:14:05 +00001153{
Brian Curtind25aef52011-06-13 15:16:04 -05001154 HINSTANCE hKernel32;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001155 DWORD (CALLBACK *Py_GetFinalPathNameByHandleA)(HANDLE, LPSTR, DWORD,
1156 DWORD);
1157
Brian Curtind25aef52011-06-13 15:16:04 -05001158 /* only recheck */
1159 if (!has_GetFinalPathNameByHandle)
1160 {
Martin v. Löwis50590f12012-01-14 17:54:09 +01001161 hKernel32 = GetModuleHandleW(L"KERNEL32");
Brian Curtind25aef52011-06-13 15:16:04 -05001162 *(FARPROC*)&Py_GetFinalPathNameByHandleA = GetProcAddress(hKernel32,
1163 "GetFinalPathNameByHandleA");
1164 *(FARPROC*)&Py_GetFinalPathNameByHandleW = GetProcAddress(hKernel32,
1165 "GetFinalPathNameByHandleW");
1166 has_GetFinalPathNameByHandle = Py_GetFinalPathNameByHandleA &&
1167 Py_GetFinalPathNameByHandleW;
1168 }
1169 return has_GetFinalPathNameByHandle;
1170}
1171
1172static BOOL
1173get_target_path(HANDLE hdl, wchar_t **target_path)
1174{
1175 int buf_size, result_length;
1176 wchar_t *buf;
1177
1178 /* We have a good handle to the target, use it to determine
1179 the target path name (then we'll call lstat on it). */
1180 buf_size = Py_GetFinalPathNameByHandleW(hdl, 0, 0,
1181 VOLUME_NAME_DOS);
1182 if(!buf_size)
1183 return FALSE;
1184
1185 buf = (wchar_t *)malloc((buf_size+1)*sizeof(wchar_t));
Brian Curtinc8be8402011-06-14 09:52:50 -05001186 if (!buf) {
1187 SetLastError(ERROR_OUTOFMEMORY);
1188 return FALSE;
1189 }
1190
Brian Curtind25aef52011-06-13 15:16:04 -05001191 result_length = Py_GetFinalPathNameByHandleW(hdl,
1192 buf, buf_size, VOLUME_NAME_DOS);
1193
1194 if(!result_length) {
1195 free(buf);
1196 return FALSE;
1197 }
1198
1199 if(!CloseHandle(hdl)) {
1200 free(buf);
1201 return FALSE;
1202 }
1203
1204 buf[result_length] = 0;
1205
1206 *target_path = buf;
1207 return TRUE;
1208}
1209
1210static int
1211win32_xstat_impl_w(const wchar_t *path, struct win32_stat *result,
1212 BOOL traverse);
1213static int
1214win32_xstat_impl(const char *path, struct win32_stat *result,
1215 BOOL traverse)
1216{
Victor Stinner26de69d2011-06-17 15:15:38 +02001217 int code;
Brian Curtind25aef52011-06-13 15:16:04 -05001218 HANDLE hFile, hFile2;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001219 BY_HANDLE_FILE_INFORMATION info;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001220 ULONG reparse_tag = 0;
Victor Stinner26de69d2011-06-17 15:15:38 +02001221 wchar_t *target_path;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001222 const char *dot;
1223
Brian Curtind25aef52011-06-13 15:16:04 -05001224 if(!check_GetFinalPathNameByHandle()) {
Brian Curtinc8be8402011-06-14 09:52:50 -05001225 /* If the OS doesn't have GetFinalPathNameByHandle, don't
1226 traverse reparse point. */
1227 traverse = FALSE;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001228 }
1229
Brian Curtinf5e76d02010-11-24 13:14:05 +00001230 hFile = CreateFileA(
1231 path,
Brian Curtind25aef52011-06-13 15:16:04 -05001232 FILE_READ_ATTRIBUTES, /* desired access */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001233 0, /* share mode */
1234 NULL, /* security attributes */
1235 OPEN_EXISTING,
1236 /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
Brian Curtind25aef52011-06-13 15:16:04 -05001237 /* FILE_FLAG_OPEN_REPARSE_POINT does not follow the symlink.
1238 Because of this, calls like GetFinalPathNameByHandle will return
1239 the symlink path agin and not the actual final path. */
1240 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS|
1241 FILE_FLAG_OPEN_REPARSE_POINT,
Brian Curtinf5e76d02010-11-24 13:14:05 +00001242 NULL);
1243
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001244 if (hFile == INVALID_HANDLE_VALUE) {
Brian Curtinf5e76d02010-11-24 13:14:05 +00001245 /* Either the target doesn't exist, or we don't have access to
1246 get a handle to it. If the former, we need to return an error.
1247 If the latter, we can use attributes_from_dir. */
1248 if (GetLastError() != ERROR_SHARING_VIOLATION)
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001249 return -1;
1250 /* Could not get attributes on open file. Fall back to
1251 reading the directory. */
1252 if (!attributes_from_dir(path, &info, &reparse_tag))
1253 /* Very strange. This should not fail now */
1254 return -1;
1255 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
1256 if (traverse) {
1257 /* Should traverse, but could not open reparse point handle */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001258 SetLastError(ERROR_SHARING_VIOLATION);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001259 return -1;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001260 }
Brian Curtinf5e76d02010-11-24 13:14:05 +00001261 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001262 } else {
1263 if (!GetFileInformationByHandle(hFile, &info)) {
1264 CloseHandle(hFile);
Brian Curtind25aef52011-06-13 15:16:04 -05001265 return -1;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001266 }
1267 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
Brian Curtind25aef52011-06-13 15:16:04 -05001268 if (!win32_get_reparse_tag(hFile, &reparse_tag))
1269 return -1;
1270
1271 /* Close the outer open file handle now that we're about to
1272 reopen it with different flags. */
1273 if (!CloseHandle(hFile))
1274 return -1;
1275
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001276 if (traverse) {
Brian Curtind25aef52011-06-13 15:16:04 -05001277 /* In order to call GetFinalPathNameByHandle we need to open
1278 the file without the reparse handling flag set. */
1279 hFile2 = CreateFileA(
1280 path, FILE_READ_ATTRIBUTES, FILE_SHARE_READ,
1281 NULL, OPEN_EXISTING,
1282 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS,
1283 NULL);
1284 if (hFile2 == INVALID_HANDLE_VALUE)
1285 return -1;
1286
1287 if (!get_target_path(hFile2, &target_path))
1288 return -1;
1289
1290 code = win32_xstat_impl_w(target_path, result, FALSE);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001291 free(target_path);
1292 return code;
1293 }
Hirokazu Yamamoto7ed117a2010-12-07 10:24:37 +00001294 } else
1295 CloseHandle(hFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001296 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001297 attribute_data_to_stat(&info, reparse_tag, result);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001298
1299 /* Set S_IEXEC if it is an .exe, .bat, ... */
1300 dot = strrchr(path, '.');
1301 if (dot) {
1302 if (stricmp(dot, ".bat") == 0 || stricmp(dot, ".cmd") == 0 ||
1303 stricmp(dot, ".exe") == 0 || stricmp(dot, ".com") == 0)
1304 result->st_mode |= 0111;
1305 }
1306 return 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001307}
1308
1309static int
Brian Curtind25aef52011-06-13 15:16:04 -05001310win32_xstat_impl_w(const wchar_t *path, struct win32_stat *result,
1311 BOOL traverse)
Brian Curtinf5e76d02010-11-24 13:14:05 +00001312{
1313 int code;
Brian Curtind25aef52011-06-13 15:16:04 -05001314 HANDLE hFile, hFile2;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001315 BY_HANDLE_FILE_INFORMATION info;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001316 ULONG reparse_tag = 0;
Victor Stinner26de69d2011-06-17 15:15:38 +02001317 wchar_t *target_path;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001318 const wchar_t *dot;
1319
Brian Curtind25aef52011-06-13 15:16:04 -05001320 if(!check_GetFinalPathNameByHandle()) {
Brian Curtinc8be8402011-06-14 09:52:50 -05001321 /* If the OS doesn't have GetFinalPathNameByHandle, don't
1322 traverse reparse point. */
1323 traverse = FALSE;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001324 }
1325
Brian Curtinf5e76d02010-11-24 13:14:05 +00001326 hFile = CreateFileW(
1327 path,
Brian Curtind25aef52011-06-13 15:16:04 -05001328 FILE_READ_ATTRIBUTES, /* desired access */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001329 0, /* share mode */
1330 NULL, /* security attributes */
1331 OPEN_EXISTING,
1332 /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
Brian Curtind25aef52011-06-13 15:16:04 -05001333 /* FILE_FLAG_OPEN_REPARSE_POINT does not follow the symlink.
1334 Because of this, calls like GetFinalPathNameByHandle will return
1335 the symlink path agin and not the actual final path. */
Victor Stinner26de69d2011-06-17 15:15:38 +02001336 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS|
Brian Curtind25aef52011-06-13 15:16:04 -05001337 FILE_FLAG_OPEN_REPARSE_POINT,
Brian Curtinf5e76d02010-11-24 13:14:05 +00001338 NULL);
1339
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001340 if (hFile == INVALID_HANDLE_VALUE) {
Brian Curtinf5e76d02010-11-24 13:14:05 +00001341 /* Either the target doesn't exist, or we don't have access to
1342 get a handle to it. If the former, we need to return an error.
1343 If the latter, we can use attributes_from_dir. */
1344 if (GetLastError() != ERROR_SHARING_VIOLATION)
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001345 return -1;
1346 /* Could not get attributes on open file. Fall back to
1347 reading the directory. */
1348 if (!attributes_from_dir_w(path, &info, &reparse_tag))
1349 /* Very strange. This should not fail now */
1350 return -1;
1351 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
1352 if (traverse) {
1353 /* Should traverse, but could not open reparse point handle */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001354 SetLastError(ERROR_SHARING_VIOLATION);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001355 return -1;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001356 }
Brian Curtinf5e76d02010-11-24 13:14:05 +00001357 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001358 } else {
1359 if (!GetFileInformationByHandle(hFile, &info)) {
1360 CloseHandle(hFile);
Brian Curtind25aef52011-06-13 15:16:04 -05001361 return -1;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001362 }
1363 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
Brian Curtind25aef52011-06-13 15:16:04 -05001364 if (!win32_get_reparse_tag(hFile, &reparse_tag))
1365 return -1;
1366
1367 /* Close the outer open file handle now that we're about to
1368 reopen it with different flags. */
1369 if (!CloseHandle(hFile))
1370 return -1;
1371
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001372 if (traverse) {
Brian Curtind25aef52011-06-13 15:16:04 -05001373 /* In order to call GetFinalPathNameByHandle we need to open
1374 the file without the reparse handling flag set. */
1375 hFile2 = CreateFileW(
1376 path, FILE_READ_ATTRIBUTES, FILE_SHARE_READ,
1377 NULL, OPEN_EXISTING,
1378 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS,
1379 NULL);
1380 if (hFile2 == INVALID_HANDLE_VALUE)
1381 return -1;
1382
1383 if (!get_target_path(hFile2, &target_path))
1384 return -1;
1385
1386 code = win32_xstat_impl_w(target_path, result, FALSE);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001387 free(target_path);
1388 return code;
1389 }
Hirokazu Yamamoto7ed117a2010-12-07 10:24:37 +00001390 } else
1391 CloseHandle(hFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001392 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001393 attribute_data_to_stat(&info, reparse_tag, result);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001394
1395 /* Set S_IEXEC if it is an .exe, .bat, ... */
1396 dot = wcsrchr(path, '.');
1397 if (dot) {
1398 if (_wcsicmp(dot, L".bat") == 0 || _wcsicmp(dot, L".cmd") == 0 ||
1399 _wcsicmp(dot, L".exe") == 0 || _wcsicmp(dot, L".com") == 0)
1400 result->st_mode |= 0111;
1401 }
1402 return 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001403}
1404
1405static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001406win32_xstat(const char *path, struct win32_stat *result, BOOL traverse)
Brian Curtinf5e76d02010-11-24 13:14:05 +00001407{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001408 /* Protocol violation: we explicitly clear errno, instead of
1409 setting it to a POSIX error. Callers should use GetLastError. */
Brian Curtind25aef52011-06-13 15:16:04 -05001410 int code = win32_xstat_impl(path, result, traverse);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001411 errno = 0;
1412 return code;
1413}
Brian Curtinf5e76d02010-11-24 13:14:05 +00001414
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001415static int
1416win32_xstat_w(const wchar_t *path, struct win32_stat *result, BOOL traverse)
1417{
1418 /* Protocol violation: we explicitly clear errno, instead of
1419 setting it to a POSIX error. Callers should use GetLastError. */
Brian Curtind25aef52011-06-13 15:16:04 -05001420 int code = win32_xstat_impl_w(path, result, traverse);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001421 errno = 0;
1422 return code;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001423}
Brian Curtind25aef52011-06-13 15:16:04 -05001424/* About the following functions: win32_lstat_w, win32_stat, win32_stat_w
Brian Curtind40e6f72010-07-08 21:39:08 +00001425
1426 In Posix, stat automatically traverses symlinks and returns the stat
1427 structure for the target. In Windows, the equivalent GetFileAttributes by
1428 default does not traverse symlinks and instead returns attributes for
1429 the symlink.
1430
1431 Therefore, win32_lstat will get the attributes traditionally, and
1432 win32_stat will first explicitly resolve the symlink target and then will
1433 call win32_lstat on that result.
1434
Ezio Melotti4969f702011-03-15 05:59:46 +02001435 The _w represent Unicode equivalents of the aforementioned ANSI functions. */
Brian Curtind40e6f72010-07-08 21:39:08 +00001436
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001437static int
Brian Curtind40e6f72010-07-08 21:39:08 +00001438win32_lstat(const char* path, struct win32_stat *result)
Martin v. Löwis14694662006-02-03 12:54:16 +00001439{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001440 return win32_xstat(path, result, FALSE);
Martin v. Löwis14694662006-02-03 12:54:16 +00001441}
1442
Victor Stinner8c62be82010-05-06 00:08:46 +00001443static int
Brian Curtind40e6f72010-07-08 21:39:08 +00001444win32_lstat_w(const wchar_t* path, struct win32_stat *result)
Martin v. Löwis14694662006-02-03 12:54:16 +00001445{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001446 return win32_xstat_w(path, result, FALSE);
Brian Curtind40e6f72010-07-08 21:39:08 +00001447}
1448
1449static int
1450win32_stat(const char* path, struct win32_stat *result)
1451{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001452 return win32_xstat(path, result, TRUE);
Brian Curtind40e6f72010-07-08 21:39:08 +00001453}
1454
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001455static int
Brian Curtind40e6f72010-07-08 21:39:08 +00001456win32_stat_w(const wchar_t* path, struct win32_stat *result)
1457{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001458 return win32_xstat_w(path, result, TRUE);
Martin v. Löwis14694662006-02-03 12:54:16 +00001459}
1460
1461static int
1462win32_fstat(int file_number, struct win32_stat *result)
1463{
Victor Stinner8c62be82010-05-06 00:08:46 +00001464 BY_HANDLE_FILE_INFORMATION info;
1465 HANDLE h;
1466 int type;
Martin v. Löwis14694662006-02-03 12:54:16 +00001467
Victor Stinner8c62be82010-05-06 00:08:46 +00001468 h = (HANDLE)_get_osfhandle(file_number);
Martin v. Löwis14694662006-02-03 12:54:16 +00001469
Victor Stinner8c62be82010-05-06 00:08:46 +00001470 /* Protocol violation: we explicitly clear errno, instead of
1471 setting it to a POSIX error. Callers should use GetLastError. */
1472 errno = 0;
Martin v. Löwis14694662006-02-03 12:54:16 +00001473
Victor Stinner8c62be82010-05-06 00:08:46 +00001474 if (h == INVALID_HANDLE_VALUE) {
1475 /* This is really a C library error (invalid file handle).
1476 We set the Win32 error to the closes one matching. */
1477 SetLastError(ERROR_INVALID_HANDLE);
1478 return -1;
1479 }
1480 memset(result, 0, sizeof(*result));
Martin v. Löwis14694662006-02-03 12:54:16 +00001481
Victor Stinner8c62be82010-05-06 00:08:46 +00001482 type = GetFileType(h);
1483 if (type == FILE_TYPE_UNKNOWN) {
1484 DWORD error = GetLastError();
1485 if (error != 0) {
Brian Curtinf5e76d02010-11-24 13:14:05 +00001486 return -1;
Victor Stinner8c62be82010-05-06 00:08:46 +00001487 }
1488 /* else: valid but unknown file */
1489 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001490
Victor Stinner8c62be82010-05-06 00:08:46 +00001491 if (type != FILE_TYPE_DISK) {
1492 if (type == FILE_TYPE_CHAR)
1493 result->st_mode = _S_IFCHR;
1494 else if (type == FILE_TYPE_PIPE)
1495 result->st_mode = _S_IFIFO;
1496 return 0;
1497 }
1498
1499 if (!GetFileInformationByHandle(h, &info)) {
1500 return -1;
1501 }
1502
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001503 attribute_data_to_stat(&info, 0, result);
Victor Stinner8c62be82010-05-06 00:08:46 +00001504 /* specific to fstat() */
Victor Stinner8c62be82010-05-06 00:08:46 +00001505 result->st_ino = (((__int64)info.nFileIndexHigh)<<32) + info.nFileIndexLow;
1506 return 0;
Martin v. Löwis14694662006-02-03 12:54:16 +00001507}
1508
1509#endif /* MS_WINDOWS */
1510
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001511PyDoc_STRVAR(stat_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001512"stat_result: Result from stat or lstat.\n\n\
1513This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001514 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001515or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
1516\n\
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001517Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
1518or st_flags, they are available as attributes only.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001519\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001520See os.stat for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001521
1522static PyStructSequence_Field stat_result_fields[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001523 {"st_mode", "protection bits"},
1524 {"st_ino", "inode"},
1525 {"st_dev", "device"},
1526 {"st_nlink", "number of hard links"},
1527 {"st_uid", "user ID of owner"},
1528 {"st_gid", "group ID of owner"},
1529 {"st_size", "total size, in bytes"},
1530 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
1531 {NULL, "integer time of last access"},
1532 {NULL, "integer time of last modification"},
1533 {NULL, "integer time of last change"},
1534 {"st_atime", "time of last access"},
1535 {"st_mtime", "time of last modification"},
1536 {"st_ctime", "time of last change"},
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001537#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00001538 {"st_blksize", "blocksize for filesystem I/O"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001539#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001540#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Victor Stinner8c62be82010-05-06 00:08:46 +00001541 {"st_blocks", "number of blocks allocated"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001542#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001543#ifdef HAVE_STRUCT_STAT_ST_RDEV
Victor Stinner8c62be82010-05-06 00:08:46 +00001544 {"st_rdev", "device type (if inode device)"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001545#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001546#ifdef HAVE_STRUCT_STAT_ST_FLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00001547 {"st_flags", "user defined flags for file"},
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001548#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001549#ifdef HAVE_STRUCT_STAT_ST_GEN
Victor Stinner8c62be82010-05-06 00:08:46 +00001550 {"st_gen", "generation number"},
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001551#endif
1552#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00001553 {"st_birthtime", "time of creation"},
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001554#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001555 {0}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001556};
1557
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001558#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001559#define ST_BLKSIZE_IDX 13
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001560#else
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001561#define ST_BLKSIZE_IDX 12
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001562#endif
1563
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001564#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001565#define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
1566#else
1567#define ST_BLOCKS_IDX ST_BLKSIZE_IDX
1568#endif
1569
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001570#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001571#define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
1572#else
1573#define ST_RDEV_IDX ST_BLOCKS_IDX
1574#endif
1575
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001576#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1577#define ST_FLAGS_IDX (ST_RDEV_IDX+1)
1578#else
1579#define ST_FLAGS_IDX ST_RDEV_IDX
1580#endif
1581
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001582#ifdef HAVE_STRUCT_STAT_ST_GEN
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001583#define ST_GEN_IDX (ST_FLAGS_IDX+1)
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001584#else
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001585#define ST_GEN_IDX ST_FLAGS_IDX
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001586#endif
1587
1588#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1589#define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
1590#else
1591#define ST_BIRTHTIME_IDX ST_GEN_IDX
1592#endif
1593
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001594static PyStructSequence_Desc stat_result_desc = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001595 "stat_result", /* name */
1596 stat_result__doc__, /* doc */
1597 stat_result_fields,
1598 10
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001599};
1600
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001601PyDoc_STRVAR(statvfs_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001602"statvfs_result: Result from statvfs or fstatvfs.\n\n\
1603This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001604 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
Guido van Rossuma4dc73e2001-10-18 20:53:15 +00001605or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001606\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001607See os.statvfs for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001608
1609static PyStructSequence_Field statvfs_result_fields[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001610 {"f_bsize", },
1611 {"f_frsize", },
1612 {"f_blocks", },
1613 {"f_bfree", },
1614 {"f_bavail", },
1615 {"f_files", },
1616 {"f_ffree", },
1617 {"f_favail", },
1618 {"f_flag", },
1619 {"f_namemax",},
1620 {0}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001621};
1622
1623static PyStructSequence_Desc statvfs_result_desc = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001624 "statvfs_result", /* name */
1625 statvfs_result__doc__, /* doc */
1626 statvfs_result_fields,
1627 10
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001628};
1629
Ross Lagerwall7807c352011-03-17 20:20:30 +02001630#if defined(HAVE_WAITID) && !defined(__APPLE__)
1631PyDoc_STRVAR(waitid_result__doc__,
1632"waitid_result: Result from waitid.\n\n\
1633This object may be accessed either as a tuple of\n\
1634 (si_pid, si_uid, si_signo, si_status, si_code),\n\
1635or via the attributes si_pid, si_uid, and so on.\n\
1636\n\
1637See os.waitid for more information.");
1638
1639static PyStructSequence_Field waitid_result_fields[] = {
1640 {"si_pid", },
1641 {"si_uid", },
1642 {"si_signo", },
1643 {"si_status", },
1644 {"si_code", },
1645 {0}
1646};
1647
1648static PyStructSequence_Desc waitid_result_desc = {
1649 "waitid_result", /* name */
1650 waitid_result__doc__, /* doc */
1651 waitid_result_fields,
1652 5
1653};
1654static PyTypeObject WaitidResultType;
1655#endif
1656
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001657static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001658static PyTypeObject StatResultType;
1659static PyTypeObject StatVFSResultType;
Benjamin Petersonbad9c2f2011-08-02 18:42:14 -05001660#if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER)
Benjamin Peterson94b580d2011-08-02 17:30:04 -05001661static PyTypeObject SchedParamType;
Benjamin Petersonbad9c2f2011-08-02 18:42:14 -05001662#endif
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001663static newfunc structseq_new;
1664
1665static PyObject *
1666statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1667{
Victor Stinner8c62be82010-05-06 00:08:46 +00001668 PyStructSequence *result;
1669 int i;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001670
Victor Stinner8c62be82010-05-06 00:08:46 +00001671 result = (PyStructSequence*)structseq_new(type, args, kwds);
1672 if (!result)
1673 return NULL;
1674 /* If we have been initialized from a tuple,
1675 st_?time might be set to None. Initialize it
1676 from the int slots. */
1677 for (i = 7; i <= 9; i++) {
1678 if (result->ob_item[i+3] == Py_None) {
1679 Py_DECREF(Py_None);
1680 Py_INCREF(result->ob_item[i]);
1681 result->ob_item[i+3] = result->ob_item[i];
1682 }
1683 }
1684 return (PyObject*)result;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001685}
1686
1687
1688
1689/* If true, st_?time is float. */
Martin v. Löwisfe33d0b2005-01-16 08:57:39 +00001690static int _stat_float_times = 1;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001691
1692PyDoc_STRVAR(stat_float_times__doc__,
1693"stat_float_times([newval]) -> oldval\n\n\
1694Determine whether os.[lf]stat represents time stamps as float objects.\n\
1695If newval is True, future calls to stat() return floats, if it is False,\n\
1696future calls return ints. \n\
1697If newval is omitted, return the current setting.\n");
1698
1699static PyObject*
1700stat_float_times(PyObject* self, PyObject *args)
1701{
Victor Stinner8c62be82010-05-06 00:08:46 +00001702 int newval = -1;
1703 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
1704 return NULL;
1705 if (newval == -1)
1706 /* Return old value */
1707 return PyBool_FromLong(_stat_float_times);
1708 _stat_float_times = newval;
1709 Py_INCREF(Py_None);
1710 return Py_None;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001711}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001712
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001713static void
Victor Stinner4195b5c2012-02-08 23:03:19 +01001714fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001715{
Victor Stinner8c62be82010-05-06 00:08:46 +00001716 PyObject *fval,*ival;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001717#if SIZEOF_TIME_T > SIZEOF_LONG
Victor Stinner8c62be82010-05-06 00:08:46 +00001718 ival = PyLong_FromLongLong((PY_LONG_LONG)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001719#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001720 ival = PyLong_FromLong((long)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001721#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001722 if (!ival)
1723 return;
Victor Stinner4195b5c2012-02-08 23:03:19 +01001724 if (_stat_float_times) {
1725 fval = PyFloat_FromDouble(sec + 1e-9*nsec);
1726 } else {
Victor Stinner8c62be82010-05-06 00:08:46 +00001727 fval = ival;
1728 Py_INCREF(fval);
1729 }
1730 PyStructSequence_SET_ITEM(v, index, ival);
1731 PyStructSequence_SET_ITEM(v, index+3, fval);
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001732}
1733
Tim Peters5aa91602002-01-30 05:46:57 +00001734/* pack a system stat C structure into the Python stat tuple
Fred Drake699f3522000-06-29 21:12:41 +00001735 (used by posix_stat() and posix_fstat()) */
1736static PyObject*
Victor Stinner4195b5c2012-02-08 23:03:19 +01001737_pystat_fromstructstat(STRUCT_STAT *st)
Fred Drake699f3522000-06-29 21:12:41 +00001738{
Victor Stinner8c62be82010-05-06 00:08:46 +00001739 unsigned long ansec, mnsec, cnsec;
1740 PyObject *v = PyStructSequence_New(&StatResultType);
1741 if (v == NULL)
1742 return NULL;
Fred Drake699f3522000-06-29 21:12:41 +00001743
Victor Stinner8c62be82010-05-06 00:08:46 +00001744 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long)st->st_mode));
Fred Drake699f3522000-06-29 21:12:41 +00001745#ifdef HAVE_LARGEFILE_SUPPORT
Victor Stinner8c62be82010-05-06 00:08:46 +00001746 PyStructSequence_SET_ITEM(v, 1,
1747 PyLong_FromLongLong((PY_LONG_LONG)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001748#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001749 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001750#endif
1751#if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00001752 PyStructSequence_SET_ITEM(v, 2,
1753 PyLong_FromLongLong((PY_LONG_LONG)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001754#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001755 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001756#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001757 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long)st->st_nlink));
1758 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long)st->st_uid));
1759 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long)st->st_gid));
Fred Drake699f3522000-06-29 21:12:41 +00001760#ifdef HAVE_LARGEFILE_SUPPORT
Victor Stinner8c62be82010-05-06 00:08:46 +00001761 PyStructSequence_SET_ITEM(v, 6,
1762 PyLong_FromLongLong((PY_LONG_LONG)st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001763#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001764 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong(st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001765#endif
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001766
Martin v. Löwis14694662006-02-03 12:54:16 +00001767#if defined(HAVE_STAT_TV_NSEC)
Victor Stinner8c62be82010-05-06 00:08:46 +00001768 ansec = st->st_atim.tv_nsec;
1769 mnsec = st->st_mtim.tv_nsec;
1770 cnsec = st->st_ctim.tv_nsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001771#elif defined(HAVE_STAT_TV_NSEC2)
Victor Stinner8c62be82010-05-06 00:08:46 +00001772 ansec = st->st_atimespec.tv_nsec;
1773 mnsec = st->st_mtimespec.tv_nsec;
1774 cnsec = st->st_ctimespec.tv_nsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001775#elif defined(HAVE_STAT_NSEC)
Victor Stinner8c62be82010-05-06 00:08:46 +00001776 ansec = st->st_atime_nsec;
1777 mnsec = st->st_mtime_nsec;
1778 cnsec = st->st_ctime_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001779#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001780 ansec = mnsec = cnsec = 0;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001781#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +01001782 fill_time(v, 7, st->st_atime, ansec);
1783 fill_time(v, 8, st->st_mtime, mnsec);
1784 fill_time(v, 9, st->st_ctime, cnsec);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001785
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001786#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00001787 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
1788 PyLong_FromLong((long)st->st_blksize));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001789#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001790#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Victor Stinner8c62be82010-05-06 00:08:46 +00001791 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
1792 PyLong_FromLong((long)st->st_blocks));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001793#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001794#ifdef HAVE_STRUCT_STAT_ST_RDEV
Victor Stinner8c62be82010-05-06 00:08:46 +00001795 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
1796 PyLong_FromLong((long)st->st_rdev));
Fred Drake699f3522000-06-29 21:12:41 +00001797#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001798#ifdef HAVE_STRUCT_STAT_ST_GEN
Victor Stinner8c62be82010-05-06 00:08:46 +00001799 PyStructSequence_SET_ITEM(v, ST_GEN_IDX,
1800 PyLong_FromLong((long)st->st_gen));
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001801#endif
1802#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00001803 {
Victor Stinner4195b5c2012-02-08 23:03:19 +01001804 PyObject *val;
1805 unsigned long bsec,bnsec;
1806 bsec = (long)st->st_birthtime;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001807#ifdef HAVE_STAT_TV_NSEC2
Victor Stinner4195b5c2012-02-08 23:03:19 +01001808 bnsec = st->st_birthtimespec.tv_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001809#else
Victor Stinner4195b5c2012-02-08 23:03:19 +01001810 bnsec = 0;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001811#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +01001812 if (_stat_float_times) {
1813 val = PyFloat_FromDouble(bsec + 1e-9*bnsec);
1814 } else {
1815 val = PyLong_FromLong((long)bsec);
1816 }
1817 PyStructSequence_SET_ITEM(v, ST_BIRTHTIME_IDX,
1818 val);
Victor Stinner8c62be82010-05-06 00:08:46 +00001819 }
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001820#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001821#ifdef HAVE_STRUCT_STAT_ST_FLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00001822 PyStructSequence_SET_ITEM(v, ST_FLAGS_IDX,
1823 PyLong_FromLong((long)st->st_flags));
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001824#endif
Fred Drake699f3522000-06-29 21:12:41 +00001825
Victor Stinner8c62be82010-05-06 00:08:46 +00001826 if (PyErr_Occurred()) {
1827 Py_DECREF(v);
1828 return NULL;
1829 }
Fred Drake699f3522000-06-29 21:12:41 +00001830
Victor Stinner8c62be82010-05-06 00:08:46 +00001831 return v;
Fred Drake699f3522000-06-29 21:12:41 +00001832}
1833
Barry Warsaw53699e91996-12-10 23:23:01 +00001834static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01001835posix_do_stat(PyObject *self, PyObject *args,
Victor Stinner8c62be82010-05-06 00:08:46 +00001836 char *format,
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001837#ifdef __VMS
Victor Stinner8c62be82010-05-06 00:08:46 +00001838 int (*statfunc)(const char *, STRUCT_STAT *, ...),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001839#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001840 int (*statfunc)(const char *, STRUCT_STAT *),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001841#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001842 char *wformat,
Victor Stinnereb5657a2011-09-30 01:44:27 +02001843 int (*wstatfunc)(const wchar_t *, STRUCT_STAT *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001844{
Victor Stinner8c62be82010-05-06 00:08:46 +00001845 STRUCT_STAT st;
1846 PyObject *opath;
1847 char *path;
1848 int res;
1849 PyObject *result;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001850
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001851#ifdef MS_WINDOWS
Victor Stinnereb5657a2011-09-30 01:44:27 +02001852 PyObject *po;
Victor Stinner4195b5c2012-02-08 23:03:19 +01001853 if (PyArg_ParseTuple(args, wformat, &po)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02001854 wchar_t *wpath = PyUnicode_AsUnicode(po);
1855 if (wpath == NULL)
1856 return NULL;
Martin v. Löwis14694662006-02-03 12:54:16 +00001857
Victor Stinner8c62be82010-05-06 00:08:46 +00001858 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00001859 res = wstatfunc(wpath, &st);
1860 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001861
Victor Stinner8c62be82010-05-06 00:08:46 +00001862 if (res != 0)
Victor Stinnereb5657a2011-09-30 01:44:27 +02001863 return win32_error_object("stat", po);
Victor Stinner4195b5c2012-02-08 23:03:19 +01001864 return _pystat_fromstructstat(&st);
Victor Stinner8c62be82010-05-06 00:08:46 +00001865 }
1866 /* Drop the argument parsing error as narrow strings
1867 are also valid. */
1868 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001869#endif
1870
Victor Stinner4195b5c2012-02-08 23:03:19 +01001871 if (!PyArg_ParseTuple(args, format,
1872 PyUnicode_FSConverter, &opath))
Victor Stinner8c62be82010-05-06 00:08:46 +00001873 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001874#ifdef MS_WINDOWS
1875 if (win32_warn_bytes_api()) {
1876 Py_DECREF(opath);
1877 return NULL;
1878 }
1879#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001880 path = PyBytes_AsString(opath);
1881 Py_BEGIN_ALLOW_THREADS
1882 res = (*statfunc)(path, &st);
1883 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001884
Victor Stinner8c62be82010-05-06 00:08:46 +00001885 if (res != 0) {
Martin v. Löwis14694662006-02-03 12:54:16 +00001886#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00001887 result = win32_error("stat", path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001888#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001889 result = posix_error_with_filename(path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001890#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001891 }
1892 else
Victor Stinner4195b5c2012-02-08 23:03:19 +01001893 result = _pystat_fromstructstat(&st);
Fred Drake699f3522000-06-29 21:12:41 +00001894
Victor Stinner8c62be82010-05-06 00:08:46 +00001895 Py_DECREF(opath);
1896 return result;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001897}
1898
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001899/* POSIX methods */
1900
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001901PyDoc_STRVAR(posix_access__doc__,
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001902"access(path, mode) -> True if granted, False otherwise\n\n\
Guido van Rossuma0b90752002-06-18 16:22:43 +00001903Use the real uid/gid to test for access to a path. Note that most\n\
1904operations will use the effective uid/gid, therefore this routine can\n\
1905be used in a suid/sgid environment to test if the invoking user has the\n\
1906specified access to the path. The mode argument can be F_OK to test\n\
1907existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001908
1909static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001910posix_access(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001911{
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001912 const char *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00001913 int mode;
1914
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001915#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00001916 DWORD attr;
Victor Stinnereb5657a2011-09-30 01:44:27 +02001917 PyObject *po;
Victor Stinner8c62be82010-05-06 00:08:46 +00001918 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02001919 wchar_t* wpath = PyUnicode_AsUnicode(po);
1920 if (wpath == NULL)
1921 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00001922 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02001923 attr = GetFileAttributesW(wpath);
Victor Stinner8c62be82010-05-06 00:08:46 +00001924 Py_END_ALLOW_THREADS
1925 goto finish;
1926 }
1927 /* Drop the argument parsing error as narrow strings
1928 are also valid. */
1929 PyErr_Clear();
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001930 if (!PyArg_ParseTuple(args, "yi:access", &path, &mode))
Victor Stinner8c62be82010-05-06 00:08:46 +00001931 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001932 if (win32_warn_bytes_api())
1933 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00001934 Py_BEGIN_ALLOW_THREADS
1935 attr = GetFileAttributesA(path);
1936 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001937finish:
Victor Stinner8c62be82010-05-06 00:08:46 +00001938 if (attr == 0xFFFFFFFF)
1939 /* File does not exist, or cannot read attributes */
1940 return PyBool_FromLong(0);
1941 /* Access is possible if either write access wasn't requested, or
1942 the file isn't read-only, or if it's a directory, as there are
1943 no read-only directories on Windows. */
1944 return PyBool_FromLong(!(mode & 2)
1945 || !(attr & FILE_ATTRIBUTE_READONLY)
1946 || (attr & FILE_ATTRIBUTE_DIRECTORY));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001947#else
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001948 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00001949 int res;
1950 if (!PyArg_ParseTuple(args, "O&i:access",
1951 PyUnicode_FSConverter, &opath, &mode))
1952 return NULL;
1953 path = PyBytes_AsString(opath);
1954 Py_BEGIN_ALLOW_THREADS
1955 res = access(path, mode);
1956 Py_END_ALLOW_THREADS
1957 Py_DECREF(opath);
1958 return PyBool_FromLong(res == 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001959#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001960}
1961
Guido van Rossumd371ff11999-01-25 16:12:23 +00001962#ifndef F_OK
1963#define F_OK 0
1964#endif
1965#ifndef R_OK
1966#define R_OK 4
1967#endif
1968#ifndef W_OK
1969#define W_OK 2
1970#endif
1971#ifndef X_OK
1972#define X_OK 1
1973#endif
1974
1975#ifdef HAVE_TTYNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001976PyDoc_STRVAR(posix_ttyname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001977"ttyname(fd) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001978Return the name of the terminal device connected to 'fd'.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001979
1980static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001981posix_ttyname(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001982{
Victor Stinner8c62be82010-05-06 00:08:46 +00001983 int id;
1984 char *ret;
Guido van Rossum94f6f721999-01-06 18:42:14 +00001985
Victor Stinner8c62be82010-05-06 00:08:46 +00001986 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
1987 return NULL;
Guido van Rossum94f6f721999-01-06 18:42:14 +00001988
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001989#if defined(__VMS)
Victor Stinner8c62be82010-05-06 00:08:46 +00001990 /* file descriptor 0 only, the default input device (stdin) */
1991 if (id == 0) {
1992 ret = ttyname();
1993 }
1994 else {
1995 ret = NULL;
1996 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001997#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001998 ret = ttyname(id);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001999#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00002000 if (ret == NULL)
2001 return posix_error();
Victor Stinner5fe6de82010-08-15 09:12:51 +00002002 return PyUnicode_DecodeFSDefault(ret);
Guido van Rossum94f6f721999-01-06 18:42:14 +00002003}
Guido van Rossumd371ff11999-01-25 16:12:23 +00002004#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00002005
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002006#ifdef HAVE_CTERMID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002007PyDoc_STRVAR(posix_ctermid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002008"ctermid() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002009Return the name of the controlling terminal for this process.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002010
2011static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00002012posix_ctermid(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002013{
Victor Stinner8c62be82010-05-06 00:08:46 +00002014 char *ret;
2015 char buffer[L_ctermid];
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002016
Greg Wardb48bc172000-03-01 21:51:56 +00002017#ifdef USE_CTERMID_R
Victor Stinner8c62be82010-05-06 00:08:46 +00002018 ret = ctermid_r(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002019#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002020 ret = ctermid(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002021#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00002022 if (ret == NULL)
2023 return posix_error();
Victor Stinner5fe6de82010-08-15 09:12:51 +00002024 return PyUnicode_DecodeFSDefault(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00002025}
2026#endif
2027
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002028PyDoc_STRVAR(posix_chdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002029"chdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002030Change the current working directory to the specified path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002031
Barry Warsaw53699e91996-12-10 23:23:01 +00002032static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002033posix_chdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002034{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002035#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00002036 return win32_1str(args, "chdir", "y:chdir", win32_chdir, "U:chdir", win32_wchdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002037#elif defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00002038 return posix_1str(args, "O&:chdir", _chdir2);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00002039#elif defined(__VMS)
Victor Stinner8c62be82010-05-06 00:08:46 +00002040 return posix_1str(args, "O&:chdir", (int (*)(const char *))chdir);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002041#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002042 return posix_1str(args, "O&:chdir", chdir);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002043#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002044}
2045
Fred Drake4d1e64b2002-04-15 19:40:07 +00002046#ifdef HAVE_FCHDIR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002047PyDoc_STRVAR(posix_fchdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002048"fchdir(fildes)\n\n\
Fred Drake4d1e64b2002-04-15 19:40:07 +00002049Change to the directory of the given file descriptor. fildes must be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002050opened on a directory, not a file.");
Fred Drake4d1e64b2002-04-15 19:40:07 +00002051
2052static PyObject *
2053posix_fchdir(PyObject *self, PyObject *fdobj)
2054{
Victor Stinner8c62be82010-05-06 00:08:46 +00002055 return posix_fildes(fdobj, fchdir);
Fred Drake4d1e64b2002-04-15 19:40:07 +00002056}
2057#endif /* HAVE_FCHDIR */
2058
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002059
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002060PyDoc_STRVAR(posix_chmod__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002061"chmod(path, mode)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002062Change the access permissions of a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002063
Barry Warsaw53699e91996-12-10 23:23:01 +00002064static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002065posix_chmod(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002066{
Victor Stinner8c62be82010-05-06 00:08:46 +00002067 PyObject *opath = NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002068 const char *path = NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00002069 int i;
2070 int res;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002071#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00002072 DWORD attr;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002073 PyObject *po;
Victor Stinner8c62be82010-05-06 00:08:46 +00002074 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02002075 wchar_t *wpath = PyUnicode_AsUnicode(po);
2076 if (wpath == NULL)
2077 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00002078 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02002079 attr = GetFileAttributesW(wpath);
Victor Stinner8c62be82010-05-06 00:08:46 +00002080 if (attr != 0xFFFFFFFF) {
2081 if (i & _S_IWRITE)
2082 attr &= ~FILE_ATTRIBUTE_READONLY;
2083 else
2084 attr |= FILE_ATTRIBUTE_READONLY;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002085 res = SetFileAttributesW(wpath, attr);
Victor Stinner8c62be82010-05-06 00:08:46 +00002086 }
2087 else
2088 res = 0;
2089 Py_END_ALLOW_THREADS
2090 if (!res)
Victor Stinnereb5657a2011-09-30 01:44:27 +02002091 return win32_error_object("chmod", po);
Victor Stinner8c62be82010-05-06 00:08:46 +00002092 Py_INCREF(Py_None);
2093 return Py_None;
2094 }
2095 /* Drop the argument parsing error as narrow strings
2096 are also valid. */
2097 PyErr_Clear();
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002098
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002099 if (!PyArg_ParseTuple(args, "yi:chmod", &path, &i))
Victor Stinner8c62be82010-05-06 00:08:46 +00002100 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002101 if (win32_warn_bytes_api())
2102 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00002103 Py_BEGIN_ALLOW_THREADS
2104 attr = GetFileAttributesA(path);
2105 if (attr != 0xFFFFFFFF) {
2106 if (i & _S_IWRITE)
2107 attr &= ~FILE_ATTRIBUTE_READONLY;
2108 else
2109 attr |= FILE_ATTRIBUTE_READONLY;
2110 res = SetFileAttributesA(path, attr);
2111 }
2112 else
2113 res = 0;
2114 Py_END_ALLOW_THREADS
2115 if (!res) {
2116 win32_error("chmod", path);
Victor Stinner8c62be82010-05-06 00:08:46 +00002117 return NULL;
2118 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002119 Py_INCREF(Py_None);
2120 return Py_None;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002121#else /* MS_WINDOWS */
Victor Stinner8c62be82010-05-06 00:08:46 +00002122 if (!PyArg_ParseTuple(args, "O&i:chmod", PyUnicode_FSConverter,
2123 &opath, &i))
2124 return NULL;
2125 path = PyBytes_AsString(opath);
2126 Py_BEGIN_ALLOW_THREADS
2127 res = chmod(path, i);
2128 Py_END_ALLOW_THREADS
2129 if (res < 0)
2130 return posix_error_with_allocated_filename(opath);
2131 Py_DECREF(opath);
2132 Py_INCREF(Py_None);
2133 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002134#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002135}
2136
Christian Heimes4e30a842007-11-30 22:12:06 +00002137#ifdef HAVE_FCHMOD
2138PyDoc_STRVAR(posix_fchmod__doc__,
2139"fchmod(fd, mode)\n\n\
2140Change the access permissions of the file given by file\n\
2141descriptor fd.");
2142
2143static PyObject *
2144posix_fchmod(PyObject *self, PyObject *args)
2145{
Victor Stinner8c62be82010-05-06 00:08:46 +00002146 int fd, mode, res;
2147 if (!PyArg_ParseTuple(args, "ii:fchmod", &fd, &mode))
2148 return NULL;
2149 Py_BEGIN_ALLOW_THREADS
2150 res = fchmod(fd, mode);
2151 Py_END_ALLOW_THREADS
2152 if (res < 0)
2153 return posix_error();
2154 Py_RETURN_NONE;
Christian Heimes4e30a842007-11-30 22:12:06 +00002155}
2156#endif /* HAVE_FCHMOD */
2157
2158#ifdef HAVE_LCHMOD
2159PyDoc_STRVAR(posix_lchmod__doc__,
2160"lchmod(path, mode)\n\n\
2161Change the access permissions of a file. If path is a symlink, this\n\
2162affects the link itself rather than the target.");
2163
2164static PyObject *
2165posix_lchmod(PyObject *self, PyObject *args)
2166{
Victor Stinner8c62be82010-05-06 00:08:46 +00002167 PyObject *opath;
2168 char *path;
2169 int i;
2170 int res;
2171 if (!PyArg_ParseTuple(args, "O&i:lchmod", PyUnicode_FSConverter,
2172 &opath, &i))
2173 return NULL;
2174 path = PyBytes_AsString(opath);
2175 Py_BEGIN_ALLOW_THREADS
2176 res = lchmod(path, i);
2177 Py_END_ALLOW_THREADS
2178 if (res < 0)
2179 return posix_error_with_allocated_filename(opath);
2180 Py_DECREF(opath);
2181 Py_RETURN_NONE;
Christian Heimes4e30a842007-11-30 22:12:06 +00002182}
2183#endif /* HAVE_LCHMOD */
2184
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002185
Thomas Wouterscf297e42007-02-23 15:07:44 +00002186#ifdef HAVE_CHFLAGS
2187PyDoc_STRVAR(posix_chflags__doc__,
2188"chflags(path, flags)\n\n\
2189Set file flags.");
2190
2191static PyObject *
2192posix_chflags(PyObject *self, PyObject *args)
2193{
Victor Stinner8c62be82010-05-06 00:08:46 +00002194 PyObject *opath;
2195 char *path;
2196 unsigned long flags;
2197 int res;
2198 if (!PyArg_ParseTuple(args, "O&k:chflags",
2199 PyUnicode_FSConverter, &opath, &flags))
2200 return NULL;
2201 path = PyBytes_AsString(opath);
2202 Py_BEGIN_ALLOW_THREADS
2203 res = chflags(path, flags);
2204 Py_END_ALLOW_THREADS
2205 if (res < 0)
2206 return posix_error_with_allocated_filename(opath);
2207 Py_DECREF(opath);
2208 Py_INCREF(Py_None);
2209 return Py_None;
Thomas Wouterscf297e42007-02-23 15:07:44 +00002210}
2211#endif /* HAVE_CHFLAGS */
2212
2213#ifdef HAVE_LCHFLAGS
2214PyDoc_STRVAR(posix_lchflags__doc__,
2215"lchflags(path, flags)\n\n\
2216Set file flags.\n\
2217This function will not follow symbolic links.");
2218
2219static PyObject *
2220posix_lchflags(PyObject *self, PyObject *args)
2221{
Victor Stinner8c62be82010-05-06 00:08:46 +00002222 PyObject *opath;
2223 char *path;
2224 unsigned long flags;
2225 int res;
2226 if (!PyArg_ParseTuple(args, "O&k:lchflags",
2227 PyUnicode_FSConverter, &opath, &flags))
2228 return NULL;
2229 path = PyBytes_AsString(opath);
2230 Py_BEGIN_ALLOW_THREADS
2231 res = lchflags(path, flags);
2232 Py_END_ALLOW_THREADS
2233 if (res < 0)
2234 return posix_error_with_allocated_filename(opath);
2235 Py_DECREF(opath);
2236 Py_INCREF(Py_None);
2237 return Py_None;
Thomas Wouterscf297e42007-02-23 15:07:44 +00002238}
2239#endif /* HAVE_LCHFLAGS */
2240
Martin v. Löwis244edc82001-10-04 22:44:26 +00002241#ifdef HAVE_CHROOT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002242PyDoc_STRVAR(posix_chroot__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002243"chroot(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002244Change root directory to path.");
Martin v. Löwis244edc82001-10-04 22:44:26 +00002245
2246static PyObject *
2247posix_chroot(PyObject *self, PyObject *args)
2248{
Victor Stinner8c62be82010-05-06 00:08:46 +00002249 return posix_1str(args, "O&:chroot", chroot);
Martin v. Löwis244edc82001-10-04 22:44:26 +00002250}
2251#endif
2252
Guido van Rossum21142a01999-01-08 21:05:37 +00002253#ifdef HAVE_FSYNC
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002254PyDoc_STRVAR(posix_fsync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002255"fsync(fildes)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002256force write of file with filedescriptor to disk.");
Guido van Rossum21142a01999-01-08 21:05:37 +00002257
2258static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00002259posix_fsync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00002260{
Stefan Krah0e803b32010-11-26 16:16:47 +00002261 return posix_fildes(fdobj, fsync);
Guido van Rossum21142a01999-01-08 21:05:37 +00002262}
2263#endif /* HAVE_FSYNC */
2264
Ross Lagerwall7807c352011-03-17 20:20:30 +02002265#ifdef HAVE_SYNC
2266PyDoc_STRVAR(posix_sync__doc__,
2267"sync()\n\n\
2268Force write of everything to disk.");
2269
2270static PyObject *
2271posix_sync(PyObject *self, PyObject *noargs)
2272{
2273 Py_BEGIN_ALLOW_THREADS
2274 sync();
2275 Py_END_ALLOW_THREADS
2276 Py_RETURN_NONE;
2277}
2278#endif
2279
Guido van Rossum21142a01999-01-08 21:05:37 +00002280#ifdef HAVE_FDATASYNC
Guido van Rossumecc23b02000-09-22 16:01:05 +00002281
Guido van Rossum7f58e2e2000-09-22 17:26:14 +00002282#ifdef __hpux
Guido van Rossumecc23b02000-09-22 16:01:05 +00002283extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
2284#endif
2285
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002286PyDoc_STRVAR(posix_fdatasync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002287"fdatasync(fildes)\n\n\
Guido van Rossum21142a01999-01-08 21:05:37 +00002288force write of file with filedescriptor to disk.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002289 does not force update of metadata.");
Guido van Rossum21142a01999-01-08 21:05:37 +00002290
2291static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00002292posix_fdatasync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00002293{
Stefan Krah0e803b32010-11-26 16:16:47 +00002294 return posix_fildes(fdobj, fdatasync);
Guido van Rossum21142a01999-01-08 21:05:37 +00002295}
2296#endif /* HAVE_FDATASYNC */
2297
2298
Fredrik Lundh10723342000-07-10 16:38:09 +00002299#ifdef HAVE_CHOWN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002300PyDoc_STRVAR(posix_chown__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002301"chown(path, uid, gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002302Change the owner and group id of path to the numeric uid and gid.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002303
Barry Warsaw53699e91996-12-10 23:23:01 +00002304static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002305posix_chown(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002306{
Victor Stinner8c62be82010-05-06 00:08:46 +00002307 PyObject *opath;
2308 char *path;
2309 long uid, gid;
2310 int res;
2311 if (!PyArg_ParseTuple(args, "O&ll:chown",
2312 PyUnicode_FSConverter, &opath,
2313 &uid, &gid))
2314 return NULL;
2315 path = PyBytes_AsString(opath);
2316 Py_BEGIN_ALLOW_THREADS
2317 res = chown(path, (uid_t) uid, (gid_t) gid);
2318 Py_END_ALLOW_THREADS
2319 if (res < 0)
2320 return posix_error_with_allocated_filename(opath);
2321 Py_DECREF(opath);
2322 Py_INCREF(Py_None);
2323 return Py_None;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002324}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002325#endif /* HAVE_CHOWN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002326
Christian Heimes4e30a842007-11-30 22:12:06 +00002327#ifdef HAVE_FCHOWN
2328PyDoc_STRVAR(posix_fchown__doc__,
2329"fchown(fd, uid, gid)\n\n\
2330Change the owner and group id of the file given by file descriptor\n\
2331fd to the numeric uid and gid.");
2332
2333static PyObject *
2334posix_fchown(PyObject *self, PyObject *args)
2335{
Victor Stinner8c62be82010-05-06 00:08:46 +00002336 int fd;
2337 long uid, gid;
2338 int res;
Victor Stinner26de69d2011-06-17 15:15:38 +02002339 if (!PyArg_ParseTuple(args, "ill:fchown", &fd, &uid, &gid))
Victor Stinner8c62be82010-05-06 00:08:46 +00002340 return NULL;
2341 Py_BEGIN_ALLOW_THREADS
2342 res = fchown(fd, (uid_t) uid, (gid_t) gid);
2343 Py_END_ALLOW_THREADS
2344 if (res < 0)
2345 return posix_error();
2346 Py_RETURN_NONE;
Christian Heimes4e30a842007-11-30 22:12:06 +00002347}
2348#endif /* HAVE_FCHOWN */
2349
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002350#ifdef HAVE_LCHOWN
2351PyDoc_STRVAR(posix_lchown__doc__,
2352"lchown(path, uid, gid)\n\n\
2353Change the owner and group id of path to the numeric uid and gid.\n\
2354This function will not follow symbolic links.");
2355
2356static PyObject *
2357posix_lchown(PyObject *self, PyObject *args)
2358{
Victor Stinner8c62be82010-05-06 00:08:46 +00002359 PyObject *opath;
2360 char *path;
2361 long uid, gid;
2362 int res;
2363 if (!PyArg_ParseTuple(args, "O&ll:lchown",
2364 PyUnicode_FSConverter, &opath,
2365 &uid, &gid))
2366 return NULL;
2367 path = PyBytes_AsString(opath);
2368 Py_BEGIN_ALLOW_THREADS
2369 res = lchown(path, (uid_t) uid, (gid_t) gid);
2370 Py_END_ALLOW_THREADS
2371 if (res < 0)
2372 return posix_error_with_allocated_filename(opath);
2373 Py_DECREF(opath);
2374 Py_INCREF(Py_None);
2375 return Py_None;
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002376}
2377#endif /* HAVE_LCHOWN */
2378
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002379
Guido van Rossum36bc6801995-06-14 22:54:23 +00002380#ifdef HAVE_GETCWD
Barry Warsaw53699e91996-12-10 23:23:01 +00002381static PyObject *
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002382posix_getcwd(int use_bytes)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002383{
Victor Stinner8c62be82010-05-06 00:08:46 +00002384 char buf[1026];
2385 char *res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002386
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002387#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00002388 if (!use_bytes) {
2389 wchar_t wbuf[1026];
2390 wchar_t *wbuf2 = wbuf;
2391 PyObject *resobj;
2392 DWORD len;
2393 Py_BEGIN_ALLOW_THREADS
2394 len = GetCurrentDirectoryW(sizeof wbuf/ sizeof wbuf[0], wbuf);
2395 /* If the buffer is large enough, len does not include the
2396 terminating \0. If the buffer is too small, len includes
2397 the space needed for the terminator. */
2398 if (len >= sizeof wbuf/ sizeof wbuf[0]) {
2399 wbuf2 = malloc(len * sizeof(wchar_t));
2400 if (wbuf2)
2401 len = GetCurrentDirectoryW(len, wbuf2);
2402 }
2403 Py_END_ALLOW_THREADS
2404 if (!wbuf2) {
2405 PyErr_NoMemory();
2406 return NULL;
2407 }
2408 if (!len) {
2409 if (wbuf2 != wbuf) free(wbuf2);
2410 return win32_error("getcwdu", NULL);
2411 }
2412 resobj = PyUnicode_FromWideChar(wbuf2, len);
2413 if (wbuf2 != wbuf) free(wbuf2);
2414 return resobj;
2415 }
Victor Stinnerf7c5ae22011-11-16 23:43:07 +01002416
2417 if (win32_warn_bytes_api())
2418 return NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002419#endif
2420
Victor Stinner8c62be82010-05-06 00:08:46 +00002421 Py_BEGIN_ALLOW_THREADS
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002422#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00002423 res = _getcwd2(buf, sizeof buf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002424#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002425 res = getcwd(buf, sizeof buf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002426#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00002427 Py_END_ALLOW_THREADS
2428 if (res == NULL)
2429 return posix_error();
2430 if (use_bytes)
2431 return PyBytes_FromStringAndSize(buf, strlen(buf));
Victor Stinner97c18ab2010-05-07 16:34:53 +00002432 return PyUnicode_DecodeFSDefault(buf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002433}
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002434
2435PyDoc_STRVAR(posix_getcwd__doc__,
2436"getcwd() -> path\n\n\
2437Return a unicode string representing the current working directory.");
2438
2439static PyObject *
2440posix_getcwd_unicode(PyObject *self)
2441{
2442 return posix_getcwd(0);
2443}
2444
2445PyDoc_STRVAR(posix_getcwdb__doc__,
2446"getcwdb() -> path\n\n\
2447Return a bytes string representing the current working directory.");
2448
2449static PyObject *
2450posix_getcwd_bytes(PyObject *self)
2451{
2452 return posix_getcwd(1);
2453}
Guido van Rossum36bc6801995-06-14 22:54:23 +00002454#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002455
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002456
Guido van Rossumb6775db1994-08-01 11:34:53 +00002457#ifdef HAVE_LINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002458PyDoc_STRVAR(posix_link__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002459"link(src, dst)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002460Create a hard link to a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002461
Barry Warsaw53699e91996-12-10 23:23:01 +00002462static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002463posix_link(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002464{
Victor Stinner8c62be82010-05-06 00:08:46 +00002465 return posix_2str(args, "O&O&:link", link);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002466}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002467#endif /* HAVE_LINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002468
Brian Curtin1b9df392010-11-24 20:24:31 +00002469#ifdef MS_WINDOWS
2470PyDoc_STRVAR(win32_link__doc__,
2471"link(src, dst)\n\n\
2472Create a hard link to a file.");
2473
2474static PyObject *
2475win32_link(PyObject *self, PyObject *args)
2476{
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002477 PyObject *src, *dst;
2478 BOOL ok;
Brian Curtin1b9df392010-11-24 20:24:31 +00002479
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002480 if (PyArg_ParseTuple(args, "UU:link", &src, &dst))
Victor Stinnereb5657a2011-09-30 01:44:27 +02002481 {
2482 wchar_t *wsrc, *wdst;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002483
2484 wsrc = PyUnicode_AsUnicode(src);
Victor Stinnereb5657a2011-09-30 01:44:27 +02002485 if (wsrc == NULL)
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002486 goto error;
2487 wdst = PyUnicode_AsUnicode(dst);
Victor Stinnereb5657a2011-09-30 01:44:27 +02002488 if (wdst == NULL)
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002489 goto error;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002490
Brian Curtinfc889c42010-11-28 23:59:46 +00002491 Py_BEGIN_ALLOW_THREADS
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002492 ok = CreateHardLinkW(wdst, wsrc, NULL);
Brian Curtinfc889c42010-11-28 23:59:46 +00002493 Py_END_ALLOW_THREADS
2494
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002495 if (!ok)
Brian Curtinfc889c42010-11-28 23:59:46 +00002496 return win32_error("link", NULL);
Brian Curtinfc889c42010-11-28 23:59:46 +00002497 Py_RETURN_NONE;
2498 }
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002499 else {
2500 PyErr_Clear();
2501 if (!PyArg_ParseTuple(args, "O&O&:link",
2502 PyUnicode_FSConverter, &src,
2503 PyUnicode_FSConverter, &dst))
2504 return NULL;
Brian Curtinfc889c42010-11-28 23:59:46 +00002505
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002506 if (win32_warn_bytes_api())
2507 goto error;
Brian Curtinfc889c42010-11-28 23:59:46 +00002508
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002509 Py_BEGIN_ALLOW_THREADS
2510 ok = CreateHardLinkA(PyBytes_AS_STRING(dst),
2511 PyBytes_AS_STRING(src),
2512 NULL);
2513 Py_END_ALLOW_THREADS
2514
2515 Py_XDECREF(src);
2516 Py_XDECREF(dst);
2517
2518 if (!ok)
2519 return win32_error("link", NULL);
2520 Py_RETURN_NONE;
2521
2522 error:
2523 Py_XDECREF(src);
2524 Py_XDECREF(dst);
Brian Curtin1b9df392010-11-24 20:24:31 +00002525 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002526 }
Brian Curtin1b9df392010-11-24 20:24:31 +00002527}
2528#endif /* MS_WINDOWS */
2529
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002530
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002531PyDoc_STRVAR(posix_listdir__doc__,
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002532"listdir([path]) -> list_of_strings\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002533Return a list containing the names of the entries in the directory.\n\
2534\n\
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002535 path: path of directory to list (default: '.')\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002536\n\
2537The list is in arbitrary order. It does not include the special\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002538entries '.' and '..' even if they are present in the directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002539
Barry Warsaw53699e91996-12-10 23:23:01 +00002540static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002541posix_listdir(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002542{
Victor Stinner8c62be82010-05-06 00:08:46 +00002543 /* XXX Should redo this putting the (now four) versions of opendir
2544 in separate files instead of having them all here... */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002545#if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002546
Victor Stinner8c62be82010-05-06 00:08:46 +00002547 PyObject *d, *v;
2548 HANDLE hFindFile;
2549 BOOL result;
2550 WIN32_FIND_DATA FileData;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002551 const char *path;
2552 Py_ssize_t pathlen;
Victor Stinner8c62be82010-05-06 00:08:46 +00002553 char namebuf[MAX_PATH+5]; /* Overallocate for \\*.*\0 */
2554 char *bufptr = namebuf;
2555 Py_ssize_t len = sizeof(namebuf)-5; /* only claim to have space for MAX_PATH */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002556
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002557 PyObject *po = NULL;
2558 if (PyArg_ParseTuple(args, "|U:listdir", &po)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002559 WIN32_FIND_DATAW wFileData;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002560 wchar_t *wnamebuf, *po_wchars;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00002561
Antoine Pitrou3d330ad2010-09-14 10:08:08 +00002562 if (po == NULL) { /* Default arg: "." */
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002563 po_wchars = L".";
2564 len = 1;
2565 } else {
Victor Stinnerbeac78b2011-10-11 21:55:01 +02002566 po_wchars = PyUnicode_AsUnicodeAndSize(po, &len);
Victor Stinnereb5657a2011-09-30 01:44:27 +02002567 if (po_wchars == NULL)
2568 return NULL;
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002569 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002570 /* Overallocate for \\*.*\0 */
Victor Stinner8c62be82010-05-06 00:08:46 +00002571 wnamebuf = malloc((len + 5) * sizeof(wchar_t));
2572 if (!wnamebuf) {
2573 PyErr_NoMemory();
2574 return NULL;
2575 }
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002576 wcscpy(wnamebuf, po_wchars);
Victor Stinner8c62be82010-05-06 00:08:46 +00002577 if (len > 0) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02002578 wchar_t wch = wnamebuf[len-1];
Victor Stinner8c62be82010-05-06 00:08:46 +00002579 if (wch != L'/' && wch != L'\\' && wch != L':')
2580 wnamebuf[len++] = L'\\';
2581 wcscpy(wnamebuf + len, L"*.*");
2582 }
2583 if ((d = PyList_New(0)) == NULL) {
2584 free(wnamebuf);
2585 return NULL;
2586 }
Antoine Pitroub73caab2010-08-09 23:39:31 +00002587 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002588 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
Antoine Pitroub73caab2010-08-09 23:39:31 +00002589 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002590 if (hFindFile == INVALID_HANDLE_VALUE) {
2591 int error = GetLastError();
2592 if (error == ERROR_FILE_NOT_FOUND) {
2593 free(wnamebuf);
2594 return d;
2595 }
2596 Py_DECREF(d);
2597 win32_error_unicode("FindFirstFileW", wnamebuf);
2598 free(wnamebuf);
2599 return NULL;
2600 }
2601 do {
2602 /* Skip over . and .. */
2603 if (wcscmp(wFileData.cFileName, L".") != 0 &&
2604 wcscmp(wFileData.cFileName, L"..") != 0) {
Victor Stinner9d3b93b2011-11-22 02:27:30 +01002605 v = PyUnicode_FromWideChar(wFileData.cFileName, wcslen(wFileData.cFileName));
Victor Stinner8c62be82010-05-06 00:08:46 +00002606 if (v == NULL) {
2607 Py_DECREF(d);
2608 d = NULL;
2609 break;
2610 }
2611 if (PyList_Append(d, v) != 0) {
2612 Py_DECREF(v);
2613 Py_DECREF(d);
2614 d = NULL;
2615 break;
2616 }
2617 Py_DECREF(v);
2618 }
2619 Py_BEGIN_ALLOW_THREADS
2620 result = FindNextFileW(hFindFile, &wFileData);
2621 Py_END_ALLOW_THREADS
2622 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2623 it got to the end of the directory. */
2624 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2625 Py_DECREF(d);
2626 win32_error_unicode("FindNextFileW", wnamebuf);
2627 FindClose(hFindFile);
2628 free(wnamebuf);
2629 return NULL;
2630 }
2631 } while (result == TRUE);
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002632
Victor Stinner8c62be82010-05-06 00:08:46 +00002633 if (FindClose(hFindFile) == FALSE) {
2634 Py_DECREF(d);
2635 win32_error_unicode("FindClose", wnamebuf);
2636 free(wnamebuf);
2637 return NULL;
2638 }
2639 free(wnamebuf);
2640 return d;
2641 }
2642 /* Drop the argument parsing error as narrow strings
2643 are also valid. */
2644 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002645
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002646 if (!PyArg_ParseTuple(args, "y#:listdir", &path, &pathlen))
Victor Stinner8c62be82010-05-06 00:08:46 +00002647 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002648 if (win32_warn_bytes_api())
2649 return NULL;
2650 if (pathlen+1 > MAX_PATH) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002651 PyErr_SetString(PyExc_ValueError, "path too long");
Victor Stinner8c62be82010-05-06 00:08:46 +00002652 return NULL;
2653 }
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002654 strcpy(namebuf, path);
2655 len = pathlen;
Victor Stinner8c62be82010-05-06 00:08:46 +00002656 if (len > 0) {
2657 char ch = namebuf[len-1];
2658 if (ch != SEP && ch != ALTSEP && ch != ':')
2659 namebuf[len++] = '/';
2660 strcpy(namebuf + len, "*.*");
2661 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002662
Victor Stinner8c62be82010-05-06 00:08:46 +00002663 if ((d = PyList_New(0)) == NULL)
2664 return NULL;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002665
Antoine Pitroub73caab2010-08-09 23:39:31 +00002666 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002667 hFindFile = FindFirstFile(namebuf, &FileData);
Antoine Pitroub73caab2010-08-09 23:39:31 +00002668 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002669 if (hFindFile == INVALID_HANDLE_VALUE) {
2670 int error = GetLastError();
2671 if (error == ERROR_FILE_NOT_FOUND)
2672 return d;
2673 Py_DECREF(d);
2674 return win32_error("FindFirstFile", namebuf);
2675 }
2676 do {
2677 /* Skip over . and .. */
2678 if (strcmp(FileData.cFileName, ".") != 0 &&
2679 strcmp(FileData.cFileName, "..") != 0) {
2680 v = PyBytes_FromString(FileData.cFileName);
2681 if (v == NULL) {
2682 Py_DECREF(d);
2683 d = NULL;
2684 break;
2685 }
2686 if (PyList_Append(d, v) != 0) {
2687 Py_DECREF(v);
2688 Py_DECREF(d);
2689 d = NULL;
2690 break;
2691 }
2692 Py_DECREF(v);
2693 }
2694 Py_BEGIN_ALLOW_THREADS
2695 result = FindNextFile(hFindFile, &FileData);
2696 Py_END_ALLOW_THREADS
2697 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2698 it got to the end of the directory. */
2699 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2700 Py_DECREF(d);
2701 win32_error("FindNextFile", namebuf);
2702 FindClose(hFindFile);
2703 return NULL;
2704 }
2705 } while (result == TRUE);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002706
Victor Stinner8c62be82010-05-06 00:08:46 +00002707 if (FindClose(hFindFile) == FALSE) {
2708 Py_DECREF(d);
2709 return win32_error("FindClose", namebuf);
2710 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002711
Victor Stinner8c62be82010-05-06 00:08:46 +00002712 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002713
Tim Peters0bb44a42000-09-15 07:44:49 +00002714#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002715
2716#ifndef MAX_PATH
2717#define MAX_PATH CCHMAXPATH
2718#endif
Martin v. Löwis011e8422009-05-05 04:43:17 +00002719 PyObject *oname;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002720 char *name, *pt;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00002721 Py_ssize_t len;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002722 PyObject *d, *v;
2723 char namebuf[MAX_PATH+5];
2724 HDIR hdir = 1;
2725 ULONG srchcnt = 1;
2726 FILEFINDBUF3 ep;
2727 APIRET rc;
2728
Victor Stinner8c62be82010-05-06 00:08:46 +00002729 if (!PyArg_ParseTuple(args, "O&:listdir",
Martin v. Löwis011e8422009-05-05 04:43:17 +00002730 PyUnicode_FSConverter, &oname))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002731 return NULL;
Victor Stinnerdcb24032010-04-22 12:08:36 +00002732 name = PyBytes_AsString(oname);
2733 len = PyBytes_GET_SIZE(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002734 if (len >= MAX_PATH) {
Victor Stinnerdcb24032010-04-22 12:08:36 +00002735 Py_DECREF(oname);
Neal Norwitz6c913782007-10-14 03:23:09 +00002736 PyErr_SetString(PyExc_ValueError, "path too long");
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002737 return NULL;
2738 }
2739 strcpy(namebuf, name);
2740 for (pt = namebuf; *pt; pt++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002741 if (*pt == ALTSEP)
2742 *pt = SEP;
2743 if (namebuf[len-1] != SEP)
2744 namebuf[len++] = SEP;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002745 strcpy(namebuf + len, "*.*");
2746
Neal Norwitz6c913782007-10-14 03:23:09 +00002747 if ((d = PyList_New(0)) == NULL) {
Victor Stinnerdcb24032010-04-22 12:08:36 +00002748 Py_DECREF(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002749 return NULL;
Alexandre Vassalotti4167ebc2007-10-14 02:54:41 +00002750 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002751
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002752 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
2753 &hdir, /* Handle to Use While Search Directory */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002754 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002755 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
2756 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
2757 FIL_STANDARD); /* Format of Entry (EAs or Not) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002758
2759 if (rc != NO_ERROR) {
2760 errno = ENOENT;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002761 return posix_error_with_allocated_filename(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002762 }
2763
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002764 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002765 do {
2766 if (ep.achName[0] == '.'
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002767 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002768 continue; /* Skip Over "." and ".." Names */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002769
2770 strcpy(namebuf, ep.achName);
2771
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002772 /* Leave Case of Name Alone -- In Native Form */
2773 /* (Removed Forced Lowercasing Code) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002774
Christian Heimes72b710a2008-05-26 13:28:38 +00002775 v = PyBytes_FromString(namebuf);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002776 if (v == NULL) {
2777 Py_DECREF(d);
2778 d = NULL;
2779 break;
2780 }
2781 if (PyList_Append(d, v) != 0) {
2782 Py_DECREF(v);
2783 Py_DECREF(d);
2784 d = NULL;
2785 break;
2786 }
2787 Py_DECREF(v);
2788 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
2789 }
2790
Victor Stinnerdcb24032010-04-22 12:08:36 +00002791 Py_DECREF(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002792 return d;
2793#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002794 PyObject *oname;
2795 char *name;
2796 PyObject *d, *v;
2797 DIR *dirp;
2798 struct dirent *ep;
2799 int arg_is_unicode = 1;
Just van Rossum96b1c902003-03-03 17:32:15 +00002800
Victor Stinner8c62be82010-05-06 00:08:46 +00002801 errno = 0;
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002802 /* v is never read, so it does not need to be initialized yet. */
2803 if (!PyArg_ParseTuple(args, "|U:listdir", &v)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002804 arg_is_unicode = 0;
2805 PyErr_Clear();
2806 }
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002807 oname = NULL;
2808 if (!PyArg_ParseTuple(args, "|O&:listdir", PyUnicode_FSConverter, &oname))
Victor Stinner8c62be82010-05-06 00:08:46 +00002809 return NULL;
Antoine Pitrou3d330ad2010-09-14 10:08:08 +00002810 if (oname == NULL) { /* Default arg: "." */
Stefan Krah0e803b32010-11-26 16:16:47 +00002811 oname = PyBytes_FromString(".");
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002812 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002813 name = PyBytes_AsString(oname);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002814 Py_BEGIN_ALLOW_THREADS
2815 dirp = opendir(name);
2816 Py_END_ALLOW_THREADS
2817 if (dirp == NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002818 return posix_error_with_allocated_filename(oname);
2819 }
2820 if ((d = PyList_New(0)) == NULL) {
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002821 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002822 closedir(dirp);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002823 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002824 Py_DECREF(oname);
2825 return NULL;
2826 }
2827 for (;;) {
2828 errno = 0;
2829 Py_BEGIN_ALLOW_THREADS
2830 ep = readdir(dirp);
2831 Py_END_ALLOW_THREADS
2832 if (ep == NULL) {
2833 if (errno == 0) {
2834 break;
2835 } else {
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002836 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002837 closedir(dirp);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002838 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002839 Py_DECREF(d);
2840 return posix_error_with_allocated_filename(oname);
2841 }
2842 }
2843 if (ep->d_name[0] == '.' &&
2844 (NAMLEN(ep) == 1 ||
2845 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
2846 continue;
Victor Stinnera45598a2010-05-14 16:35:39 +00002847 if (arg_is_unicode)
2848 v = PyUnicode_DecodeFSDefaultAndSize(ep->d_name, NAMLEN(ep));
2849 else
2850 v = PyBytes_FromStringAndSize(ep->d_name, NAMLEN(ep));
Victor Stinner8c62be82010-05-06 00:08:46 +00002851 if (v == NULL) {
Victor Stinnera45598a2010-05-14 16:35:39 +00002852 Py_CLEAR(d);
Victor Stinner8c62be82010-05-06 00:08:46 +00002853 break;
2854 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002855 if (PyList_Append(d, v) != 0) {
2856 Py_DECREF(v);
Victor Stinnera45598a2010-05-14 16:35:39 +00002857 Py_CLEAR(d);
Victor Stinner8c62be82010-05-06 00:08:46 +00002858 break;
2859 }
2860 Py_DECREF(v);
2861 }
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002862 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002863 closedir(dirp);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002864 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002865 Py_DECREF(oname);
Guido van Rossum0ee42cd1991-04-08 21:01:03 +00002866
Victor Stinner8c62be82010-05-06 00:08:46 +00002867 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002868
Tim Peters0bb44a42000-09-15 07:44:49 +00002869#endif /* which OS */
2870} /* end of posix_listdir */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002871
Antoine Pitrou8250e232011-02-25 23:41:16 +00002872#ifdef HAVE_FDOPENDIR
Charles-François Natali77940902012-02-06 19:54:48 +01002873PyDoc_STRVAR(posix_flistdir__doc__,
2874"flistdir(fd) -> list_of_strings\n\n\
Charles-François Natali76961fa2012-01-10 20:25:09 +01002875Like listdir(), but uses a file descriptor instead.");
Antoine Pitrou8250e232011-02-25 23:41:16 +00002876
2877static PyObject *
Charles-François Natali77940902012-02-06 19:54:48 +01002878posix_flistdir(PyObject *self, PyObject *args)
Antoine Pitrou8250e232011-02-25 23:41:16 +00002879{
2880 PyObject *d, *v;
2881 DIR *dirp;
2882 struct dirent *ep;
2883 int fd;
2884
2885 errno = 0;
Charles-François Natali77940902012-02-06 19:54:48 +01002886 if (!PyArg_ParseTuple(args, "i:flistdir", &fd))
Antoine Pitrou8250e232011-02-25 23:41:16 +00002887 return NULL;
Charles-François Natali76961fa2012-01-10 20:25:09 +01002888 /* closedir() closes the FD, so we duplicate it */
2889 fd = dup(fd);
2890 if (fd < 0)
2891 return posix_error();
Antoine Pitrou8250e232011-02-25 23:41:16 +00002892 Py_BEGIN_ALLOW_THREADS
2893 dirp = fdopendir(fd);
2894 Py_END_ALLOW_THREADS
2895 if (dirp == NULL) {
2896 close(fd);
2897 return posix_error();
2898 }
2899 if ((d = PyList_New(0)) == NULL) {
2900 Py_BEGIN_ALLOW_THREADS
2901 closedir(dirp);
2902 Py_END_ALLOW_THREADS
2903 return NULL;
2904 }
2905 for (;;) {
2906 errno = 0;
2907 Py_BEGIN_ALLOW_THREADS
2908 ep = readdir(dirp);
2909 Py_END_ALLOW_THREADS
2910 if (ep == NULL) {
2911 if (errno == 0) {
2912 break;
2913 } else {
2914 Py_BEGIN_ALLOW_THREADS
Charles-François Natalif2840a82012-01-08 20:30:47 +01002915 rewinddir(dirp);
Antoine Pitrou8250e232011-02-25 23:41:16 +00002916 closedir(dirp);
2917 Py_END_ALLOW_THREADS
2918 Py_DECREF(d);
2919 return posix_error();
2920 }
2921 }
2922 if (ep->d_name[0] == '.' &&
2923 (NAMLEN(ep) == 1 ||
2924 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
2925 continue;
2926 v = PyUnicode_DecodeFSDefaultAndSize(ep->d_name, NAMLEN(ep));
2927 if (v == NULL) {
2928 Py_CLEAR(d);
2929 break;
2930 }
2931 if (PyList_Append(d, v) != 0) {
2932 Py_DECREF(v);
2933 Py_CLEAR(d);
2934 break;
2935 }
2936 Py_DECREF(v);
2937 }
2938 Py_BEGIN_ALLOW_THREADS
Charles-François Natalif2840a82012-01-08 20:30:47 +01002939 rewinddir(dirp);
Antoine Pitrou8250e232011-02-25 23:41:16 +00002940 closedir(dirp);
2941 Py_END_ALLOW_THREADS
2942
2943 return d;
2944}
2945#endif
2946
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002947#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00002948/* A helper function for abspath on win32 */
2949static PyObject *
2950posix__getfullpathname(PyObject *self, PyObject *args)
2951{
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002952 const char *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00002953 char outbuf[MAX_PATH*2];
2954 char *temp;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002955 PyObject *po;
2956
2957 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po))
2958 {
2959 wchar_t *wpath;
2960 wchar_t woutbuf[MAX_PATH*2], *woutbufp = woutbuf;
2961 wchar_t *wtemp;
Victor Stinner8c62be82010-05-06 00:08:46 +00002962 DWORD result;
2963 PyObject *v;
Victor Stinnereb5657a2011-09-30 01:44:27 +02002964
2965 wpath = PyUnicode_AsUnicode(po);
2966 if (wpath == NULL)
2967 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00002968 result = GetFullPathNameW(wpath,
Victor Stinner63941882011-09-29 00:42:28 +02002969 Py_ARRAY_LENGTH(woutbuf),
Victor Stinner8c62be82010-05-06 00:08:46 +00002970 woutbuf, &wtemp);
Victor Stinner63941882011-09-29 00:42:28 +02002971 if (result > Py_ARRAY_LENGTH(woutbuf)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02002972 woutbufp = malloc(result * sizeof(wchar_t));
Victor Stinner8c62be82010-05-06 00:08:46 +00002973 if (!woutbufp)
2974 return PyErr_NoMemory();
2975 result = GetFullPathNameW(wpath, result, woutbufp, &wtemp);
2976 }
2977 if (result)
Victor Stinner9d3b93b2011-11-22 02:27:30 +01002978 v = PyUnicode_FromWideChar(woutbufp, wcslen(woutbufp));
Victor Stinner8c62be82010-05-06 00:08:46 +00002979 else
Victor Stinnereb5657a2011-09-30 01:44:27 +02002980 v = win32_error_object("GetFullPathNameW", po);
Victor Stinner8c62be82010-05-06 00:08:46 +00002981 if (woutbufp != woutbuf)
2982 free(woutbufp);
2983 return v;
2984 }
2985 /* Drop the argument parsing error as narrow strings
2986 are also valid. */
2987 PyErr_Clear();
Victor Stinnereb5657a2011-09-30 01:44:27 +02002988
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002989 if (!PyArg_ParseTuple (args, "y:_getfullpathname",
2990 &path))
Victor Stinner8c62be82010-05-06 00:08:46 +00002991 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002992 if (win32_warn_bytes_api())
2993 return NULL;
Victor Stinner63941882011-09-29 00:42:28 +02002994 if (!GetFullPathName(path, Py_ARRAY_LENGTH(outbuf),
Victor Stinner8c62be82010-05-06 00:08:46 +00002995 outbuf, &temp)) {
2996 win32_error("GetFullPathName", path);
Victor Stinner8c62be82010-05-06 00:08:46 +00002997 return NULL;
2998 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002999 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
3000 return PyUnicode_Decode(outbuf, strlen(outbuf),
3001 Py_FileSystemDefaultEncoding, NULL);
3002 }
3003 return PyBytes_FromString(outbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00003004} /* end of posix__getfullpathname */
Brian Curtind40e6f72010-07-08 21:39:08 +00003005
Brian Curtind25aef52011-06-13 15:16:04 -05003006
Brian Curtinf5e76d02010-11-24 13:14:05 +00003007
Brian Curtind40e6f72010-07-08 21:39:08 +00003008/* A helper function for samepath on windows */
3009static PyObject *
3010posix__getfinalpathname(PyObject *self, PyObject *args)
3011{
3012 HANDLE hFile;
3013 int buf_size;
3014 wchar_t *target_path;
3015 int result_length;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003016 PyObject *po, *result;
Brian Curtind40e6f72010-07-08 21:39:08 +00003017 wchar_t *path;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00003018
Victor Stinnereb5657a2011-09-30 01:44:27 +02003019 if (!PyArg_ParseTuple(args, "U|:_getfinalpathname", &po))
Brian Curtind40e6f72010-07-08 21:39:08 +00003020 return NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003021 path = PyUnicode_AsUnicode(po);
3022 if (path == NULL)
3023 return NULL;
Brian Curtind40e6f72010-07-08 21:39:08 +00003024
3025 if(!check_GetFinalPathNameByHandle()) {
3026 /* If the OS doesn't have GetFinalPathNameByHandle, return a
3027 NotImplementedError. */
3028 return PyErr_Format(PyExc_NotImplementedError,
3029 "GetFinalPathNameByHandle not available on this platform");
3030 }
3031
3032 hFile = CreateFileW(
3033 path,
3034 0, /* desired access */
3035 0, /* share mode */
3036 NULL, /* security attributes */
3037 OPEN_EXISTING,
3038 /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
3039 FILE_FLAG_BACKUP_SEMANTICS,
3040 NULL);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00003041
Victor Stinnereb5657a2011-09-30 01:44:27 +02003042 if(hFile == INVALID_HANDLE_VALUE)
3043 return win32_error_object("CreateFileW", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00003044
3045 /* We have a good handle to the target, use it to determine the
3046 target path name. */
3047 buf_size = Py_GetFinalPathNameByHandleW(hFile, 0, 0, VOLUME_NAME_NT);
3048
3049 if(!buf_size)
Victor Stinnereb5657a2011-09-30 01:44:27 +02003050 return win32_error_object("GetFinalPathNameByHandle", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00003051
3052 target_path = (wchar_t *)malloc((buf_size+1)*sizeof(wchar_t));
3053 if(!target_path)
3054 return PyErr_NoMemory();
3055
3056 result_length = Py_GetFinalPathNameByHandleW(hFile, target_path,
3057 buf_size, VOLUME_NAME_DOS);
3058 if(!result_length)
Victor Stinnereb5657a2011-09-30 01:44:27 +02003059 return win32_error_object("GetFinalPathNamyByHandle", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00003060
3061 if(!CloseHandle(hFile))
Victor Stinnereb5657a2011-09-30 01:44:27 +02003062 return win32_error_object("CloseHandle", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00003063
3064 target_path[result_length] = 0;
Victor Stinner9d3b93b2011-11-22 02:27:30 +01003065 result = PyUnicode_FromWideChar(target_path, result_length);
Brian Curtind40e6f72010-07-08 21:39:08 +00003066 free(target_path);
3067 return result;
3068
3069} /* end of posix__getfinalpathname */
Brian Curtin62857742010-09-06 17:07:27 +00003070
3071static PyObject *
3072posix__getfileinformation(PyObject *self, PyObject *args)
3073{
3074 HANDLE hFile;
3075 BY_HANDLE_FILE_INFORMATION info;
3076 int fd;
3077
3078 if (!PyArg_ParseTuple(args, "i:_getfileinformation", &fd))
3079 return NULL;
3080
Hirokazu Yamamoto26253bb2010-12-05 04:16:47 +00003081 if (!_PyVerify_fd(fd))
3082 return posix_error();
Brian Curtin62857742010-09-06 17:07:27 +00003083
3084 hFile = (HANDLE)_get_osfhandle(fd);
3085 if (hFile == INVALID_HANDLE_VALUE)
Hirokazu Yamamoto26253bb2010-12-05 04:16:47 +00003086 return posix_error();
Brian Curtin62857742010-09-06 17:07:27 +00003087
3088 if (!GetFileInformationByHandle(hFile, &info))
3089 return win32_error("_getfileinformation", NULL);
3090
3091 return Py_BuildValue("iii", info.dwVolumeSerialNumber,
3092 info.nFileIndexHigh,
3093 info.nFileIndexLow);
3094}
Brian Curtin9c669cc2011-06-08 18:17:18 -05003095
Brian Curtin95d028f2011-06-09 09:10:38 -05003096PyDoc_STRVAR(posix__isdir__doc__,
3097"Return true if the pathname refers to an existing directory.");
3098
Brian Curtin9c669cc2011-06-08 18:17:18 -05003099static PyObject *
3100posix__isdir(PyObject *self, PyObject *args)
3101{
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003102 const char *path;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003103 PyObject *po;
Brian Curtin9c669cc2011-06-08 18:17:18 -05003104 DWORD attributes;
3105
3106 if (PyArg_ParseTuple(args, "U|:_isdir", &po)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02003107 wchar_t *wpath = PyUnicode_AsUnicode(po);
3108 if (wpath == NULL)
3109 return NULL;
Brian Curtin9c669cc2011-06-08 18:17:18 -05003110
3111 attributes = GetFileAttributesW(wpath);
3112 if (attributes == INVALID_FILE_ATTRIBUTES)
3113 Py_RETURN_FALSE;
3114 goto check;
3115 }
3116 /* Drop the argument parsing error as narrow strings
3117 are also valid. */
3118 PyErr_Clear();
3119
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003120 if (!PyArg_ParseTuple(args, "y:_isdir", &path))
Brian Curtin9c669cc2011-06-08 18:17:18 -05003121 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003122 if (win32_warn_bytes_api())
3123 return NULL;
Brian Curtin9c669cc2011-06-08 18:17:18 -05003124 attributes = GetFileAttributesA(path);
3125 if (attributes == INVALID_FILE_ATTRIBUTES)
3126 Py_RETURN_FALSE;
3127
3128check:
3129 if (attributes & FILE_ATTRIBUTE_DIRECTORY)
3130 Py_RETURN_TRUE;
3131 else
3132 Py_RETURN_FALSE;
3133}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00003134#endif /* MS_WINDOWS */
Mark Hammondef8b6542001-05-13 08:04:26 +00003135
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003136PyDoc_STRVAR(posix_mkdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003137"mkdir(path [, mode=0777])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003138Create a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003139
Barry Warsaw53699e91996-12-10 23:23:01 +00003140static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003141posix_mkdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003142{
Victor Stinner8c62be82010-05-06 00:08:46 +00003143 int res;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003144 const char *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00003145 int mode = 0777;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003146
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003147#ifdef MS_WINDOWS
Victor Stinnereb5657a2011-09-30 01:44:27 +02003148 PyObject *po;
3149 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode))
3150 {
3151 wchar_t *wpath = PyUnicode_AsUnicode(po);
3152 if (wpath == NULL)
3153 return NULL;
3154
Victor Stinner8c62be82010-05-06 00:08:46 +00003155 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02003156 res = CreateDirectoryW(wpath, NULL);
Victor Stinner8c62be82010-05-06 00:08:46 +00003157 Py_END_ALLOW_THREADS
3158 if (!res)
Victor Stinnereb5657a2011-09-30 01:44:27 +02003159 return win32_error_object("mkdir", po);
Victor Stinner8c62be82010-05-06 00:08:46 +00003160 Py_INCREF(Py_None);
3161 return Py_None;
3162 }
3163 /* Drop the argument parsing error as narrow strings
3164 are also valid. */
3165 PyErr_Clear();
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003166 if (!PyArg_ParseTuple(args, "y|i:mkdir", &path, &mode))
Victor Stinner8c62be82010-05-06 00:08:46 +00003167 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003168 if (win32_warn_bytes_api())
3169 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00003170 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00003171 res = CreateDirectoryA(path, NULL);
3172 Py_END_ALLOW_THREADS
3173 if (!res) {
3174 win32_error("mkdir", path);
Victor Stinner8c62be82010-05-06 00:08:46 +00003175 return NULL;
3176 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003177 Py_INCREF(Py_None);
3178 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003179#else
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003180 PyObject *opath;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003181
Victor Stinner8c62be82010-05-06 00:08:46 +00003182 if (!PyArg_ParseTuple(args, "O&|i:mkdir",
3183 PyUnicode_FSConverter, &opath, &mode))
3184 return NULL;
3185 path = PyBytes_AsString(opath);
3186 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00003187#if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
Victor Stinner8c62be82010-05-06 00:08:46 +00003188 res = mkdir(path);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003189#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003190 res = mkdir(path, mode);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003191#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003192 Py_END_ALLOW_THREADS
3193 if (res < 0)
3194 return posix_error_with_allocated_filename(opath);
3195 Py_DECREF(opath);
3196 Py_INCREF(Py_None);
3197 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003198#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003199}
3200
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003201
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003202/* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
3203#if defined(HAVE_SYS_RESOURCE_H)
Thomas Wouterse38b2f12001-07-11 22:35:31 +00003204#include <sys/resource.h>
3205#endif
Thomas Wouterse38b2f12001-07-11 22:35:31 +00003206
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003207
3208#ifdef HAVE_NICE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003209PyDoc_STRVAR(posix_nice__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003210"nice(inc) -> new_priority\n\n\
3211Decrease the priority of process by inc and return the new priority.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003212
Barry Warsaw53699e91996-12-10 23:23:01 +00003213static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003214posix_nice(PyObject *self, PyObject *args)
Guido van Rossum775f4da1993-01-09 17:18:52 +00003215{
Victor Stinner8c62be82010-05-06 00:08:46 +00003216 int increment, value;
Guido van Rossum775f4da1993-01-09 17:18:52 +00003217
Victor Stinner8c62be82010-05-06 00:08:46 +00003218 if (!PyArg_ParseTuple(args, "i:nice", &increment))
3219 return NULL;
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00003220
Victor Stinner8c62be82010-05-06 00:08:46 +00003221 /* There are two flavours of 'nice': one that returns the new
3222 priority (as required by almost all standards out there) and the
3223 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
3224 the use of getpriority() to get the new priority.
Tim Peters5aa91602002-01-30 05:46:57 +00003225
Victor Stinner8c62be82010-05-06 00:08:46 +00003226 If we are of the nice family that returns the new priority, we
3227 need to clear errno before the call, and check if errno is filled
3228 before calling posix_error() on a returnvalue of -1, because the
3229 -1 may be the actual new priority! */
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00003230
Victor Stinner8c62be82010-05-06 00:08:46 +00003231 errno = 0;
3232 value = nice(increment);
Thomas Wouterse38b2f12001-07-11 22:35:31 +00003233#if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
Victor Stinner8c62be82010-05-06 00:08:46 +00003234 if (value == 0)
3235 value = getpriority(PRIO_PROCESS, 0);
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00003236#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003237 if (value == -1 && errno != 0)
3238 /* either nice() or getpriority() returned an error */
3239 return posix_error();
3240 return PyLong_FromLong((long) value);
Guido van Rossum775f4da1993-01-09 17:18:52 +00003241}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003242#endif /* HAVE_NICE */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003243
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003244
3245#ifdef HAVE_GETPRIORITY
3246PyDoc_STRVAR(posix_getpriority__doc__,
3247"getpriority(which, who) -> current_priority\n\n\
3248Get program scheduling priority.");
3249
3250static PyObject *
3251posix_getpriority(PyObject *self, PyObject *args)
3252{
3253 int which, who, retval;
3254
3255 if (!PyArg_ParseTuple(args, "ii", &which, &who))
3256 return NULL;
3257 errno = 0;
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003258 retval = getpriority(which, who);
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003259 if (errno != 0)
3260 return posix_error();
3261 return PyLong_FromLong((long)retval);
3262}
3263#endif /* HAVE_GETPRIORITY */
3264
3265
3266#ifdef HAVE_SETPRIORITY
3267PyDoc_STRVAR(posix_setpriority__doc__,
3268"setpriority(which, who, prio) -> None\n\n\
3269Set program scheduling priority.");
3270
3271static PyObject *
3272posix_setpriority(PyObject *self, PyObject *args)
3273{
3274 int which, who, prio, retval;
3275
3276 if (!PyArg_ParseTuple(args, "iii", &which, &who, &prio))
3277 return NULL;
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003278 retval = setpriority(which, who, prio);
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003279 if (retval == -1)
3280 return posix_error();
3281 Py_RETURN_NONE;
3282}
3283#endif /* HAVE_SETPRIORITY */
3284
3285
Barry Warsaw53699e91996-12-10 23:23:01 +00003286static PyObject *
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003287internal_rename(PyObject *self, PyObject *args, int is_replace)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003288{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003289#ifdef MS_WINDOWS
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003290 PyObject *src, *dst;
Victor Stinner8c62be82010-05-06 00:08:46 +00003291 BOOL result;
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003292 int flags = is_replace ? MOVEFILE_REPLACE_EXISTING : 0;
3293 if (PyArg_ParseTuple(args,
3294 is_replace ? "UU:replace" : "UU:rename",
3295 &src, &dst))
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003296 {
3297 wchar_t *wsrc, *wdst;
3298
3299 wsrc = PyUnicode_AsUnicode(src);
3300 if (wsrc == NULL)
3301 return NULL;
3302 wdst = PyUnicode_AsUnicode(dst);
3303 if (wdst == NULL)
3304 return NULL;
3305 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003306 result = MoveFileExW(wsrc, wdst, flags);
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003307 Py_END_ALLOW_THREADS
3308 if (!result)
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003309 return win32_error(is_replace ? "replace" : "rename", NULL);
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003310 Py_INCREF(Py_None);
3311 return Py_None;
Victor Stinner8c62be82010-05-06 00:08:46 +00003312 }
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003313 else {
3314 PyErr_Clear();
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003315 if (!PyArg_ParseTuple(args,
3316 is_replace ? "O&O&:replace" : "O&O&:rename",
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003317 PyUnicode_FSConverter, &src,
3318 PyUnicode_FSConverter, &dst))
3319 return NULL;
3320
3321 if (win32_warn_bytes_api())
3322 goto error;
3323
3324 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003325 result = MoveFileExA(PyBytes_AS_STRING(src),
3326 PyBytes_AS_STRING(dst), flags);
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003327 Py_END_ALLOW_THREADS
3328
3329 Py_XDECREF(src);
3330 Py_XDECREF(dst);
3331
3332 if (!result)
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003333 return win32_error(is_replace ? "replace" : "rename", NULL);
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003334 Py_INCREF(Py_None);
3335 return Py_None;
3336
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00003337error:
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003338 Py_XDECREF(src);
3339 Py_XDECREF(dst);
Victor Stinner8c62be82010-05-06 00:08:46 +00003340 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003341 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003342#else
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003343 return posix_2str(args,
3344 is_replace ? "O&O&:replace" : "O&O&:rename", rename);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003345#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003346}
3347
Antoine Pitrouf3b2d882012-01-30 22:08:52 +01003348PyDoc_STRVAR(posix_rename__doc__,
3349"rename(old, new)\n\n\
3350Rename a file or directory.");
3351
3352static PyObject *
3353posix_rename(PyObject *self, PyObject *args)
3354{
3355 return internal_rename(self, args, 0);
3356}
3357
3358PyDoc_STRVAR(posix_replace__doc__,
3359"replace(old, new)\n\n\
3360Rename a file or directory, overwriting the destination.");
3361
3362static PyObject *
3363posix_replace(PyObject *self, PyObject *args)
3364{
3365 return internal_rename(self, args, 1);
3366}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003367
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003368PyDoc_STRVAR(posix_rmdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003369"rmdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003370Remove a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003371
Barry Warsaw53699e91996-12-10 23:23:01 +00003372static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003373posix_rmdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003374{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003375#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00003376 return win32_1str(args, "rmdir", "y:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003377#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003378 return posix_1str(args, "O&:rmdir", rmdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003379#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003380}
3381
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003382
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003383PyDoc_STRVAR(posix_stat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01003384"stat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003385Perform a stat system call on the given path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003386
Barry Warsaw53699e91996-12-10 23:23:01 +00003387static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01003388posix_stat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003389{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003390#ifdef MS_WINDOWS
Victor Stinner4195b5c2012-02-08 23:03:19 +01003391 return posix_do_stat(self, args, "O&:stat", STAT, "U:stat", win32_stat_w);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003392#else
Victor Stinner4195b5c2012-02-08 23:03:19 +01003393 return posix_do_stat(self, args, "O&:stat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003394#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003395}
3396
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003397
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003398#ifdef HAVE_SYSTEM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003399PyDoc_STRVAR(posix_system__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003400"system(command) -> exit_status\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003401Execute the command (a string) in a subshell.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003402
Barry Warsaw53699e91996-12-10 23:23:01 +00003403static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003404posix_system(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003405{
Victor Stinner8c62be82010-05-06 00:08:46 +00003406 long sts;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00003407#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00003408 wchar_t *command;
3409 if (!PyArg_ParseTuple(args, "u:system", &command))
3410 return NULL;
Victor Stinnercfa72782010-04-16 11:45:13 +00003411
Victor Stinner8c62be82010-05-06 00:08:46 +00003412 Py_BEGIN_ALLOW_THREADS
3413 sts = _wsystem(command);
3414 Py_END_ALLOW_THREADS
Victor Stinnercfa72782010-04-16 11:45:13 +00003415#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003416 PyObject *command_obj;
3417 char *command;
3418 if (!PyArg_ParseTuple(args, "O&:system",
3419 PyUnicode_FSConverter, &command_obj))
3420 return NULL;
Victor Stinnercfa72782010-04-16 11:45:13 +00003421
Victor Stinner8c62be82010-05-06 00:08:46 +00003422 command = PyBytes_AsString(command_obj);
3423 Py_BEGIN_ALLOW_THREADS
3424 sts = system(command);
3425 Py_END_ALLOW_THREADS
3426 Py_DECREF(command_obj);
Victor Stinnercfa72782010-04-16 11:45:13 +00003427#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003428 return PyLong_FromLong(sts);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003429}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003430#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003431
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003432
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003433PyDoc_STRVAR(posix_umask__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003434"umask(new_mask) -> old_mask\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003435Set the current numeric umask and return the previous umask.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003436
Barry Warsaw53699e91996-12-10 23:23:01 +00003437static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003438posix_umask(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003439{
Victor Stinner8c62be82010-05-06 00:08:46 +00003440 int i;
3441 if (!PyArg_ParseTuple(args, "i:umask", &i))
3442 return NULL;
3443 i = (int)umask(i);
3444 if (i < 0)
3445 return posix_error();
3446 return PyLong_FromLong((long)i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003447}
3448
Brian Curtind40e6f72010-07-08 21:39:08 +00003449#ifdef MS_WINDOWS
3450
3451/* override the default DeleteFileW behavior so that directory
3452symlinks can be removed with this function, the same as with
3453Unix symlinks */
3454BOOL WINAPI Py_DeleteFileW(LPCWSTR lpFileName)
3455{
3456 WIN32_FILE_ATTRIBUTE_DATA info;
3457 WIN32_FIND_DATAW find_data;
3458 HANDLE find_data_handle;
3459 int is_directory = 0;
3460 int is_link = 0;
3461
3462 if (GetFileAttributesExW(lpFileName, GetFileExInfoStandard, &info)) {
3463 is_directory = info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00003464
Brian Curtind40e6f72010-07-08 21:39:08 +00003465 /* Get WIN32_FIND_DATA structure for the path to determine if
3466 it is a symlink */
3467 if(is_directory &&
3468 info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
3469 find_data_handle = FindFirstFileW(lpFileName, &find_data);
3470
3471 if(find_data_handle != INVALID_HANDLE_VALUE) {
3472 is_link = find_data.dwReserved0 == IO_REPARSE_TAG_SYMLINK;
3473 FindClose(find_data_handle);
3474 }
3475 }
3476 }
3477
3478 if (is_directory && is_link)
3479 return RemoveDirectoryW(lpFileName);
3480
3481 return DeleteFileW(lpFileName);
3482}
3483#endif /* MS_WINDOWS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003484
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003485PyDoc_STRVAR(posix_unlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003486"unlink(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003487Remove a file (same as remove(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003488
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003489PyDoc_STRVAR(posix_remove__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003490"remove(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003491Remove a file (same as unlink(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003492
Barry Warsaw53699e91996-12-10 23:23:01 +00003493static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003494posix_unlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003495{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003496#ifdef MS_WINDOWS
Brian Curtin74e45612010-07-09 15:58:59 +00003497 return win32_1str(args, "remove", "y:remove", DeleteFileA,
3498 "U:remove", Py_DeleteFileW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003499#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003500 return posix_1str(args, "O&:remove", unlink);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003501#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003502}
3503
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003504
Guido van Rossumb6775db1994-08-01 11:34:53 +00003505#ifdef HAVE_UNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003506PyDoc_STRVAR(posix_uname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003507"uname() -> (sysname, nodename, release, version, machine)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003508Return a tuple identifying the current operating system.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003509
Barry Warsaw53699e91996-12-10 23:23:01 +00003510static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003511posix_uname(PyObject *self, PyObject *noargs)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00003512{
Victor Stinner8c62be82010-05-06 00:08:46 +00003513 struct utsname u;
3514 int res;
Neal Norwitze241ce82003-02-17 18:17:05 +00003515
Victor Stinner8c62be82010-05-06 00:08:46 +00003516 Py_BEGIN_ALLOW_THREADS
3517 res = uname(&u);
3518 Py_END_ALLOW_THREADS
3519 if (res < 0)
3520 return posix_error();
3521 return Py_BuildValue("(sssss)",
3522 u.sysname,
3523 u.nodename,
3524 u.release,
3525 u.version,
3526 u.machine);
Guido van Rossumc39de5f1992-02-05 11:15:54 +00003527}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003528#endif /* HAVE_UNAME */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003529
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003530
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003531static int
Victor Stinnera2f7c002012-02-08 03:36:25 +01003532extract_time(PyObject *t, time_t* sec, long* nsec)
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003533{
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003534 time_t intval;
Victor Stinner8c62be82010-05-06 00:08:46 +00003535 if (PyFloat_Check(t)) {
Victor Stinnera2f7c002012-02-08 03:36:25 +01003536 double d = PyFloat_AsDouble(t);
3537 double mod;
3538 *sec = (time_t)d;
3539 mod = fmod(d, 1.0);
3540 mod *= 1e9;
3541 *nsec = (long)mod;
Martin v. Löwis076b2092002-09-10 15:04:41 +00003542 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00003543 }
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003544#if SIZEOF_TIME_T > SIZEOF_LONG
3545 intval = PyLong_AsUnsignedLongLongMask(t);
3546#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003547 intval = PyLong_AsLong(t);
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003548#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003549 if (intval == -1 && PyErr_Occurred())
3550 return -1;
3551 *sec = intval;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003552 *nsec = 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00003553 return 0;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003554}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003555
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003556PyDoc_STRVAR(posix_utime__doc__,
Brian Curtinc1b65d12011-11-07 14:18:54 -06003557"utime(path[, (atime, mtime)])\n\
3558Set the access and modified time of the file to the given values.\n\
3559If no second argument is used, set the access and modified times to\n\
3560the current time.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003561
Barry Warsaw53699e91996-12-10 23:23:01 +00003562static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003563posix_utime(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003564{
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003565#ifdef MS_WINDOWS
Brian Curtinb0d5b5d2011-11-07 10:51:18 -06003566 PyObject *arg = Py_None;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003567 PyObject *obwpath;
Victor Stinner8c62be82010-05-06 00:08:46 +00003568 wchar_t *wpath = NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003569 const char *apath;
Victor Stinner8c62be82010-05-06 00:08:46 +00003570 HANDLE hFile;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003571 time_t atimesec, mtimesec;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003572 long ansec, mnsec;
Victor Stinner8c62be82010-05-06 00:08:46 +00003573 FILETIME atime, mtime;
3574 PyObject *result = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003575
Brian Curtin52fbea12011-11-06 13:41:17 -06003576 if (PyArg_ParseTuple(args, "U|O:utime", &obwpath, &arg)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02003577 wpath = PyUnicode_AsUnicode(obwpath);
3578 if (wpath == NULL)
3579 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00003580 Py_BEGIN_ALLOW_THREADS
3581 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
3582 NULL, OPEN_EXISTING,
3583 FILE_FLAG_BACKUP_SEMANTICS, NULL);
3584 Py_END_ALLOW_THREADS
3585 if (hFile == INVALID_HANDLE_VALUE)
Victor Stinnereb5657a2011-09-30 01:44:27 +02003586 return win32_error_object("utime", obwpath);
3587 }
3588 else {
Victor Stinner8c62be82010-05-06 00:08:46 +00003589 /* Drop the argument parsing error as narrow strings
3590 are also valid. */
3591 PyErr_Clear();
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00003592
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01003593 if (!PyArg_ParseTuple(args, "y|O:utime", &apath, &arg))
3594 return NULL;
3595 if (win32_warn_bytes_api())
Victor Stinner8c62be82010-05-06 00:08:46 +00003596 return NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02003597
Victor Stinner8c62be82010-05-06 00:08:46 +00003598 Py_BEGIN_ALLOW_THREADS
3599 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
3600 NULL, OPEN_EXISTING,
3601 FILE_FLAG_BACKUP_SEMANTICS, NULL);
3602 Py_END_ALLOW_THREADS
3603 if (hFile == INVALID_HANDLE_VALUE) {
3604 win32_error("utime", apath);
Victor Stinner8c62be82010-05-06 00:08:46 +00003605 return NULL;
3606 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003607 }
3608
Brian Curtinb0d5b5d2011-11-07 10:51:18 -06003609 if (arg == Py_None) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003610 SYSTEMTIME now;
3611 GetSystemTime(&now);
3612 if (!SystemTimeToFileTime(&now, &mtime) ||
3613 !SystemTimeToFileTime(&now, &atime)) {
3614 win32_error("utime", NULL);
3615 goto done;
Stefan Krah0e803b32010-11-26 16:16:47 +00003616 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003617 }
3618 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3619 PyErr_SetString(PyExc_TypeError,
3620 "utime() arg 2 must be a tuple (atime, mtime)");
3621 goto done;
3622 }
3623 else {
3624 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Victor Stinnera2f7c002012-02-08 03:36:25 +01003625 &atimesec, &ansec) == -1)
Victor Stinner8c62be82010-05-06 00:08:46 +00003626 goto done;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003627 time_t_to_FILE_TIME(atimesec, ansec, &atime);
Victor Stinner8c62be82010-05-06 00:08:46 +00003628 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Victor Stinnera2f7c002012-02-08 03:36:25 +01003629 &mtimesec, &mnsec) == -1)
Victor Stinner8c62be82010-05-06 00:08:46 +00003630 goto done;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003631 time_t_to_FILE_TIME(mtimesec, mnsec, &mtime);
Victor Stinner8c62be82010-05-06 00:08:46 +00003632 }
3633 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
3634 /* Avoid putting the file name into the error here,
3635 as that may confuse the user into believing that
3636 something is wrong with the file, when it also
3637 could be the time stamp that gives a problem. */
3638 win32_error("utime", NULL);
Antoine Pitroue85da7a2011-01-06 18:25:55 +00003639 goto done;
Victor Stinner8c62be82010-05-06 00:08:46 +00003640 }
3641 Py_INCREF(Py_None);
3642 result = Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003643done:
Victor Stinner8c62be82010-05-06 00:08:46 +00003644 CloseHandle(hFile);
3645 return result;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003646#else /* MS_WINDOWS */
Thomas Wouters477c8d52006-05-27 19:21:47 +00003647
Victor Stinner8c62be82010-05-06 00:08:46 +00003648 PyObject *opath;
3649 char *path;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003650 time_t atime, mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003651 long ansec, mnsec;
Victor Stinner8c62be82010-05-06 00:08:46 +00003652 int res;
Brian Curtinb0d5b5d2011-11-07 10:51:18 -06003653 PyObject* arg = Py_None;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003654
Brian Curtin52fbea12011-11-06 13:41:17 -06003655 if (!PyArg_ParseTuple(args, "O&|O:utime",
Victor Stinner8c62be82010-05-06 00:08:46 +00003656 PyUnicode_FSConverter, &opath, &arg))
3657 return NULL;
3658 path = PyBytes_AsString(opath);
Brian Curtinb0d5b5d2011-11-07 10:51:18 -06003659 if (arg == Py_None) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003660 /* optional time values not given */
3661 Py_BEGIN_ALLOW_THREADS
3662 res = utime(path, NULL);
3663 Py_END_ALLOW_THREADS
3664 }
3665 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3666 PyErr_SetString(PyExc_TypeError,
3667 "utime() arg 2 must be a tuple (atime, mtime)");
3668 Py_DECREF(opath);
3669 return NULL;
3670 }
3671 else {
3672 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Victor Stinnera2f7c002012-02-08 03:36:25 +01003673 &atime, &ansec) == -1) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003674 Py_DECREF(opath);
3675 return NULL;
3676 }
3677 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Victor Stinnera2f7c002012-02-08 03:36:25 +01003678 &mtime, &mnsec) == -1) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003679 Py_DECREF(opath);
3680 return NULL;
3681 }
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003682
3683 Py_BEGIN_ALLOW_THREADS
3684 {
3685#ifdef HAVE_UTIMENSAT
3686 struct timespec buf[2];
3687 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003688 buf[0].tv_nsec = ansec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003689 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003690 buf[1].tv_nsec = mnsec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003691 res = utimensat(AT_FDCWD, path, buf, 0);
3692#elif defined(HAVE_UTIMES)
3693 struct timeval buf[2];
3694 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003695 buf[0].tv_usec = ansec / 1000;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003696 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003697 buf[1].tv_usec = mnsec / 1000;
Victor Stinner8c62be82010-05-06 00:08:46 +00003698 res = utimes(path, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003699#elif defined(HAVE_UTIME_H)
3700 /* XXX should define struct utimbuf instead, above */
3701 struct utimbuf buf;
3702 buf.actime = atime;
3703 buf.modtime = mtime;
3704 res = utime(path, &buf);
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003705#else
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003706 time_t buf[2];
3707 buf[0] = atime;
3708 buf[1] = mtime;
3709 res = utime(path, buf);
3710#endif
3711 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003712 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00003713 }
3714 if (res < 0) {
3715 return posix_error_with_allocated_filename(opath);
3716 }
3717 Py_DECREF(opath);
3718 Py_INCREF(Py_None);
3719 return Py_None;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003720#undef UTIME_EXTRACT
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003721#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003722}
3723
Ross Lagerwall7807c352011-03-17 20:20:30 +02003724#ifdef HAVE_FUTIMES
3725PyDoc_STRVAR(posix_futimes__doc__,
Brian Curtinc1b65d12011-11-07 14:18:54 -06003726"futimes(fd[, (atime, mtime)])\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003727Set the access and modified time of the file specified by the file\n\
Brian Curtinc1b65d12011-11-07 14:18:54 -06003728descriptor fd to the given values. If no second argument is used, set the\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003729access and modified times to the current time.");
3730
3731static PyObject *
3732posix_futimes(PyObject *self, PyObject *args)
3733{
3734 int res, fd;
Brian Curtinc1b65d12011-11-07 14:18:54 -06003735 PyObject* arg = Py_None;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003736 time_t atime, mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003737 long ansec, mnsec;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003738
Brian Curtinc1b65d12011-11-07 14:18:54 -06003739 if (!PyArg_ParseTuple(args, "i|O:futimes", &fd, &arg))
Ross Lagerwall7807c352011-03-17 20:20:30 +02003740 return NULL;
3741
3742 if (arg == Py_None) {
3743 /* optional time values not given */
3744 Py_BEGIN_ALLOW_THREADS
3745 res = futimes(fd, NULL);
3746 Py_END_ALLOW_THREADS
3747 }
3748 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3749 PyErr_SetString(PyExc_TypeError,
3750 "futimes() arg 2 must be a tuple (atime, mtime)");
3751 return NULL;
3752 }
3753 else {
3754 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Victor Stinnera2f7c002012-02-08 03:36:25 +01003755 &atime, &ansec) == -1) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003756 return NULL;
3757 }
3758 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Victor Stinnera2f7c002012-02-08 03:36:25 +01003759 &mtime, &mnsec) == -1) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003760 return NULL;
3761 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003762 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003763 {
3764#ifdef HAVE_FUTIMENS
3765 struct timespec buf[2];
3766 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003767 buf[0].tv_nsec = ansec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003768 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003769 buf[1].tv_nsec = mnsec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003770 res = futimens(fd, buf);
3771#else
3772 struct timeval buf[2];
3773 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003774 buf[0].tv_usec = ansec / 1000;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003775 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003776 buf[1].tv_usec = mnsec / 1000;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003777 res = futimes(fd, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003778#endif
3779 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003780 Py_END_ALLOW_THREADS
3781 }
3782 if (res < 0)
3783 return posix_error();
3784 Py_RETURN_NONE;
3785}
3786#endif
3787
3788#ifdef HAVE_LUTIMES
3789PyDoc_STRVAR(posix_lutimes__doc__,
Brian Curtinc1b65d12011-11-07 14:18:54 -06003790"lutimes(path[, (atime, mtime)])\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003791Like utime(), but if path is a symbolic link, it is not dereferenced.");
3792
3793static PyObject *
3794posix_lutimes(PyObject *self, PyObject *args)
3795{
Brian Curtinc1b65d12011-11-07 14:18:54 -06003796 PyObject *opath;
3797 PyObject *arg = Py_None;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003798 const char *path;
3799 int res;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003800 time_t atime, mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003801 long ansec, mnsec;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003802
Brian Curtinc1b65d12011-11-07 14:18:54 -06003803 if (!PyArg_ParseTuple(args, "O&|O:lutimes",
Ross Lagerwall7807c352011-03-17 20:20:30 +02003804 PyUnicode_FSConverter, &opath, &arg))
3805 return NULL;
3806 path = PyBytes_AsString(opath);
3807 if (arg == Py_None) {
3808 /* optional time values not given */
3809 Py_BEGIN_ALLOW_THREADS
3810 res = lutimes(path, NULL);
3811 Py_END_ALLOW_THREADS
3812 }
3813 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3814 PyErr_SetString(PyExc_TypeError,
3815 "lutimes() arg 2 must be a tuple (atime, mtime)");
3816 Py_DECREF(opath);
3817 return NULL;
3818 }
3819 else {
3820 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Victor Stinnera2f7c002012-02-08 03:36:25 +01003821 &atime, &ansec) == -1) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003822 Py_DECREF(opath);
3823 return NULL;
3824 }
3825 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Victor Stinnera2f7c002012-02-08 03:36:25 +01003826 &mtime, &mnsec) == -1) {
Ross Lagerwall7807c352011-03-17 20:20:30 +02003827 Py_DECREF(opath);
3828 return NULL;
3829 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003830 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003831 {
3832#ifdef HAVE_UTIMENSAT
3833 struct timespec buf[2];
3834 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003835 buf[0].tv_nsec = ansec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003836 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003837 buf[1].tv_nsec = mnsec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003838 res = utimensat(AT_FDCWD, path, buf, AT_SYMLINK_NOFOLLOW);
3839#else
3840 struct timeval buf[2];
3841 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003842 buf[0].tv_usec = ansec / 1000;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003843 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01003844 buf[1].tv_usec = mnsec / 1000;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003845 res = lutimes(path, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07003846#endif
3847 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003848 Py_END_ALLOW_THREADS
3849 }
3850 Py_DECREF(opath);
3851 if (res < 0)
3852 return posix_error();
3853 Py_RETURN_NONE;
3854}
3855#endif
3856
3857#ifdef HAVE_FUTIMENS
3858PyDoc_STRVAR(posix_futimens__doc__,
Brian Curtinc1b65d12011-11-07 14:18:54 -06003859"futimens(fd[, (atime_sec, atime_nsec), (mtime_sec, mtime_nsec)])\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003860Updates the timestamps of a file specified by the file descriptor fd, with\n\
3861nanosecond precision.\n\
Brian Curtinc1b65d12011-11-07 14:18:54 -06003862If no second argument is given, set atime and mtime to the current time.\n\
Ross Lagerwall7807c352011-03-17 20:20:30 +02003863If *_nsec is specified as UTIME_NOW, the timestamp is updated to the\n\
3864current time.\n\
3865If *_nsec is specified as UTIME_OMIT, the timestamp is not updated.");
3866
3867static PyObject *
3868posix_futimens(PyObject *self, PyObject *args)
3869{
3870 int res, fd;
Brian Curtinc1b65d12011-11-07 14:18:54 -06003871 PyObject *atime = Py_None;
3872 PyObject *mtime = Py_None;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003873 struct timespec buf[2];
3874
Brian Curtinc1b65d12011-11-07 14:18:54 -06003875 if (!PyArg_ParseTuple(args, "i|OO:futimens",
Ross Lagerwall7807c352011-03-17 20:20:30 +02003876 &fd, &atime, &mtime))
3877 return NULL;
3878 if (atime == Py_None && mtime == Py_None) {
3879 /* optional time values not given */
3880 Py_BEGIN_ALLOW_THREADS
3881 res = futimens(fd, NULL);
3882 Py_END_ALLOW_THREADS
3883 }
3884 else if (!PyTuple_Check(atime) || PyTuple_Size(atime) != 2) {
3885 PyErr_SetString(PyExc_TypeError,
3886 "futimens() arg 2 must be a tuple (atime_sec, atime_nsec)");
3887 return NULL;
3888 }
3889 else if (!PyTuple_Check(mtime) || PyTuple_Size(mtime) != 2) {
3890 PyErr_SetString(PyExc_TypeError,
3891 "futimens() arg 3 must be a tuple (mtime_sec, mtime_nsec)");
3892 return NULL;
3893 }
3894 else {
3895 if (!PyArg_ParseTuple(atime, "ll:futimens",
3896 &(buf[0].tv_sec), &(buf[0].tv_nsec))) {
3897 return NULL;
3898 }
3899 if (!PyArg_ParseTuple(mtime, "ll:futimens",
3900 &(buf[1].tv_sec), &(buf[1].tv_nsec))) {
3901 return NULL;
3902 }
3903 Py_BEGIN_ALLOW_THREADS
3904 res = futimens(fd, buf);
3905 Py_END_ALLOW_THREADS
3906 }
3907 if (res < 0)
3908 return posix_error();
3909 Py_RETURN_NONE;
3910}
3911#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003912
Guido van Rossum3b066191991-06-04 19:40:25 +00003913/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003914
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003915PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003916"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003917Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003918
Barry Warsaw53699e91996-12-10 23:23:01 +00003919static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003920posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003921{
Victor Stinner8c62be82010-05-06 00:08:46 +00003922 int sts;
3923 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
3924 return NULL;
3925 _exit(sts);
3926 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003927}
3928
Martin v. Löwis114619e2002-10-07 06:44:21 +00003929#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
3930static void
Martin v. Löwis725507b2006-03-07 12:08:51 +00003931free_string_array(char **array, Py_ssize_t count)
Martin v. Löwis114619e2002-10-07 06:44:21 +00003932{
Victor Stinner8c62be82010-05-06 00:08:46 +00003933 Py_ssize_t i;
3934 for (i = 0; i < count; i++)
3935 PyMem_Free(array[i]);
3936 PyMem_DEL(array);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003937}
Martin v. Löwis011e8422009-05-05 04:43:17 +00003938
Antoine Pitrou69f71142009-05-24 21:25:49 +00003939static
Martin v. Löwis011e8422009-05-05 04:43:17 +00003940int fsconvert_strdup(PyObject *o, char**out)
3941{
Victor Stinner8c62be82010-05-06 00:08:46 +00003942 PyObject *bytes;
3943 Py_ssize_t size;
3944 if (!PyUnicode_FSConverter(o, &bytes))
3945 return 0;
3946 size = PyBytes_GET_SIZE(bytes);
3947 *out = PyMem_Malloc(size+1);
3948 if (!*out)
3949 return 0;
3950 memcpy(*out, PyBytes_AsString(bytes), size+1);
3951 Py_DECREF(bytes);
3952 return 1;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003953}
Martin v. Löwis114619e2002-10-07 06:44:21 +00003954#endif
3955
Ross Lagerwall7807c352011-03-17 20:20:30 +02003956#if defined(HAVE_EXECV) || defined (HAVE_FEXECVE)
Victor Stinner13bb71c2010-04-23 21:41:56 +00003957static char**
3958parse_envlist(PyObject* env, Py_ssize_t *envc_ptr)
3959{
Victor Stinner8c62be82010-05-06 00:08:46 +00003960 char **envlist;
3961 Py_ssize_t i, pos, envc;
3962 PyObject *keys=NULL, *vals=NULL;
3963 PyObject *key, *val, *key2, *val2;
3964 char *p, *k, *v;
3965 size_t len;
Victor Stinner13bb71c2010-04-23 21:41:56 +00003966
Victor Stinner8c62be82010-05-06 00:08:46 +00003967 i = PyMapping_Size(env);
3968 if (i < 0)
3969 return NULL;
3970 envlist = PyMem_NEW(char *, i + 1);
3971 if (envlist == NULL) {
3972 PyErr_NoMemory();
3973 return NULL;
3974 }
3975 envc = 0;
3976 keys = PyMapping_Keys(env);
3977 vals = PyMapping_Values(env);
3978 if (!keys || !vals)
3979 goto error;
3980 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3981 PyErr_Format(PyExc_TypeError,
3982 "env.keys() or env.values() is not a list");
3983 goto error;
3984 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00003985
Victor Stinner8c62be82010-05-06 00:08:46 +00003986 for (pos = 0; pos < i; pos++) {
3987 key = PyList_GetItem(keys, pos);
3988 val = PyList_GetItem(vals, pos);
3989 if (!key || !val)
3990 goto error;
Victor Stinner13bb71c2010-04-23 21:41:56 +00003991
Victor Stinner8c62be82010-05-06 00:08:46 +00003992 if (PyUnicode_FSConverter(key, &key2) == 0)
3993 goto error;
3994 if (PyUnicode_FSConverter(val, &val2) == 0) {
3995 Py_DECREF(key2);
3996 goto error;
3997 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00003998
3999#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +00004000 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
4001 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
Victor Stinner13bb71c2010-04-23 21:41:56 +00004002#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004003 k = PyBytes_AsString(key2);
4004 v = PyBytes_AsString(val2);
4005 len = PyBytes_GET_SIZE(key2) + PyBytes_GET_SIZE(val2) + 2;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004006
Victor Stinner8c62be82010-05-06 00:08:46 +00004007 p = PyMem_NEW(char, len);
4008 if (p == NULL) {
4009 PyErr_NoMemory();
4010 Py_DECREF(key2);
4011 Py_DECREF(val2);
4012 goto error;
4013 }
4014 PyOS_snprintf(p, len, "%s=%s", k, v);
4015 envlist[envc++] = p;
4016 Py_DECREF(key2);
4017 Py_DECREF(val2);
Victor Stinner13bb71c2010-04-23 21:41:56 +00004018#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +00004019 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00004020#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004021 }
4022 Py_DECREF(vals);
4023 Py_DECREF(keys);
Victor Stinner13bb71c2010-04-23 21:41:56 +00004024
Victor Stinner8c62be82010-05-06 00:08:46 +00004025 envlist[envc] = 0;
4026 *envc_ptr = envc;
4027 return envlist;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004028
4029error:
Victor Stinner8c62be82010-05-06 00:08:46 +00004030 Py_XDECREF(keys);
4031 Py_XDECREF(vals);
4032 while (--envc >= 0)
4033 PyMem_DEL(envlist[envc]);
4034 PyMem_DEL(envlist);
4035 return NULL;
Victor Stinner13bb71c2010-04-23 21:41:56 +00004036}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004037
Ross Lagerwall7807c352011-03-17 20:20:30 +02004038static char**
4039parse_arglist(PyObject* argv, Py_ssize_t *argc)
4040{
4041 int i;
4042 char **argvlist = PyMem_NEW(char *, *argc+1);
4043 if (argvlist == NULL) {
4044 PyErr_NoMemory();
4045 return NULL;
4046 }
4047 for (i = 0; i < *argc; i++) {
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02004048 PyObject* item = PySequence_ITEM(argv, i);
4049 if (item == NULL)
4050 goto fail;
4051 if (!fsconvert_strdup(item, &argvlist[i])) {
4052 Py_DECREF(item);
Ross Lagerwall7807c352011-03-17 20:20:30 +02004053 goto fail;
4054 }
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02004055 Py_DECREF(item);
Ross Lagerwall7807c352011-03-17 20:20:30 +02004056 }
4057 argvlist[*argc] = NULL;
4058 return argvlist;
4059fail:
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02004060 *argc = i;
Ross Lagerwall7807c352011-03-17 20:20:30 +02004061 free_string_array(argvlist, *argc);
4062 return NULL;
4063}
4064#endif
4065
4066#ifdef HAVE_EXECV
4067PyDoc_STRVAR(posix_execv__doc__,
4068"execv(path, args)\n\n\
4069Execute an executable path with arguments, replacing current process.\n\
4070\n\
4071 path: path of executable file\n\
4072 args: tuple or list of strings");
4073
4074static PyObject *
4075posix_execv(PyObject *self, PyObject *args)
4076{
4077 PyObject *opath;
4078 char *path;
4079 PyObject *argv;
4080 char **argvlist;
4081 Py_ssize_t argc;
4082
4083 /* execv has two arguments: (path, argv), where
4084 argv is a list or tuple of strings. */
4085
4086 if (!PyArg_ParseTuple(args, "O&O:execv",
4087 PyUnicode_FSConverter,
4088 &opath, &argv))
4089 return NULL;
4090 path = PyBytes_AsString(opath);
4091 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
4092 PyErr_SetString(PyExc_TypeError,
4093 "execv() arg 2 must be a tuple or list");
4094 Py_DECREF(opath);
4095 return NULL;
4096 }
4097 argc = PySequence_Size(argv);
4098 if (argc < 1) {
4099 PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
4100 Py_DECREF(opath);
4101 return NULL;
4102 }
4103
4104 argvlist = parse_arglist(argv, &argc);
4105 if (argvlist == NULL) {
4106 Py_DECREF(opath);
4107 return NULL;
4108 }
4109
4110 execv(path, argvlist);
4111
4112 /* If we get here it's definitely an error */
4113
4114 free_string_array(argvlist, argc);
4115 Py_DECREF(opath);
4116 return posix_error();
4117}
4118
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004119PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004120"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004121Execute a path with arguments and environment, replacing current process.\n\
4122\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004123 path: path of executable file\n\
4124 args: tuple or list of arguments\n\
4125 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004126
Barry Warsaw53699e91996-12-10 23:23:01 +00004127static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004128posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004129{
Victor Stinner8c62be82010-05-06 00:08:46 +00004130 PyObject *opath;
4131 char *path;
4132 PyObject *argv, *env;
4133 char **argvlist;
4134 char **envlist;
Ross Lagerwall7807c352011-03-17 20:20:30 +02004135 Py_ssize_t argc, envc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004136
Victor Stinner8c62be82010-05-06 00:08:46 +00004137 /* execve has three arguments: (path, argv, env), where
4138 argv is a list or tuple of strings and env is a dictionary
4139 like posix.environ. */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004140
Victor Stinner8c62be82010-05-06 00:08:46 +00004141 if (!PyArg_ParseTuple(args, "O&OO:execve",
4142 PyUnicode_FSConverter,
4143 &opath, &argv, &env))
4144 return NULL;
4145 path = PyBytes_AsString(opath);
Ross Lagerwall7807c352011-03-17 20:20:30 +02004146 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00004147 PyErr_SetString(PyExc_TypeError,
4148 "execve() arg 2 must be a tuple or list");
4149 goto fail_0;
4150 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02004151 argc = PySequence_Size(argv);
Victor Stinner8c62be82010-05-06 00:08:46 +00004152 if (!PyMapping_Check(env)) {
4153 PyErr_SetString(PyExc_TypeError,
4154 "execve() arg 3 must be a mapping object");
4155 goto fail_0;
4156 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004157
Ross Lagerwall7807c352011-03-17 20:20:30 +02004158 argvlist = parse_arglist(argv, &argc);
Victor Stinner8c62be82010-05-06 00:08:46 +00004159 if (argvlist == NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00004160 goto fail_0;
4161 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004162
Victor Stinner8c62be82010-05-06 00:08:46 +00004163 envlist = parse_envlist(env, &envc);
4164 if (envlist == NULL)
4165 goto fail_1;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004166
Victor Stinner8c62be82010-05-06 00:08:46 +00004167 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00004168
Victor Stinner8c62be82010-05-06 00:08:46 +00004169 /* If we get here it's definitely an error */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004170
Victor Stinner8c62be82010-05-06 00:08:46 +00004171 (void) posix_error();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004172
Victor Stinner8c62be82010-05-06 00:08:46 +00004173 while (--envc >= 0)
4174 PyMem_DEL(envlist[envc]);
4175 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004176 fail_1:
Ross Lagerwall7807c352011-03-17 20:20:30 +02004177 free_string_array(argvlist, argc);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004178 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004179 Py_DECREF(opath);
4180 return NULL;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004181}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00004182#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00004183
Ross Lagerwall7807c352011-03-17 20:20:30 +02004184#ifdef HAVE_FEXECVE
4185PyDoc_STRVAR(posix_fexecve__doc__,
4186"fexecve(fd, args, env)\n\n\
4187Execute the program specified by a file descriptor with arguments and\n\
4188environment, replacing the current process.\n\
4189\n\
4190 fd: file descriptor of executable\n\
4191 args: tuple or list of arguments\n\
4192 env: dictionary of strings mapping to strings");
4193
4194static PyObject *
4195posix_fexecve(PyObject *self, PyObject *args)
4196{
4197 int fd;
4198 PyObject *argv, *env;
4199 char **argvlist;
4200 char **envlist;
4201 Py_ssize_t argc, envc;
4202
4203 if (!PyArg_ParseTuple(args, "iOO:fexecve",
4204 &fd, &argv, &env))
4205 return NULL;
4206 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
4207 PyErr_SetString(PyExc_TypeError,
4208 "fexecve() arg 2 must be a tuple or list");
4209 return NULL;
4210 }
4211 argc = PySequence_Size(argv);
4212 if (!PyMapping_Check(env)) {
4213 PyErr_SetString(PyExc_TypeError,
4214 "fexecve() arg 3 must be a mapping object");
4215 return NULL;
4216 }
4217
4218 argvlist = parse_arglist(argv, &argc);
4219 if (argvlist == NULL)
4220 return NULL;
4221
4222 envlist = parse_envlist(env, &envc);
4223 if (envlist == NULL)
4224 goto fail;
4225
4226 fexecve(fd, argvlist, envlist);
4227
4228 /* If we get here it's definitely an error */
4229
4230 (void) posix_error();
4231
4232 while (--envc >= 0)
4233 PyMem_DEL(envlist[envc]);
4234 PyMem_DEL(envlist);
4235 fail:
4236 free_string_array(argvlist, argc);
4237 return NULL;
4238}
4239#endif /* HAVE_FEXECVE */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004240
Guido van Rossuma1065681999-01-25 23:20:23 +00004241#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004242PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004243"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00004244Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00004245\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004246 mode: mode of process creation\n\
4247 path: path of executable file\n\
4248 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00004249
4250static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004251posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00004252{
Victor Stinner8c62be82010-05-06 00:08:46 +00004253 PyObject *opath;
4254 char *path;
4255 PyObject *argv;
4256 char **argvlist;
4257 int mode, i;
4258 Py_ssize_t argc;
4259 Py_intptr_t spawnval;
4260 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossuma1065681999-01-25 23:20:23 +00004261
Victor Stinner8c62be82010-05-06 00:08:46 +00004262 /* spawnv has three arguments: (mode, path, argv), where
4263 argv is a list or tuple of strings. */
Guido van Rossuma1065681999-01-25 23:20:23 +00004264
Victor Stinner8c62be82010-05-06 00:08:46 +00004265 if (!PyArg_ParseTuple(args, "iO&O:spawnv", &mode,
4266 PyUnicode_FSConverter,
4267 &opath, &argv))
4268 return NULL;
4269 path = PyBytes_AsString(opath);
4270 if (PyList_Check(argv)) {
4271 argc = PyList_Size(argv);
4272 getitem = PyList_GetItem;
4273 }
4274 else if (PyTuple_Check(argv)) {
4275 argc = PyTuple_Size(argv);
4276 getitem = PyTuple_GetItem;
4277 }
4278 else {
4279 PyErr_SetString(PyExc_TypeError,
4280 "spawnv() arg 2 must be a tuple or list");
4281 Py_DECREF(opath);
4282 return NULL;
4283 }
Guido van Rossuma1065681999-01-25 23:20:23 +00004284
Victor Stinner8c62be82010-05-06 00:08:46 +00004285 argvlist = PyMem_NEW(char *, argc+1);
4286 if (argvlist == NULL) {
4287 Py_DECREF(opath);
4288 return PyErr_NoMemory();
4289 }
4290 for (i = 0; i < argc; i++) {
4291 if (!fsconvert_strdup((*getitem)(argv, i),
4292 &argvlist[i])) {
4293 free_string_array(argvlist, i);
4294 PyErr_SetString(
4295 PyExc_TypeError,
4296 "spawnv() arg 2 must contain only strings");
4297 Py_DECREF(opath);
4298 return NULL;
4299 }
4300 }
4301 argvlist[argc] = NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00004302
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004303#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004304 Py_BEGIN_ALLOW_THREADS
4305 spawnval = spawnv(mode, path, argvlist);
4306 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004307#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004308 if (mode == _OLD_P_OVERLAY)
4309 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00004310
Victor Stinner8c62be82010-05-06 00:08:46 +00004311 Py_BEGIN_ALLOW_THREADS
4312 spawnval = _spawnv(mode, path, argvlist);
4313 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004314#endif
Tim Peters5aa91602002-01-30 05:46:57 +00004315
Victor Stinner8c62be82010-05-06 00:08:46 +00004316 free_string_array(argvlist, argc);
4317 Py_DECREF(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00004318
Victor Stinner8c62be82010-05-06 00:08:46 +00004319 if (spawnval == -1)
4320 return posix_error();
4321 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00004322#if SIZEOF_LONG == SIZEOF_VOID_P
Victor Stinner8c62be82010-05-06 00:08:46 +00004323 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004324#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004325 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004326#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00004327}
4328
4329
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004330PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004331"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00004332Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00004333\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004334 mode: mode of process creation\n\
4335 path: path of executable file\n\
4336 args: tuple or list of arguments\n\
4337 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00004338
4339static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004340posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00004341{
Victor Stinner8c62be82010-05-06 00:08:46 +00004342 PyObject *opath;
4343 char *path;
4344 PyObject *argv, *env;
4345 char **argvlist;
4346 char **envlist;
4347 PyObject *res = NULL;
Antoine Pitrou22e41552010-08-15 18:07:50 +00004348 int mode;
4349 Py_ssize_t argc, i, envc;
Victor Stinner8c62be82010-05-06 00:08:46 +00004350 Py_intptr_t spawnval;
4351 PyObject *(*getitem)(PyObject *, Py_ssize_t);
4352 Py_ssize_t lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00004353
Victor Stinner8c62be82010-05-06 00:08:46 +00004354 /* spawnve has four arguments: (mode, path, argv, env), where
4355 argv is a list or tuple of strings and env is a dictionary
4356 like posix.environ. */
Guido van Rossuma1065681999-01-25 23:20:23 +00004357
Victor Stinner8c62be82010-05-06 00:08:46 +00004358 if (!PyArg_ParseTuple(args, "iO&OO:spawnve", &mode,
4359 PyUnicode_FSConverter,
4360 &opath, &argv, &env))
4361 return NULL;
4362 path = PyBytes_AsString(opath);
4363 if (PyList_Check(argv)) {
4364 argc = PyList_Size(argv);
4365 getitem = PyList_GetItem;
4366 }
4367 else if (PyTuple_Check(argv)) {
4368 argc = PyTuple_Size(argv);
4369 getitem = PyTuple_GetItem;
4370 }
4371 else {
4372 PyErr_SetString(PyExc_TypeError,
4373 "spawnve() arg 2 must be a tuple or list");
4374 goto fail_0;
4375 }
4376 if (!PyMapping_Check(env)) {
4377 PyErr_SetString(PyExc_TypeError,
4378 "spawnve() arg 3 must be a mapping object");
4379 goto fail_0;
4380 }
Guido van Rossuma1065681999-01-25 23:20:23 +00004381
Victor Stinner8c62be82010-05-06 00:08:46 +00004382 argvlist = PyMem_NEW(char *, argc+1);
4383 if (argvlist == NULL) {
4384 PyErr_NoMemory();
4385 goto fail_0;
4386 }
4387 for (i = 0; i < argc; i++) {
4388 if (!fsconvert_strdup((*getitem)(argv, i),
4389 &argvlist[i]))
4390 {
4391 lastarg = i;
4392 goto fail_1;
4393 }
4394 }
4395 lastarg = argc;
4396 argvlist[argc] = NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00004397
Victor Stinner8c62be82010-05-06 00:08:46 +00004398 envlist = parse_envlist(env, &envc);
4399 if (envlist == NULL)
4400 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00004401
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004402#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004403 Py_BEGIN_ALLOW_THREADS
4404 spawnval = spawnve(mode, path, argvlist, envlist);
4405 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004406#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004407 if (mode == _OLD_P_OVERLAY)
4408 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00004409
Victor Stinner8c62be82010-05-06 00:08:46 +00004410 Py_BEGIN_ALLOW_THREADS
4411 spawnval = _spawnve(mode, path, argvlist, envlist);
4412 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004413#endif
Tim Peters25059d32001-12-07 20:35:43 +00004414
Victor Stinner8c62be82010-05-06 00:08:46 +00004415 if (spawnval == -1)
4416 (void) posix_error();
4417 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00004418#if SIZEOF_LONG == SIZEOF_VOID_P
Victor Stinner8c62be82010-05-06 00:08:46 +00004419 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004420#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004421 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004422#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00004423
Victor Stinner8c62be82010-05-06 00:08:46 +00004424 while (--envc >= 0)
4425 PyMem_DEL(envlist[envc]);
4426 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004427 fail_1:
Victor Stinner8c62be82010-05-06 00:08:46 +00004428 free_string_array(argvlist, lastarg);
Martin v. Löwis114619e2002-10-07 06:44:21 +00004429 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004430 Py_DECREF(opath);
4431 return res;
Guido van Rossuma1065681999-01-25 23:20:23 +00004432}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004433
4434/* OS/2 supports spawnvp & spawnvpe natively */
4435#if defined(PYOS_OS2)
4436PyDoc_STRVAR(posix_spawnvp__doc__,
4437"spawnvp(mode, file, args)\n\n\
4438Execute the program 'file' in a new process, using the environment\n\
4439search path to find the file.\n\
4440\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004441 mode: mode of process creation\n\
4442 file: executable file name\n\
4443 args: tuple or list of strings");
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004444
4445static PyObject *
4446posix_spawnvp(PyObject *self, PyObject *args)
4447{
Victor Stinner8c62be82010-05-06 00:08:46 +00004448 PyObject *opath;
4449 char *path;
4450 PyObject *argv;
4451 char **argvlist;
4452 int mode, i, argc;
4453 Py_intptr_t spawnval;
4454 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004455
Victor Stinner8c62be82010-05-06 00:08:46 +00004456 /* spawnvp has three arguments: (mode, path, argv), where
4457 argv is a list or tuple of strings. */
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004458
Victor Stinner8c62be82010-05-06 00:08:46 +00004459 if (!PyArg_ParseTuple(args, "iO&O:spawnvp", &mode,
4460 PyUnicode_FSConverter,
4461 &opath, &argv))
4462 return NULL;
4463 path = PyBytes_AsString(opath);
4464 if (PyList_Check(argv)) {
4465 argc = PyList_Size(argv);
4466 getitem = PyList_GetItem;
4467 }
4468 else if (PyTuple_Check(argv)) {
4469 argc = PyTuple_Size(argv);
4470 getitem = PyTuple_GetItem;
4471 }
4472 else {
4473 PyErr_SetString(PyExc_TypeError,
4474 "spawnvp() arg 2 must be a tuple or list");
4475 Py_DECREF(opath);
4476 return NULL;
4477 }
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004478
Victor Stinner8c62be82010-05-06 00:08:46 +00004479 argvlist = PyMem_NEW(char *, argc+1);
4480 if (argvlist == NULL) {
4481 Py_DECREF(opath);
4482 return PyErr_NoMemory();
4483 }
4484 for (i = 0; i < argc; i++) {
4485 if (!fsconvert_strdup((*getitem)(argv, i),
4486 &argvlist[i])) {
4487 free_string_array(argvlist, i);
4488 PyErr_SetString(
4489 PyExc_TypeError,
4490 "spawnvp() arg 2 must contain only strings");
4491 Py_DECREF(opath);
4492 return NULL;
4493 }
4494 }
4495 argvlist[argc] = NULL;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004496
Victor Stinner8c62be82010-05-06 00:08:46 +00004497 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004498#if defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004499 spawnval = spawnvp(mode, path, argvlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004500#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004501 spawnval = _spawnvp(mode, path, argvlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004502#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004503 Py_END_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004504
Victor Stinner8c62be82010-05-06 00:08:46 +00004505 free_string_array(argvlist, argc);
4506 Py_DECREF(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004507
Victor Stinner8c62be82010-05-06 00:08:46 +00004508 if (spawnval == -1)
4509 return posix_error();
4510 else
4511 return Py_BuildValue("l", (long) spawnval);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004512}
4513
4514
4515PyDoc_STRVAR(posix_spawnvpe__doc__,
4516"spawnvpe(mode, file, args, env)\n\n\
4517Execute the program 'file' in a new process, using the environment\n\
4518search path to find the file.\n\
4519\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004520 mode: mode of process creation\n\
4521 file: executable file name\n\
4522 args: tuple or list of arguments\n\
4523 env: dictionary of strings mapping to strings");
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004524
4525static PyObject *
4526posix_spawnvpe(PyObject *self, PyObject *args)
4527{
Ross Lagerwalldcfde5a2011-11-04 07:09:14 +02004528 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00004529 char *path;
4530 PyObject *argv, *env;
4531 char **argvlist;
4532 char **envlist;
4533 PyObject *res=NULL;
Antoine Pitrou22e41552010-08-15 18:07:50 +00004534 int mode;
4535 Py_ssize_t argc, i, envc;
Victor Stinner8c62be82010-05-06 00:08:46 +00004536 Py_intptr_t spawnval;
4537 PyObject *(*getitem)(PyObject *, Py_ssize_t);
4538 int lastarg = 0;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004539
Victor Stinner8c62be82010-05-06 00:08:46 +00004540 /* spawnvpe has four arguments: (mode, path, argv, env), where
4541 argv is a list or tuple of strings and env is a dictionary
4542 like posix.environ. */
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004543
Victor Stinner8c62be82010-05-06 00:08:46 +00004544 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
4545 PyUnicode_FSConverter,
4546 &opath, &argv, &env))
4547 return NULL;
4548 path = PyBytes_AsString(opath);
4549 if (PyList_Check(argv)) {
4550 argc = PyList_Size(argv);
4551 getitem = PyList_GetItem;
4552 }
4553 else if (PyTuple_Check(argv)) {
4554 argc = PyTuple_Size(argv);
4555 getitem = PyTuple_GetItem;
4556 }
4557 else {
4558 PyErr_SetString(PyExc_TypeError,
4559 "spawnvpe() arg 2 must be a tuple or list");
4560 goto fail_0;
4561 }
4562 if (!PyMapping_Check(env)) {
4563 PyErr_SetString(PyExc_TypeError,
4564 "spawnvpe() arg 3 must be a mapping object");
4565 goto fail_0;
4566 }
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004567
Victor Stinner8c62be82010-05-06 00:08:46 +00004568 argvlist = PyMem_NEW(char *, argc+1);
4569 if (argvlist == NULL) {
4570 PyErr_NoMemory();
4571 goto fail_0;
4572 }
4573 for (i = 0; i < argc; i++) {
4574 if (!fsconvert_strdup((*getitem)(argv, i),
4575 &argvlist[i]))
4576 {
4577 lastarg = i;
4578 goto fail_1;
4579 }
4580 }
4581 lastarg = argc;
4582 argvlist[argc] = NULL;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004583
Victor Stinner8c62be82010-05-06 00:08:46 +00004584 envlist = parse_envlist(env, &envc);
4585 if (envlist == NULL)
4586 goto fail_1;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004587
Victor Stinner8c62be82010-05-06 00:08:46 +00004588 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004589#if defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004590 spawnval = spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004591#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004592 spawnval = _spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004593#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004594 Py_END_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004595
Victor Stinner8c62be82010-05-06 00:08:46 +00004596 if (spawnval == -1)
4597 (void) posix_error();
4598 else
4599 res = Py_BuildValue("l", (long) spawnval);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004600
Victor Stinner8c62be82010-05-06 00:08:46 +00004601 while (--envc >= 0)
4602 PyMem_DEL(envlist[envc]);
4603 PyMem_DEL(envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004604 fail_1:
Victor Stinner8c62be82010-05-06 00:08:46 +00004605 free_string_array(argvlist, lastarg);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004606 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004607 Py_DECREF(opath);
4608 return res;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004609}
4610#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00004611#endif /* HAVE_SPAWNV */
4612
4613
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004614#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004615PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004616"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004617Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
4618\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004619Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004620
4621static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004622posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004623{
Victor Stinner8c62be82010-05-06 00:08:46 +00004624 pid_t pid;
4625 int result = 0;
4626 _PyImport_AcquireLock();
4627 pid = fork1();
4628 if (pid == 0) {
4629 /* child: this clobbers and resets the import lock. */
4630 PyOS_AfterFork();
4631 } else {
4632 /* parent: release the import lock. */
4633 result = _PyImport_ReleaseLock();
4634 }
4635 if (pid == -1)
4636 return posix_error();
4637 if (result < 0) {
4638 /* Don't clobber the OSError if the fork failed. */
4639 PyErr_SetString(PyExc_RuntimeError,
4640 "not holding the import lock");
4641 return NULL;
4642 }
4643 return PyLong_FromPid(pid);
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004644}
4645#endif
4646
4647
Guido van Rossumad0ee831995-03-01 10:34:45 +00004648#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004649PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004650"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004651Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004652Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004653
Barry Warsaw53699e91996-12-10 23:23:01 +00004654static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004655posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004656{
Victor Stinner8c62be82010-05-06 00:08:46 +00004657 pid_t pid;
4658 int result = 0;
4659 _PyImport_AcquireLock();
4660 pid = fork();
4661 if (pid == 0) {
4662 /* child: this clobbers and resets the import lock. */
4663 PyOS_AfterFork();
4664 } else {
4665 /* parent: release the import lock. */
4666 result = _PyImport_ReleaseLock();
4667 }
4668 if (pid == -1)
4669 return posix_error();
4670 if (result < 0) {
4671 /* Don't clobber the OSError if the fork failed. */
4672 PyErr_SetString(PyExc_RuntimeError,
4673 "not holding the import lock");
4674 return NULL;
4675 }
4676 return PyLong_FromPid(pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00004677}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004678#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004679
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004680#ifdef HAVE_SCHED_H
4681
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +02004682#ifdef HAVE_SCHED_GET_PRIORITY_MAX
4683
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004684PyDoc_STRVAR(posix_sched_get_priority_max__doc__,
4685"sched_get_priority_max(policy)\n\n\
4686Get the maximum scheduling priority for *policy*.");
4687
4688static PyObject *
4689posix_sched_get_priority_max(PyObject *self, PyObject *args)
4690{
4691 int policy, max;
4692
4693 if (!PyArg_ParseTuple(args, "i:sched_get_priority_max", &policy))
4694 return NULL;
4695 max = sched_get_priority_max(policy);
4696 if (max < 0)
4697 return posix_error();
4698 return PyLong_FromLong(max);
4699}
4700
4701PyDoc_STRVAR(posix_sched_get_priority_min__doc__,
4702"sched_get_priority_min(policy)\n\n\
4703Get the minimum scheduling priority for *policy*.");
4704
4705static PyObject *
4706posix_sched_get_priority_min(PyObject *self, PyObject *args)
4707{
4708 int policy, min;
4709
4710 if (!PyArg_ParseTuple(args, "i:sched_get_priority_min", &policy))
4711 return NULL;
4712 min = sched_get_priority_min(policy);
4713 if (min < 0)
4714 return posix_error();
4715 return PyLong_FromLong(min);
4716}
4717
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +02004718#endif /* HAVE_SCHED_GET_PRIORITY_MAX */
4719
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004720#ifdef HAVE_SCHED_SETSCHEDULER
4721
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004722PyDoc_STRVAR(posix_sched_getscheduler__doc__,
4723"sched_getscheduler(pid)\n\n\
4724Get the scheduling policy for the process with a PID of *pid*.\n\
4725Passing a PID of 0 returns the scheduling policy for the calling process.");
4726
4727static PyObject *
4728posix_sched_getscheduler(PyObject *self, PyObject *args)
4729{
4730 pid_t pid;
4731 int policy;
4732
4733 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":sched_getscheduler", &pid))
4734 return NULL;
4735 policy = sched_getscheduler(pid);
4736 if (policy < 0)
4737 return posix_error();
4738 return PyLong_FromLong(policy);
4739}
4740
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004741#endif
4742
4743#if defined(HAVE_SCHED_SETSCHEDULER) || defined(HAVE_SCHED_SETPARAM)
4744
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004745static PyObject *
4746sched_param_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
4747{
4748 PyObject *res, *priority;
Benjamin Peterson4e36d5a2011-08-02 19:56:11 -05004749 static char *kwlist[] = {"sched_priority", NULL};
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004750
4751 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O:sched_param", kwlist, &priority))
4752 return NULL;
4753 res = PyStructSequence_New(type);
4754 if (!res)
4755 return NULL;
4756 Py_INCREF(priority);
4757 PyStructSequence_SET_ITEM(res, 0, priority);
4758 return res;
4759}
4760
4761PyDoc_STRVAR(sched_param__doc__,
4762"sched_param(sched_priority): A scheduling parameter.\n\n\
4763Current has only one field: sched_priority");
4764
4765static PyStructSequence_Field sched_param_fields[] = {
4766 {"sched_priority", "the scheduling priority"},
4767 {0}
4768};
4769
4770static PyStructSequence_Desc sched_param_desc = {
4771 "sched_param", /* name */
4772 sched_param__doc__, /* doc */
4773 sched_param_fields,
4774 1
4775};
4776
4777static int
4778convert_sched_param(PyObject *param, struct sched_param *res)
4779{
4780 long priority;
4781
4782 if (Py_TYPE(param) != &SchedParamType) {
4783 PyErr_SetString(PyExc_TypeError, "must have a sched_param object");
4784 return 0;
4785 }
4786 priority = PyLong_AsLong(PyStructSequence_GET_ITEM(param, 0));
4787 if (priority == -1 && PyErr_Occurred())
4788 return 0;
4789 if (priority > INT_MAX || priority < INT_MIN) {
4790 PyErr_SetString(PyExc_OverflowError, "sched_priority out of range");
4791 return 0;
4792 }
4793 res->sched_priority = Py_SAFE_DOWNCAST(priority, long, int);
4794 return 1;
4795}
4796
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004797#endif
4798
4799#ifdef HAVE_SCHED_SETSCHEDULER
4800
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004801PyDoc_STRVAR(posix_sched_setscheduler__doc__,
4802"sched_setscheduler(pid, policy, param)\n\n\
4803Set the scheduling policy, *policy*, for *pid*.\n\
4804If *pid* is 0, the calling process is changed.\n\
4805*param* is an instance of sched_param.");
4806
4807static PyObject *
4808posix_sched_setscheduler(PyObject *self, PyObject *args)
4809{
4810 pid_t pid;
4811 int policy;
4812 struct sched_param param;
4813
4814 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "iO&:sched_setscheduler",
4815 &pid, &policy, &convert_sched_param, &param))
4816 return NULL;
Jesus Cea9c822272011-09-10 01:40:52 +02004817
4818 /*
Jesus Cea54b01492011-09-10 01:53:19 +02004819 ** sched_setscheduler() returns 0 in Linux, but the previous
4820 ** scheduling policy under Solaris/Illumos, and others.
4821 ** On error, -1 is returned in all Operating Systems.
Jesus Cea9c822272011-09-10 01:40:52 +02004822 */
4823 if (sched_setscheduler(pid, policy, &param) == -1)
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004824 return posix_error();
4825 Py_RETURN_NONE;
4826}
4827
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004828#endif
4829
4830#ifdef HAVE_SCHED_SETPARAM
4831
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004832PyDoc_STRVAR(posix_sched_getparam__doc__,
4833"sched_getparam(pid) -> sched_param\n\n\
4834Returns scheduling parameters for the process with *pid* as an instance of the\n\
4835sched_param class. A PID of 0 means the calling process.");
4836
4837static PyObject *
4838posix_sched_getparam(PyObject *self, PyObject *args)
4839{
4840 pid_t pid;
4841 struct sched_param param;
4842 PyObject *res, *priority;
4843
4844 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":sched_getparam", &pid))
4845 return NULL;
4846 if (sched_getparam(pid, &param))
4847 return posix_error();
4848 res = PyStructSequence_New(&SchedParamType);
4849 if (!res)
4850 return NULL;
4851 priority = PyLong_FromLong(param.sched_priority);
4852 if (!priority) {
4853 Py_DECREF(res);
4854 return NULL;
4855 }
4856 PyStructSequence_SET_ITEM(res, 0, priority);
4857 return res;
4858}
4859
4860PyDoc_STRVAR(posix_sched_setparam__doc__,
4861"sched_setparam(pid, param)\n\n\
4862Set scheduling parameters for a process with PID *pid*.\n\
4863A PID of 0 means the calling process.");
4864
4865static PyObject *
4866posix_sched_setparam(PyObject *self, PyObject *args)
4867{
4868 pid_t pid;
4869 struct sched_param param;
4870
4871 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "O&:sched_setparam",
4872 &pid, &convert_sched_param, &param))
4873 return NULL;
4874 if (sched_setparam(pid, &param))
4875 return posix_error();
4876 Py_RETURN_NONE;
4877}
4878
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004879#endif
4880
4881#ifdef HAVE_SCHED_RR_GET_INTERVAL
4882
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004883PyDoc_STRVAR(posix_sched_rr_get_interval__doc__,
4884"sched_rr_get_interval(pid) -> float\n\n\
4885Return the round-robin quantum for the process with PID *pid* in seconds.");
4886
4887static PyObject *
4888posix_sched_rr_get_interval(PyObject *self, PyObject *args)
4889{
4890 pid_t pid;
4891 struct timespec interval;
4892
4893 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":sched_rr_get_interval", &pid))
4894 return NULL;
4895 if (sched_rr_get_interval(pid, &interval))
4896 return posix_error();
4897 return PyFloat_FromDouble((double)interval.tv_sec + 1e-9*interval.tv_nsec);
4898}
4899
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -05004900#endif
4901
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004902PyDoc_STRVAR(posix_sched_yield__doc__,
4903"sched_yield()\n\n\
4904Voluntarily relinquish the CPU.");
4905
4906static PyObject *
4907posix_sched_yield(PyObject *self, PyObject *noargs)
4908{
4909 if (sched_yield())
4910 return posix_error();
4911 Py_RETURN_NONE;
4912}
4913
Benjamin Peterson2740af82011-08-02 17:41:34 -05004914#ifdef HAVE_SCHED_SETAFFINITY
4915
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004916typedef struct {
4917 PyObject_HEAD;
4918 Py_ssize_t size;
4919 int ncpus;
4920 cpu_set_t *set;
4921} Py_cpu_set;
4922
4923static PyTypeObject cpu_set_type;
4924
4925static void
4926cpu_set_dealloc(Py_cpu_set *set)
4927{
4928 assert(set->set);
4929 CPU_FREE(set->set);
4930 Py_TYPE(set)->tp_free(set);
4931}
4932
4933static Py_cpu_set *
4934make_new_cpu_set(PyTypeObject *type, Py_ssize_t size)
4935{
4936 Py_cpu_set *set;
4937
4938 if (size < 0) {
4939 PyErr_SetString(PyExc_ValueError, "negative size");
4940 return NULL;
4941 }
4942 set = (Py_cpu_set *)type->tp_alloc(type, 0);
4943 if (!set)
4944 return NULL;
4945 set->ncpus = size;
4946 set->size = CPU_ALLOC_SIZE(size);
4947 set->set = CPU_ALLOC(size);
4948 if (!set->set) {
4949 type->tp_free(set);
4950 PyErr_NoMemory();
4951 return NULL;
4952 }
4953 CPU_ZERO_S(set->size, set->set);
4954 return set;
4955}
4956
4957static PyObject *
4958cpu_set_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
4959{
4960 int size;
4961
4962 if (!_PyArg_NoKeywords("cpu_set()", kwargs) ||
4963 !PyArg_ParseTuple(args, "i:cpu_set", &size))
4964 return NULL;
4965 return (PyObject *)make_new_cpu_set(type, size);
4966}
4967
4968static PyObject *
4969cpu_set_repr(Py_cpu_set *set)
4970{
4971 return PyUnicode_FromFormat("<cpu_set with %li entries>", set->ncpus);
Victor Stinner63941882011-09-29 00:42:28 +02004972}
Benjamin Peterson94b580d2011-08-02 17:30:04 -05004973
4974static Py_ssize_t
4975cpu_set_len(Py_cpu_set *set)
4976{
4977 return set->ncpus;
4978}
4979
4980static int
4981_get_cpu(Py_cpu_set *set, const char *requester, PyObject *args)
4982{
4983 int cpu;
4984 if (!PyArg_ParseTuple(args, requester, &cpu))
4985 return -1;
4986 if (cpu < 0) {
4987 PyErr_SetString(PyExc_ValueError, "cpu < 0 not valid");
4988 return -1;
4989 }
4990 if (cpu >= set->ncpus) {
4991 PyErr_SetString(PyExc_ValueError, "cpu too large for set");
4992 return -1;
4993 }
4994 return cpu;
4995}
4996
4997PyDoc_STRVAR(cpu_set_set_doc,
4998"cpu_set.set(i)\n\n\
4999Add CPU *i* to the set.");
5000
5001static PyObject *
5002cpu_set_set(Py_cpu_set *set, PyObject *args)
5003{
5004 int cpu = _get_cpu(set, "i|set", args);
5005 if (cpu == -1)
5006 return NULL;
5007 CPU_SET_S(cpu, set->size, set->set);
5008 Py_RETURN_NONE;
5009}
5010
5011PyDoc_STRVAR(cpu_set_count_doc,
5012"cpu_set.count() -> int\n\n\
5013Return the number of CPUs active in the set.");
5014
5015static PyObject *
5016cpu_set_count(Py_cpu_set *set, PyObject *noargs)
5017{
5018 return PyLong_FromLong(CPU_COUNT_S(set->size, set->set));
5019}
5020
5021PyDoc_STRVAR(cpu_set_clear_doc,
5022"cpu_set.clear(i)\n\n\
5023Remove CPU *i* from the set.");
5024
5025static PyObject *
5026cpu_set_clear(Py_cpu_set *set, PyObject *args)
5027{
5028 int cpu = _get_cpu(set, "i|clear", args);
5029 if (cpu == -1)
5030 return NULL;
5031 CPU_CLR_S(cpu, set->size, set->set);
5032 Py_RETURN_NONE;
5033}
5034
5035PyDoc_STRVAR(cpu_set_isset_doc,
5036"cpu_set.isset(i) -> bool\n\n\
5037Test if CPU *i* is in the set.");
5038
5039static PyObject *
5040cpu_set_isset(Py_cpu_set *set, PyObject *args)
5041{
5042 int cpu = _get_cpu(set, "i|isset", args);
5043 if (cpu == -1)
5044 return NULL;
5045 if (CPU_ISSET_S(cpu, set->size, set->set))
5046 Py_RETURN_TRUE;
5047 Py_RETURN_FALSE;
5048}
5049
5050PyDoc_STRVAR(cpu_set_zero_doc,
5051"cpu_set.zero()\n\n\
5052Clear the cpu_set.");
5053
5054static PyObject *
5055cpu_set_zero(Py_cpu_set *set, PyObject *noargs)
5056{
5057 CPU_ZERO_S(set->size, set->set);
5058 Py_RETURN_NONE;
5059}
5060
5061static PyObject *
5062cpu_set_richcompare(Py_cpu_set *set, Py_cpu_set *other, int op)
5063{
5064 int eq;
5065
Brian Curtindfc80e32011-08-10 20:28:54 -05005066 if ((op != Py_EQ && op != Py_NE) || Py_TYPE(other) != &cpu_set_type)
5067 Py_RETURN_NOTIMPLEMENTED;
5068
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005069 eq = set->ncpus == other->ncpus && CPU_EQUAL_S(set->size, set->set, other->set);
5070 if ((op == Py_EQ) ? eq : !eq)
5071 Py_RETURN_TRUE;
5072 else
5073 Py_RETURN_FALSE;
5074}
5075
5076#define CPU_SET_BINOP(name, op) \
5077 static PyObject * \
5078 do_cpu_set_##name(Py_cpu_set *left, Py_cpu_set *right, Py_cpu_set *res) { \
5079 if (res) { \
5080 Py_INCREF(res); \
5081 } \
5082 else { \
Benjamin Petersone870fe62011-08-02 17:44:26 -05005083 res = make_new_cpu_set(&cpu_set_type, left->ncpus); \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005084 if (!res) \
5085 return NULL; \
5086 } \
Benjamin Peterson9b374bf2011-08-02 18:22:30 -05005087 if (Py_TYPE(right) != &cpu_set_type || left->ncpus != right->ncpus) { \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005088 Py_DECREF(res); \
Brian Curtindfc80e32011-08-10 20:28:54 -05005089 Py_RETURN_NOTIMPLEMENTED; \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005090 } \
Benjamin Peterson8f7bdd32011-08-02 18:34:30 -05005091 assert(left->size == right->size && right->size == res->size); \
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005092 op(res->size, res->set, left->set, right->set); \
5093 return (PyObject *)res; \
5094 } \
5095 static PyObject * \
5096 cpu_set_##name(Py_cpu_set *left, Py_cpu_set *right) { \
5097 return do_cpu_set_##name(left, right, NULL); \
5098 } \
5099 static PyObject * \
5100 cpu_set_i##name(Py_cpu_set *left, Py_cpu_set *right) { \
5101 return do_cpu_set_##name(left, right, left); \
5102 } \
5103
5104CPU_SET_BINOP(and, CPU_AND_S)
5105CPU_SET_BINOP(or, CPU_OR_S)
5106CPU_SET_BINOP(xor, CPU_XOR_S)
5107#undef CPU_SET_BINOP
5108
5109PyDoc_STRVAR(cpu_set_doc,
5110"cpu_set(size)\n\n\
5111Create an empty mask of CPUs.");
5112
5113static PyNumberMethods cpu_set_as_number = {
5114 0, /*nb_add*/
5115 0, /*nb_subtract*/
5116 0, /*nb_multiply*/
5117 0, /*nb_remainder*/
5118 0, /*nb_divmod*/
5119 0, /*nb_power*/
5120 0, /*nb_negative*/
5121 0, /*nb_positive*/
5122 0, /*nb_absolute*/
5123 0, /*nb_bool*/
5124 0, /*nb_invert*/
5125 0, /*nb_lshift*/
5126 0, /*nb_rshift*/
5127 (binaryfunc)cpu_set_and, /*nb_and*/
5128 (binaryfunc)cpu_set_xor, /*nb_xor*/
5129 (binaryfunc)cpu_set_or, /*nb_or*/
5130 0, /*nb_int*/
5131 0, /*nb_reserved*/
5132 0, /*nb_float*/
5133 0, /*nb_inplace_add*/
5134 0, /*nb_inplace_subtract*/
5135 0, /*nb_inplace_multiply*/
5136 0, /*nb_inplace_remainder*/
5137 0, /*nb_inplace_power*/
5138 0, /*nb_inplace_lshift*/
5139 0, /*nb_inplace_rshift*/
5140 (binaryfunc)cpu_set_iand, /*nb_inplace_and*/
5141 (binaryfunc)cpu_set_ixor, /*nb_inplace_xor*/
5142 (binaryfunc)cpu_set_ior, /*nb_inplace_or*/
5143};
5144
5145static PySequenceMethods cpu_set_as_sequence = {
5146 (lenfunc)cpu_set_len, /* sq_length */
5147};
5148
5149static PyMethodDef cpu_set_methods[] = {
5150 {"clear", (PyCFunction)cpu_set_clear, METH_VARARGS, cpu_set_clear_doc},
5151 {"count", (PyCFunction)cpu_set_count, METH_NOARGS, cpu_set_count_doc},
5152 {"isset", (PyCFunction)cpu_set_isset, METH_VARARGS, cpu_set_isset_doc},
5153 {"set", (PyCFunction)cpu_set_set, METH_VARARGS, cpu_set_set_doc},
5154 {"zero", (PyCFunction)cpu_set_zero, METH_NOARGS, cpu_set_zero_doc},
5155 {NULL, NULL} /* sentinel */
5156};
5157
5158static PyTypeObject cpu_set_type = {
5159 PyVarObject_HEAD_INIT(&PyType_Type, 0)
5160 "posix.cpu_set", /* tp_name */
5161 sizeof(Py_cpu_set), /* tp_basicsize */
5162 0, /* tp_itemsize */
5163 /* methods */
5164 (destructor)cpu_set_dealloc, /* tp_dealloc */
5165 0, /* tp_print */
5166 0, /* tp_getattr */
5167 0, /* tp_setattr */
5168 0, /* tp_reserved */
5169 (reprfunc)cpu_set_repr, /* tp_repr */
5170 &cpu_set_as_number, /* tp_as_number */
5171 &cpu_set_as_sequence, /* tp_as_sequence */
5172 0, /* tp_as_mapping */
5173 PyObject_HashNotImplemented, /* tp_hash */
5174 0, /* tp_call */
5175 0, /* tp_str */
5176 PyObject_GenericGetAttr, /* tp_getattro */
5177 0, /* tp_setattro */
5178 0, /* tp_as_buffer */
5179 Py_TPFLAGS_DEFAULT, /* tp_flags */
5180 cpu_set_doc, /* tp_doc */
5181 0, /* tp_traverse */
5182 0, /* tp_clear */
5183 (richcmpfunc)cpu_set_richcompare, /* tp_richcompare */
5184 0, /* tp_weaklistoffset */
5185 0, /* tp_iter */
5186 0, /* tp_iternext */
5187 cpu_set_methods, /* tp_methods */
5188 0, /* tp_members */
5189 0, /* tp_getset */
5190 0, /* tp_base */
5191 0, /* tp_dict */
5192 0, /* tp_descr_get */
5193 0, /* tp_descr_set */
5194 0, /* tp_dictoffset */
5195 0, /* tp_init */
5196 PyType_GenericAlloc, /* tp_alloc */
5197 cpu_set_new, /* tp_new */
5198 PyObject_Del, /* tp_free */
5199};
5200
5201PyDoc_STRVAR(posix_sched_setaffinity__doc__,
5202"sched_setaffinity(pid, cpu_set)\n\n\
5203Set the affinity of the process with PID *pid* to *cpu_set*.");
5204
5205static PyObject *
5206posix_sched_setaffinity(PyObject *self, PyObject *args)
5207{
5208 pid_t pid;
5209 Py_cpu_set *cpu_set;
5210
Benjamin Petersona17a5d62011-08-09 16:49:13 -05005211 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "O!:sched_setaffinity",
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005212 &pid, &cpu_set_type, &cpu_set))
5213 return NULL;
5214 if (sched_setaffinity(pid, cpu_set->size, cpu_set->set))
5215 return posix_error();
5216 Py_RETURN_NONE;
5217}
5218
5219PyDoc_STRVAR(posix_sched_getaffinity__doc__,
5220"sched_getaffinity(pid, ncpus) -> cpu_set\n\n\
5221Return the affinity of the process with PID *pid*.\n\
5222The returned cpu_set will be of size *ncpus*.");
5223
5224static PyObject *
5225posix_sched_getaffinity(PyObject *self, PyObject *args)
5226{
5227 pid_t pid;
5228 int ncpus;
5229 Py_cpu_set *res;
5230
Benjamin Peterson7ac92142011-08-03 08:54:26 -05005231 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:sched_getaffinity",
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005232 &pid, &ncpus))
5233 return NULL;
5234 res = make_new_cpu_set(&cpu_set_type, ncpus);
5235 if (!res)
5236 return NULL;
5237 if (sched_getaffinity(pid, res->size, res->set)) {
5238 Py_DECREF(res);
5239 return posix_error();
5240 }
5241 return (PyObject *)res;
5242}
5243
Benjamin Peterson2740af82011-08-02 17:41:34 -05005244#endif /* HAVE_SCHED_SETAFFINITY */
5245
Benjamin Peterson94b580d2011-08-02 17:30:04 -05005246#endif /* HAVE_SCHED_H */
5247
Neal Norwitzb59798b2003-03-21 01:43:31 +00005248/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00005249/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
5250#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00005251#define DEV_PTY_FILE "/dev/ptc"
5252#define HAVE_DEV_PTMX
5253#else
5254#define DEV_PTY_FILE "/dev/ptmx"
5255#endif
5256
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005257#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00005258#ifdef HAVE_PTY_H
5259#include <pty.h>
5260#else
5261#ifdef HAVE_LIBUTIL_H
5262#include <libutil.h>
Ronald Oussoren755740f2010-02-07 19:56:39 +00005263#else
5264#ifdef HAVE_UTIL_H
5265#include <util.h>
5266#endif /* HAVE_UTIL_H */
Fred Drake8cef4cf2000-06-28 16:40:38 +00005267#endif /* HAVE_LIBUTIL_H */
5268#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00005269#ifdef HAVE_STROPTS_H
5270#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005271#endif
5272#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00005273
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005274#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005275PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005276"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005277Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00005278
5279static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005280posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00005281{
Victor Stinner8c62be82010-05-06 00:08:46 +00005282 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00005283#ifndef HAVE_OPENPTY
Victor Stinner8c62be82010-05-06 00:08:46 +00005284 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00005285#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005286#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Victor Stinner8c62be82010-05-06 00:08:46 +00005287 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005288#ifdef sun
Victor Stinner8c62be82010-05-06 00:08:46 +00005289 extern char *ptsname(int fildes);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005290#endif
5291#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00005292
Thomas Wouters70c21a12000-07-14 14:28:33 +00005293#ifdef HAVE_OPENPTY
Victor Stinner8c62be82010-05-06 00:08:46 +00005294 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
5295 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00005296#elif defined(HAVE__GETPTY)
Victor Stinner8c62be82010-05-06 00:08:46 +00005297 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
5298 if (slave_name == NULL)
5299 return posix_error();
Thomas Wouters70c21a12000-07-14 14:28:33 +00005300
Victor Stinner8c62be82010-05-06 00:08:46 +00005301 slave_fd = open(slave_name, O_RDWR);
5302 if (slave_fd < 0)
5303 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005304#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005305 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
5306 if (master_fd < 0)
5307 return posix_error();
5308 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
5309 /* change permission of slave */
5310 if (grantpt(master_fd) < 0) {
5311 PyOS_setsig(SIGCHLD, sig_saved);
5312 return posix_error();
5313 }
5314 /* unlock slave */
5315 if (unlockpt(master_fd) < 0) {
5316 PyOS_setsig(SIGCHLD, sig_saved);
5317 return posix_error();
5318 }
5319 PyOS_setsig(SIGCHLD, sig_saved);
5320 slave_name = ptsname(master_fd); /* get name of slave */
5321 if (slave_name == NULL)
5322 return posix_error();
5323 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
5324 if (slave_fd < 0)
5325 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00005326#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Victor Stinner8c62be82010-05-06 00:08:46 +00005327 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
5328 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00005329#ifndef __hpux
Victor Stinner8c62be82010-05-06 00:08:46 +00005330 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00005331#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005332#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00005333#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00005334
Victor Stinner8c62be82010-05-06 00:08:46 +00005335 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00005336
Fred Drake8cef4cf2000-06-28 16:40:38 +00005337}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00005338#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00005339
5340#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005341PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005342"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00005343Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
5344Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005345To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00005346
5347static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005348posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00005349{
Victor Stinner8c62be82010-05-06 00:08:46 +00005350 int master_fd = -1, result = 0;
5351 pid_t pid;
Tim Peters5aa91602002-01-30 05:46:57 +00005352
Victor Stinner8c62be82010-05-06 00:08:46 +00005353 _PyImport_AcquireLock();
5354 pid = forkpty(&master_fd, NULL, NULL, NULL);
5355 if (pid == 0) {
5356 /* child: this clobbers and resets the import lock. */
5357 PyOS_AfterFork();
5358 } else {
5359 /* parent: release the import lock. */
5360 result = _PyImport_ReleaseLock();
5361 }
5362 if (pid == -1)
5363 return posix_error();
5364 if (result < 0) {
5365 /* Don't clobber the OSError if the fork failed. */
5366 PyErr_SetString(PyExc_RuntimeError,
5367 "not holding the import lock");
5368 return NULL;
5369 }
5370 return Py_BuildValue("(Ni)", PyLong_FromPid(pid), master_fd);
Fred Drake8cef4cf2000-06-28 16:40:38 +00005371}
5372#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005373
Ross Lagerwall7807c352011-03-17 20:20:30 +02005374
Guido van Rossumad0ee831995-03-01 10:34:45 +00005375#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005376PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005377"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005378Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005379
Barry Warsaw53699e91996-12-10 23:23:01 +00005380static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005381posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005382{
Victor Stinner8c62be82010-05-06 00:08:46 +00005383 return PyLong_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005384}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005385#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005386
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005387
Guido van Rossumad0ee831995-03-01 10:34:45 +00005388#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005389PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005390"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005391Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005392
Barry Warsaw53699e91996-12-10 23:23:01 +00005393static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005394posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005395{
Victor Stinner8c62be82010-05-06 00:08:46 +00005396 return PyLong_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005397}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005398#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005399
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005400
Guido van Rossumad0ee831995-03-01 10:34:45 +00005401#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005402PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005403"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005404Return the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005405
Barry Warsaw53699e91996-12-10 23:23:01 +00005406static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005407posix_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005408{
Victor Stinner8c62be82010-05-06 00:08:46 +00005409 return PyLong_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005410}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005411#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005412
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005413
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005414PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005415"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005416Return the current process id");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005417
Barry Warsaw53699e91996-12-10 23:23:01 +00005418static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005419posix_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005420{
Victor Stinner8c62be82010-05-06 00:08:46 +00005421 return PyLong_FromPid(getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00005422}
5423
Ross Lagerwallb0ae53d2011-06-10 07:30:30 +02005424#ifdef HAVE_GETGROUPLIST
5425PyDoc_STRVAR(posix_getgrouplist__doc__,
5426"getgrouplist(user, group) -> list of groups to which a user belongs\n\n\
5427Returns a list of groups to which a user belongs.\n\n\
5428 user: username to lookup\n\
5429 group: base group id of the user");
5430
5431static PyObject *
5432posix_getgrouplist(PyObject *self, PyObject *args)
5433{
5434#ifdef NGROUPS_MAX
5435#define MAX_GROUPS NGROUPS_MAX
5436#else
5437 /* defined to be 16 on Solaris7, so this should be a small number */
5438#define MAX_GROUPS 64
5439#endif
5440
5441 const char *user;
5442 int i, ngroups;
5443 PyObject *list;
5444#ifdef __APPLE__
5445 int *groups, basegid;
5446#else
5447 gid_t *groups, basegid;
5448#endif
5449 ngroups = MAX_GROUPS;
5450
5451 if (!PyArg_ParseTuple(args, "si", &user, &basegid))
5452 return NULL;
5453
5454#ifdef __APPLE__
5455 groups = PyMem_Malloc(ngroups * sizeof(int));
5456#else
5457 groups = PyMem_Malloc(ngroups * sizeof(gid_t));
5458#endif
5459 if (groups == NULL)
5460 return PyErr_NoMemory();
5461
5462 if (getgrouplist(user, basegid, groups, &ngroups) == -1) {
5463 PyMem_Del(groups);
5464 return posix_error();
5465 }
5466
5467 list = PyList_New(ngroups);
5468 if (list == NULL) {
5469 PyMem_Del(groups);
5470 return NULL;
5471 }
5472
5473 for (i = 0; i < ngroups; i++) {
5474 PyObject *o = PyLong_FromUnsignedLong((unsigned long)groups[i]);
5475 if (o == NULL) {
5476 Py_DECREF(list);
5477 PyMem_Del(groups);
5478 return NULL;
5479 }
5480 PyList_SET_ITEM(list, i, o);
5481 }
5482
5483 PyMem_Del(groups);
5484
5485 return list;
5486}
5487#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005488
Fred Drakec9680921999-12-13 16:37:25 +00005489#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005490PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005491"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005492Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00005493
5494static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005495posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00005496{
5497 PyObject *result = NULL;
5498
Fred Drakec9680921999-12-13 16:37:25 +00005499#ifdef NGROUPS_MAX
5500#define MAX_GROUPS NGROUPS_MAX
5501#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005502 /* defined to be 16 on Solaris7, so this should be a small number */
Fred Drakec9680921999-12-13 16:37:25 +00005503#define MAX_GROUPS 64
5504#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00005505 gid_t grouplist[MAX_GROUPS];
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005506
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005507 /* On MacOSX getgroups(2) can return more than MAX_GROUPS results
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005508 * This is a helper variable to store the intermediate result when
5509 * that happens.
5510 *
5511 * To keep the code readable the OSX behaviour is unconditional,
5512 * according to the POSIX spec this should be safe on all unix-y
5513 * systems.
5514 */
5515 gid_t* alt_grouplist = grouplist;
Victor Stinner8c62be82010-05-06 00:08:46 +00005516 int n;
Fred Drakec9680921999-12-13 16:37:25 +00005517
Victor Stinner8c62be82010-05-06 00:08:46 +00005518 n = getgroups(MAX_GROUPS, grouplist);
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005519 if (n < 0) {
5520 if (errno == EINVAL) {
5521 n = getgroups(0, NULL);
5522 if (n == -1) {
5523 return posix_error();
5524 }
5525 if (n == 0) {
5526 /* Avoid malloc(0) */
5527 alt_grouplist = grouplist;
5528 } else {
5529 alt_grouplist = PyMem_Malloc(n * sizeof(gid_t));
5530 if (alt_grouplist == NULL) {
5531 errno = EINVAL;
5532 return posix_error();
5533 }
5534 n = getgroups(n, alt_grouplist);
5535 if (n == -1) {
5536 PyMem_Free(alt_grouplist);
5537 return posix_error();
5538 }
5539 }
5540 } else {
5541 return posix_error();
5542 }
5543 }
5544 result = PyList_New(n);
5545 if (result != NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00005546 int i;
5547 for (i = 0; i < n; ++i) {
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005548 PyObject *o = PyLong_FromLong((long)alt_grouplist[i]);
Victor Stinner8c62be82010-05-06 00:08:46 +00005549 if (o == NULL) {
Stefan Krah0e803b32010-11-26 16:16:47 +00005550 Py_DECREF(result);
5551 result = NULL;
5552 break;
Fred Drakec9680921999-12-13 16:37:25 +00005553 }
Victor Stinner8c62be82010-05-06 00:08:46 +00005554 PyList_SET_ITEM(result, i, o);
Fred Drakec9680921999-12-13 16:37:25 +00005555 }
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00005556 }
5557
5558 if (alt_grouplist != grouplist) {
5559 PyMem_Free(alt_grouplist);
Victor Stinner8c62be82010-05-06 00:08:46 +00005560 }
Neal Norwitze241ce82003-02-17 18:17:05 +00005561
Fred Drakec9680921999-12-13 16:37:25 +00005562 return result;
5563}
5564#endif
5565
Antoine Pitroub7572f02009-12-02 20:46:48 +00005566#ifdef HAVE_INITGROUPS
5567PyDoc_STRVAR(posix_initgroups__doc__,
5568"initgroups(username, gid) -> None\n\n\
5569Call the system initgroups() to initialize the group access list with all of\n\
5570the groups of which the specified username is a member, plus the specified\n\
5571group id.");
5572
5573static PyObject *
5574posix_initgroups(PyObject *self, PyObject *args)
5575{
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005576 PyObject *oname;
Victor Stinner8c62be82010-05-06 00:08:46 +00005577 char *username;
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005578 int res;
Victor Stinner8c62be82010-05-06 00:08:46 +00005579 long gid;
Antoine Pitroub7572f02009-12-02 20:46:48 +00005580
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005581 if (!PyArg_ParseTuple(args, "O&l:initgroups",
5582 PyUnicode_FSConverter, &oname, &gid))
Victor Stinner8c62be82010-05-06 00:08:46 +00005583 return NULL;
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005584 username = PyBytes_AS_STRING(oname);
Antoine Pitroub7572f02009-12-02 20:46:48 +00005585
Victor Stinner61ec5dc2010-08-15 09:22:44 +00005586 res = initgroups(username, (gid_t) gid);
5587 Py_DECREF(oname);
5588 if (res == -1)
Victor Stinner8c62be82010-05-06 00:08:46 +00005589 return PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitroub7572f02009-12-02 20:46:48 +00005590
Victor Stinner8c62be82010-05-06 00:08:46 +00005591 Py_INCREF(Py_None);
5592 return Py_None;
Antoine Pitroub7572f02009-12-02 20:46:48 +00005593}
5594#endif
5595
Martin v. Löwis606edc12002-06-13 21:09:11 +00005596#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00005597PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005598"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00005599Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00005600
5601static PyObject *
5602posix_getpgid(PyObject *self, PyObject *args)
5603{
Victor Stinner8c62be82010-05-06 00:08:46 +00005604 pid_t pid, pgid;
5605 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":getpgid", &pid))
5606 return NULL;
5607 pgid = getpgid(pid);
5608 if (pgid < 0)
5609 return posix_error();
5610 return PyLong_FromPid(pgid);
Martin v. Löwis606edc12002-06-13 21:09:11 +00005611}
5612#endif /* HAVE_GETPGID */
5613
5614
Guido van Rossumb6775db1994-08-01 11:34:53 +00005615#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005616PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005617"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005618Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005619
Barry Warsaw53699e91996-12-10 23:23:01 +00005620static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005621posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00005622{
Guido van Rossumb6775db1994-08-01 11:34:53 +00005623#ifdef GETPGRP_HAVE_ARG
Victor Stinner8c62be82010-05-06 00:08:46 +00005624 return PyLong_FromPid(getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005625#else /* GETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00005626 return PyLong_FromPid(getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005627#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00005628}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005629#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00005630
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005631
Guido van Rossumb6775db1994-08-01 11:34:53 +00005632#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005633PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005634"setpgrp()\n\n\
Senthil Kumaran684760a2010-06-17 16:48:06 +00005635Make this process the process group leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005636
Barry Warsaw53699e91996-12-10 23:23:01 +00005637static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005638posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00005639{
Guido van Rossum64933891994-10-20 21:56:42 +00005640#ifdef SETPGRP_HAVE_ARG
Victor Stinner8c62be82010-05-06 00:08:46 +00005641 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00005642#else /* SETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00005643 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00005644#endif /* SETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00005645 return posix_error();
5646 Py_INCREF(Py_None);
5647 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00005648}
5649
Guido van Rossumb6775db1994-08-01 11:34:53 +00005650#endif /* HAVE_SETPGRP */
5651
Guido van Rossumad0ee831995-03-01 10:34:45 +00005652#ifdef HAVE_GETPPID
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005653
5654#ifdef MS_WINDOWS
5655#include <tlhelp32.h>
5656
5657static PyObject*
5658win32_getppid()
5659{
5660 HANDLE snapshot;
5661 pid_t mypid;
5662 PyObject* result = NULL;
5663 BOOL have_record;
5664 PROCESSENTRY32 pe;
5665
5666 mypid = getpid(); /* This function never fails */
5667
5668 snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
5669 if (snapshot == INVALID_HANDLE_VALUE)
5670 return PyErr_SetFromWindowsErr(GetLastError());
5671
5672 pe.dwSize = sizeof(pe);
5673 have_record = Process32First(snapshot, &pe);
5674 while (have_record) {
5675 if (mypid == (pid_t)pe.th32ProcessID) {
5676 /* We could cache the ulong value in a static variable. */
5677 result = PyLong_FromPid((pid_t)pe.th32ParentProcessID);
5678 break;
5679 }
5680
5681 have_record = Process32Next(snapshot, &pe);
5682 }
5683
5684 /* If our loop exits and our pid was not found (result will be NULL)
5685 * then GetLastError will return ERROR_NO_MORE_FILES. This is an
5686 * error anyway, so let's raise it. */
5687 if (!result)
5688 result = PyErr_SetFromWindowsErr(GetLastError());
5689
5690 CloseHandle(snapshot);
5691
5692 return result;
5693}
5694#endif /*MS_WINDOWS*/
5695
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005696PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005697"getppid() -> ppid\n\n\
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005698Return the parent's process id. If the parent process has already exited,\n\
5699Windows machines will still return its id; others systems will return the id\n\
5700of the 'init' process (1).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005701
Barry Warsaw53699e91996-12-10 23:23:01 +00005702static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005703posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005704{
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005705#ifdef MS_WINDOWS
5706 return win32_getppid();
5707#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005708 return PyLong_FromPid(getppid());
Guido van Rossumad0ee831995-03-01 10:34:45 +00005709#endif
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00005710}
5711#endif /* HAVE_GETPPID */
Guido van Rossum85e3b011991-06-03 12:42:10 +00005712
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005713
Fred Drake12c6e2d1999-12-14 21:25:03 +00005714#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005715PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005716"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005717Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00005718
5719static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005720posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00005721{
Victor Stinner8c62be82010-05-06 00:08:46 +00005722 PyObject *result = NULL;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005723#ifdef MS_WINDOWS
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005724 wchar_t user_name[UNLEN + 1];
Victor Stinner63941882011-09-29 00:42:28 +02005725 DWORD num_chars = Py_ARRAY_LENGTH(user_name);
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005726
5727 if (GetUserNameW(user_name, &num_chars)) {
5728 /* num_chars is the number of unicode chars plus null terminator */
5729 result = PyUnicode_FromWideChar(user_name, num_chars - 1);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005730 }
5731 else
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005732 result = PyErr_SetFromWindowsErr(GetLastError());
5733#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005734 char *name;
5735 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00005736
Victor Stinner8c62be82010-05-06 00:08:46 +00005737 errno = 0;
5738 name = getlogin();
5739 if (name == NULL) {
5740 if (errno)
Victor Stinnere039ffe2010-08-15 09:33:08 +00005741 posix_error();
Fred Drake12c6e2d1999-12-14 21:25:03 +00005742 else
Victor Stinnere039ffe2010-08-15 09:33:08 +00005743 PyErr_SetString(PyExc_OSError, "unable to determine login name");
Victor Stinner8c62be82010-05-06 00:08:46 +00005744 }
5745 else
Victor Stinnere039ffe2010-08-15 09:33:08 +00005746 result = PyUnicode_DecodeFSDefault(name);
Victor Stinner8c62be82010-05-06 00:08:46 +00005747 errno = old_errno;
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005748#endif
Fred Drake12c6e2d1999-12-14 21:25:03 +00005749 return result;
5750}
Brian Curtine8e4b3b2010-09-23 20:04:14 +00005751#endif /* HAVE_GETLOGIN */
Fred Drake12c6e2d1999-12-14 21:25:03 +00005752
Guido van Rossumad0ee831995-03-01 10:34:45 +00005753#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005754PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005755"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005756Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005757
Barry Warsaw53699e91996-12-10 23:23:01 +00005758static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005759posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00005760{
Victor Stinner8c62be82010-05-06 00:08:46 +00005761 return PyLong_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00005762}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005763#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00005764
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005765
Guido van Rossumad0ee831995-03-01 10:34:45 +00005766#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005767PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005768"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005769Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005770
Barry Warsaw53699e91996-12-10 23:23:01 +00005771static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005772posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005773{
Victor Stinner8c62be82010-05-06 00:08:46 +00005774 pid_t pid;
5775 int sig;
5776 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:kill", &pid, &sig))
5777 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00005778#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005779 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
5780 APIRET rc;
5781 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005782 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005783
5784 } else if (sig == XCPT_SIGNAL_KILLPROC) {
5785 APIRET rc;
5786 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005787 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005788
5789 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00005790 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005791#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005792 if (kill(pid, sig) == -1)
5793 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00005794#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00005795 Py_INCREF(Py_None);
5796 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00005797}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005798#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00005799
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005800#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005801PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005802"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005803Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005804
5805static PyObject *
5806posix_killpg(PyObject *self, PyObject *args)
5807{
Victor Stinner8c62be82010-05-06 00:08:46 +00005808 int sig;
5809 pid_t pgid;
5810 /* XXX some man pages make the `pgid` parameter an int, others
5811 a pid_t. Since getpgrp() returns a pid_t, we assume killpg should
5812 take the same type. Moreover, pid_t is always at least as wide as
5813 int (else compilation of this module fails), which is safe. */
5814 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:killpg", &pgid, &sig))
5815 return NULL;
5816 if (killpg(pgid, sig) == -1)
5817 return posix_error();
5818 Py_INCREF(Py_None);
5819 return Py_None;
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005820}
5821#endif
5822
Brian Curtineb24d742010-04-12 17:16:38 +00005823#ifdef MS_WINDOWS
5824PyDoc_STRVAR(win32_kill__doc__,
5825"kill(pid, sig)\n\n\
5826Kill a process with a signal.");
5827
5828static PyObject *
5829win32_kill(PyObject *self, PyObject *args)
5830{
Amaury Forgeot d'Arc0a589c92010-05-15 20:35:12 +00005831 PyObject *result;
Victor Stinner8c62be82010-05-06 00:08:46 +00005832 DWORD pid, sig, err;
5833 HANDLE handle;
Brian Curtineb24d742010-04-12 17:16:38 +00005834
Victor Stinner8c62be82010-05-06 00:08:46 +00005835 if (!PyArg_ParseTuple(args, "kk:kill", &pid, &sig))
5836 return NULL;
Brian Curtineb24d742010-04-12 17:16:38 +00005837
Victor Stinner8c62be82010-05-06 00:08:46 +00005838 /* Console processes which share a common console can be sent CTRL+C or
5839 CTRL+BREAK events, provided they handle said events. */
5840 if (sig == CTRL_C_EVENT || sig == CTRL_BREAK_EVENT) {
5841 if (GenerateConsoleCtrlEvent(sig, pid) == 0) {
5842 err = GetLastError();
5843 PyErr_SetFromWindowsErr(err);
5844 }
5845 else
5846 Py_RETURN_NONE;
5847 }
Brian Curtineb24d742010-04-12 17:16:38 +00005848
Victor Stinner8c62be82010-05-06 00:08:46 +00005849 /* If the signal is outside of what GenerateConsoleCtrlEvent can use,
5850 attempt to open and terminate the process. */
5851 handle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, pid);
5852 if (handle == NULL) {
5853 err = GetLastError();
5854 return PyErr_SetFromWindowsErr(err);
5855 }
Brian Curtineb24d742010-04-12 17:16:38 +00005856
Victor Stinner8c62be82010-05-06 00:08:46 +00005857 if (TerminateProcess(handle, sig) == 0) {
5858 err = GetLastError();
5859 result = PyErr_SetFromWindowsErr(err);
5860 } else {
5861 Py_INCREF(Py_None);
5862 result = Py_None;
5863 }
Brian Curtineb24d742010-04-12 17:16:38 +00005864
Victor Stinner8c62be82010-05-06 00:08:46 +00005865 CloseHandle(handle);
5866 return result;
Brian Curtineb24d742010-04-12 17:16:38 +00005867}
5868#endif /* MS_WINDOWS */
5869
Guido van Rossumc0125471996-06-28 18:55:32 +00005870#ifdef HAVE_PLOCK
5871
5872#ifdef HAVE_SYS_LOCK_H
5873#include <sys/lock.h>
5874#endif
5875
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005876PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005877"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005878Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005879
Barry Warsaw53699e91996-12-10 23:23:01 +00005880static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005881posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00005882{
Victor Stinner8c62be82010-05-06 00:08:46 +00005883 int op;
5884 if (!PyArg_ParseTuple(args, "i:plock", &op))
5885 return NULL;
5886 if (plock(op) == -1)
5887 return posix_error();
5888 Py_INCREF(Py_None);
5889 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00005890}
5891#endif
5892
Guido van Rossumb6775db1994-08-01 11:34:53 +00005893#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005894PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005895"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005896Set the current process's user id.");
5897
Barry Warsaw53699e91996-12-10 23:23:01 +00005898static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005899posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005900{
Victor Stinner8c62be82010-05-06 00:08:46 +00005901 long uid_arg;
5902 uid_t uid;
5903 if (!PyArg_ParseTuple(args, "l:setuid", &uid_arg))
5904 return NULL;
5905 uid = uid_arg;
5906 if (uid != uid_arg) {
5907 PyErr_SetString(PyExc_OverflowError, "user id too big");
5908 return NULL;
5909 }
5910 if (setuid(uid) < 0)
5911 return posix_error();
5912 Py_INCREF(Py_None);
5913 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005914}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005915#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005916
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005917
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005918#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005919PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005920"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005921Set the current process's effective user id.");
5922
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005923static PyObject *
5924posix_seteuid (PyObject *self, PyObject *args)
5925{
Victor Stinner8c62be82010-05-06 00:08:46 +00005926 long euid_arg;
5927 uid_t euid;
5928 if (!PyArg_ParseTuple(args, "l", &euid_arg))
5929 return NULL;
5930 euid = euid_arg;
5931 if (euid != euid_arg) {
5932 PyErr_SetString(PyExc_OverflowError, "user id too big");
5933 return NULL;
5934 }
5935 if (seteuid(euid) < 0) {
5936 return posix_error();
5937 } else {
5938 Py_INCREF(Py_None);
5939 return Py_None;
5940 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005941}
5942#endif /* HAVE_SETEUID */
5943
5944#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005945PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005946"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005947Set the current process's effective group id.");
5948
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005949static PyObject *
5950posix_setegid (PyObject *self, PyObject *args)
5951{
Victor Stinner8c62be82010-05-06 00:08:46 +00005952 long egid_arg;
5953 gid_t egid;
5954 if (!PyArg_ParseTuple(args, "l", &egid_arg))
5955 return NULL;
5956 egid = egid_arg;
5957 if (egid != egid_arg) {
5958 PyErr_SetString(PyExc_OverflowError, "group id too big");
5959 return NULL;
5960 }
5961 if (setegid(egid) < 0) {
5962 return posix_error();
5963 } else {
5964 Py_INCREF(Py_None);
5965 return Py_None;
5966 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005967}
5968#endif /* HAVE_SETEGID */
5969
5970#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005971PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00005972"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005973Set the current process's real and effective user ids.");
5974
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005975static PyObject *
5976posix_setreuid (PyObject *self, PyObject *args)
5977{
Victor Stinner8c62be82010-05-06 00:08:46 +00005978 long ruid_arg, euid_arg;
5979 uid_t ruid, euid;
5980 if (!PyArg_ParseTuple(args, "ll", &ruid_arg, &euid_arg))
5981 return NULL;
5982 if (ruid_arg == -1)
5983 ruid = (uid_t)-1; /* let the compiler choose how -1 fits */
5984 else
5985 ruid = ruid_arg; /* otherwise, assign from our long */
5986 if (euid_arg == -1)
5987 euid = (uid_t)-1;
5988 else
5989 euid = euid_arg;
5990 if ((euid_arg != -1 && euid != euid_arg) ||
5991 (ruid_arg != -1 && ruid != ruid_arg)) {
5992 PyErr_SetString(PyExc_OverflowError, "user id too big");
5993 return NULL;
5994 }
5995 if (setreuid(ruid, euid) < 0) {
5996 return posix_error();
5997 } else {
5998 Py_INCREF(Py_None);
5999 return Py_None;
6000 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006001}
6002#endif /* HAVE_SETREUID */
6003
6004#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006005PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00006006"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006007Set the current process's real and effective group ids.");
6008
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006009static PyObject *
6010posix_setregid (PyObject *self, PyObject *args)
6011{
Victor Stinner8c62be82010-05-06 00:08:46 +00006012 long rgid_arg, egid_arg;
6013 gid_t rgid, egid;
6014 if (!PyArg_ParseTuple(args, "ll", &rgid_arg, &egid_arg))
6015 return NULL;
6016 if (rgid_arg == -1)
6017 rgid = (gid_t)-1; /* let the compiler choose how -1 fits */
6018 else
6019 rgid = rgid_arg; /* otherwise, assign from our long */
6020 if (egid_arg == -1)
6021 egid = (gid_t)-1;
6022 else
6023 egid = egid_arg;
6024 if ((egid_arg != -1 && egid != egid_arg) ||
6025 (rgid_arg != -1 && rgid != rgid_arg)) {
6026 PyErr_SetString(PyExc_OverflowError, "group id too big");
6027 return NULL;
6028 }
6029 if (setregid(rgid, egid) < 0) {
6030 return posix_error();
6031 } else {
6032 Py_INCREF(Py_None);
6033 return Py_None;
6034 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00006035}
6036#endif /* HAVE_SETREGID */
6037
Guido van Rossumb6775db1994-08-01 11:34:53 +00006038#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006039PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006040"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006041Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006042
Barry Warsaw53699e91996-12-10 23:23:01 +00006043static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006044posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00006045{
Victor Stinner8c62be82010-05-06 00:08:46 +00006046 long gid_arg;
6047 gid_t gid;
6048 if (!PyArg_ParseTuple(args, "l:setgid", &gid_arg))
6049 return NULL;
6050 gid = gid_arg;
6051 if (gid != gid_arg) {
6052 PyErr_SetString(PyExc_OverflowError, "group id too big");
6053 return NULL;
6054 }
6055 if (setgid(gid) < 0)
6056 return posix_error();
6057 Py_INCREF(Py_None);
6058 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00006059}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006060#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00006061
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006062#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006063PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006064"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006065Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006066
6067static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00006068posix_setgroups(PyObject *self, PyObject *groups)
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006069{
Victor Stinner8c62be82010-05-06 00:08:46 +00006070 int i, len;
6071 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00006072
Victor Stinner8c62be82010-05-06 00:08:46 +00006073 if (!PySequence_Check(groups)) {
6074 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
6075 return NULL;
6076 }
6077 len = PySequence_Size(groups);
6078 if (len > MAX_GROUPS) {
6079 PyErr_SetString(PyExc_ValueError, "too many groups");
6080 return NULL;
6081 }
6082 for(i = 0; i < len; i++) {
6083 PyObject *elem;
6084 elem = PySequence_GetItem(groups, i);
6085 if (!elem)
6086 return NULL;
6087 if (!PyLong_Check(elem)) {
6088 PyErr_SetString(PyExc_TypeError,
6089 "groups must be integers");
6090 Py_DECREF(elem);
6091 return NULL;
6092 } else {
6093 unsigned long x = PyLong_AsUnsignedLong(elem);
6094 if (PyErr_Occurred()) {
6095 PyErr_SetString(PyExc_TypeError,
6096 "group id too big");
6097 Py_DECREF(elem);
6098 return NULL;
6099 }
6100 grouplist[i] = x;
6101 /* read back the value to see if it fitted in gid_t */
6102 if (grouplist[i] != x) {
6103 PyErr_SetString(PyExc_TypeError,
6104 "group id too big");
6105 Py_DECREF(elem);
6106 return NULL;
6107 }
6108 }
6109 Py_DECREF(elem);
6110 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006111
Victor Stinner8c62be82010-05-06 00:08:46 +00006112 if (setgroups(len, grouplist) < 0)
6113 return posix_error();
6114 Py_INCREF(Py_None);
6115 return Py_None;
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00006116}
6117#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006118
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006119#if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
6120static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006121wait_helper(pid_t pid, int status, struct rusage *ru)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006122{
Victor Stinner8c62be82010-05-06 00:08:46 +00006123 PyObject *result;
6124 static PyObject *struct_rusage;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02006125 _Py_IDENTIFIER(struct_rusage);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006126
Victor Stinner8c62be82010-05-06 00:08:46 +00006127 if (pid == -1)
6128 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006129
Victor Stinner8c62be82010-05-06 00:08:46 +00006130 if (struct_rusage == NULL) {
6131 PyObject *m = PyImport_ImportModuleNoBlock("resource");
6132 if (m == NULL)
6133 return NULL;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006134 struct_rusage = _PyObject_GetAttrId(m, &PyId_struct_rusage);
Victor Stinner8c62be82010-05-06 00:08:46 +00006135 Py_DECREF(m);
6136 if (struct_rusage == NULL)
6137 return NULL;
6138 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006139
Victor Stinner8c62be82010-05-06 00:08:46 +00006140 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
6141 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
6142 if (!result)
6143 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006144
6145#ifndef doubletime
6146#define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
6147#endif
6148
Victor Stinner8c62be82010-05-06 00:08:46 +00006149 PyStructSequence_SET_ITEM(result, 0,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006150 PyFloat_FromDouble(doubletime(ru->ru_utime)));
Victor Stinner8c62be82010-05-06 00:08:46 +00006151 PyStructSequence_SET_ITEM(result, 1,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006152 PyFloat_FromDouble(doubletime(ru->ru_stime)));
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006153#define SET_INT(result, index, value)\
Victor Stinner8c62be82010-05-06 00:08:46 +00006154 PyStructSequence_SET_ITEM(result, index, PyLong_FromLong(value))
6155 SET_INT(result, 2, ru->ru_maxrss);
6156 SET_INT(result, 3, ru->ru_ixrss);
6157 SET_INT(result, 4, ru->ru_idrss);
6158 SET_INT(result, 5, ru->ru_isrss);
6159 SET_INT(result, 6, ru->ru_minflt);
6160 SET_INT(result, 7, ru->ru_majflt);
6161 SET_INT(result, 8, ru->ru_nswap);
6162 SET_INT(result, 9, ru->ru_inblock);
6163 SET_INT(result, 10, ru->ru_oublock);
6164 SET_INT(result, 11, ru->ru_msgsnd);
6165 SET_INT(result, 12, ru->ru_msgrcv);
6166 SET_INT(result, 13, ru->ru_nsignals);
6167 SET_INT(result, 14, ru->ru_nvcsw);
6168 SET_INT(result, 15, ru->ru_nivcsw);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006169#undef SET_INT
6170
Victor Stinner8c62be82010-05-06 00:08:46 +00006171 if (PyErr_Occurred()) {
6172 Py_DECREF(result);
6173 return NULL;
6174 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006175
Victor Stinner8c62be82010-05-06 00:08:46 +00006176 return Py_BuildValue("NiN", PyLong_FromPid(pid), status, result);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006177}
6178#endif /* HAVE_WAIT3 || HAVE_WAIT4 */
6179
6180#ifdef HAVE_WAIT3
6181PyDoc_STRVAR(posix_wait3__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006182"wait3(options) -> (pid, status, rusage)\n\n\
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006183Wait for completion of a child process.");
6184
6185static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006186posix_wait3(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006187{
Victor Stinner8c62be82010-05-06 00:08:46 +00006188 pid_t pid;
6189 int options;
6190 struct rusage ru;
6191 WAIT_TYPE status;
6192 WAIT_STATUS_INT(status) = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006193
Victor Stinner4195b5c2012-02-08 23:03:19 +01006194 if (!PyArg_ParseTuple(args, "i:wait3", &options))
Victor Stinner8c62be82010-05-06 00:08:46 +00006195 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006196
Victor Stinner8c62be82010-05-06 00:08:46 +00006197 Py_BEGIN_ALLOW_THREADS
6198 pid = wait3(&status, options, &ru);
6199 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006200
Victor Stinner4195b5c2012-02-08 23:03:19 +01006201 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006202}
6203#endif /* HAVE_WAIT3 */
6204
6205#ifdef HAVE_WAIT4
6206PyDoc_STRVAR(posix_wait4__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006207"wait4(pid, options) -> (pid, status, rusage)\n\n\
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006208Wait for completion of a given child process.");
6209
6210static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006211posix_wait4(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006212{
Victor Stinner8c62be82010-05-06 00:08:46 +00006213 pid_t pid;
6214 int options;
6215 struct rusage ru;
6216 WAIT_TYPE status;
6217 WAIT_STATUS_INT(status) = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006218
Victor Stinner4195b5c2012-02-08 23:03:19 +01006219 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:wait4", &pid, &options))
Victor Stinner8c62be82010-05-06 00:08:46 +00006220 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006221
Victor Stinner8c62be82010-05-06 00:08:46 +00006222 Py_BEGIN_ALLOW_THREADS
6223 pid = wait4(pid, &status, options, &ru);
6224 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006225
Victor Stinner4195b5c2012-02-08 23:03:19 +01006226 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006227}
6228#endif /* HAVE_WAIT4 */
6229
Ross Lagerwall7807c352011-03-17 20:20:30 +02006230#if defined(HAVE_WAITID) && !defined(__APPLE__)
6231PyDoc_STRVAR(posix_waitid__doc__,
6232"waitid(idtype, id, options) -> waitid_result\n\n\
6233Wait for the completion of one or more child processes.\n\n\
6234idtype can be P_PID, P_PGID or P_ALL.\n\
6235id specifies the pid to wait on.\n\
6236options is constructed from the ORing of one or more of WEXITED, WSTOPPED\n\
6237or WCONTINUED and additionally may be ORed with WNOHANG or WNOWAIT.\n\
6238Returns either waitid_result or None if WNOHANG is specified and there are\n\
6239no children in a waitable state.");
6240
6241static PyObject *
6242posix_waitid(PyObject *self, PyObject *args)
6243{
6244 PyObject *result;
6245 idtype_t idtype;
6246 id_t id;
6247 int options, res;
6248 siginfo_t si;
6249 si.si_pid = 0;
6250 if (!PyArg_ParseTuple(args, "i" _Py_PARSE_PID "i:waitid", &idtype, &id, &options))
6251 return NULL;
6252 Py_BEGIN_ALLOW_THREADS
6253 res = waitid(idtype, id, &si, options);
6254 Py_END_ALLOW_THREADS
6255 if (res == -1)
6256 return posix_error();
6257
6258 if (si.si_pid == 0)
6259 Py_RETURN_NONE;
6260
6261 result = PyStructSequence_New(&WaitidResultType);
6262 if (!result)
6263 return NULL;
6264
6265 PyStructSequence_SET_ITEM(result, 0, PyLong_FromPid(si.si_pid));
6266 PyStructSequence_SET_ITEM(result, 1, PyLong_FromPid(si.si_uid));
6267 PyStructSequence_SET_ITEM(result, 2, PyLong_FromLong((long)(si.si_signo)));
6268 PyStructSequence_SET_ITEM(result, 3, PyLong_FromLong((long)(si.si_status)));
6269 PyStructSequence_SET_ITEM(result, 4, PyLong_FromLong((long)(si.si_code)));
6270 if (PyErr_Occurred()) {
6271 Py_DECREF(result);
6272 return NULL;
6273 }
6274
6275 return result;
6276}
6277#endif
6278
Guido van Rossumb6775db1994-08-01 11:34:53 +00006279#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006280PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006281"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006282Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006283
Barry Warsaw53699e91996-12-10 23:23:01 +00006284static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006285posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00006286{
Victor Stinner8c62be82010-05-06 00:08:46 +00006287 pid_t pid;
6288 int options;
6289 WAIT_TYPE status;
6290 WAIT_STATUS_INT(status) = 0;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00006291
Victor Stinner8c62be82010-05-06 00:08:46 +00006292 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:waitpid", &pid, &options))
6293 return NULL;
6294 Py_BEGIN_ALLOW_THREADS
6295 pid = waitpid(pid, &status, options);
6296 Py_END_ALLOW_THREADS
6297 if (pid == -1)
6298 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006299
Victor Stinner8c62be82010-05-06 00:08:46 +00006300 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum21803b81992-08-09 12:55:27 +00006301}
6302
Tim Petersab034fa2002-02-01 11:27:43 +00006303#elif defined(HAVE_CWAIT)
6304
6305/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006306PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006307"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006308"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00006309
6310static PyObject *
6311posix_waitpid(PyObject *self, PyObject *args)
6312{
Victor Stinner8c62be82010-05-06 00:08:46 +00006313 Py_intptr_t pid;
6314 int status, options;
Tim Petersab034fa2002-02-01 11:27:43 +00006315
Victor Stinner8c62be82010-05-06 00:08:46 +00006316 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:waitpid", &pid, &options))
6317 return NULL;
6318 Py_BEGIN_ALLOW_THREADS
6319 pid = _cwait(&status, pid, options);
6320 Py_END_ALLOW_THREADS
6321 if (pid == -1)
6322 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006323
Victor Stinner8c62be82010-05-06 00:08:46 +00006324 /* shift the status left a byte so this is more like the POSIX waitpid */
6325 return Py_BuildValue("Ni", PyLong_FromPid(pid), status << 8);
Tim Petersab034fa2002-02-01 11:27:43 +00006326}
6327#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006328
Guido van Rossumad0ee831995-03-01 10:34:45 +00006329#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006330PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006331"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006332Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006333
Barry Warsaw53699e91996-12-10 23:23:01 +00006334static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006335posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00006336{
Victor Stinner8c62be82010-05-06 00:08:46 +00006337 pid_t pid;
6338 WAIT_TYPE status;
6339 WAIT_STATUS_INT(status) = 0;
Neal Norwitze241ce82003-02-17 18:17:05 +00006340
Victor Stinner8c62be82010-05-06 00:08:46 +00006341 Py_BEGIN_ALLOW_THREADS
6342 pid = wait(&status);
6343 Py_END_ALLOW_THREADS
6344 if (pid == -1)
6345 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006346
Victor Stinner8c62be82010-05-06 00:08:46 +00006347 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum85e3b011991-06-03 12:42:10 +00006348}
Guido van Rossumad0ee831995-03-01 10:34:45 +00006349#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00006350
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006351
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006352PyDoc_STRVAR(posix_lstat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01006353"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006354Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006355
Barry Warsaw53699e91996-12-10 23:23:01 +00006356static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01006357posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006358{
Guido van Rossumb6775db1994-08-01 11:34:53 +00006359#ifdef HAVE_LSTAT
Victor Stinner4195b5c2012-02-08 23:03:19 +01006360 return posix_do_stat(self, args, "O&:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00006361#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006362#ifdef MS_WINDOWS
Victor Stinner4195b5c2012-02-08 23:03:19 +01006363 return posix_do_stat(self, args, "O&:lstat", win32_lstat, "U:lstat",
Brian Curtind40e6f72010-07-08 21:39:08 +00006364 win32_lstat_w);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006365#else
Victor Stinner4195b5c2012-02-08 23:03:19 +01006366 return posix_do_stat(self, args, "O&:lstat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006367#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00006368#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006369}
6370
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006371
Guido van Rossumb6775db1994-08-01 11:34:53 +00006372#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006373PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006374"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006375Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006376
Barry Warsaw53699e91996-12-10 23:23:01 +00006377static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006378posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006379{
Victor Stinner8c62be82010-05-06 00:08:46 +00006380 PyObject* v;
6381 char buf[MAXPATHLEN];
6382 PyObject *opath;
6383 char *path;
6384 int n;
6385 int arg_is_unicode = 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00006386
Victor Stinner8c62be82010-05-06 00:08:46 +00006387 if (!PyArg_ParseTuple(args, "O&:readlink",
6388 PyUnicode_FSConverter, &opath))
6389 return NULL;
6390 path = PyBytes_AsString(opath);
6391 v = PySequence_GetItem(args, 0);
6392 if (v == NULL) {
6393 Py_DECREF(opath);
6394 return NULL;
6395 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00006396
Victor Stinner8c62be82010-05-06 00:08:46 +00006397 if (PyUnicode_Check(v)) {
6398 arg_is_unicode = 1;
6399 }
6400 Py_DECREF(v);
Thomas Wouters89f507f2006-12-13 04:49:30 +00006401
Victor Stinner8c62be82010-05-06 00:08:46 +00006402 Py_BEGIN_ALLOW_THREADS
6403 n = readlink(path, buf, (int) sizeof buf);
6404 Py_END_ALLOW_THREADS
6405 if (n < 0)
6406 return posix_error_with_allocated_filename(opath);
Thomas Wouters89f507f2006-12-13 04:49:30 +00006407
Victor Stinner8c62be82010-05-06 00:08:46 +00006408 Py_DECREF(opath);
Victor Stinnera45598a2010-05-14 16:35:39 +00006409 if (arg_is_unicode)
6410 return PyUnicode_DecodeFSDefaultAndSize(buf, n);
6411 else
6412 return PyBytes_FromStringAndSize(buf, n);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006413}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006414#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006415
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006416
Brian Curtin52173d42010-12-02 18:29:18 +00006417#if defined(HAVE_SYMLINK) && !defined(MS_WINDOWS)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006418PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006419"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00006420Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006421
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006422static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006423posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006424{
Victor Stinner8c62be82010-05-06 00:08:46 +00006425 return posix_2str(args, "O&O&:symlink", symlink);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006426}
6427#endif /* HAVE_SYMLINK */
6428
Brian Curtind40e6f72010-07-08 21:39:08 +00006429#if !defined(HAVE_READLINK) && defined(MS_WINDOWS)
6430
6431PyDoc_STRVAR(win_readlink__doc__,
6432"readlink(path) -> path\n\n\
6433Return a string representing the path to which the symbolic link points.");
6434
Brian Curtind40e6f72010-07-08 21:39:08 +00006435/* Windows readlink implementation */
6436static PyObject *
6437win_readlink(PyObject *self, PyObject *args)
6438{
6439 wchar_t *path;
6440 DWORD n_bytes_returned;
6441 DWORD io_result;
Victor Stinnereb5657a2011-09-30 01:44:27 +02006442 PyObject *po, *result;
Brian Curtind40e6f72010-07-08 21:39:08 +00006443 HANDLE reparse_point_handle;
6444
6445 char target_buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
6446 REPARSE_DATA_BUFFER *rdb = (REPARSE_DATA_BUFFER *)target_buffer;
6447 wchar_t *print_name;
6448
6449 if (!PyArg_ParseTuple(args,
Victor Stinnereb5657a2011-09-30 01:44:27 +02006450 "U:readlink",
6451 &po))
6452 return NULL;
6453 path = PyUnicode_AsUnicode(po);
6454 if (path == NULL)
Brian Curtind40e6f72010-07-08 21:39:08 +00006455 return NULL;
6456
6457 /* First get a handle to the reparse point */
6458 Py_BEGIN_ALLOW_THREADS
6459 reparse_point_handle = CreateFileW(
6460 path,
6461 0,
6462 0,
6463 0,
6464 OPEN_EXISTING,
6465 FILE_FLAG_OPEN_REPARSE_POINT|FILE_FLAG_BACKUP_SEMANTICS,
6466 0);
6467 Py_END_ALLOW_THREADS
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006468
Brian Curtind40e6f72010-07-08 21:39:08 +00006469 if (reparse_point_handle==INVALID_HANDLE_VALUE)
Victor Stinnereb5657a2011-09-30 01:44:27 +02006470 return win32_error_object("readlink", po);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006471
Brian Curtind40e6f72010-07-08 21:39:08 +00006472 Py_BEGIN_ALLOW_THREADS
6473 /* New call DeviceIoControl to read the reparse point */
6474 io_result = DeviceIoControl(
6475 reparse_point_handle,
6476 FSCTL_GET_REPARSE_POINT,
6477 0, 0, /* in buffer */
6478 target_buffer, sizeof(target_buffer),
6479 &n_bytes_returned,
6480 0 /* we're not using OVERLAPPED_IO */
6481 );
6482 CloseHandle(reparse_point_handle);
6483 Py_END_ALLOW_THREADS
6484
6485 if (io_result==0)
Victor Stinnereb5657a2011-09-30 01:44:27 +02006486 return win32_error_object("readlink", po);
Brian Curtind40e6f72010-07-08 21:39:08 +00006487
6488 if (rdb->ReparseTag != IO_REPARSE_TAG_SYMLINK)
6489 {
6490 PyErr_SetString(PyExc_ValueError,
6491 "not a symbolic link");
6492 return NULL;
6493 }
Brian Curtin74e45612010-07-09 15:58:59 +00006494 print_name = rdb->SymbolicLinkReparseBuffer.PathBuffer +
6495 rdb->SymbolicLinkReparseBuffer.PrintNameOffset;
6496
6497 result = PyUnicode_FromWideChar(print_name,
6498 rdb->SymbolicLinkReparseBuffer.PrintNameLength/2);
Brian Curtind40e6f72010-07-08 21:39:08 +00006499 return result;
6500}
6501
6502#endif /* !defined(HAVE_READLINK) && defined(MS_WINDOWS) */
6503
Brian Curtin52173d42010-12-02 18:29:18 +00006504#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtind40e6f72010-07-08 21:39:08 +00006505
6506/* Grab CreateSymbolicLinkW dynamically from kernel32 */
6507static int has_CreateSymbolicLinkW = 0;
6508static DWORD (CALLBACK *Py_CreateSymbolicLinkW)(LPWSTR, LPWSTR, DWORD);
6509static int
6510check_CreateSymbolicLinkW()
6511{
6512 HINSTANCE hKernel32;
6513 /* only recheck */
6514 if (has_CreateSymbolicLinkW)
6515 return has_CreateSymbolicLinkW;
Martin v. Löwis50590f12012-01-14 17:54:09 +01006516 hKernel32 = GetModuleHandleW(L"KERNEL32");
Brian Curtin74e45612010-07-09 15:58:59 +00006517 *(FARPROC*)&Py_CreateSymbolicLinkW = GetProcAddress(hKernel32,
6518 "CreateSymbolicLinkW");
Brian Curtind40e6f72010-07-08 21:39:08 +00006519 if (Py_CreateSymbolicLinkW)
6520 has_CreateSymbolicLinkW = 1;
6521 return has_CreateSymbolicLinkW;
6522}
6523
6524PyDoc_STRVAR(win_symlink__doc__,
6525"symlink(src, dst, target_is_directory=False)\n\n\
6526Create a symbolic link pointing to src named dst.\n\
6527target_is_directory is required if the target is to be interpreted as\n\
6528a directory.\n\
6529This function requires Windows 6.0 or greater, and raises a\n\
6530NotImplementedError otherwise.");
6531
6532static PyObject *
6533win_symlink(PyObject *self, PyObject *args, PyObject *kwargs)
6534{
6535 static char *kwlist[] = {"src", "dest", "target_is_directory", NULL};
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006536 PyObject *osrc, *odest;
6537 PyObject *usrc = NULL, *udest = NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02006538 wchar_t *wsrc, *wdest;
Brian Curtind40e6f72010-07-08 21:39:08 +00006539 int target_is_directory = 0;
6540 DWORD res;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006541
Brian Curtind40e6f72010-07-08 21:39:08 +00006542 if (!check_CreateSymbolicLinkW())
6543 {
6544 /* raise NotImplementedError */
6545 return PyErr_Format(PyExc_NotImplementedError,
6546 "CreateSymbolicLinkW not found");
6547 }
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006548 if (!PyArg_ParseTupleAndKeywords(
6549 args, kwargs, "OO|i:symlink", kwlist,
6550 &osrc, &odest, &target_is_directory))
Brian Curtind40e6f72010-07-08 21:39:08 +00006551 return NULL;
Brian Curtin3b4499c2010-12-28 14:31:47 +00006552
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006553 usrc = win32_decode_filename(osrc);
6554 if (!usrc)
6555 return NULL;
6556 udest = win32_decode_filename(odest);
6557 if (!udest)
6558 goto error;
6559
Brian Curtin3b4499c2010-12-28 14:31:47 +00006560 if (win32_can_symlink == 0)
6561 return PyErr_Format(PyExc_OSError, "symbolic link privilege not held");
6562
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006563 wsrc = PyUnicode_AsUnicode(usrc);
Victor Stinnereb5657a2011-09-30 01:44:27 +02006564 if (wsrc == NULL)
6565 goto error;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006566 wdest = PyUnicode_AsUnicode(udest);
Victor Stinnereb5657a2011-09-30 01:44:27 +02006567 if (wsrc == NULL)
6568 goto error;
6569
Brian Curtind40e6f72010-07-08 21:39:08 +00006570 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006571 res = Py_CreateSymbolicLinkW(wdest, wsrc, target_is_directory);
Brian Curtind40e6f72010-07-08 21:39:08 +00006572 Py_END_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006573
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006574 Py_DECREF(usrc);
6575 Py_DECREF(udest);
Brian Curtind40e6f72010-07-08 21:39:08 +00006576 if (!res)
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006577 return win32_error_object("symlink", osrc);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006578
Brian Curtind40e6f72010-07-08 21:39:08 +00006579 Py_INCREF(Py_None);
6580 return Py_None;
Victor Stinnereb5657a2011-09-30 01:44:27 +02006581
6582error:
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006583 Py_XDECREF(usrc);
6584 Py_XDECREF(udest);
Victor Stinnereb5657a2011-09-30 01:44:27 +02006585 return NULL;
Brian Curtind40e6f72010-07-08 21:39:08 +00006586}
Brian Curtin52173d42010-12-02 18:29:18 +00006587#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006588
6589#ifdef HAVE_TIMES
Guido van Rossumd48f2521997-12-05 22:19:34 +00006590#if defined(PYCC_VACPP) && defined(PYOS_OS2)
6591static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00006592system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00006593{
6594 ULONG value = 0;
6595
6596 Py_BEGIN_ALLOW_THREADS
6597 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
6598 Py_END_ALLOW_THREADS
6599
6600 return value;
6601}
6602
6603static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006604posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00006605{
Guido van Rossumd48f2521997-12-05 22:19:34 +00006606 /* Currently Only Uptime is Provided -- Others Later */
Victor Stinner8c62be82010-05-06 00:08:46 +00006607 return Py_BuildValue("ddddd",
6608 (double)0 /* t.tms_utime / HZ */,
6609 (double)0 /* t.tms_stime / HZ */,
6610 (double)0 /* t.tms_cutime / HZ */,
6611 (double)0 /* t.tms_cstime / HZ */,
6612 (double)system_uptime() / 1000);
Guido van Rossumd48f2521997-12-05 22:19:34 +00006613}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006614#else /* not OS2 */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00006615#define NEED_TICKS_PER_SECOND
6616static long ticks_per_second = -1;
Barry Warsaw53699e91996-12-10 23:23:01 +00006617static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006618posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00006619{
Victor Stinner8c62be82010-05-06 00:08:46 +00006620 struct tms t;
6621 clock_t c;
6622 errno = 0;
6623 c = times(&t);
6624 if (c == (clock_t) -1)
6625 return posix_error();
6626 return Py_BuildValue("ddddd",
6627 (double)t.tms_utime / ticks_per_second,
6628 (double)t.tms_stime / ticks_per_second,
6629 (double)t.tms_cutime / ticks_per_second,
6630 (double)t.tms_cstime / ticks_per_second,
6631 (double)c / ticks_per_second);
Guido van Rossum22db57e1992-04-05 14:25:30 +00006632}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006633#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00006634#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006635
6636
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006637#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00006638#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00006639static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006640posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00006641{
Victor Stinner8c62be82010-05-06 00:08:46 +00006642 FILETIME create, exit, kernel, user;
6643 HANDLE hProc;
6644 hProc = GetCurrentProcess();
6645 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
6646 /* The fields of a FILETIME structure are the hi and lo part
6647 of a 64-bit value expressed in 100 nanosecond units.
6648 1e7 is one second in such units; 1e-7 the inverse.
6649 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
6650 */
6651 return Py_BuildValue(
6652 "ddddd",
6653 (double)(user.dwHighDateTime*429.4967296 +
6654 user.dwLowDateTime*1e-7),
6655 (double)(kernel.dwHighDateTime*429.4967296 +
6656 kernel.dwLowDateTime*1e-7),
6657 (double)0,
6658 (double)0,
6659 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00006660}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006661#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006662
6663#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006664PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006665"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006666Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00006667#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00006668
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006669
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00006670#ifdef HAVE_GETSID
6671PyDoc_STRVAR(posix_getsid__doc__,
6672"getsid(pid) -> sid\n\n\
6673Call the system call getsid().");
6674
6675static PyObject *
6676posix_getsid(PyObject *self, PyObject *args)
6677{
Victor Stinner8c62be82010-05-06 00:08:46 +00006678 pid_t pid;
6679 int sid;
6680 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":getsid", &pid))
6681 return NULL;
6682 sid = getsid(pid);
6683 if (sid < 0)
6684 return posix_error();
6685 return PyLong_FromLong((long)sid);
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00006686}
6687#endif /* HAVE_GETSID */
6688
6689
Guido van Rossumb6775db1994-08-01 11:34:53 +00006690#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006691PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006692"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006693Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006694
Barry Warsaw53699e91996-12-10 23:23:01 +00006695static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006696posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00006697{
Victor Stinner8c62be82010-05-06 00:08:46 +00006698 if (setsid() < 0)
6699 return posix_error();
6700 Py_INCREF(Py_None);
6701 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00006702}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006703#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00006704
Guido van Rossumb6775db1994-08-01 11:34:53 +00006705#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006706PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006707"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006708Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006709
Barry Warsaw53699e91996-12-10 23:23:01 +00006710static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006711posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00006712{
Victor Stinner8c62be82010-05-06 00:08:46 +00006713 pid_t pid;
6714 int pgrp;
6715 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:setpgid", &pid, &pgrp))
6716 return NULL;
6717 if (setpgid(pid, pgrp) < 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_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00006723
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006724
Guido van Rossumb6775db1994-08-01 11:34:53 +00006725#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006726PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006727"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006728Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006729
Barry Warsaw53699e91996-12-10 23:23:01 +00006730static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006731posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00006732{
Victor Stinner8c62be82010-05-06 00:08:46 +00006733 int fd;
6734 pid_t pgid;
6735 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
6736 return NULL;
6737 pgid = tcgetpgrp(fd);
6738 if (pgid < 0)
6739 return posix_error();
6740 return PyLong_FromPid(pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00006741}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006742#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00006743
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006744
Guido van Rossumb6775db1994-08-01 11:34:53 +00006745#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006746PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006747"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006748Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006749
Barry Warsaw53699e91996-12-10 23:23:01 +00006750static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006751posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00006752{
Victor Stinner8c62be82010-05-06 00:08:46 +00006753 int fd;
6754 pid_t pgid;
6755 if (!PyArg_ParseTuple(args, "i" _Py_PARSE_PID ":tcsetpgrp", &fd, &pgid))
6756 return NULL;
6757 if (tcsetpgrp(fd, pgid) < 0)
6758 return posix_error();
6759 Py_INCREF(Py_None);
6760 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00006761}
Guido van Rossumb6775db1994-08-01 11:34:53 +00006762#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00006763
Guido van Rossum687dd131993-05-17 08:34:16 +00006764/* Functions acting on file descriptors */
6765
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006766PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006767"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006768Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006769
Barry Warsaw53699e91996-12-10 23:23:01 +00006770static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006771posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006772{
Victor Stinner8c62be82010-05-06 00:08:46 +00006773 PyObject *ofile;
6774 char *file;
6775 int flag;
6776 int mode = 0777;
6777 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006778
6779#ifdef MS_WINDOWS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006780 PyObject *po;
Victor Stinner26de69d2011-06-17 15:15:38 +02006781 if (PyArg_ParseTuple(args, "Ui|i:open", &po, &flag, &mode)) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02006782 wchar_t *wpath = PyUnicode_AsUnicode(po);
6783 if (wpath == NULL)
6784 return NULL;
6785
Victor Stinner8c62be82010-05-06 00:08:46 +00006786 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02006787 fd = _wopen(wpath, flag, mode);
Victor Stinner8c62be82010-05-06 00:08:46 +00006788 Py_END_ALLOW_THREADS
6789 if (fd < 0)
6790 return posix_error();
6791 return PyLong_FromLong((long)fd);
6792 }
6793 /* Drop the argument parsing error as narrow strings
6794 are also valid. */
6795 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00006796#endif
6797
Victor Stinner26de69d2011-06-17 15:15:38 +02006798 if (!PyArg_ParseTuple(args, "O&i|i:open",
Victor Stinner8c62be82010-05-06 00:08:46 +00006799 PyUnicode_FSConverter, &ofile,
6800 &flag, &mode))
6801 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01006802#ifdef MS_WINDOWS
6803 if (win32_warn_bytes_api()) {
6804 Py_DECREF(ofile);
6805 return NULL;
6806 }
6807#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006808 file = PyBytes_AsString(ofile);
6809 Py_BEGIN_ALLOW_THREADS
6810 fd = open(file, flag, mode);
6811 Py_END_ALLOW_THREADS
6812 if (fd < 0)
6813 return posix_error_with_allocated_filename(ofile);
6814 Py_DECREF(ofile);
6815 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006816}
6817
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006818
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006819PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006820"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006821Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006822
Barry Warsaw53699e91996-12-10 23:23:01 +00006823static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006824posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006825{
Victor Stinner8c62be82010-05-06 00:08:46 +00006826 int fd, res;
6827 if (!PyArg_ParseTuple(args, "i:close", &fd))
6828 return NULL;
6829 if (!_PyVerify_fd(fd))
6830 return posix_error();
6831 Py_BEGIN_ALLOW_THREADS
6832 res = close(fd);
6833 Py_END_ALLOW_THREADS
6834 if (res < 0)
6835 return posix_error();
6836 Py_INCREF(Py_None);
6837 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006838}
6839
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006840
Victor Stinner8c62be82010-05-06 00:08:46 +00006841PyDoc_STRVAR(posix_closerange__doc__,
Christian Heimesfdab48e2008-01-20 09:06:41 +00006842"closerange(fd_low, fd_high)\n\n\
6843Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
6844
6845static PyObject *
6846posix_closerange(PyObject *self, PyObject *args)
6847{
Victor Stinner8c62be82010-05-06 00:08:46 +00006848 int fd_from, fd_to, i;
6849 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
6850 return NULL;
6851 Py_BEGIN_ALLOW_THREADS
6852 for (i = fd_from; i < fd_to; i++)
6853 if (_PyVerify_fd(i))
6854 close(i);
6855 Py_END_ALLOW_THREADS
6856 Py_RETURN_NONE;
Christian Heimesfdab48e2008-01-20 09:06:41 +00006857}
6858
6859
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006860PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006861"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006862Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006863
Barry Warsaw53699e91996-12-10 23:23:01 +00006864static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006865posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006866{
Victor Stinner8c62be82010-05-06 00:08:46 +00006867 int fd;
6868 if (!PyArg_ParseTuple(args, "i:dup", &fd))
6869 return NULL;
6870 if (!_PyVerify_fd(fd))
6871 return posix_error();
Victor Stinner8c62be82010-05-06 00:08:46 +00006872 fd = dup(fd);
Victor Stinner8c62be82010-05-06 00:08:46 +00006873 if (fd < 0)
6874 return posix_error();
6875 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006876}
6877
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006878
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006879PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00006880"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006881Duplicate 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_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006885{
Victor Stinner8c62be82010-05-06 00:08:46 +00006886 int fd, fd2, res;
6887 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
6888 return NULL;
6889 if (!_PyVerify_fd_dup2(fd, fd2))
6890 return posix_error();
Victor Stinner8c62be82010-05-06 00:08:46 +00006891 res = dup2(fd, fd2);
Victor Stinner8c62be82010-05-06 00:08:46 +00006892 if (res < 0)
6893 return posix_error();
6894 Py_INCREF(Py_None);
6895 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006896}
6897
Ross Lagerwall7807c352011-03-17 20:20:30 +02006898#ifdef HAVE_LOCKF
6899PyDoc_STRVAR(posix_lockf__doc__,
6900"lockf(fd, cmd, len)\n\n\
6901Apply, test or remove a POSIX lock on an open file descriptor.\n\n\
6902fd is an open file descriptor.\n\
6903cmd specifies the command to use - one of F_LOCK, F_TLOCK, F_ULOCK or\n\
6904F_TEST.\n\
6905len specifies the section of the file to lock.");
6906
6907static PyObject *
6908posix_lockf(PyObject *self, PyObject *args)
6909{
6910 int fd, cmd, res;
6911 off_t len;
6912 if (!PyArg_ParseTuple(args, "iiO&:lockf",
6913 &fd, &cmd, _parse_off_t, &len))
6914 return NULL;
6915
6916 Py_BEGIN_ALLOW_THREADS
6917 res = lockf(fd, cmd, len);
6918 Py_END_ALLOW_THREADS
6919
6920 if (res < 0)
6921 return posix_error();
6922
6923 Py_RETURN_NONE;
6924}
6925#endif
6926
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006927
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006928PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006929"lseek(fd, pos, how) -> newpos\n\n\
Victor Stinnere83f8992011-12-17 23:15:09 +01006930Set the current position of a file descriptor.\n\
6931Return the new cursor position in bytes, starting from the beginning.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006932
Barry Warsaw53699e91996-12-10 23:23:01 +00006933static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006934posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006935{
Victor Stinner8c62be82010-05-06 00:08:46 +00006936 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006937#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00006938 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006939#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006940 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006941#endif
Ross Lagerwall8e749672011-03-17 21:54:07 +02006942 PyObject *posobj;
6943 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Victor Stinner8c62be82010-05-06 00:08:46 +00006944 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00006945#ifdef SEEK_SET
Victor Stinner8c62be82010-05-06 00:08:46 +00006946 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
6947 switch (how) {
6948 case 0: how = SEEK_SET; break;
6949 case 1: how = SEEK_CUR; break;
6950 case 2: how = SEEK_END; break;
6951 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006952#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00006953
Ross Lagerwall8e749672011-03-17 21:54:07 +02006954#if !defined(HAVE_LARGEFILE_SUPPORT)
6955 pos = PyLong_AsLong(posobj);
6956#else
6957 pos = PyLong_AsLongLong(posobj);
6958#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006959 if (PyErr_Occurred())
6960 return NULL;
Guido van Rossum94f6f721999-01-06 18:42:14 +00006961
Victor Stinner8c62be82010-05-06 00:08:46 +00006962 if (!_PyVerify_fd(fd))
6963 return posix_error();
6964 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006965#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00006966 res = _lseeki64(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00006967#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006968 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00006969#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006970 Py_END_ALLOW_THREADS
6971 if (res < 0)
6972 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00006973
6974#if !defined(HAVE_LARGEFILE_SUPPORT)
Victor Stinner8c62be82010-05-06 00:08:46 +00006975 return PyLong_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006976#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006977 return PyLong_FromLongLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006978#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00006979}
6980
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006981
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006982PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006983"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006984Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006985
Barry Warsaw53699e91996-12-10 23:23:01 +00006986static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006987posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006988{
Victor Stinner8c62be82010-05-06 00:08:46 +00006989 int fd, size;
6990 Py_ssize_t n;
6991 PyObject *buffer;
6992 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
6993 return NULL;
6994 if (size < 0) {
6995 errno = EINVAL;
6996 return posix_error();
6997 }
6998 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
6999 if (buffer == NULL)
7000 return NULL;
Stefan Krah99439262010-11-26 12:58:05 +00007001 if (!_PyVerify_fd(fd)) {
7002 Py_DECREF(buffer);
Victor Stinner8c62be82010-05-06 00:08:46 +00007003 return posix_error();
Stefan Krah99439262010-11-26 12:58:05 +00007004 }
Victor Stinner8c62be82010-05-06 00:08:46 +00007005 Py_BEGIN_ALLOW_THREADS
7006 n = read(fd, PyBytes_AS_STRING(buffer), size);
7007 Py_END_ALLOW_THREADS
7008 if (n < 0) {
7009 Py_DECREF(buffer);
7010 return posix_error();
7011 }
7012 if (n != size)
7013 _PyBytes_Resize(&buffer, n);
7014 return buffer;
Guido van Rossum687dd131993-05-17 08:34:16 +00007015}
7016
Ross Lagerwall7807c352011-03-17 20:20:30 +02007017#if (defined(HAVE_SENDFILE) && (defined(__FreeBSD__) || defined(__DragonFly__) \
7018 || defined(__APPLE__))) || defined(HAVE_READV) || defined(HAVE_WRITEV)
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007019static Py_ssize_t
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007020iov_setup(struct iovec **iov, Py_buffer **buf, PyObject *seq, int cnt, int type)
7021{
7022 int i, j;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007023 Py_ssize_t blen, total = 0;
7024
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007025 *iov = PyMem_New(struct iovec, cnt);
7026 if (*iov == NULL) {
7027 PyErr_NoMemory();
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007028 return total;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007029 }
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007030
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007031 *buf = PyMem_New(Py_buffer, cnt);
7032 if (*buf == NULL) {
7033 PyMem_Del(*iov);
7034 PyErr_NoMemory();
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007035 return total;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007036 }
7037
7038 for (i = 0; i < cnt; i++) {
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02007039 PyObject *item = PySequence_GetItem(seq, i);
7040 if (item == NULL)
7041 goto fail;
7042 if (PyObject_GetBuffer(item, &(*buf)[i], type) == -1) {
7043 Py_DECREF(item);
7044 goto fail;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007045 }
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02007046 Py_DECREF(item);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007047 (*iov)[i].iov_base = (*buf)[i].buf;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007048 blen = (*buf)[i].len;
7049 (*iov)[i].iov_len = blen;
7050 total += blen;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007051 }
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007052 return total;
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02007053
7054fail:
7055 PyMem_Del(*iov);
7056 for (j = 0; j < i; j++) {
7057 PyBuffer_Release(&(*buf)[j]);
7058 }
7059 PyMem_Del(*buf);
7060 return 0;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007061}
7062
7063static void
7064iov_cleanup(struct iovec *iov, Py_buffer *buf, int cnt)
7065{
7066 int i;
7067 PyMem_Del(iov);
7068 for (i = 0; i < cnt; i++) {
7069 PyBuffer_Release(&buf[i]);
7070 }
7071 PyMem_Del(buf);
7072}
7073#endif
7074
Ross Lagerwall7807c352011-03-17 20:20:30 +02007075#ifdef HAVE_READV
7076PyDoc_STRVAR(posix_readv__doc__,
7077"readv(fd, buffers) -> bytesread\n\n\
7078Read from a file descriptor into a number of writable buffers. buffers\n\
7079is an arbitrary sequence of writable buffers.\n\
7080Returns the total number of bytes read.");
7081
7082static PyObject *
7083posix_readv(PyObject *self, PyObject *args)
7084{
7085 int fd, cnt;
7086 Py_ssize_t n;
7087 PyObject *seq;
7088 struct iovec *iov;
7089 Py_buffer *buf;
7090
7091 if (!PyArg_ParseTuple(args, "iO:readv", &fd, &seq))
7092 return NULL;
7093 if (!PySequence_Check(seq)) {
7094 PyErr_SetString(PyExc_TypeError,
7095 "readv() arg 2 must be a sequence");
7096 return NULL;
7097 }
7098 cnt = PySequence_Size(seq);
7099
7100 if (!iov_setup(&iov, &buf, seq, cnt, PyBUF_WRITABLE))
7101 return NULL;
7102
7103 Py_BEGIN_ALLOW_THREADS
7104 n = readv(fd, iov, cnt);
7105 Py_END_ALLOW_THREADS
7106
7107 iov_cleanup(iov, buf, cnt);
7108 return PyLong_FromSsize_t(n);
7109}
7110#endif
7111
7112#ifdef HAVE_PREAD
7113PyDoc_STRVAR(posix_pread__doc__,
7114"pread(fd, buffersize, offset) -> string\n\n\
7115Read from a file descriptor, fd, at a position of offset. It will read up\n\
7116to buffersize number of bytes. The file offset remains unchanged.");
7117
7118static PyObject *
7119posix_pread(PyObject *self, PyObject *args)
7120{
7121 int fd, size;
7122 off_t offset;
7123 Py_ssize_t n;
7124 PyObject *buffer;
7125 if (!PyArg_ParseTuple(args, "iiO&:pread", &fd, &size, _parse_off_t, &offset))
7126 return NULL;
7127
7128 if (size < 0) {
7129 errno = EINVAL;
7130 return posix_error();
7131 }
7132 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
7133 if (buffer == NULL)
7134 return NULL;
7135 if (!_PyVerify_fd(fd)) {
7136 Py_DECREF(buffer);
7137 return posix_error();
7138 }
7139 Py_BEGIN_ALLOW_THREADS
7140 n = pread(fd, PyBytes_AS_STRING(buffer), size, offset);
7141 Py_END_ALLOW_THREADS
7142 if (n < 0) {
7143 Py_DECREF(buffer);
7144 return posix_error();
7145 }
7146 if (n != size)
7147 _PyBytes_Resize(&buffer, n);
7148 return buffer;
7149}
7150#endif
7151
7152PyDoc_STRVAR(posix_write__doc__,
7153"write(fd, string) -> byteswritten\n\n\
7154Write a string to a file descriptor.");
7155
7156static PyObject *
7157posix_write(PyObject *self, PyObject *args)
7158{
7159 Py_buffer pbuf;
7160 int fd;
7161 Py_ssize_t size, len;
7162
7163 if (!PyArg_ParseTuple(args, "iy*:write", &fd, &pbuf))
7164 return NULL;
7165 if (!_PyVerify_fd(fd)) {
7166 PyBuffer_Release(&pbuf);
7167 return posix_error();
7168 }
7169 len = pbuf.len;
7170 Py_BEGIN_ALLOW_THREADS
7171#if defined(MS_WIN64) || defined(MS_WINDOWS)
7172 if (len > INT_MAX)
7173 len = INT_MAX;
7174 size = write(fd, pbuf.buf, (int)len);
7175#else
7176 size = write(fd, pbuf.buf, len);
7177#endif
7178 Py_END_ALLOW_THREADS
7179 PyBuffer_Release(&pbuf);
7180 if (size < 0)
7181 return posix_error();
7182 return PyLong_FromSsize_t(size);
7183}
7184
7185#ifdef HAVE_SENDFILE
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007186PyDoc_STRVAR(posix_sendfile__doc__,
7187"sendfile(out, in, offset, nbytes) -> byteswritten\n\
7188sendfile(out, in, offset, nbytes, headers=None, trailers=None, flags=0)\n\
7189 -> byteswritten\n\
7190Copy nbytes bytes from file descriptor in to file descriptor out.");
7191
7192static PyObject *
7193posix_sendfile(PyObject *self, PyObject *args, PyObject *kwdict)
7194{
7195 int in, out;
7196 Py_ssize_t ret;
7197 off_t offset;
7198
7199#if defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__)
7200#ifndef __APPLE__
7201 Py_ssize_t len;
7202#endif
7203 PyObject *headers = NULL, *trailers = NULL;
7204 Py_buffer *hbuf, *tbuf;
7205 off_t sbytes;
7206 struct sf_hdtr sf;
7207 int flags = 0;
7208 sf.headers = NULL;
7209 sf.trailers = NULL;
Benjamin Petersond8a43b42011-02-26 21:35:16 +00007210 static char *keywords[] = {"out", "in",
7211 "offset", "count",
7212 "headers", "trailers", "flags", NULL};
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007213
7214#ifdef __APPLE__
7215 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiO&O&|OOi:sendfile",
Georg Brandla391b112011-02-25 15:23:18 +00007216 keywords, &out, &in, _parse_off_t, &offset, _parse_off_t, &sbytes,
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007217#else
7218 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiO&n|OOi:sendfile",
Georg Brandla391b112011-02-25 15:23:18 +00007219 keywords, &out, &in, _parse_off_t, &offset, &len,
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007220#endif
7221 &headers, &trailers, &flags))
7222 return NULL;
7223 if (headers != NULL) {
7224 if (!PySequence_Check(headers)) {
7225 PyErr_SetString(PyExc_TypeError,
7226 "sendfile() headers must be a sequence or None");
7227 return NULL;
7228 } else {
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007229 Py_ssize_t i = 0; /* Avoid uninitialized warning */
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007230 sf.hdr_cnt = PySequence_Size(headers);
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007231 if (sf.hdr_cnt > 0 &&
7232 !(i = iov_setup(&(sf.headers), &hbuf,
7233 headers, sf.hdr_cnt, PyBUF_SIMPLE)))
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007234 return NULL;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007235#ifdef __APPLE__
7236 sbytes += i;
7237#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007238 }
7239 }
7240 if (trailers != NULL) {
7241 if (!PySequence_Check(trailers)) {
7242 PyErr_SetString(PyExc_TypeError,
7243 "sendfile() trailers must be a sequence or None");
7244 return NULL;
7245 } else {
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007246 Py_ssize_t i = 0; /* Avoid uninitialized warning */
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007247 sf.trl_cnt = PySequence_Size(trailers);
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007248 if (sf.trl_cnt > 0 &&
7249 !(i = iov_setup(&(sf.trailers), &tbuf,
7250 trailers, sf.trl_cnt, PyBUF_SIMPLE)))
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007251 return NULL;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00007252#ifdef __APPLE__
7253 sbytes += i;
7254#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007255 }
7256 }
7257
7258 Py_BEGIN_ALLOW_THREADS
7259#ifdef __APPLE__
7260 ret = sendfile(in, out, offset, &sbytes, &sf, flags);
7261#else
7262 ret = sendfile(in, out, offset, len, &sf, &sbytes, flags);
7263#endif
7264 Py_END_ALLOW_THREADS
7265
7266 if (sf.headers != NULL)
7267 iov_cleanup(sf.headers, hbuf, sf.hdr_cnt);
7268 if (sf.trailers != NULL)
7269 iov_cleanup(sf.trailers, tbuf, sf.trl_cnt);
7270
7271 if (ret < 0) {
7272 if ((errno == EAGAIN) || (errno == EBUSY)) {
7273 if (sbytes != 0) {
7274 // some data has been sent
7275 goto done;
7276 }
7277 else {
7278 // no data has been sent; upper application is supposed
7279 // to retry on EAGAIN or EBUSY
7280 return posix_error();
7281 }
7282 }
7283 return posix_error();
7284 }
7285 goto done;
7286
7287done:
7288 #if !defined(HAVE_LARGEFILE_SUPPORT)
7289 return Py_BuildValue("l", sbytes);
7290 #else
7291 return Py_BuildValue("L", sbytes);
7292 #endif
7293
7294#else
7295 Py_ssize_t count;
7296 PyObject *offobj;
7297 static char *keywords[] = {"out", "in",
7298 "offset", "count", NULL};
7299 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiOn:sendfile",
7300 keywords, &out, &in, &offobj, &count))
7301 return NULL;
7302#ifdef linux
7303 if (offobj == Py_None) {
7304 Py_BEGIN_ALLOW_THREADS
7305 ret = sendfile(out, in, NULL, count);
7306 Py_END_ALLOW_THREADS
7307 if (ret < 0)
7308 return posix_error();
Giampaolo Rodola'ff1a7352011-04-19 09:47:16 +02007309 return Py_BuildValue("n", ret);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007310 }
7311#endif
Antoine Pitroudcc20b82011-02-26 13:38:35 +00007312 if (!_parse_off_t(offobj, &offset))
7313 return NULL;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00007314 Py_BEGIN_ALLOW_THREADS
7315 ret = sendfile(out, in, &offset, count);
7316 Py_END_ALLOW_THREADS
7317 if (ret < 0)
7318 return posix_error();
7319 return Py_BuildValue("n", ret);
7320#endif
7321}
7322#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007323
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007324PyDoc_STRVAR(posix_fstat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01007325"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007326Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007327
Barry Warsaw53699e91996-12-10 23:23:01 +00007328static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01007329posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00007330{
Victor Stinner8c62be82010-05-06 00:08:46 +00007331 int fd;
7332 STRUCT_STAT st;
7333 int res;
Victor Stinner4195b5c2012-02-08 23:03:19 +01007334 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
Victor Stinner8c62be82010-05-06 00:08:46 +00007335 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00007336#ifdef __VMS
Victor Stinner8c62be82010-05-06 00:08:46 +00007337 /* on OpenVMS we must ensure that all bytes are written to the file */
7338 fsync(fd);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00007339#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00007340 if (!_PyVerify_fd(fd))
7341 return posix_error();
7342 Py_BEGIN_ALLOW_THREADS
7343 res = FSTAT(fd, &st);
7344 Py_END_ALLOW_THREADS
7345 if (res != 0) {
Martin v. Löwis14694662006-02-03 12:54:16 +00007346#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00007347 return win32_error("fstat", NULL);
Martin v. Löwis14694662006-02-03 12:54:16 +00007348#else
Victor Stinner8c62be82010-05-06 00:08:46 +00007349 return posix_error();
Martin v. Löwis14694662006-02-03 12:54:16 +00007350#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00007351 }
Tim Peters5aa91602002-01-30 05:46:57 +00007352
Victor Stinner4195b5c2012-02-08 23:03:19 +01007353 return _pystat_fromstructstat(&st);
Guido van Rossum687dd131993-05-17 08:34:16 +00007354}
7355
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007356PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007357"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00007358Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007359connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00007360
7361static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00007362posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00007363{
Victor Stinner8c62be82010-05-06 00:08:46 +00007364 int fd;
7365 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
7366 return NULL;
7367 if (!_PyVerify_fd(fd))
7368 return PyBool_FromLong(0);
7369 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00007370}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007371
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007372#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007373PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007374"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007375Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007376
Barry Warsaw53699e91996-12-10 23:23:01 +00007377static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00007378posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00007379{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007380#if defined(PYOS_OS2)
7381 HFILE read, write;
7382 APIRET rc;
7383
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007384 rc = DosCreatePipe( &read, &write, 4096);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007385 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00007386 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007387
7388 return Py_BuildValue("(ii)", read, write);
7389#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007390#if !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00007391 int fds[2];
7392 int res;
Victor Stinner8c62be82010-05-06 00:08:46 +00007393 res = pipe(fds);
Victor Stinner8c62be82010-05-06 00:08:46 +00007394 if (res != 0)
7395 return posix_error();
7396 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007397#else /* MS_WINDOWS */
Victor Stinner8c62be82010-05-06 00:08:46 +00007398 HANDLE read, write;
7399 int read_fd, write_fd;
7400 BOOL ok;
Victor Stinner8c62be82010-05-06 00:08:46 +00007401 ok = CreatePipe(&read, &write, NULL, 0);
Victor Stinner8c62be82010-05-06 00:08:46 +00007402 if (!ok)
7403 return win32_error("CreatePipe", NULL);
7404 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
7405 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
7406 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007407#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00007408#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00007409}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007410#endif /* HAVE_PIPE */
7411
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007412#ifdef HAVE_PIPE2
7413PyDoc_STRVAR(posix_pipe2__doc__,
Charles-François Natali368f34b2011-06-06 19:49:47 +02007414"pipe2(flags) -> (read_end, write_end)\n\n\
7415Create a pipe with flags set atomically.\n\
7416flags can be constructed by ORing together one or more of these values:\n\
7417O_NONBLOCK, O_CLOEXEC.\n\
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007418");
7419
7420static PyObject *
Charles-François Natali368f34b2011-06-06 19:49:47 +02007421posix_pipe2(PyObject *self, PyObject *arg)
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007422{
Charles-François Natali368f34b2011-06-06 19:49:47 +02007423 int flags;
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007424 int fds[2];
7425 int res;
7426
Charles-François Natali368f34b2011-06-06 19:49:47 +02007427 flags = PyLong_AsLong(arg);
7428 if (flags == -1 && PyErr_Occurred())
Charles-François Natalidaafdd52011-05-29 20:07:40 +02007429 return NULL;
7430
7431 res = pipe2(fds, flags);
7432 if (res != 0)
7433 return posix_error();
7434 return Py_BuildValue("(ii)", fds[0], fds[1]);
7435}
7436#endif /* HAVE_PIPE2 */
7437
Ross Lagerwall7807c352011-03-17 20:20:30 +02007438#ifdef HAVE_WRITEV
7439PyDoc_STRVAR(posix_writev__doc__,
7440"writev(fd, buffers) -> byteswritten\n\n\
7441Write the contents of buffers to a file descriptor, where buffers is an\n\
7442arbitrary sequence of buffers.\n\
7443Returns the total bytes written.");
7444
7445static PyObject *
7446posix_writev(PyObject *self, PyObject *args)
7447{
7448 int fd, cnt;
7449 Py_ssize_t res;
7450 PyObject *seq;
7451 struct iovec *iov;
7452 Py_buffer *buf;
7453 if (!PyArg_ParseTuple(args, "iO:writev", &fd, &seq))
7454 return NULL;
7455 if (!PySequence_Check(seq)) {
7456 PyErr_SetString(PyExc_TypeError,
7457 "writev() arg 2 must be a sequence");
7458 return NULL;
7459 }
7460 cnt = PySequence_Size(seq);
7461
7462 if (!iov_setup(&iov, &buf, seq, cnt, PyBUF_SIMPLE)) {
7463 return NULL;
7464 }
7465
7466 Py_BEGIN_ALLOW_THREADS
7467 res = writev(fd, iov, cnt);
7468 Py_END_ALLOW_THREADS
7469
7470 iov_cleanup(iov, buf, cnt);
7471 return PyLong_FromSsize_t(res);
7472}
7473#endif
7474
7475#ifdef HAVE_PWRITE
7476PyDoc_STRVAR(posix_pwrite__doc__,
7477"pwrite(fd, string, offset) -> byteswritten\n\n\
7478Write string to a file descriptor, fd, from offset, leaving the file\n\
7479offset unchanged.");
7480
7481static PyObject *
7482posix_pwrite(PyObject *self, PyObject *args)
7483{
7484 Py_buffer pbuf;
7485 int fd;
7486 off_t offset;
7487 Py_ssize_t size;
7488
7489 if (!PyArg_ParseTuple(args, "iy*O&:pwrite", &fd, &pbuf, _parse_off_t, &offset))
7490 return NULL;
7491
7492 if (!_PyVerify_fd(fd)) {
7493 PyBuffer_Release(&pbuf);
7494 return posix_error();
7495 }
7496 Py_BEGIN_ALLOW_THREADS
7497 size = pwrite(fd, pbuf.buf, (size_t)pbuf.len, offset);
7498 Py_END_ALLOW_THREADS
7499 PyBuffer_Release(&pbuf);
7500 if (size < 0)
7501 return posix_error();
7502 return PyLong_FromSsize_t(size);
7503}
7504#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007505
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007506#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007507PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00007508"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007509Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007510
Barry Warsaw53699e91996-12-10 23:23:01 +00007511static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007512posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007513{
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007514 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00007515 char *filename;
7516 int mode = 0666;
7517 int res;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007518 if (!PyArg_ParseTuple(args, "O&|i:mkfifo", PyUnicode_FSConverter, &opath,
7519 &mode))
Victor Stinner8c62be82010-05-06 00:08:46 +00007520 return NULL;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007521 filename = PyBytes_AS_STRING(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007522 Py_BEGIN_ALLOW_THREADS
7523 res = mkfifo(filename, mode);
7524 Py_END_ALLOW_THREADS
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007525 Py_DECREF(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007526 if (res < 0)
7527 return posix_error();
7528 Py_INCREF(Py_None);
7529 return Py_None;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007530}
7531#endif
7532
7533
Neal Norwitz11690112002-07-30 01:08:28 +00007534#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007535PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00007536"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007537Create a filesystem node (file, device special file or named pipe)\n\
7538named filename. mode specifies both the permissions to use and the\n\
7539type of node to be created, being combined (bitwise OR) with one of\n\
7540S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007541device defines the newly created device special file (probably using\n\
7542os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00007543
7544
7545static PyObject *
7546posix_mknod(PyObject *self, PyObject *args)
7547{
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007548 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00007549 char *filename;
7550 int mode = 0600;
7551 int device = 0;
7552 int res;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007553 if (!PyArg_ParseTuple(args, "O&|ii:mknod", PyUnicode_FSConverter, &opath,
7554 &mode, &device))
Victor Stinner8c62be82010-05-06 00:08:46 +00007555 return NULL;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007556 filename = PyBytes_AS_STRING(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007557 Py_BEGIN_ALLOW_THREADS
7558 res = mknod(filename, mode, device);
7559 Py_END_ALLOW_THREADS
Benjamin Petersond4efbf92010-08-11 19:20:42 +00007560 Py_DECREF(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00007561 if (res < 0)
7562 return posix_error();
7563 Py_INCREF(Py_None);
7564 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007565}
7566#endif
7567
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007568#ifdef HAVE_DEVICE_MACROS
7569PyDoc_STRVAR(posix_major__doc__,
7570"major(device) -> major number\n\
7571Extracts a device major number from a raw device number.");
7572
7573static PyObject *
7574posix_major(PyObject *self, PyObject *args)
7575{
Victor Stinner8c62be82010-05-06 00:08:46 +00007576 int device;
7577 if (!PyArg_ParseTuple(args, "i:major", &device))
7578 return NULL;
7579 return PyLong_FromLong((long)major(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007580}
7581
7582PyDoc_STRVAR(posix_minor__doc__,
7583"minor(device) -> minor number\n\
7584Extracts a device minor number from a raw device number.");
7585
7586static PyObject *
7587posix_minor(PyObject *self, PyObject *args)
7588{
Victor Stinner8c62be82010-05-06 00:08:46 +00007589 int device;
7590 if (!PyArg_ParseTuple(args, "i:minor", &device))
7591 return NULL;
7592 return PyLong_FromLong((long)minor(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007593}
7594
7595PyDoc_STRVAR(posix_makedev__doc__,
7596"makedev(major, minor) -> device number\n\
7597Composes a raw device number from the major and minor device numbers.");
7598
7599static PyObject *
7600posix_makedev(PyObject *self, PyObject *args)
7601{
Victor Stinner8c62be82010-05-06 00:08:46 +00007602 int major, minor;
7603 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
7604 return NULL;
7605 return PyLong_FromLong((long)makedev(major, minor));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00007606}
7607#endif /* device macros */
7608
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007609
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007610#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007611PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007612"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007613Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007614
Barry Warsaw53699e91996-12-10 23:23:01 +00007615static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007616posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007617{
Victor Stinner8c62be82010-05-06 00:08:46 +00007618 int fd;
7619 off_t length;
7620 int res;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007621
Ross Lagerwall7807c352011-03-17 20:20:30 +02007622 if (!PyArg_ParseTuple(args, "iO&:ftruncate", &fd, _parse_off_t, &length))
Victor Stinner8c62be82010-05-06 00:08:46 +00007623 return NULL;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007624
Victor Stinner8c62be82010-05-06 00:08:46 +00007625 Py_BEGIN_ALLOW_THREADS
7626 res = ftruncate(fd, length);
7627 Py_END_ALLOW_THREADS
7628 if (res < 0)
7629 return posix_error();
7630 Py_INCREF(Py_None);
7631 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00007632}
7633#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00007634
Ross Lagerwall7807c352011-03-17 20:20:30 +02007635#ifdef HAVE_TRUNCATE
7636PyDoc_STRVAR(posix_truncate__doc__,
7637"truncate(path, length)\n\n\
7638Truncate the file given by path to length bytes.");
7639
7640static PyObject *
7641posix_truncate(PyObject *self, PyObject *args)
7642{
7643 PyObject *opath;
7644 const char *path;
7645 off_t length;
7646 int res;
7647
7648 if (!PyArg_ParseTuple(args, "O&O&:truncate",
7649 PyUnicode_FSConverter, &opath, _parse_off_t, &length))
7650 return NULL;
7651 path = PyBytes_AsString(opath);
7652
7653 Py_BEGIN_ALLOW_THREADS
7654 res = truncate(path, length);
7655 Py_END_ALLOW_THREADS
7656 Py_DECREF(opath);
7657 if (res < 0)
7658 return posix_error();
7659 Py_RETURN_NONE;
7660}
7661#endif
7662
7663#ifdef HAVE_POSIX_FALLOCATE
7664PyDoc_STRVAR(posix_posix_fallocate__doc__,
7665"posix_fallocate(fd, offset, len)\n\n\
7666Ensures that enough disk space is allocated for the file specified by fd\n\
7667starting from offset and continuing for len bytes.");
7668
7669static PyObject *
7670posix_posix_fallocate(PyObject *self, PyObject *args)
7671{
7672 off_t len, offset;
7673 int res, fd;
7674
7675 if (!PyArg_ParseTuple(args, "iO&O&:posix_fallocate",
7676 &fd, _parse_off_t, &offset, _parse_off_t, &len))
7677 return NULL;
7678
7679 Py_BEGIN_ALLOW_THREADS
7680 res = posix_fallocate(fd, offset, len);
7681 Py_END_ALLOW_THREADS
7682 if (res != 0) {
7683 errno = res;
7684 return posix_error();
7685 }
7686 Py_RETURN_NONE;
7687}
7688#endif
7689
7690#ifdef HAVE_POSIX_FADVISE
7691PyDoc_STRVAR(posix_posix_fadvise__doc__,
7692"posix_fadvise(fd, offset, len, advice)\n\n\
7693Announces an intention to access data in a specific pattern thus allowing\n\
7694the kernel to make optimizations.\n\
7695The advice applies to the region of the file specified by fd starting at\n\
7696offset and continuing for len bytes.\n\
7697advice is one of POSIX_FADV_NORMAL, POSIX_FADV_SEQUENTIAL,\n\
7698POSIX_FADV_RANDOM, POSIX_FADV_NOREUSE, POSIX_FADV_WILLNEED or\n\
7699POSIX_FADV_DONTNEED.");
7700
7701static PyObject *
7702posix_posix_fadvise(PyObject *self, PyObject *args)
7703{
7704 off_t len, offset;
7705 int res, fd, advice;
7706
7707 if (!PyArg_ParseTuple(args, "iO&O&i:posix_fadvise",
7708 &fd, _parse_off_t, &offset, _parse_off_t, &len, &advice))
7709 return NULL;
7710
7711 Py_BEGIN_ALLOW_THREADS
7712 res = posix_fadvise(fd, offset, len, advice);
7713 Py_END_ALLOW_THREADS
7714 if (res != 0) {
7715 errno = res;
7716 return posix_error();
7717 }
7718 Py_RETURN_NONE;
7719}
7720#endif
7721
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007722#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007723PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007724"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007725Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00007726
Fred Drake762e2061999-08-26 17:23:54 +00007727/* Save putenv() parameters as values here, so we can collect them when they
7728 * get re-set with another call for the same key. */
7729static PyObject *posix_putenv_garbage;
7730
Tim Peters5aa91602002-01-30 05:46:57 +00007731static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007732posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007733{
Victor Stinner84ae1182010-05-06 22:05:07 +00007734 PyObject *newstr = NULL;
Thomas Hellerf78f12a2007-11-08 19:33:05 +00007735#ifdef MS_WINDOWS
Victor Stinner65170952011-11-22 22:16:17 +01007736 PyObject *os1, *os2;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007737 wchar_t *newenv;
Victor Stinner65170952011-11-22 22:16:17 +01007738
Victor Stinner8c62be82010-05-06 00:08:46 +00007739 if (!PyArg_ParseTuple(args,
Victor Stinner9d3b93b2011-11-22 02:27:30 +01007740 "UU:putenv",
Victor Stinner65170952011-11-22 22:16:17 +01007741 &os1, &os2))
Victor Stinner8c62be82010-05-06 00:08:46 +00007742 return NULL;
Guido van Rossumd48f2521997-12-05 22:19:34 +00007743
Victor Stinner65170952011-11-22 22:16:17 +01007744 newstr = PyUnicode_FromFormat("%U=%U", os1, os2);
Victor Stinner84ae1182010-05-06 22:05:07 +00007745 if (newstr == NULL) {
7746 PyErr_NoMemory();
7747 goto error;
7748 }
Victor Stinner65170952011-11-22 22:16:17 +01007749 if (_MAX_ENV < PyUnicode_GET_LENGTH(newstr)) {
7750 PyErr_Format(PyExc_ValueError,
7751 "the environment variable is longer than %u characters",
7752 _MAX_ENV);
7753 goto error;
7754 }
7755
Victor Stinner8c62be82010-05-06 00:08:46 +00007756 newenv = PyUnicode_AsUnicode(newstr);
Victor Stinnereb5657a2011-09-30 01:44:27 +02007757 if (newenv == NULL)
7758 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00007759 if (_wputenv(newenv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007760 posix_error();
Victor Stinner84ae1182010-05-06 22:05:07 +00007761 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00007762 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00007763#else
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007764 PyObject *os1, *os2;
7765 char *s1, *s2;
7766 char *newenv;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007767
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007768 if (!PyArg_ParseTuple(args,
7769 "O&O&:putenv",
7770 PyUnicode_FSConverter, &os1,
7771 PyUnicode_FSConverter, &os2))
7772 return NULL;
7773 s1 = PyBytes_AsString(os1);
7774 s2 = PyBytes_AsString(os2);
Guido van Rossumad0ee831995-03-01 10:34:45 +00007775
Victor Stinner65170952011-11-22 22:16:17 +01007776 newstr = PyBytes_FromFormat("%s=%s", s1, s2);
Victor Stinner9d3b93b2011-11-22 02:27:30 +01007777 if (newstr == NULL) {
7778 PyErr_NoMemory();
7779 goto error;
7780 }
7781
Victor Stinner8c62be82010-05-06 00:08:46 +00007782 newenv = PyBytes_AS_STRING(newstr);
Victor Stinner8c62be82010-05-06 00:08:46 +00007783 if (putenv(newenv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007784 posix_error();
Victor Stinner84ae1182010-05-06 22:05:07 +00007785 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00007786 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00007787#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007788
Victor Stinner8c62be82010-05-06 00:08:46 +00007789 /* Install the first arg and newstr in posix_putenv_garbage;
7790 * this will cause previous value to be collected. This has to
7791 * happen after the real putenv() call because the old value
7792 * was still accessible until then. */
Victor Stinner65170952011-11-22 22:16:17 +01007793 if (PyDict_SetItem(posix_putenv_garbage, os1, newstr)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007794 /* really not much we can do; just leak */
7795 PyErr_Clear();
7796 }
7797 else {
7798 Py_DECREF(newstr);
7799 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00007800
Martin v. Löwis011e8422009-05-05 04:43:17 +00007801#ifndef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00007802 Py_DECREF(os1);
7803 Py_DECREF(os2);
Martin v. Löwis011e8422009-05-05 04:43:17 +00007804#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007805 Py_RETURN_NONE;
7806
7807error:
7808#ifndef MS_WINDOWS
7809 Py_DECREF(os1);
7810 Py_DECREF(os2);
7811#endif
7812 Py_XDECREF(newstr);
7813 return NULL;
Guido van Rossumb6775db1994-08-01 11:34:53 +00007814}
Guido van Rossumb6a47161997-09-15 22:54:34 +00007815#endif /* putenv */
7816
Guido van Rossumc524d952001-10-19 01:31:59 +00007817#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007818PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007819"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007820Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00007821
7822static PyObject *
7823posix_unsetenv(PyObject *self, PyObject *args)
7824{
Victor Stinner65170952011-11-22 22:16:17 +01007825 PyObject *name;
Victor Stinner984890f2011-11-24 13:53:38 +01007826#ifndef HAVE_BROKEN_UNSETENV
Victor Stinner60b385e2011-11-22 22:01:28 +01007827 int err;
Victor Stinner984890f2011-11-24 13:53:38 +01007828#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007829
7830 if (!PyArg_ParseTuple(args, "O&:unsetenv",
Guido van Rossumc524d952001-10-19 01:31:59 +00007831
Victor Stinner65170952011-11-22 22:16:17 +01007832 PyUnicode_FSConverter, &name))
Guido van Rossumc524d952001-10-19 01:31:59 +00007833 return NULL;
Guido van Rossumc524d952001-10-19 01:31:59 +00007834
Victor Stinner984890f2011-11-24 13:53:38 +01007835#ifdef HAVE_BROKEN_UNSETENV
7836 unsetenv(PyBytes_AS_STRING(name));
7837#else
Victor Stinner65170952011-11-22 22:16:17 +01007838 err = unsetenv(PyBytes_AS_STRING(name));
Benjamin Peterson4bb867d2011-11-22 23:12:49 -06007839 if (err) {
Benjamin Petersone8eb0e82011-11-22 23:14:47 -06007840 Py_DECREF(name);
Victor Stinner60b385e2011-11-22 22:01:28 +01007841 return posix_error();
Benjamin Peterson4bb867d2011-11-22 23:12:49 -06007842 }
Victor Stinner984890f2011-11-24 13:53:38 +01007843#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00007844
Victor Stinner8c62be82010-05-06 00:08:46 +00007845 /* Remove the key from posix_putenv_garbage;
7846 * this will cause it to be collected. This has to
7847 * happen after the real unsetenv() call because the
7848 * old value was still accessible until then.
7849 */
Victor Stinner65170952011-11-22 22:16:17 +01007850 if (PyDict_DelItem(posix_putenv_garbage, name)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007851 /* really not much we can do; just leak */
7852 PyErr_Clear();
7853 }
Victor Stinner65170952011-11-22 22:16:17 +01007854 Py_DECREF(name);
Victor Stinner84ae1182010-05-06 22:05:07 +00007855 Py_RETURN_NONE;
Guido van Rossumc524d952001-10-19 01:31:59 +00007856}
7857#endif /* unsetenv */
7858
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007859PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007860"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007861Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00007862
Guido van Rossumf68d8e52001-04-14 17:55:09 +00007863static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007864posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00007865{
Victor Stinner8c62be82010-05-06 00:08:46 +00007866 int code;
7867 char *message;
7868 if (!PyArg_ParseTuple(args, "i:strerror", &code))
7869 return NULL;
7870 message = strerror(code);
7871 if (message == NULL) {
7872 PyErr_SetString(PyExc_ValueError,
7873 "strerror() argument out of range");
7874 return NULL;
7875 }
Victor Stinner1b579672011-12-17 05:47:23 +01007876 return PyUnicode_DecodeLocale(message, "surrogateescape");
Guido van Rossumb6a47161997-09-15 22:54:34 +00007877}
Guido van Rossumb6a47161997-09-15 22:54:34 +00007878
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00007879
Guido van Rossumc9641791998-08-04 15:26:23 +00007880#ifdef HAVE_SYS_WAIT_H
7881
Fred Drake106c1a02002-04-23 15:58:02 +00007882#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007883PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007884"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007885Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00007886
7887static PyObject *
7888posix_WCOREDUMP(PyObject *self, PyObject *args)
7889{
Victor Stinner8c62be82010-05-06 00:08:46 +00007890 WAIT_TYPE status;
7891 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00007892
Victor Stinner8c62be82010-05-06 00:08:46 +00007893 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
7894 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00007895
Victor Stinner8c62be82010-05-06 00:08:46 +00007896 return PyBool_FromLong(WCOREDUMP(status));
Fred Drake106c1a02002-04-23 15:58:02 +00007897}
7898#endif /* WCOREDUMP */
7899
7900#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007901PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007902"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00007903Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007904job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00007905
7906static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00007907posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00007908{
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:WCONTINUED", &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(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00007916}
7917#endif /* WIFCONTINUED */
7918
Guido van Rossumc9641791998-08-04 15:26:23 +00007919#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007920PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007921"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007922Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007923
7924static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007925posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007926{
Victor Stinner8c62be82010-05-06 00:08:46 +00007927 WAIT_TYPE status;
7928 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007929
Victor Stinner8c62be82010-05-06 00:08:46 +00007930 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
7931 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007932
Victor Stinner8c62be82010-05-06 00:08:46 +00007933 return PyBool_FromLong(WIFSTOPPED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007934}
7935#endif /* WIFSTOPPED */
7936
7937#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007938PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007939"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007940Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007941
7942static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007943posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007944{
Victor Stinner8c62be82010-05-06 00:08:46 +00007945 WAIT_TYPE status;
7946 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007947
Victor Stinner8c62be82010-05-06 00:08:46 +00007948 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
7949 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007950
Victor Stinner8c62be82010-05-06 00:08:46 +00007951 return PyBool_FromLong(WIFSIGNALED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007952}
7953#endif /* WIFSIGNALED */
7954
7955#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007956PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007957"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00007958Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007959system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007960
7961static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007962posix_WIFEXITED(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:WIFEXITED", &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(WIFEXITED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007971}
7972#endif /* WIFEXITED */
7973
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00007974#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007975PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007976"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007977Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007978
7979static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007980posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007981{
Victor Stinner8c62be82010-05-06 00:08:46 +00007982 WAIT_TYPE status;
7983 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007984
Victor Stinner8c62be82010-05-06 00:08:46 +00007985 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
7986 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007987
Victor Stinner8c62be82010-05-06 00:08:46 +00007988 return Py_BuildValue("i", WEXITSTATUS(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007989}
7990#endif /* WEXITSTATUS */
7991
7992#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007993PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007994"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00007995Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007996value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007997
7998static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007999posix_WTERMSIG(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:WTERMSIG", &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", WTERMSIG(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008008}
8009#endif /* WTERMSIG */
8010
8011#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008012PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008013"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008014Return the signal that stopped the process that provided\n\
8015the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00008016
8017static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008018posix_WSTOPSIG(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:WSTOPSIG", &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", WSTOPSIG(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00008027}
8028#endif /* WSTOPSIG */
8029
8030#endif /* HAVE_SYS_WAIT_H */
8031
8032
Thomas Wouters477c8d52006-05-27 19:21:47 +00008033#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossumd5753e11999-10-19 13:29:23 +00008034#ifdef _SCO_DS
8035/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
8036 needed definitions in sys/statvfs.h */
8037#define _SVID3
8038#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00008039#include <sys/statvfs.h>
8040
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008041static PyObject*
8042_pystatvfs_fromstructstatvfs(struct statvfs st) {
Victor Stinner8c62be82010-05-06 00:08:46 +00008043 PyObject *v = PyStructSequence_New(&StatVFSResultType);
8044 if (v == NULL)
8045 return NULL;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008046
8047#if !defined(HAVE_LARGEFILE_SUPPORT)
Victor Stinner8c62be82010-05-06 00:08:46 +00008048 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
8049 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
8050 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long) st.f_blocks));
8051 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long) st.f_bfree));
8052 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long) st.f_bavail));
8053 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long) st.f_files));
8054 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong((long) st.f_ffree));
8055 PyStructSequence_SET_ITEM(v, 7, PyLong_FromLong((long) st.f_favail));
8056 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
8057 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008058#else
Victor Stinner8c62be82010-05-06 00:08:46 +00008059 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
8060 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
8061 PyStructSequence_SET_ITEM(v, 2,
8062 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
8063 PyStructSequence_SET_ITEM(v, 3,
8064 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
8065 PyStructSequence_SET_ITEM(v, 4,
8066 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
8067 PyStructSequence_SET_ITEM(v, 5,
8068 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
8069 PyStructSequence_SET_ITEM(v, 6,
8070 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
8071 PyStructSequence_SET_ITEM(v, 7,
8072 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
8073 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
8074 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008075#endif
8076
Victor Stinner8c62be82010-05-06 00:08:46 +00008077 return v;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008078}
8079
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008080PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008081"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008082Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00008083
8084static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008085posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00008086{
Victor Stinner8c62be82010-05-06 00:08:46 +00008087 int fd, res;
8088 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008089
Victor Stinner8c62be82010-05-06 00:08:46 +00008090 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
8091 return NULL;
8092 Py_BEGIN_ALLOW_THREADS
8093 res = fstatvfs(fd, &st);
8094 Py_END_ALLOW_THREADS
8095 if (res != 0)
8096 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008097
Victor Stinner8c62be82010-05-06 00:08:46 +00008098 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00008099}
Thomas Wouters477c8d52006-05-27 19:21:47 +00008100#endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00008101
8102
Thomas Wouters477c8d52006-05-27 19:21:47 +00008103#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossum94f6f721999-01-06 18:42:14 +00008104#include <sys/statvfs.h>
8105
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008106PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008107"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008108Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00008109
8110static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008111posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00008112{
Victor Stinner6fa67772011-09-20 04:04:33 +02008113 PyObject *path;
Victor Stinner8c62be82010-05-06 00:08:46 +00008114 int res;
8115 struct statvfs st;
Victor Stinner6fa67772011-09-20 04:04:33 +02008116 if (!PyArg_ParseTuple(args, "O&:statvfs", PyUnicode_FSConverter, &path))
Victor Stinner8c62be82010-05-06 00:08:46 +00008117 return NULL;
8118 Py_BEGIN_ALLOW_THREADS
Victor Stinner6fa67772011-09-20 04:04:33 +02008119 res = statvfs(PyBytes_AS_STRING(path), &st);
Victor Stinner8c62be82010-05-06 00:08:46 +00008120 Py_END_ALLOW_THREADS
Victor Stinner6fa67772011-09-20 04:04:33 +02008121 if (res != 0) {
8122 posix_error_with_filename(PyBytes_AS_STRING(path));
8123 Py_DECREF(path);
8124 return NULL;
8125 }
8126 Py_DECREF(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00008127
Victor Stinner8c62be82010-05-06 00:08:46 +00008128 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00008129}
8130#endif /* HAVE_STATVFS */
8131
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008132#ifdef MS_WINDOWS
8133PyDoc_STRVAR(win32__getdiskusage__doc__,
8134"_getdiskusage(path) -> (total, free)\n\n\
8135Return disk usage statistics about the given path as (total, free) tuple.");
8136
8137static PyObject *
8138win32__getdiskusage(PyObject *self, PyObject *args)
8139{
8140 BOOL retval;
8141 ULARGE_INTEGER _, total, free;
Victor Stinner6139c1b2011-11-09 22:14:14 +01008142 const wchar_t *path;
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008143
Victor Stinner6139c1b2011-11-09 22:14:14 +01008144 if (! PyArg_ParseTuple(args, "u", &path))
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008145 return NULL;
8146
8147 Py_BEGIN_ALLOW_THREADS
Victor Stinner6139c1b2011-11-09 22:14:14 +01008148 retval = GetDiskFreeSpaceExW(path, &_, &total, &free);
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +02008149 Py_END_ALLOW_THREADS
8150 if (retval == 0)
8151 return PyErr_SetFromWindowsErr(0);
8152
8153 return Py_BuildValue("(LL)", total.QuadPart, free.QuadPart);
8154}
8155#endif
8156
8157
Fred Drakec9680921999-12-13 16:37:25 +00008158/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
8159 * It maps strings representing configuration variable names to
8160 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00008161 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00008162 * rarely-used constants. There are three separate tables that use
8163 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00008164 *
8165 * This code is always included, even if none of the interfaces that
8166 * need it are included. The #if hackery needed to avoid it would be
8167 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00008168 */
8169struct constdef {
8170 char *name;
8171 long value;
8172};
8173
Fred Drake12c6e2d1999-12-14 21:25:03 +00008174static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008175conv_confname(PyObject *arg, int *valuep, struct constdef *table,
Guido van Rossum7d5baac2007-08-27 23:24:46 +00008176 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00008177{
Christian Heimes217cfd12007-12-02 14:31:20 +00008178 if (PyLong_Check(arg)) {
Stefan Krah0e803b32010-11-26 16:16:47 +00008179 *valuep = PyLong_AS_LONG(arg);
8180 return 1;
Fred Drake12c6e2d1999-12-14 21:25:03 +00008181 }
Guido van Rossumbce56a62007-05-10 18:04:33 +00008182 else {
Stefan Krah0e803b32010-11-26 16:16:47 +00008183 /* look up the value in the table using a binary search */
8184 size_t lo = 0;
8185 size_t mid;
8186 size_t hi = tablesize;
8187 int cmp;
8188 const char *confname;
8189 if (!PyUnicode_Check(arg)) {
8190 PyErr_SetString(PyExc_TypeError,
8191 "configuration names must be strings or integers");
8192 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00008193 }
Stefan Krah0e803b32010-11-26 16:16:47 +00008194 confname = _PyUnicode_AsString(arg);
8195 if (confname == NULL)
8196 return 0;
8197 while (lo < hi) {
8198 mid = (lo + hi) / 2;
8199 cmp = strcmp(confname, table[mid].name);
8200 if (cmp < 0)
8201 hi = mid;
8202 else if (cmp > 0)
8203 lo = mid + 1;
8204 else {
8205 *valuep = table[mid].value;
8206 return 1;
8207 }
8208 }
8209 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
8210 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00008211 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00008212}
8213
8214
8215#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
8216static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00008217#ifdef _PC_ABI_AIO_XFER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008218 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
Fred Draked86ed291999-12-15 15:34:33 +00008219#endif
8220#ifdef _PC_ABI_ASYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008221 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
Fred Draked86ed291999-12-15 15:34:33 +00008222#endif
Fred Drakec9680921999-12-13 16:37:25 +00008223#ifdef _PC_ASYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008224 {"PC_ASYNC_IO", _PC_ASYNC_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008225#endif
8226#ifdef _PC_CHOWN_RESTRICTED
Victor Stinner8c62be82010-05-06 00:08:46 +00008227 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
Fred Drakec9680921999-12-13 16:37:25 +00008228#endif
8229#ifdef _PC_FILESIZEBITS
Victor Stinner8c62be82010-05-06 00:08:46 +00008230 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
Fred Drakec9680921999-12-13 16:37:25 +00008231#endif
8232#ifdef _PC_LAST
Victor Stinner8c62be82010-05-06 00:08:46 +00008233 {"PC_LAST", _PC_LAST},
Fred Drakec9680921999-12-13 16:37:25 +00008234#endif
8235#ifdef _PC_LINK_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008236 {"PC_LINK_MAX", _PC_LINK_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008237#endif
8238#ifdef _PC_MAX_CANON
Victor Stinner8c62be82010-05-06 00:08:46 +00008239 {"PC_MAX_CANON", _PC_MAX_CANON},
Fred Drakec9680921999-12-13 16:37:25 +00008240#endif
8241#ifdef _PC_MAX_INPUT
Victor Stinner8c62be82010-05-06 00:08:46 +00008242 {"PC_MAX_INPUT", _PC_MAX_INPUT},
Fred Drakec9680921999-12-13 16:37:25 +00008243#endif
8244#ifdef _PC_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008245 {"PC_NAME_MAX", _PC_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008246#endif
8247#ifdef _PC_NO_TRUNC
Victor Stinner8c62be82010-05-06 00:08:46 +00008248 {"PC_NO_TRUNC", _PC_NO_TRUNC},
Fred Drakec9680921999-12-13 16:37:25 +00008249#endif
8250#ifdef _PC_PATH_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008251 {"PC_PATH_MAX", _PC_PATH_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008252#endif
8253#ifdef _PC_PIPE_BUF
Victor Stinner8c62be82010-05-06 00:08:46 +00008254 {"PC_PIPE_BUF", _PC_PIPE_BUF},
Fred Drakec9680921999-12-13 16:37:25 +00008255#endif
8256#ifdef _PC_PRIO_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008257 {"PC_PRIO_IO", _PC_PRIO_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008258#endif
8259#ifdef _PC_SOCK_MAXBUF
Victor Stinner8c62be82010-05-06 00:08:46 +00008260 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
Fred Drakec9680921999-12-13 16:37:25 +00008261#endif
8262#ifdef _PC_SYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008263 {"PC_SYNC_IO", _PC_SYNC_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008264#endif
8265#ifdef _PC_VDISABLE
Victor Stinner8c62be82010-05-06 00:08:46 +00008266 {"PC_VDISABLE", _PC_VDISABLE},
Fred Drakec9680921999-12-13 16:37:25 +00008267#endif
Jesus Cea7e9065c2010-10-25 13:02:04 +00008268#ifdef _PC_ACL_ENABLED
8269 {"PC_ACL_ENABLED", _PC_ACL_ENABLED},
8270#endif
8271#ifdef _PC_MIN_HOLE_SIZE
8272 {"PC_MIN_HOLE_SIZE", _PC_MIN_HOLE_SIZE},
8273#endif
8274#ifdef _PC_ALLOC_SIZE_MIN
8275 {"PC_ALLOC_SIZE_MIN", _PC_ALLOC_SIZE_MIN},
8276#endif
8277#ifdef _PC_REC_INCR_XFER_SIZE
8278 {"PC_REC_INCR_XFER_SIZE", _PC_REC_INCR_XFER_SIZE},
8279#endif
8280#ifdef _PC_REC_MAX_XFER_SIZE
8281 {"PC_REC_MAX_XFER_SIZE", _PC_REC_MAX_XFER_SIZE},
8282#endif
8283#ifdef _PC_REC_MIN_XFER_SIZE
8284 {"PC_REC_MIN_XFER_SIZE", _PC_REC_MIN_XFER_SIZE},
8285#endif
8286#ifdef _PC_REC_XFER_ALIGN
8287 {"PC_REC_XFER_ALIGN", _PC_REC_XFER_ALIGN},
8288#endif
8289#ifdef _PC_SYMLINK_MAX
8290 {"PC_SYMLINK_MAX", _PC_SYMLINK_MAX},
8291#endif
8292#ifdef _PC_XATTR_ENABLED
8293 {"PC_XATTR_ENABLED", _PC_XATTR_ENABLED},
8294#endif
8295#ifdef _PC_XATTR_EXISTS
8296 {"PC_XATTR_EXISTS", _PC_XATTR_EXISTS},
8297#endif
8298#ifdef _PC_TIMESTAMP_RESOLUTION
8299 {"PC_TIMESTAMP_RESOLUTION", _PC_TIMESTAMP_RESOLUTION},
8300#endif
Fred Drakec9680921999-12-13 16:37:25 +00008301};
8302
Fred Drakec9680921999-12-13 16:37:25 +00008303static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008304conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00008305{
8306 return conv_confname(arg, valuep, posix_constants_pathconf,
8307 sizeof(posix_constants_pathconf)
8308 / sizeof(struct constdef));
8309}
8310#endif
8311
8312#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008313PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008314"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00008315Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008316If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00008317
8318static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008319posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008320{
8321 PyObject *result = NULL;
8322 int name, fd;
8323
Fred Drake12c6e2d1999-12-14 21:25:03 +00008324 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
8325 conv_path_confname, &name)) {
Stefan Krah0e803b32010-11-26 16:16:47 +00008326 long limit;
Fred Drakec9680921999-12-13 16:37:25 +00008327
Stefan Krah0e803b32010-11-26 16:16:47 +00008328 errno = 0;
8329 limit = fpathconf(fd, name);
8330 if (limit == -1 && errno != 0)
8331 posix_error();
8332 else
8333 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00008334 }
8335 return result;
8336}
8337#endif
8338
8339
8340#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008341PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008342"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00008343Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008344If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00008345
8346static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008347posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008348{
8349 PyObject *result = NULL;
8350 int name;
8351 char *path;
8352
8353 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
8354 conv_path_confname, &name)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00008355 long limit;
Fred Drakec9680921999-12-13 16:37:25 +00008356
Victor Stinner8c62be82010-05-06 00:08:46 +00008357 errno = 0;
8358 limit = pathconf(path, name);
8359 if (limit == -1 && errno != 0) {
8360 if (errno == EINVAL)
Stefan Krah99439262010-11-26 12:58:05 +00008361 /* could be a path or name problem */
8362 posix_error();
Fred Drakec9680921999-12-13 16:37:25 +00008363 else
Stefan Krah99439262010-11-26 12:58:05 +00008364 posix_error_with_filename(path);
Victor Stinner8c62be82010-05-06 00:08:46 +00008365 }
8366 else
8367 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00008368 }
8369 return result;
8370}
8371#endif
8372
8373#ifdef HAVE_CONFSTR
8374static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00008375#ifdef _CS_ARCHITECTURE
Victor Stinner8c62be82010-05-06 00:08:46 +00008376 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
Fred Draked86ed291999-12-15 15:34:33 +00008377#endif
Mark Dickinson876d7c82010-04-16 12:47:52 +00008378#ifdef _CS_GNU_LIBC_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008379 {"CS_GNU_LIBC_VERSION", _CS_GNU_LIBC_VERSION},
Mark Dickinson876d7c82010-04-16 12:47:52 +00008380#endif
8381#ifdef _CS_GNU_LIBPTHREAD_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008382 {"CS_GNU_LIBPTHREAD_VERSION", _CS_GNU_LIBPTHREAD_VERSION},
Mark Dickinson876d7c82010-04-16 12:47:52 +00008383#endif
Fred Draked86ed291999-12-15 15:34:33 +00008384#ifdef _CS_HOSTNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008385 {"CS_HOSTNAME", _CS_HOSTNAME},
Fred Draked86ed291999-12-15 15:34:33 +00008386#endif
8387#ifdef _CS_HW_PROVIDER
Victor Stinner8c62be82010-05-06 00:08:46 +00008388 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
Fred Draked86ed291999-12-15 15:34:33 +00008389#endif
8390#ifdef _CS_HW_SERIAL
Victor Stinner8c62be82010-05-06 00:08:46 +00008391 {"CS_HW_SERIAL", _CS_HW_SERIAL},
Fred Draked86ed291999-12-15 15:34:33 +00008392#endif
8393#ifdef _CS_INITTAB_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008394 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00008395#endif
Fred Drakec9680921999-12-13 16:37:25 +00008396#ifdef _CS_LFS64_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008397 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008398#endif
8399#ifdef _CS_LFS64_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008400 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008401#endif
8402#ifdef _CS_LFS64_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008403 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008404#endif
8405#ifdef _CS_LFS64_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008406 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008407#endif
8408#ifdef _CS_LFS_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008409 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008410#endif
8411#ifdef _CS_LFS_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008412 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008413#endif
8414#ifdef _CS_LFS_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008415 {"CS_LFS_LIBS", _CS_LFS_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008416#endif
8417#ifdef _CS_LFS_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008418 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008419#endif
Fred Draked86ed291999-12-15 15:34:33 +00008420#ifdef _CS_MACHINE
Victor Stinner8c62be82010-05-06 00:08:46 +00008421 {"CS_MACHINE", _CS_MACHINE},
Fred Draked86ed291999-12-15 15:34:33 +00008422#endif
Fred Drakec9680921999-12-13 16:37:25 +00008423#ifdef _CS_PATH
Victor Stinner8c62be82010-05-06 00:08:46 +00008424 {"CS_PATH", _CS_PATH},
Fred Drakec9680921999-12-13 16:37:25 +00008425#endif
Fred Draked86ed291999-12-15 15:34:33 +00008426#ifdef _CS_RELEASE
Victor Stinner8c62be82010-05-06 00:08:46 +00008427 {"CS_RELEASE", _CS_RELEASE},
Fred Draked86ed291999-12-15 15:34:33 +00008428#endif
8429#ifdef _CS_SRPC_DOMAIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008430 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
Fred Draked86ed291999-12-15 15:34:33 +00008431#endif
8432#ifdef _CS_SYSNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008433 {"CS_SYSNAME", _CS_SYSNAME},
Fred Draked86ed291999-12-15 15:34:33 +00008434#endif
8435#ifdef _CS_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008436 {"CS_VERSION", _CS_VERSION},
Fred Draked86ed291999-12-15 15:34:33 +00008437#endif
Fred Drakec9680921999-12-13 16:37:25 +00008438#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008439 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008440#endif
8441#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008442 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008443#endif
8444#ifdef _CS_XBS5_ILP32_OFF32_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008445 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008446#endif
8447#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008448 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008449#endif
8450#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008451 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008452#endif
8453#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008454 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008455#endif
8456#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008457 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008458#endif
8459#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008460 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008461#endif
8462#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008463 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008464#endif
8465#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008466 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008467#endif
8468#ifdef _CS_XBS5_LP64_OFF64_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008469 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008470#endif
8471#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008472 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008473#endif
8474#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008475 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008476#endif
8477#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008478 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008479#endif
8480#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00008481 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00008482#endif
8483#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00008484 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00008485#endif
Fred Draked86ed291999-12-15 15:34:33 +00008486#ifdef _MIPS_CS_AVAIL_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00008487 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00008488#endif
8489#ifdef _MIPS_CS_BASE
Victor Stinner8c62be82010-05-06 00:08:46 +00008490 {"MIPS_CS_BASE", _MIPS_CS_BASE},
Fred Draked86ed291999-12-15 15:34:33 +00008491#endif
8492#ifdef _MIPS_CS_HOSTID
Victor Stinner8c62be82010-05-06 00:08:46 +00008493 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
Fred Draked86ed291999-12-15 15:34:33 +00008494#endif
8495#ifdef _MIPS_CS_HW_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008496 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00008497#endif
8498#ifdef _MIPS_CS_NUM_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00008499 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00008500#endif
8501#ifdef _MIPS_CS_OSREL_MAJ
Victor Stinner8c62be82010-05-06 00:08:46 +00008502 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
Fred Draked86ed291999-12-15 15:34:33 +00008503#endif
8504#ifdef _MIPS_CS_OSREL_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008505 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
Fred Draked86ed291999-12-15 15:34:33 +00008506#endif
8507#ifdef _MIPS_CS_OSREL_PATCH
Victor Stinner8c62be82010-05-06 00:08:46 +00008508 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
Fred Draked86ed291999-12-15 15:34:33 +00008509#endif
8510#ifdef _MIPS_CS_OS_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00008511 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00008512#endif
8513#ifdef _MIPS_CS_OS_PROVIDER
Victor Stinner8c62be82010-05-06 00:08:46 +00008514 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
Fred Draked86ed291999-12-15 15:34:33 +00008515#endif
8516#ifdef _MIPS_CS_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00008517 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00008518#endif
8519#ifdef _MIPS_CS_SERIAL
Victor Stinner8c62be82010-05-06 00:08:46 +00008520 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
Fred Draked86ed291999-12-15 15:34:33 +00008521#endif
8522#ifdef _MIPS_CS_VENDOR
Victor Stinner8c62be82010-05-06 00:08:46 +00008523 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
Fred Draked86ed291999-12-15 15:34:33 +00008524#endif
Fred Drakec9680921999-12-13 16:37:25 +00008525};
8526
8527static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008528conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00008529{
8530 return conv_confname(arg, valuep, posix_constants_confstr,
8531 sizeof(posix_constants_confstr)
8532 / sizeof(struct constdef));
8533}
8534
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008535PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008536"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008537Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00008538
8539static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008540posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008541{
8542 PyObject *result = NULL;
8543 int name;
Victor Stinnercb043522010-09-10 23:49:04 +00008544 char buffer[255];
Stefan Krah99439262010-11-26 12:58:05 +00008545 int len;
Fred Drakec9680921999-12-13 16:37:25 +00008546
Victor Stinnercb043522010-09-10 23:49:04 +00008547 if (!PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name))
8548 return NULL;
8549
8550 errno = 0;
8551 len = confstr(name, buffer, sizeof(buffer));
8552 if (len == 0) {
8553 if (errno) {
8554 posix_error();
8555 return NULL;
Fred Drakec9680921999-12-13 16:37:25 +00008556 }
8557 else {
Victor Stinnercb043522010-09-10 23:49:04 +00008558 Py_RETURN_NONE;
Fred Drakec9680921999-12-13 16:37:25 +00008559 }
8560 }
Victor Stinnercb043522010-09-10 23:49:04 +00008561
8562 if ((unsigned int)len >= sizeof(buffer)) {
8563 char *buf = PyMem_Malloc(len);
8564 if (buf == NULL)
8565 return PyErr_NoMemory();
8566 confstr(name, buf, len);
8567 result = PyUnicode_DecodeFSDefaultAndSize(buf, len-1);
8568 PyMem_Free(buf);
8569 }
8570 else
8571 result = PyUnicode_DecodeFSDefaultAndSize(buffer, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00008572 return result;
8573}
8574#endif
8575
8576
8577#ifdef HAVE_SYSCONF
8578static struct constdef posix_constants_sysconf[] = {
8579#ifdef _SC_2_CHAR_TERM
Victor Stinner8c62be82010-05-06 00:08:46 +00008580 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
Fred Drakec9680921999-12-13 16:37:25 +00008581#endif
8582#ifdef _SC_2_C_BIND
Victor Stinner8c62be82010-05-06 00:08:46 +00008583 {"SC_2_C_BIND", _SC_2_C_BIND},
Fred Drakec9680921999-12-13 16:37:25 +00008584#endif
8585#ifdef _SC_2_C_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00008586 {"SC_2_C_DEV", _SC_2_C_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00008587#endif
8588#ifdef _SC_2_C_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008589 {"SC_2_C_VERSION", _SC_2_C_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00008590#endif
8591#ifdef _SC_2_FORT_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00008592 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00008593#endif
8594#ifdef _SC_2_FORT_RUN
Victor Stinner8c62be82010-05-06 00:08:46 +00008595 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
Fred Drakec9680921999-12-13 16:37:25 +00008596#endif
8597#ifdef _SC_2_LOCALEDEF
Victor Stinner8c62be82010-05-06 00:08:46 +00008598 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
Fred Drakec9680921999-12-13 16:37:25 +00008599#endif
8600#ifdef _SC_2_SW_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00008601 {"SC_2_SW_DEV", _SC_2_SW_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00008602#endif
8603#ifdef _SC_2_UPE
Victor Stinner8c62be82010-05-06 00:08:46 +00008604 {"SC_2_UPE", _SC_2_UPE},
Fred Drakec9680921999-12-13 16:37:25 +00008605#endif
8606#ifdef _SC_2_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008607 {"SC_2_VERSION", _SC_2_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00008608#endif
Fred Draked86ed291999-12-15 15:34:33 +00008609#ifdef _SC_ABI_ASYNCHRONOUS_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008610 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
Fred Draked86ed291999-12-15 15:34:33 +00008611#endif
8612#ifdef _SC_ACL
Victor Stinner8c62be82010-05-06 00:08:46 +00008613 {"SC_ACL", _SC_ACL},
Fred Draked86ed291999-12-15 15:34:33 +00008614#endif
Fred Drakec9680921999-12-13 16:37:25 +00008615#ifdef _SC_AIO_LISTIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008616 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008617#endif
Fred Drakec9680921999-12-13 16:37:25 +00008618#ifdef _SC_AIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008619 {"SC_AIO_MAX", _SC_AIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008620#endif
8621#ifdef _SC_AIO_PRIO_DELTA_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008622 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008623#endif
8624#ifdef _SC_ARG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008625 {"SC_ARG_MAX", _SC_ARG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008626#endif
8627#ifdef _SC_ASYNCHRONOUS_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008628 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008629#endif
8630#ifdef _SC_ATEXIT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008631 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008632#endif
Fred Draked86ed291999-12-15 15:34:33 +00008633#ifdef _SC_AUDIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008634 {"SC_AUDIT", _SC_AUDIT},
Fred Draked86ed291999-12-15 15:34:33 +00008635#endif
Fred Drakec9680921999-12-13 16:37:25 +00008636#ifdef _SC_AVPHYS_PAGES
Victor Stinner8c62be82010-05-06 00:08:46 +00008637 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
Fred Drakec9680921999-12-13 16:37:25 +00008638#endif
8639#ifdef _SC_BC_BASE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008640 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008641#endif
8642#ifdef _SC_BC_DIM_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008643 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008644#endif
8645#ifdef _SC_BC_SCALE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008646 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008647#endif
8648#ifdef _SC_BC_STRING_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008649 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008650#endif
Fred Draked86ed291999-12-15 15:34:33 +00008651#ifdef _SC_CAP
Victor Stinner8c62be82010-05-06 00:08:46 +00008652 {"SC_CAP", _SC_CAP},
Fred Draked86ed291999-12-15 15:34:33 +00008653#endif
Fred Drakec9680921999-12-13 16:37:25 +00008654#ifdef _SC_CHARCLASS_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008655 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008656#endif
8657#ifdef _SC_CHAR_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008658 {"SC_CHAR_BIT", _SC_CHAR_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00008659#endif
8660#ifdef _SC_CHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008661 {"SC_CHAR_MAX", _SC_CHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008662#endif
8663#ifdef _SC_CHAR_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008664 {"SC_CHAR_MIN", _SC_CHAR_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008665#endif
8666#ifdef _SC_CHILD_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008667 {"SC_CHILD_MAX", _SC_CHILD_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008668#endif
8669#ifdef _SC_CLK_TCK
Victor Stinner8c62be82010-05-06 00:08:46 +00008670 {"SC_CLK_TCK", _SC_CLK_TCK},
Fred Drakec9680921999-12-13 16:37:25 +00008671#endif
8672#ifdef _SC_COHER_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008673 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008674#endif
8675#ifdef _SC_COLL_WEIGHTS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008676 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008677#endif
8678#ifdef _SC_DCACHE_ASSOC
Victor Stinner8c62be82010-05-06 00:08:46 +00008679 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
Fred Drakec9680921999-12-13 16:37:25 +00008680#endif
8681#ifdef _SC_DCACHE_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008682 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008683#endif
8684#ifdef _SC_DCACHE_LINESZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008685 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
Fred Drakec9680921999-12-13 16:37:25 +00008686#endif
8687#ifdef _SC_DCACHE_SZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008688 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
Fred Drakec9680921999-12-13 16:37:25 +00008689#endif
8690#ifdef _SC_DCACHE_TBLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008691 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008692#endif
8693#ifdef _SC_DELAYTIMER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008694 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008695#endif
8696#ifdef _SC_EQUIV_CLASS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008697 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008698#endif
8699#ifdef _SC_EXPR_NEST_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008700 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008701#endif
8702#ifdef _SC_FSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00008703 {"SC_FSYNC", _SC_FSYNC},
Fred Drakec9680921999-12-13 16:37:25 +00008704#endif
8705#ifdef _SC_GETGR_R_SIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008706 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008707#endif
8708#ifdef _SC_GETPW_R_SIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008709 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008710#endif
8711#ifdef _SC_ICACHE_ASSOC
Victor Stinner8c62be82010-05-06 00:08:46 +00008712 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
Fred Drakec9680921999-12-13 16:37:25 +00008713#endif
8714#ifdef _SC_ICACHE_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008715 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00008716#endif
8717#ifdef _SC_ICACHE_LINESZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008718 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
Fred Drakec9680921999-12-13 16:37:25 +00008719#endif
8720#ifdef _SC_ICACHE_SZ
Victor Stinner8c62be82010-05-06 00:08:46 +00008721 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
Fred Drakec9680921999-12-13 16:37:25 +00008722#endif
Fred Draked86ed291999-12-15 15:34:33 +00008723#ifdef _SC_INF
Victor Stinner8c62be82010-05-06 00:08:46 +00008724 {"SC_INF", _SC_INF},
Fred Draked86ed291999-12-15 15:34:33 +00008725#endif
Fred Drakec9680921999-12-13 16:37:25 +00008726#ifdef _SC_INT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008727 {"SC_INT_MAX", _SC_INT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008728#endif
8729#ifdef _SC_INT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008730 {"SC_INT_MIN", _SC_INT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008731#endif
8732#ifdef _SC_IOV_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008733 {"SC_IOV_MAX", _SC_IOV_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008734#endif
Fred Draked86ed291999-12-15 15:34:33 +00008735#ifdef _SC_IP_SECOPTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008736 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
Fred Draked86ed291999-12-15 15:34:33 +00008737#endif
Fred Drakec9680921999-12-13 16:37:25 +00008738#ifdef _SC_JOB_CONTROL
Victor Stinner8c62be82010-05-06 00:08:46 +00008739 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
Fred Drakec9680921999-12-13 16:37:25 +00008740#endif
Fred Draked86ed291999-12-15 15:34:33 +00008741#ifdef _SC_KERN_POINTERS
Victor Stinner8c62be82010-05-06 00:08:46 +00008742 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
Fred Draked86ed291999-12-15 15:34:33 +00008743#endif
8744#ifdef _SC_KERN_SIM
Victor Stinner8c62be82010-05-06 00:08:46 +00008745 {"SC_KERN_SIM", _SC_KERN_SIM},
Fred Draked86ed291999-12-15 15:34:33 +00008746#endif
Fred Drakec9680921999-12-13 16:37:25 +00008747#ifdef _SC_LINE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008748 {"SC_LINE_MAX", _SC_LINE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008749#endif
8750#ifdef _SC_LOGIN_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008751 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008752#endif
8753#ifdef _SC_LOGNAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008754 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008755#endif
8756#ifdef _SC_LONG_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008757 {"SC_LONG_BIT", _SC_LONG_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00008758#endif
Fred Draked86ed291999-12-15 15:34:33 +00008759#ifdef _SC_MAC
Victor Stinner8c62be82010-05-06 00:08:46 +00008760 {"SC_MAC", _SC_MAC},
Fred Draked86ed291999-12-15 15:34:33 +00008761#endif
Fred Drakec9680921999-12-13 16:37:25 +00008762#ifdef _SC_MAPPED_FILES
Victor Stinner8c62be82010-05-06 00:08:46 +00008763 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
Fred Drakec9680921999-12-13 16:37:25 +00008764#endif
8765#ifdef _SC_MAXPID
Victor Stinner8c62be82010-05-06 00:08:46 +00008766 {"SC_MAXPID", _SC_MAXPID},
Fred Drakec9680921999-12-13 16:37:25 +00008767#endif
8768#ifdef _SC_MB_LEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008769 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008770#endif
8771#ifdef _SC_MEMLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +00008772 {"SC_MEMLOCK", _SC_MEMLOCK},
Fred Drakec9680921999-12-13 16:37:25 +00008773#endif
8774#ifdef _SC_MEMLOCK_RANGE
Victor Stinner8c62be82010-05-06 00:08:46 +00008775 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
Fred Drakec9680921999-12-13 16:37:25 +00008776#endif
8777#ifdef _SC_MEMORY_PROTECTION
Victor Stinner8c62be82010-05-06 00:08:46 +00008778 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
Fred Drakec9680921999-12-13 16:37:25 +00008779#endif
8780#ifdef _SC_MESSAGE_PASSING
Victor Stinner8c62be82010-05-06 00:08:46 +00008781 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
Fred Drakec9680921999-12-13 16:37:25 +00008782#endif
Fred Draked86ed291999-12-15 15:34:33 +00008783#ifdef _SC_MMAP_FIXED_ALIGNMENT
Victor Stinner8c62be82010-05-06 00:08:46 +00008784 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
Fred Draked86ed291999-12-15 15:34:33 +00008785#endif
Fred Drakec9680921999-12-13 16:37:25 +00008786#ifdef _SC_MQ_OPEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008787 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008788#endif
8789#ifdef _SC_MQ_PRIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008790 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008791#endif
Fred Draked86ed291999-12-15 15:34:33 +00008792#ifdef _SC_NACLS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008793 {"SC_NACLS_MAX", _SC_NACLS_MAX},
Fred Draked86ed291999-12-15 15:34:33 +00008794#endif
Fred Drakec9680921999-12-13 16:37:25 +00008795#ifdef _SC_NGROUPS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008796 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008797#endif
8798#ifdef _SC_NL_ARGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008799 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008800#endif
8801#ifdef _SC_NL_LANGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008802 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008803#endif
8804#ifdef _SC_NL_MSGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008805 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008806#endif
8807#ifdef _SC_NL_NMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008808 {"SC_NL_NMAX", _SC_NL_NMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008809#endif
8810#ifdef _SC_NL_SETMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008811 {"SC_NL_SETMAX", _SC_NL_SETMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008812#endif
8813#ifdef _SC_NL_TEXTMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008814 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
Fred Drakec9680921999-12-13 16:37:25 +00008815#endif
8816#ifdef _SC_NPROCESSORS_CONF
Victor Stinner8c62be82010-05-06 00:08:46 +00008817 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
Fred Drakec9680921999-12-13 16:37:25 +00008818#endif
8819#ifdef _SC_NPROCESSORS_ONLN
Victor Stinner8c62be82010-05-06 00:08:46 +00008820 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
Fred Drakec9680921999-12-13 16:37:25 +00008821#endif
Fred Draked86ed291999-12-15 15:34:33 +00008822#ifdef _SC_NPROC_CONF
Victor Stinner8c62be82010-05-06 00:08:46 +00008823 {"SC_NPROC_CONF", _SC_NPROC_CONF},
Fred Draked86ed291999-12-15 15:34:33 +00008824#endif
8825#ifdef _SC_NPROC_ONLN
Victor Stinner8c62be82010-05-06 00:08:46 +00008826 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
Fred Draked86ed291999-12-15 15:34:33 +00008827#endif
Fred Drakec9680921999-12-13 16:37:25 +00008828#ifdef _SC_NZERO
Victor Stinner8c62be82010-05-06 00:08:46 +00008829 {"SC_NZERO", _SC_NZERO},
Fred Drakec9680921999-12-13 16:37:25 +00008830#endif
8831#ifdef _SC_OPEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008832 {"SC_OPEN_MAX", _SC_OPEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008833#endif
8834#ifdef _SC_PAGESIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008835 {"SC_PAGESIZE", _SC_PAGESIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008836#endif
8837#ifdef _SC_PAGE_SIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008838 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008839#endif
8840#ifdef _SC_PASS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008841 {"SC_PASS_MAX", _SC_PASS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008842#endif
8843#ifdef _SC_PHYS_PAGES
Victor Stinner8c62be82010-05-06 00:08:46 +00008844 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
Fred Drakec9680921999-12-13 16:37:25 +00008845#endif
8846#ifdef _SC_PII
Victor Stinner8c62be82010-05-06 00:08:46 +00008847 {"SC_PII", _SC_PII},
Fred Drakec9680921999-12-13 16:37:25 +00008848#endif
8849#ifdef _SC_PII_INTERNET
Victor Stinner8c62be82010-05-06 00:08:46 +00008850 {"SC_PII_INTERNET", _SC_PII_INTERNET},
Fred Drakec9680921999-12-13 16:37:25 +00008851#endif
8852#ifdef _SC_PII_INTERNET_DGRAM
Victor Stinner8c62be82010-05-06 00:08:46 +00008853 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
Fred Drakec9680921999-12-13 16:37:25 +00008854#endif
8855#ifdef _SC_PII_INTERNET_STREAM
Victor Stinner8c62be82010-05-06 00:08:46 +00008856 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
Fred Drakec9680921999-12-13 16:37:25 +00008857#endif
8858#ifdef _SC_PII_OSI
Victor Stinner8c62be82010-05-06 00:08:46 +00008859 {"SC_PII_OSI", _SC_PII_OSI},
Fred Drakec9680921999-12-13 16:37:25 +00008860#endif
8861#ifdef _SC_PII_OSI_CLTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008862 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
Fred Drakec9680921999-12-13 16:37:25 +00008863#endif
8864#ifdef _SC_PII_OSI_COTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008865 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
Fred Drakec9680921999-12-13 16:37:25 +00008866#endif
8867#ifdef _SC_PII_OSI_M
Victor Stinner8c62be82010-05-06 00:08:46 +00008868 {"SC_PII_OSI_M", _SC_PII_OSI_M},
Fred Drakec9680921999-12-13 16:37:25 +00008869#endif
8870#ifdef _SC_PII_SOCKET
Victor Stinner8c62be82010-05-06 00:08:46 +00008871 {"SC_PII_SOCKET", _SC_PII_SOCKET},
Fred Drakec9680921999-12-13 16:37:25 +00008872#endif
8873#ifdef _SC_PII_XTI
Victor Stinner8c62be82010-05-06 00:08:46 +00008874 {"SC_PII_XTI", _SC_PII_XTI},
Fred Drakec9680921999-12-13 16:37:25 +00008875#endif
8876#ifdef _SC_POLL
Victor Stinner8c62be82010-05-06 00:08:46 +00008877 {"SC_POLL", _SC_POLL},
Fred Drakec9680921999-12-13 16:37:25 +00008878#endif
8879#ifdef _SC_PRIORITIZED_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008880 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008881#endif
8882#ifdef _SC_PRIORITY_SCHEDULING
Victor Stinner8c62be82010-05-06 00:08:46 +00008883 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
Fred Drakec9680921999-12-13 16:37:25 +00008884#endif
8885#ifdef _SC_REALTIME_SIGNALS
Victor Stinner8c62be82010-05-06 00:08:46 +00008886 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
Fred Drakec9680921999-12-13 16:37:25 +00008887#endif
8888#ifdef _SC_RE_DUP_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008889 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008890#endif
8891#ifdef _SC_RTSIG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008892 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008893#endif
8894#ifdef _SC_SAVED_IDS
Victor Stinner8c62be82010-05-06 00:08:46 +00008895 {"SC_SAVED_IDS", _SC_SAVED_IDS},
Fred Drakec9680921999-12-13 16:37:25 +00008896#endif
8897#ifdef _SC_SCHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008898 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008899#endif
8900#ifdef _SC_SCHAR_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008901 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008902#endif
8903#ifdef _SC_SELECT
Victor Stinner8c62be82010-05-06 00:08:46 +00008904 {"SC_SELECT", _SC_SELECT},
Fred Drakec9680921999-12-13 16:37:25 +00008905#endif
8906#ifdef _SC_SEMAPHORES
Victor Stinner8c62be82010-05-06 00:08:46 +00008907 {"SC_SEMAPHORES", _SC_SEMAPHORES},
Fred Drakec9680921999-12-13 16:37:25 +00008908#endif
8909#ifdef _SC_SEM_NSEMS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008910 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008911#endif
8912#ifdef _SC_SEM_VALUE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008913 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008914#endif
8915#ifdef _SC_SHARED_MEMORY_OBJECTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008916 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
Fred Drakec9680921999-12-13 16:37:25 +00008917#endif
8918#ifdef _SC_SHRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008919 {"SC_SHRT_MAX", _SC_SHRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008920#endif
8921#ifdef _SC_SHRT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008922 {"SC_SHRT_MIN", _SC_SHRT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008923#endif
8924#ifdef _SC_SIGQUEUE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008925 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008926#endif
8927#ifdef _SC_SIGRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008928 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008929#endif
8930#ifdef _SC_SIGRT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008931 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008932#endif
Fred Draked86ed291999-12-15 15:34:33 +00008933#ifdef _SC_SOFTPOWER
Victor Stinner8c62be82010-05-06 00:08:46 +00008934 {"SC_SOFTPOWER", _SC_SOFTPOWER},
Fred Draked86ed291999-12-15 15:34:33 +00008935#endif
Fred Drakec9680921999-12-13 16:37:25 +00008936#ifdef _SC_SPLIT_CACHE
Victor Stinner8c62be82010-05-06 00:08:46 +00008937 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
Fred Drakec9680921999-12-13 16:37:25 +00008938#endif
8939#ifdef _SC_SSIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008940 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008941#endif
8942#ifdef _SC_STACK_PROT
Victor Stinner8c62be82010-05-06 00:08:46 +00008943 {"SC_STACK_PROT", _SC_STACK_PROT},
Fred Drakec9680921999-12-13 16:37:25 +00008944#endif
8945#ifdef _SC_STREAM_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008946 {"SC_STREAM_MAX", _SC_STREAM_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008947#endif
8948#ifdef _SC_SYNCHRONIZED_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008949 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008950#endif
8951#ifdef _SC_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00008952 {"SC_THREADS", _SC_THREADS},
Fred Drakec9680921999-12-13 16:37:25 +00008953#endif
8954#ifdef _SC_THREAD_ATTR_STACKADDR
Victor Stinner8c62be82010-05-06 00:08:46 +00008955 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
Fred Drakec9680921999-12-13 16:37:25 +00008956#endif
8957#ifdef _SC_THREAD_ATTR_STACKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008958 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008959#endif
8960#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
Victor Stinner8c62be82010-05-06 00:08:46 +00008961 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
Fred Drakec9680921999-12-13 16:37:25 +00008962#endif
8963#ifdef _SC_THREAD_KEYS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008964 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008965#endif
8966#ifdef _SC_THREAD_PRIORITY_SCHEDULING
Victor Stinner8c62be82010-05-06 00:08:46 +00008967 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
Fred Drakec9680921999-12-13 16:37:25 +00008968#endif
8969#ifdef _SC_THREAD_PRIO_INHERIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008970 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
Fred Drakec9680921999-12-13 16:37:25 +00008971#endif
8972#ifdef _SC_THREAD_PRIO_PROTECT
Victor Stinner8c62be82010-05-06 00:08:46 +00008973 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
Fred Drakec9680921999-12-13 16:37:25 +00008974#endif
8975#ifdef _SC_THREAD_PROCESS_SHARED
Victor Stinner8c62be82010-05-06 00:08:46 +00008976 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
Fred Drakec9680921999-12-13 16:37:25 +00008977#endif
8978#ifdef _SC_THREAD_SAFE_FUNCTIONS
Victor Stinner8c62be82010-05-06 00:08:46 +00008979 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
Fred Drakec9680921999-12-13 16:37:25 +00008980#endif
8981#ifdef _SC_THREAD_STACK_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008982 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008983#endif
8984#ifdef _SC_THREAD_THREADS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008985 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008986#endif
8987#ifdef _SC_TIMERS
Victor Stinner8c62be82010-05-06 00:08:46 +00008988 {"SC_TIMERS", _SC_TIMERS},
Fred Drakec9680921999-12-13 16:37:25 +00008989#endif
8990#ifdef _SC_TIMER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008991 {"SC_TIMER_MAX", _SC_TIMER_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008992#endif
8993#ifdef _SC_TTY_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008994 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008995#endif
8996#ifdef _SC_TZNAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008997 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008998#endif
8999#ifdef _SC_T_IOV_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009000 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009001#endif
9002#ifdef _SC_UCHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009003 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009004#endif
9005#ifdef _SC_UINT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009006 {"SC_UINT_MAX", _SC_UINT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009007#endif
9008#ifdef _SC_UIO_MAXIOV
Victor Stinner8c62be82010-05-06 00:08:46 +00009009 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
Fred Drakec9680921999-12-13 16:37:25 +00009010#endif
9011#ifdef _SC_ULONG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009012 {"SC_ULONG_MAX", _SC_ULONG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009013#endif
9014#ifdef _SC_USHRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009015 {"SC_USHRT_MAX", _SC_USHRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00009016#endif
9017#ifdef _SC_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00009018 {"SC_VERSION", _SC_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00009019#endif
9020#ifdef _SC_WORD_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00009021 {"SC_WORD_BIT", _SC_WORD_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00009022#endif
9023#ifdef _SC_XBS5_ILP32_OFF32
Victor Stinner8c62be82010-05-06 00:08:46 +00009024 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
Fred Drakec9680921999-12-13 16:37:25 +00009025#endif
9026#ifdef _SC_XBS5_ILP32_OFFBIG
Victor Stinner8c62be82010-05-06 00:08:46 +00009027 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
Fred Drakec9680921999-12-13 16:37:25 +00009028#endif
9029#ifdef _SC_XBS5_LP64_OFF64
Victor Stinner8c62be82010-05-06 00:08:46 +00009030 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
Fred Drakec9680921999-12-13 16:37:25 +00009031#endif
9032#ifdef _SC_XBS5_LPBIG_OFFBIG
Victor Stinner8c62be82010-05-06 00:08:46 +00009033 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
Fred Drakec9680921999-12-13 16:37:25 +00009034#endif
9035#ifdef _SC_XOPEN_CRYPT
Victor Stinner8c62be82010-05-06 00:08:46 +00009036 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
Fred Drakec9680921999-12-13 16:37:25 +00009037#endif
9038#ifdef _SC_XOPEN_ENH_I18N
Victor Stinner8c62be82010-05-06 00:08:46 +00009039 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
Fred Drakec9680921999-12-13 16:37:25 +00009040#endif
9041#ifdef _SC_XOPEN_LEGACY
Victor Stinner8c62be82010-05-06 00:08:46 +00009042 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
Fred Drakec9680921999-12-13 16:37:25 +00009043#endif
9044#ifdef _SC_XOPEN_REALTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00009045 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
Fred Drakec9680921999-12-13 16:37:25 +00009046#endif
9047#ifdef _SC_XOPEN_REALTIME_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00009048 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
Fred Drakec9680921999-12-13 16:37:25 +00009049#endif
9050#ifdef _SC_XOPEN_SHM
Victor Stinner8c62be82010-05-06 00:08:46 +00009051 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
Fred Drakec9680921999-12-13 16:37:25 +00009052#endif
9053#ifdef _SC_XOPEN_UNIX
Victor Stinner8c62be82010-05-06 00:08:46 +00009054 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
Fred Drakec9680921999-12-13 16:37:25 +00009055#endif
9056#ifdef _SC_XOPEN_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00009057 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00009058#endif
9059#ifdef _SC_XOPEN_XCU_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00009060 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00009061#endif
9062#ifdef _SC_XOPEN_XPG2
Victor Stinner8c62be82010-05-06 00:08:46 +00009063 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
Fred Drakec9680921999-12-13 16:37:25 +00009064#endif
9065#ifdef _SC_XOPEN_XPG3
Victor Stinner8c62be82010-05-06 00:08:46 +00009066 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
Fred Drakec9680921999-12-13 16:37:25 +00009067#endif
9068#ifdef _SC_XOPEN_XPG4
Victor Stinner8c62be82010-05-06 00:08:46 +00009069 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
Fred Drakec9680921999-12-13 16:37:25 +00009070#endif
9071};
9072
9073static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009074conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00009075{
9076 return conv_confname(arg, valuep, posix_constants_sysconf,
9077 sizeof(posix_constants_sysconf)
9078 / sizeof(struct constdef));
9079}
9080
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009081PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00009082"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009083Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00009084
9085static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009086posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00009087{
9088 PyObject *result = NULL;
9089 int name;
9090
9091 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
9092 int value;
9093
9094 errno = 0;
9095 value = sysconf(name);
9096 if (value == -1 && errno != 0)
9097 posix_error();
9098 else
Christian Heimes217cfd12007-12-02 14:31:20 +00009099 result = PyLong_FromLong(value);
Fred Drakec9680921999-12-13 16:37:25 +00009100 }
9101 return result;
9102}
9103#endif
9104
9105
Fred Drakebec628d1999-12-15 18:31:10 +00009106/* This code is used to ensure that the tables of configuration value names
9107 * are in sorted order as required by conv_confname(), and also to build the
9108 * the exported dictionaries that are used to publish information about the
9109 * names available on the host platform.
9110 *
9111 * Sorting the table at runtime ensures that the table is properly ordered
9112 * when used, even for platforms we're not able to test on. It also makes
9113 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00009114 */
Fred Drakebec628d1999-12-15 18:31:10 +00009115
9116static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009117cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00009118{
9119 const struct constdef *c1 =
Victor Stinner8c62be82010-05-06 00:08:46 +00009120 (const struct constdef *) v1;
Fred Drakebec628d1999-12-15 18:31:10 +00009121 const struct constdef *c2 =
Victor Stinner8c62be82010-05-06 00:08:46 +00009122 (const struct constdef *) v2;
Fred Drakebec628d1999-12-15 18:31:10 +00009123
9124 return strcmp(c1->name, c2->name);
9125}
9126
9127static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00009128setup_confname_table(struct constdef *table, size_t tablesize,
Victor Stinner8c62be82010-05-06 00:08:46 +00009129 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00009130{
Fred Drakebec628d1999-12-15 18:31:10 +00009131 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00009132 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00009133
9134 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
9135 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00009136 if (d == NULL)
Victor Stinner8c62be82010-05-06 00:08:46 +00009137 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009138
Barry Warsaw3155db32000-04-13 15:20:40 +00009139 for (i=0; i < tablesize; ++i) {
Victor Stinner8c62be82010-05-06 00:08:46 +00009140 PyObject *o = PyLong_FromLong(table[i].value);
9141 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
9142 Py_XDECREF(o);
9143 Py_DECREF(d);
9144 return -1;
9145 }
9146 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00009147 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00009148 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00009149}
9150
Fred Drakebec628d1999-12-15 18:31:10 +00009151/* Return -1 on failure, 0 on success. */
9152static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00009153setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00009154{
9155#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00009156 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00009157 sizeof(posix_constants_pathconf)
9158 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00009159 "pathconf_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00009160 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009161#endif
9162#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00009163 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00009164 sizeof(posix_constants_confstr)
9165 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00009166 "confstr_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00009167 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009168#endif
9169#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00009170 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00009171 sizeof(posix_constants_sysconf)
9172 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00009173 "sysconf_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00009174 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00009175#endif
Fred Drakebec628d1999-12-15 18:31:10 +00009176 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00009177}
Fred Draked86ed291999-12-15 15:34:33 +00009178
9179
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009180PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00009181"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009182Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009183in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009184
9185static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00009186posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009187{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009188 abort();
9189 /*NOTREACHED*/
9190 Py_FatalError("abort() called from Python code didn't abort!");
9191 return NULL;
9192}
Fred Drakebec628d1999-12-15 18:31:10 +00009193
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00009194#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009195PyDoc_STRVAR(win32_startfile__doc__,
Georg Brandlf4f44152006-02-18 22:29:33 +00009196"startfile(filepath [, operation]) - Start a file with its associated\n\
9197application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00009198\n\
Georg Brandlf4f44152006-02-18 22:29:33 +00009199When \"operation\" is not specified or \"open\", this acts like\n\
9200double-clicking the file in Explorer, or giving the file name as an\n\
9201argument to the DOS \"start\" command: the file is opened with whatever\n\
9202application (if any) its extension is associated.\n\
9203When another \"operation\" is given, it specifies what should be done with\n\
9204the file. A typical operation is \"print\".\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00009205\n\
9206startfile returns as soon as the associated application is launched.\n\
9207There is no option to wait for the application to close, and no way\n\
9208to retrieve the application's exit status.\n\
9209\n\
9210The filepath is relative to the current directory. If you want to use\n\
9211an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009212the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00009213
9214static PyObject *
9215win32_startfile(PyObject *self, PyObject *args)
9216{
Victor Stinner8c62be82010-05-06 00:08:46 +00009217 PyObject *ofilepath;
9218 char *filepath;
9219 char *operation = NULL;
Victor Stinnereb5657a2011-09-30 01:44:27 +02009220 wchar_t *wpath, *woperation;
Victor Stinner8c62be82010-05-06 00:08:46 +00009221 HINSTANCE rc;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00009222
Victor Stinnereb5657a2011-09-30 01:44:27 +02009223 PyObject *unipath, *uoperation = NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00009224 if (!PyArg_ParseTuple(args, "U|s:startfile",
9225 &unipath, &operation)) {
9226 PyErr_Clear();
9227 goto normal;
9228 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00009229
Victor Stinner8c62be82010-05-06 00:08:46 +00009230 if (operation) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02009231 uoperation = PyUnicode_DecodeASCII(operation,
Victor Stinner8c62be82010-05-06 00:08:46 +00009232 strlen(operation), NULL);
Victor Stinnereb5657a2011-09-30 01:44:27 +02009233 if (!uoperation) {
Victor Stinner8c62be82010-05-06 00:08:46 +00009234 PyErr_Clear();
9235 operation = NULL;
9236 goto normal;
9237 }
9238 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00009239
Victor Stinnereb5657a2011-09-30 01:44:27 +02009240 wpath = PyUnicode_AsUnicode(unipath);
9241 if (wpath == NULL)
9242 goto normal;
9243 if (uoperation) {
9244 woperation = PyUnicode_AsUnicode(uoperation);
9245 if (woperation == NULL)
9246 goto normal;
9247 }
9248 else
9249 woperation = NULL;
9250
Victor Stinner8c62be82010-05-06 00:08:46 +00009251 Py_BEGIN_ALLOW_THREADS
Victor Stinnereb5657a2011-09-30 01:44:27 +02009252 rc = ShellExecuteW((HWND)0, woperation, wpath,
9253 NULL, NULL, SW_SHOWNORMAL);
Victor Stinner8c62be82010-05-06 00:08:46 +00009254 Py_END_ALLOW_THREADS
9255
Victor Stinnereb5657a2011-09-30 01:44:27 +02009256 Py_XDECREF(uoperation);
Victor Stinner8c62be82010-05-06 00:08:46 +00009257 if (rc <= (HINSTANCE)32) {
Victor Stinnereb5657a2011-09-30 01:44:27 +02009258 win32_error_object("startfile", unipath);
9259 return NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00009260 }
9261 Py_INCREF(Py_None);
9262 return Py_None;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009263
9264normal:
Victor Stinner8c62be82010-05-06 00:08:46 +00009265 if (!PyArg_ParseTuple(args, "O&|s:startfile",
9266 PyUnicode_FSConverter, &ofilepath,
9267 &operation))
9268 return NULL;
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01009269 if (win32_warn_bytes_api()) {
9270 Py_DECREF(ofilepath);
9271 return NULL;
9272 }
Victor Stinner8c62be82010-05-06 00:08:46 +00009273 filepath = PyBytes_AsString(ofilepath);
9274 Py_BEGIN_ALLOW_THREADS
9275 rc = ShellExecute((HWND)0, operation, filepath,
9276 NULL, NULL, SW_SHOWNORMAL);
9277 Py_END_ALLOW_THREADS
9278 if (rc <= (HINSTANCE)32) {
9279 PyObject *errval = win32_error("startfile", filepath);
9280 Py_DECREF(ofilepath);
9281 return errval;
9282 }
9283 Py_DECREF(ofilepath);
9284 Py_INCREF(Py_None);
9285 return Py_None;
Tim Petersf58a7aa2000-09-22 10:05:54 +00009286}
9287#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009288
Martin v. Löwis438b5342002-12-27 10:16:42 +00009289#ifdef HAVE_GETLOADAVG
9290PyDoc_STRVAR(posix_getloadavg__doc__,
9291"getloadavg() -> (float, float, float)\n\n\
9292Return the number of processes in the system run queue averaged over\n\
9293the last 1, 5, and 15 minutes or raises OSError if the load average\n\
9294was unobtainable");
9295
9296static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00009297posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00009298{
9299 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00009300 if (getloadavg(loadavg, 3)!=3) {
Stefan Krah0e803b32010-11-26 16:16:47 +00009301 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
9302 return NULL;
Martin v. Löwis438b5342002-12-27 10:16:42 +00009303 } else
Stefan Krah0e803b32010-11-26 16:16:47 +00009304 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
Martin v. Löwis438b5342002-12-27 10:16:42 +00009305}
9306#endif
9307
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009308#ifdef MS_WINDOWS
9309
9310PyDoc_STRVAR(win32_urandom__doc__,
9311"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00009312Return n random bytes suitable for cryptographic use.");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009313
9314typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
9315 LPCSTR pszContainer, LPCSTR pszProvider, DWORD dwProvType,\
9316 DWORD dwFlags );
9317typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV hProv, DWORD dwLen,\
9318 BYTE *pbBuffer );
9319
9320static CRYPTGENRANDOM pCryptGenRandom = NULL;
Thomas Wouters89d996e2007-09-08 17:39:28 +00009321/* This handle is never explicitly released. Instead, the operating
9322 system will release it when the process terminates. */
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009323static HCRYPTPROV hCryptProv = 0;
9324
Tim Peters4ad82172004-08-30 17:02:04 +00009325static PyObject*
9326win32_urandom(PyObject *self, PyObject *args)
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009327{
Victor Stinner8c62be82010-05-06 00:08:46 +00009328 int howMany;
9329 PyObject* result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009330
Victor Stinner8c62be82010-05-06 00:08:46 +00009331 /* Read arguments */
9332 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
9333 return NULL;
9334 if (howMany < 0)
9335 return PyErr_Format(PyExc_ValueError,
9336 "negative argument not allowed");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009337
Victor Stinner8c62be82010-05-06 00:08:46 +00009338 if (hCryptProv == 0) {
9339 HINSTANCE hAdvAPI32 = NULL;
9340 CRYPTACQUIRECONTEXTA pCryptAcquireContext = NULL;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009341
Victor Stinner8c62be82010-05-06 00:08:46 +00009342 /* Obtain handle to the DLL containing CryptoAPI
9343 This should not fail */
Martin v. Löwis50590f12012-01-14 17:54:09 +01009344 hAdvAPI32 = GetModuleHandleW(L"advapi32.dll");
Victor Stinner8c62be82010-05-06 00:08:46 +00009345 if(hAdvAPI32 == NULL)
9346 return win32_error("GetModuleHandle", NULL);
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009347
Victor Stinner8c62be82010-05-06 00:08:46 +00009348 /* Obtain pointers to the CryptoAPI functions
9349 This will fail on some early versions of Win95 */
9350 pCryptAcquireContext = (CRYPTACQUIRECONTEXTA)GetProcAddress(
9351 hAdvAPI32,
9352 "CryptAcquireContextA");
9353 if (pCryptAcquireContext == NULL)
9354 return PyErr_Format(PyExc_NotImplementedError,
9355 "CryptAcquireContextA not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009356
Victor Stinner8c62be82010-05-06 00:08:46 +00009357 pCryptGenRandom = (CRYPTGENRANDOM)GetProcAddress(
9358 hAdvAPI32, "CryptGenRandom");
9359 if (pCryptGenRandom == NULL)
9360 return PyErr_Format(PyExc_NotImplementedError,
9361 "CryptGenRandom not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009362
Victor Stinner8c62be82010-05-06 00:08:46 +00009363 /* Acquire context */
9364 if (! pCryptAcquireContext(&hCryptProv, NULL, NULL,
9365 PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
9366 return win32_error("CryptAcquireContext", NULL);
9367 }
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009368
Victor Stinner8c62be82010-05-06 00:08:46 +00009369 /* Allocate bytes */
9370 result = PyBytes_FromStringAndSize(NULL, howMany);
9371 if (result != NULL) {
9372 /* Get random data */
9373 memset(PyBytes_AS_STRING(result), 0, howMany); /* zero seed */
9374 if (! pCryptGenRandom(hCryptProv, howMany, (unsigned char*)
9375 PyBytes_AS_STRING(result))) {
9376 Py_DECREF(result);
9377 return win32_error("CryptGenRandom", NULL);
9378 }
9379 }
9380 return result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009381}
9382#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00009383
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00009384PyDoc_STRVAR(device_encoding__doc__,
9385"device_encoding(fd) -> str\n\n\
9386Return a string describing the encoding of the device\n\
9387if the output is a terminal; else return None.");
9388
9389static PyObject *
9390device_encoding(PyObject *self, PyObject *args)
9391{
Victor Stinner8c62be82010-05-06 00:08:46 +00009392 int fd;
Victor Stinner7870bdf2011-05-23 18:12:52 +02009393#if defined(MS_WINDOWS) || defined(MS_WIN64)
9394 UINT cp;
9395#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009396 if (!PyArg_ParseTuple(args, "i:device_encoding", &fd))
9397 return NULL;
9398 if (!_PyVerify_fd(fd) || !isatty(fd)) {
9399 Py_INCREF(Py_None);
9400 return Py_None;
9401 }
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00009402#if defined(MS_WINDOWS) || defined(MS_WIN64)
Victor Stinner7870bdf2011-05-23 18:12:52 +02009403 if (fd == 0)
9404 cp = GetConsoleCP();
9405 else if (fd == 1 || fd == 2)
9406 cp = GetConsoleOutputCP();
9407 else
9408 cp = 0;
9409 /* GetConsoleCP() and GetConsoleOutputCP() return 0 if the application
9410 has no console */
9411 if (cp != 0)
9412 return PyUnicode_FromFormat("cp%u", (unsigned int)cp);
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00009413#elif defined(CODESET)
Victor Stinner8c62be82010-05-06 00:08:46 +00009414 {
9415 char *codeset = nl_langinfo(CODESET);
9416 if (codeset != NULL && codeset[0] != 0)
9417 return PyUnicode_FromString(codeset);
9418 }
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00009419#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009420 Py_INCREF(Py_None);
9421 return Py_None;
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00009422}
9423
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009424#ifdef __VMS
9425/* Use openssl random routine */
9426#include <openssl/rand.h>
9427PyDoc_STRVAR(vms_urandom__doc__,
9428"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00009429Return n random bytes suitable for cryptographic use.");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009430
9431static PyObject*
9432vms_urandom(PyObject *self, PyObject *args)
9433{
Victor Stinner8c62be82010-05-06 00:08:46 +00009434 int howMany;
9435 PyObject* result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009436
Victor Stinner8c62be82010-05-06 00:08:46 +00009437 /* Read arguments */
9438 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
9439 return NULL;
9440 if (howMany < 0)
9441 return PyErr_Format(PyExc_ValueError,
9442 "negative argument not allowed");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009443
Victor Stinner8c62be82010-05-06 00:08:46 +00009444 /* Allocate bytes */
9445 result = PyBytes_FromStringAndSize(NULL, howMany);
9446 if (result != NULL) {
9447 /* Get random data */
9448 if (RAND_pseudo_bytes((unsigned char*)
9449 PyBytes_AS_STRING(result),
9450 howMany) < 0) {
9451 Py_DECREF(result);
9452 return PyErr_Format(PyExc_ValueError,
9453 "RAND_pseudo_bytes");
9454 }
9455 }
9456 return result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009457}
9458#endif
9459
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009460#ifdef HAVE_SETRESUID
9461PyDoc_STRVAR(posix_setresuid__doc__,
9462"setresuid(ruid, euid, suid)\n\n\
9463Set the current process's real, effective, and saved user ids.");
9464
9465static PyObject*
9466posix_setresuid (PyObject *self, PyObject *args)
9467{
Victor Stinner8c62be82010-05-06 00:08:46 +00009468 /* We assume uid_t is no larger than a long. */
9469 long ruid, euid, suid;
9470 if (!PyArg_ParseTuple(args, "lll", &ruid, &euid, &suid))
9471 return NULL;
9472 if (setresuid(ruid, euid, suid) < 0)
9473 return posix_error();
9474 Py_RETURN_NONE;
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009475}
9476#endif
9477
9478#ifdef HAVE_SETRESGID
9479PyDoc_STRVAR(posix_setresgid__doc__,
9480"setresgid(rgid, egid, sgid)\n\n\
9481Set the current process's real, effective, and saved group ids.");
9482
9483static PyObject*
9484posix_setresgid (PyObject *self, PyObject *args)
9485{
Victor Stinner8c62be82010-05-06 00:08:46 +00009486 /* We assume uid_t is no larger than a long. */
9487 long rgid, egid, sgid;
9488 if (!PyArg_ParseTuple(args, "lll", &rgid, &egid, &sgid))
9489 return NULL;
9490 if (setresgid(rgid, egid, sgid) < 0)
9491 return posix_error();
9492 Py_RETURN_NONE;
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009493}
9494#endif
9495
9496#ifdef HAVE_GETRESUID
9497PyDoc_STRVAR(posix_getresuid__doc__,
9498"getresuid() -> (ruid, euid, suid)\n\n\
9499Get tuple of the current process's real, effective, and saved user ids.");
9500
9501static PyObject*
9502posix_getresuid (PyObject *self, PyObject *noargs)
9503{
Victor Stinner8c62be82010-05-06 00:08:46 +00009504 uid_t ruid, euid, suid;
9505 long l_ruid, l_euid, l_suid;
9506 if (getresuid(&ruid, &euid, &suid) < 0)
9507 return posix_error();
9508 /* Force the values into long's as we don't know the size of uid_t. */
9509 l_ruid = ruid;
9510 l_euid = euid;
9511 l_suid = suid;
9512 return Py_BuildValue("(lll)", l_ruid, l_euid, l_suid);
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009513}
9514#endif
9515
9516#ifdef HAVE_GETRESGID
9517PyDoc_STRVAR(posix_getresgid__doc__,
9518"getresgid() -> (rgid, egid, sgid)\n\n\
Georg Brandla9b51d22010-09-05 17:07:12 +00009519Get tuple of the current process's real, effective, and saved group ids.");
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009520
9521static PyObject*
9522posix_getresgid (PyObject *self, PyObject *noargs)
9523{
Victor Stinner8c62be82010-05-06 00:08:46 +00009524 uid_t rgid, egid, sgid;
9525 long l_rgid, l_egid, l_sgid;
9526 if (getresgid(&rgid, &egid, &sgid) < 0)
9527 return posix_error();
9528 /* Force the values into long's as we don't know the size of uid_t. */
9529 l_rgid = rgid;
9530 l_egid = egid;
9531 l_sgid = sgid;
9532 return Py_BuildValue("(lll)", l_rgid, l_egid, l_sgid);
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009533}
9534#endif
9535
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009536/* Posix *at family of functions:
9537 faccessat, fchmodat, fchownat, fstatat, futimesat,
9538 linkat, mkdirat, mknodat, openat, readlinkat, renameat, symlinkat,
9539 unlinkat, utimensat, mkfifoat */
9540
9541#ifdef HAVE_FACCESSAT
9542PyDoc_STRVAR(posix_faccessat__doc__,
9543"faccessat(dirfd, path, mode, flags=0) -> True if granted, False otherwise\n\n\
9544Like access() but if path is relative, it is taken as relative to dirfd.\n\
9545flags is optional and can be constructed by ORing together zero or more\n\
9546of these values: AT_SYMLINK_NOFOLLOW, AT_EACCESS.\n\
9547If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9548is interpreted relative to the current working directory.");
9549
9550static PyObject *
9551posix_faccessat(PyObject *self, PyObject *args)
9552{
9553 PyObject *opath;
9554 char *path;
9555 int mode;
9556 int res;
9557 int dirfd, flags = 0;
9558 if (!PyArg_ParseTuple(args, "iO&i|i:faccessat",
9559 &dirfd, PyUnicode_FSConverter, &opath, &mode, &flags))
9560 return NULL;
9561 path = PyBytes_AsString(opath);
9562 Py_BEGIN_ALLOW_THREADS
9563 res = faccessat(dirfd, path, mode, flags);
9564 Py_END_ALLOW_THREADS
9565 Py_DECREF(opath);
9566 return PyBool_FromLong(res == 0);
9567}
9568#endif
9569
9570#ifdef HAVE_FCHMODAT
9571PyDoc_STRVAR(posix_fchmodat__doc__,
9572"fchmodat(dirfd, path, mode, flags=0)\n\n\
9573Like chmod() but if path is relative, it is taken as relative to dirfd.\n\
9574flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9575If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9576is interpreted relative to the current working directory.");
9577
9578static PyObject *
9579posix_fchmodat(PyObject *self, PyObject *args)
9580{
9581 int dirfd, mode, res;
9582 int flags = 0;
9583 PyObject *opath;
9584 char *path;
9585
9586 if (!PyArg_ParseTuple(args, "iO&i|i:fchmodat",
9587 &dirfd, PyUnicode_FSConverter, &opath, &mode, &flags))
9588 return NULL;
9589
9590 path = PyBytes_AsString(opath);
9591
9592 Py_BEGIN_ALLOW_THREADS
9593 res = fchmodat(dirfd, path, mode, flags);
9594 Py_END_ALLOW_THREADS
9595 Py_DECREF(opath);
9596 if (res < 0)
9597 return posix_error();
9598 Py_RETURN_NONE;
9599}
9600#endif /* HAVE_FCHMODAT */
9601
9602#ifdef HAVE_FCHOWNAT
9603PyDoc_STRVAR(posix_fchownat__doc__,
9604"fchownat(dirfd, path, uid, gid, flags=0)\n\n\
9605Like chown() but if path is relative, it is taken as relative to dirfd.\n\
9606flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9607If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9608is interpreted relative to the current working directory.");
9609
9610static PyObject *
9611posix_fchownat(PyObject *self, PyObject *args)
9612{
9613 PyObject *opath;
9614 int dirfd, res;
9615 long uid, gid;
9616 int flags = 0;
9617 char *path;
Giampaolo Rodola'ff1a7352011-04-19 09:47:16 +02009618
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009619 if (!PyArg_ParseTuple(args, "iO&ll|i:fchownat",
9620 &dirfd, PyUnicode_FSConverter, &opath, &uid, &gid, &flags))
9621 return NULL;
9622
9623 path = PyBytes_AsString(opath);
9624
9625 Py_BEGIN_ALLOW_THREADS
9626 res = fchownat(dirfd, path, (uid_t) uid, (gid_t) gid, flags);
9627 Py_END_ALLOW_THREADS
9628 Py_DECREF(opath);
9629 if (res < 0)
9630 return posix_error();
9631 Py_RETURN_NONE;
9632}
9633#endif /* HAVE_FCHOWNAT */
9634
9635#ifdef HAVE_FSTATAT
9636PyDoc_STRVAR(posix_fstatat__doc__,
Victor Stinner4195b5c2012-02-08 23:03:19 +01009637"fstatat(dirfd, path, flags=0) -> stat result\n\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009638Like stat() but if path is relative, it is taken as relative to dirfd.\n\
9639flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9640If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9641is interpreted relative to the current working directory.");
9642
9643static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +01009644posix_fstatat(PyObject *self, PyObject *args)
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009645{
9646 PyObject *opath;
9647 char *path;
9648 STRUCT_STAT st;
9649 int dirfd, res, flags = 0;
9650
Victor Stinner4195b5c2012-02-08 23:03:19 +01009651 if (!PyArg_ParseTuple(args, "iO&|i:fstatat",
9652 &dirfd, PyUnicode_FSConverter, &opath, &flags))
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009653 return NULL;
9654 path = PyBytes_AsString(opath);
9655
9656 Py_BEGIN_ALLOW_THREADS
9657 res = fstatat(dirfd, path, &st, flags);
9658 Py_END_ALLOW_THREADS
9659 Py_DECREF(opath);
9660 if (res != 0)
9661 return posix_error();
9662
Victor Stinner4195b5c2012-02-08 23:03:19 +01009663 return _pystat_fromstructstat(&st);
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009664}
9665#endif
9666
9667#ifdef HAVE_FUTIMESAT
9668PyDoc_STRVAR(posix_futimesat__doc__,
Brian Curtin7ef53ef2011-11-07 14:38:24 -06009669"futimesat(dirfd, path[, (atime, mtime)])\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009670Like utime() but if path is relative, it is taken as relative to dirfd.\n\
9671If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9672is interpreted relative to the current working directory.");
9673
9674static PyObject *
9675posix_futimesat(PyObject *self, PyObject *args)
9676{
9677 PyObject *opath;
9678 char *path;
9679 int res, dirfd;
Brian Curtin7ef53ef2011-11-07 14:38:24 -06009680 PyObject* arg = Py_None;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009681 time_t atime, mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009682 long ansec, mnsec;
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009683
Brian Curtin7ef53ef2011-11-07 14:38:24 -06009684 if (!PyArg_ParseTuple(args, "iO&|O:futimesat",
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009685 &dirfd, PyUnicode_FSConverter, &opath, &arg))
9686 return NULL;
9687 path = PyBytes_AsString(opath);
9688 if (arg == Py_None) {
9689 /* optional time values not given */
9690 Py_BEGIN_ALLOW_THREADS
9691 res = futimesat(dirfd, path, NULL);
9692 Py_END_ALLOW_THREADS
9693 }
9694 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
9695 PyErr_SetString(PyExc_TypeError,
9696 "futimesat() arg 3 must be a tuple (atime, mtime)");
9697 Py_DECREF(opath);
9698 return NULL;
9699 }
9700 else {
9701 if (extract_time(PyTuple_GET_ITEM(arg, 0),
Victor Stinnera2f7c002012-02-08 03:36:25 +01009702 &atime, &ansec) == -1) {
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009703 Py_DECREF(opath);
9704 return NULL;
9705 }
9706 if (extract_time(PyTuple_GET_ITEM(arg, 1),
Victor Stinnera2f7c002012-02-08 03:36:25 +01009707 &mtime, &mnsec) == -1) {
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009708 Py_DECREF(opath);
9709 return NULL;
9710 }
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009711
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009712 Py_BEGIN_ALLOW_THREADS
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009713 {
9714#ifdef HAVE_UTIMENSAT
9715 struct timespec buf[2];
9716 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009717 buf[0].tv_nsec = ansec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009718 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009719 buf[1].tv_nsec = mnsec;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009720 res = utimensat(dirfd, path, buf, 0);
9721#else
9722 struct timeval buf[2];
9723 buf[0].tv_sec = atime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009724 buf[0].tv_usec = ansec / 1000;
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009725 buf[1].tv_sec = mtime;
Victor Stinnera2f7c002012-02-08 03:36:25 +01009726 buf[1].tv_usec = mnsec / 1000;
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009727 res = futimesat(dirfd, path, buf);
Larry Hastings9e3e70b2011-09-08 19:29:07 -07009728#endif
9729 }
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009730 Py_END_ALLOW_THREADS
9731 }
9732 Py_DECREF(opath);
9733 if (res < 0) {
9734 return posix_error();
9735 }
9736 Py_RETURN_NONE;
9737}
9738#endif
9739
9740#ifdef HAVE_LINKAT
9741PyDoc_STRVAR(posix_linkat__doc__,
9742"linkat(srcfd, srcpath, dstfd, dstpath, flags=0)\n\n\
9743Like link() but if srcpath is relative, it is taken as relative to srcfd\n\
9744and if dstpath is relative, it is taken as relative to dstfd.\n\
9745flags is optional and may be 0 or AT_SYMLINK_FOLLOW.\n\
9746If srcpath is relative and srcfd is the special value AT_FDCWD, then\n\
9747srcpath is interpreted relative to the current working directory. This\n\
9748also applies for dstpath.");
9749
9750static PyObject *
9751posix_linkat(PyObject *self, PyObject *args)
9752{
9753 PyObject *osrc, *odst;
9754 char *src, *dst;
9755 int res, srcfd, dstfd;
9756 int flags = 0;
9757
9758 if (!PyArg_ParseTuple(args, "iO&iO&|i:linkat",
9759 &srcfd, PyUnicode_FSConverter, &osrc, &dstfd, PyUnicode_FSConverter, &odst, &flags))
9760 return NULL;
9761 src = PyBytes_AsString(osrc);
9762 dst = PyBytes_AsString(odst);
9763 Py_BEGIN_ALLOW_THREADS
9764 res = linkat(srcfd, src, dstfd, dst, flags);
9765 Py_END_ALLOW_THREADS
9766 Py_DECREF(osrc);
9767 Py_DECREF(odst);
9768 if (res < 0)
9769 return posix_error();
9770 Py_RETURN_NONE;
9771}
9772#endif /* HAVE_LINKAT */
9773
9774#ifdef HAVE_MKDIRAT
9775PyDoc_STRVAR(posix_mkdirat__doc__,
9776"mkdirat(dirfd, path, mode=0o777)\n\n\
9777Like mkdir() but if path is relative, it is taken as relative to dirfd.\n\
9778If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9779is interpreted relative to the current working directory.");
9780
9781static PyObject *
9782posix_mkdirat(PyObject *self, PyObject *args)
9783{
9784 int res, dirfd;
9785 PyObject *opath;
9786 char *path;
9787 int mode = 0777;
9788
9789 if (!PyArg_ParseTuple(args, "iO&|i:mkdirat",
9790 &dirfd, PyUnicode_FSConverter, &opath, &mode))
9791 return NULL;
9792 path = PyBytes_AsString(opath);
9793 Py_BEGIN_ALLOW_THREADS
9794 res = mkdirat(dirfd, path, mode);
9795 Py_END_ALLOW_THREADS
9796 Py_DECREF(opath);
9797 if (res < 0)
9798 return posix_error();
9799 Py_RETURN_NONE;
9800}
9801#endif
9802
9803#if defined(HAVE_MKNODAT) && defined(HAVE_MAKEDEV)
9804PyDoc_STRVAR(posix_mknodat__doc__,
9805"mknodat(dirfd, path, mode=0o600, device=0)\n\n\
9806Like mknod() but if path is relative, it is taken as relative to dirfd.\n\
9807If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9808is interpreted relative to the current working directory.");
9809
9810static PyObject *
9811posix_mknodat(PyObject *self, PyObject *args)
9812{
9813 PyObject *opath;
9814 char *filename;
9815 int mode = 0600;
9816 int device = 0;
9817 int res, dirfd;
9818 if (!PyArg_ParseTuple(args, "iO&|ii:mknodat", &dirfd,
9819 PyUnicode_FSConverter, &opath, &mode, &device))
9820 return NULL;
9821 filename = PyBytes_AS_STRING(opath);
9822 Py_BEGIN_ALLOW_THREADS
9823 res = mknodat(dirfd, filename, mode, device);
9824 Py_END_ALLOW_THREADS
9825 Py_DECREF(opath);
9826 if (res < 0)
9827 return posix_error();
9828 Py_RETURN_NONE;
9829}
9830#endif
9831
9832#ifdef HAVE_OPENAT
9833PyDoc_STRVAR(posix_openat__doc__,
9834"openat(dirfd, path, flag, mode=0o777) -> fd\n\n\
9835Like open() but if path is relative, it is taken as relative to dirfd.\n\
9836If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9837is interpreted relative to the current working directory.");
9838
9839static PyObject *
9840posix_openat(PyObject *self, PyObject *args)
9841{
9842 PyObject *ofile;
9843 char *file;
9844 int flag, dirfd, fd;
9845 int mode = 0777;
9846
9847 if (!PyArg_ParseTuple(args, "iO&i|i:openat",
9848 &dirfd, PyUnicode_FSConverter, &ofile,
9849 &flag, &mode))
9850 return NULL;
9851 file = PyBytes_AsString(ofile);
9852 Py_BEGIN_ALLOW_THREADS
9853 fd = openat(dirfd, file, flag, mode);
9854 Py_END_ALLOW_THREADS
9855 Py_DECREF(ofile);
9856 if (fd < 0)
9857 return posix_error();
9858 return PyLong_FromLong((long)fd);
9859}
9860#endif
9861
9862#ifdef HAVE_READLINKAT
9863PyDoc_STRVAR(posix_readlinkat__doc__,
9864"readlinkat(dirfd, path) -> path\n\n\
9865Like readlink() but if path is relative, it is taken as relative to dirfd.\n\
9866If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9867is interpreted relative to the current working directory.");
9868
9869static PyObject *
9870posix_readlinkat(PyObject *self, PyObject *args)
9871{
9872 PyObject *v, *opath;
9873 char buf[MAXPATHLEN];
9874 char *path;
9875 int n, dirfd;
9876 int arg_is_unicode = 0;
9877
9878 if (!PyArg_ParseTuple(args, "iO&:readlinkat",
9879 &dirfd, PyUnicode_FSConverter, &opath))
9880 return NULL;
9881 path = PyBytes_AsString(opath);
9882 v = PySequence_GetItem(args, 1);
9883 if (v == NULL) {
9884 Py_DECREF(opath);
9885 return NULL;
9886 }
9887
9888 if (PyUnicode_Check(v)) {
9889 arg_is_unicode = 1;
9890 }
9891 Py_DECREF(v);
9892
9893 Py_BEGIN_ALLOW_THREADS
9894 n = readlinkat(dirfd, path, buf, (int) sizeof buf);
9895 Py_END_ALLOW_THREADS
9896 Py_DECREF(opath);
9897 if (n < 0)
9898 return posix_error();
9899
9900 if (arg_is_unicode)
9901 return PyUnicode_DecodeFSDefaultAndSize(buf, n);
9902 else
9903 return PyBytes_FromStringAndSize(buf, n);
9904}
9905#endif /* HAVE_READLINKAT */
9906
9907#ifdef HAVE_RENAMEAT
9908PyDoc_STRVAR(posix_renameat__doc__,
9909"renameat(olddirfd, oldpath, newdirfd, newpath)\n\n\
9910Like rename() but if oldpath is relative, it is taken as relative to\n\
9911olddirfd and if newpath is relative, it is taken as relative to newdirfd.\n\
9912If oldpath is relative and olddirfd is the special value AT_FDCWD, then\n\
9913oldpath is interpreted relative to the current working directory. This\n\
9914also applies for newpath.");
9915
9916static PyObject *
9917posix_renameat(PyObject *self, PyObject *args)
9918{
9919 int res;
9920 PyObject *opathold, *opathnew;
9921 char *opath, *npath;
9922 int oldfd, newfd;
9923
9924 if (!PyArg_ParseTuple(args, "iO&iO&:renameat",
9925 &oldfd, PyUnicode_FSConverter, &opathold, &newfd, PyUnicode_FSConverter, &opathnew))
9926 return NULL;
9927 opath = PyBytes_AsString(opathold);
9928 npath = PyBytes_AsString(opathnew);
9929 Py_BEGIN_ALLOW_THREADS
9930 res = renameat(oldfd, opath, newfd, npath);
9931 Py_END_ALLOW_THREADS
9932 Py_DECREF(opathold);
9933 Py_DECREF(opathnew);
9934 if (res < 0)
9935 return posix_error();
9936 Py_RETURN_NONE;
9937}
9938#endif
9939
9940#if HAVE_SYMLINKAT
9941PyDoc_STRVAR(posix_symlinkat__doc__,
9942"symlinkat(src, dstfd, dst)\n\n\
9943Like symlink() but if dst is relative, it is taken as relative to dstfd.\n\
9944If dst is relative and dstfd is the special value AT_FDCWD, then dst\n\
9945is interpreted relative to the current working directory.");
9946
9947static PyObject *
9948posix_symlinkat(PyObject *self, PyObject *args)
9949{
9950 int res, dstfd;
9951 PyObject *osrc, *odst;
9952 char *src, *dst;
9953
9954 if (!PyArg_ParseTuple(args, "O&iO&:symlinkat",
9955 PyUnicode_FSConverter, &osrc, &dstfd, PyUnicode_FSConverter, &odst))
9956 return NULL;
9957 src = PyBytes_AsString(osrc);
9958 dst = PyBytes_AsString(odst);
9959 Py_BEGIN_ALLOW_THREADS
9960 res = symlinkat(src, dstfd, dst);
9961 Py_END_ALLOW_THREADS
9962 Py_DECREF(osrc);
9963 Py_DECREF(odst);
9964 if (res < 0)
9965 return posix_error();
9966 Py_RETURN_NONE;
9967}
9968#endif /* HAVE_SYMLINKAT */
9969
9970#ifdef HAVE_UNLINKAT
9971PyDoc_STRVAR(posix_unlinkat__doc__,
9972"unlinkat(dirfd, path, flags=0)\n\n\
9973Like unlink() but if path is relative, it is taken as relative to dirfd.\n\
9974flags is optional and may be 0 or AT_REMOVEDIR. If AT_REMOVEDIR is\n\
9975specified, unlinkat() behaves like rmdir().\n\
9976If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9977is interpreted relative to the current working directory.");
9978
9979static PyObject *
9980posix_unlinkat(PyObject *self, PyObject *args)
9981{
9982 int dirfd, res, flags = 0;
9983 PyObject *opath;
9984 char *path;
9985
9986 if (!PyArg_ParseTuple(args, "iO&|i:unlinkat",
9987 &dirfd, PyUnicode_FSConverter, &opath, &flags))
9988 return NULL;
9989 path = PyBytes_AsString(opath);
9990 Py_BEGIN_ALLOW_THREADS
9991 res = unlinkat(dirfd, path, flags);
9992 Py_END_ALLOW_THREADS
9993 Py_DECREF(opath);
9994 if (res < 0)
9995 return posix_error();
9996 Py_RETURN_NONE;
9997}
9998#endif
9999
10000#ifdef HAVE_UTIMENSAT
10001PyDoc_STRVAR(posix_utimensat__doc__,
Brian Curtin569b4942011-11-07 16:09:20 -060010002"utimensat(dirfd, path[, atime=(atime_sec, atime_nsec),\n\
10003 mtime=(mtime_sec, mtime_nsec), flags=0])\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010004utimensat(dirfd, path, None, None, flags)\n\n\
10005Updates the timestamps of a file with nanosecond precision. If path is\n\
10006relative, it is taken as relative to dirfd.\n\
Brian Curtin569b4942011-11-07 16:09:20 -060010007If atime and mtime are both None, which is the default, set atime and\n\
10008mtime to the current time.\n\
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010009flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
10010If path is relative and dirfd is the special value AT_FDCWD, then path\n\
10011is interpreted relative to the current working directory.\n\
10012If *_nsec is specified as UTIME_NOW, the timestamp is updated to the\n\
10013current time.\n\
10014If *_nsec is specified as UTIME_OMIT, the timestamp is not updated.");
10015
10016static PyObject *
Brian Curtin569b4942011-11-07 16:09:20 -060010017posix_utimensat(PyObject *self, PyObject *args, PyObject *kwargs)
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010018{
10019 PyObject *opath;
10020 char *path;
10021 int res, dirfd, flags = 0;
Brian Curtin569b4942011-11-07 16:09:20 -060010022 PyObject *atime = Py_None;
10023 PyObject *mtime = Py_None;
10024
10025 static char *kwlist[] = {"dirfd", "path", "atime", "mtime", "flags", NULL};
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010026
10027 struct timespec buf[2];
10028
Brian Curtin569b4942011-11-07 16:09:20 -060010029 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "iO&|OOi:utimensat", kwlist,
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010030 &dirfd, PyUnicode_FSConverter, &opath, &atime, &mtime, &flags))
10031 return NULL;
10032 path = PyBytes_AsString(opath);
10033 if (atime == Py_None && mtime == Py_None) {
10034 /* optional time values not given */
10035 Py_BEGIN_ALLOW_THREADS
10036 res = utimensat(dirfd, path, NULL, flags);
10037 Py_END_ALLOW_THREADS
10038 }
10039 else if (!PyTuple_Check(atime) || PyTuple_Size(atime) != 2) {
10040 PyErr_SetString(PyExc_TypeError,
10041 "utimensat() arg 3 must be a tuple (atime_sec, atime_nsec)");
10042 Py_DECREF(opath);
10043 return NULL;
10044 }
10045 else if (!PyTuple_Check(mtime) || PyTuple_Size(mtime) != 2) {
10046 PyErr_SetString(PyExc_TypeError,
10047 "utimensat() arg 4 must be a tuple (mtime_sec, mtime_nsec)");
10048 Py_DECREF(opath);
10049 return NULL;
10050 }
10051 else {
10052 if (!PyArg_ParseTuple(atime, "ll:utimensat",
10053 &(buf[0].tv_sec), &(buf[0].tv_nsec))) {
10054 Py_DECREF(opath);
10055 return NULL;
10056 }
10057 if (!PyArg_ParseTuple(mtime, "ll:utimensat",
10058 &(buf[1].tv_sec), &(buf[1].tv_nsec))) {
10059 Py_DECREF(opath);
10060 return NULL;
10061 }
10062 Py_BEGIN_ALLOW_THREADS
10063 res = utimensat(dirfd, path, buf, flags);
10064 Py_END_ALLOW_THREADS
10065 }
10066 Py_DECREF(opath);
10067 if (res < 0) {
10068 return posix_error();
10069 }
10070 Py_RETURN_NONE;
10071}
10072#endif
10073
10074#ifdef HAVE_MKFIFOAT
10075PyDoc_STRVAR(posix_mkfifoat__doc__,
10076"mkfifoat(dirfd, path, mode=0o666)\n\n\
10077Like mkfifo() but if path is relative, it is taken as relative to dirfd.\n\
10078If path is relative and dirfd is the special value AT_FDCWD, then path\n\
10079is interpreted relative to the current working directory.");
10080
10081static PyObject *
10082posix_mkfifoat(PyObject *self, PyObject *args)
10083{
10084 PyObject *opath;
10085 char *filename;
10086 int mode = 0666;
10087 int res, dirfd;
10088 if (!PyArg_ParseTuple(args, "iO&|i:mkfifoat",
10089 &dirfd, PyUnicode_FSConverter, &opath, &mode))
10090 return NULL;
10091 filename = PyBytes_AS_STRING(opath);
10092 Py_BEGIN_ALLOW_THREADS
10093 res = mkfifoat(dirfd, filename, mode);
10094 Py_END_ALLOW_THREADS
10095 Py_DECREF(opath);
10096 if (res < 0)
10097 return posix_error();
10098 Py_RETURN_NONE;
10099}
10100#endif
10101
Benjamin Peterson9428d532011-09-14 11:45:52 -040010102#ifdef USE_XATTRS
Benjamin Peterson799bd802011-08-31 22:15:17 -040010103
10104static int
10105try_getxattr(const char *path, const char *name,
10106 ssize_t (*get)(const char *, const char *, void *, size_t),
10107 Py_ssize_t buf_size, PyObject **res)
10108{
10109 PyObject *value;
10110 Py_ssize_t len;
10111
10112 assert(buf_size <= XATTR_SIZE_MAX);
10113 value = PyBytes_FromStringAndSize(NULL, buf_size);
10114 if (!value)
10115 return 0;
10116 Py_BEGIN_ALLOW_THREADS;
10117 len = get(path, name, PyBytes_AS_STRING(value), buf_size);
10118 Py_END_ALLOW_THREADS;
10119 if (len < 0) {
10120 Py_DECREF(value);
10121 if (errno == ERANGE) {
10122 value = NULL;
10123 }
10124 else {
10125 posix_error();
10126 return 0;
10127 }
10128 }
10129 else if (len != buf_size) {
10130 /* Can only shrink. */
10131 _PyBytes_Resize(&value, len);
10132 }
10133 *res = value;
10134 return 1;
10135}
10136
10137static PyObject *
10138getxattr_common(const char *path, PyObject *name_obj,
10139 ssize_t (*get)(const char *, const char *, void *, size_t))
10140{
10141 PyObject *value;
10142 const char *name = PyBytes_AS_STRING(name_obj);
10143
10144 /* Try a small value first. */
10145 if (!try_getxattr(path, name, get, 128, &value))
10146 return NULL;
10147 if (value)
10148 return value;
10149 /* Now the maximum possible one. */
10150 if (!try_getxattr(path, name, get, XATTR_SIZE_MAX, &value))
10151 return NULL;
10152 assert(value);
10153 return value;
10154}
10155
10156PyDoc_STRVAR(posix_getxattr__doc__,
10157"getxattr(path, attr) -> value\n\n\
10158Return the value of extended attribute *name* on *path*.");
10159
10160static PyObject *
10161posix_getxattr(PyObject *self, PyObject *args)
10162{
10163 PyObject *path, *res, *name;
10164
10165 if (!PyArg_ParseTuple(args, "O&O&:getxattr", PyUnicode_FSConverter, &path,
10166 PyUnicode_FSConverter, &name))
10167 return NULL;
10168 res = getxattr_common(PyBytes_AS_STRING(path), name, getxattr);
10169 Py_DECREF(path);
10170 Py_DECREF(name);
10171 return res;
10172}
10173
10174PyDoc_STRVAR(posix_lgetxattr__doc__,
10175"lgetxattr(path, attr) -> value\n\n\
10176Like getxattr but don't follow symlinks.");
10177
10178static PyObject *
10179posix_lgetxattr(PyObject *self, PyObject *args)
10180{
10181 PyObject *path, *res, *name;
10182
10183 if (!PyArg_ParseTuple(args, "O&O&:lgetxattr", PyUnicode_FSConverter, &path,
10184 PyUnicode_FSConverter, &name))
10185 return NULL;
10186 res = getxattr_common(PyBytes_AS_STRING(path), name, lgetxattr);
10187 Py_DECREF(path);
10188 Py_DECREF(name);
10189 return res;
10190}
10191
10192static ssize_t
10193wrap_fgetxattr(const char *path, const char *name, void *value, size_t size)
10194{
10195 /* Hack to share code. */
10196 return fgetxattr((int)(Py_uintptr_t)path, name, value, size);
10197}
10198
10199PyDoc_STRVAR(posix_fgetxattr__doc__,
10200"fgetxattr(fd, attr) -> value\n\n\
10201Like getxattr but operate on a fd instead of a path.");
10202
10203static PyObject *
10204posix_fgetxattr(PyObject *self, PyObject *args)
10205{
10206 PyObject *res, *name;
10207 int fd;
10208
10209 if (!PyArg_ParseTuple(args, "iO&:fgetxattr", &fd, PyUnicode_FSConverter, &name))
10210 return NULL;
10211 res = getxattr_common((const char *)(Py_uintptr_t)fd, name, wrap_fgetxattr);
10212 Py_DECREF(name);
10213 return res;
10214}
10215
10216PyDoc_STRVAR(posix_setxattr__doc__,
10217"setxattr(path, attr, value, flags=0)\n\n\
10218Set extended attribute *attr* on *path* to *value*.");
10219
10220static PyObject *
10221posix_setxattr(PyObject *self, PyObject *args)
10222{
10223 PyObject *path, *name;
10224 Py_buffer data;
10225 int flags = 0, err;
10226
10227 if (!PyArg_ParseTuple(args, "O&O&y*|i:setxattr", PyUnicode_FSConverter,
10228 &path, PyUnicode_FSConverter, &name, &data, &flags))
10229 return NULL;
10230 Py_BEGIN_ALLOW_THREADS;
10231 err = setxattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name),
10232 data.buf, data.len, flags);
10233 Py_END_ALLOW_THREADS;
10234 Py_DECREF(path);
10235 Py_DECREF(name);
10236 PyBuffer_Release(&data);
10237 if (err)
10238 return posix_error();
10239 Py_RETURN_NONE;
10240}
10241
10242PyDoc_STRVAR(posix_lsetxattr__doc__,
10243"lsetxattr(path, attr, value, flags=0)\n\n\
10244Like setxattr but don't follow symlinks.");
10245
10246static PyObject *
10247posix_lsetxattr(PyObject *self, PyObject *args)
10248{
10249 PyObject *path, *name;
10250 Py_buffer data;
10251 int flags = 0, err;
10252
10253 if (!PyArg_ParseTuple(args, "O&O&y*|i:lsetxattr", PyUnicode_FSConverter,
10254 &path, PyUnicode_FSConverter, &name, &data, &flags))
10255 return NULL;
10256 Py_BEGIN_ALLOW_THREADS;
10257 err = lsetxattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name),
10258 data.buf, data.len, flags);
10259 Py_END_ALLOW_THREADS;
10260 Py_DECREF(path);
10261 Py_DECREF(name);
10262 PyBuffer_Release(&data);
10263 if (err)
10264 return posix_error();
10265 Py_RETURN_NONE;
10266}
10267
10268PyDoc_STRVAR(posix_fsetxattr__doc__,
10269"fsetxattr(fd, attr, value, flags=0)\n\n\
10270Like setxattr but operates on *fd* instead of a path.");
10271
10272static PyObject *
10273posix_fsetxattr(PyObject *self, PyObject *args)
10274{
10275 Py_buffer data;
10276 const char *name;
10277 int fd, flags = 0, err;
10278
10279 if (!PyArg_ParseTuple(args, "iO&y*|i:fsetxattr", &fd, PyUnicode_FSConverter,
10280 &name, &data, &flags))
10281 return NULL;
10282 Py_BEGIN_ALLOW_THREADS;
10283 err = fsetxattr(fd, PyBytes_AS_STRING(name), data.buf, data.len, flags);
10284 Py_END_ALLOW_THREADS;
10285 Py_DECREF(name);
10286 PyBuffer_Release(&data);
10287 if (err)
10288 return posix_error();
10289 Py_RETURN_NONE;
10290}
10291
10292PyDoc_STRVAR(posix_removexattr__doc__,
10293"removexattr(path, attr)\n\n\
10294Remove extended attribute *attr* on *path*.");
10295
10296static PyObject *
10297posix_removexattr(PyObject *self, PyObject *args)
10298{
10299 PyObject *path, *name;
10300 int err;
10301
10302 if (!PyArg_ParseTuple(args, "O&O&:removexattr", PyUnicode_FSConverter, &path,
10303 PyUnicode_FSConverter, &name))
10304 return NULL;
10305 Py_BEGIN_ALLOW_THREADS;
10306 err = removexattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name));
10307 Py_END_ALLOW_THREADS;
10308 Py_DECREF(path);
10309 Py_DECREF(name);
10310 if (err)
10311 return posix_error();
10312 Py_RETURN_NONE;
10313}
10314
10315PyDoc_STRVAR(posix_lremovexattr__doc__,
10316"lremovexattr(path, attr)\n\n\
10317Like removexattr but don't follow symlinks.");
10318
10319static PyObject *
10320posix_lremovexattr(PyObject *self, PyObject *args)
10321{
10322 PyObject *path, *name;
10323 int err;
10324
10325 if (!PyArg_ParseTuple(args, "O&O&:lremovexattr", PyUnicode_FSConverter, &path,
10326 PyUnicode_FSConverter, &name))
10327 return NULL;
10328 Py_BEGIN_ALLOW_THREADS;
10329 err = lremovexattr(PyBytes_AS_STRING(path), PyBytes_AS_STRING(name));
10330 Py_END_ALLOW_THREADS;
10331 Py_DECREF(path);
10332 Py_DECREF(name);
10333 if (err)
10334 return posix_error();
10335 Py_RETURN_NONE;
10336}
10337
10338PyDoc_STRVAR(posix_fremovexattr__doc__,
10339"fremovexattr(fd, attr)\n\n\
10340Like removexattr but operates on a file descriptor.");
10341
10342static PyObject *
10343posix_fremovexattr(PyObject *self, PyObject *args)
10344{
10345 PyObject *name;
10346 int fd, err;
10347
10348 if (!PyArg_ParseTuple(args, "iO&:fremovexattr", &fd,
10349 PyUnicode_FSConverter, &name))
10350 return NULL;
10351 Py_BEGIN_ALLOW_THREADS;
10352 err = fremovexattr(fd, PyBytes_AS_STRING(name));
10353 Py_END_ALLOW_THREADS;
10354 Py_DECREF(name);
10355 if (err)
10356 return posix_error();
10357 Py_RETURN_NONE;
10358}
10359
10360static Py_ssize_t
10361try_listxattr(const char *path, ssize_t (*list)(const char *, char *, size_t),
10362 Py_ssize_t buf_size, char **buf)
10363{
10364 Py_ssize_t len;
10365
10366 *buf = PyMem_MALLOC(buf_size);
10367 if (!*buf) {
10368 PyErr_NoMemory();
10369 return -1;
10370 }
10371 Py_BEGIN_ALLOW_THREADS;
10372 len = list(path, *buf, buf_size);
10373 Py_END_ALLOW_THREADS;
10374 if (len < 0) {
10375 PyMem_FREE(*buf);
10376 if (errno != ERANGE)
10377 posix_error();
10378 return -1;
10379 }
10380 return len;
10381}
10382
10383static PyObject *
10384listxattr_common(const char *path, ssize_t (*list)(const char *, char *, size_t))
10385{
10386 PyObject *res, *attr;
10387 Py_ssize_t len, err, start, i;
10388 char *buf;
10389
10390 len = try_listxattr(path, list, 256, &buf);
10391 if (len < 0) {
10392 if (PyErr_Occurred())
10393 return NULL;
10394 len = try_listxattr(path, list, XATTR_LIST_MAX, &buf);
10395 if (len < 0)
10396 return NULL;
10397 }
10398 res = PyList_New(0);
10399 if (!res) {
10400 PyMem_FREE(buf);
10401 return NULL;
10402 }
10403 for (start = i = 0; i < len; i++) {
10404 if (!buf[i]) {
10405 attr = PyUnicode_DecodeFSDefaultAndSize(&buf[start], i - start);
10406 if (!attr) {
10407 Py_DECREF(res);
10408 PyMem_FREE(buf);
10409 return NULL;
10410 }
10411 err = PyList_Append(res, attr);
10412 Py_DECREF(attr);
10413 if (err) {
10414 Py_DECREF(res);
10415 PyMem_FREE(buf);
10416 return NULL;
10417 }
10418 start = i + 1;
10419 }
10420 }
10421 PyMem_FREE(buf);
10422 return res;
10423}
10424
10425PyDoc_STRVAR(posix_listxattr__doc__,
10426"listxattr(path)\n\n\
10427Return a list of extended attributes on *path*.");
10428
10429static PyObject *
10430posix_listxattr(PyObject *self, PyObject *args)
10431{
10432 PyObject *path, *res;
10433
10434 if (!PyArg_ParseTuple(args, "O&:listxattr", PyUnicode_FSConverter, &path))
10435 return NULL;
10436 res = listxattr_common(PyBytes_AS_STRING(path), listxattr);
10437 Py_DECREF(path);
10438 return res;
10439}
10440
10441PyDoc_STRVAR(posix_llistxattr__doc__,
10442"llistxattr(path)\n\n\
10443Like listxattr but don't follow symlinks..");
10444
10445static PyObject *
10446posix_llistxattr(PyObject *self, PyObject *args)
10447{
10448 PyObject *path, *res;
10449
10450 if (!PyArg_ParseTuple(args, "O&:llistxattr", PyUnicode_FSConverter, &path))
10451 return NULL;
10452 res = listxattr_common(PyBytes_AS_STRING(path), llistxattr);
10453 Py_DECREF(path);
10454 return res;
10455}
10456
10457static ssize_t
10458wrap_flistxattr(const char *path, char *buf, size_t len)
10459{
10460 /* Hack to share code. */
10461 return flistxattr((int)(Py_uintptr_t)path, buf, len);
10462}
10463
10464PyDoc_STRVAR(posix_flistxattr__doc__,
10465"flistxattr(path)\n\n\
10466Like flistxattr but operates on a file descriptor.");
10467
10468static PyObject *
10469posix_flistxattr(PyObject *self, PyObject *args)
10470{
10471 long fd;
10472
10473 if (!PyArg_ParseTuple(args, "i:flistxattr", &fd))
10474 return NULL;
10475 return listxattr_common((const char *)(Py_uintptr_t)fd, wrap_flistxattr);
10476}
10477
Benjamin Peterson9428d532011-09-14 11:45:52 -040010478#endif /* USE_XATTRS */
Benjamin Peterson799bd802011-08-31 22:15:17 -040010479
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010480static PyMethodDef posix_methods[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +000010481 {"access", posix_access, METH_VARARGS, posix_access__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010482#ifdef HAVE_TTYNAME
Victor Stinner8c62be82010-05-06 00:08:46 +000010483 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010484#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010485 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +000010486#ifdef HAVE_CHFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +000010487 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +000010488#endif /* HAVE_CHFLAGS */
Victor Stinner8c62be82010-05-06 00:08:46 +000010489 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010490#ifdef HAVE_FCHMOD
Victor Stinner8c62be82010-05-06 00:08:46 +000010491 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010492#endif /* HAVE_FCHMOD */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010493#ifdef HAVE_CHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010494 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010495#endif /* HAVE_CHOWN */
Christian Heimes4e30a842007-11-30 22:12:06 +000010496#ifdef HAVE_LCHMOD
Victor Stinner8c62be82010-05-06 00:08:46 +000010497 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010498#endif /* HAVE_LCHMOD */
10499#ifdef HAVE_FCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010500 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +000010501#endif /* HAVE_FCHOWN */
Thomas Wouterscf297e42007-02-23 15:07:44 +000010502#ifdef HAVE_LCHFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +000010503 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +000010504#endif /* HAVE_LCHFLAGS */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +000010505#ifdef HAVE_LCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010506 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +000010507#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +000010508#ifdef HAVE_CHROOT
Victor Stinner8c62be82010-05-06 00:08:46 +000010509 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
Martin v. Löwis244edc82001-10-04 22:44:26 +000010510#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010511#ifdef HAVE_CTERMID
Victor Stinner8c62be82010-05-06 00:08:46 +000010512 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +000010513#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010514#ifdef HAVE_GETCWD
Victor Stinner8c62be82010-05-06 00:08:46 +000010515 {"getcwd", (PyCFunction)posix_getcwd_unicode,
10516 METH_NOARGS, posix_getcwd__doc__},
10517 {"getcwdb", (PyCFunction)posix_getcwd_bytes,
10518 METH_NOARGS, posix_getcwdb__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010519#endif
10520#ifdef HAVE_LINK
Victor Stinner8c62be82010-05-06 00:08:46 +000010521 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010522#endif /* HAVE_LINK */
Victor Stinner8c62be82010-05-06 00:08:46 +000010523 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
Antoine Pitrou8250e232011-02-25 23:41:16 +000010524#ifdef HAVE_FDOPENDIR
Charles-François Natali77940902012-02-06 19:54:48 +010010525 {"flistdir", posix_flistdir, METH_VARARGS, posix_flistdir__doc__},
Antoine Pitrou8250e232011-02-25 23:41:16 +000010526#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +010010527 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +000010528 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010529#ifdef HAVE_NICE
Victor Stinner8c62be82010-05-06 00:08:46 +000010530 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum775f4da1993-01-09 17:18:52 +000010531#endif /* HAVE_NICE */
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +000010532#ifdef HAVE_GETPRIORITY
10533 {"getpriority", posix_getpriority, METH_VARARGS, posix_getpriority__doc__},
10534#endif /* HAVE_GETPRIORITY */
10535#ifdef HAVE_SETPRIORITY
10536 {"setpriority", posix_setpriority, METH_VARARGS, posix_setpriority__doc__},
10537#endif /* HAVE_SETPRIORITY */
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010538#ifdef HAVE_READLINK
Victor Stinner8c62be82010-05-06 00:08:46 +000010539 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010540#endif /* HAVE_READLINK */
Brian Curtind40e6f72010-07-08 21:39:08 +000010541#if !defined(HAVE_READLINK) && defined(MS_WINDOWS)
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +000010542 {"readlink", win_readlink, METH_VARARGS, win_readlink__doc__},
Brian Curtind40e6f72010-07-08 21:39:08 +000010543#endif /* !defined(HAVE_READLINK) && defined(MS_WINDOWS) */
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +000010544 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
Antoine Pitrouf3b2d882012-01-30 22:08:52 +010010545 {"replace", posix_replace, METH_VARARGS, posix_replace__doc__},
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +000010546 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
Victor Stinner4195b5c2012-02-08 23:03:19 +010010547 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +000010548 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Brian Curtin52173d42010-12-02 18:29:18 +000010549#if defined(HAVE_SYMLINK) && !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +000010550 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossumc39de5f1992-02-05 11:15:54 +000010551#endif /* HAVE_SYMLINK */
Brian Curtin52173d42010-12-02 18:29:18 +000010552#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000010553 {"symlink", (PyCFunction)win_symlink, METH_VARARGS | METH_KEYWORDS,
Brian Curtin52173d42010-12-02 18:29:18 +000010554 win_symlink__doc__},
10555#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
Guido van Rossum85e3b011991-06-03 12:42:10 +000010556#ifdef HAVE_SYSTEM
Victor Stinner8c62be82010-05-06 00:08:46 +000010557 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010558#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010559 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010560#ifdef HAVE_UNAME
Victor Stinner8c62be82010-05-06 00:08:46 +000010561 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010562#endif /* HAVE_UNAME */
Victor Stinner8c62be82010-05-06 00:08:46 +000010563 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
10564 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
10565 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010566#ifdef HAVE_FUTIMES
10567 {"futimes", posix_futimes, METH_VARARGS, posix_futimes__doc__},
10568#endif
10569#ifdef HAVE_LUTIMES
10570 {"lutimes", posix_lutimes, METH_VARARGS, posix_lutimes__doc__},
10571#endif
10572#ifdef HAVE_FUTIMENS
10573 {"futimens", posix_futimens, METH_VARARGS, posix_futimens__doc__},
10574#endif
Guido van Rossum0ee42cd1991-04-08 21:01:03 +000010575#ifdef HAVE_TIMES
Victor Stinner8c62be82010-05-06 00:08:46 +000010576 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum3b066191991-06-04 19:40:25 +000010577#endif /* HAVE_TIMES */
Victor Stinner8c62be82010-05-06 00:08:46 +000010578 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossum3b066191991-06-04 19:40:25 +000010579#ifdef HAVE_EXECV
Victor Stinner8c62be82010-05-06 00:08:46 +000010580 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
10581 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossum3b066191991-06-04 19:40:25 +000010582#endif /* HAVE_EXECV */
Ross Lagerwall7807c352011-03-17 20:20:30 +020010583#ifdef HAVE_FEXECVE
10584 {"fexecve", posix_fexecve, METH_VARARGS, posix_fexecve__doc__},
10585#endif
Guido van Rossuma1065681999-01-25 23:20:23 +000010586#ifdef HAVE_SPAWNV
Victor Stinner8c62be82010-05-06 00:08:46 +000010587 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
10588 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +000010589#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +000010590 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
10591 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +000010592#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +000010593#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +000010594#ifdef HAVE_FORK1
Victor Stinner8c62be82010-05-06 00:08:46 +000010595 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +000010596#endif /* HAVE_FORK1 */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010597#ifdef HAVE_FORK
Victor Stinner8c62be82010-05-06 00:08:46 +000010598 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010599#endif /* HAVE_FORK */
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010600#ifdef HAVE_SCHED_H
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +020010601#ifdef HAVE_SCHED_GET_PRIORITY_MAX
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010602 {"sched_get_priority_max", posix_sched_get_priority_max, METH_VARARGS, posix_sched_get_priority_max__doc__},
10603 {"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 +020010604#endif
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010605#ifdef HAVE_SCHED_SETPARAM
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010606 {"sched_getparam", posix_sched_getparam, METH_VARARGS, posix_sched_getparam__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010607#endif
10608#ifdef HAVE_SCHED_SETSCHEDULER
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010609 {"sched_getscheduler", posix_sched_getscheduler, METH_VARARGS, posix_sched_getscheduler__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010610#endif
10611#ifdef HAVE_SCHED_RR_GET_INTERVAL
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010612 {"sched_rr_get_interval", posix_sched_rr_get_interval, METH_VARARGS, posix_sched_rr_get_interval__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010613#endif
10614#ifdef HAVE_SCHED_SETPARAM
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010615 {"sched_setparam", posix_sched_setparam, METH_VARARGS, posix_sched_setparam__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010616#endif
10617#ifdef HAVE_SCHED_SETSCHEDULER
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010618 {"sched_setscheduler", posix_sched_setscheduler, METH_VARARGS, posix_sched_setscheduler__doc__},
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -050010619#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010620 {"sched_yield", posix_sched_yield, METH_NOARGS, posix_sched_yield__doc__},
Benjamin Peterson2740af82011-08-02 17:41:34 -050010621#ifdef HAVE_SCHED_SETAFFINITY
Benjamin Peterson94b580d2011-08-02 17:30:04 -050010622 {"sched_setaffinity", posix_sched_setaffinity, METH_VARARGS, posix_sched_setaffinity__doc__},
10623 {"sched_getaffinity", posix_sched_getaffinity, METH_VARARGS, posix_sched_getaffinity__doc__},
10624#endif
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +020010625#endif /* HAVE_SCHED_H */
Martin v. Löwis24a880b2002-12-31 12:55:15 +000010626#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Victor Stinner8c62be82010-05-06 00:08:46 +000010627 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +000010628#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +000010629#ifdef HAVE_FORKPTY
Victor Stinner8c62be82010-05-06 00:08:46 +000010630 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +000010631#endif /* HAVE_FORKPTY */
Guido van Rossum3b066191991-06-04 19:40:25 +000010632#ifdef HAVE_GETEGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010633 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +000010634#endif /* HAVE_GETEGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010635#ifdef HAVE_GETEUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010636 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossum46003ff1992-05-15 11:05:24 +000010637#endif /* HAVE_GETEUID */
10638#ifdef HAVE_GETGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010639 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010640#endif /* HAVE_GETGID */
Ross Lagerwallb0ae53d2011-06-10 07:30:30 +020010641#ifdef HAVE_GETGROUPLIST
10642 {"getgrouplist", posix_getgrouplist, METH_VARARGS, posix_getgrouplist__doc__},
10643#endif
Fred Drakec9680921999-12-13 16:37:25 +000010644#ifdef HAVE_GETGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +000010645 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010646#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010647 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +000010648#ifdef HAVE_GETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010649 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010650#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +000010651#ifdef HAVE_GETPPID
Victor Stinner8c62be82010-05-06 00:08:46 +000010652 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010653#endif /* HAVE_GETPPID */
10654#ifdef HAVE_GETUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010655 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010656#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +000010657#ifdef HAVE_GETLOGIN
Victor Stinner8c62be82010-05-06 00:08:46 +000010658 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +000010659#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +000010660#ifdef HAVE_KILL
Victor Stinner8c62be82010-05-06 00:08:46 +000010661 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010662#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +000010663#ifdef HAVE_KILLPG
Victor Stinner8c62be82010-05-06 00:08:46 +000010664 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
Martin v. Löwisb2c92f42002-02-16 23:35:41 +000010665#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +000010666#ifdef HAVE_PLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000010667 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +000010668#endif /* HAVE_PLOCK */
Thomas Heller8b7a9572007-08-31 06:44:36 +000010669#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +000010670 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
10671 {"kill", win32_kill, METH_VARARGS, win32_kill__doc__},
Brian Curtin1b9df392010-11-24 20:24:31 +000010672 {"link", win32_link, METH_VARARGS, win32_link__doc__},
Thomas Heller8b7a9572007-08-31 06:44:36 +000010673#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +000010674#ifdef HAVE_SETUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010675 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010676#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010677#ifdef HAVE_SETEUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010678 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010679#endif /* HAVE_SETEUID */
10680#ifdef HAVE_SETEGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010681 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010682#endif /* HAVE_SETEGID */
10683#ifdef HAVE_SETREUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010684 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010685#endif /* HAVE_SETREUID */
10686#ifdef HAVE_SETREGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010687 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +000010688#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010689#ifdef HAVE_SETGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010690 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010691#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +000010692#ifdef HAVE_SETGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +000010693 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
Martin v. Löwis61c5edf2001-10-18 04:06:00 +000010694#endif /* HAVE_SETGROUPS */
Antoine Pitroub7572f02009-12-02 20:46:48 +000010695#ifdef HAVE_INITGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +000010696 {"initgroups", posix_initgroups, METH_VARARGS, posix_initgroups__doc__},
Antoine Pitroub7572f02009-12-02 20:46:48 +000010697#endif /* HAVE_INITGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +000010698#ifdef HAVE_GETPGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010699 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
Martin v. Löwis606edc12002-06-13 21:09:11 +000010700#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010701#ifdef HAVE_SETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010702 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010703#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +000010704#ifdef HAVE_WAIT
Victor Stinner8c62be82010-05-06 00:08:46 +000010705 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +000010706#endif /* HAVE_WAIT */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010707#ifdef HAVE_WAIT3
Victor Stinner4195b5c2012-02-08 23:03:19 +010010708 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010709#endif /* HAVE_WAIT3 */
10710#ifdef HAVE_WAIT4
Victor Stinner4195b5c2012-02-08 23:03:19 +010010711 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010712#endif /* HAVE_WAIT4 */
Ross Lagerwall7807c352011-03-17 20:20:30 +020010713#if defined(HAVE_WAITID) && !defined(__APPLE__)
10714 {"waitid", posix_waitid, METH_VARARGS, posix_waitid__doc__},
10715#endif
Tim Petersab034fa2002-02-01 11:27:43 +000010716#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Victor Stinner8c62be82010-05-06 00:08:46 +000010717 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010718#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +000010719#ifdef HAVE_GETSID
Victor Stinner8c62be82010-05-06 00:08:46 +000010720 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
Martin v. Löwis49ee14d2003-11-10 06:35:36 +000010721#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010722#ifdef HAVE_SETSID
Victor Stinner8c62be82010-05-06 00:08:46 +000010723 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010724#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010725#ifdef HAVE_SETPGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010726 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010727#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010728#ifdef HAVE_TCGETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010729 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010730#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +000010731#ifdef HAVE_TCSETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +000010732 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000010733#endif /* HAVE_TCSETPGRP */
Victor Stinner8c62be82010-05-06 00:08:46 +000010734 {"open", posix_open, METH_VARARGS, posix_open__doc__},
10735 {"close", posix_close, METH_VARARGS, posix_close__doc__},
10736 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
10737 {"device_encoding", device_encoding, METH_VARARGS, device_encoding__doc__},
10738 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
10739 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010740#ifdef HAVE_LOCKF
10741 {"lockf", posix_lockf, METH_VARARGS, posix_lockf__doc__},
10742#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010743 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
10744 {"read", posix_read, METH_VARARGS, posix_read__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010745#ifdef HAVE_READV
10746 {"readv", posix_readv, METH_VARARGS, posix_readv__doc__},
10747#endif
10748#ifdef HAVE_PREAD
10749 {"pread", posix_pread, METH_VARARGS, posix_pread__doc__},
10750#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010751 {"write", posix_write, METH_VARARGS, posix_write__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +020010752#ifdef HAVE_WRITEV
10753 {"writev", posix_writev, METH_VARARGS, posix_writev__doc__},
10754#endif
10755#ifdef HAVE_PWRITE
10756 {"pwrite", posix_pwrite, METH_VARARGS, posix_pwrite__doc__},
10757#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +000010758#ifdef HAVE_SENDFILE
10759 {"sendfile", (PyCFunction)posix_sendfile, METH_VARARGS | METH_KEYWORDS,
10760 posix_sendfile__doc__},
10761#endif
Victor Stinner4195b5c2012-02-08 23:03:19 +010010762 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +000010763 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010764#ifdef HAVE_PIPE
Victor Stinner8c62be82010-05-06 00:08:46 +000010765 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010766#endif
Charles-François Natalidaafdd52011-05-29 20:07:40 +020010767#ifdef HAVE_PIPE2
Charles-François Natali368f34b2011-06-06 19:49:47 +020010768 {"pipe2", posix_pipe2, METH_O, posix_pipe2__doc__},
Charles-François Natalidaafdd52011-05-29 20:07:40 +020010769#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010770#ifdef HAVE_MKFIFO
Victor Stinner8c62be82010-05-06 00:08:46 +000010771 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010772#endif
Neal Norwitz11690112002-07-30 01:08:28 +000010773#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Victor Stinner8c62be82010-05-06 00:08:46 +000010774 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
Martin v. Löwis06a83e92002-04-14 10:19:44 +000010775#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +000010776#ifdef HAVE_DEVICE_MACROS
Victor Stinner8c62be82010-05-06 00:08:46 +000010777 {"major", posix_major, METH_VARARGS, posix_major__doc__},
10778 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
10779 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
Martin v. Löwisdbe3f762002-10-10 14:27:30 +000010780#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010781#ifdef HAVE_FTRUNCATE
Victor Stinner8c62be82010-05-06 00:08:46 +000010782 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +000010783#endif
Ross Lagerwall7807c352011-03-17 20:20:30 +020010784#ifdef HAVE_TRUNCATE
10785 {"truncate", posix_truncate, METH_VARARGS, posix_truncate__doc__},
10786#endif
10787#ifdef HAVE_POSIX_FALLOCATE
10788 {"posix_fallocate", posix_posix_fallocate, METH_VARARGS, posix_posix_fallocate__doc__},
10789#endif
10790#ifdef HAVE_POSIX_FADVISE
10791 {"posix_fadvise", posix_posix_fadvise, METH_VARARGS, posix_posix_fadvise__doc__},
10792#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +000010793#ifdef HAVE_PUTENV
Victor Stinner8c62be82010-05-06 00:08:46 +000010794 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +000010795#endif
Guido van Rossumc524d952001-10-19 01:31:59 +000010796#ifdef HAVE_UNSETENV
Victor Stinner8c62be82010-05-06 00:08:46 +000010797 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
Guido van Rossumc524d952001-10-19 01:31:59 +000010798#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010799 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +000010800#ifdef HAVE_FCHDIR
Victor Stinner8c62be82010-05-06 00:08:46 +000010801 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +000010802#endif
Guido van Rossum21142a01999-01-08 21:05:37 +000010803#ifdef HAVE_FSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000010804 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +000010805#endif
Ross Lagerwall7807c352011-03-17 20:20:30 +020010806#ifdef HAVE_SYNC
10807 {"sync", posix_sync, METH_NOARGS, posix_sync__doc__},
10808#endif
Guido van Rossum21142a01999-01-08 21:05:37 +000010809#ifdef HAVE_FDATASYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000010810 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +000010811#endif
Guido van Rossumc9641791998-08-04 15:26:23 +000010812#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +000010813#ifdef WCOREDUMP
Victor Stinner8c62be82010-05-06 00:08:46 +000010814 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
Fred Drake106c1a02002-04-23 15:58:02 +000010815#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +000010816#ifdef WIFCONTINUED
Victor Stinner8c62be82010-05-06 00:08:46 +000010817 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +000010818#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +000010819#ifdef WIFSTOPPED
Victor Stinner8c62be82010-05-06 00:08:46 +000010820 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010821#endif /* WIFSTOPPED */
10822#ifdef WIFSIGNALED
Victor Stinner8c62be82010-05-06 00:08:46 +000010823 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010824#endif /* WIFSIGNALED */
10825#ifdef WIFEXITED
Victor Stinner8c62be82010-05-06 00:08:46 +000010826 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010827#endif /* WIFEXITED */
10828#ifdef WEXITSTATUS
Victor Stinner8c62be82010-05-06 00:08:46 +000010829 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010830#endif /* WEXITSTATUS */
10831#ifdef WTERMSIG
Victor Stinner8c62be82010-05-06 00:08:46 +000010832 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010833#endif /* WTERMSIG */
10834#ifdef WSTOPSIG
Victor Stinner8c62be82010-05-06 00:08:46 +000010835 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +000010836#endif /* WSTOPSIG */
10837#endif /* HAVE_SYS_WAIT_H */
Thomas Wouters477c8d52006-05-27 19:21:47 +000010838#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Victor Stinner8c62be82010-05-06 00:08:46 +000010839 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +000010840#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +000010841#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Victor Stinner8c62be82010-05-06 00:08:46 +000010842 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +000010843#endif
Fred Drakec9680921999-12-13 16:37:25 +000010844#ifdef HAVE_CONFSTR
Victor Stinner8c62be82010-05-06 00:08:46 +000010845 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010846#endif
10847#ifdef HAVE_SYSCONF
Victor Stinner8c62be82010-05-06 00:08:46 +000010848 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010849#endif
10850#ifdef HAVE_FPATHCONF
Victor Stinner8c62be82010-05-06 00:08:46 +000010851 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010852#endif
10853#ifdef HAVE_PATHCONF
Victor Stinner8c62be82010-05-06 00:08:46 +000010854 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +000010855#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010856 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000010857#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +000010858 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
Brian Curtind40e6f72010-07-08 21:39:08 +000010859 {"_getfinalpathname", posix__getfinalpathname, METH_VARARGS, NULL},
Brian Curtin62857742010-09-06 17:07:27 +000010860 {"_getfileinformation", posix__getfileinformation, METH_VARARGS, NULL},
Brian Curtin95d028f2011-06-09 09:10:38 -050010861 {"_isdir", posix__isdir, METH_VARARGS, posix__isdir__doc__},
Giampaolo Rodola'210e7ca2011-07-01 13:55:36 +020010862 {"_getdiskusage", win32__getdiskusage, METH_VARARGS, win32__getdiskusage__doc__},
Mark Hammondef8b6542001-05-13 08:04:26 +000010863#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +000010864#ifdef HAVE_GETLOADAVG
Victor Stinner8c62be82010-05-06 00:08:46 +000010865 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +000010866#endif
Martin v. Löwisdc3883f2004-08-29 15:46:35 +000010867 #ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +000010868 {"urandom", win32_urandom, METH_VARARGS, win32_urandom__doc__},
Martin v. Löwisdc3883f2004-08-29 15:46:35 +000010869 #endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +000010870 #ifdef __VMS
Victor Stinner8c62be82010-05-06 00:08:46 +000010871 {"urandom", vms_urandom, METH_VARARGS, vms_urandom__doc__},
Thomas Wouters0e3f5912006-08-11 14:57:12 +000010872 #endif
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010873#ifdef HAVE_SETRESUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010874 {"setresuid", posix_setresuid, METH_VARARGS, posix_setresuid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010875#endif
10876#ifdef HAVE_SETRESGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010877 {"setresgid", posix_setresgid, METH_VARARGS, posix_setresgid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010878#endif
10879#ifdef HAVE_GETRESUID
Victor Stinner8c62be82010-05-06 00:08:46 +000010880 {"getresuid", posix_getresuid, METH_NOARGS, posix_getresuid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010881#endif
10882#ifdef HAVE_GETRESGID
Victor Stinner8c62be82010-05-06 00:08:46 +000010883 {"getresgid", posix_getresgid, METH_NOARGS, posix_getresgid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000010884#endif
10885
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010886/* posix *at family of functions */
10887#ifdef HAVE_FACCESSAT
10888 {"faccessat", posix_faccessat, METH_VARARGS, posix_faccessat__doc__},
10889#endif
10890#ifdef HAVE_FCHMODAT
10891 {"fchmodat", posix_fchmodat, METH_VARARGS, posix_fchmodat__doc__},
10892#endif /* HAVE_FCHMODAT */
10893#ifdef HAVE_FCHOWNAT
10894 {"fchownat", posix_fchownat, METH_VARARGS, posix_fchownat__doc__},
10895#endif /* HAVE_FCHOWNAT */
10896#ifdef HAVE_FSTATAT
Victor Stinner4195b5c2012-02-08 23:03:19 +010010897 {"fstatat", posix_fstatat, METH_VARARGS, posix_fstatat__doc__},
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010898#endif
10899#ifdef HAVE_FUTIMESAT
10900 {"futimesat", posix_futimesat, METH_VARARGS, posix_futimesat__doc__},
10901#endif
10902#ifdef HAVE_LINKAT
10903 {"linkat", posix_linkat, METH_VARARGS, posix_linkat__doc__},
10904#endif /* HAVE_LINKAT */
10905#ifdef HAVE_MKDIRAT
10906 {"mkdirat", posix_mkdirat, METH_VARARGS, posix_mkdirat__doc__},
10907#endif
10908#if defined(HAVE_MKNODAT) && defined(HAVE_MAKEDEV)
10909 {"mknodat", posix_mknodat, METH_VARARGS, posix_mknodat__doc__},
10910#endif
10911#ifdef HAVE_OPENAT
10912 {"openat", posix_openat, METH_VARARGS, posix_openat__doc__},
10913#endif
10914#ifdef HAVE_READLINKAT
10915 {"readlinkat", posix_readlinkat, METH_VARARGS, posix_readlinkat__doc__},
10916#endif /* HAVE_READLINKAT */
10917#ifdef HAVE_RENAMEAT
10918 {"renameat", posix_renameat, METH_VARARGS, posix_renameat__doc__},
10919#endif
10920#if HAVE_SYMLINKAT
10921 {"symlinkat", posix_symlinkat, METH_VARARGS, posix_symlinkat__doc__},
10922#endif /* HAVE_SYMLINKAT */
10923#ifdef HAVE_UNLINKAT
10924 {"unlinkat", posix_unlinkat, METH_VARARGS, posix_unlinkat__doc__},
10925#endif
10926#ifdef HAVE_UTIMENSAT
Jesus Cead03a4912011-11-08 17:28:04 +010010927 {"utimensat", (PyCFunction)posix_utimensat,
10928 METH_VARARGS | METH_KEYWORDS,
Brian Curtin569b4942011-11-07 16:09:20 -060010929 posix_utimensat__doc__},
Antoine Pitrouf65132d2011-02-25 23:25:17 +000010930#endif
10931#ifdef HAVE_MKFIFOAT
10932 {"mkfifoat", posix_mkfifoat, METH_VARARGS, posix_mkfifoat__doc__},
10933#endif
Benjamin Peterson9428d532011-09-14 11:45:52 -040010934#ifdef USE_XATTRS
Benjamin Peterson799bd802011-08-31 22:15:17 -040010935 {"setxattr", posix_setxattr, METH_VARARGS, posix_setxattr__doc__},
10936 {"lsetxattr", posix_lsetxattr, METH_VARARGS, posix_lsetxattr__doc__},
10937 {"fsetxattr", posix_fsetxattr, METH_VARARGS, posix_fsetxattr__doc__},
10938 {"getxattr", posix_getxattr, METH_VARARGS, posix_getxattr__doc__},
10939 {"lgetxattr", posix_lgetxattr, METH_VARARGS, posix_lgetxattr__doc__},
10940 {"fgetxattr", posix_fgetxattr, METH_VARARGS, posix_fgetxattr__doc__},
10941 {"removexattr", posix_removexattr, METH_VARARGS, posix_removexattr__doc__},
10942 {"lremovexattr", posix_lremovexattr, METH_VARARGS, posix_lremovexattr__doc__},
10943 {"fremovexattr", posix_fremovexattr, METH_VARARGS, posix_fremovexattr__doc__},
10944 {"listxattr", posix_listxattr, METH_VARARGS, posix_listxattr__doc__},
10945 {"llistxattr", posix_llistxattr, METH_VARARGS, posix_llistxattr__doc__},
10946 {"flistxattr", posix_flistxattr, METH_VARARGS, posix_flistxattr__doc__},
10947#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010948 {NULL, NULL} /* Sentinel */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010949};
10950
10951
Barry Warsaw4a342091996-12-19 23:50:02 +000010952static int
Fred Drake4d1e64b2002-04-15 19:40:07 +000010953ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +000010954{
Victor Stinner8c62be82010-05-06 00:08:46 +000010955 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +000010956}
10957
Guido van Rossumd48f2521997-12-05 22:19:34 +000010958#if defined(PYOS_OS2)
10959/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +000010960static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +000010961{
10962 APIRET rc;
10963 ULONG values[QSV_MAX+1];
10964 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +000010965 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +000010966
10967 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +000010968 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +000010969 Py_END_ALLOW_THREADS
10970
10971 if (rc != NO_ERROR) {
10972 os2_error(rc);
10973 return -1;
10974 }
10975
Fred Drake4d1e64b2002-04-15 19:40:07 +000010976 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
10977 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
10978 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
10979 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
10980 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
10981 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
10982 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000010983
10984 switch (values[QSV_VERSION_MINOR]) {
10985 case 0: ver = "2.00"; break;
10986 case 10: ver = "2.10"; break;
10987 case 11: ver = "2.11"; break;
10988 case 30: ver = "3.00"; break;
10989 case 40: ver = "4.00"; break;
10990 case 50: ver = "5.00"; break;
10991 default:
Tim Peters885d4572001-11-28 20:27:42 +000010992 PyOS_snprintf(tmp, sizeof(tmp),
Victor Stinner8c62be82010-05-06 00:08:46 +000010993 "%d-%d", values[QSV_VERSION_MAJOR],
Tim Peters885d4572001-11-28 20:27:42 +000010994 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +000010995 ver = &tmp[0];
10996 }
10997
10998 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +000010999 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +000011000 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000011001
11002 /* Add Indicator of Which Drive was Used to Boot the System */
11003 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
11004 tmp[1] = ':';
11005 tmp[2] = '\0';
11006
Fred Drake4d1e64b2002-04-15 19:40:07 +000011007 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +000011008}
11009#endif
11010
Brian Curtin52173d42010-12-02 18:29:18 +000011011#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000011012static int
Brian Curtin52173d42010-12-02 18:29:18 +000011013enable_symlink()
11014{
11015 HANDLE tok;
11016 TOKEN_PRIVILEGES tok_priv;
11017 LUID luid;
11018 int meth_idx = 0;
11019
11020 if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &tok))
Brian Curtin3b4499c2010-12-28 14:31:47 +000011021 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +000011022
11023 if (!LookupPrivilegeValue(NULL, SE_CREATE_SYMBOLIC_LINK_NAME, &luid))
Brian Curtin3b4499c2010-12-28 14:31:47 +000011024 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +000011025
11026 tok_priv.PrivilegeCount = 1;
11027 tok_priv.Privileges[0].Luid = luid;
11028 tok_priv.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
11029
11030 if (!AdjustTokenPrivileges(tok, FALSE, &tok_priv,
11031 sizeof(TOKEN_PRIVILEGES),
11032 (PTOKEN_PRIVILEGES) NULL, (PDWORD) NULL))
Brian Curtin3b4499c2010-12-28 14:31:47 +000011033 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +000011034
Brian Curtin3b4499c2010-12-28 14:31:47 +000011035 /* ERROR_NOT_ALL_ASSIGNED returned when the privilege can't be assigned. */
11036 return GetLastError() == ERROR_NOT_ALL_ASSIGNED ? 0 : 1;
Brian Curtin52173d42010-12-02 18:29:18 +000011037}
11038#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
11039
Barry Warsaw4a342091996-12-19 23:50:02 +000011040static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +000011041all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +000011042{
Guido van Rossum94f6f721999-01-06 18:42:14 +000011043#ifdef F_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011044 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011045#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +000011046#ifdef R_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011047 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011048#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +000011049#ifdef W_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011050 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011051#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +000011052#ifdef X_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011053 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011054#endif
Fred Drakec9680921999-12-13 16:37:25 +000011055#ifdef NGROUPS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +000011056 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
Fred Drakec9680921999-12-13 16:37:25 +000011057#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +000011058#ifdef TMP_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +000011059 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
Fred Drake5ab8eaf1999-12-09 21:13:07 +000011060#endif
Fred Drake106c1a02002-04-23 15:58:02 +000011061#ifdef WCONTINUED
Victor Stinner8c62be82010-05-06 00:08:46 +000011062 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
Fred Drake106c1a02002-04-23 15:58:02 +000011063#endif
Barry Warsaw4a342091996-12-19 23:50:02 +000011064#ifdef WNOHANG
Victor Stinner8c62be82010-05-06 00:08:46 +000011065 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011066#endif
Fred Drake106c1a02002-04-23 15:58:02 +000011067#ifdef WUNTRACED
Victor Stinner8c62be82010-05-06 00:08:46 +000011068 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
Fred Drake106c1a02002-04-23 15:58:02 +000011069#endif
Barry Warsaw4a342091996-12-19 23:50:02 +000011070#ifdef O_RDONLY
Victor Stinner8c62be82010-05-06 00:08:46 +000011071 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011072#endif
11073#ifdef O_WRONLY
Victor Stinner8c62be82010-05-06 00:08:46 +000011074 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011075#endif
11076#ifdef O_RDWR
Victor Stinner8c62be82010-05-06 00:08:46 +000011077 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011078#endif
11079#ifdef O_NDELAY
Victor Stinner8c62be82010-05-06 00:08:46 +000011080 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011081#endif
11082#ifdef O_NONBLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000011083 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011084#endif
11085#ifdef O_APPEND
Victor Stinner8c62be82010-05-06 00:08:46 +000011086 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011087#endif
11088#ifdef O_DSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011089 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011090#endif
11091#ifdef O_RSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011092 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011093#endif
11094#ifdef O_SYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011095 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011096#endif
11097#ifdef O_NOCTTY
Victor Stinner8c62be82010-05-06 00:08:46 +000011098 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011099#endif
11100#ifdef O_CREAT
Victor Stinner8c62be82010-05-06 00:08:46 +000011101 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011102#endif
11103#ifdef O_EXCL
Victor Stinner8c62be82010-05-06 00:08:46 +000011104 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011105#endif
11106#ifdef O_TRUNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011107 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +000011108#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +000011109#ifdef O_BINARY
Victor Stinner8c62be82010-05-06 00:08:46 +000011110 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
Guido van Rossum98d9d091997-08-08 21:48:51 +000011111#endif
11112#ifdef O_TEXT
Victor Stinner8c62be82010-05-06 00:08:46 +000011113 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
Guido van Rossum98d9d091997-08-08 21:48:51 +000011114#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011115#ifdef O_LARGEFILE
Victor Stinner8c62be82010-05-06 00:08:46 +000011116 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011117#endif
Skip Montanaro5ff14922005-05-16 02:42:22 +000011118#ifdef O_SHLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000011119 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
Skip Montanaro5ff14922005-05-16 02:42:22 +000011120#endif
11121#ifdef O_EXLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +000011122 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
Skip Montanaro5ff14922005-05-16 02:42:22 +000011123#endif
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +000011124#ifdef PRIO_PROCESS
11125 if (ins(d, "PRIO_PROCESS", (long)PRIO_PROCESS)) return -1;
11126#endif
11127#ifdef PRIO_PGRP
11128 if (ins(d, "PRIO_PGRP", (long)PRIO_PGRP)) return -1;
11129#endif
11130#ifdef PRIO_USER
11131 if (ins(d, "PRIO_USER", (long)PRIO_USER)) return -1;
11132#endif
Charles-François Natali1e045b12011-05-22 20:42:32 +020011133#ifdef O_CLOEXEC
11134 if (ins(d, "O_CLOEXEC", (long)O_CLOEXEC)) return -1;
11135#endif
Antoine Pitrouf65132d2011-02-25 23:25:17 +000011136/* posix - constants for *at functions */
11137#ifdef AT_SYMLINK_NOFOLLOW
11138 if (ins(d, "AT_SYMLINK_NOFOLLOW", (long)AT_SYMLINK_NOFOLLOW)) return -1;
11139#endif
11140#ifdef AT_EACCESS
11141 if (ins(d, "AT_EACCESS", (long)AT_EACCESS)) return -1;
11142#endif
11143#ifdef AT_FDCWD
11144 if (ins(d, "AT_FDCWD", (long)AT_FDCWD)) return -1;
11145#endif
11146#ifdef AT_REMOVEDIR
11147 if (ins(d, "AT_REMOVEDIR", (long)AT_REMOVEDIR)) return -1;
11148#endif
11149#ifdef AT_SYMLINK_FOLLOW
11150 if (ins(d, "AT_SYMLINK_FOLLOW", (long)AT_SYMLINK_FOLLOW)) return -1;
11151#endif
11152#ifdef UTIME_NOW
11153 if (ins(d, "UTIME_NOW", (long)UTIME_NOW)) return -1;
11154#endif
11155#ifdef UTIME_OMIT
11156 if (ins(d, "UTIME_OMIT", (long)UTIME_OMIT)) return -1;
11157#endif
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +000011158
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011159
Tim Peters5aa91602002-01-30 05:46:57 +000011160/* MS Windows */
11161#ifdef O_NOINHERIT
Victor Stinner8c62be82010-05-06 00:08:46 +000011162 /* Don't inherit in child processes. */
11163 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011164#endif
11165#ifdef _O_SHORT_LIVED
Victor Stinner8c62be82010-05-06 00:08:46 +000011166 /* Optimize for short life (keep in memory). */
11167 /* MS forgot to define this one with a non-underscore form too. */
11168 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011169#endif
11170#ifdef O_TEMPORARY
Victor Stinner8c62be82010-05-06 00:08:46 +000011171 /* Automatically delete when last handle is closed. */
11172 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011173#endif
11174#ifdef O_RANDOM
Victor Stinner8c62be82010-05-06 00:08:46 +000011175 /* Optimize for random access. */
11176 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011177#endif
11178#ifdef O_SEQUENTIAL
Victor Stinner8c62be82010-05-06 00:08:46 +000011179 /* Optimize for sequential access. */
11180 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +000011181#endif
11182
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011183/* GNU extensions. */
Alexandre Vassalottibee32532008-05-16 18:15:12 +000011184#ifdef O_ASYNC
Victor Stinner8c62be82010-05-06 00:08:46 +000011185 /* Send a SIGIO signal whenever input or output
11186 becomes available on file descriptor */
11187 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
Alexandre Vassalottibee32532008-05-16 18:15:12 +000011188#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011189#ifdef O_DIRECT
Victor Stinner8c62be82010-05-06 00:08:46 +000011190 /* Direct disk access. */
11191 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011192#endif
11193#ifdef O_DIRECTORY
Victor Stinner8c62be82010-05-06 00:08:46 +000011194 /* Must be a directory. */
11195 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011196#endif
11197#ifdef O_NOFOLLOW
Victor Stinner8c62be82010-05-06 00:08:46 +000011198 /* Do not follow links. */
11199 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +000011200#endif
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +000011201#ifdef O_NOATIME
Victor Stinner8c62be82010-05-06 00:08:46 +000011202 /* Do not update the access time. */
11203 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +000011204#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +000011205
Victor Stinner8c62be82010-05-06 00:08:46 +000011206 /* These come from sysexits.h */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011207#ifdef EX_OK
Victor Stinner8c62be82010-05-06 00:08:46 +000011208 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011209#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011210#ifdef EX_USAGE
Victor Stinner8c62be82010-05-06 00:08:46 +000011211 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011212#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011213#ifdef EX_DATAERR
Victor Stinner8c62be82010-05-06 00:08:46 +000011214 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011215#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011216#ifdef EX_NOINPUT
Victor Stinner8c62be82010-05-06 00:08:46 +000011217 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011218#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011219#ifdef EX_NOUSER
Victor Stinner8c62be82010-05-06 00:08:46 +000011220 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011221#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011222#ifdef EX_NOHOST
Victor Stinner8c62be82010-05-06 00:08:46 +000011223 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011224#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011225#ifdef EX_UNAVAILABLE
Victor Stinner8c62be82010-05-06 00:08:46 +000011226 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011227#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011228#ifdef EX_SOFTWARE
Victor Stinner8c62be82010-05-06 00:08:46 +000011229 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011230#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011231#ifdef EX_OSERR
Victor Stinner8c62be82010-05-06 00:08:46 +000011232 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011233#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011234#ifdef EX_OSFILE
Victor Stinner8c62be82010-05-06 00:08:46 +000011235 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011236#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011237#ifdef EX_CANTCREAT
Victor Stinner8c62be82010-05-06 00:08:46 +000011238 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011239#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011240#ifdef EX_IOERR
Victor Stinner8c62be82010-05-06 00:08:46 +000011241 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011242#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011243#ifdef EX_TEMPFAIL
Victor Stinner8c62be82010-05-06 00:08:46 +000011244 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011245#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011246#ifdef EX_PROTOCOL
Victor Stinner8c62be82010-05-06 00:08:46 +000011247 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011248#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011249#ifdef EX_NOPERM
Victor Stinner8c62be82010-05-06 00:08:46 +000011250 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011251#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011252#ifdef EX_CONFIG
Victor Stinner8c62be82010-05-06 00:08:46 +000011253 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011254#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011255#ifdef EX_NOTFOUND
Victor Stinner8c62be82010-05-06 00:08:46 +000011256 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +000011257#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +000011258
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +000011259 /* statvfs */
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +000011260#ifdef ST_RDONLY
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +000011261 if (ins(d, "ST_RDONLY", (long)ST_RDONLY)) return -1;
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +000011262#endif /* ST_RDONLY */
11263#ifdef ST_NOSUID
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +000011264 if (ins(d, "ST_NOSUID", (long)ST_NOSUID)) return -1;
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +000011265#endif /* ST_NOSUID */
11266
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +000011267 /* FreeBSD sendfile() constants */
11268#ifdef SF_NODISKIO
11269 if (ins(d, "SF_NODISKIO", (long)SF_NODISKIO)) return -1;
11270#endif
11271#ifdef SF_MNOWAIT
11272 if (ins(d, "SF_MNOWAIT", (long)SF_MNOWAIT)) return -1;
11273#endif
11274#ifdef SF_SYNC
11275 if (ins(d, "SF_SYNC", (long)SF_SYNC)) return -1;
11276#endif
11277
Ross Lagerwall7807c352011-03-17 20:20:30 +020011278 /* constants for posix_fadvise */
11279#ifdef POSIX_FADV_NORMAL
11280 if (ins(d, "POSIX_FADV_NORMAL", (long)POSIX_FADV_NORMAL)) return -1;
11281#endif
11282#ifdef POSIX_FADV_SEQUENTIAL
11283 if (ins(d, "POSIX_FADV_SEQUENTIAL", (long)POSIX_FADV_SEQUENTIAL)) return -1;
11284#endif
11285#ifdef POSIX_FADV_RANDOM
11286 if (ins(d, "POSIX_FADV_RANDOM", (long)POSIX_FADV_RANDOM)) return -1;
11287#endif
11288#ifdef POSIX_FADV_NOREUSE
11289 if (ins(d, "POSIX_FADV_NOREUSE", (long)POSIX_FADV_NOREUSE)) return -1;
11290#endif
11291#ifdef POSIX_FADV_WILLNEED
11292 if (ins(d, "POSIX_FADV_WILLNEED", (long)POSIX_FADV_WILLNEED)) return -1;
11293#endif
11294#ifdef POSIX_FADV_DONTNEED
11295 if (ins(d, "POSIX_FADV_DONTNEED", (long)POSIX_FADV_DONTNEED)) return -1;
11296#endif
11297
11298 /* constants for waitid */
11299#if defined(HAVE_SYS_WAIT_H) && defined(HAVE_WAITID)
11300 if (ins(d, "P_PID", (long)P_PID)) return -1;
11301 if (ins(d, "P_PGID", (long)P_PGID)) return -1;
11302 if (ins(d, "P_ALL", (long)P_ALL)) return -1;
11303#endif
11304#ifdef WEXITED
11305 if (ins(d, "WEXITED", (long)WEXITED)) return -1;
11306#endif
11307#ifdef WNOWAIT
11308 if (ins(d, "WNOWAIT", (long)WNOWAIT)) return -1;
11309#endif
11310#ifdef WSTOPPED
11311 if (ins(d, "WSTOPPED", (long)WSTOPPED)) return -1;
11312#endif
11313#ifdef CLD_EXITED
11314 if (ins(d, "CLD_EXITED", (long)CLD_EXITED)) return -1;
11315#endif
11316#ifdef CLD_DUMPED
11317 if (ins(d, "CLD_DUMPED", (long)CLD_DUMPED)) return -1;
11318#endif
11319#ifdef CLD_TRAPPED
11320 if (ins(d, "CLD_TRAPPED", (long)CLD_TRAPPED)) return -1;
11321#endif
11322#ifdef CLD_CONTINUED
11323 if (ins(d, "CLD_CONTINUED", (long)CLD_CONTINUED)) return -1;
11324#endif
11325
11326 /* constants for lockf */
11327#ifdef F_LOCK
11328 if (ins(d, "F_LOCK", (long)F_LOCK)) return -1;
11329#endif
11330#ifdef F_TLOCK
11331 if (ins(d, "F_TLOCK", (long)F_TLOCK)) return -1;
11332#endif
11333#ifdef F_ULOCK
11334 if (ins(d, "F_ULOCK", (long)F_ULOCK)) return -1;
11335#endif
11336#ifdef F_TEST
11337 if (ins(d, "F_TEST", (long)F_TEST)) return -1;
11338#endif
11339
11340 /* constants for futimens */
11341#ifdef UTIME_NOW
11342 if (ins(d, "UTIME_NOW", (long)UTIME_NOW)) return -1;
11343#endif
11344#ifdef UTIME_OMIT
11345 if (ins(d, "UTIME_OMIT", (long)UTIME_OMIT)) return -1;
11346#endif
11347
Guido van Rossum246bc171999-02-01 23:54:31 +000011348#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000011349#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +000011350 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
11351 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
11352 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
11353 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
11354 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
11355 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
11356 if (ins(d, "P_PM", (long)P_PM)) return -1;
11357 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
11358 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
11359 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
11360 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
11361 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
11362 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
11363 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
11364 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
11365 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
11366 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
11367 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
11368 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
11369 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000011370#else
Victor Stinner8c62be82010-05-06 00:08:46 +000011371 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
11372 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
11373 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
11374 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
11375 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +000011376#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000011377#endif
Guido van Rossum246bc171999-02-01 23:54:31 +000011378
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011379#ifdef HAVE_SCHED_H
Jesus Ceaf2cb4e82011-09-09 23:55:42 +020011380 if (ins(d, "SCHED_OTHER", (long)SCHED_OTHER)) return -1;
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011381 if (ins(d, "SCHED_FIFO", (long)SCHED_FIFO)) return -1;
11382 if (ins(d, "SCHED_RR", (long)SCHED_RR)) return -1;
11383#ifdef SCHED_SPORADIC
11384 if (ins(d, "SCHED_SPORADIC", (long)SCHED_SPORADIC) return -1;
11385#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011386#ifdef SCHED_BATCH
11387 if (ins(d, "SCHED_BATCH", (long)SCHED_BATCH)) return -1;
11388#endif
11389#ifdef SCHED_IDLE
11390 if (ins(d, "SCHED_IDLE", (long)SCHED_IDLE)) return -1;
11391#endif
11392#ifdef SCHED_RESET_ON_FORK
11393 if (ins(d, "SCHED_RESET_ON_FORK", (long)SCHED_RESET_ON_FORK)) return -1;
11394#endif
Jesus Ceaf2cb4e82011-09-09 23:55:42 +020011395#ifdef SCHED_SYS
11396 if (ins(d, "SCHED_SYS", (long)SCHED_SYS)) return -1;
11397#endif
11398#ifdef SCHED_IA
11399 if (ins(d, "SCHED_IA", (long)SCHED_IA)) return -1;
11400#endif
11401#ifdef SCHED_FSS
11402 if (ins(d, "SCHED_FSS", (long)SCHED_FSS)) return -1;
11403#endif
11404#ifdef SCHED_FX
11405 if (ins(d, "SCHED_FX", (long)SCHED_FSS)) return -1;
11406#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011407#endif
11408
Benjamin Peterson9428d532011-09-14 11:45:52 -040011409#ifdef USE_XATTRS
Benjamin Peterson799bd802011-08-31 22:15:17 -040011410 if (ins(d, "XATTR_CREATE", (long)XATTR_CREATE)) return -1;
11411 if (ins(d, "XATTR_REPLACE", (long)XATTR_REPLACE)) return -1;
11412 if (ins(d, "XATTR_SIZE_MAX", (long)XATTR_SIZE_MAX)) return -1;
11413#endif
11414
Victor Stinner8b905bd2011-10-25 13:34:04 +020011415#ifdef RTLD_LAZY
11416 if (PyModule_AddIntMacro(d, RTLD_LAZY)) return -1;
11417#endif
11418#ifdef RTLD_NOW
11419 if (PyModule_AddIntMacro(d, RTLD_NOW)) return -1;
11420#endif
11421#ifdef RTLD_GLOBAL
11422 if (PyModule_AddIntMacro(d, RTLD_GLOBAL)) return -1;
11423#endif
11424#ifdef RTLD_LOCAL
11425 if (PyModule_AddIntMacro(d, RTLD_LOCAL)) return -1;
11426#endif
11427#ifdef RTLD_NODELETE
11428 if (PyModule_AddIntMacro(d, RTLD_NODELETE)) return -1;
11429#endif
11430#ifdef RTLD_NOLOAD
11431 if (PyModule_AddIntMacro(d, RTLD_NOLOAD)) return -1;
11432#endif
11433#ifdef RTLD_DEEPBIND
11434 if (PyModule_AddIntMacro(d, RTLD_DEEPBIND)) return -1;
11435#endif
11436
Guido van Rossumd48f2521997-12-05 22:19:34 +000011437#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +000011438 if (insertvalues(d)) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000011439#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011440 return 0;
Barry Warsaw4a342091996-12-19 23:50:02 +000011441}
11442
11443
Tim Peters5aa91602002-01-30 05:46:57 +000011444#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Martin v. Löwis1a214512008-06-11 05:26:20 +000011445#define INITFUNC PyInit_nt
Guido van Rossum0cb96de1997-10-01 04:29:29 +000011446#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +000011447
11448#elif defined(PYOS_OS2)
Martin v. Löwis1a214512008-06-11 05:26:20 +000011449#define INITFUNC PyInit_os2
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000011450#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +000011451
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000011452#else
Martin v. Löwis1a214512008-06-11 05:26:20 +000011453#define INITFUNC PyInit_posix
Guido van Rossum0cb96de1997-10-01 04:29:29 +000011454#define MODNAME "posix"
11455#endif
11456
Martin v. Löwis1a214512008-06-11 05:26:20 +000011457static struct PyModuleDef posixmodule = {
Victor Stinner8c62be82010-05-06 00:08:46 +000011458 PyModuleDef_HEAD_INIT,
11459 MODNAME,
11460 posix__doc__,
11461 -1,
11462 posix_methods,
11463 NULL,
11464 NULL,
11465 NULL,
11466 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +000011467};
11468
11469
Mark Hammondfe51c6d2002-08-02 02:27:13 +000011470PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +000011471INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +000011472{
Victor Stinner8c62be82010-05-06 00:08:46 +000011473 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +000011474
Brian Curtin52173d42010-12-02 18:29:18 +000011475#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000011476 win32_can_symlink = enable_symlink();
Brian Curtin52173d42010-12-02 18:29:18 +000011477#endif
11478
Victor Stinner8c62be82010-05-06 00:08:46 +000011479 m = PyModule_Create(&posixmodule);
11480 if (m == NULL)
11481 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +000011482
Victor Stinner8c62be82010-05-06 00:08:46 +000011483 /* Initialize environ dictionary */
11484 v = convertenviron();
11485 Py_XINCREF(v);
11486 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
11487 return NULL;
11488 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +000011489
Victor Stinner8c62be82010-05-06 00:08:46 +000011490 if (all_ins(m))
11491 return NULL;
Barry Warsaw4a342091996-12-19 23:50:02 +000011492
Victor Stinner8c62be82010-05-06 00:08:46 +000011493 if (setup_confname_tables(m))
11494 return NULL;
Fred Drakebec628d1999-12-15 18:31:10 +000011495
Victor Stinner8c62be82010-05-06 00:08:46 +000011496 Py_INCREF(PyExc_OSError);
11497 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +000011498
Benjamin Peterson2740af82011-08-02 17:41:34 -050011499#ifdef HAVE_SCHED_SETAFFINITY
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011500 if (PyType_Ready(&cpu_set_type) < 0)
11501 return NULL;
11502 Py_INCREF(&cpu_set_type);
11503 PyModule_AddObject(m, "cpu_set", (PyObject *)&cpu_set_type);
Benjamin Peterson2740af82011-08-02 17:41:34 -050011504#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011505
Guido van Rossumb3d39562000-01-31 18:41:26 +000011506#ifdef HAVE_PUTENV
Victor Stinner8c62be82010-05-06 00:08:46 +000011507 if (posix_putenv_garbage == NULL)
11508 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +000011509#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +000011510
Victor Stinner8c62be82010-05-06 00:08:46 +000011511 if (!initialized) {
Ross Lagerwall7807c352011-03-17 20:20:30 +020011512#if defined(HAVE_WAITID) && !defined(__APPLE__)
11513 waitid_result_desc.name = MODNAME ".waitid_result";
11514 PyStructSequence_InitType(&WaitidResultType, &waitid_result_desc);
11515#endif
11516
Victor Stinner8c62be82010-05-06 00:08:46 +000011517 stat_result_desc.name = MODNAME ".stat_result";
11518 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
11519 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
11520 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
11521 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
11522 structseq_new = StatResultType.tp_new;
11523 StatResultType.tp_new = statresult_new;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011524
Victor Stinner8c62be82010-05-06 00:08:46 +000011525 statvfs_result_desc.name = MODNAME ".statvfs_result";
11526 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011527#ifdef NEED_TICKS_PER_SECOND
11528# if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
Victor Stinner8c62be82010-05-06 00:08:46 +000011529 ticks_per_second = sysconf(_SC_CLK_TCK);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011530# elif defined(HZ)
Victor Stinner8c62be82010-05-06 00:08:46 +000011531 ticks_per_second = HZ;
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011532# else
Victor Stinner8c62be82010-05-06 00:08:46 +000011533 ticks_per_second = 60; /* magic fallback value; may be bogus */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000011534# endif
11535#endif
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011536
Benjamin Peterson0163c9a2011-08-02 18:11:38 -050011537#if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER)
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011538 sched_param_desc.name = MODNAME ".sched_param";
11539 PyStructSequence_InitType(&SchedParamType, &sched_param_desc);
11540 SchedParamType.tp_new = sched_param_new;
11541#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011542 }
Ross Lagerwall7807c352011-03-17 20:20:30 +020011543#if defined(HAVE_WAITID) && !defined(__APPLE__)
11544 Py_INCREF((PyObject*) &WaitidResultType);
11545 PyModule_AddObject(m, "waitid_result", (PyObject*) &WaitidResultType);
11546#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011547 Py_INCREF((PyObject*) &StatResultType);
11548 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
11549 Py_INCREF((PyObject*) &StatVFSResultType);
11550 PyModule_AddObject(m, "statvfs_result",
11551 (PyObject*) &StatVFSResultType);
Benjamin Petersone3298dd2011-08-02 18:40:46 -050011552
11553#if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER)
Benjamin Peterson94b580d2011-08-02 17:30:04 -050011554 Py_INCREF(&SchedParamType);
11555 PyModule_AddObject(m, "sched_param", (PyObject *)&SchedParamType);
Benjamin Petersone3298dd2011-08-02 18:40:46 -050011556#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000011557 initialized = 1;
Thomas Wouters477c8d52006-05-27 19:21:47 +000011558
11559#ifdef __APPLE__
Victor Stinner8c62be82010-05-06 00:08:46 +000011560 /*
11561 * Step 2 of weak-linking support on Mac OS X.
11562 *
11563 * The code below removes functions that are not available on the
11564 * currently active platform.
11565 *
11566 * This block allow one to use a python binary that was build on
11567 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
11568 * OSX 10.4.
11569 */
Thomas Wouters477c8d52006-05-27 19:21:47 +000011570#ifdef HAVE_FSTATVFS
Victor Stinner8c62be82010-05-06 00:08:46 +000011571 if (fstatvfs == NULL) {
11572 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
11573 return NULL;
11574 }
11575 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011576#endif /* HAVE_FSTATVFS */
11577
11578#ifdef HAVE_STATVFS
Victor Stinner8c62be82010-05-06 00:08:46 +000011579 if (statvfs == NULL) {
11580 if (PyObject_DelAttrString(m, "statvfs") == -1) {
11581 return NULL;
11582 }
11583 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011584#endif /* HAVE_STATVFS */
11585
11586# ifdef HAVE_LCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000011587 if (lchown == NULL) {
11588 if (PyObject_DelAttrString(m, "lchown") == -1) {
11589 return NULL;
11590 }
11591 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011592#endif /* HAVE_LCHOWN */
11593
11594
11595#endif /* __APPLE__ */
Victor Stinner8c62be82010-05-06 00:08:46 +000011596 return m;
Thomas Wouters477c8d52006-05-27 19:21:47 +000011597
Guido van Rossumb6775db1994-08-01 11:34:53 +000011598}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011599
11600#ifdef __cplusplus
11601}
11602#endif