added: logger, session destroy

2022-11-29 13:59:44 +05:00
parent 67ee5974be
commit 12080c106a
14 changed files with 240 additions and 3746 deletions
@@ -14,40 +14,12 @@ const crypto = require('crypto')

 var querystring = require('querystring')

-// write startup server history
-var logs_dir = './logs'
-if (!fs.existsSync(logs_dir))
-    fs.mkdirSync(logs_dir)
+const http_client = require('../lib/http_client.js')
+const logger = require('../lib/logger')

-fs.appendFileSync(`./${logs_dir}/launch_history.log`, (node_time.create().format('Y-m-d H:M:S')).toString() + '\n')
-
-async function PostSessionServer(data, ip, port, callback) {
-    // Build the post string from an object
-    // An object of options to indicate where to post to
-    var post_options = {
-        host: ip,
-        port: port,
-        method: 'POST',
-        headers: {
-            'Content-Type': 'applicaiton/json',
-            'Content-Length': Buffer.byteLength(data)
-        }
-    }
-    // Set up the request
-    var request = http.request(post_options, function(res) {
-        res.setEncoding('utf8')
-        res.on('data', async function (answer) {
-            callback(answer)
-        })
-    })
-    request.on('error', function(e) {
-        console.error(e)
-        callback('{"msg":"SESSION_SERVER_DISABLED"}')
-    })
-    // post the data
-    request.write(data)
-    request.end()
-}
+// logger
+const logger_init = new logger('./logs/init.log')
+const logger_runtime = new logger('./logs/runtime.log')

 // response function
 async function create_response(request, callback) {
@@ -59,7 +31,6 @@ async function create_response(request, callback) {
        if (request.verb == 'GET_TITLES') {
            var title = database.collection('title')
            response = await title.find().skip(request.start).limit(request.count).toArray()
-            response = JSON.stringify(response)
            callback(response)

        } else if (request.verb == 'PLAN_SESSION') {
@@ -72,7 +43,7 @@ async function create_response(request, callback) {
            var servers = await session_server.find({'title':request.title}).toArray()
            
            if (!servers.length) {
-                response = '{"msg":"SERVERS_NOT_FOUND"}'
+                response = {msg:'SERVERS_NOT_FOUND'}
                callback(response)
                return
            }
@@ -90,35 +61,58 @@ async function create_response(request, callback) {
                }
            }
            if (free_server == null) {
-                response = '{"msg":"SESSION_LIMIT_REACHED"}'
+                response = {msg:'SESSION_LIMIT_REACHED'}
                callback(response)
                return
            }
-            PostSessionServer(`{"verb":"CREATE_SESSION", "title":"${request.title}"}`, free_server.ip, free_server.port, async function(answer) {
+            
+            var session_id = crypto.randomBytes(16).toString('base64')
+
+            // create session on session server
+            new http_client(free_server.ip, free_server.port).post(
+                `{"verb":"CREATE_SESSION", "title":"${request.title}", "session_id":"${session_id}"}`, async function(answer) {
+
                var json_answer = await JSON.parse(answer)

                if (json_answer.link == null) {
-                    response = '{"msg":"SESSION_SERVER_NOT_WORKING"}'
+                    response = {msg:'SESSION_SERVER_NOT_WORKING'}
                    callback(response)
                    return
                }
+
+                // generate random code for session access
+                var code = Math.floor(1000 + Math.random() * 9000)
+                while ((await session_active.find({'code':code}).toArray()).length) {
+                    code = Math.floor(1000 + Math.random() * 9000)
+                }
+
                // add session to database
                await session_active.insertOne({
                    server_id:free_server.server_id,
-                    link:json_answer.link
+                    session_id:session_id,
+                    link:json_answer.link,
+                    connection_code:code
                })
-                response = `{"msg":"SESSION_CREATED", "link":"${json_answer.link}"}`
+                response = {msg:'SESSION_CREATED', link:json_answer.link}
                callback(response)
+                return
+            },
+            function(error) {
+                logger_runtime.error(error)
+                callback({msg:'SESSION_SERVER_DISABLED'})
+                return
            })

        } else if (request.verb == 'CONNECT_SESSION') {
            callback(response)
+            return
        } else {
-            response = '{"msg":"UNKNOWN_VERB"}'
+            response = {msg:'UNKNOWN_VERB'}
            callback(response)
+            return
        }
    } catch(e) {
-        console.error(e)
+        logger_runtime.error(e)
    }
 }

@@ -127,32 +121,32 @@ async function create_response(request, callback) {
 const server = http.createServer(function(request, response) {
    if (request.method == 'POST') {
        var body = ''
-        
        request.on('data', function(data) {
            body += data
        })
-
        request.on('end', function() {
-            console.log('received: "', body, '"')
-            response.writeHead(200, {'Content-Type': 'message'})
+            logger_runtime.log('received: ', body)
+            response.writeHead(200, {'Content-Type': 'application/json'})
            try {
                create_response(JSON.parse(body), function(message) {
-                    response.end(message)
-                    console.log('response:', message)
+                    response.end(JSON.stringify(message))
+                    logger_runtime.log('response:', message)
                })
            }
            catch(e) {
-                console.error(e)
+                logger_runtime.error(e)
            }
        })
    }
    else {
-        var message = 'UNDEFINED_REQUEST' + ': ' + request.method.toString()
-        response.writeHead(200, {'Content-Type': 'message'})
-        response.end(message)
-        console.log(message)
+        response.writeHead(501, {'Content-Type': 'application/json'})
+        response.end(JSON.stringify({msg:'UNKNOWN_METHOD'}))
    }
 })

-server.listen(config.port, config.ip)
-console.log(`Listening at http://${config.ip}:${config.port}`)
+try {
+    server.listen(config.port, config.ip)
+    logger_init.log(`Listening at http://${config.ip}:${config.port}`)
+} catch (e) {
+    logger_init.error(e)
+}
@@ -1,18 +0,0 @@
-{
-  "name": "coordinator",
-  "version": "1.0.0",
-  "description": "",
-  "main": "coordinator.js",
-  "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
-  },
-  "keywords": [],
-  "author": "",
-  "license": "ISC",
-  "type": "commonjs",
-  "dependencies": {
-    "express": "^4.18.2",
-    "mongodb": "^4.12.0",
-    "node-datetime": "^2.1.2"
-  }
-}
@@ -0,0 +1,36 @@
+const http = require('http')
+
+module.exports = class http_client {
+    constructor(ip, port) {
+        this.#ip = ip
+        this.#port = port
+    }
+    async post(data, callback_answer, callback_error) {
+        // Build the post string from an object
+        // An object of options to indicate where to post to
+        var post_options = {
+            host: this.#ip,
+            port: this.#port,
+            method: 'POST',
+            headers: {
+                'Content-Type': 'applicaiton/json',
+                'Content-Length': Buffer.byteLength(data)
+            }
+        }
+        // Set up the request
+        var request = http.request(post_options, function(res) {
+            res.setEncoding('utf8')
+            res.on('data', async function (answer) {
+                callback_answer(answer)
+            })
+        })
+        request.on('error', function(e) {
+            callback_error(e)
+        })
+        // post the data
+        request.write(data)
+        request.end()
+    }
+    #ip
+    #port
+}
@@ -0,0 +1,34 @@
+const fs = require('fs')
+const path = require('path')
+const node_time = require('node-datetime')
+const util = require('util')
+
+module.exports = class logger {
+    constructor(filepath) {
+        this.#filepath = filepath
+        this.#filename = path.basename(filepath)
+        var dirname = path.dirname(filepath)
+        
+        if (!fs.existsSync(dirname))
+            fs.mkdirSync(dirname, {recursive:true})
+    }
+    error(...arg) {
+        this.#log('error', arg)
+    }
+    log(...arg) {
+        this.#log('', arg)
+    }
+
+    #log(message = '', ...arg) {
+        var args = ''
+        arg.forEach(value => {
+            args += util.format(value) + ' '
+        })
+        args = (node_time.create().format('Y-m-d H:M:S')).toString() + ((message != '') ? ': ' + message : '') + ': ' + args
+        console.log(this.#filename + ': ' + args)
+        fs.appendFileSync(this.#filepath, args + '\n')
+    }
+    
+    #filepath
+    #filename
+}
@@ -1,4 +1,3 @@
-
 class port {
    constructor(value) {
        this.#value = value
@@ -11,7 +11,8 @@
      "dependencies": {
        "express": "^4.18.2",
        "mongodb": "^4.12.0",
-        "node-datetime": "^2.1.2"
+        "node-datetime": "^2.1.2",
+        "signal": "^7.0.6"
      }
    },
    "node_modules/@aws-crypto/ie11-detection": {
@@ -1592,6 +1593,11 @@
        "node": ">= 0.8"
      }
    },
+    "node_modules/operation": {
+      "version": "5.0.2",
+      "resolved": "https://registry.npmjs.org/operation/-/operation-5.0.2.tgz",
+      "integrity": "sha512-tFuFcNZMND6vA40IX+faGKE4xrsM9IIqgHf4peTqB1aMLIB0RwjFMzMq5DP0SgWuUEcKGWJFqOCh7xlH2lP50A=="
+    },
    "node_modules/parseurl": {
      "version": "1.3.3",
      "resolved": "https://registry.npmjs.org/parseurl/-/parseurl-1.3.3.tgz",
@@ -1757,6 +1763,14 @@
        "url": "https://github.com/sponsors/ljharb"
      }
    },
+    "node_modules/signal": {
+      "version": "7.0.6",
+      "resolved": "https://registry.npmjs.org/signal/-/signal-7.0.6.tgz",
+      "integrity": "sha512-9SnDOIATtoRktT5ePD68F+B5i3ZugSBHB1HrBhaI/kkhaMBaMMatoDn+ajdgnFE5Xz1hWliLkpwVEnj3+S+ntg==",
+      "dependencies": {
+        "operation": "5.0.x"
+      }
+    },
    "node_modules/smart-buffer": {
      "version": "4.2.0",
      "resolved": "https://registry.npmjs.org/smart-buffer/-/smart-buffer-4.2.0.tgz",
@@ -3163,6 +3177,11 @@
        "ee-first": "1.1.1"
      }
    },
+    "operation": {
+      "version": "5.0.2",
+      "resolved": "https://registry.npmjs.org/operation/-/operation-5.0.2.tgz",
+      "integrity": "sha512-tFuFcNZMND6vA40IX+faGKE4xrsM9IIqgHf4peTqB1aMLIB0RwjFMzMq5DP0SgWuUEcKGWJFqOCh7xlH2lP50A=="
+    },
    "parseurl": {
      "version": "1.3.3",
      "resolved": "https://registry.npmjs.org/parseurl/-/parseurl-1.3.3.tgz",
@@ -3283,6 +3302,14 @@
        "object-inspect": "^1.9.0"
      }
    },
+    "signal": {
+      "version": "7.0.6",
+      "resolved": "https://registry.npmjs.org/signal/-/signal-7.0.6.tgz",
+      "integrity": "sha512-9SnDOIATtoRktT5ePD68F+B5i3ZugSBHB1HrBhaI/kkhaMBaMMatoDn+ajdgnFE5Xz1hWliLkpwVEnj3+S+ntg==",
+      "requires": {
+        "operation": "5.0.x"
+      }
+    },
    "smart-buffer": {
      "version": "4.2.0",
      "resolved": "https://registry.npmjs.org/smart-buffer/-/smart-buffer-4.2.0.tgz",
@@ -13,6 +13,7 @@
  "dependencies": {
    "express": "^4.18.2",
    "mongodb": "^4.12.0",
-    "node-datetime": "^2.1.2"
+    "node-datetime": "^2.1.2",
+    "signal": "^7.0.6"
  }
 }
@@ -4,7 +4,7 @@ const http = require('http')
 // filesystem
 const fs = require('fs')
 // server config
-const config = require('./config.json');
+const config = require('./config.json')
 // node date time
 const node_time = require('node-datetime')
 // titles for process creation
@@ -12,16 +12,15 @@ const titles = require('./titles.json')

 const { spawn } = require('node:child_process')

-const port_alloc = require('./port_alloc.js')
-
 const child_process = require('child_process')

-// write startup server history
-var logs_dir = './logs'
-if (!fs.existsSync(logs_dir))
-    fs.mkdirSync(logs_dir)
+const port_alloc = require('../lib/port_alloc.js')
+const http_client = require('../lib/http_client.js')
+const logger = require('../lib/logger')

-fs.appendFileSync(`./${logs_dir}/launch_history.log`, (node_time.create().format('Y-m-d H:M:S')).toString() + '\n')
+// logger
+const logger_init = new logger('./logs/init.log')
+const logger_runtime = new logger('./logs/runtime.log')

 var app_port_alloc = new port_alloc(config.session_ports.app_begin, config.session_ports.count)
 var http_port_alloc = new port_alloc(config.session_ports.http_begin, config.session_ports.count)
@@ -30,73 +29,83 @@ var http_port_alloc = new port_alloc(config.session_ports.http_begin, config.ses
 async function create_response(request, callback) {
    var response = ''
    try {
-        switch(request.verb) {
-            case 'CREATE_SESSION':
-                var app_info
-                titles.forEach((title) => {
-                    if (title.title == request.title) {
-                        app_info = title
-                    }
-                })
-                if (app_info == null) {
-                    response = '{"msg":"TITLE_NOT_EXISTS"}'
-                    break
+        if (request.verb == 'CREATE_SESSION') {
+            var app_info
+            titles.forEach((title) => {
+                if (title.title == request.title) {
+                    app_info = title
                }
+            })
+            if (app_info == null) {
+                response = {msg:'TITLE_NOT_EXISTS'}
+                callback(response)
+                return
+            }

-                var app_port
-                var http_port
-                try {
-                    app_port = app_port_alloc.get()
-                    http_port = http_port_alloc.get()
-                } catch(e) {
-                    console.error(e)
-                    response = '{"msg":"PORT_BUSY"}'
-                    break
+            var app_port
+            var http_port
+            try {
+                app_port = app_port_alloc.get()
+                http_port = http_port_alloc.get()
+            } catch(e) {
+                logger_runtime.error(e)
+                response = {msg:'PORT_BUSY'}
+                callback(response)
+                return
+            }
+            
+            // start app_proc
+            const app_proc = spawn(app_info.path, [].concat(config.app_args_static, 
+                [config.app_args_runtime.ip + config.ip, config.app_args_runtime.port + app_port.toString()]), {
+                    detached: true
+                })
+            app_proc.on('error', function(err) {
+                logger_runtime.error(err)
+                response = {msg:'APP_PROC_ERROR'}
+                callback(response)
+                return
+            })
+            app_proc.on('close', (code) => {
+                app_port_alloc.free(app_port)
+            })
+
+            // start webrtc server
+            const webrtc_proc = child_process.fork(config.webrtc_args_static.cirrus_path, [http_port.toString(), app_port.toString()], {
+                    detached: true
+                })
+            webrtc_proc.on('error', function(err) {
+                logger_runtime.error(err)
+                response = {msg:'WEBRTC_PROC_ERROR'}
+                callback(response)
+                return
+            })
+            webrtc_proc.on('close', (code) => {
+                http_port_alloc.free(http_port)
+                // if webrtc server closed, kill app proc and send session end message to server
+                if (process.platform === 'win32') {
+                    spawn('taskkill', ['/pid', app_proc.pid, '/f', '/t'])
                }
-                
-                // start app_proc
-                const app_proc = spawn(app_info.path, [].concat(config.app_args_static, 
-                    [config.app_args_runtime.ip + config.ip, config.app_args_runtime.port + app_port.toString()]), {
-                        detached: true
-                    })
-                app_proc.on('error', function(err) {
-                    console.error(err)
-                    response = '{"msg":"APP_PROC_ERROR"}'
-                    callback(response)
-                    return
-                })
-                app_proc.on('close', (code) => {
-                    app_port_alloc.free(app_port)
-                })
+                else {
+                    app_proc.kill('SIGINT')
+                }
+                //new http_client(ip, port).post()
+            })

-                // start webrtc server
-                //console.log('http_port: ', http_port, ' app_port: ', app_port)
-
-                const webrtc_proc = child_process.fork(config.webrtc_args_static.cirrus_path, [http_port.toString(), app_port.toString()], {
-                        detached: true
-                    })
-                webrtc_proc.on('error', function(err) {
-                    console.error(err)
-                    response = '{"msg":"WEBRTC_PROC_ERROR"}'
-                    callback(response)
-                    return
-                })
-                webrtc_proc.on('close', (code) => {
-                    http_port_alloc.free(http_port)
-                })
-
-
-                response = `{"msg":"SESSION_CREATED", "link":"http://${config.ip+':'+http_port}"}`
-                break
-            case 'CONNECT_SESSION':
-                break
-            default:
-                response = '{"msg":"UNKNOWN_VERB"}'
+            response = {msg:'SESSION_CREATED', link:`http://${config.ip+':'+http_port}`}
+            callback(response)
+            return
+        } else if(request.verb == 'CONNECT_SESSION') {
+            response = ''
+            callback(response)
+            return
+        } else {
+            response = {msg:'UNKNOWN_VERB'}
+            callback(response)
+            return
        }
-        callback(response)
    }
    catch(e) {
-        console.error(e)
+        logger_runtime.error(e)
    }
 }

@@ -105,32 +114,33 @@ async function create_response(request, callback) {
 const server = http.createServer(function(request, response) {
    if (request.method == 'POST') {
        var body = ''
-        
        request.on('data', function(data) {
            body += data
        })
-
        request.on('end', function() {
-            console.log('received: "', body, '"')
-            response.writeHead(200, {'Content-Type': 'message'})
+            logger_runtime.log('received: ', body)
+            response.writeHead(200, {'Content-Type': 'application/json'})
            try {
                create_response(JSON.parse(body), function(message) {
-                    response.end(message)
-                    console.log('response:', message)
+                    response.end(JSON.stringify(message))
+                    logger_runtime.log('response:', message)
                })
            }
            catch(e) {
-                console.error(e)
+                logger_runtime.error(e)
            }
        })
    }
    else {
-        var message = 'UNDEFINED_REQUEST' + ': ' + request.method.toString()
-        response.writeHead(200, {'Content-Type': 'message'})
-        response.end(message)
-        console.log(message)
+        response.writeHead(501, {'Content-Type': 'application/json'})
+        response.end(JSON.stringify({msg:'UNKNOWN_METHOD'}))
    }
 })

-server.listen(config.port, config.ip)
-console.log(`Listening at http://${config.ip}:${config.port}`)
+
+try {
+    server.listen(config.port, config.ip)
+    logger_init.log(`Listening at http://${config.ip}:${config.port}`)
+} catch (e) {
+    logger_init.error(e)
+}