mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
279 lines
6.7 KiB
Go
279 lines
6.7 KiB
Go
package plugin // import "github.com/docker/docker/plugin"
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/pkg/mount"
|
|
"github.com/docker/docker/pkg/stringid"
|
|
"github.com/docker/docker/pkg/system"
|
|
"github.com/docker/docker/plugin/v2"
|
|
"github.com/opencontainers/runtime-spec/specs-go"
|
|
"github.com/pkg/errors"
|
|
"gotest.tools/skip"
|
|
)
|
|
|
|
func TestManagerWithPluginMounts(t *testing.T) {
|
|
skip.If(t, os.Getuid() != 0, "skipping test that requires root")
|
|
root, err := ioutil.TempDir("", "test-store-with-plugin-mounts")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer system.EnsureRemoveAll(root)
|
|
|
|
s := NewStore()
|
|
managerRoot := filepath.Join(root, "manager")
|
|
p1 := newTestPlugin(t, "test1", "testcap", managerRoot)
|
|
|
|
p2 := newTestPlugin(t, "test2", "testcap", managerRoot)
|
|
p2.PluginObj.Enabled = true
|
|
|
|
m, err := NewManager(
|
|
ManagerConfig{
|
|
Store: s,
|
|
Root: managerRoot,
|
|
ExecRoot: filepath.Join(root, "exec"),
|
|
CreateExecutor: func(*Manager) (Executor, error) { return nil, nil },
|
|
LogPluginEvent: func(_, _, _ string) {},
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := s.Add(p1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := s.Add(p2); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Create a mount to simulate a plugin that has created it's own mounts
|
|
p2Mount := filepath.Join(p2.Rootfs, "testmount")
|
|
if err := os.MkdirAll(p2Mount, 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := mount.Mount("tmpfs", p2Mount, "tmpfs", ""); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := m.Remove(p1.GetID(), &types.PluginRmConfig{ForceRemove: true}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if mounted, err := mount.Mounted(p2Mount); !mounted || err != nil {
|
|
t.Fatalf("expected %s to be mounted, err: %v", p2Mount, err)
|
|
}
|
|
}
|
|
|
|
func newTestPlugin(t *testing.T, name, cap, root string) *v2.Plugin {
|
|
id := stringid.GenerateNonCryptoID()
|
|
rootfs := filepath.Join(root, id)
|
|
if err := os.MkdirAll(rootfs, 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
p := v2.Plugin{PluginObj: types.Plugin{ID: id, Name: name}}
|
|
p.Rootfs = rootfs
|
|
iType := types.PluginInterfaceType{Capability: cap, Prefix: "docker", Version: "1.0"}
|
|
i := types.PluginConfigInterface{Socket: "plugin.sock", Types: []types.PluginInterfaceType{iType}}
|
|
p.PluginObj.Config.Interface = i
|
|
p.PluginObj.ID = id
|
|
|
|
return &p
|
|
}
|
|
|
|
type simpleExecutor struct {
|
|
}
|
|
|
|
func (e *simpleExecutor) Create(id string, spec specs.Spec, stdout, stderr io.WriteCloser) error {
|
|
return errors.New("Create failed")
|
|
}
|
|
|
|
func (e *simpleExecutor) Restore(id string, stdout, stderr io.WriteCloser) (bool, error) {
|
|
return false, nil
|
|
}
|
|
|
|
func (e *simpleExecutor) IsRunning(id string) (bool, error) {
|
|
return false, nil
|
|
}
|
|
|
|
func (e *simpleExecutor) Signal(id string, signal int) error {
|
|
return nil
|
|
}
|
|
|
|
func TestCreateFailed(t *testing.T) {
|
|
root, err := ioutil.TempDir("", "test-create-failed")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer system.EnsureRemoveAll(root)
|
|
|
|
s := NewStore()
|
|
managerRoot := filepath.Join(root, "manager")
|
|
p := newTestPlugin(t, "create", "testcreate", managerRoot)
|
|
|
|
m, err := NewManager(
|
|
ManagerConfig{
|
|
Store: s,
|
|
Root: managerRoot,
|
|
ExecRoot: filepath.Join(root, "exec"),
|
|
CreateExecutor: func(*Manager) (Executor, error) { return &simpleExecutor{}, nil },
|
|
LogPluginEvent: func(_, _, _ string) {},
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := s.Add(p); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := m.enable(p, &controller{}, false); err == nil {
|
|
t.Fatalf("expected Create failed error, got %v", err)
|
|
}
|
|
|
|
if err := m.Remove(p.GetID(), &types.PluginRmConfig{ForceRemove: true}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
type executorWithRunning struct {
|
|
m *Manager
|
|
root string
|
|
exitChans map[string]chan struct{}
|
|
}
|
|
|
|
func (e *executorWithRunning) Create(id string, spec specs.Spec, stdout, stderr io.WriteCloser) error {
|
|
sockAddr := filepath.Join(e.root, id, "plugin.sock")
|
|
ch := make(chan struct{})
|
|
if e.exitChans == nil {
|
|
e.exitChans = make(map[string]chan struct{})
|
|
}
|
|
e.exitChans[id] = ch
|
|
listenTestPlugin(sockAddr, ch)
|
|
return nil
|
|
}
|
|
|
|
func (e *executorWithRunning) IsRunning(id string) (bool, error) {
|
|
return true, nil
|
|
}
|
|
func (e *executorWithRunning) Restore(id string, stdout, stderr io.WriteCloser) (bool, error) {
|
|
return true, nil
|
|
}
|
|
|
|
func (e *executorWithRunning) Signal(id string, signal int) error {
|
|
ch := e.exitChans[id]
|
|
ch <- struct{}{}
|
|
<-ch
|
|
e.m.HandleExitEvent(id)
|
|
return nil
|
|
}
|
|
|
|
func TestPluginAlreadyRunningOnStartup(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
root, err := ioutil.TempDir("", t.Name())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer system.EnsureRemoveAll(root)
|
|
|
|
for _, test := range []struct {
|
|
desc string
|
|
config ManagerConfig
|
|
}{
|
|
{
|
|
desc: "live-restore-disabled",
|
|
config: ManagerConfig{
|
|
LogPluginEvent: func(_, _, _ string) {},
|
|
},
|
|
},
|
|
{
|
|
desc: "live-restore-enabled",
|
|
config: ManagerConfig{
|
|
LogPluginEvent: func(_, _, _ string) {},
|
|
LiveRestoreEnabled: true,
|
|
},
|
|
},
|
|
} {
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
config := test.config
|
|
desc := test.desc
|
|
t.Parallel()
|
|
|
|
p := newTestPlugin(t, desc, desc, config.Root)
|
|
p.PluginObj.Enabled = true
|
|
|
|
// Need a short-ish path here so we don't run into unix socket path length issues.
|
|
config.ExecRoot, err = ioutil.TempDir("", "plugintest")
|
|
|
|
executor := &executorWithRunning{root: config.ExecRoot}
|
|
config.CreateExecutor = func(m *Manager) (Executor, error) { executor.m = m; return executor, nil }
|
|
|
|
if err := executor.Create(p.GetID(), specs.Spec{}, nil, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
root := filepath.Join(root, desc)
|
|
config.Root = filepath.Join(root, "manager")
|
|
if err := os.MkdirAll(filepath.Join(config.Root, p.GetID()), 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !p.IsEnabled() {
|
|
t.Fatal("plugin should be enabled")
|
|
}
|
|
if err := (&Manager{config: config}).save(p); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
s := NewStore()
|
|
config.Store = s
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer system.EnsureRemoveAll(config.ExecRoot)
|
|
|
|
m, err := NewManager(config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer m.Shutdown()
|
|
|
|
p = s.GetAll()[p.GetID()] // refresh `p` with what the manager knows
|
|
if p.Client() == nil {
|
|
t.Fatal("plugin client should not be nil")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func listenTestPlugin(sockAddr string, exit chan struct{}) (net.Listener, error) {
|
|
if err := os.MkdirAll(filepath.Dir(sockAddr), 0755); err != nil {
|
|
return nil, err
|
|
}
|
|
l, err := net.Listen("unix", sockAddr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
go func() {
|
|
for {
|
|
conn, err := l.Accept()
|
|
if err != nil {
|
|
return
|
|
}
|
|
conn.Close()
|
|
}
|
|
}()
|
|
go func() {
|
|
<-exit
|
|
l.Close()
|
|
os.Remove(sockAddr)
|
|
exit <- struct{}{}
|
|
}()
|
|
return l, nil
|
|
}
|