1 files changed, 12 insertions, 9 deletions

diff --git a/proxy.c b/proxy.c
index 63b4c4f..a5c6bb7 100644
--- a/proxy.c
+++ b/proxy.c
@@ -19,6 +19,7 @@
 
 /* Opentracker */
 #include "ot_batch.h"
+#include "ot_fdstate.h"
 #include "ot_ip.h"
 #include "ot_mutex.h"
 #include "ot_random.h"
@@ -148,7 +149,7 @@ size_t add_peer_to_torrent_proxy(ot_hash hash, ot_peer *peer, size_t peer_size)
 size_t remove_peer_from_torrent_proxy(ot_hash hash, ot_peer *peer, size_t peer_size) {
   int         exactmatch;
   ot_vector  *torrents_list = mutex_bucket_lock_by_hash(hash);
-  ot_torrent *torrent       = binary_search(hash, torrents_list->data, torrents_list->size, sizeof(ot_torrent), OT_HASH_COMPARE_SIZE, &exactmatch);
+  ot_torrent *torrent       = vector_binary_search(hash, torrents_list->data, torrents_list->size, sizeof(ot_torrent), OT_HASH_COMPARE_SIZE, &exactmatch);
 
   if (exactmatch) {
     ot_peerlist *peer_list = peer_list = peer_size == OT_PEER_SIZE6 ? torrent->peer_list6 : torrent->peer_list4;
@@ -267,7 +268,7 @@ static void handle_reconnects(void) {
         continue; /* No socket for you */
       io_fd(newfd);
       io_wantwrite(newfd); /* So we will be informed when it is connected */
-      io_setcookie(newfd, g_connections + i);
+      fdstate_open(newfd, g_connections + i, 0);
 
       /* Prepare connection info block */
       reset_info_block(g_connections + i);
@@ -310,7 +311,7 @@ static void handle_accept(int64_t serversocket) {
       g_connections[i].port = port;
       g_connections[i].fd   = newfd;
 
-      io_setcookie(newfd, g_connections + i);
+      fdstate_open(newfd, g_connections + i, 0);
 
       /* We expect the connecting side to begin with its tracker_id */
       io_wantread(newfd);
@@ -325,7 +326,7 @@ static void handle_read(int64_t peersocket) {
   int         i;
   int64_t     datalen;
   uint32_t    tracker_id;
-  proxy_peer *peer = io_getcookie(peersocket);
+  proxy_peer *peer = fdstate_cookie(peersocket);
 
   if (!peer) {
     /* Can't happen ;) */
@@ -391,7 +392,7 @@ static void handle_read(int64_t peersocket) {
 
 /* Can write new sync data to the stream */
 static void handle_write(int64_t peersocket) {
-  proxy_peer *peer = io_getcookie(peersocket);
+  proxy_peer *peer = fdstate_cookie(peersocket);
 
   if (!peer) {
     /* Can't happen ;) */
@@ -464,7 +465,7 @@ static void server_mainloop() {
 
     /* Loop over readable sockets */
     while ((sock = io_canread()) != -1) {
-      const void *cookie = io_getcookie(sock);
+      const void *cookie = fdstate_cookie(sock);
       if ((uintptr_t)cookie == FLAG_SERVERSOCKET)
         handle_accept(sock);
       else
@@ -493,7 +494,7 @@ static int64_t ot_try_bind(ot_ip6 ip, uint16_t port) {
   if (!io_fd(sock))
     panic("io_fd");
 
-  io_setcookie(sock, (void *)FLAG_SERVERSOCKET);
+  fdstate_open(sock, (void *)FLAG_SERVERSOCKET, 0);
   io_wantread(sock);
   return sock;
 }
@@ -506,7 +507,7 @@ static int scan_ip6_port(const char *src, ot_ip6 ip, uint16_t *port) {
     ++s;
   if (*s == '[')
     ++s, ++bracket; /* for v6 style notation */
-  if (!(off = ip_scan(s, ip)))
+  if (!(off = ip_parse(s, ip)))
     return 0;
   s += off;
   if (*s == 0 || isspace(*s))
@@ -521,7 +522,7 @@ static int scan_ip6_port(const char *src, ot_ip6 ip, uint16_t *port) {
     if (*(s++) != ':')
       return 0;
   }
-  if (!(off = scan_udec(s, &value, 65535)))
+  if (!(off = scan_uint(s, &value, 65535)))
     return 0;
   *port = (uint16_t)value;
   return off + s - src;
@@ -535,6 +536,8 @@ int main(int argc, char **argv) {
   int              scanon = 1, lbound = 0, sbound = 0;
 
   random_init();
+  if (fdstate_init() == -1)
+    panic("fdstate_init");
   g_tracker_id = random_uint32();
 
   while (scanon) {