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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
#! /usr/bin/env python
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
'''Testing for the JSON file emitted by DMD heap scan mode when running SmokeDMD.'''
from __future__ import print_function, division
import argparse
import gzip
import json
import sys
# The DMD output version this script handles.
outputVersion = 5
def parseCommandLine():
description = '''
Ensure that DMD heap scan mode creates the correct output when run with SmokeDMD.
This is only for testing. Input files can be gzipped.
'''
p = argparse.ArgumentParser(description=description)
p.add_argument('--clamp-contents', action='store_true',
help='expect that the contents of the JSON input file have had their addresses clamped')
p.add_argument('input_file',
help='a file produced by DMD')
return p.parse_args(sys.argv[1:])
def checkScanContents(contents, expected):
if len(contents) != len(expected):
raise Exception("Expected " + str(len(expected)) + " things in contents but found " + str(len(contents)))
for i in range(len(expected)):
if contents[i] != expected[i]:
raise Exception("Expected to find " + expected[i] + " at offset " + str(i) + " but found " + contents[i])
def main():
args = parseCommandLine()
# Handle gzipped input if necessary.
isZipped = args.input_file.endswith('.gz')
opener = gzip.open if isZipped else open
with opener(args.input_file, 'rb') as f:
j = json.load(f)
if j['version'] != outputVersion:
raise Exception("'version' property isn't '{:d}'".format(outputVersion))
invocation = j['invocation']
mode = invocation['mode']
if mode != 'scan':
raise Exception("bad 'mode' property: '{:s}'".format(mode))
blockList = j['blockList']
if len(blockList) != 1:
raise Exception("Expected only one block")
b = blockList[0]
# The expected values are based on hard-coded values in SmokeDMD.cpp.
if args.clamp_contents:
expected = ['0', '0', '0', b['addr'], b['addr']]
else:
addr = int(b['addr'], 16)
expected = ['123', '0', str(format(addr - 1, 'x')), b['addr'],
str(format(addr + 1, 'x')), '0']
checkScanContents(b['contents'], expected)
if __name__ == '__main__':
main()
|
