/*
|
* =====================================================================================
|
*
|
* Filename: tcp_proxy.cpp
|
*
|
* Description:
|
*
|
* Version: 1.0
|
* Created: 2021年05月19日 15时04分15秒
|
* Revision: none
|
* Compiler: gcc
|
*
|
* Author: Li Chao (), lichao@aiotlink.com
|
* Organization:
|
*
|
* =====================================================================================
|
*/
|
#include "tcp_proxy.h"
|
#include "defs.h"
|
#include "shm_socket.h"
|
#include "tcp_connection.h"
|
|
TcpProxy::TcpProxy() :
|
run_(false) {}
|
|
TcpProxy::~TcpProxy() {}
|
|
bool TcpProxy::Start(bhome_shm::SharedMemory &shm)
|
{
|
Stop();
|
bool cur = false;
|
if (!run_.compare_exchange_strong(cur, true)) { return false; }
|
|
auto &mq = GetCenterInfo(shm)->mq_tcp_proxy_;
|
local_.reset(new ShmSocket(mq.offset_, shm, mq.id_));
|
auto localProc = [this](ShmSocket &sock, MsgI &msg, BHMsgHead &head) {
|
auto &dest = head.dest();
|
if (dest.ip().empty() || dest.port() == 0) { return; }
|
Request(dest.ip(), dest.port(), msg.content());
|
};
|
local_->Start(1, localProc);
|
|
auto proxyProc = [this]() {
|
while (run_) {
|
io_context_.run_one_for(std::chrono::milliseconds(100));
|
}
|
};
|
std::thread(proxyProc).swap(worker_);
|
return true;
|
}
|
|
void TcpProxy::Stop()
|
{
|
bool cur = true;
|
if (run_.compare_exchange_strong(cur, false)) {
|
if (worker_.joinable()) {
|
worker_.join();
|
}
|
local_.reset();
|
}
|
}
|
|
bool TcpProxy::Request(const std::string &ip, int port, std::string &&content)
|
{
|
if (content.empty()) { return false; }
|
|
tcp::endpoint dest(ip::address::from_string(ip), port);
|
TcpRequest1::Create(io_context_, dest, std::move(content), *local_);
|
}
|
