int SMBCALL smb_addhashes(smb_t* smb, hash_t** hashes, BOOL skip_marked)
{
int retval;
size_t h;
COUNT_LIST_ITEMS(hashes, h);
if(!h) /* nothing to add */
return(SMB_SUCCESS);
if((retval=smb_open_hash(smb))!=SMB_SUCCESS)
return(retval);
fseek(smb->hash_fp,0,SEEK_END);
for(h=0;hashes[h]!=NULL;h++) {
/* skip hashes marked by smb_findhash() */
if(skip_marked && hashes[h]->flags&SMB_HASH_MARKED)
continue;
/* can't think of any reason to strip SMB_HASH_MARKED flag right now */
if(smb_fwrite(smb,hashes[h],sizeof(hash_t),smb->hash_fp)!=sizeof(hash_t)) {
retval=SMB_ERR_WRITE;
break;
}
}
smb_close_hash(smb);
return(retval);
}
size_t strListCount(const str_list_t list)
{
size_t i;
COUNT_LIST_ITEMS(list,i);
return(i);
}
/* If return value is SMB_ERROR_NOT_FOUND, hash file is left open */
int SMBCALL smb_findhash(smb_t* smb, hash_t** compare, hash_t* found_hash,
long source_mask, BOOL mark)
{
int retval;
BOOL found=FALSE;
size_t c,count;
hash_t hash;
if(found_hash!=NULL)
memset(found_hash,0,sizeof(hash_t));
if((retval=smb_open_hash(smb))!=SMB_SUCCESS)
return(retval);
COUNT_LIST_ITEMS(compare, count);
if(count) {
rewind(smb->hash_fp);
while(!feof(smb->hash_fp)) {
if(smb_fread(smb,&hash,sizeof(hash),smb->hash_fp)!=sizeof(hash))
break;
if(hash.flags==0)
continue; /* invalid hash record (!?) */
if((source_mask&(1<<hash.source))==0) /* not checking this source type */
continue;
for(c=0;compare[c]!=NULL;c++) {
if(compare[c]->source!=hash.source)
continue; /* wrong source */
if(compare[c]->length!=hash.length)
continue; /* wrong source length */
if(compare[c]->flags&SMB_HASH_MARKED)
continue; /* already marked */
if((compare[c]->flags&SMB_HASH_PROC_MASK)!=(hash.flags&SMB_HASH_PROC_MASK))
continue; /* wrong pre-process flags */
if((compare[c]->flags&hash.flags&SMB_HASH_MASK)==0)
continue; /* no matching hashes */
if(compare[c]->flags&hash.flags&SMB_HASH_CRC16
&& compare[c]->crc16!=hash.crc16)
continue; /* wrong crc-16 */
if(compare[c]->flags&hash.flags&SMB_HASH_CRC32
&& compare[c]->crc32!=hash.crc32)
continue; /* wrong crc-32 */
if(compare[c]->flags&hash.flags&SMB_HASH_MD5
&& memcmp(compare[c]->md5,hash.md5,sizeof(hash.md5)))
continue; /* wrong crc-16 */
/* successful match! */
break; /* can't match more than one, so stop comparing */
}
if(compare[c]==NULL)
continue; /* no match */
found=TRUE;
if(found_hash!=NULL)
memcpy(found_hash,&hash,sizeof(hash));
if(!mark)
break;
compare[c]->flags|=SMB_HASH_MARKED;
}
if(found) {
smb_close_hash(smb);
return(SMB_SUCCESS);
}
}
/* hash file left open */
return(SMB_ERR_NOT_FOUND);
}