#include "sdp.h" #include #include #include #include #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_attributes { GQueue list; GHashTable *hash; GHashTable *lists_hash; }; struct sdp_session { str s; struct sdp_origin origin; struct sdp_connection connection; int rr, rs; struct sdp_attributes attributes; GQueue media_streams; }; struct sdp_media { str s; str media_type; str port; str transport; /* ... format list */ long int port_num; int port_count; struct sdp_connection connection; int rr, rs; struct sdp_attributes attributes; }; struct attribute_rtcp { long int port_num; struct network_address address; }; struct attribute_candidate { str foundation; str component_str; str transport; str priority_str; /* incomplete */ unsigned long component; unsigned long priority; int parsed:1; }; struct sdp_attribute { str full_line, /* including a= and \r\n */ line_value, /* without a= and without \r\n */ name, /* just "rtpmap" */ value, /* just "8 PCMA/8000" */ key, /* "rtpmap:8" */ param; /* "PCMA/8000" */ enum { ATTR_OTHER = 0, ATTR_RTCP, ATTR_CANDIDATE, ATTR_ICE, } attr; union { struct attribute_rtcp rtcp; struct attribute_candidate candidate; } u; }; static const char ice_chars[] = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"; static char ice_foundation[17]; static str ice_foundation_str; /* 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->field)) 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; } static void attrs_init(struct sdp_attributes *a) { g_queue_init(&a->list); a->hash = g_hash_table_new(str_hash, str_equal); a->lists_hash = g_hash_table_new_full(str_hash, str_equal, NULL, (GDestroyNotify) g_queue_free); } static int parse_attribute_rtcp(struct sdp_attribute *output) { char *ep, *start, *end; end = output->value.s + output->value.len; output->attr = ATTR_RTCP; output->u.rtcp.port_num = strtol(output->value.s, &ep, 10); if (ep == output->value.s) return -1; if (output->u.rtcp.port_num <= 0 || output->u.rtcp.port_num > 0xffff) { output->u.rtcp.port_num = 0; return -1; } if (*ep != ' ') return 0; ep++; if (ep >= end) return 0; start = ep; EXTRACT_NETWORK_ADDRESS(u.rtcp.address); return 0; } static int parse_attribute_candidate(struct sdp_attribute *output) { char *end, *start, *ep; start = output->value.s; end = start + output->value.len; output->attr = ATTR_CANDIDATE; EXTRACT_TOKEN(u.candidate.foundation); EXTRACT_TOKEN(u.candidate.component_str); EXTRACT_TOKEN(u.candidate.transport); EXTRACT_TOKEN(u.candidate.priority_str); output->u.candidate.component = strtoul(output->u.candidate.component_str.s, &ep, 10); if (ep == output->u.candidate.component_str.s) return -1; output->u.candidate.priority = strtoul(output->u.candidate.priority_str.s, &ep, 10); if (ep == output->u.candidate.priority_str.s) return -1; output->u.candidate.parsed = 1; return 0; } static void parse_attribute(struct sdp_attribute *a) { a->name = a->line_value; str_chr_str(&a->value, &a->name, ':'); if (a->value.s) { a->name.len -= a->value.len; a->value.s++; a->value.len--; a->key = a->name; str_chr_str(&a->param, &a->value, ' '); if (a->param.s) { a->key.len += 1 + (a->value.len - a->param.len); a->param.s++; a->param.len--; if (!a->param.len) a->param.s = NULL; } else a->key.len += 1 + a->value.len; } switch (a->name.len) { case 4: if (!str_cmp(&a->name, "rtcp")) parse_attribute_rtcp(a); break; case 7: if (!str_cmp(&a->name, "ice-pwd")) a->attr = ATTR_ICE; break; case 8: if (!str_cmp(&a->name, "ice-lite")) a->attr = ATTR_ICE; break; case 9: if (!str_cmp(&a->name, "candidate")) parse_attribute_candidate(a); else if (!str_cmp(&a->name, "ice-ufrag")) a->attr = ATTR_ICE; break; case 11: if (!str_cmp(&a->name, "ice-options")) a->attr = ATTR_ICE; break; case 12: if (!str_cmp(&a->name, "ice-mismatch")) a->attr = ATTR_ICE; break; case 17: if (!str_cmp(&a->name, "remote-candidates")) a->attr = ATTR_ICE; break; } } 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; struct sdp_attributes *attrs; struct sdp_attribute *attr; str *adj_s; GQueue *attr_queue; 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 (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); attrs_init(&session->attributes); g_queue_push_tail(sessions, session); media = NULL; session->s.s = b; session->rr = session->rs = -1; 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)); attrs_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); media->s.s = b; media->rr = media->rs = -1; break; case 'c': errstr = "Error parsing c= line"; if (parse_connection(value, line_end, media ? &media->connection : &session->connection)) goto error; break; case 'a': attr = g_slice_alloc0(sizeof(*attr)); attr->full_line.s = b; attr->full_line.len = next_line ? (next_line - b) : (line_end - b); attr->line_value.s = value; attr->line_value.len = line_end - value; parse_attribute(attr); attrs = media ? &media->attributes : &session->attributes; g_queue_push_tail(&attrs->list, attr); g_hash_table_insert(attrs->hash, &attr->name, attr); if (attr->key.s) g_hash_table_insert(attrs->hash, &attr->key, attr); attr_queue = g_hash_table_lookup(attrs->lists_hash, &attr->name); if (!attr_queue) g_hash_table_insert(attrs->lists_hash, &attr->name, (attr_queue = g_queue_new())); g_queue_push_tail(attr_queue, attr); break; case 'b': /* RR:0 */ if (line_end - value < 4) break; if (!memcmp(value, "RR:", 3)) *(media ? &media->rr : &session->rr) = (line_end - value == 4 && value[3] == '0') ? 0 : 1; else if (!memcmp(value, "RS:", 3)) *(media ? &media->rs : &session->rs) = (line_end - value == 4 && value[3] == '0') ? 0 : 1; break; case 's': case 'i': case 'u': case 'e': case 'p': case 't': case 'r': case 'z': case 'k': break; default: errstr = "Unknown SDP line type found"; goto error; } adj_s = media ? &media->s : &session->s; adj_s->len = (next_line ? : end) - adj_s->s; 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(struct sdp_attributes *a) { struct sdp_attribute *attr; g_hash_table_destroy(a->hash); g_hash_table_destroy(a->lists_hash); while ((attr = g_queue_pop_head(&a->list))) { g_slice_free1(sizeof(*attr), attr); } } 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); } } static int fill_stream_address(struct stream_input *si, struct sdp_media *media, struct sdp_session *session) { if (media->connection.parsed) si->stream.ip46 = media->connection.address.parsed; else if (session->connection.parsed) si->stream.ip46 = session->connection.address.parsed; else return -1; return 0; } static int fill_stream(struct stream_input *si, struct sdp_media *media, struct sdp_session *session, int offset) { if (fill_stream_address(si, media, session)) return -1; /* we ignore the media type */ si->stream.port = (media->port_num + (offset * 2)) & 0xffff; return 0; } static int fill_stream_rtcp(struct stream_input *si, struct sdp_media *media, struct sdp_session *session, int port) { if (fill_stream_address(si, media, session)) return -1; si->stream.port = port; return 0; } int sdp_streams(const GQueue *sessions, GQueue *streams, GHashTable *streamhash) { struct sdp_session *session; struct sdp_media *media; struct stream_input *si; GList *l, *k; const char *errstr; int i, num; str s; struct sdp_attribute *attr; 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; si = NULL; for (i = 0; i < media->port_count; i++) { si = g_slice_alloc0(sizeof(*si)); errstr = "No address info found for stream"; if (fill_stream(si, media, session, i)) goto error; if (i == 0 && g_hash_table_contains(streamhash, si)) { g_slice_free1(sizeof(*si), si); continue; } si->stream.num = ++num; si->consecutive_num = (i == 0) ? media->port_count : 1; g_hash_table_insert(streamhash, si, si); g_queue_push_tail(streams, si); } if (!si || media->port_count != 1) continue; str_init(&s, "rtcp"); /* XXX use the enum for hash instead? */ attr = g_hash_table_lookup(media->attributes.hash, &s); if (!attr || !attr->u.rtcp.port_num) continue; if (attr->u.rtcp.port_num == si->stream.port + 1) continue; si->has_rtcp = 1; si = g_slice_alloc0(sizeof(*si)); if (fill_stream_rtcp(si, media, session, attr->u.rtcp.port_num)) goto error; si->stream.num = ++num; si->consecutive_num = 1; g_hash_table_insert(streamhash, si, si); g_queue_push_tail(streams, si); } } return 0; error: mylog(LOG_WARNING, "Failed to extract streams from SDP: %s", errstr); if (si) g_slice_free1(sizeof(*si), si); return -1; } struct sdp_chopper *sdp_chopper_new(str *input) { struct sdp_chopper *c = g_slice_alloc0(sizeof(*c)); c->input = input; c->chunk = g_string_chunk_new(512); c->iov = g_array_new(0, 0, sizeof(struct iovec)); return c; } static void chopper_append(struct sdp_chopper *c, const char *s, int len) { struct iovec *iov; g_array_set_size(c->iov, ++c->iov_num); iov = &g_array_index(c->iov, struct iovec, c->iov_num - 1); iov->iov_base = (void *) s; iov->iov_len = len; c->str_len += len; } static inline void chopper_append_c(struct sdp_chopper *c, const char *s) { chopper_append(c, s, strlen(s)); } static inline void chopper_append_str(struct sdp_chopper *c, const str *s) { chopper_append(c, s->s, s->len); } static void chopper_append_dup(struct sdp_chopper *c, const char *s, int len) { return chopper_append(c, g_string_chunk_insert_len(c->chunk, s, len), len); } static void chopper_append_printf(struct sdp_chopper *c, const char *fmt, ...) __attribute__((format(printf,2,3))); static void chopper_append_printf(struct sdp_chopper *c, const char *fmt, ...) { char buf[32]; int l; va_list va; va_start(va, fmt); l = vsnprintf(buf, sizeof(buf) - 1, fmt, va); va_end(va); chopper_append(c, g_string_chunk_insert_len(c->chunk, buf, l), l); } static int copy_up_to_ptr(struct sdp_chopper *chop, const char *b) { int offset, len; offset = b - 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; } chopper_append(chop, chop->input->s + chop->position, len); chop->position += len; return 0; } static int copy_up_to(struct sdp_chopper *chop, str *where) { return copy_up_to_ptr(chop, where->s); } static int copy_up_to_end_of(struct sdp_chopper *chop, str *where) { return copy_up_to_ptr(chop, where->s + where->len); } static void copy_remainder(struct sdp_chopper *chop) { copy_up_to_ptr(chop, chop->input->s + chop->input->len); } static int skip_over(struct sdp_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 void fill_relays(struct streamrelay **rtp, struct streamrelay **rtcp, GList *m, int off, struct stream_input *sip) { *rtp = &((struct callstream *) m->data)->peers[off].rtps[0]; if (rtcp) *rtcp = &((struct callstream *) m->data)->peers[off].rtps[1]; if (sip && sip->has_rtcp && m->next) *rtcp = &((struct callstream *) m->next->data)->peers[off].rtps[0]; } static int replace_media_port(struct sdp_chopper *chop, struct sdp_media *media, struct streamrelay *sr) { str *port = &media->port; if (copy_up_to(chop, port)) return -1; chopper_append_printf(chop, "%hu", sr->fd.localport); if (skip_over(chop, port)) return -1; return 0; } static int replace_consecutive_port_count(struct sdp_chopper *chop, struct sdp_media *media, struct streamrelay *rtp, GList *m, int off) { int cons; struct streamrelay *sr; if (media->port_count == 1) return 0; for (cons = 1; cons < media->port_count; cons++) { m = m->next; if (!m) goto warn; fill_relays(&sr, NULL, m, off, NULL); if (sr->fd.localport != rtp->fd.localport + cons * 2) { warn: mylog(LOG_WARN, "Failed to handle consecutive ports"); break; } } chopper_append_printf(chop, "/%i", cons); return 0; } static int insert_ice_address(struct sdp_chopper *chop, struct sdp_ng_flags *flags, struct streamrelay *sr) { char buf[64]; int len; if (!flags->trust_address && flags->received_from_family.len == 3 && flags->received_from_address.len) chopper_append_str(chop, &flags->received_from_address); else { call_stream_address(buf, sr->up, SAF_ICE, &len); chopper_append_dup(chop, buf, len); } chopper_append_printf(chop, " %hu", sr->fd.localport); return 0; } static int replace_network_address(struct sdp_chopper *chop, struct network_address *address, struct streamrelay *sr, struct sdp_ng_flags *flags) { char buf[64]; int len; if (copy_up_to(chop, &address->address_type)) return -1; if (!flags->trust_address && flags->received_from_family.len == 3 && flags->received_from_address.len) { chopper_append_str(chop, &flags->received_from_family); chopper_append_c(chop, " "); chopper_append_str(chop, &flags->received_from_address); } else { call_stream_address(buf, sr->up, SAF_NG, &len); chopper_append_dup(chop, buf, len); } if (skip_over(chop, &address->address)) return -1; return 0; } void sdp_chopper_destroy(struct sdp_chopper *chop) { g_string_chunk_free(chop->chunk); g_array_free(chop->iov, 1); g_slice_free1(sizeof(*chop), chop); } static void random_string(char *buf, int len) { while (len--) *buf++ = ice_chars[random() % strlen(ice_chars)]; } static void create_random_string(struct call *call, str *s, int len) { char buf[30]; assert(len < sizeof(buf)); if (s->s) return; random_string(buf, len); call_str_cpy_len(call, s, buf, len); } static int process_session_attributes(struct sdp_chopper *chop, struct sdp_attributes *attrs, struct sdp_ng_flags *flags) { GList *l; struct sdp_attribute *attr; for (l = attrs->list.head; l; l = l->next) { attr = l->data; switch (attr->attr) { case ATTR_ICE: case ATTR_CANDIDATE: if (!flags->ice_remove && !flags->ice_force) break; goto strip; default: break; } continue; strip: if (copy_up_to(chop, &attr->full_line)) return -1; if (skip_over(chop, &attr->full_line)) return -1; } return 0; } static int process_media_attributes(struct sdp_chopper *chop, struct sdp_attributes *attrs, struct sdp_ng_flags *flags) { GList *l; struct sdp_attribute *attr; for (l = attrs->list.head; l; l = l->next) { attr = l->data; switch (attr->attr) { case ATTR_ICE: case ATTR_CANDIDATE: if (!flags->ice_remove && !flags->ice_force) break; goto strip; case ATTR_RTCP: goto strip; default: break; } continue; strip: if (copy_up_to(chop, &attr->full_line)) return -1; if (skip_over(chop, &attr->full_line)) return -1; } return 0; } static GList *find_stream_num(GList *m, int num) { /* XXX use a hash instead? must link input streams to output streams */ while (m && ((struct callstream *) m->data)->num < num) m = m->next; while (m && ((struct callstream *) m->data)->num > num) m = m->prev; return m; } static int has_rtcp(struct sdp_session *session, struct sdp_media *media) { if ((media->rr == -1 ? session->rr : media->rr) != 0 && (media->rs == -1 ? session->rs : media->rs) != 0) return 1; return 0; } static unsigned long prio_calc(unsigned int pref) { return (1 << 24) * 126 + (1 << 8) * pref + 256 * 1; } static unsigned long new_priority(struct sdp_media *media) { str s; GQueue *cands; unsigned int pref; unsigned long prio; GList *l; struct attribute_candidate *c; pref = 65535; prio = prio_calc(pref); if (!media) goto out; str_init(&s, "candidate"); cands = g_hash_table_lookup(media->attributes.lists_hash, &s); for (l = cands->head; l; l = l->next) { c = l->data; while (c->priority >= prio) { pref--; prio = prio_calc(pref); } } out: return prio; } int sdp_replace(struct sdp_chopper *chop, GQueue *sessions, struct call *call, enum call_opmode opmode, struct sdp_ng_flags *flags, GHashTable *streamhash) { struct sdp_session *session; struct sdp_media *media; GList *l, *k, *m; int off; struct stream_input si, *sip; struct streamrelay *rtp, *rtcp; unsigned long priority; off = opmode; m = call->callstreams->head; for (l = sessions->head; l; l = l->next) { session = l->data; fill_relays(&rtp, &rtcp, m, off, NULL); if (session->origin.parsed && flags->replace_origin) { if (replace_network_address(chop, &session->origin.address, rtp, flags)) goto error; } if (session->connection.parsed) { if (replace_network_address(chop, &session->connection.address, rtp, flags)) goto error; } if (process_session_attributes(chop, &session->attributes, flags)) goto error; if (flags->ice_force) { /* XXX locking here? */ create_random_string(call, &rtp->up->ice_ufrag, 8); create_random_string(call, &rtp->up->ice_pwd, 28); copy_up_to_end_of(chop, &session->s); chopper_append_c(chop, "a=ice-lite\r\na=ice-ufrag:"); chopper_append_str(chop, &rtp->up->ice_ufrag); chopper_append_c(chop, "\r\na=ice-pwd:"); chopper_append_str(chop, &rtp->up->ice_pwd); chopper_append_c(chop, "\r\n"); } for (k = session->media_streams.head; k; k = k->next) { media = k->data; if (fill_stream(&si, media, session, 0)) goto error; sip = g_hash_table_lookup(streamhash, &si); if (!sip) goto error; m = find_stream_num(m, sip->stream.num); if (!m) goto error; fill_relays(&rtp, &rtcp, m, off, sip); if (replace_media_port(chop, media, rtp)) goto error; if (replace_consecutive_port_count(chop, media, rtp, m, off)) goto error; if (media->connection.parsed && flags->replace_sess_conn) { if (replace_network_address(chop, &media->connection.address, rtp, flags)) goto error; } if (process_media_attributes(chop, &media->attributes, flags)) goto error; copy_up_to_end_of(chop, &media->s); if (has_rtcp(session, media)) { chopper_append_c(chop, "a=rtcp:"); chopper_append_printf(chop, "%hu", rtcp->fd.localport); chopper_append_c(chop, "\r\n"); } if (!flags->ice_remove) { if (flags->ice_force) { priority = new_priority(NULL); rtp->stun = 1; rtcp->stun = 1; } else priority = new_priority(media); chopper_append_c(chop, "a=candidate:"); chopper_append_str(chop, &ice_foundation_str); chopper_append_printf(chop, " 1 UDP %lu ", priority); insert_ice_address(chop, flags, rtp); chopper_append_c(chop, " typ host\r\n"); if (has_rtcp(session, media)) { chopper_append_c(chop, "a=candidate:"); chopper_append_str(chop, &ice_foundation_str); chopper_append_printf(chop, " 2 UDP %lu ", priority - 1); insert_ice_address(chop, flags, rtcp); chopper_append_c(chop, " typ host\r\n"); } } } } copy_remainder(chop); return 0; error: mylog(LOG_ERROR, "Error rewriting SDP"); return -1; } void sdp_init() { random_string(ice_foundation, sizeof(ice_foundation) - 1); ice_foundation_str.s = ice_foundation; ice_foundation_str.len = sizeof(ice_foundation) - 1; }