summaryrefslogtreecommitdiff
path: root/lldb/test/API/functionalities/postmortem/FreeBSDKernel/tools/copy-sparse.py
blob: c572351749131dd39069b18929e53683fcc3de7a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#!/usr/bin/env python

import argparse
import re
import sys


def main():
    argp = argparse.ArgumentParser()
    argp.add_argument('infile', type=argparse.FileType('rb'),
                      help='Input vmcore file')
    argp.add_argument('outfile', type=argparse.FileType('wb'),
                      help='Output vmcore file')
    args = argp.parse_args()

    inf = args.infile
    outf = args.outfile
    line_re = re.compile(r"^% RD: (\d+) (\d+)")

    # copy the first chunk that usually includes ELF headers
    # (not output by patched libfbsdvmcore since libelf reads this)
    outf.write(inf.read(1024))

    for l in sys.stdin:
        m = line_re.match(l)
        offset, size = [int(x) for x in m.groups()]

        inf.seek(offset)
        outf.seek(offset)
        outf.write(inf.read(size))


if __name__ == "__main__":
    main()