src/test/ui/tcp-stress.rs

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