net-knocker 0.0.1 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7d734c93c4e362b454bace35ef5f0e46efd227139cdafdb3c19c030a98961f85
-  data.tar.gz: 53de1160612bbd364785b5e97ea84c343eeb27080621fbb0702bc13b0f3329fc
+  metadata.gz: d2b9e2759985d6e60c6ff108157155b7862d703c2e1836284f89f2786e58d199
+  data.tar.gz: dce4cd990ad60f354628f15e83bb2fe9cadfd7d1bb1ea973e7211447711107a7
 SHA512:
-  metadata.gz: c611fc7c1d7eaba3e86db70a4c386e98f8d394d1909d0ebee5f1972149384cf723e94470a20eea622ecb1083fabe588a3fed093ab841136569ec5d06d672d0db
-  data.tar.gz: f8b6973f49d8d1152a4adee0d71c42e5c70c40adfb1fac8ab745695439c1bbd6c9cc047346a47530a811c77bae9c9dfbd859a9212d0a06a9ea53e552bb156a69
+  metadata.gz: 43ae20154a32b3a730952bec81a0b0c5a2159c7038b9d1276e4ff11fbf8edbcfb07ae8b6bfc6021c0d892ec47b797a6db14f2a13a7f496fe221c2bcc5540114b
+  data.tar.gz: 3b3783e2c1d6a7d7b842f6e93f07ad4b0ca2dbe1e759b94e39ea7942c085d3f2b35a7532bbbff77ee8f74bd6fd24b64dfd6bb74ee242089d4ac4337a94f642fa

data/.pryrc

@@ -1 +1,3 @@
+# frozen_string_literal: true
+
 require_relative 'lib/net/knocker'

data/.rubocop.yml

@@ -1,5 +1,5 @@
 AllCops:
-  TargetRubyVersion: 2.4
+  TargetRubyVersion: 2.5
   NewCops: enable
 
 Style/StringLiterals:
@@ -13,3 +13,6 @@ Style/Documentation:
 
 Layout/LineLength:
   Max: 120
+
+Metrics/MethodLength:
+  Max: 20

data/lib/net/knocker/env.rb

@@ -0,0 +1,79 @@
+# frozen_string_literal: true
+
+require 'logger'
+require 'net/https'
+require 'uri'
+
+module Net
+  class Knocker
+    module Env
+      def env_secret
+        @env_secret ||= ENV.fetch('NET_KNOCKER_SECRET')
+      end
+
+      def http_client
+        @http_client ||= Net::HTTP.new(url.host, url.port).tap do |http|
+          http.use_ssl = true
+          http.cert = ssl_certificate
+          http.key = ssl_key
+          http.set_debug_output $stderr if ENV.key? 'NET_KNOCKER_HTTP_DEBUG'
+        end
+      end
+
+      def should_omit?
+        @should_omit ||= (%w[NET_KNOCKER_URL NET_KNOCKER_SECRET] - ENV.keys).any?
+      end
+
+      def ssl_key
+        @ssl_key ||= begin
+          key_class = case ssl_certificate.signature_algorithm
+                      when /^ecdsa/
+                        OpenSSL::PKey::EC
+                      when /^rsa/
+                        OpenSSL::PKey::RSA
+                      when /^dsa/
+                        OpenSSL::PKey::DSA
+                      else
+                        raise "Unknown key type: #{ssl_certificate.signature_algorithm}"
+                      end
+          key_class.new(env_secret)
+        end
+      end
+
+      def info(method = :info, msg)
+        tags = [self.class.name, Process.pid, method.to_s.upcase].join('][')
+        log.send(method, "[#{tags}] #{msg}")
+      end
+
+      def error(msg)
+        info(__method__, msg)
+      end
+
+      def warn(msg)
+        info(__method__, msg)
+      end
+
+      def debug(msg)
+        info(__method__, msg)
+      end
+
+      def ssl_certificate
+        @ssl_certificate ||= OpenSSL::X509::Certificate.new(env_secret)
+      end
+
+      def url
+        @url ||= URI.parse(ENV.fetch('NET_KNOCKER_URL')) unless should_omit?
+      end
+
+      def log
+        @log  ||= if (defined? logger) && (logger.respond_to? :debug)
+                    logger
+                  elsif (defined? Rails.logger) && (Rails.logger.respond_to? :debug)
+                    Rails.logger
+                  else
+                    ::Logger.new($stderr)
+                  end
+      end
+    end
+  end
+end

data/lib/net/knocker/peer.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+module Net
+  class Knocker
+    class Peer
+      def self.env_var_patterns
+        @env_var_patterns ||= %W[NET_KNOCKER_URL NET_KNOCKER_SECRET #{ENV['NET_KNOCKER_URL']}].freeze
+      end
+
+      # filter out /proc/self/environ, /proc/thread-self/environ, /proc/<PID>/environ" and "/proc/1/environ"
+      def self.self_patterns
+        @self_patterns ||= %r{proc/((?:thread-)?self|#{Process.pid}|1)/environ}.freeze
+      end
+
+      def self.environ_paths
+        Dir['/proc/*/environ'].grep_v(self_patterns)
+                              .map { |p| Pathname.new(p) }
+                              .compact
+      end
+
+      def self.pid_with_content
+        environ_paths.map do |path|
+          content = begin
+            path.binread
+                .split("\0")
+                .select { |e| env_var_patterns.any? { |p| e.include? p } }
+                .compact
+          rescue StandardError
+            next
+          end
+          pid = path.to_s.split('/').third.to_s
+          [pid, content] unless content.empty? || pid.empty?
+        end
+        .compact
+        .to_h
+      end
+
+      def self.any?(&block)
+        pid_with_content.any?(&block)
+      end
+    end
+  end
+end

data/lib/net/knocker/version.rb

@@ -2,6 +2,6 @@
 
 module Net
   class Knocker
-    VERSION = '0.0.1'
+    VERSION = '0.0.4'
   end
 end

data/lib/net/knocker.rb

@@ -1,89 +1,45 @@
 # frozen_string_literal: true
+
 require_relative './knocker/version'
-require 'logger'
-require 'net/https'
-require 'uri'
+require_relative './knocker/env'
+require_relative './knocker/peer'
 
 module Net
   class Knocker
+    include Net::Knocker::Env
+
     def self.check
       new
     end
 
-    def self.skip_checkin?
-      new.skip_checkin?
-    end
-
-    def checkin_uri
-      @checkin_uri ||= URI.parse(ENV.fetch('NET_KNOCKER_URL'))
-    end
-
-    def log
-      @log ||= if defined? logger
-                  logger
-                elsif defined? Rails.logger
-                Rails.logger
-                else
-                  ::Logger.new($stderr)
-                end
+    def self.should_omit?
+      new.should_omit?
     end
 
     def in
-      return false if skip_checkin?
+      send_request 'PUT'
+    end
 
-      check_connection 'PUT'
+    def msg_end
+      "to #{url}" unless url.to_s.empty?
     end
 
     def out
-      return false if skip_checkin?
+      return debug("not sending DELETE request #{msg_end})") if should_omit?
+      return info("sending DELETE request is blocked by PID(s) #{Peer.pid_with_content.keys.join(',')}") if Peer.any?
 
-      check_connection 'DELETE'
-    end
-
-    def skip_checkin?
-      @skip_checkin ||= %w[NET_KNOCKER_URL NET_KNOCKER_SECRET].all? { |k| !(ENV.key? k) }
+      send_request 'DELETE'
     end
 
     private
 
-    def check_connection(http_method)
-      http_client.send_request(http_method, '/') do |response|
-        log.debug("[#{self.class}][#{__method__}] Public IP: #{response.body} ")
-      end
-    end
-
-    def env_secret
-      @env_secret ||= ENV.fetch('NET_KNOCKER_SECRET')
-    end
-
-    def http_client
-      @http_client ||= Net::HTTP.new(checkin_uri.host, checkin_uri.port).tap do |http|
-        http.use_ssl = true
-        http.cert = ssl_certificate
-        http.key = ssl_key
-        http.set_debug_output $stderr if ENV.key? 'NET_KNOCKER_HTTP_DEBUG'
-      end
-    end
+    def send_request(http_method)
+      return debug("not sending #{http_method} request #{msg_end}") if should_omit?
 
-    def ssl_key
-      @ssl_key ||= begin
-        key_class = case ssl_certificate.signature_algorithm
-                    when /^ecdsa/
-                      OpenSSL::PKey::EC
-                    when /^rsa/
-                      OpenSSL::PKey::RSA
-                    when /^dsa/
-                      OpenSSL::PKey::DSA
-                    else
-                      raise "Unknown key type: #{ssl_certificate.signature_algorithm}"
-                    end
-        key_class.new(env_secret)
+      http_client.send_request(http_method, '/') do |response|
+        debug("[#{self.class}][#{__method__}] Public IP: #{response.body} ")
       end
     end
-
-    def ssl_certificate
-      @ssl_certificate ||= OpenSSL::X509::Certificate.new(env_secret)
-    end
   end
 end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: net-knocker
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.4
 platform: ruby
 authors:
 - Tio Teath
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-04-10 00:00:00.000000000 Z
+date: 2022-04-13 00:00:00.000000000 Z
 dependencies: []
 description: 
 email:
@@ -31,6 +31,8 @@ files:
 - bin/console
 - bin/setup
 - lib/net/knocker.rb
+- lib/net/knocker/env.rb
+- lib/net/knocker/peer.rb
 - lib/net/knocker/version.rb
 - net-knocker.gemspec
 homepage: https://github.com/todosmodos/net-knocker