Ошибка в прокси сервере на C++ - C++ - Ответ 1847625

@niXman · 17.07.2011, 19:00

в общем, пока ты копашился, я наваял на коленке tcp-proxy-server. ко всем просьба, сильно не стебаться если найдете ошибку в поделке

C++

#include <string>
#include <iostream>
 
#include <boost/asio.hpp>
#include <boost/bind.hpp>
#include <boost/array.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>
 
#include <boost/program_options.hpp>
 
/***************************************************************************/
 
struct user_session: boost::enable_shared_from_this<user_session> {
   enum { default_buffer_size = 1024*32 };
 
   user_session(boost::asio::io_service& io_service)
      :_client_socket(io_service),
      _proxy_socket(io_service)
   {}
 
   boost::asio::ip::tcp::socket& get_socket() {
      return _client_socket;
   }
 
   void start(const std::string& proxy_ip, std::size_t proxy_port) {
      _proxy_socket.async_connect(
         boost::asio::ip::tcp::endpoint(boost::asio::ip::address::from_string(proxy_ip), proxy_port),
         boost::bind(
            &user_session::handle_connect,
            shared_from_this(),
            boost::asio::placeholders::error
         )
      );
   }
 
   void handle_connect(const boost::system::error_code& error) {
      if ( !error ) {
         _proxy_socket.async_read_some(
            boost::asio::buffer(_proxy_buffer, default_buffer_size),
            boost::bind(
               &user_session::handle_proxy_read,
               shared_from_this(),
               boost::asio::placeholders::error,
               boost::asio::placeholders::bytes_transferred
            )
         );
 
         _client_socket.async_read_some(
            boost::asio::buffer(_client_buffer, default_buffer_size),
            boost::bind(
               &user_session::handle_client_read,
               shared_from_this(),
               boost::asio::placeholders::error,
               boost::asio::placeholders::bytes_transferred
            )
         );
      } else {
         std::cerr << error.message() << std::endl;
         close();
      }
 
   }
 
   void handle_client_read(const boost::system::error_code& error, const std::size_t bytes_transferred) {
      if ( !error ) {
         boost::asio::async_write(
            _proxy_socket,
            boost::asio::buffer(_client_buffer, bytes_transferred),
            boost::asio::transfer_all(),
            boost::bind(
               &user_session::handle_proxy_write,
               shared_from_this(),
               boost::asio::placeholders::error
            )
         );
      } else {
         std::cerr << error.message() << std::endl;
         close();
      }
   }
 
   void handle_proxy_write(const boost::system::error_code& error) {
      if ( !error ) {
         _client_socket.async_read_some(
            boost::asio::buffer(_client_buffer, default_buffer_size),
            boost::bind(
               &user_session::handle_client_read,
               shared_from_this(),
               boost::asio::placeholders::error,
               boost::asio::placeholders::bytes_transferred
            )
         );
      } else {
         std::cerr << error.message() << std::endl;
         close();
      }
   }
 
   void handle_proxy_read(const boost::system::error_code& error, const std::size_t bytes_transferred) {
      if ( !error ) {
         boost::asio::async_write(
            _client_socket,
            boost::asio::buffer(_proxy_buffer, bytes_transferred),
            boost::asio::transfer_all(),
            boost::bind(
               &user_session::handle_client_write,
               shared_from_this(),
               boost::asio::placeholders::error
            )
         );
      } else if ( error == boost::asio::error::eof ) {
         close();
      } else {
         std::cerr << error.message() << std::endl;
         close();
      }
   }
 
   void handle_client_write(const boost::system::error_code& error) {
      if ( !error ) {
         _proxy_socket.async_read_some(
            boost::asio::buffer(_proxy_buffer, default_buffer_size),
            boost::bind(
               &user_session::handle_proxy_read,
               shared_from_this(),
               boost::asio::placeholders::error,
               boost::asio::placeholders::bytes_transferred
            )
         );
      } else {
         std::cerr << error.message() << std::endl;
         close();
      }
   }
 
   void close() {
      if ( _client_socket.is_open() ) {
         _client_socket.shutdown(boost::asio::ip::tcp::socket::shutdown_both);
         _client_socket.close();
      }
      if ( _proxy_socket.is_open() ) {
         _proxy_socket.shutdown(boost::asio::ip::tcp::socket::shutdown_both);
         _proxy_socket.close();
      }
   }
 
private:
   boost::asio::ip::tcp::socket _client_socket;
   boost::asio::ip::tcp::socket _proxy_socket;
   boost::array<char, default_buffer_size> _client_buffer;
   boost::array<char, default_buffer_size> _proxy_buffer;
};
 
/***************************************************************************/
 
struct connection_acceptor {
   connection_acceptor(
      boost::asio::io_service& io_service,
      const std::string& local_ip,
      boost::uint16_t local_port,
      const std::string remote_ip,
      boost::uint16_t remote_port
   ): _io_service(io_service),
      _acceptor(io_service, boost::asio::ip::tcp::endpoint(boost::asio::ip::address::from_string(local_ip), local_port)),
      _remote_ip(remote_ip),
      _remote_port(remote_port)
   {}
 
   boost::shared_ptr<user_session> create_connection() {
      return boost::shared_ptr<user_session>(new user_session(_io_service));
   }
 
   bool start_accept() {
      try {
         _connection = create_connection();
         _acceptor.async_accept(
            _connection->get_socket(),
            boost::bind(
               &connection_acceptor::handle_accept,
               this,
               boost::asio::placeholders::error
            )
         );
      } catch( const std::exception& e ) {
         std::cerr << "acceptor exception: " << e.what() << std::endl;
         return false;
      }
      return true;
   }
 
   void handle_accept(const boost::system::error_code& error) {
      if ( !error ) {
         _connection->start(_remote_ip, _remote_port);
         if ( !start_accept() ) {
            std::cerr << "Filed to accept." << std::endl;
         }
      } else {
         std::cerr << error.message() << std::endl;
      }
   }
 
private:
   boost::asio::io_service& _io_service;
   boost::asio::ip::tcp::acceptor _acceptor;
   boost::shared_ptr<user_session> _connection;
   std::string _remote_ip;
   boost::uint16_t _remote_port;
};
 
/***************************************************************************/
 
int main(int argc, char** argv) {
   try {
      boost::asio::io_service ios;
 
      namespace po = boost::program_options;
 
      std::string local_ip, remote_ip;
      boost::uint16_t local_port, remote_port = 0;
 
      po::options_description description("allowed options");
      description.add_options()
         ("help,h", "produce help message")
         ("local-ip",     po::value<std::string>(&local_ip)->default_value("127.0.0.1"), "local IP-address")
         ("local-port",   po::value<boost::uint16_t>(&local_port)->default_value(44599), "local port")
         ("remote-ip",    po::value<std::string>(&remote_ip), "remote IP-address")
         ("remote-port",  po::value<boost::uint16_t>(&remote_port), "remote port");
 
      po::variables_map options;
      try {
         po::store(po::parse_command_line(argc, argv, description), options);
         po::notify(options);
      } catch ( const std::exception& ex ) {
         std::cout << "command line error: " << ex.what() << std::endl;
         return 1;
      }
      if ( options.count("help") ) {
         std::cout << description << std::endl;
         return 0;
      }
      if ( !options.count("remote-ip") ) {
         std::cout << "please choose remote IP" << std::endl;
         return 1;
      }
      if ( !options.count("remote-port") ) {
         std::cout << "please choose remote port" << std::endl;
         return 1;
      }
 
      connection_acceptor acceptor(ios, local_ip, local_port, remote_ip, remote_port);
      acceptor.start_accept();
 
      ios.run();
   } catch ( const std::exception& ex ) {
      std::cout << "(exception): " << ex.what() << std::endl;
      return 1;
   }
 
   return 0;
}
 
/***************************************************************************/

замечательно работает.

с помощью MinGW и GCC собирать так:

g++ proxy.cpp -lboost_system -lboost_thread -lboost_program_options -oproxy

запускать так:

proxy --local-ip=127.0.0.1 --local-port=44550 --remote-ip=23.45.67.89 --remote-port=33445

MinGW у меня в подписи.

оригинал тут: http://forum.try-catch.ru/index.php?topic=702.0

Вернуться к обсуждению:
Ошибка в прокси сервере на C++ C++

Следующий ответ

Другие темы раздела
C++ Заготовка socket сервера https://www.cyberforum.ru/ cpp-networks/ thread333499.html В интернете видел множество примеров данной реализации у меня же вылазит ошибка 1>main.cpp 1>c:\users\adminer\documents\visual studio 2008\projects\server_for_tank_1\server_for_tank_1\main.cpp(36) : error C2679: бинарный '=': не найден оператор, принимающий правый операнд типа 'unsigned long' (или приемлемое преобразование отсутствует) 1> c:\program files\microsoft...	простой веб-сервер C++ Добрго времени суток! Написал простой веб-сервер. Возникли трудности с обменом данных по протоколу HTTP. Вот мой код с раскрытыми функциями обмена: // Сервер.cpp: определяет точку входа для консольного приложения. // #include "stdafx.h" #include "winsock.h" #include <iostream> #define PORT_ADDR 80 // порт подключения using namespace std; const int WINSOCK_VERSION =...
C++ На XP работает, а на Windows Server 2003 нет #include <winsock2.h> char sbuf="\0",sbuf2="\0"; STARTUPINFO si; PROCESS_INFORMATION pi; int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, https://www.cyberforum.ru/ cpp-networks/ thread332828.html	C++ Работа в фоновом режиме, как сделать? #include <winsock2.h> char sbuf="\0",sbuf2="\0"; STARTUPINFO si; PROCESS_INFORMATION pi; int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, https://www.cyberforum.ru/ cpp-networks/ thread332290.html
Информация о системе C++ Может есть у кого в копилке: 1. Метод определения свободного места на диске. 2. Метод определения процессора и объема оперативной памяти свободного места на диске. Все желательно без WinAPI, в идеале кросплатформенно и без вызова внешних утилит. На ОС windows, Linux, MacOS.... Поделитесь, пожалуйста.	C++ MMP mail.ru Агент протокол! https://www.cyberforum.ru/ cpp-networks/ thread330829.html Мне нужно реализовать соединение по протоколу маил агента! Я скачал заголовочный файл подключил и вобщем вот код WSADATA wsadata; SOCKADDR_IN sin; SOCKET s; hostent * hp; #pragma comment(lib, "ws2_32.lib") //Инициализация соккета
C++ Считать исходный код интернет страницы Доброго времени суток!!! Парни у меня к вам большая просьба,помогите с написанием програмы на C++! Нужно написать консольную програму в которой адрес сайта вводиш с клавиатуры и програма сохраняет исходный код страницы в файл!!! Помогите плиз!!! Ума не приложу как это сделать!!! https://www.cyberforum.ru/ cpp-networks/ thread329576.html	Простейший почтовый клиент POP3 на С++ C++ Уважаемый форумчане, мне нужна программа, которая выдает количество писем на почтовом ящике с помощью протокола POP3. Другими словами, как заставить выполнить протокол следующие команды : USER PASS QUIT STAT и прочитать то, что он возвращает.
C++ C++ общие вопросы https://www.cyberforum.ru/ cpp-networks/ thread327725.html Информация на эту тему так расхлёстана, что сводит с ума новичка в первые недели)))) Вобщем есть Win32 API со стандартными функциями работы со строками массивами графикой.. это и есть эти iostream.h conio.h string.h файлы? А где узнать о других API? например работы с сокетами или какойнибудь БД? их нужно загружать отдельно или большенство уже есть в компиляторе? Просто очень, очень сложно...	C++ Программа обновлятор https://www.cyberforum.ru/ cpp-networks/ thread327239.html Здравствуйте. ;) Поступило задание сделать программу для обновлений других программ через интернет. Программу надо написать на чистом С++. Тут возникает вопрос перекачки данных(файлов), с которым я не знаком. Не подскажите в какую сторону рыть и что читать ?) P.S. сам думаю что связанно с сокетами, но не уверен )
Отправка аттачмент с письмо C++ Доброго времени суток. Подскажите кто знает, отправляю E-mail с помощью WinSock, как к письму приложить аттачмент?	C++ сокеты - переконнект соединения https://www.cyberforum.ru/ cpp-networks/ thread321629.html Как все-таки правильно закрыть Windows-сокетное соединение со стороны клиента, чтобы порт стал доступен сразу же. Если соединение закрывает серверная сторона, клиент может сразу же долбиться на тот же самый порт, несмотря на TIME_WAIT статус порта. А если инициатором закрытия будет клиент - никакими способами, кроме linger не получается. И попутно: может кто объяснит, зачем, например, strcpy,...

	17.07.2011, 19:00