Alex 'AdUser' Z
8 years ago
1 changed files with 162 additions and 0 deletions
@ -0,0 +1,162 @@ |
|||||||
|
/* Copyright 2016 Alex 'AdUser' Z (ad_user@runbox.com)
|
||||||
|
* |
||||||
|
* This program is free software; you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU General Public License version 2 as |
||||||
|
* published by the Free Software Foundation. |
||||||
|
*/ |
||||||
|
#include "source.h" |
||||||
|
|
||||||
|
#include <sys/types.h> |
||||||
|
#include <fcntl.h> |
||||||
|
#include <sys/socket.h> |
||||||
|
#include <netinet/in.h> |
||||||
|
#include <arpa/inet.h> |
||||||
|
#include <netdb.h> |
||||||
|
|
||||||
|
#define DEFAULT_MCAST_ADDR "239.255.186.1" |
||||||
|
#define DEFAULT_MCAST_PORT 3370 |
||||||
|
|
||||||
|
struct _config { |
||||||
|
char name[32]; |
||||||
|
char error[256]; |
||||||
|
void (*errcb)(char *errstr); |
||||||
|
char addr[INET_STRADDRLEN]; |
||||||
|
char iface[16]; |
||||||
|
uint16_t port; |
||||||
|
int sock; |
||||||
|
}; |
||||||
|
|
||||||
|
static void |
||||||
|
errcb_stub(char *str) { |
||||||
|
assert(str != NULL); |
||||||
|
(void)(str); |
||||||
|
} |
||||||
|
|
||||||
|
cfg_t * |
||||||
|
create(const char *init) { |
||||||
|
cfg_t *cfg = NULL; |
||||||
|
assert(init != NULL); |
||||||
|
if ((cfg = calloc(1, sizeof(cfg_t))) == NULL) |
||||||
|
return NULL; |
||||||
|
strlcpy(cfg->name, init, sizeof(cfg->name)); |
||||||
|
strlcpy(cfg->addr, DEFAULT_MCAST_ADDR, sizeof(cfg->addr)); |
||||||
|
cfg->port = DEFAULT_MCAST_PORT; |
||||||
|
cfg->errcb = &errcb_stub; |
||||||
|
return cfg; |
||||||
|
} |
||||||
|
|
||||||
|
bool |
||||||
|
config(cfg_t *cfg, const char *key, const char *value) { |
||||||
|
assert(cfg != NULL); |
||||||
|
assert(key != NULL); |
||||||
|
assert(value != NULL); |
||||||
|
|
||||||
|
if (strcmp(key, "address") == 0) { |
||||||
|
if (strncmp(value, "239.255.", 8) != 0) { |
||||||
|
strlcpy(cfg->error, "mcast address should be inside 239.255.0.0/16 block"); |
||||||
|
return false; |
||||||
|
} |
||||||
|
strlcpy(cfg->addr, value, sizeof(cfg->addr)); |
||||||
|
return true; |
||||||
|
} |
||||||
|
if (strcmp(key, "port") == 0) { |
||||||
|
cfg->port = atoi(value); |
||||||
|
return true; |
||||||
|
} |
||||||
|
if (strcmp(key, "iface") == 0) { |
||||||
|
strlcpy(cfg->iface, value, sizeof(cfg->iface)); |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
bool |
||||||
|
ready(cfg_t *cfg) { |
||||||
|
assert(cfg != NULL); |
||||||
|
|
||||||
|
if (cfg->addr[0] != '\0') |
||||||
|
return true; |
||||||
|
|
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
char * |
||||||
|
error(cfg_t *cfg) { |
||||||
|
assert(cfg != NULL); |
||||||
|
|
||||||
|
return cfg->error; |
||||||
|
} |
||||||
|
|
||||||
|
void |
||||||
|
errcb(cfg_t *cfg, void (*cb)(char *errstr)) { |
||||||
|
assert(cfg != NULL); |
||||||
|
assert(cb != NULL); |
||||||
|
|
||||||
|
cfg->errcb = cb; |
||||||
|
} |
||||||
|
|
||||||
|
bool |
||||||
|
start(cfg_t *cfg) { |
||||||
|
struct addrinfo hints; |
||||||
|
struct addrinfo *result; |
||||||
|
int opt; |
||||||
|
|
||||||
|
assert(cfg != NULL); |
||||||
|
|
||||||
|
memset(&hints, 0, sizeof(struct addrinfo)); |
||||||
|
hints.ai_family = AF_INET; |
||||||
|
hints.ai_socktype = SOCK_DGRAM; |
||||||
|
hints.ai_protocol = 0; |
||||||
|
|
||||||
|
int ret = getaddrinfo(cfg->addr, cfg->port, &hints, &result); |
||||||
|
for (struct addrinfo *rp = result; rp != NULL; rp = rp->ai_next) { |
||||||
|
cfg->sock = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol); |
||||||
|
if (cfg->sock == -1) |
||||||
|
continue; |
||||||
|
if ((opt = fcntl(cfg->sock, F_GETFL, 0)) < 0) |
||||||
|
continue; |
||||||
|
fcntl(cfg->sock, F_SETFL, opt | O_NONBLOCK); |
||||||
|
opt = 1; |
||||||
|
setsockopt(cfg->sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)); |
||||||
|
if (connect(cfg->sock, rp->ai_addr, rp->ai_addrlen) == 0) { |
||||||
|
break; /* success */ |
||||||
|
} |
||||||
|
freeaddrinfo(result); |
||||||
|
|
||||||
|
/* setsockopt -- socket(7) -- SO_BINDTODEVICE */ |
||||||
|
|
||||||
|
if (cfg->sock < 0) { |
||||||
|
snprintf(cfg->error, sizeof(cfg->error), "can't bind to %s:%s", cfg->host, cfg->port); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
/* TODO */ |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
bool |
||||||
|
stop(cfg_t *cfg) { |
||||||
|
assert(cfg != NULL); |
||||||
|
|
||||||
|
/* TODO */ |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
bool |
||||||
|
next(cfg_t *cfg, char *buf, size_t bufsize, bool reset) { |
||||||
|
assert(cfg != NULL); |
||||||
|
assert(buf != NULL); |
||||||
|
assert(bufsize > 0); |
||||||
|
|
||||||
|
/* TODO */ |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
void |
||||||
|
destroy(cfg_t *cfg) { |
||||||
|
f2b_port_t *next; |
||||||
|
assert(cfg != NULL); |
||||||
|
|
||||||
|
free(cfg); |
||||||
|
} |
Loading…
Reference in new issue