IFileNode* FileCollectorImpl::AddFileNode(const file_id_t& id, file_size_t pos, bool_t bFinish, const char_t* strDir, const char_t* strName) {
if (NULL == strDir || NULL == strName) { return NULL; }
if (IFileNode::INVALID_FILE_ID == id) { return NULL; }
hash_value_t hash = char_t_hash()(strDir);
/* find */
FileNode* pFileNode = findFileNode(hash, strName);
if (NULL == pFileNode) {
pFileNode = new FileNode(strDir, NULL, strName, id, pos, bFinish);
if (NULL == pFileNode) { return NULL; }
// add to map
addFileNode(hash, pFileNode->LocalName(), pFileNode);
}
if (IFileNode::INVALID_FILE_ID != id) {
// update id
pFileNode->SetID(id);
addFileNode(id, pFileNode);
}
return pFileNode;
}
/*
* indexes a word, calling other functions to check and add to hashtable
* returns 1 on success, otherwise it returns 0
*/
int indexWord( char *key, char *filename ) {
if( key == NULL || filename == NULL ) { // ya dun gooffed
return 0;
}
TokenPtr word, search;
HASH_FIND_STR(words, key, search);
// if the word exists in the hash table, check its fileNodes
if ( search ) {
// shouldn't happen in current implementation
if ( search -> fileHead == NULL ) {
addFileNode( search, filename );
} else if ( !strcmp(filename, (search->fileHead->filename)) ) { // file already exists for word
search->fileHead->tokenCount++;
} else { // file doesn't exist for word
addFileNode( search, filename );
}
} else { // word doesn't exist in the hashtable, create new word and file
if ( (word = (TokenPtr) malloc(sizeof(struct Token))) == NULL ) exit(-1);
char *newKey = (char *) malloc(sizeof(char) * strlen(key));
strcpy(newKey, key);
word -> key = newKey;
word -> fileHead = NULL;
HASH_ADD_STR( words, key, word );
addFileNode( word, filename );
}
return 1;
}
bool_t FileCollectorImpl::searchAllFile(collector_log_type_e collector_log_type
,CollectRule* pCollectRule, file_collect_callback_t pfn_cb, void* context) {
ASSERT(pCollectRule);
ASSERT(pfn_cb);
// now
struct tm today;
time_t now = time(NULL);
LOCALTIME(today, now);
WIN32_FIND_DATA find_file_data;
//__try
{
HANDLE find_handle = FindFirstFile(pCollectRule->find_dir, &find_file_data);
if (find_handle != INVALID_HANDLE_VALUE) {
do {
// Don't count current or parent directories.
if ((STRCMP(find_file_data.cFileName, _STR("..")) == 0) ||
(STRCMP(find_file_data.cFileName, _STR(".")) == 0))
continue;
if (find_file_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { continue; }
/*
long result = CompareFileTime(&find_file_data.ftCreationTime, // NOLINT
&comparison_filetime);
// File was created after or on comparison time
if ((result == 1) || (result == 0))
++file_count;
*/
// is we need file
//uint32_t nMatchLen = STRLEN(find_file_data.cFileName);
if (TRUE == pCollectRule->regExExclude.didCompile()
&& TRUE == pCollectRule->regExExclude.isMatchOnly(find_file_data.cFileName)
) { continue; }
if (FALSE == pCollectRule->regExInclude.isMatchOnly(find_file_data.cFileName)) { continue; }
// test file
// char_t strPath[MAX_PATH + 1] = {0};
// SNPRINTF(strPath, MAX_PATH, MAX_PATH, "%s/%s", pCollectRule->dir, find_file_data.cFileName);
// AutoReleaseFile autoRelFile(OpenFile(strPath, strAttrOpenRead));
// if (NULL == autoRelFile) { continue; }
// find map
FileNode* pFileNode;
if (MEM_NINI_LOG == collector_log_type) {
pFileNode = findFileNode(pCollectRule->mml_hash, find_file_data.cFileName);
}
else {
pFileNode = findFileNode(pCollectRule->hash, find_file_data.cFileName);
}
if (NULL == pFileNode) {
if (TRUE == m_bAfterToDay) {
// time control
SYSTEMTIME nCreateTime;
SYSTEMTIME stUTC;
if (FALSE == FileTimeToSystemTime(&(find_file_data.ftCreationTime), &stUTC)
|| FALSE == SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &nCreateTime)
) { continue; }
if (nCreateTime.wYear != today.tm_year + 1900
|| nCreateTime.wMonth != today.tm_mon + 1
|| nCreateTime.wDay != today.tm_mday
) { continue; }
}
// char_t strPath[MAX_PATH + 1] = {0};
// SNPRINTF(strPath, MAX_PATH, MAX_PATH, "%s/%s", pCollectRule->dir, find_file_data.cFileName);
// AutoReleaseFile AutoRelFile(::OpenFile(strPath, strAttrOpenRead));
// if (NULL == AutoRelFile) { continue; }
//
pFileNode = new FileNode(pCollectRule->dir
, MEM_NINI_LOG == collector_log_type ? pCollectRule->GetMMLDir() : NULL
, find_file_data.cFileName);
if (NULL == pFileNode) { continue; }
// update file info
FindData2FileInfo(&pFileNode->FileInfo(), find_file_data);
// add to map
addFileNode(pCollectRule->hash, pFileNode->LocalName(), pFileNode);
// new file
if (RC_S_OK != (pfn_cb)(/*this, */pFileNode, context)) { break; }
}
else {
ASSERT(pFileNode);
// update file info
PlatformFileInfo& fileInfo = pFileNode->FileInfo();
FindData2FileInfo(&fileInfo, find_file_data);
if (pFileNode->GetSendSize() < fileInfo.size) {
// new data
if (RC_S_OK != (pfn_cb)(/*this, */pFileNode, context)) { break; }
}
}
} while (FindNextFile(find_handle, &find_file_data));
FindClose(find_handle);
}
}
// __except(EXCEPTION_EXECUTE_HANDLER) {
//
// return TRUE;
// }
return TRUE;
}