180814287SRaphael Isemann //===-- GDBRemoteCommunicationClientTest.cpp ------------------------------===//
256d7262bSPavel Labath //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
656d7262bSPavel Labath //
756d7262bSPavel Labath //===----------------------------------------------------------------------===//
856d7262bSPavel Labath #include "Plugins/Process/gdb-remote/GDBRemoteCommunicationClient.h"
947776cbdSPavel Labath #include "GDBRemoteTestUtils.h"
102f1fbaebSPavel Labath #include "lldb/Core/ModuleSpec.h"
1147776cbdSPavel Labath #include "lldb/Host/XML.h"
1248d1427cSStephane Sezer #include "lldb/Target/MemoryRegionInfo.h"
13666cc0b2SZachary Turner #include "lldb/Utility/DataBuffer.h"
14f2a8bccfSPavel Labath #include "lldb/Utility/StructuredData.h"
1593df2fbeSPavel Labath #include "lldb/lldb-enumerations.h"
16b42b48e0SPavel Labath #include "llvm/ADT/ArrayRef.h"
1793df2fbeSPavel Labath #include "llvm/Testing/Support/Error.h"
1847776cbdSPavel Labath #include "gmock/gmock.h"
1947776cbdSPavel Labath #include <future>
205ea091a8SDavid Spickett #include <limits>
2156d7262bSPavel Labath 
2256d7262bSPavel Labath using namespace lldb_private::process_gdb_remote;
2356d7262bSPavel Labath using namespace lldb_private;
2456d7262bSPavel Labath using namespace lldb;
253aa04910SPavel Labath using namespace llvm;
2656d7262bSPavel Labath 
27b9c1b51eSKate Stone namespace {
2856d7262bSPavel Labath 
2956d7262bSPavel Labath typedef GDBRemoteCommunication::PacketResult PacketResult;
3056d7262bSPavel Labath 
31b9c1b51eSKate Stone struct TestClient : public GDBRemoteCommunicationClient {
TestClient__anon8a707d100111::TestClient3256d7262bSPavel Labath   TestClient() { m_send_acks = false; }
3356d7262bSPavel Labath };
3456d7262bSPavel Labath 
Handle_QThreadSuffixSupported(MockServer & server,bool supported)35b9c1b51eSKate Stone void Handle_QThreadSuffixSupported(MockServer &server, bool supported) {
3656d7262bSPavel Labath   StringExtractorGDBRemote request;
3756d7262bSPavel Labath   ASSERT_EQ(PacketResult::Success, server.GetPacket(request));
3856d7262bSPavel Labath   ASSERT_EQ("QThreadSuffixSupported", request.GetStringRef());
3956d7262bSPavel Labath   if (supported)
4056d7262bSPavel Labath     ASSERT_EQ(PacketResult::Success, server.SendOKResponse());
4156d7262bSPavel Labath   else
4256d7262bSPavel Labath     ASSERT_EQ(PacketResult::Success, server.SendUnimplementedResponse(nullptr));
4356d7262bSPavel Labath }
4456d7262bSPavel Labath 
HandlePacket(MockServer & server,const testing::Matcher<const std::string &> & expected,StringRef response)4547776cbdSPavel Labath void HandlePacket(MockServer &server,
4647776cbdSPavel Labath                   const testing::Matcher<const std::string &> &expected,
4747776cbdSPavel Labath                   StringRef response) {
4856d7262bSPavel Labath   StringExtractorGDBRemote request;
4956d7262bSPavel Labath   ASSERT_EQ(PacketResult::Success, server.GetPacket(request));
50777180a3SBenjamin Kramer   ASSERT_THAT(std::string(request.GetStringRef()), expected);
5156d7262bSPavel Labath   ASSERT_EQ(PacketResult::Success, server.SendPacket(response));
5256d7262bSPavel Labath }
5356d7262bSPavel Labath 
54b9c1b51eSKate Stone uint8_t all_registers[] = {'@', 'A', 'B', 'C', 'D', 'E', 'F', 'G',
55b9c1b51eSKate Stone                            'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O'};
56b42b48e0SPavel Labath std::string all_registers_hex = "404142434445464748494a4b4c4d4e4f";
57b42b48e0SPavel Labath uint8_t one_register[] = {'A', 'B', 'C', 'D'};
58b42b48e0SPavel Labath std::string one_register_hex = "41424344";
594b6f9591SPavel Labath 
6056d7262bSPavel Labath } // end anonymous namespace
6156d7262bSPavel Labath 
6293df2fbeSPavel Labath class GDBRemoteCommunicationClientTest : public GDBRemoteTest {
6393df2fbeSPavel Labath public:
SetUp()6493df2fbeSPavel Labath   void SetUp() override {
659e046f02SJonas Devlieghere     ASSERT_THAT_ERROR(GDBRemoteCommunication::ConnectLocally(client, server),
669e046f02SJonas Devlieghere                       llvm::Succeeded());
6793df2fbeSPavel Labath   }
6856d7262bSPavel Labath 
6993df2fbeSPavel Labath protected:
7056d7262bSPavel Labath   TestClient client;
7156d7262bSPavel Labath   MockServer server;
7293df2fbeSPavel Labath };
7356d7262bSPavel Labath 
TEST_F(GDBRemoteCommunicationClientTest,WriteRegister)7493df2fbeSPavel Labath TEST_F(GDBRemoteCommunicationClientTest, WriteRegister) {
7556d7262bSPavel Labath   const lldb::tid_t tid = 0x47;
7656d7262bSPavel Labath   const uint32_t reg_num = 4;
77b9c1b51eSKate Stone   std::future<bool> write_result = std::async(std::launch::async, [&] {
78b9c1b51eSKate Stone     return client.WriteRegister(tid, reg_num, one_register);
79b9c1b51eSKate Stone   });
805c95ee4dSPavel Labath 
815c95ee4dSPavel Labath   Handle_QThreadSuffixSupported(server, true);
8256d7262bSPavel Labath 
83b42b48e0SPavel Labath   HandlePacket(server, "P4=" + one_register_hex + ";thread:0047;", "OK");
8456d7262bSPavel Labath   ASSERT_TRUE(write_result.get());
8556d7262bSPavel Labath 
86b9c1b51eSKate Stone   write_result = std::async(std::launch::async, [&] {
87b9c1b51eSKate Stone     return client.WriteAllRegisters(tid, all_registers);
88b9c1b51eSKate Stone   });
8956d7262bSPavel Labath 
90b42b48e0SPavel Labath   HandlePacket(server, "G" + all_registers_hex + ";thread:0047;", "OK");
9156d7262bSPavel Labath   ASSERT_TRUE(write_result.get());
9256d7262bSPavel Labath }
9356d7262bSPavel Labath 
TEST_F(GDBRemoteCommunicationClientTest,WriteRegisterNoSuffix)94b9c1b51eSKate Stone TEST_F(GDBRemoteCommunicationClientTest, WriteRegisterNoSuffix) {
9556d7262bSPavel Labath   const lldb::tid_t tid = 0x47;
9656d7262bSPavel Labath   const uint32_t reg_num = 4;
97b9c1b51eSKate Stone   std::future<bool> write_result = std::async(std::launch::async, [&] {
98b9c1b51eSKate Stone     return client.WriteRegister(tid, reg_num, one_register);
99b9c1b51eSKate Stone   });
10056d7262bSPavel Labath 
1015c95ee4dSPavel Labath   Handle_QThreadSuffixSupported(server, false);
1023c326955SMichał Górny   HandlePacket(server, "Hg47", "OK");
103b42b48e0SPavel Labath   HandlePacket(server, "P4=" + one_register_hex, "OK");
10456d7262bSPavel Labath   ASSERT_TRUE(write_result.get());
10556d7262bSPavel Labath 
106b9c1b51eSKate Stone   write_result = std::async(std::launch::async, [&] {
107b9c1b51eSKate Stone     return client.WriteAllRegisters(tid, all_registers);
108b9c1b51eSKate Stone   });
10956d7262bSPavel Labath 
110b42b48e0SPavel Labath   HandlePacket(server, "G" + all_registers_hex, "OK");
11156d7262bSPavel Labath   ASSERT_TRUE(write_result.get());
11256d7262bSPavel Labath }
1134b6f9591SPavel Labath 
TEST_F(GDBRemoteCommunicationClientTest,ReadRegister)114b9c1b51eSKate Stone TEST_F(GDBRemoteCommunicationClientTest, ReadRegister) {
1154b6f9591SPavel Labath   const lldb::tid_t tid = 0x47;
1164b6f9591SPavel Labath   const uint32_t reg_num = 4;
117b9c1b51eSKate Stone   std::future<bool> async_result = std::async(
118b9c1b51eSKate Stone       std::launch::async, [&] { return client.GetpPacketSupported(tid); });
1195c95ee4dSPavel Labath   Handle_QThreadSuffixSupported(server, true);
120b42b48e0SPavel Labath   HandlePacket(server, "p0;thread:0047;", one_register_hex);
1214b6f9591SPavel Labath   ASSERT_TRUE(async_result.get());
1224b6f9591SPavel Labath 
123b9c1b51eSKate Stone   std::future<DataBufferSP> read_result = std::async(
124b9c1b51eSKate Stone       std::launch::async, [&] { return client.ReadRegister(tid, reg_num); });
1254b6f9591SPavel Labath   HandlePacket(server, "p4;thread:0047;", "41424344");
126b42b48e0SPavel Labath   auto buffer_sp = read_result.get();
127b42b48e0SPavel Labath   ASSERT_TRUE(bool(buffer_sp));
128b9c1b51eSKate Stone   ASSERT_EQ(0,
129b9c1b51eSKate Stone             memcmp(buffer_sp->GetBytes(), one_register, sizeof one_register));
1304b6f9591SPavel Labath 
131b9c1b51eSKate Stone   read_result = std::async(std::launch::async,
132b9c1b51eSKate Stone                            [&] { return client.ReadAllRegisters(tid); });
133b42b48e0SPavel Labath   HandlePacket(server, "g;thread:0047;", all_registers_hex);
134b42b48e0SPavel Labath   buffer_sp = read_result.get();
135b42b48e0SPavel Labath   ASSERT_TRUE(bool(buffer_sp));
136b9c1b51eSKate Stone   ASSERT_EQ(0,
137b9c1b51eSKate Stone             memcmp(buffer_sp->GetBytes(), all_registers, sizeof all_registers));
1384b6f9591SPavel Labath }
1394b6f9591SPavel Labath 
TEST_F(GDBRemoteCommunicationClientTest,SaveRestoreRegistersNoSuffix)140b9c1b51eSKate Stone TEST_F(GDBRemoteCommunicationClientTest, SaveRestoreRegistersNoSuffix) {
1414b6f9591SPavel Labath   const lldb::tid_t tid = 0x47;
1424b6f9591SPavel Labath   uint32_t save_id;
143b9c1b51eSKate Stone   std::future<bool> async_result = std::async(std::launch::async, [&] {
144b9c1b51eSKate Stone     return client.SaveRegisterState(tid, save_id);
145b9c1b51eSKate Stone   });
1465c95ee4dSPavel Labath   Handle_QThreadSuffixSupported(server, false);
1473c326955SMichał Górny   HandlePacket(server, "Hg47", "OK");
1484b6f9591SPavel Labath   HandlePacket(server, "QSaveRegisterState", "1");
1494b6f9591SPavel Labath   ASSERT_TRUE(async_result.get());
1504b6f9591SPavel Labath   EXPECT_EQ(1u, save_id);
1514b6f9591SPavel Labath 
152b9c1b51eSKate Stone   async_result = std::async(std::launch::async, [&] {
153b9c1b51eSKate Stone     return client.RestoreRegisterState(tid, save_id);
154b9c1b51eSKate Stone   });
1554b6f9591SPavel Labath   HandlePacket(server, "QRestoreRegisterState:1", "OK");
1564b6f9591SPavel Labath   ASSERT_TRUE(async_result.get());
1574b6f9591SPavel Labath }
15827402d2aSPavel Labath 
TEST_F(GDBRemoteCommunicationClientTest,SyncThreadState)159b9c1b51eSKate Stone TEST_F(GDBRemoteCommunicationClientTest, SyncThreadState) {
16027402d2aSPavel Labath   const lldb::tid_t tid = 0x47;
161b9c1b51eSKate Stone   std::future<bool> async_result = std::async(
162b9c1b51eSKate Stone       std::launch::async, [&] { return client.SyncThreadState(tid); });
16327402d2aSPavel Labath   HandlePacket(server, "qSyncThreadStateSupported", "OK");
16427402d2aSPavel Labath   HandlePacket(server, "QSyncThreadState:0047;", "OK");
16527402d2aSPavel Labath   ASSERT_TRUE(async_result.get());
16627402d2aSPavel Labath }
1672f1fbaebSPavel Labath 
TEST_F(GDBRemoteCommunicationClientTest,GetModulesInfo)1682f1fbaebSPavel Labath TEST_F(GDBRemoteCommunicationClientTest, GetModulesInfo) {
1692f1fbaebSPavel Labath   llvm::Triple triple("i386-pc-linux");
1702f1fbaebSPavel Labath 
1716e5e49fbSPavel Labath   FileSpec file_specs[] = {
1728f3be7a3SJonas Devlieghere       FileSpec("/foo/bar.so", FileSpec::Style::posix),
1738f3be7a3SJonas Devlieghere       FileSpec("/foo/baz.so", FileSpec::Style::posix),
174763f1c45SPavel Labath 
175763f1c45SPavel Labath       // This is a bit dodgy but we currently depend on GetModulesInfo not
176763f1c45SPavel Labath       // performing denormalization. It can go away once the users
177763f1c45SPavel Labath       // (DynamicLoaderPOSIXDYLD, at least) correctly set the path syntax for
178763f1c45SPavel Labath       // the FileSpecs they create.
1798f3be7a3SJonas Devlieghere       FileSpec("/foo/baw.so", FileSpec::Style::windows),
180763f1c45SPavel Labath   };
1812f1fbaebSPavel Labath   std::future<llvm::Optional<std::vector<ModuleSpec>>> async_result =
1822f1fbaebSPavel Labath       std::async(std::launch::async,
1832f1fbaebSPavel Labath                  [&] { return client.GetModulesInfo(file_specs, triple); });
1842f1fbaebSPavel Labath   HandlePacket(
1852f1fbaebSPavel Labath       server, "jModulesInfo:["
1862f1fbaebSPavel Labath               R"({"file":"/foo/bar.so","triple":"i386-pc-linux"},)"
187763f1c45SPavel Labath               R"({"file":"/foo/baz.so","triple":"i386-pc-linux"},)"
188763f1c45SPavel Labath               R"({"file":"/foo/baw.so","triple":"i386-pc-linux"}])",
1892f1fbaebSPavel Labath       R"([{"uuid":"404142434445464748494a4b4c4d4e4f","triple":"i386-pc-linux",)"
1902f1fbaebSPavel Labath       R"("file_path":"/foo/bar.so","file_offset":0,"file_size":1234}]])");
1912f1fbaebSPavel Labath 
1922f1fbaebSPavel Labath   auto result = async_result.get();
193e5f568a4SKazu Hirata   ASSERT_TRUE(result.has_value());
1942f1fbaebSPavel Labath   ASSERT_EQ(1u, result->size());
195*5cff5142SKazu Hirata   EXPECT_EQ("/foo/bar.so", result.value()[0].GetFileSpec().GetPath());
196*5cff5142SKazu Hirata   EXPECT_EQ(triple, result.value()[0].GetArchitecture().GetTriple());
1972f93fd1fSPavel Labath   EXPECT_EQ(UUID::fromData("@ABCDEFGHIJKLMNO", 16),
198*5cff5142SKazu Hirata             result.value()[0].GetUUID());
199*5cff5142SKazu Hirata   EXPECT_EQ(0u, result.value()[0].GetObjectOffset());
200*5cff5142SKazu Hirata   EXPECT_EQ(1234u, result.value()[0].GetObjectSize());
2012f1fbaebSPavel Labath }
2022f1fbaebSPavel Labath 
TEST_F(GDBRemoteCommunicationClientTest,GetModulesInfo_UUID20)2038c92c899SPavel Labath TEST_F(GDBRemoteCommunicationClientTest, GetModulesInfo_UUID20) {
2048c92c899SPavel Labath   llvm::Triple triple("i386-pc-linux");
2058c92c899SPavel Labath 
2068f3be7a3SJonas Devlieghere   FileSpec file_spec("/foo/bar.so", FileSpec::Style::posix);
2078c92c899SPavel Labath   std::future<llvm::Optional<std::vector<ModuleSpec>>> async_result =
2088c92c899SPavel Labath       std::async(std::launch::async,
2098c92c899SPavel Labath                  [&] { return client.GetModulesInfo(file_spec, triple); });
2108c92c899SPavel Labath   HandlePacket(
2118c92c899SPavel Labath       server,
2128c92c899SPavel Labath       "jModulesInfo:["
2138c92c899SPavel Labath       R"({"file":"/foo/bar.so","triple":"i386-pc-linux"}])",
2148c92c899SPavel Labath       R"([{"uuid":"404142434445464748494a4b4c4d4e4f50515253","triple":"i386-pc-linux",)"
2158c92c899SPavel Labath       R"("file_path":"/foo/bar.so","file_offset":0,"file_size":1234}]])");
2168c92c899SPavel Labath 
2178c92c899SPavel Labath   auto result = async_result.get();
218e5f568a4SKazu Hirata   ASSERT_TRUE(result.has_value());
2198c92c899SPavel Labath   ASSERT_EQ(1u, result->size());
220*5cff5142SKazu Hirata   EXPECT_EQ("/foo/bar.so", result.value()[0].GetFileSpec().GetPath());
221*5cff5142SKazu Hirata   EXPECT_EQ(triple, result.value()[0].GetArchitecture().GetTriple());
2222f93fd1fSPavel Labath   EXPECT_EQ(UUID::fromData("@ABCDEFGHIJKLMNOPQRS", 20),
223*5cff5142SKazu Hirata             result.value()[0].GetUUID());
224*5cff5142SKazu Hirata   EXPECT_EQ(0u, result.value()[0].GetObjectOffset());
225*5cff5142SKazu Hirata   EXPECT_EQ(1234u, result.value()[0].GetObjectSize());
2268c92c899SPavel Labath }
2278c92c899SPavel Labath 
TEST_F(GDBRemoteCommunicationClientTest,GetModulesInfoInvalidResponse)2282f1fbaebSPavel Labath TEST_F(GDBRemoteCommunicationClientTest, GetModulesInfoInvalidResponse) {
2292f1fbaebSPavel Labath   llvm::Triple triple("i386-pc-linux");
2308f3be7a3SJonas Devlieghere   FileSpec file_spec("/foo/bar.so", FileSpec::Style::posix);
2312f1fbaebSPavel Labath 
2322f1fbaebSPavel Labath   const char *invalid_responses[] = {
2332f1fbaebSPavel Labath       // no UUID
2342f1fbaebSPavel Labath       R"([{"triple":"i386-pc-linux",)"
2358c92c899SPavel Labath       R"("file_path":"/foo/bar.so","file_offset":0,"file_size":1234}]])",
2368c92c899SPavel Labath       // invalid UUID
2378c92c899SPavel Labath       R"([{"uuid":"XXXXXX","triple":"i386-pc-linux",)"
2388c92c899SPavel Labath       R"("file_path":"/foo/bar.so","file_offset":0,"file_size":1234}]])",
2392f1fbaebSPavel Labath       // no triple
2402f1fbaebSPavel Labath       R"([{"uuid":"404142434445464748494a4b4c4d4e4f",)"
2418c92c899SPavel Labath       R"("file_path":"/foo/bar.so","file_offset":0,"file_size":1234}]])",
2422f1fbaebSPavel Labath       // no file_path
2432f1fbaebSPavel Labath       R"([{"uuid":"404142434445464748494a4b4c4d4e4f","triple":"i386-pc-linux",)"
2448c92c899SPavel Labath       R"("file_offset":0,"file_size":1234}]])",
2452f1fbaebSPavel Labath       // no file_offset
2462f1fbaebSPavel Labath       R"([{"uuid":"404142434445464748494a4b4c4d4e4f","triple":"i386-pc-linux",)"
2478c92c899SPavel Labath       R"("file_path":"/foo/bar.so","file_size":1234}]])",
2482f1fbaebSPavel Labath       // no file_size
2492f1fbaebSPavel Labath       R"([{"uuid":"404142434445464748494a4b4c4d4e4f","triple":"i386-pc-linux",)"
2508c92c899SPavel Labath       R"("file_path":"/foo/bar.so","file_offset":0}]])",
2512f1fbaebSPavel Labath   };
2522f1fbaebSPavel Labath 
2532f1fbaebSPavel Labath   for (const char *response : invalid_responses) {
2542f1fbaebSPavel Labath     std::future<llvm::Optional<std::vector<ModuleSpec>>> async_result =
2552f1fbaebSPavel Labath         std::async(std::launch::async,
2562f1fbaebSPavel Labath                    [&] { return client.GetModulesInfo(file_spec, triple); });
2572f1fbaebSPavel Labath     HandlePacket(
2582f1fbaebSPavel Labath         server,
2592f1fbaebSPavel Labath         R"(jModulesInfo:[{"file":"/foo/bar.so","triple":"i386-pc-linux"}])",
2602f1fbaebSPavel Labath         response);
2612f1fbaebSPavel Labath 
2628c92c899SPavel Labath     auto result = async_result.get();
2638c92c899SPavel Labath     ASSERT_TRUE(result);
2648c92c899SPavel Labath     ASSERT_EQ(0u, result->size()) << "response was: " << response;
2652f1fbaebSPavel Labath   }
2662f1fbaebSPavel Labath }
2673aa04910SPavel Labath 
TEST_F(GDBRemoteCommunicationClientTest,TestPacketSpeedJSON)2683aa04910SPavel Labath TEST_F(GDBRemoteCommunicationClientTest, TestPacketSpeedJSON) {
26993df2fbeSPavel Labath   std::thread server_thread([this] {
2702fd9a1e0SPavel Labath     for (;;) {
2713aa04910SPavel Labath       StringExtractorGDBRemote request;
2723aa04910SPavel Labath       PacketResult result = server.GetPacket(request);
2733aa04910SPavel Labath       if (result == PacketResult::ErrorDisconnected)
2743aa04910SPavel Labath         return;
2753aa04910SPavel Labath       ASSERT_EQ(PacketResult::Success, result);
2763aa04910SPavel Labath       StringRef ref = request.GetStringRef();
2773aa04910SPavel Labath       ASSERT_TRUE(ref.consume_front("qSpeedTest:response_size:"));
2783aa04910SPavel Labath       int size;
2793aa04910SPavel Labath       ASSERT_FALSE(ref.consumeInteger(10, size)) << "ref: " << ref;
2803aa04910SPavel Labath       std::string response(size, 'X');
2813aa04910SPavel Labath       ASSERT_EQ(PacketResult::Success, server.SendPacket(response));
2822fd9a1e0SPavel Labath     }
2833aa04910SPavel Labath   });
2843aa04910SPavel Labath 
2853aa04910SPavel Labath   StreamString ss;
2862fd9a1e0SPavel Labath   client.TestPacketSpeed(10, 32, 32, 4096, true, ss);
2873aa04910SPavel Labath   client.Disconnect();
2883aa04910SPavel Labath   server_thread.join();
2893aa04910SPavel Labath 
2902f7cfaf4SPavel Labath   GTEST_LOG_(INFO) << "Formatted output: " << ss.GetData();
291adcd0268SBenjamin Kramer   auto object_sp = StructuredData::ParseJSON(std::string(ss.GetString()));
2923aa04910SPavel Labath   ASSERT_TRUE(bool(object_sp));
2933aa04910SPavel Labath   auto dict_sp = object_sp->GetAsDictionary();
2943aa04910SPavel Labath   ASSERT_TRUE(bool(dict_sp));
2953aa04910SPavel Labath 
2963aa04910SPavel Labath   object_sp = dict_sp->GetValueForKey("packet_speeds");
2973aa04910SPavel Labath   ASSERT_TRUE(bool(object_sp));
2983aa04910SPavel Labath   dict_sp = object_sp->GetAsDictionary();
2993aa04910SPavel Labath   ASSERT_TRUE(bool(dict_sp));
3003aa04910SPavel Labath 
3013aa04910SPavel Labath   int num_packets;
3023aa04910SPavel Labath   ASSERT_TRUE(dict_sp->GetValueForKeyAsInteger("num_packets", num_packets))
3033aa04910SPavel Labath       << ss.GetString();
3043aa04910SPavel Labath   ASSERT_EQ(10, num_packets);
3053aa04910SPavel Labath }
3067993cc5eSEugene Zemtsov 
TEST_F(GDBRemoteCommunicationClientTest,SendSignalsToIgnore)3077993cc5eSEugene Zemtsov TEST_F(GDBRemoteCommunicationClientTest, SendSignalsToIgnore) {
30897206d57SZachary Turner   std::future<Status> result = std::async(std::launch::async, [&] {
3097993cc5eSEugene Zemtsov     return client.SendSignalsToIgnore({2, 3, 5, 7, 0xB, 0xD, 0x11});
3107993cc5eSEugene Zemtsov   });
3117993cc5eSEugene Zemtsov 
3127993cc5eSEugene Zemtsov   HandlePacket(server, "QPassSignals:02;03;05;07;0b;0d;11", "OK");
3137993cc5eSEugene Zemtsov   EXPECT_TRUE(result.get().Success());
3147993cc5eSEugene Zemtsov 
3157993cc5eSEugene Zemtsov   result = std::async(std::launch::async, [&] {
3167993cc5eSEugene Zemtsov     return client.SendSignalsToIgnore(std::vector<int32_t>());
3177993cc5eSEugene Zemtsov   });
3187993cc5eSEugene Zemtsov 
3197993cc5eSEugene Zemtsov   HandlePacket(server, "QPassSignals:", "OK");
3207993cc5eSEugene Zemtsov   EXPECT_TRUE(result.get().Success());
3217993cc5eSEugene Zemtsov }
32248d1427cSStephane Sezer 
TEST_F(GDBRemoteCommunicationClientTest,GetMemoryRegionInfo)32348d1427cSStephane Sezer TEST_F(GDBRemoteCommunicationClientTest, GetMemoryRegionInfo) {
32448d1427cSStephane Sezer   const lldb::addr_t addr = 0xa000;
32548d1427cSStephane Sezer   MemoryRegionInfo region_info;
32697206d57SZachary Turner   std::future<Status> result = std::async(std::launch::async, [&] {
32748d1427cSStephane Sezer     return client.GetMemoryRegionInfo(addr, region_info);
32848d1427cSStephane Sezer   });
32948d1427cSStephane Sezer 
33048d1427cSStephane Sezer   HandlePacket(server,
33148d1427cSStephane Sezer       "qMemoryRegionInfo:a000",
33248d1427cSStephane Sezer       "start:a000;size:2000;permissions:rx;name:2f666f6f2f6261722e736f;");
33347776cbdSPavel Labath   if (XMLDocument::XMLEnabled()) {
33447776cbdSPavel Labath     // In case we have XML support, this will also do a "qXfer:memory-map".
33547776cbdSPavel Labath     // Preceeded by a query for supported extensions. Pretend we don't support
33647776cbdSPavel Labath     // that.
33747776cbdSPavel Labath     HandlePacket(server, testing::StartsWith("qSupported:"), "");
33847776cbdSPavel Labath   }
33948d1427cSStephane Sezer   EXPECT_TRUE(result.get().Success());
34047776cbdSPavel Labath   EXPECT_EQ(addr, region_info.GetRange().GetRangeBase());
34147776cbdSPavel Labath   EXPECT_EQ(0x2000u, region_info.GetRange().GetByteSize());
34247776cbdSPavel Labath   EXPECT_EQ(MemoryRegionInfo::eYes, region_info.GetReadable());
34347776cbdSPavel Labath   EXPECT_EQ(MemoryRegionInfo::eNo, region_info.GetWritable());
34447776cbdSPavel Labath   EXPECT_EQ(MemoryRegionInfo::eYes, region_info.GetExecutable());
34547776cbdSPavel Labath   EXPECT_EQ("/foo/bar.so", region_info.GetName().GetStringRef());
34632541685SDavid Spickett   EXPECT_EQ(MemoryRegionInfo::eDontKnow, region_info.GetMemoryTagged());
34732541685SDavid Spickett 
34832541685SDavid Spickett   result = std::async(std::launch::async, [&] {
34932541685SDavid Spickett     return client.GetMemoryRegionInfo(addr, region_info);
35032541685SDavid Spickett   });
35132541685SDavid Spickett 
35232541685SDavid Spickett   HandlePacket(server, "qMemoryRegionInfo:a000",
35332541685SDavid Spickett                "start:a000;size:2000;flags:;");
35432541685SDavid Spickett   EXPECT_TRUE(result.get().Success());
35532541685SDavid Spickett   EXPECT_EQ(MemoryRegionInfo::eNo, region_info.GetMemoryTagged());
35632541685SDavid Spickett 
35732541685SDavid Spickett   result = std::async(std::launch::async, [&] {
35832541685SDavid Spickett     return client.GetMemoryRegionInfo(addr, region_info);
35932541685SDavid Spickett   });
36032541685SDavid Spickett 
36132541685SDavid Spickett   HandlePacket(server, "qMemoryRegionInfo:a000",
36232541685SDavid Spickett                "start:a000;size:2000;flags: mt  zz mt  ;");
36332541685SDavid Spickett   EXPECT_TRUE(result.get().Success());
36432541685SDavid Spickett   EXPECT_EQ(MemoryRegionInfo::eYes, region_info.GetMemoryTagged());
36548d1427cSStephane Sezer }
36648d1427cSStephane Sezer 
TEST_F(GDBRemoteCommunicationClientTest,GetMemoryRegionInfoInvalidResponse)36748d1427cSStephane Sezer TEST_F(GDBRemoteCommunicationClientTest, GetMemoryRegionInfoInvalidResponse) {
36848d1427cSStephane Sezer   const lldb::addr_t addr = 0x4000;
36948d1427cSStephane Sezer   MemoryRegionInfo region_info;
37097206d57SZachary Turner   std::future<Status> result = std::async(std::launch::async, [&] {
37148d1427cSStephane Sezer     return client.GetMemoryRegionInfo(addr, region_info);
37248d1427cSStephane Sezer   });
37348d1427cSStephane Sezer 
37448d1427cSStephane Sezer   HandlePacket(server, "qMemoryRegionInfo:4000", "start:4000;size:0000;");
37547776cbdSPavel Labath   if (XMLDocument::XMLEnabled()) {
37647776cbdSPavel Labath     // In case we have XML support, this will also do a "qXfer:memory-map".
37747776cbdSPavel Labath     // Preceeded by a query for supported extensions. Pretend we don't support
37847776cbdSPavel Labath     // that.
37947776cbdSPavel Labath     HandlePacket(server, testing::StartsWith("qSupported:"), "");
38047776cbdSPavel Labath   }
38148d1427cSStephane Sezer   EXPECT_FALSE(result.get().Success());
38248d1427cSStephane Sezer }
383e714c4f5SRavitheja Addepally 
TEST_F(GDBRemoteCommunicationClientTest,SendTraceSupportedPacket)3840b697561SWalter Erquinigo TEST_F(GDBRemoteCommunicationClientTest, SendTraceSupportedPacket) {
3850b697561SWalter Erquinigo   TraceSupportedResponse trace_type;
386d2f18e6bSWalter Erquinigo   std::string error_message;
387d2f18e6bSWalter Erquinigo   auto callback = [&] {
388379f24ffSJim Ingham     std::chrono::seconds timeout(10);
3890b697561SWalter Erquinigo     if (llvm::Expected<TraceSupportedResponse> trace_type_or_err =
390379f24ffSJim Ingham             client.SendTraceSupported(timeout)) {
391d2f18e6bSWalter Erquinigo       trace_type = *trace_type_or_err;
392d2f18e6bSWalter Erquinigo       error_message = "";
393d2f18e6bSWalter Erquinigo       return true;
394d2f18e6bSWalter Erquinigo     } else {
395d2f18e6bSWalter Erquinigo       trace_type = {};
396d2f18e6bSWalter Erquinigo       error_message = llvm::toString(trace_type_or_err.takeError());
397d2f18e6bSWalter Erquinigo       return false;
398d2f18e6bSWalter Erquinigo     }
399d2f18e6bSWalter Erquinigo   };
400d2f18e6bSWalter Erquinigo 
40121555fffSWalter Erquinigo   // Success response
40221555fffSWalter Erquinigo   {
403d2f18e6bSWalter Erquinigo     std::future<bool> result = std::async(std::launch::async, callback);
40421555fffSWalter Erquinigo 
40521555fffSWalter Erquinigo     HandlePacket(
4060b697561SWalter Erquinigo         server, "jLLDBTraceSupported",
40721555fffSWalter Erquinigo         R"({"name":"intel-pt","description":"Intel Processor Trace"}])");
40821555fffSWalter Erquinigo 
409d2f18e6bSWalter Erquinigo     EXPECT_TRUE(result.get());
410d2f18e6bSWalter Erquinigo     ASSERT_STREQ(trace_type.name.c_str(), "intel-pt");
411d2f18e6bSWalter Erquinigo     ASSERT_STREQ(trace_type.description.c_str(), "Intel Processor Trace");
41221555fffSWalter Erquinigo   }
41321555fffSWalter Erquinigo 
41421555fffSWalter Erquinigo   // Error response - wrong json
41521555fffSWalter Erquinigo   {
416d2f18e6bSWalter Erquinigo     std::future<bool> result = std::async(std::launch::async, callback);
41721555fffSWalter Erquinigo 
4180b697561SWalter Erquinigo     HandlePacket(server, "jLLDBTraceSupported", R"({"type":"intel-pt"}])");
41921555fffSWalter Erquinigo 
420d2f18e6bSWalter Erquinigo     EXPECT_FALSE(result.get());
421a4ee79c8SWalter Erquinigo     ASSERT_STREQ(error_message.c_str(), "missing value at TraceSupportedResponse.description");
42221555fffSWalter Erquinigo   }
42321555fffSWalter Erquinigo 
42421555fffSWalter Erquinigo   // Error response
42521555fffSWalter Erquinigo   {
426d2f18e6bSWalter Erquinigo     std::future<bool> result = std::async(std::launch::async, callback);
42721555fffSWalter Erquinigo 
4280b697561SWalter Erquinigo     HandlePacket(server, "jLLDBTraceSupported", "E23");
429d2f18e6bSWalter Erquinigo 
430d2f18e6bSWalter Erquinigo     EXPECT_FALSE(result.get());
43121555fffSWalter Erquinigo   }
43221555fffSWalter Erquinigo 
43321555fffSWalter Erquinigo   // Error response with error message
43421555fffSWalter Erquinigo   {
435d2f18e6bSWalter Erquinigo     std::future<bool> result = std::async(std::launch::async, callback);
43621555fffSWalter Erquinigo 
4370b697561SWalter Erquinigo     HandlePacket(server, "jLLDBTraceSupported",
43821555fffSWalter Erquinigo                  "E23;50726F63657373206E6F742072756E6E696E672E");
439d2f18e6bSWalter Erquinigo 
440d2f18e6bSWalter Erquinigo     EXPECT_FALSE(result.get());
441d2f18e6bSWalter Erquinigo     ASSERT_STREQ(error_message.c_str(), "Process not running.");
44221555fffSWalter Erquinigo   }
44321555fffSWalter Erquinigo }
44421555fffSWalter Erquinigo 
TEST_F(GDBRemoteCommunicationClientTest,GetQOffsets)445d4eca120SPavel Labath TEST_F(GDBRemoteCommunicationClientTest, GetQOffsets) {
446d4eca120SPavel Labath   const auto &GetQOffsets = [&](llvm::StringRef response) {
447d4eca120SPavel Labath     std::future<Optional<QOffsets>> result = std::async(
448d4eca120SPavel Labath         std::launch::async, [&] { return client.GetQOffsets(); });
449d4eca120SPavel Labath 
450d4eca120SPavel Labath     HandlePacket(server, "qOffsets", response);
451d4eca120SPavel Labath     return result.get();
452d4eca120SPavel Labath   };
453d4eca120SPavel Labath   EXPECT_EQ((QOffsets{false, {0x1234, 0x1234}}),
454d4eca120SPavel Labath             GetQOffsets("Text=1234;Data=1234"));
455d4eca120SPavel Labath   EXPECT_EQ((QOffsets{false, {0x1234, 0x1234, 0x1234}}),
456d4eca120SPavel Labath             GetQOffsets("Text=1234;Data=1234;Bss=1234"));
457d4eca120SPavel Labath   EXPECT_EQ((QOffsets{true, {0x1234}}), GetQOffsets("TextSeg=1234"));
458d4eca120SPavel Labath   EXPECT_EQ((QOffsets{true, {0x1234, 0x2345}}),
459d4eca120SPavel Labath             GetQOffsets("TextSeg=1234;DataSeg=2345"));
460d4eca120SPavel Labath 
461d4eca120SPavel Labath   EXPECT_EQ(llvm::None, GetQOffsets("E05"));
462d4eca120SPavel Labath   EXPECT_EQ(llvm::None, GetQOffsets("Text=bogus"));
463d4eca120SPavel Labath   EXPECT_EQ(llvm::None, GetQOffsets("Text=1234"));
464d4eca120SPavel Labath   EXPECT_EQ(llvm::None, GetQOffsets("Text=1234;Data=1234;"));
465d4eca120SPavel Labath   EXPECT_EQ(llvm::None, GetQOffsets("Text=1234;Data=1234;Bss=1234;"));
466d4eca120SPavel Labath   EXPECT_EQ(llvm::None, GetQOffsets("TEXTSEG=1234"));
467d4eca120SPavel Labath   EXPECT_EQ(llvm::None, GetQOffsets("TextSeg=0x1234"));
468d4eca120SPavel Labath   EXPECT_EQ(llvm::None, GetQOffsets("TextSeg=12345678123456789"));
469d4eca120SPavel Labath }
4706e5c4a44SDavid Spickett 
4715d343620SDavid Spickett static void
check_qmemtags(TestClient & client,MockServer & server,size_t read_len,int32_t type,const char * packet,llvm::StringRef response,llvm::Optional<std::vector<uint8_t>> expected_tag_data)4725d343620SDavid Spickett check_qmemtags(TestClient &client, MockServer &server, size_t read_len,
473555cd031SDavid Spickett                int32_t type, const char *packet, llvm::StringRef response,
4745d343620SDavid Spickett                llvm::Optional<std::vector<uint8_t>> expected_tag_data) {
475555cd031SDavid Spickett   const auto &ReadMemoryTags = [&]() {
4765d343620SDavid Spickett     std::future<DataBufferSP> result = std::async(std::launch::async, [&] {
477555cd031SDavid Spickett       return client.ReadMemoryTags(0xDEF0, read_len, type);
4785d343620SDavid Spickett     });
4795d343620SDavid Spickett 
4805d343620SDavid Spickett     HandlePacket(server, packet, response);
4815d343620SDavid Spickett     return result.get();
4825d343620SDavid Spickett   };
4835d343620SDavid Spickett 
484555cd031SDavid Spickett   auto result = ReadMemoryTags();
4855d343620SDavid Spickett   if (expected_tag_data) {
4865d343620SDavid Spickett     ASSERT_TRUE(result);
4875d343620SDavid Spickett     llvm::ArrayRef<uint8_t> expected(*expected_tag_data);
4885d343620SDavid Spickett     llvm::ArrayRef<uint8_t> got = result->GetData();
4895d343620SDavid Spickett     ASSERT_THAT(expected, testing::ContainerEq(got));
4905d343620SDavid Spickett   } else {
4915d343620SDavid Spickett     ASSERT_FALSE(result);
4925d343620SDavid Spickett   }
4935d343620SDavid Spickett }
4945d343620SDavid Spickett 
TEST_F(GDBRemoteCommunicationClientTest,ReadMemoryTags)4955d343620SDavid Spickett TEST_F(GDBRemoteCommunicationClientTest, ReadMemoryTags) {
4965d343620SDavid Spickett   // Zero length reads are valid
497555cd031SDavid Spickett   check_qmemtags(client, server, 0, 1, "qMemTags:def0,0:1", "m",
4985d343620SDavid Spickett                  std::vector<uint8_t>{});
4995d343620SDavid Spickett 
500555cd031SDavid Spickett   // Type can be negative. Put into the packet as the raw bytes
501555cd031SDavid Spickett   // (as opposed to a literal -1)
502555cd031SDavid Spickett   check_qmemtags(client, server, 0, -1, "qMemTags:def0,0:ffffffff", "m",
503555cd031SDavid Spickett                  std::vector<uint8_t>{});
504555cd031SDavid Spickett   check_qmemtags(client, server, 0, std::numeric_limits<int32_t>::min(),
505555cd031SDavid Spickett                  "qMemTags:def0,0:80000000", "m", std::vector<uint8_t>{});
506555cd031SDavid Spickett   check_qmemtags(client, server, 0, std::numeric_limits<int32_t>::max(),
507555cd031SDavid Spickett                  "qMemTags:def0,0:7fffffff", "m", std::vector<uint8_t>{});
508555cd031SDavid Spickett 
5095d343620SDavid Spickett   // The client layer does not check the length of the received data.
5105d343620SDavid Spickett   // All we need is the "m" and for the decode to use all of the chars
511555cd031SDavid Spickett   check_qmemtags(client, server, 32, 2, "qMemTags:def0,20:2", "m09",
5125d343620SDavid Spickett                  std::vector<uint8_t>{0x9});
5135d343620SDavid Spickett 
5145d343620SDavid Spickett   // Zero length response is fine as long as the "m" is present
515555cd031SDavid Spickett   check_qmemtags(client, server, 0, 0x34, "qMemTags:def0,0:34", "m",
5165d343620SDavid Spickett                  std::vector<uint8_t>{});
5175d343620SDavid Spickett 
5185d343620SDavid Spickett   // Normal responses
519555cd031SDavid Spickett   check_qmemtags(client, server, 16, 1, "qMemTags:def0,10:1", "m66",
5205d343620SDavid Spickett                  std::vector<uint8_t>{0x66});
521555cd031SDavid Spickett   check_qmemtags(client, server, 32, 1, "qMemTags:def0,20:1", "m0102",
5225d343620SDavid Spickett                  std::vector<uint8_t>{0x1, 0x2});
5235d343620SDavid Spickett 
5245d343620SDavid Spickett   // Empty response is an error
525555cd031SDavid Spickett   check_qmemtags(client, server, 17, 1, "qMemTags:def0,11:1", "", llvm::None);
5265d343620SDavid Spickett   // Usual error response
527555cd031SDavid Spickett   check_qmemtags(client, server, 17, 1, "qMemTags:def0,11:1", "E01",
528555cd031SDavid Spickett                  llvm::None);
5295d343620SDavid Spickett   // Leading m missing
530555cd031SDavid Spickett   check_qmemtags(client, server, 17, 1, "qMemTags:def0,11:1", "01", llvm::None);
5315d343620SDavid Spickett   // Anything other than m is an error
532555cd031SDavid Spickett   check_qmemtags(client, server, 17, 1, "qMemTags:def0,11:1", "z01",
533555cd031SDavid Spickett                  llvm::None);
5345d343620SDavid Spickett   // Decoding tag data doesn't use all the chars in the packet
535555cd031SDavid Spickett   check_qmemtags(client, server, 32, 1, "qMemTags:def0,20:1", "m09zz",
536555cd031SDavid Spickett                  llvm::None);
5375d343620SDavid Spickett   // Data that is not hex bytes
538555cd031SDavid Spickett   check_qmemtags(client, server, 32, 1, "qMemTags:def0,20:1", "mhello",
5395d343620SDavid Spickett                  llvm::None);
5405d343620SDavid Spickett   // Data is not a complete hex char
541555cd031SDavid Spickett   check_qmemtags(client, server, 32, 1, "qMemTags:def0,20:1", "m9", llvm::None);
5425d343620SDavid Spickett   // Data has a trailing hex char
543555cd031SDavid Spickett   check_qmemtags(client, server, 32, 1, "qMemTags:def0,20:1", "m01020",
5445d343620SDavid Spickett                  llvm::None);
5455d343620SDavid Spickett }
5465ea091a8SDavid Spickett 
check_Qmemtags(TestClient & client,MockServer & server,lldb::addr_t addr,size_t len,int32_t type,const std::vector<uint8_t> & tags,const char * packet,llvm::StringRef response,bool should_succeed)5475ea091a8SDavid Spickett static void check_Qmemtags(TestClient &client, MockServer &server,
5485ea091a8SDavid Spickett                            lldb::addr_t addr, size_t len, int32_t type,
5495ea091a8SDavid Spickett                            const std::vector<uint8_t> &tags, const char *packet,
5505ea091a8SDavid Spickett                            llvm::StringRef response, bool should_succeed) {
5515ea091a8SDavid Spickett   const auto &WriteMemoryTags = [&]() {
5525ea091a8SDavid Spickett     std::future<Status> result = std::async(std::launch::async, [&] {
5535ea091a8SDavid Spickett       return client.WriteMemoryTags(addr, len, type, tags);
5545ea091a8SDavid Spickett     });
5555ea091a8SDavid Spickett 
5565ea091a8SDavid Spickett     HandlePacket(server, packet, response);
5575ea091a8SDavid Spickett     return result.get();
5585ea091a8SDavid Spickett   };
5595ea091a8SDavid Spickett 
5605ea091a8SDavid Spickett   auto result = WriteMemoryTags();
5615ea091a8SDavid Spickett   if (should_succeed)
5625ea091a8SDavid Spickett     ASSERT_TRUE(result.Success());
5635ea091a8SDavid Spickett   else
5645ea091a8SDavid Spickett     ASSERT_TRUE(result.Fail());
5655ea091a8SDavid Spickett }
5665ea091a8SDavid Spickett 
TEST_F(GDBRemoteCommunicationClientTest,WriteMemoryTags)5675ea091a8SDavid Spickett TEST_F(GDBRemoteCommunicationClientTest, WriteMemoryTags) {
5685ea091a8SDavid Spickett   check_Qmemtags(client, server, 0xABCD, 0x20, 1,
5695ea091a8SDavid Spickett                  std::vector<uint8_t>{0x12, 0x34}, "QMemTags:abcd,20:1:1234",
5705ea091a8SDavid Spickett                  "OK", true);
5715ea091a8SDavid Spickett 
5725ea091a8SDavid Spickett   // The GDB layer doesn't care that the number of tags !=
5735ea091a8SDavid Spickett   // the length of the write.
5745ea091a8SDavid Spickett   check_Qmemtags(client, server, 0x4321, 0x20, 9, std::vector<uint8_t>{},
5755ea091a8SDavid Spickett                  "QMemTags:4321,20:9:", "OK", true);
5765ea091a8SDavid Spickett 
5775ea091a8SDavid Spickett   check_Qmemtags(client, server, 0x8877, 0x123, 0x34,
5785ea091a8SDavid Spickett                  std::vector<uint8_t>{0x55, 0x66, 0x77},
5795ea091a8SDavid Spickett                  "QMemTags:8877,123:34:556677", "E01", false);
5805ea091a8SDavid Spickett 
5815ea091a8SDavid Spickett   // Type is a signed integer but is packed as its raw bytes,
5825ea091a8SDavid Spickett   // instead of having a +/-.
5835ea091a8SDavid Spickett   check_Qmemtags(client, server, 0x456789, 0, -1, std::vector<uint8_t>{0x99},
5845ea091a8SDavid Spickett                  "QMemTags:456789,0:ffffffff:99", "E03", false);
5855ea091a8SDavid Spickett   check_Qmemtags(client, server, 0x456789, 0,
5865ea091a8SDavid Spickett                  std::numeric_limits<int32_t>::max(),
5875ea091a8SDavid Spickett                  std::vector<uint8_t>{0x99}, "QMemTags:456789,0:7fffffff:99",
5885ea091a8SDavid Spickett                  "E03", false);
5895ea091a8SDavid Spickett   check_Qmemtags(client, server, 0x456789, 0,
5905ea091a8SDavid Spickett                  std::numeric_limits<int32_t>::min(),
5915ea091a8SDavid Spickett                  std::vector<uint8_t>{0x99}, "QMemTags:456789,0:80000000:99",
5925ea091a8SDavid Spickett                  "E03", false);
5935ea091a8SDavid Spickett }
594