ZLMediaKit/src/Rtmp/RtmpProtocol.h

/*
 * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
 *
 * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
 *
 * Use of this source code is governed by MIT license that can be found in the
 * LICENSE file in the root of the source tree. All contributing project authors
 * may be found in the AUTHORS file in the root of the source tree.
 */

#ifndef SRC_RTMP_RTMPPROTOCOL_H_
#define SRC_RTMP_RTMPPROTOCOL_H_

#include <memory>
#include <string>
#include <functional>
#include "amf.h"
#include "Rtmp.h"
#include "Util/util.h"
#include "Util/logger.h"
#include "Util/TimeTicker.h"
#include "Network/Socket.h"
#include "Util/ResourcePool.h"

using namespace std;
using namespace toolkit;

namespace mediakit {

class RtmpProtocol {
public:
    RtmpProtocol();
    virtual ~RtmpProtocol();
    //作为客户端发送c0c1，等待s0s1s2并且回调
    void startClientSession(const function<void()> &cb);
    void onParseRtmp(const char *pcRawData,int iSize);
    void reset();
protected:
    virtual void onSendRawData(const Buffer::Ptr &buffer) = 0;
    virtual void onRtmpChunk(RtmpPacket &chunkData) = 0;
    virtual void onStreamBegin(uint32_t ui32StreamId){
        _ui32StreamId = ui32StreamId;
    }
    virtual void onStreamEof(uint32_t ui32StreamId){};
    virtual void onStreamDry(uint32_t ui32StreamId){};
protected:
    void sendAcknowledgement(uint32_t ui32Size);
    void sendAcknowledgementSize(uint32_t ui32Size);
    void sendPeerBandwidth(uint32_t ui32Size);
    void sendChunkSize(uint32_t ui32Size);
    void sendPingRequest(uint32_t ui32TimeStamp = ::time(NULL));
    void sendPingResponse(uint32_t ui32TimeStamp = ::time(NULL));
    void sendSetBufferLength(uint32_t ui32StreamId, uint32_t ui32Length);
    void sendUserControl(uint16_t ui16EventType, uint32_t ui32EventData);
    void sendUserControl(uint16_t ui16EventType, const string &strEventData);

    void sendInvoke(const string &strCmd, const AMFValue &val);
    void sendRequest(int iCmd, const string &str);
    void sendResponse(int iType, const string &str);
    void sendRtmp(uint8_t ui8Type, uint32_t ui32StreamId, const std::string &strBuf, uint32_t ui32TimeStamp, int iChunkID);
    void sendRtmp(uint8_t ui8Type, uint32_t ui32StreamId, const Buffer::Ptr &buffer, uint32_t ui32TimeStamp, int iChunkID);
protected:
    int _iReqID = 0;
    uint32_t _ui32StreamId = STREAM_CONTROL;
    int _iNowStreamID = 0;
    int _iNowChunkID = 0;
    bool _bDataStarted = false;
    inline BufferRaw::Ptr obtainBuffer();
    inline BufferRaw::Ptr obtainBuffer(const void *data, int len);
    //ResourcePool<BufferRaw,MAX_SEND_PKT> _bufferPool;
private:
    void handle_S0S1S2(const function<void()> &cb);
    void handle_C0C1();
    void handle_C1_simple();
#ifdef ENABLE_OPENSSL
    void handle_C1_complex();
    string get_C1_digest(const uint8_t *ptr,char **digestPos);
    string get_C1_key(const uint8_t *ptr);
    void check_C1_Digest(const string &digest,const string &data);
    void send_complex_S0S1S2(int schemeType,const string &digest);
#endif //ENABLE_OPENSSL

    void handle_C2();
    void handle_rtmp();
    void handle_rtmpChunk(RtmpPacket &chunkData);

private:
    ////////////ChunkSize////////////
    size_t _iChunkLenIn = DEFAULT_CHUNK_LEN;
    size_t _iChunkLenOut = DEFAULT_CHUNK_LEN;
    ////////////Acknowledgement////////////
    uint32_t _ui32ByteSent = 0;
    uint32_t _ui32LastSent = 0;
    uint32_t _ui32WinSize = 0;
    ///////////PeerBandwidth///////////
    uint32_t _ui32Bandwidth = 2500000;
    uint8_t _ui8LimitType = 2;
    ////////////Chunk////////////
    unordered_map<int, RtmpPacket> _mapChunkData;
    //////////Rtmp parser//////////
    string _strRcvBuf;
    function<void()> _nextHandle;
};

} /* namespace mediakit */

#endif /* SRC_RTMP_RTMPPROTOCOL_H_ */