You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
rtpengine/daemon/sdp.c

498 lines
10 KiB

#include <glib.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include "sdp.h"
#include "call.h"
#include "log.h"
#include "str.h"
#include "call.h"
struct network_address {
str network_type;
str address_type;
str address;
struct in6_addr parsed;
};
struct sdp_origin {
str username;
str session_id;
str version;
struct network_address address;
int parsed:1;
};
struct sdp_connection {
struct network_address address;
int parsed:1;
};
struct sdp_session {
struct sdp_origin origin;
struct sdp_connection connection;
GQueue attributes;
GQueue media_streams;
};
struct sdp_media {
str media_type;
str port;
str transport;
/* ... format list */
long int port_num;
int port_count;
struct sdp_connection connection;
GQueue attributes;
};
struct string_chopper {
str *input;
GString *output;
int position;
};
/* hack hack */
static inline int inet_pton_str(int af, str *src, void *dst) {
char *s = src->s;
char p;
int ret;
p = s[src->len];
s[src->len] = '\0';
ret = inet_pton(af, src->s, dst);
s[src->len] = p;
return ret;
}
static int parse_address(struct network_address *address) {
struct in_addr in4;
if (address->network_type.len != 2)
return -1;
if (memcmp(address->network_type.s, "IN", 2)
&& memcmp(address->network_type.s, "in", 2))
return -1;
if (address->address_type.len != 3)
return -1;
if (!memcmp(address->address_type.s, "IP4", 3)
|| !memcmp(address->address_type.s, "ip4", 3)) {
if (inet_pton_str(AF_INET, &address->address, &in4) != 1)
return -1;
in4_to_6(&address->parsed, in4.s_addr);
}
else if (!memcmp(address->address_type.s, "IP6", 3)
|| !memcmp(address->address_type.s, "ip6", 3)) {
if (inet_pton_str(AF_INET6, &address->address, &address->parsed) != 1)
return -1;
}
else
return -1;
return 0;
}
static inline int extract_token(char **sp, char *end, str *out) {
char *space;
out->s = *sp;
space = memchr(*sp, ' ', end - *sp);
if (space == *sp || end == *sp)
return -1;
if (!space) {
out->len = end - *sp;
*sp = end;
}
else {
out->len = space - *sp;
*sp = space + 1;
}
return 0;
}
#define EXTRACT_TOKEN(field) if (extract_token(&start, end, &output->field)) return -1
#define EXTRACT_NETWORK_ADDRESS(field) \
EXTRACT_TOKEN(field.network_type); \
EXTRACT_TOKEN(field.address_type); \
EXTRACT_TOKEN(field.address); \
if (parse_address(&output->address)) return -1
static int parse_origin(char *start, char *end, struct sdp_origin *output) {
if (output->parsed)
return -1;
EXTRACT_TOKEN(username);
EXTRACT_TOKEN(session_id);
EXTRACT_TOKEN(version);
EXTRACT_NETWORK_ADDRESS(address);
output->parsed = 1;
return 0;
}
static int parse_connection(char *start, char *end, struct sdp_connection *output) {
if (output->parsed)
return -1;
EXTRACT_NETWORK_ADDRESS(address);
output->parsed = 1;
return 0;
}
static int parse_media(char *start, char *end, struct sdp_media *output) {
char *ep;
EXTRACT_TOKEN(media_type);
EXTRACT_TOKEN(port);
EXTRACT_TOKEN(transport);
output->port_num = strtol(output->port.s, &ep, 10);
if (ep == output->port.s)
return -1;
if (output->port_num <= 0 || output->port_num > 0xffff)
return -1;
if (*ep == '/') {
output->port_count = atoi(ep + 1);
if (output->port_count <= 0)
return -1;
if (output->port_count > 10) /* unsupported */
return -1;
}
else
output->port_count = 1;
return 0;
}
int sdp_parse(str *body, GQueue *sessions) {
char *b, *end, *value, *line_end, *next_line;
struct sdp_session *session = NULL;
struct sdp_media *media = NULL;
const char *errstr;
str *attribute;
b = body->s;
end = str_end(body);
while (b && b < end - 1) {
errstr = "Missing '=' sign";
if (b[1] != '=')
goto error;
value = &b[2];
line_end = memchr(value, '\n', end - value);
if (!line_end) {
/* assume missing LF at end of body */
line_end = end;
next_line = NULL;
}
else {
next_line = line_end + 1;
if (next_line >= end)
next_line = NULL;
if (line_end[-1] == '\r')
line_end--;
}
switch (b[0]) {
case 'v':
errstr = "Error in v= line";
if (line_end != value + 1)
goto error;
if (value[0] != '0')
goto error;
session = g_slice_alloc0(sizeof(*session));
g_queue_init(&session->media_streams);
g_queue_init(&session->attributes);
g_queue_push_tail(sessions, session);
media = NULL;
break;
case 'o':
errstr = "o= line found within media section";
if (media)
goto error;
errstr = "Error parsing o= line";
if (parse_origin(value, line_end, &session->origin))
goto error;
break;
case 'm':
media = g_slice_alloc0(sizeof(*media));
g_queue_init(&media->attributes);
errstr = "Error parsing m= line";
if (parse_media(value, line_end, media))
goto error;
g_queue_push_tail(&session->media_streams, media);
break;
case 'c':
errstr = "Error parsing c= line";
if (parse_connection(value, line_end,
media ? &media->connection : &session->connection))
goto error;
break;
case 'a':
attribute = g_slice_alloc(sizeof(*attribute));
attribute->s = value;
attribute->len = line_end - value;
g_queue_push_tail(media ? &media->attributes : &session->attributes,
attribute);
break;
case 's':
case 'i':
case 'u':
case 'e':
case 'p':
case 'b':
case 't':
case 'r':
case 'z':
case 'k':
break;
default:
errstr = "Unknown SDP line type found";
goto error;
}
b = next_line;
}
return 0;
error:
mylog(LOG_WARNING, "Error parsing SDP at offset %li: %s", b - body->s, errstr);
sdp_free(sessions);
return -1;
}
static void __free_attributes(GQueue *a) {
str *str;
while ((str = g_queue_pop_head(a))) {
g_slice_free1(sizeof(*str), str);
}
}
void sdp_free(GQueue *sessions) {
struct sdp_session *session;
struct sdp_media *media;
while ((session = g_queue_pop_head(sessions))) {
while ((media = g_queue_pop_head(&session->media_streams))) {
__free_attributes(&media->attributes);
g_slice_free1(sizeof(*media), media);
}
__free_attributes(&session->attributes);
g_slice_free1(sizeof(*session), session);
}
}
int sdp_streams(const GQueue *sessions, GQueue *streams) {
struct sdp_session *session;
struct sdp_media *media;
struct stream *stream;
GList *l, *k;
const char *errstr;
int i, num;
num = 0;
for (l = sessions->head; l; l = l->next) {
session = l->data;
for (k = session->media_streams.head; k; k = k->next) {
media = k->data;
for (i = 0; i < media->port_count; i++) {
stream = g_slice_alloc0(sizeof(*stream));
errstr = "No address info found for stream";
if (media->connection.parsed)
stream->ip46 = media->connection.address.parsed;
else if (session->connection.parsed)
stream->ip46 = session->connection.address.parsed;
else
goto error;
/* XXX ports must be consecutive */
/* XXX check for RTP type */
stream->port = (media->port_num + (i * 2)) & 0xffff;
stream->num = ++num;
g_queue_push_tail(streams, stream);
}
}
}
return 0;
error:
mylog(LOG_WARNING, "Failed to extract streams from SDP: %s", errstr);
return -1;
}
static void chopper_init(struct string_chopper *c, str *input) {
c->input = input;
c->output = g_string_new_str();
c->position = 0;
}
static int copy_up_to(struct string_chopper *chop, str *where) {
int offset, len;
offset = where->s - chop->input->s;
assert(offset >= 0);
assert(offset < chop->input->len);
len = offset - chop->position;
if (len < 0) {
mylog(LOG_WARNING, "Malformed SDP, cannot rewrite");
return -1;
}
g_string_append_len(chop->output, chop->input->s + chop->position, len);
chop->position += len;
return 0;
}
static void copy_remainder(struct string_chopper *chop) {
int len;
len = chop->input->len - chop->position;
assert(len >= 0);
g_string_append_len(chop->output, chop->input->s + chop->position, len);
chop->position += len;
}
static int skip_over(struct string_chopper *chop, str *where) {
int offset, len;
offset = (where->s - chop->input->s) + where->len;
assert(offset >= 0);
assert(offset < chop->input->len);
len = offset - chop->position;
if (len < 0) {
mylog(LOG_WARNING, "Malformed SDP, cannot rewrite");
return -1;
}
chop->position += len;
return 0;
}
static int replace_port(struct string_chopper *chop, str *port, GList *m, int off) {
struct callstream *cs;
struct streamrelay *sr;
if (!m) {
mylog(LOG_ERROR, "BUG! Ran out of streams");
return -1;
}
cs = m->data;
sr = &cs->peers[off].rtps[0];
if (copy_up_to(chop, port))
return -1;
g_string_append_printf(chop->output, "%hu", sr->localport);
if (skip_over(chop, port))
return -1;
return 0;
}
static int replace_network_address(struct string_chopper *chop, struct network_address *address, GList *m, int off) {
struct callstream *cs;
struct peer *peer;
if (!m) {
mylog(LOG_ERROR, "BUG! Ran out of streams");
return -1;
}
cs = m->data;
peer = &cs->peers[off];
if (copy_up_to(chop, &address->address_type))
return -1;
call_stream_address(chop->output, peer, 2);
if (skip_over(chop, &address->address))
return -1;
return 0;
}
static str *chopper_done(struct string_chopper *chop) {
str *ret;
ret = g_string_free_str(chop->output);
return ret;
}
static void chopper_destroy(struct string_chopper *chop) {
g_string_free(chop->output, TRUE);
}
/* XXX use stream numbers as index */
/* XXX use port numbers as index */
/* XXX get rid of num/off parameters? */
/* XXX use iovec based rewriting */
str *sdp_replace(str *body, GQueue *sessions, struct call *call, int num, int off) {
struct sdp_session *session;
struct sdp_media *media;
GList *l, *k, *m;
struct string_chopper chop;
chopper_init(&chop, body);
m = call->callstreams->head;
for (l = sessions->head; l; l = l->next) {
session = l->data;
if (session->origin.parsed) {
if (replace_network_address(&chop, &session->origin.address, m, off))
goto error;
}
if (session->connection.parsed) {
if (replace_network_address(&chop, &session->connection.address, m, off))
goto error;
}
for (k = session->media_streams.head; k; k = k->next) {
media = k->data;
/* XXX take multiple ports into account */
if (replace_port(&chop, &media->port, m, off))
goto error;
if (media->connection.parsed) {
if (replace_network_address(&chop, &media->connection.address, m, off))
goto error;
}
}
}
copy_remainder(&chop);
return chopper_done(&chop);
error:
mylog(LOG_ERROR, "Error rewriting SDP");
chopper_destroy(&chop);
return NULL;
}