X-Git-Url: http://www.git.cypherpunks.ru/?p=gocheese.git;a=blobdiff_plain;f=fileutils.go;h=34e6e748956922fb8d7a79ac234828514d3da960;hp=b1b9ef95714edea8321e5832341f4d005a19f3fa;hb=HEAD;hpb=0d804c602fbc55f58d04fa9657950f3a6aac6d81

diff --git a/fileutils.go b/fileutils.go
index b1b9ef9..9655d0a 100644
--- a/fileutils.go
+++ b/fileutils.go
@@ -1,37 +1,50 @@
-/*
-GoCheese -- Python private package repository and caching proxy
-Copyright (C) 2019-2020 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
-the Free Software Foundation, version 3 of the License.
-
-This program is distributed in the hope that it will be useful,
-but WITHOUT ANY WARRANTY; without even the implied warranty of
-MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-GNU General Public License for more details.
-
-You should have received a copy of the GNU General Public License
-along with this program.  If not, see <http://www.gnu.org/licenses/>.
-*/
+// GoCheese -- Python private package repository and caching proxy
+// Copyright (C) 2019-2024 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
+// the Free Software Foundation, version 3 of the License.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 package main
 
 import (
+	"log"
+	"net/http"
 	"os"
 	"path/filepath"
-	"strconv"
 	"time"
 )
 
+var (
+	NoSync   = os.Getenv("GOCHEESE_NO_SYNC") == "1"
+	UmaskCur int
+)
+
 func TempFile(dir string) (*os.File, error) {
-	// Assume that probability of suffix collision is negligible
-	suffix := strconv.FormatInt(time.Now().UnixNano()+int64(os.Getpid()), 16)
-	name := filepath.Join(dir, "nncp"+suffix)
-	return os.OpenFile(name, os.O_RDWR|os.O_CREATE|os.O_EXCL, os.FileMode(0666))
+	tmp, err := os.CreateTemp(dir, "gocheese")
+	if err != nil {
+		return nil, err
+	}
+	err = os.Chmod(tmp.Name(), os.FileMode(0666&^UmaskCur))
+	if err != nil {
+		tmp.Close()
+		return nil, err
+	}
+	return tmp, nil
 }
 
 func DirSync(dirPath string) error {
+	if NoSync {
+		return nil
+	}
 	fd, err := os.Open(dirPath)
 	if err != nil {
 		return err
@@ -44,7 +57,7 @@ func DirSync(dirPath string) error {
 	return fd.Close()
 }
 
-func WriteFileSync(dirPath, filePath string, data []byte) error {
+func WriteFileSync(dirPath, filePath string, data []byte, mtime time.Time) error {
 	dst, err := TempFile(dirPath)
 	if err != nil {
 		return err
@@ -54,14 +67,32 @@ func WriteFileSync(dirPath, filePath string, data []byte) error {
 		dst.Close()
 		return err
 	}
-	if err = dst.Sync(); err != nil {
-		os.Remove(dst.Name())
-		dst.Close()
-		return err
+	if !NoSync {
+		if err = dst.Sync(); err != nil {
+			os.Remove(dst.Name())
+			dst.Close()
+			return err
+		}
 	}
 	dst.Close()
+	if err = os.Chtimes(dst.Name(), mtime, mtime); err != nil {
+		return err
+	}
 	if err = os.Rename(dst.Name(), filePath); err != nil {
 		return err
 	}
 	return DirSync(dirPath)
 }
+
+func mkdirForPkg(w http.ResponseWriter, r *http.Request, pkgName string) bool {
+	path := filepath.Join(Root, pkgName)
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		if err = os.Mkdir(path, os.FileMode(0777)); err != nil {
+			log.Println("error", r.RemoteAddr, "mkdir", pkgName, err)
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return false
+		}
+		log.Println(r.RemoteAddr, "mkdir", pkgName)
+	}
+	return true
+}