use of com.biglybt.pif.messaging.generic.GenericMessageEndpoint in project BiglyBT by BiglySoftware.
the class MessageManagerImpl method registerGenericMessageType.
@Override
public GenericMessageRegistration registerGenericMessageType(final String _type, final String description, final int stream_crypto, final GenericMessageHandler handler) throws MessageException {
final String type = "AEGEN:" + _type;
final byte[] type_bytes = type.getBytes();
final byte[][] shared_secrets = new byte[][] { new SHA1Simple().calculateHash(type_bytes) };
synchronized (message_handlers) {
message_handlers.put(type, handler);
}
final NetworkManager.ByteMatcher matcher = new NetworkManager.ByteMatcher() {
@Override
public int matchThisSizeOrBigger() {
return (maxSize());
}
@Override
public int maxSize() {
return type_bytes.length;
}
@Override
public int minSize() {
return maxSize();
}
@Override
public Object matches(TransportHelper transport, ByteBuffer to_compare, int port) {
int old_limit = to_compare.limit();
to_compare.limit(to_compare.position() + maxSize());
boolean matches = to_compare.equals(ByteBuffer.wrap(type_bytes));
// restore buffer structure
to_compare.limit(old_limit);
return matches ? "" : null;
}
@Override
public Object minMatches(TransportHelper transport, ByteBuffer to_compare, int port) {
return (matches(transport, to_compare, port));
}
@Override
public byte[][] getSharedSecrets() {
return (shared_secrets);
}
@Override
public int getSpecificPort() {
return (-1);
}
};
NetworkManager.getSingleton().requestIncomingConnectionRouting(matcher, new NetworkManager.RoutingListener() {
@Override
public void connectionRouted(final NetworkConnection connection, Object routing_data) {
try {
ByteBuffer[] skip_buffer = { ByteBuffer.allocate(type_bytes.length) };
connection.getTransport().read(skip_buffer, 0, 1);
if (skip_buffer[0].remaining() != 0) {
Debug.out("incomplete read");
}
GenericMessageEndpointImpl endpoint = new GenericMessageEndpointImpl(connection.getEndpoint());
GenericMessageConnectionDirect direct_connection = GenericMessageConnectionDirect.receive(endpoint, type, description, stream_crypto, shared_secrets);
GenericMessageConnectionImpl new_connection = new GenericMessageConnectionImpl(MessageManagerImpl.this, direct_connection);
direct_connection.connect(connection);
if (handler.accept(new_connection)) {
new_connection.accepted();
} else {
connection.close("connection not accepted");
}
} catch (Throwable e) {
Debug.printStackTrace(e);
connection.close(e == null ? null : Debug.getNestedExceptionMessage(e));
}
}
@Override
public boolean autoCryptoFallback() {
return (stream_crypto != MessageManager.STREAM_ENCRYPTION_RC4_REQUIRED);
}
}, new MessageStreamFactory() {
@Override
public MessageStreamEncoder createEncoder() {
return new GenericMessageEncoder();
}
@Override
public MessageStreamDecoder createDecoder() {
return new GenericMessageDecoder(type, description);
}
});
return (new GenericMessageRegistration() {
@Override
public GenericMessageEndpoint createEndpoint(InetSocketAddress notional_target) {
return (new GenericMessageEndpointImpl(notional_target));
}
@Override
public GenericMessageConnection createConnection(GenericMessageEndpoint endpoint) throws MessageException {
return (new GenericMessageConnectionImpl(MessageManagerImpl.this, type, description, (GenericMessageEndpointImpl) endpoint, stream_crypto, shared_secrets));
}
@Override
public void cancel() {
NetworkManager.getSingleton().cancelIncomingConnectionRouting(matcher);
synchronized (message_handlers) {
message_handlers.remove(type);
}
}
});
}
use of com.biglybt.pif.messaging.generic.GenericMessageEndpoint in project BiglyBT by BiglySoftware.
the class BuddyPluginBuddy method outgoingConnection.
protected GenericMessageConnection outgoingConnection() throws BuddyPluginException {
GenericMessageRegistration msg_registration = plugin.getMessageRegistration();
if (msg_registration == null) {
throw (new BuddyPluginException("Messaging system unavailable"));
}
InetAddress ip = getIP();
if (ip == null) {
throw (new BuddyPluginException("Friend offline (no usable IP address)"));
}
InetSocketAddress tcp_target = null;
InetSocketAddress udp_target = null;
int tcp_port = getTCPPort();
if (tcp_port > 0) {
tcp_target = new InetSocketAddress(ip, tcp_port);
}
int udp_port = getUDPPort();
if (udp_port > 0) {
udp_target = new InetSocketAddress(ip, udp_port);
}
InetSocketAddress notional_target = tcp_target;
if (notional_target == null) {
notional_target = udp_target;
}
if (notional_target == null) {
throw (new BuddyPluginException("Friend offline (no usable protocols)"));
}
GenericMessageEndpoint endpoint = msg_registration.createEndpoint(notional_target);
if (tcp_target != null) {
endpoint.addTCP(tcp_target);
}
if (udp_target != null) {
endpoint.addUDP(udp_target);
}
GenericMessageConnection con = null;
try {
last_connect_attempt = SystemTime.getCurrentTime();
con = msg_registration.createConnection(endpoint);
plugin.addRateLimiters(con);
String reason = "Friend: Outgoing connection establishment";
SESecurityManager sec_man = plugin.getSecurityManager();
con = sec_man.getSTSConnection(con, sec_man.getPublicKey(SEPublicKey.KEY_TYPE_ECC_192, reason), new SEPublicKeyLocator() {
@Override
public boolean accept(Object context, SEPublicKey other_key) {
String other_key_str = Base32.encode(other_key.encodeRawPublicKey());
if (other_key_str.equals(public_key)) {
consec_connect_fails = 0;
return (true);
} else {
log(getString() + ": connection failed due to pk mismatch");
return (false);
}
}
}, reason, BuddyPlugin.BLOCK_CRYPTO);
con.connect();
return (con);
} catch (Throwable e) {
if (con != null) {
consec_connect_fails++;
try {
con.close();
} catch (Throwable f) {
log("Failed to close connection", f);
}
}
throw (new BuddyPluginException("Failed to send message", e));
}
}
Aggregations