blob: 057b424c92e0a2f14cb2def39ee6d6409c1a6b5c [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001/***********************************************************
Guido van Rossum6610ad91995-01-04 19:07:38 +00002Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam,
3The Netherlands.
Guido van Rossumf70e43a1991-02-19 12:39:46 +00004
5 All Rights Reserved
6
Guido van Rossumd266eb41996-10-25 14:44:06 +00007Permission to use, copy, modify, and distribute this software and its
8documentation for any purpose and without fee is hereby granted,
Guido van Rossumf70e43a1991-02-19 12:39:46 +00009provided that the above copyright notice appear in all copies and that
Guido van Rossumd266eb41996-10-25 14:44:06 +000010both that copyright notice and this permission notice appear in
Guido van Rossumf70e43a1991-02-19 12:39:46 +000011supporting documentation, and that the names of Stichting Mathematisch
Guido van Rossumd266eb41996-10-25 14:44:06 +000012Centrum or CWI or Corporation for National Research Initiatives or
13CNRI not be used in advertising or publicity pertaining to
14distribution of the software without specific, written prior
15permission.
Guido van Rossumf70e43a1991-02-19 12:39:46 +000016
Guido van Rossumd266eb41996-10-25 14:44:06 +000017While CWI is the initial source for this software, a modified version
18is made available by the Corporation for National Research Initiatives
19(CNRI) at the Internet address ftp://ftp.python.org.
20
21STICHTING MATHEMATISCH CENTRUM AND CNRI DISCLAIM ALL WARRANTIES WITH
22REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF
23MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH
24CENTRUM OR CNRI BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL
25DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
26PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
27TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
28PERFORMANCE OF THIS SOFTWARE.
Guido van Rossumf70e43a1991-02-19 12:39:46 +000029
30******************************************************************/
31
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000032/* File object implementation */
33
Guido van Rossumc0b618a1997-05-02 03:12:38 +000034#include "Python.h"
Guido van Rossumb6775db1994-08-01 11:34:53 +000035#include "structmember.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000036
Guido van Rossum685a38e1996-12-05 21:54:17 +000037#ifdef HAVE_UNISTD_H
38#include <unistd.h>
39#endif
40
Guido van Rossumb8199141997-05-06 15:23:24 +000041#ifdef MS_WIN32
42#define ftruncate _chsize
43#define fileno _fileno
44#define HAVE_FTRUNCATE
45#endif
46
Guido van Rossum295d1711995-02-19 15:55:19 +000047#ifdef THINK_C
48#define HAVE_FOPENRF
49#endif
Jack Jansene08dea191995-04-23 22:12:47 +000050#ifdef __MWERKS__
51/* Mwerks fopen() doesn't always set errno */
52#define NO_FOPEN_ERRNO
53#endif
Guido van Rossum295d1711995-02-19 15:55:19 +000054
Guido van Rossumc0b618a1997-05-02 03:12:38 +000055#define BUF(v) PyString_AS_STRING((PyStringObject *)v)
Guido van Rossumce5ba841991-03-06 13:06:18 +000056
Guido van Rossumf1dc5661993-07-05 10:31:29 +000057#include <errno.h>
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000058
59typedef struct {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000060 PyObject_HEAD
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000061 FILE *f_fp;
Guido van Rossumc0b618a1997-05-02 03:12:38 +000062 PyObject *f_name;
63 PyObject *f_mode;
64 int (*f_close) Py_PROTO((FILE *));
Guido van Rossumeb183da1991-04-04 10:44:06 +000065 int f_softspace; /* Flag used by 'print' command */
Guido van Rossumc0b618a1997-05-02 03:12:38 +000066} PyFileObject;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000067
68FILE *
Guido van Rossumc0b618a1997-05-02 03:12:38 +000069PyFile_AsFile(f)
70 PyObject *f;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000071{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000072 if (f == NULL || !PyFile_Check(f))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000073 return NULL;
Guido van Rossum3165fe61992-09-25 21:59:05 +000074 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +000075 return ((PyFileObject *)f)->f_fp;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000076}
77
Guido van Rossumc0b618a1997-05-02 03:12:38 +000078PyObject *
79PyFile_Name(f)
80 PyObject *f;
Guido van Rossumdb3165e1993-10-18 17:06:59 +000081{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000082 if (f == NULL || !PyFile_Check(f))
Guido van Rossumdb3165e1993-10-18 17:06:59 +000083 return NULL;
84 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +000085 return ((PyFileObject *)f)->f_name;
Guido van Rossumdb3165e1993-10-18 17:06:59 +000086}
87
Guido van Rossumc0b618a1997-05-02 03:12:38 +000088PyObject *
89PyFile_FromFile(fp, name, mode, close)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000090 FILE *fp;
91 char *name;
92 char *mode;
Guido van Rossumc0b618a1997-05-02 03:12:38 +000093 int (*close) Py_FPROTO((FILE *));
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000094{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000095 PyFileObject *f = PyObject_NEW(PyFileObject, &PyFile_Type);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000096 if (f == NULL)
97 return NULL;
98 f->f_fp = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +000099 f->f_name = PyString_FromString(name);
100 f->f_mode = PyString_FromString(mode);
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000101 f->f_close = close;
Guido van Rossumeb183da1991-04-04 10:44:06 +0000102 f->f_softspace = 0;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000103 if (f->f_name == NULL || f->f_mode == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000104 Py_DECREF(f);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000105 return NULL;
106 }
107 f->f_fp = fp;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000108 return (PyObject *) f;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000109}
110
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000111PyObject *
112PyFile_FromString(name, mode)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000113 char *name, *mode;
114{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000115 extern int fclose Py_PROTO((FILE *));
116 PyFileObject *f;
117 f = (PyFileObject *) PyFile_FromFile((FILE *)NULL, name, mode, fclose);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000118 if (f == NULL)
119 return NULL;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000120#ifdef HAVE_FOPENRF
Guido van Rossuma08095a1991-02-13 23:25:27 +0000121 if (*mode == '*') {
122 FILE *fopenRF();
123 f->f_fp = fopenRF(name, mode+1);
124 }
125 else
126#endif
Guido van Rossumff4949e1992-08-05 19:58:53 +0000127 {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000128 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000129 f->f_fp = fopen(name, mode);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000130 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000131 }
Guido van Rossuma08095a1991-02-13 23:25:27 +0000132 if (f->f_fp == NULL) {
Jack Jansene08dea191995-04-23 22:12:47 +0000133#ifdef NO_FOPEN_ERRNO
134 if ( errno == 0 ) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000135 PyErr_SetString(PyExc_IOError, "Cannot open file");
136 Py_DECREF(f);
Jack Jansene08dea191995-04-23 22:12:47 +0000137 return NULL;
138 }
139#endif
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000140 PyErr_SetFromErrno(PyExc_IOError);
141 Py_DECREF(f);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000142 return NULL;
143 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000144 return (PyObject *)f;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000145}
146
Guido van Rossumb6775db1994-08-01 11:34:53 +0000147void
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000148PyFile_SetBufSize(f, bufsize)
149 PyObject *f;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000150 int bufsize;
151{
152 if (bufsize >= 0) {
153#ifdef HAVE_SETVBUF
154 int type;
155 switch (bufsize) {
156 case 0:
157 type = _IONBF;
158 break;
159 case 1:
160 type = _IOLBF;
161 bufsize = BUFSIZ;
162 break;
163 default:
164 type = _IOFBF;
165 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000166 setvbuf(((PyFileObject *)f)->f_fp, (char *)NULL,
167 type, bufsize);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000168#endif /* HAVE_SETVBUF */
169 }
170}
171
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000172static PyObject *
Guido van Rossumd7297e61992-07-06 14:19:26 +0000173err_closed()
174{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000175 PyErr_SetString(PyExc_ValueError, "I/O operation on closed file");
Guido van Rossumd7297e61992-07-06 14:19:26 +0000176 return NULL;
177}
178
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000179/* Methods */
180
181static void
Guido van Rossum3f5da241990-12-20 15:06:42 +0000182file_dealloc(f)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000183 PyFileObject *f;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000184{
Guido van Rossumff4949e1992-08-05 19:58:53 +0000185 if (f->f_fp != NULL && f->f_close != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000186 Py_BEGIN_ALLOW_THREADS
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000187 (*f->f_close)(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000188 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000189 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000190 if (f->f_name != NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000191 Py_DECREF(f->f_name);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000192 if (f->f_mode != NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000193 Py_DECREF(f->f_mode);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000194 free((char *)f);
195}
196
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000197static PyObject *
Guido van Rossum3f5da241990-12-20 15:06:42 +0000198file_repr(f)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000199 PyFileObject *f;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000200{
201 char buf[300];
Guido van Rossume35399e1993-01-10 18:33:56 +0000202 sprintf(buf, "<%s file '%.256s', mode '%.10s' at %lx>",
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000203 f->f_fp == NULL ? "closed" : "open",
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000204 PyString_AsString(f->f_name),
205 PyString_AsString(f->f_mode),
Guido van Rossume35399e1993-01-10 18:33:56 +0000206 (long)f);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000207 return PyString_FromString(buf);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000208}
209
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000210static PyObject *
Guido van Rossum3f5da241990-12-20 15:06:42 +0000211file_close(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000212 PyFileObject *f;
213 PyObject *args;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000214{
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000215 int sts = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000216 if (!PyArg_NoArgs(args))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000217 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000218 if (f->f_fp != NULL) {
Guido van Rossumff4949e1992-08-05 19:58:53 +0000219 if (f->f_close != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000220 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000221 errno = 0;
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000222 sts = (*f->f_close)(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000223 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000224 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000225 f->f_fp = NULL;
226 }
Guido van Rossumfebd5511992-03-04 16:39:24 +0000227 if (sts == EOF)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000228 return PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000229 if (sts != 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000230 return PyInt_FromLong((long)sts);
231 Py_INCREF(Py_None);
232 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000233}
234
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000235static PyObject *
Guido van Rossumce5ba841991-03-06 13:06:18 +0000236file_seek(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000237 PyFileObject *f;
238 PyObject *args;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000239{
Guido van Rossumce5ba841991-03-06 13:06:18 +0000240 long offset;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000241 int whence;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000242 int ret;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000243
Guido van Rossumd7297e61992-07-06 14:19:26 +0000244 if (f->f_fp == NULL)
245 return err_closed();
246 whence = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000247 if (!PyArg_Parse(args, "l", &offset)) {
248 PyErr_Clear();
249 if (!PyArg_Parse(args, "(li)", &offset, &whence))
Guido van Rossumce5ba841991-03-06 13:06:18 +0000250 return NULL;
251 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000252 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000253 errno = 0;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000254 ret = fseek(f->f_fp, offset, whence);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000255 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000256 if (ret != 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000257 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumfebd5511992-03-04 16:39:24 +0000258 clearerr(f->f_fp);
259 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000260 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000261 Py_INCREF(Py_None);
262 return Py_None;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000263}
264
Guido van Rossumd7047b31995-01-02 19:07:15 +0000265#ifdef HAVE_FTRUNCATE
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000266static PyObject *
Guido van Rossumd7047b31995-01-02 19:07:15 +0000267file_truncate(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000268 PyFileObject *f;
269 PyObject *args;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000270{
271 long newsize;
272 int ret;
273
274 if (f->f_fp == NULL)
275 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000276 if (!PyArg_Parse(args, "l", &newsize)) {
277 PyErr_Clear();
278 if (!PyArg_NoArgs(args))
Guido van Rossumd7047b31995-01-02 19:07:15 +0000279 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000280 Py_BEGIN_ALLOW_THREADS
Guido van Rossumd7047b31995-01-02 19:07:15 +0000281 errno = 0;
282 newsize = ftell(f->f_fp); /* default to current position*/
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000283 Py_END_ALLOW_THREADS
Guido van Rossumd7047b31995-01-02 19:07:15 +0000284 if (newsize == -1L) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000285 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumd7047b31995-01-02 19:07:15 +0000286 clearerr(f->f_fp);
287 return NULL;
288 }
289 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000290 Py_BEGIN_ALLOW_THREADS
Guido van Rossumd7047b31995-01-02 19:07:15 +0000291 errno = 0;
292 ret = fflush(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000293 Py_END_ALLOW_THREADS
Guido van Rossumd7047b31995-01-02 19:07:15 +0000294 if (ret == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000295 Py_BEGIN_ALLOW_THREADS
Guido van Rossumd7047b31995-01-02 19:07:15 +0000296 errno = 0;
297 ret = ftruncate(fileno(f->f_fp), newsize);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000298 Py_END_ALLOW_THREADS
Guido van Rossumd7047b31995-01-02 19:07:15 +0000299 }
300 if (ret != 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000301 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumd7047b31995-01-02 19:07:15 +0000302 clearerr(f->f_fp);
303 return NULL;
304 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000305 Py_INCREF(Py_None);
306 return Py_None;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000307}
308#endif /* HAVE_FTRUNCATE */
309
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000310static PyObject *
Guido van Rossumce5ba841991-03-06 13:06:18 +0000311file_tell(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000312 PyFileObject *f;
313 PyObject *args;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000314{
315 long offset;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000316 if (f->f_fp == NULL)
317 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000318 if (!PyArg_NoArgs(args))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000319 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000320 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000321 errno = 0;
322 offset = ftell(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000323 Py_END_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000324 if (offset == -1L) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000325 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumfebd5511992-03-04 16:39:24 +0000326 clearerr(f->f_fp);
327 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000328 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000329 return PyInt_FromLong(offset);
Guido van Rossumce5ba841991-03-06 13:06:18 +0000330}
331
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000332static PyObject *
Guido van Rossumed233a51992-06-23 09:07:03 +0000333file_fileno(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000334 PyFileObject *f;
335 PyObject *args;
Guido van Rossumed233a51992-06-23 09:07:03 +0000336{
Guido van Rossumd7297e61992-07-06 14:19:26 +0000337 if (f->f_fp == NULL)
338 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000339 if (!PyArg_NoArgs(args))
Guido van Rossumed233a51992-06-23 09:07:03 +0000340 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000341 return PyInt_FromLong((long) fileno(f->f_fp));
Guido van Rossumed233a51992-06-23 09:07:03 +0000342}
343
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000344static PyObject *
Guido van Rossumce5ba841991-03-06 13:06:18 +0000345file_flush(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000346 PyFileObject *f;
347 PyObject *args;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000348{
Guido van Rossumff4949e1992-08-05 19:58:53 +0000349 int res;
350
Guido van Rossumd7297e61992-07-06 14:19:26 +0000351 if (f->f_fp == NULL)
352 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000353 if (!PyArg_NoArgs(args))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000354 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000355 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000356 errno = 0;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000357 res = fflush(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000358 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000359 if (res != 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000360 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumfebd5511992-03-04 16:39:24 +0000361 clearerr(f->f_fp);
362 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000363 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000364 Py_INCREF(Py_None);
365 return Py_None;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000366}
367
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000368static PyObject *
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000369file_isatty(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000370 PyFileObject *f;
371 PyObject *args;
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000372{
Guido van Rossumff4949e1992-08-05 19:58:53 +0000373 long res;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000374 if (f->f_fp == NULL)
375 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000376 if (!PyArg_NoArgs(args))
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000377 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000378 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000379 res = isatty((int)fileno(f->f_fp));
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000380 Py_END_ALLOW_THREADS
381 return PyInt_FromLong(res);
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000382}
383
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000384/* We expect that fstat exists on most systems.
385 It's confirmed on Unix, Mac and Windows.
386 If you don't have it, add #define DONT_HAVE_FSTAT to your config.h. */
387#ifndef DONT_HAVE_FSTAT
388#define HAVE_FSTAT
389
390#include <sys/types.h>
391#include <sys/stat.h>
392
393#endif
394
395#if BUFSIZ < 8192
396#define SMALLCHUNK 8192
397#else
398#define SMALLCHUNK BUFSIZ
399#endif
400
401#define BIGCHUNK (512*1024)
402
403static size_t
404new_buffersize(f, currentsize)
405 PyFileObject *f;
406 size_t currentsize;
407{
408#ifdef HAVE_FSTAT
409 long pos, end;
410 struct stat st;
411 if (fstat(fileno(f->f_fp), &st) == 0) {
412 end = st.st_size;
413 pos = ftell(f->f_fp);
414 if (end > pos && pos >= 0)
415 return end - pos + 1;
416 }
417#endif
418 if (currentsize > SMALLCHUNK) {
419 /* Keep doubling until we reach BIGCHUNK;
420 then keep adding BIGCHUNK. */
421 if (currentsize <= BIGCHUNK)
422 return currentsize + currentsize;
423 else
424 return currentsize + BIGCHUNK;
425 }
426 return currentsize + SMALLCHUNK;
427}
428
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000429static PyObject *
Guido van Rossumce5ba841991-03-06 13:06:18 +0000430file_read(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000431 PyFileObject *f;
432 PyObject *args;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000433{
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000434 long bytesrequested;
435 size_t bytesread, buffersize, chunksize;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000436 PyObject *v;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000437
Guido van Rossumd7297e61992-07-06 14:19:26 +0000438 if (f->f_fp == NULL)
439 return err_closed();
Guido van Rossumb6775db1994-08-01 11:34:53 +0000440 if (args == NULL)
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000441 bytesrequested = -1;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000442 else {
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000443 if (!PyArg_Parse(args, "l", &bytesrequested))
Guido van Rossumb6775db1994-08-01 11:34:53 +0000444 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000445 }
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000446 if (bytesrequested < 0)
447 buffersize = new_buffersize(f, 0);
448 else
449 buffersize = bytesrequested;
450 v = PyString_FromStringAndSize((char *)NULL, buffersize);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000451 if (v == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000452 return NULL;
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000453 bytesread = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000454 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000455 for (;;) {
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000456 chunksize = fread(BUF(v) + bytesread, 1,
457 buffersize - bytesread, f->f_fp);
Guido van Rossumce5ba841991-03-06 13:06:18 +0000458 /* XXX Error check? */
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000459 if (chunksize == 0)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000460 break;
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000461 bytesread += chunksize;
462 if (bytesread < buffersize)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000463 break;
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000464 if (bytesrequested < 0) {
465 buffersize = new_buffersize(f, buffersize);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000466 Py_BLOCK_THREADS
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000467 if (_PyString_Resize(&v, buffersize) < 0)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000468 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000469 Py_UNBLOCK_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000470 }
471 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000472 Py_END_ALLOW_THREADS
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000473 if (bytesread != buffersize)
474 _PyString_Resize(&v, bytesread);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000475 return v;
476}
477
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000478static PyObject *
479file_readinto(f, args)
480 PyFileObject *f;
481 PyObject *args;
482{
483 char *ptr;
484 int ntodo, ndone, nnow;
485
486 if (f->f_fp == NULL)
487 return err_closed();
488 if (!PyArg_Parse(args, "w#", &ptr, &ntodo))
489 return NULL;
490 ndone = 0;
491 /*
492 ** XXXX Is this correct? Other threads may see partially-completed
493 ** reads if they look at the object we're reading into...
494 */
495 Py_BEGIN_ALLOW_THREADS
496 while(ntodo > 0) {
497 nnow = fread(ptr+ndone, 1, ntodo, f->f_fp);
498 if (nnow < 0 ) {
499 PyErr_SetFromErrno(PyExc_IOError);
500 clearerr(f->f_fp);
501 return NULL;
502 }
503 if (nnow == 0)
504 break;
505 ndone += nnow;
506 ntodo -= nnow;
507 }
508 Py_END_ALLOW_THREADS
509 return PyInt_FromLong(ndone);
510}
511
512
Guido van Rossum0bd24411991-04-04 15:21:57 +0000513/* Internal routine to get a line.
514 Size argument interpretation:
515 > 0: max length;
516 = 0: read arbitrary line;
517 < 0: strip trailing '\n', raise EOFError if EOF reached immediately
Guido van Rossumce5ba841991-03-06 13:06:18 +0000518*/
519
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000520static PyObject *
Guido van Rossum0bd24411991-04-04 15:21:57 +0000521getline(f, n)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000522 PyFileObject *f;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000523 int n;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000524{
Guido van Rossumce5ba841991-03-06 13:06:18 +0000525 register FILE *fp;
526 register int c;
527 register char *buf, *end;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000528 int n1, n2;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000529 PyObject *v;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000530
Guido van Rossumc10aa771992-07-31 12:42:38 +0000531 fp = f->f_fp;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000532 n2 = n > 0 ? n : 100;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000533 v = PyString_FromStringAndSize((char *)NULL, n2);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000534 if (v == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000535 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000536 buf = BUF(v);
537 end = buf + n2;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000538
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000539 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000540 for (;;) {
Guido van Rossum0bd24411991-04-04 15:21:57 +0000541 if ((c = getc(fp)) == EOF) {
Guido van Rossum76ad8ed1991-06-03 10:54:55 +0000542 clearerr(fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000543 if (PyErr_CheckSignals()) {
544 Py_BLOCK_THREADS
545 Py_DECREF(v);
Guido van Rossum0bd24411991-04-04 15:21:57 +0000546 return NULL;
547 }
548 if (n < 0 && buf == BUF(v)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000549 Py_BLOCK_THREADS
550 Py_DECREF(v);
551 PyErr_SetString(PyExc_EOFError,
Guido van Rossum201be051991-12-24 13:26:41 +0000552 "EOF when reading a line");
Guido van Rossum0bd24411991-04-04 15:21:57 +0000553 return NULL;
554 }
Guido van Rossumce5ba841991-03-06 13:06:18 +0000555 break;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000556 }
557 if ((*buf++ = c) == '\n') {
558 if (n < 0)
559 buf--;
560 break;
561 }
Guido van Rossumce5ba841991-03-06 13:06:18 +0000562 if (buf == end) {
Guido van Rossum0bd24411991-04-04 15:21:57 +0000563 if (n > 0)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000564 break;
565 n1 = n2;
566 n2 += 1000;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000567 Py_BLOCK_THREADS
568 if (_PyString_Resize(&v, n2) < 0)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000569 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000570 Py_UNBLOCK_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000571 buf = BUF(v) + n1;
572 end = BUF(v) + n2;
573 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000574 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000575 Py_END_ALLOW_THREADS
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000576
Guido van Rossumce5ba841991-03-06 13:06:18 +0000577 n1 = buf - BUF(v);
578 if (n1 != n2)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000579 _PyString_Resize(&v, n1);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000580 return v;
581}
582
Guido van Rossum0bd24411991-04-04 15:21:57 +0000583/* External C interface */
584
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000585PyObject *
586PyFile_GetLine(f, n)
587 PyObject *f;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000588 int n;
589{
Guido van Rossum3165fe61992-09-25 21:59:05 +0000590 if (f == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000591 PyErr_BadInternalCall();
Guido van Rossum0bd24411991-04-04 15:21:57 +0000592 return NULL;
593 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000594 if (!PyFile_Check(f)) {
595 PyObject *reader;
596 PyObject *args;
597 PyObject *result;
598 reader = PyObject_GetAttrString(f, "readline");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000599 if (reader == NULL)
600 return NULL;
601 if (n <= 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000602 args = Py_BuildValue("()");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000603 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000604 args = Py_BuildValue("(i)", n);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000605 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000606 Py_DECREF(reader);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000607 return NULL;
608 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000609 result = PyEval_CallObject(reader, args);
610 Py_DECREF(reader);
611 Py_DECREF(args);
612 if (result != NULL && !PyString_Check(result)) {
613 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000614 result = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000615 PyErr_SetString(PyExc_TypeError,
Guido van Rossum3165fe61992-09-25 21:59:05 +0000616 "object.readline() returned non-string");
617 }
618 if (n < 0 && result != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000619 char *s = PyString_AsString(result);
620 int len = PyString_Size(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000621 if (len == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000622 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000623 result = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000624 PyErr_SetString(PyExc_EOFError,
Guido van Rossum3165fe61992-09-25 21:59:05 +0000625 "EOF when reading a line");
626 }
627 else if (s[len-1] == '\n') {
628 if (result->ob_refcnt == 1)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000629 _PyString_Resize(&result, len-1);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000630 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000631 PyObject *v;
632 v = PyString_FromStringAndSize(s,
633 len-1);
634 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000635 result = v;
636 }
637 }
638 }
639 return result;
640 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000641 if (((PyFileObject*)f)->f_fp == NULL)
Guido van Rossumd7297e61992-07-06 14:19:26 +0000642 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000643 return getline((PyFileObject *)f, n);
Guido van Rossum0bd24411991-04-04 15:21:57 +0000644}
645
646/* Python method */
647
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000648static PyObject *
Guido van Rossum0bd24411991-04-04 15:21:57 +0000649file_readline(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000650 PyFileObject *f;
651 PyObject *args;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000652{
653 int n;
654
Guido van Rossumd7297e61992-07-06 14:19:26 +0000655 if (f->f_fp == NULL)
656 return err_closed();
Guido van Rossum0bd24411991-04-04 15:21:57 +0000657 if (args == NULL)
658 n = 0; /* Unlimited */
659 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000660 if (!PyArg_Parse(args, "i", &n))
Guido van Rossum0bd24411991-04-04 15:21:57 +0000661 return NULL;
Guido van Rossumd3f9a1a1995-07-10 23:32:26 +0000662 if (n == 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000663 return PyString_FromString("");
Guido van Rossumd3f9a1a1995-07-10 23:32:26 +0000664 if (n < 0)
665 n = 0;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000666 }
667
Guido van Rossum51415a71992-03-27 17:23:38 +0000668 return getline(f, n);
Guido van Rossum0bd24411991-04-04 15:21:57 +0000669}
670
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000671static PyObject *
Guido van Rossumce5ba841991-03-06 13:06:18 +0000672file_readlines(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000673 PyFileObject *f;
674 PyObject *args;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000675{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000676 PyObject *list;
677 PyObject *line;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000678
Guido van Rossumd7297e61992-07-06 14:19:26 +0000679 if (f->f_fp == NULL)
680 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000681 if (!PyArg_NoArgs(args))
Guido van Rossum0bd24411991-04-04 15:21:57 +0000682 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000683 if ((list = PyList_New(0)) == NULL)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000684 return NULL;
685 for (;;) {
Guido van Rossum0bd24411991-04-04 15:21:57 +0000686 line = getline(f, 0);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000687 if (line != NULL && PyString_Size(line) == 0) {
688 Py_DECREF(line);
Guido van Rossumce5ba841991-03-06 13:06:18 +0000689 break;
690 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000691 if (line == NULL || PyList_Append(list, line) != 0) {
692 Py_DECREF(list);
693 Py_XDECREF(line);
Guido van Rossumce5ba841991-03-06 13:06:18 +0000694 return NULL;
695 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000696 Py_DECREF(line);
Guido van Rossumce5ba841991-03-06 13:06:18 +0000697 }
698 return list;
699}
700
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000701static PyObject *
Guido van Rossum3f5da241990-12-20 15:06:42 +0000702file_write(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000703 PyFileObject *f;
704 PyObject *args;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000705{
Guido van Rossumd7297e61992-07-06 14:19:26 +0000706 char *s;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000707 int n, n2;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000708 if (f->f_fp == NULL)
709 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000710 if (!PyArg_Parse(args, "s#", &s, &n))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000711 return NULL;
Guido van Rossumeb183da1991-04-04 10:44:06 +0000712 f->f_softspace = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000713 Py_BEGIN_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000714 errno = 0;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000715 n2 = fwrite(s, 1, n, f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000716 Py_END_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000717 if (n2 != n) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000718 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumfebd5511992-03-04 16:39:24 +0000719 clearerr(f->f_fp);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000720 return NULL;
721 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000722 Py_INCREF(Py_None);
723 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000724}
725
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000726static PyObject *
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000727file_writelines(f, args)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000728 PyFileObject *f;
729 PyObject *args;
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000730{
731 int i, n;
732 if (f->f_fp == NULL)
733 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000734 if (args == NULL || !PyList_Check(args)) {
735 PyErr_SetString(PyExc_TypeError,
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000736 "writelines() requires list of strings");
737 return NULL;
738 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000739 n = PyList_Size(args);
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000740 f->f_softspace = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000741 Py_BEGIN_ALLOW_THREADS
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000742 errno = 0;
743 for (i = 0; i < n; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000744 PyObject *line = PyList_GetItem(args, i);
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000745 int len;
746 int nwritten;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000747 if (!PyString_Check(line)) {
748 Py_BLOCK_THREADS
749 PyErr_SetString(PyExc_TypeError,
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000750 "writelines() requires list of strings");
751 return NULL;
752 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000753 len = PyString_Size(line);
754 nwritten = fwrite(PyString_AsString(line), 1, len, f->f_fp);
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000755 if (nwritten != len) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000756 Py_BLOCK_THREADS
757 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000758 clearerr(f->f_fp);
759 return NULL;
760 }
761 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000762 Py_END_ALLOW_THREADS
763 Py_INCREF(Py_None);
764 return Py_None;
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000765}
766
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000767static PyMethodDef file_methods[] = {
768 {"close", (PyCFunction)file_close, 0},
769 {"flush", (PyCFunction)file_flush, 0},
770 {"fileno", (PyCFunction)file_fileno, 0},
771 {"isatty", (PyCFunction)file_isatty, 0},
772 {"read", (PyCFunction)file_read, 0},
773 {"readline", (PyCFunction)file_readline, 0},
774 {"readlines", (PyCFunction)file_readlines, 0},
775 {"seek", (PyCFunction)file_seek, 0},
Guido van Rossumd7047b31995-01-02 19:07:15 +0000776#ifdef HAVE_FTRUNCATE
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000777 {"truncate", (PyCFunction)file_truncate, 0},
Guido van Rossumd7047b31995-01-02 19:07:15 +0000778#endif
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000779 {"tell", (PyCFunction)file_tell, 0},
780 {"write", (PyCFunction)file_write, 0},
781 {"writelines", (PyCFunction)file_writelines, 0},
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000782 {"readinto", (PyCFunction)file_readinto, 0},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000783 {NULL, NULL} /* sentinel */
784};
785
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000786#define OFF(x) offsetof(PyFileObject, x)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000787
788static struct memberlist file_memberlist[] = {
789 {"softspace", T_INT, OFF(f_softspace)},
790 {"mode", T_OBJECT, OFF(f_mode), RO},
791 {"name", T_OBJECT, OFF(f_name), RO},
792 /* getattr(f, "closed") is implemented without this table */
793 {"closed", T_INT, 0, RO},
794 {NULL} /* Sentinel */
795};
796
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000797static PyObject *
Guido van Rossum3f5da241990-12-20 15:06:42 +0000798file_getattr(f, name)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000799 PyFileObject *f;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000800 char *name;
801{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000802 PyObject *res;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000803
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000804 res = Py_FindMethod(file_methods, (PyObject *)f, name);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000805 if (res != NULL)
806 return res;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000807 PyErr_Clear();
Guido van Rossumb6775db1994-08-01 11:34:53 +0000808 if (strcmp(name, "closed") == 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000809 return PyInt_FromLong((long)(f->f_fp == 0));
810 return PyMember_Get((char *)f, file_memberlist, name);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000811}
812
813static int
814file_setattr(f, name, v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000815 PyFileObject *f;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000816 char *name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000817 PyObject *v;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000818{
819 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000820 PyErr_SetString(PyExc_AttributeError,
821 "can't delete file attributes");
Guido van Rossumb6775db1994-08-01 11:34:53 +0000822 return -1;
823 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000824 return PyMember_Set((char *)f, file_memberlist, name, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000825}
826
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000827PyTypeObject PyFile_Type = {
828 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000829 0,
830 "file",
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000831 sizeof(PyFileObject),
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000832 0,
Guido van Rossumb6775db1994-08-01 11:34:53 +0000833 (destructor)file_dealloc, /*tp_dealloc*/
Guido van Rossum7066dd71992-09-17 17:54:56 +0000834 0, /*tp_print*/
Guido van Rossumb6775db1994-08-01 11:34:53 +0000835 (getattrfunc)file_getattr, /*tp_getattr*/
836 (setattrfunc)file_setattr, /*tp_setattr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000837 0, /*tp_compare*/
Guido van Rossumb6775db1994-08-01 11:34:53 +0000838 (reprfunc)file_repr, /*tp_repr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000839};
Guido van Rossumeb183da1991-04-04 10:44:06 +0000840
841/* Interface for the 'soft space' between print items. */
842
843int
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000844PyFile_SoftSpace(f, newflag)
845 PyObject *f;
Guido van Rossumeb183da1991-04-04 10:44:06 +0000846 int newflag;
847{
848 int oldflag = 0;
Guido van Rossum3165fe61992-09-25 21:59:05 +0000849 if (f == NULL) {
850 /* Do nothing */
851 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000852 else if (PyFile_Check(f)) {
853 oldflag = ((PyFileObject *)f)->f_softspace;
854 ((PyFileObject *)f)->f_softspace = newflag;
Guido van Rossumeb183da1991-04-04 10:44:06 +0000855 }
Guido van Rossum3165fe61992-09-25 21:59:05 +0000856 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000857 PyObject *v;
858 v = PyObject_GetAttrString(f, "softspace");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000859 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000860 PyErr_Clear();
Guido van Rossum3165fe61992-09-25 21:59:05 +0000861 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000862 if (PyInt_Check(v))
863 oldflag = PyInt_AsLong(v);
864 Py_DECREF(v);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000865 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000866 v = PyInt_FromLong((long)newflag);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000867 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000868 PyErr_Clear();
Guido van Rossum3165fe61992-09-25 21:59:05 +0000869 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000870 if (PyObject_SetAttrString(f, "softspace", v) != 0)
871 PyErr_Clear();
872 Py_DECREF(v);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000873 }
874 }
Guido van Rossumeb183da1991-04-04 10:44:06 +0000875 return oldflag;
876}
Guido van Rossum3165fe61992-09-25 21:59:05 +0000877
878/* Interfaces to write objects/strings to file-like objects */
879
880int
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000881PyFile_WriteObject(v, f, flags)
882 PyObject *v;
883 PyObject *f;
Guido van Rossum3165fe61992-09-25 21:59:05 +0000884 int flags;
885{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000886 PyObject *writer, *value, *args, *result;
Guido van Rossum3165fe61992-09-25 21:59:05 +0000887 if (f == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000888 PyErr_SetString(PyExc_TypeError, "writeobject with NULL file");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000889 return -1;
890 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000891 else if (PyFile_Check(f)) {
892 FILE *fp = PyFile_AsFile(f);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000893 if (fp == NULL) {
894 err_closed();
895 return -1;
896 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000897 return PyObject_Print(v, fp, flags);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000898 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000899 writer = PyObject_GetAttrString(f, "write");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000900 if (writer == NULL)
901 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000902 if (flags & Py_PRINT_RAW)
903 value = PyObject_Str(v);
Guido van Rossumc6004111993-11-05 10:22:19 +0000904 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000905 value = PyObject_Repr(v);
Guido van Rossumc6004111993-11-05 10:22:19 +0000906 if (value == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000907 Py_DECREF(writer);
Guido van Rossumc6004111993-11-05 10:22:19 +0000908 return -1;
Guido van Rossum3165fe61992-09-25 21:59:05 +0000909 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000910 args = Py_BuildValue("(O)", value);
Guido van Rossumd3f9a1a1995-07-10 23:32:26 +0000911 if (value == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000912 Py_DECREF(value);
913 Py_DECREF(writer);
Guido van Rossumd3f9a1a1995-07-10 23:32:26 +0000914 return -1;
915 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000916 result = PyEval_CallObject(writer, args);
917 Py_DECREF(args);
918 Py_DECREF(value);
919 Py_DECREF(writer);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000920 if (result == NULL)
921 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000922 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000923 return 0;
924}
925
926void
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000927PyFile_WriteString(s, f)
Guido van Rossum3165fe61992-09-25 21:59:05 +0000928 char *s;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000929 PyObject *f;
Guido van Rossum3165fe61992-09-25 21:59:05 +0000930{
931 if (f == NULL) {
932 /* Do nothing */
933 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000934 else if (PyFile_Check(f)) {
935 FILE *fp = PyFile_AsFile(f);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000936 if (fp != NULL)
937 fputs(s, fp);
938 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000939 else if (!PyErr_Occurred()) {
940 PyObject *v = PyString_FromString(s);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000941 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000942 PyErr_Clear();
Guido van Rossum3165fe61992-09-25 21:59:05 +0000943 }
944 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000945 if (PyFile_WriteObject(v, f, Py_PRINT_RAW) != 0)
946 PyErr_Clear();
947 Py_DECREF(v);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000948 }
949 }
950}