def scan_virus(self, scan_task):
try:
sroot_id = None
hroot_id = None
if scan_task.scan_commit_id:
sroot_id = commit_mgr.get_commit_root_id(
scan_task.repo_id, 1, scan_task.scan_commit_id)
if scan_task.head_commit_id:
hroot_id = commit_mgr.get_commit_root_id(
scan_task.repo_id, 1, scan_task.head_commit_id)
differ = CommitDiffer(scan_task.repo_id, 1, sroot_id, hroot_id)
scan_files = differ.diff()
if len(scan_files) == 0:
logger.debug('No change occur for repo %.8s, skip virus scan.',
scan_task.repo_id)
self.db_oper.update_vscan_record(scan_task.repo_id,
scan_task.head_commit_id)
return
else:
logger.info('Start to scan virus for repo %.8s.',
scan_task.repo_id)
vnum = 0
nvnum = 0
nfailed = 0
vrecords = []
for scan_file in scan_files:
fpath, fid, fsize = scan_file
if not self.should_scan_file(fpath, fsize):
continue
ret = self.scan_file_virus(scan_task.repo_id, fid, fpath)
if ret == 0:
logger.debug('File %s virus scan by %s: OK.', fpath,
self.settings.scan_cmd)
nvnum += 1
elif ret == 1:
logger.info('File %s virus scan by %s: Found virus.',
fpath, self.settings.scan_cmd)
vnum += 1
fpath = fpath if isinstance(
fpath, unicode) else fpath.decode('utf-8')
vrecords.append(
(scan_task.repo_id, scan_task.head_commit_id, fpath))
else:
logger.debug('File %s virus scan by %s: Failed.', fpath,
self.settings.scan_cmd)
nfailed += 1
if nfailed == 0:
ret = 0
if len(vrecords) > 0:
ret = self.db_oper.add_virus_record(vrecords)
if ret == 0 and self.settings.enable_send_mail:
self.send_email(vrecords)
if ret == 0:
self.db_oper.update_vscan_record(scan_task.repo_id,
scan_task.head_commit_id)
logger.info(
'Virus scan for repo %.8s finished: %d virus, %d non virus, %d failed.',
scan_task.repo_id, vnum, nvnum, nfailed)
except Exception as e:
logger.warning('Failed to scan virus for repo %.8s: %s.',
scan_task.repo_id, e)
def get_repo_root_seafdir(repo):
root_id = commit_mgr.get_commit_root_id(repo.id, repo.version, repo.head_cmmt_id)
return fs_mgr.load_seafdir(repo.store_id, repo.version, root_id)
def get_repo_root_seafdir(repo):
root_id = commit_mgr.get_commit_root_id(repo.id, repo.version,
repo.head_cmmt_id)
return fs_mgr.load_seafdir(repo.store_id, repo.version, root_id)