gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2024-10-04 15:11:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update worker.rb

Browse Source
This commit is contained in:
Namhyeon Go 2024-03-07 16:28:23 +09:00 committed by GitHub
parent 1d665793bc
commit 968791c233
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 51 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
assets/ruby/worker.rb
View File

@ -1,5 +1,8 @@
require 'socket' require 'socket'
require 'json' require 'json'
require 'openssl'
require 'base64'
require 'timeout'
DEFAULT_SOCKET_TIMEOUT = 1 DEFAULT_SOCKET_TIMEOUT = 1
STATEFUL_SOCKET_TIMEOUT = 30 STATEFUL_SOCKET_TIMEOUT = 30
@ -44,10 +47,13 @@ end
def read_from_remote_server(remote_address, remote_port, scheme, data = nil, conn = nil, buffer_size = 8192, id = '') def read_from_remote_server(remote_address, remote_port, scheme, data = nil, conn = nil, buffer_size = 8192, id = '')
if ["https", "ssl", "tls"].include?(scheme) if ["https", "ssl", "tls"].include?(scheme)
remote_address = "tls://" + remote_address ssl_context = OpenSSL::SSL::SSLContext.new
sock = OpenSSL::SSL::SSLSocket.new(TCPSocket.open(remote_address, remote_port), ssl_context)
sock.connect
else
sock = TCPSocket.open(remote_address, remote_port)
end end
sock = TCPSocket.open(remote_address, remote_port)
if sock.nil? if sock.nil?
error = { error = {
"status" => 502, "status" => 502,
@ -102,8 +108,23 @@ def relay_request(params, id = '')
scheme = params['scheme'] scheme = params['scheme']
datetime = params['datetime'] # format: %Y-%m-%d %H:%M:%S.%f datetime = params['datetime'] # format: %Y-%m-%d %H:%M:%S.%f
if ["https", "ssl", "tls"].include?(scheme) begin
remote_address = "tls://" + remote_address Timeout.timeout(DEFAULT_SOCKET_TIMEOUT) do
if ["https", "ssl", "tls"].include?(scheme)
ssl_context = OpenSSL::SSL::SSLContext.new
sock = OpenSSL::SSL::SSLSocket.new(TCPSocket.open(remote_address, remote_port), ssl_context)
sock.connect
else
sock = TCPSocket.open(remote_address, remote_port)
end
end
rescue Timeout::Error
error = {
"status" => 504,
"message" => "Gateway Timeout"
}
puts jsonrpc2_error_encode(error, id)
return
end end
case request_header['@method'][0] case request_header['@method'][0]
@ -130,27 +151,36 @@ def relay_connect(params, id = '')
datetime = params['datetime'] # format: %Y-%m-%d %H:%M:%S.%f datetime = params['datetime'] # format: %Y-%m-%d %H:%M:%S.%f
starttime = Time.now.to_f starttime = Time.now.to_f
conn = TCPSocket.open(client_address, client_port)
if conn.nil? begin
Timeout.timeout(STATEFUL_SOCKET_TIMEOUT) do
if ["https", "ssl", "tls"].include?(scheme)
ssl_context = OpenSSL::SSL::SSLContext.new
conn = OpenSSL::SSL::SSLSocket.new(TCPSocket.open(client_address, client_port), ssl_context)
conn.connect
else
conn = TCPSocket.open(client_address, client_port)
end
end
rescue Timeout::Error
error = { error = {
"status" => 502, "status" => 504,
"code" => error_code, "message" => "Gateway Timeout"
"message" => error_message,
"_params" => params
} }
puts jsonrpc2_error_encode(error, id) puts jsonrpc2_error_encode(error, id)
else return
stoptime = Time.now.to_f
connection_speed = ((stoptime - starttime) * 1000).to_i
data = jsonrpc2_encode("relay_accept", {
"success" => true,
"connection_speed" => connection_speed
}, id)
conn.write(data + "\r\n\r\n")
read_from_remote_server(remote_address, remote_port, scheme, nil, conn, buffer_size, id)
conn.close
end end
stoptime = Time.now.to_f
connection_speed = ((stoptime - starttime) * 1000).to_i
data = jsonrpc2_encode("relay_accept", {
"success" => true,
"connection_speed" => connection_speed
}, id)
conn.write(data + "\r\n\r\n")
read_from_remote_server(remote_address, remote_port, scheme, nil, conn, buffer_size, id)
conn.close
end end
context = JSON.parse(STDIN.read) context = JSON.parse(STDIN.read)