Add IPC for FuzzerRemote on Linux

[aarongreen]

Summary

This change adds an implementation of the IPC layer for FuzzerRemote. It translates calls to the FuzzerProxy* methods of the FuzzerRemoteInterface into messages sent over a fuzzer::ipc::Connection, and messages received over that connection into calls to the FuzzerRemote* methods.

This is change 19 of (at least) 20 for cross-process fuzzing support.

[aarongreen]

Rebase

[aarongreen]

compiler-rt/lib/fuzzer/ipc/CMakeLists.txt

3

Hmm, this should have been applied from the get-go when ipc/ was created. I'll move it up the chain and rebase (again).

[aarongreen]

Multiprocess fuzzing will not be supported by the libFuzzer maintainers. Fuchsia has implemented a new approach with their Component Fuzzing Framework (RFC-117).

compiler-rt/lib/fuzzer/ipc/CMakeLists.txt

 include(CompilerRTCompile)
 
-set(LIBFUZZER_IPC_CFLAGS
-  ${LIBFUZZER_CFLAGS}
-  -I${COMPILER_RT_SOURCE_DIR}/lib/fuzzer)
-
 add_libfuzzer_objects(RTfuzzer_ipc_base
   SOURCES IPCLinux.cpp
-  HEADERS IPCLinux.h
-  CFLAGS ${LIBFUZZER_IPC_CFLAGS})
+  HEADERS IPCLinux.h)
 
 add_libfuzzer_objects(RTfuzzer_ipc
   SOURCES IPCQuickExit.cpp
-  HEADERS IPCQuickExit.h
-  CFLAGS ${LIBFUZZER_IPC_CFLAGS})
+  HEADERS IPCQuickExit.h)
 
 add_libfuzzer_objects(RTfuzzer_ipc_proxy
   SOURCES IPCProxyLinux.cpp
-  HEADERS IPCProxyLinux.h IPCRemoteLinux.h
-  CFLAGS ${LIBFUZZER_IPC_CFLAGS})
+  HEADERS IPCProxyLinux.h IPCRemoteLinux.h)
+
+add_libfuzzer_objects(RTfuzzer_ipc_remote
+  SOURCES IPCRemoteLinux.cpp
+  HEADERS IPCRemoteLinux.h)

compiler-rt/lib/fuzzer/ipc/IPCRemoteLinux.cpp

+//===- IPCRemoteLinux.cpp -------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+// IPC routines used by the FuzzerRemote on Linux.
+//===----------------------------------------------------------------------===//
+
+#include "FuzzerPlatform.h"
+#if LIBFUZZER_LINUX
+
+#include "FuzzerUtil.h"
+
+#include "FuzzerExtFunctions.h"
+#include "FuzzerRemoteInterface.h"
+#include "IPCLinux.h"
+#include "IPCProxyLinux.h"
+#include "IPCRemoteLinux.h"
+#include <cassert>
+#include <sys/types.h>
+#include <unistd.h>
+
+namespace fuzzer {
+namespace ipc {
+namespace {
+
+// This class represents a single FuzzerProxy-side endpoint for a connection to
+// a FuzzerRemote process.
+class RemoteConnection final : public Connection {
+public:
+  static RemoteConnection &GetInstance() {
+    static RemoteConnection Singleton;
+    return Singleton;
+  }
+
+  ~RemoteConnection() override { Reset(); }
+
+  void Connect(void *Options, size_t OptionsLen) {
+    PID = getpid();
+    ReceivedOptions = Options;
+    ReceivedOptionsLen = OptionsLen;
+    if (!Connection::Connect(fuzzer::ipc::kProxyAddressEnvVar)) {
+      Printf("ERROR: Failed to connect to proxy.\n");
+      exit(1);
+    }
+  }
+
+  bool DoHandshake() override {
+    Connection::DoHandshake();
+    std::lock_guard<std::mutex> Lock(GetRxMutex());
+    ProxyOrdinal O;
+    if (!ReceiveLocked(&O) || O != kOptions ||
+        !ReceiveLocked(ReceivedOptions, ReceivedOptionsLen))
+      return false;
+    ReceivedOptions = nullptr;
+    ReceivedOptionsLen = 0;
+    return true;
+  }
+
+  void AddCoverage(uint8_t *CountersBegin, uint8_t *CountersEnd,
+                   const uintptr_t *PCsBegin, const uintptr_t *PCsEnd) {
+    if (CountersEnd <= CountersBegin || PCsEnd <= PCsBegin)
+      return;
+    {
+      std::lock_guard<std::mutex> TxLock(GetTxMutex());
+      SendLocked(kAddCoverage);
+      size_t Len = (CountersEnd - CountersBegin) * sizeof(uint8_t);
+      SendMemoryLocked(CountersBegin, Len, /* Updatable= */ true);
+      Len = (PCsEnd - PCsBegin) * sizeof(uintptr_t);
+      SendMemoryLocked(PCsBegin, Len, /* Updatable= */ false);
+    }
+    WaitForAcknowledgement();
+  }
+
+  void ExecutionStarted() {
+    std::lock_guard<std::mutex> TxLock(GetTxMutex());
+    SendLocked(kExecutionStarted);
+  }
+
+  void ExecutionFinished(int HasMoreMallocsThanFrees) {
+    UpdateSharedMemory();
+    std::lock_guard<std::mutex> TxLock(GetTxMutex());
+    SendLocked(kExecutionFinished);
+    SendLocked(HasMoreMallocsThanFrees);
+  }
+
+  void CrashCallback() {
+    DispatchOnCurrentThread(
+        [this]() REQUIRES(GetTxMutex()) { SendLocked(kCrashCallback); });
+  }
+
+  void DeathCallback() {
+    DispatchOnCurrentThread(
+        [this]() REQUIRES(GetTxMutex()) { SendLocked(kDeathCallback); });
+  }
+
+  void ExitCallback() {
+    DispatchOnCurrentThread(
+        [this]() REQUIRES(GetTxMutex()) { SendLocked(kExitCallback); });
+  }
+
+  void LeakCallback() {
+    // LeakCallback is only ever invoked from the ReceiverThread in response to
+    // either a kFinishExecution or kDetectLeaksAtExit message. Additionally,
+    // it doesn't have an associated stack trace, so this can exit immediately.
+    if (EF->__sanitizer_acquire_crash_state &&
+        !EF->__sanitizer_acquire_crash_state())
+      return;
+    {
+      std::lock_guard<std::mutex> TxLock(GetTxMutex());
+      SendLocked(kLeakCallback);
+    }
+  }
+
+  void MallocLimitCallback(size_t Size) {
+    DispatchOnCurrentThread([this, Size]() REQUIRES(GetTxMutex()) {
+      SendLocked(kMallocLimitCallback);
+      SendLocked(Size);
+    });
+  }
+
+  void RssLimitCallback() {
+    DispatchOnCurrentThread(
+        [this]() REQUIRES(GetTxMutex()) { SendLocked(kRssLimitCallback); });
+  }
+
+private:
+  bool DispatchOneLocked() override REQUIRES(GetRxMutex()) {
+    ProxyOrdinal O;
+    if (!ReceiveLocked(&O))
+      return false;
+    switch (O) {
+    case kProxyAck: {
+      HandleAcknowledgement();
+      break;
+    }
+    case kStartExecution: {
+      uint32_t Options;
+      if (!ReceiveLocked(&Options))
+        return false;
+      FuzzerRemoteStartExecution(PID, Options);
+      break;
+    }
+    case kFinishExecution: {
+      FuzzerRemoteFinishExecution(PID);
+      break;
+    }
+    case kPrintPC: {
+      char SymbolizedFMT[kMaxDescLen];
+      char FallbackFMT[kMaxDescLen];
+      uintptr_t PC;
+      if (!ReceiveStringLocked(SymbolizedFMT, sizeof(SymbolizedFMT)) ||
+          !ReceiveStringLocked(FallbackFMT, sizeof(FallbackFMT)) ||
+          !ReceiveLocked(&PC)) {
+        return false;
+      }
+      FuzzerRemotePrintPC(PID, SymbolizedFMT, FallbackFMT, PC);
+      Acknowledge();
+      break;
+    }
+    case kDescribePC: {
+      char SymbolizedFMT[kMaxDescLen];
+      uintptr_t PC;
+      char Desc[kMaxDescLen];
+      size_t DescLen;
+      if (!ReceiveStringLocked(SymbolizedFMT, sizeof(SymbolizedFMT)) ||
+          !ReceiveLocked(&PC) || !ReceiveLocked(&DescLen))
+        return false;
+      assert(DescLen <= kMaxDescLen);
+      FuzzerRemoteDescribePC(PID, SymbolizedFMT, PC, Desc, DescLen);
+      UniqueLock TxLock(GetTxMutex());
+      SendLocked(kDescribePCResponse);
+      SendStringLocked(Desc);
+      break;
+    }
+    case kPrintStackTrace: {
+      FuzzerRemotePrintStackTrace(PID);
+      Acknowledge();
+      break;
+    }
+    case kPrintMemoryProfile: {
+      FuzzerRemotePrintMemoryProfile(PID);
+      Acknowledge();
+      break;
+    }
+    case kDetectLeaksAtExit: {
+      FuzzerRemoteDetectLeaksAtExit(PID);
+      Acknowledge();
+      break;
+    }
+    default:
+      Printf("ERROR: protocol: unknown request: %u\n", O);
+      Close();
+      exit(1);
+    }
+    return true;
+  }
+
+  template <typename TxCallback> void DispatchOnCurrentThread(TxCallback CB) {
+    if (EF->__sanitizer_acquire_crash_state &&
+        !EF->__sanitizer_acquire_crash_state())
+      return;
+    {
+      UniqueLock TxLock(GetTxMutex());
+      CB();
+    }
+    StopReceiving();
+    DispatchLoop();
+  }
+
+  pid_t PID;
+  void *ReceivedOptions;
+  size_t ReceivedOptionsLen;
+  std::condition_variable OnCloseCV;
+};
+
+} // namespace
+
+} // namespace ipc
+} // namespace fuzzer
+
+// FuzzerRemoteInterface and FuzzerMonitor implementations. The IPC layer looks
+// like FuzzerProxy to the FuzzerRemotes.
+
+extern "C" {
+
+ATTRIBUTE_INTERFACE void FuzzerProxyConnect(unsigned long PID, void *Options,
+                                            size_t OptionsLen) {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.Connect(Options, OptionsLen);
+}
+
+ATTRIBUTE_INTERFACE uintptr_t FuzzerProxyAddCoverage(unsigned long PID,
+                                                     uint8_t *CountersBegin,
+                                                     uint8_t *CountersEnd,
+                                                     const uintptr_t *PCsBegin,
+                                                     const uintptr_t *PCsEnd) {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.AddCoverage(CountersBegin, CountersEnd, PCsBegin, PCsEnd);
+  return fuzzer::kInvalidIdx; // Ignored by FuzzerRemote.
+}
+
+ATTRIBUTE_INTERFACE void FuzzerProxyExecutionStarted(unsigned long PID) {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.ExecutionStarted();
+}
+
+ATTRIBUTE_INTERFACE void
+FuzzerProxyExecutionFinished(unsigned long PID, int HasMoreMallocsThanFrees) {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.ExecutionFinished(HasMoreMallocsThanFrees);
+}
+
+ATTRIBUTE_INTERFACE void FuzzerCrashSignalCallback(unsigned long PID) {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.CrashCallback();
+}
+
+ATTRIBUTE_INTERFACE void FuzzerDeathCallback() {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.DeathCallback();
+}
+
+ATTRIBUTE_INTERFACE void FuzzerExitCallback(unsigned long PID) {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.ExitCallback();
+}
+
+ATTRIBUTE_INTERFACE void FuzzerLeakCallback(unsigned long PID) {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.LeakCallback();
+}
+
+ATTRIBUTE_INTERFACE void FuzzerMallocLimitCallback(unsigned long PID,
+                                                   size_t Size) {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.MallocLimitCallback(Size);
+}
+
+ATTRIBUTE_INTERFACE void FuzzerRssLimitCallback(unsigned long PID) {
+  auto &RC = fuzzer::ipc::RemoteConnection::GetInstance();
+  RC.RssLimitCallback();
+}
+
+} // extern "C"
+
+#endif // LIBFUZZER_LINUX

compiler-rt/lib/fuzzer/tests/ipc/CMakeLists.txt

OBJECT_LIBS RTfuzzer_base
                 RTfuzzer_ipc_test_util)
 
   generate_libfuzzer_unittests(FuzzerIPCProxyUnitTests ${arch}
     SOURCES IPCProxyLinuxUnittest.cpp
     OBJECT_LIBS RTfuzzer_base
                 RTfuzzer_ipc_base
                 RTfuzzer_ipc_test_util
                 RTfuzzer_ipc_proxy)
+
+  generate_libfuzzer_unittests(FuzzerIPCRemoteUnitTests ${arch}
+    SOURCES IPCRemoteLinuxUnittest.cpp
+    OBJECT_LIBS RTfuzzer_base
+                RTfuzzer_ipc_base
+                RTfuzzer_ipc_test_util
+                RTfuzzer_ipc_remote)
 endif()

compiler-rt/lib/fuzzer/tests/ipc/IPCRemoteLinuxUnittest.cpp

+//===- IPCRemoteLinuxUnittest.cpp -----------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+// Unit test of IPCRemoteLinux.
+//
+// This is kept separate from other unit tests as it must link against the
+// ipc_remote library, which has symbols that conflict with the fuzzer and
+// fuzzer_remote runtimes and with the fuzzer_ipc_proxy library, e.g. the
+// FuzzerRemoteInterface symbols.
+//===----------------------------------------------------------------------===//
+
+#include "FuzzerPlatform.h"
+#if LIBFUZZER_LINUX
+
+#include "FuzzerMonitor.h"
+#include "FuzzerRemoteInterface.h"
+#include "ipc/IPCProxyLinux.h"
+#include "ipc/IPCRemoteLinux.h"
+#include "tests/FuzzerTestUtil.h"
+#include "tests/ipc/IPCTestUtilLinux.h"
+#include "gtest/gtest.h"
+#include <sys/mman.h>
+
+namespace fuzzer {
+namespace ipc {
+namespace {
+
+// Test fixtures
+
+class IPCRemoteLinuxTest : public TestBase {
+public:
+  template <typename Responder> static void OnRemoteReceive(Responder R) {
+    ReceiveAtNear(R);
+  }
+
+protected:
+  void SetUp() override {
+    TestBase::SetUp();
+    Proxy.Start(kProxyAddressEnvVar);
+    // Terminated = false;
+  }
+
+  void Connect() {
+    uint8_t Ignored;
+    DoRemoteSend([&]() { FuzzerProxyConnect(PID, &Ignored, sizeof(Ignored)); });
+    Proxy.WaitForConnection();
+    OnProxyReceive([&](TestConnection *TC) {
+      EXPECT_TRUE(TC->Send(kOptions));
+      EXPECT_TRUE(TC->Send(Ignored));
+    });
+  }
+
+  // Rename the base names for clarity.
+  template <typename Requester> void DoRemoteSend(Requester R) {
+    SendFromNear(R);
+  }
+
+  template <typename Responder> void OnProxyReceive(Responder R) {
+    ReceiveAtFar(R);
+  }
+
+  template <typename Requester> void DoProxySend(Requester R) {
+    SendFromFar(R);
+  }
+
+  TestConnection *GetFarConnection() override {
+    return Proxy.GetTestConnection(PID);
+  }
+
+  TestServer Proxy;
+
+  void TearDown() override {
+    Proxy.Stop();
+    TestBase::TearDown();
+  }
+};
+
+// Unit tests
+
+TEST_F(IPCRemoteLinuxTest, Connect) {
+  constexpr size_t kOptionsSize = 0x2020;
+  uint8_t OptionsOut[kOptionsSize];
+
+  DoRemoteSend(
+      [&]() { FuzzerProxyConnect(PID, &OptionsOut, sizeof(OptionsOut)); });
+  Proxy.WaitForConnection();
+
+  uint8_t OptionsIn[kOptionsSize];
+  for (size_t i = 0; i < kOptionsSize; ++i)
+    OptionsIn[i] = Pick<uint8_t>();
+
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Send(kOptions));
+    EXPECT_TRUE(TC->Send(OptionsIn, sizeof(OptionsIn)));
+  });
+
+  EXPECT_EQ(memcmp(OptionsIn, OptionsOut, sizeof(OptionsIn)), 0);
+}
+
+TEST_F(IPCRemoteLinuxTest, AddCoverage) {
+  Connect();
+
+  // Initialize coverage.
+  FakeDSO<0x1000> DSO;
+  DoRemoteSend([&]() {
+    FuzzerProxyAddCoverage(PID, DSO.Counters, DSO.CountersEnd(), DSO.PCs,
+                           DSO.PCsEnd());
+  });
+
+  void *Addr1, *Addr2;
+  size_t Len1, Len2;
+  RemoteOrdinal Ordinal;
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Receive(&Ordinal));
+    EXPECT_TRUE(TC->ReceiveMemory(&Addr1, &Len1));
+    EXPECT_TRUE(TC->ReceiveMemory(&Addr2, &Len2));
+    EXPECT_TRUE(TC->Send(Connection::kAcknowledgement));
+  });
+  EXPECT_EQ(Ordinal, kAddCoverage);
+
+  uint8_t *CountersBegin = reinterpret_cast<uint8_t *>(Addr1);
+  ASSERT_EQ(Len1, sizeof(DSO.Counters));
+  EXPECT_EQ(memcmp(CountersBegin, DSO.Counters, Len1), 0);
+  ASSERT_NE(munmap(Addr1, Len1), -1);
+
+  uintptr_t *PCsBegin = reinterpret_cast<uintptr_t *>(Addr2);
+  ASSERT_EQ(Len2, sizeof(DSO.PCs));
+  EXPECT_EQ(memcmp(PCsBegin, DSO.PCs, Len2), 0);
+  ASSERT_NE(munmap(Addr2, Len2), -1);
+}
+
+static struct : public StaticTestMessage {
+  uint32_t ExecOptions;
+} StartExecutionReq;
+
+extern "C" void FuzzerRemoteStartExecution(unsigned long PID,
+                                           uint32_t ExecOptions) {
+  IPCRemoteLinuxTest::OnRemoteReceive([&]() {
+    StartExecutionReq.PID = PID;
+    StartExecutionReq.ExecOptions = ExecOptions;
+  });
+}
+
+TEST_F(IPCRemoteLinuxTest, StartExecution) {
+  Connect();
+  auto ExecOptions = Pick<uint32_t>();
+  DoProxySend([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Send(kStartExecution));
+    EXPECT_TRUE(TC->Send(ExecOptions));
+  });
+  EXPECT_EQ(StartExecutionReq.PID, PID);
+  EXPECT_EQ(StartExecutionReq.ExecOptions, ExecOptions);
+}
+
+static StaticTestMessage FinishExecutionReq;
+
+extern "C" void FuzzerRemoteFinishExecution(unsigned long PID) {
+  IPCRemoteLinuxTest::OnRemoteReceive([&]() { FinishExecutionReq.PID = PID; });
+}
+
+TEST_F(IPCRemoteLinuxTest, FinishExecution) {
+  Connect();
+  DoProxySend(
+      [&](TestConnection *TC) { EXPECT_TRUE(TC->Send(kFinishExecution)); });
+  EXPECT_EQ(FinishExecutionReq.PID, PID);
+}
+
+TEST_F(IPCRemoteLinuxTest, ExecutionFinished) {
+  Connect();
+
+  // Initialize coverage.
+  FakeDSO<0x1000> DSO;
+  DoRemoteSend([&]() {
+    FuzzerProxyAddCoverage(PID, DSO.Counters, DSO.CountersEnd(), DSO.PCs,
+                           DSO.PCsEnd());
+  });
+
+  RemoteOrdinal Ordinal;
+  void *Addr1, *Addr2;
+  size_t Len1, Len2;
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Receive(&Ordinal));
+    EXPECT_TRUE(TC->ReceiveMemory(&Addr1, &Len1));
+    EXPECT_TRUE(TC->ReceiveMemory(&Addr2, &Len2));
+    EXPECT_TRUE(TC->Send(Connection::kAcknowledgement));
+  });
+  EXPECT_EQ(Ordinal, kAddCoverage);
+
+  uint8_t *CountersBegin = reinterpret_cast<uint8_t *>(Addr1);
+  ASSERT_EQ(Len1, sizeof(DSO.Counters));
+
+  // Simulate finishing an iteration. Counters should be updated.
+  for (size_t i = 0; i < DSO.NumPCs; ++i)
+    DSO.Counters[i] = Pick<uint8_t>();
+  int HasMoreMallocsThanFreesIn = 8;
+  DoRemoteSend(
+      [&]() { FuzzerProxyExecutionFinished(PID, HasMoreMallocsThanFreesIn); });
+
+  int HasMoreMallocsThanFreesOut;
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Receive(&Ordinal));
+    EXPECT_TRUE(TC->Receive(&HasMoreMallocsThanFreesOut));
+  });
+  EXPECT_EQ(Ordinal, kExecutionFinished);
+  EXPECT_EQ(HasMoreMallocsThanFreesIn, HasMoreMallocsThanFreesOut);
+  EXPECT_EQ(memcmp(CountersBegin, DSO.Counters, Len1), 0);
+  ASSERT_NE(munmap(Addr1, Len1), -1);
+  ASSERT_NE(munmap(Addr2, Len2), -1);
+}
+
+TEST_F(IPCRemoteLinuxTest, CrashCallback) {
+  Connect();
+  DoRemoteSend([&]() { FuzzerCrashSignalCallback(PID); });
+
+  RemoteOrdinal Ordinal;
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Receive(&Ordinal));
+    TC->Close();
+  });
+  EXPECT_EQ(Ordinal, kCrashCallback);
+}
+
+TEST_F(IPCRemoteLinuxTest, DeathCallback) {
+  Connect();
+  DoRemoteSend([&]() { FuzzerDeathCallback(); });
+
+  RemoteOrdinal Ordinal;
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Receive(&Ordinal));
+    TC->Close();
+  });
+  EXPECT_EQ(Ordinal, kDeathCallback);
+}
+
+TEST_F(IPCRemoteLinuxTest, ExitCallback) {
+  Connect();
+  DoRemoteSend([&]() { FuzzerExitCallback(PID); });
+
+  RemoteOrdinal Ordinal;
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Receive(&Ordinal));
+    TC->Close();
+  });
+  EXPECT_EQ(Ordinal, kExitCallback);
+}
+
+TEST_F(IPCRemoteLinuxTest, LeakCallback) {
+  Connect();
+  DoRemoteSend([&]() { FuzzerLeakCallback(PID); });
+
+  RemoteOrdinal Ordinal;
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Receive(&Ordinal));
+    TC->Close();
+  });
+  EXPECT_EQ(Ordinal, kLeakCallback);
+}
+
+TEST_F(IPCRemoteLinuxTest, MallocLimitCallback) {
+  Connect();
+  size_t SizeIn = std::numeric_limits<size_t>::max();
+  DoRemoteSend([&]() { FuzzerMallocLimitCallback(PID, SizeIn); });
+
+  RemoteOrdinal Ordinal;
+  size_t SizeOut;
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Receive(&Ordinal));
+    EXPECT_TRUE(TC->Receive(&SizeOut));
+    TC->Close();
+  });
+  EXPECT_EQ(Ordinal, kMallocLimitCallback);
+  EXPECT_EQ(SizeIn, SizeOut);
+}
+
+TEST_F(IPCRemoteLinuxTest, RssLimitCallback) {
+  Connect();
+  DoRemoteSend([&]() { FuzzerRssLimitCallback(PID); });
+
+  RemoteOrdinal Ordinal;
+  OnProxyReceive([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Receive(&Ordinal));
+    TC->Close();
+  });
+  EXPECT_EQ(Ordinal, kRssLimitCallback);
+}
+
+static struct : public StaticTestMessage {
+  char SymbolizedFMT[kMaxDescLen];
+  char FallbackFMT[kMaxDescLen];
+  uintptr_t PC;
+} PrintPCReq;
+
+extern "C" void FuzzerRemotePrintPC(unsigned long PID,
+                                    const char *SymbolizedFMT,
+                                    const char *FallbackFMT, uintptr_t PC) {
+  IPCRemoteLinuxTest::OnRemoteReceive([&]() {
+    PrintPCReq.PID = PID;
+    strncpy(PrintPCReq.SymbolizedFMT, SymbolizedFMT,
+            sizeof(PrintPCReq.SymbolizedFMT));
+    strncpy(PrintPCReq.FallbackFMT, FallbackFMT,
+            sizeof(PrintPCReq.FallbackFMT));
+    PrintPCReq.PC = PC;
+  });
+}
+
+TEST_F(IPCRemoteLinuxTest, PrintPC) {
+  Connect();
+  const char *SymbolizedFMT = "symbolized format";
+  const char *FallbackFMT = "fallback format";
+  auto PC = Pick<uintptr_t>();
+  DoProxySend([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Send(kPrintPC));
+    EXPECT_TRUE(TC->SendString(SymbolizedFMT));
+    EXPECT_TRUE(TC->SendString(FallbackFMT));
+    EXPECT_TRUE(TC->Send(PC));
+  });
+
+  EXPECT_EQ(PrintPCReq.PID, PID);
+  EXPECT_STREQ(PrintPCReq.SymbolizedFMT, SymbolizedFMT);
+  EXPECT_STREQ(PrintPCReq.FallbackFMT, FallbackFMT);
+  EXPECT_EQ(PrintPCReq.PC, PC);
+}
+
+static struct : public StaticTestMessage {
+  char SymbolizedFMT[kMaxDescLen];
+  uintptr_t PC;
+} DescribePCReq;
+
+extern "C" void FuzzerRemoteDescribePC(unsigned long PID,
+                                       const char *SymbolizedFMT, uintptr_t PC,
+                                       char *Desc, size_t DescLen) {
+  IPCRemoteLinuxTest::OnRemoteReceive([&]() {
+    DescribePCReq.PID = PID;
+    strncpy(DescribePCReq.SymbolizedFMT, SymbolizedFMT,
+            sizeof(DescribePCReq.SymbolizedFMT));
+    DescribePCReq.PC = PC;
+    snprintf(Desc, DescLen, SymbolizedFMT, PC);
+  });
+}
+
+TEST_F(IPCRemoteLinuxTest, DescribePC) {
+  Connect();
+  const char *SymbolizedFMT = "symbolized format: %lu";
+  auto PC = Pick<uintptr_t>();
+  char Desc[kMaxDescLen];
+  DoProxySend([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Send(kDescribePC));
+    EXPECT_TRUE(TC->SendString(SymbolizedFMT));
+    EXPECT_TRUE(TC->Send(PC));
+    EXPECT_TRUE(TC->Send(sizeof(Desc)));
+  });
+
+  EXPECT_EQ(DescribePCReq.PID, PID);
+  EXPECT_STREQ(DescribePCReq.SymbolizedFMT, SymbolizedFMT);
+  EXPECT_EQ(DescribePCReq.PC, PC);
+
+  auto *TC = GetFarConnection();
+  RemoteOrdinal Ordinal;
+  EXPECT_TRUE(TC->Receive(&Ordinal));
+  EXPECT_TRUE(TC->ReceiveString(Desc, sizeof(Desc)));
+
+  char Expected[sizeof(Desc)];
+  snprintf(Expected, sizeof(Expected), SymbolizedFMT, PC);
+  EXPECT_EQ(Ordinal, kDescribePCResponse);
+  EXPECT_STREQ(Desc, Expected);
+}
+
+static StaticTestMessage PrintStackTraceReq;
+
+extern "C" void FuzzerRemotePrintStackTrace(unsigned long PID) {
+  IPCRemoteLinuxTest::OnRemoteReceive([&]() { PrintStackTraceReq.PID = PID; });
+}
+
+TEST_F(IPCRemoteLinuxTest, PrintStackTrace) {
+  Connect();
+  DoProxySend(
+      [&](TestConnection *TC) { EXPECT_TRUE(TC->Send(kPrintStackTrace)); });
+  EXPECT_EQ(PrintStackTraceReq.PID, PID);
+}
+
+static StaticTestMessage PrintMemoryProfileReq;
+
+extern "C" void FuzzerRemotePrintMemoryProfile(unsigned long PID) {
+  IPCRemoteLinuxTest::OnRemoteReceive(
+      [&]() { PrintMemoryProfileReq.PID = PID; });
+}
+
+TEST_F(IPCRemoteLinuxTest, PrintMemoryProfile) {
+  Connect();
+  DoProxySend(
+      [&](TestConnection *TC) { EXPECT_TRUE(TC->Send(kPrintMemoryProfile)); });
+  EXPECT_EQ(PrintMemoryProfileReq.PID, PID);
+}
+
+static struct : public StaticTestMessage { int ExitCode; } TerminateReq;
+
+extern "C" void FuzzerRemoteTerminate(unsigned long PID, int ExitCode) {
+  // Handle "normal" exits caused by disconnecting the proxy from the remote.
+  if (ExitCode == 0)
+    return;
+  IPCRemoteLinuxTest::OnRemoteReceive([&]() {
+    TerminateReq.PID = PID;
+    TerminateReq.ExitCode = ExitCode;
+  });
+}
+
+TEST_F(IPCRemoteLinuxTest, Terminate) {
+  Connect();
+  int ExitCode = 1;
+  DoProxySend([&](TestConnection *TC) {
+    EXPECT_TRUE(TC->Send(kTerminate));
+    EXPECT_TRUE(TC->Send(ExitCode));
+  });
+
+  EXPECT_EQ(TerminateReq.PID, PID);
+  EXPECT_EQ(TerminateReq.ExitCode, ExitCode);
+}
+
+} // namespace
+} // namespace ipc
+} // namespace fuzzer
+
+#endif // LIBFUZZER_LINUX

compiler-rt/test/fuzzer/CMakeLists.txt

 endif()
 
 if(COMPILER_RT_INCLUDE_TESTS)
   list(APPEND LIBFUZZER_TEST_DEPS FuzzerUnitTests)
   list(APPEND LIBFUZZER_TEST_DEPS FuzzedDataProviderUnitTests)
   list(APPEND LIBFUZZER_TEST_DEPS FuzzerRemoteUnitTests)
   list(APPEND LIBFUZZER_TEST_DEPS FuzzerIPCUnitTests)
   list(APPEND LIBFUZZER_TEST_DEPS FuzzerIPCProxyUnitTests)
+  list(APPEND LIBFUZZER_TEST_DEPS FuzzerIPCRemoteUnitTests)
 endif()
 
 add_custom_target(check-fuzzer)
 
 if(COMPILER_RT_INCLUDE_TESTS)
   # libFuzzer unit tests.
   configure_lit_site_cfg(
     ${CMAKE_CURRENT_SOURCE_DIR}/unit/lit.site.cfg.py.in