diff options
| author | Jonny Lamb <jonny.lamb@collabora.co.uk> | 2010-03-15 21:41:29 +0000 |
|---|---|---|
| committer | Louis-Francis Ratté-Boulianne <louis-francis.ratte-boulianne@collabora.co.uk> | 2010-06-03 16:42:30 -0400 |
| commit | 671f160be02c78ea41e55c931025e2856ec868f8 (patch) | |
| tree | 3cf2f3b65748aac12820aa723330079754865b0e | |
| parent | 98eb001f2ebf5ddaa17a7beefb0b851b5ce05d4c (diff) | |
file transfer: raise NotImplemented if called with unknown socket type
Signed-off-by: Jonny Lamb <jonny.lamb@collabora.co.uk>
| -rw-r--r-- | butterfly/channel/filetransfer.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/butterfly/channel/filetransfer.py b/butterfly/channel/filetransfer.py index 4334fdb..d60f62e 100644 --- a/butterfly/channel/filetransfer.py +++ b/butterfly/channel/filetransfer.py @@ -118,6 +118,10 @@ class ButterflyFileTransferChannel(telepathy.server.ChannelTypeFileTransfer): def AcceptFile(self, address_type, access_control, param, offset): self._receiving = True logger.debug("Accept file") + + if address_type not in self.socket_types.keys(): + raise telepathy.NotImplemented("Socket type %u is unsupported" % address_type) + self.socket = self.add_listener() self.channel = self.add_io_channel(self.socket) self.set_state(telepathy.FILE_TRANSFER_STATE_PENDING, @@ -130,6 +134,10 @@ class ButterflyFileTransferChannel(telepathy.server.ChannelTypeFileTransfer): def ProvideFile(self, address_type, access_control, param): self._receiving = False logger.debug("Provide file") + + if address_type not in self.socket_types.keys(): + raise telepathy.NotImplemented("Socket type %u is unsupported" % address_type) + self.socket = self.add_listener() self.channel = self.add_io_channel(self.socket) return self.socket.getsockname() |