add more shrimple

This commit is contained in:
Tove 2026-04-14 20:22:49 +02:00
parent 63dfd9871c
commit 3d94e01522
3 changed files with 13 additions and 2 deletions

1
.gitignore vendored
View file

@ -1 +1,2 @@
/target
.direnv

View file

@ -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 || {

View file

@ -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(),