echo_server.c
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <arpa/inet.h> #include <sys/socket.h> void error_handling(char *message); int main(int argc, char *argv[]) { int serv_sock; int clnt_sock; struct sockaddr_in serv_addr; struct sockaddr_in clnt_addr; socklen_t clnt_addr_size; int str_len; char message[1024]; if(argc!=2) { exit(1); } //TCP socket //1.安装电话机 serv_sock=socket(PF_INET, SOCK_STREAM, 0); if(serv_sock == -1) error_handling("socket error!"); //2.调用bind分配电话号码 memset(&serv_addr, 0, sizeof(serv_addr)); serv_addr.sin_family = AF_INET; //IPV4协议族 //INADDR_ANY表示自动获取服务器端IP地址, //如果计算机只有一个NIC(网络接口卡),则可以这么配置,如果有多个,则需要手动配置 //服务器IP地址的数量与NIC数量有关 serv_addr.sin_addr.s_addr = htonl(INADDR_ANY); //主机字节序(host)转换成网络字节序(net)(大端序) serv_addr.sin_port = htons(atoi(argv[1])); //端口号 //表示请把进入IP...、9190端口的数据传给我 if(bind(serv_sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1) error_handling("bind error"); //3.调用listen连接电话线(进入等待连接请求状态) //5表示连接请求等待队列为5 if(listen(serv_sock, 5) == -1) error_handling("listen error"); //4.调用accept拿起话筒(受理连接请求) clnt_addr_size = sizeof(clnt_addr); int i; for(i=0;i<5;i++) { //服务器调用accept函数时进入阻塞状态,直到有请求进入 //第二个参数保存发起连接请求的客户端地址 clnt_sock=accept(serv_sock, (struct sockaddr*) &clnt_addr, &clnt_addr_size); if(clnt_sock == -1) error_handling("accept error"); else printf("connect client: %d \n", i+1); //read函数返回接受的字节数(成功时),遇到文件结尾返回0,失败时返回-1 //对于一个套接字的read和write默认是阻塞的,有数据时才会有返回 while((str_len=read(clnt_sock, message, 1024))!=0) { //由于线程阻塞,所以在while循环中不能进行printf显示,跳出循环后才会显示 //printf("%d", str_len); write(clnt_sock, message, str_len); } close(clnt_sock); } close(serv_sock); return 0; } void error_handling(char *message) { fputs(message, stderr); fputc('\n', stderr); exit(1); }
echo_client.c
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <arpa/inet.h> #include <sys/socket.h> #define BUF_SIZE 1024 void error_handling(char *message); int main(int argc, char *argv[]) { int sock; struct sockaddr_in serv_addr; int str_len; char message[BUF_SIZE]; if(argc!=3) { exit(1); } sock=socket(PF_INET, SOCK_STREAM, 0); if(sock == -1) error_handling("socket error!"); memset(&serv_addr, 0, sizeof(serv_addr)); serv_addr.sin_family = AF_INET; //基于字符串的IP地址初始化 //也可以使用 int inet_aton(const char* string, struct in_addr* addr); //反向转换 char* inet_ntoa(struct in_addr adr); serv_addr.sin_addr.s_addr = inet_addr(argv[1]); serv_addr.sin_port = htons(atoi(argv[2])); //向服务器发起连接请求 //注意:客户端IP地址在调用connect时会自动分配 if(connect(sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1) error_handling("connect error"); else puts("Connected....."); while(1) { fputs("Input Message: ", stdout); fgets(message, BUF_SIZE, stdin); if(!strcmp(message, "q\n") || !strcmp(message, "Q\n")) break; // //存在问题:多次调用write函数传递的字符串有可能一次性传输到服务器 // //客户端有可能在尚未受到全部数据包时就调用read函数 // write(sock, message, strlen(message)); // str_len=read(sock, message, BUF_SIZE-1); //记录接收字符的大小有效解决一次读取不全的问题 str_len=write(sock, message, strlen(message)); int recv_len=0; while(recv_len<str_len) { int recv_cnt = read(sock, &message[recv_len], BUF_SIZE-1); if(recv_cnt==-1) error_handling("read error"); recv_len+=recv_cnt; } message[str_len]=0; printf("Message from server: %s", message); } close(sock); return 0; } void error_handling(char *message) { fputs(message, stderr); fputc('\n', stderr); exit(1); }