1import lldb
2from intelpt_testcase import *
3from lldbsuite.test.lldbtest import *
4from lldbsuite.test import lldbutil
5from lldbsuite.test.decorators import *
6
7class TestTraceLoad(TraceIntelPTTestCaseBase):
8
9    mydir = TestBase.compute_mydir(__file__)
10    NO_DEBUG_INFO_TESTCASE = True
11
12    def testLoadTrace(self):
13        src_dir = self.getSourceDir()
14        trace_definition_file = os.path.join(src_dir, "intelpt-trace", "trace.json")
15        self.expect("trace load -v " + trace_definition_file, substrs=["intel-pt"])
16
17        target = self.dbg.GetSelectedTarget()
18        process = target.GetProcess()
19        self.assertEqual(process.GetProcessID(), 1234)
20
21        self.assertEqual(process.GetNumThreads(), 1)
22        self.assertEqual(process.GetThreadAtIndex(0).GetThreadID(), 3842849)
23
24        self.assertEqual(target.GetNumModules(), 1)
25        module = target.GetModuleAtIndex(0)
26        path = module.GetFileSpec()
27        self.assertEqual(path.fullpath, os.path.join(src_dir, "intelpt-trace", "a.out"))
28        self.assertGreater(module.GetNumSections(), 0)
29        self.assertEqual(module.GetSectionAtIndex(0).GetFileAddress(), 0x400000)
30
31        self.assertEqual("6AA9A4E2-6F28-2F33-377D-59FECE874C71-5B41261A", module.GetUUIDString())
32
33        # check that the Process and Thread objects were created correctly
34        self.expect("thread info", substrs=["tid = 3842849"])
35        self.expect("thread list", substrs=["Process 1234 stopped", "tid = 3842849"])
36        self.expect("thread trace dump info", substrs=['''Trace technology: intel-pt
37
38thread #1: tid = 3842849
39  Total number of instructions: 21
40
41  Memory usage:
42    Raw trace size: 4 KiB
43    Total approximate memory usage (excluding raw trace): 1.27 KiB
44    Average memory usage per instruction (excluding raw trace): 61.76 bytes
45
46  Timing:
47    Decoding instructions: ''', '''s
48
49  Events:
50    Number of instructions with events: 1
51    Number of individual events: 1
52      paused: 1
53
54  Errors:
55    Number of TSC decoding errors: 0'''])
56
57    def testLoadInvalidTraces(self):
58        src_dir = self.getSourceDir()
59        # We test first an invalid type
60        self.expect("trace load -v " + os.path.join(src_dir, "intelpt-trace", "trace_bad.json"), error=True,
61          substrs=['''error: expected object at traceSession.processes[0]
62
63Context:
64{
65  "processes": [
66    /* error: expected object */
67    123
68  ],
69  "trace": { ... }
70}
71
72Schema:
73{
74  "trace": {
75    "type": "intel-pt",
76    "cpuInfo": {
77      "vendor": "intel" | "unknown",
78      "family": integer,
79      "model": integer,
80      "stepping": integer
81    }
82  },'''])
83
84        # Now we test a missing field in the global session file
85        self.expect("trace load -v " + os.path.join(src_dir, "intelpt-trace", "trace_bad2.json"), error=True,
86            substrs=['error: missing value at traceSession.processes[1].triple', "Context", "Schema"])
87
88        # Now we test a missing field in the intel-pt settings
89        self.expect("trace load -v " + os.path.join(src_dir, "intelpt-trace", "trace_bad4.json"), error=True,
90            substrs=['''error: missing value at traceSession.trace.cpuInfo.family
91
92Context:
93{
94  "processes": [],
95  "trace": {
96    "cpuInfo": /* error: missing value */ {
97      "model": 79,
98      "stepping": 1,
99      "vendor": "intel"
100    },
101    "type": "intel-pt"
102  }
103}''', "Schema"])
104
105        # Now we test an incorrect load address in the intel-pt settings
106        self.expect("trace load -v " + os.path.join(src_dir, "intelpt-trace", "trace_bad5.json"), error=True,
107            substrs=['error: expected numeric string at traceSession.processes[0].modules[0].loadAddress',
108                     '"loadAddress": /* error: expected numeric string */ 400000,', "Schema"])
109
110        # The following wrong schema will have a valid target and an invalid one. In the case of failure,
111        # no targets should be created.
112        self.assertEqual(self.dbg.GetNumTargets(), 0)
113        self.expect("trace load -v " + os.path.join(src_dir, "intelpt-trace", "trace_bad3.json"), error=True,
114            substrs=['error: missing value at traceSession.processes[1].pid'])
115        self.assertEqual(self.dbg.GetNumTargets(), 0)
116