Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 1 | :mod:`asynchat` --- Asynchronous socket command/response handler |
| 2 | ================================================================ |
| 3 | |
| 4 | .. module:: asynchat |
| 5 | :synopsis: Support for asynchronous command/response protocols. |
| 6 | .. moduleauthor:: Sam Rushing <rushing@nightmare.com> |
Josiah Carlson | 35bf9bf | 2008-07-07 04:24:24 +0000 | [diff] [blame] | 7 | .. sectionauthor:: Steve Holden <sholden@holdenweb.com> |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 8 | |
Raymond Hettinger | ead4975 | 2011-01-24 16:28:06 +0000 | [diff] [blame] | 9 | **Source code:** :source:`Lib/asynchat.py` |
| 10 | |
| 11 | -------------- |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 12 | |
Guido van Rossum | 4da459c | 2013-11-22 12:27:45 -0800 | [diff] [blame] | 13 | .. note:: |
| 14 | |
| 15 | This module exists for backwards compatibility only. For new code we |
| 16 | recommend using :mod:`asyncio`. |
Guido van Rossum | aa40775 | 2013-11-22 11:57:35 -0800 | [diff] [blame] | 17 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 18 | This module builds on the :mod:`asyncore` infrastructure, simplifying |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 19 | asynchronous clients and servers and making it easier to handle protocols |
| 20 | whose elements are terminated by arbitrary strings, or are of variable length. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 21 | :mod:`asynchat` defines the abstract class :class:`async_chat` that you |
| 22 | subclass, providing implementations of the :meth:`collect_incoming_data` and |
| 23 | :meth:`found_terminator` methods. It uses the same asynchronous loop as |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 24 | :mod:`asyncore`, and the two types of channel, :class:`asyncore.dispatcher` |
| 25 | and :class:`asynchat.async_chat`, can freely be mixed in the channel map. |
| 26 | Typically an :class:`asyncore.dispatcher` server channel generates new |
| 27 | :class:`asynchat.async_chat` channel objects as it receives incoming |
| 28 | connection requests. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 29 | |
| 30 | |
| 31 | .. class:: async_chat() |
| 32 | |
| 33 | This class is an abstract subclass of :class:`asyncore.dispatcher`. To make |
| 34 | practical use of the code you must subclass :class:`async_chat`, providing |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 35 | meaningful :meth:`collect_incoming_data` and :meth:`found_terminator` |
| 36 | methods. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 37 | The :class:`asyncore.dispatcher` methods can be used, although not all make |
| 38 | sense in a message/response context. |
| 39 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 40 | Like :class:`asyncore.dispatcher`, :class:`async_chat` defines a set of |
| 41 | events that are generated by an analysis of socket conditions after a |
Georg Brandl | 60203b4 | 2010-10-06 10:11:56 +0000 | [diff] [blame] | 42 | :c:func:`select` call. Once the polling loop has been started the |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 43 | :class:`async_chat` object's methods are called by the event-processing |
| 44 | framework with no action on the part of the programmer. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 45 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 46 | Two class attributes can be modified, to improve performance, or possibly |
| 47 | even to conserve memory. |
| 48 | |
| 49 | |
| 50 | .. data:: ac_in_buffer_size |
| 51 | |
| 52 | The asynchronous input buffer size (default ``4096``). |
| 53 | |
| 54 | |
| 55 | .. data:: ac_out_buffer_size |
| 56 | |
| 57 | The asynchronous output buffer size (default ``4096``). |
| 58 | |
| 59 | Unlike :class:`asyncore.dispatcher`, :class:`async_chat` allows you to |
| 60 | define a first-in-first-out queue (fifo) of *producers*. A producer need |
| 61 | have only one method, :meth:`more`, which should return data to be |
| 62 | transmitted on the channel. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 63 | The producer indicates exhaustion (*i.e.* that it contains no more data) by |
Serhiy Storchaka | 5e028ae | 2014-02-06 21:10:41 +0200 | [diff] [blame] | 64 | having its :meth:`more` method return the empty bytes object. At this point |
| 65 | the :class:`async_chat` object removes the producer from the fifo and starts |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 66 | using the next producer, if any. When the producer fifo is empty the |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 67 | :meth:`handle_write` method does nothing. You use the channel object's |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 68 | :meth:`set_terminator` method to describe how to recognize the end of, or |
| 69 | an important breakpoint in, an incoming transmission from the remote |
| 70 | endpoint. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 71 | |
| 72 | To build a functioning :class:`async_chat` subclass your input methods |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 73 | :meth:`collect_incoming_data` and :meth:`found_terminator` must handle the |
| 74 | data that the channel receives asynchronously. The methods are described |
| 75 | below. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 76 | |
| 77 | |
| 78 | .. method:: async_chat.close_when_done() |
| 79 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 80 | Pushes a ``None`` on to the producer fifo. When this producer is popped off |
| 81 | the fifo it causes the channel to be closed. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 82 | |
| 83 | |
| 84 | .. method:: async_chat.collect_incoming_data(data) |
| 85 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 86 | Called with *data* holding an arbitrary amount of received data. The |
| 87 | default method, which must be overridden, raises a |
| 88 | :exc:`NotImplementedError` exception. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 89 | |
| 90 | |
| 91 | .. method:: async_chat.discard_buffers() |
| 92 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 93 | In emergencies this method will discard any data held in the input and/or |
| 94 | output buffers and the producer fifo. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 95 | |
| 96 | |
| 97 | .. method:: async_chat.found_terminator() |
| 98 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 99 | Called when the incoming data stream matches the termination condition set |
| 100 | by :meth:`set_terminator`. The default method, which must be overridden, |
| 101 | raises a :exc:`NotImplementedError` exception. The buffered input data |
| 102 | should be available via an instance attribute. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 103 | |
| 104 | |
| 105 | .. method:: async_chat.get_terminator() |
| 106 | |
| 107 | Returns the current terminator for the channel. |
| 108 | |
| 109 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 110 | .. method:: async_chat.push(data) |
| 111 | |
Giampaolo RodolĂ | 8130290 | 2010-04-29 20:45:01 +0000 | [diff] [blame] | 112 | Pushes data on to the channel's fifo to ensure its transmission. |
| 113 | This is all you need to do to have the channel write the data out to the |
| 114 | network, although it is possible to use your own producers in more complex |
| 115 | schemes to implement encryption and chunking, for example. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 116 | |
| 117 | |
| 118 | .. method:: async_chat.push_with_producer(producer) |
| 119 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 120 | Takes a producer object and adds it to the producer fifo associated with |
| 121 | the channel. When all currently-pushed producers have been exhausted the |
| 122 | channel will consume this producer's data by calling its :meth:`more` |
| 123 | method and send the data to the remote endpoint. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 124 | |
| 125 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 126 | .. method:: async_chat.set_terminator(term) |
| 127 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 128 | Sets the terminating condition to be recognized on the channel. ``term`` |
| 129 | may be any of three types of value, corresponding to three different ways |
| 130 | to handle incoming protocol data. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 131 | |
| 132 | +-----------+---------------------------------------------+ |
| 133 | | term | Description | |
| 134 | +===========+=============================================+ |
| 135 | | *string* | Will call :meth:`found_terminator` when the | |
| 136 | | | string is found in the input stream | |
| 137 | +-----------+---------------------------------------------+ |
| 138 | | *integer* | Will call :meth:`found_terminator` when the | |
| 139 | | | indicated number of characters have been | |
| 140 | | | received | |
| 141 | +-----------+---------------------------------------------+ |
| 142 | | ``None`` | The channel continues to collect data | |
| 143 | | | forever | |
| 144 | +-----------+---------------------------------------------+ |
| 145 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 146 | Note that any data following the terminator will be available for reading |
| 147 | by the channel after :meth:`found_terminator` is called. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 148 | |
| 149 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 150 | .. _asynchat-example: |
| 151 | |
| 152 | asynchat Example |
| 153 | ---------------- |
| 154 | |
| 155 | The following partial example shows how HTTP requests can be read with |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 156 | :class:`async_chat`. A web server might create an |
| 157 | :class:`http_request_handler` object for each incoming client connection. |
| 158 | Notice that initially the channel terminator is set to match the blank line at |
| 159 | the end of the HTTP headers, and a flag indicates that the headers are being |
| 160 | read. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 161 | |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 162 | Once the headers have been read, if the request is of type POST (indicating |
| 163 | that further data are present in the input stream) then the |
| 164 | ``Content-Length:`` header is used to set a numeric terminator to read the |
| 165 | right amount of data from the channel. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 166 | |
| 167 | The :meth:`handle_request` method is called once all relevant input has been |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 168 | marshalled, after setting the channel terminator to ``None`` to ensure that |
| 169 | any extraneous data sent by the web client are ignored. :: |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 170 | |
Giampaolo Rodola' | 0fb41b5 | 2012-05-15 15:46:00 +0200 | [diff] [blame] | 171 | |
| 172 | import asynchat |
| 173 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 174 | class http_request_handler(asynchat.async_chat): |
| 175 | |
Benjamin Peterson | 9bc9351 | 2008-09-22 22:10:59 +0000 | [diff] [blame] | 176 | def __init__(self, sock, addr, sessions, log): |
| 177 | asynchat.async_chat.__init__(self, sock=sock) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 178 | self.addr = addr |
| 179 | self.sessions = sessions |
| 180 | self.ibuffer = [] |
Josiah Carlson | 1893ce7 | 2008-07-07 04:23:14 +0000 | [diff] [blame] | 181 | self.obuffer = b"" |
| 182 | self.set_terminator(b"\r\n\r\n") |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 183 | self.reading_headers = True |
| 184 | self.handling = False |
| 185 | self.cgi_data = None |
| 186 | self.log = log |
| 187 | |
| 188 | def collect_incoming_data(self, data): |
| 189 | """Buffer the data""" |
| 190 | self.ibuffer.append(data) |
| 191 | |
| 192 | def found_terminator(self): |
| 193 | if self.reading_headers: |
| 194 | self.reading_headers = False |
Serhiy Storchaka | 5e028ae | 2014-02-06 21:10:41 +0200 | [diff] [blame] | 195 | self.parse_headers(b"".join(self.ibuffer)) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 196 | self.ibuffer = [] |
Josiah Carlson | 1893ce7 | 2008-07-07 04:23:14 +0000 | [diff] [blame] | 197 | if self.op.upper() == b"POST": |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 198 | clen = self.headers.getheader("content-length") |
| 199 | self.set_terminator(int(clen)) |
| 200 | else: |
| 201 | self.handling = True |
| 202 | self.set_terminator(None) |
| 203 | self.handle_request() |
| 204 | elif not self.handling: |
| 205 | self.set_terminator(None) # browsers sometimes over-send |
Josiah Carlson | 1893ce7 | 2008-07-07 04:23:14 +0000 | [diff] [blame] | 206 | self.cgi_data = parse(self.headers, b"".join(self.ibuffer)) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 207 | self.handling = True |
| 208 | self.ibuffer = [] |
| 209 | self.handle_request() |