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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
#pragma once
#include <locale>
#include <absl/strings/numbers.h>
#include <signal.h>
#include "libmain/common-args.hh"
#include "libutil/args.hh"
#include "libutil/util.hh"
namespace nix {
class Exit : public std::exception {
public:
int status;
Exit() : status(0) {}
Exit(int status) : status(status) {}
virtual ~Exit();
};
int handleExceptions(const std::string& programName,
const std::function<void()>& fun);
void initNix();
void parseCmdLine(
int argc, char** argv,
std::function<bool(Strings::iterator& arg, const Strings::iterator& end)>
parseArg);
void parseCmdLine(
const std::string& programName, const Strings& args,
std::function<bool(Strings::iterator& arg, const Strings::iterator& end)>
parseArg);
void printVersion(const std::string& programName);
/* Ugh. No better place to put this. */
void printGCWarning();
class Store;
void printMissing(const ref<Store>& store, const PathSet& paths);
void printMissing(const ref<Store>& store, const PathSet& willBuild,
const PathSet& willSubstitute, const PathSet& unknown,
unsigned long long downloadSize, unsigned long long narSize);
std::string getArg(const std::string& opt, Strings::iterator& i,
const Strings::iterator& end);
template <class N>
N getIntArg(const std::string& opt, Strings::iterator& i,
const Strings::iterator& end, bool allowUnit) {
++i;
if (i == end) {
throw UsageError(format("'%1%' requires an argument") % opt);
}
std::string s = *i;
N multiplier = 1;
if (allowUnit && !s.empty()) {
char u = std::toupper(*s.rbegin());
if (std::isalpha(u)) {
if (u == 'K') {
multiplier = 1ULL << 10;
} else if (u == 'M') {
multiplier = 1ULL << 20;
} else if (u == 'G') {
multiplier = 1ULL << 30;
} else if (u == 'T') {
multiplier = 1ULL << 40;
} else {
throw UsageError(format("invalid unit specifier '%1%'") % u);
}
s.resize(s.size() - 1);
}
}
N n;
if (!absl::SimpleAtoi(s, &n)) {
throw UsageError(format("'%1%' requires an integer argument") % opt);
}
return n * multiplier;
}
struct LegacyArgs : public MixCommonArgs {
std::function<bool(Strings::iterator& arg, const Strings::iterator& end)>
parseArg;
LegacyArgs(
const std::string& programName,
std::function<bool(Strings::iterator& arg, const Strings::iterator& end)>
parseArg);
bool processFlag(Strings::iterator& pos, Strings::iterator end) override;
bool processArgs(const Strings& args, bool finish) override;
};
/* Show the manual page for the specified program. */
void showManPage(const std::string& name);
/* The constructor of this class starts a pager if stdout is a
terminal and $PAGER is set. Stdout is redirected to the pager. */
class RunPager {
public:
RunPager();
~RunPager();
private:
Pid pid;
};
extern volatile ::sig_atomic_t blockInt;
/* GC helpers. */
std::string showBytes(unsigned long long bytes);
struct GCResults;
struct PrintFreed {
bool show;
const GCResults& results;
PrintFreed(bool show, const GCResults& results)
: show(show), results(results) {}
~PrintFreed();
};
/* Install a SIGSEGV handler to detect stack overflows. */
void detectStackOverflow();
} // namespace nix
|
