Regenerate META.yml
[p5sagit/p5-mst-13.2.git] / win32 / win32io.c
1 #define PERL_NO_GET_CONTEXT
2 #define WIN32_LEAN_AND_MEAN
3 #define WIN32IO_IS_STDIO
4 #include <tchar.h>
5 #ifdef __GNUC__
6 #define Win32_Winsock
7 #endif
8 #include <windows.h>
9
10 #include <sys/stat.h>
11 #include "EXTERN.h"
12 #include "perl.h"
13
14 #ifdef PERLIO_LAYERS
15
16 #include "perliol.h"
17
18 #define NO_XSLOCKS
19 #include "XSUB.h"
20
21
22 /* Bottom-most level for Win32 case */
23
24 typedef struct
25 {
26  struct _PerlIO base;       /* The generic part */
27  HANDLE         h;          /* OS level handle */
28  IV             refcnt;     /* REFCNT for the "fd" this represents */
29  int            fd;         /* UNIX like file descriptor - index into fdtable */
30 } PerlIOWin32;
31
32 PerlIOWin32 *fdtable[256];
33 IV max_open_fd = -1;
34
35 IV
36 PerlIOWin32_popped(pTHX_ PerlIO *f)
37 {
38  PerlIOWin32 *s = PerlIOSelf(f,PerlIOWin32);
39  if (--s->refcnt > 0)
40   {
41    *f = PerlIOBase(f)->next;
42    return 1;
43   }
44  fdtable[s->fd] = NULL;
45  return 0;
46 }
47
48 IV
49 PerlIOWin32_fileno(pTHX_ PerlIO *f)
50 {
51  return PerlIOSelf(f,PerlIOWin32)->fd;
52 }
53
54 IV
55 PerlIOWin32_pushed(pTHX_ PerlIO *f, const char *mode, SV *arg, PerlIO_funcs *tab)
56 {
57  IV code = PerlIOBase_pushed(aTHX_ f,mode,arg,tab);
58  if (*PerlIONext(f))
59   {
60    PerlIOWin32 *s = PerlIOSelf(f,PerlIOWin32);
61    s->fd     = PerlIO_fileno(PerlIONext(f));
62   }
63  PerlIOBase(f)->flags |= PERLIO_F_OPEN;
64  return code;
65 }
66
67 PerlIO *
68 PerlIOWin32_open(pTHX_ PerlIO_funcs *self, PerlIO_list_t *layers, IV n, const char *mode, int fd, int imode, int perm, PerlIO *f, int narg, SV **args)
69 {
70  const char *tmode = mode;
71  HANDLE h = INVALID_HANDLE_VALUE;
72  if (f)
73   {
74    /* Close if already open */
75    if (PerlIOBase(f)->flags & PERLIO_F_OPEN)
76     (*PerlIOBase(f)->tab->Close)(aTHX_ f);
77   }
78  if (narg > 0)
79   {
80    char *path = SvPV_nolen(*args);
81    DWORD  access = 0;
82    DWORD  share  = 0;
83    DWORD  create = -1;
84    DWORD  attr   = FILE_ATTRIBUTE_NORMAL;
85    if (*mode == '#')
86     {
87      /* sysopen - imode is UNIX-like O_RDONLY etc.
88         - do_open has converted that back to string form in mode as well
89         - perm is UNIX like permissions
90       */
91      mode++;
92     }
93    else
94     {
95      /* Normal open - decode mode string */
96     }
97    switch(*mode)
98     {
99      case 'r':
100       access  = GENERIC_READ;
101       create  = OPEN_EXISTING;
102       if (*++mode == '+')
103        {
104         access |= GENERIC_WRITE;
105         create  = OPEN_ALWAYS;
106         mode++;
107        }
108       break;
109
110      case 'w':
111       access  = GENERIC_WRITE;
112       create  = TRUNCATE_EXISTING;
113       if (*++mode == '+')
114        {
115         access |= GENERIC_READ;
116         mode++;
117        }
118       break;
119
120      case 'a':
121       access = GENERIC_WRITE;
122       create  = OPEN_ALWAYS;
123       if (*++mode == '+')
124        {
125         access |= GENERIC_READ;
126         mode++;
127        }
128       break;
129     }
130    if (*mode == 'b')
131     {
132      mode++;
133     }
134    else if (*mode == 't')
135     {
136      mode++;
137     }
138    if (*mode || create == -1)
139     {
140      SETERRNO(EINVAL,LIB$_INVARG);
141      return NULL;
142     }
143    if (!(access & GENERIC_WRITE))
144     share = FILE_SHARE_READ;
145    h = CreateFile(path,access,share,NULL,create,attr,NULL);
146    if (h == INVALID_HANDLE_VALUE)
147     {
148      if (create == TRUNCATE_EXISTING)
149       h = CreateFile(path,access,share,NULL,(create = OPEN_ALWAYS),attr,NULL);
150     }
151   }
152  else
153   {
154    /* fd open */
155    h = INVALID_HANDLE_VALUE;
156    if (fd >= 0 && fd <= max_open_fd)
157     {
158      PerlIOWin32 *s = fdtable[fd];
159      if (s)
160       {
161        s->refcnt++;
162        if (!f)
163         f = PerlIO_allocate(aTHX);
164        *f = &s->base;
165        return f;
166       }
167     }
168    if (*mode == 'I')
169     {
170      mode++;
171      switch(fd)
172       {
173        case 0:
174         h = GetStdHandle(STD_INPUT_HANDLE);
175         break;
176        case 1:
177         h = GetStdHandle(STD_OUTPUT_HANDLE);
178         break;
179        case 2:
180         h = GetStdHandle(STD_ERROR_HANDLE);
181         break;
182       }
183     }
184   }
185  if (h != INVALID_HANDLE_VALUE)
186   fd = win32_open_osfhandle((intptr_t) h, PerlIOUnix_oflags(tmode));
187  if (fd >= 0)
188   {
189    PerlIOWin32 *s;
190    if (!f)
191     f = PerlIO_allocate(aTHX);
192    s = PerlIOSelf(PerlIO_push(aTHX_ f,self,tmode,PerlIOArg),PerlIOWin32);
193    s->h      = h;
194    s->fd     = fd;
195    s->refcnt = 1;
196    if (fd >= 0)
197     {
198      fdtable[fd] = s;
199      if (fd > max_open_fd)
200       max_open_fd = fd;
201     }
202    return f;
203   }
204  if (f)
205   {
206    /* FIXME: pop layers ??? */
207   }
208  return NULL;
209 }
210
211 SSize_t
212 PerlIOWin32_read(pTHX_ PerlIO *f, void *vbuf, Size_t count)
213 {
214  PerlIOWin32 *s = PerlIOSelf(f,PerlIOWin32);
215  DWORD len;
216  if (!(PerlIOBase(f)->flags & PERLIO_F_CANREAD))
217   return 0;
218  if (ReadFile(s->h,vbuf,count,&len,NULL))
219   {
220    return len;
221   }
222  else
223   {
224    if (GetLastError() != NO_ERROR)
225     {
226      PerlIOBase(f)->flags |= PERLIO_F_ERROR;
227      return -1;
228     }
229    else
230     {
231      if (count != 0)
232       PerlIOBase(f)->flags |= PERLIO_F_EOF;
233      return 0;
234     }
235   }
236 }
237
238 SSize_t
239 PerlIOWin32_write(pTHX_ PerlIO *f, const void *vbuf, Size_t count)
240 {
241  PerlIOWin32 *s = PerlIOSelf(f,PerlIOWin32);
242  DWORD len;
243  if (WriteFile(s->h,vbuf,count,&len,NULL))
244   {
245    return len;
246   }
247  else
248   {
249    PerlIOBase(f)->flags |= PERLIO_F_ERROR;
250    return -1;
251   }
252 }
253
254 IV
255 PerlIOWin32_seek(pTHX_ PerlIO *f, Off_t offset, int whence)
256 {
257  static const DWORD where[3] = { FILE_BEGIN, FILE_CURRENT, FILE_END };
258  PerlIOWin32 *s = PerlIOSelf(f,PerlIOWin32);
259  DWORD high = (sizeof(offset) > sizeof(DWORD)) ? (DWORD)(offset >> 32) : 0;
260  DWORD low  = (DWORD) offset;
261  DWORD res  = SetFilePointer(s->h,(LONG)low,(LONG *)&high,where[whence]);
262  if (res != 0xFFFFFFFF || GetLastError() != NO_ERROR)
263   {
264    return 0;
265   }
266  else
267   {
268    return -1;
269   }
270 }
271
272 Off_t
273 PerlIOWin32_tell(pTHX_ PerlIO *f)
274 {
275  PerlIOWin32 *s = PerlIOSelf(f,PerlIOWin32);
276  DWORD high = 0;
277  DWORD res  = SetFilePointer(s->h,0,(LONG *)&high,FILE_CURRENT);
278  if (res != 0xFFFFFFFF || GetLastError() != NO_ERROR)
279   {
280    return ((Off_t) high << 32) | res;
281   }
282  return (Off_t) -1;
283 }
284
285 IV
286 PerlIOWin32_close(pTHX_ PerlIO *f)
287 {
288  PerlIOWin32 *s = PerlIOSelf(f,PerlIOWin32);
289  if (s->refcnt == 1)
290   {
291    IV code = 0; 
292 #if 0
293    /* This does not do pipes etc. correctly */  
294    if (!CloseHandle(s->h))
295     {
296      s->h = INVALID_HANDLE_VALUE;
297      return -1;
298     }
299 #else
300     PerlIOBase(f)->flags &= ~PERLIO_F_OPEN;
301     return win32_close(s->fd);
302 #endif
303   }
304  return 0;
305 }
306
307 PerlIO *
308 PerlIOWin32_dup(pTHX_ PerlIO *f, PerlIO *o, CLONE_PARAMS *params, int flags)
309 {
310  PerlIOWin32 *os = PerlIOSelf(f,PerlIOWin32);
311  HANDLE proc = GetCurrentProcess();
312  HANDLE new;
313  if (DuplicateHandle(proc, os->h, proc, &new, 0, FALSE,  DUPLICATE_SAME_ACCESS))
314   {
315    char mode[8];
316    int fd = win32_open_osfhandle((intptr_t) new, PerlIOUnix_oflags(PerlIO_modestr(o,mode)));
317    if (fd >= 0)
318     {
319      f = PerlIOBase_dup(aTHX_ f, o, params, flags);
320      if (f)
321       {
322        PerlIOWin32 *fs = PerlIOSelf(f,PerlIOWin32);
323        fs->h  = new;
324        fs->fd = fd;
325        fs->refcnt = 1;
326        fdtable[fd] = fs;
327        if (fd > max_open_fd)
328         max_open_fd = fd;
329       }
330      else
331       {
332        win32_close(fd);
333       }
334     }
335    else
336     {
337      CloseHandle(new);
338     }
339   }
340  return f;
341 }
342
343 PERLIO_FUNCS_DECL(PerlIO_win32) = {
344  sizeof(PerlIO_funcs),
345  "win32",
346  sizeof(PerlIOWin32),
347  PERLIO_K_RAW,
348  PerlIOWin32_pushed,
349  PerlIOWin32_popped,
350  PerlIOWin32_open,
351  PerlIOBase_binmode,
352  NULL,                 /* getarg */
353  PerlIOWin32_fileno,
354  PerlIOWin32_dup,
355  PerlIOWin32_read,
356  PerlIOBase_unread,
357  PerlIOWin32_write,
358  PerlIOWin32_seek,
359  PerlIOWin32_tell,
360  PerlIOWin32_close,
361  PerlIOBase_noop_ok,   /* flush */
362  PerlIOBase_noop_fail, /* fill */
363  PerlIOBase_eof,
364  PerlIOBase_error,
365  PerlIOBase_clearerr,
366  PerlIOBase_setlinebuf,
367  NULL, /* get_base */
368  NULL, /* get_bufsiz */
369  NULL, /* get_ptr */
370  NULL, /* get_cnt */
371  NULL, /* set_ptrcnt */
372 };
373
374 #endif
375