adds welcome message to wsperf, lots of wsperf cleanup

This commit is contained in:
Peter Thorson
2012-03-10 11:06:22 -06:00
parent 7042091a61
commit 5e8ed9f64f

View File

@@ -1,191 +0,0 @@
/*
* Copyright (c) 2011, Peter Thorson. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* * Neither the name of the WebSocket++ Project nor the
* names of its contributors may be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL PETER THORSON BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/
#ifndef WSPERF_CASE__AUTOBAHN_HPP
#define WSPERF_CASE__AUTOBAHN_HPP
#include "case.hpp"
namespace wsperf {
// test class for 9.1.* and 9.2.*
class test_9_1_X : public case_handler {
public:
test_9_1_X(int minor, int subtest)
: m_minor(minor),
m_subtest(subtest)
{
// needs more C++11 intializer lists =(
m_test_sizes[0] = 65536;
m_test_sizes[1] = 262144;
m_test_sizes[2] = 1048576;
m_test_sizes[3] = 4194304;
m_test_sizes[4] = 8388608;
m_test_sizes[5] = 16777216;
}
void on_open(connection_ptr con) {
std::stringstream o;
o << "Test 9." << m_minor << "." << m_subtest;
m_name = o.str();
m_data.reserve(m_test_sizes[m_subtest-1]);
int timeout = 10000; // 10 sec
// extend timeout to 100 sec for the longer tests
if ((m_minor == 1 && m_subtest >= 3) || (m_minor == 2 && m_subtest >= 5)) {
timeout = 100000;
}
if (m_minor == 1) {
fill_utf8(m_data,m_test_sizes[m_subtest-1],true);
start(con,timeout);
con->send(m_data,websocketpp::frame::opcode::TEXT);
} else if (m_minor == 2) {
fill_binary(m_data,m_test_sizes[m_subtest-1],true);
start(con,timeout);
con->send(m_data,websocketpp::frame::opcode::BINARY);
} else {
std::cout << " has unknown definition." << std::endl;
}
}
void on_message(connection_ptr con,websocketpp::message::data_ptr msg) {
m_timer->cancel();
mark();
// Check whether the echoed data matches exactly
m_pass = ((msg->get_payload() == m_data) ? PASS : FAIL);
this->end(con);
}
private:
int m_minor;
int m_subtest;
int m_test_sizes[6];
std::string m_data;
};
// test class for 9.7.* and 9.8.*
class test_9_7_X : public case_handler {
public:
test_9_7_X(int minor, int subtest)
: m_minor(minor),
m_subtest(subtest),
m_acks(0)
{
// needs more C++11 intializer lists =(
m_test_sizes[0] = 0;
m_test_sizes[1] = 16;
m_test_sizes[2] = 64;
m_test_sizes[3] = 256;
m_test_sizes[4] = 1024;
m_test_sizes[5] = 4096;
m_test_timeouts[0] = 60000;
m_test_timeouts[1] = 60000;
m_test_timeouts[2] = 60000;
m_test_timeouts[3] = 120000;
m_test_timeouts[4] = 240000;
m_test_timeouts[5] = 480000;
m_iterations = 1000;
}
void on_open(connection_ptr con) {
std::stringstream o;
o << "Test 9." << m_minor << "." << m_subtest;
m_name = o.str();
// Get a message buffer from the connection
m_msg = con->get_data_message();
// reserve space in our local data buffer
m_data.reserve(m_test_sizes[m_subtest-1]);
// Reset message to appropriate opcode and fill local buffer with
// appropriate types of random data.
if (m_minor == 7) {
fill_utf8(m_data,m_test_sizes[m_subtest-1],true);
m_msg->reset(websocketpp::frame::opcode::TEXT);
} else if (m_minor == 8) {
fill_binary(m_data,m_test_sizes[m_subtest-1],true);
m_msg->reset(websocketpp::frame::opcode::BINARY);
} else {
std::cout << " has unknown definition." << std::endl;
return;
}
m_msg->set_payload(m_data);
// start test timer with 60-480 second timeout based on test length
start(con,m_test_timeouts[m_subtest-1]);
con->send(m_msg);
// send 1000 copies of prepared message
/*for (int i = 0; i < m_iterations; i++) {
con->send(msg);
}*/
}
void on_message(connection_ptr con, message_ptr msg) {
if (msg->get_payload() == m_data) {
m_acks++;
mark();
} else {
mark();
m_timer->cancel();
m_msg.reset();
this->end(con);
}
if (m_acks == m_iterations) {
m_pass = PASS;
mark();
m_timer->cancel();
m_msg.reset();
this->end(con);
} else {
con->send(m_msg);
}
}
private:
int m_minor;
int m_subtest;
int m_test_timeouts[6];
size_t m_test_sizes[6];
int m_iterations;
std::string m_data;
int m_acks;
message_ptr m_msg;
};
} // namespace wsperf
#endif // WSPERF_CASE__AUTOBAHN_HPP