| |
| #include <stdint.h> |
| #include <stdlib.h> |
| #include <stdio.h> |
| #include <errno.h> |
| |
| #include <pthread.h> |
| |
| #include <talloc.h> |
| |
| #include <osmocom/core/linuxlist.h> |
| |
| #include "bankd.h" |
| |
| /* thread-safe lookup of map by client:slot */ |
| struct bankd_slot_mapping *bankd_slotmap_by_client(struct bankd *bankd, const struct client_slot *client) |
| { |
| struct bankd_slot_mapping *map; |
| |
| pthread_rwlock_rdlock(&bankd->slot_mappings_rwlock); |
| llist_for_each_entry(map, &bankd->slot_mappings, list) { |
| if (client_slot_equals(&map->client, client)) { |
| pthread_rwlock_unlock(&bankd->slot_mappings_rwlock); |
| return map; |
| } |
| } |
| pthread_rwlock_unlock(&bankd->slot_mappings_rwlock); |
| return NULL; |
| } |
| |
| /* thread-safe lookup of map by bank:slot */ |
| struct bankd_slot_mapping *bankd_slotmap_by_bank(struct bankd *bankd, const struct bank_slot *bank) |
| { |
| struct bankd_slot_mapping *map; |
| |
| pthread_rwlock_rdlock(&bankd->slot_mappings_rwlock); |
| llist_for_each_entry(map, &bankd->slot_mappings, list) { |
| if (bank_slot_equals(&map->bank, bank)) { |
| pthread_rwlock_unlock(&bankd->slot_mappings_rwlock); |
| return map; |
| } |
| } |
| pthread_rwlock_unlock(&bankd->slot_mappings_rwlock); |
| return NULL; |
| |
| } |
| |
| /* thread-safe creating of a new bank<->client map */ |
| int bankd_slotmap_add(struct bankd *bankd, const struct bank_slot *bank, const struct client_slot *client) |
| { |
| struct bankd_slot_mapping *map; |
| |
| /* We assume a single thread (main thread) will ever update the mappings, |
| * and hence we don't have any races by first grabbing + releasing the read |
| * lock twice before grabbing the writelock below */ |
| |
| map = bankd_slotmap_by_bank(bankd, bank); |
| if (map) { |
| fprintf(stderr, "BANKD %u:%u already in use, cannot add new map\n", |
| bank->bank_id, bank->slot_nr); |
| return -EBUSY; |
| } |
| |
| map = bankd_slotmap_by_client(bankd, client); |
| if (map) { |
| fprintf(stderr, "CLIENT %u:%u already in use, cannot add new map\n", |
| client->client_id, client->slot_nr); |
| return -EBUSY; |
| } |
| |
| /* allocate new mapping and add to list of mappings */ |
| map = talloc_zero(bankd, struct bankd_slot_mapping); |
| if (!map) |
| return -ENOMEM; |
| |
| map->bank = *bank; |
| map->client = *client; |
| |
| pthread_rwlock_wrlock(&bankd->slot_mappings_rwlock); |
| llist_add_tail(&map->list, &bankd->slot_mappings); |
| pthread_rwlock_unlock(&bankd->slot_mappings_rwlock); |
| |
| printf("Added Slot Map C(%u:%u) <-> B(%u:%u)\n", |
| map->client.client_id, map->client.slot_nr, map->bank.bank_id, map->bank.slot_nr); |
| |
| return 0; |
| } |
| |
| /* thread-safe removal of a bank<->client map */ |
| void bankd_slotmap_del(struct bankd *bankd, struct bankd_slot_mapping *map) |
| { |
| printf("Deleting Slot Map C(%u:%u) <-> B(%u:%u)\n", |
| map->client.client_id, map->client.slot_nr, map->bank.bank_id, map->bank.slot_nr); |
| |
| pthread_rwlock_wrlock(&bankd->slot_mappings_rwlock); |
| llist_del(&map->list); |
| pthread_rwlock_unlock(&bankd->slot_mappings_rwlock); |
| |
| talloc_free(map); |
| } |