RtmpProtocol.h 4.34 KB
Newer Older
xiongziliang committed
1
/*
xiongziliang committed
2
 * MIT License
xzl committed
3
 *
xiongziliang committed
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
 * Copyright (c) 2016 xiongziliang <771730766@qq.com>
 *
 * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
xzl committed
25 26 27 28 29
 */

#ifndef SRC_RTMP_RTMPPROTOCOL_H_
#define SRC_RTMP_RTMPPROTOCOL_H_

xiongzilaing committed
30
#include <memory>
xzl committed
31 32 33
#include <string>
#include <functional>
#include "amf.h"
xiongzilaing committed
34 35 36 37 38
#include "Rtmp.h"
#include "Util/util.h"
#include "Util/logger.h"
#include "Util/TimeTicker.h"
#include "Network/Socket.h"
771730766@qq.com committed
39
#include "Util/ResourcePool.h"
xzl committed
40 41 42 43 44 45 46 47 48 49 50 51 52 53 54

using namespace std;
using namespace ZL::Util;
using namespace ZL::Network;

namespace ZL {
namespace Rtmp {

class RtmpProtocol {
public:
	RtmpProtocol();
	virtual ~RtmpProtocol();
	//作为客户端发送c0c1,等待s0s1s2并且回调
	void startClientSession(const function<void()> &cb);
	void onParseRtmp(const char *pcRawData,int iSize);
xiongziliang committed
55
	void reset();
xzl committed
56
protected:
57
	virtual void onSendRawData(const Buffer::Ptr &buffer) = 0;
xzl committed
58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
	virtual void onRtmpChunk(RtmpPacket &chunkData) = 0;
	virtual void onStreamBegin(uint32_t ui32StreamId){
		m_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);
78
	void sendRtmp(uint8_t ui8Type, uint32_t ui32StreamId, const std::string &strBuf, uint32_t ui32TimeStamp, int iChunkID);
xzl committed
79 80 81
protected:
	int m_iReqID = 0;
	uint32_t m_ui32StreamId = STREAM_CONTROL;
xiongziliang committed
82 83 84
	int m_iNowStreamID = 0;
	int m_iNowChunkID = 0;
	bool m_bDataStarted = false;
85 86 87
	inline BufferRaw::Ptr obtainBuffer();
	inline BufferRaw::Ptr obtainBuffer(const void *data, int len);
	//ResourcePool<BufferRaw,MAX_SEND_PKT> m_bufferPool;
xzl committed
88 89 90
private:
	void handle_S0S1S2(const function<void()> &cb);
	void handle_C0C1();
91
	void handle_C1_simple();
92
#ifdef ENABLE_OPENSSL
93 94 95 96 97
	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);
98
#endif //ENABLE_OPENSSL
99

xzl committed
100 101 102 103
	void handle_C2();
	void handle_rtmp();
	void handle_rtmpChunk(RtmpPacket &chunkData);

104
private:
xzl committed
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125
	////////////ChunkSize////////////
	size_t m_iChunkLenIn = DEFAULT_CHUNK_LEN;
	size_t m_iChunkLenOut = DEFAULT_CHUNK_LEN;
	////////////Acknowledgement////////////
	uint32_t m_ui32ByteSent = 0;
	uint32_t m_ui32LastSent = 0;
	uint32_t m_ui32WinSize = 0;
	///////////PeerBandwidth///////////
	uint32_t m_ui32Bandwidth = 2500000;
	uint8_t m_ui8LimitType = 2;
	////////////Chunk////////////
	unordered_map<int, RtmpPacket> m_mapChunkData;
	//////////Rtmp parser//////////
	string m_strRcvBuf;
	function<void()> m_nextHandle;
};

} /* namespace Rtmp */
} /* namespace ZL */

#endif /* SRC_RTMP_RTMPPROTOCOL_H_ */