wip

livekit · Sep 4, 2023 · 75cd73e · 75cd73e
1 parent 61d27c3
commit 75cd73e
Show file tree

Hide file tree

Showing 12 changed files with 174 additions and 315 deletions.
diff --git a/examples/wgpu_room/src/service.rs b/examples/wgpu_room/src/service.rs
@@ -196,18 +196,7 @@ async fn service_task(inner: Arc<ServiceInner>, mut cmd_rx: mpsc::UnboundedRecei
                 if let Some(state) = running_state.as_ref() {
                     let e2ee_manager = state.room.e2ee_manager();
                     if let Some(key_provider) = e2ee_manager.key_provider() {
-                        // Ratchet all local cryptors
-                        e2ee_manager.frame_cryptors().iter().for_each(
-                            |((participant_identity, _), cryptor)| {
-                                if participant_identity.as_str()
-                                    != state.room.local_participant().identity().as_str()
-                                {
-                                    return;
-                                }
-
-                                key_provider.ratchet_key(cryptor.participant_id(), 0);
-                            },
-                        );
+                        key_provider.ratchet_shared_key(0);
                     }
                 }
             }

diff --git a/livekit-webrtc/src/frame_cryptor.rs b/livekit-webrtc/src/frame_cryptor.rs
diff --git a/livekit-webrtc/src/lib.rs b/livekit-webrtc/src/lib.rs
@@ -60,11 +60,11 @@ pub mod video_frame;
 pub mod video_source;
 pub mod video_stream;
 pub mod video_track;
-pub mod frame_cryptor;
 
 #[cfg(not(target_arch = "wasm32"))]
 pub mod native {
     pub use crate::imp::audio_resampler;
+    pub use crate::imp::frame_cryptor;
     pub use crate::imp::yuv_helper;
     pub use webrtc_sys::webrtc::ffi::create_random_uuid;
 }