File: //usr/local/CyberCP/plogical/__pycache__/dnsUtilities.cpython-310.pyc
o
��hɚ � @ s� d dl Z d dlZej�d� d dlZe j�dd� ze�� W n Y d dlm Z
d dl Z d dl Z zd dl
mZmZ d dlmZmZ W n Y d dlZd dlmZ G d d
� d
�ZdS ) � Nz/usr/local/CyberCP�DJANGO_SETTINGS_MODULEzCyberCP.settings)�CyberCPLogFileWriter)�Domains�Records)�
PDNSStatus� SlaveServers)�ProcessUtilitiesc @ s� e Zd ZdZdZdZdZdZdd� Zd#d d
�Z e
d d � �Z e
d
d� �Z e
dd� �Z
e
dd� �Ze
dd� �Ze
dd� �Ze
dd� �Ze
dd� �Ze
dd � �Ze
d d � �Ze
d d � �Ze
d!d"� �ZdS )$�DNSz/etc/nsd/nsd.confz/usr/local/lsws/conf/zones/z/usr/local/lsws/conf/zonesz#/home/cyberpanel/defaultNameserversz/home/cyberpanel/CloudFlarec C sd dt j| jjf }tj�|�r0t|d��� }|d � d�| _
|d � d�| _ |d � d�| _ dS dS )Nz%s%s�rr �
� � )
r �CFPath�admin�userName�os�path�exists�open� readlines�rstrip�email�key�status)�self�cfFile�data� r �+/usr/local/CyberCP/plogical/dnsUtilities.py�
loadCFKeys# s zDNS.loadCFKeysNc C sF �z|| _ | �� �r|d krn | jdkrnW dS tj| j| jd�}zE|dd�}|jj|d�}t|dd� d �D ].}|d
}t j
j|d �}t j
j |j
d �} | D ]}
t�|||
j|
j|
j|
j|
j� qM W W d
S W n5 tjjy� } ztj�t| �� W Y d } ~ n d } ~ w ty� } ztj�t| �� W Y d } ~ nd } ~ ww z6|jj d|d�d�} | d
}t j
j|d �}t j
j |j
d �} | D ]}
t�|||
j|
j|
j|
j|
j� q�W W d
S tjjy� } z
dt| �fW Y d } ~ W S d } ~ w t�y } z
dt| �fW Y d } ~ W S d } ~ ww W d S t �y" }
z dt| �fW Y d }
~
S d }
~
ww )N�Enable)r zSync not enabled.�r �token�2 ��name�per_page��paramsc S � | d S �Nr% r ��vr r r �<lambda>E � z DNS.cfTemplate.<locals>.<lambda>�r �id�r% )� domain_id�r NF)�
jump_startr% �r r ) r r r �
CloudFlarer r �zones�get�sortedr �objectsr �filterr0 r �createDNSRecordCloudFlarer% �type�content�prio�ttl�
exceptions�CloudFlareAPIError�loggingr � writeToFile�str� Exception�post�
BaseException)r �
zoneDomainr � enableCheck�cfr( r7 �zone�domain�records�record�e� zone_info�msgr r r �
cfTemplate0 sf
�
�
� �������5��zDNS.cfTemplatec C s� �z�d}t |�}|�� }|�dd�d }dd l}|jd d�}|| �}|jd |j } |j}
t|
�dk�rat j
j | d�� � dk�r_z<t
j
jdd�} | jd krtt || d d
�} | �� tj
�� D ]}
t| | j| d |
jd dddd
� }|�� q]nt || dd
�} W n t || dd
�} Y | �� | jdkr�tj�tj�r�t tjd��� }|D ] }
t|
�dkr�t| | j| d |
� d�d dddd
� }|�� q�n(t| | j| d d| d dddd
� }|�� t| | j| d d| d dddd
� }|�� d| d | d }t� | | d|dd � t� | | d|dd � d| }t� | |d| dd � d| }t� | |d| dd � | }t� | | d|d d � t� | |d|dd � t� | | d d | d dd � t� | d | d d!dd � t� | d"| d d#dd � �n t j
j | d�� � dk�rz t
j
jdd�} | jd k�r�t || d d
�} nt || dd
�} W n t || dd
�} Y | �� d| d | d }t� | | d|dd � t� | | d|dd � d| }t� | |d| dd � d| }t� | |d| dd � | }t� | |d|d d � t� | |d|dd � t� | | d d | d dd � t� | d | d d!dd � t� | d"| d d#dd � t j
j| d�} |
d | }t� | |d|dd � d$| � d%�d&k�r>t� | d'| d|dd � d| }t� | |d|dd � |}t� | |d|d d � t� | |d d | d dd � t� | d | d d!dd � t� | d"| d d#dd � t � � t j!k�s�t � � t j"k�r�d(}t �#|� t� }|�$| |� W d S t%�y� } zt&j'�(d)| d* t)|� � W Y d }~d S d }~ww )+Nz/etc/cyberpanel/machineIPr r r �� cache_dir�.r1 )�pk�MASTER�r r% r= �NS� � � domainOwnerr2 r% r= r> r@ r? �disabled�auth�NATIVEr
� zns1.%szns2.%szns1.z hostmaster.z 1 10800 3600 1209600 3600�SOA�Azwww.�CNAMEzftp.�MX�
�TXTzv=spf1 a mx ip4:z ~allz_dmarc.zv=DMARC1; p=none;z _domainkey.z t=y; o=~;zmail.%sz mail.mail���zmail.�sudo systemctl restart pdnsz.We had errors while creating DNS records for: �. Error message: )*r �read�split�
tldextract�
TLDExtractrM �suffix� subdomain�lenr r: r; �countr r8 r= �saver �allr r0 � slaveServerr r r r �defaultNameServersPathr r �createDNSRecord�findr � decideDistro�ubuntu�ubuntu20� executionerrS rH rC r rD rE )rM r �ipFile�f�ipData� ipAddressrm �no_cache_extract�
extractDomain�topLevelDomain� subDomain�pdnsrL �itemsrO � defaultNSr> �
cNameValue�mxValue�actualSubDomain�command�dnsrR r r r � dnsTemplatem s
�
� �
� ��
� �
� �
���zDNS.dnsTemplatec C s �zddd l }|jd d�}|| �}|jd |j }|j}tjj|d�}d| d }d| }t� t
� |��� d�} | �
d �d
}
| �d �d } tjj|d
| d��� dkrlt||jd
| d| |
| � dddd d� } | �� ntjj|d
| d�}
| |
| � |
_|
�� t|�dkr�tjj|d
| d��� dkr�t||jd
| d| |
| � dddd d� } | �� ntjj|d
| d�}
| |
| � |
_|
�� t�� tjks�t�� tjkr�d}t�|� t� }|j |_ |� � �r`t j |j |j d�}|j!dk�rcz1| dd�}|j"j|d�}t#|dd� d�D ]}|d }t�$||d
| d| |
| � dd� �qW W d S t j%j&�yC } zt'j(�)t*|�� W Y d }~W d S d }~w t+�y_ } zt'j(�)t*|�� W Y d }~W d S d }~ww W d S W d S t,�y� } zt'j(�)d | d t*|� � W Y d }~d S d }~ww ) Nr rT rV r1 z/etc/opendkim/keys/z /default.txt�cat zutf-8�(r
�)r zdefault._domainkey.)r] r% rg r[ r\ z systemctl restart pdnsr! r r# r$ r' c S r) r* r r+ r r r r- z r. z'DNS.createDKIMRecords.<locals>.<lambda>r/ r0 z.We had errors while creating DKIM record for: rj )-rm rn rM ro rp r r: r8 �
subprocess� check_output�shlexrl �decode�index�rindexr r; rr r0 rs r> rq r ry rz r{ r| r r r r6 r r r r7 r9 r< rA rB rC r rD rE rF rH )rM rm r� r� r� r� rL r r� �output� leftIndex�
rightIndexrO �rcrdr� rK r( r7 rP rR r r r �createDKIMRecords1 s�
�
�
�� � ������zDNS.createDKIMRecordsc C s zt jj| d�W S Y dS )Nr1 r )r r: r8 )�virtualHostNamer r r �
getZoneObject� s zDNS.getZoneObjectc
C s� z3|� d�dkr|�dd�}|�dd�}|dkr!|||||d�}n||||d�}| jjj||d � W d S tyP } ztj�t |�d
� W Y d }~d S d }~ww ) N�DKIMrh z
� �"r )r% r= r> r@ �priority)r% r= r> r� r5 z . [createDNSRecordCloudFlare])
rx �replacer7 � dns_recordsrG rH rC r rD rE ) rK rL r% r= �valuer� r@ �
dns_recordrR r r r r< � s "��z DNS.createDNSRecordCloudFlarec C s� �z>t jj|||d��� dkrW d S | jdkrEz)t jj| dd�D ] }|j�d�}tt|d �d �|d< d� |�|_|�
� q W n Y |d kr�t jj|||d��� dkr�t | | j |||||ddd
� }|�
� t �
� t jksvt �
� t jkr�d } t �| �}
|
�d �d
kr�d} t �| � W d S |dkr�t jj|||d��� dkr�t | | j |||||ddd
� }|�
� t �
� t jks�t �
� t jkr�d } t �| �}
|
�d �d
kr�d} t �| � W d S |dk�r%t jj|||d��� dk�r"t | | j |||||ddd
� }|�
� t �
� t jk�s t �
� t jk�r"d } t �| �}
|
�d �d
k�r"d} t �| � W d S |dk�rgt | | j ||||t|�ddd
� }|�
� t �
� t jk�sNt �
� t jk�rdd } t �| �}
|
�d �d
k�rdd} t �| � W d S t jj||d��� dk�r�t | | j |||||ddd
� }|�
� t �
� t jk�s�t �
� t jk�r�d } t �| �}
|
�d �d
k�r�d} t �| � z�t� } | j| _| �� tj| j| jd�} | jdk�r)z*| jdd�}
| jj |
d�}t |dd� d�D ]} | d } t� | | |||||� �q�W n; tj j �y } zt!j"�#t|�� W Y d }~n'd }~w t$�y( } zt!j"�#t|�� W Y d }~nd }~ww W W d S W W d S W W d S W W d S Y W d S t%�y] } zt!j"�#t|�d � W Y d }~d S d }~ww ) N)r% r= r> r rX rb )r] r= � r
r rZ r\ z?ls -la /etc/systemd/system/multi-user.target.wants/pdns.servicez No such filerh ri rg re )r% r= r! r r# r$ r' c S r) r* r r+ r r r r- 2 r. z%DNS.createDNSRecord.<locals>.<lambda>r/ r0 z [createDNSRecord])&r r: r; rr r= r> rl rE �int�joinrs r0 r ry rz r{ �outputExecutionerrx r| r r r r6 r r r r% r7 r8 r9 r< rA rB rC r rD rF rH )rL r% r= r� r� r@ �getSOA�
soaContentrO r� �resultr� rK r( r7 rP rR r r r rw � s
��
�
�
�
�
� � ���"��zDNS.createDNSRecordc C �* zt jj| d�}|�� W d S Y d S �Nr1 )r r: r8 �delete)r� �delZoner r r �
deleteDNSZoneA �
zDNS.deleteDNSZonec C s* zt || dd�}|�� W d S Y d S )Nr` rY )r rs )r� r rL r r r �
createDNSZoneJ r� zDNS.createDNSZonec C s0 zt jj| d�}|�� |j�� W S Y d S r� )r r: r8 rs � records_setrt )r� rL r r r �
getDNSRecordsS s zDNS.getDNSRecordsc C s zt j�� W S Y d S �N)r r: rt r r r r � getDNSZones] s zDNS.getDNSZonesc C r� )N)r0 )r r: r8 r� )�recordID� delRecordr r r �deleteDNSRecordd r� zDNS.deleteDNSRecordc
C s zbddl m} d}| d�}t�d|� ��}|�d�dkr`d|� d �}d
|� d |� �}t�|d d � t�� tjks?t�� tjkrBd
}nd}d|� d�}d
|� d |� �}t�|d d � d}t�|� W dS t y� } zt
j � dt
|�� �� dt
|�fW Y d }~S d }~ww )Nr )�
generate_pass�/root/.acme.sh/account.conf� r� �PDNS_Urlrh zH
PDNS_Url='http://localhost:8081'
PDNS_ServerId='localhost'
PDNS_Token='�'
�echo "�" >> Tz/etc/pdns/pdns.confz/etc/powerdns/pdns.confz�
# Turn on the webserver API
webserver=yes
webserver-address=0.0.0.0
webserver-port=8081
# Set the API key for accessing the API
api=yes
api-key=z!
webserver-allow-from=0.0.0.0/0
zsystemctl restart pdnsr3 z ConfigurePowerDNSInAcme, Error: )�plogical.randomPasswordr� r r� rx r| ry �centos�cent8rH rC r rD rE ) r� r �APIKey�CurrentContent� PDNSContentr� �PDNSPath�PDNSConfrR r r r �ConfigurePowerDNSInAcmem s6 � �
��zDNS.ConfigurePowerDNSInAcmec
C s� z'd}d|� �}t �|� d| � d|� d�}d|� d|� �}t �|d d� W d S tyJ } ztj�d
t|�� �� d t|�fW Y d }~S d }~ww ) Nr� z+sed -i '/SAVED_CF_Key/d;/SAVED_CF_Email/d' z
SAVED_CF_Key='z'
SAVED_CF_Email='r� r� r� Tr3 z"ConfigureCloudflareInAcme, Error: r )r r| rH rC r rD rE )� SAVED_CF_Key�SAVED_CF_Emailr r� � CFContentrR r r r �ConfigureCloudflareInAcme� s"
����z DNS.ConfigureCloudflareInAcmer� )�__name__�
__module__� __qualname__�nsd_base�zones_base_dir�create_zone_dirrv r r rS � staticmethodr� r� r� r< rw r� r� r� r� r� r� r� r r r r r sH
=
F
Y
2r )r �sysr �append�django�environ�
setdefault�setup�plogicalr rC r� r� �
dns.modelsr r �manageServices.modelsr r r6 �plogical.processUtilitiesr r r r r r �<module> s&