一个简单的socket通信聊天程序

知识点：socket通信 多线程的使用，以便于同时进行发送和读取操作

 

编译

服务端serv.c    客户端client.c 

g++ -o serv serv.c -lpthread

g++ -o client client.c  -lpthread

 

分别在服务端和客户端启动进程

./serv

 

客户端要加上服务端的ip作为参数，端口在程序里面有默认值

./client 192.168.1.102

 

代码很粗糙，从socket通信的代码上进行的修改

 

serv.c 

#include <stdio.h> #include <stdlib.h> #include <errno.h> #include <string.h> #include <sys/types.h> #include <netinet/in.h> #include <sys/socket.h> #include <sys/wait.h> #include <string.h> #include <arpa/inet.h> #include <pthread.h> //#include <unistd.h> #define MYPORT 3490 /* the port users will be connecting to */ #define BACKLOG 10 /* how many pending connections queue will hold */ #define STRSIZE 1000 char strEcho[1000] = {0}; void * msend(void* p) { int new_fd = *(int*)p; while(1) { fgets(strEcho, 1000, stdin); printf("=========/n", new_fd); write(new_fd, strEcho, strlen(strEcho), 0); } } main() { // char strEcho[1000] = {0}; int sockfd, new_fd; /* listen on sock_fd, new connection on new_fd */ struct sockaddr_in my_addr; /* my address information */ struct sockaddr_in their_addr; /* connector's address information */ int sin_size; int len; char str[STRSIZE] = {0}; int count = 0; if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { perror("socket"); exit(1); } my_addr.sin_family = AF_INET; /* host byte order */ my_addr.sin_port = htons(MYPORT); /* short, network byte order */ my_addr.sin_addr.s_addr = INADDR_ANY; /* auto-fill with my IP */ bzero(&(my_addr.sin_zero), 8); /* zero the rest of the struct */ if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof(struct sockaddr)) / == -1) { perror("bind"); exit(1); } if (listen(sockfd, BACKLOG) == -1) { perror("listen"); exit(1); } printf("服务端启动/n"); while(1) { /* main accept() loop */ sin_size = sizeof(struct sockaddr_in); if ((new_fd = accept(sockfd, (struct sockaddr *)&their_addr, / &sin_size)) < 0 ) { perror("accept"); continue; } printf("server: got connection from %s/n", / inet_ntoa(their_addr.sin_addr)); if (!fork()) { if (write(new_fd, "Hello, world!/n", 14, 0) == -1) perror("send"); /* bzero(str, STRSIZE); len = recv(new_fd, str, STRSIZE, 0); if (len < 0) { printf("failed/n"); } else { printf("验证通过/n"); } */ pthread_t id; pthread_create(&id, NULL, msend, &new_fd); while (1) { // write(new_fd, strEcho, strlen(strEcho), 0); bzero(str, STRSIZE); len = read(new_fd, str, STRSIZE, 0); printf("///////"); if (len > 0) { printf("接收到:%s from %s/n", str, inet_ntoa(their_addr.sin_addr)); /* write(new_fd, strEcho, strlen(strEcho), 0);*/ } else { printf("fail"); break; } if (strcmp(str, "bye") == 0) { break; } } close(new_fd); exit(0); } close(new_fd); /* parent doesn't need this */ while(waitpid(-1,NULL,WNOHANG) > 0); /* clean up child processes */ } }

 

 

client.c

#include <stdio.h> #include <stdlib.h> #include <errno.h> #include <string.h> #include <netdb.h> #include <sys/types.h> #include <netinet/in.h> #include <sys/socket.h> #include <pthread.h> //#include <unistd.h> #define PORT 3490 /* the port client will be connecting to */ #define MAXDATASIZE 1000 /* max number of bytes we can get at once */ char strEcho[1000] = {0}; void * mread(void* p) { int new_fd = *(int*)p; // fgets(strEcho, 1000, stdin); int len = 0; while (1) { bzero(strEcho, 1000); len = read(new_fd, strEcho, 1000, 0); if(len < 0) { exit(-2); } strEcho[len] = '/0'; printf("/nReceived:==========/n%s/n", strEcho); } } int main(int argc, char *argv[]) { int sockfd, numbytes; char buf[MAXDATASIZE]; struct hostent *he; struct sockaddr_in their_addr; /* connector's address information */ int len; if (argc != 2) { fprintf(stderr,"usage: client hostname/n"); exit(1); } if ((he = gethostbyname(argv[1])) == NULL) { /* get the host info */ herror("gethostbyname"); exit(1); } if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { perror("socket"); exit(1); } their_addr.sin_family = AF_INET; /* host byte order */ their_addr.sin_port = htons(PORT); /* short, network byte order */ their_addr.sin_addr = *((struct in_addr *)he->h_addr); bzero(&(their_addr.sin_zero), 8); /* zero the rest of the struct */ if (connect(sockfd, (struct sockaddr *)&their_addr, / sizeof(struct sockaddr)) == -1) { perror("connect"); exit(1); } if ((numbytes = read(sockfd, buf, MAXDATASIZE, 0)) == -1) { perror("recv"); exit(1); } buf[numbytes] = '/0'; printf("Received: %s",buf); pthread_t id; pthread_create(&id, NULL, mread, &sockfd); while (1) { bzero(buf, MAXDATASIZE); printf("put in the massage:"); // scanf("%s", buf); fgets(buf, MAXDATASIZE, stdin); len = write(sockfd, buf, strlen(buf), 0); if (len > 0) { printf("send/n"); // if (numbytes = read(sockfd, buf, MAXDATASIZE, 0) == -1) // {exit(1);} // buf[numbytes] = '/0'; // printf("%d hi %s",numbytes, buf); } else { printf("fail/n"); break; } } printf("client close/n"); close(sockfd); return 0; }