X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=src%2Ftmp.go;h=4a43eaa41a2675969f0d7109fe362f5304c93866;hb=HEAD;hp=62ec5bbc9cbdd4d253858666c850168595ab3ba9;hpb=4e08a1c97600e0372680e86a651f916c70e89342;p=nncp.git

diff --git a/src/tmp.go b/src/tmp.go
index 62ec5bb..c4f57c1 100644
--- a/src/tmp.go
+++ b/src/tmp.go
@@ -1,6 +1,6 @@
 /*
 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
-Copyright (C) 2016-2020 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2016-2023 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,16 +19,21 @@ package nncp
 
 import (
 	"bufio"
+	"fmt"
 	"hash"
 	"io"
 	"os"
 	"path/filepath"
 	"strconv"
 	"time"
-
-	"golang.org/x/crypto/blake2b"
 )
 
+var NoSync bool
+
+func init() {
+	NoSync = os.Getenv(CfgNoSync) != ""
+}
+
 func TempFile(dir, prefix string) (*os.File, error) {
 	// Assume that probability of suffix collision is negligible
 	suffix := strconv.FormatInt(time.Now().UnixNano()+int64(os.Getpid()), 16)
@@ -38,13 +43,14 @@ func TempFile(dir, prefix string) (*os.File, error) {
 
 func (ctx *Ctx) NewTmpFile() (*os.File, error) {
 	jobsPath := filepath.Join(ctx.Spool, "tmp")
-	var err error
-	if err = os.MkdirAll(jobsPath, os.FileMode(0777)); err != nil {
+	if err := ensureDir(jobsPath); err != nil {
 		return nil, err
 	}
 	fd, err := TempFile(jobsPath, "")
 	if err == nil {
-		ctx.LogD("tmp", SDS{"src": fd.Name()}, "created")
+		ctx.LogD("tmp", LEs{{"Src", fd.Name()}}, func(les LEs) string {
+			return "Temporary file created: " + fd.Name()
+		})
 	}
 	return fd, err
 }
@@ -61,10 +67,7 @@ func (ctx *Ctx) NewTmpFileWHash() (*TmpFileWHash, error) {
 	if err != nil {
 		return nil, err
 	}
-	hsh, err := blake2b.New256(nil)
-	if err != nil {
-		return nil, err
-	}
+	hsh := MTHNew(0, 0)
 	return &TmpFileWHash{
 		W:   bufio.NewWriter(io.MultiWriter(hsh, tmp)),
 		Fd:  tmp,
@@ -74,42 +77,57 @@ func (ctx *Ctx) NewTmpFileWHash() (*TmpFileWHash, error) {
 }
 
 func (tmp *TmpFileWHash) Cancel() {
-	tmp.Fd.Truncate(0) // #nosec G104
-	tmp.Fd.Close() // #nosec G104
-	os.Remove(tmp.Fd.Name()) // #nosec G104
+	tmp.Fd.Truncate(0)
+	tmp.Fd.Close()
+	os.Remove(tmp.Fd.Name())
 }
 
 func DirSync(dirPath string) error {
+	if NoSync {
+		return nil
+	}
 	fd, err := os.Open(dirPath)
 	if err != nil {
 		return err
 	}
 	err = fd.Sync()
 	if err != nil {
-		fd.Close() // #nosec G104
+		fd.Close()
 		return err
 	}
 	return fd.Close()
 }
 
+func (tmp *TmpFileWHash) Checksum() string {
+	return Base32Codec.EncodeToString(tmp.Hsh.Sum(nil))
+}
+
 func (tmp *TmpFileWHash) Commit(dir string) error {
 	var err error
-	if err = os.MkdirAll(dir, os.FileMode(0777)); err != nil {
+	if err = ensureDir(dir); err != nil {
 		return err
 	}
 	if err = tmp.W.Flush(); err != nil {
-		tmp.Fd.Close() // #nosec G104
+		tmp.Fd.Close()
 		return err
 	}
-	if err = tmp.Fd.Sync(); err != nil {
-		tmp.Fd.Close() // #nosec G104
-		return err
+	if !NoSync {
+		if err = tmp.Fd.Sync(); err != nil {
+			tmp.Fd.Close()
+			return err
+		}
 	}
 	if err = tmp.Fd.Close(); err != nil {
 		return err
 	}
-	checksum := Base32Codec.EncodeToString(tmp.Hsh.Sum(nil))
-	tmp.ctx.LogD("tmp", SDS{"src": tmp.Fd.Name(), "dst": checksum}, "commit")
+	checksum := tmp.Checksum()
+	tmp.ctx.LogD(
+		"tmp-rename",
+		LEs{{"Src", tmp.Fd.Name()}, {"Dst", checksum}},
+		func(les LEs) string {
+			return fmt.Sprintf("Temporary file: %s -> %s", tmp.Fd.Name(), checksum)
+		},
+	)
 	if err = os.Rename(tmp.Fd.Name(), filepath.Join(dir, checksum)); err != nil {
 		return err
 	}