lldb/source/API/SBEnvironment.cpp

   1 //===-- SBEnvironment.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/API/SBEnvironment.h"
  10 #include "Utils.h"
  11 #include "lldb/API/SBStringList.h"
  12 #include "lldb/Utility/ConstString.h"
  13 #include "lldb/Utility/Environment.h"
  14 #include "lldb/Utility/Instrumentation.h"
  15
  16 using namespace lldb;
  17 using namespace lldb_private;
  18
  19 SBEnvironment::SBEnvironment() : m_opaque_up(new Environment()) {
  20   LLDB_INSTRUMENT_VA(this);
  21 }
  22
  23 SBEnvironment::SBEnvironment(const SBEnvironment &rhs)
  24     : m_opaque_up(clone(rhs.m_opaque_up)) {
  25   LLDB_INSTRUMENT_VA(this, rhs);
  26 }
  27
  28 SBEnvironment::SBEnvironment(Environment rhs)
  29     : m_opaque_up(new Environment(std::move(rhs))) {}
  30
  31 SBEnvironment::~SBEnvironment() = default;
  32
  33 const SBEnvironment &SBEnvironment::operator=(const SBEnvironment &rhs) {
  34   LLDB_INSTRUMENT_VA(this, rhs);
  35
  36   if (this != &rhs)
  37     m_opaque_up = clone(rhs.m_opaque_up);
  38   return *this;
  39 }
  40
  41 size_t SBEnvironment::GetNumValues() {
  42   LLDB_INSTRUMENT_VA(this);
  43
  44   return m_opaque_up->size();
  45 }
  46
  47 const char *SBEnvironment::Get(const char *name) {
  48   LLDB_INSTRUMENT_VA(this, name);
  49
  50   auto entry = m_opaque_up->find(name);
  51   if (entry == m_opaque_up->end()) {
  52     return nullptr;
  53   }
  54   return ConstString(entry->second).AsCString("");
  55 }
  56
  57 const char *SBEnvironment::GetNameAtIndex(size_t index) {
  58   LLDB_INSTRUMENT_VA(this, index);
  59
  60   if (index >= GetNumValues())
  61     return nullptr;
  62   return ConstString(std::next(m_opaque_up->begin(), index)->first())
  63       .AsCString("");
  64 }
  65
  66 const char *SBEnvironment::GetValueAtIndex(size_t index) {
  67   LLDB_INSTRUMENT_VA(this, index);
  68
  69   if (index >= GetNumValues())
  70     return nullptr;
  71   return ConstString(std::next(m_opaque_up->begin(), index)->second)
  72       .AsCString("");
  73 }
  74
  75 bool SBEnvironment::Set(const char *name, const char *value, bool overwrite) {
  76   LLDB_INSTRUMENT_VA(this, name, value, overwrite);
  77
  78   if (overwrite) {
  79     m_opaque_up->insert_or_assign(name, std::string(value));
  80     return true;
  81   }
  82   return m_opaque_up->try_emplace(name, std::string(value)).second;
  83 }
  84
  85 bool SBEnvironment::Unset(const char *name) {
  86   LLDB_INSTRUMENT_VA(this, name);
  87
  88   return m_opaque_up->erase(name);
  89 }
  90
  91 SBStringList SBEnvironment::GetEntries() {
  92   LLDB_INSTRUMENT_VA(this);
  93
  94   SBStringList entries;
  95   for (const auto &KV : *m_opaque_up) {
  96     entries.AppendString(Environment::compose(KV).c_str());
  97   }
  98   return entries;
  99 }
 100
 101 void SBEnvironment::PutEntry(const char *name_and_value) {
 102   LLDB_INSTRUMENT_VA(this, name_and_value);
 103
 104   auto split = llvm::StringRef(name_and_value).split('=');
 105   m_opaque_up->insert_or_assign(split.first.str(), split.second.str());
 106 }
 107
 108 void SBEnvironment::SetEntries(const SBStringList &entries, bool append) {
 109   LLDB_INSTRUMENT_VA(this, entries, append);
 110
 111   if (!append)
 112     m_opaque_up->clear();
 113   for (size_t i = 0; i < entries.GetSize(); i++) {
 114     PutEntry(entries.GetStringAtIndex(i));
 115   }
 116 }
 117
 118 void SBEnvironment::Clear() {
 119   LLDB_INSTRUMENT_VA(this);
 120
 121   m_opaque_up->clear();
 122 }
 123
 124 Environment &SBEnvironment::ref() const { return *m_opaque_up; }