Error as WifimitmError

wifimitm/access.py

@@ -13,6 +13,7 @@
 import subprocess
 from contextlib import contextmanager
 
+from wifimitm.common import WifimitmError
 from .model import WirelessInterface, WirelessAccessPoint
 from .wep import WepAttacker
 from .wpa2 import Wpa2Attacker
@@ -23,7 +24,7 @@
 logger = logging.getLogger(__name__)
 
 
-class NotCrackedError(Exception):
+class NotCrackedError(WifimitmError):
  pass
 
 

wifimitm/common.py

@@ -24,6 +24,10 @@
 logger = logging.getLogger(__name__)
 
 
+class WifimitmError(Exception):
+ pass
+
+
 def csv_row_station_bssid(row):
  """
  Provide associated bssid of given station.

wifimitm/impersonation.py

@@ -12,6 +12,7 @@
 from enum import Enum, unique
 from typing import Optional, Dict
 
+from wifimitm.common import WifimitmError
 from wifimitm.updatableProcess import UpdatableProcess
 from .model import WirelessInterface, WirelessAccessPoint
 
@@ -112,7 +113,7 @@ def parse_from_line(cls, line: str):
  client_id = parts[4]
  return cls(expiration, mac_address, ip_address, hostname, client_id)
 
- class UnexpectedTerminationError(Exception):
+ class UnexpectedTerminationError(WifimitmError):
  pass
 
  def __init__(self,

wifimitm/requirements.py

@@ -13,6 +13,8 @@
 from abc import ABC, abstractmethod
 from typing import List
 
+from wifimitm.common import WifimitmError
+
 __author__ = 'Martin Vondracek'
 __email__ = '[email protected]'
 
@@ -41,7 +43,7 @@ def msg(self) -> str:
  """
 
 
-class RequirementError(Exception):
+class RequirementError(WifimitmError):
  def __init__(self, requirement: Requirement):
  """
  :type requirement: Requirement

wifimitm/wpa2.py

@@ -35,15 +35,15 @@
 import pkg_resources
 
 from .model import WirelessAccessPoint, WirelessInterface
-from .common import WirelessCapturer, deauthenticate
+from .common import WirelessCapturer, deauthenticate, WifimitmError
 
 __author__ = 'Martin Vondracek'
 __email__ = '[email protected]'
 
 logger = logging.getLogger(__name__)
 
 
-class Wpa2Error(Exception):
+class Wpa2Error(WifimitmError):
  pass