Message sender and message receiver merged

opt/ahenk/base/messaging/AnonymousMessager.py

@@ -0,0 +1,132 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>
+# Author: İsmail BAŞARAN <ismail.basaran@tubitak.gov.tr> <basaran.ismaill@gmail.com>
+import slixmpp, asyncio, sys
+sys.path.append('../..')
+from slixmpp.exceptions import IqError, IqTimeout
+from base.Scope import Scope
+
+
+class AnonymousMessager(slixmpp.ClientXMPP):
+
+    def __init__(self,message,file_path):
+        # global scope of ahenk
+        scope = Scope().getInstance()
+
+        self.logger = scope.getLogger()
+        self.configuration_manager = scope.getConfigurationManager()
+        self.registration=scope.getRegistration()
+        self.event_manager = scope.getEventManager()
+
+        self.my_jid=str(self.configuration_manager.get('CONNECTION', 'host'))
+
+
+        slixmpp.ClientXMPP.__init__(self, self.my_jid,None)
+
+        self.message=None
+        self.file=None
+        self.receiver=self.configuration_manager.get('CONNECTION', 'receiverjid')+'@'+self.configuration_manager.get('CONNECTION', 'host')+'/Smack'
+
+        if file_path is not None and file_path!='':
+            self.file=open(file_path, 'rb')
+        if message is not None:
+            self.message=message
+
+        self.logger.debug('[MessageSender] XMPP Receiver parameters were set')
+
+        self.add_listeners()
+        self.register_extensions()
+
+    def add_listeners(self):
+        self.add_event_handler("session_start", self.session_start)
+        self.add_event_handler("message", self.recv_direct_message)
+        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)
+        self.logger.debug('[MessageSender] Event handlers were added')
+
+
+    def recv_direct_message(self, msg):
+        if msg['type'] in ('chat', 'normal'):
+            self.logger.debug("[MessageSender] Received message: %s -> %s" % (msg['from'], msg['body']))
+            self.disconnect()
+            self.logger.debug('[MessageSender] Disconnecting...')
+            self.logger.debug('[MessageSender] Fired event is: confirm_registration')
+            self.event_manager.fireEvent('confirm_registration',str(msg['body']))
+            ##TODO type fire -- only anonymous account can fire confirm_registration
+
+    @asyncio.coroutine
+    def session_start(self, event):
+        self.logger.debug('[MessageSender] Session was started')
+        self.get_roster()
+        self.send_presence()
+
+        if self.message is not None:
+            self.send_direct_message(self.message)
+
+        if self.file is not None:
+            self.logger.debug('[MessageSender] Sending file: '+self.file.name)
+            try:
+                self.logger.debug('[MessageSender] Handshaking for file transfering...')
+                # Open the S5B stream in which to write to.
+                proxy = yield from self['xep_0065'].handshake(self.receiver)
+                # Send the entire file.
+                self.logger.debug('[MessageSender] Started to streaming 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):
+                self.logger.error('[MessageSender] File transfer errored')
+            else:
+                self.logger.debug('[MessageSender] File transfer finished successfully')
+            finally:
+                self.file.close()
+
+    def stream_opened(self, sid):
+        self.logger.debug('[MessageSender] Stream was opened. Stream id: '+str(self.stream_id))
+        self.file = open(self.receive_file_path+self.stream_id, 'wb')
+        return self.file
+
+    def stream_data(self, data):
+        self.logger.debug('[MessageSender] Receiving file...')
+        self.file.write(data)
+
+    def stream_closed(self, exception):
+        self.logger.debug('[MessageSender] Stream was closed')
+        self.file.close()
+        self.logger.debug('[MessageSender] Disconnecting...')
+        self.disconnect()
+
+    def send_direct_message(self,msg):
+        self.logger.debug('[MessageSender] Sending message: '+msg)
+        self.send_message(mto=self.receiver,mbody=msg,mtype='normal')
+
+    def connect_to_server(self):# Connect to the XMPP server and start processing XMPP stanzas.
+        try:
+            self.logger.debug('[MessageSender] Connecting to server...')
+            self.connect()
+            self.process(forever=False)
+            self.logger.debug('[MessageSender] Connection were established successfully')
+            return True
+        except Exception as e:
+            self.logger.error('[MessageSender] Connection to server is failed! '+e)
+            return False
+
+    def register_extensions(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', {'auto_accept': True}) # SOCKS5 Bytestreams
+            self.register_plugin('xep_0047', {'auto_accept': True}) # In-band Bytestreams
+
+            self.logger.debug('[MessageSender] Extension were registered: xep_0030,xep_0045,xep_0199,xep_0065,xep_0047')
+            return True
+        except Exception as e:
+            self.logger.error('[MessageSender] Extension registration is failed!(%s)\n' % (e.errno, e.strerror))
+            return False

opt/ahenk/base/messaging/Messager.py

@@ -4,11 +4,12 @@
 # Author: İsmail BAŞARAN <ismail.basaran@tubitak.gov.tr> <basaran.ismaill@gmail.com>
 import json, os, asyncio, slixmpp, sys
 sys.path.append('../..')
+from slixmpp.exceptions import IqError, IqTimeout
 from base.Scope import Scope
 
 
 
-class MessageReceiver(slixmpp.ClientXMPP):
+class Messager(slixmpp.ClientXMPP):
 
     def __init__(self):
         # global scope of ahenk
@@ -19,34 +20,20 @@ class MessageReceiver(slixmpp.ClientXMPP):
         self.event_manger=scope.getEventManager()
         self.execution_manager=scope.getExecutionManager()
 
-        self.my_jid=self.get_jid_id()
+        self.my_jid=str(self.configuration_manager.get('CONNECTION', 'uid')+'@'+self.configuration_manager.get('CONNECTION', 'host')+'/receiver')
-        self.my_pass=self.get_password()
+        self.my_pass=str(self.configuration_manager.get('CONNECTION', 'password'))
 
         slixmpp.ClientXMPP.__init__(self, self.my_jid,self.my_pass)
 
-        self.room=None
+        self.file = None
         self.receiver=self.configuration_manager.get('CONNECTION', 'receiverjid')+'@'+self.configuration_manager.get('CONNECTION', 'host')+'/Smack'
         self.nick = self.configuration_manager.get('CONNECTION', 'nick')
         self.receive_file_path=self.configuration_manager.get('CONNECTION', 'receiveFileParam')
-        self.logger.debug('[MessageReceiver] XMPP Receiver parameters were set')
+        self.logger.debug('[Messager] XMPP Receiver parameters were set')
 
         self.register_extensions()
         self.add_listeners()
-        self.connect()
 
-    def get_jid_id(self):
-        if self.configuration_manager.get('CONNECTION', 'uid') == "" or  self.configuration_manager.get('CONNECTION', 'uid') is None:
-            self.logger.debug('[MessageReceiver] Parameters were set as anonymous account')
-            return str(self.configuration_manager.get('CONNECTION', 'host'))
-        else:
-            self.logger.debug('[MessageReceiver] Parameters were set as defined account')
-            return str(self.configuration_manager.get('CONNECTION', 'uid')+'@'+self.configuration_manager.get('CONNECTION', 'host')+'/receiver')
-
-    def get_password(self):
-        if self.configuration_manager.get('CONNECTION', 'password') == "" or  self.configuration_manager.get('CONNECTION', 'password') is None:
-            return None
-        else:
-            return str(self.configuration_manager.get('CONNECTION', 'password'))
 
     def add_listeners(self):
         self.add_event_handler("session_start", self.session_start)
@@ -60,65 +47,78 @@ class MessageReceiver(slixmpp.ClientXMPP):
         self.add_event_handler("ibb_stream_data", self.stream_data)
         self.add_event_handler("ibb_stream_end", self.stream_closed)
 
-        self.logger.debug('[MessageReceiver] Event handlers were added')
+        self.logger.debug('[Messager] Event handlers were added')
 
     def stream_opened(self, sid):
-        self.logger.debug('[MessageReceiver] Stream was opened. Stream id: '+str(self.stream_id))
+        self.logger.debug('[Messager] Stream was opened. Stream id: '+str(self.stream_id))
         self.file = open(self.receive_file_path+self.stream_id, 'wb')
         return self.file
 
     def stream_data(self, data):
-        self.logger.debug('[MessageReceiver] Receiving file...')
+        self.logger.debug('[Messager] Receiving file...')
         self.file.write(data)
 
     def stream_closed(self, exception):
-        self.logger.debug('[MessageReceiver] Stream was closed')
+        self.logger.debug('[Messager] Stream was closed')
         self.file.close()
-        self.set_file_name_to_md5()
+        self.set_file_name_md5()
 
     def session_start(self, event):
-        self.logger.debug('[MessageReceiver] Session was started')
+        self.logger.debug('[Messager] Session was started')
         self.get_roster()
         self.send_presence()
 
-    def send_direct_message(self,msg):
+    #TODO need check
-        self.logger.debug('[MessageReceiver] Sending message: '+msg)
+    def send_file(self,file_path):
-        self.send_message(mto=self.receiver,mbody=msg,mtype='normal')
+        self.file=open(file_path, 'rb')
-
+        self.logger.debug('[Messager] Sending file: '+self.file.name)
-    def invite_auto_accept(self, inv):
+        self.logger.debug('[MessageSender] Sending file: '+self.file.name)
-        self.room=inv['from']
+        try:
-        self.logger.debug('[MessageReceiver] (%s) invite is accepted' % str(self.room))
+            self.logger.debug('[MessageSender] Handshaking for file transfering...')
-        self.plugin['xep_0045'].joinMUC(self.room,self.nick,wait=True)
+            # Open the S5B stream in which to write to.
-        self.send_message(mto=self.room.bare,mbody="Hi all!",mtype='groupchat')
+            proxy = yield from self['xep_0065'].handshake(self.receiver)
-        return self.room
+            # Send the entire file.
-
+            self.logger.debug('[MessageSender] Started to streaming file...')
-    def recv_muc_message(self, msg):
+            while True:
-        if msg['mucnick'] != self.nick:
+                data = self.file.read(1048576)
-            self.logger.debug('[MessageReceiver] %s : %s' % (str(msg['from']),str(msg['body'])) )
+                if not data:
-            self.send_message(mto=msg['from'].bare,mbody="I got it, %s." % msg['mucnick'],mtype='groupchat')
+                    break
+                yield from proxy.write(data)
+            # And finally close the stream.
+            proxy.transport.write_eof()
+        except (IqError, IqTimeout):
+            self.logger.error('[MessageSender] File transfer errored')
         else:
-            self.logger.debug('[MessageReceiver] %s : %s' % (str(msg['mucnick']),str(msg['body'])))
+            self.logger.debug('[MessageSender] File transfer finished successfully')
+        finally:
+            self.file.close()
+
+    def send_direct_message(self,msg):
+        self.logger.debug('[Messager] Sending message: '+msg)
+        self.send_message(mto=self.receiver,mbody=msg,mtype='normal')
 
     def recv_direct_message(self, msg):
         if msg['type'] in ('chat', 'normal'):
             j = json.loads(str(msg['body']))
             type =j['type']
-            self.logger.debug('[MessageReceiver] Fired event is: '+type)
+            self.logger.debug('[Messager] Fired event is: '+type)
             self.event_manger.fireEvent(type,str(msg['body']).lower())
 
     def connect_to_server(self):# Connect to the XMPP server and start processing XMPP stanzas.
         try:
-            self.logger.debug('[MessageReceiver] Connecting to server as thread')
+            self.logger.debug('[Messager] Connecting to server as thread')
-            loop = asyncio.get_event_loop()
+            loop = asyncio.new_event_loop()
-            loop.run_until_complete(self.process())
+            asyncio.set_event_loop(loop)
-            self.logger.debug('[MessageReceiver] Connection were established successfully')
+            self.connect()
+            self.process()
+            self.logger.debug('[Messager] Connection were established successfully')
             return True
         except Exception as e:
-            self.logger.error('[MessageReceiver] Connection to server is failed! '+e)
+            self.logger.error('[Messager] Connection to server is failed! '+e)
             return False
 
-    def set_file_name_to_md5(self):
+    def set_file_name_md5(self):
-        self.logger.debug('[MessageReceiver] Renaming file as md5 hash')
+        self.logger.debug('[Messager] Renaming file as md5 hash')
         md5_hash=self.execution_manager.get_md5_file(self.file.name)
         os.rename(self.file.name,self.receive_file_path+md5_hash)
 
@@ -135,3 +135,20 @@ class MessageReceiver(slixmpp.ClientXMPP):
         except Exception as e:
             self.logger.error('Extension registration is failed!(%s)\n' % (e.errno, e.strerror))
             return False
+
+        """
+    def invite_auto_accept(self, inv):
+        self.room=inv['from']
+        self.logger.debug('[Messager] (%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):
+        if msg['mucnick'] != self.nick:
+            self.logger.debug('[Messager] %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:
+            self.logger.debug('[Messager] %s : %s' % (str(msg['mucnick']),str(msg['body'])))
+    """