DHTNATPuncherImplpublic class DHTNATPuncherImpl extends Object implements DHTNATPuncher
Fields Summary |
---|
private static boolean | TESTING | private static boolean | TRACE | private static final int | RT_BIND_REQUEST | private static final int | RT_BIND_REPLY | private static final int | RT_PUNCH_REQUEST | private static final int | RT_PUNCH_REPLY | private static final int | RT_CONNECT_REQUEST | private static final int | RT_CONNECT_REPLY | private static final int | RT_TUNNEL_INBOUND | private static final int | RT_TUNNEL_OUTBOUND | private static final int | RT_QUERY_REQUEST | private static final int | RT_QUERY_REPLY | private static final int | RT_CLOSE_REQUEST | private static final int | RT_CLOSE_REPLY | private static final int | RESP_OK | private static final int | RESP_NOT_OK | private static final int | RESP_FAILED | private static byte[] | transfer_handler_key | private boolean | started | private DHTNATPuncherAdapter | adapter | private com.aelitis.azureus.core.dht.DHT | dht | private com.aelitis.azureus.core.dht.DHTLogger | logger | private org.gudy.azureus2.plugins.PluginInterface | plugin_interface | private Formatters | formatters | private UTTimer | timer | private static final int | REPUBLISH_TIME_MIN | private static final int | TRANSFER_TIMEOUT | private static final int | RENDEZVOUS_LOOKUP_TIMEOUT | private static final int | TUNNEL_TIMEOUT | private static final int | RENDEZVOUS_SERVER_MAX | private static final int | RENDEZVOUS_SERVER_TIMEOUT | private static final int | RENDEZVOUS_CLIENT_PING_PERIOD | private static final int | RENDEZVOUS_PING_FAIL_LIMIT | private Monitor | server_mon | private Map | rendezvous_bindings | private long | last_publish | private Monitor | pub_mon | private boolean | publish_in_progress | private volatile com.aelitis.azureus.core.dht.transport.DHTTransportContact | rendezvous_local_contact | private volatile com.aelitis.azureus.core.dht.transport.DHTTransportContact | rendezvous_target | private volatile com.aelitis.azureus.core.dht.transport.DHTTransportContact | last_ok_rendezvous | private static final int | FAILED_RENDEZVOUS_HISTORY_MAX | private Map | failed_rendezvous | private boolean | rendezvous_thread_running | private Map | explicit_rendezvous_map | private Monitor | punch_mon | private List | oustanding_punches | private com.aelitis.azureus.core.dht.transport.DHTTransportContact | current_local | private com.aelitis.azureus.core.dht.transport.DHTTransportContact | current_target | private int | rendevzous_fail_count |
Constructors Summary |
---|
public DHTNATPuncherImpl(DHTNATPuncherAdapter _adapter, com.aelitis.azureus.core.dht.DHT _dht)
adapter = _adapter;
dht = _dht;
logger = dht.getLogger();
plugin_interface = dht.getLogger().getPluginInterface();
formatters = plugin_interface.getUtilities().getFormatters();
pub_mon = plugin_interface.getUtilities().getMonitor();
server_mon = plugin_interface.getUtilities().getMonitor();
punch_mon = plugin_interface.getUtilities().getMonitor();
timer = plugin_interface.getUtilities().createTimer(
"DHTNATPuncher:refresher", true );
|
Methods Summary |
---|
public boolean | active()
return( rendezvous_local_contact != null );
| protected com.aelitis.azureus.core.dht.transport.udp.DHTTransportUDPContact | decodeContact(byte[] bytes)
try{
ByteArrayInputStream bais = new ByteArrayInputStream( bytes );
DataInputStream dis = new DataInputStream( bais );
return((DHTTransportUDPContact)dht.getTransport().importContact( dis ));
}catch( Throwable e ){
log(e);
return( null );
}
| public void | destroy()
try{
server_mon.enter();
Iterator it = rendezvous_bindings.values().iterator();
while( it.hasNext()){
Object[] entry = (Object[])it.next();
final DHTTransportUDPContact contact = (DHTTransportUDPContact)entry[0];
new AEThread( "DHTNATPuncher:destroy", true )
{
public void
runSupport()
{
sendClose( contact );
}
}.start();
}
}catch( Throwable e ){
log( e );
}finally{
server_mon.exit();
}
| protected byte[] | encodeContact(com.aelitis.azureus.core.dht.transport.DHTTransportContact contact)
try{
ByteArrayOutputStream baos = new ByteArrayOutputStream();
DataOutputStream dos = new DataOutputStream(baos);
contact.exportContact( dos );
dos.close();
return( baos.toByteArray());
}catch( Throwable e ){
log( e );
return( null );
}
| protected byte[] | encodePublishValue(com.aelitis.azureus.core.dht.transport.DHTTransportContact contact)
try{
ByteArrayOutputStream baos = new ByteArrayOutputStream();
DataOutputStream dos = new DataOutputStream(baos);
dos.writeByte( 0 ); // version
contact.exportContact( dos );
dos.close();
return( baos.toByteArray());
}catch( Throwable e ){
log( e );
return( new byte[0]);
}
| protected byte[] | getPublishKey(com.aelitis.azureus.core.dht.transport.DHTTransportContact contact)
byte[] id = contact.getID();
byte[] suffix = ":DHTNATPuncher".getBytes();
byte[] res = new byte[id.length + suffix.length];
System.arraycopy( id, 0, res, 0, id.length );
System.arraycopy( suffix, 0, res, id.length, suffix.length );
return( res );
| protected com.aelitis.azureus.core.dht.transport.DHTTransportContact | getRendezvous(java.lang.String reason, com.aelitis.azureus.core.dht.transport.DHTTransportContact target)
DHTTransportContact explicit = (DHTTransportContact)explicit_rendezvous_map.get( target.getAddress());
if ( explicit != null ){
return( explicit );
}
byte[] key = getPublishKey( target );
final DHTTransportValue[] result_value = {null};
final Semaphore sem = plugin_interface.getUtilities().getSemaphore();
dht.get( key,
reason + ": lookup for '" + target.getString() + "'",
(byte)0,
1,
RENDEZVOUS_LOOKUP_TIMEOUT,
false, true,
new DHTOperationAdapter()
{
public void
read(
DHTTransportContact contact,
DHTTransportValue value )
{
result_value[0] = value;
sem.release();
}
public void
complete(
boolean timeout )
{
sem.release();
}
});
sem.reserve();
DHTTransportContact result = null;
if ( result_value[0] != null ){
byte[] bytes = result_value[0].getValue();
try{
ByteArrayInputStream bais = new ByteArrayInputStream( bytes );
DataInputStream dis = new DataInputStream( bais );
byte version = dis.readByte();
if ( version != 0 ){
throw( new Exception( "Unsupported rendezvous version '" + version + "'" ));
}
result = dht.getTransport().importContact( dis );
}catch( Throwable e ){
log(e);
}
}
log( "Lookup of rendezvous for " + target.getString() + " -> " + ( result==null?"None":result.getString()));
return( result );
| protected void | log(java.lang.String str)
if ( TRACE ){
System.out.println( str );
}
logger.log( "NATPuncher: " + str );
| protected void | log(java.lang.Throwable e)
if ( TRACE ){
e.printStackTrace();
}
logger.log( "NATPuncher: error occurred" );
logger.log(e);
| public boolean | operational()
DHTTransportContact ok = last_ok_rendezvous;
if ( ok != null && ok == rendezvous_target ){
return( true );
}
return( false );
| protected void | publish(boolean force)
long now = plugin_interface.getUtilities().getCurrentSystemTime();
if ( now < last_publish && !force ){
last_publish = now;
}else{
if ( force || now - last_publish >= REPUBLISH_TIME_MIN ){
last_publish = now;
plugin_interface.getUtilities().createThread(
"DHTNATPuncher:publisher",
new Runnable()
{
public void
run()
{
try{
pub_mon.enter();
if ( publish_in_progress ){
return;
}
publish_in_progress = true;
}finally{
pub_mon.exit();
}
try{
publishSupport();
}finally{
try{
pub_mon.enter();
publish_in_progress = false;
}finally{
pub_mon.exit();
}
}
}
});
}
}
| protected void | publishSupport()
DHTTransport transport = dht.getTransport();
if ( TESTING || !transport.isReachable() ){
DHTTransportContact local_contact = transport.getLocalContact();
// see if the rendezvous has failed and therefore we are required to find a new one
boolean force =
rendezvous_target != null &&
failed_rendezvous.containsKey( rendezvous_target.getAddress());
if ( rendezvous_local_contact != null && !force ){
if ( local_contact.getAddress().equals( rendezvous_local_contact.getAddress())){
// already running for the current local contact
return;
}
}
DHTTransportContact explicit = (DHTTransportContact)explicit_rendezvous_map.get( local_contact.getAddress());
if ( explicit != null ){
try{
pub_mon.enter();
rendezvous_local_contact = local_contact;
rendezvous_target = explicit;
runRendezvous();
}finally{
pub_mon.exit();
}
}else{
final DHTTransportContact[] new_rendezvous_target = { null };
DHTTransportContact[] reachables = dht.getTransport().getReachableContacts();
int reachables_tried = 0;
int reachables_skipped = 0;
final Semaphore sem = plugin_interface.getUtilities().getSemaphore();
for (int i=0;i<reachables.length;i++){
DHTTransportContact contact = reachables[i];
try{
pub_mon.enter();
// see if we've found a good one yet
if ( new_rendezvous_target[0] != null ){
break;
}
// skip any known bad ones
if ( failed_rendezvous.containsKey( contact.getAddress())){
reachables_skipped++;
sem.release();
continue;
}
}finally{
pub_mon.exit();
}
if ( i > 0 ){
try{
Thread.sleep( 1000 );
}catch( Throwable e ){
}
}
reachables_tried++;
contact.sendPing(
new DHTTransportReplyHandlerAdapter()
{
public void
pingReply(
DHTTransportContact ok_contact )
{
trace( "Punch:" + ok_contact.getString() + " OK" );
try{
pub_mon.enter();
if ( new_rendezvous_target[0] == null ){
new_rendezvous_target[0] = ok_contact;
}
}finally{
pub_mon.exit();
sem.release();
}
}
public void
failed(
DHTTransportContact failed_contact,
Throwable e )
{
try{
trace( "Punch:" + failed_contact.getString() + " Failed" );
}finally{
sem.release();
}
}
});
}
for (int i=0;i<reachables.length;i++){
sem.reserve();
try{
pub_mon.enter();
if ( new_rendezvous_target[0] != null ){
rendezvous_target = new_rendezvous_target[0];
rendezvous_local_contact = local_contact;
log( "Rendezvous found: " + rendezvous_local_contact.getString() + " -> " + rendezvous_target.getString());
runRendezvous();
break;
}
}finally{
pub_mon.exit();
}
}
if ( new_rendezvous_target[0] == null ){
log( "No rendezvous found: candidates=" + reachables.length +",tried="+ reachables_tried+",skipped=" +reachables_skipped );
try{
pub_mon.enter();
rendezvous_local_contact = null;
rendezvous_target = null;
}finally{
pub_mon.exit();
}
}
}
}else{
try{
pub_mon.enter();
rendezvous_local_contact = null;
rendezvous_target = null;
}finally{
pub_mon.exit();
}
}
| public java.util.Map | punch(java.lang.String reason, java.net.InetSocketAddress[] target, com.aelitis.azureus.core.dht.transport.DHTTransportContact[] rendezvous_used, java.util.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 );
}
| public java.util.Map | punch(java.lang.String reason, com.aelitis.azureus.core.dht.transport.DHTTransportContact _target, com.aelitis.azureus.core.dht.transport.DHTTransportContact[] rendezvous_used, java.util.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 );
| protected void | receiveBind(com.aelitis.azureus.core.dht.transport.udp.DHTTransportUDPContact originator, java.util.Map request, java.util.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 | receiveClose(com.aelitis.azureus.core.dht.transport.udp.DHTTransportUDPContact originator, java.util.Map request, java.util.Map response)
trace( "received close request" );
final DHTTransportContact current_target = rendezvous_target;
if ( current_target != null && Arrays.equals( current_target.getID(), originator.getID())){
new AEThread( "DHTNATPuncher:close", true )
{
public void
runSupport()
{
rendezvousFailed( current_target, true );
}
}.start();
}
response.put( "ok", new Long(1));
| protected void | receiveConnect(com.aelitis.azureus.core.dht.transport.DHTTransportContact rendezvous, java.util.Map request, java.util.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 void | receivePunch(com.aelitis.azureus.core.dht.transport.udp.DHTTransportUDPContact originator, java.util.Map request, java.util.Map response)
trace( "received punch request" );
boolean ok = false;
try{
server_mon.enter();
String target_str = new String((byte[])request.get( "target" ));
Object[] entry = (Object[])rendezvous_bindings.get( target_str );
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"));
}finally{
server_mon.exit();
}
response.put( "ok", new Long(ok?1:0));
| protected void | receiveQuery(com.aelitis.azureus.core.dht.transport.udp.DHTTransportUDPContact originator, java.util.Map request, java.util.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));
| protected byte[] | receiveRequest(com.aelitis.azureus.core.dht.transport.udp.DHTTransportUDPContact originator, byte[] data)
try{
Map res = receiveRequest( originator, formatters.bDecode( data ));
if ( res == null ){
return( null );
}
return( formatters.bEncode( res ));
}catch( Throwable e ){
log(e);
return( null );
}
| protected java.util.Map | receiveRequest(com.aelitis.azureus.core.dht.transport.udp.DHTTransportUDPContact originator, java.util.Map data)
int type = ((Long)data.get("type")).intValue();
Map response = new HashMap();
switch( type ){
case RT_BIND_REQUEST:
{
response.put( "type", new Long( RT_BIND_REPLY ));
receiveBind( originator, data, response );
break;
}
case RT_CLOSE_REQUEST:
{
response.put( "type", new Long( RT_CLOSE_REPLY ));
receiveClose( originator, data, response );
break;
}
case RT_QUERY_REQUEST:
{
response.put( "type", new Long( RT_QUERY_REPLY ));
receiveQuery( originator, data, response );
break;
}
case RT_PUNCH_REQUEST:
{
response.put( "type", new Long( RT_PUNCH_REPLY ));
receivePunch( originator, data, response );
break;
}
case RT_CONNECT_REQUEST:
{
response.put( "type", new Long( RT_CONNECT_REPLY ));
receiveConnect( originator, data, response );
break;
}
case RT_TUNNEL_INBOUND:
{
receiveTunnelInbound( originator, data );
response = null;
break;
}
case RT_TUNNEL_OUTBOUND:
{
receiveTunnelOutbound( originator, data );
response = null;
break;
}
default:
{
response = null;
break;
}
}
return( response );
| protected void | receiveTunnelInbound(com.aelitis.azureus.core.dht.transport.udp.DHTTransportUDPContact originator, java.util.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 | receiveTunnelOutbound(com.aelitis.azureus.core.dht.transport.DHTTransportContact originator, java.util.Map data)
log( "Received tunnel outbound message from " + originator.getString());
| protected void | rendezvousFailed(com.aelitis.azureus.core.dht.transport.DHTTransportContact current_target, boolean tidy)
log( "Rendezvous " + (tidy?"closed":"failed") + ": " + current_target.getString());
try{
pub_mon.enter();
failed_rendezvous.put( current_target.getAddress(), "" );
}finally{
pub_mon.exit();
}
publish( true );
| protected void | runRendezvous()
try{
pub_mon.enter();
if ( !rendezvous_thread_running ){
rendezvous_thread_running = true;
plugin_interface.getUtilities().createThread(
"DHTNatPuncher:rendevzous",
new Runnable()
{
public void
run()
{
runRendezvousSupport();
}
});
}
}finally{
pub_mon.exit();
}
| protected void | runRendezvousSupport()
SimpleTimer.addPeriodicEvent(
"DHTNAT:cp",
RENDEZVOUS_CLIENT_PING_PERIOD,
new TimerEventPerformer() {
public void perform( TimerEvent ev ) {
try{
DHTTransportContact latest_local;
DHTTransportContact latest_target;
try{
pub_mon.enter();
latest_local = rendezvous_local_contact;
latest_target = rendezvous_target;
}finally{
pub_mon.exit();
}
if ( current_local != null || latest_local != null ){
// one's not null, worthwhile further investigation
if ( current_local != latest_local ){
// local has changed, remove existing publish
if ( current_local != null ){
log( "Removing publish for " + current_local.getString() + " -> " + current_target.getString());
dht.remove(
getPublishKey( current_local ),
"DHTNatPuncher: removal of publish",
new DHTOperationListener()
{
public void
searching(
DHTTransportContact contact,
int level,
int active_searches )
{}
public void
found(
DHTTransportContact contact )
{}
public void
diversified()
{
}
public void
read(
DHTTransportContact contact,
DHTTransportValue value )
{}
public void
wrote(
DHTTransportContact contact,
DHTTransportValue value )
{}
public void
complete(
boolean timeout )
{}
});
}
if ( latest_local != null ){
log( "Adding publish for " + latest_local.getString() + " -> " + latest_target.getString());
rendevzous_fail_count = RENDEZVOUS_PING_FAIL_LIMIT - 2; // only 2 attempts to start with
dht.put(
getPublishKey( latest_local ),
"NAT Traversal: rendezvous publish",
encodePublishValue( latest_target ),
DHT.FLAG_SINGLE_VALUE,
new DHTOperationListener()
{
public void
searching(
DHTTransportContact contact,
int level,
int active_searches )
{}
public void
found(
DHTTransportContact contact )
{}
public void
diversified()
{
}
public void
read(
DHTTransportContact contact,
DHTTransportValue value )
{}
public void
wrote(
DHTTransportContact contact,
DHTTransportValue value )
{}
public void
complete(
boolean timeout )
{}
});
}
}else if ( current_target != latest_target ){
// here current_local == latest_local and neither is null!
// target changed, update publish
log( "Updating publish for " + latest_local.getString() + " -> " + latest_target.getString());
rendevzous_fail_count = RENDEZVOUS_PING_FAIL_LIMIT - 2; // only 2 attempts to start with
dht.put(
getPublishKey( latest_local ),
"DHTNatPuncher: update publish",
encodePublishValue( latest_target ),
DHT.FLAG_SINGLE_VALUE,
new DHTOperationListener()
{
public void
searching(
DHTTransportContact contact,
int level,
int active_searches )
{}
public void
found(
DHTTransportContact contact )
{}
public void
diversified()
{
}
public void
read(
DHTTransportContact contact,
DHTTransportValue value )
{}
public void
wrote(
DHTTransportContact contact,
DHTTransportValue value )
{}
public void
complete(
boolean timeout )
{}
});
}
}
current_local = latest_local;
current_target = latest_target;
if ( current_target != null ){
int bind_result = sendBind( current_target );
if ( bind_result == RESP_OK ){
trace( "Rendezvous:" + current_target.getString() + " OK" );
rendevzous_fail_count = 0;
if ( last_ok_rendezvous != current_target ){
last_ok_rendezvous = current_target;
log( "Rendezvous " + latest_target.getString() + " operational" );
}
}else{
if ( bind_result == RESP_NOT_OK ){
// denied access
rendevzous_fail_count = RENDEZVOUS_PING_FAIL_LIMIT;
}else{
rendevzous_fail_count++;
}
if ( rendevzous_fail_count == RENDEZVOUS_PING_FAIL_LIMIT ){
rendezvousFailed( current_target, false );
}
}
}
}catch( Throwable e ){
log(e);
}
}
}
);
| protected int | sendBind(com.aelitis.azureus.core.dht.transport.DHTTransportContact target)
try{
Map request = new HashMap();
request.put("type", new Long( RT_BIND_REQUEST ));
Map response = sendRequest( target, request, TRANSFER_TIMEOUT );
if ( response == null ){
return( RESP_FAILED );
}
if (((Long)response.get( "type" )).intValue() == RT_BIND_REPLY ){
int result = ((Long)response.get("ok")).intValue();
trace( "received bind reply: " + (result==0?"failed":"ok" ));
if ( result == 1 ){
return( RESP_OK );
}
}
return( RESP_NOT_OK );
}catch( Throwable e ){
log(e);
return( RESP_FAILED );
}
| protected int | sendClose(com.aelitis.azureus.core.dht.transport.DHTTransportContact target)
try{
Map request = new HashMap();
request.put("type", new Long( RT_CLOSE_REQUEST ));
Map response = sendRequest( target, request, TRANSFER_TIMEOUT );
if ( response == null ){
return( RESP_FAILED );
}
if (((Long)response.get( "type" )).intValue() == RT_CLOSE_REPLY ){
int result = ((Long)response.get("ok")).intValue();
trace( "received close reply: " + (result==0?"failed":"ok" ));
if ( result == 1 ){
return( RESP_OK );
}
}
return( RESP_NOT_OK );
}catch( Throwable e ){
log(e);
return( RESP_FAILED );
}
| protected java.util.Map | sendConnect(com.aelitis.azureus.core.dht.transport.DHTTransportContact target, com.aelitis.azureus.core.dht.transport.DHTTransportContact originator, java.util.Map originator_client_data)
try{
Map request = new HashMap();
request.put("type", new Long( RT_CONNECT_REQUEST ));
request.put("origin", encodeContact( originator ));
request.put( "port", new Long( ((DHTTransportUDPContact)originator).getTransportAddress().getPort()));
if ( originator_client_data != null ){
request.put( "client_data", originator_client_data );
}
Map response = sendRequest( target, request, TRANSFER_TIMEOUT );
if ( response == null ){
return( null );
}
if (((Long)response.get( "type" )).intValue() == RT_CONNECT_REPLY ){
int result = ((Long)response.get("ok")).intValue();
trace( "received connect reply: " + (result==0?"failed":"ok" ));
if ( result == 1 ){
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 );
}
| public java.util.Map | sendMessage(java.net.InetSocketAddress rendezvous, java.net.InetSocketAddress target, java.util.Map message)
try{
DHTTransportUDP transport = (DHTTransportUDP)dht.getTransport();
DHTTransportUDPContact rend_contact = transport.importContact( rendezvous, transport.getProtocolVersion());
DHTTransportUDPContact target_contact = transport.importContact( target, transport.getProtocolVersion());
Map result = sendPunch( rend_contact, target_contact, message, true );
return( result );
}catch( Throwable e ){
Debug.printStackTrace(e);
return( null );
}
| protected java.util.Map | sendPunch(com.aelitis.azureus.core.dht.transport.DHTTransportContact rendezvous, com.aelitis.azureus.core.dht.transport.udp.DHTTransportUDPContact target, java.util.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();
}
}
| protected int | sendQuery(com.aelitis.azureus.core.dht.transport.DHTTransportContact target)
try{
Map request = new HashMap();
request.put("type", new Long( RT_QUERY_REQUEST ));
Map response = sendRequest( target, request, TRANSFER_TIMEOUT );
if ( response == null ){
return( RESP_FAILED );
}
if (((Long)response.get( "type" )).intValue() == RT_QUERY_REPLY ){
int result = ((Long)response.get("ok")).intValue();
trace( "received query reply: " + (result==0?"failed":"ok" ));
if ( result == 1 ){
return( RESP_OK );
}
}
return( RESP_NOT_OK );
}catch( Throwable e ){
log(e);
return( RESP_FAILED );
}
| protected byte[] | sendRequest(com.aelitis.azureus.core.dht.transport.DHTTransportContact target, byte[] data, int timeout)
try{
return(
dht.getTransport().writeReadTransfer(
new DHTTransportProgressListener()
{
public void
reportSize(
long size )
{
}
public void
reportActivity(
String str )
{
}
public void
reportCompleteness(
int percent )
{
}
},
target,
transfer_handler_key,
data,
timeout ));
}catch( DHTTransportException e ){
// log(e); timeout most likely
return( null );
}
| protected java.util.Map | sendRequest(com.aelitis.azureus.core.dht.transport.DHTTransportContact target, java.util.Map data, int timeout)
try{
byte[] res = sendRequest( target, formatters.bEncode( data ), timeout );
if ( res == null ){
return( null );
}
return( formatters.bDecode( res ));
}catch( Throwable e ){
log(e);
return( null );
}
| protected boolean | sendTunnelInbound(com.aelitis.azureus.core.dht.transport.DHTTransportContact target)
log( "Sending tunnel inbound message to " + target.getString());
try{
Map message = new HashMap();
message.put( "type", new Long( RT_TUNNEL_INBOUND ));
return( sendTunnelMessage( target, message ));
}catch( Throwable e ){
log(e);
return( false );
}
| protected boolean | sendTunnelMessage(com.aelitis.azureus.core.dht.transport.DHTTransportContact target, java.util.Map data)
try{
return( sendTunnelMessage( target, formatters.bEncode( data )));
}catch( Throwable e ){
log(e);
return( false );
}
| protected boolean | sendTunnelMessage(com.aelitis.azureus.core.dht.transport.DHTTransportContact target, byte[] data)
try{
dht.getTransport().writeTransfer(
new DHTTransportProgressListener()
{
public void
reportSize(
long size )
{
}
public void
reportActivity(
String str )
{
}
public void
reportCompleteness(
int percent )
{
}
},
target,
transfer_handler_key,
new byte[0],
data,
TUNNEL_TIMEOUT );
return( true );
}catch( DHTTransportException e ){
// log(e); timeout most likely
return( false );
}
| protected boolean | sendTunnelOutbound(com.aelitis.azureus.core.dht.transport.DHTTransportContact target)
log( "Sending tunnel outbound message to " + target.getString());
try{
Map message = new HashMap();
message.put( "type", new Long( RT_TUNNEL_OUTBOUND ));
return( sendTunnelMessage( target, message ));
}catch( Throwable e ){
log(e);
return( false );
}
| public void | setRendezvous(com.aelitis.azureus.core.dht.transport.DHTTransportContact target, com.aelitis.azureus.core.dht.transport.DHTTransportContact rendezvous)
explicit_rendezvous_map.put( target.getAddress(), rendezvous );
if ( target.getAddress().equals( dht.getTransport().getLocalContact().getAddress())){
publish( true );
}
| public void | start()
if ( started ){
return;
}
started = true;
DHTTransport transport = dht.getTransport();
transport.addListener(
new DHTTransportListener()
{
public void
localContactChanged(
DHTTransportContact local_contact )
{
publish( false );
}
public void
currentAddress(
String address )
{
}
public void
reachabilityChanged(
boolean reacheable )
{
publish( false );
}
});
transport.registerTransferHandler(
transfer_handler_key,
new DHTTransportTransferHandler()
{
public String
getName()
{
return( "NAT Traversal" );
}
public byte[]
handleRead(
DHTTransportContact originator,
byte[] key )
{
return( null );
}
public byte[]
handleWrite(
DHTTransportContact originator,
byte[] key,
byte[] value )
{
return( receiveRequest((DHTTransportUDPContact)originator, value ));
}
});
timer.addPeriodicEvent(
REPUBLISH_TIME_MIN,
new UTTimerEventPerformer()
{
public void
perform(
UTTimerEvent event )
{
publish( false );
}
});
timer.addPeriodicEvent(
RENDEZVOUS_SERVER_TIMEOUT/2,
new UTTimerEventPerformer()
{
public void
perform(
UTTimerEvent event )
{
long now = plugin_interface.getUtilities().getCurrentSystemTime();
try{
server_mon.enter();
Iterator it = rendezvous_bindings.values().iterator();
while( it.hasNext()){
Object[] entry = (Object[])it.next();
long time = ((Long)entry[1]).longValue();
boolean removed = false;
if ( time > now ){
// clock change, easiest approach is to remove it
it.remove();
removed = true;
}else if ( now - time > RENDEZVOUS_SERVER_TIMEOUT ){
// timeout
it.remove();
removed = true;
}
if ( removed ){
log( "Rendezvous " + ((DHTTransportContact)entry[0]).getString() + " removed due to inactivity" );
}
}
}finally{
server_mon.exit();
}
}
});
publish( false );
| protected void | trace(java.lang.String str)
if ( TRACE ){
System.out.println( str );
}
|
|