enviroblyd 0.4.5 → 0.4.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 794c9805c66ea61e73486ccc08f71077daa2e2de80e6f9c5d510ff62f038f9f5
-  data.tar.gz: 40f92e9aafc2351298c65f71ee251eb766a31abc0835ac3170ceb257ba854bfe
+  metadata.gz: a8467f960365419e9f02f14d3d3186a511787dff28798fd8883ac328a2443e9c
+  data.tar.gz: 2f39dbca93e49cf69ab2bb9aaed76fe89ed2ebd7867baebba640c8ea665082ae
 SHA512:
-  metadata.gz: 4d13d6792c28d62cd0ccf872e2368dc9396523cf59d3b67b0f471e8315dad6c66a3a7f8be3d507d2e7d8ef1bae7df06c1397b587a2b56de757426d448693d4be
-  data.tar.gz: 8e0641adc0ef80bac699bd52702cbccb2ee7cdbe3d5b2d41d19f7f1e3baa27e10c6423d57a0a571d61c9f22c9f7bb088d33b8014f38067831d5bd1083b4d24b6
+  metadata.gz: dc3ea42ebc9125f88ec1828b4ebdf9032ba2482d1db359860b9b7c117ae0cb12eb57369531bd760cb54b2563f64c63e3afb89a304a574c23fa19bc7ff4334214
+  data.tar.gz: 3c9baed9045d458595627b186f5be800abe5da2df36beea47e22fd478bc28e25e984814de9731118040b4a65f16fabea10923d06daaa0f419639ace81f0be67f

data/lib/enviroblyd/command.rb

@@ -1,23 +1,26 @@
 # frozen_string_literal: true
 
 require "open3"
+require "json"
 
 class Enviroblyd::Command
   DEFAULT_TIMEOUT_SECONDS = 5 * 60
   DEFAULT_RUNTIME = "/bin/bash"
 
-  def self.run(params)
-    Thread.new do
-      new(params).run
-    end
-  end
-
-  def initialize(params)
+  def initialize(message)
+    params = parse_message message
     @url = params.fetch "url"
     @script = params.fetch "script"
     @runtime = params.fetch "runtime", DEFAULT_RUNTIME
     @timeout = params.fetch "timeout", DEFAULT_TIMEOUT_SECONDS
     @stdout = @stderr = @exit_code = nil
+    @valid = true
+  rescue NoMethodError, KeyError
+    @valid = false
+  end
+
+  def valid?
+    @valid
   end
 
   def run
@@ -45,4 +48,10 @@ class Enviroblyd::Command
         }
       }
     end
+
+    def parse_message(message)
+      JSON.parse message
+    rescue
+      nil
+    end
 end

data/lib/enviroblyd/daemon.rb

@@ -1,43 +1,47 @@
 # frozen_string_literal: true
 
 require "socket"
-require "json"
 
 class Enviroblyd::Daemon
   MAX_MESSAGE_SIZE = 6000 # bytes
-  LISTEN_PORT = ENV.fetch("ENVIROBLYD_PORT", 63106).to_i
+  PORT = ENV.fetch("ENVIROBLYD_PORT", 63106).to_i
 
   def self.start
     imds = Enviroblyd::IMDS.new
     host = imds.private_ipv4
-    new.listen(host) do
-      Enviroblyd::Web.register
-    end
+    daemon = new(host)
+    daemon.listen
   end
 
-  def listen(host)
-    server = TCPServer.new host, LISTEN_PORT
-    puts "Listening on #{host}:#{LISTEN_PORT}"
+  def initialize(host)
+    @host = host
+  end
 
-    yield
+  def listen
+    server = TCPServer.new @host, PORT
+    puts "Listening on #{@host}:#{PORT}"
+    Enviroblyd::Web.register
 
     loop do
       Thread.start(server.accept) do |client|
-        params =
-          begin
-            JSON.parse client.recv(MAX_MESSAGE_SIZE)
-          rescue
-            nil
-          end
-
-        if params.nil?
-          client.puts "Error parsing JSON"
-        else
-          client.puts "OK"
+        message = client.recv(MAX_MESSAGE_SIZE)
+        command = Enviroblyd::Command.new message
+
+        unless command.valid?
+          $stderr.puts "Invalid message received: #{message}"
+          client.puts "Invalid message"
+          next
         end
 
+        client.puts "OK"
+        client.close
+        command.run
+      ensure
         client.close
-        Enviroblyd::Command.run(params) if params
+        command = nil
+        message = nil
+        client = nil
+        GC.start
       end
     end
   end

data/lib/enviroblyd/imds.rb

@@ -6,7 +6,8 @@ class Enviroblyd::IMDS
 
   def initialize
     @token = Enviroblyd::Web.http("http://#{IMDS_HOST}/latest/api/token",
-      type: Net::HTTP::Put, headers: { "X-aws-ec2-metadata-token-ttl-seconds" => TOKEN_TTL_SECONDS.to_s }).
+      type: Net::HTTP::Put,
+      headers: { "X-aws-ec2-metadata-token-ttl-seconds" => TOKEN_TTL_SECONDS.to_s }).
       body.strip
   end
 

data/lib/enviroblyd/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Enviroblyd
-  VERSION = "0.4.5"
+  VERSION = "0.4.6"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: enviroblyd
 version: !ruby/object:Gem::Version
-  version: 0.4.5
+  version: 0.4.6
 platform: ruby
 authors:
 - Robert Starsi
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-06 00:00:00.000000000 Z
+date: 2024-09-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor