gsh/buffered_dispatcher.py

   1 # This program is free software; you can redistribute it and/or modify
   2 # it under the terms of the GNU General Public License as published by
   3 # the Free Software Foundation; either version 2 of the License, or
   4 # (at your option) any later version.
   5 #
   6 # This program is distributed in the hope that it will be useful,
   7 # but WITHOUT ANY WARRANTY; without even the implied warranty of
   8 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   9 # GNU Library General Public License for more details.
  10 #
  11 # You should have received a copy of the GNU General Public License
  12 # along with this program; if not, write to the Free Software
  13 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  14 #
  15 # See the COPYING file for license information.
  16 #
  17 # Copyright (c) 2006, 2007, 2008 Guillaume Chazarain <guichaz@gmail.com>
  18
  19 import asyncore
  20 import errno
  21
  22 from gsh.console import console_output
  23
  24 class buffered_dispatcher(asyncore.file_dispatcher):
  25     """A dispatcher with a write buffer to allow asynchronous writers, and a
  26     read buffer to permit line oriented manipulations"""
  27
  28     # 1 MiB should be enough for everybody
  29     MAX_BUFFER_SIZE = 1 * 1024 * 1024
  30
  31     def __init__(self, fd):
  32         asyncore.file_dispatcher.__init__(self, fd)
  33         self.fd = fd
  34         self.read_buffer = ''
  35         self.write_buffer = ''
  36
  37     def handle_read(self):
  38         """Some data can be read"""
  39         new_data = ''
  40         buffer_length = len(self.read_buffer)
  41         try:
  42             while buffer_length < buffered_dispatcher.MAX_BUFFER_SIZE:
  43                 try:
  44                     piece = self.recv(4096)
  45                 except OSError, e:
  46                     if e.errno == errno.EAGAIN:
  47                         # End of the available data
  48                         break
  49                     elif e.errno == errno.EIO and new_data:
  50                         # Hopefully we could read an error message before the
  51                         # actual termination
  52                         break
  53                     else:
  54                         raise
  55                 new_data += piece
  56                 buffer_length += len(piece)
  57         finally:
  58             new_data = new_data.replace('\r', '\n')
  59             self.read_buffer += new_data
  60         return new_data
  61
  62     def readable(self):
  63         """No need to ask data if our buffer is already full"""
  64         return len(self.read_buffer) < buffered_dispatcher.MAX_BUFFER_SIZE
  65
  66     def writable(self):
  67         """Do we have something to write?"""
  68         return self.write_buffer != ''
  69
  70     def dispatch_write(self, buf):
  71         """Augment the buffer with stuff to write when possible"""
  72         self.write_buffer += buf
  73         if len(self.write_buffer) > buffered_dispatcher.MAX_BUFFER_SIZE:
  74             console_output('Buffer too big (%d) for %s\n' %
  75                                             (len(self.write_buffer), str(self)))
  76             raise asyncore.ExitNow(1)
  77