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/>.
29 "golang.org/x/crypto/blake2b"
32 func TempFile(dir, prefix string) (*os.File, error) {
33 // Assume that probability of suffix collision is negligible
34 suffix := strconv.FormatInt(time.Now().UnixNano()+int64(os.Getpid()), 16)
35 name := filepath.Join(dir, "nncp"+prefix+suffix)
36 return os.OpenFile(name, os.O_RDWR|os.O_CREATE|os.O_EXCL, os.FileMode(0666))
39 func (ctx *Ctx) NewTmpFile() (*os.File, error) {
40 jobsPath := filepath.Join(ctx.Spool, "tmp")
42 if err = os.MkdirAll(jobsPath, os.FileMode(0777)); err != nil {
45 fd, err := TempFile(jobsPath, "")
47 ctx.LogD("tmp", SDS{"src": fd.Name()}, "created")
52 type TmpFileWHash struct {
59 func (ctx *Ctx) NewTmpFileWHash() (*TmpFileWHash, error) {
60 tmp, err := ctx.NewTmpFile()
64 hsh, err := blake2b.New256(nil)
69 W: bufio.NewWriter(io.MultiWriter(hsh, tmp)),
76 func (tmp *TmpFileWHash) Cancel() {
77 tmp.Fd.Truncate(0) // #nosec G104
78 tmp.Fd.Close() // #nosec G104
79 os.Remove(tmp.Fd.Name()) // #nosec G104
82 func DirSync(dirPath string) error {
83 fd, err := os.Open(dirPath)
89 fd.Close() // #nosec G104
95 func (tmp *TmpFileWHash) Commit(dir string) error {
97 if err = os.MkdirAll(dir, os.FileMode(0777)); err != nil {
100 if err = tmp.W.Flush(); err != nil {
101 tmp.Fd.Close() // #nosec G104
104 if err = tmp.Fd.Sync(); err != nil {
105 tmp.Fd.Close() // #nosec G104
108 if err = tmp.Fd.Close(); err != nil {
111 checksum := Base32Codec.EncodeToString(tmp.Hsh.Sum(nil))
112 tmp.ctx.LogD("tmp", SDS{"src": tmp.Fd.Name(), "dst": checksum}, "commit")
113 if err = os.Rename(tmp.Fd.Name(), filepath.Join(dir, checksum)); err != nil {