def connect_to_sm(self, ip, port, server_conn=False): """ connect to specific IP:PORT """ logger.debug( 'Try to connect to Site Manager and establish DTN connection') conn_send = DTN._tcp_connect(ip, port) if conn_send is not None: # send information conn_send.send('%s %s\n' % (self.sh, self.target)) data = conn_send.recv(1024) port_recv = 0 if data == EXIST_ERR: logger.debug( 'DTN connection already exists between these two Site Managers' ) return False sh = data.split()[0] port_recv = int(data.split()[1]) tar = ' '.join(data.split()[2:]) logger.debug('SH: %s, Port: %d, Target: %s' % (sh, port_recv, tar)) # check if site manager is connected if self.dtn.has_key(sh) and self.dtn[sh] is not None: conn_send.send(EXIST_ERR) logger.debug( 'DTN connection already exists between these two Site Managers' ) return False else: conn_send.send(SUCCESS_INFO) # connect to ip using another port conn_recv = DTN._tcp_connect(ip, port_recv) if conn_recv is not None: # Ready dtn_conn = DTNConnection(conn_send, conn_recv, self.sh, sh, tar, self, server_conn=server_conn) self.dtn[sh] = dtn_conn if not self.last_hash.has_key(sh): self.last_hash[sh] = '' logger.info('New DTN connection established') dtn_conn.start() return True return False
def handle_dtn_listen(self, s): logger.info('DTN connection request') conn_recv, remote = s.accept() data = conn_recv.recv(1024) sh = data.split()[0] tar = ' '.join(data.split()[1:]) logger.debug('SH: %s, Target: %s' % (sh, tar)) if self.dtn.has_key(sh): logger.debug(self.dtn[sh]) if self.dtn.has_key(sh) and self.dtn[sh] is not None: conn_recv.send(EXIST_ERR) logger.debug( 'DTN connection already exists between these two Site Managers' ) return False # Generate a random port for listening random_n = 1 listener = None while True: try: listener = DTN._tcp_listen(self.my_ip, self.dtn_port + random_n) except: random_n += 1 continue break conn_recv.send('%s %d %s' % (self.sh, self.dtn_port + random_n, self.target)) data = conn_recv.recv(1024) if data == EXIST_ERR: logger.debug( 'DTN connection already exists between these two Site Managers' ) return False # wait for connection listener.settimeout(5) try: conn_send, remote = listener.accept() except socket.timeout: logger.debug('timeout') return False # remote[0] is IP # remote[1] is PORT #conn_send = DTN._tcp_connect(remote[0], port) if conn_send is not None: # send my SH info #conn_send.send('%s %s' % (self.sh, self.target)) #data = conn_send.recv(1024) #if data == EXIST_ERR: #logger.debug('DTN connection already exists between these two Site Managers') #return #else: #logger.debug('Good') # Ready dtn_conn = DTNConnection(conn_send, conn_recv, self.sh, sh, tar, self) self.dtn[sh] = dtn_conn if not self.last_hash.has_key(sh): self.last_hash[sh] = '' #if sh == SERVER_SH_INFO: #self.server_connected = True logger.info('New DTN connection established') dtn_conn.start() return True return False