|
|
@@ -132,92 +132,4 @@ impl RendezvousServer {
|
|
132
|
132
|
}
|
|
133
|
133
|
Ok(())
|
|
134
|
134
|
}
|
|
135
|
|
-}
|
|
136
|
|
-
|
|
137
|
|
-#[cfg(test)]
|
|
138
|
|
-mod tests {
|
|
139
|
|
- use super::*;
|
|
140
|
|
- use hbb_common::tokio;
|
|
141
|
|
-
|
|
142
|
|
- #[allow(unused_must_use)]
|
|
143
|
|
- #[tokio::main]
|
|
144
|
|
- async fn test_rs_async() {
|
|
145
|
|
- let mut port_server: u16 = 0;
|
|
146
|
|
- let socket = FramedSocket::new("127.0.0.1:0").await.unwrap();
|
|
147
|
|
- if let SocketAddr::V4(addr) = socket.get_ref().local_addr().unwrap() {
|
|
148
|
|
- port_server = addr.port();
|
|
149
|
|
- }
|
|
150
|
|
- drop(socket);
|
|
151
|
|
- let addr_server = format!("127.0.0.1:{}", port_server);
|
|
152
|
|
- let f1 = RendezvousServer::start(&addr_server);
|
|
153
|
|
- let addr_server = addr_server.parse().unwrap();
|
|
154
|
|
- let f2 = punch_hole(addr_server);
|
|
155
|
|
- tokio::try_join!(f1, f2);
|
|
156
|
|
- }
|
|
157
|
|
-
|
|
158
|
|
- async fn punch_hole(addr_server: SocketAddr) -> ResultType<()> {
|
|
159
|
|
- // B register it to server
|
|
160
|
|
- let mut socket_b = FramedSocket::new("127.0.0.1:0").await?;
|
|
161
|
|
- let local_addr_b = socket_b.get_ref().local_addr().unwrap();
|
|
162
|
|
- let mut msg_out = RendezvousMessage::new();
|
|
163
|
|
- msg_out.set_register_peer(RegisterPeer {
|
|
164
|
|
- id: "123".to_string(),
|
|
165
|
|
- ..Default::default()
|
|
166
|
|
- });
|
|
167
|
|
- socket_b.send(&msg_out, addr_server).await?;
|
|
168
|
|
-
|
|
169
|
|
- // A send punch request to server
|
|
170
|
|
- let mut socket_a = FramedSocket::new("127.0.0.1:0").await?;
|
|
171
|
|
- let local_addr_a = socket_a.get_ref().local_addr().unwrap();
|
|
172
|
|
- msg_out.set_punch_hole_request(PunchHoleRequest {
|
|
173
|
|
- id: "123".to_string(),
|
|
174
|
|
- ..Default::default()
|
|
175
|
|
- });
|
|
176
|
|
- socket_a.send(&msg_out, addr_server).await?;
|
|
177
|
|
-
|
|
178
|
|
- println!(
|
|
179
|
|
- "A {:?} request punch hole to B {:?} via server {:?}",
|
|
180
|
|
- local_addr_a, local_addr_b, addr_server,
|
|
181
|
|
- );
|
|
182
|
|
-
|
|
183
|
|
- // on B side, responsed to A's punch request forwarded from server
|
|
184
|
|
- if let Some(Ok((bytes, addr))) = socket_b.next_timeout(1000).await {
|
|
185
|
|
- assert_eq!(addr_server, addr);
|
|
186
|
|
- let msg_in = parse_from_bytes::<RendezvousMessage>(&bytes)?;
|
|
187
|
|
- let remote_addr_a = AddrMangle::decode(&msg_in.get_punch_hole().socket_addr);
|
|
188
|
|
- assert_eq!(local_addr_a, remote_addr_a);
|
|
189
|
|
-
|
|
190
|
|
- // B punch A
|
|
191
|
|
- socket_b
|
|
192
|
|
- .get_mut()
|
|
193
|
|
- .send_to(&b"SYN"[..], &remote_addr_a)
|
|
194
|
|
- .await?;
|
|
195
|
|
-
|
|
196
|
|
- msg_out.set_punch_hole_sent(PunchHoleSent {
|
|
197
|
|
- socket_addr: AddrMangle::encode(&remote_addr_a),
|
|
198
|
|
- ..Default::default()
|
|
199
|
|
- });
|
|
200
|
|
- socket_b.send(&msg_out, addr_server).await?;
|
|
201
|
|
- } else {
|
|
202
|
|
- panic!("failed");
|
|
203
|
|
- }
|
|
204
|
|
-
|
|
205
|
|
- // on A side
|
|
206
|
|
- socket_a.next().await; // skip "SYN"
|
|
207
|
|
- if let Some(Ok((bytes, addr))) = socket_a.next_timeout(1000).await {
|
|
208
|
|
- assert_eq!(addr_server, addr);
|
|
209
|
|
- let msg_in = parse_from_bytes::<RendezvousMessage>(&bytes)?;
|
|
210
|
|
- let remote_addr_b = AddrMangle::decode(&msg_in.get_punch_hole_response().socket_addr);
|
|
211
|
|
- assert_eq!(local_addr_b, remote_addr_b);
|
|
212
|
|
- } else {
|
|
213
|
|
- panic!("failed");
|
|
214
|
|
- }
|
|
215
|
|
-
|
|
216
|
|
- Err("done".into())
|
|
217
|
|
- }
|
|
218
|
|
-
|
|
219
|
|
- #[test]
|
|
220
|
|
- fn test_rs() {
|
|
221
|
|
- self::test_rs_async();
|
|
222
|
|
- }
|
|
223
|
|
-}
|
|
|
135
|
+}
|