mirror of
https://github.com/checkpoint-restore/criu
synced 2025-08-28 04:48:16 +00:00
38 lines
837 B
Python
38 lines
837 B
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
import socket, os, imp, sys
|
||
|
|
|
||
|
|
p = os.getcwd()
|
||
|
|
sys.path.append(p)
|
||
|
|
import rpc_pb2 as rpc
|
||
|
|
|
||
|
|
# Connect to service socket
|
||
|
|
s = socket.socket(socket.AF_UNIX, socket.SOCK_SEQPACKET)
|
||
|
|
s.connect('criu_service.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.RESTORE
|
||
|
|
|
||
|
|
req.opts.images_dir_fd = os.open('imgs_loop', 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.RESTORE:
|
||
|
|
print 'Unexpected msg type'
|
||
|
|
sys.exit(-1)
|
||
|
|
else:
|
||
|
|
if resp.success:
|
||
|
|
print 'Restore success'
|
||
|
|
else:
|
||
|
|
print 'Restore fail'
|
||
|
|
sys.exit(-1)
|
||
|
|
print "PID of the restored program is %d\n" %(resp.restore.pid)
|