ronin-support 0.5.2 → 1.0.0.beta2

data/lib/ronin/support/network/esmtp/mixin.rb

@@ -0,0 +1,115 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support 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 Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/network/smtp/mixin'
+
+module Ronin
+  module Support
+    module Network
+      module ESMTP
+        #
+        # Provides helper methods for communicating with ESMTP services.
+        #
+        module Mixin
+          include SMTP::Mixin
+
+          #
+          # Alias for {SMTP::Mixin#smtp_message}.
+          #
+          # @param [Hash{Symbol => Object}] kwargs
+          #   Additional keyword arguments for {SMTP::Mixin#smtp_message}.
+          #
+          # @return [SMTP::Email]
+          #   The new email object.
+          #
+          # @see SMTP.message
+          #
+          # @api public
+          #
+          def esmtp_message(**kwargs,&block)
+            smtp_message(**kwargs,&block)
+          end
+
+          #
+          # Creates a connection to the ESMTP server.
+          #
+          # @param [String] host
+          #   The host to connect to.
+          #
+          # @param [Hash{Symbol => Object}] kwargs
+          #   Additional keyword arguments for {#smtp_connect}.
+          #
+          # @option kwargs [Integer] :port (SMTP::DEFAULT_PORT)
+          #   The port to connect to.
+          #
+          # @option kwargs [Boolean, Hash] :ssl
+          #   Additional SSL options.
+          #
+          # @option :ssl [Boolean] :verify
+          #   Specifies that the SSL certificate should be verified.
+          #
+          # @option :ssl [String] :certs
+          #   The path to the file containing CA certs of the server.
+          #
+          # @option kwargs [String] :helo
+          #   The HELO domain.
+          #
+          # @option kwargs [Symbol] :auth
+          #   The type of authentication to use.
+          #   May be one of the following:
+          #
+          #   * `:login`
+          #   * `:plain`
+          #   * `:cram_md5`
+          #
+          # @option kwargs [String] :user
+          #   The user-name to authenticate with.
+          #
+          # @option kwargs [String] :password
+          #   The password to authenticate with.
+          #
+          # @yield [esmtp]
+          #   If a block is given, it will be passed an ESMTP enabled session.
+          #   Once the block returns the ESMTP session will be closed.
+          #
+          # @yieldparam [Net::SMTP] esmtp
+          #   The ESMTP session.
+          #
+          # @return [Net::SMTP, nil]
+          #   The ESMTP enabled session. If a block is given, `nil` will be
+          #   returned.
+          #
+          # @api public
+          #
+          def esmtp_connect(host,**kwargs)
+            if block_given?
+              smtp_connect(host,**kwargs) do |smtp|
+                smtp.esmtp = true
+                yield smtp
+              end
+            else
+              smtp = smtp_connect(host,**kwargs)
+              smtp.esmtp = true
+              return smtp
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/network/exceptions.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support 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 Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ipaddr'
+
+module Ronin
+  module Support
+    module Network
+      #
+      # Exception for when a given IP address is invalid.
+      #
+      class InvalidIP < IPAddr::InvalidAddressError
+      end
+
+      #
+      # Exception for when a given hostname is invalid.
+      #
+      class InvalidHostname < RuntimeError
+      end
+
+      #
+      # Exception for when a given email address is invalid.
+      #
+      class InvalidEmailAddress < RuntimeError
+      end
+    end
+  end
+end

data/lib/ronin/support/network/ftp/mixin.rb

@@ -0,0 +1,106 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support 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 Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/network/dns/idn'
+
+begin
+  require 'net/ftp'
+rescue LoadError => error
+  warn "ronin/network/ftp requires the net-ftp gem be listed in the Gemfile."
+  raise(error)
+end
+
+module Ronin
+  module Support
+    module Network
+      module FTP
+        #
+        # Provides helper methods for communicating with FTP servers.
+        #
+        module Mixin
+          # Default FTP port
+          DEFAULT_PORT = 21
+
+          # Default FTP user
+          DEFAULT_USER = 'anonymous'
+
+          #
+          # Creates a connection to the FTP server.
+          #
+          # @param [String] host
+          #   The host to connect to.
+          #
+          # @param [Integer] port
+          #   The port to connect to.
+          #
+          # @param [String] user
+          #   The user to authenticate with.
+          #
+          # @param [String] password
+          #   The password to authenticate with.
+          #
+          # @param [String] account
+          #   The FTP account information to send via the `ACCT` command.
+          #
+          # @param [Boolean] passive
+          #   Specifies whether the FTP session should use passive mode.
+          #
+          # @yield [ftp]
+          #   If a block is given, it will be passed an FTP session object.
+          #   Once the block returns, the FTP session will be closed.
+          #
+          # @yieldparam [Net::FTP] ftp
+          #   The FTP session.
+          #
+          # @return [Net::FTP, nil]
+          #   The FTP session. If a block is given, then `nil` will be returned.
+          #
+          # @example
+          #   ftp_connect('www.example.com', user: 'joe', password: 'secret')
+          #
+          # @example
+          #   ftp_connect('www.example.com', user: 'joe', password: 'secret') do |ftp|
+          #     # ...
+          #   end
+          #
+          # @api public
+          #
+          def ftp_connect(host, port:     DEFAULT_PORT,
+                                user:     DEFAULT_USER,
+                                password: nil,
+                                account:  nil,
+                                passive:  true)
+            host = DNS::IDN.to_ascii(host)
+
+            ftp = Net::FTP.new
+            ftp.connect(host,port)
+            ftp.login(user,password,account)
+            ftp.passive = passive
+
+            if block_given?
+              yield ftp
+              ftp.close
+            else
+              return ftp
+            end
+          end
+        end
+      end
+    end
+  end
+end