mirror of
https://github.com/pineappleEA/pineapple-src.git
synced 2024-11-29 15:58:24 -05:00
136 lines
3.1 KiB
C++
Executable File
136 lines
3.1 KiB
C++
Executable File
//
|
|
// simplesvr.cc
|
|
//
|
|
// Copyright (c) 2019 Yuji Hirose. All rights reserved.
|
|
// MIT License
|
|
//
|
|
|
|
#include <cstdio>
|
|
#include <httplib.h>
|
|
#include <iostream>
|
|
|
|
#define SERVER_CERT_FILE "./cert.pem"
|
|
#define SERVER_PRIVATE_KEY_FILE "./key.pem"
|
|
|
|
using namespace httplib;
|
|
using namespace std;
|
|
|
|
string dump_headers(const Headers &headers) {
|
|
string s;
|
|
char buf[BUFSIZ];
|
|
|
|
for (const auto &x : headers) {
|
|
snprintf(buf, sizeof(buf), "%s: %s\n", x.first.c_str(), x.second.c_str());
|
|
s += buf;
|
|
}
|
|
|
|
return s;
|
|
}
|
|
|
|
string dump_multipart_files(const MultipartFormDataMap &files) {
|
|
string s;
|
|
char buf[BUFSIZ];
|
|
|
|
s += "--------------------------------\n";
|
|
|
|
for (const auto &x : files) {
|
|
const auto &name = x.first;
|
|
const auto &file = x.second;
|
|
|
|
snprintf(buf, sizeof(buf), "name: %s\n", name.c_str());
|
|
s += buf;
|
|
|
|
snprintf(buf, sizeof(buf), "filename: %s\n", file.filename.c_str());
|
|
s += buf;
|
|
|
|
snprintf(buf, sizeof(buf), "content type: %s\n", file.content_type.c_str());
|
|
s += buf;
|
|
|
|
snprintf(buf, sizeof(buf), "text length: %zu\n", file.content.size());
|
|
s += buf;
|
|
|
|
s += "----------------\n";
|
|
}
|
|
|
|
return s;
|
|
}
|
|
|
|
string log(const Request &req, const Response &res) {
|
|
string s;
|
|
char buf[BUFSIZ];
|
|
|
|
s += "================================\n";
|
|
|
|
snprintf(buf, sizeof(buf), "%s %s %s", req.method.c_str(),
|
|
req.version.c_str(), req.path.c_str());
|
|
s += buf;
|
|
|
|
string query;
|
|
for (auto it = req.params.begin(); it != req.params.end(); ++it) {
|
|
const auto &x = *it;
|
|
snprintf(buf, sizeof(buf), "%c%s=%s",
|
|
(it == req.params.begin()) ? '?' : '&', x.first.c_str(),
|
|
x.second.c_str());
|
|
query += buf;
|
|
}
|
|
snprintf(buf, sizeof(buf), "%s\n", query.c_str());
|
|
s += buf;
|
|
|
|
s += dump_headers(req.headers);
|
|
s += dump_multipart_files(req.files);
|
|
|
|
s += "--------------------------------\n";
|
|
|
|
snprintf(buf, sizeof(buf), "%d\n", res.status);
|
|
s += buf;
|
|
s += dump_headers(res.headers);
|
|
|
|
return s;
|
|
}
|
|
|
|
int main(int argc, const char **argv) {
|
|
if (argc > 1 && string("--help") == argv[1]) {
|
|
cout << "usage: simplesvr [PORT] [DIR]" << endl;
|
|
return 1;
|
|
}
|
|
|
|
#ifdef CPPHTTPLIB_OPENSSL_SUPPORT
|
|
SSLServer svr(SERVER_CERT_FILE, SERVER_PRIVATE_KEY_FILE);
|
|
#else
|
|
Server svr;
|
|
#endif
|
|
|
|
svr.Post("/multipart", [](const Request &req, Response &res) {
|
|
auto body = dump_headers(req.headers) + dump_multipart_files(req.files);
|
|
|
|
res.set_content(body, "text/plain");
|
|
});
|
|
|
|
svr.set_error_handler([](const Request & /*req*/, Response &res) {
|
|
const char *fmt = "<p>Error Status: <span style='color:red;'>%d</span></p>";
|
|
char buf[BUFSIZ];
|
|
snprintf(buf, sizeof(buf), fmt, res.status);
|
|
res.set_content(buf, "text/html");
|
|
});
|
|
|
|
svr.set_logger(
|
|
[](const Request &req, const Response &res) { cout << log(req, res); });
|
|
|
|
auto port = 8080;
|
|
if (argc > 1) { port = atoi(argv[1]); }
|
|
|
|
auto base_dir = "./";
|
|
if (argc > 2) { base_dir = argv[2]; }
|
|
|
|
if (!svr.set_mount_point("/", base_dir)) {
|
|
cout << "The specified base directory doesn't exist...";
|
|
return 1;
|
|
}
|
|
|
|
cout << "The server started at port " << port << "..." << endl;
|
|
|
|
svr.listen("localhost", port);
|
|
|
|
return 0;
|
|
}
|