7 #include "recv_thread.h"
9 static bool dragonnet_peer_init(DragonnetPeer *p, char *addr, void (*on_recv_type)(struct dragonnet_peer *, u16))
11 pthread_rwlock_init(&p->mu, NULL);
12 pthread_rwlock_wrlock(&p->mu);
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;
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) {
25 struct sockaddr_in6 sock_name;
26 socklen_t sock_namelen = sizeof sock_name;
28 if (getsockname(p->sock, (struct sockaddr *) &sock_name, &sock_namelen) < 0) {
29 perror("getsockname");
33 p->laddr = dragonnet_addr_parse_sock(sock_name);
35 pthread_rwlock_unlock(&p->mu);
39 DragonnetPeer *dragonnet_connect(char *addr, void (*on_recv_type)(struct dragonnet_peer *, u16))
41 DragonnetPeer *p = malloc(sizeof *p);
42 if (!dragonnet_peer_init(p, addr, on_recv_type)) {
43 dragonnet_peer_delete(p);
50 void dragonnet_peer_run(DragonnetPeer *p)
52 pthread_rwlock_wrlock(&p->mu);
53 pthread_create(&p->recv_thread, NULL, &dragonnet_peer_recv_thread, p);
54 pthread_rwlock_unlock(&p->mu);
56 while (p->state < DRAGONNET_PEER_ACTIVE);
59 void dragonnet_peer_close(DragonnetPeer *p)
61 pthread_rwlock_wrlock(&p->mu);
63 pthread_t recv_thread = p->recv_thread;
64 if (p->state == DRAGONNET_PEER_ACTIVE)
65 shutdown(p->sock, SHUT_RDWR);
67 pthread_rwlock_unlock(&p->mu);
69 pthread_cancel(recv_thread);
70 pthread_join(recv_thread, NULL);
73 void dragonnet_peer_delete(DragonnetPeer *p)
75 pthread_rwlock_destroy(&p->mu);