mirror of
https://github.com/dolphin-emu/dolphin.git
synced 2025-07-21 05:09:34 -06:00
Reformat all the things. Have fun with merge conflicts.
This commit is contained in:
@ -1,20 +1,20 @@
|
||||
// This file is public domain, in case it's useful to anyone. -comex
|
||||
|
||||
// The central server implementation.
|
||||
#include <arpa/inet.h>
|
||||
#include <cerrno>
|
||||
#include <cstdio>
|
||||
#include <cstdlib>
|
||||
#include <cstring>
|
||||
#include <fcntl.h>
|
||||
#include <unistd.h>
|
||||
#include <unordered_map>
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
#include <arpa/inet.h>
|
||||
#include <netinet/in.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/time.h>
|
||||
#include <sys/types.h>
|
||||
#include <unistd.h>
|
||||
#include <unordered_map>
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
#include "Common/TraversalProto.h"
|
||||
|
||||
#define DEBUG 1
|
||||
@ -24,436 +24,431 @@ static u64 currentTime;
|
||||
|
||||
struct OutgoingPacketInfo
|
||||
{
|
||||
TraversalPacket packet;
|
||||
TraversalRequestId misc;
|
||||
sockaddr_in6 dest;
|
||||
int tries;
|
||||
u64 sendTime;
|
||||
TraversalPacket packet;
|
||||
TraversalRequestId misc;
|
||||
sockaddr_in6 dest;
|
||||
int tries;
|
||||
u64 sendTime;
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
struct EvictEntry
|
||||
{
|
||||
u64 updateTime;
|
||||
T value;
|
||||
u64 updateTime;
|
||||
T value;
|
||||
};
|
||||
|
||||
template <typename V>
|
||||
struct EvictFindResult
|
||||
{
|
||||
bool found;
|
||||
V* value;
|
||||
bool found;
|
||||
V* value;
|
||||
};
|
||||
|
||||
template <typename K, typename V>
|
||||
EvictFindResult<V> EvictFind(std::unordered_map<K, EvictEntry<V>>& map, const K& key, bool refresh = false)
|
||||
EvictFindResult<V> EvictFind(std::unordered_map<K, EvictEntry<V>>& map, const K& key,
|
||||
bool refresh = false)
|
||||
{
|
||||
retry:
|
||||
const u64 expiryTime = 30 * 1000000; // 30s
|
||||
EvictFindResult<V> result;
|
||||
if (map.bucket_count())
|
||||
{
|
||||
auto bucket = map.bucket(key);
|
||||
auto it = map.begin(bucket);
|
||||
for (; it != map.end(bucket); ++it)
|
||||
{
|
||||
if (currentTime - it->second.updateTime > expiryTime)
|
||||
{
|
||||
map.erase(it->first);
|
||||
goto retry;
|
||||
}
|
||||
if (it->first == key)
|
||||
{
|
||||
if (refresh)
|
||||
it->second.updateTime = currentTime;
|
||||
result.found = true;
|
||||
result.value = &it->second.value;
|
||||
return result;
|
||||
}
|
||||
}
|
||||
}
|
||||
retry:
|
||||
const u64 expiryTime = 30 * 1000000; // 30s
|
||||
EvictFindResult<V> result;
|
||||
if (map.bucket_count())
|
||||
{
|
||||
auto bucket = map.bucket(key);
|
||||
auto it = map.begin(bucket);
|
||||
for (; it != map.end(bucket); ++it)
|
||||
{
|
||||
if (currentTime - it->second.updateTime > expiryTime)
|
||||
{
|
||||
map.erase(it->first);
|
||||
goto retry;
|
||||
}
|
||||
if (it->first == key)
|
||||
{
|
||||
if (refresh)
|
||||
it->second.updateTime = currentTime;
|
||||
result.found = true;
|
||||
result.value = &it->second.value;
|
||||
return result;
|
||||
}
|
||||
}
|
||||
}
|
||||
#if DEBUG
|
||||
printf("failed to find key '");
|
||||
for (size_t i = 0; i < sizeof(key); i++) {
|
||||
printf("%02x", ((u8*) &key)[i]);
|
||||
}
|
||||
printf("'\n");
|
||||
printf("failed to find key '");
|
||||
for (size_t i = 0; i < sizeof(key); i++)
|
||||
{
|
||||
printf("%02x", ((u8*)&key)[i]);
|
||||
}
|
||||
printf("'\n");
|
||||
#endif
|
||||
result.found = false;
|
||||
return result;
|
||||
result.found = false;
|
||||
return result;
|
||||
}
|
||||
|
||||
template <typename K, typename V>
|
||||
V* EvictSet(std::unordered_map<K, EvictEntry<V>>& map, const K& key)
|
||||
{
|
||||
// can't use a local_iterator to emplace...
|
||||
auto& result = map[key];
|
||||
result.updateTime = currentTime;
|
||||
return &result.value;
|
||||
// can't use a local_iterator to emplace...
|
||||
auto& result = map[key];
|
||||
result.updateTime = currentTime;
|
||||
return &result.value;
|
||||
}
|
||||
|
||||
namespace std
|
||||
{
|
||||
template <>
|
||||
struct hash<TraversalHostId>
|
||||
{
|
||||
size_t operator()(const TraversalHostId& id) const
|
||||
{
|
||||
auto p = (u32*) id.data();
|
||||
return p[0] ^ ((p[1] << 13) | (p[1] >> 19));
|
||||
}
|
||||
};
|
||||
template <>
|
||||
struct hash<TraversalHostId>
|
||||
{
|
||||
size_t operator()(const TraversalHostId& id) const
|
||||
{
|
||||
auto p = (u32*)id.data();
|
||||
return p[0] ^ ((p[1] << 13) | (p[1] >> 19));
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
static int sock;
|
||||
static int urandomFd;
|
||||
static std::unordered_map<
|
||||
TraversalRequestId,
|
||||
OutgoingPacketInfo
|
||||
> outgoingPackets;
|
||||
static std::unordered_map<
|
||||
TraversalHostId,
|
||||
EvictEntry<TraversalInetAddress>
|
||||
> connectedClients;
|
||||
static std::unordered_map<TraversalRequestId, OutgoingPacketInfo> outgoingPackets;
|
||||
static std::unordered_map<TraversalHostId, EvictEntry<TraversalInetAddress>> connectedClients;
|
||||
|
||||
static TraversalInetAddress MakeInetAddress(const sockaddr_in6& addr)
|
||||
{
|
||||
if (addr.sin6_family != AF_INET6)
|
||||
{
|
||||
fprintf(stderr, "bad sockaddr_in6\n");
|
||||
exit(1);
|
||||
}
|
||||
u32* words = (u32*) addr.sin6_addr.s6_addr;
|
||||
TraversalInetAddress result = {0};
|
||||
if (words[0] == 0 && words[1] == 0 && words[2] == 0xffff0000)
|
||||
{
|
||||
result.isIPV6 = false;
|
||||
result.address[0] = words[3];
|
||||
}
|
||||
else
|
||||
{
|
||||
result.isIPV6 = true;
|
||||
memcpy(result.address, words, sizeof(result.address));
|
||||
}
|
||||
result.port = addr.sin6_port;
|
||||
return result;
|
||||
if (addr.sin6_family != AF_INET6)
|
||||
{
|
||||
fprintf(stderr, "bad sockaddr_in6\n");
|
||||
exit(1);
|
||||
}
|
||||
u32* words = (u32*)addr.sin6_addr.s6_addr;
|
||||
TraversalInetAddress result = {0};
|
||||
if (words[0] == 0 && words[1] == 0 && words[2] == 0xffff0000)
|
||||
{
|
||||
result.isIPV6 = false;
|
||||
result.address[0] = words[3];
|
||||
}
|
||||
else
|
||||
{
|
||||
result.isIPV6 = true;
|
||||
memcpy(result.address, words, sizeof(result.address));
|
||||
}
|
||||
result.port = addr.sin6_port;
|
||||
return result;
|
||||
}
|
||||
|
||||
static sockaddr_in6 MakeSinAddr(const TraversalInetAddress& addr)
|
||||
{
|
||||
sockaddr_in6 result;
|
||||
sockaddr_in6 result;
|
||||
#ifdef SIN6_LEN
|
||||
result.sin6_len = sizeof(result);
|
||||
result.sin6_len = sizeof(result);
|
||||
#endif
|
||||
result.sin6_family = AF_INET6;
|
||||
result.sin6_port = addr.port;
|
||||
result.sin6_flowinfo = 0;
|
||||
if (addr.isIPV6)
|
||||
{
|
||||
memcpy(&result.sin6_addr, addr.address, 16);
|
||||
}
|
||||
else
|
||||
{
|
||||
u32* words = (u32*) result.sin6_addr.s6_addr;
|
||||
words[0] = 0;
|
||||
words[1] = 0;
|
||||
words[2] = 0xffff0000;
|
||||
words[3] = addr.address[0];
|
||||
}
|
||||
result.sin6_scope_id = 0;
|
||||
return result;
|
||||
result.sin6_family = AF_INET6;
|
||||
result.sin6_port = addr.port;
|
||||
result.sin6_flowinfo = 0;
|
||||
if (addr.isIPV6)
|
||||
{
|
||||
memcpy(&result.sin6_addr, addr.address, 16);
|
||||
}
|
||||
else
|
||||
{
|
||||
u32* words = (u32*)result.sin6_addr.s6_addr;
|
||||
words[0] = 0;
|
||||
words[1] = 0;
|
||||
words[2] = 0xffff0000;
|
||||
words[3] = addr.address[0];
|
||||
}
|
||||
result.sin6_scope_id = 0;
|
||||
return result;
|
||||
}
|
||||
|
||||
static void GetRandomBytes(void* output, size_t size)
|
||||
{
|
||||
static u8 bytes[8192];
|
||||
static size_t bytesLeft = 0;
|
||||
if (bytesLeft < size)
|
||||
{
|
||||
ssize_t rv = read(urandomFd, bytes, sizeof(bytes));
|
||||
if (rv != sizeof(bytes))
|
||||
{
|
||||
perror("read from /dev/urandom");
|
||||
exit(1);
|
||||
}
|
||||
bytesLeft = sizeof(bytes);
|
||||
}
|
||||
memcpy(output, bytes + (bytesLeft -= size), size);
|
||||
static u8 bytes[8192];
|
||||
static size_t bytesLeft = 0;
|
||||
if (bytesLeft < size)
|
||||
{
|
||||
ssize_t rv = read(urandomFd, bytes, sizeof(bytes));
|
||||
if (rv != sizeof(bytes))
|
||||
{
|
||||
perror("read from /dev/urandom");
|
||||
exit(1);
|
||||
}
|
||||
bytesLeft = sizeof(bytes);
|
||||
}
|
||||
memcpy(output, bytes + (bytesLeft -= size), size);
|
||||
}
|
||||
|
||||
static void GetRandomHostId(TraversalHostId* hostId)
|
||||
{
|
||||
char buf[9];
|
||||
u32 num;
|
||||
GetRandomBytes(&num, sizeof(num));
|
||||
sprintf(buf, "%08x", num);
|
||||
memcpy(hostId->data(), buf, 8);
|
||||
char buf[9];
|
||||
u32 num;
|
||||
GetRandomBytes(&num, sizeof(num));
|
||||
sprintf(buf, "%08x", num);
|
||||
memcpy(hostId->data(), buf, 8);
|
||||
}
|
||||
|
||||
static const char* SenderName(sockaddr_in6* addr)
|
||||
{
|
||||
static char buf[INET6_ADDRSTRLEN + 10];
|
||||
inet_ntop(PF_INET6, &addr->sin6_addr, buf, sizeof(buf));
|
||||
sprintf(buf + strlen(buf), ":%d", ntohs(addr->sin6_port));
|
||||
return buf;
|
||||
static char buf[INET6_ADDRSTRLEN + 10];
|
||||
inet_ntop(PF_INET6, &addr->sin6_addr, buf, sizeof(buf));
|
||||
sprintf(buf + strlen(buf), ":%d", ntohs(addr->sin6_port));
|
||||
return buf;
|
||||
}
|
||||
|
||||
static void TrySend(const void* buffer, size_t size, sockaddr_in6* addr)
|
||||
{
|
||||
#if DEBUG
|
||||
printf("-> %d %llu %s\n", ((TraversalPacket*) buffer)->type,
|
||||
(long long)((TraversalPacket*) buffer)->requestId, SenderName(addr));
|
||||
printf("-> %d %llu %s\n", ((TraversalPacket*)buffer)->type,
|
||||
(long long)((TraversalPacket*)buffer)->requestId, SenderName(addr));
|
||||
#endif
|
||||
if ((size_t) sendto(sock, buffer, size, 0, (sockaddr*) addr, sizeof(*addr)) != size)
|
||||
{
|
||||
perror("sendto");
|
||||
}
|
||||
if ((size_t)sendto(sock, buffer, size, 0, (sockaddr*)addr, sizeof(*addr)) != size)
|
||||
{
|
||||
perror("sendto");
|
||||
}
|
||||
}
|
||||
|
||||
static TraversalPacket* AllocPacket(const sockaddr_in6& dest, TraversalRequestId misc = 0)
|
||||
{
|
||||
TraversalRequestId requestId;
|
||||
GetRandomBytes(&requestId, sizeof(requestId));
|
||||
OutgoingPacketInfo* info = &outgoingPackets[requestId];
|
||||
info->dest = dest;
|
||||
info->misc = misc;
|
||||
info->tries = 0;
|
||||
info->sendTime = currentTime;
|
||||
TraversalPacket* result = &info->packet;
|
||||
memset(result, 0, sizeof(*result));
|
||||
result->requestId = requestId;
|
||||
return result;
|
||||
TraversalRequestId requestId;
|
||||
GetRandomBytes(&requestId, sizeof(requestId));
|
||||
OutgoingPacketInfo* info = &outgoingPackets[requestId];
|
||||
info->dest = dest;
|
||||
info->misc = misc;
|
||||
info->tries = 0;
|
||||
info->sendTime = currentTime;
|
||||
TraversalPacket* result = &info->packet;
|
||||
memset(result, 0, sizeof(*result));
|
||||
result->requestId = requestId;
|
||||
return result;
|
||||
}
|
||||
|
||||
static void SendPacket(OutgoingPacketInfo* info)
|
||||
{
|
||||
info->tries++;
|
||||
info->sendTime = currentTime;
|
||||
TrySend(&info->packet, sizeof(info->packet), &info->dest);
|
||||
info->tries++;
|
||||
info->sendTime = currentTime;
|
||||
TrySend(&info->packet, sizeof(info->packet), &info->dest);
|
||||
}
|
||||
|
||||
|
||||
static void ResendPackets()
|
||||
{
|
||||
std::vector<std::pair<TraversalInetAddress, TraversalRequestId>> todoFailures;
|
||||
todoFailures.clear();
|
||||
for (auto it = outgoingPackets.begin(); it != outgoingPackets.end();)
|
||||
{
|
||||
OutgoingPacketInfo* info = &it->second;
|
||||
if (currentTime - info->sendTime >= (u64) (300000 * info->tries))
|
||||
{
|
||||
if (info->tries >= NUMBER_OF_TRIES)
|
||||
{
|
||||
if (info->packet.type == TraversalPacketPleaseSendPacket)
|
||||
{
|
||||
todoFailures.push_back(std::make_pair(info->packet.pleaseSendPacket.address, info->misc));
|
||||
}
|
||||
it = outgoingPackets.erase(it);
|
||||
continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
SendPacket(info);
|
||||
}
|
||||
}
|
||||
++it;
|
||||
}
|
||||
std::vector<std::pair<TraversalInetAddress, TraversalRequestId>> todoFailures;
|
||||
todoFailures.clear();
|
||||
for (auto it = outgoingPackets.begin(); it != outgoingPackets.end();)
|
||||
{
|
||||
OutgoingPacketInfo* info = &it->second;
|
||||
if (currentTime - info->sendTime >= (u64)(300000 * info->tries))
|
||||
{
|
||||
if (info->tries >= NUMBER_OF_TRIES)
|
||||
{
|
||||
if (info->packet.type == TraversalPacketPleaseSendPacket)
|
||||
{
|
||||
todoFailures.push_back(std::make_pair(info->packet.pleaseSendPacket.address, info->misc));
|
||||
}
|
||||
it = outgoingPackets.erase(it);
|
||||
continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
SendPacket(info);
|
||||
}
|
||||
}
|
||||
++it;
|
||||
}
|
||||
|
||||
for (const auto& p : todoFailures)
|
||||
{
|
||||
TraversalPacket* fail = AllocPacket(MakeSinAddr(p.first));
|
||||
fail->type = TraversalPacketConnectFailed;
|
||||
fail->connectFailed.requestId = p.second;
|
||||
fail->connectFailed.reason = TraversalConnectFailedClientDidntRespond;
|
||||
}
|
||||
for (const auto& p : todoFailures)
|
||||
{
|
||||
TraversalPacket* fail = AllocPacket(MakeSinAddr(p.first));
|
||||
fail->type = TraversalPacketConnectFailed;
|
||||
fail->connectFailed.requestId = p.second;
|
||||
fail->connectFailed.reason = TraversalConnectFailedClientDidntRespond;
|
||||
}
|
||||
}
|
||||
|
||||
static void HandlePacket(TraversalPacket* packet, sockaddr_in6* addr)
|
||||
{
|
||||
#if DEBUG
|
||||
printf("<- %d %llu %s\n", packet->type, (long long)packet->requestId, SenderName(addr));
|
||||
printf("<- %d %llu %s\n", packet->type, (long long)packet->requestId, SenderName(addr));
|
||||
#endif
|
||||
bool packetOk = true;
|
||||
switch (packet->type)
|
||||
{
|
||||
case TraversalPacketAck:
|
||||
{
|
||||
auto it = outgoingPackets.find(packet->requestId);
|
||||
if (it == outgoingPackets.end())
|
||||
break;
|
||||
bool packetOk = true;
|
||||
switch (packet->type)
|
||||
{
|
||||
case TraversalPacketAck:
|
||||
{
|
||||
auto it = outgoingPackets.find(packet->requestId);
|
||||
if (it == outgoingPackets.end())
|
||||
break;
|
||||
|
||||
OutgoingPacketInfo* info = &it->second;
|
||||
OutgoingPacketInfo* info = &it->second;
|
||||
|
||||
if (info->packet.type == TraversalPacketPleaseSendPacket)
|
||||
{
|
||||
TraversalPacket* ready = AllocPacket(MakeSinAddr(info->packet.pleaseSendPacket.address));
|
||||
if (packet->ack.ok)
|
||||
{
|
||||
ready->type = TraversalPacketConnectReady;
|
||||
ready->connectReady.requestId = info->misc;
|
||||
ready->connectReady.address = MakeInetAddress(info->dest);
|
||||
}
|
||||
else
|
||||
{
|
||||
ready->type = TraversalPacketConnectFailed;
|
||||
ready->connectFailed.requestId = info->misc;
|
||||
ready->connectFailed.reason = TraversalConnectFailedClientFailure;
|
||||
}
|
||||
}
|
||||
if (info->packet.type == TraversalPacketPleaseSendPacket)
|
||||
{
|
||||
TraversalPacket* ready = AllocPacket(MakeSinAddr(info->packet.pleaseSendPacket.address));
|
||||
if (packet->ack.ok)
|
||||
{
|
||||
ready->type = TraversalPacketConnectReady;
|
||||
ready->connectReady.requestId = info->misc;
|
||||
ready->connectReady.address = MakeInetAddress(info->dest);
|
||||
}
|
||||
else
|
||||
{
|
||||
ready->type = TraversalPacketConnectFailed;
|
||||
ready->connectFailed.requestId = info->misc;
|
||||
ready->connectFailed.reason = TraversalConnectFailedClientFailure;
|
||||
}
|
||||
}
|
||||
|
||||
outgoingPackets.erase(it);
|
||||
break;
|
||||
}
|
||||
case TraversalPacketPing:
|
||||
{
|
||||
auto r = EvictFind(connectedClients, packet->ping.hostId, true);
|
||||
packetOk = r.found;
|
||||
break;
|
||||
}
|
||||
case TraversalPacketHelloFromClient:
|
||||
{
|
||||
u8 ok = packet->helloFromClient.protoVersion <= TraversalProtoVersion;
|
||||
TraversalPacket* reply = AllocPacket(*addr);
|
||||
reply->type = TraversalPacketHelloFromServer;
|
||||
reply->helloFromServer.ok = ok;
|
||||
if (ok)
|
||||
{
|
||||
TraversalHostId hostId;
|
||||
TraversalInetAddress* iaddr;
|
||||
// not that there is any significant change of
|
||||
// duplication, but...
|
||||
GetRandomHostId(&hostId);
|
||||
while (true)
|
||||
{
|
||||
auto r = EvictFind(connectedClients, hostId);
|
||||
if (!r.found)
|
||||
{
|
||||
iaddr = EvictSet(connectedClients, hostId);
|
||||
break;
|
||||
}
|
||||
}
|
||||
outgoingPackets.erase(it);
|
||||
break;
|
||||
}
|
||||
case TraversalPacketPing:
|
||||
{
|
||||
auto r = EvictFind(connectedClients, packet->ping.hostId, true);
|
||||
packetOk = r.found;
|
||||
break;
|
||||
}
|
||||
case TraversalPacketHelloFromClient:
|
||||
{
|
||||
u8 ok = packet->helloFromClient.protoVersion <= TraversalProtoVersion;
|
||||
TraversalPacket* reply = AllocPacket(*addr);
|
||||
reply->type = TraversalPacketHelloFromServer;
|
||||
reply->helloFromServer.ok = ok;
|
||||
if (ok)
|
||||
{
|
||||
TraversalHostId hostId;
|
||||
TraversalInetAddress* iaddr;
|
||||
// not that there is any significant change of
|
||||
// duplication, but...
|
||||
GetRandomHostId(&hostId);
|
||||
while (true)
|
||||
{
|
||||
auto r = EvictFind(connectedClients, hostId);
|
||||
if (!r.found)
|
||||
{
|
||||
iaddr = EvictSet(connectedClients, hostId);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
*iaddr = MakeInetAddress(*addr);
|
||||
*iaddr = MakeInetAddress(*addr);
|
||||
|
||||
reply->helloFromServer.yourAddress = *iaddr;
|
||||
reply->helloFromServer.yourHostId = hostId;
|
||||
}
|
||||
break;
|
||||
}
|
||||
case TraversalPacketConnectPlease:
|
||||
{
|
||||
TraversalHostId& hostId = packet->connectPlease.hostId;
|
||||
auto r = EvictFind(connectedClients, hostId);
|
||||
if (!r.found)
|
||||
{
|
||||
TraversalPacket* reply = AllocPacket(*addr);
|
||||
reply->type = TraversalPacketConnectFailed;
|
||||
reply->connectFailed.requestId = packet->requestId;
|
||||
reply->connectFailed.reason = TraversalConnectFailedNoSuchClient;
|
||||
}
|
||||
else
|
||||
{
|
||||
TraversalPacket* please = AllocPacket(MakeSinAddr(*r.value), packet->requestId);
|
||||
please->type = TraversalPacketPleaseSendPacket;
|
||||
please->pleaseSendPacket.address = MakeInetAddress(*addr);
|
||||
}
|
||||
break;
|
||||
}
|
||||
default:
|
||||
fprintf(stderr, "received unknown packet type %d from %s\n", packet->type, SenderName(addr));
|
||||
}
|
||||
if (packet->type != TraversalPacketAck)
|
||||
{
|
||||
TraversalPacket ack = {};
|
||||
ack.type = TraversalPacketAck;
|
||||
ack.requestId = packet->requestId;
|
||||
ack.ack.ok = packetOk;
|
||||
TrySend(&ack, sizeof(ack), addr);
|
||||
}
|
||||
reply->helloFromServer.yourAddress = *iaddr;
|
||||
reply->helloFromServer.yourHostId = hostId;
|
||||
}
|
||||
break;
|
||||
}
|
||||
case TraversalPacketConnectPlease:
|
||||
{
|
||||
TraversalHostId& hostId = packet->connectPlease.hostId;
|
||||
auto r = EvictFind(connectedClients, hostId);
|
||||
if (!r.found)
|
||||
{
|
||||
TraversalPacket* reply = AllocPacket(*addr);
|
||||
reply->type = TraversalPacketConnectFailed;
|
||||
reply->connectFailed.requestId = packet->requestId;
|
||||
reply->connectFailed.reason = TraversalConnectFailedNoSuchClient;
|
||||
}
|
||||
else
|
||||
{
|
||||
TraversalPacket* please = AllocPacket(MakeSinAddr(*r.value), packet->requestId);
|
||||
please->type = TraversalPacketPleaseSendPacket;
|
||||
please->pleaseSendPacket.address = MakeInetAddress(*addr);
|
||||
}
|
||||
break;
|
||||
}
|
||||
default:
|
||||
fprintf(stderr, "received unknown packet type %d from %s\n", packet->type, SenderName(addr));
|
||||
}
|
||||
if (packet->type != TraversalPacketAck)
|
||||
{
|
||||
TraversalPacket ack = {};
|
||||
ack.type = TraversalPacketAck;
|
||||
ack.requestId = packet->requestId;
|
||||
ack.ack.ok = packetOk;
|
||||
TrySend(&ack, sizeof(ack), addr);
|
||||
}
|
||||
}
|
||||
|
||||
int main()
|
||||
{
|
||||
int rv;
|
||||
int rv;
|
||||
|
||||
urandomFd = open("/dev/urandom", O_RDONLY);
|
||||
if (urandomFd < 0)
|
||||
{
|
||||
perror("open /dev/urandom");
|
||||
return 1;
|
||||
}
|
||||
urandomFd = open("/dev/urandom", O_RDONLY);
|
||||
if (urandomFd < 0)
|
||||
{
|
||||
perror("open /dev/urandom");
|
||||
return 1;
|
||||
}
|
||||
|
||||
sock = socket(PF_INET6, SOCK_DGRAM, 0);
|
||||
if (sock == -1)
|
||||
{
|
||||
perror("socket");
|
||||
return 1;
|
||||
}
|
||||
int no = 0;
|
||||
rv = setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY, &no, sizeof(no));
|
||||
if (rv < 0)
|
||||
{
|
||||
perror("setsockopt IPV6_V6ONLY");
|
||||
return 1;
|
||||
}
|
||||
in6_addr any = IN6ADDR_ANY_INIT;
|
||||
sockaddr_in6 addr;
|
||||
sock = socket(PF_INET6, SOCK_DGRAM, 0);
|
||||
if (sock == -1)
|
||||
{
|
||||
perror("socket");
|
||||
return 1;
|
||||
}
|
||||
int no = 0;
|
||||
rv = setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY, &no, sizeof(no));
|
||||
if (rv < 0)
|
||||
{
|
||||
perror("setsockopt IPV6_V6ONLY");
|
||||
return 1;
|
||||
}
|
||||
in6_addr any = IN6ADDR_ANY_INIT;
|
||||
sockaddr_in6 addr;
|
||||
#ifdef SIN6_LEN
|
||||
addr.sin6_len = sizeof(addr);
|
||||
addr.sin6_len = sizeof(addr);
|
||||
#endif
|
||||
addr.sin6_family = AF_INET6;
|
||||
addr.sin6_port = htons(6262);
|
||||
addr.sin6_flowinfo = 0;
|
||||
addr.sin6_addr = any;
|
||||
addr.sin6_scope_id = 0;
|
||||
addr.sin6_family = AF_INET6;
|
||||
addr.sin6_port = htons(6262);
|
||||
addr.sin6_flowinfo = 0;
|
||||
addr.sin6_addr = any;
|
||||
addr.sin6_scope_id = 0;
|
||||
|
||||
rv = bind(sock, (sockaddr*) &addr, sizeof(addr));
|
||||
if (rv < 0)
|
||||
{
|
||||
perror("bind");
|
||||
return 1;
|
||||
}
|
||||
rv = bind(sock, (sockaddr*)&addr, sizeof(addr));
|
||||
if (rv < 0)
|
||||
{
|
||||
perror("bind");
|
||||
return 1;
|
||||
}
|
||||
|
||||
timeval tv;
|
||||
tv.tv_sec = 0;
|
||||
tv.tv_usec = 300000;
|
||||
rv = setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
|
||||
if (rv < 0)
|
||||
{
|
||||
perror("setsockopt SO_RCVTIMEO");
|
||||
return 1;
|
||||
}
|
||||
timeval tv;
|
||||
tv.tv_sec = 0;
|
||||
tv.tv_usec = 300000;
|
||||
rv = setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
|
||||
if (rv < 0)
|
||||
{
|
||||
perror("setsockopt SO_RCVTIMEO");
|
||||
return 1;
|
||||
}
|
||||
|
||||
while (true)
|
||||
{
|
||||
sockaddr_in6 raddr;
|
||||
socklen_t addrLen = sizeof(raddr);
|
||||
TraversalPacket packet;
|
||||
// note: switch to recvmmsg (yes, mmsg) if this becomes
|
||||
// expensive
|
||||
rv = recvfrom(sock, &packet, sizeof(packet), 0, (sockaddr*) &raddr, &addrLen);
|
||||
if (gettimeofday(&tv, nullptr) < 0)
|
||||
{
|
||||
perror("gettimeofday");
|
||||
exit(1);
|
||||
}
|
||||
currentTime = (u64) tv.tv_sec * 1000000 + tv.tv_usec;
|
||||
if (rv < 0)
|
||||
{
|
||||
if (errno != EINTR && errno != EAGAIN)
|
||||
{
|
||||
perror("recvfrom");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
else if ((size_t) rv < sizeof(packet))
|
||||
{
|
||||
fprintf(stderr, "received short packet from %s\n", SenderName(&raddr));
|
||||
}
|
||||
else
|
||||
{
|
||||
HandlePacket(&packet, &raddr);
|
||||
}
|
||||
ResendPackets();
|
||||
}
|
||||
while (true)
|
||||
{
|
||||
sockaddr_in6 raddr;
|
||||
socklen_t addrLen = sizeof(raddr);
|
||||
TraversalPacket packet;
|
||||
// note: switch to recvmmsg (yes, mmsg) if this becomes
|
||||
// expensive
|
||||
rv = recvfrom(sock, &packet, sizeof(packet), 0, (sockaddr*)&raddr, &addrLen);
|
||||
if (gettimeofday(&tv, nullptr) < 0)
|
||||
{
|
||||
perror("gettimeofday");
|
||||
exit(1);
|
||||
}
|
||||
currentTime = (u64)tv.tv_sec * 1000000 + tv.tv_usec;
|
||||
if (rv < 0)
|
||||
{
|
||||
if (errno != EINTR && errno != EAGAIN)
|
||||
{
|
||||
perror("recvfrom");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
else if ((size_t)rv < sizeof(packet))
|
||||
{
|
||||
fprintf(stderr, "received short packet from %s\n", SenderName(&raddr));
|
||||
}
|
||||
else
|
||||
{
|
||||
HandlePacket(&packet, &raddr);
|
||||
}
|
||||
ResendPackets();
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user