android 移植好的librtmp 2.4

/* * Copyright (C) 2005-2008 Team XBMC * http://www.xbmc.org * Copyright (C) 2008-2009 Andrej Stepanchuk * Copyright (C) 2009-2010 Howard Chu * * This file is part of librtmp. * * librtmp 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, * or (at your option) any later version. * * librtmp 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 General Public License for more details. * * You should have received a copy of the GNU Lesser General Public License * along with librtmp see the file COPYING. If not, write to * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, * Boston, MA 02110-1301, USA. * http://www.gnu.org/copyleft/lgpl.html */ #include <stdint.h> #include <stdlib.h> #include <string.h> #include <assert.h> #include <time.h> #include "rtmp_sys.h" #include "log.h" #ifdef CRYPTO #ifdef USE_POLARSSL #include <polarssl/havege.h> #include <polarssl/md5.h> #include <polarssl/base64.h> #define MD5_DIGEST_LENGTH 16 static const char *my_dhm_P = "E4004C1F94182000103D883A448B3F80" \ "2CE4B44A83301270002C20D0321CFD00" \ "11CCEF784C26A400F43DFB901BCA7538" \ "F2C6B176001CF5A0FD16D2C48B1D0C1C" \ "F6AC8E1DA6BCC3B4E1F96B0564965300" \ "FFA1D0B601EB2800F489AA512C4B248C" \ "01F76949A60BB7F00A40B1EAB64BDD48" \ "E8A700D60B7F1200FA8E77B0A979DABF"; static const char *my_dhm_G = "4"; #elif defined(USE_GNUTLS) #include <gnutls/gnutls.h> #define MD5_DIGEST_LENGTH 16 #include <nettle/base64.h> #include <nettle/md5.h> #else /* USE_OPENSSL */ #include <openssl/ssl.h> #include <openssl/rc4.h> #include <openssl/md5.h> #include <openssl/bio.h> #include <openssl/buffer.h> #endif TLS_CTX RTMP_TLS_ctx; #endif #define RTMP_SIG_SIZE 1536 #define RTMP_LARGE_HEADER_SIZE 12 static const int packetSize[] = { 12, 8, 4, 1 }; int RTMP_ctrlC; const char RTMPProtocolStrings[][7] = { "RTMP", "RTMPT", "RTMPE", "RTMPTE", "RTMPS", "RTMPTS", "", "", "RTMFP" }; const char RTMPProtocolStringsLower[][7] = { "rtmp", "rtmpt", "rtmpe", "rtmpte", "rtmps", "rtmpts", "", "", "rtmfp" }; static const char *RTMPT_cmds[] = { "open", "send", "idle", "close" }; typedef enum { RTMPT_OPEN=0, RTMPT_SEND, RTMPT_IDLE, RTMPT_CLOSE } RTMPTCmd; static int DumpMetaData(AMFObject *obj); static int HandShake(RTMP *r, int FP9HandShake); static int SocksNegotiate(RTMP *r); static int SendConnectPacket(RTMP *r, RTMPPacket *cp); static int SendCheckBW(RTMP *r); static int SendCheckBWResult(RTMP *r, double txn); static int SendDeleteStream(RTMP *r, double dStreamId); static int SendFCSubscribe(RTMP *r, AVal *subscribepath); static int SendPlay(RTMP *r); static int SendBytesReceived(RTMP *r); static int SendUsherToken(RTMP *r, AVal *usherToken); #if 0 /* unused */ static int SendBGHasStream(RTMP *r, double dId, AVal *playpath); #endif static int HandleInvoke(RTMP *r, const char *body, unsigned int nBodySize); static int HandleMetadata(RTMP *r, char *body, unsigned int len); static void HandleChangeChunkSize(RTMP *r, const RTMPPacket *packet); static void HandleAudio(RTMP *r, const RTMPPacket *packet); static void HandleVideo(RTMP *r, const RTMPPacket *packet); static void HandleCtrl(RTMP *r, const RTMPPacket *packet); static void HandleServerBW(RTMP *r, const RTMPPacket *packet); static void HandleClientBW(RTMP *r, const RTMPPacket *packet); static int ReadN(RTMP *r, char *buffer, int n); static int WriteN(RTMP *r, const char *buffer, int n); static void DecodeTEA(AVal *key, AVal *text); static int HTTP_Post(RTMP *r, RTMPTCmd cmd, const char *buf, int len); static int HTTP_read(RTMP *r, int fill); static void CloseInternal(RTMP *r, int reconnect); #ifndef _WIN32 static int clk_tck; #endif #ifdef CRYPTO #include "handshake.h" #endif uint32_t RTMP_GetTime() { #ifdef _DEBUG return 0; #elif defined(_WIN32) return timeGetTime(); #else struct tms t; if (!clk_tck) clk_tck = sysconf(_SC_CLK_TCK); return times(&t) * 1000 / clk_tck; #endif } void RTMP_UserInterrupt() { RTMP_ctrlC = TRUE; } void RTMPPacket_Reset(RTMPPacket *p) { p->m_headerType = 0; p->m_packetType = 0; p->m_nChannel = 0; p->m_nTimeStamp = 0; p->m_nInfoField2 = 0; p->m_hasAbsTimestamp = FALSE; p->m_nBodySize = 0; p->m_nBytesRead = 0; } int RTMPPacket_Alloc(RTMPPacket *p, uint32_t nSize) { char *ptr; if (nSize > SIZE_MAX - RTMP_MAX_HEADER_SIZE) return FALSE; ptr = calloc(1, nSize + RTMP_MAX_HEADER_SIZE); if (!ptr) return FALSE; p->m_body = ptr + RTMP_MAX_HEADER_SIZE; p->m_nBytesRead = 0; return TRUE; } void RTMPPacket_Free(RTMPPacket *p) { if (p->m_body) { free(p->m_body - RTMP_MAX_HEADER_SIZE); p->m_body = NULL; } } void RTMPPacket_Dump(RTMPPacket *p) { RTMP_Log(RTMP_LOGDEBUG, "RTMP PACKET: packet type: 0x%02x. channel: 0x%02x. info 1: %d info 2: %d. Body size: %u. body: 0x%02x", p->m_packetType, p->m_nChannel, p->m_nTimeStamp, p->m_nInfoField2, p->m_nBodySize, p->m_body ? (unsigned char)p->m_body[0] : 0); } int RTMP_LibVersion() { return RTMP_LIB_VERSION; } void RTMP_TLS_Init() { #ifdef CRYPTO #ifdef USE_POLARSSL /* Do this regardless of NO_SSL, we use havege for rtmpe too */ RTMP_TLS_ctx = calloc(1,sizeof(struct tls_ctx)); havege_init(&RTMP_TLS_ctx->hs); #elif defined(USE_GNUTLS) && !defined(NO_SSL) /* Technically we need to initialize libgcrypt ourselves if * we're not going to call gnutls_global_init(). Ignoring this * for now. */ gnutls_global_init(); RTMP_TLS_ctx = malloc(sizeof(struct tls_ctx)); gnutls_certificate_allocate_credentials(&RTMP_TLS_ctx->cred); gnutls_priority_init(&RTMP_TLS_ctx->prios, "NORMAL", NULL); gnutls_certificate_set_x509_trust_file(RTMP_TLS_ctx->cred, "ca.pem", GNUTLS_X509_FMT_PEM); #elif !defined(NO_SSL) /* USE_OPENSSL */ /* libcrypto doesn't need anything special */ SSL_load_error_strings(); SSL_library_init(); OpenSSL_add_all_digests(); RTMP_TLS_ctx = SSL_CTX_new(SSLv23_method()); SSL_CTX_set_options(RTMP_TLS_ctx, SSL_OP_ALL); SSL_CTX_set_default_verify_paths(RTMP_TLS_ctx); #endif #endif } void * RTMP_TLS_AllocServerContext(const char* cert, const char* key) { void *ctx = NULL; #ifdef CRYPTO if (!RTMP_TLS_ctx) RTMP_TLS_Init(); #ifdef USE_POLARSSL tls_server_ctx *tc = ctx = calloc(1, sizeof(struct tls_server_ctx)); tc->dhm_P = my_dhm_P; tc->dhm_G = my_dhm_G; tc->hs = &RTMP_TLS_ctx->hs; if (x509parse_crtfile(&tc->cert, cert)) { free(tc); return NULL; } if (x509parse_keyfile(&tc->key, key, NULL)) { x509_free(&tc->cert); free(tc); return NULL; } #elif defined(USE_GNUTLS) && !defined(NO_SSL) gnutls_certificate_allocate_credentials((gnutls_certificate_credentials*) &ctx); if (gnutls_certificate_set_x509_key_file(ctx, cert, key, GNUTLS_X509_FMT_PEM) != 0) { gnutls_certificate_free_credentials(ctx); return NULL; } #elif !defined(NO_SSL) /* USE_OPENSSL */ ctx = SSL_CTX_new(SSLv23_server_method()); if (!SSL_CTX_use_certificate_chain_file(ctx, cert)) { SSL_CTX_free(ctx); return NULL; } if (!SSL_CTX_use_PrivateKey_file(ctx, key, SSL_FILETYPE_PEM)) { SSL_CTX_free(ctx); return NULL; } #endif #endif return ctx; } void RTMP_TLS_FreeServerContext(void *ctx) { #ifdef CRYPTO #ifdef USE_POLARSSL x509_free(&((tls_server_ctx*)ctx)->cert); rsa_free(&((tls_server_ctx*)ctx)->key); free(ctx); #elif defined(USE_GNUTLS) && !defined(NO_SSL) gnutls_certificate_free_credentials(ctx); #elif !defined(NO_SSL) /* USE_OPENSSL */ SSL_CTX_free(ctx); #endif #endif } RTMP * RTMP_Alloc() { return calloc(1, sizeof(RTMP)); } void RTMP_Free(RTMP *r) { free(r); } void RTMP