simono41/shellinabox
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix some corner cases, where partial matches would not be found.

Browse source 
git-svn-id: https://shellinabox.googlecode.com/svn/trunk@92 0da03de8-d603-11dd-86c2-0f8696b7b6f9
This commit is contained in:
zodiac 2009-03-30 07:23:40 +00:00
parent 3edcc43298
commit c19ed2f3ca
1 changed files with 22 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
libhttp/trie.c
View file

@ -169,7 +169,7 @@ void addToTrie(struct Trie *trie, const char *key, char *value) {
addLeafToTrie(newNode, *key, key + 1, len - 1, value);
} else {
initTrie(newNode, trie->destructor, trie->arg);
trie->value = value;
newNode->value = value;
}
}
}
@ -178,23 +178,28 @@ char *getFromTrie(const struct Trie *trie, const char *key, char **diff) {
if (diff) {
*diff = NULL;
}
struct Trie *partial = NULL;
char *partialKey = NULL;
for (;;) {
if (trie->idx > 0) {
if (memcmp(trie->key, key, trie->idx)) {
if (diff) {
*diff = partialKey;
return partial->value;
}
return NULL;
}
key += trie->idx;
}
int partial = -1;
for (int i = 0; ; i++) {
if (i >= trie->numChildren) {
if (diff && partial >= 0) {
if (diff && partial != NULL) {
// If the caller provided a "diff" pointer, then we allow partial
// matches for the longest possible prefix that is a key in the
// trie. Upon return, the "diff" pointer points to the first
// character in the key does not match.
*diff = (char *)key;
return trie->children[partial].value;
*diff = partialKey;
return partial->value;
}
return NULL;
} else if (*key == trie->children[i].ch) {
@ -204,13 +209,20 @@ char *getFromTrie(const struct Trie *trie, const char *key, char **diff) {
}
return trie->children[i].value;
}
for (int j = i + 1; j < trie->numChildren; j++) {
if (!trie->children[j].ch) {
partial = trie->children + j;
partialKey = (char *)key;
break;
}
}
trie = &trie->children[i];
key++;
break;
} else if (!trie->children[i].ch) {
partial = i;
partial = trie->children + i;
partialKey = (char *)key;
}
}
}
}