fix: md count send

src/forward/mod.rs

@@ -1,7 +1,6 @@
 use std::io::Cursor;
 use std::sync::Arc;
 
-use crate::forward::forward_internal::{get_peer_key, PeerForwardInternal};
 use anyhow::Result;
 use log::info;
 use tokio::sync::Mutex;
@@ -12,6 +11,10 @@ use webrtc::peer_connection::sdp::session_description::RTCSessionDescription;
 use webrtc::peer_connection::RTCPeerConnection;
 use webrtc::rtp_transceiver::rtp_codec::RTPCodecType;
 use webrtc::sdp::{MediaDescription, SessionDescription};
+
+use crate::forward::forward_internal::{get_peer_key, PeerForwardInternal};
+use crate::media;
+
 mod forward_internal;
 mod track_match;
 
@@ -191,19 +194,7 @@ fn parse_ice_candidate(content: String) -> Result<Vec<RTCIceCandidateInit>> {
 
 fn get_media_descriptions(sd: SessionDescription, publish: bool) -> Result<Vec<MediaDescription>> {
     let mut media_descriptions = sd.media_descriptions;
-    media_descriptions.retain(|media_description| {
-        let mut is_publish = false;
-        for attribute in &media_description.attributes {
-            match attribute.key.as_str() {
-                "sendonly" | "simulcast:send" => {
-                    is_publish = true;
-                    break;
-                }
-                _ => {}
-            }
-        }
-        publish == is_publish
-    });
+    media_descriptions.retain(|md| publish == (media::count_send(md) > 0));
     let mut video = false;
     let mut audio = false;
     for md in &media_descriptions {

src/media/mod.rs

@@ -74,3 +74,26 @@ pub fn codecs_from_media_description(
 
     Ok(out)
 }
+
+pub fn count_send(md: &MediaDescription) -> usize {
+    let mut count = 0;
+    let mut minus = 0;
+    for attribute in &md.attributes {
+        match attribute.key.as_str() {
+            "sendonly" => {
+                count += 1;
+                minus += 1;
+            }
+            "simulcast" => {
+                let val = attribute.value.clone().unwrap_or("".to_string());
+                if !val.starts_with("send ") {
+                    break;
+                }
+                count += val.replace("send ", "").split(";").count() - minus;
+                break;
+            }
+            _ => {}
+        }
+    }
+    count
+}