netwerk/dns/prepare_tlds.py
author arthur.iakab <aiakab@mozilla.com>
Tue, 03 Apr 2018 18:30:58 +0300
changeset 411462 5ec55f7a95f94cf39d57cdf36abecab914ea48eb
parent 373406 276db25c9e92c04073c7fc97e0029ded2e0885d2
child 429636 20668aaf20ebbef0dad5b18b0db001bee589da3e
permissions -rw-r--r--
Backed out changeset 3ff8ce674396 (bug 1447460) on request of tomprince a=backout

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

import codecs
import encodings.idna
import imp
import os
import re
import sys

"""
Processes a file containing effective TLD data.  See the following URL for a
description of effective TLDs and of the file format that this script
processes (although for the latter you're better off just reading this file's
short source code).

http://wiki.mozilla.org/Gecko:Effective_TLD_Service
"""

def getEffectiveTLDs(path):
  file = codecs.open(path, "r", "UTF-8")
  entries = []
  domains = set()
  for line in file:
    # line always contains a line terminator unless the file is empty
    if len(line) == 0:
      raise StopIteration
    line = line.rstrip()
    # comment, empty, or superfluous line for explicitness purposes
    if line.startswith("//") or "." not in line:
      continue
    line = re.split(r"[ \t\n]", line, 1)[0]
    entry = EffectiveTLDEntry(line)
    domain = entry.domain()
    assert domain not in domains, \
           "repeating domain %s makes no sense" % domain
    domains.add(domain)
    yield entry

def _normalizeHostname(domain):
  """
  Normalizes the given domain, component by component.  ASCII components are
  lowercased, while non-ASCII components are processed using the ToASCII
  algorithm.
  """
  def convertLabel(label):
    if _isASCII(label):
      return label.lower()
    return encodings.idna.ToASCII(label)
  return ".".join(map(convertLabel, domain.split(".")))

def _isASCII(s):
  "True if s consists entirely of ASCII characters, false otherwise."
  for c in s:
    if ord(c) > 127:
      return False
  return True

class EffectiveTLDEntry:
  """
  Stores an entry in an effective-TLD name file.
  """

  _exception = False
  _wild = False

  def __init__(self, line):
    """
    Creates a TLD entry from a line of data, which must have been stripped of
    the line ending.
    """
    if line.startswith("!"):
      self._exception = True
      domain = line[1:]
    elif line.startswith("*."):
      self._wild = True
      domain = line[2:]
    else:
      domain = line
    self._domain = _normalizeHostname(domain)

  def domain(self):
    "The domain this represents."
    return self._domain

  def exception(self):
    "True if this entry's domain denotes does not denote an effective TLD."
    return self._exception

  def wild(self):
    "True if this entry represents a class of effective TLDs."
    return self._wild


#################
# DO EVERYTHING #
#################

def main(output, effective_tld_filename):
  """
  effective_tld_filename is the effective TLD file to parse.
  A C++ array of a binary representation of a DAFSA representing the
  eTLD file is then printed to output.
  """

  # Find and load the `make_dafsa.py` script under xpcom/ds.
  tld_dir = os.path.dirname(effective_tld_filename)
  make_dafsa_py = os.path.join(tld_dir, '../../xpcom/ds/make_dafsa.py')
  sys.path.append(os.path.dirname(make_dafsa_py))
  with open(make_dafsa_py, 'r') as fh:
    make_dafsa = imp.load_module('script', fh, make_dafsa_py,
                                 ('.py', 'r', imp.PY_SOURCE))

  def typeEnum(etld):
    """
    Maps the flags to the DAFSA's enum types.
    """
    if etld.exception():
      return 1
    elif etld.wild():
      return 2
    else:
      return 0

  def dafsa_words():
    """
    make_dafsa expects lines of the form "<domain_name><enum_value>"
    """
    for etld in getEffectiveTLDs(effective_tld_filename):
      yield "%s%d" % (etld.domain(), typeEnum(etld))

  output.write(make_dafsa.words_to_cxx(dafsa_words()))

if __name__ == '__main__':
    main(sys.stdout, sys.argv[1])