package com.ard.zlm.service.impl;
|
|
import com.ard.common.core.constant.Constants;
|
import com.ard.common.core.constant.SecurityConstants;
|
import com.ard.common.core.domain.R;
|
import com.ard.gb28181.api.RemoteGb28181Service;
|
import com.ard.gb28181.api.domain.Device;
|
import com.ard.zlm.service.ZlmStreamService;
|
import com.ard.zlm.domain.StreamChannel;
|
import com.ard.zlm.api.domain.RTPServerParam;
|
import com.ard.zlm.api.domain.StreamPullPlay;
|
import com.ard.zlm.api.domain.ZlmMediaServer;
|
import com.ard.zlm.api.utils.MediaServerUtils;
|
import com.ard.zlm.common.InviteSessionType;
|
import com.ard.zlm.config.UserSetting;
|
import com.ard.zlm.domain.StreamAuthorityInfo;
|
import com.ard.zlm.hook.ResultForOnPublish;
|
import com.ard.zlm.service.IInviteStreamService;
|
import com.ard.zlm.service.IMediaServerService;
|
import com.ard.zlm.service.IMediaService;
|
import com.ard.zlm.service.IRedisCatchStorage;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.DigestUtils;
|
|
import java.util.Map;
|
|
@Slf4j
|
@Service
|
public class MediaServiceImpl implements IMediaService {
|
|
@Autowired
|
private ZlmStreamService zlmStreamService;
|
|
@Autowired
|
private IMediaServerService mediaServerService;
|
|
@Autowired
|
private IInviteStreamService inviteStreamService;
|
|
@Autowired
|
private UserSetting userSetting;
|
|
@Autowired
|
private IRedisCatchStorage redisCatchStorage;
|
|
@Autowired
|
private RemoteGb28181Service remoteGb28181Service;
|
|
@Override
|
public boolean closeStreamOnNoneReader(String mediaServerId, String app, String stream, String schema) {
|
|
// R<StreamChannel> r = zlmStreamService.getQsDeviceStream(stream, SecurityConstants.INNER);
|
// if (r.getCode() != Constants.SUCCESS) {
|
// return false;
|
// }
|
//
|
// StreamChannel data = r.getData();
|
// if (data == null) {
|
// return false;
|
// }
|
//
|
// // 拉流代理
|
// if ("rtsp".equals(app) || "rtmp".equals(app) || "flv".equals(app) || "hls".equals(app)) {
|
// if ("1".equals(data.getEnableDisableNoneReader())) {
|
// // 无人观看停用
|
// // 修改数据
|
// StreamPullPlay streamPullPlay = new StreamPullPlay();
|
// streamPullPlay.setDeviceId(data.getId());
|
// streamPullPlay.setStreamKey(data.getStreamKey());
|
// streamPullPlay.setMediaServerId(data.getMediaServerId());
|
//
|
// mediaServerService.stopStreamPullPlay(streamPullPlay);
|
// return true;
|
// } else {
|
// return false;
|
// }
|
// } else if ("haikang".equals(app) || "haikang_isup".equals(app) || "dahua".equals(app)) {
|
// if ("1".equals(data.getEnableDisableNoneReader())) {
|
// // 无人观看停用
|
// RTPServerParam rtpServerParam = new RTPServerParam();
|
// rtpServerParam.setId(data.getId());
|
// rtpServerParam.setType(data.getType());
|
// rtpServerParam.setStreamId(stream);
|
// mediaServerService.stopRtpPlay(rtpServerParam);
|
// return true;
|
// } else {
|
// return false;
|
// }
|
// } else if ("push".equals(app)) {
|
// if ("1".equals(data.getEnableDisableNoneReader())) {
|
// return true;
|
// } else {
|
// return false;
|
// }
|
// } else if ("gb28181".equals(app)) {
|
// if ("1".equals(data.getEnableDisableNoneReader())) {
|
// R<Device> deviceR = remoteGb28181Service.getDeviceByDeviceId(data.getGbDeviceId(), SecurityConstants.INNER);
|
// if (deviceR.getCode() == Constants.SUCCESS && deviceR.getData() != null) {
|
// mediaServerService.stopGb28181Play(InviteSessionType.PLAY, data, deviceR.getData(), data.getDeviceCode());
|
// }
|
// return true;
|
// } else {
|
// return false;
|
// }
|
// }
|
return true;
|
}
|
|
@Override
|
public ResultForOnPublish authenticatePublish(ZlmMediaServer mediaServer, String app, String stream, String params) {
|
ResultForOnPublish result = new ResultForOnPublish();
|
result.setEnable_audio(true);
|
|
// 海康sdk 海康isup 大华sdk
|
if ("haikang".equals(app) || "haikang_isup".equals(app) || "dahua".equals(app)) {
|
R<StreamChannel> r = zlmStreamService.getQsDeviceStream(stream, SecurityConstants.INNER);
|
|
if (r.getCode() != Constants.SUCCESS) {
|
result.setEnable_mp4(false);
|
} else if (r.getData() == null) {
|
result.setEnable_mp4(false);
|
} else {
|
result.setEnable_mp4("1".equals(r.getData().getEnableMp4()));
|
}
|
}
|
// 推流
|
if ("push".equals(app)) {
|
R<StreamChannel> r = zlmStreamService.getQsDeviceStream(stream, SecurityConstants.INNER);
|
|
if (r.getCode() != Constants.SUCCESS) {
|
result.setEnable_mp4(false);
|
} else if (r.getData() == null) {
|
result.setEnable_mp4(false);
|
} else {
|
result.setEnable_mp4("1".equals(r.getData().getEnableMp4()));
|
}
|
if (userSetting.getPushAuthority()) {
|
// 对于推流进行鉴权
|
Map<String, String> paramMap = MediaServerUtils.urlParamToMap(params);
|
// 推流鉴权
|
if (params == null) {
|
log.info("推流鉴权失败： 缺少必要参数：sign=md5(user表的pushKey)");
|
throw new RuntimeException("Unauthorized");
|
}
|
|
String sign = paramMap.get("sign");
|
if (sign == null) {
|
log.info("推流鉴权失败： 缺少必要参数：sign=md5");
|
throw new RuntimeException("Unauthorized");
|
}
|
sign = sign.replaceAll("/$", "");
|
// 推流自定义播放鉴权码
|
String callId = paramMap.get("callId");
|
// 鉴权配置
|
String checkStr = callId == null ? userSetting.getPushKey() : (callId + "_" + userSetting.getPushKey());
|
String checkSign = DigestUtils.md5DigestAsHex(checkStr.getBytes());
|
if (!checkSign.equals(sign)) {
|
log.info("推流鉴权失败： sign 无权限: callId={}. sign={}", callId, sign);
|
throw new RuntimeException("推流鉴权失败： sign 无权限: callId=" + callId + ". sign=" + sign);
|
}
|
|
StreamAuthorityInfo streamAuthorityInfo = StreamAuthorityInfo.getInstanceByHook(app, stream, mediaServer.getId());
|
streamAuthorityInfo.setCallId(callId);
|
streamAuthorityInfo.setSign(sign);
|
// 鉴权通过
|
redisCatchStorage.updateStreamAuthorityInfo(app, stream, streamAuthorityInfo);
|
}
|
}
|
|
return result;
|
}
|
}
|
