o hh,@spddlZddlZddlZddlZddlmZmZmZmZddl m Z ej ej dde eZGdddZdS)N)ListDictAnyOptional)get_db_managerz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformatc@seZdZdZd2dedefddZdd Zd d Zd ee ee ffd dZ ded e e ee ffddZ de ee fd efddZdede ee fd efddZded efddZded efddZded efddZdeded e ee fd efd!d"Zded e e ee ffd#d$Zded%ed efd&d'Zd ee ee ffd(d)Zded efd*d+Zded,ee ee fd efd-d.Zded ee ee ffd/d0Zd1S)3 RuleManageruF 规则管理器，负责规则的增删改查和持久化。 data/honeypot.dbconfig/rules.jsondb_path rules_filecCs>||_g|_t|_|dkrddlm}|||_|dS)Nr r)DatabaseManager)r rulesr db_manager core.databaser _load_rules)selfr r rr/root/ive/core/rule_manager.py__init__s   zRuleManager.__init__c Cstj|jrgz=t|jddd}t|}t|tr!| dgng|_ Wdn1s.wYt dt |j d|jWdStyf}zt d|jd |g|_ WYd}~dSd}~wwt d |jd g|_ dS) u从文件加载全局规则rutf-8encodingrNzLoaded z global rules from zFailed to load rules from : z Rules file z/ not found. Initializing with empty rules list.)ospathexistsr openjsonload isinstancedictgetrloggerinfolen Exceptionerrorwarning)rfdataerrrrs $ zRuleManager._load_rulesc CszDtjtj|jddt|jddd%}tjd|ji|ddd t d t |jd |jWd Wd S1s=wYWd St yc}zt d |jd|WYd }~d Sd }~ww)u将全局规则保存到文件T)exist_okwrrrF) ensure_asciiindentzSaved z global rules to NzFailed to save rules to r)rmakedirsrdirnamer rr dumprr%r&r'r(r))rr+r-rrr _save_rules,s &&zRuleManager._save_rulesreturncCs |jS)u获取所有全局规则)rcopy)rrrr get_all_rules7s zRuleManager.get_all_rulesrule_idcCs&|jD] }|d|kr|SqdS)u根据ID获取全局规则idN)rr$rr:rulerrrget_rule_by_id;s zRuleManager.get_rule_by_idr=cCsX|ds tddS||dr td|dddS|j||dS)u添加新全局规则r;zRule must have an 'id'FRule with ID 'z' already existsT)r$r%r)r>rappendr6)rr=rrradd_ruleBs   zRuleManager.add_rule updated_rulecCsPt|jD]\}}|d|kr||j|<|dSqtd|ddS)u更新全局规则r;Tr?z' not found for updateF enumeraterr$r6r%r*)rr:rBir=rrr update_ruleOs zRuleManager.update_rulecCsNt|jD]\}}|d|kr|j|=|dSqtd|ddS)u删除全局规则r;Tr?z' not found for deletionFrC)rr:rEr=rrr delete_ruleYszRuleManager.delete_rulecCs&||}|rd|d<|dSdS)u启用全局规则TenabledFr>r6r<rrr enable_rulec zRuleManager.enable_rulecCs&||}|rd|d<|dSdS)u禁用全局规则FrHTrIr<rrr disable_rulelrKzRuleManager.disable_ruleenv_iddriverconfigc Csz;|j+}|}|d||tj|dddf|td|d WdWdS1s4wYWdSt j yOt d|d YdSt j ym}zt d |d |WYd}~dSd}~ww) u 在数据库中创建一个新环境。 :param env_id: 环境唯一标识符。 :param driver: 使用的驱动名称。 :param config: 环境配置字典。 :return: 操作是否成功。 zIINSERT INTO environments (id, driver, config, status) VALUES (?, ?, ?, ?)Fr1stopped Environment z created in database.NT already exists in database.zError creating environment r)rget_connectioncursorexecuter dumpscommitr%r&sqlite3IntegrityErrorr*Errorr))rrMrNrOconnrUr-rrrcreate_environmentws& ( zRuleManager.create_environmentc CszB|j2}|}|d|f|}|r/t|}t|d|d<|WdWSWdWdS1s;wYWdStj ya}zt d|d|WYd}~dSd}~ww)u 从数据库获取环境信息。 :param env_id: 环境唯一标识符。 :return: 环境信息字典，如果未找到则返回 None。 z'SELECT * FROM environments WHERE id = ?rONzError getting environment r) rrTrUrVZfetchoner#r loadsrYr[r%r))rrMr\rUrowenv_datar-rrrget_environments(     zRuleManager.get_environmentstatusc Csz4|j$}|}|d||f|td|d WdWdS1s-wYWdStjyHt d|dYdStj yf}zt d|d|WYd}~dSd}~ww) Nz/UPDATE environments set status = ? where id = ?rRz update in database.TrSFzError updateing environment r) rrTrUrVrXr%r&rYrZr*r[r))rrMrbr\rUr-rrrupdate_environments& ( zRuleManager.update_environmentc CszA|j1}|}|d|}g}|D]}t|}t|d|d<||q|WdWS1s:wYWdSt j y^}zt d|gWYd}~Sd}~ww)u` 从数据库列出所有环境。 :return: 环境信息字典列表。 zSELECT * FROM environmentsrONzError listing environments: ) rrTrUrVfetchallr#r r^r@rYr[r%r))rr\rUrows environmentsr_r`r-rrrlist_environmentss"   ( zRuleManager.list_environmentsc CszK|j;}|}|d|f||jdkr.td|d WdWdStd|d WdWdS1sDwYWdSt j yj}zt d |d |WYd}~dSd}~ww) u 从数据库删除环境。由于设置了 CASCADE，environment_rules 表中的相关记录也会被自动删除。 :param env_id: 环境唯一标识符。 :return: 操作是否成功。 z%DELETE FROM environments WHERE id = ?rrRz deleted from database.NTz not found for deletion.FzError deleting environment r) rrTrUrVrXZrowcountr%r&r*rYr[r))rrMr\rUr-rrrdelete_environments"  ( zRuleManager.delete_environmentrc CszQ|jA}|}|d|ft|D]\}}tj|dd}|d|||fq|t dt |d|d WdWd S1sJwYWdSt j yp}zt d |d |WYd}~dSd}~ww) u 为指定环境设置（覆盖）规则列表。 :param env_id: 环境唯一标识符。 :param rules: 要关联到环境的规则对象列表。 :return: 操作是否成功。 z6DELETE FROM environment_rules WHERE environment_id = ?FrPzVINSERT INTO environment_rules (environment_id, rule_data, rule_order) VALUES (?, ?, ?)zSet z rules for environment .NTz$Error setting rules for environment r)rrTrUrVrDr rWrXr%r&r'rYr[r)) rrMrr\rUindexr=Z rule_jsonr-rrrset_environment_ruless& (z!RuleManager.set_environment_rulesc Csz=|j-}|}|d|f|}g}|D]}t|d}||q|WdWS1s6wYWdStj y]}zt d|d|gWYd}~Sd}~ww)u 获取指定环境的规则列表。 :param env_id: 环境唯一标识符。 :return: 与该环境关联的规则对象列表。 zTSELECT rule_data FROM environment_rules WHERE environment_id = ? ORDER BY rule_orderZ rule_dataNz$Error getting rules for environment r) rrTrUrVrdr r^r@rYr[r%r)) rrMr\rUrerr_rule_objr-rrrget_environment_ruless&  (z!RuleManager.get_environment_rulesN)r r )__name__ __module__ __qualname____doc__strrrr6rrrr9rr>boolrArFrGrJrLr]rarcrgrhrkrmrrrrr s&      " ""r )r rloggingrYtypingrrrrrr basicConfigINFO getLoggerrnr%r rrrrs