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