From aa6bce35c710750b68d4e6c53095e9be4e1afd8d Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期一, 05 九月 2022 09:44:24 +0800
Subject: [PATCH] Merge branch 'wvp-28181-2.0'

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java |  210 ++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 200 insertions(+), 10 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java
index 5286aa5..bd3d4df 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java
@@ -1,12 +1,18 @@
 package com.genersoft.iot.vmp.gb28181.transmit.event.request.impl;
 
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.genersoft.iot.vmp.conf.DynamicTask;
+import com.genersoft.iot.vmp.conf.SipConfig;
 import com.genersoft.iot.vmp.conf.UserSetting;
 import com.genersoft.iot.vmp.gb28181.bean.*;
 import com.genersoft.iot.vmp.gb28181.event.SipSubscribe;
+import com.genersoft.iot.vmp.gb28181.session.AudioBroadcastManager;
 import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager;
 import com.genersoft.iot.vmp.gb28181.transmit.SIPProcessorObserver;
+import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;
+import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage;
 import com.genersoft.iot.vmp.gb28181.transmit.cmd.ISIPCommander;
 import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander;
 import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommanderFroPlatform;
@@ -15,11 +21,11 @@
 import com.genersoft.iot.vmp.gb28181.utils.SipUtils;
 import com.genersoft.iot.vmp.media.zlm.ZlmHttpHookSubscribe;
 import com.genersoft.iot.vmp.media.zlm.ZLMMediaListManager;
+import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils;
 import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory;
-import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem;
-import com.genersoft.iot.vmp.media.zlm.dto.StreamProxyItem;
-import com.genersoft.iot.vmp.media.zlm.dto.StreamPushItem;
+import com.genersoft.iot.vmp.media.zlm.dto.*;
 import com.genersoft.iot.vmp.service.IMediaServerService;
+import com.genersoft.iot.vmp.service.IMediaService;
 import com.genersoft.iot.vmp.service.IPlayService;
 import com.genersoft.iot.vmp.service.IStreamProxyService;
 import com.genersoft.iot.vmp.service.IStreamPushService;
@@ -30,8 +36,12 @@
 import com.genersoft.iot.vmp.storager.IVideoManagerStorage;
 import com.genersoft.iot.vmp.utils.DateUtil;
 import com.genersoft.iot.vmp.utils.SerializeUtils;
+import com.genersoft.iot.vmp.vmanager.bean.AudioBroadcastResult;
+import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
 import gov.nist.javax.sdp.TimeDescriptionImpl;
 import gov.nist.javax.sdp.fields.TimeField;
+import gov.nist.javax.sip.message.SIPRequest;
+import gov.nist.javax.sip.stack.SIPDialog;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.InitializingBean;
@@ -66,6 +76,7 @@
 
     @Autowired
     private IStreamPushService streamPushService;
+
     @Autowired
     private IStreamProxyService streamProxyService;
 
@@ -84,11 +95,20 @@
     @Autowired
     private ISIPCommander commander;
 
+	@Autowired
+	private AudioBroadcastManager audioBroadcastManager;
+
     @Autowired
     private ZLMRTPServerFactory zlmrtpServerFactory;
 
     @Autowired
     private IMediaServerService mediaServerService;
+
+    @Autowired
+    private IMediaService mediaService;
+
+	@Autowired
+	private ZLMRESTfulUtils zlmresTfulUtils;
 
     @Autowired
     private SIPProcessorObserver sipProcessorObserver;
@@ -101,6 +121,16 @@
 
     @Autowired
     private ZLMMediaListManager mediaListManager;
+
+	@Autowired
+	private DeferredResultHolder resultHolder;
+
+	@Autowired
+	private ZlmHttpHookSubscribe subscribe;
+
+	@Autowired
+	private SipConfig config;
+
 
 
     @Autowired
@@ -136,7 +166,7 @@
             // 鏌ヨ璇锋眰鏄惁鏉ヨ嚜涓婄骇骞冲彴\璁惧
             ParentPlatform platform = storager.queryParentPlatByServerGBId(requesterId);
             if (platform == null) {
-                inviteFromDeviceHandle(evt, requesterId);
+                inviteFromDeviceHandle(evt, requesterId, channelId);
             } else {
                 // 鏌ヨ骞冲彴涓嬫槸鍚︽湁璇ラ�氶亾
                 DeviceChannel channel = storager.queryChannelInParentPlatform(requesterId, channelId);
@@ -746,10 +776,16 @@
         }
     }
 
-    public void inviteFromDeviceHandle(RequestEvent evt, String requesterId) throws InvalidArgumentException, ParseException, SipException, SdpException {
+    public void inviteFromDeviceHandle(RequestEvent evt, String requesterId, String channelId) throws InvalidArgumentException, ParseException, SipException, SdpException {
 
         // 闈炰笂绾у钩鍙拌姹傦紝鏌ヨ鏄惁璁惧璇锋眰锛堥�氬父涓烘帴鏀惰闊冲箍鎾殑璁惧锛�
         Device device = redisCatchStorage.getDevice(requesterId);
+        AudioBroadcastCatch audioBroadcastCatch = audioBroadcastManager.get(requesterId, channelId);
+        if (audioBroadcastCatch == null) {
+            logger.warn("鏉ヨ嚜璁惧鐨処nvite璇锋眰闈炶闊冲箍鎾紝宸插拷鐣ワ紝requesterId锛� {}/{}", requesterId, channelId);
+            responseAck(evt, Response.FORBIDDEN);
+            return;
+        }
         Request request = evt.getRequest();
         if (device != null) {
             logger.info("鏀跺埌璁惧" + requesterId + "鐨勮闊冲箍鎾璉nvite璇锋眰");
@@ -762,7 +798,7 @@
             int ssrcIndex = contentString.indexOf("y=");
             if (ssrcIndex > 0) {
                 substring = contentString.substring(0, ssrcIndex);
-                ssrc = contentString.substring(ssrcIndex + 2, ssrcIndex + 12);
+                ssrc = contentString.substring(ssrcIndex + 2, ssrcIndex + 12).trim();
             }
             ssrcIndex = substring.indexOf("f=");
             if (ssrcIndex > 0) {
@@ -772,13 +808,13 @@
 
             //  鑾峰彇鏀寔鐨勬牸寮�
             Vector mediaDescriptions = sdp.getMediaDescriptions(true);
+
             // 鏌ョ湅鏄惁鏀寔PS 璐熻浇96
             int port = -1;
-            //boolean recvonly = false;
             boolean mediaTransmissionTCP = false;
             Boolean tcpActive = null;
             for (int i = 0; i < mediaDescriptions.size(); i++) {
-                MediaDescription mediaDescription = (MediaDescription) mediaDescriptions.get(i);
+                MediaDescription mediaDescription = (MediaDescription)mediaDescriptions.get(i);
                 Media media = mediaDescription.getMedia();
 
                 Vector mediaFormats = media.getMediaFormats(false);
@@ -806,10 +842,164 @@
                 responseAck(evt, Response.UNSUPPORTED_MEDIA_TYPE); // 涓嶆敮鎸佺殑鏍煎紡锛屽彂415
                 return;
             }
-            String username = sdp.getOrigin().getUsername();
             String addressStr = sdp.getOrigin().getAddress();
-            logger.info("璁惧{}璇锋眰璇煶娴侊紝鍦板潃锛歿}:{}锛宻src锛歿}", username, addressStr, port, ssrc);
+            logger.info("璁惧{}璇锋眰璇煶娴侊紝鍦板潃锛歿}:{}锛宻src锛歿}", requesterId, addressStr, port, ssrc);
 
+            MediaServerItem mediaServerItem = playService.getNewMediaServerItem(device);
+            if (mediaServerItem == null) {
+                logger.warn("鏈壘鍒板彲鐢ㄧ殑zlm");
+                responseAck(evt, Response.BUSY_HERE);
+                return;
+            }
+            SendRtpItem sendRtpItem = zlmrtpServerFactory.createSendRtpItem(mediaServerItem, addressStr, port, ssrc, requesterId,
+                    device.getDeviceId(), audioBroadcastCatch.getChannelId(),
+                    mediaTransmissionTCP);
+            if (sendRtpItem == null) {
+                logger.warn("鏈嶅姟鍣ㄧ鍙h祫婧愪笉瓒�");
+                responseAck(evt, Response.BUSY_HERE);
+                return;
+            }
+            sendRtpItem.setTcp(mediaTransmissionTCP);
+            if (tcpActive != null) {
+                sendRtpItem.setTcpActive(tcpActive);
+            }
+            String app = "broadcast";
+            String stream = device.getDeviceId() + "_" + audioBroadcastCatch.getChannelId();
+
+            CallIdHeader callIdHeader = (CallIdHeader) request.getHeader(CallIdHeader.NAME);
+            sendRtpItem.setPlayType(InviteStreamType.PLAY);
+            sendRtpItem.setCallId(callIdHeader.getCallId());
+            sendRtpItem.setPlatformId(requesterId);
+            sendRtpItem.setStatus(1);
+            sendRtpItem.setApp(app);
+            sendRtpItem.setStreamId(stream);
+            sendRtpItem.setPt(8);
+            sendRtpItem.setUsePs(false);
+            sendRtpItem.setOnlyAudio(true);
+            redisCatchStorage.updateSendRTPSever(sendRtpItem);
+
+            // hook鐩戝惉绛夊緟璁惧鎺ㄦ祦涓婃潵
+            // 娣诲姞璁㈤槄
+            HookSubscribeForStreamChange subscribeKey = HookSubscribeFactory.on_stream_changed(app, stream, true, "rtsp", mediaServerItem.getId());
+
+            String finalSsrc = ssrc;
+            // 娴佸凡缁忓瓨鍦ㄦ椂鐩存帴鎺ㄦ祦
+                // 璁剧疆绛夊緟鎺ㄦ祦鐨勮秴鏃�; 榛樿20s
+                String waiteStreamTimeoutTaskKey = "waite-stream-" + device.getDeviceId() + audioBroadcastCatch.getChannelId();
+                dynamicTask.startDelay(waiteStreamTimeoutTaskKey, ()->{
+                    logger.info("绛夊緟鎺ㄦ祦瓒呮椂: {}/{}", app, stream);
+                    subscribe.removeSubscribe(subscribeKey);
+                    playService.stopAudioBroadcast(device.getDeviceId(), audioBroadcastCatch.getChannelId());
+                    // 鍙戦�乥ye
+                    try {
+                        responseAck(evt, Response.BUSY_HERE);
+                    } catch (SipException e) {
+                        throw new RuntimeException(e);
+                    } catch (InvalidArgumentException e) {
+                        throw new RuntimeException(e);
+                    } catch (ParseException e) {
+                        throw new RuntimeException(e);
+                    }
+                }, 20*1000);
+
+                boolean finalMediaTransmissionTCP = mediaTransmissionTCP;
+                subscribe.addSubscribe(subscribeKey,
+                        (MediaServerItem mediaServerItemInUse, JSONObject json)->{
+                            logger.info("鏀跺埌璇煶瀵硅鎺ㄦ祦");
+                            dynamicTask.stop(waiteStreamTimeoutTaskKey);
+                            MediaItem mediaItem = JSON.toJavaObject(json, MediaItem.class);
+                            Integer audioCodecId = null;
+                            if (mediaItem.getTracks() != null && mediaItem.getTracks().size() > 0) {
+                                for (int i = 0; i < mediaItem.getTracks().size(); i++) {
+                                    MediaItem.MediaTrack mediaTrack = mediaItem.getTracks().get(i);
+                                    if (mediaTrack.getCodecType() == 1) {
+                                        audioCodecId = mediaTrack.getCodecId();
+                                        break;
+                                    }
+                                }
+                            }
+
+                            try {
+                                sendRtpItem.setStatus(2);
+                                redisCatchStorage.updateSendRTPSever(sendRtpItem);
+                                StringBuffer content = new StringBuffer(200);
+                                content.append("v=0\r\n");
+                                content.append("o="+ config.getId() +" "+ sdp.getOrigin().getSessionId() +" " + sdp.getOrigin().getSessionVersion()  + " IN IP4 "+mediaServerItem.getSdpIp()+"\r\n");
+                                content.append("s=Play\r\n");
+                                content.append("c=IN IP4 "+mediaServerItem.getSdpIp()+"\r\n");
+                                content.append("t=0 0\r\n");
+                                if (audioCodecId == null) {
+                                    if (finalMediaTransmissionTCP) {
+                                        content.append("m=audio "+ sendRtpItem.getLocalPort()+" TCP/RTP/AVP 8\r\n");
+                                    }else {
+                                        content.append("m=audio "+ sendRtpItem.getLocalPort()+" RTP/AVP 8\r\n");
+                                    }
+
+                                    content.append("a=rtpmap:8 PCMA/8000\r\n");
+                                }else {
+                                    if (audioCodecId == 4) {
+                                        if (finalMediaTransmissionTCP) {
+                                            content.append("m=audio "+ sendRtpItem.getLocalPort()+" TCP/RTP/AVP 0\r\n");
+                                        }else {
+                                            content.append("m=audio "+ sendRtpItem.getLocalPort()+" RTP/AVP 0\r\n");
+                                        }
+                                        content.append("a=rtpmap:0 PCMU/8000\r\n");
+                                    }else {
+                                        if (finalMediaTransmissionTCP) {
+                                            content.append("m=audio "+ sendRtpItem.getLocalPort()+" TCP/RTP/AVP 8\r\n");
+                                        }else {
+                                            content.append("m=audio "+ sendRtpItem.getLocalPort()+" RTP/AVP 8\r\n");
+                                        }
+                                        content.append("a=rtpmap:8 PCMA/8000\r\n");
+                                    }
+                                }
+                                content.append("a=sendonly\r\n");
+                                if (sendRtpItem.isTcp()) {
+                                    content.append("a=connection:new\r\n");
+                                    if (!sendRtpItem.isTcpActive()) {
+                                        content.append("a=setup:active\r\n");
+                                    }else {
+                                        content.append("a=setup:passive\r\n");
+                                    }
+                                }
+                                content.append("y="+ finalSsrc + "\r\n");
+                                content.append("f=v/////a/1/8/1\r\n");
+
+                                ParentPlatform parentPlatform = new ParentPlatform();
+                                parentPlatform.setServerIP(device.getIp());
+                                parentPlatform.setServerPort(device.getPort());
+                                parentPlatform.setServerGBId(device.getDeviceId());
+
+                                responseSdpAck(evt, content.toString(), parentPlatform);
+                                Dialog dialog = evt.getDialog();
+                                audioBroadcastCatch.setDialog((SIPDialog) dialog);
+                                audioBroadcastCatch.setRequest((SIPRequest) request);
+                                audioBroadcastManager.update(audioBroadcastCatch);
+                            } catch (SipException e) {
+                                throw new RuntimeException(e);
+                            } catch (InvalidArgumentException e) {
+                                throw new RuntimeException(e);
+                            } catch (ParseException e) {
+                                throw new RuntimeException(e);
+                            } catch (SdpParseException e) {
+                                throw new RuntimeException(e);
+                            }
+                        });
+//            }
+            String key = DeferredResultHolder.CALLBACK_CMD_BROADCAST + device.getDeviceId();
+            WVPResult<AudioBroadcastResult> wvpResult = new WVPResult<>();
+            wvpResult.setCode(0);
+            wvpResult.setMsg("success");
+            AudioBroadcastResult audioBroadcastResult = new AudioBroadcastResult();
+            audioBroadcastResult.setApp(app);
+            audioBroadcastResult.setStream(stream);
+            audioBroadcastResult.setStreamInfo(mediaService.getStreamInfoByAppAndStream(mediaServerItem, app, stream, null, null, null,false));
+            audioBroadcastResult.setCodec("G.711");
+            wvpResult.setData(audioBroadcastResult);
+            RequestMessage requestMessage = new RequestMessage();
+            requestMessage.setKey(key);
+            requestMessage.setData(wvpResult);
+            resultHolder.invokeAllResult(requestMessage);
         } else {
             logger.warn("鏉ヨ嚜鏃犳晥璁惧/骞冲彴鐨勮姹�");
             responseAck(evt, Response.BAD_REQUEST);

--
Gitblit v1.8.0