LibRTMP源代码分析5：建立网络连接

  网络连接的基本步骤在第一篇文章中有所介绍，此处不再重复。源代码中的 RTMP_Connect(...) 用于建立RTMP网络连接。

/**

 * @brief 建立RTMP中的网络连接(NetConnection).

 *  a) 创建并设置目标socket，包括ip地址和端口号

 *  b) 建立socket连接，设置socket的超时和接收、发送缓冲区的大小

 *  c) 握手操作

 *  d) 发送含有connect命令的数据报，用于建立RTMP连接

 */

int RTMP_Connect(RTMP *r, RTMPPacket *cp, struct sockaddr *dst)

{

// socket结构体

struct sockaddr_in service;

if (!r->Link.hostname.av_len)

return FALSE;

// 设置socket地址

memset(&service, 0, sizeof(struct sockaddr_in));

service.sin_family = AF_INET;

if (r->Link.socksport)

{

/* Connect via SOCKS，使用SOCKS连接 */

if (!add_addr_info(&service, &r->Link.sockshost, r->Link.socksport))

return FALSE;

        }

else

{

/* Connect directly, 直接连接 */

if (!add_addr_info(&service, &r->Link.hostname, r->Link.port))

return FALSE;

       }

if (r->m_nstub == 1)

{

// 设置socket的ip地址和端口号

rtmp_sockaddr_set_ip((struct sockaddr*)&service, inet_addr(r->m_stubip));

service.sin_port = htons(r->m_stubport);

}

// 第0次连接，主要用于建立socket连接，并未开始真正的建立RTMP连接

// 设置socket的超时和socket接收、发送缓冲区的大小

if (!RTMP_Connect0(r, (struct sockaddr *)&service))

return FALSE;

if (dst) 

memcpy(dst, &service, sizeof(struct sockaddr));

r->m_bSendCounter = TRUE;

// 第1次连接，真正建立RTMP连接，主要包括握手和发送connect命令

return RTMP_Connect1(r, cp);

}

 RTMP_Connect()函数主要调用了上述红色标记出来的四个函数来完成RTMP网络连接，接下来我们分别来分析这四个函数。

/**

 * @brief 设置scoket的ip地址

 *

 * @param a  : 待设置的scoket指针

 * @param ip : 32位的网络字节序IP地址

 */

void rtmp_sockaddr_set_ip(struct sockaddr *a, unsigned long ip)

{

union { struct sockaddr_in ain; struct sockaddr addr; } ts;

ts.addr = a[0];

ts.ain.sin_addr.s_addr = ip;

a[0] = ts.addr;

}

/**

 * @brief 建立socket连接. r创建一个socket连接到service指定的网络地址。

 *  设置socket接收数据的超时时限以及socket接收和发送缓冲区的大小。

 */

int RTMP_Connect0(RTMP *r, struct sockaddr *service)

{

int recv_size = 128*1024;

int send_size = 128*1024;

int len = sizeof(int);

int on  = 1;

r->m_sb.sb_timedout = FALSE;

r->m_pausing = 0;

r->m_fDuration = 0.0;

// 创建一个流式socket

r->m_sb.sb_socket = r->m_sock.socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

if (r->m_sb.sb_socket != -1)

{

// 将刚刚创建的socket连接至service指定的网络地址

if (r->m_sock.connect(r->m_sb.sb_socket, service, sizeof(struct sockaddr)) <0)

{

int err = r->m_sock.getsockerr();

RTMP_Log(RTMP_LOGERROR, "%s, failed to connect socket. %d (%s)", __FUNCTION__, err, strerror(err));

RTMP_Close(r);

return FALSE;

}

// 指定了端口号，这个不是必需的.

if (r->Link.socksport)

{

RTMP_Log(RTMP_LOGDEBUG, "%s ... SOCKS negotiation", __FUNCTION__);

if (!SocksNegotiate(r))

{

RTMP_Log(RTMP_LOGERROR, "%s, SOCKS negotiation failed.", __FUNCTION__);

RTMP_Close(r);

return FALSE;

}

}

        }

else // 创建socket失败

{

RTMP_Log(RTMP_LOGERROR, "%s, failed to create socket. Error: %d", __FUNCTION__, r->m_sock.getsockerr());

return FALSE;

        }

/* set timeout,设置超时 */

if (r->m_sock.setsockopt) 

{

// 获取设定的超时时限，单位毫秒

SET_RCVTIMEO(tv, r->Link.timeout);

// 设置socket接收数据的超时时限，单位毫秒

if (r->m_sock.setsockopt(r->m_sb.sb_socket, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof(tv)))

{

RTMP_Log(RTMP_LOGERROR, "%s, Setting socket timeout to %ds failed!", __FUNCTION__, r->Link.timeout);

}

// 设置socket接收和发送缓冲区的大小

r->m_sock.setsockopt(r->m_sb.sb_socket, SOL_SOCKET, SO_RCVBUF, (char*)&recv_size, len);

r->m_sock.setsockopt(r->m_sb.sb_socket, SOL_SOCKET, SO_SNDBUF, (char*)&send_size, len);

}

return TRUE;

}

第一次连接RTMP_Connect1( )函数主要做了两件事情： 

 1） HandShake（）完成握手，之前已经分析过了；

 2） SendConnectPacket（）发送connect命令，用于建立RTMP连接。接下来就具体分析一下这个函数。