C实现socket 服务端和客户端
一. 服务端和客户端服务流程图二. 服务端的实现1.1 服务端和客户端公用文件tcpSocket.h, tcpSocket.cpp1.2 服务端主程序 server.cpp三 客户端的实现3.1 客户端主程序 client.cpp一. 服务端和客户端服务流程图
二. 服务端的实现
1.1 服务端和客户端公用文件tcpSocket.h, tcpSocket.cpp
#ifndef _TCPSOCKET_H_#define _TCPSOCKET_H#include<WinSock2.h> // windows平台网络库头文件#include<WS2tcpip.h>#pragma comment(lib, "ws2_32.lib") // 库文件#include<stdbool.h>#include<stdio.h>// 服务器端口#define PORT 8899 // [0, 65536)// 封装错误提示#define err(errMsg) printf("[line:%d] %s failed code %d\n", __LINE__, errMsg, WSAGetLastError());// 初始化socketbool init_Socket();// 关闭socketbool close_Socket();// 发送消息bool sendMsg(int fd, const char* msg);// 接收消息bool recvMsg(int fd, char* buf, int bufSize);#endif // !_TCPSOCKET_H_
#include "tcpSocket.h"bool init_Socket(){WSADATA wsadata;if (0 != WSAStartup(MAKEWORD(2, 2), &wsadata)) {err("WSAStartup");return false;}return true;}bool close_Socket(){if (0 != WSACleanup()) {err("WSACleanup");return false;}return true;}// 发送消息bool sendMsg(int fd, const char* msg) {int ret = send(fd, msg, strlen(msg) + 1, 0);if (ret == -1) {err("send msg");return false;}return true;}// 接收消息bool recvMsg(int fd, char* buf, int bufSize) {//printf("buf:%d\n", sizeof(buf)); // 这里的大小为指针的大小int len = recv(fd, buf, bufSize, 0);if (len > 0) {printf("服务端说: %s\n", buf);return true;}else if (len == 0){printf("服务器端已经断开了...\n");}else {perror("recv");}return false;}
1.2 服务端主程序 server.cpp
#include "../tcpSocket/tcpSocket.h"#include <thread>// 客户端信息结构体struct SockInfo {sockaddr_in addr;int fd;};// 客户端工作函数void* working(SockInfo *pinfo);// 客户端数组SockInfo infos[512] = {0 };int main() {// 初始化网络库init_Socket();// 1. 创建空 socket// param1: 地址协议族 ipv4 ipv6// param2: 传输协议类型 流式套接字 数据报// param3: 使用具体的某个传输协议SOCKET fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);if (fd == -1){err("socket");return -1;}// 2. 给socket绑定ip地址和端口号sockaddr_in saddr;saddr.sin_family = AF_INET;saddr.sin_port = htons(PORT);saddr.sin_addr.S_un.S_addr = INADDR_ANY;int ret = bind(fd, (sockaddr*)&saddr, sizeof(saddr));if (ret == -1) {err("bind");return -1;}//3. 监听ret = listen(fd, 10);if (ret == -1){err("listen");return -1;}printf("等待客户端连接...\n");// 初始化存客户端的数组int max = sizeof(infos) / sizeof(infos[0]);for (int i = 0; i < max; ++i){infos[i].fd = -1;}//4. 如果有客户端请求连接 int addrLen = sizeof(sockaddr_in);while (1) {// 为新客户端连接准备存储位置SockInfo* pinfo = nullptr;for (int i = 0; i < max; ++i){if (infos[i].fd == -1) {pinfo = &infos[i];break;}}// 挂起，等待新连接，有新连接则运行int cfd = accept(fd, (struct sockaddr*)&pinfo->addr, &addrLen);if (cfd == -1) {err("accept");break;}pinfo->fd = cfd;// 有客户端连接，启动工作线程std::thread Worker(working, pinfo);Worker.detach();}closesocket(fd);close_Socket();return 0;}/** 打印客户端信息，接收和发送客户端消息*/void* working(SockInfo *pinfo){//SockInfo* pinfo = (SockInfo*)arg;// 连接成功，打印客户端的IP和端口char ip[32];printf("客户端fd: %d, IP：%s, 端口:%d\n",pinfo->fd,inet_ntop(AF_INET, &pinfo->addr.sin_addr.S_un, ip, sizeof(ip)),ntohs(pinfo->addr.sin_port));//5. 可以和客户端进行通信// 从指定的socket接受消息while (1) {char buf[1024];// 接收消息if (recvMsg(pinfo->fd, buf, sizeof(buf))) {sendMsg(pinfo->fd, buf);}}int ret = closesocket(pinfo->fd);if (ret == 0){printf("释放客户端fd：%d\n", pinfo->fd);}pinfo->fd = -1;return NULL;}
三 客户端的实现
3.1 客户端主程序 client.cpp
#include <iostream>#include "../tcpSocket/tcpSocket.h"int main(){init_Socket();// 1. 创建空 socket// param1: 地址协议族 ipv4 ipv6// param2: 传输协议类型 流式套接字 数据报// param3: 使用具体的某个传输协议SOCKET fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);if (fd == -1){err("socket");return -1;}// 2. 与服务器建立连接struct sockaddr_in addr;addr.sin_family = AF_INET;addr.sin_port = htons(PORT);inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.S_un.S_addr);int ret = connect(fd, (sockaddr*)&addr, sizeof(addr));if ( ret == -1){err("connect");return -1;}int number = 0;while (1) {// 给服务器发送消息sendMsg(fd, "你好，我是客户端");// 接收服务器消息char recvBuf[BUFSIZ];memset(recvBuf, 0, sizeof(recvBuf));if (!recvMsg(fd, recvBuf, sizeof(recvBuf))){break;}printf("接收到的消息：%s\n", recvBuf);Sleep(1000);}closesocket(fd);close_Socket();return 0;}
