Git-Mirrors/haveno
1
0
Fork 0
mirror of https://github.com/haveno-dex/haveno.git synced 2025-05-02 06:36:22 -04:00
Code Issues Projects Releases Packages Wiki Activity

Refactorings, small improvements/fixes

Browse source
This commit is contained in:
Manfred Karrer 2016-01-31 00:38:03 +01:00
parent 0b5cdc2581
commit 186f85c44e
5 changed files with 36 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

64
network/src/main/java/io/bitsquare/p2p/network/Connection.java
View file

@ -23,10 +23,7 @@ import java.io.*;
import java.net.Socket;
import java.net.SocketException;
import java.net.SocketTimeoutException;
import java.util.Date;
import java.util.List;
import java.util.Optional;
import java.util.UUID;
import java.util.*;
import java.util.concurrent.*;
import static com.google.common.base.Preconditions.checkArgument;
@ -39,8 +36,8 @@ import static com.google.common.base.Preconditions.checkNotNull;
public class Connection implements MessageListener {
private static final Logger log = LoggerFactory.getLogger(Connection.class);
private static final int MAX_MSG_SIZE = 100 * 1024; // 100 kb of compressed data
private static final int MAX_MSG_PER_SEC = 10; // With MAX_MSG_SIZE of 100kb results in bandwidth of 10 mbit/sec
private static final int MAX_MSG_PER_10SEC = 50; // With MAX_MSG_SIZE of 100kb results in bandwidth of 5 mbit/sec for 10 sec
private static final int MSG_THROTTLE_PER_SEC = 10; // With MAX_MSG_SIZE of 100kb results in bandwidth of 10 mbit/sec
private static final int MSG_THROTTLE_PER_10SEC = 50; // With MAX_MSG_SIZE of 100kb results in bandwidth of 5 mbit/sec for 10 sec
//timeout on blocking Socket operations like ServerSocket.accept() or SocketInputStream.read()
private static final int SOCKET_TIMEOUT = 30 * 60 * 1000; // 30 min.
@ -86,7 +83,7 @@ public class Connection implements MessageListener {
private final boolean useCompression = false;
private PeerType peerType;
private final ObjectProperty<NodeAddress> nodeAddressProperty = new SimpleObjectProperty<>();
private List<Long> messageTimeStamps = new ArrayList<>();
///////////////////////////////////////////////////////////////////////////////////////////
// Constructor
@ -204,6 +201,26 @@ public class Connection implements MessageListener {
sharedModel.reportIllegalRequest(illegalRequest);
}
public boolean violatesThrottleLimit() {
long now = System.currentTimeMillis();
boolean violated = false;
if (messageTimeStamps.size() >= MSG_THROTTLE_PER_SEC) {
// check if we got more than 10 msg per sec.
long compareTo = messageTimeStamps.get(messageTimeStamps.size() - MSG_THROTTLE_PER_SEC);
violated = now - compareTo < 1000;
}
if (messageTimeStamps.size() >= MSG_THROTTLE_PER_10SEC) {
// check if we got more than 50 msg per 10 sec.
long compareTo = messageTimeStamps.get(messageTimeStamps.size() - MSG_THROTTLE_PER_10SEC);
violated = violated || now - compareTo < 10000;
// we limit to max 50 entries
messageTimeStamps.remove(0);
}
messageTimeStamps.add(now);
return violated;
}
///////////////////////////////////////////////////////////////////////////////////////////
// MessageListener implementation
@ -428,7 +445,6 @@ public class Connection implements MessageListener {
private Date lastActivityDate;
private volatile boolean stopped;
private ConnectionListener.Reason shutDownReason;
private List<Long> messageTimeStamps = new CopyOnWriteArrayList<>();
public SharedModel(Connection connection, Socket socket) {
Log.traceCall();
@ -518,27 +534,6 @@ public class Connection implements MessageListener {
this.stopped = true;
}
private boolean tooManyMessages() {
long now = System.currentTimeMillis();
boolean exceeds = false;
if (messageTimeStamps.size() >= MAX_MSG_PER_SEC) {
// check if we got more than 10 msg per sec.
long compareTo = messageTimeStamps.get(messageTimeStamps.size() - MAX_MSG_PER_SEC);
exceeds = now - compareTo < 1000;
}
if (messageTimeStamps.size() >= MAX_MSG_PER_10SEC) {
// check if we got more than 50 msg per 10 sec.
long compareTo = messageTimeStamps.get(messageTimeStamps.size() - MAX_MSG_PER_10SEC);
exceeds = exceeds || now - compareTo < 10000;
// we limit to max 50 entries
messageTimeStamps.remove(0);
}
messageTimeStamps.add(now);
return exceeds;
}
public synchronized ConnectionListener.Reason getShutDownReason() {
return shutDownReason;
}
@ -635,6 +630,12 @@ public class Connection implements MessageListener {
sharedModel.reportIllegalRequest(IllegalRequest.MaxSizeExceeded);
return;
}
if (sharedModel.connection.violatesThrottleLimit()) {
sharedModel.reportIllegalRequest(IllegalRequest.ViolatedThrottleLimit);
return;
}
if (!(serializable instanceof Message)) {
sharedModel.reportIllegalRequest(IllegalRequest.InvalidDataType);
return;
@ -646,11 +647,6 @@ public class Connection implements MessageListener {
return;
}
if (sharedModel.tooManyMessages()) {
sharedModel.reportIllegalRequest(IllegalRequest.TooManyMessages);
return;
}
Connection connection = sharedModel.connection;
sharedModel.updateLastActivityDate();
if (message instanceof CloseConnectionMessage) {

2
network/src/main/java/io/bitsquare/p2p/network/IllegalRequest.java
View file

@ -4,7 +4,7 @@ public enum IllegalRequest {
MaxSizeExceeded(1),
InvalidDataType(0),
WrongNetworkId(0),
TooManyMessages(1);
ViolatedThrottleLimit(1);
public final int maxTolerance;