ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

2021SC@SDUSC 山大智云 7.病毒扫描

2021-11-12 17:30:14  阅读:143  来源: 互联网

标签:2021SC scan self 山大智云 repo commit SDUSC id def


病毒扫描

病毒扫描包下的文件组成

__init__.py
commit_differ.py:文件差异对比
db_oper.py:数据库操作
models.py:数据库模型
run_virus_scan.py:主启动文件
scan_settings.py:病毒扫描设置
thread_pool.py:线程池声明
virus_scan.py:病毒扫描函数

scan_settings(关于扫描的设置)

class Settings():
   __init__():
   parse_config():
   parse_scan_config():
   parse_send_mail_config():
   is_enabled():
补充一:configParser:读取配置对象

parse_scan_config:分析扫描的配置。

该方法对配置中的virus_scan中的多个属性进行检查。
包括scan_command,virus_code,nonvirus_code,scan_interval,scan_size_limit,scan_skip_ext,threads.
大体结构如下:
if cfg.has_option('virus_scan', 'scan_command'):
    self.scan_cmd = cfg.get('virus_scan', 'scan_command')
if not self.scan_cmd:
    logger.info('[virus_scan] scan_command option is not found in %s, disable virus scan.' %
                        seaf_conf)
    return False

parse_send_mail_config:对邮件的病毒扫描进行配置的信息。

引入了utils.config中的两个方法:get_opt_from_conf_or_env,parse_bool

get_opt_from_conf_or_env:在配置中查找属性,没有的话,如果有设置环境变量中的变量的话,调用os.environ(os模块中获取环境变量的方法)

def get_opt_from_conf_or_env(config, section, key, env_key=None, default=None):
    '''Get option value from events.conf. If not specified in events.conf,
    check the environment variable.

    '''
    try:
        return config.get(section, key)
    except configparser.NoOptionError:
        if env_key is None:
            return default
        else:
            return os.environ.get(env_key.upper(), default)

parse_bool:判断输入是否为true

符合要求类型:1和lower(v)="true"

返回parse_send_mail_config方法

通过get_opt_from_conf_or_env读取配置'SEAHUB EMAIL'中的 'enabled'和'seahubdir'属性。
然后根据配置返回mail时的virus scan配置

parse方法:读取config并调用上述方法进行设置。

commit_differ.py(差异对比)

和seafobj中的commit_differ相似,为差异对比。

首先是辅助函数make_path(dirname,filename)

def make_path(dirname, filename):
    if dirname == '/':
        return dirname + filename
    else:
        return '/'.join((dirname, filename))

该函数用于连接文件夹名和文件名

class CommitDiffer

    def __init__(self, repo_id, version, root1, root2):
        self.repo_id = repo_id
        self.version = version
        self.root1 = root1
        self.root2 = root2

def diff为差异对比函数

model.py(数据库模型)

存放数据库模型

VirusScanRecord:repo_id(主键), scan_commit_id
VirusFile:vid(主键,自增),repo_id, commit_id, file_path, has_deleted, has_ignored

db_oper.py(数据库的操作)

文件结构

class DBOper:
   def __init__(self,settings)
   def get_repo_list(self)
   def get_scan_commit_id(self,repo_id)
   def update_vscan_record(self,repo_id,scan_commit_id)
   add_virus_record(self,records)
def get_virus_files(session, repo_id, has_handled, start, limit)
def delete_virus_file(session, vid)
def operate_virus_file(session, vid, ignore)
def get_virus_file_by_vid(session, vid)

def get_virus_file_by_vid(session,vid):从VirusFile表中通过vid来获取对象

q = session.query(VirusFile).filter(VirusFile.vid == vid)
return q.first()

def operate_virus_file(sessoin,vid,ignore):修改指定vid的has_ignored属性

q = session.query(VirusFile).filter(VirusFile.vid == vid)
r = q.first()
r.has_ignored = ignore
session.commit()
return 0

def delete_virus_file(session, vid):修改指定vid的has_deleted属性为1

q = session.query(VirusFile).filter(VirusFile.vid == vid)
r = q.first()
r.has_deleted = 1
session.commit()
return 0

def get_virus_files(session, repo_id, has_handled, start, limit):

获取VirusFile对象(start:起始位置,has_handled:属性要求,limit:条数限制)

q = session.query(VirusFile)
if repo_id:
    q = q.filter(VirusFile.repo_id == repo_id)
if has_handled is not None:
    if has_handled:
        q = q.filter(or_(VirusFile.has_deleted == 1, VirusFile.has_ignored == 1))
    else:
        q = q.filter(and_(VirusFile.has_deleted == 0, VirusFile.has_ignored == 0))
q = q.slice(start, start+limit)
return q.all()

class DBOper

def get_scan_commit_id(self,repo_id):根据repo_id查询VirusScanRecord表中的scan_commit_id

q = session.query(VirusScanRecord).filter(VirusScanRecord.repo_id == repo_id)
r = q.first()
scan_commit_id = r.scan_commit_id if r else None
return scan_commit_id

def update_vscan_record(self, repo_id, scan_commit_id)

标签:2021SC,scan,self,山大智云,repo,commit,SDUSC,id,def
来源: https://blog.csdn.net/curiouity/article/details/121292510

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有