blob: ea8fe9dccfe909e5bc01f8f8aa4db20d5159785b [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* File object implementation */
3
Guido van Rossumc0b618a1997-05-02 03:12:38 +00004#include "Python.h"
Guido van Rossumb6775db1994-08-01 11:34:53 +00005#include "structmember.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006
Guido van Rossumff7e83d1999-08-27 20:39:37 +00007#ifndef DONT_HAVE_SYS_TYPES_H
Guido van Rossum41498431999-01-07 22:09:51 +00008#include <sys/types.h>
Guido van Rossumff7e83d1999-08-27 20:39:37 +00009#endif /* DONT_HAVE_SYS_TYPES_H */
Guido van Rossum41498431999-01-07 22:09:51 +000010
Guido van Rossumeceebb82000-06-28 20:57:07 +000011/* We expect that fstat exists on most systems.
12 It's confirmed on Unix, Mac and Windows.
13 If you don't have it, add #define DONT_HAVE_FSTAT to your config.h. */
14#ifndef DONT_HAVE_FSTAT
15#define HAVE_FSTAT
16
17#ifndef DONT_HAVE_SYS_TYPES_H
18#include <sys/types.h>
19#endif
20
21#ifndef DONT_HAVE_SYS_STAT_H
22#include <sys/stat.h>
23#else
24#ifdef HAVE_STAT_H
25#include <stat.h>
26#endif
27#endif
28
29#endif /* DONT_HAVE_FSTAT */
30
Guido van Rossum685a38e1996-12-05 21:54:17 +000031#ifdef HAVE_UNISTD_H
32#include <unistd.h>
33#endif
34
Guido van Rossumb8199141997-05-06 15:23:24 +000035#ifdef MS_WIN32
Guido van Rossumb8199141997-05-06 15:23:24 +000036#define fileno _fileno
Trent Mickf29f47b2000-08-11 19:02:59 +000037/* can (almost fully) duplicate with _chsize, see file_truncate */
Guido van Rossumb8199141997-05-06 15:23:24 +000038#define HAVE_FTRUNCATE
39#endif
40
Guido van Rossumf2044e11998-04-28 16:05:59 +000041#ifdef macintosh
42#ifdef USE_GUSI
43#define HAVE_FTRUNCATE
44#endif
45#endif
46
Jack Jansene08dea191995-04-23 22:12:47 +000047#ifdef __MWERKS__
48/* Mwerks fopen() doesn't always set errno */
49#define NO_FOPEN_ERRNO
50#endif
Guido van Rossum295d1711995-02-19 15:55:19 +000051
Guido van Rossumc0b618a1997-05-02 03:12:38 +000052#define BUF(v) PyString_AS_STRING((PyStringObject *)v)
Guido van Rossumce5ba841991-03-06 13:06:18 +000053
Guido van Rossumff7e83d1999-08-27 20:39:37 +000054#ifndef DONT_HAVE_ERRNO_H
Guido van Rossumf1dc5661993-07-05 10:31:29 +000055#include <errno.h>
Guido van Rossumff7e83d1999-08-27 20:39:37 +000056#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000057
Trent Mickf29f47b2000-08-11 19:02:59 +000058/* define the appropriate 64-bit capable tell() function */
Guido van Rossum1a5e5832000-09-21 22:15:29 +000059#if defined(MS_WIN64)
60#define TELL64 _telli64
Guido van Rossumecaa7772000-11-13 19:48:22 +000061#elif defined(__NetBSD__) || defined(__OpenBSD__) || defined(_HAVE_BSDI) || defined(__APPLE__)
Guido van Rossum1a5e5832000-09-21 22:15:29 +000062/* NOTE: this is only used on older
63 NetBSD prior to f*o() funcions */
64#define TELL64(fd) lseek((fd),0,SEEK_CUR)
Trent Mickf29f47b2000-08-11 19:02:59 +000065#endif
66
67
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000068typedef struct {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000069 PyObject_HEAD
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000070 FILE *f_fp;
Guido van Rossumc0b618a1997-05-02 03:12:38 +000071 PyObject *f_name;
72 PyObject *f_mode;
Tim Petersdbd9ba62000-07-09 03:09:57 +000073 int (*f_close)(FILE *);
Guido van Rossumeb183da1991-04-04 10:44:06 +000074 int f_softspace; /* Flag used by 'print' command */
Guido van Rossum4c08d552000-03-10 22:55:18 +000075 int f_binary; /* Flag which indicates whether the file is open
76 open in binary (1) or test (0) mode */
Guido van Rossumc0b618a1997-05-02 03:12:38 +000077} PyFileObject;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000078
79FILE *
Fred Drakefd99de62000-07-09 05:02:18 +000080PyFile_AsFile(PyObject *f)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000081{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000082 if (f == NULL || !PyFile_Check(f))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000083 return NULL;
Guido van Rossum3165fe61992-09-25 21:59:05 +000084 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +000085 return ((PyFileObject *)f)->f_fp;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000086}
87
Guido van Rossumc0b618a1997-05-02 03:12:38 +000088PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +000089PyFile_Name(PyObject *f)
Guido van Rossumdb3165e1993-10-18 17:06:59 +000090{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000091 if (f == NULL || !PyFile_Check(f))
Guido van Rossumdb3165e1993-10-18 17:06:59 +000092 return NULL;
93 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +000094 return ((PyFileObject *)f)->f_name;
Guido van Rossumdb3165e1993-10-18 17:06:59 +000095}
96
Guido van Rossumc0b618a1997-05-02 03:12:38 +000097PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +000098PyFile_FromFile(FILE *fp, char *name, char *mode, int (*close)(FILE *))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000099{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000100 PyFileObject *f = PyObject_NEW(PyFileObject, &PyFile_Type);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000101 if (f == NULL)
102 return NULL;
103 f->f_fp = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000104 f->f_name = PyString_FromString(name);
105 f->f_mode = PyString_FromString(mode);
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000106 f->f_close = close;
Guido van Rossumeb183da1991-04-04 10:44:06 +0000107 f->f_softspace = 0;
Guido van Rossum4c08d552000-03-10 22:55:18 +0000108 if (strchr(mode,'b') != NULL)
109 f->f_binary = 1;
110 else
111 f->f_binary = 0;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000112 if (f->f_name == NULL || f->f_mode == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000113 Py_DECREF(f);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000114 return NULL;
115 }
116 f->f_fp = fp;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000117 return (PyObject *) f;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000118}
119
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000120PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000121PyFile_FromString(char *name, char *mode)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000122{
Tim Petersdbd9ba62000-07-09 03:09:57 +0000123 extern int fclose(FILE *);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000124 PyFileObject *f;
125 f = (PyFileObject *) PyFile_FromFile((FILE *)NULL, name, mode, fclose);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000126 if (f == NULL)
127 return NULL;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000128#ifdef HAVE_FOPENRF
Guido van Rossuma08095a1991-02-13 23:25:27 +0000129 if (*mode == '*') {
130 FILE *fopenRF();
131 f->f_fp = fopenRF(name, mode+1);
132 }
133 else
134#endif
Guido van Rossumff4949e1992-08-05 19:58:53 +0000135 {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000136 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000137 f->f_fp = fopen(name, mode);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000138 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000139 }
Guido van Rossuma08095a1991-02-13 23:25:27 +0000140 if (f->f_fp == NULL) {
Jack Jansene08dea191995-04-23 22:12:47 +0000141#ifdef NO_FOPEN_ERRNO
Barry Warsaw52ddc0e1998-07-23 16:07:02 +0000142 /* Metroworks only, not testable, so unchanged */
Jack Jansene08dea191995-04-23 22:12:47 +0000143 if ( errno == 0 ) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000144 PyErr_SetString(PyExc_IOError, "Cannot open file");
145 Py_DECREF(f);
Jack Jansene08dea191995-04-23 22:12:47 +0000146 return NULL;
147 }
148#endif
Barry Warsaw52ddc0e1998-07-23 16:07:02 +0000149 PyErr_SetFromErrnoWithFilename(PyExc_IOError, name);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000150 Py_DECREF(f);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000151 return NULL;
152 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000153 return (PyObject *)f;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000154}
155
Guido van Rossumb6775db1994-08-01 11:34:53 +0000156void
Fred Drakefd99de62000-07-09 05:02:18 +0000157PyFile_SetBufSize(PyObject *f, int bufsize)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000158{
159 if (bufsize >= 0) {
160#ifdef HAVE_SETVBUF
161 int type;
162 switch (bufsize) {
163 case 0:
164 type = _IONBF;
165 break;
166 case 1:
167 type = _IOLBF;
168 bufsize = BUFSIZ;
169 break;
170 default:
171 type = _IOFBF;
172 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000173 setvbuf(((PyFileObject *)f)->f_fp, (char *)NULL,
174 type, bufsize);
Guido van Rossumf8b4de01998-03-06 15:32:40 +0000175#else /* !HAVE_SETVBUF */
176 if (bufsize <= 1)
177 setbuf(((PyFileObject *)f)->f_fp, (char *)NULL);
178#endif /* !HAVE_SETVBUF */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000179 }
180}
181
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000182static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000183err_closed(void)
Guido van Rossumd7297e61992-07-06 14:19:26 +0000184{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000185 PyErr_SetString(PyExc_ValueError, "I/O operation on closed file");
Guido van Rossumd7297e61992-07-06 14:19:26 +0000186 return NULL;
187}
188
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000189/* Methods */
190
191static void
Fred Drakefd99de62000-07-09 05:02:18 +0000192file_dealloc(PyFileObject *f)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000193{
Guido van Rossumff4949e1992-08-05 19:58:53 +0000194 if (f->f_fp != NULL && f->f_close != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000195 Py_BEGIN_ALLOW_THREADS
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000196 (*f->f_close)(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000197 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000198 }
Guido van Rossum1109fbc1998-04-10 22:16:39 +0000199 if (f->f_name != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000200 Py_DECREF(f->f_name);
Guido van Rossum1109fbc1998-04-10 22:16:39 +0000201 }
202 if (f->f_mode != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000203 Py_DECREF(f->f_mode);
Guido van Rossum1109fbc1998-04-10 22:16:39 +0000204 }
Guido van Rossumb18618d2000-05-03 23:44:39 +0000205 PyObject_DEL(f);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000206}
207
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000208static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000209file_repr(PyFileObject *f)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000210{
211 char buf[300];
Fred Drakea44d3532000-06-30 15:01:00 +0000212 sprintf(buf, "<%s file '%.256s', mode '%.10s' at %p>",
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000213 f->f_fp == NULL ? "closed" : "open",
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000214 PyString_AsString(f->f_name),
215 PyString_AsString(f->f_mode),
Fred Drakea44d3532000-06-30 15:01:00 +0000216 f);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000217 return PyString_FromString(buf);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000218}
219
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000220static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000221file_close(PyFileObject *f, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000222{
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000223 int sts = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000224 if (!PyArg_NoArgs(args))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000225 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000226 if (f->f_fp != NULL) {
Guido van Rossumff4949e1992-08-05 19:58:53 +0000227 if (f->f_close != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000228 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000229 errno = 0;
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000230 sts = (*f->f_close)(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000231 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000232 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000233 f->f_fp = NULL;
234 }
Guido van Rossumfebd5511992-03-04 16:39:24 +0000235 if (sts == EOF)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000236 return PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000237 if (sts != 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000238 return PyInt_FromLong((long)sts);
239 Py_INCREF(Py_None);
240 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000241}
242
Trent Mickf29f47b2000-08-11 19:02:59 +0000243
244/* a portable fseek() function
245 return 0 on success, non-zero on failure (with errno set) */
246int
Trent Mickf29f47b2000-08-11 19:02:59 +0000247#if defined(HAVE_LARGEFILE_SUPPORT) && SIZEOF_OFF_T < 8 && SIZEOF_FPOS_T >= 8
Fred Drake8ce159a2000-08-31 05:18:54 +0000248_portable_fseek(FILE *fp, fpos_t offset, int whence)
Trent Mickf29f47b2000-08-11 19:02:59 +0000249#else
Fred Drake8ce159a2000-08-31 05:18:54 +0000250_portable_fseek(FILE *fp, off_t offset, int whence)
Trent Mickf29f47b2000-08-11 19:02:59 +0000251#endif
Trent Mickf29f47b2000-08-11 19:02:59 +0000252{
253#if defined(HAVE_FSEEKO)
254 return fseeko(fp, offset, whence);
255#elif defined(HAVE_FSEEK64)
256 return fseek64(fp, offset, whence);
Fred Drakedb810ac2000-10-06 20:42:33 +0000257#elif defined(__BEOS__)
258 return _fseek(fp, offset, whence);
Trent Mickf29f47b2000-08-11 19:02:59 +0000259#elif defined(HAVE_LARGEFILE_SUPPORT) && SIZEOF_FPOS_T >= 8
260 /* lacking a 64-bit capable fseek() (as Win64 does) use a 64-bit capable
261 fsetpos() and tell() to implement fseek()*/
262 fpos_t pos;
263 switch (whence) {
264 case SEEK_CUR:
265 if (fgetpos(fp, &pos) != 0)
266 return -1;
267 offset += pos;
268 break;
269 case SEEK_END:
270 /* do a "no-op" seek first to sync the buffering so that
271 the low-level tell() can be used correctly */
272 if (fseek(fp, 0, SEEK_END) != 0)
273 return -1;
274 if ((pos = TELL64(fileno(fp))) == -1L)
275 return -1;
276 offset += pos;
277 break;
278 /* case SEEK_SET: break; */
279 }
280 return fsetpos(fp, &offset);
281#else
282 return fseek(fp, offset, whence);
283#endif
284}
285
286
287/* a portable ftell() function
288 Return -1 on failure with errno set appropriately, current file
289 position on success */
290#if defined(HAVE_LARGEFILE_SUPPORT) && SIZEOF_OFF_T < 8 && SIZEOF_FPOS_T >= 8
291fpos_t
292#else
293off_t
294#endif
Fred Drake8ce159a2000-08-31 05:18:54 +0000295_portable_ftell(FILE* fp)
Trent Mickf29f47b2000-08-11 19:02:59 +0000296{
297#if defined(HAVE_FTELLO) && defined(HAVE_LARGEFILE_SUPPORT)
298 return ftello(fp);
299#elif defined(HAVE_FTELL64) && defined(HAVE_LARGEFILE_SUPPORT)
300 return ftell64(fp);
301#elif SIZEOF_FPOS_T >= 8 && defined(HAVE_LARGEFILE_SUPPORT)
302 fpos_t pos;
303 if (fgetpos(fp, &pos) != 0)
304 return -1;
305 return pos;
306#else
307 return ftell(fp);
308#endif
309}
310
311
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000312static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000313file_seek(PyFileObject *f, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000314{
Guido van Rossumd7297e61992-07-06 14:19:26 +0000315 int whence;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000316 int ret;
Trent Mickf29f47b2000-08-11 19:02:59 +0000317#if defined(HAVE_LARGEFILE_SUPPORT) && SIZEOF_OFF_T < 8 && SIZEOF_FPOS_T >= 8
318 fpos_t offset, pos;
319#else
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000320 off_t offset;
Trent Mickf29f47b2000-08-11 19:02:59 +0000321#endif /* !MS_WIN64 */
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000322 PyObject *offobj;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000323
Guido van Rossumd7297e61992-07-06 14:19:26 +0000324 if (f->f_fp == NULL)
325 return err_closed();
326 whence = 0;
Guido van Rossum43713e52000-02-29 13:59:29 +0000327 if (!PyArg_ParseTuple(args, "O|i:seek", &offobj, &whence))
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000328 return NULL;
329#if !defined(HAVE_LARGEFILE_SUPPORT)
330 offset = PyInt_AsLong(offobj);
331#else
332 offset = PyLong_Check(offobj) ?
333 PyLong_AsLongLong(offobj) : PyInt_AsLong(offobj);
334#endif
335 if (PyErr_Occurred())
Guido van Rossum88303191999-01-04 17:22:18 +0000336 return NULL;
Trent Mickf29f47b2000-08-11 19:02:59 +0000337
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000338 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000339 errno = 0;
Trent Mickf29f47b2000-08-11 19:02:59 +0000340 ret = _portable_fseek(f->f_fp, offset, whence);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000341 Py_END_ALLOW_THREADS
Trent Mickf29f47b2000-08-11 19:02:59 +0000342
Guido van Rossumff4949e1992-08-05 19:58:53 +0000343 if (ret != 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000344 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumfebd5511992-03-04 16:39:24 +0000345 clearerr(f->f_fp);
346 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000347 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000348 Py_INCREF(Py_None);
349 return Py_None;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000350}
351
Trent Mickf29f47b2000-08-11 19:02:59 +0000352
Guido van Rossumd7047b31995-01-02 19:07:15 +0000353#ifdef HAVE_FTRUNCATE
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000354static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000355file_truncate(PyFileObject *f, PyObject *args)
Guido van Rossumd7047b31995-01-02 19:07:15 +0000356{
Guido van Rossumd7047b31995-01-02 19:07:15 +0000357 int ret;
Trent Mickf29f47b2000-08-11 19:02:59 +0000358#if defined(HAVE_LARGEFILE_SUPPORT) && SIZEOF_OFF_T < 8 && SIZEOF_FPOS_T >= 8
359 fpos_t newsize;
360#else
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000361 off_t newsize;
Trent Mickf29f47b2000-08-11 19:02:59 +0000362#endif
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000363 PyObject *newsizeobj;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000364
365 if (f->f_fp == NULL)
366 return err_closed();
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000367 newsizeobj = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +0000368 if (!PyArg_ParseTuple(args, "|O:truncate", &newsizeobj))
Guido van Rossum88303191999-01-04 17:22:18 +0000369 return NULL;
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000370 if (newsizeobj != NULL) {
371#if !defined(HAVE_LARGEFILE_SUPPORT)
372 newsize = PyInt_AsLong(newsizeobj);
373#else
374 newsize = PyLong_Check(newsizeobj) ?
375 PyLong_AsLongLong(newsizeobj) :
376 PyInt_AsLong(newsizeobj);
377#endif
378 if (PyErr_Occurred())
379 return NULL;
380 } else {
381 /* Default to current position*/
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000382 Py_BEGIN_ALLOW_THREADS
Guido van Rossumd7047b31995-01-02 19:07:15 +0000383 errno = 0;
Trent Mickf29f47b2000-08-11 19:02:59 +0000384 newsize = _portable_ftell(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000385 Py_END_ALLOW_THREADS
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000386 if (newsize == -1) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000387 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumd7047b31995-01-02 19:07:15 +0000388 clearerr(f->f_fp);
389 return NULL;
390 }
391 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000392 Py_BEGIN_ALLOW_THREADS
Guido van Rossumd7047b31995-01-02 19:07:15 +0000393 errno = 0;
394 ret = fflush(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000395 Py_END_ALLOW_THREADS
Trent Mickf29f47b2000-08-11 19:02:59 +0000396 if (ret != 0) goto onioerror;
397
398#ifdef MS_WIN32
399 /* can use _chsize; if, however, the newsize overflows 32-bits then
400 _chsize is *not* adequate; in this case, an OverflowError is raised */
401 if (newsize > LONG_MAX) {
402 PyErr_SetString(PyExc_OverflowError,
403 "the new size is too long for _chsize (it is limited to 32-bit values)");
Guido van Rossumd7047b31995-01-02 19:07:15 +0000404 return NULL;
Trent Mickf29f47b2000-08-11 19:02:59 +0000405 } else {
406 Py_BEGIN_ALLOW_THREADS
407 errno = 0;
408 ret = _chsize(fileno(f->f_fp), newsize);
409 Py_END_ALLOW_THREADS
410 if (ret != 0) goto onioerror;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000411 }
Trent Mickf29f47b2000-08-11 19:02:59 +0000412#else
413 Py_BEGIN_ALLOW_THREADS
414 errno = 0;
415 ret = ftruncate(fileno(f->f_fp), newsize);
416 Py_END_ALLOW_THREADS
417 if (ret != 0) goto onioerror;
418#endif /* !MS_WIN32 */
419
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000420 Py_INCREF(Py_None);
421 return Py_None;
Trent Mickf29f47b2000-08-11 19:02:59 +0000422
423onioerror:
424 PyErr_SetFromErrno(PyExc_IOError);
425 clearerr(f->f_fp);
426 return NULL;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000427}
428#endif /* HAVE_FTRUNCATE */
429
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000430static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000431file_tell(PyFileObject *f, PyObject *args)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000432{
Trent Mickf29f47b2000-08-11 19:02:59 +0000433#if defined(HAVE_LARGEFILE_SUPPORT) && SIZEOF_OFF_T < 8 && SIZEOF_FPOS_T >= 8
434 fpos_t pos;
435#else
436 off_t pos;
437#endif
438
Guido van Rossumd7297e61992-07-06 14:19:26 +0000439 if (f->f_fp == NULL)
440 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000441 if (!PyArg_NoArgs(args))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000442 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000443 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000444 errno = 0;
Trent Mickf29f47b2000-08-11 19:02:59 +0000445 pos = _portable_ftell(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000446 Py_END_ALLOW_THREADS
Trent Mickf29f47b2000-08-11 19:02:59 +0000447 if (pos == -1) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000448 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumfebd5511992-03-04 16:39:24 +0000449 clearerr(f->f_fp);
450 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000451 }
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000452#if !defined(HAVE_LARGEFILE_SUPPORT)
Trent Mickf29f47b2000-08-11 19:02:59 +0000453 return PyInt_FromLong(pos);
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000454#else
Trent Mickf29f47b2000-08-11 19:02:59 +0000455 return PyLong_FromLongLong(pos);
Guido van Rossum3c9fe0c1999-01-06 18:51:17 +0000456#endif
Guido van Rossumce5ba841991-03-06 13:06:18 +0000457}
458
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000459static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000460file_fileno(PyFileObject *f, PyObject *args)
Guido van Rossumed233a51992-06-23 09:07:03 +0000461{
Guido van Rossumd7297e61992-07-06 14:19:26 +0000462 if (f->f_fp == NULL)
463 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000464 if (!PyArg_NoArgs(args))
Guido van Rossumed233a51992-06-23 09:07:03 +0000465 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000466 return PyInt_FromLong((long) fileno(f->f_fp));
Guido van Rossumed233a51992-06-23 09:07:03 +0000467}
468
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000469static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000470file_flush(PyFileObject *f, PyObject *args)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000471{
Guido van Rossumff4949e1992-08-05 19:58:53 +0000472 int res;
473
Guido van Rossumd7297e61992-07-06 14:19:26 +0000474 if (f->f_fp == NULL)
475 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000476 if (!PyArg_NoArgs(args))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000477 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000478 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000479 errno = 0;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000480 res = fflush(f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000481 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000482 if (res != 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000483 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumfebd5511992-03-04 16:39:24 +0000484 clearerr(f->f_fp);
485 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000486 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000487 Py_INCREF(Py_None);
488 return Py_None;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000489}
490
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000491static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000492file_isatty(PyFileObject *f, PyObject *args)
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000493{
Guido van Rossumff4949e1992-08-05 19:58:53 +0000494 long res;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000495 if (f->f_fp == NULL)
496 return err_closed();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000497 if (!PyArg_NoArgs(args))
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000498 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000499 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +0000500 res = isatty((int)fileno(f->f_fp));
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000501 Py_END_ALLOW_THREADS
502 return PyInt_FromLong(res);
Guido van Rossuma1ab7fa1991-06-04 19:37:39 +0000503}
504
Guido van Rossumff7e83d1999-08-27 20:39:37 +0000505
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000506#if BUFSIZ < 8192
507#define SMALLCHUNK 8192
508#else
509#define SMALLCHUNK BUFSIZ
510#endif
511
Guido van Rossum3c259041999-01-14 19:00:14 +0000512#if SIZEOF_INT < 4
513#define BIGCHUNK (512 * 32)
514#else
515#define BIGCHUNK (512 * 1024)
516#endif
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000517
518static size_t
Fred Drakefd99de62000-07-09 05:02:18 +0000519new_buffersize(PyFileObject *f, size_t currentsize)
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000520{
521#ifdef HAVE_FSTAT
522 long pos, end;
523 struct stat st;
524 if (fstat(fileno(f->f_fp), &st) == 0) {
525 end = st.st_size;
Guido van Rossumcada2931998-12-11 20:44:56 +0000526 /* The following is not a bug: we really need to call lseek()
527 *and* ftell(). The reason is that some stdio libraries
528 mistakenly flush their buffer when ftell() is called and
529 the lseek() call it makes fails, thereby throwing away
530 data that cannot be recovered in any way. To avoid this,
531 we first test lseek(), and only call ftell() if lseek()
532 works. We can't use the lseek() value either, because we
533 need to take the amount of buffered data into account.
534 (Yet another reason why stdio stinks. :-) */
Guido van Rossum91aaa921998-05-05 22:21:35 +0000535 pos = lseek(fileno(f->f_fp), 0L, SEEK_CUR);
536 if (pos >= 0)
537 pos = ftell(f->f_fp);
Guido van Rossumd30dc0a1998-04-27 19:01:08 +0000538 if (pos < 0)
539 clearerr(f->f_fp);
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000540 if (end > pos && pos >= 0)
Guido van Rossumcada2931998-12-11 20:44:56 +0000541 return currentsize + end - pos + 1;
Guido van Rossumdcb5e7f1998-03-03 22:36:10 +0000542 /* Add 1 so if the file were to grow we'd notice. */
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000543 }
544#endif
545 if (currentsize > SMALLCHUNK) {
546 /* Keep doubling until we reach BIGCHUNK;
547 then keep adding BIGCHUNK. */
548 if (currentsize <= BIGCHUNK)
549 return currentsize + currentsize;
550 else
551 return currentsize + BIGCHUNK;
552 }
553 return currentsize + SMALLCHUNK;
554}
555
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000556static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000557file_read(PyFileObject *f, PyObject *args)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000558{
Guido van Rossum789a1611997-05-10 22:33:55 +0000559 long bytesrequested = -1;
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000560 size_t bytesread, buffersize, chunksize;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000561 PyObject *v;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000562
Guido van Rossumd7297e61992-07-06 14:19:26 +0000563 if (f->f_fp == NULL)
564 return err_closed();
Guido van Rossum43713e52000-02-29 13:59:29 +0000565 if (!PyArg_ParseTuple(args, "|l:read", &bytesrequested))
Guido van Rossum789a1611997-05-10 22:33:55 +0000566 return NULL;
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000567 if (bytesrequested < 0)
Guido van Rossumff1ccbf1999-04-10 15:48:23 +0000568 buffersize = new_buffersize(f, (size_t)0);
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000569 else
570 buffersize = bytesrequested;
Trent Mickf29f47b2000-08-11 19:02:59 +0000571 if (buffersize > INT_MAX) {
572 PyErr_SetString(PyExc_OverflowError,
573 "requested number of bytes is more than a Python string can hold");
574 return NULL;
575 }
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000576 v = PyString_FromStringAndSize((char *)NULL, buffersize);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000577 if (v == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000578 return NULL;
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000579 bytesread = 0;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000580 for (;;) {
Guido van Rossum6263d541997-05-10 22:07:25 +0000581 Py_BEGIN_ALLOW_THREADS
582 errno = 0;
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000583 chunksize = fread(BUF(v) + bytesread, 1,
584 buffersize - bytesread, f->f_fp);
Guido van Rossum6263d541997-05-10 22:07:25 +0000585 Py_END_ALLOW_THREADS
586 if (chunksize == 0) {
587 if (!ferror(f->f_fp))
588 break;
589 PyErr_SetFromErrno(PyExc_IOError);
590 clearerr(f->f_fp);
591 Py_DECREF(v);
592 return NULL;
593 }
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000594 bytesread += chunksize;
595 if (bytesread < buffersize)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000596 break;
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000597 if (bytesrequested < 0) {
Guido van Rossumcada2931998-12-11 20:44:56 +0000598 buffersize = new_buffersize(f, buffersize);
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000599 if (_PyString_Resize(&v, buffersize) < 0)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000600 return NULL;
601 }
602 }
Guido van Rossum5449b6e1997-05-09 22:27:31 +0000603 if (bytesread != buffersize)
604 _PyString_Resize(&v, bytesread);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000605 return v;
606}
607
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000608static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000609file_readinto(PyFileObject *f, PyObject *args)
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000610{
611 char *ptr;
Trent Mickf29f47b2000-08-11 19:02:59 +0000612 size_t ntodo, ndone, nnow;
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000613
614 if (f->f_fp == NULL)
615 return err_closed();
616 if (!PyArg_Parse(args, "w#", &ptr, &ntodo))
617 return NULL;
618 ndone = 0;
Guido van Rossum6263d541997-05-10 22:07:25 +0000619 while (ntodo > 0) {
620 Py_BEGIN_ALLOW_THREADS
621 errno = 0;
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000622 nnow = fread(ptr+ndone, 1, ntodo, f->f_fp);
Guido van Rossum6263d541997-05-10 22:07:25 +0000623 Py_END_ALLOW_THREADS
624 if (nnow == 0) {
625 if (!ferror(f->f_fp))
626 break;
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000627 PyErr_SetFromErrno(PyExc_IOError);
628 clearerr(f->f_fp);
629 return NULL;
630 }
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000631 ndone += nnow;
632 ntodo -= nnow;
633 }
Trent Mickf29f47b2000-08-11 19:02:59 +0000634 return PyInt_FromLong((long)ndone);
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000635}
636
637
Guido van Rossum0bd24411991-04-04 15:21:57 +0000638/* Internal routine to get a line.
639 Size argument interpretation:
640 > 0: max length;
641 = 0: read arbitrary line;
642 < 0: strip trailing '\n', raise EOFError if EOF reached immediately
Guido van Rossumce5ba841991-03-06 13:06:18 +0000643*/
644
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000645static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000646get_line(PyFileObject *f, int n)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000647{
Andrew M. Kuchling4b2b4452000-11-29 02:53:22 +0000648 register FILE *fp = f->f_fp;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000649 register int c;
Andrew M. Kuchling4b2b4452000-11-29 02:53:22 +0000650 char *buf, *end;
Trent Mickf29f47b2000-08-11 19:02:59 +0000651 size_t n1, n2;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000652 PyObject *v;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000653
Andrew M. Kuchling1221e6d2000-11-30 18:27:50 +0000654#if defined(HAVE_GETLINE) && defined(_GNU_SOURCE)
Andrew M. Kuchling4b2b4452000-11-29 02:53:22 +0000655 /* Use GNU libc extension getline() for arbitrary-sized lines */
656 if (n == 0) {
657 size_t size = 0;
658 buf = NULL;
659 Py_BEGIN_ALLOW_THREADS
660 n1 = getline(&buf, &size, fp);
661 Py_END_ALLOW_THREADS
662 if (n1 == -1) {
Fred Drakee7e190e2000-12-20 00:55:07 +0000663 if (buf){
664 free(buf);
665 }
Andrew M. Kuchling4b2b4452000-11-29 02:53:22 +0000666 clearerr(fp);
667 if (PyErr_CheckSignals()) {
668 return NULL;
669 }
670 if (n < 0 && feof(fp)) {
671 PyErr_SetString(PyExc_EOFError,
672 "EOF when reading a line");
673 return NULL;
674 }
675 return PyString_FromStringAndSize(NULL, 0);
676 }
677 /* No error */
678
679 v = PyString_FromStringAndSize(buf, n1);
680 free(buf);
681 return v;
682 }
683#endif
684
Guido van Rossum0bd24411991-04-04 15:21:57 +0000685 n2 = n > 0 ? n : 100;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000686 v = PyString_FromStringAndSize((char *)NULL, n2);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000687 if (v == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000688 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000689 buf = BUF(v);
690 end = buf + n2;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000691
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000692 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000693 for (;;) {
Guido van Rossum0bd24411991-04-04 15:21:57 +0000694 if ((c = getc(fp)) == EOF) {
Guido van Rossum76ad8ed1991-06-03 10:54:55 +0000695 clearerr(fp);
Guido van Rossumf5181541997-11-07 19:20:34 +0000696 Py_BLOCK_THREADS
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000697 if (PyErr_CheckSignals()) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000698 Py_DECREF(v);
Guido van Rossum0bd24411991-04-04 15:21:57 +0000699 return NULL;
700 }
701 if (n < 0 && buf == BUF(v)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000702 Py_DECREF(v);
703 PyErr_SetString(PyExc_EOFError,
Guido van Rossum201be051991-12-24 13:26:41 +0000704 "EOF when reading a line");
Guido van Rossum0bd24411991-04-04 15:21:57 +0000705 return NULL;
706 }
Guido van Rossumf5181541997-11-07 19:20:34 +0000707 Py_UNBLOCK_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000708 break;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000709 }
710 if ((*buf++ = c) == '\n') {
711 if (n < 0)
712 buf--;
713 break;
714 }
Guido van Rossumce5ba841991-03-06 13:06:18 +0000715 if (buf == end) {
Guido van Rossum0bd24411991-04-04 15:21:57 +0000716 if (n > 0)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000717 break;
718 n1 = n2;
719 n2 += 1000;
Trent Mickf29f47b2000-08-11 19:02:59 +0000720 if (n2 > INT_MAX) {
721 PyErr_SetString(PyExc_OverflowError,
722 "line is longer than a Python string can hold");
723 return NULL;
724 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000725 Py_BLOCK_THREADS
726 if (_PyString_Resize(&v, n2) < 0)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000727 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000728 Py_UNBLOCK_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000729 buf = BUF(v) + n1;
730 end = BUF(v) + n2;
731 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000732 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000733 Py_END_ALLOW_THREADS
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000734
Guido van Rossumce5ba841991-03-06 13:06:18 +0000735 n1 = buf - BUF(v);
736 if (n1 != n2)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000737 _PyString_Resize(&v, n1);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000738 return v;
739}
740
Guido van Rossum0bd24411991-04-04 15:21:57 +0000741/* External C interface */
742
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000743PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000744PyFile_GetLine(PyObject *f, int n)
Guido van Rossum0bd24411991-04-04 15:21:57 +0000745{
Guido van Rossum3165fe61992-09-25 21:59:05 +0000746 if (f == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000747 PyErr_BadInternalCall();
Guido van Rossum0bd24411991-04-04 15:21:57 +0000748 return NULL;
749 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000750 if (!PyFile_Check(f)) {
751 PyObject *reader;
752 PyObject *args;
753 PyObject *result;
754 reader = PyObject_GetAttrString(f, "readline");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000755 if (reader == NULL)
756 return NULL;
757 if (n <= 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000758 args = Py_BuildValue("()");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000759 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000760 args = Py_BuildValue("(i)", n);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000761 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000762 Py_DECREF(reader);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000763 return NULL;
764 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000765 result = PyEval_CallObject(reader, args);
766 Py_DECREF(reader);
767 Py_DECREF(args);
768 if (result != NULL && !PyString_Check(result)) {
769 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000770 result = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000771 PyErr_SetString(PyExc_TypeError,
Guido van Rossum3165fe61992-09-25 21:59:05 +0000772 "object.readline() returned non-string");
773 }
774 if (n < 0 && result != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000775 char *s = PyString_AsString(result);
776 int len = PyString_Size(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000777 if (len == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000778 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000779 result = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000780 PyErr_SetString(PyExc_EOFError,
Guido van Rossum3165fe61992-09-25 21:59:05 +0000781 "EOF when reading a line");
782 }
783 else if (s[len-1] == '\n') {
784 if (result->ob_refcnt == 1)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000785 _PyString_Resize(&result, len-1);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000786 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000787 PyObject *v;
788 v = PyString_FromStringAndSize(s,
789 len-1);
790 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000791 result = v;
792 }
793 }
794 }
795 return result;
796 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000797 if (((PyFileObject*)f)->f_fp == NULL)
Guido van Rossumd7297e61992-07-06 14:19:26 +0000798 return err_closed();
Marc-André Lemburg1f468602000-07-05 15:32:40 +0000799 return get_line((PyFileObject *)f, n);
Guido van Rossum0bd24411991-04-04 15:21:57 +0000800}
801
802/* Python method */
803
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000804static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000805file_readline(PyFileObject *f, PyObject *args)
Guido van Rossum0bd24411991-04-04 15:21:57 +0000806{
Guido van Rossum789a1611997-05-10 22:33:55 +0000807 int n = -1;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000808
Guido van Rossumd7297e61992-07-06 14:19:26 +0000809 if (f->f_fp == NULL)
810 return err_closed();
Guido van Rossum43713e52000-02-29 13:59:29 +0000811 if (!PyArg_ParseTuple(args, "|i:readline", &n))
Guido van Rossum789a1611997-05-10 22:33:55 +0000812 return NULL;
813 if (n == 0)
814 return PyString_FromString("");
815 if (n < 0)
816 n = 0;
Marc-André Lemburg1f468602000-07-05 15:32:40 +0000817 return get_line(f, n);
Guido van Rossum0bd24411991-04-04 15:21:57 +0000818}
819
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000820static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000821file_readlines(PyFileObject *f, PyObject *args)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000822{
Guido van Rossum789a1611997-05-10 22:33:55 +0000823 long sizehint = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000824 PyObject *list;
825 PyObject *line;
Guido van Rossum6263d541997-05-10 22:07:25 +0000826 char small_buffer[SMALLCHUNK];
827 char *buffer = small_buffer;
828 size_t buffersize = SMALLCHUNK;
829 PyObject *big_buffer = NULL;
830 size_t nfilled = 0;
831 size_t nread;
Guido van Rossum789a1611997-05-10 22:33:55 +0000832 size_t totalread = 0;
Guido van Rossum6263d541997-05-10 22:07:25 +0000833 char *p, *q, *end;
834 int err;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000835
Guido van Rossumd7297e61992-07-06 14:19:26 +0000836 if (f->f_fp == NULL)
837 return err_closed();
Guido van Rossum43713e52000-02-29 13:59:29 +0000838 if (!PyArg_ParseTuple(args, "|l:readlines", &sizehint))
Guido van Rossum0bd24411991-04-04 15:21:57 +0000839 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000840 if ((list = PyList_New(0)) == NULL)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000841 return NULL;
842 for (;;) {
Guido van Rossum6263d541997-05-10 22:07:25 +0000843 Py_BEGIN_ALLOW_THREADS
844 errno = 0;
845 nread = fread(buffer+nfilled, 1, buffersize-nfilled, f->f_fp);
846 Py_END_ALLOW_THREADS
847 if (nread == 0) {
Guido van Rossum789a1611997-05-10 22:33:55 +0000848 sizehint = 0;
Guido van Rossum3da3fce1998-02-19 20:46:48 +0000849 if (!ferror(f->f_fp))
Guido van Rossum6263d541997-05-10 22:07:25 +0000850 break;
851 PyErr_SetFromErrno(PyExc_IOError);
852 clearerr(f->f_fp);
853 error:
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000854 Py_DECREF(list);
Guido van Rossum6263d541997-05-10 22:07:25 +0000855 list = NULL;
856 goto cleanup;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000857 }
Guido van Rossum789a1611997-05-10 22:33:55 +0000858 totalread += nread;
Guido van Rossum6263d541997-05-10 22:07:25 +0000859 p = memchr(buffer+nfilled, '\n', nread);
860 if (p == NULL) {
861 /* Need a larger buffer to fit this line */
862 nfilled += nread;
863 buffersize *= 2;
Trent Mickf29f47b2000-08-11 19:02:59 +0000864 if (buffersize > INT_MAX) {
865 PyErr_SetString(PyExc_OverflowError,
Fred Drake661ea262000-10-24 19:57:45 +0000866 "line is longer than a Python string can hold");
Trent Mickf29f47b2000-08-11 19:02:59 +0000867 goto error;
868 }
Guido van Rossum6263d541997-05-10 22:07:25 +0000869 if (big_buffer == NULL) {
870 /* Create the big buffer */
871 big_buffer = PyString_FromStringAndSize(
872 NULL, buffersize);
873 if (big_buffer == NULL)
874 goto error;
875 buffer = PyString_AS_STRING(big_buffer);
876 memcpy(buffer, small_buffer, nfilled);
877 }
878 else {
879 /* Grow the big buffer */
880 _PyString_Resize(&big_buffer, buffersize);
881 buffer = PyString_AS_STRING(big_buffer);
882 }
883 continue;
884 }
885 end = buffer+nfilled+nread;
886 q = buffer;
887 do {
888 /* Process complete lines */
889 p++;
890 line = PyString_FromStringAndSize(q, p-q);
891 if (line == NULL)
892 goto error;
893 err = PyList_Append(list, line);
894 Py_DECREF(line);
895 if (err != 0)
896 goto error;
897 q = p;
898 p = memchr(q, '\n', end-q);
899 } while (p != NULL);
900 /* Move the remaining incomplete line to the start */
901 nfilled = end-q;
902 memmove(buffer, q, nfilled);
Guido van Rossum789a1611997-05-10 22:33:55 +0000903 if (sizehint > 0)
904 if (totalread >= (size_t)sizehint)
905 break;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000906 }
Guido van Rossum6263d541997-05-10 22:07:25 +0000907 if (nfilled != 0) {
908 /* Partial last line */
909 line = PyString_FromStringAndSize(buffer, nfilled);
910 if (line == NULL)
911 goto error;
Guido van Rossum789a1611997-05-10 22:33:55 +0000912 if (sizehint > 0) {
913 /* Need to complete the last line */
Marc-André Lemburg1f468602000-07-05 15:32:40 +0000914 PyObject *rest = get_line(f, 0);
Guido van Rossum789a1611997-05-10 22:33:55 +0000915 if (rest == NULL) {
916 Py_DECREF(line);
917 goto error;
918 }
919 PyString_Concat(&line, rest);
920 Py_DECREF(rest);
921 if (line == NULL)
922 goto error;
923 }
Guido van Rossum6263d541997-05-10 22:07:25 +0000924 err = PyList_Append(list, line);
925 Py_DECREF(line);
926 if (err != 0)
927 goto error;
928 }
929 cleanup:
Guido van Rossum1109fbc1998-04-10 22:16:39 +0000930 if (big_buffer) {
Guido van Rossum6263d541997-05-10 22:07:25 +0000931 Py_DECREF(big_buffer);
Guido van Rossum1109fbc1998-04-10 22:16:39 +0000932 }
Guido van Rossumce5ba841991-03-06 13:06:18 +0000933 return list;
934}
935
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000936static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000937file_write(PyFileObject *f, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000938{
Guido van Rossumd7297e61992-07-06 14:19:26 +0000939 char *s;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000940 int n, n2;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000941 if (f->f_fp == NULL)
942 return err_closed();
Guido van Rossum4c08d552000-03-10 22:55:18 +0000943 if (!PyArg_Parse(args, f->f_binary ? "s#" : "t#", &s, &n))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000944 return NULL;
Guido van Rossumeb183da1991-04-04 10:44:06 +0000945 f->f_softspace = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000946 Py_BEGIN_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000947 errno = 0;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000948 n2 = fwrite(s, 1, n, f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000949 Py_END_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000950 if (n2 != n) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000951 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumfebd5511992-03-04 16:39:24 +0000952 clearerr(f->f_fp);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000953 return NULL;
954 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000955 Py_INCREF(Py_None);
956 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000957}
958
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000959static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000960file_writelines(PyFileObject *f, PyObject *args)
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000961{
Guido van Rossumee70ad12000-03-13 16:27:06 +0000962#define CHUNKSIZE 1000
963 PyObject *list, *line;
964 PyObject *result;
965 int i, j, index, len, nwritten, islist;
966
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000967 if (f->f_fp == NULL)
968 return err_closed();
Guido van Rossumee70ad12000-03-13 16:27:06 +0000969 if (args == NULL || !PySequence_Check(args)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000970 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +0000971 "writelines() argument must be a sequence of strings");
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000972 return NULL;
973 }
Guido van Rossumee70ad12000-03-13 16:27:06 +0000974 islist = PyList_Check(args);
975
976 /* Strategy: slurp CHUNKSIZE lines into a private list,
977 checking that they are all strings, then write that list
978 without holding the interpreter lock, then come back for more. */
979 index = 0;
980 if (islist)
981 list = NULL;
982 else {
983 list = PyList_New(CHUNKSIZE);
984 if (list == NULL)
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000985 return NULL;
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000986 }
Guido van Rossumee70ad12000-03-13 16:27:06 +0000987 result = NULL;
988
989 for (;;) {
990 if (islist) {
991 Py_XDECREF(list);
992 list = PyList_GetSlice(args, index, index+CHUNKSIZE);
993 if (list == NULL)
994 return NULL;
995 j = PyList_GET_SIZE(list);
996 }
997 else {
998 for (j = 0; j < CHUNKSIZE; j++) {
999 line = PySequence_GetItem(args, index+j);
1000 if (line == NULL) {
1001 if (PyErr_ExceptionMatches(
Marc-André Lemburg6ef68b52000-08-25 22:39:50 +00001002 PyExc_IndexError)) {
Guido van Rossumee70ad12000-03-13 16:27:06 +00001003 PyErr_Clear();
1004 break;
1005 }
1006 /* Some other error occurred.
1007 XXX We may lose some output. */
1008 goto error;
1009 }
Guido van Rossumee70ad12000-03-13 16:27:06 +00001010 PyList_SetItem(list, j, line);
1011 }
1012 }
1013 if (j == 0)
1014 break;
1015
Marc-André Lemburg6ef68b52000-08-25 22:39:50 +00001016 /* Check that all entries are indeed strings. If not,
1017 apply the same rules as for file.write() and
1018 convert the results to strings. This is slow, but
1019 seems to be the only way since all conversion APIs
1020 could potentially execute Python code. */
1021 for (i = 0; i < j; i++) {
1022 PyObject *v = PyList_GET_ITEM(list, i);
1023 if (!PyString_Check(v)) {
1024 const char *buffer;
1025 int len;
1026 if (((f->f_binary &&
1027 PyObject_AsReadBuffer(v,
1028 (const void**)&buffer,
1029 &len)) ||
1030 PyObject_AsCharBuffer(v,
1031 &buffer,
1032 &len))) {
1033 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00001034 "writelines() argument must be a sequence of strings");
Marc-André Lemburg6ef68b52000-08-25 22:39:50 +00001035 goto error;
1036 }
1037 line = PyString_FromStringAndSize(buffer,
1038 len);
1039 if (line == NULL)
1040 goto error;
1041 Py_DECREF(v);
Marc-André Lemburgf5e96fa2000-08-25 22:49:05 +00001042 PyList_SET_ITEM(list, i, line);
Marc-André Lemburg6ef68b52000-08-25 22:39:50 +00001043 }
1044 }
1045
1046 /* Since we are releasing the global lock, the
1047 following code may *not* execute Python code. */
Guido van Rossumee70ad12000-03-13 16:27:06 +00001048 Py_BEGIN_ALLOW_THREADS
1049 f->f_softspace = 0;
1050 errno = 0;
1051 for (i = 0; i < j; i++) {
Marc-André Lemburg6ef68b52000-08-25 22:39:50 +00001052 line = PyList_GET_ITEM(list, i);
Guido van Rossumee70ad12000-03-13 16:27:06 +00001053 len = PyString_GET_SIZE(line);
1054 nwritten = fwrite(PyString_AS_STRING(line),
1055 1, len, f->f_fp);
1056 if (nwritten != len) {
1057 Py_BLOCK_THREADS
1058 PyErr_SetFromErrno(PyExc_IOError);
1059 clearerr(f->f_fp);
1060 goto error;
1061 }
1062 }
1063 Py_END_ALLOW_THREADS
1064
1065 if (j < CHUNKSIZE)
1066 break;
1067 index += CHUNKSIZE;
1068 }
1069
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001070 Py_INCREF(Py_None);
Guido van Rossumee70ad12000-03-13 16:27:06 +00001071 result = Py_None;
1072 error:
1073 Py_XDECREF(list);
1074 return result;
Guido van Rossum5a2a6831993-10-25 09:59:04 +00001075}
1076
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001077static PyMethodDef file_methods[] = {
Guido van Rossum789a1611997-05-10 22:33:55 +00001078 {"readline", (PyCFunction)file_readline, 1},
Guido van Rossum74ba2471997-07-13 03:56:50 +00001079 {"read", (PyCFunction)file_read, 1},
1080 {"write", (PyCFunction)file_write, 0},
1081 {"fileno", (PyCFunction)file_fileno, 0},
Guido van Rossum88303191999-01-04 17:22:18 +00001082 {"seek", (PyCFunction)file_seek, 1},
Guido van Rossumd7047b31995-01-02 19:07:15 +00001083#ifdef HAVE_FTRUNCATE
Guido van Rossum88303191999-01-04 17:22:18 +00001084 {"truncate", (PyCFunction)file_truncate, 1},
Guido van Rossumd7047b31995-01-02 19:07:15 +00001085#endif
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001086 {"tell", (PyCFunction)file_tell, 0},
Guido van Rossumfdf95dd1997-05-05 22:15:02 +00001087 {"readinto", (PyCFunction)file_readinto, 0},
Guido van Rossum74ba2471997-07-13 03:56:50 +00001088 {"readlines", (PyCFunction)file_readlines, 1},
1089 {"writelines", (PyCFunction)file_writelines, 0},
1090 {"flush", (PyCFunction)file_flush, 0},
1091 {"close", (PyCFunction)file_close, 0},
1092 {"isatty", (PyCFunction)file_isatty, 0},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001093 {NULL, NULL} /* sentinel */
1094};
1095
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001096#define OFF(x) offsetof(PyFileObject, x)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001097
1098static struct memberlist file_memberlist[] = {
1099 {"softspace", T_INT, OFF(f_softspace)},
1100 {"mode", T_OBJECT, OFF(f_mode), RO},
1101 {"name", T_OBJECT, OFF(f_name), RO},
1102 /* getattr(f, "closed") is implemented without this table */
1103 {"closed", T_INT, 0, RO},
1104 {NULL} /* Sentinel */
1105};
1106
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001107static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +00001108file_getattr(PyFileObject *f, char *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001109{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001110 PyObject *res;
Guido van Rossumb6775db1994-08-01 11:34:53 +00001111
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001112 res = Py_FindMethod(file_methods, (PyObject *)f, name);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001113 if (res != NULL)
1114 return res;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001115 PyErr_Clear();
Guido van Rossumb6775db1994-08-01 11:34:53 +00001116 if (strcmp(name, "closed") == 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001117 return PyInt_FromLong((long)(f->f_fp == 0));
1118 return PyMember_Get((char *)f, file_memberlist, name);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001119}
1120
1121static int
Fred Drakefd99de62000-07-09 05:02:18 +00001122file_setattr(PyFileObject *f, char *name, PyObject *v)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001123{
1124 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001125 PyErr_SetString(PyExc_AttributeError,
1126 "can't delete file attributes");
Guido van Rossumb6775db1994-08-01 11:34:53 +00001127 return -1;
1128 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001129 return PyMember_Set((char *)f, file_memberlist, name, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001130}
1131
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001132PyTypeObject PyFile_Type = {
1133 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001134 0,
1135 "file",
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001136 sizeof(PyFileObject),
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001137 0,
Guido van Rossumb6775db1994-08-01 11:34:53 +00001138 (destructor)file_dealloc, /*tp_dealloc*/
Guido van Rossum7066dd71992-09-17 17:54:56 +00001139 0, /*tp_print*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001140 (getattrfunc)file_getattr, /*tp_getattr*/
1141 (setattrfunc)file_setattr, /*tp_setattr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001142 0, /*tp_compare*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001143 (reprfunc)file_repr, /*tp_repr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001144};
Guido van Rossumeb183da1991-04-04 10:44:06 +00001145
1146/* Interface for the 'soft space' between print items. */
1147
1148int
Fred Drakefd99de62000-07-09 05:02:18 +00001149PyFile_SoftSpace(PyObject *f, int newflag)
Guido van Rossumeb183da1991-04-04 10:44:06 +00001150{
1151 int oldflag = 0;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001152 if (f == NULL) {
1153 /* Do nothing */
1154 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001155 else if (PyFile_Check(f)) {
1156 oldflag = ((PyFileObject *)f)->f_softspace;
1157 ((PyFileObject *)f)->f_softspace = newflag;
Guido van Rossumeb183da1991-04-04 10:44:06 +00001158 }
Guido van Rossum3165fe61992-09-25 21:59:05 +00001159 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001160 PyObject *v;
1161 v = PyObject_GetAttrString(f, "softspace");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001162 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001163 PyErr_Clear();
Guido van Rossum3165fe61992-09-25 21:59:05 +00001164 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001165 if (PyInt_Check(v))
1166 oldflag = PyInt_AsLong(v);
1167 Py_DECREF(v);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001168 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001169 v = PyInt_FromLong((long)newflag);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001170 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001171 PyErr_Clear();
Guido van Rossum3165fe61992-09-25 21:59:05 +00001172 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001173 if (PyObject_SetAttrString(f, "softspace", v) != 0)
1174 PyErr_Clear();
1175 Py_DECREF(v);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001176 }
1177 }
Guido van Rossumeb183da1991-04-04 10:44:06 +00001178 return oldflag;
1179}
Guido van Rossum3165fe61992-09-25 21:59:05 +00001180
1181/* Interfaces to write objects/strings to file-like objects */
1182
1183int
Fred Drakefd99de62000-07-09 05:02:18 +00001184PyFile_WriteObject(PyObject *v, PyObject *f, int flags)
Guido van Rossum3165fe61992-09-25 21:59:05 +00001185{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001186 PyObject *writer, *value, *args, *result;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001187 if (f == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001188 PyErr_SetString(PyExc_TypeError, "writeobject with NULL file");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001189 return -1;
1190 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001191 else if (PyFile_Check(f)) {
1192 FILE *fp = PyFile_AsFile(f);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001193 if (fp == NULL) {
1194 err_closed();
1195 return -1;
1196 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001197 return PyObject_Print(v, fp, flags);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001198 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001199 writer = PyObject_GetAttrString(f, "write");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001200 if (writer == NULL)
1201 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001202 if (flags & Py_PRINT_RAW)
1203 value = PyObject_Str(v);
Guido van Rossumc6004111993-11-05 10:22:19 +00001204 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001205 value = PyObject_Repr(v);
Guido van Rossumc6004111993-11-05 10:22:19 +00001206 if (value == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001207 Py_DECREF(writer);
Guido van Rossumc6004111993-11-05 10:22:19 +00001208 return -1;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001209 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001210 args = Py_BuildValue("(O)", value);
Guido van Rossume9eec541997-05-22 14:02:25 +00001211 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001212 Py_DECREF(value);
1213 Py_DECREF(writer);
Guido van Rossumd3f9a1a1995-07-10 23:32:26 +00001214 return -1;
1215 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001216 result = PyEval_CallObject(writer, args);
1217 Py_DECREF(args);
1218 Py_DECREF(value);
1219 Py_DECREF(writer);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001220 if (result == NULL)
1221 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001222 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001223 return 0;
1224}
1225
Guido van Rossum27a60b11997-05-22 22:25:11 +00001226int
Fred Drakefd99de62000-07-09 05:02:18 +00001227PyFile_WriteString(char *s, PyObject *f)
Guido van Rossum3165fe61992-09-25 21:59:05 +00001228{
1229 if (f == NULL) {
Guido van Rossum27a60b11997-05-22 22:25:11 +00001230 /* Should be caused by a pre-existing error */
Fred Drakefd99de62000-07-09 05:02:18 +00001231 if (!PyErr_Occurred())
Guido van Rossum27a60b11997-05-22 22:25:11 +00001232 PyErr_SetString(PyExc_SystemError,
1233 "null file for PyFile_WriteString");
1234 return -1;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001235 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001236 else if (PyFile_Check(f)) {
1237 FILE *fp = PyFile_AsFile(f);
Guido van Rossum27a60b11997-05-22 22:25:11 +00001238 if (fp == NULL) {
1239 err_closed();
1240 return -1;
1241 }
1242 fputs(s, fp);
1243 return 0;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001244 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001245 else if (!PyErr_Occurred()) {
1246 PyObject *v = PyString_FromString(s);
Guido van Rossum27a60b11997-05-22 22:25:11 +00001247 int err;
1248 if (v == NULL)
1249 return -1;
1250 err = PyFile_WriteObject(v, f, Py_PRINT_RAW);
1251 Py_DECREF(v);
1252 return err;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001253 }
Guido van Rossum74ba2471997-07-13 03:56:50 +00001254 else
1255 return -1;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001256}
Andrew M. Kuchling06051ed2000-07-13 23:56:54 +00001257
1258/* Try to get a file-descriptor from a Python object. If the object
1259 is an integer or long integer, its value is returned. If not, the
1260 object's fileno() method is called if it exists; the method must return
1261 an integer or long integer, which is returned as the file descriptor value.
1262 -1 is returned on failure.
1263*/
1264
1265int PyObject_AsFileDescriptor(PyObject *o)
1266{
1267 int fd;
1268 PyObject *meth;
1269
1270 if (PyInt_Check(o)) {
1271 fd = PyInt_AsLong(o);
1272 }
1273 else if (PyLong_Check(o)) {
1274 fd = PyLong_AsLong(o);
1275 }
1276 else if ((meth = PyObject_GetAttrString(o, "fileno")) != NULL)
1277 {
1278 PyObject *fno = PyEval_CallObject(meth, NULL);
1279 Py_DECREF(meth);
1280 if (fno == NULL)
1281 return -1;
1282
1283 if (PyInt_Check(fno)) {
1284 fd = PyInt_AsLong(fno);
1285 Py_DECREF(fno);
1286 }
1287 else if (PyLong_Check(fno)) {
1288 fd = PyLong_AsLong(fno);
1289 Py_DECREF(fno);
1290 }
1291 else {
1292 PyErr_SetString(PyExc_TypeError,
1293 "fileno() returned a non-integer");
1294 Py_DECREF(fno);
1295 return -1;
1296 }
1297 }
1298 else {
1299 PyErr_SetString(PyExc_TypeError,
1300 "argument must be an int, or have a fileno() method.");
1301 return -1;
1302 }
1303
1304 if (fd < 0) {
1305 PyErr_Format(PyExc_ValueError,
1306 "file descriptor cannot be a negative integer (%i)",
1307 fd);
1308 return -1;
1309 }
1310 return fd;
1311}
1312
1313
1314