src/mesh/gnunet-mesh.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2012 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  * @file mesh/gnunet-mesh.c
  23  * @brief Print information about mesh tunnels and peers.
  24  * @author Bartlomiej Polot
  25  */
  26 #include "platform.h"
  27 #include "gnunet_util_lib.h"
  28 #include "gnunet_mesh_service.h"
  29
  30
  31 /**
  32  * Option -m.
  33  */
  34 static int monitor_connections;
  35
  36 /**
  37  * Option -i.
  38  */
  39 static int get_info;
  40
  41 /**
  42  * Option --tunnel
  43  */
  44 static char *tunnel_id;
  45
  46 /**
  47  * Option --connection
  48  */
  49 static char *conn_id;
  50
  51 /**
  52  * Option --channel
  53  */
  54 static char *channel_id;
  55
  56 /**
  57  * Port to listen on (-p).
  58  */
  59 static uint32_t listen_port;
  60
  61 /**
  62  * Peer to connect to.
  63  */
  64 static char *target_id;
  65
  66 /**
  67  * Port to connect to
  68  */
  69 static uint32_t target_port;
  70
  71 /**
  72  * Data pending in netcat mode.
  73  */
  74 size_t data_size;
  75
  76
  77 /**
  78  * Mesh handle.
  79  */
  80 static struct GNUNET_MESH_Handle *mh;
  81
  82 /**
  83  * Channel handle.
  84  */
  85 static struct GNUNET_MESH_Channel *ch;
  86
  87 /**
  88  * Shutdown task handle.
  89  */
  90 GNUNET_SCHEDULER_TaskIdentifier sd;
  91
  92
  93
  94 static void
  95 listen_stdio (void);
  96
  97
  98
  99 /**
 100  * Task run in monitor mode when the user presses CTRL-C to abort.
 101  * Stops monitoring activity.
 102  *
 103  * @param cls Closure (unused).
 104  * @param tc scheduler context
 105  */
 106 static void
 107 shutdown_task (void *cls,
 108                const struct GNUNET_SCHEDULER_TaskContext *tc)
 109 {
 110   if (NULL != ch)
 111   {
 112     GNUNET_MESH_channel_destroy (ch);
 113     ch = NULL;
 114   }
 115   if (NULL != mh)
 116   {
 117     GNUNET_MESH_disconnect (mh);
 118         mh = NULL;
 119   }
 120 }
 121
 122
 123 /**
 124  * Function called to notify a client about the connection
 125  * begin ready to queue more data.  "buf" will be
 126  * NULL and "size" zero if the connection was closed for
 127  * writing in the meantime.
 128  *
 129  * FIXME
 130  *
 131  * @param cls closure
 132  * @param size number of bytes available in buf
 133  * @param buf where the callee should write the message
 134  * @return number of bytes written to buf
 135  */
 136 size_t
 137 data_ready (void *cls, size_t size, void *buf)
 138 {
 139   struct GNUNET_MessageHeader *msg;
 140   size_t total_size;
 141
 142   if (NULL == buf || 0 == size)
 143   {
 144     GNUNET_SCHEDULER_shutdown();
 145     return 0;
 146   }
 147
 148   total_size = data_size + sizeof (struct GNUNET_MessageHeader);
 149   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "sending %u bytes\n", data_size);
 150   GNUNET_assert (size >= total_size);
 151
 152   msg = buf;
 153   msg->size = ntohs (total_size);
 154   msg->type = ntohs (GNUNET_MESSAGE_TYPE_MESH_CLI);
 155   memcpy (&msg[1], cls, data_size);
 156   listen_stdio ();
 157
 158   return total_size;
 159 }
 160
 161
 162 /**
 163  * Task run in monitor mode when the user presses CTRL-C to abort.
 164  * Stops monitoring activity.
 165  *
 166  * @param cls Closure (unused).
 167  * @param tc scheduler context
 168  */
 169 static void
 170 read_stdio (void *cls,
 171             const struct GNUNET_SCHEDULER_TaskContext *tc)
 172 {
 173   char buf[60000];
 174
 175   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 176   {
 177     return;
 178   }
 179
 180   data_size = read (0, buf, 60000);
 181   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "stdio read %u bytes\n", data_size);
 182   if (data_size < 1)
 183   {
 184     GNUNET_SCHEDULER_shutdown();
 185     return;
 186   }
 187   GNUNET_MESH_notify_transmit_ready (ch, GNUNET_NO,
 188                                      GNUNET_TIME_UNIT_FOREVER_REL,
 189                                      data_size
 190                                      + sizeof (struct GNUNET_MessageHeader),
 191                                      &data_ready, buf);
 192 }
 193
 194
 195 /**
 196  * Start listening to stdin
 197  */
 198 static void
 199 listen_stdio (void)
 200 {
 201   struct GNUNET_NETWORK_FDSet *rs;
 202
 203   rs = GNUNET_NETWORK_fdset_create ();
 204   GNUNET_NETWORK_fdset_set_native (rs, 0);
 205   GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
 206                                GNUNET_TIME_UNIT_FOREVER_REL,
 207                                rs, NULL,
 208                                &read_stdio, NULL);
 209   GNUNET_NETWORK_fdset_destroy (rs);
 210 }
 211
 212
 213 /**
 214  * Function called whenever a channel is destroyed.  Should clean up
 215  * any associated state.
 216  *
 217  * It must NOT call #GNUNET_MESH_channel_destroy on the channel.
 218  *
 219  * @param cls closure (set from #GNUNET_MESH_connect)
 220  * @param channel connection to the other end (henceforth invalid)
 221  * @param channel_ctx place where local state associated
 222  *                   with the channel is stored
 223  */
 224 static void
 225 channel_ended (void *cls,
 226                const struct GNUNET_MESH_Channel *channel,
 227                void *channel_ctx)
 228 {
 229   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Channel ended!\n");
 230   GNUNET_break (channel == ch);
 231   ch = NULL;
 232   GNUNET_SCHEDULER_shutdown ();
 233 }
 234
 235
 236 /**
 237  * Method called whenever another peer has added us to a channel
 238  * the other peer initiated.
 239  * Only called (once) upon reception of data with a message type which was
 240  * subscribed to in #GNUNET_MESH_connect.
 241  *
 242  * A call to #GNUNET_MESH_channel_destroy causes te channel to be ignored. In
 243  * this case the handler MUST return NULL.
 244  *
 245  * @param cls closure
 246  * @param channel new handle to the channel
 247  * @param initiator peer that started the channel
 248  * @param port Port this channel is for.
 249  * @param options MeshOption flag field, with all active option bits set to 1.
 250  *
 251  * @return initial channel context for the channel
 252  *         (can be NULL -- that's not an error)
 253  */
 254 static void *
 255 channel_incoming (void *cls,
 256                   struct GNUNET_MESH_Channel * channel,
 257                   const struct GNUNET_PeerIdentity * initiator,
 258                   uint32_t port, enum GNUNET_MESH_ChannelOption options)
 259 {
 260   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 261               "Incoming channel %p on port %u\n",
 262               channel, port);
 263   if (NULL != ch)
 264   {
 265     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "A channel already exists\n");
 266     return NULL;
 267   }
 268   if (0 == listen_port)
 269   {
 270     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Not listening to channels\n");
 271     return NULL;
 272   }
 273   ch = channel;
 274   listen_stdio ();
 275   return NULL;
 276 }
 277
 278
 279
 280 /**
 281  * Call MESH's monitor API, get info of one connection.
 282  *
 283  * @param cls Closure (unused).
 284  * @param tc TaskContext
 285  */
 286 static void
 287 create_channel (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 288 {
 289   struct GNUNET_PeerIdentity pid;
 290
 291   GNUNET_assert (NULL == ch);
 292
 293   if (GNUNET_OK !=
 294       GNUNET_CRYPTO_eddsa_public_key_from_string (target_id,
 295                                                   strlen (target_id),
 296                                                   &pid.public_key))
 297   {
 298     FPRINTF (stderr,
 299              _("Invalid target `%s'\n"),
 300              target_id);
 301     GNUNET_SCHEDULER_shutdown ();
 302     return;
 303   }
 304   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connecting to `%s'\n", target_id);
 305   ch = GNUNET_MESH_channel_create (mh, NULL, &pid, target_port,
 306                                    GNUNET_MESH_OPTION_DEFAULT);
 307   listen_stdio ();
 308 }
 309
 310
 311 /**
 312  * Function called whenever a message is received.
 313  *
 314  * Each time the function must call #GNUNET_MESH_receive_done on the channel
 315  * in order to receive the next message. This doesn't need to be immediate:
 316  * can be delayed if some processing is done on the message.
 317  *
 318  * @param cls Closure (set from #GNUNET_MESH_connect).
 319  * @param channel Connection to the other end.
 320  * @param channel_ctx Place to store local state associated with the channel.
 321  * @param message The actual message.
 322  * @return #GNUNET_OK to keep the channel open,
 323  *         #GNUNET_SYSERR to close it (signal serious error).
 324  */
 325 static int
 326 data_callback (void *cls,
 327                struct GNUNET_MESH_Channel *channel,
 328                void **channel_ctx,
 329                const struct GNUNET_MessageHeader *message)
 330 {
 331   int16_t len;
 332   GNUNET_break (ch == channel);
 333
 334   len = ntohs (message->size) - sizeof (*message);
 335   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got %u bytes\n", len);
 336   FPRINTF (stdout, "%.*s", len, (char *) &message[1]);
 337   return GNUNET_OK;
 338 }
 339
 340
 341 /**
 342  * Method called to retrieve information about each tunnel the mesh peer
 343  * is aware of.
 344  *
 345  * @param cls Closure.
 346  * @param tunnel_number Tunnel number.
 347  * @param origin that started the tunnel (owner).
 348  * @param target other endpoint of the tunnel
 349  */
 350 void /* FIXME static */
 351 tunnels_callback (void *cls,
 352                   uint32_t tunnel_number,
 353                   const struct GNUNET_PeerIdentity *origin,
 354                   const struct GNUNET_PeerIdentity *target)
 355 {
 356   FPRINTF (stdout, "Tunnel %s [%u]\n",
 357            GNUNET_i2s_full (origin), tunnel_number);
 358   FPRINTF (stdout, "\n");
 359 }
 360
 361
 362 /**
 363  * Method called to retrieve information about each tunnel the mesh peer
 364  * is aware of.
 365  *
 366  * @param cls Closure.
 367  * @param peer Peer in the tunnel's tree.
 368  * @param parent Parent of the current peer. All 0 when peer is root.
 369  *
 370  */
 371 void /* FIXME static */
 372 tunnel_callback (void *cls,
 373                  const struct GNUNET_PeerIdentity *peer,
 374                  const struct GNUNET_PeerIdentity *parent)
 375 {
 376 }
 377
 378
 379 /**
 380  * Call MESH's monitor API, get all tunnels known to peer.
 381  *
 382  * @param cls Closure (unused).
 383  * @param tc TaskContext
 384  */
 385 static void
 386 get_tunnels (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 387 {
 388   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 389   {
 390     return;
 391   }
 392 //   GNUNET_MESH_get_tunnels (mh, &tunnels_callback, NULL);
 393   if (GNUNET_YES != monitor_connections)
 394   {
 395     GNUNET_SCHEDULER_shutdown();
 396   }
 397 }
 398
 399
 400 /**
 401  * Call MESH's monitor API, get info of one tunnel.
 402  *
 403  * @param cls Closure (unused).
 404  * @param tc TaskContext
 405  */
 406 static void
 407 show_tunnel (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 408 {
 409   struct GNUNET_PeerIdentity pid;
 410
 411   if (GNUNET_OK !=
 412       GNUNET_CRYPTO_eddsa_public_key_from_string (tunnel_id,
 413                                                   strlen (tunnel_id),
 414                                                   &pid.public_key))
 415   {
 416     fprintf (stderr,
 417              _("Invalid tunnel owner `%s'\n"),
 418              tunnel_id);
 419     GNUNET_SCHEDULER_shutdown();
 420     return;
 421   }
 422 //   GNUNET_MESH_show_tunnel (mh, &pid, 0, tunnel_callback, NULL);
 423 }
 424
 425
 426 /**
 427  * Call MESH's monitor API, get info of one channel.
 428  *
 429  * @param cls Closure (unused).
 430  * @param tc TaskContext
 431  */
 432 static void
 433 show_channel (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 434 {
 435
 436 }
 437
 438
 439 /**
 440  * Call MESH's monitor API, get info of one connection.
 441  *
 442  * @param cls Closure (unused).
 443  * @param tc TaskContext
 444  */
 445 static void
 446 show_connection (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 447 {
 448
 449 }
 450
 451
 452 /**
 453  * Main function that will be run by the scheduler.
 454  *
 455  * @param cls closure
 456  * @param args remaining command-line arguments
 457  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
 458  * @param cfg configuration
 459  */
 460 static void
 461 run (void *cls, char *const *args, const char *cfgfile,
 462      const struct GNUNET_CONFIGURATION_Handle *cfg)
 463 {
 464   GNUNET_MESH_InboundChannelNotificationHandler *newch = NULL;
 465   GNUNET_MESH_ChannelEndHandler *endch = NULL;
 466   static const struct GNUNET_MESH_MessageHandler handlers[] = {
 467     {&data_callback, GNUNET_MESSAGE_TYPE_MESH_CLI, 0},
 468     {NULL, 0, 0} /* FIXME add option to monitor msg types */
 469   };
 470   static uint32_t *ports = NULL;
 471   /* FIXME add option to monitor apps */
 472
 473   target_id = args[0];
 474   target_port = args[0] && args[1] ? atoi(args[1]) : 0;
 475   if ( (0 != get_info
 476         || 0 != monitor_connections
 477         || NULL != tunnel_id
 478         || NULL != conn_id
 479         || NULL != channel_id)
 480        && target_id != NULL)
 481   {
 482     FPRINTF (stderr, _("You must NOT give a TARGET when using options\n"));
 483     return;
 484   }
 485
 486   if (NULL != target_id)
 487   {
 488     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 489                 "Creating channel to %s\n",
 490                 target_id);
 491     GNUNET_SCHEDULER_add_now (&create_channel, NULL);
 492     endch = &channel_ended;
 493   }
 494   else if (0 != listen_port)
 495   {
 496     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Listen\n");
 497     newch = &channel_incoming;
 498     endch = &channel_ended;
 499     ports = GNUNET_malloc (sizeof (uint32_t) * 2);
 500     ports[0] = listen_port;
 501   }
 502   else if (NULL != tunnel_id)
 503   {
 504     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Show tunnel\n");
 505     GNUNET_SCHEDULER_add_now (&show_tunnel, NULL);
 506   }
 507   else if (NULL != channel_id)
 508   {
 509     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Show channel\n");
 510     GNUNET_SCHEDULER_add_now (&show_channel, NULL);
 511   }
 512   else if (NULL != conn_id)
 513   {
 514     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Show connection\n");
 515     GNUNET_SCHEDULER_add_now (&show_connection, NULL);
 516   }
 517   else if (GNUNET_YES == get_info)
 518   {
 519     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Show all tunnels\n");
 520     GNUNET_SCHEDULER_add_now (&get_tunnels, NULL);
 521   }
 522   else
 523   {
 524     FPRINTF (stderr, "No action requested\n");
 525     return;
 526   }
 527
 528   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connecting to mesh\n");
 529   mh = GNUNET_MESH_connect (cfg,
 530                             NULL, /* cls */
 531                             newch, /* new channel */
 532                             endch, /* cleaner */
 533                             handlers,
 534                             ports);
 535   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Done\n");
 536   if (NULL == mh)
 537     GNUNET_SCHEDULER_add_now (shutdown_task, NULL);
 538   else
 539     sd = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
 540                                        shutdown_task, NULL);
 541
 542 }
 543
 544
 545 /**
 546  * The main function to obtain peer information.
 547  *
 548  * @param argc number of arguments from the command line
 549  * @param argv command line arguments
 550  * @return 0 ok, 1 on error
 551  */
 552 int
 553 main (int argc, char *const *argv)
 554 {
 555   int res;
 556   const char helpstr[] = "Create channels and retreive info about meshs status.";
 557   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
 558     {'a', "channel", "TUNNEL_ID:CHANNEL_ID",
 559      gettext_noop ("provide information about a particular channel"),
 560      GNUNET_YES, &GNUNET_GETOPT_set_string, &channel_id},
 561     {'b', "connection", "TUNNEL_ID:CONNECTION_ID",
 562      gettext_noop ("provide information about a particular connection"),
 563      GNUNET_YES, &GNUNET_GETOPT_set_string, &conn_id},
 564     {'i', "info", NULL,
 565      gettext_noop ("provide information about all tunnels"),
 566      GNUNET_NO, &GNUNET_GETOPT_set_one, &get_info},
 567     {'m', "monitor", NULL,
 568      gettext_noop ("provide information about all tunnels (continuously) NOT IMPLEMENTED"), /* FIXME */
 569      GNUNET_NO, &GNUNET_GETOPT_set_one, &monitor_connections},
 570     {'p', "port", NULL,
 571      gettext_noop ("port to listen to (default; 0)"),
 572      GNUNET_YES, &GNUNET_GETOPT_set_uint, &listen_port},
 573     {'t', "tunnel", "TUNNEL_ID",
 574      gettext_noop ("provide information about a particular tunnel"),
 575      GNUNET_YES, &GNUNET_GETOPT_set_string, &tunnel_id},
 576     GNUNET_GETOPT_OPTION_END
 577   };
 578
 579   if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
 580     return 2;
 581
 582   res = GNUNET_PROGRAM_run (argc, argv, "gnunet-mesh (OPTIONS | TARGET PORT)",
 583                             gettext_noop (helpstr),
 584                             options, &run, NULL);
 585
 586   GNUNET_free ((void *) argv);
 587
 588   if (GNUNET_OK == res)
 589     return 0;
 590   else
 591     return 1;
 592 }
 593
 594 /* end of gnunet-mesh.c */