commmain.c

这是个unix环境下的通讯程序

//该程序在1号机运行，当与2号机建立连接后，一个进程继续监听，另创建两个进程。。
//一个负责向2号机发送数据，一个负责从2号机接收数据。
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <signal.h>

#include "UnionSocket.h"
#include "UnionReg.h"
#include "CommMsg.h"

// Golobal variables definition
int	g_iSckHDL;
int	g_iBrotherPID;
int	g_iPortOfHost;
TUnionIDOfMsgBuf	g_iHostReqBuf;
TUnionIDOfMsgBuf	g_iHostResBuf;

// Functions definition
void DealSigUsr1();
void DealSigUsr2();

int InitializeResource()
{
	int	iRet;

	if ((iRet = UnionInitHostMsgCommCFG()) != 0)
	{
		UnionUserErrLog("in InitializeResource::UnionInitHostMsgCommCFG Failure! iRet = [%d]\n",iRet);
		return(iRet);
	}
	else
		UnionLog("in InitializeResource::UnionInitHostMsgCommCFG OK!\n");

	g_iHostReqBuf=UnionInitializeMessageBuffer(UnionGetUserIDOfReqBufForHost());
	if (g_iHostReqBuf < 0)
	{
		UnionUserErrLog("in InitializeResource::UnionInitializeMessageBuffer Failure! return = [%d]\n",g_iHostReqBuf);
		return(iRet);
	}
	
	g_iHostResBuf=UnionInitializeMessageBuffer(UnionGetUserIDOfResBufForHost());
	if (g_iHostResBuf < 0)
	{
		UnionUserErrLog("in InitializeResource::UnionInitializeMessageBuffer Failure! return = [%d]\n",g_iHostResBuf);
		return(iRet);
	}

	return(0);
}

int ReleaseResource()
{
	return(0);
}

int main(int argc,char *argv[])
{
	int	iRet;
			
	if (argc != 2)
	{
		printf("Usage:: CommMain PortOfListen\n");
		return(-1);
	}

	if (atoi(argv[1]) <= 0)
	{
		printf("Port of host listen QZJ [%s] is error!\n",argv[2]);
		return(-3);
	}

	g_iPortOfHost = atoi(argv[1]);
	
	if (UnionCreateProcess() > 0)
		return(0);
	
	UnionIgnoreSignals();
	signal(SIGUSR1,DealSigUsr1);

	if ((iRet=InitializeResource()) != 0)
	{
		UnionUserErrLog("in main:: InitializeResource! return=[%d]\n",iRet);
		return(-1);
	}

	if ((iRet = UnionTCPIPServer(g_iPortOfHost,regName,UnionTCPIPTaskServer)) < 0)
		UnionUserErrLog("in main:: UnionTCPIPServer Failure! return=[%d]\n",iRet);
	else
		iRet = 0;
	
	ReleaseResource();

	return(iRet);
}

int UnionTCPIPTaskServer(int iSocketID,char *pIPAddr,int iPort)
{
	int	iRet,iPID,iBreak=0;

	UnionSuccessLog("in UnionTCPIPTaskServer:: Client [%s] [%d] is connected now!\n",pIPAddr,iPort);

	g_iSckHDL = iSocketID;

	for (;;)
	{
		g_iBrotherPID = getpid();		// Set Father Process as Brother Process of Child Process.
		if ((iPID = UnionCreateProcess()) > 0)	// Father Process responsible for sending message to snet.
		{
			g_iBrotherPID = iPID;		// Set Child Process as Brother Process of Father Process.
			if ((iRet = UnionSendMessageToQZJ()) < 0)
			{
				UnionUserErrLog("in UnionTCPIPTaskServer:: UnionSendMessageToQZJ! return=[%d]\n",iRet);
			}
			//UnionKillTaskByProcID(g_iBrotherPID);	// 通知兄弟进程退出
			kill(g_iBrotherPID,SIGUSR1);
			UnionSuccessLog("in UnionTCPIPTaskServer:: Parent process: Send SIGUSR1 to brother process [%d]\n",g_iBrotherPID);
			iBreak = 1;
		}
		else if (iPID == 0)
		{
			if ((iRet = UnionReceiveMessageFromQZJ()) < 0)
			{
				UnionUserErrLog("in UnionTCPIPTaskServer:: UnionReceiveMessageFromQZJ! return=[%d]\n",iRet);
			}
			//UnionKillTaskByProcID(g_iBrotherPID);	// 通知兄弟进程退出。
			kill(g_iBrotherPID,SIGUSR1);
			UnionSuccessLog("in UnionTCPIPTaskServer:: Child process: Send SIGUSR1 to brother process [%d]\n",g_iBrotherPID);
			iBreak = 1;
		}
		else
		{
			UnionUserErrLog("in UnionTCPIPTaskServer:: UnionCreateProcess fail! return=[%d]\n",iPID);
			continue;
			
		}
		if (iBreak == 1) break;
	}

	return(iRet);
}	

int UnionSendMessageToQZJ()
{
	int	iRet,iLenOfMsg;
	unsigned char	caSendBuf[1024];
	
	for (;;)
	{
		UnionLog("in UnionSendMessageToQZJ:: Waiting receive [%ld] type of message from Host...\n",UnionGetMsgTypeForHostComm());
		memset(caSendBuf,0,sizeof(caSendBuf));
		// 从主机的响应消息队列中读出需要发送到2号机的交易包
		iLenOfMsg=UnionGetSpecifiedMessageFromBuffer(g_iHostResBuf,UnionGetMsgTypeForHostComm(),&caSendBuf[2]);
		if (iLenOfMsg < 0)
		{
			UnionUserErrLog("in UnionSendMessageToQZJ:: UnionGetSpecifiedMessageFromBuffer! return=[%d]\n",iLenOfMsg);
			continue;
		}
		UnionLog("in UnionSendMessageToQZJ:: Receive [%d] bytes from host message buffer=[%s]\n",iLenOfMsg,&caSendBuf[2]);

		caSendBuf[0] = iLenOfMsg/256;
		caSendBuf[1] = iLenOfMsg%256;
		if ((iRet = UnionSendToSocket(g_iSckHDL,(unsigned char *)caSendBuf,iLenOfMsg+2)) < 0)
		{
			UnionUserErrLog("in UnionSendMessageToQZJ:: UnionSendToSocket! return=[%d]\n",iRet);
			break;
		}
		UnionLog("in UnionSendMessageToQZJ:: Send the host message to QZJ successfully!\n");
	}
	UnionCloseSocket(g_iSckHDL);

	return(-1);
}

int UnionReceiveMessageFromQZJ()
{
	int	iRet,iMsgLen;
	char	caHostTime[10],caBuf[10];
	unsigned char	caRecvBuf[1024];

	for (;;)
	{
		UnionLog("in UnionReceiveMessageFromQZJ:: Waiting receive message from QZJ...\n");
		memset(caRecvBuf,0,sizeof(caRecvBuf));
		if ((iRet = UnionReceiveFromSocketUntilLen(g_iSckHDL,caRecvBuf,2)) != 2)
		{
			UnionUserErrLog("in UnionReceiveMessageFromQZJ:: UnionReceiveFromSocketUntilLen fail! Received bytes=[%d], Expected bytes=[2]\n",iRet);
			break;
		}
		iMsgLen = caRecvBuf[0]*256 + caRecvBuf[1];
		UnionLog("in UnionReceiveMessageFromQZJ:: Receive buffer length = [%02x %02x] [%d]\n",caRecvBuf[0],caRecvBuf[1],iMsgLen);

		if ((iRet = UnionReceiveFromSocketUntilLen(g_iSckHDL,caRecvBuf,iMsgLen)) != iMsgLen)
		{
			UnionUserErrLog("in UnionReceiveMessageFromQZJ:: UnionReceiveFromSocketUntilLen fail! Received bytes=[%d], Expected bytes=[%d]\n",iRet,iMsgLen);
			break;
		}
		UnionLog("in UnionReceiveMessageFromQZJ:: Receive [%d] bytes from QZJ=[%s]\n",iMsgLen,caRecvBuf);

		// 置换交易请求包中的交易时间为主机的系统时间
		memset(caHostTime,0,sizeof(caHostTime));
		memset(caBuf,0,sizeof(caBuf));
		UnionGetSystemTime(caBuf);
		memcpy(caHostTime,caBuf,2);
		caHostTime[2]=':';
		memcpy(&caHostTime[3],&caBuf[2],2);
		caHostTime[5]=':';
		memcpy(&caHostTime[6],&caBuf[4],2);
		memcpy(&caRecvBuf[73],caHostTime,8);
		UnionLog("in UnionReceiveMessageFromQZJ:: Host time = [%s][%s]\n",caBuf,caHostTime);
		
		// 将从2号机接收到的交易包发送到主机的请求消息队列
		iRet=UnionAddMessageToBuffer(g_iHostReqBuf,UnionGetMsgTypeForHostComm(),(unsigned char *)caRecvBuf,iMsgLen);
		if (iRet != 0)
		{
			UnionUserErrLog("in UnionReceiveMessageFromQZJ:: UnionAddMessageToBuffer! return=[%d]\n",iRet);
			continue;
		}
		UnionLog("in UnionReceiveMessageFromQZJ:: Send the QZJ message to host message buffer successfully!\n");
	}
	UnionCloseSocket(g_iSckHDL);

	return(-1);
}

void DealSigUsr1()
{
	UnionLog("in DealSigUsr1:: Task Received SigUsr1!\n");
	UnionCloseSocket(g_iSckHDL);
	ReleaseResource();
	exit(-1);
}

// Functions called by other programs
int UnionUnregisterTask()
{
	return(0);
}

int UnionRegisterTask(char *fmt,...)
{
	return(0);
}

int UnionCreateProcess()
{
	int i;
	int pid;

	for (i=0;i<3;i++)
	{
		if ((pid = fork()) != -1)
			break;
		else
		{
			UnionSystemErrLog("in UnionCreateProcess:: fork!\n");
			usleep(20);
		}
	}
	return(pid);
}

int UnionIgnoreSignals()
{
	setsid();

	signal(SIGALRM,SIG_IGN);
	signal(SIGINT,SIG_IGN);
	signal(SIGQUIT,SIG_IGN);
	signal(SIGHUP,SIG_IGN);
	signal(SIGCLD,SIG_IGN);
	return(0);
}