群聊程序的linux代码实现

#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <pthread.h>

#define BUFLEN 255

struct sockaddr_in addr,peeraddr;
int sockfd;

void * thr_send(void *arg)
{
	char sendmsg[BUFLEN + 1];
	
	while(1)
	{
		/* 接受用户输入 */
		bzero(sendmsg, BUFLEN + 1);
		if (fgets(sendmsg, BUFLEN, stdin) == (char *) EOF)
			exit(0);
		/* 发送消息 */
		if (sendto(sockfd, sendmsg, strlen(sendmsg), 0,(struct sockaddr *) &addr,sizeof(struct sockaddr_in)) < 0) 
		{
			printf("sendto error!\n");
			exit(3);
		}
	}
}

void * thr_recv(void *arg)
{
	char recvmsg[BUFLEN + 1];
	unsigned int socklen, n;
	
	socklen = sizeof(struct sockaddr_in);
	while(1)
	{
		bzero(recvmsg, BUFLEN + 1);
		n = recvfrom(sockfd, recvmsg, BUFLEN, MSG_DONTWAIT,(struct sockaddr *) &peeraddr, &socklen);
		if(n != -1)
		{
			/* 成功接收到数据报 */
			recvmsg[n] = 0;
			printf("n = %d, socklen = %d", n, socklen);
			printf("%s:%s", inet_ntoa(peeraddr.sin_addr),recvmsg);
		}
	}
}

int main(void)
{
	unsigned int socklen;	
	struct ip_mreq mreq; 
	char loop;
	pthread_t tidSend, tidRecv;
	int err;
	
	sockfd = socket(AF_INET, SOCK_DGRAM, 0);
	/* 设置要加入组播的地址 */
	if (inet_pton(AF_INET, "224.0.1.1", &mreq.imr_multiaddr.s_addr) <= 0) 
	{
		printf("wrong group address!\n");
		exit(0);
	}

	/* 设置发送组播消息的源主机的地址信息 */
	mreq.imr_interface.s_addr = htonl(INADDR_ANY);

	/* 把本机加入组播地址，即本机网卡作为组播成员，只有加入组才能收到组播消息 */
	if (setsockopt(sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq,sizeof(struct ip_mreq)) == -1) 
	{
		perror("setsockopt error: IP_ADD_MEMBERSHIP");
		exit(-1);
	}
	loop = 0;
	if (setsockopt(sockfd, IPPROTO_IP, IP_MULTICAST_LOOP, &loop,sizeof(loop)) == -1) 
	{
		perror("setsockopt error: IP_MULTICAST_LOOP");
		exit(-1);
	}
	
	socklen = sizeof(struct sockaddr_in);
	memset(&addr, 0, socklen);
	addr.sin_family = AF_INET;
	addr.sin_port = htons(23456);
	if (inet_pton(AF_INET, "224.0.1.1", &addr.sin_addr) <= 0) 
	{
		printf("ip address error!\n");
		exit(0);
	}
	
	/* 绑定端口和IP信息到socket上 */
	if (bind(sockfd, (struct sockaddr *) &addr,sizeof(struct sockaddr_in)) == -1) 
	{
		printf("Bind error\n");
		exit(0);
	}	

	socklen = sizeof(struct sockaddr_in);

	/* 设置对方的端口和IP信息 */
	memset(&peeraddr, 0, socklen);
	peeraddr.sin_family = AF_INET;
	
	err = pthread_create(&tidSend, NULL, thr_send, NULL);
	if (err != 0)
	{
		printf("can't create thread send: %s\n", strerror(err));
		exit(-1);
	}
	err = pthread_create(&tidRecv, NULL, thr_recv, NULL);
	if (err != 0)
	{
		printf("can't create thread recv: %s\n", strerror(err));
		exit(-1);
	}
	err = pthread_join(tidSend, NULL);
	if (err != 0)
	{
		printf("can't join thread send: %s\n", strerror(err));
	}
	err = pthread_join(tidRecv, NULL);
	if (err != 0)
	{
		printf("can't join thread recv: %s\n", strerror(err));
	}
	
	return 0;
}