sco/pokerth
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Trying to fix connectivity tool.

Browse Source
This commit is contained in:
lotodore
2011-01-17 10:44:47 +00:00
parent 4a6367cea3
commit 0255568b80
2 changed files with 270 additions and 262 deletions
Download Patch File Download Diff File Expand all files Collapse all files
connectivity.pro
+1 -1
View File
@@ -15,7 +15,7 @@ MOC_DIR = mocs
OBJECTS_DIR = obj OBJECTS_DIR = obj
DEFINES += PREFIX=\"$${PREFIX}\" DEFINES += PREFIX=\"$${PREFIX}\"
QT -= core gui QT -= core gui
QMAKE_CXXFLAGS += -std=gnu++0x #QMAKE_CXXFLAGS += -std=gnu++0x
#PRECOMPILED_HEADER = src/pch_lib.h #PRECOMPILED_HEADER = src/pch_lib.h
INCLUDEPATH += . \ INCLUDEPATH += . \
src/connectivity.cpp
+269 -261
View File
@@ -33,10 +33,12 @@ using namespace std;
using boost::asio::ip::tcp; using boost::asio::ip::tcp;
namespace po = boost::program_options; namespace po = boost::program_options;
#define BUF_SIZE 1024
// Global receive buffer // Global receive buffer
boost::array<char, 512> recBuf; boost::array<char, BUF_SIZE> recBuf;
size_t recBufPos = 0; size_t recBufPos = 0;
boost::array<char, 512> sendBuf; boost::array<char, BUF_SIZE> sendBuf;
static int static int
net_packet_print_to_string(const void *buffer, size_t size, void *packetStr) net_packet_print_to_string(const void *buffer, size_t size, void *packetStr)
@@ -51,34 +53,31 @@ xer_encode(&asn_DEF_PokerTHMessage, msg, XER_F_BASIC, &net_packet_print_to_strin
cout << packetString << endl;*/ cout << packetString << endl;*/
PokerTHMessage_t * PokerTHMessage_t *
receiveMessage(tcp::socket &socket, bool recursed = false) receiveMessage(tcp::socket &socket)
{ {
PokerTHMessage_t *msg = NULL; PokerTHMessage_t *msg = NULL;
asn_dec_rval_t retVal = ber_decode(0, &asn_DEF_PokerTHMessage, (void **)&msg, recBuf.data(), recBufPos); do
if(retVal.code == RC_OK)
{ {
if (retVal.consumed < recBufPos) asn_dec_rval_t retVal = ber_decode(0, &asn_DEF_PokerTHMessage, (void **)&msg, recBuf.data(), recBufPos);
if(retVal.code == RC_OK)
{ {
recBufPos -= retVal.consumed; if (retVal.consumed < recBufPos)
memmove(recBuf.c_array(), recBuf.c_array() + retVal.consumed, recBufPos); {
recBufPos -= retVal.consumed;
memmove(recBuf.c_array(), recBuf.c_array() + retVal.consumed, recBufPos);
}
else
{
recBufPos = 0;
}
} }
else else
{ {
recBufPos = 0; // Free the partially decoded message (if applicable).
ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg);
recBufPos += socket.receive(boost::asio::buffer(recBuf.c_array() + recBufPos, BUF_SIZE - recBufPos));
} }
} } while (recBufPos < BUF_SIZE && msg != NULL);
else
{
// Free the partially decoded message (if applicable).
ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg);
if (!recursed)
{
recBufPos += socket.receive(boost::asio::buffer(recBuf.c_array() + recBufPos, 512 - recBufPos));
return receiveMessage(socket, true);
}
else
return msg = NULL;
}
return msg; return msg;
} }
@@ -88,7 +87,7 @@ sendMessage(tcp::socket &socket, PokerTHMessage_t *msg)
bool retVal = false; bool retVal = false;
if (msg) if (msg)
{ {
asn_enc_rval_t e = der_encode_to_buffer(&asn_DEF_PokerTHMessage, msg, sendBuf.data(), 512); asn_enc_rval_t e = der_encode_to_buffer(&asn_DEF_PokerTHMessage, msg, sendBuf.data(), BUF_SIZE);
if (e.encoded != -1) if (e.encoded != -1)
{ {
socket.send(boost::asio::buffer(sendBuf.data(), e.encoded)); socket.send(boost::asio::buffer(sendBuf.data(), e.encoded));
@@ -102,271 +101,280 @@ sendMessage(tcp::socket &socket, PokerTHMessage_t *msg)
int int
main(int argc, char *argv[]) main(int argc, char *argv[])
{ {
// Check command line options. try
po::options_description desc("Allowed options"); {
desc.add_options() // Check command line options.
("help,h", "produce help message") po::options_description desc("Allowed options");
("server,s", po::value<string>(), "PokerTH server name") desc.add_options()
("port,P", po::value<string>(), "PokerTH server port") ("help,h", "produce help message")
("mode,m", po::value<int>(), "set mode (0=connection test, 1=lag test)") ("server,s", po::value<string>(), "PokerTH server name")
("username,u", po::value<string>(), "user name used for test") ("port,P", po::value<string>(), "PokerTH server port")
("password,p", po::value<string>(), "password used for test") ("mode,m", po::value<int>(), "set mode (0=connection test, 1=lag test)")
; ("username,u", po::value<string>(), "user name used for test")
("password,p", po::value<string>(), "password used for test")
;
po::variables_map vm; po::variables_map vm;
po::store(po::parse_command_line(argc, argv, desc), vm); po::store(po::parse_command_line(argc, argv, desc), vm);
po::notify(vm); po::notify(vm);
if (vm.count("help")) if (vm.count("help"))
{ {
cout << desc << endl; cout << desc << endl;
return 1; return 1;
} }
if (!vm.count("server") || !vm.count("port") || !vm.count("mode") || !vm.count("username")) if (!vm.count("server") || !vm.count("port") || !vm.count("mode") || !vm.count("username"))
{ {
cout << "Missing option!" << endl << desc << endl; cout << "Missing option!" << endl << desc << endl;
return 1; return 1;
} }
string server(vm["server"].as<string>()); string server(vm["server"].as<string>());
string port = vm["port"].as<string>(); string port = vm["port"].as<string>();
int mode = vm["mode"].as<int>(); int mode = vm["mode"].as<int>();
string username(vm["username"].as<string>()); string username(vm["username"].as<string>());
string password; string password;
if (vm.count("password")) if (vm.count("password"))
{ {
password = vm["password"].as<string>(); password = vm["password"].as<string>();
} }
// Initialise gsasl. // Initialise gsasl.
Gsasl *authContext; Gsasl *authContext;
Gsasl_session *authSession; Gsasl_session *authSession;
int res = gsasl_init(&authContext); int res = gsasl_init(&authContext);
if (res != GSASL_OK) if (res != GSASL_OK)
{ {
cout << "gsasl init failed" << endl; cout << "gsasl init failed" << endl;
return 1; return 1;
} }
if (!gsasl_client_support_p(authContext, "SCRAM-SHA-1")) if (!gsasl_client_support_p(authContext, "SCRAM-SHA-1"))
{ {
gsasl_done(authContext); gsasl_done(authContext);
cout << "This version of gsasl does not support SCRAM-SHA-1" << endl; cout << "This version of gsasl does not support SCRAM-SHA-1" << endl;
return 1; return 1;
} }
// Connect to the PokerTH server. // Connect to the PokerTH server.
boost::timers::portable::microsec_timer perfTimer; boost::timers::portable::microsec_timer perfTimer;
boost::asio::io_service io_service; boost::asio::io_service io_service;
tcp::resolver resolver(io_service); tcp::resolver resolver(io_service);
tcp::resolver::query query(server, port); tcp::resolver::query query(server, port);
tcp::resolver::iterator endpoint_iterator = resolver.resolve(query); tcp::resolver::iterator endpoint_iterator = resolver.resolve(query);
tcp::resolver::iterator end; tcp::resolver::iterator end;
tcp::socket socket(io_service); tcp::socket socket(io_service);
boost::system::error_code error = boost::asio::error::host_not_found; boost::system::error_code error = boost::asio::error::host_not_found;
while (error && endpoint_iterator != end) while (error && endpoint_iterator != end)
{ {
socket.close(); socket.close();
socket.connect(*endpoint_iterator++, error); socket.connect(*endpoint_iterator++, error);
} }
if (error) if (error)
{ {
cout << "Connect failed" << endl; cout << "Connect failed" << endl;
return 1; return 1;
} }
if (mode == 1) if (mode == 1)
{ {
cout << "Connect.value " << perfTimer.elapsed().total_milliseconds() << endl; cout << "Connect.value " << perfTimer.elapsed().total_milliseconds() << endl;
} }
perfTimer.restart(); perfTimer.restart();
// Receive server information // Receive server information
PokerTHMessage_t *msg = receiveMessage(socket); PokerTHMessage_t *msg = receiveMessage(socket);
if (!msg || msg->present != PokerTHMessage_PR_announceMessage) if (!msg || msg->present != PokerTHMessage_PR_announceMessage)
{ {
cout << "Announce failed" << endl; cout << "Announce failed" << endl;
return 1; return 1;
} }
ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg); ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg);
// Send init // Send init
msg = (PokerTHMessage_t *)calloc(1, sizeof(PokerTHMessage_t)); msg = (PokerTHMessage_t *)calloc(1, sizeof(PokerTHMessage_t));
msg->present = PokerTHMessage_PR_initMessage; msg->present = PokerTHMessage_PR_initMessage;
InitMessage_t *netInit = &msg->choice.initMessage; InitMessage_t *netInit = &msg->choice.initMessage;
netInit->requestedVersion.major = 1; netInit->requestedVersion.major = 1;
netInit->requestedVersion.minor = 0; netInit->requestedVersion.minor = 0;
if (password.empty()) if (password.empty())
{ {
netInit->login.present = login_PR_guestLogin; netInit->login.present = login_PR_guestLogin;
GuestLogin_t *guestLogin = &netInit->login.choice.guestLogin; GuestLogin_t *guestLogin = &netInit->login.choice.guestLogin;
OCTET_STRING_fromBuf(&guestLogin->nickName, OCTET_STRING_fromBuf(&guestLogin->nickName,
username.c_str(), username.c_str(),
username.length()); username.length());
if (!sendMessage(socket, msg))
{
cout << "Init guest failed" << endl;
return 1;
}
}
else
{
int errorCode = gsasl_client_start(authContext, "SCRAM-SHA-1", &authSession);
if (errorCode == GSASL_OK)
{
gsasl_property_set(authSession, GSASL_AUTHID, username.c_str());
gsasl_property_set(authSession, GSASL_PASSWORD, password.c_str());
netInit->login.present = login_PR_authenticatedLogin;
AuthenticatedLogin_t *authLogin = &netInit->login.choice.authenticatedLogin;
char *tmpOut;
size_t tmpOutSize;
string nextGsaslMsg;
errorCode = gsasl_step(authSession, NULL, 0, &tmpOut, &tmpOutSize);
if (errorCode == GSASL_NEEDS_MORE)
{
nextGsaslMsg = string(tmpOut, tmpOutSize);
}
else
{
cout << "gsasl step 1 failed" << endl;
return 1;
}
gsasl_free(tmpOut);
OCTET_STRING_fromBuf(&authLogin->clientUserData,
nextGsaslMsg.c_str(),
nextGsaslMsg.length());
if (!sendMessage(socket, msg))
{
cout << "Init auth request failed" << endl;
return 1;
}
msg = receiveMessage(socket);
if (!msg || msg->present != PokerTHMessage_PR_authMessage)
{
cout << "Auth request failed" << endl;
return 1;
}
AuthMessage_t *netAuth = &msg->choice.authMessage;
AuthServerChallenge_t *netChallenge = &netAuth->choice.authServerChallenge;
string challengeStr = STL_STRING_FROM_OCTET_STRING(netChallenge->serverChallenge);
errorCode = gsasl_step(authSession, challengeStr.c_str(), challengeStr.size(), &tmpOut, &tmpOutSize);
if (errorCode == GSASL_NEEDS_MORE)
{
nextGsaslMsg = string(tmpOut, tmpOutSize);
}
else
{
cout << "gsasl step 2 failed" << endl;
return 1;
}
gsasl_free(tmpOut);
ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg);
msg = (PokerTHMessage_t *)calloc(1, sizeof(PokerTHMessage_t));
msg->present = PokerTHMessage_PR_authMessage;
AuthMessage_t *outAuth = &msg->choice.authMessage;
outAuth->present = AuthMessage_PR_authClientResponse;
AuthClientResponse_t *outResponse = &outAuth->choice.authClientResponse;
OCTET_STRING_fromBuf(&outResponse->clientResponse,
nextGsaslMsg.c_str(),
nextGsaslMsg.length());
if (!sendMessage(socket, msg))
{
cout << "Init auth response failed" << endl;
return 1;
}
msg = receiveMessage(socket);
if (!msg || msg->present != PokerTHMessage_PR_authMessage)
{
cout << "Auth response failed" << endl;
return 1;
}
}
}
// Receive init ack
msg = receiveMessage(socket);
if (!msg || msg->present != PokerTHMessage_PR_initAckMessage)
{
cout << "Init ack failed" << endl;
return 1;
}
ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg);
if (mode == 1)
{
cout << "Init.value " << perfTimer.elapsed().total_milliseconds() << endl;
}
perfTimer.restart();
// Send create game
msg = (PokerTHMessage_t *)calloc(1, sizeof(PokerTHMessage_t));
msg->present = PokerTHMessage_PR_joinGameRequestMessage;
JoinGameRequestMessage_t *netJoinGame = &msg->choice.joinGameRequestMessage;
string tmpGamePassword("blah123");
netJoinGame->password = OCTET_STRING_new_fromBuf(
&asn_DEF_UTF8String,
tmpGamePassword.c_str(),
tmpGamePassword.length());
netJoinGame->joinGameAction.present = joinGameAction_PR_joinNewGame;
JoinNewGame_t *joinNew = &netJoinGame->joinGameAction.choice.joinNewGame;
string tmpGameName("_perftest_do_not_join_" + username);
joinNew->gameInfo.netGameType = netGameType_normalGame;
joinNew->gameInfo.maxNumPlayers = 10;
joinNew->gameInfo.raiseIntervalMode.present = raiseIntervalMode_PR_raiseEveryHands;
joinNew->gameInfo.raiseIntervalMode.choice.raiseEveryHands = 5;
joinNew->gameInfo.endRaiseMode = endRaiseMode_keepLastBlind;
joinNew->gameInfo.proposedGuiSpeed = 5;
joinNew->gameInfo.delayBetweenHands = 6;
joinNew->gameInfo.playerActionTimeout = 10;
joinNew->gameInfo.endRaiseSmallBlindValue = 0;
joinNew->gameInfo.firstSmallBlind = 50;
joinNew->gameInfo.startMoney = 2000;
OCTET_STRING_fromBuf(&joinNew->gameInfo.gameName,
tmpGameName.c_str(),
tmpGameName.length());
if (!sendMessage(socket, msg)) if (!sendMessage(socket, msg))
{ {
cout << "Init guest failed" << endl; cout << "Create game failed" << endl;
return 1; return 1;
} }
} msg = NULL;
else // Receive join game ack
{ do
int errorCode = gsasl_client_start(authContext, "SCRAM-SHA-1", &authSession);
if (errorCode == GSASL_OK)
{ {
gsasl_property_set(authSession, GSASL_AUTHID, username.c_str());
gsasl_property_set(authSession, GSASL_PASSWORD, password.c_str());
netInit->login.present = login_PR_authenticatedLogin;
AuthenticatedLogin_t *authLogin = &netInit->login.choice.authenticatedLogin;
char *tmpOut;
size_t tmpOutSize;
string nextGsaslMsg;
errorCode = gsasl_step(authSession, NULL, 0, &tmpOut, &tmpOutSize);
if (errorCode == GSASL_NEEDS_MORE)
{
nextGsaslMsg = string(tmpOut, tmpOutSize);
}
else
{
cout << "gsasl step 1 failed" << endl;
return 1;
}
gsasl_free(tmpOut);
OCTET_STRING_fromBuf(&authLogin->clientUserData,
nextGsaslMsg.c_str(),
nextGsaslMsg.length());
if (!sendMessage(socket, msg))
{
cout << "Init auth request failed" << endl;
return 1;
}
msg = receiveMessage(socket);
if (!msg || msg->present != PokerTHMessage_PR_authMessage)
{
cout << "Auth request failed" << endl;
return 1;
}
AuthMessage_t *netAuth = &msg->choice.authMessage;
AuthServerChallenge_t *netChallenge = &netAuth->choice.authServerChallenge;
string challengeStr = STL_STRING_FROM_OCTET_STRING(netChallenge->serverChallenge);
errorCode = gsasl_step(authSession, challengeStr.c_str(), challengeStr.size(), &tmpOut, &tmpOutSize);
if (errorCode == GSASL_NEEDS_MORE)
{
nextGsaslMsg = string(tmpOut, tmpOutSize);
}
else
{
cout << "gsasl step 2 failed" << endl;
return 1;
}
gsasl_free(tmpOut);
ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg); ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg);
msg = (PokerTHMessage_t *)calloc(1, sizeof(PokerTHMessage_t));
msg->present = PokerTHMessage_PR_authMessage;
AuthMessage_t *outAuth = &msg->choice.authMessage;
outAuth->present = AuthMessage_PR_authClientResponse;
AuthClientResponse_t *outResponse = &outAuth->choice.authClientResponse;
OCTET_STRING_fromBuf(&outResponse->clientResponse,
nextGsaslMsg.c_str(),
nextGsaslMsg.length());
if (!sendMessage(socket, msg))
{
cout << "Init auth response failed" << endl;
return 1;
}
msg = receiveMessage(socket); msg = receiveMessage(socket);
if (!msg || msg->present != PokerTHMessage_PR_authMessage) if (!msg)
{ {
cout << "Auth response failed" << endl; cout << "Receive in lobby failed" << endl;
return 1; return 1;
} }
if (msg->present == PokerTHMessage_PR_errorMessage)
{
cout << "Received error" << endl;
return 1;
}
} while (msg->present != PokerTHMessage_PR_joinGameReplyMessage);
if (msg->choice.joinGameReplyMessage.joinGameResult.present != joinGameResult_PR_joinGameAck)
{
cout << "Join game ack failed" << endl;
return 1;
} }
}
// Receive init ack
msg = receiveMessage(socket);
if (!msg || msg->present != PokerTHMessage_PR_initAckMessage)
{
cout << "Init ack failed" << endl;
return 1;
}
ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg);
if (mode == 1)
{
cout << "Init.value " << perfTimer.elapsed().total_milliseconds() << endl;
}
perfTimer.restart();
// Send create game
msg = (PokerTHMessage_t *)calloc(1, sizeof(PokerTHMessage_t));
msg->present = PokerTHMessage_PR_joinGameRequestMessage;
JoinGameRequestMessage_t *netJoinGame = &msg->choice.joinGameRequestMessage;
string tmpGamePassword("blah123");
netJoinGame->password = OCTET_STRING_new_fromBuf(
&asn_DEF_UTF8String,
tmpGamePassword.c_str(),
tmpGamePassword.length());
netJoinGame->joinGameAction.present = joinGameAction_PR_joinNewGame;
JoinNewGame_t *joinNew = &netJoinGame->joinGameAction.choice.joinNewGame;
string tmpGameName("_perftest_do_not_join_" + username);
joinNew->gameInfo.netGameType = netGameType_normalGame;
joinNew->gameInfo.maxNumPlayers = 10;
joinNew->gameInfo.raiseIntervalMode.present = raiseIntervalMode_PR_raiseEveryHands;
joinNew->gameInfo.raiseIntervalMode.choice.raiseEveryHands = 5;
joinNew->gameInfo.endRaiseMode = endRaiseMode_keepLastBlind;
joinNew->gameInfo.proposedGuiSpeed = 5;
joinNew->gameInfo.delayBetweenHands = 6;
joinNew->gameInfo.playerActionTimeout = 10;
joinNew->gameInfo.endRaiseSmallBlindValue = 0;
joinNew->gameInfo.firstSmallBlind = 50;
joinNew->gameInfo.startMoney = 2000;
OCTET_STRING_fromBuf(&joinNew->gameInfo.gameName,
tmpGameName.c_str(),
tmpGameName.length());
if (!sendMessage(socket, msg))
{
cout << "Create game failed" << endl;
return 1;
}
msg = NULL;
// Receive join game ack
do
{
ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg); ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg);
msg = receiveMessage(socket);
if (!msg) if (mode == 1)
{ {
cout << "Receive in lobby failed" << endl; cout << "CreateGame.value " << perfTimer.elapsed().total_milliseconds() << endl;
return 1;
} }
if (msg->present == PokerTHMessage_PR_errorMessage) else
{ {
cout << "Received error" << endl; cout << "Success" << endl;
return 1;
} }
} while (msg->present != PokerTHMessage_PR_joinGameReplyMessage); perfTimer.restart();
if (msg->choice.joinGameReplyMessage.joinGameResult.present != joinGameResult_PR_joinGameAck) gsasl_done(authContext);
}
catch (...)
{ {
cout << "Join game ack failed" << endl; cout << "Exception caught" << endl;
return 1; return 1;
} }
ASN_STRUCT_FREE(asn_DEF_PokerTHMessage, msg);
if (mode == 1)
{
cout << "CreateGame.value " << perfTimer.elapsed().total_milliseconds() << endl;
}
else
{
cout << "Success" << endl;
}
perfTimer.restart();
gsasl_done(authContext);
return 0; return 0;
} }