go-gost/x
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add plugin system

Browse Source
This commit is contained in:
ginuerzh
2023-04-18 20:52:56 +08:00
parent 7576651a67
commit 32c8188351
48 changed files with 761 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
admission/wrapper/conn.go
View File

@ -1,6 +1,7 @@
package wrapper
import (
"context"
"errors"
"io"
"net"
@ -33,7 +34,7 @@ func WrapConn(admission admission.Admission, c net.Conn) net.Conn {
func (c *serverConn) Read(b []byte) (n int, err error) {
if c.admission != nil &&
!c.admission.Admit(c.RemoteAddr().String()) {
!c.admission.Admit(context.Background(), c.RemoteAddr().String()) {
err = io.EOF
return
}
@ -79,7 +80,7 @@ func (c *packetConn) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
}
if c.admission != nil &&
!c.admission.Admit(addr.String()) {
!c.admission.Admit(context.Background(), addr.String()) {
continue
}
@ -143,7 +144,7 @@ func (c *udpConn) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
return
}
if c.admission != nil &&
!c.admission.Admit(addr.String()) {
!c.admission.Admit(context.Background(), addr.String()) {
continue
}
return
@ -158,7 +159,7 @@ func (c *udpConn) ReadFromUDP(b []byte) (n int, addr *net.UDPAddr, err error) {
return
}
if c.admission != nil &&
!c.admission.Admit(addr.String()) {
!c.admission.Admit(context.Background(), addr.String()) {
continue
}
return
@ -176,7 +177,7 @@ func (c *udpConn) ReadMsgUDP(b, oob []byte) (n, oobn, flags int, addr *net.UDPAd
return
}
if c.admission != nil &&
!c.admission.Admit(addr.String()) {
!c.admission.Admit(context.Background(), addr.String()) {
continue
}
return

3
admission/wrapper/listener.go
View File

@ -1,6 +1,7 @@
package wrapper
import (
"context"
"net"
"github.com/go-gost/core/admission"
@ -28,7 +29,7 @@ func (ln *listener) Accept() (net.Conn, error) {
return nil, err
}
if ln.admission != nil &&
!ln.admission.Admit(c.RemoteAddr().String()) {
!ln.admission.Admit(context.Background(), c.RemoteAddr().String()) {
c.Close()
continue
}