57 lines
1.2 KiB
C++
57 lines
1.2 KiB
C++
|
#pragma once
|
||
|
|
|
||
|
|
#include <nall/http/role.hpp>
|
||
|
|
|
||
|
|
namespace nall { namespace HTTP {
|
||
|
|
|
||
|
|
struct Client : Role {
|
||
|
|
inline auto open(const string& hostname, unsigned port = 80) -> bool;
|
||
|
|
inline auto upload(const Request& request) -> bool;
|
||
|
|
inline auto download(const Request& request) -> Response;
|
||
|
|
inline auto close() -> void;
|
||
|
|
~Client() { close(); }
|
||
|
|
|
||
|
|
private:
|
||
|
|
signed fd = -1;
|
||
|
|
addrinfo* info = nullptr;
|
||
|
|
};
|
||
|
|
|
||
|
|
auto Client::open(const string& hostname, unsigned port) -> bool {
|
||
|
|
addrinfo hint = {0};
|
||
|
|
hint.ai_family = AF_UNSPEC;
|
||
|
|
hint.ai_socktype = SOCK_STREAM;
|
||
|
|
hint.ai_flags = AI_ADDRCONFIG;
|
||
|
|
|
||
|
|
if(getaddrinfo(hostname, string{port}, &hint, &info) != 0) return close(), false;
|
||
|
|
|
||
|
|
fd = socket(info->ai_family, info->ai_socktype, info->ai_protocol);
|
||
|
|
if(fd < 0) return close(), false;
|
||
|
|
|
||
|
|
if(connect(fd, info->ai_addr, info->ai_addrlen) < 0) return close(), false;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
auto Client::upload(const Request& request) -> bool {
|
||
|
|
return Role::upload(fd, request);
|
||
|
|
}
|
||
|
|
|
||
|
|
auto Client::download(const Request& request) -> Response {
|
||
|
|
Response response(request);
|
||
|
|
Role::download(fd, response);
|
||
|
|
return response;
|
||
|
|
}
|
||
|
|
|
||
|
|
auto Client::close() -> void {
|
||
|
|
if(fd) {
|
||
|
|
::close(fd);
|
||
|
|
fd = -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(info) {
|
||
|
|
freeaddrinfo(info);
|
||
|
|
info = nullptr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}}
|