main.h

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/wait.h>
#include <signal.h>
#include <list>
#include <unistd.h>
#include <ctype.h>
#include <map>
#include <openssl/sha.h>
#include <stdlib.h>
#include <string.h>
#include <set>
#include <queue>

#define HEADER_SIZE 27
#define SHA_DIGEST_LENGTH 20

using namespace std ;

//keeps the join responses
extern set<struct joinResNode> joinResponse ;
//keeps the status messgae
extern set< set<struct node> > statusResponse ;
extern map<struct node, list<string> > statusResponseTypeFiles ;

struct node{
	unsigned short int portNo;
	unsigned char hostname[256];

	bool operator<(const struct node& node1) const{
		return node1.portNo < portNo ;
	}
	bool operator==(const struct node& node1) const{
		return ((node1.portNo == portNo) && !strcmp((char *)node1.hostname, (char *)hostname)) ;
	}
};

struct joinResNode{
	unsigned short int portNo;
	unsigned char hostname[256];
	uint32_t location ;

	bool operator<(const struct joinResNode& node1) const{
		return node1.location > location ;
	}
	bool operator==(const struct joinResNode& node1) const{
		return ((node1.portNo == portNo) && !strcmp((char *)node1.hostname, (char *)hostname) && (node1.location == location)) ;
	}
};

extern map<struct node, int> nodeConnectionMap ;				// To store all the neighboring nodes
extern pthread_mutex_t nodeConnectionMapLock ;

struct metaData
{
	unsigned int fileNumber;
	unsigned char fileID[20];
	unsigned char fileName[256];
	unsigned long int fileSize;
	unsigned char sha1[20];
	unsigned char nonce[20];
	list<string > *keywords;
	//map<string, int> keywords;
	unsigned char bitVector[128];
};
//message structure that goes into the messgae queues
struct Message{
	uint8_t type;
	unsigned char *buffer ;
	unsigned char *query ;
	unsigned char *metadata;
	unsigned char *fileName ;
	uint8_t ttl ;
	uint32_t location ;
	int status ;								// 0 - originated from here
	unsigned char uoid[SHA_DIGEST_LENGTH] ;
	bool fromConnect ;							// 1 - The message was created by the node which 
										// initiated the connection
	int buffer_len ;
	uint8_t status_type ;	
	unsigned char query_type ;
	uint8_t errorCode;
} ;

//information for the neighbors stored in this structures
struct connectionNode{
	list<struct Message> MessageQ ;
	pthread_mutex_t mesQLock ;
	pthread_cond_t mesQCv ;
	int shutDown  ;
	unsigned int keepAliveTimer;
	int keepAliveTimeOut;	
	int myReadId;
	int myWriteId;
	int isReady;
	bool joinFlag;
	struct node n;
};


struct parsedDeleteMessage
{
	unsigned char fileName[256];
	unsigned char sha1[20];
	unsigned char nonce[20];
	unsigned char password[20];	
};

extern unsigned char tempLogFile[512], tempInitFile[512];
extern bool shutDown ;
extern int accept_pid;
extern int nSocket_accept;
extern int keepAlive_pid;
extern int toBeClosed;
extern int joinTimeOutFlag;
extern int inJoinNetwork;
extern int statusTimerFlag ;
extern int checkTimerFlag ;
extern int searchTimerFlag ;
extern int node_pid;
extern int softRestartFlag ;
//extern int globalFileNumber ;
extern int globalSearchCount ;
extern int currentCacheSize;
extern unsigned char extFile[256];
extern unsigned char filesDir[256];
extern FILE *f_log;
extern map<int, struct connectionNode> connectionMap ;				// Stores all the info related to a connection
extern list<pthread_t > childThreadList ;
extern map<pthread_t , bool > myConnectThread ;
extern map<string, list<int> > bitVectorIndexMap;
extern map<string, list<int> > fileNameIndexMap;
extern map<string, list<int> > sha1IndexMap;
extern map<string, int> fileIDMap;
extern map<int, struct metaData> getFileIDMap;
extern list<int > cacheLRU;
//extern list<struct metaData> metadataList;
extern pthread_mutex_t connectionMapLock ;
extern pthread_mutex_t statusMsgLock ;
extern pthread_mutex_t searchMsgLock ;
extern pthread_mutex_t getMsgLock ;
extern pthread_mutex_t logEntryLock ;
extern pthread_cond_t statusMsgCV;
extern pthread_cond_t searchMsgCV;
extern pthread_cond_t getMsgCV;
extern list<string> tmpFileNameList ;

//pakcet structre stored in the cache at the nodes
struct Packet{

	struct node receivedFrom ;
	int status;								//  0 - originally Sent from here, 
										//  1 - Forwarded from here, route back, else -1
	int sockfd ;
	int msgLifeTime;	
	int status_type ;
};

extern map<string, struct Packet> MessageDB ;					// Keeps a track of all the messages it sends/forwards
extern pthread_mutex_t MessageDBLock ;

// Thread function declarations
void *keyboard_thread(void *) ;
void *timer_thread(void *) ;
void *accept_connectionsT(void *);						// Waits for other nodes to connect
void *read_thread(void *) ;
void *write_thread(void *) ;
void *connectBeacon(void *);

// Function declarations
int isBeaconNode(struct node n);
int connectTo(unsigned char *, unsigned int) ;
extern pthread_t k_thread;
void notifyMessageSend(int resSock, uint8_t errorCode);
void pushMessageinQ(int, struct Message ) ;
void closeConnection(int) ;
void joinNetwork() ;
void getStatus() ;
void writeToStatusFile() ;
void getStatusTypeFiles() ;
void writeToStatusFile_TypeFiles() ;
unsigned char *createLogEntry(unsigned char mode, int origin, unsigned char header[HEADER_SIZE], unsigned char *buffer);
void writeLogEntry(unsigned char *logEntry);
void eraseValueInMap(int val);
extern unsigned char *GetUOID(char *, unsigned char *, long unsigned int) ;
void initiateCheck() ;
void init() ;
void cleanup() ;
void writeMetaData(struct metaData metadata, int globalFileNumber);
int updateGlobalFileNumber();
void writeData(struct metaData, int globalFileNumber);
void populateBitVectorIndexMap(unsigned char*, unsigned int);
void populateSha1IndexMap(unsigned char*, unsigned int);
void populateFileNameIndexMap(unsigned char*, unsigned int);
unsigned char* toHex(unsigned char *str, int len);
struct metaData populateMetaData(int fileNumber);
void initiateSearch(unsigned char, unsigned char *) ;
string MetaDataToStr(struct metaData) ;
struct metaData populateMetaDataFromString(unsigned char *input);
struct metaData populateMetaDataFromCPPString(string);
int searchResponseDisplay(list<struct metaData> metadataList, int count);
void writeLRUToFile();
void readLRUFromFile();
void updateLRU(int fileNumber);
void removeFromLRU();
int storeInLRU(struct metaData metadata, unsigned int fileNumber);
list<int> getAllFiles() ;
struct parsedDeleteMessage parseDeleteMessage(unsigned char *message);
void deleteFile(struct parsedDeleteMessage pd);
int getFileNumberFromIndex(unsigned char *fileName, unsigned char *nonce);
void deleteFromIndex(int);
void deleteAllFiles();
void initiateDelete(unsigned char *) ;
void initiateStore(string, string) ;
void writeFileToCache(unsigned char *metadata_str, unsigned char *fileName);
void writeFileToPermanent(unsigned char *metadata_str, unsigned char *fileName);
void initiateGet(struct metaData) ;
struct metaData populateMetaDataFromString_noFileID(unsigned char *input1);
int doesFileExist(struct metaData );
void writeToFileFromData(unsigned char fileName[], int fileNumber);