mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-16 01:40:52 +08:00
9697f95b8f
This reverts commit 15d999759e
.
190 lines
4.4 KiB
C++
190 lines
4.4 KiB
C++
// http_aio.cpp : 定义控制台应用程序的入口点。
|
||
//
|
||
|
||
#include "stdafx.h"
|
||
#include <assert.h>
|
||
|
||
typedef struct {
|
||
ACL_ASTREAM *stream;
|
||
HTTP_HDR_REQ *hdr_req;
|
||
HTTP_HDR_RES *hdr_res;
|
||
HTTP_RES *http_res;
|
||
} CTX;
|
||
|
||
static char __server_addr[128];
|
||
static bool __stop = false;
|
||
|
||
static int on_close(ACL_ASTREAM *stream acl_unused, void *context)
|
||
{
|
||
CTX *ctx = (CTX *) context;
|
||
|
||
http_hdr_req_free(ctx->hdr_req);
|
||
|
||
// 当读到完整 HTTP 响应头时,HTTP 响应体对象应该也创建,当释放响应体
|
||
// 对象时,响应头对象会一起被释放;如果在读响应头时出错,则响应体对象
|
||
// 并未创建,所以只需释放响应头对象
|
||
if (ctx->http_res) {
|
||
http_res_free(ctx->http_res);
|
||
} else if (ctx->hdr_res) {
|
||
http_hdr_res_free(ctx->hdr_res);
|
||
}
|
||
|
||
acl_myfree(ctx);
|
||
|
||
printf("%s(%d)\n", __FUNCTION__, __LINE__);
|
||
__stop = true;
|
||
return 0;
|
||
}
|
||
|
||
static int on_timeout(ACL_ASTREAM *stream acl_unused, void *context acl_unused)
|
||
{
|
||
printf("%s(%d)\n", __FUNCTION__, __LINE__);
|
||
return 0;
|
||
}
|
||
|
||
static int read_respond_body_ready(int status, char *data, int dlen,
|
||
void *arg acl_unused)
|
||
{
|
||
ACL_VSTRING *buf = acl_vstring_alloc(256);
|
||
|
||
acl_vstring_memcpy(buf, data, dlen);
|
||
ACL_VSTRING_TERMINATE(buf);
|
||
printf("status=%d >>%s", status, acl_vstring_str(buf));
|
||
acl_vstring_free(buf);
|
||
|
||
return 0;
|
||
}
|
||
|
||
static int on_http_hdr(int status, void *arg)
|
||
{
|
||
CTX *ctx = (CTX*) arg;
|
||
|
||
printf("%s: status=%d\r\n", __FUNCTION__, status);
|
||
|
||
http_hdr_print(&ctx->hdr_res->hdr, "---respond---");
|
||
|
||
// 构建 HTTP 响应体
|
||
ctx->http_res = http_res_new(ctx->hdr_res);
|
||
|
||
// 异步读取 HTTP 服务端的响应体
|
||
http_res_body_get_async(ctx->http_res, ctx->stream,
|
||
read_respond_body_ready, ctx, 0);
|
||
|
||
return 0;
|
||
}
|
||
|
||
static int on_connect(ACL_ASTREAM *stream, void *context)
|
||
{
|
||
const char *addr = (const char*) context;
|
||
|
||
if (stream == NULL) {
|
||
int err = acl_last_error();
|
||
printf("connect %s failed, errno=%d, %s\r\n", addr, err,
|
||
err < 0 ? acl_dns_serror(err) : acl_last_serror());
|
||
__stop = true;
|
||
return -1;
|
||
}
|
||
|
||
printf(">>>> connect %s ok!\r\n", addr);
|
||
|
||
CTX *ctx = (CTX*) acl_mycalloc(1, sizeof(CTX));
|
||
ctx->stream = stream;
|
||
|
||
// 构建 HTTP 请求头
|
||
ctx->hdr_req = http_hdr_req_create("/", "GET", "HTTP/1.1");
|
||
// 设置短连接模式
|
||
http_hdr_entry_replace(&ctx->hdr_req->hdr, "Connection", "Close", 1);
|
||
// 设置内容类型
|
||
http_hdr_put_str(&ctx->hdr_req->hdr, "Content-Type", "text/plain");
|
||
// 设置主机字段
|
||
http_hdr_put_str(&ctx->hdr_req->hdr, "Host", __server_addr);
|
||
|
||
// 设置读超时回调函数
|
||
acl_aio_add_timeo_hook(stream, on_timeout, ctx);
|
||
// 设置关闭回调函数
|
||
acl_aio_add_close_hook(stream, on_close, ctx);
|
||
|
||
// 创建动态内存并将 HTTP 请求头拷贝至其中
|
||
ACL_VSTRING *buf = acl_vstring_alloc(1024);
|
||
http_hdr_build_request(ctx->hdr_req, buf);
|
||
|
||
// 发送 HTTP 请求
|
||
acl_aio_writen(stream, acl_vstring_str(buf), ACL_VSTRING_LEN(buf));
|
||
|
||
// 释放动态内存
|
||
acl_vstring_free(buf);
|
||
|
||
// 创建 HTTP 响应对象
|
||
ctx->hdr_res = http_hdr_res_new();
|
||
|
||
// 异步读取服务端响应的 HTTP 头
|
||
http_hdr_res_get_async(ctx->hdr_res, stream, on_http_hdr, ctx, 0);
|
||
|
||
return 0;
|
||
}
|
||
|
||
static void usage(const char *procname)
|
||
{
|
||
printf("usage: %s -h[help]\r\n"
|
||
" -s server_addr\r\n"
|
||
" -N name_server\r\n"
|
||
" -t connect_timeout\r\n"
|
||
" -T dns_timeout\r\n",
|
||
procname);
|
||
}
|
||
|
||
int main(int argc, char* argv[])
|
||
{
|
||
ACL_AIO *aio = acl_aio_create(ACL_EVENT_SELECT);
|
||
char name_server[128];
|
||
int ch, connect_timeout = 5, dns_timeout = 1;
|
||
|
||
acl_aio_set_keep_read(aio, 1);
|
||
|
||
snprintf(__server_addr, sizeof(__server_addr), "www.baidu.com:80");
|
||
snprintf(name_server, sizeof(name_server), "8.8.8.8:53");
|
||
|
||
while ((ch = getopt(argc, argv, "hs:N:t:T:")) > 0) {
|
||
switch (ch) {
|
||
case 'h':
|
||
usage(argv[0]);
|
||
return 0;
|
||
case 's':
|
||
snprintf(__server_addr, sizeof(__server_addr), "%s", optarg);
|
||
break;
|
||
case 'N':
|
||
snprintf(name_server, sizeof(name_server), "%s", optarg);
|
||
break;
|
||
case 't':
|
||
connect_timeout = atoi(optarg);
|
||
break;
|
||
case 'T':
|
||
dns_timeout = atoi(optarg);
|
||
break;
|
||
default:
|
||
usage(argv[0]);
|
||
return 1;
|
||
}
|
||
}
|
||
|
||
// 设置域名解析服务器地址
|
||
acl_aio_set_dns(aio, name_server, dns_timeout);
|
||
|
||
// 异步连接指定 WEB 服务器
|
||
if (acl_aio_connect_addr(aio, __server_addr, connect_timeout,
|
||
on_connect, __server_addr) == -1) {
|
||
|
||
printf("connect %s error\r\n", __server_addr);
|
||
return 1;
|
||
}
|
||
|
||
// 异步事件循环
|
||
while (!__stop) {
|
||
acl_aio_loop(aio);
|
||
}
|
||
|
||
acl_aio_free(aio);
|
||
return 0;
|
||
}
|
||
|