Once again syncing after too long an absence
[p5sagit/p5-mst-13.2.git] / vms / ext / Stdio / Stdio.xs
1 /* VMS::Stdio - VMS extensions to stdio routines 
2  *
3  * Version:  2.2
4  * Author:   Charles Bailey  bailey@newman.upenn.edu
5  * Revised:  18-Jul-1998
6  *
7  */
8
9 #include "EXTERN.h"
10 #include "perl.h"
11 #include "XSUB.h"
12 #include <file.h>
13 #include <iodef.h>
14 #include <rms.h>
15 #include <starlet.h>
16
17 static bool
18 constant(name, pval)
19 char *name;
20 IV *pval;
21 {
22     if (strnNE(name, "O_", 2)) return FALSE;
23
24     if (strEQ(name, "O_APPEND"))
25 #ifdef O_APPEND
26         { *pval = O_APPEND; return TRUE; }
27 #else
28         return FALSE;
29 #endif
30     if (strEQ(name, "O_CREAT"))
31 #ifdef O_CREAT
32         { *pval = O_CREAT; return TRUE; }
33 #else
34         return FALSE;
35 #endif
36     if (strEQ(name, "O_EXCL"))
37 #ifdef O_EXCL
38         { *pval = O_EXCL; return TRUE; }
39 #else
40         return FALSE;
41 #endif
42     if (strEQ(name, "O_NDELAY"))
43 #ifdef O_NDELAY
44         { *pval = O_NDELAY; return TRUE; }
45 #else
46         return FALSE;
47 #endif
48     if (strEQ(name, "O_NOWAIT"))
49 #ifdef O_NOWAIT
50         { *pval = O_NOWAIT; return TRUE; }
51 #else
52         return FALSE;
53 #endif
54     if (strEQ(name, "O_RDONLY"))
55 #ifdef O_RDONLY
56         { *pval = O_RDONLY; return TRUE; }
57 #else
58         return FALSE;
59 #endif
60     if (strEQ(name, "O_RDWR"))
61 #ifdef O_RDWR
62         { *pval = O_RDWR; return TRUE; }
63 #else
64         return FALSE;
65 #endif
66     if (strEQ(name, "O_TRUNC"))
67 #ifdef O_TRUNC
68         { *pval = O_TRUNC; return TRUE; }
69 #else
70         return FALSE;
71 #endif
72     if (strEQ(name, "O_WRONLY"))
73 #ifdef O_WRONLY
74         { *pval = O_WRONLY; return TRUE; }
75 #else
76         return FALSE;
77 #endif
78
79     return FALSE;
80 }
81
82
83 static SV *
84 newFH(FILE *fp, char type) {
85     SV *rv;
86     GV **stashp, *gv = (GV *)NEWSV(0,0);
87     HV *stash;
88     IO *io;
89
90     /* Find stash for VMS::Stdio.  We don't do this once at boot
91      * to allow for possibility of threaded Perl with per-thread
92      * symbol tables.  This code (through io = ...) is really
93      * equivalent to gv_fetchpv("VMS::Stdio::__FH__",TRUE,SVt_PVIO),
94      * with a little less overhead, and good exercise for me. :-) */
95     stashp = (GV **)hv_fetch(PL_defstash,"VMS::",5,TRUE);
96     if (!stashp || *stashp == (GV *)&PL_sv_undef) return Nullsv;
97     if (!(stash = GvHV(*stashp))) stash = GvHV(*stashp) = newHV();
98     stashp = (GV **)hv_fetch(GvHV(*stashp),"Stdio::",7,TRUE);
99     if (!stashp || *stashp == (GV *)&PL_sv_undef) return Nullsv;
100     if (!(stash = GvHV(*stashp))) stash = GvHV(*stashp) = newHV();
101
102     /* Set up GV to point to IO, and then take reference */
103     gv_init(gv,stash,"__FH__",6,0);
104     io = GvIOp(gv) = newIO();
105     IoIFP(io) = fp;
106     if (type != '<') IoOFP(io) = fp;
107     IoTYPE(io) = type;
108     rv = newRV((SV *)gv);
109     SvREFCNT_dec(gv);
110     return sv_bless(rv,stash);
111 }
112
113 MODULE = VMS::Stdio  PACKAGE = VMS::Stdio
114
115 void
116 constant(name)
117         char *  name
118         PROTOTYPE: $
119         CODE:
120         IV i;
121         if (constant(name, &i))
122             ST(0) = sv_2mortal(newSViv(i));
123         else
124             ST(0) = &PL_sv_undef;
125
126 void
127 binmode(fh)
128         SV *    fh
129         PROTOTYPE: $
130         CODE:
131             IO *io = sv_2io(fh);
132             FILE *fp = io ? IoOFP(io) : NULL;
133             char iotype = io ? IoTYPE(io) : '\0';
134             char filespec[NAM$C_MAXRSS], *acmode, *s, *colon, *dirend = Nullch;
135             int ret = 0, saverrno = errno, savevmserrno = vaxc$errno;
136             fpos_t pos;
137             if (fp == NULL || strchr(">was+-|",iotype) == Nullch) {
138               set_errno(EBADF); set_vaxc_errno(SS$_IVCHAN); XSRETURN_UNDEF;
139             }
140             if (!fgetname(fp,filespec)) XSRETURN_UNDEF;
141             for (s = filespec; *s; s++) {
142               if (*s == ':') colon = s;
143               else if (*s == ']' || *s == '>') dirend = s;
144             }
145             /* Looks like a tmpfile, which will go away if reopened */
146             if (s == dirend + 3) {
147               set_errno(EBADF); set_vaxc_errno(RMS$_IOP); XSRETURN_UNDEF;
148             }
149             /* If we've got a non-file-structured device, clip off the trailing
150              * junk, and don't lose sleep if we can't get a stream position.  */
151             if (dirend == Nullch) *(colon+1) = '\0'; 
152             if (iotype != '-' && (ret = fgetpos(fp, &pos)) == -1 && dirend)
153               XSRETURN_UNDEF;
154             switch (iotype) {
155               case '<': case 'r':           acmode = "rb";                      break;
156               case '>': case 'w': case '|':
157                 /* use 'a' instead of 'w' to avoid creating new file;
158                    fsetpos below will take care of restoring file position */
159               case 'a':                     acmode = "ab";                      break;
160               case '+':  case 's':          acmode = "rb+";                     break;
161               case '-':                     acmode = fileno(fp) ? "ab" : "rb";  break;
162               /* iotype'll be null for the SYS$INPUT:/SYS$OUTPUT:/SYS$ERROR: files */
163               /* since we didn't really open them and can't really */
164               /* reopen them */
165               case 0:                       XSRETURN_UNDEF;
166               default:
167                 if (PL_dowarn) warn("Unrecognized iotype %c for %s in binmode",
168                                  iotype, filespec);
169                 acmode = "rb+";
170             }
171             if (freopen(filespec,acmode,fp) == NULL) XSRETURN_UNDEF;
172             if (iotype != '-' && ret != -1 && fsetpos(fp,&pos) == -1) XSRETURN_UNDEF;
173             if (ret == -1) { set_errno(saverrno); set_vaxc_errno(savevmserrno); }
174             XSRETURN_YES;
175
176
177 void
178 flush(fp)
179         FILE *  fp
180         PROTOTYPE: $
181         CODE:
182             if (fflush(fp)) { ST(0) = &PL_sv_undef; }
183             else            { clearerr(fp); ST(0) = &PL_sv_yes; }
184
185 char *
186 getname(fp)
187         FILE *  fp
188         PROTOTYPE: $
189         CODE:
190             char fname[NAM$C_MAXRSS+1];
191             ST(0) = sv_newmortal();
192             if (fgetname(fp,fname) != NULL) sv_setpv(ST(0),fname);
193
194 void
195 rewind(fp)
196         FILE *  fp
197         PROTOTYPE: $
198         CODE:
199             ST(0) = rewind(fp) ? &PL_sv_undef : &PL_sv_yes;
200
201 void
202 remove(name)
203         char *name
204         PROTOTYPE: $
205         CODE:
206             ST(0) = remove(name) ? &PL_sv_undef : &PL_sv_yes;
207
208 void
209 setdef(...)
210         PROTOTYPE: @
211         CODE:
212             char vmsdef[NAM$C_MAXRSS+1], es[NAM$C_MAXRSS], sep;
213             unsigned long int retsts;
214             struct FAB deffab = cc$rms_fab;
215             struct NAM defnam = cc$rms_nam;
216             struct dsc$descriptor_s dirdsc = {0, DSC$K_DTYPE_T, DSC$K_CLASS_S, 0};
217             STRLEN n_a;
218             if (items) {
219                 SV *defsv = ST(items-1);  /* mimic chdir() */
220                 ST(0) = &PL_sv_undef;
221                 if (!SvPOK(defsv)) { SETERRNO(EINVAL,LIB$_INVARG); XSRETURN(1); }
222                 if (tovmsspec(SvPV(defsv,n_a),vmsdef) == NULL) { XSRETURN(1); }
223                 deffab.fab$l_fna = vmsdef; deffab.fab$b_fns = strlen(vmsdef);
224             }
225             else {
226                 deffab.fab$l_fna = "SYS$LOGIN"; deffab.fab$b_fns = 9;
227                 EXTEND(sp,1);  ST(0) = &PL_sv_undef;
228             }
229             defnam.nam$l_esa = es;  defnam.nam$b_ess = sizeof es;
230             deffab.fab$l_nam = &defnam;
231             retsts = sys$parse(&deffab,0,0);
232             if (retsts & 1) {
233                 if (defnam.nam$v_wildcard) retsts = RMS$_WLD;
234                 else if (defnam.nam$b_name || defnam.nam$b_type > 1 ||
235                      defnam.nam$b_ver > 1) retsts = RMS$_DIR;
236                 }
237             defnam.nam$b_nop |= NAM$M_SYNCHK; defnam.nam$l_rlf = NULL; deffab.fab$b_dns = 0;
238             if (!(retsts & 1)) {
239                 set_vaxc_errno(retsts);
240                 switch (retsts) {
241                     case RMS$_DNF:
242                         set_errno(ENOENT); break;
243                     case RMS$_SYN: case RMS$_DIR: case RMS$_DEV:
244                         set_errno(EINVAL); break;
245                     case RMS$_PRV:
246                         set_errno(EACCES); break;
247                     default:
248                         set_errno(EVMSERR); break;
249                 }
250                 (void) sys$parse(&deffab,0,0);  /* free up context */
251                 XSRETURN(1);
252             }
253             sep = *defnam.nam$l_dir;
254             *defnam.nam$l_dir = '\0';
255             my_setenv("SYS$DISK",defnam.nam$b_node ? defnam.nam$l_node : defnam.nam$l_dev);
256             *defnam.nam$l_dir = sep;
257             dirdsc.dsc$a_pointer = defnam.nam$l_dir; dirdsc.dsc$w_length = defnam.nam$b_dir;
258             if ((retsts = sys$setddir(&dirdsc,0,0)) & 1) ST(0) = &PL_sv_yes;
259             else { set_errno(EVMSERR); set_vaxc_errno(retsts); }
260             (void) sys$parse(&deffab,0,0);  /* free up context */
261
262 void
263 sync(fp)
264         FILE *  fp
265         PROTOTYPE: $
266         CODE:
267             if (fsync(fileno(fp))) { ST(0) = &PL_sv_undef; }
268             else                   { clearerr(fp); ST(0) = &PL_sv_yes; }
269
270 char *
271 tmpnam()
272         PROTOTYPE:
273         CODE:
274             char fname[L_tmpnam];
275             ST(0) = sv_newmortal();
276             if (tmpnam(fname) != NULL) sv_setpv(ST(0),fname);
277
278 void
279 vmsopen(spec,...)
280         char *  spec
281         PROTOTYPE: @
282         CODE:
283             char *args[8],mode[3] = {'r','\0','\0'}, type = '<';
284             register int i, myargc;
285             FILE *fp;
286             STRLEN n_a;
287         
288             if (!spec || !*spec) {
289                SETERRNO(EINVAL,LIB$_INVARG);
290                XSRETURN_UNDEF;
291             }
292             if (items > 9) croak("too many args");
293         
294             /* First, set up name and mode args from perl's string */
295             if (*spec == '+') {
296               mode[1] = '+';
297               spec++;
298             }
299             if (*spec == '>') {
300               if (*(spec+1) == '>') *mode = 'a', spec += 2;
301               else *mode = 'w',  spec++;
302             }
303             else if (*spec == '<') spec++;
304             myargc = items - 1;
305             for (i = 0; i < myargc; i++) args[i] = SvPV(ST(i+1),n_a);
306             /* This hack brought to you by C's opaque arglist management */
307             switch (myargc) {
308               case 0:
309                 fp = fopen(spec,mode);
310                 break;
311               case 1:
312                 fp = fopen(spec,mode,args[0]);
313                 break;
314               case 2:
315                 fp = fopen(spec,mode,args[0],args[1]);
316                 break;
317               case 3:
318                 fp = fopen(spec,mode,args[0],args[1],args[2]);
319                 break;
320               case 4:
321                 fp = fopen(spec,mode,args[0],args[1],args[2],args[3]);
322                 break;
323               case 5:
324                 fp = fopen(spec,mode,args[0],args[1],args[2],args[3],args[4]);
325                 break;
326               case 6:
327                 fp = fopen(spec,mode,args[0],args[1],args[2],args[3],args[4],args[5]);
328                 break;
329               case 7:
330                 fp = fopen(spec,mode,args[0],args[1],args[2],args[3],args[4],args[5],args[6]);
331                 break;
332               case 8:
333                 fp = fopen(spec,mode,args[0],args[1],args[2],args[3],args[4],args[5],args[6],args[7]);
334                 break;
335             }
336             if (fp != Nullfp) {
337               SV *fh = newFH(fp,(mode[1] ? '+' : (mode[0] == 'r' ? '<' : (mode[0] == 'a' ? 'a' : '>'))));
338               ST(0) = (fh ? sv_2mortal(fh) : &PL_sv_undef);
339             }
340             else { ST(0) = &PL_sv_undef; }
341
342 void
343 vmssysopen(spec,mode,perm,...)
344         char *  spec
345         int     mode
346         int     perm
347         PROTOTYPE: @
348         CODE:
349             char *args[8];
350             int i, myargc, fd;
351             FILE *fp;
352             SV *fh;
353             STRLEN n_a;
354             if (!spec || !*spec) {
355                SETERRNO(EINVAL,LIB$_INVARG);
356                XSRETURN_UNDEF;
357             }
358             if (items > 11) croak("too many args");
359             myargc = items - 3;
360             for (i = 0; i < myargc; i++) args[i] = SvPV(ST(i+3),n_a);
361             /* More fun with C calls; can't combine with above because
362                args 2,3 of different types in fopen() and open() */
363             switch (myargc) {
364               case 0:
365                 fd = open(spec,mode,perm);
366                 break;
367               case 1:
368                 fd = open(spec,mode,perm,args[0]);
369                 break;
370               case 2:
371                 fd = open(spec,mode,perm,args[0],args[1]);
372                 break;
373               case 3:
374                 fd = open(spec,mode,perm,args[0],args[1],args[2]);
375                 break;
376               case 4:
377                 fd = open(spec,mode,perm,args[0],args[1],args[2],args[3]);
378                 break;
379               case 5:
380                 fd = open(spec,mode,perm,args[0],args[1],args[2],args[3],args[4]);
381                 break;
382               case 6:
383                 fd = open(spec,mode,perm,args[0],args[1],args[2],args[3],args[4],args[5]);
384                 break;
385               case 7:
386                 fd = open(spec,mode,perm,args[0],args[1],args[2],args[3],args[4],args[5],args[6]);
387                 break;
388               case 8:
389                 fd = open(spec,mode,perm,args[0],args[1],args[2],args[3],args[4],args[5],args[6],args[7]);
390                 break;
391             }
392             i = mode & 3;
393             if (fd >= 0 &&
394                ((fp = fdopen(fd, &("r\000w\000r+"[2*i]))) != Nullfp)) {
395               SV *fh = newFH(fp,"<>++"[i]);
396               ST(0) = (fh ? sv_2mortal(fh) : &PL_sv_undef);
397             }
398             else { ST(0) = &PL_sv_undef; }
399
400 void
401 waitfh(fp)
402         FILE *  fp
403         PROTOTYPE: $
404         CODE:
405             ST(0) = fwait(fp) ? &PL_sv_undef : &PL_sv_yes;
406
407 void
408 writeof(mysv)
409         SV *    mysv
410         PROTOTYPE: $
411         CODE:
412             char devnam[257], *cp;
413             unsigned long int chan, iosb[2], retsts, retsts2;
414             struct dsc$descriptor devdsc = {0, DSC$K_DTYPE_T, DSC$K_CLASS_S, devnam};
415             IO *io = sv_2io(mysv);
416             FILE *fp = io ? IoOFP(io) : NULL;
417             if (fp == NULL || strchr(">was+-|",IoTYPE(io)) == Nullch) {
418               set_errno(EBADF); set_vaxc_errno(SS$_IVCHAN); XSRETURN_UNDEF;
419             }
420             if (fgetname(fp,devnam) == Nullch) { ST(0) = &PL_sv_undef; XSRETURN(1); }
421             if ((cp = strrchr(devnam,':')) != NULL) *(cp+1) = '\0';
422             devdsc.dsc$w_length = strlen(devnam);
423             retsts = sys$assign(&devdsc,&chan,0,0);
424             if (retsts & 1) retsts = sys$qiow(0,chan,IO$_WRITEOF,iosb,0,0,0,0,0,0,0,0);
425             if (retsts & 1) retsts = iosb[0];
426             retsts2 = sys$dassgn(chan);  /* Be sure to deassign the channel */
427             if (retsts & 1) retsts = retsts2;
428             if (retsts & 1) { ST(0) = &PL_sv_yes; }
429             else {
430               set_vaxc_errno(retsts);
431               switch (retsts) {
432                 case SS$_EXQUOTA:  case SS$_INSFMEM:  case SS$_MBFULL:
433                 case SS$_MBTOOSML: case SS$_NOIOCHAN: case SS$_NOLINKS:
434                 case SS$_BUFFEROVF:
435                   set_errno(ENOSPC); break;
436                 case SS$_ILLIOFUNC: case SS$_DEVOFFLINE: case SS$_NOSUCHDEV:
437                   set_errno(EBADF);  break;
438                 case SS$_NOPRIV:
439                   set_errno(EACCES); break;
440                 default:  /* Includes "shouldn't happen" cases that might map */
441                   set_errno(EVMSERR); break;         /* to other errno values */
442               }
443               ST(0) = &PL_sv_undef;
444             }