trema 0.2.5 → 0.2.6

data/src/management/echo.c

@@ -0,0 +1,185 @@
+/*
+ * Management command to check if a specified application/service is alive or not.
+ *
+ * Author: Yasunobu Chiba
+ *
+ * Copyright (C) 2012 NEC Corporation
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License, version 2, as
+ * published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+
+#include <assert.h>
+#include <errno.h>
+#include <stdio.h>
+#include <string.h>
+#include <time.h>
+#include "trema.h"
+
+
+#define SUB_TIMESPEC( _a, _b, _return )                       \
+  do {                                                        \
+    ( _return )->tv_sec = ( _a )->tv_sec - ( _b )->tv_sec;    \
+    ( _return )->tv_nsec = ( _a )->tv_nsec - ( _b )->tv_nsec; \
+    if ( ( _return )->tv_nsec < 0 ) {                         \
+      ( _return )->tv_sec--;                                  \
+      ( _return )->tv_nsec += 1000000000;                     \
+    }                                                         \
+  }                                                           \
+  while ( 0 )
+
+
+static char service_name[ MESSENGER_SERVICE_NAME_LENGTH ];
+
+
+void
+usage( void ) {
+  printf( "Usage: echo SERVICE_NAME\n" );
+}
+
+
+static void
+print_usage_and_exit( void ) {
+  usage();
+  exit( EXIT_FAILURE );
+}
+
+
+static void
+parse_arguments( int argc, char **argv ) {
+  assert( argv != NULL );
+
+  if ( argc != 2 ) {
+    print_usage_and_exit();
+  }
+
+  if ( argv[ 1 ] == NULL || ( argv[ 1 ] != NULL && strlen( argv[ 1 ] ) == 0 ) ) {
+    print_usage_and_exit();
+  }
+
+  memset( service_name, '\0', MESSENGER_SERVICE_NAME_LENGTH );
+  snprintf( service_name, MESSENGER_SERVICE_NAME_LENGTH, "%s", get_management_service_name( argv[ 1 ] ) );
+}
+
+
+static void
+timeout( void *user_data ) {
+  UNUSED( user_data );
+
+  printf( "Timeout.\n" );
+
+  delete_timer_event( timeout, NULL );
+
+  stop_trema();
+}
+
+
+static void
+handle_reply( uint16_t tag, void *data, size_t length, void *user_data ) {
+  UNUSED( user_data );
+
+  assert( tag == MESSENGER_MANAGEMENT_REPLY );
+  assert( data != NULL );
+  assert( length == sizeof( management_echo_reply ) );
+
+  management_echo_reply *reply = data;
+  assert( ntohs( reply->header.type ) == MANAGEMENT_ECHO_REPLY );
+  assert( ntohl( reply->header.length ) == sizeof( management_echo_reply ) );
+
+  if ( reply->header.status != MANAGEMENT_REQUEST_SUCCEEDED ) {
+    printf( "Failed to execute echo request/reply transaction.\n" );
+    stop_trema();
+    return;
+  }
+
+  struct timespec sent_at = { 0, 0 };
+  sent_at.tv_sec = ( time_t ) ntohl( reply->sent_at.tv_sec );
+  sent_at.tv_nsec = ( long ) ntohl( reply->sent_at.tv_nsec );
+
+  struct timespec received_at = { 0, 0 };
+  received_at.tv_sec = ( time_t ) ntohl( reply->received_at.tv_sec );
+  received_at.tv_nsec = ( long ) ntohl( reply->received_at.tv_nsec );
+
+  struct timespec now = { 0, 0 };
+  clock_gettime( CLOCK_REALTIME, &now );
+
+  struct timespec roundtrip = { 0, 0 };
+  SUB_TIMESPEC( &now, &sent_at, &roundtrip );
+
+  long double roundtrip_msec = roundtrip.tv_sec * 1000 + ( long double ) roundtrip.tv_nsec / 1000000;
+
+  printf( "Reqeust sent at: %d.%09d\n", ( int ) sent_at.tv_sec, ( int ) sent_at.tv_nsec );
+  printf( "Reqeust received at: %d.%09d\n", ( int ) received_at.tv_sec, ( int ) received_at.tv_nsec );
+  printf( "Reply received at: %d.%09d\n", ( int ) now.tv_sec, ( int ) now.tv_nsec );
+  printf( "Roundtrip time: %Lf [msec]\n", roundtrip_msec );
+
+  stop_trema();
+}
+
+
+static void
+send_echo_request() {
+  management_echo_request request;
+  memset( &request, 0, sizeof( management_echo_request ) );
+  request.header.type = MANAGEMENT_ECHO_REQUEST;
+  request.header.length = htonl( sizeof( management_echo_request ) );
+
+  struct timespec now = { 0, 0 };
+  int retval = clock_gettime( CLOCK_REALTIME, &now );
+  if ( retval < 0 ) {
+    printf( "Failed to retrieve clock ( errno = %s [%d] ).\n", strerror( errno ), errno );
+    exit( EXIT_FAILURE );
+  }
+  request.sent_at.tv_sec = htonl( ( uint32_t ) now.tv_sec );
+  request.sent_at.tv_nsec = htonl( ( uint32_t ) now.tv_nsec );
+
+  bool ret = send_request_message( service_name, get_trema_name(), MESSENGER_MANAGEMENT_REQUEST,
+                                   &request, sizeof( management_echo_request ), NULL );
+  if ( !ret ) {
+    printf( "Failed to send an echo request to %s.\n", service_name );
+    exit( EXIT_FAILURE );
+  }
+}
+
+
+int
+main( int argc, char *argv[] ) {
+  // Initialize the Trema world
+  init_trema( &argc, &argv );
+
+  // Parse arguments
+  parse_arguments( argc, argv );
+
+  // Set a handler to handle echo reply
+  add_message_replied_callback( get_trema_name(), handle_reply );
+
+  // Send an echo request
+  send_echo_request();
+
+  // Set timeout
+  add_periodic_event_callback( 5, timeout, NULL );
+
+  // Main loop
+  start_trema();
+
+  return EXIT_SUCCESS;
+}
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */

data/src/management/set_logging_level.c

@@ -0,0 +1,153 @@
+/*
+ * Management command to change logging level in operation.
+ *
+ * Author: Yasunobu Chiba
+ *
+ * Copyright (C) 2012 NEC Corporation
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License, version 2, as
+ * published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+
+#include <assert.h>
+#include <errno.h>
+#include <stdio.h>
+#include <string.h>
+#include "trema.h"
+
+
+static char service_name[ MESSENGER_SERVICE_NAME_LENGTH ];
+static char logging_level[ LOGGING_LEVEL_STR_LENGTH ];
+
+
+void
+usage( void ) {
+  printf( "Usage: set_logging_level SERVICE_NAME LEVEL\n" );
+}
+
+
+static void
+print_usage_and_exit( void ) {
+  usage();
+  exit( EXIT_FAILURE );
+}
+
+
+static void
+parse_arguments( int argc, char **argv ) {
+  assert( argv != NULL );
+
+  if ( argc != 3 ) {
+    print_usage_and_exit();
+  }
+
+  if ( argv[ 1 ] == NULL || ( argv[ 1 ] != NULL && strlen( argv[ 1 ] ) == 0 ) ) {
+    print_usage_and_exit();
+  }
+
+  if ( argv[ 2 ] == NULL ||
+       ( argv[ 2 ] != NULL && ( strlen( argv[ 2 ] ) == 0 || strlen( argv[ 2 ] ) >= LOGGING_LEVEL_STR_LENGTH ) ) ) {
+    print_usage_and_exit();
+  }
+
+  memset( service_name, '\0', MESSENGER_SERVICE_NAME_LENGTH );
+  snprintf( service_name, MESSENGER_SERVICE_NAME_LENGTH, "%s", get_management_service_name( argv[ 1 ] ) );
+
+  memset( logging_level, '\0', LOGGING_LEVEL_STR_LENGTH );
+  snprintf( logging_level, LOGGING_LEVEL_STR_LENGTH, "%s", argv[ 2 ] );
+}
+
+
+static void
+timeout( void *user_data ) {
+  UNUSED( user_data );
+
+  printf( "Timeout.\n" );
+
+  delete_timer_event( timeout, NULL );
+
+  stop_trema();
+}
+
+
+static void
+handle_reply( uint16_t tag, void *data, size_t length, void *user_data ) {
+  UNUSED( user_data );
+
+  assert( tag == MESSENGER_MANAGEMENT_REPLY );
+  assert( data != NULL );
+  assert( length == sizeof( management_set_logging_level_reply ) );
+
+  management_set_logging_level_reply *reply = data;
+  assert( ntohs( reply->header.type ) == MANAGEMENT_SET_LOGGING_LEVEL_REPLY );
+  assert( ntohl( reply->header.length ) == sizeof( management_set_logging_level_reply ) );
+
+  if ( reply->header.status == MANAGEMENT_REQUEST_SUCCEEDED ) {
+    printf( "Logging level is set to %s.\n", logging_level );
+  }
+  else {
+    printf( "Failed to set logging level.\n" );
+  }
+
+  stop_trema();
+}
+
+
+static void
+send_set_logging_level_request() {
+  management_set_logging_level_request request;
+  memset( &request, 0, sizeof( management_set_logging_level_request ) );
+  request.header.type = htons( MANAGEMENT_SET_LOGGING_LEVEL_REQUEST );
+  request.header.length = htonl( sizeof( management_set_logging_level_request ) );
+  strncpy( request.level, logging_level, sizeof( request.level ) );
+
+  bool ret = send_request_message( service_name, get_trema_name(), MESSENGER_MANAGEMENT_REQUEST,
+                                   &request, sizeof( management_set_logging_level_request ), NULL );
+  if ( !ret ) {
+    printf( "Failed to send a set logging level request to %s.\n", service_name );
+    exit( EXIT_FAILURE );
+  }
+}
+
+
+int
+main( int argc, char *argv[] ) {
+  // Initialize the Trema world
+  init_trema( &argc, &argv );
+
+  // Parse arguments
+  parse_arguments( argc, argv );
+
+  // Set a handler to handle set logging level reply
+  add_message_replied_callback( get_trema_name(), handle_reply );
+
+  // Send a set logging level request
+  send_set_logging_level_request();
+
+  // Set timeout
+  add_periodic_event_callback( 5, timeout, NULL );
+
+  // Main loop
+  start_trema();
+
+  return EXIT_SUCCESS;
+}
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */

data/src/management/show_stats.c

@@ -0,0 +1,150 @@
+/*
+ * Management command to show statistics.
+ *
+ * Author: Yasunobu Chiba
+ *
+ * Copyright (C) 2012 NEC Corporation
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License, version 2, as
+ * published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+
+#include <assert.h>
+#include <errno.h>
+#include <stdio.h>
+#include <string.h>
+#include "trema.h"
+
+
+static char service_name[ MESSENGER_SERVICE_NAME_LENGTH ];
+
+
+void
+usage( void ) {
+  printf( "Usage: show_stats SERVICE_NAME\n" );
+}
+
+
+static void
+print_usage_and_exit( void ) {
+  usage();
+  exit( EXIT_FAILURE );
+}
+
+
+static void
+parse_arguments( int argc, char **argv ) {
+  assert( argv != NULL );
+
+  if ( argc != 2 ) {
+    print_usage_and_exit();
+  }
+
+  if ( argv[ 1 ] == NULL || ( argv[ 1 ] != NULL && strlen( argv[ 1 ] ) == 0 ) ) {
+    print_usage_and_exit();
+  }
+
+  memset( service_name, '\0', MESSENGER_SERVICE_NAME_LENGTH );
+  snprintf( service_name, MESSENGER_SERVICE_NAME_LENGTH, "%s", get_management_service_name( argv[ 1 ] ) );
+}
+
+
+static void
+timeout( void *user_data ) {
+  UNUSED( user_data );
+
+  printf( "Timeout.\n" );
+
+  delete_timer_event( timeout, NULL );
+
+  stop_trema();
+}
+
+
+static void
+handle_reply( uint16_t tag, void *data, size_t length, void *user_data ) {
+  UNUSED( user_data );
+
+  assert( tag == MESSENGER_MANAGEMENT_REPLY );
+  assert( data != NULL );
+  assert( length >= offsetof( management_show_stats_reply, entries ) );
+
+  management_show_stats_reply *reply = data;
+  assert( ntohs( reply->header.type ) == MANAGEMENT_SHOW_STATS_REPLY );
+  assert( ntohl( reply->header.length ) == length );
+
+  if ( reply->header.status == MANAGEMENT_REQUEST_FAILED ) {
+    printf( "Failed to show statistics.\n" );
+    stop_trema();
+    return;
+  }
+
+  size_t entries_length = length - offsetof( management_show_stats_reply, entries );
+  unsigned int n_entries = ( unsigned int ) entries_length / sizeof( stat_entry );
+  stat_entry *entry = reply->entries;
+  for ( unsigned int i = 0; i < n_entries; i++ ) {
+    printf( "%s: %" PRIu64 "\n", entry->key, ntohll( entry->value ) );
+    entry++;
+  }
+
+  stop_trema();
+}
+
+
+static void
+send_show_stats_request() {
+  management_show_stats_request request;
+  memset( &request, 0, sizeof( management_show_stats_request ) );
+  request.header.type = htons( MANAGEMENT_SHOW_STATS_REQUEST );
+  request.header.length = htonl( sizeof( management_show_stats_request ) );
+
+  bool ret = send_request_message( service_name, get_trema_name(), MESSENGER_MANAGEMENT_REQUEST,
+                                   &request, sizeof( management_show_stats_request ), NULL );
+  if ( !ret ) {
+    printf( "Failed to send a show stats request to %s.\n", service_name );
+    exit( EXIT_FAILURE );
+  }
+}
+
+
+int
+main( int argc, char *argv[] ) {
+  // Initialize the Trema world
+  init_trema( &argc, &argv );
+
+  // Parse arguments
+  parse_arguments( argc, argv );
+
+  // Set a handler to handle show stats reply
+  add_message_replied_callback( get_trema_name(), handle_reply );
+
+  // Send a show stats request
+  send_show_stats_request();
+
+  // Set timeout
+  add_periodic_event_callback( 5, timeout, NULL );
+
+  // Main loop
+  start_trema();
+
+  return EXIT_SUCCESS;
+}
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */