Commit 5dd883b3 authored by nanahira's avatar nanahira

neos implementation

parent 4ac3f97b
Pipeline #17322 passed with stages
in 8 minutes and 11 seconds
// Generated by CoffeeScript 2.5.1 // Generated by CoffeeScript 2.6.1
(function() { (function() {
var Replay, ReplayReader, fs, lzma, replayHeader; var Replay, ReplayReader, fs, lzma, replayHeader;
......
...@@ -156,16 +156,23 @@ class YGOProMessagesHelper { ...@@ -156,16 +156,23 @@ class YGOProMessagesHelper {
} }
return sendBuffer; return sendBuffer;
} }
sendMessage(socket, protostr, info) { send(socket, buffer) {
const sendBuffer = this.prepareMessage(protostr, info);
socket.write(sendBuffer);
}
sendMessageAsync(socket, protostr, info) {
const sendBuffer = this.prepareMessage(protostr, info);
return new Promise(done => { return new Promise(done => {
socket.write(sendBuffer, done); if (socket['isWs']) {
const ws = socket;
// @ts-ignore
ws.send(buffer, {}, done);
}
else {
const sock = socket;
sock.write(buffer, done);
}
}); });
} }
sendMessage(socket, protostr, info) {
const sendBuffer = this.prepareMessage(protostr, info);
return this.send(socket, sendBuffer);
}
addHandler(protostr, handler, synchronous, priority) { addHandler(protostr, handler, synchronous, priority) {
if (priority < 0 || priority > 4) { if (priority < 0 || priority > 4) {
throw "Invalid priority: " + priority; throw "Invalid priority: " + priority;
......
...@@ -196,16 +196,22 @@ export class YGOProMessagesHelper { ...@@ -196,16 +196,22 @@ export class YGOProMessagesHelper {
return sendBuffer; return sendBuffer;
} }
sendMessage(socket: net.Socket, protostr: string, info?: string | Buffer | any) { send(socket: net.Socket | WebSocket, buffer: Buffer) {
const sendBuffer = this.prepareMessage(protostr, info); return new Promise<Error | undefined>(done => {
socket.write(sendBuffer); if (socket['isWs']) {
const ws = socket as WebSocket;
// @ts-ignore
ws.send(buffer, {}, done);
} else {
const sock = socket as net.Socket;
sock.write(buffer, done);
}
})
} }
sendMessageAsync(socket: net.Socket, protostr: string, info?: string | Buffer | any): Promise<Error> { sendMessage(socket: net.Socket | WebSocket, protostr: string, info?: string | Buffer | any): Promise<Error> {
const sendBuffer = this.prepareMessage(protostr, info); const sendBuffer = this.prepareMessage(protostr, info);
return new Promise(done => { return this.send(socket, sendBuffer);
socket.write(sendBuffer, done);
});
} }
addHandler(protostr: string, handler: (buffer: Buffer, info: any, datas: Buffer[], params: any) => Promise<boolean | string>, synchronous: boolean, priority: number) { addHandler(protostr: string, handler: (buffer: Buffer, info: any, datas: Buffer[], params: any) => Promise<boolean | string>, synchronous: boolean, priority: number) {
......
...@@ -196,6 +196,15 @@ ...@@ -196,6 +196,15 @@
"banCount": 0, "banCount": 0,
"ttl": 600 "ttl": 600
}, },
"neos": {
"enabled": false,
"port": 7977,
"trusted_proxies": [
"127.0.0.1/8",
"::1/128"
],
"ip_header": "x-forwarded-for"
},
"test_mode": { "test_mode": {
"watch_public_hand": false, "watch_public_hand": false,
"no_connect_count_limit": false, "no_connect_count_limit": false,
......
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
"deepmerge": "^4.2.2", "deepmerge": "^4.2.2",
"formidable": "^1.2.6", "formidable": "^1.2.6",
"geoip-country-lite": "^1.0.0", "geoip-country-lite": "^1.0.0",
"ip6addr": "^0.2.5",
"jszip": "^3.5.0", "jszip": "^3.5.0",
"load-json-file": "^6.2.0", "load-json-file": "^6.2.0",
"lzma": "^2.3.2", "lzma": "^2.3.2",
...@@ -35,6 +36,7 @@ ...@@ -35,6 +36,7 @@
}, },
"devDependencies": { "devDependencies": {
"@types/bunyan": "^1.8.8", "@types/bunyan": "^1.8.8",
"@types/ip6addr": "^0.2.3",
"@types/node": "^17.0.19", "@types/node": "^17.0.19",
"@types/underscore": "^1.11.4", "@types/underscore": "^1.11.4",
"@types/ws": "^8.5.3", "@types/ws": "^8.5.3",
...@@ -79,6 +81,12 @@ ...@@ -79,6 +81,12 @@
"@types/node": "*" "@types/node": "*"
} }
}, },
"node_modules/@types/ip6addr": {
"version": "0.2.3",
"resolved": "https://registry.npmjs.org/@types/ip6addr/-/ip6addr-0.2.3.tgz",
"integrity": "sha512-oe7hzc+P9DU6+gql8+bLKuUf4WL4aakyCSXZMZq2cjhhGK75qYwH1zJ4s94XOlnb4cAhrGKwnbrmMBaqDK8+Ww==",
"dev": true
},
"node_modules/@types/node": { "node_modules/@types/node": {
"version": "17.0.19", "version": "17.0.19",
"resolved": "https://registry.npmjs.org/@types/node/-/node-17.0.19.tgz", "resolved": "https://registry.npmjs.org/@types/node/-/node-17.0.19.tgz",
...@@ -1187,6 +1195,34 @@ ...@@ -1187,6 +1195,34 @@
"node": "*" "node": "*"
} }
}, },
"node_modules/ip6addr": {
"version": "0.2.5",
"resolved": "https://registry.npmjs.org/ip6addr/-/ip6addr-0.2.5.tgz",
"integrity": "sha512-9RGGSB6Zc9Ox5DpDGFnJdIeF0AsqXzdH+FspCfPPaU/L/4tI6P+5lIoFUFm9JXs9IrJv1boqAaNCQmoDADTSKQ==",
"dependencies": {
"assert-plus": "^1.0.0",
"jsprim": "^2.0.2"
}
},
"node_modules/ip6addr/node_modules/json-schema": {
"version": "0.4.0",
"resolved": "https://registry.npmjs.org/json-schema/-/json-schema-0.4.0.tgz",
"integrity": "sha512-es94M3nTIfsEPisRafak+HDLfHXnKBhV3vU5eqPcS3flIWqcxJWgXHXiey3YrpaNsanY5ei1VoYEbOzijuq9BA=="
},
"node_modules/ip6addr/node_modules/jsprim": {
"version": "2.0.2",
"resolved": "https://registry.npmjs.org/jsprim/-/jsprim-2.0.2.tgz",
"integrity": "sha512-gqXddjPqQ6G40VdnI6T6yObEC+pDNvyP95wdQhkWkg7crHH3km5qP1FsOXEkzEQwnz6gz5qGTn1c2Y52wP3OyQ==",
"engines": [
"node >=0.6.0"
],
"dependencies": {
"assert-plus": "1.0.0",
"extsprintf": "1.3.0",
"json-schema": "0.4.0",
"verror": "1.10.0"
}
},
"node_modules/is-arrayish": { "node_modules/is-arrayish": {
"version": "0.2.1", "version": "0.2.1",
"resolved": "https://registry.npmjs.org/is-arrayish/-/is-arrayish-0.2.1.tgz", "resolved": "https://registry.npmjs.org/is-arrayish/-/is-arrayish-0.2.1.tgz",
...@@ -3112,6 +3148,12 @@ ...@@ -3112,6 +3148,12 @@
"@types/node": "*" "@types/node": "*"
} }
}, },
"@types/ip6addr": {
"version": "0.2.3",
"resolved": "https://registry.npmjs.org/@types/ip6addr/-/ip6addr-0.2.3.tgz",
"integrity": "sha512-oe7hzc+P9DU6+gql8+bLKuUf4WL4aakyCSXZMZq2cjhhGK75qYwH1zJ4s94XOlnb4cAhrGKwnbrmMBaqDK8+Ww==",
"dev": true
},
"@types/node": { "@types/node": {
"version": "17.0.19", "version": "17.0.19",
"resolved": "https://registry.npmjs.org/@types/node/-/node-17.0.19.tgz", "resolved": "https://registry.npmjs.org/@types/node/-/node-17.0.19.tgz",
...@@ -3954,6 +3996,33 @@ ...@@ -3954,6 +3996,33 @@
"resolved": "https://registry.npmjs.org/ini/-/ini-1.3.5.tgz", "resolved": "https://registry.npmjs.org/ini/-/ini-1.3.5.tgz",
"integrity": "sha512-RZY5huIKCMRWDUqZlEi72f/lmXKMvuszcMBduliQ3nnWbx9X/ZBQO7DijMEYS9EhHBb2qacRUMtC7svLwe0lcw==" "integrity": "sha512-RZY5huIKCMRWDUqZlEi72f/lmXKMvuszcMBduliQ3nnWbx9X/ZBQO7DijMEYS9EhHBb2qacRUMtC7svLwe0lcw=="
}, },
"ip6addr": {
"version": "0.2.5",
"resolved": "https://registry.npmjs.org/ip6addr/-/ip6addr-0.2.5.tgz",
"integrity": "sha512-9RGGSB6Zc9Ox5DpDGFnJdIeF0AsqXzdH+FspCfPPaU/L/4tI6P+5lIoFUFm9JXs9IrJv1boqAaNCQmoDADTSKQ==",
"requires": {
"assert-plus": "^1.0.0",
"jsprim": "^2.0.2"
},
"dependencies": {
"json-schema": {
"version": "0.4.0",
"resolved": "https://registry.npmjs.org/json-schema/-/json-schema-0.4.0.tgz",
"integrity": "sha512-es94M3nTIfsEPisRafak+HDLfHXnKBhV3vU5eqPcS3flIWqcxJWgXHXiey3YrpaNsanY5ei1VoYEbOzijuq9BA=="
},
"jsprim": {
"version": "2.0.2",
"resolved": "https://registry.npmjs.org/jsprim/-/jsprim-2.0.2.tgz",
"integrity": "sha512-gqXddjPqQ6G40VdnI6T6yObEC+pDNvyP95wdQhkWkg7crHH3km5qP1FsOXEkzEQwnz6gz5qGTn1c2Y52wP3OyQ==",
"requires": {
"assert-plus": "1.0.0",
"extsprintf": "1.3.0",
"json-schema": "0.4.0",
"verror": "1.10.0"
}
}
}
},
"is-arrayish": { "is-arrayish": {
"version": "0.2.1", "version": "0.2.1",
"resolved": "https://registry.npmjs.org/is-arrayish/-/is-arrayish-0.2.1.tgz", "resolved": "https://registry.npmjs.org/is-arrayish/-/is-arrayish-0.2.1.tgz",
......
...@@ -17,6 +17,7 @@ ...@@ -17,6 +17,7 @@
"deepmerge": "^4.2.2", "deepmerge": "^4.2.2",
"formidable": "^1.2.6", "formidable": "^1.2.6",
"geoip-country-lite": "^1.0.0", "geoip-country-lite": "^1.0.0",
"ip6addr": "^0.2.5",
"jszip": "^3.5.0", "jszip": "^3.5.0",
"load-json-file": "^6.2.0", "load-json-file": "^6.2.0",
"lzma": "^2.3.2", "lzma": "^2.3.2",
...@@ -46,6 +47,7 @@ ...@@ -46,6 +47,7 @@
}, },
"devDependencies": { "devDependencies": {
"@types/bunyan": "^1.8.8", "@types/bunyan": "^1.8.8",
"@types/ip6addr": "^0.2.3",
"@types/node": "^17.0.19", "@types/node": "^17.0.19",
"@types/underscore": "^1.11.4", "@types/underscore": "^1.11.4",
"@types/ws": "^8.5.3", "@types/ws": "^8.5.3",
......
// Generated by CoffeeScript 2.5.1 // Generated by CoffeeScript 2.6.1
(function() { (function() {
/* /*
Main script of new dashboard account system. Main script of new dashboard account system.
......
...@@ -590,17 +590,31 @@ init = () -> ...@@ -590,17 +590,31 @@ init = () ->
log.info "NOTE: server not open due to config, ", settings.modules.stop log.info "NOTE: server not open due to config, ", settings.modules.stop
http_server = http.createServer(httpRequestListener) http_server = http.createServer(httpRequestListener)
http_server.listen settings.modules.http.port main_http_server = http_server
if settings.modules.http.ssl.enabled if settings.modules.http.ssl.enabled
https = require 'https' https = require 'https'
options = httpsOptions =
cert: await fs.promises.readFile(settings.modules.http.ssl.cert) cert: await fs.promises.readFile(settings.modules.http.ssl.cert)
key: await fs.promises.readFile(settings.modules.http.ssl.key) key: await fs.promises.readFile(settings.modules.http.ssl.key)
https_server = https.createServer(options, httpRequestListener) https_server = https.createServer(httpsOptions, httpRequestListener)
if settings.modules.http.websocket_roomlist and roomlist
roomlist.init https_server, ROOM_all
https_server.listen settings.modules.http.ssl.port https_server.listen settings.modules.http.ssl.port
main_http_server = https_server
if settings.modules.http.websocket_roomlist and roomlist
roomlist.init main_http_server, ROOM_all
http_server.listen settings.modules.http.port
if settings.modules.neos.enabled
ws = require 'ws'
neosHttpServer = null
if settings.modules.http.ssl.enabled
neosHttpServer = https.createServer(httpsOptions, neosRequestListener)
else
neosHttpServer = http.createServer()
neosWsServer = new ws.WebSocketServer({server: neosHttpServer})
neosWsServer.on 'connection', neosRequestListener
neosHttpServer.listen settings.modules.neos.port
mkdirList = [ mkdirList = [
"./plugins", "./plugins",
...@@ -1149,9 +1163,9 @@ CLIENT_send_replays = global.CLIENT_send_replays = (client, room) -> ...@@ -1149,9 +1163,9 @@ CLIENT_send_replays = global.CLIENT_send_replays = (client, room) ->
SOCKET_flush_data = global.SOCKET_flush_data = (sk, datas) -> SOCKET_flush_data = global.SOCKET_flush_data = (sk, datas) ->
if !sk or sk.closed if !sk or sk.closed
return false return false
for buffer in datas while datas.length
sk.write(buffer) buffer = datas.shift()
datas.splice(0, datas.length) await ygopro.helper.send(sk, buffer)
return true return true
getSeedTimet = global.getSeedTimet = (count) -> getSeedTimet = global.getSeedTimet = (count) ->
...@@ -1347,7 +1361,7 @@ class Room ...@@ -1347,7 +1361,7 @@ class Room
@port = parseInt data @port = parseInt data
_.each @players, (player)=> _.each @players, (player)=>
player.server.connect @port, '127.0.0.1', -> player.server.connect @port, '127.0.0.1', ->
player.server.write buffer for buffer in player.pre_establish_buffers await ygopro.helper.send(player.server, buffer) for buffer in player.pre_establish_buffers
player.established = true player.established = true
player.pre_establish_buffers = [] player.pre_establish_buffers = []
return return
...@@ -1577,7 +1591,7 @@ class Room ...@@ -1577,7 +1591,7 @@ class Room
if @established if @established
roomlist.update(this) if !@windbot and @duel_stage == ygopro.constants.DUEL_STAGE.BEGIN and settings.modules.http.websocket_roomlist roomlist.update(this) if !@windbot and @duel_stage == ygopro.constants.DUEL_STAGE.BEGIN and settings.modules.http.websocket_roomlist
client.server.connect @port, '127.0.0.1', -> client.server.connect @port, '127.0.0.1', ->
client.server.write buffer for buffer in client.pre_establish_buffers await ygopro.helper.send(client.server, buffer) for buffer in client.pre_establish_buffers
client.established = true client.established = true
client.pre_establish_buffers = [] client.pre_establish_buffers = []
return return
...@@ -1724,7 +1738,7 @@ class Room ...@@ -1724,7 +1738,7 @@ class Room
@watchers.push client @watchers.push client
ygopro.stoc_send_chat(client, "${watch_watching}", ygopro.constants.COLORS.BABYBLUE) ygopro.stoc_send_chat(client, "${watch_watching}", ygopro.constants.COLORS.BABYBLUE)
for buffer in @watcher_buffers for buffer in @watcher_buffers
client.write buffer await ygopro.helper.send(client, buffer)
return true return true
else else
ygopro.stoc_die(client, "${watch_denied}") ygopro.stoc_die(client, "${watch_denied}")
...@@ -1758,7 +1772,8 @@ class Room ...@@ -1758,7 +1772,8 @@ class Room
# 网络连接 # 网络连接
netRequestHandler = (client) -> netRequestHandler = (client) ->
client.ip = client.remoteAddress if !client.isWs
client.ip = client.remoteAddress
client.is_local = client.ip and (client.ip.includes('127.0.0.1') or client.ip.includes(real_windbot_server_ip)) client.is_local = client.ip and (client.ip.includes('127.0.0.1') or client.ip.includes(real_windbot_server_ip))
connect_count = ROOM_connected_ip[client.ip] or 0 connect_count = ROOM_connected_ip[client.ip] or 0
...@@ -1780,46 +1795,38 @@ netRequestHandler = (client) -> ...@@ -1780,46 +1795,38 @@ netRequestHandler = (client) ->
client.setTimeout(2000) #连接前超时2秒 client.setTimeout(2000) #连接前超时2秒
# 释放处理 # 释放处理
client.on 'close', (had_error) -> closeHandler = (error) ->
#log.info "client closed", client.name, had_error #log.info "client closed", client.name, had_error
room=ROOM_all[client.rid]
connect_count = ROOM_connected_ip[client.ip]
if connect_count > 0
connect_count--
ROOM_connected_ip[client.ip] = connect_count
#log.info "disconnect", client.ip, ROOM_connected_ip[client.ip] #log.info "disconnect", client.ip, ROOM_connected_ip[client.ip]
unless client.closed if client.closed
client.closed = true return
if settings.modules.heartbeat_detection.enabled
CLIENT_heartbeat_unregister(client)
if room
if !CLIENT_reconnect_register(client, client.rid)
room.disconnect(client)
else if !client.had_new_reconnection
SERVER_kick(client.server)
return
client.on 'error', (error)->
#log.info "client error", client.name, error
room=ROOM_all[client.rid] room=ROOM_all[client.rid]
connect_count = ROOM_connected_ip[client.ip] connect_count = ROOM_connected_ip[client.ip]
if connect_count > 0 if connect_count > 0
connect_count-- connect_count--
ROOM_connected_ip[client.ip] = connect_count ROOM_connected_ip[client.ip] = connect_count
#log.info "err disconnect", client.ip, ROOM_connected_ip[client.ip] client.closed = true
unless client.closed if settings.modules.heartbeat_detection.enabled
client.closed = true CLIENT_heartbeat_unregister(client)
if room if room
if !CLIENT_reconnect_register(client, client.rid, error) if !CLIENT_reconnect_register(client, client.rid, error)
room.disconnect(client, error) room.disconnect(client)
else if !client.had_new_reconnection else if !client.had_new_reconnection
SERVER_kick(client.server) SERVER_kick(client.server)
return return
if client.isWs
client.on 'close', (code, reason) ->
closeHandler()
client.on 'timeout', ()->
unless settings.modules.reconnect.enabled and (disconnect_list[CLIENT_get_authorize_key(client)] or client.had_new_reconnection)
client.destroy()
return
else
client.on 'close', (had_error) ->
closeHandler(had_error ? 'unknown' : undefined)
client.on 'error', closeHandler
client.on 'timeout', ()->
unless settings.modules.reconnect.enabled and (disconnect_list[CLIENT_get_authorize_key(client)] or client.had_new_reconnection)
client.destroy()
return
server.on 'close', (had_error) -> server.on 'close', (had_error) ->
server.closed = true unless server.closed server.closed = true unless server.closed
...@@ -1867,9 +1874,8 @@ netRequestHandler = (client) -> ...@@ -1867,9 +1874,8 @@ netRequestHandler = (client) ->
ygopro.stoc_die(client, "${cloud_replay_error}") ygopro.stoc_die(client, "${cloud_replay_error}")
return return
ygopro.stoc_send_chat(client, "${cloud_replay_playing} #{replay.getDisplayString()}", ygopro.constants.COLORS.BABYBLUE) ygopro.stoc_send_chat(client, "${cloud_replay_playing} #{replay.getDisplayString()}", ygopro.constants.COLORS.BABYBLUE)
client.write replay_buffer, ()-> await ygopro.helper.send(client, replay_buffer)
CLIENT_kick(client) CLIENT_kick(client)
return
return return
# 需要重构 # 需要重构
...@@ -1877,7 +1883,7 @@ netRequestHandler = (client) -> ...@@ -1877,7 +1883,7 @@ netRequestHandler = (client) ->
client.pre_establish_buffers = new Array() client.pre_establish_buffers = new Array()
client.on 'data', (ctos_buffer) -> dataHandler = (ctos_buffer) ->
if client.is_post_watcher if client.is_post_watcher
room=ROOM_all[client.rid] room=ROOM_all[client.rid]
if room if room
...@@ -1895,7 +1901,7 @@ netRequestHandler = (client) -> ...@@ -1895,7 +1901,7 @@ netRequestHandler = (client) ->
ROOM_bad_ip[client.ip] = 1 ROOM_bad_ip[client.ip] = 1
CLIENT_kick(client) CLIENT_kick(client)
return return
room.watcher.write(buffer) for buffer in handle_data.datas await ygopro.helper.send(room.watcher, buffer) for buffer in handle_data.datas
else else
ctos_filter = null ctos_filter = null
preconnect = false preconnect = false
...@@ -1921,12 +1927,17 @@ netRequestHandler = (client) -> ...@@ -1921,12 +1927,17 @@ netRequestHandler = (client) ->
if client.closed || !client.server if client.closed || !client.server
return return
if client.established if client.established
client.server.write buffer for buffer in handle_data.datas await ygopro.helper.send(client.server, buffer) for buffer in handle_data.datas
else else
client.pre_establish_buffers.push buffer for buffer in handle_data.datas client.pre_establish_buffers = client.pre_establish_buffers.concat(handle_data.datas)
return return
if client.isWs
client.on 'message', dataHandler
else
client.on 'data', dataHandler
# 服务端到客户端(stoc) # 服务端到客户端(stoc)
server.on 'data', (stoc_buffer)-> server.on 'data', (stoc_buffer)->
handle_data = await ygopro.helper.handleBuffer(stoc_buffer, "STOC", null, { handle_data = await ygopro.helper.handleBuffer(stoc_buffer, "STOC", null, {
...@@ -1939,7 +1950,7 @@ netRequestHandler = (client) -> ...@@ -1939,7 +1950,7 @@ netRequestHandler = (client) ->
server.destroy() server.destroy()
return return
if server.client and !server.client.closed if server.client and !server.client.closed
server.client.write buffer for buffer in handle_data.datas await ygopro.helper.send(server.client, buffer) for buffer in handle_data.datas
return return
return return
...@@ -2416,7 +2427,7 @@ ygopro.stoc_follow 'JOIN_GAME', false, (buffer, info, client, server, datas)-> ...@@ -2416,7 +2427,7 @@ ygopro.stoc_follow 'JOIN_GAME', false, (buffer, info, client, server, datas)->
return unless room return unless room
room.watcher_buffers.push data room.watcher_buffers.push data
for w in room.watchers for w in room.watchers
w.write data if w #a WTF fix ygopro.helper.send(w, data) if w #a WTF fix
return return
watcher.on 'error', (error)-> watcher.on 'error', (error)->
...@@ -2850,7 +2861,7 @@ ygopro.stoc_follow 'FIELD_FINISH', true, (buffer, info, client, server, datas)-> ...@@ -2850,7 +2861,7 @@ ygopro.stoc_follow 'FIELD_FINISH', true, (buffer, info, client, server, datas)->
ygopro.stoc_follow 'DUEL_END', false, (buffer, info, client, server, datas)-> ygopro.stoc_follow 'DUEL_END', false, (buffer, info, client, server, datas)->
room=ROOM_all[client.rid] room=ROOM_all[client.rid]
return unless room and settings.modules.replay_delay and room.hostinfo.mode == 1 return unless room and settings.modules.replay_delay and room.hostinfo.mode == 1
SOCKET_flush_data(client, datas) await SOCKET_flush_data(client, datas)
CLIENT_send_replays(client, room) CLIENT_send_replays(client, room)
if !room.replays_sent_to_watchers if !room.replays_sent_to_watchers
room.replays_sent_to_watchers = true room.replays_sent_to_watchers = true
...@@ -3879,4 +3890,23 @@ if true ...@@ -3879,4 +3890,23 @@ if true
response.end() response.end()
return return
ip6addr = require('ip6addr')
neosRequestListener = (client, req) ->
physicalAddress = req.socket.remoteAddress
if settings.modules.neos.trusted_proxies.some((trusted) ->
cidr = if trusted.includes('/') then ip6addr.createCIDR(trusted) else ip6addr.createAddrRange(trusted, trusted)
return cidr.contains(physicalAddress)
)
ipHeader = req.headers[settings.modules.neos.trusted_proxy_header]
if ipHeader
client.ip = ipHeader.split(',')[0].trim()
if !client.ip
client.ip = physicalAddress
client.setTimeout = () -> true
client.destroy = () -> client.close()
client.isWs = true
netRequestHandler(client)
init() init()
// Generated by CoffeeScript 2.6.1 // Generated by CoffeeScript 2.6.1
(function() { (function() {
// 标准库 // 标准库
var CLIENT_get_authorize_key, CLIENT_get_kick_reconnect_target, CLIENT_heartbeat_register, CLIENT_heartbeat_unregister, CLIENT_import_data, CLIENT_is_able_to_kick_reconnect, CLIENT_is_able_to_reconnect, CLIENT_is_banned_by_mc, CLIENT_is_player, CLIENT_kick, CLIENT_kick_reconnect, CLIENT_pre_reconnect, CLIENT_reconnect, CLIENT_reconnect_register, CLIENT_reconnect_unregister, CLIENT_send_pre_reconnect_info, CLIENT_send_reconnect_info, CLIENT_send_replays, Q, ROOM_all, ROOM_bad_ip, ROOM_ban_player, ROOM_clear_disconnect, ROOM_connected_ip, ROOM_find_by_name, ROOM_find_by_pid, ROOM_find_by_port, ROOM_find_by_title, ROOM_find_or_create_ai, ROOM_find_or_create_by_name, ROOM_find_or_create_random, ROOM_kick, ROOM_player_flee, ROOM_player_get_score, ROOM_player_lose, ROOM_player_win, ROOM_players_oppentlist, ROOM_unwelcome, ROOM_validate, ReplayParser, ResolveData, Room, SERVER_clear_disconnect, SERVER_kick, SOCKET_flush_data, _, _async, addCallback, athleticChecker, auth, axios, badwordR, badwords, ban_user, bunyan, challonge, checkFileExists, createDirectoryIfNotExists, crypto, dataManager, deck_name_match, dialogues, disconnect_list, exec, execFile, fs, geoip, getDuelLogQueryFromQs, getSeedTimet, get_memory_usage, http, httpRequestListener, importOldConfig, import_datas, init, lflists, loadJSON, loadJSONAsync, loadLFList, loadRemoteData, load_dialogues, load_tips, log, long_resolve_cards, memory_usage, merge, moment, moment_long_ago_string, moment_now, moment_now_string, net, netRequestHandler, os, osu, path, qs, real_windbot_server_ip, release_disconnect, report_to_big_brother, request, roomlist, rooms_count, setting_change, setting_get, setting_save, settings, spawn, spawnSync, spawn_windbot, tips, url, users_cache, util, wait_room_start, wait_room_start_arena, windbot_looplimit, windbot_process, windbots, ygopro, zlib; var CLIENT_get_authorize_key, CLIENT_get_kick_reconnect_target, CLIENT_heartbeat_register, CLIENT_heartbeat_unregister, CLIENT_import_data, CLIENT_is_able_to_kick_reconnect, CLIENT_is_able_to_reconnect, CLIENT_is_banned_by_mc, CLIENT_is_player, CLIENT_kick, CLIENT_kick_reconnect, CLIENT_pre_reconnect, CLIENT_reconnect, CLIENT_reconnect_register, CLIENT_reconnect_unregister, CLIENT_send_pre_reconnect_info, CLIENT_send_reconnect_info, CLIENT_send_replays, Q, ROOM_all, ROOM_bad_ip, ROOM_ban_player, ROOM_clear_disconnect, ROOM_connected_ip, ROOM_find_by_name, ROOM_find_by_pid, ROOM_find_by_port, ROOM_find_by_title, ROOM_find_or_create_ai, ROOM_find_or_create_by_name, ROOM_find_or_create_random, ROOM_kick, ROOM_player_flee, ROOM_player_get_score, ROOM_player_lose, ROOM_player_win, ROOM_players_oppentlist, ROOM_unwelcome, ROOM_validate, ReplayParser, ResolveData, Room, SERVER_clear_disconnect, SERVER_kick, SOCKET_flush_data, _, _async, addCallback, athleticChecker, auth, axios, badwordR, badwords, ban_user, bunyan, challonge, checkFileExists, createDirectoryIfNotExists, crypto, dataManager, deck_name_match, dialogues, disconnect_list, exec, execFile, fs, geoip, getDuelLogQueryFromQs, getSeedTimet, get_memory_usage, http, httpRequestListener, importOldConfig, import_datas, init, ip6addr, lflists, loadJSON, loadJSONAsync, loadLFList, loadRemoteData, load_dialogues, load_tips, log, long_resolve_cards, memory_usage, merge, moment, moment_long_ago_string, moment_now, moment_now_string, neosRequestListener, net, netRequestHandler, os, osu, path, qs, real_windbot_server_ip, release_disconnect, report_to_big_brother, request, roomlist, rooms_count, setting_change, setting_get, setting_save, settings, spawn, spawnSync, spawn_windbot, tips, url, users_cache, util, wait_room_start, wait_room_start_arena, windbot_looplimit, windbot_process, windbots, ygopro, zlib;
net = require('net'); net = require('net');
...@@ -304,7 +304,7 @@ ...@@ -304,7 +304,7 @@
}; };
init = async function() { init = async function() {
var AthleticChecker, Challonge, DataManager, chat_color, config, cppversion, defaultConfig, default_data, dirPath, dns, e, get_rooms_count, http_server, https, https_server, imported, j, key, keysFromEnv, l, len, len1, len2, m, mkdirList, options, pgClient, pg_client, pg_query, plugin_filename, plugin_list, plugin_path, postData, settingKey, val, valFromDefault; var AthleticChecker, Challonge, DataManager, chat_color, config, cppversion, defaultConfig, default_data, dirPath, dns, e, get_rooms_count, http_server, https, httpsOptions, https_server, imported, j, key, keysFromEnv, l, len, len1, len2, m, main_http_server, mkdirList, neosHttpServer, neosWsServer, pgClient, pg_client, pg_query, plugin_filename, plugin_list, plugin_path, postData, settingKey, val, valFromDefault, ws;
log.info('Reading config.'); log.info('Reading config.');
await createDirectoryIfNotExists("./config"); await createDirectoryIfNotExists("./config");
await importOldConfig(); await importOldConfig();
...@@ -764,18 +764,34 @@ ...@@ -764,18 +764,34 @@
log.info("NOTE: server not open due to config, ", settings.modules.stop); log.info("NOTE: server not open due to config, ", settings.modules.stop);
} }
http_server = http.createServer(httpRequestListener); http_server = http.createServer(httpRequestListener);
http_server.listen(settings.modules.http.port); main_http_server = http_server;
if (settings.modules.http.ssl.enabled) { if (settings.modules.http.ssl.enabled) {
https = require('https'); https = require('https');
options = { httpsOptions = {
cert: (await fs.promises.readFile(settings.modules.http.ssl.cert)), cert: (await fs.promises.readFile(settings.modules.http.ssl.cert)),
key: (await fs.promises.readFile(settings.modules.http.ssl.key)) key: (await fs.promises.readFile(settings.modules.http.ssl.key))
}; };
https_server = https.createServer(options, httpRequestListener); https_server = https.createServer(httpsOptions, httpRequestListener);
if (settings.modules.http.websocket_roomlist && roomlist) {
roomlist.init(https_server, ROOM_all);
}
https_server.listen(settings.modules.http.ssl.port); https_server.listen(settings.modules.http.ssl.port);
main_http_server = https_server;
}
if (settings.modules.http.websocket_roomlist && roomlist) {
roomlist.init(main_http_server, ROOM_all);
}
http_server.listen(settings.modules.http.port);
if (settings.modules.neos.enabled) {
ws = require('ws');
neosHttpServer = null;
if (settings.modules.http.ssl.enabled) {
neosHttpServer = https.createServer(httpsOptions, neosRequestListener);
} else {
neosHttpServer = http.createServer();
}
neosWsServer = new ws.WebSocketServer({
server: neosHttpServer
});
neosWsServer.on('connection', neosRequestListener);
neosHttpServer.listen(settings.modules.neos.port);
} }
mkdirList = ["./plugins", settings.modules.tournament_mode.deck_path, settings.modules.tournament_mode.replay_path, settings.modules.tournament_mode.log_save_path, settings.modules.deck_log.local]; mkdirList = ["./plugins", settings.modules.tournament_mode.deck_path, settings.modules.tournament_mode.replay_path, settings.modules.tournament_mode.log_save_path, settings.modules.deck_log.local];
for (l = 0, len1 = mkdirList.length; l < len1; l++) { for (l = 0, len1 = mkdirList.length; l < len1; l++) {
...@@ -1515,16 +1531,15 @@ ...@@ -1515,16 +1531,15 @@
return true; return true;
}; };
SOCKET_flush_data = global.SOCKET_flush_data = function(sk, datas) { SOCKET_flush_data = global.SOCKET_flush_data = async function(sk, datas) {
var buffer, j, len; var buffer;
if (!sk || sk.closed) { if (!sk || sk.closed) {
return false; return false;
} }
for (j = 0, len = datas.length; j < len; j++) { while (datas.length) {
buffer = datas[j]; buffer = datas.shift();
sk.write(buffer); await ygopro.helper.send(sk, buffer);
} }
datas.splice(0, datas.length);
return true; return true;
}; };
...@@ -1756,12 +1771,12 @@ ...@@ -1756,12 +1771,12 @@
} }
this.port = parseInt(data); this.port = parseInt(data);
_.each(this.players, (player) => { _.each(this.players, (player) => {
player.server.connect(this.port, '127.0.0.1', function() { player.server.connect(this.port, '127.0.0.1', async function() {
var buffer, len, m, ref; var buffer, len, m, ref;
ref = player.pre_establish_buffers; ref = player.pre_establish_buffers;
for (m = 0, len = ref.length; m < len; m++) { for (m = 0, len = ref.length; m < len; m++) {
buffer = ref[m]; buffer = ref[m];
player.server.write(buffer); await ygopro.helper.send(player.server, buffer);
} }
player.established = true; player.established = true;
player.pre_establish_buffers = []; player.pre_establish_buffers = [];
...@@ -2097,12 +2112,12 @@ ...@@ -2097,12 +2112,12 @@
if (!this.windbot && this.duel_stage === ygopro.constants.DUEL_STAGE.BEGIN && settings.modules.http.websocket_roomlist) { if (!this.windbot && this.duel_stage === ygopro.constants.DUEL_STAGE.BEGIN && settings.modules.http.websocket_roomlist) {
roomlist.update(this); roomlist.update(this);
} }
client.server.connect(this.port, '127.0.0.1', function() { client.server.connect(this.port, '127.0.0.1', async function() {
var buffer, j, len, ref; var buffer, j, len, ref;
ref = client.pre_establish_buffers; ref = client.pre_establish_buffers;
for (j = 0, len = ref.length; j < len; j++) { for (j = 0, len = ref.length; j < len; j++) {
buffer = ref[j]; buffer = ref[j];
client.server.write(buffer); await ygopro.helper.send(client.server, buffer);
} }
client.established = true; client.established = true;
client.pre_establish_buffers = []; client.pre_establish_buffers = [];
...@@ -2316,7 +2331,7 @@ ...@@ -2316,7 +2331,7 @@
})); }));
} }
join_post_watch(client) { async join_post_watch(client) {
var buffer, j, len, ref; var buffer, j, len, ref;
if (this.duel_stage !== ygopro.constants.DUEL_STAGE.BEGIN) { if (this.duel_stage !== ygopro.constants.DUEL_STAGE.BEGIN) {
if (settings.modules.cloud_replay.enable_halfway_watch && !this.hostinfo.no_watch) { if (settings.modules.cloud_replay.enable_halfway_watch && !this.hostinfo.no_watch) {
...@@ -2329,7 +2344,7 @@ ...@@ -2329,7 +2344,7 @@
ref = this.watcher_buffers; ref = this.watcher_buffers;
for (j = 0, len = ref.length; j < len; j++) { for (j = 0, len = ref.length; j < len; j++) {
buffer = ref[j]; buffer = ref[j];
client.write(buffer); await ygopro.helper.send(client, buffer);
} }
return true; return true;
} else { } else {
...@@ -2383,8 +2398,10 @@ ...@@ -2383,8 +2398,10 @@
// 网络连接 // 网络连接
netRequestHandler = function(client) { netRequestHandler = function(client) {
var connect_count, server; var closeHandler, connect_count, dataHandler, server;
client.ip = client.remoteAddress; if (!client.isWs) {
client.ip = client.remoteAddress;
}
client.is_local = client.ip && (client.ip.includes('127.0.0.1') || client.ip.includes(real_windbot_server_ip)); client.is_local = client.ip && (client.ip.includes('127.0.0.1') || client.ip.includes(real_windbot_server_ip));
connect_count = ROOM_connected_ip[client.ip] || 0; connect_count = ROOM_connected_ip[client.ip] || 0;
if (!settings.modules.test_mode.no_connect_count_limit && !client.is_local) { if (!settings.modules.test_mode.no_connect_count_limit && !client.is_local) {
...@@ -2404,56 +2421,48 @@ ...@@ -2404,56 +2421,48 @@
client.setTimeout(2000); //连接前超时2秒 client.setTimeout(2000); //连接前超时2秒
// 释放处理 // 释放处理
client.on('close', function(had_error) { closeHandler = function(error) {
var room; var room;
//log.info "client closed", client.name, had_error //log.info "client closed", client.name, had_error
room = ROOM_all[client.rid];
connect_count = ROOM_connected_ip[client.ip];
if (connect_count > 0) {
connect_count--;
}
ROOM_connected_ip[client.ip] = connect_count;
//log.info "disconnect", client.ip, ROOM_connected_ip[client.ip] //log.info "disconnect", client.ip, ROOM_connected_ip[client.ip]
if (!client.closed) { if (client.closed) {
client.closed = true; return;
if (settings.modules.heartbeat_detection.enabled) {
CLIENT_heartbeat_unregister(client);
}
if (room) {
if (!CLIENT_reconnect_register(client, client.rid)) {
room.disconnect(client);
}
} else if (!client.had_new_reconnection) {
SERVER_kick(client.server);
}
} }
});
client.on('error', function(error) {
var room;
//log.info "client error", client.name, error
room = ROOM_all[client.rid]; room = ROOM_all[client.rid];
connect_count = ROOM_connected_ip[client.ip]; connect_count = ROOM_connected_ip[client.ip];
if (connect_count > 0) { if (connect_count > 0) {
connect_count--; connect_count--;
} }
ROOM_connected_ip[client.ip] = connect_count; ROOM_connected_ip[client.ip] = connect_count;
//log.info "err disconnect", client.ip, ROOM_connected_ip[client.ip] client.closed = true;
if (!client.closed) { if (settings.modules.heartbeat_detection.enabled) {
client.closed = true; CLIENT_heartbeat_unregister(client);
if (room) {
if (!CLIENT_reconnect_register(client, client.rid, error)) {
room.disconnect(client, error);
}
} else if (!client.had_new_reconnection) {
SERVER_kick(client.server);
}
} }
}); if (room) {
client.on('timeout', function() { if (!CLIENT_reconnect_register(client, client.rid, error)) {
if (!(settings.modules.reconnect.enabled && (disconnect_list[CLIENT_get_authorize_key(client)] || client.had_new_reconnection))) { room.disconnect(client);
client.destroy(); }
} else if (!client.had_new_reconnection) {
SERVER_kick(client.server);
} }
}); };
if (client.isWs) {
client.on('close', function(code, reason) {
return closeHandler();
});
client.on('timeout', function() {
if (!(settings.modules.reconnect.enabled && (disconnect_list[CLIENT_get_authorize_key(client)] || client.had_new_reconnection))) {
client.destroy();
}
});
} else {
client.on('close', function(had_error) {
return closeHandler(had_error != null ? had_error : {
'unknown': void 0
});
});
}
client.on('error', closeHandler);
server.on('close', function(had_error) { server.on('close', function(had_error) {
var room; var room;
if (!server.closed) { if (!server.closed) {
...@@ -2514,16 +2523,15 @@ ...@@ -2514,16 +2523,15 @@
return; return;
} }
ygopro.stoc_send_chat(client, `\${cloud_replay_playing} ${replay.getDisplayString()}`, ygopro.constants.COLORS.BABYBLUE); ygopro.stoc_send_chat(client, `\${cloud_replay_playing} ${replay.getDisplayString()}`, ygopro.constants.COLORS.BABYBLUE);
client.write(replay_buffer, function() { await ygopro.helper.send(client, replay_buffer);
CLIENT_kick(client); CLIENT_kick(client);
});
}; };
} }
// 需要重构 // 需要重构
// 客户端到服务端(ctos)协议分析 // 客户端到服务端(ctos)协议分析
client.pre_establish_buffers = new Array(); client.pre_establish_buffers = new Array();
client.on('data', async function(ctos_buffer) { dataHandler = async function(ctos_buffer) {
var bad_ip_count, buffer, ctos_filter, handle_data, j, l, len, len1, len2, m, preconnect, ref, ref1, ref2, room; var bad_ip_count, buffer, ctos_filter, handle_data, j, l, len, len1, preconnect, ref, ref1, room;
if (client.is_post_watcher) { if (client.is_post_watcher) {
room = ROOM_all[client.rid]; room = ROOM_all[client.rid];
if (room) { if (room) {
...@@ -2547,7 +2555,7 @@ ...@@ -2547,7 +2555,7 @@
ref = handle_data.datas; ref = handle_data.datas;
for (j = 0, len = ref.length; j < len; j++) { for (j = 0, len = ref.length; j < len; j++) {
buffer = ref[j]; buffer = ref[j];
room.watcher.write(buffer); await ygopro.helper.send(room.watcher, buffer);
} }
} }
} else { } else {
...@@ -2584,17 +2592,18 @@ ...@@ -2584,17 +2592,18 @@
ref1 = handle_data.datas; ref1 = handle_data.datas;
for (l = 0, len1 = ref1.length; l < len1; l++) { for (l = 0, len1 = ref1.length; l < len1; l++) {
buffer = ref1[l]; buffer = ref1[l];
client.server.write(buffer); await ygopro.helper.send(client.server, buffer);
} }
} else { } else {
ref2 = handle_data.datas; client.pre_establish_buffers = client.pre_establish_buffers.concat(handle_data.datas);
for (m = 0, len2 = ref2.length; m < len2; m++) {
buffer = ref2[m];
client.pre_establish_buffers.push(buffer);
}
} }
} }
}); };
if (client.isWs) {
client.on('message', dataHandler);
} else {
client.on('data', dataHandler);
}
// 服务端到客户端(stoc) // 服务端到客户端(stoc)
server.on('data', async function(stoc_buffer) { server.on('data', async function(stoc_buffer) {
var buffer, handle_data, j, len, ref; var buffer, handle_data, j, len, ref;
...@@ -2613,7 +2622,7 @@ ...@@ -2613,7 +2622,7 @@
ref = handle_data.datas; ref = handle_data.datas;
for (j = 0, len = ref.length; j < len; j++) { for (j = 0, len = ref.length; j < len; j++) {
buffer = ref[j]; buffer = ref[j];
server.client.write(buffer); await ygopro.helper.send(server.client, buffer);
} }
} }
}); });
...@@ -3177,7 +3186,7 @@ ...@@ -3177,7 +3186,7 @@
for (l = 0, len1 = ref1.length; l < len1; l++) { for (l = 0, len1 = ref1.length; l < len1; l++) {
w = ref1[l]; w = ref1[l];
if (w) { //a WTF fix if (w) { //a WTF fix
w.write(data); ygopro.helper.send(w, data);
} }
} }
}); });
...@@ -3781,7 +3790,7 @@ ...@@ -3781,7 +3790,7 @@
if (!(room && settings.modules.replay_delay && room.hostinfo.mode === 1)) { if (!(room && settings.modules.replay_delay && room.hostinfo.mode === 1)) {
return; return;
} }
SOCKET_flush_data(client, datas); await SOCKET_flush_data(client, datas);
CLIENT_send_replays(client, room); CLIENT_send_replays(client, room);
if (!room.replays_sent_to_watchers) { if (!room.replays_sent_to_watchers) {
room.replays_sent_to_watchers = true; room.replays_sent_to_watchers = true;
...@@ -5169,6 +5178,34 @@ ...@@ -5169,6 +5178,34 @@
}; };
} }
ip6addr = require('ip6addr');
neosRequestListener = function(client, req) {
var ipHeader, physicalAddress;
physicalAddress = req.socket.remoteAddress;
if (settings.modules.neos.trusted_proxies.some(function(trusted) {
var cidr;
cidr = trusted.includes('/') ? ip6addr.createCIDR(trusted) : ip6addr.createAddrRange(trusted, trusted);
return cidr.contains(physicalAddress);
})) {
ipHeader = req.headers[settings.modules.neos.trusted_proxy_header];
if (ipHeader) {
client.ip = ipHeader.split(',')[0].trim();
}
}
if (!client.ip) {
client.ip = physicalAddress;
}
client.setTimeout = function() {
return true;
};
client.destroy = function() {
return client.close();
};
client.isWs = true;
return netRequestHandler(client);
};
init(); init();
}).call(this); }).call(this);
...@@ -68,7 +68,7 @@ translateHandler = (handler) -> ...@@ -68,7 +68,7 @@ translateHandler = (handler) ->
line="[Server]: "+line line="[Server]: "+line
for o,r of @i18nR[client.lang] for o,r of @i18nR[client.lang]
line=line.replace(r.regex, r.text) line=line.replace(r.regex, r.text)
@stoc_send client, 'CHAT', { await @stoc_send client, 'CHAT', {
player: player player: player
msg: line msg: line
} }
...@@ -105,8 +105,8 @@ translateHandler = (handler) -> ...@@ -105,8 +105,8 @@ translateHandler = (handler) ->
return return
@stoc_die = (client, msg)-> @stoc_die = (client, msg)->
@stoc_send_chat(client, msg, @constants.COLORS.RED) await @stoc_send_chat(client, msg, @constants.COLORS.RED)
@stoc_send client, 'ERROR_MSG', { await @stoc_send client, 'ERROR_MSG', {
msg: 1 msg: 1
code: 9 code: 9
} if client } if client
......
// Generated by CoffeeScript 2.5.1 // Generated by CoffeeScript 2.6.1
(function() { (function() {
var Struct, YGOProMessagesHelper, _, loadJSON, translateHandler; var Struct, YGOProMessagesHelper, _, loadJSON, translateHandler;
...@@ -95,7 +95,7 @@ ...@@ -95,7 +95,7 @@
}; };
//util //util
this.stoc_send_chat = function(client, msg, player = 8) { this.stoc_send_chat = async function(client, msg, player = 8) {
var i, len, line, o, r, ref, ref1; var i, len, line, o, r, ref, ref1;
if (!client) { if (!client) {
console.log("err stoc_send_chat"); console.log("err stoc_send_chat");
...@@ -112,7 +112,7 @@ ...@@ -112,7 +112,7 @@
r = ref1[o]; r = ref1[o];
line = line.replace(r.regex, r.text); line = line.replace(r.regex, r.text);
} }
this.stoc_send(client, 'CHAT', { await this.stoc_send(client, 'CHAT', {
player: player, player: player,
msg: line msg: line
}); });
...@@ -173,10 +173,10 @@ ...@@ -173,10 +173,10 @@
} }
}; };
this.stoc_die = function(client, msg) { this.stoc_die = async function(client, msg) {
this.stoc_send_chat(client, msg, this.constants.COLORS.RED); await this.stoc_send_chat(client, msg, this.constants.COLORS.RED);
if (client) { if (client) {
this.stoc_send(client, 'ERROR_MSG', { await this.stoc_send(client, 'ERROR_MSG', {
msg: 1, msg: 1,
code: 9 code: 9
}); });
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment