56 lines
1.4 KiB
C++

#include "net.hpp"
#include <sys/types.h>
Socket::Socket() {
sock = socket(AF_INET, SOCK_DGRAM, 0);
if (sock < 0) {
throw runtime_error("Failed to create socket");
}
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
}
Socket::~Socket() {
close(sock);
}
void Socket::connect(const char *ip, uint16_t port) {
server_addr.sin_port = htons(port);
if (inet_pton(AF_INET, ip, &server_addr.sin_addr) <= 0) {
throw runtime_error("Invalid IP address");
}
}
void Socket::send() {
ssize_t sent_bytes = sendto(sock, &packet_data, sizeof(packet_data), 0,
(struct sockaddr *)&server_addr, sizeof(server_addr));
if (sent_bytes < 0) {
throw runtime_error("Failed to send data");
}
}
void Socket::send(const api_packet &data) {
ssize_t sent_bytes = sendto(sock, &data, sizeof(data), 0,
(struct sockaddr *)&server_addr, sizeof(server_addr));
if (sent_bytes < 0) {
throw runtime_error("Failed to send data");
}
}
void Socket::set_packet(const api_packet &data) {
packet_data = data;
}
api_packet Socket::create_packet(uint8_t tracker_id) {
api_packet pkt;
memset(&pkt, 0, sizeof(pkt));
pkt.message_type = 65; // Handshake message type
pkt.tracker_id = tracker_id;
return pkt;
}
api_packet Socket::get_packet() {
return packet_data;
}