/
serve.cpp
66 lines (56 loc) · 2.26 KB
/
serve.cpp
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
#include "internal/Version.h"
#include "seasocks/AccessControl.h"
#include "seasocks/UsernameAccessControl.h"
#include "seasocks/PrintfLogger.h"
#include "seasocks/Server.h"
#include "seasocks/SsoAuthenticator.h"
#include "seasocks/StringUtil.h"
#include <tclap/CmdLine.h>
#include <iostream>
#include <memory>
using namespace SeaSocks;
using namespace TCLAP;
namespace {
const char description[] = "Serve static files over HTTP.";
}
int main(int argc, const char* argv[]) {
CmdLine cmd(description, ' ', SEASOCKS_VERSION_STRING);
ValueArg<int> portArg("p", "port", "Listen for incoming connections on PORT", false, 80, "PORT", cmd);
SwitchArg ssoArg("s", "enable-sso", "Protect content with Single Sign On", cmd);
SwitchArg verboseArg("v", "verbose", "Output verbose debug information", cmd);
ValueArg<std::string> ssoEnvironmentArg(
"e", "sso-environment", "Use SSO-ENV as the SSO environment (default PROD)",
false, "PROD", "SSO-ENV", cmd);
ValueArg<std::string> authorizedUsersArg(
"u", "authorized-users", "Serve content only to USERS (comma-separated)",
false, "", "USERS", cmd);
ValueArg<int> lameTimeoutArg(
"", "lame-connection-timeout", "Time out lame connections after SECS seconds",
false, 0, "SECS", cmd);
UnlabeledValueArg<std::string> rootArg("serve-from-dir", "Use DIR as file serving root", true, "", "DIR", cmd);
cmd.parse(argc, argv);
std::shared_ptr<Logger> logger(
new PrintfLogger(verboseArg.getValue() ? Logger::DEBUG : Logger::ACCESS));
Server server(logger);
if (lameTimeoutArg.isSet()) {
server.setLameConnectionTimeoutSeconds(lameTimeoutArg.getValue());
}
if (ssoArg.getValue()) {
auto ssoOptions = SsoOptions::environment(ssoEnvironmentArg.getValue());
if (ssoOptions.authServer.size() == 0) {
std::cerr << "Unrecognized SSO environment '" << ssoEnvironmentArg.getValue() << "'" << std::endl;
return 1;
}
auto users = split(authorizedUsersArg.getValue(), ',');
if (users.empty()) {
std::cerr << "No users specified" << std::endl;
return 2;
}
std::set<std::string> userSet;
userSet.insert(users.begin(), users.end());
ssoOptions.accessController.reset(new UsernameAccessControl(userSet));
server.enableSingleSignOn(ssoOptions);
}
server.serve(rootArg.getValue().c_str(), portArg.getValue());
return 0;
}