WSAAsyncSelect模型

afx_msg LRESULT CSmallFileDlg::OnAsyncSelect(WPARAM wParam, LPARAM lParam)
{
	switch (WSAGETSELECTEVENT(lParam))
	{
	case FD_CONNECT:
		if (WSAGETSELECTERROR(lParam))
		{
			MessageBox(GetErrorString(WSAGETSELECTERROR(lParam), L"FD_CONNECT"));
		}
		else
		{
			MessageBox(L"连接成功");
		}
		break;


	case FD_READ:
		{
			recvBuffer();
			static WinFile fr(L"C:\\Users\\jiaw\\Desktop\\2.flv", false);
			static size_t totalWritten = 0;
			size_t currentWritten;
			int ret = fr.write(m_input.peek(), m_input.readableBytes(), ¤tWritten);
			if (0 != ret) {
				MessageBox(GetErrorString(ret, L"FD_READ"));
				::shutdown(m_socket, SD_BOTH);
				::closesocket(wParam);
			} else {
				totalWritten += currentWritten;
				m_input.retrieve(currentWritten);
				fr.seek(totalWritten);
			}
		}
		break;


	case FD_WRITE:
		sendBuffer(); // send当网络缓冲区满的时候会产生WSAEWOULDBLOCK错误码，此时触发写事件继续写入
		break;


	case FD_CLOSE:
		if (WSAGETSELECTERROR(lParam))
		{
			MessageBox(GetErrorString(WSAGetLastError(), L"FD_CLOSE"));
		}
		::shutdown(m_socket, SD_BOTH);
		::closesocket(wParam);
		MessageBox(L"close");
		break;
	}


	return 0;
}


void CSmallFileDlg::OnBnClickedConnect()
{
	const char *ip = "192.168.239.150";
	unsigned short port = 5000;
	struct sockaddr_in sin;
	memset(&sin, 0, sizeof(sin));
	sin.sin_family = AF_INET;
	sin.sin_addr.S_un.S_addr = inet_addr(ip);
	sin.sin_port = htons(port);
	m_socket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (INVALID_SOCKET == m_socket) {
		MessageBox(GetErrorString(WSAGetLastError(), L"socket"));
		return;
	}

	// 会自动将socket设为非阻塞
	if (SOCKET_ERROR == WSAAsyncSelect(m_socket, m_hWnd, WM_ASYNC_SELECT, FD_READ | FD_WRITE | FD_CONNECT | FD_CLOSE)) {
		MessageBox(GetErrorString(WSAGetLastError(), L"WSAAsyncSelect"));
		::closesocket(m_socket);
	}


	if (SOCKET_ERROR == ::connect(m_socket, (sockaddr*)&sin, sizeof(sin))) {
		int errCode = WSAGetLastError();
		if (WSAEWOULDBLOCK != errCode) { // 非阻塞连接处理WSAEWOULDBLOCK
			MessageBox(GetErrorString(WSAGetLastError(), L"connect"));
			::closesocket(m_socket);
			return;
		}
	}
}

void CSmallFileDlg::OnBnClickedDisconnect()
{
	::shutdown(m_socket, SD_BOTH);
	::closesocket(m_socket);
}

void CSmallFileDlg::OnBnClickedSend()
{
	WinFile fr(L"C:\\Users\\jiaw\\Desktop\\1.flv", true);
	char buf[65536];
	size_t result;
	DWORD total = fr.size();
	DWORD nread = 0;
	while (nread != total) {
		int ret = fr.read(buf, sizeof(buf), &result);
		if (0 != ret) {
			MessageBox(GetErrorString(ret));
			break;
		}


		nread += result;
		fr.seek(nread); // 设置下次的读取位置，从0开始偏移


		m_output.append(buf, result);
		sendBuffer();
	}
}

void CSmallFileDlg::sendBuffer()
{
	int ret = ::send(m_socket, m_output.peek(), m_output.readableBytes(), 0);
	if (ret > 0)
	{
		m_output.retrieve(ret);
		m_sendNum += ret; // 发送消息计数
		UpdateData(FALSE);
	}
	else if (SOCKET_ERROR == ret)
	{
		int errCode = WSAGetLastError();
		if (WSAEWOULDBLOCK != errCode)
		{
			MessageBox(GetErrorString(errCode), L"send");
		}
	}
}

void CSmallFileDlg::recvBuffer()
{
	char buf[65536] = {0};
	int ret = ::recv(m_socket, buf, sizeof(buf) - 1, 0);
	if (ret > 0)
	{
		m_recvNum += ret; // 接收消息计数
		UpdateData(FALSE);
		m_input.append(buf, ret);
	}
	else if (0 == ret)
	{
		MessageBox(L"连接优雅的关闭");
	}
	else
	{
		MessageBox(GetErrorString(WSAGetLastError(), L"recv"));
	}
}

// 产生随机字符串
std::string BuildRandString(int num)
{
	static unsigned int s_add = 0;
	std::string ret;
	srand((unsigned int)time(NULL) + (s_add++));
	for (int i=0; i<num; )
	{
		char buf[17] = {0};
		_itoa_s(rand(), buf, 0x10);
		ret += buf;
		i += strlen(buf);
	}
	return ret.substr(0, num);
}

void DebugLog(const char *fmt, ...)
{
	va_list ap;
	va_start(ap, fmt);
	char buf[1024] = {0};
	vsnprintf_s(buf, sizeof(buf), fmt, ap);
	MessageBoxA(NULL, buf, "debug log", MB_OK);
	va_end(ap);
}