maride/goknockr
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update whitelist time permanently while connected

Browse Source
This commit is contained in:
maride 2017-09-28 23:41:47 +02:00
parent f1ec646068
commit 1531e2c16b
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
knockr.go
View File

@ -47,6 +47,11 @@ func stats() {
} }
} }
func get_address_from_conn(c net.Conn) string {
host, _, _ := net.SplitHostPort(c.RemoteAddr().String())
return host
}
func listener(port int, listen_func func(c net.Conn)) { func listener(port int, listen_func func(c net.Conn)) {
// Set up listening sockets on specified port and hand over to specified listen_func // Set up listening sockets on specified port and hand over to specified listen_func
ln, err := net.Listen("tcp", fmt.Sprintf(":%d", port)) ln, err := net.Listen("tcp", fmt.Sprintf(":%d", port))
@ -69,7 +74,7 @@ func listener(port int, listen_func func(c net.Conn)) {
func whitelist_handler(c net.Conn) { func whitelist_handler(c net.Conn) {
// Handler function for whitelist socket connections, whitelisting the connecting host // Handler function for whitelist socket connections, whitelisting the connecting host
host, _, _ := net.SplitHostPort(c.RemoteAddr().String()) host := get_address_from_conn(c)
if is_blacklisted(host) { if is_blacklisted(host) {
if arguments.Verbose { if arguments.Verbose {
@ -85,7 +90,7 @@ func whitelist_handler(c net.Conn) {
func blacklist_handler(c net.Conn) { func blacklist_handler(c net.Conn) {
// Handler which blocks every host connecting to it. // Handler which blocks every host connecting to it.
// Useful to place it on port (whitelistPort-1) to crash port scanners. // Useful to place it on port (whitelistPort-1) to crash port scanners.
host, _, _ := net.SplitHostPort(c.RemoteAddr().String()) host := get_address_from_conn(c)
if ! is_whitelisted(host) { if ! is_whitelisted(host) {
if arguments.Verbose { if arguments.Verbose {
@ -102,7 +107,7 @@ func blacklist_handler(c net.Conn) {
func gateway_handler(c net.Conn) { func gateway_handler(c net.Conn) {
// Filter connections whether or not the connecting host is whitelisted // Filter connections whether or not the connecting host is whitelisted
host, _, _ := net.SplitHostPort(c.RemoteAddr().String()) host := get_address_from_conn(c)
if is_blacklisted(host) { if is_blacklisted(host) {
if arguments.Verbose { if arguments.Verbose {
@ -188,17 +193,19 @@ func proxy(c net.Conn) {
fmt.Println("[ERR] Proxy connection to server failed") fmt.Println("[ERR] Proxy connection to server failed")
fmt.Println(" Error is ", err) fmt.Println(" Error is ", err)
} else { } else {
go proxy_writefunc(c, ln, &traffic_in) host := get_address_from_conn(c)
proxy_writefunc(ln, c, &traffic_out) go proxy_writefunc(c, ln, &traffic_in, host)
proxy_writefunc(ln, c, &traffic_out, host)
} }
} }
func proxy_writefunc(a net.Conn, b net.Conn, written_bytes *int64) { func proxy_writefunc(a net.Conn, b net.Conn, written_bytes *int64, addr string) {
var delta int64 = 0 var delta int64 = 0
var err error var err error
for err == nil { for err == nil {
delta, err = io.CopyN(a, b, 1024) delta, err = io.CopyN(a, b, 1024)
*written_bytes += delta *written_bytes += delta
update_whitelist_time(addr)
} }
} }