UDT源码剖析（六）：UDT::socket()过程代码注释

调用路线
UDT::socket()->CUDT::socket()->CUDTUnited::newSocket()

UDTSOCKET CUDT::socket(int af, int type, int)
 
{
 
   // 如果垃圾收集没启动，那么这里自动调用startup
 
   // 所以手动调用startup其实不是必须的，但是为了接受作者的风格，我们也养成这样的习惯吧
 
   // 起码startup你可以在启动时就调用，这可以让第一次创建Socket的过程显得快一些
 
   if (!s_UDTUnited.m_bGCStatus)
 
      s_UDTUnited.startup();
 
 
 
   try
 
   {
 
      // 直接调用Socket单元的newSocket
 
      return s_UDTUnited.newSocket(af, type);
 
   }
 
   catch (CUDTException& e)
 
   {
 
      s_UDTUnited.setError(new CUDTException(e));
 
      return INVALID_SOCK;
 
   }
 
   catch (bad_alloc&)
 
   {
 
      s_UDTUnited.setError(new CUDTException(3, 2, 0));
 
      return INVALID_SOCK;
 
   }
 
   catch (...)
 
   {
 
      s_UDTUnited.setError(new CUDTException(-1, 0, 0));
 
      return INVALID_SOCK;
 
   }
 
}

UDTSOCKET CUDTUnited::newSocket(const int& af, const int& type)
 
{
 
   // 只允许SOCK_STREAM和SOCK_DGRAM两种类型
 
   // 前面说sendfile和recvfile的时候说了这两个类型会被转化为UDT自己的类型，下面会看到转化的地方
 
   if ((type != SOCK_STREAM) && (type != SOCK_DGRAM))
 
      throw CUDTException(5, 3, 0);
 
 
 
   // 声明一个新的Socket指针
 
   CUDTSocket* ns = NULL;
 
 
 
   try
 
   {
 
      // 创建Socket对象
 
      ns = new CUDTSocket;
 
 
 
      // 创建Socket的core对象
 
      ns->m_pUDT = new CUDT;
 
 
 
      // 根据ipv4、ipv6的标准设置初始地址信息
 
      if (AF_INET == af)
 
      {
 
         ns->m_pSelfAddr = (sockaddr*)(new sockaddr_in);
 
         ((sockaddr_in*)(ns->m_pSelfAddr))->sin_port = 0;
 
      }
 
      else
 
      {
 
         ns->m_pSelfAddr = (sockaddr*)(new sockaddr_in6);
 
         ((sockaddr_in6*)(ns->m_pSelfAddr))->sin6_port = 0;
 
      }
 
   }
 
   catch (...)
 
   {
 
      delete ns;
 
      throw CUDTException(3, 2, 0);
 
   }
 
 
 
   // 加锁分配ID，然后解锁，ID初始为随机数，然后逐次减一
 
   CGuard::enterCS(m_IDLock);
 
   ns->m_SocketID = -- m_SocketID;
 
   CGuard::leaveCS(m_IDLock);
 
 
 
   // 设置状态为INIT
 
   ns->m_Status = INIT;
 
   // 监听SocketID为0，说明目前为独立Socket
 
   ns->m_ListenSocket = 0;
 
   // 设置Core的SocketID
 
   ns->m_pUDT->m_SocketID = ns->m_SocketID;
 
   // 这里进行自己的类型映射转换
 
   ns->m_pUDT->m_iSockType = (SOCK_STREAM == type) ? UDT_STREAM : UDT_DGRAM;
 
   // 协议版本，ipv4或者ipv6
 
   ns->m_pUDT->m_iIPversion = ns->m_iIPversion = af;
 
   // UDT网络信息的缓存，这个对象在CUDTUnited构造时创建，在CUDTUnited对象析构时销毁
 
   ns->m_pUDT->m_pCache = m_pCache;
 
 
 
   // protect the m_Sockets structure.
 
   // 将这个新的socket加入socket列表，如果加入过程中有任何异常，则socket创建失败
 
   // 注意这个CUDTUnited，我们叫它Socket单元，是一个socket的管理器
 
   // 而非单个socket业务对象，这个逻辑我们以后再分析
 
   CGuard::enterCS(m_ControlLock);
 
   try
 
   {
 
      m_Sockets[ns->m_SocketID] = ns;
 
   }
 
   catch (...)
 
   {
 
      //failure and rollback
 
      delete ns;
 
      ns = NULL;
 
   }
 
   CGuard::leaveCS(m_ControlLock);
 
 
 
   // 判断socket对象指针是否正常
 
   if (NULL == ns)
 
      throw CUDTException(3, 2, 0);
 
 
 
   // 返回新socket的ID
 
   return ns->m_SocketID;
 
}