def EditMappings(self): do_refresh = False for index in self._mappings_list_ctrl.GetAllSelected(): (description, internal_ip, internal_port, external_ip, external_port, protocol, duration) = self._mappings_list_ctrl.GetClientData(index) with ClientGUIDialogs.DialogInputUPnPMapping( self, external_port, protocol, internal_port, description, duration) as dlg: if dlg.ShowModal() == wx.ID_OK: (external_port, protocol, internal_port, description, duration) = dlg.GetInfo() HydrusNATPunch.RemoveUPnPMapping(external_port, protocol) internal_client = HydrusNATPunch.GetLocalIP() HydrusNATPunch.AddUPnPMapping(internal_client, internal_port, external_port, protocol, description, duration=duration) do_refresh = True if do_refresh: self._RefreshMappings()
def test_upnp( self ): internal_client = HydrusNATPunch.GetLocalIP() internal_port = random.randint( 1000, 1500 ) external_port = random.randint( 1000, 1500 ) description_tcp = 'hydrus test tcp' description_udp = 'hydrus test udp' HydrusNATPunch.AddUPnPMapping( internal_client, internal_port, external_port, 'TCP', description_tcp ) HydrusNATPunch.AddUPnPMapping( internal_client, internal_port, external_port, 'UDP', description_udp ) mappings = HydrusNATPunch.GetUPnPMappings() external_ip_address = mappings[0][3] mappings_without_lease_times = [ mapping[:-1] for mapping in mappings ] self.assertIn( ( description_tcp, internal_client, internal_port, external_ip_address, external_port, 'TCP' ), mappings_without_lease_times ) self.assertIn( ( description_udp, internal_client, internal_port, external_ip_address, external_port, 'UDP' ), mappings_without_lease_times ) HydrusNATPunch.RemoveUPnPMapping( external_port, 'TCP' ) HydrusNATPunch.RemoveUPnPMapping( external_port, 'UDP' ) mappings = HydrusNATPunch.GetUPnPMappings() mappings_without_lease_times = [ mapping[:-1] for mapping in mappings ] self.assertNotIn( ( description_tcp, internal_client, internal_port, external_ip_address, external_port, 'TCP' ), mappings_without_lease_times ) self.assertNotIn( ( description_udp, internal_client, internal_port, external_ip_address, external_port, 'UDP' ), mappings_without_lease_times )
def DAEMONUPnP(controller): try: local_ip = HydrusNATPunch.GetLocalIP() current_mappings = HydrusNATPunch.GetUPnPMappings() our_mappings = {(internal_client, internal_port): external_port for (description, internal_client, internal_port, external_ip_address, external_port, protocol, enabled) in current_mappings} except: return # This IGD probably doesn't support UPnP, so don't spam the user with errors they can't fix! services = HG.server_controller.GetServices() for service in services: internal_port = service.GetPort() upnp_port = service.GetUPnPPort() if (local_ip, internal_port) in our_mappings: current_external_port = our_mappings[(local_ip, internal_port)] if upnp_port is None or upnp_port != current_external_port: HydrusNATPunch.RemoveUPnPMapping(current_external_port, 'TCP') for service in services: internal_port = service.GetPort() upnp_port = service.GetUPnPPort() if upnp_port is not None and (local_ip, internal_port) not in our_mappings: external_port = upnp_port protocol = 'TCP' service_type = service.GetServiceType() description = HC.service_string_lookup[ service_type] + ' at ' + local_ip + ':' + str(internal_port) duration = 3600 HydrusNATPunch.AddUPnPMapping(local_ip, internal_port, external_port, protocol, description, duration=duration)
def DAEMONUPnP(controller): try: local_ip = HydrusNATPunch.GetLocalIP() current_mappings = HydrusNATPunch.GetUPnPMappings() our_mappings = {(internal_client, internal_port): external_port for (description, internal_client, internal_port, external_ip_address, external_port, protocol, enabled) in current_mappings} except: return # This IGD probably doesn't support UPnP, so don't spam the user with errors they can't fix! services_info = controller.Read('services_info') for (service_key, service_type, options) in services_info: internal_port = options['port'] upnp = options['upnp'] if (local_ip, internal_port) in our_mappings: current_external_port = our_mappings[(local_ip, internal_port)] if current_external_port != upnp: HydrusNATPunch.RemoveUPnPMapping(current_external_port, 'TCP') for (service_key, service_type, options) in services_info: internal_port = options['port'] upnp = options['upnp'] if upnp is not None and (local_ip, internal_port) not in our_mappings: external_port = upnp protocol = 'TCP' description = HC.service_string_lookup[ service_type] + ' at ' + local_ip + ':' + str(internal_port) duration = 3600 HydrusNATPunch.AddUPnPMapping(local_ip, internal_port, external_port, protocol, description, duration=duration)
def EventAddCustomMapping(self, event): do_refresh = False external_port = HC.DEFAULT_SERVICE_PORT protocol = 'TCP' internal_port = HC.DEFAULT_SERVICE_PORT description = 'hydrus service' duration = 0 with ClientGUIDialogs.DialogInputUPnPMapping(self, external_port, protocol, internal_port, description, duration) as dlg: if dlg.ShowModal() == wx.ID_OK: (external_port, protocol, internal_port, description, duration) = dlg.GetInfo() for (existing_description, existing_internal_ip, existing_internal_port, existing_external_ip, existing_external_port, existing_protocol, existing_lease) in self._mappings: if external_port == existing_external_port and protocol == existing_protocol: wx.MessageBox('That external port already exists!') return internal_client = HydrusNATPunch.GetLocalIP() HydrusNATPunch.AddUPnPMapping(internal_client, internal_port, external_port, protocol, description, duration=duration) do_refresh = True if do_refresh: self._RefreshMappings()
def DAEMONUPnP(controller): try: local_ip = HydrusNATPunch.GetLocalIP() current_mappings = HydrusNATPunch.GetUPnPMappings() our_mappings = {(internal_client, internal_port): external_port for (description, internal_client, internal_port, external_ip_address, external_port, protocol, enabled) in current_mappings} except: return # This IGD probably doesn't support UPnP, so don't spam the user with errors they can't fix! services = controller.services_manager.GetServices((HC.LOCAL_BOORU, )) for service in services: internal_port = service.GetPort() if (local_ip, internal_port) in our_mappings: current_external_port = our_mappings[(local_ip, internal_port)] upnp_port = service.GetUPnPPort() if upnp_port is None or current_external_port != upnp_port: HydrusNATPunch.RemoveUPnPMapping(current_external_port, 'TCP') for service in services: internal_port = service.GetPort() upnp_port = service.GetUPnPPort() if upnp_port is not None: if (local_ip, internal_port) not in our_mappings: service_type = service.GetServiceType() protocol = 'TCP' description = HC.service_string_lookup[ service_type] + ' at ' + local_ip + ':' + str( internal_port) duration = 3600 try: HydrusNATPunch.AddUPnPMapping(local_ip, internal_port, upnp_port, protocol, description, duration=duration) except HydrusExceptions.FirewallException: HydrusData.Print( 'The UPnP Daemon tried to add ' + local_ip + ':' + internal_port + '->external:' + upnp_port + ' but it failed due to router error. Please try it manually to get a full log of what happened.' ) return except: raise
def DAEMONUPnP(controller): try: local_ip = HydrusNATPunch.GetLocalIP() current_mappings = HydrusNATPunch.GetUPnPMappings() our_mappings = {(internal_client, internal_port): external_port for (description, internal_client, internal_port, external_ip_address, external_port, protocol, enabled) in current_mappings} except: return # This IGD probably doesn't support UPnP, so don't spam the user with errors they can't fix! services = controller.GetServicesManager().GetServices((HC.LOCAL_BOORU, )) for service in services: info = service.GetInfo() internal_port = info['port'] upnp = info['upnp'] if (local_ip, internal_port) in our_mappings: current_external_port = our_mappings[(local_ip, internal_port)] if upnp is None or current_external_port != upnp: HydrusNATPunch.RemoveUPnPMapping(current_external_port, 'TCP') for service in services: info = service.GetInfo() internal_port = info['port'] upnp = info['upnp'] if upnp is not None: if (local_ip, internal_port) not in our_mappings: service_type = service.GetServiceType() external_port = upnp protocol = 'TCP' description = HC.service_string_lookup[ service_type] + ' at ' + local_ip + ':' + str( internal_port) duration = 3600 HydrusNATPunch.AddUPnPMapping(local_ip, internal_port, external_port, protocol, description, duration=duration)