2020-07-02 15:44:44 +08:00
|
|
|
#include "stdafx.h"
|
2022-08-31 08:37:27 +08:00
|
|
|
#include "../../util.h"
|
2020-07-02 15:44:44 +08:00
|
|
|
#include "https_request.h"
|
|
|
|
|
2023-09-03 21:20:35 +08:00
|
|
|
static bool __debug = false;
|
|
|
|
|
2020-07-02 15:44:44 +08:00
|
|
|
static void usage(const char* procname)
|
|
|
|
{
|
|
|
|
printf("usage: %s -h [help]\r\n"
|
2022-08-21 22:01:51 +08:00
|
|
|
" -f path of mbedtls or polarssl\r\n"
|
|
|
|
" -s server_addr [default: 127.0.0.1:1443]\r\n"
|
2020-07-02 15:44:44 +08:00
|
|
|
" -H host\r\n"
|
|
|
|
" -U url\r\n"
|
|
|
|
" -L data_length [default: 1024]\r\n"
|
|
|
|
" -c cocurrent [default: 1]\r\n"
|
|
|
|
" -S [use ssl, default: no]\r\n"
|
2023-09-03 21:20:35 +08:00
|
|
|
" -n count [default: 10]\r\n"
|
|
|
|
" -D [if show data]\r\n"
|
|
|
|
, procname);
|
2020-07-02 15:44:44 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
int main(int argc, char* argv[])
|
|
|
|
{
|
|
|
|
int ch, cocurrent = 1, count = 10;
|
|
|
|
bool use_ssl = false;
|
|
|
|
acl::string server_addr("127.0.0.1:1443"), host;
|
2022-08-21 22:01:51 +08:00
|
|
|
acl::string libpath;
|
2020-07-02 15:44:44 +08:00
|
|
|
acl::string url("/");
|
|
|
|
|
|
|
|
acl::acl_cpp_init();
|
|
|
|
acl::log::stdout_open(true);
|
|
|
|
|
2023-09-03 21:20:35 +08:00
|
|
|
while ((ch = getopt(argc, argv, "hf:s:c:n:SH:U:D")) > 0) {
|
2020-07-02 15:44:44 +08:00
|
|
|
switch (ch) {
|
|
|
|
case 'h':
|
|
|
|
usage(argv[0]);
|
|
|
|
return 0;
|
|
|
|
case 'f':
|
|
|
|
libpath = optarg;
|
|
|
|
break;
|
|
|
|
case 'c':
|
|
|
|
cocurrent = atoi(optarg);
|
|
|
|
break;
|
|
|
|
case 'n':
|
|
|
|
count = atoi(optarg);
|
|
|
|
break;
|
|
|
|
case 's':
|
|
|
|
server_addr = optarg;
|
|
|
|
break;
|
|
|
|
case 'S':
|
|
|
|
use_ssl = true;
|
|
|
|
break;
|
|
|
|
case 'H':
|
|
|
|
host = optarg;
|
|
|
|
break;
|
|
|
|
case 'U':
|
|
|
|
url = optarg;
|
|
|
|
break;
|
2023-09-03 21:20:35 +08:00
|
|
|
case 'D':
|
|
|
|
__debug = true;
|
|
|
|
break;
|
2020-07-02 15:44:44 +08:00
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
acl::sslbase_conf* ssl_conf = NULL;
|
|
|
|
if (libpath.find("mbedtls")) {
|
|
|
|
ssl_conf = new acl::mbedtls_conf(false);
|
|
|
|
acl::mbedtls_conf::set_libpath(libpath.c_str());
|
|
|
|
if (!acl::mbedtls_conf::load()) {
|
|
|
|
printf("load %s error\r\n", libpath.c_str());
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
} else if (libpath.find("polarssl")) {
|
|
|
|
ssl_conf = new acl::polarssl_conf;
|
|
|
|
acl::polarssl_conf::set_libpath(libpath);
|
|
|
|
if (!acl::polarssl_conf::load()) {
|
|
|
|
printf("load %s error\r\n", libpath.c_str());
|
|
|
|
return 1;
|
|
|
|
}
|
2022-08-22 22:03:38 +08:00
|
|
|
} else if (libpath.find("libssl")) {
|
|
|
|
const std::vector<acl::string>& libs = libpath.split2(";, \t");
|
|
|
|
if (libs.size() != 2) {
|
|
|
|
printf("invalid libpath=%s\r\n", libpath.c_str());
|
2023-09-03 21:20:35 +08:00
|
|
|
printf("usage: -f 'libcrypto.so;libssl.so'\r\n");
|
2022-08-22 22:03:38 +08:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
acl::openssl_conf::set_libpath(libs[0], libs[1]);
|
|
|
|
|
|
|
|
if (!acl::openssl_conf::load()) {
|
|
|
|
printf("load ssl error\r\n");
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2022-08-21 22:01:51 +08:00
|
|
|
ssl_conf = new acl::openssl_conf(false);
|
2020-07-02 15:44:44 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
if (host.empty()) {
|
|
|
|
host = server_addr;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct timeval begin;
|
|
|
|
gettimeofday(&begin, NULL);
|
|
|
|
|
|
|
|
std::list<https_request*> threads;
|
|
|
|
|
|
|
|
for (int i = 0; i < cocurrent; i++) {
|
|
|
|
https_request* thread = new https_request(
|
2023-09-03 21:20:35 +08:00
|
|
|
use_ssl ? ssl_conf : NULL, server_addr, host, url, __debug);
|
2020-07-02 15:44:44 +08:00
|
|
|
|
|
|
|
thread->set_detachable(false);
|
|
|
|
threads.push_back(thread);
|
|
|
|
thread->start();
|
|
|
|
}
|
|
|
|
|
|
|
|
std::list<https_request*>::iterator it = threads.begin();
|
|
|
|
for (; it != threads.end(); ++it) {
|
|
|
|
if ((*it)->wait(NULL)) {
|
|
|
|
printf("wait thread(%lu) ok\r\n", (*it)->thread_id());
|
|
|
|
} else {
|
|
|
|
printf("wait one thread(%lu) error\r\n",
|
|
|
|
(*it)->thread_id());
|
|
|
|
}
|
|
|
|
|
|
|
|
delete *it;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct timeval end;
|
|
|
|
gettimeofday(&end, NULL);
|
|
|
|
|
|
|
|
double spent = util::stamp_sub(&end, &begin);
|
|
|
|
printf("total: %d, spent: %.2f, speed: %.f\r\n", count,
|
|
|
|
spent, (count * 1000) / (spent > 1 ? spent : 1));
|
|
|
|
|
|
|
|
delete ssl_conf;
|
|
|
|
return 0;
|
|
|
|
}
|