generated from AfonsoCMSousa/CPP-Template
85 lines
2.1 KiB
C
85 lines
2.1 KiB
C
#include "socket.h"
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
|
|
int connect_udp_socket(const char* ip, uint16_t port) {
|
|
#ifdef _WIN32
|
|
static int wsa_initialized = 0;
|
|
if (!wsa_initialized) {
|
|
WSADATA wsaData;
|
|
if (WSAStartup(MAKEWORD(2,2), &wsaData) != 0) {
|
|
fprintf(stderr, "WSAStartup failed\n");
|
|
return -1;
|
|
}
|
|
wsa_initialized = 1;
|
|
}
|
|
#endif
|
|
|
|
int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
|
|
if (sockfd < 0) {
|
|
perror("socket creation failed");
|
|
return -1;
|
|
}
|
|
|
|
struct sockaddr_in servaddr;
|
|
memset(&servaddr, 0, sizeof(servaddr));
|
|
|
|
servaddr.sin_family = AF_INET;
|
|
servaddr.sin_port = htons(port);
|
|
servaddr.sin_addr.s_addr = inet_addr(ip);
|
|
|
|
if (connect(sockfd, (const struct sockaddr*)&servaddr, sizeof(servaddr)) < 0) {
|
|
perror("connection to the server failed");
|
|
CLOSESOCKET(sockfd);
|
|
return -1;
|
|
}
|
|
|
|
return sockfd;
|
|
}
|
|
|
|
int bind_udp_socket(int sockfd, const char* ip, uint16_t port) {
|
|
struct sockaddr_in servaddr;
|
|
memset(&servaddr, 0, sizeof(servaddr));
|
|
|
|
servaddr.sin_family = AF_INET;
|
|
servaddr.sin_addr.s_addr = inet_addr(ip);
|
|
servaddr.sin_port = htons(port);
|
|
|
|
if (bind(sockfd, (const struct sockaddr*)&servaddr, sizeof(servaddr)) < 0) {
|
|
perror("bind failed");
|
|
CLOSESOCKET(sockfd);
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
ssize_t send_udp_message(int sockfd, const char *message, const char *dest_ip, uint16_t dest_port) {
|
|
struct sockaddr_in destaddr;
|
|
memset(&destaddr, 0, sizeof(destaddr));
|
|
|
|
destaddr.sin_family = AF_INET;
|
|
destaddr.sin_port = htons(dest_port);
|
|
destaddr.sin_addr.s_addr = inet_addr(dest_ip);
|
|
|
|
#ifdef _WIN32
|
|
int n = sendto(sockfd, message, (int)strlen(message), 0, (const struct sockaddr*)&destaddr, sizeof(destaddr));
|
|
#else
|
|
ssize_t n = sendto(sockfd, message, strlen(message), 0, (const struct sockaddr*)&destaddr, sizeof(destaddr));
|
|
#endif
|
|
|
|
if (n < 0) {
|
|
perror("sendto failed");
|
|
return -1;
|
|
}
|
|
|
|
return n;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
void cleanup_sockets() {
|
|
WSACleanup();
|
|
}
|
|
#endif
|
|
|