Linux下網絡編程-簡易poll服務器和客戶端

簡易poll服務器:

//編譯命令：g++ -g xxx.cpp -o xxx
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <poll.h>int main(int argc, char *argv[])
{// 1.創建套接字int serv_sock=socket(PF_INET, SOCK_STREAM, 0);struct sockaddr_in serv_adr;memset(&serv_adr, 0, sizeof(serv_adr));serv_adr.sin_family=AF_INET;serv_adr.sin_addr.s_addr=htonl(INADDR_ANY);serv_adr.sin_port=htons(atoi(argv[1]));// 2. 綁定 ip, portif(bind(serv_sock, (struct sockaddr*) &serv_adr, sizeof(serv_adr))==-1)error_handling("bind() error");// 3. 監聽if(listen(serv_sock, 5)==-1)error_handling("listen() error");//4 等待連接->循環//檢測->都緩沖區，委托內核去處理//數據初始化，創建自定義得文件描述集struct pollfd fds[1024];//初始化for(int i=0;i<1024;i++){fds[i].fd=-1;fds[i].events=POLLIN;}fds[0].fd=serv_sock;int maxfd=0;while (1) {//委托內核檢測if(poll(fds, maxfd+1, -1)==-1){perror("select");exit(0);}//檢測的度緩沖區有變化//有新連接if(fds[0].revents&POLLIN){// 接收連接請求struct sockaddr_in sockcli;socklen_t len = sizeof(sockcli);// 這個accept是不會阻塞的int connfd = accept(serv_sock, (struct sockaddr*)&sockcli, &len);//委托內核檢測connfd的讀緩沖區int i;for(i=0;i<1024;i++){if(fds[i].fd==-1){fds[i].fd=connfd;break;}}maxfd=i>maxfd?i:maxfd;}//通信。有客戶端發送數據過來for(int i=1;i<=maxfd;i++){//如果在集合中，說明都緩沖區有數據if(fds[i].revents&POLLIN){char buf[128];int ret = read(fds[i].fd, buf, sizeof(buf));if(ret == -1){perror("read");exit(0);}else if(ret == 0){printf("對方已經關閉了連接...\n");close(fds[i].fd);fds[i].fd = -1;}else{printf("客戶端say: %s\n", buf);write(fds[i].fd, buf, strlen(buf)+1);}}}}close(serv_sock);return 0;
}

簡易客戶端:

//編譯命令：g++ -g xxx.cpp -o xxx
#include <netinet/in.h>
#include <cstdio>
#include <iostream>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/select.h>
#include <poll.h>
using namespace std;int main()
{//創建socket對象int Client = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);if (Client == -1)//判斷是否創建成功{cout << "創建失敗" << endl;close(Client);return -1;}const char* ip = "127.0.0.1";//保存ip地址unsigned short port=9527;//保存端口//設置服務端sockaddr_in addrServer;addrServer.sin_addr.s_addr = inet_addr(ip);addrServer.sin_port = htons(port);addrServer.sin_family = AF_INET;//連接if (connect(Client, (sockaddr*)&addrServer, sizeof(addrServer))== -1){cout << "連接失敗" << endl;close(Client);return -1;}cout << "連接服務器成功" << endl;//接收和發送數據char rec[1024];char se[1024];while(1){memset(rec, 0, sizeof(rec));memset(se, 0, sizeof(se));scanf("%s", se);if (strcmp(se, "quit") == 0 || strcmp(se, "exit") == 0)break;if (send(Client, se, sizeof(se), 0) < 0)break;cout << "發送：" << se << endl;if (recv(Client, rec, sizeof(rec), 0) < 0)break;cout << "接收：" << rec << endl;}cout << "已與服務端斷開連接" << endl;close(Client);return 0;
}