mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-08-04 08:52:16 +02:00
1064 lines
42 KiB
Python
1064 lines
42 KiB
Python
import logging
|
|
|
|
from datetime import date
|
|
from string import digits
|
|
from django_fsm import FSMField, transition # type: ignore
|
|
|
|
from django.db import models
|
|
|
|
from epplibwrapper import (
|
|
CLIENT as registry,
|
|
commands,
|
|
common as epp,
|
|
RegistryError,
|
|
ErrorCode,
|
|
)
|
|
|
|
from .utility.domain_field import DomainField
|
|
from .utility.domain_helper import DomainHelper
|
|
from .utility.time_stamped_model import TimeStampedModel
|
|
|
|
from .public_contact import PublicContact
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Domain(TimeStampedModel, DomainHelper):
|
|
"""
|
|
Manage the lifecycle of domain names.
|
|
|
|
The registry is the source of truth for this data and this model exists:
|
|
1. To tie ownership information in the registrar to
|
|
DNS entries in the registry
|
|
|
|
~~~ HOW TO USE THIS CLASS ~~~
|
|
|
|
A) You can create a Domain object with just a name. `Domain(name="something.gov")`.
|
|
B) Saving the Domain object will not contact the registry, as it may be useful
|
|
to have Domain objects in an `UNKNOWN` pre-created state.
|
|
C) Domain properties are lazily loaded. Accessing `my_domain.expiration_date` will
|
|
contact the registry, if a cached copy does not exist.
|
|
D) Domain creation is lazy. If `my_domain.expiration_date` finds that `my_domain`
|
|
does not exist in the registry, it will ask the registry to create it.
|
|
F) Created is _not_ the same as active aka live on the internet.
|
|
G) Activation is controlled by the registry. It will happen automatically when the
|
|
domain meets the required checks.
|
|
"""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self._cache = {}
|
|
super(Domain, self).__init__(*args, **kwargs)
|
|
|
|
class Status(models.TextChoices):
|
|
"""
|
|
The status codes we can receive from the registry.
|
|
|
|
These are detailed in RFC 5731 in section 2.3.
|
|
https://www.rfc-editor.org/std/std69.txt
|
|
"""
|
|
|
|
# Requests to delete the object MUST be rejected.
|
|
CLIENT_DELETE_PROHIBITED = "clientDeleteProhibited"
|
|
SERVER_DELETE_PROHIBITED = "serverDeleteProhibited"
|
|
|
|
# DNS delegation information MUST NOT be published for the object.
|
|
CLIENT_HOLD = "clientHold"
|
|
SERVER_HOLD = "serverHold"
|
|
|
|
# Requests to renew the object MUST be rejected.
|
|
CLIENT_RENEW_PROHIBITED = "clientRenewProhibited"
|
|
SERVER_RENEW_PROHIBITED = "serverRenewProhibited"
|
|
|
|
# Requests to transfer the object MUST be rejected.
|
|
CLIENT_TRANSFER_PROHIBITED = "clientTransferProhibited"
|
|
SERVER_TRANSFER_PROHIBITED = "serverTransferProhibited"
|
|
|
|
# Requests to update the object (other than to remove this status)
|
|
# MUST be rejected.
|
|
CLIENT_UPDATE_PROHIBITED = "clientUpdateProhibited"
|
|
SERVER_UPDATE_PROHIBITED = "serverUpdateProhibited"
|
|
|
|
# Delegation information has not been associated with the object.
|
|
# This is the default status when a domain object is first created
|
|
# and there are no associated host objects for the DNS delegation.
|
|
# This status can also be set by the server when all host-object
|
|
# associations are removed.
|
|
INACTIVE = "inactive"
|
|
|
|
# This is the normal status value for an object that has no pending
|
|
# operations or prohibitions. This value is set and removed by the
|
|
# server as other status values are added or removed.
|
|
OK = "ok"
|
|
|
|
# A transform command has been processed for the object, but the
|
|
# action has not been completed by the server. Server operators can
|
|
# delay action completion for a variety of reasons, such as to allow
|
|
# for human review or third-party action. A transform command that
|
|
# is processed, but whose requested action is pending, is noted with
|
|
# response code 1001.
|
|
DNS_NEEDED = "pendingCreate"
|
|
PENDING_DELETE = "pendingDelete"
|
|
PENDING_RENEW = "pendingRenew"
|
|
PENDING_TRANSFER = "pendingTransfer"
|
|
PENDING_UPDATE = "pendingUpdate"
|
|
|
|
class State(models.TextChoices):
|
|
"""These capture (some of) the states a domain object can be in."""
|
|
|
|
# the state is indeterminate
|
|
UNKNOWN = "unknown"
|
|
|
|
# The domain object exists in the registry
|
|
# but nameservers don't exist for it yet
|
|
DNS_NEEDED = "dns needed"
|
|
|
|
# Domain has had nameservers set, may or may not be active
|
|
READY = "ready"
|
|
|
|
# Registrar manually changed state to client hold
|
|
ON_HOLD = "client hold"
|
|
|
|
# previously existed but has been deleted from the registry
|
|
DELETED = "deleted"
|
|
|
|
class Cache(property):
|
|
"""
|
|
Python descriptor to turn class methods into properties.
|
|
|
|
The purpose of subclassing `property` rather than using it directly
|
|
as a decorator (`@Cache`) is to insert generic code to run
|
|
before or after _all_ properties are accessed, modified, or deleted.
|
|
|
|
As an example:
|
|
|
|
domain = Domain(name="example.gov")
|
|
domain.save()
|
|
<--- insert code here
|
|
date = domain.creation_date
|
|
<--- or here
|
|
(...other stuff...)
|
|
"""
|
|
|
|
def __get__(self, obj, objtype=None):
|
|
"""Called during get. Example: `r = domain.registrant`."""
|
|
return super().__get__(obj, objtype)
|
|
|
|
def __set__(self, obj, value):
|
|
"""Called during set. Example: `domain.registrant = 'abc123'`."""
|
|
super().__set__(obj, value)
|
|
# always invalidate cache after sending updates to the registry
|
|
obj._invalidate_cache()
|
|
|
|
def __delete__(self, obj):
|
|
"""Called during delete. Example: `del domain.registrant`."""
|
|
super().__delete__(obj)
|
|
|
|
@classmethod
|
|
def available(cls, domain: str) -> bool:
|
|
"""Check if a domain is available."""
|
|
if not cls.string_could_be_domain(domain):
|
|
raise ValueError("Not a valid domain: %s" % str(domain))
|
|
req = commands.CheckDomain([domain])
|
|
return registry.send(req, cleaned=True).res_data[0].avail
|
|
|
|
@classmethod
|
|
def registered(cls, domain: str) -> bool:
|
|
"""Check if a domain is _not_ available."""
|
|
return not cls.available(domain)
|
|
|
|
@Cache
|
|
def contacts(self) -> dict[str, str]:
|
|
"""
|
|
Get a dictionary of registry IDs for the contacts for this domain.
|
|
|
|
IDs are provided as strings, e.g.
|
|
|
|
{ PublicContact.ContactTypeChoices.REGISTRANT: "jd1234",
|
|
PublicContact.ContactTypeChoices.ADMINISTRATIVE: "sh8013",...}
|
|
"""
|
|
raise NotImplementedError()
|
|
|
|
@Cache
|
|
def creation_date(self) -> date:
|
|
"""Get the `cr_date` element from the registry."""
|
|
return self._get_property("cr_date")
|
|
|
|
@Cache
|
|
def last_transferred_date(self) -> date:
|
|
"""Get the `tr_date` element from the registry."""
|
|
raise NotImplementedError()
|
|
|
|
@Cache
|
|
def last_updated_date(self) -> date:
|
|
"""Get the `up_date` element from the registry."""
|
|
return self._get_property("up_date")
|
|
|
|
@Cache
|
|
def expiration_date(self) -> date:
|
|
"""Get or set the `ex_date` element from the registry."""
|
|
return self._get_property("ex_date")
|
|
|
|
@expiration_date.setter # type: ignore
|
|
def expiration_date(self, ex_date: date):
|
|
raise NotImplementedError()
|
|
|
|
@Cache
|
|
def password(self) -> str:
|
|
"""
|
|
Get the `auth_info.pw` element from the registry. Not a real password.
|
|
|
|
This `auth_info` element is required by the EPP protocol, but the registry is
|
|
using a different mechanism to ensure unauthorized clients cannot perform
|
|
actions on domains they do not own. This field provides no security features.
|
|
It is not a secret.
|
|
"""
|
|
raise NotImplementedError()
|
|
|
|
@Cache
|
|
def nameservers(self) -> list[tuple[str]]:
|
|
"""
|
|
Get or set a complete list of nameservers for this domain.
|
|
|
|
Hosts are provided as a list of tuples, e.g.
|
|
|
|
[("ns1.example.com",), ("ns1.example.gov", "0.0.0.0")]
|
|
|
|
Subordinate hosts (something.your-domain.gov) MUST have IP addresses,
|
|
while non-subordinate hosts MUST NOT.
|
|
"""
|
|
try:
|
|
hosts = self._get_property("hosts")
|
|
except Exception as err:
|
|
# Don't throw error as this is normal for a new domain
|
|
logger.info("Domain is missing nameservers %s" % err)
|
|
return []
|
|
|
|
hostList = []
|
|
for host in hosts:
|
|
# TODO - this should actually have a second tuple value with the ip address
|
|
# ignored because uncertain if we will even have a way to display mult.
|
|
# and adresses can be a list of mult address
|
|
hostList.append((host["name"],))
|
|
|
|
return hostList
|
|
|
|
def _check_host(self, hostnames: list[str]):
|
|
"""check if host is available, True if available
|
|
returns boolean"""
|
|
checkCommand = commands.CheckHost(hostnames)
|
|
try:
|
|
response = registry.send(checkCommand, cleaned=True)
|
|
return response.res_data[0].avail
|
|
except RegistryError as err:
|
|
logger.warning(
|
|
"Couldn't check hosts %s. Errorcode was %s, error was %s",
|
|
hostnames,
|
|
err.code,
|
|
err,
|
|
)
|
|
return False
|
|
|
|
def _create_host(self, host, addrs):
|
|
"""Call _check_host first before using this function,
|
|
This creates the host object in the registry
|
|
doesn't add the created host to the domain
|
|
returns ErrorCode (int)"""
|
|
logger.info("Creating host")
|
|
if addrs is not None:
|
|
addresses = [epp.Ip(addr=addr) for addr in addrs]
|
|
request = commands.CreateHost(name=host, addrs=addresses)
|
|
else:
|
|
request = commands.CreateHost(name=host)
|
|
|
|
try:
|
|
logger.info("_create_host()-> sending req as %s" % request)
|
|
response = registry.send(request, cleaned=True)
|
|
return response.code
|
|
except RegistryError as e:
|
|
logger.error("Error _create_host, code was %s error was %s" % (e.code, e))
|
|
return e.code
|
|
|
|
@nameservers.setter # type: ignore
|
|
def nameservers(self, hosts: list[tuple[str]]):
|
|
"""host should be a tuple of type str, str,... where the elements are
|
|
Fully qualified host name, addresses associated with the host
|
|
example: [(ns1.okay.gov, 127.0.0.1, others ips)]"""
|
|
# TODO: ticket #848 finish this implementation
|
|
# must delete nameservers as well or update
|
|
# ip version checking may need to be added in a different ticket
|
|
|
|
if len(hosts) > 13:
|
|
raise ValueError(
|
|
"Too many hosts provided, you may not have more than 13 nameservers."
|
|
)
|
|
logger.info("Setting nameservers")
|
|
logger.info(hosts)
|
|
for hostTuple in hosts:
|
|
host = hostTuple[0]
|
|
addrs = None
|
|
if len(hostTuple) > 1:
|
|
addrs = hostTuple[1:]
|
|
avail = self._check_host([host])
|
|
if avail:
|
|
createdCode = self._create_host(host=host, addrs=addrs)
|
|
|
|
# update the domain obj
|
|
if createdCode == ErrorCode.COMMAND_COMPLETED_SUCCESSFULLY:
|
|
# add host to domain
|
|
request = commands.UpdateDomain(
|
|
name=self.name, add=[epp.HostObjSet([host])]
|
|
)
|
|
|
|
try:
|
|
registry.send(request, cleaned=True)
|
|
except RegistryError as e:
|
|
logger.error(
|
|
"Error adding nameserver, code was %s error was %s"
|
|
% (e.code, e)
|
|
)
|
|
|
|
try:
|
|
self.ready()
|
|
self.save()
|
|
except Exception as err:
|
|
logger.info(
|
|
"nameserver setter checked for create state "
|
|
"and it did not succeed. Error: %s" % err
|
|
)
|
|
# TODO - handle removed nameservers here will need to change the state
|
|
# then go back to DNS_NEEDED
|
|
|
|
@Cache
|
|
def statuses(self) -> list[str]:
|
|
"""
|
|
Get or set the domain `status` elements from the registry.
|
|
|
|
A domain's status indicates various properties. See Domain.Status.
|
|
"""
|
|
# implementation note: the Status object from EPP stores the string in
|
|
# a dataclass property `state`, not to be confused with the `state` field here
|
|
if "statuses" not in self._cache:
|
|
self._fetch_cache()
|
|
if "statuses" not in self._cache:
|
|
raise Exception("Can't retreive status from domain info")
|
|
else:
|
|
return self._cache["statuses"]
|
|
|
|
@statuses.setter # type: ignore
|
|
def statuses(self, statuses: list[str]):
|
|
# TODO: there are a long list of rules in the RFC about which statuses
|
|
# can be combined; check that here and raise errors for invalid combinations -
|
|
# some statuses cannot be set by the client at all
|
|
raise NotImplementedError()
|
|
|
|
@Cache
|
|
def registrant_contact(self) -> PublicContact:
|
|
"""Get or set the registrant for this domain."""
|
|
raise NotImplementedError()
|
|
|
|
@registrant_contact.setter # type: ignore
|
|
def registrant_contact(self, contact: PublicContact):
|
|
"""Registrant is set when a domain is created,
|
|
so follow on additions will update the current registrant"""
|
|
|
|
logger.info("making registrant contact")
|
|
self._set_singleton_contact(
|
|
contact=contact, expectedType=contact.ContactTypeChoices.REGISTRANT
|
|
)
|
|
|
|
@Cache
|
|
def administrative_contact(self) -> PublicContact:
|
|
"""Get or set the admin contact for this domain."""
|
|
raise NotImplementedError()
|
|
|
|
@administrative_contact.setter # type: ignore
|
|
def administrative_contact(self, contact: PublicContact):
|
|
logger.info("making admin contact")
|
|
if contact.contact_type != contact.ContactTypeChoices.ADMINISTRATIVE:
|
|
raise ValueError(
|
|
"Cannot set a registrant contact with a different contact type"
|
|
)
|
|
self._make_contact_in_registry(contact=contact)
|
|
self._update_domain_with_contact(contact, rem=False)
|
|
|
|
def get_default_security_contact(self):
|
|
logger.info("getting default sec contact")
|
|
contact = PublicContact.get_default_security()
|
|
contact.domain = self
|
|
return contact
|
|
|
|
def _update_epp_contact(self, contact: PublicContact):
|
|
"""Sends UpdateContact to update the actual contact object,
|
|
domain object remains unaffected
|
|
should be used when changing email address
|
|
or other contact info on an existing domain
|
|
"""
|
|
updateContact = commands.UpdateContact(
|
|
id=contact.registry_id,
|
|
# type: ignore
|
|
postal_info=self._make_epp_contact_postal_info(contact=contact),
|
|
email=contact.email,
|
|
voice=contact.voice,
|
|
fax=contact.fax,
|
|
) # type: ignore
|
|
|
|
try:
|
|
registry.send(updateContact, cleaned=True)
|
|
except RegistryError as e:
|
|
logger.error(
|
|
"Error updating contact, code was %s error was %s" % (e.code, e)
|
|
)
|
|
# TODO - ticket 433 human readable error handling here
|
|
|
|
def _update_domain_with_contact(self, contact: PublicContact, rem=False):
|
|
"""adds or removes a contact from a domain"""
|
|
logger.info(
|
|
"_update_domain_with_contact() received type %s " % contact.contact_type
|
|
)
|
|
domainContact = epp.DomainContact(
|
|
contact=contact.registry_id, type=contact.contact_type
|
|
)
|
|
|
|
updateDomain = commands.UpdateDomain(name=self.name, add=[domainContact])
|
|
if rem:
|
|
updateDomain = commands.UpdateDomain(name=self.name, rem=[domainContact])
|
|
|
|
try:
|
|
registry.send(updateDomain, cleaned=True)
|
|
except RegistryError as e:
|
|
logger.error(
|
|
"Error changing contact on a domain. Error code is %s error was %s"
|
|
% (e.code, e)
|
|
)
|
|
action = "add"
|
|
if rem:
|
|
action = "remove"
|
|
|
|
raise Exception(
|
|
"Can't %s the contact of type %s" % (action, contact.contact_type)
|
|
)
|
|
|
|
@Cache
|
|
def security_contact(self) -> PublicContact:
|
|
"""Get or set the security contact for this domain."""
|
|
|
|
# get the contacts: call _get_property(contacts=True)
|
|
# if contacts exist and security contact is in the contact list
|
|
# return that contact
|
|
# else call the setter
|
|
# send the public default contact
|
|
try:
|
|
contacts = self._get_property("contacts")
|
|
for contact in contacts:
|
|
if contact.type == PublicContact.ContactTypeChoices.SECURITY:
|
|
return contact
|
|
|
|
except Exception as err: # use better error handling
|
|
logger.info("Couldn't get contact %s" % err)
|
|
|
|
# TODO - remove this! ideally it should return None,
|
|
# but error handling needs to be
|
|
# added on the security email page so that it can handle it being none
|
|
return self.get_default_security_contact()
|
|
|
|
def _add_registrant_to_existing_domain(self, contact: PublicContact):
|
|
self._update_epp_contact(contact=contact)
|
|
|
|
updateDomain = commands.UpdateDomain(
|
|
name=self.name, registrant=contact.registry_id
|
|
)
|
|
try:
|
|
registry.send(updateDomain, cleaned=True)
|
|
except RegistryError as e:
|
|
logger.error(
|
|
"Error changing to new registrant error code is %s, error is %s"
|
|
% (e.code, e)
|
|
)
|
|
# TODO-error handling better here?
|
|
|
|
def _set_singleton_contact(self, contact: PublicContact, expectedType: str): # noqa
|
|
"""Sets the contacts by adding them to the registry as new contacts,
|
|
updates the contact if it is already in epp,
|
|
deletes any additional contacts of the matching type for this domain
|
|
does not create the PublicContact object, this should be made beforehand
|
|
(call save() on a public contact to trigger the contact setters
|
|
which inturn call this function)
|
|
Raises ValueError if expected type doesn't match the contact type"""
|
|
if expectedType != contact.contact_type:
|
|
raise ValueError(
|
|
"Cannot set a contact with a different contact type,"
|
|
" expected type was %s" % expectedType
|
|
)
|
|
|
|
isRegistrant = contact.contact_type == contact.ContactTypeChoices.REGISTRANT
|
|
isEmptySecurity = (
|
|
contact.contact_type == contact.ContactTypeChoices.SECURITY
|
|
and contact.email == ""
|
|
)
|
|
|
|
# get publicContact objects that have the matching
|
|
# domain and type but a different id
|
|
# like in highlander we there can only be one
|
|
hasOtherContact = (
|
|
PublicContact.objects.exclude(registry_id=contact.registry_id)
|
|
.filter(domain=self, contact_type=contact.contact_type)
|
|
.exists()
|
|
)
|
|
|
|
# if no record exists with this contact type
|
|
# make contact in registry, duplicate and errors handled there
|
|
errorCode = self._make_contact_in_registry(contact)
|
|
|
|
# contact is already added to the domain, but something may have changed on it
|
|
alreadyExistsInRegistry = errorCode == ErrorCode.OBJECT_EXISTS
|
|
# if an error occured besides duplication, stop
|
|
if (
|
|
not alreadyExistsInRegistry
|
|
and errorCode != ErrorCode.COMMAND_COMPLETED_SUCCESSFULLY
|
|
):
|
|
# TODO- ticket #433 look here for error handling
|
|
raise Exception("Unable to add contact to registry")
|
|
|
|
# contact doesn't exist on the domain yet
|
|
logger.info("_set_singleton_contact()-> contact has been added to the registry")
|
|
|
|
# if has conflicting contacts in our db remove them
|
|
if hasOtherContact:
|
|
logger.info(
|
|
"_set_singleton_contact()-> updating domain, removing old contact"
|
|
)
|
|
if isEmptySecurity:
|
|
existing_contact = PublicContact.objects.filter(
|
|
domain=self, contact_type=contact.contact_type
|
|
).get()
|
|
else:
|
|
existing_contact = (
|
|
PublicContact.objects.exclude(registry_id=contact.registry_id)
|
|
.filter(domain=self, contact_type=contact.contact_type)
|
|
.get()
|
|
)
|
|
if isRegistrant:
|
|
# send update domain only for registant contacts
|
|
existing_contact.delete()
|
|
self._add_registrant_to_existing_domain(contact)
|
|
else:
|
|
# remove the old contact and add a new one
|
|
try:
|
|
self._update_domain_with_contact(contact=existing_contact, rem=True)
|
|
existing_contact.delete()
|
|
if isEmptySecurity:
|
|
# security is empty so set the default security object again
|
|
self.get_default_security_contact().save()
|
|
except Exception as err:
|
|
logger.error(
|
|
"Raising error after removing and adding a new contact"
|
|
)
|
|
raise (err)
|
|
|
|
# TODO- This could switch to just creating a
|
|
# list of ones to remove and a list of ones to add
|
|
# or Change it to add contacts before deleting the old ones
|
|
|
|
# update domain with contact or update the contact itself
|
|
if not isEmptySecurity:
|
|
if not alreadyExistsInRegistry and not isRegistrant:
|
|
self._update_domain_with_contact(contact=contact, rem=False)
|
|
# if already exists just update
|
|
elif alreadyExistsInRegistry:
|
|
current_contact = PublicContact.objects.filter(
|
|
registry_id=contact.registry_id
|
|
).get()
|
|
|
|
if current_contact.email != contact.email:
|
|
self._update_epp_contact(contact=contact)
|
|
else:
|
|
logger.info("removing security contact and setting default again")
|
|
|
|
# get the current contact registry id for security
|
|
current_contact = PublicContact.objects.filter(
|
|
registry_id=contact.registry_id
|
|
).get()
|
|
|
|
# don't let user delete the default without adding a new email
|
|
if current_contact.email != PublicContact.get_default_security().email:
|
|
# remove the contact
|
|
self._update_domain_with_contact(contact=current_contact, rem=True)
|
|
current_contact.delete()
|
|
# add new contact
|
|
security_contact = self.get_default_security_contact()
|
|
security_contact.save()
|
|
|
|
@security_contact.setter # type: ignore
|
|
def security_contact(self, contact: PublicContact):
|
|
"""makes the contact in the registry,
|
|
for security the public contact should have the org or registrant information
|
|
from domain information (not domain application)
|
|
and should have the security email from DomainApplication"""
|
|
logger.info("making security contact in registry")
|
|
self._set_singleton_contact(
|
|
contact, expectedType=contact.ContactTypeChoices.SECURITY
|
|
)
|
|
|
|
@Cache
|
|
def technical_contact(self) -> PublicContact:
|
|
"""Get or set the tech contact for this domain."""
|
|
raise NotImplementedError()
|
|
|
|
@technical_contact.setter # type: ignore
|
|
def technical_contact(self, contact: PublicContact):
|
|
logger.info("making technical contact")
|
|
self._set_singleton_contact(
|
|
contact, expectedType=contact.ContactTypeChoices.TECHNICAL
|
|
)
|
|
|
|
def is_active(self) -> bool:
|
|
"""Currently just returns if the state is created,
|
|
because then it should be live, theoretically.
|
|
Post mvp this should indicate
|
|
Is the domain live on the inter webs?
|
|
could be replaced with request to see if ok status is set
|
|
"""
|
|
return self.state == self.State.READY
|
|
|
|
def transfer(self):
|
|
"""Going somewhere. Not implemented."""
|
|
raise NotImplementedError()
|
|
|
|
def renew(self):
|
|
"""Time to renew. Not implemented."""
|
|
raise NotImplementedError()
|
|
|
|
def get_security_email(self):
|
|
logger.info("get_security_email-> getting the contact ")
|
|
secContact = self.security_contact
|
|
return secContact.email
|
|
|
|
def clientHoldStatus(self):
|
|
return epp.Status(state=self.Status.CLIENT_HOLD, description="", lang="en")
|
|
|
|
def _place_client_hold(self):
|
|
"""This domain should not be active.
|
|
may raises RegistryError, should be caught or handled correctly by caller"""
|
|
request = commands.UpdateDomain(name=self.name, add=[self.clientHoldStatus()])
|
|
registry.send(request)
|
|
|
|
def _remove_client_hold(self):
|
|
"""This domain is okay to be active.
|
|
may raises RegistryError, should be caught or handled correctly by caller"""
|
|
request = commands.UpdateDomain(name=self.name, rem=[self.clientHoldStatus()])
|
|
registry.send(request)
|
|
|
|
def _delete_domain(self):
|
|
"""This domain should be deleted from the registry
|
|
may raises RegistryError, should be caught or handled correctly by caller"""
|
|
request = commands.DeleteDomain(name=self.name)
|
|
registry.send(request)
|
|
|
|
def __str__(self) -> str:
|
|
return self.name
|
|
|
|
name = DomainField(
|
|
max_length=253,
|
|
blank=False,
|
|
default=None, # prevent saving without a value
|
|
unique=True,
|
|
help_text="Fully qualified domain name",
|
|
)
|
|
|
|
state = FSMField(
|
|
max_length=21,
|
|
choices=State.choices,
|
|
default=State.UNKNOWN,
|
|
protected=True, # cannot change state directly, particularly in Django admin
|
|
help_text="Very basic info about the lifecycle of this domain object",
|
|
)
|
|
|
|
def isActive(self):
|
|
return self.state == Domain.State.CREATED
|
|
|
|
# ForeignKey on UserDomainRole creates a "permissions" member for
|
|
# all of the user-roles that are in place for this domain
|
|
|
|
# ManyToManyField on User creates a "users" member for all of the
|
|
# users who have some role on this domain
|
|
|
|
# ForeignKey on DomainInvitation creates an "invitations" member for
|
|
# all of the invitations that have been sent for this domain
|
|
|
|
def _validate_host_tuples(self, hosts: list[tuple[str]]):
|
|
"""
|
|
Helper function. Validate hostnames and IP addresses.
|
|
|
|
Raises:
|
|
ValueError if hostname or IP address appears invalid or mismatched.
|
|
"""
|
|
for host in hosts:
|
|
hostname = host[0].lower()
|
|
addresses: tuple[str] = host[1:] # type: ignore
|
|
if not bool(Domain.HOST_REGEX.match(hostname)):
|
|
raise ValueError("Invalid hostname: %s." % hostname)
|
|
if len(hostname) > Domain.MAX_LENGTH:
|
|
raise ValueError("Too long hostname: %s" % hostname)
|
|
|
|
is_subordinate = hostname.split(".", 1)[-1] == self.name
|
|
if is_subordinate and len(addresses) == 0:
|
|
raise ValueError(
|
|
"Must supply IP addresses for subordinate host %s" % hostname
|
|
)
|
|
if not is_subordinate and len(addresses) > 0:
|
|
raise ValueError("Must not supply IP addresses for %s" % hostname)
|
|
|
|
for address in addresses:
|
|
allow = set(":." + digits)
|
|
if any(c not in allow for c in address):
|
|
raise ValueError("Invalid IP address: %s." % address)
|
|
|
|
def _get_or_create_domain(self):
|
|
"""Try to fetch info about this domain. Create it if it does not exist."""
|
|
already_tried_to_create = False
|
|
exitEarly = False
|
|
count = 0
|
|
while not exitEarly and count < 3:
|
|
try:
|
|
logger.info("Getting domain info from epp")
|
|
req = commands.InfoDomain(name=self.name)
|
|
domainInfo = registry.send(req, cleaned=True).res_data[0]
|
|
exitEarly = True
|
|
return domainInfo
|
|
except RegistryError as e:
|
|
count += 1
|
|
|
|
if already_tried_to_create:
|
|
logger.error("Already tried to create")
|
|
logger.error(e)
|
|
logger.error(e.code)
|
|
raise e
|
|
if e.code == ErrorCode.OBJECT_DOES_NOT_EXIST:
|
|
# avoid infinite loop
|
|
already_tried_to_create = True
|
|
self.pendingCreate()
|
|
self.save()
|
|
else:
|
|
logger.error(e)
|
|
logger.error(e.code)
|
|
raise e
|
|
|
|
def addRegistrant(self):
|
|
registrant = PublicContact.get_default_registrant()
|
|
registrant.domain = self
|
|
registrant.save() # calls the registrant_contact.setter
|
|
return registrant.registry_id
|
|
|
|
@transition(field="state", source=State.UNKNOWN, target=State.DNS_NEEDED)
|
|
def pendingCreate(self):
|
|
logger.info("Changing to dns_needed")
|
|
|
|
registrantID = self.addRegistrant()
|
|
|
|
# TODO-notes no chg item for registrant in the epplib should
|
|
|
|
req = commands.CreateDomain(
|
|
name=self.name,
|
|
registrant=registrantID,
|
|
auth_info=epp.DomainAuthInfo(pw="2fooBAR123fooBaz"), # not a password
|
|
)
|
|
|
|
try:
|
|
registry.send(req, cleaned=True)
|
|
|
|
except RegistryError as err:
|
|
if err.code != ErrorCode.OBJECT_EXISTS:
|
|
raise err
|
|
|
|
self.addAllDefaults()
|
|
|
|
def addAllDefaults(self):
|
|
security_contact = self.get_default_security_contact()
|
|
security_contact.save()
|
|
|
|
technical_contact = PublicContact.get_default_technical()
|
|
technical_contact.domain = self
|
|
technical_contact.save()
|
|
|
|
administrative_contact = PublicContact.get_default_administrative()
|
|
administrative_contact.domain = self
|
|
administrative_contact.save()
|
|
|
|
@transition(field="state", source=State.DNS_NEEDED, target=State.ON_HOLD)
|
|
def clientHold(self):
|
|
"""place a clienthold on a domain (no longer should resolve)"""
|
|
# TODO - ensure all requirements for client hold are made here
|
|
# (check prohibited statuses)
|
|
logger.info("clientHold()-> inside clientHold")
|
|
self._place_client_hold()
|
|
# TODO -on the client hold ticket any additional error handling here
|
|
|
|
@transition(field="state", source=State.ON_HOLD, target=State.DNS_NEEDED)
|
|
def revertClientHold(self):
|
|
"""undo a clienthold placed on a domain"""
|
|
|
|
logger.info("clientHold()-> inside clientHold")
|
|
self._remove_client_hold()
|
|
# TODO -on the client hold ticket any additional error handling here
|
|
|
|
@transition(field="state", source=State.ON_HOLD, target=State.DELETED)
|
|
def deleted(self):
|
|
"""domain is deleted in epp but is saved in our database"""
|
|
# TODO Domains may not be deleted if:
|
|
# a child host is being used by
|
|
# another .gov domains. The host must be first removed
|
|
# and/or renamed before the parent domain may be deleted.
|
|
logger.info("pendingCreate()-> inside pending create")
|
|
self._delete_domain()
|
|
# TODO - delete ticket any additional error handling here
|
|
|
|
@transition(
|
|
field="state",
|
|
source=[State.DNS_NEEDED],
|
|
target=State.READY,
|
|
)
|
|
def ready(self):
|
|
"""Transition to the ready state
|
|
domain should have nameservers and all contacts
|
|
and now should be considered live on a domain
|
|
"""
|
|
# TODO - in nameservers tickets 848 and 562
|
|
# check here if updates need to be made
|
|
nameserverList = self.nameservers
|
|
logger.info("Changing to ready state")
|
|
if len(nameserverList) < 2 or len(nameserverList) > 13:
|
|
raise ValueError("Not ready to become created, cannot transition yet")
|
|
logger.info("able to transition to ready state")
|
|
|
|
def _disclose_fields(self, contact: PublicContact):
|
|
"""creates a disclose object that can be added to a contact Create using
|
|
.disclose= <this function> on the command before sending.
|
|
if item is security email then make sure email is visable"""
|
|
isSecurity = contact.contact_type == contact.ContactTypeChoices.SECURITY
|
|
DF = epp.DiscloseField
|
|
fields = {DF.FAX, DF.VOICE, DF.ADDR}
|
|
|
|
if not isSecurity or (
|
|
isSecurity and contact.email == PublicContact.get_default_security().email
|
|
):
|
|
fields.add(DF.EMAIL)
|
|
return epp.Disclose(
|
|
flag=False,
|
|
fields=fields,
|
|
types={DF.ADDR: "loc"},
|
|
)
|
|
|
|
def _make_epp_contact_postal_info(self, contact: PublicContact): # type: ignore
|
|
return epp.PostalInfo( # type: ignore
|
|
name=contact.name,
|
|
addr=epp.ContactAddr(
|
|
street=[
|
|
getattr(contact, street)
|
|
for street in ["street1", "street2", "street3"]
|
|
if hasattr(contact, street)
|
|
], # type: ignore
|
|
city=contact.city,
|
|
pc=contact.pc,
|
|
cc=contact.cc,
|
|
sp=contact.sp,
|
|
),
|
|
org=contact.org,
|
|
type="loc",
|
|
)
|
|
|
|
def _make_contact_in_registry(self, contact: PublicContact):
|
|
"""Create the contact in the registry, ignore duplicate contact errors
|
|
returns int corresponding to ErrorCode values"""
|
|
|
|
create = commands.CreateContact(
|
|
id=contact.registry_id,
|
|
postal_info=self._make_epp_contact_postal_info(contact=contact),
|
|
email=contact.email,
|
|
voice=contact.voice,
|
|
fax=contact.fax,
|
|
auth_info=epp.ContactAuthInfo(pw="2fooBAR123fooBaz"),
|
|
) # type: ignore
|
|
# security contacts should only show email addresses, for now
|
|
create.disclose = self._disclose_fields(contact=contact)
|
|
try:
|
|
registry.send(create, cleaned=True)
|
|
return ErrorCode.COMMAND_COMPLETED_SUCCESSFULLY
|
|
except RegistryError as err:
|
|
# don't throw an error if it is just saying this is a duplicate contact
|
|
if err.code != ErrorCode.OBJECT_EXISTS:
|
|
logger.error(
|
|
"Registry threw error for contact id %s"
|
|
" contact type is %s,"
|
|
" error code is\n %s"
|
|
" full error is %s",
|
|
contact.registry_id,
|
|
contact.contact_type,
|
|
err.code,
|
|
err,
|
|
)
|
|
# TODO - Error handling here
|
|
|
|
else:
|
|
logger.warning(
|
|
"Registrar tried to create duplicate contact for id %s",
|
|
contact.registry_id,
|
|
)
|
|
return err.code
|
|
|
|
def _request_contact_info(self, contact: PublicContact):
|
|
req = commands.InfoContact(id=contact.registry_id)
|
|
return registry.send(req, cleaned=True).res_data[0]
|
|
|
|
def _get_or_create_contact(self, contact: PublicContact):
|
|
"""Try to fetch info about a contact. Create it if it does not exist."""
|
|
|
|
try:
|
|
return self._request_contact_info(contact)
|
|
|
|
except RegistryError as e:
|
|
if e.code == ErrorCode.OBJECT_DOES_NOT_EXIST:
|
|
logger.info(
|
|
"_get_or_create_contact()-> contact doesn't exist so making it"
|
|
)
|
|
contact.domain = self
|
|
contact.save() # this will call the function based on type of contact
|
|
return self._request_contact_info(contact=contact)
|
|
else:
|
|
logger.error(
|
|
"Registry threw error for contact id %s"
|
|
" contact type is %s,"
|
|
" error code is\n %s"
|
|
" full error is %s",
|
|
contact.registry_id,
|
|
contact.contact_type,
|
|
e.code,
|
|
e,
|
|
)
|
|
|
|
raise e
|
|
|
|
def _update_or_create_host(self, host):
|
|
raise NotImplementedError()
|
|
|
|
def _delete_host(self, host):
|
|
raise NotImplementedError()
|
|
|
|
def _fetch_cache(self, fetch_hosts=False, fetch_contacts=False):
|
|
"""Contact registry for info about a domain."""
|
|
try:
|
|
# get info from registry
|
|
data = self._get_or_create_domain()
|
|
# extract properties from response
|
|
# (Ellipsis is used to mean "null")
|
|
cache = {
|
|
"auth_info": getattr(data, "auth_info", ...),
|
|
"_contacts": getattr(data, "contacts", ...),
|
|
"cr_date": getattr(data, "cr_date", ...),
|
|
"ex_date": getattr(data, "ex_date", ...),
|
|
"_hosts": getattr(data, "hosts", ...),
|
|
"name": getattr(data, "name", ...),
|
|
"registrant": getattr(data, "registrant", ...),
|
|
"statuses": getattr(data, "statuses", ...),
|
|
"tr_date": getattr(data, "tr_date", ...),
|
|
"up_date": getattr(data, "up_date", ...),
|
|
}
|
|
|
|
# remove null properties (to distinguish between "a value of None" and null)
|
|
cleaned = {k: v for k, v in cache.items() if v is not ...}
|
|
|
|
# statuses can just be a list no need to keep the epp object
|
|
if "statuses" in cleaned.keys():
|
|
cleaned["statuses"] = [status.state for status in cleaned["statuses"]]
|
|
# get contact info, if there are any
|
|
if (
|
|
# fetch_contacts and
|
|
"_contacts" in cleaned
|
|
and isinstance(cleaned["_contacts"], list)
|
|
and len(cleaned["_contacts"])
|
|
):
|
|
cleaned["contacts"] = []
|
|
for domainContact in cleaned["_contacts"]:
|
|
# we do not use _get_or_create_* because we expect the object we
|
|
# just asked the registry for still exists --
|
|
# if not, that's a problem
|
|
|
|
# TODO- discuss-should we check if contact is in public contacts
|
|
# and add it if not- this is really to keep in mine the transisiton
|
|
req = commands.InfoContact(id=domainContact.contact)
|
|
data = registry.send(req, cleaned=True).res_data[0]
|
|
|
|
# extract properties from response
|
|
# (Ellipsis is used to mean "null")
|
|
contact = {
|
|
"id": id,
|
|
"type": domainContact.type,
|
|
"auth_info": getattr(data, "auth_info", ...),
|
|
"cr_date": getattr(data, "cr_date", ...),
|
|
"disclose": getattr(data, "disclose", ...),
|
|
"email": getattr(data, "email", ...),
|
|
"fax": getattr(data, "fax", ...),
|
|
"postal_info": getattr(data, "postal_info", ...),
|
|
"statuses": getattr(data, "statuses", ...),
|
|
"tr_date": getattr(data, "tr_date", ...),
|
|
"up_date": getattr(data, "up_date", ...),
|
|
"voice": getattr(data, "voice", ...),
|
|
}
|
|
|
|
cleaned["contacts"].append(
|
|
{k: v for k, v in contact.items() if v is not ...}
|
|
)
|
|
|
|
# get nameserver info, if there are any
|
|
if (
|
|
# fetch_hosts and
|
|
"_hosts" in cleaned
|
|
and isinstance(cleaned["_hosts"], list)
|
|
and len(cleaned["_hosts"])
|
|
):
|
|
# TODO- add elif in cache set it to be the old cache value
|
|
# no point in removing
|
|
cleaned["hosts"] = []
|
|
for name in cleaned["_hosts"]:
|
|
# we do not use _get_or_create_* because we expect the object we
|
|
# just asked the registry for still exists --
|
|
# if not, that's a problem
|
|
req = commands.InfoHost(name=name)
|
|
data = registry.send(req, cleaned=True).res_data[0]
|
|
# extract properties from response
|
|
# (Ellipsis is used to mean "null")
|
|
host = {
|
|
"name": name,
|
|
"addrs": getattr(data, "addrs", ...),
|
|
"cr_date": getattr(data, "cr_date", ...),
|
|
"statuses": getattr(data, "statuses", ...),
|
|
"tr_date": getattr(data, "tr_date", ...),
|
|
"up_date": getattr(data, "up_date", ...),
|
|
}
|
|
cleaned["hosts"].append(
|
|
{k: v for k, v in host.items() if v is not ...}
|
|
)
|
|
|
|
# replace the prior cache with new data
|
|
self._cache = cleaned
|
|
|
|
except RegistryError as e:
|
|
logger.error(e)
|
|
|
|
def _invalidate_cache(self):
|
|
"""Remove cache data when updates are made."""
|
|
self._cache = {}
|
|
|
|
def _get_property(self, property):
|
|
"""Get some piece of info about a domain."""
|
|
if property not in self._cache:
|
|
self._fetch_cache(
|
|
fetch_hosts=(property == "hosts"),
|
|
fetch_contacts=(property == "contacts"),
|
|
)
|
|
|
|
if property in self._cache:
|
|
return self._cache[property]
|
|
else:
|
|
raise KeyError(
|
|
"Requested key %s was not found in registry cache." % str(property)
|
|
)
|