[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index] [Xen-devel] [Patch v6 02/13] tools/python: Scripts relating to migrationv2 streams
* format.py contains structure and constants as per the stream specificion. * convert-legacy-stream.py will take a legacy migration stream as an input, and produce a v2 stream as an output. * verify-stream-v2.py will verify a stream against the v2 specification. format.py lives as part of the regular xen library, while convert-legacy-stream and verify-stream-v2 are installed as standalone scripts into PRIVATE_BINDIR. Signed-off-by: Andrew Cooper <andrew.cooper3@xxxxxxxxxx> CC: Ian Campbell <Ian.Campbell@xxxxxxxxxx> CC: Ian Jackson <Ian.Jackson@xxxxxxxxxxxxx> --- v6: * Move to be part of tools/python and installed in proper locations --- tools/python/Makefile | 5 + tools/python/scripts/convert-legacy-stream.py | 622 +++++++++++++++++++++++++ tools/python/scripts/verify-stream-v2.py | 456 ++++++++++++++++++ tools/python/setup.py | 1 + tools/python/xen/streamv2/format.py | 148 ++++++ 5 files changed, 1232 insertions(+) create mode 100755 tools/python/scripts/convert-legacy-stream.py create mode 100755 tools/python/scripts/verify-stream-v2.py create mode 100644 tools/python/xen/streamv2/__init__.py create mode 100644 tools/python/xen/streamv2/format.py diff --git a/tools/python/Makefile b/tools/python/Makefile index c914332..f1038a1 100644 --- a/tools/python/Makefile +++ b/tools/python/Makefile @@ -20,9 +20,14 @@ build: genpath genwrap.py $(XEN_ROOT)/tools/libxl/libxl_types.idl \ .PHONY: install install: + $(INSTALL_DIR) $(DESTDIR)$(PRIVATE_BINDIR) + CC="$(CC)" CFLAGS="$(CFLAGS) $(APPEND_LDFLAGS)" $(PYTHON) setup.py install \ $(PYTHON_PREFIX_ARG) --root="$(DESTDIR)" --force + $(INSTALL_PROG) scripts/convert-legacy-stream.py $(DESTDIR)$(PRIVATE_BINDIR)/convert-legacy-stream + $(INSTALL_PROG) scripts/verify-stream-v2.py $(DESTDIR)$(PRIVATE_BINDIR)/verify-stream-v2 + .PHONY: test test: export LD_LIBRARY_PATH=$$(readlink -f ../libxc):$$(readlink -f ../xenstore); $(PYTHON) test.py -b -u diff --git a/tools/python/scripts/convert-legacy-stream.py b/tools/python/scripts/convert-legacy-stream.py new file mode 100755 index 0000000..af2b365 --- /dev/null +++ b/tools/python/scripts/convert-legacy-stream.py @@ -0,0 +1,622 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +import sys +import struct +import os, os.path +import syslog +import traceback + +from xen.streamv2 import format as sv2 + +__version__ = 1 + +fin = None # Input file/fd +fout = None # Output file/fd +twidth = 0 # Legacy toolstack bitness (32 or 64) +pv = None # Boolean (pv or hvm) +qemu = True # Boolean - process qemu record? +log_to_syslog = False # Boolean - Log to syslog instead of stdout/err? +verbose = False # Boolean - Summarise stream contents + +def stream_read(_ = None): + return fin.read(_) + +def stream_write(_): + return fout.write(_) + +def info(msg): + """Info message, routed to appropriate destination""" + if verbose: + if log_to_syslog: + for line in msg.split("\n"): + syslog.syslog(syslog.LOG_INFO, line) + else: + print msg + +def err(msg): + """Error message, routed to appropriate destination""" + if log_to_syslog: + for line in msg.split("\n"): + syslog.syslog(syslog.LOG_ERR, line) + else: + print >> sys.stderr, msg + +class StreamError(StandardError): + pass + +class VM(object): + + def __init__(self): + # Common + self.p2m_size = 0 + + # PV + self.max_vcpu_id = 0 + self.online_vcpu_map = [] + self.width = 0 + self.levels = 0 + self.basic_len = 0 + self.extd = False + self.xsave_len = 0 + +def write_ihdr(): + stream_write(struct.pack(sv2.IHDR_FORMAT, + sv2.IHDR_MARKER, # Marker + sv2.IHDR_IDENT, # Ident + 2, # Version + sv2.IHDR_OPT_LE, # Options + 0, 0)) # Reserved + +def write_dhdr(): + if pv: + dtype = sv2.DHDR_TYPE_x86_pv + else: + dtype = sv2.DHDR_TYPE_x86_hvm + + stream_write(struct.pack(sv2.DHDR_FORMAT, + dtype, # Type + 12, # Page size + 0, # Reserved + 0, # Xen major (converted) + __version__)) # Xen minor (converted) + +def write_record(rt, *argl): + alldata = ''.join(argl) + length = len(alldata) + + record = struct.pack(sv2.RH_FORMAT, rt, length) + alldata + plen = (8 - (length & 7)) & 7 + record += '\x00' * plen + + stream_write(record) + +def write_pv_info(vm): + write_record(sv2.REC_TYPE_x86_pv_info, + struct.pack(sv2.X86_PV_INFO_FORMAT, + vm.width, vm.levels, 0, 0)) + +def write_pv_p2m_frames(vm, pfns): + write_record(sv2.REC_TYPE_x86_pv_p2m_frames, + struct.pack(sv2.X86_PV_P2M_FRAMES_FORMAT, + 0, vm.p2m_size - 1), + struct.pack("Q" * len(pfns), *pfns)) + +def write_pv_vcpu_basic(vcpu_id, data): + write_record(sv2.REC_TYPE_x86_pv_vcpu_basic, + struct.pack(sv2.X86_PV_VCPU_HDR_FORMAT, vcpu_id, 0), + data) + +def write_pv_vcpu_extd(vcpu_id, data): + write_record(sv2.REC_TYPE_x86_pv_vcpu_extended, + struct.pack(sv2.X86_PV_VCPU_HDR_FORMAT, vcpu_id, 0), + data) + +def write_pv_vcpu_xsave(vcpu_id, data): + write_record(sv2.REC_TYPE_x86_pv_vcpu_xsave, + struct.pack(sv2.X86_PV_VCPU_HDR_FORMAT, vcpu_id, 0), + data) + +def write_page_data(pfns, pages): + if fout is None: # Safe copying 1M buffers around for no reason + return + + new_pfns = [(((x & 0xf0000000) << 32) | (x & 0x0fffffff)) for x in pfns] + + # Optimise the needless buffer copying in write_record() + stream_write(struct.pack(sv2.RH_FORMAT, + sv2.REC_TYPE_page_data, + 8 + (len(new_pfns) * 8) + len(pages))) + stream_write(struct.pack(sv2.PAGE_DATA_FORMAT, len(new_pfns), 0)) + stream_write(struct.pack("Q" * len(new_pfns), *new_pfns)) + stream_write(pages) + +def write_tsc_info(mode, khz, nsec, incarn): + write_record(sv2.REC_TYPE_tsc_info, + struct.pack(sv2.TSC_INFO_FORMAT, + mode, khz, nsec, incarn, 0)) + +def write_hvm_params(params): + if pv: + raise StreamError("HVM-only param in PV stream") + elif len(params) % 2: + raise RuntimeError("Expected even length list of hvm parameters") + + write_record(sv2.REC_TYPE_hvm_params, + struct.pack(sv2.HVM_PARAMS_FORMAT, len(params) / 2, 0), + struct.pack("Q" * len(params), *params)) + + +def rdexact(nr_bytes): + """Read exactly nr_bytes from fin""" + _ = stream_read(nr_bytes) + if len(_) != nr_bytes: + raise IOError("Stream truncated") + return _ + +def unpack_exact(fmt): + """Unpack a format from fin""" + sz = struct.calcsize(fmt) + return struct.unpack(fmt, rdexact(sz)) + +def unpack_ulongs(nr_ulongs): + if twidth == 32: + return unpack_exact("I" * nr_ulongs) + else: + return unpack_exact("Q" * nr_ulongs) + +def skip_xl_header(): + + hdr = rdexact(32) + if hdr != "Xen saved domain, xl format\n \0 \r": + raise StreamError("No xl header") + + opts = rdexact(16) + _, _, _, optlen = struct.unpack("=IIII", opts) + + optdata = rdexact(optlen) + + info("Skipped xl header") + + stream_write(hdr) + stream_write(opts) + stream_write(optdata) + +def read_pv_extended_info(vm): + + marker, = unpack_ulongs(1) + + if twidth == 32: + expected = 0xffffffff + else: + expected = 0xffffffffffffffff + + if marker != expected: + raise StreamError("Unexpected extended info marker 0x%x" % (marker, )) + + total_length, = unpack_exact("I") + so_far = 0 + + info("Extended Info: length 0x%x" % (total_length, )) + + while so_far < total_length: + + blkid, datasz = unpack_exact("=4sI") + so_far += 8 + + info(" Record type: %s, size 0x%x" % (blkid, datasz)) + + data = rdexact(datasz) + so_far += datasz + + # Eww, but this is how it is done :( + if blkid == "vcpu": + + vm.basic_len = datasz + + if datasz == 0x1430: + vm.width = 8 + vm.levels = 4 + info(" 64bit domain, 4 levels") + elif datasz == 0xaf0: + vm.width = 4 + vm.levels = 3 + info(" 32bit domain, 3 levels") + else: + raise StreamError("Unable to determine guest width/level") + + write_pv_info(vm) + + elif blkid == "extv": + vm.extd = True + + elif blkid == "xcnt": + vm.xsave_len, = struct.unpack("I", data[:4]) + info("xcnt sz 0x%x" % (vm.xsave_len, )) + + else: + raise StreamError("Unrecognised extended block") + + + if so_far != total_length: + raise StreamError("Overshot Extended Info size by %d bytes" + % (so_far - total_length,)) + +def read_pv_p2m_frames(vm): + fpp = 4096 / vm.width + p2m_frame_len = (vm.p2m_size - 1) / fpp + 1 + + info("P2M frames: fpp %d, p2m_frame_len %d" % (fpp, p2m_frame_len)) + write_pv_p2m_frames(vm, unpack_ulongs(p2m_frame_len)) + +def read_pv_tail(vm): + + nr_unmapped_pfns, = unpack_exact("I") + + if nr_unmapped_pfns != 0: + # "Unmapped" pfns are bogus + _ = unpack_ulongs(nr_unmapped_pfns) + info("discarding %d bogus 'unmapped pfns'" % (nr_unmapped_pfns, )) + #raise StreamError("Found bogus 'unmapped pfns'") + + for vcpu_id in vm.online_vcpu_map: + + basic = rdexact(vm.basic_len) + info("Got VCPU basic (size 0x%x)" % (vm.basic_len, )) + write_pv_vcpu_basic(vcpu_id, basic) + + if vm.extd: + extd = rdexact(128) + info("Got VCPU extd (size 0x%x)" % (128, )) + write_pv_vcpu_extd(vcpu_id, extd) + + if vm.xsave_len: + mask, size = unpack_exact("QQ") + assert vm.xsave_len - 16 == size + + xsave = rdexact(size) + info("Got VCPU xsave (mask 0x%x, size 0x%x)" % (mask, size)) + write_pv_vcpu_xsave(vcpu_id, xsave) + + shinfo = rdexact(4096) + info("Got shinfo") + + write_record(sv2.REC_TYPE_shared_info, shinfo) + write_record(sv2.REC_TYPE_end, "") + + +def read_chunks(vm): + + hvm_params = [] + + while True: + + marker, = unpack_exact("=i") + if marker <= 0: + info("Chunk: type 0x%x" % (marker, )) + + if marker == 0: + info(" End") + + if hvm_params: + write_hvm_params(hvm_params) + + return + + elif marker > 0: + + if marker > 1024: + raise StreamError("Page batch (%d) exceeded MAX_BATCH" + % (marker, )) + pfns = unpack_ulongs(marker) + + # xc_domain_save() leaves many XEN_DOMCTL_PFINFO_XTAB records for + # sequences of pfns it cant map. Drop these. + pfns = [ x for x in pfns if x != 0xf0000000 ] + + if len(set(pfns)) != len(pfns): + raise StreamError("Duplicate pfns in batch") + + # print "0x[", + # for pfn in pfns: + # print "%x" % (pfn, ), + # print "]" + + nr_pages = len([x for x in pfns if (x & 0xf0000000) < 0xd0000000]) + + #print " Page Batch, %d PFNs, %d pages" % (marker, nr_pages) + pages = rdexact(nr_pages * 4096) + + write_page_data(pfns, pages) + + elif marker == -1: # XC_SAVE_ID_ENABLE_VERIFY_MODE + # Verify mode... Seemingly nothing to do... + pass + + elif marker == -2: # XC_SAVE_ID_VCPU_INFO + max_id, = unpack_exact("i") + + if max_id > 4095: + raise StreamError("Vcpu max_id out of range: %d > 4095" + % (max_id, ) ) + + vm.max_vcpu_id = max_id + bitmap = unpack_exact("Q" * ((max_id/64) + 1)) + + for idx, word in enumerate(bitmap): + bit_idx = 0 + + while word > 0: + if word & 1: + vm.online_vcpu_map.append((idx * 64) + bit_idx) + + bit_idx += 1 + word >>= 1 + + info(" Vcpu info: max_id %d, online map %s" + % (vm.max_vcpu_id, vm.online_vcpu_map)) + + elif marker == -3: # XC_SAVE_ID_HVM_IDENT_PT + _, ident_pt = unpack_exact("=IQ") + info(" EPT Identity Pagetable 0x%x" % (ident_pt, )) + hvm_params.extend([12, # HVM_PARAM_IDENT_PT + ident_pt]) + + elif marker == -4: # XC_SAVE_ID_HVM_VM86_TSS + _, vm86_tss = unpack_exact("=IQ") + info(" VM86 TSS: 0x%x" % (vm86_tss, )) + hvm_params.extend([15, # HVM_PARAM_VM86_TSS + vm86_tss]) + + elif marker == -5: # XC_SAVE_ID_TMEM + raise RuntimeError("todo") + + elif marker == -6: # XC_SAVE_ID_TMEM_EXTRA + raise RuntimeError("todo") + + elif marker == -7: # XC_SAVE_ID_TSC_INFO + mode, nsec, khz, incarn = unpack_exact("=IQII") + info(" TSC_INFO: mode %s, %d ns, %d khz, %d incarn" + % (mode, nsec, khz, incarn)) + write_tsc_info(mode, khz, nsec, incarn) + + elif marker == -8: # XC_SAVE_ID_HVM_CONSOLE_PFN + _, console_pfn = unpack_exact("=IQ") + info(" Console pfn 0x%x" % (console_pfn, )) + hvm_params.extend([17, # HVM_PARAM_CONSOLE_PFN + console_pfn]) + + elif marker == -9: # XC_SAVE_ID_LAST_CHECKPOINT + info(" Last Checkpoint") + # Nothing to do + + elif marker == -10: # XC_SAVE_ID_HVM_ACPI_IOPORTS_LOCATION + _, loc = unpack_exact("=IQ") + info(" ACPI ioport location 0x%x" % (loc, )) + hvm_params.extend([19, # HVM_PARAM_ACPI_IOPORTS_LOCATION + loc]) + + elif marker == -11: # XC_SAVE_ID_HVM_VIRIDIAN + _, loc = unpack_exact("=IQ") + info(" Viridian location 0x%x" % (loc, )) + hvm_params.extend([9, # HVM_PARAM_VIRIDIAN + loc]) + + elif marker == -12: # XC_SAVE_ID_COMPRESSED_DATA + sz, = unpack_exact("I") + data = rdexact(sz) + info(" Compressed Data: sz 0x%x" % (sz, )) + raise RuntimeError("todo") + + elif marker == -13: # XC_SAVE_ID_ENABLE_COMPRESSION + raise RuntimeError("todo") + + elif marker == -14: # XC_SAVE_ID_HVM_GENERATION_ID_ADDR + _, genid_loc = unpack_exact("=IQ") + info(" Generation ID Address 0x%x" % (genid_loc, )) + hvm_params.extend([32, # HVM_PARAM_VM_GENERATION_ID_ADDR + genid_loc]) + + elif marker == -15: # XC_SAVE_ID_HVM_PAGING_RING_PFN + _, paging_ring_pfn = unpack_exact("=IQ") + info(" Paging ring pfn 0x%x" % (paging_ring_pfn, )) + hvm_params.extend([27, # HVM_PARAM_PAGING_RING_PFN + paging_ring_pfn]) + + elif marker == -16: # XC_SAVE_ID_HVM_ACCESS_RING_PFN + _, access_ring_pfn = unpack_exact("=IQ") + info(" Access ring pfn 0x%x" % (access_ring_pfn, )) + hvm_params.extend([28, # HVM_PARAM_ACCESS_RING_PFN + access_ring_pfn]) + + elif marker == -17: # XC_SAVE_ID_HVM_SHARING_RING_PFN + _, sharing_ring_pfn = unpack_exact("=IQ") + info(" Sharing ring pfn 0x%x" % (sharing_ring_pfn, )) + hvm_params.extend([29, # HVM_PARAM_SHARING_RING_PFN + sharing_ring_pfn]) + + elif marker == -18: + sz, = unpack_exact("I") + data = rdexact(sz) + info(" Toolstack Data: sz 0x%x" % (sz, )) + err("TODO - fix libxl's use of this") + write_record(sv2.REC_TYPE_toolstack, data) + + else: + raise StreamError("Unrecognised chunk") + +def read_hvm_tail(vm): + + io, bufio, store = unpack_exact("QQQ") + info("Magic pfns: 0x%x 0x%x 0x%x" % (io, bufio, store)) + write_hvm_params([5, io, # HVM_PARAM_IOREQ_PFN + 6, bufio, # HVM_PARAM_BUFIOREQ_PFN + 1, store]) # HVM_PARAM_STORE_PFN + + blobsz, = unpack_exact("I") + info("Got HVM Context (0x%x bytes)" % (blobsz, )) + blob = rdexact(blobsz) + + write_record(sv2.REC_TYPE_hvm_context, blob) + write_record(sv2.REC_TYPE_end, "") + + + +def read_qemu(vm): + + rawsig = rdexact(21) + sig, = struct.unpack("21s", rawsig) + info("Qemu signature: %s" % (sig, )) + + if sig == "DeviceModelRecord0002": + rawsz = rdexact(4) + sz, = struct.unpack("I", rawsz) + qdata = rdexact(sz) + + stream_write(rawsig) + stream_write(rawsz) + stream_write(qdata) + + else: + raise RuntimeError("Unrecognised Qemu sig '%s'" % (sig, )) + + +def read_vm(vm): + + try: + + vm.p2m_size, = unpack_ulongs(1) + info("P2M Size: 0x%x" % (vm.p2m_size,)) + + write_ihdr() + write_dhdr() + + if pv: + read_pv_extended_info(vm) + read_pv_p2m_frames(vm) + + read_chunks(vm) + + if pv: + read_pv_tail(vm) + else: + read_hvm_tail(vm) + + if qemu: + read_qemu(vm) + + except (IOError, StreamError, ): + err("Stream Error:") + err(traceback.format_exc()) + return 1 + + except RuntimeError: + err("Script Error:") + err(traceback.format_exc()) + err("Please fix me") + return 2 + return 0 + +def open_file_or_fd(val, mode): + """ + If 'val' looks like a decimal integer, open it as an fd. If not, try to + open it as a regular file. + """ + + fd = -1 + try: + # Does it look like an integer? + try: + fd = int(val, 10) + except ValueError: + pass + + # Try to open it... + if fd != -1: + return os.fdopen(fd, mode, 0) + else: + return open(val, mode, 0) + + except StandardError, e: + if fd != -1: + err("Unable to open fd %d: %s" % (fd, e)) + else: + err("Unable to open file '%s': %s" % (val, e)) + + raise SystemExit(1) + + +def main(argv): + from optparse import OptionParser + global fin, fout, twidth, pv, qemu, verbose + + # Change stdout to be line-buffered. + sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 1) + + parser = OptionParser(version = __version__, + usage = ("%prog [options] -i INPUT -o OUTPUT" + " -w WIDTH -g GUEST"), + description = + "Convert a legacy stream to a v2 stream") + + # Required options + parser.add_option("-i", "--in", dest = "fin", metavar = "<FD or FILE>", + help = "Legacy input to convert") + parser.add_option("-o", "--out", dest = "fout", metavar = "<FD or FILE>", + help = "v2 format output") + parser.add_option("-w", "--width", dest = "twidth", + metavar = "<32/64>", choices = ["32", "64"], + help = "Legacy toolstack bitness") + parser.add_option("-g", "--guest-type", dest = "gtype", + metavar = "<pv/hvm>", choices = ["pv", "hvm"], + help = "Type of guest in stream") + + # Optional options + parser.add_option("-v", "--verbose", action = "store_true", default = False, + help = "Summarise stream contents") + parser.add_option("-x", "--xl", action = "store_true", default = False, + help = ("Is an `xl` header present in the stream?" + " (default no)")) + parser.add_option("--skip-qemu", action = "store_true", default = False, + help = ("Skip processing of the qemu tail?" + " (default no)")) + parser.add_option("--syslog", action = "store_true", default = False, + help = "Log to syslog instead of stdout/stderr") + + opts, _ = parser.parse_args() + + if (opts.fin is None or opts.fout is None or + opts.twidth is None or opts.gtype is None): + + parser.print_help(sys.stderr) + raise SystemExit(1) + + if opts.syslog: + global log_to_syslog + + syslog.openlog(os.path.basename(argv[0])) + log_to_syslog = True + + fin = open_file_or_fd(opts.fin, "rb") + fout = open_file_or_fd(opts.fout, "wb") + twidth = int(opts.twidth) + pv = opts.gtype == "pv" + verbose = opts.verbose + if opts.skip_qemu: + qemu = False + + if opts.xl: + skip_xl_header() + + rc = read_vm(VM()) + fout.close() + + return rc + +if __name__ == "__main__": + try: + sys.exit(main(sys.argv)) + except SystemExit, e: + sys.exit(e.code) + except KeyboardInterrupt: + sys.exit(1) diff --git a/tools/python/scripts/verify-stream-v2.py b/tools/python/scripts/verify-stream-v2.py new file mode 100755 index 0000000..3e70527 --- /dev/null +++ b/tools/python/scripts/verify-stream-v2.py @@ -0,0 +1,456 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +""" Verify a v2 format migration stream """ + +import sys +import struct +import os, os.path +import syslog +import traceback + +from xen.streamv2.format import * + +fin = None # Input file/fd +log_to_syslog = False # Boolean - Log to syslog instead of stdout/err? +verbose = False # Boolean - Summarise stream contents +quiet = False # Boolean - Suppress error printing + +def info(msg): + """Info message, routed to appropriate destination""" + if not quiet and verbose: + if log_to_syslog: + for line in msg.split("\n"): + syslog.syslog(syslog.LOG_INFO, line) + else: + print msg + +def err(msg): + """Error message, routed to appropriate destination""" + if not quiet: + if log_to_syslog: + for line in msg.split("\n"): + syslog.syslog(syslog.LOG_ERR, line) + else: + print >> sys.stderr, msg + +def stream_read(_ = None): + """Read from input""" + return fin.read(_) + +def rdexact(nr_bytes): + """Read exactly nr_bytes from fin""" + _ = stream_read(nr_bytes) + if len(_) != nr_bytes: + raise IOError("Stream truncated") + return _ + +def unpack_exact(fmt): + """Unpack a format from fin""" + sz = struct.calcsize(fmt) + return struct.unpack(fmt, rdexact(sz)) + +class StreamError(StandardError): + """Error with the stream""" + pass + +class RecordError(StandardError): + """Error with a record in the stream""" + pass + +def skip_xl_header(): + """Skip over an xl header in the stream""" + + hdr = rdexact(32) + if hdr != "Xen saved domain, xl format\n \0 \r": + raise StreamError("No xl header") + + _, _, _, optlen = unpack_exact("=IIII") + _ = rdexact(optlen) + + info("Skipped xl header") + +def verify_ihdr(): + """ Verify an image header """ + + marker, ident, version, options, res1, res2 = unpack_exact(IHDR_FORMAT) + + if marker != IHDR_MARKER: + raise StreamError("Bad image marker: Expected 0x%x, got 0x%x" + % (IHDR_MARKER, marker)) + + if ident != IHDR_IDENT: + raise StreamError("Bad image id: Expected 0x%x, got 0x%x" + % (IHDR_IDENT, ident)) + + if version != 2: + raise StreamError("Unknown image version: Expected 2, got %d" + % (version, )) + + if options & IHDR_OPT_RESZ_MASK: + raise StreamError("Reserved bits set in image options field: 0x%x" + % (options & IHDR_OPT_RESZ_MASK)) + + if res1 != 0 or res2 != 0: + raise StreamError("Reserved bits set in image header: 0x%04x:0x%08x" + % (res1, res2)) + + if ( sys.byteorder == "little" and + (options & IHDR_OPT_ENDIAN_) != IHDR_OPT_LE ): + raise StreamError("Stream is not native endianess - unable to validate") + + if options & IHDR_OPT_BE: + info("Image Header: big endian") + else: + info("Image Header: little endian") + +def verify_dhdr(): + """ Verify a domain header """ + + gtype, page_shift, res1, major, minor = unpack_exact(DHDR_FORMAT) + + if gtype not in dhdr_type_to_str: + raise StreamError("Unrecognised domain type 0x%x" % (gtype, )) + + if res1 != 0: + raise StreamError("Reserved bits set in domain header 0x%04x" + % (res1, )) + + if page_shift != 12: + raise StreamError("Page shift expected to be 12. Got %d" + % (page_shift, )) + + if major == 0: + info("Domain Header: legacy converted %s" + % (dhdr_type_to_str[gtype], )) + else: + info("Domain Header: %s from Xen %d.%d" + % (dhdr_type_to_str[gtype], major, minor)) + + +def verify_record_end(content): + """ End record """ + + if len(content) != 0: + raise RecordError("End record with non-zero length") + +def verify_page_data(content): + """ Page Data record """ + minsz = struct.calcsize(PAGE_DATA_FORMAT) + + if len(content) <= minsz: + raise RecordError("PAGE_DATA record must be at least %d bytes long" + % (minsz, )) + + count, res1 = struct.unpack(PAGE_DATA_FORMAT, content[:minsz]) + + if res1 != 0: + raise StreamError("Reserved bits set in PAGE_DATA record 0x%04x" + % (res1, )) + + pfnsz = count * 8 + if (len(content) - minsz) < pfnsz: + raise RecordError("PAGE_DATA record must contain a pfn record for " + "each count") + + pfns = list(struct.unpack("=%dQ" % (count,), content[minsz:minsz + pfnsz])) + + nr_pages = 0 + for idx, pfn in enumerate(pfns): + + if pfn & PAGE_DATA_PFN_RESZ_MASK: + raise RecordError("Reserved bits set in pfn[%d]: 0x%016x", + idx, pfn & PAGE_DATA_PFN_RESZ_MASK) + + if pfn >> PAGE_DATA_TYPE_SHIFT in (5, 6, 7, 8): + raise RecordError("Invalid type value in pfn[%d]: 0x%016x", + idx, pfn & PAGE_DATA_TYPE_LTAB_MASK) + + # We expect page data for each normal page or pagetable + if PAGE_DATA_TYPE_NOTAB <= (pfn & PAGE_DATA_TYPE_LTABTYPE_MASK) \ + <= PAGE_DATA_TYPE_L4TAB: + nr_pages += 1 + + pagesz = nr_pages * 4096 + if len(content) != minsz + pfnsz + pagesz: + raise RecordError("Expected %u + %u + %u, got %u" + % (minsz, pfnsz, pagesz, len(content))) + + +def verify_record_x86_pv_vcpu_generic(content, name): + """ Generic for all REC_TYPE_x86_pv_vcpu_{basic,extended,xsave,msrs} """ + minsz = struct.calcsize(X86_PV_VCPU_HDR_FORMAT) + + if len(content) <= minsz: + raise RecordError("X86_PV_VCPU_%s record length must be at least %d" + " bytes long" % (name, minsz)) + + vcpuid, res1 = struct.unpack(X86_PV_VCPU_HDR_FORMAT, content[:minsz]) + + if res1 != 0: + raise StreamError("Reserved bits set in x86_pv_vcpu_%s record 0x%04x" + % (name, res1)) + + info(" vcpu%d %s context, %d bytes" % (vcpuid, name, len(content) - minsz)) + + +def verify_x86_pv_info(content): + """ x86 PV Info record """ + + expectedsz = struct.calcsize(X86_PV_INFO_FORMAT) + if len(content) != expectedsz: + raise RecordError("x86_pv_info: expected length of %d, got %d" + % (expectedsz, len(content))) + + width, levels, res1, res2 = struct.unpack(X86_PV_INFO_FORMAT, content) + + if width not in (4, 8): + raise RecordError("Expected width of 4 or 8, got %d" % (width, )) + + if levels not in (3, 4): + raise RecordError("Expected levels of 3 or 4, got %d" % (levels, )) + + if res1 != 0 or res2 != 0: + raise StreamError("Reserved bits set in X86_PV_INFO: 0x%04x 0x%08x" + % (res1, res2)) + + bitness = {4:32, 8:64}[width] + + info(" %sbit guest, %d levels of pagetables" % (bitness, levels)) + +def verify_x86_pv_p2m_frames(content): + """ x86 PV p2m frames record """ + + if len(content) % 8 != 0: + raise RecordError("Length expected to be a multiple of 8, not %d" + % (len(content), )) + + start, end = struct.unpack("=II", content[:8]) + + info(" Start pfn 0x%x, End 0x%x" % (start, end)) + +def verify_record_shared_info(content): + """ shared info record """ + + if len(content) != 4096: + raise RecordError("Length expected to be 4906 bytes, not %d" + % (len(content), )) + +def verify_record_tsc_info(content): + """ tsc info record """ + + sz = struct.calcsize(TSC_INFO_FORMAT) + + if len(content) != sz: + raise RecordError("Length should be %u bytes" % (sz, )) + + mode, khz, nsec, incarn, res1 = struct.unpack(TSC_INFO_FORMAT, content) + + if res1 != 0: + raise StreamError("Reserved bits set in TSC_INFO: 0x%08x" % (res1, )) + + info(" Mode %u, %u kHz, %u ns, incarnation %d" % (mode, khz, nsec, incarn)) + +def verify_record_hvm_context(content): + """ hvm context record """ + + if len(content) == 0: + raise RecordError("Zero length HVM context") + +def verify_record_hvm_params(content): + """ hvm params record """ + + sz = struct.calcsize(HVM_PARAMS_FORMAT) + + if len(content) < sz: + raise RecordError("Length should be at least %u bytes" % (sz, )) + + count, rsvd = struct.unpack(HVM_PARAMS_FORMAT, content[:sz]) + + if rsvd != 0: + raise RecordError("Reserved field not zero (0x%04x)" % (rsvd, )) + + sz += count * struct.calcsize(HVM_PARAMS_ENTRY_FORMAT) + + if len(content) != sz: + raise RecordError("Length should be %u bytes" % (sz, )) + +def verify_toolstack(_): + """ toolstack record """ + info(" TODO: remove") + +def verify_record_verify(content): + """ verify record """ + + if len(content) != 0: + raise RecordError("Verify record with non-zero length") + +def verify_record_saving_cpu(content): + """ saving cpu record """ + info(" Saving cpu: '%s'" % (content,)) + +record_verifiers = { + REC_TYPE_end : verify_record_end, + REC_TYPE_page_data : verify_page_data, + + REC_TYPE_x86_pv_info: verify_x86_pv_info, + REC_TYPE_x86_pv_p2m_frames: verify_x86_pv_p2m_frames, + + REC_TYPE_x86_pv_vcpu_basic : + lambda x: verify_record_x86_pv_vcpu_generic(x, "basic"), + REC_TYPE_x86_pv_vcpu_extended : + lambda x: verify_record_x86_pv_vcpu_generic(x, "extended"), + REC_TYPE_x86_pv_vcpu_xsave : + lambda x: verify_record_x86_pv_vcpu_generic(x, "xsave"), + REC_TYPE_x86_pv_vcpu_msrs : + lambda x: verify_record_x86_pv_vcpu_generic(x, "msrs"), + + REC_TYPE_shared_info: verify_record_shared_info, + REC_TYPE_tsc_info: verify_record_tsc_info, + + REC_TYPE_hvm_context: verify_record_hvm_context, + REC_TYPE_hvm_params: verify_record_hvm_params, + REC_TYPE_toolstack: verify_toolstack, + REC_TYPE_verify: verify_record_verify, + + REC_TYPE_saving_cpu: verify_record_saving_cpu, +} + +squahsed_page_data_records = 0 +def verify_record(): + """ Verify a record """ + global squahsed_page_data_records + + rtype, length = unpack_exact(RH_FORMAT) + + if rtype not in rec_type_to_str: + raise StreamError("Unrecognised record type %x" % (rtype, )) + + contentsz = (length + 7) & ~7 + content = rdexact(contentsz) + + padding = content[length:] + if padding != "\x00" * len(padding): + raise StreamError("Padding containing non0 bytes found") + + if rtype != REC_TYPE_page_data: + + if squahsed_page_data_records > 0: + info("Squashed %d Page Data records together" + % (squahsed_page_data_records, )) + squahsed_page_data_records = 0 + + info("Record: %s, length %d" % (rec_type_to_str[rtype], length)) + + else: + squahsed_page_data_records += 1 + + if rtype not in record_verifiers: + raise RuntimeError("No verification function") + else: + record_verifiers[rtype](content[:length]) + + return rtype + +def read_stream(): + """ Read an entire stream """ + + try: + verify_ihdr() + verify_dhdr() + + while verify_record() != REC_TYPE_end: + pass + + except (IOError, StreamError, RecordError): + err("Stream Error:") + err(traceback.format_exc()) + return 1 + + except StandardError: + err("Script Error:") + err(traceback.format_exc()) + err("Please fix me") + return 2 + + return 0 + +def open_file_or_fd(val, mode, buffering): + """ + If 'val' looks like a decimal integer, open it as an fd. If not, try to + open it as a regular file. + """ + + fd = -1 + try: + # Does it look like an integer? + try: + fd = int(val, 10) + except ValueError: + pass + + # Try to open it... + if fd != -1: + return os.fdopen(fd, mode, buffering) + else: + return open(val, mode, buffering) + + except StandardError, e: + if fd != -1: + err("Unable to open fd %d: %s: %s" % + (fd, e.__class__.__name__, e)) + else: + err("Unable to open file '%s': %s: %s" % + (val, __class__.__name__, e)) + + raise SystemExit(2) + +def main(argv): + from optparse import OptionParser + global fin, quiet, verbose + + # Change stdout to be line-buffered. + sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 1) + + parser = OptionParser(usage = "%prog [options]", + description = + "Verify a stream according to the v2 spec") + + # Optional options + parser.add_option("-i", "--in", dest = "fin", metavar = "<FD or FILE>", + default = "0", + help = "v2 format stream to verify (defaults to stdin)") + parser.add_option("-v", "--verbose", action = "store_true", default = False, + help = "Summarise stream contents") + parser.add_option("-q", "--quiet", action = "store_true", default = False, + help = "Suppress all logging/errors") + parser.add_option("-x", "--xl", action = "store_true", default = False, + help = ("Is an `xl` header present in the stream?" + " (default no)")) + parser.add_option("--syslog", action = "store_true", default = False, + help = "Log to syslog instead of stdout/stderr") + + opts, _ = parser.parse_args() + + if opts.syslog: + global log_to_syslog + + syslog.openlog(os.path.basename(argv[0])) + log_to_syslog = True + + verbose = opts.verbose + quiet = opts.quiet + fin = open_file_or_fd(opts.fin, "rb", 0) + + if opts.xl: + skip_xl_header() + + return read_stream() + +if __name__ == "__main__": + try: + sys.exit(main(sys.argv)) + except SystemExit, e: + sys.exit(e.code) + except KeyboardInterrupt: + sys.exit(2) diff --git a/tools/python/setup.py b/tools/python/setup.py index 17ebb4a..52c6118 100644 --- a/tools/python/setup.py +++ b/tools/python/setup.py @@ -43,6 +43,7 @@ setup(name = 'xen', version = '3.0', description = 'Xen', packages = ['xen', + 'xen.streamv2', 'xen.lowlevel', ], ext_package = "xen.lowlevel", diff --git a/tools/python/xen/streamv2/__init__.py b/tools/python/xen/streamv2/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tools/python/xen/streamv2/format.py b/tools/python/xen/streamv2/format.py new file mode 100644 index 0000000..25b8638 --- /dev/null +++ b/tools/python/xen/streamv2/format.py @@ -0,0 +1,148 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +# Python structures for the Migration v2 stream format. +# See docs/specs/libxc-migration-stream.pandoc + +# Image Header +IHDR_FORMAT = "!QIIHHI" + +IHDR_MARKER = 0xffffffffffffffff +IHDR_IDENT = 0x58454E46 # "XENF" in ASCII +IHDR_VERSION = 2 + +IHDR_OPT_ENDIAN_ = 0 +IHDR_OPT_LE = (0 << IHDR_OPT_ENDIAN_) +IHDR_OPT_BE = (1 << IHDR_OPT_ENDIAN_) + +IHDR_OPT_RESZ_MASK = 0xfffe + +# Domain Header +DHDR_FORMAT = "IHHII" + +DHDR_TYPE_x86_pv = 0x00000001 +DHDR_TYPE_x86_hvm = 0x00000002 +DHDR_TYPE_x86_pvh = 0x00000003 +DHDR_TYPE_arm = 0x00000004 + +dhdr_type_to_str = { + DHDR_TYPE_x86_pv : "x86 PV", + DHDR_TYPE_x86_hvm : "x86 HVM", + DHDR_TYPE_x86_pvh : "x86 PVH", + DHDR_TYPE_arm : "ARM", +} + +RH_FORMAT = "II" + +REC_TYPE_end = 0x00000000 +REC_TYPE_page_data = 0x00000001 +REC_TYPE_x86_pv_info = 0x00000002 +REC_TYPE_x86_pv_p2m_frames = 0x00000003 +REC_TYPE_x86_pv_vcpu_basic = 0x00000004 +REC_TYPE_x86_pv_vcpu_extended = 0x00000005 +REC_TYPE_x86_pv_vcpu_xsave = 0x00000006 +REC_TYPE_shared_info = 0x00000007 +REC_TYPE_tsc_info = 0x00000008 +REC_TYPE_hvm_context = 0x00000009 +REC_TYPE_hvm_params = 0x0000000a +REC_TYPE_toolstack = 0x0000000b +REC_TYPE_x86_pv_vcpu_msrs = 0x0000000c +REC_TYPE_verify = 0x0000000d + +REC_TYPE_saving_cpu = 0x80000000 + +rec_type_to_str = { + REC_TYPE_end : "End", + REC_TYPE_page_data : "Page data", + REC_TYPE_x86_pv_info : "x86 PV info", + REC_TYPE_x86_pv_p2m_frames : "x86 PV P2M frames", + REC_TYPE_x86_pv_vcpu_basic : "x86 PV vcpu basic", + REC_TYPE_x86_pv_vcpu_extended : "x86 PV vcpu extended", + REC_TYPE_x86_pv_vcpu_xsave : "x86 PV vcpu xsave", + REC_TYPE_shared_info : "Shared info", + REC_TYPE_tsc_info : "TSC info", + REC_TYPE_hvm_context : "HVM context", + REC_TYPE_hvm_params : "HVM params", + REC_TYPE_toolstack : "Toolstack", + REC_TYPE_x86_pv_vcpu_msrs : "x86 PV vcpu msrs", + REC_TYPE_verify : "Verify", + + REC_TYPE_saving_cpu : "Saving CPU", +} + +# page_data +PAGE_DATA_FORMAT = "II" +PAGE_DATA_PFN_MASK = (1L << 52) - 1 +PAGE_DATA_PFN_RESZ_MASK = ((1L << 60) - 1) & ~((1L << 52) - 1) + +# flags from xen/public/domctl.h: XEN_DOMCTL_PFINFO_* shifted by 32 bits +PAGE_DATA_TYPE_SHIFT = 60 +PAGE_DATA_TYPE_LTABTYPE_MASK = (0x7L << PAGE_DATA_TYPE_SHIFT) +PAGE_DATA_TYPE_LTAB_MASK = (0xfL << PAGE_DATA_TYPE_SHIFT) +PAGE_DATA_TYPE_LPINTAB = (0x8L << PAGE_DATA_TYPE_SHIFT) # Pinned pagetable + +PAGE_DATA_TYPE_NOTAB = (0x0L << PAGE_DATA_TYPE_SHIFT) # Regular page +PAGE_DATA_TYPE_L1TAB = (0x1L << PAGE_DATA_TYPE_SHIFT) # L1 pagetable +PAGE_DATA_TYPE_L2TAB = (0x2L << PAGE_DATA_TYPE_SHIFT) # L2 pagetable +PAGE_DATA_TYPE_L3TAB = (0x3L << PAGE_DATA_TYPE_SHIFT) # L3 pagetable +PAGE_DATA_TYPE_L4TAB = (0x4L << PAGE_DATA_TYPE_SHIFT) # L4 pagetable +PAGE_DATA_TYPE_BROKEN = (0xdL << PAGE_DATA_TYPE_SHIFT) # Broken +PAGE_DATA_TYPE_XALLOC = (0xeL << PAGE_DATA_TYPE_SHIFT) # Allocate-only +PAGE_DATA_TYPE_XTAB = (0xfL << PAGE_DATA_TYPE_SHIFT) # Invalid + +# x86_pv_info +X86_PV_INFO_FORMAT = "BBHI" + +X86_PV_P2M_FRAMES_FORMAT = "II" + +# x86_pv_vcpu_{basic,extended,xsave,msrs} +X86_PV_VCPU_HDR_FORMAT = "II" + +# tsc_info +TSC_INFO_FORMAT = "IIQII" + +# hvm_params +HVM_PARAMS_ENTRY_FORMAT = "QQ" +HVM_PARAMS_FORMAT = "II" + +# +# libxl format +# + +LIBXL_QEMU_SIGNATURE = "DeviceModelRecord0002" +LIBXL_QEMU_RECORD_HDR = "=%dsI" % (len(LIBXL_QEMU_SIGNATURE), ) + + +def __main(): + """ Basic sanity check for format sizes """ + + import sys + from struct import calcsize + + ok = True + for fmt, sz in [ ("IHDR_FORMAT", 24), + ("DHDR_FORMAT", 16), + ("RH_FORMAT", 8), + + ("PAGE_DATA_FORMAT", 8), + ("X86_PV_INFO_FORMAT", 8), + ("X86_PV_P2M_FRAMES_FORMAT", 8), + ("X86_PV_VCPU_HDR_FORMAT", 8), + ("TSC_INFO_FORMAT", 24), + ("HVM_PARAMS_ENTRY_FORMAT", 16), + ("HVM_PARAMS_FORMAT", 8), + ]: + + realsz = calcsize(getattr(sys.modules[__name__], fmt)) + if realsz != sz: + print "%s is %d bytes but expected %d" % (fmt, realsz, sz) + ok = False + + if ok: + sys.exit(0) + else: + sys.exit(1) + +# If run as a python script alone, confirm some expected sizes +if __name__ == "__main__": + __main() -- 1.7.10.4 _______________________________________________ Xen-devel mailing list Xen-devel@xxxxxxxxxxxxx http://lists.xen.org/xen-devel
|
Lists.xenproject.org is hosted with RackSpace, monitoring our |