peer.c

   1 #include <assert.h>
   2 #include <stdbool.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5
   6 #include "peer.h"
   7 #include "recv_thread.h"
   8
   9 static bool dragonnet_peer_init(DragonnetPeer *p, char *addr, void (*on_recv_type)(struct dragonnet_peer *, u16))
  10 {
  11         pthread_rwlock_init(&p->mu, NULL);
  12         pthread_rwlock_wrlock(&p->mu);
  13
  14         p->sock = socket(AF_INET6, SOCK_STREAM, 0);
  15         p->raddr = dragonnet_addr_parse_str(addr);
  16         p->on_recv_type = on_recv_type;
  17
  18         struct sockaddr_in6 sock_addr = dragonnet_addr_sock(p->raddr);
  19         if (connect(p->sock, (const struct sockaddr *) &sock_addr,
  20                         sizeof sock_addr) < 0) {
  21                 perror("connect");
  22                 return false;
  23         }
  24
  25         struct sockaddr_in6 sock_name;
  26         socklen_t sock_namelen = sizeof sock_name;
  27
  28         if (getsockname(p->sock, (struct sockaddr *) &sock_name, &sock_namelen) < 0) {
  29                 perror("getsockname");
  30                 return false;
  31         }
  32
  33         p->laddr = dragonnet_addr_parse_sock(sock_name);
  34
  35         pthread_rwlock_unlock(&p->mu);
  36         return true;
  37 }
  38
  39 DragonnetPeer *dragonnet_connect(char *addr, void (*on_recv_type)(struct dragonnet_peer *, u16))
  40 {
  41         DragonnetPeer *p = malloc(sizeof *p);
  42         if (!dragonnet_peer_init(p, addr, on_recv_type)) {
  43                 dragonnet_peer_delete(p);
  44                 return NULL;
  45         }
  46
  47         return p;
  48 }
  49
  50 void dragonnet_peer_run(DragonnetPeer *p)
  51 {
  52         pthread_rwlock_wrlock(&p->mu);
  53         pthread_create(&p->recv_thread, NULL, &dragonnet_peer_recv_thread, p);
  54         pthread_rwlock_unlock(&p->mu);
  55
  56         while (p->state < DRAGONNET_PEER_ACTIVE);
  57 }
  58
  59 void dragonnet_peer_close(DragonnetPeer *p)
  60 {
  61         pthread_rwlock_wrlock(&p->mu);
  62
  63         pthread_t recv_thread = p->recv_thread;
  64         if (p->state == DRAGONNET_PEER_ACTIVE)
  65                 shutdown(p->sock, SHUT_RDWR);
  66
  67         pthread_rwlock_unlock(&p->mu);
  68
  69         pthread_cancel(recv_thread);
  70         pthread_join(recv_thread, NULL);
  71 }
  72
  73 void dragonnet_peer_delete(DragonnetPeer *p)
  74 {
  75         pthread_rwlock_destroy(&p->mu);
  76         free(p);
  77 }