Merge pull request #741 from mitmproxy/proxy-refactor-cb

Proxy Refactor

1 files changed, 34 insertions, 0 deletions

diff --git a/libmproxy/exceptions.py b/libmproxy/exceptions.py
new file mode 100644
index 00000000..f34d9707
--- /dev/null
+++ b/libmproxy/exceptions.py
@@ -0,0 +1,34 @@
+from __future__ import (absolute_import, print_function, division)
+
+
+class ProxyException(Exception):
+    """
+    Base class for all exceptions thrown by libmproxy.
+    """
+    def __init__(self, message, cause=None):
+        """
+        :param message: Error Message
+        :param cause: Exception object that caused this exception to be thrown.
+        """
+        super(ProxyException, self).__init__(message)
+        self.cause = cause
+
+
+class ProtocolException(ProxyException):
+    pass
+
+
+class Socks5Exception(ProtocolException):
+    pass
+
+
+class HttpException(ProtocolException):
+    pass
+
+
+class InvalidCredentials(HttpException):
+    pass
+
+
+class ServerException(ProxyException):
+    pass