Server IP : 85.214.239.14 / Your IP : 3.135.247.237 Web Server : Apache/2.4.62 (Debian) System : Linux h2886529.stratoserver.net 4.9.0 #1 SMP Tue Jan 9 19:45:01 MSK 2024 x86_64 User : www-data ( 33) PHP Version : 7.4.18 Disable Function : pcntl_alarm,pcntl_fork,pcntl_waitpid,pcntl_wait,pcntl_wifexited,pcntl_wifstopped,pcntl_wifsignaled,pcntl_wifcontinued,pcntl_wexitstatus,pcntl_wtermsig,pcntl_wstopsig,pcntl_signal,pcntl_signal_get_handler,pcntl_signal_dispatch,pcntl_get_last_error,pcntl_strerror,pcntl_sigprocmask,pcntl_sigwaitinfo,pcntl_sigtimedwait,pcntl_exec,pcntl_getpriority,pcntl_setpriority,pcntl_async_signals,pcntl_unshare, MySQL : OFF | cURL : OFF | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : OFF Directory : /proc/self/root/lib/python3/dist-packages/libcloud/dns/drivers/ |
Upload File : |
# Licensed to the Apache Software Foundation (ASF) under one or more # contributor license agreements. See the NOTICE file distributed with # this work for additional information regarding copyright ownership. # The ASF licenses this file to You under the Apache License, Version 2.0 # (the "License"); you may not use this file except in compliance with # the License.You may obtain a copy of the License at # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. """ RcodeZero DNS Driver """ import json import hashlib import re from libcloud.common.base import ConnectionKey, JsonResponse from libcloud.common.exceptions import BaseHTTPError from libcloud.common.types import InvalidCredsError, MalformedResponseError from libcloud.dns.base import DNSDriver, Zone, Record from libcloud.dns.types import ZoneDoesNotExistError, ZoneAlreadyExistsError from libcloud.dns.types import Provider, RecordType from libcloud.utils.py3 import httplib API_HOST = 'my.rcodezero.at' __all__ = [ 'RcodeZeroDNSDriver', ] class RcodeZeroResponse(JsonResponse): def success(self): i = int(self.status) return 200 <= i <= 299 def parse_error(self): if self.status == httplib.UNAUTHORIZED: raise InvalidCredsError( 'Invalid API key. Check https://my.rcodezero.at/enableapi') errors = [] try: body = self.parse_body() except MalformedResponseError as e: body = '%s: %s' % (e.value, e.body) try: errors = [body['message']] except TypeError: return '%s (HTTP Code: %d)' % (body, self.status) except KeyError: pass return '%s (HTTP Code: %d)' % (' '.join(errors), self.status) class RcodeZeroConnection(ConnectionKey): responseCls = RcodeZeroResponse host = API_HOST def add_default_headers(self, headers): headers['Authorization'] = 'Bearer ' + self.key headers['Accept'] = 'application/json' return headers class RcodeZeroDNSDriver(DNSDriver): type = Provider.RCODEZERO name = 'RcodeZero DNS' website = 'https://www.rcodezero.at/' connectionCls = RcodeZeroConnection RECORD_TYPE_MAP = { RecordType.A: 'A', RecordType.AAAA: 'AAAA', RecordType.AFSDB: 'AFSDB', RecordType.ALIAS: 'ALIAS', RecordType.CERT: 'CERT', RecordType.CNAME: 'CNAME', RecordType.DNAME: 'DNAME', RecordType.DNSKEY: 'DNSKEY', RecordType.DS: 'DS', RecordType.HINFO: 'HINFO', RecordType.KEY: 'KEY', RecordType.LOC: 'LOC', RecordType.MX: 'MX', RecordType.NAPTR: 'NAPTR', RecordType.NS: 'NS', RecordType.NSEC: 'NSEC', RecordType.OPENPGPKEY: 'OPENPGPKEY', RecordType.PTR: 'PTR', RecordType.RP: 'RP', RecordType.RRSIG: 'RRSIG', RecordType.SOA: 'SOA', RecordType.SPF: 'SPF', RecordType.SRV: 'SRV', RecordType.SSHFP: 'SSHFP', RecordType.SRV: 'SRV', RecordType.TLSA: 'TLSA', RecordType.TXT: 'TXT', } def __init__(self, key, secret=None, secure=True, host=None, port=None, api_version='v1', **kwargs): """ :param key: API token to be used (required) :type key: ``str`` :param secret: Password to be used, ignored by RcodeZero :type key: ``str`` :param secure: Whether to use HTTPS (default) or HTTP. :type secure: ``bool`` :param host: Hostname used for connections. :type host: ``str`` :param port: Port used for connections. :type port: ``int`` :param api_version: Specifies the API version to use. ``v1`` is currently the only valid option (and default) :type api_version: ``str`` :return: ``None`` """ if api_version == 'v1': self.api_root = '/api/v1' else: raise NotImplementedError('Unsupported API version: %s' % api_version) super(RcodeZeroDNSDriver, self).__init__(key=key, secure=secure, host=host, port=port, **kwargs) def create_record(self, name, zone, type, data, extra=None): """ Create a new record in a given, existing zone. :param name: name of the new record without the domain name, for example "www". :type name: ``str`` :param zone: Zone in which the requested record is created. :type zone: :class:`Zone` :param type: DNS resource record type (A, AAAA, ...). :type type: :class:`RecordType` :param data: Data for the record (depending on the record type). :type data: ``str`` :param extra: Extra attributes: 'ttl', 'disabled' :type extra: ``dict`` :rtype: :class:`Record` """ action = '%s/zones/%s/rrsets' % (self.api_root, zone.id) payload = self._to_patchrequest( zone.id, None, name, type, data, extra, 'add') try: self.connection.request(action=action, data=json.dumps(payload), method='PATCH') except BaseHTTPError as e: if e.code == httplib.UNPROCESSABLE_ENTITY and \ e.message.startswith('Could not find domain'): raise ZoneDoesNotExistError(zone_id=zone.id, driver=self, value=e.message) raise e if extra is not None and extra.get('ttl', None) is not None: ttl = extra['ttl'] else: ttl = None return Record(id=None, name=name, data=data, type=type, zone=zone, ttl=ttl, driver=self) def create_zone(self, domain, type='master', ttl=None, extra={}): """ Create a new zone. :param name: Zone domain name (e.g. example.com) :type name: ``str`` :param domain: Zone type ('master' / 'slave'). (required). :type domain: :class:`Zone` :param ttl: TTL for new records. (optional). Ignored by RcodeZero. RcodeZero uses record specific TTLs. :type ttl: ``int`` :param extra: Extra attributes: 'masters' (for type=slave): ``extra={'masters': ['193.0.2.2','2001:db8::2']}`` sets the Master nameservers for a type=slave zone. :type extra: ``dict`` :rtype: :class:`Zone` """ action = '%s/zones' % (self.api_root) if type.lower() == 'slave' and (extra is None or extra.get('masters', None) is None): msg = 'Master IPs required for slave zones' raise ValueError(msg) payload = {'domain': domain.lower(), 'type': type.lower()} payload.update(extra) zone_id = domain + '.' try: self.connection.request(action=action, data=json.dumps(payload), method='POST') except BaseHTTPError as e: if e.code == httplib.UNPROCESSABLE_ENTITY and \ e.message.find("Zone '%s' already configured for your account" % domain): raise ZoneAlreadyExistsError(zone_id=zone_id, driver=self, value=e.message) raise e return Zone(id=zone_id, domain=domain, type=type.lower(), ttl=None, driver=self, extra=extra) def update_zone(self, zone, domain, type=None, ttl=None, extra=None): """ Update an existing zone. :param zone: Zone to update. :type zone: :class:`Zone` :param domain: Zone domain name (e.g. example.com) :type domain: ``str`` :param type: Zone type ('master' / 'slave'). :type type: ``str`` :param ttl: not supported. RcodeZero uses RRSet-specific TTLs :type ttl: ``int`` :param extra: Extra attributes: 'masters' (for type=slave) ``extra={'masters': ['193.0.2.2','2001:db8::2']}`` sets the Master nameserver addresses for a type=slave zone :type extra: ``dict`` :rtype: :class:`Zone` """ action = '%s/zones/%s' % (self.api_root, domain) if type is None: type = zone.type if type.lower() == 'slave' and (extra is None or extra.get('masters', None) is None): msg = 'Master IPs required for slave zones' raise ValueError(msg) payload = {'domain': domain.lower(), 'type': type.lower()} if extra is not None: payload.update(extra) try: self.connection.request(action=action, data=json.dumps(payload), method='PUT') except BaseHTTPError as e: if e.code == httplib.UNPROCESSABLE_ENTITY and \ e.message.startswith("Domain '%s' update failed" % domain): raise ZoneAlreadyExistsError(zone_id=zone.id, driver=self, value=e.message) raise e return Zone(id=zone.id, domain=domain, type=type, ttl=None, driver=self, extra=extra) def delete_record(self, record): """ Delete a record in a given zone. :param record: record to delete (record object) :type record: `Record` :rtype: ``bool`` """ action = '%s/zones/%s/rrsets' % (self.api_root, record.zone.id) payload = self._to_patchrequest( record.zone.id, None, record.name, record.type, record.data, record.extra, 'delete') try: self.connection.request(action=action, data=json.dumps(payload), method='PATCH') except BaseHTTPError as e: if e.code == httplib.UNPROCESSABLE_ENTITY and \ e.message.startswith('Could not find domain'): raise ZoneDoesNotExistError(zone_id=record.zone.id, driver=self, value=e.message) raise e return True def delete_zone(self, zone): """ Delete a zone and all its records. :param zone: zone to delete :type zone: `Zone` :rtype: ``bool`` """ action = '%s/zones/%s' % (self.api_root, zone.id) try: self.connection.request(action=action, method='DELETE') except BaseHTTPError: return False return True def get_zone(self, zone_id): """ Get a Zone object. :param zone_id: name of the zone, for example "example.com". :type zone_id: ``str`` :rtype: :class:`Zone` :raises: ZoneDoesNotExistError: if zone could not be found. """ action = '%s/zones/%s' % (self.api_root, zone_id) try: response = self.connection.request(action=action, method='GET') except BaseHTTPError as e: if e.code == httplib.NOT_FOUND: raise ZoneDoesNotExistError(zone_id=zone_id, driver=self, value=e.message) raise e return self._to_zone(response.object) def get_record(self, zone_id, record_id): """ Return a Record instance. :param zone_id: ID of the required zone :type zone_id: ``str`` :param record_id: ID of the required record :type record_id: ``str`` :rtype: :class:`Record` """ records = self.list_records(Zone(id=zone_id, domain=zone_id, type=None, ttl=None, driver=self, extra=None)) foundrecords = list(filter(lambda x: x.id == record_id, records)) if len(foundrecords) > 0: return(foundrecords[0]) else: return(None) def list_records(self, zone): """ Return a list of all record objects for the given zone. :param zone: Zone object to list records for. :type zone: :class:`Zone` :return: ``list`` of :class:`Record` """ action = '%s/zones/%s/rrsets?page_size=-1' % (self.api_root, zone.id) try: response = self.connection.request(action=action, method='GET') except BaseHTTPError as e: if e.code == httplib.UNPROCESSABLE_ENTITY and \ e.message.startswith('Could not find domain'): raise ZoneDoesNotExistError(zone_id=zone.id, driver=self, value=e.message) raise e return self._to_records(response.object['data'], zone) def list_zones(self): """ Return a list of zone objects for this account. :return: ``list`` of :class:`Zone` """ action = '%s/zones?page_size=-1' % (self.api_root) response = self.connection.request(action=action, method='GET') return self._to_zones(response.object['data']) def update_record(self, record, name, type, data, extra=None): """ Update an existing record. :param record: Record object to update. :type record: :class:`Record` :param name: name of the new record, for example "www". :type name: ``str`` :param type: DNS resource record type (A, AAAA, ...). :type type: :class:`RecordType` :param data: Data for the record (depending on the record type). :type data: ``str`` :param extra: Extra attributes: 'ttl','disabled' (optional) :type extra: ``dict`` :rtype: :class:`Record` """ action = '%s/zones/%s/rrsets' % (self.api_root, record.zone.id) payload = self._to_patchrequest( record.zone.id, record, name, type, data, record.extra, 'update') try: self.connection.request(action=action, data=json.dumps(payload), method='PATCH') except BaseHTTPError as e: if e.code == httplib.UNPROCESSABLE_ENTITY and \ e.message.startswith('Could not find domain'): raise ZoneDoesNotExistError(zone_id=record.zone.id, driver=self, value=e.message) raise e if not (extra is None or extra.get('ttl', None) is None): ttl = extra['ttl'] else: ttl = record.ttl return Record(id=hashlib.md5(str(name + ' ' + data).encode('utf-8')).hexdigest(), name=name, data=data, type=type, zone=record.zone, driver=self, ttl=ttl, extra=extra) def _to_zone(self, item): extra = {} for e in ['dnssec_status', 'dnssec_status_detail', 'dnssec_ksk_status', 'dnssec_ksk_status_detail', 'dnssec_ds', 'dnssec_dnskey', 'dnssec_safe_to_unsign', 'dnssec', 'masters', 'serial', 'created', 'last_check']: if e in item: extra[e] = item[e] return Zone(id=item['domain'], domain=item['domain'], type=item['type'].lower(), ttl=None, driver=self, extra=extra) def _to_zones(self, items): zones = [] for item in items: zones.append(self._to_zone(item)) return zones def _to_records(self, items, zone): records = [] for item in items: for record in item['records']: extra = {} extra['disabled'] = record['disabled'] # strip domain and trailing dot from recordname recordname = re.sub('.' + zone.id + '$', '', item['name'][:-1]) records.append( Record(id=hashlib.md5(str(recordname + ' ' + record['content']). encode('utf-8')).hexdigest(), name=recordname, data=record['content'], type=item['type'], zone=zone, driver=self, ttl=item['ttl'], extra=extra)) return records # rcodezero supports only rrset, so we must create rrsets from the given # record def _to_patchrequest(self, zone, record, name, type, data, extra, action): rrset = {} cur_records = self.list_records( Zone(id=zone, domain=None, type=None, ttl=None, driver=self)) if name != '': rrset['name'] = name + '.' + zone + '.' else: rrset['name'] = zone + '.' rrset['type'] = type rrset['changetype'] = action rrset['records'] = [] if not (extra is None or extra.get('ttl', None) is None): rrset['ttl'] = extra['ttl'] content = {} if not action == 'delete': content['content'] = data if not (extra is None or extra.get('disabled', None) is None): content['disabled'] = extra['disabled'] rrset['records'].append(content) id = hashlib.md5(str(name + ' ' + data).encode('utf-8')).hexdigest() # check if rrset contains more than one record. if yes we need to create an # update request for r in cur_records: if action == 'update' and r.id == record.id: # do not include records which should be updated in the update # request continue if name == r.name and r.id != id: # we have other records with the same name so make an update # request rrset['changetype'] = 'update' content = {} content['content'] = r.data if not (r.extra is None or r.extra.get('disabled', None) is None): content['disabled'] = r.extra['disabled'] rrset['records'].append(content) request = list() request.append(rrset) return request