mirror of
https://github.com/checkpoint-restore/criu.git
synced 2026-01-23 10:16:41 +00:00
3825c3e5d4
Running crit tool 4 times per test (decode, encode, decode --pretty and encode back again) is way too slow. The majority of time, as it turned out, goes on python load and arguments parsing. The en- and de-coding works pretty fast. So doing re-code logic in one python script for ALL images is way way faster -- ~1 hour vs ~1 minute on my box. Signed-off-by: Pavel Emelyanov <xemul@parallels.com>
67 lines
1.3 KiB
Python
Executable file
67 lines
1.3 KiB
Python
Executable file
#!/bin/env python
|
|
|
|
import pycriu
|
|
import sys
|
|
import os
|
|
import subprocess
|
|
|
|
find = subprocess.Popen(['find', 'test/dump/', '-name', '*.img'],
|
|
stdout = subprocess.PIPE)
|
|
|
|
test_pass = True
|
|
|
|
def recode_and_check(imgf, o_img, pretty):
|
|
try:
|
|
pb = pycriu.images.loads(o_img, pretty)
|
|
except pycriu.images.MagicException as me:
|
|
print "%s magic %x error" % (imgf, me.magic)
|
|
return False
|
|
except:
|
|
print "%s %sdecode fails" % (imgf, pretty and 'pretty ' or '')
|
|
return False
|
|
|
|
try:
|
|
r_img = pycriu.images.dumps(pb)
|
|
except:
|
|
print "%s %sencode fails" % (imgf, pretty and 'pretty ' or '')
|
|
return False
|
|
|
|
if o_img != r_img:
|
|
print "%s %srecode mismatch" % (imgf, pretty and 'pretty ' or '')
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
for imgf in find.stdout.readlines():
|
|
imgf = imgf.strip()
|
|
imgf_b = os.path.basename(imgf)
|
|
|
|
if imgf_b.startswith('pages-'):
|
|
continue
|
|
if imgf_b.startswith('iptables-'):
|
|
continue
|
|
if imgf_b.startswith('ip6tables-'):
|
|
continue
|
|
if imgf_b.startswith('route-'):
|
|
continue
|
|
if imgf_b.startswith('route6-'):
|
|
continue
|
|
if imgf_b.startswith('ifaddr-'):
|
|
continue
|
|
if imgf_b.startswith('tmpfs-'):
|
|
continue
|
|
|
|
o_img = open(imgf).read()
|
|
if not recode_and_check(imgf, o_img, False):
|
|
test_pass = False
|
|
if not recode_and_check(imgf, o_img, True):
|
|
test_pass = False
|
|
|
|
find.wait()
|
|
|
|
if not test_pass:
|
|
print "FAIL"
|
|
sys.exit(1)
|
|
|
|
print "PASS"
|