blob: 398b3bae310666be4bbc3a0180363a4c02d2885b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
//===-- Environment.h -------------------------------------------*- C++ -*-===//
//
// 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
//
//===----------------------------------------------------------------------===//
#ifndef LLDB_UTILITY_ENVIRONMENT_H
#define LLDB_UTILITY_ENVIRONMENT_H
#include "llvm/ADT/StringMap.h"
#include "llvm/Support/Allocator.h"
#include "llvm/Support/FormatProviders.h"
namespace lldb_private {
class Environment : private llvm::StringMap<std::string> {
using Base = llvm::StringMap<std::string>;
public:
class Envp {
public:
Envp(Envp &&RHS) = default;
Envp &operator=(Envp &&RHS) = default;
char *const *get() const { return Data; }
operator char *const *() const { return get(); }
private:
explicit Envp(const Environment &Env);
char *make_entry(llvm::StringRef Key, llvm::StringRef Value);
Envp(const Envp &) = delete;
Envp &operator=(const Envp &) = delete;
friend class Environment;
llvm::BumpPtrAllocator Allocator;
char **Data;
};
using Base::const_iterator;
using Base::iterator;
using Base::value_type;
using Base::begin;
using Base::clear;
using Base::count;
using Base::empty;
using Base::end;
using Base::erase;
using Base::find;
using Base::insert;
using Base::lookup;
using Base::size;
using Base::try_emplace;
using Base::operator[];
Environment() : Base() {}
Environment(const Environment &RHS) : Base(RHS) {}
Environment(Environment &&RHS) : Base(std::move(RHS)) {}
Environment(char *const *Env)
: Environment(const_cast<const char *const *>(Env)) {}
Environment(const char *const *Env);
Environment &operator=(Environment RHS) {
Base::operator=(std::move(RHS));
return *this;
}
std::pair<iterator, bool> insert(llvm::StringRef KeyEqValue) {
return insert(KeyEqValue.split('='));
}
void insert(const_iterator first, const_iterator last);
Envp getEnvp() const { return Envp(*this); }
static std::string compose(const value_type &KeyValue) {
return (KeyValue.first() + "=" + KeyValue.second).str();
}
};
} // namespace lldb_private
namespace llvm {
template <> struct format_provider<lldb_private::Environment> {
static void format(const lldb_private::Environment &Env, raw_ostream &Stream,
StringRef Style) {
for (const auto &KV : Env)
Stream << "env[" << KV.first() << "] = " << KV.second << "\n";
}
};
} // namespace llvm
#endif // #ifndef LLDB_UTILITY_ENVIRONMENT_H
|
