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.
 
 
 
 
 
 

128 lines
3.3 KiB

static int extmap_find(const void *key, const void *ele) {
const uint8_t *k = key, *e = ele;
if (*k < *e)
return -1;
if (*k > *e)
return 1;
return 0;
}
static bool extmap_has_ext(uint8_t id, struct rtpengine_output *o) {
uint8_t *const *match = bsearch(&id, o->output.extmap_filter, o->output.num_extmap_filter,
sizeof(*o->output.extmap_filter), extmap_find);
return match != NULL;
}
static bool apply_extmap_filter_ext(uint8_t id, size_t len, size_t size,
unsigned char **r, unsigned char **w, unsigned char *end,
unsigned int *count,
struct rtpengine_output *o)
{
if (*r + size + len > end)
return false;
if (extmap_has_ext(id, o)) {
// retain ext
if (*r != *w)
memmove(*w, *r, len + size);
*w += len + size;
(*count)++;
}
// else: skip over & filter out
*r += len + size;
return true;
}
static void apply_extmap_filter_finish(unsigned char *r, unsigned char *w, unsigned int count,
struct sk_buff *skb, struct rtp_parsed *rtp)
{
if (r == w)
return; // everything left as it was
if (count == 0) {
// no extensions, remove header and trim packet
rtp->rtp_header->v_p_x_cc &= ~0x10;
size_t pull = rtp->payload - (unsigned char *) rtp->ext_hdr;
memmove(rtp->ext_hdr, rtp->payload, rtp->payload_len);
rtp->payload = (unsigned char *) rtp->ext_hdr;
rtp->ext_hdr = NULL;
rtp->extension_len = 0;
skb_trim(skb, skb->len - pull);
return;
}
// shift up payload and trim packet
size_t size = w - rtp->extension;
size_t padded = (size + 3L) & ~3L;
memset(w, 0, padded - size);
rtp->ext_hdr->length = htons(padded / 4);
size_t pull = rtp->extension_len - padded;
rtp->extension_len = padded;
memmove(rtp->payload - pull, rtp->payload, rtp->payload_len);
rtp->payload -= pull;
skb_trim(skb, skb->len - pull);
}
static void apply_extmap_filter_short(struct sk_buff *skb, struct rtpengine_output *o, struct rtp_parsed *rtp) {
unsigned char *r = rtp->extension; // reader
unsigned char *end = r + rtp->extension_len;
unsigned char *w = r; // writer
unsigned int count = 0; // non-padding extensions
// XXX partly shared code
while (r < end) {
uint8_t id_len = r[0];
if (id_len == '\0') {
// padding
r++; // don't copy padding to *w
continue;
}
uint8_t id = id_len >> 4;
uint8_t len = (id_len & 0xf) + 1;
if (!apply_extmap_filter_ext(id, len, 1, &r, &w, end, &count, o))
break;
}
apply_extmap_filter_finish(r, w, count, skb, rtp);
}
static void apply_extmap_filter_long(struct sk_buff *skb, struct rtpengine_output *o, struct rtp_parsed *rtp) {
unsigned char *r = rtp->extension; // reader
unsigned char *end = r + rtp->extension_len;
unsigned char *w = r; // writer
unsigned int count = 0; // non-padding extensions
// XXX partly shared code
while (r < end) {
uint8_t id = r[0];
if (id == '\0') {
// padding
r++; // don't copy padding to *w
continue;
}
uint8_t len = r[1];
if (!apply_extmap_filter_ext(id, len, 2, &r, &w, end, &count, o))
break;
}
apply_extmap_filter_finish(r, w, count, skb, rtp);
}
static void apply_extmap_filter(struct sk_buff *skb, struct rtpengine_output *o, struct rtp_parsed *rtp) {
if (!rtp->ext_hdr)
return;
if (ntohs(rtp->ext_hdr->undefined) == 0xbede)
apply_extmap_filter_short(skb, o, rtp);
else if ((ntohs(rtp->ext_hdr->undefined) & 0xfff0) == 0x0100)
apply_extmap_filter_long(skb, o, rtp);
// else: leave untouched
}