2 #include <dragonnet/listen.h>
3 #include <dragonnet/recv.h>
7 #include <sys/socket.h>
14 static bool dragonnet_peer_init_accepted(DragonnetPeer *p, int sock,
15 struct sockaddr_in6 addr, DragonnetListener *l)
17 pthread_mutex_init(&p->mtx, NULL);
21 p->raddr = dragonnet_addr_parse_sock(addr);
22 p->on_disconnect = l->on_disconnect;
23 p->on_recv = l->on_recv;
24 p->on_recv_type = l->on_recv_type;
29 static DragonnetPeer *dragonnet_peer_accept(int sock, struct sockaddr_in6 addr,
32 DragonnetPeer *p = malloc(sizeof *p);
33 if (!dragonnet_peer_init_accepted(p, sock, addr, l)) {
34 pthread_mutex_destroy(&p->mtx);
46 DragonnetListener *dragonnet_listener_new(char *addr)
48 DragonnetListener *l = malloc(sizeof *l);
49 pthread_rwlock_init(&l->mu, NULL);
52 l->sock = socket(AF_INET6, SOCK_STREAM, 0);
54 l->on_disconnect = NULL;
56 l->on_recv_type = calloc(sizeof *l->on_recv_type, dragonnet_num_types);
59 if (setsockopt(l->sock, SOL_SOCKET, SO_REUSEADDR, &so_reuseaddr,
60 sizeof so_reuseaddr) < 0) {
62 dragonnet_listener_delete(l);
66 l->laddr = dragonnet_addr_parse_str(addr);
67 struct sockaddr_in6 ai_addr = dragonnet_addr_sock(l->laddr);
69 if (bind(l->sock, (const struct sockaddr *) &ai_addr, sizeof ai_addr) < 0) {
71 dragonnet_listener_delete(l);
75 if (listen(l->sock, 10) < 0) {
77 dragonnet_listener_delete(l);
84 static void *listener_main(void *g_listener)
86 DragonnetListener *l = (DragonnetListener *) g_listener;
89 struct sockaddr_in6 clt_addr;
90 socklen_t clt_addrlen = sizeof clt_addr;
92 pthread_rwlock_rdlock(&l->mu);
94 pthread_rwlock_unlock(&l->mu);
96 int clt_sock = accept(sock, (struct sockaddr *) &clt_addr, &clt_addrlen);
102 DragonnetPeer *p = dragonnet_peer_accept(clt_sock, clt_addr, l);
106 pthread_rwlock_rdlock(&l->mu);
107 void (*on_connect)(DragonnetPeer *) = l->on_connect;
108 pthread_rwlock_unlock(&l->mu);
110 if (on_connect != NULL)
113 dragonnet_peer_run(p);
119 void dragonnet_listener_run(DragonnetListener *l)
121 pthread_create(&l->accept_thread, NULL, &listener_main, l);
124 void dragonnet_listener_close(DragonnetListener *l)
127 pthread_cancel(l->accept_thread);
128 pthread_join(l->accept_thread, NULL);
131 void dragonnet_listener_delete(DragonnetListener *l)
133 pthread_rwlock_destroy(&l->mu);