rtmppkt.h

/*
 * RTMP packet utilities
 * Copyright (c) 2009 Konstantin 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
 */

#ifndef AVFORMAT_RTMPPKT_H
#define AVFORMAT_RTMPPKT_H

#include "libavcodec/bytestream.h"
#include "avformat.h"
#include "url.h"

#define RTMP_CHANNELS 65599

enum RTMPChannel {
    RTMP_NETWORK_CHANNEL = 2,   
    RTMP_SYSTEM_CHANNEL,        
    RTMP_AUDIO_CHANNEL,         
    RTMP_VIDEO_CHANNEL   = 6,   
    RTMP_SOURCE_CHANNEL  = 8,   
};

typedef enum RTMPPacketType {
    RTMP_PT_CHUNK_SIZE   =  1,  
    RTMP_PT_BYTES_READ   =  3,  
    RTMP_PT_USER_CONTROL,       
    RTMP_PT_WINDOW_ACK_SIZE,    
    RTMP_PT_SET_PEER_BW,        
    RTMP_PT_AUDIO        =  8,  
    RTMP_PT_VIDEO,              
    RTMP_PT_FLEX_STREAM  = 15,  
    RTMP_PT_FLEX_OBJECT,        
    RTMP_PT_FLEX_MESSAGE,       
    RTMP_PT_NOTIFY,             
    RTMP_PT_SHARED_OBJ,         
    RTMP_PT_INVOKE,             
    RTMP_PT_METADATA     = 22,  
} RTMPPacketType;

enum RTMPPacketSize {
    RTMP_PS_TWELVEBYTES = 0, 
    RTMP_PS_EIGHTBYTES,      
    RTMP_PS_FOURBYTES,       
    RTMP_PS_ONEBYTE          
};

typedef struct RTMPPacket {
    int            channel_id; 
    RTMPPacketType type;       
    uint32_t       timestamp;  
    uint32_t       ts_field;   
    uint32_t       extra;      
    uint8_t        *data;      
    int            size;       
    int            offset;     
    int            read;       
} RTMPPacket;

int ff_rtmp_packet_create(RTMPPacket *pkt, int channel_id, RTMPPacketType type,
                          int timestamp, int size);

void ff_rtmp_packet_destroy(RTMPPacket *pkt);

int ff_rtmp_packet_read(URLContext *h, RTMPPacket *p,
                        int chunk_size, RTMPPacket **prev_pkt,
                        int *nb_prev_pkt);
int ff_rtmp_packet_read_internal(URLContext *h, RTMPPacket *p, int chunk_size,
                                 RTMPPacket **prev_pkt, int *nb_prev_pkt,
                                 uint8_t c);

int ff_rtmp_packet_write(URLContext *h, RTMPPacket *p,
                         int chunk_size, RTMPPacket **prev_pkt,
                         int *nb_prev_pkt);

void ff_rtmp_packet_dump(void *ctx, RTMPPacket *p);

int ff_rtmp_check_alloc_array(RTMPPacket **prev_pkt, int *nb_prev_pkt,
                              int channel);

int ff_amf_tag_size(const uint8_t *data, const uint8_t *data_end);

int ff_amf_get_field_value(const uint8_t *data, const uint8_t *data_end,
                           const uint8_t *name, uint8_t *dst, int dst_size);

void ff_amf_write_bool(uint8_t **dst, int val);

void ff_amf_write_number(uint8_t **dst, double num);

void ff_amf_write_string(uint8_t **dst, const char *str);

void ff_amf_write_string2(uint8_t **dst, const char *str1, const char *str2);

void ff_amf_write_null(uint8_t **dst);

void ff_amf_write_object_start(uint8_t **dst);

void ff_amf_write_field_name(uint8_t **dst, const char *str);

void ff_amf_write_object_end(uint8_t **dst);

int ff_amf_read_bool(GetByteContext *gbc, int *val);

int ff_amf_read_number(GetByteContext *gbc, double *val);

int ff_amf_get_string(GetByteContext *bc, uint8_t *str,
                      int strsize, int *length);

int ff_amf_read_string(GetByteContext *gbc, uint8_t *str,
                       int strsize, int *length);

int ff_amf_read_null(GetByteContext *gbc);

int ff_amf_match_string(const uint8_t *data, int size, const char *str);
 // AMF funcs

#endif /* AVFORMAT_RTMPPKT_H */