aboutsummaryrefslogtreecommitdiff
path: root/openssl/src/ssl/tests.rs
diff options
context:
space:
mode:
authorAlex Crichton <[email protected]>2015-09-22 21:34:12 -0700
committerAlex Crichton <[email protected]>2015-10-22 08:45:29 -0700
commitc895b9f09f25410b6e0046d71d42439f7a56c1be (patch)
tree581268a8f53487e072f29af0b79ae8774a5d0269 /openssl/src/ssl/tests.rs
parentMerge pull request #272 from dropbox/async_support (diff)
downloadrust-openssl-c895b9f09f25410b6e0046d71d42439f7a56c1be.tar.xz
rust-openssl-c895b9f09f25410b6e0046d71d42439f7a56c1be.zip
Get nonblocking tests working on OSX/Windows
Diffstat (limited to 'openssl/src/ssl/tests.rs')
-rw-r--r--openssl/src/ssl/tests.rs953
1 files changed, 0 insertions, 953 deletions
diff --git a/openssl/src/ssl/tests.rs b/openssl/src/ssl/tests.rs
deleted file mode 100644
index 8335bc53..00000000
--- a/openssl/src/ssl/tests.rs
+++ /dev/null
@@ -1,953 +0,0 @@
-#![allow(unused_imports)]
-
-use std::fs::File;
-use std::io::prelude::*;
-use std::io::{self, BufReader};
-use std::iter;
-use std::net::{TcpStream, TcpListener, SocketAddr};
-use std::path::Path;
-use std::process::{Command, Child, Stdio, ChildStdin};
-use std::thread;
-
-use crypto::hash::Type::{SHA256};
-use ssl;
-use ssl::SslMethod;
-use ssl::SslMethod::Sslv23;
-use ssl::{SslContext, SslStream, VerifyCallback};
-use ssl::SSL_VERIFY_PEER;
-use x509::X509StoreContext;
-use x509::X509FileType;
-use x509::X509;
-use crypto::pkey::PKey;
-
-#[cfg(feature="dtlsv1")]
-use std::net::UdpSocket;
-#[cfg(feature="dtlsv1")]
-use ssl::SslMethod::Dtlsv1;
-#[cfg(feature="sslv2")]
-use ssl::SslMethod::Sslv2;
-#[cfg(feature="dtlsv1")]
-use net2::UdpSocketExt;
-
-fn next_addr() -> SocketAddr {
- use std::sync::atomic::{AtomicUsize, ATOMIC_USIZE_INIT, Ordering};
- static PORT: AtomicUsize = ATOMIC_USIZE_INIT;
- let port = 15411 + PORT.fetch_add(1, Ordering::SeqCst);
-
- format!("127.0.0.1:{}", port).parse().unwrap()
-}
-
-struct Server {
- p: Child,
-}
-
-impl Server {
- fn spawn(args: &[&str], input: Option<Box<FnMut(ChildStdin) + Send>>)
- -> (Server, SocketAddr) {
- let addr = next_addr();
- let mut child = Command::new("openssl").arg("s_server")
- .arg("-accept").arg(addr.port().to_string())
- .args(args)
- .arg("-cert").arg("cert.pem")
- .arg("-key").arg("key.pem")
- .arg("-no_dhe")
- .current_dir("test")
- .stdout(Stdio::null())
- .stderr(Stdio::null())
- .stdin(Stdio::piped())
- .spawn().unwrap();
- let stdin = child.stdin.take().unwrap();
- if let Some(mut input) = input {
- thread::spawn(move || input(stdin));
- }
- (Server { p: child }, addr)
- }
-
- fn new_tcp(args: &[&str]) -> (Server, TcpStream) {
- let (server, addr) = Server::spawn(args, None);
- loop {
- match TcpStream::connect(&addr) {
- Ok(s) => return (server, s),
- Err(ref e) if e.kind() == io::ErrorKind::ConnectionRefused => {
- thread::sleep_ms(100);
- }
- Err(e) => panic!("wut: {}", e),
- }
- }
- }
-
- fn new() -> (Server, TcpStream) {
- Server::new_tcp(&["-www"])
- }
-
- #[cfg(any(feature = "alpn", feature = "npn"))]
- fn new_alpn() -> (Server, TcpStream) {
- Server::new_tcp(&["-www", "-nextprotoneg", "http/1.1,spdy/3.1",
- "-alpn", "http/1.1,spdy/3.1"])
- }
-
- #[cfg(feature = "dtlsv1")]
- fn new_dtlsv1<I>(input: I) -> (Server, UdpConnected)
- where I: IntoIterator<Item=&'static str>,
- I::IntoIter: Send + 'static
- {
- let mut input = input.into_iter();
- let (s, addr) = Server::spawn(&["-dtls1"], Some(Box::new(move |mut io| {
- for s in input.by_ref() {
- if io.write_all(s.as_bytes()).is_err() {
- break
- }
- }
- })));
- // Need to wait for the UDP socket to get bound in our child process,
- // but don't currently have a great way to do that so just wait for a
- // bit.
- thread::sleep_ms(100);
- let socket = UdpSocket::bind(next_addr()).unwrap();
- socket.connect(&addr).unwrap();
- (s, UdpConnected(socket))
- }
-}
-
-impl Drop for Server {
- fn drop(&mut self) {
- let _ = self.p.kill();
- let _ = self.p.wait();
- }
-}
-
-#[cfg(feature = "dtlsv1")]
-struct UdpConnected(UdpSocket);
-
-#[cfg(feature = "dtlsv1")]
-impl Read for UdpConnected {
- fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
- self.0.recv_from(buf).map(|(s, _)| s)
- }
-}
-
-#[cfg(feature = "dtlsv1")]
-impl Write for UdpConnected {
- #[cfg(unix)]
- fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
- use std::os::unix::prelude::*;
- use libc;
- let n = unsafe {
- libc::send(self.0.as_raw_fd(), buf.as_ptr() as *const _,
- buf.len() as libc::size_t, 0)
- };
- if n < 0 {
- Err(io::Error::last_os_error())
- } else {
- Ok(n as usize)
- }
- }
-
- #[cfg(windows)]
- fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
- use std::os::windows::prelude::*;
- use libc;
- let n = unsafe {
- libc::send(self.0.as_raw_socket(), buf.as_ptr() as *const _,
- buf.len() as libc::c_int, 0)
- };
- if n < 0 {
- Err(io::Error::last_os_error())
- } else {
- Ok(n as usize)
- }
- }
-
- fn flush(&mut self) -> io::Result<()> { Ok(()) }
-}
-
-macro_rules! run_test(
- ($module:ident, $blk:expr) => (
- #[cfg(test)]
- mod $module {
- use std::io;
- use std::io::prelude::*;
- use std::path::Path;
- use std::net::UdpSocket;
- use std::net::TcpStream;
- use ssl;
- use ssl::SslMethod;
- use ssl::{SslContext, Ssl, SslStream, VerifyCallback};
- use ssl::SSL_VERIFY_PEER;
- use crypto::hash::Type::SHA256;
- use x509::X509StoreContext;
- use serialize::hex::FromHex;
- use super::Server;
-
- #[test]
- fn sslv23() {
- let (_s, stream) = Server::new();
- $blk(SslMethod::Sslv23, stream);
- }
-
- #[test]
- #[cfg(feature="dtlsv1")]
- fn dtlsv1() {
- let (_s, stream) = Server::new_dtlsv1(Some("hello"));
- $blk(SslMethod::Dtlsv1, stream);
- }
- }
- );
-);
-
-run_test!(new_ctx, |method, _| {
- SslContext::new(method).unwrap();
-});
-
-run_test!(new_sslstream, |method, stream| {
- SslStream::connect_generic(&SslContext::new(method).unwrap(), stream).unwrap();
-});
-
-run_test!(get_ssl_method, |method, _| {
- let ssl = Ssl::new(&SslContext::new(method).unwrap()).unwrap();
- assert_eq!(ssl.get_ssl_method(), Some(method));
-});
-
-run_test!(verify_untrusted, |method, stream| {
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
-
- match SslStream::connect_generic(&ctx, stream) {
- Ok(_) => panic!("expected failure"),
- Err(err) => println!("error {:?}", err)
- }
-});
-
-run_test!(verify_trusted, |method, stream| {
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
-
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- match SslStream::connect_generic(&ctx, stream) {
- Ok(_) => (),
- Err(err) => panic!("Expected success, got {:?}", err)
- }
-});
-
-run_test!(verify_untrusted_callback_override_ok, |method, stream| {
- fn callback(_preverify_ok: bool, _x509_ctx: &X509StoreContext) -> bool {
- true
- }
-
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, Some(callback as VerifyCallback));
-
- match SslStream::connect_generic(&ctx, stream) {
- Ok(_) => (),
- Err(err) => panic!("Expected success, got {:?}", err)
- }
-});
-
-run_test!(verify_untrusted_callback_override_bad, |method, stream| {
- fn callback(_preverify_ok: bool, _x509_ctx: &X509StoreContext) -> bool {
- false
- }
-
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, Some(callback as VerifyCallback));
-
- assert!(SslStream::connect_generic(&ctx, stream).is_err());
-});
-
-run_test!(verify_trusted_callback_override_ok, |method, stream| {
- fn callback(_preverify_ok: bool, _x509_ctx: &X509StoreContext) -> bool {
- true
- }
-
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, Some(callback as VerifyCallback));
-
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- match SslStream::connect_generic(&ctx, stream) {
- Ok(_) => (),
- Err(err) => panic!("Expected success, got {:?}", err)
- }
-});
-
-run_test!(verify_trusted_callback_override_bad, |method, stream| {
- fn callback(_preverify_ok: bool, _x509_ctx: &X509StoreContext) -> bool {
- false
- }
-
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, Some(callback as VerifyCallback));
-
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- assert!(SslStream::connect_generic(&ctx, stream).is_err());
-});
-
-run_test!(verify_callback_load_certs, |method, stream| {
- fn callback(_preverify_ok: bool, x509_ctx: &X509StoreContext) -> bool {
- assert!(x509_ctx.get_current_cert().is_some());
- true
- }
-
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, Some(callback as VerifyCallback));
-
- assert!(SslStream::connect_generic(&ctx, stream).is_ok());
-});
-
-run_test!(verify_trusted_get_error_ok, |method, stream| {
- fn callback(_preverify_ok: bool, x509_ctx: &X509StoreContext) -> bool {
- assert!(x509_ctx.get_error().is_none());
- true
- }
-
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, Some(callback as VerifyCallback));
-
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- assert!(SslStream::connect_generic(&ctx, stream).is_ok());
-});
-
-run_test!(verify_trusted_get_error_err, |method, stream| {
- fn callback(_preverify_ok: bool, x509_ctx: &X509StoreContext) -> bool {
- assert!(x509_ctx.get_error().is_some());
- false
- }
-
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, Some(callback as VerifyCallback));
-
- assert!(SslStream::connect_generic(&ctx, stream).is_err());
-});
-
-run_test!(verify_callback_data, |method, stream| {
- fn callback(_preverify_ok: bool, x509_ctx: &X509StoreContext, node_id: &Vec<u8>) -> bool {
- let cert = x509_ctx.get_current_cert();
- match cert {
- None => false,
- Some(cert) => {
- let fingerprint = cert.fingerprint(SHA256).unwrap();
- &fingerprint == node_id
- }
- }
- }
- let mut ctx = SslContext::new(method).unwrap();
-
- // Node id was generated as SHA256 hash of certificate "test/cert.pem"
- // in DER format.
- // Command: openssl x509 -in test/cert.pem -outform DER | openssl dgst -sha256
- // Please update if "test/cert.pem" will ever change
- let node_hash_str = "db400bb62f1b1f29c3b8f323b8f7d9dea724fdcd67104ef549c772ae3749655b";
- let node_id = node_hash_str.from_hex().unwrap();
- ctx.set_verify_with_data(SSL_VERIFY_PEER, callback, node_id);
- ctx.set_verify_depth(1);
-
- match SslStream::connect_generic(&ctx, stream) {
- Ok(_) => (),
- Err(err) => panic!("Expected success, got {:?}", err)
- }
-});
-
-// Make sure every write call translates to a write call to the underlying socket.
-#[test]
-fn test_write_hits_stream() {
- let listener = TcpListener::bind(next_addr()).unwrap();
- let addr = listener.local_addr().unwrap();
-
- let guard = thread::spawn(move || {
- let ctx = SslContext::new(Sslv23).unwrap();
- let stream = TcpStream::connect(addr).unwrap();
- let mut stream = SslStream::connect_generic(&ctx, stream).unwrap();
-
- stream.write_all(b"hello").unwrap();
- stream
- });
-
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_certificate_file(&Path::new("test/cert.pem"), X509FileType::PEM).unwrap();
- ctx.set_private_key_file(&Path::new("test/key.pem"), X509FileType::PEM).unwrap();
- let stream = listener.accept().unwrap().0;
- let mut stream = SslStream::accept(&ctx, stream).unwrap();
-
- let mut buf = [0; 5];
- assert_eq!(5, stream.read(&mut buf).unwrap());
- assert_eq!(&b"hello"[..], &buf[..]);
- guard.join().unwrap();
-}
-
-#[test]
-fn test_set_certificate_and_private_key() {
- let key_path = Path::new("test/key.pem");
- let cert_path = Path::new("test/cert.pem");
- let mut key_file = File::open(&key_path)
- .ok()
- .expect("Failed to open `test/key.pem`");
- let mut cert_file = File::open(&cert_path)
- .ok()
- .expect("Failed to open `test/cert.pem`");
-
- let key = PKey::private_key_from_pem(&mut key_file).unwrap();
- let cert = X509::from_pem(&mut cert_file).unwrap();
-
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_private_key(&key).unwrap();
- ctx.set_certificate(&cert).unwrap();
-
- assert!(ctx.check_private_key().is_ok());
-}
-
-run_test!(get_ctx_options, |method, _| {
- let mut ctx = SslContext::new(method).unwrap();
- ctx.get_options();
-});
-
-run_test!(set_ctx_options, |method, _| {
- let mut ctx = SslContext::new(method).unwrap();
- let opts = ctx.set_options(ssl::SSL_OP_NO_TICKET);
- assert!(opts.contains(ssl::SSL_OP_NO_TICKET));
- assert!(!opts.contains(ssl::SSL_OP_CISCO_ANYCONNECT));
- let more_opts = ctx.set_options(ssl::SSL_OP_CISCO_ANYCONNECT);
- assert!(more_opts.contains(ssl::SSL_OP_NO_TICKET));
- assert!(more_opts.contains(ssl::SSL_OP_CISCO_ANYCONNECT));
-});
-
-run_test!(clear_ctx_options, |method, _| {
- let mut ctx = SslContext::new(method).unwrap();
- ctx.set_options(ssl::SSL_OP_ALL);
- let opts = ctx.clear_options(ssl::SSL_OP_ALL);
- assert!(!opts.contains(ssl::SSL_OP_ALL));
-});
-
-#[test]
-fn test_write() {
- let (_s, stream) = Server::new();
- let mut stream = SslStream::connect_generic(&SslContext::new(Sslv23).unwrap(), stream).unwrap();
- stream.write_all("hello".as_bytes()).unwrap();
- stream.flush().unwrap();
- stream.write_all(" there".as_bytes()).unwrap();
- stream.flush().unwrap();
-}
-
-#[test]
-fn test_write_direct() {
- let (_s, stream) = Server::new();
- let mut stream = SslStream::connect(&SslContext::new(Sslv23).unwrap(), stream).unwrap();
- stream.write_all("hello".as_bytes()).unwrap();
- stream.flush().unwrap();
- stream.write_all(" there".as_bytes()).unwrap();
- stream.flush().unwrap();
-}
-
-run_test!(get_peer_certificate, |method, stream| {
- let stream = SslStream::connect_generic(&SslContext::new(method).unwrap(),
- stream).unwrap();
- let cert = stream.get_peer_certificate().unwrap();
- let fingerprint = cert.fingerprint(SHA256).unwrap();
- let node_hash_str = "db400bb62f1b1f29c3b8f323b8f7d9dea724fdcd67104ef549c772ae3749655b";
- let node_id = node_hash_str.from_hex().unwrap();
- assert_eq!(node_id, fingerprint)
-});
-
-#[test]
-#[cfg(feature = "dtlsv1")]
-fn test_write_dtlsv1() {
- let (_s, stream) = Server::new_dtlsv1(iter::repeat("y\n"));
-
- let mut stream = SslStream::connect_generic(&SslContext::new(Dtlsv1).unwrap(),
- stream).unwrap();
- stream.write_all(b"hello").unwrap();
- stream.flush().unwrap();
- stream.write_all(b" there").unwrap();
- stream.flush().unwrap();
-}
-
-#[test]
-fn test_read() {
- let (_s, tcp) = Server::new();
- let mut stream = SslStream::connect_generic(&SslContext::new(Sslv23).unwrap(), tcp).unwrap();
- stream.write_all("GET /\r\n\r\n".as_bytes()).unwrap();
- stream.flush().unwrap();
- io::copy(&mut stream, &mut io::sink()).ok().expect("read error");
-}
-
-#[test]
-fn test_read_direct() {
- let (_s, tcp) = Server::new();
- let mut stream = SslStream::connect(&SslContext::new(Sslv23).unwrap(), tcp).unwrap();
- stream.write_all("GET /\r\n\r\n".as_bytes()).unwrap();
- stream.flush().unwrap();
- io::copy(&mut stream, &mut io::sink()).ok().expect("read error");
-}
-
-#[test]
-fn test_pending() {
- let (_s, tcp) = Server::new();
- let mut stream = SslStream::connect_generic(&SslContext::new(Sslv23).unwrap(), tcp).unwrap();
- stream.write_all("GET /\r\n\r\n".as_bytes()).unwrap();
- stream.flush().unwrap();
-
- // wait for the response and read first byte...
- let mut buf = [0u8; 16*1024];
- stream.read(&mut buf[..1]).unwrap();
-
- let pending = stream.pending();
- let len = stream.read(&mut buf[1..]).unwrap();
-
- assert_eq!(pending, len);
-
- stream.read(&mut buf[..1]).unwrap();
-
- let pending = stream.pending();
- let len = stream.read(&mut buf[1..]).unwrap();
- assert_eq!(pending, len);
-}
-
-#[test]
-fn test_state() {
- let (_s, tcp) = Server::new();
- let stream = SslStream::connect_generic(&SslContext::new(Sslv23).unwrap(), tcp).unwrap();
- assert_eq!(stream.get_state_string(), "SSLOK ");
- assert_eq!(stream.get_state_string_long(), "SSL negotiation finished successfully");
-}
-
-/// Tests that connecting with the client using ALPN, but the server not does not
-/// break the existing connection behavior.
-#[test]
-#[cfg(feature = "alpn")]
-fn test_connect_with_unilateral_alpn() {
- let (_s, stream) = Server::new();
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_alpn_protocols(&[b"http/1.1", b"spdy/3.1"]);
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- let stream = match SslStream::new(&ctx, stream) {
- Ok(stream) => stream,
- Err(err) => panic!("Expected success, got {:?}", err)
- };
- // Since the socket to which we connected is not configured to use ALPN,
- // there should be no selected protocol...
- assert!(stream.get_selected_alpn_protocol().is_none());
-}
-
-/// Tests that connecting with the client using NPN, but the server not does not
-/// break the existing connection behavior.
-#[test]
-#[cfg(feature = "npn")]
-fn test_connect_with_unilateral_npn() {
- let (_s, stream) = Server::new();
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_npn_protocols(&[b"http/1.1", b"spdy/3.1"]);
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- let stream = match SslStream::connect_generic(&ctx, stream) {
- Ok(stream) => stream,
- Err(err) => panic!("Expected success, got {:?}", err)
- };
- // Since the socket to which we connected is not configured to use NPN,
- // there should be no selected protocol...
- assert!(stream.get_selected_npn_protocol().is_none());
-}
-
-/// Tests that when both the client as well as the server use ALPN and their
-/// lists of supported protocols have an overlap, the correct protocol is chosen.
-#[test]
-#[cfg(feature = "alpn")]
-fn test_connect_with_alpn_successful_multiple_matching() {
- let (_s, stream) = Server::new_alpn();
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_alpn_protocols(&[b"spdy/3.1", b"http/1.1"]);
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- let stream = match SslStream::new(&ctx, stream) {
- Ok(stream) => stream,
- Err(err) => panic!("Expected success, got {:?}", err)
- };
- // The server prefers "http/1.1", so that is chosen, even though the client
- // would prefer "spdy/3.1"
- assert_eq!(b"http/1.1", stream.get_selected_alpn_protocol().unwrap());
-}
-
-/// Tests that when both the client as well as the server use NPN and their
-/// lists of supported protocols have an overlap, the correct protocol is chosen.
-#[test]
-#[cfg(feature = "npn")]
-fn test_connect_with_npn_successful_multiple_matching() {
- let (_s, stream) = Server::new_alpn();
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_npn_protocols(&[b"spdy/3.1", b"http/1.1"]);
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- let stream = match SslStream::connect_generic(&ctx, stream) {
- Ok(stream) => stream,
- Err(err) => panic!("Expected success, got {:?}", err)
- };
- // The server prefers "http/1.1", so that is chosen, even though the client
- // would prefer "spdy/3.1"
- assert_eq!(b"http/1.1", stream.get_selected_npn_protocol().unwrap());
-}
-
-/// Tests that when both the client as well as the server use ALPN and their
-/// lists of supported protocols have an overlap -- with only ONE protocol
-/// being valid for both.
-#[test]
-#[cfg(feature = "alpn")]
-fn test_connect_with_alpn_successful_single_match() {
- let (_s, stream) = Server::new_alpn();
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_alpn_protocols(&[b"spdy/3.1"]);
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- let stream = match SslStream::new(&ctx, stream) {
- Ok(stream) => stream,
- Err(err) => panic!("Expected success, got {:?}", err)
- };
- // The client now only supports one of the server's protocols, so that one
- // is used.
- assert_eq!(b"spdy/3.1", stream.get_selected_alpn_protocol().unwrap());
-}
-
-
-/// Tests that when both the client as well as the server use NPN and their
-/// lists of supported protocols have an overlap -- with only ONE protocol
-/// being valid for both.
-#[test]
-#[cfg(feature = "npn")]
-fn test_connect_with_npn_successful_single_match() {
- let (_s, stream) = Server::new_alpn();
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_npn_protocols(&[b"spdy/3.1"]);
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- let stream = match SslStream::connect_generic(&ctx, stream) {
- Ok(stream) => stream,
- Err(err) => panic!("Expected success, got {:?}", err)
- };
- // The client now only supports one of the server's protocols, so that one
- // is used.
- assert_eq!(b"spdy/3.1", stream.get_selected_npn_protocol().unwrap());
-}
-
-/// Tests that when the `SslStream` is created as a server stream, the protocols
-/// are correctly advertised to the client.
-#[test]
-#[cfg(feature = "npn")]
-fn test_npn_server_advertise_multiple() {
- let listener = TcpListener::bind(next_addr()).unwrap();
- let localhost = listener.local_addr().unwrap();
- // We create a different context instance for the server...
- let listener_ctx = {
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_npn_protocols(&[b"http/1.1", b"spdy/3.1"]);
- assert!(ctx.set_certificate_file(
- &Path::new("test/cert.pem"), X509FileType::PEM).is_ok());
- ctx.set_private_key_file(
- &Path::new("test/key.pem"), X509FileType::PEM).unwrap();
- ctx
- };
- // Have the listener wait on the connection in a different thread.
- thread::spawn(move || {
- let (stream, _) = listener.accept().unwrap();
- let _ = SslStream::accept(&listener_ctx, stream).unwrap();
- });
-
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_npn_protocols(&[b"spdy/3.1"]);
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- // Now connect to the socket and make sure the protocol negotiation works...
- let stream = TcpStream::connect(localhost).unwrap();
- let stream = match SslStream::connect_generic(&ctx, stream) {
- Ok(stream) => stream,
- Err(err) => panic!("Expected success, got {:?}", err)
- };
- // SPDY is selected since that's the only thing the client supports.
- assert_eq!(b"spdy/3.1", stream.get_selected_npn_protocol().unwrap());
-}
-
-/// Tests that when the `SslStream` is created as a server stream, the protocols
-/// are correctly advertised to the client.
-#[test]
-#[cfg(feature = "alpn")]
-fn test_alpn_server_advertise_multiple() {
- let listener = TcpListener::bind(next_addr()).unwrap();
- let localhost = listener.local_addr().unwrap();
- // We create a different context instance for the server...
- let listener_ctx = {
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_alpn_protocols(&[b"http/1.1", b"spdy/3.1"]);
- assert!(ctx.set_certificate_file(
- &Path::new("test/cert.pem"), X509FileType::PEM).is_ok());
- ctx.set_private_key_file(
- &Path::new("test/key.pem"), X509FileType::PEM).unwrap();
- ctx
- };
- // Have the listener wait on the connection in a different thread.
- thread::spawn(move || {
- let (stream, _) = listener.accept().unwrap();
- let _ = SslStream::accept(&listener_ctx, stream).unwrap();
- });
-
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_alpn_protocols(&[b"spdy/3.1"]);
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- // Now connect to the socket and make sure the protocol negotiation works...
- let stream = TcpStream::connect(localhost).unwrap();
- let stream = match SslStream::new(&ctx, stream) {
- Ok(stream) => stream,
- Err(err) => panic!("Expected success, got {:?}", err)
- };
- // SPDY is selected since that's the only thing the client supports.
- assert_eq!(b"spdy/3.1", stream.get_selected_alpn_protocol().unwrap());
-}
-
-/// Test that Servers supporting ALPN don't report a protocol when none of their protocols match
-/// the client's reported protocol.
-#[test]
-#[cfg(feature = "alpn")]
-fn test_alpn_server_select_none() {
- let listener = TcpListener::bind(next_addr()).unwrap();
- let localhost = listener.local_addr().unwrap();
- // We create a different context instance for the server...
- let listener_ctx = {
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_alpn_protocols(&[b"http/1.1", b"spdy/3.1"]);
- assert!(ctx.set_certificate_file(
- &Path::new("test/cert.pem"), X509FileType::PEM).is_ok());
- ctx.set_private_key_file(
- &Path::new("test/key.pem"), X509FileType::PEM).unwrap();
- ctx
- };
- // Have the listener wait on the connection in a different thread.
- thread::spawn(move || {
- let (stream, _) = listener.accept().unwrap();
- let _ = SslStream::accept(&listener_ctx, stream).unwrap();
- });
-
- let mut ctx = SslContext::new(Sslv23).unwrap();
- ctx.set_verify(SSL_VERIFY_PEER, None);
- ctx.set_alpn_protocols(&[b"http/2"]);
- match ctx.set_CA_file(&Path::new("test/cert.pem")) {
- Ok(_) => {}
- Err(err) => panic!("Unexpected error {:?}", err)
- }
- // Now connect to the socket and make sure the protocol negotiation works...
- let stream = TcpStream::connect(localhost).unwrap();
- let stream = match SslStream::new(&ctx, stream) {
- Ok(stream) => stream,
- Err(err) => panic!("Expected success, got {:?}", err)
- };
-
- // Since the protocols from the server and client don't overlap at all, no protocol is selected
- assert_eq!(None, stream.get_selected_alpn_protocol());
-}
-
-
-#[cfg(feature="dtlsv1")]
-#[cfg(test)]
-mod dtlsv1 {
- use serialize::hex::FromHex;
- use std::net::TcpStream;
- use std::thread;
-
- use crypto::hash::Type::{SHA256};
- use ssl::SslMethod;
- use ssl::SslMethod::Dtlsv1;
- use ssl::{SslContext, SslStream, VerifyCallback};
- use ssl::SSL_VERIFY_PEER;
- use x509::{X509StoreContext};
-
- const PROTOCOL:SslMethod = Dtlsv1;
-
- #[test]
- fn test_new_ctx() {
- SslContext::new(PROTOCOL).unwrap();
- }
-}
-
-#[test]
-#[cfg(feature = "dtlsv1")]
-fn test_read_dtlsv1() {
- let (_s, stream) = Server::new_dtlsv1(Some("hello"));
-
- let mut stream = SslStream::connect_generic(&SslContext::new(Dtlsv1).unwrap(), stream).unwrap();
- let mut buf = [0u8;100];
- assert!(stream.read(&mut buf).is_ok());
-}
-
-#[test]
-#[cfg(feature = "sslv2")]
-fn test_sslv2_connect_failure() {
- let (_s, tcp) = Server::new_tcp(&["-no_ssl2", "-www"]);
- SslStream::connect_generic(&SslContext::new(Sslv2).unwrap(), tcp).err().unwrap();
-}
-
-#[cfg(target_os = "linux")]
-mod nonblocking_tests {
- extern crate nix;
-
- use std::io::Write;
- use std::net::TcpStream;
- use std::os::unix::io::AsRawFd;
-
- use super::Server;
- use self::nix::sys::epoll;
- use self::nix::fcntl;
- use ssl;
- use ssl::error::NonblockingSslError;
- use ssl::SslMethod;
- use ssl::SslMethod::Sslv23;
- use ssl::{SslContext, NonblockingSslStream};
-
- fn wait_io(stream: &NonblockingSslStream<TcpStream>, read: bool, timeout_ms: isize) -> bool {
- let fd = stream.as_raw_fd();
- let ep = epoll::epoll_create().unwrap();
- let event = if read {
- epoll::EpollEvent {
- events: epoll::EPOLLIN | epoll::EPOLLERR,
- data: 0,
- }
- } else {
- epoll::EpollEvent {
- events: epoll::EPOLLOUT,
- data: 0,
- }
- };
- epoll::epoll_ctl(ep, epoll::EpollOp::EpollCtlAdd, fd, &event).unwrap();
- let mut events = [event];
- let count = epoll::epoll_wait(ep, &mut events, timeout_ms).unwrap();
- epoll::epoll_ctl(ep, epoll::EpollOp::EpollCtlDel, fd, &event).unwrap();
- assert!(count <= 1);
- count == 1
- }
-
- fn make_nonblocking(stream: &TcpStream) {
- let fd = stream.as_raw_fd();
- fcntl::fcntl(fd, fcntl::FcntlArg::F_SETFL(fcntl::O_NONBLOCK)).unwrap();
- }
-
- #[test]
- fn test_write_nonblocking() {
- let (_s, stream) = Server::new();
- make_nonblocking(&stream);
- let mut stream = NonblockingSslStream::connect(&SslContext::new(Sslv23).unwrap(), stream).unwrap();
-
- let mut iterations = 0;
- loop {
- iterations += 1;
- if iterations > 7 {
- // Probably a safe assumption for the foreseeable future of openssl.
- panic!("Too many read/write round trips in handshake!!");
- }
- let result = stream.write("hello".as_bytes());
- match result {
- Ok(_) => {
- break;
- },
- Err(NonblockingSslError::WantRead) => {
- assert!(wait_io(&stream, true, 1000));
- },
- Err(NonblockingSslError::WantWrite) => {
- assert!(wait_io(&stream, false, 1000));
- },
- Err(other) => {
- panic!("Unexpected SSL Error: {:?}", other);
- },
- }
- }
-
- // Second write should succeed immediately--plenty of space in kernel buffer,
- // and handshake just completed.
- stream.write(" there".as_bytes()).unwrap();
- }
-
- #[test]
- fn test_read_nonblocking() {
- let (_s, stream) = Server::new();
- make_nonblocking(&stream);
- let mut stream = NonblockingSslStream::connect(&SslContext::new(Sslv23).unwrap(), stream).unwrap();
-
- let mut iterations = 0;
- loop {
- iterations += 1;
- if iterations > 7 {
- // Probably a safe assumption for the foreseeable future of openssl.
- panic!("Too many read/write round trips in handshake!!");
- }
- let result = stream.write("GET /\r\n\r\n".as_bytes());
- match result {
- Ok(n) => {
- assert_eq!(n, 9);
- break;
- },
- Err(NonblockingSslError::WantRead) => {
- assert!(wait_io(&stream, true, 1000));
- },
- Err(NonblockingSslError::WantWrite) => {
- assert!(wait_io(&stream, false, 1000));
- },
- Err(other) => {
- panic!("Unexpected SSL Error: {:?}", other);
- },
- }
- }
- let mut input_buffer = [0u8; 1500];
- let result = stream.read(&mut input_buffer);
- let bytes_read = match result {
- Ok(n) => {
- // This branch is unlikely, but on an overloaded VM with
- // unlucky context switching, the response could actually
- // be in the receive buffer before we issue the read() syscall...
- n
- },
- Err(NonblockingSslError::WantRead) => {
- assert!(wait_io(&stream, true, 3000));
- // Second read should return application data.
- stream.read(&mut input_buffer).unwrap()
- },
- Err(other) => {
- panic!("Unexpected SSL Error: {:?}", other);
- },
- };
- assert!(bytes_read >= 5);
- assert_eq!(&input_buffer[..5], b"HTTP/");
- }
-}