Exemplo n.º 1
0
def tcp_recieveItem(csock, caddr,dispatcher,keys):
    try:
#        if caddr[0]=='83.201.26.105':
#            print 'start-1'
        SizeStruct = struct.Struct( "!I" )          
        size_data = csock.recv(4)
        size, = struct.unpack("!I",size_data)
        result2 =''        
        #@Ali
        start=0
        packetSize=5000
        dieCounter=0
        while len(result2)!=size:
            if (len(result2)+packetSize) < size:
                packet=csock.recv(packetSize) # recieve 100 KB
                
                result2=result2+packet
                start=start+len(packet)
            else:
                if dieCounter>=100:
                    print 'error: unable to recieve all bytes from '+str(caddr)
                    print 'error: bytes to send = '+str(size)
                    print 'error: bytes recieved = '+str(len(result2))
                    csock.close()
                    return
                packet=csock.recv(size-start)
                
                dieCounter+=1
                result2=result2+packet
                start=start+len(packet)
                if len(result2)!=size:
                    time.sleep(1)
        try:
            dec_data = crypto.decrypt( keys, result2 )
        except:
            print 'Communication Manager: error in decrypt'
            print traceback.print_exc()
            return
                     
        try:            
            msg = pickle.loads( dec_data )
        except:
            print 'TCP Thread Error: Cant load Pickle'
            print type(dec_data)
            print traceback.print_exc()
            return
                
        msg.type = "R" + msg.type[1:]
        try:
            msg.fromip = msg.myIP #@Ali
        except:
            msg.fromip = (caddr[0],'','')
        msg.isSSLPacket=False
        dispatcher.add( msg )

        csock.close()
    except socket.error, e:
        # A socket error
        print 'Recieve Socket Error'
        print socket.error
Exemplo n.º 2
0
def udp_serverthread( dispatcher, udpSSock ):
    udpSSock.bind( ("0.0.0.0", int(myPubaddr[2]))  )#@Ali
    logging.info( "UDP server listening on port {0}...".format( myPubaddr[2] ) )#@Ali
    f = open(constants.nid_filename, 'r')
    nodeid = f.readline()
    f.close()
    keys = crypto.loadKeyPrivate(nodeid+"_N_")
    while True:
        try:
            data, addr = udpSSock.recvfrom( 30000 )
            try:
                dec_data = crypto.decrypt( keys, data ) #esko: p2p message decryption 
                
            except:
                print 'Communication Manager: error in p2p decrypt'
                print
                print traceback.print_exc() 
            
            m = pickle.loads( dec_data )
            
            m.isSSLPacket=False
            job = Job( "R_P2P_" + m.type, m )#@Ali addr contains the IP and the random port, so the sender attach the MYIP field with IP and its listening ports
            dispatcher.add( job )
            logging.info( "Received datagram from {0}".format( addr ) )
        except socket.error, e:
    # A socket error
            pass
        except IOError, e:
            if e.errno == errno.EPIPE:
                print "broken pipe"
 def udpMsg_handler(self,item,id):
     try:
         #print "UDP/SSL handler"
         dec_data = b64decode(crypto.decrypt( self.keys, item ))
         m = pickle.loads( dec_data )
         m.myIP=(id,str(self.hostSSLport),str(self.hostSSLport))
         m.isSSLPacket=True
         m.isClient=self.isClient
         job = Job( "R_P2P_" + m.type, m )
         #print m
         self.dispatcher.add( job )
     except:
         print 'Error in upd packet pickle load'
 def tcpMsg_handler(self,item,id):
     try:
         dec_data = b64decode(crypto.decrypt( self.keys, item ))
     except:
         print 'SSL TCP Handler: error in decrypt'
         print traceback.print_exc()
         return          
     try:            
         msg = pickle.loads( dec_data )
     except:
         print 'SSL TCP Handler: Cant load Pickle'
         print traceback.print_exc()
         return
     msg.type = "R" + msg.type[1:]
     
     msg.fromip = (id,str(self.hostSSLport),str(self.hostSSLport))
     msg.myIP=(id,str(self.hostSSLport),str(self.hostSSLport))
     msg.isSSLPacket=True
     self.dispatcher.add( msg )