mirror of https://github.com/python/cpython.git
559 lines
9.9 KiB
C
559 lines
9.9 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.
|
|
|
|
******************************************************************/
|
|
|
|
/* File object implementation */
|
|
|
|
#include "allobjects.h"
|
|
#include "modsupport.h"
|
|
#include "ceval.h"
|
|
|
|
#define BUF(v) GETSTRINGVALUE((stringobject *)v)
|
|
|
|
#include "errno.h"
|
|
#ifndef errno
|
|
extern int errno;
|
|
#endif
|
|
|
|
typedef struct {
|
|
OB_HEAD
|
|
FILE *f_fp;
|
|
object *f_name;
|
|
object *f_mode;
|
|
int (*f_close) PROTO((FILE *));
|
|
int f_softspace; /* Flag used by 'print' command */
|
|
} fileobject;
|
|
|
|
FILE *
|
|
getfilefile(f)
|
|
object *f;
|
|
{
|
|
if (!is_fileobject(f) || ((fileobject *)f)->f_fp == NULL) {
|
|
err_badcall();
|
|
return NULL;
|
|
}
|
|
return ((fileobject *)f)->f_fp;
|
|
}
|
|
|
|
object *
|
|
newopenfileobject(fp, name, mode, close)
|
|
FILE *fp;
|
|
char *name;
|
|
char *mode;
|
|
int (*close) FPROTO((FILE *));
|
|
{
|
|
fileobject *f = NEWOBJ(fileobject, &Filetype);
|
|
if (f == NULL)
|
|
return NULL;
|
|
f->f_fp = NULL;
|
|
f->f_name = newstringobject(name);
|
|
f->f_mode = newstringobject(mode);
|
|
f->f_close = close;
|
|
f->f_softspace = 0;
|
|
if (f->f_name == NULL || f->f_mode == NULL) {
|
|
DECREF(f);
|
|
return NULL;
|
|
}
|
|
f->f_fp = fp;
|
|
return (object *) f;
|
|
}
|
|
|
|
object *
|
|
newfileobject(name, mode)
|
|
char *name, *mode;
|
|
{
|
|
extern int fclose PROTO((FILE *));
|
|
fileobject *f;
|
|
f = (fileobject *) newopenfileobject((FILE *)NULL, name, mode, fclose);
|
|
if (f == NULL)
|
|
return NULL;
|
|
#ifdef THINK_C
|
|
if (*mode == '*') {
|
|
FILE *fopenRF();
|
|
f->f_fp = fopenRF(name, mode+1);
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
BGN_SAVE
|
|
f->f_fp = fopen(name, mode);
|
|
END_SAVE
|
|
}
|
|
if (f->f_fp == NULL) {
|
|
err_errno(IOError);
|
|
DECREF(f);
|
|
return NULL;
|
|
}
|
|
return (object *)f;
|
|
}
|
|
|
|
static object *
|
|
err_closed()
|
|
{
|
|
err_setstr(ValueError, "I/O operation on closed file");
|
|
return NULL;
|
|
}
|
|
|
|
/* Methods */
|
|
|
|
static void
|
|
file_dealloc(f)
|
|
fileobject *f;
|
|
{
|
|
if (f->f_fp != NULL && f->f_close != NULL) {
|
|
BGN_SAVE
|
|
(*f->f_close)(f->f_fp);
|
|
END_SAVE
|
|
}
|
|
if (f->f_name != NULL)
|
|
DECREF(f->f_name);
|
|
if (f->f_mode != NULL)
|
|
DECREF(f->f_mode);
|
|
free((char *)f);
|
|
}
|
|
|
|
static int
|
|
file_print(f, fp, flags)
|
|
fileobject *f;
|
|
FILE *fp;
|
|
int flags;
|
|
{
|
|
fprintf(fp, "<%s file ", f->f_fp == NULL ? "closed" : "open");
|
|
if (printobject(f->f_name, fp, flags) != 0)
|
|
return -1;
|
|
fprintf(fp, ", mode ");
|
|
if (printobject(f->f_mode, fp, flags) != 0)
|
|
return -1;
|
|
fprintf(fp, ">");
|
|
return 0;
|
|
}
|
|
|
|
static object *
|
|
file_repr(f)
|
|
fileobject *f;
|
|
{
|
|
char buf[300];
|
|
/* XXX This differs from file_print if the filename contains
|
|
quotes or other funny characters. */
|
|
sprintf(buf, "<%s file '%.256s', mode '%.10s'>",
|
|
f->f_fp == NULL ? "closed" : "open",
|
|
getstringvalue(f->f_name),
|
|
getstringvalue(f->f_mode));
|
|
return newstringobject(buf);
|
|
}
|
|
|
|
static object *
|
|
file_close(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
int sts = 0;
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
if (f->f_fp != NULL) {
|
|
if (f->f_close != NULL) {
|
|
BGN_SAVE
|
|
errno = 0;
|
|
sts = (*f->f_close)(f->f_fp);
|
|
END_SAVE
|
|
}
|
|
f->f_fp = NULL;
|
|
}
|
|
if (sts == EOF)
|
|
return err_errno(IOError);
|
|
if (sts != 0)
|
|
return newintobject((long)sts);
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static object *
|
|
file_seek(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
long offset;
|
|
int whence;
|
|
int ret;
|
|
|
|
if (f->f_fp == NULL)
|
|
return err_closed();
|
|
whence = 0;
|
|
if (!getargs(args, "l", &offset)) {
|
|
err_clear();
|
|
if (!getargs(args, "(li)", &offset, &whence))
|
|
return NULL;
|
|
}
|
|
BGN_SAVE
|
|
errno = 0;
|
|
ret = fseek(f->f_fp, offset, whence);
|
|
END_SAVE
|
|
if (ret != 0) {
|
|
err_errno(IOError);
|
|
clearerr(f->f_fp);
|
|
return NULL;
|
|
}
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static object *
|
|
file_tell(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
long offset;
|
|
if (f->f_fp == NULL)
|
|
return err_closed();
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
BGN_SAVE
|
|
errno = 0;
|
|
offset = ftell(f->f_fp);
|
|
END_SAVE
|
|
if (offset == -1L) {
|
|
err_errno(IOError);
|
|
clearerr(f->f_fp);
|
|
return NULL;
|
|
}
|
|
return newintobject(offset);
|
|
}
|
|
|
|
static object *
|
|
file_fileno(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
if (f->f_fp == NULL)
|
|
return err_closed();
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
return newintobject((long) fileno(f->f_fp));
|
|
}
|
|
|
|
static object *
|
|
file_flush(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
int res;
|
|
|
|
if (f->f_fp == NULL)
|
|
return err_closed();
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
BGN_SAVE
|
|
errno = 0;
|
|
res = fflush(f->f_fp);
|
|
END_SAVE
|
|
if (res != 0) {
|
|
err_errno(IOError);
|
|
clearerr(f->f_fp);
|
|
return NULL;
|
|
}
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static object *
|
|
file_isatty(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
long res;
|
|
if (f->f_fp == NULL)
|
|
return err_closed();
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
BGN_SAVE
|
|
res = isatty((int)fileno(f->f_fp));
|
|
END_SAVE
|
|
return newintobject(res);
|
|
}
|
|
|
|
static object *
|
|
file_read(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
int n, n1, n2, n3;
|
|
object *v;
|
|
|
|
if (f->f_fp == NULL)
|
|
return err_closed();
|
|
if (args == NULL) {
|
|
n = 0;
|
|
if (n < 0) {
|
|
err_setstr(ValueError, "negative read count");
|
|
return NULL;
|
|
}
|
|
}
|
|
else if (!getargs(args, "i", &n))
|
|
return NULL;
|
|
|
|
n2 = n != 0 ? n : BUFSIZ;
|
|
v = newsizedstringobject((char *)NULL, n2);
|
|
if (v == NULL)
|
|
return NULL;
|
|
n1 = 0;
|
|
BGN_SAVE
|
|
for (;;) {
|
|
n3 = fread(BUF(v)+n1, 1, n2-n1, f->f_fp);
|
|
/* XXX Error check? */
|
|
if (n3 == 0)
|
|
break;
|
|
n1 += n3;
|
|
if (n1 == n)
|
|
break;
|
|
if (n == 0) {
|
|
n2 = n1 + BUFSIZ;
|
|
RET_SAVE
|
|
if (resizestring(&v, n2) < 0)
|
|
return NULL;
|
|
RES_SAVE
|
|
}
|
|
}
|
|
END_SAVE
|
|
if (n1 != n2)
|
|
resizestring(&v, n1);
|
|
return v;
|
|
}
|
|
|
|
/* Internal routine to get a line.
|
|
Size argument interpretation:
|
|
> 0: max length;
|
|
= 0: read arbitrary line;
|
|
< 0: strip trailing '\n', raise EOFError if EOF reached immediately
|
|
*/
|
|
|
|
object *
|
|
getline(f, n)
|
|
fileobject *f;
|
|
int n;
|
|
{
|
|
register FILE *fp;
|
|
register int c;
|
|
register char *buf, *end;
|
|
int n1, n2;
|
|
object *v;
|
|
|
|
fp = f->f_fp;
|
|
n2 = n > 0 ? n : 100;
|
|
v = newsizedstringobject((char *)NULL, n2);
|
|
if (v == NULL)
|
|
return NULL;
|
|
buf = BUF(v);
|
|
end = buf + n2;
|
|
|
|
BGN_SAVE
|
|
for (;;) {
|
|
if ((c = getc(fp)) == EOF) {
|
|
clearerr(fp);
|
|
if (intrcheck()) {
|
|
RET_SAVE
|
|
DECREF(v);
|
|
err_set(KeyboardInterrupt);
|
|
return NULL;
|
|
}
|
|
if (n < 0 && buf == BUF(v)) {
|
|
RET_SAVE
|
|
DECREF(v);
|
|
err_setstr(EOFError,
|
|
"EOF when reading a line");
|
|
return NULL;
|
|
}
|
|
break;
|
|
}
|
|
if ((*buf++ = c) == '\n') {
|
|
if (n < 0)
|
|
buf--;
|
|
break;
|
|
}
|
|
if (buf == end) {
|
|
if (n > 0)
|
|
break;
|
|
n1 = n2;
|
|
n2 += 1000;
|
|
RET_SAVE
|
|
if (resizestring(&v, n2) < 0)
|
|
return NULL;
|
|
RES_SAVE
|
|
buf = BUF(v) + n1;
|
|
end = BUF(v) + n2;
|
|
}
|
|
}
|
|
END_SAVE
|
|
|
|
n1 = buf - BUF(v);
|
|
if (n1 != n2)
|
|
resizestring(&v, n1);
|
|
return v;
|
|
}
|
|
|
|
/* External C interface */
|
|
|
|
object *
|
|
filegetline(f, n)
|
|
object *f;
|
|
int n;
|
|
{
|
|
if (f == NULL || !is_fileobject(f)) {
|
|
err_badcall();
|
|
return NULL;
|
|
}
|
|
if (((fileobject*)f)->f_fp == NULL)
|
|
return err_closed();
|
|
return getline((fileobject *)f, n);
|
|
}
|
|
|
|
/* Python method */
|
|
|
|
static object *
|
|
file_readline(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
int n;
|
|
|
|
if (f->f_fp == NULL)
|
|
return err_closed();
|
|
if (args == NULL)
|
|
n = 0; /* Unlimited */
|
|
else {
|
|
if (!getintarg(args, &n))
|
|
return NULL;
|
|
if (n < 0) {
|
|
err_setstr(ValueError, "negative readline count");
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
return getline(f, n);
|
|
}
|
|
|
|
static object *
|
|
file_readlines(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
object *list;
|
|
object *line;
|
|
|
|
if (f->f_fp == NULL)
|
|
return err_closed();
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
if ((list = newlistobject(0)) == NULL)
|
|
return NULL;
|
|
for (;;) {
|
|
line = getline(f, 0);
|
|
if (line != NULL && getstringsize(line) == 0) {
|
|
DECREF(line);
|
|
break;
|
|
}
|
|
if (line == NULL || addlistitem(list, line) != 0) {
|
|
DECREF(list);
|
|
XDECREF(line);
|
|
return NULL;
|
|
}
|
|
DECREF(line);
|
|
}
|
|
return list;
|
|
}
|
|
|
|
static object *
|
|
file_write(f, args)
|
|
fileobject *f;
|
|
object *args;
|
|
{
|
|
char *s;
|
|
int n, n2;
|
|
if (f->f_fp == NULL)
|
|
return err_closed();
|
|
if (!getargs(args, "s#", &s, &n))
|
|
return NULL;
|
|
f->f_softspace = 0;
|
|
BGN_SAVE
|
|
errno = 0;
|
|
n2 = fwrite(s, 1, n, f->f_fp);
|
|
END_SAVE
|
|
if (n2 != n) {
|
|
err_errno(IOError);
|
|
clearerr(f->f_fp);
|
|
return NULL;
|
|
}
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static struct methodlist file_methods[] = {
|
|
{"close", file_close},
|
|
{"flush", file_flush},
|
|
{"fileno", file_fileno},
|
|
{"isatty", file_isatty},
|
|
{"read", file_read},
|
|
{"readline", file_readline},
|
|
{"readlines", file_readlines},
|
|
{"seek", file_seek},
|
|
{"tell", file_tell},
|
|
{"write", file_write},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
static object *
|
|
file_getattr(f, name)
|
|
fileobject *f;
|
|
char *name;
|
|
{
|
|
return findmethod(file_methods, (object *)f, name);
|
|
}
|
|
|
|
typeobject Filetype = {
|
|
OB_HEAD_INIT(&Typetype)
|
|
0,
|
|
"file",
|
|
sizeof(fileobject),
|
|
0,
|
|
file_dealloc, /*tp_dealloc*/
|
|
file_print, /*tp_print*/
|
|
file_getattr, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
0, /*tp_compare*/
|
|
file_repr, /*tp_repr*/
|
|
};
|
|
|
|
/* Interface for the 'soft space' between print items. */
|
|
|
|
int
|
|
softspace(f, newflag)
|
|
object *f;
|
|
int newflag;
|
|
{
|
|
int oldflag = 0;
|
|
if (f != NULL && is_fileobject(f)) {
|
|
oldflag = ((fileobject *)f)->f_softspace;
|
|
((fileobject *)f)->f_softspace = newflag;
|
|
}
|
|
return oldflag;
|
|
}
|