mirror of
https://github.com/xvzc/SpoofDPI.git
synced 2024-12-22 14:26:31 +00:00
add net module and fixed the other modules
This commit is contained in:
parent
00b22acc25
commit
452a7e343f
@ -1,13 +1,42 @@
|
||||
package proxy
|
||||
package net
|
||||
|
||||
import (
|
||||
"net"
|
||||
// "github.com/xvzc/SpoofDPI/util"
|
||||
)
|
||||
|
||||
const BUF_SIZE = 1024
|
||||
|
||||
func ReadBytes(conn net.Conn) ([]byte, error) {
|
||||
type Conn struct {
|
||||
Conn net.Conn
|
||||
}
|
||||
|
||||
func (conn *Conn) Close() {
|
||||
conn.Conn.Close()
|
||||
}
|
||||
|
||||
func (conn *Conn) Read(b []byte) (n int, err error) {
|
||||
return conn.Conn.Read(b)
|
||||
}
|
||||
|
||||
func (conn *Conn) Write(b []byte) (n int, err error) {
|
||||
return conn.Conn.Write(b)
|
||||
}
|
||||
|
||||
func (conn *Conn) WriteChunks(c [][]byte) (n int, err error) {
|
||||
total := 0
|
||||
for i := 0; i < len(c); i++ {
|
||||
b, err := conn.Write(c[i])
|
||||
if err != nil {
|
||||
return 0, nil
|
||||
}
|
||||
|
||||
b += total
|
||||
}
|
||||
|
||||
return total, nil
|
||||
}
|
||||
|
||||
func (conn *Conn) ReadBytes() ([]byte, error) {
|
||||
ret := make([]byte, 0)
|
||||
buf := make([]byte, BUF_SIZE)
|
||||
|
||||
@ -26,9 +55,9 @@ func ReadBytes(conn net.Conn) ([]byte, error) {
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
func Serve(from net.Conn, to net.Conn, proto string) {
|
||||
func (from *Conn) Serve(to Conn, proto string) {
|
||||
for {
|
||||
buf, err := ReadBytes(from)
|
||||
buf, err := from.ReadBytes()
|
||||
if err != nil {
|
||||
// util.Debug("["+proto+"]"+"Error reading from ", from.RemoteAddr())
|
||||
// util.Debug(err, " Closing the connection.. ")
|
23
net/dial.go
Normal file
23
net/dial.go
Normal file
@ -0,0 +1,23 @@
|
||||
package net
|
||||
|
||||
import (
|
||||
"net"
|
||||
)
|
||||
|
||||
func Listen(network, address string) (Listener, error) {
|
||||
l, err := net.Listen(network, address)
|
||||
if err != nil {
|
||||
return Listener{}, err
|
||||
}
|
||||
|
||||
return Listener{Listener: l}, nil
|
||||
}
|
||||
|
||||
func Dial(network, address string) (Conn, error) {
|
||||
conn, err := net.Dial(network, address)
|
||||
if err != nil {
|
||||
return Conn{}, err
|
||||
}
|
||||
|
||||
return Conn{Conn: conn}, nil
|
||||
}
|
18
net/listener.go
Normal file
18
net/listener.go
Normal file
@ -0,0 +1,18 @@
|
||||
package net
|
||||
|
||||
import (
|
||||
"net"
|
||||
)
|
||||
|
||||
type Listener struct {
|
||||
Listener net.Listener
|
||||
}
|
||||
|
||||
func (l *Listener) Accept() (Conn, error) {
|
||||
conn, err := l.Listener.Accept()
|
||||
if err != nil {
|
||||
return Conn{}, err
|
||||
}
|
||||
|
||||
return Conn{Conn: conn}, nil
|
||||
}
|
1
net/net.go
Normal file
1
net/net.go
Normal file
@ -0,0 +1 @@
|
||||
package net
|
@ -1,27 +1,3 @@
|
||||
package proxy
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
|
||||
"github.com/xvzc/SpoofDPI/packet"
|
||||
// "github.com/xvzc/SpoofDPI/util"
|
||||
)
|
||||
|
||||
func HandleHttp(clientConn net.Conn, ip string, p *packet.HttpPacket) {
|
||||
remoteConn, err := net.Dial("tcp", ip+":80") // create connection to server
|
||||
if err != nil {
|
||||
// util.Debug(err)
|
||||
return
|
||||
}
|
||||
defer remoteConn.Close()
|
||||
|
||||
// util.Debug("[HTTP] Connected to the server.")
|
||||
|
||||
go Serve(remoteConn, clientConn, "HTTP")
|
||||
|
||||
// util.Debug("[HTTP] Sending request to the server")
|
||||
fmt.Fprintf(remoteConn, string(*p.Raw))
|
||||
|
||||
Serve(clientConn, remoteConn, "HTTP")
|
||||
}
|
||||
// "github.com/xvzc/SpoofDPI/util"
|
||||
|
@ -2,12 +2,29 @@ package proxy
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
|
||||
"github.com/xvzc/SpoofDPI/net"
|
||||
"github.com/xvzc/SpoofDPI/packet"
|
||||
"github.com/xvzc/SpoofDPI/util"
|
||||
)
|
||||
|
||||
func HandleHttp(clientConn net.Conn, ip string, p *packet.HttpPacket) {
|
||||
remoteConn, err := net.Dial("tcp", ip+":80") // create connection to server
|
||||
if err != nil {
|
||||
// util.Debug(err)
|
||||
return
|
||||
}
|
||||
defer remoteConn.Close()
|
||||
|
||||
// util.Debug("[HTTP] Connected to the server.")
|
||||
|
||||
go remoteConn.Serve(clientConn, "HTTP")
|
||||
|
||||
// util.Debug("[HTTP] Sending request to the server")
|
||||
fmt.Fprintf(remoteConn.Conn, string(*p.Raw))
|
||||
|
||||
go clientConn.Serve(remoteConn, "HTTP")
|
||||
}
|
||||
|
||||
func HandleHttps(clientConn net.Conn, ip string, r *packet.HttpPacket) {
|
||||
// Create a connection to the requested server
|
||||
remoteConn, err := net.Dial("tcp", ip+":443")
|
||||
@ -20,11 +37,11 @@ func HandleHttps(clientConn net.Conn, ip string, r *packet.HttpPacket) {
|
||||
// util.Debug("[HTTPS] Connected to the server.")
|
||||
|
||||
// Send self generated response for connect request
|
||||
fmt.Fprintf(clientConn, "HTTP/1.1 200 Connection Established\r\n\r\n")
|
||||
fmt.Fprintf(clientConn.Conn, "HTTP/1.1 200 Connection Established\r\n\r\n")
|
||||
// util.Debug("[HTTPS] Sent 200 Connection Estabalished")
|
||||
|
||||
// Read client hello
|
||||
clientHello, err := ReadBytes(clientConn)
|
||||
clientHello, err := clientConn.ReadBytes()
|
||||
if err != nil {
|
||||
// util.Debug("[HTTPS] Error reading client hello: ", err)
|
||||
// util.Debug("Closing connection ", clientConn.RemoteAddr())
|
||||
@ -33,18 +50,16 @@ func HandleHttps(clientConn net.Conn, ip string, r *packet.HttpPacket) {
|
||||
// util.Debug(clientConn.RemoteAddr(), "[HTTPS] Client sent hello", len(clientHello))
|
||||
|
||||
// Generate a go routine that reads from the server
|
||||
go Serve(remoteConn, clientConn, "HTTPS")
|
||||
go remoteConn.Serve(clientConn, "HTTPS")
|
||||
|
||||
// Send chunked request
|
||||
chunks := util.BytesToChunks(clientHello)
|
||||
for i := 0; i < len(chunks); i++ {
|
||||
_, write_err := remoteConn.Write(chunks[i])
|
||||
if write_err != nil {
|
||||
// util.Debug("[HTTPS] Error writing to the client:", write_err)
|
||||
break
|
||||
}
|
||||
pkt := packet.NewHttpsPacket(&clientHello)
|
||||
|
||||
chunks := pkt.SplitInChunks()
|
||||
|
||||
if _, err := remoteConn.WriteChunks(chunks); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// Read from the client
|
||||
Serve(clientConn, remoteConn, "HTTPS")
|
||||
clientConn.Serve(remoteConn, "HTTPS")
|
||||
}
|
||||
|
@ -2,10 +2,10 @@ package proxy
|
||||
|
||||
import (
|
||||
"log"
|
||||
"net"
|
||||
"os"
|
||||
|
||||
"github.com/xvzc/SpoofDPI/doh"
|
||||
"github.com/xvzc/SpoofDPI/net"
|
||||
"github.com/xvzc/SpoofDPI/packet"
|
||||
)
|
||||
|
||||
@ -40,7 +40,7 @@ func (p *Proxy) Start() {
|
||||
go func() {
|
||||
defer clientConn.Close()
|
||||
|
||||
b, err := ReadBytes(clientConn)
|
||||
b, err := clientConn.ReadBytes()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
@ -17,14 +17,6 @@ func ParseArgs() (string, string, bool) {
|
||||
return *port, *dns, *debug
|
||||
}
|
||||
|
||||
func BytesToChunks(buf []byte) [][]byte {
|
||||
if len(buf) < 1 {
|
||||
return [][]byte{buf}
|
||||
}
|
||||
|
||||
return [][]byte{buf[:1], buf[1:]}
|
||||
}
|
||||
|
||||
func PrintWelcome(port string, dns string, debug bool) {
|
||||
cyan := pterm.NewLettersFromStringWithStyle("Spoof", pterm.NewStyle(pterm.FgCyan))
|
||||
purple := pterm.NewLettersFromStringWithStyle("DPI", pterm.NewStyle(pterm.FgLightMagenta))
|
||||
|
Loading…
Reference in New Issue
Block a user