websocketServer.js 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644
  1. import WorkerManager from './workerManager.js';
  2. import {hex_md5} from "./md5.js";
  3. function WebSocketServer(options) {
  4. let videoElement = null;
  5. let canvasElement = null;
  6. let ROIElement =null;
  7. let websocket = null;
  8. let wsURL = null;
  9. let rtspURL = null;
  10. let username = null;
  11. let password = null;
  12. let CSeq = 1;
  13. let IsDescribe = false; //RTSP响应报文中,describe时有两段,以'\r\n'分段
  14. let currentState = "Options";
  15. let describekey = false;
  16. let Authentication = '\r\n'; //认证,信令最后四个字节为'\r\n\r\n',为补足,默认为'\r\n'
  17. let sessionID = '';
  18. let rtspSDPData = {};
  19. let SDPinfo = []; //SDP信息
  20. let setupSDPIndex = 0;
  21. let getParameterInterval = null; //保活
  22. let AACCodecInfo = null;
  23. //RTP包处理相关
  24. let rtspinterleave = null;
  25. let RTPPacketTotalSize = 0;
  26. let rtpheader = null;
  27. let rtpPacketArray = null;
  28. let workerManager = null;
  29. let connectFailCallback = null;
  30. let lastStreamTime = null; //记录收到码流的时间
  31. let getStreamInterval = null;
  32. let noStreamCallback = null;
  33. const RTSP_INTERLEAVE_LENGTH = 4; //交织头占4个字节
  34. const RTSP_STATE = {
  35. OK: 200,
  36. UNAUTHORIZED: 401,
  37. NOTFOUND: 404,
  38. INVALID_RANGE: 457,
  39. NOTSERVICE: 503,
  40. DISCONNECT: 999
  41. };
  42. const SEND_GETPARM_INTERVAL = 20000; //保活时间
  43. function constructor({video, canvas, ROI,wsUrl, rtspUrl, user, pwd} = {options}) {
  44. videoElement = video;
  45. canvasElement = canvas;
  46. ROIElement = ROI;
  47. wsURL = wsUrl;
  48. rtspURL = rtspUrl;
  49. username = user;
  50. password = pwd;
  51. }
  52. constructor.prototype = {
  53. init() {
  54. workerManager = new WorkerManager();
  55. workerManager.init(videoElement,canvasElement, ROIElement);
  56. },
  57. connect() {
  58. websocket = new WebSocket(wsURL);
  59. websocket.binaryType = 'arraybuffer';
  60. websocket.onmessage = ReceiveMessage;
  61. websocket.onopen = () => {
  62. let option = StringToU8Array("OPTIONS " + rtspURL + " RTSP/1.0\r\nCSeq: " + CSeq + "\r\n\r\n");
  63. websocket.send(option);
  64. //console.log('websocket connect')
  65. };
  66. websocket.onerror = ()=> {
  67. if(connectFailCallback) {
  68. connectFailCallback('websocket connect fail');
  69. }
  70. }
  71. },
  72. close() {
  73. clearInterval(getParameterInterval);
  74. clearInterval(getStreamInterval);
  75. SendRtspCommand(CommandConstructor("TEARDOWN", null));
  76. websocket.close();
  77. if(workerManager) {
  78. workerManager.terminate();
  79. }
  80. },
  81. setCallBack(event, callback) {
  82. switch (event) {
  83. case 'error':
  84. connectFailCallback = ()=>{
  85. callback();
  86. this.close();
  87. };
  88. break;
  89. case 'noStream':
  90. noStreamCallback = ()=>{
  91. callback();
  92. this.close();
  93. };
  94. break;
  95. case 'canplay':
  96. case 'initialCompleted':
  97. case 'ROIFinished':
  98. workerManager.setEventCallBack(event, callback);
  99. break;
  100. default:
  101. console.log('unsupport event');
  102. }
  103. },
  104. updateInfo(obj) {
  105. workerManager.updateInfo(obj);
  106. },
  107. setROIDrawer(ROIElement) {
  108. workerManager.setROIDrawer(ROIElement);
  109. },
  110. };
  111. return new constructor(options);
  112. /**
  113. * websocket消息处理函数
  114. * @param event
  115. * @constructor
  116. */
  117. function ReceiveMessage(event) {
  118. let data = event.data;
  119. let receiveUint8 = new Uint8Array(data);
  120. let PreceiveUint8 = new Uint8Array(receiveUint8.length);
  121. PreceiveUint8.set(receiveUint8, 0);
  122. let dataLength = PreceiveUint8.length;
  123. // if(dataLength < 10) {
  124. // //console.log(String.fromCharCode.apply(null, PreceiveUint8))
  125. // }
  126. while (dataLength > 0) {
  127. if (PreceiveUint8[0] != 36) {//非$符号表示RTSP
  128. //console.log(PreceiveUint8[0], PreceiveUint8[1], PreceiveUint8[2], PreceiveUint8[3], PreceiveUint8[4])
  129. //console.log(PreceiveUint8.length)
  130. let PreceiveMsg = String.fromCharCode.apply(null, PreceiveUint8);
  131. //console.log(PreceiveMsg)
  132. let rtspendpos = null;
  133. if (IsDescribe === true) {
  134. rtspendpos = PreceiveMsg.lastIndexOf("\r\n");
  135. IsDescribe = false
  136. } else {
  137. rtspendpos = PreceiveMsg.search("\r\n\r\n");
  138. }
  139. let rtspstartpos = PreceiveMsg.search("RTSP");
  140. if (rtspstartpos !== -1) {
  141. if (rtspendpos !== -1) {
  142. let RTSPResArray = PreceiveUint8.subarray(rtspstartpos, rtspendpos + RTSP_INTERLEAVE_LENGTH);
  143. PreceiveUint8 = PreceiveUint8.subarray(rtspendpos + RTSP_INTERLEAVE_LENGTH);
  144. let receiveMsg = String.fromCharCode.apply(null, RTSPResArray);
  145. RTSPResHandler(receiveMsg);
  146. dataLength = PreceiveUint8.length;
  147. } else {
  148. dataLength = PreceiveUint8.length;
  149. return
  150. }
  151. } else {
  152. PreceiveUint8 = new Uint8Array;
  153. return
  154. }
  155. } else { //$表示RTP和RTCP
  156. //console.log('RTP开始');
  157. //console.log(PreceiveUint8.length)
  158. // if(PreceiveUint8.length == 4) {
  159. // console.log(PreceiveUint8)
  160. // }
  161. lastStreamTime = Date.now();
  162. rtspinterleave = PreceiveUint8.subarray(0, RTSP_INTERLEAVE_LENGTH);
  163. //console.log(rtspinterleave)
  164. RTPPacketTotalSize = rtspinterleave[2] * 256 + rtspinterleave[3];
  165. if (RTPPacketTotalSize + RTSP_INTERLEAVE_LENGTH <= PreceiveUint8.length) {
  166. rtpheader = PreceiveUint8.subarray(RTSP_INTERLEAVE_LENGTH, 16);
  167. rtpPacketArray = PreceiveUint8.subarray(16, RTPPacketTotalSize + RTSP_INTERLEAVE_LENGTH);
  168. //rtpCallback(rtspinterleave, rtpheader, rtpPacketArray);
  169. workerManager.parseRtpData(rtspinterleave, rtpheader, rtpPacketArray);
  170. PreceiveUint8 = PreceiveUint8.subarray(RTPPacketTotalSize + RTSP_INTERLEAVE_LENGTH);
  171. //console.log('PreceiveUint8.length: ' + PreceiveUint8.length)
  172. dataLength = PreceiveUint8.length;
  173. } else {
  174. dataLength = PreceiveUint8.length;
  175. //console.count('11111111111')
  176. //console.log(PreceiveUint8)
  177. return
  178. }
  179. }
  180. }
  181. }
  182. /**
  183. * 将字符串转为arrayBuffer
  184. * @param string
  185. */
  186. function StringToU8Array(string) {
  187. CSeq++;
  188. //console.log(string)
  189. let stringLength = string.length;
  190. let outputUint8Array = new Uint8Array(new ArrayBuffer(stringLength));
  191. for (let i = 0; i < stringLength; i++) {
  192. outputUint8Array[i] = string.charCodeAt(i);
  193. }
  194. //console.log(outputUint8Array)
  195. return outputUint8Array;
  196. //return string;
  197. }
  198. /**
  199. * 处理收到的RTSP信令,解析后发送下一条
  200. * @param stringMessage
  201. * @constructor
  202. */
  203. function RTSPResHandler(stringMessage) {
  204. //console.log(stringMessage)
  205. //let seekPoint = stringMessage.search("CSeq: ") + 5;
  206. let rtspResponseMsg = parseRtsp(stringMessage);
  207. //console.log(rtspResponseMsg)
  208. if (rtspResponseMsg.ResponseCode === RTSP_STATE.UNAUTHORIZED && Authentication === "\r\n") { //需要鉴权
  209. if(currentState === "Describe") {
  210. IsDescribe = false;
  211. describekey = false;
  212. }
  213. username= getUser(rtspURL).username;
  214. password = getUser(rtspURL).password;
  215. //console.log(rtspResponseMsg)
  216. SendRtspCommand(formDigest(rtspResponseMsg));
  217. Authentication = "\r\n";
  218. } else if (rtspResponseMsg.ResponseCode === RTSP_STATE.OK) { //服务器端返回成功
  219. switch (currentState) {
  220. case 'Options':
  221. currentState = "Describe";
  222. SendRtspCommand(CommandConstructor("DESCRIBE", null));
  223. break;
  224. case "Describe":
  225. rtspSDPData = parseDescribeResponse(stringMessage);
  226. if (typeof rtspResponseMsg.ContentBase !== "undefined") {
  227. rtspSDPData.ContentBase = rtspResponseMsg.ContentBase
  228. }
  229. //console.log(rtspSDPData.Sessions)
  230. for (let idx = 0; idx < rtspSDPData.Sessions.length; idx++) {
  231. let sdpInfoObj = {};
  232. if (rtspSDPData.Sessions[idx].CodecMime === "H264" ) { //暂时只支持H264
  233. sdpInfoObj.codecName = rtspSDPData.Sessions[idx].CodecMime;
  234. //sdpInfoObj.trackID = rtspSDPData.Sessions[idx].ControlURL;
  235. sdpInfoObj.trackID = rtspSDPData.Sessions[idx].trackID;
  236. sdpInfoObj.ClockFreq = rtspSDPData.Sessions[idx].ClockFreq;
  237. sdpInfoObj.Port = parseInt(rtspSDPData.Sessions[idx].Port);
  238. if (typeof rtspSDPData.Sessions[idx].Framerate !== "undefined") {
  239. sdpInfoObj.Framerate = parseInt(rtspSDPData.Sessions[idx].Framerate)
  240. }
  241. if(typeof rtspSDPData.Sessions[idx].SPS !== "undefined") {
  242. sdpInfoObj.SPS = rtspSDPData.Sessions[idx].SPS;
  243. }
  244. SDPinfo.push(sdpInfoObj)
  245. } else {
  246. console.log("Unknown codec type:", rtspSDPData.Sessions[idx].CodecMime, rtspSDPData.Sessions[idx].ControlURL)
  247. }
  248. }
  249. setupSDPIndex = 0;
  250. currentState = "Setup";
  251. //console.log(SDPinfo[setupSDPIndex])
  252. SendRtspCommand(CommandConstructor("SETUP", SDPinfo[setupSDPIndex].trackID, setupSDPIndex));
  253. //SendRtspCommand(CommandConstructor("SETUP", 'track1'));
  254. break;
  255. case "Setup":
  256. sessionID = rtspResponseMsg.SessionID;
  257. //多路流(如音频流)
  258. //在Describe中暂时只解析H264视频流,因此SDPinfo.length始终为1
  259. if (setupSDPIndex < SDPinfo.length) {
  260. SDPinfo[setupSDPIndex].RtpInterlevedID = rtspResponseMsg.RtpInterlevedID;
  261. SDPinfo[setupSDPIndex].RtcpInterlevedID = rtspResponseMsg.RtcpInterlevedID;
  262. setupSDPIndex += 1;
  263. if (setupSDPIndex !== SDPinfo.length) {
  264. SendRtspCommand(CommandConstructor("SETUP", SDPinfo[setupSDPIndex].trackID, setupSDPIndex));
  265. } else {
  266. workerManager.sendSdpInfo(SDPinfo);
  267. currentState = "Play";
  268. SendRtspCommand(CommandConstructor("PLAY"));
  269. }
  270. }
  271. sessionID = rtspResponseMsg.SessionID;
  272. //开始播放后,发送GET_PARAMETER进行保活
  273. clearInterval(getParameterInterval);
  274. getParameterInterval = setInterval(function () {
  275. SendRtspCommand(CommandConstructor("GET_PARAMETER", null))
  276. }, SEND_GETPARM_INTERVAL);
  277. getStreamInterval = setInterval(()=>{
  278. if(!getBitStream()) {
  279. console.log('超时!');
  280. noStreamCallback && noStreamCallback();
  281. }
  282. }, 5000);
  283. break;
  284. case "Play":
  285. break;
  286. default:
  287. console.log('暂不支持的信令');
  288. break;
  289. }
  290. } else if (rtspResponseMsg.ResponseCode === RTSP_STATE.NOTSERVICE) { //服务不可用
  291. } else if (rtspResponseMsg.ResponseCode === RTSP_STATE.NOTFOUND) { //Not Found
  292. }
  293. }
  294. /**
  295. * 发送rtsp信令
  296. * @param sendMessage
  297. * @constructor
  298. */
  299. function SendRtspCommand(sendMessage) {
  300. //console.log(sendMessage)
  301. if (websocket !== null && websocket.readyState === WebSocket.OPEN) {
  302. if (describekey === false) {
  303. let describeCmd = sendMessage.search("DESCRIBE");
  304. if (describeCmd !== -1) {
  305. IsDescribe = true;
  306. describekey = true;
  307. }
  308. }
  309. //console.log(sendMessage)
  310. websocket.send(StringToU8Array(sendMessage))
  311. } else {
  312. console.log('websocket未连接')
  313. }
  314. }
  315. /**
  316. * 组装RTSP信令
  317. * @param method
  318. * @param trackID
  319. * @returns {*}
  320. * @constructor
  321. */
  322. function CommandConstructor(method, trackID, interleaved) {
  323. let sendMessage;
  324. switch (method) {
  325. case"OPTIONS":
  326. case"TEARDOWN":
  327. case"SET_PARAMETERS":
  328. case"DESCRIBE":
  329. //TODO: 保活
  330. sendMessage = method + " " + rtspURL + " RTSP/1.0\r\nCSeq: " + CSeq + "\r\n" + Authentication;
  331. break;
  332. case"SETUP":
  333. //console.log(trackID)
  334. //TODO 多trackID的时候测试一下
  335. sendMessage = method + " " + rtspURL + "/" + trackID + " RTSP/1.0\r\nCSeq: " + CSeq + Authentication + "Transport:RTP/AVP/TCP;unicast;interleaved=" + 2 * interleaved + "-" + (2 * interleaved + 1) + "\r\n";
  336. if(sessionID == 0) {
  337. sendMessage += "\r\n";
  338. } else {
  339. sendMessage += "Session: " + sessionID + "\r\n\r\n";
  340. }
  341. break;
  342. case"PLAY":
  343. sendMessage = method + " " + rtspURL + " RTSP/1.0\r\nCSeq: " + CSeq + "\r\nSession: " + sessionID + "\r\n" + "Range: npt=0.000-\r\n" + Authentication;
  344. break;
  345. case"PAUSE":
  346. sendMessage = method + " " + rtspURL + " RTSP/1.0\r\nCSeq: " + CSeq + "\r\nSession: " + sessionID + "\r\n\r\n";
  347. break;
  348. case"GET_PARAMETER":
  349. sendMessage = method + " " + rtspURL + " RTSP/1.0\r\nCSeq: " + CSeq + "\r\nSession: " + sessionID + "\r\n" + Authentication;
  350. break;
  351. default:
  352. console.log('暂不支持的RTSP信令');
  353. }
  354. //console.log(sendMessage);
  355. return sendMessage;
  356. }
  357. /**
  358. * 解析RTSP信令
  359. * @param message1
  360. */
  361. function parseRtsp(message1) {
  362. let RtspResponseData = {};
  363. let cnt = 0, cnt1 = 0, ttt = null, LineTokens = null;
  364. let message = null;
  365. if (message1.search("Content-Type: application/sdp") !== -1) {
  366. let messageTok = message1.split("\r\n\r\n");
  367. message = messageTok[0]
  368. } else {
  369. message = message1
  370. }
  371. let TokenziedResponseLines = message.split("\r\n");
  372. let ResponseCodeTokens = TokenziedResponseLines[0].split(" ");
  373. if (ResponseCodeTokens.length > 2) {
  374. RtspResponseData.ResponseCode = parseInt(ResponseCodeTokens[1]);
  375. RtspResponseData.ResponseMessage = ResponseCodeTokens[2]
  376. }
  377. if (RtspResponseData.ResponseCode === RTSP_STATE.OK) {
  378. for (cnt = 1; cnt < TokenziedResponseLines.length; cnt++) {
  379. LineTokens = TokenziedResponseLines[cnt].split(":");
  380. if (LineTokens[0] === "Public") {
  381. RtspResponseData.MethodsSupported = LineTokens[1].split(",")
  382. } else if (LineTokens[0] === "CSeq") {
  383. RtspResponseData.CSeq = parseInt(LineTokens[1])
  384. } else if (LineTokens[0] === "Content-Type") {
  385. RtspResponseData.ContentType = LineTokens[1];
  386. if (RtspResponseData.ContentType.search("application/sdp") !== -1) {
  387. RtspResponseData.SDPData = parseDescribeResponse(message1)
  388. }
  389. } else if (LineTokens[0] === "Content-Length") {
  390. RtspResponseData.ContentLength = parseInt(LineTokens[1])
  391. } else if (LineTokens[0] === "Content-Base") {
  392. let ppos = TokenziedResponseLines[cnt].search("Content-Base:");
  393. if (ppos !== -1) {
  394. RtspResponseData.ContentBase = TokenziedResponseLines[cnt].substr(ppos + 13)
  395. }
  396. } else if (LineTokens[0] === "Session") {
  397. let SessionTokens = LineTokens[1].split(";");
  398. //RtspResponseData.SessionID = parseInt(SessionTokens[0])
  399. //console.log(SessionTokens[0])
  400. RtspResponseData.SessionID = SessionTokens[0].trim();
  401. } else if (LineTokens[0] === "Transport") {
  402. let TransportTokens = LineTokens[1].split(";");
  403. for (cnt1 = 0; cnt1 < TransportTokens.length; cnt1++) {
  404. let tpos = TransportTokens[cnt1].search("interleaved=");
  405. if (tpos !== -1) {
  406. let interleaved = TransportTokens[cnt1].substr(tpos + 12);
  407. let interleavedTokens = interleaved.split("-");
  408. if (interleavedTokens.length > 1) {
  409. RtspResponseData.RtpInterlevedID = parseInt(interleavedTokens[0]);
  410. RtspResponseData.RtcpInterlevedID = parseInt(interleavedTokens[1])
  411. }
  412. }
  413. }
  414. } else if (LineTokens[0] === "RTP-Info") {
  415. LineTokens[1] = TokenziedResponseLines[cnt].substr(9);
  416. let RTPInfoTokens = LineTokens[1].split(",");
  417. RtspResponseData.RTPInfoList = [];
  418. for (cnt1 = 0; cnt1 < RTPInfoTokens.length; cnt1++) {
  419. let RtpTokens = RTPInfoTokens[cnt1].split(";");
  420. let RtpInfo = {};
  421. for (let cnt2 = 0; cnt2 < RtpTokens.length; cnt2++) {
  422. let poss = RtpTokens[cnt2].search("url=");
  423. if (poss !== -1) {
  424. RtpInfo.URL = RtpTokens[cnt2].substr(poss + 4)
  425. }
  426. poss = RtpTokens[cnt2].search("seq=");
  427. if (poss !== -1) {
  428. RtpInfo.Seq = parseInt(RtpTokens[cnt2].substr(poss + 4))
  429. }
  430. }
  431. RtspResponseData.RTPInfoList.push(RtpInfo)
  432. }
  433. }
  434. }
  435. } else if (RtspResponseData.ResponseCode === RTSP_STATE.UNAUTHORIZED) {
  436. for (cnt = 1; cnt < TokenziedResponseLines.length; cnt++) {
  437. LineTokens = TokenziedResponseLines[cnt].split(":");
  438. if (LineTokens[0] === "CSeq") {
  439. RtspResponseData.CSeq = parseInt(LineTokens[1])
  440. } else if (LineTokens[0] === "WWW-Authenticate") {
  441. let AuthTokens = LineTokens[1].split(",");
  442. for (cnt1 = 0; cnt1 < AuthTokens.length; cnt1++) {
  443. let pos = AuthTokens[cnt1].search("Digest realm=");
  444. if (pos !== -1) {
  445. ttt = AuthTokens[cnt1].substr(pos + 13);
  446. let realmtok = ttt.split('"');
  447. RtspResponseData.Realm = realmtok[1]
  448. }
  449. pos = AuthTokens[cnt1].search("nonce=");
  450. if (pos !== -1) {
  451. ttt = AuthTokens[cnt1].substr(pos + 6);
  452. let noncetok = ttt.split('"');
  453. RtspResponseData.Nonce = noncetok[1]
  454. }
  455. }
  456. }
  457. }
  458. }
  459. return RtspResponseData
  460. }
  461. /**
  462. * 解析Describe信令
  463. * @param message1
  464. */
  465. function parseDescribeResponse(message1) {
  466. //console.log(message1)
  467. let SDPData = {};
  468. let Sessions = [];
  469. SDPData.Sessions = Sessions;
  470. let message = null;
  471. if (message1.search("Content-Type: application/sdp") !== -1) {
  472. let messageTok = message1.split("\r\n\r\n");
  473. message = messageTok[1]
  474. } else {
  475. message = message1
  476. }
  477. let TokenziedDescribe = message.split("\r\n");
  478. let mediaFound = false;
  479. for (let cnt = 0; cnt < TokenziedDescribe.length; cnt++) {
  480. let SDPLineTokens = TokenziedDescribe[cnt].split("=");
  481. if (SDPLineTokens.length > 0) {
  482. switch (SDPLineTokens[0]) {
  483. case"a":
  484. let aLineToken = SDPLineTokens[1].split(":");
  485. if (aLineToken.length > 1) {
  486. if (aLineToken[0] === "control") {
  487. let pos = TokenziedDescribe[cnt].search("control:");
  488. if (mediaFound === true) {
  489. if (pos !== -1) {
  490. SDPData.Sessions[SDPData.Sessions.length - 1].ControlURL = TokenziedDescribe[cnt].substr(pos + 8);
  491. let trackPos = TokenziedDescribe[cnt].search("track");
  492. SDPData.Sessions[SDPData.Sessions.length - 1].trackID = TokenziedDescribe[cnt].substr(trackPos);
  493. }
  494. } else {
  495. if (pos !== -1) {
  496. SDPData.BaseURL = TokenziedDescribe[cnt].substr(pos + 8)
  497. }
  498. }
  499. } else if (aLineToken[0] === "rtpmap") {
  500. //console.log(aLineToken)
  501. let rtpmapLine = aLineToken[1].split(" ");
  502. //console.log(rtpmapLine)
  503. SDPData.Sessions[SDPData.Sessions.length - 1].PayloadType = rtpmapLine[0];
  504. let MimeLine = rtpmapLine[1].split("/");
  505. SDPData.Sessions[SDPData.Sessions.length - 1].CodecMime = MimeLine[0];
  506. if (MimeLine.length > 1) {
  507. SDPData.Sessions[SDPData.Sessions.length - 1].ClockFreq = MimeLine[1]
  508. }
  509. } else if (aLineToken[0] === "framesize") {
  510. let framesizeLine = aLineToken[1].split(" ");
  511. if (framesizeLine.length > 1) {
  512. let framesizeinf = framesizeLine[1].split("-");
  513. SDPData.Sessions[SDPData.Sessions.length - 1].Width = framesizeinf[0];
  514. SDPData.Sessions[SDPData.Sessions.length - 1].Height = framesizeinf[1]
  515. }
  516. } else if (aLineToken[0] === "framerate") {
  517. SDPData.Sessions[SDPData.Sessions.length - 1].Framerate = aLineToken[1]
  518. } else if (aLineToken[0] === "fmtp") {
  519. let sessLine = TokenziedDescribe[cnt].split(" ");
  520. if (sessLine.length < 2) {
  521. continue
  522. }
  523. for (let ii = 1; ii < sessLine.length; ii++) {
  524. let sessToken = sessLine[ii].split(";");
  525. let sessprmcnt = 0;
  526. for (sessprmcnt = 0; sessprmcnt < sessToken.length; sessprmcnt++) {
  527. let ppos = sessToken[sessprmcnt].search("mode=");
  528. if (ppos !== -1) {
  529. SDPData.Sessions[SDPData.Sessions.length - 1].mode = sessToken[sessprmcnt].substr(ppos + 5)
  530. }
  531. ppos = sessToken[sessprmcnt].search("config=");
  532. if (ppos !== -1) {
  533. SDPData.Sessions[SDPData.Sessions.length - 1].config = sessToken[sessprmcnt].substr(ppos + 7);
  534. AACCodecInfo.config = SDPData.Sessions[SDPData.Sessions.length - 1].config;
  535. AACCodecInfo.clockFreq = SDPData.Sessions[SDPData.Sessions.length - 1].ClockFreq;
  536. AACCodecInfo.bitrate = SDPData.Sessions[SDPData.Sessions.length - 1].Bitrate
  537. }
  538. ppos = sessToken[sessprmcnt].search("sprop-vps=");
  539. if (ppos !== -1) {
  540. SDPData.Sessions[SDPData.Sessions.length - 1].VPS = sessToken[sessprmcnt].substr(ppos + 10)
  541. }
  542. ppos = sessToken[sessprmcnt].search("sprop-sps=");
  543. if (ppos !== -1) {
  544. SDPData.Sessions[SDPData.Sessions.length - 1].SPS = sessToken[sessprmcnt].substr(ppos + 10)
  545. }
  546. ppos = sessToken[sessprmcnt].search("sprop-pps=");
  547. if (ppos !== -1) {
  548. SDPData.Sessions[SDPData.Sessions.length - 1].PPS = sessToken[sessprmcnt].substr(ppos + 10)
  549. }
  550. ppos = sessToken[sessprmcnt].search("sprop-parameter-sets=");
  551. if (ppos !== -1) {
  552. let SPSPPS = sessToken[sessprmcnt].substr(ppos + 21);
  553. let SPSPPSTokenized = SPSPPS.split(",");
  554. if (SPSPPSTokenized.length > 1) {
  555. SDPData.Sessions[SDPData.Sessions.length - 1].SPS = SPSPPSTokenized[0];
  556. SDPData.Sessions[SDPData.Sessions.length - 1].PPS = SPSPPSTokenized[1]
  557. }
  558. }
  559. }
  560. }
  561. }
  562. }
  563. break;
  564. case"m":
  565. let mLineToken = SDPLineTokens[1].split(" ");
  566. let Session = {};
  567. Session.Type = mLineToken[0];
  568. Session.Port = mLineToken[1];
  569. Session.Payload = mLineToken[3];
  570. SDPData.Sessions.push(Session);
  571. mediaFound = true;
  572. break;
  573. case"b":
  574. if (mediaFound === true) {
  575. let bLineToken = SDPLineTokens[1].split(":");
  576. SDPData.Sessions[SDPData.Sessions.length - 1].Bitrate = bLineToken[1]
  577. }
  578. break
  579. }
  580. }
  581. }
  582. return SDPData
  583. };
  584. function formDigest(message) {
  585. let {Nonce, Realm} = message;
  586. //Realm = '54c415830ec4';
  587. //Nonce = 'fb01c51948704e59eb5a474b33caff8b';
  588. let user = {
  589. username: username,
  590. password: password,
  591. };
  592. let hex1 = hex_md5(user.username + ":" + Realm + ":" + user.password);
  593. let hex2 = hex_md5(currentState.toUpperCase() + ":" + rtspURL);
  594. let responce = hex_md5(hex1 + ":" + Nonce + ":" + hex2);
  595. Authentication = 'Authorization: Digest username="' + user.username + '", realm="' + Realm + '", nonce="' + Nonce + '",uri="' + rtspURL + '", response="' + responce + '"\r\n' + "Accept: application/sdp\r\n" + '\r\n';
  596. return currentState.toUpperCase() + " " + rtspURL + " RTSP/1.0\r\nCSeq: " + CSeq + "\r\n" + Authentication;
  597. }
  598. function getUser(rtspUrl) {
  599. let user = rtspUrl.split('rtsp://')[1].split('@')[0];
  600. let username = user.split(':')[0],
  601. password = user.split(':')[1];
  602. return {username, password};
  603. }
  604. function getBitStream() {
  605. if(lastStreamTime === null) {
  606. lastStreamTime = Date.now();
  607. } else {
  608. //console.log(Date.now() - lastStreamTime)
  609. return Date.now() - lastStreamTime < 5000;
  610. }
  611. }
  612. }
  613. export default WebSocketServer;