WSAAsynSelect模型

最新推荐文章于 2022-11-06 22:50:54 发布

DemonDante

最新推荐文章于 2022-11-06 22:50:54 发布

阅读量625

点赞数

CC 4.0 BY-SA版权

分类专栏：网络编程文章标签： socket null callback command stream windows

本文链接：https://blog.youkuaiyun.com/DemonDante/article/details/6654383

网络编程专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个使用WSAAsynSelect实现的简单服务器示例，通过该示例可以了解如何设置监听套接字，并接收客户端连接及数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

// WSAAsynSelect.cpp : 定义应用程序的入口点。
//

#include "stdafx.h"
#include "WSAAsynSelect.h"
#include <WinSock2.h>
#define WM_SOCKET WM_USER+101
#define MAX_LOADSTRING 100
#pragma comment(lib,"ws2_32.lib")
#include <stdio.h>

// 全局变量:
HINSTANCE hInst;								// 当前实例
TCHAR szTitle[MAX_LOADSTRING];					// 标题栏文本
TCHAR szWindowClass[MAX_LOADSTRING];			// 主窗口类名

// 此代码模块中包含的函数的前向声明:
ATOM				MyRegisterClass(HINSTANCE hInstance);
HWND				InitInstance(HINSTANCE, int);
LRESULT CALLBACK	WndProc(HWND, UINT, WPARAM, LPARAM);
INT_PTR CALLBACK	About(HWND, UINT, WPARAM, LPARAM);

int APIENTRY _tWinMain(HINSTANCE hInstance,
                     HINSTANCE hPrevInstance,
                     LPTSTR    lpCmdLine,
                     int       nCmdShow)
{
	UNREFERENCED_PARAMETER(hPrevInstance);
	UNREFERENCED_PARAMETER(lpCmdLine);

 	// TODO: 在此放置代码。
	MSG msg;
	HACCEL hAccelTable;

	// 初始化全局字符串
	LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
	LoadString(hInstance, IDC_WSAASYNSELECT, szWindowClass, MAX_LOADSTRING);
	MyRegisterClass(hInstance);

	HWND hWnd;
	// 执行应用程序初始化:
	if ((hWnd=InitInstance (hInstance, nCmdShow))==NULL)
	{
		return FALSE;
	}

	hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_WSAASYNSELECT));

	WSADATA ws;
	if (WSAStartup(MAKEWORD(2,2),&ws))
	{
		MessageBox(NULL,"WSAStartup Failed","fail",0);
		return 1;
	}
	SOCKET s=socket(AF_INET,SOCK_STREAM,0);
	if (s==INVALID_SOCKET)
	{
		MessageBox(NULL,"socket Failed","failed",0);
		return 1;
	}
	sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
	sin.sin_port=htons(5555);

	if (bind(s,(sockaddr*)&sin,sizeof(sin))==SOCKET_ERROR)
	{
		MessageBox(NULL,"bind failed","fail",0);
		return 1;
	}
	if (listen(s,3)==SOCKET_ERROR)
	{
		MessageBox(NULL,"listen failed","failed",0);
		return 1;
	}
	else
	{
		MessageBox(NULL,"listening...","OK",0);
	}
	WSAAsyncSelect(s,hWnd,WM_SOCKET,FD_ACCEPT|FD_CLOSE);

	// 主消息循环:
	while (GetMessage(&msg, NULL, 0, 0))
	{
		if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))
		{
			TranslateMessage(&msg);
			DispatchMessage(&msg);
		}
	}

	return (int) msg.wParam;
}



//
//  函数: MyRegisterClass()
//
//  目的: 注册窗口类。
//
//  注释:
//
//    仅当希望
//    此代码与添加到 Windows 95 中的“RegisterClassEx”
//    函数之前的 Win32 系统兼容时，才需要此函数及其用法。调用此函数十分重要，
//    这样应用程序就可以获得关联的
//    “格式正确的”小图标。
//
ATOM MyRegisterClass(HINSTANCE hInstance)
{
	WNDCLASSEX wcex;

	wcex.cbSize = sizeof(WNDCLASSEX);

	wcex.style			= CS_HREDRAW | CS_VREDRAW;
	wcex.lpfnWndProc	= WndProc;
	wcex.cbClsExtra		= 0;
	wcex.cbWndExtra		= 0;
	wcex.hInstance		= hInstance;
	wcex.hIcon			= LoadIcon(hInstance, MAKEINTRESOURCE(IDI_WSAASYNSELECT));
	wcex.hCursor		= LoadCursor(NULL, IDC_ARROW);
	wcex.hbrBackground	= (HBRUSH)(COLOR_WINDOW+1);
	wcex.lpszMenuName	= MAKEINTRESOURCE(IDC_WSAASYNSELECT);
	wcex.lpszClassName	= szWindowClass;
	wcex.hIconSm		= LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_SMALL));

	return RegisterClassEx(&wcex);
}

//
//   函数: InitInstance(HINSTANCE, int)
//
//   目的: 保存实例句柄并创建主窗口
//
//   注释:
//
//        在此函数中，我们在全局变量中保存实例句柄并
//        创建和显示主程序窗口。
//
HWND InitInstance(HINSTANCE hInstance, int nCmdShow)
{
   HWND hWnd;

   hInst = hInstance; // 将实例句柄存储在全局变量中

   hWnd = CreateWindow(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,
      CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInstance, NULL);

   if (NULL==hWnd)
   {
      return FALSE;
   }

   ShowWindow(hWnd, nCmdShow);
   UpdateWindow(hWnd);

   return hWnd;
}

//
//  函数: WndProc(HWND, UINT, WPARAM, LPARAM)
//
//  目的: 处理主窗口的消息。
//
//  WM_COMMAND	- 处理应用程序菜单
//  WM_PAINT	- 绘制主窗口
//  WM_DESTROY	- 发送退出消息并返回
//
//
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
	int wmId, wmEvent;
	PAINTSTRUCT ps;
	HDC hdc;

	switch (message)
	{
	case WM_SOCKET:
		{
			if (WSAGETSELECTERROR(lParam))
			{
				closesocket(wParam);
				return 0;
			}
			switch(WSAGETSELECTEVENT(lParam))
			{
			case FD_ACCEPT:
				{
					sockaddr_in sin;
					int len=sizeof(sin);
					SOCKET sock_new=accept(wParam,(sockaddr*)&sin,&len);
					char buf[100];
					memset(buf,0,100);
					sprintf(buf,"来自:%s 的客户端",inet_ntoa(sin.sin_addr));
					MessageBox(NULL,buf,"warnning",MB_OK);
					WSAAsyncSelect(sock_new,hWnd,WM_SOCKET,FD_READ|FD_CLOSE);
					break;
				}
			case FD_READ:
				{
					char cbuf[256];
					memset(cbuf,0,sizeof(cbuf));
					int cRecv=recv(wParam,cbuf,sizeof(cbuf),0);
					MessageBox(NULL,cbuf,"收到的消息",MB_OK);
					char sbuf[]="I am the server";
					send(wParam,sbuf,sizeof(sbuf),0);
					break;
				}
			case FD_CLOSE:
				{
					closesocket(wParam);
					break;
				}
			}

			break;
		}
	case WM_COMMAND:
		wmId    = LOWORD(wParam);
		wmEvent = HIWORD(wParam);
		// 分析菜单选择:
		switch (wmId)
		{
		case IDM_ABOUT:
			DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
			break;
		case IDM_EXIT:
			DestroyWindow(hWnd);
			break;
		default:
			return DefWindowProc(hWnd, message, wParam, lParam);
		}
		break;
	case WM_PAINT:
		hdc = BeginPaint(hWnd, &ps);
		// TODO: 在此添加任意绘图代码...
		EndPaint(hWnd, &ps);
		break;
	case WM_DESTROY:
		PostQuitMessage(0);
		break;
	default:
		return DefWindowProc(hWnd, message, wParam, lParam);
	}
	return 0;
}

// “关于”框的消息处理程序。
INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
	UNREFERENCED_PARAMETER(lParam);
	switch (message)
	{
	case WM_INITDIALOG:
		return (INT_PTR)TRUE;

	case WM_COMMAND:
		if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
		{
			EndDialog(hDlg, LOWORD(wParam));
			return (INT_PTR)TRUE;
		}
		break;
	}
	return (INT_PTR)FALSE;
}

一个比较简单的WSAAsynSelect模型，窗口是vs2008自己生成，直接用，只添加了一点点关于WSAAsynSelect的代码而已