mirror of
https://github.com/sogou/workflow.git
synced 2026-10-19 05:31:19 +08:00
Xie Han
125 lines
2.9 KiB
C++
125 lines
2.9 KiB
C++
#include <signal.h>
|
|
#include <fcntl.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <utility>
|
|
#include <string>
|
|
#include "workflow/HttpMessage.h"
|
|
#include "workflow/WFHttpServer.h"
|
|
#include "workflow/WFTaskFactory.h"
|
|
#include "workflow/Workflow.h"
|
|
|
|
#ifndef _WIN32
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
#include <io.h>
|
|
# define open _open
|
|
# define O_RDONLY _O_RDONLY
|
|
# define close _close
|
|
# define lseek _lseek
|
|
#endif
|
|
|
|
using namespace protocol;
|
|
|
|
void pread_callback(WFFileIOTask *task)
|
|
{
|
|
FileIOArgs *args = task->get_args();
|
|
long ret = task->get_retval();
|
|
HttpResponse *resp = (HttpResponse *)task->user_data;
|
|
|
|
close(args->fd);
|
|
if (task->get_state() != WFT_STATE_SUCCESS || ret < 0)
|
|
{
|
|
resp->set_status_code("503");
|
|
resp->append_output_body("<html>503 Internal Server Error.</html>");
|
|
}
|
|
else /* Use '_nocopy' carefully. */
|
|
resp->append_output_body_nocopy(args->buf, ret);
|
|
}
|
|
|
|
void process(WFHttpTask *server_task, const char *root)
|
|
{
|
|
HttpRequest *req = server_task->get_req();
|
|
HttpResponse *resp = server_task->get_resp();
|
|
const char *uri = req->get_request_uri();
|
|
const char *p = uri;
|
|
|
|
printf("Request-URI: %s\n", uri);
|
|
while (*p && *p != '?')
|
|
p++;
|
|
|
|
std::string abs_path(uri, p - uri);
|
|
abs_path = root + abs_path;
|
|
if (abs_path.back() == '/')
|
|
abs_path += "index.html";
|
|
|
|
resp->add_header_pair("Server", "Sogou C++ Workflow Server");
|
|
// ------ we will support fileio with iocp as soon as possible
|
|
// int fd = open(abs_path.c_str(), O_RDONLY);
|
|
// if (fd >= 0)
|
|
// {
|
|
// size_t size = lseek(fd, 0, SEEK_END);
|
|
// void *buf = malloc(size); /* As an example, assert(buf != NULL); */
|
|
// WFFileIOTask *pread_task;
|
|
//
|
|
// pread_task = WFTaskFactory::create_pread_task(fd, buf, size, 0,
|
|
// pread_callback);
|
|
// /* To implement a more complicated server, please use series' context
|
|
// * instead of tasks' user_data to pass/store internal data. */
|
|
// pread_task->user_data = resp; /* pass resp pointer to pread task. */
|
|
// server_task->user_data = buf; /* to free() in callback() */
|
|
// server_task->set_callback([](WFHttpTask *t){ free(t->user_data); });
|
|
// series_of(server_task)->push_back(pread_task);
|
|
// }
|
|
// else
|
|
{
|
|
resp->set_status_code("404");
|
|
resp->append_output_body("<html>404 Not Found.</html>");
|
|
}
|
|
}
|
|
|
|
void sig_handler(int signo) { }
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
if (argc != 2 && argc != 3 && argc != 5)
|
|
{
|
|
fprintf(stderr, "%s <port> [root path] [cert file] [key file]\n",
|
|
argv[0]);
|
|
exit(1);
|
|
}
|
|
|
|
signal(SIGINT, sig_handler);
|
|
|
|
unsigned short port = atoi(argv[1]);
|
|
const char *root = (argc >= 3 ? argv[2] : ".");
|
|
auto&& proc = std::bind(process, std::placeholders::_1, root);
|
|
WFHttpServer server(proc);
|
|
int ret;
|
|
|
|
if (argc == 5)
|
|
ret = server.start(port, argv[3], argv[4]); /* https server */
|
|
else
|
|
ret = server.start(port);
|
|
|
|
if (ret == 0)
|
|
{
|
|
#ifndef _WIN32
|
|
pause();
|
|
#else
|
|
getchar();
|
|
#endif
|
|
server.stop();
|
|
}
|
|
else
|
|
{
|
|
perror("start server");
|
|
exit(1);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|