Fixes for https client over AutoSocket.

This commit is contained in:
Arthur Britto
2013-02-01 19:02:08 -08:00
parent 628a1bdd5f
commit c25f480cb8
3 changed files with 39 additions and 31 deletions

View File

@@ -13,6 +13,9 @@
#include <boost/system/error_code.hpp> #include <boost/system/error_code.hpp>
#include "Config.h" #include "Config.h"
#include "Log.h"
SETUP_LOG();
using namespace boost::system; using namespace boost::system;
using namespace boost::asio; using namespace boost::asio;
@@ -77,14 +80,15 @@ void HttpsClient::httpsGet(
void HttpsClient::httpsNext() void HttpsClient::httpsNext()
{ {
std::cerr << "Fetch: " << mDeqSites[0] << std::endl; cLog(lsTRACE) << "Fetch: " << mDeqSites[0];
boost::shared_ptr<boost::asio::ip::tcp::resolver::query> query(new boost::asio::ip::tcp::resolver::query(mDeqSites[0], boost::lexical_cast<std::string>(mPort), boost::shared_ptr<boost::asio::ip::tcp::resolver::query> query(new boost::asio::ip::tcp::resolver::query(mDeqSites[0], boost::lexical_cast<std::string>(mPort),
ip::resolver_query_base::numeric_service)); ip::resolver_query_base::numeric_service));
mQuery = query; mQuery = query;
mDeadline.expires_from_now(mTimeout, mShutdown); mDeadline.expires_from_now(mTimeout, mShutdown);
std::cerr << "expires_from_now: " << mShutdown.message() << std::endl; cLog(lsTRACE) << "expires_from_now: " << mShutdown.message();
if (!mShutdown) if (!mShutdown)
{ {
@@ -97,7 +101,7 @@ std::cerr << "expires_from_now: " << mShutdown.message() << std::endl;
if (!mShutdown) if (!mShutdown)
{ {
std::cerr << "Resolving: " << mDeqSites[0] << std::endl; cLog(lsTRACE) << "Resolving: " << mDeqSites[0];
mResolver.async_resolve(*mQuery, mResolver.async_resolve(*mQuery,
boost::bind( boost::bind(
@@ -116,20 +120,20 @@ void HttpsClient::handleDeadline(const boost::system::error_code& ecResult)
if (ecResult == boost::asio::error::operation_aborted) if (ecResult == boost::asio::error::operation_aborted)
{ {
// Timer canceled because deadline no longer needed. // Timer canceled because deadline no longer needed.
std::cerr << "Deadline cancelled." << std::endl; cLog(lsTRACE) << "Deadline cancelled.";
nothing(); // Aborter is done. nothing(); // Aborter is done.
} }
else if (ecResult) else if (ecResult)
{ {
std::cerr << "Deadline error: " << mDeqSites[0] << ": " << ecResult.message() << std::endl; cLog(lsTRACE) << "Deadline error: " << mDeqSites[0] << ": " << ecResult.message();
// Can't do anything sound. // Can't do anything sound.
abort(); abort();
} }
else else
{ {
std::cerr << "Deadline arrived." << std::endl; cLog(lsTRACE) << "Deadline arrived.";
// Mark us as shutting down. // Mark us as shutting down.
// XXX Use our own error code. // XXX Use our own error code.
@@ -153,7 +157,7 @@ void HttpsClient::handleShutdown(
{ {
if (ecResult) if (ecResult)
{ {
std::cerr << "Shutdown error: " << mDeqSites[0] << ": " << ecResult.message() << std::endl; cLog(lsTRACE) << "Shutdown error: " << mDeqSites[0] << ": " << ecResult.message();
} }
} }
@@ -167,13 +171,13 @@ void HttpsClient::handleResolve(
if (mShutdown) if (mShutdown)
{ {
std::cerr << "Resolve error: " << mDeqSites[0] << ": " << mShutdown.message() << std::endl; cLog(lsTRACE) << "Resolve error: " << mDeqSites[0] << ": " << mShutdown.message();
invokeComplete(mShutdown); invokeComplete(mShutdown);
} }
else else
{ {
std::cerr << "Resolve complete." << std::endl; cLog(lsTRACE) << "Resolve complete.";
boost::asio::async_connect( boost::asio::async_connect(
mSocket.lowest_layer(), mSocket.lowest_layer(),
@@ -192,17 +196,18 @@ void HttpsClient::handleConnect(const boost::system::error_code& ecResult)
if (mShutdown) if (mShutdown)
{ {
std::cerr << "Connect error: " << mShutdown.message() << std::endl; cLog(lsTRACE) << "Connect error: " << mShutdown.message();
} }
if (!mShutdown) if (!mShutdown)
{ {
std::cerr << "Connected." << std::endl; cLog(lsTRACE) << "Connected.";
mShutdown = mSocket.verify(mDeqSites[0]); mShutdown = mSocket.verify(mDeqSites[0]);
if (mShutdown) if (mShutdown)
{ {
std::cerr << "set_verify_callback: " << mDeqSites[0] << ": " << mShutdown.message() << std::endl; cLog(lsTRACE) << "set_verify_callback: " << mDeqSites[0] << ": " << mShutdown.message();
} }
} }
@@ -232,13 +237,14 @@ void HttpsClient::handleRequest(const boost::system::error_code& ecResult)
if (mShutdown) if (mShutdown)
{ {
std::cerr << "Handshake error:" << mShutdown.message() << std::endl; cLog(lsTRACE) << "Handshake error:" << mShutdown.message();
invokeComplete(mShutdown); invokeComplete(mShutdown);
} }
else else
{ {
std::cerr << "Session started." << std::endl; cLog(lsTRACE) << "Session started.";
mBuild(mRequest, mDeqSites[0]); mBuild(mRequest, mDeqSites[0]);
mSocket.async_write( mSocket.async_write(
@@ -257,32 +263,32 @@ void HttpsClient::handleWrite(const boost::system::error_code& ecResult, std::si
if (mShutdown) if (mShutdown)
{ {
std::cerr << "Write error: " << mShutdown.message() << std::endl; cLog(lsTRACE) << "Write error: " << mShutdown.message();
invokeComplete(mShutdown); invokeComplete(mShutdown);
} }
else else
{ {
std::cerr << "Wrote." << std::endl; cLog(lsTRACE) << "Wrote.";
boost::asio::async_read( mSocket.async_read(
mSocket,
mResponse, mResponse,
boost::asio::transfer_all(), boost::asio::transfer_all(),
boost::bind(&HttpsClient::handleData, boost::bind(&HttpsClient::handleData,
shared_from_this(), shared_from_this(),
boost::asio::placeholders::error)); boost::asio::placeholders::error,
boost::asio::placeholders::bytes_transferred));
} }
} }
void HttpsClient::handleData(const boost::system::error_code& ecResult) void HttpsClient::handleData(const boost::system::error_code& ecResult, std::size_t bytes_transferred)
{ {
if (!mShutdown) if (!mShutdown)
mShutdown = ecResult; mShutdown = ecResult;
if (mShutdown && mShutdown != boost::asio::error::eof) if (mShutdown && mShutdown != boost::asio::error::eof)
{ {
std::cerr << "Read error: " << mShutdown.message() << std::endl; cLog(lsTRACE) << "Read error: " << mShutdown.message();
invokeComplete(mShutdown); invokeComplete(mShutdown);
} }
@@ -290,13 +296,14 @@ void HttpsClient::handleData(const boost::system::error_code& ecResult)
{ {
if (mShutdown) if (mShutdown)
{ {
std::cerr << "Complete." << std::endl; cLog(lsTRACE) << "Complete.";
nothing(); nothing();
} }
else else
{ {
// XXX According to boost example code, this is what we should expect for success. // XXX According to boost example code, this is what we should expect for success.
std::cerr << "Complete, no eof." << std::endl; cLog(lsTRACE) << "Complete, no eof.";
} }
parseData(); parseData();
@@ -312,7 +319,7 @@ void HttpsClient::invokeComplete(const boost::system::error_code& ecResult, int
if (ecCancel) if (ecCancel)
{ {
std::cerr << "Deadline cancel error: " << ecCancel.message() << std::endl; cLog(lsTRACE) << "Deadline cancel error: " << ecCancel.message();
} }
mDeqSites.pop_front(); mDeqSites.pop_front();
@@ -340,20 +347,21 @@ void HttpsClient::parseData()
static boost::regex reStatus("\\`HTTP/1\\S+ (\\d{3}) .*\\'"); // HTTP/1.1 200 OK static boost::regex reStatus("\\`HTTP/1\\S+ (\\d{3}) .*\\'"); // HTTP/1.1 200 OK
static boost::regex reBody("\\`(?:.*?\\r\\n\\r\\n){1}(.*)\\'"); static boost::regex reBody("\\`(?:.*?\\r\\n\\r\\n){1}(.*)\\'");
boost::smatch smMatch; boost::smatch smStatus;
boost::smatch smBody;
bool bMatch = boost::regex_match(strData, smMatch, reStatus) // Match status code. bool bMatch = boost::regex_match(strData, smStatus, reStatus) // Match status code.
&& boost::regex_match(strData, smMatch, reBody); // Match body. && boost::regex_match(strData, smBody, reBody); // Match body.
// std::cerr << "Data:" << strData << std::endl; // std::cerr << "Data:" << strData << std::endl;
// std::cerr << "Match: " << bMatch << std::endl; // std::cerr << "Match: " << bMatch << std::endl;
// std::cerr << "Body:" << smMatch[1] << std::endl; // std::cerr << "Body:" << smBody[1] << std::endl;
if (bMatch) if (bMatch)
{ {
boost::system::error_code noErr; boost::system::error_code noErr;
invokeComplete(noErr, lexical_cast_st<int>(smMatch[1]), smMatch[1]); invokeComplete(noErr, lexical_cast_st<int>(smStatus[1]), smBody[1]);
} }
else else
{ {

View File

@@ -50,7 +50,7 @@ private:
void handleWrite(const boost::system::error_code& ecResult, std::size_t bytes_transferred); void handleWrite(const boost::system::error_code& ecResult, std::size_t bytes_transferred);
void handleData(const boost::system::error_code& ecResult); void handleData(const boost::system::error_code& ecResult, std::size_t bytes_transferred);
void handleShutdown(const boost::system::error_code& ecResult); void handleShutdown(const boost::system::error_code& ecResult);

View File

@@ -45,7 +45,7 @@ std::string createHTTPPost(const std::string& strHost, const std::string& strPat
s << "POST " s << "POST "
<< (strPath.empty() ? "/" : strPath) << (strPath.empty() ? "/" : strPath)
<< " HTTP/1.1\r\n" << " HTTP/1.0\r\n"
<< "User-Agent: " SYSTEM_NAME "-json-rpc/" << FormatFullVersion() << "\r\n" << "User-Agent: " SYSTEM_NAME "-json-rpc/" << FormatFullVersion() << "\r\n"
<< "Host: " << strHost << "\r\n" << "Host: " << strHost << "\r\n"
<< "Content-Type: application/json\r\n" << "Content-Type: application/json\r\n"