socket編程實現的多人局域網聊天(udp)多線程

服務器端:

/*
  * File Name: socket_udp_server.c
  * Author: eagle
  * mail: [email protected]
  * 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"); //主機ip
	server_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("%d\n",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:%s\n", 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: [email protected]
  * 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);
			}
}