fix srt bridge hevc to rtmp error.

try to fix #4428.
Cause: rtmp do not support hevc, rtmp enhanced do.

Author: suzp1984

trunk/src/app/srs_app_srt_source.cpp

@@ -671,6 +671,7 @@ srs_error_t SrsSrtFrameBuilder::check_vps_sps_pps_change(SrsTsMessage *msg)
 
     // ts tbn to flv tbn.
     uint32_t dts = (uint32_t)(msg->dts / 90);
+    uint32_t pts = (uint32_t)(msg->pts / 90);
 
     std::string sh;
     SrsUniquePtr<SrsRawHEVCStream> hevc(new SrsRawHEVCStream());
@@ -682,10 +683,16 @@ srs_error_t SrsSrtFrameBuilder::check_vps_sps_pps_change(SrsTsMessage *msg)
     // h265 packet to flv packet.
     char *flv = NULL;
     int nb_flv = 0;
-    if ((err = hevc->mux_avc2flv(sh, SrsVideoAvcFrameTypeKeyFrame, SrsVideoAvcFrameTraitSequenceHeader, dts, dts, &flv, &nb_flv)) != srs_success) {
-        return srs_error_wrap(err, "avc to flv");
-    }
-
+    if ((err = hevc->mux_avc2flv_enhanced(sh,
+                                          SrsVideoAvcFrameTypeKeyFrame,
+                                          SrsVideoHEVCFrameTraitPacketTypeSequenceStart,
+                                          dts,
+                                          pts,
+                                          &flv,
+                                          &nb_flv)) != srs_success) {
+        return srs_error_wrap(err, "hevc sh to flv");
+    }
+    
     SrsMessageHeader header;
     header.initialize_video(nb_flv, dts, video_streamid_);
     SrsRtmpCommonMessage rtmp;
@@ -713,8 +720,6 @@ srs_error_t SrsSrtFrameBuilder::on_hevc_frame(SrsTsMessage *msg, vector<pair<cha
 
     // ts tbn to flv tbn.
     uint32_t dts = (uint32_t)(msg->dts / 90);
-    uint32_t pts = (uint32_t)(msg->pts / 90);
-    int32_t cts = pts - dts;
 
     // for IDR frame, the frame is keyframe.
     SrsVideoAvcFrameType frame_type = SrsVideoAvcFrameTypeInterFrame;
@@ -736,16 +741,9 @@ srs_error_t SrsSrtFrameBuilder::on_hevc_frame(SrsTsMessage *msg, vector<pair<cha
     SrsBuffer payload(rtmp.payload(), rtmp.size());
 
     // Write 5bytes video tag header.
-
-    // @see: E.4.3 Video Tags, video_file_format_spec_v10_1.pdf, page 78
-    // Frame Type, Type of video frame.
-    // CodecID, Codec Identifier.
-    // set the rtmp header
-    payload.write_1bytes((frame_type << 4) | SrsVideoCodecIdHEVC);
-    // hevc_type: nalu
-    payload.write_1bytes(0x01);
-    // composition time
-    payload.write_3bytes(cts);
+    // @see: https://veovera.org/docs/enhanced/enhanced-rtmp-v1.pdf, page 8
+    payload.write_1bytes(SRS_FLV_IS_EX_HEADER | (frame_type << 4) | SrsVideoHEVCFrameTraitPacketTypeCodedFramesX);
+    payload.write_4bytes(0x68766331); // 'h' 'v' 'c' '1'
 
     // Write video nalus.
     for (size_t i = 0; i != ipb_frames.size(); ++i) {