Add HTTP protocol tarpit

1 files changed, 5 insertions, 3 deletions

diff --git a/handle.go b/handle.go
index 918f3a8..c153b7d 100644
--- a/handle.go
+++ b/handle.go
@@ -9,12 +9,14 @@ import (
 type empty struct{}
 type protoHandler func(net.Conn, time.Duration)
 
-func protocolHandler(proto string) (protoHandler, error) {
+func protocolHandler(proto string) (protoHandler, uint16, error) {
 	switch proto {
 	case "ssh":
-		return sshHandler, nil
+		return sshHandler, 22, nil
+	case "http":
+		return httpHandler, 80, nil
 	default:
-		return nil, fmt.Errorf("unknown protocol '%s'", proto)
+		return nil, 0, fmt.Errorf("unknown protocol '%s'", proto)
 	}
 }