Added command-line options

This commit is contained in:
Tatsuhiro Tsujikawa 2012-06-06 23:58:19 +09:00
parent 8da4938031
commit 8e0a1c91a3
5 changed files with 191 additions and 18 deletions

View File

@ -31,8 +31,12 @@
#include <signal.h> #include <signal.h>
#include <netinet/in.h> #include <netinet/in.h>
#include <arpa/inet.h> #include <arpa/inet.h>
#include <unistd.h>
#include <getopt.h>
#include <limits>
#include <cstdlib> #include <cstdlib>
#include <iostream>
#include <openssl/ssl.h> #include <openssl/ssl.h>
#include <openssl/err.h> #include <openssl/err.h>
@ -61,14 +65,17 @@ int cache_downstream_host_address()
addrinfo hints; addrinfo hints;
int rv; int rv;
char service[10]; char service[10];
snprintf(service, sizeof(service), "%u", get_config()->downstream_port); snprintf(service, sizeof(service), "%u", get_config()->downstream_port);
memset(&hints, 0, sizeof(addrinfo)); memset(&hints, 0, sizeof(addrinfo));
hints.ai_family = AF_UNSPEC; hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM; hints.ai_socktype = SOCK_STREAM;
#ifdef AI_ADDRCONFIG #ifdef AI_ADDRCONFIG
hints.ai_flags |= AI_ADDRCONFIG; hints.ai_flags |= AI_ADDRCONFIG;
#endif // AI_ADDRCONFIG #endif // AI_ADDRCONFIG
addrinfo *res; addrinfo *res;
rv = getaddrinfo(get_config()->downstream_host, service, &hints, &res); rv = getaddrinfo(get_config()->downstream_host, service, &hints, &res);
if(rv != 0) { if(rv != 0) {
LOG(FATAL) << "Unable to get downstream address: " << gai_strerror(rv); LOG(FATAL) << "Unable to get downstream address: " << gai_strerror(rv);
@ -121,7 +128,8 @@ evconnlistener* create_evlistener(ListenHandler *handler, int family)
addrinfo *res, *rp; addrinfo *res, *rp;
r = getaddrinfo(get_config()->host, service, &hints, &res); r = getaddrinfo(get_config()->host, service, &hints, &res);
if(r != 0) { if(r != 0) {
LOG(ERROR) << "getaddrinfo: " << gai_strerror(r); LOG(ERROR) << "Unable to get address for " << get_config()->host << ": "
<< gai_strerror(r);
return NULL; return NULL;
} }
for(rp = res; rp; rp = rp->ai_next) { for(rp = res; rp; rp = rp->ai_next) {
@ -195,7 +203,7 @@ int event_loop()
if(!evlistener6 && !evlistener4) { if(!evlistener6 && !evlistener4) {
LOG(FATAL) << "Failed to listen on address " LOG(FATAL) << "Failed to listen on address "
<< get_config()->host << ", port " << get_config()->port; << get_config()->host << ", port " << get_config()->port;
DIE(); exit(EXIT_FAILURE);
} }
if(get_config()->num_worker > 1) { if(get_config()->num_worker > 1) {
@ -216,21 +224,11 @@ int event_loop()
} }
} // namespace } // namespace
int main(int argc, char **argv) namespace {
void fill_default_config()
{ {
struct sigaction act;
memset(&act, 0, sizeof(struct sigaction));
act.sa_handler = SIG_IGN;
sigaction(SIGPIPE, &act, 0);
OpenSSL_add_all_algorithms();
SSL_load_error_strings();
SSL_library_init();
Log::set_severity_level(INFO);
create_config();
mod_config()->server_name = "shrpx spdylay/"SPDYLAY_VERSION; mod_config()->server_name = "shrpx spdylay/"SPDYLAY_VERSION;
mod_config()->host = "localhost";
mod_config()->port = 3000; mod_config()->port = 3000;
mod_config()->private_key_file = "server.key"; mod_config()->private_key_file = "server.key";
mod_config()->cert_file = "server.crt"; mod_config()->cert_file = "server.crt";
@ -252,7 +250,159 @@ int main(int argc, char **argv)
mod_config()->downstream_host = "localhost"; mod_config()->downstream_host = "localhost";
mod_config()->downstream_port = 80; mod_config()->downstream_port = 80;
char hostport[256];
mod_config()->num_worker = 4;
mod_config()->spdy_max_concurrent_streams =
SPDYLAY_INITIAL_MAX_CONCURRENT_STREAMS;
}
} // namespace
namespace {
int split_host_port(char *host, size_t hostlen, uint16_t *port_ptr,
const char *hostport)
{
// host and port in |hostport| is separated by single ','.
const char *p = strchr(hostport, ',');
if(!p) {
std::cerr << "Invalid host, port: " << hostport << std::endl;
return -1;
}
size_t len = p-hostport;
if(hostlen < len+1) {
std::cerr << "Hostname too long: " << hostport << std::endl;
return -1;
}
memcpy(host, hostport, len);
host[len] = '\0';
errno = 0;
unsigned long d = strtoul(p+1, 0, 10);
if(errno == 0 && 1 <= d && d <= std::numeric_limits<uint16_t>::max()) {
*port_ptr = d;
return 0;
} else {
std::cerr << "Port is invalid: " << p+1 << std::endl;
return -1;
}
}
} // namespace
namespace {
void print_usage(std::ostream& out)
{
out << "Usage: shrpx [-Dh] [-b <HOST,PORT>] [-f <HOST,PORT>] [-n <CORES>]\n"
<< " [-c <NUM>] [-L <LEVEL>] <PRIVATE_KEY> <CERT>\n"
<< "\n"
<< "A reverse proxy for SPDY/HTTPS.\n"
<< std::endl;
}
} // namespace
namespace {
void print_help(std::ostream& out)
{
fill_default_config();
print_usage(out);
out << "\n"
<< "OPTIONS:\n"
<< " -b, --backend=<HOST,PORT>\n"
<< " Set backend host and port.\n"
<< " Default: '"
<< get_config()->downstream_host << ","
<< get_config()->downstream_port << "'\n"
<< " -f, --frontend=<HOST,PORT>\n"
<< " Set frontend host and port.\n"
<< " Default: '"
<< get_config()->host << "," << get_config()->port << "'\n"
<< " -n, --workers=<CORES>\n"
<< " Set the number of worker threads.\n"
<< " -c, --spdy-max-concurrent-streams=<NUM>\n"
<< " Set the maximum number of the concurrent\n"
<< " streams in one SPDY session.\n"
<< " -L, --log-level=<LEVEL>\n"
<< " Set the severity level of log output.\n"
<< " INFO, WARNING, ERROR and FATAL\n"
<< " -D, --daemon Run in a background. If -D is used, the\n"
<< " current working directory is changed to '/'.\n"
<< " -h, --help Print this help.\n"
<< std::endl;
}
} // namespace
int main(int argc, char **argv)
{
Log::set_severity_level(WARNING);
create_config();
fill_default_config();
char frontend_host[NI_MAXHOST];
uint16_t frontend_port;
char backend_host[NI_MAXHOST];
uint16_t backend_port;
while(1) {
static option long_options[] = {
{"backend", required_argument, 0, 'b' },
{"frontend", required_argument, 0, 'f' },
{"workers", required_argument, 0, 'n' },
{"spdy-max-concurrent-streams", required_argument, 0, 'c' },
{"log-level", required_argument, 0, 'L' },
{"daemon", no_argument, 0, 'D' },
{"help", no_argument, 0, 'h' },
{0, 0, 0, 0 }
};
int option_index = 0;
int c = getopt_long(argc, argv, "DL:b:c:f:n:h", long_options,
&option_index);
if(c == -1) {
break;
}
switch(c) {
case 'D':
mod_config()->daemon = true;
break;
case 'h':
print_help(std::cout);
exit(EXIT_SUCCESS);
case 'L':
if(Log::set_severity_level_by_name(optarg) == -1) {
std::cerr << "Invalid severity level: " << optarg << std::endl;
exit(EXIT_SUCCESS);
}
break;
case 'b':
if(split_host_port(backend_host, sizeof(backend_host),
&backend_port, optarg) == -1) {
exit(EXIT_FAILURE);
} else {
mod_config()->downstream_host = backend_host;
mod_config()->downstream_port = backend_port;
}
break;
case 'f':
if(split_host_port(frontend_host, sizeof(frontend_host),
&frontend_port, optarg) == -1) {
exit(EXIT_FAILURE);
} else {
mod_config()->host = frontend_host;
mod_config()->port = frontend_port;
}
break;
case 'n':
mod_config()->num_worker = strtol(optarg, 0, 10);
break;
case 'c':
mod_config()->spdy_max_concurrent_streams = strtol(optarg, 0, 10);
break;
case '?':
exit(EXIT_FAILURE);
default:
break;
}
}
char hostport[NI_MAXHOST];
if(get_config()->downstream_port == 80) { if(get_config()->downstream_port == 80) {
mod_config()->downstream_hostport = get_config()->downstream_host; mod_config()->downstream_hostport = get_config()->downstream_host;
} else { } else {
@ -260,11 +410,19 @@ int main(int argc, char **argv)
get_config()->downstream_host, get_config()->downstream_port); get_config()->downstream_host, get_config()->downstream_port);
mod_config()->downstream_hostport = hostport; mod_config()->downstream_hostport = hostport;
} }
if(cache_downstream_host_address() == -1) { if(cache_downstream_host_address() == -1) {
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
mod_config()->num_worker = 4; struct sigaction act;
memset(&act, 0, sizeof(struct sigaction));
act.sa_handler = SIG_IGN;
sigaction(SIGPIPE, &act, 0);
OpenSSL_add_all_algorithms();
SSL_load_error_strings();
SSL_library_init();
event_loop(); event_loop();
return 0; return 0;

View File

@ -39,7 +39,8 @@ Config::Config()
downstream_port(0), downstream_port(0),
downstream_hostport(0), downstream_hostport(0),
downstream_addrlen(0), downstream_addrlen(0),
num_worker(0) num_worker(0),
spdy_max_concurrent_streams(0)
{} {}
namespace { namespace {

View File

@ -65,6 +65,7 @@ struct Config {
timeval downstream_read_timeout; timeval downstream_read_timeout;
timeval downstream_write_timeout; timeval downstream_write_timeout;
size_t num_worker; size_t num_worker;
size_t spdy_max_concurrent_streams;
Config(); Config();
}; };

View File

@ -25,6 +25,7 @@
#include "shrpx_log.h" #include "shrpx_log.h"
#include <cstdio> #include <cstdio>
#include <cstring>
namespace shrpx { namespace shrpx {
@ -39,6 +40,17 @@ void Log::set_severity_level(int severity)
severity_thres_ = severity; severity_thres_ = severity;
} }
int Log::set_severity_level_by_name(const char *name)
{
for(size_t i = 0, max = sizeof(SEVERITY_STR)/sizeof(char*); i < max; ++i) {
if(strcmp(SEVERITY_STR[i], name) == 0) {
severity_thres_ = i;
return 0;
}
}
return -1;
}
Log::Log(int severity, const char *filename, int linenum) Log::Log(int severity, const char *filename, int linenum)
: severity_(severity), : severity_(severity),
filename_(filename), filename_(filename),

View File

@ -49,6 +49,7 @@ public:
return *this; return *this;
} }
static void set_severity_level(int severity); static void set_severity_level(int severity);
static int set_severity_level_by_name(const char *name);
private: private:
int severity_; int severity_;
const char *filename_; const char *filename_;