From c233eb6f948258dcf081e1d58ee371c7853a115c Mon Sep 17 00:00:00 2001 From: smail Date: Tue, 22 Sep 2026 03:51:39 +0800 Subject: [PATCH] =?UTF-8?q?=E6=94=B6=E7=B4=A7=E5=8F=91=E8=A1=8C=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E8=AF=BB=E5=86=99=E6=9D=83=E9=99=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/internal/release/release.go | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/backend/internal/release/release.go b/backend/internal/release/release.go index 1feacd3..6b39e63 100644 --- a/backend/internal/release/release.go +++ b/backend/internal/release/release.go @@ -12,6 +12,7 @@ import ( "time" "git.pcux.cn/xuans/Smail/internal/buildinfo" + securefile "git.pcux.cn/xuans/Smail/utils/file" ) var ( @@ -191,7 +192,7 @@ func replacePair(stage, targetDir string, names []string) error { } func ReadManifest(targetDir string) (Manifest, error) { - data, err := os.ReadFile(filepath.Join(targetDir, "release.json")) + data, err := securefile.ReadRegularFile(filepath.Join(targetDir, "release.json"), 1<<20) if err != nil { return Manifest{}, err } @@ -210,23 +211,28 @@ func executableNames() (string, string) { } func validateExecutableSource(path string) error { - info, err := os.Lstat(path) + input, err := securefile.OpenRegularFile(path, os.O_RDONLY, 0) if err != nil { return err } - if info.Mode()&os.ModeSymlink != 0 || !info.Mode().IsRegular() || info.Size() == 0 { + defer input.Close() + info, err := input.Stat() + if err != nil { + return err + } + if !info.Mode().IsRegular() || info.Size() == 0 { return ErrInvalidPair } return nil } func copyExecutable(source, target string, mode os.FileMode) error { - input, err := os.Open(source) + input, err := securefile.OpenRegularFile(source, os.O_RDONLY, 0) if err != nil { return err } defer input.Close() - output, err := os.OpenFile(target, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, mode) + output, err := securefile.CreateExclusiveFile(target, mode) if err != nil { return err } @@ -247,7 +253,7 @@ func writeManifest(path string, manifest Manifest) error { return fmt.Errorf("encode release manifest: %w", err) } data = append(data, '\n') - if err := os.WriteFile(path, data, 0640); err != nil { + if err := securefile.WriteRegularFile(path, data, 0600); err != nil { return fmt.Errorf("write release manifest: %w", err) } return nil