SAFE public projects git trees. - safe/jmp/linux-2.6/blob - drivers/staging/batman-adv/bitarray.c

   1 /*
   2  * Copyright (C) 2006-2009 B.A.T.M.A.N. contributors:
   3  *
   4  * Simon Wunderlich, Marek Lindner
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of version 2 of the GNU General Public
   8  * License as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  18  * 02110-1301, USA
  19  *
  20  */
  21
  22 #include "main.h"
  23 #include "bitarray.h"
  24
  25 /* returns true if the corresponding bit in the given seq_bits indicates true
  26  * and curr_seqno is within range of last_seqno */
  27 uint8_t get_bit_status(TYPE_OF_WORD *seq_bits, uint16_t last_seqno,
  28                        uint16_t curr_seqno)
  29 {
  30         int16_t diff, word_offset, word_num;
  31
  32         diff = last_seqno - curr_seqno;
  33         if (diff < 0 || diff >= TQ_LOCAL_WINDOW_SIZE) {
  34                 return 0;
  35         } else {
  36                 /* which word */
  37                 word_num = (last_seqno - curr_seqno) / WORD_BIT_SIZE;
  38                 /* which position in the selected word */
  39                 word_offset = (last_seqno - curr_seqno) % WORD_BIT_SIZE;
  40
  41                 if (seq_bits[word_num] & 1 << word_offset)
  42                         return 1;
  43                 else
  44                         return 0;
  45         }
  46 }
  47
  48 /* turn corresponding bit on, so we can remember that we got the packet */
  49 void bit_mark(TYPE_OF_WORD *seq_bits, int32_t n)
  50 {
  51         int32_t word_offset, word_num;
  52
  53         /* if too old, just drop it */
  54         if (n < 0 || n >= TQ_LOCAL_WINDOW_SIZE)
  55                 return;
  56
  57         /* which word */
  58         word_num = n / WORD_BIT_SIZE;
  59         /* which position in the selected word */
  60         word_offset = n % WORD_BIT_SIZE;
  61
  62         seq_bits[word_num] |= 1 << word_offset; /* turn the position on */
  63 }
  64
  65 /* shift the packet array by n places. */
  66 void bit_shift(TYPE_OF_WORD *seq_bits, int32_t n)
  67 {
  68         int32_t word_offset, word_num;
  69         int32_t i;
  70
  71         if (n <= 0)
  72                 return;
  73
  74         word_offset = n % WORD_BIT_SIZE;/* shift how much inside each word */
  75         word_num = n / WORD_BIT_SIZE;   /* shift over how much (full) words */
  76
  77         for (i = NUM_WORDS - 1; i > word_num; i--) {
  78                 /* going from old to new, so we don't overwrite the data we copy
  79                  * from.
  80                  *
  81                  * left is high, right is low: FEDC BA98 7654 3210
  82                  *                                        ^^ ^^
  83                  *                             vvvv
  84                  * ^^^^ = from, vvvvv =to, we'd have word_num==1 and
  85                  * word_offset==WORD_BIT_SIZE/2 ????? in this example.
  86                  * (=24 bits)
  87                  *
  88                  * our desired output would be: 9876 5432 1000 0000
  89                  * */
  90
  91                 seq_bits[i] =
  92                         (seq_bits[i - word_num] << word_offset) +
  93                         /* take the lower port from the left half, shift it left
  94                          * to its final position */
  95                         (seq_bits[i - word_num - 1] >>
  96                          (WORD_BIT_SIZE-word_offset));
  97                 /* and the upper part of the right half and shift it left to
  98                  * it's position */
  99                 /* for our example that would be: word[0] = 9800 + 0076 =
 100                  * 9876 */
 101         }
 102         /* now for our last word, i==word_num, we only have the it's "left"
 103          * half. that's the 1000 word in our example.*/
 104
 105         seq_bits[i] = (seq_bits[i - word_num] << word_offset);
 106
 107         /* pad the rest with 0, if there is anything */
 108         i--;
 109
 110         for (; i >= 0; i--)
 111                 seq_bits[i] = 0;
 112 }
 113
 114
 115 /* receive and process one packet, returns 1 if received seq_num is considered
 116  * new, 0 if old  */
 117 char bit_get_packet(TYPE_OF_WORD *seq_bits, int16_t seq_num_diff,
 118                     int8_t set_mark)
 119 {
 120         int i;
 121
 122         /* we already got a sequence number higher than this one, so we just
 123          * mark it. this should wrap around the integer just fine */
 124         if ((seq_num_diff < 0) && (seq_num_diff >= -TQ_LOCAL_WINDOW_SIZE)) {
 125                 if (set_mark)
 126                         bit_mark(seq_bits, -seq_num_diff);
 127                 return 0;
 128         }
 129
 130         /* it seems we missed a lot of packets or the other host restarted */
 131         if ((seq_num_diff > TQ_LOCAL_WINDOW_SIZE) ||
 132             (seq_num_diff < -TQ_LOCAL_WINDOW_SIZE)) {
 133
 134                 if (seq_num_diff > TQ_LOCAL_WINDOW_SIZE)
 135                         bat_dbg(DBG_BATMAN,
 136                                 "We missed a lot of packets (%i) !\n",
 137                                 seq_num_diff-1);
 138
 139                 if (-seq_num_diff > TQ_LOCAL_WINDOW_SIZE)
 140                         bat_dbg(DBG_BATMAN,
 141                                 "Other host probably restarted !\n");
 142
 143                 for (i = 0; i < NUM_WORDS; i++)
 144                         seq_bits[i] = 0;
 145
 146                 if (set_mark)
 147                         seq_bits[0] = 1;  /* we only have the latest packet */
 148         } else {
 149                 bit_shift(seq_bits, seq_num_diff);
 150
 151                 if (set_mark)
 152                         bit_mark(seq_bits, 0);
 153         }
 154
 155         return 1;
 156 }
 157
 158 /* count the hamming weight, how many good packets did we receive? just count
 159  * the 1's. The inner loop uses the Kernighan algorithm, see
 160  * http://graphics.stanford.edu/~seander/bithacks.html#CountBitsSetKernighan
 161  */
 162 int bit_packet_count(TYPE_OF_WORD *seq_bits)
 163 {
 164         int i, hamming = 0;
 165         TYPE_OF_WORD word;
 166
 167         for (i = 0; i < NUM_WORDS; i++) {
 168                 word = seq_bits[i];
 169
 170                 while (word) {
 171                         word &= word-1;
 172                         hamming++;
 173                 }
 174         }
 175         return hamming;
 176 }