nghttp2/lib/nghttp2_map.c

190 lines
5.0 KiB
C
Raw Normal View History

/*
2014-03-30 12:09:21 +02:00
* nghttp2 - HTTP/2 C Library
*
* Copyright (c) 2012 Tatsuhiro Tsujikawa
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
2013-07-12 17:19:03 +02:00
#include "nghttp2_map.h"
2013-12-04 16:41:42 +01:00
#include <string.h>
#define INITIAL_TABLE_LENGTH 256
int nghttp2_map_init(nghttp2_map *map, nghttp2_mem *mem) {
map->mem = mem;
2013-12-04 16:41:42 +01:00
map->tablelen = INITIAL_TABLE_LENGTH;
map->table =
nghttp2_mem_calloc(mem, map->tablelen, sizeof(nghttp2_map_entry *));
2014-11-27 15:39:04 +01:00
if (map->table == NULL) {
2013-12-04 16:41:42 +01:00
return NGHTTP2_ERR_NOMEM;
2012-10-05 16:41:49 +02:00
}
2013-12-04 16:41:42 +01:00
map->size = 0;
2013-12-04 16:41:42 +01:00
return 0;
2012-10-05 16:41:49 +02:00
}
void nghttp2_map_free(nghttp2_map *map) {
nghttp2_mem_free(map->mem, map->table);
}
2012-10-05 16:41:49 +02:00
2013-07-12 17:19:03 +02:00
void nghttp2_map_each_free(nghttp2_map *map,
int (*func)(nghttp2_map_entry *entry, void *ptr),
2014-11-27 15:39:04 +01:00
void *ptr) {
2015-09-23 07:41:53 +02:00
uint32_t i;
2014-11-27 15:39:04 +01:00
for (i = 0; i < map->tablelen; ++i) {
2013-12-04 16:41:42 +01:00
nghttp2_map_entry *entry;
2014-11-27 15:39:04 +01:00
for (entry = map->table[i]; entry;) {
2013-12-04 16:41:42 +01:00
nghttp2_map_entry *next = entry->next;
func(entry, ptr);
entry = next;
}
map->table[i] = NULL;
2012-10-05 16:41:49 +02:00
}
}
2013-07-12 17:19:03 +02:00
int nghttp2_map_each(nghttp2_map *map,
int (*func)(nghttp2_map_entry *entry, void *ptr),
2014-11-27 15:39:04 +01:00
void *ptr) {
2013-12-04 16:41:42 +01:00
int rv;
2015-09-23 07:41:53 +02:00
uint32_t i;
2014-11-27 15:39:04 +01:00
for (i = 0; i < map->tablelen; ++i) {
2013-12-04 16:41:42 +01:00
nghttp2_map_entry *entry;
2014-11-27 15:39:04 +01:00
for (entry = map->table[i]; entry; entry = entry->next) {
2013-12-04 16:41:42 +01:00
rv = func(entry, ptr);
2014-11-27 15:39:04 +01:00
if (rv != 0) {
2013-12-04 16:41:42 +01:00
return rv;
}
2012-10-05 16:41:49 +02:00
}
}
return 0;
}
2014-11-27 15:39:04 +01:00
void nghttp2_map_entry_init(nghttp2_map_entry *entry, key_type key) {
entry->key = key;
2013-12-04 16:41:42 +01:00
entry->next = NULL;
}
2014-05-10 11:27:53 +02:00
/* Same hash function in android HashMap source code. */
2013-12-04 16:41:42 +01:00
/* The |mod| must be power of 2 */
2015-09-23 07:41:53 +02:00
static uint32_t hash(int32_t key, uint32_t mod) {
uint32_t h = (uint32_t)key;
2013-12-04 16:41:42 +01:00
h ^= (h >> 20) ^ (h >> 12);
2014-05-10 11:27:53 +02:00
h ^= (h >> 7) ^ (h >> 4);
return h & (mod - 1);
2013-12-04 16:41:42 +01:00
}
2015-09-23 07:41:53 +02:00
static int insert(nghttp2_map_entry **table, uint32_t tablelen,
2014-11-27 15:39:04 +01:00
nghttp2_map_entry *entry) {
2015-09-23 07:41:53 +02:00
uint32_t h = hash(entry->key, tablelen);
2014-11-27 15:39:04 +01:00
if (table[h] == NULL) {
2013-12-04 16:41:42 +01:00
table[h] = entry;
} else {
nghttp2_map_entry *p;
/* We won't allow duplicated key, so check it out. */
2014-11-27 15:39:04 +01:00
for (p = table[h]; p; p = p->next) {
if (p->key == entry->key) {
2013-12-04 16:41:42 +01:00
return NGHTTP2_ERR_INVALID_ARGUMENT;
}
}
2013-12-04 16:41:42 +01:00
entry->next = table[h];
table[h] = entry;
}
2013-12-04 16:41:42 +01:00
return 0;
}
2013-12-04 16:41:42 +01:00
/* new_tablelen must be power of 2 */
2015-09-23 07:41:53 +02:00
static int resize(nghttp2_map *map, uint32_t new_tablelen) {
uint32_t i;
2013-12-04 16:41:42 +01:00
nghttp2_map_entry **new_table;
new_table =
nghttp2_mem_calloc(map->mem, new_tablelen, sizeof(nghttp2_map_entry *));
2014-11-27 15:39:04 +01:00
if (new_table == NULL) {
2013-12-04 16:41:42 +01:00
return NGHTTP2_ERR_NOMEM;
}
2014-11-27 15:39:04 +01:00
for (i = 0; i < map->tablelen; ++i) {
2013-12-04 16:41:42 +01:00
nghttp2_map_entry *entry;
2014-11-27 15:39:04 +01:00
for (entry = map->table[i]; entry;) {
2013-12-04 16:41:42 +01:00
nghttp2_map_entry *next = entry->next;
entry->next = NULL;
/* This function must succeed */
insert(new_table, new_tablelen, entry);
entry = next;
}
}
nghttp2_mem_free(map->mem, map->table);
2013-12-04 16:41:42 +01:00
map->tablelen = new_tablelen;
map->table = new_table;
2013-12-04 16:41:42 +01:00
return 0;
}
2014-11-27 15:39:04 +01:00
int nghttp2_map_insert(nghttp2_map *map, nghttp2_map_entry *new_entry) {
2013-12-04 16:41:42 +01:00
int rv;
2013-12-05 11:26:16 +01:00
/* Load factor is 0.75 */
2014-11-27 15:39:04 +01:00
if ((map->size + 1) * 4 > map->tablelen * 3) {
2013-12-04 16:41:42 +01:00
rv = resize(map, map->tablelen * 2);
2014-11-27 15:39:04 +01:00
if (rv != 0) {
2013-12-04 16:41:42 +01:00
return rv;
}
}
2013-12-04 16:41:42 +01:00
rv = insert(map->table, map->tablelen, new_entry);
2014-11-27 15:39:04 +01:00
if (rv != 0) {
2013-12-04 16:41:42 +01:00
return rv;
}
++map->size;
return 0;
}
2014-11-27 15:39:04 +01:00
nghttp2_map_entry *nghttp2_map_find(nghttp2_map *map, key_type key) {
2015-09-23 07:41:53 +02:00
uint32_t h;
2013-12-04 16:41:42 +01:00
nghttp2_map_entry *entry;
h = hash(key, map->tablelen);
2014-11-27 15:39:04 +01:00
for (entry = map->table[h]; entry; entry = entry->next) {
if (entry->key == key) {
return entry;
}
}
return NULL;
}
2014-11-27 15:39:04 +01:00
int nghttp2_map_remove(nghttp2_map *map, key_type key) {
2015-09-23 07:41:53 +02:00
uint32_t h;
2016-05-14 04:34:51 +02:00
nghttp2_map_entry **dst;
2013-12-04 16:41:42 +01:00
h = hash(key, map->tablelen);
2016-05-14 04:34:51 +02:00
for (dst = &map->table[h]; *dst; dst = &(*dst)->next) {
if ((*dst)->key != key) {
continue;
2012-01-25 18:28:58 +01:00
}
2016-05-14 04:34:51 +02:00
*dst = (*dst)->next;
--map->size;
return 0;
}
2013-12-04 16:41:42 +01:00
return NGHTTP2_ERR_INVALID_ARGUMENT;
}
2014-11-27 15:39:04 +01:00
size_t nghttp2_map_size(nghttp2_map *map) { return map->size; }