1
2
3
4
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
|
import { server, io } from './site'
import * as proxy from './proxy'
let relay_connected = false
export const client = io.of('/client')
client.on('connect', bind_client)
export const relay = (() => {
let relay;
if (process.env.EXPRESS_CONNECTS_TO_RELAY === 'true') {
const relay_endpoint = (process.env.SECURE_PROXY ? 'https' : 'http') + "://" + process.env.PROXY_REMOTE + '/client'
console.log('Connecting to relay on ' + relay_endpoint)
relay = require('socket.io-client').connect(relay_endpoint)
proxy.attach(server)
bind_relay(relay)
} else {
relay = io.of('/relay')
relay.on('connect', bind_relay)
}
return relay
})()
const syscall_lookup = {}
function bind_relay(socket) {
console.log('Relay connected')
relay_connected = true
client.emit('system_res', { type: 'relay_connected' })
socket.on('res', data => {
// console.log('Received response', data.cmd)
client.emit('res', data)
})
socket.on('status', data => {
client.emit('status', data)
})
socket.on('system_res', data => {
if (process.env.CACHE_SYSCALLS) {
const id = make_server_id(data)
syscall_lookup[id] = data
}
client.emit('system_res', data)
})
socket.on('task_res', data => {
client.emit('task_res', data)
})
socket.on('frame', (data) => {
client.volatile.emit('frame', data)
})
socket.on('disconnect', () => {
console.log('Relay disconnected')
relay_connected = false
client.emit('system_res', { type: 'relay_disconnected' })
})
}
function make_client_id(data){
return [data.cmd, JSON.stringify(data.payload)].join('_')
}
function make_server_id(data){
return [data.type, JSON.stringify(data.cmd || data.dir)].join('_')
}
function bind_client(socket){
console.log('Client connected')
if (relay_connected) {
socket.emit('system_res', { type: 'relay_connected' })
relay.emit('system', { cmd: 'get_status' })
} else {
socket.emit('system_res', { type: 'relay_disconnected' })
}
socket.emit('system_res', {
type: 'site',
site: {
name: process.env.SITE_NAME || 'untitled'
}
})
socket.on('cmd', data => {
// console.log('Client sent command', data)
relay.emit('cmd', data)
})
socket.on('system', data => {
// console.log('Client sent system command', data)
if (process.env.CACHE_SYSCALLS) {
const id = make_client_id(data)
console.log('client', id)
// console.log(id in syscall_lookup)
if (id in syscall_lookup) {
const cached = syscall_lookup[id]
syscall_lookup[id].uuid = data.uuid
return socket.emit('system_res', cached)
}
}
relay.emit('system', data)
})
socket.on('task', data => {
// console.log('Client sent task command', data)
relay.emit('task', data)
})
socket.on('disconnect', () => {
console.log('Client disconnected')
})
}
|
