git.sesse.net Git - ffmpeg/blob - libavformat/rtmpproto.c

   1 /*
   2  * RTMP network protocol
   3  * Copyright (c) 2009 Konstantin Shishkov
   4  *
   5  * This file is part of Libav.
   6  *
   7  * Libav is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * Libav is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with Libav; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 /**
  23  * @file
  24  * RTMP protocol
  25  */
  26
  27 #include "libavcodec/bytestream.h"
  28 #include "libavutil/avstring.h"
  29 #include "libavutil/base64.h"
  30 #include "libavutil/hmac.h"
  31 #include "libavutil/intfloat.h"
  32 #include "libavutil/lfg.h"
  33 #include "libavutil/md5.h"
  34 #include "libavutil/opt.h"
  35 #include "libavutil/random_seed.h"
  36 #include "avformat.h"
  37 #include "internal.h"
  38
  39 #include "network.h"
  40
  41 #include "flv.h"
  42 #include "rtmp.h"
  43 #include "rtmpcrypt.h"
  44 #include "rtmppkt.h"
  45 #include "url.h"
  46
  47 #if CONFIG_ZLIB
  48 #include <zlib.h>
  49 #endif
  50
  51 #define APP_MAX_LENGTH 128
  52 #define PLAYPATH_MAX_LENGTH 256
  53 #define TCURL_MAX_LENGTH 512
  54 #define FLASHVER_MAX_LENGTH 64
  55 #define RTMP_PKTDATA_DEFAULT_SIZE 4096
  56 #define RTMP_HEADER 11
  57
  58 /** RTMP protocol handler state */
  59 typedef enum {
  60     STATE_START,      ///< client has not done anything yet
  61     STATE_HANDSHAKED, ///< client has performed handshake
  62     STATE_FCPUBLISH,  ///< client FCPublishing stream (for output)
  63     STATE_PLAYING,    ///< client has started receiving multimedia data from server
  64     STATE_SEEKING,    ///< client has started the seek operation. Back on STATE_PLAYING when the time comes
  65     STATE_PUBLISHING, ///< client has started sending multimedia data to server (for output)
  66     STATE_RECEIVING,  ///< received a publish command (for input)
  67     STATE_SENDING,    ///< received a play command (for output)
  68     STATE_STOPPED,    ///< the broadcast has been stopped
  69 } ClientState;
  70
  71 typedef struct TrackedMethod {
  72     char *name;
  73     int id;
  74 } TrackedMethod;
  75
  76 /** protocol handler context */
  77 typedef struct RTMPContext {
  78     const AVClass *class;
  79     URLContext*   stream;                     ///< TCP stream used in interactions with RTMP server
  80     RTMPPacket    *prev_pkt[2];               ///< packet history used when reading and sending packets ([0] for reading, [1] for writing)
  81     int           nb_prev_pkt[2];             ///< number of elements in prev_pkt
  82     int           in_chunk_size;              ///< size of the chunks incoming RTMP packets are divided into
  83     int           out_chunk_size;             ///< size of the chunks outgoing RTMP packets are divided into
  84     int           is_input;                   ///< input/output flag
  85     char          *playpath;                  ///< stream identifier to play (with possible "mp4:" prefix)
  86     int           live;                       ///< 0: recorded, -1: live, -2: both
  87     char          *app;                       ///< name of application
  88     char          *conn;                      ///< append arbitrary AMF data to the Connect message
  89     ClientState   state;                      ///< current state
  90     int           stream_id;                  ///< ID assigned by the server for the stream
  91     uint8_t*      flv_data;                   ///< buffer with data for demuxer
  92     int           flv_size;                   ///< current buffer size
  93     int           flv_off;                    ///< number of bytes read from current buffer
  94     int           flv_nb_packets;             ///< number of flv packets published
  95     RTMPPacket    out_pkt;                    ///< rtmp packet, created from flv a/v or metadata (for output)
  96     uint32_t      client_report_size;         ///< number of bytes after which client should report to server
  97     uint32_t      bytes_read;                 ///< number of bytes read from server
  98     uint32_t      last_bytes_read;            ///< number of bytes read last reported to server
  99     uint32_t      last_timestamp;             ///< last timestamp received in a packet
 100     int           skip_bytes;                 ///< number of bytes to skip from the input FLV stream in the next write call
 101     int           has_audio;                  ///< presence of audio data
 102     int           has_video;                  ///< presence of video data
 103     int           received_metadata;          ///< Indicates if we have received metadata about the streams
 104     uint8_t       flv_header[RTMP_HEADER];    ///< partial incoming flv packet header
 105     int           flv_header_bytes;           ///< number of initialized bytes in flv_header
 106     int           nb_invokes;                 ///< keeps track of invoke messages
 107     char*         tcurl;                      ///< url of the target stream
 108     char*         flashver;                   ///< version of the flash plugin
 109     char*         swfhash;                    ///< SHA256 hash of the decompressed SWF file (32 bytes)
 110     int           swfhash_len;                ///< length of the SHA256 hash
 111     int           swfsize;                    ///< size of the decompressed SWF file
 112     char*         swfurl;                     ///< url of the swf player
 113     char*         swfverify;                  ///< URL to player swf file, compute hash/size automatically
 114     char          swfverification[42];        ///< hash of the SWF verification
 115     char*         pageurl;                    ///< url of the web page
 116     char*         subscribe;                  ///< name of live stream to subscribe
 117     int           server_bw;                  ///< server bandwidth
 118     int           client_buffer_time;         ///< client buffer time in ms
 119     int           flush_interval;             ///< number of packets flushed in the same request (RTMPT only)
 120     int           encrypted;                  ///< use an encrypted connection (RTMPE only)
 121     TrackedMethod*tracked_methods;            ///< tracked methods buffer
 122     int           nb_tracked_methods;         ///< number of tracked methods
 123     int           tracked_methods_size;       ///< size of the tracked methods buffer
 124     int           listen;                     ///< listen mode flag
 125     int           listen_timeout;             ///< listen timeout to wait for new connections
 126     int           nb_streamid;                ///< The next stream id to return on createStream calls
 127     double        duration;                   ///< Duration of the stream in seconds as returned by the server (only valid if non-zero)
 128     char          username[50];
 129     char          password[50];
 130     char          auth_params[500];
 131     int           do_reconnect;
 132     int           auth_tried;
 133 } RTMPContext;
 134
 135 #define PLAYER_KEY_OPEN_PART_LEN 30   ///< length of partial key used for first client digest signing
 136 /** Client key used for digest signing */
 137 static const uint8_t rtmp_player_key[] = {
 138     'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
 139     'F', 'l', 'a', 's', 'h', ' ', 'P', 'l', 'a', 'y', 'e', 'r', ' ', '0', '0', '1',
 140
 141     0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1, 0x02,
 142     0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB, 0x93, 0xB8,
 143     0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE
 144 };
 145
 146 #define SERVER_KEY_OPEN_PART_LEN 36   ///< length of partial key used for first server digest signing
 147 /** Key used for RTMP server digest signing */
 148 static const uint8_t rtmp_server_key[] = {
 149     'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
 150     'F', 'l', 'a', 's', 'h', ' ', 'M', 'e', 'd', 'i', 'a', ' ',
 151     'S', 'e', 'r', 'v', 'e', 'r', ' ', '0', '0', '1',
 152
 153     0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1, 0x02,
 154     0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB, 0x93, 0xB8,
 155     0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE
 156 };
 157
 158 static int handle_chunk_size(URLContext *s, RTMPPacket *pkt);
 159
 160 static int add_tracked_method(RTMPContext *rt, const char *name, int id)
 161 {
 162     int err;
 163
 164     if (rt->nb_tracked_methods + 1 > rt->tracked_methods_size) {
 165         rt->tracked_methods_size = (rt->nb_tracked_methods + 1) * 2;
 166         if ((err = av_reallocp(&rt->tracked_methods, rt->tracked_methods_size *
 167                                sizeof(*rt->tracked_methods))) < 0) {
 168             rt->nb_tracked_methods = 0;
 169             rt->tracked_methods_size = 0;
 170             return err;
 171         }
 172     }
 173
 174     rt->tracked_methods[rt->nb_tracked_methods].name = av_strdup(name);
 175     if (!rt->tracked_methods[rt->nb_tracked_methods].name)
 176         return AVERROR(ENOMEM);
 177     rt->tracked_methods[rt->nb_tracked_methods].id = id;
 178     rt->nb_tracked_methods++;
 179
 180     return 0;
 181 }
 182
 183 static void del_tracked_method(RTMPContext *rt, int index)
 184 {
 185     memmove(&rt->tracked_methods[index], &rt->tracked_methods[index + 1],
 186             sizeof(*rt->tracked_methods) * (rt->nb_tracked_methods - index - 1));
 187     rt->nb_tracked_methods--;
 188 }
 189
 190 static int find_tracked_method(URLContext *s, RTMPPacket *pkt, int offset,
 191                                char **tracked_method)
 192 {
 193     RTMPContext *rt = s->priv_data;
 194     GetByteContext gbc;
 195     double pkt_id;
 196     int ret;
 197     int i;
 198
 199     bytestream2_init(&gbc, pkt->data + offset, pkt->size - offset);
 200     if ((ret = ff_amf_read_number(&gbc, &pkt_id)) < 0)
 201         return ret;
 202
 203     for (i = 0; i < rt->nb_tracked_methods; i++) {
 204         if (rt->tracked_methods[i].id != pkt_id)
 205             continue;
 206
 207         *tracked_method = rt->tracked_methods[i].name;
 208         del_tracked_method(rt, i);
 209         break;
 210     }
 211
 212     return 0;
 213 }
 214
 215 static void free_tracked_methods(RTMPContext *rt)
 216 {
 217     int i;
 218
 219     for (i = 0; i < rt->nb_tracked_methods; i ++)
 220         av_free(rt->tracked_methods[i].name);
 221     av_free(rt->tracked_methods);
 222     rt->tracked_methods      = NULL;
 223     rt->tracked_methods_size = 0;
 224     rt->nb_tracked_methods   = 0;
 225 }
 226
 227 static int rtmp_send_packet(RTMPContext *rt, RTMPPacket *pkt, int track)
 228 {
 229     int ret;
 230
 231     if (pkt->type == RTMP_PT_INVOKE && track) {
 232         GetByteContext gbc;
 233         char name[128];
 234         double pkt_id;
 235         int len;
 236
 237         bytestream2_init(&gbc, pkt->data, pkt->size);
 238         if ((ret = ff_amf_read_string(&gbc, name, sizeof(name), &len)) < 0)
 239             goto fail;
 240
 241         if ((ret = ff_amf_read_number(&gbc, &pkt_id)) < 0)
 242             goto fail;
 243
 244         if ((ret = add_tracked_method(rt, name, pkt_id)) < 0)
 245             goto fail;
 246     }
 247
 248     ret = ff_rtmp_packet_write(rt->stream, pkt, rt->out_chunk_size,
 249                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
 250 fail:
 251     ff_rtmp_packet_destroy(pkt);
 252     return ret;
 253 }
 254
 255 static int rtmp_write_amf_data(URLContext *s, char *param, uint8_t **p)
 256 {
 257     char *field, *value;
 258     char type;
 259
 260     /* The type must be B for Boolean, N for number, S for string, O for
 261      * object, or Z for null. For Booleans the data must be either 0 or 1 for
 262      * FALSE or TRUE, respectively. Likewise for Objects the data must be
 263      * 0 or 1 to end or begin an object, respectively. Data items in subobjects
 264      * may be named, by prefixing the type with 'N' and specifying the name
 265      * before the value (ie. NB:myFlag:1). This option may be used multiple times
 266      * to construct arbitrary AMF sequences. */
 267     if (param[0] && param[1] == ':') {
 268         type = param[0];
 269         value = param + 2;
 270     } else if (param[0] == 'N' && param[1] && param[2] == ':') {
 271         type = param[1];
 272         field = param + 3;
 273         value = strchr(field, ':');
 274         if (!value)
 275             goto fail;
 276         *value = '\0';
 277         value++;
 278
 279         ff_amf_write_field_name(p, field);
 280     } else {
 281         goto fail;
 282     }
 283
 284     switch (type) {
 285     case 'B':
 286         ff_amf_write_bool(p, value[0] != '0');
 287         break;
 288     case 'S':
 289         ff_amf_write_string(p, value);
 290         break;
 291     case 'N':
 292         ff_amf_write_number(p, strtod(value, NULL));
 293         break;
 294     case 'Z':
 295         ff_amf_write_null(p);
 296         break;
 297     case 'O':
 298         if (value[0] != '0')
 299             ff_amf_write_object_start(p);
 300         else
 301             ff_amf_write_object_end(p);
 302         break;
 303     default:
 304         goto fail;
 305         break;
 306     }
 307
 308     return 0;
 309
 310 fail:
 311     av_log(s, AV_LOG_ERROR, "Invalid AMF parameter: %s\n", param);
 312     return AVERROR(EINVAL);
 313 }
 314
 315 /**
 316  * Generate 'connect' call and send it to the server.
 317  */
 318 static int gen_connect(URLContext *s, RTMPContext *rt)
 319 {
 320     RTMPPacket pkt;
 321     uint8_t *p;
 322     int ret;
 323
 324     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
 325                                      0, 4096)) < 0)
 326         return ret;
 327
 328     p = pkt.data;
 329
 330     ff_amf_write_string(&p, "connect");
 331     ff_amf_write_number(&p, ++rt->nb_invokes);
 332     ff_amf_write_object_start(&p);
 333     ff_amf_write_field_name(&p, "app");
 334     ff_amf_write_string2(&p, rt->app, rt->auth_params);
 335
 336     if (!rt->is_input) {
 337         ff_amf_write_field_name(&p, "type");
 338         ff_amf_write_string(&p, "nonprivate");
 339     }
 340     ff_amf_write_field_name(&p, "flashVer");
 341     ff_amf_write_string(&p, rt->flashver);
 342
 343     if (rt->swfurl) {
 344         ff_amf_write_field_name(&p, "swfUrl");
 345         ff_amf_write_string(&p, rt->swfurl);
 346     }
 347
 348     ff_amf_write_field_name(&p, "tcUrl");
 349     ff_amf_write_string2(&p, rt->tcurl, rt->auth_params);
 350     if (rt->is_input) {
 351         ff_amf_write_field_name(&p, "fpad");
 352         ff_amf_write_bool(&p, 0);
 353         ff_amf_write_field_name(&p, "capabilities");
 354         ff_amf_write_number(&p, 15.0);
 355
 356         /* Tell the server we support all the audio codecs except
 357          * SUPPORT_SND_INTEL (0x0008) and SUPPORT_SND_UNUSED (0x0010)
 358          * which are unused in the RTMP protocol implementation. */
 359         ff_amf_write_field_name(&p, "audioCodecs");
 360         ff_amf_write_number(&p, 4071.0);
 361         ff_amf_write_field_name(&p, "videoCodecs");
 362         ff_amf_write_number(&p, 252.0);
 363         ff_amf_write_field_name(&p, "videoFunction");
 364         ff_amf_write_number(&p, 1.0);
 365
 366         if (rt->pageurl) {
 367             ff_amf_write_field_name(&p, "pageUrl");
 368             ff_amf_write_string(&p, rt->pageurl);
 369         }
 370     }
 371     ff_amf_write_object_end(&p);
 372
 373     if (rt->conn) {
 374         char *param = rt->conn;
 375
 376         // Write arbitrary AMF data to the Connect message.
 377         while (param) {
 378             char *sep;
 379             param += strspn(param, " ");
 380             if (!*param)
 381                 break;
 382             sep = strchr(param, ' ');
 383             if (sep)
 384                 *sep = '\0';
 385             if ((ret = rtmp_write_amf_data(s, param, &p)) < 0) {
 386                 // Invalid AMF parameter.
 387                 ff_rtmp_packet_destroy(&pkt);
 388                 return ret;
 389             }
 390
 391             if (sep)
 392                 param = sep + 1;
 393             else
 394                 break;
 395         }
 396     }
 397
 398     pkt.size = p - pkt.data;
 399
 400     return rtmp_send_packet(rt, &pkt, 1);
 401 }
 402
 403 static int read_connect(URLContext *s, RTMPContext *rt)
 404 {
 405     RTMPPacket pkt = { 0 };
 406     uint8_t *p;
 407     const uint8_t *cp;
 408     int ret;
 409     char command[64];
 410     int stringlen;
 411     double seqnum;
 412     uint8_t tmpstr[256];
 413     GetByteContext gbc;
 414
 415     if ((ret = ff_rtmp_packet_read(rt->stream, &pkt, rt->in_chunk_size,
 416                                    &rt->prev_pkt[0], &rt->nb_prev_pkt[0])) < 0)
 417         return ret;
 418
 419     if (pkt.type == RTMP_PT_CHUNK_SIZE) {
 420         if ((ret = handle_chunk_size(s, &pkt)) < 0)
 421             return ret;
 422         ff_rtmp_packet_destroy(&pkt);
 423         if ((ret = ff_rtmp_packet_read(rt->stream, &pkt, rt->in_chunk_size,
 424                                        &rt->prev_pkt[0], &rt->nb_prev_pkt[0])) < 0)
 425             return ret;
 426     }
 427
 428     cp = pkt.data;
 429     bytestream2_init(&gbc, cp, pkt.size);
 430     if (ff_amf_read_string(&gbc, command, sizeof(command), &stringlen)) {
 431         av_log(s, AV_LOG_ERROR, "Unable to read command string\n");
 432         ff_rtmp_packet_destroy(&pkt);
 433         return AVERROR_INVALIDDATA;
 434     }
 435     if (strcmp(command, "connect")) {
 436         av_log(s, AV_LOG_ERROR, "Expecting connect, got %s\n", command);
 437         ff_rtmp_packet_destroy(&pkt);
 438         return AVERROR_INVALIDDATA;
 439     }
 440     ret = ff_amf_read_number(&gbc, &seqnum);
 441     if (ret)
 442         av_log(s, AV_LOG_WARNING, "SeqNum not found\n");
 443     /* Here one could parse an AMF Object with data as flashVers and others. */
 444     ret = ff_amf_get_field_value(gbc.buffer,
 445                                  gbc.buffer + bytestream2_get_bytes_left(&gbc),
 446                                  "app", tmpstr, sizeof(tmpstr));
 447     if (ret)
 448         av_log(s, AV_LOG_WARNING, "App field not found in connect\n");
 449     if (!ret && strcmp(tmpstr, rt->app))
 450         av_log(s, AV_LOG_WARNING, "App field don't match up: %s <-> %s\n",
 451                tmpstr, rt->app);
 452     ff_rtmp_packet_destroy(&pkt);
 453
 454     // Send Window Acknowledgement Size (as defined in specification)
 455     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL,
 456                                      RTMP_PT_SERVER_BW, 0, 4)) < 0)
 457         return ret;
 458     p = pkt.data;
 459     bytestream_put_be32(&p, rt->server_bw);
 460     pkt.size = p - pkt.data;
 461     ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
 462                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
 463     ff_rtmp_packet_destroy(&pkt);
 464     if (ret < 0)
 465         return ret;
 466     // Send Peer Bandwidth
 467     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL,
 468                                      RTMP_PT_CLIENT_BW, 0, 5)) < 0)
 469         return ret;
 470     p = pkt.data;
 471     bytestream_put_be32(&p, rt->server_bw);
 472     bytestream_put_byte(&p, 2); // dynamic
 473     pkt.size = p - pkt.data;
 474     ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
 475                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
 476     ff_rtmp_packet_destroy(&pkt);
 477     if (ret < 0)
 478         return ret;
 479
 480     // Ping request
 481     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL,
 482                                      RTMP_PT_PING, 0, 6)) < 0)
 483         return ret;
 484
 485     p = pkt.data;
 486     bytestream_put_be16(&p, 0); // 0 -> Stream Begin
 487     bytestream_put_be32(&p, 0);
 488     ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
 489                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
 490     ff_rtmp_packet_destroy(&pkt);
 491     if (ret < 0)
 492         return ret;
 493
 494     // Chunk size
 495     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL,
 496                                      RTMP_PT_CHUNK_SIZE, 0, 4)) < 0)
 497         return ret;
 498
 499     p = pkt.data;
 500     bytestream_put_be32(&p, rt->out_chunk_size);
 501     ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
 502                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
 503     ff_rtmp_packet_destroy(&pkt);
 504     if (ret < 0)
 505         return ret;
 506
 507     // Send _result NetConnection.Connect.Success to connect
 508     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL,
 509                                      RTMP_PT_INVOKE, 0,
 510                                      RTMP_PKTDATA_DEFAULT_SIZE)) < 0)
 511         return ret;
 512
 513     p = pkt.data;
 514     ff_amf_write_string(&p, "_result");
 515     ff_amf_write_number(&p, seqnum);
 516
 517     ff_amf_write_object_start(&p);
 518     ff_amf_write_field_name(&p, "fmsVer");
 519     ff_amf_write_string(&p, "FMS/3,0,1,123");
 520     ff_amf_write_field_name(&p, "capabilities");
 521     ff_amf_write_number(&p, 31);
 522     ff_amf_write_object_end(&p);
 523
 524     ff_amf_write_object_start(&p);
 525     ff_amf_write_field_name(&p, "level");
 526     ff_amf_write_string(&p, "status");
 527     ff_amf_write_field_name(&p, "code");
 528     ff_amf_write_string(&p, "NetConnection.Connect.Success");
 529     ff_amf_write_field_name(&p, "description");
 530     ff_amf_write_string(&p, "Connection succeeded.");
 531     ff_amf_write_field_name(&p, "objectEncoding");
 532     ff_amf_write_number(&p, 0);
 533     ff_amf_write_object_end(&p);
 534
 535     pkt.size = p - pkt.data;
 536     ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
 537                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
 538     ff_rtmp_packet_destroy(&pkt);
 539     if (ret < 0)
 540         return ret;
 541
 542     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL,
 543                                      RTMP_PT_INVOKE, 0, 30)) < 0)
 544         return ret;
 545     p = pkt.data;
 546     ff_amf_write_string(&p, "onBWDone");
 547     ff_amf_write_number(&p, 0);
 548     ff_amf_write_null(&p);
 549     ff_amf_write_number(&p, 8192);
 550     pkt.size = p - pkt.data;
 551     ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
 552                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
 553     ff_rtmp_packet_destroy(&pkt);
 554
 555     return ret;
 556 }
 557
 558 /**
 559  * Generate 'releaseStream' call and send it to the server. It should make
 560  * the server release some channel for media streams.
 561  */
 562 static int gen_release_stream(URLContext *s, RTMPContext *rt)
 563 {
 564     RTMPPacket pkt;
 565     uint8_t *p;
 566     int ret;
 567
 568     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
 569                                      0, 29 + strlen(rt->playpath))) < 0)
 570         return ret;
 571
 572     av_log(s, AV_LOG_DEBUG, "Releasing stream...\n");
 573     p = pkt.data;
 574     ff_amf_write_string(&p, "releaseStream");
 575     ff_amf_write_number(&p, ++rt->nb_invokes);
 576     ff_amf_write_null(&p);
 577     ff_amf_write_string(&p, rt->playpath);
 578
 579     return rtmp_send_packet(rt, &pkt, 1);
 580 }
 581
 582 /**
 583  * Generate 'FCPublish' call and send it to the server. It should make
 584  * the server prepare for receiving media streams.
 585  */
 586 static int gen_fcpublish_stream(URLContext *s, RTMPContext *rt)
 587 {
 588     RTMPPacket pkt;
 589     uint8_t *p;
 590     int ret;
 591
 592     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
 593                                      0, 25 + strlen(rt->playpath))) < 0)
 594         return ret;
 595
 596     av_log(s, AV_LOG_DEBUG, "FCPublish stream...\n");
 597     p = pkt.data;
 598     ff_amf_write_string(&p, "FCPublish");
 599     ff_amf_write_number(&p, ++rt->nb_invokes);
 600     ff_amf_write_null(&p);
 601     ff_amf_write_string(&p, rt->playpath);
 602
 603     return rtmp_send_packet(rt, &pkt, 1);
 604 }
 605
 606 /**
 607  * Generate 'FCUnpublish' call and send it to the server. It should make
 608  * the server destroy stream.
 609  */
 610 static int gen_fcunpublish_stream(URLContext *s, RTMPContext *rt)
 611 {
 612     RTMPPacket pkt;
 613     uint8_t *p;
 614     int ret;
 615
 616     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
 617                                      0, 27 + strlen(rt->playpath))) < 0)
 618         return ret;
 619
 620     av_log(s, AV_LOG_DEBUG, "UnPublishing stream...\n");
 621     p = pkt.data;
 622     ff_amf_write_string(&p, "FCUnpublish");
 623     ff_amf_write_number(&p, ++rt->nb_invokes);
 624     ff_amf_write_null(&p);
 625     ff_amf_write_string(&p, rt->playpath);
 626
 627     return rtmp_send_packet(rt, &pkt, 0);
 628 }
 629
 630 /**
 631  * Generate 'createStream' call and send it to the server. It should make
 632  * the server allocate some channel for media streams.
 633  */
 634 static int gen_create_stream(URLContext *s, RTMPContext *rt)
 635 {
 636     RTMPPacket pkt;
 637     uint8_t *p;
 638     int ret;
 639
 640     av_log(s, AV_LOG_DEBUG, "Creating stream...\n");
 641
 642     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
 643                                      0, 25)) < 0)
 644         return ret;
 645
 646     p = pkt.data;
 647     ff_amf_write_string(&p, "createStream");
 648     ff_amf_write_number(&p, ++rt->nb_invokes);
 649     ff_amf_write_null(&p);
 650
 651     return rtmp_send_packet(rt, &pkt, 1);
 652 }
 653
 654
 655 /**
 656  * Generate 'deleteStream' call and send it to the server. It should make
 657  * the server remove some channel for media streams.
 658  */
 659 static int gen_delete_stream(URLContext *s, RTMPContext *rt)
 660 {
 661     RTMPPacket pkt;
 662     uint8_t *p;
 663     int ret;
 664
 665     av_log(s, AV_LOG_DEBUG, "Deleting stream...\n");
 666
 667     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
 668                                      0, 34)) < 0)
 669         return ret;
 670
 671     p = pkt.data;
 672     ff_amf_write_string(&p, "deleteStream");
 673     ff_amf_write_number(&p, ++rt->nb_invokes);
 674     ff_amf_write_null(&p);
 675     ff_amf_write_number(&p, rt->stream_id);
 676
 677     return rtmp_send_packet(rt, &pkt, 0);
 678 }
 679
 680 /**
 681  * Generate 'getStreamLength' call and send it to the server. If the server
 682  * knows the duration of the selected stream, it will reply with the duration
 683  * in seconds.
 684  */
 685 static int gen_get_stream_length(URLContext *s, RTMPContext *rt)
 686 {
 687     RTMPPacket pkt;
 688     uint8_t *p;
 689     int ret;
 690
 691     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SOURCE_CHANNEL, RTMP_PT_INVOKE,
 692                                      0, 31 + strlen(rt->playpath))) < 0)
 693         return ret;
 694
 695     p = pkt.data;
 696     ff_amf_write_string(&p, "getStreamLength");
 697     ff_amf_write_number(&p, ++rt->nb_invokes);
 698     ff_amf_write_null(&p);
 699     ff_amf_write_string(&p, rt->playpath);
 700
 701     return rtmp_send_packet(rt, &pkt, 1);
 702 }
 703
 704 /**
 705  * Generate client buffer time and send it to the server.
 706  */
 707 static int gen_buffer_time(URLContext *s, RTMPContext *rt)
 708 {
 709     RTMPPacket pkt;
 710     uint8_t *p;
 711     int ret;
 712
 713     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_PING,
 714                                      1, 10)) < 0)
 715         return ret;
 716
 717     p = pkt.data;
 718     bytestream_put_be16(&p, 3);
 719     bytestream_put_be32(&p, rt->stream_id);
 720     bytestream_put_be32(&p, rt->client_buffer_time);
 721
 722     return rtmp_send_packet(rt, &pkt, 0);
 723 }
 724
 725 /**
 726  * Generate 'play' call and send it to the server, then ping the server
 727  * to start actual playing.
 728  */
 729 static int gen_play(URLContext *s, RTMPContext *rt)
 730 {
 731     RTMPPacket pkt;
 732     uint8_t *p;
 733     int ret;
 734
 735     av_log(s, AV_LOG_DEBUG, "Sending play command for '%s'\n", rt->playpath);
 736
 737     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SOURCE_CHANNEL, RTMP_PT_INVOKE,
 738                                      0, 29 + strlen(rt->playpath))) < 0)
 739         return ret;
 740
 741     pkt.extra = rt->stream_id;
 742
 743     p = pkt.data;
 744     ff_amf_write_string(&p, "play");
 745     ff_amf_write_number(&p, ++rt->nb_invokes);
 746     ff_amf_write_null(&p);
 747     ff_amf_write_string(&p, rt->playpath);
 748     ff_amf_write_number(&p, rt->live * 1000);
 749
 750     return rtmp_send_packet(rt, &pkt, 1);
 751 }
 752
 753 static int gen_seek(URLContext *s, RTMPContext *rt, int64_t timestamp)
 754 {
 755     RTMPPacket pkt;
 756     uint8_t *p;
 757     int ret;
 758
 759     av_log(s, AV_LOG_DEBUG, "Sending seek command for timestamp %"PRId64"\n",
 760            timestamp);
 761
 762     if ((ret = ff_rtmp_packet_create(&pkt, 3, RTMP_PT_INVOKE, 0, 26)) < 0)
 763         return ret;
 764
 765     pkt.extra = rt->stream_id;
 766
 767     p = pkt.data;
 768     ff_amf_write_string(&p, "seek");
 769     ff_amf_write_number(&p, 0); //no tracking back responses
 770     ff_amf_write_null(&p); //as usual, the first null param
 771     ff_amf_write_number(&p, timestamp); //where we want to jump
 772
 773     return rtmp_send_packet(rt, &pkt, 1);
 774 }
 775
 776 /**
 777  * Generate a pause packet that either pauses or unpauses the current stream.
 778  */
 779 static int gen_pause(URLContext *s, RTMPContext *rt, int pause, uint32_t timestamp)
 780 {
 781     RTMPPacket pkt;
 782     uint8_t *p;
 783     int ret;
 784
 785     av_log(s, AV_LOG_DEBUG, "Sending pause command for timestamp %d\n",
 786            timestamp);
 787
 788     if ((ret = ff_rtmp_packet_create(&pkt, 3, RTMP_PT_INVOKE, 0, 29)) < 0)
 789         return ret;
 790
 791     pkt.extra = rt->stream_id;
 792
 793     p = pkt.data;
 794     ff_amf_write_string(&p, "pause");
 795     ff_amf_write_number(&p, 0); //no tracking back responses
 796     ff_amf_write_null(&p); //as usual, the first null param
 797     ff_amf_write_bool(&p, pause); // pause or unpause
 798     ff_amf_write_number(&p, timestamp); //where we pause the stream
 799
 800     return rtmp_send_packet(rt, &pkt, 1);
 801 }
 802
 803 /**
 804  * Generate 'publish' call and send it to the server.
 805  */
 806 static int gen_publish(URLContext *s, RTMPContext *rt)
 807 {
 808     RTMPPacket pkt;
 809     uint8_t *p;
 810     int ret;
 811
 812     av_log(s, AV_LOG_DEBUG, "Sending publish command for '%s'\n", rt->playpath);
 813
 814     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SOURCE_CHANNEL, RTMP_PT_INVOKE,
 815                                      0, 30 + strlen(rt->playpath))) < 0)
 816         return ret;
 817
 818     pkt.extra = rt->stream_id;
 819
 820     p = pkt.data;
 821     ff_amf_write_string(&p, "publish");
 822     ff_amf_write_number(&p, ++rt->nb_invokes);
 823     ff_amf_write_null(&p);
 824     ff_amf_write_string(&p, rt->playpath);
 825     ff_amf_write_string(&p, "live");
 826
 827     return rtmp_send_packet(rt, &pkt, 1);
 828 }
 829
 830 /**
 831  * Generate ping reply and send it to the server.
 832  */
 833 static int gen_pong(URLContext *s, RTMPContext *rt, RTMPPacket *ppkt)
 834 {
 835     RTMPPacket pkt;
 836     uint8_t *p;
 837     int ret;
 838
 839     if (ppkt->size < 6) {
 840         av_log(s, AV_LOG_ERROR, "Too short ping packet (%d)\n",
 841                ppkt->size);
 842         return AVERROR_INVALIDDATA;
 843     }
 844
 845     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_PING,
 846                                      ppkt->timestamp + 1, 6)) < 0)
 847         return ret;
 848
 849     p = pkt.data;
 850     bytestream_put_be16(&p, 7);
 851     bytestream_put_be32(&p, AV_RB32(ppkt->data+2));
 852
 853     return rtmp_send_packet(rt, &pkt, 0);
 854 }
 855
 856 /**
 857  * Generate SWF verification message and send it to the server.
 858  */
 859 static int gen_swf_verification(URLContext *s, RTMPContext *rt)
 860 {
 861     RTMPPacket pkt;
 862     uint8_t *p;
 863     int ret;
 864
 865     av_log(s, AV_LOG_DEBUG, "Sending SWF verification...\n");
 866     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_PING,
 867                                      0, 44)) < 0)
 868         return ret;
 869
 870     p = pkt.data;
 871     bytestream_put_be16(&p, 27);
 872     memcpy(p, rt->swfverification, 42);
 873
 874     return rtmp_send_packet(rt, &pkt, 0);
 875 }
 876
 877 /**
 878  * Generate server bandwidth message and send it to the server.
 879  */
 880 static int gen_server_bw(URLContext *s, RTMPContext *rt)
 881 {
 882     RTMPPacket pkt;
 883     uint8_t *p;
 884     int ret;
 885
 886     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_SERVER_BW,
 887                                      0, 4)) < 0)
 888         return ret;
 889
 890     p = pkt.data;
 891     bytestream_put_be32(&p, rt->server_bw);
 892
 893     return rtmp_send_packet(rt, &pkt, 0);
 894 }
 895
 896 /**
 897  * Generate check bandwidth message and send it to the server.
 898  */
 899 static int gen_check_bw(URLContext *s, RTMPContext *rt)
 900 {
 901     RTMPPacket pkt;
 902     uint8_t *p;
 903     int ret;
 904
 905     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
 906                                      0, 21)) < 0)
 907         return ret;
 908
 909     p = pkt.data;
 910     ff_amf_write_string(&p, "_checkbw");
 911     ff_amf_write_number(&p, ++rt->nb_invokes);
 912     ff_amf_write_null(&p);
 913
 914     return rtmp_send_packet(rt, &pkt, 1);
 915 }
 916
 917 /**
 918  * Generate report on bytes read so far and send it to the server.
 919  */
 920 static int gen_bytes_read(URLContext *s, RTMPContext *rt, uint32_t ts)
 921 {
 922     RTMPPacket pkt;
 923     uint8_t *p;
 924     int ret;
 925
 926     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_BYTES_READ,
 927                                      ts, 4)) < 0)
 928         return ret;
 929
 930     p = pkt.data;
 931     bytestream_put_be32(&p, rt->bytes_read);
 932
 933     return rtmp_send_packet(rt, &pkt, 0);
 934 }
 935
 936 static int gen_fcsubscribe_stream(URLContext *s, RTMPContext *rt,
 937                                   const char *subscribe)
 938 {
 939     RTMPPacket pkt;
 940     uint8_t *p;
 941     int ret;
 942
 943     if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
 944                                      0, 27 + strlen(subscribe))) < 0)
 945         return ret;
 946
 947     p = pkt.data;
 948     ff_amf_write_string(&p, "FCSubscribe");
 949     ff_amf_write_number(&p, ++rt->nb_invokes);
 950     ff_amf_write_null(&p);
 951     ff_amf_write_string(&p, subscribe);
 952
 953     return rtmp_send_packet(rt, &pkt, 1);
 954 }
 955
 956 int ff_rtmp_calc_digest(const uint8_t *src, int len, int gap,
 957                         const uint8_t *key, int keylen, uint8_t *dst)
 958 {
 959     AVHMAC *hmac;
 960
 961     hmac = av_hmac_alloc(AV_HMAC_SHA256);
 962     if (!hmac)
 963         return AVERROR(ENOMEM);
 964
 965     av_hmac_init(hmac, key, keylen);
 966     if (gap <= 0) {
 967         av_hmac_update(hmac, src, len);
 968     } else { //skip 32 bytes used for storing digest
 969         av_hmac_update(hmac, src, gap);
 970         av_hmac_update(hmac, src + gap + 32, len - gap - 32);
 971     }
 972     av_hmac_final(hmac, dst, 32);
 973
 974     av_hmac_free(hmac);
 975
 976     return 0;
 977 }
 978
 979 int ff_rtmp_calc_digest_pos(const uint8_t *buf, int off, int mod_val,
 980                             int add_val)
 981 {
 982     int i, digest_pos = 0;
 983
 984     for (i = 0; i < 4; i++)
 985         digest_pos += buf[i + off];
 986     digest_pos = digest_pos % mod_val + add_val;
 987
 988     return digest_pos;
 989 }
 990
 991 /**
 992  * Put HMAC-SHA2 digest of packet data (except for the bytes where this digest
 993  * will be stored) into that packet.
 994  *
 995  * @param buf handshake data (1536 bytes)
 996  * @param encrypted use an encrypted connection (RTMPE)
 997  * @return offset to the digest inside input data
 998  */
 999 static int rtmp_handshake_imprint_with_digest(uint8_t *buf, int encrypted)
1000 {
1001     int ret, digest_pos;
1002
1003     if (encrypted)
1004         digest_pos = ff_rtmp_calc_digest_pos(buf, 772, 728, 776);
1005     else
1006         digest_pos = ff_rtmp_calc_digest_pos(buf, 8, 728, 12);
1007
1008     ret = ff_rtmp_calc_digest(buf, RTMP_HANDSHAKE_PACKET_SIZE, digest_pos,
1009                               rtmp_player_key, PLAYER_KEY_OPEN_PART_LEN,
1010                               buf + digest_pos);
1011     if (ret < 0)
1012         return ret;
1013
1014     return digest_pos;
1015 }
1016
1017 /**
1018  * Verify that the received server response has the expected digest value.
1019  *
1020  * @param buf handshake data received from the server (1536 bytes)
1021  * @param off position to search digest offset from
1022  * @return 0 if digest is valid, digest position otherwise
1023  */
1024 static int rtmp_validate_digest(uint8_t *buf, int off)
1025 {
1026     uint8_t digest[32];
1027     int ret, digest_pos;
1028
1029     digest_pos = ff_rtmp_calc_digest_pos(buf, off, 728, off + 4);
1030
1031     ret = ff_rtmp_calc_digest(buf, RTMP_HANDSHAKE_PACKET_SIZE, digest_pos,
1032                               rtmp_server_key, SERVER_KEY_OPEN_PART_LEN,
1033                               digest);
1034     if (ret < 0)
1035         return ret;
1036
1037     if (!memcmp(digest, buf + digest_pos, 32))
1038         return digest_pos;
1039     return 0;
1040 }
1041
1042 static int rtmp_calc_swf_verification(URLContext *s, RTMPContext *rt,
1043                                       uint8_t *buf)
1044 {
1045     uint8_t *p;
1046     int ret;
1047
1048     if (rt->swfhash_len != 32) {
1049         av_log(s, AV_LOG_ERROR,
1050                "Hash of the decompressed SWF file is not 32 bytes long.\n");
1051         return AVERROR(EINVAL);
1052     }
1053
1054     p = &rt->swfverification[0];
1055     bytestream_put_byte(&p, 1);
1056     bytestream_put_byte(&p, 1);
1057     bytestream_put_be32(&p, rt->swfsize);
1058     bytestream_put_be32(&p, rt->swfsize);
1059
1060     if ((ret = ff_rtmp_calc_digest(rt->swfhash, 32, 0, buf, 32, p)) < 0)
1061         return ret;
1062
1063     return 0;
1064 }
1065
1066 #if CONFIG_ZLIB
1067 static int rtmp_uncompress_swfplayer(uint8_t *in_data, int64_t in_size,
1068                                      uint8_t **out_data, int64_t *out_size)
1069 {
1070     z_stream zs = { 0 };
1071     void *ptr;
1072     int size;
1073     int ret = 0;
1074
1075     zs.avail_in = in_size;
1076     zs.next_in  = in_data;
1077     ret = inflateInit(&zs);
1078     if (ret != Z_OK)
1079         return AVERROR_UNKNOWN;
1080
1081     do {
1082         uint8_t tmp_buf[16384];
1083
1084         zs.avail_out = sizeof(tmp_buf);
1085         zs.next_out  = tmp_buf;
1086
1087         ret = inflate(&zs, Z_NO_FLUSH);
1088         if (ret != Z_OK && ret != Z_STREAM_END) {
1089             ret = AVERROR_UNKNOWN;
1090             goto fail;
1091         }
1092
1093         size = sizeof(tmp_buf) - zs.avail_out;
1094         if (!(ptr = av_realloc(*out_data, *out_size + size))) {
1095             ret = AVERROR(ENOMEM);
1096             goto fail;
1097         }
1098         *out_data = ptr;
1099
1100         memcpy(*out_data + *out_size, tmp_buf, size);
1101         *out_size += size;
1102     } while (zs.avail_out == 0);
1103
1104 fail:
1105     inflateEnd(&zs);
1106     return ret;
1107 }
1108 #endif
1109
1110 static int rtmp_calc_swfhash(URLContext *s)
1111 {
1112     RTMPContext *rt = s->priv_data;
1113     uint8_t *in_data = NULL, *out_data = NULL, *swfdata;
1114     int64_t in_size;
1115     URLContext *stream;
1116     char swfhash[32];
1117     int swfsize;
1118     int ret = 0;
1119
1120     /* Get the SWF player file. */
1121     if ((ret = ffurl_open(&stream, rt->swfverify, AVIO_FLAG_READ,
1122                           &s->interrupt_callback, NULL, s->protocols, s)) < 0) {
1123         av_log(s, AV_LOG_ERROR, "Cannot open connection %s.\n", rt->swfverify);
1124         goto fail;
1125     }
1126
1127     if ((in_size = ffurl_seek(stream, 0, AVSEEK_SIZE)) < 0) {
1128         ret = AVERROR(EIO);
1129         goto fail;
1130     }
1131
1132     if (!(in_data = av_malloc(in_size))) {
1133         ret = AVERROR(ENOMEM);
1134         goto fail;
1135     }
1136
1137     if ((ret = ffurl_read_complete(stream, in_data, in_size)) < 0)
1138         goto fail;
1139
1140     if (in_size < 3) {
1141         ret = AVERROR_INVALIDDATA;
1142         goto fail;
1143     }
1144
1145     if (!memcmp(in_data, "CWS", 3)) {
1146 #if CONFIG_ZLIB
1147         int64_t out_size;
1148         /* Decompress the SWF player file using Zlib. */
1149         if (!(out_data = av_malloc(8))) {
1150             ret = AVERROR(ENOMEM);
1151             goto fail;
1152         }
1153         *in_data = 'F'; // magic stuff
1154         memcpy(out_data, in_data, 8);
1155         out_size = 8;
1156
1157         if ((ret = rtmp_uncompress_swfplayer(in_data + 8, in_size - 8,
1158                                              &out_data, &out_size)) < 0)
1159             goto fail;
1160         swfsize = out_size;
1161         swfdata = out_data;
1162 #else
1163         av_log(s, AV_LOG_ERROR,
1164                "Zlib is required for decompressing the SWF player file.\n");
1165         ret = AVERROR(EINVAL);
1166         goto fail;
1167 #endif
1168     } else {
1169         swfsize = in_size;
1170         swfdata = in_data;
1171     }
1172
1173     /* Compute the SHA256 hash of the SWF player file. */
1174     if ((ret = ff_rtmp_calc_digest(swfdata, swfsize, 0,
1175                                    "Genuine Adobe Flash Player 001", 30,
1176                                    swfhash)) < 0)
1177         goto fail;
1178
1179     /* Set SWFVerification parameters. */
1180     av_opt_set_bin(rt, "rtmp_swfhash", swfhash, 32, 0);
1181     rt->swfsize = swfsize;
1182
1183 fail:
1184     av_freep(&in_data);
1185     av_freep(&out_data);
1186     ffurl_close(stream);
1187     return ret;
1188 }
1189
1190 /**
1191  * Perform handshake with the server by means of exchanging pseudorandom data
1192  * signed with HMAC-SHA2 digest.
1193  *
1194  * @return 0 if handshake succeeds, negative value otherwise
1195  */
1196 static int rtmp_handshake(URLContext *s, RTMPContext *rt)
1197 {
1198     AVLFG rnd;
1199     uint8_t tosend    [RTMP_HANDSHAKE_PACKET_SIZE+1] = {
1200         3,                // unencrypted data
1201         0, 0, 0, 0,       // client uptime
1202         RTMP_CLIENT_VER1,
1203         RTMP_CLIENT_VER2,
1204         RTMP_CLIENT_VER3,
1205         RTMP_CLIENT_VER4,
1206     };
1207     uint8_t clientdata[RTMP_HANDSHAKE_PACKET_SIZE];
1208     uint8_t serverdata[RTMP_HANDSHAKE_PACKET_SIZE+1];
1209     int i;
1210     int server_pos, client_pos;
1211     uint8_t digest[32], signature[32];
1212     int ret, type = 0;
1213
1214     av_log(s, AV_LOG_DEBUG, "Handshaking...\n");
1215
1216     av_lfg_init(&rnd, 0xDEADC0DE);
1217     // generate handshake packet - 1536 bytes of pseudorandom data
1218     for (i = 9; i <= RTMP_HANDSHAKE_PACKET_SIZE; i++)
1219         tosend[i] = av_lfg_get(&rnd) >> 24;
1220
1221     if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
1222         /* When the client wants to use RTMPE, we have to change the command
1223          * byte to 0x06 which means to use encrypted data and we have to set
1224          * the flash version to at least 9.0.115.0. */
1225         tosend[0] = 6;
1226         tosend[5] = 128;
1227         tosend[6] = 0;
1228         tosend[7] = 3;
1229         tosend[8] = 2;
1230
1231         /* Initialize the Diffie-Hellmann context and generate the public key
1232          * to send to the server. */
1233         if ((ret = ff_rtmpe_gen_pub_key(rt->stream, tosend + 1)) < 0)
1234             return ret;
1235     }
1236
1237     client_pos = rtmp_handshake_imprint_with_digest(tosend + 1, rt->encrypted);
1238     if (client_pos < 0)
1239         return client_pos;
1240
1241     if ((ret = ffurl_write(rt->stream, tosend,
1242                            RTMP_HANDSHAKE_PACKET_SIZE + 1)) < 0) {
1243         av_log(s, AV_LOG_ERROR, "Cannot write RTMP handshake request\n");
1244         return ret;
1245     }
1246
1247     if ((ret = ffurl_read_complete(rt->stream, serverdata,
1248                                    RTMP_HANDSHAKE_PACKET_SIZE + 1)) < 0) {
1249         av_log(s, AV_LOG_ERROR, "Cannot read RTMP handshake response\n");
1250         return ret;
1251     }
1252
1253     if ((ret = ffurl_read_complete(rt->stream, clientdata,
1254                                    RTMP_HANDSHAKE_PACKET_SIZE)) < 0) {
1255         av_log(s, AV_LOG_ERROR, "Cannot read RTMP handshake response\n");
1256         return ret;
1257     }
1258
1259     av_log(s, AV_LOG_DEBUG, "Type answer %d\n", serverdata[0]);
1260     av_log(s, AV_LOG_DEBUG, "Server version %d.%d.%d.%d\n",
1261            serverdata[5], serverdata[6], serverdata[7], serverdata[8]);
1262
1263     if (rt->is_input && serverdata[5] >= 3) {
1264         server_pos = rtmp_validate_digest(serverdata + 1, 772);
1265         if (server_pos < 0)
1266             return server_pos;
1267
1268         if (!server_pos) {
1269             type = 1;
1270             server_pos = rtmp_validate_digest(serverdata + 1, 8);
1271             if (server_pos < 0)
1272                 return server_pos;
1273
1274             if (!server_pos) {
1275                 av_log(s, AV_LOG_ERROR, "Server response validating failed\n");
1276                 return AVERROR(EIO);
1277             }
1278         }
1279
1280         /* Generate SWFVerification token (SHA256 HMAC hash of decompressed SWF,
1281          * key are the last 32 bytes of the server handshake. */
1282         if (rt->swfsize) {
1283             if ((ret = rtmp_calc_swf_verification(s, rt, serverdata + 1 +
1284                                                   RTMP_HANDSHAKE_PACKET_SIZE - 32)) < 0)
1285                 return ret;
1286         }
1287
1288         ret = ff_rtmp_calc_digest(tosend + 1 + client_pos, 32, 0,
1289                                   rtmp_server_key, sizeof(rtmp_server_key),
1290                                   digest);
1291         if (ret < 0)
1292             return ret;
1293
1294         ret = ff_rtmp_calc_digest(clientdata, RTMP_HANDSHAKE_PACKET_SIZE - 32,
1295                                   0, digest, 32, signature);
1296         if (ret < 0)
1297             return ret;
1298
1299         if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
1300             /* Compute the shared secret key sent by the server and initialize
1301              * the RC4 encryption. */
1302             if ((ret = ff_rtmpe_compute_secret_key(rt->stream, serverdata + 1,
1303                                                    tosend + 1, type)) < 0)
1304                 return ret;
1305
1306             /* Encrypt the signature received by the server. */
1307             ff_rtmpe_encrypt_sig(rt->stream, signature, digest, serverdata[0]);
1308         }
1309
1310         if (memcmp(signature, clientdata + RTMP_HANDSHAKE_PACKET_SIZE - 32, 32)) {
1311             av_log(s, AV_LOG_ERROR, "Signature mismatch\n");
1312             return AVERROR(EIO);
1313         }
1314
1315         for (i = 0; i < RTMP_HANDSHAKE_PACKET_SIZE; i++)
1316             tosend[i] = av_lfg_get(&rnd) >> 24;
1317         ret = ff_rtmp_calc_digest(serverdata + 1 + server_pos, 32, 0,
1318                                   rtmp_player_key, sizeof(rtmp_player_key),
1319                                   digest);
1320         if (ret < 0)
1321             return ret;
1322
1323         ret = ff_rtmp_calc_digest(tosend, RTMP_HANDSHAKE_PACKET_SIZE - 32, 0,
1324                                   digest, 32,
1325                                   tosend + RTMP_HANDSHAKE_PACKET_SIZE - 32);
1326         if (ret < 0)
1327             return ret;
1328
1329         if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
1330             /* Encrypt the signature to be send to the server. */
1331             ff_rtmpe_encrypt_sig(rt->stream, tosend +
1332                                  RTMP_HANDSHAKE_PACKET_SIZE - 32, digest,
1333                                  serverdata[0]);
1334         }
1335
1336         // write reply back to the server
1337         if ((ret = ffurl_write(rt->stream, tosend,
1338                                RTMP_HANDSHAKE_PACKET_SIZE)) < 0)
1339             return ret;
1340
1341         if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
1342             /* Set RC4 keys for encryption and update the keystreams. */
1343             if ((ret = ff_rtmpe_update_keystream(rt->stream)) < 0)
1344                 return ret;
1345         }
1346     } else {
1347         if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
1348             /* Compute the shared secret key sent by the server and initialize
1349              * the RC4 encryption. */
1350             if ((ret = ff_rtmpe_compute_secret_key(rt->stream, serverdata + 1,
1351                             tosend + 1, 1)) < 0)
1352                 return ret;
1353
1354             if (serverdata[0] == 9) {
1355                 /* Encrypt the signature received by the server. */
1356                 ff_rtmpe_encrypt_sig(rt->stream, signature, digest,
1357                                      serverdata[0]);
1358             }
1359         }
1360
1361         if ((ret = ffurl_write(rt->stream, serverdata + 1,
1362                                RTMP_HANDSHAKE_PACKET_SIZE)) < 0)
1363             return ret;
1364
1365         if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
1366             /* Set RC4 keys for encryption and update the keystreams. */
1367             if ((ret = ff_rtmpe_update_keystream(rt->stream)) < 0)
1368                 return ret;
1369         }
1370     }
1371
1372     return 0;
1373 }
1374
1375 static int rtmp_receive_hs_packet(RTMPContext* rt, uint32_t *first_int,
1376                                   uint32_t *second_int, char *arraydata,
1377                                   int size)
1378 {
1379     int inoutsize;
1380
1381     inoutsize = ffurl_read_complete(rt->stream, arraydata,
1382                                     RTMP_HANDSHAKE_PACKET_SIZE);
1383     if (inoutsize <= 0)
1384         return AVERROR(EIO);
1385     if (inoutsize != RTMP_HANDSHAKE_PACKET_SIZE) {
1386         av_log(rt, AV_LOG_ERROR, "Erroneous Message size %d"
1387                " not following standard\n", (int)inoutsize);
1388         return AVERROR(EINVAL);
1389     }
1390
1391     *first_int  = AV_RB32(arraydata);
1392     *second_int = AV_RB32(arraydata + 4);
1393     return 0;
1394 }
1395
1396 static int rtmp_send_hs_packet(RTMPContext* rt, uint32_t first_int,
1397                                uint32_t second_int, char *arraydata, int size)
1398 {
1399     int inoutsize;
1400
1401     AV_WB32(arraydata, first_int);
1402     AV_WB32(arraydata + 4, second_int);
1403     inoutsize = ffurl_write(rt->stream, arraydata,
1404                             RTMP_HANDSHAKE_PACKET_SIZE);
1405     if (inoutsize != RTMP_HANDSHAKE_PACKET_SIZE) {
1406         av_log(rt, AV_LOG_ERROR, "Unable to write answer\n");
1407         return AVERROR(EIO);
1408     }
1409
1410     return 0;
1411 }
1412
1413 /**
1414  * rtmp handshake server side
1415  */
1416 static int rtmp_server_handshake(URLContext *s, RTMPContext *rt)
1417 {
1418     uint8_t buffer[RTMP_HANDSHAKE_PACKET_SIZE];
1419     uint32_t hs_epoch;
1420     uint32_t hs_my_epoch;
1421     uint8_t hs_c1[RTMP_HANDSHAKE_PACKET_SIZE];
1422     uint8_t hs_s1[RTMP_HANDSHAKE_PACKET_SIZE];
1423     uint32_t zeroes;
1424     uint32_t temp       = 0;
1425     int randomidx       = 0;
1426     int inoutsize       = 0;
1427     int ret;
1428
1429     inoutsize = ffurl_read_complete(rt->stream, buffer, 1);       // Receive C0
1430     if (inoutsize <= 0) {
1431         av_log(s, AV_LOG_ERROR, "Unable to read handshake\n");
1432         return AVERROR(EIO);
1433     }
1434     // Check Version
1435     if (buffer[0] != 3) {
1436         av_log(s, AV_LOG_ERROR, "RTMP protocol version mismatch\n");
1437         return AVERROR(EIO);
1438     }
1439     if (ffurl_write(rt->stream, buffer, 1) <= 0) {                 // Send S0
1440         av_log(s, AV_LOG_ERROR,
1441                "Unable to write answer - RTMP S0\n");
1442         return AVERROR(EIO);
1443     }
1444     /* Receive C1 */
1445     ret = rtmp_receive_hs_packet(rt, &hs_epoch, &zeroes, hs_c1,
1446                                  RTMP_HANDSHAKE_PACKET_SIZE);
1447     if (ret) {
1448         av_log(s, AV_LOG_ERROR, "RTMP Handshake C1 Error\n");
1449         return ret;
1450     }
1451     /* Send S1 */
1452     /* By now same epoch will be sent */
1453     hs_my_epoch = hs_epoch;
1454     /* Generate random */
1455     for (randomidx = 8; randomidx < (RTMP_HANDSHAKE_PACKET_SIZE);
1456          randomidx += 4)
1457         AV_WB32(hs_s1 + randomidx, av_get_random_seed());
1458
1459     ret = rtmp_send_hs_packet(rt, hs_my_epoch, 0, hs_s1,
1460                               RTMP_HANDSHAKE_PACKET_SIZE);
1461     if (ret) {
1462         av_log(s, AV_LOG_ERROR, "RTMP Handshake S1 Error\n");
1463         return ret;
1464     }
1465     /* Send S2 */
1466     ret = rtmp_send_hs_packet(rt, hs_epoch, 0, hs_c1,
1467                               RTMP_HANDSHAKE_PACKET_SIZE);
1468     if (ret) {
1469         av_log(s, AV_LOG_ERROR, "RTMP Handshake S2 Error\n");
1470         return ret;
1471     }
1472     /* Receive C2 */
1473     ret = rtmp_receive_hs_packet(rt, &temp, &zeroes, buffer,
1474                                  RTMP_HANDSHAKE_PACKET_SIZE);
1475     if (ret) {
1476         av_log(s, AV_LOG_ERROR, "RTMP Handshake C2 Error\n");
1477         return ret;
1478     }
1479     if (temp != hs_my_epoch)
1480         av_log(s, AV_LOG_WARNING,
1481                "Erroneous C2 Message epoch does not match up with C1 epoch\n");
1482     if (memcmp(buffer + 8, hs_s1 + 8,
1483                RTMP_HANDSHAKE_PACKET_SIZE - 8))
1484         av_log(s, AV_LOG_WARNING,
1485                "Erroneous C2 Message random does not match up\n");
1486
1487     return 0;
1488 }
1489
1490 static int handle_chunk_size(URLContext *s, RTMPPacket *pkt)
1491 {
1492     RTMPContext *rt = s->priv_data;
1493     int ret;
1494
1495     if (pkt->size < 4) {
1496         av_log(s, AV_LOG_ERROR,
1497                "Too short chunk size change packet (%d)\n",
1498                pkt->size);
1499         return AVERROR_INVALIDDATA;
1500     }
1501
1502     if (!rt->is_input) {
1503         /* Send the same chunk size change packet back to the server,
1504          * setting the outgoing chunk size to the same as the incoming one. */
1505         if ((ret = ff_rtmp_packet_write(rt->stream, pkt, rt->out_chunk_size,
1506                                         &rt->prev_pkt[1], &rt->nb_prev_pkt[1])) < 0)
1507             return ret;
1508         rt->out_chunk_size = AV_RB32(pkt->data);
1509     }
1510
1511     rt->in_chunk_size = AV_RB32(pkt->data);
1512     if (rt->in_chunk_size <= 0) {
1513         av_log(s, AV_LOG_ERROR, "Incorrect chunk size %d\n",
1514                rt->in_chunk_size);
1515         return AVERROR_INVALIDDATA;
1516     }
1517     av_log(s, AV_LOG_DEBUG, "New incoming chunk size = %d\n",
1518            rt->in_chunk_size);
1519
1520     return 0;
1521 }
1522
1523 static int handle_ping(URLContext *s, RTMPPacket *pkt)
1524 {
1525     RTMPContext *rt = s->priv_data;
1526     int t, ret;
1527
1528     if (pkt->size < 2) {
1529         av_log(s, AV_LOG_ERROR, "Too short ping packet (%d)\n",
1530                pkt->size);
1531         return AVERROR_INVALIDDATA;
1532     }
1533
1534     t = AV_RB16(pkt->data);
1535     if (t == 6) {
1536         if ((ret = gen_pong(s, rt, pkt)) < 0)
1537             return ret;
1538     } else if (t == 26) {
1539         if (rt->swfsize) {
1540             if ((ret = gen_swf_verification(s, rt)) < 0)
1541                 return ret;
1542         } else {
1543             av_log(s, AV_LOG_WARNING, "Ignoring SWFVerification request.\n");
1544         }
1545     }
1546
1547     return 0;
1548 }
1549
1550 static int handle_client_bw(URLContext *s, RTMPPacket *pkt)
1551 {
1552     RTMPContext *rt = s->priv_data;
1553
1554     if (pkt->size < 4) {
1555         av_log(s, AV_LOG_ERROR,
1556                "Client bandwidth report packet is less than 4 bytes long (%d)\n",
1557                pkt->size);
1558         return AVERROR_INVALIDDATA;
1559     }
1560
1561     rt->client_report_size = AV_RB32(pkt->data);
1562     if (rt->client_report_size <= 0) {
1563         av_log(s, AV_LOG_ERROR, "Incorrect client bandwidth %d\n",
1564                 rt->client_report_size);
1565         return AVERROR_INVALIDDATA;
1566
1567     }
1568     av_log(s, AV_LOG_DEBUG, "Client bandwidth = %d\n", rt->client_report_size);
1569     rt->client_report_size >>= 1;
1570
1571     return 0;
1572 }
1573
1574 static int handle_server_bw(URLContext *s, RTMPPacket *pkt)
1575 {
1576     RTMPContext *rt = s->priv_data;
1577
1578     if (pkt->size < 4) {
1579         av_log(s, AV_LOG_ERROR,
1580                "Too short server bandwidth report packet (%d)\n",
1581                pkt->size);
1582         return AVERROR_INVALIDDATA;
1583     }
1584
1585     rt->server_bw = AV_RB32(pkt->data);
1586     if (rt->server_bw <= 0) {
1587         av_log(s, AV_LOG_ERROR, "Incorrect server bandwidth %d\n",
1588                rt->server_bw);
1589         return AVERROR_INVALIDDATA;
1590     }
1591     av_log(s, AV_LOG_DEBUG, "Server bandwidth = %d\n", rt->server_bw);
1592
1593     return 0;
1594 }
1595
1596 static int do_adobe_auth(RTMPContext *rt, const char *user, const char *salt,
1597                          const char *opaque, const char *challenge)
1598 {
1599     uint8_t hash[16];
1600     char hashstr[AV_BASE64_SIZE(sizeof(hash))], challenge2[10];
1601     struct AVMD5 *md5 = av_md5_alloc();
1602     if (!md5)
1603         return AVERROR(ENOMEM);
1604
1605     snprintf(challenge2, sizeof(challenge2), "%08x", av_get_random_seed());
1606
1607     av_md5_init(md5);
1608     av_md5_update(md5, user, strlen(user));
1609     av_md5_update(md5, salt, strlen(salt));
1610     av_md5_update(md5, rt->password, strlen(rt->password));
1611     av_md5_final(md5, hash);
1612     av_base64_encode(hashstr, sizeof(hashstr), hash,
1613                      sizeof(hash));
1614     av_md5_init(md5);
1615     av_md5_update(md5, hashstr, strlen(hashstr));
1616     if (opaque)
1617         av_md5_update(md5, opaque, strlen(opaque));
1618     else if (challenge)
1619         av_md5_update(md5, challenge, strlen(challenge));
1620     av_md5_update(md5, challenge2, strlen(challenge2));
1621     av_md5_final(md5, hash);
1622     av_base64_encode(hashstr, sizeof(hashstr), hash,
1623                      sizeof(hash));
1624     snprintf(rt->auth_params, sizeof(rt->auth_params),
1625              "?authmod=%s&user=%s&challenge=%s&response=%s",
1626              "adobe", user, challenge2, hashstr);
1627     if (opaque)
1628         av_strlcatf(rt->auth_params, sizeof(rt->auth_params),
1629                     "&opaque=%s", opaque);
1630
1631     av_free(md5);
1632     return 0;
1633 }
1634
1635 static int do_llnw_auth(RTMPContext *rt, const char *user, const char *nonce)
1636 {
1637     uint8_t hash[16];
1638     char hashstr1[33], hashstr2[33];
1639     const char *realm = "live";
1640     const char *method = "publish";
1641     const char *qop = "auth";
1642     const char *nc = "00000001";
1643     char cnonce[10];
1644     struct AVMD5 *md5 = av_md5_alloc();
1645     if (!md5)
1646         return AVERROR(ENOMEM);
1647
1648     snprintf(cnonce, sizeof(cnonce), "%08x", av_get_random_seed());
1649
1650     av_md5_init(md5);
1651     av_md5_update(md5, user, strlen(user));
1652     av_md5_update(md5, ":", 1);
1653     av_md5_update(md5, realm, strlen(realm));
1654     av_md5_update(md5, ":", 1);
1655     av_md5_update(md5, rt->password, strlen(rt->password));
1656     av_md5_final(md5, hash);
1657     ff_data_to_hex(hashstr1, hash, 16, 1);
1658     hashstr1[32] = '\0';
1659
1660     av_md5_init(md5);
1661     av_md5_update(md5, method, strlen(method));
1662     av_md5_update(md5, ":/", 2);
1663     av_md5_update(md5, rt->app, strlen(rt->app));
1664     if (!strchr(rt->app, '/'))
1665         av_md5_update(md5, "/_definst_", strlen("/_definst_"));
1666     av_md5_final(md5, hash);
1667     ff_data_to_hex(hashstr2, hash, 16, 1);
1668     hashstr2[32] = '\0';
1669
1670     av_md5_init(md5);
1671     av_md5_update(md5, hashstr1, strlen(hashstr1));
1672     av_md5_update(md5, ":", 1);
1673     if (nonce)
1674         av_md5_update(md5, nonce, strlen(nonce));
1675     av_md5_update(md5, ":", 1);
1676     av_md5_update(md5, nc, strlen(nc));
1677     av_md5_update(md5, ":", 1);
1678     av_md5_update(md5, cnonce, strlen(cnonce));
1679     av_md5_update(md5, ":", 1);
1680     av_md5_update(md5, qop, strlen(qop));
1681     av_md5_update(md5, ":", 1);
1682     av_md5_update(md5, hashstr2, strlen(hashstr2));
1683     av_md5_final(md5, hash);
1684     ff_data_to_hex(hashstr1, hash, 16, 1);
1685
1686     snprintf(rt->auth_params, sizeof(rt->auth_params),
1687              "?authmod=%s&user=%s&nonce=%s&cnonce=%s&nc=%s&response=%s",
1688              "llnw", user, nonce, cnonce, nc, hashstr1);
1689
1690     av_free(md5);
1691     return 0;
1692 }
1693
1694 static int handle_connect_error(URLContext *s, const char *desc)
1695 {
1696     RTMPContext *rt = s->priv_data;
1697     char buf[300], *ptr, authmod[15];
1698     int i = 0, ret = 0;
1699     const char *user = "", *salt = "", *opaque = NULL,
1700                *challenge = NULL, *cptr = NULL, *nonce = NULL;
1701
1702     if (!(cptr = strstr(desc, "authmod=adobe")) &&
1703         !(cptr = strstr(desc, "authmod=llnw"))) {
1704         av_log(s, AV_LOG_ERROR,
1705                "Unknown connect error (unsupported authentication method?)\n");
1706         return AVERROR_UNKNOWN;
1707     }
1708     cptr += strlen("authmod=");
1709     while (*cptr && *cptr != ' ' && i < sizeof(authmod) - 1)
1710         authmod[i++] = *cptr++;
1711     authmod[i] = '\0';
1712
1713     if (!rt->username[0] || !rt->password[0]) {
1714         av_log(s, AV_LOG_ERROR, "No credentials set\n");
1715         return AVERROR_UNKNOWN;
1716     }
1717
1718     if (strstr(desc, "?reason=authfailed")) {
1719         av_log(s, AV_LOG_ERROR, "Incorrect username/password\n");
1720         return AVERROR_UNKNOWN;
1721     } else if (strstr(desc, "?reason=nosuchuser")) {
1722         av_log(s, AV_LOG_ERROR, "Incorrect username\n");
1723         return AVERROR_UNKNOWN;
1724     }
1725
1726     if (rt->auth_tried) {
1727         av_log(s, AV_LOG_ERROR, "Authentication failed\n");
1728         return AVERROR_UNKNOWN;
1729     }
1730
1731     rt->auth_params[0] = '\0';
1732
1733     if (strstr(desc, "code=403 need auth")) {
1734         snprintf(rt->auth_params, sizeof(rt->auth_params),
1735                  "?authmod=%s&user=%s", authmod, rt->username);
1736         return 0;
1737     }
1738
1739     if (!(cptr = strstr(desc, "?reason=needauth"))) {
1740         av_log(s, AV_LOG_ERROR, "No auth parameters found\n");
1741         return AVERROR_UNKNOWN;
1742     }
1743
1744     av_strlcpy(buf, cptr + 1, sizeof(buf));
1745     ptr = buf;
1746
1747     while (ptr) {
1748         char *next  = strchr(ptr, '&');
1749         char *value = strchr(ptr, '=');
1750         if (next)
1751             *next++ = '\0';
1752         if (value)
1753             *value++ = '\0';
1754         if (!strcmp(ptr, "user")) {
1755             user = value;
1756         } else if (!strcmp(ptr, "salt")) {
1757             salt = value;
1758         } else if (!strcmp(ptr, "opaque")) {
1759             opaque = value;
1760         } else if (!strcmp(ptr, "challenge")) {
1761             challenge = value;
1762         } else if (!strcmp(ptr, "nonce")) {
1763             nonce = value;
1764         }
1765         ptr = next;
1766     }
1767
1768     if (!strcmp(authmod, "adobe")) {
1769         if ((ret = do_adobe_auth(rt, user, salt, opaque, challenge)) < 0)
1770             return ret;
1771     } else {
1772         if ((ret = do_llnw_auth(rt, user, nonce)) < 0)
1773             return ret;
1774     }
1775
1776     rt->auth_tried = 1;
1777     return 0;
1778 }
1779
1780 static int handle_invoke_error(URLContext *s, RTMPPacket *pkt)
1781 {
1782     RTMPContext *rt = s->priv_data;
1783     const uint8_t *data_end = pkt->data + pkt->size;
1784     char *tracked_method = NULL;
1785     int level = AV_LOG_ERROR;
1786     uint8_t tmpstr[256];
1787     int ret;
1788
1789     if ((ret = find_tracked_method(s, pkt, 9, &tracked_method)) < 0)
1790         return ret;
1791
1792     if (!ff_amf_get_field_value(pkt->data + 9, data_end,
1793                                 "description", tmpstr, sizeof(tmpstr))) {
1794         if (tracked_method && (!strcmp(tracked_method, "_checkbw")      ||
1795                                !strcmp(tracked_method, "releaseStream") ||
1796                                !strcmp(tracked_method, "FCSubscribe")   ||
1797                                !strcmp(tracked_method, "FCPublish"))) {
1798             /* Gracefully ignore Adobe-specific historical artifact errors. */
1799             level = AV_LOG_WARNING;
1800             ret = 0;
1801         } else if (tracked_method && !strcmp(tracked_method, "getStreamLength")) {
1802             level = rt->live ? AV_LOG_DEBUG : AV_LOG_WARNING;
1803             ret = 0;
1804         } else if (tracked_method && !strcmp(tracked_method, "connect")) {
1805             ret = handle_connect_error(s, tmpstr);
1806             if (!ret) {
1807                 rt->do_reconnect = 1;
1808                 level = AV_LOG_VERBOSE;
1809             }
1810         } else
1811             ret = AVERROR_UNKNOWN;
1812         av_log(s, level, "Server error: %s\n", tmpstr);
1813     }
1814
1815     av_free(tracked_method);
1816     return ret;
1817 }
1818
1819 static int write_begin(URLContext *s)
1820 {
1821     RTMPContext *rt = s->priv_data;
1822     PutByteContext pbc;
1823     RTMPPacket spkt = { 0 };
1824     int ret;
1825
1826     // Send Stream Begin 1
1827     if ((ret = ff_rtmp_packet_create(&spkt, RTMP_NETWORK_CHANNEL,
1828                                      RTMP_PT_PING, 0, 6)) < 0) {
1829         av_log(s, AV_LOG_ERROR, "Unable to create response packet\n");
1830         return ret;
1831     }
1832
1833     bytestream2_init_writer(&pbc, spkt.data, spkt.size);
1834     bytestream2_put_be16(&pbc, 0);          // 0 -> Stream Begin
1835     bytestream2_put_be32(&pbc, rt->nb_streamid);
1836
1837     ret = ff_rtmp_packet_write(rt->stream, &spkt, rt->out_chunk_size,
1838                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
1839
1840     ff_rtmp_packet_destroy(&spkt);
1841
1842     return ret;
1843 }
1844
1845 static int write_status(URLContext *s, RTMPPacket *pkt,
1846                         const char *status, const char *filename)
1847 {
1848     RTMPContext *rt = s->priv_data;
1849     RTMPPacket spkt = { 0 };
1850     char statusmsg[128];
1851     uint8_t *pp;
1852     int ret;
1853
1854     if ((ret = ff_rtmp_packet_create(&spkt, RTMP_SYSTEM_CHANNEL,
1855                                      RTMP_PT_INVOKE, 0,
1856                                      RTMP_PKTDATA_DEFAULT_SIZE)) < 0) {
1857         av_log(s, AV_LOG_ERROR, "Unable to create response packet\n");
1858         return ret;
1859     }
1860
1861     pp = spkt.data;
1862     spkt.extra = pkt->extra;
1863     ff_amf_write_string(&pp, "onStatus");
1864     ff_amf_write_number(&pp, 0);
1865     ff_amf_write_null(&pp);
1866
1867     ff_amf_write_object_start(&pp);
1868     ff_amf_write_field_name(&pp, "level");
1869     ff_amf_write_string(&pp, "status");
1870     ff_amf_write_field_name(&pp, "code");
1871     ff_amf_write_string(&pp, status);
1872     ff_amf_write_field_name(&pp, "description");
1873     snprintf(statusmsg, sizeof(statusmsg),
1874              "%s is now published", filename);
1875     ff_amf_write_string(&pp, statusmsg);
1876     ff_amf_write_field_name(&pp, "details");
1877     ff_amf_write_string(&pp, filename);
1878     ff_amf_write_object_end(&pp);
1879
1880     spkt.size = pp - spkt.data;
1881     ret = ff_rtmp_packet_write(rt->stream, &spkt, rt->out_chunk_size,
1882                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
1883     ff_rtmp_packet_destroy(&spkt);
1884
1885     return ret;
1886 }
1887
1888 static int send_invoke_response(URLContext *s, RTMPPacket *pkt)
1889 {
1890     RTMPContext *rt = s->priv_data;
1891     double seqnum;
1892     char filename[128];
1893     char command[64];
1894     int stringlen;
1895     char *pchar;
1896     const uint8_t *p = pkt->data;
1897     uint8_t *pp      = NULL;
1898     RTMPPacket spkt  = { 0 };
1899     GetByteContext gbc;
1900     int ret;
1901
1902     bytestream2_init(&gbc, p, pkt->size);
1903     if (ff_amf_read_string(&gbc, command, sizeof(command),
1904                            &stringlen)) {
1905         av_log(s, AV_LOG_ERROR, "Error in PT_INVOKE\n");
1906         return AVERROR_INVALIDDATA;
1907     }
1908
1909     ret = ff_amf_read_number(&gbc, &seqnum);
1910     if (ret)
1911         return ret;
1912     ret = ff_amf_read_null(&gbc);
1913     if (ret)
1914         return ret;
1915     if (!strcmp(command, "FCPublish") ||
1916         !strcmp(command, "publish")) {
1917         ret = ff_amf_read_string(&gbc, filename,
1918                                  sizeof(filename), &stringlen);
1919         if (ret) {
1920             if (ret == AVERROR(EINVAL))
1921                 av_log(s, AV_LOG_ERROR, "Unable to parse stream name - name too long?\n");
1922             else
1923                 av_log(s, AV_LOG_ERROR, "Unable to parse stream name\n");
1924             return ret;
1925         }
1926         // check with url
1927         if (s->filename) {
1928             pchar = strrchr(s->filename, '/');
1929             if (!pchar) {
1930                 av_log(s, AV_LOG_WARNING,
1931                        "Unable to find / in url %s, bad format\n",
1932                        s->filename);
1933                 pchar = s->filename;
1934             }
1935             pchar++;
1936             if (strcmp(pchar, filename))
1937                 av_log(s, AV_LOG_WARNING, "Unexpected stream %s, expecting"
1938                        " %s\n", filename, pchar);
1939         }
1940         rt->state = STATE_RECEIVING;
1941     }
1942
1943     if (!strcmp(command, "FCPublish")) {
1944         if ((ret = ff_rtmp_packet_create(&spkt, RTMP_SYSTEM_CHANNEL,
1945                                          RTMP_PT_INVOKE, 0,
1946                                          RTMP_PKTDATA_DEFAULT_SIZE)) < 0) {
1947             av_log(s, AV_LOG_ERROR, "Unable to create response packet\n");
1948             return ret;
1949         }
1950         pp = spkt.data;
1951         ff_amf_write_string(&pp, "onFCPublish");
1952     } else if (!strcmp(command, "publish")) {
1953         ret = write_begin(s);
1954         if (ret < 0)
1955             return ret;
1956
1957         // Send onStatus(NetStream.Publish.Start)
1958         return write_status(s, pkt, "NetStream.Publish.Start",
1959                            filename);
1960     } else if (!strcmp(command, "play")) {
1961         ret = write_begin(s);
1962         if (ret < 0)
1963             return ret;
1964         rt->state = STATE_SENDING;
1965         return write_status(s, pkt, "NetStream.Play.Start",
1966                             filename);
1967     } else {
1968         if ((ret = ff_rtmp_packet_create(&spkt, RTMP_SYSTEM_CHANNEL,
1969                                          RTMP_PT_INVOKE, 0,
1970                                          RTMP_PKTDATA_DEFAULT_SIZE)) < 0) {
1971             av_log(s, AV_LOG_ERROR, "Unable to create response packet\n");
1972             return ret;
1973         }
1974         pp = spkt.data;
1975         ff_amf_write_string(&pp, "_result");
1976         ff_amf_write_number(&pp, seqnum);
1977         ff_amf_write_null(&pp);
1978         if (!strcmp(command, "createStream")) {
1979             rt->nb_streamid++;
1980             if (rt->nb_streamid == 0 || rt->nb_streamid == 2)
1981                 rt->nb_streamid++; /* Values 0 and 2 are reserved */
1982             ff_amf_write_number(&pp, rt->nb_streamid);
1983             /* By now we don't control which streams are removed in
1984              * deleteStream. There is no stream creation control
1985              * if a client creates more than 2^32 - 2 streams. */
1986         }
1987     }
1988     spkt.size = pp - spkt.data;
1989     ret = ff_rtmp_packet_write(rt->stream, &spkt, rt->out_chunk_size,
1990                                &rt->prev_pkt[1], &rt->nb_prev_pkt[1]);
1991     ff_rtmp_packet_destroy(&spkt);
1992     return ret;
1993 }
1994
1995 /**
1996  * Read the AMF_NUMBER response ("_result") to a function call
1997  * (e.g. createStream()). This response should be made up of the AMF_STRING
1998  * "result", a NULL object and then the response encoded as AMF_NUMBER. On a
1999  * successful response, we will return set the value to number (otherwise number
2000  * will not be changed).
2001  *
2002  * @return 0 if reading the value succeeds, negative value otherwise
2003  */
2004 static int read_number_result(RTMPPacket *pkt, double *number)
2005 {
2006     // We only need to fit "_result" in this.
2007     uint8_t strbuffer[8];
2008     int stringlen;
2009     double numbuffer;
2010     GetByteContext gbc;
2011
2012     bytestream2_init(&gbc, pkt->data, pkt->size);
2013
2014     // Value 1/4: "_result" as AMF_STRING
2015     if (ff_amf_read_string(&gbc, strbuffer, sizeof(strbuffer), &stringlen))
2016         return AVERROR_INVALIDDATA;
2017     if (strcmp(strbuffer, "_result"))
2018         return AVERROR_INVALIDDATA;
2019     // Value 2/4: The callee reference number
2020     if (ff_amf_read_number(&gbc, &numbuffer))
2021         return AVERROR_INVALIDDATA;
2022     // Value 3/4: Null
2023     if (ff_amf_read_null(&gbc))
2024         return AVERROR_INVALIDDATA;
2025     // Value 4/4: The response as AMF_NUMBER
2026     if (ff_amf_read_number(&gbc, &numbuffer))
2027         return AVERROR_INVALIDDATA;
2028     else
2029         *number = numbuffer;
2030
2031     return 0;
2032 }
2033
2034 static int handle_invoke_result(URLContext *s, RTMPPacket *pkt)
2035 {
2036     RTMPContext *rt = s->priv_data;
2037     char *tracked_method = NULL;
2038     int ret = 0;
2039
2040     if ((ret = find_tracked_method(s, pkt, 10, &tracked_method)) < 0)
2041         return ret;
2042
2043     if (!tracked_method) {
2044         /* Ignore this reply when the current method is not tracked. */
2045         return ret;
2046     }
2047
2048     if (!strcmp(tracked_method, "connect")) {
2049         if (!rt->is_input) {
2050             if ((ret = gen_release_stream(s, rt)) < 0)
2051                 goto fail;
2052
2053             if ((ret = gen_fcpublish_stream(s, rt)) < 0)
2054                 goto fail;
2055         } else {
2056             if ((ret = gen_server_bw(s, rt)) < 0)
2057                 goto fail;
2058         }
2059
2060         if ((ret = gen_create_stream(s, rt)) < 0)
2061             goto fail;
2062
2063         if (rt->is_input) {
2064             /* Send the FCSubscribe command when the name of live
2065              * stream is defined by the user or if it's a live stream. */
2066             if (rt->subscribe) {
2067                 if ((ret = gen_fcsubscribe_stream(s, rt, rt->subscribe)) < 0)
2068                     goto fail;
2069             } else if (rt->live == -1) {
2070                 if ((ret = gen_fcsubscribe_stream(s, rt, rt->playpath)) < 0)
2071                     goto fail;
2072             }
2073         }
2074     } else if (!strcmp(tracked_method, "createStream")) {
2075         double stream_id;
2076         if (read_number_result(pkt, &stream_id)) {
2077             av_log(s, AV_LOG_WARNING, "Unexpected reply on connect()\n");
2078         } else {
2079             rt->stream_id = stream_id;
2080         }
2081
2082         if (!rt->is_input) {
2083             if ((ret = gen_publish(s, rt)) < 0)
2084                 goto fail;
2085         } else {
2086             if (rt->live != -1) {
2087                 if ((ret = gen_get_stream_length(s, rt)) < 0)
2088                     goto fail;
2089             }
2090             if ((ret = gen_play(s, rt)) < 0)
2091                 goto fail;
2092             if ((ret = gen_buffer_time(s, rt)) < 0)
2093                 goto fail;
2094         }
2095     } else if (!strcmp(tracked_method, "getStreamLength")) {
2096         if (read_number_result(pkt, &rt->duration)) {
2097             av_log(s, AV_LOG_WARNING, "Unexpected reply on getStreamLength()\n");
2098         }
2099     }
2100
2101 fail:
2102     av_free(tracked_method);
2103     return ret;
2104 }
2105
2106 static int handle_invoke_status(URLContext *s, RTMPPacket *pkt)
2107 {
2108     RTMPContext *rt = s->priv_data;
2109     const uint8_t *data_end = pkt->data + pkt->size;
2110     const uint8_t *ptr = pkt->data + RTMP_HEADER;
2111     uint8_t tmpstr[256];
2112     int i, t;
2113
2114     for (i = 0; i < 2; i++) {
2115         t = ff_amf_tag_size(ptr, data_end);
2116         if (t < 0)
2117             return 1;
2118         ptr += t;
2119     }
2120
2121     t = ff_amf_get_field_value(ptr, data_end, "level", tmpstr, sizeof(tmpstr));
2122     if (!t && !strcmp(tmpstr, "error")) {
2123         t = ff_amf_get_field_value(ptr, data_end,
2124                                    "description", tmpstr, sizeof(tmpstr));
2125         if (t || !tmpstr[0])
2126             t = ff_amf_get_field_value(ptr, data_end, "code",
2127                                        tmpstr, sizeof(tmpstr));
2128         if (!t)
2129             av_log(s, AV_LOG_ERROR, "Server error: %s\n", tmpstr);
2130         return -1;
2131     }
2132
2133     t = ff_amf_get_field_value(ptr, data_end, "code", tmpstr, sizeof(tmpstr));
2134     if (!t && !strcmp(tmpstr, "NetStream.Play.Start")) rt->state = STATE_PLAYING;
2135     if (!t && !strcmp(tmpstr, "NetStream.Play.Stop")) rt->state = STATE_STOPPED;
2136     if (!t && !strcmp(tmpstr, "NetStream.Play.UnpublishNotify")) rt->state = STATE_STOPPED;
2137     if (!t && !strcmp(tmpstr, "NetStream.Publish.Start")) rt->state = STATE_PUBLISHING;
2138     if (!t && !strcmp(tmpstr, "NetStream.Seek.Notify")) rt->state = STATE_PLAYING;
2139
2140     return 0;
2141 }
2142
2143 static int handle_invoke(URLContext *s, RTMPPacket *pkt)
2144 {
2145     RTMPContext *rt = s->priv_data;
2146     int ret = 0;
2147
2148     //TODO: check for the messages sent for wrong state?
2149     if (ff_amf_match_string(pkt->data, pkt->size, "_error")) {
2150         if ((ret = handle_invoke_error(s, pkt)) < 0)
2151             return ret;
2152     } else if (ff_amf_match_string(pkt->data, pkt->size, "_result")) {
2153         if ((ret = handle_invoke_result(s, pkt)) < 0)
2154             return ret;
2155     } else if (ff_amf_match_string(pkt->data, pkt->size, "onStatus")) {
2156         if ((ret = handle_invoke_status(s, pkt)) < 0)
2157             return ret;
2158     } else if (ff_amf_match_string(pkt->data, pkt->size, "onBWDone")) {
2159         if ((ret = gen_check_bw(s, rt)) < 0)
2160             return ret;
2161     } else if (ff_amf_match_string(pkt->data, pkt->size, "releaseStream") ||
2162                ff_amf_match_string(pkt->data, pkt->size, "FCPublish")     ||
2163                ff_amf_match_string(pkt->data, pkt->size, "publish")       ||
2164                ff_amf_match_string(pkt->data, pkt->size, "play")          ||
2165                ff_amf_match_string(pkt->data, pkt->size, "_checkbw")      ||
2166                ff_amf_match_string(pkt->data, pkt->size, "createStream")) {
2167         if ((ret = send_invoke_response(s, pkt)) < 0)
2168             return ret;
2169     }
2170
2171     return ret;
2172 }
2173
2174 static int update_offset(RTMPContext *rt, int size)
2175 {
2176     int old_flv_size;
2177
2178     // generate packet header and put data into buffer for FLV demuxer
2179     if (rt->flv_off < rt->flv_size) {
2180         // There is old unread data in the buffer, thus append at the end
2181         old_flv_size  = rt->flv_size;
2182         rt->flv_size += size;
2183     } else {
2184         // All data has been read, write the new data at the start of the buffer
2185         old_flv_size = 0;
2186         rt->flv_size = size;
2187         rt->flv_off  = 0;
2188     }
2189
2190     return old_flv_size;
2191 }
2192
2193 static int append_flv_data(RTMPContext *rt, RTMPPacket *pkt, int skip)
2194 {
2195     int old_flv_size, ret;
2196     PutByteContext pbc;
2197     const uint8_t *data = pkt->data + skip;
2198     const int size      = pkt->size - skip;
2199     uint32_t ts         = pkt->timestamp;
2200
2201     if (pkt->type == RTMP_PT_AUDIO) {
2202         rt->has_audio = 1;
2203     } else if (pkt->type == RTMP_PT_VIDEO) {
2204         rt->has_video = 1;
2205     }
2206
2207     old_flv_size = update_offset(rt, size + 15);
2208
2209     if ((ret = av_reallocp(&rt->flv_data, rt->flv_size)) < 0) {
2210         rt->flv_size = rt->flv_off = 0;
2211         return ret;
2212     }
2213     bytestream2_init_writer(&pbc, rt->flv_data, rt->flv_size);
2214     bytestream2_skip_p(&pbc, old_flv_size);
2215     bytestream2_put_byte(&pbc, pkt->type);
2216     bytestream2_put_be24(&pbc, size);
2217     bytestream2_put_be24(&pbc, ts);
2218     bytestream2_put_byte(&pbc, ts >> 24);
2219     bytestream2_put_be24(&pbc, 0);
2220     bytestream2_put_buffer(&pbc, data, size);
2221     bytestream2_put_be32(&pbc, size + RTMP_HEADER);
2222
2223     return 0;
2224 }
2225
2226 static int handle_notify(URLContext *s, RTMPPacket *pkt)
2227 {
2228     RTMPContext *rt  = s->priv_data;
2229     uint8_t commandbuffer[64];
2230     char statusmsg[128];
2231     int stringlen, ret, skip = 0;
2232     GetByteContext gbc;
2233
2234     bytestream2_init(&gbc, pkt->data, pkt->size);
2235     if (ff_amf_read_string(&gbc, commandbuffer, sizeof(commandbuffer),
2236                            &stringlen))
2237         return AVERROR_INVALIDDATA;
2238
2239     if (!strcmp(commandbuffer, "onMetaData")) {
2240         // metadata properties should be stored in a mixed array
2241         if (bytestream2_get_byte(&gbc) == AMF_DATA_TYPE_MIXEDARRAY) {
2242             // We have found a metaData Array so flv can determine the streams
2243             // from this.
2244             rt->received_metadata = 1;
2245             // skip 32-bit max array index
2246             bytestream2_skip(&gbc, 4);
2247             while (bytestream2_get_bytes_left(&gbc) > 3) {
2248                 if (ff_amf_get_string(&gbc, statusmsg, sizeof(statusmsg),
2249                                       &stringlen))
2250                     return AVERROR_INVALIDDATA;
2251                 // We do not care about the content of the property (yet).
2252                 stringlen = ff_amf_tag_size(gbc.buffer, gbc.buffer_end);
2253                 if (stringlen < 0)
2254                     return AVERROR_INVALIDDATA;
2255                 bytestream2_skip(&gbc, stringlen);
2256
2257                 // The presence of the following properties indicates that the
2258                 // respective streams are present.
2259                 if (!strcmp(statusmsg, "videocodecid")) {
2260                     rt->has_video = 1;
2261                 }
2262                 if (!strcmp(statusmsg, "audiocodecid")) {
2263                     rt->has_audio = 1;
2264                 }
2265             }
2266             if (bytestream2_get_be24(&gbc) != AMF_END_OF_OBJECT)
2267                 return AVERROR_INVALIDDATA;
2268         }
2269     }
2270
2271     // Skip the @setDataFrame string and validate it is a notification
2272     if (!strcmp(commandbuffer, "@setDataFrame")) {
2273         skip = gbc.buffer - pkt->data;
2274         ret = ff_amf_read_string(&gbc, statusmsg,
2275                                  sizeof(statusmsg), &stringlen);
2276         if (ret < 0)
2277             return AVERROR_INVALIDDATA;
2278     }
2279
2280     return append_flv_data(rt, pkt, skip);
2281 }
2282
2283 /**
2284  * Parse received packet and possibly perform some action depending on
2285  * the packet contents.
2286  * @return 0 for no errors, negative values for serious errors which prevent
2287  *         further communications, positive values for uncritical errors
2288  */
2289 static int rtmp_parse_result(URLContext *s, RTMPContext *rt, RTMPPacket *pkt)
2290 {
2291     int ret;
2292
2293 #ifdef DEBUG
2294     ff_rtmp_packet_dump(s, pkt);
2295 #endif
2296
2297     switch (pkt->type) {
2298     case RTMP_PT_BYTES_READ:
2299         av_log(s, AV_LOG_TRACE, "received bytes read report\n");
2300         break;
2301     case RTMP_PT_CHUNK_SIZE:
2302         if ((ret = handle_chunk_size(s, pkt)) < 0)
2303             return ret;
2304         break;
2305     case RTMP_PT_PING:
2306         if ((ret = handle_ping(s, pkt)) < 0)
2307             return ret;
2308         break;
2309     case RTMP_PT_CLIENT_BW:
2310         if ((ret = handle_client_bw(s, pkt)) < 0)
2311             return ret;
2312         break;
2313     case RTMP_PT_SERVER_BW:
2314         if ((ret = handle_server_bw(s, pkt)) < 0)
2315             return ret;
2316         break;
2317     case RTMP_PT_INVOKE:
2318         if ((ret = handle_invoke(s, pkt)) < 0)
2319             return ret;
2320         break;
2321     case RTMP_PT_VIDEO:
2322     case RTMP_PT_AUDIO:
2323     case RTMP_PT_METADATA:
2324     case RTMP_PT_NOTIFY:
2325         /* Audio, Video and Metadata packets are parsed in get_packet() */
2326         break;
2327     default:
2328         av_log(s, AV_LOG_VERBOSE, "Unknown packet type received 0x%02X\n", pkt->type);
2329         break;
2330     }
2331     return 0;
2332 }
2333
2334 static int handle_metadata(RTMPContext *rt, RTMPPacket *pkt)
2335 {
2336     int ret, old_flv_size, type;
2337     const uint8_t *next;
2338     uint8_t *p;
2339     uint32_t size;
2340     uint32_t ts, cts, pts = 0;
2341
2342     old_flv_size = update_offset(rt, pkt->size);
2343
2344     if ((ret = av_reallocp(&rt->flv_data, rt->flv_size)) < 0) {
2345         rt->flv_size = rt->flv_off = 0;
2346         return ret;
2347     }
2348
2349     next = pkt->data;
2350     p    = rt->flv_data + old_flv_size;
2351
2352     /* copy data while rewriting timestamps */
2353     ts = pkt->timestamp;
2354
2355     while (next - pkt->data < pkt->size - RTMP_HEADER) {
2356         type = bytestream_get_byte(&next);
2357         size = bytestream_get_be24(&next);
2358         cts  = bytestream_get_be24(&next);
2359         cts |= bytestream_get_byte(&next) << 24;
2360         if (!pts)
2361             pts = cts;
2362         ts += cts - pts;
2363         pts = cts;
2364         if (size + 3 + 4 > pkt->data + pkt->size - next)
2365             break;
2366         bytestream_put_byte(&p, type);
2367         bytestream_put_be24(&p, size);
2368         bytestream_put_be24(&p, ts);
2369         bytestream_put_byte(&p, ts >> 24);
2370         memcpy(p, next, size + 3 + 4);
2371         p    += size + 3;
2372         bytestream_put_be32(&p, size + RTMP_HEADER);
2373         next += size + 3 + 4;
2374     }
2375     if (p != rt->flv_data + rt->flv_size) {
2376         av_log(NULL, AV_LOG_WARNING, "Incomplete flv packets in "
2377                                      "RTMP_PT_METADATA packet\n");
2378         rt->flv_size = p - rt->flv_data;
2379     }
2380
2381     return 0;
2382 }
2383
2384 /**
2385  * Interact with the server by receiving and sending RTMP packets until
2386  * there is some significant data (media data or expected status notification).
2387  *
2388  * @param s          reading context
2389  * @param for_header non-zero value tells function to work until it
2390  * gets notification from the server that playing has been started,
2391  * otherwise function will work until some media data is received (or
2392  * an error happens)
2393  * @return 0 for successful operation, negative value in case of error
2394  */
2395 static int get_packet(URLContext *s, int for_header)
2396 {
2397     RTMPContext *rt = s->priv_data;
2398     int ret;
2399
2400     if (rt->state == STATE_STOPPED)
2401         return AVERROR_EOF;
2402
2403     for (;;) {
2404         RTMPPacket rpkt = { 0 };
2405         if ((ret = ff_rtmp_packet_read(rt->stream, &rpkt,
2406                                        rt->in_chunk_size, &rt->prev_pkt[0],
2407                                        &rt->nb_prev_pkt[0])) <= 0) {
2408             if (ret == 0) {
2409                 return AVERROR(EAGAIN);
2410             } else {
2411                 return AVERROR(EIO);
2412             }
2413         }
2414
2415         // Track timestamp for later use
2416         rt->last_timestamp = rpkt.timestamp;
2417
2418         rt->bytes_read += ret;
2419         if (rt->bytes_read > rt->last_bytes_read + rt->client_report_size) {
2420             av_log(s, AV_LOG_DEBUG, "Sending bytes read report\n");
2421             if ((ret = gen_bytes_read(s, rt, rpkt.timestamp + 1)) < 0)
2422                 return ret;
2423             rt->last_bytes_read = rt->bytes_read;
2424         }
2425
2426         ret = rtmp_parse_result(s, rt, &rpkt);
2427
2428         // At this point we must check if we are in the seek state and continue
2429         // with the next packet. handle_invoke will get us out of this state
2430         // when the right message is encountered
2431         if (rt->state == STATE_SEEKING) {
2432             ff_rtmp_packet_destroy(&rpkt);
2433             // We continue, let the natural flow of things happen:
2434             // AVERROR(EAGAIN) or handle_invoke gets us out of here
2435             continue;
2436         }
2437
2438         if (ret < 0) {//serious error in current packet
2439             ff_rtmp_packet_destroy(&rpkt);
2440             return ret;
2441         }
2442         if (rt->do_reconnect && for_header) {
2443             ff_rtmp_packet_destroy(&rpkt);
2444             return 0;
2445         }
2446         if (rt->state == STATE_STOPPED) {
2447             ff_rtmp_packet_destroy(&rpkt);
2448             return AVERROR_EOF;
2449         }
2450         if (for_header && (rt->state == STATE_PLAYING    ||
2451                            rt->state == STATE_PUBLISHING ||
2452                            rt->state == STATE_SENDING    ||
2453                            rt->state == STATE_RECEIVING)) {
2454             ff_rtmp_packet_destroy(&rpkt);
2455             return 0;
2456         }
2457         if (!rpkt.size || !rt->is_input) {
2458             ff_rtmp_packet_destroy(&rpkt);
2459             continue;
2460         }
2461         if (rpkt.type == RTMP_PT_VIDEO || rpkt.type == RTMP_PT_AUDIO) {
2462             ret = append_flv_data(rt, &rpkt, 0);
2463             ff_rtmp_packet_destroy(&rpkt);
2464             return ret;
2465         } else if (rpkt.type == RTMP_PT_NOTIFY) {
2466             ret = handle_notify(s, &rpkt);
2467             ff_rtmp_packet_destroy(&rpkt);
2468             return ret;
2469         } else if (rpkt.type == RTMP_PT_METADATA) {
2470             ret = handle_metadata(rt, &rpkt);
2471             ff_rtmp_packet_destroy(&rpkt);
2472             return 0;
2473         }
2474         ff_rtmp_packet_destroy(&rpkt);
2475     }
2476 }
2477
2478 static int rtmp_close(URLContext *h)
2479 {
2480     RTMPContext *rt = h->priv_data;
2481     int ret = 0, i, j;
2482
2483     if (!rt->is_input) {
2484         rt->flv_data = NULL;
2485         if (rt->out_pkt.size)
2486             ff_rtmp_packet_destroy(&rt->out_pkt);
2487         if (rt->state > STATE_FCPUBLISH)
2488             ret = gen_fcunpublish_stream(h, rt);
2489     }
2490     if (rt->state > STATE_HANDSHAKED)
2491         ret = gen_delete_stream(h, rt);
2492     for (i = 0; i < 2; i++) {
2493         for (j = 0; j < rt->nb_prev_pkt[i]; j++)
2494             ff_rtmp_packet_destroy(&rt->prev_pkt[i][j]);
2495         av_freep(&rt->prev_pkt[i]);
2496     }
2497
2498     free_tracked_methods(rt);
2499     av_freep(&rt->flv_data);
2500     ffurl_close(rt->stream);
2501     return ret;
2502 }
2503
2504 /**
2505  * Insert a fake onMetadata packet into the FLV stream to notify the FLV
2506  * demuxer about the duration of the stream.
2507  *
2508  * This should only be done if there was no real onMetadata packet sent by the
2509  * server at the start of the stream and if we were able to retrieve a valid
2510  * duration via a getStreamLength call.
2511  *
2512  * @return 0 for successful operation, negative value in case of error
2513  */
2514 static int inject_fake_duration_metadata(RTMPContext *rt)
2515 {
2516     // We need to insert the metadata packet directly after the FLV
2517     // header, i.e. we need to move all other already read data by the
2518     // size of our fake metadata packet.
2519
2520     uint8_t* p;
2521     // Keep old flv_data pointer
2522     uint8_t* old_flv_data = rt->flv_data;
2523     // Allocate a new flv_data pointer with enough space for the additional package
2524     if (!(rt->flv_data = av_malloc(rt->flv_size + 55))) {
2525         rt->flv_data = old_flv_data;
2526         return AVERROR(ENOMEM);
2527     }
2528
2529     // Copy FLV header
2530     memcpy(rt->flv_data, old_flv_data, 13);
2531     // Copy remaining packets
2532     memcpy(rt->flv_data + 13 + 55, old_flv_data + 13, rt->flv_size - 13);
2533     // Increase the size by the injected packet
2534     rt->flv_size += 55;
2535     // Delete the old FLV data
2536     av_free(old_flv_data);
2537
2538     p = rt->flv_data + 13;
2539     bytestream_put_byte(&p, FLV_TAG_TYPE_META);
2540     bytestream_put_be24(&p, 40); // size of data part (sum of all parts below)
2541     bytestream_put_be24(&p, 0);  // timestamp
2542     bytestream_put_be32(&p, 0);  // reserved
2543
2544     // first event name as a string
2545     bytestream_put_byte(&p, AMF_DATA_TYPE_STRING);
2546     // "onMetaData" as AMF string
2547     bytestream_put_be16(&p, 10);
2548     bytestream_put_buffer(&p, "onMetaData", 10);
2549
2550     // mixed array (hash) with size and string/type/data tuples
2551     bytestream_put_byte(&p, AMF_DATA_TYPE_MIXEDARRAY);
2552     bytestream_put_be32(&p, 1); // metadata_count
2553
2554     // "duration" as AMF string
2555     bytestream_put_be16(&p, 8);
2556     bytestream_put_buffer(&p, "duration", 8);
2557     bytestream_put_byte(&p, AMF_DATA_TYPE_NUMBER);
2558     bytestream_put_be64(&p, av_double2int(rt->duration));
2559
2560     // Finalise object
2561     bytestream_put_be16(&p, 0); // Empty string
2562     bytestream_put_byte(&p, AMF_END_OF_OBJECT);
2563     bytestream_put_be32(&p, 40 + RTMP_HEADER); // size of data part (sum of all parts above)
2564
2565     return 0;
2566 }
2567
2568 /**
2569  * Open RTMP connection and verify that the stream can be played.
2570  *
2571  * URL syntax: rtmp://server[:port][/app][/playpath]
2572  *             where 'app' is first one or two directories in the path
2573  *             (e.g. /ondemand/, /flash/live/, etc.)
2574  *             and 'playpath' is a file name (the rest of the path,
2575  *             may be prefixed with "mp4:")
2576  */
2577 static int rtmp_open(URLContext *s, const char *uri, int flags)
2578 {
2579     RTMPContext *rt = s->priv_data;
2580     char proto[8], hostname[256], path[1024], auth[100], *fname;
2581     char *old_app, *qmark, fname_buffer[1024];
2582     uint8_t buf[2048];
2583     int port;
2584     AVDictionary *opts = NULL;
2585     int ret;
2586
2587     if (rt->listen_timeout > 0)
2588         rt->listen = 1;
2589
2590     rt->is_input = !(flags & AVIO_FLAG_WRITE);
2591
2592     av_url_split(proto, sizeof(proto), auth, sizeof(auth),
2593                  hostname, sizeof(hostname), &port,
2594                  path, sizeof(path), s->filename);
2595
2596     if (strchr(path, ' ')) {
2597         av_log(s, AV_LOG_WARNING,
2598                "Detected librtmp style URL parameters, these aren't supported "
2599                "by the libavformat internal RTMP handler currently enabled. "
2600                "See the documentation for the correct way to pass parameters.\n");
2601     }
2602
2603     if (auth[0]) {
2604         char *ptr = strchr(auth, ':');
2605         if (ptr) {
2606             *ptr = '\0';
2607             av_strlcpy(rt->username, auth, sizeof(rt->username));
2608             av_strlcpy(rt->password, ptr + 1, sizeof(rt->password));
2609         }
2610     }
2611
2612     if (rt->listen && strcmp(proto, "rtmp")) {
2613         av_log(s, AV_LOG_ERROR, "rtmp_listen not available for %s\n",
2614                proto);
2615         return AVERROR(EINVAL);
2616     }
2617     if (!strcmp(proto, "rtmpt") || !strcmp(proto, "rtmpts")) {
2618         if (!strcmp(proto, "rtmpts"))
2619             av_dict_set(&opts, "ffrtmphttp_tls", "1", 1);
2620
2621         /* open the http tunneling connection */
2622         ff_url_join(buf, sizeof(buf), "ffrtmphttp", NULL, hostname, port, NULL);
2623     } else if (!strcmp(proto, "rtmps")) {
2624         /* open the tls connection */
2625         if (port < 0)
2626             port = RTMPS_DEFAULT_PORT;
2627         ff_url_join(buf, sizeof(buf), "tls", NULL, hostname, port, NULL);
2628     } else if (!strcmp(proto, "rtmpe") || (!strcmp(proto, "rtmpte"))) {
2629         if (!strcmp(proto, "rtmpte"))
2630             av_dict_set(&opts, "ffrtmpcrypt_tunneling", "1", 1);
2631
2632         /* open the encrypted connection */
2633         ff_url_join(buf, sizeof(buf), "ffrtmpcrypt", NULL, hostname, port, NULL);
2634         rt->encrypted = 1;
2635     } else {
2636         /* open the tcp connection */
2637         if (port < 0)
2638             port = RTMP_DEFAULT_PORT;
2639         if (rt->listen)
2640             ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port,
2641                         "?listen&listen_timeout=%d",
2642                         rt->listen_timeout * 1000);
2643         else
2644             ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port, NULL);
2645     }
2646
2647 reconnect:
2648     if ((ret = ffurl_open(&rt->stream, buf, AVIO_FLAG_READ_WRITE,
2649                           &s->interrupt_callback, &opts, s->protocols, s)) < 0) {
2650         av_log(s , AV_LOG_ERROR, "Cannot open connection %s\n", buf);
2651         goto fail;
2652     }
2653
2654     if (rt->swfverify) {
2655         if ((ret = rtmp_calc_swfhash(s)) < 0)
2656             goto fail;
2657     }
2658
2659     rt->state = STATE_START;
2660     if (!rt->listen && (ret = rtmp_handshake(s, rt)) < 0)
2661         goto fail;
2662     if (rt->listen && (ret = rtmp_server_handshake(s, rt)) < 0)
2663         goto fail;
2664
2665     rt->out_chunk_size = 128;
2666     rt->in_chunk_size  = 128; // Probably overwritten later
2667     rt->state = STATE_HANDSHAKED;
2668
2669     // Keep the application name when it has been defined by the user.
2670     old_app = rt->app;
2671
2672     rt->app = av_malloc(APP_MAX_LENGTH);
2673     if (!rt->app) {
2674         ret = AVERROR(ENOMEM);
2675         goto fail;
2676     }
2677
2678     //extract "app" part from path
2679     qmark = strchr(path, '?');
2680     if (qmark && strstr(qmark, "slist=")) {
2681         char* amp;
2682         // After slist we have the playpath, the full path is used as app
2683         av_strlcpy(rt->app, path + 1, APP_MAX_LENGTH);
2684         fname = strstr(path, "slist=") + 6;
2685         // Strip any further query parameters from fname
2686         amp = strchr(fname, '&');
2687         if (amp) {
2688             av_strlcpy(fname_buffer, fname, FFMIN(amp - fname + 1,
2689                                                   sizeof(fname_buffer)));
2690             fname = fname_buffer;
2691         }
2692     } else if (!strncmp(path, "/ondemand/", 10)) {
2693         fname = path + 10;
2694         memcpy(rt->app, "ondemand", 9);
2695     } else {
2696         char *next = *path ? path + 1 : path;
2697         char *p = strchr(next, '/');
2698         if (!p) {
2699             fname = next;
2700             rt->app[0] = '\0';
2701         } else {
2702             // make sure we do not mismatch a playpath for an application instance
2703             char *c = strchr(p + 1, ':');
2704             fname = strchr(p + 1, '/');
2705             if (!fname || (c && c < fname)) {
2706                 fname = p + 1;
2707                 av_strlcpy(rt->app, path + 1, FFMIN(p - path, APP_MAX_LENGTH));
2708             } else {
2709                 fname++;
2710                 av_strlcpy(rt->app, path + 1, FFMIN(fname - path - 1, APP_MAX_LENGTH));
2711             }
2712         }
2713     }
2714
2715     if (old_app) {
2716         // The name of application has been defined by the user, override it.
2717         av_free(rt->app);
2718         rt->app = old_app;
2719     }
2720
2721     if (!rt->playpath) {
2722         int len = strlen(fname);
2723
2724         rt->playpath = av_malloc(PLAYPATH_MAX_LENGTH);
2725         if (!rt->playpath) {
2726             ret = AVERROR(ENOMEM);
2727             goto fail;
2728         }
2729
2730         if (!strchr(fname, ':') && len >= 4 &&
2731             (!strcmp(fname + len - 4, ".f4v") ||
2732              !strcmp(fname + len - 4, ".mp4"))) {
2733             memcpy(rt->playpath, "mp4:", 5);
2734         } else {
2735             if (len >= 4 && !strcmp(fname + len - 4, ".flv"))
2736                 fname[len - 4] = '\0';
2737             rt->playpath[0] = 0;
2738         }
2739         av_strlcat(rt->playpath, fname, PLAYPATH_MAX_LENGTH);
2740     }
2741
2742     if (!rt->tcurl) {
2743         rt->tcurl = av_malloc(TCURL_MAX_LENGTH);
2744         if (!rt->tcurl) {
2745             ret = AVERROR(ENOMEM);
2746             goto fail;
2747         }
2748         ff_url_join(rt->tcurl, TCURL_MAX_LENGTH, proto, NULL, hostname,
2749                     port, "/%s", rt->app);
2750     }
2751
2752     if (!rt->flashver) {
2753         rt->flashver = av_malloc(FLASHVER_MAX_LENGTH);
2754         if (!rt->flashver) {
2755             ret = AVERROR(ENOMEM);
2756             goto fail;
2757         }
2758         if (rt->is_input) {
2759             snprintf(rt->flashver, FLASHVER_MAX_LENGTH, "%s %d,%d,%d,%d",
2760                     RTMP_CLIENT_PLATFORM, RTMP_CLIENT_VER1, RTMP_CLIENT_VER2,
2761                     RTMP_CLIENT_VER3, RTMP_CLIENT_VER4);
2762         } else {
2763             snprintf(rt->flashver, FLASHVER_MAX_LENGTH,
2764                     "FMLE/3.0 (compatible; %s)", LIBAVFORMAT_IDENT);
2765         }
2766     }
2767
2768     rt->client_report_size = 1048576;
2769     rt->bytes_read = 0;
2770     rt->has_audio = 0;
2771     rt->has_video = 0;
2772     rt->received_metadata = 0;
2773     rt->last_bytes_read = 0;
2774     rt->server_bw = 2500000;
2775     rt->duration = 0;
2776
2777     av_log(s, AV_LOG_DEBUG, "Proto = %s, path = %s, app = %s, fname = %s\n",
2778            proto, path, rt->app, rt->playpath);
2779     if (!rt->listen) {
2780         if ((ret = gen_connect(s, rt)) < 0)
2781             goto fail;
2782     } else {
2783         if ((ret = read_connect(s, s->priv_data)) < 0)
2784             goto fail;
2785     }
2786
2787     do {
2788         ret = get_packet(s, 1);
2789     } while (ret == AVERROR(EAGAIN));
2790     if (ret < 0)
2791         goto fail;
2792
2793     if (rt->do_reconnect) {
2794         int i;
2795         ffurl_close(rt->stream);
2796         rt->stream       = NULL;
2797         rt->do_reconnect = 0;
2798         rt->nb_invokes   = 0;
2799         for (i = 0; i < 2; i++)
2800             memset(rt->prev_pkt[i], 0,
2801                    sizeof(**rt->prev_pkt) * rt->nb_prev_pkt[i]);
2802         free_tracked_methods(rt);
2803         goto reconnect;
2804     }
2805
2806     if (rt->is_input) {
2807         // generate FLV header for demuxer
2808         rt->flv_size = 13;
2809         if ((ret = av_reallocp(&rt->flv_data, rt->flv_size)) < 0)
2810             goto fail;
2811         rt->flv_off  = 0;
2812         memcpy(rt->flv_data, "FLV\1\0\0\0\0\011\0\0\0\0", rt->flv_size);
2813
2814         // Read packets until we reach the first A/V packet or read metadata.
2815         // If there was a metadata package in front of the A/V packets, we can
2816         // build the FLV header from this. If we do not receive any metadata,
2817         // the FLV decoder will allocate the needed streams when their first
2818         // audio or video packet arrives.
2819         while (!rt->has_audio && !rt->has_video && !rt->received_metadata) {
2820             if ((ret = get_packet(s, 0)) < 0)
2821                goto fail;
2822         }
2823
2824         // Either after we have read the metadata or (if there is none) the
2825         // first packet of an A/V stream, we have a better knowledge about the
2826         // streams, so set the FLV header accordingly.
2827         if (rt->has_audio) {
2828             rt->flv_data[4] |= FLV_HEADER_FLAG_HASAUDIO;
2829         }
2830         if (rt->has_video) {
2831             rt->flv_data[4] |= FLV_HEADER_FLAG_HASVIDEO;
2832         }
2833
2834         // If we received the first packet of an A/V stream and no metadata but
2835         // the server returned a valid duration, create a fake metadata packet
2836         // to inform the FLV decoder about the duration.
2837         if (!rt->received_metadata && rt->duration > 0) {
2838             if ((ret = inject_fake_duration_metadata(rt)) < 0)
2839                 goto fail;
2840         }
2841     } else {
2842         rt->flv_size = 0;
2843         rt->flv_data = NULL;
2844         rt->flv_off  = 0;
2845         rt->skip_bytes = 13;
2846     }
2847
2848     s->max_packet_size = rt->stream->max_packet_size;
2849     s->is_streamed     = 1;
2850     return 0;
2851
2852 fail:
2853     av_dict_free(&opts);
2854     rtmp_close(s);
2855     return ret;
2856 }
2857
2858 static int rtmp_read(URLContext *s, uint8_t *buf, int size)
2859 {
2860     RTMPContext *rt = s->priv_data;
2861     int orig_size = size;
2862     int ret;
2863
2864     while (size > 0) {
2865         int data_left = rt->flv_size - rt->flv_off;
2866
2867         if (data_left >= size) {
2868             memcpy(buf, rt->flv_data + rt->flv_off, size);
2869             rt->flv_off += size;
2870             return orig_size;
2871         }
2872         if (data_left > 0) {
2873             memcpy(buf, rt->flv_data + rt->flv_off, data_left);
2874             buf  += data_left;
2875             size -= data_left;
2876             rt->flv_off = rt->flv_size;
2877             return data_left;
2878         }
2879         if ((ret = get_packet(s, 0)) < 0)
2880            return ret;
2881     }
2882     return orig_size;
2883 }
2884
2885 static int64_t rtmp_seek(URLContext *s, int stream_index, int64_t timestamp,
2886                          int flags)
2887 {
2888     RTMPContext *rt = s->priv_data;
2889     int ret;
2890     av_log(s, AV_LOG_DEBUG,
2891            "Seek on stream index %d at timestamp %"PRId64" with flags %08x\n",
2892            stream_index, timestamp, flags);
2893     if ((ret = gen_seek(s, rt, timestamp)) < 0) {
2894         av_log(s, AV_LOG_ERROR,
2895                "Unable to send seek command on stream index %d at timestamp "
2896                "%"PRId64" with flags %08x\n",
2897                stream_index, timestamp, flags);
2898         return ret;
2899     }
2900     rt->flv_off = rt->flv_size;
2901     rt->state = STATE_SEEKING;
2902     return timestamp;
2903 }
2904
2905 static int rtmp_pause(URLContext *s, int pause)
2906 {
2907     RTMPContext *rt = s->priv_data;
2908     int ret;
2909     av_log(s, AV_LOG_DEBUG, "Pause at timestamp %d\n",
2910            rt->last_timestamp);
2911     if ((ret = gen_pause(s, rt, pause, rt->last_timestamp)) < 0) {
2912         av_log(s, AV_LOG_ERROR, "Unable to send pause command at timestamp %d\n",
2913                rt->last_timestamp);
2914         return ret;
2915     }
2916     return 0;
2917 }
2918
2919 static int rtmp_write(URLContext *s, const uint8_t *buf, int size)
2920 {
2921     RTMPContext *rt = s->priv_data;
2922     int size_temp = size;
2923     int pktsize, pkttype, copy;
2924     uint32_t ts;
2925     const uint8_t *buf_temp = buf;
2926     uint8_t c;
2927     int ret;
2928
2929     do {
2930         if (rt->skip_bytes) {
2931             int skip = FFMIN(rt->skip_bytes, size_temp);
2932             buf_temp       += skip;
2933             size_temp      -= skip;
2934             rt->skip_bytes -= skip;
2935             continue;
2936         }
2937
2938         if (rt->flv_header_bytes < RTMP_HEADER) {
2939             const uint8_t *header = rt->flv_header;
2940             int channel = RTMP_AUDIO_CHANNEL;
2941             copy = FFMIN(RTMP_HEADER - rt->flv_header_bytes, size_temp);
2942             bytestream_get_buffer(&buf_temp, rt->flv_header + rt->flv_header_bytes, copy);
2943             rt->flv_header_bytes += copy;
2944             size_temp            -= copy;
2945             if (rt->flv_header_bytes < RTMP_HEADER)
2946                 break;
2947
2948             pkttype = bytestream_get_byte(&header);
2949             pktsize = bytestream_get_be24(&header);
2950             ts = bytestream_get_be24(&header);
2951             ts |= bytestream_get_byte(&header) << 24;
2952             bytestream_get_be24(&header);
2953             rt->flv_size = pktsize;
2954
2955             if (pkttype == RTMP_PT_VIDEO)
2956                 channel = RTMP_VIDEO_CHANNEL;
2957
2958             if (((pkttype == RTMP_PT_VIDEO || pkttype == RTMP_PT_AUDIO) && ts == 0) ||
2959                 pkttype == RTMP_PT_NOTIFY) {
2960                 if ((ret = ff_rtmp_check_alloc_array(&rt->prev_pkt[1],
2961                                                      &rt->nb_prev_pkt[1],
2962                                                      channel)) < 0)
2963                     return ret;
2964                 // Force sending a full 12 bytes header by clearing the
2965                 // channel id, to make it not match a potential earlier
2966                 // packet in the same channel.
2967                 rt->prev_pkt[1][channel].channel_id = 0;
2968             }
2969
2970             //this can be a big packet, it's better to send it right here
2971             if ((ret = ff_rtmp_packet_create(&rt->out_pkt, channel,
2972                                              pkttype, ts, pktsize)) < 0)
2973                 return ret;
2974
2975             rt->out_pkt.extra = rt->stream_id;
2976             rt->flv_data = rt->out_pkt.data;
2977         }
2978
2979         copy = FFMIN(rt->flv_size - rt->flv_off, size_temp);
2980         bytestream_get_buffer(&buf_temp, rt->flv_data + rt->flv_off, copy);
2981         rt->flv_off += copy;
2982         size_temp   -= copy;
2983
2984         if (rt->flv_off == rt->flv_size) {
2985             rt->skip_bytes = 4;
2986
2987             if (rt->out_pkt.type == RTMP_PT_NOTIFY) {
2988                 // For onMetaData and |RtmpSampleAccess packets, we want
2989                 // @setDataFrame prepended to the packet before it gets sent.
2990                 // However, not all RTMP_PT_NOTIFY packets (e.g., onTextData
2991                 // and onCuePoint).
2992                 uint8_t commandbuffer[64];
2993                 int stringlen = 0;
2994                 GetByteContext gbc;
2995
2996                 bytestream2_init(&gbc, rt->flv_data, rt->flv_size);
2997                 if (!ff_amf_read_string(&gbc, commandbuffer, sizeof(commandbuffer),
2998                                         &stringlen)) {
2999                     if (!strcmp(commandbuffer, "onMetaData") ||
3000                         !strcmp(commandbuffer, "|RtmpSampleAccess")) {
3001                         uint8_t *ptr;
3002                         if ((ret = av_reallocp(&rt->out_pkt.data, rt->out_pkt.size + 16)) < 0) {
3003                             rt->flv_size = rt->flv_off = rt->flv_header_bytes = 0;
3004                             return ret;
3005                         }
3006                         memmove(rt->out_pkt.data + 16, rt->out_pkt.data, rt->out_pkt.size);
3007                         rt->out_pkt.size += 16;
3008                         ptr = rt->out_pkt.data;
3009                         ff_amf_write_string(&ptr, "@setDataFrame");
3010                     }
3011                 }
3012             }
3013
3014             if ((ret = rtmp_send_packet(rt, &rt->out_pkt, 0)) < 0)
3015                 return ret;
3016             rt->flv_size = 0;
3017             rt->flv_off = 0;
3018             rt->flv_header_bytes = 0;
3019             rt->flv_nb_packets++;
3020         }
3021     } while (buf_temp - buf < size);
3022
3023     if (rt->flv_nb_packets < rt->flush_interval)
3024         return size;
3025     rt->flv_nb_packets = 0;
3026
3027     /* set stream into nonblocking mode */
3028     rt->stream->flags |= AVIO_FLAG_NONBLOCK;
3029
3030     /* try to read one byte from the stream */
3031     ret = ffurl_read(rt->stream, &c, 1);
3032
3033     /* switch the stream back into blocking mode */
3034     rt->stream->flags &= ~AVIO_FLAG_NONBLOCK;
3035
3036     if (ret == AVERROR(EAGAIN)) {
3037         /* no incoming data to handle */
3038         return size;
3039     } else if (ret < 0) {
3040         return ret;
3041     } else if (ret == 1) {
3042         RTMPPacket rpkt = { 0 };
3043
3044         if ((ret = ff_rtmp_packet_read_internal(rt->stream, &rpkt,
3045                                                 rt->in_chunk_size,
3046                                                 &rt->prev_pkt[0],
3047                                                 &rt->nb_prev_pkt[0], c)) <= 0)
3048              return ret;
3049
3050         if ((ret = rtmp_parse_result(s, rt, &rpkt)) < 0)
3051             return ret;
3052
3053         ff_rtmp_packet_destroy(&rpkt);
3054     }
3055
3056     return size;
3057 }
3058
3059 #define OFFSET(x) offsetof(RTMPContext, x)
3060 #define DEC AV_OPT_FLAG_DECODING_PARAM
3061 #define ENC AV_OPT_FLAG_ENCODING_PARAM
3062
3063 static const AVOption rtmp_options[] = {
3064     {"rtmp_app", "Name of application to connect to on the RTMP server", OFFSET(app), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
3065     {"rtmp_buffer", "Set buffer time in milliseconds. The default is 3000.", OFFSET(client_buffer_time), AV_OPT_TYPE_INT, {.i64 = 3000}, 0, INT_MAX, DEC|ENC},
3066     {"rtmp_conn", "Append arbitrary AMF data to the Connect message", OFFSET(conn), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
3067     {"rtmp_flashver", "Version of the Flash plugin used to run the SWF player.", OFFSET(flashver), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
3068     {"rtmp_flush_interval", "Number of packets flushed in the same request (RTMPT only).", OFFSET(flush_interval), AV_OPT_TYPE_INT, {.i64 = 10}, 0, INT_MAX, ENC},
3069     {"rtmp_live", "Specify that the media is a live stream.", OFFSET(live), AV_OPT_TYPE_INT, {.i64 = -2}, INT_MIN, INT_MAX, DEC, "rtmp_live"},
3070     {"any", "both", 0, AV_OPT_TYPE_CONST, {.i64 = -2}, 0, 0, DEC, "rtmp_live"},
3071     {"live", "live stream", 0, AV_OPT_TYPE_CONST, {.i64 = -1}, 0, 0, DEC, "rtmp_live"},
3072     {"recorded", "recorded stream", 0, AV_OPT_TYPE_CONST, {.i64 = 0}, 0, 0, DEC, "rtmp_live"},
3073     {"rtmp_pageurl", "URL of the web page in which the media was embedded. By default no value will be sent.", OFFSET(pageurl), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC},
3074     {"rtmp_playpath", "Stream identifier to play or to publish", OFFSET(playpath), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
3075     {"rtmp_subscribe", "Name of live stream to subscribe to. Defaults to rtmp_playpath.", OFFSET(subscribe), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC},
3076     {"rtmp_swfhash", "SHA256 hash of the decompressed SWF file (32 bytes).", OFFSET(swfhash), AV_OPT_TYPE_BINARY, .flags = DEC},
3077     {"rtmp_swfsize", "Size of the decompressed SWF file, required for SWFVerification.", OFFSET(swfsize), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, DEC},
3078     {"rtmp_swfurl", "URL of the SWF player. By default no value will be sent", OFFSET(swfurl), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
3079     {"rtmp_swfverify", "URL to player swf file, compute hash/size automatically.", OFFSET(swfverify), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC},
3080     {"rtmp_tcurl", "URL of the target stream. Defaults to proto://host[:port]/app.", OFFSET(tcurl), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
3081     {"rtmp_listen", "Listen for incoming rtmp connections", OFFSET(listen), AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, DEC, "rtmp_listen" },
3082     {"listen",      "Listen for incoming rtmp connections", OFFSET(listen), AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, DEC, "rtmp_listen" },
3083     {"timeout", "Maximum timeout (in seconds) to wait for incoming connections. -1 is infinite. Implies -rtmp_listen 1",  OFFSET(listen_timeout), AV_OPT_TYPE_INT, {.i64 = -1}, INT_MIN, INT_MAX, DEC, "rtmp_listen" },
3084     { NULL },
3085 };
3086
3087 #define RTMP_PROTOCOL(flavor)                    \
3088 static const AVClass flavor##_class = {          \
3089     .class_name = #flavor,                       \
3090     .item_name  = av_default_item_name,          \
3091     .option     = rtmp_options,                  \
3092     .version    = LIBAVUTIL_VERSION_INT,         \
3093 };                                               \
3094                                                  \
3095 const URLProtocol ff_##flavor##_protocol = {     \
3096     .name           = #flavor,                   \
3097     .url_open       = rtmp_open,                 \
3098     .url_read       = rtmp_read,                 \
3099     .url_read_seek  = rtmp_seek,                 \
3100     .url_read_pause = rtmp_pause,                \
3101     .url_write      = rtmp_write,                \
3102     .url_close      = rtmp_close,                \
3103     .priv_data_size = sizeof(RTMPContext),       \
3104     .flags          = URL_PROTOCOL_FLAG_NETWORK, \
3105     .priv_data_class= &flavor##_class,           \
3106 };
3107
3108
3109 RTMP_PROTOCOL(rtmp)
3110 RTMP_PROTOCOL(rtmpe)
3111 RTMP_PROTOCOL(rtmps)
3112 RTMP_PROTOCOL(rtmpt)
3113 RTMP_PROTOCOL(rtmpte)
3114 RTMP_PROTOCOL(rtmpts)