girl 9.1.0 → 9.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ddce98f59f1102d50d58fd02775ed1305f3babb3a2b7998a4ed86f39438e2599
-  data.tar.gz: b51d851640addd0acb8d5367b9d6295a67036e720436c6d0bea9ed352222f9bc
+  metadata.gz: 673184bfb0785b65e56e4715c2b007481ae91d16620fbc11e3f6e12c50f511b5
+  data.tar.gz: 9a37c381bbf266cac12239b479214979202a8b715a1db0c5f443f8ecd3ca046a
 SHA512:
-  metadata.gz: 469527cb74ea0a0aab7af6358dcd345852c8c9c25b9adbcb69f70acaa582592b696a2549d92e3970d55b0a607fe400f107c8445e075fffb6c22bc4d82837289a
-  data.tar.gz: 1ce78e2d003752fb847c11ec505960c82a241d4d0b8ebde1047e7ac788532bb35ad6d985cf76067e20c638a788736336a35a2381b146664d4c391a5e3441d39f
+  metadata.gz: d291af3cfc1ff8813e187062a55b93ad9f0c6a52f7bd14e19c4638becd98c96451da21146a167111c5da01d2f20b6fc31711143e71d82e82bf5cf68d8e420efb
+  data.tar.gz: 9a6e13e0f756afd3a024f7c7c386fd18e5978488bd1909fd69dbabffacd25e71d8d0c286920ec86665ed69766cee1464ba1d4a8fcab94055fdf82d405392d96b

data/lib/girl/proxy.rb

@@ -23,8 +23,12 @@ module Girl
 
       conf = JSON.parse( IO.binread( config_path ), symbolize_names: true )
       puts "load #{ config_path } #{ conf.inspect }"
+      redir_host = conf[ :redir_host ]
       redir_port = conf[ :redir_port ]
       memd_port = conf[ :memd_port ]
+      relayd_host = conf[ :relayd_host ]
+      relayd_port = conf[ :relayd_port ]
+      tspd_host = conf[ :tspd_host ]
       tspd_port = conf[ :tspd_port ]
       proxyd_host = conf[ :proxyd_host ]
       proxyd_port = conf[ :proxyd_port ]
@@ -54,7 +58,6 @@ module Girl
       h_src_underhalf = conf[ :h_src_underhalf ] # V
       h_dst_overflow = conf[ :h_dst_overflow ]   # W
       h_dst_underhalf = conf[ :h_dst_underhalf ] # X
-      
       expire_connecting = conf[ :expire_connecting ]     # 连接多久没有建成关闭（秒）
       expire_long_after = conf[ :expire_long_after ]     # 长连接多久没有新流量关闭（秒）
       expire_proxy_after = conf[ :expire_proxy_after ]   # proxy多久没有收到流量重建（秒）
@@ -62,8 +65,12 @@ module Girl
       expire_short_after = conf[ :expire_short_after ]   # 短连接创建多久后关闭（秒）
       is_debug = conf[ :is_debug ]
 
+      redir_host = redir_host ? redir_host.to_s : '0.0.0.0'
       redir_port = redir_port ? redir_port.to_i : 6666
       memd_port = memd_port ? memd_port.to_i : redir_port + 1
+      relayd_host = relayd_host ? relayd_host.to_s : '0.0.0.0'
+      relayd_port = relayd_port ? relayd_port.to_i : redir_port + 2
+      tspd_host = tspd_host ? tspd_host.to_s : '0.0.0.0'
       tspd_port = tspd_port ? tspd_port.to_i : 7777
       raise "missing proxyd host" unless proxyd_host
       proxyd_port = proxyd_port ? proxyd_port.to_i : 6060
@@ -128,8 +135,8 @@ module Girl
         end
       end
 
-      puts "girl proxy #{ Girl::VERSION } #{ im }"
-      puts "#{ redir_port } #{ tspd_port } #{ proxyd_host } #{ proxyd_port } #{ appd_host } #{ appd_port } #{ nameservers.inspect } #{ is_client_fastopen } #{ is_server_fastopen }"
+      puts "girl proxy #{ Girl::VERSION } #{ im } #{ redir_port } #{ relayd_port } #{ tspd_port }"
+      puts "#{ proxyd_host } #{ proxyd_port } #{ appd_host } #{ appd_port } #{ nameservers.inspect } #{ is_client_fastopen } #{ is_server_fastopen }"
       puts "#{ direct_path } #{ directs.size } directs"
       puts "#{ remote_path } #{ remotes.size } remotes"
 
@@ -139,8 +146,12 @@ module Girl
       end
 
       worker = Girl::ProxyWorker.new(
+        redir_host,
         redir_port,
         memd_port,
+        relayd_host,
+        relayd_port,
+        tspd_host,
         tspd_port,
         proxyd_host,
         proxyd_port,

data/lib/girl/proxy_worker.rb

@@ -3,8 +3,12 @@ module Girl
     include Dns
 
     def initialize(
+      redir_host,
       redir_port,
       memd_port,
+      relayd_host,
+      relayd_port,
+      tspd_host,
       tspd_port,
       proxyd_host,
       proxyd_port,
@@ -51,14 +55,16 @@ module Girl
       @remotes = remotes
       @local_ips = Socket.ip_address_list.select{ | info | info.ipv4? }.map{ | info | info.ip_address }
       @update_roles = [ :dns, :dst, :mem, :p1, :src, :rsv ] # 参与淘汰的角色
-      @updates_limit = 1008  # 淘汰池上限，1015(mac) - info, infod, memd, proxy, redir, rsvd, tspd
+      @updates_limit = 1007  # 淘汰池上限，1015(mac) - info, infod, memd, proxy, redir, relayd, rsvd, tspd
       @eliminate_count = 0   # 淘汰次数
       @reads = []            # 读池
       @writes = []           # 写池
-      @roles = {}            # sock =>  :dns / :dst / :infod / :mem / :memd / :p1 / :proxy / :redir / :rsv / :rsvd / :src / :tspd
+      @roles = {}            # sock =>  :dns / :dst / :girl / :infod / :mem / :memd / :p1 / :proxy / :redir / :relay / :relayd / :rsv / :rsvd / :src / :tspd
       @updates = {}          # sock => updated_at
       @proxy_infos = {}      # proxy => { :is_syn :paused_p1s :paused_srcs :rbuff :recv_at :wbuff }
       @mem_infos = {}        # mem => { :wbuff }
+      @relay_infos = {}      # relay => { :addrinfo :closing :girl :overflowing :wbuff }
+      @girl_infos = {}       # girl => { :closing :connected :is_syn :overflowing :relay :wbuff }
       @src_infos = {}        # src => { :addrinfo :closing :destination_domain :destination_port :dst :is_connect :overflowing :proxy_proto :proxy_type :rbuff :src_id :wbuff }
       @dst_infos = {}        # dst => { :closing :connected :domain :ip :overflowing :port :src :wbuff }
       @dns_infos = {}        # dns => { :domain :src }
@@ -100,11 +106,12 @@ module Girl
       @is_client_fastopen = is_client_fastopen
       @is_server_fastopen = is_server_fastopen
 
-      new_a_redir( redir_port )
+      new_a_redir( redir_host, redir_port )
       new_a_infod( redir_port )
       new_a_memd( memd_port )
-      new_a_rsvd( tspd_port )
-      new_a_tspd( tspd_port )
+      new_a_relayd( relayd_host, relayd_port )
+      new_a_rsvd( tspd_host, tspd_port )
+      new_a_tspd( tspd_host, tspd_port )
       new_a_proxy
     end
 
@@ -123,6 +130,8 @@ module Girl
             read_dns( sock )
           when :dst then
             read_dst( sock )
+          when :girl then
+            read_girl( sock )
           when :infod then
             read_infod( sock )
           when :mem then
@@ -135,6 +144,10 @@ module Girl
             read_proxy( sock )
           when :redir then
             read_redir( sock )
+          when :relay then
+            read_relay( sock )
+          when :relayd then
+            read_relayd( sock )
           when :rsv then
             read_rsv( sock )
           when :rsvd then
@@ -154,12 +167,16 @@ module Girl
           case role
           when :dst then
             write_dst( sock )
+          when :girl then
+            write_girl( sock )
           when :mem then
             write_mem( sock )
           when :p1 then
             write_p1( sock )
           when :proxy then
             write_proxy( sock )
+          when :relay then
+            write_relay( sock )
           when :src then
             write_src( sock )
           else
@@ -184,6 +201,12 @@ module Girl
       dst_info[ :wbuff ] << data
       bytesize = dst_info[ :wbuff ].bytesize
 
+      if bytesize >= CLOSE_ABOVE then
+        puts "close overflow dst #{ dst_info[ :domain ] }"
+        close_dst( dst )
+        return
+      end
+
       if !dst_info[ :overflowing ] && ( bytesize >= WBUFF_LIMIT ) then
         puts "dst overflow pause src #{ dst_info[ :domain ] }"
         @reads.delete( dst_info[ :src ] )
@@ -193,6 +216,27 @@ module Girl
       add_write( dst )
     end
 
+    def add_girl_wbuff( girl, data )
+      return if girl.nil? || girl.closed? || data.nil? || data.empty?
+      girl_info = @girl_infos[ girl ]
+      girl_info[ :wbuff ] << data
+      bytesize = girl_info[ :wbuff ].bytesize
+
+      if bytesize >= CLOSE_ABOVE then
+        puts "close overflow girl"
+        close_girl( girl )
+        return
+      end
+
+      if !girl_info[ :overflowing ] && ( bytesize >= WBUFF_LIMIT ) then
+        puts "girl overflow pause relay"
+        @reads.delete( girl_info[ :relay ] )
+        girl_info[ :overflowing ] = true
+      end
+
+      add_write( girl )
+    end
+
     def add_mem_wbuff( mem, data )
       return if mem.nil? || mem.closed? || data.nil? || data.empty?
       mem_info = @mem_infos[ mem ]
@@ -219,7 +263,7 @@ module Girl
         add_proxy_wbuff( pack_a_chunk( msg ) )
         p1_info[ :overflowing ] = true
       end
-      
+
       add_write( p1 )
     end
 
@@ -253,6 +297,27 @@ module Girl
       end
     end
 
+    def add_relay_wbuff( relay, data )
+      return if relay.nil? || relay.closed? || data.nil? || data.empty?
+      relay_info = @relay_infos[ relay ]
+      relay_info[ :wbuff ] << data
+      bytesize = relay_info[ :wbuff ].bytesize
+
+      if bytesize >= CLOSE_ABOVE then
+        puts "close overflow relay #{ relay_info[ :addrinfo ].ip_unpack.inspect }"
+        close_relay( relay )
+        return
+      end
+
+      if !relay_info[ :overflowing ] && ( bytesize >= WBUFF_LIMIT ) then
+        puts "relay overflow pause girl #{ relay_info[ :addrinfo ].ip_unpack.inspect }"
+        @reads.delete( relay_info[ :girl ] )
+        relay_info[ :overflowing ] = true
+      end
+
+      add_write( relay )
+    end
+
     def add_socks5_conn_reply( src )
       # +----+-----+-------+------+----------+----------+
       # |VER | REP |  RSV  | ATYP | BND.ADDR | BND.PORT |
@@ -334,6 +399,15 @@ module Girl
       end
     end
 
+    def check_expire_girls
+      now = Time.new
+
+      @girl_infos.select{ | girl, info | info[ :connected ] ? ( now.to_i - @updates[ girl ].to_i >= @expire_long_after ) : ( now.to_i - @updates[ girl ].to_i >= @expire_connecting ) }.each do | girl, _ |
+        puts "expire girl" if @is_debug
+        close_girl( girl )
+      end
+    end
+
     def check_expire_mems
       now = Time.new
 
@@ -361,6 +435,15 @@ module Girl
       end
     end
 
+    def check_expire_relays
+      now = Time.new
+
+      @relay_infos.select{ | relay, _ | now.to_i - @updates[ relay ].to_i >= @expire_long_after }.each do | relay, info |
+        puts "expire relay #{ info[ :addrinfo ].ip_unpack.inspect }" if @is_debug
+        close_relay( relay )
+      end
+    end
+
     def check_expire_rsvs
       now = Time.new
 
@@ -396,6 +479,15 @@ module Girl
       dst_info
     end
 
+    def close_girl( girl )
+      return nil if girl.nil? || girl.closed?
+      close_sock( girl )
+      girl_info = @girl_infos.delete( girl )
+      puts "close girl" if @is_debug
+      set_relay_closing( girl_info[ :relay ] ) if girl_info
+      girl_info
+    end
+
     def close_mem( mem )
       return nil if mem.nil? || mem.closed?
       close_sock( mem )
@@ -406,7 +498,7 @@ module Girl
       return nil if p1.nil? || p1.closed?
       close_sock( p1 )
       p1_info = @p1_infos.delete( p1 )
-      
+
       unless @proxy.closed? then
         proxy_info = @proxy_infos[ @proxy ]
         proxy_info[ :paused_p1s ].delete( p1 )
@@ -429,6 +521,15 @@ module Girl
       proxy_info
     end
 
+    def close_relay( relay )
+      return nil if relay.nil? || relay.closed?
+      close_sock( relay )
+      relay_info = @relay_infos.delete( relay )
+      puts "close relay" if @is_debug
+      set_girl_closing( relay_info[ :girl ] ) if relay_info
+      relay_info
+    end
+
     def close_rsv( rsv )
       return nil if rsv.nil? || rsv.closed?
       close_sock( rsv )
@@ -453,7 +554,7 @@ module Girl
       src_id = src_info[ :src_id ]
       domain = src_info[ :destination_domain ]
       puts "close src #{ domain }" if @is_debug
-      
+
       if src_info[ :proxy_type ] == :direct then
         set_dst_closing( src_info[ :dst ] )
       elsif ( src_info[ :proxy_type ] == :remote ) && !@proxy.closed? then
@@ -612,14 +713,14 @@ module Girl
       domain = src_info[ :destination_domain ]
       port = src_info[ :destination_port ]
 
-      if @local_ips.include?( ip ) && [ @redir_port, @tspd_port ].include?( port ) then
+      if @local_ips.include?( ip ) && [ @redir_port, @relayd_port, @tspd_port ].include?( port ) then
         puts "ignore #{ ip }:#{ port }"
         close_src( src )
         return
       end
 
-      if [ domain, ip ].include?( @proxyd_host )  then
-        # 访问远端，直连
+      if [ domain, ip ].include?( @proxyd_host ) && ![ 80, 443 ].include?( port ) then
+        # 访问远端非http端口，直连
         puts "direct #{ ip } #{ port }"
         new_a_dst( ip, src )
         return
@@ -709,13 +810,52 @@ module Girl
       end
     end
 
+    def new_a_girl( relay )
+      return if relay.nil? || relay.closed?
+      check_expire_girls
+
+      begin
+        girl = Socket.new( Socket::AF_INET, Socket::SOCK_STREAM, 0 )
+      rescue Exception => e
+        puts "new a girl #{ e.class }"
+        close_girl( girl )
+        return
+      end
+
+      girl.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1 )
+
+      begin
+        girl.connect_nonblock( @proxyd_addr )
+      rescue IO::WaitWritable
+      rescue Exception => e
+        puts "girl connect proxyd #{ e.class }"
+        girl.close
+        close_girl( girl )
+        return
+      end
+
+      girl_info = {
+        closing: false,
+        connected: false,
+        is_syn: @is_client_fastopen,
+        overflowing: false,
+        relay: relay,
+        wbuff: ''
+      }
+
+      @girl_infos[ girl ] = girl_info
+      add_read( girl, :girl )
+      add_write( girl )
+      girl
+    end
+
     def new_a_infod( infod_port )
-      infod_ip = '127.0.0.1'
-      infod_addr = Socket.sockaddr_in( infod_port, infod_ip )
+      infod_host = '127.0.0.1'
+      infod_addr = Socket.sockaddr_in( infod_port, infod_host )
       infod = Socket.new( Socket::AF_INET, Socket::SOCK_DGRAM, 0 )
       infod.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEPORT, 1 ) if RUBY_PLATFORM.include?( 'linux' )
       infod.bind( infod_addr )
-      puts "infod bind on #{ infod_ip } #{ infod_port }"
+      puts "infod bind on #{ infod_host } #{ infod_port }"
       add_read( infod, :infod )
       info = Socket.new( Socket::AF_INET, Socket::SOCK_DGRAM, 0 )
       @infod_addr = infod_addr
@@ -724,14 +864,14 @@ module Girl
     end
 
     def new_a_memd( memd_port )
-      memd_ip = '127.0.0.1'
+      memd_host = '127.0.0.1'
       memd = Socket.new( Socket::AF_INET, Socket::SOCK_STREAM, 0 )
       memd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1 )
       memd.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEPORT, 1 ) if RUBY_PLATFORM.include?( 'linux' )
       memd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_FASTOPEN, 5 ) if @is_server_fastopen
-      memd.bind( Socket.sockaddr_in( memd_port, memd_ip ) )
+      memd.bind( Socket.sockaddr_in( memd_port, memd_host ) )
       memd.listen( 5 )
-      puts "memd listen on #{ memd_ip } #{ memd_port }"
+      puts "memd listen on #{ memd_host } #{ memd_port }"
       add_read( memd, :memd )
     end
 
@@ -807,21 +947,33 @@ module Girl
       proxy_info
     end
 
-    def new_a_redir( redir_port )
-      redir_ip = '0.0.0.0'
+    def new_a_redir( redir_host, redir_port )
       redir = Socket.new( Socket::AF_INET, Socket::SOCK_STREAM, 0 )
       redir.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1 )
       redir.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEADDR, 1 )
       redir.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEPORT, 1 ) if RUBY_PLATFORM.include?( 'linux' )
       redir.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_FASTOPEN, BACKLOG ) if @is_server_fastopen
-      redir.bind( Socket.sockaddr_in( redir_port, redir_ip ) )
+      redir.bind( Socket.sockaddr_in( redir_port, redir_host ) )
       redir.listen( BACKLOG )
-      puts "redir listen on #{ redir_ip } #{ redir_port }"
+      puts "redir listen on #{ redir_host } #{ redir_port }"
       add_read( redir, :redir )
       @redir_port = redir_port
       @redir_local_address = redir.local_address
     end
 
+    def new_a_relayd( relayd_host, relayd_port )
+      relayd = Socket.new( Socket::AF_INET, Socket::SOCK_STREAM, 0 )
+      relayd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1 )
+      relayd.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEADDR, 1 )
+      relayd.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEPORT, 1 ) if RUBY_PLATFORM.include?( 'linux' )
+      relayd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_FASTOPEN, BACKLOG ) if @is_server_fastopen
+      relayd.bind( Socket.sockaddr_in( relayd_port, relayd_host ) )
+      relayd.listen( BACKLOG )
+      puts "relayd listen on #{ relayd_host } #{ relayd_port }"
+      add_read( relayd, :relayd )
+      @relayd_port = relayd_port
+    end
+
     def new_a_rsv( data, addrinfo, domain, type )
       check_expire_rsvs
       rsv = Socket.new( Socket::AF_INET, Socket::SOCK_DGRAM, 0 )
@@ -844,27 +996,25 @@ module Girl
       add_read( rsv, :rsv )
     end
 
-    def new_a_rsvd( rsvd_port )
-      rsvd_ip = '0.0.0.0'
-      rsvd_addr = Socket.sockaddr_in( rsvd_port, rsvd_ip )
+    def new_a_rsvd( rsvd_host, rsvd_port )
+      rsvd_addr = Socket.sockaddr_in( rsvd_port, rsvd_host )
       rsvd = Socket.new( Socket::AF_INET, Socket::SOCK_DGRAM, 0 )
       rsvd.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEPORT, 1 ) if RUBY_PLATFORM.include?( 'linux' )
       rsvd.bind( rsvd_addr )
-      puts "rsvd bind on #{ rsvd_ip } #{ rsvd_port }"
+      puts "rsvd bind on #{ rsvd_host } #{ rsvd_port }"
       add_read( rsvd, :rsvd )
       @rsvd = rsvd
     end
 
-    def new_a_tspd( tspd_port )
-      tspd_ip = '0.0.0.0'
+    def new_a_tspd( tspd_host, tspd_port )
       tspd = Socket.new( Socket::AF_INET, Socket::SOCK_STREAM, 0 )
       tspd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1 )
       tspd.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEADDR, 1 )
       tspd.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEPORT, 1 ) if RUBY_PLATFORM.include?( 'linux' )
       tspd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_FASTOPEN, BACKLOG ) if @is_server_fastopen
-      tspd.bind( Socket.sockaddr_in( tspd_port, tspd_ip ) )
+      tspd.bind( Socket.sockaddr_in( tspd_port, tspd_host ) )
       tspd.listen( BACKLOG )
-      puts "tspd listen on #{ tspd_ip } #{ tspd_port }"
+      puts "tspd listen on #{ tspd_host } #{ tspd_port }"
       add_read( tspd, :tspd )
       @tspd_port = tspd_port
     end
@@ -955,6 +1105,22 @@ module Girl
       add_src_wbuff( src, data )
     end
 
+    def read_girl( girl )
+      begin
+        data = girl.read_nonblock( READ_SIZE )
+      rescue Errno::ENOTCONN => e
+        return
+      rescue Exception => e
+        close_girl( girl )
+        return
+      end
+
+      set_update( girl )
+      girl_info = @girl_infos[ girl ]
+      relay = girl_info[ :relay ]
+      add_relay_wbuff( relay, data )
+    end
+
     def read_infod( infod )
       begin
         data, addrinfo, rflags, *controls = infod.recvmsg
@@ -1026,6 +1192,8 @@ module Girl
           updates: @updates.size,
           proxy_infos: @proxy_infos.size,
           mem_infos: @mem_infos.size,
+          relay_infos: @relay_infos.size,
+          girl_infos: @girl_infos.size,
           src_infos: @src_infos.size,
           dst_infos: @dst_infos.size,
           dns_infos: @dns_infos.size,
@@ -1148,6 +1316,47 @@ module Girl
       add_read( src, :src )
     end
 
+    def read_relay( relay )
+      begin
+        data = relay.read_nonblock( READ_SIZE )
+      rescue Errno::ENOTCONN => e
+        return
+      rescue Exception => e
+        close_relay( relay )
+        return
+      end
+
+      set_update( relay )
+      relay_info = @relay_infos[ relay ]
+      girl = relay_info[ :girl ]
+      add_girl_wbuff( girl, data )
+    end
+
+    def read_relayd( relayd )
+      check_expire_relays
+
+      begin
+        relay, addrinfo = relayd.accept_nonblock
+      rescue IO::WaitReadable, Errno::EINTR => e
+        puts "relayd accept #{ e.class }"
+        return
+      end
+
+      puts "relayd accept a relay #{ addrinfo.ip_unpack.inspect }"
+
+      relay_info = {
+        addrinfo: addrinfo,
+        closing: false,
+        girl: nil,
+        overflowing: false,
+        wbuff: ''
+      }
+
+      @relay_infos[ relay ] = relay_info
+      add_read( relay, :relay )
+      relay_info[ :girl ] = new_a_girl( relay )
+    end
+
     def read_rsv( rsv )
       begin
         data, addrinfo, rflags, *controls = rsv.recvmsg
@@ -1409,7 +1618,7 @@ module Girl
         unless @proxy.closed? then
           proxy_info = @proxy_infos[ @proxy ]
           bytesize = proxy_info[ :wbuff ].bytesize
-  
+
           if ( bytesize >= WBUFF_LIMIT ) && !proxy_info[ :paused_srcs ].include?( src ) then
             puts "proxy overflow pause src #{ src_id } #{ src_info[ :destination_domain ] }"
             @reads.delete( src )
@@ -1477,7 +1686,7 @@ module Girl
     def resolve_domain( domain, src )
       return if src.nil? || src.closed?
 
-      unless domain =~ /^[0-9a-zA-Z\-\.]{1,63}$/ then
+      unless domain =~ /^(?=^.{3,255}$)[a-zA-Z0-9][-a-zA-Z0-9]{0,62}(\.[a-zA-Z0-9][-a-zA-Z0-9]{0,62})+$/ then
         # 忽略非法域名
         puts "ignore #{ domain }"
         close_src( src )
@@ -1559,6 +1768,14 @@ module Girl
       add_write( dst )
     end
 
+    def set_girl_closing( girl )
+      return if girl.nil? || girl.closed?
+      girl_info = @girl_infos[ girl ]
+      return if girl_info.nil? || girl_info[ :closing ]
+      girl_info[ :closing ] = true
+      add_write( girl )
+    end
+
     def set_p1_closing( p1 )
       return if p1.nil? || p1.closed?
       p1_info = @p1_infos[ p1 ]
@@ -1567,6 +1784,14 @@ module Girl
       add_write( p1 )
     end
 
+    def set_relay_closing( relay )
+      return if relay.nil? || relay.closed?
+      relay_info = @relay_infos[ relay ]
+      return if relay_info.nil? || relay_info[ :closing ]
+      relay_info[ :closing ] = true
+      add_write( relay )
+    end
+
     def set_remote( src )
       return if src.nil? || src.closed?
 
@@ -1574,7 +1799,7 @@ module Girl
         close_src( src )
         return
       end
-      
+
       src_info = @src_infos[ src ]
       src_info[ :proxy_type ] = :remote
 
@@ -1627,10 +1852,14 @@ module Girl
             close_dns( _sock )
           when :dst
             close_dst( _sock )
+          when :girl
+            close_girl( _sock )
           when :mem
             close_mem( _sock )
           when :p1
             close_p1( _sock )
+          when :relay
+            close_relay( _sock )
           when :rsv
             close_rsv( _sock )
           when :src
@@ -1685,6 +1914,52 @@ module Girl
       end
     end
 
+    def write_girl( girl )
+      if girl.closed? then
+        puts "write closed girl?"
+        return
+      end
+
+      girl_info = @girl_infos[ girl ]
+      girl_info[ :connected ] = true
+      data = girl_info[ :wbuff ]
+
+      if data.empty? then
+        if girl_info[ :closing ] then
+          close_girl( girl )
+        else
+          @writes.delete( girl )
+        end
+
+        return
+      end
+
+      begin
+        if girl_info[ :is_syn ] then
+          written = girl.sendmsg_nonblock( data, 536870912, @proxyd_addr )
+          girl_info[ :is_syn ] = false
+        else
+          written = girl.write_nonblock( data )
+        end
+      rescue Errno::EINPROGRESS
+        return
+      rescue Exception => e
+        close_girl( girl )
+        return
+      end
+
+      set_update( girl )
+      data = data[ written..-1 ]
+      girl_info[ :wbuff ] = data
+      bytesize = girl_info[ :wbuff ].bytesize
+
+      if girl_info[ :overflowing ] && ( bytesize < RESUME_BELOW ) then
+        puts "girl underhalf"
+        add_read( girl_info[ :relay ] )
+        girl_info[ :overflowing ] = false
+      end
+    end
+
     def write_mem( mem )
       if mem.closed? then
         puts "write closed mem?"
@@ -1797,7 +2072,7 @@ module Girl
           proxy_info[ :paused_srcs ].each{ | src | add_read( src ) }
           proxy_info[ :paused_srcs ].clear
         end
-  
+
         if proxy_info[ :paused_p1s ].any? then
           puts "proxy underhalf resume p1s #{ proxy_info[ :paused_p1s ].size }"
           proxy_info[ :paused_p1s ].each{ | p1 | add_read( p1 ) }
@@ -1838,7 +2113,7 @@ module Girl
       data = data[ written..-1 ]
       src_info[ :wbuff ] = data
       bytesize = src_info[ :wbuff ].bytesize
-      
+
       if src_info[ :overflowing ] && ( bytesize < RESUME_BELOW ) then
         src_id = src_info[ :src_id ]
         domain = src_info[ :destination_domain ]
@@ -1856,5 +2131,45 @@ module Girl
       end
     end
 
+    def write_relay( relay )
+      if relay.closed? then
+        puts "write closed relay?"
+        return
+      end
+
+      relay_info = @relay_infos[ relay ]
+      data = relay_info[ :wbuff ]
+
+      if data.empty? then
+        if relay_info[ :closing ] then
+          close_relay( relay )
+        else
+          @writes.delete( relay )
+        end
+
+        return
+      end
+
+      begin
+        written = relay.write_nonblock( data )
+      rescue Errno::EINPROGRESS
+        return
+      rescue Exception => e
+        close_relay( relay )
+        return
+      end
+
+      set_update( relay )
+      data = data[ written..-1 ]
+      relay_info[ :wbuff ] = data
+      bytesize = relay_info[ :wbuff ].bytesize
+
+      if relay_info[ :overflowing ] && ( bytesize < RESUME_BELOW ) then
+        puts "relay underhalf"
+        add_read( relay_info[ :girl ] )
+        relay_info[ :overflowing ] = false
+      end
+    end
+
   end
 end

data/lib/girl/proxyd_worker.rb

@@ -170,7 +170,7 @@ module Girl
         close_dst( dst )
         return
       end
-      
+
       if !dst_info[ :overflowing ] && ( bytesize >= WBUFF_LIMIT ) then
         proxy = dst_info[ :proxy ]
         puts "add h_dst_overflow #{ im } #{ src_id } #{ domain }"
@@ -209,7 +209,7 @@ module Girl
         close_p2( p2 )
         return
       end
-      
+
       if !p2_info[ :overflowing ] && ( bytesize >= WBUFF_LIMIT ) then
         proxy = im_info[ :proxy ]
         puts "add h_p2_overflow #{ im } #{ p2_id }"
@@ -641,12 +641,12 @@ module Girl
     end
 
     def new_a_infod( infod_port )
-      infod_ip = '127.0.0.1'
-      infod_addr = Socket.sockaddr_in( infod_port, infod_ip )
+      infod_host = '127.0.0.1'
+      infod_addr = Socket.sockaddr_in( infod_port, infod_host )
       infod = Socket.new( Socket::AF_INET, Socket::SOCK_DGRAM, 0 )
       infod.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEPORT, 1 ) if RUBY_PLATFORM.include?( 'linux' )
       infod.bind( infod_addr )
-      puts "infod bind on #{ infod_ip } #{ infod_port }"
+      puts "infod bind on #{ infod_host } #{ infod_port }"
       add_read( infod, :infod )
       info = Socket.new( Socket::AF_INET, Socket::SOCK_DGRAM, 0 )
       @infod_addr = infod_addr
@@ -655,14 +655,14 @@ module Girl
     end
 
     def new_a_memd( memd_port )
-      memd_ip = '127.0.0.1'
+      memd_host = '127.0.0.1'
       memd = Socket.new( Socket::AF_INET, Socket::SOCK_STREAM, 0 )
       memd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1 )
       memd.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEPORT, 1 ) if RUBY_PLATFORM.include?( 'linux' )
       memd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_FASTOPEN, 5 ) if @is_server_fastopen
-      memd.bind( Socket.sockaddr_in( memd_port, memd_ip ) )
+      memd.bind( Socket.sockaddr_in( memd_port, memd_host ) )
       memd.listen( 5 )
-      puts "memd listen on #{ memd_ip } #{ memd_port }"
+      puts "memd listen on #{ memd_host } #{ memd_port }"
       add_read( memd, :memd )
     end
 
@@ -715,14 +715,14 @@ module Girl
     end
 
     def new_a_proxyd( proxyd_port )
-      proxyd_ip = '0.0.0.0'
+      proxyd_host = '0.0.0.0'
       proxyd = Socket.new( Socket::AF_INET, Socket::SOCK_STREAM, 0 )
       proxyd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1 )
       proxyd.setsockopt( Socket::SOL_SOCKET, Socket::SO_REUSEPORT, 1 ) if RUBY_PLATFORM.include?( 'linux' )
       proxyd.setsockopt( Socket::IPPROTO_TCP, Socket::TCP_FASTOPEN, BACKLOG ) if @is_server_fastopen
-      proxyd.bind( Socket.sockaddr_in( proxyd_port, proxyd_ip ) )
+      proxyd.bind( Socket.sockaddr_in( proxyd_port, proxyd_host ) )
       proxyd.listen( BACKLOG )
-      puts "proxyd listen on #{ proxyd_ip } #{ proxyd_port }"
+      puts "proxyd listen on #{ proxyd_host } #{ proxyd_port }"
       add_read( proxyd, :proxyd )
     end
 
@@ -1120,7 +1120,7 @@ module Girl
       domain = domain_port[ 0...colon_idx ]
       port = domain_port[ ( colon_idx + 1 )..-1 ].to_i
 
-      if ( domain !~ /^[0-9a-zA-Z\-\.]{1,63}$/ ) || ( domain =~ /^((0\.\d{1,3}\.\d{1,3}\.\d{1,3})|(10\.\d{1,3}\.\d{1,3}\.\d{1,3})|(127\.\d{1,3}\.\d{1,3}\.\d{1,3})|(169\.254\.\d{1,3}\.\d{1,3})|(172\.((1[6-9])|(2\d)|(3[01]))\.\d{1,3}\.\d{1,3})|(192\.168\.\d{1,3}\.\d{1,3})|(255\.255\.255\.255)|(localhost))$/ ) then
+      if ( domain !~ /^(?=^.{3,255}$)[a-zA-Z0-9][-a-zA-Z0-9]{0,62}(\.[a-zA-Z0-9][-a-zA-Z0-9]{0,62})+$/ ) || ( domain =~ /^((0\.\d{1,3}\.\d{1,3}\.\d{1,3})|(10\.\d{1,3}\.\d{1,3}\.\d{1,3})|(127\.\d{1,3}\.\d{1,3}\.\d{1,3})|(169\.254\.\d{1,3}\.\d{1,3})|(172\.((1[6-9])|(2\d)|(3[01]))\.\d{1,3}\.\d{1,3})|(192\.168\.\d{1,3}\.\d{1,3})|(255\.255\.255\.255)|(localhost))$/ ) then
         # 忽略非法域名，内网地址
         puts "ignore #{ domain }"
         return
@@ -1348,7 +1348,7 @@ module Girl
       data = data[ written..-1 ]
       p2_info[ :wbuff ] = data
       bytesize = p2_info[ :wbuff ].bytesize
-      
+
       if p2_info[ :overflowing ] && ( bytesize < RESUME_BELOW ) then
         proxy = im_info[ :proxy ]
         p2_id = p2_info[ :p2_id ]
@@ -1396,7 +1396,7 @@ module Girl
           proxy_info[ :paused_dsts ].each{ | dst | add_read( dst ) }
           proxy_info[ :paused_dsts ].clear
         end
-  
+
         if proxy_info[ :paused_p2s ].any? then
           puts "proxy underhalf resume p2s #{ im } #{ proxy_info[ :paused_p2s ].size }"
           proxy_info[ :paused_p2s ].each{ | p2 | add_read( p2 ) }

data/lib/girl/version.rb

@@ -1,3 +1,3 @@
 module Girl
-  VERSION = '9.1.0'.freeze
+  VERSION = '9.1.2'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: girl
 version: !ruby/object:Gem::Version
-  version: 9.1.0
+  version: 9.1.2
 platform: ruby
 authors:
 - takafan
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-02-18 00:00:00.000000000 Z
+date: 2024-03-12 00:00:00.000000000 Z
 dependencies: []
 description: escape evil.
 email: