mirror of
https://github.com/criyle/go-sandbox.git
synced 2025-11-04 14:49:53 +08:00
112 lines
1.7 KiB
Go
112 lines
1.7 KiB
Go
package forkexec
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"syscall"
|
|
"testing"
|
|
|
|
"github.com/criyle/go-sandbox/pkg/mount"
|
|
)
|
|
|
|
func TestFork_DropCaps(t *testing.T) {
|
|
t.Parallel()
|
|
r := Runner{
|
|
Args: []string{"/bin/echo"},
|
|
CloneFlags: syscall.CLONE_NEWUSER,
|
|
DropCaps: true,
|
|
}
|
|
_, err := r.Start()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestFork_ETXTBSY(t *testing.T) {
|
|
f, err := ioutil.TempFile("", "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.Remove(f.Name())
|
|
defer f.Close()
|
|
|
|
if err := f.Chmod(0777); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
echo, err := os.Open("/bin/echo")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer echo.Close()
|
|
|
|
_, err = io.Copy(f, echo)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
r := Runner{
|
|
Args: []string{f.Name()},
|
|
ExecFile: f.Fd(),
|
|
}
|
|
_, err = r.Start()
|
|
if err != syscall.ETXTBSY {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestFork_OK(t *testing.T) {
|
|
t.Parallel()
|
|
f, err := ioutil.TempFile("", "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.Remove(f.Name())
|
|
|
|
if err := f.Chmod(0777); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
echo, err := os.Open("/bin/echo")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer echo.Close()
|
|
|
|
_, err = io.Copy(f, echo)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
f.Close()
|
|
|
|
r := Runner{
|
|
Args: []string{f.Name()},
|
|
}
|
|
_, err = r.Start()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestFork_ENOENT(t *testing.T) {
|
|
t.Parallel()
|
|
m, err := mount.NewBuilder().
|
|
WithMount(
|
|
mount.Mount{
|
|
Source: "NOT_EXISTS",
|
|
}).Build()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r := Runner{
|
|
Args: []string{"/bin/echo"},
|
|
CloneFlags: syscall.CLONE_NEWNS | syscall.CLONE_NEWUSER,
|
|
Mounts: m,
|
|
}
|
|
_, err = r.Start()
|
|
if err != syscall.ENOENT {
|
|
t.Fatal(err)
|
|
}
|
|
}
|