rtmppkt.c 21 KB
Newer Older
1 2
/*
 * RTMP input format
3
 * Copyright (c) 2009 Konstantin Shishkov
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include "libavcodec/bytestream.h"
#include "libavutil/avstring.h"
24
#include "libavutil/intfloat.h"
25 26 27 28
#include "avformat.h"

#include "rtmppkt.h"
#include "flv.h"
29
#include "url.h"
30 31 32 33 34 35 36 37 38 39

void ff_amf_write_bool(uint8_t **dst, int val)
{
    bytestream_put_byte(dst, AMF_DATA_TYPE_BOOL);
    bytestream_put_byte(dst, val);
}

void ff_amf_write_number(uint8_t **dst, double val)
{
    bytestream_put_byte(dst, AMF_DATA_TYPE_NUMBER);
40
    bytestream_put_be64(dst, av_double2int(val));
41 42 43 44 45 46 47 48 49
}

void ff_amf_write_string(uint8_t **dst, const char *str)
{
    bytestream_put_byte(dst, AMF_DATA_TYPE_STRING);
    bytestream_put_be16(dst, strlen(str));
    bytestream_put_buffer(dst, str, strlen(str));
}

50 51 52 53 54 55 56 57 58 59 60 61 62
void ff_amf_write_string2(uint8_t **dst, const char *str1, const char *str2)
{
    int len1 = 0, len2 = 0;
    if (str1)
        len1 = strlen(str1);
    if (str2)
        len2 = strlen(str2);
    bytestream_put_byte(dst, AMF_DATA_TYPE_STRING);
    bytestream_put_be16(dst, len1 + len2);
    bytestream_put_buffer(dst, str1, len1);
    bytestream_put_buffer(dst, str2, len2);
}

63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
void ff_amf_write_null(uint8_t **dst)
{
    bytestream_put_byte(dst, AMF_DATA_TYPE_NULL);
}

void ff_amf_write_object_start(uint8_t **dst)
{
    bytestream_put_byte(dst, AMF_DATA_TYPE_OBJECT);
}

void ff_amf_write_field_name(uint8_t **dst, const char *str)
{
    bytestream_put_be16(dst, strlen(str));
    bytestream_put_buffer(dst, str, strlen(str));
}

void ff_amf_write_object_end(uint8_t **dst)
{
    /* first two bytes are field name length = 0,
     * AMF object should end with it and end marker
     */
    bytestream_put_be24(dst, AMF_DATA_TYPE_OBJECT_END);
}

87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
int ff_amf_read_bool(GetByteContext *bc, int *val)
{
    if (bytestream2_get_byte(bc) != AMF_DATA_TYPE_BOOL)
        return AVERROR_INVALIDDATA;
    *val = bytestream2_get_byte(bc);
    return 0;
}

int ff_amf_read_number(GetByteContext *bc, double *val)
{
    uint64_t read;
    if (bytestream2_get_byte(bc) != AMF_DATA_TYPE_NUMBER)
        return AVERROR_INVALIDDATA;
    read = bytestream2_get_be64(bc);
    *val = av_int2double(read);
    return 0;
}

105 106
int ff_amf_get_string(GetByteContext *bc, uint8_t *str,
                      int strsize, int *length)
107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
{
    int stringlen = 0;
    int readsize;
    stringlen = bytestream2_get_be16(bc);
    if (stringlen + 1 > strsize)
        return AVERROR(EINVAL);
    readsize = bytestream2_get_buffer(bc, str, stringlen);
    if (readsize != stringlen) {
        av_log(NULL, AV_LOG_WARNING,
               "Unable to read as many bytes as AMF string signaled\n");
    }
    str[readsize] = '\0';
    *length = FFMIN(stringlen, readsize);
    return 0;
}

123 124 125 126 127 128 129 130
int ff_amf_read_string(GetByteContext *bc, uint8_t *str,
                       int strsize, int *length)
{
    if (bytestream2_get_byte(bc) != AMF_DATA_TYPE_STRING)
        return AVERROR_INVALIDDATA;
    return ff_amf_get_string(bc, str, strsize, length);
}

131 132 133 134 135 136 137
int ff_amf_read_null(GetByteContext *bc)
{
    if (bytestream2_get_byte(bc) != AMF_DATA_TYPE_NULL)
        return AVERROR_INVALIDDATA;
    return 0;
}

138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
int ff_rtmp_check_alloc_array(RTMPPacket **prev_pkt, int *nb_prev_pkt,
                              int channel)
{
    int nb_alloc;
    RTMPPacket *ptr;
    if (channel < *nb_prev_pkt)
        return 0;

    nb_alloc = channel + 16;
    // This can't use the av_reallocp family of functions, since we
    // would need to free each element in the array before the array
    // itself is freed.
    ptr = av_realloc_array(*prev_pkt, nb_alloc, sizeof(**prev_pkt));
    if (!ptr)
        return AVERROR(ENOMEM);
    memset(ptr + *nb_prev_pkt, 0, (nb_alloc - *nb_prev_pkt) * sizeof(*ptr));
    *prev_pkt = ptr;
    *nb_prev_pkt = nb_alloc;
    return 0;
}

159
int ff_rtmp_packet_read(URLContext *h, RTMPPacket *p,
160
                        int chunk_size, RTMPPacket **prev_pkt, int *nb_prev_pkt)
161
{
162 163 164 165 166
    uint8_t hdr;

    if (ffurl_read(h, &hdr, 1) != 1)
        return AVERROR(EIO);

167 168
    return ff_rtmp_packet_read_internal(h, p, chunk_size, prev_pkt,
                                        nb_prev_pkt, hdr);
169 170
}

171
static int rtmp_packet_read_one_chunk(URLContext *h, RTMPPacket *p,
172 173
                                      int chunk_size, RTMPPacket **prev_pkt_ptr,
                                      int *nb_prev_pkt, uint8_t hdr)
174 175
{

176 177
    uint8_t buf[16];
    int channel_id, timestamp, size;
178
    uint32_t ts_field; // non-extended timestamp or delta field
179
    uint32_t extra = 0;
180
    enum RTMPPacketType type;
181
    int written = 0;
182
    int ret, toread;
183
    RTMPPacket *prev_pkt;
184

185
    written++;
186 187
    channel_id = hdr & 0x3F;

188 189
    if (channel_id < 2) { //special case for channel number >= 64
        buf[1] = 0;
190
        if (ffurl_read_complete(h, buf, channel_id + 1) != channel_id + 1)
191
            return AVERROR(EIO);
192
        written += channel_id + 1;
193 194
        channel_id = AV_RL16(buf) + 64;
    }
195 196 197 198
    if ((ret = ff_rtmp_check_alloc_array(prev_pkt_ptr, nb_prev_pkt,
                                         channel_id)) < 0)
        return ret;
    prev_pkt = *prev_pkt_ptr;
199 200 201
    size  = prev_pkt[channel_id].size;
    type  = prev_pkt[channel_id].type;
    extra = prev_pkt[channel_id].extra;
202

203
    hdr >>= 6; // header size indicator
204
    if (hdr == RTMP_PS_ONEBYTE) {
205
        ts_field = prev_pkt[channel_id].ts_field;
206
    } else {
207
        if (ffurl_read_complete(h, buf, 3) != 3)
208
            return AVERROR(EIO);
209
        written += 3;
210
        ts_field = AV_RB24(buf);
211
        if (hdr != RTMP_PS_FOURBYTES) {
212
            if (ffurl_read_complete(h, buf, 3) != 3)
213
                return AVERROR(EIO);
214 215
            written += 3;
            size = AV_RB24(buf);
216
            if (ffurl_read_complete(h, buf, 1) != 1)
217
                return AVERROR(EIO);
218
            written++;
219
            type = buf[0];
220
            if (hdr == RTMP_PS_TWELVEBYTES) {
221
                if (ffurl_read_complete(h, buf, 4) != 4)
222
                    return AVERROR(EIO);
223
                written += 4;
224 225 226
                extra = AV_RL32(buf);
            }
        }
227 228 229 230 231 232 233
    }
    if (ts_field == 0xFFFFFF) {
        if (ffurl_read_complete(h, buf, 4) != 4)
            return AVERROR(EIO);
        timestamp = AV_RB32(buf);
    } else {
        timestamp = ts_field;
234
    }
235 236 237
    if (hdr != RTMP_PS_TWELVEBYTES)
        timestamp += prev_pkt[channel_id].timestamp;

238 239 240 241 242 243 244 245
    if (prev_pkt[channel_id].read && size != prev_pkt[channel_id].size) {
        av_log(NULL, AV_LOG_ERROR, "RTMP packet size mismatch %d != %d\n",
                size,
                prev_pkt[channel_id].size);
        ff_rtmp_packet_destroy(&prev_pkt[channel_id]);
        prev_pkt[channel_id].read = 0;
    }

246 247 248 249 250 251
    if (!prev_pkt[channel_id].read) {
        if ((ret = ff_rtmp_packet_create(p, channel_id, type, timestamp,
                                         size)) < 0)
            return ret;
        p->read = written;
        p->offset = 0;
252
        prev_pkt[channel_id].ts_field   = ts_field;
253 254 255 256 257 258 259 260
        prev_pkt[channel_id].timestamp  = timestamp;
    } else {
        // previous packet in this channel hasn't completed reading
        RTMPPacket *prev = &prev_pkt[channel_id];
        p->data          = prev->data;
        p->size          = prev->size;
        p->channel_id    = prev->channel_id;
        p->type          = prev->type;
261
        p->ts_field      = prev->ts_field;
262 263 264 265 266 267
        p->extra         = prev->extra;
        p->offset        = prev->offset;
        p->read          = prev->read + written;
        p->timestamp     = prev->timestamp;
        prev->data       = NULL;
    }
268 269 270 271
    p->extra = extra;
    // save history
    prev_pkt[channel_id].channel_id = channel_id;
    prev_pkt[channel_id].type       = type;
272
    prev_pkt[channel_id].size       = size;
273
    prev_pkt[channel_id].extra      = extra;
274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289
    size = size - p->offset;

    toread = FFMIN(size, chunk_size);
    if (ffurl_read_complete(h, p->data + p->offset, toread) != toread) {
        ff_rtmp_packet_destroy(p);
        return AVERROR(EIO);
    }
    size      -= toread;
    p->read   += toread;
    p->offset += toread;

    if (size > 0) {
       RTMPPacket *prev = &prev_pkt[channel_id];
       prev->data = p->data;
       prev->read = p->read;
       prev->offset = p->offset;
290
       p->data      = NULL;
291 292 293 294 295 296 297 298
       return AVERROR(EAGAIN);
    }

    prev_pkt[channel_id].read = 0; // read complete; reset if needed
    return p->read;
}

int ff_rtmp_packet_read_internal(URLContext *h, RTMPPacket *p, int chunk_size,
299 300
                                 RTMPPacket **prev_pkt, int *nb_prev_pkt,
                                 uint8_t hdr)
301 302
{
    while (1) {
303 304
        int ret = rtmp_packet_read_one_chunk(h, p, chunk_size, prev_pkt,
                                             nb_prev_pkt, hdr);
305 306 307 308
        if (ret > 0 || ret != AVERROR(EAGAIN))
            return ret;

        if (ffurl_read(h, &hdr, 1) != 1)
309 310 311 312 313
            return AVERROR(EIO);
    }
}

int ff_rtmp_packet_write(URLContext *h, RTMPPacket *pkt,
314 315
                         int chunk_size, RTMPPacket **prev_pkt_ptr,
                         int *nb_prev_pkt)
316 317 318 319
{
    uint8_t pkt_hdr[16], *p = pkt_hdr;
    int mode = RTMP_PS_TWELVEBYTES;
    int off = 0;
320
    int written = 0;
321
    int ret;
322
    RTMPPacket *prev_pkt;
323 324
    int use_delta; // flag if using timestamp delta, not RTMP_PS_TWELVEBYTES
    uint32_t timestamp; // full 32-bit timestamp or delta value
325 326 327 328 329

    if ((ret = ff_rtmp_check_alloc_array(prev_pkt_ptr, nb_prev_pkt,
                                         pkt->channel_id)) < 0)
        return ret;
    prev_pkt = *prev_pkt_ptr;
330

331
    //if channel_id = 0, this is first presentation of prev_pkt, send full hdr.
332
    use_delta = prev_pkt[pkt->channel_id].channel_id &&
333
        pkt->extra == prev_pkt[pkt->channel_id].extra &&
334 335 336 337 338 339 340
        pkt->timestamp >= prev_pkt[pkt->channel_id].timestamp;

    timestamp = pkt->timestamp;
    if (use_delta) {
        timestamp -= prev_pkt[pkt->channel_id].timestamp;
    }
    if (timestamp >= 0xFFFFFF) {
341
        pkt->ts_field = 0xFFFFFF;
342
    } else {
343
        pkt->ts_field = timestamp;
344 345 346
    }

    if (use_delta) {
347
        if (pkt->type == prev_pkt[pkt->channel_id].type &&
348
            pkt->size == prev_pkt[pkt->channel_id].size) {
349
            mode = RTMP_PS_FOURBYTES;
350
            if (pkt->ts_field == prev_pkt[pkt->channel_id].ts_field)
351 352 353 354 355 356
                mode = RTMP_PS_ONEBYTE;
        } else {
            mode = RTMP_PS_EIGHTBYTES;
        }
    }

357 358 359 360 361 362 363 364 365
    if (pkt->channel_id < 64) {
        bytestream_put_byte(&p, pkt->channel_id | (mode << 6));
    } else if (pkt->channel_id < 64 + 256) {
        bytestream_put_byte(&p, 0               | (mode << 6));
        bytestream_put_byte(&p, pkt->channel_id - 64);
    } else {
        bytestream_put_byte(&p, 1               | (mode << 6));
        bytestream_put_le16(&p, pkt->channel_id - 64);
    }
366
    if (mode != RTMP_PS_ONEBYTE) {
367
        bytestream_put_be24(&p, pkt->ts_field);
368
        if (mode != RTMP_PS_FOURBYTES) {
369
            bytestream_put_be24(&p, pkt->size);
370 371 372 373 374
            bytestream_put_byte(&p, pkt->type);
            if (mode == RTMP_PS_TWELVEBYTES)
                bytestream_put_le32(&p, pkt->extra);
        }
    }
375
    if (pkt->ts_field == 0xFFFFFF)
376
        bytestream_put_be32(&p, timestamp);
377 378 379
    // save history
    prev_pkt[pkt->channel_id].channel_id = pkt->channel_id;
    prev_pkt[pkt->channel_id].type       = pkt->type;
380
    prev_pkt[pkt->channel_id].size       = pkt->size;
381
    prev_pkt[pkt->channel_id].timestamp  = pkt->timestamp;
382
    prev_pkt[pkt->channel_id].ts_field   = pkt->ts_field;
383 384
    prev_pkt[pkt->channel_id].extra      = pkt->extra;

385 386
    if ((ret = ffurl_write(h, pkt_hdr, p - pkt_hdr)) < 0)
        return ret;
387 388 389
    written = p - pkt_hdr + pkt->size;
    while (off < pkt->size) {
        int towrite = FFMIN(chunk_size, pkt->size - off);
390 391
        if ((ret = ffurl_write(h, pkt->data + off, towrite)) < 0)
            return ret;
392
        off += towrite;
393
        if (off < pkt->size) {
394
            uint8_t marker = 0xC0 | pkt->channel_id;
395 396
            if ((ret = ffurl_write(h, &marker, 1)) < 0)
                return ret;
397
            written++;
398 399 400 401 402 403 404
            if (pkt->ts_field == 0xFFFFFF) {
                uint8_t ts_header[4];
                AV_WB32(ts_header, timestamp);
                if ((ret = ffurl_write(h, ts_header, 4)) < 0)
                    return ret;
                written += 4;
            }
405 406
        }
    }
407
    return written;
408 409 410 411 412
}

int ff_rtmp_packet_create(RTMPPacket *pkt, int channel_id, RTMPPacketType type,
                          int timestamp, int size)
{
413
    if (size) {
414
        pkt->data = av_realloc(NULL, size);
Martin Storsjö's avatar
Martin Storsjö committed
415 416
        if (!pkt->data)
            return AVERROR(ENOMEM);
417
    }
418
    pkt->size       = size;
419 420 421 422
    pkt->channel_id = channel_id;
    pkt->type       = type;
    pkt->timestamp  = timestamp;
    pkt->extra      = 0;
423
    pkt->ts_field   = 0;
424 425 426 427 428 429 430 431 432

    return 0;
}

void ff_rtmp_packet_destroy(RTMPPacket *pkt)
{
    if (!pkt)
        return;
    av_freep(&pkt->data);
433
    pkt->size = 0;
434 435 436 437 438
}

int ff_amf_tag_size(const uint8_t *data, const uint8_t *data_end)
{
    const uint8_t *base = data;
439 440 441
    AMFDataType type;
    unsigned nb   = -1;
    int parse_key = 1;
442 443 444

    if (data >= data_end)
        return -1;
445
    switch ((type = *data++)) {
446 447 448 449 450
    case AMF_DATA_TYPE_NUMBER:      return 9;
    case AMF_DATA_TYPE_BOOL:        return 2;
    case AMF_DATA_TYPE_STRING:      return 3 + AV_RB16(data);
    case AMF_DATA_TYPE_LONG_STRING: return 5 + AV_RB32(data);
    case AMF_DATA_TYPE_NULL:        return 1;
451
    case AMF_DATA_TYPE_DATE:        return 11;
452
    case AMF_DATA_TYPE_ARRAY:
453 454 455
        parse_key = 0;
    case AMF_DATA_TYPE_MIXEDARRAY:
        nb = bytestream_get_be32(&data);
456
    case AMF_DATA_TYPE_OBJECT:
457
        while (nb-- > 0 || type != AMF_DATA_TYPE_ARRAY) {
458
            int t;
459 460 461 462 463 464 465 466 467
            if (parse_key) {
                int size = bytestream_get_be16(&data);
                if (!size) {
                    data++;
                    break;
                }
                if (size < 0 || size >= data_end - data)
                    return -1;
                data += size;
468 469
            }
            t = ff_amf_tag_size(data, data_end);
470
            if (t < 0 || t >= data_end - data)
471 472 473 474 475 476 477 478 479 480 481 482 483 484 485
                return -1;
            data += t;
        }
        return data - base;
    case AMF_DATA_TYPE_OBJECT_END:  return 1;
    default:                        return -1;
    }
}

int ff_amf_get_field_value(const uint8_t *data, const uint8_t *data_end,
                           const uint8_t *name, uint8_t *dst, int dst_size)
{
    int namelen = strlen(name);
    int len;

486 487 488 489 490 491
    while (*data != AMF_DATA_TYPE_OBJECT && data < data_end) {
        len = ff_amf_tag_size(data, data_end);
        if (len < 0)
            len = data_end - data;
        data += len;
    }
492 493
    if (data_end - data < 3)
        return -1;
494
    data++;
495 496 497 498
    for (;;) {
        int size = bytestream_get_be16(&data);
        if (!size)
            break;
499
        if (size < 0 || size >= data_end - data)
500 501 502 503 504
            return -1;
        data += size;
        if (size == namelen && !memcmp(data-size, name, namelen)) {
            switch (*data++) {
            case AMF_DATA_TYPE_NUMBER:
505
                snprintf(dst, dst_size, "%g", av_int2double(AV_RB64(data)));
506 507 508 509 510 511 512 513 514 515 516 517 518 519
                break;
            case AMF_DATA_TYPE_BOOL:
                snprintf(dst, dst_size, "%s", *data ? "true" : "false");
                break;
            case AMF_DATA_TYPE_STRING:
                len = bytestream_get_be16(&data);
                av_strlcpy(dst, data, FFMIN(len+1, dst_size));
                break;
            default:
                return -1;
            }
            return 0;
        }
        len = ff_amf_tag_size(data, data_end);
520
        if (len < 0 || len >= data_end - data)
521 522 523 524 525
            return -1;
        data += len;
    }
    return -1;
}
526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547

static const char* rtmp_packet_type(int type)
{
    switch (type) {
    case RTMP_PT_CHUNK_SIZE:     return "chunk size";
    case RTMP_PT_BYTES_READ:     return "bytes read";
    case RTMP_PT_PING:           return "ping";
    case RTMP_PT_SERVER_BW:      return "server bandwidth";
    case RTMP_PT_CLIENT_BW:      return "client bandwidth";
    case RTMP_PT_AUDIO:          return "audio packet";
    case RTMP_PT_VIDEO:          return "video packet";
    case RTMP_PT_FLEX_STREAM:    return "Flex shared stream";
    case RTMP_PT_FLEX_OBJECT:    return "Flex shared object";
    case RTMP_PT_FLEX_MESSAGE:   return "Flex shared message";
    case RTMP_PT_NOTIFY:         return "notification";
    case RTMP_PT_SHARED_OBJ:     return "shared object";
    case RTMP_PT_INVOKE:         return "invoke";
    case RTMP_PT_METADATA:       return "metadata";
    default:                     return "unknown";
    }
}

548 549
static void amf_tag_contents(void *ctx, const uint8_t *data,
                             const uint8_t *data_end)
550
{
551
    unsigned int size, nb = -1;
552
    char buf[1024];
553 554
    AMFDataType type;
    int parse_key = 1;
555 556 557

    if (data >= data_end)
        return;
558
    switch ((type = *data++)) {
559
    case AMF_DATA_TYPE_NUMBER:
560
        av_log(ctx, AV_LOG_DEBUG, " number %g\n", av_int2double(AV_RB64(data)));
561 562 563 564 565 566
        return;
    case AMF_DATA_TYPE_BOOL:
        av_log(ctx, AV_LOG_DEBUG, " bool %d\n", *data);
        return;
    case AMF_DATA_TYPE_STRING:
    case AMF_DATA_TYPE_LONG_STRING:
567
        if (type == AMF_DATA_TYPE_STRING) {
568 569
            size = bytestream_get_be16(&data);
        } else {
570
            size = bytestream_get_be32(&data);
571
        }
572
        size = FFMIN(size, sizeof(buf) - 1);
573 574 575 576 577 578 579 580
        memcpy(buf, data, size);
        buf[size] = 0;
        av_log(ctx, AV_LOG_DEBUG, " string '%s'\n", buf);
        return;
    case AMF_DATA_TYPE_NULL:
        av_log(ctx, AV_LOG_DEBUG, " NULL\n");
        return;
    case AMF_DATA_TYPE_ARRAY:
581 582 583
        parse_key = 0;
    case AMF_DATA_TYPE_MIXEDARRAY:
        nb = bytestream_get_be32(&data);
584 585
    case AMF_DATA_TYPE_OBJECT:
        av_log(ctx, AV_LOG_DEBUG, " {\n");
586
        while (nb-- > 0 || type != AMF_DATA_TYPE_ARRAY) {
587
            int t;
588 589 590 591 592 593 594 595 596 597 598 599 600 601
            if (parse_key) {
                size = bytestream_get_be16(&data);
                size = FFMIN(size, sizeof(buf) - 1);
                if (!size) {
                    av_log(ctx, AV_LOG_DEBUG, " }\n");
                    data++;
                    break;
                }
                memcpy(buf, data, size);
                buf[size] = 0;
                if (size >= data_end - data)
                    return;
                data += size;
                av_log(ctx, AV_LOG_DEBUG, "  %s: ", buf);
602
            }
603
            amf_tag_contents(ctx, data, data_end);
604
            t = ff_amf_tag_size(data, data_end);
605
            if (t < 0 || t >= data_end - data)
606 607 608 609 610 611 612 613 614 615 616 617 618 619 620
                return;
            data += t;
        }
        return;
    case AMF_DATA_TYPE_OBJECT_END:
        av_log(ctx, AV_LOG_DEBUG, " }\n");
        return;
    default:
        return;
    }
}

void ff_rtmp_packet_dump(void *ctx, RTMPPacket *p)
{
    av_log(ctx, AV_LOG_DEBUG, "RTMP packet type '%s'(%d) for channel %d, timestamp %d, extra field %d size %d\n",
621
           rtmp_packet_type(p->type), p->type, p->channel_id, p->timestamp, p->extra, p->size);
622
    if (p->type == RTMP_PT_INVOKE || p->type == RTMP_PT_NOTIFY) {
623
        uint8_t *src = p->data, *src_end = p->data + p->size;
624 625
        while (src < src_end) {
            int sz;
626
            amf_tag_contents(ctx, src, src_end);
627 628 629 630 631 632 633 634 635 636 637
            sz = ff_amf_tag_size(src, src_end);
            if (sz < 0)
                break;
            src += sz;
        }
    } else if (p->type == RTMP_PT_SERVER_BW){
        av_log(ctx, AV_LOG_DEBUG, "Server BW = %d\n", AV_RB32(p->data));
    } else if (p->type == RTMP_PT_CLIENT_BW){
        av_log(ctx, AV_LOG_DEBUG, "Client BW = %d\n", AV_RB32(p->data));
    } else if (p->type != RTMP_PT_AUDIO && p->type != RTMP_PT_VIDEO && p->type != RTMP_PT_METADATA) {
        int i;
638
        for (i = 0; i < p->size; i++)
639 640 641 642
            av_log(ctx, AV_LOG_DEBUG, " %02X", p->data[i]);
        av_log(ctx, AV_LOG_DEBUG, "\n");
    }
}
643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675

int ff_amf_match_string(const uint8_t *data, int size, const char *str)
{
    int len = strlen(str);
    int amf_len, type;

    if (size < 1)
        return 0;

    type = *data++;

    if (type != AMF_DATA_TYPE_LONG_STRING &&
        type != AMF_DATA_TYPE_STRING)
        return 0;

    if (type == AMF_DATA_TYPE_LONG_STRING) {
        if ((size -= 4 + 1) < 0)
            return 0;
        amf_len = bytestream_get_be32(&data);
    } else {
        if ((size -= 2 + 1) < 0)
            return 0;
        amf_len = bytestream_get_be16(&data);
    }

    if (amf_len > size)
        return 0;

    if (amf_len != len)
        return 0;

    return !memcmp(data, str, len);
}