OwlCyberSecurity - MANAGER
Edit File: update.cpython-36.pyc
3 �b�W ' � @ s` d Z ddlZddlZddlZddlZddlZddlZddlZddl m Z G dd� dej j �Z dS )zDNS Dynamic Update Support� N� )� string_typesc sx e Zd Zejjddejjf� fdd� Zddd�Z dd� Z dd � Z d d � Z d d � Z dd� Zddd�Zd� fdd� Z� ZS )�UpdateNc s� t t| �j� | jtjjtjj�O _t|t �r>tj j |�}|| _ t|t �rZtj j |�}|| _| j| j| j |tjjddd� |dk r�| j|||d� dS )a� Initialize a new DNS Update object. @param zone: The zone which is being updated. @type zone: A dns.name.Name or string @param rdclass: The class of the zone; defaults to dns.rdataclass.IN. @type rdclass: An int designating the class, or a string whose value is the name of a class. @param keyring: The TSIG keyring to use; defaults to None. @type keyring: dict @param keyname: The name of the TSIG key to use; defaults to None. The key must be defined in the keyring. If a keyring is specified but a keyname is not, then the key used will be the first key in the keyring. Note that the order of keys in a dictionary is not defined, so applications should supply a keyname when a keyring is used, unless they know the keyring contains only one key. @type keyname: dns.name.Name or string @param keyalgorithm: The TSIG algorithm to use; defaults to dns.tsig.default_algorithm. Constants for TSIG algorithms are defined in dns.tsig, and the currently implemented algorithms are HMAC_MD5, HMAC_SHA1, HMAC_SHA224, HMAC_SHA256, HMAC_SHA384, and HMAC_SHA512. @type keyalgorithm: string T)ZcreateZ force_uniqueN)� algorithm)�superr �__init__�flags�dnsZopcodeZto_flagsZUPDATE� isinstancer �name� from_text�origin� rdataclass� zone_rdclass� find_rrsetZquestion� rdatatypeZSOAZuse_tsig)�selfZzoneZrdclassZkeyringZkeynameZ keyalgorithm)� __class__� � /usr/lib/python3.6/update.pyr s zUpdate.__init__c C sB |dkr| j }|j� }| j||| j|j||dd�}|j||� dS )z&Add a single RR to the update section.NT)� authority�coversr r �rdtype�add)r r �ttl�rdZdeleting�sectionr Zrrsetr r r �_add_rrE s zUpdate._add_rrc G s: t |t�rtjj|d�}t |d tjj�rnx@|D ]8}|rF| j||j� x |D ]}| j ||j ||d� qLW q0W n�t |�}t |j d��}t |d tjj�r�|r�| j||d j� x�|D ]}| j ||||d� q�W nh|j d�}t |t�r�tjj|�}|r�| j||� x6|D ].} tjj| j|| | j�}| j ||||d� �qW dS )a� Add records. The first argument is the replace mode. If false, RRs are added to an existing RRset; if true, the RRset is replaced with the specified contents. The second argument is the section to add to. The third argument is always a name. The other arguments can be: - rdataset... - ttl, rdata... - ttl, rdtype, string...Nr )r )r r r r r �rdataset�Rdataset�deleter r r �list�int�pop�rdata�Rdatar r r ) r �replacer r �args�rdsr r r �sr r r �_addO s0 z Update._addc G s | j d| j|f|�� dS )z�Add records. The first argument is always a name. The other arguments can be: - rdataset... - ttl, rdata... - ttl, rdtype, string...FN)r* r )r r r' r r r r w s z Update.addc G sf t |t�rtjj|d�}t|�dkrR| j| j|tjj tj j tj j tj j dd� �nt |d tj j �r�x�|D ]&}x |D ]}| j|d|tjj � qtW qjW n�t|�}t |d tjj�r�x�|D ]}| j|d|tjj � q�W n�|jd�}t |t�r�tj j|�}t|�dk�r(| j| j|| j|tj j tjj dd� n:x8|D ]0}tjj| j||| j�}| j|d|tjj � �q.W dS )z�Delete records. The first argument is always a name. The other arguments can be: - I{nothing} - rdataset... - rdata... - rdtype, [string...]Nr T)r r r r r �lenr r r �ANYr �NONEr r r r! r$ r% r# r r )r r r' r( r r r) r r r r � s6 z Update.deletec G s | j d| j|f|�� dS )aQ Replace records. The first argument is always a name. The other arguments can be: - rdataset... - ttl, rdata... - ttl, rdtype, string... Note that if you want to replace the entire node, you should do a delete of the name followed by one or more calls to add.TN)r* r )r r r' r r r r& � s zUpdate.replacec G s� t |t�rtjj|d�}t|�dkrL| j| j|tjj tj j tj j ddd� n�t |d tj j �s|t |d tjj�s|t|�dkr�t |d tj j �s�t|�}|jdd� | jd| j|f|�� n@|d }t |t�r�tj j|�}| j| j|tjj |tj j ddd� dS )aB Require that an owner name (and optionally an rdata type, or specific rdataset) exists as a prerequisite to the execution of the update. The first argument is always a name. The other arguments can be: - rdataset... - rdata... - rdtype, string...Nr Tr F)r r r r r r+ r �answerr r, r r- r r r$ r% r! �insertr* )r r r' r r r r �present� s* zUpdate.presentc C s� t |t�rtjj|d�}|dkrH| j| j|tjjtj j tj jddd� n8t |t�r^tj j|�}| j| j|tjj|tj jddd� dS )z�Require that an owner name (and optionally an rdata type) does not exist as a prerequisite to the execution of the update.NT) r r r r r r r. r r- r r, )r r r r r r �absent� s z Update.absent� c s |dkr| j }tt| �j||�S )zcReturn a string containing the update in DNS compressed wire format. @rtype: stringN)r r r �to_wire)r r Zmax_size)r r r r3 � s zUpdate.to_wire)NN)N)Nr2 )�__name__� __module__� __qualname__r r �INZtsigZdefault_algorithmr r r* r r r& r0 r1 r3 � __classcell__r r )r r r s % ( +$ r )�__doc__Z dns.messager Zdns.nameZ dns.opcodeZ dns.rdataZdns.rdataclassZ dns.rdatasetZdns.tsigZ_compatr �messageZMessager r r r r �<module> s