mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-15 09:20:52 +08:00
188 lines
4.0 KiB
C++
188 lines
4.0 KiB
C++
#include "stdafx.h"
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include "http_servlet.h"
|
|
|
|
#define STACK_SIZE 128000
|
|
|
|
static int __rw_timeout = 0;
|
|
static acl::string __ssl_crt("./ssl_crt.pem");
|
|
static acl::string __ssl_key("./ssl_key.pem");
|
|
static acl::sslbase_conf* __ssl_conf;
|
|
|
|
static void http_server(ACL_FIBER *, void *ctx)
|
|
{
|
|
acl::socket_stream *conn = (acl::socket_stream *) ctx;
|
|
|
|
printf("start one http_server\r\n");
|
|
|
|
acl::sslbase_io* ssl = __ssl_conf->create(false);
|
|
|
|
if (conn->setup_hook(ssl) == ssl) {
|
|
printf("setup_hook error\r\n");
|
|
ssl->destroy();
|
|
delete conn;
|
|
return;
|
|
}
|
|
|
|
#if 0
|
|
if (!ssl->handshake()) {
|
|
printf("ssl handshake error\r\n");
|
|
ssl->destroy();
|
|
delete conn;
|
|
return;
|
|
}
|
|
if (!ssl->handshake_ok()) {
|
|
printf("ssl handshake error\r\n");
|
|
ssl->destroy();
|
|
delete conn;
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
printf("ssl handshake_ok\r\n");
|
|
|
|
acl::memcache_session session("127.0.0.1:11211");
|
|
http_servlet servlet(conn, &session);
|
|
servlet.setLocalCharset("gb2312");
|
|
|
|
while (true) {
|
|
if (!servlet.doRun()) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
printf("close one connection: %d, %s\r\n", conn->sock_handle(),
|
|
acl::last_serror());
|
|
delete conn;
|
|
}
|
|
|
|
static void ssl_init(acl::sslbase_conf& conf, const acl::string& crt,
|
|
const acl::string& key)
|
|
{
|
|
conf.enable_cache(1);
|
|
|
|
if (!conf.add_cert(crt)) {
|
|
printf("load %s error\r\n", crt.c_str());
|
|
exit (1);
|
|
}
|
|
|
|
if (!conf.set_key(key)) {
|
|
printf("set_key %s error\r\n", key.c_str());
|
|
exit (1);
|
|
}
|
|
|
|
printf(">>> ssl_init ok, ssl_crt: %s, ssl_key: %s\r\n",
|
|
crt.c_str(), key.c_str());
|
|
}
|
|
|
|
static void fiber_accept(ACL_FIBER *, void *ctx)
|
|
{
|
|
const char* addr = (const char* ) ctx;
|
|
acl::server_socket server;
|
|
|
|
ssl_init(*__ssl_conf, __ssl_crt, __ssl_key);
|
|
|
|
if (!server.open(addr)) {
|
|
printf("open %s error\r\n", addr);
|
|
exit (1);
|
|
} else {
|
|
printf(">>> listen %s ok\r\n", addr);
|
|
}
|
|
|
|
while (true) {
|
|
acl::socket_stream* client = server.accept();
|
|
if (client == NULL) {
|
|
printf("accept failed: %s\r\n", acl::last_serror());
|
|
break;
|
|
}
|
|
|
|
client->set_rw_timeout(__rw_timeout);
|
|
printf("accept one: %d\r\n", client->sock_handle());
|
|
acl_fiber_create(http_server, client, STACK_SIZE);
|
|
}
|
|
|
|
exit (0);
|
|
}
|
|
|
|
static void usage(const char* procname)
|
|
{
|
|
printf("usage: %s -h [help]\r\n"
|
|
" -l ssl_lib_path\r\n"
|
|
" -s listen_addr\r\n"
|
|
" -r rw_timeout\r\n"
|
|
" -c ssl_crt.pem\r\n"
|
|
" -k ssl_key.pem\r\n", procname);
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
acl::string addr(":9001");
|
|
#ifdef __APPLE__
|
|
acl::string libpath("../libmbedtls_all.dylib");
|
|
#else
|
|
acl::string libpath("../libmbedtls_all.so");
|
|
#endif
|
|
int ch;
|
|
|
|
acl::acl_cpp_init();
|
|
acl::log::stdout_open(true);
|
|
|
|
while ((ch = getopt(argc, argv, "hs:r:c:k:l:")) > 0) {
|
|
switch (ch) {
|
|
case 'h':
|
|
usage(argv[0]);
|
|
return 0;
|
|
case 's':
|
|
addr = optarg;
|
|
break;
|
|
case 'r':
|
|
__rw_timeout = atoi(optarg);
|
|
break;
|
|
case 'c':
|
|
__ssl_crt = optarg;
|
|
break;
|
|
case 'k':
|
|
__ssl_key = optarg;
|
|
break;
|
|
case 'l':
|
|
libpath = optarg;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (libpath.find("mbedtls") != NULL) {
|
|
const std::vector<acl::string>& libs = libpath.split2("; \t");
|
|
if (libs.size() != 3) {
|
|
printf("invalid libpath=%s\r\n", libpath.c_str());
|
|
return 1;
|
|
}
|
|
|
|
acl::mbedtls_conf::set_libpath(libs[0], libs[1], libs[2]);
|
|
if (acl::mbedtls_conf::load()) {
|
|
__ssl_conf = new acl::mbedtls_conf(true);
|
|
} else {
|
|
printf("load %s error\r\n", libpath.c_str());
|
|
return 1;
|
|
}
|
|
} else if (libpath.find("polarlssl") != NULL) {
|
|
acl::polarssl_conf::set_libpath(libpath);
|
|
if (acl::polarssl_conf::load()) {
|
|
__ssl_conf = new acl::polarssl_conf;
|
|
} else {
|
|
printf("load %s error\r\n", libpath.c_str());
|
|
return 1;
|
|
}
|
|
} else {
|
|
printf("invalid ssl lib=%s\r\n", libpath.c_str());
|
|
return 1;
|
|
}
|
|
|
|
acl_fiber_create(fiber_accept, addr.c_str(), STACK_SIZE);
|
|
acl_fiber_schedule();
|
|
delete __ssl_conf;
|
|
return 0;
|
|
}
|