123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215 |
- /* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- /*
- * Resource allocation code... the code here is responsible for making
- * sure that nothing leaks.
- *
- * rst --- 4/95 --- 6/95
- */
- #include "fspr_private.h"
- #include "fspr_general.h"
- #include "fspr_pools.h"
- #include "fspr_tables.h"
- #include "fspr_strings.h"
- #include "fspr_lib.h"
- #if APR_HAVE_STDLIB_H
- #include <stdlib.h>
- #endif
- #if APR_HAVE_STRING_H
- #include <string.h>
- #endif
- #if APR_HAVE_STRINGS_H
- #include <strings.h>
- #endif
- #if APR_POOL_DEBUG && APR_HAVE_STDIO_H
- #include <stdio.h>
- #endif
- /*****************************************************************
- * This file contains array and fspr_table_t functions only.
- */
- /*****************************************************************
- *
- * The 'array' functions...
- */
- static void make_array_core(fspr_array_header_t *res, fspr_pool_t *p,
- int nelts, int elt_size, int clear)
- {
- /*
- * Assure sanity if someone asks for
- * array of zero elts.
- */
- if (nelts < 1) {
- nelts = 1;
- }
- if (clear) {
- res->elts = fspr_pcalloc(p, nelts * elt_size);
- }
- else {
- res->elts = fspr_palloc(p, nelts * elt_size);
- }
- res->pool = p;
- res->elt_size = elt_size;
- res->nelts = 0; /* No active elements yet... */
- res->nalloc = nelts; /* ...but this many allocated */
- }
- APR_DECLARE(int) fspr_is_empty_array(const fspr_array_header_t *a)
- {
- return ((a == NULL) || (a->nelts == 0));
- }
- APR_DECLARE(fspr_array_header_t *) fspr_array_make(fspr_pool_t *p,
- int nelts, int elt_size)
- {
- fspr_array_header_t *res;
- res = (fspr_array_header_t *) fspr_palloc(p, sizeof(fspr_array_header_t));
- make_array_core(res, p, nelts, elt_size, 1);
- return res;
- }
- APR_DECLARE(void) fspr_array_clear(fspr_array_header_t *arr)
- {
- arr->nelts = 0;
- }
- APR_DECLARE(void *) fspr_array_pop(fspr_array_header_t *arr)
- {
- if (fspr_is_empty_array(arr)) {
- return NULL;
- }
-
- return arr->elts + (arr->elt_size * (--arr->nelts));
- }
- APR_DECLARE(void *) fspr_array_push(fspr_array_header_t *arr)
- {
- if (arr->nelts == arr->nalloc) {
- int new_size = (arr->nalloc <= 0) ? 1 : arr->nalloc * 2;
- char *new_data;
- new_data = fspr_palloc(arr->pool, arr->elt_size * new_size);
- memcpy(new_data, arr->elts, arr->nalloc * arr->elt_size);
- memset(new_data + arr->nalloc * arr->elt_size, 0,
- arr->elt_size * (new_size - arr->nalloc));
- arr->elts = new_data;
- arr->nalloc = new_size;
- }
- ++arr->nelts;
- return arr->elts + (arr->elt_size * (arr->nelts - 1));
- }
- static void *fspr_array_push_noclear(fspr_array_header_t *arr)
- {
- if (arr->nelts == arr->nalloc) {
- int new_size = (arr->nalloc <= 0) ? 1 : arr->nalloc * 2;
- char *new_data;
- new_data = fspr_palloc(arr->pool, arr->elt_size * new_size);
- memcpy(new_data, arr->elts, arr->nalloc * arr->elt_size);
- arr->elts = new_data;
- arr->nalloc = new_size;
- }
- ++arr->nelts;
- return arr->elts + (arr->elt_size * (arr->nelts - 1));
- }
- APR_DECLARE(void) fspr_array_cat(fspr_array_header_t *dst,
- const fspr_array_header_t *src)
- {
- int elt_size = dst->elt_size;
- if (dst->nelts + src->nelts > dst->nalloc) {
- int new_size = (dst->nalloc <= 0) ? 1 : dst->nalloc * 2;
- char *new_data;
- while (dst->nelts + src->nelts > new_size) {
- new_size *= 2;
- }
- new_data = fspr_pcalloc(dst->pool, elt_size * new_size);
- memcpy(new_data, dst->elts, dst->nalloc * elt_size);
- dst->elts = new_data;
- dst->nalloc = new_size;
- }
- memcpy(dst->elts + dst->nelts * elt_size, src->elts,
- elt_size * src->nelts);
- dst->nelts += src->nelts;
- }
- APR_DECLARE(fspr_array_header_t *) fspr_array_copy(fspr_pool_t *p,
- const fspr_array_header_t *arr)
- {
- fspr_array_header_t *res =
- (fspr_array_header_t *) fspr_palloc(p, sizeof(fspr_array_header_t));
- make_array_core(res, p, arr->nalloc, arr->elt_size, 0);
- memcpy(res->elts, arr->elts, arr->elt_size * arr->nelts);
- res->nelts = arr->nelts;
- memset(res->elts + res->elt_size * res->nelts, 0,
- res->elt_size * (res->nalloc - res->nelts));
- return res;
- }
- /* This cute function copies the array header *only*, but arranges
- * for the data section to be copied on the first push or arraycat.
- * It's useful when the elements of the array being copied are
- * read only, but new stuff *might* get added on the end; we have the
- * overhead of the full copy only where it is really needed.
- */
- static APR_INLINE void copy_array_hdr_core(fspr_array_header_t *res,
- const fspr_array_header_t *arr)
- {
- res->elts = arr->elts;
- res->elt_size = arr->elt_size;
- res->nelts = arr->nelts;
- res->nalloc = arr->nelts; /* Force overflow on push */
- }
- APR_DECLARE(fspr_array_header_t *)
- fspr_array_copy_hdr(fspr_pool_t *p,
- const fspr_array_header_t *arr)
- {
- fspr_array_header_t *res;
- res = (fspr_array_header_t *) fspr_palloc(p, sizeof(fspr_array_header_t));
- res->pool = p;
- copy_array_hdr_core(res, arr);
- return res;
- }
- /* The above is used here to avoid consing multiple new array bodies... */
- APR_DECLARE(fspr_array_header_t *)
- fspr_array_append(fspr_pool_t *p,
- const fspr_array_header_t *first,
- const fspr_array_header_t *second)
- {
- fspr_array_header_t *res = fspr_array_copy_hdr(p, first);
- fspr_array_cat(res, second);
- return res;
- }
- /* fspr_array_pstrcat generates a new string from the fspr_pool_t containing
- * the concatenated sequence of substrings referenced as elements within
- * the array. The string will be empty if all substrings are empty or null,
- * or if there are no elements in the array.
- * If sep is non-NUL, it will be inserted between elements as a separator.
- */
- APR_DECLARE(char *) fspr_array_pstrcat(fspr_pool_t *p,
- const fspr_array_header_t *arr,
- const char sep)
- {
- char *cp, *res, **strpp;
- fspr_size_t len;
- int i;
- if (arr->nelts <= 0 || arr->elts == NULL) { /* Empty table? */
- return (char *) fspr_pcalloc(p, 1);
- }
- /* Pass one --- find length of required string */
- len = 0;
- for (i = 0, strpp = (char **) arr->elts; ; ++strpp) {
- if (strpp && *strpp != NULL) {
- len += strlen(*strpp);
- }
- if (++i >= arr->nelts) {
- break;
- }
- if (sep) {
- ++len;
- }
- }
- /* Allocate the required string */
- res = (char *) fspr_palloc(p, len + 1);
- cp = res;
- /* Pass two --- copy the argument strings into the result space */
- for (i = 0, strpp = (char **) arr->elts; ; ++strpp) {
- if (strpp && *strpp != NULL) {
- len = strlen(*strpp);
- memcpy(cp, *strpp, len);
- cp += len;
- }
- if (++i >= arr->nelts) {
- break;
- }
- if (sep) {
- *cp++ = sep;
- }
- }
- *cp = '\0';
- /* Return the result string */
- return res;
- }
- /*****************************************************************
- *
- * The "table" functions.
- */
- #if APR_CHARSET_EBCDIC
- #define CASE_MASK 0xbfbfbfbf
- #else
- #define CASE_MASK 0xdfdfdfdf
- #endif
- #define TABLE_HASH_SIZE 32
- #define TABLE_INDEX_MASK 0x1f
- #define TABLE_HASH(key) (TABLE_INDEX_MASK & *(unsigned char *)(key))
- #define TABLE_INDEX_IS_INITIALIZED(t, i) ((t)->index_initialized & (1 << (i)))
- #define TABLE_SET_INDEX_INITIALIZED(t, i) ((t)->index_initialized |= (1 << (i)))
- /* Compute the "checksum" for a key, consisting of the first
- * 4 bytes, normalized for case-insensitivity and packed into
- * an int...this checksum allows us to do a single integer
- * comparison as a fast check to determine whether we can
- * skip a strcasecmp
- */
- #define COMPUTE_KEY_CHECKSUM(key, checksum) \
- { \
- const char *k = (key); \
- fspr_uint32_t c = (fspr_uint32_t)*k; \
- (checksum) = c; \
- (checksum) <<= 8; \
- if (c) { \
- c = (fspr_uint32_t)*++k; \
- checksum |= c; \
- } \
- (checksum) <<= 8; \
- if (c) { \
- c = (fspr_uint32_t)*++k; \
- checksum |= c; \
- } \
- (checksum) <<= 8; \
- if (c) { \
- c = (fspr_uint32_t)*++k; \
- checksum |= c; \
- } \
- checksum &= CASE_MASK; \
- }
- /** The opaque string-content table type */
- struct fspr_table_t {
- /* This has to be first to promote backwards compatibility with
- * older modules which cast a fspr_table_t * to an fspr_array_header_t *...
- * they should use the fspr_table_elts() function for most of the
- * cases they do this for.
- */
- /** The underlying array for the table */
- fspr_array_header_t a;
- #ifdef MAKE_TABLE_PROFILE
- /** Who created the array. */
- void *creator;
- #endif
- /* An index to speed up table lookups. The way this works is:
- * - Hash the key into the index:
- * - index_first[TABLE_HASH(key)] is the offset within
- * the table of the first entry with that key
- * - index_last[TABLE_HASH(key)] is the offset within
- * the table of the last entry with that key
- * - If (and only if) there is no entry in the table whose
- * key hashes to index element i, then the i'th bit
- * of index_initialized will be zero. (Check this before
- * trying to use index_first[i] or index_last[i]!)
- */
- fspr_uint32_t index_initialized;
- int index_first[TABLE_HASH_SIZE];
- int index_last[TABLE_HASH_SIZE];
- };
- /*
- * NOTICE: if you tweak this you should look at is_empty_table()
- * and table_elts() in alloc.h
- */
- #ifdef MAKE_TABLE_PROFILE
- static fspr_table_entry_t *table_push(fspr_table_t *t)
- {
- if (t->a.nelts == t->a.nalloc) {
- return NULL;
- }
- return (fspr_table_entry_t *) fspr_array_push_noclear(&t->a);
- }
- #else /* MAKE_TABLE_PROFILE */
- #define table_push(t) ((fspr_table_entry_t *) fspr_array_push_noclear(&(t)->a))
- #endif /* MAKE_TABLE_PROFILE */
- APR_DECLARE(const fspr_array_header_t *) fspr_table_elts(const fspr_table_t *t)
- {
- return (const fspr_array_header_t *)t;
- }
- APR_DECLARE(int) fspr_is_empty_table(const fspr_table_t *t)
- {
- return ((t == NULL) || (t->a.nelts == 0));
- }
- APR_DECLARE(fspr_table_t *) fspr_table_make(fspr_pool_t *p, int nelts)
- {
- fspr_table_t *t = fspr_palloc(p, sizeof(fspr_table_t));
- make_array_core(&t->a, p, nelts, sizeof(fspr_table_entry_t), 0);
- #ifdef MAKE_TABLE_PROFILE
- t->creator = __builtin_return_address(0);
- #endif
- t->index_initialized = 0;
- return t;
- }
- APR_DECLARE(fspr_table_t *) fspr_table_copy(fspr_pool_t *p, const fspr_table_t *t)
- {
- fspr_table_t *new = fspr_palloc(p, sizeof(fspr_table_t));
- #if APR_POOL_DEBUG
- /* we don't copy keys and values, so it's necessary that t->a.pool
- * have a life span at least as long as p
- */
- if (!fspr_pool_is_ancestor(t->a.pool, p)) {
- fprintf(stderr, "fspr_table_copy: t's pool is not an ancestor of p\n");
- abort();
- }
- #endif
- make_array_core(&new->a, p, t->a.nalloc, sizeof(fspr_table_entry_t), 0);
- memcpy(new->a.elts, t->a.elts, t->a.nelts * sizeof(fspr_table_entry_t));
- new->a.nelts = t->a.nelts;
- memcpy(new->index_first, t->index_first, sizeof(int) * TABLE_HASH_SIZE);
- memcpy(new->index_last, t->index_last, sizeof(int) * TABLE_HASH_SIZE);
- new->index_initialized = t->index_initialized;
- return new;
- }
- static void table_reindex(fspr_table_t *t)
- {
- int i;
- int hash;
- fspr_table_entry_t *next_elt = (fspr_table_entry_t *) t->a.elts;
- t->index_initialized = 0;
- for (i = 0; i < t->a.nelts; i++, next_elt++) {
- hash = TABLE_HASH(next_elt->key);
- t->index_last[hash] = i;
- if (!TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- t->index_first[hash] = i;
- TABLE_SET_INDEX_INITIALIZED(t, hash);
- }
- }
- }
- APR_DECLARE(void) fspr_table_clear(fspr_table_t *t)
- {
- t->a.nelts = 0;
- t->index_initialized = 0;
- }
- APR_DECLARE(const char *) fspr_table_get(const fspr_table_t *t, const char *key)
- {
- fspr_table_entry_t *next_elt;
- fspr_table_entry_t *end_elt;
- fspr_uint32_t checksum;
- int hash;
- if (key == NULL) {
- return NULL;
- }
- hash = TABLE_HASH(key);
- if (!TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- return NULL;
- }
- COMPUTE_KEY_CHECKSUM(key, checksum);
- next_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_first[hash];;
- end_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_last[hash];
- for (; next_elt <= end_elt; next_elt++) {
- if ((checksum == next_elt->key_checksum) &&
- !strcasecmp(next_elt->key, key)) {
- return next_elt->val;
- }
- }
- return NULL;
- }
- APR_DECLARE(void) fspr_table_set(fspr_table_t *t, const char *key,
- const char *val)
- {
- fspr_table_entry_t *next_elt;
- fspr_table_entry_t *end_elt;
- fspr_table_entry_t *table_end;
- fspr_uint32_t checksum;
- int hash;
- COMPUTE_KEY_CHECKSUM(key, checksum);
- hash = TABLE_HASH(key);
- if (!TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- t->index_first[hash] = t->a.nelts;
- TABLE_SET_INDEX_INITIALIZED(t, hash);
- goto add_new_elt;
- }
- next_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_first[hash];;
- end_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_last[hash];
- table_end =((fspr_table_entry_t *) t->a.elts) + t->a.nelts;
- for (; next_elt <= end_elt; next_elt++) {
- if ((checksum == next_elt->key_checksum) &&
- !strcasecmp(next_elt->key, key)) {
- /* Found an existing entry with the same key, so overwrite it */
- int must_reindex = 0;
- fspr_table_entry_t *dst_elt = NULL;
- next_elt->val = fspr_pstrdup(t->a.pool, val);
- /* Remove any other instances of this key */
- for (next_elt++; next_elt <= end_elt; next_elt++) {
- if ((checksum == next_elt->key_checksum) &&
- !strcasecmp(next_elt->key, key)) {
- t->a.nelts--;
- if (!dst_elt) {
- dst_elt = next_elt;
- }
- }
- else if (dst_elt) {
- *dst_elt++ = *next_elt;
- must_reindex = 1;
- }
- }
- /* If we've removed anything, shift over the remainder
- * of the table (note that the previous loop didn't
- * run to the end of the table, just to the last match
- * for the index)
- */
- if (dst_elt) {
- for (; next_elt < table_end; next_elt++) {
- *dst_elt++ = *next_elt;
- }
- must_reindex = 1;
- }
- if (must_reindex) {
- table_reindex(t);
- }
- return;
- }
- }
- add_new_elt:
- t->index_last[hash] = t->a.nelts;
- next_elt = (fspr_table_entry_t *) table_push(t);
- next_elt->key = fspr_pstrdup(t->a.pool, key);
- next_elt->val = fspr_pstrdup(t->a.pool, val);
- next_elt->key_checksum = checksum;
- }
- APR_DECLARE(void) fspr_table_setn(fspr_table_t *t, const char *key,
- const char *val)
- {
- fspr_table_entry_t *next_elt;
- fspr_table_entry_t *end_elt;
- fspr_table_entry_t *table_end;
- fspr_uint32_t checksum;
- int hash;
- COMPUTE_KEY_CHECKSUM(key, checksum);
- hash = TABLE_HASH(key);
- if (!TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- t->index_first[hash] = t->a.nelts;
- TABLE_SET_INDEX_INITIALIZED(t, hash);
- goto add_new_elt;
- }
- next_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_first[hash];;
- end_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_last[hash];
- table_end =((fspr_table_entry_t *) t->a.elts) + t->a.nelts;
- for (; next_elt <= end_elt; next_elt++) {
- if ((checksum == next_elt->key_checksum) &&
- !strcasecmp(next_elt->key, key)) {
- /* Found an existing entry with the same key, so overwrite it */
- int must_reindex = 0;
- fspr_table_entry_t *dst_elt = NULL;
- next_elt->val = (char *)val;
- /* Remove any other instances of this key */
- for (next_elt++; next_elt <= end_elt; next_elt++) {
- if ((checksum == next_elt->key_checksum) &&
- !strcasecmp(next_elt->key, key)) {
- t->a.nelts--;
- if (!dst_elt) {
- dst_elt = next_elt;
- }
- }
- else if (dst_elt) {
- *dst_elt++ = *next_elt;
- must_reindex = 1;
- }
- }
- /* If we've removed anything, shift over the remainder
- * of the table (note that the previous loop didn't
- * run to the end of the table, just to the last match
- * for the index)
- */
- if (dst_elt) {
- for (; next_elt < table_end; next_elt++) {
- *dst_elt++ = *next_elt;
- }
- must_reindex = 1;
- }
- if (must_reindex) {
- table_reindex(t);
- }
- return;
- }
- }
- add_new_elt:
- t->index_last[hash] = t->a.nelts;
- next_elt = (fspr_table_entry_t *) table_push(t);
- next_elt->key = (char *)key;
- next_elt->val = (char *)val;
- next_elt->key_checksum = checksum;
- }
- APR_DECLARE(void) fspr_table_unset(fspr_table_t *t, const char *key)
- {
- fspr_table_entry_t *next_elt;
- fspr_table_entry_t *end_elt;
- fspr_table_entry_t *dst_elt;
- fspr_uint32_t checksum;
- int hash;
- int must_reindex;
- hash = TABLE_HASH(key);
- if (!TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- return;
- }
- COMPUTE_KEY_CHECKSUM(key, checksum);
- next_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_first[hash];
- end_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_last[hash];
- must_reindex = 0;
- for (; next_elt <= end_elt; next_elt++) {
- if ((checksum == next_elt->key_checksum) &&
- !strcasecmp(next_elt->key, key)) {
- /* Found a match: remove this entry, plus any additional
- * matches for the same key that might follow
- */
- fspr_table_entry_t *table_end = ((fspr_table_entry_t *) t->a.elts) +
- t->a.nelts;
- t->a.nelts--;
- dst_elt = next_elt;
- for (next_elt++; next_elt <= end_elt; next_elt++) {
- if ((checksum == next_elt->key_checksum) &&
- !strcasecmp(next_elt->key, key)) {
- t->a.nelts--;
- }
- else {
- *dst_elt++ = *next_elt;
- }
- }
- /* Shift over the remainder of the table (note that
- * the previous loop didn't run to the end of the table,
- * just to the last match for the index)
- */
- for (; next_elt < table_end; next_elt++) {
- *dst_elt++ = *next_elt;
- }
- must_reindex = 1;
- break;
- }
- }
- if (must_reindex) {
- table_reindex(t);
- }
- }
- APR_DECLARE(void) fspr_table_merge(fspr_table_t *t, const char *key,
- const char *val)
- {
- fspr_table_entry_t *next_elt;
- fspr_table_entry_t *end_elt;
- fspr_uint32_t checksum;
- int hash;
- COMPUTE_KEY_CHECKSUM(key, checksum);
- hash = TABLE_HASH(key);
- if (!TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- t->index_first[hash] = t->a.nelts;
- TABLE_SET_INDEX_INITIALIZED(t, hash);
- goto add_new_elt;
- }
- next_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_first[hash];
- end_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_last[hash];
- for (; next_elt <= end_elt; next_elt++) {
- if ((checksum == next_elt->key_checksum) &&
- !strcasecmp(next_elt->key, key)) {
- /* Found an existing entry with the same key, so merge with it */
- next_elt->val = fspr_pstrcat(t->a.pool, next_elt->val, ", ",
- val, NULL);
- return;
- }
- }
- add_new_elt:
- t->index_last[hash] = t->a.nelts;
- next_elt = (fspr_table_entry_t *) table_push(t);
- next_elt->key = fspr_pstrdup(t->a.pool, key);
- next_elt->val = fspr_pstrdup(t->a.pool, val);
- next_elt->key_checksum = checksum;
- }
- APR_DECLARE(void) fspr_table_mergen(fspr_table_t *t, const char *key,
- const char *val)
- {
- fspr_table_entry_t *next_elt;
- fspr_table_entry_t *end_elt;
- fspr_uint32_t checksum;
- int hash;
- #if APR_POOL_DEBUG
- {
- if (!fspr_pool_is_ancestor(fspr_pool_find(key), t->a.pool)) {
- fprintf(stderr, "fspr_table_mergen: key not in ancestor pool of t\n");
- abort();
- }
- if (!fspr_pool_is_ancestor(fspr_pool_find(val), t->a.pool)) {
- fprintf(stderr, "fspr_table_mergen: key not in ancestor pool of t\n");
- abort();
- }
- }
- #endif
- COMPUTE_KEY_CHECKSUM(key, checksum);
- hash = TABLE_HASH(key);
- if (!TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- t->index_first[hash] = t->a.nelts;
- TABLE_SET_INDEX_INITIALIZED(t, hash);
- goto add_new_elt;
- }
- next_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_first[hash];;
- end_elt = ((fspr_table_entry_t *) t->a.elts) + t->index_last[hash];
- for (; next_elt <= end_elt; next_elt++) {
- if ((checksum == next_elt->key_checksum) &&
- !strcasecmp(next_elt->key, key)) {
- /* Found an existing entry with the same key, so merge with it */
- next_elt->val = fspr_pstrcat(t->a.pool, next_elt->val, ", ",
- val, NULL);
- return;
- }
- }
- add_new_elt:
- t->index_last[hash] = t->a.nelts;
- next_elt = (fspr_table_entry_t *) table_push(t);
- next_elt->key = (char *)key;
- next_elt->val = (char *)val;
- next_elt->key_checksum = checksum;
- }
- APR_DECLARE(void) fspr_table_add(fspr_table_t *t, const char *key,
- const char *val)
- {
- fspr_table_entry_t *elts;
- fspr_uint32_t checksum;
- int hash;
- hash = TABLE_HASH(key);
- t->index_last[hash] = t->a.nelts;
- if (!TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- t->index_first[hash] = t->a.nelts;
- TABLE_SET_INDEX_INITIALIZED(t, hash);
- }
- COMPUTE_KEY_CHECKSUM(key, checksum);
- elts = (fspr_table_entry_t *) table_push(t);
- elts->key = fspr_pstrdup(t->a.pool, key);
- elts->val = fspr_pstrdup(t->a.pool, val);
- elts->key_checksum = checksum;
- }
- APR_DECLARE(void) fspr_table_addn(fspr_table_t *t, const char *key,
- const char *val)
- {
- fspr_table_entry_t *elts;
- fspr_uint32_t checksum;
- int hash;
- #if APR_POOL_DEBUG
- {
- if (!fspr_pool_is_ancestor(fspr_pool_find(key), t->a.pool)) {
- fprintf(stderr, "fspr_table_addn: key not in ancestor pool of t\n");
- abort();
- }
- if (!fspr_pool_is_ancestor(fspr_pool_find(val), t->a.pool)) {
- fprintf(stderr, "fspr_table_addn: key not in ancestor pool of t\n");
- abort();
- }
- }
- #endif
- hash = TABLE_HASH(key);
- t->index_last[hash] = t->a.nelts;
- if (!TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- t->index_first[hash] = t->a.nelts;
- TABLE_SET_INDEX_INITIALIZED(t, hash);
- }
- COMPUTE_KEY_CHECKSUM(key, checksum);
- elts = (fspr_table_entry_t *) table_push(t);
- elts->key = (char *)key;
- elts->val = (char *)val;
- elts->key_checksum = checksum;
- }
- APR_DECLARE(fspr_table_t *) fspr_table_overlay(fspr_pool_t *p,
- const fspr_table_t *overlay,
- const fspr_table_t *base)
- {
- fspr_table_t *res;
- #if APR_POOL_DEBUG
- /* we don't copy keys and values, so it's necessary that
- * overlay->a.pool and base->a.pool have a life span at least
- * as long as p
- */
- if (!fspr_pool_is_ancestor(overlay->a.pool, p)) {
- fprintf(stderr,
- "fspr_table_overlay: overlay's pool is not an ancestor of p\n");
- abort();
- }
- if (!fspr_pool_is_ancestor(base->a.pool, p)) {
- fprintf(stderr,
- "fspr_table_overlay: base's pool is not an ancestor of p\n");
- abort();
- }
- #endif
- res = fspr_palloc(p, sizeof(fspr_table_t));
- /* behave like append_arrays */
- res->a.pool = p;
- copy_array_hdr_core(&res->a, &overlay->a);
- fspr_array_cat(&res->a, &base->a);
- table_reindex(res);
- return res;
- }
- /* And now for something completely abstract ...
- * For each key value given as a vararg:
- * run the function pointed to as
- * int comp(void *r, char *key, char *value);
- * on each valid key-value pair in the fspr_table_t t that matches the vararg key,
- * or once for every valid key-value pair if the vararg list is empty,
- * until the function returns false (0) or we finish the table.
- *
- * Note that we restart the traversal for each vararg, which means that
- * duplicate varargs will result in multiple executions of the function
- * for each matching key. Note also that if the vararg list is empty,
- * only one traversal will be made and will cut short if comp returns 0.
- *
- * Note that the table_get and table_merge functions assume that each key in
- * the fspr_table_t is unique (i.e., no multiple entries with the same key). This
- * function does not make that assumption, since it (unfortunately) isn't
- * true for some of Apache's tables.
- *
- * Note that rec is simply passed-on to the comp function, so that the
- * caller can pass additional info for the task.
- *
- * ADDENDUM for fspr_table_vdo():
- *
- * The caching api will allow a user to walk the header values:
- *
- * fspr_status_t fspr_cache_el_header_walk(fspr_cache_el *el,
- * int (*comp)(void *, const char *, const char *), void *rec, ...);
- *
- * So it can be ..., however from there I use a callback that use a va_list:
- *
- * fspr_status_t (*cache_el_header_walk)(fspr_cache_el *el,
- * int (*comp)(void *, const char *, const char *), void *rec, va_list);
- *
- * To pass those ...'s on down to the actual module that will handle walking
- * their headers, in the file case this is actually just an fspr_table - and
- * rather than reimplementing fspr_table_do (which IMHO would be bad) I just
- * called it with the va_list. For mod_shmem_cache I don't need it since I
- * can't use fspr_table's, but mod_file_cache should (though a good hash would
- * be better, but that's a different issue :).
- *
- * So to make mod_file_cache easier to maintain, it's a good thing
- */
- APR_DECLARE_NONSTD(int) fspr_table_do(fspr_table_do_callback_fn_t *comp,
- void *rec, const fspr_table_t *t, ...)
- {
- int rv;
- va_list vp;
- va_start(vp, t);
- rv = fspr_table_vdo(comp, rec, t, vp);
- va_end(vp);
- return rv;
- }
- /* XXX: do the semantics of this routine make any sense? Right now,
- * if the caller passed in a non-empty va_list of keys to search for,
- * the "early termination" facility only terminates on *that* key; other
- * keys will continue to process. Note that this only has any effect
- * at all if there are multiple entries in the table with the same key,
- * otherwise the called function can never effectively early-terminate
- * this function, as the zero return value is effectively ignored.
- *
- * Note also that this behavior is at odds with the behavior seen if an
- * empty va_list is passed in -- in that case, a zero return value terminates
- * the entire fspr_table_vdo (which is what I think should happen in
- * both cases).
- *
- * If nobody objects soon, I'm going to change the order of the nested
- * loops in this function so that any zero return value from the (*comp)
- * function will cause a full termination of fspr_table_vdo. I'm hesitant
- * at the moment because these (funky) semantics have been around for a
- * very long time, and although Apache doesn't seem to use them at all,
- * some third-party vendor might. I can only think of one possible reason
- * the existing semantics would make any sense, and it's very Apache-centric,
- * which is this: if (*comp) is looking for matches of a particular
- * substring in request headers (let's say it's looking for a particular
- * cookie name in the Set-Cookie headers), then maybe it wants to be
- * able to stop searching early as soon as it finds that one and move
- * on to the next key. That's only an optimization of course, but changing
- * the behavior of this function would mean that any code that tried
- * to do that would stop working right.
- *
- * Sigh. --JCW, 06/28/02
- */
- APR_DECLARE(int) fspr_table_vdo(fspr_table_do_callback_fn_t *comp,
- void *rec, const fspr_table_t *t, va_list vp)
- {
- char *argp;
- fspr_table_entry_t *elts = (fspr_table_entry_t *) t->a.elts;
- int vdorv = 1;
- argp = va_arg(vp, char *);
- do {
- int rv = 1, i;
- if (argp) {
- /* Scan for entries that match the next key */
- int hash = TABLE_HASH(argp);
- if (TABLE_INDEX_IS_INITIALIZED(t, hash)) {
- fspr_uint32_t checksum;
- COMPUTE_KEY_CHECKSUM(argp, checksum);
- for (i = t->index_first[hash];
- rv && (i <= t->index_last[hash]); ++i) {
- if (elts[i].key && (checksum == elts[i].key_checksum) &&
- !strcasecmp(elts[i].key, argp)) {
- rv = (*comp) (rec, elts[i].key, elts[i].val);
- }
- }
- }
- }
- else {
- /* Scan the entire table */
- for (i = 0; rv && (i < t->a.nelts); ++i) {
- if (elts[i].key) {
- rv = (*comp) (rec, elts[i].key, elts[i].val);
- }
- }
- }
- if (rv == 0) {
- vdorv = 0;
- }
- } while (argp && ((argp = va_arg(vp, char *)) != NULL));
- return vdorv;
- }
- static fspr_table_entry_t **table_mergesort(fspr_pool_t *pool,
- fspr_table_entry_t **values,
- fspr_size_t n)
- {
- /* Bottom-up mergesort, based on design in Sedgewick's "Algorithms
- * in C," chapter 8
- */
- fspr_table_entry_t **values_tmp =
- (fspr_table_entry_t **)fspr_palloc(pool, n * sizeof(fspr_table_entry_t*));
- fspr_size_t i;
- fspr_size_t blocksize;
- /* First pass: sort pairs of elements (blocksize=1) */
- for (i = 0; i + 1 < n; i += 2) {
- if (strcasecmp(values[i]->key, values[i + 1]->key) > 0) {
- fspr_table_entry_t *swap = values[i];
- values[i] = values[i + 1];
- values[i + 1] = swap;
- }
- }
- /* Merge successively larger blocks */
- blocksize = 2;
- while (blocksize < n) {
- fspr_table_entry_t **dst = values_tmp;
- fspr_size_t next_start;
- fspr_table_entry_t **swap;
- /* Merge consecutive pairs blocks of the next blocksize.
- * Within a block, elements are in sorted order due to
- * the previous iteration.
- */
- for (next_start = 0; next_start + blocksize < n;
- next_start += (blocksize + blocksize)) {
- fspr_size_t block1_start = next_start;
- fspr_size_t block2_start = block1_start + blocksize;
- fspr_size_t block1_end = block2_start;
- fspr_size_t block2_end = block2_start + blocksize;
- if (block2_end > n) {
- /* The last block may be smaller than blocksize */
- block2_end = n;
- }
- for (;;) {
- /* Merge the next two blocks:
- * Pick the smaller of the next element from
- * block 1 and the next element from block 2.
- * Once either of the blocks is emptied, copy
- * over all the remaining elements from the
- * other block
- */
- if (block1_start == block1_end) {
- for (; block2_start < block2_end; block2_start++) {
- *dst++ = values[block2_start];
- }
- break;
- }
- else if (block2_start == block2_end) {
- for (; block1_start < block1_end; block1_start++) {
- *dst++ = values[block1_start];
- }
- break;
- }
- if (strcasecmp(values[block1_start]->key,
- values[block2_start]->key) > 0) {
- *dst++ = values[block2_start++];
- }
- else {
- *dst++ = values[block1_start++];
- }
- }
- }
- /* If n is not a multiple of 2*blocksize, some elements
- * will be left over at the end of the array.
- */
- for (i = dst - values_tmp; i < n; i++) {
- values_tmp[i] = values[i];
- }
- /* The output array of this pass becomes the input
- * array of the next pass, and vice versa
- */
- swap = values_tmp;
- values_tmp = values;
- values = swap;
- blocksize += blocksize;
- }
- return values;
- }
- APR_DECLARE(void) fspr_table_compress(fspr_table_t *t, unsigned flags)
- {
- fspr_table_entry_t **sort_array;
- fspr_table_entry_t **sort_next;
- fspr_table_entry_t **sort_end;
- fspr_table_entry_t *table_next;
- fspr_table_entry_t **last;
- int i;
- int dups_found;
- if (t->a.nelts <= 1) {
- return;
- }
- /* Copy pointers to all the table elements into an
- * array and sort to allow for easy detection of
- * duplicate keys
- */
- sort_array = (fspr_table_entry_t **)
- fspr_palloc(t->a.pool, t->a.nelts * sizeof(fspr_table_entry_t*));
- sort_next = sort_array;
- table_next = (fspr_table_entry_t *)t->a.elts;
- i = t->a.nelts;
- do {
- *sort_next++ = table_next++;
- } while (--i);
- /* Note: the merge is done with mergesort instead of quicksort
- * because mergesort is a stable sort and runs in n*log(n)
- * time regardless of its inputs (quicksort is quadratic in
- * the worst case)
- */
- sort_array = table_mergesort(t->a.pool, sort_array, t->a.nelts);
- /* Process any duplicate keys */
- dups_found = 0;
- sort_next = sort_array;
- sort_end = sort_array + t->a.nelts;
- last = sort_next++;
- while (sort_next < sort_end) {
- if (((*sort_next)->key_checksum == (*last)->key_checksum) &&
- !strcasecmp((*sort_next)->key, (*last)->key)) {
- fspr_table_entry_t **dup_last = sort_next + 1;
- dups_found = 1;
- while ((dup_last < sort_end) &&
- ((*dup_last)->key_checksum == (*last)->key_checksum) &&
- !strcasecmp((*dup_last)->key, (*last)->key)) {
- dup_last++;
- }
- dup_last--; /* Elements from last through dup_last, inclusive,
- * all have the same key
- */
- if (flags == APR_OVERLAP_TABLES_MERGE) {
- fspr_size_t len = 0;
- fspr_table_entry_t **next = last;
- char *new_val;
- char *val_dst;
- do {
- len += strlen((*next)->val);
- len += 2; /* for ", " or trailing null */
- } while (++next <= dup_last);
- new_val = (char *)fspr_palloc(t->a.pool, len);
- val_dst = new_val;
- next = last;
- for (;;) {
- strcpy(val_dst, (*next)->val);
- val_dst += strlen((*next)->val);
- next++;
- if (next > dup_last) {
- *val_dst = 0;
- break;
- }
- else {
- *val_dst++ = ',';
- *val_dst++ = ' ';
- }
- }
- (*last)->val = new_val;
- }
- else { /* overwrite */
- (*last)->val = (*dup_last)->val;
- }
- do {
- (*sort_next)->key = NULL;
- } while (++sort_next <= dup_last);
- }
- else {
- last = sort_next++;
- }
- }
- /* Shift elements to the left to fill holes left by removing duplicates */
- if (dups_found) {
- fspr_table_entry_t *src = (fspr_table_entry_t *)t->a.elts;
- fspr_table_entry_t *dst = (fspr_table_entry_t *)t->a.elts;
- fspr_table_entry_t *last_elt = src + t->a.nelts;
- do {
- if (src->key) {
- *dst++ = *src;
- }
- } while (++src < last_elt);
- t->a.nelts -= (int)(last_elt - dst);
- }
- table_reindex(t);
- }
- static void fspr_table_cat(fspr_table_t *t, const fspr_table_t *s)
- {
- const int n = t->a.nelts;
- register int idx;
- fspr_array_cat(&t->a,&s->a);
- if (n == 0) {
- memcpy(t->index_first,s->index_first,sizeof(int) * TABLE_HASH_SIZE);
- memcpy(t->index_last, s->index_last, sizeof(int) * TABLE_HASH_SIZE);
- t->index_initialized = s->index_initialized;
- return;
- }
- for (idx = 0; idx < TABLE_HASH_SIZE; ++idx) {
- if (TABLE_INDEX_IS_INITIALIZED(s, idx)) {
- t->index_last[idx] = s->index_last[idx] + n;
- if (!TABLE_INDEX_IS_INITIALIZED(t, idx)) {
- t->index_first[idx] = s->index_first[idx] + n;
- }
- }
- }
- t->index_initialized |= s->index_initialized;
- }
- APR_DECLARE(void) fspr_table_overlap(fspr_table_t *a, const fspr_table_t *b,
- unsigned flags)
- {
- if (a->a.nelts + b->a.nelts == 0) {
- return;
- }
- #if APR_POOL_DEBUG
- /* Since the keys and values are not copied, it's required that
- * b->a.pool has a lifetime at least as long as a->a.pool. */
- if (!fspr_pool_is_ancestor(b->a.pool, a->a.pool)) {
- fprintf(stderr, "fspr_table_overlap: b's pool is not an ancestor of a's\n");
- abort();
- }
- #endif
- fspr_table_cat(a, b);
- fspr_table_compress(a, flags);
- }
|