Fixed system test and added some fixes for Java backwards compatibility
This commit is contained in:
parent
322bddcc4f
commit
aee5debdd2
@ -133,13 +133,12 @@ internal open class DefaultMessageListener(
|
|||||||
}
|
}
|
||||||
|
|
||||||
protected fun receive(objectMessage: ObjectMessage, broadcast: Broadcast) {
|
protected fun receive(objectMessage: ObjectMessage, broadcast: Broadcast) {
|
||||||
val tag = if (broadcast is V5Broadcast) broadcast.tag else null
|
val tag = (broadcast as? V5Broadcast)?.tag
|
||||||
for (subscription in ctx.addressRepository.getSubscriptions(broadcast.version)) {
|
ctx.addressRepository.getSubscriptions(broadcast.version)
|
||||||
if (tag != null && !Arrays.equals(tag, subscription.tag)) {
|
.filter { tag == null || Arrays.equals(tag, it.tag) }
|
||||||
continue
|
.forEach {
|
||||||
}
|
|
||||||
try {
|
try {
|
||||||
broadcast.decrypt(subscription.publicDecryptionKey)
|
broadcast.decrypt(it.publicDecryptionKey)
|
||||||
if (!objectMessage.isSignatureValid(broadcast.plaintext!!.from.pubkey!!)) {
|
if (!objectMessage.isSignatureValid(broadcast.plaintext!!.from.pubkey!!)) {
|
||||||
LOG.warn("Broadcast with IV " + objectMessage.inventoryVector + " was successfully decrypted, but signature check failed. Ignoring.")
|
LOG.warn("Broadcast with IV " + objectMessage.inventoryVector + " was successfully decrypted, but signature check failed. Ignoring.")
|
||||||
} else {
|
} else {
|
||||||
|
@ -36,14 +36,14 @@ import java.util.*
|
|||||||
import java.util.Collections
|
import java.util.Collections
|
||||||
import kotlin.collections.HashSet
|
import kotlin.collections.HashSet
|
||||||
|
|
||||||
fun message(encoding: Plaintext.Encoding, subject: String, body: String): ByteArray = when (encoding) {
|
internal fun message(encoding: Plaintext.Encoding, subject: String, body: String): ByteArray = when (encoding) {
|
||||||
SIMPLE -> "Subject:$subject\nBody:$body".toByteArray()
|
SIMPLE -> "Subject:$subject\nBody:$body".toByteArray()
|
||||||
EXTENDED -> Message.Builder().subject(subject).body(body).build().zip()
|
EXTENDED -> Message.Builder().subject(subject).body(body).build().zip()
|
||||||
TRIVIAL -> (subject + body).toByteArray()
|
TRIVIAL -> (subject + body).toByteArray()
|
||||||
IGNORE -> ByteArray(0)
|
IGNORE -> ByteArray(0)
|
||||||
}
|
}
|
||||||
|
|
||||||
fun ackData(type: Plaintext.Type, ackData: ByteArray?): ByteArray? {
|
internal fun ackData(type: Plaintext.Type, ackData: ByteArray?): ByteArray? {
|
||||||
if (ackData != null) {
|
if (ackData != null) {
|
||||||
return ackData
|
return ackData
|
||||||
} else if (type == MSG) {
|
} else if (type == MSG) {
|
||||||
@ -67,6 +67,7 @@ class Plaintext private constructor(
|
|||||||
val conversationId: UUID = UUID.randomUUID(),
|
val conversationId: UUID = UUID.randomUUID(),
|
||||||
var inventoryVector: InventoryVector? = null,
|
var inventoryVector: InventoryVector? = null,
|
||||||
var signature: ByteArray? = null,
|
var signature: ByteArray? = null,
|
||||||
|
sent: Long? = null,
|
||||||
val received: Long? = null,
|
val received: Long? = null,
|
||||||
var initialHash: ByteArray? = null,
|
var initialHash: ByteArray? = null,
|
||||||
ttl: Long = TTL.msg,
|
ttl: Long = TTL.msg,
|
||||||
@ -117,7 +118,7 @@ class Plaintext private constructor(
|
|||||||
}
|
}
|
||||||
|
|
||||||
val encoding: Encoding? by lazy { Encoding.fromCode(encodingCode) }
|
val encoding: Encoding? by lazy { Encoding.fromCode(encodingCode) }
|
||||||
var sent: Long? = null
|
var sent: Long? = sent
|
||||||
private set
|
private set
|
||||||
var retries: Int = 0
|
var retries: Int = 0
|
||||||
private set
|
private set
|
||||||
@ -145,9 +146,9 @@ class Plaintext private constructor(
|
|||||||
type = type,
|
type = type,
|
||||||
from = from,
|
from = from,
|
||||||
to = to,
|
to = to,
|
||||||
encoding = encoding.code,
|
encodingCode = encoding.code,
|
||||||
message = message,
|
message = message,
|
||||||
ackMessage = ackData(type, ackData),
|
ackData = ackData(type, ackData),
|
||||||
conversationId = conversationId,
|
conversationId = conversationId,
|
||||||
inventoryVector = inventoryVector,
|
inventoryVector = inventoryVector,
|
||||||
signature = signature,
|
signature = signature,
|
||||||
@ -214,9 +215,9 @@ class Plaintext private constructor(
|
|||||||
type = type,
|
type = type,
|
||||||
from = from,
|
from = from,
|
||||||
to = to,
|
to = to,
|
||||||
encoding = encoding.code,
|
encoding = encoding,
|
||||||
message = message(encoding, subject, body),
|
message = message(encoding, subject, body),
|
||||||
ackMessage = ackData(type, ackData),
|
ackData = ackData(type, ackData),
|
||||||
conversationId = conversationId,
|
conversationId = conversationId,
|
||||||
inventoryVector = null,
|
inventoryVector = null,
|
||||||
signature = null,
|
signature = null,
|
||||||
@ -248,6 +249,7 @@ class Plaintext private constructor(
|
|||||||
conversationId = builder.conversation ?: UUID.randomUUID(),
|
conversationId = builder.conversation ?: UUID.randomUUID(),
|
||||||
inventoryVector = builder.inventoryVector,
|
inventoryVector = builder.inventoryVector,
|
||||||
signature = builder.signature,
|
signature = builder.signature,
|
||||||
|
sent = builder.sent,
|
||||||
received = builder.received,
|
received = builder.received,
|
||||||
initialHash = null,
|
initialHash = null,
|
||||||
ttl = builder.ttl,
|
ttl = builder.ttl,
|
||||||
|
@ -195,7 +195,7 @@ class CryptoBox : Streamable {
|
|||||||
companion object {
|
companion object {
|
||||||
private val LOG = LoggerFactory.getLogger(CryptoBox::class.java)
|
private val LOG = LoggerFactory.getLogger(CryptoBox::class.java)
|
||||||
|
|
||||||
fun read(stream: InputStream, length: Int): CryptoBox {
|
@JvmStatic fun read(stream: InputStream, length: Int): CryptoBox {
|
||||||
val counter = AccessCounter()
|
val counter = AccessCounter()
|
||||||
return Builder()
|
return Builder()
|
||||||
.IV(Decode.bytes(stream, 16, counter))
|
.IV(Decode.bytes(stream, 16, counter))
|
||||||
|
@ -53,7 +53,7 @@ class GenericPayload(version: Long, override val stream: Long, val data: ByteArr
|
|||||||
}
|
}
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
fun read(version: Long, stream: Long, `is`: InputStream, length: Int): GenericPayload {
|
@JvmStatic fun read(version: Long, stream: Long, `is`: InputStream, length: Int): GenericPayload {
|
||||||
return GenericPayload(version, stream, Decode.bytes(`is`, length))
|
return GenericPayload(version, stream, Decode.bytes(`is`, length))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -58,7 +58,7 @@ class GetPubkey : ObjectPayload {
|
|||||||
}
|
}
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
fun read(`is`: InputStream, stream: Long, length: Int, version: Long): GetPubkey {
|
@JvmStatic fun read(`is`: InputStream, stream: Long, length: Int, version: Long): GetPubkey {
|
||||||
return GetPubkey(version, stream, Decode.bytes(`is`, length))
|
return GetPubkey(version, stream, Decode.bytes(`is`, length))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -96,7 +96,7 @@ class Msg : ObjectPayload, Encrypted, PlaintextHolder {
|
|||||||
companion object {
|
companion object {
|
||||||
val ACK_LENGTH = 32
|
val ACK_LENGTH = 32
|
||||||
|
|
||||||
fun read(`in`: InputStream, stream: Long, length: Int): Msg {
|
@JvmStatic fun read(`in`: InputStream, stream: Long, length: Int): Msg {
|
||||||
return Msg(stream, CryptoBox.read(`in`, length))
|
return Msg(stream, CryptoBox.read(`in`, length))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -26,7 +26,7 @@ enum class ObjectType constructor(val number: Long) {
|
|||||||
BROADCAST(3);
|
BROADCAST(3);
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
fun fromNumber(number: Long): ObjectType? {
|
@JvmStatic fun fromNumber(number: Long): ObjectType? {
|
||||||
return values().firstOrNull { it.number == number }
|
return values().firstOrNull { it.number == number }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -80,7 +80,7 @@ open class V2Pubkey constructor(version: Long, override val stream: Long, overri
|
|||||||
}
|
}
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
fun read(`in`: InputStream, stream: Long): V2Pubkey {
|
@JvmStatic fun read(`in`: InputStream, stream: Long): V2Pubkey {
|
||||||
return V2Pubkey(
|
return V2Pubkey(
|
||||||
version = 2,
|
version = 2,
|
||||||
stream = stream,
|
stream = stream,
|
||||||
|
@ -134,7 +134,7 @@ class V3Pubkey protected constructor(
|
|||||||
}
|
}
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
fun read(`is`: InputStream, stream: Long): V3Pubkey {
|
@JvmStatic fun read(`is`: InputStream, stream: Long): V3Pubkey {
|
||||||
return V3Pubkey(
|
return V3Pubkey(
|
||||||
version = 3,
|
version = 3,
|
||||||
stream = stream,
|
stream = stream,
|
||||||
|
@ -52,7 +52,7 @@ open class V4Broadcast : Broadcast {
|
|||||||
}
|
}
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
fun read(`in`: InputStream, stream: Long, length: Int): V4Broadcast {
|
@JvmStatic fun read(`in`: InputStream, stream: Long, length: Int): V4Broadcast {
|
||||||
return V4Broadcast(4, stream, CryptoBox.read(`in`, length), null)
|
return V4Broadcast(4, stream, CryptoBox.read(`in`, length), null)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -127,7 +127,7 @@ class V4Pubkey : Pubkey, Encrypted {
|
|||||||
}
|
}
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
fun read(`in`: InputStream, stream: Long, length: Int, encrypted: Boolean): V4Pubkey {
|
@JvmStatic fun read(`in`: InputStream, stream: Long, length: Int, encrypted: Boolean): V4Pubkey {
|
||||||
if (encrypted)
|
if (encrypted)
|
||||||
return V4Pubkey(stream,
|
return V4Pubkey(stream,
|
||||||
Decode.bytes(`in`, 32),
|
Decode.bytes(`in`, 32),
|
||||||
|
@ -51,7 +51,7 @@ class V5Broadcast : V4Broadcast {
|
|||||||
}
|
}
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
fun read(`is`: InputStream, stream: Long, length: Int): V5Broadcast {
|
@JvmStatic fun read(`is`: InputStream, stream: Long, length: Int): V5Broadcast {
|
||||||
return V5Broadcast(stream, Decode.bytes(`is`, 32), CryptoBox.read(`is`, length - 32))
|
return V5Broadcast(stream, Decode.bytes(`is`, 32), CryptoBox.read(`is`, length - 32))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -28,7 +28,8 @@ package ch.dissem.bitmessage.utils
|
|||||||
class Property private constructor(val name: String, val value: Any? = null, val properties: Array<Property> = emptyArray()) {
|
class Property private constructor(val name: String, val value: Any? = null, val properties: Array<Property> = emptyArray()) {
|
||||||
|
|
||||||
constructor(name: String, value: Any) : this(name = name, value = value, properties = emptyArray())
|
constructor(name: String, value: Any) : this(name = name, value = value, properties = emptyArray())
|
||||||
constructor(name: String, vararg properties: Property) : this(name, null, Array(properties.size, { i -> properties[i] }))
|
constructor(name: String, vararg properties: Property) : this(name, null, arrayOf(*properties))
|
||||||
|
constructor(name: String, properties: List<Property>) : this(name, null, properties.toTypedArray())
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns the property if available or `null` otherwise.
|
* Returns the property if available or `null` otherwise.
|
||||||
|
@ -40,7 +40,7 @@ import java.util.*
|
|||||||
* As Spongycastle can't be used on the Oracle JVM, and Bouncycastle doesn't work properly on Android (thanks, Google),
|
* As Spongycastle can't be used on the Oracle JVM, and Bouncycastle doesn't work properly on Android (thanks, Google),
|
||||||
* this is the Spongycastle implementation.
|
* this is the Spongycastle implementation.
|
||||||
*/
|
*/
|
||||||
class SpongyCryptography : AbstractCryptography(BouncyCastleProvider()) {
|
open class SpongyCryptography : AbstractCryptography(BouncyCastleProvider()) {
|
||||||
|
|
||||||
override fun crypt(encrypt: Boolean, data: ByteArray, key_e: ByteArray, initializationVector: ByteArray): ByteArray {
|
override fun crypt(encrypt: Boolean, data: ByteArray, key_e: ByteArray, initializationVector: ByteArray): ByteArray {
|
||||||
val cipher = PaddedBufferedBlockCipher(
|
val cipher = PaddedBufferedBlockCipher(
|
||||||
|
@ -33,5 +33,5 @@ dependencies {
|
|||||||
compile 'com.h2database:h2:1.4.194'
|
compile 'com.h2database:h2:1.4.194'
|
||||||
compile 'org.apache.commons:commons-lang3:3.5'
|
compile 'org.apache.commons:commons-lang3:3.5'
|
||||||
testCompile 'junit:junit:4.12'
|
testCompile 'junit:junit:4.12'
|
||||||
testCompile 'org.mockito:mockito-core:2.7.21'
|
testCompile 'com.nhaarman:mockito-kotlin:1.4.0'
|
||||||
}
|
}
|
||||||
|
@ -22,13 +22,11 @@ import ch.dissem.bitmessage.entity.Plaintext;
|
|||||||
import ch.dissem.bitmessage.networking.nio.NioNetworkHandler;
|
import ch.dissem.bitmessage.networking.nio.NioNetworkHandler;
|
||||||
import ch.dissem.bitmessage.ports.DefaultLabeler;
|
import ch.dissem.bitmessage.ports.DefaultLabeler;
|
||||||
import ch.dissem.bitmessage.ports.Labeler;
|
import ch.dissem.bitmessage.ports.Labeler;
|
||||||
import ch.dissem.bitmessage.ports.NetworkHandler;
|
|
||||||
import ch.dissem.bitmessage.repository.*;
|
import ch.dissem.bitmessage.repository.*;
|
||||||
import ch.dissem.bitmessage.utils.TTL;
|
import ch.dissem.bitmessage.utils.TTL;
|
||||||
import org.junit.After;
|
import org.junit.After;
|
||||||
import org.junit.Before;
|
import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.mockito.Mockito;
|
|
||||||
import org.slf4j.Logger;
|
import org.slf4j.Logger;
|
||||||
import org.slf4j.LoggerFactory;
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
@ -37,6 +35,9 @@ import java.util.concurrent.TimeUnit;
|
|||||||
|
|
||||||
import static ch.dissem.bitmessage.entity.payload.Pubkey.Feature.DOES_ACK;
|
import static ch.dissem.bitmessage.entity.payload.Pubkey.Feature.DOES_ACK;
|
||||||
import static ch.dissem.bitmessage.utils.UnixTime.MINUTE;
|
import static ch.dissem.bitmessage.utils.UnixTime.MINUTE;
|
||||||
|
import static com.nhaarman.mockito_kotlin.MockitoKt.spy;
|
||||||
|
import static com.nhaarman.mockito_kotlin.MockitoKt.timeout;
|
||||||
|
import static com.nhaarman.mockito_kotlin.MockitoKt.verify;
|
||||||
import static org.hamcrest.CoreMatchers.equalTo;
|
import static org.hamcrest.CoreMatchers.equalTo;
|
||||||
import static org.junit.Assert.assertThat;
|
import static org.junit.Assert.assertThat;
|
||||||
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.ArgumentMatchers.any;
|
||||||
@ -65,7 +66,7 @@ public class SystemTest {
|
|||||||
int bobPort = port++;
|
int bobPort = port++;
|
||||||
{
|
{
|
||||||
JdbcConfig aliceDB = new JdbcConfig("jdbc:h2:mem:alice;DB_CLOSE_DELAY=-1", "sa", "");
|
JdbcConfig aliceDB = new JdbcConfig("jdbc:h2:mem:alice;DB_CLOSE_DELAY=-1", "sa", "");
|
||||||
aliceLabeler = Mockito.spy(new DebugLabeler("Alice"));
|
aliceLabeler = spy(new DebugLabeler("Alice"));
|
||||||
TestListener aliceListener = new TestListener();
|
TestListener aliceListener = new TestListener();
|
||||||
alice = new BitmessageContext.Builder()
|
alice = new BitmessageContext.Builder()
|
||||||
.addressRepo(new JdbcAddressRepository(aliceDB))
|
.addressRepo(new JdbcAddressRepository(aliceDB))
|
||||||
@ -110,17 +111,17 @@ public class SystemTest {
|
|||||||
bob.shutdown();
|
bob.shutdown();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test(timeout = 60_000)
|
@Test(timeout = 120_000)
|
||||||
public void ensureAliceCanSendMessageToBob() throws Exception {
|
public void ensureAliceCanSendMessageToBob() throws Exception {
|
||||||
String originalMessage = UUID.randomUUID().toString();
|
String originalMessage = UUID.randomUUID().toString();
|
||||||
alice.send(aliceIdentity, new BitmessageAddress(bobIdentity.getAddress()), "Subject", originalMessage);
|
alice.send(aliceIdentity, new BitmessageAddress(bobIdentity.getAddress()), "Subject", originalMessage);
|
||||||
|
|
||||||
Plaintext plaintext = bobListener.get(15, TimeUnit.MINUTES);
|
Plaintext plaintext = bobListener.get(2, TimeUnit.MINUTES);
|
||||||
|
|
||||||
assertThat(plaintext.getType(), equalTo(Plaintext.Type.MSG));
|
assertThat(plaintext.getType(), equalTo(Plaintext.Type.MSG));
|
||||||
assertThat(plaintext.getText(), equalTo(originalMessage));
|
assertThat(plaintext.getText(), equalTo(originalMessage));
|
||||||
|
|
||||||
Mockito.verify(aliceLabeler, Mockito.timeout(TimeUnit.MINUTES.toMillis(15)).atLeastOnce())
|
verify(aliceLabeler, timeout(TimeUnit.MINUTES.toMillis(2)).atLeastOnce())
|
||||||
.markAsAcknowledged(any());
|
.markAsAcknowledged(any());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -75,7 +75,7 @@ class Connection(
|
|||||||
|
|
||||||
fun send(payload: MessagePayload) = io.send(payload)
|
fun send(payload: MessagePayload) = io.send(payload)
|
||||||
|
|
||||||
protected fun handleMessage(payload: MessagePayload) {
|
private fun handleMessage(payload: MessagePayload) {
|
||||||
when (state) {
|
when (state) {
|
||||||
State.CONNECTING -> initializer!!.handleCommand(payload)
|
State.CONNECTING -> initializer!!.handleCommand(payload)
|
||||||
State.ACTIVE -> receiveMessage(payload)
|
State.ACTIVE -> receiveMessage(payload)
|
||||||
@ -108,7 +108,7 @@ class Connection(
|
|||||||
private fun receiveMessage(objectMessage: ObjectMessage) {
|
private fun receiveMessage(objectMessage: ObjectMessage) {
|
||||||
requestedObjects.remove(objectMessage.inventoryVector)
|
requestedObjects.remove(objectMessage.inventoryVector)
|
||||||
if (ctx.inventory.contains(objectMessage)) {
|
if (ctx.inventory.contains(objectMessage)) {
|
||||||
LOG.trace("Received object " + objectMessage.inventoryVector + " - already in inventory")
|
LOG.trace("Received object ${objectMessage.inventoryVector} - already in inventory")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
try {
|
try {
|
||||||
@ -122,7 +122,7 @@ class Connection(
|
|||||||
LOG.warn(e.message)
|
LOG.warn(e.message)
|
||||||
// DebugUtils.saveToFile(objectMessage); // this line must not be committed active
|
// DebugUtils.saveToFile(objectMessage); // this line must not be committed active
|
||||||
} catch (e: IOException) {
|
} catch (e: IOException) {
|
||||||
LOG.error("Stream " + objectMessage.stream + ", object type " + objectMessage.type + ": " + e.message, e)
|
LOG.error("Stream ${objectMessage.stream}, object type ${objectMessage.type}: ${e.message}", e)
|
||||||
} finally {
|
} finally {
|
||||||
if (commonRequestedObjects.remove(objectMessage.inventoryVector) == null) {
|
if (commonRequestedObjects.remove(objectMessage.inventoryVector) == null) {
|
||||||
LOG.debug("Received object that wasn't requested.")
|
LOG.debug("Received object that wasn't requested.")
|
||||||
@ -131,7 +131,7 @@ class Connection(
|
|||||||
}
|
}
|
||||||
|
|
||||||
private fun receiveMessage(addr: Addr) {
|
private fun receiveMessage(addr: Addr) {
|
||||||
LOG.trace("Received " + addr.addresses.size + " addresses.")
|
LOG.trace("Received ${addr.addresses.size} addresses.")
|
||||||
ctx.nodeRegistry.offerAddresses(addr.addresses)
|
ctx.nodeRegistry.offerAddresses(addr.addresses)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -431,7 +431,7 @@ class NioNetworkHandler : NetworkHandler, InternalContext.ContextHolder {
|
|||||||
}
|
}
|
||||||
return Property("network",
|
return Property("network",
|
||||||
Property("connectionManager", if (isRunning) "running" else "stopped"),
|
Property("connectionManager", if (isRunning) "running" else "stopped"),
|
||||||
Property("connections", *streamProperties.toTypedArray()),
|
Property("connections", streamProperties),
|
||||||
Property("requestedObjects", requestedObjects.size)
|
Property("requestedObjects", requestedObjects.size)
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
@ -1,49 +0,0 @@
|
|||||||
/*
|
|
||||||
* 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.networking;
|
|
||||||
|
|
||||||
import ch.dissem.bitmessage.entity.valueobject.NetworkAddress;
|
|
||||||
import ch.dissem.bitmessage.ports.NodeRegistry;
|
|
||||||
|
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Empty {@link NodeRegistry} that doesn't do anything, but shouldn't break things either.
|
|
||||||
*/
|
|
||||||
class TestNodeRegistry implements NodeRegistry {
|
|
||||||
private List<NetworkAddress> nodes;
|
|
||||||
|
|
||||||
public TestNodeRegistry(NetworkAddress... nodes) {
|
|
||||||
this.nodes = Arrays.asList(nodes);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void clear() {
|
|
||||||
// no op
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public List<NetworkAddress> getKnownAddresses(int limit, long... streams) {
|
|
||||||
return nodes;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void offerAddresses(List<NetworkAddress> addresses) {
|
|
||||||
// Ignore
|
|
||||||
}
|
|
||||||
}
|
|
@ -0,0 +1,41 @@
|
|||||||
|
/*
|
||||||
|
* 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.networking
|
||||||
|
|
||||||
|
import ch.dissem.bitmessage.entity.valueobject.NetworkAddress
|
||||||
|
import ch.dissem.bitmessage.ports.NodeRegistry
|
||||||
|
|
||||||
|
import java.util.Arrays
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Empty [NodeRegistry] that doesn't do anything, but shouldn't break things either.
|
||||||
|
*/
|
||||||
|
internal class TestNodeRegistry(vararg nodes: NetworkAddress) : NodeRegistry {
|
||||||
|
private val nodes: List<NetworkAddress> = listOf(*nodes)
|
||||||
|
|
||||||
|
override fun clear() {
|
||||||
|
// no op
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun getKnownAddresses(limit: Int, vararg streams: Long): List<NetworkAddress> {
|
||||||
|
return nodes
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun offerAddresses(addresses: List<NetworkAddress>) {
|
||||||
|
// Ignore
|
||||||
|
}
|
||||||
|
}
|
@ -98,7 +98,7 @@ class JdbcMessageRepository(private val config: JdbcConfig) : AbstractMessageRep
|
|||||||
builder.id(id)
|
builder.id(id)
|
||||||
builder.IV(InventoryVector.fromHash(iv))
|
builder.IV(InventoryVector.fromHash(iv))
|
||||||
builder.from(ctx.addressRepository.getAddress(rs.getString("sender"))!!)
|
builder.from(ctx.addressRepository.getAddress(rs.getString("sender"))!!)
|
||||||
builder.to(ctx.addressRepository.getAddress(rs.getString("recipient")))
|
rs.getString("recipient")?.let { builder.to(ctx.addressRepository.getAddress(it)) }
|
||||||
builder.ackData(rs.getBytes("ack_data"))
|
builder.ackData(rs.getBytes("ack_data"))
|
||||||
builder.sent(rs.getObject("sent") as Long?)
|
builder.sent(rs.getObject("sent") as Long?)
|
||||||
builder.received(rs.getObject("received") as Long?)
|
builder.received(rs.getObject("received") as Long?)
|
||||||
|
Loading…
Reference in New Issue
Block a user