yuzu/cpp-httplib
1
0
Fork 0
mirror of https://github.com/yhirose/cpp-httplib synced 2024-11-21 14:29:10 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added capture and setting of REMOTE_ADDR in request.

Browse source
This commit is contained in:
Kevin B. Carpenter 2018-03-06 08:20:51 -07:00
parent 25aa0b34c3
commit 73fa115803
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
httplib.h
View file

@ -157,6 +157,7 @@ public:
virtual int read(char* ptr, size_t size) = 0; virtual int read(char* ptr, size_t size) = 0;
virtual int write(const char* ptr, size_t size1) = 0; virtual int write(const char* ptr, size_t size1) = 0;
virtual int write(const char* ptr) = 0; virtual int write(const char* ptr) = 0;
virtual std::string get_remote_addr() = 0;
template <typename ...Args> template <typename ...Args>
void write_format(const char* fmt, const Args& ...args); void write_format(const char* fmt, const Args& ...args);
@ -170,6 +171,26 @@ public:
virtual int read(char* ptr, size_t size); virtual int read(char* ptr, size_t size);
virtual int write(const char* ptr, size_t size); virtual int write(const char* ptr, size_t size);
virtual int write(const char* ptr); virtual int write(const char* ptr);
std::string get_remote_addr() {
socklen_t len;
struct sockaddr_storage addr;
char ipstr[INET6_ADDRSTRLEN];
len = sizeof addr;
getpeername(sock_, (struct sockaddr*)&addr, &len);
// deal with both IPv4 and IPv6:
if (addr.ss_family == AF_INET) {
struct sockaddr_in *s = (struct sockaddr_in *)&addr;
inet_ntop(AF_INET, &s->sin_addr, ipstr, sizeof ipstr);
} else { // AF_INET6
struct sockaddr_in6 *s = (struct sockaddr_in6 *)&addr;
inet_ntop(AF_INET6, &s->sin6_addr, ipstr, sizeof ipstr);
}
return ipstr;
}
private: private:
socket_t sock_; socket_t sock_;
@ -1583,6 +1604,8 @@ inline bool Server::process_request(Stream& strm, bool last_connection)
ret = false; ret = false;
} }
req.set_header("REMOTE_ADDR", strm.get_remote_addr().c_str());
// Body // Body
if (req.method == "POST") { if (req.method == "POST") {
if (!detail::read_content(strm, req)) { if (!detail::read_content(strm, req)) {