Limnoria/src/socketDrivers.py

197 lines
7.2 KiB
Python
Raw Normal View History

2003-09-01 10:06:55 +02:00
#!/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$"
2003-11-25 09:38:19 +01:00
import fix
2003-09-01 10:06:55 +02:00
import time
2004-01-18 08:58:26 +01:00
import atexit
import select
2003-09-01 10:06:55 +02:00
import socket
from itertools import imap
2003-09-01 10:06:55 +02:00
import log
2003-09-01 10:06:55 +02:00
import conf
import world
2003-09-01 10:06:55 +02:00
import drivers
import ircmsgs
import schedule
instances = 0
2004-01-18 08:58:26 +01:00
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)
2003-09-01 10:06:55 +02:00
class SocketDriver(drivers.IrcDriver):
def __init__(self, (server, port), irc, reconnectWaits=(0, 60, 300)):
global instances
instances += 1
2004-01-18 08:58:26 +01:00
conf.supybot.drivers.poll.setValue(originalPoll / instances)
2003-09-01 10:06:55 +02:00
self.server = (server, port)
drivers.IrcDriver.__init__(self) # Must come after server is set.
2003-09-01 10:06:55 +02:00
self.irc = irc
self.irc.driver = self
self.inbuffer = ''
self.outbuffer = ''
self.connected = False
self.eagains = 0
self.reconnectWaitsIndex = 0
self.reconnectWaits = reconnectWaits
2003-09-01 10:06:55 +02:00
self.reconnect()
def _sendIfMsgs(self):
msgs = [self.irc.takeMsg()]
2003-09-01 10:06:55 +02:00
while msgs[-1] is not None:
msgs.append(self.irc.takeMsg())
del msgs[-1]
self.outbuffer += ''.join(imap(str, msgs))
2003-09-01 10:06:55 +02:00
if self.outbuffer:
try:
sent = self.conn.send(self.outbuffer)
self.outbuffer = self.outbuffer[sent:]
self.eagains = 0
except socket.error, e:
# (11, 'Resource temporarily unavailable') raised if connect
# hasn't finished yet.
if e.args[0] != 11 and self.eagains > 120:
log.warning('Disconnect from %s: %s', self.server, e)
2004-01-02 23:29:14 +01:00
self.reconnect(wait=True)
else:
log.debug('Got EAGAIN, current count: %s', self.eagains)
self.eagains += 1
2003-09-01 10:06:55 +02:00
def run(self):
if not self.connected:
2004-01-18 08:58:26 +01:00
# 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())
2003-09-01 10:06:55 +02:00
return
self._sendIfMsgs()
try:
self.inbuffer += self.conn.recv(1024)
self.eagains = 0
2003-09-01 10:06:55 +02:00
lines = self.inbuffer.split('\n')
self.inbuffer = lines.pop()
for line in lines:
start = time.time()
2003-09-01 10:06:55 +02:00
msg = ircmsgs.IrcMsg(line)
2004-02-16 09:10:50 +01:00
#log.debug('Time to parse IrcMsg: %s', time.time()-start)
self.irc.feedMsg(msg)
2003-09-01 10:06:55 +02:00
except socket.timeout:
pass
except socket.error, e:
# Same as with _sendIfMsgs.
if e.args[0] != 11 or self.eagains > 120:
log.warning('Disconnect from %s: %s', self.server, e)
2004-01-02 23:29:14 +01:00
self.reconnect(wait=True)
else:
log.debug('Got EAGAIN, current count: %s', self.eagains)
self.eagains += 1
return
2003-09-01 10:06:55 +02:00
self._sendIfMsgs()
2004-01-01 21:12:01 +01:00
def reconnect(self, wait=False):
2004-01-05 13:03:54 +01:00
if self.connected:
2004-02-14 01:24:32 +01:00
log.info('Reconnect called on driver for %s.' % self.irc)
2004-01-05 13:03:54 +01:00
self.conn.close()
elif not wait:
2004-02-14 01:24:32 +01:00
log.info('Connecting to %s.' % ':'.join(map(str, self.server)))
2004-01-05 13:03:54 +01:00
self.connected = False
2004-01-01 21:12:01 +01:00
if wait:
2004-01-03 16:51:53 +01:00
log.info('Reconnect waiting.')
2004-01-01 21:12:01 +01:00
self._scheduleReconnect()
return
2004-01-05 13:03:54 +01:00
self.irc.reset()
2003-09-01 10:06:55 +02:00
self.conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
2004-01-18 08:58:26 +01:00
# We allow more time for the connect here, since it might take longer.
# At least 10 seconds.
self.conn.settimeout(max(10, conf.supybot.drivers.poll()*10))
if self.reconnectWaitsIndex < len(self.reconnectWaits)-1:
self.reconnectWaitsIndex += 1
try:
self.conn.connect(self.server)
2004-01-18 08:58:26 +01:00
self.conn.settimeout(conf.supybot.drivers.poll())
except socket.error, e:
if e.args[0] == 115:
now = time.time()
when = now + 60
whenS = log.timestamp(when)
log.info('Connection in progress, scheduling connectedness '
'check for %s', whenS)
schedule.addEvent(self._checkAndWriteOrReconnect, when)
else:
2004-03-25 13:14:41 +01:00
log.warning('Error connecting to %s: %s', self.server[0], e)
2004-01-01 21:12:01 +01:00
self.reconnect(wait=True)
return
self.connected = True
self.reconnectWaitPeriodsIndex = 0
def _checkAndWriteOrReconnect(self):
log.debug('Checking whether we are connected.')
(_, w, _) = select.select([], [self.conn], [], 0)
if w:
log.info('Socket is writable, it might be connected.')
self.connected = True
self.reconnectWaitPeriodsIndex = 0
else:
2004-03-25 13:14:41 +01:00
log.warning('Error connecting to %s: Timed out.', self.server[0])
self.reconnect()
2004-01-01 21:12:01 +01:00
def _scheduleReconnect(self):
when = time.time() + self.reconnectWaits[self.reconnectWaitsIndex]
if not world.dying:
whenS = log.timestamp(when)
log.info('Scheduling reconnect to %s at %s', self.server, whenS)
schedule.addEvent(self.reconnect, when)
2003-09-01 10:06:55 +02:00
def die(self):
log.info('Driver for %s dying.', self.irc)
self.conn.close()
# self.irc.die() Kill off the ircs yourself, jerk!
def name(self):
return '%s%s' % (self.__class__.__name__, self.server)
2003-09-01 10:06:55 +02:00
Driver = SocketDriver
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: