mirror of
https://github.com/checkpoint-restore/criu.git
synced 2026-07-17 16:47:50 +00:00
147660f02f
Use __future__ imports to keep this working for python2 Signed-off-by: Radostin Stoyanov <rstoyanov1@gmail.com> Signed-off-by: Andrei Vagin <avagin@gmail.com>
81 lines
2.1 KiB
Python
Executable file
81 lines
2.1 KiB
Python
Executable file
#!/usr/bin/python2
|
|
|
|
import socket, os, sys
|
|
import rpc_pb2 as rpc
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(description="Test dump/restore using 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())
|
|
|
|
# Connect to service socket
|
|
s = socket.socket(socket.AF_UNIX, socket.SOCK_SEQPACKET)
|
|
s.connect(args['socket'])
|
|
|
|
# Create criu msg, set it's type to dump request
|
|
# and set dump options. Checkout more options in protobuf/rpc.proto
|
|
req = rpc.criu_req()
|
|
req.type = rpc.DUMP
|
|
req.opts.leave_running = True
|
|
req.opts.log_level = 4
|
|
req.opts.images_dir_fd = os.open(args['dir'], os.O_DIRECTORY)
|
|
|
|
# Send request
|
|
s.send(req.SerializeToString())
|
|
|
|
# Recv response
|
|
resp = rpc.criu_resp()
|
|
MAX_MSG_SIZE = 1024
|
|
resp.ParseFromString(s.recv(MAX_MSG_SIZE))
|
|
|
|
if resp.type != rpc.DUMP:
|
|
print('Unexpected msg type')
|
|
sys.exit(-1)
|
|
else:
|
|
if resp.success:
|
|
print('Success')
|
|
else:
|
|
print('Fail')
|
|
sys.exit(-1)
|
|
|
|
if resp.dump.restored:
|
|
print('Restored')
|
|
|
|
# Connect to service socket
|
|
s = socket.socket(socket.AF_UNIX, socket.SOCK_SEQPACKET)
|
|
s.connect(args['socket'])
|
|
|
|
# Create criu msg, set it's type to dump request
|
|
# and set dump options. Checkout more options in protobuf/rpc.proto
|
|
req = rpc.criu_req()
|
|
req.type = rpc.VERSION
|
|
|
|
# Send request
|
|
s.send(req.SerializeToString())
|
|
|
|
# Recv response
|
|
resp = rpc.criu_resp()
|
|
MAX_MSG_SIZE = 1024
|
|
resp.ParseFromString(s.recv(MAX_MSG_SIZE))
|
|
|
|
if resp.type != rpc.VERSION:
|
|
print('RPC: Unexpected msg type')
|
|
sys.exit(-1)
|
|
else:
|
|
if resp.success:
|
|
print('RPC: Success')
|
|
print('CRIU major %d' % resp.version.major)
|
|
print('CRIU minor %d' % resp.version.minor)
|
|
if resp.version.HasField('gitid'):
|
|
print('CRIU gitid %s' % resp.version.gitid)
|
|
if resp.version.HasField('sublevel'):
|
|
print('CRIU sublevel %s' % resp.version.sublevel)
|
|
if resp.version.HasField('extra'):
|
|
print('CRIU extra %s' % resp.version.extra)
|
|
if resp.version.HasField('name'):
|
|
print('CRIU name %s' % resp.version.name)
|
|
else:
|
|
print('Fail')
|
|
sys.exit(-1)
|