diff options
Diffstat (limited to 'cloudinit/sources/helpers')
| -rw-r--r-- | cloudinit/sources/helpers/hetzner.py | 19 | 
1 files changed, 19 insertions, 0 deletions
| diff --git a/cloudinit/sources/helpers/hetzner.py b/cloudinit/sources/helpers/hetzner.py index 2554530d..72edb023 100644 --- a/cloudinit/sources/helpers/hetzner.py +++ b/cloudinit/sources/helpers/hetzner.py @@ -7,6 +7,9 @@ from cloudinit import log as logging  from cloudinit import url_helper  from cloudinit import util +import base64 +import binascii +  LOG = logging.getLogger(__name__) @@ -24,3 +27,19 @@ def read_userdata(url, timeout=2, sec_between=2, retries=30):      if not response.ok():          raise RuntimeError("unable to read userdata at %s" % url)      return response.contents + + +def maybe_b64decode(data: bytes) -> bytes: +    """base64 decode data + +    If data is base64 encoded bytes, return b64decode(data). +    If not, return data unmodified. + +    @param data: data as bytes. TypeError is raised if not bytes. +    """ +    if not isinstance(data, bytes): +        raise TypeError("data is '%s', expected bytes" % type(data)) +    try: +        return base64.b64decode(data, validate=True) +    except binascii.Error: +        return data | 
