forked from security/hakurei
Ophestra
6f719bc3c1
The mutex is not really doing anything, none of these methods make sense when called concurrently anyway. The copylocks analysis is still satisfied by the noCopy struct. Signed-off-by: Ophestra <cat@gensokyo.uk>
67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
package system
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"syscall"
|
|
)
|
|
|
|
// CopyFile appends [TmpfileOp] to [I].
|
|
func (sys *I) CopyFile(payload *[]byte, src string, cap int, n int64) *I {
|
|
buf := new(bytes.Buffer)
|
|
buf.Grow(cap)
|
|
sys.ops = append(sys.ops, &TmpfileOp{payload, src, n, buf})
|
|
return sys
|
|
}
|
|
|
|
// TmpfileOp reads up to n bytes from src and writes the resulting byte slice to payload.
|
|
type TmpfileOp struct {
|
|
payload *[]byte
|
|
src string
|
|
|
|
n int64
|
|
buf *bytes.Buffer
|
|
}
|
|
|
|
func (t *TmpfileOp) Type() Enablement { return Process }
|
|
func (t *TmpfileOp) apply(*I) error {
|
|
msg.Verbose("copying", t)
|
|
|
|
if t.payload == nil {
|
|
// this is a misuse of the API; do not return an error message
|
|
return errors.New("invalid payload")
|
|
}
|
|
|
|
if b, err := os.Stat(t.src); err != nil {
|
|
return newOpError("tmpfile", err, false)
|
|
} else {
|
|
if b.IsDir() {
|
|
return newOpError("tmpfile", &os.PathError{Op: "stat", Path: t.src, Err: syscall.EISDIR}, false)
|
|
}
|
|
if s := b.Size(); s > t.n {
|
|
return newOpError("tmpfile", &os.PathError{Op: "stat", Path: t.src, Err: syscall.ENOMEM}, false)
|
|
}
|
|
}
|
|
|
|
if f, err := os.Open(t.src); err != nil {
|
|
return newOpError("tmpfile", err, false)
|
|
} else if _, err = io.CopyN(t.buf, f, t.n); err != nil {
|
|
return newOpError("tmpfile", err, false)
|
|
}
|
|
|
|
*t.payload = t.buf.Bytes()
|
|
return nil
|
|
}
|
|
func (t *TmpfileOp) revert(*I, *Criteria) error { t.buf.Reset(); return nil }
|
|
|
|
func (t *TmpfileOp) Is(o Op) bool {
|
|
target, ok := o.(*TmpfileOp)
|
|
return ok && t != nil && target != nil &&
|
|
t.src == target.src && t.n == target.n
|
|
}
|
|
func (t *TmpfileOp) Path() string { return t.src }
|
|
func (t *TmpfileOp) String() string { return fmt.Sprintf("up to %d bytes from %q", t.n, t.src) }
|