pop3.rb in ronin-0.1.3

- old
+ new

@@ -1,11 +1,11 @@
 #
 #--
 # Ronin - A ruby development platform designed for information security
 # and data exploration tasks.
 #
-# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
 # the Free Software Foundation; either version 2 of the License, or
 # (at your option) any later version.
@@ -28,28 +28,26 @@
   module Sessions
     module POP3
       include Session
 
       setup_session do
-        parameter :pop3_host, :description => 'POP3 host'
-        parameter :pop3_port, :description => 'POP3 port'
+        parameter :host, :description => 'POP3 host'
+        parameter :port, :description => 'POP3 port'
 
         parameter :pop3_user, :description => 'POP3 user'
         parameter :pop3_password, :description => 'POP3 password'
       end
 
       protected
 
       def pop3_connect(options={},&block)
-        unless @pop3_host
-          raise(ParamNotFound,"Missing parameter #{describe_param(:pop3_host).dump}",caller)
-        end
+        require_params :host
 
-        options[:port] ||= @pop3_port
+        options[:port] ||= @port
         options[:user] ||= @pop3_user
         options[:password] ||= @pop3_password
 
-        return ::Net.pop3_connect(@pop3_host,options,&block)
+        return ::Net.pop3_connect(@host,options,&block)
       end
 
       def pop3_session(options={},&block)
         pop3_connect(options) do |sess|
           block.call(sess) if block