0
0
mirror of https://github.com/python/cpython.git synced 2024-12-01 11:15:56 +01:00
cpython/Modules/posixmodule.c
Guido van Rossum 7066dd75c5 * Makefile: added IMGFILE; moved some stuff around.
* flmodule.c: added some missing functions; changed readonly flags of
  some data members based upon FORMS documentation.
* listobject.c: fixed int/long arg lint bug (bites PC compilers).
* several: removed redundant print methods (repr is good enough).
* posixmodule.c: added (still experimental) process group functions.
1992-09-17 17:54:56 +00:00

1044 lines
18 KiB
C

/***********************************************************
Copyright 1991, 1992 by Stichting Mathematisch Centrum, Amsterdam, The
Netherlands.
All Rights Reserved
Permission to use, copy, modify, and distribute this software and its
documentation for any purpose and without fee is hereby granted,
provided that the above copyright notice appear in all copies and that
both that copyright notice and this permission notice appear in
supporting documentation, and that the names of Stichting Mathematisch
Centrum or CWI not be used in advertising or publicity pertaining to
distribution of the software without specific, written prior permission.
STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO
THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE
FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
******************************************************************/
/* POSIX module implementation */
#ifdef AMOEBA
#define NO_LSTAT
#define SYSV
#endif
#ifdef MSDOS
#define NO_LSTAT
#define NO_UNAME
#endif
#ifdef __sgi
#define DO_PG
#endif
#include <signal.h>
#include <string.h>
#include <setjmp.h>
#include <sys/types.h>
#include <sys/stat.h>
#ifdef DO_TIMES
#include <sys/times.h>
#include <sys/param.h>
#include <errno.h>
#endif
#ifdef SYSV
#define UTIME_STRUCT
#include <dirent.h>
#define direct dirent
#ifdef i386
#define mode_t int
#endif
#else /* !SYSV */
#ifndef MSDOS
#include <sys/dir.h>
#endif
#endif /* !SYSV */
#ifndef NO_UNISTD
#include <unistd.h> /* Take this out and hope the best if it doesn't exist */
#endif
#include "allobjects.h"
#include "modsupport.h"
#include "ceval.h"
extern char *strerror PROTO((int));
/* Return a dictionary corresponding to the POSIX environment table */
extern char **environ;
static object *
convertenviron()
{
object *d;
char **e;
d = newdictobject();
if (d == NULL)
return NULL;
if (environ == NULL)
return d;
/* XXX This part ignores errors */
for (e = environ; *e != NULL; e++) {
object *v;
char *p = strchr(*e, '=');
if (p == NULL)
continue;
v = newstringobject(p+1);
if (v == NULL)
continue;
*p = '\0';
(void) dictinsert(d, *e, v);
*p = '=';
DECREF(v);
}
return d;
}
static object *PosixError; /* Exception posix.error */
/* Set a POSIX-specific error from errno, and return NULL */
static object *
posix_error()
{
return err_errno(PosixError);
}
/* POSIX generic methods */
static object *
posix_1str(args, func)
object *args;
int (*func) FPROTO((const char *));
{
char *path1;
int res;
if (!getstrarg(args, &path1))
return NULL;
BGN_SAVE
res = (*func)(path1);
END_SAVE
if (res < 0)
return posix_error();
INCREF(None);
return None;
}
static object *
posix_2str(args, func)
object *args;
int (*func) FPROTO((const char *, const char *));
{
char *path1, *path2;
int res;
if (!getstrstrarg(args, &path1, &path2))
return NULL;
BGN_SAVE
res = (*func)(path1, path2);
END_SAVE
if (res < 0)
return posix_error();
INCREF(None);
return None;
}
static object *
posix_strint(args, func)
object *args;
int (*func) FPROTO((const char *, int));
{
char *path;
int i;
int res;
if (!getstrintarg(args, &path, &i))
return NULL;
BGN_SAVE
res = (*func)(path, i);
END_SAVE
if (res < 0)
return posix_error();
INCREF(None);
return None;
}
static object *
posix_do_stat(self, args, statfunc)
object *self;
object *args;
int (*statfunc) FPROTO((const char *, struct stat *));
{
struct stat st;
char *path;
object *v;
int res;
if (!getstrarg(args, &path))
return NULL;
BGN_SAVE
res = (*statfunc)(path, &st);
END_SAVE
if (res != 0)
return posix_error();
v = newtupleobject(10);
if (v == NULL)
return NULL;
#define SET(i, st_member) settupleitem(v, i, newintobject((long)st.st_member))
SET(0, st_mode);
SET(1, st_ino);
SET(2, st_dev);
SET(3, st_nlink);
SET(4, st_uid);
SET(5, st_gid);
SET(6, st_size);
SET(7, st_atime);
SET(8, st_mtime);
SET(9, st_ctime);
#undef SET
if (err_occurred()) {
DECREF(v);
return NULL;
}
return v;
}
/* POSIX methods */
static object *
posix_chdir(self, args)
object *self;
object *args;
{
extern int chdir PROTO((const char *));
return posix_1str(args, chdir);
}
static object *
posix_chmod(self, args)
object *self;
object *args;
{
extern int chmod PROTO((const char *, mode_t));
return posix_strint(args, chmod);
}
static object *
posix_getcwd(self, args)
object *self;
object *args;
{
char buf[1026];
char *res;
extern char *getcwd PROTO((char *, int));
if (!getnoarg(args))
return NULL;
BGN_SAVE
res = getcwd(buf, sizeof buf);
END_SAVE
if (res == NULL)
return posix_error();
return newstringobject(buf);
}
#ifndef MSDOS
static object *
posix_link(self, args)
object *self;
object *args;
{
extern int link PROTO((const char *, const char *));
return posix_2str(args, link);
}
#endif /* !MSDOS */
static object *
posix_listdir(self, args)
object *self;
object *args;
{
char *name;
object *d, *v;
#ifdef MSDOS
struct ffblk ep;
int rv;
if (!getstrarg(args, &name))
return NULL;
if (findfirst(name, &ep, 0) == -1)
return posix_error();
if ((d = newlistobject(0)) == NULL)
return NULL;
do {
v = newstringobject(ep.ff_name);
if (v == NULL) {
DECREF(d);
d = NULL;
break;
}
if (addlistitem(d, v) != 0) {
DECREF(v);
DECREF(d);
d = NULL;
break;
}
DECREF(v);
} while ((rv = findnext(&ep)) == 0);
#else /* !MSDOS */
DIR *dirp;
struct direct *ep;
if (!getstrarg(args, &name))
return NULL;
BGN_SAVE
if ((dirp = opendir(name)) == NULL) {
RET_SAVE
return posix_error();
}
if ((d = newlistobject(0)) == NULL) {
closedir(dirp);
RET_SAVE
return NULL;
}
while ((ep = readdir(dirp)) != NULL) {
v = newstringobject(ep->d_name);
if (v == NULL) {
DECREF(d);
d = NULL;
break;
}
if (addlistitem(d, v) != 0) {
DECREF(v);
DECREF(d);
d = NULL;
break;
}
DECREF(v);
}
closedir(dirp);
END_SAVE
#endif /* !MSDOS */
return d;
}
static object *
posix_mkdir(self, args)
object *self;
object *args;
{
extern int mkdir PROTO((const char *, mode_t));
return posix_strint(args, mkdir);
}
#ifdef i386
int
rename(from, to)
char *from;
char *to;
{
int status;
/* XXX Shouldn't this unlink the destination first? */
status = link(from, to);
if (status != 0)
return status;
return unlink(from);
}
#endif /* i386 */
static object *
posix_rename(self, args)
object *self;
object *args;
{
extern int rename PROTO((const char *, const char *));
return posix_2str(args, rename);
}
static object *
posix_rmdir(self, args)
object *self;
object *args;
{
extern int rmdir PROTO((const char *));
return posix_1str(args, rmdir);
}
static object *
posix_stat(self, args)
object *self;
object *args;
{
extern int stat PROTO((const char *, struct stat *));
return posix_do_stat(self, args, stat);
}
static object *
posix_system(self, args)
object *self;
object *args;
{
char *command;
long sts;
if (!getstrarg(args, &command))
return NULL;
BGN_SAVE
sts = system(command);
END_SAVE
return newintobject(sts);
}
#ifndef MSDOS
static object *
posix_umask(self, args)
object *self;
object *args;
{
int i;
if (!getintarg(args, &i))
return NULL;
i = umask(i);
if (i < 0)
return posix_error();
return newintobject((long)i);
}
#endif /* !MSDOS */
static object *
posix_unlink(self, args)
object *self;
object *args;
{
extern int unlink PROTO((const char *));
return posix_1str(args, unlink);
}
#ifndef NO_UNAME
#include <sys/utsname.h>
static object *
posix_uname(self, args)
object *self;
object *args;
{
extern int uname PROTO((struct utsname *));
struct utsname u;
object *v;
int res;
if (!getnoarg(args))
return NULL;
BGN_SAVE
res = uname(&u);
END_SAVE
if (res < 0)
return posix_error();
v = newtupleobject(5);
if (v == NULL)
return NULL;
#define SET(i, member) settupleitem(v, i, newstringobject(u.member))
SET(0, sysname);
SET(1, nodename);
SET(2, release);
SET(3, version);
SET(4, machine);
#undef SET
if (err_occurred()) {
DECREF(v);
return NULL;
}
return v;
}
#endif /* NO_UNAME */
#ifdef UTIME_STRUCT
#include <utime.h>
#endif
static object *
posix_utime(self, args)
object *self;
object *args;
{
char *path;
int res;
#ifdef UTIME_STRUCT
struct utimbuf buf;
#define ATIME buf.actime
#define MTIME buf.modtime
#define UTIME_ARG &buf
#else
time_t buf[2];
#define ATIME buf[0]
#define MTIME buf[1]
#define UTIME_ARG buf
#endif
if (!getargs(args, "(s(ll))", &path, &ATIME, &MTIME))
return NULL;
BGN_SAVE
res = utime(path, UTIME_ARG);
END_SAVE
if (res < 0)
return posix_error();
INCREF(None);
return None;
#undef UTIME_ARG
#undef ATIME
#undef MTIME
}
#ifndef MSDOS
/* Process operations */
static object *
posix__exit(self, args)
object *self;
object *args;
{
int sts;
if (!getintarg(args, &sts))
return NULL;
_exit(sts);
/* NOTREACHED */
}
/* XXX To do: exece, execp */
static object *
posix_exec(self, args)
object *self;
object *args;
{
char *path;
object *argv;
char **argvlist;
int i, argc;
object *(*getitem) PROTO((object *, int));
/* exec has two arguments: (path, argv), where
argv is a list or tuple of strings. */
if (!getargs(args, "(sO)", &path, &argv))
return NULL;
if (is_listobject(argv)) {
argc = getlistsize(argv);
getitem = getlistitem;
}
else if (is_tupleobject(argv)) {
argc = gettuplesize(argv);
getitem = gettupleitem;
}
else {
badarg:
err_badarg();
return NULL;
}
argvlist = NEW(char *, argc+1);
if (argvlist == NULL)
return NULL;
for (i = 0; i < argc; i++) {
if (!getstrarg((*getitem)(argv, i), &argvlist[i])) {
DEL(argvlist);
goto badarg;
}
}
argvlist[argc] = NULL;
execv(path, argvlist);
/* If we get here it's definitely an error */
DEL(argvlist);
return posix_error();
}
static object *
posix_fork(self, args)
object *self;
object *args;
{
int pid;
if (!getnoarg(args))
return NULL;
pid = fork();
if (pid == -1)
return posix_error();
return newintobject((long)pid);
}
static object *
posix_getegid(self, args)
object *self;
object *args;
{
if (!getnoarg(args))
return NULL;
return newintobject((long)getegid());
}
static object *
posix_geteuid(self, args)
object *self;
object *args;
{
if (!getnoarg(args))
return NULL;
return newintobject((long)geteuid());
}
static object *
posix_getgid(self, args)
object *self;
object *args;
{
if (!getnoarg(args))
return NULL;
return newintobject((long)getgid());
}
static object *
posix_getpid(self, args)
object *self;
object *args;
{
if (!getnoarg(args))
return NULL;
return newintobject((long)getpid());
}
static object *
posix_getpgrp(self, args)
object *self;
object *args;
{
if (!getnoarg(args))
return NULL;
#ifdef SYSV
return newintobject((long)getpgrp());
#else
return newintobject((long)getpgrp(0));
#endif
}
static object *
posix_setpgrp(self, args)
object *self;
object *args;
{
if (!getnoarg(args))
return NULL;
#ifdef SYSV
if (setpgrp() < 0)
#else
if (setpgrp(0, 0) < 0)
#endif
return err_errno(PosixError);
INCREF(None);
return None;
}
static object *
posix_getppid(self, args)
object *self;
object *args;
{
if (!getnoarg(args))
return NULL;
return newintobject((long)getppid());
}
static object *
posix_getuid(self, args)
object *self;
object *args;
{
if (!getnoarg(args))
return NULL;
return newintobject((long)getuid());
}
static object *
posix_kill(self, args)
object *self;
object *args;
{
int pid, sig;
if (!getintintarg(args, &pid, &sig))
return NULL;
if (kill(pid, sig) == -1)
return posix_error();
INCREF(None);
return None;
}
static object *
posix_popen(self, args)
object *self;
object *args;
{
extern int pclose PROTO((FILE *));
char *name, *mode;
FILE *fp;
if (!getargs(args, "(ss)", &name, &mode))
return NULL;
BGN_SAVE
fp = popen(name, mode);
END_SAVE
if (fp == NULL)
return posix_error();
/* From now on, ignore SIGPIPE and let the error checking
do the work. */
(void) signal(SIGPIPE, SIG_IGN);
return newopenfileobject(fp, name, mode, pclose);
}
static object *
posix_waitpid(self, args)
object *self;
object *args;
{
#ifdef NO_WAITPID
err_setstr(PosixError,
"posix.waitpid() not supported on this system");
return NULL;
#else
int pid, options, sts;
if (!getargs(args, "(ii)", &pid, &options))
return NULL;
BGN_SAVE
pid = waitpid(pid, &sts, options);
END_SAVE
if (pid == -1)
return posix_error();
else
return mkvalue("ii", pid, sts);
#endif
}
static object *
posix_wait(self, args)
object *self;
object *args;
{
int pid, sts;
if (args != NULL)
return posix_waitpid(self, args); /* BW compat */
BGN_SAVE
pid = wait(&sts);
END_SAVE
if (pid == -1)
return posix_error();
else
return mkvalue("ii", pid, sts);
}
#endif /* MSDOS */
static object *
posix_lstat(self, args)
object *self;
object *args;
{
#ifdef NO_LSTAT
#define lstat stat
#endif
extern int lstat PROTO((const char *, struct stat *));
return posix_do_stat(self, args, lstat);
}
static object *
posix_readlink(self, args)
object *self;
object *args;
{
#ifdef NO_LSTAT
err_setstr(PosixError, "readlink not implemented on this system");
return NULL;
#else
char buf[1024]; /* XXX Should use MAXPATHLEN */
char *path;
int n;
if (!getstrarg(args, &path))
return NULL;
BGN_SAVE
n = readlink(path, buf, (int) sizeof buf);
END_SAVE
if (n < 0)
return posix_error();
return newsizedstringobject(buf, n);
#endif
}
static object *
posix_symlink(self, args)
object *self;
object *args;
{
#ifdef NO_LSTAT
err_setstr(PosixError, "symlink not implemented on this system");
return NULL;
#else
extern int symlink PROTO((const char *, const char *));
return posix_2str(args, symlink);
#endif
}
#ifdef DO_TIMES
static object *
posix_times(self, args)
object *self;
object *args;
{
struct tms t;
clock_t c;
object *tuple;
if (!getnoarg(args))
return NULL;
errno = 0;
c = times(&t);
if (c == (clock_t) -1) {
err_errno(PosixError);
return NULL;
}
tuple = newtupleobject(4);
if (tuple == NULL)
return NULL;
settupleitem(tuple, 0, newfloatobject((double)t.tms_utime / HZ));
settupleitem(tuple, 1, newfloatobject((double)t.tms_stime / HZ));
settupleitem(tuple, 2, newfloatobject((double)t.tms_cutime / HZ));
settupleitem(tuple, 3, newfloatobject((double)t.tms_cstime / HZ));
if (err_occurred()) {
DECREF(tuple);
return NULL;
}
return tuple;
}
#endif /* DO_TIMES */
#ifdef DO_PG
static object *
posix_setsid(self, args)
object *self;
object *args;
{
if (!getnoarg(args))
return NULL;
if (setsid() < 0) {
err_errno(PosixError);
return NULL;
}
INCREF(None);
return None;
}
static object *
posix_setpgid(self, args)
object *self;
object *args;
{
int pid, pgrp;
if (!getargs(args, "(ii)", &pid, &pgrp))
return NULL;
if (setpgid(pid, pgrp) < 0) {
err_errno(PosixError);
return NULL;
}
INCREF(None);
return None;
}
static object *
posix_tcgetpgrp(self, args)
object *self;
object *args;
{
int fd, pgid;
if (!getargs(args, "i", &fd))
return NULL;
pgid = tcgetpgrp(fd);
if (pgid < 0) {
err_errno(PosixError);
return NULL;
}
return newintobject((long)pgid);
}
static object *
posix_tcsetpgrp(self, args)
object *self;
object *args;
{
int fd, pgid;
if (!getargs(args, "(ii)", &fd, &pgid))
return NULL;
if (tcsetpgrp(fd, pgid) < 0) {
err_errno(PosixError);
return NULL;
}
INCREF(None);
return None;
}
#endif /* DO_PG */
static struct methodlist posix_methods[] = {
{"chdir", posix_chdir},
{"chmod", posix_chmod},
{"getcwd", posix_getcwd},
#ifndef MSDOS
{"link", posix_link},
#endif
{"listdir", posix_listdir},
{"lstat", posix_lstat},
{"mkdir", posix_mkdir},
{"readlink", posix_readlink},
{"rename", posix_rename},
{"rmdir", posix_rmdir},
{"stat", posix_stat},
{"symlink", posix_symlink},
{"system", posix_system},
#ifndef MSDOS
{"umask", posix_umask},
#endif
#ifndef NO_UNAME
{"uname", posix_uname},
#endif
{"unlink", posix_unlink},
{"utime", posix_utime},
#ifdef DO_TIMES
{"times", posix_times},
#endif
#ifndef MSDOS
{"_exit", posix__exit},
{"exec", posix_exec},
{"fork", posix_fork},
{"getegid", posix_getegid},
{"geteuid", posix_geteuid},
{"getgid", posix_getgid},
{"getpid", posix_getpid},
{"getpgrp", posix_getpgrp},
{"getppid", posix_getppid},
{"getuid", posix_getuid},
{"kill", posix_kill},
{"popen", posix_popen},
{"setpgrp", posix_setpgrp},
{"wait", posix_wait},
{"waitpid", posix_waitpid},
#endif
#ifdef DO_PG
{"setsid", posix_setsid},
{"setpgid", posix_setpgid},
{"tcgetpgrp", posix_tcgetpgrp},
{"tcsetpgrp", posix_tcsetpgrp},
#endif
{NULL, NULL} /* Sentinel */
};
void
initposix()
{
object *m, *d, *v;
m = initmodule("posix", posix_methods);
d = getmoduledict(m);
/* Initialize posix.environ dictionary */
v = convertenviron();
if (v == NULL || dictinsert(d, "environ", v) != 0)
fatal("can't define posix.environ");
DECREF(v);
/* Initialize posix.error exception */
PosixError = newstringobject("posix.error");
if (PosixError == NULL || dictinsert(d, "error", PosixError) != 0)
fatal("can't define posix.error");
}
/* Function used elsewhere to get a file's modification time */
long
getmtime(path)
char *path;
{
struct stat st;
if (stat(path, &st) != 0)
return -1;
else
return st.st_mtime;
}
#ifdef MSDOS
/* A small "compatibility library" for TurboC under MS-DOS */
#include <sir.h>
#include <io.h>
#include <dos.h>
#include <fcntl.h>
int
chmod(path, mode)
char *path;
int mode;
{
return _chmod(path, 1, mode);
}
int
utime(path, times)
char *path;
time_t times[2];
{
struct date dt;
struct time tm;
struct ftime dft;
int fh;
unixtodos(tv[0].tv_sec,&dt,&tm);
dft.ft_tsec = tm.ti_sec; dft.ft_min = tm.ti_min;
dft.ft_hour = tm.ti_hour; dft.ft_day = dt.da_day;
dft.ft_month = dt.da_mon;
dft.ft_year = (dt.da_year - 1980); /* this is for TC library */
if ((fh = open(path,O_RDWR)) < 0)
return posix_error(); /* can't open file to set time */
if (setftime(fh,&dft) < 0)
{
close(fh);
return posix_error();
}
close(fh); /* close the temp handle */
}
#endif /* MSDOS */