364 lines
11 KiB
C++
364 lines
11 KiB
C++
#include "Participant.h"
|
|
|
|
// #include <ws2tcpip.h>
|
|
|
|
#define BUF_SIZE 1024
|
|
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
#pragma comment(lib, "ws2_32.lib")
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
#include <arpa/inet.h>
|
|
#include <netinet/in.h>
|
|
#include <sys/socket.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h> // For fcntl
|
|
#endif
|
|
|
|
Participant::Participant() {}
|
|
|
|
Participant::Participant(const char *ipAddress, int port) {
|
|
this->ipAddress = ipAddress;
|
|
this->port = port;
|
|
|
|
this->participants.push_back(this);
|
|
|
|
int randomPort = (rand() % (65535 - 49152 + 1)) + 49152;
|
|
SetupUDP(randomPort, ipAddress, port);
|
|
}
|
|
|
|
void Passer::Control::Participant::SetupUDP(int localPort,
|
|
const char *remoteIpAddress,
|
|
int remotePort) {
|
|
// Create a UDP socket
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
// Windows-specific Winsock initialization
|
|
WSADATA wsaData;
|
|
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
|
|
std::cerr << "WSAStartup failed" << std::endl;
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
this->sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
this->sock = socket(AF_INET, SOCK_DGRAM, 0);
|
|
#endif
|
|
|
|
if (this->sock < 0) {
|
|
std::cerr << "Error creating socket" << std::endl;
|
|
return;
|
|
}
|
|
|
|
// Set the socket to non-blocking mode
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
u_long mode = 1; // 1 to enable non-blocking socket
|
|
ioctlsocket(this->sock, FIONBIO, &mode);
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
int flags = fcntl(this->sock, F_GETFL, 0);
|
|
fcntl(this->sock, F_SETFL, flags | O_NONBLOCK);
|
|
#endif
|
|
|
|
if (remotePort != 0) {
|
|
// Set up the address to send to
|
|
memset(&remote_addr, 0, sizeof(remote_addr));
|
|
remote_addr.sin_family = AF_INET;
|
|
remote_addr.sin_port = htons((u_short)remotePort);
|
|
if (inet_pton(AF_INET, remoteIpAddress, &remote_addr.sin_addr) <= 0) {
|
|
std::cerr << "Invalid address" << std::endl;
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
closesocket(sock);
|
|
WSACleanup();
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
close(sock);
|
|
#endif
|
|
return;
|
|
}
|
|
}
|
|
|
|
// Set up the receiving address
|
|
memset(&server_addr, 0, sizeof(server_addr));
|
|
server_addr.sin_family = AF_INET;
|
|
server_addr.sin_port = htons((u_short)localPort);
|
|
if (inet_pton(AF_INET, "0.0.0.0", &server_addr.sin_addr) <= 0) {
|
|
std::cerr << "Invalid address" << std::endl;
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
closesocket(sock);
|
|
WSACleanup();
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
close(sock);
|
|
#endif
|
|
return;
|
|
}
|
|
|
|
// Bind the socket to the specified port
|
|
if (bind(this->sock, (const struct sockaddr *)&server_addr,
|
|
sizeof(server_addr)) < 0) {
|
|
std::cerr << "Bind failed" << std::endl;
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
closesocket(sock);
|
|
WSACleanup();
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
close(sock);
|
|
#endif
|
|
return;
|
|
}
|
|
|
|
// Bind the socket to the specified port
|
|
// if (bind(this->sock, (const struct sockaddr *)&server_addr,
|
|
// sizeof(server_addr)) < 0) {
|
|
// std::cerr << "Bind failed" << std::endl;
|
|
// #if defined(_WIN32) || defined(_WIN64)
|
|
// closesocket(sock);
|
|
// WSACleanup();
|
|
// #elif defined(__unix__) || defined(__APPLE__)
|
|
// close(sock);
|
|
// #endif
|
|
// return;
|
|
// }
|
|
|
|
// Set up the broadcast address
|
|
memset(&broadcast_addr, 0, sizeof(broadcast_addr));
|
|
broadcast_addr.sin_family = AF_INET;
|
|
broadcast_addr.sin_port = htons((u_short)remotePort);
|
|
if (inet_pton(AF_INET, "255.255.255.255", &broadcast_addr.sin_addr) <= 0) {
|
|
std::cerr << "Invalid address" << std::endl;
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
closesocket(sock);
|
|
WSACleanup();
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
close(sock);
|
|
#endif
|
|
return;
|
|
}
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
BOOL broadcast = TRUE;
|
|
if (setsockopt(sock, SOL_SOCKET, SO_BROADCAST, (char *)&broadcast,
|
|
sizeof(broadcast)) == SOCKET_ERROR) {
|
|
std::cerr << "Setting socket option for broadcast failed" << std::endl;
|
|
}
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
int broadcastEnable = 1;
|
|
if (setsockopt(sock, SOL_SOCKET, SO_BROADCAST, &broadcastEnable, sizeof(broadcastEnable)) < 0) {
|
|
std::cerr << "Failed to set socket options" << std::endl;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void Participant::Update(unsigned long currentTimeMs) {
|
|
if (currentTimeMs > this->nextPublishMe) {
|
|
ClientMsg* msg = new ClientMsg(this->networkId);
|
|
this->Publish(msg);
|
|
delete msg;
|
|
std::cout << this->name << " published ClientMsg\n";
|
|
this->nextPublishMe = currentTimeMs + this->publishInterval;
|
|
}
|
|
this->ReceiveUDP();
|
|
|
|
Thing::UpdateAll(currentTimeMs);
|
|
}
|
|
|
|
void Participant::ReceiveUDP() {
|
|
// char ip_str[INET_ADDRSTRLEN];
|
|
// inet_ntop(AF_INET, &(server_addr.sin_addr), ip_str, INET_ADDRSTRLEN);
|
|
// std::cout << this->name << " Receive on " << ip_str << ":"
|
|
// << ntohs(server_addr.sin_port) << "\n";
|
|
|
|
sockaddr_in client_addr;
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
int len = sizeof(client_addr);
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
socklen_t len = sizeof(client_addr);
|
|
#endif
|
|
int packetSize = recvfrom(this->sock, buffer, sizeof(buffer), 0,
|
|
(struct sockaddr *)&client_addr, &len);
|
|
// std::cout << "received data " << packetSize << "\n";
|
|
if (packetSize < 0) {
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
int error_code = WSAGetLastError(); // Get the error code on Windows
|
|
if (error_code != WSAEWOULDBLOCK)
|
|
std::cerr << "recvfrom failed with error: " << error_code << std::endl;
|
|
#else
|
|
//std::cerr << "recvfrom failed with error: " << packetSize << std::endl;
|
|
#endif
|
|
} else if (packetSize > 0) {
|
|
char sender_ipAddress[INET_ADDRSTRLEN];
|
|
inet_ntop(AF_INET, &(client_addr.sin_addr), sender_ipAddress,
|
|
INET_ADDRSTRLEN);
|
|
int sender_port = ntohs(client_addr.sin_port);
|
|
|
|
Participant *remoteParticipant =
|
|
this->GetParticipant(sender_ipAddress, sender_port);
|
|
if (remoteParticipant == nullptr) {
|
|
remoteParticipant = this->AddParticipant(sender_ipAddress, sender_port);
|
|
// std::cout << "New sender " << sender_ipAddress << ":"
|
|
// << sender_port << "\n";
|
|
// std::cout << "New remote participant " << remoteParticipant->ipAddress << ":"
|
|
// << remoteParticipant->port << " "
|
|
// << (int)remoteParticipant->networkId << "\n";
|
|
}
|
|
|
|
ReceiveData(packetSize, remoteParticipant);
|
|
}
|
|
}
|
|
|
|
Participant *Participant::GetParticipant(const char *ipAddress, int port) {
|
|
for (Participant *participant : this->participants) {
|
|
if (participant->ipAddress == ipAddress && participant->port == port)
|
|
return participant;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
Participant *Participant::AddParticipant(const char *ipAddress, int port) {
|
|
Participant *participant = new Participant(ipAddress, port);
|
|
participant->networkId = (unsigned char)this->participants.size();
|
|
this->participants.push_back(participant);
|
|
return participant;
|
|
}
|
|
|
|
#pragma region Send
|
|
|
|
void Participant::SendThingInfo(Thing *thing) {
|
|
std::cout << "Send thing info\n";
|
|
IMessage *msg = new ThingMsg(this->networkId, thing);
|
|
this->Send(msg);
|
|
delete msg;
|
|
msg = new NameMsg(this->networkId, thing);
|
|
this->Send(msg);
|
|
delete msg;
|
|
msg = new ModelUrlMsg(this->networkId, thing);
|
|
this->Send(msg);
|
|
delete msg;
|
|
// this->Send(&ModelUrlMsg(this->networkId, thing));
|
|
}
|
|
|
|
bool Participant::Send(IMessage *msg) {
|
|
// Send the message to the specified address and port
|
|
int bufferSize = msg->Serialize(this->buffer);
|
|
// std::cout << "buffer size " << bufferSize << "\n";
|
|
if (bufferSize <= 0)
|
|
return true;
|
|
|
|
char ip_str[INET_ADDRSTRLEN];
|
|
inet_ntop(AF_INET, &(remote_addr.sin_addr), ip_str, INET_ADDRSTRLEN);
|
|
std::cout << "Send to " << ip_str << ":" << ntohs(remote_addr.sin_port)
|
|
<< "\n";
|
|
int sent_bytes = sendto(sock, this->buffer, bufferSize, 0,
|
|
(struct sockaddr *)&remote_addr, sizeof(remote_addr));
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
if (sent_bytes <= SOCKET_ERROR) {
|
|
int error_code = WSAGetLastError(); // Get the error code on Windows
|
|
std::cerr << "sendto failed with error: " << error_code << std::endl;
|
|
closesocket(sock);
|
|
WSACleanup();
|
|
return false;
|
|
}
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
if (sent_bytes < 0) {
|
|
std::cerr << "sendto failed with error: " << sent_bytes << std::endl;
|
|
close(sock);
|
|
return false;
|
|
}
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
bool Participant::Publish(IMessage *msg) {
|
|
int bufferSize = msg->Serialize(this->buffer);
|
|
if (bufferSize <= 0)
|
|
return true;
|
|
|
|
char ip_str[INET_ADDRSTRLEN];
|
|
inet_ntop(AF_INET, &(broadcast_addr.sin_addr), ip_str, INET_ADDRSTRLEN);
|
|
std::cout << "Publish to " << ip_str << ":" << ntohs(broadcast_addr.sin_port)
|
|
<< "\n";
|
|
int sent_bytes =
|
|
sendto(sock, this->buffer, bufferSize, 0,
|
|
(struct sockaddr *)&broadcast_addr, sizeof(broadcast_addr));
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
if (sent_bytes <= SOCKET_ERROR) {
|
|
int error_code = WSAGetLastError(); // Get the error code on Windows
|
|
std::cerr << "sendto failed with error: " << error_code << std::endl;
|
|
closesocket(sock);
|
|
WSACleanup();
|
|
return false;
|
|
}
|
|
#elif defined(__unix__) || defined(__APPLE__)
|
|
if (sent_bytes < 0) {
|
|
std::cerr << "sendto failed with error: " << sent_bytes << std::endl;
|
|
close(sock);
|
|
return false;
|
|
}
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
// Send
|
|
#pragma endregion
|
|
|
|
#pragma region Receive
|
|
|
|
void Participant::ReceiveData(unsigned char bufferSize,
|
|
Participant *remoteParticipant) {
|
|
unsigned char msgId = this->buffer[0];
|
|
switch (msgId) {
|
|
case ClientMsg::id: {
|
|
ClientMsg* msg = new ClientMsg(this->buffer);
|
|
Process(remoteParticipant, msg);
|
|
delete msg;
|
|
}
|
|
case NetworkIdMsg::id: {
|
|
NetworkIdMsg* msg = new NetworkIdMsg(this->buffer);
|
|
Process(remoteParticipant, msg);
|
|
delete msg;
|
|
} break;
|
|
case InvestigateMsg::id: {
|
|
InvestigateMsg msg = InvestigateMsg(this->buffer);
|
|
ProcessInvestigateMsg(msg);
|
|
} break;
|
|
case ThingMsg::id: {
|
|
ThingMsg msg = ThingMsg(this->buffer);
|
|
ProcessThingMsg(msg);
|
|
} break;
|
|
case PoseMsg::id: {
|
|
PoseMsg msg = PoseMsg(this->buffer);
|
|
ProcessPoseMsg(msg);
|
|
} break;
|
|
case CustomMsg::id: {
|
|
CustomMsg msg = CustomMsg(this->buffer);
|
|
ProcessCustomMsg(msg);
|
|
} break;
|
|
};
|
|
}
|
|
|
|
void Participant::Process(Participant *sender, ClientMsg *msg) {}
|
|
|
|
void Participant::Process(Participant *sender, NetworkIdMsg *msg) {
|
|
std::cout << this->name << " receive network id " << (int)this->networkId
|
|
<< " " << (int)msg->networkId << "\n";
|
|
if (this->networkId != msg->networkId) {
|
|
this->networkId = msg->networkId;
|
|
Thing **allThings = Thing::GetAllThings();
|
|
for (uint16_t ix = 0; ix < THING_STORE_SIZE; ix++) {
|
|
Thing *thing = allThings[ix];
|
|
if (thing == nullptr)
|
|
continue;
|
|
|
|
sender->SendThingInfo(thing);
|
|
}
|
|
}
|
|
}
|
|
|
|
void Participant::ProcessInvestigateMsg(InvestigateMsg msg) {}
|
|
|
|
void Participant::ProcessThingMsg(ThingMsg msg) {}
|
|
|
|
void Passer::Control::Participant::ProcessPoseMsg(PoseMsg msg) {}
|
|
|
|
void Participant::ProcessCustomMsg(CustomMsg msg) {}
|
|
|
|
// Receive
|
|
#pragma endregion |