protected void receiveTunnelInbound(DHTTransportUDPContact originator, Map data) { log("Received tunnel inbound message from " + originator.getString()); try { punch_mon.enter(); for (int i = 0; i < oustanding_punches.size(); i++) { Object[] wait_data = (Object[]) oustanding_punches.get(i); DHTTransportContact wait_contact = (DHTTransportContact) wait_data[0]; if (originator.getAddress().getAddress().equals(wait_contact.getAddress().getAddress())) { wait_data[2] = new Integer(originator.getTransportAddress().getPort()); ((AESemaphore) wait_data[1]).release(); } } } finally { punch_mon.exit(); } }
protected void receiveBind(DHTTransportUDPContact originator, Map request, Map response) { trace("received bind request"); boolean ok = true; boolean log = true; try { server_mon.enter(); Object[] entry = (Object[]) rendezvous_bindings.get(originator.getAddress().toString()); if (entry == null) { if (rendezvous_bindings.size() == RENDEZVOUS_SERVER_MAX) { ok = false; } } else { // already present, no need to log again log = false; } if (ok) { long now = plugin_interface.getUtilities().getCurrentSystemTime(); rendezvous_bindings.put( originator.getAddress().toString(), new Object[] {originator, new Long(now)}); response.put("port", new Long(originator.getAddress().getPort())); } } finally { server_mon.exit(); } if (log) { log("Rendezvous request from " + originator.getString() + " " + (ok ? "accepted" : "denied")); } response.put("ok", new Long(ok ? 1 : 0)); }
protected void receivePunch(DHTTransportUDPContact originator, Map request, Map response) { trace("received punch request"); boolean ok = false; String target_str = new String((byte[]) request.get("target")); Object[] entry; try { server_mon.enter(); entry = (Object[]) rendezvous_bindings.get(target_str); } finally { server_mon.exit(); } if (entry != null) { DHTTransportUDPContact target = (DHTTransportUDPContact) entry[0]; Map target_client_data = sendConnect(target, originator, (Map) request.get("client_data")); if (target_client_data != null) { response.put("client_data", target_client_data); response.put("port", new Long(target.getTransportAddress().getPort())); ok = true; } } log( "Rendezvous punch request from " + originator.getString() + " to " + target_str + " " + (ok ? "initiated" : "failed")); response.put("ok", new Long(ok ? 1 : 0)); }
protected void receiveQuery(DHTTransportUDPContact originator, Map request, Map response) { trace("received query request"); InetSocketAddress address = originator.getTransportAddress(); response.put("ip", address.getAddress().getHostAddress().getBytes()); response.put("port", new Long(address.getPort())); response.put("ok", new Long(1)); }
public Map punch( String reason, DHTTransportContact _target, DHTTransportContact[] rendezvous_used, Map originator_client_data) { DHTTransportUDPContact target = (DHTTransportUDPContact) _target; try { DHTTransportContact rendezvous = getRendezvous(reason, target); if (rendezvous_used != null && rendezvous_used.length > 0) { rendezvous_used[0] = rendezvous; } if (rendezvous == null) { return (null); } Map target_client_data = sendPunch(rendezvous, target, originator_client_data, false); if (target_client_data != null) { log(" punch to " + target.getString() + " succeeded"); return (target_client_data); } } catch (Throwable e) { log(e); } log(" punch to " + target.getString() + " failed"); return (null); }
public Map punch( String reason, InetSocketAddress[] target, DHTTransportContact[] rendezvous_used, Map originator_client_data) { try { DHTTransportUDP transport = (DHTTransportUDP) dht.getTransport(); DHTTransportUDPContact contact = transport.importContact(target[0], transport.getProtocolVersion()); Map result = punch(reason, contact, rendezvous_used, originator_client_data); target[0] = contact.getTransportAddress(); return (result); } catch (Throwable e) { Debug.printStackTrace(e); return (null); } }
protected void receiveClose(DHTTransportUDPContact originator, Map request, Map response) { trace("received close request"); final DHTTransportContact current_target = rendezvous_target; if (current_target != null && Arrays.equals(current_target.getID(), originator.getID())) { new AEThread2("DHTNATPuncher:close", true) { public void run() { rendezvousFailed(current_target, true); } }.start(); } response.put("ok", new Long(1)); }
protected void receiveConnect(DHTTransportContact rendezvous, Map request, Map response) { trace("received connect request"); boolean ok = false; // ensure that we've received this from our current rendezvous node DHTTransportContact rt = rendezvous_target; if (rt != null && rt.getAddress().equals(rendezvous.getAddress())) { final DHTTransportUDPContact target = decodeContact((byte[]) request.get("origin")); if (target != null) { int transport_port = 0; Long indirect_port = (Long) request.get("port"); if (indirect_port != null) { transport_port = indirect_port.intValue(); } if (transport_port != 0) { InetSocketAddress existing_address = target.getTransportAddress(); if (transport_port != existing_address.getPort()) { target.setTransportAddress( new InetSocketAddress(existing_address.getAddress(), transport_port)); } } Map originator_client_data = (Map) request.get("client_data"); boolean no_tunnel = false; if (originator_client_data == null) { originator_client_data = new HashMap(); } else { no_tunnel = originator_client_data.get("_notunnel") != null; } if (no_tunnel) { log("Received message from " + target.getString()); } else { log("Received connect request from " + target.getString()); // ping the origin a few times to try and establish a tunnel UTTimerEvent event = timer.addPeriodicEvent( 3000, new UTTimerEventPerformer() { private int pings = 1; public void perform(UTTimerEvent ev) { if (pings > 3) { ev.cancel(); return; } pings++; if (sendTunnelInbound(target)) { ev.cancel(); } } }); if (sendTunnelInbound(target)) { event.cancel(); } } Map client_data = adapter.getClientData(target.getTransportAddress(), originator_client_data); if (client_data == null) { client_data = new HashMap(); } response.put("client_data", client_data); ok = true; } else { log("Connect request: failed to decode target"); } } else { log("Connect request from invalid rendezvous: " + rendezvous.getString()); } response.put("ok", new Long(ok ? 1 : 0)); }
protected Map sendPunch( DHTTransportContact rendezvous, final DHTTransportUDPContact target, Map originator_client_data, boolean no_tunnel) { AESemaphore wait_sem = new AESemaphore("DHTNatPuncher::sendPunch"); Object[] wait_data = new Object[] {target, wait_sem, new Integer(0)}; try { try { punch_mon.enter(); oustanding_punches.add(wait_data); } finally { punch_mon.exit(); } Map request = new HashMap(); request.put("type", new Long(RT_PUNCH_REQUEST)); request.put("target", target.getAddress().toString().getBytes()); if (originator_client_data != null) { if (no_tunnel) { originator_client_data.put("_notunnel", new Long(1)); } request.put("client_data", originator_client_data); } // for a message payload (i.e. no_tunnel) we double the initiator timeout to give // more chance for reasonable size messages to get through as they have to go through // 2 xfer processes Map response = sendRequest(rendezvous, request, no_tunnel ? TRANSFER_TIMEOUT * 2 : TRANSFER_TIMEOUT); if (response == null) { return (null); } if (((Long) response.get("type")).intValue() == RT_PUNCH_REPLY) { int result = ((Long) response.get("ok")).intValue(); trace( "received " + (no_tunnel ? "message" : "punch") + " reply: " + (result == 0 ? "failed" : "ok")); if (result == 1) { // pick up port changes from the rendezvous Long indirect_port = (Long) response.get("port"); if (indirect_port != null) { int transport_port = indirect_port.intValue(); if (transport_port != 0) { InetSocketAddress existing_address = target.getTransportAddress(); if (transport_port != existing_address.getPort()) { target.setTransportAddress( new InetSocketAddress(existing_address.getAddress(), transport_port)); } } } if (!no_tunnel) { // ping the target a few times to try and establish a tunnel UTTimerEvent event = timer.addPeriodicEvent( 3000, new UTTimerEventPerformer() { private int pings = 1; public void perform(UTTimerEvent event) { if (pings > 3) { event.cancel(); return; } pings++; if (sendTunnelOutbound(target)) { event.cancel(); } } }); if (sendTunnelOutbound(target)) { event.cancel(); } // give the other end a few seconds to kick off some tunnel events to us if (wait_sem.reserve(10000)) { event.cancel(); } } // routers often fiddle with the port when not mapped so we need to grab the right one to // use // for direct communication // first priority goes to direct tunnel messages received int transport_port = 0; try { punch_mon.enter(); transport_port = ((Integer) wait_data[2]).intValue(); } finally { punch_mon.exit(); } if (transport_port != 0) { InetSocketAddress existing_address = target.getTransportAddress(); if (transport_port != existing_address.getPort()) { target.setTransportAddress( new InetSocketAddress(existing_address.getAddress(), transport_port)); } } Map target_client_data = (Map) response.get("client_data"); if (target_client_data == null) { target_client_data = new HashMap(); } return (target_client_data); } } return (null); } catch (Throwable e) { log(e); return (null); } finally { try { punch_mon.enter(); oustanding_punches.remove(wait_data); } finally { punch_mon.exit(); } } }