p2p2 0.16.0 → 0.21.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0ff46f98fa6ec31663730e3a2f79fa49b0196823c2b99688daaf4f34a8c91bd3
-  data.tar.gz: 24b9449407dc43bc37a23b77e37e39548ad6313a52eaacea359fec96da786ef1
+  metadata.gz: 57e288ebc806c2661f7fa5314a9be3fa16bbb1dc2deac96cfce9f3ca65c97bea
+  data.tar.gz: e7872fc331f68d07aaf16b02ea95fd3cdcc09f122acc4b16a51eeb49c5cac854
 SHA512:
-  metadata.gz: b86a9e22e55b4b31339877b3fdfc294b24720a9f906347019aa84f59a3274ea93f20ad0552cfd4203dcdc9510bc0920eb7096af490f2397d1b016f795255ef4e
-  data.tar.gz: 724486226b02372d27b5b32534141e0e4952e5825bf35ab7ef3cc13f5cdea8181f28cf2446fbf71b4e4b5f82d20f6cfe585ba30ce0cb42e582167e6a47e89b7b
+  metadata.gz: 665ea51b965d27fdd38fabcdbab4ab7932a9cad6ca70d4503a4c452597979eee6598706b4f8d15670a188742a14375805cb8f9eb2e919c7c48bc8917f2a3a0b4
+  data.tar.gz: 7263370934487b9e3937931cbefb13feedce4b14c30fb3c25f6c3119f41fe9d878e4b18a70cbb733267bcc95b6ba39abe097a4bcb1cf35b474cad04dd1379a92

data/lib/p2p2/custom.rb

@@ -1,4 +1,4 @@
-module P2pd
+module P2p2
   module Custom
     def encode( data )
       # overwrite me, you'll be free

data/lib/p2p2/head.rb

@@ -6,11 +6,12 @@ module P2p2
   RESUME_BELOW          = 50_000           # 降到多少以下恢复写
   EXPIRE_NEW            = 10               # 创建之后多久没有流量进来，过期
   EXPIRE_AFTER          = 300              # 多久没有新流量，过期
-  CHECK_EXPIRE_INTERVAL = 30               # 检查过期间隔
+  CHECK_EXPIRE_INTERVAL = 60               # 检查过期间隔
   STATUS_INTERVAL       = 0.5              # 发送状态间隔
   SEND_STATUS_UNTIL     = 10               # 持续的告之对面状态，直到没有流量往来，持续多少秒
   BREAK_SEND_MISS       = 10_000           # miss包个数上限，达到上限忽略要后面的段，可控碎片缓存
   CONFUSE_UNTIL         = 5                # 混淆前几个包
+  HEARTBEAT_INTERVAL    = 5                # 心跳间隔
   UPDATE_ROOM_INTERVAL  = 60               # 刷新房间间隔
   PEER_ADDR             = 1
   HEARTBEAT             = 2

data/lib/p2p2/p1_worker.rb

@@ -28,8 +28,7 @@ module P2p2
     #
     def looping
       puts "#{ Time.new } looping"
-      loop_update_room
-      loop_check_expire
+      loop_heartbeat
       loop_check_status
 
       loop do
@@ -80,67 +79,65 @@ module P2p2
     private
 
     ##
-    # loop update room
+    # loop heartbeat
     #
-    def loop_update_room
+    def loop_heartbeat( check_at = Time.new )
       Thread.new do
         loop do
-          sleep UPDATE_ROOM_INTERVAL
+          sleep HEARTBEAT_INTERVAL
 
           @mutex.synchronize do
-            if !@tund.closed? && @tund_info[ :peer_addr ].nil?
-              add_tund_ctlmsg( @room, @p2pd_addr )
-              next_tick
-            end
-          end
-        end
-      end
-    end
-
-    ##
-    # loop check expire
-    #
-    def loop_check_expire
-      Thread.new do
-        loop do
-          sleep CHECK_EXPIRE_INTERVAL
-
-          @mutex.synchronize do
-            need_trigger = false
             now = Time.new
 
-            if !@tund.closed? && @tund_info[ :tun_addr ]
-              if now - @tund_info[ :last_recv_at ] > EXPIRE_AFTER
-                puts "#{ Time.new } expire tund"
-                set_is_closing( @tund )
-              else
-                # puts "debug1 #{ Time.new } heartbeat"
-                add_tund_ctlmsg( pack_a_heartbeat )
-
-                @tund_info[ :dst_exts ].each do | dst_local_port, dst_ext |
-                  if dst_ext[ :dst ].closed? && ( now - dst_ext[ :last_continue_at ] > EXPIRE_AFTER )
-                    puts "#{ Time.new } expire dst ext #{ dst_local_port }"
-                    del_dst_ext( dst_local_port )
+            unless @tund.closed?
+              if @tund_info[ :peer_addr ]
+                if @tund_info[ :tun_addr ]
+                  if now - check_at >= CHECK_EXPIRE_INTERVAL
+                    if now - @tund_info[ :last_recv_at ] > EXPIRE_AFTER
+                      puts "#{ Time.new } expire tund"
+                      set_is_closing( @tund )
+                    else
+                      @tund_info[ :dst_exts ].each do | dst_local_port, dst_ext |
+                        if dst_ext[ :dst ].closed? && ( now - dst_ext[ :last_continue_at ] > EXPIRE_AFTER )
+                          puts "#{ Time.new } expire dst ext #{ dst_local_port }"
+                          del_dst_ext( dst_local_port )
+                        end
+                      end
+                    end
+
+                    @dst_infos.each do | dst, dst_info |
+                      if dst_info[ :last_recv_at ].nil? && ( now - dst_info[ :created_at ] > EXPIRE_NEW )
+                        puts "#{ Time.new } expire dst"
+                        set_is_closing( dst )
+                      end
+                    end
+
+                    check_at = now
                   end
-                end
-              end
 
-              need_trigger = true
-            end
-
-            @dst_infos.each do | dst, dst_info |
-              is_expired = dst_info[ :last_recv_at ].nil? && ( now - dst_info[ :created_at ] > EXPIRE_NEW )
+                  # puts "debug2 heartbeat"
+                  add_tund_ctlmsg( pack_a_heartbeat )
+                  next_tick
+                elsif now - @tund_info[ :created_at ] > EXPIRE_NEW
+                  # no tun addr
+                  puts "#{ Time.new } expire new tund"
+                  set_is_closing( @tund )
+                  next_tick
+                end
+              else
+                # no peer addr
+                if now - check_at >= UPDATE_ROOM_INTERVAL
+                  data = @room
+                  check_at = now
+                else
+                  data = [ rand( 128 ) ].pack( 'C' )
+                end
 
-              if is_expired
-                puts "#{ Time.new } expire dst"
-                set_is_closing( dst )
-                need_trigger = true
+                # puts "debug2 update room"
+                add_tund_ctlmsg( data, @p2pd_addr )
+                next_tick
               end
             end
-
-            if need_trigger
-              next_tick
-            end
           end
         end
       end
@@ -252,6 +249,8 @@ module P2p2
     # is match tun addr
     #
     def is_match_tun_addr( addrinfo )
+      return false unless @tund_info[ :tun_addr ]
+
       from_addr = addrinfo.to_sockaddr
 
       if from_addr != @tund_info[ :tun_addr ]
@@ -260,6 +259,7 @@ module P2p2
       end
 
       @tund_info[ :last_recv_at ] = Time.new
+
       true
     end
 

data/lib/p2p2/p2_worker.rb

@@ -28,7 +28,7 @@ module P2p2
     #
     def looping
       puts "#{ Time.new } looping"
-      loop_check_expire
+      loop_heartbeat
       loop_check_status
 
       loop do
@@ -81,49 +81,51 @@ module P2p2
     private
 
     ##
-    # loop check expire
+    # loop heartbeat
     #
-    def loop_check_expire
+    def loop_heartbeat( check_at = Time.new )
       Thread.new do
         loop do
-          sleep CHECK_EXPIRE_INTERVAL
+          sleep HEARTBEAT_INTERVAL
 
           @mutex.synchronize do
-            need_trigger = false
             now = Time.new
 
-            if @tun && !@tun.closed? && @tun_info[ :tund_addr ]
-              if now - @tun_info[ :last_recv_at ] > EXPIRE_AFTER
-                puts "#{ Time.new } expire tun"
-                set_is_closing( @tun )
-              else
-                # puts "debug1 #{ Time.new } heartbeat"
-                add_tun_ctlmsg( pack_a_heartbeat )
-
-                @tun_info[ :src_exts ].each do | src_id, src_ext |
-                  if src_ext[ :src ].closed? && ( now - src_ext[ :last_continue_at ] > EXPIRE_AFTER )
-                    puts "#{ Time.new } expire src ext #{ src_id }"
-                    del_src_ext( src_id )
+            if @tun && !@tun.closed? && @tun_info[ :peer_addr ]
+              if @tun_info[ :tund_addr ]
+                if now - check_at >= CHECK_EXPIRE_INTERVAL
+                  if now - @tun_info[ :last_recv_at ] > EXPIRE_AFTER
+                    puts "#{ Time.new } expire tun"
+                    set_is_closing( @tun )
+                  else
+                    @tun_info[ :src_exts ].each do | src_id, src_ext |
+                      if src_ext[ :src ].closed? && ( now - src_ext[ :last_continue_at ] > EXPIRE_AFTER )
+                        puts "#{ Time.new } expire src ext #{ src_id }"
+                        del_src_ext( src_id )
+                      end
+                    end
                   end
-                end
-              end
 
-              need_trigger = true
-            end
+                  @src_infos.each do | src, src_info |
+                    if src_info[ :last_recv_at ].nil? && ( now - src_info[ :created_at ] > EXPIRE_NEW )
+                      puts "#{ Time.new } expire src"
+                      set_is_closing( src )
+                    end
+                  end
 
-            @src_infos.each do | src, src_info |
-              is_expired = src_info[ :last_recv_at ].nil? && ( now - src_info[ :created_at ] > EXPIRE_NEW )
+                  check_at = now
+                end
 
-              if is_expired
-                puts "#{ Time.new } expire src"
-                set_is_closing( src )
-                need_trigger = true
+                # puts "debug2 heartbeat"
+                add_tun_ctlmsg( pack_a_heartbeat )
+                next_tick
+              elsif now - @tun_info[ :created_at ] > EXPIRE_NEW
+                # no tund addr
+                puts "#{ Time.new } expire new tun"
+                set_is_closing( @tun )
+                next_tick
               end
             end
-
-            if need_trigger
-              next_tick
-            end
           end
         end
       end
@@ -275,6 +277,8 @@ module P2p2
     # is match tund addr
     #
     def is_match_tund_addr( addrinfo )
+      return false unless @tun_info[ :tund_addr ]
+
       from_addr = addrinfo.to_sockaddr
 
       if from_addr != @tun_info[ :tund_addr ]
@@ -283,6 +287,7 @@ module P2p2
       end
 
       @tun_info[ :last_recv_at ] = Time.new
+
       true
     end
 

data/lib/p2p2/version.rb

@@ -1,3 +1,3 @@
 module P2p2
-  VERSION = '0.16.0'.freeze
+  VERSION = '0.21.0'.freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: p2p2
 version: !ruby/object:Gem::Version
-  version: 0.16.0
+  version: 0.21.0
 platform: ruby
 authors:
 - takafan