mirror of
https://github.com/yhirose/cpp-httplib
synced 2024-11-21 06:26:02 -07:00
Added httpsvrkit.h and sample.
This commit is contained in:
yhirose
parent
c568deaba1
commit
dfe95a669d
3 changed files with 330 additions and 0 deletions
17
example/Makefile
Normal file
17
example/Makefile
Normal file
|
|
@ -0,0 +1,17 @@
|
|||
|
||||
USE_CLANG = 1
|
||||
|
||||
ifdef USE_CLANG
|
||||
CC = clang++
|
||||
CFLAGS = -std=c++0x -stdlib=libc++
|
||||
else
|
||||
CC = g++
|
||||
CFLAGS = -std=c++11
|
||||
endif
|
||||
|
||||
sample : sample.cc ../httpsvrkit.h
|
||||
$(CC) -o sample $(CFLAGS) -I.. sample.cc
|
||||
|
||||
.PHONY : test
|
||||
test: sample
|
||||
./sample
|
||||
36
example/sample.cc
Normal file
36
example/sample.cc
Normal file
|
|
@ -0,0 +1,36 @@
|
|||
//
|
||||
// sample.cc
|
||||
//
|
||||
// Copyright (c) 2012 Yuji Hirose. All rights reserved.
|
||||
// The Boost Software License 1.0
|
||||
//
|
||||
|
||||
#include <httpsvrkit.h>
|
||||
#include <cstdio>
|
||||
|
||||
int main(void)
|
||||
{
|
||||
using namespace httpsvrkit;
|
||||
|
||||
Server svr;
|
||||
|
||||
svr.post("/", [](const Request& /*req*/, Response& res) {
|
||||
res.body_ = "<html><head></head><body><ul></ul></body></html>";
|
||||
});
|
||||
|
||||
svr.post("/item", [](const Request& req, Response& res) {
|
||||
res.body_ = req.pattern_;
|
||||
});
|
||||
|
||||
svr.get("/item/:name", [](const Request& req, Response& res) {
|
||||
try {
|
||||
res.body_ = req.params_.at("name");
|
||||
} catch (...) {
|
||||
// Error...
|
||||
}
|
||||
});
|
||||
|
||||
svr.run("0.0.0.0", 1234);
|
||||
}
|
||||
|
||||
// vim: et ts=4 sw=4 cin cino={1s ff=unix
|
||||
277
httpsvrkit.h
Normal file
277
httpsvrkit.h
Normal file
|
|
@ -0,0 +1,277 @@
|
|||
//
|
||||
// httpsvrkit.h
|
||||
//
|
||||
// Copyright (c) 2012 Yuji Hirose. All rights reserved.
|
||||
// The Boost Software License 1.0
|
||||
//
|
||||
|
||||
#include <functional>
|
||||
#include <map>
|
||||
#include <regex>
|
||||
#include <string>
|
||||
#include <assert.h>
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <winsock2.h>
|
||||
#include <ws2tcpip.h>
|
||||
|
||||
typedef SOCKET socket_t;
|
||||
|
||||
int inet_aton(const char* strptr, struct in_addr* addrptr)
|
||||
{
|
||||
unsigned long addr = inet_addr(strptr);
|
||||
if (addr == ULONG_MAX)
|
||||
return 0;
|
||||
addrptr->s_addr = addr;
|
||||
return 1;
|
||||
}
|
||||
#else
|
||||
#include <pthread.h>
|
||||
#include <unistd.h>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
|
||||
typedef int socket_t;
|
||||
#endif
|
||||
|
||||
namespace httpsvrkit
|
||||
{
|
||||
|
||||
// HTTP request
|
||||
class Request {
|
||||
public:
|
||||
std::map<std::string, std::string> headers_;
|
||||
std::string body_;
|
||||
|
||||
std::string pattern_;
|
||||
std::map<std::string, std::string> params_;
|
||||
};
|
||||
|
||||
// HTTP response
|
||||
class Response {
|
||||
public:
|
||||
std::multimap<std::string, std::string> headers_;
|
||||
std::string body_;
|
||||
};
|
||||
|
||||
// HTTP server
|
||||
class Server {
|
||||
public:
|
||||
typedef std::function<void (const Request&, Response& res)> Handler;
|
||||
|
||||
Server();
|
||||
~Server();
|
||||
|
||||
void get(const std::string& pattern, Handler handler);
|
||||
void post(const std::string& pattern, Handler handler);
|
||||
|
||||
bool run(const std::string& ipaddr, int port);
|
||||
void stop();
|
||||
|
||||
private:
|
||||
void process_request(int fd);
|
||||
|
||||
const size_t BUFSIZ_REQUESTLINE = 2048;
|
||||
|
||||
socket_t sock_;
|
||||
std::multimap<std::string, Handler> handlers_;
|
||||
};
|
||||
|
||||
// Implementation
|
||||
|
||||
template <typename Fn>
|
||||
void fdopen_b(int fd, const char* md, Fn fn)
|
||||
{
|
||||
#ifdef _WIN32
|
||||
int osfhandle = _open_osfhandle(fd, _O_RDONLY);
|
||||
FILE* fp = fdopen(osfhandle, md);
|
||||
#else
|
||||
FILE* fp = fdopen(fd, md);
|
||||
#endif
|
||||
|
||||
if (fp) {
|
||||
fn(fp);
|
||||
fclose(fp);
|
||||
|
||||
#ifdef _WIN32
|
||||
close(osfhandle);
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
inline socket_t create_socket(const std::string& ipaddr, int port)
|
||||
{
|
||||
// Create a server socket
|
||||
socket_t sock = socket(AF_INET, SOCK_STREAM, 0);
|
||||
if (sock == -1) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Make 'reuse address' option available
|
||||
int yes = 1;
|
||||
setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char*)&yes, sizeof(yes));
|
||||
|
||||
// Bind the socket to the given address
|
||||
struct sockaddr_in addr;
|
||||
addr.sin_family = AF_INET;
|
||||
addr.sin_port = htons((uint16_t)port);
|
||||
|
||||
if (inet_aton(ipaddr.c_str(), &addr.sin_addr) <= 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (::bind(sock, (struct sockaddr*)&addr, sizeof(addr)) != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Listen through 5 channels
|
||||
if (listen(sock, 5) != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
return sock;
|
||||
}
|
||||
|
||||
inline void close_socket(socket_t sock)
|
||||
{
|
||||
#ifdef _WIN32
|
||||
closesocket(sock);
|
||||
#else
|
||||
shutdown(sock, SHUT_RDWR);
|
||||
close(sock);
|
||||
#endif
|
||||
}
|
||||
|
||||
inline Server::Server()
|
||||
: sock_(-1)
|
||||
{
|
||||
#ifdef _WIN32
|
||||
WSADATA wsaData;
|
||||
WSAStartup(0x0002, &wsaData);
|
||||
#endif
|
||||
}
|
||||
|
||||
inline Server::~Server()
|
||||
{
|
||||
#ifdef _WIN32
|
||||
WSACleanup();
|
||||
#endif
|
||||
}
|
||||
|
||||
inline void Server::get(const std::string& pattern, Handler handler)
|
||||
{
|
||||
handlers_.insert(std::make_pair(pattern, handler));
|
||||
}
|
||||
|
||||
inline void Server::post(const std::string& pattern, Handler handler)
|
||||
{
|
||||
handlers_.insert(std::make_pair(pattern, handler));
|
||||
}
|
||||
|
||||
inline bool Server::run(const std::string& ipaddr, int port)
|
||||
{
|
||||
sock_ = create_socket(ipaddr, port);
|
||||
if (sock_ == -1) {
|
||||
return false;
|
||||
}
|
||||
|
||||
for (;;) {
|
||||
socket_t fd = accept(sock_, NULL, NULL);
|
||||
if (fd == -1) {
|
||||
// The server socket was closed by user.
|
||||
if (sock_ == -1) {
|
||||
return true;
|
||||
}
|
||||
|
||||
close_socket(sock_);
|
||||
return false;
|
||||
}
|
||||
|
||||
process_request(fd);
|
||||
close(fd);
|
||||
}
|
||||
|
||||
// NOTREACHED
|
||||
}
|
||||
|
||||
inline void Server::stop()
|
||||
{
|
||||
close_socket(sock_);
|
||||
sock_ = -1;
|
||||
}
|
||||
|
||||
inline bool parse_request_line(const char* s, std::string& cmd, std::string& url)
|
||||
{
|
||||
std::regex re("(GET|POST) (.+) HTTP/1\\.1\r\n");
|
||||
std::cmatch m;
|
||||
if (std::regex_match(s, m, re)) {
|
||||
cmd = std::string(m[1]);
|
||||
url = std::string(m[2]);
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
inline void write_plain_text(int fd, const char* s)
|
||||
{
|
||||
fdopen_b(fd, "w", [=](FILE* fp) {
|
||||
fprintf(fp, "HTTP/1.0 200 OK\r\n");
|
||||
fprintf(fp, "Content-type: text/plain\r\n");
|
||||
fprintf(fp, "Connection: close\r\n");
|
||||
fprintf(fp, "\r\n");
|
||||
fprintf(fp, "%s", s);
|
||||
});
|
||||
}
|
||||
|
||||
inline void write_error(int fd, int code)
|
||||
{
|
||||
const char* msg = NULL;
|
||||
|
||||
switch (code) {
|
||||
case 400:
|
||||
msg = "Bad Request";
|
||||
break;
|
||||
case 404:
|
||||
msg = "Not Found";
|
||||
break;
|
||||
default:
|
||||
code = 500;
|
||||
msg = "Internal Server Error";
|
||||
break;
|
||||
}
|
||||
|
||||
assert(msg);
|
||||
|
||||
fdopen_b(fd, "w", [=](FILE* fp) {
|
||||
fprintf(fp, "HTTP/1.0 %d %s\r\n", code, msg);
|
||||
fprintf(fp, "Content-type: text/plain\r\n");
|
||||
fprintf(fp, "Connection: close\r\n");
|
||||
fprintf(fp, "\r\n");
|
||||
fprintf(fp, "Status: %d\r\n", code);
|
||||
});
|
||||
}
|
||||
|
||||
inline void Server::process_request(int fd)
|
||||
{
|
||||
fdopen_b(fd, "r", [=](FILE* fp) {
|
||||
// Parse request line
|
||||
char request_line[BUFSIZ_REQUESTLINE];
|
||||
fgets(request_line, BUFSIZ_REQUESTLINE, fp);
|
||||
|
||||
std::string cmd, url;
|
||||
if (!parse_request_line(request_line, cmd, url)) {
|
||||
write_error(fd, 400);
|
||||
return;
|
||||
}
|
||||
|
||||
// Write content
|
||||
char content[BUFSIZ];
|
||||
sprintf(content, "cmd: %s, url: %s\n", cmd.c_str(), url.c_str());
|
||||
write_plain_text(fd, content);
|
||||
});
|
||||
}
|
||||
|
||||
} // namespace httpsvrkit
|
||||
|
||||
// vim: et ts=4 sw=4 cin cino={1s ff=unix
|
||||
|
||||
Loading…
Reference in a new issue