mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-14 08:50:53 +08:00
259 lines
6.5 KiB
C++
259 lines
6.5 KiB
C++
// ssl_client.cpp : 定义控制台应用程序的入口点。
|
||
//
|
||
|
||
#include "stdafx.h"
|
||
#include "lib_acl.h"
|
||
#include <iostream>
|
||
#include "acl_cpp/acl_cpp_init.hpp"
|
||
#include "acl_cpp/stdlib/log.hpp"
|
||
#include "acl_cpp/http/http_header.hpp"
|
||
#include "acl_cpp/stdlib/string.hpp"
|
||
#include "acl_cpp/stream/socket_stream.hpp"
|
||
#include "acl_cpp/stream/polarssl_io.hpp"
|
||
#include "acl_cpp/stream/polarssl_conf.hpp"
|
||
#include "acl_cpp/http/http_client.hpp"
|
||
|
||
static acl::polarssl_conf* __ssl_conf;
|
||
|
||
static void test0(int i)
|
||
{
|
||
acl::socket_stream client;
|
||
acl::string addr("127.0.0.1:441");
|
||
if (!client.open(addr.c_str(), 60, 60)) {
|
||
std::cout << "connect " << addr.c_str()
|
||
<< " error!" << std::endl;
|
||
return;
|
||
}
|
||
|
||
acl::polarssl_io* ssl = new acl::polarssl_io(*__ssl_conf, false);
|
||
if (client.setup_hook(ssl) == ssl) {
|
||
std::cout << "open ssl " << addr.c_str()
|
||
<< " error!" << std::endl;
|
||
ssl->destroy();
|
||
return;
|
||
}
|
||
|
||
char line[1024];
|
||
memset(line, 'x', sizeof(line));
|
||
line[1023] = 0;
|
||
line[1022] = '\n';
|
||
if (client.write(line, strlen(line)) == -1) {
|
||
std::cout << "write to " << addr.c_str()
|
||
<< " error!" << std::endl;
|
||
return;
|
||
}
|
||
|
||
size_t n = sizeof(line);
|
||
if (!client.gets(line, &n)) {
|
||
std::cout << "gets from " << addr.c_str()
|
||
<< " error!" << std::endl;
|
||
return;
|
||
}
|
||
if (i < 10) {
|
||
std::cout << ">>gets: " << line << std::endl;
|
||
}
|
||
}
|
||
|
||
static void test1(const char* domain, int port, bool use_gzip, bool use_ssl)
|
||
{
|
||
// 连接 WEB 服务器过程
|
||
|
||
acl::string addr;
|
||
addr << domain << ':' << port;
|
||
|
||
acl::socket_stream client;
|
||
if (!client.open(addr.c_str(), 60, 60)) {
|
||
std::cout << "connect " << addr.c_str()
|
||
<< " error!" << std::endl;
|
||
return;
|
||
}
|
||
|
||
// 如果使用 SSL 方式，则进行 SSL 握手过程
|
||
if (use_ssl) {
|
||
acl::polarssl_io* ssl = new acl::polarssl_io(*__ssl_conf, false);
|
||
if (client.setup_hook(ssl) == ssl) {
|
||
std::cout << "open ssl client " << addr.c_str()
|
||
<< " error!" << std::endl;
|
||
ssl->destroy();
|
||
return;
|
||
}
|
||
}
|
||
|
||
// 构建 HTTP 请求头
|
||
acl::http_header header;
|
||
header.set_url("/")
|
||
.set_host(domain)
|
||
.accept_gzip(use_gzip)
|
||
.set_keep_alive(false);
|
||
// mail.126.com 比较土鳖，有时客户端要求非压缩数据其也会返回压缩数据，所以此处
|
||
// 强制要求非压缩数据
|
||
if (!use_gzip) {
|
||
header.add_entry("Accept-Encoding", "text/plain");
|
||
}
|
||
|
||
acl::string request;
|
||
header.build_request(request);
|
||
|
||
std::cout << "request(len: " << request.length() << "):" << std::endl;
|
||
std::cout << "----------------------------------------" << std::endl;
|
||
std::cout << request.c_str();
|
||
std::cout << "----------------------------------------" << std::endl;
|
||
|
||
// 发送 HTTP GET 请求头
|
||
if (!client.write(request)) {
|
||
std::cout << "write to " << addr.c_str() <<
|
||
" error!" << std::endl;
|
||
return;
|
||
}
|
||
|
||
// 读取 HTTP 数据体过程
|
||
|
||
char buf[8192];
|
||
size_t size;
|
||
int ret;
|
||
|
||
while (true) {
|
||
size = sizeof(buf) - 1;
|
||
if ((ret = client.read(buf, size, false)) == -1) {
|
||
std::cout << "read over!" << std::endl;
|
||
break;
|
||
}
|
||
buf[ret] = 0;
|
||
std::cout << buf;
|
||
}
|
||
}
|
||
|
||
static void test2(const char* domain, int port, bool use_gzip, bool use_ssl)
|
||
{
|
||
// 连接 WEB 服务器过程
|
||
|
||
acl::string addr;
|
||
addr << domain << ':' << port;
|
||
|
||
acl::http_client client;
|
||
if (!client.open(addr.c_str(), 60, 60, use_gzip)) {
|
||
std::cout << "connect " << addr.c_str()
|
||
<< " error!" << std::endl;
|
||
return;
|
||
}
|
||
|
||
if (use_ssl) {
|
||
// 创建 SSL 对象并与网络客户端连接流绑定，当流对象被释放前该 SSL 对象
|
||
// 将由流对象内部通过调用 stream_hook::destroy() 释放
|
||
acl::polarssl_io* ssl = new acl::polarssl_io(*__ssl_conf, false);
|
||
if (client.get_stream().setup_hook(ssl) == ssl) {
|
||
std::cout << "open ssl client " << addr.c_str()
|
||
<< " error!" << std::endl;
|
||
ssl->destroy();
|
||
return;
|
||
}
|
||
}
|
||
|
||
// 构建 HTTP 请求头
|
||
|
||
acl::http_header header;
|
||
header.set_url("/")
|
||
.set_host(domain)
|
||
.accept_gzip(use_gzip)
|
||
.add_entry("User-Agent", "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:31.0) Gecko/20100101 Firefox/31.0")
|
||
.add_entry("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8")
|
||
.add_entry("Accept-Languge", "zh-cn,en;q=0.5")
|
||
.add_entry("Pragma", "no-cache")
|
||
.add_entry("Cache-Control", "no-cache");
|
||
|
||
acl::string request;
|
||
header.build_request(request);
|
||
|
||
std::cout << "request:" << std::endl;
|
||
std::cout << "----------------------------------------" << std::endl;
|
||
std::cout << request.c_str();
|
||
std::cout << "----------------------------------------" << std::endl;
|
||
|
||
// 发送 HTTP GET 请求头
|
||
|
||
if (!client.write_head(header)) {
|
||
std::cout << "write to " << addr.c_str()
|
||
<< " error!" << std::endl;
|
||
return;
|
||
}
|
||
|
||
// 读取 HTTP 响应头
|
||
if (!client.read_head()) {
|
||
std::cout << "read http respond header error!" << std::endl;
|
||
return;
|
||
}
|
||
|
||
client.get_respond_head(&request);
|
||
std::cout << "respond header:" << std::endl;
|
||
std::cout << request.c_str();
|
||
|
||
// 读取服务器响应的 HTTP 数据体过程
|
||
|
||
char buf[8192];
|
||
size_t size;
|
||
int ret;
|
||
|
||
while (true) {
|
||
size = sizeof(buf) - 1;
|
||
if ((ret = client.read_body(buf, size)) <= 0) {
|
||
std::cout << "read over!" << std::endl;
|
||
break;
|
||
}
|
||
buf[ret] = 0;
|
||
std::cout << buf;
|
||
}
|
||
}
|
||
|
||
int main(int argc, char* argv[])
|
||
{
|
||
acl::acl_cpp_init();
|
||
acl::log::stdout_open(true);
|
||
|
||
#if defined(_WIN32) || defined(_WIN64)
|
||
acl::polarssl_conf::set_libpath("libpolarssl.dll");
|
||
#else
|
||
acl::polarssl_conf::set_libpath("../libpolarssl.so");
|
||
#endif
|
||
acl::polarssl_conf::load();
|
||
|
||
__ssl_conf = new acl::polarssl_conf;
|
||
|
||
int n = 1;
|
||
if (argc >= 2) {
|
||
n = atoi(argv[1]);
|
||
}
|
||
if (n <= 0) {
|
||
n = 100;
|
||
}
|
||
ACL_METER_TIME("---------- begin ----------");
|
||
for (int i = 0; i < 0; i++) {
|
||
test0(i);
|
||
}
|
||
ACL_METER_TIME("---------- end ----------");
|
||
|
||
// 126 的 SSL 传输时当 HTTP 请求头中的 Host 值为 mail.126.com:443 时其 nginx
|
||
// 会报错，只能是：Host: mail.126.com，土鳖
|
||
|
||
test1("mail.126.com", 443, false, true);
|
||
printf("\r\nenter any key to continue ..."); fflush(stdout); getchar();
|
||
|
||
test2("mail.126.com", 443, false, true);
|
||
printf("\r\nenter any key to continue ..."); fflush(stdout); getchar();
|
||
|
||
test2("mail.qq.com", 443, false, true);
|
||
printf("\r\nenter any key to continue ..."); fflush(stdout); getchar();
|
||
|
||
test2("mail.sohu.com", 443, false, true);
|
||
printf("\r\nenter any key to continue ..."); fflush(stdout); getchar();
|
||
|
||
test2("mail.sina.com.cn", 443, false, true);
|
||
printf("\r\nenter any key to continue ..."); fflush(stdout); getchar();
|
||
|
||
test2("127.0.0.1", 2443, false, true);
|
||
|
||
printf("Over, enter any key to exit!\n");
|
||
getchar();
|
||
delete __ssl_conf;
|
||
return (0);
|
||
}
|