RTMPdump源码分析: NetConnection(5)

函数调用结构图

RTMPDump (libRTMP)的整体的函数调用结构图如下图所示。


RTMP_Connect()函数用于建立RTMP中的NetConnection;

[cpp]  view plain  copy
  1. //连接  
  2. int  
  3. RTMP_Connect(RTMP *r, RTMPPacket *cp)  
  4. {  
  5.     //Socket结构体  
  6.   struct sockaddr_in service;  
  7.   if (!r->Link.hostname.av_len)  
  8.     return FALSE;  
  9.   
  10.   memset(&service, 0, sizeof(struct sockaddr_in));  
  11.   service.sin_family = AF_INET;  
  12.   
  13.   if (r->Link.socksport)  
  14.     {  
  15.     //加入地址信息  
  16.       /* 使用SOCKS连接 */  
  17.       if (!add_addr_info(&service, &r->Link.sockshost, r->Link.socksport))  
  18.     return FALSE;  
  19.     }  
  20.   else  
  21.     {  
  22.       /* 直接连接 */  
  23.       if (!add_addr_info(&service, &r->Link.hostname, r->Link.port))  
  24.     return FALSE;  
  25.     }  
  26.   //-----------------  
  27.   r->dlg->AppendCInfo("建立连接:第0次连接。开始建立Socket连接");  
  28.   //-----------------------------  
  29.   if (!RTMP_Connect0(r, (struct sockaddr *)&service)){  
  30.     r->dlg->AppendCInfo("建立连接:第0次连接。建立Socket连接失败");  
  31.     return FALSE;  
  32.   }  
  33.   //-----------------  
  34.   r->dlg->AppendCInfo("建立连接:第0次连接。建立Socket连接成功");  
  35.   //-----------------------------  
  36.   r->m_bSendCounter = TRUE;  
  37.   
  38.   return RTMP_Connect1(r, cp);  
  39. }  
可见RTMP_Connect0()主要用于建立Socket连接,并未开始真正的建立RTMP连接。

RTMP_Connect0()

[cpp]  view plain  copy
  1. //sockaddr是Linux网络编程的地址结构体一种,其定义如下:  
  2. //struct sockaddr{   
  3. //  unsigned short sa_family;   
  4. //  char sa_data[14];  
  5. //};   
  6. //说明:sa_family:是地址家族,也称作,协议族,一般都是“AF_xxx”的形式。通常大多用的是都是AF_INET。  
  7. //  sa_data:是14字节协议地址。  
  8. //有时不使用sockaddr,而使用sockaddr_in(多用在windows)(等价)  
  9. //struct sockaddr_in {  
  10. //  short int sin_family;              /* Address family */  
  11. //  unsigned short int sin_port;       /* Port number */  
  12. //  struct in_addr sin_addr;           /* Internet address */  
  13. //  unsigned char sin_zero[8];         /* Same size as struct sockaddr */  
  14. //};  
  15. //union {  
  16. //   struct{  
  17. //      unsigned char s_b1,s_b2,s_b3,s_b4;  
  18. //         } S_un_b;  
  19. //    struct {  
  20. //      unsigned short s_w1,s_w2;  
  21. //         } S_un_w;  
  22. //    unsigned long S_addr;  
  23. //   } S_un;  
  24. //} in_addr;  
  25. //第0次连接,建立Socket连接  
  26. int  
  27. RTMP_Connect0(RTMP *r, struct sockaddr * service)  
  28. {  
  29.   int on = 1;  
  30.   r->m_sb.sb_timedout = FALSE;  
  31.   r->m_pausing = 0;  
  32.   r->m_fDuration = 0.0;  
  33.   //创建一个Socket,并把Socket序号赋值给相应变量  
  34.   //-----------------  
  35.   r->dlg->AppendCInfo("建立连接:第0次连接。create一个Socket");  
  36.   //-----------------------------  
  37.   r->m_sb.sb_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);  
  38.   if (r->m_sb.sb_socket != -1)  
  39.     {  
  40.   
  41.     //定义函数 int connect (int sockfd,struct sockaddr * serv_addr,int addrlen);     
  42.     //函数说明 connect()用来将参数sockfd 的Socket(刚刚创建)连至参数serv_addr   
  43.     //指定的网络地址。参数addrlen为sockaddr的结构长度。   
  44.     //连接  
  45.     RTMP_LogPrintf("建立Socket连接!\n");  
  46.     //-----------------  
  47.     r->dlg->AppendCInfo("建立连接:第0次连接。connect该Socket");  
  48.     //-----------------------------  
  49.       if (connect(r->m_sb.sb_socket, service, sizeof(struct sockaddr)) < 0)  
  50.     {  
  51.         //-----------------  
  52.         r->dlg->AppendCInfo("建立连接:第0次连接。connect该Socket失败");  
  53.         //-----------------------------  
  54.       int err = GetSockError();  
  55.       RTMP_Log(RTMP_LOGERROR, "%s, failed to connect socket. %d (%s)",  
  56.           __FUNCTION__, err, strerror(err));  
  57.       RTMP_Close(r);  
  58.       return FALSE;  
  59.     }  
  60.       //-----------------  
  61.       r->dlg->AppendCInfo("建立连接:第0次连接。connect该Socket成功");  
  62.       //-----------------------------  
  63.       //指定了端口号。注:这不是必需的。  
  64.       if (r->Link.socksport)  
  65.     {  
  66.       RTMP_Log(RTMP_LOGDEBUG, "%s ... SOCKS negotiation", __FUNCTION__);  
  67.       //谈判?发送数据报以进行谈判?!  
  68.       if (!SocksNegotiate(r))  
  69.         {  
  70.           RTMP_Log(RTMP_LOGERROR, "%s, SOCKS negotiation failed.", __FUNCTION__);  
  71.           RTMP_Close(r);  
  72.           return FALSE;  
  73.         }  
  74.     }  
  75.     }  
  76.   else  
  77.     {  
  78.       RTMP_Log(RTMP_LOGERROR, "%s, failed to create socket. Error: %d", __FUNCTION__,  
  79.       GetSockError());  
  80.       return FALSE;  
  81.     }  
  82.   
  83.   /* set timeout */  
  84.   //超时  
  85.   {  
  86.     SET_RCVTIMEO(tv, r->Link.timeout);  
  87.     if (setsockopt  
  88.         (r->m_sb.sb_socket, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof(tv)))  
  89.       {  
  90.         RTMP_Log(RTMP_LOGERROR, "%s, Setting socket timeout to %ds failed!",  
  91.         __FUNCTION__, r->Link.timeout);  
  92.       }  
  93.   }  
  94.   
  95.   setsockopt(r->m_sb.sb_socket, IPPROTO_TCP, TCP_NODELAY, (char *) &on, sizeof(on));  
  96.   
  97.   return TRUE;  
  98. }  

再来看看RTMP_Connect1(),这是真正建立RTMP连接的函数:

RTMP_Connect1()

[cpp]  view plain  copy
  1. //第1次连接,从握手开始  
  2. int  
  3. RTMP_Connect1(RTMP *r, RTMPPacket *cp)  
  4. {  
  5.   if (r->Link.protocol & RTMP_FEATURE_SSL)  
  6.     {  
  7. #if defined(CRYPTO) && !defined(NO_SSL)  
  8.       TLS_client(RTMP_TLS_ctx, r->m_sb.sb_ssl);  
  9.       TLS_setfd((SSL *)r->m_sb.sb_ssl, r->m_sb.sb_socket);  
  10.       if (TLS_connect((SSL *)r->m_sb.sb_ssl) < 0)  
  11.     {  
  12.       RTMP_Log(RTMP_LOGERROR, "%s, TLS_Connect failed", __FUNCTION__);  
  13.       RTMP_Close(r);  
  14.       return FALSE;  
  15.     }  
  16. #else  
  17.       RTMP_Log(RTMP_LOGERROR, "%s, no SSL/TLS support", __FUNCTION__);  
  18.       RTMP_Close(r);  
  19.       return FALSE;  
  20.   
  21. #endif  
  22.     }  
  23.   //使用HTTP  
  24.   if (r->Link.protocol & RTMP_FEATURE_HTTP)  
  25.     {  
  26.       r->m_msgCounter = 1;  
  27.       r->m_clientID.av_val = NULL;  
  28.       r->m_clientID.av_len = 0;  
  29.       HTTP_Post(r, RTMPT_OPEN, "", 1);  
  30.       HTTP_read(r, 1);  
  31.       r->m_msgCounter = 0;  
  32.     }  
  33.   RTMP_Log(RTMP_LOGDEBUG, "%s, ... connected, handshaking", __FUNCTION__);  
  34.   //握手----------------  
  35.   r->dlg->AppendCInfo("建立连接:第1次连接。开始握手(HandShake)");  
  36.   //-----------------------------  
  37.   RTMP_LogPrintf("开始握手(HandShake)!\n");  
  38.   if (!HandShake(r, TRUE))  
  39.     {  
  40.         //----------------  
  41.         r->dlg->AppendCInfo("建立连接:第1次连接。握手(HandShake)失败!");  
  42.         //-----------------------------  
  43.       RTMP_Log(RTMP_LOGERROR, "%s, handshake failed.", __FUNCTION__);  
  44.       RTMP_Close(r);  
  45.       return FALSE;  
  46.     }  
  47.   //----------------  
  48.   r->dlg->AppendCInfo("建立连接:第1次连接。握手(HandShake)成功");  
  49.   //-----------------------------  
  50.   RTMP_LogPrintf("握手(HandShake)完毕!\n");  
  51.   RTMP_Log(RTMP_LOGDEBUG, "%s, handshaked", __FUNCTION__);  
  52.   //发送“connect”命令--------------  
  53.   //----------------  
  54.   r->dlg->AppendCInfo("建立连接:第1次连接。开始建立网络连接(NetConnection)");  
  55.   //-----------------------------  
  56.   RTMP_LogPrintf("开始建立网络连接(NetConnection)!\n");  
  57.   //----------------  
  58.   r->dlg->AppendCInfo("发送数据。消息 命令 (typeID=20) (Connect)。");  
  59.   //-----------------------------  
  60.   if (!SendConnectPacket(r, cp))  
  61.     {  
  62.         //----------------  
  63.         r->dlg->AppendCInfo("建立连接:第1次连接。建立网络连接(NetConnection)失败!");  
  64.         //-----------------------------  
  65.       RTMP_Log(RTMP_LOGERROR, "%s, RTMP connect failed.", __FUNCTION__);  
  66.       RTMP_Close(r);  
  67.       return FALSE;  
  68.     }  
  69.   //----------------  
  70.   r->dlg->AppendCInfo("建立连接:第1次连接。建立网络连接(NetConnection)成功");  
  71.   //-----------------------------  
  72.   RTMP_LogPrintf("命令消息“Connect”发送完毕!\n");  
  73.   return TRUE;  
  74. }  


你可能感兴趣的:(多媒体协议)