[CRIU] [PATCH v2] crit: Use ns_pid, ns_sid and ns_gid

[Andrei Vagin]

Applied, thanks!

On Tue, Sep 05, 2017 at 11:19:41AM +0300, Kirill Tkhai wrote:
> Reflect commit 3086b0323cf8 "pstree: Dump and restore NSpid, NSsid etc"
> in crit.
> 
> Signed-off-by: Kirill Tkhai <ktkhai at virtuozzo.com>
> ---
> v2: Support old dumps without ns_pid too.
> 
>  crit/crit |   14 ++++++++------
>  1 file changed, 8 insertions(+), 6 deletions(-)
> 
> diff --git a/crit/crit b/crit/crit
> index a6a6832b5..7099ec81e 100755
> --- a/crit/crit
> +++ b/crit/crit
> @@ -49,20 +49,22 @@ import pycriu
>  	json.dump(infs, sys.stdout, indent = 4)
>  	print
>  
> +def get_task_id(p, val):
> +	return p[val] if val in p else p['ns_' + val][0]
>  #
>  # Explorers
>  #
>  
>  class ps_item:
>  	def __init__(self, p, core):
> -		self.pid = p['pid']
> +		self.pid = get_task_id(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'],
> +	print "%7d%7d%7d   %s%s" % (p.pid, get_task_id(p.p, 'pgid'), get_task_id(p.p, 'sid'),
>  			' ' * (4 * depth), p.core['tc']['comm'])
>  	for kid in p.kids:
>  		show_ps(kid, opts, depth + 1)
> @@ -71,7 +73,7 @@ import pycriu
>  	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']))
> +		core = pycriu.images.load(dinf(opts, 'core-%d.img' % get_task_id(p, 'pid')))
>  		ps = ps_item(p, core['entries'][0])
>  		pss[ps.pid] = ps
>  
> @@ -138,7 +140,7 @@ files_cache = { }
>  def explore_fds(opts):
>  	ps_img = pycriu.images.load(dinf(opts, 'pstree.img'))
>  	for p in ps_img['entries']:
> -		pid = p['pid']
> +		pid = get_task_id(p, 'pid')
>  		idi = pycriu.images.load(dinf(opts, 'ids-%s.img' % pid))
>  		fdt = idi['entries'][0]['files_id']
>  		fdi = pycriu.images.load(dinf(opts, 'fdinfo-%d.img' % fdt))
> @@ -170,7 +172,7 @@ files_cache = { }
>  	ps_img = pycriu.images.load(dinf(opts, 'pstree.img'))
>  	vids = vma_id()
>  	for p in ps_img['entries']:
> -		pid = p['pid']
> +		pid = get_task_id(p, 'pid')
>  		mmi = pycriu.images.load(dinf(opts, 'mm-%d.img' % pid))['entries'][0]
>  
>  		print "%d" % pid
> @@ -215,7 +217,7 @@ files_cache = { }
>  def explore_rss(opts):
>  	ps_img = pycriu.images.load(dinf(opts, 'pstree.img'))
>  	for p in ps_img['entries']:
> -		pid = p['pid']
> +		pid = get_task_id(p, 'pid')
>  		vmas = pycriu.images.load(dinf(opts, 'mm-%d.img' % pid))['entries'][0]['vmas']
>  		pms = pycriu.images.load(dinf(opts, 'pagemap-%d.img' % pid))['entries']
>  
>