investtools-ftpd 1.0.1

data/lib/ftpd/cmd_mdtm.rb

@@ -0,0 +1,27 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdMdtm < CommandHandler
+
+    def cmd_mdtm(path)
+      ensure_logged_in
+      ensure_file_system_supports :dir
+      ensure_file_system_supports :file_info
+      syntax_error unless path
+      path = File.expand_path(path, name_prefix)
+      ensure_accessible(path)
+      ensure_exists(path)
+      info = file_system.file_info(path)
+      mtime = info.mtime.utc
+      # We would like to report fractional seconds, too.  Sadly, the
+      # spec declares that we may not report more precision than is
+      # actually there, and there is no spec or API to tell us how
+      # many fractional digits are significant.
+      mtime = mtime.strftime("%Y%m%d%H%M%S")
+      reply "213 #{mtime}"
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_mkd.rb

@@ -0,0 +1,23 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdMkd < CommandHandler
+
+    def cmd_mkd(argument)
+      syntax_error unless argument
+      ensure_logged_in
+      ensure_file_system_supports :mkdir
+      path = File.expand_path(argument, name_prefix)
+      ensure_accessible path
+      ensure_exists File.dirname(path)
+      ensure_directory File.dirname(path)
+      ensure_does_not_exist path
+      file_system.mkdir path
+      reply %Q'257 "#{path}" created'
+    end
+    alias cmd_xmkd :cmd_mkd
+
+  end
+
+end

data/lib/ftpd/cmd_mode.rb

@@ -0,0 +1,27 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdMode < CommandHandler
+
+    def cmd_mode(argument)
+      syntax_error unless argument
+      ensure_logged_in
+      name, implemented = TRANSMISSION_MODES[argument]
+      error "Invalid mode code", 504 unless name
+      error "Mode not implemented", 504 unless implemented
+      self.mode = argument
+      reply "200 Mode set to #{name}"
+    end
+
+    private
+
+    TRANSMISSION_MODES = {
+      'B'=>['Block', false],
+      'C'=>['Compressed', false],
+      'S'=>['Stream', true],
+    }
+
+  end
+
+end

data/lib/ftpd/cmd_nlst.rb

@@ -0,0 +1,27 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdNlst < CommandHandler
+
+    def cmd_nlst(argument)
+      close_data_server_socket_when_done do
+        ensure_logged_in
+        ensure_file_system_supports :dir
+        path = list_path(argument)
+        path = File.expand_path(path, name_prefix)
+        transmit_file(StringIO.new(name_list(path)), 'A')
+      end
+    end
+
+    private
+
+    def name_list(path)
+      path_list(path).map do |path|
+        File.basename(path) + "\n"
+      end.join
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_noop.rb

@@ -0,0 +1,14 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdNoop < CommandHandler
+
+    def cmd_noop(argument)
+      syntax_error if argument
+      reply "200 Nothing done"
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_opts.rb

@@ -0,0 +1,14 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdOpts < CommandHandler
+
+    def cmd_opts(argument)
+      syntax_error unless argument
+      error 'Unsupported option', 501
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_pasv.rb

@@ -0,0 +1,28 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdPasv < CommandHandler
+
+    def cmd_pasv(argument)
+      ensure_logged_in
+      ensure_not_epsv_all
+      if data_server
+        reply "200 Already in passive mode"
+      else
+        interface = socket.addr[3]
+        self.data_server = TCPServer.new(interface, 0)
+        ip = data_server.addr[3]
+        port = data_server.addr[1]
+        quads = [
+          ip.scan(/\d+/),
+          port >> 8,
+          port & 0xff,
+        ].flatten.join(',')
+        reply "227 Entering passive mode (#{quads})"
+      end
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_pbsz.rb

@@ -0,0 +1,23 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdPbsz < CommandHandler
+
+    def cmd_pbsz(buffer_size)
+      ensure_tls_supported
+      syntax_error unless buffer_size =~ /^\d+$/
+      buffer_size = buffer_size.to_i
+      unless socket.encrypted?
+        error "PBSZ must be preceded by AUTH", 503
+      end
+      unless buffer_size == 0
+        error "PBSZ=0", 501
+      end
+      reply "200 PBSZ=0"
+      self.protection_buffer_size_set = true
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_port.rb

@@ -0,0 +1,28 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  # The Data Port (PORT) command.
+
+  class CmdPort < CommandHandler
+
+    def cmd_port(argument)
+      ensure_logged_in
+      ensure_not_epsv_all
+      pieces = argument.split(/,/)
+      syntax_error unless pieces.size == 6
+      pieces.collect! do |s|
+        syntax_error unless s =~ /^\d{1,3}$/
+        i = s.to_i
+        syntax_error unless (0..255) === i
+        i
+      end
+      hostname = pieces[0..3].join('.')
+      port = pieces[4] << 8 | pieces[5]
+      set_active_mode_address hostname, port
+      reply "200 PORT command successful"
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_prot.rb

@@ -0,0 +1,34 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdProt < CommandHandler
+
+    def cmd_prot(level_arg)
+      level_code = level_arg.upcase
+      unless protection_buffer_size_set
+        error "PROT must be preceded by PBSZ", 503
+      end
+      level = DATA_CHANNEL_PROTECTION_LEVELS[level_code]
+      unless level
+        error "Unknown protection level", 504
+      end
+      unless level == :private
+        error "Unsupported protection level #{level}", 536
+      end
+      self.data_channel_protection_level = level
+      reply "200 Data protection level #{level_code}"
+    end
+
+    private
+
+    DATA_CHANNEL_PROTECTION_LEVELS = {
+      'C'=>:clear,
+      'S'=>:safe,
+      'E'=>:confidential,
+      'P'=>:private
+    }
+
+  end
+
+end

data/lib/ftpd/cmd_pwd.rb

@@ -0,0 +1,15 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdPwd < CommandHandler
+
+    def cmd_pwd(argument)
+      ensure_logged_in
+      pwd 257
+    end
+    alias cmd_xpwd :cmd_pwd
+
+  end
+
+end

data/lib/ftpd/cmd_quit.rb

@@ -0,0 +1,18 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  # The Quit (QUIT) command
+
+  class CmdQuit < CommandHandler
+
+    def cmd_quit(argument)
+      syntax_error if argument
+      ensure_logged_in
+      reply "221 Byebye"
+      self.logged_in = false
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_rein.rb

@@ -0,0 +1,13 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdRein < CommandHandler
+
+    def cmd_rein(argument)
+      unimplemented_error
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_rename.rb

@@ -0,0 +1,32 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdRename < CommandHandler
+
+    def cmd_rnfr(argument)
+      ensure_logged_in
+      ensure_file_system_supports :rename
+      syntax_error unless argument
+      from_path = File.expand_path(argument, name_prefix)
+      ensure_accessible from_path
+      ensure_exists from_path
+      @rename_from_path = from_path
+      reply '350 RNFR accepted; ready for destination'
+      expect 'rnto'
+    end
+
+    def cmd_rnto(argument)
+      ensure_logged_in
+      ensure_file_system_supports :rename
+      syntax_error unless argument
+      to_path = File.expand_path(argument, name_prefix)
+      ensure_accessible to_path
+      ensure_does_not_exist to_path
+      file_system.rename(@rename_from_path, to_path)
+      reply '250 Rename successful'
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_rest.rb

@@ -0,0 +1,13 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdRest < CommandHandler
+
+    def cmd_rest(argument)
+      unimplemented_error
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_retr.rb

@@ -0,0 +1,24 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdRetr < CommandHandler
+
+    def cmd_retr(argument)
+      close_data_server_socket_when_done do
+        ensure_logged_in
+        ensure_file_system_supports :read
+        path = argument
+        syntax_error unless path
+        path = File.expand_path(path, name_prefix)
+        ensure_accessible path
+        ensure_exists path
+        file_system.read(path) do |file|
+          transmit_file file
+        end
+      end
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_rmd.rb

@@ -0,0 +1,22 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdRmd < CommandHandler
+
+    def cmd_rmd(argument)
+      syntax_error unless argument
+      ensure_logged_in
+      ensure_file_system_supports :rmdir
+      path = File.expand_path(argument, name_prefix)
+      ensure_accessible path
+      ensure_exists path
+      ensure_directory path
+      file_system.rmdir path
+      reply '250 RMD command successful'
+    end
+    alias cmd_xrmd :cmd_rmd
+
+  end
+
+end

data/lib/ftpd/cmd_site.rb

@@ -0,0 +1,13 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdSite < CommandHandler
+
+    def cmd_site(argument)
+      unimplemented_error
+    end
+
+  end
+
+end

data/lib/ftpd/cmd_size.rb

@@ -0,0 +1,29 @@
+require_relative 'command_handler'
+
+module Ftpd
+
+  class CmdSize < CommandHandler
+
+    def cmd_size(path)
+      ensure_logged_in
+      ensure_file_system_supports :read
+      syntax_error unless path
+      path = File.expand_path(path, name_prefix)
+      ensure_accessible(path)
+      ensure_exists(path)
+      file_system.read(path) do |file|
+        if data_type == 'A'
+          output = StringIO.new
+          io = Ftpd::Stream.new(output, data_type)
+          io.write(file)
+          size = output.size
+        else
+          size = file.size
+        end
+        reply "213 #{size}"
+      end
+    end
+
+  end
+
+end