Merge pull request #1 from kescherCode/kescher/unix-socket

Add unix domain socket listening support

hellpot.go

@@ -4,18 +4,38 @@ import (
 	"context"
 	"github.com/gorilla/mux"
 	"github.com/yunginnanet/HellPot/src/config"
+	"net"
 	"net/http"
 	"os"
 	"os/signal"
+	"strings"
 	"syscall"
 	"time"
 )
 
 const robotsTxt = "User-agent: *\r\n"
 
+func listenOnUnixSocket(addr string, srv *http.Server) error {
+	var err error
+	var unixAddr *net.UnixAddr
+	var unixListener *net.UnixListener
+	socketPath := strings.TrimPrefix(addr, "unix:")
+	unixAddr, err = net.ResolveUnixAddr("unix", socketPath)
+	if err == nil {
+		// Always unlink sockets before listening on them
+		_ = syscall.Unlink(socketPath)
+		unixListener, err = net.ListenUnix("unix", unixAddr)
+		if err == nil {
+			err = srv.Serve(unixListener)
+		}
+	}
+	return err
+}
+
 func startPot() {
 	addr := config.BindAddr
 	port := config.BindPort
+	listenOnUnixDomain := true
 
 	// subscribe to SIGINT signals
 	stopChan := make(chan os.Signal, 1)
@@ -25,13 +45,22 @@ func startPot() {
 
 	r.HandleFunc("/{path}", DefaultHoneypot)
 
-	srv := &http.Server{Addr: addr + ":" + port, Handler: r}
+	srv := &http.Server{Handler: r}
+	if !strings.HasPrefix(addr, "unix:") {
+		listenOnUnixDomain = false
+		srv.Addr = addr + ":" + port
+	}
 
 	go func() {
 		log.Info().Str("bind_addr", addr).Str("bind_port", port).
 			Msg("Listening and serving HTTP...")
 		// service connections
-		err := srv.ListenAndServe()
+		var err error
+		if listenOnUnixDomain {
+			err = listenOnUnixSocket(addr, srv)
+		} else {
+			err = srv.ListenAndServe()
+		}
 		log.Warn().Err(err).Msg("HTTP_STOP")
 	}()