Logging to opened file descriptor

[nncp.git] / src / ctx.go

diff --git a/src/ctx.go b/src/ctx.go
index 63d1fc27353dac5682fee2ce24b0d0adde87fdf9..58dba44df6156381216ee90e5043587d9b538039 100644 (file)
--- a/src/ctx.go
+++ b/src/ctx.go
@@ -21,13 +21,12 @@ import (
        "errors"
        "fmt"
        "io/ioutil"
-       "log"
        "os"
        "path/filepath"
+       "strconv"
+       "strings"
 
        "syscall"
-
-       "golang.org/x/sys/unix"
 )
 
 type Ctx struct {
@@ -88,9 +87,7 @@ func (ctx *Ctx) ensureRxDir(nodeId *NodeId) error {
 }
 
 func CtxFromCmdline(
-       cfgPath,
-       spoolPath,
-       logPath string,
+       cfgPath, spoolPath, logPath string,
        quiet, showPrgrs, omitPrgrs, debug bool,
 ) (*Ctx, error) {
        env := os.Getenv(CfgPathEnv)
@@ -100,11 +97,27 @@ func CtxFromCmdline(
        if showPrgrs && omitPrgrs {
                return nil, errors.New("simultaneous -progress and -noprogress")
        }
-       cfgRaw, err := ioutil.ReadFile(cfgPath)
+       fi, err := os.Stat(cfgPath)
        if err != nil {
                return nil, err
        }
-       ctx, err := CfgParse(cfgRaw)
+       var cfg *CfgJSON
+       if fi.IsDir() {
+               cfg, err = DirToCfg(cfgPath)
+               if err != nil {
+                       return nil, err
+               }
+       } else {
+               cfgRaw, err := ioutil.ReadFile(cfgPath)
+               if err != nil {
+                       return nil, err
+               }
+               cfg, err = CfgParse(cfgRaw)
+               if err != nil {
+                       return nil, err
+               }
+       }
+       ctx, err := Cfg2Ctx(cfg)
        if err != nil {
                return nil, err
        }
@@ -124,6 +137,18 @@ func CtxFromCmdline(
        } else {
                ctx.LogPath = logPath
        }
+       if strings.HasPrefix(ctx.LogPath, LogFdPrefix) {
+               ptr, err := strconv.ParseUint(
+                       strings.TrimPrefix(ctx.LogPath, LogFdPrefix), 10, 64,
+               )
+               if err != nil {
+                       return nil, err
+               }
+               LogFd = os.NewFile(uintptr(ptr), CfgLogEnv)
+               if LogFd == nil {
+                       return nil, errors.New("can not open:" + ctx.LogPath)
+               }
+       }
        if showPrgrs {
                ctx.ShowPrgrs = true
        }
@@ -135,14 +160,6 @@ func CtxFromCmdline(
        return ctx, nil
 }
 
-func (ctx *Ctx) IsEnoughSpace(want int64) bool {
-       var s unix.Statfs_t
-       if err := unix.Statfs(ctx.Spool, &s); err != nil {
-               log.Fatalln(err)
-       }
-       return int64(s.Bavail)*int64(s.Bsize) > want
-}
-
 func (ctx *Ctx) Umask() {
        if ctx.UmaskForce != nil {
                syscall.Umask(*ctx.UmaskForce)