socket编程实现的多人局域网聊天(udp)多线程

socket编程实现的多人局域网聊天(udp)多线程

服务器端:

/** File Name: socket_udp_server.c* Author: eagle* mail: zhaocheng121@163* Created Time: 2015-10-08 18:43*/#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <net/if.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>char Name[100][40];int findclient(struct sockaddr_in client_address, struct sockaddr_in client[], int *client_num, char name[]);int main(int argc, char *argv[])
{int server_socketfd, client_socketfd;int server_len,client_len;struct sockaddr_in server_address;struct sockaddr_in client_address;struct sockaddr_in client[100];//记录客户端信息int i, byte, client_num = 0;char recv[100], send[100];int flag;<pre name="code" class="cpp">        /*hostip 本机ip地址*/int inet_sock;struct ifreq ifr;char hostip[15];inet_sock = socket(AF_INET, SOCK_DGRAM,0);strcpy(ifr.ifr_name, "wlan0");if (ioctl(inet_sock, SIOCGIFADDR, &ifr) < 0)perror("ioctl");strcpy(hostip, inet_ntoa(((struct sockaddr_in *)&(ifr.ifr_addr))->sin_addr));/*上面部分为获取本机ip*/

/*create socket and bind*/server_socketfd = socket(AF_INET, SOCK_DGRAM, 0); //创建socket对象server_address.sin_family = AF_INET;              server_address.sin_addr.s_addr = inet_addr("192.168.99.184"); //主机ipserver_address.sin_port = 9734;                     //主机端口server_len = sizeof(server_address);bind(server_socketfd, (struct sockaddr *) & server_address, server_len);client_len = sizeof(struct sockaddr_in);memset(&client_address, 0, sizeof(struct sockaddr_in));
while(1){if((byte = recvfrom(server_socketfd, &recv, sizeof(recv), 0,  //从客户端接受消息(struct sockaddr *)&client_address, //存储客户端地址信息(socklen_t *)&client_len)) == -1){perror("recvfrom");exit(EXIT_FAILURE);}flag = findclient(client_address, client, &client_num, recv);
//	printf("%dn",flag);if(flag == -1) {strcpy(send, "Welcome to chating room");if((byte = sendto(server_socketfd, &send, sizeof(send), 0, (struct sockaddr *)&client_address,(socklen_t)client_len)) == -1){perror("send");exit(EXIT_FAILURE);}} //给刚加入的客户端发送确认信息//printf("recv:%sn", recv);//printf("send:");//scanf("%s", send);else {strcpy(send, Name[flag]);strcat(send, ":");strcat(send, recv);for(i = 0; i < client_num; i++){//	if(client[i].sin_port != client_address.sin_port) {if((client[i].sin_addr.s_addr != client_address.sin_addr.s_addr) || (client[i].sin_port != client_address.sin_port)) {client_len = sizeof(client[i]);if((byte = sendto(server_socketfd, &send, sizeof(send), 0, (struct sockaddr *)&client[i],(socklen_t)client_len)) == -1){perror("send");exit(EXIT_FAILURE);}}}}
}
//	shutdown(client_socketfd, 2);
//	shutdown(server_socketfd, 2);return 0;
}int findclient(struct sockaddr_in client_address, struct sockaddr_in client[], int *client_num, char name[])
{int i;if((*client_num) == 0) {
//		client[0].sin_family = client_address.sin_family;
//		client[0].sin_port = client_address.sin_port;client[0] = client_address;strcpy(Name[0], name);(*client_num)++;return -1; //没有客户端信息}else {for(i = 0; i < (*client_num); i++) {if(client[i].sin_addr.s_addr == client_address.sin_addr.s_addr && client[i].sin_port == client_address.sin_port)return i; //列表中有客户端的信息}client[i] = client_address;strcpy(Name[i], name);(*client_num)++;return -1; //没有客户信息}
}

<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);">客户端:(本机测试时需将hostip改为127.0.0.1</span>

/** File Name: socket_udp_server.c* Author: eagle* mail: zhaocheng121@163* Created Time: 2015-10-08 18:43*/#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <net/if.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>char name[40];void first_message(int socketfd, struct sockaddr_in server_address, int server_len);int main(int argc, char *argv[])
{int socketfd;int server_len,client_len;struct sockaddr_in server_address;struct sockaddr_in client_address;int i,byte;char recv[100], send[100];int port;srand(time(0));port = 9000 + rand() % 1000;
//char recv, send;/*hostip 本机ip地址*/int inet_sock;struct ifreq ifr;char hostip[15];inet_sock = socket(AF_INET, SOCK_DGRAM,0);strcpy(ifr.ifr_name, "wlan0");if (ioctl(inet_sock, SIOCGIFADDR, &ifr) < 0)perror("ioctl");strcpy(hostip, inet_ntoa(((struct sockaddr_in *)&(ifr.ifr_addr))->sin_addr));/*上面部分为获取本机ip*/socketfd = socket(AF_INET, SOCK_DGRAM, 0); //创建socket对象server_address.sin_family = AF_INET;              server_address.sin_addr.s_addr = inet_addr("192.168.99.184"); //主机ip(需要自己设置)server_address.sin_port = 9734;                     //主机端口server_len = sizeof(server_address);client_address.sin_family = AF_INET;              client_address.sin_addr.s_addr = inet_addr(hostip); client_address.sin_port = htons(port);                    client_len = sizeof(client_address);bind(socketfd, (struct sockaddr *) & client_address, client_len);/*第一次连接时向服务器发送识别信息*/first_message(socketfd, server_address, server_len);if(fork()==0) {while(1) {//	printf("send:");//scanf("%s", send);gets(send);if((byte = sendto(socketfd, &send, sizeof(send), 0, (struct sockaddr *)&server_address,(socklen_t)server_len)) == -1){perror("send");exit(EXIT_FAILURE);}}}else {while(1) {if((byte = recvfrom(socketfd, &recv, sizeof(recv), 0,  //从客户端接受消息(struct sockaddr *)&server_address, //存储客户端地址信息(socklen_t *)&server_len)) == -1){perror("recvprom");exit(EXIT_FAILURE);}printf("recv:");puts(recv);printf("n");}}shutdown(socketfd, 2);return 0;
}void first_message(int socketfd, struct sockaddr_in server_address, int server_len)
{char name[40];//struct message mess;printf("please input your name: ");gets(name);//puts(mess.ID);if((sendto(socketfd, &name, sizeof(name), 0, (struct sockaddr *)&server_address,(socklen_t)server_len)) == -1){perror("send");exit(EXIT_FAILURE);}
}