mirror of
https://github.com/gin-gonic/gin.git
synced 2026-06-07 21:08:17 +08:00
Merge ec746b406056edd05042f58199ed834d8a085fa3 into d75fcd4c9ab260e5225de590f1f0f8c0e0e12d11
This commit is contained in:
commit
6ace805722
10
context.go
10
context.go
@ -728,11 +728,17 @@ func (c *Context) SaveUploadedFile(file *multipart.FileHeader, dst string, perm
|
||||
mode = perm[0]
|
||||
}
|
||||
dir := filepath.Dir(dst)
|
||||
// Check if directory exists to only chmod newly created directories
|
||||
_, statErr := os.Stat(dir)
|
||||
dirExists := !os.IsNotExist(statErr)
|
||||
if err = os.MkdirAll(dir, mode); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = os.Chmod(dir, mode); err != nil {
|
||||
return err
|
||||
// Only chmod if directory was newly created (fixes #4622)
|
||||
if !dirExists {
|
||||
if err = os.Chmod(dir, mode); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
out, err := os.Create(dst)
|
||||
|
||||
@ -247,14 +247,19 @@ func TestSaveUploadedFileWithPermission(t *testing.T) {
|
||||
f, err := c.FormFile("file")
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, "permission_test", f.Filename)
|
||||
|
||||
// Save into a not-yet-existing subdirectory so the requested mode is
|
||||
// applied by MkdirAll/Chmod and the assertion is independent of the
|
||||
// test runner's working-directory permissions.
|
||||
parent := t.TempDir()
|
||||
newDir := filepath.Join(parent, "perm_subdir")
|
||||
dst := filepath.Join(newDir, "permission_test")
|
||||
|
||||
var mode fs.FileMode = 0o755
|
||||
require.NoError(t, c.SaveUploadedFile(f, "permission_test", mode))
|
||||
t.Cleanup(func() {
|
||||
assert.NoError(t, os.Remove("permission_test"))
|
||||
})
|
||||
info, err := os.Stat(filepath.Dir("permission_test"))
|
||||
require.NoError(t, c.SaveUploadedFile(f, dst, mode))
|
||||
info, err := os.Stat(newDir)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, info.Mode().Perm(), mode)
|
||||
assert.Equal(t, mode, info.Mode().Perm())
|
||||
}
|
||||
|
||||
func TestSaveUploadedFileWithPermissionFailed(t *testing.T) {
|
||||
@ -275,6 +280,39 @@ func TestSaveUploadedFileWithPermissionFailed(t *testing.T) {
|
||||
require.Error(t, c.SaveUploadedFile(f, "test/permission_test", mode))
|
||||
}
|
||||
|
||||
// TestSaveUploadedFileExistingDir verifies that SaveUploadedFile does not
|
||||
// alter permissions on a pre-existing target directory (regression test for
|
||||
// #4622, where chmod on a system dir like /tmp would fail).
|
||||
func TestSaveUploadedFileExistingDir(t *testing.T) {
|
||||
buf := new(bytes.Buffer)
|
||||
mw := multipart.NewWriter(buf)
|
||||
w, err := mw.CreateFormFile("file", "existing_dir_test")
|
||||
require.NoError(t, err)
|
||||
_, err = w.Write([]byte("existing_dir_test"))
|
||||
require.NoError(t, err)
|
||||
mw.Close()
|
||||
c, _ := CreateTestContext(httptest.NewRecorder())
|
||||
c.Request, _ = http.NewRequest(http.MethodPost, "/", buf)
|
||||
c.Request.Header.Set("Content-Type", mw.FormDataContentType())
|
||||
f, err := c.FormFile("file")
|
||||
require.NoError(t, err)
|
||||
|
||||
// Create a target directory with a non-default permission and
|
||||
// confirm SaveUploadedFile leaves the existing perm untouched.
|
||||
dir := t.TempDir()
|
||||
var existingPerm fs.FileMode = 0o700
|
||||
require.NoError(t, os.Chmod(dir, existingPerm))
|
||||
|
||||
dst := filepath.Join(dir, "existing_dir_test")
|
||||
var mode fs.FileMode = 0o755
|
||||
require.NoError(t, c.SaveUploadedFile(f, dst, mode))
|
||||
|
||||
info, err := os.Stat(dir)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, existingPerm, info.Mode().Perm(),
|
||||
"existing directory permissions must not be modified")
|
||||
}
|
||||
|
||||
func TestContextReset(t *testing.T) {
|
||||
router := New()
|
||||
c := router.allocateContext(0)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user