generated from AfonsoCMSousa/CPP-Template
117 lines
3.2 KiB
C++
117 lines
3.2 KiB
C++
#include "net.hpp"
|
|
#include "log.c"
|
|
#include <sys/types.h>
|
|
|
|
Socket::Socket() {
|
|
sock_server = socket(AF_INET, SOCK_DGRAM, 0);
|
|
sock_unix = -1;
|
|
if (sock_server < 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_server);
|
|
close(sock_unix);
|
|
}
|
|
|
|
void Socket::connect_server(const char *ip, uint16_t port) {
|
|
this->server_port_input = 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::bind_server(const char *ip, uint16_t port) {
|
|
struct sockaddr_in bind_addr;
|
|
memset(&bind_addr, 0, sizeof(bind_addr));
|
|
bind_addr.sin_family = AF_INET;
|
|
bind_addr.sin_port = htons(port);
|
|
bind_addr.sin_addr.s_addr = inet_addr(ip);
|
|
|
|
if (bind(sock_server, (const struct sockaddr *)&bind_addr, sizeof(bind_addr)) < 0) {
|
|
log_error("Failed to bind UDP socket on %s:%d\n", ip, port);
|
|
throw runtime_error("Failed to bind UDP socket");
|
|
}
|
|
|
|
log_info("Successfully bound UDP socket on %s:%d\n", ip, port);
|
|
}
|
|
|
|
void Socket::connect_unix(const char *ip, uint16_t port) {
|
|
sock_unix = socket(AF_UNIX, SOCK_STREAM, 0);
|
|
if (sock_unix < 0) {
|
|
throw std::runtime_error("Failed to create UNIX socket");
|
|
}
|
|
|
|
this->server_addr_unix.sun_family = AF_UNIX;
|
|
strncpy(this->server_addr_unix.sun_path, ip, sizeof(this->server_addr_unix.sun_path) - 1);
|
|
|
|
if (connect(sock_unix, (struct sockaddr *)&server_addr_unix, sizeof(server_addr_unix)) < 0) {
|
|
throw std::runtime_error("Failed to connect to UNIX socket");
|
|
}
|
|
}
|
|
|
|
void Socket::send_server() {
|
|
ssize_t sent_bytes = sendto(sock_server, &packet_data, sizeof(packet_data), 0, (struct sockaddr *)&server_addr, sizeof(server_addr));
|
|
if (sent_bytes < 0) {
|
|
throw runtime_error("Failed to send data");
|
|
}
|
|
}
|
|
|
|
ssize_t Socket::receive_server(void *buffer, size_t len) {
|
|
ssize_t recv_bytes = recv(sock_server, buffer, len, 0);
|
|
if (recv_bytes < 0) {
|
|
log_error("Failed to receive data from server socket\n");
|
|
return -1;
|
|
}
|
|
return recv_bytes;
|
|
}
|
|
|
|
void Socket::send_unix() {
|
|
ssize_t sent_bytes = send(sock_unix, &packet_data, sizeof(packet_data), 0);
|
|
if (sent_bytes < 0) {
|
|
throw runtime_error("Failed to send data to UNIX socket");
|
|
}
|
|
}
|
|
|
|
void Socket::send_server(const api_packet &data) {
|
|
ssize_t sent_bytes = sendto(sock_server, &data, sizeof(data), 0, (struct sockaddr *)&server_addr, sizeof(server_addr));
|
|
if (sent_bytes < 0) {
|
|
throw runtime_error("Failed to send data");
|
|
}
|
|
}
|
|
|
|
void Socket::send_server(const void *data, size_t len) {
|
|
ssize_t sent_bytes = sendto(sock_server, data, len, 0, (struct sockaddr *)&server_addr, sizeof(server_addr));
|
|
if (sent_bytes < 0) {
|
|
throw runtime_error("Failed to send data");
|
|
}
|
|
}
|
|
|
|
void Socket::send_unix(const api_packet &data) {
|
|
ssize_t sent_bytes = send(sock_unix, &data, sizeof(data), 0);
|
|
if (sent_bytes < 0) {
|
|
throw runtime_error("Failed to send data to UNIX socket");
|
|
}
|
|
}
|
|
|
|
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;
|
|
}
|