3 #include <dragonnet/listen.h>
4 #include <dragonnet/recv.h>
11 #include <sys/socket.h>
18 static bool dragonnet_peer_init_accepted(DragonnetPeer *p, int sock,
19 struct sockaddr_in6 addr, DragonnetListener *l)
21 pthread_mutex_init(&p->mtx, NULL);
25 p->raddr = dragonnet_addr_parse_sock(addr);
26 p->on_disconnect = l->on_disconnect;
27 p->on_recv = l->on_recv;
28 p->on_recv_type = l->on_recv_type;
33 static DragonnetPeer *dragonnet_peer_accept(int sock, struct sockaddr_in6 addr,
36 DragonnetPeer *p = malloc(sizeof *p);
37 if (!dragonnet_peer_init_accepted(p, sock, addr, l)) {
38 pthread_mutex_destroy(&p->mtx);
50 DragonnetListener *dragonnet_listener_new(char *addr)
52 DragonnetListener *l = malloc(sizeof *l);
55 l->sock = socket(AF_INET6, SOCK_STREAM, 0);
57 l->on_disconnect = NULL;
59 l->on_recv_type = calloc(sizeof *l->on_recv_type, dragonnet_num_types);
62 if (setsockopt(l->sock, SOL_SOCKET, SO_REUSEADDR, &so_reuseaddr,
63 sizeof so_reuseaddr) < 0) {
65 dragonnet_listener_delete(l);
69 l->laddr = dragonnet_addr_parse_str(addr);
70 struct sockaddr_in6 ai_addr = dragonnet_addr_sock(l->laddr);
72 if (bind(l->sock, (const struct sockaddr *) &ai_addr, sizeof ai_addr) < 0) {
74 dragonnet_listener_delete(l);
78 if (listen(l->sock, 10) < 0) {
80 dragonnet_listener_delete(l);
87 static void *listener_main(void *g_listener)
90 pthread_setname_np(pthread_self(), "listen");
93 DragonnetListener *l = (DragonnetListener *) g_listener;
96 struct sockaddr_in6 clt_addr;
97 socklen_t clt_addrlen = sizeof clt_addr;
99 int clt_sock = accept(l->sock, (struct sockaddr *) &clt_addr, &clt_addrlen);
106 DragonnetPeer *p = dragonnet_peer_accept(clt_sock, clt_addr, l);
110 void (*on_connect)(DragonnetPeer *) = l->on_connect;
112 if (on_connect != NULL)
115 dragonnet_peer_run(p);
121 void dragonnet_listener_run(DragonnetListener *l)
123 pthread_create(&l->accept_thread, NULL, &listener_main, l);
126 void dragonnet_listener_close(DragonnetListener *l)
130 pthread_kill(l->accept_thread, SIGINT);
131 pthread_join(l->accept_thread, NULL);
134 void dragonnet_listener_delete(DragonnetListener *l)
136 free(l->on_recv_type);