def get_uname_from_mess(self, mess, full_jid=False):
nick = self.get_sender_username(mess)
node = mess.getFrom().getNode()
# Private message
if nick == node:
return mess.getFrom() if full_jid else nick
# MUC message
try:
jid = self.nick_dict[node][nick]
except KeyError:
jid = JID("default")
return jid if full_jid else jid.getNode()
def send(self, recipients, message):
"""Send message to recipients via xmpp."""
jid = JID(self.user)
if self.server:
server = self.server
else:
server = jid.getDomain()
cl = Client(server, port=self.port, debug=[])
if not cl.connect():
raise IOError("Couldn't connect to xmpp server %s" % server)
if not cl.auth(jid.getNode(), self.password, resource=self.resource):
cl.Connection.disconnect()
raise IOError("Xmpp auth erro using %s to %s", jid, server)
for recip in recipients:
cl.send(Message(recip[2], message))
class XMPPClient(object):
def __init__(self):
self.jid = JID(config.get('im', 'account'))
self.jid.setResource('xtunnel')
self.password = config.get('im', 'password')
self.debug = config.get('config', 'debug') == 'true'
self.status = 'Internal %s %s' % (tap.ip, tap.mac)
if config.has_option('im', 'ip'):
eip = config.get('im', 'ip')
eport = int(config.get('im', 'port'))
self.status = 'External %s %s %s %s' % (tap.ip, tap.mac, eip,
eport)
self.reconnect()
def connect(self):
if not self.client.connect():
print 'Failed to connect to the XMPP server.\nExiting...'
return False
auth = self.client.auth(self.jid.getNode(), self.password,
self.jid.getResource())
if not auth:
print 'Failed to authenticate you.\nExiting...'
return False
self.client.RegisterHandler('message', self.handle_message)
self.client.RegisterHandler('presence', self.handle_presence)
self.client.send(Presence(status=self.status))
return True
def handle_presence(self, dispatcher, presence):
type_ = presence.getType()
jid = presence.getFrom()
# When I login to the gtalk server, I have set my resource name to
# 'xtunnel'. But when I get the presence information, the server
# will add some random characters after the resource string.
# Following lines is to deal with this situation.
if not jid.getResource().startswith('xtunnel'):
return
jid.setResource('xtunnel')
if jid == self.jid:
return # TODO google kick me out?
if type_ is None: # available
if presence.getStatus():
info = presence.getStatus().split()
if len(info) == 3 and info[0] == 'Internal':
[ip, mac] = info[1:]
host = Host(jid, ip, mac)
hosts.add_host(host)
elif len(info) == 5 and info[0] == 'External':
[ip, mac, eip, eport] = info[1:]
if self.status.startswith('External') and tap.mac > mac:
host = Host(jid, ip, mac)
else:
host = Host(jid, ip, mac, eip, int(eport))
hosts.add_host(host)
elif type_ == 'unavailable':
hosts.remove_host(jid)
def handle_message(self, dispatcher, message):
if message.getType() != 'normal':
return
tap.write(base64.b64decode(message.getBody()))
def read(self):
try:
self.client.Process()
except:
self.reconnect()
def send_frame(self, jid, frame):
message = base64.b64encode(repr(frame))
try:
self.client.send(Message(to=jid, typ='normal', body=message))
except:
self.reconnect()
def fileno(self):
return self.client.Connection._sock.fileno()
def reconnect(self):
try:
time.sleep(7)
self.client.disconnect()
except:
pass
if self.debug:
self.client = Client(self.jid.getDomain())
else:
self.client = Client(self.jid.getDomain(), debug=[])
try:
if not self.connect():
self.reconnect()
except:
self.reconnect()