123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460 |
- /*
- * RTMP network protocol
- * Copyright (c) 2009 Kostya Shishkov
- *
- * This file is part of FFmpeg.
- *
- * FFmpeg is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 2.1 of the License, or (at your option) any later version.
- *
- * FFmpeg is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with FFmpeg; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
- */
- /**
- * @file
- * RTMP protocol
- */
- #include "libavcodec/bytestream.h"
- #include "libavutil/avstring.h"
- #include "libavutil/intfloat.h"
- #include "libavutil/lfg.h"
- #include "libavutil/opt.h"
- #include "libavutil/random_seed.h"
- #include "libavutil/sha.h"
- #include "avformat.h"
- #include "internal.h"
- #include "network.h"
- #include "flv.h"
- #include "rtmp.h"
- #include "rtmpcrypt.h"
- #include "rtmppkt.h"
- #include "url.h"
- #if CONFIG_ZLIB
- #include <zlib.h>
- #endif
- //#define DEBUG
- #define APP_MAX_LENGTH 1024
- #define PLAYPATH_MAX_LENGTH 256
- #define TCURL_MAX_LENGTH 512
- #define FLASHVER_MAX_LENGTH 64
- #define RTMP_PKTDATA_DEFAULT_SIZE 4096
- /** RTMP protocol handler state */
- typedef enum {
- STATE_START, ///< client has not done anything yet
- STATE_HANDSHAKED, ///< client has performed handshake
- STATE_FCPUBLISH, ///< client FCPublishing stream (for output)
- STATE_PLAYING, ///< client has started receiving multimedia data from server
- STATE_PUBLISHING, ///< client has started sending multimedia data to server (for output)
- STATE_RECEIVING, ///< received a publish command (for input)
- STATE_STOPPED, ///< the broadcast has been stopped
- } ClientState;
- typedef struct TrackedMethod {
- char *name;
- int id;
- } TrackedMethod;
- /** protocol handler context */
- typedef struct RTMPContext {
- const AVClass *class;
- URLContext* stream; ///< TCP stream used in interactions with RTMP server
- RTMPPacket prev_pkt[2][RTMP_CHANNELS]; ///< packet history used when reading and sending packets
- int in_chunk_size; ///< size of the chunks incoming RTMP packets are divided into
- int out_chunk_size; ///< size of the chunks outgoing RTMP packets are divided into
- int is_input; ///< input/output flag
- char *playpath; ///< stream identifier to play (with possible "mp4:" prefix)
- int live; ///< 0: recorded, -1: live, -2: both
- char *app; ///< name of application
- char *conn; ///< append arbitrary AMF data to the Connect message
- ClientState state; ///< current state
- int main_channel_id; ///< an additional channel ID which is used for some invocations
- uint8_t* flv_data; ///< buffer with data for demuxer
- int flv_size; ///< current buffer size
- int flv_off; ///< number of bytes read from current buffer
- int flv_nb_packets; ///< number of flv packets published
- RTMPPacket out_pkt; ///< rtmp packet, created from flv a/v or metadata (for output)
- uint32_t client_report_size; ///< number of bytes after which client should report to server
- uint32_t bytes_read; ///< number of bytes read from server
- uint32_t last_bytes_read; ///< number of bytes read last reported to server
- int skip_bytes; ///< number of bytes to skip from the input FLV stream in the next write call
- uint8_t flv_header[11]; ///< partial incoming flv packet header
- int flv_header_bytes; ///< number of initialized bytes in flv_header
- int nb_invokes; ///< keeps track of invoke messages
- char* tcurl; ///< url of the target stream
- char* flashver; ///< version of the flash plugin
- char* swfhash; ///< SHA256 hash of the decompressed SWF file (32 bytes)
- int swfhash_len; ///< length of the SHA256 hash
- int swfsize; ///< size of the decompressed SWF file
- char* swfurl; ///< url of the swf player
- char* swfverify; ///< URL to player swf file, compute hash/size automatically
- char swfverification[42]; ///< hash of the SWF verification
- char* pageurl; ///< url of the web page
- char* subscribe; ///< name of live stream to subscribe
- int server_bw; ///< server bandwidth
- int client_buffer_time; ///< client buffer time in ms
- int flush_interval; ///< number of packets flushed in the same request (RTMPT only)
- int encrypted; ///< use an encrypted connection (RTMPE only)
- TrackedMethod*tracked_methods; ///< tracked methods buffer
- int nb_tracked_methods; ///< number of tracked methods
- int tracked_methods_size; ///< size of the tracked methods buffer
- int listen; ///< listen mode flag
- int listen_timeout; ///< listen timeout to wait for new connections
- int nb_streamid; ///< The next stream id to return on createStream calls
- } RTMPContext;
- #define PLAYER_KEY_OPEN_PART_LEN 30 ///< length of partial key used for first client digest signing
- /** Client key used for digest signing */
- static const uint8_t rtmp_player_key[] = {
- 'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
- 'F', 'l', 'a', 's', 'h', ' ', 'P', 'l', 'a', 'y', 'e', 'r', ' ', '0', '0', '1',
- 0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1, 0x02,
- 0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB, 0x93, 0xB8,
- 0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE
- };
- #define SERVER_KEY_OPEN_PART_LEN 36 ///< length of partial key used for first server digest signing
- /** Key used for RTMP server digest signing */
- static const uint8_t rtmp_server_key[] = {
- 'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
- 'F', 'l', 'a', 's', 'h', ' ', 'M', 'e', 'd', 'i', 'a', ' ',
- 'S', 'e', 'r', 'v', 'e', 'r', ' ', '0', '0', '1',
- 0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1, 0x02,
- 0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB, 0x93, 0xB8,
- 0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE
- };
- static int add_tracked_method(RTMPContext *rt, const char *name, int id)
- {
- void *ptr;
- if (rt->nb_tracked_methods + 1 > rt->tracked_methods_size) {
- rt->tracked_methods_size = (rt->nb_tracked_methods + 1) * 2;
- ptr = av_realloc(rt->tracked_methods,
- rt->tracked_methods_size * sizeof(*rt->tracked_methods));
- if (!ptr)
- return AVERROR(ENOMEM);
- rt->tracked_methods = ptr;
- }
- rt->tracked_methods[rt->nb_tracked_methods].name = av_strdup(name);
- if (!rt->tracked_methods[rt->nb_tracked_methods].name)
- return AVERROR(ENOMEM);
- rt->tracked_methods[rt->nb_tracked_methods].id = id;
- rt->nb_tracked_methods++;
- return 0;
- }
- static void del_tracked_method(RTMPContext *rt, int index)
- {
- memmove(&rt->tracked_methods[index], &rt->tracked_methods[index + 1],
- sizeof(*rt->tracked_methods) * (rt->nb_tracked_methods - index - 1));
- rt->nb_tracked_methods--;
- }
- static int find_tracked_method(URLContext *s, RTMPPacket *pkt, int offset,
- char **tracked_method)
- {
- RTMPContext *rt = s->priv_data;
- GetByteContext gbc;
- double pkt_id;
- int ret;
- int i;
- bytestream2_init(&gbc, pkt->data + offset, pkt->data_size - offset);
- if ((ret = ff_amf_read_number(&gbc, &pkt_id)) < 0)
- return ret;
- for (i = 0; i < rt->nb_tracked_methods; i++) {
- if (rt->tracked_methods[i].id != pkt_id)
- continue;
- *tracked_method = rt->tracked_methods[i].name;
- del_tracked_method(rt, i);
- break;
- }
- return 0;
- }
- static void free_tracked_methods(RTMPContext *rt)
- {
- int i;
- for (i = 0; i < rt->nb_tracked_methods; i ++)
- av_free(rt->tracked_methods[i].name);
- av_free(rt->tracked_methods);
- }
- static int rtmp_send_packet(RTMPContext *rt, RTMPPacket *pkt, int track)
- {
- int ret;
- if (pkt->type == RTMP_PT_INVOKE && track) {
- GetByteContext gbc;
- char name[128];
- double pkt_id;
- int len;
- bytestream2_init(&gbc, pkt->data, pkt->data_size);
- if ((ret = ff_amf_read_string(&gbc, name, sizeof(name), &len)) < 0)
- goto fail;
- if ((ret = ff_amf_read_number(&gbc, &pkt_id)) < 0)
- goto fail;
- if ((ret = add_tracked_method(rt, name, pkt_id)) < 0)
- goto fail;
- }
- ret = ff_rtmp_packet_write(rt->stream, pkt, rt->out_chunk_size,
- rt->prev_pkt[1]);
- fail:
- ff_rtmp_packet_destroy(pkt);
- return ret;
- }
- static int rtmp_write_amf_data(URLContext *s, char *param, uint8_t **p)
- {
- char *field, *value;
- char type;
- /* The type must be B for Boolean, N for number, S for string, O for
- * object, or Z for null. For Booleans the data must be either 0 or 1 for
- * FALSE or TRUE, respectively. Likewise for Objects the data must be
- * 0 or 1 to end or begin an object, respectively. Data items in subobjects
- * may be named, by prefixing the type with 'N' and specifying the name
- * before the value (ie. NB:myFlag:1). This option may be used multiple times
- * to construct arbitrary AMF sequences. */
- if (param[0] && param[1] == ':') {
- type = param[0];
- value = param + 2;
- } else if (param[0] == 'N' && param[1] && param[2] == ':') {
- type = param[1];
- field = param + 3;
- value = strchr(field, ':');
- if (!value)
- goto fail;
- *value = '\0';
- value++;
- if (!field || !value)
- goto fail;
- ff_amf_write_field_name(p, field);
- } else {
- goto fail;
- }
- switch (type) {
- case 'B':
- ff_amf_write_bool(p, value[0] != '0');
- break;
- case 'S':
- ff_amf_write_string(p, value);
- break;
- case 'N':
- ff_amf_write_number(p, strtod(value, NULL));
- break;
- case 'Z':
- ff_amf_write_null(p);
- break;
- case 'O':
- if (value[0] != '0')
- ff_amf_write_object_start(p);
- else
- ff_amf_write_object_end(p);
- break;
- default:
- goto fail;
- break;
- }
- return 0;
- fail:
- av_log(s, AV_LOG_ERROR, "Invalid AMF parameter: %s\n", param);
- return AVERROR(EINVAL);
- }
- /**
- * Generate 'connect' call and send it to the server.
- */
- static int gen_connect(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
- 0, 4096 + APP_MAX_LENGTH)) < 0)
- return ret;
- p = pkt.data;
- ff_amf_write_string(&p, "connect");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_object_start(&p);
- ff_amf_write_field_name(&p, "app");
- ff_amf_write_string(&p, rt->app);
- if (!rt->is_input) {
- ff_amf_write_field_name(&p, "type");
- ff_amf_write_string(&p, "nonprivate");
- }
- ff_amf_write_field_name(&p, "flashVer");
- ff_amf_write_string(&p, rt->flashver);
- if (rt->swfurl) {
- ff_amf_write_field_name(&p, "swfUrl");
- ff_amf_write_string(&p, rt->swfurl);
- }
- ff_amf_write_field_name(&p, "tcUrl");
- ff_amf_write_string(&p, rt->tcurl);
- if (rt->is_input) {
- ff_amf_write_field_name(&p, "fpad");
- ff_amf_write_bool(&p, 0);
- ff_amf_write_field_name(&p, "capabilities");
- ff_amf_write_number(&p, 15.0);
- /* Tell the server we support all the audio codecs except
- * SUPPORT_SND_INTEL (0x0008) and SUPPORT_SND_UNUSED (0x0010)
- * which are unused in the RTMP protocol implementation. */
- ff_amf_write_field_name(&p, "audioCodecs");
- ff_amf_write_number(&p, 4071.0);
- ff_amf_write_field_name(&p, "videoCodecs");
- ff_amf_write_number(&p, 252.0);
- ff_amf_write_field_name(&p, "videoFunction");
- ff_amf_write_number(&p, 1.0);
- if (rt->pageurl) {
- ff_amf_write_field_name(&p, "pageUrl");
- ff_amf_write_string(&p, rt->pageurl);
- }
- }
- ff_amf_write_object_end(&p);
- if (rt->conn) {
- char *param = rt->conn;
- // Write arbitrary AMF data to the Connect message.
- while (param != NULL) {
- char *sep;
- param += strspn(param, " ");
- if (!*param)
- break;
- sep = strchr(param, ' ');
- if (sep)
- *sep = '\0';
- if ((ret = rtmp_write_amf_data(s, param, &p)) < 0) {
- // Invalid AMF parameter.
- ff_rtmp_packet_destroy(&pkt);
- return ret;
- }
- if (sep)
- param = sep + 1;
- else
- break;
- }
- }
- pkt.data_size = p - pkt.data;
- return rtmp_send_packet(rt, &pkt, 1);
- }
- static int read_connect(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt = { 0 };
- uint8_t *p;
- const uint8_t *cp;
- int ret;
- char command[64];
- int stringlen;
- double seqnum;
- uint8_t tmpstr[256];
- GetByteContext gbc;
- if ((ret = ff_rtmp_packet_read(rt->stream, &pkt, rt->in_chunk_size,
- rt->prev_pkt[1])) < 0)
- return ret;
- cp = pkt.data;
- bytestream2_init(&gbc, cp, pkt.data_size);
- if (ff_amf_read_string(&gbc, command, sizeof(command), &stringlen)) {
- av_log(s, AV_LOG_ERROR, "Unable to read command string\n");
- ff_rtmp_packet_destroy(&pkt);
- return AVERROR_INVALIDDATA;
- }
- if (strcmp(command, "connect")) {
- av_log(s, AV_LOG_ERROR, "Expecting connect, got %s\n", command);
- ff_rtmp_packet_destroy(&pkt);
- return AVERROR_INVALIDDATA;
- }
- ret = ff_amf_read_number(&gbc, &seqnum);
- if (ret)
- av_log(s, AV_LOG_WARNING, "SeqNum not found\n");
- /* Here one could parse an AMF Object with data as flashVers and others. */
- ret = ff_amf_get_field_value(gbc.buffer,
- gbc.buffer + bytestream2_get_bytes_left(&gbc),
- "app", tmpstr, sizeof(tmpstr));
- if (ret)
- av_log(s, AV_LOG_WARNING, "App field not found in connect\n");
- if (!ret && strcmp(tmpstr, rt->app))
- av_log(s, AV_LOG_WARNING, "App field don't match up: %s <-> %s\n",
- tmpstr, rt->app);
- ff_rtmp_packet_destroy(&pkt);
- // Send Window Acknowledgement Size (as defined in speficication)
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL,
- RTMP_PT_SERVER_BW, 0, 4)) < 0)
- return ret;
- p = pkt.data;
- bytestream_put_be32(&p, rt->server_bw);
- pkt.data_size = p - pkt.data;
- ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
- rt->prev_pkt[1]);
- ff_rtmp_packet_destroy(&pkt);
- if (ret < 0)
- return ret;
- // Send Peer Bandwidth
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL,
- RTMP_PT_CLIENT_BW, 0, 5)) < 0)
- return ret;
- p = pkt.data;
- bytestream_put_be32(&p, rt->server_bw);
- bytestream_put_byte(&p, 2); // dynamic
- pkt.data_size = p - pkt.data;
- ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
- rt->prev_pkt[1]);
- ff_rtmp_packet_destroy(&pkt);
- if (ret < 0)
- return ret;
- // Ping request
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL,
- RTMP_PT_PING, 0, 6)) < 0)
- return ret;
- p = pkt.data;
- bytestream_put_be16(&p, 0); // 0 -> Stream Begin
- bytestream_put_be32(&p, 0);
- ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
- rt->prev_pkt[1]);
- ff_rtmp_packet_destroy(&pkt);
- if (ret < 0)
- return ret;
- // Chunk size
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL,
- RTMP_PT_CHUNK_SIZE, 0, 4)) < 0)
- return ret;
- p = pkt.data;
- bytestream_put_be32(&p, rt->out_chunk_size);
- ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
- rt->prev_pkt[1]);
- ff_rtmp_packet_destroy(&pkt);
- if (ret < 0)
- return ret;
- // Send result_ NetConnection.Connect.Success to connect
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL,
- RTMP_PT_INVOKE, 0,
- RTMP_PKTDATA_DEFAULT_SIZE)) < 0)
- return ret;
- p = pkt.data;
- ff_amf_write_string(&p, "_result");
- ff_amf_write_number(&p, seqnum);
- ff_amf_write_object_start(&p);
- ff_amf_write_field_name(&p, "fmsVer");
- ff_amf_write_string(&p, "FMS/3,0,1,123");
- ff_amf_write_field_name(&p, "capabilities");
- ff_amf_write_number(&p, 31);
- ff_amf_write_object_end(&p);
- ff_amf_write_object_start(&p);
- ff_amf_write_field_name(&p, "level");
- ff_amf_write_string(&p, "status");
- ff_amf_write_field_name(&p, "code");
- ff_amf_write_string(&p, "NetConnection.Connect.Success");
- ff_amf_write_field_name(&p, "description");
- ff_amf_write_string(&p, "Connection succeeded.");
- ff_amf_write_field_name(&p, "objectEncoding");
- ff_amf_write_number(&p, 0);
- ff_amf_write_object_end(&p);
- pkt.data_size = p - pkt.data;
- ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
- rt->prev_pkt[1]);
- ff_rtmp_packet_destroy(&pkt);
- if (ret < 0)
- return ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL,
- RTMP_PT_INVOKE, 0, 30)) < 0)
- return ret;
- p = pkt.data;
- ff_amf_write_string(&p, "onBWDone");
- ff_amf_write_number(&p, 0);
- ff_amf_write_null(&p);
- ff_amf_write_number(&p, 8192);
- pkt.data_size = p - pkt.data;
- ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->out_chunk_size,
- rt->prev_pkt[1]);
- ff_rtmp_packet_destroy(&pkt);
- return ret;
- }
- /**
- * Generate 'releaseStream' call and send it to the server. It should make
- * the server release some channel for media streams.
- */
- static int gen_release_stream(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
- 0, 29 + strlen(rt->playpath))) < 0)
- return ret;
- av_log(s, AV_LOG_DEBUG, "Releasing stream...\n");
- p = pkt.data;
- ff_amf_write_string(&p, "releaseStream");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_null(&p);
- ff_amf_write_string(&p, rt->playpath);
- return rtmp_send_packet(rt, &pkt, 1);
- }
- /**
- * Generate 'FCPublish' call and send it to the server. It should make
- * the server preapare for receiving media streams.
- */
- static int gen_fcpublish_stream(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
- 0, 25 + strlen(rt->playpath))) < 0)
- return ret;
- av_log(s, AV_LOG_DEBUG, "FCPublish stream...\n");
- p = pkt.data;
- ff_amf_write_string(&p, "FCPublish");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_null(&p);
- ff_amf_write_string(&p, rt->playpath);
- return rtmp_send_packet(rt, &pkt, 1);
- }
- /**
- * Generate 'FCUnpublish' call and send it to the server. It should make
- * the server destroy stream.
- */
- static int gen_fcunpublish_stream(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
- 0, 27 + strlen(rt->playpath))) < 0)
- return ret;
- av_log(s, AV_LOG_DEBUG, "UnPublishing stream...\n");
- p = pkt.data;
- ff_amf_write_string(&p, "FCUnpublish");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_null(&p);
- ff_amf_write_string(&p, rt->playpath);
- return rtmp_send_packet(rt, &pkt, 0);
- }
- /**
- * Generate 'createStream' call and send it to the server. It should make
- * the server allocate some channel for media streams.
- */
- static int gen_create_stream(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- av_log(s, AV_LOG_DEBUG, "Creating stream...\n");
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
- 0, 25)) < 0)
- return ret;
- p = pkt.data;
- ff_amf_write_string(&p, "createStream");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_null(&p);
- return rtmp_send_packet(rt, &pkt, 1);
- }
- /**
- * Generate 'deleteStream' call and send it to the server. It should make
- * the server remove some channel for media streams.
- */
- static int gen_delete_stream(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- av_log(s, AV_LOG_DEBUG, "Deleting stream...\n");
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
- 0, 34)) < 0)
- return ret;
- p = pkt.data;
- ff_amf_write_string(&p, "deleteStream");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_null(&p);
- ff_amf_write_number(&p, rt->main_channel_id);
- return rtmp_send_packet(rt, &pkt, 0);
- }
- /**
- * Generate client buffer time and send it to the server.
- */
- static int gen_buffer_time(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_PING,
- 1, 10)) < 0)
- return ret;
- p = pkt.data;
- bytestream_put_be16(&p, 3);
- bytestream_put_be32(&p, rt->main_channel_id);
- bytestream_put_be32(&p, rt->client_buffer_time);
- return rtmp_send_packet(rt, &pkt, 0);
- }
- /**
- * Generate 'play' call and send it to the server, then ping the server
- * to start actual playing.
- */
- static int gen_play(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- av_log(s, AV_LOG_DEBUG, "Sending play command for '%s'\n", rt->playpath);
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_VIDEO_CHANNEL, RTMP_PT_INVOKE,
- 0, 29 + strlen(rt->playpath))) < 0)
- return ret;
- pkt.extra = rt->main_channel_id;
- p = pkt.data;
- ff_amf_write_string(&p, "play");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_null(&p);
- ff_amf_write_string(&p, rt->playpath);
- ff_amf_write_number(&p, rt->live);
- return rtmp_send_packet(rt, &pkt, 1);
- }
- /**
- * Generate 'publish' call and send it to the server.
- */
- static int gen_publish(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- av_log(s, AV_LOG_DEBUG, "Sending publish command for '%s'\n", rt->playpath);
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SOURCE_CHANNEL, RTMP_PT_INVOKE,
- 0, 30 + strlen(rt->playpath))) < 0)
- return ret;
- pkt.extra = rt->main_channel_id;
- p = pkt.data;
- ff_amf_write_string(&p, "publish");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_null(&p);
- ff_amf_write_string(&p, rt->playpath);
- ff_amf_write_string(&p, "live");
- return rtmp_send_packet(rt, &pkt, 1);
- }
- /**
- * Generate ping reply and send it to the server.
- */
- static int gen_pong(URLContext *s, RTMPContext *rt, RTMPPacket *ppkt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if (ppkt->data_size < 6) {
- av_log(s, AV_LOG_ERROR, "Too short ping packet (%d)\n",
- ppkt->data_size);
- return AVERROR_INVALIDDATA;
- }
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_PING,
- ppkt->timestamp + 1, 6)) < 0)
- return ret;
- p = pkt.data;
- bytestream_put_be16(&p, 7);
- bytestream_put_be32(&p, AV_RB32(ppkt->data+2));
- return rtmp_send_packet(rt, &pkt, 0);
- }
- /**
- * Generate SWF verification message and send it to the server.
- */
- static int gen_swf_verification(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- av_log(s, AV_LOG_DEBUG, "Sending SWF verification...\n");
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_PING,
- 0, 44)) < 0)
- return ret;
- p = pkt.data;
- bytestream_put_be16(&p, 27);
- memcpy(p, rt->swfverification, 42);
- return rtmp_send_packet(rt, &pkt, 0);
- }
- /**
- * Generate server bandwidth message and send it to the server.
- */
- static int gen_server_bw(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_SERVER_BW,
- 0, 4)) < 0)
- return ret;
- p = pkt.data;
- bytestream_put_be32(&p, rt->server_bw);
- return rtmp_send_packet(rt, &pkt, 0);
- }
- /**
- * Generate check bandwidth message and send it to the server.
- */
- static int gen_check_bw(URLContext *s, RTMPContext *rt)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
- 0, 21)) < 0)
- return ret;
- p = pkt.data;
- ff_amf_write_string(&p, "_checkbw");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_null(&p);
- return rtmp_send_packet(rt, &pkt, 1);
- }
- /**
- * Generate report on bytes read so far and send it to the server.
- */
- static int gen_bytes_read(URLContext *s, RTMPContext *rt, uint32_t ts)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_BYTES_READ,
- ts, 4)) < 0)
- return ret;
- p = pkt.data;
- bytestream_put_be32(&p, rt->bytes_read);
- return rtmp_send_packet(rt, &pkt, 0);
- }
- static int gen_fcsubscribe_stream(URLContext *s, RTMPContext *rt,
- const char *subscribe)
- {
- RTMPPacket pkt;
- uint8_t *p;
- int ret;
- if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
- 0, 27 + strlen(subscribe))) < 0)
- return ret;
- p = pkt.data;
- ff_amf_write_string(&p, "FCSubscribe");
- ff_amf_write_number(&p, ++rt->nb_invokes);
- ff_amf_write_null(&p);
- ff_amf_write_string(&p, subscribe);
- return rtmp_send_packet(rt, &pkt, 1);
- }
- int ff_rtmp_calc_digest(const uint8_t *src, int len, int gap,
- const uint8_t *key, int keylen, uint8_t *dst)
- {
- struct AVSHA *sha;
- uint8_t hmac_buf[64+32] = {0};
- int i;
- sha = av_mallocz(av_sha_size);
- if (!sha)
- return AVERROR(ENOMEM);
- if (keylen < 64) {
- memcpy(hmac_buf, key, keylen);
- } else {
- av_sha_init(sha, 256);
- av_sha_update(sha,key, keylen);
- av_sha_final(sha, hmac_buf);
- }
- for (i = 0; i < 64; i++)
- hmac_buf[i] ^= HMAC_IPAD_VAL;
- av_sha_init(sha, 256);
- av_sha_update(sha, hmac_buf, 64);
- if (gap <= 0) {
- av_sha_update(sha, src, len);
- } else { //skip 32 bytes used for storing digest
- av_sha_update(sha, src, gap);
- av_sha_update(sha, src + gap + 32, len - gap - 32);
- }
- av_sha_final(sha, hmac_buf + 64);
- for (i = 0; i < 64; i++)
- hmac_buf[i] ^= HMAC_IPAD_VAL ^ HMAC_OPAD_VAL; //reuse XORed key for opad
- av_sha_init(sha, 256);
- av_sha_update(sha, hmac_buf, 64+32);
- av_sha_final(sha, dst);
- av_free(sha);
- return 0;
- }
- int ff_rtmp_calc_digest_pos(const uint8_t *buf, int off, int mod_val,
- int add_val)
- {
- int i, digest_pos = 0;
- for (i = 0; i < 4; i++)
- digest_pos += buf[i + off];
- digest_pos = digest_pos % mod_val + add_val;
- return digest_pos;
- }
- /**
- * Put HMAC-SHA2 digest of packet data (except for the bytes where this digest
- * will be stored) into that packet.
- *
- * @param buf handshake data (1536 bytes)
- * @param encrypted use an encrypted connection (RTMPE)
- * @return offset to the digest inside input data
- */
- static int rtmp_handshake_imprint_with_digest(uint8_t *buf, int encrypted)
- {
- int ret, digest_pos;
- if (encrypted)
- digest_pos = ff_rtmp_calc_digest_pos(buf, 772, 728, 776);
- else
- digest_pos = ff_rtmp_calc_digest_pos(buf, 8, 728, 12);
- ret = ff_rtmp_calc_digest(buf, RTMP_HANDSHAKE_PACKET_SIZE, digest_pos,
- rtmp_player_key, PLAYER_KEY_OPEN_PART_LEN,
- buf + digest_pos);
- if (ret < 0)
- return ret;
- return digest_pos;
- }
- /**
- * Verify that the received server response has the expected digest value.
- *
- * @param buf handshake data received from the server (1536 bytes)
- * @param off position to search digest offset from
- * @return 0 if digest is valid, digest position otherwise
- */
- static int rtmp_validate_digest(uint8_t *buf, int off)
- {
- uint8_t digest[32];
- int ret, digest_pos;
- digest_pos = ff_rtmp_calc_digest_pos(buf, off, 728, off + 4);
- ret = ff_rtmp_calc_digest(buf, RTMP_HANDSHAKE_PACKET_SIZE, digest_pos,
- rtmp_server_key, SERVER_KEY_OPEN_PART_LEN,
- digest);
- if (ret < 0)
- return ret;
- if (!memcmp(digest, buf + digest_pos, 32))
- return digest_pos;
- return 0;
- }
- static int rtmp_calc_swf_verification(URLContext *s, RTMPContext *rt,
- uint8_t *buf)
- {
- uint8_t *p;
- int ret;
- if (rt->swfhash_len != 32) {
- av_log(s, AV_LOG_ERROR,
- "Hash of the decompressed SWF file is not 32 bytes long.\n");
- return AVERROR(EINVAL);
- }
- p = &rt->swfverification[0];
- bytestream_put_byte(&p, 1);
- bytestream_put_byte(&p, 1);
- bytestream_put_be32(&p, rt->swfsize);
- bytestream_put_be32(&p, rt->swfsize);
- if ((ret = ff_rtmp_calc_digest(rt->swfhash, 32, 0, buf, 32, p)) < 0)
- return ret;
- return 0;
- }
- #if CONFIG_ZLIB
- static int rtmp_uncompress_swfplayer(uint8_t *in_data, int64_t in_size,
- uint8_t **out_data, int64_t *out_size)
- {
- z_stream zs = { 0 };
- void *ptr;
- int size;
- int ret = 0;
- zs.avail_in = in_size;
- zs.next_in = in_data;
- ret = inflateInit(&zs);
- if (ret != Z_OK)
- return AVERROR_UNKNOWN;
- do {
- uint8_t tmp_buf[16384];
- zs.avail_out = sizeof(tmp_buf);
- zs.next_out = tmp_buf;
- ret = inflate(&zs, Z_NO_FLUSH);
- if (ret != Z_OK && ret != Z_STREAM_END) {
- ret = AVERROR_UNKNOWN;
- goto fail;
- }
- size = sizeof(tmp_buf) - zs.avail_out;
- if (!(ptr = av_realloc(*out_data, *out_size + size))) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- *out_data = ptr;
- memcpy(*out_data + *out_size, tmp_buf, size);
- *out_size += size;
- } while (zs.avail_out == 0);
- fail:
- inflateEnd(&zs);
- return ret;
- }
- #endif
- static int rtmp_calc_swfhash(URLContext *s)
- {
- RTMPContext *rt = s->priv_data;
- uint8_t *in_data = NULL, *out_data = NULL, *swfdata;
- int64_t in_size, out_size;
- URLContext *stream;
- char swfhash[32];
- int swfsize;
- int ret = 0;
- /* Get the SWF player file. */
- if ((ret = ffurl_open(&stream, rt->swfverify, AVIO_FLAG_READ,
- &s->interrupt_callback, NULL)) < 0) {
- av_log(s, AV_LOG_ERROR, "Cannot open connection %s.\n", rt->swfverify);
- goto fail;
- }
- if ((in_size = ffurl_seek(stream, 0, AVSEEK_SIZE)) < 0) {
- ret = AVERROR(EIO);
- goto fail;
- }
- if (!(in_data = av_malloc(in_size))) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- if ((ret = ffurl_read_complete(stream, in_data, in_size)) < 0)
- goto fail;
- if (in_size < 3) {
- ret = AVERROR_INVALIDDATA;
- goto fail;
- }
- if (!memcmp(in_data, "CWS", 3)) {
- /* Decompress the SWF player file using Zlib. */
- if (!(out_data = av_malloc(8))) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- *in_data = 'F'; // magic stuff
- memcpy(out_data, in_data, 8);
- out_size = 8;
- #if CONFIG_ZLIB
- if ((ret = rtmp_uncompress_swfplayer(in_data + 8, in_size - 8,
- &out_data, &out_size)) < 0)
- goto fail;
- #else
- av_log(s, AV_LOG_ERROR,
- "Zlib is required for decompressing the SWF player file.\n");
- ret = AVERROR(EINVAL);
- goto fail;
- #endif
- swfsize = out_size;
- swfdata = out_data;
- } else {
- swfsize = in_size;
- swfdata = in_data;
- }
- /* Compute the SHA256 hash of the SWF player file. */
- if ((ret = ff_rtmp_calc_digest(swfdata, swfsize, 0,
- "Genuine Adobe Flash Player 001", 30,
- swfhash)) < 0)
- goto fail;
- /* Set SWFVerification parameters. */
- av_opt_set_bin(rt, "rtmp_swfhash", swfhash, 32, 0);
- rt->swfsize = swfsize;
- fail:
- av_freep(&in_data);
- av_freep(&out_data);
- ffurl_close(stream);
- return ret;
- }
- /**
- * Perform handshake with the server by means of exchanging pseudorandom data
- * signed with HMAC-SHA2 digest.
- *
- * @return 0 if handshake succeeds, negative value otherwise
- */
- static int rtmp_handshake(URLContext *s, RTMPContext *rt)
- {
- AVLFG rnd;
- uint8_t tosend [RTMP_HANDSHAKE_PACKET_SIZE+1] = {
- 3, // unencrypted data
- 0, 0, 0, 0, // client uptime
- RTMP_CLIENT_VER1,
- RTMP_CLIENT_VER2,
- RTMP_CLIENT_VER3,
- RTMP_CLIENT_VER4,
- };
- uint8_t clientdata[RTMP_HANDSHAKE_PACKET_SIZE];
- uint8_t serverdata[RTMP_HANDSHAKE_PACKET_SIZE+1];
- int i;
- int server_pos, client_pos;
- uint8_t digest[32], signature[32];
- int ret, type = 0;
- av_log(s, AV_LOG_DEBUG, "Handshaking...\n");
- av_lfg_init(&rnd, 0xDEADC0DE);
- // generate handshake packet - 1536 bytes of pseudorandom data
- for (i = 9; i <= RTMP_HANDSHAKE_PACKET_SIZE; i++)
- tosend[i] = av_lfg_get(&rnd) >> 24;
- if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
- /* When the client wants to use RTMPE, we have to change the command
- * byte to 0x06 which means to use encrypted data and we have to set
- * the flash version to at least 9.0.115.0. */
- tosend[0] = 6;
- tosend[5] = 128;
- tosend[6] = 0;
- tosend[7] = 3;
- tosend[8] = 2;
- /* Initialize the Diffie-Hellmann context and generate the public key
- * to send to the server. */
- if ((ret = ff_rtmpe_gen_pub_key(rt->stream, tosend + 1)) < 0)
- return ret;
- }
- client_pos = rtmp_handshake_imprint_with_digest(tosend + 1, rt->encrypted);
- if (client_pos < 0)
- return client_pos;
- if ((ret = ffurl_write(rt->stream, tosend,
- RTMP_HANDSHAKE_PACKET_SIZE + 1)) < 0) {
- av_log(s, AV_LOG_ERROR, "Cannot write RTMP handshake request\n");
- return ret;
- }
- if ((ret = ffurl_read_complete(rt->stream, serverdata,
- RTMP_HANDSHAKE_PACKET_SIZE + 1)) < 0) {
- av_log(s, AV_LOG_ERROR, "Cannot read RTMP handshake response\n");
- return ret;
- }
- if ((ret = ffurl_read_complete(rt->stream, clientdata,
- RTMP_HANDSHAKE_PACKET_SIZE)) < 0) {
- av_log(s, AV_LOG_ERROR, "Cannot read RTMP handshake response\n");
- return ret;
- }
- av_log(s, AV_LOG_DEBUG, "Type answer %d\n", serverdata[0]);
- av_log(s, AV_LOG_DEBUG, "Server version %d.%d.%d.%d\n",
- serverdata[5], serverdata[6], serverdata[7], serverdata[8]);
- if (rt->is_input && serverdata[5] >= 3) {
- server_pos = rtmp_validate_digest(serverdata + 1, 772);
- if (server_pos < 0)
- return server_pos;
- if (!server_pos) {
- type = 1;
- server_pos = rtmp_validate_digest(serverdata + 1, 8);
- if (server_pos < 0)
- return server_pos;
- if (!server_pos) {
- av_log(s, AV_LOG_ERROR, "Server response validating failed\n");
- return AVERROR(EIO);
- }
- }
- /* Generate SWFVerification token (SHA256 HMAC hash of decompressed SWF,
- * key are the last 32 bytes of the server handshake. */
- if (rt->swfsize) {
- if ((ret = rtmp_calc_swf_verification(s, rt, serverdata + 1 +
- RTMP_HANDSHAKE_PACKET_SIZE - 32)) < 0)
- return ret;
- }
- ret = ff_rtmp_calc_digest(tosend + 1 + client_pos, 32, 0,
- rtmp_server_key, sizeof(rtmp_server_key),
- digest);
- if (ret < 0)
- return ret;
- ret = ff_rtmp_calc_digest(clientdata, RTMP_HANDSHAKE_PACKET_SIZE - 32,
- 0, digest, 32, signature);
- if (ret < 0)
- return ret;
- if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
- /* Compute the shared secret key sent by the server and initialize
- * the RC4 encryption. */
- if ((ret = ff_rtmpe_compute_secret_key(rt->stream, serverdata + 1,
- tosend + 1, type)) < 0)
- return ret;
- /* Encrypt the signature received by the server. */
- ff_rtmpe_encrypt_sig(rt->stream, signature, digest, serverdata[0]);
- }
- if (memcmp(signature, clientdata + RTMP_HANDSHAKE_PACKET_SIZE - 32, 32)) {
- av_log(s, AV_LOG_ERROR, "Signature mismatch\n");
- return AVERROR(EIO);
- }
- for (i = 0; i < RTMP_HANDSHAKE_PACKET_SIZE; i++)
- tosend[i] = av_lfg_get(&rnd) >> 24;
- ret = ff_rtmp_calc_digest(serverdata + 1 + server_pos, 32, 0,
- rtmp_player_key, sizeof(rtmp_player_key),
- digest);
- if (ret < 0)
- return ret;
- ret = ff_rtmp_calc_digest(tosend, RTMP_HANDSHAKE_PACKET_SIZE - 32, 0,
- digest, 32,
- tosend + RTMP_HANDSHAKE_PACKET_SIZE - 32);
- if (ret < 0)
- return ret;
- if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
- /* Encrypt the signature to be send to the server. */
- ff_rtmpe_encrypt_sig(rt->stream, tosend +
- RTMP_HANDSHAKE_PACKET_SIZE - 32, digest,
- serverdata[0]);
- }
- // write reply back to the server
- if ((ret = ffurl_write(rt->stream, tosend,
- RTMP_HANDSHAKE_PACKET_SIZE)) < 0)
- return ret;
- if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
- /* Set RC4 keys for encryption and update the keystreams. */
- if ((ret = ff_rtmpe_update_keystream(rt->stream)) < 0)
- return ret;
- }
- } else {
- if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
- /* Compute the shared secret key sent by the server and initialize
- * the RC4 encryption. */
- if ((ret = ff_rtmpe_compute_secret_key(rt->stream, serverdata + 1,
- tosend + 1, 1)) < 0)
- return ret;
- if (serverdata[0] == 9) {
- /* Encrypt the signature received by the server. */
- ff_rtmpe_encrypt_sig(rt->stream, signature, digest,
- serverdata[0]);
- }
- }
- if ((ret = ffurl_write(rt->stream, serverdata + 1,
- RTMP_HANDSHAKE_PACKET_SIZE)) < 0)
- return ret;
- if (CONFIG_FFRTMPCRYPT_PROTOCOL && rt->encrypted) {
- /* Set RC4 keys for encryption and update the keystreams. */
- if ((ret = ff_rtmpe_update_keystream(rt->stream)) < 0)
- return ret;
- }
- }
- return 0;
- }
- static int rtmp_receive_hs_packet(RTMPContext* rt, uint32_t *first_int,
- uint32_t *second_int, char *arraydata,
- int size)
- {
- int inoutsize;
- inoutsize = ffurl_read_complete(rt->stream, arraydata,
- RTMP_HANDSHAKE_PACKET_SIZE);
- if (inoutsize <= 0)
- return AVERROR(EIO);
- if (inoutsize != RTMP_HANDSHAKE_PACKET_SIZE) {
- av_log(rt, AV_LOG_ERROR, "Erroneous Message size %d"
- " not following standard\n", (int)inoutsize);
- return AVERROR(EINVAL);
- }
- *first_int = AV_RB32(arraydata);
- *second_int = AV_RB32(arraydata + 4);
- return 0;
- }
- static int rtmp_send_hs_packet(RTMPContext* rt, uint32_t first_int,
- uint32_t second_int, char *arraydata, int size)
- {
- int inoutsize;
- AV_WB32(arraydata, first_int);
- AV_WB32(arraydata + 4, first_int);
- inoutsize = ffurl_write(rt->stream, arraydata,
- RTMP_HANDSHAKE_PACKET_SIZE);
- if (inoutsize != RTMP_HANDSHAKE_PACKET_SIZE) {
- av_log(rt, AV_LOG_ERROR, "Unable to write answer\n");
- return AVERROR(EIO);
- }
- return 0;
- }
- /**
- * rtmp handshake server side
- */
- static int rtmp_server_handshake(URLContext *s, RTMPContext *rt)
- {
- uint8_t buffer[RTMP_HANDSHAKE_PACKET_SIZE];
- uint32_t hs_epoch;
- uint32_t hs_my_epoch;
- uint8_t hs_c1[RTMP_HANDSHAKE_PACKET_SIZE];
- uint8_t hs_s1[RTMP_HANDSHAKE_PACKET_SIZE];
- uint32_t zeroes;
- uint32_t temp = 0;
- int randomidx = 0;
- int inoutsize = 0;
- int ret;
- inoutsize = ffurl_read_complete(rt->stream, buffer, 1); // Receive C0
- if (inoutsize <= 0) {
- av_log(s, AV_LOG_ERROR, "Unable to read handshake\n");
- return AVERROR(EIO);
- }
- // Check Version
- if (buffer[0] != 3) {
- av_log(s, AV_LOG_ERROR, "RTMP protocol version mismatch\n");
- return AVERROR(EIO);
- }
- if (ffurl_write(rt->stream, buffer, 1) <= 0) { // Send S0
- av_log(s, AV_LOG_ERROR,
- "Unable to write answer - RTMP S0\n");
- return AVERROR(EIO);
- }
- /* Receive C1 */
- ret = rtmp_receive_hs_packet(rt, &hs_epoch, &zeroes, hs_c1,
- RTMP_HANDSHAKE_PACKET_SIZE);
- if (ret) {
- av_log(s, AV_LOG_ERROR, "RTMP Handshake C1 Error\n");
- return ret;
- }
- if (zeroes)
- av_log(s, AV_LOG_WARNING, "Erroneous C1 Message zero != 0\n");
- /* Send S1 */
- /* By now same epoch will be sent */
- hs_my_epoch = hs_epoch;
- /* Generate random */
- for (randomidx = 0; randomidx < (RTMP_HANDSHAKE_PACKET_SIZE);
- randomidx += 4)
- AV_WB32(hs_s1 + 8 + randomidx, av_get_random_seed());
- ret = rtmp_send_hs_packet(rt, hs_my_epoch, 0, hs_s1,
- RTMP_HANDSHAKE_PACKET_SIZE);
- if (ret) {
- av_log(s, AV_LOG_ERROR, "RTMP Handshake S1 Error\n");
- return ret;
- }
- /* Send S2 */
- ret = rtmp_send_hs_packet(rt, hs_epoch, 0, hs_c1,
- RTMP_HANDSHAKE_PACKET_SIZE);
- if (ret) {
- av_log(s, AV_LOG_ERROR, "RTMP Handshake S2 Error\n");
- return ret;
- }
- /* Receive C2 */
- ret = rtmp_receive_hs_packet(rt, &temp, &zeroes, buffer,
- RTMP_HANDSHAKE_PACKET_SIZE);
- if (ret) {
- av_log(s, AV_LOG_ERROR, "RTMP Handshake C2 Error\n");
- return ret;
- }
- if (temp != hs_my_epoch)
- av_log(s, AV_LOG_WARNING,
- "Erroneous C2 Message epoch does not match up with C1 epoch\n");
- if (memcmp(buffer + 8, hs_s1 + 8,
- RTMP_HANDSHAKE_PACKET_SIZE - 8))
- av_log(s, AV_LOG_WARNING,
- "Erroneous C2 Message random does not match up\n");
- return 0;
- }
- static int handle_chunk_size(URLContext *s, RTMPPacket *pkt)
- {
- RTMPContext *rt = s->priv_data;
- int ret;
- if (pkt->data_size < 4) {
- av_log(s, AV_LOG_ERROR,
- "Too short chunk size change packet (%d)\n",
- pkt->data_size);
- return AVERROR_INVALIDDATA;
- }
- if (!rt->is_input) {
- /* Send the same chunk size change packet back to the server,
- * setting the outgoing chunk size to the same as the incoming one. */
- if ((ret = ff_rtmp_packet_write(rt->stream, pkt, rt->out_chunk_size,
- rt->prev_pkt[1])) < 0)
- return ret;
- rt->out_chunk_size = AV_RB32(pkt->data);
- }
- rt->in_chunk_size = AV_RB32(pkt->data);
- if (rt->in_chunk_size <= 0) {
- av_log(s, AV_LOG_ERROR, "Incorrect chunk size %d\n",
- rt->in_chunk_size);
- return AVERROR_INVALIDDATA;
- }
- av_log(s, AV_LOG_DEBUG, "New incoming chunk size = %d\n",
- rt->in_chunk_size);
- return 0;
- }
- static int handle_ping(URLContext *s, RTMPPacket *pkt)
- {
- RTMPContext *rt = s->priv_data;
- int t, ret;
- if (pkt->data_size < 2) {
- av_log(s, AV_LOG_ERROR, "Too short ping packet (%d)\n",
- pkt->data_size);
- return AVERROR_INVALIDDATA;
- }
- t = AV_RB16(pkt->data);
- if (t == 6) {
- if ((ret = gen_pong(s, rt, pkt)) < 0)
- return ret;
- } else if (t == 26) {
- if (rt->swfsize) {
- if ((ret = gen_swf_verification(s, rt)) < 0)
- return ret;
- } else {
- av_log(s, AV_LOG_WARNING, "Ignoring SWFVerification request.\n");
- }
- }
- return 0;
- }
- static int handle_client_bw(URLContext *s, RTMPPacket *pkt)
- {
- RTMPContext *rt = s->priv_data;
- if (pkt->data_size < 4) {
- av_log(s, AV_LOG_ERROR,
- "Client bandwidth report packet is less than 4 bytes long (%d)\n",
- pkt->data_size);
- return AVERROR_INVALIDDATA;
- }
- rt->client_report_size = AV_RB32(pkt->data);
- if (rt->client_report_size <= 0) {
- av_log(s, AV_LOG_ERROR, "Incorrect client bandwidth %d\n",
- rt->client_report_size);
- return AVERROR_INVALIDDATA;
- }
- av_log(s, AV_LOG_DEBUG, "Client bandwidth = %d\n", rt->client_report_size);
- rt->client_report_size >>= 1;
- return 0;
- }
- static int handle_server_bw(URLContext *s, RTMPPacket *pkt)
- {
- RTMPContext *rt = s->priv_data;
- if (pkt->data_size < 4) {
- av_log(s, AV_LOG_ERROR,
- "Too short server bandwidth report packet (%d)\n",
- pkt->data_size);
- return AVERROR_INVALIDDATA;
- }
- rt->server_bw = AV_RB32(pkt->data);
- if (rt->server_bw <= 0) {
- av_log(s, AV_LOG_ERROR, "Incorrect server bandwidth %d\n",
- rt->server_bw);
- return AVERROR_INVALIDDATA;
- }
- av_log(s, AV_LOG_DEBUG, "Server bandwidth = %d\n", rt->server_bw);
- return 0;
- }
- static int handle_invoke_error(URLContext *s, RTMPPacket *pkt)
- {
- const uint8_t *data_end = pkt->data + pkt->data_size;
- char *tracked_method = NULL;
- int level = AV_LOG_ERROR;
- uint8_t tmpstr[256];
- int ret;
- if ((ret = find_tracked_method(s, pkt, 9, &tracked_method)) < 0)
- return ret;
- if (!ff_amf_get_field_value(pkt->data + 9, data_end,
- "description", tmpstr, sizeof(tmpstr))) {
- if (tracked_method && (!strcmp(tracked_method, "_checkbw") ||
- !strcmp(tracked_method, "releaseStream") ||
- !strcmp(tracked_method, "FCSubscribe") ||
- !strcmp(tracked_method, "FCPublish"))) {
- /* Gracefully ignore Adobe-specific historical artifact errors. */
- level = AV_LOG_WARNING;
- ret = 0;
- } else
- ret = -1;
- av_log(s, level, "Server error: %s\n", tmpstr);
- }
- av_free(tracked_method);
- return ret;
- }
- static int send_invoke_response(URLContext *s, RTMPPacket *pkt)
- {
- RTMPContext *rt = s->priv_data;
- double seqnum;
- char filename[64];
- char command[64];
- char statusmsg[128];
- int stringlen;
- char *pchar;
- const uint8_t *p = pkt->data;
- uint8_t *pp = NULL;
- RTMPPacket spkt = { 0 };
- GetByteContext gbc;
- int ret;
- bytestream2_init(&gbc, p, pkt->data_size);
- if (ff_amf_read_string(&gbc, command, sizeof(command),
- &stringlen)) {
- av_log(s, AV_LOG_ERROR, "Error in PT_INVOKE\n");
- return AVERROR_INVALIDDATA;
- }
- ret = ff_amf_read_number(&gbc, &seqnum);
- if (ret)
- return ret;
- ret = ff_amf_read_null(&gbc);
- if (ret)
- return ret;
- if (!strcmp(command, "FCPublish") ||
- !strcmp(command, "publish")) {
- ret = ff_amf_read_string(&gbc, filename,
- sizeof(filename), &stringlen);
- // check with url
- if (s->filename) {
- pchar = strrchr(s->filename, '/');
- if (!pchar) {
- av_log(s, AV_LOG_WARNING,
- "Unable to find / in url %s, bad format\n",
- s->filename);
- pchar = s->filename;
- }
- pchar++;
- if (strcmp(pchar, filename))
- av_log(s, AV_LOG_WARNING, "Unexpected stream %s, expecting"
- " %s\n", filename, pchar);
- }
- rt->state = STATE_RECEIVING;
- }
- if (!strcmp(command, "FCPublish")) {
- if ((ret = ff_rtmp_packet_create(&spkt, RTMP_SYSTEM_CHANNEL,
- RTMP_PT_INVOKE, 0,
- RTMP_PKTDATA_DEFAULT_SIZE)) < 0) {
- av_log(s, AV_LOG_ERROR, "Unable to create response packet\n");
- return ret;
- }
- pp = spkt.data;
- ff_amf_write_string(&pp, "onFCPublish");
- } else if (!strcmp(command, "publish")) {
- PutByteContext pbc;
- // Send Stream Begin 1
- if ((ret = ff_rtmp_packet_create(&spkt, RTMP_NETWORK_CHANNEL,
- RTMP_PT_PING, 0, 6)) < 0) {
- av_log(s, AV_LOG_ERROR, "Unable to create response packet\n");
- return ret;
- }
- pp = spkt.data;
- bytestream2_init_writer(&pbc, pp, spkt.data_size);
- bytestream2_put_be16(&pbc, 0); // 0 -> Stream Begin
- bytestream2_put_be32(&pbc, rt->nb_streamid);
- ret = ff_rtmp_packet_write(rt->stream, &spkt, rt->out_chunk_size,
- rt->prev_pkt[1]);
- ff_rtmp_packet_destroy(&spkt);
- if (ret < 0)
- return ret;
- // Send onStatus(NetStream.Publish.Start)
- if ((ret = ff_rtmp_packet_create(&spkt, RTMP_SYSTEM_CHANNEL,
- RTMP_PT_INVOKE, 0,
- RTMP_PKTDATA_DEFAULT_SIZE)) < 0) {
- av_log(s, AV_LOG_ERROR, "Unable to create response packet\n");
- return ret;
- }
- spkt.extra = pkt->extra;
- pp = spkt.data;
- ff_amf_write_string(&pp, "onStatus");
- ff_amf_write_number(&pp, 0);
- ff_amf_write_null(&pp);
- ff_amf_write_object_start(&pp);
- ff_amf_write_field_name(&pp, "level");
- ff_amf_write_string(&pp, "status");
- ff_amf_write_field_name(&pp, "code");
- ff_amf_write_string(&pp, "NetStream.Publish.Start");
- ff_amf_write_field_name(&pp, "description");
- snprintf(statusmsg, sizeof(statusmsg),
- "%s is now published", filename);
- ff_amf_write_string(&pp, statusmsg);
- ff_amf_write_field_name(&pp, "details");
- ff_amf_write_string(&pp, filename);
- ff_amf_write_field_name(&pp, "clientid");
- snprintf(statusmsg, sizeof(statusmsg), "%s", LIBAVFORMAT_IDENT);
- ff_amf_write_string(&pp, statusmsg);
- ff_amf_write_object_end(&pp);
- } else {
- if ((ret = ff_rtmp_packet_create(&spkt, RTMP_SYSTEM_CHANNEL,
- RTMP_PT_INVOKE, 0,
- RTMP_PKTDATA_DEFAULT_SIZE)) < 0) {
- av_log(s, AV_LOG_ERROR, "Unable to create response packet\n");
- return ret;
- }
- pp = spkt.data;
- ff_amf_write_string(&pp, "_result");
- ff_amf_write_number(&pp, seqnum);
- ff_amf_write_null(&pp);
- if (!strcmp(command, "createStream")) {
- rt->nb_streamid++;
- if (rt->nb_streamid == 0 || rt->nb_streamid == 2)
- rt->nb_streamid++; /* Values 0 and 2 are reserved */
- ff_amf_write_number(&pp, rt->nb_streamid);
- /* By now we don't control which streams are removed in
- * deleteStream. There is no stream creation control
- * if a client creates more than 2^32 - 2 streams. */
- }
- }
- spkt.data_size = pp - spkt.data;
- ret = ff_rtmp_packet_write(rt->stream, &spkt, rt->out_chunk_size,
- rt->prev_pkt[1]);
- ff_rtmp_packet_destroy(&spkt);
- return ret;
- }
- static int handle_invoke_result(URLContext *s, RTMPPacket *pkt)
- {
- RTMPContext *rt = s->priv_data;
- char *tracked_method = NULL;
- int ret = 0;
- if ((ret = find_tracked_method(s, pkt, 10, &tracked_method)) < 0)
- return ret;
- if (!tracked_method) {
- /* Ignore this reply when the current method is not tracked. */
- return ret;
- }
- if (!memcmp(tracked_method, "connect", 7)) {
- if (!rt->is_input) {
- if ((ret = gen_release_stream(s, rt)) < 0)
- goto fail;
- if ((ret = gen_fcpublish_stream(s, rt)) < 0)
- goto fail;
- } else {
- if ((ret = gen_server_bw(s, rt)) < 0)
- goto fail;
- }
- if ((ret = gen_create_stream(s, rt)) < 0)
- goto fail;
- if (rt->is_input) {
- /* Send the FCSubscribe command when the name of live
- * stream is defined by the user or if it's a live stream. */
- if (rt->subscribe) {
- if ((ret = gen_fcsubscribe_stream(s, rt, rt->subscribe)) < 0)
- goto fail;
- } else if (rt->live == -1) {
- if ((ret = gen_fcsubscribe_stream(s, rt, rt->playpath)) < 0)
- goto fail;
- }
- }
- } else if (!memcmp(tracked_method, "createStream", 12)) {
- //extract a number from the result
- if (pkt->data[10] || pkt->data[19] != 5 || pkt->data[20]) {
- av_log(s, AV_LOG_WARNING, "Unexpected reply on connect()\n");
- } else {
- rt->main_channel_id = av_int2double(AV_RB64(pkt->data + 21));
- }
- if (!rt->is_input) {
- if ((ret = gen_publish(s, rt)) < 0)
- goto fail;
- } else {
- if ((ret = gen_play(s, rt)) < 0)
- goto fail;
- if ((ret = gen_buffer_time(s, rt)) < 0)
- goto fail;
- }
- }
- fail:
- av_free(tracked_method);
- return ret;
- }
- static int handle_invoke_status(URLContext *s, RTMPPacket *pkt)
- {
- RTMPContext *rt = s->priv_data;
- const uint8_t *data_end = pkt->data + pkt->data_size;
- const uint8_t *ptr = pkt->data + 11;
- uint8_t tmpstr[256];
- int i, t;
- for (i = 0; i < 2; i++) {
- t = ff_amf_tag_size(ptr, data_end);
- if (t < 0)
- return 1;
- ptr += t;
- }
- t = ff_amf_get_field_value(ptr, data_end, "level", tmpstr, sizeof(tmpstr));
- if (!t && !strcmp(tmpstr, "error")) {
- if (!ff_amf_get_field_value(ptr, data_end,
- "description", tmpstr, sizeof(tmpstr)))
- av_log(s, AV_LOG_ERROR, "Server error: %s\n", tmpstr);
- return -1;
- }
- t = ff_amf_get_field_value(ptr, data_end, "code", tmpstr, sizeof(tmpstr));
- if (!t && !strcmp(tmpstr, "NetStream.Play.Start")) rt->state = STATE_PLAYING;
- if (!t && !strcmp(tmpstr, "NetStream.Play.Stop")) rt->state = STATE_STOPPED;
- if (!t && !strcmp(tmpstr, "NetStream.Play.UnpublishNotify")) rt->state = STATE_STOPPED;
- if (!t && !strcmp(tmpstr, "NetStream.Publish.Start")) rt->state = STATE_PUBLISHING;
- return 0;
- }
- static int handle_invoke(URLContext *s, RTMPPacket *pkt)
- {
- RTMPContext *rt = s->priv_data;
- int ret = 0;
- //TODO: check for the messages sent for wrong state?
- if (!memcmp(pkt->data, "\002\000\006_error", 9)) {
- if ((ret = handle_invoke_error(s, pkt)) < 0)
- return ret;
- } else if (!memcmp(pkt->data, "\002\000\007_result", 10)) {
- if ((ret = handle_invoke_result(s, pkt)) < 0)
- return ret;
- } else if (!memcmp(pkt->data, "\002\000\010onStatus", 11)) {
- if ((ret = handle_invoke_status(s, pkt)) < 0)
- return ret;
- } else if (!memcmp(pkt->data, "\002\000\010onBWDone", 11)) {
- if ((ret = gen_check_bw(s, rt)) < 0)
- return ret;
- } else if (!memcmp(pkt->data, "\002\000\015releaseStream", 16) ||
- !memcmp(pkt->data, "\002\000\011FCPublish", 12) ||
- !memcmp(pkt->data, "\002\000\007publish", 10) ||
- !memcmp(pkt->data, "\002\000\010_checkbw", 11) ||
- !memcmp(pkt->data, "\002\000\014createStream", 15)) {
- if ((ret = send_invoke_response(s, pkt)) < 0)
- return ret;
- }
- return ret;
- }
- static int handle_notify(URLContext *s, RTMPPacket *pkt) {
- RTMPContext *rt = s->priv_data;
- const uint8_t *p = NULL;
- uint8_t *cp = NULL;
- uint8_t commandbuffer[64];
- char statusmsg[128];
- int stringlen;
- GetByteContext gbc;
- PutByteContext pbc;
- uint32_t ts;
- int old_flv_size;
- const uint8_t *datatowrite;
- unsigned datatowritelength;
- p = pkt->data;
- bytestream2_init(&gbc, p, pkt->data_size);
- if (ff_amf_read_string(&gbc, commandbuffer, sizeof(commandbuffer),
- &stringlen))
- return AVERROR_INVALIDDATA;
- if (!strcmp(commandbuffer, "@setDataFrame")) {
- datatowrite = gbc.buffer;
- datatowritelength = bytestream2_get_bytes_left(&gbc);
- if (ff_amf_read_string(&gbc, statusmsg,
- sizeof(statusmsg), &stringlen))
- return AVERROR_INVALIDDATA;
- if (strcmp(statusmsg, "onMetaData")) {
- av_log(s, AV_LOG_INFO, "Expecting onMetadata but got %s\n",
- statusmsg);
- return 0;
- }
- /* Provide ECMAArray to flv */
- ts = pkt->timestamp;
- // generate packet header and put data into buffer for FLV demuxer
- if (rt->flv_off < rt->flv_size) {
- old_flv_size = rt->flv_size;
- rt->flv_size += datatowritelength + 15;
- } else {
- old_flv_size = 0;
- rt->flv_size = datatowritelength + 15;
- rt->flv_off = 0;
- }
- cp = av_realloc(rt->flv_data, rt->flv_size);
- if (!cp)
- return AVERROR(ENOMEM);
- rt->flv_data = cp;
- bytestream2_init_writer(&pbc, cp, rt->flv_size);
- bytestream2_skip_p(&pbc, old_flv_size);
- bytestream2_put_byte(&pbc, pkt->type);
- bytestream2_put_be24(&pbc, datatowritelength);
- bytestream2_put_be24(&pbc, ts);
- bytestream2_put_byte(&pbc, ts >> 24);
- bytestream2_put_be24(&pbc, 0);
- bytestream2_put_buffer(&pbc, datatowrite, datatowritelength);
- bytestream2_put_be32(&pbc, 0);
- }
- return 0;
- }
- /**
- * Parse received packet and possibly perform some action depending on
- * the packet contents.
- * @return 0 for no errors, negative values for serious errors which prevent
- * further communications, positive values for uncritical errors
- */
- static int rtmp_parse_result(URLContext *s, RTMPContext *rt, RTMPPacket *pkt)
- {
- int ret;
- #ifdef DEBUG
- ff_rtmp_packet_dump(s, pkt);
- #endif
- switch (pkt->type) {
- case RTMP_PT_BYTES_READ:
- av_dlog(s, "received bytes read report\n");
- break;
- case RTMP_PT_CHUNK_SIZE:
- if ((ret = handle_chunk_size(s, pkt)) < 0)
- return ret;
- break;
- case RTMP_PT_PING:
- if ((ret = handle_ping(s, pkt)) < 0)
- return ret;
- break;
- case RTMP_PT_CLIENT_BW:
- if ((ret = handle_client_bw(s, pkt)) < 0)
- return ret;
- break;
- case RTMP_PT_SERVER_BW:
- if ((ret = handle_server_bw(s, pkt)) < 0)
- return ret;
- break;
- case RTMP_PT_INVOKE:
- if ((ret = handle_invoke(s, pkt)) < 0)
- return ret;
- break;
- case RTMP_PT_VIDEO:
- case RTMP_PT_AUDIO:
- case RTMP_PT_METADATA:
- case RTMP_PT_NOTIFY:
- /* Audio, Video and Metadata packets are parsed in get_packet() */
- break;
- default:
- av_log(s, AV_LOG_VERBOSE, "Unknown packet type received 0x%02X\n", pkt->type);
- break;
- }
- return 0;
- }
- /**
- * Interact with the server by receiving and sending RTMP packets until
- * there is some significant data (media data or expected status notification).
- *
- * @param s reading context
- * @param for_header non-zero value tells function to work until it
- * gets notification from the server that playing has been started,
- * otherwise function will work until some media data is received (or
- * an error happens)
- * @return 0 for successful operation, negative value in case of error
- */
- static int get_packet(URLContext *s, int for_header)
- {
- RTMPContext *rt = s->priv_data;
- int ret;
- uint8_t *p;
- const uint8_t *next;
- uint32_t data_size;
- uint32_t ts, cts, pts=0;
- if (rt->state == STATE_STOPPED)
- return AVERROR_EOF;
- for (;;) {
- RTMPPacket rpkt = { 0 };
- if ((ret = ff_rtmp_packet_read(rt->stream, &rpkt,
- rt->in_chunk_size, rt->prev_pkt[0])) <= 0) {
- if (ret == 0) {
- return AVERROR(EAGAIN);
- } else {
- return AVERROR(EIO);
- }
- }
- rt->bytes_read += ret;
- if (rt->bytes_read - rt->last_bytes_read > rt->client_report_size) {
- av_log(s, AV_LOG_DEBUG, "Sending bytes read report\n");
- if ((ret = gen_bytes_read(s, rt, rpkt.timestamp + 1)) < 0)
- return ret;
- rt->last_bytes_read = rt->bytes_read;
- }
- ret = rtmp_parse_result(s, rt, &rpkt);
- if (ret < 0) {//serious error in current packet
- ff_rtmp_packet_destroy(&rpkt);
- return ret;
- }
- if (rt->state == STATE_STOPPED) {
- ff_rtmp_packet_destroy(&rpkt);
- return AVERROR_EOF;
- }
- if (for_header && (rt->state == STATE_PLAYING ||
- rt->state == STATE_PUBLISHING ||
- rt->state == STATE_RECEIVING)) {
- ff_rtmp_packet_destroy(&rpkt);
- return 0;
- }
- if (!rpkt.data_size || !rt->is_input) {
- ff_rtmp_packet_destroy(&rpkt);
- continue;
- }
- if (rpkt.type == RTMP_PT_VIDEO || rpkt.type == RTMP_PT_AUDIO ||
- (rpkt.type == RTMP_PT_NOTIFY && !memcmp("\002\000\012onMetaData", rpkt.data, 13))) {
- ts = rpkt.timestamp;
- // generate packet header and put data into buffer for FLV demuxer
- rt->flv_off = 0;
- rt->flv_size = rpkt.data_size + 15;
- rt->flv_data = p = av_realloc(rt->flv_data, rt->flv_size);
- bytestream_put_byte(&p, rpkt.type);
- bytestream_put_be24(&p, rpkt.data_size);
- bytestream_put_be24(&p, ts);
- bytestream_put_byte(&p, ts >> 24);
- bytestream_put_be24(&p, 0);
- bytestream_put_buffer(&p, rpkt.data, rpkt.data_size);
- bytestream_put_be32(&p, 0);
- ff_rtmp_packet_destroy(&rpkt);
- return 0;
- } else if (rpkt.type == RTMP_PT_NOTIFY) {
- ret = handle_notify(s, &rpkt);
- ff_rtmp_packet_destroy(&rpkt);
- if (ret) {
- av_log(s, AV_LOG_ERROR, "Handle notify error\n");
- return ret;
- }
- return 0;
- } else if (rpkt.type == RTMP_PT_METADATA) {
- // we got raw FLV data, make it available for FLV demuxer
- rt->flv_off = 0;
- rt->flv_size = rpkt.data_size;
- rt->flv_data = av_realloc(rt->flv_data, rt->flv_size);
- /* rewrite timestamps */
- next = rpkt.data;
- ts = rpkt.timestamp;
- while (next - rpkt.data < rpkt.data_size - 11) {
- next++;
- data_size = bytestream_get_be24(&next);
- p=next;
- cts = bytestream_get_be24(&next);
- cts |= bytestream_get_byte(&next) << 24;
- if (pts==0)
- pts=cts;
- ts += cts - pts;
- pts = cts;
- bytestream_put_be24(&p, ts);
- bytestream_put_byte(&p, ts >> 24);
- next += data_size + 3 + 4;
- }
- memcpy(rt->flv_data, rpkt.data, rpkt.data_size);
- ff_rtmp_packet_destroy(&rpkt);
- return 0;
- }
- ff_rtmp_packet_destroy(&rpkt);
- }
- }
- static int rtmp_close(URLContext *h)
- {
- RTMPContext *rt = h->priv_data;
- int ret = 0;
- if (!rt->is_input) {
- rt->flv_data = NULL;
- if (rt->out_pkt.data_size)
- ff_rtmp_packet_destroy(&rt->out_pkt);
- if (rt->state > STATE_FCPUBLISH)
- ret = gen_fcunpublish_stream(h, rt);
- }
- if (rt->state > STATE_HANDSHAKED)
- ret = gen_delete_stream(h, rt);
- free_tracked_methods(rt);
- av_freep(&rt->flv_data);
- ffurl_close(rt->stream);
- return ret;
- }
- /**
- * Open RTMP connection and verify that the stream can be played.
- *
- * URL syntax: rtmp://server[:port][/app][/playpath]
- * where 'app' is first one or two directories in the path
- * (e.g. /ondemand/, /flash/live/, etc.)
- * and 'playpath' is a file name (the rest of the path,
- * may be prefixed with "mp4:")
- */
- static int rtmp_open(URLContext *s, const char *uri, int flags)
- {
- RTMPContext *rt = s->priv_data;
- char proto[8], hostname[256], path[1024], *fname;
- char *old_app;
- uint8_t buf[2048];
- int port;
- AVDictionary *opts = NULL;
- int ret;
- if (rt->listen_timeout > 0)
- rt->listen = 1;
- rt->is_input = !(flags & AVIO_FLAG_WRITE);
- av_url_split(proto, sizeof(proto), NULL, 0, hostname, sizeof(hostname), &port,
- path, sizeof(path), s->filename);
- if (rt->listen && strcmp(proto, "rtmp")) {
- av_log(s, AV_LOG_ERROR, "rtmp_listen not available for %s\n",
- proto);
- return AVERROR(EINVAL);
- }
- if (!strcmp(proto, "rtmpt") || !strcmp(proto, "rtmpts")) {
- if (!strcmp(proto, "rtmpts"))
- av_dict_set(&opts, "ffrtmphttp_tls", "1", 1);
- /* open the http tunneling connection */
- ff_url_join(buf, sizeof(buf), "ffrtmphttp", NULL, hostname, port, NULL);
- } else if (!strcmp(proto, "rtmps")) {
- /* open the tls connection */
- if (port < 0)
- port = RTMPS_DEFAULT_PORT;
- ff_url_join(buf, sizeof(buf), "tls", NULL, hostname, port, NULL);
- } else if (!strcmp(proto, "rtmpe") || (!strcmp(proto, "rtmpte"))) {
- if (!strcmp(proto, "rtmpte"))
- av_dict_set(&opts, "ffrtmpcrypt_tunneling", "1", 1);
- /* open the encrypted connection */
- ff_url_join(buf, sizeof(buf), "ffrtmpcrypt", NULL, hostname, port, NULL);
- rt->encrypted = 1;
- } else {
- /* open the tcp connection */
- if (port < 0)
- port = RTMP_DEFAULT_PORT;
- if (rt->listen)
- ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port,
- "?listen&listen_timeout=%d",
- rt->listen_timeout * 1000);
- else
- ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port, NULL);
- }
- if ((ret = ffurl_open(&rt->stream, buf, AVIO_FLAG_READ_WRITE,
- &s->interrupt_callback, &opts)) < 0) {
- av_log(s , AV_LOG_ERROR, "Cannot open connection %s\n", buf);
- goto fail;
- }
- if (rt->swfverify) {
- if ((ret = rtmp_calc_swfhash(s)) < 0)
- goto fail;
- }
- rt->state = STATE_START;
- if (!rt->listen && (ret = rtmp_handshake(s, rt)) < 0)
- goto fail;
- if (rt->listen && (ret = rtmp_server_handshake(s, rt)) < 0)
- goto fail;
- rt->out_chunk_size = 128;
- rt->in_chunk_size = 128; // Probably overwritten later
- rt->state = STATE_HANDSHAKED;
- // Keep the application name when it has been defined by the user.
- old_app = rt->app;
- rt->app = av_malloc(APP_MAX_LENGTH);
- if (!rt->app) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- //extract "app" part from path
- if (!strncmp(path, "/ondemand/", 10)) {
- fname = path + 10;
- memcpy(rt->app, "ondemand", 9);
- } else {
- char *next = *path ? path + 1 : path;
- char *p = strchr(next, '/');
- if (!p) {
- fname = next;
- rt->app[0] = '\0';
- } else {
- // make sure we do not mismatch a playpath for an application instance
- char *c = strchr(p + 1, ':');
- fname = strchr(p + 1, '/');
- if (!fname || (c && c < fname)) {
- fname = p + 1;
- av_strlcpy(rt->app, path + 1, FFMIN(p - path, APP_MAX_LENGTH));
- } else {
- fname++;
- av_strlcpy(rt->app, path + 1, FFMIN(fname - path - 1, APP_MAX_LENGTH));
- }
- }
- }
- if (old_app) {
- // The name of application has been defined by the user, override it.
- if (strlen(old_app) >= APP_MAX_LENGTH) {
- ret = AVERROR(EINVAL);
- goto fail;
- }
- av_free(rt->app);
- rt->app = old_app;
- }
- if (!rt->playpath) {
- int len = strlen(fname);
- rt->playpath = av_malloc(PLAYPATH_MAX_LENGTH);
- if (!rt->playpath) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- if (!strchr(fname, ':') && len >= 4 &&
- (!strcmp(fname + len - 4, ".f4v") ||
- !strcmp(fname + len - 4, ".mp4"))) {
- memcpy(rt->playpath, "mp4:", 5);
- } else if (len >= 4 && !strcmp(fname + len - 4, ".flv")) {
- fname[len - 4] = '\0';
- } else {
- rt->playpath[0] = 0;
- }
- strncat(rt->playpath, fname, PLAYPATH_MAX_LENGTH - 5);
- }
- if (!rt->tcurl) {
- rt->tcurl = av_malloc(TCURL_MAX_LENGTH);
- if (!rt->tcurl) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- ff_url_join(rt->tcurl, TCURL_MAX_LENGTH, proto, NULL, hostname,
- port, "/%s", rt->app);
- }
- if (!rt->flashver) {
- rt->flashver = av_malloc(FLASHVER_MAX_LENGTH);
- if (!rt->flashver) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- if (rt->is_input) {
- snprintf(rt->flashver, FLASHVER_MAX_LENGTH, "%s %d,%d,%d,%d",
- RTMP_CLIENT_PLATFORM, RTMP_CLIENT_VER1, RTMP_CLIENT_VER2,
- RTMP_CLIENT_VER3, RTMP_CLIENT_VER4);
- } else {
- snprintf(rt->flashver, FLASHVER_MAX_LENGTH,
- "FMLE/3.0 (compatible; %s)", LIBAVFORMAT_IDENT);
- }
- }
- rt->client_report_size = 1048576;
- rt->bytes_read = 0;
- rt->last_bytes_read = 0;
- rt->server_bw = 2500000;
- av_log(s, AV_LOG_DEBUG, "Proto = %s, path = %s, app = %s, fname = %s\n",
- proto, path, rt->app, rt->playpath);
- if (!rt->listen) {
- if ((ret = gen_connect(s, rt)) < 0)
- goto fail;
- } else {
- if (read_connect(s, s->priv_data) < 0)
- goto fail;
- rt->is_input = 1;
- }
- do {
- ret = get_packet(s, 1);
- } while (ret == EAGAIN);
- if (ret < 0)
- goto fail;
- if (rt->is_input) {
- // generate FLV header for demuxer
- rt->flv_size = 13;
- rt->flv_data = av_realloc(rt->flv_data, rt->flv_size);
- rt->flv_off = 0;
- memcpy(rt->flv_data, "FLV\1\5\0\0\0\011\0\0\0\0", rt->flv_size);
- } else {
- rt->flv_size = 0;
- rt->flv_data = NULL;
- rt->flv_off = 0;
- rt->skip_bytes = 13;
- }
- s->max_packet_size = rt->stream->max_packet_size;
- s->is_streamed = 1;
- return 0;
- fail:
- av_dict_free(&opts);
- rtmp_close(s);
- return ret;
- }
- static int rtmp_read(URLContext *s, uint8_t *buf, int size)
- {
- RTMPContext *rt = s->priv_data;
- int orig_size = size;
- int ret;
- while (size > 0) {
- int data_left = rt->flv_size - rt->flv_off;
- if (data_left >= size) {
- memcpy(buf, rt->flv_data + rt->flv_off, size);
- rt->flv_off += size;
- return orig_size;
- }
- if (data_left > 0) {
- memcpy(buf, rt->flv_data + rt->flv_off, data_left);
- buf += data_left;
- size -= data_left;
- rt->flv_off = rt->flv_size;
- return data_left;
- }
- if ((ret = get_packet(s, 0)) < 0)
- return ret;
- }
- return orig_size;
- }
- static int rtmp_write(URLContext *s, const uint8_t *buf, int size)
- {
- RTMPContext *rt = s->priv_data;
- int size_temp = size;
- int pktsize, pkttype;
- uint32_t ts;
- const uint8_t *buf_temp = buf;
- uint8_t c;
- int ret;
- do {
- if (rt->skip_bytes) {
- int skip = FFMIN(rt->skip_bytes, size_temp);
- buf_temp += skip;
- size_temp -= skip;
- rt->skip_bytes -= skip;
- continue;
- }
- if (rt->flv_header_bytes < 11) {
- const uint8_t *header = rt->flv_header;
- int copy = FFMIN(11 - rt->flv_header_bytes, size_temp);
- bytestream_get_buffer(&buf_temp, rt->flv_header + rt->flv_header_bytes, copy);
- rt->flv_header_bytes += copy;
- size_temp -= copy;
- if (rt->flv_header_bytes < 11)
- break;
- pkttype = bytestream_get_byte(&header);
- pktsize = bytestream_get_be24(&header);
- ts = bytestream_get_be24(&header);
- ts |= bytestream_get_byte(&header) << 24;
- bytestream_get_be24(&header);
- rt->flv_size = pktsize;
- //force 12bytes header
- if (((pkttype == RTMP_PT_VIDEO || pkttype == RTMP_PT_AUDIO) && ts == 0) ||
- pkttype == RTMP_PT_NOTIFY) {
- if (pkttype == RTMP_PT_NOTIFY)
- pktsize += 16;
- rt->prev_pkt[1][RTMP_SOURCE_CHANNEL].channel_id = 0;
- }
- //this can be a big packet, it's better to send it right here
- if ((ret = ff_rtmp_packet_create(&rt->out_pkt, RTMP_SOURCE_CHANNEL,
- pkttype, ts, pktsize)) < 0)
- return ret;
- rt->out_pkt.extra = rt->main_channel_id;
- rt->flv_data = rt->out_pkt.data;
- if (pkttype == RTMP_PT_NOTIFY)
- ff_amf_write_string(&rt->flv_data, "@setDataFrame");
- }
- if (rt->flv_size - rt->flv_off > size_temp) {
- bytestream_get_buffer(&buf_temp, rt->flv_data + rt->flv_off, size_temp);
- rt->flv_off += size_temp;
- size_temp = 0;
- } else {
- bytestream_get_buffer(&buf_temp, rt->flv_data + rt->flv_off, rt->flv_size - rt->flv_off);
- size_temp -= rt->flv_size - rt->flv_off;
- rt->flv_off += rt->flv_size - rt->flv_off;
- }
- if (rt->flv_off == rt->flv_size) {
- rt->skip_bytes = 4;
- if ((ret = rtmp_send_packet(rt, &rt->out_pkt, 0)) < 0)
- return ret;
- rt->flv_size = 0;
- rt->flv_off = 0;
- rt->flv_header_bytes = 0;
- rt->flv_nb_packets++;
- }
- } while (buf_temp - buf < size);
- if (rt->flv_nb_packets < rt->flush_interval)
- return size;
- rt->flv_nb_packets = 0;
- /* set stream into nonblocking mode */
- rt->stream->flags |= AVIO_FLAG_NONBLOCK;
- /* try to read one byte from the stream */
- ret = ffurl_read(rt->stream, &c, 1);
- /* switch the stream back into blocking mode */
- rt->stream->flags &= ~AVIO_FLAG_NONBLOCK;
- if (ret == AVERROR(EAGAIN)) {
- /* no incoming data to handle */
- return size;
- } else if (ret < 0) {
- return ret;
- } else if (ret == 1) {
- RTMPPacket rpkt = { 0 };
- if ((ret = ff_rtmp_packet_read_internal(rt->stream, &rpkt,
- rt->in_chunk_size,
- rt->prev_pkt[0], c)) <= 0)
- return ret;
- if ((ret = rtmp_parse_result(s, rt, &rpkt)) < 0)
- return ret;
- ff_rtmp_packet_destroy(&rpkt);
- }
- return size;
- }
- #define OFFSET(x) offsetof(RTMPContext, x)
- #define DEC AV_OPT_FLAG_DECODING_PARAM
- #define ENC AV_OPT_FLAG_ENCODING_PARAM
- static const AVOption rtmp_options[] = {
- {"rtmp_app", "Name of application to connect to on the RTMP server", OFFSET(app), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
- {"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},
- {"rtmp_conn", "Append arbitrary AMF data to the Connect message", OFFSET(conn), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
- {"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},
- {"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},
- {"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"},
- {"any", "both", 0, AV_OPT_TYPE_CONST, {.i64 = -2}, 0, 0, DEC, "rtmp_live"},
- {"live", "live stream", 0, AV_OPT_TYPE_CONST, {.i64 = -1}, 0, 0, DEC, "rtmp_live"},
- {"recorded", "recorded stream", 0, AV_OPT_TYPE_CONST, {.i64 = 0}, 0, 0, DEC, "rtmp_live"},
- {"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},
- {"rtmp_playpath", "Stream identifier to play or to publish", OFFSET(playpath), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
- {"rtmp_subscribe", "Name of live stream to subscribe to. Defaults to rtmp_playpath.", OFFSET(subscribe), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC},
- {"rtmp_swfhash", "SHA256 hash of the decompressed SWF file (32 bytes).", OFFSET(swfhash), AV_OPT_TYPE_BINARY, .flags = DEC},
- {"rtmp_swfsize", "Size of the decompressed SWF file, required for SWFVerification.", OFFSET(swfsize), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, DEC},
- {"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},
- {"rtmp_swfverify", "URL to player swf file, compute hash/size automatically.", OFFSET(swfverify), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC},
- {"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},
- {"rtmp_listen", "Listen for incoming rtmp connections", OFFSET(listen), AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, DEC, "rtmp_listen" },
- {"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" },
- { NULL },
- };
- #define RTMP_PROTOCOL(flavor) \
- static const AVClass flavor##_class = { \
- .class_name = #flavor, \
- .item_name = av_default_item_name, \
- .option = rtmp_options, \
- .version = LIBAVUTIL_VERSION_INT, \
- }; \
- \
- URLProtocol ff_##flavor##_protocol = { \
- .name = #flavor, \
- .url_open = rtmp_open, \
- .url_read = rtmp_read, \
- .url_write = rtmp_write, \
- .url_close = rtmp_close, \
- .priv_data_size = sizeof(RTMPContext), \
- .flags = URL_PROTOCOL_FLAG_NETWORK, \
- .priv_data_class= &flavor##_class, \
- };
- RTMP_PROTOCOL(rtmp)
- RTMP_PROTOCOL(rtmpe)
- RTMP_PROTOCOL(rtmps)
- RTMP_PROTOCOL(rtmpt)
- RTMP_PROTOCOL(rtmpte)
- RTMP_PROTOCOL(rtmpts)
|