2 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
3 Copyright (C) 2016-2021 Sergey Matveev <stargrave@stargrave.org>
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, version 3 of the License.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>.
30 "golang.org/x/sys/unix"
36 Neigh map[NodeId]*Node
37 Alias map[string]*NodeId
46 NotifyFile *FromToJSON
47 NotifyFreq *FromToJSON
48 NotifyExec map[string]*FromToJSON
51 func (ctx *Ctx) FindNode(id string) (*Node, error) {
52 nodeId, known := ctx.Alias[id]
54 return ctx.Neigh[*nodeId], nil
56 nodeId, err := NodeIdFromString(id)
60 node, known := ctx.Neigh[*nodeId]
62 return nil, errors.New("Unknown node")
67 func (ctx *Ctx) ensureRxDir(nodeId *NodeId) error {
68 dirPath := filepath.Join(ctx.Spool, nodeId.String(), string(TRx))
69 logMsg := func(les LEs) string {
70 return fmt.Sprintf("Ensuring directory %s existence", dirPath)
72 if err := os.MkdirAll(dirPath, os.FileMode(0777)); err != nil {
73 ctx.LogE("dir-ensure-mkdir", LEs{{"Dir", dirPath}}, err, logMsg)
76 fd, err := os.Open(dirPath)
78 ctx.LogE("dir-ensure-open", LEs{{"Dir", dirPath}}, err, logMsg)
88 quiet, showPrgrs, omitPrgrs, debug bool,
90 env := os.Getenv(CfgPathEnv)
94 if showPrgrs && omitPrgrs {
95 return nil, errors.New("simultaneous -progress and -noprogress")
97 cfgRaw, err := ioutil.ReadFile(cfgPath)
101 ctx, err := CfgParse(cfgRaw)
106 env = os.Getenv(CfgSpoolEnv)
111 ctx.Spool = spoolPath
114 env = os.Getenv(CfgLogEnv)
119 ctx.LogPath = logPath
124 if quiet || omitPrgrs {
125 ctx.ShowPrgrs = false
132 func (ctx *Ctx) IsEnoughSpace(want int64) bool {
134 if err := unix.Statfs(ctx.Spool, &s); err != nil {
137 return int64(s.Bavail)*int64(s.Bsize) > want
140 func (ctx *Ctx) Umask() {
141 if ctx.UmaskForce != nil {
142 syscall.Umask(*ctx.UmaskForce)