[英]Boost::Asio : Problem with async_read
我正在編寫一個簡單的 ProxyServer 來分析包並將它們發送到另一個服務器實例,例如:
客戶端 -> MyProxy -> SQLServer ->
客戶端 <- MyProxy <- SQLServer <-
它應該在無限循環中運行。 我現在的問題是代理似乎丟失了包,有時甚至掛起。 當我添加大量調試信息(寫入控制台)時,ProxyServer 更加穩定。 看來 ProxyServer 太快了.. :-)
我很確定我做錯了什么,這是我的 session class 的代碼(代碼來自 Boost::Asio 示例)。
#include "session.h"
#include <iostream>
using namespace std;
session::session(boost::asio::io_service& io_service)
: socket_(io_service)
, sqlsocket_(io_service)
, io_service_(io_service)
, resolver(io_service)
{
cout << "session::session()" << endl;
}
session::~session()
{
cout << "session::~session()" << endl;
cout << "closing session ..." << endl;
}
tcp::socket& session::socket()
{
return socket_;
}
void session::start()
{
cout << "session::start()" << endl;
cout << "starting session ..." << endl;
// connect to the sqlserver database
tcp::resolver::query query("192.168.1.50", "1317");
tcp::resolver::iterator endpoint_iterator = resolver.resolve(query);
tcp::endpoint endpoint = *endpoint_iterator;
sqlsocket_.async_connect(endpoint,
boost::bind(&session::handle_sqlserver_connect, this,
boost::asio::placeholders::error, ++endpoint_iterator));
// TODO: connect to the connector
}
void session::handle_read(const boost::system::error_code& error,
size_t bytes_transferred)
{
cout << "session::handle_read()" << endl;
if (!error)
{
cout << "session::handle_read() (read: "
<< bytes_transferred << ")"
<< endl;
boost::asio::async_write(sqlsocket_,
boost::asio::buffer(data_, bytes_transferred),
boost::bind(&session::handle_sqlserver_write, this,
boost::asio::placeholders::error, bytes_transferred));
}
else
{
delete this;
}
}
void session::handle_sqlserver_read(const boost::system::error_code& error,
size_t bytes_transferred)
{
cout << "session::handle_sqlserver_read()" << endl;
if (!error)
{
cout << "session::handle_sqlserver_read() (read: "
<< bytes_transferred << ")"
<< endl;
boost::asio::async_write(socket_,
boost::asio::buffer(data_, bytes_transferred),
boost::bind(&session::handle_write, this,
boost::asio::placeholders::error, bytes_transferred));
}
else
{
delete this;
}
}
void session::handle_write(const boost::system::error_code& error,
size_t bytes_transferred)
{
static int count = 0;
cout << ++count << ". session::handle_write()" << endl;
if (!error)
{
cout << "session::handle_write() (read: "
<< bytes_transferred << ")"
<< endl;
socket_.async_read_some(boost::asio::buffer(data_, max_length),
boost::bind(&session::handle_read, this,
boost::asio::placeholders::error,
boost::asio::placeholders::bytes_transferred));
}
else
{
delete this;
}
}
void session::handle_sqlserver_write(const boost::system::error_code& error,
size_t bytes_transferred)
{
cout << "session::handle_sqlserver_write()" << endl;
if (!error)
{
cout << "session::handle_sqlserver_write() (read: "
<< bytes_transferred << ")"
<< endl;
sqlsocket_.async_read_some(boost::asio::buffer(data_, max_length),
boost::bind(&session::handle_sqlserver_read, this,
boost::asio::placeholders::error,
boost::asio::placeholders::bytes_transferred));
}
else
{
delete this;
}
}
void session::handle_sqlserver_connect(const boost::system::error_code& error,
tcp::resolver::iterator endpoint_iterator)
{
cout << "session::handle_sqlserver_connect()" << endl;
if (!error)
{
socket_.async_read_some(boost::asio::buffer(data_, max_length),
boost::bind(&session::handle_read, this,
boost::asio::placeholders::error,
boost::asio::placeholders::bytes_transferred));
}
else if (endpoint_iterator != tcp::resolver::iterator())
{
sqlsocket_.close();
tcp::endpoint endpoint = *endpoint_iterator;
sqlsocket_.async_connect(endpoint,
boost::bind(&session::handle_sqlserver_connect, this,
boost::asio::placeholders::error, ++endpoint_iterator));
}
}
對於我的代理類型,我是否需要使用其他方法而不是 async_*? 我正在從我的公司想要重新啟動的一些舊項目中移植代碼,但使用的是 boost 而不是以前使用的 Winsock 東西。
知道可能是什么問題嗎?
舊代碼做了這樣的事情:帶有accept方法調用的main方法創建了兩個線程
CreateThread(0, 0, (LPTHREAD_START_ROUTINE)listenatclient, (LPVOID)cs, 0, 0);
CreateThread(0, 0, (LPTHREAD_START_ROUTINE)listenatserver, (LPVOID)cs, 0, 0);
並且線程調用了以下函數:
void listenatclient(LPVOID connection)
{
connection_s* cs = (connection_s*)connection;
char inMessagecli[MSG_SIZE];
int rcount = 0;
...
do
{
memset(inMessagecli, 0, MSG_SIZE);
rcount = recv((SOCKET)cs->client, inMessagecli, MSG_SIZE, 0);
if (rcount != SOCKET_ERROR)
{
// analyze package
...
send((SOCKET)cs->server, inMessagecli, rcount, 0);
}
} while (rcount > 0);
}
void listenatserver(LPVOID connection)
{
connection_s* cs = (connection_s*)connection;
char inMessageserv[MSG_SIZE];
int rcount = 0;
do
{
memset(inMessageserv, 0, MSG_SIZE);
rcount = recv((SOCKET)cs->server, inMessageserv, MSG_SIZE, 0);
if (rcount != SOCKET_ERROR)
{
send((SOCKET)cs->client, inMessageserv, rcount, 0);
}
} while (rcount > 0);
}
[編輯]:我嘗試同時為客戶端和 sqlserver 運行 async_read 命令,但現在我總是崩潰,有時在 boost::bind 中,有時在 boost 庫的其他部分。
似乎發生的是創建了 2 或 3 個連接(3 個會話)。 在關閉第一個 session 時,崩潰似乎發生在第二個 session 中。
是 boost asio 不安全還是我在這里做錯了什么:-)?
我在這里發布了小 ProxyServer 的代碼:
session.h:鏈接
session.cpp:鏈接
server.h:鏈接
server.cpp:鏈接
ProxyServer.cpp:鏈接
我懷疑正在發生的事情是您的 async_read_some 調用之一正在返回“一些”數據,但不足以讓 SQL 服務器滿意它已收到完整的請求。 您的代碼始終遵循 read_from_client -> send_to_server -> read_from_server -> send_to_client 的路徑。 它不處理您需要 read_from_client -> send_to_server ->read_from_client -> send_to_server -> read_from_server -> send_to_client 的情況。
到目前為止,您編寫的代碼不會執行與原始代碼相同的操作。 具體來說,舊代碼同時監聽 sockets 上的讀取。 幸運的是,由於您使用的是 ASIO,因此您不需要弄亂線程。 只需在 sockets 上同時發出 async_read_some 請求並異步處理它們。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.