trema 0.1.3

data/src/examples/traffic_monitor/counter.h

@@ -0,0 +1,48 @@
+/*
+ * Traffic counter.
+ *
+ * Author: Kazushi SUGYO
+ *
+ * Copyright (C) 2008-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.
+ */
+
+
+#ifndef COUNTER_H
+#define COUNTER_H
+
+
+typedef struct {
+  uint8_t mac[ ETH_ADDRLEN ];
+  uint64_t packet_count;
+  uint64_t byte_count;
+} counter;
+
+
+hash_table *create_counter( void );
+void add_counter( hash_table *db, uint8_t *mac, uint64_t packet_count, uint64_t byte_count );
+void foreach_counter( hash_table *db, void function( uint8_t *mac, counter *counter, void *user_data ), void *user_data );
+void delete_counter( hash_table *db );
+
+
+#endif // COUNTER_H
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */

data/src/examples/traffic_monitor/counter.rb

@@ -0,0 +1,46 @@
+#
+# Traffic counter.
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-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.
+#
+
+
+class Counter
+  def initialize
+    @db = {}
+  end
+
+
+  def add mac, packet_count, byte_count
+    @db[ mac ] ||= { :packet_count => 0, :byte_count => 0 }
+    @db[ mac ][ :packet_count ] += packet_count
+    @db[ mac ][ :byte_count ] += byte_count
+  end
+
+
+  def each_pair &block
+    @db.each_pair &block
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/src/examples/traffic_monitor/fdb.c

@@ -0,0 +1,76 @@
+/*
+ * Traffic counter.
+ *
+ * Author: Kazushi SUGYO
+ *
+ * Copyright (C) 2008-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 "trema.h"
+#include "fdb.h"
+
+
+hash_table *
+create_fdb() {
+  return create_hash( compare_mac, hash_mac );
+}
+
+
+uint16_t
+lookup_fdb( hash_table *db, uint8_t *mac ) {
+  fdb *fdb = lookup_hash_entry( db, mac );
+  return ( uint16_t ) ( fdb == NULL ? ENTRY_NOT_FOUND_IN_FDB : fdb->port_number );
+
+}
+
+
+void
+learn_fdb( hash_table *db, uint8_t *mac, uint16_t port_number ) {
+  fdb *fdb = lookup_hash_entry( db, mac );
+  if ( fdb == NULL ) {
+    fdb = xmalloc( sizeof( fdb ) );
+    memcpy( fdb->mac, mac, ETH_ADDRLEN );
+    fdb->port_number = port_number;
+    insert_hash_entry( db, fdb->mac, fdb );
+  }
+  else {
+    fdb->port_number = port_number;
+  }
+}
+
+
+static void
+free_each_value( void *key, void *value, void *user_data ) {
+  UNUSED( key );
+  UNUSED( user_data );
+  xfree( value );
+}
+
+
+void
+delete_fdb( hash_table *db ) {
+  foreach_hash( db, free_each_value, NULL );
+  delete_hash( db );
+}
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */

data/src/examples/traffic_monitor/fdb.h

@@ -0,0 +1,50 @@
+/*
+ * Traffic counter.
+ *
+ * Author: Kazushi SUGYO
+ *
+ * Copyright (C) 2008-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.
+ */
+
+
+#ifndef FDB_H
+#define FDB_H
+
+
+typedef struct {
+  uint8_t mac[ ETH_ADDRLEN ];
+  uint16_t port_number;
+} fdb;
+
+
+#define ENTRY_NOT_FOUND_IN_FDB OFPP_NONE
+
+
+hash_table *create_fdb( void );
+uint16_t lookup_fdb( hash_table *db, uint8_t *mac );
+void learn_fdb( hash_table *db, uint8_t *mac, uint16_t port_number );
+void delete_fdb( hash_table *db );
+
+
+#endif // FDB_H
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */

data/src/examples/traffic_monitor/fdb.rb

@@ -0,0 +1,44 @@
+#
+# Forwarding database (FDB) of layer-2 switch.
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-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.
+#
+
+
+class FDB
+  def initialize
+    @db = {}
+  end
+
+
+  def lookup mac
+    @db[ mac ]
+  end
+
+
+  def learn mac, port_number
+    @db[ mac ] = port_number
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/src/examples/traffic_monitor/traffic-monitor.rb

@@ -0,0 +1,100 @@
+#
+# Simple layer-2 switch with traffic monitoring.
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-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.
+#
+
+
+require "counter"
+require "fdb"
+
+
+class TrafficMonitor < Trema::Controller
+  periodic_timer_event :show_counter, 10
+
+
+  def start
+    @counter = Counter.new
+    @fdb = FDB.new
+  end
+
+
+  def packet_in datapath_id, message
+    macsa = message.macsa
+    macda = message.macda
+
+    @fdb.learn macsa, message.in_port
+    @counter.add macsa, 1, message.total_len
+    out_port = @fdb.lookup( macda )
+    if out_port
+      packet_out datapath_id, message, out_port
+      flow_mod datapath_id, macsa, macda, out_port
+    else
+      flood datapath_id, message
+    end
+  end
+
+
+  def flow_removed datapath_id, message
+    @counter.add message.match.dl_src, message.packet_count, message.byte_count
+  end
+
+
+  ##############################################################################
+  private
+  ##############################################################################
+
+
+  def show_counter
+    puts Time.now
+    @counter.each_pair do | mac, counter |
+      puts "#{ mac } #{ counter[ :packet_count ] } packets (#{ counter[ :byte_count ] } bytes)"
+    end
+  end
+
+
+  def flow_mod datapath_id, macsa, macda, out_port
+    send_flow_mod_add(
+      datapath_id,
+      :hard_timeout => 10,
+      :match => Match.new( :dl_src => macsa, :dl_dst => macda ),
+      :actions => Trema::ActionOutput.new( out_port )
+    )
+  end
+
+
+  def packet_out datapath_id, message, out_port
+    send_packet_out(
+      datapath_id,
+      :packet_in => message,
+      :actions => Trema::ActionOutput.new( out_port )
+    )
+  end
+
+
+  def flood datapath_id, message
+    packet_out datapath_id, message, OFPP_FLOOD
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/src/examples/traffic_monitor/traffic_monitor.c

@@ -0,0 +1,163 @@
+/*
+ * Traffic counter.
+ *
+ * Author: Kazushi SUGYO
+ *
+ * Copyright (C) 2008-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 <inttypes.h>
+#include "trema.h"
+#include "fdb.h"
+#include "counter.h"
+
+
+typedef struct {
+  hash_table *counter;
+  hash_table *fdb;
+} traffic;
+
+
+static void
+send_flow_mod( uint64_t datapath_id, uint8_t *macsa, uint8_t *macda, uint16_t out_port ) {
+  struct ofp_match match;
+  memset( &match, 0, sizeof( match ) );
+  match.wildcards = ( OFPFW_ALL & ~( OFPFW_DL_SRC | OFPFW_DL_DST ) );
+  memcpy( match.dl_src, macsa, OFP_ETH_ALEN );
+  memcpy( match.dl_dst, macda, OFP_ETH_ALEN );
+  uint16_t idle_timeout = 0;
+  uint16_t hard_timeout = 10;
+  uint16_t priority = UINT16_MAX;
+  uint32_t buffer_id = UINT32_MAX;
+  uint16_t out_port_for_delete_command = OFPP_NONE;
+  uint16_t flags = OFPFF_SEND_FLOW_REM;
+  openflow_actions *actions = create_actions();
+  append_action_output( actions, out_port, UINT16_MAX );
+  buffer *flow_mod = create_flow_mod( get_transaction_id(), match, get_cookie(), OFPFC_ADD,
+                                      idle_timeout, hard_timeout, priority, buffer_id,
+                                      out_port_for_delete_command, flags, actions );
+  delete_actions( actions );
+
+  send_openflow_message( datapath_id, flow_mod );
+  free_buffer( flow_mod );
+}
+
+
+static void
+send_packet_out( uint64_t datapath_id, packet_in *message, uint16_t out_port ) {
+  uint32_t buffer_id = message->buffer_id;
+  uint16_t in_port = message->in_port;
+  if ( datapath_id != message->datapath_id ) {
+    buffer_id = UINT32_MAX;
+    in_port = OFPP_NONE;
+  }
+  openflow_actions *actions = create_actions();
+  append_action_output( actions, out_port, UINT16_MAX );
+  const buffer *data = NULL;
+  if ( buffer_id == UINT32_MAX ) {
+    data = message->data;
+  }
+  buffer *packet_out = create_packet_out( get_transaction_id(), buffer_id, in_port, actions, data );
+  delete_actions( actions );
+
+  send_openflow_message( datapath_id, packet_out );
+  free_buffer( packet_out );
+}
+
+
+static void
+do_flooding( uint64_t datapath_id, packet_in *message ) {
+  send_packet_out( datapath_id, message, OFPP_FLOOD );
+}
+
+
+static void
+handle_packet_in( uint64_t datapath_id, packet_in message ) {
+  UNUSED( datapath_id );
+  packet_info packet_info = get_packet_info( message.data );
+  traffic *db = message.user_data;
+
+  uint8_t *macsa = packet_info.eth_macsa;
+  uint8_t *macda = packet_info.eth_macda;
+
+  learn_fdb( db->fdb, macsa, message.in_port );
+  add_counter( db->counter, macsa, 1, message.data->length );
+  uint16_t out_port = lookup_fdb( db->fdb, macda );
+  if ( out_port != ENTRY_NOT_FOUND_IN_FDB ) {
+     send_packet_out( datapath_id, &message, out_port );
+     send_flow_mod( datapath_id, macsa, macda, out_port );
+  }
+  else {
+     do_flooding( datapath_id, &message );
+  }
+}
+
+
+static void
+handle_flow_removed( uint64_t datapath_id, flow_removed message ) {
+  UNUSED( datapath_id );
+  traffic *db = message.user_data;
+  add_counter( db->counter, message.match.dl_src, message.packet_count, message.byte_count );
+}
+
+
+static void
+show_each_counter( uint8_t *mac, counter *counter, void *user_data ) {
+  UNUSED( user_data );
+  printf( "%02x:%02x:%02x:%02x:%02x:%02x %" PRIu64 " packets (%" PRIu64 " bytes)\n",
+          mac[ 0 ], mac[ 1 ], mac[ 2 ], mac[ 3 ], mac[ 4 ], mac[ 5 ],
+          counter->packet_count, counter->byte_count );
+}
+
+
+static void
+show_counter( void *user_data ) {
+  traffic *db = user_data;
+  time_t now = time( NULL );
+  printf( "%s", ctime( &now ) );
+  foreach_counter( db->counter, show_each_counter, NULL );
+}
+
+
+int
+main( int argc, char *argv[] ) {
+  init_trema( &argc, &argv );
+
+  traffic db;
+  db.counter = create_counter();
+  db.fdb = create_fdb();
+
+  add_periodic_event_callback( 10, show_counter, &db );
+
+  set_packet_in_handler( handle_packet_in, &db );
+  set_flow_removed_handler( handle_flow_removed, &db );
+
+  start_trema();
+
+  delete_fdb( db.fdb );
+  delete_counter( db.counter );
+
+  return 0;
+}
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */