Merge branch 'for-linus' of git://git.kernel.org/pub/scm/linux/kernel/git/dtor/input
[linux-drm-fsl-dcu.git] / net / dccp / ccids / ccid2.c
index 4a7b8751256059ceea2f8efe419550bcd3e826c6..248d20f4c7c4edb45d94715b244331b760e77775 100644 (file)
  */
 
 /*
- * This implementation should follow: draft-ietf-dccp-ccid2-10.txt
+ * This implementation should follow RFC 4341
  *
  * BUGS:
  * - sequence number wrapping
- * - jiffies wrapping
  */
 
-#include <linux/config.h>
 #include "../ccid.h"
 #include "../dccp.h"
 #include "ccid2.h"
 
-static int ccid2_debug;
-
-#if 0
-#define CCID2_DEBUG
-#endif
-
-#ifdef CCID2_DEBUG
-#define ccid2_pr_debug(format, a...) \
-        do { if (ccid2_debug) \
-                printk(KERN_DEBUG "%s: " format, __FUNCTION__, ##a); \
-        } while (0)
-#else
-#define ccid2_pr_debug(format, a...)
-#endif
-
-static const int ccid2_seq_len = 128;
 
-static inline struct ccid2_hc_tx_sock *ccid2_hc_tx_sk(const struct sock *sk)
-{
-       return dccp_sk(sk)->dccps_hc_tx_ccid_private;
-}
-
-static inline struct ccid2_hc_rx_sock *ccid2_hc_rx_sk(const struct sock *sk)
-{
-       return dccp_sk(sk)->dccps_hc_rx_ccid_private;
-}
+#ifdef CONFIG_IP_DCCP_CCID2_DEBUG
+static int ccid2_debug;
+#define ccid2_pr_debug(format, a...)   DCCP_PR_DEBUG(ccid2_debug, format, ##a)
 
-#ifdef CCID2_DEBUG
 static void ccid2_hc_tx_check_sanity(const struct ccid2_hc_tx_sock *hctx)
 {
        int len = 0;
-       struct ccid2_seq *seqp;
        int pipe = 0;
-
-       seqp = hctx->ccid2hctx_seqh;
+       struct ccid2_seq *seqp = hctx->ccid2hctx_seqh;
 
        /* there is data in the chain */
        if (seqp != hctx->ccid2hctx_seqt) {
@@ -79,17 +52,16 @@ static void ccid2_hc_tx_check_sanity(const struct ccid2_hc_tx_sock *hctx)
                        pipe++;
 
                while (seqp != hctx->ccid2hctx_seqt) {
-                       struct ccid2_seq *prev;
+                       struct ccid2_seq *prev = seqp->ccid2s_prev;
 
-                       prev = seqp->ccid2s_prev;
                        len++;
                        if (!prev->ccid2s_acked)
                                pipe++;
 
                        /* packets are sent sequentially */
                        BUG_ON(seqp->ccid2s_seq <= prev->ccid2s_seq);
-                       BUG_ON(seqp->ccid2s_sent < prev->ccid2s_sent);
-                       BUG_ON(len > ccid2_seq_len);
+                       BUG_ON(time_before(seqp->ccid2s_sent,
+                                          prev->ccid2s_sent));
 
                        seqp = prev;
                }
@@ -101,18 +73,59 @@ static void ccid2_hc_tx_check_sanity(const struct ccid2_hc_tx_sock *hctx)
        do {
                seqp = seqp->ccid2s_prev;
                len++;
-               BUG_ON(len > ccid2_seq_len);
-       } while(seqp != hctx->ccid2hctx_seqh);
+       } while (seqp != hctx->ccid2hctx_seqh);
 
-       BUG_ON(len != ccid2_seq_len);
        ccid2_pr_debug("total len=%d\n", len);
+       BUG_ON(len != hctx->ccid2hctx_seqbufc * CCID2_SEQBUF_LEN);
 }
 #else
-#define ccid2_hc_tx_check_sanity(hctx) do {} while (0)
+#define ccid2_pr_debug(format, a...)
+#define ccid2_hc_tx_check_sanity(hctx)
 #endif
 
-static int ccid2_hc_tx_send_packet(struct sock *sk,
-                                  struct sk_buff *skb, int len)
+static int ccid2_hc_tx_alloc_seq(struct ccid2_hc_tx_sock *hctx, int num,
+                                gfp_t gfp)
+{
+       struct ccid2_seq *seqp;
+       int i;
+
+       /* check if we have space to preserve the pointer to the buffer */
+       if (hctx->ccid2hctx_seqbufc >= (sizeof(hctx->ccid2hctx_seqbuf) /
+                                       sizeof(struct ccid2_seq*)))
+               return -ENOMEM;
+
+       /* allocate buffer and initialize linked list */
+       seqp = kmalloc(sizeof(*seqp) * num, gfp);
+       if (seqp == NULL)
+               return -ENOMEM;
+
+       for (i = 0; i < (num - 1); i++) {
+               seqp[i].ccid2s_next = &seqp[i + 1];
+               seqp[i + 1].ccid2s_prev = &seqp[i];
+       }
+       seqp[num - 1].ccid2s_next = seqp;
+       seqp->ccid2s_prev = &seqp[num - 1];
+
+       /* This is the first allocation.  Initiate the head and tail.  */
+       if (hctx->ccid2hctx_seqbufc == 0)
+               hctx->ccid2hctx_seqh = hctx->ccid2hctx_seqt = seqp;
+       else {
+               /* link the existing list with the one we just created */
+               hctx->ccid2hctx_seqh->ccid2s_next = seqp;
+               seqp->ccid2s_prev = hctx->ccid2hctx_seqh;
+
+               hctx->ccid2hctx_seqt->ccid2s_prev = &seqp[num - 1];
+               seqp[num - 1].ccid2s_next = hctx->ccid2hctx_seqt;
+       }
+
+       /* store the original pointer to the buffer so we can free it */
+       hctx->ccid2hctx_seqbuf[hctx->ccid2hctx_seqbufc] = seqp;
+       hctx->ccid2hctx_seqbufc++;
+
+       return 0;
+}
+
+static int ccid2_hc_tx_send_packet(struct sock *sk, struct sk_buff *skb)
 {
        struct ccid2_hc_tx_sock *hctx;
 
@@ -126,7 +139,7 @@ static int ccid2_hc_tx_send_packet(struct sock *sk,
                return 0;
        }
 
-        hctx = ccid2_hc_tx_sk(sk);
+       hctx = ccid2_hc_tx_sk(sk);
 
        ccid2_pr_debug("pipe=%d cwnd=%d\n", hctx->ccid2hctx_pipe,
                       hctx->ccid2hctx_cwnd);
@@ -139,7 +152,7 @@ static int ccid2_hc_tx_send_packet(struct sock *sk,
                }
        }
 
-       return 100; /* XXX */
+       return 1; /* XXX CCID should dequeue when ready instead of polling */
 }
 
 static void ccid2_change_l_ack_ratio(struct sock *sk, int val)
@@ -151,7 +164,7 @@ static void ccid2_change_l_ack_ratio(struct sock *sk, int val)
         * -sorbo.
         */
        if (val != 2) {
-               struct ccid2_hc_tx_sock *hctx = ccid2_hc_tx_sk(sk);
+               const struct ccid2_hc_tx_sock *hctx = ccid2_hc_tx_sk(sk);
                int max = hctx->ccid2hctx_cwnd / 2;
 
                /* round up */
@@ -167,10 +180,8 @@ static void ccid2_change_l_ack_ratio(struct sock *sk, int val)
        dp->dccps_l_ack_ratio = val;
 }
 
-static void ccid2_change_cwnd(struct sock *sk, int val)
+static void ccid2_change_cwnd(struct ccid2_hc_tx_sock *hctx, int val)
 {
-       struct ccid2_hc_tx_sock *hctx = ccid2_hc_tx_sk(sk);
-
        if (val == 0)
                val = 1;
 
@@ -181,6 +192,17 @@ static void ccid2_change_cwnd(struct sock *sk, int val)
        hctx->ccid2hctx_cwnd = val;
 }
 
+static void ccid2_change_srtt(struct ccid2_hc_tx_sock *hctx, long val)
+{
+       ccid2_pr_debug("change SRTT to %ld\n", val);
+       hctx->ccid2hctx_srtt = val;
+}
+
+static void ccid2_change_pipe(struct ccid2_hc_tx_sock *hctx, long val)
+{
+       hctx->ccid2hctx_pipe = val;
+}
+
 static void ccid2_start_rto_timer(struct sock *sk);
 
 static void ccid2_hc_tx_rto_expire(unsigned long data)
@@ -189,9 +211,6 @@ static void ccid2_hc_tx_rto_expire(unsigned long data)
        struct ccid2_hc_tx_sock *hctx = ccid2_hc_tx_sk(sk);
        long s;
 
-       /* XXX I don't think i'm locking correctly
-        * -sorbo.
-        */
        bh_lock_sock(sk);
        if (sock_owned_by_user(sk)) {
                sk_reset_timer(sk, &hctx->ccid2hctx_rtotimer,
@@ -213,11 +232,11 @@ static void ccid2_hc_tx_rto_expire(unsigned long data)
        ccid2_start_rto_timer(sk);
 
        /* adjust pipe, cwnd etc */
-       hctx->ccid2hctx_pipe = 0;
+       ccid2_change_pipe(hctx, 0);
        hctx->ccid2hctx_ssthresh = hctx->ccid2hctx_cwnd >> 1;
        if (hctx->ccid2hctx_ssthresh < 2)
                hctx->ccid2hctx_ssthresh = 2;
-       ccid2_change_cwnd(sk, 1);
+       ccid2_change_cwnd(hctx, 1);
 
        /* clear state about stuff we sent */
        hctx->ccid2hctx_seqt    = hctx->ccid2hctx_seqh;
@@ -234,7 +253,7 @@ static void ccid2_hc_tx_rto_expire(unsigned long data)
        ccid2_hc_tx_check_sanity(hctx);
 out:
        bh_unlock_sock(sk);
-/*     sock_put(sk); */
+       sock_put(sk);
 }
 
 static void ccid2_start_rto_timer(struct sock *sk)
@@ -248,17 +267,18 @@ static void ccid2_start_rto_timer(struct sock *sk)
                       jiffies + hctx->ccid2hctx_rto);
 }
 
-static void ccid2_hc_tx_packet_sent(struct sock *sk, int more, int len)
+static void ccid2_hc_tx_packet_sent(struct sock *sk, int more, unsigned int len)
 {
        struct dccp_sock *dp = dccp_sk(sk);
        struct ccid2_hc_tx_sock *hctx = ccid2_hc_tx_sk(sk);
+       struct ccid2_seq *next;
        u64 seq;
 
        ccid2_hc_tx_check_sanity(hctx);
 
        BUG_ON(!hctx->ccid2hctx_sendwait);
        hctx->ccid2hctx_sendwait = 0;
-       hctx->ccid2hctx_pipe++;
+       ccid2_change_pipe(hctx, hctx->ccid2hctx_pipe + 1);
        BUG_ON(hctx->ccid2hctx_pipe < 0);
 
        /* There is an issue.  What if another packet is sent between
@@ -271,15 +291,23 @@ static void ccid2_hc_tx_packet_sent(struct sock *sk, int more, int len)
        hctx->ccid2hctx_seqh->ccid2s_seq   = seq;
        hctx->ccid2hctx_seqh->ccid2s_acked = 0;
        hctx->ccid2hctx_seqh->ccid2s_sent  = jiffies;
-       hctx->ccid2hctx_seqh = hctx->ccid2hctx_seqh->ccid2s_next;
 
-       ccid2_pr_debug("cwnd=%d pipe=%d\n", hctx->ccid2hctx_cwnd,
-                      hctx->ccid2hctx_pipe);
+       next = hctx->ccid2hctx_seqh->ccid2s_next;
+       /* check if we need to alloc more space */
+       if (next == hctx->ccid2hctx_seqt) {
+               int rc;
 
-       if (hctx->ccid2hctx_seqh == hctx->ccid2hctx_seqt) {
-               /* XXX allocate more space */
-               WARN_ON(1);
+               ccid2_pr_debug("allocating more space in history\n");
+               rc = ccid2_hc_tx_alloc_seq(hctx, CCID2_SEQBUF_LEN, GFP_KERNEL);
+               BUG_ON(rc); /* XXX what do we do? */
+
+               next = hctx->ccid2hctx_seqh->ccid2s_next;
+               BUG_ON(next == hctx->ccid2hctx_seqt);
        }
+       hctx->ccid2hctx_seqh = next;
+
+       ccid2_pr_debug("cwnd=%d pipe=%d\n", hctx->ccid2hctx_cwnd,
+                      hctx->ccid2hctx_pipe);
 
        hctx->ccid2hctx_sent++;
 
@@ -288,49 +316,46 @@ static void ccid2_hc_tx_packet_sent(struct sock *sk, int more, int len)
        /* We had an ack loss in this window... */
        if (hctx->ccid2hctx_ackloss) {
                if (hctx->ccid2hctx_arsent >= hctx->ccid2hctx_cwnd) {
-                       hctx->ccid2hctx_arsent = 0;
-                       hctx->ccid2hctx_ackloss = 0;
+                       hctx->ccid2hctx_arsent  = 0;
+                       hctx->ccid2hctx_ackloss = 0;
                }
-       }
-       /* No acks lost up to now... */
-       else {
+       } else {
+               /* No acks lost up to now... */
                /* decrease ack ratio if enough packets were sent */
                if (dp->dccps_l_ack_ratio > 1) {
                        /* XXX don't calculate denominator each time */
-                       int denom;
+                       int denom = dp->dccps_l_ack_ratio * dp->dccps_l_ack_ratio -
+                                   dp->dccps_l_ack_ratio;
 
-                       denom = dp->dccps_l_ack_ratio * dp->dccps_l_ack_ratio -
-                               dp->dccps_l_ack_ratio;
                        denom = hctx->ccid2hctx_cwnd * hctx->ccid2hctx_cwnd / denom;
 
                        if (hctx->ccid2hctx_arsent >= denom) {
                                ccid2_change_l_ack_ratio(sk, dp->dccps_l_ack_ratio - 1);
                                hctx->ccid2hctx_arsent = 0;
                        }
-               }
-               /* we can't increase ack ratio further [1] */
-               else {
+               } else {
+                       /* we can't increase ack ratio further [1] */
                        hctx->ccid2hctx_arsent = 0; /* or maybe set it to cwnd*/
                }
        }
 
        /* setup RTO timer */
-       if (!timer_pending(&hctx->ccid2hctx_rtotimer)) {
+       if (!timer_pending(&hctx->ccid2hctx_rtotimer))
                ccid2_start_rto_timer(sk);
-       }
-#ifdef CCID2_DEBUG
+
+#ifdef CONFIG_IP_DCCP_CCID2_DEBUG
        ccid2_pr_debug("pipe=%d\n", hctx->ccid2hctx_pipe);
-       ccid2_pr_debug("Sent: seq=%llu\n", seq);
+       ccid2_pr_debug("Sent: seq=%llu\n", (unsigned long long)seq);
        do {
                struct ccid2_seq *seqp = hctx->ccid2hctx_seqt;
 
                while (seqp != hctx->ccid2hctx_seqh) {
                        ccid2_pr_debug("out seq=%llu acked=%d time=%lu\n",
-                                      seqp->ccid2s_seq, seqp->ccid2s_acked,
-                                      seqp->ccid2s_sent);
+                                      (unsigned long long)seqp->ccid2s_seq,
+                                      seqp->ccid2s_acked, seqp->ccid2s_sent);
                        seqp = seqp->ccid2s_next;
                }
-       } while(0);
+       } while (0);
        ccid2_pr_debug("=========\n");
        ccid2_hc_tx_check_sanity(hctx);
 #endif
@@ -343,13 +368,13 @@ static void ccid2_hc_tx_packet_sent(struct sock *sk, int more, int len)
 static int ccid2_ackvector(struct sock *sk, struct sk_buff *skb, int offset,
                           unsigned char **vec, unsigned char *veclen)
 {
-        const struct dccp_hdr *dh = dccp_hdr(skb);
-        unsigned char *options = (unsigned char *)dh + dccp_hdr_len(skb);
-        unsigned char *opt_ptr;
-        const unsigned char *opt_end = (unsigned char *)dh +
-                                        (dh->dccph_doff * 4);
-        unsigned char opt, len;
-        unsigned char *value;
+       const struct dccp_hdr *dh = dccp_hdr(skb);
+       unsigned char *options = (unsigned char *)dh + dccp_hdr_len(skb);
+       unsigned char *opt_ptr;
+       const unsigned char *opt_end = (unsigned char *)dh +
+                                       (dh->dccph_doff * 4);
+       unsigned char opt, len;
+       unsigned char *value;
 
        BUG_ON(offset < 0);
        options += offset;
@@ -358,29 +383,29 @@ static int ccid2_ackvector(struct sock *sk, struct sk_buff *skb, int offset,
                return -1;
 
        while (opt_ptr != opt_end) {
-                opt   = *opt_ptr++;
-                len   = 0;
-                value = NULL;
-
-                /* Check if this isn't a single byte option */
-                if (opt > DCCPO_MAX_RESERVED) {
-                        if (opt_ptr == opt_end)
-                                goto out_invalid_option;
-
-                        len = *opt_ptr++;
-                        if (len < 3)
-                                goto out_invalid_option;
-                        /*
-                         * Remove the type and len fields, leaving
-                         * just the value size
-                         */
-                        len     -= 2;
-                        value   = opt_ptr;
-                        opt_ptr += len;
-
-                        if (opt_ptr > opt_end)
-                                goto out_invalid_option;
-                }
+               opt   = *opt_ptr++;
+               len   = 0;
+               value = NULL;
+
+               /* Check if this isn't a single byte option */
+               if (opt > DCCPO_MAX_RESERVED) {
+                       if (opt_ptr == opt_end)
+                               goto out_invalid_option;
+
+                       len = *opt_ptr++;
+                       if (len < 3)
+                               goto out_invalid_option;
+                       /*
+                        * Remove the type and len fields, leaving
+                        * just the value size
+                        */
+                       len     -= 2;
+                       value   = opt_ptr;
+                       opt_ptr += len;
+
+                       if (opt_ptr > opt_end)
+                               goto out_invalid_option;
+               }
 
                switch (opt) {
                case DCCPO_ACK_VECTOR_0:
@@ -388,25 +413,26 @@ static int ccid2_ackvector(struct sock *sk, struct sk_buff *skb, int offset,
                        *vec    = value;
                        *veclen = len;
                        return offset + (opt_ptr - options);
-                       break;
                }
        }
 
        return -1;
 
 out_invalid_option:
-       BUG_ON(1); /* should never happen... options were previously parsed ! */
+       DCCP_BUG("Invalid option - this should not happen (previous parsing)!");
        return -1;
 }
 
-static void ccid2_hc_tx_kill_rto_timer(struct ccid2_hc_tx_sock *hctx)
+static void ccid2_hc_tx_kill_rto_timer(struct sock *sk)
 {
-       if (del_timer(&hctx->ccid2hctx_rtotimer))
-               ccid2_pr_debug("deleted RTO timer\n");
+       struct ccid2_hc_tx_sock *hctx = ccid2_hc_tx_sk(sk);
+
+       sk_stop_timer(sk, &hctx->ccid2hctx_rtotimer);
+       ccid2_pr_debug("deleted RTO timer\n");
 }
 
 static inline void ccid2_new_ack(struct sock *sk,
-                                struct ccid2_seq *seqp,
+                                struct ccid2_seq *seqp,
                                 unsigned int *maxincr)
 {
        struct ccid2_hc_tx_sock *hctx = ccid2_hc_tx_sk(sk);
@@ -420,42 +446,42 @@ static inline void ccid2_new_ack(struct sock *sk,
                        /* increase every 2 acks */
                        hctx->ccid2hctx_ssacks++;
                        if (hctx->ccid2hctx_ssacks == 2) {
-                               ccid2_change_cwnd(sk, hctx->ccid2hctx_cwnd + 1);
+                               ccid2_change_cwnd(hctx, hctx->ccid2hctx_cwnd+1);
                                hctx->ccid2hctx_ssacks = 0;
                                *maxincr = *maxincr - 1;
                        }
-               }
-               /* increased cwnd enough for this single ack */
-               else {
+               } else {
+                       /* increased cwnd enough for this single ack */
                        hctx->ccid2hctx_ssacks = 0;
                }
-       }
-       else {
+       } else {
                hctx->ccid2hctx_ssacks = 0;
                hctx->ccid2hctx_acks++;
 
                if (hctx->ccid2hctx_acks >= hctx->ccid2hctx_cwnd) {
-                       ccid2_change_cwnd(sk, hctx->ccid2hctx_cwnd + 1);
+                       ccid2_change_cwnd(hctx, hctx->ccid2hctx_cwnd + 1);
                        hctx->ccid2hctx_acks = 0;
                }
        }
 
        /* update RTO */
        if (hctx->ccid2hctx_srtt == -1 ||
-           (jiffies - hctx->ccid2hctx_lastrtt) >= hctx->ccid2hctx_srtt) {
-               unsigned long r = jiffies - seqp->ccid2s_sent;
+           time_after(jiffies, hctx->ccid2hctx_lastrtt + hctx->ccid2hctx_srtt)) {
+               unsigned long r = (long)jiffies - (long)seqp->ccid2s_sent;
                int s;
 
                /* first measurement */
                if (hctx->ccid2hctx_srtt == -1) {
                        ccid2_pr_debug("R: %lu Time=%lu seq=%llu\n",
-                                      r, jiffies, seqp->ccid2s_seq);
-                       hctx->ccid2hctx_srtt = r;
+                                      r, jiffies,
+                                      (unsigned long long)seqp->ccid2s_seq);
+                       ccid2_change_srtt(hctx, r);
                        hctx->ccid2hctx_rttvar = r >> 1;
-               }
-               else {
+               } else {
                        /* RTTVAR */
                        long tmp = hctx->ccid2hctx_srtt - r;
+                       long srtt;
+
                        if (tmp < 0)
                                tmp *= -1;
 
@@ -465,10 +491,12 @@ static inline void ccid2_new_ack(struct sock *sk,
                        hctx->ccid2hctx_rttvar += tmp;
 
                        /* SRTT */
-                       hctx->ccid2hctx_srtt *= 7;
-                       hctx->ccid2hctx_srtt >>= 3;
+                       srtt = hctx->ccid2hctx_srtt;
+                       srtt *= 7;
+                       srtt >>= 3;
                        tmp = r >> 3;
-                       hctx->ccid2hctx_srtt += tmp;
+                       srtt += tmp;
+                       ccid2_change_srtt(hctx, srtt);
                }
                s = hctx->ccid2hctx_rttvar << 2;
                /* clock granularity is 1 when based on jiffies */
@@ -490,23 +518,41 @@ static inline void ccid2_new_ack(struct sock *sk,
                hctx->ccid2hctx_lastrtt = jiffies;
 
                ccid2_pr_debug("srtt: %ld rttvar: %ld rto: %ld (HZ=%d) R=%lu\n",
-                              hctx->ccid2hctx_srtt, hctx->ccid2hctx_rttvar,
-                              hctx->ccid2hctx_rto, HZ, r);
+                              hctx->ccid2hctx_srtt, hctx->ccid2hctx_rttvar,
+                              hctx->ccid2hctx_rto, HZ, r);
                hctx->ccid2hctx_sent = 0;
        }
 
        /* we got a new ack, so re-start RTO timer */
-       ccid2_hc_tx_kill_rto_timer(hctx);
+       ccid2_hc_tx_kill_rto_timer(sk);
        ccid2_start_rto_timer(sk);
 }
 
-static void ccid2_hc_tx_dec_pipe(struct ccid2_hc_tx_sock *hctx)
+static void ccid2_hc_tx_dec_pipe(struct sock *sk)
 {
-       hctx->ccid2hctx_pipe--;
+       struct ccid2_hc_tx_sock *hctx = ccid2_hc_tx_sk(sk);
+
+       ccid2_change_pipe(hctx, hctx->ccid2hctx_pipe-1);
        BUG_ON(hctx->ccid2hctx_pipe < 0);
 
        if (hctx->ccid2hctx_pipe == 0)
-               ccid2_hc_tx_kill_rto_timer(hctx);
+               ccid2_hc_tx_kill_rto_timer(sk);
+}
+
+static void ccid2_congestion_event(struct ccid2_hc_tx_sock *hctx,
+                                  struct ccid2_seq *seqp)
+{
+       if (time_before(seqp->ccid2s_sent, hctx->ccid2hctx_last_cong)) {
+               ccid2_pr_debug("Multiple losses in an RTT---treating as one\n");
+               return;
+       }
+
+       hctx->ccid2hctx_last_cong = jiffies;
+
+       ccid2_change_cwnd(hctx, hctx->ccid2hctx_cwnd >> 1);
+       hctx->ccid2hctx_ssthresh = hctx->ccid2hctx_cwnd;
+       if (hctx->ccid2hctx_ssthresh < 2)
+               hctx->ccid2hctx_ssthresh = 2;
 }
 
 static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
@@ -519,7 +565,6 @@ static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
        unsigned char veclen;
        int offset = 0;
        int done = 0;
-       int loss = 0;
        unsigned int maxincr = 0;
 
        ccid2_hc_tx_check_sanity(hctx);
@@ -534,12 +579,10 @@ static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
        if (hctx->ccid2hctx_rpdupack == -1) {
                hctx->ccid2hctx_rpdupack = 0;
                hctx->ccid2hctx_rpseq = seqno;
-       }
-       else {
+       } else {
                /* check if packet is consecutive */
-               if ((hctx->ccid2hctx_rpseq + 1) == seqno) {
+               if ((hctx->ccid2hctx_rpseq + 1) == seqno)
                        hctx->ccid2hctx_rpseq++;
-               }
                /* it's a later packet */
                else if (after48(seqno, hctx->ccid2hctx_rpseq)) {
                        hctx->ccid2hctx_rpdupack++;
@@ -547,7 +590,6 @@ static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
                        /* check if we got enough dupacks */
                        if (hctx->ccid2hctx_rpdupack >=
                            hctx->ccid2hctx_numdupack) {
-
                                hctx->ccid2hctx_rpdupack = -1; /* XXX lame */
                                hctx->ccid2hctx_rpseq = 0;
 
@@ -565,13 +607,22 @@ static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
        case DCCP_PKT_ACK:
        case DCCP_PKT_DATAACK:
                break;
-
        default:
                return;
        }
 
        ackno = DCCP_SKB_CB(skb)->dccpd_ack_seq;
-       seqp = hctx->ccid2hctx_seqh->ccid2s_prev;
+       if (after48(ackno, hctx->ccid2hctx_high_ack))
+               hctx->ccid2hctx_high_ack = ackno;
+
+       seqp = hctx->ccid2hctx_seqt;
+       while (before48(seqp->ccid2s_seq, ackno)) {
+               seqp = seqp->ccid2s_next;
+               if (seqp == hctx->ccid2hctx_seqh) {
+                       seqp = hctx->ccid2hctx_seqh->ccid2s_prev;
+                       break;
+               }
+       }
 
        /* If in slow-start, cwnd can increase at most Ack Ratio / 2 packets for
         * this single ack.  I round up.
@@ -589,8 +640,9 @@ static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
                        u64 ackno_end_rl;
 
                        dccp_set_seqno(&ackno_end_rl, ackno - rl);
-                       ccid2_pr_debug("ackvec start:%llu end:%llu\n", ackno,
-                                      ackno_end_rl);
+                       ccid2_pr_debug("ackvec start:%llu end:%llu\n",
+                                      (unsigned long long)ackno,
+                                      (unsigned long long)ackno_end_rl);
                        /* if the seqno we are analyzing is larger than the
                         * current ackno, then move towards the tail of our
                         * seqnos.
@@ -609,25 +661,24 @@ static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
                         * run length
                         */
                        while (between48(seqp->ccid2s_seq,ackno_end_rl,ackno)) {
-                               const u8 state = (*vector &
-                                                 DCCP_ACKVEC_STATE_MASK) >> 6;
+                               const u8 state = *vector &
+                                                DCCP_ACKVEC_STATE_MASK;
 
                                /* new packet received or marked */
                                if (state != DCCP_ACKVEC_STATE_NOT_RECEIVED &&
                                    !seqp->ccid2s_acked) {
-                                       if (state ==
+                                       if (state ==
                                            DCCP_ACKVEC_STATE_ECN_MARKED) {
-                                               loss = 1;
-                                       }
-                                       else {
+                                               ccid2_congestion_event(hctx,
+                                                                      seqp);
+                                       } else
                                                ccid2_new_ack(sk, seqp,
                                                              &maxincr);
-                                       }
 
                                        seqp->ccid2s_acked = 1;
                                        ccid2_pr_debug("Got ack for %llu\n",
-                                                      seqp->ccid2s_seq);
-                                       ccid2_hc_tx_dec_pipe(hctx);
+                                                      (unsigned long long)seqp->ccid2s_seq);
+                                       ccid2_hc_tx_dec_pipe(sk);
                                }
                                if (seqp == hctx->ccid2hctx_seqt) {
                                        done = 1;
@@ -649,18 +700,23 @@ static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
        /* The state about what is acked should be correct now
         * Check for NUMDUPACK
         */
-       seqp = hctx->ccid2hctx_seqh->ccid2s_prev;
+       seqp = hctx->ccid2hctx_seqt;
+       while (before48(seqp->ccid2s_seq, hctx->ccid2hctx_high_ack)) {
+               seqp = seqp->ccid2s_next;
+               if (seqp == hctx->ccid2hctx_seqh) {
+                       seqp = hctx->ccid2hctx_seqh->ccid2s_prev;
+                       break;
+               }
+       }
        done = 0;
        while (1) {
                if (seqp->ccid2s_acked) {
                        done++;
-                       if (done == hctx->ccid2hctx_numdupack) {
+                       if (done == hctx->ccid2hctx_numdupack)
                                break;
-                       }
                }
-               if (seqp == hctx->ccid2hctx_seqt) {
+               if (seqp == hctx->ccid2hctx_seqt)
                        break;
-               }
                seqp = seqp->ccid2s_prev;
        }
 
@@ -673,8 +729,14 @@ static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
                /* check for lost packets */
                while (1) {
                        if (!seqp->ccid2s_acked) {
-                               loss = 1;
-                               ccid2_hc_tx_dec_pipe(hctx);
+                               ccid2_pr_debug("Packet lost: %llu\n",
+                                              (unsigned long long)seqp->ccid2s_seq);
+                               /* XXX need to traverse from tail -> head in
+                                * order to detect multiple congestion events in
+                                * one ack vector.
+                                */
+                               ccid2_congestion_event(hctx, seqp);
+                               ccid2_hc_tx_dec_pipe(sk);
                        }
                        if (seqp == hctx->ccid2hctx_seqt)
                                break;
@@ -692,62 +754,34 @@ static void ccid2_hc_tx_packet_recv(struct sock *sk, struct sk_buff *skb)
                hctx->ccid2hctx_seqt = hctx->ccid2hctx_seqt->ccid2s_next;
        }
 
-       if (loss) {
-               /* XXX do bit shifts guarantee a 0 as the new bit? */
-               ccid2_change_cwnd(sk, hctx->ccid2hctx_cwnd >> 1);
-               hctx->ccid2hctx_ssthresh = hctx->ccid2hctx_cwnd;
-               if (hctx->ccid2hctx_ssthresh < 2)
-                       hctx->ccid2hctx_ssthresh = 2;
-       }
-
        ccid2_hc_tx_check_sanity(hctx);
 }
 
-static int ccid2_hc_tx_init(struct sock *sk)
+static int ccid2_hc_tx_init(struct ccid *ccid, struct sock *sk)
 {
-       struct dccp_sock *dp = dccp_sk(sk);
-        struct ccid2_hc_tx_sock *hctx;
-       int seqcount = ccid2_seq_len;
-       int i;
-
-        dp->dccps_hc_tx_ccid_private = kzalloc(sizeof(*hctx), gfp_any());
-        if (dp->dccps_hc_tx_ccid_private == NULL)
-                return -ENOMEM;
-
-        hctx = ccid2_hc_tx_sk(sk);
+       struct ccid2_hc_tx_sock *hctx = ccid_priv(ccid);
 
-       /* XXX init variables with proper values */
-       hctx->ccid2hctx_cwnd      = 1;
-       hctx->ccid2hctx_ssthresh  = 10;
+       ccid2_change_cwnd(hctx, 1);
+       /* Initialize ssthresh to infinity.  This means that we will exit the
+        * initial slow-start after the first packet loss.  This is what we
+        * want.
+        */
+       hctx->ccid2hctx_ssthresh  = ~0;
        hctx->ccid2hctx_numdupack = 3;
+       hctx->ccid2hctx_seqbufc   = 0;
 
        /* XXX init ~ to window size... */
-       hctx->ccid2hctx_seqbuf = kmalloc(sizeof(*hctx->ccid2hctx_seqbuf) *
-                                        seqcount, gfp_any());
-       if (hctx->ccid2hctx_seqbuf == NULL) {
-               kfree(dp->dccps_hc_tx_ccid_private);
-               dp->dccps_hc_tx_ccid_private = NULL;
+       if (ccid2_hc_tx_alloc_seq(hctx, CCID2_SEQBUF_LEN, GFP_ATOMIC) != 0)
                return -ENOMEM;
-       }
-       for (i = 0; i < (seqcount - 1); i++) {
-               hctx->ccid2hctx_seqbuf[i].ccid2s_next =
-                                       &hctx->ccid2hctx_seqbuf[i + 1];
-               hctx->ccid2hctx_seqbuf[i + 1].ccid2s_prev =
-                                       &hctx->ccid2hctx_seqbuf[i];
-       }
-       hctx->ccid2hctx_seqbuf[seqcount - 1].ccid2s_next =
-                                       hctx->ccid2hctx_seqbuf;
-       hctx->ccid2hctx_seqbuf->ccid2s_prev =
-                                       &hctx->ccid2hctx_seqbuf[seqcount - 1];
 
-       hctx->ccid2hctx_seqh     = hctx->ccid2hctx_seqbuf;
-       hctx->ccid2hctx_seqt     = hctx->ccid2hctx_seqh;
        hctx->ccid2hctx_sent     = 0;
        hctx->ccid2hctx_rto      = 3 * HZ;
-       hctx->ccid2hctx_srtt     = -1;
+       ccid2_change_srtt(hctx, -1);
        hctx->ccid2hctx_rttvar   = -1;
        hctx->ccid2hctx_lastrtt  = 0;
        hctx->ccid2hctx_rpdupack = -1;
+       hctx->ccid2hctx_last_cong = jiffies;
+       hctx->ccid2hctx_high_ack = 0;
 
        hctx->ccid2hctx_rtotimer.function = &ccid2_hc_tx_rto_expire;
        hctx->ccid2hctx_rtotimer.data     = (unsigned long)sk;
@@ -759,15 +793,14 @@ static int ccid2_hc_tx_init(struct sock *sk)
 
 static void ccid2_hc_tx_exit(struct sock *sk)
 {
-       struct dccp_sock *dp = dccp_sk(sk);
-        struct ccid2_hc_tx_sock *hctx = dp->dccps_hc_tx_ccid_private;
-
-       ccid2_hc_tx_kill_rto_timer(hctx);
+       struct ccid2_hc_tx_sock *hctx = ccid2_hc_tx_sk(sk);
+       int i;
 
-       kfree(hctx->ccid2hctx_seqbuf);
+       ccid2_hc_tx_kill_rto_timer(sk);
 
-       kfree(dp->dccps_hc_tx_ccid_private);
-       dp->dccps_hc_tx_ccid_private = NULL;
+       for (i = 0; i < hctx->ccid2hctx_seqbufc; i++)
+               kfree(hctx->ccid2hctx_seqbuf[i]);
+       hctx->ccid2hctx_seqbufc = 0;
 }
 
 static void ccid2_hc_rx_packet_recv(struct sock *sk, struct sk_buff *skb)
@@ -787,38 +820,24 @@ static void ccid2_hc_rx_packet_recv(struct sock *sk, struct sk_buff *skb)
        }
 }
 
-static int ccid2_hc_rx_init(struct sock *sk)
-{
-       struct dccp_sock *dp = dccp_sk(sk);
-        dp->dccps_hc_rx_ccid_private = kzalloc(sizeof(struct ccid2_hc_rx_sock),
-                                              gfp_any());
-        return dp->dccps_hc_rx_ccid_private == NULL ? -ENOMEM : 0;
-}
-
-static void ccid2_hc_rx_exit(struct sock *sk)
-{
-       struct dccp_sock *dp = dccp_sk(sk);
-
-       kfree(dp->dccps_hc_rx_ccid_private);
-       dp->dccps_hc_rx_ccid_private = NULL;
-}
-
-static struct ccid ccid2 = {
-       .ccid_id                = 2,
+static struct ccid_operations ccid2 = {
+       .ccid_id                = DCCPC_CCID2,
        .ccid_name              = "ccid2",
        .ccid_owner             = THIS_MODULE,
+       .ccid_hc_tx_obj_size    = sizeof(struct ccid2_hc_tx_sock),
        .ccid_hc_tx_init        = ccid2_hc_tx_init,
        .ccid_hc_tx_exit        = ccid2_hc_tx_exit,
        .ccid_hc_tx_send_packet = ccid2_hc_tx_send_packet,
        .ccid_hc_tx_packet_sent = ccid2_hc_tx_packet_sent,
        .ccid_hc_tx_packet_recv = ccid2_hc_tx_packet_recv,
-       .ccid_hc_rx_init        = ccid2_hc_rx_init,
-       .ccid_hc_rx_exit        = ccid2_hc_rx_exit,
+       .ccid_hc_rx_obj_size    = sizeof(struct ccid2_hc_rx_sock),
        .ccid_hc_rx_packet_recv = ccid2_hc_rx_packet_recv,
 };
 
+#ifdef CONFIG_IP_DCCP_CCID2_DEBUG
 module_param(ccid2_debug, int, 0444);
 MODULE_PARM_DESC(ccid2_debug, "Enable debug messages");
+#endif
 
 static __init int ccid2_module_init(void)
 {
@@ -833,6 +852,6 @@ static __exit void ccid2_module_exit(void)
 module_exit(ccid2_module_exit);
 
 MODULE_AUTHOR("Andrea Bittau <a.bittau@cs.ucl.ac.uk>");
-MODULE_DESCRIPTION("DCCP TCP CCID2 CCID");
+MODULE_DESCRIPTION("DCCP TCP-Like (CCID2) CCID");
 MODULE_LICENSE("GPL");
 MODULE_ALIAS("net-dccp-ccid-2");