7 #include <visiontransfer/internalinformation.h>
8 #include <visiontransfer/networking.h>
9 #include <visiontransfer/datachannelservicebase.h>
13 using namespace visiontransfer;
14 using namespace visiontransfer::internal;
16 namespace visiontransfer {
19 DataChannelServiceBase::DataChannelServiceBase() {
21 if((dataChannelSocket = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0) {
22 throw std::runtime_error(
"Error creating data channel service socket!");
25 Networking::enableReuseAddress(dataChannelSocket,
true);
28 sockaddr_in localAddr;
29 memset(&localAddr, 0,
sizeof(localAddr));
30 localAddr.sin_family = AF_INET;
31 localAddr.sin_port = htons(InternalInformation::DATACHANNELSERVICE_PORT);
32 localAddr.sin_addr.s_addr = htonl(INADDR_ANY);
33 if(::bind(dataChannelSocket, (sockaddr *)&localAddr,
sizeof(localAddr)) != 0) {
34 throw std::runtime_error(
"Error binding dataChannel socket!");
37 Networking::setSocketBlocking(dataChannelSocket,
false);
40 DataChannelServiceBase::~DataChannelServiceBase() {
41 Networking::closeSocket(dataChannelSocket);
44 void DataChannelServiceBase::process() {
45 static unsigned char buffer[100000];
46 static sockaddr_in senderAddress;
47 static socklen_t senderLength = (socklen_t)
sizeof(senderAddress);
52 received = recvfrom(dataChannelSocket, (
char*) buffer,
sizeof(buffer), 0, (sockaddr *)&senderAddress, &senderLength);
56 message.header.channelID = (DataChannel::ID) raw->channelID;
57 message.header.channelType = (DataChannel::Type) raw->channelType;
58 message.header.payloadSize = ntohl(raw->payloadSize);
61 std::cerr <<
"DataChannelServiceBase: Size mismatch in UDP message, type " << (int) message.header.channelType <<
" ID " << (
int) message.header.channelID <<
" - discarded!" << std::endl;
63 if (!(message.header.channelType)) {
64 handleChannel0Message(message, &senderAddress);
67 auto it = channels.find(message.header.channelID);
68 if (it != channels.end()) {
69 it->second->handleMessage(message, &senderAddress);
78 for (
auto& kv: channels) {
85 int DataChannelServiceBase::sendDataInternal(
unsigned char* compiledMessage,
unsigned int messageSize, sockaddr_in* recipient) {
86 if (!recipient)
throw std::runtime_error(
"Requested sendDataInternal without recipient address");
90 if (messageSize != reportedSize)
throw std::runtime_error(
"Message size does not match");
92 result = sendto(dataChannelSocket, (
char*) compiledMessage, reportedSize, 0, (sockaddr*) recipient,
sizeof(*recipient));
93 if (result != (
int) reportedSize) {
94 std::cerr <<
"Error sending DataChannel message to " << inet_ntoa(recipient->sin_addr) <<
": " << Networking::getLastErrorString() << std::endl;
95 throw std::runtime_error(
"Error during sendto");
101 int DataChannelServiceBase::sendDataIsolatedPacket(DataChannel::ID
id, DataChannel::Type type,
unsigned char* data,
unsigned int dataSize, sockaddr_in* recipient) {
103 unsigned char* buf =
new unsigned char[msgSize]();
105 header->channelID = id;
106 header->channelType = type;
107 header->payloadSize = htonl(dataSize);
110 int result = sendDataInternal(buf, msgSize, recipient);
115 DataChannel::ID DataChannelServiceBase::registerChannel(std::shared_ptr<DataChannel> channel) {
117 DataChannel::ID
id = (DataChannel::ID) channel->getChannelType();
118 if (channels.count(
id)) {
122 if (!channel->initialize())
return 0;
123 channel->setChannelID(
id);
124 channels[id] = channel;
125 channel->setService(shared_from_this());
128 int DataChannel::sendData(
unsigned char* data,
unsigned int dataLen, sockaddr_in* recipient) {
129 if (
auto srv = service.lock()) {
130 return srv->sendDataIsolatedPacket(channelID, getChannelType(), data, dataLen, recipient);