/*****************************************************************************
 * receive.c - partner to "send.c". Opens a port and reports the port number
 * to standard error. A port number can be specified which overrides the
 * automatic selection. Any data arriving on this port is then copied to
 * standard out and the process exits.
 *
 * Phil Gibbs - Trinem Consulting (pgibbs@trinem.co.uk)
 *****************************************************************************/

#include <sys/select.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <netinet/in.h>
#include <netdb.h>
#include <stdio.h>
#include <signal.h>
#include <errno.h>
#include <rpc/xdr.h>
#include <assert.h>

int EstablishIncomingConnection(int *base_port)
{
       int sock;
       struct sockaddr_in server;
       int length;

       sock = socket(AF_INET, SOCK_STREAM, 0);
       assert(sock>0);
       server.sin_family = AF_INET;
       server.sin_addr.s_addr = INADDR_ANY;
       server.sin_port = htons(*base_port);
       assert(!bind(sock, (struct sockaddr *)&server, sizeof(server)));
       length = sizeof(server);
       assert(!getsockname(sock, (struct sockaddr *)&server, &length));
       listen(sock, 5);   
       *base_port=htons(server.sin_port);
       return sock;
}

void CopySocketToStandardOut(int msgsock)
{
	char	Buffer[2048];
	int	BytesRead;
	int	res;

	while (BytesRead=read(msgsock,Buffer,sizeof(Buffer)))
	{
		res=write(1,Buffer,BytesRead);
		if (res<BytesRead)
		{
			if (res!=-1) fprintf(stderr,"Failed to write\n");
			else perror("Error writing");
			exit(1);
		}
	}
}

int WaitForConnectionRequest(int sock)
{
	int msgsock;

	msgsock=accept(sock,0,0);
	assert(msgsock != -1);
	fprintf(stderr,"Receiving...\n");
	CopySocketToStandardOut(msgsock);
	close(msgsock);
}


main(int argc,char **argv)
{
	int Port;
	int Socket;
	Port=(argc>1)?atol(argv[1]):0;
	Socket=EstablishIncomingConnection(&Port);
	fprintf(stderr,"Listening on port %d\n",Port);
	WaitForConnectionRequest(Socket);
}