314 lines
14 KiB
Racket
314 lines
14 KiB
Racket
;; https://github.com/racket/unix-socket
|
|
;; License: Apache 2.0/MIT
|
|
;; Support for UNIX domain sockets.
|
|
#lang racket/base
|
|
(require racket/contract
|
|
racket/match
|
|
(rename-in ffi/unsafe (-> -->))
|
|
ffi/unsafe/atomic
|
|
ffi/unsafe/custodian
|
|
ffi/unsafe/define
|
|
ffi/unsafe/schedule
|
|
ffi/unsafe/port
|
|
ffi/file
|
|
"unix-socket-ffi.rkt")
|
|
(provide unix-socket-available?
|
|
unix-socket-listener?
|
|
unix-socket-path?
|
|
unix-socket-connect
|
|
unix-socket-listen
|
|
(contract-out
|
|
[unix-socket-close-listener
|
|
(-> unix-socket-listener? any)]
|
|
[unix-socket-accept
|
|
(-> unix-socket-listener? (values input-port? output-port?))]
|
|
[unix-socket-accept-evt
|
|
(-> unix-socket-listener? evt?)]))
|
|
|
|
(define (unix-socket-path? v)
|
|
(and (unix-socket-path->bytes v) #t))
|
|
|
|
(define (unix-socket-path->bytes path)
|
|
(if (path-string? path)
|
|
;; On all platforms, normal path of up to UNIX-PATH-MAX bytes after
|
|
;; conversion to absolute is considered valid and shall be accepted.
|
|
(let ([bstr (path->bytes (cleanse-path (path->complete-path path)))])
|
|
(and (<= (bytes-length bstr) UNIX-PATH-MAX) bstr))
|
|
|
|
;; On Linux, paths may be in so-called abstract namespace where they
|
|
;; start with #\nul and do not have a corresponding socket file.
|
|
;; We accept such paths only as byte strings because we don't know
|
|
;; the correct encoding.
|
|
(and (eq? platform 'linux)
|
|
(bytes? path)
|
|
(> (bytes-length path) 0)
|
|
(<= (bytes-length path) UNIX-PATH-MAX)
|
|
(= (bytes-ref path 0) 0)
|
|
path)))
|
|
|
|
(define (check-available who)
|
|
(unless unix-socket-available?
|
|
(error who "unix domain sockets are not supported on this platform")))
|
|
|
|
;; do-make-sockaddr : Symbol Path/String -> (values Sockaddr-Pointer Nat)
|
|
(define (do-make-sockaddr who path)
|
|
(when (path-string? path)
|
|
(security-guard-check-file who path '(read write)))
|
|
(define path-bytes (unix-socket-path->bytes path))
|
|
(define sockaddr (make-sockaddr path-bytes))
|
|
(define addrlen (+ (ctype-sizeof _ushort) (bytes-length path-bytes)))
|
|
(values sockaddr addrlen))
|
|
|
|
;; do-make-socket : Symbol Nat -> (values FD Cust-Reg)
|
|
;; Creates nonblocking socket, registers w/ custodian (returning registration).
|
|
;; Should be called in atomic mode.
|
|
(define (do-make-socket who mode)
|
|
(define socket-fd (socket AF-UNIX mode 0))
|
|
(unless (positive? socket-fd)
|
|
(error who "failed to create socket~a"
|
|
(errno-error-lines (saved-errno))))
|
|
(set-fd-nonblocking who socket-fd)
|
|
(values socket-fd (register-custodian-shutdown socket-fd close/unregister)))
|
|
|
|
;; set-fd-nonblocking : Symbol Nat -> Void
|
|
(define (set-fd-nonblocking who fd)
|
|
(unless (zero? (fcntl fd F_SETFL O_NONBLOCK))
|
|
(close fd)
|
|
(error who "failed to set non-blocking mode~a"
|
|
(errno-error-lines (saved-errno)))))
|
|
|
|
;; close/unregister : Nat Cust-Reg/#f -> Void
|
|
(define (close/unregister fd [reg #f])
|
|
(close fd)
|
|
(socket->semaphore fd 'remove)
|
|
(when reg (unregister-custodian-shutdown fd reg)))
|
|
|
|
;; make-socket-ports : Symbol FD Cust-Reg/#f -> (values Input-Port Output-Port)
|
|
(define (make-socket-ports who socket-fd reg)
|
|
(with-handlers ([(lambda (e) #t)
|
|
(lambda (exn)
|
|
(close/unregister socket-fd reg)
|
|
(raise exn))])
|
|
(define-values (in out)
|
|
(unsafe-file-descriptor->port socket-fd #"unix-socket-port" '(read write)))
|
|
;; closing the ports closes socket-fd, so custodian no longer needs to manage directly
|
|
(when reg (unregister-custodian-shutdown socket-fd reg))
|
|
(define fd+ports (list socket-fd in out))
|
|
(values (wrap-input-port in fd+ports) (wrap-output-port out fd+ports))))
|
|
|
|
;; wrap-output-port : Output-Port (List FD Port Port) -> Output-Port
|
|
;; Wrap port, override close to shutdown write side of socket.
|
|
(define (wrap-output-port out fd+ports)
|
|
(define (close)
|
|
(when out (close-output-port out)) ;; may block, so avoid in custodian shutdown
|
|
(call-as-atomic
|
|
(lambda ()
|
|
(when creg (unregister-custodian-shutdown out* creg) (set! creg #f))
|
|
(when fd+ports (do-shutdown fd+ports #t) (set! fd+ports #f)))))
|
|
(define (get-write-evt buf start end) (write-bytes-avail-evt buf out start end))
|
|
(define buffer-mode (make-buffer-mode-fun out))
|
|
(define out*
|
|
(make-output-port 'unix-socket out out close #f get-write-evt #f #f void 1 buffer-mode))
|
|
(define creg (register-custodian-shutdown out* (lambda (p) (set! out #f) (close-output-port p))))
|
|
out*)
|
|
|
|
;; wrap-input-port : Input-Port (List FD Port Port) -> Input-Port
|
|
(define (wrap-input-port in fd+ports)
|
|
(define (close)
|
|
(when in (close-input-port in))
|
|
(call-as-atomic
|
|
(lambda ()
|
|
(when creg (unregister-custodian-shutdown in* creg) (set! creg #f))
|
|
(when fd+ports (do-shutdown fd+ports #f) (set! fd+ports #f)))))
|
|
(define (get-progress-evt) (port-progress-evt in))
|
|
(define (commit k progress done) (port-commit-peeked k progress done in))
|
|
(define buffer-mode (make-buffer-mode-fun in))
|
|
(define in*
|
|
(make-input-port 'unix-socket in in close get-progress-evt commit #f void 1 buffer-mode))
|
|
(define creg (register-custodian-shutdown in* (lambda (p) (set! in #f) (close-input-port p))))
|
|
in*)
|
|
|
|
(define (make-buffer-mode-fun port)
|
|
(case-lambda [() (file-stream-buffer-mode port)]
|
|
[(mode) (file-stream-buffer-mode port mode)]))
|
|
|
|
;; do-shutdown : (List FD Port Port) Boolean -> Void
|
|
;; Requirements:
|
|
;; - want to shutdown RD/WR when corresponding port closed
|
|
;; - want to shutdown *after* port closed to avoid low-level errors
|
|
;; - must *not* call shutdown after *both* ports closed (fd is stale)
|
|
;; So: okay to call shutdown if either of the ports is still open.
|
|
(define (do-shutdown fd+ports output?)
|
|
(define socket-fd (car fd+ports))
|
|
(define ports (cdr fd+ports))
|
|
(unless (andmap port-closed? ports)
|
|
(unless (zero? (shutdown socket-fd (if output? SHUT_WR SHUT_RD)))
|
|
;; ENOTCONN is okay; the other side may have disconnected.
|
|
(unless (= (saved-errno) ENOTCONN)
|
|
(error (if output? 'close-output-port/unix-socket 'close-input-port/unix-socket)
|
|
"error from shutdown~a" (errno-error-lines (saved-errno)))))))
|
|
|
|
;; ============================================================
|
|
;; Connect
|
|
|
|
;; unix-socket-connect : Path/String [Symbol] -> (values Input-Port Output-Port)
|
|
(define (unix-socket-connect path [mode 'SOCK-STREAM])
|
|
(check-available 'unix-socket-connect)
|
|
(define-values (sockaddr addrlen) (do-make-sockaddr 'unix-socket-connect path))
|
|
(define connect-k
|
|
;; Non-blocking connect may succeed immediately or require waiting to see.
|
|
;; - If succeeds immediately, make ports in same atomic block
|
|
;; - If wait, must exit atomic mode to sync
|
|
;; So we return a procedure to be applied in non-atomic mode that does
|
|
;; whatever needs doing.
|
|
(call-as-atomic
|
|
(lambda ()
|
|
(when (custodian-shut-down? (current-custodian))
|
|
(error 'unix-socket-connect "the custodian has been shut down"))
|
|
(define-values (socket-fd reg) (do-make-socket 'unix-socket-connect
|
|
(match mode
|
|
['SOCK-STREAM SOCK-STREAM]
|
|
['SOCK-SEQPACKET SOCK-SEQPACKET])))
|
|
(define r (connect socket-fd sockaddr addrlen))
|
|
(define errno (saved-errno))
|
|
(cond [(= r 0) ;; connected
|
|
(define-values (in out) (make-socket-ports 'unix-socket-connect socket-fd reg))
|
|
(lambda () (values in out))]
|
|
[(= errno EINPROGRESS) ;; wait and see
|
|
(define sema (socket->semaphore socket-fd 'write))
|
|
(lambda () ;; called in non-atomic mode!
|
|
(sync sema)
|
|
;; FIXME: check custodian hasn't been shut down?
|
|
(call-as-atomic
|
|
(lambda ()
|
|
(define errno (getsockopt socket-fd SOL_SOCKET SO_ERROR))
|
|
(cond [(= errno 0)
|
|
(make-socket-ports 'unix-socket-connect socket-fd reg)]
|
|
[else
|
|
(close/unregister socket-fd reg)
|
|
(error 'unix-socket-connect
|
|
"failed to connect socket (non-blocking)\n path: ~e~a"
|
|
path (errno-error-lines errno))]))))]
|
|
[else
|
|
(close/unregister socket-fd reg)
|
|
(error 'unix-socket-connect "failed to connect socket\n path: ~e~a"
|
|
path (errno-error-lines (saved-errno)))]))))
|
|
(connect-k))
|
|
|
|
|
|
;; ============================================================
|
|
;; Listen & Accept
|
|
|
|
;; A Listener is (unix-socket-listener Nat/#f Cust-Reg/#f)
|
|
;; States:
|
|
;; OPEN: (unix-socket-listener Nat Cust-Reg)
|
|
;; CLOSED: (unix-socket-listener #f #f)
|
|
;; Only transition allowed is OPEN to CLOSED, must happen atomically.
|
|
(struct unix-socket-listener (fd reg)
|
|
#:mutable
|
|
#:property prop:evt
|
|
(lambda (self)
|
|
(call-as-atomic
|
|
(lambda ()
|
|
(wrap-evt
|
|
;; ready when fd is readable OR listener is closed
|
|
;; If closed after evt creation, then fd-sema becomes ready
|
|
;; when fd closed and fd-sema unregistered.
|
|
(cond [(unix-socket-listener-fd self)
|
|
=> (lambda (fd) (socket->semaphore fd 'read))]
|
|
[else always-evt])
|
|
(lambda (r) self))))))
|
|
|
|
;; unix-socket-listen : Path/String [Nat] [Symbol] -> Unix-Socket-Listener
|
|
(define (unix-socket-listen path [backlog 4] [mode 'SOCK-STREAM])
|
|
(check-available 'unix-socket-listen)
|
|
(define-values (sockaddr addrlen) (do-make-sockaddr 'unix-socket-listen path))
|
|
(call-as-atomic
|
|
(lambda ()
|
|
(define-values (socket-fd reg) (do-make-socket 'unix-socket-listen
|
|
(match mode
|
|
['SOCK-STREAM SOCK-STREAM]
|
|
['SOCK-SEQPACKET SOCK-SEQPACKET])))
|
|
(unless (zero? (bind socket-fd sockaddr addrlen))
|
|
(close/unregister socket-fd reg)
|
|
(error 'unix-socket-listen "failed to bind socket\n path: ~e~a"
|
|
path (errno-error-lines (saved-errno))))
|
|
(unless (zero? (listen socket-fd backlog))
|
|
(close/unregister socket-fd reg)
|
|
(error 'unix-socket-listen "failed to listen\n path: ~e~a"
|
|
path (errno-error-lines (saved-errno))))
|
|
(define listener (unix-socket-listener socket-fd #f))
|
|
(set-unix-socket-listener-reg! listener
|
|
(register-custodian-shutdown listener do-close-listener))
|
|
(unregister-custodian-shutdown socket-fd reg)
|
|
listener)))
|
|
|
|
;; unix-socket-close-listener : Listener -> Void
|
|
(define (unix-socket-close-listener l)
|
|
(call-as-atomic (lambda () (do-close-listener l #t))))
|
|
|
|
(define (do-close-listener l [unregister? #f])
|
|
(define fd (unix-socket-listener-fd l))
|
|
(define reg (unix-socket-listener-reg l))
|
|
(when fd
|
|
(set-unix-socket-listener-fd! l #f)
|
|
(set-unix-socket-listener-reg! l #f)
|
|
(when unregister? (unregister-custodian-shutdown l reg))
|
|
(close/unregister fd)
|
|
(void)))
|
|
|
|
;; ----------------------------------------
|
|
|
|
(struct accept-evt (who listener cust) ;; <: (Evt-of (-> (list Input-Port Output-Port)))
|
|
#:property prop:evt
|
|
(unsafe-poller (lambda (self maybe-wakeups) (accept-poll self maybe-wakeups))))
|
|
|
|
;; unix-socket-accept : Unix-Socket-Listener -> (values Input-Port Output-Port)
|
|
(define (unix-socket-accept l)
|
|
(apply values ((sync (accept-evt 'unix-socket-accept l (current-custodian))))))
|
|
|
|
;; unix-socket-accept-evt : Unix-Socket-Listener -> (Evt-of (list Input-Port Output-Port))
|
|
(define (unix-socket-accept-evt l)
|
|
(wrap-evt (accept-evt 'unix-socket-accept-evt l (current-custodian)) (lambda (r) (r))))
|
|
|
|
;; accept-poll : Accept-Evt (U #f Wakeups) -> (U (values List #f) (values #f Evt))
|
|
(define (accept-poll accept-evt maybe-wakeups)
|
|
(define l (accept-evt-listener accept-evt))
|
|
(define who (accept-evt-who accept-evt))
|
|
(define lfd (unix-socket-listener-fd l))
|
|
(cond [(eq? lfd #f)
|
|
(values (list (lambda () (error who "unix socket listener is closed")))
|
|
#f)]
|
|
[(custodian-shut-down? (accept-evt-cust accept-evt))
|
|
(error '|unix-socket-accept-evt poll| "the custodian has been shut down")]
|
|
[lfd
|
|
(cond [maybe-wakeups (accept-poll/sleep who accept-evt maybe-wakeups lfd)]
|
|
[else (accept-poll/check who accept-evt lfd)])]))
|
|
|
|
(define (accept-poll/sleep who accept-evt wakeups lfd)
|
|
;; No need to register wakeup for custodian; if custodian is shut down, then
|
|
;; lfd semaphore becomes ready when it is unregistered
|
|
(unsafe-poll-ctx-fd-wakeup wakeups lfd 'read)
|
|
(values #f accept-evt))
|
|
|
|
(define (accept-poll/check who accept-evt lfd)
|
|
(define fd (accept lfd))
|
|
(cond [(< fd 0)
|
|
(let ([errno (saved-errno)])
|
|
(cond [(or (= errno EAGAIN) (= errno EWOULDBLOCK) (= errno EINTR))
|
|
(values #f accept-evt)]
|
|
[else
|
|
(values (list (lambda ()
|
|
(error who "failed to accept socket~a"
|
|
(errno-error-lines errno))))
|
|
#f)]))]
|
|
[else
|
|
(define cust (accept-evt-cust accept-evt))
|
|
(define r
|
|
(with-handlers ([(lambda (e) #t)
|
|
(lambda (e) (lambda () (raise e)))])
|
|
(parameterize ((current-custodian cust))
|
|
(define-values (in out) (make-socket-ports who fd #f))
|
|
(lambda () (list in out)))))
|
|
(values (list r) #f)]))
|