mirror of https://gitee.com/openkylin/linux.git
fib_trie: Update meaning of pos to represent unchecked bits
This change moves the pos value to the other side of the "bits" field. By doing this it actually simplifies a significant amount of code in the trie. For example when halving a tree we know that the bit lost exists at oldnode->pos, and if we inflate the tree the new bit being add is at tn->pos. Previously to find those bits you would have to subtract pos and bits from the keylength or start with a value of (1 << 31) and then shift that. There are a number of spots throughout the code that benefit from this. In the case of the hot-path searches the main advantage is that we can drop 2 or more operations from the search path as we no longer need to compute the value for the index to be shifted by and can instead just use the raw pos value. In addition the tkey_extract_bits is now defunct and can be replaced by get_index since the two operations were doing the same thing, but now get_index does it much more quickly as it is only an xor and shift versus a pair of shifts and a subtraction. Signed-off-by: Alexander Duyck <alexander.h.duyck@redhat.com> Signed-off-by: David S. Miller <davem@davemloft.net>
This commit is contained in:
parent
836a0123c9
commit
e9b44019d4
|
@ -90,8 +90,7 @@ typedef unsigned int t_key;
|
||||||
#define IS_TNODE(n) ((n)->bits)
|
#define IS_TNODE(n) ((n)->bits)
|
||||||
#define IS_LEAF(n) (!(n)->bits)
|
#define IS_LEAF(n) (!(n)->bits)
|
||||||
|
|
||||||
#define get_shift(_kv) (KEYLENGTH - (_kv)->pos - (_kv)->bits)
|
#define get_index(_key, _kv) (((_key) ^ (_kv)->key) >> (_kv)->pos)
|
||||||
#define get_index(_key, _kv) (((_key) ^ (_kv)->key) >> get_shift(_kv))
|
|
||||||
|
|
||||||
struct tnode {
|
struct tnode {
|
||||||
t_key key;
|
t_key key;
|
||||||
|
@ -209,80 +208,63 @@ static inline struct tnode *tnode_get_child_rcu(const struct tnode *tn, unsigned
|
||||||
return rcu_dereference_rtnl(tn->child[i]);
|
return rcu_dereference_rtnl(tn->child[i]);
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline t_key mask_pfx(t_key k, unsigned int l)
|
/* To understand this stuff, an understanding of keys and all their bits is
|
||||||
{
|
* necessary. Every node in the trie has a key associated with it, but not
|
||||||
return (l == 0) ? 0 : k >> (KEYLENGTH-l) << (KEYLENGTH-l);
|
* all of the bits in that key are significant.
|
||||||
}
|
*
|
||||||
|
* Consider a node 'n' and its parent 'tp'.
|
||||||
static inline t_key tkey_extract_bits(t_key a, unsigned int offset, unsigned int bits)
|
*
|
||||||
{
|
* If n is a leaf, every bit in its key is significant. Its presence is
|
||||||
if (offset < KEYLENGTH)
|
* necessitated by path compression, since during a tree traversal (when
|
||||||
return ((t_key)(a << offset)) >> (KEYLENGTH - bits);
|
* searching for a leaf - unless we are doing an insertion) we will completely
|
||||||
else
|
* ignore all skipped bits we encounter. Thus we need to verify, at the end of
|
||||||
return 0;
|
* a potentially successful search, that we have indeed been walking the
|
||||||
}
|
* correct key path.
|
||||||
|
*
|
||||||
/*
|
* Note that we can never "miss" the correct key in the tree if present by
|
||||||
To understand this stuff, an understanding of keys and all their bits is
|
* following the wrong path. Path compression ensures that segments of the key
|
||||||
necessary. Every node in the trie has a key associated with it, but not
|
* that are the same for all keys with a given prefix are skipped, but the
|
||||||
all of the bits in that key are significant.
|
* skipped part *is* identical for each node in the subtrie below the skipped
|
||||||
|
* bit! trie_insert() in this implementation takes care of that.
|
||||||
Consider a node 'n' and its parent 'tp'.
|
*
|
||||||
|
* if n is an internal node - a 'tnode' here, the various parts of its key
|
||||||
If n is a leaf, every bit in its key is significant. Its presence is
|
* have many different meanings.
|
||||||
necessitated by path compression, since during a tree traversal (when
|
*
|
||||||
searching for a leaf - unless we are doing an insertion) we will completely
|
* Example:
|
||||||
ignore all skipped bits we encounter. Thus we need to verify, at the end of
|
* _________________________________________________________________
|
||||||
a potentially successful search, that we have indeed been walking the
|
* | i | i | i | i | i | i | i | N | N | N | S | S | S | S | S | C |
|
||||||
correct key path.
|
* -----------------------------------------------------------------
|
||||||
|
* 31 30 29 28 27 26 25 24 23 22 21 20 19 18 17 16
|
||||||
Note that we can never "miss" the correct key in the tree if present by
|
*
|
||||||
following the wrong path. Path compression ensures that segments of the key
|
* _________________________________________________________________
|
||||||
that are the same for all keys with a given prefix are skipped, but the
|
* | C | C | C | u | u | u | u | u | u | u | u | u | u | u | u | u |
|
||||||
skipped part *is* identical for each node in the subtrie below the skipped
|
* -----------------------------------------------------------------
|
||||||
bit! trie_insert() in this implementation takes care of that - note the
|
* 15 14 13 12 11 10 9 8 7 6 5 4 3 2 1 0
|
||||||
call to tkey_sub_equals() in trie_insert().
|
*
|
||||||
|
* tp->pos = 22
|
||||||
if n is an internal node - a 'tnode' here, the various parts of its key
|
* tp->bits = 3
|
||||||
have many different meanings.
|
* n->pos = 13
|
||||||
|
* n->bits = 4
|
||||||
Example:
|
*
|
||||||
_________________________________________________________________
|
* First, let's just ignore the bits that come before the parent tp, that is
|
||||||
| i | i | i | i | i | i | i | N | N | N | S | S | S | S | S | C |
|
* the bits from (tp->pos + tp->bits) to 31. They are *known* but at this
|
||||||
-----------------------------------------------------------------
|
* point we do not use them for anything.
|
||||||
0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
|
*
|
||||||
|
* The bits from (tp->pos) to (tp->pos + tp->bits - 1) - "N", above - are the
|
||||||
_________________________________________________________________
|
* index into the parent's child array. That is, they will be used to find
|
||||||
| C | C | C | u | u | u | u | u | u | u | u | u | u | u | u | u |
|
* 'n' among tp's children.
|
||||||
-----------------------------------------------------------------
|
*
|
||||||
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
|
* The bits from (n->pos + n->bits) to (tn->pos - 1) - "S" - are skipped bits
|
||||||
|
* for the node n.
|
||||||
tp->pos = 7
|
*
|
||||||
tp->bits = 3
|
* All the bits we have seen so far are significant to the node n. The rest
|
||||||
n->pos = 15
|
* of the bits are really not needed or indeed known in n->key.
|
||||||
n->bits = 4
|
*
|
||||||
|
* The bits from (n->pos) to (n->pos + n->bits - 1) - "C" - are the index into
|
||||||
First, let's just ignore the bits that come before the parent tp, that is
|
* n's child array, and will of course be different for each child.
|
||||||
the bits from 0 to (tp->pos-1). They are *known* but at this point we do
|
*
|
||||||
not use them for anything.
|
* The rest of the bits, from 0 to (n->pos + n->bits), are completely unknown
|
||||||
|
* at this point.
|
||||||
The bits from (tp->pos) to (tp->pos + tp->bits - 1) - "N", above - are the
|
|
||||||
index into the parent's child array. That is, they will be used to find
|
|
||||||
'n' among tp's children.
|
|
||||||
|
|
||||||
The bits from (tp->pos + tp->bits) to (n->pos - 1) - "S" - are skipped bits
|
|
||||||
for the node n.
|
|
||||||
|
|
||||||
All the bits we have seen so far are significant to the node n. The rest
|
|
||||||
of the bits are really not needed or indeed known in n->key.
|
|
||||||
|
|
||||||
The bits from (n->pos) to (n->pos + n->bits - 1) - "C" - are the index into
|
|
||||||
n's child array, and will of course be different for each child.
|
|
||||||
|
|
||||||
|
|
||||||
The rest of the bits, from (n->pos + n->bits) onward, are completely unknown
|
|
||||||
at this point.
|
|
||||||
|
|
||||||
*/
|
*/
|
||||||
|
|
||||||
static const int halve_threshold = 25;
|
static const int halve_threshold = 25;
|
||||||
|
@ -367,7 +349,7 @@ static struct tnode *leaf_new(t_key key)
|
||||||
* as the nodes are searched
|
* as the nodes are searched
|
||||||
*/
|
*/
|
||||||
l->key = key;
|
l->key = key;
|
||||||
l->pos = KEYLENGTH;
|
l->pos = 0;
|
||||||
/* set bits to 0 indicating we are not a tnode */
|
/* set bits to 0 indicating we are not a tnode */
|
||||||
l->bits = 0;
|
l->bits = 0;
|
||||||
|
|
||||||
|
@ -400,7 +382,7 @@ static struct tnode *tnode_new(t_key key, int pos, int bits)
|
||||||
tn->parent = NULL;
|
tn->parent = NULL;
|
||||||
tn->pos = pos;
|
tn->pos = pos;
|
||||||
tn->bits = bits;
|
tn->bits = bits;
|
||||||
tn->key = mask_pfx(key, pos);
|
tn->key = (shift < KEYLENGTH) ? (key >> shift) << shift : 0;
|
||||||
tn->full_children = 0;
|
tn->full_children = 0;
|
||||||
tn->empty_children = 1<<bits;
|
tn->empty_children = 1<<bits;
|
||||||
}
|
}
|
||||||
|
@ -410,14 +392,12 @@ static struct tnode *tnode_new(t_key key, int pos, int bits)
|
||||||
return tn;
|
return tn;
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/* Check whether a tnode 'n' is "full", i.e. it is an internal node
|
||||||
* Check whether a tnode 'n' is "full", i.e. it is an internal node
|
|
||||||
* and no bits are skipped. See discussion in dyntree paper p. 6
|
* and no bits are skipped. See discussion in dyntree paper p. 6
|
||||||
*/
|
*/
|
||||||
|
|
||||||
static inline int tnode_full(const struct tnode *tn, const struct tnode *n)
|
static inline int tnode_full(const struct tnode *tn, const struct tnode *n)
|
||||||
{
|
{
|
||||||
return n && IS_TNODE(n) && (n->pos == (tn->pos + tn->bits));
|
return n && ((n->pos + n->bits) == tn->pos) && IS_TNODE(n);
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline void put_child(struct tnode *tn, int i,
|
static inline void put_child(struct tnode *tn, int i,
|
||||||
|
@ -641,11 +621,12 @@ static struct tnode *inflate(struct trie *t, struct tnode *oldtnode)
|
||||||
{
|
{
|
||||||
int olen = tnode_child_length(oldtnode);
|
int olen = tnode_child_length(oldtnode);
|
||||||
struct tnode *tn;
|
struct tnode *tn;
|
||||||
|
t_key m;
|
||||||
int i;
|
int i;
|
||||||
|
|
||||||
pr_debug("In inflate\n");
|
pr_debug("In inflate\n");
|
||||||
|
|
||||||
tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits + 1);
|
tn = tnode_new(oldtnode->key, oldtnode->pos - 1, oldtnode->bits + 1);
|
||||||
|
|
||||||
if (!tn)
|
if (!tn)
|
||||||
return ERR_PTR(-ENOMEM);
|
return ERR_PTR(-ENOMEM);
|
||||||
|
@ -656,21 +637,18 @@ static struct tnode *inflate(struct trie *t, struct tnode *oldtnode)
|
||||||
* fails. In case of failure we return the oldnode and inflate
|
* fails. In case of failure we return the oldnode and inflate
|
||||||
* of tnode is ignored.
|
* of tnode is ignored.
|
||||||
*/
|
*/
|
||||||
|
for (i = 0, m = 1u << tn->pos; i < olen; i++) {
|
||||||
|
struct tnode *inode = tnode_get_child(oldtnode, i);
|
||||||
|
|
||||||
for (i = 0; i < olen; i++) {
|
if (tnode_full(oldtnode, inode) && (inode->bits > 1)) {
|
||||||
struct tnode *inode;
|
|
||||||
|
|
||||||
inode = tnode_get_child(oldtnode, i);
|
|
||||||
if (tnode_full(oldtnode, inode) && inode->bits > 1) {
|
|
||||||
struct tnode *left, *right;
|
struct tnode *left, *right;
|
||||||
t_key m = ~0U << (KEYLENGTH - 1) >> inode->pos;
|
|
||||||
|
|
||||||
left = tnode_new(inode->key&(~m), inode->pos + 1,
|
left = tnode_new(inode->key & ~m, inode->pos,
|
||||||
inode->bits - 1);
|
inode->bits - 1);
|
||||||
if (!left)
|
if (!left)
|
||||||
goto nomem;
|
goto nomem;
|
||||||
|
|
||||||
right = tnode_new(inode->key|m, inode->pos + 1,
|
right = tnode_new(inode->key | m, inode->pos,
|
||||||
inode->bits - 1);
|
inode->bits - 1);
|
||||||
|
|
||||||
if (!right) {
|
if (!right) {
|
||||||
|
@ -694,9 +672,7 @@ static struct tnode *inflate(struct trie *t, struct tnode *oldtnode)
|
||||||
|
|
||||||
/* A leaf or an internal node with skipped bits */
|
/* A leaf or an internal node with skipped bits */
|
||||||
if (!tnode_full(oldtnode, inode)) {
|
if (!tnode_full(oldtnode, inode)) {
|
||||||
put_child(tn,
|
put_child(tn, get_index(inode->key, tn), inode);
|
||||||
tkey_extract_bits(inode->key, tn->pos, tn->bits),
|
|
||||||
inode);
|
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -767,7 +743,7 @@ static struct tnode *halve(struct trie *t, struct tnode *oldtnode)
|
||||||
|
|
||||||
pr_debug("In halve\n");
|
pr_debug("In halve\n");
|
||||||
|
|
||||||
tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits - 1);
|
tn = tnode_new(oldtnode->key, oldtnode->pos + 1, oldtnode->bits - 1);
|
||||||
|
|
||||||
if (!tn)
|
if (!tn)
|
||||||
return ERR_PTR(-ENOMEM);
|
return ERR_PTR(-ENOMEM);
|
||||||
|
@ -787,7 +763,7 @@ static struct tnode *halve(struct trie *t, struct tnode *oldtnode)
|
||||||
if (left && right) {
|
if (left && right) {
|
||||||
struct tnode *newn;
|
struct tnode *newn;
|
||||||
|
|
||||||
newn = tnode_new(left->key, tn->pos + tn->bits, 1);
|
newn = tnode_new(left->key, oldtnode->pos, 1);
|
||||||
|
|
||||||
if (!newn)
|
if (!newn)
|
||||||
goto nomem;
|
goto nomem;
|
||||||
|
@ -915,7 +891,7 @@ static void trie_rebalance(struct trie *t, struct tnode *tn)
|
||||||
key = tn->key;
|
key = tn->key;
|
||||||
|
|
||||||
while (tn != NULL && (tp = node_parent(tn)) != NULL) {
|
while (tn != NULL && (tp = node_parent(tn)) != NULL) {
|
||||||
cindex = tkey_extract_bits(key, tp->pos, tp->bits);
|
cindex = get_index(key, tp);
|
||||||
wasfull = tnode_full(tp, tnode_get_child(tp, cindex));
|
wasfull = tnode_full(tp, tnode_get_child(tp, cindex));
|
||||||
tn = resize(t, tn);
|
tn = resize(t, tn);
|
||||||
|
|
||||||
|
@ -1005,11 +981,8 @@ static struct list_head *fib_insert_node(struct trie *t, u32 key, int plen)
|
||||||
*/
|
*/
|
||||||
if (n) {
|
if (n) {
|
||||||
struct tnode *tn;
|
struct tnode *tn;
|
||||||
int newpos;
|
|
||||||
|
|
||||||
newpos = KEYLENGTH - __fls(n->key ^ key) - 1;
|
tn = tnode_new(key, __fls(key ^ n->key), 1);
|
||||||
|
|
||||||
tn = tnode_new(key, newpos, 1);
|
|
||||||
if (!tn) {
|
if (!tn) {
|
||||||
free_leaf_info(li);
|
free_leaf_info(li);
|
||||||
node_free(l);
|
node_free(l);
|
||||||
|
@ -1559,12 +1532,7 @@ static int trie_flush_leaf(struct tnode *l)
|
||||||
static struct tnode *leaf_walk_rcu(struct tnode *p, struct tnode *c)
|
static struct tnode *leaf_walk_rcu(struct tnode *p, struct tnode *c)
|
||||||
{
|
{
|
||||||
do {
|
do {
|
||||||
t_key idx;
|
t_key idx = c ? idx = get_index(c->key, p) + 1 : 0;
|
||||||
|
|
||||||
if (c)
|
|
||||||
idx = tkey_extract_bits(c->key, p->pos, p->bits) + 1;
|
|
||||||
else
|
|
||||||
idx = 0;
|
|
||||||
|
|
||||||
while (idx < 1u << p->bits) {
|
while (idx < 1u << p->bits) {
|
||||||
c = tnode_get_child_rcu(p, idx++);
|
c = tnode_get_child_rcu(p, idx++);
|
||||||
|
@ -1851,7 +1819,7 @@ static struct tnode *fib_trie_get_next(struct fib_trie_iter *iter)
|
||||||
/* Current node exhausted, pop back up */
|
/* Current node exhausted, pop back up */
|
||||||
p = node_parent_rcu(tn);
|
p = node_parent_rcu(tn);
|
||||||
if (p) {
|
if (p) {
|
||||||
cindex = tkey_extract_bits(tn->key, p->pos, p->bits)+1;
|
cindex = get_index(tn->key, p) + 1;
|
||||||
tn = p;
|
tn = p;
|
||||||
--iter->depth;
|
--iter->depth;
|
||||||
goto rescan;
|
goto rescan;
|
||||||
|
@ -2187,9 +2155,9 @@ static int fib_trie_seq_show(struct seq_file *seq, void *v)
|
||||||
__be32 prf = htonl(n->key);
|
__be32 prf = htonl(n->key);
|
||||||
|
|
||||||
seq_indent(seq, iter->depth-1);
|
seq_indent(seq, iter->depth-1);
|
||||||
seq_printf(seq, " +-- %pI4/%d %d %d %d\n",
|
seq_printf(seq, " +-- %pI4/%zu %u %u %u\n",
|
||||||
&prf, n->pos, n->bits, n->full_children,
|
&prf, KEYLENGTH - n->pos - n->bits, n->bits,
|
||||||
n->empty_children);
|
n->full_children, n->empty_children);
|
||||||
} else {
|
} else {
|
||||||
struct leaf_info *li;
|
struct leaf_info *li;
|
||||||
__be32 val = htonl(n->key);
|
__be32 val = htonl(n->key);
|
||||||
|
|
Loading…
Reference in New Issue