add magic_endpoint_conn_type_cb call in main.c client side code

Author: ramfox

main.c

@@ -179,6 +179,24 @@ run_server (MagicEndpointConfig_t * config, slice_ref_uint8_t alpn_slice, bool j
   return 0;
 }
 
+typedef struct ConnectionStatus {
+  MagicEndpointResult_t res;
+  ConnectionType_t conn_type;
+} ConnectionStatus;
+
+void
+callback(
+  void const * ctx,
+  MagicEndpointResult_t res,
+  ConnectionType_t conn_type
+)
+{
+  ConnectionStatus *cs;
+  cs = (ConnectionStatus *)ctx;
+  cs->res = res;
+  cs->conn_type = conn_type;
+}
+
 int
 run_client (
   MagicEndpointConfig_t * config,
@@ -238,6 +256,10 @@ run_client (
     return -1;
   }
 
+  ConnectionStatus *conn_status;
+  conn_status = malloc(sizeof(ConnectionStatus));
+  magic_endpoint_conn_type_cb(ep, (const void *)conn_status, &node_addr.node_id, callback);
+
   SendStream_t * send_stream = send_stream_default();
   ret = connection_open_uni(&conn, &send_stream);
   if (ret != 0) {
@@ -307,6 +329,30 @@ run_client (
   recv_str[read] = '\0';
   printf("received: '%s'\n", recv_str);
 
+  // check that we were able to use the conn_type callback
+  if (conn_status->res != MAGIC_ENDPOINT_RESULT_OK) {
+    fprintf(stderr, "callback failed to send a connection type\n");
+    return -1;
+  } else {
+    switch (conn_status->conn_type) {
+      case CONNECTION_TYPE_DIRECT:
+        printf("had a direct connection\n");
+        break;
+      case CONNECTION_TYPE_RELAY:
+        printf("had a relay connection\n");
+        break;
+      case CONNECTION_TYPE_MIXED:
+        printf("had a mixed connection\n");
+        break;
+      case CONNECTION_TYPE_NONE:
+        fprintf(stderr, "callback reported no connection\n");
+        return -1;
+      default:
+        fprintf(stderr, "unknown connection type reported: %i\n", conn_status->conn_type);
+        return -1;
+    }
+  }
+
   // finish
   ret = send_stream_finish(send_stream);
   if (ret != 0) {
@@ -317,6 +363,7 @@ run_client (
   // cleanup
   free(recv_str);
   free(recv_buffer);
+  free(conn_status);
   recv_stream_free(recv_stream);
   connection_free(conn);