3 #include <dragonnet/addr.h>
4 #include <dragonnet/listen.h>
5 #include <dragonnet/recv.h>
12 #include <sys/socket.h>
19 static bool dragonnet_peer_init_accepted(DragonnetPeer *p, int sock,
20 char *addr, DragonnetListener *l)
22 pthread_mutex_init(&p->mtx, NULL);
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, char *addr,
36 DragonnetPeer *p = malloc(sizeof *p);
37 if (!dragonnet_peer_init_accepted(p, sock, addr, l)) {
38 pthread_mutex_destroy(&p->mtx);
51 DragonnetListener *dragonnet_listener_new(char *addr)
53 struct addrinfo *info = dragonnet_str2addr(addr);
57 DragonnetListener *l = malloc(sizeof *l);
60 l->sock = socket(info->ai_family, info->ai_socktype, info->ai_protocol);
61 l->address = dragonnet_addr2str(info->ai_addr, info->ai_addrlen);
63 l->on_disconnect = NULL;
65 l->on_recv_type = calloc(sizeof *l->on_recv_type, dragonnet_num_types);
68 if (setsockopt(l->sock, SOL_SOCKET, SO_REUSEADDR, &so_reuseaddr,
69 sizeof so_reuseaddr) < 0) {
72 dragonnet_listener_delete(l);
76 if (bind(l->sock, info->ai_addr, info->ai_addrlen) < 0) {
79 dragonnet_listener_delete(l);
85 if (listen(l->sock, 10) < 0) {
87 dragonnet_listener_delete(l);
94 static void *listener_main(void *g_listener)
97 pthread_setname_np(pthread_self(), "listen");
100 DragonnetListener *l = (DragonnetListener *) g_listener;
103 struct sockaddr_storage clt_addr;
104 socklen_t clt_addrlen = sizeof clt_addr;
106 int clt_sock = accept(l->sock, (struct sockaddr *) &clt_addr, &clt_addrlen);
113 char *clt_addstr = dragonnet_addr2str((struct sockaddr *) &clt_addr, clt_addrlen);
114 DragonnetPeer *p = dragonnet_peer_accept(clt_sock, clt_addstr, l);
118 void (*on_connect)(DragonnetPeer *) = l->on_connect;
120 if (on_connect != NULL)
123 dragonnet_peer_run(p);
129 void dragonnet_listener_run(DragonnetListener *l)
131 pthread_create(&l->accept_thread, NULL, &listener_main, l);
134 void dragonnet_listener_close(DragonnetListener *l)
138 pthread_kill(l->accept_thread, SIGINT);
139 pthread_join(l->accept_thread, NULL);
142 void dragonnet_listener_delete(DragonnetListener *l)
144 free(l->on_recv_type);