dht.h

   1 /*
   2  * Copyright (c) 2016 Mohamed Aslan <maslan@sce.carleton.ca>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #ifndef DHT_H
  18 #define DHT_H
  19
  20 #include <sys/socket.h>
  21
  22 #define DHT_USEUDP      (1 << 0)
  23
  24
  25 struct dht_node {
  26         uint64_t token;
  27         int port;
  28         int socket;
  29         int ev_queue;
  30         int n_replicas;
  31         int n_peers;
  32         int def_r;
  33         int def_w;
  34         uint8_t ready;
  35         uint32_t flags;
  36         struct sockaddr_storage saddr;
  37         struct dht_options *opts;
  38         struct dht_peer *peers;
  39         void *ht;
  40 };
  41
  42 struct dht_peer {
  43         uint64_t token;
  44         int sock;
  45         time_t last_recv;
  46         struct sockaddr_storage s_addr;
  47 };
  48
  49 struct dht_options {
  50         uint64_t (*partitioner)(const char *);
  51         int (*ht_init)(void *);
  52         int (*ht_put)(void *, const char *, const char *);
  53         int (*ht_get)(void *, const char *, char **);
  54         int (*ht_del)(void *, const char *);
  55 };
  56
  57 int dht_init(struct dht_node *, int, int, uint32_t, struct dht_options *);
  58 void dht_event_loop(struct dht_node *);
  59 void dht_add_peer(struct dht_node *, const char *, int);
  60 int dht_put_tunable(int, const char *, int);
  61 int dht_get_tunable(int, const char *, char **, int);
  62 int dht_put(int, const char *);
  63 int dht_get(int, const char *, char **);
  64
  65 #endif
  66