Class: Girl::ProxyWorker

Inherits:
Object
  • Object
show all
Includes:
Dns
Defined in:
lib/girl/proxy_worker.rb

Instance Method Summary collapse

Methods included from Dns

#pack_a_query, #pack_domain, #seek_dn, #seek_ip, #seek_question, #seek_question_dn, #seek_rr_ip

Constructor Details

#initialize(redir_host, redir_port, memd_port, relayd_host, relayd_port, tspd_host, tspd_port, proxyd_host, proxyd_port, nameservers, im, directs, remotes, appd_host, appd_port, head_len, h_a_new_source, h_a_new_p2, h_dst_close, h_heartbeat, h_p1_close, h_p2_close, h_p2_traffic, h_p1_overflow, h_p1_underhalf, h_p2_overflow, h_p2_underhalf, h_query, h_response, h_src_close, h_traffic, h_src_overflow, h_src_underhalf, h_dst_overflow, h_dst_underhalf, expire_connecting, expire_long_after, expire_proxy_after, expire_resolv_cache, expire_short_after, is_debug, is_client_fastopen, is_server_fastopen) ⇒ ProxyWorker

Returns a new instance of ProxyWorker.



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/girl/proxy_worker.rb', line 5

def initialize(
  redir_host,
  redir_port,
  memd_port,
  relayd_host,
  relayd_port,
  tspd_host,
  tspd_port,
  proxyd_host,
  proxyd_port,
  nameservers,
  im,
  directs,
  remotes,
  appd_host,
  appd_port,
  head_len,
  h_a_new_source,
  h_a_new_p2,
  h_dst_close,
  h_heartbeat,
  h_p1_close,
  h_p2_close,
  h_p2_traffic,
  h_p1_overflow,
  h_p1_underhalf,
  h_p2_overflow,
  h_p2_underhalf,
  h_query,
  h_response,
  h_src_close,
  h_traffic,
  h_src_overflow,
  h_src_underhalf,
  h_dst_overflow,
  h_dst_underhalf,
  expire_connecting,
  expire_long_after,
  expire_proxy_after,
  expire_resolv_cache,
  expire_short_after,
  is_debug,
  is_client_fastopen,
  is_server_fastopen )

  @proxyd_host = proxyd_host
  @proxyd_addr = Socket.sockaddr_in( proxyd_port, proxyd_host )
  @nameserver_addrs = nameservers.map{ | n | Socket.sockaddr_in( 53, n ) }
  @im = im
  @directs = directs
  @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 = 1007  # 淘汰池上限，1015(mac) - info, infod, memd, proxy, redir, relayd, rsvd, tspd
  @eliminate_count = 0   # 淘汰次数
  @reads = []            # 读池
  @writes = []           # 写池
  @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 }
  @rsv_infos = {}        # rsv => { :addrinfo :domain :type }
  @near_infos = {}       # near_id => { :addrinfo :created_at :domain :id :type }
  @resolv_caches = {}    # domain => [ ip, created_at ]
  @is_direct_caches = {} # ip => true / false
  @response_caches = {}  # domain => [ response, created_at, ip, is_remote ]
  @response6_caches = {} # domain => [ response, created_at, ip, is_remote ]
  @p1_infos = {}         # p1 => { :closing :connected :overflowing :p2_id :wbuff }
  @appd_addr = Socket.sockaddr_in( appd_port, appd_host )

  @head_len = head_len
  @h_a_new_source = h_a_new_source
  @h_a_new_p2 = h_a_new_p2
  @h_dst_close = h_dst_close
  @h_heartbeat = h_heartbeat
  @h_p1_close = h_p1_close
  @h_p2_close = h_p2_close
  @h_p2_traffic = h_p2_traffic
  @h_p1_overflow = h_p1_overflow
  @h_p1_underhalf = h_p1_underhalf
  @h_p2_overflow = h_p2_overflow
  @h_p2_underhalf = h_p2_underhalf
  @h_query = h_query
  @h_response = h_response
  @h_src_close = h_src_close
  @h_traffic = h_traffic
  @h_src_overflow = h_src_overflow
  @h_src_underhalf = h_src_underhalf
  @h_dst_overflow = h_dst_overflow
  @h_dst_underhalf = h_dst_underhalf
  @expire_connecting = expire_connecting
  @expire_long_after = expire_long_after
  @expire_proxy_after = expire_proxy_after
  @expire_resolv_cache = expire_resolv_cache
  @expire_short_after = expire_short_after
  @is_debug = is_debug
  @is_client_fastopen = is_client_fastopen
  @is_server_fastopen = is_server_fastopen

  new_a_redir( redir_host, redir_port )
  new_a_infod( redir_port )
  new_a_memd( memd_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

Instance Method Details

#looping ⇒ Object 



118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
 
# File 'lib/girl/proxy_worker.rb', line 118

def looping
  puts "looping"
  loop_heartbeat

  loop do
    rs, ws = IO.select( @reads, @writes )

    rs.each do | sock |
      role = @roles[ sock ]

      case role
      when :dns then
        read_dns( sock )
      when :dst then
        read_dst( sock )
      when :girl then
        read_girl( sock )
      when :infod then
        read_infod( sock )
      when :mem then
        read_mem( sock )
      when :memd then
        read_memd( sock )
      when :p1 then
        read_p1( sock )
      when :proxy then
        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
        read_rsvd( sock )
      when :src then
        read_src( sock )
      when :tspd then
        read_tspd( sock )
      else
        close_sock( sock )
      end
    end

    ws.each do | sock |
      role = @roles[ sock ]

      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
        close_sock( sock )
      end
    end
  end
rescue Interrupt => e
  puts e.class
  quit!
end

#quit! ⇒ Object 



192
193
194
 
# File 'lib/girl/proxy_worker.rb', line 192

def quit!
  exit
end