Ophestra
fe7d208cf7
This replaces the pipes object and integrates context into helper process lifecycle. Signed-off-by: Ophestra <cat@gensokyo.uk>
69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package seccomp
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"syscall"
|
|
|
|
"git.gensokyo.uk/security/fortify/helper/proc"
|
|
)
|
|
|
|
// New returns an inactive Encoder instance.
|
|
func New(opts SyscallOpts) *Encoder { return &Encoder{newExporter(opts)} }
|
|
|
|
/*
|
|
An Encoder writes a BPF program to an output stream.
|
|
|
|
Methods of Encoder are not safe for concurrent use.
|
|
|
|
An Encoder must not be copied after first use.
|
|
*/
|
|
type Encoder struct {
|
|
*exporter
|
|
}
|
|
|
|
func (e *Encoder) Read(p []byte) (n int, err error) {
|
|
if err = e.prepare(); err != nil {
|
|
return
|
|
}
|
|
return e.r.Read(p)
|
|
}
|
|
|
|
func (e *Encoder) Close() error {
|
|
if e.r == nil {
|
|
return syscall.EINVAL
|
|
}
|
|
|
|
// this hangs if the cgo thread fails to exit
|
|
return errors.Join(e.closeWrite(), <-e.exportErr)
|
|
}
|
|
|
|
// NewFile returns an instance of exporter implementing [proc.File].
|
|
func NewFile(opts SyscallOpts) proc.File { return &File{opts: opts} }
|
|
|
|
// File implements [proc.File] and provides access to the read end of exporter pipe.
|
|
type File struct {
|
|
opts SyscallOpts
|
|
proc.BaseFile
|
|
}
|
|
|
|
func (f *File) ErrCount() int { return 2 }
|
|
func (f *File) Fulfill(ctx context.Context, dispatchErr func(error)) error {
|
|
e := newExporter(f.opts)
|
|
if err := e.prepare(); err != nil {
|
|
return err
|
|
}
|
|
f.Set(e.r)
|
|
go func() {
|
|
select {
|
|
case err := <-e.exportErr:
|
|
dispatchErr(nil)
|
|
dispatchErr(err)
|
|
case <-ctx.Done():
|
|
dispatchErr(e.closeWrite())
|
|
dispatchErr(<-e.exportErr)
|
|
}
|
|
}()
|
|
return nil
|
|
}
|