1 files changed, 134 insertions, 0 deletions
diff --git a/guile/tests/session-record-port.scm b/guile/tests/session-record-port.scm
new file mode 100644
index 0000000..6a7ec03
--- /dev/null
+++ b/guile/tests/session-record-port.scm
@@ -0,0 +1,134 @@
+;;; GnuTLS --- Guile bindings for GnuTLS.
+;;; Copyright (C) 2007-2012, 2014, 2016, 2022 Free Software Foundation, Inc.
+;;;
+;;; GnuTLS is free software; you can redistribute it and/or
+;;; modify it under the terms of the GNU Lesser General Public
+;;; License as published by the Free Software Foundation; either
+;;; version 2.1 of the License, or (at your option) any later version.
+;;;
+;;; GnuTLS is distributed in the hope that it will be useful,
+;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
+;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+;;; Lesser General Public License for more details.
+;;;
+;;; You should have received a copy of the GNU Lesser General Public
+;;; License along with GnuTLS; if not, write to the Free Software
+;;; Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+;;; Written by Ludovic Courtès <ludo@chbouib.org>.
+
+
+;;;
+;;; Test session establishment using anonymous authentication.  Exercise the
+;;; `session-record-port' API.
+;;;
+
+(use-modules (gnutls)
+             (gnutls build tests)
+             (srfi srfi-4))
+
+;; TLS session settings.
+(define priorities
+  "NONE:+VERS-TLS1.2:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH")
+
+;; Message sent by the client.
+(define %message (apply u8vector (iota 256)))
+
+(define (import-something import-proc file fmt)
+  (let* ((path (search-path %load-path file))
+         (size (stat:size (stat path)))
+         (raw  (make-u8vector size)))
+    (uniform-vector-read! raw (open-input-file path))
+    (import-proc raw fmt)))
+
+(define (import-dh-params file)
+  (import-something pkcs3-import-dh-parameters file
+                    x509-certificate-format/pem))
+
+;; Debugging.
+;; (set-log-level! 100)
+;; (set-log-procedure! (lambda (level str)
+;;                       (format #t "[~a|~a] ~a" (getpid) level str)))
+
+(run-test
+ (lambda ()
+   ;; Stress the GC.  In 0.0, this triggered an abort due to
+   ;; "scm_unprotect_object called during GC".
+   (let ((sessions (map (lambda (i)
+                          (make-session connection-end/server))
+                        (iota 123))))
+     (for-each session-record-port sessions)
+     (gc)(gc)(gc))
+
+   ;; Stress the GC.  The session associated with each port in PORTS should
+   ;; remain reachable.
+   (let ((ports (map session-record-port
+                     (map (lambda (i)
+                            (make-session connection-end/server))
+                          (iota 123)))))
+     (gc)(gc)(gc)
+     (for-each (lambda (p)
+                 (catch 'gnutls-error
+                   (lambda ()
+                     (read p))
+                   (lambda (key . args)
+                     #t)))
+               ports))
+
+   ;; Try using the record port for I/O.
+   (let ((socket-pair (socketpair PF_UNIX SOCK_STREAM 0)))
+     (with-child-process pid
+
+       ;; server-side
+       (let ((server (make-session connection-end/server)))
+         (set-session-priorities! server priorities)
+
+         (set-session-transport-fd! server (fileno (cdr socket-pair)))
+         (let ((cred (make-anonymous-server-credentials))
+               (dh-params (import-dh-params "dh-parameters.pem")))
+           ;; Note: DH parameter generation can take some time.
+           (set-anonymous-server-dh-parameters! cred dh-params)
+           (set-session-credentials! server cred))
+         (set-session-dh-prime-bits! server 1024)
+
+         (handshake server)
+         (let* ((buf (make-u8vector (u8vector-length %message)))
+                (amount
+                 (uniform-vector-read! buf (session-record-port server))))
+           (bye server close-request/rdwr)
+
+           ;; Make sure we got everything right.
+           (and (eq? (session-record-port server)
+                     (session-record-port server))
+                (zero? (cdr (waitpid pid)))
+                (= amount (u8vector-length %message))
+                (equal? buf %message)
+                (eof-object?
+                 (read-char (session-record-port server)))
+
+                ;; Close the port and make sure its 'close' procedure is
+                ;; called.
+                (let* ((closed? #f)
+                       (port  (session-record-port server))
+                       (close (lambda (p)
+                                (format #t "closing port ~s~%" p)
+                                (set! closed? (eq? p port)))))
+                  (set-session-record-port-close! port close)
+                  (close-port port)
+                  closed?))))
+
+       ;; client-side (child process)
+       (let ((client (make-session connection-end/client)))
+         (set-session-priorities! client priorities)
+
+         (set-session-transport-port! client (car socket-pair))
+         (set-session-credentials! client (make-anonymous-client-credentials))
+         (set-session-dh-prime-bits! client 1024)
+
+         (handshake client)
+         (uniform-vector-write %message (session-record-port client))
+         (bye client close-request/rdwr)
+
+         (primitive-exit))))))
+
+;;; arch-tag: e873226a-d0b6-4a93-87ec-a1b5ad2ae8a2