blob: f9b3a431673af64d5ccfbef79e0fe029d6c6801c [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* POSIX module implementation */
3
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004/* This file is also used for Windows NT/MS-Win and OS/2. In that case the
5 module actually calls itself 'nt' or 'os2', not 'posix', and a few
6 functions are either unimplemented or implemented differently. The source
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007 assumes that for Windows NT, the macro 'MS_WINDOWS' is defined independent
Guido van Rossumad0ee831995-03-01 10:34:45 +00008 of the compiler used. Different compilers define their own feature
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00009 test macro, e.g. '__BORLANDC__' or '_MSC_VER'. For OS/2, the compiler
10 independent macro PYOS_OS2 should be defined. On OS/2 the default
11 compiler is assumed to be IBM's VisualAge C++ (VACPP). PYCC_GCC is used
12 as the compiler specific macro for the EMX port of gcc to OS/2. */
Guido van Rossumad0ee831995-03-01 10:34:45 +000013
Guido van Rossuma4916fa1996-05-23 22:58:55 +000014/* See also ../Dos/dosmodule.c */
Guido van Rossumad0ee831995-03-01 10:34:45 +000015
Thomas Wouters477c8d52006-05-27 19:21:47 +000016#ifdef __APPLE__
17 /*
Victor Stinner8c62be82010-05-06 00:08:46 +000018 * Step 1 of support for weak-linking a number of symbols existing on
Thomas Wouters477c8d52006-05-27 19:21:47 +000019 * OSX 10.4 and later, see the comment in the #ifdef __APPLE__ block
20 * at the end of this file for more information.
21 */
22# pragma weak lchown
23# pragma weak statvfs
24# pragma weak fstatvfs
25
26#endif /* __APPLE__ */
27
Thomas Wouters68bc4f92006-03-01 01:05:10 +000028#define PY_SSIZE_T_CLEAN
29
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000030#include "Python.h"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000031
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000032#if defined(__VMS)
Victor Stinnerb90db4c2011-04-26 22:48:24 +020033# error "PEP 11: VMS is now unsupported, code will be removed in Python 3.4"
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000034# include <unixio.h>
Martin v. Löwis79acb9e2002-12-06 12:48:53 +000035#endif /* defined(__VMS) */
36
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000037#ifdef __cplusplus
38extern "C" {
39#endif
40
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000041PyDoc_STRVAR(posix__doc__,
Guido van Rossumec4f4ac1997-06-02 22:20:51 +000042"This module provides access to operating system functionality that is\n\
43standardized by the C Standard and the POSIX standard (a thinly\n\
44disguised Unix interface). Refer to the library manual and\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000045corresponding Unix manual entries for more information on calls.");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000046
Martin v. Löwis0073f2e2002-11-21 23:52:35 +000047
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000048#if defined(PYOS_OS2)
Victor Stinnerb90db4c2011-04-26 22:48:24 +020049#error "PEP 11: OS/2 is now unsupported, code will be removed in Python 3.4"
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000050#define INCL_DOS
51#define INCL_DOSERRORS
52#define INCL_DOSPROCESS
53#define INCL_NOPMAPI
54#include <os2.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000055#if defined(PYCC_GCC)
56#include <ctype.h>
57#include <io.h>
58#include <stdio.h>
59#include <process.h>
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000060#endif
Andrew MacIntyreda4d6cb2004-03-29 11:53:38 +000061#include "osdefs.h"
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000062#endif
63
Ross Lagerwall4d076da2011-03-18 06:56:53 +020064#ifdef HAVE_SYS_UIO_H
65#include <sys/uio.h>
66#endif
67
Thomas Wouters0e3f5912006-08-11 14:57:12 +000068#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000069#include <sys/types.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000070#endif /* HAVE_SYS_TYPES_H */
71
72#ifdef HAVE_SYS_STAT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000073#include <sys/stat.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000074#endif /* HAVE_SYS_STAT_H */
Guido van Rossuma6535fd2001-10-18 19:44:10 +000075
Guido van Rossum36bc6801995-06-14 22:54:23 +000076#ifdef HAVE_SYS_WAIT_H
Victor Stinner8c62be82010-05-06 00:08:46 +000077#include <sys/wait.h> /* For WNOHANG */
Guido van Rossum36bc6801995-06-14 22:54:23 +000078#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +000079
Thomas Wouters0e3f5912006-08-11 14:57:12 +000080#ifdef HAVE_SIGNAL_H
Guido van Rossuma376cc51996-12-05 23:43:35 +000081#include <signal.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000082#endif
Guido van Rossuma376cc51996-12-05 23:43:35 +000083
Guido van Rossumb6775db1994-08-01 11:34:53 +000084#ifdef HAVE_FCNTL_H
85#include <fcntl.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +000086#endif /* HAVE_FCNTL_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +000087
Guido van Rossuma6535fd2001-10-18 19:44:10 +000088#ifdef HAVE_GRP_H
89#include <grp.h>
90#endif
91
Barry Warsaw5676bd12003-01-07 20:57:09 +000092#ifdef HAVE_SYSEXITS_H
93#include <sysexits.h>
94#endif /* HAVE_SYSEXITS_H */
95
Anthony Baxter8a560de2004-10-13 15:30:56 +000096#ifdef HAVE_SYS_LOADAVG_H
97#include <sys/loadavg.h>
98#endif
99
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +0000100#ifdef HAVE_LANGINFO_H
101#include <langinfo.h>
102#endif
103
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000104#ifdef HAVE_SYS_SENDFILE_H
105#include <sys/sendfile.h>
106#endif
107
108#if defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__)
109#ifdef HAVE_SYS_SOCKET_H
110#include <sys/socket.h>
111#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000112#endif
113
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000114/* Various compilers have only certain posix functions */
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +0000115/* XXX Gosh I wish these were all moved into pyconfig.h */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000116#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000117#include <process.h>
118#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000119#if defined(__WATCOMC__) && !defined(__QNX__) /* Watcom compiler */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000120#define HAVE_GETCWD 1
121#define HAVE_OPENDIR 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000122#define HAVE_SYSTEM 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000123#if defined(__OS2__)
124#define HAVE_EXECV 1
125#define HAVE_WAIT 1
Guido van Rossumad0ee831995-03-01 10:34:45 +0000126#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000127#include <process.h>
128#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000129#ifdef __BORLANDC__ /* Borland compiler */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000130#define HAVE_EXECV 1
131#define HAVE_GETCWD 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000132#define HAVE_OPENDIR 1
133#define HAVE_PIPE 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000134#define HAVE_SYSTEM 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000135#define HAVE_WAIT 1
136#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000137#ifdef _MSC_VER /* Microsoft compiler */
Guido van Rossum8d665e61996-06-26 18:22:49 +0000138#define HAVE_GETCWD 1
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +0000139#define HAVE_GETPPID 1
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000140#define HAVE_GETLOGIN 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000141#define HAVE_SPAWNV 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000142#define HAVE_EXECV 1
143#define HAVE_PIPE 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000144#define HAVE_SYSTEM 1
145#define HAVE_CWAIT 1
146#define HAVE_FSYNC 1
Tim Peters11b23062003-04-23 02:39:17 +0000147#define fsync _commit
Andrew MacIntyre6c73af22002-03-03 03:07:07 +0000148#else
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000149#if defined(PYOS_OS2) && defined(PYCC_GCC) || defined(__VMS)
150/* Everything needed is defined in PC/os2emx/pyconfig.h or vms/pyconfig.h */
Victor Stinner8c62be82010-05-06 00:08:46 +0000151#else /* all other compilers */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000152/* Unix functions that the configure script doesn't check for */
153#define HAVE_EXECV 1
154#define HAVE_FORK 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000155#if defined(__USLC__) && defined(__SCO_VERSION__) /* SCO UDK Compiler */
Guido van Rossum2242f2f2001-04-11 20:58:20 +0000156#define HAVE_FORK1 1
157#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000158#define HAVE_GETCWD 1
159#define HAVE_GETEGID 1
160#define HAVE_GETEUID 1
161#define HAVE_GETGID 1
162#define HAVE_GETPPID 1
163#define HAVE_GETUID 1
164#define HAVE_KILL 1
165#define HAVE_OPENDIR 1
166#define HAVE_PIPE 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000167#define HAVE_SYSTEM 1
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000168#define HAVE_WAIT 1
Victor Stinner8c62be82010-05-06 00:08:46 +0000169#define HAVE_TTYNAME 1
Martin v. Löwis7a924e62003-03-05 14:15:21 +0000170#endif /* PYOS_OS2 && PYCC_GCC && __VMS */
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000171#endif /* _MSC_VER */
172#endif /* __BORLANDC__ */
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000173#endif /* ! __WATCOMC__ || __QNX__ */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000174#endif /* ! __IBMC__ */
Guido van Rossumad0ee831995-03-01 10:34:45 +0000175
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000176#ifndef _MSC_VER
Guido van Rossum36bc6801995-06-14 22:54:23 +0000177
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000178#if defined(__sgi)&&_COMPILER_VERSION>=700
179/* declare ctermid_r if compiling with MIPSPro 7.x in ANSI C mode
180 (default) */
181extern char *ctermid_r(char *);
182#endif
183
Thomas Wouters1e0c2f42000-07-24 16:06:23 +0000184#ifndef HAVE_UNISTD_H
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000185#if defined(PYCC_VACPP)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000186extern int mkdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000187#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000188#if ( defined(__WATCOMC__) || defined(_MSC_VER) ) && !defined(__QNX__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000189extern int mkdir(const char *);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000190#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000191extern int mkdir(const char *, mode_t);
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000192#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000193#endif
194#if defined(__IBMC__) || defined(__IBMCPP__)
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000195extern int chdir(char *);
196extern int rmdir(char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000197#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000198extern int chdir(const char *);
199extern int rmdir(const char *);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000200#endif
Tim Peters58e0a8c2001-05-14 22:32:33 +0000201#ifdef __BORLANDC__
202extern int chmod(const char *, int);
203#else
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000204extern int chmod(const char *, mode_t);
Tim Peters58e0a8c2001-05-14 22:32:33 +0000205#endif
Christian Heimes4e30a842007-11-30 22:12:06 +0000206/*#ifdef HAVE_FCHMOD
207extern int fchmod(int, mode_t);
208#endif*/
209/*#ifdef HAVE_LCHMOD
210extern int lchmod(const char *, mode_t);
211#endif*/
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000212extern int chown(const char *, uid_t, gid_t);
213extern char *getcwd(char *, int);
214extern char *strerror(int);
215extern int link(const char *, const char *);
216extern int rename(const char *, const char *);
217extern int stat(const char *, struct stat *);
218extern int unlink(const char *);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000219#ifdef HAVE_SYMLINK
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000220extern int symlink(const char *, const char *);
Guido van Rossuma38a5031995-02-17 15:11:36 +0000221#endif /* HAVE_SYMLINK */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000222#ifdef HAVE_LSTAT
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000223extern int lstat(const char *, struct stat *);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000224#endif /* HAVE_LSTAT */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000225#endif /* !HAVE_UNISTD_H */
Guido van Rossum36bc6801995-06-14 22:54:23 +0000226
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000227#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000228
Guido van Rossumb6775db1994-08-01 11:34:53 +0000229#ifdef HAVE_UTIME_H
230#include <utime.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000231#endif /* HAVE_UTIME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000232
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000233#ifdef HAVE_SYS_UTIME_H
234#include <sys/utime.h>
235#define HAVE_UTIME_H /* pretend we do for the rest of this file */
236#endif /* HAVE_SYS_UTIME_H */
237
Guido van Rossumb6775db1994-08-01 11:34:53 +0000238#ifdef HAVE_SYS_TIMES_H
239#include <sys/times.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000240#endif /* HAVE_SYS_TIMES_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000241
242#ifdef HAVE_SYS_PARAM_H
243#include <sys/param.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000244#endif /* HAVE_SYS_PARAM_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000245
246#ifdef HAVE_SYS_UTSNAME_H
247#include <sys/utsname.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000248#endif /* HAVE_SYS_UTSNAME_H */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000249
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000250#ifdef HAVE_DIRENT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000251#include <dirent.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000252#define NAMLEN(dirent) strlen((dirent)->d_name)
253#else
Guido van Rossumc5a0f531997-12-02 20:36:02 +0000254#if defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000255#include <direct.h>
256#define NAMLEN(dirent) strlen((dirent)->d_name)
257#else
Guido van Rossumb6775db1994-08-01 11:34:53 +0000258#define dirent direct
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000259#define NAMLEN(dirent) (dirent)->d_namlen
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000260#endif
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000261#ifdef HAVE_SYS_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000262#include <sys/ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000263#endif
264#ifdef HAVE_SYS_DIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000265#include <sys/dir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000266#endif
267#ifdef HAVE_NDIR_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000268#include <ndir.h>
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000269#endif
270#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000271
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000272#ifdef _MSC_VER
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000273#ifdef HAVE_DIRECT_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000274#include <direct.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000275#endif
276#ifdef HAVE_IO_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000277#include <io.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000278#endif
279#ifdef HAVE_PROCESS_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000280#include <process.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000281#endif
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000282#ifndef VOLUME_NAME_DOS
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +0000283#define VOLUME_NAME_DOS 0x0
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000284#endif
285#ifndef VOLUME_NAME_NT
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +0000286#define VOLUME_NAME_NT 0x2
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000287#endif
288#ifndef IO_REPARSE_TAG_SYMLINK
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +0000289#define IO_REPARSE_TAG_SYMLINK (0xA000000CL)
Raymond Hettinger0291c9f2010-08-01 21:10:35 +0000290#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000291#include "osdefs.h"
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000292#include <malloc.h>
Guido van Rossumb6775db1994-08-01 11:34:53 +0000293#include <windows.h>
Victor Stinner8c62be82010-05-06 00:08:46 +0000294#include <shellapi.h> /* for ShellExecute() */
Brian Curtine8e4b3b2010-09-23 20:04:14 +0000295#include <lmcons.h> /* for UNLEN */
Brian Curtin52173d42010-12-02 18:29:18 +0000296#ifdef SE_CREATE_SYMBOLIC_LINK_NAME /* Available starting with Vista */
297#define HAVE_SYMLINK
Brian Curtin3b4499c2010-12-28 14:31:47 +0000298static int win32_can_symlink = 0;
Brian Curtin52173d42010-12-02 18:29:18 +0000299#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000300#endif /* _MSC_VER */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000301
Guido van Rossumd48f2521997-12-05 22:19:34 +0000302#if defined(PYCC_VACPP) && defined(PYOS_OS2)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000303#include <io.h>
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000304#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000305
Tim Petersbc2e10e2002-03-03 23:17:02 +0000306#ifndef MAXPATHLEN
Thomas Wouters477c8d52006-05-27 19:21:47 +0000307#if defined(PATH_MAX) && PATH_MAX > 1024
308#define MAXPATHLEN PATH_MAX
309#else
Tim Petersbc2e10e2002-03-03 23:17:02 +0000310#define MAXPATHLEN 1024
Thomas Wouters477c8d52006-05-27 19:21:47 +0000311#endif
Tim Petersbc2e10e2002-03-03 23:17:02 +0000312#endif /* MAXPATHLEN */
313
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000314#ifdef UNION_WAIT
315/* Emulate some macros on systems that have a union instead of macros */
316
317#ifndef WIFEXITED
318#define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
319#endif
320
321#ifndef WEXITSTATUS
322#define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
323#endif
324
325#ifndef WTERMSIG
326#define WTERMSIG(u_wait) ((u_wait).w_termsig)
327#endif
328
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000329#define WAIT_TYPE union wait
330#define WAIT_STATUS_INT(s) (s.w_status)
331
332#else /* !UNION_WAIT */
333#define WAIT_TYPE int
334#define WAIT_STATUS_INT(s) (s)
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000335#endif /* UNION_WAIT */
336
Greg Wardb48bc172000-03-01 21:51:56 +0000337/* Don't use the "_r" form if we don't need it (also, won't have a
338 prototype for it, at least on Solaris -- maybe others as well?). */
339#if defined(HAVE_CTERMID_R) && defined(WITH_THREAD)
340#define USE_CTERMID_R
341#endif
342
Fred Drake699f3522000-06-29 21:12:41 +0000343/* choose the appropriate stat and fstat functions and return structs */
Guido van Rossum64529cd2000-06-30 22:45:12 +0000344#undef STAT
Antoine Pitroue47e0932011-01-19 15:21:35 +0000345#undef FSTAT
346#undef STRUCT_STAT
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000347#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +0000348# define STAT win32_stat
349# define FSTAT win32_fstat
350# define STRUCT_STAT struct win32_stat
Fred Drake699f3522000-06-29 21:12:41 +0000351#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000352# define STAT stat
353# define FSTAT fstat
354# define STRUCT_STAT struct stat
Fred Drake699f3522000-06-29 21:12:41 +0000355#endif
356
Tim Peters11b23062003-04-23 02:39:17 +0000357#if defined(MAJOR_IN_MKDEV)
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000358#include <sys/mkdev.h>
359#else
360#if defined(MAJOR_IN_SYSMACROS)
361#include <sys/sysmacros.h>
362#endif
Neal Norwitz3d949422002-04-20 13:46:43 +0000363#if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
364#include <sys/mkdev.h>
365#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +0000366#endif
Fred Drake699f3522000-06-29 21:12:41 +0000367
Georg Brandla391b112011-02-25 15:23:18 +0000368static int
369_parse_off_t(PyObject* arg, void* addr)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000370{
371#if !defined(HAVE_LARGEFILE_SUPPORT)
372 *((off_t*)addr) = PyLong_AsLong(arg);
373#else
Antoine Pitroudcc20b82011-02-26 13:38:35 +0000374 *((off_t*)addr) = PyLong_AsLongLong(arg);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +0000375#endif
376 if (PyErr_Occurred())
377 return 0;
378 return 1;
379}
380
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000381#if defined _MSC_VER && _MSC_VER >= 1400
382/* Microsoft CRT in VS2005 and higher will verify that a filehandle is
383 * valid and throw an assertion if it isn't.
384 * Normally, an invalid fd is likely to be a C program error and therefore
385 * an assertion can be useful, but it does contradict the POSIX standard
386 * which for write(2) states:
387 * "Otherwise, -1 shall be returned and errno set to indicate the error."
388 * "[EBADF] The fildes argument is not a valid file descriptor open for
389 * writing."
390 * Furthermore, python allows the user to enter any old integer
391 * as a fd and should merely raise a python exception on error.
392 * The Microsoft CRT doesn't provide an official way to check for the
393 * validity of a file descriptor, but we can emulate its internal behaviour
Victor Stinner8c62be82010-05-06 00:08:46 +0000394 * by using the exported __pinfo data member and knowledge of the
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000395 * internal structures involved.
396 * The structures below must be updated for each version of visual studio
397 * according to the file internal.h in the CRT source, until MS comes
398 * up with a less hacky way to do this.
399 * (all of this is to avoid globally modifying the CRT behaviour using
400 * _set_invalid_parameter_handler() and _CrtSetReportMode())
401 */
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000402/* The actual size of the structure is determined at runtime.
403 * Only the first items must be present.
404 */
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000405typedef struct {
Victor Stinner8c62be82010-05-06 00:08:46 +0000406 intptr_t osfhnd;
407 char osfile;
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000408} my_ioinfo;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000409
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000410extern __declspec(dllimport) char * __pioinfo[];
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000411#define IOINFO_L2E 5
412#define IOINFO_ARRAY_ELTS (1 << IOINFO_L2E)
413#define IOINFO_ARRAYS 64
414#define _NHANDLE_ (IOINFO_ARRAYS * IOINFO_ARRAY_ELTS)
415#define FOPEN 0x01
416#define _NO_CONSOLE_FILENO (intptr_t)-2
417
418/* This function emulates what the windows CRT does to validate file handles */
419int
420_PyVerify_fd(int fd)
421{
Victor Stinner8c62be82010-05-06 00:08:46 +0000422 const int i1 = fd >> IOINFO_L2E;
423 const int i2 = fd & ((1 << IOINFO_L2E) - 1);
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000424
Antoine Pitrou22e41552010-08-15 18:07:50 +0000425 static size_t sizeof_ioinfo = 0;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000426
Victor Stinner8c62be82010-05-06 00:08:46 +0000427 /* Determine the actual size of the ioinfo structure,
428 * as used by the CRT loaded in memory
429 */
430 if (sizeof_ioinfo == 0 && __pioinfo[0] != NULL) {
431 sizeof_ioinfo = _msize(__pioinfo[0]) / IOINFO_ARRAY_ELTS;
432 }
433 if (sizeof_ioinfo == 0) {
434 /* This should not happen... */
435 goto fail;
436 }
437
438 /* See that it isn't a special CLEAR fileno */
439 if (fd != _NO_CONSOLE_FILENO) {
440 /* Microsoft CRT would check that 0<=fd<_nhandle but we can't do that. Instead
441 * we check pointer validity and other info
442 */
443 if (0 <= i1 && i1 < IOINFO_ARRAYS && __pioinfo[i1] != NULL) {
444 /* finally, check that the file is open */
445 my_ioinfo* info = (my_ioinfo*)(__pioinfo[i1] + i2 * sizeof_ioinfo);
446 if (info->osfile & FOPEN) {
447 return 1;
448 }
449 }
450 }
Kristján Valur Jónssonf64e6512009-04-13 10:16:14 +0000451 fail:
Victor Stinner8c62be82010-05-06 00:08:46 +0000452 errno = EBADF;
453 return 0;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000454}
455
456/* the special case of checking dup2. The target fd must be in a sensible range */
457static int
458_PyVerify_fd_dup2(int fd1, int fd2)
459{
Victor Stinner8c62be82010-05-06 00:08:46 +0000460 if (!_PyVerify_fd(fd1))
461 return 0;
462 if (fd2 == _NO_CONSOLE_FILENO)
463 return 0;
464 if ((unsigned)fd2 < _NHANDLE_)
465 return 1;
466 else
467 return 0;
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000468}
469#else
470/* dummy version. _PyVerify_fd() is already defined in fileobject.h */
471#define _PyVerify_fd_dup2(A, B) (1)
472#endif
473
Brian Curtinfc1be6d2010-11-24 13:23:18 +0000474#ifdef MS_WINDOWS
Brian Curtinf5e76d02010-11-24 13:14:05 +0000475/* The following structure was copied from
476 http://msdn.microsoft.com/en-us/library/ms791514.aspx as the required
477 include doesn't seem to be present in the Windows SDK (at least as included
478 with Visual Studio Express). */
479typedef struct _REPARSE_DATA_BUFFER {
480 ULONG ReparseTag;
481 USHORT ReparseDataLength;
482 USHORT Reserved;
483 union {
484 struct {
485 USHORT SubstituteNameOffset;
486 USHORT SubstituteNameLength;
487 USHORT PrintNameOffset;
488 USHORT PrintNameLength;
489 ULONG Flags;
490 WCHAR PathBuffer[1];
491 } SymbolicLinkReparseBuffer;
492
493 struct {
494 USHORT SubstituteNameOffset;
495 USHORT SubstituteNameLength;
496 USHORT PrintNameOffset;
497 USHORT PrintNameLength;
498 WCHAR PathBuffer[1];
499 } MountPointReparseBuffer;
500
501 struct {
502 UCHAR DataBuffer[1];
503 } GenericReparseBuffer;
504 };
505} REPARSE_DATA_BUFFER, *PREPARSE_DATA_BUFFER;
506
507#define REPARSE_DATA_BUFFER_HEADER_SIZE FIELD_OFFSET(REPARSE_DATA_BUFFER,\
508 GenericReparseBuffer)
509#define MAXIMUM_REPARSE_DATA_BUFFER_SIZE ( 16 * 1024 )
510
511static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +0000512win32_read_link(HANDLE reparse_point_handle, ULONG *reparse_tag, wchar_t **target_path)
Brian Curtinf5e76d02010-11-24 13:14:05 +0000513{
514 char target_buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
515 REPARSE_DATA_BUFFER *rdb = (REPARSE_DATA_BUFFER *)target_buffer;
516 DWORD n_bytes_returned;
517 const wchar_t *ptr;
518 wchar_t *buf;
519 size_t len;
520
521 if (0 == DeviceIoControl(
522 reparse_point_handle,
523 FSCTL_GET_REPARSE_POINT,
524 NULL, 0, /* in buffer */
525 target_buffer, sizeof(target_buffer),
526 &n_bytes_returned,
527 NULL)) /* we're not using OVERLAPPED_IO */
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +0000528 return -1;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000529
530 if (reparse_tag)
531 *reparse_tag = rdb->ReparseTag;
532
533 if (target_path) {
534 switch (rdb->ReparseTag) {
535 case IO_REPARSE_TAG_SYMLINK:
536 /* XXX: Maybe should use SubstituteName? */
537 ptr = rdb->SymbolicLinkReparseBuffer.PathBuffer +
538 rdb->SymbolicLinkReparseBuffer.PrintNameOffset/sizeof(WCHAR);
539 len = rdb->SymbolicLinkReparseBuffer.PrintNameLength/sizeof(WCHAR);
540 break;
541 case IO_REPARSE_TAG_MOUNT_POINT:
542 ptr = rdb->MountPointReparseBuffer.PathBuffer +
543 rdb->MountPointReparseBuffer.SubstituteNameOffset/sizeof(WCHAR);
544 len = rdb->MountPointReparseBuffer.SubstituteNameLength/sizeof(WCHAR);
545 break;
546 default:
547 SetLastError(ERROR_REPARSE_TAG_MISMATCH); /* XXX: Proper error code? */
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +0000548 return -1;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000549 }
550 buf = (wchar_t *)malloc(sizeof(wchar_t)*(len+1));
551 if (!buf) {
552 SetLastError(ERROR_OUTOFMEMORY);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +0000553 return -1;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000554 }
555 wcsncpy(buf, ptr, len);
556 buf[len] = L'\0';
557 if (wcsncmp(buf, L"\\??\\", 4) == 0)
558 buf[1] = L'\\';
559 *target_path = buf;
560 }
561
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +0000562 return 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +0000563}
Brian Curtinfc1be6d2010-11-24 13:23:18 +0000564#endif /* MS_WINDOWS */
Brian Curtinf5e76d02010-11-24 13:14:05 +0000565
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000566/* Return a dictionary corresponding to the POSIX environment table */
Jack Jansenea0c3822002-08-01 21:57:49 +0000567#ifdef WITH_NEXT_FRAMEWORK
568/* On Darwin/MacOSX a shared library or framework has no access to
569** environ directly, we must obtain it with _NSGetEnviron().
570*/
571#include <crt_externs.h>
572static char **environ;
573#elif !defined(_MSC_VER) && ( !defined(__WATCOMC__) || defined(__QNX__) )
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000574extern char **environ;
Guido van Rossuma4916fa1996-05-23 22:58:55 +0000575#endif /* !_MSC_VER */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000576
Barry Warsaw53699e91996-12-10 23:23:01 +0000577static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000578convertenviron(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000579{
Victor Stinner8c62be82010-05-06 00:08:46 +0000580 PyObject *d;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000581#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +0000582 wchar_t **e;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000583#else
Victor Stinner8c62be82010-05-06 00:08:46 +0000584 char **e;
Thomas Hellerf78f12a2007-11-08 19:33:05 +0000585#endif
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +0000586#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +0000587 APIRET rc;
588 char buffer[1024]; /* OS/2 Provides a Documented Max of 1024 Chars */
589#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +0000590
Victor Stinner8c62be82010-05-06 00:08:46 +0000591 d = PyDict_New();
592 if (d == NULL)
593 return NULL;
594#ifdef WITH_NEXT_FRAMEWORK
595 if (environ == NULL)
596 environ = *_NSGetEnviron();
597#endif
598#ifdef MS_WINDOWS
599 /* _wenviron must be initialized in this way if the program is started
600 through main() instead of wmain(). */
601 _wgetenv(L"");
602 if (_wenviron == NULL)
603 return d;
604 /* This part ignores errors */
605 for (e = _wenviron; *e != NULL; e++) {
606 PyObject *k;
607 PyObject *v;
608 wchar_t *p = wcschr(*e, L'=');
609 if (p == NULL)
610 continue;
611 k = PyUnicode_FromWideChar(*e, (Py_ssize_t)(p-*e));
612 if (k == NULL) {
613 PyErr_Clear();
614 continue;
Guido van Rossumd48f2521997-12-05 22:19:34 +0000615 }
Victor Stinner8c62be82010-05-06 00:08:46 +0000616 v = PyUnicode_FromWideChar(p+1, wcslen(p+1));
617 if (v == NULL) {
618 PyErr_Clear();
619 Py_DECREF(k);
620 continue;
Guido van Rossumd48f2521997-12-05 22:19:34 +0000621 }
Victor Stinner8c62be82010-05-06 00:08:46 +0000622 if (PyDict_GetItem(d, k) == NULL) {
623 if (PyDict_SetItem(d, k, v) != 0)
624 PyErr_Clear();
625 }
626 Py_DECREF(k);
627 Py_DECREF(v);
628 }
629#else
630 if (environ == NULL)
631 return d;
632 /* This part ignores errors */
633 for (e = environ; *e != NULL; e++) {
634 PyObject *k;
635 PyObject *v;
636 char *p = strchr(*e, '=');
637 if (p == NULL)
638 continue;
Victor Stinner84ae1182010-05-06 22:05:07 +0000639 k = PyBytes_FromStringAndSize(*e, (int)(p-*e));
Victor Stinner8c62be82010-05-06 00:08:46 +0000640 if (k == NULL) {
641 PyErr_Clear();
642 continue;
643 }
Victor Stinner84ae1182010-05-06 22:05:07 +0000644 v = PyBytes_FromStringAndSize(p+1, strlen(p+1));
Victor Stinner8c62be82010-05-06 00:08:46 +0000645 if (v == NULL) {
646 PyErr_Clear();
647 Py_DECREF(k);
648 continue;
649 }
650 if (PyDict_GetItem(d, k) == NULL) {
651 if (PyDict_SetItem(d, k, v) != 0)
652 PyErr_Clear();
653 }
654 Py_DECREF(k);
655 Py_DECREF(v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000656 }
657#endif
Victor Stinner8c62be82010-05-06 00:08:46 +0000658#if defined(PYOS_OS2)
659 rc = DosQueryExtLIBPATH(buffer, BEGIN_LIBPATH);
660 if (rc == NO_ERROR) { /* (not a type, envname is NOT 'BEGIN_LIBPATH') */
661 PyObject *v = PyBytes_FromString(buffer);
662 PyDict_SetItemString(d, "BEGINLIBPATH", v);
663 Py_DECREF(v);
664 }
665 rc = DosQueryExtLIBPATH(buffer, END_LIBPATH);
666 if (rc == NO_ERROR) { /* (not a typo, envname is NOT 'END_LIBPATH') */
667 PyObject *v = PyBytes_FromString(buffer);
668 PyDict_SetItemString(d, "ENDLIBPATH", v);
669 Py_DECREF(v);
670 }
671#endif
672 return d;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000673}
674
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000675/* Set a POSIX-specific error from errno, and return NULL */
676
Barry Warsawd58d7641998-07-23 16:14:40 +0000677static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000678posix_error(void)
Guido van Rossumad0ee831995-03-01 10:34:45 +0000679{
Victor Stinner8c62be82010-05-06 00:08:46 +0000680 return PyErr_SetFromErrno(PyExc_OSError);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000681}
Barry Warsawd58d7641998-07-23 16:14:40 +0000682static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +0000683posix_error_with_filename(char* name)
Barry Warsawd58d7641998-07-23 16:14:40 +0000684{
Victor Stinner8c62be82010-05-06 00:08:46 +0000685 return PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);
Barry Warsawd58d7641998-07-23 16:14:40 +0000686}
687
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000688
Mark Hammondef8b6542001-05-13 08:04:26 +0000689static PyObject *
Martin v. Löwis011e8422009-05-05 04:43:17 +0000690posix_error_with_allocated_filename(PyObject* name)
Mark Hammondef8b6542001-05-13 08:04:26 +0000691{
Victor Stinner77ccd6d2010-05-08 00:36:42 +0000692 PyObject *name_str, *rc;
693 name_str = PyUnicode_DecodeFSDefaultAndSize(PyBytes_AsString(name),
694 PyBytes_GET_SIZE(name));
Victor Stinner8c62be82010-05-06 00:08:46 +0000695 Py_DECREF(name);
Victor Stinner77ccd6d2010-05-08 00:36:42 +0000696 rc = PyErr_SetFromErrnoWithFilenameObject(PyExc_OSError,
697 name_str);
698 Py_XDECREF(name_str);
Victor Stinner8c62be82010-05-06 00:08:46 +0000699 return rc;
Mark Hammondef8b6542001-05-13 08:04:26 +0000700}
701
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000702#ifdef MS_WINDOWS
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000703static PyObject *
Brian Curtind40e6f72010-07-08 21:39:08 +0000704win32_error(char* function, const char* filename)
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000705{
Victor Stinner8c62be82010-05-06 00:08:46 +0000706 /* XXX We should pass the function name along in the future.
707 (winreg.c also wants to pass the function name.)
708 This would however require an additional param to the
709 Windows error object, which is non-trivial.
710 */
711 errno = GetLastError();
712 if (filename)
713 return PyErr_SetFromWindowsErrWithFilename(errno, filename);
714 else
715 return PyErr_SetFromWindowsErr(errno);
Fredrik Lundhffb9c772000-07-09 14:49:51 +0000716}
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000717
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000718static PyObject *
719win32_error_unicode(char* function, Py_UNICODE* filename)
720{
Victor Stinner8c62be82010-05-06 00:08:46 +0000721 /* XXX - see win32_error for comments on 'function' */
722 errno = GetLastError();
723 if (filename)
724 return PyErr_SetFromWindowsErrWithUnicodeFilename(errno, filename);
725 else
726 return PyErr_SetFromWindowsErr(errno);
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000727}
728
Thomas Wouters477c8d52006-05-27 19:21:47 +0000729static int
Hirokazu Yamamotod7e4c082008-08-17 09:30:15 +0000730convert_to_unicode(PyObject **param)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000731{
Victor Stinner8c62be82010-05-06 00:08:46 +0000732 if (PyUnicode_CheckExact(*param))
733 Py_INCREF(*param);
734 else if (PyUnicode_Check(*param))
735 /* For a Unicode subtype that's not a Unicode object,
736 return a true Unicode object with the same data. */
737 *param = PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(*param),
738 PyUnicode_GET_SIZE(*param));
739 else
740 *param = PyUnicode_FromEncodedObject(*param,
741 Py_FileSystemDefaultEncoding,
742 "strict");
743 return (*param) != NULL;
Mark Hammondc2e85bd2002-10-03 05:10:39 +0000744}
745
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000746#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000747
Guido van Rossumd48f2521997-12-05 22:19:34 +0000748#if defined(PYOS_OS2)
749/**********************************************************************
750 * Helper Function to Trim and Format OS/2 Messages
751 **********************************************************************/
Victor Stinner8c62be82010-05-06 00:08:46 +0000752static void
Guido van Rossumd48f2521997-12-05 22:19:34 +0000753os2_formatmsg(char *msgbuf, int msglen, char *reason)
754{
755 msgbuf[msglen] = '\0'; /* OS/2 Doesn't Guarantee a Terminator */
756
757 if (strlen(msgbuf) > 0) { /* If Non-Empty Msg, Trim CRLF */
758 char *lastc = &msgbuf[ strlen(msgbuf)-1 ];
759
Neal Norwitz30b5c5d2005-12-19 06:05:18 +0000760 while (lastc > msgbuf && isspace(Py_CHARMASK(*lastc)))
Guido van Rossumd48f2521997-12-05 22:19:34 +0000761 *lastc-- = '\0'; /* Trim Trailing Whitespace (CRLF) */
762 }
763
764 /* Add Optional Reason Text */
765 if (reason) {
766 strcat(msgbuf, " : ");
767 strcat(msgbuf, reason);
768 }
769}
770
771/**********************************************************************
772 * Decode an OS/2 Operating System Error Code
773 *
774 * A convenience function to lookup an OS/2 error code and return a
775 * text message we can use to raise a Python exception.
776 *
777 * Notes:
778 * The messages for errors returned from the OS/2 kernel reside in
779 * the file OSO001.MSG in the \OS2 directory hierarchy.
780 *
781 **********************************************************************/
Victor Stinner8c62be82010-05-06 00:08:46 +0000782static char *
Guido van Rossumd48f2521997-12-05 22:19:34 +0000783os2_strerror(char *msgbuf, int msgbuflen, int errorcode, char *reason)
784{
785 APIRET rc;
786 ULONG msglen;
787
788 /* Retrieve Kernel-Related Error Message from OSO001.MSG File */
789 Py_BEGIN_ALLOW_THREADS
790 rc = DosGetMessage(NULL, 0, msgbuf, msgbuflen,
791 errorcode, "oso001.msg", &msglen);
792 Py_END_ALLOW_THREADS
793
794 if (rc == NO_ERROR)
795 os2_formatmsg(msgbuf, msglen, reason);
796 else
Tim Peters1ceb5fb2001-11-28 20:32:57 +0000797 PyOS_snprintf(msgbuf, msgbuflen,
Victor Stinner8c62be82010-05-06 00:08:46 +0000798 "unknown OS error #%d", errorcode);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000799
800 return msgbuf;
801}
802
803/* Set an OS/2-specific error and return NULL. OS/2 kernel
804 errors are not in a global variable e.g. 'errno' nor are
805 they congruent with posix error numbers. */
806
Victor Stinner8c62be82010-05-06 00:08:46 +0000807static PyObject *
808os2_error(int code)
Guido van Rossumd48f2521997-12-05 22:19:34 +0000809{
810 char text[1024];
811 PyObject *v;
812
813 os2_strerror(text, sizeof(text), code, "");
814
815 v = Py_BuildValue("(is)", code, text);
816 if (v != NULL) {
Barry Warsawca74da41999-02-09 19:31:45 +0000817 PyErr_SetObject(PyExc_OSError, v);
Guido van Rossumd48f2521997-12-05 22:19:34 +0000818 Py_DECREF(v);
819 }
820 return NULL; /* Signal to Python that an Exception is Pending */
821}
822
823#endif /* OS2 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000824
825/* POSIX generic methods */
826
Barry Warsaw53699e91996-12-10 23:23:01 +0000827static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +0000828posix_fildes(PyObject *fdobj, int (*func)(int))
829{
Victor Stinner8c62be82010-05-06 00:08:46 +0000830 int fd;
831 int res;
832 fd = PyObject_AsFileDescriptor(fdobj);
833 if (fd < 0)
834 return NULL;
835 if (!_PyVerify_fd(fd))
836 return posix_error();
837 Py_BEGIN_ALLOW_THREADS
838 res = (*func)(fd);
839 Py_END_ALLOW_THREADS
840 if (res < 0)
841 return posix_error();
842 Py_INCREF(Py_None);
843 return Py_None;
Fred Drake4d1e64b2002-04-15 19:40:07 +0000844}
Guido van Rossum21142a01999-01-08 21:05:37 +0000845
846static PyObject *
Thomas Wouters477c8d52006-05-27 19:21:47 +0000847posix_1str(PyObject *args, char *format, int (*func)(const char*))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000848{
Victor Stinner8c62be82010-05-06 00:08:46 +0000849 PyObject *opath1 = NULL;
850 char *path1;
851 int res;
852 if (!PyArg_ParseTuple(args, format,
853 PyUnicode_FSConverter, &opath1))
854 return NULL;
855 path1 = PyBytes_AsString(opath1);
856 Py_BEGIN_ALLOW_THREADS
857 res = (*func)(path1);
858 Py_END_ALLOW_THREADS
859 if (res < 0)
860 return posix_error_with_allocated_filename(opath1);
861 Py_DECREF(opath1);
862 Py_INCREF(Py_None);
863 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000864}
865
Barry Warsaw53699e91996-12-10 23:23:01 +0000866static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +0000867posix_2str(PyObject *args,
Victor Stinner8c62be82010-05-06 00:08:46 +0000868 char *format,
869 int (*func)(const char *, const char *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000870{
Victor Stinner8c62be82010-05-06 00:08:46 +0000871 PyObject *opath1 = NULL, *opath2 = NULL;
872 char *path1, *path2;
873 int res;
874 if (!PyArg_ParseTuple(args, format,
875 PyUnicode_FSConverter, &opath1,
876 PyUnicode_FSConverter, &opath2)) {
877 return NULL;
878 }
879 path1 = PyBytes_AsString(opath1);
880 path2 = PyBytes_AsString(opath2);
881 Py_BEGIN_ALLOW_THREADS
882 res = (*func)(path1, path2);
883 Py_END_ALLOW_THREADS
884 Py_DECREF(opath1);
885 Py_DECREF(opath2);
886 if (res != 0)
887 /* XXX how to report both path1 and path2??? */
888 return posix_error();
889 Py_INCREF(Py_None);
890 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000891}
892
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +0000893#ifdef MS_WINDOWS
Thomas Wouters477c8d52006-05-27 19:21:47 +0000894static PyObject*
Victor Stinner8c62be82010-05-06 00:08:46 +0000895win32_1str(PyObject* args, char* func,
896 char* format, BOOL (__stdcall *funcA)(LPCSTR),
897 char* wformat, BOOL (__stdcall *funcW)(LPWSTR))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000898{
Victor Stinner8c62be82010-05-06 00:08:46 +0000899 PyObject *uni;
900 char *ansi;
901 BOOL result;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +0000902
Victor Stinner8c62be82010-05-06 00:08:46 +0000903 if (!PyArg_ParseTuple(args, wformat, &uni))
904 PyErr_Clear();
905 else {
906 Py_BEGIN_ALLOW_THREADS
907 result = funcW(PyUnicode_AsUnicode(uni));
908 Py_END_ALLOW_THREADS
909 if (!result)
910 return win32_error_unicode(func, PyUnicode_AsUnicode(uni));
911 Py_INCREF(Py_None);
912 return Py_None;
913 }
914 if (!PyArg_ParseTuple(args, format, &ansi))
915 return NULL;
916 Py_BEGIN_ALLOW_THREADS
917 result = funcA(ansi);
918 Py_END_ALLOW_THREADS
919 if (!result)
920 return win32_error(func, ansi);
921 Py_INCREF(Py_None);
922 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000923
924}
925
926/* This is a reimplementation of the C library's chdir function,
927 but one that produces Win32 errors instead of DOS error codes.
928 chdir is essentially a wrapper around SetCurrentDirectory; however,
929 it also needs to set "magic" environment variables indicating
930 the per-drive current directory, which are of the form =<drive>: */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000931static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000932win32_chdir(LPCSTR path)
933{
Victor Stinner8c62be82010-05-06 00:08:46 +0000934 char new_path[MAX_PATH+1];
935 int result;
936 char env[4] = "=x:";
Thomas Wouters477c8d52006-05-27 19:21:47 +0000937
Victor Stinner8c62be82010-05-06 00:08:46 +0000938 if(!SetCurrentDirectoryA(path))
939 return FALSE;
940 result = GetCurrentDirectoryA(MAX_PATH+1, new_path);
941 if (!result)
942 return FALSE;
943 /* In the ANSI API, there should not be any paths longer
944 than MAX_PATH. */
945 assert(result <= MAX_PATH+1);
946 if (strncmp(new_path, "\\\\", 2) == 0 ||
947 strncmp(new_path, "//", 2) == 0)
948 /* UNC path, nothing to do. */
949 return TRUE;
950 env[1] = new_path[0];
951 return SetEnvironmentVariableA(env, new_path);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000952}
953
954/* The Unicode version differs from the ANSI version
955 since the current directory might exceed MAX_PATH characters */
Benjamin Peterson206e3072008-10-19 14:07:49 +0000956static BOOL __stdcall
Thomas Wouters477c8d52006-05-27 19:21:47 +0000957win32_wchdir(LPCWSTR path)
958{
Victor Stinner8c62be82010-05-06 00:08:46 +0000959 wchar_t _new_path[MAX_PATH+1], *new_path = _new_path;
960 int result;
961 wchar_t env[4] = L"=x:";
Thomas Wouters477c8d52006-05-27 19:21:47 +0000962
Victor Stinner8c62be82010-05-06 00:08:46 +0000963 if(!SetCurrentDirectoryW(path))
964 return FALSE;
965 result = GetCurrentDirectoryW(MAX_PATH+1, new_path);
966 if (!result)
967 return FALSE;
968 if (result > MAX_PATH+1) {
969 new_path = malloc(result * sizeof(wchar_t));
970 if (!new_path) {
971 SetLastError(ERROR_OUTOFMEMORY);
972 return FALSE;
973 }
974 result = GetCurrentDirectoryW(result, new_path);
975 if (!result) {
976 free(new_path);
977 return FALSE;
978 }
979 }
980 if (wcsncmp(new_path, L"\\\\", 2) == 0 ||
981 wcsncmp(new_path, L"//", 2) == 0)
982 /* UNC path, nothing to do. */
983 return TRUE;
984 env[1] = new_path[0];
985 result = SetEnvironmentVariableW(env, new_path);
986 if (new_path != _new_path)
987 free(new_path);
988 return result;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000989}
990#endif
991
Martin v. Löwis14694662006-02-03 12:54:16 +0000992#ifdef MS_WINDOWS
993/* The CRT of Windows has a number of flaws wrt. its stat() implementation:
994 - time stamps are restricted to second resolution
995 - file modification times suffer from forth-and-back conversions between
996 UTC and local time
997 Therefore, we implement our own stat, based on the Win32 API directly.
998*/
Victor Stinner8c62be82010-05-06 00:08:46 +0000999#define HAVE_STAT_NSEC 1
Martin v. Löwis14694662006-02-03 12:54:16 +00001000
1001struct win32_stat{
1002 int st_dev;
1003 __int64 st_ino;
1004 unsigned short st_mode;
1005 int st_nlink;
1006 int st_uid;
1007 int st_gid;
1008 int st_rdev;
1009 __int64 st_size;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001010 time_t st_atime;
Martin v. Löwis14694662006-02-03 12:54:16 +00001011 int st_atime_nsec;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001012 time_t st_mtime;
Martin v. Löwis14694662006-02-03 12:54:16 +00001013 int st_mtime_nsec;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001014 time_t st_ctime;
Martin v. Löwis14694662006-02-03 12:54:16 +00001015 int st_ctime_nsec;
1016};
1017
1018static __int64 secs_between_epochs = 11644473600; /* Seconds between 1.1.1601 and 1.1.1970 */
1019
1020static void
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001021FILE_TIME_to_time_t_nsec(FILETIME *in_ptr, time_t *time_out, int* nsec_out)
Martin v. Löwis14694662006-02-03 12:54:16 +00001022{
Victor Stinner8c62be82010-05-06 00:08:46 +00001023 /* XXX endianness. Shouldn't matter, as all Windows implementations are little-endian */
1024 /* Cannot simply cast and dereference in_ptr,
1025 since it might not be aligned properly */
1026 __int64 in;
1027 memcpy(&in, in_ptr, sizeof(in));
1028 *nsec_out = (int)(in % 10000000) * 100; /* FILETIME is in units of 100 nsec. */
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001029 *time_out = Py_SAFE_DOWNCAST((in / 10000000) - secs_between_epochs, __int64, time_t);
Martin v. Löwis14694662006-02-03 12:54:16 +00001030}
1031
Thomas Wouters477c8d52006-05-27 19:21:47 +00001032static void
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00001033time_t_to_FILE_TIME(time_t time_in, int nsec_in, FILETIME *out_ptr)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001034{
Victor Stinner8c62be82010-05-06 00:08:46 +00001035 /* XXX endianness */
1036 __int64 out;
1037 out = time_in + secs_between_epochs;
1038 out = out * 10000000 + nsec_in / 100;
1039 memcpy(out_ptr, &out, sizeof(out));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001040}
1041
Martin v. Löwis14694662006-02-03 12:54:16 +00001042/* Below, we *know* that ugo+r is 0444 */
1043#if _S_IREAD != 0400
1044#error Unsupported C library
1045#endif
1046static int
1047attributes_to_mode(DWORD attr)
1048{
Victor Stinner8c62be82010-05-06 00:08:46 +00001049 int m = 0;
1050 if (attr & FILE_ATTRIBUTE_DIRECTORY)
1051 m |= _S_IFDIR | 0111; /* IFEXEC for user,group,other */
1052 else
1053 m |= _S_IFREG;
1054 if (attr & FILE_ATTRIBUTE_READONLY)
1055 m |= 0444;
1056 else
1057 m |= 0666;
1058 return m;
Martin v. Löwis14694662006-02-03 12:54:16 +00001059}
1060
1061static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001062attribute_data_to_stat(BY_HANDLE_FILE_INFORMATION *info, ULONG reparse_tag, struct win32_stat *result)
Martin v. Löwis14694662006-02-03 12:54:16 +00001063{
Victor Stinner8c62be82010-05-06 00:08:46 +00001064 memset(result, 0, sizeof(*result));
1065 result->st_mode = attributes_to_mode(info->dwFileAttributes);
1066 result->st_size = (((__int64)info->nFileSizeHigh)<<32) + info->nFileSizeLow;
1067 FILE_TIME_to_time_t_nsec(&info->ftCreationTime, &result->st_ctime, &result->st_ctime_nsec);
1068 FILE_TIME_to_time_t_nsec(&info->ftLastWriteTime, &result->st_mtime, &result->st_mtime_nsec);
1069 FILE_TIME_to_time_t_nsec(&info->ftLastAccessTime, &result->st_atime, &result->st_atime_nsec);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001070 result->st_nlink = info->nNumberOfLinks;
Brian Curtin1b9df392010-11-24 20:24:31 +00001071 result->st_ino = (((__int64)info->nFileIndexHigh)<<32) + info->nFileIndexLow;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001072 if (reparse_tag == IO_REPARSE_TAG_SYMLINK) {
1073 /* first clear the S_IFMT bits */
1074 result->st_mode ^= (result->st_mode & 0170000);
1075 /* now set the bits that make this a symlink */
1076 result->st_mode |= 0120000;
1077 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001078
Victor Stinner8c62be82010-05-06 00:08:46 +00001079 return 0;
Martin v. Löwis14694662006-02-03 12:54:16 +00001080}
1081
Guido van Rossumd8faa362007-04-27 19:54:29 +00001082static BOOL
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001083attributes_from_dir(LPCSTR pszFile, BY_HANDLE_FILE_INFORMATION *info, ULONG *reparse_tag)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001084{
Victor Stinner8c62be82010-05-06 00:08:46 +00001085 HANDLE hFindFile;
1086 WIN32_FIND_DATAA FileData;
1087 hFindFile = FindFirstFileA(pszFile, &FileData);
1088 if (hFindFile == INVALID_HANDLE_VALUE)
1089 return FALSE;
1090 FindClose(hFindFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001091 memset(info, 0, sizeof(*info));
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001092 *reparse_tag = 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001093 info->dwFileAttributes = FileData.dwFileAttributes;
1094 info->ftCreationTime = FileData.ftCreationTime;
1095 info->ftLastAccessTime = FileData.ftLastAccessTime;
1096 info->ftLastWriteTime = FileData.ftLastWriteTime;
1097 info->nFileSizeHigh = FileData.nFileSizeHigh;
1098 info->nFileSizeLow = FileData.nFileSizeLow;
1099/* info->nNumberOfLinks = 1; */
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001100 if (FileData.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
1101 *reparse_tag = FileData.dwReserved0;
Victor Stinner8c62be82010-05-06 00:08:46 +00001102 return TRUE;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001103}
1104
1105static BOOL
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001106attributes_from_dir_w(LPCWSTR pszFile, BY_HANDLE_FILE_INFORMATION *info, ULONG *reparse_tag)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001107{
Victor Stinner8c62be82010-05-06 00:08:46 +00001108 HANDLE hFindFile;
1109 WIN32_FIND_DATAW FileData;
1110 hFindFile = FindFirstFileW(pszFile, &FileData);
1111 if (hFindFile == INVALID_HANDLE_VALUE)
1112 return FALSE;
1113 FindClose(hFindFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001114 memset(info, 0, sizeof(*info));
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001115 *reparse_tag = 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001116 info->dwFileAttributes = FileData.dwFileAttributes;
1117 info->ftCreationTime = FileData.ftCreationTime;
1118 info->ftLastAccessTime = FileData.ftLastAccessTime;
1119 info->ftLastWriteTime = FileData.ftLastWriteTime;
1120 info->nFileSizeHigh = FileData.nFileSizeHigh;
1121 info->nFileSizeLow = FileData.nFileSizeLow;
1122/* info->nNumberOfLinks = 1; */
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001123 if (FileData.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
1124 *reparse_tag = FileData.dwReserved0;
Victor Stinner8c62be82010-05-06 00:08:46 +00001125 return TRUE;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001126}
1127
Brian Curtinf5e76d02010-11-24 13:14:05 +00001128#ifndef SYMLOOP_MAX
1129#define SYMLOOP_MAX ( 88 )
1130#endif
1131
1132static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001133win32_xstat_impl_w(const wchar_t *path, struct win32_stat *result, BOOL traverse, int depth);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001134
1135static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001136win32_xstat_impl(const char *path, struct win32_stat *result, BOOL traverse, int depth)
Brian Curtinf5e76d02010-11-24 13:14:05 +00001137{
1138 int code;
1139 HANDLE hFile;
1140 BY_HANDLE_FILE_INFORMATION info;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001141 ULONG reparse_tag = 0;
Hirokazu Yamamoto74673512010-12-05 02:48:08 +00001142 wchar_t *target_path;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001143 const char *dot;
1144
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001145 if (depth > SYMLOOP_MAX) {
1146 SetLastError(ERROR_CANT_RESOLVE_FILENAME); /* XXX: ELOOP? */
1147 return -1;
1148 }
1149
Brian Curtinf5e76d02010-11-24 13:14:05 +00001150 hFile = CreateFileA(
1151 path,
1152 0, /* desired access */
1153 0, /* share mode */
1154 NULL, /* security attributes */
1155 OPEN_EXISTING,
1156 /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
1157 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS|FILE_FLAG_OPEN_REPARSE_POINT,
1158 NULL);
1159
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001160 if (hFile == INVALID_HANDLE_VALUE) {
Brian Curtinf5e76d02010-11-24 13:14:05 +00001161 /* Either the target doesn't exist, or we don't have access to
1162 get a handle to it. If the former, we need to return an error.
1163 If the latter, we can use attributes_from_dir. */
1164 if (GetLastError() != ERROR_SHARING_VIOLATION)
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001165 return -1;
1166 /* Could not get attributes on open file. Fall back to
1167 reading the directory. */
1168 if (!attributes_from_dir(path, &info, &reparse_tag))
1169 /* Very strange. This should not fail now */
1170 return -1;
1171 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
1172 if (traverse) {
1173 /* Should traverse, but could not open reparse point handle */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001174 SetLastError(ERROR_SHARING_VIOLATION);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001175 return -1;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001176 }
Brian Curtinf5e76d02010-11-24 13:14:05 +00001177 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001178 } else {
1179 if (!GetFileInformationByHandle(hFile, &info)) {
1180 CloseHandle(hFile);
1181 return -1;;
1182 }
1183 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
1184 code = win32_read_link(hFile, &reparse_tag, traverse ? &target_path : NULL);
1185 CloseHandle(hFile);
1186 if (code < 0)
1187 return code;
1188 if (traverse) {
1189 code = win32_xstat_impl_w(target_path, result, traverse, depth + 1);
1190 free(target_path);
1191 return code;
1192 }
Hirokazu Yamamoto7ed117a2010-12-07 10:24:37 +00001193 } else
1194 CloseHandle(hFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001195 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001196 attribute_data_to_stat(&info, reparse_tag, result);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001197
1198 /* Set S_IEXEC if it is an .exe, .bat, ... */
1199 dot = strrchr(path, '.');
1200 if (dot) {
1201 if (stricmp(dot, ".bat") == 0 || stricmp(dot, ".cmd") == 0 ||
1202 stricmp(dot, ".exe") == 0 || stricmp(dot, ".com") == 0)
1203 result->st_mode |= 0111;
1204 }
1205 return 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001206}
1207
1208static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001209win32_xstat_impl_w(const wchar_t *path, struct win32_stat *result, BOOL traverse, int depth)
Brian Curtinf5e76d02010-11-24 13:14:05 +00001210{
1211 int code;
1212 HANDLE hFile;
1213 BY_HANDLE_FILE_INFORMATION info;
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001214 ULONG reparse_tag = 0;
1215 wchar_t *target_path;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001216 const wchar_t *dot;
1217
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001218 if (depth > SYMLOOP_MAX) {
1219 SetLastError(ERROR_CANT_RESOLVE_FILENAME); /* XXX: ELOOP? */
1220 return -1;
1221 }
1222
Brian Curtinf5e76d02010-11-24 13:14:05 +00001223 hFile = CreateFileW(
1224 path,
1225 0, /* desired access */
1226 0, /* share mode */
1227 NULL, /* security attributes */
1228 OPEN_EXISTING,
1229 /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
1230 FILE_ATTRIBUTE_NORMAL|FILE_FLAG_BACKUP_SEMANTICS|FILE_FLAG_OPEN_REPARSE_POINT,
1231 NULL);
1232
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001233 if (hFile == INVALID_HANDLE_VALUE) {
Brian Curtinf5e76d02010-11-24 13:14:05 +00001234 /* Either the target doesn't exist, or we don't have access to
1235 get a handle to it. If the former, we need to return an error.
1236 If the latter, we can use attributes_from_dir. */
1237 if (GetLastError() != ERROR_SHARING_VIOLATION)
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001238 return -1;
1239 /* Could not get attributes on open file. Fall back to
1240 reading the directory. */
1241 if (!attributes_from_dir_w(path, &info, &reparse_tag))
1242 /* Very strange. This should not fail now */
1243 return -1;
1244 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
1245 if (traverse) {
1246 /* Should traverse, but could not open reparse point handle */
Brian Curtinf5e76d02010-11-24 13:14:05 +00001247 SetLastError(ERROR_SHARING_VIOLATION);
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001248 return -1;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001249 }
Brian Curtinf5e76d02010-11-24 13:14:05 +00001250 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001251 } else {
1252 if (!GetFileInformationByHandle(hFile, &info)) {
1253 CloseHandle(hFile);
1254 return -1;;
1255 }
1256 if (info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
1257 code = win32_read_link(hFile, &reparse_tag, traverse ? &target_path : NULL);
1258 CloseHandle(hFile);
1259 if (code < 0)
1260 return code;
1261 if (traverse) {
1262 code = win32_xstat_impl_w(target_path, result, traverse, depth + 1);
1263 free(target_path);
1264 return code;
1265 }
Hirokazu Yamamoto7ed117a2010-12-07 10:24:37 +00001266 } else
1267 CloseHandle(hFile);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001268 }
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001269 attribute_data_to_stat(&info, reparse_tag, result);
Brian Curtinf5e76d02010-11-24 13:14:05 +00001270
1271 /* Set S_IEXEC if it is an .exe, .bat, ... */
1272 dot = wcsrchr(path, '.');
1273 if (dot) {
1274 if (_wcsicmp(dot, L".bat") == 0 || _wcsicmp(dot, L".cmd") == 0 ||
1275 _wcsicmp(dot, L".exe") == 0 || _wcsicmp(dot, L".com") == 0)
1276 result->st_mode |= 0111;
1277 }
1278 return 0;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001279}
1280
1281static int
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001282win32_xstat(const char *path, struct win32_stat *result, BOOL traverse)
Brian Curtinf5e76d02010-11-24 13:14:05 +00001283{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001284 /* Protocol violation: we explicitly clear errno, instead of
1285 setting it to a POSIX error. Callers should use GetLastError. */
1286 int code = win32_xstat_impl(path, result, traverse, 0);
1287 errno = 0;
1288 return code;
1289}
Brian Curtinf5e76d02010-11-24 13:14:05 +00001290
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001291static int
1292win32_xstat_w(const wchar_t *path, struct win32_stat *result, BOOL traverse)
1293{
1294 /* Protocol violation: we explicitly clear errno, instead of
1295 setting it to a POSIX error. Callers should use GetLastError. */
1296 int code = win32_xstat_impl_w(path, result, traverse, 0);
1297 errno = 0;
1298 return code;
Brian Curtinf5e76d02010-11-24 13:14:05 +00001299}
1300
Brian Curtind40e6f72010-07-08 21:39:08 +00001301/* About the following functions: win32_lstat, win32_lstat_w, win32_stat,
1302 win32_stat_w
1303
1304 In Posix, stat automatically traverses symlinks and returns the stat
1305 structure for the target. In Windows, the equivalent GetFileAttributes by
1306 default does not traverse symlinks and instead returns attributes for
1307 the symlink.
1308
1309 Therefore, win32_lstat will get the attributes traditionally, and
1310 win32_stat will first explicitly resolve the symlink target and then will
1311 call win32_lstat on that result.
1312
Ezio Melotti4969f702011-03-15 05:59:46 +02001313 The _w represent Unicode equivalents of the aforementioned ANSI functions. */
Brian Curtind40e6f72010-07-08 21:39:08 +00001314
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001315static int
Brian Curtind40e6f72010-07-08 21:39:08 +00001316win32_lstat(const char* path, struct win32_stat *result)
Martin v. Löwis14694662006-02-03 12:54:16 +00001317{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001318 return win32_xstat(path, result, FALSE);
Martin v. Löwis14694662006-02-03 12:54:16 +00001319}
1320
Victor Stinner8c62be82010-05-06 00:08:46 +00001321static int
Brian Curtind40e6f72010-07-08 21:39:08 +00001322win32_lstat_w(const wchar_t* path, struct win32_stat *result)
Martin v. Löwis14694662006-02-03 12:54:16 +00001323{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001324 return win32_xstat_w(path, result, FALSE);
Brian Curtind40e6f72010-07-08 21:39:08 +00001325}
1326
1327static int
1328win32_stat(const char* path, struct win32_stat *result)
1329{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001330 return win32_xstat(path, result, TRUE);
Brian Curtind40e6f72010-07-08 21:39:08 +00001331}
1332
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001333static int
Brian Curtind40e6f72010-07-08 21:39:08 +00001334win32_stat_w(const wchar_t* path, struct win32_stat *result)
1335{
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001336 return win32_xstat_w(path, result, TRUE);
Martin v. Löwis14694662006-02-03 12:54:16 +00001337}
1338
1339static int
1340win32_fstat(int file_number, struct win32_stat *result)
1341{
Victor Stinner8c62be82010-05-06 00:08:46 +00001342 BY_HANDLE_FILE_INFORMATION info;
1343 HANDLE h;
1344 int type;
Martin v. Löwis14694662006-02-03 12:54:16 +00001345
Victor Stinner8c62be82010-05-06 00:08:46 +00001346 h = (HANDLE)_get_osfhandle(file_number);
Martin v. Löwis14694662006-02-03 12:54:16 +00001347
Victor Stinner8c62be82010-05-06 00:08:46 +00001348 /* Protocol violation: we explicitly clear errno, instead of
1349 setting it to a POSIX error. Callers should use GetLastError. */
1350 errno = 0;
Martin v. Löwis14694662006-02-03 12:54:16 +00001351
Victor Stinner8c62be82010-05-06 00:08:46 +00001352 if (h == INVALID_HANDLE_VALUE) {
1353 /* This is really a C library error (invalid file handle).
1354 We set the Win32 error to the closes one matching. */
1355 SetLastError(ERROR_INVALID_HANDLE);
1356 return -1;
1357 }
1358 memset(result, 0, sizeof(*result));
Martin v. Löwis14694662006-02-03 12:54:16 +00001359
Victor Stinner8c62be82010-05-06 00:08:46 +00001360 type = GetFileType(h);
1361 if (type == FILE_TYPE_UNKNOWN) {
1362 DWORD error = GetLastError();
1363 if (error != 0) {
Brian Curtinf5e76d02010-11-24 13:14:05 +00001364 return -1;
Victor Stinner8c62be82010-05-06 00:08:46 +00001365 }
1366 /* else: valid but unknown file */
1367 }
Martin v. Löwis14694662006-02-03 12:54:16 +00001368
Victor Stinner8c62be82010-05-06 00:08:46 +00001369 if (type != FILE_TYPE_DISK) {
1370 if (type == FILE_TYPE_CHAR)
1371 result->st_mode = _S_IFCHR;
1372 else if (type == FILE_TYPE_PIPE)
1373 result->st_mode = _S_IFIFO;
1374 return 0;
1375 }
1376
1377 if (!GetFileInformationByHandle(h, &info)) {
1378 return -1;
1379 }
1380
Hirokazu Yamamoto427d3142010-12-04 10:16:05 +00001381 attribute_data_to_stat(&info, 0, result);
Victor Stinner8c62be82010-05-06 00:08:46 +00001382 /* specific to fstat() */
Victor Stinner8c62be82010-05-06 00:08:46 +00001383 result->st_ino = (((__int64)info.nFileIndexHigh)<<32) + info.nFileIndexLow;
1384 return 0;
Martin v. Löwis14694662006-02-03 12:54:16 +00001385}
1386
1387#endif /* MS_WINDOWS */
1388
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001389PyDoc_STRVAR(stat_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001390"stat_result: Result from stat or lstat.\n\n\
1391This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001392 (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001393or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
1394\n\
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001395Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
1396or st_flags, they are available as attributes only.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001397\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001398See os.stat for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001399
1400static PyStructSequence_Field stat_result_fields[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001401 {"st_mode", "protection bits"},
1402 {"st_ino", "inode"},
1403 {"st_dev", "device"},
1404 {"st_nlink", "number of hard links"},
1405 {"st_uid", "user ID of owner"},
1406 {"st_gid", "group ID of owner"},
1407 {"st_size", "total size, in bytes"},
1408 /* The NULL is replaced with PyStructSequence_UnnamedField later. */
1409 {NULL, "integer time of last access"},
1410 {NULL, "integer time of last modification"},
1411 {NULL, "integer time of last change"},
1412 {"st_atime", "time of last access"},
1413 {"st_mtime", "time of last modification"},
1414 {"st_ctime", "time of last change"},
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001415#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00001416 {"st_blksize", "blocksize for filesystem I/O"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001417#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001418#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Victor Stinner8c62be82010-05-06 00:08:46 +00001419 {"st_blocks", "number of blocks allocated"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001420#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001421#ifdef HAVE_STRUCT_STAT_ST_RDEV
Victor Stinner8c62be82010-05-06 00:08:46 +00001422 {"st_rdev", "device type (if inode device)"},
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001423#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001424#ifdef HAVE_STRUCT_STAT_ST_FLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00001425 {"st_flags", "user defined flags for file"},
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001426#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001427#ifdef HAVE_STRUCT_STAT_ST_GEN
Victor Stinner8c62be82010-05-06 00:08:46 +00001428 {"st_gen", "generation number"},
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001429#endif
1430#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00001431 {"st_birthtime", "time of creation"},
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001432#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001433 {0}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001434};
1435
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001436#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001437#define ST_BLKSIZE_IDX 13
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001438#else
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001439#define ST_BLKSIZE_IDX 12
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001440#endif
1441
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001442#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001443#define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
1444#else
1445#define ST_BLOCKS_IDX ST_BLKSIZE_IDX
1446#endif
1447
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001448#ifdef HAVE_STRUCT_STAT_ST_RDEV
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001449#define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
1450#else
1451#define ST_RDEV_IDX ST_BLOCKS_IDX
1452#endif
1453
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001454#ifdef HAVE_STRUCT_STAT_ST_FLAGS
1455#define ST_FLAGS_IDX (ST_RDEV_IDX+1)
1456#else
1457#define ST_FLAGS_IDX ST_RDEV_IDX
1458#endif
1459
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001460#ifdef HAVE_STRUCT_STAT_ST_GEN
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001461#define ST_GEN_IDX (ST_FLAGS_IDX+1)
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001462#else
Martin v. Löwisf09582e2005-08-14 21:42:34 +00001463#define ST_GEN_IDX ST_FLAGS_IDX
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001464#endif
1465
1466#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
1467#define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
1468#else
1469#define ST_BIRTHTIME_IDX ST_GEN_IDX
1470#endif
1471
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001472static PyStructSequence_Desc stat_result_desc = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001473 "stat_result", /* name */
1474 stat_result__doc__, /* doc */
1475 stat_result_fields,
1476 10
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001477};
1478
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001479PyDoc_STRVAR(statvfs_result__doc__,
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001480"statvfs_result: Result from statvfs or fstatvfs.\n\n\
1481This object may be accessed either as a tuple of\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00001482 (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
Guido van Rossuma4dc73e2001-10-18 20:53:15 +00001483or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001484\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001485See os.statvfs for more information.");
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001486
1487static PyStructSequence_Field statvfs_result_fields[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001488 {"f_bsize", },
1489 {"f_frsize", },
1490 {"f_blocks", },
1491 {"f_bfree", },
1492 {"f_bavail", },
1493 {"f_files", },
1494 {"f_ffree", },
1495 {"f_favail", },
1496 {"f_flag", },
1497 {"f_namemax",},
1498 {0}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001499};
1500
1501static PyStructSequence_Desc statvfs_result_desc = {
Victor Stinner8c62be82010-05-06 00:08:46 +00001502 "statvfs_result", /* name */
1503 statvfs_result__doc__, /* doc */
1504 statvfs_result_fields,
1505 10
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001506};
1507
Ross Lagerwall7807c352011-03-17 20:20:30 +02001508#if defined(HAVE_WAITID) && !defined(__APPLE__)
1509PyDoc_STRVAR(waitid_result__doc__,
1510"waitid_result: Result from waitid.\n\n\
1511This object may be accessed either as a tuple of\n\
1512 (si_pid, si_uid, si_signo, si_status, si_code),\n\
1513or via the attributes si_pid, si_uid, and so on.\n\
1514\n\
1515See os.waitid for more information.");
1516
1517static PyStructSequence_Field waitid_result_fields[] = {
1518 {"si_pid", },
1519 {"si_uid", },
1520 {"si_signo", },
1521 {"si_status", },
1522 {"si_code", },
1523 {0}
1524};
1525
1526static PyStructSequence_Desc waitid_result_desc = {
1527 "waitid_result", /* name */
1528 waitid_result__doc__, /* doc */
1529 waitid_result_fields,
1530 5
1531};
1532static PyTypeObject WaitidResultType;
1533#endif
1534
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001535static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001536static PyTypeObject StatResultType;
1537static PyTypeObject StatVFSResultType;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001538static newfunc structseq_new;
1539
1540static PyObject *
1541statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1542{
Victor Stinner8c62be82010-05-06 00:08:46 +00001543 PyStructSequence *result;
1544 int i;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001545
Victor Stinner8c62be82010-05-06 00:08:46 +00001546 result = (PyStructSequence*)structseq_new(type, args, kwds);
1547 if (!result)
1548 return NULL;
1549 /* If we have been initialized from a tuple,
1550 st_?time might be set to None. Initialize it
1551 from the int slots. */
1552 for (i = 7; i <= 9; i++) {
1553 if (result->ob_item[i+3] == Py_None) {
1554 Py_DECREF(Py_None);
1555 Py_INCREF(result->ob_item[i]);
1556 result->ob_item[i+3] = result->ob_item[i];
1557 }
1558 }
1559 return (PyObject*)result;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001560}
1561
1562
1563
1564/* If true, st_?time is float. */
Martin v. Löwisfe33d0b2005-01-16 08:57:39 +00001565static int _stat_float_times = 1;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001566
1567PyDoc_STRVAR(stat_float_times__doc__,
1568"stat_float_times([newval]) -> oldval\n\n\
1569Determine whether os.[lf]stat represents time stamps as float objects.\n\
1570If newval is True, future calls to stat() return floats, if it is False,\n\
1571future calls return ints. \n\
1572If newval is omitted, return the current setting.\n");
1573
1574static PyObject*
1575stat_float_times(PyObject* self, PyObject *args)
1576{
Victor Stinner8c62be82010-05-06 00:08:46 +00001577 int newval = -1;
1578 if (!PyArg_ParseTuple(args, "|i:stat_float_times", &newval))
1579 return NULL;
1580 if (newval == -1)
1581 /* Return old value */
1582 return PyBool_FromLong(_stat_float_times);
1583 _stat_float_times = newval;
1584 Py_INCREF(Py_None);
1585 return Py_None;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001586}
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001587
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001588static void
1589fill_time(PyObject *v, int index, time_t sec, unsigned long nsec)
1590{
Victor Stinner8c62be82010-05-06 00:08:46 +00001591 PyObject *fval,*ival;
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001592#if SIZEOF_TIME_T > SIZEOF_LONG
Victor Stinner8c62be82010-05-06 00:08:46 +00001593 ival = PyLong_FromLongLong((PY_LONG_LONG)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001594#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001595 ival = PyLong_FromLong((long)sec);
Martin v. Löwisf607bda2002-10-16 18:27:39 +00001596#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001597 if (!ival)
1598 return;
1599 if (_stat_float_times) {
1600 fval = PyFloat_FromDouble(sec + 1e-9*nsec);
1601 } else {
1602 fval = ival;
1603 Py_INCREF(fval);
1604 }
1605 PyStructSequence_SET_ITEM(v, index, ival);
1606 PyStructSequence_SET_ITEM(v, index+3, fval);
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001607}
1608
Tim Peters5aa91602002-01-30 05:46:57 +00001609/* pack a system stat C structure into the Python stat tuple
Fred Drake699f3522000-06-29 21:12:41 +00001610 (used by posix_stat() and posix_fstat()) */
1611static PyObject*
Martin v. Löwis14694662006-02-03 12:54:16 +00001612_pystat_fromstructstat(STRUCT_STAT *st)
Fred Drake699f3522000-06-29 21:12:41 +00001613{
Victor Stinner8c62be82010-05-06 00:08:46 +00001614 unsigned long ansec, mnsec, cnsec;
1615 PyObject *v = PyStructSequence_New(&StatResultType);
1616 if (v == NULL)
1617 return NULL;
Fred Drake699f3522000-06-29 21:12:41 +00001618
Victor Stinner8c62be82010-05-06 00:08:46 +00001619 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long)st->st_mode));
Fred Drake699f3522000-06-29 21:12:41 +00001620#ifdef HAVE_LARGEFILE_SUPPORT
Victor Stinner8c62be82010-05-06 00:08:46 +00001621 PyStructSequence_SET_ITEM(v, 1,
1622 PyLong_FromLongLong((PY_LONG_LONG)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001623#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001624 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long)st->st_ino));
Fred Drake699f3522000-06-29 21:12:41 +00001625#endif
1626#if defined(HAVE_LONG_LONG) && !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00001627 PyStructSequence_SET_ITEM(v, 2,
1628 PyLong_FromLongLong((PY_LONG_LONG)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001629#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001630 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long)st->st_dev));
Fred Drake699f3522000-06-29 21:12:41 +00001631#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001632 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long)st->st_nlink));
1633 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long)st->st_uid));
1634 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long)st->st_gid));
Fred Drake699f3522000-06-29 21:12:41 +00001635#ifdef HAVE_LARGEFILE_SUPPORT
Victor Stinner8c62be82010-05-06 00:08:46 +00001636 PyStructSequence_SET_ITEM(v, 6,
1637 PyLong_FromLongLong((PY_LONG_LONG)st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001638#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001639 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong(st->st_size));
Fred Drake699f3522000-06-29 21:12:41 +00001640#endif
Martin v. Löwis94717ed2002-09-09 14:24:16 +00001641
Martin v. Löwis14694662006-02-03 12:54:16 +00001642#if defined(HAVE_STAT_TV_NSEC)
Victor Stinner8c62be82010-05-06 00:08:46 +00001643 ansec = st->st_atim.tv_nsec;
1644 mnsec = st->st_mtim.tv_nsec;
1645 cnsec = st->st_ctim.tv_nsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001646#elif defined(HAVE_STAT_TV_NSEC2)
Victor Stinner8c62be82010-05-06 00:08:46 +00001647 ansec = st->st_atimespec.tv_nsec;
1648 mnsec = st->st_mtimespec.tv_nsec;
1649 cnsec = st->st_ctimespec.tv_nsec;
Martin v. Löwis14694662006-02-03 12:54:16 +00001650#elif defined(HAVE_STAT_NSEC)
Victor Stinner8c62be82010-05-06 00:08:46 +00001651 ansec = st->st_atime_nsec;
1652 mnsec = st->st_mtime_nsec;
1653 cnsec = st->st_ctime_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001654#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001655 ansec = mnsec = cnsec = 0;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001656#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001657 fill_time(v, 7, st->st_atime, ansec);
1658 fill_time(v, 8, st->st_mtime, mnsec);
1659 fill_time(v, 9, st->st_ctime, cnsec);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001660
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001661#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00001662 PyStructSequence_SET_ITEM(v, ST_BLKSIZE_IDX,
1663 PyLong_FromLong((long)st->st_blksize));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001664#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001665#ifdef HAVE_STRUCT_STAT_ST_BLOCKS
Victor Stinner8c62be82010-05-06 00:08:46 +00001666 PyStructSequence_SET_ITEM(v, ST_BLOCKS_IDX,
1667 PyLong_FromLong((long)st->st_blocks));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001668#endif
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001669#ifdef HAVE_STRUCT_STAT_ST_RDEV
Victor Stinner8c62be82010-05-06 00:08:46 +00001670 PyStructSequence_SET_ITEM(v, ST_RDEV_IDX,
1671 PyLong_FromLong((long)st->st_rdev));
Fred Drake699f3522000-06-29 21:12:41 +00001672#endif
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001673#ifdef HAVE_STRUCT_STAT_ST_GEN
Victor Stinner8c62be82010-05-06 00:08:46 +00001674 PyStructSequence_SET_ITEM(v, ST_GEN_IDX,
1675 PyLong_FromLong((long)st->st_gen));
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001676#endif
1677#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00001678 {
1679 PyObject *val;
1680 unsigned long bsec,bnsec;
1681 bsec = (long)st->st_birthtime;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001682#ifdef HAVE_STAT_TV_NSEC2
Victor Stinner8c62be82010-05-06 00:08:46 +00001683 bnsec = st->st_birthtimespec.tv_nsec;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001684#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001685 bnsec = 0;
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001686#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001687 if (_stat_float_times) {
1688 val = PyFloat_FromDouble(bsec + 1e-9*bnsec);
1689 } else {
1690 val = PyLong_FromLong((long)bsec);
1691 }
1692 PyStructSequence_SET_ITEM(v, ST_BIRTHTIME_IDX,
1693 val);
1694 }
Martin v. Löwisebd9d5b2005-08-09 15:00:59 +00001695#endif
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001696#ifdef HAVE_STRUCT_STAT_ST_FLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00001697 PyStructSequence_SET_ITEM(v, ST_FLAGS_IDX,
1698 PyLong_FromLong((long)st->st_flags));
Hye-Shik Chang5f937a72005-06-02 13:09:30 +00001699#endif
Fred Drake699f3522000-06-29 21:12:41 +00001700
Victor Stinner8c62be82010-05-06 00:08:46 +00001701 if (PyErr_Occurred()) {
1702 Py_DECREF(v);
1703 return NULL;
1704 }
Fred Drake699f3522000-06-29 21:12:41 +00001705
Victor Stinner8c62be82010-05-06 00:08:46 +00001706 return v;
Fred Drake699f3522000-06-29 21:12:41 +00001707}
1708
Barry Warsaw53699e91996-12-10 23:23:01 +00001709static PyObject *
Tim Peters11b23062003-04-23 02:39:17 +00001710posix_do_stat(PyObject *self, PyObject *args,
Victor Stinner8c62be82010-05-06 00:08:46 +00001711 char *format,
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001712#ifdef __VMS
Victor Stinner8c62be82010-05-06 00:08:46 +00001713 int (*statfunc)(const char *, STRUCT_STAT *, ...),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001714#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001715 int (*statfunc)(const char *, STRUCT_STAT *),
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001716#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001717 char *wformat,
1718 int (*wstatfunc)(const Py_UNICODE *, STRUCT_STAT *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001719{
Victor Stinner8c62be82010-05-06 00:08:46 +00001720 STRUCT_STAT st;
1721 PyObject *opath;
1722 char *path;
1723 int res;
1724 PyObject *result;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001725
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001726#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00001727 PyUnicodeObject *po;
1728 if (PyArg_ParseTuple(args, wformat, &po)) {
1729 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
Martin v. Löwis14694662006-02-03 12:54:16 +00001730
Victor Stinner8c62be82010-05-06 00:08:46 +00001731 Py_BEGIN_ALLOW_THREADS
1732 /* PyUnicode_AS_UNICODE result OK without
1733 thread lock as it is a simple dereference. */
1734 res = wstatfunc(wpath, &st);
1735 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001736
Victor Stinner8c62be82010-05-06 00:08:46 +00001737 if (res != 0)
1738 return win32_error_unicode("stat", wpath);
1739 return _pystat_fromstructstat(&st);
1740 }
1741 /* Drop the argument parsing error as narrow strings
1742 are also valid. */
1743 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001744#endif
1745
Victor Stinner8c62be82010-05-06 00:08:46 +00001746 if (!PyArg_ParseTuple(args, format,
1747 PyUnicode_FSConverter, &opath))
1748 return NULL;
1749 path = PyBytes_AsString(opath);
1750 Py_BEGIN_ALLOW_THREADS
1751 res = (*statfunc)(path, &st);
1752 Py_END_ALLOW_THREADS
Martin v. Löwis14694662006-02-03 12:54:16 +00001753
Victor Stinner8c62be82010-05-06 00:08:46 +00001754 if (res != 0) {
Martin v. Löwis14694662006-02-03 12:54:16 +00001755#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00001756 result = win32_error("stat", path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001757#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001758 result = posix_error_with_filename(path);
Martin v. Löwis14694662006-02-03 12:54:16 +00001759#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001760 }
1761 else
1762 result = _pystat_fromstructstat(&st);
Fred Drake699f3522000-06-29 21:12:41 +00001763
Victor Stinner8c62be82010-05-06 00:08:46 +00001764 Py_DECREF(opath);
1765 return result;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001766}
1767
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001768/* POSIX methods */
1769
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001770PyDoc_STRVAR(posix_access__doc__,
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001771"access(path, mode) -> True if granted, False otherwise\n\n\
Guido van Rossuma0b90752002-06-18 16:22:43 +00001772Use the real uid/gid to test for access to a path. Note that most\n\
1773operations will use the effective uid/gid, therefore this routine can\n\
1774be used in a suid/sgid environment to test if the invoking user has the\n\
1775specified access to the path. The mode argument can be F_OK to test\n\
1776existence, or the inclusive-OR of R_OK, W_OK, and X_OK.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001777
1778static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001779posix_access(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001780{
Victor Stinner8c62be82010-05-06 00:08:46 +00001781 PyObject *opath;
1782 char *path;
1783 int mode;
1784
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001785#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00001786 DWORD attr;
1787 PyUnicodeObject *po;
1788 if (PyArg_ParseTuple(args, "Ui:access", &po, &mode)) {
1789 Py_BEGIN_ALLOW_THREADS
1790 /* PyUnicode_AS_UNICODE OK without thread lock as
1791 it is a simple dereference. */
1792 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1793 Py_END_ALLOW_THREADS
1794 goto finish;
1795 }
1796 /* Drop the argument parsing error as narrow strings
1797 are also valid. */
1798 PyErr_Clear();
1799 if (!PyArg_ParseTuple(args, "O&i:access",
1800 PyUnicode_FSConverter, &opath, &mode))
1801 return NULL;
1802 path = PyBytes_AsString(opath);
1803 Py_BEGIN_ALLOW_THREADS
1804 attr = GetFileAttributesA(path);
1805 Py_END_ALLOW_THREADS
1806 Py_DECREF(opath);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001807finish:
Victor Stinner8c62be82010-05-06 00:08:46 +00001808 if (attr == 0xFFFFFFFF)
1809 /* File does not exist, or cannot read attributes */
1810 return PyBool_FromLong(0);
1811 /* Access is possible if either write access wasn't requested, or
1812 the file isn't read-only, or if it's a directory, as there are
1813 no read-only directories on Windows. */
1814 return PyBool_FromLong(!(mode & 2)
1815 || !(attr & FILE_ATTRIBUTE_READONLY)
1816 || (attr & FILE_ATTRIBUTE_DIRECTORY));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001817#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001818 int res;
1819 if (!PyArg_ParseTuple(args, "O&i:access",
1820 PyUnicode_FSConverter, &opath, &mode))
1821 return NULL;
1822 path = PyBytes_AsString(opath);
1823 Py_BEGIN_ALLOW_THREADS
1824 res = access(path, mode);
1825 Py_END_ALLOW_THREADS
1826 Py_DECREF(opath);
1827 return PyBool_FromLong(res == 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001828#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001829}
1830
Guido van Rossumd371ff11999-01-25 16:12:23 +00001831#ifndef F_OK
1832#define F_OK 0
1833#endif
1834#ifndef R_OK
1835#define R_OK 4
1836#endif
1837#ifndef W_OK
1838#define W_OK 2
1839#endif
1840#ifndef X_OK
1841#define X_OK 1
1842#endif
1843
1844#ifdef HAVE_TTYNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001845PyDoc_STRVAR(posix_ttyname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001846"ttyname(fd) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001847Return the name of the terminal device connected to 'fd'.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00001848
1849static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001850posix_ttyname(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00001851{
Victor Stinner8c62be82010-05-06 00:08:46 +00001852 int id;
1853 char *ret;
Guido van Rossum94f6f721999-01-06 18:42:14 +00001854
Victor Stinner8c62be82010-05-06 00:08:46 +00001855 if (!PyArg_ParseTuple(args, "i:ttyname", &id))
1856 return NULL;
Guido van Rossum94f6f721999-01-06 18:42:14 +00001857
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001858#if defined(__VMS)
Victor Stinner8c62be82010-05-06 00:08:46 +00001859 /* file descriptor 0 only, the default input device (stdin) */
1860 if (id == 0) {
1861 ret = ttyname();
1862 }
1863 else {
1864 ret = NULL;
1865 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001866#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001867 ret = ttyname(id);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001868#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001869 if (ret == NULL)
1870 return posix_error();
Victor Stinner5fe6de82010-08-15 09:12:51 +00001871 return PyUnicode_DecodeFSDefault(ret);
Guido van Rossum94f6f721999-01-06 18:42:14 +00001872}
Guido van Rossumd371ff11999-01-25 16:12:23 +00001873#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00001874
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001875#ifdef HAVE_CTERMID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001876PyDoc_STRVAR(posix_ctermid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001877"ctermid() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001878Return the name of the controlling terminal for this process.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001879
1880static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00001881posix_ctermid(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001882{
Victor Stinner8c62be82010-05-06 00:08:46 +00001883 char *ret;
1884 char buffer[L_ctermid];
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001885
Greg Wardb48bc172000-03-01 21:51:56 +00001886#ifdef USE_CTERMID_R
Victor Stinner8c62be82010-05-06 00:08:46 +00001887 ret = ctermid_r(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001888#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001889 ret = ctermid(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001890#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00001891 if (ret == NULL)
1892 return posix_error();
Victor Stinner5fe6de82010-08-15 09:12:51 +00001893 return PyUnicode_DecodeFSDefault(buffer);
Fred Drake5ab8eaf1999-12-09 21:13:07 +00001894}
1895#endif
1896
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001897PyDoc_STRVAR(posix_chdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001898"chdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001899Change the current working directory to the specified path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001900
Barry Warsaw53699e91996-12-10 23:23:01 +00001901static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001902posix_chdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001903{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001904#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00001905 return win32_1str(args, "chdir", "y:chdir", win32_chdir, "U:chdir", win32_wchdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00001906#elif defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00001907 return posix_1str(args, "O&:chdir", _chdir2);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00001908#elif defined(__VMS)
Victor Stinner8c62be82010-05-06 00:08:46 +00001909 return posix_1str(args, "O&:chdir", (int (*)(const char *))chdir);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001910#else
Victor Stinner8c62be82010-05-06 00:08:46 +00001911 return posix_1str(args, "O&:chdir", chdir);
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00001912#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001913}
1914
Fred Drake4d1e64b2002-04-15 19:40:07 +00001915#ifdef HAVE_FCHDIR
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001916PyDoc_STRVAR(posix_fchdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001917"fchdir(fildes)\n\n\
Fred Drake4d1e64b2002-04-15 19:40:07 +00001918Change to the directory of the given file descriptor. fildes must be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001919opened on a directory, not a file.");
Fred Drake4d1e64b2002-04-15 19:40:07 +00001920
1921static PyObject *
1922posix_fchdir(PyObject *self, PyObject *fdobj)
1923{
Victor Stinner8c62be82010-05-06 00:08:46 +00001924 return posix_fildes(fdobj, fchdir);
Fred Drake4d1e64b2002-04-15 19:40:07 +00001925}
1926#endif /* HAVE_FCHDIR */
1927
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001928
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001929PyDoc_STRVAR(posix_chmod__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00001930"chmod(path, mode)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001931Change the access permissions of a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00001932
Barry Warsaw53699e91996-12-10 23:23:01 +00001933static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00001934posix_chmod(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001935{
Victor Stinner8c62be82010-05-06 00:08:46 +00001936 PyObject *opath = NULL;
1937 char *path = NULL;
1938 int i;
1939 int res;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001940#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00001941 DWORD attr;
1942 PyUnicodeObject *po;
1943 if (PyArg_ParseTuple(args, "Ui|:chmod", &po, &i)) {
1944 Py_BEGIN_ALLOW_THREADS
1945 attr = GetFileAttributesW(PyUnicode_AS_UNICODE(po));
1946 if (attr != 0xFFFFFFFF) {
1947 if (i & _S_IWRITE)
1948 attr &= ~FILE_ATTRIBUTE_READONLY;
1949 else
1950 attr |= FILE_ATTRIBUTE_READONLY;
1951 res = SetFileAttributesW(PyUnicode_AS_UNICODE(po), attr);
1952 }
1953 else
1954 res = 0;
1955 Py_END_ALLOW_THREADS
1956 if (!res)
1957 return win32_error_unicode("chmod",
1958 PyUnicode_AS_UNICODE(po));
1959 Py_INCREF(Py_None);
1960 return Py_None;
1961 }
1962 /* Drop the argument parsing error as narrow strings
1963 are also valid. */
1964 PyErr_Clear();
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00001965
Victor Stinner8c62be82010-05-06 00:08:46 +00001966 if (!PyArg_ParseTuple(args, "O&i:chmod", PyUnicode_FSConverter,
1967 &opath, &i))
1968 return NULL;
1969 path = PyBytes_AsString(opath);
1970 Py_BEGIN_ALLOW_THREADS
1971 attr = GetFileAttributesA(path);
1972 if (attr != 0xFFFFFFFF) {
1973 if (i & _S_IWRITE)
1974 attr &= ~FILE_ATTRIBUTE_READONLY;
1975 else
1976 attr |= FILE_ATTRIBUTE_READONLY;
1977 res = SetFileAttributesA(path, attr);
1978 }
1979 else
1980 res = 0;
1981 Py_END_ALLOW_THREADS
1982 if (!res) {
1983 win32_error("chmod", path);
1984 Py_DECREF(opath);
1985 return NULL;
1986 }
1987 Py_DECREF(opath);
1988 Py_INCREF(Py_None);
1989 return Py_None;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00001990#else /* MS_WINDOWS */
Victor Stinner8c62be82010-05-06 00:08:46 +00001991 if (!PyArg_ParseTuple(args, "O&i:chmod", PyUnicode_FSConverter,
1992 &opath, &i))
1993 return NULL;
1994 path = PyBytes_AsString(opath);
1995 Py_BEGIN_ALLOW_THREADS
1996 res = chmod(path, i);
1997 Py_END_ALLOW_THREADS
1998 if (res < 0)
1999 return posix_error_with_allocated_filename(opath);
2000 Py_DECREF(opath);
2001 Py_INCREF(Py_None);
2002 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002003#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002004}
2005
Christian Heimes4e30a842007-11-30 22:12:06 +00002006#ifdef HAVE_FCHMOD
2007PyDoc_STRVAR(posix_fchmod__doc__,
2008"fchmod(fd, mode)\n\n\
2009Change the access permissions of the file given by file\n\
2010descriptor fd.");
2011
2012static PyObject *
2013posix_fchmod(PyObject *self, PyObject *args)
2014{
Victor Stinner8c62be82010-05-06 00:08:46 +00002015 int fd, mode, res;
2016 if (!PyArg_ParseTuple(args, "ii:fchmod", &fd, &mode))
2017 return NULL;
2018 Py_BEGIN_ALLOW_THREADS
2019 res = fchmod(fd, mode);
2020 Py_END_ALLOW_THREADS
2021 if (res < 0)
2022 return posix_error();
2023 Py_RETURN_NONE;
Christian Heimes4e30a842007-11-30 22:12:06 +00002024}
2025#endif /* HAVE_FCHMOD */
2026
2027#ifdef HAVE_LCHMOD
2028PyDoc_STRVAR(posix_lchmod__doc__,
2029"lchmod(path, mode)\n\n\
2030Change the access permissions of a file. If path is a symlink, this\n\
2031affects the link itself rather than the target.");
2032
2033static PyObject *
2034posix_lchmod(PyObject *self, PyObject *args)
2035{
Victor Stinner8c62be82010-05-06 00:08:46 +00002036 PyObject *opath;
2037 char *path;
2038 int i;
2039 int res;
2040 if (!PyArg_ParseTuple(args, "O&i:lchmod", PyUnicode_FSConverter,
2041 &opath, &i))
2042 return NULL;
2043 path = PyBytes_AsString(opath);
2044 Py_BEGIN_ALLOW_THREADS
2045 res = lchmod(path, i);
2046 Py_END_ALLOW_THREADS
2047 if (res < 0)
2048 return posix_error_with_allocated_filename(opath);
2049 Py_DECREF(opath);
2050 Py_RETURN_NONE;
Christian Heimes4e30a842007-11-30 22:12:06 +00002051}
2052#endif /* HAVE_LCHMOD */
2053
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002054
Thomas Wouterscf297e42007-02-23 15:07:44 +00002055#ifdef HAVE_CHFLAGS
2056PyDoc_STRVAR(posix_chflags__doc__,
2057"chflags(path, flags)\n\n\
2058Set file flags.");
2059
2060static PyObject *
2061posix_chflags(PyObject *self, PyObject *args)
2062{
Victor Stinner8c62be82010-05-06 00:08:46 +00002063 PyObject *opath;
2064 char *path;
2065 unsigned long flags;
2066 int res;
2067 if (!PyArg_ParseTuple(args, "O&k:chflags",
2068 PyUnicode_FSConverter, &opath, &flags))
2069 return NULL;
2070 path = PyBytes_AsString(opath);
2071 Py_BEGIN_ALLOW_THREADS
2072 res = chflags(path, flags);
2073 Py_END_ALLOW_THREADS
2074 if (res < 0)
2075 return posix_error_with_allocated_filename(opath);
2076 Py_DECREF(opath);
2077 Py_INCREF(Py_None);
2078 return Py_None;
Thomas Wouterscf297e42007-02-23 15:07:44 +00002079}
2080#endif /* HAVE_CHFLAGS */
2081
2082#ifdef HAVE_LCHFLAGS
2083PyDoc_STRVAR(posix_lchflags__doc__,
2084"lchflags(path, flags)\n\n\
2085Set file flags.\n\
2086This function will not follow symbolic links.");
2087
2088static PyObject *
2089posix_lchflags(PyObject *self, PyObject *args)
2090{
Victor Stinner8c62be82010-05-06 00:08:46 +00002091 PyObject *opath;
2092 char *path;
2093 unsigned long flags;
2094 int res;
2095 if (!PyArg_ParseTuple(args, "O&k:lchflags",
2096 PyUnicode_FSConverter, &opath, &flags))
2097 return NULL;
2098 path = PyBytes_AsString(opath);
2099 Py_BEGIN_ALLOW_THREADS
2100 res = lchflags(path, flags);
2101 Py_END_ALLOW_THREADS
2102 if (res < 0)
2103 return posix_error_with_allocated_filename(opath);
2104 Py_DECREF(opath);
2105 Py_INCREF(Py_None);
2106 return Py_None;
Thomas Wouterscf297e42007-02-23 15:07:44 +00002107}
2108#endif /* HAVE_LCHFLAGS */
2109
Martin v. Löwis244edc82001-10-04 22:44:26 +00002110#ifdef HAVE_CHROOT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002111PyDoc_STRVAR(posix_chroot__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002112"chroot(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002113Change root directory to path.");
Martin v. Löwis244edc82001-10-04 22:44:26 +00002114
2115static PyObject *
2116posix_chroot(PyObject *self, PyObject *args)
2117{
Victor Stinner8c62be82010-05-06 00:08:46 +00002118 return posix_1str(args, "O&:chroot", chroot);
Martin v. Löwis244edc82001-10-04 22:44:26 +00002119}
2120#endif
2121
Guido van Rossum21142a01999-01-08 21:05:37 +00002122#ifdef HAVE_FSYNC
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002123PyDoc_STRVAR(posix_fsync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002124"fsync(fildes)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002125force write of file with filedescriptor to disk.");
Guido van Rossum21142a01999-01-08 21:05:37 +00002126
2127static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00002128posix_fsync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00002129{
Stefan Krah0e803b32010-11-26 16:16:47 +00002130 return posix_fildes(fdobj, fsync);
Guido van Rossum21142a01999-01-08 21:05:37 +00002131}
2132#endif /* HAVE_FSYNC */
2133
Ross Lagerwall7807c352011-03-17 20:20:30 +02002134#ifdef HAVE_SYNC
2135PyDoc_STRVAR(posix_sync__doc__,
2136"sync()\n\n\
2137Force write of everything to disk.");
2138
2139static PyObject *
2140posix_sync(PyObject *self, PyObject *noargs)
2141{
2142 Py_BEGIN_ALLOW_THREADS
2143 sync();
2144 Py_END_ALLOW_THREADS
2145 Py_RETURN_NONE;
2146}
2147#endif
2148
Guido van Rossum21142a01999-01-08 21:05:37 +00002149#ifdef HAVE_FDATASYNC
Guido van Rossumecc23b02000-09-22 16:01:05 +00002150
Guido van Rossum7f58e2e2000-09-22 17:26:14 +00002151#ifdef __hpux
Guido van Rossumecc23b02000-09-22 16:01:05 +00002152extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
2153#endif
2154
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002155PyDoc_STRVAR(posix_fdatasync__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002156"fdatasync(fildes)\n\n\
Guido van Rossum21142a01999-01-08 21:05:37 +00002157force write of file with filedescriptor to disk.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002158 does not force update of metadata.");
Guido van Rossum21142a01999-01-08 21:05:37 +00002159
2160static PyObject *
Fred Drake4d1e64b2002-04-15 19:40:07 +00002161posix_fdatasync(PyObject *self, PyObject *fdobj)
Guido van Rossum21142a01999-01-08 21:05:37 +00002162{
Stefan Krah0e803b32010-11-26 16:16:47 +00002163 return posix_fildes(fdobj, fdatasync);
Guido van Rossum21142a01999-01-08 21:05:37 +00002164}
2165#endif /* HAVE_FDATASYNC */
2166
2167
Fredrik Lundh10723342000-07-10 16:38:09 +00002168#ifdef HAVE_CHOWN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002169PyDoc_STRVAR(posix_chown__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002170"chown(path, uid, gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002171Change the owner and group id of path to the numeric uid and gid.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002172
Barry Warsaw53699e91996-12-10 23:23:01 +00002173static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002174posix_chown(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002175{
Victor Stinner8c62be82010-05-06 00:08:46 +00002176 PyObject *opath;
2177 char *path;
2178 long uid, gid;
2179 int res;
2180 if (!PyArg_ParseTuple(args, "O&ll:chown",
2181 PyUnicode_FSConverter, &opath,
2182 &uid, &gid))
2183 return NULL;
2184 path = PyBytes_AsString(opath);
2185 Py_BEGIN_ALLOW_THREADS
2186 res = chown(path, (uid_t) uid, (gid_t) gid);
2187 Py_END_ALLOW_THREADS
2188 if (res < 0)
2189 return posix_error_with_allocated_filename(opath);
2190 Py_DECREF(opath);
2191 Py_INCREF(Py_None);
2192 return Py_None;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002193}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002194#endif /* HAVE_CHOWN */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002195
Christian Heimes4e30a842007-11-30 22:12:06 +00002196#ifdef HAVE_FCHOWN
2197PyDoc_STRVAR(posix_fchown__doc__,
2198"fchown(fd, uid, gid)\n\n\
2199Change the owner and group id of the file given by file descriptor\n\
2200fd to the numeric uid and gid.");
2201
2202static PyObject *
2203posix_fchown(PyObject *self, PyObject *args)
2204{
Victor Stinner8c62be82010-05-06 00:08:46 +00002205 int fd;
2206 long uid, gid;
2207 int res;
2208 if (!PyArg_ParseTuple(args, "ill:chown", &fd, &uid, &gid))
2209 return NULL;
2210 Py_BEGIN_ALLOW_THREADS
2211 res = fchown(fd, (uid_t) uid, (gid_t) gid);
2212 Py_END_ALLOW_THREADS
2213 if (res < 0)
2214 return posix_error();
2215 Py_RETURN_NONE;
Christian Heimes4e30a842007-11-30 22:12:06 +00002216}
2217#endif /* HAVE_FCHOWN */
2218
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002219#ifdef HAVE_LCHOWN
2220PyDoc_STRVAR(posix_lchown__doc__,
2221"lchown(path, uid, gid)\n\n\
2222Change the owner and group id of path to the numeric uid and gid.\n\
2223This function will not follow symbolic links.");
2224
2225static PyObject *
2226posix_lchown(PyObject *self, PyObject *args)
2227{
Victor Stinner8c62be82010-05-06 00:08:46 +00002228 PyObject *opath;
2229 char *path;
2230 long uid, gid;
2231 int res;
2232 if (!PyArg_ParseTuple(args, "O&ll:lchown",
2233 PyUnicode_FSConverter, &opath,
2234 &uid, &gid))
2235 return NULL;
2236 path = PyBytes_AsString(opath);
2237 Py_BEGIN_ALLOW_THREADS
2238 res = lchown(path, (uid_t) uid, (gid_t) gid);
2239 Py_END_ALLOW_THREADS
2240 if (res < 0)
2241 return posix_error_with_allocated_filename(opath);
2242 Py_DECREF(opath);
2243 Py_INCREF(Py_None);
2244 return Py_None;
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00002245}
2246#endif /* HAVE_LCHOWN */
2247
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002248
Guido van Rossum36bc6801995-06-14 22:54:23 +00002249#ifdef HAVE_GETCWD
Barry Warsaw53699e91996-12-10 23:23:01 +00002250static PyObject *
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002251posix_getcwd(int use_bytes)
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002252{
Victor Stinner8c62be82010-05-06 00:08:46 +00002253 char buf[1026];
2254 char *res;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002255
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002256#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00002257 if (!use_bytes) {
2258 wchar_t wbuf[1026];
2259 wchar_t *wbuf2 = wbuf;
2260 PyObject *resobj;
2261 DWORD len;
2262 Py_BEGIN_ALLOW_THREADS
2263 len = GetCurrentDirectoryW(sizeof wbuf/ sizeof wbuf[0], wbuf);
2264 /* If the buffer is large enough, len does not include the
2265 terminating \0. If the buffer is too small, len includes
2266 the space needed for the terminator. */
2267 if (len >= sizeof wbuf/ sizeof wbuf[0]) {
2268 wbuf2 = malloc(len * sizeof(wchar_t));
2269 if (wbuf2)
2270 len = GetCurrentDirectoryW(len, wbuf2);
2271 }
2272 Py_END_ALLOW_THREADS
2273 if (!wbuf2) {
2274 PyErr_NoMemory();
2275 return NULL;
2276 }
2277 if (!len) {
2278 if (wbuf2 != wbuf) free(wbuf2);
2279 return win32_error("getcwdu", NULL);
2280 }
2281 resobj = PyUnicode_FromWideChar(wbuf2, len);
2282 if (wbuf2 != wbuf) free(wbuf2);
2283 return resobj;
2284 }
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002285#endif
2286
Victor Stinner8c62be82010-05-06 00:08:46 +00002287 Py_BEGIN_ALLOW_THREADS
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002288#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00002289 res = _getcwd2(buf, sizeof buf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002290#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002291 res = getcwd(buf, sizeof buf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002292#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00002293 Py_END_ALLOW_THREADS
2294 if (res == NULL)
2295 return posix_error();
2296 if (use_bytes)
2297 return PyBytes_FromStringAndSize(buf, strlen(buf));
Victor Stinner97c18ab2010-05-07 16:34:53 +00002298 return PyUnicode_DecodeFSDefault(buf);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002299}
Guido van Rossumf0af3e32008-10-02 18:55:37 +00002300
2301PyDoc_STRVAR(posix_getcwd__doc__,
2302"getcwd() -> path\n\n\
2303Return a unicode string representing the current working directory.");
2304
2305static PyObject *
2306posix_getcwd_unicode(PyObject *self)
2307{
2308 return posix_getcwd(0);
2309}
2310
2311PyDoc_STRVAR(posix_getcwdb__doc__,
2312"getcwdb() -> path\n\n\
2313Return a bytes string representing the current working directory.");
2314
2315static PyObject *
2316posix_getcwd_bytes(PyObject *self)
2317{
2318 return posix_getcwd(1);
2319}
Guido van Rossum36bc6801995-06-14 22:54:23 +00002320#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002321
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002322
Guido van Rossumb6775db1994-08-01 11:34:53 +00002323#ifdef HAVE_LINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002324PyDoc_STRVAR(posix_link__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00002325"link(src, dst)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002326Create a hard link to a file.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002327
Barry Warsaw53699e91996-12-10 23:23:01 +00002328static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002329posix_link(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002330{
Victor Stinner8c62be82010-05-06 00:08:46 +00002331 return posix_2str(args, "O&O&:link", link);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002332}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002333#endif /* HAVE_LINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002334
Brian Curtin1b9df392010-11-24 20:24:31 +00002335#ifdef MS_WINDOWS
2336PyDoc_STRVAR(win32_link__doc__,
2337"link(src, dst)\n\n\
2338Create a hard link to a file.");
2339
2340static PyObject *
2341win32_link(PyObject *self, PyObject *args)
2342{
2343 PyObject *osrc, *odst;
2344 char *src, *dst;
2345 BOOL rslt;
2346
Brian Curtinfc889c42010-11-28 23:59:46 +00002347 PyUnicodeObject *usrc, *udst;
2348 if (PyArg_ParseTuple(args, "UU:link", &usrc, &udst)) {
2349 Py_BEGIN_ALLOW_THREADS
2350 rslt = CreateHardLinkW(PyUnicode_AS_UNICODE(udst),
2351 PyUnicode_AS_UNICODE(usrc), NULL);
2352 Py_END_ALLOW_THREADS
2353
2354 if (rslt == 0)
2355 return win32_error("link", NULL);
2356
2357 Py_RETURN_NONE;
2358 }
2359
2360 /* Narrow strings also valid. */
2361 PyErr_Clear();
2362
Brian Curtin1b9df392010-11-24 20:24:31 +00002363 if (!PyArg_ParseTuple(args, "O&O&:link", PyUnicode_FSConverter, &osrc,
2364 PyUnicode_FSConverter, &odst))
2365 return NULL;
2366
2367 src = PyBytes_AsString(osrc);
2368 dst = PyBytes_AsString(odst);
2369
2370 Py_BEGIN_ALLOW_THREADS
Brian Curtinfc889c42010-11-28 23:59:46 +00002371 rslt = CreateHardLinkA(dst, src, NULL);
Brian Curtin1b9df392010-11-24 20:24:31 +00002372 Py_END_ALLOW_THREADS
2373
Stefan Krah30b341f2010-11-27 11:44:18 +00002374 Py_DECREF(osrc);
2375 Py_DECREF(odst);
Brian Curtin1b9df392010-11-24 20:24:31 +00002376 if (rslt == 0)
Brian Curtinfc889c42010-11-28 23:59:46 +00002377 return win32_error("link", NULL);
Brian Curtin1b9df392010-11-24 20:24:31 +00002378
2379 Py_RETURN_NONE;
2380}
2381#endif /* MS_WINDOWS */
2382
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002383
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002384PyDoc_STRVAR(posix_listdir__doc__,
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002385"listdir([path]) -> list_of_strings\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002386Return a list containing the names of the entries in the directory.\n\
2387\n\
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002388 path: path of directory to list (default: '.')\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002389\n\
2390The list is in arbitrary order. It does not include the special\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002391entries '.' and '..' even if they are present in the directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00002392
Barry Warsaw53699e91996-12-10 23:23:01 +00002393static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00002394posix_listdir(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +00002395{
Victor Stinner8c62be82010-05-06 00:08:46 +00002396 /* XXX Should redo this putting the (now four) versions of opendir
2397 in separate files instead of having them all here... */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002398#if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002399
Victor Stinner8c62be82010-05-06 00:08:46 +00002400 PyObject *d, *v;
2401 HANDLE hFindFile;
2402 BOOL result;
2403 WIN32_FIND_DATA FileData;
2404 PyObject *opath;
2405 char namebuf[MAX_PATH+5]; /* Overallocate for \\*.*\0 */
2406 char *bufptr = namebuf;
2407 Py_ssize_t len = sizeof(namebuf)-5; /* only claim to have space for MAX_PATH */
Guido van Rossumb6775db1994-08-01 11:34:53 +00002408
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002409 PyObject *po = NULL;
2410 if (PyArg_ParseTuple(args, "|U:listdir", &po)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002411 WIN32_FIND_DATAW wFileData;
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002412 Py_UNICODE *wnamebuf, *po_wchars;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00002413
Antoine Pitrou3d330ad2010-09-14 10:08:08 +00002414 if (po == NULL) { /* Default arg: "." */
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002415 po_wchars = L".";
2416 len = 1;
2417 } else {
2418 po_wchars = PyUnicode_AS_UNICODE(po);
2419 len = PyUnicode_GET_SIZE(po);
2420 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002421 /* Overallocate for \\*.*\0 */
Victor Stinner8c62be82010-05-06 00:08:46 +00002422 wnamebuf = malloc((len + 5) * sizeof(wchar_t));
2423 if (!wnamebuf) {
2424 PyErr_NoMemory();
2425 return NULL;
2426 }
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002427 wcscpy(wnamebuf, po_wchars);
Victor Stinner8c62be82010-05-06 00:08:46 +00002428 if (len > 0) {
2429 Py_UNICODE wch = wnamebuf[len-1];
2430 if (wch != L'/' && wch != L'\\' && wch != L':')
2431 wnamebuf[len++] = L'\\';
2432 wcscpy(wnamebuf + len, L"*.*");
2433 }
2434 if ((d = PyList_New(0)) == NULL) {
2435 free(wnamebuf);
2436 return NULL;
2437 }
Antoine Pitroub73caab2010-08-09 23:39:31 +00002438 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002439 hFindFile = FindFirstFileW(wnamebuf, &wFileData);
Antoine Pitroub73caab2010-08-09 23:39:31 +00002440 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002441 if (hFindFile == INVALID_HANDLE_VALUE) {
2442 int error = GetLastError();
2443 if (error == ERROR_FILE_NOT_FOUND) {
2444 free(wnamebuf);
2445 return d;
2446 }
2447 Py_DECREF(d);
2448 win32_error_unicode("FindFirstFileW", wnamebuf);
2449 free(wnamebuf);
2450 return NULL;
2451 }
2452 do {
2453 /* Skip over . and .. */
2454 if (wcscmp(wFileData.cFileName, L".") != 0 &&
2455 wcscmp(wFileData.cFileName, L"..") != 0) {
2456 v = PyUnicode_FromUnicode(wFileData.cFileName, wcslen(wFileData.cFileName));
2457 if (v == NULL) {
2458 Py_DECREF(d);
2459 d = NULL;
2460 break;
2461 }
2462 if (PyList_Append(d, v) != 0) {
2463 Py_DECREF(v);
2464 Py_DECREF(d);
2465 d = NULL;
2466 break;
2467 }
2468 Py_DECREF(v);
2469 }
2470 Py_BEGIN_ALLOW_THREADS
2471 result = FindNextFileW(hFindFile, &wFileData);
2472 Py_END_ALLOW_THREADS
2473 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2474 it got to the end of the directory. */
2475 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2476 Py_DECREF(d);
2477 win32_error_unicode("FindNextFileW", wnamebuf);
2478 FindClose(hFindFile);
2479 free(wnamebuf);
2480 return NULL;
2481 }
2482 } while (result == TRUE);
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002483
Victor Stinner8c62be82010-05-06 00:08:46 +00002484 if (FindClose(hFindFile) == FALSE) {
2485 Py_DECREF(d);
2486 win32_error_unicode("FindClose", wnamebuf);
2487 free(wnamebuf);
2488 return NULL;
2489 }
2490 free(wnamebuf);
2491 return d;
2492 }
2493 /* Drop the argument parsing error as narrow strings
2494 are also valid. */
2495 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002496
Victor Stinner8c62be82010-05-06 00:08:46 +00002497 if (!PyArg_ParseTuple(args, "O&:listdir",
2498 PyUnicode_FSConverter, &opath))
2499 return NULL;
2500 if (PyBytes_GET_SIZE(opath)+1 > MAX_PATH) {
2501 PyErr_SetString(PyExc_ValueError, "path too long");
2502 Py_DECREF(opath);
2503 return NULL;
2504 }
2505 strcpy(namebuf, PyBytes_AsString(opath));
2506 len = PyObject_Size(opath);
Stefan Krah2a7feee2010-11-27 22:06:49 +00002507 Py_DECREF(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00002508 if (len > 0) {
2509 char ch = namebuf[len-1];
2510 if (ch != SEP && ch != ALTSEP && ch != ':')
2511 namebuf[len++] = '/';
2512 strcpy(namebuf + len, "*.*");
2513 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002514
Victor Stinner8c62be82010-05-06 00:08:46 +00002515 if ((d = PyList_New(0)) == NULL)
2516 return NULL;
Guido van Rossumb6775db1994-08-01 11:34:53 +00002517
Antoine Pitroub73caab2010-08-09 23:39:31 +00002518 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002519 hFindFile = FindFirstFile(namebuf, &FileData);
Antoine Pitroub73caab2010-08-09 23:39:31 +00002520 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002521 if (hFindFile == INVALID_HANDLE_VALUE) {
2522 int error = GetLastError();
2523 if (error == ERROR_FILE_NOT_FOUND)
2524 return d;
2525 Py_DECREF(d);
2526 return win32_error("FindFirstFile", namebuf);
2527 }
2528 do {
2529 /* Skip over . and .. */
2530 if (strcmp(FileData.cFileName, ".") != 0 &&
2531 strcmp(FileData.cFileName, "..") != 0) {
2532 v = PyBytes_FromString(FileData.cFileName);
2533 if (v == NULL) {
2534 Py_DECREF(d);
2535 d = NULL;
2536 break;
2537 }
2538 if (PyList_Append(d, v) != 0) {
2539 Py_DECREF(v);
2540 Py_DECREF(d);
2541 d = NULL;
2542 break;
2543 }
2544 Py_DECREF(v);
2545 }
2546 Py_BEGIN_ALLOW_THREADS
2547 result = FindNextFile(hFindFile, &FileData);
2548 Py_END_ALLOW_THREADS
2549 /* FindNextFile sets error to ERROR_NO_MORE_FILES if
2550 it got to the end of the directory. */
2551 if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
2552 Py_DECREF(d);
2553 win32_error("FindNextFile", namebuf);
2554 FindClose(hFindFile);
2555 return NULL;
2556 }
2557 } while (result == TRUE);
Guido van Rossumb6775db1994-08-01 11:34:53 +00002558
Victor Stinner8c62be82010-05-06 00:08:46 +00002559 if (FindClose(hFindFile) == FALSE) {
2560 Py_DECREF(d);
2561 return win32_error("FindClose", namebuf);
2562 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00002563
Victor Stinner8c62be82010-05-06 00:08:46 +00002564 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002565
Tim Peters0bb44a42000-09-15 07:44:49 +00002566#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002567
2568#ifndef MAX_PATH
2569#define MAX_PATH CCHMAXPATH
2570#endif
Martin v. Löwis011e8422009-05-05 04:43:17 +00002571 PyObject *oname;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002572 char *name, *pt;
Thomas Wouters68bc4f92006-03-01 01:05:10 +00002573 Py_ssize_t len;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002574 PyObject *d, *v;
2575 char namebuf[MAX_PATH+5];
2576 HDIR hdir = 1;
2577 ULONG srchcnt = 1;
2578 FILEFINDBUF3 ep;
2579 APIRET rc;
2580
Victor Stinner8c62be82010-05-06 00:08:46 +00002581 if (!PyArg_ParseTuple(args, "O&:listdir",
Martin v. Löwis011e8422009-05-05 04:43:17 +00002582 PyUnicode_FSConverter, &oname))
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002583 return NULL;
Victor Stinnerdcb24032010-04-22 12:08:36 +00002584 name = PyBytes_AsString(oname);
2585 len = PyBytes_GET_SIZE(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002586 if (len >= MAX_PATH) {
Victor Stinnerdcb24032010-04-22 12:08:36 +00002587 Py_DECREF(oname);
Neal Norwitz6c913782007-10-14 03:23:09 +00002588 PyErr_SetString(PyExc_ValueError, "path too long");
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002589 return NULL;
2590 }
2591 strcpy(namebuf, name);
2592 for (pt = namebuf; *pt; pt++)
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002593 if (*pt == ALTSEP)
2594 *pt = SEP;
2595 if (namebuf[len-1] != SEP)
2596 namebuf[len++] = SEP;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002597 strcpy(namebuf + len, "*.*");
2598
Neal Norwitz6c913782007-10-14 03:23:09 +00002599 if ((d = PyList_New(0)) == NULL) {
Victor Stinnerdcb24032010-04-22 12:08:36 +00002600 Py_DECREF(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002601 return NULL;
Alexandre Vassalotti4167ebc2007-10-14 02:54:41 +00002602 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002603
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002604 rc = DosFindFirst(namebuf, /* Wildcard Pattern to Match */
2605 &hdir, /* Handle to Use While Search Directory */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002606 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002607 &ep, sizeof(ep), /* Structure to Receive Directory Entry */
2608 &srchcnt, /* Max and Actual Count of Entries Per Iteration */
2609 FIL_STANDARD); /* Format of Entry (EAs or Not) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002610
2611 if (rc != NO_ERROR) {
2612 errno = ENOENT;
Martin v. Löwis011e8422009-05-05 04:43:17 +00002613 return posix_error_with_allocated_filename(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002614 }
2615
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002616 if (srchcnt > 0) { /* If Directory is NOT Totally Empty, */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002617 do {
2618 if (ep.achName[0] == '.'
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00002619 && (ep.achName[1] == '\0' || (ep.achName[1] == '.' && ep.achName[2] == '\0')))
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002620 continue; /* Skip Over "." and ".." Names */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002621
2622 strcpy(namebuf, ep.achName);
2623
Guido van Rossumc5a0f531997-12-02 20:36:02 +00002624 /* Leave Case of Name Alone -- In Native Form */
2625 /* (Removed Forced Lowercasing Code) */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002626
Christian Heimes72b710a2008-05-26 13:28:38 +00002627 v = PyBytes_FromString(namebuf);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002628 if (v == NULL) {
2629 Py_DECREF(d);
2630 d = NULL;
2631 break;
2632 }
2633 if (PyList_Append(d, v) != 0) {
2634 Py_DECREF(v);
2635 Py_DECREF(d);
2636 d = NULL;
2637 break;
2638 }
2639 Py_DECREF(v);
2640 } while (DosFindNext(hdir, &ep, sizeof(ep), &srchcnt) == NO_ERROR && srchcnt > 0);
2641 }
2642
Victor Stinnerdcb24032010-04-22 12:08:36 +00002643 Py_DECREF(oname);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00002644 return d;
2645#else
Victor Stinner8c62be82010-05-06 00:08:46 +00002646 PyObject *oname;
2647 char *name;
2648 PyObject *d, *v;
2649 DIR *dirp;
2650 struct dirent *ep;
2651 int arg_is_unicode = 1;
Just van Rossum96b1c902003-03-03 17:32:15 +00002652
Victor Stinner8c62be82010-05-06 00:08:46 +00002653 errno = 0;
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002654 /* v is never read, so it does not need to be initialized yet. */
2655 if (!PyArg_ParseTuple(args, "|U:listdir", &v)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002656 arg_is_unicode = 0;
2657 PyErr_Clear();
2658 }
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002659 oname = NULL;
2660 if (!PyArg_ParseTuple(args, "|O&:listdir", PyUnicode_FSConverter, &oname))
Victor Stinner8c62be82010-05-06 00:08:46 +00002661 return NULL;
Antoine Pitrou3d330ad2010-09-14 10:08:08 +00002662 if (oname == NULL) { /* Default arg: "." */
Stefan Krah0e803b32010-11-26 16:16:47 +00002663 oname = PyBytes_FromString(".");
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +00002664 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002665 name = PyBytes_AsString(oname);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002666 Py_BEGIN_ALLOW_THREADS
2667 dirp = opendir(name);
2668 Py_END_ALLOW_THREADS
2669 if (dirp == NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00002670 return posix_error_with_allocated_filename(oname);
2671 }
2672 if ((d = PyList_New(0)) == NULL) {
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002673 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002674 closedir(dirp);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002675 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002676 Py_DECREF(oname);
2677 return NULL;
2678 }
2679 for (;;) {
2680 errno = 0;
2681 Py_BEGIN_ALLOW_THREADS
2682 ep = readdir(dirp);
2683 Py_END_ALLOW_THREADS
2684 if (ep == NULL) {
2685 if (errno == 0) {
2686 break;
2687 } else {
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002688 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002689 closedir(dirp);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002690 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002691 Py_DECREF(d);
2692 return posix_error_with_allocated_filename(oname);
2693 }
2694 }
2695 if (ep->d_name[0] == '.' &&
2696 (NAMLEN(ep) == 1 ||
2697 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
2698 continue;
Victor Stinnera45598a2010-05-14 16:35:39 +00002699 if (arg_is_unicode)
2700 v = PyUnicode_DecodeFSDefaultAndSize(ep->d_name, NAMLEN(ep));
2701 else
2702 v = PyBytes_FromStringAndSize(ep->d_name, NAMLEN(ep));
Victor Stinner8c62be82010-05-06 00:08:46 +00002703 if (v == NULL) {
Victor Stinnera45598a2010-05-14 16:35:39 +00002704 Py_CLEAR(d);
Victor Stinner8c62be82010-05-06 00:08:46 +00002705 break;
2706 }
Victor Stinner8c62be82010-05-06 00:08:46 +00002707 if (PyList_Append(d, v) != 0) {
2708 Py_DECREF(v);
Victor Stinnera45598a2010-05-14 16:35:39 +00002709 Py_CLEAR(d);
Victor Stinner8c62be82010-05-06 00:08:46 +00002710 break;
2711 }
2712 Py_DECREF(v);
2713 }
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002714 Py_BEGIN_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002715 closedir(dirp);
Antoine Pitroud3ccde82010-09-04 17:21:57 +00002716 Py_END_ALLOW_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00002717 Py_DECREF(oname);
Guido van Rossum0ee42cd1991-04-08 21:01:03 +00002718
Victor Stinner8c62be82010-05-06 00:08:46 +00002719 return d;
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002720
Tim Peters0bb44a42000-09-15 07:44:49 +00002721#endif /* which OS */
2722} /* end of posix_listdir */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002723
Antoine Pitrou8250e232011-02-25 23:41:16 +00002724#ifdef HAVE_FDOPENDIR
2725PyDoc_STRVAR(posix_fdlistdir__doc__,
2726"fdlistdir(fd) -> list_of_strings\n\n\
2727Like listdir(), but uses a file descriptor instead.\n\
2728After succesful execution of this function, fd will be closed.");
2729
2730static PyObject *
2731posix_fdlistdir(PyObject *self, PyObject *args)
2732{
2733 PyObject *d, *v;
2734 DIR *dirp;
2735 struct dirent *ep;
2736 int fd;
2737
2738 errno = 0;
2739 if (!PyArg_ParseTuple(args, "i:fdlistdir", &fd))
2740 return NULL;
2741 Py_BEGIN_ALLOW_THREADS
2742 dirp = fdopendir(fd);
2743 Py_END_ALLOW_THREADS
2744 if (dirp == NULL) {
2745 close(fd);
2746 return posix_error();
2747 }
2748 if ((d = PyList_New(0)) == NULL) {
2749 Py_BEGIN_ALLOW_THREADS
2750 closedir(dirp);
2751 Py_END_ALLOW_THREADS
2752 return NULL;
2753 }
2754 for (;;) {
2755 errno = 0;
2756 Py_BEGIN_ALLOW_THREADS
2757 ep = readdir(dirp);
2758 Py_END_ALLOW_THREADS
2759 if (ep == NULL) {
2760 if (errno == 0) {
2761 break;
2762 } else {
2763 Py_BEGIN_ALLOW_THREADS
2764 closedir(dirp);
2765 Py_END_ALLOW_THREADS
2766 Py_DECREF(d);
2767 return posix_error();
2768 }
2769 }
2770 if (ep->d_name[0] == '.' &&
2771 (NAMLEN(ep) == 1 ||
2772 (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
2773 continue;
2774 v = PyUnicode_DecodeFSDefaultAndSize(ep->d_name, NAMLEN(ep));
2775 if (v == NULL) {
2776 Py_CLEAR(d);
2777 break;
2778 }
2779 if (PyList_Append(d, v) != 0) {
2780 Py_DECREF(v);
2781 Py_CLEAR(d);
2782 break;
2783 }
2784 Py_DECREF(v);
2785 }
2786 Py_BEGIN_ALLOW_THREADS
2787 closedir(dirp);
2788 Py_END_ALLOW_THREADS
2789
2790 return d;
2791}
2792#endif
2793
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002794#ifdef MS_WINDOWS
Mark Hammondef8b6542001-05-13 08:04:26 +00002795/* A helper function for abspath on win32 */
2796static PyObject *
2797posix__getfullpathname(PyObject *self, PyObject *args)
2798{
Victor Stinner8c62be82010-05-06 00:08:46 +00002799 PyObject *opath;
2800 char *path;
2801 char outbuf[MAX_PATH*2];
2802 char *temp;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00002803#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00002804 PyUnicodeObject *po;
2805 if (PyArg_ParseTuple(args, "U|:_getfullpathname", &po)) {
2806 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
2807 Py_UNICODE woutbuf[MAX_PATH*2], *woutbufp = woutbuf;
2808 Py_UNICODE *wtemp;
2809 DWORD result;
2810 PyObject *v;
2811 result = GetFullPathNameW(wpath,
2812 sizeof(woutbuf)/sizeof(woutbuf[0]),
2813 woutbuf, &wtemp);
2814 if (result > sizeof(woutbuf)/sizeof(woutbuf[0])) {
2815 woutbufp = malloc(result * sizeof(Py_UNICODE));
2816 if (!woutbufp)
2817 return PyErr_NoMemory();
2818 result = GetFullPathNameW(wpath, result, woutbufp, &wtemp);
2819 }
2820 if (result)
2821 v = PyUnicode_FromUnicode(woutbufp, wcslen(woutbufp));
2822 else
2823 v = win32_error_unicode("GetFullPathNameW", wpath);
2824 if (woutbufp != woutbuf)
2825 free(woutbufp);
2826 return v;
2827 }
2828 /* Drop the argument parsing error as narrow strings
2829 are also valid. */
2830 PyErr_Clear();
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00002831
Mark Hammondc2e85bd2002-10-03 05:10:39 +00002832#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00002833 if (!PyArg_ParseTuple (args, "O&:_getfullpathname",
2834 PyUnicode_FSConverter, &opath))
2835 return NULL;
2836 path = PyBytes_AsString(opath);
2837 if (!GetFullPathName(path, sizeof(outbuf)/sizeof(outbuf[0]),
2838 outbuf, &temp)) {
2839 win32_error("GetFullPathName", path);
2840 Py_DECREF(opath);
2841 return NULL;
2842 }
2843 Py_DECREF(opath);
2844 if (PyUnicode_Check(PyTuple_GetItem(args, 0))) {
2845 return PyUnicode_Decode(outbuf, strlen(outbuf),
2846 Py_FileSystemDefaultEncoding, NULL);
2847 }
2848 return PyBytes_FromString(outbuf);
Mark Hammondef8b6542001-05-13 08:04:26 +00002849} /* end of posix__getfullpathname */
Brian Curtind40e6f72010-07-08 21:39:08 +00002850
Brian Curtinf5e76d02010-11-24 13:14:05 +00002851/* Grab GetFinalPathNameByHandle dynamically from kernel32 */
2852static int has_GetFinalPathNameByHandle = 0;
2853static DWORD (CALLBACK *Py_GetFinalPathNameByHandleA)(HANDLE, LPSTR, DWORD,
2854 DWORD);
2855static DWORD (CALLBACK *Py_GetFinalPathNameByHandleW)(HANDLE, LPWSTR, DWORD,
2856 DWORD);
2857static int
2858check_GetFinalPathNameByHandle()
2859{
2860 HINSTANCE hKernel32;
2861 /* only recheck */
2862 if (!has_GetFinalPathNameByHandle)
2863 {
2864 hKernel32 = GetModuleHandle("KERNEL32");
2865 *(FARPROC*)&Py_GetFinalPathNameByHandleA = GetProcAddress(hKernel32,
2866 "GetFinalPathNameByHandleA");
2867 *(FARPROC*)&Py_GetFinalPathNameByHandleW = GetProcAddress(hKernel32,
2868 "GetFinalPathNameByHandleW");
2869 has_GetFinalPathNameByHandle = Py_GetFinalPathNameByHandleA &&
2870 Py_GetFinalPathNameByHandleW;
2871 }
2872 return has_GetFinalPathNameByHandle;
2873}
2874
Brian Curtind40e6f72010-07-08 21:39:08 +00002875/* A helper function for samepath on windows */
2876static PyObject *
2877posix__getfinalpathname(PyObject *self, PyObject *args)
2878{
2879 HANDLE hFile;
2880 int buf_size;
2881 wchar_t *target_path;
2882 int result_length;
2883 PyObject *result;
2884 wchar_t *path;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00002885
Brian Curtin94622b02010-09-24 00:03:39 +00002886 if (!PyArg_ParseTuple(args, "u|:_getfinalpathname", &path)) {
Brian Curtind40e6f72010-07-08 21:39:08 +00002887 return NULL;
2888 }
2889
2890 if(!check_GetFinalPathNameByHandle()) {
2891 /* If the OS doesn't have GetFinalPathNameByHandle, return a
2892 NotImplementedError. */
2893 return PyErr_Format(PyExc_NotImplementedError,
2894 "GetFinalPathNameByHandle not available on this platform");
2895 }
2896
2897 hFile = CreateFileW(
2898 path,
2899 0, /* desired access */
2900 0, /* share mode */
2901 NULL, /* security attributes */
2902 OPEN_EXISTING,
2903 /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
2904 FILE_FLAG_BACKUP_SEMANTICS,
2905 NULL);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00002906
Brian Curtind40e6f72010-07-08 21:39:08 +00002907 if(hFile == INVALID_HANDLE_VALUE) {
2908 return win32_error_unicode("GetFinalPathNamyByHandle", path);
Brian Curtin74e45612010-07-09 15:58:59 +00002909 return PyErr_Format(PyExc_RuntimeError,
2910 "Could not get a handle to file.");
Brian Curtind40e6f72010-07-08 21:39:08 +00002911 }
2912
2913 /* We have a good handle to the target, use it to determine the
2914 target path name. */
2915 buf_size = Py_GetFinalPathNameByHandleW(hFile, 0, 0, VOLUME_NAME_NT);
2916
2917 if(!buf_size)
2918 return win32_error_unicode("GetFinalPathNameByHandle", path);
2919
2920 target_path = (wchar_t *)malloc((buf_size+1)*sizeof(wchar_t));
2921 if(!target_path)
2922 return PyErr_NoMemory();
2923
2924 result_length = Py_GetFinalPathNameByHandleW(hFile, target_path,
2925 buf_size, VOLUME_NAME_DOS);
2926 if(!result_length)
2927 return win32_error_unicode("GetFinalPathNamyByHandle", path);
2928
2929 if(!CloseHandle(hFile))
2930 return win32_error_unicode("GetFinalPathNameByHandle", path);
2931
2932 target_path[result_length] = 0;
2933 result = PyUnicode_FromUnicode(target_path, result_length);
2934 free(target_path);
2935 return result;
2936
2937} /* end of posix__getfinalpathname */
Brian Curtin62857742010-09-06 17:07:27 +00002938
2939static PyObject *
2940posix__getfileinformation(PyObject *self, PyObject *args)
2941{
2942 HANDLE hFile;
2943 BY_HANDLE_FILE_INFORMATION info;
2944 int fd;
2945
2946 if (!PyArg_ParseTuple(args, "i:_getfileinformation", &fd))
2947 return NULL;
2948
Hirokazu Yamamoto26253bb2010-12-05 04:16:47 +00002949 if (!_PyVerify_fd(fd))
2950 return posix_error();
Brian Curtin62857742010-09-06 17:07:27 +00002951
2952 hFile = (HANDLE)_get_osfhandle(fd);
2953 if (hFile == INVALID_HANDLE_VALUE)
Hirokazu Yamamoto26253bb2010-12-05 04:16:47 +00002954 return posix_error();
Brian Curtin62857742010-09-06 17:07:27 +00002955
2956 if (!GetFileInformationByHandle(hFile, &info))
2957 return win32_error("_getfileinformation", NULL);
2958
2959 return Py_BuildValue("iii", info.dwVolumeSerialNumber,
2960 info.nFileIndexHigh,
2961 info.nFileIndexLow);
2962}
Brian Curtin9c669cc2011-06-08 18:17:18 -05002963
2964static PyObject *
2965posix__isdir(PyObject *self, PyObject *args)
2966{
2967 PyObject *opath;
2968 char *path;
2969 PyUnicodeObject *po;
2970 DWORD attributes;
2971
2972 if (PyArg_ParseTuple(args, "U|:_isdir", &po)) {
2973 Py_UNICODE *wpath = PyUnicode_AS_UNICODE(po);
2974
2975 attributes = GetFileAttributesW(wpath);
2976 if (attributes == INVALID_FILE_ATTRIBUTES)
2977 Py_RETURN_FALSE;
2978 goto check;
2979 }
2980 /* Drop the argument parsing error as narrow strings
2981 are also valid. */
2982 PyErr_Clear();
2983
2984 if (!PyArg_ParseTuple(args, "O&:_isdir",
2985 PyUnicode_FSConverter, &opath))
2986 return NULL;
2987
2988 path = PyBytes_AsString(opath);
2989 attributes = GetFileAttributesA(path);
2990 if (attributes == INVALID_FILE_ATTRIBUTES)
2991 Py_RETURN_FALSE;
2992
2993check:
2994 if (attributes & FILE_ATTRIBUTE_DIRECTORY)
2995 Py_RETURN_TRUE;
2996 else
2997 Py_RETURN_FALSE;
2998}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002999#endif /* MS_WINDOWS */
Mark Hammondef8b6542001-05-13 08:04:26 +00003000
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003001PyDoc_STRVAR(posix_mkdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003002"mkdir(path [, mode=0777])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003003Create a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003004
Barry Warsaw53699e91996-12-10 23:23:01 +00003005static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003006posix_mkdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003007{
Victor Stinner8c62be82010-05-06 00:08:46 +00003008 int res;
3009 PyObject *opath;
3010 char *path;
3011 int mode = 0777;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003012
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003013#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00003014 PyUnicodeObject *po;
3015 if (PyArg_ParseTuple(args, "U|i:mkdir", &po, &mode)) {
3016 Py_BEGIN_ALLOW_THREADS
3017 /* PyUnicode_AS_UNICODE OK without thread lock as
3018 it is a simple dereference. */
3019 res = CreateDirectoryW(PyUnicode_AS_UNICODE(po), NULL);
3020 Py_END_ALLOW_THREADS
3021 if (!res)
3022 return win32_error_unicode("mkdir", PyUnicode_AS_UNICODE(po));
3023 Py_INCREF(Py_None);
3024 return Py_None;
3025 }
3026 /* Drop the argument parsing error as narrow strings
3027 are also valid. */
3028 PyErr_Clear();
3029 if (!PyArg_ParseTuple(args, "O&|i:mkdir",
3030 PyUnicode_FSConverter, &opath, &mode))
3031 return NULL;
3032 path = PyBytes_AsString(opath);
3033 Py_BEGIN_ALLOW_THREADS
3034 /* PyUnicode_AS_UNICODE OK without thread lock as
3035 it is a simple dereference. */
3036 res = CreateDirectoryA(path, NULL);
3037 Py_END_ALLOW_THREADS
3038 if (!res) {
3039 win32_error("mkdir", path);
3040 Py_DECREF(opath);
3041 return NULL;
3042 }
3043 Py_DECREF(opath);
3044 Py_INCREF(Py_None);
3045 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003046#else
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003047
Victor Stinner8c62be82010-05-06 00:08:46 +00003048 if (!PyArg_ParseTuple(args, "O&|i:mkdir",
3049 PyUnicode_FSConverter, &opath, &mode))
3050 return NULL;
3051 path = PyBytes_AsString(opath);
3052 Py_BEGIN_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00003053#if ( defined(__WATCOMC__) || defined(PYCC_VACPP) ) && !defined(__QNX__)
Victor Stinner8c62be82010-05-06 00:08:46 +00003054 res = mkdir(path);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003055#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003056 res = mkdir(path, mode);
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003057#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003058 Py_END_ALLOW_THREADS
3059 if (res < 0)
3060 return posix_error_with_allocated_filename(opath);
3061 Py_DECREF(opath);
3062 Py_INCREF(Py_None);
3063 return Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003064#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003065}
3066
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003067
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003068/* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
3069#if defined(HAVE_SYS_RESOURCE_H)
Thomas Wouterse38b2f12001-07-11 22:35:31 +00003070#include <sys/resource.h>
3071#endif
Thomas Wouterse38b2f12001-07-11 22:35:31 +00003072
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003073
3074#ifdef HAVE_NICE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003075PyDoc_STRVAR(posix_nice__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003076"nice(inc) -> new_priority\n\n\
3077Decrease the priority of process by inc and return the new priority.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003078
Barry Warsaw53699e91996-12-10 23:23:01 +00003079static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003080posix_nice(PyObject *self, PyObject *args)
Guido van Rossum775f4da1993-01-09 17:18:52 +00003081{
Victor Stinner8c62be82010-05-06 00:08:46 +00003082 int increment, value;
Guido van Rossum775f4da1993-01-09 17:18:52 +00003083
Victor Stinner8c62be82010-05-06 00:08:46 +00003084 if (!PyArg_ParseTuple(args, "i:nice", &increment))
3085 return NULL;
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00003086
Victor Stinner8c62be82010-05-06 00:08:46 +00003087 /* There are two flavours of 'nice': one that returns the new
3088 priority (as required by almost all standards out there) and the
3089 Linux/FreeBSD/BSDI one, which returns '0' on success and advices
3090 the use of getpriority() to get the new priority.
Tim Peters5aa91602002-01-30 05:46:57 +00003091
Victor Stinner8c62be82010-05-06 00:08:46 +00003092 If we are of the nice family that returns the new priority, we
3093 need to clear errno before the call, and check if errno is filled
3094 before calling posix_error() on a returnvalue of -1, because the
3095 -1 may be the actual new priority! */
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00003096
Victor Stinner8c62be82010-05-06 00:08:46 +00003097 errno = 0;
3098 value = nice(increment);
Thomas Wouterse38b2f12001-07-11 22:35:31 +00003099#if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
Victor Stinner8c62be82010-05-06 00:08:46 +00003100 if (value == 0)
3101 value = getpriority(PRIO_PROCESS, 0);
Thomas Woutersc2c12dc2001-07-11 14:45:34 +00003102#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003103 if (value == -1 && errno != 0)
3104 /* either nice() or getpriority() returned an error */
3105 return posix_error();
3106 return PyLong_FromLong((long) value);
Guido van Rossum775f4da1993-01-09 17:18:52 +00003107}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003108#endif /* HAVE_NICE */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003109
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003110
3111#ifdef HAVE_GETPRIORITY
3112PyDoc_STRVAR(posix_getpriority__doc__,
3113"getpriority(which, who) -> current_priority\n\n\
3114Get program scheduling priority.");
3115
3116static PyObject *
3117posix_getpriority(PyObject *self, PyObject *args)
3118{
3119 int which, who, retval;
3120
3121 if (!PyArg_ParseTuple(args, "ii", &which, &who))
3122 return NULL;
3123 errno = 0;
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003124 retval = getpriority(which, who);
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003125 if (errno != 0)
3126 return posix_error();
3127 return PyLong_FromLong((long)retval);
3128}
3129#endif /* HAVE_GETPRIORITY */
3130
3131
3132#ifdef HAVE_SETPRIORITY
3133PyDoc_STRVAR(posix_setpriority__doc__,
3134"setpriority(which, who, prio) -> None\n\n\
3135Set program scheduling priority.");
3136
3137static PyObject *
3138posix_setpriority(PyObject *self, PyObject *args)
3139{
3140 int which, who, prio, retval;
3141
3142 if (!PyArg_ParseTuple(args, "iii", &which, &who, &prio))
3143 return NULL;
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003144 retval = setpriority(which, who, prio);
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00003145 if (retval == -1)
3146 return posix_error();
3147 Py_RETURN_NONE;
3148}
3149#endif /* HAVE_SETPRIORITY */
3150
3151
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003152PyDoc_STRVAR(posix_rename__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003153"rename(old, new)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003154Rename a file or directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003155
Barry Warsaw53699e91996-12-10 23:23:01 +00003156static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003157posix_rename(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003158{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003159#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00003160 PyObject *o1, *o2;
3161 char *p1, *p2;
3162 BOOL result;
3163 if (!PyArg_ParseTuple(args, "OO:rename", &o1, &o2))
3164 goto error;
3165 if (!convert_to_unicode(&o1))
3166 goto error;
3167 if (!convert_to_unicode(&o2)) {
3168 Py_DECREF(o1);
3169 goto error;
3170 }
3171 Py_BEGIN_ALLOW_THREADS
3172 result = MoveFileW(PyUnicode_AsUnicode(o1),
3173 PyUnicode_AsUnicode(o2));
3174 Py_END_ALLOW_THREADS
3175 Py_DECREF(o1);
3176 Py_DECREF(o2);
3177 if (!result)
3178 return win32_error("rename", NULL);
3179 Py_INCREF(Py_None);
3180 return Py_None;
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00003181error:
Victor Stinner8c62be82010-05-06 00:08:46 +00003182 PyErr_Clear();
3183 if (!PyArg_ParseTuple(args, "ss:rename", &p1, &p2))
3184 return NULL;
3185 Py_BEGIN_ALLOW_THREADS
3186 result = MoveFileA(p1, p2);
3187 Py_END_ALLOW_THREADS
3188 if (!result)
3189 return win32_error("rename", NULL);
3190 Py_INCREF(Py_None);
3191 return Py_None;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003192#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003193 return posix_2str(args, "O&O&:rename", rename);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003194#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003195}
3196
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003197
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003198PyDoc_STRVAR(posix_rmdir__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003199"rmdir(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003200Remove a directory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003201
Barry Warsaw53699e91996-12-10 23:23:01 +00003202static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003203posix_rmdir(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003204{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003205#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00003206 return win32_1str(args, "rmdir", "y:rmdir", RemoveDirectoryA, "U:rmdir", RemoveDirectoryW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003207#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003208 return posix_1str(args, "O&:rmdir", rmdir);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003209#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003210}
3211
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003212
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003213PyDoc_STRVAR(posix_stat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003214"stat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003215Perform a stat system call on the given path.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003216
Barry Warsaw53699e91996-12-10 23:23:01 +00003217static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003218posix_stat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003219{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003220#ifdef MS_WINDOWS
Brian Curtind40e6f72010-07-08 21:39:08 +00003221 return posix_do_stat(self, args, "O&:stat", STAT, "U:stat", win32_stat_w);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003222#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003223 return posix_do_stat(self, args, "O&:stat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003224#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003225}
3226
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003227
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003228#ifdef HAVE_SYSTEM
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003229PyDoc_STRVAR(posix_system__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003230"system(command) -> exit_status\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003231Execute the command (a string) in a subshell.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003232
Barry Warsaw53699e91996-12-10 23:23:01 +00003233static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003234posix_system(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003235{
Victor Stinner8c62be82010-05-06 00:08:46 +00003236 long sts;
Amaury Forgeot d'Arc90ebd3e2007-11-20 01:52:14 +00003237#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00003238 wchar_t *command;
3239 if (!PyArg_ParseTuple(args, "u:system", &command))
3240 return NULL;
Victor Stinnercfa72782010-04-16 11:45:13 +00003241
Victor Stinner8c62be82010-05-06 00:08:46 +00003242 Py_BEGIN_ALLOW_THREADS
3243 sts = _wsystem(command);
3244 Py_END_ALLOW_THREADS
Victor Stinnercfa72782010-04-16 11:45:13 +00003245#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003246 PyObject *command_obj;
3247 char *command;
3248 if (!PyArg_ParseTuple(args, "O&:system",
3249 PyUnicode_FSConverter, &command_obj))
3250 return NULL;
Victor Stinnercfa72782010-04-16 11:45:13 +00003251
Victor Stinner8c62be82010-05-06 00:08:46 +00003252 command = PyBytes_AsString(command_obj);
3253 Py_BEGIN_ALLOW_THREADS
3254 sts = system(command);
3255 Py_END_ALLOW_THREADS
3256 Py_DECREF(command_obj);
Victor Stinnercfa72782010-04-16 11:45:13 +00003257#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003258 return PyLong_FromLong(sts);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003259}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003260#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003261
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003262
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003263PyDoc_STRVAR(posix_umask__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003264"umask(new_mask) -> old_mask\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003265Set the current numeric umask and return the previous umask.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003266
Barry Warsaw53699e91996-12-10 23:23:01 +00003267static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003268posix_umask(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003269{
Victor Stinner8c62be82010-05-06 00:08:46 +00003270 int i;
3271 if (!PyArg_ParseTuple(args, "i:umask", &i))
3272 return NULL;
3273 i = (int)umask(i);
3274 if (i < 0)
3275 return posix_error();
3276 return PyLong_FromLong((long)i);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003277}
3278
Brian Curtind40e6f72010-07-08 21:39:08 +00003279#ifdef MS_WINDOWS
3280
3281/* override the default DeleteFileW behavior so that directory
3282symlinks can be removed with this function, the same as with
3283Unix symlinks */
3284BOOL WINAPI Py_DeleteFileW(LPCWSTR lpFileName)
3285{
3286 WIN32_FILE_ATTRIBUTE_DATA info;
3287 WIN32_FIND_DATAW find_data;
3288 HANDLE find_data_handle;
3289 int is_directory = 0;
3290 int is_link = 0;
3291
3292 if (GetFileAttributesExW(lpFileName, GetFileExInfoStandard, &info)) {
3293 is_directory = info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00003294
Brian Curtind40e6f72010-07-08 21:39:08 +00003295 /* Get WIN32_FIND_DATA structure for the path to determine if
3296 it is a symlink */
3297 if(is_directory &&
3298 info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
3299 find_data_handle = FindFirstFileW(lpFileName, &find_data);
3300
3301 if(find_data_handle != INVALID_HANDLE_VALUE) {
3302 is_link = find_data.dwReserved0 == IO_REPARSE_TAG_SYMLINK;
3303 FindClose(find_data_handle);
3304 }
3305 }
3306 }
3307
3308 if (is_directory && is_link)
3309 return RemoveDirectoryW(lpFileName);
3310
3311 return DeleteFileW(lpFileName);
3312}
3313#endif /* MS_WINDOWS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003314
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003315PyDoc_STRVAR(posix_unlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003316"unlink(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003317Remove a file (same as remove(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003318
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003319PyDoc_STRVAR(posix_remove__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003320"remove(path)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003321Remove a file (same as unlink(path)).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003322
Barry Warsaw53699e91996-12-10 23:23:01 +00003323static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003324posix_unlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003325{
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003326#ifdef MS_WINDOWS
Brian Curtin74e45612010-07-09 15:58:59 +00003327 return win32_1str(args, "remove", "y:remove", DeleteFileA,
3328 "U:remove", Py_DeleteFileW);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003329#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003330 return posix_1str(args, "O&:remove", unlink);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00003331#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003332}
3333
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003334
Guido van Rossumb6775db1994-08-01 11:34:53 +00003335#ifdef HAVE_UNAME
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003336PyDoc_STRVAR(posix_uname__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003337"uname() -> (sysname, nodename, release, version, machine)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003338Return a tuple identifying the current operating system.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003339
Barry Warsaw53699e91996-12-10 23:23:01 +00003340static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00003341posix_uname(PyObject *self, PyObject *noargs)
Guido van Rossumc39de5f1992-02-05 11:15:54 +00003342{
Victor Stinner8c62be82010-05-06 00:08:46 +00003343 struct utsname u;
3344 int res;
Neal Norwitze241ce82003-02-17 18:17:05 +00003345
Victor Stinner8c62be82010-05-06 00:08:46 +00003346 Py_BEGIN_ALLOW_THREADS
3347 res = uname(&u);
3348 Py_END_ALLOW_THREADS
3349 if (res < 0)
3350 return posix_error();
3351 return Py_BuildValue("(sssss)",
3352 u.sysname,
3353 u.nodename,
3354 u.release,
3355 u.version,
3356 u.machine);
Guido van Rossumc39de5f1992-02-05 11:15:54 +00003357}
Guido van Rossumb6775db1994-08-01 11:34:53 +00003358#endif /* HAVE_UNAME */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003359
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003360static int
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003361extract_time(PyObject *t, time_t* sec, long* usec)
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003362{
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003363 time_t intval;
Victor Stinner8c62be82010-05-06 00:08:46 +00003364 if (PyFloat_Check(t)) {
3365 double tval = PyFloat_AsDouble(t);
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003366 PyObject *intobj = PyNumber_Long(t);
Victor Stinner8c62be82010-05-06 00:08:46 +00003367 if (!intobj)
3368 return -1;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003369#if SIZEOF_TIME_T > SIZEOF_LONG
3370 intval = PyLong_AsUnsignedLongLongMask(intobj);
3371#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003372 intval = PyLong_AsLong(intobj);
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003373#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003374 Py_DECREF(intobj);
3375 if (intval == -1 && PyErr_Occurred())
3376 return -1;
3377 *sec = intval;
3378 *usec = (long)((tval - intval) * 1e6); /* can't exceed 1000000 */
3379 if (*usec < 0)
3380 /* If rounding gave us a negative number,
3381 truncate. */
3382 *usec = 0;
Martin v. Löwis076b2092002-09-10 15:04:41 +00003383 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00003384 }
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003385#if SIZEOF_TIME_T > SIZEOF_LONG
3386 intval = PyLong_AsUnsignedLongLongMask(t);
3387#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003388 intval = PyLong_AsLong(t);
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003389#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003390 if (intval == -1 && PyErr_Occurred())
3391 return -1;
3392 *sec = intval;
3393 *usec = 0;
3394 return 0;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003395}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003396
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003397PyDoc_STRVAR(posix_utime__doc__,
Thomas Wouters477c8d52006-05-27 19:21:47 +00003398"utime(path, (atime, mtime))\n\
Fred Drakef7ce04d2002-06-20 18:31:21 +00003399utime(path, None)\n\n\
Barry Warsaw3cef8562000-05-01 16:17:24 +00003400Set the access and modified time of the file to the given values. If the\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003401second form is used, set the access and modified times to the current time.");
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_utime(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003405{
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003406#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00003407 PyObject *arg;
3408 PyUnicodeObject *obwpath;
3409 wchar_t *wpath = NULL;
3410 PyObject *oapath;
3411 char *apath;
3412 HANDLE hFile;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003413 time_t atimesec, mtimesec;
3414 long ausec, musec;
Victor Stinner8c62be82010-05-06 00:08:46 +00003415 FILETIME atime, mtime;
3416 PyObject *result = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003417
Victor Stinner8c62be82010-05-06 00:08:46 +00003418 if (PyArg_ParseTuple(args, "UO|:utime", &obwpath, &arg)) {
3419 wpath = PyUnicode_AS_UNICODE(obwpath);
3420 Py_BEGIN_ALLOW_THREADS
3421 hFile = CreateFileW(wpath, FILE_WRITE_ATTRIBUTES, 0,
3422 NULL, OPEN_EXISTING,
3423 FILE_FLAG_BACKUP_SEMANTICS, NULL);
3424 Py_END_ALLOW_THREADS
3425 if (hFile == INVALID_HANDLE_VALUE)
3426 return win32_error_unicode("utime", wpath);
3427 } else
3428 /* Drop the argument parsing error as narrow strings
3429 are also valid. */
3430 PyErr_Clear();
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00003431
Victor Stinner8c62be82010-05-06 00:08:46 +00003432 if (!wpath) {
3433 if (!PyArg_ParseTuple(args, "O&O:utime",
3434 PyUnicode_FSConverter, &oapath, &arg))
3435 return NULL;
3436 apath = PyBytes_AsString(oapath);
3437 Py_BEGIN_ALLOW_THREADS
3438 hFile = CreateFileA(apath, FILE_WRITE_ATTRIBUTES, 0,
3439 NULL, OPEN_EXISTING,
3440 FILE_FLAG_BACKUP_SEMANTICS, NULL);
3441 Py_END_ALLOW_THREADS
3442 if (hFile == INVALID_HANDLE_VALUE) {
3443 win32_error("utime", apath);
3444 Py_DECREF(oapath);
3445 return NULL;
3446 }
3447 Py_DECREF(oapath);
3448 }
3449
3450 if (arg == Py_None) {
3451 SYSTEMTIME now;
3452 GetSystemTime(&now);
3453 if (!SystemTimeToFileTime(&now, &mtime) ||
3454 !SystemTimeToFileTime(&now, &atime)) {
3455 win32_error("utime", NULL);
3456 goto done;
Stefan Krah0e803b32010-11-26 16:16:47 +00003457 }
Victor Stinner8c62be82010-05-06 00:08:46 +00003458 }
3459 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3460 PyErr_SetString(PyExc_TypeError,
3461 "utime() arg 2 must be a tuple (atime, mtime)");
3462 goto done;
3463 }
3464 else {
3465 if (extract_time(PyTuple_GET_ITEM(arg, 0),
3466 &atimesec, &ausec) == -1)
3467 goto done;
3468 time_t_to_FILE_TIME(atimesec, 1000*ausec, &atime);
3469 if (extract_time(PyTuple_GET_ITEM(arg, 1),
3470 &mtimesec, &musec) == -1)
3471 goto done;
3472 time_t_to_FILE_TIME(mtimesec, 1000*musec, &mtime);
3473 }
3474 if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
3475 /* Avoid putting the file name into the error here,
3476 as that may confuse the user into believing that
3477 something is wrong with the file, when it also
3478 could be the time stamp that gives a problem. */
3479 win32_error("utime", NULL);
Antoine Pitroue85da7a2011-01-06 18:25:55 +00003480 goto done;
Victor Stinner8c62be82010-05-06 00:08:46 +00003481 }
3482 Py_INCREF(Py_None);
3483 result = Py_None;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003484done:
Victor Stinner8c62be82010-05-06 00:08:46 +00003485 CloseHandle(hFile);
3486 return result;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003487#else /* MS_WINDOWS */
Thomas Wouters477c8d52006-05-27 19:21:47 +00003488
Victor Stinner8c62be82010-05-06 00:08:46 +00003489 PyObject *opath;
3490 char *path;
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +00003491 time_t atime, mtime;
3492 long ausec, musec;
Victor Stinner8c62be82010-05-06 00:08:46 +00003493 int res;
3494 PyObject* arg;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003495
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003496#if defined(HAVE_UTIMES)
Victor Stinner8c62be82010-05-06 00:08:46 +00003497 struct timeval buf[2];
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003498#define ATIME buf[0].tv_sec
3499#define MTIME buf[1].tv_sec
3500#elif defined(HAVE_UTIME_H)
Guido van Rossum6d8841c1997-08-14 19:57:39 +00003501/* XXX should define struct utimbuf instead, above */
Victor Stinner8c62be82010-05-06 00:08:46 +00003502 struct utimbuf buf;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003503#define ATIME buf.actime
3504#define MTIME buf.modtime
3505#define UTIME_ARG &buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003506#else /* HAVE_UTIMES */
Victor Stinner8c62be82010-05-06 00:08:46 +00003507 time_t buf[2];
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003508#define ATIME buf[0]
3509#define MTIME buf[1]
3510#define UTIME_ARG buf
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003511#endif /* HAVE_UTIMES */
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003512
Mark Hammond817c9292003-12-03 01:22:38 +00003513
Victor Stinner8c62be82010-05-06 00:08:46 +00003514 if (!PyArg_ParseTuple(args, "O&O:utime",
3515 PyUnicode_FSConverter, &opath, &arg))
3516 return NULL;
3517 path = PyBytes_AsString(opath);
3518 if (arg == Py_None) {
3519 /* optional time values not given */
3520 Py_BEGIN_ALLOW_THREADS
3521 res = utime(path, NULL);
3522 Py_END_ALLOW_THREADS
3523 }
3524 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3525 PyErr_SetString(PyExc_TypeError,
3526 "utime() arg 2 must be a tuple (atime, mtime)");
3527 Py_DECREF(opath);
3528 return NULL;
3529 }
3530 else {
3531 if (extract_time(PyTuple_GET_ITEM(arg, 0),
3532 &atime, &ausec) == -1) {
3533 Py_DECREF(opath);
3534 return NULL;
3535 }
3536 if (extract_time(PyTuple_GET_ITEM(arg, 1),
3537 &mtime, &musec) == -1) {
3538 Py_DECREF(opath);
3539 return NULL;
3540 }
3541 ATIME = atime;
3542 MTIME = mtime;
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003543#ifdef HAVE_UTIMES
Victor Stinner8c62be82010-05-06 00:08:46 +00003544 buf[0].tv_usec = ausec;
3545 buf[1].tv_usec = musec;
3546 Py_BEGIN_ALLOW_THREADS
3547 res = utimes(path, buf);
3548 Py_END_ALLOW_THREADS
Martin v. Löwis6aa9fdb2002-09-10 09:16:13 +00003549#else
Victor Stinner8c62be82010-05-06 00:08:46 +00003550 Py_BEGIN_ALLOW_THREADS
3551 res = utime(path, UTIME_ARG);
3552 Py_END_ALLOW_THREADS
Mark Hammond817c9292003-12-03 01:22:38 +00003553#endif /* HAVE_UTIMES */
Victor Stinner8c62be82010-05-06 00:08:46 +00003554 }
3555 if (res < 0) {
3556 return posix_error_with_allocated_filename(opath);
3557 }
3558 Py_DECREF(opath);
3559 Py_INCREF(Py_None);
3560 return Py_None;
Guido van Rossum1ff6cb41991-04-08 20:59:13 +00003561#undef UTIME_ARG
3562#undef ATIME
3563#undef MTIME
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00003564#endif /* MS_WINDOWS */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003565}
3566
Ross Lagerwall7807c352011-03-17 20:20:30 +02003567#ifdef HAVE_FUTIMES
3568PyDoc_STRVAR(posix_futimes__doc__,
3569"futimes(fd, (atime, mtime))\n\
3570futimes(fd, None)\n\n\
3571Set the access and modified time of the file specified by the file\n\
3572descriptor fd to the given values. If the second form is used, set the\n\
3573access and modified times to the current time.");
3574
3575static PyObject *
3576posix_futimes(PyObject *self, PyObject *args)
3577{
3578 int res, fd;
3579 PyObject* arg;
3580 struct timeval buf[2];
3581 long ausec, musec;
3582
3583 if (!PyArg_ParseTuple(args, "iO:futimes", &fd, &arg))
3584 return NULL;
3585
3586 if (arg == Py_None) {
3587 /* optional time values not given */
3588 Py_BEGIN_ALLOW_THREADS
3589 res = futimes(fd, NULL);
3590 Py_END_ALLOW_THREADS
3591 }
3592 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3593 PyErr_SetString(PyExc_TypeError,
3594 "futimes() arg 2 must be a tuple (atime, mtime)");
3595 return NULL;
3596 }
3597 else {
3598 if (extract_time(PyTuple_GET_ITEM(arg, 0),
3599 &(buf[0].tv_sec), &ausec) == -1) {
3600 return NULL;
3601 }
3602 if (extract_time(PyTuple_GET_ITEM(arg, 1),
3603 &(buf[1].tv_sec), &musec) == -1) {
3604 return NULL;
3605 }
3606 buf[0].tv_usec = ausec;
3607 buf[1].tv_usec = musec;
3608 Py_BEGIN_ALLOW_THREADS
3609 res = futimes(fd, buf);
3610 Py_END_ALLOW_THREADS
3611 }
3612 if (res < 0)
3613 return posix_error();
3614 Py_RETURN_NONE;
3615}
3616#endif
3617
3618#ifdef HAVE_LUTIMES
3619PyDoc_STRVAR(posix_lutimes__doc__,
3620"lutimes(path, (atime, mtime))\n\
3621lutimes(path, None)\n\n\
3622Like utime(), but if path is a symbolic link, it is not dereferenced.");
3623
3624static PyObject *
3625posix_lutimes(PyObject *self, PyObject *args)
3626{
3627 PyObject *opath, *arg;
3628 const char *path;
3629 int res;
3630 struct timeval buf[2];
3631 long ausec, musec;
3632
3633 if (!PyArg_ParseTuple(args, "O&O:lutimes",
3634 PyUnicode_FSConverter, &opath, &arg))
3635 return NULL;
3636 path = PyBytes_AsString(opath);
3637 if (arg == Py_None) {
3638 /* optional time values not given */
3639 Py_BEGIN_ALLOW_THREADS
3640 res = lutimes(path, NULL);
3641 Py_END_ALLOW_THREADS
3642 }
3643 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
3644 PyErr_SetString(PyExc_TypeError,
3645 "lutimes() arg 2 must be a tuple (atime, mtime)");
3646 Py_DECREF(opath);
3647 return NULL;
3648 }
3649 else {
3650 if (extract_time(PyTuple_GET_ITEM(arg, 0),
3651 &(buf[0].tv_sec), &ausec) == -1) {
3652 Py_DECREF(opath);
3653 return NULL;
3654 }
3655 if (extract_time(PyTuple_GET_ITEM(arg, 1),
3656 &(buf[1].tv_sec), &musec) == -1) {
3657 Py_DECREF(opath);
3658 return NULL;
3659 }
3660 buf[0].tv_usec = ausec;
3661 buf[1].tv_usec = musec;
3662 Py_BEGIN_ALLOW_THREADS
3663 res = lutimes(path, buf);
3664 Py_END_ALLOW_THREADS
3665 }
3666 Py_DECREF(opath);
3667 if (res < 0)
3668 return posix_error();
3669 Py_RETURN_NONE;
3670}
3671#endif
3672
3673#ifdef HAVE_FUTIMENS
3674PyDoc_STRVAR(posix_futimens__doc__,
3675"futimens(fd, (atime_sec, atime_nsec), (mtime_sec, mtime_nsec))\n\
3676futimens(fd, None, None)\n\n\
3677Updates the timestamps of a file specified by the file descriptor fd, with\n\
3678nanosecond precision.\n\
3679The second form sets atime and mtime to the current time.\n\
3680If *_nsec is specified as UTIME_NOW, the timestamp is updated to the\n\
3681current time.\n\
3682If *_nsec is specified as UTIME_OMIT, the timestamp is not updated.");
3683
3684static PyObject *
3685posix_futimens(PyObject *self, PyObject *args)
3686{
3687 int res, fd;
3688 PyObject *atime, *mtime;
3689 struct timespec buf[2];
3690
3691 if (!PyArg_ParseTuple(args, "iOO:futimens",
3692 &fd, &atime, &mtime))
3693 return NULL;
3694 if (atime == Py_None && mtime == Py_None) {
3695 /* optional time values not given */
3696 Py_BEGIN_ALLOW_THREADS
3697 res = futimens(fd, NULL);
3698 Py_END_ALLOW_THREADS
3699 }
3700 else if (!PyTuple_Check(atime) || PyTuple_Size(atime) != 2) {
3701 PyErr_SetString(PyExc_TypeError,
3702 "futimens() arg 2 must be a tuple (atime_sec, atime_nsec)");
3703 return NULL;
3704 }
3705 else if (!PyTuple_Check(mtime) || PyTuple_Size(mtime) != 2) {
3706 PyErr_SetString(PyExc_TypeError,
3707 "futimens() arg 3 must be a tuple (mtime_sec, mtime_nsec)");
3708 return NULL;
3709 }
3710 else {
3711 if (!PyArg_ParseTuple(atime, "ll:futimens",
3712 &(buf[0].tv_sec), &(buf[0].tv_nsec))) {
3713 return NULL;
3714 }
3715 if (!PyArg_ParseTuple(mtime, "ll:futimens",
3716 &(buf[1].tv_sec), &(buf[1].tv_nsec))) {
3717 return NULL;
3718 }
3719 Py_BEGIN_ALLOW_THREADS
3720 res = futimens(fd, buf);
3721 Py_END_ALLOW_THREADS
3722 }
3723 if (res < 0)
3724 return posix_error();
3725 Py_RETURN_NONE;
3726}
3727#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00003728
Guido van Rossum3b066191991-06-04 19:40:25 +00003729/* Process operations */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003730
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003731PyDoc_STRVAR(posix__exit__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003732"_exit(status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003733Exit to the system with specified status, without normal exit processing.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003734
Barry Warsaw53699e91996-12-10 23:23:01 +00003735static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003736posix__exit(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00003737{
Victor Stinner8c62be82010-05-06 00:08:46 +00003738 int sts;
3739 if (!PyArg_ParseTuple(args, "i:_exit", &sts))
3740 return NULL;
3741 _exit(sts);
3742 return NULL; /* Make gcc -Wall happy */
Guido van Rossum85e3b011991-06-03 12:42:10 +00003743}
3744
Martin v. Löwis114619e2002-10-07 06:44:21 +00003745#if defined(HAVE_EXECV) || defined(HAVE_SPAWNV)
3746static void
Martin v. Löwis725507b2006-03-07 12:08:51 +00003747free_string_array(char **array, Py_ssize_t count)
Martin v. Löwis114619e2002-10-07 06:44:21 +00003748{
Victor Stinner8c62be82010-05-06 00:08:46 +00003749 Py_ssize_t i;
3750 for (i = 0; i < count; i++)
3751 PyMem_Free(array[i]);
3752 PyMem_DEL(array);
Martin v. Löwis114619e2002-10-07 06:44:21 +00003753}
Martin v. Löwis011e8422009-05-05 04:43:17 +00003754
Antoine Pitrou69f71142009-05-24 21:25:49 +00003755static
Martin v. Löwis011e8422009-05-05 04:43:17 +00003756int fsconvert_strdup(PyObject *o, char**out)
3757{
Victor Stinner8c62be82010-05-06 00:08:46 +00003758 PyObject *bytes;
3759 Py_ssize_t size;
3760 if (!PyUnicode_FSConverter(o, &bytes))
3761 return 0;
3762 size = PyBytes_GET_SIZE(bytes);
3763 *out = PyMem_Malloc(size+1);
3764 if (!*out)
3765 return 0;
3766 memcpy(*out, PyBytes_AsString(bytes), size+1);
3767 Py_DECREF(bytes);
3768 return 1;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003769}
Martin v. Löwis114619e2002-10-07 06:44:21 +00003770#endif
3771
Ross Lagerwall7807c352011-03-17 20:20:30 +02003772#if defined(HAVE_EXECV) || defined (HAVE_FEXECVE)
Victor Stinner13bb71c2010-04-23 21:41:56 +00003773static char**
3774parse_envlist(PyObject* env, Py_ssize_t *envc_ptr)
3775{
Victor Stinner8c62be82010-05-06 00:08:46 +00003776 char **envlist;
3777 Py_ssize_t i, pos, envc;
3778 PyObject *keys=NULL, *vals=NULL;
3779 PyObject *key, *val, *key2, *val2;
3780 char *p, *k, *v;
3781 size_t len;
Victor Stinner13bb71c2010-04-23 21:41:56 +00003782
Victor Stinner8c62be82010-05-06 00:08:46 +00003783 i = PyMapping_Size(env);
3784 if (i < 0)
3785 return NULL;
3786 envlist = PyMem_NEW(char *, i + 1);
3787 if (envlist == NULL) {
3788 PyErr_NoMemory();
3789 return NULL;
3790 }
3791 envc = 0;
3792 keys = PyMapping_Keys(env);
3793 vals = PyMapping_Values(env);
3794 if (!keys || !vals)
3795 goto error;
3796 if (!PyList_Check(keys) || !PyList_Check(vals)) {
3797 PyErr_Format(PyExc_TypeError,
3798 "env.keys() or env.values() is not a list");
3799 goto error;
3800 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00003801
Victor Stinner8c62be82010-05-06 00:08:46 +00003802 for (pos = 0; pos < i; pos++) {
3803 key = PyList_GetItem(keys, pos);
3804 val = PyList_GetItem(vals, pos);
3805 if (!key || !val)
3806 goto error;
Victor Stinner13bb71c2010-04-23 21:41:56 +00003807
Victor Stinner8c62be82010-05-06 00:08:46 +00003808 if (PyUnicode_FSConverter(key, &key2) == 0)
3809 goto error;
3810 if (PyUnicode_FSConverter(val, &val2) == 0) {
3811 Py_DECREF(key2);
3812 goto error;
3813 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00003814
3815#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +00003816 /* Omit Pseudo-Env Vars that Would Confuse Programs if Passed On */
3817 if (stricmp(k, "BEGINLIBPATH") != 0 && stricmp(k, "ENDLIBPATH") != 0) {
Victor Stinner13bb71c2010-04-23 21:41:56 +00003818#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003819 k = PyBytes_AsString(key2);
3820 v = PyBytes_AsString(val2);
3821 len = PyBytes_GET_SIZE(key2) + PyBytes_GET_SIZE(val2) + 2;
Victor Stinner13bb71c2010-04-23 21:41:56 +00003822
Victor Stinner8c62be82010-05-06 00:08:46 +00003823 p = PyMem_NEW(char, len);
3824 if (p == NULL) {
3825 PyErr_NoMemory();
3826 Py_DECREF(key2);
3827 Py_DECREF(val2);
3828 goto error;
3829 }
3830 PyOS_snprintf(p, len, "%s=%s", k, v);
3831 envlist[envc++] = p;
3832 Py_DECREF(key2);
3833 Py_DECREF(val2);
Victor Stinner13bb71c2010-04-23 21:41:56 +00003834#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +00003835 }
Victor Stinner13bb71c2010-04-23 21:41:56 +00003836#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00003837 }
3838 Py_DECREF(vals);
3839 Py_DECREF(keys);
Victor Stinner13bb71c2010-04-23 21:41:56 +00003840
Victor Stinner8c62be82010-05-06 00:08:46 +00003841 envlist[envc] = 0;
3842 *envc_ptr = envc;
3843 return envlist;
Victor Stinner13bb71c2010-04-23 21:41:56 +00003844
3845error:
Victor Stinner8c62be82010-05-06 00:08:46 +00003846 Py_XDECREF(keys);
3847 Py_XDECREF(vals);
3848 while (--envc >= 0)
3849 PyMem_DEL(envlist[envc]);
3850 PyMem_DEL(envlist);
3851 return NULL;
Victor Stinner13bb71c2010-04-23 21:41:56 +00003852}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003853
Ross Lagerwall7807c352011-03-17 20:20:30 +02003854static char**
3855parse_arglist(PyObject* argv, Py_ssize_t *argc)
3856{
3857 int i;
3858 char **argvlist = PyMem_NEW(char *, *argc+1);
3859 if (argvlist == NULL) {
3860 PyErr_NoMemory();
3861 return NULL;
3862 }
3863 for (i = 0; i < *argc; i++) {
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02003864 PyObject* item = PySequence_ITEM(argv, i);
3865 if (item == NULL)
3866 goto fail;
3867 if (!fsconvert_strdup(item, &argvlist[i])) {
3868 Py_DECREF(item);
Ross Lagerwall7807c352011-03-17 20:20:30 +02003869 goto fail;
3870 }
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02003871 Py_DECREF(item);
Ross Lagerwall7807c352011-03-17 20:20:30 +02003872 }
3873 argvlist[*argc] = NULL;
3874 return argvlist;
3875fail:
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02003876 *argc = i;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003877 free_string_array(argvlist, *argc);
3878 return NULL;
3879}
3880#endif
3881
3882#ifdef HAVE_EXECV
3883PyDoc_STRVAR(posix_execv__doc__,
3884"execv(path, args)\n\n\
3885Execute an executable path with arguments, replacing current process.\n\
3886\n\
3887 path: path of executable file\n\
3888 args: tuple or list of strings");
3889
3890static PyObject *
3891posix_execv(PyObject *self, PyObject *args)
3892{
3893 PyObject *opath;
3894 char *path;
3895 PyObject *argv;
3896 char **argvlist;
3897 Py_ssize_t argc;
3898
3899 /* execv has two arguments: (path, argv), where
3900 argv is a list or tuple of strings. */
3901
3902 if (!PyArg_ParseTuple(args, "O&O:execv",
3903 PyUnicode_FSConverter,
3904 &opath, &argv))
3905 return NULL;
3906 path = PyBytes_AsString(opath);
3907 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
3908 PyErr_SetString(PyExc_TypeError,
3909 "execv() arg 2 must be a tuple or list");
3910 Py_DECREF(opath);
3911 return NULL;
3912 }
3913 argc = PySequence_Size(argv);
3914 if (argc < 1) {
3915 PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
3916 Py_DECREF(opath);
3917 return NULL;
3918 }
3919
3920 argvlist = parse_arglist(argv, &argc);
3921 if (argvlist == NULL) {
3922 Py_DECREF(opath);
3923 return NULL;
3924 }
3925
3926 execv(path, argvlist);
3927
3928 /* If we get here it's definitely an error */
3929
3930 free_string_array(argvlist, argc);
3931 Py_DECREF(opath);
3932 return posix_error();
3933}
3934
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003935PyDoc_STRVAR(posix_execve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00003936"execve(path, args, env)\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003937Execute a path with arguments and environment, replacing current process.\n\
3938\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00003939 path: path of executable file\n\
3940 args: tuple or list of arguments\n\
3941 env: dictionary of strings mapping to strings");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00003942
Barry Warsaw53699e91996-12-10 23:23:01 +00003943static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00003944posix_execve(PyObject *self, PyObject *args)
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003945{
Victor Stinner8c62be82010-05-06 00:08:46 +00003946 PyObject *opath;
3947 char *path;
3948 PyObject *argv, *env;
3949 char **argvlist;
3950 char **envlist;
Ross Lagerwall7807c352011-03-17 20:20:30 +02003951 Py_ssize_t argc, envc;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003952
Victor Stinner8c62be82010-05-06 00:08:46 +00003953 /* execve has three arguments: (path, argv, env), where
3954 argv is a list or tuple of strings and env is a dictionary
3955 like posix.environ. */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003956
Victor Stinner8c62be82010-05-06 00:08:46 +00003957 if (!PyArg_ParseTuple(args, "O&OO:execve",
3958 PyUnicode_FSConverter,
3959 &opath, &argv, &env))
3960 return NULL;
3961 path = PyBytes_AsString(opath);
Ross Lagerwall7807c352011-03-17 20:20:30 +02003962 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003963 PyErr_SetString(PyExc_TypeError,
3964 "execve() arg 2 must be a tuple or list");
3965 goto fail_0;
3966 }
Ross Lagerwall7807c352011-03-17 20:20:30 +02003967 argc = PySequence_Size(argv);
Victor Stinner8c62be82010-05-06 00:08:46 +00003968 if (!PyMapping_Check(env)) {
3969 PyErr_SetString(PyExc_TypeError,
3970 "execve() arg 3 must be a mapping object");
3971 goto fail_0;
3972 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003973
Ross Lagerwall7807c352011-03-17 20:20:30 +02003974 argvlist = parse_arglist(argv, &argc);
Victor Stinner8c62be82010-05-06 00:08:46 +00003975 if (argvlist == NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00003976 goto fail_0;
3977 }
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003978
Victor Stinner8c62be82010-05-06 00:08:46 +00003979 envlist = parse_envlist(env, &envc);
3980 if (envlist == NULL)
3981 goto fail_1;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003982
Victor Stinner8c62be82010-05-06 00:08:46 +00003983 execve(path, argvlist, envlist);
Tim Peters5aa91602002-01-30 05:46:57 +00003984
Victor Stinner8c62be82010-05-06 00:08:46 +00003985 /* If we get here it's definitely an error */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003986
Victor Stinner8c62be82010-05-06 00:08:46 +00003987 (void) posix_error();
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003988
Victor Stinner8c62be82010-05-06 00:08:46 +00003989 while (--envc >= 0)
3990 PyMem_DEL(envlist[envc]);
3991 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003992 fail_1:
Ross Lagerwall7807c352011-03-17 20:20:30 +02003993 free_string_array(argvlist, argc);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00003994 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00003995 Py_DECREF(opath);
3996 return NULL;
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003997}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00003998#endif /* HAVE_EXECV */
Guido van Rossumc6dcc9f1993-11-05 10:15:19 +00003999
Ross Lagerwall7807c352011-03-17 20:20:30 +02004000#ifdef HAVE_FEXECVE
4001PyDoc_STRVAR(posix_fexecve__doc__,
4002"fexecve(fd, args, env)\n\n\
4003Execute the program specified by a file descriptor with arguments and\n\
4004environment, replacing the current process.\n\
4005\n\
4006 fd: file descriptor of executable\n\
4007 args: tuple or list of arguments\n\
4008 env: dictionary of strings mapping to strings");
4009
4010static PyObject *
4011posix_fexecve(PyObject *self, PyObject *args)
4012{
4013 int fd;
4014 PyObject *argv, *env;
4015 char **argvlist;
4016 char **envlist;
4017 Py_ssize_t argc, envc;
4018
4019 if (!PyArg_ParseTuple(args, "iOO:fexecve",
4020 &fd, &argv, &env))
4021 return NULL;
4022 if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
4023 PyErr_SetString(PyExc_TypeError,
4024 "fexecve() arg 2 must be a tuple or list");
4025 return NULL;
4026 }
4027 argc = PySequence_Size(argv);
4028 if (!PyMapping_Check(env)) {
4029 PyErr_SetString(PyExc_TypeError,
4030 "fexecve() arg 3 must be a mapping object");
4031 return NULL;
4032 }
4033
4034 argvlist = parse_arglist(argv, &argc);
4035 if (argvlist == NULL)
4036 return NULL;
4037
4038 envlist = parse_envlist(env, &envc);
4039 if (envlist == NULL)
4040 goto fail;
4041
4042 fexecve(fd, argvlist, envlist);
4043
4044 /* If we get here it's definitely an error */
4045
4046 (void) posix_error();
4047
4048 while (--envc >= 0)
4049 PyMem_DEL(envlist[envc]);
4050 PyMem_DEL(envlist);
4051 fail:
4052 free_string_array(argvlist, argc);
4053 return NULL;
4054}
4055#endif /* HAVE_FEXECVE */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004056
Guido van Rossuma1065681999-01-25 23:20:23 +00004057#ifdef HAVE_SPAWNV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004058PyDoc_STRVAR(posix_spawnv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004059"spawnv(mode, path, args)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00004060Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00004061\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004062 mode: mode of process creation\n\
4063 path: path of executable file\n\
4064 args: tuple or list of strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00004065
4066static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004067posix_spawnv(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00004068{
Victor Stinner8c62be82010-05-06 00:08:46 +00004069 PyObject *opath;
4070 char *path;
4071 PyObject *argv;
4072 char **argvlist;
4073 int mode, i;
4074 Py_ssize_t argc;
4075 Py_intptr_t spawnval;
4076 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Guido van Rossuma1065681999-01-25 23:20:23 +00004077
Victor Stinner8c62be82010-05-06 00:08:46 +00004078 /* spawnv has three arguments: (mode, path, argv), where
4079 argv is a list or tuple of strings. */
Guido van Rossuma1065681999-01-25 23:20:23 +00004080
Victor Stinner8c62be82010-05-06 00:08:46 +00004081 if (!PyArg_ParseTuple(args, "iO&O:spawnv", &mode,
4082 PyUnicode_FSConverter,
4083 &opath, &argv))
4084 return NULL;
4085 path = PyBytes_AsString(opath);
4086 if (PyList_Check(argv)) {
4087 argc = PyList_Size(argv);
4088 getitem = PyList_GetItem;
4089 }
4090 else if (PyTuple_Check(argv)) {
4091 argc = PyTuple_Size(argv);
4092 getitem = PyTuple_GetItem;
4093 }
4094 else {
4095 PyErr_SetString(PyExc_TypeError,
4096 "spawnv() arg 2 must be a tuple or list");
4097 Py_DECREF(opath);
4098 return NULL;
4099 }
Guido van Rossuma1065681999-01-25 23:20:23 +00004100
Victor Stinner8c62be82010-05-06 00:08:46 +00004101 argvlist = PyMem_NEW(char *, argc+1);
4102 if (argvlist == NULL) {
4103 Py_DECREF(opath);
4104 return PyErr_NoMemory();
4105 }
4106 for (i = 0; i < argc; i++) {
4107 if (!fsconvert_strdup((*getitem)(argv, i),
4108 &argvlist[i])) {
4109 free_string_array(argvlist, i);
4110 PyErr_SetString(
4111 PyExc_TypeError,
4112 "spawnv() arg 2 must contain only strings");
4113 Py_DECREF(opath);
4114 return NULL;
4115 }
4116 }
4117 argvlist[argc] = NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00004118
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004119#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004120 Py_BEGIN_ALLOW_THREADS
4121 spawnval = spawnv(mode, path, argvlist);
4122 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004123#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004124 if (mode == _OLD_P_OVERLAY)
4125 mode = _P_OVERLAY;
Tim Peters5aa91602002-01-30 05:46:57 +00004126
Victor Stinner8c62be82010-05-06 00:08:46 +00004127 Py_BEGIN_ALLOW_THREADS
4128 spawnval = _spawnv(mode, path, argvlist);
4129 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004130#endif
Tim Peters5aa91602002-01-30 05:46:57 +00004131
Victor Stinner8c62be82010-05-06 00:08:46 +00004132 free_string_array(argvlist, argc);
4133 Py_DECREF(opath);
Guido van Rossuma1065681999-01-25 23:20:23 +00004134
Victor Stinner8c62be82010-05-06 00:08:46 +00004135 if (spawnval == -1)
4136 return posix_error();
4137 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00004138#if SIZEOF_LONG == SIZEOF_VOID_P
Victor Stinner8c62be82010-05-06 00:08:46 +00004139 return Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004140#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004141 return Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004142#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00004143}
4144
4145
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004146PyDoc_STRVAR(posix_spawnve__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004147"spawnve(mode, path, args, env)\n\n\
Tim Peters25059d32001-12-07 20:35:43 +00004148Execute the program 'path' in a new process.\n\
Guido van Rossuma1065681999-01-25 23:20:23 +00004149\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004150 mode: mode of process creation\n\
4151 path: path of executable file\n\
4152 args: tuple or list of arguments\n\
4153 env: dictionary of strings mapping to strings");
Guido van Rossuma1065681999-01-25 23:20:23 +00004154
4155static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004156posix_spawnve(PyObject *self, PyObject *args)
Guido van Rossuma1065681999-01-25 23:20:23 +00004157{
Victor Stinner8c62be82010-05-06 00:08:46 +00004158 PyObject *opath;
4159 char *path;
4160 PyObject *argv, *env;
4161 char **argvlist;
4162 char **envlist;
4163 PyObject *res = NULL;
Antoine Pitrou22e41552010-08-15 18:07:50 +00004164 int mode;
4165 Py_ssize_t argc, i, envc;
Victor Stinner8c62be82010-05-06 00:08:46 +00004166 Py_intptr_t spawnval;
4167 PyObject *(*getitem)(PyObject *, Py_ssize_t);
4168 Py_ssize_t lastarg = 0;
Guido van Rossuma1065681999-01-25 23:20:23 +00004169
Victor Stinner8c62be82010-05-06 00:08:46 +00004170 /* spawnve has four arguments: (mode, path, argv, env), where
4171 argv is a list or tuple of strings and env is a dictionary
4172 like posix.environ. */
Guido van Rossuma1065681999-01-25 23:20:23 +00004173
Victor Stinner8c62be82010-05-06 00:08:46 +00004174 if (!PyArg_ParseTuple(args, "iO&OO:spawnve", &mode,
4175 PyUnicode_FSConverter,
4176 &opath, &argv, &env))
4177 return NULL;
4178 path = PyBytes_AsString(opath);
4179 if (PyList_Check(argv)) {
4180 argc = PyList_Size(argv);
4181 getitem = PyList_GetItem;
4182 }
4183 else if (PyTuple_Check(argv)) {
4184 argc = PyTuple_Size(argv);
4185 getitem = PyTuple_GetItem;
4186 }
4187 else {
4188 PyErr_SetString(PyExc_TypeError,
4189 "spawnve() arg 2 must be a tuple or list");
4190 goto fail_0;
4191 }
4192 if (!PyMapping_Check(env)) {
4193 PyErr_SetString(PyExc_TypeError,
4194 "spawnve() arg 3 must be a mapping object");
4195 goto fail_0;
4196 }
Guido van Rossuma1065681999-01-25 23:20:23 +00004197
Victor Stinner8c62be82010-05-06 00:08:46 +00004198 argvlist = PyMem_NEW(char *, argc+1);
4199 if (argvlist == NULL) {
4200 PyErr_NoMemory();
4201 goto fail_0;
4202 }
4203 for (i = 0; i < argc; i++) {
4204 if (!fsconvert_strdup((*getitem)(argv, i),
4205 &argvlist[i]))
4206 {
4207 lastarg = i;
4208 goto fail_1;
4209 }
4210 }
4211 lastarg = argc;
4212 argvlist[argc] = NULL;
Guido van Rossuma1065681999-01-25 23:20:23 +00004213
Victor Stinner8c62be82010-05-06 00:08:46 +00004214 envlist = parse_envlist(env, &envc);
4215 if (envlist == NULL)
4216 goto fail_1;
Guido van Rossuma1065681999-01-25 23:20:23 +00004217
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004218#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004219 Py_BEGIN_ALLOW_THREADS
4220 spawnval = spawnve(mode, path, argvlist, envlist);
4221 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004222#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004223 if (mode == _OLD_P_OVERLAY)
4224 mode = _P_OVERLAY;
Tim Peters25059d32001-12-07 20:35:43 +00004225
Victor Stinner8c62be82010-05-06 00:08:46 +00004226 Py_BEGIN_ALLOW_THREADS
4227 spawnval = _spawnve(mode, path, argvlist, envlist);
4228 Py_END_ALLOW_THREADS
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004229#endif
Tim Peters25059d32001-12-07 20:35:43 +00004230
Victor Stinner8c62be82010-05-06 00:08:46 +00004231 if (spawnval == -1)
4232 (void) posix_error();
4233 else
Fredrik Lundhe25cfd82000-07-09 13:10:40 +00004234#if SIZEOF_LONG == SIZEOF_VOID_P
Victor Stinner8c62be82010-05-06 00:08:46 +00004235 res = Py_BuildValue("l", (long) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004236#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004237 res = Py_BuildValue("L", (PY_LONG_LONG) spawnval);
Fred Drake699f3522000-06-29 21:12:41 +00004238#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00004239
Victor Stinner8c62be82010-05-06 00:08:46 +00004240 while (--envc >= 0)
4241 PyMem_DEL(envlist[envc]);
4242 PyMem_DEL(envlist);
Guido van Rossum0847c5c2002-12-13 18:36:22 +00004243 fail_1:
Victor Stinner8c62be82010-05-06 00:08:46 +00004244 free_string_array(argvlist, lastarg);
Martin v. Löwis114619e2002-10-07 06:44:21 +00004245 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004246 Py_DECREF(opath);
4247 return res;
Guido van Rossuma1065681999-01-25 23:20:23 +00004248}
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004249
4250/* OS/2 supports spawnvp & spawnvpe natively */
4251#if defined(PYOS_OS2)
4252PyDoc_STRVAR(posix_spawnvp__doc__,
4253"spawnvp(mode, file, args)\n\n\
4254Execute the program 'file' in a new process, using the environment\n\
4255search path to find the file.\n\
4256\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004257 mode: mode of process creation\n\
4258 file: executable file name\n\
4259 args: tuple or list of strings");
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004260
4261static PyObject *
4262posix_spawnvp(PyObject *self, PyObject *args)
4263{
Victor Stinner8c62be82010-05-06 00:08:46 +00004264 PyObject *opath;
4265 char *path;
4266 PyObject *argv;
4267 char **argvlist;
4268 int mode, i, argc;
4269 Py_intptr_t spawnval;
4270 PyObject *(*getitem)(PyObject *, Py_ssize_t);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004271
Victor Stinner8c62be82010-05-06 00:08:46 +00004272 /* spawnvp has three arguments: (mode, path, argv), where
4273 argv is a list or tuple of strings. */
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004274
Victor Stinner8c62be82010-05-06 00:08:46 +00004275 if (!PyArg_ParseTuple(args, "iO&O:spawnvp", &mode,
4276 PyUnicode_FSConverter,
4277 &opath, &argv))
4278 return NULL;
4279 path = PyBytes_AsString(opath);
4280 if (PyList_Check(argv)) {
4281 argc = PyList_Size(argv);
4282 getitem = PyList_GetItem;
4283 }
4284 else if (PyTuple_Check(argv)) {
4285 argc = PyTuple_Size(argv);
4286 getitem = PyTuple_GetItem;
4287 }
4288 else {
4289 PyErr_SetString(PyExc_TypeError,
4290 "spawnvp() arg 2 must be a tuple or list");
4291 Py_DECREF(opath);
4292 return NULL;
4293 }
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004294
Victor Stinner8c62be82010-05-06 00:08:46 +00004295 argvlist = PyMem_NEW(char *, argc+1);
4296 if (argvlist == NULL) {
4297 Py_DECREF(opath);
4298 return PyErr_NoMemory();
4299 }
4300 for (i = 0; i < argc; i++) {
4301 if (!fsconvert_strdup((*getitem)(argv, i),
4302 &argvlist[i])) {
4303 free_string_array(argvlist, i);
4304 PyErr_SetString(
4305 PyExc_TypeError,
4306 "spawnvp() arg 2 must contain only strings");
4307 Py_DECREF(opath);
4308 return NULL;
4309 }
4310 }
4311 argvlist[argc] = NULL;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004312
Victor Stinner8c62be82010-05-06 00:08:46 +00004313 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004314#if defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004315 spawnval = spawnvp(mode, path, argvlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004316#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004317 spawnval = _spawnvp(mode, path, argvlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004318#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004319 Py_END_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004320
Victor Stinner8c62be82010-05-06 00:08:46 +00004321 free_string_array(argvlist, argc);
4322 Py_DECREF(opath);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004323
Victor Stinner8c62be82010-05-06 00:08:46 +00004324 if (spawnval == -1)
4325 return posix_error();
4326 else
4327 return Py_BuildValue("l", (long) spawnval);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004328}
4329
4330
4331PyDoc_STRVAR(posix_spawnvpe__doc__,
4332"spawnvpe(mode, file, args, env)\n\n\
4333Execute the program 'file' in a new process, using the environment\n\
4334search path to find the file.\n\
4335\n\
Victor Stinner8c62be82010-05-06 00:08:46 +00004336 mode: mode of process creation\n\
4337 file: executable file name\n\
4338 args: tuple or list of arguments\n\
4339 env: dictionary of strings mapping to strings");
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004340
4341static PyObject *
4342posix_spawnvpe(PyObject *self, PyObject *args)
4343{
Victor Stinner8c62be82010-05-06 00:08:46 +00004344 PyObject *opath
4345 char *path;
4346 PyObject *argv, *env;
4347 char **argvlist;
4348 char **envlist;
4349 PyObject *res=NULL;
Antoine Pitrou22e41552010-08-15 18:07:50 +00004350 int mode;
4351 Py_ssize_t argc, i, envc;
Victor Stinner8c62be82010-05-06 00:08:46 +00004352 Py_intptr_t spawnval;
4353 PyObject *(*getitem)(PyObject *, Py_ssize_t);
4354 int lastarg = 0;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004355
Victor Stinner8c62be82010-05-06 00:08:46 +00004356 /* spawnvpe has four arguments: (mode, path, argv, env), where
4357 argv is a list or tuple of strings and env is a dictionary
4358 like posix.environ. */
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004359
Victor Stinner8c62be82010-05-06 00:08:46 +00004360 if (!PyArg_ParseTuple(args, "ietOO:spawnvpe", &mode,
4361 PyUnicode_FSConverter,
4362 &opath, &argv, &env))
4363 return NULL;
4364 path = PyBytes_AsString(opath);
4365 if (PyList_Check(argv)) {
4366 argc = PyList_Size(argv);
4367 getitem = PyList_GetItem;
4368 }
4369 else if (PyTuple_Check(argv)) {
4370 argc = PyTuple_Size(argv);
4371 getitem = PyTuple_GetItem;
4372 }
4373 else {
4374 PyErr_SetString(PyExc_TypeError,
4375 "spawnvpe() arg 2 must be a tuple or list");
4376 goto fail_0;
4377 }
4378 if (!PyMapping_Check(env)) {
4379 PyErr_SetString(PyExc_TypeError,
4380 "spawnvpe() arg 3 must be a mapping object");
4381 goto fail_0;
4382 }
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004383
Victor Stinner8c62be82010-05-06 00:08:46 +00004384 argvlist = PyMem_NEW(char *, argc+1);
4385 if (argvlist == NULL) {
4386 PyErr_NoMemory();
4387 goto fail_0;
4388 }
4389 for (i = 0; i < argc; i++) {
4390 if (!fsconvert_strdup((*getitem)(argv, i),
4391 &argvlist[i]))
4392 {
4393 lastarg = i;
4394 goto fail_1;
4395 }
4396 }
4397 lastarg = argc;
4398 argvlist[argc] = NULL;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004399
Victor Stinner8c62be82010-05-06 00:08:46 +00004400 envlist = parse_envlist(env, &envc);
4401 if (envlist == NULL)
4402 goto fail_1;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004403
Victor Stinner8c62be82010-05-06 00:08:46 +00004404 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004405#if defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004406 spawnval = spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004407#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004408 spawnval = _spawnvpe(mode, path, argvlist, envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004409#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004410 Py_END_ALLOW_THREADS
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004411
Victor Stinner8c62be82010-05-06 00:08:46 +00004412 if (spawnval == -1)
4413 (void) posix_error();
4414 else
4415 res = Py_BuildValue("l", (long) spawnval);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004416
Victor Stinner8c62be82010-05-06 00:08:46 +00004417 while (--envc >= 0)
4418 PyMem_DEL(envlist[envc]);
4419 PyMem_DEL(envlist);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004420 fail_1:
Victor Stinner8c62be82010-05-06 00:08:46 +00004421 free_string_array(argvlist, lastarg);
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004422 fail_0:
Victor Stinner8c62be82010-05-06 00:08:46 +00004423 Py_DECREF(opath);
4424 return res;
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00004425}
4426#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00004427#endif /* HAVE_SPAWNV */
4428
4429
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004430#ifdef HAVE_FORK1
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004431PyDoc_STRVAR(posix_fork1__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004432"fork1() -> pid\n\n\
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004433Fork a child process with a single multiplexed (i.e., not bound) thread.\n\
4434\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004435Return 0 to child process and PID of child to parent process.");
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004436
4437static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004438posix_fork1(PyObject *self, PyObject *noargs)
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004439{
Victor Stinner8c62be82010-05-06 00:08:46 +00004440 pid_t pid;
4441 int result = 0;
4442 _PyImport_AcquireLock();
4443 pid = fork1();
4444 if (pid == 0) {
4445 /* child: this clobbers and resets the import lock. */
4446 PyOS_AfterFork();
4447 } else {
4448 /* parent: release the import lock. */
4449 result = _PyImport_ReleaseLock();
4450 }
4451 if (pid == -1)
4452 return posix_error();
4453 if (result < 0) {
4454 /* Don't clobber the OSError if the fork failed. */
4455 PyErr_SetString(PyExc_RuntimeError,
4456 "not holding the import lock");
4457 return NULL;
4458 }
4459 return PyLong_FromPid(pid);
Guido van Rossum2242f2f2001-04-11 20:58:20 +00004460}
4461#endif
4462
4463
Guido van Rossumad0ee831995-03-01 10:34:45 +00004464#ifdef HAVE_FORK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004465PyDoc_STRVAR(posix_fork__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004466"fork() -> pid\n\n\
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004467Fork a child process.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004468Return 0 to child process and PID of child to parent process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004469
Barry Warsaw53699e91996-12-10 23:23:01 +00004470static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004471posix_fork(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004472{
Victor Stinner8c62be82010-05-06 00:08:46 +00004473 pid_t pid;
4474 int result = 0;
4475 _PyImport_AcquireLock();
4476 pid = fork();
4477 if (pid == 0) {
4478 /* child: this clobbers and resets the import lock. */
4479 PyOS_AfterFork();
4480 } else {
4481 /* parent: release the import lock. */
4482 result = _PyImport_ReleaseLock();
4483 }
4484 if (pid == -1)
4485 return posix_error();
4486 if (result < 0) {
4487 /* Don't clobber the OSError if the fork failed. */
4488 PyErr_SetString(PyExc_RuntimeError,
4489 "not holding the import lock");
4490 return NULL;
4491 }
4492 return PyLong_FromPid(pid);
Guido van Rossum85e3b011991-06-03 12:42:10 +00004493}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004494#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004495
Neal Norwitzb59798b2003-03-21 01:43:31 +00004496/* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
Neal Norwitz2deaddb2003-03-21 03:08:31 +00004497/* IRIX has both /dev/ptc and /dev/ptmx, use ptmx */
4498#if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
Neal Norwitzb59798b2003-03-21 01:43:31 +00004499#define DEV_PTY_FILE "/dev/ptc"
4500#define HAVE_DEV_PTMX
4501#else
4502#define DEV_PTY_FILE "/dev/ptmx"
4503#endif
4504
Martin v. Löwis24a880b2002-12-31 12:55:15 +00004505#if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX)
Fred Drake8cef4cf2000-06-28 16:40:38 +00004506#ifdef HAVE_PTY_H
4507#include <pty.h>
4508#else
4509#ifdef HAVE_LIBUTIL_H
4510#include <libutil.h>
Ronald Oussoren755740f2010-02-07 19:56:39 +00004511#else
4512#ifdef HAVE_UTIL_H
4513#include <util.h>
4514#endif /* HAVE_UTIL_H */
Fred Drake8cef4cf2000-06-28 16:40:38 +00004515#endif /* HAVE_LIBUTIL_H */
4516#endif /* HAVE_PTY_H */
Martin v. Löwis14e73b12003-01-01 09:51:12 +00004517#ifdef HAVE_STROPTS_H
4518#include <stropts.h>
Martin v. Löwis24a880b2002-12-31 12:55:15 +00004519#endif
4520#endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00004521
Martin v. Löwis24a880b2002-12-31 12:55:15 +00004522#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004523PyDoc_STRVAR(posix_openpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004524"openpty() -> (master_fd, slave_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004525Open a pseudo-terminal, returning open fd's for both master and slave end.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00004526
4527static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004528posix_openpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00004529{
Victor Stinner8c62be82010-05-06 00:08:46 +00004530 int master_fd, slave_fd;
Thomas Wouters70c21a12000-07-14 14:28:33 +00004531#ifndef HAVE_OPENPTY
Victor Stinner8c62be82010-05-06 00:08:46 +00004532 char * slave_name;
Thomas Wouters70c21a12000-07-14 14:28:33 +00004533#endif
Martin v. Löwis24a880b2002-12-31 12:55:15 +00004534#if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
Victor Stinner8c62be82010-05-06 00:08:46 +00004535 PyOS_sighandler_t sig_saved;
Martin v. Löwis24a880b2002-12-31 12:55:15 +00004536#ifdef sun
Victor Stinner8c62be82010-05-06 00:08:46 +00004537 extern char *ptsname(int fildes);
Martin v. Löwis24a880b2002-12-31 12:55:15 +00004538#endif
4539#endif
Thomas Wouters70c21a12000-07-14 14:28:33 +00004540
Thomas Wouters70c21a12000-07-14 14:28:33 +00004541#ifdef HAVE_OPENPTY
Victor Stinner8c62be82010-05-06 00:08:46 +00004542 if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
4543 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00004544#elif defined(HAVE__GETPTY)
Victor Stinner8c62be82010-05-06 00:08:46 +00004545 slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
4546 if (slave_name == NULL)
4547 return posix_error();
Thomas Wouters70c21a12000-07-14 14:28:33 +00004548
Victor Stinner8c62be82010-05-06 00:08:46 +00004549 slave_fd = open(slave_name, O_RDWR);
4550 if (slave_fd < 0)
4551 return posix_error();
Martin v. Löwis24a880b2002-12-31 12:55:15 +00004552#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004553 master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
4554 if (master_fd < 0)
4555 return posix_error();
4556 sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
4557 /* change permission of slave */
4558 if (grantpt(master_fd) < 0) {
4559 PyOS_setsig(SIGCHLD, sig_saved);
4560 return posix_error();
4561 }
4562 /* unlock slave */
4563 if (unlockpt(master_fd) < 0) {
4564 PyOS_setsig(SIGCHLD, sig_saved);
4565 return posix_error();
4566 }
4567 PyOS_setsig(SIGCHLD, sig_saved);
4568 slave_name = ptsname(master_fd); /* get name of slave */
4569 if (slave_name == NULL)
4570 return posix_error();
4571 slave_fd = open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
4572 if (slave_fd < 0)
4573 return posix_error();
Neal Norwitzb59798b2003-03-21 01:43:31 +00004574#if !defined(__CYGWIN__) && !defined(HAVE_DEV_PTC)
Victor Stinner8c62be82010-05-06 00:08:46 +00004575 ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
4576 ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
Neal Norwitz6700e472002-12-31 16:16:07 +00004577#ifndef __hpux
Victor Stinner8c62be82010-05-06 00:08:46 +00004578 ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
Neal Norwitz6700e472002-12-31 16:16:07 +00004579#endif /* __hpux */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00004580#endif /* HAVE_CYGWIN */
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00004581#endif /* HAVE_OPENPTY */
Thomas Wouters70c21a12000-07-14 14:28:33 +00004582
Victor Stinner8c62be82010-05-06 00:08:46 +00004583 return Py_BuildValue("(ii)", master_fd, slave_fd);
Thomas Wouters70c21a12000-07-14 14:28:33 +00004584
Fred Drake8cef4cf2000-06-28 16:40:38 +00004585}
Martin v. Löwis24a880b2002-12-31 12:55:15 +00004586#endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
Fred Drake8cef4cf2000-06-28 16:40:38 +00004587
4588#ifdef HAVE_FORKPTY
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004589PyDoc_STRVAR(posix_forkpty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004590"forkpty() -> (pid, master_fd)\n\n\
Fred Drake8cef4cf2000-06-28 16:40:38 +00004591Fork a new process with a new pseudo-terminal as controlling tty.\n\n\
4592Like fork(), return 0 as pid to child process, and PID of child to parent.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004593To both, return fd of newly opened pseudo-terminal.\n");
Fred Drake8cef4cf2000-06-28 16:40:38 +00004594
4595static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004596posix_forkpty(PyObject *self, PyObject *noargs)
Fred Drake8cef4cf2000-06-28 16:40:38 +00004597{
Victor Stinner8c62be82010-05-06 00:08:46 +00004598 int master_fd = -1, result = 0;
4599 pid_t pid;
Tim Peters5aa91602002-01-30 05:46:57 +00004600
Victor Stinner8c62be82010-05-06 00:08:46 +00004601 _PyImport_AcquireLock();
4602 pid = forkpty(&master_fd, NULL, NULL, NULL);
4603 if (pid == 0) {
4604 /* child: this clobbers and resets the import lock. */
4605 PyOS_AfterFork();
4606 } else {
4607 /* parent: release the import lock. */
4608 result = _PyImport_ReleaseLock();
4609 }
4610 if (pid == -1)
4611 return posix_error();
4612 if (result < 0) {
4613 /* Don't clobber the OSError if the fork failed. */
4614 PyErr_SetString(PyExc_RuntimeError,
4615 "not holding the import lock");
4616 return NULL;
4617 }
4618 return Py_BuildValue("(Ni)", PyLong_FromPid(pid), master_fd);
Fred Drake8cef4cf2000-06-28 16:40:38 +00004619}
4620#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004621
Ross Lagerwall7807c352011-03-17 20:20:30 +02004622
Guido van Rossumad0ee831995-03-01 10:34:45 +00004623#ifdef HAVE_GETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004624PyDoc_STRVAR(posix_getegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004625"getegid() -> egid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004626Return the current process's effective group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004627
Barry Warsaw53699e91996-12-10 23:23:01 +00004628static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004629posix_getegid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00004630{
Victor Stinner8c62be82010-05-06 00:08:46 +00004631 return PyLong_FromLong((long)getegid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00004632}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004633#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00004634
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004635
Guido van Rossumad0ee831995-03-01 10:34:45 +00004636#ifdef HAVE_GETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004637PyDoc_STRVAR(posix_geteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004638"geteuid() -> euid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004639Return the current process's effective user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004640
Barry Warsaw53699e91996-12-10 23:23:01 +00004641static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004642posix_geteuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00004643{
Victor Stinner8c62be82010-05-06 00:08:46 +00004644 return PyLong_FromLong((long)geteuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00004645}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004646#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00004647
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004648
Guido van Rossumad0ee831995-03-01 10:34:45 +00004649#ifdef HAVE_GETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004650PyDoc_STRVAR(posix_getgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004651"getgid() -> gid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004652Return the current process's group id.");
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_getgid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00004656{
Victor Stinner8c62be82010-05-06 00:08:46 +00004657 return PyLong_FromLong((long)getgid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00004658}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004659#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00004660
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004661
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004662PyDoc_STRVAR(posix_getpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004663"getpid() -> pid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004664Return the current process id");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004665
Barry Warsaw53699e91996-12-10 23:23:01 +00004666static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004667posix_getpid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004668{
Victor Stinner8c62be82010-05-06 00:08:46 +00004669 return PyLong_FromPid(getpid());
Guido van Rossum85e3b011991-06-03 12:42:10 +00004670}
4671
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004672
Fred Drakec9680921999-12-13 16:37:25 +00004673#ifdef HAVE_GETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004674PyDoc_STRVAR(posix_getgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004675"getgroups() -> list of group IDs\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004676Return list of supplemental group IDs for the process.");
Fred Drakec9680921999-12-13 16:37:25 +00004677
4678static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004679posix_getgroups(PyObject *self, PyObject *noargs)
Fred Drakec9680921999-12-13 16:37:25 +00004680{
4681 PyObject *result = NULL;
4682
Fred Drakec9680921999-12-13 16:37:25 +00004683#ifdef NGROUPS_MAX
4684#define MAX_GROUPS NGROUPS_MAX
4685#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004686 /* defined to be 16 on Solaris7, so this should be a small number */
Fred Drakec9680921999-12-13 16:37:25 +00004687#define MAX_GROUPS 64
4688#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004689 gid_t grouplist[MAX_GROUPS];
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00004690
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00004691 /* On MacOSX getgroups(2) can return more than MAX_GROUPS results
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00004692 * This is a helper variable to store the intermediate result when
4693 * that happens.
4694 *
4695 * To keep the code readable the OSX behaviour is unconditional,
4696 * according to the POSIX spec this should be safe on all unix-y
4697 * systems.
4698 */
4699 gid_t* alt_grouplist = grouplist;
Victor Stinner8c62be82010-05-06 00:08:46 +00004700 int n;
Fred Drakec9680921999-12-13 16:37:25 +00004701
Victor Stinner8c62be82010-05-06 00:08:46 +00004702 n = getgroups(MAX_GROUPS, grouplist);
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00004703 if (n < 0) {
4704 if (errno == EINVAL) {
4705 n = getgroups(0, NULL);
4706 if (n == -1) {
4707 return posix_error();
4708 }
4709 if (n == 0) {
4710 /* Avoid malloc(0) */
4711 alt_grouplist = grouplist;
4712 } else {
4713 alt_grouplist = PyMem_Malloc(n * sizeof(gid_t));
4714 if (alt_grouplist == NULL) {
4715 errno = EINVAL;
4716 return posix_error();
4717 }
4718 n = getgroups(n, alt_grouplist);
4719 if (n == -1) {
4720 PyMem_Free(alt_grouplist);
4721 return posix_error();
4722 }
4723 }
4724 } else {
4725 return posix_error();
4726 }
4727 }
4728 result = PyList_New(n);
4729 if (result != NULL) {
Victor Stinner8c62be82010-05-06 00:08:46 +00004730 int i;
4731 for (i = 0; i < n; ++i) {
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00004732 PyObject *o = PyLong_FromLong((long)alt_grouplist[i]);
Victor Stinner8c62be82010-05-06 00:08:46 +00004733 if (o == NULL) {
Stefan Krah0e803b32010-11-26 16:16:47 +00004734 Py_DECREF(result);
4735 result = NULL;
4736 break;
Fred Drakec9680921999-12-13 16:37:25 +00004737 }
Victor Stinner8c62be82010-05-06 00:08:46 +00004738 PyList_SET_ITEM(result, i, o);
Fred Drakec9680921999-12-13 16:37:25 +00004739 }
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00004740 }
4741
4742 if (alt_grouplist != grouplist) {
4743 PyMem_Free(alt_grouplist);
Victor Stinner8c62be82010-05-06 00:08:46 +00004744 }
Neal Norwitze241ce82003-02-17 18:17:05 +00004745
Fred Drakec9680921999-12-13 16:37:25 +00004746 return result;
4747}
4748#endif
4749
Antoine Pitroub7572f02009-12-02 20:46:48 +00004750#ifdef HAVE_INITGROUPS
4751PyDoc_STRVAR(posix_initgroups__doc__,
4752"initgroups(username, gid) -> None\n\n\
4753Call the system initgroups() to initialize the group access list with all of\n\
4754the groups of which the specified username is a member, plus the specified\n\
4755group id.");
4756
4757static PyObject *
4758posix_initgroups(PyObject *self, PyObject *args)
4759{
Victor Stinner61ec5dc2010-08-15 09:22:44 +00004760 PyObject *oname;
Victor Stinner8c62be82010-05-06 00:08:46 +00004761 char *username;
Victor Stinner61ec5dc2010-08-15 09:22:44 +00004762 int res;
Victor Stinner8c62be82010-05-06 00:08:46 +00004763 long gid;
Antoine Pitroub7572f02009-12-02 20:46:48 +00004764
Victor Stinner61ec5dc2010-08-15 09:22:44 +00004765 if (!PyArg_ParseTuple(args, "O&l:initgroups",
4766 PyUnicode_FSConverter, &oname, &gid))
Victor Stinner8c62be82010-05-06 00:08:46 +00004767 return NULL;
Victor Stinner61ec5dc2010-08-15 09:22:44 +00004768 username = PyBytes_AS_STRING(oname);
Antoine Pitroub7572f02009-12-02 20:46:48 +00004769
Victor Stinner61ec5dc2010-08-15 09:22:44 +00004770 res = initgroups(username, (gid_t) gid);
4771 Py_DECREF(oname);
4772 if (res == -1)
Victor Stinner8c62be82010-05-06 00:08:46 +00004773 return PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitroub7572f02009-12-02 20:46:48 +00004774
Victor Stinner8c62be82010-05-06 00:08:46 +00004775 Py_INCREF(Py_None);
4776 return Py_None;
Antoine Pitroub7572f02009-12-02 20:46:48 +00004777}
4778#endif
4779
Martin v. Löwis606edc12002-06-13 21:09:11 +00004780#ifdef HAVE_GETPGID
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00004781PyDoc_STRVAR(posix_getpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004782"getpgid(pid) -> pgid\n\n\
Neal Norwitz0c2c17c2002-06-13 21:22:11 +00004783Call the system call getpgid().");
Martin v. Löwis606edc12002-06-13 21:09:11 +00004784
4785static PyObject *
4786posix_getpgid(PyObject *self, PyObject *args)
4787{
Victor Stinner8c62be82010-05-06 00:08:46 +00004788 pid_t pid, pgid;
4789 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":getpgid", &pid))
4790 return NULL;
4791 pgid = getpgid(pid);
4792 if (pgid < 0)
4793 return posix_error();
4794 return PyLong_FromPid(pgid);
Martin v. Löwis606edc12002-06-13 21:09:11 +00004795}
4796#endif /* HAVE_GETPGID */
4797
4798
Guido van Rossumb6775db1994-08-01 11:34:53 +00004799#ifdef HAVE_GETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004800PyDoc_STRVAR(posix_getpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004801"getpgrp() -> pgrp\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004802Return the current process group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004803
Barry Warsaw53699e91996-12-10 23:23:01 +00004804static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004805posix_getpgrp(PyObject *self, PyObject *noargs)
Guido van Rossum04814471991-06-04 20:23:49 +00004806{
Guido van Rossumb6775db1994-08-01 11:34:53 +00004807#ifdef GETPGRP_HAVE_ARG
Victor Stinner8c62be82010-05-06 00:08:46 +00004808 return PyLong_FromPid(getpgrp(0));
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004809#else /* GETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00004810 return PyLong_FromPid(getpgrp());
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00004811#endif /* GETPGRP_HAVE_ARG */
Guido van Rossum04814471991-06-04 20:23:49 +00004812}
Guido van Rossumb6775db1994-08-01 11:34:53 +00004813#endif /* HAVE_GETPGRP */
Guido van Rossum04814471991-06-04 20:23:49 +00004814
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004815
Guido van Rossumb6775db1994-08-01 11:34:53 +00004816#ifdef HAVE_SETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004817PyDoc_STRVAR(posix_setpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004818"setpgrp()\n\n\
Senthil Kumaran684760a2010-06-17 16:48:06 +00004819Make this process the process group leader.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004820
Barry Warsaw53699e91996-12-10 23:23:01 +00004821static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004822posix_setpgrp(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00004823{
Guido van Rossum64933891994-10-20 21:56:42 +00004824#ifdef SETPGRP_HAVE_ARG
Victor Stinner8c62be82010-05-06 00:08:46 +00004825 if (setpgrp(0, 0) < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00004826#else /* SETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00004827 if (setpgrp() < 0)
Guido van Rossum64933891994-10-20 21:56:42 +00004828#endif /* SETPGRP_HAVE_ARG */
Victor Stinner8c62be82010-05-06 00:08:46 +00004829 return posix_error();
4830 Py_INCREF(Py_None);
4831 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00004832}
4833
Guido van Rossumb6775db1994-08-01 11:34:53 +00004834#endif /* HAVE_SETPGRP */
4835
Guido van Rossumad0ee831995-03-01 10:34:45 +00004836#ifdef HAVE_GETPPID
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00004837
4838#ifdef MS_WINDOWS
4839#include <tlhelp32.h>
4840
4841static PyObject*
4842win32_getppid()
4843{
4844 HANDLE snapshot;
4845 pid_t mypid;
4846 PyObject* result = NULL;
4847 BOOL have_record;
4848 PROCESSENTRY32 pe;
4849
4850 mypid = getpid(); /* This function never fails */
4851
4852 snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
4853 if (snapshot == INVALID_HANDLE_VALUE)
4854 return PyErr_SetFromWindowsErr(GetLastError());
4855
4856 pe.dwSize = sizeof(pe);
4857 have_record = Process32First(snapshot, &pe);
4858 while (have_record) {
4859 if (mypid == (pid_t)pe.th32ProcessID) {
4860 /* We could cache the ulong value in a static variable. */
4861 result = PyLong_FromPid((pid_t)pe.th32ParentProcessID);
4862 break;
4863 }
4864
4865 have_record = Process32Next(snapshot, &pe);
4866 }
4867
4868 /* If our loop exits and our pid was not found (result will be NULL)
4869 * then GetLastError will return ERROR_NO_MORE_FILES. This is an
4870 * error anyway, so let's raise it. */
4871 if (!result)
4872 result = PyErr_SetFromWindowsErr(GetLastError());
4873
4874 CloseHandle(snapshot);
4875
4876 return result;
4877}
4878#endif /*MS_WINDOWS*/
4879
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004880PyDoc_STRVAR(posix_getppid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004881"getppid() -> ppid\n\n\
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00004882Return the parent's process id. If the parent process has already exited,\n\
4883Windows machines will still return its id; others systems will return the id\n\
4884of the 'init' process (1).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004885
Barry Warsaw53699e91996-12-10 23:23:01 +00004886static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004887posix_getppid(PyObject *self, PyObject *noargs)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004888{
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00004889#ifdef MS_WINDOWS
4890 return win32_getppid();
4891#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004892 return PyLong_FromPid(getppid());
Guido van Rossumad0ee831995-03-01 10:34:45 +00004893#endif
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00004894}
4895#endif /* HAVE_GETPPID */
Guido van Rossum85e3b011991-06-03 12:42:10 +00004896
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004897
Fred Drake12c6e2d1999-12-14 21:25:03 +00004898#ifdef HAVE_GETLOGIN
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004899PyDoc_STRVAR(posix_getlogin__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004900"getlogin() -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004901Return the actual login name.");
Fred Drake12c6e2d1999-12-14 21:25:03 +00004902
4903static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004904posix_getlogin(PyObject *self, PyObject *noargs)
Fred Drake12c6e2d1999-12-14 21:25:03 +00004905{
Victor Stinner8c62be82010-05-06 00:08:46 +00004906 PyObject *result = NULL;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00004907#ifdef MS_WINDOWS
Brian Curtine8e4b3b2010-09-23 20:04:14 +00004908 wchar_t user_name[UNLEN + 1];
4909 DWORD num_chars = sizeof(user_name)/sizeof(user_name[0]);
4910
4911 if (GetUserNameW(user_name, &num_chars)) {
4912 /* num_chars is the number of unicode chars plus null terminator */
4913 result = PyUnicode_FromWideChar(user_name, num_chars - 1);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00004914 }
4915 else
Brian Curtine8e4b3b2010-09-23 20:04:14 +00004916 result = PyErr_SetFromWindowsErr(GetLastError());
4917#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004918 char *name;
4919 int old_errno = errno;
Fred Drake12c6e2d1999-12-14 21:25:03 +00004920
Victor Stinner8c62be82010-05-06 00:08:46 +00004921 errno = 0;
4922 name = getlogin();
4923 if (name == NULL) {
4924 if (errno)
Victor Stinnere039ffe2010-08-15 09:33:08 +00004925 posix_error();
Fred Drake12c6e2d1999-12-14 21:25:03 +00004926 else
Victor Stinnere039ffe2010-08-15 09:33:08 +00004927 PyErr_SetString(PyExc_OSError, "unable to determine login name");
Victor Stinner8c62be82010-05-06 00:08:46 +00004928 }
4929 else
Victor Stinnere039ffe2010-08-15 09:33:08 +00004930 result = PyUnicode_DecodeFSDefault(name);
Victor Stinner8c62be82010-05-06 00:08:46 +00004931 errno = old_errno;
Brian Curtine8e4b3b2010-09-23 20:04:14 +00004932#endif
Fred Drake12c6e2d1999-12-14 21:25:03 +00004933 return result;
4934}
Brian Curtine8e4b3b2010-09-23 20:04:14 +00004935#endif /* HAVE_GETLOGIN */
Fred Drake12c6e2d1999-12-14 21:25:03 +00004936
Guido van Rossumad0ee831995-03-01 10:34:45 +00004937#ifdef HAVE_GETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004938PyDoc_STRVAR(posix_getuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004939"getuid() -> uid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004940Return the current process's user id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004941
Barry Warsaw53699e91996-12-10 23:23:01 +00004942static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00004943posix_getuid(PyObject *self, PyObject *noargs)
Guido van Rossum46003ff1992-05-15 11:05:24 +00004944{
Victor Stinner8c62be82010-05-06 00:08:46 +00004945 return PyLong_FromLong((long)getuid());
Guido van Rossum46003ff1992-05-15 11:05:24 +00004946}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004947#endif
Guido van Rossum46003ff1992-05-15 11:05:24 +00004948
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004949
Guido van Rossumad0ee831995-03-01 10:34:45 +00004950#ifdef HAVE_KILL
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004951PyDoc_STRVAR(posix_kill__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004952"kill(pid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004953Kill a process with a signal.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00004954
Barry Warsaw53699e91996-12-10 23:23:01 +00004955static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00004956posix_kill(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00004957{
Victor Stinner8c62be82010-05-06 00:08:46 +00004958 pid_t pid;
4959 int sig;
4960 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:kill", &pid, &sig))
4961 return NULL;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +00004962#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004963 if (sig == XCPT_SIGNAL_INTR || sig == XCPT_SIGNAL_BREAK) {
4964 APIRET rc;
4965 if ((rc = DosSendSignalException(pid, sig)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004966 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004967
4968 } else if (sig == XCPT_SIGNAL_KILLPROC) {
4969 APIRET rc;
4970 if ((rc = DosKillProcess(DKP_PROCESS, pid)) != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00004971 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004972
4973 } else
Guido van Rossumc5a0f531997-12-02 20:36:02 +00004974 return NULL; /* Unrecognized Signal Requested */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004975#else
Victor Stinner8c62be82010-05-06 00:08:46 +00004976 if (kill(pid, sig) == -1)
4977 return posix_error();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004978#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00004979 Py_INCREF(Py_None);
4980 return Py_None;
Guido van Rossum85e3b011991-06-03 12:42:10 +00004981}
Guido van Rossumad0ee831995-03-01 10:34:45 +00004982#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00004983
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004984#ifdef HAVE_KILLPG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004985PyDoc_STRVAR(posix_killpg__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00004986"killpg(pgid, sig)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004987Kill a process group with a signal.");
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00004988
4989static PyObject *
4990posix_killpg(PyObject *self, PyObject *args)
4991{
Victor Stinner8c62be82010-05-06 00:08:46 +00004992 int sig;
4993 pid_t pgid;
4994 /* XXX some man pages make the `pgid` parameter an int, others
4995 a pid_t. Since getpgrp() returns a pid_t, we assume killpg should
4996 take the same type. Moreover, pid_t is always at least as wide as
4997 int (else compilation of this module fails), which is safe. */
4998 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:killpg", &pgid, &sig))
4999 return NULL;
5000 if (killpg(pgid, sig) == -1)
5001 return posix_error();
5002 Py_INCREF(Py_None);
5003 return Py_None;
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00005004}
5005#endif
5006
Brian Curtineb24d742010-04-12 17:16:38 +00005007#ifdef MS_WINDOWS
5008PyDoc_STRVAR(win32_kill__doc__,
5009"kill(pid, sig)\n\n\
5010Kill a process with a signal.");
5011
5012static PyObject *
5013win32_kill(PyObject *self, PyObject *args)
5014{
Amaury Forgeot d'Arc0a589c92010-05-15 20:35:12 +00005015 PyObject *result;
Victor Stinner8c62be82010-05-06 00:08:46 +00005016 DWORD pid, sig, err;
5017 HANDLE handle;
Brian Curtineb24d742010-04-12 17:16:38 +00005018
Victor Stinner8c62be82010-05-06 00:08:46 +00005019 if (!PyArg_ParseTuple(args, "kk:kill", &pid, &sig))
5020 return NULL;
Brian Curtineb24d742010-04-12 17:16:38 +00005021
Victor Stinner8c62be82010-05-06 00:08:46 +00005022 /* Console processes which share a common console can be sent CTRL+C or
5023 CTRL+BREAK events, provided they handle said events. */
5024 if (sig == CTRL_C_EVENT || sig == CTRL_BREAK_EVENT) {
5025 if (GenerateConsoleCtrlEvent(sig, pid) == 0) {
5026 err = GetLastError();
5027 PyErr_SetFromWindowsErr(err);
5028 }
5029 else
5030 Py_RETURN_NONE;
5031 }
Brian Curtineb24d742010-04-12 17:16:38 +00005032
Victor Stinner8c62be82010-05-06 00:08:46 +00005033 /* If the signal is outside of what GenerateConsoleCtrlEvent can use,
5034 attempt to open and terminate the process. */
5035 handle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, pid);
5036 if (handle == NULL) {
5037 err = GetLastError();
5038 return PyErr_SetFromWindowsErr(err);
5039 }
Brian Curtineb24d742010-04-12 17:16:38 +00005040
Victor Stinner8c62be82010-05-06 00:08:46 +00005041 if (TerminateProcess(handle, sig) == 0) {
5042 err = GetLastError();
5043 result = PyErr_SetFromWindowsErr(err);
5044 } else {
5045 Py_INCREF(Py_None);
5046 result = Py_None;
5047 }
Brian Curtineb24d742010-04-12 17:16:38 +00005048
Victor Stinner8c62be82010-05-06 00:08:46 +00005049 CloseHandle(handle);
5050 return result;
Brian Curtineb24d742010-04-12 17:16:38 +00005051}
5052#endif /* MS_WINDOWS */
5053
Guido van Rossumc0125471996-06-28 18:55:32 +00005054#ifdef HAVE_PLOCK
5055
5056#ifdef HAVE_SYS_LOCK_H
5057#include <sys/lock.h>
5058#endif
5059
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005060PyDoc_STRVAR(posix_plock__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005061"plock(op)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005062Lock program segments into memory.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005063
Barry Warsaw53699e91996-12-10 23:23:01 +00005064static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005065posix_plock(PyObject *self, PyObject *args)
Guido van Rossumc0125471996-06-28 18:55:32 +00005066{
Victor Stinner8c62be82010-05-06 00:08:46 +00005067 int op;
5068 if (!PyArg_ParseTuple(args, "i:plock", &op))
5069 return NULL;
5070 if (plock(op) == -1)
5071 return posix_error();
5072 Py_INCREF(Py_None);
5073 return Py_None;
Guido van Rossumc0125471996-06-28 18:55:32 +00005074}
5075#endif
5076
Guido van Rossumb6775db1994-08-01 11:34:53 +00005077#ifdef HAVE_SETUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005078PyDoc_STRVAR(posix_setuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005079"setuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005080Set the current process's user id.");
5081
Barry Warsaw53699e91996-12-10 23:23:01 +00005082static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005083posix_setuid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005084{
Victor Stinner8c62be82010-05-06 00:08:46 +00005085 long uid_arg;
5086 uid_t uid;
5087 if (!PyArg_ParseTuple(args, "l:setuid", &uid_arg))
5088 return NULL;
5089 uid = uid_arg;
5090 if (uid != uid_arg) {
5091 PyErr_SetString(PyExc_OverflowError, "user id too big");
5092 return NULL;
5093 }
5094 if (setuid(uid) < 0)
5095 return posix_error();
5096 Py_INCREF(Py_None);
5097 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005098}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005099#endif /* HAVE_SETUID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005100
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005101
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005102#ifdef HAVE_SETEUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005103PyDoc_STRVAR(posix_seteuid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005104"seteuid(uid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005105Set the current process's effective user id.");
5106
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005107static PyObject *
5108posix_seteuid (PyObject *self, PyObject *args)
5109{
Victor Stinner8c62be82010-05-06 00:08:46 +00005110 long euid_arg;
5111 uid_t euid;
5112 if (!PyArg_ParseTuple(args, "l", &euid_arg))
5113 return NULL;
5114 euid = euid_arg;
5115 if (euid != euid_arg) {
5116 PyErr_SetString(PyExc_OverflowError, "user id too big");
5117 return NULL;
5118 }
5119 if (seteuid(euid) < 0) {
5120 return posix_error();
5121 } else {
5122 Py_INCREF(Py_None);
5123 return Py_None;
5124 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005125}
5126#endif /* HAVE_SETEUID */
5127
5128#ifdef HAVE_SETEGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005129PyDoc_STRVAR(posix_setegid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005130"setegid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005131Set the current process's effective group id.");
5132
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005133static PyObject *
5134posix_setegid (PyObject *self, PyObject *args)
5135{
Victor Stinner8c62be82010-05-06 00:08:46 +00005136 long egid_arg;
5137 gid_t egid;
5138 if (!PyArg_ParseTuple(args, "l", &egid_arg))
5139 return NULL;
5140 egid = egid_arg;
5141 if (egid != egid_arg) {
5142 PyErr_SetString(PyExc_OverflowError, "group id too big");
5143 return NULL;
5144 }
5145 if (setegid(egid) < 0) {
5146 return posix_error();
5147 } else {
5148 Py_INCREF(Py_None);
5149 return Py_None;
5150 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005151}
5152#endif /* HAVE_SETEGID */
5153
5154#ifdef HAVE_SETREUID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005155PyDoc_STRVAR(posix_setreuid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00005156"setreuid(ruid, euid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005157Set the current process's real and effective user ids.");
5158
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005159static PyObject *
5160posix_setreuid (PyObject *self, PyObject *args)
5161{
Victor Stinner8c62be82010-05-06 00:08:46 +00005162 long ruid_arg, euid_arg;
5163 uid_t ruid, euid;
5164 if (!PyArg_ParseTuple(args, "ll", &ruid_arg, &euid_arg))
5165 return NULL;
5166 if (ruid_arg == -1)
5167 ruid = (uid_t)-1; /* let the compiler choose how -1 fits */
5168 else
5169 ruid = ruid_arg; /* otherwise, assign from our long */
5170 if (euid_arg == -1)
5171 euid = (uid_t)-1;
5172 else
5173 euid = euid_arg;
5174 if ((euid_arg != -1 && euid != euid_arg) ||
5175 (ruid_arg != -1 && ruid != ruid_arg)) {
5176 PyErr_SetString(PyExc_OverflowError, "user id too big");
5177 return NULL;
5178 }
5179 if (setreuid(ruid, euid) < 0) {
5180 return posix_error();
5181 } else {
5182 Py_INCREF(Py_None);
5183 return Py_None;
5184 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005185}
5186#endif /* HAVE_SETREUID */
5187
5188#ifdef HAVE_SETREGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005189PyDoc_STRVAR(posix_setregid__doc__,
Neal Norwitz94f1d712004-02-16 01:26:34 +00005190"setregid(rgid, egid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005191Set the current process's real and effective group ids.");
5192
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005193static PyObject *
5194posix_setregid (PyObject *self, PyObject *args)
5195{
Victor Stinner8c62be82010-05-06 00:08:46 +00005196 long rgid_arg, egid_arg;
5197 gid_t rgid, egid;
5198 if (!PyArg_ParseTuple(args, "ll", &rgid_arg, &egid_arg))
5199 return NULL;
5200 if (rgid_arg == -1)
5201 rgid = (gid_t)-1; /* let the compiler choose how -1 fits */
5202 else
5203 rgid = rgid_arg; /* otherwise, assign from our long */
5204 if (egid_arg == -1)
5205 egid = (gid_t)-1;
5206 else
5207 egid = egid_arg;
5208 if ((egid_arg != -1 && egid != egid_arg) ||
5209 (rgid_arg != -1 && rgid != rgid_arg)) {
5210 PyErr_SetString(PyExc_OverflowError, "group id too big");
5211 return NULL;
5212 }
5213 if (setregid(rgid, egid) < 0) {
5214 return posix_error();
5215 } else {
5216 Py_INCREF(Py_None);
5217 return Py_None;
5218 }
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00005219}
5220#endif /* HAVE_SETREGID */
5221
Guido van Rossumb6775db1994-08-01 11:34:53 +00005222#ifdef HAVE_SETGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005223PyDoc_STRVAR(posix_setgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005224"setgid(gid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005225Set the current process's group id.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005226
Barry Warsaw53699e91996-12-10 23:23:01 +00005227static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005228posix_setgid(PyObject *self, PyObject *args)
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005229{
Victor Stinner8c62be82010-05-06 00:08:46 +00005230 long gid_arg;
5231 gid_t gid;
5232 if (!PyArg_ParseTuple(args, "l:setgid", &gid_arg))
5233 return NULL;
5234 gid = gid_arg;
5235 if (gid != gid_arg) {
5236 PyErr_SetString(PyExc_OverflowError, "group id too big");
5237 return NULL;
5238 }
5239 if (setgid(gid) < 0)
5240 return posix_error();
5241 Py_INCREF(Py_None);
5242 return Py_None;
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005243}
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00005244#endif /* HAVE_SETGID */
Guido van Rossuma3d78fb1993-11-10 09:23:53 +00005245
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005246#ifdef HAVE_SETGROUPS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005247PyDoc_STRVAR(posix_setgroups__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005248"setgroups(list)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005249Set the groups of the current process to list.");
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005250
5251static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00005252posix_setgroups(PyObject *self, PyObject *groups)
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005253{
Victor Stinner8c62be82010-05-06 00:08:46 +00005254 int i, len;
5255 gid_t grouplist[MAX_GROUPS];
Tim Peters5aa91602002-01-30 05:46:57 +00005256
Victor Stinner8c62be82010-05-06 00:08:46 +00005257 if (!PySequence_Check(groups)) {
5258 PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
5259 return NULL;
5260 }
5261 len = PySequence_Size(groups);
5262 if (len > MAX_GROUPS) {
5263 PyErr_SetString(PyExc_ValueError, "too many groups");
5264 return NULL;
5265 }
5266 for(i = 0; i < len; i++) {
5267 PyObject *elem;
5268 elem = PySequence_GetItem(groups, i);
5269 if (!elem)
5270 return NULL;
5271 if (!PyLong_Check(elem)) {
5272 PyErr_SetString(PyExc_TypeError,
5273 "groups must be integers");
5274 Py_DECREF(elem);
5275 return NULL;
5276 } else {
5277 unsigned long x = PyLong_AsUnsignedLong(elem);
5278 if (PyErr_Occurred()) {
5279 PyErr_SetString(PyExc_TypeError,
5280 "group id too big");
5281 Py_DECREF(elem);
5282 return NULL;
5283 }
5284 grouplist[i] = x;
5285 /* read back the value to see if it fitted in gid_t */
5286 if (grouplist[i] != x) {
5287 PyErr_SetString(PyExc_TypeError,
5288 "group id too big");
5289 Py_DECREF(elem);
5290 return NULL;
5291 }
5292 }
5293 Py_DECREF(elem);
5294 }
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005295
Victor Stinner8c62be82010-05-06 00:08:46 +00005296 if (setgroups(len, grouplist) < 0)
5297 return posix_error();
5298 Py_INCREF(Py_None);
5299 return Py_None;
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00005300}
5301#endif /* HAVE_SETGROUPS */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005302
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005303#if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
5304static PyObject *
Christian Heimes292d3512008-02-03 16:51:08 +00005305wait_helper(pid_t pid, int status, struct rusage *ru)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005306{
Victor Stinner8c62be82010-05-06 00:08:46 +00005307 PyObject *result;
5308 static PyObject *struct_rusage;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005309
Victor Stinner8c62be82010-05-06 00:08:46 +00005310 if (pid == -1)
5311 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005312
Victor Stinner8c62be82010-05-06 00:08:46 +00005313 if (struct_rusage == NULL) {
5314 PyObject *m = PyImport_ImportModuleNoBlock("resource");
5315 if (m == NULL)
5316 return NULL;
5317 struct_rusage = PyObject_GetAttrString(m, "struct_rusage");
5318 Py_DECREF(m);
5319 if (struct_rusage == NULL)
5320 return NULL;
5321 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005322
Victor Stinner8c62be82010-05-06 00:08:46 +00005323 /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
5324 result = PyStructSequence_New((PyTypeObject*) struct_rusage);
5325 if (!result)
5326 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005327
5328#ifndef doubletime
5329#define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
5330#endif
5331
Victor Stinner8c62be82010-05-06 00:08:46 +00005332 PyStructSequence_SET_ITEM(result, 0,
5333 PyFloat_FromDouble(doubletime(ru->ru_utime)));
5334 PyStructSequence_SET_ITEM(result, 1,
5335 PyFloat_FromDouble(doubletime(ru->ru_stime)));
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005336#define SET_INT(result, index, value)\
Victor Stinner8c62be82010-05-06 00:08:46 +00005337 PyStructSequence_SET_ITEM(result, index, PyLong_FromLong(value))
5338 SET_INT(result, 2, ru->ru_maxrss);
5339 SET_INT(result, 3, ru->ru_ixrss);
5340 SET_INT(result, 4, ru->ru_idrss);
5341 SET_INT(result, 5, ru->ru_isrss);
5342 SET_INT(result, 6, ru->ru_minflt);
5343 SET_INT(result, 7, ru->ru_majflt);
5344 SET_INT(result, 8, ru->ru_nswap);
5345 SET_INT(result, 9, ru->ru_inblock);
5346 SET_INT(result, 10, ru->ru_oublock);
5347 SET_INT(result, 11, ru->ru_msgsnd);
5348 SET_INT(result, 12, ru->ru_msgrcv);
5349 SET_INT(result, 13, ru->ru_nsignals);
5350 SET_INT(result, 14, ru->ru_nvcsw);
5351 SET_INT(result, 15, ru->ru_nivcsw);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005352#undef SET_INT
5353
Victor Stinner8c62be82010-05-06 00:08:46 +00005354 if (PyErr_Occurred()) {
5355 Py_DECREF(result);
5356 return NULL;
5357 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005358
Victor Stinner8c62be82010-05-06 00:08:46 +00005359 return Py_BuildValue("NiN", PyLong_FromPid(pid), status, result);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005360}
5361#endif /* HAVE_WAIT3 || HAVE_WAIT4 */
5362
5363#ifdef HAVE_WAIT3
5364PyDoc_STRVAR(posix_wait3__doc__,
5365"wait3(options) -> (pid, status, rusage)\n\n\
5366Wait for completion of a child process.");
5367
5368static PyObject *
5369posix_wait3(PyObject *self, PyObject *args)
5370{
Victor Stinner8c62be82010-05-06 00:08:46 +00005371 pid_t pid;
5372 int options;
5373 struct rusage ru;
5374 WAIT_TYPE status;
5375 WAIT_STATUS_INT(status) = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005376
Victor Stinner8c62be82010-05-06 00:08:46 +00005377 if (!PyArg_ParseTuple(args, "i:wait3", &options))
5378 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005379
Victor Stinner8c62be82010-05-06 00:08:46 +00005380 Py_BEGIN_ALLOW_THREADS
5381 pid = wait3(&status, options, &ru);
5382 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005383
Victor Stinner8c62be82010-05-06 00:08:46 +00005384 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005385}
5386#endif /* HAVE_WAIT3 */
5387
5388#ifdef HAVE_WAIT4
5389PyDoc_STRVAR(posix_wait4__doc__,
5390"wait4(pid, options) -> (pid, status, rusage)\n\n\
5391Wait for completion of a given child process.");
5392
5393static PyObject *
5394posix_wait4(PyObject *self, PyObject *args)
5395{
Victor Stinner8c62be82010-05-06 00:08:46 +00005396 pid_t pid;
5397 int options;
5398 struct rusage ru;
5399 WAIT_TYPE status;
5400 WAIT_STATUS_INT(status) = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005401
Victor Stinner8c62be82010-05-06 00:08:46 +00005402 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:wait4", &pid, &options))
5403 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005404
Victor Stinner8c62be82010-05-06 00:08:46 +00005405 Py_BEGIN_ALLOW_THREADS
5406 pid = wait4(pid, &status, options, &ru);
5407 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005408
Victor Stinner8c62be82010-05-06 00:08:46 +00005409 return wait_helper(pid, WAIT_STATUS_INT(status), &ru);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005410}
5411#endif /* HAVE_WAIT4 */
5412
Ross Lagerwall7807c352011-03-17 20:20:30 +02005413#if defined(HAVE_WAITID) && !defined(__APPLE__)
5414PyDoc_STRVAR(posix_waitid__doc__,
5415"waitid(idtype, id, options) -> waitid_result\n\n\
5416Wait for the completion of one or more child processes.\n\n\
5417idtype can be P_PID, P_PGID or P_ALL.\n\
5418id specifies the pid to wait on.\n\
5419options is constructed from the ORing of one or more of WEXITED, WSTOPPED\n\
5420or WCONTINUED and additionally may be ORed with WNOHANG or WNOWAIT.\n\
5421Returns either waitid_result or None if WNOHANG is specified and there are\n\
5422no children in a waitable state.");
5423
5424static PyObject *
5425posix_waitid(PyObject *self, PyObject *args)
5426{
5427 PyObject *result;
5428 idtype_t idtype;
5429 id_t id;
5430 int options, res;
5431 siginfo_t si;
5432 si.si_pid = 0;
5433 if (!PyArg_ParseTuple(args, "i" _Py_PARSE_PID "i:waitid", &idtype, &id, &options))
5434 return NULL;
5435 Py_BEGIN_ALLOW_THREADS
5436 res = waitid(idtype, id, &si, options);
5437 Py_END_ALLOW_THREADS
5438 if (res == -1)
5439 return posix_error();
5440
5441 if (si.si_pid == 0)
5442 Py_RETURN_NONE;
5443
5444 result = PyStructSequence_New(&WaitidResultType);
5445 if (!result)
5446 return NULL;
5447
5448 PyStructSequence_SET_ITEM(result, 0, PyLong_FromPid(si.si_pid));
5449 PyStructSequence_SET_ITEM(result, 1, PyLong_FromPid(si.si_uid));
5450 PyStructSequence_SET_ITEM(result, 2, PyLong_FromLong((long)(si.si_signo)));
5451 PyStructSequence_SET_ITEM(result, 3, PyLong_FromLong((long)(si.si_status)));
5452 PyStructSequence_SET_ITEM(result, 4, PyLong_FromLong((long)(si.si_code)));
5453 if (PyErr_Occurred()) {
5454 Py_DECREF(result);
5455 return NULL;
5456 }
5457
5458 return result;
5459}
5460#endif
5461
Guido van Rossumb6775db1994-08-01 11:34:53 +00005462#ifdef HAVE_WAITPID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005463PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005464"waitpid(pid, options) -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005465Wait for completion of a given child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005466
Barry Warsaw53699e91996-12-10 23:23:01 +00005467static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005468posix_waitpid(PyObject *self, PyObject *args)
Guido van Rossum85e3b011991-06-03 12:42:10 +00005469{
Victor Stinner8c62be82010-05-06 00:08:46 +00005470 pid_t pid;
5471 int options;
5472 WAIT_TYPE status;
5473 WAIT_STATUS_INT(status) = 0;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00005474
Victor Stinner8c62be82010-05-06 00:08:46 +00005475 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:waitpid", &pid, &options))
5476 return NULL;
5477 Py_BEGIN_ALLOW_THREADS
5478 pid = waitpid(pid, &status, options);
5479 Py_END_ALLOW_THREADS
5480 if (pid == -1)
5481 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005482
Victor Stinner8c62be82010-05-06 00:08:46 +00005483 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum21803b81992-08-09 12:55:27 +00005484}
5485
Tim Petersab034fa2002-02-01 11:27:43 +00005486#elif defined(HAVE_CWAIT)
5487
5488/* MS C has a variant of waitpid() that's usable for most purposes. */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005489PyDoc_STRVAR(posix_waitpid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005490"waitpid(pid, options) -> (pid, status << 8)\n\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005491"Wait for completion of a given process. options is ignored on Windows.");
Tim Petersab034fa2002-02-01 11:27:43 +00005492
5493static PyObject *
5494posix_waitpid(PyObject *self, PyObject *args)
5495{
Victor Stinner8c62be82010-05-06 00:08:46 +00005496 Py_intptr_t pid;
5497 int status, options;
Tim Petersab034fa2002-02-01 11:27:43 +00005498
Victor Stinner8c62be82010-05-06 00:08:46 +00005499 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:waitpid", &pid, &options))
5500 return NULL;
5501 Py_BEGIN_ALLOW_THREADS
5502 pid = _cwait(&status, pid, options);
5503 Py_END_ALLOW_THREADS
5504 if (pid == -1)
5505 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005506
Victor Stinner8c62be82010-05-06 00:08:46 +00005507 /* shift the status left a byte so this is more like the POSIX waitpid */
5508 return Py_BuildValue("Ni", PyLong_FromPid(pid), status << 8);
Tim Petersab034fa2002-02-01 11:27:43 +00005509}
5510#endif /* HAVE_WAITPID || HAVE_CWAIT */
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005511
Guido van Rossumad0ee831995-03-01 10:34:45 +00005512#ifdef HAVE_WAIT
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005513PyDoc_STRVAR(posix_wait__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005514"wait() -> (pid, status)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005515Wait for completion of a child process.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005516
Barry Warsaw53699e91996-12-10 23:23:01 +00005517static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005518posix_wait(PyObject *self, PyObject *noargs)
Guido van Rossum21803b81992-08-09 12:55:27 +00005519{
Victor Stinner8c62be82010-05-06 00:08:46 +00005520 pid_t pid;
5521 WAIT_TYPE status;
5522 WAIT_STATUS_INT(status) = 0;
Neal Norwitze241ce82003-02-17 18:17:05 +00005523
Victor Stinner8c62be82010-05-06 00:08:46 +00005524 Py_BEGIN_ALLOW_THREADS
5525 pid = wait(&status);
5526 Py_END_ALLOW_THREADS
5527 if (pid == -1)
5528 return posix_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005529
Victor Stinner8c62be82010-05-06 00:08:46 +00005530 return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
Guido van Rossum85e3b011991-06-03 12:42:10 +00005531}
Guido van Rossumad0ee831995-03-01 10:34:45 +00005532#endif
Guido van Rossum85e3b011991-06-03 12:42:10 +00005533
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005534
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005535PyDoc_STRVAR(posix_lstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005536"lstat(path) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005537Like stat(path), but do not follow symbolic links.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005538
Barry Warsaw53699e91996-12-10 23:23:01 +00005539static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005540posix_lstat(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005541{
Guido van Rossumb6775db1994-08-01 11:34:53 +00005542#ifdef HAVE_LSTAT
Victor Stinner8c62be82010-05-06 00:08:46 +00005543 return posix_do_stat(self, args, "O&:lstat", lstat, NULL, NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +00005544#else /* !HAVE_LSTAT */
Mark Hammondc2e85bd2002-10-03 05:10:39 +00005545#ifdef MS_WINDOWS
Brian Curtind40e6f72010-07-08 21:39:08 +00005546 return posix_do_stat(self, args, "O&:lstat", STAT, "U:lstat",
5547 win32_lstat_w);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00005548#else
Victor Stinner8c62be82010-05-06 00:08:46 +00005549 return posix_do_stat(self, args, "O&:lstat", STAT, NULL, NULL);
Mark Hammondc2e85bd2002-10-03 05:10:39 +00005550#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00005551#endif /* !HAVE_LSTAT */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005552}
5553
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005554
Guido van Rossumb6775db1994-08-01 11:34:53 +00005555#ifdef HAVE_READLINK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005556PyDoc_STRVAR(posix_readlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005557"readlink(path) -> path\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005558Return a string representing the path to which the symbolic link points.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005559
Barry Warsaw53699e91996-12-10 23:23:01 +00005560static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005561posix_readlink(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005562{
Victor Stinner8c62be82010-05-06 00:08:46 +00005563 PyObject* v;
5564 char buf[MAXPATHLEN];
5565 PyObject *opath;
5566 char *path;
5567 int n;
5568 int arg_is_unicode = 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00005569
Victor Stinner8c62be82010-05-06 00:08:46 +00005570 if (!PyArg_ParseTuple(args, "O&:readlink",
5571 PyUnicode_FSConverter, &opath))
5572 return NULL;
5573 path = PyBytes_AsString(opath);
5574 v = PySequence_GetItem(args, 0);
5575 if (v == NULL) {
5576 Py_DECREF(opath);
5577 return NULL;
5578 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00005579
Victor Stinner8c62be82010-05-06 00:08:46 +00005580 if (PyUnicode_Check(v)) {
5581 arg_is_unicode = 1;
5582 }
5583 Py_DECREF(v);
Thomas Wouters89f507f2006-12-13 04:49:30 +00005584
Victor Stinner8c62be82010-05-06 00:08:46 +00005585 Py_BEGIN_ALLOW_THREADS
5586 n = readlink(path, buf, (int) sizeof buf);
5587 Py_END_ALLOW_THREADS
5588 if (n < 0)
5589 return posix_error_with_allocated_filename(opath);
Thomas Wouters89f507f2006-12-13 04:49:30 +00005590
Victor Stinner8c62be82010-05-06 00:08:46 +00005591 Py_DECREF(opath);
Victor Stinnera45598a2010-05-14 16:35:39 +00005592 if (arg_is_unicode)
5593 return PyUnicode_DecodeFSDefaultAndSize(buf, n);
5594 else
5595 return PyBytes_FromStringAndSize(buf, n);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005596}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005597#endif /* HAVE_READLINK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005598
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005599
Brian Curtin52173d42010-12-02 18:29:18 +00005600#if defined(HAVE_SYMLINK) && !defined(MS_WINDOWS)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005601PyDoc_STRVAR(posix_symlink__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005602"symlink(src, dst)\n\n\
Brett Cannon807413d2003-06-11 00:18:09 +00005603Create a symbolic link pointing to src named dst.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005604
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005605static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005606posix_symlink(PyObject *self, PyObject *args)
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005607{
Victor Stinner8c62be82010-05-06 00:08:46 +00005608 return posix_2str(args, "O&O&:symlink", symlink);
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005609}
5610#endif /* HAVE_SYMLINK */
5611
Brian Curtind40e6f72010-07-08 21:39:08 +00005612#if !defined(HAVE_READLINK) && defined(MS_WINDOWS)
5613
5614PyDoc_STRVAR(win_readlink__doc__,
5615"readlink(path) -> path\n\n\
5616Return a string representing the path to which the symbolic link points.");
5617
Brian Curtind40e6f72010-07-08 21:39:08 +00005618/* Windows readlink implementation */
5619static PyObject *
5620win_readlink(PyObject *self, PyObject *args)
5621{
5622 wchar_t *path;
5623 DWORD n_bytes_returned;
5624 DWORD io_result;
5625 PyObject *result;
5626 HANDLE reparse_point_handle;
5627
5628 char target_buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
5629 REPARSE_DATA_BUFFER *rdb = (REPARSE_DATA_BUFFER *)target_buffer;
5630 wchar_t *print_name;
5631
5632 if (!PyArg_ParseTuple(args,
5633 "u:readlink",
5634 &path))
5635 return NULL;
5636
5637 /* First get a handle to the reparse point */
5638 Py_BEGIN_ALLOW_THREADS
5639 reparse_point_handle = CreateFileW(
5640 path,
5641 0,
5642 0,
5643 0,
5644 OPEN_EXISTING,
5645 FILE_FLAG_OPEN_REPARSE_POINT|FILE_FLAG_BACKUP_SEMANTICS,
5646 0);
5647 Py_END_ALLOW_THREADS
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005648
Brian Curtind40e6f72010-07-08 21:39:08 +00005649 if (reparse_point_handle==INVALID_HANDLE_VALUE)
5650 {
5651 return win32_error_unicode("readlink", path);
5652 }
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005653
Brian Curtind40e6f72010-07-08 21:39:08 +00005654 Py_BEGIN_ALLOW_THREADS
5655 /* New call DeviceIoControl to read the reparse point */
5656 io_result = DeviceIoControl(
5657 reparse_point_handle,
5658 FSCTL_GET_REPARSE_POINT,
5659 0, 0, /* in buffer */
5660 target_buffer, sizeof(target_buffer),
5661 &n_bytes_returned,
5662 0 /* we're not using OVERLAPPED_IO */
5663 );
5664 CloseHandle(reparse_point_handle);
5665 Py_END_ALLOW_THREADS
5666
5667 if (io_result==0)
5668 {
5669 return win32_error_unicode("readlink", path);
5670 }
5671
5672 if (rdb->ReparseTag != IO_REPARSE_TAG_SYMLINK)
5673 {
5674 PyErr_SetString(PyExc_ValueError,
5675 "not a symbolic link");
5676 return NULL;
5677 }
Brian Curtin74e45612010-07-09 15:58:59 +00005678 print_name = rdb->SymbolicLinkReparseBuffer.PathBuffer +
5679 rdb->SymbolicLinkReparseBuffer.PrintNameOffset;
5680
5681 result = PyUnicode_FromWideChar(print_name,
5682 rdb->SymbolicLinkReparseBuffer.PrintNameLength/2);
Brian Curtind40e6f72010-07-08 21:39:08 +00005683 return result;
5684}
5685
5686#endif /* !defined(HAVE_READLINK) && defined(MS_WINDOWS) */
5687
Brian Curtin52173d42010-12-02 18:29:18 +00005688#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtind40e6f72010-07-08 21:39:08 +00005689
5690/* Grab CreateSymbolicLinkW dynamically from kernel32 */
5691static int has_CreateSymbolicLinkW = 0;
5692static DWORD (CALLBACK *Py_CreateSymbolicLinkW)(LPWSTR, LPWSTR, DWORD);
5693static int
5694check_CreateSymbolicLinkW()
5695{
5696 HINSTANCE hKernel32;
5697 /* only recheck */
5698 if (has_CreateSymbolicLinkW)
5699 return has_CreateSymbolicLinkW;
5700 hKernel32 = GetModuleHandle("KERNEL32");
Brian Curtin74e45612010-07-09 15:58:59 +00005701 *(FARPROC*)&Py_CreateSymbolicLinkW = GetProcAddress(hKernel32,
5702 "CreateSymbolicLinkW");
Brian Curtind40e6f72010-07-08 21:39:08 +00005703 if (Py_CreateSymbolicLinkW)
5704 has_CreateSymbolicLinkW = 1;
5705 return has_CreateSymbolicLinkW;
5706}
5707
5708PyDoc_STRVAR(win_symlink__doc__,
5709"symlink(src, dst, target_is_directory=False)\n\n\
5710Create a symbolic link pointing to src named dst.\n\
5711target_is_directory is required if the target is to be interpreted as\n\
5712a directory.\n\
5713This function requires Windows 6.0 or greater, and raises a\n\
5714NotImplementedError otherwise.");
5715
5716static PyObject *
5717win_symlink(PyObject *self, PyObject *args, PyObject *kwargs)
5718{
5719 static char *kwlist[] = {"src", "dest", "target_is_directory", NULL};
5720 PyObject *src, *dest;
5721 int target_is_directory = 0;
5722 DWORD res;
5723 WIN32_FILE_ATTRIBUTE_DATA src_info;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005724
Brian Curtind40e6f72010-07-08 21:39:08 +00005725 if (!check_CreateSymbolicLinkW())
5726 {
5727 /* raise NotImplementedError */
5728 return PyErr_Format(PyExc_NotImplementedError,
5729 "CreateSymbolicLinkW not found");
5730 }
5731 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|i:symlink",
5732 kwlist, &src, &dest, &target_is_directory))
5733 return NULL;
Brian Curtin3b4499c2010-12-28 14:31:47 +00005734
5735 if (win32_can_symlink == 0)
5736 return PyErr_Format(PyExc_OSError, "symbolic link privilege not held");
5737
Brian Curtind40e6f72010-07-08 21:39:08 +00005738 if (!convert_to_unicode(&src)) { return NULL; }
5739 if (!convert_to_unicode(&dest)) {
5740 Py_DECREF(src);
5741 return NULL;
5742 }
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005743
Brian Curtind40e6f72010-07-08 21:39:08 +00005744 /* if src is a directory, ensure target_is_directory==1 */
5745 if(
5746 GetFileAttributesExW(
5747 PyUnicode_AsUnicode(src), GetFileExInfoStandard, &src_info
5748 ))
5749 {
5750 target_is_directory = target_is_directory ||
5751 (src_info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY);
5752 }
5753
5754 Py_BEGIN_ALLOW_THREADS
5755 res = Py_CreateSymbolicLinkW(
5756 PyUnicode_AsUnicode(dest),
5757 PyUnicode_AsUnicode(src),
5758 target_is_directory);
5759 Py_END_ALLOW_THREADS
5760 Py_DECREF(src);
5761 Py_DECREF(dest);
5762 if (!res)
5763 {
5764 return win32_error_unicode("symlink", PyUnicode_AsUnicode(src));
5765 }
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00005766
Brian Curtind40e6f72010-07-08 21:39:08 +00005767 Py_INCREF(Py_None);
5768 return Py_None;
5769}
Brian Curtin52173d42010-12-02 18:29:18 +00005770#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005771
5772#ifdef HAVE_TIMES
Guido van Rossumd48f2521997-12-05 22:19:34 +00005773#if defined(PYCC_VACPP) && defined(PYOS_OS2)
5774static long
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00005775system_uptime(void)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005776{
5777 ULONG value = 0;
5778
5779 Py_BEGIN_ALLOW_THREADS
5780 DosQuerySysInfo(QSV_MS_COUNT, QSV_MS_COUNT, &value, sizeof(value));
5781 Py_END_ALLOW_THREADS
5782
5783 return value;
5784}
5785
5786static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005787posix_times(PyObject *self, PyObject *noargs)
Guido van Rossumd48f2521997-12-05 22:19:34 +00005788{
Guido van Rossumd48f2521997-12-05 22:19:34 +00005789 /* Currently Only Uptime is Provided -- Others Later */
Victor Stinner8c62be82010-05-06 00:08:46 +00005790 return Py_BuildValue("ddddd",
5791 (double)0 /* t.tms_utime / HZ */,
5792 (double)0 /* t.tms_stime / HZ */,
5793 (double)0 /* t.tms_cutime / HZ */,
5794 (double)0 /* t.tms_cstime / HZ */,
5795 (double)system_uptime() / 1000);
Guido van Rossumd48f2521997-12-05 22:19:34 +00005796}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005797#else /* not OS2 */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +00005798#define NEED_TICKS_PER_SECOND
5799static long ticks_per_second = -1;
Barry Warsaw53699e91996-12-10 23:23:01 +00005800static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005801posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum22db57e1992-04-05 14:25:30 +00005802{
Victor Stinner8c62be82010-05-06 00:08:46 +00005803 struct tms t;
5804 clock_t c;
5805 errno = 0;
5806 c = times(&t);
5807 if (c == (clock_t) -1)
5808 return posix_error();
5809 return Py_BuildValue("ddddd",
5810 (double)t.tms_utime / ticks_per_second,
5811 (double)t.tms_stime / ticks_per_second,
5812 (double)t.tms_cutime / ticks_per_second,
5813 (double)t.tms_cstime / ticks_per_second,
5814 (double)c / ticks_per_second);
Guido van Rossum22db57e1992-04-05 14:25:30 +00005815}
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005816#endif /* not OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +00005817#endif /* HAVE_TIMES */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005818
5819
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005820#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00005821#define HAVE_TIMES /* so the method table will pick it up */
Barry Warsaw53699e91996-12-10 23:23:01 +00005822static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005823posix_times(PyObject *self, PyObject *noargs)
Guido van Rossum14ed0b21994-09-29 09:50:09 +00005824{
Victor Stinner8c62be82010-05-06 00:08:46 +00005825 FILETIME create, exit, kernel, user;
5826 HANDLE hProc;
5827 hProc = GetCurrentProcess();
5828 GetProcessTimes(hProc, &create, &exit, &kernel, &user);
5829 /* The fields of a FILETIME structure are the hi and lo part
5830 of a 64-bit value expressed in 100 nanosecond units.
5831 1e7 is one second in such units; 1e-7 the inverse.
5832 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
5833 */
5834 return Py_BuildValue(
5835 "ddddd",
5836 (double)(user.dwHighDateTime*429.4967296 +
5837 user.dwLowDateTime*1e-7),
5838 (double)(kernel.dwHighDateTime*429.4967296 +
5839 kernel.dwLowDateTime*1e-7),
5840 (double)0,
5841 (double)0,
5842 (double)0);
Guido van Rossum14ed0b21994-09-29 09:50:09 +00005843}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005844#endif /* MS_WINDOWS */
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005845
5846#ifdef HAVE_TIMES
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005847PyDoc_STRVAR(posix_times__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005848"times() -> (utime, stime, cutime, cstime, elapsed_time)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005849Return a tuple of floating point numbers indicating process times.");
Guido van Rossumbfaf3d61997-12-29 20:02:27 +00005850#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00005851
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005852
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00005853#ifdef HAVE_GETSID
5854PyDoc_STRVAR(posix_getsid__doc__,
5855"getsid(pid) -> sid\n\n\
5856Call the system call getsid().");
5857
5858static PyObject *
5859posix_getsid(PyObject *self, PyObject *args)
5860{
Victor Stinner8c62be82010-05-06 00:08:46 +00005861 pid_t pid;
5862 int sid;
5863 if (!PyArg_ParseTuple(args, _Py_PARSE_PID ":getsid", &pid))
5864 return NULL;
5865 sid = getsid(pid);
5866 if (sid < 0)
5867 return posix_error();
5868 return PyLong_FromLong((long)sid);
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00005869}
5870#endif /* HAVE_GETSID */
5871
5872
Guido van Rossumb6775db1994-08-01 11:34:53 +00005873#ifdef HAVE_SETSID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005874PyDoc_STRVAR(posix_setsid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005875"setsid()\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005876Call the system call setsid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005877
Barry Warsaw53699e91996-12-10 23:23:01 +00005878static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00005879posix_setsid(PyObject *self, PyObject *noargs)
Guido van Rossumc2670a01992-09-13 20:07:29 +00005880{
Victor Stinner8c62be82010-05-06 00:08:46 +00005881 if (setsid() < 0)
5882 return posix_error();
5883 Py_INCREF(Py_None);
5884 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00005885}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005886#endif /* HAVE_SETSID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00005887
Guido van Rossumb6775db1994-08-01 11:34:53 +00005888#ifdef HAVE_SETPGID
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005889PyDoc_STRVAR(posix_setpgid__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005890"setpgid(pid, pgrp)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005891Call the system call setpgid().");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005892
Barry Warsaw53699e91996-12-10 23:23:01 +00005893static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005894posix_setpgid(PyObject *self, PyObject *args)
Guido van Rossumc2670a01992-09-13 20:07:29 +00005895{
Victor Stinner8c62be82010-05-06 00:08:46 +00005896 pid_t pid;
5897 int pgrp;
5898 if (!PyArg_ParseTuple(args, _Py_PARSE_PID "i:setpgid", &pid, &pgrp))
5899 return NULL;
5900 if (setpgid(pid, pgrp) < 0)
5901 return posix_error();
5902 Py_INCREF(Py_None);
5903 return Py_None;
Guido van Rossumc2670a01992-09-13 20:07:29 +00005904}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005905#endif /* HAVE_SETPGID */
Guido van Rossumc2670a01992-09-13 20:07:29 +00005906
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005907
Guido van Rossumb6775db1994-08-01 11:34:53 +00005908#ifdef HAVE_TCGETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005909PyDoc_STRVAR(posix_tcgetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005910"tcgetpgrp(fd) -> pgid\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005911Return the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005912
Barry Warsaw53699e91996-12-10 23:23:01 +00005913static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005914posix_tcgetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00005915{
Victor Stinner8c62be82010-05-06 00:08:46 +00005916 int fd;
5917 pid_t pgid;
5918 if (!PyArg_ParseTuple(args, "i:tcgetpgrp", &fd))
5919 return NULL;
5920 pgid = tcgetpgrp(fd);
5921 if (pgid < 0)
5922 return posix_error();
5923 return PyLong_FromPid(pgid);
Guido van Rossum7066dd71992-09-17 17:54:56 +00005924}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005925#endif /* HAVE_TCGETPGRP */
Guido van Rossum7066dd71992-09-17 17:54:56 +00005926
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005927
Guido van Rossumb6775db1994-08-01 11:34:53 +00005928#ifdef HAVE_TCSETPGRP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005929PyDoc_STRVAR(posix_tcsetpgrp__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005930"tcsetpgrp(fd, pgid)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005931Set the process group associated with the terminal given by a fd.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005932
Barry Warsaw53699e91996-12-10 23:23:01 +00005933static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005934posix_tcsetpgrp(PyObject *self, PyObject *args)
Guido van Rossum7066dd71992-09-17 17:54:56 +00005935{
Victor Stinner8c62be82010-05-06 00:08:46 +00005936 int fd;
5937 pid_t pgid;
5938 if (!PyArg_ParseTuple(args, "i" _Py_PARSE_PID ":tcsetpgrp", &fd, &pgid))
5939 return NULL;
5940 if (tcsetpgrp(fd, pgid) < 0)
5941 return posix_error();
5942 Py_INCREF(Py_None);
5943 return Py_None;
Guido van Rossum7066dd71992-09-17 17:54:56 +00005944}
Guido van Rossumb6775db1994-08-01 11:34:53 +00005945#endif /* HAVE_TCSETPGRP */
Guido van Rossum22db57e1992-04-05 14:25:30 +00005946
Guido van Rossum687dd131993-05-17 08:34:16 +00005947/* Functions acting on file descriptors */
5948
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005949PyDoc_STRVAR(posix_open__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005950"open(filename, flag [, mode=0777]) -> fd\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005951Open a file (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005952
Barry Warsaw53699e91996-12-10 23:23:01 +00005953static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005954posix_open(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00005955{
Victor Stinner8c62be82010-05-06 00:08:46 +00005956 PyObject *ofile;
5957 char *file;
5958 int flag;
5959 int mode = 0777;
5960 int fd;
Mark Hammondc2e85bd2002-10-03 05:10:39 +00005961
5962#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00005963 PyUnicodeObject *po;
5964 if (PyArg_ParseTuple(args, "Ui|i:mkdir", &po, &flag, &mode)) {
5965 Py_BEGIN_ALLOW_THREADS
5966 /* PyUnicode_AS_UNICODE OK without thread
5967 lock as it is a simple dereference. */
5968 fd = _wopen(PyUnicode_AS_UNICODE(po), flag, mode);
5969 Py_END_ALLOW_THREADS
5970 if (fd < 0)
5971 return posix_error();
5972 return PyLong_FromLong((long)fd);
5973 }
5974 /* Drop the argument parsing error as narrow strings
5975 are also valid. */
5976 PyErr_Clear();
Mark Hammondc2e85bd2002-10-03 05:10:39 +00005977#endif
5978
Victor Stinner8c62be82010-05-06 00:08:46 +00005979 if (!PyArg_ParseTuple(args, "O&i|i",
5980 PyUnicode_FSConverter, &ofile,
5981 &flag, &mode))
5982 return NULL;
5983 file = PyBytes_AsString(ofile);
5984 Py_BEGIN_ALLOW_THREADS
5985 fd = open(file, flag, mode);
5986 Py_END_ALLOW_THREADS
5987 if (fd < 0)
5988 return posix_error_with_allocated_filename(ofile);
5989 Py_DECREF(ofile);
5990 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00005991}
5992
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005993
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005994PyDoc_STRVAR(posix_close__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00005995"close(fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00005996Close a file descriptor (for low level IO).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00005997
Barry Warsaw53699e91996-12-10 23:23:01 +00005998static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00005999posix_close(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006000{
Victor Stinner8c62be82010-05-06 00:08:46 +00006001 int fd, res;
6002 if (!PyArg_ParseTuple(args, "i:close", &fd))
6003 return NULL;
6004 if (!_PyVerify_fd(fd))
6005 return posix_error();
6006 Py_BEGIN_ALLOW_THREADS
6007 res = close(fd);
6008 Py_END_ALLOW_THREADS
6009 if (res < 0)
6010 return posix_error();
6011 Py_INCREF(Py_None);
6012 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006013}
6014
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006015
Victor Stinner8c62be82010-05-06 00:08:46 +00006016PyDoc_STRVAR(posix_closerange__doc__,
Christian Heimesfdab48e2008-01-20 09:06:41 +00006017"closerange(fd_low, fd_high)\n\n\
6018Closes all file descriptors in [fd_low, fd_high), ignoring errors.");
6019
6020static PyObject *
6021posix_closerange(PyObject *self, PyObject *args)
6022{
Victor Stinner8c62be82010-05-06 00:08:46 +00006023 int fd_from, fd_to, i;
6024 if (!PyArg_ParseTuple(args, "ii:closerange", &fd_from, &fd_to))
6025 return NULL;
6026 Py_BEGIN_ALLOW_THREADS
6027 for (i = fd_from; i < fd_to; i++)
6028 if (_PyVerify_fd(i))
6029 close(i);
6030 Py_END_ALLOW_THREADS
6031 Py_RETURN_NONE;
Christian Heimesfdab48e2008-01-20 09:06:41 +00006032}
6033
6034
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006035PyDoc_STRVAR(posix_dup__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006036"dup(fd) -> fd2\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006037Return a duplicate of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006038
Barry Warsaw53699e91996-12-10 23:23:01 +00006039static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006040posix_dup(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006041{
Victor Stinner8c62be82010-05-06 00:08:46 +00006042 int fd;
6043 if (!PyArg_ParseTuple(args, "i:dup", &fd))
6044 return NULL;
6045 if (!_PyVerify_fd(fd))
6046 return posix_error();
Victor Stinner8c62be82010-05-06 00:08:46 +00006047 fd = dup(fd);
Victor Stinner8c62be82010-05-06 00:08:46 +00006048 if (fd < 0)
6049 return posix_error();
6050 return PyLong_FromLong((long)fd);
Guido van Rossum687dd131993-05-17 08:34:16 +00006051}
6052
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006053
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006054PyDoc_STRVAR(posix_dup2__doc__,
Andrew M. Kuchling8135fd52004-01-16 13:18:42 +00006055"dup2(old_fd, new_fd)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006056Duplicate file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006057
Barry Warsaw53699e91996-12-10 23:23:01 +00006058static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006059posix_dup2(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006060{
Victor Stinner8c62be82010-05-06 00:08:46 +00006061 int fd, fd2, res;
6062 if (!PyArg_ParseTuple(args, "ii:dup2", &fd, &fd2))
6063 return NULL;
6064 if (!_PyVerify_fd_dup2(fd, fd2))
6065 return posix_error();
Victor Stinner8c62be82010-05-06 00:08:46 +00006066 res = dup2(fd, fd2);
Victor Stinner8c62be82010-05-06 00:08:46 +00006067 if (res < 0)
6068 return posix_error();
6069 Py_INCREF(Py_None);
6070 return Py_None;
Guido van Rossum687dd131993-05-17 08:34:16 +00006071}
6072
Ross Lagerwall7807c352011-03-17 20:20:30 +02006073#ifdef HAVE_LOCKF
6074PyDoc_STRVAR(posix_lockf__doc__,
6075"lockf(fd, cmd, len)\n\n\
6076Apply, test or remove a POSIX lock on an open file descriptor.\n\n\
6077fd is an open file descriptor.\n\
6078cmd specifies the command to use - one of F_LOCK, F_TLOCK, F_ULOCK or\n\
6079F_TEST.\n\
6080len specifies the section of the file to lock.");
6081
6082static PyObject *
6083posix_lockf(PyObject *self, PyObject *args)
6084{
6085 int fd, cmd, res;
6086 off_t len;
6087 if (!PyArg_ParseTuple(args, "iiO&:lockf",
6088 &fd, &cmd, _parse_off_t, &len))
6089 return NULL;
6090
6091 Py_BEGIN_ALLOW_THREADS
6092 res = lockf(fd, cmd, len);
6093 Py_END_ALLOW_THREADS
6094
6095 if (res < 0)
6096 return posix_error();
6097
6098 Py_RETURN_NONE;
6099}
6100#endif
6101
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006102
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006103PyDoc_STRVAR(posix_lseek__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006104"lseek(fd, pos, how) -> newpos\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006105Set the current position of a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006106
Barry Warsaw53699e91996-12-10 23:23:01 +00006107static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006108posix_lseek(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006109{
Victor Stinner8c62be82010-05-06 00:08:46 +00006110 int fd, how;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006111#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00006112 PY_LONG_LONG pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006113#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006114 off_t pos, res;
Fred Drake699f3522000-06-29 21:12:41 +00006115#endif
Ross Lagerwall8e749672011-03-17 21:54:07 +02006116 PyObject *posobj;
6117 if (!PyArg_ParseTuple(args, "iOi:lseek", &fd, &posobj, &how))
Victor Stinner8c62be82010-05-06 00:08:46 +00006118 return NULL;
Guido van Rossum687dd131993-05-17 08:34:16 +00006119#ifdef SEEK_SET
Victor Stinner8c62be82010-05-06 00:08:46 +00006120 /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
6121 switch (how) {
6122 case 0: how = SEEK_SET; break;
6123 case 1: how = SEEK_CUR; break;
6124 case 2: how = SEEK_END; break;
6125 }
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00006126#endif /* SEEK_END */
Guido van Rossum94f6f721999-01-06 18:42:14 +00006127
Ross Lagerwall8e749672011-03-17 21:54:07 +02006128#if !defined(HAVE_LARGEFILE_SUPPORT)
6129 pos = PyLong_AsLong(posobj);
6130#else
6131 pos = PyLong_AsLongLong(posobj);
6132#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006133 if (PyErr_Occurred())
6134 return NULL;
Guido van Rossum94f6f721999-01-06 18:42:14 +00006135
Victor Stinner8c62be82010-05-06 00:08:46 +00006136 if (!_PyVerify_fd(fd))
6137 return posix_error();
6138 Py_BEGIN_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006139#if defined(MS_WIN64) || defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00006140 res = _lseeki64(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00006141#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006142 res = lseek(fd, pos, how);
Fred Drake699f3522000-06-29 21:12:41 +00006143#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006144 Py_END_ALLOW_THREADS
6145 if (res < 0)
6146 return posix_error();
Guido van Rossum94f6f721999-01-06 18:42:14 +00006147
6148#if !defined(HAVE_LARGEFILE_SUPPORT)
Victor Stinner8c62be82010-05-06 00:08:46 +00006149 return PyLong_FromLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006150#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006151 return PyLong_FromLongLong(res);
Guido van Rossum94f6f721999-01-06 18:42:14 +00006152#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00006153}
6154
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006155
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006156PyDoc_STRVAR(posix_read__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006157"read(fd, buffersize) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006158Read a file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006159
Barry Warsaw53699e91996-12-10 23:23:01 +00006160static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006161posix_read(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006162{
Victor Stinner8c62be82010-05-06 00:08:46 +00006163 int fd, size;
6164 Py_ssize_t n;
6165 PyObject *buffer;
6166 if (!PyArg_ParseTuple(args, "ii:read", &fd, &size))
6167 return NULL;
6168 if (size < 0) {
6169 errno = EINVAL;
6170 return posix_error();
6171 }
6172 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
6173 if (buffer == NULL)
6174 return NULL;
Stefan Krah99439262010-11-26 12:58:05 +00006175 if (!_PyVerify_fd(fd)) {
6176 Py_DECREF(buffer);
Victor Stinner8c62be82010-05-06 00:08:46 +00006177 return posix_error();
Stefan Krah99439262010-11-26 12:58:05 +00006178 }
Victor Stinner8c62be82010-05-06 00:08:46 +00006179 Py_BEGIN_ALLOW_THREADS
6180 n = read(fd, PyBytes_AS_STRING(buffer), size);
6181 Py_END_ALLOW_THREADS
6182 if (n < 0) {
6183 Py_DECREF(buffer);
6184 return posix_error();
6185 }
6186 if (n != size)
6187 _PyBytes_Resize(&buffer, n);
6188 return buffer;
Guido van Rossum687dd131993-05-17 08:34:16 +00006189}
6190
Ross Lagerwall7807c352011-03-17 20:20:30 +02006191#if (defined(HAVE_SENDFILE) && (defined(__FreeBSD__) || defined(__DragonFly__) \
6192 || defined(__APPLE__))) || defined(HAVE_READV) || defined(HAVE_WRITEV)
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006193static Py_ssize_t
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006194iov_setup(struct iovec **iov, Py_buffer **buf, PyObject *seq, int cnt, int type)
6195{
6196 int i, j;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006197 Py_ssize_t blen, total = 0;
6198
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006199 *iov = PyMem_New(struct iovec, cnt);
6200 if (*iov == NULL) {
6201 PyErr_NoMemory();
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006202 return total;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006203 }
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006204
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006205 *buf = PyMem_New(Py_buffer, cnt);
6206 if (*buf == NULL) {
6207 PyMem_Del(*iov);
6208 PyErr_NoMemory();
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006209 return total;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006210 }
6211
6212 for (i = 0; i < cnt; i++) {
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02006213 PyObject *item = PySequence_GetItem(seq, i);
6214 if (item == NULL)
6215 goto fail;
6216 if (PyObject_GetBuffer(item, &(*buf)[i], type) == -1) {
6217 Py_DECREF(item);
6218 goto fail;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006219 }
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02006220 Py_DECREF(item);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006221 (*iov)[i].iov_base = (*buf)[i].buf;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006222 blen = (*buf)[i].len;
6223 (*iov)[i].iov_len = blen;
6224 total += blen;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006225 }
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006226 return total;
Ross Lagerwall9ad63e02011-03-19 09:11:14 +02006227
6228fail:
6229 PyMem_Del(*iov);
6230 for (j = 0; j < i; j++) {
6231 PyBuffer_Release(&(*buf)[j]);
6232 }
6233 PyMem_Del(*buf);
6234 return 0;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006235}
6236
6237static void
6238iov_cleanup(struct iovec *iov, Py_buffer *buf, int cnt)
6239{
6240 int i;
6241 PyMem_Del(iov);
6242 for (i = 0; i < cnt; i++) {
6243 PyBuffer_Release(&buf[i]);
6244 }
6245 PyMem_Del(buf);
6246}
6247#endif
6248
Ross Lagerwall7807c352011-03-17 20:20:30 +02006249#ifdef HAVE_READV
6250PyDoc_STRVAR(posix_readv__doc__,
6251"readv(fd, buffers) -> bytesread\n\n\
6252Read from a file descriptor into a number of writable buffers. buffers\n\
6253is an arbitrary sequence of writable buffers.\n\
6254Returns the total number of bytes read.");
6255
6256static PyObject *
6257posix_readv(PyObject *self, PyObject *args)
6258{
6259 int fd, cnt;
6260 Py_ssize_t n;
6261 PyObject *seq;
6262 struct iovec *iov;
6263 Py_buffer *buf;
6264
6265 if (!PyArg_ParseTuple(args, "iO:readv", &fd, &seq))
6266 return NULL;
6267 if (!PySequence_Check(seq)) {
6268 PyErr_SetString(PyExc_TypeError,
6269 "readv() arg 2 must be a sequence");
6270 return NULL;
6271 }
6272 cnt = PySequence_Size(seq);
6273
6274 if (!iov_setup(&iov, &buf, seq, cnt, PyBUF_WRITABLE))
6275 return NULL;
6276
6277 Py_BEGIN_ALLOW_THREADS
6278 n = readv(fd, iov, cnt);
6279 Py_END_ALLOW_THREADS
6280
6281 iov_cleanup(iov, buf, cnt);
6282 return PyLong_FromSsize_t(n);
6283}
6284#endif
6285
6286#ifdef HAVE_PREAD
6287PyDoc_STRVAR(posix_pread__doc__,
6288"pread(fd, buffersize, offset) -> string\n\n\
6289Read from a file descriptor, fd, at a position of offset. It will read up\n\
6290to buffersize number of bytes. The file offset remains unchanged.");
6291
6292static PyObject *
6293posix_pread(PyObject *self, PyObject *args)
6294{
6295 int fd, size;
6296 off_t offset;
6297 Py_ssize_t n;
6298 PyObject *buffer;
6299 if (!PyArg_ParseTuple(args, "iiO&:pread", &fd, &size, _parse_off_t, &offset))
6300 return NULL;
6301
6302 if (size < 0) {
6303 errno = EINVAL;
6304 return posix_error();
6305 }
6306 buffer = PyBytes_FromStringAndSize((char *)NULL, size);
6307 if (buffer == NULL)
6308 return NULL;
6309 if (!_PyVerify_fd(fd)) {
6310 Py_DECREF(buffer);
6311 return posix_error();
6312 }
6313 Py_BEGIN_ALLOW_THREADS
6314 n = pread(fd, PyBytes_AS_STRING(buffer), size, offset);
6315 Py_END_ALLOW_THREADS
6316 if (n < 0) {
6317 Py_DECREF(buffer);
6318 return posix_error();
6319 }
6320 if (n != size)
6321 _PyBytes_Resize(&buffer, n);
6322 return buffer;
6323}
6324#endif
6325
6326PyDoc_STRVAR(posix_write__doc__,
6327"write(fd, string) -> byteswritten\n\n\
6328Write a string to a file descriptor.");
6329
6330static PyObject *
6331posix_write(PyObject *self, PyObject *args)
6332{
6333 Py_buffer pbuf;
6334 int fd;
6335 Py_ssize_t size, len;
6336
6337 if (!PyArg_ParseTuple(args, "iy*:write", &fd, &pbuf))
6338 return NULL;
6339 if (!_PyVerify_fd(fd)) {
6340 PyBuffer_Release(&pbuf);
6341 return posix_error();
6342 }
6343 len = pbuf.len;
6344 Py_BEGIN_ALLOW_THREADS
6345#if defined(MS_WIN64) || defined(MS_WINDOWS)
6346 if (len > INT_MAX)
6347 len = INT_MAX;
6348 size = write(fd, pbuf.buf, (int)len);
6349#else
6350 size = write(fd, pbuf.buf, len);
6351#endif
6352 Py_END_ALLOW_THREADS
6353 PyBuffer_Release(&pbuf);
6354 if (size < 0)
6355 return posix_error();
6356 return PyLong_FromSsize_t(size);
6357}
6358
6359#ifdef HAVE_SENDFILE
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006360PyDoc_STRVAR(posix_sendfile__doc__,
6361"sendfile(out, in, offset, nbytes) -> byteswritten\n\
6362sendfile(out, in, offset, nbytes, headers=None, trailers=None, flags=0)\n\
6363 -> byteswritten\n\
6364Copy nbytes bytes from file descriptor in to file descriptor out.");
6365
6366static PyObject *
6367posix_sendfile(PyObject *self, PyObject *args, PyObject *kwdict)
6368{
6369 int in, out;
6370 Py_ssize_t ret;
6371 off_t offset;
6372
6373#if defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__)
6374#ifndef __APPLE__
6375 Py_ssize_t len;
6376#endif
6377 PyObject *headers = NULL, *trailers = NULL;
6378 Py_buffer *hbuf, *tbuf;
6379 off_t sbytes;
6380 struct sf_hdtr sf;
6381 int flags = 0;
6382 sf.headers = NULL;
6383 sf.trailers = NULL;
Benjamin Petersond8a43b42011-02-26 21:35:16 +00006384 static char *keywords[] = {"out", "in",
6385 "offset", "count",
6386 "headers", "trailers", "flags", NULL};
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006387
6388#ifdef __APPLE__
6389 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiO&O&|OOi:sendfile",
Georg Brandla391b112011-02-25 15:23:18 +00006390 keywords, &out, &in, _parse_off_t, &offset, _parse_off_t, &sbytes,
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006391#else
6392 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiO&n|OOi:sendfile",
Georg Brandla391b112011-02-25 15:23:18 +00006393 keywords, &out, &in, _parse_off_t, &offset, &len,
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006394#endif
6395 &headers, &trailers, &flags))
6396 return NULL;
6397 if (headers != NULL) {
6398 if (!PySequence_Check(headers)) {
6399 PyErr_SetString(PyExc_TypeError,
6400 "sendfile() headers must be a sequence or None");
6401 return NULL;
6402 } else {
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006403 Py_ssize_t i = 0; /* Avoid uninitialized warning */
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006404 sf.hdr_cnt = PySequence_Size(headers);
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006405 if (sf.hdr_cnt > 0 &&
6406 !(i = iov_setup(&(sf.headers), &hbuf,
6407 headers, sf.hdr_cnt, PyBUF_SIMPLE)))
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006408 return NULL;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006409#ifdef __APPLE__
6410 sbytes += i;
6411#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006412 }
6413 }
6414 if (trailers != NULL) {
6415 if (!PySequence_Check(trailers)) {
6416 PyErr_SetString(PyExc_TypeError,
6417 "sendfile() trailers must be a sequence or None");
6418 return NULL;
6419 } else {
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006420 Py_ssize_t i = 0; /* Avoid uninitialized warning */
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006421 sf.trl_cnt = PySequence_Size(trailers);
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006422 if (sf.trl_cnt > 0 &&
6423 !(i = iov_setup(&(sf.trailers), &tbuf,
6424 trailers, sf.trl_cnt, PyBUF_SIMPLE)))
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006425 return NULL;
Giampaolo Rodolàacdad9a2011-03-03 16:10:51 +00006426#ifdef __APPLE__
6427 sbytes += i;
6428#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006429 }
6430 }
6431
6432 Py_BEGIN_ALLOW_THREADS
6433#ifdef __APPLE__
6434 ret = sendfile(in, out, offset, &sbytes, &sf, flags);
6435#else
6436 ret = sendfile(in, out, offset, len, &sf, &sbytes, flags);
6437#endif
6438 Py_END_ALLOW_THREADS
6439
6440 if (sf.headers != NULL)
6441 iov_cleanup(sf.headers, hbuf, sf.hdr_cnt);
6442 if (sf.trailers != NULL)
6443 iov_cleanup(sf.trailers, tbuf, sf.trl_cnt);
6444
6445 if (ret < 0) {
6446 if ((errno == EAGAIN) || (errno == EBUSY)) {
6447 if (sbytes != 0) {
6448 // some data has been sent
6449 goto done;
6450 }
6451 else {
6452 // no data has been sent; upper application is supposed
6453 // to retry on EAGAIN or EBUSY
6454 return posix_error();
6455 }
6456 }
6457 return posix_error();
6458 }
6459 goto done;
6460
6461done:
6462 #if !defined(HAVE_LARGEFILE_SUPPORT)
6463 return Py_BuildValue("l", sbytes);
6464 #else
6465 return Py_BuildValue("L", sbytes);
6466 #endif
6467
6468#else
6469 Py_ssize_t count;
6470 PyObject *offobj;
6471 static char *keywords[] = {"out", "in",
6472 "offset", "count", NULL};
6473 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "iiOn:sendfile",
6474 keywords, &out, &in, &offobj, &count))
6475 return NULL;
6476#ifdef linux
6477 if (offobj == Py_None) {
6478 Py_BEGIN_ALLOW_THREADS
6479 ret = sendfile(out, in, NULL, count);
6480 Py_END_ALLOW_THREADS
6481 if (ret < 0)
6482 return posix_error();
Giampaolo Rodola'ff1a7352011-04-19 09:47:16 +02006483 return Py_BuildValue("n", ret);
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006484 }
6485#endif
Antoine Pitroudcc20b82011-02-26 13:38:35 +00006486 if (!_parse_off_t(offobj, &offset))
6487 return NULL;
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00006488 Py_BEGIN_ALLOW_THREADS
6489 ret = sendfile(out, in, &offset, count);
6490 Py_END_ALLOW_THREADS
6491 if (ret < 0)
6492 return posix_error();
6493 return Py_BuildValue("n", ret);
6494#endif
6495}
6496#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006497
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006498PyDoc_STRVAR(posix_fstat__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006499"fstat(fd) -> stat result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006500Like stat(), but for an open file descriptor.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006501
Barry Warsaw53699e91996-12-10 23:23:01 +00006502static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006503posix_fstat(PyObject *self, PyObject *args)
Guido van Rossum687dd131993-05-17 08:34:16 +00006504{
Victor Stinner8c62be82010-05-06 00:08:46 +00006505 int fd;
6506 STRUCT_STAT st;
6507 int res;
6508 if (!PyArg_ParseTuple(args, "i:fstat", &fd))
6509 return NULL;
Martin v. Löwis7a924e62003-03-05 14:15:21 +00006510#ifdef __VMS
Victor Stinner8c62be82010-05-06 00:08:46 +00006511 /* on OpenVMS we must ensure that all bytes are written to the file */
6512 fsync(fd);
Martin v. Löwis7a924e62003-03-05 14:15:21 +00006513#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006514 if (!_PyVerify_fd(fd))
6515 return posix_error();
6516 Py_BEGIN_ALLOW_THREADS
6517 res = FSTAT(fd, &st);
6518 Py_END_ALLOW_THREADS
6519 if (res != 0) {
Martin v. Löwis14694662006-02-03 12:54:16 +00006520#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00006521 return win32_error("fstat", NULL);
Martin v. Löwis14694662006-02-03 12:54:16 +00006522#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006523 return posix_error();
Martin v. Löwis14694662006-02-03 12:54:16 +00006524#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006525 }
Tim Peters5aa91602002-01-30 05:46:57 +00006526
Victor Stinner8c62be82010-05-06 00:08:46 +00006527 return _pystat_fromstructstat(&st);
Guido van Rossum687dd131993-05-17 08:34:16 +00006528}
6529
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006530PyDoc_STRVAR(posix_isatty__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006531"isatty(fd) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00006532Return True if the file descriptor 'fd' is an open file descriptor\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006533connected to the slave end of a terminal.");
Skip Montanaro1517d842000-07-19 14:34:14 +00006534
6535static PyObject *
Thomas Wouters616607a2000-07-19 14:45:40 +00006536posix_isatty(PyObject *self, PyObject *args)
Skip Montanaro1517d842000-07-19 14:34:14 +00006537{
Victor Stinner8c62be82010-05-06 00:08:46 +00006538 int fd;
6539 if (!PyArg_ParseTuple(args, "i:isatty", &fd))
6540 return NULL;
6541 if (!_PyVerify_fd(fd))
6542 return PyBool_FromLong(0);
6543 return PyBool_FromLong(isatty(fd));
Skip Montanaro1517d842000-07-19 14:34:14 +00006544}
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006545
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006546#ifdef HAVE_PIPE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006547PyDoc_STRVAR(posix_pipe__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006548"pipe() -> (read_end, write_end)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006549Create a pipe.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006550
Barry Warsaw53699e91996-12-10 23:23:01 +00006551static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00006552posix_pipe(PyObject *self, PyObject *noargs)
Guido van Rossum687dd131993-05-17 08:34:16 +00006553{
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00006554#if defined(PYOS_OS2)
6555 HFILE read, write;
6556 APIRET rc;
6557
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00006558 rc = DosCreatePipe( &read, &write, 4096);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00006559 if (rc != NO_ERROR)
Guido van Rossumd48f2521997-12-05 22:19:34 +00006560 return os2_error(rc);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00006561
6562 return Py_BuildValue("(ii)", read, write);
6563#else
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006564#if !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00006565 int fds[2];
6566 int res;
Victor Stinner8c62be82010-05-06 00:08:46 +00006567 res = pipe(fds);
Victor Stinner8c62be82010-05-06 00:08:46 +00006568 if (res != 0)
6569 return posix_error();
6570 return Py_BuildValue("(ii)", fds[0], fds[1]);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006571#else /* MS_WINDOWS */
Victor Stinner8c62be82010-05-06 00:08:46 +00006572 HANDLE read, write;
6573 int read_fd, write_fd;
6574 BOOL ok;
Victor Stinner8c62be82010-05-06 00:08:46 +00006575 ok = CreatePipe(&read, &write, NULL, 0);
Victor Stinner8c62be82010-05-06 00:08:46 +00006576 if (!ok)
6577 return win32_error("CreatePipe", NULL);
6578 read_fd = _open_osfhandle((Py_intptr_t)read, 0);
6579 write_fd = _open_osfhandle((Py_intptr_t)write, 1);
6580 return Py_BuildValue("(ii)", read_fd, write_fd);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00006581#endif /* MS_WINDOWS */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00006582#endif
Guido van Rossum687dd131993-05-17 08:34:16 +00006583}
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006584#endif /* HAVE_PIPE */
6585
Charles-François Natalidaafdd52011-05-29 20:07:40 +02006586#ifdef HAVE_PIPE2
6587PyDoc_STRVAR(posix_pipe2__doc__,
Charles-François Natali368f34b2011-06-06 19:49:47 +02006588"pipe2(flags) -> (read_end, write_end)\n\n\
6589Create a pipe with flags set atomically.\n\
6590flags can be constructed by ORing together one or more of these values:\n\
6591O_NONBLOCK, O_CLOEXEC.\n\
Charles-François Natalidaafdd52011-05-29 20:07:40 +02006592");
6593
6594static PyObject *
Charles-François Natali368f34b2011-06-06 19:49:47 +02006595posix_pipe2(PyObject *self, PyObject *arg)
Charles-François Natalidaafdd52011-05-29 20:07:40 +02006596{
Charles-François Natali368f34b2011-06-06 19:49:47 +02006597 int flags;
Charles-François Natalidaafdd52011-05-29 20:07:40 +02006598 int fds[2];
6599 int res;
6600
Charles-François Natali368f34b2011-06-06 19:49:47 +02006601 flags = PyLong_AsLong(arg);
6602 if (flags == -1 && PyErr_Occurred())
Charles-François Natalidaafdd52011-05-29 20:07:40 +02006603 return NULL;
6604
6605 res = pipe2(fds, flags);
6606 if (res != 0)
6607 return posix_error();
6608 return Py_BuildValue("(ii)", fds[0], fds[1]);
6609}
6610#endif /* HAVE_PIPE2 */
6611
Ross Lagerwall7807c352011-03-17 20:20:30 +02006612#ifdef HAVE_WRITEV
6613PyDoc_STRVAR(posix_writev__doc__,
6614"writev(fd, buffers) -> byteswritten\n\n\
6615Write the contents of buffers to a file descriptor, where buffers is an\n\
6616arbitrary sequence of buffers.\n\
6617Returns the total bytes written.");
6618
6619static PyObject *
6620posix_writev(PyObject *self, PyObject *args)
6621{
6622 int fd, cnt;
6623 Py_ssize_t res;
6624 PyObject *seq;
6625 struct iovec *iov;
6626 Py_buffer *buf;
6627 if (!PyArg_ParseTuple(args, "iO:writev", &fd, &seq))
6628 return NULL;
6629 if (!PySequence_Check(seq)) {
6630 PyErr_SetString(PyExc_TypeError,
6631 "writev() arg 2 must be a sequence");
6632 return NULL;
6633 }
6634 cnt = PySequence_Size(seq);
6635
6636 if (!iov_setup(&iov, &buf, seq, cnt, PyBUF_SIMPLE)) {
6637 return NULL;
6638 }
6639
6640 Py_BEGIN_ALLOW_THREADS
6641 res = writev(fd, iov, cnt);
6642 Py_END_ALLOW_THREADS
6643
6644 iov_cleanup(iov, buf, cnt);
6645 return PyLong_FromSsize_t(res);
6646}
6647#endif
6648
6649#ifdef HAVE_PWRITE
6650PyDoc_STRVAR(posix_pwrite__doc__,
6651"pwrite(fd, string, offset) -> byteswritten\n\n\
6652Write string to a file descriptor, fd, from offset, leaving the file\n\
6653offset unchanged.");
6654
6655static PyObject *
6656posix_pwrite(PyObject *self, PyObject *args)
6657{
6658 Py_buffer pbuf;
6659 int fd;
6660 off_t offset;
6661 Py_ssize_t size;
6662
6663 if (!PyArg_ParseTuple(args, "iy*O&:pwrite", &fd, &pbuf, _parse_off_t, &offset))
6664 return NULL;
6665
6666 if (!_PyVerify_fd(fd)) {
6667 PyBuffer_Release(&pbuf);
6668 return posix_error();
6669 }
6670 Py_BEGIN_ALLOW_THREADS
6671 size = pwrite(fd, pbuf.buf, (size_t)pbuf.len, offset);
6672 Py_END_ALLOW_THREADS
6673 PyBuffer_Release(&pbuf);
6674 if (size < 0)
6675 return posix_error();
6676 return PyLong_FromSsize_t(size);
6677}
6678#endif
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006679
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006680#ifdef HAVE_MKFIFO
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006681PyDoc_STRVAR(posix_mkfifo__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00006682"mkfifo(filename [, mode=0666])\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006683Create a FIFO (a POSIX named pipe).");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006684
Barry Warsaw53699e91996-12-10 23:23:01 +00006685static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006686posix_mkfifo(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006687{
Benjamin Petersond4efbf92010-08-11 19:20:42 +00006688 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00006689 char *filename;
6690 int mode = 0666;
6691 int res;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00006692 if (!PyArg_ParseTuple(args, "O&|i:mkfifo", PyUnicode_FSConverter, &opath,
6693 &mode))
Victor Stinner8c62be82010-05-06 00:08:46 +00006694 return NULL;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00006695 filename = PyBytes_AS_STRING(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00006696 Py_BEGIN_ALLOW_THREADS
6697 res = mkfifo(filename, mode);
6698 Py_END_ALLOW_THREADS
Benjamin Petersond4efbf92010-08-11 19:20:42 +00006699 Py_DECREF(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00006700 if (res < 0)
6701 return posix_error();
6702 Py_INCREF(Py_None);
6703 return Py_None;
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006704}
6705#endif
6706
6707
Neal Norwitz11690112002-07-30 01:08:28 +00006708#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006709PyDoc_STRVAR(posix_mknod__doc__,
Neal Norwitzc18b3082002-10-11 22:19:42 +00006710"mknod(filename [, mode=0600, device])\n\n\
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006711Create a filesystem node (file, device special file or named pipe)\n\
6712named filename. mode specifies both the permissions to use and the\n\
6713type of node to be created, being combined (bitwise OR) with one of\n\
6714S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. For S_IFCHR and S_IFBLK,\n\
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006715device defines the newly created device special file (probably using\n\
6716os.makedev()), otherwise it is ignored.");
Martin v. Löwis06a83e92002-04-14 10:19:44 +00006717
6718
6719static PyObject *
6720posix_mknod(PyObject *self, PyObject *args)
6721{
Benjamin Petersond4efbf92010-08-11 19:20:42 +00006722 PyObject *opath;
Victor Stinner8c62be82010-05-06 00:08:46 +00006723 char *filename;
6724 int mode = 0600;
6725 int device = 0;
6726 int res;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00006727 if (!PyArg_ParseTuple(args, "O&|ii:mknod", PyUnicode_FSConverter, &opath,
6728 &mode, &device))
Victor Stinner8c62be82010-05-06 00:08:46 +00006729 return NULL;
Benjamin Petersond4efbf92010-08-11 19:20:42 +00006730 filename = PyBytes_AS_STRING(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00006731 Py_BEGIN_ALLOW_THREADS
6732 res = mknod(filename, mode, device);
6733 Py_END_ALLOW_THREADS
Benjamin Petersond4efbf92010-08-11 19:20:42 +00006734 Py_DECREF(opath);
Victor Stinner8c62be82010-05-06 00:08:46 +00006735 if (res < 0)
6736 return posix_error();
6737 Py_INCREF(Py_None);
6738 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006739}
6740#endif
6741
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006742#ifdef HAVE_DEVICE_MACROS
6743PyDoc_STRVAR(posix_major__doc__,
6744"major(device) -> major number\n\
6745Extracts a device major number from a raw device number.");
6746
6747static PyObject *
6748posix_major(PyObject *self, PyObject *args)
6749{
Victor Stinner8c62be82010-05-06 00:08:46 +00006750 int device;
6751 if (!PyArg_ParseTuple(args, "i:major", &device))
6752 return NULL;
6753 return PyLong_FromLong((long)major(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006754}
6755
6756PyDoc_STRVAR(posix_minor__doc__,
6757"minor(device) -> minor number\n\
6758Extracts a device minor number from a raw device number.");
6759
6760static PyObject *
6761posix_minor(PyObject *self, PyObject *args)
6762{
Victor Stinner8c62be82010-05-06 00:08:46 +00006763 int device;
6764 if (!PyArg_ParseTuple(args, "i:minor", &device))
6765 return NULL;
6766 return PyLong_FromLong((long)minor(device));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006767}
6768
6769PyDoc_STRVAR(posix_makedev__doc__,
6770"makedev(major, minor) -> device number\n\
6771Composes a raw device number from the major and minor device numbers.");
6772
6773static PyObject *
6774posix_makedev(PyObject *self, PyObject *args)
6775{
Victor Stinner8c62be82010-05-06 00:08:46 +00006776 int major, minor;
6777 if (!PyArg_ParseTuple(args, "ii:makedev", &major, &minor))
6778 return NULL;
6779 return PyLong_FromLong((long)makedev(major, minor));
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00006780}
6781#endif /* device macros */
6782
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006783
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006784#ifdef HAVE_FTRUNCATE
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006785PyDoc_STRVAR(posix_ftruncate__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006786"ftruncate(fd, length)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006787Truncate a file to a specified length.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006788
Barry Warsaw53699e91996-12-10 23:23:01 +00006789static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006790posix_ftruncate(PyObject *self, PyObject *args)
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006791{
Victor Stinner8c62be82010-05-06 00:08:46 +00006792 int fd;
6793 off_t length;
6794 int res;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006795
Ross Lagerwall7807c352011-03-17 20:20:30 +02006796 if (!PyArg_ParseTuple(args, "iO&:ftruncate", &fd, _parse_off_t, &length))
Victor Stinner8c62be82010-05-06 00:08:46 +00006797 return NULL;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006798
Victor Stinner8c62be82010-05-06 00:08:46 +00006799 Py_BEGIN_ALLOW_THREADS
6800 res = ftruncate(fd, length);
6801 Py_END_ALLOW_THREADS
6802 if (res < 0)
6803 return posix_error();
6804 Py_INCREF(Py_None);
6805 return Py_None;
Guido van Rossuma4916fa1996-05-23 22:58:55 +00006806}
6807#endif
Guido van Rossum22db57e1992-04-05 14:25:30 +00006808
Ross Lagerwall7807c352011-03-17 20:20:30 +02006809#ifdef HAVE_TRUNCATE
6810PyDoc_STRVAR(posix_truncate__doc__,
6811"truncate(path, length)\n\n\
6812Truncate the file given by path to length bytes.");
6813
6814static PyObject *
6815posix_truncate(PyObject *self, PyObject *args)
6816{
6817 PyObject *opath;
6818 const char *path;
6819 off_t length;
6820 int res;
6821
6822 if (!PyArg_ParseTuple(args, "O&O&:truncate",
6823 PyUnicode_FSConverter, &opath, _parse_off_t, &length))
6824 return NULL;
6825 path = PyBytes_AsString(opath);
6826
6827 Py_BEGIN_ALLOW_THREADS
6828 res = truncate(path, length);
6829 Py_END_ALLOW_THREADS
6830 Py_DECREF(opath);
6831 if (res < 0)
6832 return posix_error();
6833 Py_RETURN_NONE;
6834}
6835#endif
6836
6837#ifdef HAVE_POSIX_FALLOCATE
6838PyDoc_STRVAR(posix_posix_fallocate__doc__,
6839"posix_fallocate(fd, offset, len)\n\n\
6840Ensures that enough disk space is allocated for the file specified by fd\n\
6841starting from offset and continuing for len bytes.");
6842
6843static PyObject *
6844posix_posix_fallocate(PyObject *self, PyObject *args)
6845{
6846 off_t len, offset;
6847 int res, fd;
6848
6849 if (!PyArg_ParseTuple(args, "iO&O&:posix_fallocate",
6850 &fd, _parse_off_t, &offset, _parse_off_t, &len))
6851 return NULL;
6852
6853 Py_BEGIN_ALLOW_THREADS
6854 res = posix_fallocate(fd, offset, len);
6855 Py_END_ALLOW_THREADS
6856 if (res != 0) {
6857 errno = res;
6858 return posix_error();
6859 }
6860 Py_RETURN_NONE;
6861}
6862#endif
6863
6864#ifdef HAVE_POSIX_FADVISE
6865PyDoc_STRVAR(posix_posix_fadvise__doc__,
6866"posix_fadvise(fd, offset, len, advice)\n\n\
6867Announces an intention to access data in a specific pattern thus allowing\n\
6868the kernel to make optimizations.\n\
6869The advice applies to the region of the file specified by fd starting at\n\
6870offset and continuing for len bytes.\n\
6871advice is one of POSIX_FADV_NORMAL, POSIX_FADV_SEQUENTIAL,\n\
6872POSIX_FADV_RANDOM, POSIX_FADV_NOREUSE, POSIX_FADV_WILLNEED or\n\
6873POSIX_FADV_DONTNEED.");
6874
6875static PyObject *
6876posix_posix_fadvise(PyObject *self, PyObject *args)
6877{
6878 off_t len, offset;
6879 int res, fd, advice;
6880
6881 if (!PyArg_ParseTuple(args, "iO&O&i:posix_fadvise",
6882 &fd, _parse_off_t, &offset, _parse_off_t, &len, &advice))
6883 return NULL;
6884
6885 Py_BEGIN_ALLOW_THREADS
6886 res = posix_fadvise(fd, offset, len, advice);
6887 Py_END_ALLOW_THREADS
6888 if (res != 0) {
6889 errno = res;
6890 return posix_error();
6891 }
6892 Py_RETURN_NONE;
6893}
6894#endif
6895
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006896#ifdef HAVE_PUTENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006897PyDoc_STRVAR(posix_putenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00006898"putenv(key, value)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00006899Change or add an environment variable.");
Guido van Rossumec4f4ac1997-06-02 22:20:51 +00006900
Fred Drake762e2061999-08-26 17:23:54 +00006901/* Save putenv() parameters as values here, so we can collect them when they
6902 * get re-set with another call for the same key. */
6903static PyObject *posix_putenv_garbage;
6904
Tim Peters5aa91602002-01-30 05:46:57 +00006905static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00006906posix_putenv(PyObject *self, PyObject *args)
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006907{
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006908#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00006909 wchar_t *s1, *s2;
6910 wchar_t *newenv;
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006911#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006912 PyObject *os1, *os2;
6913 char *s1, *s2;
6914 char *newenv;
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006915#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00006916 PyObject *newstr = NULL;
Victor Stinner8c62be82010-05-06 00:08:46 +00006917 size_t len;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00006918
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006919#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00006920 if (!PyArg_ParseTuple(args,
6921 "uu:putenv",
6922 &s1, &s2))
6923 return NULL;
Martin v. Löwis011e8422009-05-05 04:43:17 +00006924#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006925 if (!PyArg_ParseTuple(args,
6926 "O&O&:putenv",
6927 PyUnicode_FSConverter, &os1,
6928 PyUnicode_FSConverter, &os2))
6929 return NULL;
6930 s1 = PyBytes_AsString(os1);
6931 s2 = PyBytes_AsString(os2);
Martin v. Löwis011e8422009-05-05 04:43:17 +00006932#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00006933
6934#if defined(PYOS_OS2)
6935 if (stricmp(s1, "BEGINLIBPATH") == 0) {
6936 APIRET rc;
6937
Guido van Rossumd48f2521997-12-05 22:19:34 +00006938 rc = DosSetExtLIBPATH(s2, BEGIN_LIBPATH);
Victor Stinner84ae1182010-05-06 22:05:07 +00006939 if (rc != NO_ERROR) {
6940 os2_error(rc);
6941 goto error;
6942 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00006943
6944 } else if (stricmp(s1, "ENDLIBPATH") == 0) {
6945 APIRET rc;
6946
Guido van Rossumd48f2521997-12-05 22:19:34 +00006947 rc = DosSetExtLIBPATH(s2, END_LIBPATH);
Victor Stinner84ae1182010-05-06 22:05:07 +00006948 if (rc != NO_ERROR) {
6949 os2_error(rc);
6950 goto error;
6951 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00006952 } else {
6953#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00006954 /* XXX This can leak memory -- not easy to fix :-( */
6955 /* len includes space for a trailing \0; the size arg to
6956 PyBytes_FromStringAndSize does not count that */
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006957#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00006958 len = wcslen(s1) + wcslen(s2) + 2;
6959 newstr = PyUnicode_FromUnicode(NULL, (int)len - 1);
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006960#else
Victor Stinner84ae1182010-05-06 22:05:07 +00006961 len = PyBytes_GET_SIZE(os1) + PyBytes_GET_SIZE(os2) + 2;
Victor Stinner8c62be82010-05-06 00:08:46 +00006962 newstr = PyBytes_FromStringAndSize(NULL, (int)len - 1);
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006963#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00006964 if (newstr == NULL) {
6965 PyErr_NoMemory();
6966 goto error;
6967 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006968#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00006969 newenv = PyUnicode_AsUnicode(newstr);
6970 _snwprintf(newenv, len, L"%s=%s", s1, s2);
6971 if (_wputenv(newenv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00006972 posix_error();
Victor Stinner84ae1182010-05-06 22:05:07 +00006973 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00006974 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006975#else
Victor Stinner8c62be82010-05-06 00:08:46 +00006976 newenv = PyBytes_AS_STRING(newstr);
6977 PyOS_snprintf(newenv, len, "%s=%s", s1, s2);
6978 if (putenv(newenv)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00006979 posix_error();
Victor Stinner84ae1182010-05-06 22:05:07 +00006980 goto error;
Victor Stinner8c62be82010-05-06 00:08:46 +00006981 }
Thomas Hellerf78f12a2007-11-08 19:33:05 +00006982#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00006983
Victor Stinner8c62be82010-05-06 00:08:46 +00006984 /* Install the first arg and newstr in posix_putenv_garbage;
6985 * this will cause previous value to be collected. This has to
6986 * happen after the real putenv() call because the old value
6987 * was still accessible until then. */
6988 if (PyDict_SetItem(posix_putenv_garbage,
Victor Stinner84ae1182010-05-06 22:05:07 +00006989#ifdef MS_WINDOWS
6990 PyTuple_GET_ITEM(args, 0),
6991#else
6992 os1,
6993#endif
6994 newstr)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00006995 /* really not much we can do; just leak */
6996 PyErr_Clear();
6997 }
6998 else {
6999 Py_DECREF(newstr);
7000 }
Guido van Rossumd48f2521997-12-05 22:19:34 +00007001
7002#if defined(PYOS_OS2)
7003 }
7004#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007005
Martin v. Löwis011e8422009-05-05 04:43:17 +00007006#ifndef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00007007 Py_DECREF(os1);
7008 Py_DECREF(os2);
Martin v. Löwis011e8422009-05-05 04:43:17 +00007009#endif
Victor Stinner84ae1182010-05-06 22:05:07 +00007010 Py_RETURN_NONE;
7011
7012error:
7013#ifndef MS_WINDOWS
7014 Py_DECREF(os1);
7015 Py_DECREF(os2);
7016#endif
7017 Py_XDECREF(newstr);
7018 return NULL;
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007019}
Guido van Rossumb6a47161997-09-15 22:54:34 +00007020#endif /* putenv */
7021
Guido van Rossumc524d952001-10-19 01:31:59 +00007022#ifdef HAVE_UNSETENV
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007023PyDoc_STRVAR(posix_unsetenv__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007024"unsetenv(key)\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007025Delete an environment variable.");
Guido van Rossumc524d952001-10-19 01:31:59 +00007026
7027static PyObject *
7028posix_unsetenv(PyObject *self, PyObject *args)
7029{
Victor Stinner84ae1182010-05-06 22:05:07 +00007030#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00007031 char *s1;
Guido van Rossumc524d952001-10-19 01:31:59 +00007032
Victor Stinner8c62be82010-05-06 00:08:46 +00007033 if (!PyArg_ParseTuple(args, "s:unsetenv", &s1))
7034 return NULL;
Victor Stinner84ae1182010-05-06 22:05:07 +00007035#else
7036 PyObject *os1;
7037 char *s1;
7038
7039 if (!PyArg_ParseTuple(args, "O&:unsetenv",
7040 PyUnicode_FSConverter, &os1))
7041 return NULL;
7042 s1 = PyBytes_AsString(os1);
7043#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00007044
Victor Stinner8c62be82010-05-06 00:08:46 +00007045 unsetenv(s1);
Guido van Rossumc524d952001-10-19 01:31:59 +00007046
Victor Stinner8c62be82010-05-06 00:08:46 +00007047 /* Remove the key from posix_putenv_garbage;
7048 * this will cause it to be collected. This has to
7049 * happen after the real unsetenv() call because the
7050 * old value was still accessible until then.
7051 */
7052 if (PyDict_DelItem(posix_putenv_garbage,
Victor Stinner84ae1182010-05-06 22:05:07 +00007053#ifdef MS_WINDOWS
7054 PyTuple_GET_ITEM(args, 0)
7055#else
7056 os1
7057#endif
7058 )) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007059 /* really not much we can do; just leak */
7060 PyErr_Clear();
7061 }
Guido van Rossumc524d952001-10-19 01:31:59 +00007062
Victor Stinner84ae1182010-05-06 22:05:07 +00007063#ifndef MS_WINDOWS
7064 Py_DECREF(os1);
7065#endif
7066 Py_RETURN_NONE;
Guido van Rossumc524d952001-10-19 01:31:59 +00007067}
7068#endif /* unsetenv */
7069
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007070PyDoc_STRVAR(posix_strerror__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007071"strerror(code) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007072Translate an error code to a message string.");
Guido van Rossumb6a47161997-09-15 22:54:34 +00007073
Guido van Rossumf68d8e52001-04-14 17:55:09 +00007074static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007075posix_strerror(PyObject *self, PyObject *args)
Guido van Rossumb6a47161997-09-15 22:54:34 +00007076{
Victor Stinner8c62be82010-05-06 00:08:46 +00007077 int code;
7078 char *message;
7079 if (!PyArg_ParseTuple(args, "i:strerror", &code))
7080 return NULL;
7081 message = strerror(code);
7082 if (message == NULL) {
7083 PyErr_SetString(PyExc_ValueError,
7084 "strerror() argument out of range");
7085 return NULL;
7086 }
7087 return PyUnicode_FromString(message);
Guido van Rossumb6a47161997-09-15 22:54:34 +00007088}
Guido van Rossumb6a47161997-09-15 22:54:34 +00007089
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00007090
Guido van Rossumc9641791998-08-04 15:26:23 +00007091#ifdef HAVE_SYS_WAIT_H
7092
Fred Drake106c1a02002-04-23 15:58:02 +00007093#ifdef WCOREDUMP
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007094PyDoc_STRVAR(posix_WCOREDUMP__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007095"WCOREDUMP(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007096Return True if the process returning 'status' was dumped to a core file.");
Fred Drake106c1a02002-04-23 15:58:02 +00007097
7098static PyObject *
7099posix_WCOREDUMP(PyObject *self, PyObject *args)
7100{
Victor Stinner8c62be82010-05-06 00:08:46 +00007101 WAIT_TYPE status;
7102 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00007103
Victor Stinner8c62be82010-05-06 00:08:46 +00007104 if (!PyArg_ParseTuple(args, "i:WCOREDUMP", &WAIT_STATUS_INT(status)))
7105 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00007106
Victor Stinner8c62be82010-05-06 00:08:46 +00007107 return PyBool_FromLong(WCOREDUMP(status));
Fred Drake106c1a02002-04-23 15:58:02 +00007108}
7109#endif /* WCOREDUMP */
7110
7111#ifdef WIFCONTINUED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007112PyDoc_STRVAR(posix_WIFCONTINUED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007113"WIFCONTINUED(status) -> bool\n\n\
Fred Drake106c1a02002-04-23 15:58:02 +00007114Return True if the process returning 'status' was continued from a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007115job control stop.");
Fred Drake106c1a02002-04-23 15:58:02 +00007116
7117static PyObject *
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00007118posix_WIFCONTINUED(PyObject *self, PyObject *args)
Fred Drake106c1a02002-04-23 15:58:02 +00007119{
Victor Stinner8c62be82010-05-06 00:08:46 +00007120 WAIT_TYPE status;
7121 WAIT_STATUS_INT(status) = 0;
Fred Drake106c1a02002-04-23 15:58:02 +00007122
Victor Stinner8c62be82010-05-06 00:08:46 +00007123 if (!PyArg_ParseTuple(args, "i:WCONTINUED", &WAIT_STATUS_INT(status)))
7124 return NULL;
Fred Drake106c1a02002-04-23 15:58:02 +00007125
Victor Stinner8c62be82010-05-06 00:08:46 +00007126 return PyBool_FromLong(WIFCONTINUED(status));
Fred Drake106c1a02002-04-23 15:58:02 +00007127}
7128#endif /* WIFCONTINUED */
7129
Guido van Rossumc9641791998-08-04 15:26:23 +00007130#ifdef WIFSTOPPED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007131PyDoc_STRVAR(posix_WIFSTOPPED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007132"WIFSTOPPED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007133Return True if the process returning 'status' was stopped.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007134
7135static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007136posix_WIFSTOPPED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007137{
Victor Stinner8c62be82010-05-06 00:08:46 +00007138 WAIT_TYPE status;
7139 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007140
Victor Stinner8c62be82010-05-06 00:08:46 +00007141 if (!PyArg_ParseTuple(args, "i:WIFSTOPPED", &WAIT_STATUS_INT(status)))
7142 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007143
Victor Stinner8c62be82010-05-06 00:08:46 +00007144 return PyBool_FromLong(WIFSTOPPED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007145}
7146#endif /* WIFSTOPPED */
7147
7148#ifdef WIFSIGNALED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007149PyDoc_STRVAR(posix_WIFSIGNALED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007150"WIFSIGNALED(status) -> bool\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007151Return True if the process returning 'status' was terminated by a signal.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007152
7153static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007154posix_WIFSIGNALED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007155{
Victor Stinner8c62be82010-05-06 00:08:46 +00007156 WAIT_TYPE status;
7157 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007158
Victor Stinner8c62be82010-05-06 00:08:46 +00007159 if (!PyArg_ParseTuple(args, "i:WIFSIGNALED", &WAIT_STATUS_INT(status)))
7160 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007161
Victor Stinner8c62be82010-05-06 00:08:46 +00007162 return PyBool_FromLong(WIFSIGNALED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007163}
7164#endif /* WIFSIGNALED */
7165
7166#ifdef WIFEXITED
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007167PyDoc_STRVAR(posix_WIFEXITED__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007168"WIFEXITED(status) -> bool\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00007169Return true if the process returning 'status' exited using the exit()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007170system call.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007171
7172static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007173posix_WIFEXITED(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007174{
Victor Stinner8c62be82010-05-06 00:08:46 +00007175 WAIT_TYPE status;
7176 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007177
Victor Stinner8c62be82010-05-06 00:08:46 +00007178 if (!PyArg_ParseTuple(args, "i:WIFEXITED", &WAIT_STATUS_INT(status)))
7179 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007180
Victor Stinner8c62be82010-05-06 00:08:46 +00007181 return PyBool_FromLong(WIFEXITED(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007182}
7183#endif /* WIFEXITED */
7184
Guido van Rossum54ecc3d1999-01-27 17:53:11 +00007185#ifdef WEXITSTATUS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007186PyDoc_STRVAR(posix_WEXITSTATUS__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007187"WEXITSTATUS(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007188Return the process return code from 'status'.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007189
7190static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007191posix_WEXITSTATUS(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007192{
Victor Stinner8c62be82010-05-06 00:08:46 +00007193 WAIT_TYPE status;
7194 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007195
Victor Stinner8c62be82010-05-06 00:08:46 +00007196 if (!PyArg_ParseTuple(args, "i:WEXITSTATUS", &WAIT_STATUS_INT(status)))
7197 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007198
Victor Stinner8c62be82010-05-06 00:08:46 +00007199 return Py_BuildValue("i", WEXITSTATUS(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007200}
7201#endif /* WEXITSTATUS */
7202
7203#ifdef WTERMSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007204PyDoc_STRVAR(posix_WTERMSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007205"WTERMSIG(status) -> integer\n\n\
Fred Drake7e3535c1999-02-02 16:37:11 +00007206Return the signal that terminated the process that provided the 'status'\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007207value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007208
7209static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007210posix_WTERMSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007211{
Victor Stinner8c62be82010-05-06 00:08:46 +00007212 WAIT_TYPE status;
7213 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007214
Victor Stinner8c62be82010-05-06 00:08:46 +00007215 if (!PyArg_ParseTuple(args, "i:WTERMSIG", &WAIT_STATUS_INT(status)))
7216 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007217
Victor Stinner8c62be82010-05-06 00:08:46 +00007218 return Py_BuildValue("i", WTERMSIG(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007219}
7220#endif /* WTERMSIG */
7221
7222#ifdef WSTOPSIG
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007223PyDoc_STRVAR(posix_WSTOPSIG__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007224"WSTOPSIG(status) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007225Return the signal that stopped the process that provided\n\
7226the 'status' value.");
Guido van Rossumc9641791998-08-04 15:26:23 +00007227
7228static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007229posix_WSTOPSIG(PyObject *self, PyObject *args)
Guido van Rossumc9641791998-08-04 15:26:23 +00007230{
Victor Stinner8c62be82010-05-06 00:08:46 +00007231 WAIT_TYPE status;
7232 WAIT_STATUS_INT(status) = 0;
Tim Peters5aa91602002-01-30 05:46:57 +00007233
Victor Stinner8c62be82010-05-06 00:08:46 +00007234 if (!PyArg_ParseTuple(args, "i:WSTOPSIG", &WAIT_STATUS_INT(status)))
7235 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +00007236
Victor Stinner8c62be82010-05-06 00:08:46 +00007237 return Py_BuildValue("i", WSTOPSIG(status));
Guido van Rossumc9641791998-08-04 15:26:23 +00007238}
7239#endif /* WSTOPSIG */
7240
7241#endif /* HAVE_SYS_WAIT_H */
7242
7243
Thomas Wouters477c8d52006-05-27 19:21:47 +00007244#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossumd5753e11999-10-19 13:29:23 +00007245#ifdef _SCO_DS
7246/* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
7247 needed definitions in sys/statvfs.h */
7248#define _SVID3
7249#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00007250#include <sys/statvfs.h>
7251
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007252static PyObject*
7253_pystatvfs_fromstructstatvfs(struct statvfs st) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007254 PyObject *v = PyStructSequence_New(&StatVFSResultType);
7255 if (v == NULL)
7256 return NULL;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007257
7258#if !defined(HAVE_LARGEFILE_SUPPORT)
Victor Stinner8c62be82010-05-06 00:08:46 +00007259 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
7260 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
7261 PyStructSequence_SET_ITEM(v, 2, PyLong_FromLong((long) st.f_blocks));
7262 PyStructSequence_SET_ITEM(v, 3, PyLong_FromLong((long) st.f_bfree));
7263 PyStructSequence_SET_ITEM(v, 4, PyLong_FromLong((long) st.f_bavail));
7264 PyStructSequence_SET_ITEM(v, 5, PyLong_FromLong((long) st.f_files));
7265 PyStructSequence_SET_ITEM(v, 6, PyLong_FromLong((long) st.f_ffree));
7266 PyStructSequence_SET_ITEM(v, 7, PyLong_FromLong((long) st.f_favail));
7267 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
7268 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007269#else
Victor Stinner8c62be82010-05-06 00:08:46 +00007270 PyStructSequence_SET_ITEM(v, 0, PyLong_FromLong((long) st.f_bsize));
7271 PyStructSequence_SET_ITEM(v, 1, PyLong_FromLong((long) st.f_frsize));
7272 PyStructSequence_SET_ITEM(v, 2,
7273 PyLong_FromLongLong((PY_LONG_LONG) st.f_blocks));
7274 PyStructSequence_SET_ITEM(v, 3,
7275 PyLong_FromLongLong((PY_LONG_LONG) st.f_bfree));
7276 PyStructSequence_SET_ITEM(v, 4,
7277 PyLong_FromLongLong((PY_LONG_LONG) st.f_bavail));
7278 PyStructSequence_SET_ITEM(v, 5,
7279 PyLong_FromLongLong((PY_LONG_LONG) st.f_files));
7280 PyStructSequence_SET_ITEM(v, 6,
7281 PyLong_FromLongLong((PY_LONG_LONG) st.f_ffree));
7282 PyStructSequence_SET_ITEM(v, 7,
7283 PyLong_FromLongLong((PY_LONG_LONG) st.f_favail));
7284 PyStructSequence_SET_ITEM(v, 8, PyLong_FromLong((long) st.f_flag));
7285 PyStructSequence_SET_ITEM(v, 9, PyLong_FromLong((long) st.f_namemax));
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007286#endif
7287
Victor Stinner8c62be82010-05-06 00:08:46 +00007288 return v;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007289}
7290
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007291PyDoc_STRVAR(posix_fstatvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007292"fstatvfs(fd) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007293Perform an fstatvfs system call on the given fd.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00007294
7295static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007296posix_fstatvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00007297{
Victor Stinner8c62be82010-05-06 00:08:46 +00007298 int fd, res;
7299 struct statvfs st;
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007300
Victor Stinner8c62be82010-05-06 00:08:46 +00007301 if (!PyArg_ParseTuple(args, "i:fstatvfs", &fd))
7302 return NULL;
7303 Py_BEGIN_ALLOW_THREADS
7304 res = fstatvfs(fd, &st);
7305 Py_END_ALLOW_THREADS
7306 if (res != 0)
7307 return posix_error();
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007308
Victor Stinner8c62be82010-05-06 00:08:46 +00007309 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00007310}
Thomas Wouters477c8d52006-05-27 19:21:47 +00007311#endif /* HAVE_FSTATVFS && HAVE_SYS_STATVFS_H */
Guido van Rossum94f6f721999-01-06 18:42:14 +00007312
7313
Thomas Wouters477c8d52006-05-27 19:21:47 +00007314#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Guido van Rossum94f6f721999-01-06 18:42:14 +00007315#include <sys/statvfs.h>
7316
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007317PyDoc_STRVAR(posix_statvfs__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007318"statvfs(path) -> statvfs result\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007319Perform a statvfs system call on the given path.");
Guido van Rossum94f6f721999-01-06 18:42:14 +00007320
7321static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007322posix_statvfs(PyObject *self, PyObject *args)
Guido van Rossum94f6f721999-01-06 18:42:14 +00007323{
Victor Stinner8c62be82010-05-06 00:08:46 +00007324 char *path;
7325 int res;
7326 struct statvfs st;
7327 if (!PyArg_ParseTuple(args, "s:statvfs", &path))
7328 return NULL;
7329 Py_BEGIN_ALLOW_THREADS
7330 res = statvfs(path, &st);
7331 Py_END_ALLOW_THREADS
7332 if (res != 0)
7333 return posix_error_with_filename(path);
Guido van Rossum98bf58f2001-10-18 20:34:25 +00007334
Victor Stinner8c62be82010-05-06 00:08:46 +00007335 return _pystatvfs_fromstructstatvfs(st);
Guido van Rossum94f6f721999-01-06 18:42:14 +00007336}
7337#endif /* HAVE_STATVFS */
7338
Fred Drakec9680921999-12-13 16:37:25 +00007339/* This is used for fpathconf(), pathconf(), confstr() and sysconf().
7340 * It maps strings representing configuration variable names to
7341 * integer values, allowing those functions to be called with the
Thomas Wouters7e474022000-07-16 12:04:32 +00007342 * magic names instead of polluting the module's namespace with tons of
Fred Drake12c6e2d1999-12-14 21:25:03 +00007343 * rarely-used constants. There are three separate tables that use
7344 * these definitions.
Fred Drakebec628d1999-12-15 18:31:10 +00007345 *
7346 * This code is always included, even if none of the interfaces that
7347 * need it are included. The #if hackery needed to avoid it would be
7348 * sufficiently pervasive that it's not worth the loss of readability.
Fred Drakec9680921999-12-13 16:37:25 +00007349 */
7350struct constdef {
7351 char *name;
7352 long value;
7353};
7354
Fred Drake12c6e2d1999-12-14 21:25:03 +00007355static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007356conv_confname(PyObject *arg, int *valuep, struct constdef *table,
Guido van Rossum7d5baac2007-08-27 23:24:46 +00007357 size_t tablesize)
Fred Drake12c6e2d1999-12-14 21:25:03 +00007358{
Christian Heimes217cfd12007-12-02 14:31:20 +00007359 if (PyLong_Check(arg)) {
Stefan Krah0e803b32010-11-26 16:16:47 +00007360 *valuep = PyLong_AS_LONG(arg);
7361 return 1;
Fred Drake12c6e2d1999-12-14 21:25:03 +00007362 }
Guido van Rossumbce56a62007-05-10 18:04:33 +00007363 else {
Stefan Krah0e803b32010-11-26 16:16:47 +00007364 /* look up the value in the table using a binary search */
7365 size_t lo = 0;
7366 size_t mid;
7367 size_t hi = tablesize;
7368 int cmp;
7369 const char *confname;
7370 if (!PyUnicode_Check(arg)) {
7371 PyErr_SetString(PyExc_TypeError,
7372 "configuration names must be strings or integers");
7373 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00007374 }
Stefan Krah0e803b32010-11-26 16:16:47 +00007375 confname = _PyUnicode_AsString(arg);
7376 if (confname == NULL)
7377 return 0;
7378 while (lo < hi) {
7379 mid = (lo + hi) / 2;
7380 cmp = strcmp(confname, table[mid].name);
7381 if (cmp < 0)
7382 hi = mid;
7383 else if (cmp > 0)
7384 lo = mid + 1;
7385 else {
7386 *valuep = table[mid].value;
7387 return 1;
7388 }
7389 }
7390 PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
7391 return 0;
Victor Stinner8c62be82010-05-06 00:08:46 +00007392 }
Fred Drake12c6e2d1999-12-14 21:25:03 +00007393}
7394
7395
7396#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
7397static struct constdef posix_constants_pathconf[] = {
Fred Draked86ed291999-12-15 15:34:33 +00007398#ifdef _PC_ABI_AIO_XFER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007399 {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
Fred Draked86ed291999-12-15 15:34:33 +00007400#endif
7401#ifdef _PC_ABI_ASYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00007402 {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
Fred Draked86ed291999-12-15 15:34:33 +00007403#endif
Fred Drakec9680921999-12-13 16:37:25 +00007404#ifdef _PC_ASYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00007405 {"PC_ASYNC_IO", _PC_ASYNC_IO},
Fred Drakec9680921999-12-13 16:37:25 +00007406#endif
7407#ifdef _PC_CHOWN_RESTRICTED
Victor Stinner8c62be82010-05-06 00:08:46 +00007408 {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
Fred Drakec9680921999-12-13 16:37:25 +00007409#endif
7410#ifdef _PC_FILESIZEBITS
Victor Stinner8c62be82010-05-06 00:08:46 +00007411 {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
Fred Drakec9680921999-12-13 16:37:25 +00007412#endif
7413#ifdef _PC_LAST
Victor Stinner8c62be82010-05-06 00:08:46 +00007414 {"PC_LAST", _PC_LAST},
Fred Drakec9680921999-12-13 16:37:25 +00007415#endif
7416#ifdef _PC_LINK_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007417 {"PC_LINK_MAX", _PC_LINK_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007418#endif
7419#ifdef _PC_MAX_CANON
Victor Stinner8c62be82010-05-06 00:08:46 +00007420 {"PC_MAX_CANON", _PC_MAX_CANON},
Fred Drakec9680921999-12-13 16:37:25 +00007421#endif
7422#ifdef _PC_MAX_INPUT
Victor Stinner8c62be82010-05-06 00:08:46 +00007423 {"PC_MAX_INPUT", _PC_MAX_INPUT},
Fred Drakec9680921999-12-13 16:37:25 +00007424#endif
7425#ifdef _PC_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007426 {"PC_NAME_MAX", _PC_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007427#endif
7428#ifdef _PC_NO_TRUNC
Victor Stinner8c62be82010-05-06 00:08:46 +00007429 {"PC_NO_TRUNC", _PC_NO_TRUNC},
Fred Drakec9680921999-12-13 16:37:25 +00007430#endif
7431#ifdef _PC_PATH_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007432 {"PC_PATH_MAX", _PC_PATH_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007433#endif
7434#ifdef _PC_PIPE_BUF
Victor Stinner8c62be82010-05-06 00:08:46 +00007435 {"PC_PIPE_BUF", _PC_PIPE_BUF},
Fred Drakec9680921999-12-13 16:37:25 +00007436#endif
7437#ifdef _PC_PRIO_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00007438 {"PC_PRIO_IO", _PC_PRIO_IO},
Fred Drakec9680921999-12-13 16:37:25 +00007439#endif
7440#ifdef _PC_SOCK_MAXBUF
Victor Stinner8c62be82010-05-06 00:08:46 +00007441 {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
Fred Drakec9680921999-12-13 16:37:25 +00007442#endif
7443#ifdef _PC_SYNC_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00007444 {"PC_SYNC_IO", _PC_SYNC_IO},
Fred Drakec9680921999-12-13 16:37:25 +00007445#endif
7446#ifdef _PC_VDISABLE
Victor Stinner8c62be82010-05-06 00:08:46 +00007447 {"PC_VDISABLE", _PC_VDISABLE},
Fred Drakec9680921999-12-13 16:37:25 +00007448#endif
Jesus Cea7e9065c2010-10-25 13:02:04 +00007449#ifdef _PC_ACL_ENABLED
7450 {"PC_ACL_ENABLED", _PC_ACL_ENABLED},
7451#endif
7452#ifdef _PC_MIN_HOLE_SIZE
7453 {"PC_MIN_HOLE_SIZE", _PC_MIN_HOLE_SIZE},
7454#endif
7455#ifdef _PC_ALLOC_SIZE_MIN
7456 {"PC_ALLOC_SIZE_MIN", _PC_ALLOC_SIZE_MIN},
7457#endif
7458#ifdef _PC_REC_INCR_XFER_SIZE
7459 {"PC_REC_INCR_XFER_SIZE", _PC_REC_INCR_XFER_SIZE},
7460#endif
7461#ifdef _PC_REC_MAX_XFER_SIZE
7462 {"PC_REC_MAX_XFER_SIZE", _PC_REC_MAX_XFER_SIZE},
7463#endif
7464#ifdef _PC_REC_MIN_XFER_SIZE
7465 {"PC_REC_MIN_XFER_SIZE", _PC_REC_MIN_XFER_SIZE},
7466#endif
7467#ifdef _PC_REC_XFER_ALIGN
7468 {"PC_REC_XFER_ALIGN", _PC_REC_XFER_ALIGN},
7469#endif
7470#ifdef _PC_SYMLINK_MAX
7471 {"PC_SYMLINK_MAX", _PC_SYMLINK_MAX},
7472#endif
7473#ifdef _PC_XATTR_ENABLED
7474 {"PC_XATTR_ENABLED", _PC_XATTR_ENABLED},
7475#endif
7476#ifdef _PC_XATTR_EXISTS
7477 {"PC_XATTR_EXISTS", _PC_XATTR_EXISTS},
7478#endif
7479#ifdef _PC_TIMESTAMP_RESOLUTION
7480 {"PC_TIMESTAMP_RESOLUTION", _PC_TIMESTAMP_RESOLUTION},
7481#endif
Fred Drakec9680921999-12-13 16:37:25 +00007482};
7483
Fred Drakec9680921999-12-13 16:37:25 +00007484static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007485conv_path_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00007486{
7487 return conv_confname(arg, valuep, posix_constants_pathconf,
7488 sizeof(posix_constants_pathconf)
7489 / sizeof(struct constdef));
7490}
7491#endif
7492
7493#ifdef HAVE_FPATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007494PyDoc_STRVAR(posix_fpathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007495"fpathconf(fd, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00007496Return the configuration limit name for the file descriptor fd.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007497If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00007498
7499static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007500posix_fpathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00007501{
7502 PyObject *result = NULL;
7503 int name, fd;
7504
Fred Drake12c6e2d1999-12-14 21:25:03 +00007505 if (PyArg_ParseTuple(args, "iO&:fpathconf", &fd,
7506 conv_path_confname, &name)) {
Stefan Krah0e803b32010-11-26 16:16:47 +00007507 long limit;
Fred Drakec9680921999-12-13 16:37:25 +00007508
Stefan Krah0e803b32010-11-26 16:16:47 +00007509 errno = 0;
7510 limit = fpathconf(fd, name);
7511 if (limit == -1 && errno != 0)
7512 posix_error();
7513 else
7514 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00007515 }
7516 return result;
7517}
7518#endif
7519
7520
7521#ifdef HAVE_PATHCONF
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007522PyDoc_STRVAR(posix_pathconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007523"pathconf(path, name) -> integer\n\n\
Fred Drakec9680921999-12-13 16:37:25 +00007524Return the configuration limit name for the file or directory path.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007525If there is no limit, return -1.");
Fred Drakec9680921999-12-13 16:37:25 +00007526
7527static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007528posix_pathconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00007529{
7530 PyObject *result = NULL;
7531 int name;
7532 char *path;
7533
7534 if (PyArg_ParseTuple(args, "sO&:pathconf", &path,
7535 conv_path_confname, &name)) {
Victor Stinner8c62be82010-05-06 00:08:46 +00007536 long limit;
Fred Drakec9680921999-12-13 16:37:25 +00007537
Victor Stinner8c62be82010-05-06 00:08:46 +00007538 errno = 0;
7539 limit = pathconf(path, name);
7540 if (limit == -1 && errno != 0) {
7541 if (errno == EINVAL)
Stefan Krah99439262010-11-26 12:58:05 +00007542 /* could be a path or name problem */
7543 posix_error();
Fred Drakec9680921999-12-13 16:37:25 +00007544 else
Stefan Krah99439262010-11-26 12:58:05 +00007545 posix_error_with_filename(path);
Victor Stinner8c62be82010-05-06 00:08:46 +00007546 }
7547 else
7548 result = PyLong_FromLong(limit);
Fred Drakec9680921999-12-13 16:37:25 +00007549 }
7550 return result;
7551}
7552#endif
7553
7554#ifdef HAVE_CONFSTR
7555static struct constdef posix_constants_confstr[] = {
Fred Draked86ed291999-12-15 15:34:33 +00007556#ifdef _CS_ARCHITECTURE
Victor Stinner8c62be82010-05-06 00:08:46 +00007557 {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
Fred Draked86ed291999-12-15 15:34:33 +00007558#endif
Mark Dickinson876d7c82010-04-16 12:47:52 +00007559#ifdef _CS_GNU_LIBC_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00007560 {"CS_GNU_LIBC_VERSION", _CS_GNU_LIBC_VERSION},
Mark Dickinson876d7c82010-04-16 12:47:52 +00007561#endif
7562#ifdef _CS_GNU_LIBPTHREAD_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00007563 {"CS_GNU_LIBPTHREAD_VERSION", _CS_GNU_LIBPTHREAD_VERSION},
Mark Dickinson876d7c82010-04-16 12:47:52 +00007564#endif
Fred Draked86ed291999-12-15 15:34:33 +00007565#ifdef _CS_HOSTNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00007566 {"CS_HOSTNAME", _CS_HOSTNAME},
Fred Draked86ed291999-12-15 15:34:33 +00007567#endif
7568#ifdef _CS_HW_PROVIDER
Victor Stinner8c62be82010-05-06 00:08:46 +00007569 {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
Fred Draked86ed291999-12-15 15:34:33 +00007570#endif
7571#ifdef _CS_HW_SERIAL
Victor Stinner8c62be82010-05-06 00:08:46 +00007572 {"CS_HW_SERIAL", _CS_HW_SERIAL},
Fred Draked86ed291999-12-15 15:34:33 +00007573#endif
7574#ifdef _CS_INITTAB_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00007575 {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00007576#endif
Fred Drakec9680921999-12-13 16:37:25 +00007577#ifdef _CS_LFS64_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007578 {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007579#endif
7580#ifdef _CS_LFS64_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007581 {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007582#endif
7583#ifdef _CS_LFS64_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00007584 {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00007585#endif
7586#ifdef _CS_LFS64_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007587 {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007588#endif
7589#ifdef _CS_LFS_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007590 {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007591#endif
7592#ifdef _CS_LFS_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007593 {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007594#endif
7595#ifdef _CS_LFS_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00007596 {"CS_LFS_LIBS", _CS_LFS_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00007597#endif
7598#ifdef _CS_LFS_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007599 {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007600#endif
Fred Draked86ed291999-12-15 15:34:33 +00007601#ifdef _CS_MACHINE
Victor Stinner8c62be82010-05-06 00:08:46 +00007602 {"CS_MACHINE", _CS_MACHINE},
Fred Draked86ed291999-12-15 15:34:33 +00007603#endif
Fred Drakec9680921999-12-13 16:37:25 +00007604#ifdef _CS_PATH
Victor Stinner8c62be82010-05-06 00:08:46 +00007605 {"CS_PATH", _CS_PATH},
Fred Drakec9680921999-12-13 16:37:25 +00007606#endif
Fred Draked86ed291999-12-15 15:34:33 +00007607#ifdef _CS_RELEASE
Victor Stinner8c62be82010-05-06 00:08:46 +00007608 {"CS_RELEASE", _CS_RELEASE},
Fred Draked86ed291999-12-15 15:34:33 +00007609#endif
7610#ifdef _CS_SRPC_DOMAIN
Victor Stinner8c62be82010-05-06 00:08:46 +00007611 {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
Fred Draked86ed291999-12-15 15:34:33 +00007612#endif
7613#ifdef _CS_SYSNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00007614 {"CS_SYSNAME", _CS_SYSNAME},
Fred Draked86ed291999-12-15 15:34:33 +00007615#endif
7616#ifdef _CS_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00007617 {"CS_VERSION", _CS_VERSION},
Fred Draked86ed291999-12-15 15:34:33 +00007618#endif
Fred Drakec9680921999-12-13 16:37:25 +00007619#ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007620 {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007621#endif
7622#ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007623 {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007624#endif
7625#ifdef _CS_XBS5_ILP32_OFF32_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00007626 {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00007627#endif
7628#ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007629 {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007630#endif
7631#ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007632 {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007633#endif
7634#ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007635 {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007636#endif
7637#ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00007638 {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00007639#endif
7640#ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007641 {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007642#endif
7643#ifdef _CS_XBS5_LP64_OFF64_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007644 {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007645#endif
7646#ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007647 {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007648#endif
7649#ifdef _CS_XBS5_LP64_OFF64_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00007650 {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00007651#endif
7652#ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007653 {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007654#endif
7655#ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007656 {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007657#endif
7658#ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007659 {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007660#endif
7661#ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
Victor Stinner8c62be82010-05-06 00:08:46 +00007662 {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
Fred Drakec9680921999-12-13 16:37:25 +00007663#endif
7664#ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00007665 {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
Fred Drakec9680921999-12-13 16:37:25 +00007666#endif
Fred Draked86ed291999-12-15 15:34:33 +00007667#ifdef _MIPS_CS_AVAIL_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00007668 {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00007669#endif
7670#ifdef _MIPS_CS_BASE
Victor Stinner8c62be82010-05-06 00:08:46 +00007671 {"MIPS_CS_BASE", _MIPS_CS_BASE},
Fred Draked86ed291999-12-15 15:34:33 +00007672#endif
7673#ifdef _MIPS_CS_HOSTID
Victor Stinner8c62be82010-05-06 00:08:46 +00007674 {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
Fred Draked86ed291999-12-15 15:34:33 +00007675#endif
7676#ifdef _MIPS_CS_HW_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00007677 {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00007678#endif
7679#ifdef _MIPS_CS_NUM_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00007680 {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00007681#endif
7682#ifdef _MIPS_CS_OSREL_MAJ
Victor Stinner8c62be82010-05-06 00:08:46 +00007683 {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
Fred Draked86ed291999-12-15 15:34:33 +00007684#endif
7685#ifdef _MIPS_CS_OSREL_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00007686 {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
Fred Draked86ed291999-12-15 15:34:33 +00007687#endif
7688#ifdef _MIPS_CS_OSREL_PATCH
Victor Stinner8c62be82010-05-06 00:08:46 +00007689 {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
Fred Draked86ed291999-12-15 15:34:33 +00007690#endif
7691#ifdef _MIPS_CS_OS_NAME
Victor Stinner8c62be82010-05-06 00:08:46 +00007692 {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
Fred Draked86ed291999-12-15 15:34:33 +00007693#endif
7694#ifdef _MIPS_CS_OS_PROVIDER
Victor Stinner8c62be82010-05-06 00:08:46 +00007695 {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
Fred Draked86ed291999-12-15 15:34:33 +00007696#endif
7697#ifdef _MIPS_CS_PROCESSORS
Victor Stinner8c62be82010-05-06 00:08:46 +00007698 {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
Fred Draked86ed291999-12-15 15:34:33 +00007699#endif
7700#ifdef _MIPS_CS_SERIAL
Victor Stinner8c62be82010-05-06 00:08:46 +00007701 {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
Fred Draked86ed291999-12-15 15:34:33 +00007702#endif
7703#ifdef _MIPS_CS_VENDOR
Victor Stinner8c62be82010-05-06 00:08:46 +00007704 {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
Fred Draked86ed291999-12-15 15:34:33 +00007705#endif
Fred Drakec9680921999-12-13 16:37:25 +00007706};
7707
7708static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007709conv_confstr_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00007710{
7711 return conv_confname(arg, valuep, posix_constants_confstr,
7712 sizeof(posix_constants_confstr)
7713 / sizeof(struct constdef));
7714}
7715
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007716PyDoc_STRVAR(posix_confstr__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00007717"confstr(name) -> string\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007718Return a string-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00007719
7720static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00007721posix_confstr(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00007722{
7723 PyObject *result = NULL;
7724 int name;
Victor Stinnercb043522010-09-10 23:49:04 +00007725 char buffer[255];
Stefan Krah99439262010-11-26 12:58:05 +00007726 int len;
Fred Drakec9680921999-12-13 16:37:25 +00007727
Victor Stinnercb043522010-09-10 23:49:04 +00007728 if (!PyArg_ParseTuple(args, "O&:confstr", conv_confstr_confname, &name))
7729 return NULL;
7730
7731 errno = 0;
7732 len = confstr(name, buffer, sizeof(buffer));
7733 if (len == 0) {
7734 if (errno) {
7735 posix_error();
7736 return NULL;
Fred Drakec9680921999-12-13 16:37:25 +00007737 }
7738 else {
Victor Stinnercb043522010-09-10 23:49:04 +00007739 Py_RETURN_NONE;
Fred Drakec9680921999-12-13 16:37:25 +00007740 }
7741 }
Victor Stinnercb043522010-09-10 23:49:04 +00007742
7743 if ((unsigned int)len >= sizeof(buffer)) {
7744 char *buf = PyMem_Malloc(len);
7745 if (buf == NULL)
7746 return PyErr_NoMemory();
7747 confstr(name, buf, len);
7748 result = PyUnicode_DecodeFSDefaultAndSize(buf, len-1);
7749 PyMem_Free(buf);
7750 }
7751 else
7752 result = PyUnicode_DecodeFSDefaultAndSize(buffer, len-1);
Fred Drakec9680921999-12-13 16:37:25 +00007753 return result;
7754}
7755#endif
7756
7757
7758#ifdef HAVE_SYSCONF
7759static struct constdef posix_constants_sysconf[] = {
7760#ifdef _SC_2_CHAR_TERM
Victor Stinner8c62be82010-05-06 00:08:46 +00007761 {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
Fred Drakec9680921999-12-13 16:37:25 +00007762#endif
7763#ifdef _SC_2_C_BIND
Victor Stinner8c62be82010-05-06 00:08:46 +00007764 {"SC_2_C_BIND", _SC_2_C_BIND},
Fred Drakec9680921999-12-13 16:37:25 +00007765#endif
7766#ifdef _SC_2_C_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00007767 {"SC_2_C_DEV", _SC_2_C_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00007768#endif
7769#ifdef _SC_2_C_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00007770 {"SC_2_C_VERSION", _SC_2_C_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00007771#endif
7772#ifdef _SC_2_FORT_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00007773 {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00007774#endif
7775#ifdef _SC_2_FORT_RUN
Victor Stinner8c62be82010-05-06 00:08:46 +00007776 {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
Fred Drakec9680921999-12-13 16:37:25 +00007777#endif
7778#ifdef _SC_2_LOCALEDEF
Victor Stinner8c62be82010-05-06 00:08:46 +00007779 {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
Fred Drakec9680921999-12-13 16:37:25 +00007780#endif
7781#ifdef _SC_2_SW_DEV
Victor Stinner8c62be82010-05-06 00:08:46 +00007782 {"SC_2_SW_DEV", _SC_2_SW_DEV},
Fred Drakec9680921999-12-13 16:37:25 +00007783#endif
7784#ifdef _SC_2_UPE
Victor Stinner8c62be82010-05-06 00:08:46 +00007785 {"SC_2_UPE", _SC_2_UPE},
Fred Drakec9680921999-12-13 16:37:25 +00007786#endif
7787#ifdef _SC_2_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00007788 {"SC_2_VERSION", _SC_2_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00007789#endif
Fred Draked86ed291999-12-15 15:34:33 +00007790#ifdef _SC_ABI_ASYNCHRONOUS_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00007791 {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
Fred Draked86ed291999-12-15 15:34:33 +00007792#endif
7793#ifdef _SC_ACL
Victor Stinner8c62be82010-05-06 00:08:46 +00007794 {"SC_ACL", _SC_ACL},
Fred Draked86ed291999-12-15 15:34:33 +00007795#endif
Fred Drakec9680921999-12-13 16:37:25 +00007796#ifdef _SC_AIO_LISTIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007797 {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007798#endif
Fred Drakec9680921999-12-13 16:37:25 +00007799#ifdef _SC_AIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007800 {"SC_AIO_MAX", _SC_AIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007801#endif
7802#ifdef _SC_AIO_PRIO_DELTA_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007803 {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007804#endif
7805#ifdef _SC_ARG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007806 {"SC_ARG_MAX", _SC_ARG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007807#endif
7808#ifdef _SC_ASYNCHRONOUS_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00007809 {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
Fred Drakec9680921999-12-13 16:37:25 +00007810#endif
7811#ifdef _SC_ATEXIT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007812 {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007813#endif
Fred Draked86ed291999-12-15 15:34:33 +00007814#ifdef _SC_AUDIT
Victor Stinner8c62be82010-05-06 00:08:46 +00007815 {"SC_AUDIT", _SC_AUDIT},
Fred Draked86ed291999-12-15 15:34:33 +00007816#endif
Fred Drakec9680921999-12-13 16:37:25 +00007817#ifdef _SC_AVPHYS_PAGES
Victor Stinner8c62be82010-05-06 00:08:46 +00007818 {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
Fred Drakec9680921999-12-13 16:37:25 +00007819#endif
7820#ifdef _SC_BC_BASE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007821 {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007822#endif
7823#ifdef _SC_BC_DIM_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007824 {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007825#endif
7826#ifdef _SC_BC_SCALE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007827 {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007828#endif
7829#ifdef _SC_BC_STRING_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007830 {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007831#endif
Fred Draked86ed291999-12-15 15:34:33 +00007832#ifdef _SC_CAP
Victor Stinner8c62be82010-05-06 00:08:46 +00007833 {"SC_CAP", _SC_CAP},
Fred Draked86ed291999-12-15 15:34:33 +00007834#endif
Fred Drakec9680921999-12-13 16:37:25 +00007835#ifdef _SC_CHARCLASS_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007836 {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007837#endif
7838#ifdef _SC_CHAR_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00007839 {"SC_CHAR_BIT", _SC_CHAR_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00007840#endif
7841#ifdef _SC_CHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007842 {"SC_CHAR_MAX", _SC_CHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007843#endif
7844#ifdef _SC_CHAR_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00007845 {"SC_CHAR_MIN", _SC_CHAR_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00007846#endif
7847#ifdef _SC_CHILD_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007848 {"SC_CHILD_MAX", _SC_CHILD_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007849#endif
7850#ifdef _SC_CLK_TCK
Victor Stinner8c62be82010-05-06 00:08:46 +00007851 {"SC_CLK_TCK", _SC_CLK_TCK},
Fred Drakec9680921999-12-13 16:37:25 +00007852#endif
7853#ifdef _SC_COHER_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00007854 {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00007855#endif
7856#ifdef _SC_COLL_WEIGHTS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007857 {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007858#endif
7859#ifdef _SC_DCACHE_ASSOC
Victor Stinner8c62be82010-05-06 00:08:46 +00007860 {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
Fred Drakec9680921999-12-13 16:37:25 +00007861#endif
7862#ifdef _SC_DCACHE_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00007863 {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00007864#endif
7865#ifdef _SC_DCACHE_LINESZ
Victor Stinner8c62be82010-05-06 00:08:46 +00007866 {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
Fred Drakec9680921999-12-13 16:37:25 +00007867#endif
7868#ifdef _SC_DCACHE_SZ
Victor Stinner8c62be82010-05-06 00:08:46 +00007869 {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
Fred Drakec9680921999-12-13 16:37:25 +00007870#endif
7871#ifdef _SC_DCACHE_TBLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00007872 {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00007873#endif
7874#ifdef _SC_DELAYTIMER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007875 {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007876#endif
7877#ifdef _SC_EQUIV_CLASS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007878 {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007879#endif
7880#ifdef _SC_EXPR_NEST_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007881 {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007882#endif
7883#ifdef _SC_FSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00007884 {"SC_FSYNC", _SC_FSYNC},
Fred Drakec9680921999-12-13 16:37:25 +00007885#endif
7886#ifdef _SC_GETGR_R_SIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007887 {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007888#endif
7889#ifdef _SC_GETPW_R_SIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007890 {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007891#endif
7892#ifdef _SC_ICACHE_ASSOC
Victor Stinner8c62be82010-05-06 00:08:46 +00007893 {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
Fred Drakec9680921999-12-13 16:37:25 +00007894#endif
7895#ifdef _SC_ICACHE_BLKSZ
Victor Stinner8c62be82010-05-06 00:08:46 +00007896 {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
Fred Drakec9680921999-12-13 16:37:25 +00007897#endif
7898#ifdef _SC_ICACHE_LINESZ
Victor Stinner8c62be82010-05-06 00:08:46 +00007899 {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
Fred Drakec9680921999-12-13 16:37:25 +00007900#endif
7901#ifdef _SC_ICACHE_SZ
Victor Stinner8c62be82010-05-06 00:08:46 +00007902 {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
Fred Drakec9680921999-12-13 16:37:25 +00007903#endif
Fred Draked86ed291999-12-15 15:34:33 +00007904#ifdef _SC_INF
Victor Stinner8c62be82010-05-06 00:08:46 +00007905 {"SC_INF", _SC_INF},
Fred Draked86ed291999-12-15 15:34:33 +00007906#endif
Fred Drakec9680921999-12-13 16:37:25 +00007907#ifdef _SC_INT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007908 {"SC_INT_MAX", _SC_INT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007909#endif
7910#ifdef _SC_INT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00007911 {"SC_INT_MIN", _SC_INT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00007912#endif
7913#ifdef _SC_IOV_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007914 {"SC_IOV_MAX", _SC_IOV_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007915#endif
Fred Draked86ed291999-12-15 15:34:33 +00007916#ifdef _SC_IP_SECOPTS
Victor Stinner8c62be82010-05-06 00:08:46 +00007917 {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
Fred Draked86ed291999-12-15 15:34:33 +00007918#endif
Fred Drakec9680921999-12-13 16:37:25 +00007919#ifdef _SC_JOB_CONTROL
Victor Stinner8c62be82010-05-06 00:08:46 +00007920 {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
Fred Drakec9680921999-12-13 16:37:25 +00007921#endif
Fred Draked86ed291999-12-15 15:34:33 +00007922#ifdef _SC_KERN_POINTERS
Victor Stinner8c62be82010-05-06 00:08:46 +00007923 {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
Fred Draked86ed291999-12-15 15:34:33 +00007924#endif
7925#ifdef _SC_KERN_SIM
Victor Stinner8c62be82010-05-06 00:08:46 +00007926 {"SC_KERN_SIM", _SC_KERN_SIM},
Fred Draked86ed291999-12-15 15:34:33 +00007927#endif
Fred Drakec9680921999-12-13 16:37:25 +00007928#ifdef _SC_LINE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007929 {"SC_LINE_MAX", _SC_LINE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007930#endif
7931#ifdef _SC_LOGIN_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007932 {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007933#endif
7934#ifdef _SC_LOGNAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007935 {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007936#endif
7937#ifdef _SC_LONG_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00007938 {"SC_LONG_BIT", _SC_LONG_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00007939#endif
Fred Draked86ed291999-12-15 15:34:33 +00007940#ifdef _SC_MAC
Victor Stinner8c62be82010-05-06 00:08:46 +00007941 {"SC_MAC", _SC_MAC},
Fred Draked86ed291999-12-15 15:34:33 +00007942#endif
Fred Drakec9680921999-12-13 16:37:25 +00007943#ifdef _SC_MAPPED_FILES
Victor Stinner8c62be82010-05-06 00:08:46 +00007944 {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
Fred Drakec9680921999-12-13 16:37:25 +00007945#endif
7946#ifdef _SC_MAXPID
Victor Stinner8c62be82010-05-06 00:08:46 +00007947 {"SC_MAXPID", _SC_MAXPID},
Fred Drakec9680921999-12-13 16:37:25 +00007948#endif
7949#ifdef _SC_MB_LEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007950 {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007951#endif
7952#ifdef _SC_MEMLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +00007953 {"SC_MEMLOCK", _SC_MEMLOCK},
Fred Drakec9680921999-12-13 16:37:25 +00007954#endif
7955#ifdef _SC_MEMLOCK_RANGE
Victor Stinner8c62be82010-05-06 00:08:46 +00007956 {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
Fred Drakec9680921999-12-13 16:37:25 +00007957#endif
7958#ifdef _SC_MEMORY_PROTECTION
Victor Stinner8c62be82010-05-06 00:08:46 +00007959 {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
Fred Drakec9680921999-12-13 16:37:25 +00007960#endif
7961#ifdef _SC_MESSAGE_PASSING
Victor Stinner8c62be82010-05-06 00:08:46 +00007962 {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
Fred Drakec9680921999-12-13 16:37:25 +00007963#endif
Fred Draked86ed291999-12-15 15:34:33 +00007964#ifdef _SC_MMAP_FIXED_ALIGNMENT
Victor Stinner8c62be82010-05-06 00:08:46 +00007965 {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
Fred Draked86ed291999-12-15 15:34:33 +00007966#endif
Fred Drakec9680921999-12-13 16:37:25 +00007967#ifdef _SC_MQ_OPEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007968 {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007969#endif
7970#ifdef _SC_MQ_PRIO_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007971 {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007972#endif
Fred Draked86ed291999-12-15 15:34:33 +00007973#ifdef _SC_NACLS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007974 {"SC_NACLS_MAX", _SC_NACLS_MAX},
Fred Draked86ed291999-12-15 15:34:33 +00007975#endif
Fred Drakec9680921999-12-13 16:37:25 +00007976#ifdef _SC_NGROUPS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007977 {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00007978#endif
7979#ifdef _SC_NL_ARGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007980 {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00007981#endif
7982#ifdef _SC_NL_LANGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007983 {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00007984#endif
7985#ifdef _SC_NL_MSGMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007986 {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
Fred Drakec9680921999-12-13 16:37:25 +00007987#endif
7988#ifdef _SC_NL_NMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007989 {"SC_NL_NMAX", _SC_NL_NMAX},
Fred Drakec9680921999-12-13 16:37:25 +00007990#endif
7991#ifdef _SC_NL_SETMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007992 {"SC_NL_SETMAX", _SC_NL_SETMAX},
Fred Drakec9680921999-12-13 16:37:25 +00007993#endif
7994#ifdef _SC_NL_TEXTMAX
Victor Stinner8c62be82010-05-06 00:08:46 +00007995 {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
Fred Drakec9680921999-12-13 16:37:25 +00007996#endif
7997#ifdef _SC_NPROCESSORS_CONF
Victor Stinner8c62be82010-05-06 00:08:46 +00007998 {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
Fred Drakec9680921999-12-13 16:37:25 +00007999#endif
8000#ifdef _SC_NPROCESSORS_ONLN
Victor Stinner8c62be82010-05-06 00:08:46 +00008001 {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
Fred Drakec9680921999-12-13 16:37:25 +00008002#endif
Fred Draked86ed291999-12-15 15:34:33 +00008003#ifdef _SC_NPROC_CONF
Victor Stinner8c62be82010-05-06 00:08:46 +00008004 {"SC_NPROC_CONF", _SC_NPROC_CONF},
Fred Draked86ed291999-12-15 15:34:33 +00008005#endif
8006#ifdef _SC_NPROC_ONLN
Victor Stinner8c62be82010-05-06 00:08:46 +00008007 {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
Fred Draked86ed291999-12-15 15:34:33 +00008008#endif
Fred Drakec9680921999-12-13 16:37:25 +00008009#ifdef _SC_NZERO
Victor Stinner8c62be82010-05-06 00:08:46 +00008010 {"SC_NZERO", _SC_NZERO},
Fred Drakec9680921999-12-13 16:37:25 +00008011#endif
8012#ifdef _SC_OPEN_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008013 {"SC_OPEN_MAX", _SC_OPEN_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008014#endif
8015#ifdef _SC_PAGESIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008016 {"SC_PAGESIZE", _SC_PAGESIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008017#endif
8018#ifdef _SC_PAGE_SIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008019 {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008020#endif
8021#ifdef _SC_PASS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008022 {"SC_PASS_MAX", _SC_PASS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008023#endif
8024#ifdef _SC_PHYS_PAGES
Victor Stinner8c62be82010-05-06 00:08:46 +00008025 {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
Fred Drakec9680921999-12-13 16:37:25 +00008026#endif
8027#ifdef _SC_PII
Victor Stinner8c62be82010-05-06 00:08:46 +00008028 {"SC_PII", _SC_PII},
Fred Drakec9680921999-12-13 16:37:25 +00008029#endif
8030#ifdef _SC_PII_INTERNET
Victor Stinner8c62be82010-05-06 00:08:46 +00008031 {"SC_PII_INTERNET", _SC_PII_INTERNET},
Fred Drakec9680921999-12-13 16:37:25 +00008032#endif
8033#ifdef _SC_PII_INTERNET_DGRAM
Victor Stinner8c62be82010-05-06 00:08:46 +00008034 {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
Fred Drakec9680921999-12-13 16:37:25 +00008035#endif
8036#ifdef _SC_PII_INTERNET_STREAM
Victor Stinner8c62be82010-05-06 00:08:46 +00008037 {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
Fred Drakec9680921999-12-13 16:37:25 +00008038#endif
8039#ifdef _SC_PII_OSI
Victor Stinner8c62be82010-05-06 00:08:46 +00008040 {"SC_PII_OSI", _SC_PII_OSI},
Fred Drakec9680921999-12-13 16:37:25 +00008041#endif
8042#ifdef _SC_PII_OSI_CLTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008043 {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
Fred Drakec9680921999-12-13 16:37:25 +00008044#endif
8045#ifdef _SC_PII_OSI_COTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008046 {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
Fred Drakec9680921999-12-13 16:37:25 +00008047#endif
8048#ifdef _SC_PII_OSI_M
Victor Stinner8c62be82010-05-06 00:08:46 +00008049 {"SC_PII_OSI_M", _SC_PII_OSI_M},
Fred Drakec9680921999-12-13 16:37:25 +00008050#endif
8051#ifdef _SC_PII_SOCKET
Victor Stinner8c62be82010-05-06 00:08:46 +00008052 {"SC_PII_SOCKET", _SC_PII_SOCKET},
Fred Drakec9680921999-12-13 16:37:25 +00008053#endif
8054#ifdef _SC_PII_XTI
Victor Stinner8c62be82010-05-06 00:08:46 +00008055 {"SC_PII_XTI", _SC_PII_XTI},
Fred Drakec9680921999-12-13 16:37:25 +00008056#endif
8057#ifdef _SC_POLL
Victor Stinner8c62be82010-05-06 00:08:46 +00008058 {"SC_POLL", _SC_POLL},
Fred Drakec9680921999-12-13 16:37:25 +00008059#endif
8060#ifdef _SC_PRIORITIZED_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008061 {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008062#endif
8063#ifdef _SC_PRIORITY_SCHEDULING
Victor Stinner8c62be82010-05-06 00:08:46 +00008064 {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
Fred Drakec9680921999-12-13 16:37:25 +00008065#endif
8066#ifdef _SC_REALTIME_SIGNALS
Victor Stinner8c62be82010-05-06 00:08:46 +00008067 {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
Fred Drakec9680921999-12-13 16:37:25 +00008068#endif
8069#ifdef _SC_RE_DUP_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008070 {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008071#endif
8072#ifdef _SC_RTSIG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008073 {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008074#endif
8075#ifdef _SC_SAVED_IDS
Victor Stinner8c62be82010-05-06 00:08:46 +00008076 {"SC_SAVED_IDS", _SC_SAVED_IDS},
Fred Drakec9680921999-12-13 16:37:25 +00008077#endif
8078#ifdef _SC_SCHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008079 {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008080#endif
8081#ifdef _SC_SCHAR_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008082 {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008083#endif
8084#ifdef _SC_SELECT
Victor Stinner8c62be82010-05-06 00:08:46 +00008085 {"SC_SELECT", _SC_SELECT},
Fred Drakec9680921999-12-13 16:37:25 +00008086#endif
8087#ifdef _SC_SEMAPHORES
Victor Stinner8c62be82010-05-06 00:08:46 +00008088 {"SC_SEMAPHORES", _SC_SEMAPHORES},
Fred Drakec9680921999-12-13 16:37:25 +00008089#endif
8090#ifdef _SC_SEM_NSEMS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008091 {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008092#endif
8093#ifdef _SC_SEM_VALUE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008094 {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008095#endif
8096#ifdef _SC_SHARED_MEMORY_OBJECTS
Victor Stinner8c62be82010-05-06 00:08:46 +00008097 {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
Fred Drakec9680921999-12-13 16:37:25 +00008098#endif
8099#ifdef _SC_SHRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008100 {"SC_SHRT_MAX", _SC_SHRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008101#endif
8102#ifdef _SC_SHRT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008103 {"SC_SHRT_MIN", _SC_SHRT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008104#endif
8105#ifdef _SC_SIGQUEUE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008106 {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008107#endif
8108#ifdef _SC_SIGRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008109 {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008110#endif
8111#ifdef _SC_SIGRT_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008112 {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008113#endif
Fred Draked86ed291999-12-15 15:34:33 +00008114#ifdef _SC_SOFTPOWER
Victor Stinner8c62be82010-05-06 00:08:46 +00008115 {"SC_SOFTPOWER", _SC_SOFTPOWER},
Fred Draked86ed291999-12-15 15:34:33 +00008116#endif
Fred Drakec9680921999-12-13 16:37:25 +00008117#ifdef _SC_SPLIT_CACHE
Victor Stinner8c62be82010-05-06 00:08:46 +00008118 {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
Fred Drakec9680921999-12-13 16:37:25 +00008119#endif
8120#ifdef _SC_SSIZE_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008121 {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008122#endif
8123#ifdef _SC_STACK_PROT
Victor Stinner8c62be82010-05-06 00:08:46 +00008124 {"SC_STACK_PROT", _SC_STACK_PROT},
Fred Drakec9680921999-12-13 16:37:25 +00008125#endif
8126#ifdef _SC_STREAM_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008127 {"SC_STREAM_MAX", _SC_STREAM_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008128#endif
8129#ifdef _SC_SYNCHRONIZED_IO
Victor Stinner8c62be82010-05-06 00:08:46 +00008130 {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
Fred Drakec9680921999-12-13 16:37:25 +00008131#endif
8132#ifdef _SC_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00008133 {"SC_THREADS", _SC_THREADS},
Fred Drakec9680921999-12-13 16:37:25 +00008134#endif
8135#ifdef _SC_THREAD_ATTR_STACKADDR
Victor Stinner8c62be82010-05-06 00:08:46 +00008136 {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
Fred Drakec9680921999-12-13 16:37:25 +00008137#endif
8138#ifdef _SC_THREAD_ATTR_STACKSIZE
Victor Stinner8c62be82010-05-06 00:08:46 +00008139 {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
Fred Drakec9680921999-12-13 16:37:25 +00008140#endif
8141#ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
Victor Stinner8c62be82010-05-06 00:08:46 +00008142 {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
Fred Drakec9680921999-12-13 16:37:25 +00008143#endif
8144#ifdef _SC_THREAD_KEYS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008145 {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008146#endif
8147#ifdef _SC_THREAD_PRIORITY_SCHEDULING
Victor Stinner8c62be82010-05-06 00:08:46 +00008148 {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
Fred Drakec9680921999-12-13 16:37:25 +00008149#endif
8150#ifdef _SC_THREAD_PRIO_INHERIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008151 {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
Fred Drakec9680921999-12-13 16:37:25 +00008152#endif
8153#ifdef _SC_THREAD_PRIO_PROTECT
Victor Stinner8c62be82010-05-06 00:08:46 +00008154 {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
Fred Drakec9680921999-12-13 16:37:25 +00008155#endif
8156#ifdef _SC_THREAD_PROCESS_SHARED
Victor Stinner8c62be82010-05-06 00:08:46 +00008157 {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
Fred Drakec9680921999-12-13 16:37:25 +00008158#endif
8159#ifdef _SC_THREAD_SAFE_FUNCTIONS
Victor Stinner8c62be82010-05-06 00:08:46 +00008160 {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
Fred Drakec9680921999-12-13 16:37:25 +00008161#endif
8162#ifdef _SC_THREAD_STACK_MIN
Victor Stinner8c62be82010-05-06 00:08:46 +00008163 {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
Fred Drakec9680921999-12-13 16:37:25 +00008164#endif
8165#ifdef _SC_THREAD_THREADS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008166 {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008167#endif
8168#ifdef _SC_TIMERS
Victor Stinner8c62be82010-05-06 00:08:46 +00008169 {"SC_TIMERS", _SC_TIMERS},
Fred Drakec9680921999-12-13 16:37:25 +00008170#endif
8171#ifdef _SC_TIMER_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008172 {"SC_TIMER_MAX", _SC_TIMER_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008173#endif
8174#ifdef _SC_TTY_NAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008175 {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008176#endif
8177#ifdef _SC_TZNAME_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008178 {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008179#endif
8180#ifdef _SC_T_IOV_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008181 {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008182#endif
8183#ifdef _SC_UCHAR_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008184 {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008185#endif
8186#ifdef _SC_UINT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008187 {"SC_UINT_MAX", _SC_UINT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008188#endif
8189#ifdef _SC_UIO_MAXIOV
Victor Stinner8c62be82010-05-06 00:08:46 +00008190 {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
Fred Drakec9680921999-12-13 16:37:25 +00008191#endif
8192#ifdef _SC_ULONG_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008193 {"SC_ULONG_MAX", _SC_ULONG_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008194#endif
8195#ifdef _SC_USHRT_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00008196 {"SC_USHRT_MAX", _SC_USHRT_MAX},
Fred Drakec9680921999-12-13 16:37:25 +00008197#endif
8198#ifdef _SC_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008199 {"SC_VERSION", _SC_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00008200#endif
8201#ifdef _SC_WORD_BIT
Victor Stinner8c62be82010-05-06 00:08:46 +00008202 {"SC_WORD_BIT", _SC_WORD_BIT},
Fred Drakec9680921999-12-13 16:37:25 +00008203#endif
8204#ifdef _SC_XBS5_ILP32_OFF32
Victor Stinner8c62be82010-05-06 00:08:46 +00008205 {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
Fred Drakec9680921999-12-13 16:37:25 +00008206#endif
8207#ifdef _SC_XBS5_ILP32_OFFBIG
Victor Stinner8c62be82010-05-06 00:08:46 +00008208 {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
Fred Drakec9680921999-12-13 16:37:25 +00008209#endif
8210#ifdef _SC_XBS5_LP64_OFF64
Victor Stinner8c62be82010-05-06 00:08:46 +00008211 {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
Fred Drakec9680921999-12-13 16:37:25 +00008212#endif
8213#ifdef _SC_XBS5_LPBIG_OFFBIG
Victor Stinner8c62be82010-05-06 00:08:46 +00008214 {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
Fred Drakec9680921999-12-13 16:37:25 +00008215#endif
8216#ifdef _SC_XOPEN_CRYPT
Victor Stinner8c62be82010-05-06 00:08:46 +00008217 {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
Fred Drakec9680921999-12-13 16:37:25 +00008218#endif
8219#ifdef _SC_XOPEN_ENH_I18N
Victor Stinner8c62be82010-05-06 00:08:46 +00008220 {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
Fred Drakec9680921999-12-13 16:37:25 +00008221#endif
8222#ifdef _SC_XOPEN_LEGACY
Victor Stinner8c62be82010-05-06 00:08:46 +00008223 {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
Fred Drakec9680921999-12-13 16:37:25 +00008224#endif
8225#ifdef _SC_XOPEN_REALTIME
Victor Stinner8c62be82010-05-06 00:08:46 +00008226 {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
Fred Drakec9680921999-12-13 16:37:25 +00008227#endif
8228#ifdef _SC_XOPEN_REALTIME_THREADS
Victor Stinner8c62be82010-05-06 00:08:46 +00008229 {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
Fred Drakec9680921999-12-13 16:37:25 +00008230#endif
8231#ifdef _SC_XOPEN_SHM
Victor Stinner8c62be82010-05-06 00:08:46 +00008232 {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
Fred Drakec9680921999-12-13 16:37:25 +00008233#endif
8234#ifdef _SC_XOPEN_UNIX
Victor Stinner8c62be82010-05-06 00:08:46 +00008235 {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
Fred Drakec9680921999-12-13 16:37:25 +00008236#endif
8237#ifdef _SC_XOPEN_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008238 {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00008239#endif
8240#ifdef _SC_XOPEN_XCU_VERSION
Victor Stinner8c62be82010-05-06 00:08:46 +00008241 {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
Fred Drakec9680921999-12-13 16:37:25 +00008242#endif
8243#ifdef _SC_XOPEN_XPG2
Victor Stinner8c62be82010-05-06 00:08:46 +00008244 {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
Fred Drakec9680921999-12-13 16:37:25 +00008245#endif
8246#ifdef _SC_XOPEN_XPG3
Victor Stinner8c62be82010-05-06 00:08:46 +00008247 {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
Fred Drakec9680921999-12-13 16:37:25 +00008248#endif
8249#ifdef _SC_XOPEN_XPG4
Victor Stinner8c62be82010-05-06 00:08:46 +00008250 {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
Fred Drakec9680921999-12-13 16:37:25 +00008251#endif
8252};
8253
8254static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008255conv_sysconf_confname(PyObject *arg, int *valuep)
Fred Drakec9680921999-12-13 16:37:25 +00008256{
8257 return conv_confname(arg, valuep, posix_constants_sysconf,
8258 sizeof(posix_constants_sysconf)
8259 / sizeof(struct constdef));
8260}
8261
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008262PyDoc_STRVAR(posix_sysconf__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008263"sysconf(name) -> integer\n\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008264Return an integer-valued system configuration variable.");
Fred Drakec9680921999-12-13 16:37:25 +00008265
8266static PyObject *
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008267posix_sysconf(PyObject *self, PyObject *args)
Fred Drakec9680921999-12-13 16:37:25 +00008268{
8269 PyObject *result = NULL;
8270 int name;
8271
8272 if (PyArg_ParseTuple(args, "O&:sysconf", conv_sysconf_confname, &name)) {
8273 int value;
8274
8275 errno = 0;
8276 value = sysconf(name);
8277 if (value == -1 && errno != 0)
8278 posix_error();
8279 else
Christian Heimes217cfd12007-12-02 14:31:20 +00008280 result = PyLong_FromLong(value);
Fred Drakec9680921999-12-13 16:37:25 +00008281 }
8282 return result;
8283}
8284#endif
8285
8286
Fred Drakebec628d1999-12-15 18:31:10 +00008287/* This code is used to ensure that the tables of configuration value names
8288 * are in sorted order as required by conv_confname(), and also to build the
8289 * the exported dictionaries that are used to publish information about the
8290 * names available on the host platform.
8291 *
8292 * Sorting the table at runtime ensures that the table is properly ordered
8293 * when used, even for platforms we're not able to test on. It also makes
8294 * it easier to add additional entries to the tables.
Fred Draked86ed291999-12-15 15:34:33 +00008295 */
Fred Drakebec628d1999-12-15 18:31:10 +00008296
8297static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008298cmp_constdefs(const void *v1, const void *v2)
Fred Drakebec628d1999-12-15 18:31:10 +00008299{
8300 const struct constdef *c1 =
Victor Stinner8c62be82010-05-06 00:08:46 +00008301 (const struct constdef *) v1;
Fred Drakebec628d1999-12-15 18:31:10 +00008302 const struct constdef *c2 =
Victor Stinner8c62be82010-05-06 00:08:46 +00008303 (const struct constdef *) v2;
Fred Drakebec628d1999-12-15 18:31:10 +00008304
8305 return strcmp(c1->name, c2->name);
8306}
8307
8308static int
Fredrik Lundhff7df9d2000-07-08 22:48:53 +00008309setup_confname_table(struct constdef *table, size_t tablesize,
Victor Stinner8c62be82010-05-06 00:08:46 +00008310 char *tablename, PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00008311{
Fred Drakebec628d1999-12-15 18:31:10 +00008312 PyObject *d = NULL;
Barry Warsaw3155db32000-04-13 15:20:40 +00008313 size_t i;
Fred Drakebec628d1999-12-15 18:31:10 +00008314
8315 qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
8316 d = PyDict_New();
Barry Warsaw3155db32000-04-13 15:20:40 +00008317 if (d == NULL)
Victor Stinner8c62be82010-05-06 00:08:46 +00008318 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00008319
Barry Warsaw3155db32000-04-13 15:20:40 +00008320 for (i=0; i < tablesize; ++i) {
Victor Stinner8c62be82010-05-06 00:08:46 +00008321 PyObject *o = PyLong_FromLong(table[i].value);
8322 if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
8323 Py_XDECREF(o);
8324 Py_DECREF(d);
8325 return -1;
8326 }
8327 Py_DECREF(o);
Fred Draked86ed291999-12-15 15:34:33 +00008328 }
Fred Drake4d1e64b2002-04-15 19:40:07 +00008329 return PyModule_AddObject(module, tablename, d);
Fred Draked86ed291999-12-15 15:34:33 +00008330}
8331
Fred Drakebec628d1999-12-15 18:31:10 +00008332/* Return -1 on failure, 0 on success. */
8333static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00008334setup_confname_tables(PyObject *module)
Fred Draked86ed291999-12-15 15:34:33 +00008335{
8336#if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
Fred Drakebec628d1999-12-15 18:31:10 +00008337 if (setup_confname_table(posix_constants_pathconf,
Fred Draked86ed291999-12-15 15:34:33 +00008338 sizeof(posix_constants_pathconf)
8339 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00008340 "pathconf_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00008341 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00008342#endif
8343#ifdef HAVE_CONFSTR
Fred Drakebec628d1999-12-15 18:31:10 +00008344 if (setup_confname_table(posix_constants_confstr,
Fred Draked86ed291999-12-15 15:34:33 +00008345 sizeof(posix_constants_confstr)
8346 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00008347 "confstr_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00008348 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00008349#endif
8350#ifdef HAVE_SYSCONF
Fred Drakebec628d1999-12-15 18:31:10 +00008351 if (setup_confname_table(posix_constants_sysconf,
Fred Draked86ed291999-12-15 15:34:33 +00008352 sizeof(posix_constants_sysconf)
8353 / sizeof(struct constdef),
Fred Drake4d1e64b2002-04-15 19:40:07 +00008354 "sysconf_names", module))
Stefan Krah0e803b32010-11-26 16:16:47 +00008355 return -1;
Fred Draked86ed291999-12-15 15:34:33 +00008356#endif
Fred Drakebec628d1999-12-15 18:31:10 +00008357 return 0;
Fred Draked86ed291999-12-15 15:34:33 +00008358}
Fred Draked86ed291999-12-15 15:34:33 +00008359
8360
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008361PyDoc_STRVAR(posix_abort__doc__,
Fred Drakef7ce04d2002-06-20 18:31:21 +00008362"abort() -> does not return!\n\n\
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008363Abort the interpreter immediately. This 'dumps core' or otherwise fails\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008364in the hardest way possible on the hosting operating system.");
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008365
8366static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00008367posix_abort(PyObject *self, PyObject *noargs)
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008368{
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008369 abort();
8370 /*NOTREACHED*/
8371 Py_FatalError("abort() called from Python code didn't abort!");
8372 return NULL;
8373}
Fred Drakebec628d1999-12-15 18:31:10 +00008374
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00008375#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008376PyDoc_STRVAR(win32_startfile__doc__,
Georg Brandlf4f44152006-02-18 22:29:33 +00008377"startfile(filepath [, operation]) - Start a file with its associated\n\
8378application.\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00008379\n\
Georg Brandlf4f44152006-02-18 22:29:33 +00008380When \"operation\" is not specified or \"open\", this acts like\n\
8381double-clicking the file in Explorer, or giving the file name as an\n\
8382argument to the DOS \"start\" command: the file is opened with whatever\n\
8383application (if any) its extension is associated.\n\
8384When another \"operation\" is given, it specifies what should be done with\n\
8385the file. A typical operation is \"print\".\n\
Tim Petersf58a7aa2000-09-22 10:05:54 +00008386\n\
8387startfile returns as soon as the associated application is launched.\n\
8388There is no option to wait for the application to close, and no way\n\
8389to retrieve the application's exit status.\n\
8390\n\
8391The filepath is relative to the current directory. If you want to use\n\
8392an absolute path, make sure the first character is not a slash (\"/\");\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008393the underlying Win32 ShellExecute function doesn't work if it is.");
Tim Petersf58a7aa2000-09-22 10:05:54 +00008394
8395static PyObject *
8396win32_startfile(PyObject *self, PyObject *args)
8397{
Victor Stinner8c62be82010-05-06 00:08:46 +00008398 PyObject *ofilepath;
8399 char *filepath;
8400 char *operation = NULL;
8401 HINSTANCE rc;
Hirokazu Yamamoto8223c242009-05-17 04:21:53 +00008402
Victor Stinner8c62be82010-05-06 00:08:46 +00008403 PyObject *unipath, *woperation = NULL;
8404 if (!PyArg_ParseTuple(args, "U|s:startfile",
8405 &unipath, &operation)) {
8406 PyErr_Clear();
8407 goto normal;
8408 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00008409
Victor Stinner8c62be82010-05-06 00:08:46 +00008410 if (operation) {
8411 woperation = PyUnicode_DecodeASCII(operation,
8412 strlen(operation), NULL);
8413 if (!woperation) {
8414 PyErr_Clear();
8415 operation = NULL;
8416 goto normal;
8417 }
8418 }
Hirokazu Yamamoto892a37a2009-06-28 11:07:03 +00008419
Victor Stinner8c62be82010-05-06 00:08:46 +00008420 Py_BEGIN_ALLOW_THREADS
8421 rc = ShellExecuteW((HWND)0, woperation ? PyUnicode_AS_UNICODE(woperation) : 0,
8422 PyUnicode_AS_UNICODE(unipath),
8423 NULL, NULL, SW_SHOWNORMAL);
8424 Py_END_ALLOW_THREADS
8425
8426 Py_XDECREF(woperation);
8427 if (rc <= (HINSTANCE)32) {
8428 PyObject *errval = win32_error_unicode("startfile",
8429 PyUnicode_AS_UNICODE(unipath));
8430 return errval;
8431 }
8432 Py_INCREF(Py_None);
8433 return Py_None;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008434
8435normal:
Victor Stinner8c62be82010-05-06 00:08:46 +00008436 if (!PyArg_ParseTuple(args, "O&|s:startfile",
8437 PyUnicode_FSConverter, &ofilepath,
8438 &operation))
8439 return NULL;
8440 filepath = PyBytes_AsString(ofilepath);
8441 Py_BEGIN_ALLOW_THREADS
8442 rc = ShellExecute((HWND)0, operation, filepath,
8443 NULL, NULL, SW_SHOWNORMAL);
8444 Py_END_ALLOW_THREADS
8445 if (rc <= (HINSTANCE)32) {
8446 PyObject *errval = win32_error("startfile", filepath);
8447 Py_DECREF(ofilepath);
8448 return errval;
8449 }
8450 Py_DECREF(ofilepath);
8451 Py_INCREF(Py_None);
8452 return Py_None;
Tim Petersf58a7aa2000-09-22 10:05:54 +00008453}
8454#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00008455
Martin v. Löwis438b5342002-12-27 10:16:42 +00008456#ifdef HAVE_GETLOADAVG
8457PyDoc_STRVAR(posix_getloadavg__doc__,
8458"getloadavg() -> (float, float, float)\n\n\
8459Return the number of processes in the system run queue averaged over\n\
8460the last 1, 5, and 15 minutes or raises OSError if the load average\n\
8461was unobtainable");
8462
8463static PyObject *
Neal Norwitze241ce82003-02-17 18:17:05 +00008464posix_getloadavg(PyObject *self, PyObject *noargs)
Martin v. Löwis438b5342002-12-27 10:16:42 +00008465{
8466 double loadavg[3];
Martin v. Löwis438b5342002-12-27 10:16:42 +00008467 if (getloadavg(loadavg, 3)!=3) {
Stefan Krah0e803b32010-11-26 16:16:47 +00008468 PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
8469 return NULL;
Martin v. Löwis438b5342002-12-27 10:16:42 +00008470 } else
Stefan Krah0e803b32010-11-26 16:16:47 +00008471 return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
Martin v. Löwis438b5342002-12-27 10:16:42 +00008472}
8473#endif
8474
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008475#ifdef MS_WINDOWS
8476
8477PyDoc_STRVAR(win32_urandom__doc__,
8478"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00008479Return n random bytes suitable for cryptographic use.");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008480
8481typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
8482 LPCSTR pszContainer, LPCSTR pszProvider, DWORD dwProvType,\
8483 DWORD dwFlags );
8484typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV hProv, DWORD dwLen,\
8485 BYTE *pbBuffer );
8486
8487static CRYPTGENRANDOM pCryptGenRandom = NULL;
Thomas Wouters89d996e2007-09-08 17:39:28 +00008488/* This handle is never explicitly released. Instead, the operating
8489 system will release it when the process terminates. */
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008490static HCRYPTPROV hCryptProv = 0;
8491
Tim Peters4ad82172004-08-30 17:02:04 +00008492static PyObject*
8493win32_urandom(PyObject *self, PyObject *args)
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008494{
Victor Stinner8c62be82010-05-06 00:08:46 +00008495 int howMany;
8496 PyObject* result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008497
Victor Stinner8c62be82010-05-06 00:08:46 +00008498 /* Read arguments */
8499 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
8500 return NULL;
8501 if (howMany < 0)
8502 return PyErr_Format(PyExc_ValueError,
8503 "negative argument not allowed");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008504
Victor Stinner8c62be82010-05-06 00:08:46 +00008505 if (hCryptProv == 0) {
8506 HINSTANCE hAdvAPI32 = NULL;
8507 CRYPTACQUIRECONTEXTA pCryptAcquireContext = NULL;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008508
Victor Stinner8c62be82010-05-06 00:08:46 +00008509 /* Obtain handle to the DLL containing CryptoAPI
8510 This should not fail */
8511 hAdvAPI32 = GetModuleHandle("advapi32.dll");
8512 if(hAdvAPI32 == NULL)
8513 return win32_error("GetModuleHandle", NULL);
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008514
Victor Stinner8c62be82010-05-06 00:08:46 +00008515 /* Obtain pointers to the CryptoAPI functions
8516 This will fail on some early versions of Win95 */
8517 pCryptAcquireContext = (CRYPTACQUIRECONTEXTA)GetProcAddress(
8518 hAdvAPI32,
8519 "CryptAcquireContextA");
8520 if (pCryptAcquireContext == NULL)
8521 return PyErr_Format(PyExc_NotImplementedError,
8522 "CryptAcquireContextA not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008523
Victor Stinner8c62be82010-05-06 00:08:46 +00008524 pCryptGenRandom = (CRYPTGENRANDOM)GetProcAddress(
8525 hAdvAPI32, "CryptGenRandom");
8526 if (pCryptGenRandom == NULL)
8527 return PyErr_Format(PyExc_NotImplementedError,
8528 "CryptGenRandom not found");
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008529
Victor Stinner8c62be82010-05-06 00:08:46 +00008530 /* Acquire context */
8531 if (! pCryptAcquireContext(&hCryptProv, NULL, NULL,
8532 PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
8533 return win32_error("CryptAcquireContext", NULL);
8534 }
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008535
Victor Stinner8c62be82010-05-06 00:08:46 +00008536 /* Allocate bytes */
8537 result = PyBytes_FromStringAndSize(NULL, howMany);
8538 if (result != NULL) {
8539 /* Get random data */
8540 memset(PyBytes_AS_STRING(result), 0, howMany); /* zero seed */
8541 if (! pCryptGenRandom(hCryptProv, howMany, (unsigned char*)
8542 PyBytes_AS_STRING(result))) {
8543 Py_DECREF(result);
8544 return win32_error("CryptGenRandom", NULL);
8545 }
8546 }
8547 return result;
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00008548}
8549#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00008550
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00008551PyDoc_STRVAR(device_encoding__doc__,
8552"device_encoding(fd) -> str\n\n\
8553Return a string describing the encoding of the device\n\
8554if the output is a terminal; else return None.");
8555
8556static PyObject *
8557device_encoding(PyObject *self, PyObject *args)
8558{
Victor Stinner8c62be82010-05-06 00:08:46 +00008559 int fd;
Victor Stinner7870bdf2011-05-23 18:12:52 +02008560#if defined(MS_WINDOWS) || defined(MS_WIN64)
8561 UINT cp;
8562#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00008563 if (!PyArg_ParseTuple(args, "i:device_encoding", &fd))
8564 return NULL;
8565 if (!_PyVerify_fd(fd) || !isatty(fd)) {
8566 Py_INCREF(Py_None);
8567 return Py_None;
8568 }
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00008569#if defined(MS_WINDOWS) || defined(MS_WIN64)
Victor Stinner7870bdf2011-05-23 18:12:52 +02008570 if (fd == 0)
8571 cp = GetConsoleCP();
8572 else if (fd == 1 || fd == 2)
8573 cp = GetConsoleOutputCP();
8574 else
8575 cp = 0;
8576 /* GetConsoleCP() and GetConsoleOutputCP() return 0 if the application
8577 has no console */
8578 if (cp != 0)
8579 return PyUnicode_FromFormat("cp%u", (unsigned int)cp);
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00008580#elif defined(CODESET)
Victor Stinner8c62be82010-05-06 00:08:46 +00008581 {
8582 char *codeset = nl_langinfo(CODESET);
8583 if (codeset != NULL && codeset[0] != 0)
8584 return PyUnicode_FromString(codeset);
8585 }
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00008586#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00008587 Py_INCREF(Py_None);
8588 return Py_None;
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +00008589}
8590
Thomas Wouters0e3f5912006-08-11 14:57:12 +00008591#ifdef __VMS
8592/* Use openssl random routine */
8593#include <openssl/rand.h>
8594PyDoc_STRVAR(vms_urandom__doc__,
8595"urandom(n) -> str\n\n\
Neal Norwitz93c56822007-08-26 07:10:06 +00008596Return n random bytes suitable for cryptographic use.");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00008597
8598static PyObject*
8599vms_urandom(PyObject *self, PyObject *args)
8600{
Victor Stinner8c62be82010-05-06 00:08:46 +00008601 int howMany;
8602 PyObject* result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00008603
Victor Stinner8c62be82010-05-06 00:08:46 +00008604 /* Read arguments */
8605 if (! PyArg_ParseTuple(args, "i:urandom", &howMany))
8606 return NULL;
8607 if (howMany < 0)
8608 return PyErr_Format(PyExc_ValueError,
8609 "negative argument not allowed");
Thomas Wouters0e3f5912006-08-11 14:57:12 +00008610
Victor Stinner8c62be82010-05-06 00:08:46 +00008611 /* Allocate bytes */
8612 result = PyBytes_FromStringAndSize(NULL, howMany);
8613 if (result != NULL) {
8614 /* Get random data */
8615 if (RAND_pseudo_bytes((unsigned char*)
8616 PyBytes_AS_STRING(result),
8617 howMany) < 0) {
8618 Py_DECREF(result);
8619 return PyErr_Format(PyExc_ValueError,
8620 "RAND_pseudo_bytes");
8621 }
8622 }
8623 return result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00008624}
8625#endif
8626
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00008627#ifdef HAVE_SETRESUID
8628PyDoc_STRVAR(posix_setresuid__doc__,
8629"setresuid(ruid, euid, suid)\n\n\
8630Set the current process's real, effective, and saved user ids.");
8631
8632static PyObject*
8633posix_setresuid (PyObject *self, PyObject *args)
8634{
Victor Stinner8c62be82010-05-06 00:08:46 +00008635 /* We assume uid_t is no larger than a long. */
8636 long ruid, euid, suid;
8637 if (!PyArg_ParseTuple(args, "lll", &ruid, &euid, &suid))
8638 return NULL;
8639 if (setresuid(ruid, euid, suid) < 0)
8640 return posix_error();
8641 Py_RETURN_NONE;
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00008642}
8643#endif
8644
8645#ifdef HAVE_SETRESGID
8646PyDoc_STRVAR(posix_setresgid__doc__,
8647"setresgid(rgid, egid, sgid)\n\n\
8648Set the current process's real, effective, and saved group ids.");
8649
8650static PyObject*
8651posix_setresgid (PyObject *self, PyObject *args)
8652{
Victor Stinner8c62be82010-05-06 00:08:46 +00008653 /* We assume uid_t is no larger than a long. */
8654 long rgid, egid, sgid;
8655 if (!PyArg_ParseTuple(args, "lll", &rgid, &egid, &sgid))
8656 return NULL;
8657 if (setresgid(rgid, egid, sgid) < 0)
8658 return posix_error();
8659 Py_RETURN_NONE;
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00008660}
8661#endif
8662
8663#ifdef HAVE_GETRESUID
8664PyDoc_STRVAR(posix_getresuid__doc__,
8665"getresuid() -> (ruid, euid, suid)\n\n\
8666Get tuple of the current process's real, effective, and saved user ids.");
8667
8668static PyObject*
8669posix_getresuid (PyObject *self, PyObject *noargs)
8670{
Victor Stinner8c62be82010-05-06 00:08:46 +00008671 uid_t ruid, euid, suid;
8672 long l_ruid, l_euid, l_suid;
8673 if (getresuid(&ruid, &euid, &suid) < 0)
8674 return posix_error();
8675 /* Force the values into long's as we don't know the size of uid_t. */
8676 l_ruid = ruid;
8677 l_euid = euid;
8678 l_suid = suid;
8679 return Py_BuildValue("(lll)", l_ruid, l_euid, l_suid);
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00008680}
8681#endif
8682
8683#ifdef HAVE_GETRESGID
8684PyDoc_STRVAR(posix_getresgid__doc__,
8685"getresgid() -> (rgid, egid, sgid)\n\n\
Georg Brandla9b51d22010-09-05 17:07:12 +00008686Get tuple of the current process's real, effective, and saved group ids.");
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00008687
8688static PyObject*
8689posix_getresgid (PyObject *self, PyObject *noargs)
8690{
Victor Stinner8c62be82010-05-06 00:08:46 +00008691 uid_t rgid, egid, sgid;
8692 long l_rgid, l_egid, l_sgid;
8693 if (getresgid(&rgid, &egid, &sgid) < 0)
8694 return posix_error();
8695 /* Force the values into long's as we don't know the size of uid_t. */
8696 l_rgid = rgid;
8697 l_egid = egid;
8698 l_sgid = sgid;
8699 return Py_BuildValue("(lll)", l_rgid, l_egid, l_sgid);
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00008700}
8701#endif
8702
Antoine Pitrouf65132d2011-02-25 23:25:17 +00008703/* Posix *at family of functions:
8704 faccessat, fchmodat, fchownat, fstatat, futimesat,
8705 linkat, mkdirat, mknodat, openat, readlinkat, renameat, symlinkat,
8706 unlinkat, utimensat, mkfifoat */
8707
8708#ifdef HAVE_FACCESSAT
8709PyDoc_STRVAR(posix_faccessat__doc__,
8710"faccessat(dirfd, path, mode, flags=0) -> True if granted, False otherwise\n\n\
8711Like access() but if path is relative, it is taken as relative to dirfd.\n\
8712flags is optional and can be constructed by ORing together zero or more\n\
8713of these values: AT_SYMLINK_NOFOLLOW, AT_EACCESS.\n\
8714If path is relative and dirfd is the special value AT_FDCWD, then path\n\
8715is interpreted relative to the current working directory.");
8716
8717static PyObject *
8718posix_faccessat(PyObject *self, PyObject *args)
8719{
8720 PyObject *opath;
8721 char *path;
8722 int mode;
8723 int res;
8724 int dirfd, flags = 0;
8725 if (!PyArg_ParseTuple(args, "iO&i|i:faccessat",
8726 &dirfd, PyUnicode_FSConverter, &opath, &mode, &flags))
8727 return NULL;
8728 path = PyBytes_AsString(opath);
8729 Py_BEGIN_ALLOW_THREADS
8730 res = faccessat(dirfd, path, mode, flags);
8731 Py_END_ALLOW_THREADS
8732 Py_DECREF(opath);
8733 return PyBool_FromLong(res == 0);
8734}
8735#endif
8736
8737#ifdef HAVE_FCHMODAT
8738PyDoc_STRVAR(posix_fchmodat__doc__,
8739"fchmodat(dirfd, path, mode, flags=0)\n\n\
8740Like chmod() but if path is relative, it is taken as relative to dirfd.\n\
8741flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
8742If path is relative and dirfd is the special value AT_FDCWD, then path\n\
8743is interpreted relative to the current working directory.");
8744
8745static PyObject *
8746posix_fchmodat(PyObject *self, PyObject *args)
8747{
8748 int dirfd, mode, res;
8749 int flags = 0;
8750 PyObject *opath;
8751 char *path;
8752
8753 if (!PyArg_ParseTuple(args, "iO&i|i:fchmodat",
8754 &dirfd, PyUnicode_FSConverter, &opath, &mode, &flags))
8755 return NULL;
8756
8757 path = PyBytes_AsString(opath);
8758
8759 Py_BEGIN_ALLOW_THREADS
8760 res = fchmodat(dirfd, path, mode, flags);
8761 Py_END_ALLOW_THREADS
8762 Py_DECREF(opath);
8763 if (res < 0)
8764 return posix_error();
8765 Py_RETURN_NONE;
8766}
8767#endif /* HAVE_FCHMODAT */
8768
8769#ifdef HAVE_FCHOWNAT
8770PyDoc_STRVAR(posix_fchownat__doc__,
8771"fchownat(dirfd, path, uid, gid, flags=0)\n\n\
8772Like chown() but if path is relative, it is taken as relative to dirfd.\n\
8773flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
8774If path is relative and dirfd is the special value AT_FDCWD, then path\n\
8775is interpreted relative to the current working directory.");
8776
8777static PyObject *
8778posix_fchownat(PyObject *self, PyObject *args)
8779{
8780 PyObject *opath;
8781 int dirfd, res;
8782 long uid, gid;
8783 int flags = 0;
8784 char *path;
Giampaolo Rodola'ff1a7352011-04-19 09:47:16 +02008785
Antoine Pitrouf65132d2011-02-25 23:25:17 +00008786 if (!PyArg_ParseTuple(args, "iO&ll|i:fchownat",
8787 &dirfd, PyUnicode_FSConverter, &opath, &uid, &gid, &flags))
8788 return NULL;
8789
8790 path = PyBytes_AsString(opath);
8791
8792 Py_BEGIN_ALLOW_THREADS
8793 res = fchownat(dirfd, path, (uid_t) uid, (gid_t) gid, flags);
8794 Py_END_ALLOW_THREADS
8795 Py_DECREF(opath);
8796 if (res < 0)
8797 return posix_error();
8798 Py_RETURN_NONE;
8799}
8800#endif /* HAVE_FCHOWNAT */
8801
8802#ifdef HAVE_FSTATAT
8803PyDoc_STRVAR(posix_fstatat__doc__,
8804"fstatat(dirfd, path, flags=0) -> stat result\n\n\
8805Like stat() but if path is relative, it is taken as relative to dirfd.\n\
8806flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
8807If path is relative and dirfd is the special value AT_FDCWD, then path\n\
8808is interpreted relative to the current working directory.");
8809
8810static PyObject *
8811posix_fstatat(PyObject *self, PyObject *args)
8812{
8813 PyObject *opath;
8814 char *path;
8815 STRUCT_STAT st;
8816 int dirfd, res, flags = 0;
8817
8818 if (!PyArg_ParseTuple(args, "iO&|i:fstatat",
8819 &dirfd, PyUnicode_FSConverter, &opath, &flags))
8820 return NULL;
8821 path = PyBytes_AsString(opath);
8822
8823 Py_BEGIN_ALLOW_THREADS
8824 res = fstatat(dirfd, path, &st, flags);
8825 Py_END_ALLOW_THREADS
8826 Py_DECREF(opath);
8827 if (res != 0)
8828 return posix_error();
8829
8830 return _pystat_fromstructstat(&st);
8831}
8832#endif
8833
8834#ifdef HAVE_FUTIMESAT
8835PyDoc_STRVAR(posix_futimesat__doc__,
8836"futimesat(dirfd, path, (atime, mtime))\n\
8837futimesat(dirfd, path, None)\n\n\
8838Like utime() but if path is relative, it is taken as relative to dirfd.\n\
8839If path is relative and dirfd is the special value AT_FDCWD, then path\n\
8840is interpreted relative to the current working directory.");
8841
8842static PyObject *
8843posix_futimesat(PyObject *self, PyObject *args)
8844{
8845 PyObject *opath;
8846 char *path;
8847 int res, dirfd;
8848 PyObject* arg;
8849
8850 struct timeval buf[2];
8851
8852 if (!PyArg_ParseTuple(args, "iO&O:futimesat",
8853 &dirfd, PyUnicode_FSConverter, &opath, &arg))
8854 return NULL;
8855 path = PyBytes_AsString(opath);
8856 if (arg == Py_None) {
8857 /* optional time values not given */
8858 Py_BEGIN_ALLOW_THREADS
8859 res = futimesat(dirfd, path, NULL);
8860 Py_END_ALLOW_THREADS
8861 }
8862 else if (!PyTuple_Check(arg) || PyTuple_Size(arg) != 2) {
8863 PyErr_SetString(PyExc_TypeError,
8864 "futimesat() arg 3 must be a tuple (atime, mtime)");
8865 Py_DECREF(opath);
8866 return NULL;
8867 }
8868 else {
8869 if (extract_time(PyTuple_GET_ITEM(arg, 0),
8870 &(buf[0].tv_sec), &(buf[0].tv_usec)) == -1) {
8871 Py_DECREF(opath);
8872 return NULL;
8873 }
8874 if (extract_time(PyTuple_GET_ITEM(arg, 1),
8875 &(buf[1].tv_sec), &(buf[1].tv_usec)) == -1) {
8876 Py_DECREF(opath);
8877 return NULL;
8878 }
8879 Py_BEGIN_ALLOW_THREADS
8880 res = futimesat(dirfd, path, buf);
8881 Py_END_ALLOW_THREADS
8882 }
8883 Py_DECREF(opath);
8884 if (res < 0) {
8885 return posix_error();
8886 }
8887 Py_RETURN_NONE;
8888}
8889#endif
8890
8891#ifdef HAVE_LINKAT
8892PyDoc_STRVAR(posix_linkat__doc__,
8893"linkat(srcfd, srcpath, dstfd, dstpath, flags=0)\n\n\
8894Like link() but if srcpath is relative, it is taken as relative to srcfd\n\
8895and if dstpath is relative, it is taken as relative to dstfd.\n\
8896flags is optional and may be 0 or AT_SYMLINK_FOLLOW.\n\
8897If srcpath is relative and srcfd is the special value AT_FDCWD, then\n\
8898srcpath is interpreted relative to the current working directory. This\n\
8899also applies for dstpath.");
8900
8901static PyObject *
8902posix_linkat(PyObject *self, PyObject *args)
8903{
8904 PyObject *osrc, *odst;
8905 char *src, *dst;
8906 int res, srcfd, dstfd;
8907 int flags = 0;
8908
8909 if (!PyArg_ParseTuple(args, "iO&iO&|i:linkat",
8910 &srcfd, PyUnicode_FSConverter, &osrc, &dstfd, PyUnicode_FSConverter, &odst, &flags))
8911 return NULL;
8912 src = PyBytes_AsString(osrc);
8913 dst = PyBytes_AsString(odst);
8914 Py_BEGIN_ALLOW_THREADS
8915 res = linkat(srcfd, src, dstfd, dst, flags);
8916 Py_END_ALLOW_THREADS
8917 Py_DECREF(osrc);
8918 Py_DECREF(odst);
8919 if (res < 0)
8920 return posix_error();
8921 Py_RETURN_NONE;
8922}
8923#endif /* HAVE_LINKAT */
8924
8925#ifdef HAVE_MKDIRAT
8926PyDoc_STRVAR(posix_mkdirat__doc__,
8927"mkdirat(dirfd, path, mode=0o777)\n\n\
8928Like mkdir() but if path is relative, it is taken as relative to dirfd.\n\
8929If path is relative and dirfd is the special value AT_FDCWD, then path\n\
8930is interpreted relative to the current working directory.");
8931
8932static PyObject *
8933posix_mkdirat(PyObject *self, PyObject *args)
8934{
8935 int res, dirfd;
8936 PyObject *opath;
8937 char *path;
8938 int mode = 0777;
8939
8940 if (!PyArg_ParseTuple(args, "iO&|i:mkdirat",
8941 &dirfd, PyUnicode_FSConverter, &opath, &mode))
8942 return NULL;
8943 path = PyBytes_AsString(opath);
8944 Py_BEGIN_ALLOW_THREADS
8945 res = mkdirat(dirfd, path, mode);
8946 Py_END_ALLOW_THREADS
8947 Py_DECREF(opath);
8948 if (res < 0)
8949 return posix_error();
8950 Py_RETURN_NONE;
8951}
8952#endif
8953
8954#if defined(HAVE_MKNODAT) && defined(HAVE_MAKEDEV)
8955PyDoc_STRVAR(posix_mknodat__doc__,
8956"mknodat(dirfd, path, mode=0o600, device=0)\n\n\
8957Like mknod() but if path is relative, it is taken as relative to dirfd.\n\
8958If path is relative and dirfd is the special value AT_FDCWD, then path\n\
8959is interpreted relative to the current working directory.");
8960
8961static PyObject *
8962posix_mknodat(PyObject *self, PyObject *args)
8963{
8964 PyObject *opath;
8965 char *filename;
8966 int mode = 0600;
8967 int device = 0;
8968 int res, dirfd;
8969 if (!PyArg_ParseTuple(args, "iO&|ii:mknodat", &dirfd,
8970 PyUnicode_FSConverter, &opath, &mode, &device))
8971 return NULL;
8972 filename = PyBytes_AS_STRING(opath);
8973 Py_BEGIN_ALLOW_THREADS
8974 res = mknodat(dirfd, filename, mode, device);
8975 Py_END_ALLOW_THREADS
8976 Py_DECREF(opath);
8977 if (res < 0)
8978 return posix_error();
8979 Py_RETURN_NONE;
8980}
8981#endif
8982
8983#ifdef HAVE_OPENAT
8984PyDoc_STRVAR(posix_openat__doc__,
8985"openat(dirfd, path, flag, mode=0o777) -> fd\n\n\
8986Like open() but if path is relative, it is taken as relative to dirfd.\n\
8987If path is relative and dirfd is the special value AT_FDCWD, then path\n\
8988is interpreted relative to the current working directory.");
8989
8990static PyObject *
8991posix_openat(PyObject *self, PyObject *args)
8992{
8993 PyObject *ofile;
8994 char *file;
8995 int flag, dirfd, fd;
8996 int mode = 0777;
8997
8998 if (!PyArg_ParseTuple(args, "iO&i|i:openat",
8999 &dirfd, PyUnicode_FSConverter, &ofile,
9000 &flag, &mode))
9001 return NULL;
9002 file = PyBytes_AsString(ofile);
9003 Py_BEGIN_ALLOW_THREADS
9004 fd = openat(dirfd, file, flag, mode);
9005 Py_END_ALLOW_THREADS
9006 Py_DECREF(ofile);
9007 if (fd < 0)
9008 return posix_error();
9009 return PyLong_FromLong((long)fd);
9010}
9011#endif
9012
9013#ifdef HAVE_READLINKAT
9014PyDoc_STRVAR(posix_readlinkat__doc__,
9015"readlinkat(dirfd, path) -> path\n\n\
9016Like readlink() but if path is relative, it is taken as relative to dirfd.\n\
9017If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9018is interpreted relative to the current working directory.");
9019
9020static PyObject *
9021posix_readlinkat(PyObject *self, PyObject *args)
9022{
9023 PyObject *v, *opath;
9024 char buf[MAXPATHLEN];
9025 char *path;
9026 int n, dirfd;
9027 int arg_is_unicode = 0;
9028
9029 if (!PyArg_ParseTuple(args, "iO&:readlinkat",
9030 &dirfd, PyUnicode_FSConverter, &opath))
9031 return NULL;
9032 path = PyBytes_AsString(opath);
9033 v = PySequence_GetItem(args, 1);
9034 if (v == NULL) {
9035 Py_DECREF(opath);
9036 return NULL;
9037 }
9038
9039 if (PyUnicode_Check(v)) {
9040 arg_is_unicode = 1;
9041 }
9042 Py_DECREF(v);
9043
9044 Py_BEGIN_ALLOW_THREADS
9045 n = readlinkat(dirfd, path, buf, (int) sizeof buf);
9046 Py_END_ALLOW_THREADS
9047 Py_DECREF(opath);
9048 if (n < 0)
9049 return posix_error();
9050
9051 if (arg_is_unicode)
9052 return PyUnicode_DecodeFSDefaultAndSize(buf, n);
9053 else
9054 return PyBytes_FromStringAndSize(buf, n);
9055}
9056#endif /* HAVE_READLINKAT */
9057
9058#ifdef HAVE_RENAMEAT
9059PyDoc_STRVAR(posix_renameat__doc__,
9060"renameat(olddirfd, oldpath, newdirfd, newpath)\n\n\
9061Like rename() but if oldpath is relative, it is taken as relative to\n\
9062olddirfd and if newpath is relative, it is taken as relative to newdirfd.\n\
9063If oldpath is relative and olddirfd is the special value AT_FDCWD, then\n\
9064oldpath is interpreted relative to the current working directory. This\n\
9065also applies for newpath.");
9066
9067static PyObject *
9068posix_renameat(PyObject *self, PyObject *args)
9069{
9070 int res;
9071 PyObject *opathold, *opathnew;
9072 char *opath, *npath;
9073 int oldfd, newfd;
9074
9075 if (!PyArg_ParseTuple(args, "iO&iO&:renameat",
9076 &oldfd, PyUnicode_FSConverter, &opathold, &newfd, PyUnicode_FSConverter, &opathnew))
9077 return NULL;
9078 opath = PyBytes_AsString(opathold);
9079 npath = PyBytes_AsString(opathnew);
9080 Py_BEGIN_ALLOW_THREADS
9081 res = renameat(oldfd, opath, newfd, npath);
9082 Py_END_ALLOW_THREADS
9083 Py_DECREF(opathold);
9084 Py_DECREF(opathnew);
9085 if (res < 0)
9086 return posix_error();
9087 Py_RETURN_NONE;
9088}
9089#endif
9090
9091#if HAVE_SYMLINKAT
9092PyDoc_STRVAR(posix_symlinkat__doc__,
9093"symlinkat(src, dstfd, dst)\n\n\
9094Like symlink() but if dst is relative, it is taken as relative to dstfd.\n\
9095If dst is relative and dstfd is the special value AT_FDCWD, then dst\n\
9096is interpreted relative to the current working directory.");
9097
9098static PyObject *
9099posix_symlinkat(PyObject *self, PyObject *args)
9100{
9101 int res, dstfd;
9102 PyObject *osrc, *odst;
9103 char *src, *dst;
9104
9105 if (!PyArg_ParseTuple(args, "O&iO&:symlinkat",
9106 PyUnicode_FSConverter, &osrc, &dstfd, PyUnicode_FSConverter, &odst))
9107 return NULL;
9108 src = PyBytes_AsString(osrc);
9109 dst = PyBytes_AsString(odst);
9110 Py_BEGIN_ALLOW_THREADS
9111 res = symlinkat(src, dstfd, dst);
9112 Py_END_ALLOW_THREADS
9113 Py_DECREF(osrc);
9114 Py_DECREF(odst);
9115 if (res < 0)
9116 return posix_error();
9117 Py_RETURN_NONE;
9118}
9119#endif /* HAVE_SYMLINKAT */
9120
9121#ifdef HAVE_UNLINKAT
9122PyDoc_STRVAR(posix_unlinkat__doc__,
9123"unlinkat(dirfd, path, flags=0)\n\n\
9124Like unlink() but if path is relative, it is taken as relative to dirfd.\n\
9125flags is optional and may be 0 or AT_REMOVEDIR. If AT_REMOVEDIR is\n\
9126specified, unlinkat() behaves like rmdir().\n\
9127If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9128is interpreted relative to the current working directory.");
9129
9130static PyObject *
9131posix_unlinkat(PyObject *self, PyObject *args)
9132{
9133 int dirfd, res, flags = 0;
9134 PyObject *opath;
9135 char *path;
9136
9137 if (!PyArg_ParseTuple(args, "iO&|i:unlinkat",
9138 &dirfd, PyUnicode_FSConverter, &opath, &flags))
9139 return NULL;
9140 path = PyBytes_AsString(opath);
9141 Py_BEGIN_ALLOW_THREADS
9142 res = unlinkat(dirfd, path, flags);
9143 Py_END_ALLOW_THREADS
9144 Py_DECREF(opath);
9145 if (res < 0)
9146 return posix_error();
9147 Py_RETURN_NONE;
9148}
9149#endif
9150
9151#ifdef HAVE_UTIMENSAT
9152PyDoc_STRVAR(posix_utimensat__doc__,
9153"utimensat(dirfd, path, (atime_sec, atime_nsec),\n\
9154 (mtime_sec, mtime_nsec), flags)\n\
9155utimensat(dirfd, path, None, None, flags)\n\n\
9156Updates the timestamps of a file with nanosecond precision. If path is\n\
9157relative, it is taken as relative to dirfd.\n\
9158The second form sets atime and mtime to the current time.\n\
9159flags is optional and may be 0 or AT_SYMLINK_NOFOLLOW.\n\
9160If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9161is interpreted relative to the current working directory.\n\
9162If *_nsec is specified as UTIME_NOW, the timestamp is updated to the\n\
9163current time.\n\
9164If *_nsec is specified as UTIME_OMIT, the timestamp is not updated.");
9165
9166static PyObject *
9167posix_utimensat(PyObject *self, PyObject *args)
9168{
9169 PyObject *opath;
9170 char *path;
9171 int res, dirfd, flags = 0;
9172 PyObject *atime, *mtime;
9173
9174 struct timespec buf[2];
9175
9176 if (!PyArg_ParseTuple(args, "iO&OO|i:utimensat",
9177 &dirfd, PyUnicode_FSConverter, &opath, &atime, &mtime, &flags))
9178 return NULL;
9179 path = PyBytes_AsString(opath);
9180 if (atime == Py_None && mtime == Py_None) {
9181 /* optional time values not given */
9182 Py_BEGIN_ALLOW_THREADS
9183 res = utimensat(dirfd, path, NULL, flags);
9184 Py_END_ALLOW_THREADS
9185 }
9186 else if (!PyTuple_Check(atime) || PyTuple_Size(atime) != 2) {
9187 PyErr_SetString(PyExc_TypeError,
9188 "utimensat() arg 3 must be a tuple (atime_sec, atime_nsec)");
9189 Py_DECREF(opath);
9190 return NULL;
9191 }
9192 else if (!PyTuple_Check(mtime) || PyTuple_Size(mtime) != 2) {
9193 PyErr_SetString(PyExc_TypeError,
9194 "utimensat() arg 4 must be a tuple (mtime_sec, mtime_nsec)");
9195 Py_DECREF(opath);
9196 return NULL;
9197 }
9198 else {
9199 if (!PyArg_ParseTuple(atime, "ll:utimensat",
9200 &(buf[0].tv_sec), &(buf[0].tv_nsec))) {
9201 Py_DECREF(opath);
9202 return NULL;
9203 }
9204 if (!PyArg_ParseTuple(mtime, "ll:utimensat",
9205 &(buf[1].tv_sec), &(buf[1].tv_nsec))) {
9206 Py_DECREF(opath);
9207 return NULL;
9208 }
9209 Py_BEGIN_ALLOW_THREADS
9210 res = utimensat(dirfd, path, buf, flags);
9211 Py_END_ALLOW_THREADS
9212 }
9213 Py_DECREF(opath);
9214 if (res < 0) {
9215 return posix_error();
9216 }
9217 Py_RETURN_NONE;
9218}
9219#endif
9220
9221#ifdef HAVE_MKFIFOAT
9222PyDoc_STRVAR(posix_mkfifoat__doc__,
9223"mkfifoat(dirfd, path, mode=0o666)\n\n\
9224Like mkfifo() but if path is relative, it is taken as relative to dirfd.\n\
9225If path is relative and dirfd is the special value AT_FDCWD, then path\n\
9226is interpreted relative to the current working directory.");
9227
9228static PyObject *
9229posix_mkfifoat(PyObject *self, PyObject *args)
9230{
9231 PyObject *opath;
9232 char *filename;
9233 int mode = 0666;
9234 int res, dirfd;
9235 if (!PyArg_ParseTuple(args, "iO&|i:mkfifoat",
9236 &dirfd, PyUnicode_FSConverter, &opath, &mode))
9237 return NULL;
9238 filename = PyBytes_AS_STRING(opath);
9239 Py_BEGIN_ALLOW_THREADS
9240 res = mkfifoat(dirfd, filename, mode);
9241 Py_END_ALLOW_THREADS
9242 Py_DECREF(opath);
9243 if (res < 0)
9244 return posix_error();
9245 Py_RETURN_NONE;
9246}
9247#endif
9248
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009249static PyMethodDef posix_methods[] = {
Victor Stinner8c62be82010-05-06 00:08:46 +00009250 {"access", posix_access, METH_VARARGS, posix_access__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009251#ifdef HAVE_TTYNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00009252 {"ttyname", posix_ttyname, METH_VARARGS, posix_ttyname__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009253#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009254 {"chdir", posix_chdir, METH_VARARGS, posix_chdir__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +00009255#ifdef HAVE_CHFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00009256 {"chflags", posix_chflags, METH_VARARGS, posix_chflags__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +00009257#endif /* HAVE_CHFLAGS */
Victor Stinner8c62be82010-05-06 00:08:46 +00009258 {"chmod", posix_chmod, METH_VARARGS, posix_chmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +00009259#ifdef HAVE_FCHMOD
Victor Stinner8c62be82010-05-06 00:08:46 +00009260 {"fchmod", posix_fchmod, METH_VARARGS, posix_fchmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +00009261#endif /* HAVE_FCHMOD */
Guido van Rossumb6775db1994-08-01 11:34:53 +00009262#ifdef HAVE_CHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +00009263 {"chown", posix_chown, METH_VARARGS, posix_chown__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009264#endif /* HAVE_CHOWN */
Christian Heimes4e30a842007-11-30 22:12:06 +00009265#ifdef HAVE_LCHMOD
Victor Stinner8c62be82010-05-06 00:08:46 +00009266 {"lchmod", posix_lchmod, METH_VARARGS, posix_lchmod__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +00009267#endif /* HAVE_LCHMOD */
9268#ifdef HAVE_FCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +00009269 {"fchown", posix_fchown, METH_VARARGS, posix_fchown__doc__},
Christian Heimes4e30a842007-11-30 22:12:06 +00009270#endif /* HAVE_FCHOWN */
Thomas Wouterscf297e42007-02-23 15:07:44 +00009271#ifdef HAVE_LCHFLAGS
Victor Stinner8c62be82010-05-06 00:08:46 +00009272 {"lchflags", posix_lchflags, METH_VARARGS, posix_lchflags__doc__},
Thomas Wouterscf297e42007-02-23 15:07:44 +00009273#endif /* HAVE_LCHFLAGS */
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00009274#ifdef HAVE_LCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +00009275 {"lchown", posix_lchown, METH_VARARGS, posix_lchown__doc__},
Martin v. Löwis0cec0ff2002-07-28 16:33:45 +00009276#endif /* HAVE_LCHOWN */
Martin v. Löwis244edc82001-10-04 22:44:26 +00009277#ifdef HAVE_CHROOT
Victor Stinner8c62be82010-05-06 00:08:46 +00009278 {"chroot", posix_chroot, METH_VARARGS, posix_chroot__doc__},
Martin v. Löwis244edc82001-10-04 22:44:26 +00009279#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009280#ifdef HAVE_CTERMID
Victor Stinner8c62be82010-05-06 00:08:46 +00009281 {"ctermid", posix_ctermid, METH_NOARGS, posix_ctermid__doc__},
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009282#endif
Guido van Rossum36bc6801995-06-14 22:54:23 +00009283#ifdef HAVE_GETCWD
Victor Stinner8c62be82010-05-06 00:08:46 +00009284 {"getcwd", (PyCFunction)posix_getcwd_unicode,
9285 METH_NOARGS, posix_getcwd__doc__},
9286 {"getcwdb", (PyCFunction)posix_getcwd_bytes,
9287 METH_NOARGS, posix_getcwdb__doc__},
Guido van Rossum36bc6801995-06-14 22:54:23 +00009288#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00009289#ifdef HAVE_LINK
Victor Stinner8c62be82010-05-06 00:08:46 +00009290 {"link", posix_link, METH_VARARGS, posix_link__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009291#endif /* HAVE_LINK */
Victor Stinner8c62be82010-05-06 00:08:46 +00009292 {"listdir", posix_listdir, METH_VARARGS, posix_listdir__doc__},
Antoine Pitrou8250e232011-02-25 23:41:16 +00009293#ifdef HAVE_FDOPENDIR
9294 {"fdlistdir", posix_fdlistdir, METH_VARARGS, posix_fdlistdir__doc__},
9295#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009296 {"lstat", posix_lstat, METH_VARARGS, posix_lstat__doc__},
9297 {"mkdir", posix_mkdir, METH_VARARGS, posix_mkdir__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00009298#ifdef HAVE_NICE
Victor Stinner8c62be82010-05-06 00:08:46 +00009299 {"nice", posix_nice, METH_VARARGS, posix_nice__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009300#endif /* HAVE_NICE */
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00009301#ifdef HAVE_GETPRIORITY
9302 {"getpriority", posix_getpriority, METH_VARARGS, posix_getpriority__doc__},
9303#endif /* HAVE_GETPRIORITY */
9304#ifdef HAVE_SETPRIORITY
9305 {"setpriority", posix_setpriority, METH_VARARGS, posix_setpriority__doc__},
9306#endif /* HAVE_SETPRIORITY */
Guido van Rossumb6775db1994-08-01 11:34:53 +00009307#ifdef HAVE_READLINK
Victor Stinner8c62be82010-05-06 00:08:46 +00009308 {"readlink", posix_readlink, METH_VARARGS, posix_readlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009309#endif /* HAVE_READLINK */
Brian Curtind40e6f72010-07-08 21:39:08 +00009310#if !defined(HAVE_READLINK) && defined(MS_WINDOWS)
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +00009311 {"readlink", win_readlink, METH_VARARGS, win_readlink__doc__},
Brian Curtind40e6f72010-07-08 21:39:08 +00009312#endif /* !defined(HAVE_READLINK) && defined(MS_WINDOWS) */
Amaury Forgeot d'Arc844807e2010-08-16 22:16:51 +00009313 {"rename", posix_rename, METH_VARARGS, posix_rename__doc__},
9314 {"rmdir", posix_rmdir, METH_VARARGS, posix_rmdir__doc__},
9315 {"stat", posix_stat, METH_VARARGS, posix_stat__doc__},
Victor Stinner8c62be82010-05-06 00:08:46 +00009316 {"stat_float_times", stat_float_times, METH_VARARGS, stat_float_times__doc__},
Brian Curtin52173d42010-12-02 18:29:18 +00009317#if defined(HAVE_SYMLINK) && !defined(MS_WINDOWS)
Victor Stinner8c62be82010-05-06 00:08:46 +00009318 {"symlink", posix_symlink, METH_VARARGS, posix_symlink__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009319#endif /* HAVE_SYMLINK */
Brian Curtin52173d42010-12-02 18:29:18 +00009320#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +00009321 {"symlink", (PyCFunction)win_symlink, METH_VARARGS | METH_KEYWORDS,
Brian Curtin52173d42010-12-02 18:29:18 +00009322 win_symlink__doc__},
9323#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009324#ifdef HAVE_SYSTEM
Victor Stinner8c62be82010-05-06 00:08:46 +00009325 {"system", posix_system, METH_VARARGS, posix_system__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009326#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009327 {"umask", posix_umask, METH_VARARGS, posix_umask__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00009328#ifdef HAVE_UNAME
Victor Stinner8c62be82010-05-06 00:08:46 +00009329 {"uname", posix_uname, METH_NOARGS, posix_uname__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009330#endif /* HAVE_UNAME */
Victor Stinner8c62be82010-05-06 00:08:46 +00009331 {"unlink", posix_unlink, METH_VARARGS, posix_unlink__doc__},
9332 {"remove", posix_unlink, METH_VARARGS, posix_remove__doc__},
9333 {"utime", posix_utime, METH_VARARGS, posix_utime__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +02009334#ifdef HAVE_FUTIMES
9335 {"futimes", posix_futimes, METH_VARARGS, posix_futimes__doc__},
9336#endif
9337#ifdef HAVE_LUTIMES
9338 {"lutimes", posix_lutimes, METH_VARARGS, posix_lutimes__doc__},
9339#endif
9340#ifdef HAVE_FUTIMENS
9341 {"futimens", posix_futimens, METH_VARARGS, posix_futimens__doc__},
9342#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00009343#ifdef HAVE_TIMES
Victor Stinner8c62be82010-05-06 00:08:46 +00009344 {"times", posix_times, METH_NOARGS, posix_times__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009345#endif /* HAVE_TIMES */
Victor Stinner8c62be82010-05-06 00:08:46 +00009346 {"_exit", posix__exit, METH_VARARGS, posix__exit__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009347#ifdef HAVE_EXECV
Victor Stinner8c62be82010-05-06 00:08:46 +00009348 {"execv", posix_execv, METH_VARARGS, posix_execv__doc__},
9349 {"execve", posix_execve, METH_VARARGS, posix_execve__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009350#endif /* HAVE_EXECV */
Ross Lagerwall7807c352011-03-17 20:20:30 +02009351#ifdef HAVE_FEXECVE
9352 {"fexecve", posix_fexecve, METH_VARARGS, posix_fexecve__doc__},
9353#endif
Guido van Rossuma1065681999-01-25 23:20:23 +00009354#ifdef HAVE_SPAWNV
Victor Stinner8c62be82010-05-06 00:08:46 +00009355 {"spawnv", posix_spawnv, METH_VARARGS, posix_spawnv__doc__},
9356 {"spawnve", posix_spawnve, METH_VARARGS, posix_spawnve__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00009357#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +00009358 {"spawnvp", posix_spawnvp, METH_VARARGS, posix_spawnvp__doc__},
9359 {"spawnvpe", posix_spawnvpe, METH_VARARGS, posix_spawnvpe__doc__},
Andrew MacIntyre69e18c92004-04-04 07:11:43 +00009360#endif /* PYOS_OS2 */
Guido van Rossuma1065681999-01-25 23:20:23 +00009361#endif /* HAVE_SPAWNV */
Guido van Rossum2242f2f2001-04-11 20:58:20 +00009362#ifdef HAVE_FORK1
Victor Stinner8c62be82010-05-06 00:08:46 +00009363 {"fork1", posix_fork1, METH_NOARGS, posix_fork1__doc__},
Guido van Rossum2242f2f2001-04-11 20:58:20 +00009364#endif /* HAVE_FORK1 */
Guido van Rossumad0ee831995-03-01 10:34:45 +00009365#ifdef HAVE_FORK
Victor Stinner8c62be82010-05-06 00:08:46 +00009366 {"fork", posix_fork, METH_NOARGS, posix_fork__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00009367#endif /* HAVE_FORK */
Martin v. Löwis24a880b2002-12-31 12:55:15 +00009368#if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
Victor Stinner8c62be82010-05-06 00:08:46 +00009369 {"openpty", posix_openpty, METH_NOARGS, posix_openpty__doc__},
Martin v. Löwis24a880b2002-12-31 12:55:15 +00009370#endif /* HAVE_OPENPTY || HAVE__GETPTY || HAVE_DEV_PTMX */
Fred Drake8cef4cf2000-06-28 16:40:38 +00009371#ifdef HAVE_FORKPTY
Victor Stinner8c62be82010-05-06 00:08:46 +00009372 {"forkpty", posix_forkpty, METH_NOARGS, posix_forkpty__doc__},
Fred Drake8cef4cf2000-06-28 16:40:38 +00009373#endif /* HAVE_FORKPTY */
Guido van Rossumad0ee831995-03-01 10:34:45 +00009374#ifdef HAVE_GETEGID
Victor Stinner8c62be82010-05-06 00:08:46 +00009375 {"getegid", posix_getegid, METH_NOARGS, posix_getegid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00009376#endif /* HAVE_GETEGID */
9377#ifdef HAVE_GETEUID
Victor Stinner8c62be82010-05-06 00:08:46 +00009378 {"geteuid", posix_geteuid, METH_NOARGS, posix_geteuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00009379#endif /* HAVE_GETEUID */
9380#ifdef HAVE_GETGID
Victor Stinner8c62be82010-05-06 00:08:46 +00009381 {"getgid", posix_getgid, METH_NOARGS, posix_getgid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00009382#endif /* HAVE_GETGID */
Fred Drakec9680921999-12-13 16:37:25 +00009383#ifdef HAVE_GETGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +00009384 {"getgroups", posix_getgroups, METH_NOARGS, posix_getgroups__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00009385#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009386 {"getpid", posix_getpid, METH_NOARGS, posix_getpid__doc__},
Guido van Rossumb6775db1994-08-01 11:34:53 +00009387#ifdef HAVE_GETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +00009388 {"getpgrp", posix_getpgrp, METH_NOARGS, posix_getpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009389#endif /* HAVE_GETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00009390#ifdef HAVE_GETPPID
Victor Stinner8c62be82010-05-06 00:08:46 +00009391 {"getppid", posix_getppid, METH_NOARGS, posix_getppid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00009392#endif /* HAVE_GETPPID */
9393#ifdef HAVE_GETUID
Victor Stinner8c62be82010-05-06 00:08:46 +00009394 {"getuid", posix_getuid, METH_NOARGS, posix_getuid__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00009395#endif /* HAVE_GETUID */
Fred Drake12c6e2d1999-12-14 21:25:03 +00009396#ifdef HAVE_GETLOGIN
Victor Stinner8c62be82010-05-06 00:08:46 +00009397 {"getlogin", posix_getlogin, METH_NOARGS, posix_getlogin__doc__},
Fred Drake12c6e2d1999-12-14 21:25:03 +00009398#endif
Guido van Rossumad0ee831995-03-01 10:34:45 +00009399#ifdef HAVE_KILL
Victor Stinner8c62be82010-05-06 00:08:46 +00009400 {"kill", posix_kill, METH_VARARGS, posix_kill__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00009401#endif /* HAVE_KILL */
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00009402#ifdef HAVE_KILLPG
Victor Stinner8c62be82010-05-06 00:08:46 +00009403 {"killpg", posix_killpg, METH_VARARGS, posix_killpg__doc__},
Martin v. Löwisb2c92f42002-02-16 23:35:41 +00009404#endif /* HAVE_KILLPG */
Guido van Rossumc0125471996-06-28 18:55:32 +00009405#ifdef HAVE_PLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +00009406 {"plock", posix_plock, METH_VARARGS, posix_plock__doc__},
Guido van Rossumc0125471996-06-28 18:55:32 +00009407#endif /* HAVE_PLOCK */
Thomas Heller8b7a9572007-08-31 06:44:36 +00009408#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00009409 {"startfile", win32_startfile, METH_VARARGS, win32_startfile__doc__},
9410 {"kill", win32_kill, METH_VARARGS, win32_kill__doc__},
Brian Curtin1b9df392010-11-24 20:24:31 +00009411 {"link", win32_link, METH_VARARGS, win32_link__doc__},
Thomas Heller8b7a9572007-08-31 06:44:36 +00009412#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00009413#ifdef HAVE_SETUID
Victor Stinner8c62be82010-05-06 00:08:46 +00009414 {"setuid", posix_setuid, METH_VARARGS, posix_setuid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009415#endif /* HAVE_SETUID */
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00009416#ifdef HAVE_SETEUID
Victor Stinner8c62be82010-05-06 00:08:46 +00009417 {"seteuid", posix_seteuid, METH_VARARGS, posix_seteuid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00009418#endif /* HAVE_SETEUID */
9419#ifdef HAVE_SETEGID
Victor Stinner8c62be82010-05-06 00:08:46 +00009420 {"setegid", posix_setegid, METH_VARARGS, posix_setegid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00009421#endif /* HAVE_SETEGID */
9422#ifdef HAVE_SETREUID
Victor Stinner8c62be82010-05-06 00:08:46 +00009423 {"setreuid", posix_setreuid, METH_VARARGS, posix_setreuid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00009424#endif /* HAVE_SETREUID */
9425#ifdef HAVE_SETREGID
Victor Stinner8c62be82010-05-06 00:08:46 +00009426 {"setregid", posix_setregid, METH_VARARGS, posix_setregid__doc__},
Andrew M. Kuchling8d2f2b22000-07-13 01:26:58 +00009427#endif /* HAVE_SETREGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00009428#ifdef HAVE_SETGID
Victor Stinner8c62be82010-05-06 00:08:46 +00009429 {"setgid", posix_setgid, METH_VARARGS, posix_setgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009430#endif /* HAVE_SETGID */
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00009431#ifdef HAVE_SETGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +00009432 {"setgroups", posix_setgroups, METH_O, posix_setgroups__doc__},
Martin v. Löwis61c5edf2001-10-18 04:06:00 +00009433#endif /* HAVE_SETGROUPS */
Antoine Pitroub7572f02009-12-02 20:46:48 +00009434#ifdef HAVE_INITGROUPS
Victor Stinner8c62be82010-05-06 00:08:46 +00009435 {"initgroups", posix_initgroups, METH_VARARGS, posix_initgroups__doc__},
Antoine Pitroub7572f02009-12-02 20:46:48 +00009436#endif /* HAVE_INITGROUPS */
Martin v. Löwis606edc12002-06-13 21:09:11 +00009437#ifdef HAVE_GETPGID
Victor Stinner8c62be82010-05-06 00:08:46 +00009438 {"getpgid", posix_getpgid, METH_VARARGS, posix_getpgid__doc__},
Martin v. Löwis606edc12002-06-13 21:09:11 +00009439#endif /* HAVE_GETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00009440#ifdef HAVE_SETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +00009441 {"setpgrp", posix_setpgrp, METH_NOARGS, posix_setpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009442#endif /* HAVE_SETPGRP */
Guido van Rossumad0ee831995-03-01 10:34:45 +00009443#ifdef HAVE_WAIT
Victor Stinner8c62be82010-05-06 00:08:46 +00009444 {"wait", posix_wait, METH_NOARGS, posix_wait__doc__},
Guido van Rossumad0ee831995-03-01 10:34:45 +00009445#endif /* HAVE_WAIT */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009446#ifdef HAVE_WAIT3
Victor Stinner8c62be82010-05-06 00:08:46 +00009447 {"wait3", posix_wait3, METH_VARARGS, posix_wait3__doc__},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009448#endif /* HAVE_WAIT3 */
9449#ifdef HAVE_WAIT4
Victor Stinner8c62be82010-05-06 00:08:46 +00009450 {"wait4", posix_wait4, METH_VARARGS, posix_wait4__doc__},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009451#endif /* HAVE_WAIT4 */
Ross Lagerwall7807c352011-03-17 20:20:30 +02009452#if defined(HAVE_WAITID) && !defined(__APPLE__)
9453 {"waitid", posix_waitid, METH_VARARGS, posix_waitid__doc__},
9454#endif
Tim Petersab034fa2002-02-01 11:27:43 +00009455#if defined(HAVE_WAITPID) || defined(HAVE_CWAIT)
Victor Stinner8c62be82010-05-06 00:08:46 +00009456 {"waitpid", posix_waitpid, METH_VARARGS, posix_waitpid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009457#endif /* HAVE_WAITPID */
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00009458#ifdef HAVE_GETSID
Victor Stinner8c62be82010-05-06 00:08:46 +00009459 {"getsid", posix_getsid, METH_VARARGS, posix_getsid__doc__},
Martin v. Löwis49ee14d2003-11-10 06:35:36 +00009460#endif /* HAVE_GETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00009461#ifdef HAVE_SETSID
Victor Stinner8c62be82010-05-06 00:08:46 +00009462 {"setsid", posix_setsid, METH_NOARGS, posix_setsid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009463#endif /* HAVE_SETSID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00009464#ifdef HAVE_SETPGID
Victor Stinner8c62be82010-05-06 00:08:46 +00009465 {"setpgid", posix_setpgid, METH_VARARGS, posix_setpgid__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009466#endif /* HAVE_SETPGID */
Guido van Rossumb6775db1994-08-01 11:34:53 +00009467#ifdef HAVE_TCGETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +00009468 {"tcgetpgrp", posix_tcgetpgrp, METH_VARARGS, posix_tcgetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009469#endif /* HAVE_TCGETPGRP */
Guido van Rossumb6775db1994-08-01 11:34:53 +00009470#ifdef HAVE_TCSETPGRP
Victor Stinner8c62be82010-05-06 00:08:46 +00009471 {"tcsetpgrp", posix_tcsetpgrp, METH_VARARGS, posix_tcsetpgrp__doc__},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00009472#endif /* HAVE_TCSETPGRP */
Victor Stinner8c62be82010-05-06 00:08:46 +00009473 {"open", posix_open, METH_VARARGS, posix_open__doc__},
9474 {"close", posix_close, METH_VARARGS, posix_close__doc__},
9475 {"closerange", posix_closerange, METH_VARARGS, posix_closerange__doc__},
9476 {"device_encoding", device_encoding, METH_VARARGS, device_encoding__doc__},
9477 {"dup", posix_dup, METH_VARARGS, posix_dup__doc__},
9478 {"dup2", posix_dup2, METH_VARARGS, posix_dup2__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +02009479#ifdef HAVE_LOCKF
9480 {"lockf", posix_lockf, METH_VARARGS, posix_lockf__doc__},
9481#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009482 {"lseek", posix_lseek, METH_VARARGS, posix_lseek__doc__},
9483 {"read", posix_read, METH_VARARGS, posix_read__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +02009484#ifdef HAVE_READV
9485 {"readv", posix_readv, METH_VARARGS, posix_readv__doc__},
9486#endif
9487#ifdef HAVE_PREAD
9488 {"pread", posix_pread, METH_VARARGS, posix_pread__doc__},
9489#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009490 {"write", posix_write, METH_VARARGS, posix_write__doc__},
Ross Lagerwall7807c352011-03-17 20:20:30 +02009491#ifdef HAVE_WRITEV
9492 {"writev", posix_writev, METH_VARARGS, posix_writev__doc__},
9493#endif
9494#ifdef HAVE_PWRITE
9495 {"pwrite", posix_pwrite, METH_VARARGS, posix_pwrite__doc__},
9496#endif
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00009497#ifdef HAVE_SENDFILE
9498 {"sendfile", (PyCFunction)posix_sendfile, METH_VARARGS | METH_KEYWORDS,
9499 posix_sendfile__doc__},
9500#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009501 {"fstat", posix_fstat, METH_VARARGS, posix_fstat__doc__},
9502 {"isatty", posix_isatty, METH_VARARGS, posix_isatty__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009503#ifdef HAVE_PIPE
Victor Stinner8c62be82010-05-06 00:08:46 +00009504 {"pipe", posix_pipe, METH_NOARGS, posix_pipe__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009505#endif
Charles-François Natalidaafdd52011-05-29 20:07:40 +02009506#ifdef HAVE_PIPE2
Charles-François Natali368f34b2011-06-06 19:49:47 +02009507 {"pipe2", posix_pipe2, METH_O, posix_pipe2__doc__},
Charles-François Natalidaafdd52011-05-29 20:07:40 +02009508#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009509#ifdef HAVE_MKFIFO
Victor Stinner8c62be82010-05-06 00:08:46 +00009510 {"mkfifo", posix_mkfifo, METH_VARARGS, posix_mkfifo__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009511#endif
Neal Norwitz11690112002-07-30 01:08:28 +00009512#if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
Victor Stinner8c62be82010-05-06 00:08:46 +00009513 {"mknod", posix_mknod, METH_VARARGS, posix_mknod__doc__},
Martin v. Löwis06a83e92002-04-14 10:19:44 +00009514#endif
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00009515#ifdef HAVE_DEVICE_MACROS
Victor Stinner8c62be82010-05-06 00:08:46 +00009516 {"major", posix_major, METH_VARARGS, posix_major__doc__},
9517 {"minor", posix_minor, METH_VARARGS, posix_minor__doc__},
9518 {"makedev", posix_makedev, METH_VARARGS, posix_makedev__doc__},
Martin v. Löwisdbe3f762002-10-10 14:27:30 +00009519#endif
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009520#ifdef HAVE_FTRUNCATE
Victor Stinner8c62be82010-05-06 00:08:46 +00009521 {"ftruncate", posix_ftruncate, METH_VARARGS, posix_ftruncate__doc__},
Guido van Rossuma4916fa1996-05-23 22:58:55 +00009522#endif
Ross Lagerwall7807c352011-03-17 20:20:30 +02009523#ifdef HAVE_TRUNCATE
9524 {"truncate", posix_truncate, METH_VARARGS, posix_truncate__doc__},
9525#endif
9526#ifdef HAVE_POSIX_FALLOCATE
9527 {"posix_fallocate", posix_posix_fallocate, METH_VARARGS, posix_posix_fallocate__doc__},
9528#endif
9529#ifdef HAVE_POSIX_FADVISE
9530 {"posix_fadvise", posix_posix_fadvise, METH_VARARGS, posix_posix_fadvise__doc__},
9531#endif
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00009532#ifdef HAVE_PUTENV
Victor Stinner8c62be82010-05-06 00:08:46 +00009533 {"putenv", posix_putenv, METH_VARARGS, posix_putenv__doc__},
Guido van Rossumf1af3fe1996-07-23 19:18:10 +00009534#endif
Guido van Rossumc524d952001-10-19 01:31:59 +00009535#ifdef HAVE_UNSETENV
Victor Stinner8c62be82010-05-06 00:08:46 +00009536 {"unsetenv", posix_unsetenv, METH_VARARGS, posix_unsetenv__doc__},
Guido van Rossumc524d952001-10-19 01:31:59 +00009537#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009538 {"strerror", posix_strerror, METH_VARARGS, posix_strerror__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +00009539#ifdef HAVE_FCHDIR
Victor Stinner8c62be82010-05-06 00:08:46 +00009540 {"fchdir", posix_fchdir, METH_O, posix_fchdir__doc__},
Fred Drake4d1e64b2002-04-15 19:40:07 +00009541#endif
Guido van Rossum21142a01999-01-08 21:05:37 +00009542#ifdef HAVE_FSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00009543 {"fsync", posix_fsync, METH_O, posix_fsync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00009544#endif
Ross Lagerwall7807c352011-03-17 20:20:30 +02009545#ifdef HAVE_SYNC
9546 {"sync", posix_sync, METH_NOARGS, posix_sync__doc__},
9547#endif
Guido van Rossum21142a01999-01-08 21:05:37 +00009548#ifdef HAVE_FDATASYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00009549 {"fdatasync", posix_fdatasync, METH_O, posix_fdatasync__doc__},
Guido van Rossum21142a01999-01-08 21:05:37 +00009550#endif
Guido van Rossumc9641791998-08-04 15:26:23 +00009551#ifdef HAVE_SYS_WAIT_H
Fred Drake106c1a02002-04-23 15:58:02 +00009552#ifdef WCOREDUMP
Victor Stinner8c62be82010-05-06 00:08:46 +00009553 {"WCOREDUMP", posix_WCOREDUMP, METH_VARARGS, posix_WCOREDUMP__doc__},
Fred Drake106c1a02002-04-23 15:58:02 +00009554#endif /* WCOREDUMP */
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00009555#ifdef WIFCONTINUED
Victor Stinner8c62be82010-05-06 00:08:46 +00009556 {"WIFCONTINUED",posix_WIFCONTINUED, METH_VARARGS, posix_WIFCONTINUED__doc__},
Martin v. Löwis2b41b0d2002-05-04 13:13:41 +00009557#endif /* WIFCONTINUED */
Guido van Rossumc9641791998-08-04 15:26:23 +00009558#ifdef WIFSTOPPED
Victor Stinner8c62be82010-05-06 00:08:46 +00009559 {"WIFSTOPPED", posix_WIFSTOPPED, METH_VARARGS, posix_WIFSTOPPED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00009560#endif /* WIFSTOPPED */
9561#ifdef WIFSIGNALED
Victor Stinner8c62be82010-05-06 00:08:46 +00009562 {"WIFSIGNALED", posix_WIFSIGNALED, METH_VARARGS, posix_WIFSIGNALED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00009563#endif /* WIFSIGNALED */
9564#ifdef WIFEXITED
Victor Stinner8c62be82010-05-06 00:08:46 +00009565 {"WIFEXITED", posix_WIFEXITED, METH_VARARGS, posix_WIFEXITED__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00009566#endif /* WIFEXITED */
9567#ifdef WEXITSTATUS
Victor Stinner8c62be82010-05-06 00:08:46 +00009568 {"WEXITSTATUS", posix_WEXITSTATUS, METH_VARARGS, posix_WEXITSTATUS__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00009569#endif /* WEXITSTATUS */
9570#ifdef WTERMSIG
Victor Stinner8c62be82010-05-06 00:08:46 +00009571 {"WTERMSIG", posix_WTERMSIG, METH_VARARGS, posix_WTERMSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00009572#endif /* WTERMSIG */
9573#ifdef WSTOPSIG
Victor Stinner8c62be82010-05-06 00:08:46 +00009574 {"WSTOPSIG", posix_WSTOPSIG, METH_VARARGS, posix_WSTOPSIG__doc__},
Guido van Rossumc9641791998-08-04 15:26:23 +00009575#endif /* WSTOPSIG */
9576#endif /* HAVE_SYS_WAIT_H */
Thomas Wouters477c8d52006-05-27 19:21:47 +00009577#if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
Victor Stinner8c62be82010-05-06 00:08:46 +00009578 {"fstatvfs", posix_fstatvfs, METH_VARARGS, posix_fstatvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00009579#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00009580#if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
Victor Stinner8c62be82010-05-06 00:08:46 +00009581 {"statvfs", posix_statvfs, METH_VARARGS, posix_statvfs__doc__},
Guido van Rossum94f6f721999-01-06 18:42:14 +00009582#endif
Fred Drakec9680921999-12-13 16:37:25 +00009583#ifdef HAVE_CONFSTR
Victor Stinner8c62be82010-05-06 00:08:46 +00009584 {"confstr", posix_confstr, METH_VARARGS, posix_confstr__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00009585#endif
9586#ifdef HAVE_SYSCONF
Victor Stinner8c62be82010-05-06 00:08:46 +00009587 {"sysconf", posix_sysconf, METH_VARARGS, posix_sysconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00009588#endif
9589#ifdef HAVE_FPATHCONF
Victor Stinner8c62be82010-05-06 00:08:46 +00009590 {"fpathconf", posix_fpathconf, METH_VARARGS, posix_fpathconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00009591#endif
9592#ifdef HAVE_PATHCONF
Victor Stinner8c62be82010-05-06 00:08:46 +00009593 {"pathconf", posix_pathconf, METH_VARARGS, posix_pathconf__doc__},
Fred Drakec9680921999-12-13 16:37:25 +00009594#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009595 {"abort", posix_abort, METH_NOARGS, posix_abort__doc__},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00009596#ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00009597 {"_getfullpathname", posix__getfullpathname, METH_VARARGS, NULL},
Brian Curtind40e6f72010-07-08 21:39:08 +00009598 {"_getfinalpathname", posix__getfinalpathname, METH_VARARGS, NULL},
Brian Curtin62857742010-09-06 17:07:27 +00009599 {"_getfileinformation", posix__getfileinformation, METH_VARARGS, NULL},
Brian Curtin9c669cc2011-06-08 18:17:18 -05009600 {"_isdir", posix__isdir, METH_VARARGS, NULL},
Mark Hammondef8b6542001-05-13 08:04:26 +00009601#endif
Martin v. Löwis438b5342002-12-27 10:16:42 +00009602#ifdef HAVE_GETLOADAVG
Victor Stinner8c62be82010-05-06 00:08:46 +00009603 {"getloadavg", posix_getloadavg, METH_NOARGS, posix_getloadavg__doc__},
Martin v. Löwis438b5342002-12-27 10:16:42 +00009604#endif
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009605 #ifdef MS_WINDOWS
Victor Stinner8c62be82010-05-06 00:08:46 +00009606 {"urandom", win32_urandom, METH_VARARGS, win32_urandom__doc__},
Martin v. Löwisdc3883f2004-08-29 15:46:35 +00009607 #endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009608 #ifdef __VMS
Victor Stinner8c62be82010-05-06 00:08:46 +00009609 {"urandom", vms_urandom, METH_VARARGS, vms_urandom__doc__},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009610 #endif
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009611#ifdef HAVE_SETRESUID
Victor Stinner8c62be82010-05-06 00:08:46 +00009612 {"setresuid", posix_setresuid, METH_VARARGS, posix_setresuid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009613#endif
9614#ifdef HAVE_SETRESGID
Victor Stinner8c62be82010-05-06 00:08:46 +00009615 {"setresgid", posix_setresgid, METH_VARARGS, posix_setresgid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009616#endif
9617#ifdef HAVE_GETRESUID
Victor Stinner8c62be82010-05-06 00:08:46 +00009618 {"getresuid", posix_getresuid, METH_NOARGS, posix_getresuid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009619#endif
9620#ifdef HAVE_GETRESGID
Victor Stinner8c62be82010-05-06 00:08:46 +00009621 {"getresgid", posix_getresgid, METH_NOARGS, posix_getresgid__doc__},
Martin v. Löwis7aed61a2009-11-27 14:09:49 +00009622#endif
9623
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009624/* posix *at family of functions */
9625#ifdef HAVE_FACCESSAT
9626 {"faccessat", posix_faccessat, METH_VARARGS, posix_faccessat__doc__},
9627#endif
9628#ifdef HAVE_FCHMODAT
9629 {"fchmodat", posix_fchmodat, METH_VARARGS, posix_fchmodat__doc__},
9630#endif /* HAVE_FCHMODAT */
9631#ifdef HAVE_FCHOWNAT
9632 {"fchownat", posix_fchownat, METH_VARARGS, posix_fchownat__doc__},
9633#endif /* HAVE_FCHOWNAT */
9634#ifdef HAVE_FSTATAT
9635 {"fstatat", posix_fstatat, METH_VARARGS, posix_fstatat__doc__},
9636#endif
9637#ifdef HAVE_FUTIMESAT
9638 {"futimesat", posix_futimesat, METH_VARARGS, posix_futimesat__doc__},
9639#endif
9640#ifdef HAVE_LINKAT
9641 {"linkat", posix_linkat, METH_VARARGS, posix_linkat__doc__},
9642#endif /* HAVE_LINKAT */
9643#ifdef HAVE_MKDIRAT
9644 {"mkdirat", posix_mkdirat, METH_VARARGS, posix_mkdirat__doc__},
9645#endif
9646#if defined(HAVE_MKNODAT) && defined(HAVE_MAKEDEV)
9647 {"mknodat", posix_mknodat, METH_VARARGS, posix_mknodat__doc__},
9648#endif
9649#ifdef HAVE_OPENAT
9650 {"openat", posix_openat, METH_VARARGS, posix_openat__doc__},
9651#endif
9652#ifdef HAVE_READLINKAT
9653 {"readlinkat", posix_readlinkat, METH_VARARGS, posix_readlinkat__doc__},
9654#endif /* HAVE_READLINKAT */
9655#ifdef HAVE_RENAMEAT
9656 {"renameat", posix_renameat, METH_VARARGS, posix_renameat__doc__},
9657#endif
9658#if HAVE_SYMLINKAT
9659 {"symlinkat", posix_symlinkat, METH_VARARGS, posix_symlinkat__doc__},
9660#endif /* HAVE_SYMLINKAT */
9661#ifdef HAVE_UNLINKAT
9662 {"unlinkat", posix_unlinkat, METH_VARARGS, posix_unlinkat__doc__},
9663#endif
9664#ifdef HAVE_UTIMENSAT
9665 {"utimensat", posix_utimensat, METH_VARARGS, posix_utimensat__doc__},
9666#endif
9667#ifdef HAVE_MKFIFOAT
9668 {"mkfifoat", posix_mkfifoat, METH_VARARGS, posix_mkfifoat__doc__},
9669#endif
Victor Stinner8c62be82010-05-06 00:08:46 +00009670 {NULL, NULL} /* Sentinel */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00009671};
9672
9673
Barry Warsaw4a342091996-12-19 23:50:02 +00009674static int
Fred Drake4d1e64b2002-04-15 19:40:07 +00009675ins(PyObject *module, char *symbol, long value)
Barry Warsaw4a342091996-12-19 23:50:02 +00009676{
Victor Stinner8c62be82010-05-06 00:08:46 +00009677 return PyModule_AddIntConstant(module, symbol, value);
Barry Warsaw4a342091996-12-19 23:50:02 +00009678}
9679
Guido van Rossumd48f2521997-12-05 22:19:34 +00009680#if defined(PYOS_OS2)
9681/* Insert Platform-Specific Constant Values (Strings & Numbers) of Common Use */
Fred Drake4d1e64b2002-04-15 19:40:07 +00009682static int insertvalues(PyObject *module)
Guido van Rossumd48f2521997-12-05 22:19:34 +00009683{
9684 APIRET rc;
9685 ULONG values[QSV_MAX+1];
9686 PyObject *v;
Marc-André Lemburgd4c0a9c2001-11-28 11:47:00 +00009687 char *ver, tmp[50];
Guido van Rossumd48f2521997-12-05 22:19:34 +00009688
9689 Py_BEGIN_ALLOW_THREADS
Andrew MacIntyre75e01452003-04-21 14:19:51 +00009690 rc = DosQuerySysInfo(1L, QSV_MAX, &values[1], sizeof(ULONG) * QSV_MAX);
Guido van Rossumd48f2521997-12-05 22:19:34 +00009691 Py_END_ALLOW_THREADS
9692
9693 if (rc != NO_ERROR) {
9694 os2_error(rc);
9695 return -1;
9696 }
9697
Fred Drake4d1e64b2002-04-15 19:40:07 +00009698 if (ins(module, "meminstalled", values[QSV_TOTPHYSMEM])) return -1;
9699 if (ins(module, "memkernel", values[QSV_TOTRESMEM])) return -1;
9700 if (ins(module, "memvirtual", values[QSV_TOTAVAILMEM])) return -1;
9701 if (ins(module, "maxpathlen", values[QSV_MAX_PATH_LENGTH])) return -1;
9702 if (ins(module, "maxnamelen", values[QSV_MAX_COMP_LENGTH])) return -1;
9703 if (ins(module, "revision", values[QSV_VERSION_REVISION])) return -1;
9704 if (ins(module, "timeslice", values[QSV_MIN_SLICE])) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00009705
9706 switch (values[QSV_VERSION_MINOR]) {
9707 case 0: ver = "2.00"; break;
9708 case 10: ver = "2.10"; break;
9709 case 11: ver = "2.11"; break;
9710 case 30: ver = "3.00"; break;
9711 case 40: ver = "4.00"; break;
9712 case 50: ver = "5.00"; break;
9713 default:
Tim Peters885d4572001-11-28 20:27:42 +00009714 PyOS_snprintf(tmp, sizeof(tmp),
Victor Stinner8c62be82010-05-06 00:08:46 +00009715 "%d-%d", values[QSV_VERSION_MAJOR],
Tim Peters885d4572001-11-28 20:27:42 +00009716 values[QSV_VERSION_MINOR]);
Guido van Rossumd48f2521997-12-05 22:19:34 +00009717 ver = &tmp[0];
9718 }
9719
9720 /* Add Indicator of the Version of the Operating System */
Fred Drake4d1e64b2002-04-15 19:40:07 +00009721 if (PyModule_AddStringConstant(module, "version", tmp) < 0)
Guido van Rossumd48f2521997-12-05 22:19:34 +00009722 return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +00009723
9724 /* Add Indicator of Which Drive was Used to Boot the System */
9725 tmp[0] = 'A' + values[QSV_BOOT_DRIVE] - 1;
9726 tmp[1] = ':';
9727 tmp[2] = '\0';
9728
Fred Drake4d1e64b2002-04-15 19:40:07 +00009729 return PyModule_AddStringConstant(module, "bootdrive", tmp);
Guido van Rossumd48f2521997-12-05 22:19:34 +00009730}
9731#endif
9732
Brian Curtin52173d42010-12-02 18:29:18 +00009733#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +00009734static int
Brian Curtin52173d42010-12-02 18:29:18 +00009735enable_symlink()
9736{
9737 HANDLE tok;
9738 TOKEN_PRIVILEGES tok_priv;
9739 LUID luid;
9740 int meth_idx = 0;
9741
9742 if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &tok))
Brian Curtin3b4499c2010-12-28 14:31:47 +00009743 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +00009744
9745 if (!LookupPrivilegeValue(NULL, SE_CREATE_SYMBOLIC_LINK_NAME, &luid))
Brian Curtin3b4499c2010-12-28 14:31:47 +00009746 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +00009747
9748 tok_priv.PrivilegeCount = 1;
9749 tok_priv.Privileges[0].Luid = luid;
9750 tok_priv.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
9751
9752 if (!AdjustTokenPrivileges(tok, FALSE, &tok_priv,
9753 sizeof(TOKEN_PRIVILEGES),
9754 (PTOKEN_PRIVILEGES) NULL, (PDWORD) NULL))
Brian Curtin3b4499c2010-12-28 14:31:47 +00009755 return 0;
Brian Curtin52173d42010-12-02 18:29:18 +00009756
Brian Curtin3b4499c2010-12-28 14:31:47 +00009757 /* ERROR_NOT_ALL_ASSIGNED returned when the privilege can't be assigned. */
9758 return GetLastError() == ERROR_NOT_ALL_ASSIGNED ? 0 : 1;
Brian Curtin52173d42010-12-02 18:29:18 +00009759}
9760#endif /* defined(HAVE_SYMLINK) && defined(MS_WINDOWS) */
9761
Barry Warsaw4a342091996-12-19 23:50:02 +00009762static int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +00009763all_ins(PyObject *d)
Barry Warsaw4a342091996-12-19 23:50:02 +00009764{
Guido van Rossum94f6f721999-01-06 18:42:14 +00009765#ifdef F_OK
Victor Stinner8c62be82010-05-06 00:08:46 +00009766 if (ins(d, "F_OK", (long)F_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009767#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00009768#ifdef R_OK
Victor Stinner8c62be82010-05-06 00:08:46 +00009769 if (ins(d, "R_OK", (long)R_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009770#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00009771#ifdef W_OK
Victor Stinner8c62be82010-05-06 00:08:46 +00009772 if (ins(d, "W_OK", (long)W_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009773#endif
Guido van Rossum94f6f721999-01-06 18:42:14 +00009774#ifdef X_OK
Victor Stinner8c62be82010-05-06 00:08:46 +00009775 if (ins(d, "X_OK", (long)X_OK)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009776#endif
Fred Drakec9680921999-12-13 16:37:25 +00009777#ifdef NGROUPS_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009778 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1;
Fred Drakec9680921999-12-13 16:37:25 +00009779#endif
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009780#ifdef TMP_MAX
Victor Stinner8c62be82010-05-06 00:08:46 +00009781 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1;
Fred Drake5ab8eaf1999-12-09 21:13:07 +00009782#endif
Fred Drake106c1a02002-04-23 15:58:02 +00009783#ifdef WCONTINUED
Victor Stinner8c62be82010-05-06 00:08:46 +00009784 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1;
Fred Drake106c1a02002-04-23 15:58:02 +00009785#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00009786#ifdef WNOHANG
Victor Stinner8c62be82010-05-06 00:08:46 +00009787 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009788#endif
Fred Drake106c1a02002-04-23 15:58:02 +00009789#ifdef WUNTRACED
Victor Stinner8c62be82010-05-06 00:08:46 +00009790 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1;
Fred Drake106c1a02002-04-23 15:58:02 +00009791#endif
Barry Warsaw4a342091996-12-19 23:50:02 +00009792#ifdef O_RDONLY
Victor Stinner8c62be82010-05-06 00:08:46 +00009793 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009794#endif
9795#ifdef O_WRONLY
Victor Stinner8c62be82010-05-06 00:08:46 +00009796 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009797#endif
9798#ifdef O_RDWR
Victor Stinner8c62be82010-05-06 00:08:46 +00009799 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009800#endif
9801#ifdef O_NDELAY
Victor Stinner8c62be82010-05-06 00:08:46 +00009802 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009803#endif
9804#ifdef O_NONBLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +00009805 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009806#endif
9807#ifdef O_APPEND
Victor Stinner8c62be82010-05-06 00:08:46 +00009808 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009809#endif
9810#ifdef O_DSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00009811 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009812#endif
9813#ifdef O_RSYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00009814 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009815#endif
9816#ifdef O_SYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00009817 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009818#endif
9819#ifdef O_NOCTTY
Victor Stinner8c62be82010-05-06 00:08:46 +00009820 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009821#endif
9822#ifdef O_CREAT
Victor Stinner8c62be82010-05-06 00:08:46 +00009823 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009824#endif
9825#ifdef O_EXCL
Victor Stinner8c62be82010-05-06 00:08:46 +00009826 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009827#endif
9828#ifdef O_TRUNC
Victor Stinner8c62be82010-05-06 00:08:46 +00009829 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1;
Barry Warsaw4a342091996-12-19 23:50:02 +00009830#endif
Guido van Rossum98d9d091997-08-08 21:48:51 +00009831#ifdef O_BINARY
Victor Stinner8c62be82010-05-06 00:08:46 +00009832 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1;
Guido van Rossum98d9d091997-08-08 21:48:51 +00009833#endif
9834#ifdef O_TEXT
Victor Stinner8c62be82010-05-06 00:08:46 +00009835 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1;
Guido van Rossum98d9d091997-08-08 21:48:51 +00009836#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00009837#ifdef O_LARGEFILE
Victor Stinner8c62be82010-05-06 00:08:46 +00009838 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00009839#endif
Skip Montanaro5ff14922005-05-16 02:42:22 +00009840#ifdef O_SHLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +00009841 if (ins(d, "O_SHLOCK", (long)O_SHLOCK)) return -1;
Skip Montanaro5ff14922005-05-16 02:42:22 +00009842#endif
9843#ifdef O_EXLOCK
Victor Stinner8c62be82010-05-06 00:08:46 +00009844 if (ins(d, "O_EXLOCK", (long)O_EXLOCK)) return -1;
Skip Montanaro5ff14922005-05-16 02:42:22 +00009845#endif
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00009846#ifdef PRIO_PROCESS
9847 if (ins(d, "PRIO_PROCESS", (long)PRIO_PROCESS)) return -1;
9848#endif
9849#ifdef PRIO_PGRP
9850 if (ins(d, "PRIO_PGRP", (long)PRIO_PGRP)) return -1;
9851#endif
9852#ifdef PRIO_USER
9853 if (ins(d, "PRIO_USER", (long)PRIO_USER)) return -1;
9854#endif
Charles-François Natali1e045b12011-05-22 20:42:32 +02009855#ifdef O_CLOEXEC
9856 if (ins(d, "O_CLOEXEC", (long)O_CLOEXEC)) return -1;
9857#endif
Antoine Pitrouf65132d2011-02-25 23:25:17 +00009858/* posix - constants for *at functions */
9859#ifdef AT_SYMLINK_NOFOLLOW
9860 if (ins(d, "AT_SYMLINK_NOFOLLOW", (long)AT_SYMLINK_NOFOLLOW)) return -1;
9861#endif
9862#ifdef AT_EACCESS
9863 if (ins(d, "AT_EACCESS", (long)AT_EACCESS)) return -1;
9864#endif
9865#ifdef AT_FDCWD
9866 if (ins(d, "AT_FDCWD", (long)AT_FDCWD)) return -1;
9867#endif
9868#ifdef AT_REMOVEDIR
9869 if (ins(d, "AT_REMOVEDIR", (long)AT_REMOVEDIR)) return -1;
9870#endif
9871#ifdef AT_SYMLINK_FOLLOW
9872 if (ins(d, "AT_SYMLINK_FOLLOW", (long)AT_SYMLINK_FOLLOW)) return -1;
9873#endif
9874#ifdef UTIME_NOW
9875 if (ins(d, "UTIME_NOW", (long)UTIME_NOW)) return -1;
9876#endif
9877#ifdef UTIME_OMIT
9878 if (ins(d, "UTIME_OMIT", (long)UTIME_OMIT)) return -1;
9879#endif
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00009880
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00009881
Tim Peters5aa91602002-01-30 05:46:57 +00009882/* MS Windows */
9883#ifdef O_NOINHERIT
Victor Stinner8c62be82010-05-06 00:08:46 +00009884 /* Don't inherit in child processes. */
9885 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009886#endif
9887#ifdef _O_SHORT_LIVED
Victor Stinner8c62be82010-05-06 00:08:46 +00009888 /* Optimize for short life (keep in memory). */
9889 /* MS forgot to define this one with a non-underscore form too. */
9890 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009891#endif
9892#ifdef O_TEMPORARY
Victor Stinner8c62be82010-05-06 00:08:46 +00009893 /* Automatically delete when last handle is closed. */
9894 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009895#endif
9896#ifdef O_RANDOM
Victor Stinner8c62be82010-05-06 00:08:46 +00009897 /* Optimize for random access. */
9898 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009899#endif
9900#ifdef O_SEQUENTIAL
Victor Stinner8c62be82010-05-06 00:08:46 +00009901 /* Optimize for sequential access. */
9902 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1;
Tim Peters5aa91602002-01-30 05:46:57 +00009903#endif
9904
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00009905/* GNU extensions. */
Alexandre Vassalottibee32532008-05-16 18:15:12 +00009906#ifdef O_ASYNC
Victor Stinner8c62be82010-05-06 00:08:46 +00009907 /* Send a SIGIO signal whenever input or output
9908 becomes available on file descriptor */
9909 if (ins(d, "O_ASYNC", (long)O_ASYNC)) return -1;
Alexandre Vassalottibee32532008-05-16 18:15:12 +00009910#endif
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00009911#ifdef O_DIRECT
Victor Stinner8c62be82010-05-06 00:08:46 +00009912 /* Direct disk access. */
9913 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00009914#endif
9915#ifdef O_DIRECTORY
Victor Stinner8c62be82010-05-06 00:08:46 +00009916 /* Must be a directory. */
9917 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00009918#endif
9919#ifdef O_NOFOLLOW
Victor Stinner8c62be82010-05-06 00:08:46 +00009920 /* Do not follow links. */
9921 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1;
Martin v. Löwis4fe3c272001-10-18 22:05:36 +00009922#endif
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +00009923#ifdef O_NOATIME
Victor Stinner8c62be82010-05-06 00:08:46 +00009924 /* Do not update the access time. */
9925 if (ins(d, "O_NOATIME", (long)O_NOATIME)) return -1;
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +00009926#endif
Guido van Rossumd48f2521997-12-05 22:19:34 +00009927
Victor Stinner8c62be82010-05-06 00:08:46 +00009928 /* These come from sysexits.h */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009929#ifdef EX_OK
Victor Stinner8c62be82010-05-06 00:08:46 +00009930 if (ins(d, "EX_OK", (long)EX_OK)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009931#endif /* EX_OK */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009932#ifdef EX_USAGE
Victor Stinner8c62be82010-05-06 00:08:46 +00009933 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009934#endif /* EX_USAGE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009935#ifdef EX_DATAERR
Victor Stinner8c62be82010-05-06 00:08:46 +00009936 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009937#endif /* EX_DATAERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009938#ifdef EX_NOINPUT
Victor Stinner8c62be82010-05-06 00:08:46 +00009939 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009940#endif /* EX_NOINPUT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009941#ifdef EX_NOUSER
Victor Stinner8c62be82010-05-06 00:08:46 +00009942 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009943#endif /* EX_NOUSER */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009944#ifdef EX_NOHOST
Victor Stinner8c62be82010-05-06 00:08:46 +00009945 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009946#endif /* EX_NOHOST */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009947#ifdef EX_UNAVAILABLE
Victor Stinner8c62be82010-05-06 00:08:46 +00009948 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009949#endif /* EX_UNAVAILABLE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009950#ifdef EX_SOFTWARE
Victor Stinner8c62be82010-05-06 00:08:46 +00009951 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009952#endif /* EX_SOFTWARE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009953#ifdef EX_OSERR
Victor Stinner8c62be82010-05-06 00:08:46 +00009954 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009955#endif /* EX_OSERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009956#ifdef EX_OSFILE
Victor Stinner8c62be82010-05-06 00:08:46 +00009957 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009958#endif /* EX_OSFILE */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009959#ifdef EX_CANTCREAT
Victor Stinner8c62be82010-05-06 00:08:46 +00009960 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009961#endif /* EX_CANTCREAT */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009962#ifdef EX_IOERR
Victor Stinner8c62be82010-05-06 00:08:46 +00009963 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009964#endif /* EX_IOERR */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009965#ifdef EX_TEMPFAIL
Victor Stinner8c62be82010-05-06 00:08:46 +00009966 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009967#endif /* EX_TEMPFAIL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009968#ifdef EX_PROTOCOL
Victor Stinner8c62be82010-05-06 00:08:46 +00009969 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009970#endif /* EX_PROTOCOL */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009971#ifdef EX_NOPERM
Victor Stinner8c62be82010-05-06 00:08:46 +00009972 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009973#endif /* EX_NOPERM */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009974#ifdef EX_CONFIG
Victor Stinner8c62be82010-05-06 00:08:46 +00009975 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009976#endif /* EX_CONFIG */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009977#ifdef EX_NOTFOUND
Victor Stinner8c62be82010-05-06 00:08:46 +00009978 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1;
Neal Norwitz8e914d92003-01-10 15:29:16 +00009979#endif /* EX_NOTFOUND */
Barry Warsaw5676bd12003-01-07 20:57:09 +00009980
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +00009981 /* statvfs */
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +00009982#ifdef ST_RDONLY
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +00009983 if (ins(d, "ST_RDONLY", (long)ST_RDONLY)) return -1;
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +00009984#endif /* ST_RDONLY */
9985#ifdef ST_NOSUID
Amaury Forgeot d'Arc66d00ad2010-09-10 18:11:45 +00009986 if (ins(d, "ST_NOSUID", (long)ST_NOSUID)) return -1;
Andrew M. Kuchling4ea04a32010-08-18 22:30:34 +00009987#endif /* ST_NOSUID */
9988
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00009989 /* FreeBSD sendfile() constants */
9990#ifdef SF_NODISKIO
9991 if (ins(d, "SF_NODISKIO", (long)SF_NODISKIO)) return -1;
9992#endif
9993#ifdef SF_MNOWAIT
9994 if (ins(d, "SF_MNOWAIT", (long)SF_MNOWAIT)) return -1;
9995#endif
9996#ifdef SF_SYNC
9997 if (ins(d, "SF_SYNC", (long)SF_SYNC)) return -1;
9998#endif
9999
Ross Lagerwall7807c352011-03-17 20:20:30 +020010000 /* constants for posix_fadvise */
10001#ifdef POSIX_FADV_NORMAL
10002 if (ins(d, "POSIX_FADV_NORMAL", (long)POSIX_FADV_NORMAL)) return -1;
10003#endif
10004#ifdef POSIX_FADV_SEQUENTIAL
10005 if (ins(d, "POSIX_FADV_SEQUENTIAL", (long)POSIX_FADV_SEQUENTIAL)) return -1;
10006#endif
10007#ifdef POSIX_FADV_RANDOM
10008 if (ins(d, "POSIX_FADV_RANDOM", (long)POSIX_FADV_RANDOM)) return -1;
10009#endif
10010#ifdef POSIX_FADV_NOREUSE
10011 if (ins(d, "POSIX_FADV_NOREUSE", (long)POSIX_FADV_NOREUSE)) return -1;
10012#endif
10013#ifdef POSIX_FADV_WILLNEED
10014 if (ins(d, "POSIX_FADV_WILLNEED", (long)POSIX_FADV_WILLNEED)) return -1;
10015#endif
10016#ifdef POSIX_FADV_DONTNEED
10017 if (ins(d, "POSIX_FADV_DONTNEED", (long)POSIX_FADV_DONTNEED)) return -1;
10018#endif
10019
10020 /* constants for waitid */
10021#if defined(HAVE_SYS_WAIT_H) && defined(HAVE_WAITID)
10022 if (ins(d, "P_PID", (long)P_PID)) return -1;
10023 if (ins(d, "P_PGID", (long)P_PGID)) return -1;
10024 if (ins(d, "P_ALL", (long)P_ALL)) return -1;
10025#endif
10026#ifdef WEXITED
10027 if (ins(d, "WEXITED", (long)WEXITED)) return -1;
10028#endif
10029#ifdef WNOWAIT
10030 if (ins(d, "WNOWAIT", (long)WNOWAIT)) return -1;
10031#endif
10032#ifdef WSTOPPED
10033 if (ins(d, "WSTOPPED", (long)WSTOPPED)) return -1;
10034#endif
10035#ifdef CLD_EXITED
10036 if (ins(d, "CLD_EXITED", (long)CLD_EXITED)) return -1;
10037#endif
10038#ifdef CLD_DUMPED
10039 if (ins(d, "CLD_DUMPED", (long)CLD_DUMPED)) return -1;
10040#endif
10041#ifdef CLD_TRAPPED
10042 if (ins(d, "CLD_TRAPPED", (long)CLD_TRAPPED)) return -1;
10043#endif
10044#ifdef CLD_CONTINUED
10045 if (ins(d, "CLD_CONTINUED", (long)CLD_CONTINUED)) return -1;
10046#endif
10047
10048 /* constants for lockf */
10049#ifdef F_LOCK
10050 if (ins(d, "F_LOCK", (long)F_LOCK)) return -1;
10051#endif
10052#ifdef F_TLOCK
10053 if (ins(d, "F_TLOCK", (long)F_TLOCK)) return -1;
10054#endif
10055#ifdef F_ULOCK
10056 if (ins(d, "F_ULOCK", (long)F_ULOCK)) return -1;
10057#endif
10058#ifdef F_TEST
10059 if (ins(d, "F_TEST", (long)F_TEST)) return -1;
10060#endif
10061
10062 /* constants for futimens */
10063#ifdef UTIME_NOW
10064 if (ins(d, "UTIME_NOW", (long)UTIME_NOW)) return -1;
10065#endif
10066#ifdef UTIME_OMIT
10067 if (ins(d, "UTIME_OMIT", (long)UTIME_OMIT)) return -1;
10068#endif
10069
Guido van Rossum246bc171999-02-01 23:54:31 +000010070#ifdef HAVE_SPAWNV
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000010071#if defined(PYOS_OS2) && defined(PYCC_GCC)
Victor Stinner8c62be82010-05-06 00:08:46 +000010072 if (ins(d, "P_WAIT", (long)P_WAIT)) return -1;
10073 if (ins(d, "P_NOWAIT", (long)P_NOWAIT)) return -1;
10074 if (ins(d, "P_OVERLAY", (long)P_OVERLAY)) return -1;
10075 if (ins(d, "P_DEBUG", (long)P_DEBUG)) return -1;
10076 if (ins(d, "P_SESSION", (long)P_SESSION)) return -1;
10077 if (ins(d, "P_DETACH", (long)P_DETACH)) return -1;
10078 if (ins(d, "P_PM", (long)P_PM)) return -1;
10079 if (ins(d, "P_DEFAULT", (long)P_DEFAULT)) return -1;
10080 if (ins(d, "P_MINIMIZE", (long)P_MINIMIZE)) return -1;
10081 if (ins(d, "P_MAXIMIZE", (long)P_MAXIMIZE)) return -1;
10082 if (ins(d, "P_FULLSCREEN", (long)P_FULLSCREEN)) return -1;
10083 if (ins(d, "P_WINDOWED", (long)P_WINDOWED)) return -1;
10084 if (ins(d, "P_FOREGROUND", (long)P_FOREGROUND)) return -1;
10085 if (ins(d, "P_BACKGROUND", (long)P_BACKGROUND)) return -1;
10086 if (ins(d, "P_NOCLOSE", (long)P_NOCLOSE)) return -1;
10087 if (ins(d, "P_NOSESSION", (long)P_NOSESSION)) return -1;
10088 if (ins(d, "P_QUOTE", (long)P_QUOTE)) return -1;
10089 if (ins(d, "P_TILDE", (long)P_TILDE)) return -1;
10090 if (ins(d, "P_UNRELATED", (long)P_UNRELATED)) return -1;
10091 if (ins(d, "P_DEBUGDESC", (long)P_DEBUGDESC)) return -1;
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000010092#else
Victor Stinner8c62be82010-05-06 00:08:46 +000010093 if (ins(d, "P_WAIT", (long)_P_WAIT)) return -1;
10094 if (ins(d, "P_NOWAIT", (long)_P_NOWAIT)) return -1;
10095 if (ins(d, "P_OVERLAY", (long)_OLD_P_OVERLAY)) return -1;
10096 if (ins(d, "P_NOWAITO", (long)_P_NOWAITO)) return -1;
10097 if (ins(d, "P_DETACH", (long)_P_DETACH)) return -1;
Guido van Rossum246bc171999-02-01 23:54:31 +000010098#endif
Andrew MacIntyre6c73af22002-03-03 03:07:07 +000010099#endif
Guido van Rossum246bc171999-02-01 23:54:31 +000010100
Guido van Rossumd48f2521997-12-05 22:19:34 +000010101#if defined(PYOS_OS2)
Victor Stinner8c62be82010-05-06 00:08:46 +000010102 if (insertvalues(d)) return -1;
Guido van Rossumd48f2521997-12-05 22:19:34 +000010103#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010104 return 0;
Barry Warsaw4a342091996-12-19 23:50:02 +000010105}
10106
10107
Tim Peters5aa91602002-01-30 05:46:57 +000010108#if (defined(_MSC_VER) || defined(__WATCOMC__) || defined(__BORLANDC__)) && !defined(__QNX__)
Martin v. Löwis1a214512008-06-11 05:26:20 +000010109#define INITFUNC PyInit_nt
Guido van Rossum0cb96de1997-10-01 04:29:29 +000010110#define MODNAME "nt"
Tim Peters58e0a8c2001-05-14 22:32:33 +000010111
10112#elif defined(PYOS_OS2)
Martin v. Löwis1a214512008-06-11 05:26:20 +000010113#define INITFUNC PyInit_os2
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000010114#define MODNAME "os2"
Tim Peters58e0a8c2001-05-14 22:32:33 +000010115
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000010116#else
Martin v. Löwis1a214512008-06-11 05:26:20 +000010117#define INITFUNC PyInit_posix
Guido van Rossum0cb96de1997-10-01 04:29:29 +000010118#define MODNAME "posix"
10119#endif
10120
Martin v. Löwis1a214512008-06-11 05:26:20 +000010121static struct PyModuleDef posixmodule = {
Victor Stinner8c62be82010-05-06 00:08:46 +000010122 PyModuleDef_HEAD_INIT,
10123 MODNAME,
10124 posix__doc__,
10125 -1,
10126 posix_methods,
10127 NULL,
10128 NULL,
10129 NULL,
10130 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +000010131};
10132
10133
Mark Hammondfe51c6d2002-08-02 02:27:13 +000010134PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +000010135INITFUNC(void)
Guido van Rossumb6775db1994-08-01 11:34:53 +000010136{
Victor Stinner8c62be82010-05-06 00:08:46 +000010137 PyObject *m, *v;
Tim Peters5aa91602002-01-30 05:46:57 +000010138
Brian Curtin52173d42010-12-02 18:29:18 +000010139#if defined(HAVE_SYMLINK) && defined(MS_WINDOWS)
Brian Curtin3b4499c2010-12-28 14:31:47 +000010140 win32_can_symlink = enable_symlink();
Brian Curtin52173d42010-12-02 18:29:18 +000010141#endif
10142
Victor Stinner8c62be82010-05-06 00:08:46 +000010143 m = PyModule_Create(&posixmodule);
10144 if (m == NULL)
10145 return NULL;
Tim Peters5aa91602002-01-30 05:46:57 +000010146
Victor Stinner8c62be82010-05-06 00:08:46 +000010147 /* Initialize environ dictionary */
10148 v = convertenviron();
10149 Py_XINCREF(v);
10150 if (v == NULL || PyModule_AddObject(m, "environ", v) != 0)
10151 return NULL;
10152 Py_DECREF(v);
Fred Drakec9680921999-12-13 16:37:25 +000010153
Victor Stinner8c62be82010-05-06 00:08:46 +000010154 if (all_ins(m))
10155 return NULL;
Barry Warsaw4a342091996-12-19 23:50:02 +000010156
Victor Stinner8c62be82010-05-06 00:08:46 +000010157 if (setup_confname_tables(m))
10158 return NULL;
Fred Drakebec628d1999-12-15 18:31:10 +000010159
Victor Stinner8c62be82010-05-06 00:08:46 +000010160 Py_INCREF(PyExc_OSError);
10161 PyModule_AddObject(m, "error", PyExc_OSError);
Fred Drake762e2061999-08-26 17:23:54 +000010162
Guido van Rossumb3d39562000-01-31 18:41:26 +000010163#ifdef HAVE_PUTENV
Victor Stinner8c62be82010-05-06 00:08:46 +000010164 if (posix_putenv_garbage == NULL)
10165 posix_putenv_garbage = PyDict_New();
Guido van Rossumb3d39562000-01-31 18:41:26 +000010166#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +000010167
Victor Stinner8c62be82010-05-06 00:08:46 +000010168 if (!initialized) {
Ross Lagerwall7807c352011-03-17 20:20:30 +020010169#if defined(HAVE_WAITID) && !defined(__APPLE__)
10170 waitid_result_desc.name = MODNAME ".waitid_result";
10171 PyStructSequence_InitType(&WaitidResultType, &waitid_result_desc);
10172#endif
10173
Victor Stinner8c62be82010-05-06 00:08:46 +000010174 stat_result_desc.name = MODNAME ".stat_result";
10175 stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
10176 stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
10177 stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
10178 PyStructSequence_InitType(&StatResultType, &stat_result_desc);
10179 structseq_new = StatResultType.tp_new;
10180 StatResultType.tp_new = statresult_new;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010181
Victor Stinner8c62be82010-05-06 00:08:46 +000010182 statvfs_result_desc.name = MODNAME ".statvfs_result";
10183 PyStructSequence_InitType(&StatVFSResultType, &statvfs_result_desc);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000010184#ifdef NEED_TICKS_PER_SECOND
10185# if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
Victor Stinner8c62be82010-05-06 00:08:46 +000010186 ticks_per_second = sysconf(_SC_CLK_TCK);
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000010187# elif defined(HZ)
Victor Stinner8c62be82010-05-06 00:08:46 +000010188 ticks_per_second = HZ;
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000010189# else
Victor Stinner8c62be82010-05-06 00:08:46 +000010190 ticks_per_second = 60; /* magic fallback value; may be bogus */
Martin v. Löwis05bfe1f2008-12-29 18:21:47 +000010191# endif
10192#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010193 }
Ross Lagerwall7807c352011-03-17 20:20:30 +020010194#if defined(HAVE_WAITID) && !defined(__APPLE__)
10195 Py_INCREF((PyObject*) &WaitidResultType);
10196 PyModule_AddObject(m, "waitid_result", (PyObject*) &WaitidResultType);
10197#endif
Victor Stinner8c62be82010-05-06 00:08:46 +000010198 Py_INCREF((PyObject*) &StatResultType);
10199 PyModule_AddObject(m, "stat_result", (PyObject*) &StatResultType);
10200 Py_INCREF((PyObject*) &StatVFSResultType);
10201 PyModule_AddObject(m, "statvfs_result",
10202 (PyObject*) &StatVFSResultType);
10203 initialized = 1;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010204
10205#ifdef __APPLE__
Victor Stinner8c62be82010-05-06 00:08:46 +000010206 /*
10207 * Step 2 of weak-linking support on Mac OS X.
10208 *
10209 * The code below removes functions that are not available on the
10210 * currently active platform.
10211 *
10212 * This block allow one to use a python binary that was build on
10213 * OSX 10.4 on OSX 10.3, without loosing access to new APIs on
10214 * OSX 10.4.
10215 */
Thomas Wouters477c8d52006-05-27 19:21:47 +000010216#ifdef HAVE_FSTATVFS
Victor Stinner8c62be82010-05-06 00:08:46 +000010217 if (fstatvfs == NULL) {
10218 if (PyObject_DelAttrString(m, "fstatvfs") == -1) {
10219 return NULL;
10220 }
10221 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000010222#endif /* HAVE_FSTATVFS */
10223
10224#ifdef HAVE_STATVFS
Victor Stinner8c62be82010-05-06 00:08:46 +000010225 if (statvfs == NULL) {
10226 if (PyObject_DelAttrString(m, "statvfs") == -1) {
10227 return NULL;
10228 }
10229 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000010230#endif /* HAVE_STATVFS */
10231
10232# ifdef HAVE_LCHOWN
Victor Stinner8c62be82010-05-06 00:08:46 +000010233 if (lchown == NULL) {
10234 if (PyObject_DelAttrString(m, "lchown") == -1) {
10235 return NULL;
10236 }
10237 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000010238#endif /* HAVE_LCHOWN */
10239
10240
10241#endif /* __APPLE__ */
Victor Stinner8c62be82010-05-06 00:08:46 +000010242 return m;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010243
Guido van Rossumb6775db1994-08-01 11:34:53 +000010244}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010245
10246#ifdef __cplusplus
10247}
10248#endif