blob: b8b47f887adb12b5f83d91873a35a76d5ea69a42 [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
61#elif defined(__NetBSD__) || defined(__OpenBSD__)
62/* 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{
Guido van Rossumce5ba841991-03-06 13:06:18 +0000648 register FILE *fp;
649 register int c;
650 register 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
Guido van Rossumc10aa771992-07-31 12:42:38 +0000654 fp = f->f_fp;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000655 n2 = n > 0 ? n : 100;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000656 v = PyString_FromStringAndSize((char *)NULL, n2);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000657 if (v == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000658 return NULL;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000659 buf = BUF(v);
660 end = buf + n2;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000661
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000662 Py_BEGIN_ALLOW_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000663 for (;;) {
Guido van Rossum0bd24411991-04-04 15:21:57 +0000664 if ((c = getc(fp)) == EOF) {
Guido van Rossum76ad8ed1991-06-03 10:54:55 +0000665 clearerr(fp);
Guido van Rossumf5181541997-11-07 19:20:34 +0000666 Py_BLOCK_THREADS
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000667 if (PyErr_CheckSignals()) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000668 Py_DECREF(v);
Guido van Rossum0bd24411991-04-04 15:21:57 +0000669 return NULL;
670 }
671 if (n < 0 && buf == BUF(v)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000672 Py_DECREF(v);
673 PyErr_SetString(PyExc_EOFError,
Guido van Rossum201be051991-12-24 13:26:41 +0000674 "EOF when reading a line");
Guido van Rossum0bd24411991-04-04 15:21:57 +0000675 return NULL;
676 }
Guido van Rossumf5181541997-11-07 19:20:34 +0000677 Py_UNBLOCK_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000678 break;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000679 }
680 if ((*buf++ = c) == '\n') {
681 if (n < 0)
682 buf--;
683 break;
684 }
Guido van Rossumce5ba841991-03-06 13:06:18 +0000685 if (buf == end) {
Guido van Rossum0bd24411991-04-04 15:21:57 +0000686 if (n > 0)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000687 break;
688 n1 = n2;
689 n2 += 1000;
Trent Mickf29f47b2000-08-11 19:02:59 +0000690 if (n2 > INT_MAX) {
691 PyErr_SetString(PyExc_OverflowError,
692 "line is longer than a Python string can hold");
693 return NULL;
694 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000695 Py_BLOCK_THREADS
696 if (_PyString_Resize(&v, n2) < 0)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000697 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000698 Py_UNBLOCK_THREADS
Guido van Rossumce5ba841991-03-06 13:06:18 +0000699 buf = BUF(v) + n1;
700 end = BUF(v) + n2;
701 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000702 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000703 Py_END_ALLOW_THREADS
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000704
Guido van Rossumce5ba841991-03-06 13:06:18 +0000705 n1 = buf - BUF(v);
706 if (n1 != n2)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000707 _PyString_Resize(&v, n1);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000708 return v;
709}
710
Guido van Rossum0bd24411991-04-04 15:21:57 +0000711/* External C interface */
712
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000713PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000714PyFile_GetLine(PyObject *f, int n)
Guido van Rossum0bd24411991-04-04 15:21:57 +0000715{
Guido van Rossum3165fe61992-09-25 21:59:05 +0000716 if (f == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000717 PyErr_BadInternalCall();
Guido van Rossum0bd24411991-04-04 15:21:57 +0000718 return NULL;
719 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000720 if (!PyFile_Check(f)) {
721 PyObject *reader;
722 PyObject *args;
723 PyObject *result;
724 reader = PyObject_GetAttrString(f, "readline");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000725 if (reader == NULL)
726 return NULL;
727 if (n <= 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000728 args = Py_BuildValue("()");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000729 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000730 args = Py_BuildValue("(i)", n);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000731 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000732 Py_DECREF(reader);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000733 return NULL;
734 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000735 result = PyEval_CallObject(reader, args);
736 Py_DECREF(reader);
737 Py_DECREF(args);
738 if (result != NULL && !PyString_Check(result)) {
739 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000740 result = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000741 PyErr_SetString(PyExc_TypeError,
Guido van Rossum3165fe61992-09-25 21:59:05 +0000742 "object.readline() returned non-string");
743 }
744 if (n < 0 && result != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000745 char *s = PyString_AsString(result);
746 int len = PyString_Size(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000747 if (len == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000748 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000749 result = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000750 PyErr_SetString(PyExc_EOFError,
Guido van Rossum3165fe61992-09-25 21:59:05 +0000751 "EOF when reading a line");
752 }
753 else if (s[len-1] == '\n') {
754 if (result->ob_refcnt == 1)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000755 _PyString_Resize(&result, len-1);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000756 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000757 PyObject *v;
758 v = PyString_FromStringAndSize(s,
759 len-1);
760 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +0000761 result = v;
762 }
763 }
764 }
765 return result;
766 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000767 if (((PyFileObject*)f)->f_fp == NULL)
Guido van Rossumd7297e61992-07-06 14:19:26 +0000768 return err_closed();
Marc-André Lemburg1f468602000-07-05 15:32:40 +0000769 return get_line((PyFileObject *)f, n);
Guido van Rossum0bd24411991-04-04 15:21:57 +0000770}
771
772/* Python method */
773
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000774static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000775file_readline(PyFileObject *f, PyObject *args)
Guido van Rossum0bd24411991-04-04 15:21:57 +0000776{
Guido van Rossum789a1611997-05-10 22:33:55 +0000777 int n = -1;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000778
Guido van Rossumd7297e61992-07-06 14:19:26 +0000779 if (f->f_fp == NULL)
780 return err_closed();
Guido van Rossum43713e52000-02-29 13:59:29 +0000781 if (!PyArg_ParseTuple(args, "|i:readline", &n))
Guido van Rossum789a1611997-05-10 22:33:55 +0000782 return NULL;
783 if (n == 0)
784 return PyString_FromString("");
785 if (n < 0)
786 n = 0;
Marc-André Lemburg1f468602000-07-05 15:32:40 +0000787 return get_line(f, n);
Guido van Rossum0bd24411991-04-04 15:21:57 +0000788}
789
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000790static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000791file_readlines(PyFileObject *f, PyObject *args)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000792{
Guido van Rossum789a1611997-05-10 22:33:55 +0000793 long sizehint = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000794 PyObject *list;
795 PyObject *line;
Guido van Rossum6263d541997-05-10 22:07:25 +0000796 char small_buffer[SMALLCHUNK];
797 char *buffer = small_buffer;
798 size_t buffersize = SMALLCHUNK;
799 PyObject *big_buffer = NULL;
800 size_t nfilled = 0;
801 size_t nread;
Guido van Rossum789a1611997-05-10 22:33:55 +0000802 size_t totalread = 0;
Guido van Rossum6263d541997-05-10 22:07:25 +0000803 char *p, *q, *end;
804 int err;
Guido van Rossum0bd24411991-04-04 15:21:57 +0000805
Guido van Rossumd7297e61992-07-06 14:19:26 +0000806 if (f->f_fp == NULL)
807 return err_closed();
Guido van Rossum43713e52000-02-29 13:59:29 +0000808 if (!PyArg_ParseTuple(args, "|l:readlines", &sizehint))
Guido van Rossum0bd24411991-04-04 15:21:57 +0000809 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000810 if ((list = PyList_New(0)) == NULL)
Guido van Rossumce5ba841991-03-06 13:06:18 +0000811 return NULL;
812 for (;;) {
Guido van Rossum6263d541997-05-10 22:07:25 +0000813 Py_BEGIN_ALLOW_THREADS
814 errno = 0;
815 nread = fread(buffer+nfilled, 1, buffersize-nfilled, f->f_fp);
816 Py_END_ALLOW_THREADS
817 if (nread == 0) {
Guido van Rossum789a1611997-05-10 22:33:55 +0000818 sizehint = 0;
Guido van Rossum3da3fce1998-02-19 20:46:48 +0000819 if (!ferror(f->f_fp))
Guido van Rossum6263d541997-05-10 22:07:25 +0000820 break;
821 PyErr_SetFromErrno(PyExc_IOError);
822 clearerr(f->f_fp);
823 error:
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000824 Py_DECREF(list);
Guido van Rossum6263d541997-05-10 22:07:25 +0000825 list = NULL;
826 goto cleanup;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000827 }
Guido van Rossum789a1611997-05-10 22:33:55 +0000828 totalread += nread;
Guido van Rossum6263d541997-05-10 22:07:25 +0000829 p = memchr(buffer+nfilled, '\n', nread);
830 if (p == NULL) {
831 /* Need a larger buffer to fit this line */
832 nfilled += nread;
833 buffersize *= 2;
Trent Mickf29f47b2000-08-11 19:02:59 +0000834 if (buffersize > INT_MAX) {
835 PyErr_SetString(PyExc_OverflowError,
836 "line is too long for a Python string");
837 goto error;
838 }
Guido van Rossum6263d541997-05-10 22:07:25 +0000839 if (big_buffer == NULL) {
840 /* Create the big buffer */
841 big_buffer = PyString_FromStringAndSize(
842 NULL, buffersize);
843 if (big_buffer == NULL)
844 goto error;
845 buffer = PyString_AS_STRING(big_buffer);
846 memcpy(buffer, small_buffer, nfilled);
847 }
848 else {
849 /* Grow the big buffer */
850 _PyString_Resize(&big_buffer, buffersize);
851 buffer = PyString_AS_STRING(big_buffer);
852 }
853 continue;
854 }
855 end = buffer+nfilled+nread;
856 q = buffer;
857 do {
858 /* Process complete lines */
859 p++;
860 line = PyString_FromStringAndSize(q, p-q);
861 if (line == NULL)
862 goto error;
863 err = PyList_Append(list, line);
864 Py_DECREF(line);
865 if (err != 0)
866 goto error;
867 q = p;
868 p = memchr(q, '\n', end-q);
869 } while (p != NULL);
870 /* Move the remaining incomplete line to the start */
871 nfilled = end-q;
872 memmove(buffer, q, nfilled);
Guido van Rossum789a1611997-05-10 22:33:55 +0000873 if (sizehint > 0)
874 if (totalread >= (size_t)sizehint)
875 break;
Guido van Rossumce5ba841991-03-06 13:06:18 +0000876 }
Guido van Rossum6263d541997-05-10 22:07:25 +0000877 if (nfilled != 0) {
878 /* Partial last line */
879 line = PyString_FromStringAndSize(buffer, nfilled);
880 if (line == NULL)
881 goto error;
Guido van Rossum789a1611997-05-10 22:33:55 +0000882 if (sizehint > 0) {
883 /* Need to complete the last line */
Marc-André Lemburg1f468602000-07-05 15:32:40 +0000884 PyObject *rest = get_line(f, 0);
Guido van Rossum789a1611997-05-10 22:33:55 +0000885 if (rest == NULL) {
886 Py_DECREF(line);
887 goto error;
888 }
889 PyString_Concat(&line, rest);
890 Py_DECREF(rest);
891 if (line == NULL)
892 goto error;
893 }
Guido van Rossum6263d541997-05-10 22:07:25 +0000894 err = PyList_Append(list, line);
895 Py_DECREF(line);
896 if (err != 0)
897 goto error;
898 }
899 cleanup:
Guido van Rossum1109fbc1998-04-10 22:16:39 +0000900 if (big_buffer) {
Guido van Rossum6263d541997-05-10 22:07:25 +0000901 Py_DECREF(big_buffer);
Guido van Rossum1109fbc1998-04-10 22:16:39 +0000902 }
Guido van Rossumce5ba841991-03-06 13:06:18 +0000903 return list;
904}
905
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000906static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000907file_write(PyFileObject *f, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000908{
Guido van Rossumd7297e61992-07-06 14:19:26 +0000909 char *s;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000910 int n, n2;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000911 if (f->f_fp == NULL)
912 return err_closed();
Guido van Rossum4c08d552000-03-10 22:55:18 +0000913 if (!PyArg_Parse(args, f->f_binary ? "s#" : "t#", &s, &n))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000914 return NULL;
Guido van Rossumeb183da1991-04-04 10:44:06 +0000915 f->f_softspace = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000916 Py_BEGIN_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000917 errno = 0;
Guido van Rossumd7297e61992-07-06 14:19:26 +0000918 n2 = fwrite(s, 1, n, f->f_fp);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000919 Py_END_ALLOW_THREADS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000920 if (n2 != n) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000921 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossumfebd5511992-03-04 16:39:24 +0000922 clearerr(f->f_fp);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000923 return NULL;
924 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000925 Py_INCREF(Py_None);
926 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000927}
928
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000929static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000930file_writelines(PyFileObject *f, PyObject *args)
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000931{
Guido van Rossumee70ad12000-03-13 16:27:06 +0000932#define CHUNKSIZE 1000
933 PyObject *list, *line;
934 PyObject *result;
935 int i, j, index, len, nwritten, islist;
936
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000937 if (f->f_fp == NULL)
938 return err_closed();
Guido van Rossumee70ad12000-03-13 16:27:06 +0000939 if (args == NULL || !PySequence_Check(args)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000940 PyErr_SetString(PyExc_TypeError,
Guido van Rossumee70ad12000-03-13 16:27:06 +0000941 "writelines() requires sequence of strings");
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000942 return NULL;
943 }
Guido van Rossumee70ad12000-03-13 16:27:06 +0000944 islist = PyList_Check(args);
945
946 /* Strategy: slurp CHUNKSIZE lines into a private list,
947 checking that they are all strings, then write that list
948 without holding the interpreter lock, then come back for more. */
949 index = 0;
950 if (islist)
951 list = NULL;
952 else {
953 list = PyList_New(CHUNKSIZE);
954 if (list == NULL)
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000955 return NULL;
Guido van Rossum5a2a6831993-10-25 09:59:04 +0000956 }
Guido van Rossumee70ad12000-03-13 16:27:06 +0000957 result = NULL;
958
959 for (;;) {
960 if (islist) {
961 Py_XDECREF(list);
962 list = PyList_GetSlice(args, index, index+CHUNKSIZE);
963 if (list == NULL)
964 return NULL;
965 j = PyList_GET_SIZE(list);
966 }
967 else {
968 for (j = 0; j < CHUNKSIZE; j++) {
969 line = PySequence_GetItem(args, index+j);
970 if (line == NULL) {
971 if (PyErr_ExceptionMatches(
Marc-André Lemburg6ef68b52000-08-25 22:39:50 +0000972 PyExc_IndexError)) {
Guido van Rossumee70ad12000-03-13 16:27:06 +0000973 PyErr_Clear();
974 break;
975 }
976 /* Some other error occurred.
977 XXX We may lose some output. */
978 goto error;
979 }
Guido van Rossumee70ad12000-03-13 16:27:06 +0000980 PyList_SetItem(list, j, line);
981 }
982 }
983 if (j == 0)
984 break;
985
Marc-André Lemburg6ef68b52000-08-25 22:39:50 +0000986 /* Check that all entries are indeed strings. If not,
987 apply the same rules as for file.write() and
988 convert the results to strings. This is slow, but
989 seems to be the only way since all conversion APIs
990 could potentially execute Python code. */
991 for (i = 0; i < j; i++) {
992 PyObject *v = PyList_GET_ITEM(list, i);
993 if (!PyString_Check(v)) {
994 const char *buffer;
995 int len;
996 if (((f->f_binary &&
997 PyObject_AsReadBuffer(v,
998 (const void**)&buffer,
999 &len)) ||
1000 PyObject_AsCharBuffer(v,
1001 &buffer,
1002 &len))) {
1003 PyErr_SetString(PyExc_TypeError,
1004 "writelines() requires sequences of strings");
1005 goto error;
1006 }
1007 line = PyString_FromStringAndSize(buffer,
1008 len);
1009 if (line == NULL)
1010 goto error;
1011 Py_DECREF(v);
Marc-André Lemburgf5e96fa2000-08-25 22:49:05 +00001012 PyList_SET_ITEM(list, i, line);
Marc-André Lemburg6ef68b52000-08-25 22:39:50 +00001013 }
1014 }
1015
1016 /* Since we are releasing the global lock, the
1017 following code may *not* execute Python code. */
Guido van Rossumee70ad12000-03-13 16:27:06 +00001018 Py_BEGIN_ALLOW_THREADS
1019 f->f_softspace = 0;
1020 errno = 0;
1021 for (i = 0; i < j; i++) {
Marc-André Lemburg6ef68b52000-08-25 22:39:50 +00001022 line = PyList_GET_ITEM(list, i);
Guido van Rossumee70ad12000-03-13 16:27:06 +00001023 len = PyString_GET_SIZE(line);
1024 nwritten = fwrite(PyString_AS_STRING(line),
1025 1, len, f->f_fp);
1026 if (nwritten != len) {
1027 Py_BLOCK_THREADS
1028 PyErr_SetFromErrno(PyExc_IOError);
1029 clearerr(f->f_fp);
1030 goto error;
1031 }
1032 }
1033 Py_END_ALLOW_THREADS
1034
1035 if (j < CHUNKSIZE)
1036 break;
1037 index += CHUNKSIZE;
1038 }
1039
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001040 Py_INCREF(Py_None);
Guido van Rossumee70ad12000-03-13 16:27:06 +00001041 result = Py_None;
1042 error:
1043 Py_XDECREF(list);
1044 return result;
Guido van Rossum5a2a6831993-10-25 09:59:04 +00001045}
1046
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001047static PyMethodDef file_methods[] = {
Guido van Rossum789a1611997-05-10 22:33:55 +00001048 {"readline", (PyCFunction)file_readline, 1},
Guido van Rossum74ba2471997-07-13 03:56:50 +00001049 {"read", (PyCFunction)file_read, 1},
1050 {"write", (PyCFunction)file_write, 0},
1051 {"fileno", (PyCFunction)file_fileno, 0},
Guido van Rossum88303191999-01-04 17:22:18 +00001052 {"seek", (PyCFunction)file_seek, 1},
Guido van Rossumd7047b31995-01-02 19:07:15 +00001053#ifdef HAVE_FTRUNCATE
Guido van Rossum88303191999-01-04 17:22:18 +00001054 {"truncate", (PyCFunction)file_truncate, 1},
Guido van Rossumd7047b31995-01-02 19:07:15 +00001055#endif
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001056 {"tell", (PyCFunction)file_tell, 0},
Guido van Rossumfdf95dd1997-05-05 22:15:02 +00001057 {"readinto", (PyCFunction)file_readinto, 0},
Guido van Rossum74ba2471997-07-13 03:56:50 +00001058 {"readlines", (PyCFunction)file_readlines, 1},
1059 {"writelines", (PyCFunction)file_writelines, 0},
1060 {"flush", (PyCFunction)file_flush, 0},
1061 {"close", (PyCFunction)file_close, 0},
1062 {"isatty", (PyCFunction)file_isatty, 0},
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001063 {NULL, NULL} /* sentinel */
1064};
1065
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001066#define OFF(x) offsetof(PyFileObject, x)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001067
1068static struct memberlist file_memberlist[] = {
1069 {"softspace", T_INT, OFF(f_softspace)},
1070 {"mode", T_OBJECT, OFF(f_mode), RO},
1071 {"name", T_OBJECT, OFF(f_name), RO},
1072 /* getattr(f, "closed") is implemented without this table */
1073 {"closed", T_INT, 0, RO},
1074 {NULL} /* Sentinel */
1075};
1076
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001077static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +00001078file_getattr(PyFileObject *f, char *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001079{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001080 PyObject *res;
Guido van Rossumb6775db1994-08-01 11:34:53 +00001081
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001082 res = Py_FindMethod(file_methods, (PyObject *)f, name);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001083 if (res != NULL)
1084 return res;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001085 PyErr_Clear();
Guido van Rossumb6775db1994-08-01 11:34:53 +00001086 if (strcmp(name, "closed") == 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001087 return PyInt_FromLong((long)(f->f_fp == 0));
1088 return PyMember_Get((char *)f, file_memberlist, name);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001089}
1090
1091static int
Fred Drakefd99de62000-07-09 05:02:18 +00001092file_setattr(PyFileObject *f, char *name, PyObject *v)
Guido van Rossumb6775db1994-08-01 11:34:53 +00001093{
1094 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001095 PyErr_SetString(PyExc_AttributeError,
1096 "can't delete file attributes");
Guido van Rossumb6775db1994-08-01 11:34:53 +00001097 return -1;
1098 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001099 return PyMember_Set((char *)f, file_memberlist, name, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001100}
1101
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001102PyTypeObject PyFile_Type = {
1103 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001104 0,
1105 "file",
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001106 sizeof(PyFileObject),
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001107 0,
Guido van Rossumb6775db1994-08-01 11:34:53 +00001108 (destructor)file_dealloc, /*tp_dealloc*/
Guido van Rossum7066dd71992-09-17 17:54:56 +00001109 0, /*tp_print*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001110 (getattrfunc)file_getattr, /*tp_getattr*/
1111 (setattrfunc)file_setattr, /*tp_setattr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001112 0, /*tp_compare*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001113 (reprfunc)file_repr, /*tp_repr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001114};
Guido van Rossumeb183da1991-04-04 10:44:06 +00001115
1116/* Interface for the 'soft space' between print items. */
1117
1118int
Fred Drakefd99de62000-07-09 05:02:18 +00001119PyFile_SoftSpace(PyObject *f, int newflag)
Guido van Rossumeb183da1991-04-04 10:44:06 +00001120{
1121 int oldflag = 0;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001122 if (f == NULL) {
1123 /* Do nothing */
1124 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001125 else if (PyFile_Check(f)) {
1126 oldflag = ((PyFileObject *)f)->f_softspace;
1127 ((PyFileObject *)f)->f_softspace = newflag;
Guido van Rossumeb183da1991-04-04 10:44:06 +00001128 }
Guido van Rossum3165fe61992-09-25 21:59:05 +00001129 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001130 PyObject *v;
1131 v = PyObject_GetAttrString(f, "softspace");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001132 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001133 PyErr_Clear();
Guido van Rossum3165fe61992-09-25 21:59:05 +00001134 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001135 if (PyInt_Check(v))
1136 oldflag = PyInt_AsLong(v);
1137 Py_DECREF(v);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001138 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001139 v = PyInt_FromLong((long)newflag);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001140 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001141 PyErr_Clear();
Guido van Rossum3165fe61992-09-25 21:59:05 +00001142 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001143 if (PyObject_SetAttrString(f, "softspace", v) != 0)
1144 PyErr_Clear();
1145 Py_DECREF(v);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001146 }
1147 }
Guido van Rossumeb183da1991-04-04 10:44:06 +00001148 return oldflag;
1149}
Guido van Rossum3165fe61992-09-25 21:59:05 +00001150
1151/* Interfaces to write objects/strings to file-like objects */
1152
1153int
Fred Drakefd99de62000-07-09 05:02:18 +00001154PyFile_WriteObject(PyObject *v, PyObject *f, int flags)
Guido van Rossum3165fe61992-09-25 21:59:05 +00001155{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001156 PyObject *writer, *value, *args, *result;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001157 if (f == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001158 PyErr_SetString(PyExc_TypeError, "writeobject with NULL file");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001159 return -1;
1160 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001161 else if (PyFile_Check(f)) {
1162 FILE *fp = PyFile_AsFile(f);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001163 if (fp == NULL) {
1164 err_closed();
1165 return -1;
1166 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001167 return PyObject_Print(v, fp, flags);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001168 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001169 writer = PyObject_GetAttrString(f, "write");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001170 if (writer == NULL)
1171 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001172 if (flags & Py_PRINT_RAW)
1173 value = PyObject_Str(v);
Guido van Rossumc6004111993-11-05 10:22:19 +00001174 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001175 value = PyObject_Repr(v);
Guido van Rossumc6004111993-11-05 10:22:19 +00001176 if (value == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001177 Py_DECREF(writer);
Guido van Rossumc6004111993-11-05 10:22:19 +00001178 return -1;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001179 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001180 args = Py_BuildValue("(O)", value);
Guido van Rossume9eec541997-05-22 14:02:25 +00001181 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001182 Py_DECREF(value);
1183 Py_DECREF(writer);
Guido van Rossumd3f9a1a1995-07-10 23:32:26 +00001184 return -1;
1185 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001186 result = PyEval_CallObject(writer, args);
1187 Py_DECREF(args);
1188 Py_DECREF(value);
1189 Py_DECREF(writer);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001190 if (result == NULL)
1191 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001192 Py_DECREF(result);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001193 return 0;
1194}
1195
Guido van Rossum27a60b11997-05-22 22:25:11 +00001196int
Fred Drakefd99de62000-07-09 05:02:18 +00001197PyFile_WriteString(char *s, PyObject *f)
Guido van Rossum3165fe61992-09-25 21:59:05 +00001198{
1199 if (f == NULL) {
Guido van Rossum27a60b11997-05-22 22:25:11 +00001200 /* Should be caused by a pre-existing error */
Fred Drakefd99de62000-07-09 05:02:18 +00001201 if (!PyErr_Occurred())
Guido van Rossum27a60b11997-05-22 22:25:11 +00001202 PyErr_SetString(PyExc_SystemError,
1203 "null file for PyFile_WriteString");
1204 return -1;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001205 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001206 else if (PyFile_Check(f)) {
1207 FILE *fp = PyFile_AsFile(f);
Guido van Rossum27a60b11997-05-22 22:25:11 +00001208 if (fp == NULL) {
1209 err_closed();
1210 return -1;
1211 }
1212 fputs(s, fp);
1213 return 0;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001214 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001215 else if (!PyErr_Occurred()) {
1216 PyObject *v = PyString_FromString(s);
Guido van Rossum27a60b11997-05-22 22:25:11 +00001217 int err;
1218 if (v == NULL)
1219 return -1;
1220 err = PyFile_WriteObject(v, f, Py_PRINT_RAW);
1221 Py_DECREF(v);
1222 return err;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001223 }
Guido van Rossum74ba2471997-07-13 03:56:50 +00001224 else
1225 return -1;
Guido van Rossum3165fe61992-09-25 21:59:05 +00001226}
Andrew M. Kuchling06051ed2000-07-13 23:56:54 +00001227
1228/* Try to get a file-descriptor from a Python object. If the object
1229 is an integer or long integer, its value is returned. If not, the
1230 object's fileno() method is called if it exists; the method must return
1231 an integer or long integer, which is returned as the file descriptor value.
1232 -1 is returned on failure.
1233*/
1234
1235int PyObject_AsFileDescriptor(PyObject *o)
1236{
1237 int fd;
1238 PyObject *meth;
1239
1240 if (PyInt_Check(o)) {
1241 fd = PyInt_AsLong(o);
1242 }
1243 else if (PyLong_Check(o)) {
1244 fd = PyLong_AsLong(o);
1245 }
1246 else if ((meth = PyObject_GetAttrString(o, "fileno")) != NULL)
1247 {
1248 PyObject *fno = PyEval_CallObject(meth, NULL);
1249 Py_DECREF(meth);
1250 if (fno == NULL)
1251 return -1;
1252
1253 if (PyInt_Check(fno)) {
1254 fd = PyInt_AsLong(fno);
1255 Py_DECREF(fno);
1256 }
1257 else if (PyLong_Check(fno)) {
1258 fd = PyLong_AsLong(fno);
1259 Py_DECREF(fno);
1260 }
1261 else {
1262 PyErr_SetString(PyExc_TypeError,
1263 "fileno() returned a non-integer");
1264 Py_DECREF(fno);
1265 return -1;
1266 }
1267 }
1268 else {
1269 PyErr_SetString(PyExc_TypeError,
1270 "argument must be an int, or have a fileno() method.");
1271 return -1;
1272 }
1273
1274 if (fd < 0) {
1275 PyErr_Format(PyExc_ValueError,
1276 "file descriptor cannot be a negative integer (%i)",
1277 fd);
1278 return -1;
1279 }
1280 return fd;
1281}
1282
1283
1284