dgrecv2.c

understanding unix/linux programming source code(bruce molay)

/***********************************************************************
 * dgrecv2.c  - datagram receiver
 * 	         usage: dgrecv portnum 
 * 	        action: receives messages, prints them, sends reply
 */

#include	<stdio.h>
#include	<stdlib.h>
#include	<unistd.h>
#include	<string.h>
#include	<sys/types.h>
#include	<sys/socket.h>
#include	<netinet/in.h>

#include	"dgram.h"

#define oops(m,x)  { perror(m);exit(x);}

void say_who_called(struct sockaddr_in *);
void reply_to_sender(int,char *,struct sockaddr_in *, socklen_t);

int main(int ac, char *av[])
{
	int	port;			/* use this port		*/
	int	sock;			/* for this socket		*/
	char	buf[BUFSIZ];		/* to receive data here		*/
	size_t	msglen;			/* store its length here	*/
	struct  sockaddr_in   saddr;	/* put sender's address here	*/
	socklen_t saddrlen;		/* and its length here		*/
	
	if ( ac == 1 || (port = atoi(av[1])) <= 0 ){
		fprintf(stderr,"usage: dgrecv portnumber\n");
		exit(1);
	}

    /*  get a socket and assign it a port number */

	if( (sock = make_dgram_server_socket(port)) == -1 )
		oops("cannot make socket",2);

    /* receive messaages on that socket */

	saddrlen = sizeof(saddr);
	while( (msglen = recvfrom(sock,buf,BUFSIZ,0,
		(struct sockaddr *) &saddr,&saddrlen))>0 ) {
		buf[msglen] = '\0';
		printf("dgrecv: got a message: %s\n", buf);
		say_who_called(&saddr);
		reply_to_sender(sock,buf,&saddr,saddrlen);
	}
	return 0;
}
void 
reply_to_sender(int sock,char *msg,struct sockaddr_in *addrp, socklen_t len)
{
	char	reply[BUFSIZ+BUFSIZ];

	sprintf(reply, "Thanks for your %d char message\n",  strlen(msg));
	sendto(sock,  reply,  strlen(reply), 0, 
			(struct sockaddr *) addrp, len);
}
void say_who_called(struct sockaddr_in *addrp)
{
	char	host[BUFSIZ];
	int	port;

	get_internet_address(host,BUFSIZ,&port,addrp);
	printf("  from: %s:%d\n", host, port);
}