olaf@20: /* olaf@20: * olaf@20: */ olaf@2: olaf@20: #include olaf@20: #include olaf@20: olaf@20: #include "map.h" olaf@20: olaf@20: UcxMap *ucx_map_new(size_t size) { olaf@20: UcxMap *map = (UcxMap*)malloc(sizeof(UcxMap)); olaf@20: if(map == NULL) { olaf@20: return NULL; olaf@20: } olaf@20: olaf@20: map->map = (UcxMapElement*)calloc(size, sizeof(UcxMapElement)); olaf@20: if(map->map == NULL) { olaf@20: free(map); olaf@20: return NULL; olaf@20: } olaf@20: map->size = size; olaf@20: olaf@20: return map; olaf@20: } olaf@20: olaf@20: int ucx_map_put(UcxMap *map, UcxKey key, void *data) { olaf@20: if(key.hash == 0) { olaf@20: key.hash = ucx_hash((char*)key.data, key.len); olaf@20: } olaf@20: void *kd = malloc(key.len); olaf@20: memcpy(kd, key.data, key.len); olaf@20: key.data = kd; olaf@20: olaf@20: UcxMapElement *elm = &map->map[key.hash%map->size]; olaf@20: if(elm->next != NULL) { olaf@20: while(elm->next != NULL) { olaf@20: elm = elm->next; olaf@20: } olaf@20: UcxMapElement *e = (UcxMapElement*)malloc(sizeof(UcxMapElement)); olaf@20: if(e == NULL) { olaf@20: return -1; olaf@20: } olaf@20: elm->next = e; olaf@20: elm = e; olaf@20: } olaf@20: olaf@20: elm->key = key; olaf@20: elm->data = data; olaf@20: olaf@20: return 0; olaf@20: } olaf@20: olaf@20: void* ucx_map_get(UcxMap *map, UcxKey key) { olaf@20: if(key.hash == 0) { olaf@20: key.hash = ucx_hash((char*)key.data, key.len); olaf@20: } olaf@20: olaf@20: UcxMapElement *elm = &map->map[key.hash%map->size]; olaf@20: while(elm != NULL) { olaf@20: if(elm->key.hash == key.hash) { olaf@20: int n = (key.len > elm->key.len) ? elm->key.len : key.len; olaf@20: if(memcmp(elm->key.data, key.data, n) == 0) { olaf@20: return elm->data; olaf@20: } olaf@20: } olaf@20: elm = elm->next; olaf@20: } olaf@20: olaf@20: return NULL; olaf@20: } olaf@20: olaf@20: UcxKey ucx_key(void *data, size_t len) { olaf@20: UcxKey key; olaf@20: key.data = data; olaf@20: key.len = len; olaf@20: key.hash = ucx_hash(data, len); olaf@20: return key; olaf@20: } olaf@20: olaf@20: olaf@20: int ucx_hash(char *data, size_t len) { olaf@20: /* murmur hash 2 */ olaf@20: olaf@20: int m = 0x5bd1e995; olaf@20: int r = 24; olaf@20: olaf@20: int h = 25 ^ len; olaf@20: olaf@20: int i = 0; olaf@20: while (len >= 4) { olaf@20: int k = data[i + 0] & 0xFF; olaf@20: k |= (data[i + 1] & 0xFF) << 8; olaf@20: k |= (data[i + 2] & 0xFF) << 16; olaf@20: k |= (data[i + 3] & 0xFF) << 24; olaf@20: olaf@20: k *= m; olaf@20: k ^= k >> r; olaf@20: k *= m; olaf@20: olaf@20: h *= m; olaf@20: h ^= k; olaf@20: olaf@20: i += 4; olaf@20: len -= 4; olaf@20: } olaf@20: olaf@20: switch (len) { olaf@20: case 3: h ^= (data[i + 2] & 0xFF) << 16; olaf@20: case 2: h ^= (data[i + 1] & 0xFF) << 8; olaf@20: case 1: h ^= (data[i + 0] & 0xFF); h *= m; olaf@20: } olaf@20: olaf@20: h ^= h >> 13; olaf@20: h *= m; olaf@20: h ^= h >> 15; olaf@20: olaf@20: return h; olaf@20: }