74b4f36a13c042bc6240d9504465f7edb7f41883
[kmrcl.git] / sockets.lisp
1 ;;;; -*- Mode: Lisp; Syntax: ANSI-Common-Lisp; Base: 10; Package: modlisp -*-
2 ;;;; *************************************************************************
3 ;;;; FILE IDENTIFICATION
4 ;;;;
5 ;;;; Name:          sockets.lisp
6 ;;;; Purpose:       Socket functions
7 ;;;; Programmer:    Kevin M. Rosenberg with excerpts from portableaserve
8 ;;;; Date Started:  Jun 2003
9 ;;;;
10 ;;;; $Id$
11 ;;;; *************************************************************************
12
13 (in-package #:kmrcl)
14
15 (eval-when (:compile-toplevel :load-toplevel :execute)
16   #+sbcl (require :sb-bsd-sockets)
17   #+lispworks (require "comm")
18   #+allegro (require :socket))
19
20
21 #+sbcl
22 (defun listen-to-inet-port (&key (port 0) (kind :stream) (reuse nil))
23   "Create, bind and listen to an inet socket on *:PORT.
24 setsockopt SO_REUSEADDR if :reuse is not nil"
25   (let ((socket (make-instance 'sb-bsd-sockets:inet-socket
26                                :type :stream
27                                :protocol :tcp)))
28     (if reuse
29         (setf (sb-bsd-sockets:sockopt-reuse-address socket) t))
30     (sb-bsd-sockets:socket-bind 
31      socket (sb-bsd-sockets:make-inet-address "0.0.0.0") port)
32     (sb-bsd-sockets:socket-listen socket 15)
33     socket))
34
35 (defun create-inet-listener (port &key (format :text) (reuse-address t))
36   #+cmu (ext:create-inet-listener port)
37   #+allegro
38   (socket:make-socket :connect :passive :local-port port :format format
39                       :address-family 
40                       (if (stringp port)
41                           :file
42                         (if (or (null port) (integerp port))
43                             :internet
44                           (error "illegal value for port: ~s" port)))
45                       :reuse-address reuse-address)
46   #+sbcl
47   (listen-to-inet-port :port port :reuse reuse-address)
48   #+clisp (ext:socket-server port)
49   #+openmcl 
50   (ccl:make-socket :connect :passive :local-port port
51                    :reuse-address reuse-address)
52   #-(or allegro clisp cmu sbcl openmcl)
53   (warn "create-inet-listener not supported on this implementation")
54   )
55
56 (defun make-fd-stream (socket &key input output element-type)
57   #+cmu
58   (sys:make-fd-stream socket :input input :output output
59                       :element-type element-type)
60   #+sbcl
61   (sb-bsd-sockets:socket-make-stream socket :input input :output output
62                                      :element-type element-type)
63   #-(or cmu sbcl) (declare (ignore input output element-type))
64   #-(or cmu sbcl) socket
65   )
66
67
68 (defun accept-tcp-connection (listener)
69   "Returns (VALUES stream socket)"
70   #+allegro
71   (let ((sock (socket:accept-connection listener)))
72     (values sock sock))
73   #+clisp
74   (let ((sock (ext:socket-accept listener)))
75     (values sock sock))
76   #+cmu
77   (progn
78     (mp:process-wait-until-fd-usable listener :input)
79     (let ((sock (nth-value 0 (ext:accept-tcp-connection listener))))
80       (values (sys:make-fd-stream sock :input t :output t) sock)))
81   #+sbcl
82   (when (sb-sys:wait-until-fd-usable
83          (sb-bsd-sockets:socket-file-descriptor listener) :input)
84     (let ((sock (sb-bsd-sockets:socket-accept listener)))
85       (values
86        (sb-bsd-sockets:socket-make-stream
87         sock :element-type :default :input t :output t)
88        sock)))
89   #+openmcl 
90   (let ((sock (ccl:accept-connection listener :wait t)))
91     (values sock sock))
92   #-(or allegro clisp cmu sbcl openmcl)
93   (warn "accept-tcp-connection not supported on this implementation")
94   )
95
96
97 (defmacro errorset (form display)
98   `(handler-case
99     ,form
100     (error (e)
101      (declare (ignorable e))
102      (when ,display
103        (format t "~&Error: ~A~%" e)))))
104
105 (defun close-passive-socket (socket)
106   #+allegro (close socket)
107   #+clisp (ext:socket-server-close socket)
108   #+cmu (unix:unix-close socket)
109   #+sbcl (sb-unix:unix-close
110           (sb-bsd-sockets:socket-file-descriptor socket))
111   #+openmcl (close socket)
112   #-(or allegro clisp cmu sbcl openmcl)
113   (warn "close-passive-socket not supported on this implementation")
114   )
115
116
117 (defun close-active-socket (socket)
118   #+sbcl (sb-bsd-sockets:socket-close socket)
119   #-sbcl (close socket))
120
121 (defun ipaddr-to-dotted (ipaddr &key values)
122   "Convert from 32-bit integer to dotted string."
123   (declare (type (unsigned-byte 32) ipaddr))
124   (let ((a (logand #xff (ash ipaddr -24)))
125         (b (logand #xff (ash ipaddr -16)))
126         (c (logand #xff (ash ipaddr -8)))
127         (d (logand #xff ipaddr)))
128     (if values
129         (values a b c d)
130       (format nil "~d.~d.~d.~d" a b c d))))
131
132 (defun dotted-to-ipaddr (dotted &key (errorp t))
133   "Convert from dotted string to 32-bit integer."
134   (declare (string dotted))
135   (if errorp
136       (let ((ll (delimited-string-to-list dotted #\.)))
137         (+ (ash (parse-integer (first ll)) 24) 
138            (ash (parse-integer (second ll)) 16)
139            (ash (parse-integer (third ll)) 8) 
140            (parse-integer (fourth ll))))
141     (ignore-errors
142       (let ((ll (delimited-string-to-list dotted #\.)))
143         (+ (ash (parse-integer (first ll)) 24)
144            (ash (parse-integer (second ll)) 16)
145            (ash (parse-integer (third ll)) 8) 
146            (parse-integer (fourth ll)))))))
147
148 #+sbcl
149 (defun ipaddr-to-hostname (ipaddr &key ignore-cache)
150   (when ignore-cache
151     (warn ":IGNORE-CACHE keyword in IPADDR-TO-HOSTNAME not supported."))
152   (sb-bsd-sockets:host-ent-name
153    (sb-bsd-sockets:get-host-by-address
154     (sb-bsd-sockets:make-inet-address ipaddr))))
155
156 #+sbcl
157 (defun lookup-hostname (host &key ignore-cache)
158   (when ignore-cache
159     (warn ":IGNORE-CACHE keyword in LOOKUP-HOSTNAME not supported."))
160   (if (stringp host)
161       (sb-bsd-sockets:host-ent-address
162        (sb-bsd-sockets:get-host-by-name host))
163       (dotted-to-ipaddr (ipaddr-to-dotted host))))
164
165
166 (defun make-active-socket (server port)
167   "Returns (VALUES STREAM SOCKET)"
168   #+allegro
169   (let ((sock (socket:make-socket :remote-host server
170                                   :remote-port port)))
171     (values sock sock))
172   #+lispworks
173   (let ((sock (comm:open-tcp-stream server port)))
174     (values sock sock))
175   #+sbcl
176   (let ((sock (make-instance 'sb-bsd-sockets:inet-socket
177                              :type :stream
178                              :protocol :tcp)))
179     (sb-bsd-sockets:socket-connect sock (lookup-hostname server) port)
180     (values
181      (sb-bsd-sockets:socket-make-stream
182       sock :input t :output t :element-type :default)
183      sock))
184   #+cmu
185   (let ((sock (ext:connect-to-inet-socket server port)))
186     (values
187      (sys:make-fd-stream sock :input t :output t :element-type 'base-char)
188      sock))
189   #+clisp
190   (let ((sock (ext:socket-connect port server)))
191     (values sock sock))
192   #+openmcl
193   (let ((sock (ccl:make-socket :remote-host server :remote-port port )))
194     (values sock sock))
195   )
196
197 (defun ipaddr-array-to-dotted (array)
198   (format nil "~{~D~^.~}" (coerce array 'list))
199   #+ignore
200   (format nil "~D.~D.~D.~D" 
201           (aref 0 array) (aref 1 array) (aref 2 array) (array 3 array)))
202
203 (defun remote-host (socket)
204   #+allegro (socket:ipaddr-to-dotted (socket:remote-host socket))
205   #+lispworks (nth-value 0 (comm:get-socket-peer-address socket))
206   #+sbcl (ipaddr-array-to-dotted 
207           (nth-value 0 (sb-bsd-sockets:socket-peername socket)))
208   #+cmu (nth-value 0 (ext:get-peer-host-and-port socket))
209   #+clisp (let* ((peer (ext:socket-stream-peer socket t))
210                 (stop (position #\Space peer)))
211            ;; 2.37-2.39 had do-not-resolve-p backwards
212            (if stop (subseq peer 0 stop) peer))
213   #+openmcl (ccl:remote-host socket)
214   )
215