学步园

 BOOL MsgMng::WSockInit(BOOL recv_flg)<br /> {<br /> WSADATA wsaData;<br /> if (::WSAStartup(0x0101, &wsaData) != 0)<br /> return GetSockErrorMsg("WSAStart()"), FALSE;</p> <p> if ((udp_sd = ::socket(AF_INET, SOCK_DGRAM, 0)) == INVALID_SOCKET)<br /> return GetSockErrorMsg("Please setup TCP/IP(controlpanel->network)/r/n"), FALSE;</p> <p> if (recv_flg != TRUE)<br /> return TRUE;</p> <p> if ((tcp_sd = ::socket(AF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET)<br /> return GetSockErrorMsg("Please setup2 TCP/IP(controlpanel->network)/r/n"), FALSE;</p> <p> struct sockaddr_in addr;<br /> memset(&addr, 0, sizeof(addr));<br /> addr.sin_family = AF_INET;<br /> addr.sin_addr.s_addr = local.addr;<br /> addr.sin_port = local.portNo;</p> <p> if (::bind(udp_sd, (LPSOCKADDR)&addr, sizeof(addr)) != 0)<br /> return GetSockErrorMsg("bind()"), FALSE;</p> <p> if (::bind(tcp_sd, (LPSOCKADDR)&addr, sizeof(addr)) != 0)<br /> {<br /> ::closesocket(tcp_sd);<br /> tcp_sd = INVALID_SOCKET;<br /> GetSockErrorMsg("bind(tcp) error. Can't support file attach");<br /> }</p> <p> // This function controls the I/O mode of a socket.<br /> // Set the socket I/O mode: In this case FIONBIO<br /> // enables or disables the blocking mode for the<br /> // socket based on the numerical value of iMode.<br /> // If flg = FALSE, blocking is enabled;<br /> // If flg = TRUE, non-blocking mode is enabled.<br /> // 上面是MSDN上面的解释，也就是说设置为非阻塞模式<br /> BOOL flg = TRUE;<br /> if (::ioctlsocket(udp_sd, FIONBIO, (unsigned long *)&flg) != 0)<br /> return GetSockErrorMsg("ioctlsocket(nonblock)"), FALSE;</p> <p> if (IsAvailableTCP() && ::ioctlsocket(tcp_sd, FIONBIO, (unsigned long *)&flg) != 0)<br /> return GetSockErrorMsg("ioctlsocket tcp(nonblock)"), FALSE;</p> <p> // 设置UDP允许广播<br /> flg = TRUE;<br /> if (::setsockopt(udp_sd, SOL_SOCKET, SO_BROADCAST, (char *)&flg, sizeof(flg)) != 0)<br /> return GetSockErrorMsg("setsockopt(broadcast)"), FALSE;</p> <p> // Specifies the total per-socket buffer space reserved for sends.<br /> // This is unrelated to SO_MAX_MSG_SIZE and does not necessarily<br /> // correspond to the size of a TCP send window.<br /> int buf_size = MAX_SOCKBUF, buf_minsize = MAX_SOCKBUF / 2;<br /> if (::setsockopt(udp_sd, SOL_SOCKET, SO_SNDBUF, (char *)&buf_size, sizeof(int)) != 0<br /> && ::setsockopt(udp_sd, SOL_SOCKET, SO_SNDBUF, (char *)&buf_minsize, sizeof(int)) != 0)<br /> GetSockErrorMsg("setsockopt(sendbuf)");</p> <p> // Specifies the total per-socket buffer space reserved for receives.<br /> // This is unrelated to SO_MAX_MSG_SIZE and does not necessarily<br /> // correspond to the size of the TCP receive window.<br /> buf_size = MAX_SOCKBUF, buf_minsize = MAX_SOCKBUF / 2;<br /> if (::setsockopt(udp_sd, SOL_SOCKET, SO_RCVBUF, (char *)&buf_size, sizeof(int)) != 0<br /> && ::setsockopt(udp_sd, SOL_SOCKET, SO_RCVBUF, (char *)&buf_minsize, sizeof(int)) != 0)<br /> GetSockErrorMsg("setsockopt(recvbuf)");</p> <p> // Allows the socket to be bound to an address that is already in use.<br /> // (See bind.) Not applicable on ATM sockets.<br /> // 这个套接字选项通知内核，如果端口忙，但TCP状态位于 TIME_WAIT ，可以重用端口。<br /> // 如果端口忙，而TCP状态位于其他状态，重用端口时依旧得到一个错误信息，指明"地址已经使用中"。<br /> // 如果你的服务程序停止后想立即重启，而新套接字依旧使用同一端口，此时 SO_REUSEADDR 选项非常有用。<br /> // 必须意识到，此时任何非期望数据到达，都可能导致服务程序反应混乱，不过这只是一种可能，事实上很不可能。<br /> //一个套接字由相关五元组构成，协议、本地地址、本地端口、远程地址、远程端口。<br /> // SO_REUSEADDR 仅仅表示可以重用本地本地地址、本地端口，整个相关五元组还是唯一确定的。<br /> // 所以，重启后的服务程序有可能收到非期望数据。必须慎重使用 SO_REUSEADDR 选项。<br /> flg = TRUE; // REUSE ADDR<br /> if (IsAvailableTCP() && ::setsockopt(tcp_sd, SOL_SOCKET, SO_REUSEADDR, (char *)&flg, sizeof(flg)) != 0)<br /> GetSockErrorMsg("setsockopt tcp(reuseaddr)");</p> <p> if (IsAvailableTCP() && ::listen(tcp_sd, 5) != 0)<br /> return FALSE;</p> <p> return TRUE;<br /> }