From 728dd9a060fb5c665e39616a699b4e5a041e2bf0 Mon Sep 17 00:00:00 2001 From: Dean Michael Berris Date: Thu, 5 Oct 2017 05:18:17 +0000 Subject: [PATCH] [XRay][tools] Support arg1 logging entries in the basic logging mode Summary: The arg1 logging handler changed in compiler-rt to start writing a different type for entries encountered when logging the first argument of XRay-instrumented functions. This change allows the trace loader to support reading these record types as well as prepare for when the basic (naive) mode implementation starts writing down the argument payloads. Without this change, binaries with arg1 logging support enabled start writing unreadable logs for any of the XRay tracing tools. Reviewers: pelikan Subscribers: llvm-commits Differential Revision: https://reviews.llvm.org/D38550 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@314967 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/XRay/Trace.cpp | 72 +++++++++++++----- .../X86/Inputs/naive-with-arg1-entries.xray | Bin 0 -> 192 bytes .../X86/convert-basic-arg1-to-yaml.txt | 15 ++++ 3 files changed, 66 insertions(+), 21 deletions(-) create mode 100644 test/tools/llvm-xray/X86/Inputs/naive-with-arg1-entries.xray create mode 100644 test/tools/llvm-xray/X86/convert-basic-arg1-to-yaml.txt diff --git a/lib/XRay/Trace.cpp b/lib/XRay/Trace.cpp index e1eb7a7f117..e90396959fb 100644 --- a/lib/XRay/Trace.cpp +++ b/lib/XRay/Trace.cpp @@ -82,29 +82,59 @@ Error loadNaiveFormatLog(StringRef Data, XRayFileHeader &FileHeader, for (auto S = Data.drop_front(32); !S.empty(); S = S.drop_front(32)) { DataExtractor RecordExtractor(S, true, 8); uint32_t OffsetPtr = 0; - Records.emplace_back(); - auto &Record = Records.back(); - Record.RecordType = RecordExtractor.getU16(&OffsetPtr); - Record.CPU = RecordExtractor.getU8(&OffsetPtr); - auto Type = RecordExtractor.getU8(&OffsetPtr); - switch (Type) { - case 0: - Record.Type = RecordTypes::ENTER; - break; - case 1: - Record.Type = RecordTypes::EXIT; + switch (auto RecordType = RecordExtractor.getU16(&OffsetPtr)) { + case 0: { // Normal records. + Records.emplace_back(); + auto &Record = Records.back(); + Record.RecordType = RecordType; + Record.CPU = RecordExtractor.getU8(&OffsetPtr); + auto Type = RecordExtractor.getU8(&OffsetPtr); + switch (Type) { + case 0: + Record.Type = RecordTypes::ENTER; + break; + case 1: + Record.Type = RecordTypes::EXIT; + break; + case 2: + Record.Type = RecordTypes::TAIL_EXIT; + break; + case 3: + Record.Type = RecordTypes::ENTER_ARG; + break; + default: + return make_error( + Twine("Unknown record type '") + Twine(int{Type}) + "'", + std::make_error_code(std::errc::executable_format_error)); + } + Record.FuncId = RecordExtractor.getSigned(&OffsetPtr, sizeof(int32_t)); + Record.TSC = RecordExtractor.getU64(&OffsetPtr); + Record.TId = RecordExtractor.getU32(&OffsetPtr); break; - case 2: - Record.Type = RecordTypes::TAIL_EXIT; + } + case 1: { // Arg payload record. + auto &Record = Records.back(); + // Advance two bytes to avoid padding. + OffsetPtr += 2; + int32_t FuncId = RecordExtractor.getSigned(&OffsetPtr, sizeof(int32_t)); + auto TId = RecordExtractor.getU32(&OffsetPtr); + if (Record.FuncId != FuncId || Record.TId != TId) + return make_error( + Twine("Corrupted log, found payload following non-matching " + "function + thread record. Record for ") + + Twine(Record.FuncId) + " != " + Twine(FuncId), + std::make_error_code(std::errc::executable_format_error)); + // Advance another four bytes to avoid padding. + OffsetPtr += 4; + auto Arg = RecordExtractor.getU64(&OffsetPtr); + Record.CallArgs.push_back(Arg); break; + } default: return make_error( - Twine("Unknown record type '") + Twine(int{Type}) + "'", + Twine("Unknown record type == ") + Twine(RecordType), std::make_error_code(std::errc::executable_format_error)); } - Record.FuncId = RecordExtractor.getSigned(&OffsetPtr, sizeof(int32_t)); - Record.TSC = RecordExtractor.getU64(&OffsetPtr); - Record.TId = RecordExtractor.getU32(&OffsetPtr); } return Error::success(); } @@ -234,8 +264,8 @@ Error processCustomEventMarker(FDRState &State, uint8_t RecordFirstByte, uint32_t DataSize = RecordExtractor.getU32(&OffsetPtr); uint64_t TSC = RecordExtractor.getU64(&OffsetPtr); - // FIXME: Actually represent the record through the API. For now we only skip - // through the data. + // FIXME: Actually represent the record through the API. For now we only + // skip through the data. (void)TSC; RecordSize = 16 + DataSize; return Error::success(); @@ -507,8 +537,8 @@ Error loadYAMLLog(StringRef Data, XRayFileHeader &FileHeader, Records.clear(); std::transform(Trace.Records.begin(), Trace.Records.end(), std::back_inserter(Records), [&](const YAMLXRayRecord &R) { - return XRayRecord{R.RecordType, R.CPU, R.Type, - R.FuncId, R.TSC, R.TId, R.CallArgs}; + return XRayRecord{R.RecordType, R.CPU, R.Type, R.FuncId, + R.TSC, R.TId, R.CallArgs}; }); return Error::success(); } diff --git a/test/tools/llvm-xray/X86/Inputs/naive-with-arg1-entries.xray b/test/tools/llvm-xray/X86/Inputs/naive-with-arg1-entries.xray new file mode 100644 index 0000000000000000000000000000000000000000..026065902997bc5709c43a43bcc623b6f2754c00 GIT binary patch literal 192 zcmZQ#U|{$U1Pq7gUH~$&06_*upr}Nlv9+^O00XxYx)MP~ko