Windows: large file support for hFileSize and hSeek (#1771)
[ghc-base.git] / include / HsBase.h
1 /* -----------------------------------------------------------------------------
2  *
3  * (c) The University of Glasgow 2001-2004
4  *
5  * Definitions for package `base' which are visible in Haskell land.
6  *
7  * ---------------------------------------------------------------------------*/
8
9 #ifndef __HSBASE_H__
10 #define __HSBASE_H__
11
12 #include "HsBaseConfig.h"
13
14 /* ultra-evil... */
15 #undef PACKAGE_BUGREPORT
16 #undef PACKAGE_NAME
17 #undef PACKAGE_STRING
18 #undef PACKAGE_TARNAME
19 #undef PACKAGE_VERSION
20
21 /* Needed to get the macro version of errno on some OSs (eg. Solaris).
22    We must do this, because these libs are only compiled once, but
23    must work in both single-threaded and multi-threaded programs. */
24 #define _REENTRANT 1
25
26 #include "HsFFI.h"
27
28 #include <stdio.h>
29 #include <stdlib.h>
30 #include <math.h>
31
32 #if HAVE_SYS_TYPES_H
33 #include <sys/types.h>
34 #endif
35 #if HAVE_UNISTD_H
36 #include <unistd.h>
37 #endif
38 #if HAVE_SYS_STAT_H
39 #include <sys/stat.h>
40 #endif
41 #if HAVE_FCNTL_H
42 # include <fcntl.h>
43 #endif
44 #if HAVE_TERMIOS_H
45 #include <termios.h>
46 #endif
47 #if HAVE_SIGNAL_H
48 #include <signal.h>
49 /* Ultra-ugly: OpenBSD uses broken macros for sigemptyset and sigfillset (missing casts) */
50 #if __OpenBSD__
51 #undef sigemptyset
52 #undef sigfillset
53 #endif
54 #endif
55 #if HAVE_ERRNO_H
56 #include <errno.h>
57 #endif
58 #if HAVE_STRING_H
59 #include <string.h>
60 #endif
61 #if HAVE_DIRENT_H
62 #include <dirent.h>
63 #endif
64 #if HAVE_UTIME_H
65 #include <utime.h>
66 #endif
67 #if HAVE_SYS_UTSNAME_H
68 #include <sys/utsname.h>
69 #endif
70 #if HAVE_GETTIMEOFDAY
71 #  if HAVE_SYS_TIME_H
72 #   include <sys/time.h>
73 #  endif
74 #elif HAVE_GETCLOCK
75 # if HAVE_SYS_TIMERS_H
76 #  define POSIX_4D9 1
77 #  include <sys/timers.h>
78 # endif
79 #endif
80 #if HAVE_TIME_H
81 #include <time.h>
82 #endif
83 #if HAVE_SYS_TIMEB_H
84 #include <sys/timeb.h>
85 #endif
86 #if HAVE_WINDOWS_H
87 #include <windows.h>
88 #endif
89 #if HAVE_SYS_TIMES_H
90 #include <sys/times.h>
91 #endif
92 #if HAVE_WINSOCK_H && defined(__MINGW32__)
93 #include <winsock.h>
94 #endif
95 #if HAVE_LIMITS_H
96 #include <limits.h>
97 #endif
98 #if HAVE_WCTYPE_H
99 #include <wctype.h>
100 #endif
101 #if HAVE_INTTYPES_H
102 # include <inttypes.h>
103 #elif HAVE_STDINT_H
104 # include <stdint.h>
105 #endif
106
107 #if !defined(__MINGW32__) && !defined(irix_HOST_OS)
108 # if HAVE_SYS_RESOURCE_H
109 #  include <sys/resource.h>
110 # endif
111 #endif
112
113 #if !HAVE_GETRUSAGE && HAVE_SYS_SYSCALL_H
114 # include <sys/syscall.h>
115 # if defined(SYS_GETRUSAGE)     /* hpux_HOST_OS */
116 #  define getrusage(a, b)  syscall(SYS_GETRUSAGE, a, b)
117 #  define HAVE_GETRUSAGE 1
118 # endif
119 #endif
120
121 /* For System */
122 #if HAVE_SYS_WAIT_H
123 #include <sys/wait.h>
124 #endif
125 #if HAVE_VFORK_H
126 #include <vfork.h>
127 #endif
128 #include "dirUtils.h"
129 #include "WCsubst.h"
130
131 #if defined(__MINGW32__)
132 /* in Win32Utils.c */
133 extern void maperrno (void);
134 extern HsWord64 getUSecOfDay(void);
135 #endif
136
137 #if defined(__MINGW32__)
138 #include <io.h>
139 #include <fcntl.h>
140 #include <shlobj.h>
141 #include <share.h>
142 #endif
143
144 #if HAVE_SYS_SELECT_H
145 #include <sys/select.h>
146 #endif
147
148 /* in inputReady.c */
149 extern int fdReady(int fd, int write, int msecs, int isSock);
150
151 /* in Signals.c */
152 extern HsInt nocldstop;
153
154 /* -----------------------------------------------------------------------------
155    64-bit operations, defined in longlong.c
156    -------------------------------------------------------------------------- */
157
158 #ifdef SUPPORT_LONG_LONGS
159
160 HsBool hs_gtWord64 (HsWord64, HsWord64);
161 HsBool hs_geWord64 (HsWord64, HsWord64);
162 HsBool hs_eqWord64 (HsWord64, HsWord64);
163 HsBool hs_neWord64 (HsWord64, HsWord64);
164 HsBool hs_ltWord64 (HsWord64, HsWord64);
165 HsBool hs_leWord64 (HsWord64, HsWord64);
166
167 HsBool hs_gtInt64 (HsInt64, HsInt64);
168 HsBool hs_geInt64 (HsInt64, HsInt64);
169 HsBool hs_eqInt64 (HsInt64, HsInt64);
170 HsBool hs_neInt64 (HsInt64, HsInt64);
171 HsBool hs_ltInt64 (HsInt64, HsInt64);
172 HsBool hs_leInt64 (HsInt64, HsInt64);
173
174 HsWord64 hs_remWord64  (HsWord64, HsWord64);
175 HsWord64 hs_quotWord64 (HsWord64, HsWord64);
176
177 HsInt64 hs_remInt64    (HsInt64, HsInt64);
178 HsInt64 hs_quotInt64   (HsInt64, HsInt64);
179 HsInt64 hs_negateInt64 (HsInt64);
180 HsInt64 hs_plusInt64   (HsInt64, HsInt64);
181 HsInt64 hs_minusInt64  (HsInt64, HsInt64);
182 HsInt64 hs_timesInt64  (HsInt64, HsInt64);
183
184 HsWord64 hs_and64  (HsWord64, HsWord64);
185 HsWord64 hs_or64   (HsWord64, HsWord64);
186 HsWord64 hs_xor64  (HsWord64, HsWord64);
187 HsWord64 hs_not64  (HsWord64);
188
189 HsWord64 hs_uncheckedShiftL64   (HsWord64, HsInt);
190 HsWord64 hs_uncheckedShiftRL64  (HsWord64, HsInt);
191 HsInt64  hs_uncheckedIShiftL64  (HsInt64, HsInt);
192 HsInt64  hs_uncheckedIShiftRA64 (HsInt64, HsInt);
193 HsInt64  hs_uncheckedIShiftRL64 (HsInt64, HsInt);
194
195 HsInt64  hs_intToInt64    (HsInt);
196 HsInt    hs_int64ToInt    (HsInt64);
197 HsWord64 hs_int64ToWord64 (HsInt64);
198 HsWord64 hs_wordToWord64  (HsWord);
199 HsWord   hs_word64ToWord  (HsWord64);
200 HsInt64  hs_word64ToInt64 (HsWord64);
201
202 HsWord64 hs_integerToWord64 (HsInt sa, StgByteArray /* Really: mp_limb_t* */ da);
203 HsInt64  hs_integerToInt64 (HsInt sa, StgByteArray /* Really: mp_limb_t* */ da);
204
205 #endif /* SUPPORT_LONG_LONGS */
206
207 /* -----------------------------------------------------------------------------
208    INLINE functions.
209
210    These functions are given as inlines here for when compiling via C,
211    but we also generate static versions into the cbits library for
212    when compiling to native code.
213    -------------------------------------------------------------------------- */
214
215 #ifndef INLINE
216 # if defined(_MSC_VER)
217 #  define INLINE extern __inline
218 # else
219 #  define INLINE static inline
220 # endif
221 #endif
222
223 INLINE int __hscore_get_errno(void) { return errno; }
224 INLINE void __hscore_set_errno(int e) { errno = e; }
225
226 #if !defined(_MSC_VER)
227 INLINE int __hscore_s_isreg(mode_t m)  { return S_ISREG(m);  }
228 INLINE int __hscore_s_isdir(mode_t m)  { return S_ISDIR(m);  }
229 INLINE int __hscore_s_isfifo(mode_t m) { return S_ISFIFO(m); }
230 INLINE int __hscore_s_isblk(mode_t m)  { return S_ISBLK(m);  }
231 INLINE int __hscore_s_ischr(mode_t m)  { return S_ISCHR(m);  }
232 #ifdef S_ISSOCK
233 INLINE int __hscore_s_issock(mode_t m) { return S_ISSOCK(m); }
234 #endif
235 #endif
236
237 #if !defined(_MSC_VER) && !defined(__MINGW32__) && !defined(_WIN32)
238 INLINE int
239 __hscore_sigemptyset( sigset_t *set )
240 { return sigemptyset(set); }
241
242 INLINE int
243 __hscore_sigfillset( sigset_t *set )
244 { return sigfillset(set); }
245
246 INLINE int
247 __hscore_sigaddset( sigset_t * set, int s )
248 { return sigaddset(set,s); }
249
250 INLINE int
251 __hscore_sigdelset( sigset_t * set, int s )
252 { return sigdelset(set,s); }
253
254 INLINE int
255 __hscore_sigismember( sigset_t * set, int s )
256 { return sigismember(set,s); }
257 #endif
258
259 INLINE void *
260 __hscore_memcpy_dst_off( char *dst, int dst_off, char *src, size_t sz )
261 { return memcpy(dst+dst_off, src, sz); }
262
263 INLINE void *
264 __hscore_memcpy_src_off( char *dst, char *src, int src_off, size_t sz )
265 { return memcpy(dst, src+src_off, sz); }
266
267 INLINE HsBool
268 __hscore_supportsTextMode()
269 {
270 #if defined(_MSC_VER) || defined(__MINGW32__) || defined(_WIN32)
271   return HS_BOOL_FALSE;
272 #else
273   return HS_BOOL_TRUE;
274 #endif
275 }
276
277 INLINE HsInt
278 __hscore_bufsiz()
279 {
280   return BUFSIZ;
281 }
282
283 INLINE int
284 __hscore_seek_cur()
285 {
286   return SEEK_CUR;
287 }
288
289 INLINE int
290 __hscore_o_binary()
291 {
292 #if defined(_MSC_VER)
293   return O_BINARY;
294 #else
295   return CONST_O_BINARY;
296 #endif
297 }
298
299 INLINE int
300 __hscore_o_rdonly()
301 {
302 #ifdef O_RDONLY
303   return O_RDONLY;
304 #else
305   return 0;
306 #endif
307 }
308
309 INLINE int
310 __hscore_o_wronly( void )
311 {
312 #ifdef O_WRONLY
313   return O_WRONLY;
314 #else
315   return 0;
316 #endif
317 }
318
319 INLINE int
320 __hscore_o_rdwr( void )
321 {
322 #ifdef O_RDWR
323   return O_RDWR;
324 #else
325   return 0;
326 #endif
327 }
328
329 INLINE int
330 __hscore_o_append( void )
331 {
332 #ifdef O_APPEND
333   return O_APPEND;
334 #else
335   return 0;
336 #endif
337 }
338
339 INLINE int
340 __hscore_o_creat( void )
341 {
342 #ifdef O_CREAT
343   return O_CREAT;
344 #else
345   return 0;
346 #endif
347 }
348
349 INLINE int
350 __hscore_o_excl( void )
351 {
352 #ifdef O_EXCL
353   return O_EXCL;
354 #else
355   return 0;
356 #endif
357 }
358
359 INLINE int
360 __hscore_o_trunc( void )
361 {
362 #ifdef O_TRUNC
363   return O_TRUNC;
364 #else
365   return 0;
366 #endif
367 }
368
369 INLINE int
370 __hscore_o_noctty( void )
371 {
372 #ifdef O_NOCTTY
373   return O_NOCTTY;
374 #else
375   return 0;
376 #endif
377 }
378
379 INLINE int
380 __hscore_o_nonblock( void )
381 {
382 #ifdef O_NONBLOCK
383   return O_NONBLOCK;
384 #else
385   return 0;
386 #endif
387 }
388
389 INLINE int
390 __hscore_seek_set( void )
391 {
392   return SEEK_SET;
393 }
394
395 INLINE int
396 __hscore_seek_end( void )
397 {
398   return SEEK_END;
399 }
400
401 INLINE int
402 __hscore_ftruncate( int fd, off_t where )
403 {
404 #if defined(HAVE_FTRUNCATE)
405   return ftruncate(fd,where);
406 #elif defined(HAVE__CHSIZE)
407   return _chsize(fd,where);
408 #else
409 #error at least ftruncate or _chsize functions are required to build
410 #endif
411 }
412
413 INLINE int
414 __hscore_setmode( int fd, HsBool toBin )
415 {
416 #if defined(_MSC_VER) || defined(__MINGW32__) || defined(_WIN32)
417   return setmode(fd,(toBin == HS_BOOL_TRUE) ? _O_BINARY : _O_TEXT);
418 #else
419   return 0;
420 #endif
421 }
422
423 #if __GLASGOW_HASKELL__
424
425 INLINE int
426 __hscore_PrelHandle_write( int fd, void *ptr, HsInt off, int sz )
427 {
428   return write(fd,(char *)ptr + off, sz);
429 }
430
431 INLINE int
432 __hscore_PrelHandle_read( int fd, void *ptr, HsInt off, int sz )
433 {
434   return read(fd,(char *)ptr + off, sz);
435
436 }
437
438 #if defined(_MSC_VER) || defined(__MINGW32__) || defined(_WIN32)
439 INLINE int
440 __hscore_PrelHandle_send( int fd, void *ptr, HsInt off, int sz )
441 {
442     return send(fd,(char *)ptr + off, sz, 0);
443 }
444
445 INLINE int
446 __hscore_PrelHandle_recv( int fd, void *ptr, HsInt off, int sz )
447 {
448     return recv(fd,(char *)ptr + off, sz, 0);
449 }
450 #endif
451
452 #endif /* __GLASGOW_HASKELL__ */
453
454 INLINE int
455 __hscore_mkdir( char *pathName, int mode )
456 {
457 #if defined(_MSC_VER) || defined(__MINGW32__) || defined(_WIN32)
458   return mkdir(pathName);
459 #else
460   return mkdir(pathName,mode);
461 #endif
462 }
463
464 INLINE int
465 __hscore_lstat( const char *fname, struct stat *st )
466 {
467 #if HAVE_LSTAT
468   return lstat(fname, st);
469 #else
470   return stat(fname, st);
471 #endif
472 }
473
474 INLINE char *
475 __hscore_d_name( struct dirent* d )
476 {
477   return (d->d_name);
478 }
479
480 INLINE int
481 __hscore_end_of_dir( void )
482 {
483   return READDIR_ERRNO_EOF;
484 }
485
486 INLINE void
487 __hscore_free_dirent(struct dirent *dEnt)
488 {
489 #if HAVE_READDIR_R
490   free(dEnt);
491 #endif
492 }
493
494 #if defined(__MINGW32__)
495 // We want the versions of stat/fstat/lseek that use 64-bit offsets,
496 // and you have to ask for those explicitly.  Unfortunately there
497 // doesn't seem to be a 64-bit version of truncate/ftruncate, so while
498 // hFileSize and hSeek will work with large files, hSetFileSize will not.
499 #define stat(file,buf)       _stati64(file,buf)
500 #define fstat(fd,buf)        _fstati64(fd,buf)
501 typedef struct _stati64 struct_stat;
502 typedef off64_t stsize_t;
503 #else
504 typedef struct stat struct_stat;
505 typedef off_t stsize_t;
506 #endif
507
508 INLINE HsInt
509 __hscore_sizeof_stat( void )
510 {
511   return sizeof(struct_stat);
512 }
513
514 INLINE time_t __hscore_st_mtime ( struct_stat* st ) { return st->st_mtime; }
515 INLINE stsize_t __hscore_st_size  ( struct_stat* st ) { return st->st_size; }
516 #if !defined(_MSC_VER)
517 INLINE mode_t __hscore_st_mode  ( struct_stat* st ) { return st->st_mode; }
518 INLINE mode_t __hscore_st_dev  ( struct_stat* st ) { return st->st_dev; }
519 INLINE mode_t __hscore_st_ino  ( struct_stat* st ) { return st->st_ino; }
520 #endif
521
522 #if HAVE_TERMIOS_H
523 INLINE tcflag_t __hscore_lflag( struct termios* ts ) { return ts->c_lflag; }
524
525 INLINE void
526 __hscore_poke_lflag( struct termios* ts, tcflag_t t ) { ts->c_lflag = t; }
527
528 INLINE unsigned char*
529 __hscore_ptr_c_cc( struct termios* ts )
530 { return (unsigned char*) &ts->c_cc; }
531
532 INLINE HsInt
533 __hscore_sizeof_termios( void )
534 {
535 #ifndef __MINGW32__
536   return sizeof(struct termios);
537 #else
538   return 0;
539 #endif
540 }
541 #endif
542
543 #if !defined(_MSC_VER) && !defined(__MINGW32__) && !defined(_WIN32)
544 INLINE HsInt
545 __hscore_sizeof_sigset_t( void )
546 {
547   return sizeof(sigset_t);
548 }
549 #endif
550
551 INLINE int
552 __hscore_echo( void )
553 {
554 #ifdef ECHO
555   return ECHO;
556 #else
557   return 0;
558 #endif
559
560 }
561
562 INLINE int
563 __hscore_tcsanow( void )
564 {
565 #ifdef TCSANOW
566   return TCSANOW;
567 #else
568   return 0;
569 #endif
570
571 }
572
573 INLINE int
574 __hscore_icanon( void )
575 {
576 #ifdef ICANON
577   return ICANON;
578 #else
579   return 0;
580 #endif
581 }
582
583 INLINE int __hscore_vmin( void )
584 {
585 #ifdef VMIN
586   return VMIN;
587 #else
588   return 0;
589 #endif
590 }
591
592 INLINE int __hscore_vtime( void )
593 {
594 #ifdef VTIME
595   return VTIME;
596 #else
597   return 0;
598 #endif
599 }
600
601 INLINE int __hscore_sigttou( void )
602 {
603 #ifdef SIGTTOU
604   return SIGTTOU;
605 #else
606   return 0;
607 #endif
608 }
609
610 INLINE int __hscore_sig_block( void )
611 {
612 #ifdef SIG_BLOCK
613   return SIG_BLOCK;
614 #else
615   return 0;
616 #endif
617 }
618
619 INLINE int __hscore_sig_setmask( void )
620 {
621 #ifdef SIG_SETMASK
622   return SIG_SETMASK;
623 #else
624   return 0;
625 #endif
626 }
627
628 INLINE int
629 __hscore_f_getfl( void )
630 {
631 #ifdef F_GETFL
632   return F_GETFL;
633 #else
634   return 0;
635 #endif
636 }
637
638 INLINE int
639 __hscore_f_setfl( void )
640 {
641 #ifdef F_SETFL
642   return F_SETFL;
643 #else
644   return 0;
645 #endif
646 }
647
648 // defined in rts/RtsStartup.c.
649 extern void* __hscore_get_saved_termios(int fd);
650 extern void __hscore_set_saved_termios(int fd, void* ts);
651
652 INLINE int __hscore_hs_fileno (FILE *f) { return fileno (f); }
653
654 INLINE int __hscore_open(char *file, int how, mode_t mode) {
655 #ifdef __MINGW32__
656         if ((how & O_WRONLY) || (how & O_RDWR) || (how & O_APPEND))
657           return _sopen(file,how,_SH_DENYRW,mode);
658         else
659           return _sopen(file,how,_SH_DENYWR,mode);
660 #else
661         return open(file,how,mode);
662 #endif
663 }
664
665 // These are wrapped because on some OSs (eg. Linux) they are
666 // macros which redirect to the 64-bit-off_t versions when large file
667 // support is enabled.
668 //
669 #if defined(__MINGW32__)
670 INLINE off64_t __hscore_lseek(int fd, off64_t off, int whence) {
671         return (_lseeki64(fd,off,whence));
672 }
673 #else
674 INLINE off_t __hscore_lseek(int fd, off_t off, int whence) {
675         return (lseek(fd,off,whence));
676 }
677 #endif
678
679 INLINE int __hscore_stat(char *file, struct_stat *buf) {
680         return (stat(file,buf));
681 }
682
683 INLINE int __hscore_fstat(int fd, struct_stat *buf) {
684         return (fstat(fd,buf));
685 }
686
687 // select-related stuff
688
689 #if !defined(__MINGW32__)
690 INLINE int  hsFD_SETSIZE(void) { return FD_SETSIZE; }
691 INLINE void hsFD_CLR(int fd, fd_set *fds) { FD_CLR(fd, fds); }
692 INLINE int  hsFD_ISSET(int fd, fd_set *fds) { return FD_ISSET(fd, fds); }
693 INLINE void hsFD_SET(int fd, fd_set *fds) { FD_SET(fd, fds); }
694 INLINE HsInt sizeof_fd_set(void) { return sizeof(fd_set); }
695 extern void hsFD_ZERO(fd_set *fds);
696 #endif
697
698 // gettimeofday()-related
699
700 #if !defined(__MINGW32__)
701
702 INLINE HsInt sizeofTimeVal(void) { return sizeof(struct timeval); }
703
704 INLINE HsWord64 getUSecOfDay(void)
705 {
706     struct timeval tv;
707     gettimeofday(&tv, (struct timezone *) NULL);
708     // Don't forget to cast *before* doing the arithmetic, otherwise
709     // the arithmetic happens at the type of tv_sec, which is probably
710     // only 'int'.
711     return ((HsWord64)tv.tv_sec * 1000000 + (HsWord64)tv.tv_usec);
712 }
713
714 INLINE void setTimevalTicks(struct timeval *p, HsWord64 usecs)
715 {
716     p->tv_sec  = usecs / 1000000;
717     p->tv_usec = usecs % 1000000;
718 }
719 #endif /* !defined(__MINGW32__) */
720
721 /* ToDo: write a feature test that doesn't assume 'environ' to
722  *    be in scope at link-time. */
723 extern char** environ;
724 INLINE char **__hscore_environ() { return environ; }
725
726 /* lossless conversions between pointers and integral types */
727 INLINE void *    __hscore_from_uintptr(uintptr_t n) { return (void *)n; }
728 INLINE void *    __hscore_from_intptr (intptr_t n)  { return (void *)n; }
729 INLINE uintptr_t __hscore_to_uintptr  (void *p)     { return (uintptr_t)p; }
730 INLINE intptr_t  __hscore_to_intptr   (void *p)     { return (intptr_t)p; }
731
732 #endif /* __HSBASE_H__ */
733