Linux C语言 UDP网络编程

客户端

#include <sys/socket.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <memory.h>
#include <unistd.h>

int main(int argc, char *argv[])
{
	if(argc<3)
	{
		printf("usage: %s ip port\n", argv[0]);
		exit(1);
	}

	// create socket UDP
	int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sockfd<0)
	{
		perror("socket error");
		exit(1);
	}

	// binding socket
	struct sockaddr_in serveraddr;
	memset(&serveraddr, 0, sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;                // IPV4
	serveraddr.sin_port = htons(atoi(argv[2]));     // port
	inet_pton(AF_INET, argv[1], &serveraddr.sin_addr.s_addr);  // ip

	if(connect(sockfd, (struct sockaddr*)&serveraddr,sizeof(serveraddr))<0)
	{
		perror("connect error");
		exit(1);
	}

	char buffer[1024] = "hello iotek";

	/*if(sendto(sockfd, buffer, sizeof(buffer), 0, 
		(struct sockaddr*)&serveraddr, sizeof(serveraddr)) <0)*/
	if(send(sockfd, buffer, sizeof(buffer), 0)<0)
	{
		perror("sendto error");
		exit(1);
	}
	else
	{
		memset(buffer, 0, sizeof(buffer));
		size_t size;
		// accept message
		if(recv(sockfd, buffer, sizeof(buffer), 0) <0)
		{
			perror("recv error");
			exit(1);
		}
		else
		{
			printf("%s", buffer);
		}
	}

	close(sockfd);
	return 0;
}

服务器

#include <sys/socket.h>
#include <netdb.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <signal.h>
#include <time.h>

int sockfd;

void sig_handler(int signo)
{
	if(signo == SIGINT)
	{
		printf("server close\n");
		close(sockfd);
		exit(1);
	}
}

void out_addr(struct sockaddr_in *clientaddr)
{
	char ip[16];
	int port;
	memset(ip, 0, sizeof(ip));
	inet_ntop(AF_INET, &clientaddr->sin_addr.s_addr,
			ip, sizeof(ip));
        port = ntohs(clientaddr->sin_port);
	printf("client: %s(%d)\n", ip, port);


}

void do_service()
{
	struct sockaddr_in clientaddr;
	socklen_t len = sizeof(clientaddr);
	char buffer[1024];
	memset(buffer, 0, sizeof(buffer));
	// accept message
	if(recvfrom(sockfd, buffer, sizeof(buffer), 0,
		(struct sockaddr*)&clientaddr, &len)<0)
	{
		perror("recvfrom error");
	}
	else
	{
		out_addr(&clientaddr);
		printf("client send into: %s\n", buffer);
		// send message

		long int t = time(0);
		char *ptr = ctime(&t);
		size_t size = strlen(ptr) * sizeof(char);
		if(sendto(sockfd, ptr, size, 0,
			(struct sockaddr*)&clientaddr, len) <0)
		{
			perror("sendto error");
		}

	}


}

int main(int argc, char *argv[])
{
	if(argc<2)
	{
		printf("usage: %s port\n", argv[0]);
		exit(1);
	}

	if(signal(SIGINT, sig_handler) == SIG_ERR)
	{
		perror("signal sigint error");
		exit(1);
	}

	// udp
	// create socket
	sockfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sockfd<0)
	{
		perror("socket error");
		exit(1);
	}
	int ret;
	int opt=1;
	//
	if((ret = setsockopt(sockfd, SOL_SOCKET,
		SO_REUSEADDR, &opt, sizeof(opt)))<0)
	{
		perror("setsockopt error");
		exit(1);
	}

	// bingding socket
	struct sockaddr_in serveraddr;
	memset(&serveraddr, 0, sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;           // IPV4
	serveraddr.sin_port = htons(atoi(argv[1]));// port	
	serveraddr.sin_addr.s_addr = INADDR_ANY;   // ip
	if(bind(sockfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr))<0)
	{
		perror("bind error");
		exit(1);
	}
	while(1)
	{
		do_service();
	}
	return 0;

}