o ¨ ¼hxã@s`ddlZddlZddlZddlZddlmZmZmZddlm Z m Z e  e ¡Z Gdd„dƒZdS)éN)ÚDictÚTypeÚAny)Ú DriverPluginÚCountermeasurePluginc@s€eZdZdZdefdd„Zdd„Zdedefd d „Zded e fd d „Z ded e fdd„Z dede fdd„Z dedefdd„ZdS)Ú PluginManageruU æ’件管ç†å™¨ï¼Œè´Ÿè´£åŠ è½½å’Œç®¡ç†é©±åŠ¨æ’件与å制动作æ’件。 Ú plugin_dirscCs||_i|_i|_dS)N)rÚdriversÚcountermeasures)Úselfr©r ú /root/ive/core/plugin_manager.pyÚ__init__s zPluginManager.__init__c Csú|jD]w}tj |¡st d|›d¡qtj |d¡}tj |¡r)| ||¡qz4t  |¡D],}tj ||¡}tj |¡r[tj |d¡}tj |¡rR| ||¡q/t  d|›d¡q/Wqt yz}zt  d|›d|›¡WYd}~qd}~wwdS) u扫æ并加载所有æ’件zPlugin directory z does not exist.z plugin.jsonzPlugin config z not found.zError scanning directory ú: N) rÚosÚpathÚisdirÚloggerÚwarningÚjoinÚexistsÚ_load_plugin_from_configÚlistdirÚdebugÚOSErrorÚerror)r Ú plugin_dirÚplugin_config_pathÚitemZ item_pathZsub_plugin_config_pathÚer r r Ú load_pluginss.      €ù"€ÿëzPluginManager.load_pluginsrrc Cs zt|ddd }t |¡}Wdƒn1swYWn<tjy=}zt d|›d|›¡WYd}~dSd}~wtyZ}zt d|›d|›¡WYd}~dSd}~ww| d¡}|d krl| ||¡dS|d krx|  ||¡dSt  d |›d |›¡dS) u!从é…置文件加载å•ä¸ªæ’件Úrzutf-8)ÚencodingNzError decoding JSON from rzError reading ÚtypeÚdriverZcountermeasurezUnknown plugin type 'z' in ) ÚopenÚjsonÚloadÚJSONDecodeErrorrrÚ ExceptionÚgetÚ_load_driver_pluginÚ_load_countermeasure_pluginr)r rrÚfÚconfigrZ plugin_typer r r r.s( ÿ€€€þ z&PluginManager._load_plugin_from_configr.c Cób| d¡}| d¡}|r|st d|›¡dSz[| dd¡\}}tj |tj  ||›d¡¡}|durBt d|›d |›¡WdStj  |¡}|j   |¡t ||ƒ} t| tƒrh| |j|<t d |›¡WdSt d |›d ¡WdSttfy“} zt d |›d| ›¡WYd} ~ dSd} ~ wty°} zt d|›d| ›¡WYd} ~ dSd} ~ ww)u加载驱动æ’件ÚnameÚ entry_pointz Invalid driver plugin config in NÚ.éú.pyz(Failed to create spec for driver plugin ú from zLoaded driver plugin: úPlugin z+ does not implement DriverPlugin interface.zFailed to load driver plugin rz'Unexpected error loading driver plugin )r*rrÚrsplitÚ importlibÚutilÚspec_from_file_locationrrrÚmodule_from_specÚloaderÚ exec_moduleÚgetattrÚ issubclassrr ÚinfoÚ ImportErrorÚAttributeErrorr)) r rr.r0r1Ú module_nameÚ class_nameÚspecÚmoduleÚ driver_classrr r r r+Bó2       $€$€ÿz!PluginManager._load_driver_pluginc Cr/)u加载å制动作æ’件r0r1z(Invalid countermeasure plugin config in Nr2r3r4z0Failed to create spec for countermeasure plugin r5zLoaded countermeasure plugin: r6z3 does not implement CountermeasurePlugin interface.z%Failed to load countermeasure plugin rz/Unexpected error loading countermeasure plugin )r*rrr7r8r9r:rrrr;r<r=r>r?rr r@rArBr)) r rr.r0r1rCrDrErFÚcm_classrr r r r,_rHz)PluginManager._load_countermeasure_pluginr0ÚreturncCó|j |¡}|r |ƒSdS)u获å–驱动实例N)r r*)r r0rGr r r Ú get_driver|ó zPluginManager.get_drivercCrK)u获å–å制动作实例N)r r*)r r0rIr r r Úget_countermeasureƒrMz PluginManager.get_countermeasureN)Ú__name__Ú __module__Ú __qualname__Ú__doc__Úlistrr ÚstrrÚdictr+r,rrLrrNr r r r r sr)rr&r8ÚloggingÚtypingrrrÚpluginsrrÚ getLoggerrOrrr r r r Ús