migrate to go-modules

Signed-off-by: Valentin Rothberg <rothberg@redhat.com>
This commit is contained in:
Valentin Rothberg
2019-06-24 11:29:13 +02:00
parent a3211b73c6
commit d697456dc9
1000 changed files with 97470 additions and 48284 deletions

View File

@ -0,0 +1,5 @@
test/test
test/piggie
test/phaul
image
rpc/rpc.proto

View File

@ -0,0 +1,25 @@
language: go
sudo: required
os:
- linux
go:
- "1.8"
- "1.9"
- "1.10"
env:
# Run the tests with CRIU master and criu-dev
- CRIU_BRANCH="master"
- CRIU_BRANCH="criu-dev"
install:
- sudo apt-get update
- sudo apt-get install -y libprotobuf-dev libprotobuf-c0-dev protobuf-c-compiler protobuf-compiler python-protobuf libnl-3-dev libnet-dev libcap-dev
- go get github.com/checkpoint-restore/go-criu
- git clone --single-branch -b ${CRIU_BRANCH} https://github.com/checkpoint-restore/criu.git
- cd criu; make
- sudo install -D -m 755 criu/criu /usr/sbin/
- cd ..
script:
# This builds the code without running the tests.
- make build phaul test/test test/phaul test/piggie
# Run actual test as root as it uses CRIU.
- sudo make test phaul-test

60
vendor/github.com/checkpoint-restore/go-criu/Makefile generated vendored Normal file
View File

@ -0,0 +1,60 @@
GO ?= go
CC ?= gcc
ifeq ($(GOPATH),)
export GOPATH := $(shell $(GO) env GOPATH)
endif
FIRST_GOPATH := $(firstword $(subst :, ,$(GOPATH)))
GOBIN := $(shell $(GO) env GOBIN)
ifeq ($(GOBIN),)
GOBIN := $(FIRST_GOPATH)/bin
endif
all: build test phaul phaul-test
lint:
@golint . test phaul
build:
@$(GO) build -v
test/piggie: test/piggie.c
@$(CC) $^ -o $@
test/test: test/main.go
@$(GO) build -v -o test/test test/main.go
test: test/test test/piggie
mkdir -p image
test/piggie
test/test dump `pidof piggie` image
test/test restore image
pkill -9 piggie || :
phaul:
@cd phaul; go build -v
test/phaul: test/phaul-main.go
@$(GO) build -v -o test/phaul test/phaul-main.go
phaul-test: test/phaul test/piggie
rm -rf image
test/piggie
test/phaul `pidof piggie`
pkill -9 piggie || :
clean:
@rm -f test/test test/piggie test/phaul
@rm -rf image
@rm -f rpc/rpc.proto
install.tools:
if [ ! -x "$(GOBIN)/golint" ]; then \
$(GO) get -u golang.org/x/lint/golint; \
fi
rpc/rpc.proto:
curl -s https://raw.githubusercontent.com/checkpoint-restore/criu/master/images/rpc.proto -o $@
rpc/rpc.pb.go: rpc/rpc.proto
protoc --go_out=. $^
.PHONY: build test clean lint phaul

View File

@ -1,133 +0,0 @@
package main
import (
"fmt"
"github.com/checkpoint-restore/go-criu"
"github.com/checkpoint-restore/go-criu/rpc"
"github.com/golang/protobuf/proto"
"os"
"strconv"
)
// TestNfy struct
type TestNfy struct {
criu.NoNotify
}
// PreDump test function
func (c TestNfy) PreDump() error {
fmt.Printf("TEST PRE DUMP\n")
return nil
}
func doDump(c *criu.Criu, pidS string, imgDir string, pre bool, prevImg string) error {
fmt.Printf("Dumping\n")
pid, _ := strconv.Atoi(pidS)
img, err := os.Open(imgDir)
if err != nil {
return fmt.Errorf("can't open image dir (%s)", err)
}
defer img.Close()
opts := rpc.CriuOpts{
Pid: proto.Int32(int32(pid)),
ImagesDirFd: proto.Int32(int32(img.Fd())),
LogLevel: proto.Int32(4),
LogFile: proto.String("dump.log"),
}
if prevImg != "" {
opts.ParentImg = proto.String(prevImg)
opts.TrackMem = proto.Bool(true)
}
if pre {
err = c.PreDump(opts, TestNfy{})
} else {
err = c.Dump(opts, TestNfy{})
}
if err != nil {
return fmt.Errorf("dump fail (%s)", err)
}
return nil
}
// Usage: test $act $pid $images_dir
func main() {
c := criu.MakeCriu()
// Read out CRIU version
version, err := c.GetCriuVersion()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
fmt.Println("CRIU version", version)
// Check if version at least 3.2
result, err := c.IsCriuAtLeast(30200)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
if !result {
fmt.Println("CRIU too old")
os.Exit(1)
}
act := os.Args[1]
switch act {
case "dump":
err := doDump(c, os.Args[2], os.Args[3], false, "")
if err != nil {
fmt.Print(err)
os.Exit(1)
}
case "dump2":
err := c.Prepare()
if err != nil {
fmt.Print(err)
os.Exit(1)
}
err = doDump(c, os.Args[2], os.Args[3]+"/pre", true, "")
if err != nil {
fmt.Printf("pre-dump failed")
fmt.Print(err)
os.Exit(1)
}
err = doDump(c, os.Args[2], os.Args[3], false, "./pre")
if err != nil {
fmt.Printf("dump failed")
fmt.Print(err)
os.Exit(1)
}
c.Cleanup()
case "restore":
fmt.Printf("Restoring\n")
img, err := os.Open(os.Args[2])
if err != nil {
fmt.Printf("can't open image dir")
os.Exit(1)
}
defer img.Close()
opts := rpc.CriuOpts{
ImagesDirFd: proto.Int32(int32(img.Fd())),
LogLevel: proto.Int32(4),
LogFile: proto.String("restore.log"),
}
err = c.Restore(opts, nil)
if err != nil {
fmt.Printf("Error:")
fmt.Print(err)
fmt.Printf("\n")
os.Exit(1)
}
default:
fmt.Printf("unknown action\n")
os.Exit(1)
}
fmt.Printf("Success\n")
}

View File

@ -1,192 +0,0 @@
package main
import (
"fmt"
"os"
"strconv"
"strings"
"syscall"
"github.com/checkpoint-restore/go-criu"
"github.com/checkpoint-restore/go-criu/phaul"
"github.com/checkpoint-restore/go-criu/rpc"
"github.com/golang/protobuf/proto"
)
type testLocal struct {
criu.NoNotify
r *testRemote
}
type testRemote struct {
srv *phaul.Server
}
/* Dir where test will put dump images */
const imagesDir = "image"
func prepareImages() error {
err := os.Mkdir(imagesDir, 0700)
if err != nil {
return err
}
/* Work dir for PhaulClient */
err = os.Mkdir(imagesDir+"/local", 0700)
if err != nil {
return err
}
/* Work dir for PhaulServer */
err = os.Mkdir(imagesDir+"/remote", 0700)
if err != nil {
return err
}
/* Work dir for DumpCopyRestore */
err = os.Mkdir(imagesDir+"/test", 0700)
if err != nil {
return err
}
return nil
}
func mergeImages(dumpDir, lastPreDumpDir string) error {
idir, err := os.Open(dumpDir)
if err != nil {
return err
}
defer idir.Close()
imgs, err := idir.Readdirnames(0)
if err != nil {
return err
}
for _, fname := range imgs {
if !strings.HasSuffix(fname, ".img") {
continue
}
fmt.Printf("\t%s -> %s/\n", fname, lastPreDumpDir)
err = syscall.Link(dumpDir+"/"+fname, lastPreDumpDir+"/"+fname)
if err != nil {
return err
}
}
return nil
}
func (r *testRemote) doRestore() error {
lastSrvImagesDir := r.srv.LastImagesDir()
/*
* In imagesDir we have images from dump, in the
* lastSrvImagesDir -- where server-side images
* (from page server, with pages and pagemaps) are.
* Need to put former into latter and restore from
* them.
*/
err := mergeImages(imagesDir+"/test", lastSrvImagesDir)
if err != nil {
return err
}
imgDir, err := os.Open(lastSrvImagesDir)
if err != nil {
return err
}
defer imgDir.Close()
opts := rpc.CriuOpts{
LogLevel: proto.Int32(4),
LogFile: proto.String("restore.log"),
ImagesDirFd: proto.Int32(int32(imgDir.Fd())),
}
cr := r.srv.GetCriu()
fmt.Printf("Do restore\n")
return cr.Restore(opts, nil)
}
func (l *testLocal) PostDump() error {
return l.r.doRestore()
}
func (l *testLocal) DumpCopyRestore(cr *criu.Criu, cfg phaul.Config, lastClnImagesDir string) error {
fmt.Printf("Final stage\n")
imgDir, err := os.Open(imagesDir + "/test")
if err != nil {
return err
}
defer imgDir.Close()
psi := rpc.CriuPageServerInfo{
Fd: proto.Int32(int32(cfg.Memfd)),
}
opts := rpc.CriuOpts{
Pid: proto.Int32(int32(cfg.Pid)),
LogLevel: proto.Int32(4),
LogFile: proto.String("dump.log"),
ImagesDirFd: proto.Int32(int32(imgDir.Fd())),
TrackMem: proto.Bool(true),
ParentImg: proto.String(lastClnImagesDir),
Ps: &psi,
}
fmt.Printf("Do dump\n")
return cr.Dump(opts, l)
}
func main() {
pid, _ := strconv.Atoi(os.Args[1])
fds, err := syscall.Socketpair(syscall.AF_LOCAL, syscall.SOCK_STREAM, 0)
if err != nil {
fmt.Printf("Can't make socketpair: %v\n", err)
os.Exit(1)
}
err = prepareImages()
if err != nil {
fmt.Printf("Can't prepare dirs for images: %v\n", err)
os.Exit(1)
return
}
fmt.Printf("Make server part (socket %d)\n", fds[1])
srv, err := phaul.MakePhaulServer(phaul.Config{
Pid: pid,
Memfd: fds[1],
Wdir: imagesDir + "/remote"})
if err != nil {
fmt.Printf("Unable to run a server: %v", err)
os.Exit(1)
return
}
r := &testRemote{srv}
fmt.Printf("Make client part (socket %d)\n", fds[0])
cln, err := phaul.MakePhaulClient(&testLocal{r: r}, srv,
phaul.Config{
Pid: pid,
Memfd: fds[0],
Wdir: imagesDir + "/local"})
if err != nil {
fmt.Printf("Unable to run a client: %v\n", err)
os.Exit(1)
}
fmt.Printf("Migrate\n")
err = cln.Migrate()
if err != nil {
fmt.Printf("Failed: %v\n", err)
os.Exit(1)
}
fmt.Printf("SUCCESS!\n")
}

View File

@ -1,57 +0,0 @@
#define _GNU_SOURCE
#include <stdio.h>
#include <signal.h>
#include <unistd.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <sched.h>
#define STKS (4*4096)
#ifndef CLONE_NEWPID
#define CLONE_NEWPID 0x20000000
#endif
static int do_test(void *logf)
{
int fd, i = 0;
setsid();
close(0);
close(1);
close(2);
fd = open("/dev/null", O_RDONLY);
if (fd != 0) {
dup2(fd, 0);
close(fd);
}
fd = open(logf, O_WRONLY | O_TRUNC | O_CREAT, 0600);
dup2(fd, 1);
dup2(fd, 2);
if (fd != 1 && fd != 2)
close(fd);
while (1) {
sleep(1);
printf("%d\n", i++);
fflush(stdout);
}
return 0;
}
int main(int argc, char **argv)
{
int pid;
void *stk;
stk = mmap(NULL, STKS, PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANON | MAP_GROWSDOWN, 0, 0);
pid = clone(do_test, stk + STKS, SIGCHLD | CLONE_NEWPID, argv[1]);
printf("Child forked, pid %d\n", pid);
return 0;
}