6 |
|
|
7 |
|
#include "Mounter.hpp" |
8 |
|
#include "Unmounter.hpp" |
9 |
– |
#include "Matcher/Matcher.hpp" |
9 |
|
|
10 |
|
#include <menes-api/exename.hpp> |
11 |
|
#include <menes-app/simple.hpp> |
16 |
|
int Main(const app::Options& options) |
17 |
|
{ |
18 |
|
api::Path config(Spectre2::prefix + "/etc/spectre.xml"); |
19 |
< |
bool fork(false); |
19 |
> |
enum Mode { none, pid = 1, fork } mode(none); |
20 |
|
|
21 |
|
_foreach (const app::ArgumentList, arg, app::GetArguments()) |
22 |
|
{ |
24 |
|
|
25 |
|
if (*arg == matcher("^-config=(.*)$")) |
26 |
|
config = matcher[1]; |
27 |
+ |
else if (*arg == "-pid") |
28 |
+ |
mode = pid; |
29 |
|
else if (*arg == "-fork") |
30 |
< |
fork = true; |
30 |
> |
mode = fork; |
31 |
|
else if (*arg == "-D") |
32 |
|
Spectre2::debug = true; |
33 |
|
else |
34 |
|
{ |
35 |
< |
api::Cerr << "Usage: " << Spectre2::program << " [-config=config] [-fork] [-D]" << ios::NewLine; |
35 |
> |
api::Cerr << "Usage: " << Spectre2::program << " [-config=config] [-pid|-fork] [-D]" << ios::NewLine; |
36 |
|
|
37 |
|
return 1; |
38 |
|
} |
45 |
|
return 2; |
46 |
|
} |
47 |
|
|
48 |
< |
if (fork && api::Posix::CheckError(::fork())) |
49 |
< |
return 0; |
48 |
> |
switch (mode) |
49 |
> |
{ |
50 |
> |
case pid: |
51 |
> |
try |
52 |
> |
{ |
53 |
> |
_H<xml::Document> document(xml::Parse(config.GetPath())); |
54 |
> |
api::FileReader in(*document/"spectre"/"pid"); |
55 |
|
|
56 |
< |
Spectre2 spectre(config); |
56 |
> |
ios::ReadToWrite(in, api::Cout); |
57 |
> |
} |
58 |
> |
catch (const api::Error&) |
59 |
> |
{ |
60 |
> |
return 1; |
61 |
> |
} |
62 |
> |
break; |
63 |
> |
case fork: |
64 |
> |
if (api::Posix::CheckError(::fork())) |
65 |
> |
return 0; |
66 |
> |
default: |
67 |
> |
{ |
68 |
> |
Spectre2 spectre(config); |
69 |
> |
} |
70 |
> |
} |
71 |
|
|
72 |
|
return 0; |
73 |
|
} |
82 |
|
{ |
83 |
|
{ |
84 |
|
_H<xml::Document> document(xml::Parse(config.GetPath())); |
85 |
+ |
_H<xml::Node> spectre(*document/"spectre"); |
86 |
+ |
|
87 |
+ |
pid = *spectre/"pid"; |
88 |
+ |
|
89 |
+ |
if (!(*spectre/"prefix").IsEmpty()) |
90 |
+ |
prefix = *spectre/"prefix"; |
91 |
+ |
|
92 |
+ |
if (!(*spectre/"root").IsEmpty()) |
93 |
+ |
root = *spectre/"root"; |
94 |
|
|
95 |
< |
pid = *document/"spectre"/"pid"; |
95 |
> |
if (!(*spectre/"mount").IsEmpty()) |
96 |
> |
mount = *spectre/"mount"; |
97 |
|
} |
98 |
|
|
99 |
|
{ |
120 |
|
|
121 |
|
action.sa_sigaction = stop; |
122 |
|
|
123 |
+ |
api::Posix::CheckError(::sigaction(SIGINT, &action, NULL)); |
124 |
|
api::Posix::CheckError(::sigaction(SIGTERM, &action, NULL)); |
125 |
|
} |
126 |
|
|
133 |
|
api::Posix::CheckError(::unlink(pid.NullTerminate())); |
134 |
|
} |
135 |
|
|
136 |
< |
ext::String Spectre2::program(api::GetExecutablePath().GetName()), Spectre2::prefix(_Spectre2_prefix_), Spectre2::root(_Spectre2_root_); |
136 |
> |
ext::String Spectre2::program(api::GetExecutablePath().GetName()), Spectre2::prefix(_Spectre2_prefix_), Spectre2::root(_Spectre2_root_), Spectre2::mount(_Spectre2_mount_); |
137 |
|
bool Spectre2::debug(false); |
138 |
|
ext::Vector<Daemon*> Spectre2::daemons; |
139 |
+ |
ext::RedBlackSet<Share> Spectre2::shares; |
140 |
+ |
api::ThreadMutex Spectre2::sharesLock; |
141 |
|
|
142 |
|
extern "C" |
143 |
|
{ |