lldb/unittests/Utility/ProcessInstanceInfoTest.cpp

   1 //===-- ProcessInstanceInfoTest.cpp ---------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "lldb/Target/Process.h"
  10 #include "gtest/gtest.h"
  11
  12 using namespace lldb_private;
  13
  14 namespace {
  15 /// A very simple resolver which fails for even ids and returns a simple string
  16 /// for odd ones.
  17 class DummyUserIDResolver : public UserIDResolver {
  18 protected:
  19   llvm::Optional<std::string> DoGetUserName(id_t uid) override {
  20     if (uid % 2)
  21       return ("user" + llvm::Twine(uid)).str();
  22     return llvm::None;
  23   }
  24
  25   llvm::Optional<std::string> DoGetGroupName(id_t gid) override {
  26     if (gid % 2)
  27       return ("group" + llvm::Twine(gid)).str();
  28     return llvm::None;
  29   }
  30 };
  31 } // namespace
  32
  33 TEST(ProcessInstanceInfo, Dump) {
  34   ProcessInstanceInfo info("a.out", ArchSpec("x86_64-pc-linux"), 47);
  35   info.SetUserID(1);
  36   info.SetEffectiveUserID(2);
  37   info.SetGroupID(3);
  38   info.SetEffectiveGroupID(4);
  39
  40   DummyUserIDResolver resolver;
  41   StreamString s;
  42   info.Dump(s, resolver);
  43   EXPECT_STREQ(R"(    pid = 47
  44    name = a.out
  45    file = a.out
  46    arch = x86_64-pc-linux
  47     uid = 1     (user1)
  48     gid = 3     (group3)
  49    euid = 2     ()
  50    egid = 4     ()
  51 )",
  52                s.GetData());
  53 }
  54
  55 TEST(ProcessInstanceInfo, DumpTable) {
  56   ProcessInstanceInfo info("a.out", ArchSpec("x86_64-pc-linux"), 47);
  57   info.SetUserID(1);
  58   info.SetEffectiveUserID(2);
  59   info.SetGroupID(3);
  60   info.SetEffectiveGroupID(4);
  61
  62   DummyUserIDResolver resolver;
  63   StreamString s;
  64
  65   const bool show_args = false;
  66   const bool verbose = true;
  67   ProcessInstanceInfo::DumpTableHeader(s, show_args, verbose);
  68   info.DumpAsTableRow(s, resolver, show_args, verbose);
  69   EXPECT_STREQ(
  70       R"(PID    PARENT USER       GROUP      EFF USER   EFF GROUP  TRIPLE                         ARGUMENTS
  71 ====== ====== ========== ========== ========== ========== ============================== ============================
  72 47     0      user1      group3     2          4          x86_64-pc-linux
  73 )",
  74       s.GetData());
  75 }
  76
  77 TEST(ProcessInstanceInfo, DumpTable_invalidUID) {
  78   ProcessInstanceInfo info("a.out", ArchSpec("aarch64-unknown-linux-android"), 47);
  79
  80   DummyUserIDResolver resolver;
  81   StreamString s;
  82
  83   const bool show_args = false;
  84   const bool verbose = false;
  85   ProcessInstanceInfo::DumpTableHeader(s, show_args, verbose);
  86   info.DumpAsTableRow(s, resolver, show_args, verbose);
  87   EXPECT_STREQ(
  88       R"(PID    PARENT USER       TRIPLE                         NAME
  89 ====== ====== ========== ============================== ============================
  90 47     0                 aarch64-unknown-linux-android  a.out
  91 )",
  92       s.GetData());
  93 }
  94
  95 TEST(ProcessInstanceInfoMatch, Name) {
  96   ProcessInstanceInfo info_bar, info_empty;
  97   info_bar.GetExecutableFile().SetFile("/foo/bar", FileSpec::Style::posix);
  98
  99   ProcessInstanceInfoMatch match;
 100   match.SetNameMatchType(NameMatch::Equals);
 101   match.GetProcessInfo().GetExecutableFile().SetFile("bar",
 102                                                      FileSpec::Style::posix);
 103
 104   EXPECT_TRUE(match.Matches(info_bar));
 105   EXPECT_FALSE(match.Matches(info_empty));
 106
 107   match.GetProcessInfo().GetExecutableFile() = FileSpec();
 108   EXPECT_TRUE(match.Matches(info_bar));
 109   EXPECT_TRUE(match.Matches(info_empty));
 110 }
 111
 112 TEST(ProcessInstanceInfo, Yaml) {
 113   std::string buffer;
 114   llvm::raw_string_ostream os(buffer);
 115
 116   // Serialize.
 117   ProcessInstanceInfo info("a.out", ArchSpec("x86_64-pc-linux"), 47);
 118   info.SetUserID(1);
 119   info.SetEffectiveUserID(2);
 120   info.SetGroupID(3);
 121   info.SetEffectiveGroupID(4);
 122   llvm::yaml::Output yout(os);
 123   yout << info;
 124   os.flush();
 125
 126   // Deserialize.
 127   ProcessInstanceInfo deserialized;
 128   llvm::yaml::Input yin(buffer);
 129   yin >> deserialized;
 130
 131   EXPECT_EQ(deserialized.GetNameAsStringRef(), info.GetNameAsStringRef());
 132   EXPECT_EQ(deserialized.GetArchitecture(), info.GetArchitecture());
 133   EXPECT_EQ(deserialized.GetUserID(), info.GetUserID());
 134   EXPECT_EQ(deserialized.GetGroupID(), info.GetGroupID());
 135   EXPECT_EQ(deserialized.GetEffectiveUserID(), info.GetEffectiveUserID());
 136   EXPECT_EQ(deserialized.GetEffectiveGroupID(), info.GetEffectiveGroupID());
 137 }
 138
 139 TEST(ProcessInstanceInfoList, Yaml) {
 140   std::string buffer;
 141   llvm::raw_string_ostream os(buffer);
 142
 143   // Serialize.
 144   ProcessInstanceInfo info("a.out", ArchSpec("x86_64-pc-linux"), 47);
 145   info.SetUserID(1);
 146   info.SetEffectiveUserID(2);
 147   info.SetGroupID(3);
 148   info.SetEffectiveGroupID(4);
 149   ProcessInstanceInfoList list;
 150   list.push_back(info);
 151   llvm::yaml::Output yout(os);
 152   yout << list;
 153   os.flush();
 154
 155   // Deserialize.
 156   ProcessInstanceInfoList deserialized;
 157   llvm::yaml::Input yin(buffer);
 158   yin >> deserialized;
 159
 160   ASSERT_EQ(deserialized.size(), static_cast<size_t>(1));
 161   EXPECT_EQ(deserialized[0].GetNameAsStringRef(), info.GetNameAsStringRef());
 162   EXPECT_EQ(deserialized[0].GetArchitecture(), info.GetArchitecture());
 163   EXPECT_EQ(deserialized[0].GetUserID(), info.GetUserID());
 164   EXPECT_EQ(deserialized[0].GetGroupID(), info.GetGroupID());
 165   EXPECT_EQ(deserialized[0].GetEffectiveUserID(), info.GetEffectiveUserID());
 166   EXPECT_EQ(deserialized[0].GetEffectiveGroupID(), info.GetEffectiveGroupID());
 167 }