icontrol 0.0.2 → 0.0.3

data/README.rdoc

@@ -1,5 +1,49 @@
-= icontrol
+= IControl: F5 BigIP SOAP API Client
 
+IControl allows you to easily connect to a BigIP F5 load balancer and by means of the SOAP API. You can programmatically do almost the same things that you would do throught the Web Interface.
+
+= Installing
+
+== Install the gem:
+    gem install icontrol
+
+= Configuring
+
+In order to configure you just have to set up the username, password and the server you're connecting to:
+
+	IControl.config[:user] =  'username'
+	IControl.config[:password] = 'secret'
+	IControl.config[:base_url] = "https://myf5.test.com/"
+
+= Using the library
+
+== Virtual Servers
+
+You can retreive, create, delete and modify virtual servers, for more information see IControl::LocalLB::VirtualServer. As an example of you what you can do:
+
+=== Creating a virtual Server
+ new_virtual_server = IControl::LocalLB::VirtualServer.create(:name => "foo_virtual_server",
+                                  :address => "192.168.1.1",
+                                  :port => "80",
+                                  :protocol => IControl::Common::ProtocolType::PROTOCOL_TCP,
+                                  :wildmask => "255.255.255.255",
+                         	  :type => IControl::LocalLB::VirtualServer::Type::RESOURCE_TYPE_POOL,
+                                  :default_pool =>  foo_pool,
+                                  :profiles => [])
+
+=== Obtaining an instance of a virtual server
+
+
+ my_virtual_server = IControl::LocalLB::VirtualServer.find("virtual_server_name")
+
+
+=== Changing its default pool
+
+ my_virtual_server.default_pool = IControl::LocalLB::Pool.find("my_new_default_pool")
+
+=== Destroying it
+
+ my_virtual_server.destroy
 
 == Note on Patches/Pull Requests
  

data/lib/icontrol/attributable.rb

@@ -18,10 +18,22 @@ module Attributable
   end
 
   module InstanceMethods
+
     def initialize(attributes)
       id = attributes.delete(self.class.id_name) if attributes && attributes[self.class.id_name]
       @attributes = attributes || {}
-      @attributes[:id] ||= id 
+      @attributes[:id] ||= id       
+      
+      # Now we define an alias for the id_name
+      id_name = self.class.id_name
+
+      (class << self; self; end).instance_eval do        
+        define_method(id_name) do 
+          @attributes[:id]
+        end
+      end if id_name
+
+
     end
 
     def method_missing(method_name,*args,&block)

data/lib/icontrol/base.rb

@@ -2,6 +2,7 @@
 require 'savon'
 require 'net/https'
 require 'digest/md5'
+Savon::Request.log = false
 
 # The idea is to create an object proxy to the web service client with the same structure 
 # than the IControl stuff
@@ -92,7 +93,7 @@ module IControl
   @config = {
     :user     => "",               # username and password
     :password => "",
-    :base_url => "",                # base url for the web-service (https://example.com/iControl/iControlPortal.cgi)
+    :base_url => "",                # base url for the web-service (https://example.com/)
     :test_mode => false,           # When test mode is set, the soap responses are saved (this is done to ease the testing fixtures generation)
     :test_path => File.join(File.dirname(__FILE__),"..","..","spec","fixtures"),
     :test_file_prefix => ""
@@ -146,7 +147,7 @@ module IControl
                 return @client if @client
                 @client = nil
                 if IControl.configured?
-                  @client = Savon::Client.new IControl.config[:base_url] + "?WSDL=#{name.to_s}.#{class_name.to_s}"
+                  @client = Savon::Client.new IControl.config[:base_url] + "/iControl/iControlPortal.cgi?WSDL=#{name.to_s}.#{class_name.to_s}"
                   @client.request.basic_auth IControl.config[:user],IControl.config[:password]
                   @client.request.http.ssl_client_auth( :verify_mode => OpenSSL::SSL::VERIFY_NONE )
                 else

data/lib/icontrol/common.rb

@@ -5,38 +5,12 @@ module IControl
   class NoSuchPoolException < Exception; end
   class MethodNotImplementedException < Exception; end
 
-  module ConstDefiner
-
-    def self.included(base)
-      base.extend(ClassMethods)
-    end
-    
-    module ClassMethods
-
-      def class_name
-        self.name.split("::").last
-      end
-      
-      def declare_constants(constants,parent)
-        constants.each do |const_name|
-          klass = Class.new(parent)
-          self.const_set(const_name,klass)
-        end
-      end
-      
-      def from_string(klass)
-        return eval("::#{self}::#{klass}")
-      end 
-    end
-  end
-
-
   module Common
 
     class VLANFilterList
       attr_accessor :state,:vlans
       def initialize(attributes)
-        @state = IControl::Common::EnabledState.from_string(attributes[:state])
+        @state = IControl::Common::EnabledState.const_get(attributes[:state])
         @vlans = [attributes[:vlans][:item]].flatten.compact
       end
 
@@ -72,30 +46,56 @@ module IControl
     end
 
     class SourcePortBehavior
-      include ConstDefiner
-      declare_constants [:SOURCE_PORT_PRESERVE,:SOURCE_PORT_PRESERVE_STRICT,:SOURCE_PORT_CHANGE],SourcePortBehavior
+      #   Attempt to preserve the source port (best effort). This is the default.
+      class SOURCE_PORT_PRESERVE ; VALUE = 0; end
+      #  Preserve source port. Use of the preserve-strict setting should be restricted to UDP only under very special circumstances such as nPath or transparent (that is, no translation of any other L3/L4 field), where there is a 1:1 relationship between virtual IP addresses and node addresses, or when clustered multi-processing (CMP) is disabled.
+      class SOURCE_PORT_PRESERVE_STRICT ; VALUE = 1; end
+      #  The change setting is useful for obfuscating internal network addresses. 
+      class SOURCE_PORT_CHANGE ; VALUE = 2; end
     end
-
+    
+    ## 
+    #  A list of enabled states. 
     class EnabledState      
-      include ConstDefiner
-      declare_constants [:STATE_ENABLED,:STATE_DISABLED],EnabledState
+      class STATE_DISABLED ; VALUE = 0; end
+      class STATE_ENABLED  ; VALUE = 1; end
     end
 
+    ##
+    #  A list of TMOS modules. 
     class TMOSModule
-      include ConstDefiner
-      declare_constants [:TMOS_MODULE_ASM, :TMOS_MODULE_SAM, :TMOS_MODULE_WAM], TMOSModule
+      class TMOS_MODULE_SAM ; VALUE = 0; end
+      class TMOS_MODULE_ASM ; VALUE = 1; end
+      class TMOS_MODULE_WAM ; VALUE = 2; end
     end
 
+    ##
+    #  An enumeration of protocol types.
     class ProtocolType
-      
-      include ConstDefiner
-      
-      valid_consts = [:PROTOCOL_ANY,:PROTOCOL_IPV6,:PROTOCOL_ROUTING,:PROTOCOL_NONE,
-                      :PROTOCOL_FRAGMENT,:PROTOCOL_DSTOPTS,:PROTOCOL_TCP,:PROTOCOL_UDP,
-                      :PROTOCOL_ICMP,:PROTOCOL_ICMPV6,:PROTOCOL_OSPF,:PROTOCOL_SCTP]
-      
-      declare_constants valid_consts,ProtocolType
-
+      #   Protocol is wildcard.
+      class PROTOCOL_ANY ; VALUE = 0; end
+      #  Protocol is IPv6 header.
+      class PROTOCOL_IPV6 ; VALUE = 1; end
+      #  Protocol is IPv6 routing header.
+      class PROTOCOL_ROUTING ; VALUE = 2; end
+      #  Protocol is IPv6 no next header.
+      class PROTOCOL_NONE ; VALUE = 3; end
+      #  Protocol is IPv6 fragmentation header.
+      class PROTOCOL_FRAGMENT ; VALUE = 4; end
+      #  Protocol is IPv6 destination options.
+      class PROTOCOL_DSTOPTS ; VALUE = 5; end
+      #  Protocol is TCP.
+      class PROTOCOL_TCP ; VALUE = 6; end
+      #  Protocol is UDP.
+      class PROTOCOL_UDP ; VALUE = 7; end
+      #  Protocol is ICMP.
+      class PROTOCOL_ICMP ; VALUE = 8; end
+      #  Protocol is ICMPv6.
+      class PROTOCOL_ICMPV6 ; VALUE = 9; end
+      #  Protocol is OSPF.
+      class PROTOCOL_OSPF ; VALUE = 10; end
+      #  Protocol is SCTP.             
+      class PROTOCOL_SCTP ; VALUE = 11; end
     end
 
     class IPPortDefinition

data/lib/icontrol/local_lb/common.rb

@@ -1,47 +1,76 @@
 module IControl
   module LocalLB
     
+    ##
+    #  A list of possible levels for hardware acceleration. 
     class HardwareAccelerationMode
-      include IControl::ConstDefiner
-      declare_constants  [:HW_ACCELERATION_MODE_NONE, :HW_ACCELERATION_ASSIST, :HW_ACCELERATION_MODE_FULL ], HardwareAccelerationMode
+      class HW_ACCELERATION_MODE_NONE ; VALUE = 0; end
+      class HW_ACCELERATION_MODE_ASSIST; VALUE = 1; end
+      class HW_ACCELERATION_MODE_FULL; VALUE = 2 ; end
     end
 
+    ##
+    #  An enumeration of clone pool types. 
     class ClonePoolType
-      include IControl::ConstDefiner
-      declare_constants [ :CLONE_POOL_TYPE_UNDEFINED, :CLONE_POOL_TYPE_CLIENTSIDE, :CLONE_POOL_TYPE_SERVERSIDE ], ClonePoolType
+      class CLONE_POOL_TYPE_UNDEFINED ; VALUE = 0; end
+      class CLONE_POOL_TYPE_CLIENTSIDE; VALUE = 1; end
+      class CLONE_POOL_TYPE_SERVERSIDE; VALUE = 2 ; end
     end
 
+    ##
+    #  A list of profile context types. 
     class ProfileContextType
-      include IControl::ConstDefiner
-      declare_constants [ :PROFILE_CONTEXT_TYPE_ALL, :PROFILE_CONTEXT_TYPE_CLIENT, :PROFILE_CONTEXT_TYPE_SERVER], ProfileContextType
+      class PROFILE_CONTEXT_TYPE_ALL ; VALUE = 0; end
+      class PROFILE_CONTEXT_TYPE_CLIENT; VALUE = 1; end
+      class PROFILE_CONTEXT_TYPE_SERVER; VALUE = 2 ; end
     end
-
-    class SnatType
-      include IControl::ConstDefiner
-      declare_constants [ :SNAT_TYPE_NONE, :SNAT_TYPE_TRANSLATION_ADDRESS, :SNAT_TYPE_SNATPOOL, :SNAT_TYPE_AUTOMAP ], SnatType
+    
+    ##
+    #  A list of translation types.
+    class SnatType     
+      class SNAT_TYPE_NONE ; VALUE = 0; end
+      class SNAT_TYPE_TRANSLATION_ADDRESS ; VALUE = 1; end
+      class SNAT_TYPE_SNATPOOL ; VALUE = 2; end
+      class SNAT_TYPE_AUTOMAP ; VALUE = 3 ; end
     end
 
+    ##
+    #  A list of profile types. 
     class ProfileType
-      include IControl::ConstDefiner
-
-      declare_constants [ :PROFILE_TYPE_TCP, :PROFILE_TYPE_UDP, :PROFILE_TYPE_FTP, :PROFILE_TYPE_FAST_L4, 
-                          :PROFILE_TYPE_HTTP, :PROFILE_TYPE_SERVER_SSL, :PROFILE_TYPE_CLIENT_SSL, 
-                          :PROFILE_TYPE_AUTH, :PROFILE_TYPE_PERSISTENCE, :PROFILE_TYPE_CONNECTION_POOL, 
-                          :PROFILE_TYPE_STREAM, :PROFILE_TYPE_FAST_HTTP, :PROFILE_TYPE_IIOP, 
-                          :PROFILE_TYPE_RTSP, :PROFILE_TYPE_STATISTICS, :PROFILE_TYPE_HTTPCLASS, :PROFILE_TYPE_SCTP, 
-                          :PROFILE_TYPE_INSTANCE, :PROFILE_TYPE_SIPP ], ProfileType
+      class PROFILE_TYPE_TCP ; VALUE = 0; end
+      class PROFILE_TYPE_UDP; VALUE = 1; end
+      class PROFILE_TYPE_FTP; VALUE = 2; end
+      class PROFILE_TYPE_FAST_L4; VALUE = 3; end
+      class PROFILE_TYPE_HTTP; VALUE = 4; end
+      class PROFILE_TYPE_SERVER_SSL; VALUE = 5; end
+      class PROFILE_TYPE_CLIENT_SSL; VALUE = 6; end
+      class PROFILE_TYPE_AUTH; VALUE = 7; end
+      class PROFILE_TYPE_PERSISTENCE; VALUE = 8; end
+      class PROFILE_TYPE_CONNECTION_POOL; VALUE = 9; end
+      class PROFILE_TYPE_STREAM; VALUE = 10; end
+      class PROFILE_TYPE_XML; VALUE = 11; end
+      class PROFILE_TYPE_FAST_HTTP; VALUE = 12; end
+      class PROFILE_TYPE_IIOP; VALUE = 13; end
+      class PROFILE_TYPE_RTSP; VALUE = 14; end
+      class PROFILE_TYPE_STATISTICS; VALUE = 15; end
+      class PROFILE_TYPE_HTTPCLASS; VALUE = 16; end
+      class PROFILE_TYPE_DNS; VALUE = 17; end
+      class PROFILE_TYPE_SCTP; VALUE = 18; end
+      class PROFILE_TYPE_INSTANCE; VALUE = 19; end
+      class PROFILE_TYPE_SIPP; VALUE = 20 ; end
     end
 
-    class PersistenceMode
-      include IControl::ConstDefiner
-
-      declare_constants [ :PERSISTENCE_MODE_NONE, :PERSISTENCE_MODE_SOURCE_ADDRESS_AFFINITY, 
-                          :PERSISTENCE_MODE_DESTINATION_ADDRESS_AFFINITY, :PERSISTENCE_MODE_COOKIE, 
-                          :PERSISTENCE_MODE_MSRDP, :PERSISTENCE_MODE_SSL_SID, :PERSISTENCE_MODE_SIP, 
-                          :PERSISTENCE_MODE_UIE, :PERSISTENCE_MODE_HASH ], PersistenceMode
-      
-    end
 
-    
+    class PersistenceMode
+      class PERSISTENCE_MODE_NONE ; VALUE = 0; end
+      class PERSISTENCE_MODE_SOURCE_ADDRESS_AFFINITY; VALUE = 1; end
+      class PERSISTENCE_MODE_DESTINATION_ADDRESS_AFFINITY; VALUE = 2; end
+      class PERSISTENCE_MODE_COOKIE; VALUE = 3; end
+      class PERSISTENCE_MODE_MSRDP; VALUE = 4; end
+      class PERSISTENCE_MODE_SSL_SID; VALUE = 5; end
+      class PERSISTENCE_MODE_SIP; VALUE = 6; end
+      class PERSISTENCE_MODE_UIE; VALUE = 7; end
+      class PERSISTENCE_MODE_HASH; VALUE = 8 ; end
+    end   
   end
 end

data/lib/icontrol/local_lb/virtual_server.rb

@@ -8,20 +8,34 @@
 IControl::LocalLB::VirtualServer
 class IControl::LocalLB::VirtualServer
 
-  class Type    
-    include IControl::ConstDefiner
-    valid_consts = [:RESOURCE_TYPE_POOL,:RESOURCE_TYPE_IP_FORWARDING,:RESOURCE_TYPE_L2_FORWARDING,
-                    :RESOURCE_TYPE_REJECT,:RESOURCE_TYPE_FAST_L4,:RESOURCE_TYPE_FAST_HTTP]
-    
-    declare_constants valid_consts,Type
+  ##
+  #  A list of virtual server types. 
+  module Type    
+    #   The virtual server is based on a pool.
+    class RESOURCE_TYPE_POOL ; VALUE = 0; end
+    #  The virtual server only supports IP forwarding. There is no load balancing on this type of virtual server.
+    class RESOURCE_TYPE_IP_FORWARDING ; VALUE = 1; end
+    #  The virtual server only supports L2 forwarding. There is no load balancing on this type of virtual server.
+    class RESOURCE_TYPE_L2_FORWARDING ; VALUE = 2; end
+    #  All connections going to this virtual server will be rejected, and resets will be sent.
+    class RESOURCE_TYPE_REJECT ; VALUE = 3; end
+    #  The virtual server is based on the Fast L4 profile.
+    class RESOURCE_TYPE_FAST_L4 ; VALUE = 4; end
+    #  The virtual server is based on the Fast HTTP profile. 
+    class RESOURCE_TYPE_FAST_HTTP ; VALUE = 5; end
   end
   
+  ##
+  #  A list of CMP enable modes. 
   class CMPEnableMode
-    include IControl::ConstDefiner
-    valid_consts = [:RESOURCE_TYPE_CMP_ENABLE_ALL,:RESOURCE_TYPE_CMP_ENABLE_SINGLE,
-                    :RESOURCE_TYPE_CMP_ENABLE_GROUP,:RESOURCE_TYPE_CMP_ENABLE_UNKNOWN]
-    
-    declare_constants valid_consts,CMPEnableMode
+    #   The virtual server is enabled on all processing cores.
+    class RESOURCE_TYPE_CMP_ENABLE_ALL ; VALUE = 0; end
+    #  The virtual server is enabled on a single processing core.
+    class RESOURCE_TYPE_CMP_ENABLE_SINGLE ; VALUE = 1; end
+    #  The virtual server is enabled on a group of processing cores.
+    class RESOURCE_TYPE_CMP_ENABLE_GROUP ; VALUE = 2; end
+
+    class RESOURCE_TYPE_CMP_ENABLE_UNKNOWN ; VALUE = 3; end
   end
 
   class StatisticEntry
@@ -34,7 +48,7 @@ class IControl::LocalLB::VirtualServer
         aux.virtual_server = IControl::LocalLB::VirtualServer.find(i[:virtual_server][:name])
         aux.statistics = {}
         i[:statistics][:item].each do |entry|
-          aux.statistics[IControl::Common::StatisticType.from_string(entry[:type])] = IControl::Common::ULong64.new(entry[:value])
+          aux.statistics[IControl::Common::StatisticType.const_get(entry[:type])] = IControl::Common::ULong64.new(entry[:value])
         end
         aux
       end
@@ -46,7 +60,7 @@ class IControl::LocalLB::VirtualServer
     attr_accessor :tmos_module ,:score
     def initialize(attribules)
       @score = attributes[:score]
-      @tmos_module = IControl::Common::TMOSModule.from_string(attributes[:tmos_module])
+      @tmos_module = IControl::Common::TMOSModule.const_get(attributes[:tmos_module])
     end    
   end
 =begin  
@@ -63,7 +77,7 @@ class IControl::LocalLB::VirtualServer
     attr_accessor :pool,:clone_type
     def initialize(attributes)
       @pool = IControl::LocalLB::Pool.find(attributes[:pool_name])
-      @clone_type = IControl::LocalLB::ClonePoolType.from_string(attributes[:type])
+      @clone_type = IControl::LocalLB::ClonePoolType.const_get(attributes[:type])
     end
   end
 
@@ -149,7 +163,7 @@ class IControl::LocalLB::VirtualServer
     
     attributes[:profiles].each do |i| 
       if i.class == Hash
-        profiles["item#{item+=1}"] = {"profile_context" => i["profile_context"].class_name,"profile_name" => ( i["profile"] ? i["profile"] :  "tcp" ) }
+        profiles["item#{item+=1}"] = {"profile_context" => i["profile_context"].name.split("::").last,"profile_name" => ( i["profile"] ? i["profile"] :  "tcp" ) }
       else
         profiles["item#{item+=1}"] = {"profile_context" => i.profile_context,"profile_name" => i.profile_name || "tcp" }
       end
@@ -160,12 +174,12 @@ class IControl::LocalLB::VirtualServer
             :name => attributes[:name],
             :address => attributes[:address],
             "port" => attributes[:port],
-            :protocol => (attributes[:protocol] || IControl::Common::ProtocolType::PROTOCOL_TCP).class_name
+            :protocol => (attributes[:protocol] || IControl::Common::ProtocolType::PROTOCOL_TCP).name.split("::").last
           }
         },
         "wildmasks" => {:item => attributes[:wildmask] || "255.255.255.255"},
         "resources" => {:item => {          
-            :type => (attributes[:type] || IControl::LocalLB::VirtualServer::Type::RESOURCE_TYPE_POOL).class_name,
+            :type => (attributes[:type] || IControl::LocalLB::VirtualServer::Type::RESOURCE_TYPE_POOL).name.split("::").last,
             "default_pool_name" => attributes[:default_pool] ? attributes[:default_pool].id : "" 
           }
         },
@@ -203,7 +217,7 @@ class IControl::LocalLB::VirtualServer
     IControl::LocalLB::VirtualServer.set_type do |soap|
       soap.body = {
         "virtual_servers" => {:item => id},
-        "types" => {:item => new_type.class_name}
+        "types" => {:item => new_type.name.split("::").last}
       }
     end
   end
@@ -244,12 +258,20 @@ class IControl::LocalLB::VirtualServer
     super
   end
 
+  def disable
+    self.enabled_state= IControl::Common::EnabledState::STATE_DISABLED
+  end
+
+  def enable
+    self.enabled_state= IControl::Common::EnabledState::STATE_ENABLED
+  end
+
   # Sets the enabled state of the specified virtual servers.  
   def enabled_state=(state)
     IControl::LocalLB::VirtualServer.set_enabled_state do |soap|
       soap.body = {
         "virtual_servers" => {:item => id},
-        "states" => {:item => state.class_name}
+        "states" => {:item => state.name.split("::").last}
       }
     end if state
   end
@@ -274,7 +296,7 @@ class IControl::LocalLB::VirtualServer
 
   def connection_mirror_state=(state)
     IControl::LocalLB::VirtualServer.set_connection_mirror_state do |soap|
-      soap.body = {"virtual_servers" => {:item => id},:states => {:item => state.class_name }}
+      soap.body = {"virtual_servers" => {:item => id},:states => {:item => state.name.split("::").last }}
     end
   end
   
@@ -286,7 +308,10 @@ class IControl::LocalLB::VirtualServer
   # Sets the connection limits of the specified virtual server.
   def connection_limit=(limit)
     IControl::LocalLB::VirtualServer.set_connection_limit do |soap|
-      soap.body = {"virtual_servers" => {:item => id},:limits => {:item => IControl::Common::ULong64.new(:low => limit).to_hash}}
+      limit = limit.is_a?(IControl::Common::ULong64) ? limit : IControl::Common::ULong64.new(:low => limit)
+      soap.body = {
+        "virtual_servers" => {:item => id},
+        :limits => {:item => limit.to_hash}}
     end
   end
 
@@ -363,6 +388,7 @@ private
 
 public 
 
+
   # Gets the lists of VLANs on which access to the specified Virtual Servers are enabled/disabled.
   def vlan
     super
@@ -418,10 +444,37 @@ public
     super
   end
 
+
+  def remove_all_persistence_profiles! 
+    IControl::LocalLB::VirtualServer.remove_all_persistence_profiles do |soap|
+      soap.body = {
+        "virtual_servers" => {:item => id }
+      }
+    end    
+  end
+
   def persistence_profile=(persistence_profile)
+    self.remove_all_persistence_profiles!
     add_persistence_profile(persistence_profile)
   end
 
+
+  def remove_persistence_profile(persistence_profile)
+    IControl::LocalLB::VirtualServer.remove_persistence_profile do |soap|
+      soap.body = {
+        "virtual_servers" => {:item => id},
+        "profiles" => {:item => {:item => {"profile_name" => persistence_profile.profile_name,"default_profile" => persistence_profile.default_profile}}}
+      } 
+    end if persistence_profile && persistence_profile!= ""    
+  end
+
+
+  def default_persistence_profile=(persistence_profile)
+    self.remove_persistence_profile(persistence_profile)
+    persistence_profile.default_profile = true
+    self.add_persistence_profile(persistence_profile)
+  end
+
   def add_persistence_profile(persistence_profile)
     IControl::LocalLB::VirtualServer.add_persistence_profile do |soap|
       soap.body = {
@@ -465,7 +518,7 @@ public
     IControl::LocalLB::VirtualServer.add_rule do |soap|
       item = "item"; count = 0
       my_rules = {}
-      rules.each{ |i| my_rules[item + (count +=1).to_s] =  {"rule_name" => i.class == String ? i : i.id, "priority" => count } }
+      rules.each{ |i| my_rules[item + (count +=1).to_s] =  {"rule_name" => i.class == String ? i : i.rule_name, "priority" => count } }
       soap.body = {
         "virtual_servers" => {:item => id},
         "rules" => {"value" =>  my_rules  }
@@ -524,7 +577,7 @@ public
     IControl::LocalLB::VirtualServer.set_protocol do |soap|
       soap.body = {
         "virtual_servers" => {:item => id},
-        "protocols" => {:item => protocol.class_name }
+        "protocols" => {:item => protocol.name.split("::").last }
       }
     end
   end
@@ -548,11 +601,11 @@ public
   end
 
   def enable_cmp!
-    cmp_enabled_state = IControl::Common::EnabledState::STATE_ENABLED
+    self.cmp_enabled_state = IControl::Common::EnabledState::STATE_ENABLED
   end
 
   def disable_cmp!
-    cmp_enabled_state = IControl::Common::EnabledState::STATE_DISABLED
+    self.cmp_enabled_state = IControl::Common::EnabledState::STATE_DISABLED
   end
   
   # Sets the translate cmp address state receives a EnabledState constant
@@ -572,7 +625,7 @@ public
         "virtual_servers" => {:item => id},
         "snatpools" => {:item => snat_pool.id}
       }
-    end
+    end if snat_pool
   end
 
   def destroy 
@@ -588,8 +641,19 @@ public
   end
   
   def vlan=(vlan)
-    puts "NOT IMPLEMENTED"
-    raise LocalLB::MethodNotImplementedException
+    count=0;
+    vlans = {}
+    vlan.vlans.each { |i| vlans["item#{count+=1}"]=i}
+    IControl::LocalLB::VirtualServer.set_vlan do |soap|
+      soap.body = {
+        "virtual_servers" => {:item => id},
+        "vlans" => { :item => 
+          { :state => vlan.state.name.split("::").last,
+            :vlans => vlans
+          }
+        }
+      }
+    end
   end
   
 =begin