dragonnet/peer.c

   1 #include <assert.h>
   2 #include <stdbool.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include "addr.h"
   6 #include "peer.h"
   7 #include "recv.h"
   8 #include "recv_thread.h"
   9 #include "sock.h"
  10
  11 static bool dragonnet_peer_init(DragonnetPeer *p, char *addr)
  12 {
  13         pthread_mutex_init(&p->mtx, NULL);
  14
  15         struct addrinfo *info = dragonnet_str2addr(addr);
  16         if (!info)
  17                 return false;
  18
  19         p->sock = socket(info->ai_family, info->ai_socktype, info->ai_protocol);
  20         p->address = dragonnet_addr2str(info->ai_addr, info->ai_addrlen);
  21
  22         if (connect(p->sock, info->ai_addr, info->ai_addrlen) < 0) {
  23                 freeaddrinfo(info);
  24                 perror("connect");
  25                 return false;
  26         }
  27
  28         freeaddrinfo(info);
  29
  30         p->on_disconnect = NULL;
  31         p->on_recv = NULL;
  32         p->on_recv_type = calloc(sizeof *p->on_recv_type, dragonnet_num_types); // fixme: memory leak
  33
  34         return true;
  35 }
  36
  37 DragonnetPeer *dragonnet_connect(char *addr)
  38 {
  39         DragonnetPeer *p = malloc(sizeof *p);
  40         if (!dragonnet_peer_init(p, addr)) {
  41                 pthread_mutex_destroy(&p->mtx);
  42                 free(p);
  43                 return NULL;
  44         }
  45
  46         return p;
  47 }
  48
  49 void dragonnet_peer_run(DragonnetPeer *p)
  50 {
  51         pthread_create(&p->recv_thread, NULL, &dragonnet_peer_recv_thread, p);
  52 }
  53
  54 void dragonnet_peer_shutdown(DragonnetPeer *p)
  55 {
  56         shutdown(p->sock, SHUT_RDWR);
  57 }