1 files changed, 159 insertions, 0 deletions
diff --git a/plugin/rpcplugin/sandbox/sandbox_linux_test.go b/plugin/rpcplugin/sandbox/sandbox_linux_test.go
new file mode 100644
index 000000000..2bcbf0c57
--- /dev/null
+++ b/plugin/rpcplugin/sandbox/sandbox_linux_test.go
@@ -0,0 +1,159 @@
+// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package sandbox
+
+import (
+	"context"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/mattermost/mattermost-server/plugin/rpcplugin/rpcplugintest"
+)
+
+func TestNewProcess(t *testing.T) {
+	if err := CheckSupport(); err != nil {
+		t.Skip("sandboxing not supported:", err)
+	}
+
+	dir, err := ioutil.TempDir("", "")
+	require.NoError(t, err)
+	defer os.RemoveAll(dir)
+
+	ping := filepath.Join(dir, "ping.exe")
+	rpcplugintest.CompileGo(t, `
+		package main
+
+		import (
+			"crypto/rand"
+			"fmt"
+			"io/ioutil"
+			"net/http"
+			"os"
+			"os/exec"
+			"syscall"
+
+			"github.com/stretchr/testify/assert"
+			"github.com/stretchr/testify/require"
+
+			"github.com/mattermost/mattermost-server/plugin/rpcplugin"
+		)
+
+		var failures int
+
+		type T struct {}
+		func (T) Errorf(format string, args ...interface{}) {
+			fmt.Printf(format, args...)
+			failures++
+		}
+		func (T) FailNow() {
+			os.Exit(1)
+		}
+
+		func init() {
+			if len(os.Args) > 0 && os.Args[0] == "exitImmediately" {
+				os.Exit(0)
+			}
+		}
+
+		func main() {
+			t := &T{}
+
+			pwd, err := os.Getwd()
+			assert.NoError(t, err)
+			assert.Equal(t, "/dir", pwd)
+
+			assert.Equal(t, 0, os.Getgid(), "we should see ourselves as root")
+			assert.Equal(t, 0, os.Getuid(), "we should see ourselves as root")
+
+			f, err := ioutil.TempFile("", "")
+			require.NoError(t, err, "we should be able to create temporary files")
+			f.Close()
+
+			_, err = os.Stat("ping.exe")
+			assert.NoError(t, err, "we should be able to read files in the working directory")
+
+			buf := make([]byte, 20)
+			n, err := rand.Read(buf)
+			assert.Equal(t, 20, n)
+			assert.NoError(t, err, "we should be able to read from /dev/urandom")
+
+			f, err = os.Create("/dev/zero")
+			require.NoError(t, err, "we should be able to write to /dev/zero")
+			defer f.Close()
+			n, err = f.Write([]byte("foo"))
+			assert.Equal(t, 3, n)
+			require.NoError(t, err, "we should be able to write to /dev/zero")
+
+			f, err = os.Create("/dir/foo")
+			if f != nil {
+				defer f.Close()
+			}
+			assert.Error(t, err, "we shouldn't be able to write to this read-only mount point")
+
+			_, err = ioutil.ReadFile("/etc/resolv.conf")
+			require.NoError(t, err, "we should be able to read /etc/resolv.conf")
+
+			resp, err := http.Get("https://github.com")
+			require.NoError(t, err, "we should be able to use the network")
+			resp.Body.Close()
+
+			status, err := ioutil.ReadFile("/proc/self/status")
+			require.NoError(t, err, "we should be able to read from /proc")
+			assert.Regexp(t, status, "CapEff:\\s+0000000000000000", "we should have no effective capabilities")
+
+			require.NoError(t, os.MkdirAll("/tmp/dir2", 0755))
+			err = syscall.Mount("/dir", "/tmp/dir2", "", syscall.MS_BIND, "")
+			assert.Equal(t, syscall.EPERM, err, "we shouldn't be allowed to mount things")
+
+			cmd := exec.Command("/proc/self/exe")
+			cmd.Args = []string{"exitImmediately"}
+			cmd.SysProcAttr = &syscall.SysProcAttr{
+				Pdeathsig:  syscall.SIGTERM,
+			}
+			assert.NoError(t, cmd.Run(), "we should be able to re-exec ourself")
+
+			cmd = exec.Command("/proc/self/exe")
+			cmd.Args = []string{"exitImmediately"}
+			cmd.SysProcAttr = &syscall.SysProcAttr{
+				Cloneflags: syscall.CLONE_NEWNS | syscall.CLONE_NEWUTS | syscall.CLONE_NEWIPC | syscall.CLONE_NEWPID | syscall.CLONE_NEWUSER,
+				Pdeathsig:  syscall.SIGTERM,
+			}
+			assert.Error(t, cmd.Run(), "we shouldn't be able to create new namespaces anymore")
+
+			ipc, err := rpcplugin.InheritedProcessIPC()
+			require.NoError(t, err)
+			defer ipc.Close()
+			_, err = ipc.Write([]byte("ping"))
+			require.NoError(t, err)
+
+			if failures > 0 {
+				os.Exit(1)
+			}
+		}
+	`, ping)
+
+	p, ipc, err := NewProcess(context.Background(), &Configuration{
+		MountPoints: []*MountPoint{
+			{
+				Source:      dir,
+				Destination: "/dir",
+				ReadOnly:    true,
+			},
+		},
+		WorkingDirectory: "/dir",
+	}, "/dir/ping.exe")
+	require.NoError(t, err)
+	defer ipc.Close()
+	b := make([]byte, 10)
+	n, err := ipc.Read(b)
+	require.NoError(t, err)
+	assert.Equal(t, 4, n)
+	assert.Equal(t, "ping", string(b[:4]))
+	require.NoError(t, p.Wait())
+}