tarbit 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 5fdb81b5d9d98958d081c9fb3fdc25ec216bd0b9b8d1ceb1284eb3e898482d9a
+  data.tar.gz: 5ca27088bb0c658716d09edb4236227f2ef298654eb9c166b86ec4261ffb8462
+SHA512:
+  metadata.gz: b47d842599e19cedeb839f73ae732d85bd9059cba24dae157651ca8dd4ca802154849086b2d62b37c8a726091b95481fca1828d71fd825dc0df021a753ae2509
+  data.tar.gz: 02632a2259b571ced4bccf13c6d051ea83e3e9381252a1d2025ccfc1301e818bf92ec3f7a7b1693737bcb14147e6f11381a3bf092678be13a4cfe3907ea27a0a

data/.gitignore

@@ -0,0 +1,3 @@
+.idea/
+.DS_Store
+data/*.png

data/Gemfile

@@ -0,0 +1,5 @@
+source 'https://rubygems.org'
+
+ruby "2.6.5"
+
+gemspec

data/Gemfile.lock

@@ -0,0 +1,38 @@
+PATH
+  remote: .
+  specs:
+    tarbit (1.0.0)
+      async-io
+      commander
+      gruff
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    async (1.24.2)
+      console (~> 1.0)
+      nio4r (~> 2.3)
+      timers (~> 4.1)
+    async-io (1.27.3)
+      async (~> 1.14)
+    commander (4.5.0)
+      highline (~> 2.0.0)
+    console (1.8.1)
+    gruff (0.7.0)
+      rmagick (~> 2.13, >= 2.13.4)
+    highline (2.0.3)
+    nio4r (2.5.2)
+    rmagick (2.16.0)
+    timers (4.3.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  tarbit!
+
+RUBY VERSION
+   ruby 2.6.5p114
+
+BUNDLED WITH
+   1.17.3

data/bin/tarbit

@@ -0,0 +1,28 @@
+#!/usr/bin/env ruby
+
+require 'commander/import'
+require 'tarbit'
+
+# :name is optional, otherwise uses the basename of this executable
+program :name, 'Tarbit - SSH Tarpit using Ruby'
+program :version, Tarbit::VERSION
+program :description, 'Stupid command that prints foo or bar.'
+
+command :serve do |c|
+  c.syntax = 'apollo serve [options]'
+  c.description = 'Runs the apollo server'
+  c.option '--debug', nil, 'Runs the apollo server in debug mode'
+  c.option '--suffix STRING', String, 'Adds a suffix to bar'
+  c.action do |args, options|
+    puts "Starting tarbit ssh tarpit"
+
+    server = Tarbit::Server.new
+    statistic = Tarbit::Statistic.new(server, 10)
+
+    Async do |task|
+        statistic.watch
+        server.run
+    end
+
+  end
+end

data/lib/tarbit/server.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+require 'async/io'
+require 'async/io/stream'
+require 'async/reactor'
+require 'async/io/host_endpoint'
+require 'async/io/protocol/line'
+require 'securerandom'
+
+module Tarbit
+  class Server
+    attr_reader :connections
+
+    def initialize
+      @connections = []
+    end
+
+    def run
+      endpoint = Async::IO::Endpoint.parse("tcp://localhost:1025")
+
+      Async do |task|
+        while true
+          task.sleep 1
+          Async.logger.info "Connection count: #{@connections.size}"
+        end
+      end
+
+      Async do |task|
+        endpoint.accept do |peer|
+          stream = Async::IO::Stream.new(peer)
+          Async.logger.info "New connection: #{stream}"
+
+          @connections << {
+              created_at: Date.new,
+              id: SecureRandom.uuid
+          }
+
+          while true do
+            task.sleep 1
+            stream.write "#{rand(10)}\r\n"
+          end
+        rescue Async::TimeoutError => e
+          @connections.delete stream
+          Async.logger.info "Connection closed: #{stream}"
+        end
+      end
+    end
+
+  end
+
+end

data/lib/tarbit/statistic.rb

@@ -0,0 +1,45 @@
+require 'gruff'
+require 'async'
+
+module Tarbit
+  class Statistic
+
+    def initialize(server, interval)
+      @server = server
+      @interval = interval
+      @history = []
+    end
+
+    def watch
+      Async do |task|
+        while true
+          task.sleep @interval
+          write_line_chart
+        end
+      end
+    end
+
+    private
+
+    def write_line_chart
+
+      # Add point in time
+      @history << {
+          created_at: Date.new.strftime("%B %d, %Y"),
+          connections: @server.connections.clone # Cloning instead of referencing
+      }
+
+      g = Gruff::Line.new
+      g.title = 'History of connections over time'
+
+      labels = {}
+      @history.each_with_index{ |item, index| labels[index] = item.fetch(:created_at) }
+      g.labels = labels
+
+      g.data :Bots, @history.map {|point_in_time| point_in_time.fetch(:connections).size }
+
+      g.write('data/exciting.png')
+    end
+
+  end
+end

data/lib/tarbit/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Tarbit
+  VERSION = "1.0.0"
+end

data/lib/tarbit.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+#
+
+require 'async/reactor'
+require_relative 'tarbit/server'
+require_relative 'tarbit/statistic'
+
+Signal.trap "SIGINT" do
+  exit(0)
+end
+
+module Tarbit; end

data/tarbit.gemspec

@@ -0,0 +1,30 @@
+lib = File.expand_path("lib", __dir__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "tarbit/version"
+
+Gem::Specification.new do |s|
+
+  s.name = 'tarbit'
+  s.version = Tarbit::VERSION
+
+  s.summary = "Summary"
+  s.description = "Description"
+
+  s.authors = ['Niklas Hanft']
+  s.email = 'hello@niklashanft.com'
+  s.homepage = 'https://github.com/nhh/apollo'
+  s.license = 'ISC'
+
+  s.files = `git ls-files`.split("\n")
+  s.executables << 'tarbit'
+  s.required_ruby_version = '~> 2.5'
+  s.require_path = 'lib'
+
+  # Dependencies
+  s.add_dependency 'async-io'
+  s.add_dependency 'commander'
+  s.add_dependency 'gruff'
+
+  # Development Dependencies
+
+end

metadata

@@ -0,0 +1,94 @@
+--- !ruby/object:Gem::Specification
+name: tarbit
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Niklas Hanft
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-02-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: async-io
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: commander
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: gruff
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Description
+email: hello@niklashanft.com
+executables:
+- tarbit
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- bin/tarbit
+- lib/tarbit.rb
+- lib/tarbit/server.rb
+- lib/tarbit/statistic.rb
+- lib/tarbit/version.rb
+- tarbit.gemspec
+homepage: https://github.com/nhh/apollo
+licenses:
+- ISC
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - "~>"
+    - !ruby/object:Gem::Version
+      version: '2.5'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.6
+signing_key: 
+specification_version: 4
+summary: Summary
+test_files: []