test/test_natpmp.cpp

   1 #include "libtorrent/natpmp.hpp"
   2 #include "libtorrent/socket.hpp"
   3 #include "libtorrent/connection_queue.hpp"
   4 #include <boost/bind.hpp>
   5 #include <boost/ref.hpp>
   6 #include <boost/intrusive_ptr.hpp>
   7
   8 using namespace libtorrent;
   9
  10 void callback(int mapping, int port, std::string const& err)
  11 {
  12         std::cerr << "mapping: " << mapping << ", port: " << port << ", error: \"" << err << "\"\n";
  13 }
  14
  15 int main(int argc, char* argv[])
  16 {
  17         io_service ios;
  18         std::string user_agent = "test agent";
  19
  20         if (argc != 3)
  21         {
  22                 std::cerr << "usage: " << argv[0] << " tcp-port udp-port" << std::endl;
  23                 return 1;
  24         }
  25
  26         connection_queue cc(ios);
  27         boost::intrusive_ptr<natpmp> natpmp_handler = new natpmp(ios, address_v4(), &callback);
  28
  29         deadline_timer timer(ios);
  30
  31         int tcp_map = natpmp_handler->add_mapping(natpmp::tcp, atoi(argv[1]), atoi(argv[1]));
  32         int udp_map = natpmp_handler->add_mapping(natpmp::udp, atoi(argv[2]), atoi(argv[2]));
  33
  34         timer.expires_from_now(seconds(2));
  35         timer.async_wait(boost::bind(&io_service::stop, boost::ref(ios)));
  36         std::cerr << "mapping ports TCP: " << argv[1]
  37                 << " UDP: " << argv[2] << std::endl;
  38
  39         ios.reset();
  40         ios.run();
  41         timer.expires_from_now(seconds(2));
  42         timer.async_wait(boost::bind(&io_service::stop, boost::ref(ios)));
  43         std::cerr << "removing mapping " << tcp_map << std::endl;
  44         natpmp_handler->delete_mapping(tcp_map);
  45
  46         ios.reset();
  47         ios.run();
  48         std::cerr << "removing mappings" << std::endl;
  49         natpmp_handler->close();
  50
  51         ios.reset();
  52         ios.run();
  53         std::cerr << "closing" << std::endl;
  54 }
  55
  56