mirrors/SpoofDPI
1
0
Fork 0
mirror of https://github.com/xvzc/SpoofDPI.git synced 2024-12-22 14:26:31 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix local address

Browse Source
This commit is contained in:
jerry901 2022-05-02 00:31:22 +09:00
parent dd2e9a1c55
commit 741b4bc235
2 changed files with 35 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
net/conn.go
View File

@ -40,6 +40,16 @@ func (c *Conn) Write(b []byte) (n int, err error) {
return c.conn.Write(b)
}
func (c *Conn) SetDeadLine(t time.Time) (error) {
c.conn.SetDeadline(t)
return nil
}
func (c *Conn) SetKeepAlive(b bool) (error) {
c.conn.(*net.TCPConn).SetKeepAlive(b)
return nil
}
func (conn *Conn) WriteChunks(c [][]byte) (n int, err error) {
total := 0
for i := 0; i < len(c); i++ {
@ -87,9 +97,15 @@ func (lConn *Conn) HandleHttp(p packet.HttpPacket) {
ip, err := doh.Lookup(p.Domain())
if err != nil {
log.Debug("[DOH] Error looking up for domain with ", p.Domain() , err)
// log.Error("[HTTP DOH] Error looking up for domain with ", p.Domain() , " ", err)
log.Error(lConn.RemoteAddr().String())
// log.Error(string(p.Raw()))
lConn.Write([]byte(p.Version() + " 502 Bad Gateway\r\n\r\n"))
lConn.Close()
return
}
log.Info(string(lConn.RemoteAddr().String()))
log.Debug("[DOH] Found ", ip, " with ", p.Domain())
@ -102,6 +118,7 @@ func (lConn *Conn) HandleHttp(p packet.HttpPacket) {
rConn, err := Dial("tcp", ip + port)
if err != nil {
log.Debug("[HTTPS] ", err)
lConn.Close()
return
}
@ -113,6 +130,8 @@ func (lConn *Conn) HandleHttp(p packet.HttpPacket) {
_, err = rConn.Write(p.Raw())
if err != nil {
log.Debug("[HTTP] Error sending request to ", p.Domain(), err)
lConn.Close()
rConn.Close()
return
}
@ -124,7 +143,9 @@ func (lConn *Conn) HandleHttps(p packet.HttpPacket) {
ip, err := doh.Lookup(p.Domain())
if err != nil {
log.Debug("[DOH] Error looking up for domain: ", p.Domain(), " ", err)
log.Error("[HTTPS DOH] Error looking up for domain: ", p.Domain(), " ", err)
lConn.Write([]byte(p.Version() + " 502 Bad Gateway\r\n\r\n"))
lConn.Close()
return
}
@ -139,6 +160,7 @@ func (lConn *Conn) HandleHttps(p packet.HttpPacket) {
rConn, err := Dial("tcp", ip + port)
if err != nil {
log.Debug("[HTTPS] ", err)
lConn.Close()
return
}
@ -147,6 +169,8 @@ func (lConn *Conn) HandleHttps(p packet.HttpPacket) {
_, err = lConn.Write([]byte(p.Version() + " 200 Connection Established\r\n\r\n"))
if err != nil {
log.Debug("[HTTPS] Error sending 200 Connection Established to the client", err)
lConn.Close()
rConn.Close()
return
}
log.Debug("[HTTPS] Sent 200 Connection Estabalished to the client")
@ -156,6 +180,8 @@ func (lConn *Conn) HandleHttps(p packet.HttpPacket) {
if err != nil {
log.Debug("[HTTPS] Error reading client hello from the client", err)
log.Debug("[HTTPS] Closing local connection..")
lConn.Close()
rConn.Close()
return
}
@ -171,6 +197,8 @@ func (lConn *Conn) HandleHttps(p packet.HttpPacket) {
if _, err := rConn.WriteChunks(chunks); err != nil {
log.Debug("[HTTPS] Error writing client hello to ", p.Domain(), err)
lConn.Close()
rConn.Close()
return
}
@ -178,7 +206,7 @@ func (lConn *Conn) HandleHttps(p packet.HttpPacket) {
func (from *Conn) Serve(to *Conn, proto string, fd string, td string) {
defer from.Close()
defer to.CloseWrite()
defer to.Close()
proto += " "

5
proxy/proxy.go
View File

@ -2,6 +2,7 @@ package proxy
import (
"os"
"time"
log "github.com/sirupsen/logrus"
"github.com/xvzc/SpoofDPI/net"
@ -23,7 +24,7 @@ func (p *Proxy) Port() string {
}
func (p *Proxy) Start() {
l, err := net.Listen("tcp", ":"+p.Port())
l, err := net.Listen("tcp", "127.0.0.1:"+p.Port())
if err != nil {
log.Fatal("Error creating listener: ", err)
os.Exit(1)
@ -37,6 +38,8 @@ func (p *Proxy) Start() {
log.Fatal("Error accepting connection: ", err)
continue
}
conn.SetDeadLine(time.Now().Add(3 * time.Second))
conn.SetKeepAlive(false)
log.Debug("[PROXY] Accepted a new connection from ", conn.RemoteAddr())