src/test/ui/tcp-stress.rs

   1 // run-pass
   2 // ignore-android needs extra network permissions
   3 // ignore-cloudabi no global network namespace access
   4 // ignore-emscripten no threads or sockets support
   5 // ignore-netbsd system ulimit (Too many open files)
   6 // ignore-openbsd system ulimit (Too many open files)
   7 // ignore-sgx no thread sleep support
   8
   9 use std::io::prelude::*;
  10 use std::net::{TcpListener, TcpStream};
  11 use std::process;
  12 use std::sync::mpsc::channel;
  13 use std::time::Duration;
  14 use std::thread::{self, Builder};
  15
  16 const TARGET_CNT: usize = 200;
  17
  18 fn main() {
  19     // This test has a chance to time out, try to not let it time out
  20     thread::spawn(move|| -> () {
  21         thread::sleep(Duration::from_secs(30));
  22         process::exit(1);
  23     });
  24
  25     let listener = TcpListener::bind("127.0.0.1:0").unwrap();
  26     let addr = listener.local_addr().unwrap();
  27     thread::spawn(move || -> () {
  28         loop {
  29             let mut stream = match listener.accept() {
  30                 Ok(stream) => stream.0,
  31                 Err(_) => continue,
  32             };
  33             let _ = stream.read(&mut [0]);
  34             let _ = stream.write(&[2]);
  35         }
  36     });
  37
  38     let (tx, rx) = channel();
  39
  40     let mut spawned_cnt = 0;
  41     for _ in 0..TARGET_CNT {
  42         let tx = tx.clone();
  43         let res = Builder::new().stack_size(64 * 1024).spawn(move|| {
  44             match TcpStream::connect(addr) {
  45                 Ok(mut stream) => {
  46                     let _ = stream.write(&[1]);
  47                     let _ = stream.read(&mut [0]);
  48                 },
  49                 Err(..) => {}
  50             }
  51             tx.send(()).unwrap();
  52         });
  53         if let Ok(_) = res {
  54             spawned_cnt += 1;
  55         };
  56     }
  57
  58     // Wait for all clients to exit, but don't wait for the server to exit. The
  59     // server just runs infinitely.
  60     drop(tx);
  61     for _ in 0..spawned_cnt {
  62         rx.recv().unwrap();
  63     }
  64     assert_eq!(spawned_cnt, TARGET_CNT);
  65     process::exit(0);
  66 }