mirror of
https://github.com/checkpoint-restore/criu.git
synced 2026-01-23 18:25:14 +00:00
8869d3ccc8
We want to execute this test in Jenkins. Signed-off-by: Andrew Vagin <avagin@openvz.org> Signed-off-by: Pavel Emelyanov <xemul@parallels.com>
135 lines
3 KiB
Python
Executable file
135 lines
3 KiB
Python
Executable file
#!/usr/bin/python
|
|
# Test criu errno
|
|
|
|
import socket, os, imp, sys, errno
|
|
import rpc_pb2 as rpc
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(description="Test errno reported by CRIU RPC")
|
|
parser.add_argument('socket', type = str, help = "CRIU service socket")
|
|
parser.add_argument('dir', type = str, help = "Directory where CRIU images should be placed")
|
|
|
|
args = vars(parser.parse_args())
|
|
|
|
# Prepare dir for images
|
|
class test:
|
|
def __init__(self):
|
|
self.imgs_fd = os.open(args['dir'], os.O_DIRECTORY)
|
|
self.s = -1
|
|
self._MAX_MSG_SIZE = 1024
|
|
|
|
def connect(self):
|
|
self.s = socket.socket(socket.AF_UNIX, socket.SOCK_SEQPACKET)
|
|
self.s.connect(args['socket'])
|
|
|
|
def get_base_req(self):
|
|
req = rpc.criu_req()
|
|
req.opts.log_level = 4
|
|
req.opts.images_dir_fd = self.imgs_fd
|
|
return req
|
|
|
|
def send_req(self, req):
|
|
self.connect()
|
|
self.s.send(req.SerializeToString())
|
|
|
|
def recv_resp(self):
|
|
resp = rpc.criu_resp()
|
|
resp.ParseFromString(self.s.recv(self._MAX_MSG_SIZE))
|
|
return resp
|
|
|
|
def check_resp(self, resp, typ, err):
|
|
if resp.type != typ:
|
|
raise Exception('Unexpected responce type ' + str(resp.type))
|
|
|
|
if resp.success:
|
|
raise Exception('Unexpected success = True')
|
|
|
|
if err and resp.cr_errno != err:
|
|
raise Exception('Unexpected cr_errno ' + str(resp.cr_errno))
|
|
|
|
def no_process(self):
|
|
print 'Try to dump unexisting process'
|
|
# Get pid of non-existing process.
|
|
# Suppose max_pid is not taken by any process.
|
|
with open("/proc/sys/kernel/pid_max", "r") as f:
|
|
pid = int(f.readline())
|
|
try:
|
|
os.kill(pid, 0)
|
|
except OSError:
|
|
pass
|
|
else:
|
|
raise Exception('max pid is taken')
|
|
|
|
# Ask criu to dump non-existing process.
|
|
req = self.get_base_req()
|
|
req.type = rpc.DUMP
|
|
req.opts.pid = pid
|
|
|
|
self.send_req(req)
|
|
resp = self.recv_resp()
|
|
|
|
self.check_resp(resp, rpc.DUMP, errno.ESRCH)
|
|
|
|
print 'Success'
|
|
|
|
def process_exists(self):
|
|
print 'Try to restore process which pid is already taken by other process'
|
|
|
|
# Perform self-dump
|
|
req = self.get_base_req()
|
|
req.type = rpc.DUMP
|
|
req.opts.leave_running = True
|
|
|
|
self.send_req(req)
|
|
resp = self.recv_resp()
|
|
|
|
if resp.success != True:
|
|
raise Exception('Self-dump failed')
|
|
|
|
# Ask to restore process from images of ourselves
|
|
req = self.get_base_req()
|
|
req.type = rpc.RESTORE
|
|
|
|
self.send_req(req)
|
|
resp = self.recv_resp()
|
|
|
|
self.check_resp(resp, rpc.RESTORE, errno.EEXIST)
|
|
|
|
print 'Success'
|
|
|
|
def bad_options(self):
|
|
print 'Try to send criu invalid opts'
|
|
|
|
# Subdirs are not allowed in log_file
|
|
req = self.get_base_req()
|
|
req.type = rpc.DUMP
|
|
req.opts.log_file = "../file.log"
|
|
|
|
self.send_req(req)
|
|
resp = self.recv_resp()
|
|
|
|
self.check_resp(resp, rpc.DUMP, errno.EBADRQC)
|
|
|
|
print 'Success'
|
|
|
|
def bad_request(self):
|
|
print 'Try to send criu invalid request type'
|
|
|
|
req = self.get_base_req()
|
|
req.type = rpc.NOTIFY
|
|
|
|
self.send_req(req)
|
|
resp = self.recv_resp()
|
|
|
|
self.check_resp(resp, rpc.EMPTY, None)
|
|
|
|
print 'Success'
|
|
|
|
def run(self):
|
|
self.no_process()
|
|
self.process_exists()
|
|
self.bad_options()
|
|
self.bad_request()
|
|
|
|
t = test()
|
|
t.run()
|