mirror of
https://github.com/checkpoint-restore/criu.git
synced 2026-01-23 02:14:37 +00:00
022baf868b
Shows process tree from image. The output is like
PID PGID SID COMM
1 1 1 session00
4 4 4 session00
7 7 7 session00
8 4 4 session00
11 11 11 session00
12 4 4 session00
13 13 13 session00
14 14 14 session00
15 4 4 session00
6 4 4 session00
10 9 9 session00
(the above is for session00 test).
Signed-off-by: Pavel Emelyanov <xemul@parallels.com>
Appreciated-by: avagin@openvz.org
153 lines
3.6 KiB
Python
Executable file
153 lines
3.6 KiB
Python
Executable file
#!/usr/bin/env python
|
|
import argparse
|
|
import sys
|
|
import json
|
|
import os
|
|
|
|
import pycriu
|
|
|
|
def inf(opts):
|
|
if opts['in']:
|
|
return open(opts['in'], 'r')
|
|
else:
|
|
return sys.stdin
|
|
|
|
def outf(opts):
|
|
if opts['out']:
|
|
return open(opts['out'], 'w+')
|
|
else:
|
|
return sys.stdout
|
|
|
|
def dinf(opts, name):
|
|
return open(os.path.join(opts['dir'], name))
|
|
|
|
def decode(opts):
|
|
indent = None
|
|
|
|
try:
|
|
img = pycriu.images.load(inf(opts), opts['pretty'])
|
|
except pycriu.images.MagicException as exc:
|
|
print >>sys.stderr, "Unknown magic %#x.\n"\
|
|
"Maybe you are feeding me an image with "\
|
|
"raw data(i.e. pages.img)?" % exc.magic
|
|
sys.exit(1)
|
|
|
|
if opts['pretty']:
|
|
indent = 4
|
|
|
|
f = outf(opts)
|
|
json.dump(img, f, indent=indent)
|
|
if f == sys.stdout:
|
|
f.write("\n")
|
|
|
|
def encode(opts):
|
|
img = json.load(inf(opts))
|
|
pycriu.images.dump(img, outf(opts))
|
|
|
|
def info(opts):
|
|
infs = pycriu.images.info(inf(opts))
|
|
json.dump(infs, sys.stdout, indent = 4)
|
|
print
|
|
|
|
#
|
|
# Explorers
|
|
#
|
|
|
|
class ps_item:
|
|
def __init__(self, p, core):
|
|
self.pid = p['pid']
|
|
self.ppid = p['ppid']
|
|
self.p = p
|
|
self.core = core
|
|
self.kids = []
|
|
|
|
def show_ps(p, opts, depth = 0):
|
|
print "%7d%7d%7d %s%s" % (p.pid, p.p['pgid'], p.p['sid'],
|
|
' ' * (4 * depth), p.core['tc']['comm'])
|
|
for kid in p.kids:
|
|
show_ps(kid, opts, depth + 1)
|
|
|
|
def explore_ps(opts):
|
|
pss = { }
|
|
ps_img = pycriu.images.load(dinf(opts, 'pstree.img'))
|
|
for p in ps_img['entries']:
|
|
core = pycriu.images.load(dinf(opts, 'core-%d.img' % p['pid']))
|
|
ps = ps_item(p, core['entries'][0])
|
|
pss[ps.pid] = ps
|
|
|
|
# Build tree
|
|
psr = None
|
|
for pid in pss:
|
|
p = pss[pid]
|
|
if p.ppid == 0:
|
|
psr = p
|
|
continue
|
|
|
|
pp = pss[p.ppid]
|
|
pp.kids.append(p)
|
|
|
|
print "%7s%7s%7s %s" % ('PID', 'PGID', 'SID', 'COMM')
|
|
show_ps(psr, opts)
|
|
|
|
|
|
explorers = { 'ps': explore_ps }
|
|
|
|
def explore(opts):
|
|
explorers[opts['what']](opts)
|
|
|
|
def main():
|
|
desc = 'CRiu Image Tool'
|
|
parser = argparse.ArgumentParser(description=desc,
|
|
formatter_class=argparse.RawTextHelpFormatter)
|
|
|
|
subparsers = parser.add_subparsers(help='Use crit CMD --help for command-specific help')
|
|
|
|
# Decode
|
|
decode_parser = subparsers.add_parser('decode',
|
|
help = 'convert criu image from binary type to json')
|
|
decode_parser.add_argument('--pretty',
|
|
help = 'Multiline with indents and some numerical fields in field-specific format',
|
|
action = 'store_true')
|
|
decode_parser.add_argument('-i',
|
|
'--in',
|
|
help = 'criu image in binary format to be decoded (stdin by default)')
|
|
decode_parser.add_argument('-o',
|
|
'--out',
|
|
help = 'where to put criu image in json format (stdout by default)')
|
|
decode_parser.set_defaults(func=decode)
|
|
|
|
# Encode
|
|
encode_parser = subparsers.add_parser('encode',
|
|
help = 'convert criu image from json type to binary')
|
|
encode_parser.add_argument('-i',
|
|
'--in',
|
|
help = 'criu image in json format to be encoded (stdin by default)')
|
|
encode_parser.add_argument('-o',
|
|
'--out',
|
|
help = 'where to put criu image in binary format (stdout by default)')
|
|
encode_parser.set_defaults(func=encode)
|
|
|
|
# Info
|
|
info_parser = subparsers.add_parser('info',
|
|
help = 'show info about image')
|
|
info_parser.add_argument("in")
|
|
info_parser.set_defaults(func=info)
|
|
|
|
# Explore
|
|
x_parser = subparsers.add_parser('x', help = 'explore image dir')
|
|
x_parser.add_argument('dir')
|
|
x_parser.add_argument('what', choices = [ 'ps' ])
|
|
x_parser.set_defaults(func=explore)
|
|
|
|
# Show
|
|
show_parser = subparsers.add_parser('show',
|
|
help = "convert criu image from binary to human-readable json")
|
|
show_parser.add_argument("in")
|
|
show_parser.set_defaults(func=decode, pretty=True, out=None)
|
|
|
|
opts = vars(parser.parse_args())
|
|
|
|
opts["func"](opts)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|