pcas/src/dbtools/PVSserver.c

/* only runable on work station now */

#include "PVS.h"

#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>

struct PVSnode
{
	BSDATA info;
	int alive;
	struct PVSnode* next;
};
typedef struct PVSnode PVSNODE;

static PVSNODE* ioc_list = (PVSNODE*)NULL;

static int read_pvs(BSDATA* info,int serv,char* sname);

#ifndef PVS_SERVER_PROG
int main(int argc,char** argv)
{
	BSDATA info;
	int serv;
	int rc;

	if(argc<4)
	{
		fprintf(stderr,"usage: %s IOC-name server-number [server-name]\n",
			argv[0]);
		return -1;
	}

	serv=atoi(argv[2]);
	BSsetAddress(&info,argv[1]);

	if(serv>PVS_LAST_VERB)
		rc=read_pvs(&info,serv,argv[3]);
	else
		rc=read_pvs(&info,serv,(char*)NULL);

	if(rc<0) fprintf(stderr,"read of data failed horribly\n");

	return 0;
}
#else
int main(int argc,char** argv)
{
	int soc,mlen;
	unsigned short buf,in_buf,ping;
	BSDATA info;
	PVSNODE* node;

	if(BSmakeServer(argv)<0)
	{
		fprintf(stderr,"Cannot make into a server\n");
		return -1;
	}

	if((soc=BSopenListenerUDP(PVS_UDP_PORT))<0)
	{
		fprintf(stderr,"Open of UDP listener socket failed\n");
		return -1;
	}

	buf=htons(BS_Ok); /* always sends this out */
	ping=htons(BS_Ping); /* always sends this out */

	while(1)
	{
		/* wait forever until a message comes in */

		mlen=BSreadUDP(soc,&info,7,&in_buf,sizeof(in_buf));

		/* check for errors */
		switch(mlen)
		{
		case 0: /* timeout */
			printf("Why did a timeout occur?\n");
			/* send out a ping to each of the IOCs in the ioc_list */
			for(node=ioc_list;node;node=node->next)
			{
				mlen=BStransUDP(soc,&(node->info),&ping,sizeof(ping),
					&in_buf,sizeof(in_buf));

				/* check for errors */
				switch(mlen)
				{
				case 0: /* timeout */
					printf("IOC dead\n");
					node->alive=0;
					break;
				case -1: /* error */
					printf("Communications failed\n");
					break;
				default: /* ok */
					if(node->alive==0)
					{
						switch(fork())
						{
						case -1: /* error */
							perror("fork failure");
							break;
						case 0: /* child */
							close(soc);
							BSserverClearSignals();
							sleep(1);
							if(read_pvs(&(node->info))==0)
								node->alive=1;
						default: /* parent */
							break;
						}
					}
					break;
				}
			}
			break;
		case -1: /* error */
			fprintf(stderr,"Communications failure\n");
			break;
		default: /* ok */
			if(BSwriteUDP(soc,&info,&buf,sizeof(buf))<0)
				fprintf(stderr,"respone send failed\n");
			else
			{
				node=(PVSNODE*)malloc(sizeof(PVSNODE));
				node->alive=1;
				node->info=info;
				BSsetPort(&(node->info),PVS_UDP_CPORT);
				node->next=ioc_list;
				ioc_list=node;

				switch(fork())
				{
				case -1: /* error */
					perror("fork failure");
					break;
				case 0: /* child */
					close(soc);
					BSserverClearSignals();
					sleep(1);
					return read_pvs(&info,PVS_RecList,NULL);
				default: /* parent */
					break;
				}
			}
			break;
		}
	}
	close(soc);
	return 0;
}
#endif

static int read_pvs(BSDATA* info,int serv,char* sname)
{
	BS* bs;
	int verb,size,done,len,i,port,rsize;
	char* buffer;
	char ip_from[40];
	FILE* fd;

	BSgetAddressPort(info,ip_from,&port);

	/* printf("IOC %s starting\n",ip_from); */

	/* verify ioc not already added */
	if(access(ip_from,F_OK)==0)
	{
		/* delete the existing file for this IOC */
		unlink(ip_from);
	}

	done=0;
	BSsetPort(info,PVS_TCP_PORT);

	if((bs=BSipOpenData(info))==NULL)
	{
		fprintf(stderr,"Open of socket to IOC failed\n");
		return -1;
	}

	if(serv>PVS_LAST_VERB)
		rsize=strlen(sname)+1;
	else
		rsize=0;

	if(BSsendHeader(bs,serv,rsize)<0)
	{
		fprintf(stderr,"Command send failed\n");
		return -1;
	}

	if(rsize>0)
	{
		if(BSsendData(bs,sname,rsize)<0)
		{
			fprintf(stderr,"send of command name failed\n");
			return -1;
		}
	}

#ifdef PVS_SERVER_PROG
	if((fd=fopen(ip_from,"w"))==(FILE*)NULL)
	{
		fprintf(stderr,"Open of name file failed\n");
		return -1;
	}
#else
	fd=stdout;
#endif

	buffer=(char*)malloc(PVS_TRANSFER_SIZE+2);

	while(done==0)
	{
		if(BSreceiveHeader(bs,&verb,&size)<0)
		{
			fprintf(stderr,"Receive header failed\n");
			done=-1;
		}
		else
		{
			switch(verb)
			{
			case PVS_Data: /* read a block of names */
				if((len=BSreceiveData(bs,buffer,size))<0)
				{
					fprintf(stderr,"Receive data failed\n");
				}
				else
				{
					for(i=0;i<len;i++)
					{
						if(buffer[i]==' ') buffer[i]='\n';
					}
					buffer[len]='\n';
					buffer[len+1]='\0';

					fputs(buffer,fd);
				}
				break;
			case BS_Done: /* transfers complete */
				BSclose(bs);
				done=-1;
				break;
			default:
				if(size>0) done=-1;
				break;
			}
		}
	}

#ifdef PVS_SERVER_PROG
	fclose(fd);
#endif
	free(buffer);
	return 0;
}