Skip to content

Commit

Permalink
interop tests with latest
Browse files Browse the repository at this point in the history
  • Loading branch information
@kevodwyer
kevodwyer committed Jul 28, 2023
1 parent cd42a07 commit 0f458c2
Show file tree
Hide file tree
Showing 3 changed files with 184 additions and 2 deletions.
7 changes: 5 additions & 2 deletions Containerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,10 +16,13 @@ LABEL org.opencontainers.image.licenses="MIT license"

ENV IPFS_PATH=/opt/nabu/.ipfs

VOLUMES /opt/nabu/.ipfs

WORKDIR /opt/nabu
RUN mkdir -p /opt/nabu/.ipfs
COPY --from=build /opt/nabu/target /opt/nabu

ENTRYPOINT ["java", "-cp", "/opt/nabu/nabu-v0.0.1-SNAPSHOT-jar-with-dependencies.jar", "org.peergos.Nabu", "Addresses.API", "/ip4/0.0.0.0/tcp/5001"]
# ENTRYPOINT ["java", "-cp", "/opt/nabu/nabu-v0.0.1-SNAPSHOT-jar-with-dependencies.jar", "org.peergos.Nabu"]
ENTRYPOINT ["java", "-cp", "/opt/nabu/nabu-v0.1.0-SNAPSHOT-jar-with-dependencies.jar", "org.peergos.client.InteropTestClient"]

EXPOSE 4001 5001 8080 8000
EXPOSE 4001 5001 8080 8003 8002
5 changes: 5 additions & 0 deletions pom.xml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,11 @@
</build>

<dependencies>
<dependency>
<groupId>redis.clients</groupId>
<artifactId>jedis</artifactId>
<version>4.3.0</version>
</dependency>
<dependency>
<groupId>com.github.multiformats</groupId>
<artifactId>java-multiaddr</artifactId>
Expand Down
174 changes: 174 additions & 0 deletions src/main/java/org/peergos/client/InteropTestClient.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,174 @@
package org.peergos.client;

import identify.pb.IdentifyOuterClass;
import io.libp2p.core.Host;
import io.libp2p.core.PeerId;
import io.libp2p.core.crypto.PrivKey;
import io.libp2p.core.dsl.Builder;
import io.libp2p.core.dsl.BuilderJKt;
import io.libp2p.core.multiformats.Multiaddr;
import io.libp2p.core.multistream.ProtocolBinding;
import io.libp2p.core.mux.StreamMuxerProtocol;
import io.libp2p.crypto.keys.Ed25519Kt;
import io.libp2p.etc.types.ByteArrayExtKt;
import io.libp2p.protocol.Identify;
import io.libp2p.protocol.Ping;
import io.libp2p.protocol.PingController;
import io.libp2p.security.noise.NoiseXXSecureChannel;
import io.libp2p.security.tls.TlsSecureChannel;
import io.libp2p.transport.tcp.TcpTransport;
import io.ipfs.multiaddr.MultiAddress;
import org.peergos.protocol.IdentifyBuilder;
import redis.clients.jedis.Jedis;

import java.net.*;
import java.util.*;
import java.util.stream.Collectors;

public class InteropTestClient {

private static String getLocalIPAddress() throws SocketException {
System.err.println("Getting localIP");
List<NetworkInterface> interfaces = NetworkInterface.networkInterfaces().collect(Collectors.toList());
for (NetworkInterface inter : interfaces) {
for (InterfaceAddress addr : inter.getInterfaceAddresses()) {
InetAddress address = addr.getAddress();
if (! address.isLoopbackAddress() && !(address instanceof Inet6Address))
return address.getHostAddress();
}
}
throw new IllegalStateException("Unable to determine local IPAddress");
}
private InteropTestClient(String transport, String muxer, String security, boolean is_dialer,
String ip, String redis_addr, String test_timeout_seconds) throws Exception {
if (ip == null || ip.length() == 0)
ip = "0.0.0.0";
if (! is_dialer && ip.equals("0.0.0.0"))
ip = getLocalIPAddress();

if (redis_addr == null || redis_addr.length() == 0) {
redis_addr = "redis:6379";
}
if (test_timeout_seconds == null || test_timeout_seconds.length() == 0) {
test_timeout_seconds = "180";
}
System.err.println("Parameters: transport=" + transport + " muxer=" + muxer +
" security=" + security + " is_dialer=" + is_dialer + " ip=" + ip +
" redis_addr=" + redis_addr + " test_timeout_seconds=" + test_timeout_seconds);
if (transport == null || muxer == null || security == null) {
throw new IllegalStateException("transport == null || muxer == null || security == null");
}
int port = 10000 + new Random().nextInt(50000);
Multiaddr address = Multiaddr.fromString("/ip4/" + ip + "/tcp/" + port);
List<MultiAddress> swarmAddresses = List.of(new MultiAddress(address.toString()));

List<ProtocolBinding> protocols = new ArrayList<>();
protocols.add(new Ping());

PrivKey privKey = Ed25519Kt.generateEd25519KeyPair().getFirst();
PeerId peerId = PeerId.fromPubKey(privKey.publicKey());
Multiaddr advertisedAddr = address.withP2P(peerId);
List<String> listenAddrs = new ArrayList<>();
listenAddrs.addAll(swarmAddresses.stream().map(MultiAddress::toString).collect(Collectors.toList()));
Host node = BuilderJKt.hostJ(Builder.Defaults.None, b -> {
b.getIdentity().setFactory(() -> privKey);
if (transport.equals("tcp")) {
b.getTransports().add(TcpTransport::new);
}
if (security.equals("noise")) {
b.getSecureChannels().add((k, m) -> new NoiseXXSecureChannel(k, m));
} else if(security.equals("tls")) {
b.getSecureChannels().add((k, m) -> new TlsSecureChannel(k, m, "ECDSA"));
}
List<StreamMuxerProtocol> muxers = new ArrayList<>();
if (muxer.equals("mplex")) {
muxers.add(StreamMuxerProtocol.getMplex());
} else if(muxer.equals("yamux")) {
muxers.add(StreamMuxerProtocol.getYamux());
}
b.getMuxers().addAll(muxers);

for (ProtocolBinding<?> protocol : protocols) {
b.getProtocols().add(protocol);
}

IdentifyOuterClass.Identify.Builder identifyBuilder = IdentifyOuterClass.Identify.newBuilder()
.setProtocolVersion("ipfs/0.1.0")
.setAgentVersion("nabu/v0.1.0")
.setPublicKey(ByteArrayExtKt.toProtobuf(privKey.publicKey().bytes()))
.addAllListenAddrs(listenAddrs.stream()
.map(Multiaddr::fromString)
.map(Multiaddr::serialize)
.map(ByteArrayExtKt::toProtobuf)
.collect(Collectors.toList()));
for (ProtocolBinding<?> protocol : protocols) {
identifyBuilder = identifyBuilder.addAllProtocols(protocol.getProtocolDescriptor().getAnnounceProtocols());
}
b.getProtocols().add(new Identify(identifyBuilder.build()));

for (String listenAddr : listenAddrs) {
b.getNetwork().listen(listenAddr);
}

b.getConnectionHandlers().add(conn -> System.err.println(conn.localAddress() +
" received connection from " + conn.remoteAddress() +
" on transport " + conn.transport()));
});
node.start().join();
IdentifyBuilder.addIdentifyProtocol(node);
Jedis jedis = new Jedis("http://" + redis_addr);
boolean isReady = false;
while (!isReady) {
if ("PONG".equals(jedis.ping())) {
isReady = true;
} else {
System.err.println("waiting for redis to start...");
Thread.sleep(2000);
}
}
System.err.println("My multiaddr is: " + advertisedAddr);
try {
if (is_dialer) {
List<String> listenerAddrs = jedis.blpop(Integer.parseInt(test_timeout_seconds), "listenerAddr");
if (listenerAddrs == null || listenerAddrs.isEmpty()) {
throw new IllegalStateException("listenerAddr not set");
}
String listenerAddrStr = listenerAddrs.get(listenerAddrs.size() -1);
Multiaddr listenerAddr = Multiaddr.fromString(listenerAddrStr);
System.err.println("Other peer multiaddr is: " + listenerAddr);
System.err.println("Sending ping messages to " + listenerAddr);
long handshakeStart = System.currentTimeMillis();
PingController pinger = new Ping().dial(node, listenerAddr).getController().join();
long pingRTTMilllis = pinger.ping().join();
long handshakeEnd = System.currentTimeMillis();
long handshakePlusOneRTT = handshakeEnd - handshakeStart;
System.err.println("Ping latency " + pingRTTMilllis + "ms");
String jsonResult = "{\"handshakePlusOneRTTMillis\":" + Double.valueOf(handshakePlusOneRTT) +
",\"pingRTTMilllis\":" + Double.valueOf(pingRTTMilllis) + "}";
System.out.println(jsonResult);
} else {
jedis.rpush("listenerAddr", advertisedAddr.toString());
Thread.sleep(Integer.parseInt(test_timeout_seconds) * 1000);
}
} finally {
System.err.println("Shutting down");
node.stop();
}
}
public static void main(String[] args) {
try {
String transport = System.getenv("transport"); //"tcp"
String muxer = System.getenv("muxer"); //"mplex", "yamux"
String security = System.getenv("security"); //"tls", "noise"
boolean is_dialer = Boolean.parseBoolean(System.getenv("is_dialer"));
String ip = System.getenv("ip");
String redis_addr = System.getenv("redis_addr");
String test_timeout_seconds = System.getenv("test_timeout_seconds");
new InteropTestClient(transport, muxer, security, is_dialer, ip, redis_addr, test_timeout_seconds);
} catch (Throwable t) {
System.err.println("Unexpected exit: " + t);
System.exit(-1);
}
}
}

0 comments on commit 0f458c2

Please sign in to comment.