Allow specifying permissions when creating anonymous temp files

This commit is contained in:
Kovid Goyal
2024-07-23 07:28:10 +05:30
parent 4c0ead129e
commit eaf71d1ccf
2 changed files with 25 additions and 3 deletions

View File

@@ -6,16 +6,33 @@ package utils
import (
"fmt"
"io/fs"
"os"
"golang.org/x/sys/unix"
)
var _ = fmt.Print
func CreateAnonymousTemp(dir string) (*os.File, error) {
func CreateAnonymousTemp(dir string, perms ...fs.FileMode) (*os.File, error) {
var perm fs.FileMode = unix.S_IREAD | unix.S_IWRITE
default_perm := perm
if len(perms) > 0 {
perm = perms[0]
}
f, err := os.CreateTemp(dir, "")
if err != nil {
return nil, err
}
if perm != default_perm {
err = f.Chmod(perm & fs.ModePerm)
if err != nil {
os.Remove(f.Name())
f.Close()
return nil, err
}
}
err = os.Remove(f.Name())
if err != nil {
f.Close()