Add whitelist/blacklist support for relay

e16ac5a5 · Hubix9 · ginuerzh · 4712d6c9 · e16ac5a5
Commit e16ac5a5 authored Feb 05, 2021 by Hubix9 Committed by ginuerzh Feb 06, 2021
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

relay.go relay.go +7 -0

No files found.
--- a/relay.go
+++ b/relay.go
@@ -203,6 +203,13 @@ func (h *relayHandler) Handle(conn net.Conn) {
 	if udp {
 		network = "udp"
 	}
+	if !Can(network, raddr, h.options.Whitelist, h.options.Blacklist) {
+		resp.Status = relay.StatusForbidden
+		resp.WriteTo(conn)
+		log.Logf("[relay] %s -> %s : relay to %s is forbidden",
+			conn.RemoteAddr(), conn.LocalAddr(), raddr)
+		return
+	}

 	ctx := context.TODO()
 	var cc net.Conn