fake_ftp 0.2.0 → 0.3.0

data/lib/fake_ftp/server_commands.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+  end
+end

data/lib/fake_ftp/server_commands/acct.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Acct
+      def run(*)
+        '230 WHATEVER!'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/cdup.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Cdup
+      def run(*)
+        '250 OK!'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/cwd.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Cwd
+      def run(ctx, wd, *)
+        wd = "/#{wd}" unless wd.start_with?('/')
+        ctx.workdir = wd
+        '250 OK!'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/dele.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Dele
+      def run(ctx, filename = '', *)
+        files_to_delete = ctx.store.values.select do |f|
+          if ctx.absolute?
+            ctx.abspath(::File.basename(filename)) == ctx.abspath(f.name)
+          else
+            ::File.basename(filename) == f.name
+          end
+        end
+
+        return '550 Delete operation failed.' if files_to_delete.empty?
+
+        ctx.store.reject! do |_, f|
+          files_to_delete.include?(f)
+        end
+
+        '250 Delete operation successful.'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/list.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class List
+      def run(ctx, *args)
+        if ctx.active? && ctx.command_state[:active_connection].nil?
+          ctx.respond_with('425 Ain\'t no data port!')
+          return
+        end
+
+        ctx.respond_with('150 Listing status ok, about to open data connection')
+        data_client = if ctx.active?
+                        ctx.command_state[:active_connection]
+                      else
+                        ctx.data_server.accept
+                      end
+
+        wildcards = ctx.build_wildcards(args)
+        statlines = ctx.matching_files(wildcards).map do |f|
+          %W[
+            -rw-r--r--
+            1
+            owner
+            group
+            #{f.bytes}
+            #{f.created.strftime('%b %d %H:%M')}
+            #{f.name}
+          ].join("\t")
+        end
+        data_client.write(statlines.join("\n"))
+        data_client.close
+        ctx.command_state[:active_connection] = nil
+
+        '226 List information transferred'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/mdtm.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Mdtm
+      def run(ctx, filename = '', *)
+        ctx.respond_with('501 No filename given') && return if filename.empty?
+        server_file = ctx.file(filename)
+        ctx.respond_with('550 File not found') && return if server_file.nil?
+
+        ctx.respond_with(
+          "213 #{server_file.last_modified_time.strftime('%Y%m%d%H%M%S')}"
+        )
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/mkd.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Mkd
+      def run(*)
+        '257 OK!'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/nlst.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Nlst
+      def run(ctx, *args)
+        if ctx.active? && ctx.command_state[:active_connection].nil?
+          ctx.respond_with('425 Ain\'t no data port!')
+          return
+        end
+
+        ctx.respond_with('150 Listing status ok, about to open data connection')
+        data_client = if ctx.active?
+                        ctx.command_state[:active_connection]
+                      else
+                        ctx.data_server.accept
+                      end
+
+        wildcards = ctx.build_wildcards(args)
+        matching = ctx.matching_files(wildcards).map do |f|
+          "#{f.name}\n"
+        end
+
+        data_client.write(matching.join)
+        data_client.close
+        ctx.command_state[:active_connection] = nil
+
+        '226 List information transferred'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/pass.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Pass
+      def run(*)
+        '230 logged in'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/pasv.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Pasv
+      def run(ctx, *)
+        return '502 Aww hell no, use Active' if ctx.passive_port.nil?
+        ctx.mode = :passive
+        p1 = (ctx.passive_port / 256).to_i
+        p2 = ctx.passive_port % 256
+        "227 Entering Passive Mode (127,0,0,1,#{p1},#{p2})"
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/port.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Port
+      def run(ctx, remote = '', *)
+        remote = remote.split(',')
+        remote_port = remote[4].to_i * 256 + remote[5].to_i
+        unless ctx.command_state[:active_connection].nil?
+          ctx.command_state[:active_connection].close
+          ctx.command_state[:active_connection] = nil
+        end
+        ctx.mode = :active
+        ctx.debug('_port active connection ->')
+        ctx.command_state[:active_connection] = ::TCPSocket.new(
+          '127.0.0.1', remote_port
+        )
+        ctx.debug('_port active connection <-')
+        '200 Okay'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/pwd.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Pwd
+      def run(ctx, *)
+        "257 \"#{ctx.workdir}\" is current directory"
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/quit.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Quit
+      def run(ctx, *)
+        ctx.respond_with '221 OMG bye!'
+        ctx.client&.close
+        ctx.client = nil
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/retr.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Retr
+      def run(ctx, filename = '', *)
+        ctx.respond_with('501 No filename given') if filename.empty?
+
+        f = ctx.file(filename.to_s)
+        return ctx.respond_with('550 File not found') if f.nil?
+
+        if ctx.active? && ctx.command_state[:active_connection].nil?
+          ctx.respond_with('425 Ain\'t no data port!')
+          return
+        end
+
+        ctx.respond_with('150 File status ok, about to open data connection')
+        data_client = if ctx.active?
+                        ctx.command_state[:active_connection]
+                      else
+                        ctx.data_server.accept
+                      end
+
+        data_client.write(f.data)
+
+        data_client.close
+        ctx.command_state[:active_connection] = nil
+        '226 File transferred'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/rnfr.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Rnfr
+      def run(ctx, rename_from = '', *)
+        return '501 Send path name.' if rename_from.nil? || rename_from.empty?
+
+        ctx.command_state[:rename_from] = if ctx.absolute?
+                                            ctx.abspath(rename_from)
+                                          else
+                                            rename_from
+                                          end
+        '350 Send RNTO to complete rename.'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/rnto.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Rnto
+      def run(ctx, rename_to = '', *)
+        return '501 Send path name.' if rename_to.nil? || rename_to.empty?
+        return '503 Send RNFR first.' if ctx.command_state[:rename_from].nil?
+
+        f = ctx.file(ctx.command_state[:rename_from])
+        if f.nil?
+          ctx.command_state[:rename_from] = nil
+          return '550 File not found.'
+        end
+
+        f.name = rename_to
+        ctx.command_state[:rename_from] = nil
+        '250 Path renamed.'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/site.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Site
+      def run(_, command, *)
+        "200 #{command}"
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/size.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Size
+      def run(ctx, filename, *)
+        ctx.respond_with("213 #{ctx.file(filename).bytes}")
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/stor.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Stor
+      def run(ctx, filename = '', *)
+        if ctx.active? && ctx.command_state[:active_connection].nil?
+          ctx.respond_with('425 Ain\'t no data port!')
+          return
+        end
+
+        ctx.respond_with('125 Do it!')
+        data_client = if ctx.active?
+                        ctx.command_state[:active_connection]
+                      else
+                        ctx.data_server.accept
+                      end
+
+        data = data_client.read(nil)
+        ctx.store[ctx.abspath(filename)] = FakeFtp::File.new(
+          filename.to_s, data, ctx.mode
+        )
+
+        data_client.close
+        ctx.command_state[:active_connection] = nil
+        '226 Did it!'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/type.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Type
+      def run(_, type = 'A', *)
+        case type.to_s
+        when 'A'
+          '200 Type set to A.'
+        when 'I'
+          '200 Type set to I.'
+        else
+          '504 We don\'t allow those'
+        end
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/user.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class User
+      def run(_, name, *)
+        return '230 logged in' if name.to_s == 'anonymous'
+        '331 send your password'
+      end
+    end
+  end
+end

data/lib/fake_ftp/server_commands/wat.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module FakeFtp
+  module ServerCommands
+    class Wat
+      def run(ctx, *)
+        if ctx.active? && ctx.command_state[:active_connection].nil?
+          ctx.respond_with('425 Ain\'t no data port!')
+          return
+        end
+
+        data_client = if ctx.active?
+                        ctx.command_state[:active_connection]
+                      else
+                        ctx.data_server.accept
+                      end
+        data_client.write(invisible_bike)
+        data_client.close
+        ctx.command_state[:active_connection] = nil
+        '418 Pizza Party'
+      end
+
+      private def invisible_bike
+        ::File.read(
+          ::File.expand_path(
+            '../../../../spec/fixtures/invisible_bike.jpg',
+            __FILE__
+          )
+        )
+      end
+    end
+  end
+end