C++ 基于Boost.Asio实现端口映射器

Boost.Asio 是一个功能强大的 C++ 库,用于异步编程和网络编程,它提供了跨平台的异步 I/O 操作。在这篇文章中,我们将深入分析一个使用 Boost.Asio 实现的简单端口映射服务器,该服务器能够将本地端口的数据包转发到指定的远程服务器上。

端口映射通常用于将一个网络端口上的流量转发到另一个网络端口。这对于实现网络中间人攻击、内网穿透等场景非常有用。我们将使用 Boost.Asio 提供的异步操作来实现这个简单而功能强大的端口映射服务器。

#include <list>
#include <boost/asio.hpp>
#include <boost/bind.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/function.hpp>
#include <boost/enable_shared_from_this.hpp>using boost::asio::ip::tcp;

首先,让我们简要概述代码的主要类:

  • socket_client 类:继承了 boost::enable_shared_from_thistcp::socket,用于表示客户端的套接字。
  • socket_pipe 类:表示端口映射的管道,负责在两个客户端之间传递数据。
  • async_listener 类:用于异步监听指定端口的连接请求,通过回调函数处理连接。
  • port_map_server 类:管理多个监听器,支持添加端口映射规则,并处理连接请求。

1.1 socket_client

socket_client 类继承自 boost::enable_shared_from_thistcp::socket。通过 create 静态方法创建一个 socket_client 实例,提供了共享指针的方式管理对象的生命周期。

如下代码是一个使用 Boost.Asio 库创建的异步 TCP 客户端类。

class socket_client: public boost::enable_shared_from_this<socket_client>, public tcp::socket
{
public:typedef boost::shared_ptr<socket_client> pointer;static pointer create(boost::asio::io_service& io_service){return pointer(new socket_client(io_service));}public:socket_client(boost::asio::io_service& io_service):tcp::socket(io_service){}
};

以下是对该类的概括:

  • 类名socket_client
  • 继承关系
    • 继承自 boost::enable_shared_from_this<socket_client>,这允许在异步操作中安全地使用 shared_from_this,以避免悬挂指针的问题。
    • 继承自 tcp::socket,表示该类是一个 TCP 套接字。
  • 公共成员类型
    • pointerboost::shared_ptr<socket_client> 类型的别名,用于管理该类的实例。
  • 公共静态函数
    • create:工厂方法,用于创建 socket_client 的实例。通过此方法获取了一个智能指针指向新创建的实例。
  • 公共构造函数
    • socket_client(boost::asio::io_service& io_service):构造函数,接受一个 boost::asio::io_service 引用,用于初始化基类 tcp::socket

该类的目的是提供一个异步 TCP 客户端的基本结构,使其能够与 Boost.Asio 库中的异步 I/O 操作协同工作。实际使用时,可以根据具体需求扩展该类,添加成员函数和操作,以实现特定的异步操作逻辑。

1.2 socket_pipe

socket_pipe 类用于处理两个客户端之间的数据传递。通过异步操作实现了从一个客户端读取数据,并将数据写入另一个客户端。出现错误时,会关闭两个客户端的连接。这里使用了递归的方式,实现了数据的循环传递。

如下代码是一个使用是一个 socket_pipe 类的定义,用于在两个 socket_client 实例之间建立数据传输管道。

class socket_pipe
{
public:socket_pipe(socket_client::pointer read, socket_client::pointer write):read_socket_(*read), write_socket_(*write){read_ = read;write_ = write;begin_read();}private:void begin_read(){read_socket_.async_read_some(boost::asio::buffer(data_, max_length),boost::bind(&socket_pipe::end_read, this,boost::asio::placeholders::error,boost::asio::placeholders::bytes_transferred));}void end_read(const boost::system::error_code& error, size_t bytes_transferred){if (error)handle_error(error);elsebegin_write(bytes_transferred);}void begin_write(int bytes_transferred){boost::asio::async_write(write_socket_,boost::asio::buffer(data_, bytes_transferred),boost::bind(&socket_pipe::end_write, this,boost::asio::placeholders::error));}void end_write(const boost::system::error_code& error){if (error)handle_error(error);elsebegin_read();}void handle_error(const boost::system::error_code& error){read_socket_.close();write_socket_.close();delete this;}private:socket_client& read_socket_;socket_client& write_socket_;socket_client::pointer read_;socket_client::pointer write_;enum { max_length = 1024 };char data_[max_length];
};

以下是对该类的概括:

  • 类名socket_pipe
  • 公共构造函数
    • socket_pipe(socket_client::pointer read, socket_client::pointer write):构造函数,接受两个 socket_client::pointer 实例,一个用于读取数据 (read_socket_),另一个用于写入数据 (write_socket_)。在构造函数中,调用了 begin_read 函数,启动了异步读取操作。
  • 私有成员函数
    • begin_read():启动异步读取操作,使用 read_socket_.async_read_some 异步读取数据。
    • end_read(const boost::system::error_code& error, size_t bytes_transferred):读取操作完成时的回调函数,处理可能的错误,如果没有错误则调用 begin_write 启动异步写入操作。
    • begin_write(int bytes_transferred):启动异步写入操作,使用 boost::asio::async_write 异步写入数据。
    • end_write(const boost::system::error_code& error):写入操作完成时的回调函数,处理可能的错误,如果没有错误则调用 begin_read 启动下一轮异步读取操作。
    • handle_error(const boost::system::error_code& error):处理错误的函数,关闭读取和写入套接字,并释放当前 socket_pipe 实例。
  • 私有成员变量
    • socket_client& read_socket_:引用传递的读取套接字。
    • socket_client& write_socket_:引用传递的写入套接字。
    • socket_client::pointer read_:指向读取套接字的智能指针。
    • socket_client::pointer write_:指向写入套接字的智能指针。
    • enum { max_length = 1024 };:定义了最大数据长度。
    • char data_[max_length];:存储数据的缓冲区。

该类的主要目的是在两个 socket_client 之间实现数据的双向传输,通过异步操作实现了循环的读取和写入过程。在错误处理中,如果出现错误,会关闭套接字并释放当前的 socket_pipe 实例。

1.3 async_listener

async_listener 类负责异步监听指定端口,并通过回调函数处理连接。在连接建立时,会调用用户提供的回调函数进行处理。通过 begin_accept 方法开始异步监听。

如下代码是一个使用 async_listener 类的定义,用于异步监听指定端口的连接。

class async_listener
{
public:typedef boost::function<void(socket_client::pointer client)> accept_handler;typedef boost::shared_ptr<async_listener> pointer;public:async_listener(short port, boost::asio::io_service& io_service):io_service_(io_service),acceptor_(io_service, tcp::endpoint(tcp::v4(), port)){begin_accept();}void begin_accept(){socket_client::pointer client = socket_client::create(io_service_);acceptor_.async_accept(*client,boost::bind(&async_listener::end_accept, this, client,boost::asio::placeholders::error));}void end_accept(socket_client::pointer client, const boost::system::error_code& error){if (error)handle_error(error);begin_accept();if (!handle_accept.empty())handle_accept(client);}void handle_error(const boost::system::error_code& error){}public:accept_handler handle_accept;private:tcp::acceptor acceptor_;boost::asio::io_service& io_service_;
};

以下是对该类的概括:

  • 类名async_listener
  • 公共成员类型
    • accept_handlerboost::function<void(socket_client::pointer client)> 类型的别名,用于定义连接建立时的回调函数。
    • pointerboost::shared_ptr<async_listener> 类型的别名,用于管理该类的实例。
  • 公共构造函数
    • async_listener(short port, boost::asio::io_service& io_service):构造函数,接受一个短整型端口号和一个 boost::asio::io_service 引用。在构造函数中,创建了一个 TCP 接受器 (acceptor_) 并调用 begin_accept 启动异步接受操作。
  • 公共函数
    • begin_accept():启动异步接受操作,创建一个新的 socket_client 实例,并调用 acceptor_.async_accept 异步等待连接的建立。
    • end_accept(socket_client::pointer client, const boost::system::error_code& error):异步接受操作完成时的回调函数,处理可能的错误,如果没有错误则调用 begin_accept 启动下一轮异步接受操作。如果定义了 handle_accept 回调函数,则调用它并传递新连接的 socket_client 实例。
  • 私有成员函数
    • handle_error(const boost::system::error_code& error):处理错误的函数,目前仅为空实现。
  • 公共成员变量
    • accept_handler handle_accept:用于存储用户定义的连接建立时的回调函数。
  • 私有成员变量
    • tcp::acceptor acceptor_:TCP 接受器,用于监听连接。
    • boost::asio::io_service& io_service_:引用传递的 io_service,用于执行异步操作。

该类的主要目的是实现异步监听,一旦有连接建立,就通过回调函数通知用户,并通过 handle_error 处理可能的错误。在连接建立后,会继续监听新的连接。

1.4 port_map_server

port_map_server 类管理多个监听器,支持动态添加端口映射规则。在连接建立时,会调用 handle_accept 处理连接请求。通过 begin_connect 方法开始异步连接远程服务器。

如下代码是一个 port_map_server 类的定义,它通过异步监听多个本地端口,并将连接映射到远程服务器的不同端口。

class port_map_server
{
public:port_map_server(boost::asio::io_service& io_service) :io_service_(io_service){}void add_portmap(short port, tcp::endpoint& remote_endpoint){async_listener::pointer listener(new async_listener(port, io_service_));listeners.push_back(listener);listener->handle_accept = boost::bind(&port_map_server::handle_accept, this, remote_endpoint, _1);}void handle_accept(tcp::endpoint remote_endpoint, socket_client::pointer client){begin_connect(remote_endpoint, client);}void begin_connect(tcp::endpoint& remote_endpoint, socket_client::pointer socket_local){socket_client::pointer socket_remote = socket_client::create(io_service_);socket_remote->async_connect(remote_endpoint,boost::bind(&port_map_server::end_connect, this,boost::asio::placeholders::error, socket_local, socket_remote));}void end_connect(const boost::system::error_code& error, socket_client::pointer socket_local, socket_client::pointer socket_remote){if (error){handle_error(error);}else{new socket_pipe(socket_local, socket_remote);new socket_pipe(socket_remote, socket_local);}}void handle_error(const boost::system::error_code& error){}private:boost::asio::io_service& io_service_;std::list<async_listener::pointer> listeners;
};

以下是对该类的概括:

  • 类名port_map_server
  • 公共构造函数
    • port_map_server(boost::asio::io_service& io_service):构造函数,接受一个 boost::asio::io_service 引用。
  • 公共函数
    • add_portmap(short port, tcp::endpoint& remote_endpoint):添加端口映射规则的函数。为指定端口创建一个新的 async_listener 实例,并将其添加到 listeners 列表中。同时,设置 handle_accept 回调函数,以便在新连接建立时调用 handle_accept 函数。
  • 私有成员函数
    • handle_accept(tcp::endpoint remote_endpoint, socket_client::pointer client):处理新连接建立时的回调函数。在此函数中,调用 begin_connect 启动异步连接到远程服务器的操作。
    • begin_connect(tcp::endpoint& remote_endpoint, socket_client::pointer socket_local):启动异步连接到远程服务器的操作,创建一个新的远程套接字。
    • end_connect(const boost::system::error_code& error, socket_client::pointer socket_local, socket_client::pointer socket_remote):处理异步连接操作完成时的回调函数。如果连接成功,创建两个 socket_pipe 实例,分别用于将数据从本地传输到远程和从远程传输回本地。
    • handle_error(const boost::system::error_code& error):处理错误的函数,目前仅为空实现。
  • 私有成员变量
    • boost::asio::io_service& io_service_:引用传递的 io_service,用于执行异步操作。
    • std::list<async_listener::pointer> listeners:存储多个 async_listener 实例的列表。

该类的主要目的是通过创建多个 async_listener 实例,监听多个本地端口,并在新连接建立时将其映射到远程服务器的不同端口。在连接建立后,会启动异步连接到远程服务器的操作,并创建数据传输的管道。

1.5 port_map_server

这是程序的 main 函数,负责创建一个 boost::asio::io_service 实例,设置两个远程服务器的端点,然后创建一个 port_map_server 实例并添加两个端口映射规则。最后,通过调用 io_service.run() 开始事件循环。

以下是对 main 函数的概括:

  • 函数功能
    • 创建一个 boost::asio::io_service 实例,用于管理异步操作的事件循环。
    • 定义两个远程服务器的端点 (ep1ep2),分别是 192.168.1.100:80192.168.1.200:80
    • 创建一个 port_map_server 实例,该实例使用上述 io_service
    • 通过 add_portmap 函数向 port_map_server 添加两个端口映射规则,将本地端口 5000 映射到远程服务器 192.168.1.100:80,将本地端口 6000 映射到远程服务器 192.168.1.200:80
    • 调用 io_service.run() 开始事件循环,等待异步操作的完成。
  • 异常处理
    • 使用了 trycatch 块,捕获任何可能抛出的异常,并在 catch 块中忽略异常。
  • 返回值
    • 返回整数 0 表示程序正常结束。

这个 main 函数的作用是启动异步事件循环,使得 port_map_server 开始监听指定端口,接受连接,并将连接映射到远程服务器上。

int main(int argc, char* argv[])
{try{boost::asio::io_service io_service;tcp::endpoint ep1(boost::asio::ip::address_v4::from_string("192.168.1.100"), 80);tcp::endpoint ep2(boost::asio::ip::address_v4::from_string("192.168.1.200"), 80);port_map_server server(io_service);// 访问本机5000端口,将数据包转发到 8.141.58.64:80server.add_portmap(5000, ep1);// 访问本机6000端口,将数据包转发到 8.141.58.64:80server.add_portmap(6000, ep2);io_service.run();}catch (...) {}return 0;
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/171014.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

OSG编程指南<十一>:OSG几何体操作及三维地形创建

1、简化几何体 在 OSG 中&#xff0c;场景都是由基本的绘图基元构成的&#xff0c;基本的绘图基元构成简单的几何体&#xff0c;简单的几何体构成复杂的几何体&#xff0c;复杂的几何体最终构造成复杂的场景。当多个几何体组合时&#xff0c;可能 存在多种降低场景渲染效率的原…

220. 存在重复元素 III(滑动窗口+有序集合)

Problem: 220. 存在重复元素 III 文章目录 题目思路Code 题目 给你一个整数数组 nums 和两个整数 indexDiff 和 valueDiff 。 找出满足下述条件的下标对 (i, j)&#xff1a; i ! j,abs(i - j) < indexDiffabs(nums[i] - nums[j]) < valueDiff 如果存在&#xff0c;返回…

Vue事件处理

Vue的事件处理与BOMDOMjs下的事件处理没有明显的差异&#xff0c;差别主要体现在规范化上。在模板或者DOM结构上绑定事件的指令是v-on或者符号。事件的处理函数除了直接指定处理函数之外&#xff0c;还增加了对事件简易处理函数的内联处理方式。在事件上&#xff0c;Vue增加了事…

JAVA小游戏“简易版王者荣耀”

第一步是创建项目 项目名自拟 第二部创建个包名 来规范class 然后是创建类 GameFrame 运行类 package com.sxt;import java.awt.Graphics; import java.awt.Image; import java.awt.Toolkit; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; im…

06_正则与异常处理

正则表达式 基础演示 public class Test {public static void main(String[] args) {System.out.println(check("7029622989")); // true}public static boolean check(String userId) {return userId ! null && userId.matches("[1-9]\\d{5,19}&quo…

DPDK系列之三十七IO处理

一、介绍 如果一条通信链路要想达到最优的效果&#xff0c;一定是整体上每个环节都要有最佳的节奏协调控制而不一定是每个环节都是最优。这个在计算机的数据处理上就更是明显。一般来说&#xff0c;IO的速度是最低的&#xff0c;至少在可见的时光里要想超越CPU和内存还是很难的…

Scope 模块

Scope 模块可以连接任何类型的实数信号线 (不支持复数)。 波形显示界面主要包括两个部分: Scope 独有的工具栏、波形显示区域。 波形显示界面默认是黑色背景, 当有单个信号输入时, 信号线是黄色的。 Scope 模块也有菜单栏, 只不过默认将其句柄和显示都隐藏起来, 可以通过下面…

技术类知识汇总(二)

在自己日常学习javaweb的过程中&#xff0c;做的一些笔记和总结&#xff0c;汇总如下&#xff1a; Springboot项目的静态资源(html&#xff0c;css&#xff0c;js等前端资源)默认存放目录为&#xff1a;classpath:/static classpath:/public classpath:/resources"三层架…

java-hprof 文件是什么

一、是什么 hprof 文件是 Java进程所使用的内存情况在某一时间的一次快照&#xff08;Heap Profile 的缩写&#xff09;&#xff0c;格式为java_pidxxxxx*.hprof 二、文件里面有什么 1、所有的对象信息 对象的类信息、字段信息、原生值(int, long等)及引用值 2、所有的类信…

一键下载Python各版本中的最新版

一、背景需求 下载Python的安装包非常简单&#xff0c;只需要去官网就可以了&#xff1a;https://www.python.org/downloads/windows/ 但是有时候你可能需要环境测试&#xff0c;需要安装很多版本的Python。 一个一个找倒是也可以&#xff0c;但是我做了个自动筛选的脚本&am…

Kafka-Consumer

Kafka消费者 消费者 与生产者对应的是消费者&#xff0c;应用程序可以通过KafkaConsumer来订阅主题&#xff0c;并从订阅的主题中拉取消息。 消费者与消费者组 Kafka的消费者&#xff08;Consumer&#xff09;负责订阅Kafka中的主题&#xff08;Topic&#xff09;&#xff…

Web 安全之证书透明(Certificate Transparency)详解

目录 证书透明性的概念 数字证书和颁发机构 证书透明的起源 证书透明的工作原理 证书透明的实现方法 证书透明的优点 浏览器和客户端对证书透明的支持情况 小结 证书透明&#xff08;Certificate Transparency, CT&#xff09;是网络安全领域中的一个重要概念&#xff…

车载以太网-数据链路层-MAC

文章目录 车载以太网MAC(Media Access Control)车载以太网MAC帧格式以太网MAC帧报文示例车载以太网MAC层测试内容车载以太网MAC(Media Access Control) 车载以太网MAC(Media Access Control)是一种用于车载通信系统的以太网硬件地址,用于在物理层上识别和管理数据包的传…

5 个适用于 Windows 的顶级免费数据恢复软件

对于计算机来说&#xff0c;最重要的是用户数据。除了您的数据之外&#xff0c;有关计算机的其他所有内容都是可替换的。这三个是数据丢失的最常见原因&#xff1a; 文件/文件夹删除丢失分区分区损坏 文件/文件夹删除 文件/文件夹删除是最常见的数据丢失类型。大多数时候&am…

《微信小程序开发从入门到实战》学习三十一

3.4 开发参与投票页面 3.4.9 显示投票结果 在实际使用中&#xff0c;一个用户不能对同一个投票进行重复提交&#xff0c;因此需要向服务器端提交投票结果和提交用户ID。另外页面&#xff0c;需要完善。用户提交完投票后 &#xff0c;还需要显示投票目前的结果&#xff0c;提交…

C语言进阶之笔试题详解(1)

引言&#xff1a; 对指针知识进行简单的回顾&#xff0c;然后再完成笔试题。 ✨ 猪巴戒&#xff1a;个人主页✨ 所属专栏&#xff1a;《C语言进阶》 &#x1f388;跟着猪巴戒&#xff0c;一起学习C语言&#x1f388; 目录 引言&#xff1a; 知识简单回顾 指针是什么 指针变…

1 时间序列模型入门: LSTM

0 前言 循环神经网络&#xff08;Recurrent Neural Network&#xff0c;RNN&#xff09;是一种用于处理序列数据的神经网络。相比一般的神经网络来说&#xff0c;他能够处理序列变化的数据。比如某个单词的意思会因为上文提到的内容不同而有不同的含义&#xff0c;RNN就能够很好…

2023-3年CSDN创作纪念日

机缘 今天开开心心出门去上班&#xff0c;就收到了一个csdn私信&#xff0c;打开一看说是给我惊喜来着&#xff0c;我心想csdn还能给惊喜&#xff1f;以为是有什么奖品或者周边之类的&#xff0c;结果什么也没有&#xff0c;打开就是一份信&#x1f602;。 也挺不错的&#xf…

Python基础入门例程67-NP67 遍历字典(字典)

最近的博文: Python基础入门例程66-NP66 增加元组的长度(元组)-CSDN博客 Python基础入门例程65-NP65 名单中出现过的人(元组)-CSDN博客 Python基础入门例程64-NP64 输出前三同学的成绩(元组)-CSDN博客 目录 最近的博文: 描述