github/vtun
5
0
Fork 0
mirror of https://github.com/net-byte/vtun synced 2024-03-14 10:50:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

clean code

Browse Source
This commit is contained in:
netbyte 2022-05-22 00:43:04 +08:00
parent 3724f67db0
commit 91e778a7c5
3 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
quic/quicserver.go
View File

@ -3,7 +3,6 @@ package quic
import (
"context"
"crypto/tls"
"io"
"log"
"time"
@ -50,7 +49,7 @@ func toClient(config config.Config, iface *water.Interface) {
packet := make([]byte, config.MTU)
for {
n, err := iface.Read(packet)
if err != nil || err == io.EOF || n == 0 {
if err != nil || n == 0 {
continue
}
b := packet[:n]
@ -59,10 +58,7 @@ func toClient(config config.Config, iface *water.Interface) {
if config.Obfs {
b = cipher.XOR(b)
}
_, err := v.(quic.Stream).Write(b)
if err != nil {
cache.GetCache().Delete(key)
}
v.(quic.Stream).Write(b)
}
}
}
@ -79,7 +75,7 @@ func toServer(config config.Config, session quic.Session, iface *water.Interface
for {
stream.SetReadDeadline(time.Now().Add(time.Duration(config.Timeout) * time.Second))
n, err := stream.Read(packet)
if err != nil || err == io.EOF {
if err != nil {
break
}
b := packet[:n]

3
tcp/tcpclient.go
View File

@ -1,7 +1,6 @@
package tcp
import (
"io"
"log"
"net"
"time"
@ -59,7 +58,7 @@ func tcpToTun(config config.Config, tcpconn net.Conn, iface *water.Interface) {
for {
tcpconn.SetReadDeadline(time.Now().Add(time.Duration(config.Timeout) * time.Second))
n, err := tcpconn.Read(packet)
if err != nil || err == io.EOF {
if err != nil {
break
}
b := packet[:n]

5
tcp/tcpserver.go
View File

@ -1,7 +1,6 @@
package tcp
import (
"io"
"log"
"net"
"time"
@ -40,7 +39,7 @@ func toClient(config config.Config, iface *water.Interface) {
packet := make([]byte, config.MTU)
for {
n, err := iface.Read(packet)
if err != nil || err == io.EOF || n == 0 {
if err != nil || n == 0 {
continue
}
b := packet[:n]
@ -61,7 +60,7 @@ func toServer(config config.Config, tcpconn net.Conn, iface *water.Interface) {
for {
tcpconn.SetReadDeadline(time.Now().Add(time.Duration(config.Timeout) * time.Second))
n, err := tcpconn.Read(packet)
if err != nil || err == io.EOF {
if err != nil {
break
}
b := packet[:n]