ftpd 0.0.1.pre → 0.1.0

data/lib/ftpd/temp_dir.rb

@@ -1,16 +1,15 @@
-require 'fileutils'
-require 'tmpdir'
+module Ftpd
+  module TempDir
 
-module TempDir
-
-  def make
-    Dir.mktmpdir.tap do |path|
-      at_exit do
-        FileUtils.rm_rf path
-        Dir.rmdir path if File.exists?(path)
+    def make
+      Dir.mktmpdir.tap do |path|
+        at_exit do
+          FileUtils.rm_rf path
+          Dir.rmdir path if File.exists?(path)
+        end
       end
     end
-  end
-  module_function :make
+    module_function :make
 
+  end
 end

data/lib/ftpd/tls_server.rb

@@ -1,42 +1,48 @@
-require 'openssl'
-require 'ftpd/server'
-
 module Ftpd
   class TlsServer < Server
 
+    attr_accessor :tls
+    attr_accessor :certfile_path
+
     def initialize
-      @ssl_context = make_ssl_context
       super
+      @tls = :off
+      if tls_enabled?
+        unless @certfile_path
+          raise ArgumentError, ":certfile required if tls enabled"
+        end
+      end
     end
 
     private
 
     def make_server_socket
-      ssl_server_socket = OpenSSL::SSL::SSLServer.new(super, @ssl_context);
-      ssl_server_socket.start_immediately = false
-      ssl_server_socket
+      socket = super
+      if tls_enabled?
+        socket = OpenSSL::SSL::SSLServer.new(socket, ssl_context);
+        socket.start_immediately = false
+      end
+      socket
     end
 
     def accept
       socket = @server_socket.accept
-      add_tls_methods_to_socket(socket)
+      if tls_enabled?
+        add_tls_methods_to_socket(socket)
+      end
       socket
     end
 
-    def make_ssl_context
+    def ssl_context
       context = OpenSSL::SSL::SSLContext.new
-      File.open(certfile_path) do |certfile|
+      File.open(@certfile_path) do |certfile|
         context.cert = OpenSSL::X509::Certificate.new(certfile)
         certfile.rewind
         context.key = OpenSSL::PKey::RSA.new(certfile)
       end
       context
     end
-
-    def certfile_path
-      File.expand_path('../../insecure-test-cert.pem',
-                       File.dirname(__FILE__))
-    end
+    memoize :ssl_context
 
     def add_tls_methods_to_socket(socket)
       context = @ssl_context
@@ -53,5 +59,11 @@ module Ftpd
       end
     end
 
+    private
+
+    def tls_enabled?
+      @tls != :off
+    end
+
   end
 end

data/lib/ftpd/translate_exceptions.rb

@@ -0,0 +1,44 @@
+module Ftpd
+
+  # This module provides an easy interface to ExceptionTranslator.
+
+  module TranslateExceptions
+
+    include Memoizer
+
+    def self.included(includer)
+      includer.extend ClassMethods
+    end
+
+    module ClassMethods
+
+      # Cause the named method to translate exceptions.
+
+      def translate_exceptions(method_name)
+        original_method = instance_method(method_name)
+        define_method method_name do |*args|
+          exception_translator.translate_exceptions do
+            original_method.bind(self).call *args
+          end
+        end
+      end
+
+    end
+
+    # Add exception class e to the list of exceptions to be
+    # translated.
+
+    def translate_exception(e)
+      exception_translator.register_exception e
+    end
+
+    private
+
+    def exception_translator
+      ExceptionTranslator.new
+    end
+    memoize :exception_translator
+
+  end
+
+end

data/rake_tasks/cucumber.rake

@@ -1,7 +1,9 @@
 require 'cucumber/rake/task'
 
-Cucumber::Rake::Task.new(:features) do |t|
+Cucumber::Rake::Task.new 'test:features' do |t|
   t.fork = true
+  t.cucumber_opts = '--format progress'
 end
 
-task :cucumber => [:features]
+task 'test:cucumber' => ['test:features']
+task 'cucumber' => ['test:features']

data/rake_tasks/default.rake

@@ -0,0 +1 @@
+task :default => [:test]

data/rake_tasks/spec.rake

@@ -0,0 +1,3 @@
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new 'test:spec'
+task :spec => ['test:spec']

data/rake_tasks/test.rake

@@ -0,0 +1,2 @@
+desc 'Run all tests'
+task :test => ['test:spec', 'test:cucumber']

data/sandbox/em-server.rb

@@ -0,0 +1,37 @@
+#!/usr/bin/env ruby
+
+require 'eventmachine'
+
+ module EchoServer
+
+   def post_init
+     p 'post_init'
+   end
+
+   def connection_completed
+     p 'connection_completed'
+   end
+
+   def receive_data data
+     send_data ">>>you sent: #{data}"
+     close_connection if data =~ /quit/i
+   end
+
+   def unbind
+     p 'unbind'
+  end
+end
+
+def get_port_for_fd(fd)
+  sockname = EM.get_sockname(fd)
+  port, host = Socket.unpack_sockaddr_in(sockname)
+  port
+end
+
+# Note that this will block current thread.
+ EventMachine.run {
+   fd = EventMachine.start_server "127.0.0.1", 0, EchoServer do |e|
+     p e
+   end
+  p get_port_for_fd(fd)
+ }

data/spec/disk_file_system_spec.rb

@@ -0,0 +1,239 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+module Ftpd
+  describe DiskFileSystem do
+
+    let(:data_dir) {Ftpd::TempDir.make}
+    let(:disk_file_system) {DiskFileSystem.new(data_dir)}
+    let(:missing_file_error) do
+      [Ftpd::FileSystemError, /No such file or directory/]
+    end
+    let(:is_a_directory_error) do
+      [Ftpd::FileSystemError, /Is a directory/]
+    end
+    let(:missing_path) {'missing_path'}
+
+    def data_path(path)
+      File.join(data_dir, path)
+    end
+
+    def make_directory(path)
+      Dir.mkdir data_path(path)
+    end
+
+    def write_file(path)
+      File.open(data_path(path), 'wb') do |file|
+        file.write canned_contents(path)
+      end
+    end
+
+    def read_file(path)
+      File.open(data_path(path), 'rb', &:read)
+    end
+
+    def exists?(path)
+      File.exists?(data_path(path))
+    end
+
+    def canned_contents(path)
+      "Contents of #{path}"
+    end
+
+    before(:each) do
+      write_file 'file'
+      make_directory 'dir'
+      write_file 'dir/file_in_dir'
+      make_directory 'unwritable_dir'
+      write_file 'unwritable_dir/file'
+    end
+
+    describe '#accessible?' do
+
+      context '(within tree)' do
+        specify do
+          disk_file_system.accessible?('file').should be_true
+        end
+      end
+
+      context '(outside tree)' do
+        specify do
+          disk_file_system.accessible?('../outside').should be_false
+        end
+      end
+
+    end
+
+    describe '#exists?' do
+
+      context '(exists)' do
+        specify do
+          disk_file_system.exists?('file').should be_true
+        end
+      end
+
+      context '(does not exist)' do
+        specify do
+          disk_file_system.exists?('missing').should be_false
+        end
+      end
+
+    end
+
+    describe '#directory?' do
+
+      context '(directory)' do
+        specify do
+          disk_file_system.directory?('file').should be_false
+        end
+      end
+
+      context '(file)' do
+        specify do
+          disk_file_system.directory?('dir').should be_true
+        end
+      end
+
+    end
+
+    describe '#delete' do
+
+      context '(normal)' do
+        specify do
+          disk_file_system.delete('file')
+          exists?('file').should be_false
+        end
+      end
+
+      context '(file system error)' do
+        specify do
+          expect {
+            disk_file_system.delete(missing_path)
+          }.to raise_error *missing_file_error
+        end
+      end
+
+    end
+
+    describe '#read' do
+
+      context '(normal)' do
+        let(:path) {'file'}
+        specify do
+          disk_file_system.read(path).should == canned_contents(path)
+        end
+      end
+
+      context '(file system error)' do
+        specify do
+          expect {
+            disk_file_system.read(missing_path)
+          }.to raise_error *missing_file_error
+        end
+      end
+
+    end
+
+    describe '#write' do
+
+      let(:contents) {'file contents'}
+
+      context '(normal)' do
+        let(:path) {'file_path'}
+        specify do
+          disk_file_system.write(path, contents)
+          read_file(path).should == contents
+        end
+      end
+
+      context '(file system error)' do
+        specify do
+          expect {
+            disk_file_system.write('dir', contents)
+          }.to raise_error *is_a_directory_error
+        end
+      end
+
+    end
+
+    describe '#list_short' do
+
+      subject do
+        disk_file_system.list_short(path)
+      end
+
+      shared_examples 'returns short list of root' do
+        it {should =~ /^dir$/}
+        it {should =~ /^file$/}
+        it {should =~ /^unwritable_dir$/}
+        its('lines.to_a.size') {should == 3}
+      end
+
+      context '(root)' do
+        let(:path) {'/'}
+        it_behaves_like 'returns short list of root'
+      end
+
+      context '(empty path)' do
+        let(:path) {''}
+        it_behaves_like 'returns short list of root'
+      end
+
+      context '(specific file)' do
+        let(:path) {'/file'}
+        it {should == "file\n"}
+      end
+
+      context '(specific directory)' do
+        let(:path) {'/dir'}
+        it {should == "file_in_dir\n"}
+      end
+
+      context '(missing directory)' do
+        let(:path) {'/missing/file'}
+        it {should be_empty}
+      end
+
+    end
+
+    describe '#list_long' do
+
+      subject do
+        disk_file_system.list_long(path)
+      end
+
+      shared_examples 'returns long list of root' do
+        it {should =~ /^d.*dir$/}
+        it {should =~ /^-.*file$/}
+        it {should =~ /^d.*unwritable_dir$/}
+        its('lines.to_a.size') {should == 3}
+      end
+
+      context '(root)' do
+        let(:path) {'/'}
+        it_behaves_like 'returns long list of root'
+      end
+
+      context '(empty path)' do
+        let(:path) {''}
+        it_behaves_like 'returns long list of root'
+      end
+
+      context '(specific file)' do
+        let(:path) {'/file'}
+        it {should =~ /^-.*file$/}
+      end
+
+      context '(specific directory)' do
+        let(:path) {'/dir'}
+        it {should =~ /^-.*file_in_dir\n/}
+      end
+
+      context '(missing directory)' do
+        let(:path) {'/missing/file'}
+        it {should be_empty}
+      end
+
+    end
+
+  end
+end