open_flow.rb in pio-0.24.0

- old
+ new
@@ -1,20 +1,58 @@
-module Pio
-  # OpenFlow specific types.
-  module OpenFlow
-    # OFPT_* constants.
-    HELLO = 0
-    ECHO_REQUEST = 2
-    ECHO_REPLY = 3
-    FEATURES_REQUEST = 5
-    FEATURES_REPLY = 6
-    PACKET_IN = 10
-    PORT_STATUS = 12
-    PACKET_OUT = 13
-    FLOW_MOD = 14
-  end
-end
-
+require 'pio/open_flow/datapath_id'
 require 'pio/open_flow/flags'
+require 'pio/open_flow/format'
+require 'pio/open_flow/message'
 require 'pio/open_flow/open_flow_header'
 require 'pio/open_flow/phy_port'
 require 'pio/open_flow/port_number'
+require 'pio/open_flow10'
+require 'pio/open_flow13'
+
+module Pio
+  # Common OpenFlow modules/classes.
+  module OpenFlow
+    def self.version
+      fail unless @version
+      @version
+    end
+
+    def self.switch_version(version)
+      [:Barrier, :Echo, :Features, :FlowMod, :Hello, :Match,
+       :PacketIn, :PacketOut, :SendOutPort, :PortStatus].each do |each|
+        set_message_class_name each, version
+        @version = version.to_s
+      end
+    end
+
+    # rubocop:disable MethodLength
+    def self.read(binary)
+      parser = {
+        0 => Pio::Hello,
+        2 => Pio::Echo::Request,
+        3 => Pio::Echo::Reply,
+        5 => Pio::Features::Request,
+        6 => Pio::Features::Reply,
+        10 => Pio::PacketIn,
+        12 => Pio::PortStatus
+      }
+      header = OpenFlowHeaderParser.read(binary)
+      parser.fetch(header.message_type).read(binary)
+    rescue
+      raise "Unknown message type #{header.message_type}"
+    end
+    # rubocop:enable MethodLength
+
+    def self.set_message_class_name(klass_name, version)
+      open_flow_module = Pio.const_get(version)
+      return unless open_flow_module.const_defined?(klass_name)
+      Pio.__send__ :remove_const, klass_name if Pio.const_defined?(klass_name)
+      Pio.const_set(klass_name, open_flow_module.const_get(klass_name))
+    rescue NameError
+      raise "#{version} is not supported yet."
+    end
+    private_class_method :set_message_class_name
+
+    # The default OpenFlow version is 1.0
+    switch_version 'OpenFlow10'
+  end
+end