add more shrimple
This commit is contained in:
parent
63dfd9871c
commit
3d94e01522
3 changed files with 13 additions and 2 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -1 +1,2 @@
|
|||
/target
|
||||
.direnv
|
||||
|
|
|
|||
|
|
@ -1,4 +1,5 @@
|
|||
use std::{
|
||||
env,
|
||||
io::{Read, Write},
|
||||
net::{TcpListener, UdpSocket},
|
||||
thread,
|
||||
|
|
@ -15,7 +16,11 @@ impl Ports {
|
|||
let mut tcp = tcp_listener.accept().unwrap().0;
|
||||
thread::spawn(move || {
|
||||
let udp = UdpSocket::bind(("127.0.0.1", 0)).unwrap();
|
||||
udp.connect(("127.0.0.1", self.udp)).unwrap();
|
||||
udp.connect((
|
||||
env::var("TCPUDP_OTHER").unwrap_or("127.0.0.1".to_owned()),
|
||||
self.udp,
|
||||
))
|
||||
.unwrap();
|
||||
{
|
||||
let (mut tcp, udp) = (tcp.try_clone().unwrap(), udp.try_clone().unwrap());
|
||||
thread::spawn(move || {
|
||||
|
|
|
|||
|
|
@ -1,5 +1,6 @@
|
|||
use std::{
|
||||
collections::HashMap,
|
||||
env,
|
||||
io::{Read, Write},
|
||||
net::{Shutdown, SocketAddr, TcpStream, UdpSocket},
|
||||
thread,
|
||||
|
|
@ -18,7 +19,11 @@ impl Ports {
|
|||
let (len, addr) = udp_listener.recv_from(&mut buf).unwrap();
|
||||
let mut tcp_socket = map.get(&addr);
|
||||
if tcp_socket.is_none() {
|
||||
let tcp = TcpStream::connect(("127.0.0.1", self.tcp)).unwrap();
|
||||
let tcp = TcpStream::connect((
|
||||
env::var("TCPUDP_OTHER").unwrap_or("127.0.0.1".to_owned()),
|
||||
self.tcp,
|
||||
))
|
||||
.unwrap();
|
||||
{
|
||||
let (mut tcp, addr, udp) = (
|
||||
tcp.try_clone().unwrap(),
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue