2016-02-16 17:50:37 +02:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
# -*- coding: utf-8 -*-
|
2016-02-21 01:57:26 +02:00
|
|
|
# Author: Volkan Şahin <basaran.ismaill@gmail.com>
|
|
|
|
# Author: İsmail BAŞARAN <ismail.basaran@tubitak.gov.tr> <basaran.ismaill@gmail.com>
|
2016-02-16 17:50:37 +02:00
|
|
|
|
|
|
|
import configparser
|
|
|
|
import slixmpp
|
|
|
|
from slixmpp.exceptions import IqError, IqTimeout
|
2016-02-21 03:24:54 +02:00
|
|
|
from ahenkd import AhenkDeamon
|
2016-02-16 17:50:37 +02:00
|
|
|
|
|
|
|
"""
|
|
|
|
--fetch parameters of connection from conf file
|
|
|
|
--connect xmpp
|
|
|
|
--send direct message
|
|
|
|
--receive direct message
|
|
|
|
--send muc message
|
|
|
|
--receive muc message
|
|
|
|
--listen to muc invites
|
|
|
|
--auto accept muc invites
|
|
|
|
--send auto reply to muc messages
|
|
|
|
--receive file (0065)
|
|
|
|
--send file (0065)
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
class Messaging(slixmpp.ClientXMPP):
|
|
|
|
|
2016-02-21 03:24:54 +02:00
|
|
|
def __init__(self):
|
|
|
|
|
|
|
|
# global scope of ahenk
|
|
|
|
scope = AhenkDeamon.scope()
|
2016-02-16 17:50:37 +02:00
|
|
|
|
2016-02-21 01:57:26 +02:00
|
|
|
# logger comes from ahenk deamon
|
2016-02-21 03:24:54 +02:00
|
|
|
self.logger = scope.getLogger()
|
2016-02-21 01:57:26 +02:00
|
|
|
|
|
|
|
# configurationManager comes from ahenk deamon
|
2016-02-21 03:24:54 +02:00
|
|
|
self.configurationManager = scope.getConfigurationManager()
|
2016-02-16 17:50:37 +02:00
|
|
|
|
|
|
|
#set parameters
|
2016-02-21 01:57:26 +02:00
|
|
|
slixmpp.ClientXMPP.__init__(self, self.configurationManager.get('CONNECTION', 'jid'), self.configurationManager.get('Connection_Param', 'password'))
|
|
|
|
self.nick = self.configurationManager.get('CONNECTION', 'nick')
|
|
|
|
self.receiver=self.configurationManager.get('CONNECTION','receiverJid')
|
|
|
|
self.sendfile=open(self.configurationManager.get('CONNECTION','sendFilePath'), 'rb')
|
|
|
|
self.receivefile=self.configurationManager.get('CONNECTION', 'receiveFileParam')
|
|
|
|
self.logger.info('Parameters were established')
|
2016-02-16 17:50:37 +02:00
|
|
|
|
|
|
|
self.add_event_handler("session_start", self.start)
|
|
|
|
self.room=self.add_event_handler("groupchat_invite", self.invite_auto_accept)
|
|
|
|
|
|
|
|
self.add_listeners()
|
|
|
|
#!!! you have to use modified slixmpp for file transfering
|
|
|
|
#self.send_file()
|
|
|
|
|
|
|
|
def add_listeners(self):
|
|
|
|
|
|
|
|
self.add_event_handler("groupchat_message", self.recv_muc_message)
|
|
|
|
self.add_event_handler("message", self.recv_direct_message)
|
|
|
|
|
|
|
|
#file_listeners
|
|
|
|
#self.add_event_handler("socks5_connected", self.stream_opened)
|
|
|
|
#self.add_event_handler("socks5_data", self.stream_data)
|
|
|
|
#self.add_event_handler("socks5_closed", self.stream_closed)
|
|
|
|
|
2016-02-21 01:57:26 +02:00
|
|
|
self.logger.info('Listeners were added')
|
2016-02-16 17:50:37 +02:00
|
|
|
|
|
|
|
|
|
|
|
def stream_opened(self, sid):
|
2016-02-21 01:57:26 +02:00
|
|
|
self.logger.info('Stream opened. %s', sid)
|
2016-02-16 17:50:37 +02:00
|
|
|
return open(self.receivefile, 'wb')
|
|
|
|
|
|
|
|
def stream_data(self, data):
|
2016-02-21 01:57:26 +02:00
|
|
|
self.logger.info('Stream data.')
|
2016-02-16 17:50:37 +02:00
|
|
|
self.file.write(data)
|
|
|
|
|
|
|
|
def stream_closed(self, exception):
|
2016-02-21 01:57:26 +02:00
|
|
|
self.logger.info('Stream closed. %s', exception)
|
2016-02-16 17:50:37 +02:00
|
|
|
self.file.close()
|
|
|
|
#self.disconnect()
|
|
|
|
|
|
|
|
|
|
|
|
def send_file(self):
|
|
|
|
try:
|
|
|
|
# Open the S5B stream in which to write to.
|
|
|
|
proxy = yield from self['xep_0065'].handshake(self.receiver)
|
|
|
|
|
|
|
|
# Send the entire file.
|
|
|
|
while True:
|
|
|
|
data = self.file.read(1048576)
|
|
|
|
if not data:
|
|
|
|
break
|
|
|
|
yield from proxy.write(data)
|
|
|
|
# And finally close the stream.
|
|
|
|
proxy.transport.write_eof()
|
|
|
|
except (IqError, IqTimeout):
|
|
|
|
print('File transfer errored')
|
|
|
|
else:
|
|
|
|
print('File transfer finished')
|
|
|
|
finally:
|
|
|
|
self.file.close()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def start(self, event):
|
|
|
|
self.get_roster()
|
|
|
|
self.send_presence()
|
|
|
|
|
|
|
|
|
|
|
|
def invite_auto_accept(self, inv):
|
|
|
|
|
|
|
|
self.room=inv['from']
|
|
|
|
print("(%s) invite is accepted" % str(self.room))
|
|
|
|
self.plugin['xep_0045'].joinMUC(self.room,self.nick,wait=True)
|
|
|
|
self.send_message(mto=self.room.bare,mbody="Hi all!",mtype='groupchat')
|
|
|
|
return self.room
|
|
|
|
|
|
|
|
def recv_muc_message(self, msg):#auto reply
|
|
|
|
|
|
|
|
if msg['mucnick'] != self.nick:
|
|
|
|
print("%s : %s" % (str(msg['from']),str(msg['body'])) )
|
|
|
|
self.send_message(mto=msg['from'].bare,mbody="I got it, %s." % msg['mucnick'],mtype='groupchat')
|
|
|
|
else:
|
|
|
|
print("%s : %s" % (str(msg['mucnick']),str(msg['body'])))
|
|
|
|
|
|
|
|
def send_direct_message(self,msg):
|
|
|
|
self.send_message(mto=self.receiver,mbody=msg,mtype='chat')
|
|
|
|
|
|
|
|
def recv_direct_message(self, msg):
|
|
|
|
if msg['type'] in ('chat', 'normal'):
|
|
|
|
print ("%s : %s" % (msg['from'], msg['body']))
|
|
|
|
|
2016-02-21 01:57:26 +02:00
|
|
|
def connectToServer(self):
|
|
|
|
try:
|
|
|
|
self.register_plugin('xep_0030') # Service Discovery
|
|
|
|
self.register_plugin('xep_0045') # Multi-User Chat
|
|
|
|
self.register_plugin('xep_0199') # XMPP Ping
|
|
|
|
self.register_plugin('xep_0065') # SOCKS5 Bytestreams
|
|
|
|
|
|
|
|
self.logger.info('Plugins were registered: xep_0030,xep_0045,xep_0199,xep_0065')
|
|
|
|
|
|
|
|
self.connect()
|
|
|
|
self.process()
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
|
|
self.logger.error('Connection to server failed!')
|
|
|
|
return False
|
2016-02-16 17:50:37 +02:00
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
2016-02-21 01:57:26 +02:00
|
|
|
xmpp = Messaging(None,None)
|
2016-02-16 17:50:37 +02:00
|
|
|
xmpp.register_plugin('xep_0030') # Service Discovery
|
|
|
|
xmpp.register_plugin('xep_0045') # Multi-User Chat
|
|
|
|
xmpp.register_plugin('xep_0199') # XMPP Ping
|
|
|
|
xmpp.register_plugin('xep_0065') # SOCKS5 Bytestreams
|
|
|
|
|
2016-02-21 01:57:26 +02:00
|
|
|
self.logger.info('Plugins were registered: xep_0030,xep_0045,xep_0199,xep_0065')
|
2016-02-16 17:50:37 +02:00
|
|
|
|
|
|
|
# Connect to the XMPP server and start processing XMPP stanzas.
|
|
|
|
xmpp.connect()
|
2016-02-21 01:57:26 +02:00
|
|
|
xmpp.process()
|