基于linux socket 的聊天程序

#include<stdio.h> #include<strings.h> #include<unistd.h> #include<sys/types.h> #include<sys/socket.h> #include<netinet/in.h> #include<arpa/inet.h> #define PORT 1234 #define BACKLOG 1 #define MAXDATASIZE 100 void main() { int listenfd, connectfd; struct sockaddr_in server; struct sockaddr_in client; socklen_t addrlen; char buf[MAXDATASIZE]; pid_t pid; if((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { perror("socket() error."); exit(1); } int opt = SO_REUSEADDR; setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)); bzero(&server, sizeof(server)); server.sin_family = AF_INET; server.sin_port = htons(PORT); server.sin_addr.s_addr = htonl(INADDR_ANY); if(bind(listenfd, (struct sockaddr *)&server, sizeof(server)) == -1) { perror("bind() error."); exit(1); } if(listen(listenfd, BACKLOG) == -1) { perror("listen() error."); exit(1); } addrlen = sizeof(client); while(1) { if((connectfd = accept(listenfd, (struct sockaddr *)&client, &addrlen )) == -1) { perror("accept() error."); exit(1); } if((pid = fork()) >0) { while(1) { close(listenfd); recv(connectfd, buf, MAXDATASIZE,0); printf("the client says: %s/n",buf); } } else if(pid == 0) { while(1) { close(listenfd); fgets(buf, MAXDATASIZE, stdin); send(connectfd, buf, MAXDATASIZE, 0); } } else return 1; } close(connectfd); close(listenfd); }