sockets - 在winsock中检索套接字的协议(protocol)

Question

我正在从事网络可靠性模拟工作，我需要根据服务质量百分比来模拟丢包。目前我有一个链接到 send 的 DLL , sendto , recv和 recvfrom .然后我的钩子(Hook)会根据服务质量“丢弃”数据包。

我只需要将钩子(Hook)应用于 UDP 数据包，而不干扰 TCP(TCP 用于远程调试)。

有没有一种方法可以查询 WinSock 以获取套接字绑定(bind)的协议(protocol)？

int WSAAPI HookedSend(SOCKET s, const char FAR * buf, int len, int flags)
{
  //if(s is UDP)
  //Drop according to QOS

  else
    //Send TCP packets undisturbed
    return send(s, buf, len, flags);
}

Answer 1

我认为您可以使用 getsockopt 获得套接字类型:

int optVal;
int optLen = sizeof(int);

getsockopt(socket, 
          SOL_SOCKET, 
          SO_TYPE, 
          (char*)&optVal, 
          &optLen);

if(optVal = SOCK_STREAM)
     printf("This is a TCP socket.\n");
else if(optVal = SOCK_DGRAM)
     printf("This is a UTP socket.\n");
else
     printf("Error");