mirror of
https://github.com/golang/go
synced 2024-11-08 09:36:24 -07:00
0844ff8eef
Use of a nil *File as an argument should not result in a panic, but result in the ErrInvalid error being returned. Fix the copy_file_range implementation to preserve this semantic. Fixes #40115 Change-Id: Iad5ac39664a3efb7964cf55685be636940a8db13 Reviewed-on: https://go-review.googlesource.com/c/go/+/241417 Reviewed-by: Ian Lance Taylor <iant@golang.org> Reviewed-by: Damien Neil <dneil@google.com> Run-TryBot: Ian Lance Taylor <iant@golang.org> TryBot-Result: Gobot Gobot <gobot@golang.org>
47 lines
1.0 KiB
Go
47 lines
1.0 KiB
Go
// Copyright 2020 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 (
|
|
"internal/poll"
|
|
"io"
|
|
)
|
|
|
|
var pollCopyFileRange = poll.CopyFileRange
|
|
|
|
func (f *File) readFrom(r io.Reader) (written int64, handled bool, err error) {
|
|
// copy_file_range(2) does not support destinations opened with
|
|
// O_APPEND, so don't even try.
|
|
if f.appendMode {
|
|
return 0, false, nil
|
|
}
|
|
|
|
remain := int64(1 << 62)
|
|
|
|
lr, ok := r.(*io.LimitedReader)
|
|
if ok {
|
|
remain, r = lr.N, lr.R
|
|
if remain <= 0 {
|
|
return 0, true, nil
|
|
}
|
|
}
|
|
|
|
src, ok := r.(*File)
|
|
if !ok {
|
|
return 0, false, nil
|
|
}
|
|
if src.checkValid("ReadFrom") != nil {
|
|
// Avoid returning the error as we report handled as false,
|
|
// leave further error handling as the responsibility of the caller.
|
|
return 0, false, nil
|
|
}
|
|
|
|
written, handled, err = pollCopyFileRange(&f.pfd, &src.pfd, remain)
|
|
if lr != nil {
|
|
lr.N -= written
|
|
}
|
|
return written, handled, NewSyscallError("copy_file_range", err)
|
|
}
|