Changeset 171f0d2

Timestamp:

Jul 17, 2013 9:13:19 PM (7 years ago)

Author:

zab2 <zab2@…>

Branches:

master

Children:

1e2fb4b

Parents:

8937c4b

Message:

Implement RTT update on first ack

Files:

• apps/streaming/java/src/net/i2p/client/streaming/ConnectionOptions.java (modified) (10 diffs)
• apps/streaming/java/src/net/i2p/client/streaming/TCBShare.java (modified) (1 diff)
• router/java/src/net/i2p/router/RouterVersion.java (modified) (1 diff)

apps/streaming/java/src/net/i2p/client/streaming/ConnectionOptions.java

@@ -52 +52 @@
     private int _maxConns;
     private boolean _disableRejectLog;
-
+    private double _alpha;
+    private double _beta;
+    private double _kappa;
+    
+    /** state of a connection */
+    enum AckInit {
+        INIT, // just created
+        FIRST, // first received ack
+        STEADY 
+    }
+    
+    /** LOCKING: this */
+    private AckInit _initState = AckInit.INIT;
+    
     // NOTE - almost all the options are below, but see
     // I2PSocketOptions in ministreaming for a few more
@@ -66 +79 @@
     public static final int INACTIVITY_ACTION_SEND = 2;
     
+    /* 
+     * These values are specified in RFC 6298
+     * Do not change unless you know what you're doing
+     */
+    private static final double TCP_ALPHA = 1.0/8;
+    private static final double TCP_BETA = 1.0/4; 
+    private static final double TCP_KAPPA = 4;
+    private static final String PROP_TCP_ALPHA= "i2p.streaming.alpha";
+    private static final String PROP_TCP_BETA= "i2p.streaming.beta";
+    private static final String PROP_TCP_KAPPA = "i2p.streaming.kappa";
+    
+    private static final String PROP_INITIAL_RTO = "i2p.streaming.initialRTO";
+    private static final int INITIAL_RTO = 12000;
+    
     public static final String PROP_CONNECT_DELAY = "i2p.streaming.connectDelay";
     public static final String PROP_PROFILE = "i2p.streaming.profile";
     public static final String PROP_MAX_MESSAGE_SIZE = "i2p.streaming.maxMessageSize";
     public static final String PROP_MAX_RESENDS = "i2p.streaming.maxResends";
-    public static final String PROP_INITIAL_RTT = "i2p.streaming.initialRTT";
     public static final String PROP_INITIAL_RESEND_DELAY = "i2p.streaming.initialResendDelay";
     public static final String PROP_INITIAL_ACK_DELAY = "i2p.streaming.initialAckDelay";
@@ -296 +322 @@
             setConnectDelay(opts.getConnectDelay());
             setProfile(opts.getProfile());
+            setRTTDev(opts.getRTTDev());
             setRTT(opts.getRTT());
             setRequireFullySigned(opts.getRequireFullySigned());
@@ -333 +360 @@
         setProfile(getInt(opts, PROP_PROFILE, PROFILE_BULK));
         setMaxMessageSize(getInt(opts, PROP_MAX_MESSAGE_SIZE, DEFAULT_MAX_MESSAGE_SIZE));
-        setRTT(getInt(opts, PROP_INITIAL_RTT, DEFAULT_INITIAL_RTT));
         setReceiveWindow(getInt(opts, PROP_INITIAL_RECEIVE_WINDOW, 1));
         setResendDelay(getInt(opts, PROP_INITIAL_RESEND_DELAY, 1000));
@@ -361 +387 @@
         _maxTotalConnsPerDay = getInt(opts, PROP_MAX_TOTAL_CONNS_DAY, 0);
         _maxConns = getInt(opts, PROP_MAX_STREAMS, 0);
+        
+        _alpha = getDouble(opts, PROP_TCP_ALPHA, TCP_ALPHA);
+        _beta = getDouble(opts, PROP_TCP_BETA, TCP_BETA);
+        _kappa = getDouble(opts, PROP_TCP_KAPPA, TCP_KAPPA);
+        
+        _rto = getInt(opts, PROP_INITIAL_RTO, INITIAL_RTO);
     }
     
@@ -378 +410 @@
         if (opts.containsKey(PROP_MAX_MESSAGE_SIZE))
             setMaxMessageSize(getInt(opts, PROP_MAX_MESSAGE_SIZE, Packet.MAX_PAYLOAD_SIZE));
-        if (opts.containsKey(PROP_INITIAL_RTT))
-            setRTT(getInt(opts, PROP_INITIAL_RTT, DEFAULT_INITIAL_RTT));
         if (opts.containsKey(PROP_INITIAL_RECEIVE_WINDOW))
             setReceiveWindow(getInt(opts, PROP_INITIAL_RECEIVE_WINDOW, 1));
@@ -428 +458 @@
         if (opts.containsKey(PROP_MAX_STREAMS))
             _maxConns = getInt(opts, PROP_MAX_STREAMS, 0);
+        
+        _alpha = getDouble(opts, PROP_TCP_ALPHA, TCP_ALPHA);
+        _beta = getDouble(opts, PROP_TCP_BETA, TCP_BETA);
+        _kappa = getDouble(opts, PROP_TCP_KAPPA, TCP_KAPPA);
+        _rto = getInt(opts, PROP_INITIAL_RTO, INITIAL_RTO);
     }
     
@@ -518 +553 @@
     public int getRTT() { return _rtt; }
     public void setRTT(int ms) { 
-        if (_rto == 0) { // TODO: move this out
-            _rttDev = ms / 2;
-            _rto = ms + ms / 2;
-        }
         synchronized (_trend) {
             _trend[0] = _trend[1];
@@ -545 +576 @@
     void setRTTDev(int rttDev) { _rttDev = rttDev; }
     
+    /** 
+     * mark these options as loaded from cache.
+     * affects the calculation of RTO
+     */
+    synchronized void loadedFromCache() {
+        _initState = AckInit.STEADY;
+    }
+    
+    /** 
+     * computes RTO based on formula in RFC
+     */
+    synchronized void computeRTO() {
+        switch(_initState) {
+        case INIT :
+            throw new IllegalStateException();
+        case FIRST :
+            _rttDev = _rtt / 2;
+            _rto = _rtt + _rtt / 2;
+            break;
+        case STEADY :
+            _rto = _rtt + (int) (_rttDev * _kappa);
+            break;
+        }
+        
+        if (_rto < Connection.MIN_RESEND_DELAY) 
+            _rto = (int)Connection.MIN_RESEND_DELAY;
+        else if (_rto > Connection.MAX_RESEND_DELAY)
+            _rto = (int)Connection.MAX_RESEND_DELAY;
+    }
+    
     /**
      * If we have 3 consecutive rtt increases, we are trending upwards (1), or if we have
@@ -561 +622 @@
     }
     
-    /** rtt = rtt*RTT_DAMPENING + (1-RTT_DAMPENING)*currentPacketRTT */
-    /** This is the value specified in RFC 2988, let's try it */
-    private static final double RTT_DAMPENING = 0.875;
-    
-    public void updateRTT(int measuredValue) {
-        // the rttDev calculation matches that recommended in RFC 2988 (beta = 1/4)
-        _rttDev = _rttDev + (int)(0.25d*(Math.abs(measuredValue-_rtt)-_rttDev));
-        int smoothed = (int)(RTT_DAMPENING*_rtt + (1-RTT_DAMPENING)*measuredValue);        
-        // K = 4
-        _rto = smoothed + (_rttDev<<2);
-        if (_rto < Connection.MIN_RESEND_DELAY) 
-            _rto = (int)Connection.MIN_RESEND_DELAY;
-        else if (_rto > Connection.MAX_RESEND_DELAY)
-            _rto = (int)Connection.MAX_RESEND_DELAY;
-
-        setRTT(smoothed);
+    public synchronized void updateRTT(int measuredValue) {
+        switch(_initState) {
+        case INIT:
+            _initState = AckInit.FIRST;
+            setRTT(measuredValue); // no smoothing first sample
+            break;
+        case FIRST:
+            _initState = AckInit.STEADY; 
+        case STEADY:
+            // calculation matches that recommended in RFC 6298
+            _rttDev = (int) ((1-_beta) *_rttDev  + _beta * Math.abs(measuredValue-_rtt));
+            int smoothed = (int)((1-_alpha)*_rtt + _alpha*measuredValue);        
+            setRTT(smoothed);
+        }
+        computeRTO();
     }
     

apps/streaming/java/src/net/i2p/client/streaming/TCBShare.java

@@ -100 +100 @@
                        " wdw: " + wdw );
         }
+        opts.loadedFromCache();
         opts.setRTT(rtt);
         opts.setRTTDev(rttDev);
         opts.setWindowSize(wdw);
+        opts.computeRTO();
     }
 

router/java/src/net/i2p/router/RouterVersion.java

@@ -22 +22 @@
 
     /** for example "-test" */
-    public final static String EXTRA = "";
+    public final static String EXTRA = "-979";
     public final static String FULL_VERSION = VERSION + "-" + BUILD + EXTRA;
     public static void main(String args[]) {