mitm: Disable TTS sink

credits: @Ioniq3
closes: #21

Author: manio

README.md

@@ -13,6 +13,7 @@ Currently it is intended to run as a more-or-less drop-in replacement of the `aa
 - MITM (man-in-the-middle) mode support with the following features:
   - DPI change
   - Disable media sink
+  - Disable TTS sink
   - Developer mode
 
 ## Current project status
@@ -157,6 +158,9 @@ OPTIONS:
         --disable-media-sink
             MITM: Disable media sink
 
+        --disable-tts-sink
+            MITM: Disable TTS sink
+
         --dpi <DPI>
             MITM: Force DPI (experimental)
 

src/io_uring.rs

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

src/main.rs

@@ -102,6 +102,10 @@ struct Args {
     #[clap(long, requires("mitm"))]
     disable_media_sink: bool,
 
+    /// MITM: Disable TTS sink
+    #[clap(long, requires("mitm"))]
+    disable_tts_sink: bool,
+
     /// MITM: Developer mode
     #[clap(long, requires("mitm"))]
     developer_mode: bool,
@@ -304,6 +308,7 @@ fn main() {
     let dpi = args.dpi;
     let developer_mode = args.developer_mode;
     let disable_media_sink = args.disable_media_sink;
+    let disable_tts_sink = args.disable_tts_sink;
 
     // build and spawn main tokio runtime
     let runtime = Builder::new_multi_thread().enable_all().build().unwrap();
@@ -320,6 +325,7 @@ fn main() {
         dpi,
         developer_mode,
         disable_media_sink,
+        disable_tts_sink,
     ));
 
     info!(

src/mitm.rs

@@ -15,7 +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 crate::mitm::AudioStreamType::*;
 use protobuf::text_format::print_to_string_pretty;
 use protobuf::{Enum, Message, MessageDyn};
 use protos::ControlMessageType::{self, *};
@@ -230,6 +230,7 @@ pub async fn pkt_modify_hook(
     dpi: Option<u16>,
     developer_mode: bool,
     disable_media_sink: bool,
+    disable_tts_sink: bool,
 ) -> Result<()> {
     if pkt.channel != 0 {
         return Ok(());
@@ -268,6 +269,24 @@ pub async fn pkt_modify_hook(
                 );
             }
 
+            // disable tts sink
+            if disable_tts_sink {
+                while let Some(svc) = msg.services.iter_mut().find(|svc| {
+                    !svc.media_sink_service.audio_configs.is_empty()
+                        && svc.media_sink_service.audio_type() == AUDIO_STREAM_GUIDANCE
+                }) {
+                    svc.media_sink_service
+                        .as_mut()
+                        .unwrap()
+                        .set_audio_type(AUDIO_STREAM_SYSTEM_AUDIO);
+                }
+                info!(
+                    "{} <yellow>{:?}</>: TTS sink disabled",
+                    get_name(proxy_type),
+                    control.unwrap(),
+                );
+            }
+
             // disable media sink
             if disable_media_sink {
                 msg.services
@@ -432,6 +451,7 @@ pub async fn proxy<A: Endpoint<A> + 'static>(
     dpi: Option<u16>,
     developer_mode: bool,
     disable_media_sink: bool,
+    disable_tts_sink: bool,
 ) -> Result<()> {
     let ssl = ssl_builder(proxy_type).await?;
 
@@ -511,6 +531,7 @@ pub async fn proxy<A: Endpoint<A> + 'static>(
                 dpi,
                 developer_mode,
                 disable_media_sink,
+                disable_tts_sink,
             )
             .await?;
             pkt.encrypt_payload(&mut mem_buf, &mut server).await?;