152 lines
3.9 KiB
C
152 lines
3.9 KiB
C
#include "net.h"
|
|
#include "log.h"
|
|
#include <stdio.h>
|
|
|
|
typedef struct {
|
|
struct sockaddr_in addr;
|
|
int socket;
|
|
socklen_t addr_len;
|
|
} net_info;
|
|
|
|
static net_info broadcast_info;
|
|
static net_info listen_info;
|
|
static net_info peer_info;
|
|
static int PORT;
|
|
|
|
static int init = 0;
|
|
|
|
int net_init(unsigned int port)
|
|
{
|
|
PORT = port;
|
|
// broadcast socket init
|
|
broadcast_info.socket = socket(AF_INET, SOCK_DGRAM, 0);
|
|
if (broadcast_info.socket < 0) {
|
|
log_message(LOG_ERROR, "Error creating broadcast socket");
|
|
close(broadcast_info.socket);
|
|
return 0;
|
|
}
|
|
|
|
int broadcast_enable = 1;
|
|
if (setsockopt(broadcast_info.socket, SOL_SOCKET, SO_BROADCAST, &broadcast_enable, sizeof(broadcast_enable)) < 0) return 0;
|
|
|
|
memset(&broadcast_info.addr, 0, sizeof(broadcast_info.addr));
|
|
broadcast_info.addr.sin_family = AF_INET;
|
|
broadcast_info.addr.sin_addr.s_addr = inet_addr("255.255.255.255");
|
|
broadcast_info.addr.sin_port = htons(port);
|
|
|
|
broadcast_info.addr_len = sizeof(broadcast_info.addr);
|
|
|
|
// listen socket init
|
|
listen_info.socket = socket(AF_INET, SOCK_DGRAM, 0);
|
|
if (listen_info.socket < 0) {
|
|
log_message(LOG_ERROR, "Error creating listen socket", strerror(errno));
|
|
close(listen_info.socket);
|
|
return 0;
|
|
}
|
|
|
|
int reuse_enable = 1;
|
|
if (setsockopt(listen_info.socket, SOL_SOCKET, SO_REUSEADDR, &reuse_enable, sizeof(reuse_enable)) < 0) return 0;
|
|
|
|
memset(&listen_info.addr, 0, sizeof(listen_info.addr));
|
|
listen_info.addr.sin_family = AF_INET;
|
|
listen_info.addr.sin_addr.s_addr = INADDR_BROADCAST;
|
|
listen_info.addr.sin_port = htons(port);
|
|
|
|
listen_info.addr_len =sizeof(listen_info.addr);
|
|
|
|
struct timeval tv;
|
|
tv.tv_sec = 0;
|
|
tv.tv_usec = 100000;
|
|
setsockopt(listen_info.socket, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
|
|
|
|
if (bind(listen_info.socket, (const struct sockaddr*)&listen_info.addr, sizeof(listen_info.addr)) < 0) {
|
|
log_message(LOG_ERROR, "Socket bind failed: %s", strerror(errno));
|
|
close(listen_info.socket);
|
|
return 0;
|
|
}
|
|
|
|
peer_info.socket = -1;
|
|
|
|
init = 1;
|
|
return 1;
|
|
}
|
|
|
|
int net_broadcast()
|
|
{
|
|
if (!init) return 0;;
|
|
const char* message = "broadcast message";
|
|
int retv = sendto(broadcast_info.socket, message,
|
|
strlen(message), 0,
|
|
(struct sockaddr*)&broadcast_info.addr,
|
|
broadcast_info.addr_len);
|
|
if (retv < 0) {
|
|
log_message(LOG_ERROR, "Failed to send broadcast message: %s", strerror(errno));
|
|
log_message(LOG_ERROR, "retv: %i", retv);
|
|
return 0;
|
|
}
|
|
log_message(LOG_INFO, "Broadcast message sent");
|
|
return 1;
|
|
}
|
|
|
|
int net_listen_clients()
|
|
{
|
|
if (!init) return 0;
|
|
|
|
if (peer_info.socket >= 0) {
|
|
log_message(LOG_WARNING, "Peer socket already created");
|
|
log_message(LOG_INFO,
|
|
"Peer socket address:\n%s:%d",
|
|
inet_ntoa(peer_info.addr.sin_addr),
|
|
ntohs(peer_info.addr.sin_port));
|
|
return 1;
|
|
}
|
|
|
|
peer_info.socket = socket(AF_INET, SOCK_DGRAM, 0);
|
|
if (peer_info.socket < 0) {
|
|
log_message(LOG_ERROR, "Error creating listen socket", strerror(errno));
|
|
close(peer_info.socket);
|
|
return 0;
|
|
}
|
|
|
|
char buffer[64];
|
|
|
|
memset(&peer_info.addr, 0, sizeof(peer_info.addr));
|
|
peer_info.addr.sin_family = AF_INET;
|
|
peer_info.addr.sin_addr.s_addr = INADDR_ANY;
|
|
peer_info.addr.sin_port = htons(PORT);
|
|
|
|
peer_info.addr_len = sizeof(peer_info.addr);
|
|
|
|
int retv = recvfrom(listen_info.socket, buffer, sizeof(buffer), 0,
|
|
(struct sockaddr*)&peer_info.addr,
|
|
&peer_info.addr_len);
|
|
|
|
if (retv < 0) {
|
|
log_message(LOG_ERROR, "Failed to receive broadcast message: %s", strerror(errno));
|
|
close(peer_info.socket);
|
|
peer_info.socket = -1;
|
|
return 0;
|
|
}
|
|
|
|
buffer[retv] = '\0';
|
|
log_message(LOG_INFO,
|
|
"Broadcast message received from:\n%s:%d",
|
|
inet_ntoa(peer_info.addr.sin_addr),
|
|
ntohs(peer_info.addr.sin_port));
|
|
log_message(LOG_INFO, "%s", buffer);
|
|
|
|
return 1;
|
|
}
|
|
|
|
int net_send()
|
|
{
|
|
if (!init) return 0;
|
|
return 1;
|
|
}
|
|
|
|
int net_receive()
|
|
{
|
|
if (!init) return 0;
|
|
return 1;
|
|
}
|