mirror of
https://github.com/checkpoint-restore/criu.git
synced 2026-07-18 00:58:31 +00:00
criu-ns: Convert indentation to spaces
Spaces are the preferred indentation method. https://www.python.org/dev/peps/pep-0008/#tabs-or-spaces Signed-off-by: Radostin Stoyanov <rstoyanov@fedoraproject.org>
This commit is contained in:
Radostin Stoyanov
Andrei Vagin
parent
f3d071461f
commit
8f69a58e03
1 changed files with 127 additions and 127 deletions
254
scripts/criu-ns
254
scripts/criu-ns
|
|
@ -18,190 +18,190 @@ MS_SLAVE = 1 << 19
|
|||
_libc = ctypes.CDLL(ctypes.util.find_library("c"), use_errno=True)
|
||||
|
||||
try:
|
||||
_unshare = _libc.unshare
|
||||
_unshare = _libc.unshare
|
||||
except AttributeError:
|
||||
raise OSError(errno.EINVAL, "unshare is not supported on this platform")
|
||||
raise OSError(errno.EINVAL, "unshare is not supported on this platform")
|
||||
else:
|
||||
_unshare.argtypes = [ ctypes.c_int ]
|
||||
_unshare.restype = ctypes.c_int
|
||||
_unshare.argtypes = [ ctypes.c_int ]
|
||||
_unshare.restype = ctypes.c_int
|
||||
|
||||
try:
|
||||
_setns = _libc.setns
|
||||
_setns = _libc.setns
|
||||
except AttributeError:
|
||||
raise OSError(errno.EINVAL, "setns is not supported on this platform")
|
||||
raise OSError(errno.EINVAL, "setns is not supported on this platform")
|
||||
else:
|
||||
_setns.argtypes = [ ctypes.c_int, ctypes.c_int ]
|
||||
_setns.restype = ctypes.c_int
|
||||
_setns.argtypes = [ ctypes.c_int, ctypes.c_int ]
|
||||
_setns.restype = ctypes.c_int
|
||||
|
||||
try:
|
||||
_mount = _libc.mount
|
||||
_mount = _libc.mount
|
||||
except AttributeError:
|
||||
raise OSError(errno.EINVAL, "mount is not supported on this platform")
|
||||
raise OSError(errno.EINVAL, "mount is not supported on this platform")
|
||||
else:
|
||||
_mount.argtypes = [
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_ulong,
|
||||
ctypes.c_void_p
|
||||
]
|
||||
_mount.restype = ctypes.c_int
|
||||
_mount.argtypes = [
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_ulong,
|
||||
ctypes.c_void_p
|
||||
]
|
||||
_mount.restype = ctypes.c_int
|
||||
|
||||
try:
|
||||
_umount = _libc.umount
|
||||
_umount = _libc.umount
|
||||
except AttributeError:
|
||||
raise OSError(errno.EINVAL, "umount is not supported on this platform")
|
||||
raise OSError(errno.EINVAL, "umount is not supported on this platform")
|
||||
else:
|
||||
_umount.argtypes = [ctypes.c_char]
|
||||
_umount.restype = ctypes.c_int
|
||||
_umount.argtypes = [ctypes.c_char]
|
||||
_umount.restype = ctypes.c_int
|
||||
|
||||
|
||||
def run_criu():
|
||||
print(sys.argv)
|
||||
os.execlp('criu', *['criu'] + sys.argv[1:])
|
||||
print(sys.argv)
|
||||
os.execlp('criu', *['criu'] + sys.argv[1:])
|
||||
|
||||
|
||||
def wrap_restore():
|
||||
# Unshare pid and mount namespaces
|
||||
if _unshare(CLONE_NEWNS | CLONE_NEWPID) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
# Unshare pid and mount namespaces
|
||||
if _unshare(CLONE_NEWNS | CLONE_NEWPID) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
|
||||
criu_pid = os.fork()
|
||||
if criu_pid == 0:
|
||||
# Mount new /proc
|
||||
if _mount(None, b"/", None, MS_SLAVE|MS_REC, None) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
criu_pid = os.fork()
|
||||
if criu_pid == 0:
|
||||
# Mount new /proc
|
||||
if _mount(None, b"/", None, MS_SLAVE|MS_REC, None) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
|
||||
if _mount(b'proc', b'/proc', b'proc', 0, None) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
if _mount(b'proc', b'/proc', b'proc', 0, None) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
|
||||
# Spawn CRIU binary
|
||||
run_criu()
|
||||
raise OSError(errno.ENOENT, "No such command")
|
||||
# Spawn CRIU binary
|
||||
run_criu()
|
||||
raise OSError(errno.ENOENT, "No such command")
|
||||
|
||||
# Wait for CRIU to exit and report the status back
|
||||
while True:
|
||||
try:
|
||||
(pid, status) = os.wait()
|
||||
if pid == criu_pid:
|
||||
status = os.WEXITSTATUS(status)
|
||||
break
|
||||
except OSError:
|
||||
status = -251
|
||||
break
|
||||
# Wait for CRIU to exit and report the status back
|
||||
while True:
|
||||
try:
|
||||
(pid, status) = os.wait()
|
||||
if pid == criu_pid:
|
||||
status = os.WEXITSTATUS(status)
|
||||
break
|
||||
except OSError:
|
||||
status = -251
|
||||
break
|
||||
|
||||
return status
|
||||
return status
|
||||
|
||||
|
||||
def get_varg(args):
|
||||
for i in range(1, len(sys.argv)):
|
||||
if not sys.argv[i] in args:
|
||||
continue
|
||||
for i in range(1, len(sys.argv)):
|
||||
if not sys.argv[i] in args:
|
||||
continue
|
||||
|
||||
if i + 1 >= len(sys.argv):
|
||||
break
|
||||
if i + 1 >= len(sys.argv):
|
||||
break
|
||||
|
||||
return (sys.argv[i + 1], i + 1)
|
||||
return (sys.argv[i + 1], i + 1)
|
||||
|
||||
return (None, None)
|
||||
return (None, None)
|
||||
|
||||
|
||||
|
||||
def set_pidns(tpid, pid_idx):
|
||||
# Joind pid namespace. Note, that the given pid should
|
||||
# be changed in -t option, as task lives in different
|
||||
# pid namespace.
|
||||
# Joind pid namespace. Note, that the given pid should
|
||||
# be changed in -t option, as task lives in different
|
||||
# pid namespace.
|
||||
|
||||
myns = os.stat('/proc/self/ns/pid').st_ino
|
||||
myns = os.stat('/proc/self/ns/pid').st_ino
|
||||
|
||||
ns_fd = os.open('/proc/%s/ns/pid' % tpid, os.O_RDONLY)
|
||||
if myns != os.fstat(ns_fd).st_ino:
|
||||
ns_fd = os.open('/proc/%s/ns/pid' % tpid, os.O_RDONLY)
|
||||
if myns != os.fstat(ns_fd).st_ino:
|
||||
|
||||
for l in open('/proc/%s/status' % tpid):
|
||||
if not l.startswith('NSpid:'):
|
||||
continue
|
||||
for l in open('/proc/%s/status' % tpid):
|
||||
if not l.startswith('NSpid:'):
|
||||
continue
|
||||
|
||||
ls = l.split()
|
||||
if ls[1] != tpid:
|
||||
raise OSError(errno.ESRCH, 'No such pid')
|
||||
ls = l.split()
|
||||
if ls[1] != tpid:
|
||||
raise OSError(errno.ESRCH, 'No such pid')
|
||||
|
||||
print('Replace pid {} with {}'.format(tpid, ls[2]))
|
||||
sys.argv[pid_idx] = ls[2]
|
||||
break
|
||||
else:
|
||||
raise OSError(errno.ENOENT, 'Cannot find NSpid field in proc')
|
||||
print('Replace pid {} with {}'.format(tpid, ls[2]))
|
||||
sys.argv[pid_idx] = ls[2]
|
||||
break
|
||||
else:
|
||||
raise OSError(errno.ENOENT, 'Cannot find NSpid field in proc')
|
||||
|
||||
if _setns(ns_fd, 0) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
if _setns(ns_fd, 0) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
|
||||
os.close(ns_fd)
|
||||
os.close(ns_fd)
|
||||
|
||||
|
||||
def set_mntns(tpid):
|
||||
# Join mount namespace. Trick here too -- check / and .
|
||||
# will be the same in target mntns.
|
||||
# Join mount namespace. Trick here too -- check / and .
|
||||
# will be the same in target mntns.
|
||||
|
||||
myns = os.stat('/proc/self/ns/mnt').st_ino
|
||||
ns_fd = os.open('/proc/%s/ns/mnt' % tpid, os.O_RDONLY)
|
||||
if myns != os.fstat(ns_fd).st_ino:
|
||||
root_st = os.stat('/')
|
||||
cwd_st = os.stat('.')
|
||||
cwd_path = os.path.realpath('.')
|
||||
myns = os.stat('/proc/self/ns/mnt').st_ino
|
||||
ns_fd = os.open('/proc/%s/ns/mnt' % tpid, os.O_RDONLY)
|
||||
if myns != os.fstat(ns_fd).st_ino:
|
||||
root_st = os.stat('/')
|
||||
cwd_st = os.stat('.')
|
||||
cwd_path = os.path.realpath('.')
|
||||
|
||||
if _setns(ns_fd, 0) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
if _setns(ns_fd, 0) != 0:
|
||||
_errno = ctypes.get_errno()
|
||||
raise OSError(_errno, errno.errorcode[_errno])
|
||||
|
||||
os.chdir(cwd_path)
|
||||
root_nst = os.stat('/')
|
||||
cwd_nst = os.stat('.')
|
||||
os.chdir(cwd_path)
|
||||
root_nst = os.stat('/')
|
||||
cwd_nst = os.stat('.')
|
||||
|
||||
def steq(st, nst):
|
||||
return (st.st_dev, st.st_ino) == (nst.st_dev, nst.st_ino)
|
||||
def steq(st, nst):
|
||||
return (st.st_dev, st.st_ino) == (nst.st_dev, nst.st_ino)
|
||||
|
||||
if not steq(root_st, root_nst):
|
||||
raise OSError(errno.EXDEV, 'Target ns / is not as current')
|
||||
if not steq(cwd_st, cwd_nst):
|
||||
raise OSError(errno.EXDEV, 'Target ns . is not as current')
|
||||
if not steq(root_st, root_nst):
|
||||
raise OSError(errno.EXDEV, 'Target ns / is not as current')
|
||||
if not steq(cwd_st, cwd_nst):
|
||||
raise OSError(errno.EXDEV, 'Target ns . is not as current')
|
||||
|
||||
|
||||
os.close(ns_fd)
|
||||
os.close(ns_fd)
|
||||
|
||||
|
||||
def wrap_dump():
|
||||
(pid, pid_idx) = get_varg(('-t', '--tree'))
|
||||
if pid is None:
|
||||
raise OSError(errno.EINVAL, 'No --tree option given')
|
||||
(pid, pid_idx) = get_varg(('-t', '--tree'))
|
||||
if pid is None:
|
||||
raise OSError(errno.EINVAL, 'No --tree option given')
|
||||
|
||||
set_pidns(pid, pid_idx)
|
||||
set_mntns(pid)
|
||||
set_pidns(pid, pid_idx)
|
||||
set_mntns(pid)
|
||||
|
||||
# Spawn CRIU binary
|
||||
criu_pid = os.fork()
|
||||
if criu_pid == 0:
|
||||
run_criu()
|
||||
raise OSError(errno.ENOENT, "No such command")
|
||||
# Spawn CRIU binary
|
||||
criu_pid = os.fork()
|
||||
if criu_pid == 0:
|
||||
run_criu()
|
||||
raise OSError(errno.ENOENT, "No such command")
|
||||
|
||||
# Wait for CRIU to exit and report the status back
|
||||
while True:
|
||||
try:
|
||||
(pid, status) = os.wait()
|
||||
if pid == criu_pid:
|
||||
status = os.WEXITSTATUS(status)
|
||||
break
|
||||
except OSError:
|
||||
status = -251
|
||||
break
|
||||
# Wait for CRIU to exit and report the status back
|
||||
while True:
|
||||
try:
|
||||
(pid, status) = os.wait()
|
||||
if pid == criu_pid:
|
||||
status = os.WEXITSTATUS(status)
|
||||
break
|
||||
except OSError:
|
||||
status = -251
|
||||
break
|
||||
|
||||
return status
|
||||
return status
|
||||
|
||||
|
||||
if len(sys.argv) == 1:
|
||||
print("""
|
||||
print("""
|
||||
Usage:
|
||||
{0} dump|pre-dump -t PID [<options>]
|
||||
{0} restore [<options>]
|
||||
|
|
@ -210,16 +210,16 @@ Usage:
|
|||
pre-dump pre-dump task(s) minimizing their frozen time
|
||||
restore restore a process/tree
|
||||
""".format(sys.argv[0]))
|
||||
exit(1)
|
||||
exit(1)
|
||||
|
||||
action = sys.argv[1]
|
||||
|
||||
if action == 'restore':
|
||||
res = wrap_restore()
|
||||
res = wrap_restore()
|
||||
elif action == 'dump' or action == 'pre-dump':
|
||||
res = wrap_dump()
|
||||
res = wrap_dump()
|
||||
else:
|
||||
print('Unsupported action {} for nswrap'.format(action))
|
||||
res = -1
|
||||
print('Unsupported action {} for nswrap'.format(action))
|
||||
res = -1
|
||||
|
||||
sys.exit(res)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue