Attachment 7837 Details for Bug 41006 – sync_mx_and_txt_dns_records.diff

[patch] sync_mx_and_txt_dns_records.diff

sync_mx_and_txt_dns_records.diff (text/plain), 9.15 KB, created by Arvid Requate on 2016-07-26 18:09:32 CEST

(hide)

Description:

Filename:

MIME Type:

Creator: Arvid Requate

Created: 2016-07-26 18:09:32 CEST

Size: 9.15 KB

patch

obsolete

>Index: modules/univention/s4connector/s4/dns.py
>===================================================================
>--- modules/univention/s4connector/s4/dns.py	(Revision 70207)
>+++ modules/univention/s4connector/s4/dns.py	(Arbeitskopie)
>@@ -31,6 +31,7 @@
> # /usr/share/common-licenses/AGPL-3; if not, see
> # <http://www.gnu.org/licenses/>.
> 
>+from __future__ import absolute_import
> import ldap, string
> from ldap.filter import escape_filter_chars, filter_format
> import univention.debug2 as ud
>@@ -46,6 +47,11 @@
> import time
> import sys
> 
>+from dns.rdtypes.ANY.TXT import TXT
>+from dns import rdatatype
>+from dns import rdataclass
>+from dns.tokenizer import Tokenizer
>+
> from samba.provision.sambadns import ARecord
> # def __init__(self, ip_addr, serial=1, ttl=3600):
> 
>@@ -86,6 +92,17 @@
> 		self.data.wPriority=priority
> 		self.data.nameTarget=name
> 
>+class TXTRecord(dnsp.DnssrvRpcRecord):
>+	def __init__(self, slist, priority=0, serial=1, ttl=3600):
>+		super(TXTRecord, self).__init__()
>+		self.wType=dnsp.DNS_TYPE_TXT
>+		self.dwSerial=serial
>+		self.dwTtlSeconds=ttl
>+		stringlist = dnsp.string_list()
>+		stringlist.count = len(slist)
>+		stringlist.str = slist
>+		self.data = stringlist
>+
> import univention.admin.handlers
> import univention.admin.handlers.dns.forward_zone
> import univention.admin.handlers.dns.alias
>@@ -571,6 +588,28 @@
> 			mx.append( [str(ndrRecord.data.wPriority), __append_dot(ndrRecord.data.nameTarget)] )
> 	return mx
> 
>+def __pack_txtRecord(object, dnsRecords):
>+	slist=[]
>+	for txtRecord in object['attributes'].get('tXTRecord', []):
>+		if txtRecord:
>+			ud.debug(ud.LDAP, ud.INFO, '__pack_txtRecord: %s' % txtRecord)
>+			txtRecord=univention.s4connector.s4.compatible_modstring(txtRecord)
>+			token_list = TXT.from_text(rdataclass.IN, rdatatype.TXT, Tokenizer(txtRecord)).strings
>+			ndr_txt_record = ndr_pack(TXTRecord(token_list))
>+			dnsRecords.append(ndr_txt_record)
>+			ud.debug(ud.LDAP, ud.INFO, '__pack_txtRecord: %s' % ndr_txt_record)
>+
>+def __unpack_txtRecord(object):
>+	txt=[]
>+	dnsRecords=object['attributes'].get('dnsRecord', [])
>+	for dnsRecord in dnsRecords:
>+		dnsRecord=dnsRecord.encode('latin1')
>+		ndrRecord=ndr_unpack(dnsp.DnssrvRpcRecord, dnsRecord)
>+		if ndrRecord.wType == dnsp.DNS_TYPE_TXT:
>+			txt.append(str(TXT(rdataclass.IN, rdatatype.TXT, ndrRecord.data.str)))
>+			# or: txt.append(' '.join(['"%s"' % token for token in ndrRecord.data.str]))
>+	return txt
>+
> def __pack_cName(object, dnsRecords):
> 	for c in object['attributes'].get('cNAMERecord', []):
> 		c=univention.s4connector.s4.compatible_modstring(__remove_dot(c))
>@@ -717,6 +756,8 @@
> 
> 	__pack_mxRecord(object, dnsRecords)
> 
>+	__pack_txtRecord(object, dnsRecords)
>+
> 	s4connector.lo_s4.modify(soa_dn, [('dnsRecord', old_dnsRecords, dnsRecords)])
> 
> 	return True
>@@ -870,6 +911,9 @@
> 	else:
> 		__pack_aRecord(object, dnsRecords)
> 
>+	__pack_mxRecord(object, dnsRecords)
>+	__pack_txtRecord(object, dnsRecords)
>+
> 	dnsNodeDn=s4_dns_node_base_create(s4connector, object, dnsRecords)
> 
> 	return True
>@@ -973,7 +1017,7 @@
> 			newRecord['ptr_record']=ptr[0]
> 			newRecord.modify()
> 		else:
>-			ud.debug(ud.LDAP, ud.INFO, 'ucs_ptr_record_create: do not modify host record')
>+			ud.debug(ud.LDAP, ud.INFO, 'ucs_ptr_record_create: do not modify ptr record')
> 	else:
> 		zoneDN='zoneName=%s,%s' % (zoneName, s4connector.property['dns'].ucs_default_dn)
> 
>@@ -1029,7 +1073,7 @@
> 			newRecord['cname']=c[0]
> 			newRecord.modify()
> 		else:
>-			ud.debug(ud.LDAP, ud.INFO, 'ucs_cname_create: do not modify host record')
>+			ud.debug(ud.LDAP, ud.INFO, 'ucs_cname_create: do not modify cname record')
> 	else:
> 		zoneDN='zoneName=%s,%s' % (zoneName, s4connector.property['dns'].ucs_default_dn)
> 
>@@ -1108,7 +1152,7 @@
> 				newRecord['location']=srv
> 				newRecord.modify()
> 			else:
>-				ud.debug(ud.LDAP, ud.INFO, 'ucs_srv_record_create: do not modify host record')
>+				ud.debug(ud.LDAP, ud.INFO, 'ucs_srv_record_create: do not modify srv record')
> 	else:
> 		zoneDN='zoneName=%s,%s' % (zoneName, s4connector.property['dns'].ucs_default_dn)
> 
>@@ -1189,6 +1233,79 @@
> 	dnsNodeDn=s4_dns_node_base_create(s4connector, object, dnsRecords)
> 
> 	
>+def ucs_txt_record_create(s4connector, object):
>+	_d=ud.function('ucs_txt_record_create')
>+	ud.debug(ud.LDAP, ud.INFO, 'ucs_txt_record_create: object: %s' % object)
>+	udm_property='txt'
>+
>+	zoneName = object['attributes']['zoneName'][0]
>+	relativeDomainName = object['attributes']['relativeDomainName'][0]
>+
>+	# unpack the record
>+	c = __unpack_txtRecord(object)
>+
>+	# Does a host record for this zone already exist?
>+	ol_filter = filter_format('(&(relativeDomainName=%s)(zoneName=%s))', (relativeDomainName, zoneName))
>+	searchResult=s4connector.lo.search(filter=ol_filter, unique=1)
>+	if len(searchResult) > 0:
>+		superordinate=s4connector_get_superordinate('dns/txt_record', s4connector.lo, searchResult[0][0])
>+		foundRecord= univention.admin.handlers.dns.txt_record.object(None, s4connector.lo, position=None, dn=searchResult[0][0], superordinate=superordinate, attributes=[], update_zone=False)
>+		foundRecord.open()
>+
>+		## use normalized TXT records for comparison
>+		normalized_txtRecord_list = []
>+		for txtRecord in foundRecord['txt']:
>+			normalized_txtRecord = str(TXT.from_text(rdataclass.IN, rdatatype.TXT, Tokenizer(txtRecord)))
>+			normalized_txtRecord_list.append(normalized_txtRecord)
>+		
>+		if set(normalized_txtRecord_list) != set(c):
>+			foundRecord[udm_property]=c
>+			foundRecord.modify()
>+		else:
>+			ud.debug(ud.LDAP, ud.INFO, 'ucs_txt_record_create: do not modify txt record')
>+	else:
>+		zoneDN='zoneName=%s,%s' % (zoneName, s4connector.property['dns'].ucs_default_dn)
>+
>+		superordinate=s4connector_get_superordinate('dns/txt_record', s4connector.lo, zoneDN)
>+		ud.debug(ud.LDAP, ud.INFO, 'ucs_txt_record_create: superordinate: %s' % superordinate)
>+
>+		position=univention.admin.uldap.position(zoneDN)
>+
>+		newRecord= univention.admin.handlers.dns.txt_record.object(None, s4connector.lo, position, dn=None, superordinate=superordinate, attributes=[], update_zone=False)
>+		newRecord.open()
>+		newRecord['name']=relativeDomainName
>+		newRecord[udm_property]=c
>+		newRecord.create()
>+	
>+
>+def ucs_txt_record_delete(s4connector, object):
>+	_d=ud.function('ucs_txt_record_delete')
>+	ud.debug(ud.LDAP, ud.INFO, 'ucs_txt_record_delete: object: %s' % object)
>+
>+	zoneName = object['attributes']['zoneName'][0]
>+	relativeDomainName = object['attributes']['relativeDomainName'][0]
>+
>+	ol_filter = filter_format('(&(relativeDomainName=%s)(zoneName=%s))', (relativeDomainName, zoneName))
>+	searchResult=s4connector.lo.search(filter=ol_filter, unique=1)
>+	if len(searchResult) > 0:
>+		superordinate=s4connector_get_superordinate('dns/txt_record', s4connector.lo, searchResult[0][0])
>+		newRecord= univention.admin.handlers.dns.txt_record.object(None, s4connector.lo, position=None, dn=searchResult[0][0], superordinate=superordinate, attributes=[], update_zone=False)
>+		newRecord.open()
>+		newRecord.delete()
>+	else:
>+		ud.debug(ud.LDAP, ud.INFO, 'ucs_txt_record_delete: Object was not found, filter was: %s' % ol_filter)
>+
>+	return True
>+	
>+def s4_txt_record_create(s4connector, object):
>+	_d=ud.function('s4_txt_record_create')
>+
>+	dnsRecords=[]
>+
>+	__pack_txtRecord(object, dnsRecords)
>+
>+	dnsNodeDn=s4_dns_node_base_create(s4connector, object, dnsRecords)
>+
> def ucs_zone_create(s4connector, object, dns_type):
> 	_d=ud.function('ucs_zone_create')
> 
>@@ -1328,6 +1445,8 @@
> 			return 'srv_record'
> 		if univention.admin.handlers.dns.ptr_record.identify(object['dn'], object['attributes']):
> 			return 'ptr_record'
>+		if univention.admin.handlers.dns.txt_record.identify(object['dn'], object['attributes']):
>+			return 'txt_record'
> 	return None
> 
> def _identify_dns_con_object(s4connector, object):
>@@ -1374,6 +1493,8 @@
> 					return 'srv_record'
> 				elif set((dnsp.DNS_TYPE_A, dnsp.DNS_TYPE_AAAA)) & dns_types:
> 					return 'host_record'
>+				elif dnsp.DNS_TYPE_TXT in dns_types:
>+					return 'txt_record'
> 				
> 	return None
> 	
>@@ -1436,6 +1557,13 @@
> 			s4_dns_node_base_delete(s4connector, object)
> 		# ignore move
> 
>+	elif dns_type == 'txt_record':
>+		if object['modtype'] in ['add', 'modify']:
>+			s4_txt_record_create(s4connector, object)
>+		elif object['modtype'] in ['delete']:
>+			s4_dns_node_base_delete(s4connector, object)
>+		# ignore move
>+
> 	return True
> 
> def con2ucs (s4connector, key, object):
>@@ -1486,6 +1614,12 @@
> 		elif object['modtype'] in ['delete']:
> 			ucs_srv_record_delete(s4connector, object)
> 		# ignore move
>+	elif dns_type == 'txt_record':
>+		if object['modtype'] in ['add', 'modify']:
>+			ucs_txt_record_create(s4connector, object)
>+		elif object['modtype'] in ['delete']:
>+			ucs_txt_record_delete(s4connector, object)
>+		# ignore move
> 	if dns_type in ['forward_zone', 'reverse_zone']:
> 		if object['modtype'] in ['add', 'modify']:
> 			ucs_zone_create(s4connector, object, dns_type)
>@@ -1509,7 +1643,8 @@
> 			univention.admin.handlers.dns.alias.identify(dn, attr) or\
> 			univention.admin.handlers.dns.host_record.identify(dn, attr) or\
> 			univention.admin.handlers.dns.srv_record.identify(dn, attr) or\
>-			univention.admin.handlers.dns.ptr_record.identify(dn, attr) 
>+			univention.admin.handlers.dns.ptr_record.identify(dn, attr) or\
>+			univention.admin.handlers.dns.txt_record.identify(dn, attr) 
>  
> '''
> 	Because the dns/dns.py identify function has been overwritten

Actions: View | Diff

Attachments on bug 41006: 7831 | 7833 | 7837 | 7840