判断socket和fd之间的关系

Question

在unix上，一切都是文件函数read()、write()、close( ) 在 Win32 上不受支持。

我想模拟它，但不知道如何区分WinSocks2上的sock是socket还是fd。

//returns 1 if `sock` is network socket, 
//        0 if `sock` is file desriptor (including stdio, stderr, stdout), ...
//       -1 in none of above
int is_net_socket(int sock)
{
    // ...?
}

这应该像以下内容一样工作：

int mysock  = socket(PF_INET, SOCK_STREAM, 0);
int myfd    = _open("my_file.txt", _O_RDONLY);

printf("1: %d    2: %d    3: %d    4:%d\n",
       is_net_socket(mysock),   //1
       is_net_socket(myfd),     //0
       is_net_socket(stdin),    //0
       is_net_socket(stderr));  //0

// should print "1: 1    2: 0    3: 0    4:0"

---

如何实现 is_net_socket 以便使用它，如下所示：

int my_close(int sock)
{
#if ON_WINDOWS
    switch( is_net_socket(sock) ) {
        case 1: return closesocket(sock);
        case 0: return _close(sock);
        default: //handle error...
    }
#else
    return close(sock);
#endif
}

Answer 1

不确定您从哪里得到这样的想法：Windows 不允许您将 SOCKET 句柄用作文件 - 正如 套接字句柄 页面：

套接字句柄可以选择是 Windows Sockets 2 中的文件句柄。来自 Winsock 提供程序的套接字句柄可以与其他非 Winsock 函数（例如 ReadFile、WriteFile、ReadFileEx 和 WriteFileEx）一起使用。

无论如何，至于如何在 Windows 上区分它们，请参阅函数 NtQueryObject，如果传递给它的句柄是一个开放的 SOCKET，它将返回 \Device\Tcp 的句柄名称>。请阅读“备注”部分以了解此调用返回的结构。

请注意，此方法仅适用于 XP 及更高版本，并且在 Windows 2000 上会失败（我假设它已经足够老了，不会影响您。）

Answer 2

我想你可以使用 select 来查询套接字的状态。

http://msdn.microsoft.com/en-us /library/ms740141%28VS.85%29.aspx

我建议将文件 desc 和套接字分组在一个结构中。您可以声明一个枚举来判断描述符是文件还是套接字。我知道这可能不像您想要的那样动态，但通常当您创建可移植应用程序时，最好将这些细节抽象出来。

例子：

enum type { SOCKET, FILE };

typedef struct
{
    unsigned int id;
    type dataType;
} descriptor_t;

int close(descriptor_t sock)
{
#if WIN32
    if (sock.dataType == SOCKET)
        return closesocket(sock.id);
    else
        return _close(sock.id);
#else
    return close(sock.id);
#endif
}

Answer 3

我怀疑......但我不确定，Windows 上的 fds 和套接字使用单独的名称空间。因此，套接字和文件的编号可能相同，并且当您调用 is_net_socket 时不可能知道您正在谈论哪一个。

尝试打印出套接字和文件描述符编号，看看它们是否同时彼此相同。

Answer 4

如果 Windows 'C' 库有 dup()，您可以尝试复制它，这对于套接字应该失败，但对于文件 fd 会成功。所以：

int is_net_socket(fd)
{
  return close(dup(fd)) != 0;
}

警告：未经测试的理论与未经测试的依赖关系;-) 请注意，如果您用完 fd，这将返回误导性的结果。另一个副作用是，如果它是一个文件，它将被刷新并更新其目录条目。总而言之，坦率地说，这可能很糟糕。我什至可能自己投反对票。