增加接口认证

GB28181Server.js

@@ -3,7 +3,7 @@ const SIP = require('./sip/sip');
 const context = require('./core/ctx');
 const Logger = require('./core/logger');
 const NodeSipSession = require('./GB28181Session');
-
+const NtvAuthModule = require('./NtvAuthModule');
 //GB28181 SIP服务器
 class NodeSIPServer {
     constructor(config) {
@@ -18,6 +18,9 @@ class NodeSIPServer {
         this.userinfo = {};
         //会话
         this.dialogs = {};
+
+        //ntv  认证
+        this.authModule = new NtvAuthModule(config.GB28181.authServer);
     }
 
     run() {
@@ -52,7 +55,7 @@ class NodeSIPServer {
 
         //注册&注销 请求
         context.nodeEvent.on('register', (request, remote) => {
-            this.auth(request, remote);
+            this.ntv_auth(request, remote);
         });
 
         //处理消息
@@ -76,8 +79,52 @@ class NodeSIPServer {
         this.uas.destroy();
     }
 
+    ntv_auth(request, remote){
+        let serverInfo = SIP.parseUri(request.uri);
+
+        if (!request.headers || !request.headers.to || !request.headers.from || !serverInfo.user) {
+            this.uas.send(SIP.makeResponse(request, 400, 'missing to header.'));
+            return;
+        }
+        else if (!request.headers.to.uri) {
+            this.uas.send(SIP.makeResponse(request, 400, 'missing username on to header.'));
+            return;
+        }
+        else if (!request.headers.contact || request.headers.contact.length == 0) {
+            this.uas.send(SIP.makeResponse(request, 400, 'missing contact header.'));
+            return;
+        }
+
+        let toInfo = SIP.parseUri(request.headers.to.uri);
+
+        let fromInfo = SIP.parseUri(request.headers.from.uri);
+
+        //用户 标识
+        let userId = fromInfo.user;
+
+        Logger.log(`[${userId}] register protocol=${remote.protocol} ip=${remote.address} port=${remote.port} `);
+
+        //安全性检查
+        if (toInfo.user.length != 20 || fromInfo.user.length != 20 || toInfo.user != fromInfo.user/* || serverInfo.user != this.id*/) {
+            Logger.log(`[${userId}] check fail. `);
+            return;
+        }
+
+        let that = this;
+        //TODO ntv 在这里对接接口
+        this.authModule.auth(userId,function (data){
+            if(data.code==0){
+                let auto_play = "";
+                if("err_desc" in data) auto_play = data.err_desc;
+                that.auth(request, remote,auto_play);
+            }else{
+                Logger.log(`[${userId}] 业务层认证失败：${data.err_desc} `);
+            }
+        });
+    }
+
     //身份验证
-    auth(request, remote) {
+    auth(request, remote,auto_play) {
 
         let serverInfo = SIP.parseUri(request.uri);
 
@@ -104,7 +151,7 @@ class NodeSIPServer {
         Logger.log(`[${userId}] register protocol=${remote.protocol} ip=${remote.address} port=${remote.port} `);
 
         //安全性检查
-        if (toInfo.user.length != 20 || fromInfo.user.length != 20 || toInfo.user != fromInfo.user || serverInfo.user != this.id) {
+        if (toInfo.user.length != 20 || fromInfo.user.length != 20 || toInfo.user != fromInfo.user /*|| serverInfo.user != this.id*/) {
             Logger.log(`[${userId}] check fail. `);
             return;
         }
@@ -177,7 +224,7 @@ class NodeSIPServer {
                 //新的设备加入
                 let session = new NodeSipSession(this.config, userId, { request: request, info: remote }, this.uas);
                 session.expires = expires || this.expires;
-                session.run();
+                session.run(this.authModule,auto_play);
             }
         }
         else {

GB28181Session.js

@@ -3,9 +3,13 @@ const SIP = require('./sip/sip');
 const SDP = require('./sdp/parser');
 const Logger = require('./core/logger');
 const context = require('./core/ctx');
+const NtvAuthModule = require('./NtvAuthModule');
 
 class NodeSipSession {
     constructor(config, userid, remote, uas) {
+        
+        //ntv add
+        this.config  = config;
 
         this.request = remote.request;
 
@@ -72,7 +76,7 @@ class NodeSipSession {
     }
 
     //启动
-    async run() {
+    async run(authModule,auto_play) {
 
         this.pingInterval = setInterval(() => {
 
@@ -103,6 +107,32 @@ class NodeSipSession {
 
         //获取设备目录
         this.catalog = await this.getCatalog();
+
+        //ntv 提交状态，自动开启视频... 
+        this.authModule = authModule;
+        this.auto_play  = auto_play;
+        authModule.status(this.id,1,(data)=>{
+            if(data.code==0){
+                Logger.log("设备在线状态已上报！");
+            }else{
+                Logger.log("设备在线状态已上报失败！" + data.err_desc);
+            }
+            
+        });
+        
+        await this.startRealPlay(this.id,auto_play);
+    }
+
+    //ntv add 自动发realplay消息 TODO streamServer配置项增加服务器ip选项，使摄像头可以推送到其他服务器。
+    async startRealPlay(channelId,auto_play) {
+        let config = this.config;
+        if(config.GB28181.streamServer.enable && (auto_play || config.GB28181.streamServer.auto_play)){
+            let host = config.GB28181.sipServer.host;
+            let port = config.GB28181.streamServer.listen;
+            let mode = 1;
+            Logger.log(`[${channelId}] auto send real play message`);
+            this.sendRealPlayMessage(channelId,host,port,mode);
+        }
     }
 
     // 停止
@@ -116,6 +146,15 @@ class NodeSipSession {
 
             this.isStarting = false;
             context.sessions.delete(this.id);
+
+            this.authModule.status(this.id,0,(data)=>{
+                if(data.code==0){
+                    Logger.log("设备离线状态已上报！");
+                }else{
+                    Logger.log("设备离线状态已上报失败！" + data.err_desc);
+                }
+                
+            });
         }
     }
 
@@ -855,6 +894,14 @@ class NodeSipSession {
 
                                     //ntv
                                     that.playmode='realplay';
+                                    this.authModule.play_status(this.id,1,(data)=>{
+                                        if(data.code==0){
+                                            Logger.log("视频开启状态已上报！");
+                                        }else{
+                                            Logger.log("视频开启状态已上报失败！" + data.err_desc);
+                                        }
+                                        
+                                    });
 
                                     resolve(result);
                                 }
@@ -891,6 +938,14 @@ class NodeSipSession {
                             */
                             //context.nodeEvent.emit('stopPlayed', session.ssrc);
                             delete this.dialogs[key];
+                            this.authModule.play_status(this.id,0,(data)=>{
+                                if(data.code==0){
+                                    Logger.log("视频关闭状态已上报！");
+                                }else{
+                                    Logger.log("视频关闭状态已上报失败！" + data.err_desc);
+                                }
+                                
+                            });
                         }
                     });
 

NtvAuthModule.js

@@ -0,0 +1,132 @@
+/**
+ * 信令认证模块，负责对连接信令的用户进行认证和状态记录
+ * 终端---->信令（加载本模块）---->业务系统
+ * Created by WJ on 2018-12-06
+ * @云视睿博  www.ruiboyun.com
+ */
+
+const http = require('http');
+const qs   = require('querystring');
+
+class NtvAuthModule {
+	
+	constructor(config) {
+		this.config = config;
+	}
+
+	errorMsg(error){
+		return {code:5,err_desc:error};
+	}
+
+	/**
+	 * 解析接口返回内容
+	 * 并回调,回调传输人的第二个参数是一个对象，格式同接口返回格式。
+	 * @param {*} data 
+	 * @param {*} callback 参数为json对象，结构同ntv g3接口规范
+	 */
+	parseData(data,callback) {
+		if(typeof(callback)!="function"){
+			console.log("auth module needs callback function");
+			return;
+		}
+		var obj = null;
+		try {
+			obj  = JSON.parse(data);
+			if(!"code" in obj){
+				obj.code = 5;
+				obj.err_desc = "接口返回数据内容不可理解！";
+			}
+		} catch(error) {
+			obj = this.errorMsg("解析接口返回数据失败！")
+		}
+
+		callback(obj);
+	}
+
+	/**
+	 * 请求远程接口
+	 * @param {接口的action} action 
+	 * @param {参数} data  参数键值对 对象或字符串
+	 * @param {回调} callback 
+	 */
+	postCall(host,port,path,data,callback) {
+		var returnData = '';
+		var content    = '';
+		if(typeof(data)=="object"){
+			content = qs.stringify(data);
+		}else{
+			content = data;
+		}
+
+		var options = {
+			hostname: host,
+			port: port,
+			path: path,
+			method: 'POST',
+			timeout: 2000,
+			headers: {
+				'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8'
+			}
+		};
+		var that = this;
+		var req = http.request(options, function(res) {
+			res.setEncoding('utf8');
+			res.on('data', function(chunk) {
+				returnData = returnData + chunk;
+			});
+			res.on('end', () => {
+				that.parseData(returnData,callback);
+			});
+		});
+
+		req.on('error', function(e) {
+			if(typeof(callback)=="function"){
+				that.callback(that.errorMsg(5,'无法连接业务服务器！'));
+			}
+		});
+
+		req.write(content);
+		req.end();
+	}
+
+	auth(usrId,callback){
+		var config = this.config;
+		var path = "/api/sipMgr/";
+		var paras= "action=sip_auth&id=" + usrId;
+		var host = config.host;
+		var port = config.port;
+		this.postCall(host,port,path,paras,callback);
+	}
+
+	/**
+	 * 
+	 * @param {*} usrId 
+	 * @param {*} status  1 = online
+	 * @param {*} callback 
+	 */
+	status(usrId,status,callback){
+		var config = this.config;
+		var path = "/api/sipMgr/";
+		var paras= "action=sip_status&id=" + usrId +"&status=" + status;
+		var host = config.host;
+		var port = config.port;
+		this.postCall(host,port,path,paras,callback);
+	}
+
+	/**
+	 * 
+	 * @param {*} usrId 
+	 * @param {*} status  1 = online
+	 * @param {*} callback 
+	 */
+	play_status(usrId,status,callback){
+		var config = this.config;
+		var path = "/api/sipMgr/";
+		var paras= "action=play_status&id=" + usrId +"&status=" + status;
+		var host = config.host;
+		var port = config.port;
+		this.postCall(host,port,path,paras,callback);
+	}
+}
+
+module.exports = NtvAuthModule;

package.json

@@ -1,39 +1,19 @@
 {
-  "name": "gb28181-video-agent-gateway",
-  "version": "1.0.0",
-  "description": "A Node.js implementation of GB28181 Video Streaming Agent Server",
-  "main": "index.js",
+  "name": "gb28181-server",
   "scripts": {
-    "test": "node app.js"
+    "start": "nodemon ./vag.js",
+    "pm2": "pm2 start vag.jg --name gb28181",
+    "pkg": "pkg . -t node14-linux-x64 -o sip-server"
   },
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/gb28181/vag.node.git"
+  "bin": "./vag.js",
+  "pkg": {
+
   },
-  "keywords": [
-    "gb28181",
-    "rtp",
-    "sip",
-    "PS"
-  ],
-  "author": "Yzxliu",
-  "license": "MIT",
-  "bugs": {
-    "url": "https://github.com/gb28181/vga.node/issue"
-  },
-  "homepage": "https://github.com/gb28181/vga.node",
   "dependencies": {
-    "basic-auth-connect": "^1.0.0",
-    "body-parser": "^1.19.0",
-    "chalk": "^4.1.0",
-    "dateformat": "^3.0.3",
-    "express": "^4.17.1",
-    "iconv-lite": "^0.6.0",
-    "rtp-rtcp": "0.0.2",
-    "ws": "^7.3.0",
-    "xml2js": "^0.4.23"
+
   },
-  "engines": {
-    "node": ">=8.0.0"
+  "devDependencies": {
+    "nodemon": "^14.15.4",
+    "pkg": "^4.4.9"
   }
-}
+}

stream/server.js

@@ -66,7 +66,7 @@ class NodeGB28181StreamServer {
         //RTP己处理好
         context.nodeEvent.on('rtpReadyed', this.rtpReceived.bind(this));
 
-        //停止播放,关闭推流客户端
+        //停止播放,关闭推流客户端（rtp客户端主动关闭连接）
         context.nodeEvent.on('stopPlayed', (ssrc) => {
             //ntv mod 转换成string，否者检索不到，ssrc传入的时number
             var key = '' + ssrc;
@@ -74,6 +74,16 @@ class NodeGB28181StreamServer {
                 let rtmpClient = context.publishers.get(key);
                 rtmpClient.stop();
                 context.publishers.delete(key);
+                Logger.log("rtmp session removed ",ssrc);
+            }
+        });
+
+        //ntv add 上一个函数主动关闭，或网络异常，或其他原因关闭rtmp
+        context.nodeEvent.on('rtmpClientClose', (ssrc) => {
+            var key = '' + ssrc;
+            if (context.publishers.has(key)) {
+                context.publishers.delete(key);
+                Logger.log("rtmp session removed ",ssrc);
             }
         });
     }

stream/session.js

@@ -49,7 +49,6 @@ class NodeGB28181StreamServerSession {
 
             //ntv add 
             if(this.ssrc){
-                console.log("emit stopPlayed ..........................=",this.ssrc);
                 context.nodeEvent.emit('stopPlayed',this.ssrc);
             }