157 lines
4.3 KiB
C++
157 lines
4.3 KiB
C++
/*
|
|
* Spdylay - SPDY Library
|
|
*
|
|
* Copyright (c) 2012 Tatsuhiro Tsujikawa
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
* a copy of this software and associated documentation files (the
|
|
* "Software"), to deal in the Software without restriction, including
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
* distribute, sublicense, and/or sell copies of the Software, and to
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
* the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be
|
|
* included in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
#include <unistd.h>
|
|
#include <signal.h>
|
|
#include <getopt.h>
|
|
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
#include <openssl/ssl.h>
|
|
#include <openssl/err.h>
|
|
#include <spdylay/spdylay.h>
|
|
|
|
#include "spdylay_ssl.h"
|
|
#include "SpdyServer.h"
|
|
|
|
namespace spdylay {
|
|
|
|
extern bool ssl_debug;
|
|
|
|
namespace {
|
|
void print_usage(std::ostream& out)
|
|
{
|
|
out << "Usage: spdyd [-Ddhv] PORT PRIVATE_KEY CERT" << std::endl;
|
|
}
|
|
} // namespace
|
|
|
|
namespace {
|
|
void print_help(std::ostream& out)
|
|
{
|
|
print_usage(out);
|
|
out << "\n"
|
|
<< "OPTIONS:\n"
|
|
<< " -D, --daemon Run in a background. If -D is used, the\n"
|
|
<< " current working directory is changed to '/'.\n"
|
|
<< " Therefore if this option is used, -d option\n"
|
|
<< " must be specified.\n"
|
|
<< "\n"
|
|
<< " -d, --htdocs=PATH Specify document root. If this option is not\n"
|
|
<< " specified, the document root is the current\n"
|
|
<< " working directory.\n"
|
|
<< "\n"
|
|
<< " -v, --verbose Print debug information such as reception/\n"
|
|
<< " transmission of frames and name/value pairs.\n"
|
|
<< "\n"
|
|
<< " -h, --help Print this help.\n"
|
|
<< std::endl;
|
|
}
|
|
} // namespace
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
Config config;
|
|
while(1) {
|
|
static option long_options[] = {
|
|
{"daemon", no_argument, 0, 'D' },
|
|
{"htdocs", required_argument, 0, 'd' },
|
|
{"help", no_argument, 0, 'h' },
|
|
{"verbose", no_argument, 0, 'v' },
|
|
{0, 0, 0, 0 }
|
|
};
|
|
int option_index = 0;
|
|
int c = getopt_long(argc, argv, "Dd:hv", long_options, &option_index);
|
|
if(c == -1) {
|
|
break;
|
|
}
|
|
switch(c) {
|
|
case 'D':
|
|
config.daemon = true;
|
|
break;
|
|
case 'd':
|
|
config.htdocs = optarg;
|
|
break;
|
|
case 'h':
|
|
print_help(std::cout);
|
|
exit(EXIT_SUCCESS);
|
|
case 'v':
|
|
config.verbose = true;
|
|
break;
|
|
case '?':
|
|
exit(EXIT_FAILURE);
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
if(argc-optind < 3) {
|
|
print_usage(std::cerr);
|
|
std::cerr << "Too few arguments" << std::endl;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if(config.daemon) {
|
|
if(config.htdocs.empty()) {
|
|
print_usage(std::cerr);
|
|
std::cerr << "-d option must be specified when -D is used." << std::endl;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if(daemon(0, 0) == -1) {
|
|
perror("daemon");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
if(config.htdocs.empty()) {
|
|
config.htdocs = "./";
|
|
}
|
|
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();
|
|
reset_timer();
|
|
config.port = strtol(argv[optind++], 0, 10);
|
|
config.private_key_file = argv[optind++];
|
|
config.cert_file = argv[optind++];
|
|
config.on_request_recv_callback = htdocs_on_request_recv_callback;
|
|
ssl_debug = config.verbose;
|
|
|
|
SpdyServer server(&config);
|
|
if(server.listen() == 0) {
|
|
server.run();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
} // namespace spdylay
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
return spdylay::main(argc, argv);
|
|
}
|