jyapayne/Simple-Web-Server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixes #142: a new socket is created when a https connection is reconnected, since a second SSL handshake cannot be made on the same socket. Also now removing failed connections from the connections collection.

Browse source
This commit is contained in:
eidheim 2017-07-15 07:30:27 +02:00
commit eeaf66cf2c
3 changed files with 57 additions and 71 deletions
Download patch file Download diff file Expand all files Collapse all files

84
client_http.hpp
View file

@ -2,9 +2,9 @@
#define CLIENT_HTTP_HPP
#include "utility.hpp"
#include <condition_variable>
#include <mutex>
#include <random>
#include <unordered_set>
#include <vector>
#ifdef USE_STANDALONE_ASIO
@ -384,7 +384,7 @@ namespace SimpleWeb {
std::unique_ptr<asio::ip::tcp::resolver::query> query;
std::vector<std::shared_ptr<Connection>> connections;
std::unordered_set<std::shared_ptr<Connection>> connections;
std::mutex connections_mutex;
std::shared_ptr<bool> cancel_handlers;
@ -418,7 +418,7 @@ namespace SimpleWeb {
}
if(!connection) {
connection = create_connection();
connections.emplace_back(connection);
connections.emplace(connection);
}
connection->attempt_reconnect = true;
connection->in_use = true;
@ -477,10 +477,8 @@ namespace SimpleWeb {
return;
if(!ec)
this->read(session);
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
@ -510,10 +508,8 @@ namespace SimpleWeb {
return;
if(!ec)
session->callback(ec);
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
else
@ -532,15 +528,8 @@ namespace SimpleWeb {
return;
if(!ec)
session->callback(ec);
else {
session->connection->close();
if(ec == asio::error::eof) {
error_code ec;
session->callback(ec);
}
else
session->callback(ec);
}
else
close(session, ec == asio::error::eof ? error_code() : ec);
});
}
else
@ -548,14 +537,22 @@ namespace SimpleWeb {
}
else {
if(session->connection->attempt_reconnect) {
session->connection->attempt_reconnect = false;
session->connection->close();
this->connect(session);
}
else {
session->connection->close();
session->callback(ec);
std::unique_lock<std::mutex> lock(connections_mutex);
auto it = connections.find(session->connection);
if(it != connections.end()) {
connections.erase(it);
session->connection->close();
session->connection = create_connection();
session->connection->attempt_reconnect = false;
session->connection->in_use = true;
connections.emplace(session->connection);
this->connect(session);
}
else
this->close(session, ec);
}
else
this->close(session, ec);
}
});
}
@ -607,21 +604,26 @@ namespace SimpleWeb {
return;
if(!ec)
post_process();
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
else
post_process();
}
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
void close(const std::shared_ptr<Session> &session, const error_code &ec) {
session->connection->close();
{
std::lock_guard<std::mutex> lock(connections_mutex);
connections.erase(session->connection);
}
session->callback(ec);
}
};
template <class socket_type>
@ -661,16 +663,12 @@ namespace SimpleWeb {
session->connection->socket->set_option(option, ec);
this->write(session);
}
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
else

36
client_https.hpp
View file

@ -85,38 +85,28 @@ namespace SimpleWeb {
return;
if(!ec) {
response->parse_header();
if(response->status_code.empty() || response->status_code.compare(0, 3, "200") != 0) {
session->connection->close();
session->callback(make_error_code::make_error_code(errc::permission_denied));
}
if(response->status_code.empty() || response->status_code.compare(0, 3, "200") != 0)
this->close(session, make_error_code::make_error_code(errc::permission_denied));
else
this->handshake(session);
}
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
else
this->handshake(session);
}
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
else
@ -132,10 +122,8 @@ namespace SimpleWeb {
return;
if(!ec)
this->write(session);
else {
session->connection->close();
session->callback(ec);
}
else
this->close(session, ec);
});
}
};

8
tests/io_test.cpp
View file

@ -206,7 +206,7 @@ int main() {
output << r->content.rdbuf();
assert(output.str() == "A string");
assert(client.connections.size() == 1);
connection = client.connections.front().get();
connection = client.connections.begin()->get();
}
{
@ -215,7 +215,7 @@ int main() {
output << r->content.rdbuf();
assert(output.str() == "A string");
assert(client.connections.size() == 1);
assert(connection == client.connections.front().get());
assert(connection == client.connections.begin()->get());
}
{
@ -224,7 +224,7 @@ int main() {
output << r->content.rdbuf();
assert(output.str() == "testing");
assert(client.connections.size() == 1);
assert(connection == client.connections.front().get());
assert(connection == client.connections.begin()->get());
}
{
@ -232,7 +232,7 @@ int main() {
auto r = client.request("GET", "/query_string?testing");
assert(r->content.string() == "testing");
assert(client.connections.size() == 1);
assert(connection == client.connections.front().get());
assert(connection == client.connections.begin()->get());
}
}