mitm: implement Disable media sink option

Author: manio

src/io_uring.rs

@@ -267,6 +267,7 @@ pub async fn io_loop(
     mitm: bool,
     dpi: Option<u16>,
     developer_mode: bool,
+    disable_media_sink: bool,
 ) -> Result<()> {
     info!("{} 🛰️ Starting TCP server...", NAME);
     let bind_addr = format!("0.0.0.0:{}", TCP_SERVER_PORT).parse().unwrap();
@@ -349,6 +350,7 @@ pub async fn io_loop(
                 rxr_md,
                 dpi,
                 developer_mode,
+                disable_media_sink,
             ));
             from_stream = tokio_uring::spawn(proxy(
                 ProxyType::MobileDevice,
@@ -359,6 +361,7 @@ pub async fn io_loop(
                 rxr_hu,
                 dpi,
                 developer_mode,
+                disable_media_sink,
             ));
         } else {
             // We need to copy in both directions...

src/main.rs

@@ -98,6 +98,10 @@ struct Args {
     #[clap(long, requires("mitm"))]
     dpi: Option<u16>,
 
+    /// MITM: Disable media sink
+    #[clap(long, requires("mitm"))]
+    disable_media_sink: bool,
+
     /// MITM: Developer mode
     #[clap(long, requires("mitm"))]
     developer_mode: bool,
@@ -299,6 +303,7 @@ fn main() {
     let mitm = args.mitm;
     let dpi = args.dpi;
     let developer_mode = args.developer_mode;
+    let disable_media_sink = args.disable_media_sink;
 
     // build and spawn main tokio runtime
     let runtime = Builder::new_multi_thread().enable_all().build().unwrap();
@@ -314,6 +319,7 @@ fn main() {
         mitm,
         dpi,
         developer_mode,
+        disable_media_sink,
     ));
 
     info!(

src/mitm.rs

@@ -15,6 +15,7 @@ use tokio_uring::buf::BoundedBuf;
 // protobuf stuff:
 include!(concat!(env!("OUT_DIR"), "/protos/mod.rs"));
 use crate::mitm::protos::*;
+use crate::mitm::AudioStreamType::AUDIO_STREAM_MEDIA;
 use protobuf::text_format::print_to_string_pretty;
 use protobuf::{Enum, Message, MessageDyn};
 use protos::ControlMessageType::{self, *};
@@ -221,6 +222,7 @@ pub async fn pkt_modify_hook(
     pkt: &mut Packet,
     dpi: Option<u16>,
     developer_mode: bool,
+    disable_media_sink: bool,
 ) -> Result<()> {
     if pkt.channel != 0 {
         return Ok(());
@@ -259,6 +261,17 @@ pub async fn pkt_modify_hook(
                 );
             }
 
+            // disable media sink
+            if disable_media_sink {
+                msg.services
+                    .retain(|svc| svc.media_sink_service.audio_type() != AUDIO_STREAM_MEDIA);
+                info!(
+                    "{} <yellow>{:?}</>: media sink disabled",
+                    NAME,
+                    control.unwrap(),
+                );
+            }
+
             // enabling developer mode
             if developer_mode {
                 msg.set_make("Google".into());
@@ -411,6 +424,7 @@ pub async fn proxy<A: Endpoint<A> + 'static>(
     mut rxr: Receiver<Packet>,
     dpi: Option<u16>,
     developer_mode: bool,
+    disable_media_sink: bool,
 ) -> Result<()> {
     let ssl = ssl_builder(proxy_type).await?;
 
@@ -484,7 +498,7 @@ pub async fn proxy<A: Endpoint<A> + 'static>(
     loop {
         // handling data from opposite device's thread, which needs to be transmitted
         if let Ok(mut pkt) = rx.try_recv() {
-            pkt_modify_hook(&mut pkt, dpi, developer_mode).await?;
+            pkt_modify_hook(&mut pkt, dpi, developer_mode, disable_media_sink).await?;
             pkt.encrypt_payload(&mut mem_buf, &mut server).await?;
             pkt.transmit(&mut device).await?;