test/acl
1
0
Fork 0
mirror of https://gitee.com/acl-dev/acl.git synced 2024-12-15 09:20:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
41a5f9eeaa
acl/app/jaws/global/forward.c

290 lines
8.1 KiB
C
Raw Normal View History

Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
#include "lib_acl.h"
整理了一些文件格式
2014-11-19 00:25:21 +08:00
#include "conn_cache.h"
#include "sys_patch.h"
#include "service.h"
#ifdef ACL_BCB_COMPILER
#pragma hdrstop
#endif
static int connect_close_callback(ACL_ASTREAM *astream, void *context);
static int connect_timeout_callback(ACL_ASTREAM *astream, void *context);
static int connect_callback(ACL_ASTREAM *server, void *context);
void forward_complete(CLIENT_ENTRY *entry)
{
if (entry->server) {
acl_aio_iocp_close(entry->server);
}
if (entry->client) {
acl_aio_iocp_close(entry->client);
}
}
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* <20><><EFBFBD>Դ<EFBFBD><D4B4><EFBFBD><EFBFBD>ӳ<EFBFBD><D3B3>л<EFBFBD><D0BB><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
static ACL_ASTREAM *peek_server_conn(CLIENT_ENTRY *entry)
{
ACL_ASTREAM *stream;
char addr[64];
int i;
if (entry->service->conn_cache == NULL)
return (NULL);
for (i = entry->ip_idx; i < entry->dns_ctx.ip_cnt; i++) {
snprintf(addr, sizeof(addr), "%s:%d", entry->dns_ctx.ip[i],
entry->dns_ctx.port[i] > 0
? entry->dns_ctx.port[i] : entry->server_port);
stream = conn_cache_get_stream(entry->service->conn_cache, addr, NULL);
if (stream != NULL)
return (stream);
}
return (NULL);
}
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* ȡ<><C8A1>ij<EFBFBD><C4B3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> IP:PORT <20><>ַ */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
static const char *next_server_addr(CLIENT_ENTRY *entry, char *buf, size_t size)
{
const char *myname = "next_server_addr";
SERVICE *service = entry->service;
int i;
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* <20><><EFBFBD>Ŷ<EFBFBD><C5B6><EFBFBD>һ<EFBFBD><D2BB> */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
if (entry->ip_ntry++ > entry->dns_ctx.ip_cnt) {
acl_msg_error("%s(%d): domain(%s), ip_ntry(%d) >= ip_cnt(%d)",
myname, __LINE__, entry->domain_key,
entry->ip_ntry, entry->dns_ctx.ip_cnt);
return (NULL);
}
if (entry->ip_idx == entry->dns_ctx.ip_cnt)
entry->ip_idx = 0;
for (i = entry->ip_idx; i < entry->dns_ctx.ip_cnt; i++) {
if (service->bind_ip_list) {
snprintf(buf, size, "%s@%s:%d",
service->bind_ip_list[service->bind_ip_index++],
entry->dns_ctx.ip[i],
entry->dns_ctx.port[i] > 0
? entry->dns_ctx.port[i] : entry->server_port);
if (service->bind_ip_list[service->bind_ip_index] == NULL)
service->bind_ip_index = 0;
} else {
snprintf(buf, size, "%s:%d", entry->dns_ctx.ip[i],
entry->dns_ctx.port[i] > 0
? entry->dns_ctx.port[i] : entry->server_port);
}
entry->ip_idx++;
acl_debug(23, 2) ("%s(%d): domain(%s), addr(%s)",
myname, __LINE__, entry->domain_key, buf);
return (buf);
}
return (NULL);
}
static ACL_ASTREAM *forward_connect_next(CLIENT_ENTRY *entry)
{
const char* myname = "forward_connect_next";
SERVICE *service = entry->service;
ACL_ASTREAM* server;
int i = 0;
while (1) {
if (next_server_addr(entry, entry->domain_addr,
sizeof(entry->domain_addr)) == NULL)
break;
server = acl_aio_connect(entry->client->aio,
entry->domain_addr, service->conn_timeout);
if (server != NULL)
return (server);
acl_msg_error("%s: connect server addr(%s), domain(%s), i=%d",
myname, entry->domain_addr, entry->domain_key, i);
i++;
}
return (NULL);
}
static int connect_close_callback(ACL_ASTREAM *astream, void *context)
{
const char* myname = "connect_close_callback";
CLIENT_ENTRY *entry = (CLIENT_ENTRY *) context;
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* ж<>ػص<D8BB><D8B5><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֹ<EFBFBD><D6B9><EFBFBD>ظ<EFBFBD><D8B8><EFBFBD><EFBFBD><EFBFBD> */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
acl_aio_ctl(astream,
ACL_AIO_CTL_TIMEO_HOOK_DEL, connect_timeout_callback, entry,
ACL_AIO_CTL_END);
if (entry->flag_has_replied == 1)
return (-1);
if (entry->ip_idx < entry->dns_ctx.ip_cnt) {
ACL_ASTREAM* server;
acl_debug(23, 1) ("%s(%d): begin to connect next ip(%s:%d)",
myname, __LINE__, entry->dns_ctx.ip[entry->ip_idx],
entry->server_port);
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* <20>Ͽ<EFBFBD><CFBF><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>˵<EFBFBD><CBB5><EFBFBD><EFBFBD>ӣ<EFBFBD><D3A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>˵<EFBFBD><CBB5><EFBFBD><EFBFBD><EFBFBD>
* XXX: <EFBFBD><EFBFBD>Ϊ<EFBFBD>ú<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>һЩ<EFBFBD>رջص<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>֪<EFBFBD>Ƿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ijЩ<EFBFBD>ڴ<EFBFBD>й©<EFBFBD><EFBFBD>
* ע<EFBFBD><EFBFBD><EFBFBD>˴<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>رշ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ӣ<EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Լ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ر<EFBFBD>
整理了一些文件格式
2014-11-19 00:25:21 +08:00
*/
if (entry->server && client_entry_detach(entry, acl_aio_vstream(entry->server))) {
acl_debug(23, 1) ("%s(%d): entry's freed", myname, __LINE__);
return (-1);
}
server = forward_connect_next(entry);
if (server == NULL)
goto CONNECT_ERROR;
client_entry_set_server(entry, server);
acl_aio_ctl(server,
ACL_AIO_CTL_CONNECT_HOOK_ADD, connect_callback,
ACL_AIO_CTL_CLOSE_HOOK_ADD, connect_close_callback, entry,
ACL_AIO_CTL_TIMEO_HOOK_ADD, connect_timeout_callback, entry,
ACL_AIO_CTL_CTX, entry,
ACL_AIO_CTL_END);
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* ͨ<><CDA8><EFBFBD><EFBFBD><EFBFBD><EFBFBD>-1<><31>ʹ<EFBFBD>첽<EFBFBD><ECB2BD><EFBFBD><EFBFBD><EFBFBD>ܹرշ<D8B1><D5B7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
return (-1);
}
CONNECT_ERROR:
entry->tm.connect = time(NULL) - entry->tm.stamp;
if (entry->ip_idx <= 0)
acl_debug(23, 0) ("%s(%d): internal error, ip_idx=%d, domain(%s:%d)",
myname, __LINE__, entry->ip_idx,
entry->domain_key, entry->server_port);
else
acl_debug(23, 0) ("%s(%d): connect error, addr(%s:%d), domain(%s)",
myname, __LINE__,
entry->dns_ctx.ip[entry->ip_idx - 1],
entry->server_port,
entry->domain_key);
entry->flag_has_replied = 1;
if (entry->connect_error_fn)
entry->connect_error_fn(entry);
return (-1);
}
static int connect_timeout_callback(ACL_ASTREAM *astream, void *context)
{
const char* myname = "connect_timeout_callback";
CLIENT_ENTRY *entry = (CLIENT_ENTRY *) context;
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* ж<>ػص<D8BB><D8B5><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֹ<EFBFBD><D6B9><EFBFBD>ظ<EFBFBD><D8B8><EFBFBD><EFBFBD><EFBFBD> */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
acl_aio_ctl(astream,
ACL_AIO_CTL_TIMEO_HOOK_DEL, connect_timeout_callback, entry,
ACL_AIO_CTL_END);
if (entry->flag_has_replied == 1)
return (-1);
if (entry->ip_idx < entry->dns_ctx.ip_cnt) {
ACL_ASTREAM *server;
acl_debug(23, 1) ("%s(%d): begin to connect next ip(%s:%d)",
myname, __LINE__, entry->dns_ctx.ip[entry->ip_idx],
entry->server_port);
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* <20>Ͽ<EFBFBD><CFBF><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>˵<EFBFBD><CBB5><EFBFBD><EFBFBD>ӣ<EFBFBD><D3A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>˵<EFBFBD><CBB5><EFBFBD><EFBFBD><EFBFBD>
* XXX: <EFBFBD><EFBFBD>Ϊ<EFBFBD>ú<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>һЩ<EFBFBD>رջص<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>֪<EFBFBD>Ƿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ijЩ<EFBFBD>ڴ<EFBFBD>й©<EFBFBD><EFBFBD>
* ע<EFBFBD><EFBFBD><EFBFBD>˴<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>رշ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ӣ<EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Լ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ر<EFBFBD>
整理了一些文件格式
2014-11-19 00:25:21 +08:00
*/
if (client_entry_detach(entry, acl_aio_vstream(entry->server)) == 1) {
acl_debug(3, 1) ("%s(%d): entry is freed", myname, __LINE__);
return (-1);
}
server = forward_connect_next(entry);
if (server == NULL)
goto CONNECT_ERROR;
client_entry_set_server(entry, server);
acl_aio_ctl(server,
ACL_AIO_CTL_CONNECT_HOOK_ADD, connect_callback,
ACL_AIO_CTL_CLOSE_HOOK_ADD, connect_close_callback, entry,
ACL_AIO_CTL_TIMEO_HOOK_ADD, connect_timeout_callback, entry,
ACL_AIO_CTL_CTX, entry,
ACL_AIO_CTL_END);
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* ͨ<><CDA8><EFBFBD><EFBFBD><EFBFBD><EFBFBD>-1<><31>ʹ<EFBFBD>첽<EFBFBD><ECB2BD><EFBFBD><EFBFBD><EFBFBD>ܹرշ<D8B1><D5B7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
return (-1);
}
CONNECT_ERROR:
entry->tm.connect = time(NULL) - entry->tm.stamp;
if (entry->ip_idx <= 0)
acl_debug(23, 0) ("%s(%d): internal error, ip_idx=0, domain(%s:%d)",
myname, __LINE__, entry->domain_key, entry->server_port);
else
acl_debug(23, 0) ("%s(%d): connect timeout, addr(%s:%d)",
myname, __LINE__,
entry->dns_ctx.ip[entry->ip_idx - 1],
entry->server_port);
entry->flag_has_replied = 1;
if (entry->connect_timeout_fn)
entry->connect_timeout_fn(entry);
return (-1);
}
static int connect_callback(ACL_ASTREAM *server, void *context)
{
CLIENT_ENTRY *entry = (CLIENT_ENTRY*) context;
entry->tm.connect = time(NULL) - entry->tm.stamp;
return (entry->connect_notify_fn(entry));
}
void forward_start(CLIENT_ENTRY *entry)
{
const char *myname = "forward_start";
ACL_ASTREAM *server;
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* <20>ȴ<EFBFBD><C8B4><EFBFBD><EFBFBD>ӳ<EFBFBD><D3B3>г<EFBFBD><D0B3><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
server = peek_server_conn(entry);
if (server == NULL) {
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ӳ<EFBFBD><D3B3><EFBFBD>û<EFBFBD>п<EFBFBD><D0BF><EFBFBD><EFBFBD>õ<EFBFBD><C3B5><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʼ<EFBFBD><CABC><EFBFBD>ӷ<EFBFBD><D3B7><EFBFBD><EFBFBD><EFBFBD> */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
server = forward_connect_next(entry);
if (server == NULL) {
acl_msg_error("%s: connect server_addr(%s:%d) error(%s)",
myname, entry->domain_key, entry->server_port,
acl_last_serror());
#if 1
if (entry->connect_error_fn)
entry->connect_error_fn(entry);
#else
forward_complete(entry);
#endif
return;
}
entry->flag_conn_reuse = 0;
client_entry_set_server(entry, server);
acl_aio_ctl(server,
ACL_AIO_CTL_CONNECT_HOOK_ADD, connect_callback,
ACL_AIO_CTL_CTX, entry,
ACL_AIO_CTL_CLOSE_HOOK_ADD, connect_close_callback, entry,
ACL_AIO_CTL_TIMEO_HOOK_ADD, connect_timeout_callback, entry,
ACL_AIO_CTL_END);
} else {
Revert "All source code have been changed to use utf8 charset with BOM header!" This reverts commit 15d999759eabbf2568053d9d864e06ad876a5593.
2019-07-28 10:31:56 +08:00
/* <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ӳ<EFBFBD><D3B3>е<EFBFBD><D0B5><EFBFBD><EFBFBD><EFBFBD> */
整理了一些文件格式
2014-11-19 00:25:21 +08:00
entry->flag_conn_reuse = 1;
client_entry_set_server(entry, server);
acl_aio_ctl(server, ACL_AIO_CTL_CTX, entry, ACL_AIO_CTL_END);
connect_callback(server, entry);
}
}
Reference in New Issue Copy Permalink