Path: blob/master/__pycache__/MaltegoTransform.cpython-38.pyc
329 views
U
<� `s � @ s~ d dl Z d dlmZ dZdZdZdZdZdZdZ dZ
dZ dZ d Z
d
Zd Zd ZG d
d� de�ZG dd� de�ZG dd� d�ZdS )� N)�minidomz-1�0�1�2�3�4Z
FatalErrorZ PartialError�InformZDebugc @ s� e Zd ZdZdZg Zg ZdZdZd#dd�Z d$dd�Z
d%d d
�Z d&d d �Z d'dd�Z
d(dd�Zd)dd�Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� ZdS )*�
MaltegoEntity� �d ZPhraseNc C s> |d k r|| _ |d k r || _d | _g | _d| _g | _d| _d S )Nr r
)�
entityType�value�additionalFields�weight�displayInformation�iconURL)�self�eT�v� r �6/home/neptune/Osint/holehe-maltego/MaltegoTransform.py�__init__# s zMaltegoEntity.__init__c C s |d k r|| _ d S �N)r )r r r r r �setType. s zMaltegoEntity.setTypec C s |d k r|| _ d S r )r
)r ZeVr r r �setValue2 s zMaltegoEntity.setValuec C s |d k r|| _ d S r )r )r �wr r r � setWeight6 s zMaltegoEntity.setWeight�Infoc C s |d k r| j �||g� d S r )r �append)r ZdiZdlr r r �addDisplayInformation: s z#MaltegoEntity.addDisplayInformationFc C s | j �||||g� d S r )r r )r Z fieldNameZ displayNameZ matchingRuler
r r r � addProperty> s zMaltegoEntity.addPropertyc C s |d k r|| _ d S r )r )r ZiUr r r �
setIconURLA s zMaltegoEntity.setIconURLc C s | � ddd|� d S )Nzlink#maltego.link.colorZ LinkColorr
�r )r Zcolorr r r � setLinkColorE s zMaltegoEntity.setLinkColorc C s | � ddd|� d S )Nzlink#maltego.link.styleZ LinkStyler
r" )r Zstyler r r � setLinkStyleH s zMaltegoEntity.setLinkStylec C s | � dddt|�� d S )Nzlink#maltego.link.thicknessZ Thicknessr
)r �str)r Zthickr r r �setLinkThicknessK s z MaltegoEntity.setLinkThicknessc C s | � ddd|� d S )Nzlink#maltego.link.labelZLabelr
r" )r Zlabelr r r � setLinkLabelN s zMaltegoEntity.setLinkLabelc C s | � ddd|� d S )Nz bookmark#ZBookmarkr
r" )r Zbookmarkr r r � setBookmarkQ s zMaltegoEntity.setBookmarkc C s | � ddd|� d S )Nznotes#ZNotesr
r" )r Znoter r r �setNoteT s zMaltegoEntity.setNotec C s� d}|dt | j� d 7 }|dt | j� d 7 }|dt | j� d 7 }t| j�dkr�|d 7 }tt| j��D ]4}|d
| j| d d t | j| d � d
7 }qj|d7 }t| j�dk�r�|d7 }tt| j��D ]�}t | j| d �dk�r6|dt | j| d � d t | j| d � d t | j| d � d 7 }q�|dt | j| d � d t | j| d � d t | j| d � d t | j| d � d 7 }q�|d7 }t| j�dk�r�|d| j d 7 }|d7 }|S ) Nr
z<Entity Type="�">z<Value>z</Value>z<Weight>z </Weight>r z<DisplayInformation>z
<Label Name="z " Type="text/html"><![CDATA[� z ]]></Label>z</DisplayInformation>z<AdditionalFields>� �strictz
<Field Name="z" DisplayName="� z</Field>z<Field MatchingRule="z" Name="z</AdditionalFields>z <IconURL>z
</IconURL>z </Entity>) r% r r
r �lenr �ranger r �r �r�ir r r � returnEntityW s( 2LbzMaltegoEntity.returnEntity)NN)N)N)N)Nr )NNFN)N)�__name__�
__module__� __qualname__r
r r r r r r r r r r r r! r# r$ r& r' r( r) r4 r r r r r s(
r c @ sL e Zd Zg Zg Zg Zdd� Zddd�Zddd�Zd d
� Z d d � Z
d
d� Z dS )�MaltegoTransformc C s g | _ g | _g | _d } d S r )�entities�
exceptions�
UIMessages)r r r r r w s zMaltegoTransform.__init__Nc C s t ||�}| j�|� |S r )r r9 r )r ZenTypeZenValue�mer r r � addEntity} s
zMaltegoTransform.addEntityr c C s | j �||g� d S r )r; r )r �messageZ messageTyper r r � addUIMessage� s z MaltegoTransform.addUIMessagec C s | j �|� d S r )r: r )r ZexceptionStringr r r � addException� s z MaltegoTransform.addExceptionc C sb d}|d7 }|d7 }|d7 }t t| j��D ]}|d| j| d 7 }q*|d7 }|d7 }|d7 }|S ) Nr
�<MaltegoMessage>z"<MaltegoTransformExceptionMessage>z <Exceptions>z <Exception>z
</Exceptions>z#</MaltegoTransformExceptionMessage>�</MaltegoMessage>)r0 r/ r: r1 r r r �throwExceptions� s z MaltegoTransform.throwExceptionsc C s� d}|d7 }|d7 }|d7 }t t| j��D ]}|| j| �� 7 }q*|d7 }|d7 }t t| j��D ]0}|d| j| d d | j| d
d 7 }q`|d 7 }|d
7 }|d7 }|S )Nr
rA z!<MaltegoTransformResponseMessage>z
<Entities>z </Entities>z <UIMessages>z<UIMessage MessageType="r r* r+ z </UIMessage>z
</UIMessages>z"</MaltegoTransformResponseMessage>rB )r0 r/ r9 r4 r; r1 r r r � returnOutput� s .z MaltegoTransform.returnOutput)NN)r ) r5 r6 r7 r9 r: r; r r= r? r@ rC rD r r r r r8 o s
r8 c @ s> e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )�
MaltegoMsgr
c C s t �|�}| �|d�| _| �|d�| _t| �|dd��| _| �|dd�| _i }zF|� d�d }|� d �}|D ]$}|j
d
j }| � |j
�}|||< qlW n d } Y nX i }
zF|� d �d } | � d �}|D ]$}|j
d
j } | � |j
�}
|
|
| < q�W n d } Y nX || _|
| _d S )
N�Value�WeightZLimitsZ SoftLimitZEntity�TypeZAdditionalFieldsr ZField�Namer+ ZTransformFields)r Z parseString�i_getNodeValuerF rG �int�i_getNodeAttributeValueZSliderrH �getElementsByTagName�
attributesr
� i_getText�
childNodes�
Properties�TransformSettings)r Z
MaltegoXMLZxmldocrQ ZAFNodesZSettings�nodeZAFNameZAFValueZdontcarerR ZTSNodesZTSNameZTSValuer r r r � s6
zMaltegoMsg.__init__c C s0 g }|D ] }|j |jkr|�|j� qd�|�S )Nr
)ZnodeTypeZ TEXT_NODEr �data�join)r ZnodelistZrcrS r r r rO � s
zMaltegoMsg.i_getTextc C s | � |�|�d j�S �Nr )rO rM rP )r rS �Tagr r r rJ � s zMaltegoMsg.i_getNodeValuec C s |� |�d j| jS rV )rM rN r
)r rS rW � Attributer r r rL � s z"MaltegoMsg.i_getNodeAttributeValuec C s || j �� kr| j | S d S d S r )rQ �keys�r Zskeyr r r � getProperty� s
zMaltegoMsg.getPropertyc C s || j �� kr| j | S d S d S r )rR rY rZ r r r �getTransformSetting� s
z MaltegoMsg.getTransformSettingN)r
) r5 r6 r7 r rO rJ rL r[ r\ r r r r rE � s
'rE )�sysZxml.domr ZBOOKMARK_COLOR_NONEZBOOKMARK_COLOR_BLUEZBOOKMARK_COLOR_GREENZBOOKMARK_COLOR_YELLOWZBOOKMARK_COLOR_ORANGEZBOOKMARK_COLOR_REDZLINK_STYLE_NORMALZLINK_STYLE_DASHEDZLINK_STYLE_DOTTEDZLINK_STYLE_DASHDOTZ UIM_FATALZ UIM_PARTIALZ
UIM_INFORMZ UIM_DEBUG�objectr r8 rE r r r r �<module> s$ T7