source/shadowsocks-rust
1
0
Fork 0
mirror of https://github.com/shadowsocks/shadowsocks-rust.git synced 2026-02-09 01:59:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

tests based on HTTP should read status line only

Browse Source
This commit is contained in:
zonyitoo
2021-04-28 15:59:11 +08:00
parent 04bddab6c0
commit 10f8e30b92
4 changed files with 24 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/http.rs
View File

@@ -3,7 +3,7 @@
use std::time::Duration;
use tokio::{
io::{AsyncReadExt, AsyncWriteExt},
io::{AsyncBufReadExt, AsyncReadExt, AsyncWriteExt, BufReader},
net::TcpStream,
time,
};
@@ -59,6 +59,7 @@ async fn http_proxy() {
.unwrap();
c.flush().await.unwrap();
// Proxy should close connection actively because HTTP/1.0 use short connection by default
let mut buf = Vec::new();
c.read_to_end(&mut buf).await.unwrap();
@@ -77,8 +78,10 @@ async fn http_proxy() {
.unwrap();
c.flush().await.unwrap();
let mut r = BufReader::new(c);
let mut buf = Vec::new();
c.read_to_end(&mut buf).await.unwrap();
r.read_until(b'\n', &mut buf).await.unwrap();
assert!(buf.starts_with(b"HTTP/1.0 200 OK\r\n"));
}

10
tests/socks4.rs
View File

@@ -6,7 +6,7 @@ use std::{
};
use tokio::{
io::{AsyncReadExt, AsyncWriteExt},
io::{AsyncBufReadExt, AsyncWriteExt, BufReader},
time::{self, Duration},
};
@@ -89,11 +89,11 @@ async fn socks4_relay_connect() {
c.write_all(HTTP_REQUEST).await.unwrap();
c.flush().await.unwrap();
let mut buf = Vec::new();
c.read_to_end(&mut buf).await.unwrap();
let mut r = BufReader::new(c);
println!("Got reply from server: {}", str::from_utf8(&buf).unwrap());
let mut buf = Vec::new();
r.read_until(b'\n', &mut buf).await.unwrap();
let http_status = b"HTTP/1.0 200 OK\r\n";
buf.starts_with(http_status);
assert!(buf.starts_with(http_status));
}

18
tests/socks5.rs
View File

@@ -6,7 +6,7 @@ use std::{
};
use tokio::{
io::{AsyncReadExt, AsyncWriteExt},
io::{AsyncBufReadExt, AsyncWriteExt, BufReader},
time::{self, Duration},
};
@@ -95,13 +95,13 @@ async fn socks5_relay_stream() {
c.write_all(req).await.unwrap();
c.flush().await.unwrap();
let mut buf = Vec::new();
c.read_to_end(&mut buf).await.unwrap();
let mut r = BufReader::new(c);
println!("Got reply from server: {}", str::from_utf8(&buf).unwrap());
let mut buf = Vec::new();
r.read_until(b'\n', &mut buf).await.unwrap();
let http_status = b"HTTP/1.0 200 OK\r\n";
buf.starts_with(http_status);
assert!(buf.starts_with(http_status));
}
#[tokio::test]
@@ -128,11 +128,11 @@ async fn socks5_relay_aead() {
c.write_all(req).await.unwrap();
c.flush().await.unwrap();
let mut buf = Vec::new();
c.read_to_end(&mut buf).await.unwrap();
let mut r = BufReader::new(c);
println!("Got reply from server: {}", str::from_utf8(&buf).unwrap());
let mut buf = Vec::new();
r.read_until(b'\n', &mut buf).await.unwrap();
let http_status = b"HTTP/1.0 200 OK\r\n";
buf.starts_with(http_status);
assert!(buf.starts_with(http_status));
}

12
tests/tunnel.rs
View File

@@ -1,12 +1,10 @@
#![cfg(all(feature = "local-tunnel", feature = "server"))]
use std::str;
use byte_string::ByteStr;
use log::debug;
use tokio::{
self,
io::{AsyncReadExt, AsyncWriteExt},
io::{AsyncBufReadExt, AsyncWriteExt, BufReader},
net::{TcpStream, UdpSocket},
time::{self, Duration},
};
@@ -64,13 +62,13 @@ async fn tcp_tunnel() {
stream.write_all(req).await.unwrap();
stream.flush().await.unwrap();
let mut buf = Vec::new();
stream.read_to_end(&mut buf).await.unwrap();
let mut r = BufReader::new(stream);
println!("Got reply from server: {}", str::from_utf8(&buf).unwrap());
let mut buf = Vec::new();
r.read_until(b'\n', &mut buf).await.unwrap();
let http_status = b"HTTP/1.0 200 OK\r\n";
buf.starts_with(http_status);
assert!(buf.starts_with(http_status));
}
#[tokio::test]