summaryrefslogtreecommitdiffstats
path: root/src/main.c
diff options
context:
space:
mode:
authorHarald Welte <laforge@gnumonks.org>2013-02-11 11:34:58 +0100
committerHarald Welte <laforge@gnumonks.org>2013-02-11 11:38:23 +0100
commitce94d971e1223626c96ad32373ea4ff034233b50 (patch)
treebbe05c72550d305c50e07528547cb730f59bc4f7 /src/main.c
parentf7f0c91ca8c6cae331fa65db71f8c32502fad256 (diff)
gapk: Add support for RTP input/output streams
Instead of having only file-based I/O, this enables gapk to receive and send RTP streams, e.g. from live GSM network equipment like sysmoBTS/nanoBTS. Support is currently simplistic. On transmit, there is hard-coded codec type of full-rate GSM. On receive-side, we should auto-detect the format based on frame size and/or payload type, but we don't do that yet at all.
Diffstat (limited to 'src/main.c')
-rw-r--r--src/main.c116
1 files changed, 108 insertions, 8 deletions
diff --git a/src/main.c b/src/main.c
index 9b49445..fd91d4b 100644
--- a/src/main.c
+++ b/src/main.c
@@ -25,6 +25,11 @@
#include <unistd.h>
#include <getopt.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+
+#include <osmocom/core/socket.h>
+
#include <gapk/codecs.h>
#include <gapk/formats.h>
#include <gapk/procqueue.h>
@@ -33,9 +38,17 @@
struct gapk_options
{
const char *fname_in;
+ struct {
+ const char *hostname;
+ uint16_t port;
+ } rtp_in;
const struct format_desc *fmt_in;
const char *fname_out;
+ struct {
+ const char *hostname;
+ uint16_t port;
+ } rtp_out;
const struct format_desc *fmt_out;
};
@@ -45,7 +58,7 @@ struct gapk_state
struct pq *pq;
- union {
+ struct {
struct {
FILE *fh;
} file;
@@ -54,7 +67,7 @@ struct gapk_state
} rtp;
} in;
- union {
+ struct {
struct {
FILE *fh;
} file;
@@ -75,7 +88,9 @@ print_help(char *progname)
fprintf(stdout, "\n");
fprintf(stdout, "Options:\n");
fprintf(stdout, " -i, --input-file=FILE\t\tInput file\n");
+ fprintf(stdout, " -I, --input-rtp=HOST/PORT\t\tInput RTP stream\n");
fprintf(stdout, " -o, --output-file=FILE\tOutput file\n");
+ fprintf(stdout, " -O, --output-rtp=HOST/PORT\tOutput RTP stream\n");
fprintf(stdout, " -f, --input-format=FMT\tInput format (see below)\n");
fprintf(stdout, " -g, --output-format=FMT\tOutput format (see below)\n");
fprintf(stdout, "\n");
@@ -114,16 +129,39 @@ print_help(char *progname)
}
static int
+parse_host_port(const char *host_port, const char **host)
+{
+ char *dup = strdup(host_port);
+ char *tok;
+
+ if (!dup)
+ return -ENOMEM;
+
+ tok = strtok(dup, "/");
+ if (!tok)
+ return -EINVAL;
+ *host = tok;
+
+ tok = strtok(NULL, "/");
+ if (!tok)
+ return -EINVAL;
+
+ return atoi(tok);
+}
+
+static int
parse_options(struct gapk_state *state, int argc, char *argv[])
{
const struct option long_options[] = {
{"input-file", 1, 0, 'i'},
{"output-file", 1, 0, 'o'},
+ {"input-rtp", 1, 0, 'I'},
+ {"output-rtp", 1, 0, 'O'},
{"input-format", 1, 0, 'f'},
{"output-format", 1, 0, 'g'},
{"help", 0, 0, 'h'},
};
- const char *short_options = "i:o:f:g:h";
+ const char *short_options = "i:o:I:O:f:g:h";
struct gapk_options *opt = &state->opts;
@@ -132,7 +170,7 @@ parse_options(struct gapk_state *state, int argc, char *argv[])
/* Parse */
while (1) {
- int c;
+ int c, rv;
int opt_idx;
c = getopt_long(
@@ -149,6 +187,24 @@ parse_options(struct gapk_state *state, int argc, char *argv[])
opt->fname_out = optarg;
break;
+ case 'I':
+ rv = parse_host_port(optarg, &opt->rtp_in.hostname);
+ if (rv < 0 || rv > 0xffff) {
+ fprintf(stderr, "[!] Invalid port: %d\n", rv);
+ return -EINVAL;
+ }
+ opt->rtp_in.port = rv;
+ break;
+
+ case 'O':
+ rv = parse_host_port(optarg, &opt->rtp_out.hostname);
+ if (rv < 0 || rv > 0xffff) {
+ fprintf(stderr, "[!] Invalid port: %d\n", rv);
+ return -EINVAL;
+ }
+ opt->rtp_out.port = rv;
+ break;
+
case 'f':
opt->fmt_in = fmt_get_from_name(optarg);
if (!opt->fmt_in) {
@@ -214,6 +270,18 @@ check_options(struct gapk_state *gs)
}
}
+ /* Input combinations */
+ if (gs->opts.fname_in && gs->opts.rtp_in.port) {
+ fprintf(stderr, "[!] You have to decide on either file or RTP input\n");
+ return -EINVAL;
+ }
+
+ /* Output combinations */
+ if (gs->opts.fname_out && gs->opts.rtp_out.port) {
+ fprintf(stderr, "[!] You have to decide on either file or RTP output\n");
+ return -EINVAL;
+ }
+
return 0;
}
@@ -227,6 +295,16 @@ files_open(struct gapk_state *gs)
perror("fopen");
return -errno;
}
+ } else if (gs->opts.rtp_in.port) {
+ gs->in.rtp.fd = osmo_sock_init(AF_UNSPEC, SOCK_DGRAM,
+ IPPROTO_UDP,
+ gs->opts.rtp_in.hostname,
+ gs->opts.rtp_in.port,
+ OSMO_SOCK_F_BIND);
+ if (gs->in.rtp.fd < 0) {
+ fprintf(stderr, "[!] Error while opening input socket\n");
+ return gs->in.rtp.fd;
+ }
} else
gs->in.file.fh = stdin;
@@ -237,6 +315,16 @@ files_open(struct gapk_state *gs)
perror("fopen");
return -errno;
}
+ } else if (gs->opts.rtp_out.port) {
+ gs->out.rtp.fd = osmo_sock_init(AF_UNSPEC, SOCK_DGRAM,
+ IPPROTO_UDP,
+ gs->opts.rtp_out.hostname,
+ gs->opts.rtp_out.port,
+ OSMO_SOCK_F_CONNECT);
+ if (gs->out.rtp.fd < 0) {
+ fprintf(stderr, "[!] Error while opening output socket\n");
+ return gs->out.rtp.fd;
+ }
} else
gs->out.file.fh = stdout;
@@ -248,8 +336,12 @@ files_close(struct gapk_state *gs)
{
if (gs->in.file.fh && gs->in.file.fh != stdin)
fclose(gs->in.file.fh);
+ if (gs->in.rtp.fd >= 0)
+ close(gs->in.rtp.fd);
if (gs->out.file.fh && gs->out.file.fh != stdout)
fclose(gs->out.file.fh);
+ if (gs->out.rtp.fd >= 0)
+ close(gs->out.rtp.fd);
}
static int
@@ -260,7 +352,7 @@ handle_headers(struct gapk_state *gs)
/* Input file header (remove & check it) */
len = gs->opts.fmt_in->header_len;
- if (len) {
+ if (len && gs->in.file.fh) {
uint8_t *buf;
buf = malloc(len);
@@ -281,7 +373,7 @@ handle_headers(struct gapk_state *gs)
/* Output file header (write it) */
len = gs->opts.fmt_out->header_len;
- if (len) {
+ if (len && gs->out.file.fh) {
rv = fwrite(gs->opts.fmt_out->header, len, 1, gs->out.file.fh);
if (rv != 1)
return -ENOSPC;
@@ -310,7 +402,10 @@ make_processing_chain(struct gapk_state *gs)
(fmt_out->codec_type != fmt_in->codec_type);
/* File read */
- pq_queue_file_input(gs->pq, gs->in.file.fh, fmt_in->frame_len);
+ if (gs->in.file.fh)
+ pq_queue_file_input(gs->pq, gs->in.file.fh, fmt_in->frame_len);
+ else
+ pq_queue_rtp_input(gs->pq, gs->in.rtp.fd, fmt_in->frame_len);
/* Decoding to PCM ? */
if (need_dec)
@@ -363,7 +458,10 @@ make_processing_chain(struct gapk_state *gs)
}
/* File write */
- pq_queue_file_output(gs->pq, gs->out.file.fh, fmt_out->frame_len);
+ if (gs->out.file.fh)
+ pq_queue_file_output(gs->pq, gs->out.file.fh, fmt_out->frame_len);
+ else
+ pq_queue_rtp_output(gs->pq, gs->out.rtp.fd, fmt_out->frame_len);
return 0;
}
@@ -391,6 +489,8 @@ int main(int argc, char *argv[])
/* Clear state */
memset(gs, 0x00, sizeof(struct gapk_state));
+ gs->in.rtp.fd = -1;
+ gs->out.rtp.fd = -1;
/* Parse / check options */
rv = parse_options(gs, argc, argv);