Files
swift-mirror/include/swift/Remote/CMemoryReader.h
Augusto Noronha 58df5534d2 [NFC][RemoteInspection] Add an opaque AddressSpace field to RemoteAddress
Add an extra opaque field to AddressSpace, which can be used by clients
of RemoteInspection to distinguish between different address spaces.

LLDB employs an optimization where it reads memory from files instead of
the running process whenever it can to speed up memory reads (these can
be slow when debugging something over a network). To do this, it needs
to keep track whether an address originated from a process or a file. It
currently distinguishes addresses by setting an unused high bit on the
address, but because of pointer authentication this is not a reliable
solution. In order to keep this optimization working, this patch adds an
extra opaque AddressSpace field to RemoteAddress, which LLDB can use on
its own implementation of MemoryReader to distinguish between addresses.

This patch is NFC for the other RemoteInspection clients, as it adds
extra information to RemoteAddress, which is entirely optional and if
unused should not change the behavior of the library.

Although this patch is quite big the changes are largely mechanical,
replacing threading StoredPointer with RemoteAddress.

rdar://148361743
2025-07-09 14:52:42 -07:00

117 lines
4.0 KiB
C++

//===--- CMemoryReader.h - MemoryReader wrapper for C impls -----*- C++ -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
//
// This file declares an implementation of MemoryReader that wraps the
// C interface provided by SwiftRemoteMirror.
//
//===----------------------------------------------------------------------===//
#ifndef SWIFT_REMOTE_CMEMORYREADER_H
#define SWIFT_REMOTE_CMEMORYREADER_H
#include "swift/SwiftRemoteMirror/MemoryReaderInterface.h"
#include "swift/Remote/MemoryReader.h"
struct MemoryReaderImpl {
uint8_t PointerSize;
// Opaque pointer passed to all the callback functions.
void *reader_context;
QueryDataLayoutFunction queryDataLayout;
FreeBytesFunction free;
ReadBytesFunction readBytes;
GetStringLengthFunction getStringLength;
GetSymbolAddressFunction getSymbolAddress;
};
namespace swift {
namespace remote {
/// An implementation of MemoryReader which wraps the C interface offered
/// by SwiftRemoteMirror.
class CMemoryReader final : public MemoryReader {
MemoryReaderImpl Impl;
// Check to see if an address has bits outside the ptrauth mask. This suggests
// that we're likely failing to strip a signed pointer when reading from it.
bool hasSignatureBits(RemoteAddress address) {
uint64_t addressData = address.getRawAddress();
uint64_t mask = getPtrAuthMask().value_or(~uint64_t(0));
return addressData != (addressData & mask);
}
public:
CMemoryReader(MemoryReaderImpl Impl) : Impl(Impl) {
assert(this->Impl.queryDataLayout && "No queryDataLayout implementation");
assert(this->Impl.getStringLength && "No stringLength implementation");
assert(this->Impl.readBytes && "No readBytes implementation");
}
bool queryDataLayout(DataLayoutQueryType type, void *inBuffer,
void *outBuffer) override {
return Impl.queryDataLayout(Impl.reader_context, type, inBuffer,
outBuffer) != 0;
}
RemoteAddress getSymbolAddress(const std::string &name) override {
auto addressData = Impl.getSymbolAddress(Impl.reader_context,
name.c_str(), name.size());
return RemoteAddress(addressData, RemoteAddress::DefaultAddressSpace);
}
uint64_t getStringLength(RemoteAddress address) {
assert(!hasSignatureBits(address));
return Impl.getStringLength(Impl.reader_context, address.getRawAddress());
}
bool readString(RemoteAddress address, std::string &dest) override {
assert(!hasSignatureBits(address));
auto length = getStringLength(address);
if (length == 0) {
// A length of zero unfortunately might mean either that there's a zero
// length string at the location we're trying to read, or that reading
// failed. We can do a one-byte read to tell them apart.
auto buf = readBytes(address, 1);
return buf && ((const char*)buf.get())[0] == 0;
}
auto Buf = readBytes(address, length);
if (!Buf)
return false;
dest = std::string(reinterpret_cast<const char *>(Buf.get()), length);
return true;
}
ReadBytesResult readBytes(RemoteAddress address, uint64_t size) override {
assert(!hasSignatureBits(address));
void *FreeContext;
auto Ptr = Impl.readBytes(Impl.reader_context, address.getRawAddress(),
size, &FreeContext);
auto Free = Impl.free;
if (Free == nullptr)
return ReadBytesResult(Ptr, [](const void *) {});
auto ReaderContext = Impl.reader_context;
auto freeLambda = [=](const void *Ptr) {
Free(ReaderContext, Ptr, FreeContext);
};
return ReadBytesResult(Ptr, freeLambda);
}
};
} // namespace remote
} // namespace swift
#endif