src/mesh/gnunet-service-mesh_local.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2013 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20
  21
  22 #include "platform.h"
  23 #include "gnunet_util_lib.h"
  24
  25 #include "gnunet_statistics_service.h"
  26
  27 #include "mesh.h"
  28 #include "mesh_protocol.h" /* GNUNET_MESH_Data is shared */
  29
  30 #include "gnunet-service-mesh_local.h"
  31 #include "gnunet-service-mesh_channel.h"
  32
  33 /* INFO DEBUG */
  34 #include "gnunet-service-mesh_tunnel.h"
  35
  36 #define LOG(level, ...) GNUNET_log_from(level,"mesh-loc",__VA_ARGS__)
  37
  38 /******************************************************************************/
  39 /********************************   STRUCTS  **********************************/
  40 /******************************************************************************/
  41
  42 /**
  43  * Struct containing information about a client of the service
  44  *
  45  * TODO: add a list of 'waiting' ports
  46  */
  47 struct MeshClient
  48 {
  49     /**
  50      * Linked list next
  51      */
  52   struct MeshClient *next;
  53
  54     /**
  55      * Linked list prev
  56      */
  57   struct MeshClient *prev;
  58
  59     /**
  60      * Tunnels that belong to this client, indexed by local id
  61      */
  62   struct GNUNET_CONTAINER_MultiHashMap32 *own_channels;
  63
  64     /**
  65      * Tunnels this client has accepted, indexed by incoming local id
  66      */
  67   struct GNUNET_CONTAINER_MultiHashMap32 *incoming_channels;
  68
  69     /**
  70      * Channel ID for the next incoming channel.
  71      */
  72   MESH_ChannelNumber next_chid;
  73
  74     /**
  75      * Handle to communicate with the client
  76      */
  77   struct GNUNET_SERVER_Client *handle;
  78
  79     /**
  80      * Ports that this client has declared interest in.
  81      * Indexed by port, contains *Client.
  82      */
  83   struct GNUNET_CONTAINER_MultiHashMap32 *ports;
  84
  85     /**
  86      * Whether the client is active or shutting down (don't send confirmations
  87      * to a client that is shutting down.
  88      */
  89   int shutting_down;
  90
  91     /**
  92      * ID of the client, mainly for debug messages
  93      */
  94   unsigned int id;
  95 };
  96
  97 /******************************************************************************/
  98 /*******************************   GLOBALS  ***********************************/
  99 /******************************************************************************/
 100
 101 /**
 102  * Global handle to the statistics service.
 103  */
 104 extern struct GNUNET_STATISTICS_Handle *stats;
 105
 106 /**
 107  * Handle to server lib.
 108  */
 109 static struct GNUNET_SERVER_Handle *server_handle;
 110
 111 /**
 112  * DLL with all the clients, head.
 113  */
 114 static struct MeshClient *clients_head;
 115
 116 /**
 117  * DLL with all the clients, tail.
 118  */
 119 static struct MeshClient *clients_tail;
 120
 121 /**
 122  * Next ID to assign to a client.
 123  */
 124 unsigned int next_client_id;
 125
 126 /**
 127  * All ports clients of this peer have opened.
 128  */
 129 static struct GNUNET_CONTAINER_MultiHashMap32 *ports;
 130
 131 /**
 132  * Notification context, to send messages to local clients.
 133  */
 134 static struct GNUNET_SERVER_NotificationContext *nc;
 135
 136
 137 /******************************************************************************/
 138 /********************************   STATIC  ***********************************/
 139 /******************************************************************************/
 140
 141 /**
 142  * Remove client's ports from the global hashmap on disconnect.
 143  *
 144  * @param cls Closure (unused).
 145  * @param key Port.
 146  * @param value Client structure.
 147  *
 148  * @return GNUNET_OK, keep iterating.
 149  */
 150 static int
 151 client_release_ports (void *cls,
 152                       uint32_t key,
 153                       void *value)
 154 {
 155   int res;
 156
 157   res = GNUNET_CONTAINER_multihashmap32_remove (ports, key, value);
 158   if (GNUNET_YES != res)
 159   {
 160     GNUNET_break (0);
 161     LOG (GNUNET_ERROR_TYPE_WARNING,
 162                 "Port %u by client %p was not registered.\n",
 163                 key, value);
 164   }
 165   return GNUNET_OK;
 166 }
 167
 168
 169
 170 /******************************************************************************/
 171 /********************************  HANDLES  ***********************************/
 172 /******************************************************************************/
 173
 174
 175 /**
 176  * Handler for client connection.
 177  *
 178  * @param cls Closure (unused).
 179  * @param client Client handler.
 180  */
 181 static void
 182 handle_client_connect (void *cls, struct GNUNET_SERVER_Client *client)
 183 {
 184   struct MeshClient *c;
 185
 186   LOG (GNUNET_ERROR_TYPE_DEBUG, "client connected: %p\n", client);
 187   if (NULL == client)
 188     return;
 189   c = GNUNET_new (struct MeshClient);
 190   c->handle = client;
 191   c->id = next_client_id++; /* overflow not important: just for debug */
 192   c->next_chid = GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
 193   GNUNET_SERVER_client_keep (client);
 194   GNUNET_SERVER_client_set_user_context (client, c);
 195   GNUNET_CONTAINER_DLL_insert (clients_head, clients_tail, c);
 196 }
 197
 198
 199 /**
 200  * Iterator for deleting each channel whose client endpoint disconnected.
 201  *
 202  * @param cls Closure (client that has disconnected).
 203  * @param key The local channel id (used to access the hashmap).
 204  * @param value The value stored at the key (channel to destroy).
 205  *
 206  * @return GNUNET_OK, keep iterating.
 207  */
 208 static int
 209 channel_destroy_iterator (void *cls,
 210                           uint32_t key,
 211                           void *value)
 212 {
 213   struct MeshChannel *ch = value;
 214   struct MeshClient *c = cls;
 215
 216   LOG (GNUNET_ERROR_TYPE_DEBUG,
 217               " Channel %s destroy, due to client %s shutdown.\n",
 218               GMCH_2s (ch), GML_2s (c));
 219
 220   GMCH_handle_local_destroy (ch, c, key < GNUNET_MESH_LOCAL_CHANNEL_ID_SERV);
 221   return GNUNET_OK;
 222 }
 223
 224 /**
 225  * Handler for client disconnection
 226  *
 227  * @param cls closure
 228  * @param client identification of the client; NULL
 229  *        for the last call when the server is destroyed
 230  */
 231 static void
 232 handle_client_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
 233 {
 234   struct MeshClient *c;
 235
 236   LOG (GNUNET_ERROR_TYPE_DEBUG, "client disconnected: %p\n", client);
 237   if (client == NULL)
 238   {
 239     LOG (GNUNET_ERROR_TYPE_DEBUG, "   (SERVER DOWN)\n");
 240     return;
 241   }
 242
 243   c = GML_client_get (client);
 244   if (NULL != c)
 245   {
 246     LOG (GNUNET_ERROR_TYPE_DEBUG, "matching client found (%u, %p)\n",
 247                 c->id, c);
 248     GNUNET_SERVER_client_drop (c->handle);
 249     c->shutting_down = GNUNET_YES;
 250     if (NULL != c->own_channels)
 251     {
 252       GNUNET_CONTAINER_multihashmap32_iterate (c->own_channels,
 253                                                &channel_destroy_iterator, c);
 254       GNUNET_CONTAINER_multihashmap32_destroy (c->own_channels);
 255     }
 256
 257     if (NULL != c->incoming_channels)
 258     {
 259       GNUNET_CONTAINER_multihashmap32_iterate (c->incoming_channels,
 260                                                &channel_destroy_iterator, c);
 261       GNUNET_CONTAINER_multihashmap32_destroy (c->incoming_channels);
 262     }
 263
 264     if (NULL != c->ports)
 265     {
 266       GNUNET_CONTAINER_multihashmap32_iterate (c->ports,
 267                                                &client_release_ports, c);
 268       GNUNET_CONTAINER_multihashmap32_destroy (c->ports);
 269     }
 270     GNUNET_CONTAINER_DLL_remove (clients_head, clients_tail, c);
 271     GNUNET_STATISTICS_update (stats, "# clients", -1, GNUNET_NO);
 272     LOG (GNUNET_ERROR_TYPE_DEBUG, "  client free (%p)\n", c);
 273     GNUNET_free (c);
 274   }
 275   else
 276   {
 277     LOG (GNUNET_ERROR_TYPE_WARNING, " context NULL!\n");
 278   }
 279   LOG (GNUNET_ERROR_TYPE_DEBUG, "done!\n");
 280   return;
 281 }
 282
 283
 284 /**
 285  * Handler for new clients
 286  *
 287  * @param cls closure
 288  * @param client identification of the client
 289  * @param message the actual message, which includes messages the client wants
 290  */
 291 static void
 292 handle_new_client (void *cls, struct GNUNET_SERVER_Client *client,
 293                    const struct GNUNET_MessageHeader *message)
 294 {
 295   struct GNUNET_MESH_ClientConnect *cc_msg;
 296   struct MeshClient *c;
 297   unsigned int size;
 298   uint32_t *p;
 299   unsigned int i;
 300
 301   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n");
 302   LOG (GNUNET_ERROR_TYPE_DEBUG, "new client connected %p\n", client);
 303
 304   /* Check data sanity */
 305   size = ntohs (message->size) - sizeof (struct GNUNET_MESH_ClientConnect);
 306   cc_msg = (struct GNUNET_MESH_ClientConnect *) message;
 307   if (0 != (size % sizeof (uint32_t)))
 308   {
 309     GNUNET_break (0);
 310     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 311     return;
 312   }
 313   size /= sizeof (uint32_t);
 314
 315   /* Initialize new client structure */
 316   c = GNUNET_SERVER_client_get_user_context (client, struct MeshClient);
 317   LOG (GNUNET_ERROR_TYPE_DEBUG, "  client id %u\n", c->id);
 318   LOG (GNUNET_ERROR_TYPE_DEBUG, "  client has %u ports\n", size);
 319   if (size > 0)
 320   {
 321     uint32_t u32;
 322
 323     p = (uint32_t *) &cc_msg[1];
 324     c->ports = GNUNET_CONTAINER_multihashmap32_create (size);
 325     for (i = 0; i < size; i++)
 326     {
 327       u32 = ntohl (p[i]);
 328       LOG (GNUNET_ERROR_TYPE_DEBUG, "    port: %u\n", u32);
 329
 330       /* store in client's hashmap */
 331       GNUNET_CONTAINER_multihashmap32_put (c->ports, u32, c,
 332                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
 333       /* store in global hashmap */
 334       /* FIXME only allow one client to have the port open,
 335        *       have a backup hashmap with waiting clients */
 336       GNUNET_CONTAINER_multihashmap32_put (ports, u32, c,
 337                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
 338     }
 339   }
 340
 341   c->own_channels = GNUNET_CONTAINER_multihashmap32_create (32);
 342   c->incoming_channels = GNUNET_CONTAINER_multihashmap32_create (32);
 343   GNUNET_SERVER_notification_context_add (nc, client);
 344   GNUNET_STATISTICS_update (stats, "# clients", 1, GNUNET_NO);
 345
 346   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 347   LOG (GNUNET_ERROR_TYPE_DEBUG, "new client processed\n");
 348 }
 349
 350
 351 /**
 352  * Handler for requests of new tunnels
 353  *
 354  * @param cls Closure.
 355  * @param client Identification of the client.
 356  * @param message The actual message.
 357  */
 358 static void
 359 handle_channel_create (void *cls, struct GNUNET_SERVER_Client *client,
 360                        const struct GNUNET_MessageHeader *message)
 361 {
 362   struct MeshClient *c;
 363
 364   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n");
 365   LOG (GNUNET_ERROR_TYPE_DEBUG, "new channel requested\n");
 366
 367   /* Sanity check for client registration */
 368   if (NULL == (c = GML_client_get (client)))
 369   {
 370     GNUNET_break (0);
 371     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 372     return;
 373   }
 374   LOG (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
 375
 376   /* Message size sanity check */
 377   if (sizeof (struct GNUNET_MESH_ChannelMessage) != ntohs (message->size))
 378   {
 379     GNUNET_break (0);
 380     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 381     return;
 382   }
 383
 384   if (GNUNET_OK !=
 385       GMCH_handle_local_create (c,
 386                                 (struct GNUNET_MESH_ChannelMessage *) message))
 387   {
 388     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 389     return;
 390   }
 391
 392   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 393   return;
 394 }
 395
 396
 397 /**
 398  * Handler for requests of deleting tunnels
 399  *
 400  * @param cls closure
 401  * @param client identification of the client
 402  * @param message the actual message
 403  */
 404 static void
 405 handle_channel_destroy (void *cls, struct GNUNET_SERVER_Client *client,
 406                         const struct GNUNET_MessageHeader *message)
 407 {
 408   struct GNUNET_MESH_ChannelMessage *msg;
 409   struct MeshClient *c;
 410   struct MeshChannel *ch;
 411   MESH_ChannelNumber chid;
 412
 413   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\nGot a DESTROY CHANNEL from client!\n");
 414
 415   /* Sanity check for client registration */
 416   if (NULL == (c = GML_client_get (client)))
 417   {
 418     GNUNET_break (0);
 419     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 420     return;
 421   }
 422   LOG (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
 423
 424   /* Message sanity check */
 425   if (sizeof (struct GNUNET_MESH_ChannelMessage) != ntohs (message->size))
 426   {
 427     GNUNET_break (0);
 428     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 429     return;
 430   }
 431
 432   msg = (struct GNUNET_MESH_ChannelMessage *) message;
 433
 434   /* Retrieve tunnel */
 435   chid = ntohl (msg->channel_id);
 436   LOG (GNUNET_ERROR_TYPE_DEBUG, "  for channel %X\n", chid);
 437   ch = GML_channel_get (c, chid);
 438   if (NULL == ch)
 439   {
 440     LOG (GNUNET_ERROR_TYPE_DEBUG, "  channel %X not found\n", chid);
 441     GNUNET_STATISTICS_update (stats,
 442                               "# client destroy messages on unknown channel",
 443                               1, GNUNET_NO);
 444     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 445     return;
 446   }
 447
 448   GMCH_handle_local_destroy (ch, c, chid < GNUNET_MESH_LOCAL_CHANNEL_ID_SERV);
 449
 450   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 451   return;
 452 }
 453
 454
 455 /**
 456  * Handler for client traffic
 457  *
 458  * @param cls closure
 459  * @param client identification of the client
 460  * @param message the actual message
 461  */
 462 static void
 463 handle_data (void *cls, struct GNUNET_SERVER_Client *client,
 464              const struct GNUNET_MessageHeader *message)
 465 {
 466   struct GNUNET_MESH_LocalData *msg;
 467   struct MeshClient *c;
 468   struct MeshChannel *ch;
 469   MESH_ChannelNumber chid;
 470   size_t size;
 471   int fwd;
 472
 473   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\nGot data from a client!\n");
 474
 475   /* Sanity check for client registration */
 476   if (NULL == (c = GML_client_get (client)))
 477   {
 478     GNUNET_break (0);
 479     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 480     return;
 481   }
 482   LOG (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
 483
 484   msg = (struct GNUNET_MESH_LocalData *) message;
 485
 486   /* Sanity check for message size */
 487   size = ntohs (message->size) - sizeof (struct GNUNET_MESH_LocalData);
 488   if (size < sizeof (struct GNUNET_MessageHeader))
 489   {
 490     GNUNET_break (0);
 491     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 492     return;
 493   }
 494
 495   /* Channel exists? */
 496   chid = ntohl (msg->id);
 497   LOG (GNUNET_ERROR_TYPE_DEBUG, "  on channel %X\n", chid);
 498   fwd = chid < GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
 499   ch = GML_channel_get (c, chid);
 500   if (NULL == ch)
 501   {
 502     GNUNET_STATISTICS_update (stats,
 503                               "# client data messages on unknown channel",
 504                               1, GNUNET_NO);
 505     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 506     return;
 507   }
 508
 509   if (GNUNET_OK !=
 510       GMCH_handle_local_data (ch, c,
 511                               (struct GNUNET_MessageHeader *)&msg[1], fwd))
 512   {
 513     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 514     return;
 515   }
 516
 517   LOG (GNUNET_ERROR_TYPE_DEBUG, "receive done OK\n");
 518   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 519
 520   return;
 521 }
 522
 523
 524 /**
 525  * Handler for client's ACKs for payload traffic.
 526  *
 527  * @param cls Closure (unused).
 528  * @param client Identification of the client.
 529  * @param message The actual message.
 530  */
 531 static void
 532 handle_ack (void *cls, struct GNUNET_SERVER_Client *client,
 533             const struct GNUNET_MessageHeader *message)
 534 {
 535   struct GNUNET_MESH_LocalAck *msg;
 536   struct MeshChannel *ch;
 537   struct MeshClient *c;
 538   MESH_ChannelNumber chid;
 539   int fwd;
 540
 541   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n");
 542   LOG (GNUNET_ERROR_TYPE_DEBUG, "Got a local ACK\n");
 543
 544   /* Sanity check for client registration */
 545   if (NULL == (c = GML_client_get (client)))
 546   {
 547     GNUNET_break (0);
 548     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 549     return;
 550   }
 551   LOG (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
 552
 553   msg = (struct GNUNET_MESH_LocalAck *) message;
 554
 555   /* Channel exists? */
 556   chid = ntohl (msg->channel_id);
 557   LOG (GNUNET_ERROR_TYPE_DEBUG, "  on channel %X\n", chid);
 558   ch = GML_channel_get (c, chid);
 559   LOG (GNUNET_ERROR_TYPE_DEBUG, "   -- ch %p\n", ch);
 560   if (NULL == ch)
 561   {
 562     LOG (GNUNET_ERROR_TYPE_DEBUG, "Channel %X unknown.\n", chid);
 563     LOG (GNUNET_ERROR_TYPE_DEBUG, "  for client %u.\n", c->id);
 564     GNUNET_STATISTICS_update (stats,
 565                               "# client ack messages on unknown channel",
 566                               1, GNUNET_NO);
 567     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 568     return;
 569   }
 570
 571   /* If client is root, the ACK is going FWD, therefore this is "BCK ACK". */
 572   /* If client is dest, the ACK is going BCK, therefore this is "FWD ACK" */
 573   fwd = chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
 574
 575   GMCH_handle_local_ack (ch, fwd);
 576   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 577
 578   return;
 579 }
 580
 581
 582 /**
 583  * Iterator over all tunnels to send a monitoring client info about each tunnel.
 584  *
 585  * @param cls Closure ().
 586  * @param peer Peer ID (tunnel remote peer).
 587  * @param value Tunnel info.
 588  *
 589  * @return #GNUNET_YES, to keep iterating.
 590  */
 591 static int
 592 monitor_all_tunnels_iterator (void *cls,
 593                               const struct GNUNET_PeerIdentity * peer,
 594                               void *value)
 595 {
 596   struct GNUNET_SERVER_Client *client = cls;
 597   struct MeshTunnel3 *t = value;
 598   struct GNUNET_MESH_LocalInfoTunnel msg;
 599
 600   msg.header.size = htons (sizeof (msg));
 601   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS);
 602   msg.destination = *peer;
 603   msg.channels = htonl (GMT_count_channels (t));
 604   msg.connections = htonl (GMT_count_connections (t));
 605   msg.cstate = htons ((uint16_t) GMT_get_cstate (t));
 606   msg.estate = htons ((uint16_t) GMT_get_estate (t));
 607
 608   LOG (GNUNET_ERROR_TYPE_DEBUG, "sending info about tunnel ->%s\n",
 609        GNUNET_i2s (peer));
 610
 611   GNUNET_SERVER_notification_context_unicast (nc, client,
 612                                               &msg.header, GNUNET_NO);
 613   return GNUNET_YES;
 614 }
 615
 616
 617 /**
 618  * Handler for client's INFO TUNNELS request.
 619  *
 620  * @param cls Closure (unused).
 621  * @param client Identification of the client.
 622  * @param message The actual message.
 623  */
 624 static void
 625 handle_get_tunnels (void *cls, struct GNUNET_SERVER_Client *client,
 626                     const struct GNUNET_MessageHeader *message)
 627 {
 628   struct MeshClient *c;
 629   struct GNUNET_MessageHeader reply;
 630
 631   /* Sanity check for client registration */
 632   if (NULL == (c = GML_client_get (client)))
 633   {
 634     GNUNET_break (0);
 635     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 636     return;
 637   }
 638
 639   LOG (GNUNET_ERROR_TYPE_DEBUG,
 640        "Received get tunnels request from client %u (%p)\n",
 641        c->id, client);
 642
 643   GMT_iterate_all (client, monitor_all_tunnels_iterator);
 644   reply.size = htons (sizeof (reply));
 645   reply.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS);
 646   GNUNET_SERVER_notification_context_unicast (nc, client, &reply, GNUNET_NO);
 647
 648   LOG (GNUNET_ERROR_TYPE_DEBUG,
 649        "Get tunnels request from client %u completed\n", c->id);
 650   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 651 }
 652
 653
 654 static void
 655 iter_connection (void *cls, struct MeshConnection *c)
 656 {
 657   struct GNUNET_MESH_LocalInfoTunnel *msg = cls;
 658   struct GNUNET_HashCode *h = (struct GNUNET_HashCode *) &msg[1];
 659
 660   h[msg->connections] = *(GMC_get_id (c));
 661   msg->connections++;
 662 }
 663
 664 static void
 665 iter_channel (void *cls, struct MeshChannel *ch)
 666 {
 667   struct GNUNET_MESH_LocalInfoTunnel *msg = cls;
 668   struct GNUNET_HashCode *h = (struct GNUNET_HashCode *) &msg[1];
 669   MESH_ChannelNumber *chn = (MESH_ChannelNumber *) &h[msg->connections];
 670
 671   chn[msg->channels] = GMCH_get_id (ch);
 672   msg->channels++;
 673 }
 674
 675
 676 /**
 677  * Handler for client's SHOW_TUNNEL request.
 678  *
 679  * @param cls Closure (unused).
 680  * @param client Identification of the client.
 681  * @param message The actual message.
 682  */
 683 void
 684 handle_show_tunnel (void *cls, struct GNUNET_SERVER_Client *client,
 685                     const struct GNUNET_MessageHeader *message)
 686 {
 687   const struct GNUNET_MESH_LocalInfo *msg;
 688   struct GNUNET_MESH_LocalInfoTunnel *resp;
 689   struct MeshClient *c;
 690   struct MeshTunnel3 *t;
 691   unsigned int ch_n;
 692   unsigned int c_n;
 693   size_t size;
 694
 695   /* Sanity check for client registration */
 696   if (NULL == (c = GML_client_get (client)))
 697   {
 698     GNUNET_break (0);
 699     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 700     return;
 701   }
 702
 703   msg = (struct GNUNET_MESH_LocalInfo *) message;
 704   LOG (GNUNET_ERROR_TYPE_INFO,
 705        "Received tunnel info request from client %u for tunnel %s\n",
 706        c->id, GNUNET_i2s_full(&msg->peer));
 707
 708   t = GMP_get_tunnel (GMP_get (&msg->peer));
 709   if (NULL == t)
 710   {
 711     /* We don't know the tunnel */
 712     struct GNUNET_MESH_LocalInfoTunnel warn;
 713
 714     LOG (GNUNET_ERROR_TYPE_INFO, "Tunnel %s unknown %u\n",
 715          GNUNET_i2s_full(&msg->peer), sizeof (warn));
 716     warn.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNEL);
 717     warn.header.size = htons (sizeof (warn));
 718     warn.destination = msg->peer;
 719     warn.channels = htonl (0);
 720     warn.connections = htonl (0);
 721     warn.cstate = htons (0);
 722     warn.estate = htons (0);
 723
 724     GNUNET_SERVER_notification_context_unicast (nc, client,
 725                                                 &warn.header,
 726                                                 GNUNET_NO);
 727     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 728     return;
 729   }
 730
 731   /* Initialize context */
 732   ch_n = GMT_count_channels (t);
 733   c_n = GMT_count_connections (t);
 734
 735   size = sizeof (struct GNUNET_MESH_LocalInfo);
 736   size += c_n * sizeof (struct GNUNET_HashCode);
 737   size += ch_n * sizeof (MESH_ChannelNumber);
 738
 739   resp = GNUNET_malloc (size);
 740   resp->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNEL);
 741   resp->header.size = htons (size);
 742   GMT_iterate_connections (t, &iter_connection, resp);
 743   GMT_iterate_channels (t, &iter_channel, resp);
 744   /* Do not interleave with iterators, iter_channel needs conn in HBO */
 745   resp->connections = htonl (resp->connections);
 746   resp->channels = htonl (resp->channels);
 747   resp->cstate = htons (GMT_get_cstate (t));
 748   resp->estate = htons (GMT_get_estate (t));
 749   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
 750                                               &resp->header, GNUNET_NO);
 751   GNUNET_free (resp);
 752
 753   LOG (GNUNET_ERROR_TYPE_INFO,
 754        "Show tunnel request from client %u completed. %u conn, %u ch\n",
 755        c->id, c_n, ch_n);
 756   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 757 }
 758
 759
 760 /**
 761  * Functions to handle messages from clients
 762  */
 763 static struct GNUNET_SERVER_MessageHandler client_handlers[] = {
 764   {&handle_new_client, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT, 0},
 765   {&handle_channel_create, NULL, GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE,
 766    sizeof (struct GNUNET_MESH_ChannelMessage)},
 767   {&handle_channel_destroy, NULL, GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY,
 768    sizeof (struct GNUNET_MESH_ChannelMessage)},
 769   {&handle_data, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA, 0},
 770   {&handle_ack, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK,
 771    sizeof (struct GNUNET_MESH_LocalAck)},
 772   {&handle_get_tunnels, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS,
 773    sizeof (struct GNUNET_MessageHeader)},
 774   {&handle_show_tunnel, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNEL,
 775    sizeof (struct GNUNET_MESH_LocalInfo)},
 776   {NULL, NULL, 0, 0}
 777 };
 778
 779
 780
 781 /******************************************************************************/
 782 /********************************    API    ***********************************/
 783 /******************************************************************************/
 784
 785 /**
 786  * Initialize server subsystem.
 787  *
 788  * @param handle Server handle.
 789  */
 790 void
 791 GML_init (struct GNUNET_SERVER_Handle *handle)
 792 {
 793   LOG (GNUNET_ERROR_TYPE_DEBUG, "init\n");
 794   server_handle = handle;
 795   GNUNET_SERVER_suspend (server_handle);
 796   ports = GNUNET_CONTAINER_multihashmap32_create (32);
 797 }
 798
 799
 800 /**
 801  * Install server (service) handlers and start listening to clients.
 802  */
 803 void
 804 GML_start (void)
 805 {
 806   GNUNET_SERVER_add_handlers (server_handle, client_handlers);
 807   GNUNET_SERVER_connect_notify (server_handle,  &handle_client_connect, NULL);
 808   GNUNET_SERVER_disconnect_notify (server_handle, &handle_client_disconnect,
 809                                    NULL);
 810   nc = GNUNET_SERVER_notification_context_create (server_handle, 1);
 811
 812   clients_head = NULL;
 813   clients_tail = NULL;
 814   next_client_id = 0;
 815   GNUNET_SERVER_resume (server_handle);
 816 }
 817
 818
 819 /**
 820  * Shutdown server.
 821  */
 822 void
 823 GML_shutdown (void)
 824 {
 825   if (nc != NULL)
 826   {
 827     GNUNET_SERVER_notification_context_destroy (nc);
 828     nc = NULL;
 829   }
 830 }
 831
 832
 833 /**
 834  * Get a channel from a client.
 835  *
 836  * @param c Client to check.
 837  * @param chid Channel ID, must be local (> 0x800...).
 838  *
 839  * @return non-NULL if channel exists in the clients lists
 840  */
 841 struct MeshChannel *
 842 GML_channel_get (struct MeshClient *c, MESH_ChannelNumber chid)
 843 {
 844   struct GNUNET_CONTAINER_MultiHashMap32 *map;
 845
 846   if (0 == (chid & GNUNET_MESH_LOCAL_CHANNEL_ID_CLI))
 847   {
 848     GNUNET_break_op (0);
 849     LOG (GNUNET_ERROR_TYPE_DEBUG, "CHID %X not a local chid\n", chid);
 850     return NULL;
 851   }
 852
 853   if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV)
 854     map = c->incoming_channels;
 855   else if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_CLI)
 856     map = c->own_channels;
 857   else
 858   {
 859     GNUNET_break (0);
 860     map = NULL;
 861   }
 862   if (NULL == map)
 863   {
 864     GNUNET_break (0);
 865     LOG (GNUNET_ERROR_TYPE_DEBUG,
 866          "Client %s does no t have a valid map for CHID %X\n",
 867          GML_2s (c), chid);
 868     return NULL;
 869   }
 870   return GNUNET_CONTAINER_multihashmap32_get (map, chid);
 871 }
 872
 873
 874 /**
 875  * Add a channel to a client
 876  *
 877  * @param client Client.
 878  * @param chid Channel ID.
 879  * @param ch Channel.
 880  */
 881 void
 882 GML_channel_add (struct MeshClient *client,
 883                  uint32_t chid,
 884                  struct MeshChannel *ch)
 885 {
 886   if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV)
 887     GNUNET_CONTAINER_multihashmap32_put (client->incoming_channels, chid, ch,
 888                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
 889   else if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_CLI)
 890     GNUNET_CONTAINER_multihashmap32_put (client->own_channels, chid, ch,
 891                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
 892   else
 893     GNUNET_break (0);
 894 }
 895
 896
 897 /**
 898  * Remove a channel from a client.
 899  *
 900  * @param client Client.
 901  * @param chid Channel ID.
 902  * @param ch Channel.
 903  */
 904 void
 905 GML_channel_remove (struct MeshClient *client,
 906                     uint32_t chid,
 907                     struct MeshChannel *ch)
 908 {
 909   if (GNUNET_MESH_LOCAL_CHANNEL_ID_SERV <= chid)
 910     GNUNET_break (GNUNET_YES ==
 911                   GNUNET_CONTAINER_multihashmap32_remove (client->incoming_channels,
 912                                                           chid, ch));
 913   else if (GNUNET_MESH_LOCAL_CHANNEL_ID_CLI <= chid)
 914     GNUNET_break (GNUNET_YES ==
 915                   GNUNET_CONTAINER_multihashmap32_remove (client->own_channels,
 916                                                           chid, ch));
 917   else
 918     GNUNET_break (0);
 919 }
 920
 921
 922 /**
 923  * Get the tunnel's next free local channel ID.
 924  *
 925  * @param c Client.
 926  *
 927  * @return LID of a channel free to use.
 928  */
 929 MESH_ChannelNumber
 930 GML_get_next_chid (struct MeshClient *c)
 931 {
 932   MESH_ChannelNumber chid;
 933
 934   while (NULL != GML_channel_get (c, c->next_chid))
 935   {
 936     LOG (GNUNET_ERROR_TYPE_DEBUG, "Channel %u exists...\n", c->next_chid);
 937     c->next_chid = (c->next_chid + 1) | GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
 938   }
 939   chid = c->next_chid;
 940   c->next_chid = (c->next_chid + 1) | GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
 941
 942   return chid;
 943 }
 944
 945
 946 /**
 947  * Check if client has registered with the service and has not disconnected
 948  *
 949  * @param client the client to check
 950  *
 951  * @return non-NULL if client exists in the global DLL
 952  */
 953 struct MeshClient *
 954 GML_client_get (struct GNUNET_SERVER_Client *client)
 955 {
 956   return GNUNET_SERVER_client_get_user_context (client, struct MeshClient);
 957 }
 958
 959 /**
 960  * Find a client that has opened a port
 961  *
 962  * @param port Port to check.
 963  *
 964  * @return non-NULL if a client has the port.
 965  */
 966 struct MeshClient *
 967 GML_client_get_by_port (uint32_t port)
 968 {
 969   return GNUNET_CONTAINER_multihashmap32_get (ports, port);
 970 }
 971
 972
 973 /**
 974  * Deletes a channel from a client (either owner or destination).
 975  *
 976  * @param c Client whose tunnel to delete.
 977  * @param ch Channel which should be deleted.
 978  * @param id Channel ID.
 979  */
 980 void
 981 GML_client_delete_channel (struct MeshClient *c,
 982                            struct MeshChannel *ch,
 983                            MESH_ChannelNumber id)
 984 {
 985   int res;
 986
 987   if (GNUNET_MESH_LOCAL_CHANNEL_ID_SERV <= id)
 988   {
 989     res = GNUNET_CONTAINER_multihashmap32_remove (c->incoming_channels,
 990                                                   id, ch);
 991     if (GNUNET_YES != res)
 992       LOG (GNUNET_ERROR_TYPE_DEBUG, "client_delete_channel dest KO\n");
 993   }
 994   else if (GNUNET_MESH_LOCAL_CHANNEL_ID_CLI <= id)
 995   {
 996     res = GNUNET_CONTAINER_multihashmap32_remove (c->own_channels,
 997                                                   id, ch);
 998     if (GNUNET_YES != res)
 999       LOG (GNUNET_ERROR_TYPE_DEBUG, "client_delete_tunnel root KO\n");
1000   }
1001   else
1002   {
1003     GNUNET_break (0);
1004   }
1005 }
1006
1007 /**
1008  * Build a local ACK message and send it to a local client, if needed.
1009  *
1010  * If the client was already allowed to send data, do nothing.
1011  *
1012  * @param c Client to whom send the ACK.
1013  * @param id Channel ID to use
1014  */
1015 void
1016 GML_send_ack (struct MeshClient *c, MESH_ChannelNumber id)
1017 {
1018   struct GNUNET_MESH_LocalAck msg;
1019
1020   LOG (GNUNET_ERROR_TYPE_DEBUG,
1021               "send local %s ack on %X towards %p\n",
1022               id < GNUNET_MESH_LOCAL_CHANNEL_ID_SERV ? "FWD" : "BCK", id, c);
1023
1024   msg.header.size = htons (sizeof (msg));
1025   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK);
1026   msg.channel_id = htonl (id);
1027   GNUNET_SERVER_notification_context_unicast (nc,
1028                                               c->handle,
1029                                               &msg.header,
1030                                               GNUNET_NO);
1031
1032 }
1033
1034
1035
1036 /**
1037  * Notify the client that a new incoming channel was created.
1038  *
1039  * @param c Client to notify.
1040  * @param id Channel ID.
1041  * @param port Channel's destination port.
1042  * @param opt Options (bit array).
1043  * @param peer Origin peer.
1044  */
1045 void
1046 GML_send_channel_create (struct MeshClient *c,
1047                          uint32_t id, uint32_t port, uint32_t opt,
1048                          const struct GNUNET_PeerIdentity *peer)
1049 {
1050   struct GNUNET_MESH_ChannelMessage msg;
1051
1052   msg.header.size = htons (sizeof (msg));
1053   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE);
1054   msg.channel_id = htonl (id);
1055   msg.port = htonl (port);
1056   msg.opt = htonl (opt);
1057   msg.peer = *peer;
1058   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
1059                                               &msg.header, GNUNET_NO);
1060 }
1061
1062
1063 /**
1064  * Build a local channel NACK message and send it to a local client.
1065  *
1066  * @param c Client to whom send the NACK.
1067  * @param id Channel ID to use
1068  */
1069 void
1070 GML_send_channel_nack (struct MeshClient *c, MESH_ChannelNumber id)
1071 {
1072   struct GNUNET_MESH_LocalAck msg;
1073
1074   LOG (GNUNET_ERROR_TYPE_DEBUG,
1075        "send local nack on %X towards %p\n",
1076        id, c);
1077
1078   msg.header.size = htons (sizeof (msg));
1079   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_NACK);
1080   msg.channel_id = htonl (id);
1081   GNUNET_SERVER_notification_context_unicast (nc,
1082                                               c->handle,
1083                                               &msg.header,
1084                                               GNUNET_NO);
1085
1086 }
1087
1088 /**
1089  * Notify a client that a channel is no longer valid.
1090  *
1091  * @param c Client.
1092  * @param id ID of the channel that is destroyed.
1093  */
1094 void
1095 GML_send_channel_destroy (struct MeshClient *c, uint32_t id)
1096 {
1097   struct GNUNET_MESH_ChannelMessage msg;
1098
1099   if (NULL == c)
1100   {
1101     GNUNET_break (0);
1102     return;
1103   }
1104   if (GNUNET_YES == c->shutting_down)
1105     return;
1106   msg.header.size = htons (sizeof (msg));
1107   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY);
1108   msg.channel_id = htonl (id);
1109   msg.port = htonl (0);
1110   memset (&msg.peer, 0, sizeof (msg.peer));
1111   msg.opt = htonl (0);
1112   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
1113                                               &msg.header, GNUNET_NO);
1114 }
1115
1116
1117 /**
1118  * Modify the mesh message ID from global to local and send to client.
1119  *
1120  * @param c Client to send to.
1121  * @param msg Message to modify and send.
1122  * @param id Channel ID to use (c can be both owner and client).
1123  */
1124 void
1125 GML_send_data (struct MeshClient *c,
1126                const struct GNUNET_MESH_Data *msg,
1127                MESH_ChannelNumber id)
1128 {
1129   struct GNUNET_MESH_LocalData *copy;
1130   uint16_t size = ntohs (msg->header.size) - sizeof (struct GNUNET_MESH_Data);
1131   char cbuf[size + sizeof (struct GNUNET_MESH_LocalData)];
1132
1133   if (size < sizeof (struct GNUNET_MessageHeader))
1134   {
1135     GNUNET_break_op (0);
1136     return;
1137   }
1138   if (NULL == c)
1139   {
1140     GNUNET_break (0);
1141     return;
1142   }
1143   copy = (struct GNUNET_MESH_LocalData *) cbuf;
1144   memcpy (&copy[1], &msg[1], size);
1145   copy->header.size = htons (sizeof (struct GNUNET_MESH_LocalData) + size);
1146   copy->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA);
1147   copy->id = htonl (id);
1148   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
1149                                               &copy->header, GNUNET_NO);
1150 }
1151
1152
1153 /**
1154  * Get the static string to represent a client.
1155  *
1156  * @param c Client.
1157  *
1158  * @return Static string for the client.
1159  */
1160 const char *
1161 GML_2s (const struct MeshClient *c)
1162 {
1163   static char buf[32];
1164
1165   sprintf (buf, "%u", c->id);
1166   return buf;
1167 }