71 lines
1.9 KiB
C++
71 lines
1.9 KiB
C++
|
#include <std_include.hpp>
|
||
|
|
#include <utility>
|
||
|
|
#include "stun_server.hpp"
|
||
|
|
#include "utils/cryptography.hpp"
|
||
|
|
#include "byte_buffer.hpp"
|
||
|
|
|
||
|
|
namespace demonware
|
||
|
|
{
|
||
|
|
stun_server::stun_server(std::string _name) : name_(std::move(_name))
|
||
|
|
{
|
||
|
|
this->address_ = utils::cryptography::jenkins_one_at_a_time::compute(this->name_);
|
||
|
|
}
|
||
|
|
|
||
|
|
unsigned long stun_server::get_address() const
|
||
|
|
{
|
||
|
|
return this->address_;
|
||
|
|
}
|
||
|
|
|
||
|
|
void stun_server::ip_discovery(SOCKET s, const sockaddr* to, int tolen) const
|
||
|
|
{
|
||
|
|
const uint32_t ip = 0x0100007f;
|
||
|
|
|
||
|
|
byte_buffer buffer;
|
||
|
|
buffer.set_use_data_types(false);
|
||
|
|
buffer.write_byte(31); // type
|
||
|
|
buffer.write_byte(2); // version
|
||
|
|
buffer.write_byte(0); // version
|
||
|
|
buffer.write_uint32(ip); // external ip
|
||
|
|
buffer.write_uint16(3074); // port
|
||
|
|
|
||
|
|
//Components::DemonWare::SendDatagramPacket(s, buffer, to, tolen);
|
||
|
|
}
|
||
|
|
|
||
|
|
void stun_server::nat_discovery(SOCKET s, const sockaddr* to, int tolen) const
|
||
|
|
{
|
||
|
|
const uint32_t ip = 0x0100007f;
|
||
|
|
|
||
|
|
byte_buffer buffer;
|
||
|
|
buffer.set_use_data_types(false);
|
||
|
|
buffer.write_byte(21); // type
|
||
|
|
buffer.write_byte(2); // version
|
||
|
|
buffer.write_byte(0); // version
|
||
|
|
buffer.write_uint32(ip); // external ip
|
||
|
|
buffer.write_uint16(3074); // port
|
||
|
|
buffer.write_uint32(this->get_address()); // server ip
|
||
|
|
buffer.write_uint16(3074); // server port
|
||
|
|
|
||
|
|
//Components::DemonWare::SendDatagramPacket(s, buffer, to, tolen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int stun_server::send(const SOCKET s, const char* buf, int len, const sockaddr* to, int tolen) const
|
||
|
|
{
|
||
|
|
uint8_t type, version, padding;
|
||
|
|
|
||
|
|
byte_buffer buffer(std::string(buf, len));
|
||
|
|
buffer.set_use_data_types(false);
|
||
|
|
buffer.read_byte(&type);
|
||
|
|
buffer.read_byte(&version);
|
||
|
|
buffer.read_byte(&padding);
|
||
|
|
|
||
|
|
switch (type)
|
||
|
|
{
|
||
|
|
case 30: this->ip_discovery(s, to, tolen); break;
|
||
|
|
case 20: this->nat_discovery(s, to, tolen); break;
|
||
|
|
default: break;
|
||
|
|
}
|
||
|
|
|
||
|
|
return len;
|
||
|
|
}
|
||
|
|
}
|