communicate.cpp

完成数字证书加密

#include "Communicate.h"

Communicate::Communicate(string type)
	:m_type(type)
{
	fromCA = fromADT = m_flag = 0;
	toCA = toADT = 0;

	if (m_type=="ca")
		initializeCA();
	else if (m_type=="adt") {
		initializeADT();
	}
}

Communicate::~Communicate()
{
	if (m_type=="ca") {
		freeCA();
	}
	else if (m_type=="adt") {
		freeADT();
	}
}

void Communicate::freeCA()
{
	msgctl(fromADT, IPC_RMID, 0);
	msgctl(toADT, IPC_RMID, 0);
}

void Communicate::freeADT()
{
	msgctl(fromCA, IPC_RMID, 0);
	msgctl(toCA, IPC_RMID, 0);
}

CONTENT Communicate::caGetMessageFromADT()
{
	return getMessage("ca");
}

int Communicate::caSendMessageToADT(CONTENT msg)
{
	return sendMessage(msg, "ca");
}

CONTENT Communicate::ADTGetMessageFromCA()
{
	return getMessage("adt");
}

int Communicate::ADTSendMessageToCA(CONTENT msg)
{
	return sendMessage(msg, "adt");
}

CONTENT Communicate::getMessage(string objectName)
{
	int object = 0;
	if (objectName=="ca") {
		object = fromADT;
	}
	else if (objectName=="adt") {
			object = fromCA;
	}
	MSG msg;
	while (1) {
		if (msgrcv(object, (struct msgbuf*)&msg, SIZE, PKI_SERVICE, IPC_NOWAIT) > 0) {
			break;
		}
		sleep(1);
	}
	CONTENT content;
	char* pt = msg.content;
	content.type = atol(pt);

	int pos = 0;
	while ((*(pt+pos)>=0x30) && (*(pt+pos)<=0x39)) {
		pos ++;
	}
	strcpy(content.buf, pt+pos);

	return content;
}

int Communicate::sendMessage(CONTENT msg, string objectName)
{
	int object = 0;
	if (objectName=="ca") {
		object = toADT;
	}
	else if (objectName=="adt") {
			object = toCA;
	}
	
	MSG message;
	message.type = PKI_SERVICE;
	char type[7];
	memset(type, 0, 7);
	sprintf(type, "%d", msg.type);
	strcpy(message.content, type);
	strcat(message.content, msg.buf);

	errno = 0;

	if (msgsnd(object, (struct msgbuf*)&message, SIZE, 0)== -1) {
		cout << "error : "<<strerror(errno)<<endl;
	}
	return 1;

}

void Communicate::initializeADT()
{
	if (readCAconfig("Communication", "rCACom")=="no") {
		m_flag=toCA=msgget(ftok(".",'a'), IPC_CREAT|0666);
		setCAConfigItem("Communication", "rCACom", "yes");
		if (m_flag==-1) cout<<"msgget error : " <<strerror(errno)<<endl, m_flag=0;
	}
	else {
		m_flag=toCA=msgget(ftok(".",'a'), 0666);
		setCAConfigItem("Communication", "rCACom", "no");
		if (m_flag==-1) cout<<"msgget error : " <<strerror(errno)<<endl, m_flag=0;
	}

	if (readCAconfig("Communication", "wCACom")=="no") {
		m_flag=fromCA = msgget(ftok("/ngg", 'a'), IPC_CREAT|0666);
		setCAConfigItem("Communication", "wCACom", "yes");
		if (m_flag==-1) cout<<"msgget error : " <<strerror(errno)<<endl, m_flag=0;
	}
	else {
		m_flag=fromCA = msgget(ftok("/ngg", 'a'), 0666);
		setCAConfigItem("Communication", "wCACom", "no");
		if (m_flag==-1) cout<<"msgget error : " <<strerror(errno)<<endl, m_flag=0;
	}
}

void Communicate::initializeCA()
{
	if (readCAconfig("Communication", "rCACom")=="no") {
		m_flag=fromADT=msgget(ftok(".",'a'), IPC_CREAT|0666);
		setCAConfigItem("Communication", "rCACom", "yes");
		if (m_flag==-1) cout<<"msgget error : " <<strerror(errno)<<endl, m_flag=0;
	}
	else {
		m_flag=fromADT=msgget(ftok(".",'a'), 0666);
		setCAConfigItem("Communication", "rCACom", "no");
		if (m_flag==-1) cout<<"msgget error : " <<strerror(errno)<<endl, m_flag=0;
	}

	if (readCAconfig("Communication", "wCACom")=="no") {
		m_flag=toADT = msgget(ftok("/ngg", 'a'), IPC_CREAT|0666);
		setCAConfigItem("Communication", "wCACom", "yes");
		if (m_flag==-1) cout<<"msgget error : " <<strerror(errno)<<endl, m_flag=0;
	}
	else {
		m_flag=toADT = msgget(ftok("/ngg", 'a'), 0666);
		setCAConfigItem("Communication", "wCACom", "no");
		if (m_flag==-1) cout<<"msgget error : " <<strerror(errno)<<endl, m_flag=0;
	}
}

string Communicate::readCAconfig(string keyword, string objectName)
{
	string result = "";	
	FILE *fd = fopen(CA_CONFIG_PATH, "r");
	assert(fd != NULL);
	
	if (fd == NULL) {
		result.append("");
	}
	else {
		char *line  = (char*)calloc(1, 1*K);
		string config="";
		while (!feof(fd)) {
			memset(line, 0, 1*K);
			fgets(line, 1*K, fd); 
			config.append(line);
		}
		fclose(fd);
		
		NggObject *object = NggObject::fromXML(config);
		assert( object!=NULL );
		for (unsigned int i=0; i<object->m_Contains.size(); i++) {
			if (object->m_Contains[i]->m_Name==keyword) {
				if (keyword == "Root" || keyword == "Operator" || keyword == "Administrator"|| keyword=="SerialNumber") {
					result.append(object->m_Contains[i]->m_Value);	
				}
				else {
					NggObject* nggobject = object->m_Contains[i];
					assert( nggobject!=NULL );
					for (unsigned int ii=0; ii<nggobject->m_Contains.size(); ii++) {
						if (nggobject->m_Contains[ii]->m_Name==objectName) {
							result.append(nggobject->m_Contains[ii]->m_Value);	
							break;
						}
					}
					if (nggobject) nggobject=NULL;					
				}
				break;
			}
		}

		if (object) delete object, object=NULL ;
		if (line) free(line), line=NULL;
	}	

	return result;
}

void Communicate::setCAConfigItem(string keyword, string objectName, string value)
{
	FILE *fd = fopen(CA_CONFIG_PATH, "rb");
	assert(fd != NULL);
	char *line  = (char*)calloc(1, 1*K);
	assert(line != NULL);
	string config = "";
	while (!feof(fd)) {
		memset(line, 0, 1*K);
		fgets(line, 1*K, fd); 
		config.append(line);
	}
	fclose(fd);
	
	NggObject *object = NggObject::fromXML(config);
	assert(object != NULL);
	
	for (unsigned int i=0; i<object->m_Contains.size(); i++) {
		if (object->m_Contains[i]->m_Name==keyword) {
			if (keyword == "Root" || keyword == "Operator" || keyword == "Administrator"|| keyword=="SerialNumber") {
				object->m_Contains[i]->m_Value = value;
			}
			else {
				NggObject *nggobject = object->m_Contains[i];
				assert(object != NULL);
				for (unsigned int ii=0; ii<nggobject->m_Contains.size(); ii++) {
					if (nggobject->m_Contains[ii]->m_Name==objectName) {
						nggobject->m_Contains[ii]->m_Value = value;
						break;
					}
				}				
				if (nggobject) nggobject=NULL;					
			}				
			break;
		}
	}
	fd = fopen(CA_CONFIG_PATH, "wb");
	assert( fd!=NULL );
	
	fputs(object->toXML().c_str(), fd);

	if (object) delete object, object=NULL;
	if (line) free(line), line=NULL;
	fclose(fd);
}