Saner arg structure.

This commit is contained in:
Ian Gulliver
2016-02-15 21:12:26 +00:00
parent 9b97a2c32c
commit 798b75c30d
5 changed files with 39 additions and 33 deletions

View File

@@ -9,8 +9,7 @@
struct opts {
char *backend_node;
char *backend_service;
bool dump;
};
struct client {
@@ -20,14 +19,10 @@ struct client {
static int parse_opts(int argc, char *argv[], struct opts *opts) {
int opt;
while ((opt = getopt(argc, argv, "h:p:")) != -1) {
while ((opt = getopt(argc, argv, "d")) != -1) {
switch (opt) {
case 'h':
opts->backend_node = optarg;
break;
case 'p':
opts->backend_service = optarg;
case 'd':
opts->dump = true;
break;
default:
@@ -69,11 +64,12 @@ int main(int argc, char *argv[]) {
airspy_adsb_init();
struct opts opts = {
.backend_node = "localhost",
.backend_service = "30006",
.dump = false,
};
if (parse_opts(argc, argv, &opts)) {
fprintf(stderr, "Usage: %s [-h backend_host] [-p backend_port]\n", argv[0]);
if (parse_opts(argc, argv, &opts) ||
argc - optind < 2 ||
(argc - optind) % 2 != 0) {
fprintf(stderr, "Usage: %s -d localhost 30006 [ remotehost 30002 ... ]\n", argv[0]);
return EXIT_FAILURE;
}
@@ -83,10 +79,14 @@ int main(int argc, char *argv[]) {
return EXIT_FAILURE;
}
struct backend backend = BACKEND_INIT;
if (!backend_connect(opts.backend_node, opts.backend_service, &backend, epoll_fd)) {
fprintf(stderr, "Unable to connect to %s/%s\n", opts.backend_node, opts.backend_service);
return EXIT_FAILURE;
int nbackends = (argc - optind) / 2;
struct backend backends[nbackends];
for (int i = 0, j = optind; i < nbackends && j < argc; i++, j += 2) {
backend_init(&backends[i]);
if (!backend_connect(argv[j], argv[j + 1], &backends[i], epoll_fd)) {
fprintf(stderr, "Unable to connect to %s %s\n", argv[j], argv[j + 1]);
return EXIT_FAILURE;
}
}
loop(epoll_fd);

View File

@@ -4,17 +4,29 @@
#include <sys/socket.h>
#include <netdb.h>
#include <sys/epoll.h>
#include <string.h>
#include "airspy_adsb.h"
#include "backend.h"
bool backend_autodetect_parse(struct backend *, struct packet *);
static parser parsers[] = {
airspy_adsb_parse,
};
#define NUM_PARSERS (sizeof(parsers) / sizeof(*parsers))
void backend_init(struct backend *backend) {
backend->type = PEER_BACKEND;
backend->fd = -1;
buf_init(&backend->buf);
memset(backend->parser_state, 0, PARSER_STATE_LEN);
backend->parser = backend_autodetect_parse;
}
bool backend_connect(char *node, char *service, struct backend *backend, int epoll_fd) {
assert(backend->type == PEER_BACKEND);
@@ -28,7 +40,7 @@ bool backend_connect(char *node, char *service, struct backend *backend, int epo
int gai_err = getaddrinfo(node, service, &hints, &addrs);
if (gai_err) {
fprintf(stderr, "getaddrinfo(%s/%s): %s\n", node, service, gai_strerror(gai_err));
fprintf(stderr, "getaddrinfo(%s %s): %s\n", node, service, gai_strerror(gai_err));
return false;
}
}
@@ -51,13 +63,13 @@ bool backend_connect(char *node, char *service, struct backend *backend, int epo
if (addr == NULL) {
freeaddrinfo(addrs);
fprintf(stderr, "Can't connect to %s/%s\n", node, service);
fprintf(stderr, "Can't connect to %s %s\n", node, service);
return false;
}
char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV];
if (getnameinfo(addr->ai_addr, addr->ai_addrlen, hbuf, sizeof(hbuf), sbuf, sizeof(sbuf), NI_NUMERICHOST | NI_NUMERICSERV) == 0) {
fprintf(stderr, "Connected to %s/%s\n", hbuf, sbuf);
fprintf(stderr, "Connected to %s %s\n", hbuf, sbuf);
}
}

View File

@@ -14,15 +14,7 @@ struct backend {
parser parser;
};
#define BACKEND_INIT { \
.type = PEER_BACKEND, \
.fd = -1, \
.buf = BUF_INIT, \
.parser_state = { 0 }, \
.parser = backend_autodetect_parse, \
}
void backend_init(struct backend *);
bool backend_connect(char *, char *, struct backend *, int);
bool backend_read(struct backend *);
bool backend_autodetect_parse(struct backend *, struct packet *);

View File

@@ -7,6 +7,11 @@
#include "common.h"
void buf_init(struct buf *buf) {
buf->start = 0;
buf->length = 0;
}
ssize_t buf_fill(struct buf *buf, int fd) {
if (buf->start + buf->length == BUF_LEN_MAX) {
assert(buf->start > 0);

View File

@@ -23,14 +23,11 @@ struct buf {
size_t start;
size_t length;
};
#define BUF_INIT { \
.start = 0, \
.length = 0, \
}
#define buf_chr(buff, at) ((buff)->buf[(buff)->start + (at)])
#define buf_at(buff, at) (&buf_chr(buff, at))
void buf_init(struct buf *);
ssize_t buf_fill(struct buf *, int);
void buf_consume(struct buf *, size_t);