basis/unix/unix.factor

   1 ! Copyright (C) 2005, 2007 Slava Pestov.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3
   4 USING: alien alien.c-types alien.syntax kernel libc structs sequences
   5        continuations byte-arrays strings
   6        math namespaces system combinators vocabs.loader qualified
   7        accessors stack-checker macros locals generalizations
   8        unix.types debugger io prettyprint ;
   9
  10 IN: unix
  11
  12 TYPEDEF: uint in_addr_t
  13 TYPEDEF: uint socklen_t
  14
  15 : PROT_NONE   0 ; inline
  16 : PROT_READ   1 ; inline
  17 : PROT_WRITE  2 ; inline
  18 : PROT_EXEC   4 ; inline
  19
  20 : MAP_FILE    0 ; inline
  21 : MAP_SHARED  1 ; inline
  22 : MAP_PRIVATE 2 ; inline
  23
  24 : MAP_FAILED -1 <alien> ; inline
  25
  26 : NGROUPS_MAX 16 ; inline
  27
  28 C-STRUCT: group
  29     { "char*" "gr_name" }
  30     { "char*" "gr_passwd" }
  31     { "int" "gr_gid" }
  32     { "char**" "gr_mem" } ;
  33
  34 LIBRARY: factor
  35
  36 FUNCTION: void clear_err_no ( ) ;
  37 FUNCTION: int err_no ( ) ;
  38
  39 LIBRARY: libc
  40
  41 FUNCTION: char* strerror ( int errno ) ;
  42
  43 ERROR: unix-error errno message ;
  44
  45 M: unix-error error.
  46     "Unix system call failed:" print
  47     nl
  48     dup message>> write " (" write errno>> pprint ")" print ;
  49
  50 : (io-error) ( -- * ) err_no dup strerror unix-error ;
  51
  52 : io-error ( n -- ) 0 < [ (io-error) ] when ;
  53
  54 ERROR: unix-system-call-error args errno message word ;
  55
  56 M: unix-system-call-error error.
  57     "Unix system call ``" write dup word>> pprint "'' failed:" print
  58     nl
  59     dup message>> write " (" write dup errno>> pprint ")" print
  60     nl
  61     "It was called with the following arguments:" print
  62     nl
  63     args>> stack. ;
  64
  65 MACRO:: unix-system-call ( quot -- )
  66     [let | n [ quot infer in>> ]
  67            word [ quot first ] |
  68         [
  69             n ndup quot call dup 0 < [
  70                 drop
  71                 n narray
  72                 err_no dup strerror
  73                 word unix-system-call-error
  74             ] [
  75                 n nnip
  76             ] if
  77         ]
  78     ] ;
  79
  80 FUNCTION: int accept ( int s, void* sockaddr, socklen_t* socklen ) ;
  81 FUNCTION: int bind ( int s, void* name, socklen_t namelen ) ;
  82 FUNCTION: int chdir ( char* path ) ;
  83 FUNCTION: int chown ( char* path, uid_t owner, gid_t group ) ;
  84 FUNCTION: int chroot ( char* path ) ;
  85
  86 FUNCTION: int close ( int fd ) ;
  87
  88 : close-file ( fd -- ) [ close ] unix-system-call drop ;
  89
  90 FUNCTION: int connect ( int s, void* name, socklen_t namelen ) ;
  91 FUNCTION: int dup2 ( int oldd, int newd ) ;
  92 ! FUNCTION: int dup ( int oldd ) ;
  93 : _exit ( status -- * )
  94     #! We throw to give this a terminating stack effect.
  95     "int" f "_exit" { "int" } alien-invoke "Exit failed" throw ;
  96 FUNCTION: int fchdir ( int fd ) ;
  97 FUNCTION: int fchown ( int fd, uid_t owner, gid_t group ) ;
  98 FUNCTION: int fcntl ( int fd, int cmd, int arg ) ;
  99 FUNCTION: int flock ( int fd, int operation ) ;
 100 FUNCTION: void freeaddrinfo ( addrinfo* ai ) ;
 101 FUNCTION: int futimes ( int id, timeval[2] times ) ;
 102 FUNCTION: char* gai_strerror ( int ecode ) ;
 103 FUNCTION: int getaddrinfo ( char* hostname, char* servname, addrinfo* hints, addrinfo** res ) ;
 104 FUNCTION: char* getcwd ( char* buf, size_t size ) ;
 105 FUNCTION: pid_t getpid ;
 106 FUNCTION: int getdtablesize ;
 107 FUNCTION: gid_t getegid ;
 108 FUNCTION: uid_t geteuid ;
 109 FUNCTION: gid_t getgid ;
 110 FUNCTION: int getgrgid_r ( gid_t gid, group* grp, char* buffer, size_t bufsize, group** result ) ;
 111 FUNCTION: int getgrnam_r ( char* name, group* grp, char* buffer, size_t bufsize, group** result ) ;
 112 FUNCTION: int getpwnam_r ( char* login, passwd* pwd, char* buffer, size_t bufsize, passwd** result ) ;
 113 FUNCTION: int getgroups ( int gidsetlen, gid_t* gidset ) ;
 114 FUNCTION: int gethostname ( char* name, int len ) ;
 115 FUNCTION: int getsockname ( int socket, sockaddr* address, socklen_t* address_len ) ;
 116 FUNCTION: int getpeername ( int socket, sockaddr* address, socklen_t* address_len ) ;
 117 FUNCTION: uid_t getuid ;
 118 FUNCTION: uint htonl ( uint n ) ;
 119 FUNCTION: ushort htons ( ushort n ) ;
 120 ! FUNCTION: int issetugid ;
 121 FUNCTION: int ioctl ( int fd, ulong request, char* argp ) ;
 122 FUNCTION: int lchown ( char* path, uid_t owner, gid_t group ) ;
 123 FUNCTION: int listen ( int s, int backlog ) ;
 124 FUNCTION: off_t lseek ( int fildes, off_t offset, int whence ) ;
 125 FUNCTION: void* mmap ( void* addr, size_t len, int prot, int flags, int fd, off_t offset ) ;
 126 FUNCTION: int munmap ( void* addr, size_t len ) ;
 127 FUNCTION: uint ntohl ( uint n ) ;
 128 FUNCTION: ushort ntohs ( ushort n ) ;
 129 FUNCTION: int shutdown ( int fd, int how ) ;
 130
 131 FUNCTION: int open ( char* path, int flags, int prot ) ;
 132
 133 : open-file ( path flags mode -- fd ) [ open ] unix-system-call ;
 134
 135 C-STRUCT: utimbuf
 136     { "time_t" "actime"  }
 137     { "time_t" "modtime" } ;
 138
 139 FUNCTION: int utime ( char* path, utimebuf* buf ) ;
 140
 141 : touch ( filename -- ) f [ utime ] unix-system-call drop ;
 142
 143 : change-file-times ( filename access modification -- )
 144     "utimebuf" <c-object>
 145     tuck set-utimbuf-modtime
 146     tuck set-utimbuf-actime
 147     [ utime ] unix-system-call drop ;
 148
 149 FUNCTION: int pclose ( void* file ) ;
 150 FUNCTION: int pipe ( int* filedes ) ;
 151 FUNCTION: void* popen ( char* command, char* type ) ;
 152 FUNCTION: ssize_t read ( int fd, void* buf, size_t nbytes ) ;
 153
 154 FUNCTION: ssize_t readlink ( char* path, char* buf, size_t bufsize ) ;
 155
 156 : PATH_MAX 1024 ; inline
 157
 158 : read-symbolic-link ( path -- path )
 159     PATH_MAX <byte-array> dup >r
 160     PATH_MAX
 161     [ readlink ] unix-system-call
 162     r> swap head-slice >string ;
 163
 164 FUNCTION: ssize_t recv ( int s, void* buf, size_t nbytes, int flags ) ;
 165 FUNCTION: ssize_t recvfrom ( int s, void* buf, size_t nbytes, int flags, sockaddr-in* from, socklen_t* fromlen ) ;
 166 FUNCTION: int rename ( char* from, char* to ) ;
 167 FUNCTION: int rmdir ( char* path ) ;
 168 FUNCTION: int select ( int nfds, void* readfds, void* writefds, void* exceptfds, timeval* timeout ) ;
 169 FUNCTION: ssize_t sendto ( int s, void* buf, size_t len, int flags, sockaddr-in* to, socklen_t tolen ) ;
 170 FUNCTION: int setegid ( gid_t egid ) ;
 171 FUNCTION: int seteuid ( uid_t euid ) ;
 172 FUNCTION: int setgid ( gid_t gid ) ;
 173 FUNCTION: int setgroups ( int ngroups, gid_t* gidset ) ;
 174 FUNCTION: int setregid ( gid_t rgid, gid_t egid ) ;
 175 FUNCTION: int setreuid ( uid_t ruid, uid_t euid ) ;
 176 FUNCTION: int setsockopt ( int s, int level, int optname, void* optval, socklen_t optlen ) ;
 177 FUNCTION: int setuid ( uid_t uid ) ;
 178 FUNCTION: int socket ( int domain, int type, int protocol ) ;
 179 FUNCTION: int symlink ( char* path1, char* path2 ) ;
 180 FUNCTION: int system ( char* command ) ;
 181
 182 FUNCTION: int unlink ( char* path ) ;
 183
 184 : unlink-file ( path -- ) [ unlink ] unix-system-call drop ;
 185
 186 FUNCTION: int utimes ( char* path, timeval[2] times ) ;
 187
 188 FUNCTION: ssize_t write ( int fd, void* buf, size_t nbytes ) ;
 189
 190 {
 191     { [ os linux? ] [ "unix.linux" require ] }
 192     { [ os bsd? ] [ "unix.bsd" require ] }
 193     { [ os solaris? ] [ "unix.solaris" require ] }
 194 } cond