cs_example.c

   1 #include <assert.h>
   2 #include <pthread.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5
   6 #include "deps/dragonnet/listen.h"
   7 #include "dnet-types.h"
   8
   9 static DragonnetListener *l;
  10
  11 static void connect_func(DragonnetPeer *p)
  12 {
  13         char *str_addr = dragonnet_addr_str(p->raddr);
  14         printf("new connection from %s\n", str_addr);
  15         free(str_addr);
  16 }
  17
  18 static void handle_pingpacket(DragonnetPeer *p, PingPacket *ping)
  19 {
  20         printf("PingPacket siz: 0x%08x\n", ping->compr_blob->siz);
  21         printf("PingPacket compr_siz: 0x%08x\n", ping->compr_blob->compr_siz);
  22         printf("PingPacket data: 0x%02x%02x\n", ping->compr_blob->data[0], ping->compr_blob->data[1]);
  23         free(ping->compr_blob->data);
  24
  25         dragonnet_peer_send_PongPacket(p, &(PongPacket) {
  26                 .number = 0xdba
  27         });
  28
  29         dragonnet_listener_close(l);
  30         dragonnet_listener_delete(l);
  31         l = NULL;
  32
  33         dragonnet_peer_close(p);
  34         dragonnet_peer_delete(p);
  35 }
  36
  37 static void handle_pongpacket(DragonnetPeer *p, PongPacket *pong)
  38 {
  39         printf("PongPacket number: 0x%08x\n", pong->number);
  40
  41         dragonnet_peer_close(p);
  42         dragonnet_peer_delete(p);
  43 }
  44
  45 static void *srv_func(__attribute((unused)) void *unused)
  46 {
  47         l = dragonnet_listener_new("[::1]:50000", &connect_func);
  48         assert(l != NULL);
  49
  50         dragonnet_listener_set_recv_hook(l, DRAGONNET_TYPE_PINGPACKET,
  51                         (void (*)(DragonnetPeer *, void *)) &handle_pingpacket);
  52         dragonnet_listener_run(l);
  53         return NULL;
  54 }
  55
  56 static void *clt_func(__attribute((unused)) void *unused)
  57 {
  58         while (l == NULL);
  59
  60         DragonnetPeer *p = dragonnet_connect("[::1]:50000");
  61         assert(p != NULL);
  62
  63         dragonnet_peer_set_recv_hook(p, DRAGONNET_TYPE_PONGPACKET,
  64                         (void (*)(DragonnetPeer *, void *)) &handle_pongpacket);
  65         dragonnet_peer_run(p);
  66
  67         CompressedBlob blob = malloc(sizeof *blob);
  68         blob->siz = 2;
  69         blob->data = (u8 *) "\x0d\xba";
  70         dragonnet_peer_send_PingPacket(p, &(PingPacket) {
  71                 .compr_blob = blob
  72         });
  73
  74         return NULL;
  75 }
  76
  77 int main(__attribute((unused)) int argc, __attribute((unused)) char **argv)
  78 {
  79         pthread_t srv_thread, clt_thread;
  80         pthread_create(&srv_thread, NULL, &srv_func, NULL);
  81         pthread_create(&clt_thread, NULL, &clt_func, NULL);
  82
  83         pthread_join(clt_thread, NULL);
  84         pthread_join(srv_thread, NULL);
  85
  86         while (true);
  87 }