galcon/net_client_logger.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 """
   4 Galcon networking - client-side logging.
   5 """
   6 # Copyright (C) 2007 Felix Rabe  <public@felixrabe.textdriven.com>
   7 # Copyright (C) 2007 Michael Carter
   8
   9 # Permission is hereby granted, free of charge, to any person obtaining a
  10 # copy of this software and associated documentation files (the
  11 # "Software"), to deal in the Software without restriction, including
  12 # without limitation the rights to use, copy, modify, merge, publish,
  13 # distribute, sublicense, and/or sell copies of the Software, and to permit
  14 # persons to whom the Software is furnished to do so, subject to the
  15 # following conditions:
  16 #
  17 # The above copyright notice and this permission notice shall be included
  18 # in all copies or substantial portions of the Software.
  19 #
  20 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  21 # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
  23 # IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
  24 # CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT
  25 # OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR
  26 # THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27
  28 # Recommended line length or text width: 75 characters.
  29
  30 import time
  31 from galcon.net_client_base import *
  32
  33
  34 class GalconClientCmdLoggerBaseMixin(GalconClientBase):
  35     """
  36     GalconClient base mixin for classes implementing traffic dumping.
  37     """
  38
  39     def __init__(self, *a, **kw):
  40         self.command_log_file = kw.pop("traffic_dump_file", None)
  41         super(GalconClientCmdLoggerBaseMixin, self).__init__(*a, **kw)
  42
  43
  44 class GalconClientRecvCmdLoggerMixin(GalconClientCmdLoggerBaseMixin):
  45     """
  46     GalconClient that dumps received commands to self.command_dump_file.
  47
  48     TODO: GalconClientSendCmdLoggerMixin
  49     """
  50
  51     def datagramReceived(self, datagram, host):
  52         """
  53         Override datagramReceived instead of _commands_received as we want
  54         to log *everything* (resent commands as well).
  55         """
  56         f = self.command_log_file
  57         if f is not None:
  58             f.write("\n*** COMMANDS ** SERVER ** %f ***\n" % time.time())
  59             f.write(datagram)
  60         # Call parent class' datagramReceived method:
  61         Cls = GalconClientRecvCmdLoggerMixin
  62         return super(Cls, self).datagramReceived(datagram, host)
  63