privacore-open-source-search-engine/Msg1f.cpp

#include "Msg1f.h"
#include "UdpServer.h"
#include "Conf.h"

static void handleRequest ( UdpSlot *slot , int32_t netnice );
#define  LOG_WINDOW 2048



bool Msg1f::init() {
        if ( ! g_udpServer.registerHandler ( msg_type_1f, handleRequest )) {
		log(LOG_WARN, "db: logview initialization failed"); 
		return false;
	}
	return true;
}


bool Msg1f::getLog(int32_t hostId, 
		   int32_t numBytes, 
		   void *callbackState, 
		   void ( *callback) (void *state, UdpSlot* slot)) {
	
	char* sendBuf = (char*)mmalloc(sizeof(int32_t), "Msg1fA");
	char* p = sendBuf;
	*(int32_t*)p = numBytes;
	p += sizeof(int32_t);
	g_udpServer.sendRequest(sendBuf, p - sendBuf, msg_type_1f, g_hostdb.m_hostPtrs[hostId]->m_ip, g_hostdb.m_hostPtrs[hostId]->m_port, g_hostdb.m_hostPtrs[hostId]->m_hostId, NULL, callbackState, callback, 5000);

	return false;
}


void handleRequest ( UdpSlot *slot , int32_t netnice ) {
	char *p = slot->m_readBuf;

	int32_t numBytes = *(int32_t*)p;
	p += sizeof(int32_t);


	char *filename = g_hostdb.m_logFilename;

	// running just ./gb will log to stderr...
	if ( strcmp(filename ,"/dev/stderr") == 0 ) {
		g_errno = EBADFILE;
		log(LOG_ERROR,"%s:%s:%d: call sendErrorReply.", __FILE__, __func__, __LINE__);
		g_udpServer.sendErrorReply ( slot, g_errno ); 
		return;
	}

	int32_t fd = open ( filename , O_RDONLY ,
			    getFileCreationFlags() );
			 // S_IRUSR |S_IWUSR |S_IRGRP |S_IWGRP| S_IROTH );
	if ( ! fd ) {
		log(LOG_DEBUG, "logviewer: Failed to open %s for reading: ",
		    filename);
		g_errno = EBADFILE;
		log(LOG_ERROR,"%s:%s:%d: call sendErrorReply.", __FILE__, __func__, __LINE__);
		g_udpServer.sendErrorReply ( slot, g_errno ); 
		return;
	}

	char stackSpace[LOG_WINDOW];
	char *buf = stackSpace;
	char *allocBuf = NULL;
	int32_t allocBufSize = 0;

	if(numBytes > LOG_WINDOW) {
		buf = (char*)mmalloc(numBytes, "Msg1fA");
		if(!buf) {
			log(LOG_INFO, 
			    "admin: malloc of %" PRId32" bytes failed "
			    "for logview,"
			    " falling back on stack buffer.",
			    numBytes);
			buf = stackSpace;
			numBytes = LOG_WINDOW;
		}
		else {
			allocBuf     = buf;
			allocBufSize = numBytes;
		}
	}

	lseek(fd, -1 * numBytes, SEEK_END);
	if(errno == EINVAL) {
		//oops! we seeked to before the begining of the file
		//log(LOG_WARN, "bad seek!");
		lseek(fd, 0, SEEK_SET);
	}

	int32_t numRead = read(fd, buf, numBytes-1);
	close(fd);
	if(numRead > 0)	buf[numRead-1] = '\0';
	else          {  
		buf[0] = '\0'; 
		numRead = 0;
		if(allocBuf) mfree(allocBuf,	allocBufSize, "Msg1fA");
		allocBufSize = 0;
		allocBuf = NULL;
		log(LOG_ERROR,"%s:%s:%d: call sendErrorReply.", __FILE__, __func__, __LINE__);
		g_udpServer.sendErrorReply ( slot, EBADFILE ); 
		return;
	}
	//log(LOG_DEBUG, "bytes read! %" PRId32" ", numRead);

	g_udpServer.sendReply(buf, numRead, allocBuf,allocBufSize, slot); //send
}