Limnoria/src/socketDrivers.py
2004-01-21 23:35:34 +00:00

173 lines
6.2 KiB
Python

#!/usr/bin/env python
###
# Copyright (c) 2002, Jeremiah Fincher
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions, and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions, and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of the author of this software nor the name of
# contributors to this software may be used to endorse or promote products
# derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###
"""
Contains simple socket drivers. Asyncore bugged (haha, pun!) me.
"""
from __future__ import division
__revision__ ="$Id$"
import fix
import time
import atexit
import socket
from itertools import imap
import log
import conf
import world
import drivers
import ircmsgs
import schedule
instances = 0
originalPoll = conf.supybot.drivers.poll()
def resetPoll():
log.info('Resetting supybot.drivers.poll to %s', originalPoll)
conf.supybot.drivers.poll.setValue(originalPoll)
atexit.register(resetPoll)
class SocketDriver(drivers.IrcDriver):
def __init__(self, (server, port), irc, reconnectWaits=(0, 60, 300)):
global instances
instances += 1
conf.supybot.drivers.poll.setValue(originalPoll / instances)
self.server = (server, port)
drivers.IrcDriver.__init__(self) # Must come after server is set.
self.irc = irc
self.irc.driver = self
self.inbuffer = ''
self.outbuffer = ''
self.connected = False
self.reconnectWaitsIndex = 0
self.reconnectWaits = reconnectWaits
self.reconnect()
def _sendIfMsgs(self):
try:
msgs = [self.irc.takeMsg()]
except Exception, e:
log.exception('Uncaught exception in irclib.Irc.takeMsg:')
return
while msgs[-1] is not None:
msgs.append(self.irc.takeMsg())
del msgs[-1]
self.outbuffer += ''.join(imap(str, msgs))
if self.outbuffer:
try:
sent = self.conn.send(self.outbuffer)
self.outbuffer = self.outbuffer[sent:]
except socket.error, e:
# (11, 'Resource temporarily unavailable') raised if connect
# hasn't finished yet.
if e.args[0] != 11:
log.warning('Disconnect from %s: %s', self.server, e)
self.reconnect(wait=True)
def run(self):
if not self.connected:
# We sleep here because otherwise, if we're the only driver, we'll
# spin at 100% CPU while we're disconnected.
time.sleep(conf.supybot.drivers.poll())
return
self._sendIfMsgs()
try:
self.inbuffer += self.conn.recv(1024)
lines = self.inbuffer.split('\n')
self.inbuffer = lines.pop()
for line in lines:
start = time.time()
msg = ircmsgs.IrcMsg(line)
log.debug('Time to parse IrcMsg: %s', time.time()-start)
try:
self.irc.feedMsg(msg)
except:
log.exception('Uncaught exception outside Irc object:')
except socket.timeout:
pass
except socket.error, e:
# Same as with _sendIfMsgs.
if e.args[0] != 11:
log.warning('Disconnect from %s: %s', self.server, e.args[1])
self.reconnect(wait=True)
return
self._sendIfMsgs()
def reconnect(self, wait=False):
log.info('Reconnect called on driver for %s.' % self.irc)
if self.connected:
self.conn.close()
self.connected = False
if wait:
log.info('Reconnect waiting.')
self._scheduleReconnect()
return
self.irc.reset()
self.conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# We allow more time for the connect here, since it might take longer.
self.conn.settimeout(conf.supybot.drivers.poll()*10)
if self.reconnectWaitsIndex < len(self.reconnectWaits)-1:
self.reconnectWaitsIndex += 1
try:
self.conn.connect(self.server)
self.conn.settimeout(conf.supybot.drivers.poll())
except socket.error, e:
if e.args[0] != 115:
log.warning('Error connecting to %s: %s',
self.irc.server, e.args[1])
self.reconnect(wait=True)
self.connected = True
self.reconnectWaitPeriodsIndex = 0
def die(self):
log.info('Driver for %s dying.', self.irc)
self.conn.close()
# self.irc.die() Kill off the ircs yourself, jerk!
def _scheduleReconnect(self):
when = time.time() + self.reconnectWaits[self.reconnectWaitsIndex]
when = log.timestamp(when)
if not world.dying:
log.info('Scheduling reconnect to %s at %s', self.server, when)
schedule.addEvent(self.reconnect, when)
def name(self):
return '%s%s' % (self.__class__.__name__, self.server)
Driver = SocketDriver
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: