服务端, 使用缓冲区来接受客户端数据
服务端:
- //vc 的第一套 socket, 第二套 socket, 可能存在冲突
- #define WIN32_LEAN_AND_MEAN
- #define _WINSOCK_DEPRECATED_NO_WARNINGS // 也可以放到工程属性, 预处理中
- #include <Windows.h>
- #include <WinSock2.h>
- #include <stdio.h>
- #pragma comment(lib, "ws2_32.lib")
- enum CMD
- {
- CMD_LOGIN,
- CMD_LOGIN_RET,
- CMD_LOGOUT,
- CMD_LOGOUT_RET,
- CMD_ERROR
- };
- struct DataHeader
- {
- int dataLength;
- CMD cmd;
- };
- struct Login :public DataHeader
- {
- Login()
- {
- dataLength = sizeof(Login);
- cmd = CMD_LOGIN;
- }
- char userName[32];
- char passWord[32];
- };
- struct LoginResult :public DataHeader
- {
- LoginResult()
- {
- dataLength = sizeof(LoginResult);
- cmd = CMD_LOGIN_RET;
- ret = 0;
- }
- int ret;
- };
- struct LogOut :public DataHeader
- {
- LogOut()
- {
- dataLength = sizeof(LogOut);
- cmd = CMD_LOGOUT;
- }
- char userName[32];
- };
- struct LogOutResult :public DataHeader
- {
- LogOutResult()
- {
- dataLength = sizeof(LogOutResult);
- cmd = CMD_LOGOUT_RET;
- ret = 0;
- }
- int ret;
- };
- int main()
- {
- // 启动 Windows socket 2.x 环境
- Word ver = MAKEWORD(2, 2);
- WSADATA dat;
- WSAStartup(ver, &dat);
- //-- 用 Socket API 建立简易 TCP 服务端
- //1. 建立一个 socket
- SOCKET _sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
- //2 bind 绑定用于接受客户端连接的网络端口
- sockaddr_in _sin = {};
- _sin.sin_family = AF_INET;
- _sin.sin_port = htons(4567);//host to.NET unsigned
- _sin.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
- // 使用 127.0.0.1 可以防止外网访问
- // 启用本机全部的 ip 地址可以使用, INADDR_ANY
- if (SOCKET_ERROR == bind(_sock, (sockaddr*)&_sin, sizeof(sockaddr_in)))
- {
- printf("错误, 绑定网络端口失败...\n");
- }
- else
- {
- printf("绑定网络端口成功...\n");
- }
- //3 listen 监听网络端口
- if (SOCKET_ERROR == listen(_sock, 5))
- {
- printf("错误, 监听网络端口失败...\n");
- }
- else
- {
- printf("监听网络端口成功...\n");
- }
- //4 accept 等待接受客户端连接
- sockaddr_in clientAddr = {};
- int nAddrLen = sizeof(clientAddr);
- SOCKET _cSock = INVALID_SOCKET;
- _cSock = accept(_sock, (sockaddr*)&clientAddr, &nAddrLen);
- if (INVALID_SOCKET == _cSock)
- {
- printf("错误, 接受到无效客户端 socket...\n");
- }
- printf("新客户端加入: IP=%s\n", inet_ntoa(clientAddr.sin_addr));
- while (true)
- {
- //5 接收客户端数据
- char szRecv[1024] = {};
- int nLen = recv(_cSock, szRecv, sizeof(DataHeader), 0);
- if (nLen <0)
- {
- printf("客户端已退出, 任务结束.\n");
- break;
- }
- DataHeader* pHeader = (DataHeader*)szRecv;
- switch (pHeader->cmd)
- {
- case CMD_LOGIN:
- {
- recv(_cSock, szRecv+sizeof(DataHeader), pHeader->dataLength - sizeof(DataHeader), 0);
- Login *pLogin = (Login*)szRecv;
- printf("cmd logint,user:%s,password:%s\n", pLogin->userName, pLogin->passWord);
- //----------------------------
- //----------------------------
- LoginResult longinRet = {};
- send(_cSock, (const char*)&longinRet, sizeof(LoginResult), 0);
- }
- break;
- case CMD_LOGOUT:
- {
- recv(_cSock, szRecv + sizeof(DataHeader), pHeader->dataLength - sizeof(DataHeader), 0);
- LogOut *pLogOut = (LogOut*)szRecv;
- printf("cmd logout,user:%s\n", pLogOut->userName);
- //----------------------------
- LogOutResult logOutRet = {};
- send(_cSock, (const char*)&logOutRet, sizeof(LogOutResult), 0);
- }
- break;
- default:
- {
- // 剩余的垃圾数据是否要读取出来?
- if (pHeader->dataLength>0 )
- {
- recv(_cSock, szRecv + sizeof(DataHeader), pHeader->dataLength - sizeof(DataHeader), 0);
- }
- DataHeader header = {};
- header.cmd = CMD_ERROR;
- header.dataLength = 0;
- send(_cSock, (const char*)&header, sizeof(DataHeader), 0);
- }
- break;
- }
- }
- //6 关闭套接字 closesocket
- closesocket(_sock);
- // 清除 Windows socket 环境
- WSACleanup();
- printf("已退出.\n");
- getchar();
- return 0;
- }
客户端:
- #define WIN32_LEAN_AND_MEAN
- #define _WINSOCK_DEPRECATED_NO_WARNINGS // 也可以放到工程属性, 预处理中
- //#define _CRT_SECURE_NO_WARNINGS
- #include <Windows.h>
- #include <WinSock2.h>
- #include <stdio.h>
- #pragma comment(lib, "ws2_32.lib")
- enum CMD
- {
- CMD_LOGIN,
- CMD_LOGIN_RET,
- CMD_LOGOUT,
- CMD_LOGOUT_RET,
- CMD_ERROR
- };
- struct DataHeader
- {
- int dataLength;
- CMD cmd;
- };
- struct Login:public DataHeader
- {
- Login()
- {
- dataLength = sizeof(Login);
- cmd = CMD_LOGIN;
- }
- char userName[32];
- char passWord[32];
- };
- struct LoginResult :public DataHeader
- {
- LoginResult()
- {
- dataLength = sizeof(LoginResult);
- cmd = CMD_LOGIN_RET;
- ret = 1;
- }
- int ret;
- };
- struct LogOut :public DataHeader
- {
- LogOut()
- {
- dataLength = sizeof(LogOut);
- cmd = CMD_LOGOUT;
- }
- char userName[32];
- };
- struct LogOutResult :public DataHeader
- {
- LogOutResult()
- {
- dataLength = sizeof(LogOutResult);
- cmd = CMD_LOGOUT_RET;
- ret = 1;
- }
- int ret;
- };
- int main()
- {
- // 启动 Windows socket 2.x 环境
- Word ver = MAKEWORD(2, 2);
- WSADATA dat;
- WSAStartup(ver, &dat);
- //---------------------------
- //-- 用 Socket API 建立简易 TCP 客户端
- //1 建立一个 socket
- SOCKET _sock = INVALID_SOCKET;
- _sock = socket(AF_INET, SOCK_STREAM, 0);
- if (INVALID_SOCKET == _sock)
- {
- printf("错误, 建立 Socket 失败...\n");
- }
- else
- {
- printf("建立 Socket 成功...\n");
- }
- //2 连接服务器 connect
- sockaddr_in _sin = {};
- _sin.sin_family = AF_INET;
- _sin.sin_port = htons(4567);//host to.NET unsigned
- _sin.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
- int ret = connect(_sock, (sockaddr*)&_sin, sizeof(sockaddr_in));
- if (SOCKET_ERROR == ret)
- {
- printf("错误, 连接服务端失败...\n");
- closesocket(_sock);
- // 清除 Windows socket 环境
- WSACleanup();
- printf("已退出.");
- getchar();
- return 0;
- }
- else
- {
- printf("连接服务端成功...\n");
- }
- while (true)
- {
- //3 输入请求命令
- char cmdBuf[128] = {};
- scanf_s("%s", cmdBuf, 128);
- getchar();
- //4 处理请求
- if (0 == strcmp(cmdBuf, "exit"))
- {
- printf("收到 exit 命令, 任务结束.\n");
- break;
- }
- else if(0 == strcmp(cmdBuf, "login"))
- {
- Login login = {};
- strcpy_s(login.userName, "jj");
- strcpy_s(login.passWord, "ww");
- send(_sock, (const char*)&login, sizeof(Login), 0);
- LoginResult cmd_ret = {};
- recv(_sock, (char*)&cmd_ret, sizeof(LoginResult), 0);
- printf("Login result:%d\n", cmd_ret.ret);
- }
- else if (0 == strcmp(cmdBuf, "logout"))
- {
- LogOut log_out = {};
- strcpy_s(log_out.userName, "jj");
- send(_sock, (const char*)&log_out, sizeof(LogOut), 0);
- LogOutResult cmd_ret = {};
- recv(_sock, (char*)&cmd_ret, sizeof(LogOutResult), 0);
- printf("LogOut result:%d\n", cmd_ret.ret);
- }
- else
- {
- printf("wrong cmd, please input cmd again.\n");
- }
- }
- //7 关闭套接字 closesocket
- closesocket(_sock);
- // 清除 Windows socket 环境
- WSACleanup();
- printf("已退出");
- getchar();
- return 0;
- }
来源: http://www.bubuko.com/infodetail-3716289.html