X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=src%2Fctx.go;h=c6f77c86876f644b27a8a69a3966e5ef5b0948f2;hb=e068d88291cd45a4d6b748e258077dd6c0ffb9c2;hp=28c9b38de1bfcaa5083a9b1ad82c28ac91c0f136;hpb=c8b26fe06596d26bdb14c5be85760fb3ddb197b3;p=nncp.git

diff --git a/src/ctx.go b/src/ctx.go
index 28c9b38..c6f77c8 100644
--- a/src/ctx.go
+++ b/src/ctx.go
@@ -1,6 +1,6 @@
 /*
 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
-Copyright (C) 2016-2019 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2016-2022 Sergey Matveev <stargrave@stargrave.org>
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
@@ -19,12 +19,14 @@ package nncp
 
 import (
 	"errors"
+	"fmt"
 	"io/ioutil"
-	"log"
 	"os"
 	"path/filepath"
+	"strconv"
+	"strings"
 
-	"golang.org/x/sys/unix"
+	"syscall"
 )
 
 type Ctx struct {
@@ -33,12 +35,24 @@ type Ctx struct {
 	Neigh  map[NodeId]*Node
 	Alias  map[string]*NodeId
 
+	AreaId2Area map[AreaId]*Area
+	AreaName2Id map[string]*AreaId
+
 	Spool      string
 	LogPath    string
+	UmaskForce *int
 	Quiet      bool
+	ShowPrgrs  bool
+	HdrUsage   bool
 	Debug      bool
-	NotifyFile *FromToYAML
-	NotifyFreq *FromToYAML
+	NotifyFile *FromToJSON
+	NotifyFreq *FromToJSON
+	NotifyExec map[string]*FromToJSON
+
+	MCDRxIfis []string
+	MCDTxIfis map[string]int
+
+	YggdrasilAliases map[string]string
 }
 
 func (ctx *Ctx) FindNode(id string) (*Node, error) {
@@ -57,31 +71,64 @@ func (ctx *Ctx) FindNode(id string) (*Node, error) {
 	return node, nil
 }
 
-func (ctx *Ctx) ensureRxDir(nodeId *NodeId) error {
-	dirPath := filepath.Join(ctx.Spool, nodeId.String(), string(TRx))
-	if err := os.MkdirAll(dirPath, os.FileMode(0700)); err != nil {
-		ctx.LogE("dir-ensure", SDS{"dir": dirPath, "err": err}, "")
+func ensureDir(dirs ...string) error {
+	p := filepath.Join(dirs...)
+	fi, err := os.Stat(p)
+	if err == nil {
+		if fi.IsDir() {
+			return nil
+		}
+		return fmt.Errorf("%s: is not a directory", p)
+	}
+	if !os.IsNotExist(err) {
 		return err
 	}
-	fd, err := os.Open(dirPath)
+	return os.MkdirAll(p, os.FileMode(0777))
+}
+
+func (ctx *Ctx) ensureRxDir(nodeId *NodeId) error {
+	dirPath := filepath.Join(ctx.Spool, nodeId.String(), string(TRx))
+	err := ensureDir(dirPath)
 	if err != nil {
-		ctx.LogE("dir-ensure", SDS{"dir": dirPath, "err": err}, "")
-		return err
+		ctx.LogE("dir-ensure-mkdir", LEs{{"Dir", dirPath}}, err, func(les LEs) string {
+			return fmt.Sprintf("Ensuring directory %s existence", dirPath)
+		})
 	}
-	fd.Close()
-	return nil
+	return err
 }
 
-func CtxFromCmdline(cfgPath, spoolPath, logPath string, quiet, debug bool) (*Ctx, error) {
+func CtxFromCmdline(
+	cfgPath, spoolPath, logPath string,
+	quiet, showPrgrs, omitPrgrs, debug bool,
+) (*Ctx, error) {
 	env := os.Getenv(CfgPathEnv)
 	if env != "" {
 		cfgPath = env
 	}
-	cfgRaw, err := ioutil.ReadFile(cfgPath)
+	if showPrgrs && omitPrgrs {
+		return nil, errors.New("simultaneous -progress and -noprogress")
+	}
+	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
 	}
@@ -101,15 +148,31 @@ func CtxFromCmdline(cfgPath, spoolPath, logPath string, quiet, debug bool) (*Ctx
 	} 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
+	}
+	if quiet || omitPrgrs {
+		ctx.ShowPrgrs = false
+	}
 	ctx.Quiet = quiet
 	ctx.Debug = debug
 	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)
+func (ctx *Ctx) Umask() {
+	if ctx.UmaskForce != nil {
+		syscall.Umask(*ctx.UmaskForce)
 	}
-	return int64(s.Bavail)*int64(s.Bsize) > want
 }