diff --git a/test/test.cc b/test/test.cc index a0d793b..d6610a7 100644 --- a/test/test.cc +++ b/test/test.cc @@ -54,6 +54,166 @@ MultipartFormData &get_file_value(MultipartFormDataItems &files, #endif } +#ifndef _WIN32 +class UnixSocketTest : public ::testing::Test { +protected: + void TearDown() override { std::remove(pathname_.c_str()); } + + void client_GET(const std::string &addr) { + httplib::Client cli{addr}; + cli.set_address_family(AF_UNIX); + ASSERT_TRUE(cli.is_valid()); + + const auto &result = cli.Get(pattern_); + ASSERT_TRUE(result) << "error: " << result.error(); + + const auto &resp = result.value(); + EXPECT_EQ(resp.status, StatusCode::OK_200); + EXPECT_EQ(resp.body, content_); + } + + const std::string pathname_{"./httplib-server.sock"}; + const std::string pattern_{"/hi"}; + const std::string content_{"Hello World!"}; +}; + +TEST_F(UnixSocketTest, pathname) { + httplib::Server svr; + svr.Get(pattern_, [&](const httplib::Request &, httplib::Response &res) { + res.set_content(content_, "text/plain"); + }); + + std::thread t{[&] { + ASSERT_TRUE(svr.set_address_family(AF_UNIX).listen(pathname_, 80)); + }}; + auto se = detail::scope_exit([&] { + svr.stop(); + t.join(); + ASSERT_FALSE(svr.is_running()); + }); + + svr.wait_until_ready(); + ASSERT_TRUE(svr.is_running()); + + client_GET(pathname_); +} + +#if defined(__linux__) || \ + /* __APPLE__ */ (defined(SOL_LOCAL) && defined(SO_PEERPID)) +TEST_F(UnixSocketTest, PeerPid) { + httplib::Server svr; + std::string remote_port_val; + svr.Get(pattern_, [&](const httplib::Request &req, httplib::Response &res) { + res.set_content(content_, "text/plain"); + remote_port_val = req.get_header_value("REMOTE_PORT"); + }); + + std::thread t{[&] { + ASSERT_TRUE(svr.set_address_family(AF_UNIX).listen(pathname_, 80)); + }}; + auto se = detail::scope_exit([&] { + svr.stop(); + t.join(); + ASSERT_FALSE(svr.is_running()); + }); + + svr.wait_until_ready(); + ASSERT_TRUE(svr.is_running()); + + client_GET(pathname_); + EXPECT_EQ(std::to_string(getpid()), remote_port_val); +} +#endif + +#ifdef __linux__ +TEST_F(UnixSocketTest, abstract) { + constexpr char svr_path[]{"\x00httplib-server.sock"}; + const std::string abstract_addr{svr_path, sizeof(svr_path) - 1}; + + httplib::Server svr; + svr.Get(pattern_, [&](const httplib::Request &, httplib::Response &res) { + res.set_content(content_, "text/plain"); + }); + + std::thread t{[&] { + ASSERT_TRUE(svr.set_address_family(AF_UNIX).listen(abstract_addr, 80)); + }}; + auto se = detail::scope_exit([&] { + svr.stop(); + t.join(); + ASSERT_FALSE(svr.is_running()); + }); + + svr.wait_until_ready(); + ASSERT_TRUE(svr.is_running()); + + client_GET(abstract_addr); +} +#endif + +TEST(SocketStream, is_writable_UNIX) { + int fds[2]; + ASSERT_EQ(0, socketpair(AF_UNIX, SOCK_STREAM, 0, fds)); + + const auto asSocketStream = [&](socket_t fd, + std::function func) { + return detail::process_client_socket(fd, 0, 0, 0, 0, func); + }; + asSocketStream(fds[0], [&](Stream &s0) { + EXPECT_EQ(s0.socket(), fds[0]); + EXPECT_TRUE(s0.is_writable()); + + EXPECT_EQ(0, close(fds[1])); + EXPECT_FALSE(s0.is_writable()); + + return true; + }); + EXPECT_EQ(0, close(fds[0])); +} + +TEST(SocketStream, is_writable_INET) { + sockaddr_in addr; + memset(&addr, 0, sizeof(addr)); + addr.sin_family = AF_INET; + addr.sin_port = htons(PORT + 1); + addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK); + + int disconnected_svr_sock = -1; + std::thread svr{[&] { + const int s = socket(AF_INET, SOCK_STREAM, 0); + ASSERT_LE(0, s); + ASSERT_EQ(0, ::bind(s, reinterpret_cast(&addr), sizeof(addr))); + ASSERT_EQ(0, listen(s, 1)); + ASSERT_LE(0, disconnected_svr_sock = accept(s, nullptr, nullptr)); + ASSERT_EQ(0, close(s)); + }}; + std::this_thread::sleep_for(std::chrono::milliseconds(100)); + + std::thread cli{[&] { + const int s = socket(AF_INET, SOCK_STREAM, 0); + ASSERT_LE(0, s); + ASSERT_EQ(0, connect(s, reinterpret_cast(&addr), sizeof(addr))); + ASSERT_EQ(0, close(s)); + }}; + cli.join(); + svr.join(); + ASSERT_NE(disconnected_svr_sock, -1); + + const auto asSocketStream = [&](socket_t fd, + std::function func) { + return detail::process_client_socket(fd, 0, 0, 0, 0, func); + }; + asSocketStream(disconnected_svr_sock, [&](Stream &ss) { + EXPECT_EQ(ss.socket(), disconnected_svr_sock); + EXPECT_FALSE(ss.is_writable()); + + return true; + }); + + ASSERT_EQ(0, close(disconnected_svr_sock)); +} +#endif // #ifndef _WIN32 + TEST(ClientTest, MoveConstructible) { EXPECT_FALSE(std::is_copy_constructible::value); EXPECT_TRUE(std::is_nothrow_move_constructible::value); @@ -7024,166 +7184,6 @@ TEST(MultipartFormDataTest, ContentLength) { #endif -#ifndef _WIN32 -class UnixSocketTest : public ::testing::Test { -protected: - void TearDown() override { std::remove(pathname_.c_str()); } - - void client_GET(const std::string &addr) { - httplib::Client cli{addr}; - cli.set_address_family(AF_UNIX); - ASSERT_TRUE(cli.is_valid()); - - const auto &result = cli.Get(pattern_); - ASSERT_TRUE(result) << "error: " << result.error(); - - const auto &resp = result.value(); - EXPECT_EQ(resp.status, StatusCode::OK_200); - EXPECT_EQ(resp.body, content_); - } - - const std::string pathname_{"./httplib-server.sock"}; - const std::string pattern_{"/hi"}; - const std::string content_{"Hello World!"}; -}; - -TEST_F(UnixSocketTest, pathname) { - httplib::Server svr; - svr.Get(pattern_, [&](const httplib::Request &, httplib::Response &res) { - res.set_content(content_, "text/plain"); - }); - - std::thread t{[&] { - ASSERT_TRUE(svr.set_address_family(AF_UNIX).listen(pathname_, 80)); - }}; - auto se = detail::scope_exit([&] { - svr.stop(); - t.join(); - ASSERT_FALSE(svr.is_running()); - }); - - svr.wait_until_ready(); - ASSERT_TRUE(svr.is_running()); - - client_GET(pathname_); -} - -#if defined(__linux__) || \ - /* __APPLE__ */ (defined(SOL_LOCAL) && defined(SO_PEERPID)) -TEST_F(UnixSocketTest, PeerPid) { - httplib::Server svr; - std::string remote_port_val; - svr.Get(pattern_, [&](const httplib::Request &req, httplib::Response &res) { - res.set_content(content_, "text/plain"); - remote_port_val = req.get_header_value("REMOTE_PORT"); - }); - - std::thread t{[&] { - ASSERT_TRUE(svr.set_address_family(AF_UNIX).listen(pathname_, 80)); - }}; - auto se = detail::scope_exit([&] { - svr.stop(); - t.join(); - ASSERT_FALSE(svr.is_running()); - }); - - svr.wait_until_ready(); - ASSERT_TRUE(svr.is_running()); - - client_GET(pathname_); - EXPECT_EQ(std::to_string(getpid()), remote_port_val); -} -#endif - -#ifdef __linux__ -TEST_F(UnixSocketTest, abstract) { - constexpr char svr_path[]{"\x00httplib-server.sock"}; - const std::string abstract_addr{svr_path, sizeof(svr_path) - 1}; - - httplib::Server svr; - svr.Get(pattern_, [&](const httplib::Request &, httplib::Response &res) { - res.set_content(content_, "text/plain"); - }); - - std::thread t{[&] { - ASSERT_TRUE(svr.set_address_family(AF_UNIX).listen(abstract_addr, 80)); - }}; - auto se = detail::scope_exit([&] { - svr.stop(); - t.join(); - ASSERT_FALSE(svr.is_running()); - }); - - svr.wait_until_ready(); - ASSERT_TRUE(svr.is_running()); - - client_GET(abstract_addr); -} -#endif - -TEST(SocketStream, is_writable_UNIX) { - int fds[2]; - ASSERT_EQ(0, socketpair(AF_UNIX, SOCK_STREAM, 0, fds)); - - const auto asSocketStream = [&](socket_t fd, - std::function func) { - return detail::process_client_socket(fd, 0, 0, 0, 0, func); - }; - asSocketStream(fds[0], [&](Stream &s0) { - EXPECT_EQ(s0.socket(), fds[0]); - EXPECT_TRUE(s0.is_writable()); - - EXPECT_EQ(0, close(fds[1])); - EXPECT_FALSE(s0.is_writable()); - - return true; - }); - EXPECT_EQ(0, close(fds[0])); -} - -TEST(SocketStream, is_writable_INET) { - sockaddr_in addr; - memset(&addr, 0, sizeof(addr)); - addr.sin_family = AF_INET; - addr.sin_port = htons(PORT + 1); - addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK); - - int disconnected_svr_sock = -1; - std::thread svr{[&] { - const int s = socket(AF_INET, SOCK_STREAM, 0); - ASSERT_LE(0, s); - ASSERT_EQ(0, ::bind(s, reinterpret_cast(&addr), sizeof(addr))); - ASSERT_EQ(0, listen(s, 1)); - ASSERT_LE(0, disconnected_svr_sock = accept(s, nullptr, nullptr)); - ASSERT_EQ(0, close(s)); - }}; - std::this_thread::sleep_for(std::chrono::milliseconds(100)); - - std::thread cli{[&] { - const int s = socket(AF_INET, SOCK_STREAM, 0); - ASSERT_LE(0, s); - ASSERT_EQ(0, connect(s, reinterpret_cast(&addr), sizeof(addr))); - ASSERT_EQ(0, close(s)); - }}; - cli.join(); - svr.join(); - ASSERT_NE(disconnected_svr_sock, -1); - - const auto asSocketStream = [&](socket_t fd, - std::function func) { - return detail::process_client_socket(fd, 0, 0, 0, 0, func); - }; - asSocketStream(disconnected_svr_sock, [&](Stream &ss) { - EXPECT_EQ(ss.socket(), disconnected_svr_sock); - EXPECT_FALSE(ss.is_writable()); - - return true; - }); - - ASSERT_EQ(0, close(disconnected_svr_sock)); -} -#endif // #ifndef _WIN32 - TEST(TaskQueueTest, IncreaseAtomicInteger) { static constexpr unsigned int number_of_tasks{1000000}; std::atomic_uint count{0};