source: trunk/source/level-1/l1-sockets.lisp @ 12105

Last change on this file since 12105 was 12105, checked in by gb, 10 years ago

Ignore the new FD argument to MAKE-SOCKET.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 54.4 KB
Line 
1;;;-*- Mode: Lisp; Package: CCL -*-
2;;;
3;;;   Copyright (C) 2001 Clozure Associates
4;;;   This file is part of OpenMCL. 
5;;;
6;;;   OpenMCL is licensed under the terms of the Lisp Lesser GNU Public
7;;;   License , known as the LLGPL and distributed with OpenMCL as the
8;;;   file "LICENSE".  The LLGPL consists of a preamble and the LGPL,
9;;;   which is distributed with OpenMCL as the file "LGPL".  Where these
10;;;   conflict, the preamble takes precedence. 
11;;;
12;;;   OpenMCL is referenced in the preamble as the "LIBRARY."
13;;;
14;;;   The LLGPL is also available online at
15;;;   http://opensource.franz.com/preamble.html
16
17(in-package "CCL")
18
19
20
21;;; basic socket API
22(eval-when (:compile-toplevel :load-toplevel :execute)
23  (export '(MAKE-SOCKET
24            ACCEPT-CONNECTION
25            DOTTED-TO-IPADDR
26            IPADDR-TO-DOTTED
27            IPADDR-TO-HOSTNAME
28            LOOKUP-HOSTNAME
29            LOOKUP-PORT
30            ;;with-pending-connect
31            RECEIVE-FROM
32            SEND-TO
33            SHUTDOWN
34            ;;socket-control
35            SOCKET-OS-FD
36            REMOTE-HOST
37            REMOTE-PORT
38            REMOTE-FILENAME
39            LOCAL-HOST
40            LOCAL-PORT
41            LOCAL-FILENAME
42            SOCKET-ADDRESS-FAMILY
43            SOCKET-CONNECT
44            SOCKET-FORMAT
45            SOCKET-TYPE
46            SOCKET-ERROR
47            SOCKET-ERROR-CODE
48            SOCKET-ERROR-IDENTIFIER
49            SOCKET-ERROR-SITUATION
50            WITH-OPEN-SOCKET))
51  #+windows-target
52  (defmacro check-winsock-error (form)
53    (let* ((val (gensym)))
54      `(let* ((,val ,form))
55        (if (< ,val 0)
56          (%get-winsock-error)
57          ,val))))
58  (defmacro check-socket-error (form)
59    #+windows-target `(check-winsock-error ,form)
60    #-windows-target `(int-errno-call ,form))
61  )
62
63
64#+windows-target
65(defun %get-winsock-error ()
66  (- (#_WSAGetLastError)))
67
68;;; The PPC is big-endian (uses network byte order), which makes
69;;; things like #_htonl and #_htonl no-ops.  These functions aren't
70;;; necessarily defined as functions in some header files (I'm sure
71;;; that that either complies with or violates some C standard), and
72;;; it doesn't seem to make much sense to fight that to do ff-calls
73;;; to a couple of identity functions.
74
75#+big-endian-target
76(progn
77  (defmacro HTONL (x) x)
78  (defmacro HTONS (x) x)
79  (defmacro NTOHL (x) x)
80  (defmacro NTOHS (x) x))
81
82#+little-endian-target
83(progn
84  (declaim (inline %bswap32 %bswap16))
85  (defun %bswap32 (x)
86    (declare (type (unsigned-byte 32) x))
87    (%swap-u32 x))
88  (defun %bswap16 (x)
89    (declare (type (unsigned-byte 16) x))
90    (%swap-u16 x))
91  (defmacro HTONL (x) `(%bswap32 ,x))
92  (defmacro HTONS (x) `(%bswap16 ,x))
93  (defmacro NTOHL (x) `(%bswap32 ,x))
94  (defmacro NTOHS (x) `(%bswap16 ,x)))
95
96(defparameter *default-socket-character-encoding*
97  nil)
98
99(defmethod default-character-encoding ((domain (eql :socket)))
100  *default-socket-character-encoding*)
101 
102
103;;; On some (hypothetical) little-endian platform, we might want to
104;;; define HTONL and HTONS to actually swap bytes around.
105
106(defpackage "OPENMCL-SOCKET"
107  (:use "CL")
108  (:import-from "CCL"
109                "MAKE-SOCKET"
110                "ACCEPT-CONNECTION"
111                "DOTTED-TO-IPADDR"
112                "IPADDR-TO-DOTTED"
113                "IPADDR-TO-HOSTNAME"
114                "LOOKUP-HOSTNAME"
115                "LOOKUP-PORT"
116                ;;with-pending-connect
117                "RECEIVE-FROM"
118                "SEND-TO"
119                "SHUTDOWN"
120                ;;socket-control
121                "SOCKET-OS-FD"
122                "REMOTE-HOST"
123                "REMOTE-PORT"
124                "REMOTE-FILENAME"
125                "LOCAL-HOST"
126                "LOCAL-PORT"
127                "LOCAL-FILENAME"
128                "SOCKET-ADDRESS-FAMILY"
129                "SOCKET-CONNECT"
130                "SOCKET-FORMAT"
131                "SOCKET-TYPE"
132                "SOCKET-ERROR"
133                "SOCKET-ERROR-CODE"
134                "SOCKET-ERROR-IDENTIFIER"
135                "SOCKET-ERROR-SITUATION"
136                "WITH-OPEN-SOCKET")
137  (:export  "MAKE-SOCKET"
138            "ACCEPT-CONNECTION"
139            "DOTTED-TO-IPADDR"
140            "IPADDR-TO-DOTTED"
141            "IPADDR-TO-HOSTNAME"
142            "LOOKUP-HOSTNAME"
143            "LOOKUP-PORT"
144            ;;with-pending-connect
145            "RECEIVE-FROM"
146            "SEND-TO"
147            "SHUTDOWN"
148            ;;socket-control
149            "SOCKET-OS-FD"
150            "REMOTE-HOST"
151            "REMOTE-PORT"
152            "REMOTE-FILENAME"
153            "LOCAL-HOST"
154            "LOCAL-PORT"
155            "LOCAL-FILENAME"
156            "SOCKET-ADDRESS-FAMILY"
157            "SOCKET-CONNECT"
158            "SOCKET-FORMAT"
159            "SOCKET-TYPE"
160            "SOCKET-ERROR"
161            "SOCKET-ERROR-CODE"
162            "SOCKET-ERROR-IDENTIFIER"
163            "SOCKET-ERROR-SITUATION"
164            "WITH-OPEN-SOCKET"))
165
166
167
168(define-condition socket-error (simple-stream-error)
169  ((code :initarg :code :reader socket-error-code)
170   (identifier :initform :unknown :initarg :identifier :reader socket-error-identifier)
171   (Situation :initarg :situation :reader socket-error-situation)))
172
173(define-condition socket-creation-error (simple-error)
174  ((code :initarg :code :reader socket-creation-error-code)
175   (identifier :initform :unknown :initarg :identifier :reader socket-creation-error-identifier)
176   (situation :initarg :situation :reader socket-creation-error-situation)))
177
178(defvar *socket-error-identifiers*
179  #-windows-target
180  (list #$EADDRINUSE :address-in-use
181        #$ECONNABORTED :connection-aborted
182        #$ENOBUFS :no-buffer-space
183        #$ENOMEM :no-buffer-space
184        #$ENFILE :no-buffer-space
185        #$ETIMEDOUT :connection-timed-out
186        #$ECONNREFUSED :connection-refused
187        #$ENETUNREACH :host-unreachable
188        #$EHOSTUNREACH :host-unreachable
189        #$EHOSTDOWN :host-down
190        #$ENETDOWN :network-down
191        #$EADDRNOTAVAIL :address-not-available
192        #$ENETRESET :network-reset
193        #$ECONNRESET :connection-reset
194        #$ESHUTDOWN :shutdown
195        #$EACCES :access-denied
196        #$EPERM :access-denied)
197  #+windows-target
198  (list #$WSAEADDRINUSE :address-in-use
199        #$WSAECONNABORTED :connection-aborted
200        #$WSAENOBUFS :no-buffer-space
201        #$ENOMEM :no-buffer-space
202        #$ENFILE :no-buffer-space
203        #$WSAETIMEDOUT :connection-timed-out
204        #$WSAECONNREFUSED :connection-refused
205        #$WSAENETUNREACH :host-unreachable
206        #$WSAEHOSTUNREACH :host-unreachable
207        #$WSAEHOSTDOWN :host-down
208        #$WSAENETDOWN :network-down
209        #$WSAEADDRNOTAVAIL :address-not-available
210        #$WSAENETRESET :network-reset
211        #$WSAECONNRESET :connection-reset
212        #$WSAESHUTDOWN :shutdown
213        #$EACCES :access-denied
214        #$EPERM :access-denied)
215  )
216
217
218(declaim (inline socket-call))
219(defun socket-call (stream where res)
220  (if (< res 0)
221    (socket-error stream where res)
222    res))
223
224#-windows-target
225(defun %gai-strerror (errno)
226  (let* ((err (abs errno))
227         (p (#_gai_strerror err)))
228    (if (%null-ptr-p p)
229      (format nil "Unknown nameserver error ~d" err)
230      (%get-cstring p))))
231   
232
233
234(defun socket-error (stream where errno &optional nameserver-p)
235  "Creates and signals (via error) one of two socket error
236conditions, based on the state of the arguments."
237  #+windows-target (declare (ignore nameserver-p))
238  (when (< errno 0)
239    (setq errno (- errno)))
240  (if stream
241    (error (make-condition 'socket-error
242                           :stream stream
243                           :code errno
244                           :identifier (getf *socket-error-identifiers* errno :unknown)
245                           :situation where
246                           ;; TODO: this is a constant arg, there is a way to put this
247                           ;; in the class definition, just need to remember how...
248                           :format-control "~a (error #~d) during ~a"
249                           :format-arguments (list
250                                              #+windows-target
251                                              (%windows-error-string errno)
252                                              #-windows-target
253                                              (if nameserver-p
254                                                (%gai-strerror errno)
255                                                (%strerror errno))
256                                              errno where)))
257    (error (make-condition 'socket-creation-error
258                           :code errno
259                           :identifier (getf *socket-error-identifiers* errno :unknown)
260                           :situation where
261                           ;; TODO: this is a constant arg, there is a way to put this
262                           ;; in the class definition, just need to remember how...
263                           :format-control "~a (error #~d) during socket creation or nameserver operation in ~a"
264                           :format-arguments (list
265                                              #+windows-target
266                                              (%windows-error-string errno)
267                                              #-windows-target
268                                              (if nameserver-p
269                                                (%gai-strerror errno)
270                                                (%strerror errno))
271                                              errno where)))))
272   
273
274
275;; If true, this will try to allow other cooperative processes to run
276;; while socket io is happening.  Since CCL threads are preemptively
277;; scheduled, this isn't particularly meaningful.
278(defvar *multiprocessing-socket-io* nil)
279
280(defclass socket ()
281  ())
282
283(defmacro with-open-socket ((var . args) &body body
284                            &aux (socket (make-symbol "socket"))
285                                 (done (make-symbol "done")))
286  "Execute body with var bound to the result of applying make-socket to
287make-socket-args. The socket gets closed on exit."
288  `(let (,socket ,done)
289     (unwind-protect
290         (multiple-value-prog1
291           (let ((,var (setq ,socket (make-socket ,@args))))
292             ,@body)
293           (setq ,done t))
294       (when ,socket (close ,socket :abort (not ,done))))))
295
296(defgeneric socket-address-family (socket)
297  (:documentation "Return :internet or :file, as appropriate."))
298
299(defclass ip-socket (socket)
300  ())
301
302(defmethod socket-address-family ((socket ip-socket)) :internet)
303
304(defclass file-socket (socket)
305  ())
306
307(defmethod socket-address-family ((socket file-socket)) :file)
308
309(defclass tcp-socket (ip-socket)
310  ())
311
312(defgeneric socket-type (socket)
313  (:documentation
314   "Return :stream for tcp-stream and listener-socket, and :datagram
315for udp-socket."))
316
317(defmethod socket-type ((socket tcp-socket)) :stream)
318
319(defclass stream-file-socket (file-socket)
320  ())
321
322(defmethod socket-type ((socket stream-file-socket)) :stream)
323
324
325;;; An active TCP socket is an honest-to-goodness stream.
326(defclass tcp-stream (tcp-socket)
327  ())
328
329(defclass fundamental-tcp-stream (tcp-stream
330                                  fd-stream
331                                  buffered-binary-io-stream-mixin
332                                  buffered-character-io-stream-mixin)
333    ())
334
335(make-built-in-class 'basic-tcp-stream
336                     'tcp-stream
337                     'basic-binary-io-stream
338                     'basic-character-io-stream)
339
340(defgeneric socket-connect (stream)
341 (:documentation
342   "Return :active for tcp-stream, :passive for listener-socket, and NIL
343for udp-socket"))
344
345(defmethod socket-connect ((stream tcp-stream)) :active)
346
347(defgeneric socket-format (stream)
348  (:documentation
349   "Return the socket format as specified by the :format argument to
350make-socket."))
351
352(defmethod socket-format ((stream tcp-stream))
353  (if (eq (stream-element-type stream) 'character)
354    :text
355    ;; Should distinguish between :binary and :bivalent, but hardly
356    ;; seems worth carrying around an extra slot just for that.
357    :bivalent))
358
359(defmethod socket-device ((stream tcp-stream))
360  (let ((ioblock (stream-ioblock stream nil)))
361    (and ioblock (ioblock-device ioblock))))
362
363(defmethod select-stream-class ((class tcp-stream) in-p out-p char-p)
364  (declare (ignore char-p)) ; TODO: is there any real reason to care about this?
365  ;; Yes, in general.  There is.
366  (assert (and in-p out-p) () "Non-bidirectional tcp stream?")
367  'fundamental-tcp-stream)
368
369(defmethod map-to-basic-stream-class-name ((name (eql 'tcp-stream)))
370  'basic-tcp-stream)
371
372(defmethod select-stream-class ((s (eql 'basic-tcp-stream)) in-p out-p char-p)
373  (declare (ignore char-p))
374  (assert (and in-p out-p) () "Non-bidirectional tcp stream?")
375  'basic-tcp-stream)
376
377;;; A FILE-SOCKET-STREAM is also honest. To goodness.
378(defclass file-socket-stream (stream-file-socket)
379  ())
380
381(defclass fundamental-file-socket-stream (file-socket-stream
382                                          fd-stream
383                                          buffered-binary-io-stream-mixin
384                                          buffered-character-io-stream-mixin)
385    ())
386
387(make-built-in-class 'basic-file-socket-stream
388                     'file-socket-stream
389                     'basic-binary-io-stream
390                     'basic-character-io-stream)
391
392
393(defmethod map-to-basic-stream-class-name ((name (eql 'file-socket-stream)))
394  'basic-file-socket-stream)
395
396(defmethod select-stream-class ((class file-socket-stream) in-p out-p char-p)
397  (declare (ignore char-p)) ; TODO: is there any real reason to care about this?
398  (assert (and in-p out-p) () "Non-bidirectional file-socket stream?")
399  'fundamental-file-socket-stream)
400
401(defmethod select-stream-class ((s (eql 'basic-file-socket-stream)) in-p out-p char-p)
402  (declare (ignore char-p))
403  (assert (and in-p out-p) () "Non-bidirectional file-socket stream?")
404  'basic-file-socket-stream)
405
406(defclass unconnected-socket (socket)
407  ((device :initarg :device :accessor socket-device)
408   (keys :initarg :keys :reader socket-keys)))
409
410(defmethod socket-format ((socket unconnected-socket))
411  (or (getf (socket-keys socket) :format) :text))
412
413(defgeneric close (socket &key abort)
414  (:documentation
415   "The close generic function can be applied to sockets. It releases the
416operating system resources associated with the socket."))
417
418(defmethod close ((socket unconnected-socket) &key abort)
419  (declare (ignore abort))
420  (when (socket-device socket)
421    (fd-close (socket-device socket))
422    (setf (socket-device socket) nil)
423    t))
424
425;; A passive tcp socket just generates connection streams
426(defclass listener-socket (tcp-socket unconnected-socket) ())
427
428(defmethod SOCKET-CONNECT ((stream listener-socket)) :passive)
429
430(defclass file-listener-socket (stream-file-socket unconnected-socket) ())
431
432(defmethod SOCKET-CONNECT ((stream file-listener-socket)) :passive)
433
434;;; A FILE-LISTENER-SOCKET should try to delete the filesystem
435;;; entity when closing.
436
437#-windows-target
438(defmethod close :before ((s file-listener-socket) &key abort)
439  (declare (ignore abort))
440  (let* ((path (local-socket-filename (socket-device s) s)))
441    (when path (%delete-file path))))
442
443
444;; A udp socket just sends and receives packets.
445(defclass udp-socket (ip-socket unconnected-socket) ())
446
447(defmethod socket-type ((stream udp-socket)) :datagram)
448(defmethod socket-connect ((stream udp-socket)) nil)
449
450(defgeneric socket-os-fd (socket)
451  (:documentation
452   "Return the native OS's representation of the socket, or NIL if the
453socket is closed. On Unix, this is the Unix 'file descriptor', a small
454non-negative integer. Note that it is rather dangerous to mess around
455with tcp-stream fd's, as there is all sorts of buffering and asynchronous
456I/O going on above the OS level. listener-socket and udp-socket fd's are
457safer to mess with directly as there is less magic going on."))
458
459;; Returns nil for closed stream...
460(defmethod socket-os-fd ((socket socket))
461  (socket-device socket))
462
463;; Returns nil for closed stream
464(defun local-socket-info (fd type socket)
465  (and fd
466       (rlet ((sockaddr :sockaddr_in)
467              (namelen :signed))
468             (setf (pref namelen :signed) (record-length :sockaddr_in))
469             (socket-call socket "getsockname" (c_getsockname fd sockaddr namelen))
470             (when (= #$AF_INET (pref sockaddr :sockaddr_in.sin_family))
471               (ecase type
472                 (:host (ntohl (pref sockaddr
473                                     #-(or solaris-target windows-target) :sockaddr_in.sin_addr.s_addr
474                                     #+(or solaris-target windows-target) #>sockaddr_in.sin_addr.S_un.S_addr)))
475                 (:port (ntohs (pref sockaddr :sockaddr_in.sin_port))))))))
476
477#-windows-target
478(defun path-from-unix-address (addr)
479  (when (= #$AF_UNIX (pref addr :sockaddr_un.sun_family))
480    #+darwin-target
481    (%str-from-ptr (pref addr :sockaddr_un.sun_path)
482                   (- (pref addr :sockaddr_un.sun_len) 2))
483    #-darwin-target
484    (%get-cstring (pref addr :sockaddr_un.sun_path))))
485
486#-windows-target
487(defun local-socket-filename (fd socket)
488  (and fd
489       (rlet ((addr :sockaddr_un)
490              (namelen :signed))
491         (setf (pref namelen :signed) (record-length :sockaddr_un))
492         (socket-call socket "getsockname" (c_getsockname fd addr namelen))
493         (path-from-unix-address addr))))
494
495(defmacro with-if ((var expr) &body body)
496  `(let ((,var ,expr))
497     (if ,var
498         (progn
499           ,@body))))     
500
501(defun remote-socket-info (socket type)
502  (with-if (fd (socket-device socket))
503    (rlet ((sockaddr :sockaddr_in)
504           (namelen :signed))
505          (setf (pref namelen :signed) (record-length :sockaddr_in))
506          (let ((err (c_getpeername fd sockaddr namelen)))
507            (cond ((eql err (- #+windows-target #$WSAENOTCONN #-windows-target #$ENOTCONN)) nil)
508                  ((< err 0) (socket-error socket "getpeername" err))
509                  (t
510                   (when (= #$AF_INET (pref sockaddr :sockaddr_in.sin_family))
511                     (ecase type
512                       (:host (ntohl (pref sockaddr
513                                           #-(or solaris-target windows-target) :sockaddr_in.sin_addr.s_addr
514                                           #+(or solaris-target windows-target)  #>sockaddr_in.sin_addr.S_un.S_addr)))
515                       (:port (ntohs  (pref sockaddr :sockaddr_in.sin_port)))))))))))
516
517#-windows-target
518(defun remote-socket-filename (socket)
519  (with-if (fd (socket-device socket))
520    (rlet ((addr :sockaddr_un)
521           (namelen :signed))
522          (setf (pref namelen :signed) (record-length :sockaddr_un))
523          (let* ((err (c_getsockname fd addr namelen)))
524            (cond ((eql err (- #$ENOTCONN)) nil)
525                  ((< err 0) (socket-error socket "getpeername" err))
526                  (t (path-from-unix-address addr)))))))
527
528(defgeneric local-port (socket)
529  (:documentation "Return the local port number."))
530
531(defmethod local-port ((socket socket))
532  (local-socket-info (socket-device socket) :port socket))
533
534(defgeneric local-host (socket)
535  (:documentation
536   "Return 32-bit unsigned IP address of the local host."))
537
538(defmethod local-host ((socket socket))
539  (local-socket-info (socket-device socket) :host socket))
540
541#-windows-target
542(defmethod local-filename ((socket socket))
543  (local-socket-filename (socket-device socket) socket))
544
545(defgeneric remote-host (socket)
546  (:documentation
547   "Return the 32-bit unsigned IP address of the remote host, or NIL if
548the socket is not connected."))
549
550;; Returns NIL if socket is not connected
551(defmethod remote-host ((socket socket))
552  (remote-socket-info socket :host))
553
554(defgeneric remote-port (socket)
555  (:documentation
556   "Return the remote port number, or NIL if the socket is not connected."))
557
558(defmethod remote-port ((socket socket))
559  (remote-socket-info socket :port))
560
561#-windows-target
562(defmethod remote-filename ((socket socket))
563  (remote-socket-filename socket))
564 
565(defun set-socket-fd-blocking (fd block-flag)
566  #+windows-target
567  (rlet ((argp :u_long (if block-flag 0 1)))
568    (#_ioctlsocket fd #$FIONBIO argp))
569  #-windows-target
570  (if block-flag
571    (fd-clear-flag fd #$O_NONBLOCK)
572    (fd-set-flag fd #$O_NONBLOCK)))
573
574(defun get-socket-fd-blocking (fd)
575  "returns T iff socket is in blocking mode"
576  #+windows-target (declare (ignore fd))
577  #+windows-target t
578  #-windows-target
579  (not (logtest #$O_NONBLOCK (fd-get-flags fd))))
580
581(defun set-socket-options (fd-or-socket &key 
582                           keepalive
583                           reuse-address
584                           nodelay
585                           broadcast
586                           linger
587                           address-family
588                           local-port
589                           local-host
590                           local-filename
591                           type
592                           connect
593                           out-of-band-inline
594                           &allow-other-keys)
595  ;; see man socket(7) tcp(7) ip(7)
596  (multiple-value-bind (socket fd) (etypecase fd-or-socket
597                                     (socket (values fd-or-socket (socket-device fd-or-socket)))
598                                     (integer (values nil fd-or-socket)))
599   
600    (if (null address-family)
601        (setq address-family :internet))
602    (when keepalive
603      (int-setsockopt fd #$SOL_SOCKET #$SO_KEEPALIVE 1))
604    (when reuse-address
605      (int-setsockopt fd #$SOL_SOCKET #$SO_REUSEADDR 1))
606    (when broadcast
607      (int-setsockopt fd #$SOL_SOCKET #$SO_BROADCAST 1))
608    (when out-of-band-inline
609      (int-setsockopt fd #$SOL_SOCKET #$SO_OOBINLINE 1))
610    (rlet ((plinger :linger))
611          (setf (pref plinger :linger.l_onoff) (if linger 1 0)
612                (pref plinger :linger.l_linger) (or linger 0))
613          (socket-call socket "setsockopt"
614                       (c_setsockopt fd #$SOL_SOCKET #$SO_LINGER plinger (record-length :linger))))
615    (when (eq address-family :internet)
616      (when nodelay
617        (int-setsockopt fd
618                        #+linux-target #$SOL_TCP
619                        #-linux-target #$IPPROTO_TCP
620                        #$TCP_NODELAY 1))
621      (when (or local-port local-host)
622        (let* ((proto (if (eq type :stream) "tcp" "udp"))
623               (port-n (if local-port (port-as-inet-port local-port proto) 0))
624               (host-n (if local-host (host-as-inet-host local-host) #$INADDR_ANY)))
625          ;; Darwin includes the SIN_ZERO field of the sockaddr_in when
626          ;; comparing the requested address to the addresses of configured
627          ;; interfaces (as if the zeros were somehow part of either address.)
628          ;; "rletz" zeros out the stack-allocated structure, so those zeros
629          ;; will be 0.
630          (rletz ((sockaddr :sockaddr_in))
631                 (setf (pref sockaddr :sockaddr_in.sin_family) #$AF_INET
632                       (pref sockaddr :sockaddr_in.sin_port) port-n
633                       (pref sockaddr
634                             #-(or solaris-target windows-target) :sockaddr_in.sin_addr.s_addr
635                             #+(or solaris-target windows-target) #>sockaddr_in.sin_addr.S_un.S_addr
636                             ) host-n)
637                 (socket-call socket "bind" (c_bind fd sockaddr (record-length :sockaddr_in)))))))
638    (when (and (eq address-family :file)
639               (eq connect :passive)
640               local-filename)
641      #+windows-target (error "can't create file socket on Windows")
642      #-windows-target (bind-unix-socket fd local-filename))))
643
644;; I hope the inline declaration makes the &rest/apply's go away...
645(declaim (inline make-ip-socket))
646(defun make-ip-socket (&rest keys &key type &allow-other-keys)
647  (declare (dynamic-extent keys))
648  (ecase type
649    ((nil :stream) (apply #'make-tcp-socket keys))
650    ((:datagram) (apply #'make-udp-socket keys))))
651
652(declaim (inline make-file-socket))
653(defun make-file-socket (&rest keys &key type &allow-other-keys)
654  (declare (dynamic-extent keys))
655  (ecase type
656    ((nil :stream) (apply #'make-stream-file-socket keys))
657    (:datagram (apply #'make-datagram-file-socket keys))))
658
659(defun make-socket (&rest keys
660                    &key address-family
661                    ;; List all keys here just for error checking...
662                    ;; &allow-other-keys
663                    type connect remote-host remote-port eol format
664                    keepalive reuse-address nodelay broadcast linger
665                    local-port local-host backlog class out-of-band-inline
666                    local-filename remote-filename sharing basic
667                    external-format (auto-close t)
668                    connect-timeout input-timeout output-timeout deadline
669                    fd)
670  "Create and return a new socket."
671  (declare (dynamic-extent keys))
672  (declare (ignore type connect remote-host remote-port eol format
673                   keepalive reuse-address nodelay broadcast linger
674                   local-port local-host backlog class out-of-band-inline
675                   local-filename remote-filename sharing basic external-format
676                   auto-close connect-timeout input-timeout output-timeout deadline fd))
677  (ecase address-family
678    ((:file) (apply #'make-file-socket keys))
679    ((nil :internet) (apply #'make-ip-socket keys))))
680
681
682
683(defun make-udp-socket (&rest keys &key (fd -1) &allow-other-keys)
684  (unwind-protect
685    (let (socket)
686      (when (< fd 0)
687        (setq fd (socket-call nil "socket"
688                              (c_socket #$AF_INET #$SOCK_DGRAM #$IPPROTO_UDP))))
689      (apply #'set-socket-options fd keys)
690      (setq socket (make-instance 'udp-socket
691                                  :device fd
692                                  :keys keys))
693      (setq fd -1)
694      socket)
695    (unless (< fd 0)
696      (fd-close fd))))
697
698(defun make-tcp-socket (&rest keys &key connect (fd -1) &allow-other-keys)
699  (unwind-protect
700       (let (socket)
701         (when (< fd 0)
702           (setq fd (socket-call nil "socket"
703                                 (c_socket #$AF_INET #$SOCK_STREAM #$IPPROTO_TCP))))
704         (apply #'set-socket-options fd keys)
705         (setq socket
706               (ecase connect
707                 ((nil :active) (apply #'make-tcp-stream-socket fd keys))
708                 ((:passive) (apply #'make-tcp-listener-socket fd keys))))
709         (setq fd -1)
710         socket)
711    (unless (< fd 0)
712      (fd-close fd))))
713
714(defun make-stream-file-socket (&rest keys &key connect (fd -1) &allow-other-keys)
715  (unwind-protect
716       (let (socket)
717         (when (< fd 0)
718           (setq fd (socket-call nil "socket" (c_socket #$PF_UNIX #$SOCK_STREAM 0))))
719         (apply #'set-socket-options fd keys)
720         (setq socket
721               (ecase connect
722                 ((nil :active) (apply #'make-file-stream-socket fd keys))
723                 ((:passive) (apply #'make-file-listener-socket fd keys))))
724         (setq fd -1)
725         socket)
726    (unless (< fd 0)
727      (fd-close fd))))
728
729(defun make-datagram-file-socket (&rest keys)
730  (declare (ignore keys))
731  (error "Datagram file sockets aren't implemented."))
732
733
734(defun %socket-connect (fd addr addrlen &optional timeout-in-milliseconds)
735  (let* ((err (c_connect fd addr addrlen timeout-in-milliseconds)))
736    (declare (fixnum err))
737    (unless (eql err 0) (fd-close fd) (socket-error nil "connect" err))))
738   
739(defun inet-connect (fd host-n port-n &optional timeout-in-milliseconds)
740  (rlet ((sockaddr :sockaddr_in))
741    (setf (pref sockaddr :sockaddr_in.sin_family) #$AF_INET
742          (pref sockaddr :sockaddr_in.sin_port) port-n
743          (pref sockaddr
744                #-(or solaris-target windows-target) :sockaddr_in.sin_addr.s_addr
745                #+(or solaris-target windows-target)  #>sockaddr_in.sin_addr.S_un.S_addr
746                ) host-n)
747    (%socket-connect fd sockaddr (record-length :sockaddr_in) timeout-in-milliseconds)))
748
749#-windows-target
750(defun file-socket-connect (fd remote-filename)
751  (rletz ((sockaddr :sockaddr_un))
752    (init-unix-sockaddr sockaddr remote-filename)
753    (%socket-connect fd sockaddr (record-length :sockaddr_un))))
754
755#+windows-target
756(defun file-socket-connect (fd remote-filename)
757  (declare (ignore fd))
758  (error "Can't create file socket to ~s on Windows" remote-filename))
759 
760(defun make-tcp-stream-socket (fd &rest keys
761                                  &key remote-host
762                                  remote-port
763                                  connect-timeout
764                                  deadline
765                                  &allow-other-keys)
766  (let* ((timeout-in-milliseconds
767          (if deadline
768            (max (round (- deadline (get-internal-real-time))
769                        (/ internal-time-units-per-second 1000))
770                 0)
771            (if connect-timeout
772              (round (* connect-timeout 1000))))))
773    (inet-connect fd
774                  (host-as-inet-host remote-host)
775                  (port-as-inet-port remote-port "tcp")
776                  timeout-in-milliseconds)
777    (apply #'make-tcp-stream fd keys)))
778
779(defun make-file-stream-socket (fd &rest keys
780                                   &key remote-filename
781                                   &allow-other-keys)
782  (file-socket-connect fd remote-filename)
783  (apply #'make-file-socket-stream fd keys))
784
785
786(defun make-tcp-stream (fd
787                        &key (format :bivalent)
788                             external-format
789                             (class 'tcp-stream)
790                             sharing
791                             (basic t)
792                             (auto-close t)
793                             input-timeout
794                             output-timeout
795                             deadline
796                        &allow-other-keys)
797  (let* ((external-format (normalize-external-format :socket external-format)))
798    (let ((element-type (ecase format
799                          ((nil :text) 'character)
800                          ((:binary :bivalent) '(unsigned-byte 8)))))
801      ;; TODO: check out fd-stream-advance, -listen, -eofp, -force-output, -close
802      ;; See if should specialize any of 'em.
803      (make-fd-stream fd
804                      :class class
805                      :direction :io
806                      :element-type element-type
807                      :sharing sharing
808                      :character-p (not (eq format :binary))
809                      :encoding (external-format-character-encoding external-format)
810                      :line-termination (external-format-line-termination external-format)
811                      :basic basic
812                      :auto-close auto-close
813                      :input-timeout input-timeout
814                      :output-timeout output-timeout
815                      :deadline deadline))))
816
817(defun make-file-socket-stream (fd
818                                &key (format :bivalent)
819                                external-format
820                                (class 'file-socket-stream)
821                                sharing
822                                basic
823                                (auto-close t)
824                                input-timeout
825                                output-timeout
826                                deadline
827                                &allow-other-keys)
828  (let* ((external-format (normalize-external-format :socket external-format)))
829 
830    (let ((element-type (ecase format
831                          ((nil :text) 'character)
832                          ((:binary :bivalent) '(unsigned-byte 8)))))
833      ;; TODO: check out fd-stream-advance, -listen, -eofp, -force-output, -close
834      ;; See if should specialize any of 'em.
835      (make-fd-stream fd
836                      :class class
837                      :direction :io
838                      :element-type element-type
839                      :encoding (external-format-character-encoding external-format)
840                      :line-termination (external-format-line-termination external-format)
841                      :sharing sharing
842                      :character-p (not (eq format :binary))
843                      :basic basic
844                      :auto-close auto-close
845                      :input-timeout input-timeout
846                      :output-timeout output-timeout
847                      :deadline deadline))))
848
849(defun make-tcp-listener-socket (fd &rest keys &key backlog &allow-other-keys)
850  (socket-call nil "listen" (c_listen fd (or backlog 5)))
851  (make-instance 'listener-socket
852                 :device fd
853                 :keys keys))
854
855(defun make-file-listener-socket (fd &rest keys &key backlog &allow-other-keys)
856  (socket-call nil "listen" (c_listen fd (or backlog 5)))
857  (make-instance 'file-listener-socket
858                 :device fd
859                 :keys keys))
860
861(defun socket-accept (fd wait)
862  (flet ((_accept (fd async)
863           (let ((res (c_accept fd (%null-ptr) (%null-ptr))))
864             (declare (fixnum res))
865             ;; See the inscrutable note under ERROR HANDLING in
866             ;; man accept(2). This is my best guess at what they mean...
867             (if (and async (< res 0)
868                      #+windows-target
869                      (= res #$WSAEWOULDBLOCK)
870                      #-windows-target
871                      (or (eql res (- #$ENETDOWN))
872                          (eql res (- #+linux-target #$EPROTO
873                                      #-linux-target  #$EPROTOTYPE))
874                          (eql res (- #$ENOPROTOOPT))
875                          (eql res (- #$EHOSTDOWN))
876                          (eql res (- #+linux-target #$ENONET
877                                      #-linux-target #$ENETDOWN))
878                          (eql res (- #$EHOSTUNREACH))
879                          (eql res (- #$EOPNOTSUPP))
880                          (eql res (- #$ENETUNREACH))))
881               (- #$EAGAIN)
882               res))))
883    (cond (wait
884            (with-eagain fd :input
885              (_accept fd *multiprocessing-socket-io*)))
886          (*multiprocessing-socket-io*
887            (_accept fd t))
888          (t
889            (let ((was-blocking (get-socket-fd-blocking fd)))
890              (unwind-protect
891                  (progn
892                    (set-socket-fd-blocking fd nil)
893                    (_accept fd t))
894                (set-socket-fd-blocking fd was-blocking)))))))
895
896(defun accept-socket-connection (socket wait stream-create-function)
897  (let ((listen-fd (socket-device socket))
898        (fd -1))
899    (unwind-protect
900      (progn
901        (setq fd (socket-accept listen-fd wait))
902        (cond ((>= fd 0)
903               (prog1 (apply stream-create-function fd (socket-keys socket))
904                 (setq fd -1)))
905              ((eql fd (- #$EAGAIN)) nil)
906              (t (socket-error socket "accept" fd))))
907      (when (>= fd 0)
908        (fd-close fd)))))
909
910(defgeneric accept-connection (socket &key wait)
911  (:documentation
912  "Extract the first connection on the queue of pending connections,
913accept it (i.e. complete the connection startup protocol) and return a new
914tcp-stream or file-socket-stream representing the newly established
915connection.  The tcp stream inherits any properties of the listener socket
916that are relevant (e.g. :keepalive, :nodelay, etc.) The original listener
917socket continues to be open listening for more connections, so you can call
918accept-connection on it again."))
919
920(defmethod accept-connection ((socket listener-socket) &key (wait t))
921  (accept-socket-connection socket wait #'make-tcp-stream))
922
923(defmethod accept-connection ((socket file-listener-socket) &key (wait t))
924  (accept-socket-connection socket wait #'make-file-socket-stream))
925
926(defun verify-socket-buffer (buf offset size)
927  (unless offset (setq offset 0))
928  (unless (<= (+ offset size) (length buf))
929    (report-bad-arg size `(integer 0 ,(- (length buf) offset))))
930  (multiple-value-bind (arr start) (array-data-and-offset buf)
931    (setq buf arr offset (+ offset start)))
932  ;; TODO: maybe should allow any raw vector
933  (let ((subtype (typecode buf)))
934    (unless #+ppc32-target (and (<= ppc32::min-8-bit-ivector-subtag subtype)
935                                (<= subtype ppc32::max-8-bit-ivector-subtag))
936            #+ppc64-target (= (the fixnum (logand subtype ppc64::fulltagmask))
937                              ppc64::ivector-class-8-bit)
938            #+x8632-target (and (<= x8632::min-8-bit-ivector-subtag subtype)
939                                (<= subtype x8632::max-8-bit-ivector-subtag))
940            #+x8664-target (and (>= subtype x8664::min-8-bit-ivector-subtag)
941                                (<= subtype x8664::max-8-bit-ivector-subtag))
942      (report-bad-arg buf `(or (array character)
943                               (array (unsigned-byte 8))
944                               (array (signed-byte 8))))))
945  (values buf offset))
946
947(defmethod send-to ((socket udp-socket) msg size
948                    &key remote-host remote-port offset)
949  "Send a UDP packet over a socket."
950  (let ((fd (socket-device socket)))
951    (multiple-value-setq (msg offset) (verify-socket-buffer msg offset size))
952    (unless remote-host
953      (setq remote-host (or (getf (socket-keys socket) :remote-host)
954                            (remote-socket-info socket :host))))
955    (unless remote-port
956      (setq remote-port (or (getf (socket-keys socket) :remote-port)
957                            (remote-socket-info socket :port))))
958    (rlet ((sockaddr :sockaddr_in))
959      (setf (pref sockaddr :sockaddr_in.sin_family) #$AF_INET)
960      (setf (pref sockaddr
961                  #-(or solaris-target windows-target) :sockaddr_in.sin_addr.s_addr
962                  #+(or solaris-target windows-target)  #>sockaddr_in.sin_addr.S_un.S_addr)
963            (if remote-host (host-as-inet-host remote-host) #$INADDR_ANY))
964      (setf (pref sockaddr :sockaddr_in.sin_port)
965            (if remote-port (port-as-inet-port remote-port "udp") 0))
966      (%stack-block ((bufptr size))
967        (%copy-ivector-to-ptr msg offset bufptr 0 size)
968        (socket-call socket "sendto"
969          (with-eagain fd :output
970            (c_sendto fd bufptr size 0 sockaddr (record-length :sockaddr_in))))))))
971
972(defmethod receive-from ((socket udp-socket) size &key buffer extract offset)
973  "Read a UDP packet from a socket. If no packets are available, wait for
974a packet to arrive. Returns four values:
975  The buffer with the data
976  The number of bytes read
977  The 32-bit unsigned IP address of the sender of the data
978  The port number of the sender of the data."
979  (let ((fd (socket-device socket))
980        (vec-offset offset)
981        (vec buffer)
982        (ret-size -1))
983    (when vec
984      (multiple-value-setq (vec vec-offset)
985        (verify-socket-buffer vec vec-offset size)))
986    (rlet ((sockaddr :sockaddr_in)
987           (namelen :signed))
988      (setf (pref sockaddr :sockaddr_in.sin_family) #$AF_INET)
989      (setf (pref sockaddr
990                  #-(or solaris-target windows-target) :sockaddr_in.sin_addr.s_addr
991                  #+(or solaris-target windows-target) #>sockaddr_in.sin_addr.S_un.S_addr)
992            #$INADDR_ANY)
993      (setf (pref sockaddr :sockaddr_in.sin_port) 0)
994      (setf (pref namelen :signed) (record-length :sockaddr_in))
995      (%stack-block ((bufptr size))
996        (setq ret-size (socket-call socket "recvfrom"
997                         (with-eagain fd :input
998                           (c_recvfrom fd bufptr size 0 sockaddr namelen))))
999        (unless vec
1000          (setq vec (make-array ret-size
1001                                :element-type
1002                                (ecase (socket-format socket)
1003                                  ((:text) 'base-char)
1004                                  ((:binary :bivalent) '(unsigned-byte 8))))
1005                vec-offset 0))
1006        (%copy-ptr-to-ivector bufptr 0 vec vec-offset ret-size))
1007      (values (cond ((null buffer)
1008                     vec)
1009                    ((or (not extract)
1010                         (and (eql 0 (or offset 0))
1011                              (eql ret-size (length buffer))))
1012                     buffer)
1013                    (t 
1014                     (subseq vec vec-offset (+ vec-offset ret-size))))
1015              ret-size
1016              (ntohl (pref sockaddr
1017                           #-(or solaris-target windows-target) :sockaddr_in.sin_addr.s_addr
1018                           #+(or solaris-target windows-target) #>sockaddr_in.sin_addr.S_un.S_addr))
1019              (ntohs (pref sockaddr :sockaddr_in.sin_port))))))
1020
1021(defgeneric shutdown (socket &key direction)
1022  (:documentation
1023   "Shut down part of a bidirectional connection. This is useful if e.g.
1024you need to read responses after sending an end-of-file signal."))
1025
1026(defmethod shutdown (socket &key direction)
1027  ;; TODO: should we ignore ENOTCONN error?  (at least make sure it
1028  ;; is a distinct, catchable error type).
1029  (let ((fd (socket-device socket)))
1030    (socket-call socket "shutdown"
1031      (c_shutdown fd (ecase direction
1032                       (:input 0)
1033                       (:output 1))))))
1034
1035;; Accepts port as specified by user, returns port number in network byte
1036;; order.  Protocol should be one of "tcp" or "udp".  Error if not known.
1037(defun port-as-inet-port (port proto)
1038  (or (etypecase port
1039        (fixnum (htons port))
1040        (string (_getservbyname port proto))
1041        (symbol (_getservbyname (string-downcase (symbol-name port)) proto)))
1042      (socket-error nil "getservbyname" (- #$ENOENT))))
1043
1044(defun lookup-port (port proto)
1045  "Find the port number for the specified port and protocol."
1046  (if (fixnump port)
1047    port
1048    (ntohs (port-as-inet-port port proto))))
1049
1050;; Accepts host as specified by user, returns host number in network byte
1051;; order.
1052(defun host-as-inet-host (host)
1053  (etypecase host
1054    (integer (htonl host))
1055    (string (or (and (every #'(lambda (c) (position c ".0123456789")) host)
1056                     (_inet_aton host))
1057                (multiple-value-bind (addr err) (c_gethostbyname host)
1058                  (or addr
1059                      (socket-error nil "gethostbyname" err t)))))))
1060
1061
1062(defun dotted-to-ipaddr (name &key (errorp t))
1063  "Convert a dotted-string representation of a host address to a 32-bit
1064unsigned IP address."
1065  (let ((addr (_inet_aton name)))
1066    (if addr (ntohl addr)
1067      (and errorp (error "Invalid dotted address ~s" name)))))
1068   
1069(defun lookup-hostname (host)
1070  "Convert a host spec in any of the acceptable formats into a 32-bit
1071unsigned IP address."
1072  (if (typep host 'integer)
1073    host
1074    (ntohl (host-as-inet-host host))))
1075
1076(defun ipaddr-to-dotted (addr &key values)
1077  "Convert a 32-bit unsigned IP address into octets."
1078  (let* ((a (ldb (byte 8 24) addr))
1079         (b (ldb (byte 8 16) addr))
1080         (c (ldb (byte 8  8) addr))
1081         (d (ldb (byte 8  0) addr)))
1082    (if values
1083      (values a b c d)
1084      (format nil "~d.~d.~d.~d" a b c d))))
1085
1086(defun ipaddr-to-hostname (ipaddr &key ignore-cache)
1087  "Convert a 32-bit unsigned IP address into a host name string."
1088  (declare (ignore ignore-cache))
1089  (multiple-value-bind (name err) (c_gethostbyaddr (htonl ipaddr))
1090    (or name (socket-error nil "gethostbyaddr" err t))))
1091 
1092
1093(defun int-getsockopt (socket level optname)
1094  (rlet ((valptr :signed)
1095         (vallen :signed))
1096    (setf (pref vallen :signed) 4)
1097    (let* ((err (c_getsockopt socket level optname valptr vallen)))
1098      (if (and (eql 0 err)
1099               (eql 4 (pref vallen :signed)))
1100        (pref valptr :signed)
1101        (socket-error socket "getsockopt" err)))))
1102
1103(defun timeval-setsockopt (socket level optname timeout)
1104    (multiple-value-bind (seconds micros)
1105        (microseconds timeout)
1106      (rlet ((valptr :timeval :tv_sec seconds :tv_usec micros))
1107        (socket-call socket "setsockopt"
1108          (c_setsockopt socket level optname valptr (record-length :timeval))))))
1109                   
1110(defun int-setsockopt (socket level optname optval)
1111  (rlet ((valptr :signed))
1112    (setf (pref valptr :signed) optval)
1113    (socket-call socket "setsockopt"
1114      (c_setsockopt socket level optname valptr (record-length :signed)))))
1115
1116
1117
1118           
1119(defun c_gethostbyaddr (addr-in-net-byte-order)
1120  (rletZ ((sin #>sockaddr_in))
1121    (setf (pref sin :sockaddr_in.sin_family) #$AF_INET
1122          (pref sin
1123                #+(or windows-target solaris-target) #>sockaddr_in.sin_addr.S_un.S_addr
1124                #-(or windows-target solaris-target) #>sockaddr_in.sin_addr.s_addr) addr-in-net-byte-order)
1125    #+darwin-target (setf (pref sin :sockaddr_in.sin_len) (record-length :sockaddr_in))
1126    (%stack-block ((namep #$NI_MAXHOST))
1127      (let* ((err (#_getnameinfo sin (record-length #>sockaddr_in) namep #$NI_MAXHOST (%null-ptr) 0 #$NI_NAMEREQD)))
1128        (if (eql 0 err)
1129          (%get-cstring namep)
1130          (values nil err))))))
1131               
1132(defun c_gethostbyname (name)
1133  (with-cstrs ((name (string name)))
1134    (rletZ ((hints #>addrinfo)
1135            (results :address))
1136      (setf (pref hints #>addrinfo.ai_family) #$AF_INET)
1137      (let* ((err (#_getaddrinfo name (%null-ptr) hints results)))
1138        (if (eql 0 err)
1139          (let* ((info (pref results :address))
1140                 (sin (pref info #>addrinfo.ai_addr)))
1141            (prog1
1142                #+(or windows-target solaris-target)
1143                (pref sin #>sockaddr_in.sin_addr.S_un.S_addr)
1144                #-(or windows-target solaris-target)
1145                (pref sin #>sockaddr_in.sin_addr.s_addr)
1146                (#_freeaddrinfo info)))
1147          (values nil err))))))
1148     
1149 
1150
1151 
1152
1153(defun _getservbyname (name proto)
1154  (with-cstrs ((name (string name))
1155               (proto (string proto)))
1156    (let* ((servent-ptr (%null-ptr)))
1157      (declare (dynamic-extent servent-ptr))
1158      (%setf-macptr servent-ptr (#_getservbyname name proto))
1159      (unless (%null-ptr-p servent-ptr)
1160        (pref servent-ptr :servent.s_port)))))
1161
1162(defun _inet_aton (string)
1163  (with-cstrs ((name string))
1164    #-windows-target
1165    (rlet ((addr :in_addr))
1166      (let* ((result #+freebsd-target (#___inet_aton name addr)
1167                     #-freebsd-target (#_inet_aton name addr)))
1168        (unless (eql result 0)
1169          (pref addr
1170                #-solaris-target :in_addr.s_addr
1171                #+solaris-target #>in_addr.S_un.S_addr
1172                ))))
1173    #+windows-target
1174    (rlet ((addr :sockaddr_in)
1175           (addrlenp :int (record-length :sockaddr_in)))
1176      (setf (pref addr :sockaddr_in.sin_family) #$AF_INET)
1177      (when (zerop (#_WSAStringToAddressA name #$AF_INET (%null-ptr)  addr addrlenp))
1178        (pref addr #>sockaddr_in.sin_addr.S_un.S_addr)))))
1179
1180(defun c_socket_1 (domain type protocol)
1181  #-windows-target (int-errno-call (#_socket domain type protocol))
1182  #+windows-target (let* ((handle (#_socket domain type protocol)))
1183                     (if (< handle 0)
1184                       (%get-winsock-error)
1185                       handle)))
1186
1187
1188
1189(defun c_socket (domain type protocol)
1190  (let* ((fd (c_socket_1 domain type protocol)))
1191    (when (or (eql fd (- #$EMFILE))
1192              (eql fd (- #$ENFILE)))
1193      (gc)
1194      (drain-termination-queue)
1195      (setq fd (c_socket_1 domain type protocol)))
1196    fd))
1197     
1198
1199#-windows-target
1200(defun init-unix-sockaddr (addr path)
1201  (macrolet ((sockaddr_un-path-len ()
1202               (/ (ensure-foreign-type-bits
1203                   (foreign-record-field-type 
1204                    (%find-foreign-record-type-field
1205                     (parse-foreign-type '(:struct :sockaddr_un)) :sun_path)))
1206                  8)))
1207    (let* ((name (native-translated-namestring path))
1208           (namelen (length name))
1209           (pathlen (sockaddr_un-path-len))
1210           (copylen (min (1- pathlen) namelen)))
1211      (setf (pref addr :sockaddr_un.sun_family) #$AF_UNIX)
1212      (let* ((sun-path (pref addr :sockaddr_un.sun_path)))
1213        (dotimes (i copylen)
1214          (setf (%get-unsigned-byte sun-path i)
1215                (let* ((code (char-code (schar name i))))
1216                  (if (> code 255)
1217                    (char-code #\Sub)
1218                    code))))))))
1219
1220#-windows-target
1221(defun bind-unix-socket (socketfd path)
1222  (rletz ((addr :sockaddr_un))
1223    (init-unix-sockaddr addr path)
1224    (socket-call
1225     nil
1226     "bind"
1227     (c_bind socketfd
1228             addr
1229             (+ 2
1230                (#_strlen
1231                 (pref addr :sockaddr_un.sun_path)))))))
1232     
1233
1234(defun c_bind (sockfd sockaddr addrlen)
1235  (check-socket-error (#_bind sockfd sockaddr addrlen)))
1236
1237
1238#+windows-target
1239(defun windows-connect-wait (sockfd timeout-in-milliseconds)
1240  (if (and timeout-in-milliseconds
1241           (< timeout-in-milliseconds 0))
1242    (setq timeout-in-milliseconds nil))
1243  (rlet ((writefds :fd_set)
1244         (exceptfds :fd_set)
1245         (tv :timeval :tv_sec 0 :tv_usec 0))
1246    (fd-zero writefds)
1247    (fd-zero exceptfds)
1248    (fd-set sockfd writefds)
1249    (fd-set sockfd exceptfds)
1250    (when timeout-in-milliseconds
1251      (multiple-value-bind (seconds milliseconds)
1252          (floor timeout-in-milliseconds 1000)
1253        (setf (pref tv :timeval.tv_sec) seconds
1254              (pref tv :timeval.tv_usec) (* 1000 milliseconds))))
1255    (> (#_select 1 (%null-ptr) writefds exceptfds (if timeout-in-milliseconds tv (%null-ptr))) 0)))
1256     
1257     
1258;;; If attempts to connnect are interrupted, we basically have to
1259;;; wait in #_select (or the equivalent).  There's a good rant
1260;;; about these issues in:
1261;;; <http://www.madore.org/~david/computers/connect-intr.html>
1262(defun c_connect (sockfd addr len &optional timeout-in-milliseconds)
1263  (let* ((was-blocking (get-socket-fd-blocking sockfd)))
1264    (unwind-protect
1265         (progn
1266           (set-socket-fd-blocking sockfd nil)
1267           (let* ((err (check-socket-error (#_connect sockfd addr len))))
1268             (cond ((or (eql err (- #+windows-target #$WSAEINPROGRESS
1269                                   
1270                                    #-windows-target #$EINPROGRESS))
1271                        #+windows-target (eql err (- #$WSAEWOULDBLOCK))
1272                        (eql err (- #$EINTR)))
1273                    (if
1274                      #+windows-target (windows-connect-wait sockfd timeout-in-milliseconds)
1275                      #-windows-target (process-output-wait sockfd timeout-in-milliseconds)
1276                      (- (int-getsockopt sockfd #$SOL_SOCKET #$SO_ERROR))
1277                      (- #+windows-target #$WSAETIMEDOUT #-windows-target #$ETIMEDOUT)))
1278                   (t err))))
1279      (set-socket-fd-blocking sockfd was-blocking))))
1280
1281(defun c_listen (sockfd backlog)
1282  (check-socket-error (#_listen sockfd backlog)))
1283
1284(defun c_accept (sockfd addrp addrlenp)
1285  (ignoring-eintr
1286   (check-socket-error (#_accept sockfd addrp addrlenp))))
1287
1288(defun c_getsockname (sockfd addrp addrlenp)
1289  (check-socket-error (#_getsockname sockfd addrp addrlenp)))
1290
1291(defun c_getpeername (sockfd addrp addrlenp)
1292  (check-socket-error (#_getpeername sockfd addrp addrlenp)))
1293
1294#-windows-target
1295(defun c_socketpair (domain type protocol socketsptr)
1296  (check-socket-error (#_socketpair domain type protocol socketsptr)))
1297
1298
1299(defun c_sendto (sockfd msgptr len flags addrp addrlen)
1300  (ignoring-eintr (check-socket-error (#_sendto sockfd msgptr len flags addrp addrlen))))
1301
1302(defun c_recvfrom (sockfd bufptr len flags addrp addrlenp)
1303  (ignoring-eintr (check-socket-error (#_recvfrom sockfd bufptr len flags addrp addrlenp))))
1304
1305(defun c_shutdown (sockfd how)
1306  (check-socket-error (#_shutdown sockfd how)))
1307
1308(defun c_setsockopt (sockfd level optname optvalp optlen)
1309  (check-socket-error (#_setsockopt sockfd level optname optvalp optlen)))
1310
1311(defun c_getsockopt (sockfd level optname optvalp optlenp)
1312  (check-socket-error (#_getsockopt sockfd level optname optvalp optlenp)))
1313
1314#-windows-target
1315(defun c_sendmsg (sockfd msghdrp flags)
1316  (check-socket-error (#_sendmsg sockfd msghdrp flags)))
1317
1318#-windows-target
1319(defun c_recvmsg (sockfd msghdrp flags)
1320  (check-socket-error   (#_recvmsg sockfd msghdrp flags)))
1321
1322;;; Return a list of currently configured interfaces, a la ifconfig.
1323(defstruct ip-interface
1324  name
1325  addr
1326  netmask
1327  flags
1328  address-family)
1329
1330(defun dump-buffer (p n)
1331  (dotimes (i n (progn (terpri) (terpri)))
1332    (unless (logtest i 15)
1333      (format t "~&~8,'0x: " (%ptr-to-int (%inc-ptr p i))))
1334    (format t " ~2,'0x" (%get-byte p i))))
1335
1336#-(or windows-target solaris-target)
1337(defun %get-ip-interfaces ()
1338  (rlet ((p :address (%null-ptr)))
1339    (if (zerop (#_getifaddrs p))
1340      (unwind-protect
1341           (do* ((q (%get-ptr p) (pref q :ifaddrs.ifa_next))
1342                 (res ()))
1343                ((%null-ptr-p q) (nreverse res))
1344             (let* ((addr (pref q :ifaddrs.ifa_addr)))
1345               (when (and (not (%null-ptr-p addr))
1346                          (eql (pref addr :sockaddr.sa_family) #$AF_INET))
1347                 (push (make-ip-interface
1348                        :name (%get-cstring (pref q :ifaddrs.ifa_name))
1349                        :addr (ntohl (pref addr :sockaddr_in.sin_addr.s_addr))
1350                        :netmask (ntohl
1351                                  (pref (pref q :ifaddrs.ifa_netmask)
1352                                       :sockaddr_in.sin_addr.s_addr))
1353                        :flags (pref q :ifaddrs.ifa_flags)
1354                        :address-family #$AF_INET)
1355                       res))))
1356        (#_freeifaddrs (pref p :address))))))
1357
1358#+solaris-target
1359(progn
1360  ;;; Interface translator has trouble with a lot of ioctl constants.
1361  (eval-when (:compile-toplevel :execute)
1362    (defconstant os::|SIOCGLIFNUM| #xc00c6982)
1363    (defconstant os::|SIOCGLIFCONF| #xc01069a5)
1364    (defconstant os::|SIOCGLIFADDR| #xc0786971)
1365    (defconstant os::|SIOCGLIFFLAGS| #xc0786975)
1366    (defconstant os::|SIOCGLIFNETMASK| #xc078697d)
1367    )
1368
1369(defun %get-ip-interfaces ()
1370  (let* ((sock (c_socket #$AF_INET #$SOCK_DGRAM #$IPPROTO_UDP))
1371         (res nil))
1372    (when (>= sock 0)
1373      (unwind-protect
1374           (let* ((flags (logior #$LIFC_NOXMIT #$LIFC_TEMPORARY #$LIFC_ALLZONES))
1375                  (ninterfaces (rlet ((lifnum :lifnum
1376                                        :lifn_flags flags
1377                                        :lifn_family #$AF_INET
1378                                        :lifn_count 0))
1379                                 (#_ioctl sock os::SIOCGLIFNUM :address lifnum)
1380                                 (pref lifnum :lifnum.lifn_count))))
1381             (declare (fixnum ninterfaces))
1382             (when (> ninterfaces 0)
1383               (let* ((bufsize (* ninterfaces (record-length :lifreq))))
1384                 (%stack-block ((buf bufsize :clear t))
1385                   (rlet ((lifc :lifconf
1386                            :lifc_family #$AF_INET
1387                            :lifc_flags flags
1388                            :lifc_len bufsize
1389                            :lifc_lifcu.lifcu_buf buf))
1390                     (when (>= (#_ioctl sock os::SIOCGLIFCONF :address lifc) 0)
1391                       (do* ((i 0 (1+ i))
1392                             (p (pref lifc :lifconf.lifc_lifcu.lifcu_buf)
1393                                (%inc-ptr p (record-length :lifreq))))
1394                            ((= i ninterfaces))
1395                         (let* ((name (%get-cstring (pref p :lifreq.lifr_name)))
1396                                (address-family (pref p :lifreq.lifr_lifru.lifru_addr.ss_family))
1397                                (if-flags nil)
1398                                (address nil)
1399                                (netmask nil))
1400                           (if (>= (#_ioctl sock os::SIOCGLIFFLAGS :address p)
1401                                   0)
1402                             (setq if-flags (pref p :lifreq.lifr_lifru.lifru_flags)))
1403                           (if (>= (#_ioctl sock os::SIOCGLIFADDR :address p)
1404                                   0)
1405                             (setq address (pref
1406                                            (pref p :lifreq.lifr_lifru.lifru_addr)
1407                                            #>sockaddr_in.sin_addr.S_un.S_addr)))
1408                           (if (>= (#_ioctl sock os::SIOCGLIFNETMASK :address p)
1409                                   0)
1410                             (setq netmask (pref
1411                                            (pref p :lifreq.lifr_lifru.lifru_subnet)
1412                                            #>sockaddr_in.sin_addr.S_un.S_addr)))
1413                             
1414                           (push (make-ip-interface
1415                                  :name name
1416                                  :addr (ntohl address)
1417                                  :netmask (ntohl netmask)
1418                                  :flags if-flags
1419                                  :address-family address-family)
1420                                 res)))))))))
1421        (fd-close sock)))
1422    res))
1423)
1424
1425
1426
1427
1428#+windows-target
1429(defun %get-ip-interfaces ()
1430  (let* ((handle (#_socket #$AF_INET #$SOCK_DGRAM #$IPPROTO_IP)))
1431    (unwind-protect
1432    (rlet ((realoutlen #>DWORD 0))
1433      (do* ((reservedlen (* 4 (record-length #>INTERFACE_INFO))
1434                         (* 2 reservedlen)))
1435           ()
1436        (%stack-block ((buf reservedlen))
1437          (unless (eql 0 (#_WSAIoctl
1438                          handle
1439                          #$SIO_GET_INTERFACE_LIST
1440                          (%null-ptr)
1441                          0
1442                          buf
1443                          reservedlen
1444                          realoutlen
1445                          (%null-ptr)
1446                          (%null-ptr)))
1447            (return))
1448          (let* ((noutbytes (pref realoutlen #>DWORD)))
1449            (when (< noutbytes reservedlen)
1450              (let* ((interfaces nil))
1451                (do* ((offset 0 (+ offset (record-length #>INTERFACE_INFO)))
1452                      (nameidx 0 (1+ nameidx)))
1453                     ((>= offset noutbytes))
1454                  (with-macptrs ((p (%inc-ptr buf offset)))
1455                    (push (make-ip-interface 
1456                           :name (format nil "ip~d" nameidx)
1457                           :addr (ntohl
1458                                  (pref (pref p #>INTERFACE_INFO.iiAddress)
1459                                        #>sockaddr_gen.AddressIn.sin_addr.S_un.S_addr))
1460                           :netmask (ntohl
1461                                     (pref (pref p #>INTERFACE_INFO.iiNetmask)
1462                                        #>sockaddr_gen.AddressIn.sin_addr.S_un.S_addr))
1463                           :flags (pref p #>INTERFACE_INFO.iiFlags)
1464                           :address-family #$AF_INET)
1465                          interfaces)))
1466                (return interfaces)))))))
1467      (#_CloseHandle (%int-to-ptr handle)))))
1468
1469     
1470
1471
1472(defloadvar *ip-interfaces* ())
1473
1474(defun ip-interfaces ()
1475  (or *ip-interfaces*
1476      (setq *ip-interfaces* (%get-ip-interfaces))))
1477
1478;;; This should presumably happen after a configuration change.
1479;;; How do we detect a configuration change ?
1480(defun %reset-ip-interfaces ()
1481  (setq *ip-interfaces* ()))
1482
1483;;; Return the first non-loopback interface that's up and whose address
1484;;; family is #$AF_INET.  If no such interface exists, return
1485;;; the loopback interface.
1486(defun primary-ip-interface ()
1487  (let* ((ifaces (ip-interfaces)))
1488    (or (find-if #'(lambda (i)
1489                     (and (eq #$AF_INET (ip-interface-address-family i))
1490                          (ip-interface-addr i)
1491                          (let* ((flags (ip-interface-flags i)))
1492                            (and (not (logtest #$IFF_LOOPBACK flags))
1493                                 (logtest #$IFF_UP flags)))))
1494                 ifaces)
1495        (car ifaces))))
1496
1497(defun primary-ip-interface-address ()
1498  (let* ((iface (primary-ip-interface)))
1499    (if iface
1500      (ip-interface-addr iface)
1501      (error "Can't determine primary IP interface"))))
1502         
1503         
1504(defmethod stream-io-error ((stream socket) errno where)
1505  (socket-error stream where errno))
Note: See TracBrowser for help on using the repository browser.