From 26a029d407be480d791972afb5975cf62c9360a6 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Fri, 19 Apr 2024 02:47:55 +0200
Subject: Adding upstream version 124.0.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 third_party/rust/neqo-crypto/tests/ext.rs | 99 +++++++++++++++++++++++++++++++
 1 file changed, 99 insertions(+)
 create mode 100644 third_party/rust/neqo-crypto/tests/ext.rs

(limited to 'third_party/rust/neqo-crypto/tests/ext.rs')

diff --git a/third_party/rust/neqo-crypto/tests/ext.rs b/third_party/rust/neqo-crypto/tests/ext.rs
new file mode 100644
index 0000000000..9ae81133f5
--- /dev/null
+++ b/third_party/rust/neqo-crypto/tests/ext.rs
@@ -0,0 +1,99 @@
+#![cfg_attr(feature = "deny-warnings", deny(warnings))]
+#![warn(clippy::pedantic)]
+
+use std::{cell::RefCell, rc::Rc};
+
+use neqo_crypto::{
+    constants::{HandshakeMessage, TLS_HS_CLIENT_HELLO, TLS_HS_ENCRYPTED_EXTENSIONS},
+    ext::{ExtensionHandler, ExtensionHandlerResult, ExtensionWriterResult},
+    Client, Server,
+};
+use test_fixture::fixture_init;
+
+mod handshake;
+use crate::handshake::connect;
+
+struct NoopExtensionHandler;
+impl ExtensionHandler for NoopExtensionHandler {}
+
+// This test just handshakes.  It doesn't really do anything about capturing the
+#[test]
+fn noop_extension_handler() {
+    fixture_init();
+    let mut client = Client::new("server.example", true).expect("should create client");
+    let mut server = Server::new(&["key"]).expect("should create server");
+
+    client
+        .extension_handler(0xffff, Rc::new(RefCell::new(NoopExtensionHandler)))
+        .expect("installed");
+    server
+        .extension_handler(0xffff, Rc::new(RefCell::new(NoopExtensionHandler)))
+        .expect("installed");
+
+    connect(&mut client, &mut server);
+}
+
+#[derive(Debug, Default)]
+struct SimpleExtensionHandler {
+    written: bool,
+    handled: bool,
+}
+
+impl SimpleExtensionHandler {
+    #[allow(dead_code)]
+    pub fn negotiated(&self) -> bool {
+        self.written && self.handled
+    }
+}
+
+impl ExtensionHandler for SimpleExtensionHandler {
+    fn write(&mut self, msg: HandshakeMessage, d: &mut [u8]) -> ExtensionWriterResult {
+        match msg {
+            TLS_HS_CLIENT_HELLO | TLS_HS_ENCRYPTED_EXTENSIONS => {
+                self.written = true;
+                d[0] = 77;
+                ExtensionWriterResult::Write(1)
+            }
+            _ => ExtensionWriterResult::Skip,
+        }
+    }
+
+    fn handle(&mut self, msg: HandshakeMessage, d: &[u8]) -> ExtensionHandlerResult {
+        match msg {
+            TLS_HS_CLIENT_HELLO | TLS_HS_ENCRYPTED_EXTENSIONS => {
+                self.handled = true;
+                if d.len() != 1 {
+                    ExtensionHandlerResult::Alert(50) // decode_error
+                } else if d[0] == 77 {
+                    ExtensionHandlerResult::Ok
+                } else {
+                    ExtensionHandlerResult::Alert(47) // illegal_parameter
+                }
+            }
+            _ => ExtensionHandlerResult::Alert(110), // unsupported_extension
+        }
+    }
+}
+
+#[test]
+fn simple_extension() {
+    fixture_init();
+    let mut client = Client::new("server.example", true).expect("should create client");
+    let mut server = Server::new(&["key"]).expect("should create server");
+
+    let client_handler = Rc::new(RefCell::new(SimpleExtensionHandler::default()));
+    let ch = Rc::clone(&client_handler);
+    client
+        .extension_handler(0xffff, ch)
+        .expect("client handler installed");
+    let server_handler = Rc::new(RefCell::new(SimpleExtensionHandler::default()));
+    let sh = Rc::clone(&server_handler);
+    server
+        .extension_handler(0xffff, sh)
+        .expect("server handler installed");
+
+    connect(&mut client, &mut server);
+
+    assert!(client_handler.borrow().negotiated());
+    assert!(server_handler.borrow().negotiated());
+}
-- 
cgit v1.2.3