One Hat Cyber Team

View File Name : errors.py

# Copyright (C) 2016-2017 g10 Code GmbH
# Copyright (C) 2004 Igor Belyi <belyi@users.sourceforge.net>
# Copyright (C) 2002 John Goerzen <jgoerzen@complete.org>
#
#    This library is free software; you can redistribute it and/or
#    modify it under the terms of the GNU Lesser General Public
#    License as published by the Free Software Foundation; either
#    version 2.1 of the License, or (at your option) any later version.
#
#    This library is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
#    Lesser General Public License for more details.
#
#    You should have received a copy of the GNU Lesser General Public
#    License along with this library; if not, write to the Free Software
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA

from __future__ import absolute_import, print_function, unicode_literals

from . import gpgme
from . import util

del absolute_import, print_function, unicode_literals

# To appease static analysis tools, we define some constants here.
# They are overwritten with the proper values by process_constants.
NO_ERROR = None
EOF = None

util.process_constants('GPG_ERR_', globals())
del util


class GpgError(Exception):
    """A GPG Error

    This is the base of all errors thrown by this library.

    If the error originated from GPGME, then additional information
    can be found by looking at 'code' for the error code, and 'source'
    for the errors origin.  Suitable constants for comparison are
    defined in this module.  'code_str' and 'source_str' are
    human-readable versions of the former two properties.

    If 'context' is not None, then it contains a human-readable hint
    as to where the error originated from.

    If 'results' is not None, it is a tuple containing results of the
    operation that failed.  The tuples elements are the results of the
    function that raised the error.  Some operations return results
    even though they signal an error.  Of course this information must
    be taken with a grain of salt.  But often, this information is
    useful for diagnostic uses or to give the user feedback.  Since
    the normal control flow is disrupted by the exception, the callee
    can no longer return results, hence we attach them to the
    exception objects.

    """

    def __init__(self, error=None, context=None, results=None):
        self.error = error
        self.context = context
        self.results = results

    @property
    def code(self):
        if self.error is None:
            return None
        return gpgme.gpgme_err_code(self.error)

    @property
    def code_str(self):
        if self.error is None:
            return None
        return gpgme.gpgme_strerror(self.error)

    @property
    def source(self):
        if self.error is None:
            return None
        return gpgme.gpgme_err_source(self.error)

    @property
    def source_str(self):
        if self.error is None:
            return None
        return gpgme.gpgme_strsource(self.error)

    def __str__(self):
        msgs = []
        if self.context is not None:
            msgs.append(self.context)
        if self.error is not None:
            msgs.append(self.source_str)
            msgs.append(self.code_str)
        return ': '.join(msgs)


class GPGMEError(GpgError):
    '''Generic error

    This is a generic error that wraps the underlying libraries native
    error type.  It is thrown when the low-level API is invoked and
    returns an error.  This is the error that was used in PyME.

    '''

    @classmethod
    def fromSyserror(cls):
        return cls(gpgme.gpgme_err_code_from_syserror())

    @property
    def message(self):
        return self.context

    def getstring(self):
        return str(self)

    def getcode(self):
        return self.code

    def getsource(self):
        return self.source


def errorcheck(retval, extradata=None):
    if retval:
        raise GPGMEError(retval, extradata)


class KeyNotFound(GPGMEError, KeyError):
    """Raised if a key was not found

    GPGME indicates this condition with EOF, which is not very
    idiomatic.  We raise this error that is both a GPGMEError
    indicating EOF, and a KeyError.

    """

    def __init__(self, keystr):
        self.keystr = keystr
        GPGMEError.__init__(self, EOF)

    def __str__(self):
        return self.keystr


# These errors are raised in the idiomatic interface code.


class EncryptionError(GpgError):
    pass


class InvalidRecipients(EncryptionError):
    def __init__(self, recipients, **kwargs):
        EncryptionError.__init__(self, **kwargs)
        self.recipients = recipients

    def __str__(self):
        return ", ".join("{}: {}".format(r.fpr, gpgme.gpgme_strerror(r.reason))
                         for r in self.recipients)


class DecryptionError(GpgError):
    pass


class UnsupportedAlgorithm(DecryptionError):
    def __init__(self, algorithm, **kwargs):
        DecryptionError.__init__(self, **kwargs)
        self.algorithm = algorithm

    def __str__(self):
        return self.algorithm


class SigningError(GpgError):
    pass


class InvalidSigners(SigningError):
    def __init__(self, signers, **kwargs):
        SigningError.__init__(self, **kwargs)
        self.signers = signers

    def __str__(self):
        return ", ".join("{}: {}".format(s.fpr, gpgme.gpgme_strerror(s.reason))
                         for s in self.signers)


class VerificationError(GpgError):
    def __init__(self, result, **kwargs):
        GpgError.__init__(self, **kwargs)
        self.result = result


class BadSignatures(VerificationError):
    def __str__(self):
        return ", ".join("{}: {}".format(s.fpr, gpgme.gpgme_strerror(s.status))
                         for s in self.result.signatures
                         if s.status != NO_ERROR)


class MissingSignatures(VerificationError):
    def __init__(self, result, missing, **kwargs):
        VerificationError.__init__(self, result, **kwargs)
        self.missing = missing

    def __str__(self):
        return ", ".join(k.subkeys[0].fpr for k in self.missing)