From 444398d1bb0fcb4e7e9e5ec77d689a8c78e73d07 Mon Sep 17 00:00:00 2001 From: Yuji Hirose Date: Fri, 27 Nov 2015 14:24:24 -0500 Subject: [PATCH] Added server mode. --- lint/cmdline/httplib.h | 959 +++++++++++++++++++++++++++++++++++ lint/cmdline/peglint.cc | 31 +- lint/cmdline/peglint.vcxproj | 2 + lint/cmdline/server.cc | 353 +++++++++++++ 4 files changed, 1343 insertions(+), 2 deletions(-) create mode 100644 lint/cmdline/httplib.h create mode 100644 lint/cmdline/server.cc diff --git a/lint/cmdline/httplib.h b/lint/cmdline/httplib.h new file mode 100644 index 0000000..ad47a21 --- /dev/null +++ b/lint/cmdline/httplib.h @@ -0,0 +1,959 @@ +// +// httplib.h +// +// Copyright (c) 2012 Yuji Hirose. All rights reserved. +// The Boost Software License 1.0 +// + +#ifndef _CPPHTTPLIB_HTTPSLIB_H_ +#define _CPPHTTPLIB_HTTPSLIB_H_ + +#ifdef _MSC_VER +#define _CRT_SECURE_NO_WARNINGS +#define _CRT_NONSTDC_NO_DEPRECATE + +#ifndef SO_SYNCHRONOUS_NONALERT +#define SO_SYNCHRONOUS_NONALERT 0x20; +#endif +#ifndef SO_OPENTYPE +#define SO_OPENTYPE 0x7008 +#endif +#if (_MSC_VER < 1900) +#define snprintf _snprintf_s +#endif + +#define S_ISREG(m) (((m)&S_IFREG)==S_IFREG) +#define S_ISDIR(m) (((m)&S_IFDIR)==S_IFDIR) + +#include +#include +#include +#include + +#undef min +#undef max + +typedef SOCKET socket_t; +#else +#include +#include +#include +#include +#include +#include +#include + +typedef int socket_t; +#endif + +#include +#include +#include +#include +#include +#include +#include +#include + +namespace httplib +{ + +typedef std::map Map; +typedef std::multimap MultiMap; +typedef std::smatch Match; + +struct Request { + std::string method; + std::string url; + MultiMap headers; + std::string body; + Map params; + Match matches; + + bool has_header(const char* key) const; + std::string get_header_value(const char* key) const; + void set_header(const char* key, const char* val); + + bool has_param(const char* key) const; +}; + +struct Response { + int status; + MultiMap headers; + std::string body; + + bool has_header(const char* key) const; + std::string get_header_value(const char* key) const; + void set_header(const char* key, const char* val); + + void set_redirect(const char* url); + void set_content(const char* s, size_t n, const char* content_type); + void set_content(const std::string& s, const char* content_type); + + Response() : status(-1) {} +}; + +class Server { +public: + typedef std::function Handler; + typedef std::function Logger; + + Server(); + + void get(const char* pattern, Handler handler); + void post(const char* pattern, Handler handler); + + bool set_base_dir(const char* path); + + void set_error_handler(Handler handler); + void set_logger(Logger logger); + + bool listen(const char* host, int port); + void stop(); + +private: + typedef std::vector> Handlers; + + void process_request(FILE* fp_read, FILE* fp_write); + bool read_request_line(FILE* fp, Request& req); + bool routing(Request& req, Response& res); + bool handle_file_request(Request& req, Response& res); + bool dispatch_request(Request& req, Response& res, Handlers& handlers); + + socket_t svr_sock_; + std::string base_dir_; + Handlers get_handlers_; + Handlers post_handlers_; + Handler error_handler_; + Logger logger_; +}; + +class Client { +public: + Client(const char* host, int port); + + std::shared_ptr get(const char* url); + std::shared_ptr head(const char* url); + std::shared_ptr post(const char* url, const std::string& body, const char* content_type); + std::shared_ptr post(const char* url, const Map& params); + + bool send(const Request& req, Response& res); + +private: + bool read_response_line(FILE* fp, Response& res); + + const std::string host_; + const int port_; +}; + +// Implementation +namespace detail { + +template +void split(const char* b, const char* e, char d, Fn fn) +{ + int i = 0; + int beg = 0; + + while (e ? (b + i != e) : (b[i] != '\0')) { + if (b[i] == d) { + fn(&b[beg], &b[i]); + beg = i + 1; + } + i++; + } + + if (i) { + fn(&b[beg], &b[i]); + } +} + +template +inline bool read_and_close_socket(socket_t sock, T callback) +{ + FILE* fp_read; + FILE* fp_write; +#ifdef _MSC_VER + int osfhandle = _open_osfhandle(sock, _O_RDONLY); + fp_read = _fdopen(osfhandle, "rb"); + fp_write = _fdopen(osfhandle, "wb"); +#else + fp_read = fdopen(sock, "rb"); + fp_write = fdopen(sock, "wb"); +#endif + + auto ret = callback(fp_read, fp_write); + +#ifdef _MSC_VER + sock = osfhandle; +#else + fclose(fp_read); + fclose(fp_write); +#endif + + return ret; +} + +inline int shutdown_socket(socket_t sock) +{ +#ifdef _MSC_VER + return shutdown(sock, SD_BOTH); +#else + return shutdown(sock, SHUT_RDWR); +#endif +} + +inline int close_socket(socket_t sock) +{ +#ifdef _MSC_VER + return closesocket(sock); +#else + return close(sock); +#endif +} + +template +socket_t create_socket(const char* host, int port, Fn fn) +{ +#ifdef _MSC_VER + int opt = SO_SYNCHRONOUS_NONALERT; + setsockopt(INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE, (char*)&opt, sizeof(opt)); +#endif + + // Get address info + struct addrinfo hints; + struct addrinfo *result; + + memset(&hints, 0, sizeof(struct addrinfo)); + hints.ai_family = AF_UNSPEC; + hints.ai_socktype = SOCK_STREAM; + hints.ai_flags = 0; + hints.ai_protocol = 0; + + auto service = std::to_string(port); + + if (getaddrinfo(host, service.c_str(), &hints, &result)) { + return -1; + } + + for (auto rp = result; rp; rp = rp->ai_next) { + // Create a socket + auto sock = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol); + if (sock == -1) { + continue; + } + + // Make 'reuse address' option available + int yes = 1; + setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char*)&yes, sizeof(yes)); + + // bind or connect + if (fn(sock, *rp)) { + freeaddrinfo(result); + return sock; + } + + close_socket(sock); + } + + freeaddrinfo(result); + return -1; +} + +inline socket_t create_server_socket(const char* host, int port) +{ + return create_socket(host, port, [](socket_t sock, struct addrinfo& ai) -> socket_t { + if (::bind(sock, ai.ai_addr, ai.ai_addrlen)) { + return false; + } + if (listen(sock, 5)) { // Listen through 5 channels + return false; + } + return true; + }); +} + +inline socket_t create_client_socket(const char* host, int port) +{ + return create_socket(host, port, [](socket_t sock, struct addrinfo& ai) -> socket_t { + if (connect(sock, ai.ai_addr, ai.ai_addrlen)) { + return false; + } + return true; + }); +} + +inline bool is_file(const std::string& s) +{ + struct stat st; + return stat(s.c_str(), &st) >= 0 && S_ISREG(st.st_mode); +} + +inline bool is_dir(const std::string& s) +{ + struct stat st; + return stat(s.c_str(), &st) >= 0 && S_ISDIR(st.st_mode); +} + +inline void read_file(const std::string& path, std::string& out) +{ + std::ifstream fs(path, std::ios_base::binary); + fs.seekg(0, std::ios_base::end); + auto size = fs.tellg(); + fs.seekg(0); + out.resize(size); + fs.read(&out[0], size); +} + +inline std::string get_file_extention(const std::string& path) +{ + std::smatch m; + auto pat = std::regex("\\.([a-zA-Z0-9]+)$"); + auto ret = std::regex_search(path, m, pat); + std::string content_type; + if (ret) { + return m[1].str(); + } + return std::string(); +} + +inline const char* get_content_type_from_file_extention(const std::string& ext) +{ + if (ext == "html") { + return "text/html"; + } + return "text/plain"; +} + +inline const char* status_message(int status) +{ + switch (status) { + case 200: return "OK"; + case 400: return "Bad Request"; + case 404: return "Not Found"; + default: + case 500: return "Internal Server Error"; + } +} + +inline const char* get_header_value(const MultiMap& map, const char* key, const char* def) +{ + auto it = map.find(key); + if (it != map.end()) { + return it->second.c_str(); + } + return def; +} + +inline int get_header_value_int(const MultiMap& map, const char* key, int def) +{ + auto it = map.find(key); + if (it != map.end()) { + return std::stoi(it->second); + } + return def; +} + +inline bool read_headers(FILE* fp, MultiMap& headers) +{ + static std::regex re("(.+?): (.+?)\r\n"); + + const auto BUFSIZ_HEADER = 2048; + char buf[BUFSIZ_HEADER]; + + for (;;) { + if (!fgets(buf, BUFSIZ_HEADER, fp)) { + return false; + } + if (!strcmp(buf, "\r\n")) { + break; + } + std::cmatch m; + if (std::regex_match(buf, m, re)) { + auto key = std::string(m[1]); + auto val = std::string(m[2]); + headers.insert(std::make_pair(key, val)); + } + } + + return true; +} + +template +bool read_content(T& x, FILE* fp) +{ + auto len = get_header_value_int(x.headers, "Content-Length", 0); + if (len) { + x.body.assign(len, 0); + if (!fread(&x.body[0], x.body.size(), 1, fp)) { + return false; + } + } + return true; +} + +template +inline void write_headers(FILE* fp, const T& res) +{ + fprintf(fp, "Connection: close\r\n"); + + for (const auto& x: res.headers) { + if (x.first != "Content-Type" && x.first != "Content-Length") { + fprintf(fp, "%s: %s\r\n", x.first.c_str(), x.second.c_str()); + } + } + + auto t = get_header_value(res.headers, "Content-Type", "text/plain"); + fprintf(fp, "Content-Type: %s\r\n", t); + fprintf(fp, "Content-Length: %ld\r\n", res.body.size()); + fprintf(fp, "\r\n"); +} + +inline void write_response(FILE* fp, const Request& req, const Response& res) +{ + fprintf(fp, "HTTP/1.0 %d %s\r\n", res.status, status_message(res.status)); + + write_headers(fp, res); + + if (!res.body.empty() && req.method != "HEAD") { + fprintf(fp, "%s", res.body.c_str()); + } +} + +inline std::string encode_url(const std::string& s) +{ + std::string result; + + for (auto i = 0; s[i]; i++) { + switch (s[i]) { + case ' ': result += "+"; break; + case '\'': result += "%27"; break; + case ',': result += "%2C"; break; + case ':': result += "%3A"; break; + case ';': result += "%3B"; break; + default: + if (s[i] < 0) { + result += '%'; + char hex[4]; + size_t len = snprintf(hex, sizeof(hex), "%02X", (unsigned char)s[i]); + assert(len == 2); + result.append(hex, len); + } else { + result += s[i]; + } + break; + } + } + + return result; +} + +inline bool is_hex(char c, int& v) +{ + if (0x20 <= c && isdigit(c)) { + v = c - '0'; + return true; + } else if ('A' <= c && c <= 'F') { + v = c - 'A' + 10; + return true; + } else if ('a' <= c && c <= 'f') { + v = c - 'a' + 10; + return true; + } + return false; +} + +inline int from_hex_to_i(const std::string& s, int i, int cnt, int& val) +{ + val = 0; + for (; s[i] && cnt; i++, cnt--) { + int v = 0; + if (is_hex(s[i], v)) { + val = val * 16 + v; + } else { + break; + } + } + return --i; +} + +inline size_t to_utf8(int code, char* buff) +{ + if (code < 0x0080) { + buff[0] = (code & 0x7F); + return 1; + } else if (code < 0x0800) { + buff[0] = (0xC0 | ((code >> 6) & 0x1F)); + buff[1] = (0x80 | (code & 0x3F)); + return 2; + } else if (code < 0xD800) { + buff[0] = (0xE0 | ((code >> 12) & 0xF)); + buff[1] = (0x80 | ((code >> 6) & 0x3F)); + buff[2] = (0x80 | (code & 0x3F)); + return 3; + } else if (code < 0xE000) { // D800 - DFFF is invalid... + return 0; + } else if (code < 0x10000) { + buff[0] = (0xE0 | ((code >> 12) & 0xF)); + buff[1] = (0x80 | ((code >> 6) & 0x3F)); + buff[2] = (0x80 | (code & 0x3F)); + return 3; + } else if (code < 0x110000) { + buff[0] = (0xF0 | ((code >> 18) & 0x7)); + buff[1] = (0x80 | ((code >> 12) & 0x3F)); + buff[2] = (0x80 | ((code >> 6) & 0x3F)); + buff[3] = (0x80 | (code & 0x3F)); + return 4; + } + + // NOTREACHED + return 0; +} + +inline std::string decode_url(const std::string& s) +{ + std::string result; + + for (int i = 0; s[i]; i++) { + if (s[i] == '%') { + i++; + assert(s[i]); + + if (s[i] == '%') { + result += s[i]; + } else if (s[i] == 'u') { + // Unicode + i++; + assert(s[i]); + + int val = 0; + i = from_hex_to_i(s, i, 4, val); + + char buff[4]; + size_t len = to_utf8(val, buff); + + if (len > 0) { + result.append(buff, len); + } + } else { + // HEX + int val = 0; + i = from_hex_to_i(s, i, 2, val); + result += val; + } + } else if (s[i] == '+') { + result += ' '; + } else { + result += s[i]; + } + } + + return result; +} + +inline void write_request(FILE* fp, const Request& req) +{ + auto url = encode_url(req.url); + fprintf(fp, "%s %s HTTP/1.0\r\n", req.method.c_str(), url.c_str()); + + write_headers(fp, req); + + if (!req.body.empty()) { + if (req.has_header("application/x-www-form-urlencoded")) { + fprintf(fp, "%s", encode_url(req.body).c_str()); + } else { + fprintf(fp, "%s", req.body.c_str()); + } + } +} + +inline void parse_query_text(const std::string& s, Map& params) +{ + split(&s[0], &s[s.size()], '&', [&](const char* b, const char* e) { + std::string key; + std::string val; + split(b, e, '=', [&](const char* b, const char* e) { + if (key.empty()) { + key.assign(b, e); + } else { + val.assign(b, e); + } + }); + params[key] = detail::decode_url(val); + }); +} + +#ifdef _MSC_VER +class WSInit { +public: + WSInit::WSInit() { + WSADATA wsaData; + WSAStartup(0x0002, &wsaData); + } + + WSInit::~WSInit() { + WSACleanup(); + } +}; + +static WSInit wsinit_; +#endif + +} // namespace detail + +// Request implementation +inline bool Request::has_header(const char* key) const +{ + return headers.find(key) != headers.end(); +} + +inline std::string Request::get_header_value(const char* key) const +{ + return detail::get_header_value(headers, key, ""); +} + +inline void Request::set_header(const char* key, const char* val) +{ + headers.insert(std::make_pair(key, val)); +} + +inline bool Request::has_param(const char* key) const +{ + return params.find(key) != params.end(); +} + +// Response implementation +inline bool Response::has_header(const char* key) const +{ + return headers.find(key) != headers.end(); +} + +inline std::string Response::get_header_value(const char* key) const +{ + return detail::get_header_value(headers, key, ""); +} + +inline void Response::set_header(const char* key, const char* val) +{ + headers.insert(std::make_pair(key, val)); +} + +inline void Response::set_redirect(const char* url) +{ + set_header("Location", url); + status = 302; +} + +inline void Response::set_content(const char* s, size_t n, const char* content_type) +{ + body.assign(s, n); + set_header("Content-Type", content_type); +} + +inline void Response::set_content(const std::string& s, const char* content_type) +{ + body = s; + set_header("Content-Type", content_type); +} + +// HTTP server implementation +inline Server::Server() + : svr_sock_(-1) +{ +} + +inline void Server::get(const char* pattern, Handler handler) +{ + get_handlers_.push_back(std::make_pair(std::regex(pattern), handler)); +} + +inline void Server::post(const char* pattern, Handler handler) +{ + post_handlers_.push_back(std::make_pair(std::regex(pattern), handler)); +} + +inline bool Server::set_base_dir(const char* path) +{ + if (detail::is_dir(path)) { + base_dir_ = path; + return true; + } + return false; +} + +inline void Server::set_error_handler(Handler handler) +{ + error_handler_ = handler; +} + +inline void Server::set_logger(Logger logger) +{ + logger_ = logger; +} + +inline bool Server::listen(const char* host, int port) +{ + svr_sock_ = detail::create_server_socket(host, port); + if (svr_sock_ == -1) { + return false; + } + + auto ret = true; + + for (;;) { + socket_t sock = accept(svr_sock_, NULL, NULL); + + if (sock == -1) { + if (svr_sock_ != -1) { + detail::close_socket(svr_sock_); + ret = false; + } else { + ; // The server socket was closed by user. + } + break; + } + + // TODO: should be async + detail::read_and_close_socket(sock, [this](FILE* fp_read, FILE* fp_write) { + process_request(fp_read, fp_write); + return true; + }); + } + + return ret; +} + +inline void Server::stop() +{ + detail::shutdown_socket(svr_sock_); + detail::close_socket(svr_sock_); + svr_sock_ = -1; +} + +inline bool Server::read_request_line(FILE* fp, Request& req) +{ + const auto BUFSIZ_REQUESTLINE = 2048; + char buf[BUFSIZ_REQUESTLINE]; + if (!fgets(buf, BUFSIZ_REQUESTLINE, fp)) { + return false; + } + + static std::regex re("(GET|HEAD|POST) ([^?]+)(?:\\?(.+?))? HTTP/1\\.[01]\r\n"); + + std::cmatch m; + if (std::regex_match(buf, m, re)) { + req.method = std::string(m[1]); + req.url = detail::decode_url(m[2]); + + // Parse query text + auto len = std::distance(m[3].first, m[3].second); + if (len > 0) { + detail::parse_query_text(m[3], req.params); + } + + return true; + } + + return false; +} + +inline bool Server::handle_file_request(Request& req, Response& res) +{ + if (!base_dir_.empty()) { + std::string path = base_dir_ + req.url; + + if (!path.empty() && path.back() == '/') { + path += "index.html"; + } + + if (detail::is_file(path)) { + detail::read_file(path, res.body); + res.set_header("Content-Type", + detail::get_content_type_from_file_extention( + detail::get_file_extention(path))); + res.status = 200; + return true; + } + } + + return false; +} + +inline bool Server::routing(Request& req, Response& res) +{ + if (req.method == "GET" && handle_file_request(req, res)) { + return true; + } + + if (req.method == "GET" || req.method == "HEAD") { + return dispatch_request(req, res, get_handlers_); + } else if (req.method == "POST") { + return dispatch_request(req, res, post_handlers_); + } + return false; +} + +inline bool Server::dispatch_request(Request& req, Response& res, Handlers& handlers) +{ + for (const auto& x: handlers) { + const auto& pattern = x.first; + const auto& handler = x.second; + + if (std::regex_match(req.url, req.matches, pattern)) { + handler(req, res); + return true; + } + } + return false; +} + +inline void Server::process_request(FILE* fp_read, FILE* fp_write) +{ + Request req; + Response res; + + if (!read_request_line(fp_read, req) || + !detail::read_headers(fp_read, req.headers)) { + return; + } + + if (req.method == "POST") { + if (!detail::read_content(req, fp_read)) { + return; + } + static std::string type = "application/x-www-form-urlencoded"; + if (!req.get_header_value("Content-Type").compare(0, type.size(), type)) { + detail::parse_query_text(req.body, req.params); + } + } + + if (routing(req, res)) { + if (res.status == -1) { + res.status = 200; + } + } else { + res.status = 404; + } + assert(res.status != -1); + + if (400 <= res.status && error_handler_) { + error_handler_(req, res); + } + + detail::write_response(fp_write, req, res); + fflush(fp_write); + + if (logger_) { + logger_(req, res); + } +} + +// HTTP client implementation +inline Client::Client(const char* host, int port) + : host_(host) + , port_(port) +{ +} + +inline bool Client::read_response_line(FILE* fp, Response& res) +{ + const auto BUFSIZ_RESPONSELINE = 2048; + char buf[BUFSIZ_RESPONSELINE]; + if (!fgets(buf, BUFSIZ_RESPONSELINE, fp)) { + return false; + } + + const static std::regex re("HTTP/1\\.[01] (\\d+?) .+\r\n"); + + std::cmatch m; + if (std::regex_match(buf, m, re)) { + res.status = std::stoi(std::string(m[1])); + } + + return true; +} + +inline bool Client::send(const Request& req, Response& res) +{ + auto sock = detail::create_client_socket(host_.c_str(), port_); + if (sock == -1) { + return false; + } + + return detail::read_and_close_socket(sock, [&](FILE* fp_read, FILE* fp_write) { + // Send request + detail::write_request(fp_write, req); + fflush(fp_write); + + // Receive response + if (!read_response_line(fp_read, res) || + !detail::read_headers(fp_read, res.headers)) { + return false; + } + if (req.method != "HEAD") { + if (!detail::read_content(res, fp_read)) { + return false; + } + } + + return true; + }); +} + +inline std::shared_ptr Client::get(const char* url) +{ + Request req; + req.method = "GET"; + req.url = url; + + auto res = std::make_shared(); + + return send(req, *res) ? res : nullptr; +} + +inline std::shared_ptr Client::head(const char* url) +{ + Request req; + req.method = "HEAD"; + req.url = url; + + auto res = std::make_shared(); + + return send(req, *res) ? res : nullptr; +} + +inline std::shared_ptr Client::post( + const char* url, const std::string& body, const char* content_type) +{ + Request req; + req.method = "POST"; + req.url = url; + req.set_header("Content-Type", content_type); + req.body = body; + + auto res = std::make_shared(); + + return send(req, *res) ? res : nullptr; +} + +inline std::shared_ptr Client::post( + const char* url, const Map& params) +{ + std::string query; + for (auto it = params.begin(); it != params.end(); ++it) { + if (it != params.begin()) { + query += "&"; + } + query += it->first; + query += "="; + query += it->second; + } + + return post(url, query, "application/x-www-form-urlencoded"); +} + +} // namespace httplib + +#endif + +// vim: et ts=4 sw=4 cin cino={1s ff=unix diff --git a/lint/cmdline/peglint.cc b/lint/cmdline/peglint.cc index 36265dc..1ff35f6 100644 --- a/lint/cmdline/peglint.cc +++ b/lint/cmdline/peglint.cc @@ -10,6 +10,8 @@ using namespace std; +int run_server(int port, const vector& syntax, const vector& source); + bool read_file(const char* path, vector& buff) { ifstream ifs(path, ios::in | ios::binary); @@ -29,6 +31,8 @@ int main(int argc, const char** argv) auto opt_ast = false; auto opt_optimize_ast_nodes = false; auto opt_help = false; + auto opt_server = false; + int port = 1234; vector path_list; auto argi = 1; @@ -40,16 +44,39 @@ int main(int argc, const char** argv) opt_ast = true; } else if (string("--optimize_ast_nodes") == arg || string("--opt") == arg) { opt_optimize_ast_nodes = true; + } else if (string("--server") == arg) { + opt_server = true; + if (argi < argc) { + port = std::stoi(argv[argi++]); + } } else { path_list.push_back(arg); } } - if (path_list.empty() || opt_help) { - cerr << "usage: peglint [--ast] [--optimize_ast_nodes|--opt] [grammar file path] [source file path]" << endl; + if ((path_list.empty() && !opt_server) || opt_help) { + cerr << "usage: peglint [--ast] [--optimize_ast_nodes|--opt] [--server [PORT=1234]] [grammar file path] [source file path]" << endl; return 1; } + // Sever mode + if (opt_server) { + vector syntax; + vector source; + + if (path_list.size() >= 1 && !read_file(path_list[0], syntax)) { + cerr << "can't open the grammar file." << endl; + return -1; + } + + if (path_list.size() >= 2 && !read_file(path_list[1], source)) { + cerr << "can't open the code file." << endl; + return -1; + } + + return run_server(port, syntax, source); + } + // Check PEG grammar auto syntax_path = path_list[0]; diff --git a/lint/cmdline/peglint.vcxproj b/lint/cmdline/peglint.vcxproj index 54ab492..1fd4814 100644 --- a/lint/cmdline/peglint.vcxproj +++ b/lint/cmdline/peglint.vcxproj @@ -12,9 +12,11 @@ + + {F85B641A-7538-4809-8175-C528FF632CF6} diff --git a/lint/cmdline/server.cc b/lint/cmdline/server.cc new file mode 100644 index 0000000..4b31bb6 --- /dev/null +++ b/lint/cmdline/server.cc @@ -0,0 +1,353 @@ + +#include "httplib.h" +#include "peglib.h" +#include +#include + +using namespace httplib; +using namespace std; + +string indexHTML = R"( + + + +PEG Playground + + + + +
+
+
    +
  • Grammar:
  • +
  • Valid
  • +
+
{{syntax}}
+
+
+
+
    +
  • Code:
  • +
  • Valid
  • +
+
{{source}}
+

+        

+        
+
+
+ + + + + + +)"; + +function makeJSONFormatter(string& json) +{ + auto init = make_shared(true); + + return [&json, init](size_t ln, size_t col, const string& msg) mutable { + if (!init) { + json += ","; + } + json += "{"; + json += R"("ln":)" + to_string(ln) + ","; + json += R"("col":)" + to_string(col) + ","; + json += R"("msg":")" + msg + R"(")"; + json += "}"; + *init = false; + }; +} + +bool parse_grammar(const string& text, peg::parser& peg, string& json) +{ + peg.log = makeJSONFormatter(json); + json += "["; + auto ret = peg.load_grammar(text.data(), text.size()); + json += "]"; + return ret; +} + +bool parse_code(const string& text, peg::parser& peg, string& json, shared_ptr& ast) +{ + peg.enable_ast(); + peg.log = makeJSONFormatter(json); + json += "["; + auto ret = peg.parse_n(text.data(), text.size(), ast); + json += "]"; + return ret; +} + +template +void dump_ast(const shared_ptr& ptr, string& json, int level = 0) +{ + const auto& ast = *ptr; + for (auto i = 0; i < level; i++) { + json += " "; + } + string name; + if (ast.name == ast.original_name) { + name = ast.name; + } else { + name = ast.original_name + " (" + ast.name + ")"; + } + if (ast.is_token) { + json += "- " + name + "(" + ast.token + ")\\n"; + } else { + json += "+ " + name +"\\n"; + } + for (auto node : ast.nodes) { + dump_ast(node, json, level + 1); + } +} + +std::string replace_text(std::string &s, const std::string &toReplace, const std::string &replaceWith) +{ + return(s.replace(s.find(toReplace), toReplace.length(), replaceWith)); +} + +int run_server(int port, const vector& syntax, const vector& source) +{ + Server svr; + + svr.get("/", [&](const Request& req, Response& res) { + replace_text(indexHTML, "{{syntax}}", string(syntax.data(), syntax.size())); + replace_text(indexHTML, "{{source}}", string(source.data(), source.size())); + + res.set_content(indexHTML, "text/html"); + }); + + svr.post("/parse", [](const Request& req, Response& res) { + const auto& grammarText = req.params.at("grammar"); + + string grammarResult; + string codeResult; + string astResult; + string astResultOptimized; + + peg::parser peg; + auto ret = parse_grammar(grammarText, peg, grammarResult); + + if (ret && peg) { + const auto& codeText = req.params.at("code"); + shared_ptr ast; + if (parse_code(codeText, peg, codeResult, ast)) { + dump_ast(ast, astResult); + dump_ast(peg::AstOptimizer(true).optimize(ast), astResultOptimized); + } + } + + string json; + json += "{"; + json += "\"grammar\":" + grammarResult; + if (!codeResult.empty()) { + json += ",\"code\":" + codeResult; + json += ",\"ast\":\"" + astResult + "\""; + json += ",\"astOptimized\":\"" + astResultOptimized + "\""; + } + json += "}"; + + res.set_content(json, "application/json"); + }); + + svr.set_error_handler([](const Request& req, Response& res) { + const char* fmt = "

Error Status: %d

"; + char buf[BUFSIZ]; + snprintf(buf, sizeof(buf), fmt, res.status); + res.set_content(buf, "text/html"); + }); + + cerr << "Server running at http://localhost:" << port << "/" << endl; + svr.listen("localhost", port); + + return 0; +} + +// vim: et ts=4 sw=4 cin cino={1s ff=unix