39 lines
950 B
Go
39 lines
950 B
Go
//go:build linux || darwin || freebsd
|
|
// +build linux darwin freebsd
|
|
|
|
package http
|
|
|
|
import (
|
|
"net"
|
|
"os"
|
|
"syscall"
|
|
|
|
"github.com/fasthttp/router"
|
|
"github.com/valyala/fasthttp"
|
|
|
|
"github.com/yunginnanet/HellPot/config"
|
|
)
|
|
|
|
func listenOnUnixSocket(addr string, r *router.Router) error {
|
|
var err error
|
|
var unixAddr *net.UnixAddr
|
|
var unixListener *net.UnixListener
|
|
unixAddr, err = net.ResolveUnixAddr("unix", addr)
|
|
if err == nil {
|
|
// Always unlink sockets before listening on them
|
|
_ = syscall.Unlink(addr)
|
|
// Before we set socket permissions, we want to make sure only the user HellPot is running under
|
|
// has permission to the socket.
|
|
oldmask := syscall.Umask(0o077)
|
|
unixListener, err = net.ListenUnix("unix", unixAddr)
|
|
syscall.Umask(oldmask)
|
|
if err == nil {
|
|
err = os.Chmod(unixAddr.Name, os.FileMode(config.UnixSocketPermissions))
|
|
if err == nil {
|
|
err = fasthttp.Serve(unixListener, r.Handler)
|
|
}
|
|
}
|
|
}
|
|
return err
|
|
}
|