Skip to content

Commit

Permalink
[NFC][lldb][trace] Use uint64_t when decoding and enconding json
Browse files Browse the repository at this point in the history
llvm's json parser supports uint64_t, so let's better use it for the
packets being sent between lldb and lldb-server instead of using int64_t
as an intermediate type, which might be error-prone.
  • Loading branch information
walter-erquinigo committed May 10, 2022
1 parent 39e63bd commit 9d2dd6d
Show file tree
Hide file tree
Showing 7 changed files with 30 additions and 27 deletions.
14 changes: 7 additions & 7 deletions lldb/include/lldb/Utility/TraceGDBRemotePackets.h
Expand Up @@ -45,7 +45,7 @@ struct TraceStartRequest {

/// If \a llvm::None, then this starts tracing the whole process. Otherwise,
/// only tracing for the specified threads is enabled.
llvm::Optional<std::vector<int64_t>> tids;
llvm::Optional<std::vector<lldb::tid_t>> tids;

/// \return
/// \b true if \a tids is \a None, i.e. whole process tracing.
Expand Down Expand Up @@ -73,7 +73,7 @@ struct TraceStopRequest {
std::string type;
/// If \a llvm::None, then this stops tracing the whole process. Otherwise,
/// only tracing for the specified threads is stopped.
llvm::Optional<std::vector<int64_t>> tids;
llvm::Optional<std::vector<lldb::tid_t>> tids;
};

bool fromJSON(const llvm::json::Value &value, TraceStopRequest &packet,
Expand All @@ -98,7 +98,7 @@ struct TraceBinaryData {
/// Identifier of data to fetch with jLLDBTraceGetBinaryData.
std::string kind;
/// Size in bytes for this data.
int64_t size;
uint64_t size;
};

bool fromJSON(const llvm::json::Value &value, TraceBinaryData &packet,
Expand All @@ -107,7 +107,7 @@ bool fromJSON(const llvm::json::Value &value, TraceBinaryData &packet,
llvm::json::Value toJSON(const TraceBinaryData &packet);

struct TraceThreadState {
int64_t tid;
lldb::tid_t tid;
/// List of binary data objects for this thread.
std::vector<TraceBinaryData> binaryData;
};
Expand Down Expand Up @@ -161,11 +161,11 @@ struct TraceGetBinaryDataRequest {
/// Identifier for the data.
std::string kind;
/// Optional tid if the data is related to a thread.
llvm::Optional<int64_t> tid;
llvm::Optional<lldb::tid_t> tid;
/// Offset in bytes from where to start reading the data.
int64_t offset;
uint64_t offset;
/// Number of bytes to read.
int64_t size;
uint64_t size;
};

bool fromJSON(const llvm::json::Value &value,
Expand Down
9 changes: 6 additions & 3 deletions lldb/include/lldb/Utility/TraceIntelPTGDBRemotePackets.h
Expand Up @@ -16,6 +16,9 @@
#include <chrono>

/// See docs/lldb-gdb-remote.txt for more information.
///
/// Do not use system-dependent types, like size_t, because they might cause
/// issues when compiling on arm.
namespace lldb_private {

// List of data kinds used by jLLDBGetState and jLLDBGetBinaryData.
Expand All @@ -28,20 +31,20 @@ struct IntelPTDataKinds {
/// \{
struct TraceIntelPTStartRequest : TraceStartRequest {
/// Size in bytes to use for each thread's trace buffer.
int64_t trace_buffer_size;
uint64_t trace_buffer_size;

/// Whether to enable TSC
bool enable_tsc;

/// PSB packet period
llvm::Optional<int64_t> psb_period;
llvm::Optional<uint64_t> psb_period;

/// Required when doing "process tracing".
///
/// Limit in bytes on all the thread traces started by this "process trace"
/// instance. When a thread is about to be traced and the limit would be hit,
/// then a "tracing" stop event is triggered.
llvm::Optional<int64_t> process_buffer_size_limit;
llvm::Optional<uint64_t> process_buffer_size_limit;

/// Whether to have a trace buffer per thread or per cpu core.
llvm::Optional<bool> per_core_tracing;
Expand Down
9 changes: 4 additions & 5 deletions lldb/source/Plugins/Process/Linux/IntelPTCollector.cpp
Expand Up @@ -72,10 +72,9 @@ std::vector<TraceThreadState>
IntelPTThreadTraceCollection::GetThreadStates() const {
std::vector<TraceThreadState> states;
for (const auto &it : m_thread_traces)
states.push_back({static_cast<int64_t>(it.first),
states.push_back({it.first,
{TraceBinaryData{IntelPTDataKinds::kTraceBuffer,
static_cast<int64_t>(
it.second->GetTraceBufferSize())}}});
it.second->GetTraceBufferSize()}}});
return states;
}

Expand Down Expand Up @@ -201,8 +200,8 @@ Expected<json::Value> IntelPTCollector::GetState() const {
return cpu_info.takeError();

TraceGetStateResponse state;
state.processBinaryData.push_back({IntelPTDataKinds::kProcFsCpuInfo,
static_cast<int64_t>(cpu_info->size())});
state.processBinaryData.push_back(
{IntelPTDataKinds::kProcFsCpuInfo, cpu_info->size()});

std::vector<TraceThreadState> thread_states =
m_thread_traces.GetThreadStates();
Expand Down
6 changes: 4 additions & 2 deletions lldb/source/Plugins/Trace/intel-pt/TraceIntelPT.cpp
Expand Up @@ -305,7 +305,8 @@ Error TraceIntelPT::Start(size_t trace_buffer_size,
request.trace_buffer_size = trace_buffer_size;
request.process_buffer_size_limit = total_buffer_size_limit;
request.enable_tsc = enable_tsc;
request.psb_period = psb_period.map([](size_t val) { return (int64_t)val; });
request.psb_period =
psb_period.map([](size_t val) { return static_cast<uint64_t>(val); });
request.type = GetPluginName().str();
request.per_core_tracing = per_core_tracing;
return Trace::Start(toJSON(request));
Expand Down Expand Up @@ -342,7 +343,8 @@ llvm::Error TraceIntelPT::Start(llvm::ArrayRef<lldb::tid_t> tids,
TraceIntelPTStartRequest request;
request.trace_buffer_size = trace_buffer_size;
request.enable_tsc = enable_tsc;
request.psb_period = psb_period.map([](size_t val) { return (int64_t)val; });
request.psb_period =
psb_period.map([](size_t val) { return static_cast<uint64_t>(val); });
request.type = GetPluginName().str();
request.tids.emplace();
for (lldb::tid_t tid : tids)
Expand Down
7 changes: 3 additions & 4 deletions lldb/source/Target/Trace.cpp
Expand Up @@ -154,9 +154,8 @@ Trace::GetLiveThreadBinaryData(lldb::tid_t tid, llvm::StringRef kind) {
"Tracing data \"%s\" is not available for thread %" PRIu64 ".",
kind.data(), tid);

TraceGetBinaryDataRequest request{GetPluginName().str(), kind.str(),
static_cast<int64_t>(tid), 0,
static_cast<int64_t>(*size)};
TraceGetBinaryDataRequest request{GetPluginName().str(), kind.str(), tid, 0,
*size};
return m_live_process->TraceGetBinaryData(request);
}

Expand All @@ -172,7 +171,7 @@ Trace::GetLiveProcessBinaryData(llvm::StringRef kind) {
"Tracing data \"%s\" is not available for the process.", kind.data());

TraceGetBinaryDataRequest request{GetPluginName().str(), kind.str(), None, 0,
static_cast<int64_t>(*size)};
*size};
return m_live_process->TraceGetBinaryData(request);
}

Expand Down
2 changes: 1 addition & 1 deletion lldb/source/Utility/TraceGDBRemotePackets.cpp
Expand Up @@ -48,7 +48,7 @@ TraceStopRequest::TraceStopRequest(llvm::StringRef type,
: type(type) {
tids.emplace();
for (lldb::tid_t tid : tids_)
tids->push_back(static_cast<int64_t>(tid));
tids->push_back(tid);
}

bool TraceStopRequest::IsProcessTracing() const { return !(bool)tids; }
Expand Down
10 changes: 5 additions & 5 deletions lldb/source/Utility/TraceIntelPTGDBRemotePackets.cpp
Expand Up @@ -56,24 +56,24 @@ LinuxPerfZeroTscConversion::Convert(uint64_t raw_counter_value) {
json::Value LinuxPerfZeroTscConversion::toJSON() {
return json::Value(json::Object{
{"kind", "tsc-perf-zero-conversion"},
{"time_mult", static_cast<int64_t>(m_time_mult)},
{"time_shift", static_cast<int64_t>(m_time_shift)},
{"time_zero", static_cast<int64_t>(m_time_zero)},
{"time_mult", m_time_mult},
{"time_shift", m_time_shift},
{"time_zero", m_time_zero},
});
}

bool fromJSON(const json::Value &value, TraceTscConversionUP &tsc_conversion,
json::Path path) {
ObjectMapper o(value, path);

int64_t time_mult, time_shift, time_zero;
uint64_t time_mult, time_shift, time_zero;
if (!o || !o.map("time_mult", time_mult) ||
!o.map("time_shift", time_shift) || !o.map("time_zero", time_zero))
return false;

tsc_conversion = std::make_unique<LinuxPerfZeroTscConversion>(
static_cast<uint32_t>(time_mult), static_cast<uint16_t>(time_shift),
static_cast<uint64_t>(time_zero));
time_zero);

return true;
}
Expand Down

0 comments on commit 9d2dd6d

Please sign in to comment.