mirror of
https://github.com/golang/go
synced 2024-11-19 17:14:44 -07:00
32f994acc6
On Plan 9, some file servers, like ramfs, handle the read offset when reading directories. However, the offset isn't valid anymore after directory entries have been removed between successive calls to read. This issue happens when os.RemoveAll is called on a directory that doesn't fit on a single 9P response message. In this case, the first part of the directory is read, then directory entries are removed and the second read will be incomplete because the read offset won't be valid anymore. Consequently, the content of the directory will only be partially removed. We change RemoveAll to call fd.Seek(0, 0) before calling fd.Readdirnames, so the read offset will always be reset after removing the directory entries. After adding TestRemoveAllLarge, we noticed the same issue appears on NaCl and the same fix applies as well. Fixes #22572. Change-Id: Ifc76ea7ccaf0168c34dc8ec0f400dc04db1baf8f Reviewed-on: https://go-review.googlesource.com/75974 Run-TryBot: David du Colombier <0intro@gmail.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Brad Fitzpatrick <bradfitz@golang.org>
138 lines
3.1 KiB
Go
138 lines
3.1 KiB
Go
// Copyright 2009 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package os
|
|
|
|
import (
|
|
"io"
|
|
"runtime"
|
|
"syscall"
|
|
)
|
|
|
|
// MkdirAll creates a directory named path,
|
|
// along with any necessary parents, and returns nil,
|
|
// or else returns an error.
|
|
// The permission bits perm are used for all
|
|
// directories that MkdirAll creates.
|
|
// If path is already a directory, MkdirAll does nothing
|
|
// and returns nil.
|
|
func MkdirAll(path string, perm FileMode) error {
|
|
// Fast path: if we can tell whether path is a directory or file, stop with success or error.
|
|
dir, err := Stat(path)
|
|
if err == nil {
|
|
if dir.IsDir() {
|
|
return nil
|
|
}
|
|
return &PathError{"mkdir", path, syscall.ENOTDIR}
|
|
}
|
|
|
|
// Slow path: make sure parent exists and then call Mkdir for path.
|
|
i := len(path)
|
|
for i > 0 && IsPathSeparator(path[i-1]) { // Skip trailing path separator.
|
|
i--
|
|
}
|
|
|
|
j := i
|
|
for j > 0 && !IsPathSeparator(path[j-1]) { // Scan backward over element.
|
|
j--
|
|
}
|
|
|
|
if j > 1 {
|
|
// Create parent
|
|
err = MkdirAll(path[0:j-1], perm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Parent now exists; invoke Mkdir and use its result.
|
|
err = Mkdir(path, perm)
|
|
if err != nil {
|
|
// Handle arguments like "foo/." by
|
|
// double-checking that directory doesn't exist.
|
|
dir, err1 := Lstat(path)
|
|
if err1 == nil && dir.IsDir() {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// RemoveAll removes path and any children it contains.
|
|
// It removes everything it can but returns the first error
|
|
// it encounters. If the path does not exist, RemoveAll
|
|
// returns nil (no error).
|
|
func RemoveAll(path string) error {
|
|
// Simple case: if Remove works, we're done.
|
|
err := Remove(path)
|
|
if err == nil || IsNotExist(err) {
|
|
return nil
|
|
}
|
|
|
|
// Otherwise, is this a directory we need to recurse into?
|
|
dir, serr := Lstat(path)
|
|
if serr != nil {
|
|
if serr, ok := serr.(*PathError); ok && (IsNotExist(serr.Err) || serr.Err == syscall.ENOTDIR) {
|
|
return nil
|
|
}
|
|
return serr
|
|
}
|
|
if !dir.IsDir() {
|
|
// Not a directory; return the error from Remove.
|
|
return err
|
|
}
|
|
|
|
// Directory.
|
|
fd, err := Open(path)
|
|
if err != nil {
|
|
if IsNotExist(err) {
|
|
// Race. It was deleted between the Lstat and Open.
|
|
// Return nil per RemoveAll's docs.
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
// Remove contents & return first error.
|
|
err = nil
|
|
for {
|
|
if err == nil && (runtime.GOOS == "plan9" || runtime.GOOS == "nacl") {
|
|
// Reset read offset after removing directory entries.
|
|
// See golang.org/issue/22572.
|
|
fd.Seek(0, 0)
|
|
}
|
|
names, err1 := fd.Readdirnames(100)
|
|
for _, name := range names {
|
|
err1 := RemoveAll(path + string(PathSeparator) + name)
|
|
if err == nil {
|
|
err = err1
|
|
}
|
|
}
|
|
if err1 == io.EOF {
|
|
break
|
|
}
|
|
// If Readdirnames returned an error, use it.
|
|
if err == nil {
|
|
err = err1
|
|
}
|
|
if len(names) == 0 {
|
|
break
|
|
}
|
|
}
|
|
|
|
// Close directory, because windows won't remove opened directory.
|
|
fd.Close()
|
|
|
|
// Remove directory.
|
|
err1 := Remove(path)
|
|
if err1 == nil || IsNotExist(err1) {
|
|
return nil
|
|
}
|
|
if err == nil {
|
|
err = err1
|
|
}
|
|
return err
|
|
}
|