sdk/platform-tools/systrace/catapult/telemetry/third_party/web-page-replay/third_party/dns/tsigkeyring.py

   1 # Copyright (C) 2003-2007, 2009, 2010 Nominum, Inc.
   2 #
   3 # Permission to use, copy, modify, and distribute this software and its
   4 # documentation for any purpose with or without fee is hereby granted,
   5 # provided that the above copyright notice and this permission notice
   6 # appear in all copies.
   7 #
   8 # THE SOFTWARE IS PROVIDED "AS IS" AND NOMINUM DISCLAIMS ALL WARRANTIES
   9 # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10 # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NOMINUM BE LIABLE FOR
  11 # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12 # WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
  14 # OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15
  16 """A place to store TSIG keys."""
  17
  18 import base64
  19
  20 import dns.name
  21
  22 def from_text(textring):
  23     """Convert a dictionary containing (textual DNS name, base64 secret) pairs
  24     into a binary keyring which has (dns.name.Name, binary secret) pairs.
  25     @rtype: dict"""
  26
  27     keyring = {}
  28     for keytext in textring:
  29         keyname = dns.name.from_text(keytext)
  30         secret = base64.decodestring(textring[keytext])
  31         keyring[keyname] = secret
  32     return keyring
  33
  34 def to_text(keyring):
  35     """Convert a dictionary containing (dns.name.Name, binary secret) pairs
  36     into a text keyring which has (textual DNS name, base64 secret) pairs.
  37     @rtype: dict"""
  38
  39     textring = {}
  40     for keyname in keyring:
  41         keytext = dns.name.to_text(keyname)
  42         secret = base64.encodestring(keyring[keyname])
  43         textring[keytext] = secret
  44     return textring