diff --git a/lldb/include/lldb/Host/common/NativeProcessProtocol.h b/lldb/include/lldb/Host/common/NativeProcessProtocol.h --- a/lldb/include/lldb/Host/common/NativeProcessProtocol.h +++ b/lldb/include/lldb/Host/common/NativeProcessProtocol.h @@ -83,6 +83,9 @@ Status ReadMemoryWithoutTrap(lldb::addr_t addr, void *buf, size_t size, size_t &bytes_read); + Status ReadCStringFromMemory(lldb::addr_t addr, char *buffer, size_t max_size, + size_t &total_bytes_read); + virtual Status WriteMemory(lldb::addr_t addr, const void *buf, size_t size, size_t &bytes_written) = 0; diff --git a/lldb/source/Host/common/NativeProcessProtocol.cpp b/lldb/source/Host/common/NativeProcessProtocol.cpp --- a/lldb/source/Host/common/NativeProcessProtocol.cpp +++ b/lldb/source/Host/common/NativeProcessProtocol.cpp @@ -16,6 +16,8 @@ #include "lldb/Utility/State.h" #include "lldb/lldb-enumerations.h" +#include "llvm/Support/Process.h" + using namespace lldb; using namespace lldb_private; @@ -659,6 +661,51 @@ return Status(); } +Status NativeProcessProtocol::ReadCStringFromMemory(lldb::addr_t addr, + char *buffer, + size_t max_size, + size_t &total_bytes_read) { + const size_t cache_line_size = llvm::sys::Process::getPageSizeEstimate(); + size_t bytes_read = 0; + size_t bytes_left = max_size; + addr_t curr_addr = addr; + char *curr_buffer = buffer; + total_bytes_read = 0; + Status error; + + while (bytes_left > 0 && error.Success()) { + addr_t cache_line_bytes_left = + cache_line_size - (curr_addr % cache_line_size); + addr_t bytes_to_read = std::min(bytes_left, cache_line_bytes_left); + error = ReadMemory(curr_addr, reinterpret_cast(curr_buffer), + bytes_to_read, bytes_read); + + if (bytes_read == 0) + break; + + auto str_end = std::memchr(curr_buffer, '\0', bytes_read); + if (str_end != NULL) { + total_bytes_read = (size_t)(reinterpret_cast(str_end) - buffer); + error.Clear(); + break; + } + + total_bytes_read += bytes_read; + curr_buffer += bytes_read; + curr_addr = reinterpret_cast(reinterpret_cast(curr_addr) + + bytes_read); + bytes_left -= bytes_read; + } + + // Make sure we return a null terminated string. + if (bytes_left == 0 && buffer[max_size - 1] != '\0') { + buffer[max_size - 1] = '\0'; + total_bytes_read--; + } + + return error; +} + lldb::StateType NativeProcessProtocol::GetState() const { std::lock_guard guard(m_state_mutex); return m_state; diff --git a/lldb/source/Plugins/Process/Linux/NativeProcessLinux.cpp b/lldb/source/Plugins/Process/Linux/NativeProcessLinux.cpp --- a/lldb/source/Plugins/Process/Linux/NativeProcessLinux.cpp +++ b/lldb/source/Plugins/Process/Linux/NativeProcessLinux.cpp @@ -2189,11 +2189,12 @@ return error; char name_buffer[PATH_MAX]; - error = ReadMemory(link_map.l_name, &name_buffer, sizeof(name_buffer), - bytes_read); + error = ReadCStringFromMemory(link_map.l_name, + reinterpret_cast(&name_buffer), + sizeof(name_buffer), bytes_read); if (!error.Success()) return error; - name_buffer[PATH_MAX - 1] = '\0'; + info.name = std::string(name_buffer); info.link_map = link_map_addr; info.base_addr = link_map.l_addr;