hydrar/mistserver
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added local mode to load balancer, which only allows control from local interfaces.

Browse source
This commit is contained in:
Thulinma 2017-09-29 17:21:57 +02:00
parent e05268f670
commit 09cea3324e
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/utils/util_load.cpp
View file

@ -13,6 +13,7 @@
Util::Config *cfg = 0; Util::Config *cfg = 0;
std::string passphrase; std::string passphrase;
std::string fallback; std::string fallback;
bool localMode = false;
unsigned int weight_cpu = 500; unsigned int weight_cpu = 500;
unsigned int weight_ram = 500; unsigned int weight_ram = 500;
@ -297,6 +298,12 @@ int handleRequest(Socket::Connection &conn){
if ((conn.spool() || conn.Received().size()) && H.Read(conn)){ if ((conn.spool() || conn.Received().size()) && H.Read(conn)){
// Special commands // Special commands
if (H.url.size() == 1){ if (H.url.size() == 1){
if (localMode && !conn.isLocal()){
H.SetBody("Configuration only accessible from local interfaces");
H.SendResponse("403", "Forbidden", conn);
H.Clean();
continue;
}
std::string host = H.GetVar("host"); std::string host = H.GetVar("host");
std::string viewers = H.GetVar("viewers"); std::string viewers = H.GetVar("viewers");
std::string source = H.GetVar("source"); std::string source = H.GetVar("source");
@ -640,6 +647,12 @@ int main(int argc, char **argv){
opt["value"].append((long long)weight_bonus); opt["value"].append((long long)weight_bonus);
conf.addOption("extra", opt); conf.addOption("extra", opt);
opt.null();
opt["short"] = "L";
opt["long"] = "localmode";
opt["help"] = "Control only from local interfaces, request balance from all";
conf.addOption("localmode", opt);
conf.parseArgs(argc, argv); conf.parseArgs(argc, argv);
passphrase = conf.getOption("passphrase").asStringRef(); passphrase = conf.getOption("passphrase").asStringRef();
@ -648,6 +661,8 @@ int main(int argc, char **argv){
weight_bw = conf.getInteger("bw"); weight_bw = conf.getInteger("bw");
weight_bonus = conf.getInteger("extra"); weight_bonus = conf.getInteger("extra");
fallback = conf.getString("fallback"); fallback = conf.getString("fallback");
localMode = conf.getBool("localmode");
INFO_MSG("Local control only mode is %s", localMode?"on":"off");
JSON::Value &nodes = conf.getOption("server", true); JSON::Value &nodes = conf.getOption("server", true);
conf.activate(); conf.activate();