tarbit 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5fdb81b5d9d98958d081c9fb3fdc25ec216bd0b9b8d1ceb1284eb3e898482d9a
-  data.tar.gz: 5ca27088bb0c658716d09edb4236227f2ef298654eb9c166b86ec4261ffb8462
+  metadata.gz: 94cb32b706a9f0bfd2a036dbc831aa6702b4a9795d212faa9b0c55ae9038e43c
+  data.tar.gz: 4da4a50fea4bca2a4e90bf5397cc5c302380f8623222223361c2b227401ddd43
 SHA512:
-  metadata.gz: b47d842599e19cedeb839f73ae732d85bd9059cba24dae157651ca8dd4ca802154849086b2d62b37c8a726091b95481fca1828d71fd825dc0df021a753ae2509
-  data.tar.gz: 02632a2259b571ced4bccf13c6d051ea83e3e9381252a1d2025ccfc1301e818bf92ec3f7a7b1693737bcb14147e6f11381a3bf092678be13a4cfe3907ea27a0a
+  metadata.gz: 17847c8874fe99434b8c9b1e47aaac142f506c2f2c97b5e24c9870052bdbc89751197d6e8a12f2b953a6a2a6ff35bf2032f9117a7503c69ca3919254f118c289
+  data.tar.gz: c8219d0ad15c5ece548d626a6049daf64d6aab085e2eb2a0780c2047c9794c4d83f8cfd8c7e5099f3795f29545da4f5c46d08839d20c9b5305b55e1730fe83ad

data/.gitignore

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

data/bin/tarbit

@@ -2,6 +2,7 @@
 
 require 'commander/import'
 require 'tarbit'
+require 'fileutils'
 
 # :name is optional, otherwise uses the basename of this executable
 program :name, 'Tarbit - SSH Tarpit using Ruby'
@@ -14,10 +15,12 @@ command :serve do |c|
   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|
+
+    FileUtils.mkdir_p File.expand_path ('~/.tarbit/stats/')
     puts "Starting tarbit ssh tarpit"
 
     server = Tarbit::Server.new
-    statistic = Tarbit::Statistic.new(server, 10)
+    statistic = Tarbit::Statistic.new(server, 600)
 
     Async do |task|
         statistic.watch

data/lib/tarbit/server.rb

@@ -30,17 +30,22 @@ module Tarbit
           stream = Async::IO::Stream.new(peer)
           Async.logger.info "New connection: #{stream}"
 
+          id = SecureRandom.uuid
+
           @connections << {
               created_at: Date.new,
-              id: SecureRandom.uuid
+              id: id
           }
 
           while true do
             task.sleep 1
+            if stream.eof?
+              raise Async::TimeoutError.new
+            end
             stream.write "#{rand(10)}\r\n"
           end
         rescue Async::TimeoutError => e
-          @connections.delete stream
+          @connections = @connections.reject { |stats| stats.fetch(:id) == id }
           Async.logger.info "Connection closed: #{stream}"
         end
       end

data/lib/tarbit/statistic.rb

@@ -22,6 +22,7 @@ module Tarbit
     private
 
     def write_line_chart
+      return if @server.connections.size == 0 and @history.size == 0
 
       # Add point in time
       @history << {
@@ -38,7 +39,7 @@ module Tarbit
 
       g.data :Bots, @history.map {|point_in_time| point_in_time.fetch(:connections).size }
 
-      g.write('data/exciting.png')
+      g.write(File.expand_path ('~/.tarbit/stats/line_chart.png'))
     end
 
   end

data/lib/tarbit/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tarbit
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Niklas Hanft