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         super(GalconClientCmdLoggerBaseMixin, self).__init__(*a, **kw)
  41         self._command_log_file = self._getopt("command_log_file", kw)
  42         if isinstance(self._command_log_file, basestring):
  43             self._command_log_file = file(self._command_log_file, "wb")
  44         print repr(self._command_log_file)
  45
  46     @classmethod
  47     def _modify_option_parser(cls, option_parser):
  48         C = GalconClientCmdLoggerBaseMixin
  49         option_parser = super(C, cls)._modify_option_parser(option_parser)
  50         option_parser.add_option("-l", "--command-log-file",
  51                                  dest = cls._mkopt("command_log_file"),
  52                                  metavar = "FILE", default = None,
  53                                  help = "log Galcon commands to this file")
  54         return option_parser
  55
  56
  57 class GalconClientRecvCmdLoggerMixin(GalconClientCmdLoggerBaseMixin):
  58     """
  59     GalconClient that dumps received commands to self.command_dump_file.
  60
  61     TODO: GalconClientSendCmdLoggerMixin
  62     """
  63
  64     def datagramReceived(self, datagram, host):
  65         """
  66         Override datagramReceived instead of _commands_received as we want
  67         to log *everything* (resent commands as well).
  68         """
  69         f = self._command_log_file
  70         if f is not None:
  71             f.write("\n*** COMMANDS ** SERVER ** %f ***\n" % time.time())
  72             f.write(datagram)
  73         # Call parent class' datagramReceived method:
  74         Cls = GalconClientRecvCmdLoggerMixin
  75         return super(Cls, self).datagramReceived(datagram, host)
  76