Integrate mainline
[p5sagit/p5-mst-13.2.git] / ext / IO / IO.xs
1 /*
2  * Copyright (c) 1997-8 Graham Barr <gbarr@pobox.com>. All rights reserved.
3  * This program is free software; you can redistribute it and/or
4  * modify it under the same terms as Perl itself.
5  */
6
7 #define PERL_EXT_IO
8
9 #define PERL_NO_GET_CONTEXT
10 #include "EXTERN.h"
11 #define PERLIO_NOT_STDIO 1
12 #include "perl.h"
13 #include "XSUB.h"
14 #include "poll.h"
15 #ifdef I_UNISTD
16 #  include <unistd.h>
17 #endif
18 #if defined(I_FCNTL) || defined(HAS_FCNTL)
19 #  include <fcntl.h>
20 #endif
21
22 #ifndef SIOCATMARK
23 #   ifdef I_SYS_SOCKIO
24 #       include <sys/sockio.h>
25 #   endif
26 #endif
27
28 #ifdef PerlIO
29 #ifdef MACOS_TRADITIONAL
30 #define PERLIO_IS_STDIO 1
31 #undef setbuf
32 #undef setvbuf
33 #define setvbuf         _stdsetvbuf
34 #define setbuf(f,b)     ( __sf_setbuf(f,b) )
35 #endif
36 typedef int SysRet;
37 typedef PerlIO * InputStream;
38 typedef PerlIO * OutputStream;
39 #else
40 #define PERLIO_IS_STDIO 1
41 typedef int SysRet;
42 typedef FILE * InputStream;
43 typedef FILE * OutputStream;
44 #endif
45
46 #define MY_start_subparse(fmt,flags) start_subparse(fmt,flags)
47
48 #ifndef gv_stashpvn
49 #define gv_stashpvn(str,len,flags) gv_stashpv(str,flags)
50 #endif
51
52 static int
53 not_here(char *s)
54 {
55     croak("%s not implemented on this architecture", s);
56     return -1;
57 }
58
59
60 #ifndef PerlIO
61 #define PerlIO_fileno(f) fileno(f)
62 #endif
63
64 static int
65 io_blocking(pTHX_ InputStream f, int block)
66 {
67     int RETVAL;
68     if(!f) {
69         errno = EBADF;
70         return -1;
71     }
72 #if defined(HAS_FCNTL)
73     RETVAL = fcntl(PerlIO_fileno(f), F_GETFL, 0);
74     if (RETVAL >= 0) {
75         int mode = RETVAL;
76 #ifdef O_NONBLOCK
77         /* POSIX style */
78 #if defined(O_NDELAY) && O_NDELAY != O_NONBLOCK
79         /* Ooops has O_NDELAY too - make sure we don't
80          * get SysV behaviour by mistake. */
81
82         /* E.g. In UNICOS and UNICOS/mk a F_GETFL returns an O_NDELAY
83          * after a successful F_SETFL of an O_NONBLOCK. */
84         RETVAL = RETVAL & (O_NONBLOCK | O_NDELAY) ? 0 : 1;
85
86         if (block >= 0) {
87             if ((mode & O_NDELAY) || ((block == 0) && !(mode & O_NONBLOCK))) {
88                 int ret;
89                 mode = (mode & ~O_NDELAY) | O_NONBLOCK;
90                 ret = fcntl(PerlIO_fileno(f),F_SETFL,mode);
91                 if(ret < 0)
92                     RETVAL = ret;
93             }
94             else
95               if ((mode & O_NDELAY) || ((block > 0) && (mode & O_NONBLOCK))) {
96                 int ret;
97                 mode &= ~(O_NONBLOCK | O_NDELAY);
98                 ret = fcntl(PerlIO_fileno(f),F_SETFL,mode);
99                 if(ret < 0)
100                     RETVAL = ret;
101               }
102         }
103 #else
104         /* Standard POSIX */
105         RETVAL = RETVAL & O_NONBLOCK ? 0 : 1;
106
107         if ((block == 0) && !(mode & O_NONBLOCK)) {
108             int ret;
109             mode |= O_NONBLOCK;
110             ret = fcntl(PerlIO_fileno(f),F_SETFL,mode);
111             if(ret < 0)
112                 RETVAL = ret;
113          }
114         else if ((block > 0) && (mode & O_NONBLOCK)) {
115             int ret;
116             mode &= ~O_NONBLOCK;
117             ret = fcntl(PerlIO_fileno(f),F_SETFL,mode);
118             if(ret < 0)
119                 RETVAL = ret;
120          }
121 #endif
122 #else
123         /* Not POSIX - better have O_NDELAY or we can't cope.
124          * for BSD-ish machines this is an acceptable alternative
125          * for SysV we can't tell "would block" from EOF but that is
126          * the way SysV is...
127          */
128         RETVAL = RETVAL & O_NDELAY ? 0 : 1;
129
130         if ((block == 0) && !(mode & O_NDELAY)) {
131             int ret;
132             mode |= O_NDELAY;
133             ret = fcntl(PerlIO_fileno(f),F_SETFL,mode);
134             if(ret < 0)
135                 RETVAL = ret;
136          }
137         else if ((block > 0) && (mode & O_NDELAY)) {
138             int ret;
139             mode &= ~O_NDELAY;
140             ret = fcntl(PerlIO_fileno(f),F_SETFL,mode);
141             if(ret < 0)
142                 RETVAL = ret;
143          }
144 #endif
145     }
146     return RETVAL;
147 #else
148  return -1;
149 #endif
150 }
151
152 MODULE = IO     PACKAGE = IO::Seekable  PREFIX = f
153
154 void
155 fgetpos(handle)
156         InputStream     handle
157     CODE:
158         if (handle) {
159 #ifdef PerlIO
160             ST(0) = sv_2mortal(newSV(0));
161             if (PerlIO_getpos(handle, ST(0)) != 0) {
162                 ST(0) = &PL_sv_undef;
163             }
164 #else
165             if (fgetpos(handle, &pos)) {
166                 ST(0) = &PL_sv_undef;
167             } else {
168                 ST(0) = sv_2mortal(newSVpv((char*)&pos, sizeof(Fpos_t)));
169             }
170 #endif
171         }
172         else {
173             ST(0) = &PL_sv_undef;
174             errno = EINVAL;
175         }
176
177 SysRet
178 fsetpos(handle, pos)
179         InputStream     handle
180         SV *            pos
181     CODE:
182         if (handle) {
183 #ifdef PerlIO
184             RETVAL = PerlIO_setpos(handle, pos);
185 #else
186             char *p;
187             STRLEN len;
188             if ((p = SvPV(pos,len)) && len == sizeof(Fpos_t)) {
189                 RETVAL = fsetpos(handle, (Fpos_t*)p);
190             }
191             else {
192                 RETVAL = -1;
193                 errno = EINVAL;
194             }
195 #endif
196         }
197         else {
198             RETVAL = -1;
199             errno = EINVAL;
200         }
201     OUTPUT:
202         RETVAL
203
204 MODULE = IO     PACKAGE = IO::File      PREFIX = f
205
206 void
207 new_tmpfile(packname = "IO::File")
208     char *              packname
209     PREINIT:
210         OutputStream fp;
211         GV *gv;
212     CODE:
213 #ifdef PerlIO
214         fp = PerlIO_tmpfile();
215 #else
216         fp = tmpfile();
217 #endif
218         gv = (GV*)SvREFCNT_inc(newGVgen(packname));
219         hv_delete(GvSTASH(gv), GvNAME(gv), GvNAMELEN(gv), G_DISCARD);
220         if (do_open(gv, "+>&", 3, FALSE, 0, 0, fp)) {
221             ST(0) = sv_2mortal(newRV((SV*)gv));
222             sv_bless(ST(0), gv_stashpv(packname, TRUE));
223             SvREFCNT_dec(gv);   /* undo increment in newRV() */
224         }
225         else {
226             ST(0) = &PL_sv_undef;
227             SvREFCNT_dec(gv);
228         }
229
230 MODULE = IO     PACKAGE = IO::Poll
231
232 void
233 _poll(timeout,...)
234         int timeout;
235 PPCODE:
236 {
237 #ifdef HAS_POLL
238     int nfd = (items - 1) / 2;
239     SV *tmpsv = NEWSV(999,nfd * sizeof(struct pollfd));
240     struct pollfd *fds = (struct pollfd *)SvPVX(tmpsv);
241     int i,j,ret;
242     for(i=1, j=0  ; j < nfd ; j++) {
243         fds[j].fd = SvIV(ST(i));
244         i++;
245         fds[j].events = SvIV(ST(i));
246         i++;
247         fds[j].revents = 0;
248     }
249     if((ret = poll(fds,nfd,timeout)) >= 0) {
250         for(i=1, j=0 ; j < nfd ; j++) {
251             sv_setiv(ST(i), fds[j].fd); i++;
252             sv_setiv(ST(i), fds[j].revents); i++;
253         }
254     }
255     SvREFCNT_dec(tmpsv);
256     XSRETURN_IV(ret);
257 #else
258         not_here("IO::Poll::poll");
259 #endif
260 }
261
262 MODULE = IO     PACKAGE = IO::Handle    PREFIX = io_
263
264 void
265 io_blocking(handle,blk=-1)
266         InputStream     handle
267         int             blk
268 PROTOTYPE: $;$
269 CODE:
270 {
271     int ret = io_blocking(aTHX_ handle, items == 1 ? -1 : blk ? 1 : 0);
272     if(ret >= 0)
273         XSRETURN_IV(ret);
274     else
275         XSRETURN_UNDEF;
276 }
277
278 MODULE = IO     PACKAGE = IO::Handle    PREFIX = f
279
280 int
281 ungetc(handle, c)
282         InputStream     handle
283         int             c
284     CODE:
285         if (handle)
286 #ifdef PerlIO
287             RETVAL = PerlIO_ungetc(handle, c);
288 #else
289             RETVAL = ungetc(c, handle);
290 #endif
291         else {
292             RETVAL = -1;
293             errno = EINVAL;
294         }
295     OUTPUT:
296         RETVAL
297
298 int
299 ferror(handle)
300         InputStream     handle
301     CODE:
302         if (handle)
303 #ifdef PerlIO
304             RETVAL = PerlIO_error(handle);
305 #else
306             RETVAL = ferror(handle);
307 #endif
308         else {
309             RETVAL = -1;
310             errno = EINVAL;
311         }
312     OUTPUT:
313         RETVAL
314
315 int
316 clearerr(handle)
317         InputStream     handle
318     CODE:
319         if (handle) {
320 #ifdef PerlIO
321             PerlIO_clearerr(handle);
322 #else
323             clearerr(handle);
324 #endif
325             RETVAL = 0;
326         }
327         else {
328             RETVAL = -1;
329             errno = EINVAL;
330         }
331     OUTPUT:
332         RETVAL
333
334 int
335 untaint(handle)
336        SV *     handle
337     CODE:
338 #ifdef IOf_UNTAINT
339         IO * io;
340         io = sv_2io(handle);
341         if (io) {
342             IoFLAGS(io) |= IOf_UNTAINT;
343             RETVAL = 0;
344         }
345         else {
346 #endif
347             RETVAL = -1;
348             errno = EINVAL;
349 #ifdef IOf_UNTAINT
350         }
351 #endif
352     OUTPUT:
353         RETVAL
354
355 SysRet
356 fflush(handle)
357         OutputStream    handle
358     CODE:
359         if (handle)
360 #ifdef PerlIO
361             RETVAL = PerlIO_flush(handle);
362 #else
363             RETVAL = Fflush(handle);
364 #endif
365         else {
366             RETVAL = -1;
367             errno = EINVAL;
368         }
369     OUTPUT:
370         RETVAL
371
372 void
373 setbuf(handle, ...)
374         OutputStream    handle
375     CODE:
376         if (handle)
377 #ifdef PERLIO_IS_STDIO
378         {
379             char *buf = items == 2 && SvPOK(ST(1)) ?
380               sv_grow(ST(1), BUFSIZ) : 0;
381             setbuf(handle, buf);
382         }
383 #else
384             not_here("IO::Handle::setbuf");
385 #endif
386
387 SysRet
388 setvbuf(...)
389     CODE:
390         if (items != 4)
391             Perl_croak(aTHX_ "Usage: IO::Handle::setvbuf(handle, buf, type, size)");
392 #if defined(PERLIO_IS_STDIO) && defined(_IOFBF) && defined(HAS_SETVBUF)
393     {
394         OutputStream    handle = 0;
395         char *          buf = SvPOK(ST(1)) ? sv_grow(ST(1), SvIV(ST(3))) : 0;
396         int             type;
397         int             size;
398
399         if (items == 4) {
400             handle = IoOFP(sv_2io(ST(0)));
401             buf    = SvPOK(ST(1)) ? sv_grow(ST(1), SvIV(ST(3))) : 0;
402             type   = (int)SvIV(ST(2));
403             size   = (int)SvIV(ST(3));
404         }
405         if (!handle)                    /* Try input stream. */
406             handle = IoIFP(sv_2io(ST(0)));
407         if (items == 4 && handle)
408             RETVAL = setvbuf(handle, buf, type, size);
409         else {
410             RETVAL = -1;
411             errno = EINVAL;
412         }
413     }
414 #else
415         RETVAL = (SysRet) not_here("IO::Handle::setvbuf");
416 #endif
417     OUTPUT:
418         RETVAL
419
420
421 SysRet
422 fsync(handle)
423         OutputStream handle
424     CODE:
425 #ifdef HAS_FSYNC
426         if(handle)
427             RETVAL = fsync(PerlIO_fileno(handle));
428         else {
429             RETVAL = -1;
430             errno = EINVAL;
431         }
432 #else
433         RETVAL = (SysRet) not_here("IO::Handle::sync");
434 #endif
435     OUTPUT:
436         RETVAL
437
438
439 MODULE = IO     PACKAGE = IO::Socket
440
441 SysRet
442 sockatmark (sock)
443    InputStream sock
444    PROTOTYPE: $
445    PREINIT:
446      int fd;
447    CODE:
448    {
449      fd = PerlIO_fileno(sock);
450 #ifdef HAS_SOCKATMARK
451      RETVAL = sockatmark(fd);
452 #else
453      {
454        int flag = 0;
455 #   ifdef SIOCATMARK
456 #     if defined(NETWARE) || defined(WIN32)
457        if (ioctl(fd, SIOCATMARK, (void*)&flag) != 0)
458 #     else
459        if (ioctl(fd, SIOCATMARK, &flag) != 0)
460 #     endif
461          XSRETURN_UNDEF;
462 #   else
463        not_here("IO::Socket::atmark");
464 #   endif
465        RETVAL = flag;
466      }
467 #endif
468    }
469    OUTPUT:
470      RETVAL
471
472 BOOT:
473 {
474     HV *stash;
475     /*
476      * constant subs for IO::Poll
477      */
478     stash = gv_stashpvn("IO::Poll", 8, TRUE);
479 #ifdef  POLLIN
480         newCONSTSUB(stash,"POLLIN",newSViv(POLLIN));
481 #endif
482 #ifdef  POLLPRI
483         newCONSTSUB(stash,"POLLPRI", newSViv(POLLPRI));
484 #endif
485 #ifdef  POLLOUT
486         newCONSTSUB(stash,"POLLOUT", newSViv(POLLOUT));
487 #endif
488 #ifdef  POLLRDNORM
489         newCONSTSUB(stash,"POLLRDNORM", newSViv(POLLRDNORM));
490 #endif
491 #ifdef  POLLWRNORM
492         newCONSTSUB(stash,"POLLWRNORM", newSViv(POLLWRNORM));
493 #endif
494 #ifdef  POLLRDBAND
495         newCONSTSUB(stash,"POLLRDBAND", newSViv(POLLRDBAND));
496 #endif
497 #ifdef  POLLWRBAND
498         newCONSTSUB(stash,"POLLWRBAND", newSViv(POLLWRBAND));
499 #endif
500 #ifdef  POLLNORM
501         newCONSTSUB(stash,"POLLNORM", newSViv(POLLNORM));
502 #endif
503 #ifdef  POLLERR
504         newCONSTSUB(stash,"POLLERR", newSViv(POLLERR));
505 #endif
506 #ifdef  POLLHUP
507         newCONSTSUB(stash,"POLLHUP", newSViv(POLLHUP));
508 #endif
509 #ifdef  POLLNVAL
510         newCONSTSUB(stash,"POLLNVAL", newSViv(POLLNVAL));
511 #endif
512     /*
513      * constant subs for IO::Handle
514      */
515     stash = gv_stashpvn("IO::Handle", 10, TRUE);
516 #ifdef _IOFBF
517         newCONSTSUB(stash,"_IOFBF", newSViv(_IOFBF));
518 #endif
519 #ifdef _IOLBF
520         newCONSTSUB(stash,"_IOLBF", newSViv(_IOLBF));
521 #endif
522 #ifdef _IONBF
523         newCONSTSUB(stash,"_IONBF", newSViv(_IONBF));
524 #endif
525 #ifdef SEEK_SET
526         newCONSTSUB(stash,"SEEK_SET", newSViv(SEEK_SET));
527 #endif
528 #ifdef SEEK_CUR
529         newCONSTSUB(stash,"SEEK_CUR", newSViv(SEEK_CUR));
530 #endif
531 #ifdef SEEK_END
532         newCONSTSUB(stash,"SEEK_END", newSViv(SEEK_END));
533 #endif
534 }
535