File: //lib/python3/dist-packages/samba/__pycache__/schema.cpython-310.pyc
o
�Kya�' � @ s� d Z ddlmZ ddlmZmZmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZ dd lmZ dd
lmZmZ i fd d �ZG d
d� de�Zdd� Zdd� Z ddd�ZdS )z(Functions for setting up a Samba Schema.� )� b64encode)�read_and_sub_file�substitute_var�check_all_substituted)�security)�read_ms_schema)�ndr_pack)�SamDB��
get_string)�dsdb)�
SCOPE_SUBTREE�SCOPE_ONELEVELc C s d}t j�|| �}t|�S )Na� O:SAG:SAD:AI(OA;;CR;e12b56b6-0a95-11d1-adbb-00c04fd8d5cd;;SA)(OA;;CR;1131f6aa-9c07-11d1-f79f-00c04fc2dcd2;;ED)(OA;;CR;1131f6ab-9c07-11d1-f79f-00c04fc2dcd2;;ED)(OA;;CR;1131f6ac-9c07-11d1-f79f-00c04fc2dcd2;;ED)(OA;;CR;1131f6aa-9c07-11d1-f79f-00c04fc2dcd2;;BA)(OA;;CR;1131f6ab-9c07-11d1-f79f-00c04fc2dcd2;;BA)(OA;;CR;1131f6ac-9c07-11d1-f79f-00c04fc2dcd2;;BA)(A;CI;RPLCLORC;;;AU)(A;CI;RPWPCRCCLCLORCWOWDSW;;;SA)(A;CI;RPWPCRCCDCLCLORCWOWDSDDTSW;;;SY)(OA;;CR;1131f6ad-9c07-11d1-f79f-00c04fc2dcd2;;ED)(OA;;CR;89e95b76-444d-4c62-991a-0facbeda640c;;ED)(OA;;CR;1131f6ad-9c07-11d1-f79f-00c04fc2dcd2;;BA)(OA;;CR;89e95b76-444d-4c62-991a-0facbeda640c;;BA)(OA;;CR;1131f6aa-9c07-11d1-f79f-00c04fc2dcd2;;ER)(OA;;CR;1131f6ad-9c07-11d1-f79f-00c04fc2dcd2;;ER)(OA;;CR;89e95b76-444d-4c62-991a-0facbeda640c;;ER)S:(AU;SA;WPCCDCWOWDSDDTSW;;;WD)(AU;CISA;WP;;;WD)(AU;SA;CR;;;BA)(AU;SA;CR;;;DU)(OU;SA;CR;e12b56b6-0a95-11d1-adbb-00c04fd8d5cd;;WD)(OU;SA;CR;45ec5156-db7e-47bb-b53f-dbeb2d03c40f;;WD))r �
descriptor� from_sddlr )�
domain_sid�name_map�sddl�sec� r �./usr/lib/python3/dist-packages/samba/schema.py�get_schema_descriptor$ s r c @ sj e Zd Zddddd�Z ddd�Zed d
� �Zed d � �Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dS )�Schema)z"MS-AD_Schema_2K8_R2_Attributes.txtz MS-AD_Schema_2K8_R2_Classes.txt�/ )z0Attributes_for_AD_DS__Windows_Server_2008_R2.ldfz-Classes_for_AD_DS__Windows_Server_2008_R2.ldfr )z)AD_DS_Attributes__Windows_Server_2012.ldfz&AD_DS_Classes__Windows_Server_2012.ldf�8 )z,AD_DS_Attributes__Windows_Server_2012_R2.ldfz)AD_DS_Classes__Windows_Server_2012_R2.ldf�E )� 2008_R2_old�2008_R2�2012�2012_R2Nc s� ddl m} |d u rt�� }|| _|| _tddd�| _|d ur&| j�|� t |dtj
| d �|dtj
| d ��| _ dd� � |d urSd � � fd
d �|D ��| _ t
| j d |i�| _ t| j � tt�|��} t|d
�|| d��| _tt|���d�}
t|d�||
d��| _|d ur�|| _n� |d��| _|d ur�| jd � dd � |D ��7 _t| j��d�| _d| j| jf } | �| | j | j� d S )Nr )�
setup_pathF)�
global_schema�am_rodcz ad-schema/%s� c S s6 t | d�� }|�� W d � S 1 sw Y d S )N�rb)�open�read)�file� data_filer r r � read_fileq s $�z"Schema.__init__.<locals>.read_file� c 3 s � | ] }t � |��V qd S �Nr
)�.0r' �r) r r � <genexpr>v s � �z"Schema.__init__.<locals>.<genexpr>�SCHEMADNz#provision_schema_basedn_modify.ldif)r/ �
OBJVERSION�utf8z provision_schema_basedn.ldif)r/ �
DESCRIPTORz
prefixMap.txtc s s � | ]}d | V qdS )z%s
Nr )r, �mapr r r r. � s � zdn: %s
prefixMap:: %s
)�samba.provisionr r �default_base_schema� base_schema�schemadnr �ldb�set_invocation_idr � base_schemas� schema_data�joinr r �str� get_versionr �schema_dn_modifyr r �decode�
schema_dn_add�prefixmap_data�
set_from_ldif) �selfr � invocationidr7 �files�override_prefixmap�additional_prefixmapr6 r �schema_version�descr�prefixmap_ldifr r- r �__init__R sP
�
��
�� zSchema.__init__c C s dS )z&Returns the default base schema to user r r r r r r5 � s zSchema.default_base_schemac C s t j| d S )z=Returns the base schema's object version, e.g. 47 for 2008_R2� )r r: )r6 r r r r> � s zSchema.get_versionc C s t �| j|||� d S r+ )r �_dsdb_set_schema_from_ldifr8 )rD �pf�df�dnr r r rC � s zSchema.set_from_ldifc C s| | j j|d� | j �� z!| j �d� | jd }| j �|� | j �| j� | j �| j� W n | j �� � | j � � d S )N)�urlzpdn: @ATTRIBUTES
linkID: INTEGER
dn: @INDEXLIST
@IDXATTR: linkID
@IDXATTR: attributeSyntax
@IDXGUID: objectGUID
z1objectGUID: 24e2ca70-b093-4ae8-84c0-2d7ac652a1b8
)
r8 �connect�transaction_start�add_ldifrA � modify_ldifr? r; �transaction_cancel�transaction_commit)rD �
schemadb_pathrA r r r �write_to_tmp_ldb� s
�
zSchema.write_to_tmp_ldbc C � t | j| j�S r+ )�get_linked_attributesr7 r8 �rD r r r �linked_attributes� � zSchema.linked_attributesc C r[ r+ )�get_dnsyntax_attributesr7 r8 r] r r r �dnsyntax_attributes� r_ zSchema.dnsyntax_attributesc C s t �| j||�S r+ )r � _dsdb_convert_schema_to_openldapr8 )rD �target�mappingr r r �convert_to_openldap� s zSchema.convert_to_openldap)NNNNNN)
�__name__�
__module__� __qualname__r: rL � staticmethodr5 r> rC rZ r^ ra re r r r r r @ s& �
�C
r c C s� ddg}|j d| t|d�}i }tdt|��D ])}dt|| d d �d }|j| |dtd�}|d ur?t|�|t|| d �< q|S ) N�linkID�lDAPDisplayNamezh(&(linkID=*)(!(linkID:1.2.840.113556.1.4.803:=1))(objectclass=attributeSchema)(attributeSyntax=2.5.5.1))��
expression�base�scope�attrsr zD(&(objectclass=attributeSchema)(linkID=%d)(attributeSyntax=2.5.5.1))r# )�basednrm � attributero )�searchr �range�len�int� searchoner
r= )r7 � schemaldbrp �res�
attributes�irm rc r r r r\ � s& ����r\ c C sH |j d| tddgd�}g }tdt|��D ]
}|�t|| d �� q|S )NzF(&(!(linkID=*))(objectclass=attributeSchema)(attributeSyntax=2.5.5.1))rj rk rl r )rs r rt ru �appendr= )r7 rx ry rz r{ r r r r` � s �r` �,cn=schema,cn=configuration,dc=example,dc=comNc C s* |du r t �� }nt �|�}t|| |d�S )aO Load schema for the SamDB from the AD schema files and samba4_schema.ldif
:param schemadn: DN of the schema
:param serverdn: DN of the server
Returns the schema data loaded as an object, with .ldb being a
new ldb with the schema loaded. This allows certain tests to
operate without a remote or local schema.
N)r7 rG )r �
random_sid�dom_sidr )r7 � domainsidrG r r r �ldb_with_schema� s
�r� )r} NN)�__doc__�base64r �sambar r r � samba.dcerpcr �samba.ms_schemar � samba.ndrr � samba.samdbr � samba.commonr r r8 r
r r �objectr r\ r` r� r r r r �<module> s& �