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     warn.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNEL);
 715     warn.header.size = htons (sizeof (warn));
 716     warn.destination = msg->peer;
 717     warn.channels = htonl (0);
 718     warn.connections = htonl (0);
 719     warn.cstate = htons (0);
 720     warn.estate = htons (0);
 721
 722     GNUNET_SERVER_notification_context_unicast (nc, client,
 723                                                 &warn.header,
 724                                                 GNUNET_NO);
 725     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 726     return;
 727   }
 728
 729   /* Initialize context */
 730   ch_n = GMT_count_channels (t);
 731   c_n = GMT_count_connections (t);
 732
 733   size = sizeof (struct GNUNET_MESH_LocalInfo);
 734   size += c_n * sizeof (struct GNUNET_HashCode);
 735   size += ch_n * sizeof (uint32_t);
 736
 737   resp = GNUNET_malloc (size);
 738   resp->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNEL);
 739   resp->header.size = htons (size);
 740   GMT_iterate_connections (t, &iter_connection, resp);
 741   GMT_iterate_channels (t, &iter_channel, resp);
 742   /* Do not interleave with iterators, iter_channel needs conn in HBO */
 743   resp->connections = htonl (resp->connections);
 744   resp->channels = htonl (resp->channels);
 745   resp->cstate = htons (GMT_get_cstate (t));
 746   resp->estate = htons (GMT_get_estate (t));
 747   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
 748                                               &resp->header, GNUNET_NO);
 749   GNUNET_free (resp);
 750
 751   LOG (GNUNET_ERROR_TYPE_INFO,
 752        "Show tunnel request from client %u completed\n",
 753        c->id);
 754   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 755 }
 756
 757
 758 /**
 759  * Functions to handle messages from clients
 760  */
 761 static struct GNUNET_SERVER_MessageHandler client_handlers[] = {
 762   {&handle_new_client, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT, 0},
 763   {&handle_channel_create, NULL, GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE,
 764    sizeof (struct GNUNET_MESH_ChannelMessage)},
 765   {&handle_channel_destroy, NULL, GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY,
 766    sizeof (struct GNUNET_MESH_ChannelMessage)},
 767   {&handle_data, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA, 0},
 768   {&handle_ack, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK,
 769    sizeof (struct GNUNET_MESH_LocalAck)},
 770   {&handle_get_tunnels, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS,
 771    sizeof (struct GNUNET_MessageHeader)},
 772   {&handle_show_tunnel, NULL, GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNEL,
 773    sizeof (struct GNUNET_MESH_LocalInfo)},
 774   {NULL, NULL, 0, 0}
 775 };
 776
 777
 778
 779 /******************************************************************************/
 780 /********************************    API    ***********************************/
 781 /******************************************************************************/
 782
 783 /**
 784  * Initialize server subsystem.
 785  *
 786  * @param handle Server handle.
 787  */
 788 void
 789 GML_init (struct GNUNET_SERVER_Handle *handle)
 790 {
 791   LOG (GNUNET_ERROR_TYPE_DEBUG, "init\n");
 792   server_handle = handle;
 793   GNUNET_SERVER_suspend (server_handle);
 794   ports = GNUNET_CONTAINER_multihashmap32_create (32);
 795 }
 796
 797
 798 /**
 799  * Install server (service) handlers and start listening to clients.
 800  */
 801 void
 802 GML_start (void)
 803 {
 804   GNUNET_SERVER_add_handlers (server_handle, client_handlers);
 805   GNUNET_SERVER_connect_notify (server_handle,  &handle_client_connect, NULL);
 806   GNUNET_SERVER_disconnect_notify (server_handle, &handle_client_disconnect,
 807                                    NULL);
 808   nc = GNUNET_SERVER_notification_context_create (server_handle, 1);
 809
 810   clients_head = NULL;
 811   clients_tail = NULL;
 812   next_client_id = 0;
 813   GNUNET_SERVER_resume (server_handle);
 814 }
 815
 816
 817 /**
 818  * Shutdown server.
 819  */
 820 void
 821 GML_shutdown (void)
 822 {
 823   if (nc != NULL)
 824   {
 825     GNUNET_SERVER_notification_context_destroy (nc);
 826     nc = NULL;
 827   }
 828 }
 829
 830
 831 /**
 832  * Get a channel from a client.
 833  *
 834  * @param c Client to check.
 835  * @param chid Channel ID, must be local (> 0x800...).
 836  *
 837  * @return non-NULL if channel exists in the clients lists
 838  */
 839 struct MeshChannel *
 840 GML_channel_get (struct MeshClient *c, MESH_ChannelNumber chid)
 841 {
 842   struct GNUNET_CONTAINER_MultiHashMap32 *map;
 843
 844   if (0 == (chid & GNUNET_MESH_LOCAL_CHANNEL_ID_CLI))
 845   {
 846     GNUNET_break_op (0);
 847     LOG (GNUNET_ERROR_TYPE_DEBUG, "CHID %X not a local chid\n", chid);
 848     return NULL;
 849   }
 850
 851   if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV)
 852     map = c->incoming_channels;
 853   else if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_CLI)
 854     map = c->own_channels;
 855   else
 856   {
 857     GNUNET_break (0);
 858     map = NULL;
 859   }
 860   if (NULL == map)
 861   {
 862     GNUNET_break (0);
 863     LOG (GNUNET_ERROR_TYPE_DEBUG,
 864          "Client %s does no t have a valid map for CHID %X\n",
 865          GML_2s (c), chid);
 866     return NULL;
 867   }
 868   return GNUNET_CONTAINER_multihashmap32_get (map, chid);
 869 }
 870
 871
 872 /**
 873  * Add a channel to a client
 874  *
 875  * @param client Client.
 876  * @param chid Channel ID.
 877  * @param ch Channel.
 878  */
 879 void
 880 GML_channel_add (struct MeshClient *client,
 881                  uint32_t chid,
 882                  struct MeshChannel *ch)
 883 {
 884   if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV)
 885     GNUNET_CONTAINER_multihashmap32_put (client->incoming_channels, chid, ch,
 886                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
 887   else if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_CLI)
 888     GNUNET_CONTAINER_multihashmap32_put (client->own_channels, chid, ch,
 889                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
 890   else
 891     GNUNET_break (0);
 892 }
 893
 894
 895 /**
 896  * Remove a channel from a client.
 897  *
 898  * @param client Client.
 899  * @param chid Channel ID.
 900  * @param ch Channel.
 901  */
 902 void
 903 GML_channel_remove (struct MeshClient *client,
 904                     uint32_t chid,
 905                     struct MeshChannel *ch)
 906 {
 907   if (GNUNET_MESH_LOCAL_CHANNEL_ID_SERV <= chid)
 908     GNUNET_break (GNUNET_YES ==
 909                   GNUNET_CONTAINER_multihashmap32_remove (client->incoming_channels,
 910                                                           chid, ch));
 911   else if (GNUNET_MESH_LOCAL_CHANNEL_ID_CLI <= chid)
 912     GNUNET_break (GNUNET_YES ==
 913                   GNUNET_CONTAINER_multihashmap32_remove (client->own_channels,
 914                                                           chid, ch));
 915   else
 916     GNUNET_break (0);
 917 }
 918
 919
 920 /**
 921  * Get the tunnel's next free local channel ID.
 922  *
 923  * @param c Client.
 924  *
 925  * @return LID of a channel free to use.
 926  */
 927 MESH_ChannelNumber
 928 GML_get_next_chid (struct MeshClient *c)
 929 {
 930   MESH_ChannelNumber chid;
 931
 932   while (NULL != GML_channel_get (c, c->next_chid))
 933   {
 934     LOG (GNUNET_ERROR_TYPE_DEBUG, "Channel %u exists...\n", c->next_chid);
 935     c->next_chid = (c->next_chid + 1) | GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
 936   }
 937   chid = c->next_chid;
 938   c->next_chid = (c->next_chid + 1) | GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
 939
 940   return chid;
 941 }
 942
 943
 944 /**
 945  * Check if client has registered with the service and has not disconnected
 946  *
 947  * @param client the client to check
 948  *
 949  * @return non-NULL if client exists in the global DLL
 950  */
 951 struct MeshClient *
 952 GML_client_get (struct GNUNET_SERVER_Client *client)
 953 {
 954   return GNUNET_SERVER_client_get_user_context (client, struct MeshClient);
 955 }
 956
 957 /**
 958  * Find a client that has opened a port
 959  *
 960  * @param port Port to check.
 961  *
 962  * @return non-NULL if a client has the port.
 963  */
 964 struct MeshClient *
 965 GML_client_get_by_port (uint32_t port)
 966 {
 967   return GNUNET_CONTAINER_multihashmap32_get (ports, port);
 968 }
 969
 970
 971 /**
 972  * Deletes a channel from a client (either owner or destination).
 973  *
 974  * @param c Client whose tunnel to delete.
 975  * @param ch Channel which should be deleted.
 976  * @param id Channel ID.
 977  */
 978 void
 979 GML_client_delete_channel (struct MeshClient *c,
 980                            struct MeshChannel *ch,
 981                            MESH_ChannelNumber id)
 982 {
 983   int res;
 984
 985   if (GNUNET_MESH_LOCAL_CHANNEL_ID_SERV <= id)
 986   {
 987     res = GNUNET_CONTAINER_multihashmap32_remove (c->incoming_channels,
 988                                                   id, ch);
 989     if (GNUNET_YES != res)
 990       LOG (GNUNET_ERROR_TYPE_DEBUG, "client_delete_channel dest KO\n");
 991   }
 992   else if (GNUNET_MESH_LOCAL_CHANNEL_ID_CLI <= id)
 993   {
 994     res = GNUNET_CONTAINER_multihashmap32_remove (c->own_channels,
 995                                                   id, ch);
 996     if (GNUNET_YES != res)
 997       LOG (GNUNET_ERROR_TYPE_DEBUG, "client_delete_tunnel root KO\n");
 998   }
 999   else
1000   {
1001     GNUNET_break (0);
1002   }
1003 }
1004
1005 /**
1006  * Build a local ACK message and send it to a local client, if needed.
1007  *
1008  * If the client was already allowed to send data, do nothing.
1009  *
1010  * @param c Client to whom send the ACK.
1011  * @param id Channel ID to use
1012  */
1013 void
1014 GML_send_ack (struct MeshClient *c, MESH_ChannelNumber id)
1015 {
1016   struct GNUNET_MESH_LocalAck msg;
1017
1018   LOG (GNUNET_ERROR_TYPE_DEBUG,
1019               "send local %s ack on %X towards %p\n",
1020               id < GNUNET_MESH_LOCAL_CHANNEL_ID_SERV ? "FWD" : "BCK", id, c);
1021
1022   msg.header.size = htons (sizeof (msg));
1023   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK);
1024   msg.channel_id = htonl (id);
1025   GNUNET_SERVER_notification_context_unicast (nc,
1026                                               c->handle,
1027                                               &msg.header,
1028                                               GNUNET_NO);
1029
1030 }
1031
1032
1033
1034 /**
1035  * Notify the client that a new incoming channel was created.
1036  *
1037  * @param c Client to notify.
1038  * @param id Channel ID.
1039  * @param port Channel's destination port.
1040  * @param opt Options (bit array).
1041  * @param peer Origin peer.
1042  */
1043 void
1044 GML_send_channel_create (struct MeshClient *c,
1045                          uint32_t id, uint32_t port, uint32_t opt,
1046                          const struct GNUNET_PeerIdentity *peer)
1047 {
1048   struct GNUNET_MESH_ChannelMessage msg;
1049
1050   msg.header.size = htons (sizeof (msg));
1051   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE);
1052   msg.channel_id = htonl (id);
1053   msg.port = htonl (port);
1054   msg.opt = htonl (opt);
1055   msg.peer = *peer;
1056   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
1057                                               &msg.header, GNUNET_NO);
1058 }
1059
1060
1061 /**
1062  * Build a local channel NACK message and send it to a local client.
1063  *
1064  * @param c Client to whom send the NACK.
1065  * @param id Channel ID to use
1066  */
1067 void
1068 GML_send_channel_nack (struct MeshClient *c, MESH_ChannelNumber id)
1069 {
1070   struct GNUNET_MESH_LocalAck msg;
1071
1072   LOG (GNUNET_ERROR_TYPE_DEBUG,
1073        "send local nack on %X towards %p\n",
1074        id, c);
1075
1076   msg.header.size = htons (sizeof (msg));
1077   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_NACK);
1078   msg.channel_id = htonl (id);
1079   GNUNET_SERVER_notification_context_unicast (nc,
1080                                               c->handle,
1081                                               &msg.header,
1082                                               GNUNET_NO);
1083
1084 }
1085
1086 /**
1087  * Notify a client that a channel is no longer valid.
1088  *
1089  * @param c Client.
1090  * @param id ID of the channel that is destroyed.
1091  */
1092 void
1093 GML_send_channel_destroy (struct MeshClient *c, uint32_t id)
1094 {
1095   struct GNUNET_MESH_ChannelMessage msg;
1096
1097   if (NULL == c)
1098   {
1099     GNUNET_break (0);
1100     return;
1101   }
1102   if (GNUNET_YES == c->shutting_down)
1103     return;
1104   msg.header.size = htons (sizeof (msg));
1105   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY);
1106   msg.channel_id = htonl (id);
1107   msg.port = htonl (0);
1108   memset (&msg.peer, 0, sizeof (msg.peer));
1109   msg.opt = htonl (0);
1110   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
1111                                               &msg.header, GNUNET_NO);
1112 }
1113
1114
1115 /**
1116  * Modify the mesh message ID from global to local and send to client.
1117  *
1118  * @param c Client to send to.
1119  * @param msg Message to modify and send.
1120  * @param id Channel ID to use (c can be both owner and client).
1121  */
1122 void
1123 GML_send_data (struct MeshClient *c,
1124                const struct GNUNET_MESH_Data *msg,
1125                MESH_ChannelNumber id)
1126 {
1127   struct GNUNET_MESH_LocalData *copy;
1128   uint16_t size = ntohs (msg->header.size) - sizeof (struct GNUNET_MESH_Data);
1129   char cbuf[size + sizeof (struct GNUNET_MESH_LocalData)];
1130
1131   if (size < sizeof (struct GNUNET_MessageHeader))
1132   {
1133     GNUNET_break_op (0);
1134     return;
1135   }
1136   if (NULL == c)
1137   {
1138     GNUNET_break (0);
1139     return;
1140   }
1141   copy = (struct GNUNET_MESH_LocalData *) cbuf;
1142   memcpy (&copy[1], &msg[1], size);
1143   copy->header.size = htons (sizeof (struct GNUNET_MESH_LocalData) + size);
1144   copy->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA);
1145   copy->id = htonl (id);
1146   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
1147                                               &copy->header, GNUNET_NO);
1148 }
1149
1150
1151 /**
1152  * Get the static string to represent a client.
1153  *
1154  * @param c Client.
1155  *
1156  * @return Static string for the client.
1157  */
1158 const char *
1159 GML_2s (const struct MeshClient *c)
1160 {
1161   static char buf[32];
1162
1163   sprintf (buf, "%u", c->id);
1164   return buf;
1165 }