Changeset 38 in code

Timestamp:

Oct 19, 2021, 6:09:49 PM (4 years ago)

Author:

dev

Message:

Provide ability to listen on unix sockets

File:

• trunk/partage.go (modified) (5 diffs)

trunk/partage.go

@@ -9 +9 @@
         "io/ioutil"
         "log"
+        "net"
         "net/http"
         "os"
@@ -287 +288 @@
 
 func main() {
+        var err error
         var configfile string
+        var listener net.Listener
 
         /* default values */
@@ -319 +322 @@
         }
 
+        if conf.bind[0] == '/' {
+                listener, err = net.Listen("unix", conf.bind)
+                if err != nil {
+                        log.Fatal(err)
+                }
+        } else {
+                listener, err = net.Listen("tcp", conf.bind)
+                if err != nil {
+                        log.Fatal(err)
+                }
+        }
+
         if conf.user != "" {
                 if verbose {
@@ -327 +342 @@
                         log.Fatal(err)
                 }
+
+                if listener.Addr().Network() == "unix" {
+                        os.Chown(conf.bind, uid, gid)
+                }
+
                 syscall.Setuid(uid)
                 syscall.Setgid(gid)
@@ -338 +358 @@
         }
 
-        http.ListenAndServe(conf.bind, nil)
-}
+        http.Serve(listener, nil)
+}