kscript 1.0.1 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a72c639d7028f3032a3ab8d6508308cd174a1174f698d5099f86356b1bef9f31
-  data.tar.gz: 898a2845e93a58bb1760a92850b091b84fcd374d92b138c13022e85dc6868877
+  metadata.gz: 01ab48d4e8819e3190707f87bbc3fa8a98df3046c278a6c6a625c8eca25d4585
+  data.tar.gz: 15970ccc55e4900e98d0c99a6e3ebb4ebe9555891e63393ec31459c4847d5ab9
 SHA512:
-  metadata.gz: 869a3dfa24680f0c6e95cbdb82f0d8e6f981f5868d5f5c30b0183b7d2490b64ee2ce670e08190f095eea1291f26c31293e215759ee9545eab2a1d602b36a9bba
-  data.tar.gz: 5d24ef70c21851e42b337fd8e0222d50bb9f0def1e61a0121e2edaac8eb5fd16d4233adb8560e428eada8ee5e312a2c717f5ce30826c93891988861a38ee6260
+  metadata.gz: 0e45e15a0732f678f90db60384458c542f8d038460a801a60f986c197f975321fbf8ca88e39902f5fd2d1d8c2b39b8f4adae55522253824f8cf9a5afff947a83
+  data.tar.gz: 4b0f0f5c5be88b4b248ad9e5d5a8ccca3c0f9a267da5a664afe426917881f16ec173b09e3e287d44fbe9a9aa86aa74ef651c96abaae4b34e77ab1c6b40a85ed1

data/lib/kscript/cli.rb

@@ -1,9 +1,7 @@
 # frozen_string_literal: true
 
 require 'kscript'
-require 'io/console'
 require 'kscript/banner'
-require 'thor'
 
 module Kscript
   class CLI < Thor

data/lib/kscript/logger.rb

@@ -55,7 +55,7 @@ module Kscript
       @logger.send(level, entry.to_json)
     end
 
-    # 极客风格终端输出
+    # 终端输出
     def klog(level, message, context = {})
       if human_output?
         puts "[#{level.to_s.upcase}] #{message} #{context.map { |k, v| "#{k}=#{v}" }.join(' ')}".strip

data/lib/kscript/plugins/kk_es_fingerprint_utils.rb

@@ -15,8 +15,9 @@ module Kscript
 
     # Initialize with certificate path
     # @param cert_path [String] path to the certificate file
-    def initialize(cert_path = DEFAULT_CERT_PATH)
-      @cert_path = cert_path
+    def initialize(cert_path = nil, *_args, **opts)
+      super(**opts.merge(service: 'kk_es_fingerprint'))
+      @cert_path = cert_path || self.class::DEFAULT_CERT_PATH
     end
 
     def run

data/lib/kscript/plugins/kk_ffmpeg_utils.rb

@@ -9,6 +9,10 @@ require 'kscript'
 
 module Kscript
   class KkFfmpegUtils < Base
+    def initialize(*_args, **opts)
+      super(**opts.merge(service: 'kk_ffmpeg'))
+    end
+
     def run
       with_error_handling do
         install

data/lib/kscript/plugins/kk_ip_utils.rb

@@ -7,9 +7,6 @@
 
 require 'kscript'
 
-require 'http'
-require 'json'
-
 module Kscript
   class KkIpUtils < Base
     IP_API_BASE_URL = 'http://ip-api.com/json'

data/lib/kscript/plugins/kk_jenkins_utils.rb

@@ -6,12 +6,6 @@
 # https://opensource.org/licenses/MIT
 
 require 'kscript'
-require 'http'
-require 'base64'
-require 'rexml/document'
-require 'json'
-require 'fileutils'
-require 'kscript/base'
 
 module Kscript
   class KkJenkinsUtils < Base
@@ -21,7 +15,9 @@ module Kscript
       end
     end
 
-    def initialize(jenkins_url, user, password)
+    def initialize(*args, **opts)
+      super(**opts.merge(service: 'kk_jenkins'))
+      jenkins_url, user, password = args
       @jenkins_url = jenkins_url
       @user = user
       @password = password

data/lib/kscript/plugins/kk_kibana_utils.rb

@@ -6,7 +6,6 @@
 # https://opensource.org/licenses/MIT
 
 require 'kscript'
-require 'securerandom'
 
 module Kscript
   class KkKibanaUtils < Base
@@ -16,7 +15,9 @@ module Kscript
       end
     end
 
-    def initialize(project_name, project_env, base_url, username, password)
+    def initialize(*args, **opts)
+      super(**opts.merge(service: 'kk_kibana'))
+      project_name, project_env, base_url, username, password = args
       @base_url = base_url
       @username = username
       @password = password

data/lib/kscript/plugins/kk_lvm_utils.rb

@@ -20,7 +20,8 @@ module Kscript
 
     # Initialize the LVM mounter with configuration
     # @param config [Hash] configuration options
-    def initialize(config = {}, *_args, **_opts)
+    def initialize(config = {}, *_args, **opts)
+      super(**opts.merge(service: 'kk_lvm'))
       @config = DEFAULT_CONFIG.merge(config)
     end
 

data/lib/kscript/plugins/kk_optimize_utils.rb

@@ -6,7 +6,6 @@
 # https://opensource.org/licenses/MIT
 
 require 'kscript'
-require 'open3'
 
 module Kscript
   class KkOptimizeUtils < Base

data/lib/kscript/plugins/kk_portscan_utils.rb

@@ -6,8 +6,6 @@
 # https://opensource.org/licenses/MIT
 
 require 'kscript'
-require 'socket'
-require 'timeout'
 
 module Kscript
   class KkPortscanUtils < Base
@@ -17,10 +15,11 @@ module Kscript
     # @param host [String] target host to scan
     # @param ports [Array<Integer>] list of ports to scan
     # @param thread_count [Integer] number of concurrent threads
-    def initialize(target = nil, ports = (1..1024), *_args, **opts)
+    def initialize(target = nil, ports = (1..1024), thread_count = 50, *_args, **opts)
       super(**opts.merge(service: 'kk_port_scanner'))
       @target = target
-      @ports = ports.is_a?(Range) ? ports : (1..1024)
+      @ports = parse_ports(ports)
+      @thread_count = (opts[:thread_count] || thread_count || 50).to_i
     end
 
     def run
@@ -31,23 +30,62 @@ module Kscript
 
     # Execute port scanning using multiple threads
     def scan
-      msg = "Scanning #{@target} ports #{@ports}"
+      msg = "Scanning #{@target} ports #{@ports} with concurrency=#{@thread_count}"
       if human_output?
         puts msg
       else
         logger.kinfo(msg)
       end
-      @ports.each do |port|
-        Socket.tcp(@target, port, connect_timeout: 0.5) do |_sock|
-          if human_output?
-            puts "Port #{port} is open"
-          else
-            logger.kinfo('Port open', port: port)
-            logger.kinfo("Port #{port} is open")
+      queue = Queue.new
+      @ports.each { |port| queue << port }
+      threads = []
+      @thread_count.times do
+        threads << Thread.new do
+          until queue.empty?
+            port = nil
+            begin
+              port = queue.pop(true)
+            rescue ThreadError
+              break
+            end
+            begin
+              Socket.tcp(@target, port, connect_timeout: 0.5) do |_sock|
+                if human_output?
+                  puts "Port #{port} is open"
+                else
+                  logger.kinfo('Port open', port: port)
+                  logger.kinfo("Port #{port} is open")
+                end
+              end
+            rescue Errno::ECONNREFUSED, Errno::ETIMEDOUT, SocketError
+              # closed or filtered
+            end
           end
         end
-      rescue Errno::ECONNREFUSED, Errno::ETIMEDOUT, SocketError
-        # closed or filtered
+      end
+      threads.each(&:join)
+    end
+
+    # 支持多种端口参数格式: 22,80,443 或 1..1024
+    def parse_ports(ports)
+      return ports.to_a if ports.is_a?(Range)
+
+      if ports.is_a?(String)
+        if ports.include?(',')
+          ports.split(',').map(&:to_i)
+        elsif ports.include?('..')
+          begin
+            eval(ports).to_a
+          rescue StandardError
+            (1..1024).to_a
+          end
+        else
+          [ports.to_i]
+        end
+      elsif ports.is_a?(Array)
+        ports.map(&:to_i)
+      else
+        (1..1024).to_a
       end
     end
 
@@ -56,11 +94,11 @@ module Kscript
     end
 
     def self.arguments
-      '<target_host>'
+      '<target_host> [ports] [thread_count]'
     end
 
     def self.usage
-      "kscript portscan 192.168.1.1\nkscript portscan example.com --ports=22,80,443"
+      "kscript portscan 192.168.1.1\nkscript portscan example.com 22,80,443 100\nkscript portscan 192.168.1.1 1..1024 200"
     end
 
     def self.group

data/lib/kscript/plugins/kk_projscan_utils.rb

@@ -6,7 +6,6 @@
 # https://opensource.org/licenses/MIT
 
 require 'kscript'
-require 'json'
 
 module Kscript
   class KkProjscanUtils < Base

data/lib/kscript/plugins/kk_sh_utils.rb

@@ -6,7 +6,6 @@
 # https://opensource.org/licenses/MIT
 
 require 'kscript'
-require 'nokogiri'
 
 module Kscript
   class KkShUtils < Base

data/lib/kscript/plugins/kk_syscheck_utils.rb

@@ -9,6 +9,10 @@ require 'kscript'
 
 module Kscript
   class KkSyscheckUtils < Base
+    def initialize(*_args, **opts)
+      super(**opts.merge(service: 'kk_syscheck'))
+    end
+
     def run
       with_error_handling do
         check

data/lib/kscript/plugins/kk_wg_pass_utils.rb

@@ -6,7 +6,6 @@
 # https://opensource.org/licenses/MIT
 
 require 'kscript'
-require 'bcrypt'
 
 module Kscript
   class KkWgPassUtils < Base

data/lib/kscript/utils.rb

@@ -1,8 +1,5 @@
 # frozen_string_literal: true
 
-require 'yaml'
-require 'fileutils'
-
 module Kscript
   module Utils
     class Config

data/lib/kscript/version.rb

@@ -6,5 +6,5 @@
 # https://opensource.org/licenses/MIT
 
 module Kscript
-  VERSION = '1.0.1'
+  VERSION = '1.0.3'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: kscript
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.3
 platform: ruby
 authors:
 - Kk