socket服务端和客户端实现双发聊天
相关函数
(1)socket()函数
int socket(int domain, int type, int protocol);
(2)bind()函数: IP号端口号与相应描述字赋值函数
int bind(int sockfd, const struct sockaddr *addr,socklen t addrlen);
(3)listen()函数:监听设置函数
int listen(int sockfd, int backlog);
(4)accept()函数
int accept(int sockfd, struct sockaddr *addr, socklen t *addrlen);
(5)fgets()函数
char *fgets(char *s, int size, FILE *stream);
在Linux编译C程序时使用gets,会出现:
warning: the `gets' function is dangerous and should not be used.
问题出在程序中使用了 gets ,Linux 下gcc编译器不支持这个函数,解决办法是使用 fgets
(6)atoi()函数
atoi()是C语言中的字符串转换成整型数的一个函数
(1)【头文件】#include <stdlib.h>
(2)【函数原型】int atoi (const char * str);
(3)【函数说明】atoi() 函数会扫描参数 str 字符串,跳过前面的空白字符(例如空格,tab缩进等),直到遇上数字或正负符号才开始做转换,而再遇到 非数字 或 字符串结束时(’\0’) 才结束转换,并将结果返回。函数返回转换后的整型数;如果 str 不能转换成 int 或者 str 为空字符串,那么将返回 0。
(7)itoa()函数
itoa()函数是C语言中的整型数转换成字符串的一个函数
(1)【头文件】#include <cstdlib>
(2)【函数原型】char *itoa(int value, char *string, int radix);
(3)【参数说明】
value:要转换的数据。
string:目标字符串的地址。
radix:转换后的进制数,可以是10进制、16进制等,范围必须在 2-36。
先写服务端代码:
#include <stdio.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <stdlib.h> #include <string.h> #include <unistd.h> int main(int argc,char **argv) { int s_fd; int c_fd; int n_read; char readBuf[128]; char msg[128] = {0}; struct sockaddr_in s_addr; struct sockaddr_in c_addr; if(argc != 3) { printf("param is not good\n"); exit(-1); } memset(&s_addr,0,sizeof(struct sockaddr_in)); memset(&c_addr,0,sizeof(struct sockaddr_in)); //1.socket s_fd = socket(AF_INET,SOCK_STREAM,0); if(s_fd == -1) { perror("socket"); exit(-1); } s_addr.sin_family = AF_INET; s_addr.sin_port = htons(atoi(argv[2])); inet_aton(argv[1],&s_addr.sin_addr); //2.bind bind(s_fd,(struct sockaddr *)&s_addr,sizeof(struct sockaddr_in)); //3.listen listen(s_fd,10); //4.accept int clen = sizeof(struct sockaddr_in); while(1) { c_fd = accept(s_fd,(struct sockaddr *)&c_addr,&clen); if(c_fd == -1) { perror("accept"); } printf("get connect:%s\n",inet_ntoa(c_addr.sin_addr)); if(fork() == 0) { if(fork() == 0) { while(1) { memset(msg,0,sizeof(msg)); printf("input:"); fgets(msg,sizeof(msg),stdin); write(c_fd,msg,strlen(msg)); } } //5.read while(1) { memset(readBuf,0,sizeof(readBuf)); n_read = read(c_fd,readBuf,128); if(n_read == -1) { perror("read"); } else { printf("get message:%d,%s\n",n_read,readBuf); } } break; } } return 0; }
再写客户端代码:
#include <stdio.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <stdlib.h> #include <string.h> #include <unistd.h> int main(int argc,char **argv) { int c_fd; int n_read; char readBuf[128]; // char *msg = "msg form client"; char msg[128] = {0}; struct sockaddr_in c_addr; memset(&c_addr,0,sizeof(struct sockaddr_in)); if(argc != 3) { printf("parm is not good\n"); exit(-1); } //1.socket c_fd = socket(AF_INET,SOCK_STREAM,0); if(c_fd == -1) { perror("socket"); exit(-1); } c_addr.sin_family = AF_INET; c_addr.sin_port = htons(atoi(argv[2])); inet_aton(argv[1],&c_addr.sin_addr); //2.connect if(connect(c_fd,(struct sockaddr *)&c_addr,sizeof(struct sockaddr)) == -1) { perror("connect"); exit(-1); } while(1) { if(fork() == 0) { while(1) { memset(msg,0,sizeof(msg)); printf("input:"); fgets(msg,sizeof(msg),stdin); write(c_fd,msg,strlen(msg)); } } while(1) { memset(readBuf,0,sizeof(readBuf)); n_read = read(c_fd,readBuf,128); if(n_read == -1) { perror("read"); } else { printf("get message form server:%d,%s\n",n_read,readBuf); } } } return 0; }
编译结果:
实现了双方聊天