/*
NNCP -- Node to Node copy, utilities for store-and-forward data exchange
-Copyright (C) 2016-2021 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
"errors"
"fmt"
"io/ioutil"
- "log"
"os"
"path/filepath"
+ "strconv"
+ "strings"
"syscall"
-
- "golang.org/x/sys/unix"
)
type Ctx struct {
MCDRxIfis []string
MCDTxIfis map[string]int
+
+ YggdrasilAliases map[string]string
}
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))
- logMsg := func(les LEs) string {
- return fmt.Sprintf("Ensuring directory %s existence", dirPath)
+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 err := os.MkdirAll(dirPath, os.FileMode(0777)); err != nil {
- ctx.LogE("dir-ensure-mkdir", LEs{{"Dir", dirPath}}, err, logMsg)
+ 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-open", LEs{{"Dir", dirPath}}, err, logMsg)
- return err
+ ctx.LogE("dir-ensure-mkdir", LEs{{"Dir", dirPath}}, err, func(les LEs) string {
+ return fmt.Sprintf("Ensuring directory %s existence", dirPath)
+ })
}
- return fd.Close()
+ return err
}
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
}
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("Can not stat spool:", err)
- }
- return int64(s.Bavail)*int64(s.Bsize) > want
-}
-
func (ctx *Ctx) Umask() {
if ctx.UmaskForce != nil {
syscall.Umask(*ctx.UmaskForce)