Network code now works well enough for the server to think it successfully established a connection

This commit is contained in:
Christian Basler 2015-04-07 18:48:58 +02:00
parent 3299d8ca4a
commit 35088ca033
27 changed files with 636 additions and 150 deletions

16
demo/build.gradle Normal file
View File

@ -0,0 +1,16 @@
apply plugin: 'java'
sourceCompatibility = 1.7
version = '1.0'
repositories {
mavenCentral()
}
dependencies {
compile project(':domain')
compile project(':networking')
compile project(':inventory')
compile 'org.slf4j:slf4j-simple:1.7.12'
testCompile group: 'junit', name: 'junit', version: '4.11'
}

View File

@ -0,0 +1,50 @@
/*
* Copyright 2015 Christian Basler
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package ch.dissem.bitmessage.demo;
import ch.dissem.bitmessage.Context;
import ch.dissem.bitmessage.entity.payload.ObjectPayload;
import ch.dissem.bitmessage.inventory.SimpleAddressRepository;
import ch.dissem.bitmessage.inventory.SimpleInventory;
import ch.dissem.bitmessage.networking.NetworkNode;
import ch.dissem.bitmessage.ports.NetworkHandler;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
/**
* Created by chris on 06.04.15.
*/
public class Main {
public static void main(String[] args) throws IOException {
NetworkNode networkNode = new NetworkNode();
Context.init(new SimpleInventory(), new SimpleAddressRepository(), networkNode, 48444);
Context.getInstance().addStream(1);
networkNode.setListener(new NetworkHandler.MessageListener() {
@Override
public void receive(ObjectPayload payload) {
// TODO
}
});
networkNode.start();
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
System.out.print("Enter String");
br.readLine();
}
}

View File

@ -18,8 +18,12 @@ package ch.dissem.bitmessage;
import ch.dissem.bitmessage.ports.AddressRepository;
import ch.dissem.bitmessage.ports.Inventory;
import ch.dissem.bitmessage.ports.NetworkMessageReceiver;
import ch.dissem.bitmessage.ports.NetworkMessageSender;
import ch.dissem.bitmessage.ports.NetworkHandler;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;
import java.util.TreeSet;
/**
* Created by chris on 05.04.15.
@ -31,19 +35,22 @@ public class Context {
private Inventory inventory;
private AddressRepository addressRepo;
private NetworkMessageSender sender;
private NetworkMessageReceiver receiver;
private NetworkHandler networkHandler;
private Collection<Long> streams = new TreeSet<>();
private int port;
private Context(Inventory inventory, AddressRepository addressRepo,
NetworkMessageSender sender, NetworkMessageReceiver receiver) {
NetworkHandler networkHandler, int port) {
this.inventory = inventory;
this.addressRepo = addressRepo;
this.sender = sender;
this.receiver = receiver;
this.networkHandler = networkHandler;
this.port = port;
}
public static void init(Inventory inventory, AddressRepository addressRepository, NetworkMessageSender sender, NetworkMessageReceiver receiver) {
instance = new Context(inventory, addressRepository, sender, receiver);
public static void init(Inventory inventory, AddressRepository addressRepository, NetworkHandler networkHandler, int port) {
instance = new Context(inventory, addressRepository, networkHandler, port);
}
public static Context getInstance() {
@ -57,4 +64,25 @@ public class Context {
public AddressRepository getAddressRepository() {
return addressRepo;
}
public int getPort() {
return port;
}
public long[] getStreams() {
long[] result = new long[streams.size()];
int i = 0;
for (long stream : streams) {
result[i++] = stream;
}
return result;
}
public void addStream(long stream) {
streams.add(stream);
}
public void removeStream(long stream) {
streams.remove(stream);
}
}

View File

@ -146,7 +146,7 @@ public class Version implements MessagePayload {
}
public Builder defaults() {
version = Context.CURRENT;
version = Context.CURRENT_VERSION;
services = 1;
timestamp = System.currentTimeMillis() / 1000;
nonce = new Random().nextInt();

View File

@ -0,0 +1,45 @@
/*
* Copyright 2015 Christian Basler
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package ch.dissem.bitmessage.entity.payload;
import java.io.IOException;
import java.io.OutputStream;
/**
* Created by chris on 07.04.15.
*/
public class Broadcast implements ObjectPayload {
private long stream;
private byte[] tag;
private byte[] encrypted;
public Broadcast(long stream, byte[] tag, byte[] encrypted) {
this.stream = stream;
this.tag = tag;
this.encrypted = encrypted;
}
@Override
public long getStream() {
return stream;
}
@Override
public void write(OutputStream stream) throws IOException {
}
}

View File

@ -23,12 +23,19 @@ import java.io.OutputStream;
* Created by chris on 24.03.15.
*/
public class GenericPayload implements ObjectPayload {
private long stream;
private byte[] data;
public GenericPayload(byte[] data) {
public GenericPayload(long stream, byte[] data) {
this.stream=stream;
this.data = data;
}
@Override
public long getStream() {
return stream;
}
@Override
public void write(OutputStream stream) throws IOException {
stream.write(data);

View File

@ -23,10 +23,12 @@ import java.io.OutputStream;
* Created by chris on 24.03.15.
*/
public class GetPubkey implements ObjectPayload {
private long stream;
private byte[] ripe;
private byte[] tag;
public GetPubkey(byte[] ripeOrTag) {
public GetPubkey(long stream, byte[] ripeOrTag) {
this.stream=stream;
switch (ripeOrTag.length) {
case 20:
ripe = ripeOrTag;
@ -39,6 +41,11 @@ public class GetPubkey implements ObjectPayload {
}
}
@Override
public long getStream() {
return stream;
}
@Override
public void write(OutputStream stream) throws IOException {
if (tag != null) {

View File

@ -0,0 +1,43 @@
/*
* Copyright 2015 Christian Basler
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package ch.dissem.bitmessage.entity.payload;
import java.io.IOException;
import java.io.OutputStream;
/**
* Created by chris on 07.04.15.
*/
public class Msg implements ObjectPayload {
private long stream;
private byte[] encrypted;
public Msg(long stream, byte[] encrypted) {
this.stream = stream;
this.encrypted = encrypted;
}
@Override
public long getStream() {
return stream;
}
@Override
public void write(OutputStream stream) throws IOException {
stream.write(encrypted);
}
}

View File

@ -22,4 +22,5 @@ import ch.dissem.bitmessage.entity.Streamable;
* The payload of an 'object' command. This is shared by the network.
*/
public interface ObjectPayload extends Streamable {
long getStream();
}

View File

@ -22,8 +22,6 @@ package ch.dissem.bitmessage.entity.payload;
public interface Pubkey extends ObjectPayload {
long getVersion();
long getStream();
byte[] getSigningKey();
byte[] getEncryptionKey();

View File

@ -25,11 +25,21 @@ import java.io.OutputStream;
* Created by chris on 24.03.15.
*/
public class V2Pubkey implements Pubkey {
protected long streamNumber;
protected long stream;
protected long behaviorBitfield;
protected byte[] publicSigningKey;
protected byte[] publicEncryptionKey;
protected V2Pubkey() {
}
private V2Pubkey(Builder builder) {
stream = builder.streamNumber;
behaviorBitfield = builder.behaviorBitfield;
publicSigningKey = builder.publicSigningKey;
publicEncryptionKey = builder.publicEncryptionKey;
}
@Override
public long getVersion() {
return 2;
@ -37,7 +47,7 @@ public class V2Pubkey implements Pubkey {
@Override
public long getStream() {
return streamNumber;
return stream;
}
@Override
@ -56,4 +66,38 @@ public class V2Pubkey implements Pubkey {
stream.write(publicSigningKey);
stream.write(publicEncryptionKey);
}
public static class Builder {
private long streamNumber;
private long behaviorBitfield;
private byte[] publicSigningKey;
private byte[] publicEncryptionKey;
public Builder() {
}
public Builder streamNumber(long streamNumber) {
this.streamNumber = streamNumber;
return this;
}
public Builder behaviorBitfield(long behaviorBitfield) {
this.behaviorBitfield = behaviorBitfield;
return this;
}
public Builder publicSigningKey(byte[] publicSigningKey) {
this.publicSigningKey = publicSigningKey;
return this;
}
public Builder publicEncryptionKey(byte[] publicEncryptionKey) {
this.publicEncryptionKey = publicEncryptionKey;
return this;
}
public V2Pubkey build() {
return new V2Pubkey(this);
}
}
}

View File

@ -29,6 +29,17 @@ public class V3Pubkey extends V2Pubkey {
long extraBytes;
byte[] signature;
protected V3Pubkey(Builder builder) {
stream = builder.streamNumber;
behaviorBitfield = builder.behaviorBitfield;
publicSigningKey = builder.publicSigningKey;
publicEncryptionKey = builder.publicEncryptionKey;
nonceTrialsPerByte = builder.nonceTrialsPerByte;
extraBytes = builder.extraBytes;
signature = builder.signature;
}
@Override
public void write(OutputStream stream) throws IOException {
super.write(stream);
@ -42,4 +53,57 @@ public class V3Pubkey extends V2Pubkey {
public long getVersion() {
return 3;
}
public static class Builder extends V2Pubkey.Builder {
private long streamNumber;
private long behaviorBitfield;
private byte[] publicSigningKey;
private byte[] publicEncryptionKey;
private long nonceTrialsPerByte;
private long extraBytes;
private byte[] signature;
public Builder() {
}
public Builder streamNumber(long streamNumber) {
this.streamNumber = streamNumber;
return this;
}
public Builder behaviorBitfield(long behaviorBitfield) {
this.behaviorBitfield = behaviorBitfield;
return this;
}
public Builder publicSigningKey(byte[] publicSigningKey) {
this.publicSigningKey = publicSigningKey;
return this;
}
public Builder publicEncryptionKey(byte[] publicEncryptionKey) {
this.publicEncryptionKey = publicEncryptionKey;
return this;
}
public Builder nonceTrialsPerByte(long nonceTrialsPerByte) {
this.nonceTrialsPerByte = nonceTrialsPerByte;
return this;
}
public Builder extraBytes(long extraBytes) {
this.extraBytes = extraBytes;
return this;
}
public Builder signature(byte[] signature) {
this.signature = signature;
return this;
}
public V3Pubkey build() {
return new V3Pubkey(this);
}
}
}

View File

@ -23,15 +23,22 @@ import java.io.OutputStream;
* Created by chris on 27.03.15.
*/
public class V4Pubkey implements Pubkey {
private long streamNumber;
private long stream;
private byte[] tag;
private byte[] encrypted;
private V3Pubkey decrypted;
public V4Pubkey(long stream, byte[] tag, byte[] encrypted) {
this.stream = stream;
this.tag = tag;
this.encrypted = encrypted;
}
public V4Pubkey(V3Pubkey decrypted) {
this.stream = decrypted.stream;
// TODO: this.tag = new BitmessageAddress(this).doubleHash
this.decrypted = decrypted;
// TODO: this.tag = new BitmessageAddress(this).doubleHash
}
@Override
@ -47,7 +54,7 @@ public class V4Pubkey implements Pubkey {
@Override
public long getStream() {
return streamNumber;
return stream;
}
@Override

View File

@ -21,12 +21,9 @@ import ch.dissem.bitmessage.utils.Encode;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Inet6Address;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Arrays;
import java.util.Objects;
/**
* A node's address. It's written in IPv6 format.
@ -51,6 +48,14 @@ public class NetworkAddress implements Streamable {
private byte[] ipv6;
private int port;
private NetworkAddress(Builder builder) {
time = builder.time;
stream = builder.stream;
services = builder.services;
ipv6 = builder.ipv6;
port = builder.port;
}
public int getPort() {
return port;
}
@ -63,14 +68,6 @@ public class NetworkAddress implements Streamable {
}
}
private NetworkAddress(Builder builder) {
time = builder.time;
stream = builder.stream;
services = builder.services;
ipv6 = builder.ipv6;
port = builder.port;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
@ -88,6 +85,11 @@ public class NetworkAddress implements Streamable {
return result;
}
@Override
public String toString() {
return toInetAddress() + ":" + port;
}
@Override
public void write(OutputStream stream) throws IOException {
write(stream, false);

View File

@ -17,10 +17,10 @@
package ch.dissem.bitmessage.factory;
import ch.dissem.bitmessage.entity.NetworkMessage;
import ch.dissem.bitmessage.entity.payload.GenericPayload;
import ch.dissem.bitmessage.entity.payload.GetPubkey;
import ch.dissem.bitmessage.entity.payload.ObjectPayload;
import ch.dissem.bitmessage.entity.payload.*;
import ch.dissem.bitmessage.utils.Decode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.IOException;
import java.io.InputStream;
@ -29,26 +29,63 @@ import java.io.InputStream;
* Creates {@link NetworkMessage} objects from {@link InputStream InputStreams}
*/
public class Factory {
public static final Logger LOG = LoggerFactory.getLogger(Factory.class);
public static NetworkMessage getNetworkMessage(int version, InputStream stream) throws IOException {
return new V3MessageFactory().read(stream);
}
static ObjectPayload getObjectPayload(long objectType, long version, InputStream stream, int length) throws IOException {
static ObjectPayload getObjectPayload(long objectType, long version, long streamNumber, InputStream stream, int length) throws IOException {
if (objectType < 4) {
switch ((int) objectType) {
case 0: // getpubkey
return new GetPubkey(Decode.bytes(stream, length));
return new GetPubkey(streamNumber, Decode.bytes(stream, length));
case 1: // pubkey
break;
return parsePubkey((int) version, streamNumber, stream, length);
case 2: // msg
break;
return parseMsg((int) version, streamNumber, stream, length);
case 3: // broadcast
break;
return parseBroadcast((int) version, streamNumber, stream, length);
}
throw new RuntimeException("This must not happen, someone broke something in the code!");
} else {
// passthrough message
return new GenericPayload(Decode.bytes(stream, length));
LOG.error("This should not happen, someone broke something in the code!");
}
// fallback: just store the message - we don't really care what it is
LOG.error("Unexpected object type: " + objectType);
return new GenericPayload(streamNumber, Decode.bytes(stream, length));
}
private static ObjectPayload parsePubkey(int version, long streamNumber, InputStream stream, int length) throws IOException {
switch (version) {
case 2:
return new V2Pubkey.Builder()
.streamNumber(streamNumber)
.behaviorBitfield(Decode.int64(stream))
.publicSigningKey(Decode.bytes(stream, 64))
.publicEncryptionKey(Decode.bytes(stream, 64))
.build();
case 3:
V3Pubkey.Builder v3 = new V3Pubkey.Builder()
.streamNumber(streamNumber)
.behaviorBitfield(Decode.int64(stream))
.publicSigningKey(Decode.bytes(stream, 64))
.publicEncryptionKey(Decode.bytes(stream, 64))
.nonceTrialsPerByte(Decode.varInt(stream))
.extraBytes(Decode.varInt(stream));
int sigLength = (int) Decode.varInt(stream);
v3.signature(Decode.bytes(stream, sigLength));
return v3.build();
case 4:
// TODO
}
LOG.debug("Unexpected pubkey version " + version + ", handling as generic payload object");
return new GenericPayload(streamNumber, Decode.bytes(stream, length));
}
private static ObjectPayload parseMsg(int version, long streamNumber, InputStream stream, int length) throws IOException {
return new Msg(streamNumber, Decode.bytes(stream, length));
}
private static ObjectPayload parseBroadcast(int version, long streamNumber, InputStream stream, int length) throws IOException {
return new Broadcast(streamNumber, Decode.bytes(stream, 32), Decode.bytes(stream, length - 32));
}
}

View File

@ -22,6 +22,8 @@ import ch.dissem.bitmessage.entity.valueobject.InventoryVector;
import ch.dissem.bitmessage.entity.valueobject.NetworkAddress;
import ch.dissem.bitmessage.utils.Decode;
import ch.dissem.bitmessage.utils.Security;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.ByteArrayInputStream;
import java.io.IOException;
@ -31,6 +33,8 @@ import java.io.InputStream;
* Creates protocol v3 network messages from {@link InputStream InputStreams}
*/
class V3MessageFactory {
private Logger LOG = LoggerFactory.getLogger(V3MessageFactory.class);
public NetworkMessage read(InputStream stream) throws IOException {
if (testMagic(stream)) {
String command = getCommand(stream);
@ -45,9 +49,11 @@ class V3MessageFactory {
} else {
throw new IOException("Checksum failed for message '" + command + "'");
}
}
} else {
LOG.debug("Failed test for MAGIC bytes");
return null;
}
}
private MessagePayload getPayload(String command, InputStream stream, int length) throws IOException {
switch (command) {
@ -64,6 +70,7 @@ class V3MessageFactory {
case "object":
return parseObject(stream, length);
default:
LOG.debug("Unknown command: " + command);
return null;
}
}
@ -75,7 +82,7 @@ class V3MessageFactory {
long version = Decode.varInt(stream);
long streamNumber = Decode.varInt(stream);
ObjectPayload payload = Factory.getObjectPayload(objectType, version, stream, length);
ObjectPayload payload = Factory.getObjectPayload(objectType, version, streamNumber, stream, length);
return new ObjectMessage.Builder()
.nonce(nonce)
@ -109,7 +116,7 @@ class V3MessageFactory {
long count = Decode.varInt(stream);
Addr.Builder builder = new Addr.Builder();
for (int i = 0; i < count; i++) {
builder.addAddress(parseAddress(stream));
builder.addAddress(parseAddress(stream, false));
}
return builder.build();
}
@ -118,8 +125,8 @@ class V3MessageFactory {
int version = Decode.int32(stream);
long services = Decode.int64(stream);
long timestamp = Decode.int64(stream);
NetworkAddress addrRecv = parseAddress(stream);
NetworkAddress addrFrom = parseAddress(stream);
NetworkAddress addrRecv = parseAddress(stream, true);
NetworkAddress addrFrom = parseAddress(stream, true);
long nonce = Decode.int64(stream);
String userAgent = Decode.varString(stream);
long[] streamNumbers = Decode.varIntList(stream);
@ -138,9 +145,16 @@ class V3MessageFactory {
return new InventoryVector(Decode.bytes(stream, 32));
}
private NetworkAddress parseAddress(InputStream stream) throws IOException {
long time = Decode.int64(stream);
long streamNumber = Decode.uint32(stream); // This isn't consistent, not sure if this is correct
private NetworkAddress parseAddress(InputStream stream, boolean light) throws IOException {
long time;
long streamNumber;
if (!light) {
time = Decode.int64(stream);
streamNumber = Decode.uint32(stream); // This isn't consistent, not sure if this is correct
} else {
time = 0;
streamNumber = 0;
}
long services = Decode.int64(stream);
byte[] ipv6 = Decode.bytes(stream, 16);
int port = Decode.uint16(stream);
@ -159,13 +173,21 @@ class V3MessageFactory {
private String getCommand(InputStream stream) throws IOException {
byte[] bytes = new byte[12];
stream.read(bytes);
return new String(bytes, "ASCII");
int end = -1;
for (int i = 0; i < bytes.length; i++) {
bytes[i] = (byte) stream.read();
if (end == -1) {
if (bytes[i] == 0) end = i;
} else {
if (bytes[i] != 0) throw new IOException("'\\0' padding expected for command");
}
}
return new String(bytes, 0, end, "ASCII");
}
private boolean testMagic(InputStream stream) throws IOException {
for (byte b : NetworkMessage.MAGIC_BYTES) {
if (b != stream.read()) return false;
if (b != (byte) stream.read()) return false;
}
return true;
}

View File

@ -17,7 +17,6 @@
package ch.dissem.bitmessage.ports;
import ch.dissem.bitmessage.entity.ObjectMessage;
import ch.dissem.bitmessage.entity.payload.ObjectPayload;
import ch.dissem.bitmessage.entity.valueobject.InventoryVector;
import java.util.List;
@ -26,13 +25,13 @@ import java.util.List;
* The Inventory stores and retrieves objects, cleans up outdated objects and can tell which objects are still missing.
*/
public interface Inventory {
public List<InventoryVector> getInventory(long... streams);
List<InventoryVector> getInventory(long... streams);
public List<InventoryVector> getMissing(List<InventoryVector> offer);
List<InventoryVector> getMissing(List<InventoryVector> offer);
public ObjectMessage getObject(InventoryVector vector);
ObjectMessage getObject(InventoryVector vector);
public void storeObject(ObjectMessage object);
void storeObject(ObjectMessage object);
public void cleanup();
void cleanup();
}

View File

@ -17,11 +17,18 @@
package ch.dissem.bitmessage.ports;
import ch.dissem.bitmessage.entity.payload.ObjectPayload;
import ch.dissem.bitmessage.entity.valueobject.NetworkAddress;
/**
* Sends messages
* Handles incoming messages
*/
public interface NetworkMessageSender {
void send(NetworkAddress node, ObjectPayload payload);
public interface NetworkHandler {
void setListener(MessageListener listener);
void start();
void send(ObjectPayload payload);
interface MessageListener {
void receive(ObjectPayload payload);
}
}

View File

@ -27,7 +27,10 @@ import java.nio.ByteBuffer;
public class Decode {
public static byte[] bytes(InputStream stream, int count) throws IOException {
byte[] result = new byte[count];
stream.read(result);
int off = 0;
while (off < count) {
off += stream.read(result, off, count - off);
}
return result;
}

13
inventory/build.gradle Normal file
View File

@ -0,0 +1,13 @@
apply plugin: 'java'
sourceCompatibility = 1.7
version = '1.0'
repositories {
mavenCentral()
}
dependencies {
compile project(':domain')
testCompile group: 'junit', name: 'junit', version: '4.11'
}

View File

@ -14,23 +14,24 @@
* limitations under the License.
*/
package ch.dissem.bitmessage.ports;
package ch.dissem.bitmessage.inventory;
import ch.dissem.bitmessage.entity.NetworkMessage;
import ch.dissem.bitmessage.entity.payload.ObjectPayload;
import ch.dissem.bitmessage.entity.valueobject.NetworkAddress;
import ch.dissem.bitmessage.ports.AddressRepository;
import java.io.IOException;
import java.util.Collections;
import java.util.List;
/**
* Handles incoming messages
* Created by chris on 06.04.15.
*/
public interface NetworkMessageReceiver {
void registerListener(int port, MessageListener listener) throws IOException;
public class SimpleAddressRepository implements AddressRepository {
@Override
public List<NetworkAddress> getKnownAddresses(int limit, long... streams) {
return Collections.singletonList(new NetworkAddress.Builder().ipv4(127, 0, 0, 1).port(8444).build());
}
void registerListener(NetworkAddress node, MessageListener listener) throws IOException;
interface MessageListener {
void receive(ObjectPayload payload);
@Override
public void offerAddresses(List<NetworkAddress> addresses) {
}
}

View File

@ -0,0 +1,55 @@
/*
* Copyright 2015 Christian Basler
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package ch.dissem.bitmessage.inventory;
import ch.dissem.bitmessage.entity.ObjectMessage;
import ch.dissem.bitmessage.entity.valueobject.InventoryVector;
import ch.dissem.bitmessage.ports.Inventory;
import sun.reflect.generics.reflectiveObjects.NotImplementedException;
import java.util.LinkedList;
import java.util.List;
/**
* Created by chris on 06.04.15.
*/
public class SimpleInventory implements Inventory {
@Override
public List<InventoryVector> getInventory(long... streams) {
return new LinkedList<>();
}
@Override
public List<InventoryVector> getMissing(List<InventoryVector> offer) {
return offer;
}
@Override
public ObjectMessage getObject(InventoryVector vector) {
throw new NotImplementedException();
}
@Override
public void storeObject(ObjectMessage object) {
throw new NotImplementedException();
}
@Override
public void cleanup() {
throw new NotImplementedException();
}
}

View File

@ -8,7 +8,7 @@ repositories {
}
dependencies {
compile ':domain'
compile project(':domain')
compile 'com.google.guava:guava-concurrent:r03'
testCompile 'org.slf4j:slf4j-simple:1.7.12'
testCompile 'junit:junit:4.11'

View File

@ -21,7 +21,7 @@ import ch.dissem.bitmessage.entity.*;
import ch.dissem.bitmessage.entity.valueobject.InventoryVector;
import ch.dissem.bitmessage.entity.valueobject.NetworkAddress;
import ch.dissem.bitmessage.factory.Factory;
import ch.dissem.bitmessage.ports.NetworkMessageReceiver.MessageListener;
import ch.dissem.bitmessage.ports.NetworkHandler.MessageListener;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@ -69,6 +69,14 @@ public class Connection implements Runnable {
this.node = new NetworkAddress.Builder().ip(socket.getInetAddress()).port(socket.getPort()).build();
}
public State getState() {
return state;
}
public NetworkAddress getNode() {
return node;
}
@Override
public void run() {
if (state == CLIENT) {
@ -77,6 +85,8 @@ public class Connection implements Runnable {
while (state != DISCONNECTED) {
try {
NetworkMessage msg = Factory.getNetworkMessage(version, in);
if (msg == null)
continue;
switch (state) {
case ACTIVE:
receiveMessage(msg.getPayload());
@ -90,20 +100,16 @@ public class Connection implements Runnable {
this.version = payload.getVersion();
this.streams = payload.getStreams();
send(new VerAck());
if (state == SERVER) {
state = ACTIVE;
}
sendAddresses();
sendInventory();
} else {
LOG.info("Received unsupported version " + payload.getVersion() + ", disconnecting.");
disconnect();
}
break;
case VERACK:
if (state == CLIENT) {
sendAddresses();
sendInventory();
state = ACTIVE;
} else {
if (state == SERVER) {
send(new Version.Builder().defaults().addrFrom(host).addrRecv(node).build());
}
break;
@ -155,7 +161,7 @@ public class Connection implements Runnable {
private void sendAddresses() {
List<NetworkAddress> addresses = ctx.getAddressRepository().getKnownAddresses(1000, streams);
send(new Addr.Builder().addresses(addresses).build());
sendingQueue.offer(new Addr.Builder().addresses(addresses).build());
}
private void sendInventory() {

View File

@ -16,86 +16,113 @@
package ch.dissem.bitmessage.networking;
import ch.dissem.bitmessage.entity.NetworkMessage;
import ch.dissem.bitmessage.entity.Version;
import ch.dissem.bitmessage.Context;
import ch.dissem.bitmessage.entity.payload.ObjectPayload;
import ch.dissem.bitmessage.entity.valueobject.NetworkAddress;
import ch.dissem.bitmessage.factory.Factory;
import ch.dissem.bitmessage.ports.NetworkMessageReceiver;
import ch.dissem.bitmessage.ports.NetworkMessageSender;
import ch.dissem.bitmessage.ports.NetworkHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import static ch.dissem.bitmessage.networking.Connection.State.CLIENT;
import static ch.dissem.bitmessage.networking.Connection.State.SERVER;
import static ch.dissem.bitmessage.networking.Connection.State.*;
/**
* Handles all the networky stuff.
*/
public class NetworkNode implements NetworkMessageSender, NetworkMessageReceiver {
public class NetworkNode implements NetworkHandler {
private final static Logger LOG = LoggerFactory.getLogger(NetworkNode.class);
/**
* This is only to be used where it's ignored
*/
private final static NetworkAddress LOCALHOST = new NetworkAddress.Builder().ipv4(127, 0, 0, 1).port(8444).build();
private final ExecutorService pool;
private final List<Connection> connections = new LinkedList<>();
private MessageListener listener;
public NetworkNode() {
pool = Executors.newCachedThreadPool();
// TODO: sending
// Thread sender = new Thread(new Runnable() {
// @Override
// public void run() {
// while (true) {
// try {
// NetworkMessage message = sendingQueue.take();
//
// try (Socket socket = getSocket(message.getTargetNode())) {
// message.write(socket.getOutputStream());
// } catch (Exception e) {
// e.printStackTrace();
// }
// } catch (InterruptedException e) {
// // Ignore?
// }
// }
// }
// }, "Sender");
// sender.setDaemon(true);
// sender.start();
}
@Override
public void registerListener(final int port, final MessageListener listener) throws IOException {
final ServerSocket serverSocket = new ServerSocket(port);
public void setListener(final MessageListener listener) {
if (this.listener != null) {
throw new IllegalStateException("Listener can only be set once");
}
this.listener = listener;
}
@Override
public void start() {
final Context ctx = Context.getInstance();
if (listener == null) {
throw new IllegalStateException("Listener must be set at start");
}
try {
final ServerSocket serverSocket = new ServerSocket(Context.getInstance().getPort());
pool.execute(new Runnable() {
@Override
public void run() {
NetworkAddress address = null;
try {
Socket socket = serverSocket.accept();
socket.setSoTimeout(20000);
pool.execute(new Connection(SERVER, socket, listener));
startConnection(new Connection(SERVER, socket, listener));
} catch (IOException e) {
LOG.debug(e.getMessage(), e);
}
}
});
Thread connectionManager = new Thread(new Runnable() {
@Override
public void run() {
while (!Thread.interrupted()) {
synchronized (connections) {
for (Iterator<Connection> iterator = connections.iterator(); iterator.hasNext(); ) {
Connection c = iterator.next();
if (c.getState() == DISCONNECTED) {
// Remove the current element from the iterator and the list.
iterator.remove();
}
}
}
if (connections.size() < 1) {
List<NetworkAddress> addresses = ctx.getAddressRepository().getKnownAddresses(8, ctx.getStreams());
for (NetworkAddress address : addresses) {
try {
startConnection(new Connection(CLIENT, new Socket(address.toInetAddress(), address.getPort()), listener));
} catch (IOException e) {
LOG.debug(e.getMessage(), e);
}
}
// FIXME: prevent connecting twice to the same node
}
try {
Thread.sleep(30000);
} catch (InterruptedException e) {
LOG.debug(e.getMessage(), e);
Thread.currentThread().interrupt();
}
}
}
}, "connection-manager");
connectionManager.start();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
private void startConnection(Connection c) {
synchronized (connections) {
connections.add(c);
}
pool.execute(c);
}
@Override
public void registerListener(final NetworkAddress node, final MessageListener listener) throws IOException {
pool.execute(new Connection(CLIENT, new Socket(node.toInetAddress(), node.getPort()), listener));
}
@Override
public void send(final NetworkAddress node, final NetworkMessage message) {
public void send(final ObjectPayload payload) {
// TODO: sendingQueue.add(message);
}
}

View File

@ -20,7 +20,7 @@ import ch.dissem.bitmessage.entity.NetworkMessage;
import ch.dissem.bitmessage.entity.Version;
import ch.dissem.bitmessage.entity.payload.ObjectPayload;
import ch.dissem.bitmessage.entity.valueobject.NetworkAddress;
import ch.dissem.bitmessage.ports.NetworkMessageReceiver;
import ch.dissem.bitmessage.ports.NetworkHandler;
import org.junit.Test;
/**
@ -33,13 +33,13 @@ public class NetworkNodeTest {
public void testSendMessage() throws Exception {
final Thread baseThread = Thread.currentThread();
NetworkNode net = new NetworkNode();
net.registerListener(localhost, new NetworkMessageReceiver.MessageListener() {
@Override
public void receive(ObjectPayload payload) {
System.out.println(payload);
baseThread.interrupt();
}
});
// net.setListener(localhost, new NetworkHandler.MessageListener() {
// @Override
// public void receive(ObjectPayload payload) {
// System.out.println(payload);
// baseThread.interrupt();
// }
// });
NetworkMessage ver = new NetworkMessage(
new Version.Builder()
.version(3)
@ -52,7 +52,7 @@ public class NetworkNodeTest {
.streams(1, 2)
.build()
);
net.send(localhost, ver);
// net.send(localhost, ver);
Thread.sleep(20000);
}
}

View File

@ -4,3 +4,7 @@ include 'domain'
include 'networking'
include 'inventory'
include 'demo'