This commit is contained in:
2025-10-13 20:48:35 +03:00
Unverified
parent 198ed49c80
commit bcf445916c
12 changed files with 225 additions and 300 deletions
-5
View File
@@ -49,7 +49,6 @@ function makeIV(encodedFrame: EncodedFrame): ArrayBuffer {
view.setUint32(4, metadata.synchronizationSource || 0, false); // Middle 4 bytes view.setUint32(4, metadata.synchronizationSource || 0, false); // Middle 4 bytes
view.setUint32(8, 0, false); // Last 4 bytes (padding for 12-byte IV) view.setUint32(8, 0, false); // Last 4 bytes (padding for 12-byte IV)
return ivBuffer; return ivBuffer;
} }
} catch (e) { } catch (e) {
@@ -71,8 +70,6 @@ addEventListener("rtctransform", (event) => {
const isEncrypting = mode === 'encrypt'; const isEncrypting = mode === 'encrypt';
console.log(`E2EE Worker started in ${mode.toUpperCase()} mode`);
let frameCount = 0; let frameCount = 0;
async function transform(encodedFrame: EncodedFrame, controller: TransformStreamDefaultController<EncodedFrame>) { async function transform(encodedFrame: EncodedFrame, controller: TransformStreamDefaultController<EncodedFrame>) {
@@ -131,11 +128,9 @@ addEventListener("rtctransform", (event) => {
result.set(encryptedArray, 0); result.set(encryptedArray, 0);
} }
// CRITICAL: Video frames need ArrayBuffer, not Uint8Array // CRITICAL: Video frames need ArrayBuffer, not Uint8Array
encodedFrame.data = result.buffer; encodedFrame.data = result.buffer;
controller.enqueue(encodedFrame); controller.enqueue(encodedFrame);
} catch (e) { } catch (e) {
// FAIL SECURELY: Never send unencrypted frames // FAIL SECURELY: Never send unencrypted frames
const data = new Uint8Array(encodedFrame.data); const data = new Uint8Array(encodedFrame.data);
+14 -12
View File
@@ -3,6 +3,7 @@ import { randomBytes } from "@/utils/crypto/kdf";
import { b64, ub64 } from "@/utils/utils"; import { b64, ub64 } from "@/utils/utils";
import { ecdhSharedSecret, deriveWrappingKey } from "@/utils/crypto/asymmetric"; import { ecdhSharedSecret, deriveWrappingKey } from "@/utils/crypto/asymmetric";
import { getCurrentKeys } from "@/core/api/authApi"; import { getCurrentKeys } from "@/core/api/authApi";
import type { WrappedSessionKeyPayload } from "@/core/types";
export interface CallSessionKey { export interface CallSessionKey {
key: Uint8Array; key: Uint8Array;
@@ -24,7 +25,8 @@ export interface EncryptedCallMessage {
} }
/** /**
* Generate a call session key (for initiator only) * Generates a new call session key for end-to-end encryption
* @returns Promise that resolves to a session key with its hash for display
*/ */
export async function generateCallSessionKey(): Promise<CallSessionKey> { export async function generateCallSessionKey(): Promise<CallSessionKey> {
// Generate session key material // Generate session key material
@@ -44,7 +46,7 @@ export async function generateCallSessionKey(): Promise<CallSessionKey> {
* Rotate a session key by generating a completely new key * Rotate a session key by generating a completely new key
* This provides forward secrecy for long-running calls * This provides forward secrecy for long-running calls
*/ */
export async function rotateCallSessionKey(_currentSessionKey: CallSessionKey): Promise<CallSessionKey> { export async function rotateCallSessionKey(): Promise<CallSessionKey> {
// Generate new session key material (completely independent of current key) // Generate new session key material (completely independent of current key)
const newSessionKeyMaterial = randomBytes(32); const newSessionKeyMaterial = randomBytes(32);
@@ -127,7 +129,7 @@ export async function deriveCallSessionKeyFromSharedSecret(
/** /**
* Encrypt a call signaling message with the session key * Encrypt a call signaling message with the session key
*/ */
export async function encryptCallMessage(message: any, sessionKey: Uint8Array): Promise<EncryptedCallMessage> { export async function encryptCallMessage(message: Record<string, unknown>, sessionKey: Uint8Array): Promise<EncryptedCallMessage> {
const messageKey = await importAesGcmKey(sessionKey); const messageKey = await importAesGcmKey(sessionKey);
const encrypted = await aesGcmEncrypt(messageKey, new TextEncoder().encode(JSON.stringify(message))); const encrypted = await aesGcmEncrypt(messageKey, new TextEncoder().encode(JSON.stringify(message)));
@@ -143,7 +145,7 @@ export async function encryptCallMessage(message: any, sessionKey: Uint8Array):
/** /**
* Decrypt a call signaling message * Decrypt a call signaling message
*/ */
export async function decryptCallMessage(encryptedMessage: EncryptedCallMessage, sessionKey: Uint8Array): Promise<any> { export async function decryptCallMessage(encryptedMessage: EncryptedCallMessage, sessionKey: Uint8Array): Promise<Record<string, unknown>> {
const messageKey = await importAesGcmKey(sessionKey); const messageKey = await importAesGcmKey(sessionKey);
const decrypted = await aesGcmDecrypt(messageKey, ub64(encryptedMessage.iv), ub64(encryptedMessage.ciphertext)); const decrypted = await aesGcmDecrypt(messageKey, ub64(encryptedMessage.iv), ub64(encryptedMessage.ciphertext));
return JSON.parse(new TextDecoder().decode(decrypted)); return JSON.parse(new TextDecoder().decode(decrypted));
@@ -177,14 +179,11 @@ export function generateCallEmojis(sessionKeyHash: string): string[] {
// HKDF info for CALL key wrapping (distinct from DM's info) // HKDF info for CALL key wrapping (distinct from DM's info)
const CALL_INFO = new Uint8Array([2]); const CALL_INFO = new Uint8Array([2]);
export interface WrappedSessionKeyPayload {
salt: string; // b64 salt used in HKDF
iv2: string; // b64 IV used to wrap session key
wrapped: string; // b64 ciphertext of wrapped session key
}
/** /**
* Wrap a session key for a recipient using ECDH (X25519) and AES-GCM * Wraps a call session key for a specific recipient using ECDH key exchange
* @param recipientPublicKeyB64 - The recipient's public key in base64 format
* @param sessionKey - The session key to wrap
* @returns Promise that resolves to the wrapped session key payload
*/ */
export async function wrapCallSessionKeyForRecipient(recipientPublicKeyB64: string, sessionKey: Uint8Array): Promise<WrappedSessionKeyPayload> { export async function wrapCallSessionKeyForRecipient(recipientPublicKeyB64: string, sessionKey: Uint8Array): Promise<WrappedSessionKeyPayload> {
const keys = getCurrentKeys(); const keys = getCurrentKeys();
@@ -221,7 +220,10 @@ export async function createSharedSecretAndDeriveSessionKey(
} }
/** /**
* Unwrap a received session key using sender's public key * Unwraps a call session key received from a sender using ECDH key exchange
* @param senderPublicKeyB64 - The sender's public key in base64 format
* @param payload - The wrapped session key payload
* @returns Promise that resolves to the unwrapped session key
*/ */
export async function unwrapCallSessionKeyFromSender(senderPublicKeyB64: string, payload: WrappedSessionKeyPayload): Promise<Uint8Array> { export async function unwrapCallSessionKeyFromSender(senderPublicKeyB64: string, payload: WrappedSessionKeyPayload): Promise<Uint8Array> {
const keys = getCurrentKeys(); const keys = getCurrentKeys();
+51 -49
View File
@@ -1,4 +1,4 @@
import type { CallInvite, CallSignalingMessage } from "@/core/types"; import type { CallSignalingMessage, CallAcceptData, CallRejectData, CallOfferData, CallAnswerData, CallIceCandidateData, CallEndData, CallVideoToggleData, CallScreenShareToggleData, CallInviteMessageData } from "@/core/types";
import * as WebRTC from "./webrtc"; import * as WebRTC from "./webrtc";
export interface CallState { export interface CallState {
@@ -9,6 +9,9 @@ export interface CallState {
setRemoteScreenSharing: (enabled: boolean) => void; setRemoteScreenSharing: (enabled: boolean) => void;
} }
/**
* Handles incoming WebSocket messages related to call signaling
*/
export class CallSignalingHandler { export class CallSignalingHandler {
private getState: () => CallState; private getState: () => CallState;
@@ -16,55 +19,56 @@ export class CallSignalingHandler {
this.getState = getState; this.getState = getState;
} }
/**
* Routes incoming call signaling messages to appropriate handlers
*/
handleWebSocketMessage(message: CallSignalingMessage) { handleWebSocketMessage(message: CallSignalingMessage) {
if (message.type !== "call_signaling") {
return;
}
const { data } = message; const { data } = message;
if (!data) { if (!data) {
console.warn("Received call_signaling message with no data:", message); console.warn("Received call signaling message with no data:", message);
return; return;
} }
console.log("Received signaling message:", data.type, "from user", data.fromUserId, "full data:", data); switch (message.type) {
switch (data.type) {
case "call_invite": case "call_invite":
this.handleCallInvite(data as CallInvite); this.handleCallInvite(message, data as CallInviteMessageData);
break; break;
case "call_accept": case "call_accept":
this.handleCallAccept(data); this.handleCallAccept(data as CallAcceptData);
break; break;
case "call_reject": case "call_reject":
this.handleCallReject(data); this.handleCallReject(data as CallRejectData);
break; break;
case "call_offer": case "call_offer":
this.handleCallOffer(data); this.handleCallOffer(message, data as CallOfferData);
break; break;
case "call_answer": case "call_answer":
this.handleCallAnswer(data); this.handleCallAnswer(message, data as CallAnswerData);
break; break;
case "call_ice_candidate": case "call_ice_candidate":
this.handleIceCandidate(data); this.handleIceCandidate(message, data as CallIceCandidateData);
break; break;
case "call_end": case "call_end":
this.handleCallEnd(data); this.handleCallEnd(data as CallEndData);
break; break;
case "call_session_key": case "call_session_key":
this.handleCallSessionKey(data); this.handleCallSessionKey(message);
break; break;
case "call_video_toggle": case "call_video_toggle":
this.handleVideoToggle(data); this.handleVideoToggle(message, data as CallVideoToggleData);
break; break;
case "call_screen_share_toggle": case "call_screen_share_toggle":
this.handleScreenShareToggle(data); this.handleScreenShareToggle(message, data as CallScreenShareToggleData);
break; break;
} }
} }
private async handleCallInvite(data: CallInvite) { /**
const { fromUserId, fromUsername } = data; * Handles incoming call invitation
*/
private async handleCallInvite(message: CallSignalingMessage, data: CallInviteMessageData) {
const { fromUsername } = data;
const fromUserId = message.fromUserId;
const state = this.getState(); const state = this.getState();
// First, create the peer connection in WebRTC service // First, create the peer connection in WebRTC service
@@ -74,7 +78,10 @@ export class CallSignalingHandler {
state.receiveCall(fromUserId, fromUsername); state.receiveCall(fromUserId, fromUsername);
} }
private async handleCallAccept(data: any) { /**
* Handles call acceptance from remote peer
*/
private async handleCallAccept(data: CallAcceptData) {
const { fromUserId } = data; const { fromUserId } = data;
// Initiator should create and send offer now // Initiator should create and send offer now
try { try {
@@ -84,7 +91,10 @@ export class CallSignalingHandler {
} }
} }
private handleCallReject(data: any) { /**
* Handles call rejection from remote peer
*/
private handleCallReject(data: CallRejectData) {
const state = this.getState(); const state = this.getState();
const { fromUserId } = data; const { fromUserId } = data;
@@ -97,22 +107,19 @@ export class CallSignalingHandler {
state.endCall(); state.endCall();
} }
private async handleCallOffer(data: any) { private async handleCallOffer(message: CallSignalingMessage, data: CallOfferData) {
const { fromUserId, data: offer } = data; await WebRTC.handleCallOffer(message.fromUserId, data);
await WebRTC.handleCallOffer(fromUserId, offer);
} }
private async handleCallAnswer(data: any) { private async handleCallAnswer(message: CallSignalingMessage, data: CallAnswerData) {
const { fromUserId, data: answer } = data; await WebRTC.handleCallAnswer(message.fromUserId, data);
await WebRTC.handleCallAnswer(fromUserId, answer);
} }
private async handleIceCandidate(data: any) { private async handleIceCandidate(message: CallSignalingMessage, data: CallIceCandidateData) {
const { fromUserId, data: candidate } = data; await WebRTC.handleIceCandidate(message.fromUserId, data);
await WebRTC.handleIceCandidate(fromUserId, candidate);
} }
private handleCallEnd(data: any) { private handleCallEnd(data: CallEndData) {
const state = this.getState(); const state = this.getState();
const { fromUserId } = data; const { fromUserId } = data;
@@ -125,43 +132,38 @@ export class CallSignalingHandler {
state.endCall(); state.endCall();
} }
private handleCallSessionKey({ sessionKeyHash, data, ...message }: CallSignalingMessage) { private handleCallSessionKey(message: CallSignalingMessage) {
const state = this.getState(); const state = this.getState();
const { sessionKeyHash, data } = message;
if (sessionKeyHash) { if (sessionKeyHash) {
state.setCallSessionKeyHash(sessionKeyHash); state.setCallSessionKeyHash(sessionKeyHash);
} }
if (data?.wrappedSessionKey && message.fromUserId) { if (data && 'wrappedSessionKey' in data && data.wrappedSessionKey && message.fromUserId) {
WebRTC.receiveWrappedSessionKey(message.fromUserId, data.wrappedSessionKey, sessionKeyHash); WebRTC.receiveWrappedSessionKey(message.fromUserId, data.wrappedSessionKey, sessionKeyHash);
} }
} }
private handleVideoToggle(data: any) { private handleVideoToggle(message: CallSignalingMessage, data: CallVideoToggleData) {
console.log("handleVideoToggle called with data:", data);
const state = this.getState(); const state = this.getState();
const { fromUserId, data: toggleData } = data;
if (toggleData && typeof toggleData.enabled === "boolean" && fromUserId) { if (data && typeof data.enabled === "boolean" && message.fromUserId) {
console.log("Setting remote video enabled to:", toggleData.enabled);
// Update Zustand state (for UI) // Update Zustand state (for UI)
state.setRemoteVideoEnabled(toggleData.enabled); state.setRemoteVideoEnabled(data.enabled);
// Update WebRTC internal state (for track routing) // Update WebRTC internal state (for track routing)
WebRTC.setRemoteVideoEnabled(fromUserId, toggleData.enabled); WebRTC.setRemoteVideoEnabled(message.fromUserId, data.enabled);
} else { } else {
console.warn("Invalid toggle data:", data); console.warn("Invalid toggle data:", data);
} }
} }
private handleScreenShareToggle(data: any) { private handleScreenShareToggle(message: CallSignalingMessage, data: CallScreenShareToggleData) {
console.log("handleScreenShareToggle called with data:", data);
const state = this.getState(); const state = this.getState();
const { fromUserId, data: toggleData } = data;
if (toggleData && typeof toggleData.enabled === "boolean" && fromUserId) { if (data && typeof data.enabled === "boolean" && message.fromUserId) {
console.log("Setting remote screen sharing to:", toggleData.enabled);
// Update Zustand state (for UI) // Update Zustand state (for UI)
state.setRemoteScreenSharing(toggleData.enabled); state.setRemoteScreenSharing(data.enabled);
// Update WebRTC internal state (for track routing) // Update WebRTC internal state (for track routing)
WebRTC.setRemoteScreenSharing(fromUserId, toggleData.enabled); WebRTC.setRemoteScreenSharing(message.fromUserId, data.enabled);
} else { } else {
console.warn("Invalid toggle data:", data); console.warn("Invalid toggle data:", data);
} }
+43 -173
View File
@@ -1,10 +1,16 @@
import { getAuthHeaders } from "@/core/api/authApi"; import { getAuthHeaders } from "@/core/api/authApi";
import type { CallSignalingMessage, IceServersResponse } from "@/core/types"; import type { CallSignalingMessage, IceServersResponse, WrappedSessionKeyPayload } from "@/core/types";
import { request } from "@/core/websocket"; import { request } from "@/core/websocket";
import { wrapCallSessionKeyForRecipient, unwrapCallSessionKeyFromSender, rotateCallSessionKey } from "./encryption"; import { wrapCallSessionKeyForRecipient, unwrapCallSessionKeyFromSender, rotateCallSessionKey } from "./encryption";
import { fetchUserPublicKey } from "@/core/api/dmApi"; import { fetchUserPublicKey } from "@/core/api/dmApi";
import { importAesGcmKey } from "@/utils/crypto/symmetric"; import { importAesGcmKey } from "@/utils/crypto/symmetric";
import E2EEWorker from "./e2eeWorker?worker"; import E2EEWorker from "./e2eeWorker?worker";
import { delay } from "@/utils/utils";
// Constants
const DEFAULT_ICE_SERVERS = [{ urls: "stun:fromchat.ru:3478" }];
const KEY_ROTATION_INTERVAL = 10 * 60 * 1000; // 10 minutes
const NEGOTIATION_DELAY = 100; // ms
export interface WebRTCCall { export interface WebRTCCall {
peerConnection: RTCPeerConnection; peerConnection: RTCPeerConnection;
@@ -77,12 +83,13 @@ export function setRemoteScreenShareHandler(handler: (userId: number, stream: Me
onRemoteScreenShare = handler; onRemoteScreenShare = handler;
} }
/**
* Sends a signaling message via WebSocket
*/
async function sendSignalingMessage(message: CallSignalingMessage) { async function sendSignalingMessage(message: CallSignalingMessage) {
if (!authToken) { if (!authToken) {
throw new Error("No auth token available"); throw new Error("No auth token available");
} }
console.log("Sending signaling message:", message.type, "to user", message.toUserId);
await request({ await request({
type: "call_signaling", type: "call_signaling",
@@ -94,12 +101,13 @@ async function sendSignalingMessage(message: CallSignalingMessage) {
}); });
} }
/**
* Fetches ICE servers from the backend, with fallback to default STUN server
*/
async function getIceServers(): Promise<RTCIceServer[]> { async function getIceServers(): Promise<RTCIceServer[]> {
const defaultIceServers = [{ urls: "stun:fromchat.ru:3478" }];
if (!authToken) { if (!authToken) {
console.warn("No auth token available for ICE servers"); console.warn("No auth token available for ICE servers");
return defaultIceServers; return DEFAULT_ICE_SERVERS;
} }
try { try {
@@ -118,10 +126,9 @@ async function getIceServers(): Promise<RTCIceServer[]> {
} }
// Fallback to STUN only if backend fails // Fallback to STUN only if backend fails
return defaultIceServers; return DEFAULT_ICE_SERVERS;
} }
async function createPeerConnection(userId: number): Promise<RTCPeerConnection> { async function createPeerConnection(userId: number): Promise<RTCPeerConnection> {
const iceServers = await getIceServers(); const iceServers = await getIceServers();
@@ -154,11 +161,7 @@ async function createPeerConnection(userId: number): Promise<RTCPeerConnection>
calls.set(userId, call); calls.set(userId, call);
peerConnection.addEventListener("icegatheringstatechange", () => { // Add ICE candidate event listener for sending
console.log("ICE gathering state changed:", peerConnection.iceGatheringState);
});
// Add ICE candidate event listener for debugging and sending
peerConnection.addEventListener("icecandidate", async (event) => { peerConnection.addEventListener("icecandidate", async (event) => {
if (event.candidate) { if (event.candidate) {
// Send ICE candidate to remote peer // Send ICE candidate to remote peer
@@ -180,11 +183,11 @@ async function createPeerConnection(userId: number): Promise<RTCPeerConnection>
}); });
peerConnection.addEventListener("iceconnectionstatechange", () => { peerConnection.addEventListener("iceconnectionstatechange", () => {
console.log("ICE connection state changed:", peerConnection.iceConnectionState); // ICE connection state changed
}); });
peerConnection.addEventListener("signalingstatechange", () => { peerConnection.addEventListener("signalingstatechange", () => {
console.log("Signaling state changed:", peerConnection.signalingState); // Signaling state changed
}); });
// Handle renegotiation when tracks are added/removed // Handle renegotiation when tracks are added/removed
@@ -210,10 +213,9 @@ async function createPeerConnection(userId: number): Promise<RTCPeerConnection>
const offer = await peerConnection.createOffer(); const offer = await peerConnection.createOffer();
await peerConnection.setLocalDescription(offer); await peerConnection.setLocalDescription(offer);
console.log("Sending renegotiation offer to user", userId);
await sendSignalingMessage({ await sendSignalingMessage({
type: "call_offer", type: "call_offer",
fromUserId: 0, fromUserId: 0, // Will be set by server
toUserId: userId, toUserId: userId,
data: offer data: offer
}); });
@@ -230,8 +232,6 @@ async function createPeerConnection(userId: number): Promise<RTCPeerConnection>
// Handle remote stream // Handle remote stream
peerConnection.addEventListener("track", async (event) => { peerConnection.addEventListener("track", async (event) => {
console.log("Received track:", event.track.kind, "from user", userId, "stream ID:", event.streams[0]?.id);
const [remoteStream] = event.streams; const [remoteStream] = event.streams;
const call = calls.get(userId); const call = calls.get(userId);
if (call && remoteStream) { if (call && remoteStream) {
@@ -243,96 +243,55 @@ async function createPeerConnection(userId: number): Promise<RTCPeerConnection>
const receiver = call.peerConnection.getReceivers().find(r => r.track === track); const receiver = call.peerConnection.getReceivers().find(r => r.track === track);
if (receiver && !call.transformedReceivers.has(receiver)) { if (receiver && !call.transformedReceivers.has(receiver)) {
const key = await importAesGcmKey(call.sessionKey); const key = await importAesGcmKey(call.sessionKey);
console.log(`Applying decrypt transform to newly received ${track.kind} track:`);
console.log("- sessionId:", call.sessionId);
console.log("- sessionKey (first 8 bytes):", Array.from(new Uint8Array(call.sessionKey).slice(0, 8)));
// @ts-ignore
receiver.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'decrypt', sessionId: call.sessionId }); receiver.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'decrypt', sessionId: call.sessionId });
call.transformedReceivers.add(receiver); call.transformedReceivers.add(receiver);
console.log(`Decrypt transform applied successfully to ${track.kind} track`);
} }
} catch (error) { } catch (error) {
console.error("Failed to apply E2EE to received track:", error); console.error("Failed to apply E2EE to received track:", error);
} }
} else {
console.log(`Skipping decrypt transform for ${track.kind} track - session key not available or RTCRtpScriptTransform not supported`);
console.log("Session key exists:", !!call.sessionKey);
console.log("RTCRtpScriptTransform available:", !!window.RTCRtpScriptTransform);
} }
// Determine stream type based on track kind and signaling state // Determine stream type based on track kind and signaling state
console.log(`Track received: kind=${track.kind}, isRemoteScreenSharing=${call.isRemoteScreenSharing}, isRemoteVideoEnabled=${call.isRemoteVideoEnabled}`);
if (track.kind === "video") { if (track.kind === "video") {
const receiver = call.peerConnection.getReceivers().find(r => r.track === track);
const transceiver = receiver ? call.peerConnection.getTransceivers().find(t => t.receiver === receiver) : null;
console.log("Video track transceiver mid:", transceiver?.mid);
console.log("Video sender mid:", call.videoSender ? call.peerConnection.getTransceivers().find(t => t.sender === call.videoSender)?.mid : "none");
console.log("Screen share sender mid:", call.screenShareSender ? call.peerConnection.getTransceivers().find(t => t.sender === call.screenShareSender)?.mid : "none");
let isScreenShare = false; let isScreenShare = false;
let isVideo = false; let isVideo = false;
if (call.isRemoteScreenSharing && call.isRemoteVideoEnabled) { if (call.isRemoteScreenSharing && call.isRemoteVideoEnabled) {
// Both active - route based on which one we haven't received yet // Both active - route based on which one we haven't received yet
console.log("Both features active - routing based on received track counts");
console.log("Received video tracks:", call.receivedVideoTrackCount);
console.log("Received screen share tracks:", call.receivedScreenShareTrackCount);
// Simple logic: if we haven't received video yet, this is video // Simple logic: if we haven't received video yet, this is video
// if we haven't received screen share yet, this is screen share // if we haven't received screen share yet, this is screen share
if (call.receivedVideoTrackCount === 0) { if (call.receivedVideoTrackCount === 0) {
isVideo = true; isVideo = true;
call.receivedVideoTrackCount++; call.receivedVideoTrackCount++;
console.log("Routing as video (first video track)");
} else if (call.receivedScreenShareTrackCount === 0) { } else if (call.receivedScreenShareTrackCount === 0) {
isScreenShare = true; isScreenShare = true;
call.receivedScreenShareTrackCount++; call.receivedScreenShareTrackCount++;
console.log("Routing as screen share (first screen share track)");
} else { } else {
// Both already received - this shouldn't happen, log warning // Both already received - this might be a track replacement
console.warn("Both tracks already received, but got another video track!");
console.warn("This might be a track replacement, routing as screen share by default");
isScreenShare = true; isScreenShare = true;
} }
} else if (call.isRemoteScreenSharing) { } else if (call.isRemoteScreenSharing) {
console.log("Only screen share active");
isScreenShare = true; isScreenShare = true;
call.receivedScreenShareTrackCount++; call.receivedScreenShareTrackCount++;
} else if (call.isRemoteVideoEnabled) { } else if (call.isRemoteVideoEnabled) {
console.log("Only video active");
isVideo = true; isVideo = true;
call.receivedVideoTrackCount++; call.receivedVideoTrackCount++;
} else {
console.log("Neither video nor screen share active - this shouldn't happen!");
} }
console.log("Routing decision: isScreenShare:", isScreenShare, "isVideo:", isVideo);
if (isScreenShare) { if (isScreenShare) {
console.log("Detected screen share track, notifying handler");
if (onRemoteScreenShare) { if (onRemoteScreenShare) {
onRemoteScreenShare(userId, remoteStream); onRemoteScreenShare(userId, remoteStream);
} else {
console.warn("onRemoteScreenShare handler not set!");
} }
} else if (isVideo) { } else if (isVideo) {
console.log("Detected video track, notifying handler");
if (onRemoteVideoStream) { if (onRemoteVideoStream) {
onRemoteVideoStream(userId, remoteStream); onRemoteVideoStream(userId, remoteStream);
} else {
console.warn("onRemoteVideoStream handler not set!");
} }
} }
} else if (track.kind === "audio") { } else if (track.kind === "audio") {
console.log("Detected audio track, notifying handler");
// Handle remote audio (existing behavior) // Handle remote audio (existing behavior)
call.remoteStream = remoteStream; call.remoteStream = remoteStream;
if (onRemoteStream) { if (onRemoteStream) {
onRemoteStream(userId, remoteStream); onRemoteStream(userId, remoteStream);
} else {
console.warn("onRemoteStream handler not set!");
} }
} }
} }
@@ -340,7 +299,7 @@ async function createPeerConnection(userId: number): Promise<RTCPeerConnection>
// Handle connection state changes // Handle connection state changes
peerConnection.addEventListener("connectionstatechange", () => { peerConnection.addEventListener("connectionstatechange", () => {
console.log("WebRTC connection state changed:", peerConnection.connectionState); // WebRTC connection state changed
const call = calls.get(userId); const call = calls.get(userId);
if (call) { if (call) {
if (onCallStateChange) { if (onCallStateChange) {
@@ -364,6 +323,12 @@ async function createPeerConnection(userId: number): Promise<RTCPeerConnection>
return peerConnection; return peerConnection;
} }
/**
* Initiates a call to the specified user
* @param userId - The ID of the user to call
* @param username - The username of the user to call
* @returns Promise that resolves to true if call was initiated successfully
*/
export async function initiateCall(userId: number, username: string): Promise<boolean> { export async function initiateCall(userId: number, username: string): Promise<boolean> {
try { try {
// Get user media // Get user media
@@ -400,7 +365,9 @@ export async function initiateCall(userId: number, username: string): Promise<bo
type: "call_invite", type: "call_invite",
fromUserId: 0, // Will be set by server fromUserId: 0, // Will be set by server
toUserId: userId, toUserId: userId,
data: { fromUsername: username } data: {
fromUsername: username
}
}); });
return true; return true;
@@ -448,13 +415,10 @@ export async function sendWrappedCallSessionKey(userId: number, sessionKey: Uint
async function applyE2EETransforms(call: WebRTCCall): Promise<void> { async function applyE2EETransforms(call: WebRTCCall): Promise<void> {
try { try {
// @ts-ignore
if (!call.sessionKey || !window.RTCRtpScriptTransform) { if (!call.sessionKey || !window.RTCRtpScriptTransform) {
console.log("Skipping E2EE transforms - session key or RTCRtpScriptTransform not available");
return; return;
} }
console.log("Applying E2EE transforms for call", call.remoteUserId);
const key = await importAesGcmKey(call.sessionKey); const key = await importAesGcmKey(call.sessionKey);
call.sessionCryptoKey = key; call.sessionCryptoKey = key;
@@ -462,8 +426,6 @@ async function applyE2EETransforms(call: WebRTCCall): Promise<void> {
const receivers = call.peerConnection.getReceivers(); const receivers = call.peerConnection.getReceivers();
for (const receiver of receivers) { for (const receiver of receivers) {
if (receiver.track && !call.transformedReceivers.has(receiver)) { if (receiver.track && !call.transformedReceivers.has(receiver)) {
console.log(`Applying decrypt transform to ${receiver.track.kind} receiver`);
// @ts-ignore
receiver.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'decrypt', sessionId: call.sessionId }); receiver.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'decrypt', sessionId: call.sessionId });
call.transformedReceivers.add(receiver); call.transformedReceivers.add(receiver);
} }
@@ -473,8 +435,6 @@ async function applyE2EETransforms(call: WebRTCCall): Promise<void> {
const senders = call.peerConnection.getSenders(); const senders = call.peerConnection.getSenders();
for (const sender of senders) { for (const sender of senders) {
if (sender.track && !call.transformedSenders.has(sender)) { if (sender.track && !call.transformedSenders.has(sender)) {
console.log(`Applying encrypt transform to ${sender.track.kind} sender`);
// @ts-ignore
sender.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'encrypt', sessionId: call.sessionId }); sender.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'encrypt', sessionId: call.sessionId });
call.transformedSenders.add(sender); call.transformedSenders.add(sender);
} }
@@ -491,13 +451,10 @@ export async function setSessionKey(userId: number, keyBytes: Uint8Array): Promi
return; return;
} }
console.log("setSessionKey: Setting session key for user", userId);
call.sessionKey = keyBytes; call.sessionKey = keyBytes;
call.lastKeyRotation = Date.now(); call.lastKeyRotation = Date.now();
console.log("setSessionKey: Applying E2EE transforms...");
await applyE2EETransforms(call); await applyE2EETransforms(call);
console.log("setSessionKey: E2EE transforms applied successfully");
// Start key rotation timer (rotate every 10 minutes for long calls) // Start key rotation timer (rotate every 10 minutes for long calls)
if (call.keyRotationTimer) { if (call.keyRotationTimer) {
@@ -506,7 +463,7 @@ export async function setSessionKey(userId: number, keyBytes: Uint8Array): Promi
call.keyRotationTimer = setInterval(async () => { call.keyRotationTimer = setInterval(async () => {
await rotateSessionKey(userId); await rotateSessionKey(userId);
}, 10 * 60 * 1000); // 10 minutes }, KEY_ROTATION_INTERVAL);
} }
/** /**
@@ -517,15 +474,8 @@ async function rotateSessionKey(userId: number): Promise<void> {
if (!call || !call.sessionKey) return; if (!call || !call.sessionKey) return;
try { try {
console.log("Rotating session key for call", userId);
// Generate new session key // Generate new session key
const currentSessionKey = { const newSessionKey = await rotateCallSessionKey();
key: call.sessionKey,
hash: "" // We'll generate a new hash
};
const newSessionKey = await rotateCallSessionKey(currentSessionKey);
// Update the call with new session key // Update the call with new session key
call.sessionKey = newSessionKey.key; call.sessionKey = newSessionKey.key;
@@ -533,17 +483,14 @@ async function rotateSessionKey(userId: number): Promise<void> {
// Reapply E2EE transforms with new key // Reapply E2EE transforms with new key
await applyE2EETransforms(call); await applyE2EETransforms(call);
console.log("Session key rotated successfully for call", userId);
} catch (error) { } catch (error) {
console.error("Failed to rotate session key:", error); console.error("Failed to rotate session key:", error);
} }
} }
export async function receiveWrappedSessionKey(fromUserId: number, wrappedPayload: any, sessionKeyHash?: string): Promise<void> { export async function receiveWrappedSessionKey(fromUserId: number, wrappedPayload: WrappedSessionKeyPayload, sessionKeyHash?: string): Promise<void> {
if (!authToken) return; if (!authToken) return;
try { try {
console.log("receiveWrappedSessionKey called for user", fromUserId, "hash:", sessionKeyHash);
const senderPublicKey = await fetchUserPublicKey(fromUserId, authToken); const senderPublicKey = await fetchUserPublicKey(fromUserId, authToken);
if (!senderPublicKey) { if (!senderPublicKey) {
console.error("Failed to get sender public key"); console.error("Failed to get sender public key");
@@ -554,7 +501,6 @@ export async function receiveWrappedSessionKey(fromUserId: number, wrappedPayloa
return; return;
} }
console.log("Unwrapping session key...");
// Unwrap the session key from the encrypted payload // Unwrap the session key from the encrypted payload
const unwrappedSessionKey = await unwrapCallSessionKeyFromSender(senderPublicKey, { const unwrappedSessionKey = await unwrapCallSessionKeyFromSender(senderPublicKey, {
salt: wrappedPayload.salt, salt: wrappedPayload.salt,
@@ -562,15 +508,18 @@ export async function receiveWrappedSessionKey(fromUserId: number, wrappedPayloa
wrapped: wrappedPayload.wrapped wrapped: wrappedPayload.wrapped
}); });
console.log("Session key unwrapped successfully, setting it...");
// Use the unwrapped session key directly (both sides should have the same key) // Use the unwrapped session key directly (both sides should have the same key)
await setSessionKey(fromUserId, unwrappedSessionKey); await setSessionKey(fromUserId, unwrappedSessionKey);
console.log("Session key set successfully for user", fromUserId);
} catch (e) { } catch (e) {
console.error("Failed to unwrap session key:", e); console.error("Failed to unwrap session key:", e);
} }
} }
/**
* Accepts an incoming call from the specified user
* @param userId - The ID of the user who initiated the call
* @returns Promise that resolves to true if call was accepted successfully
*/
export async function acceptCall(userId: number): Promise<boolean> { export async function acceptCall(userId: number): Promise<boolean> {
try { try {
let call = calls.get(userId); let call = calls.get(userId);
@@ -656,7 +605,7 @@ export async function onRemoteAccepted(userId: number): Promise<void> {
try { try {
// Small delay to ensure remote peer finishes processing the accept // Small delay to ensure remote peer finishes processing the accept
// This prevents race conditions where our offer arrives before they're ready // This prevents race conditions where our offer arrives before they're ready
await new Promise(resolve => setTimeout(resolve, 100)); await delay(NEGOTIATION_DELAY);
// Create offer // Create offer
const offer = await call.peerConnection.createOffer(); const offer = await call.peerConnection.createOffer();
@@ -677,21 +626,16 @@ export async function onRemoteAccepted(userId: number): Promise<void> {
async function createE2EETransform(call: WebRTCCall, sessionKey: Uint8Array, sessionId?: string): Promise<void> { async function createE2EETransform(call: WebRTCCall, sessionKey: Uint8Array, sessionId?: string): Promise<void> {
try { try {
// @ts-ignore
if (!sessionKey || !window.RTCRtpScriptTransform) { if (!sessionKey || !window.RTCRtpScriptTransform) {
console.log("Skipping E2EE transform in createE2EETransform - not supported or no session key");
return; return;
} }
console.log("Creating E2EE transform with sessionId:", sessionId);
const key = await importAesGcmKey(sessionKey); const key = await importAesGcmKey(sessionKey);
// Apply to receivers that don't already have transforms // Apply to receivers that don't already have transforms
const receivers = call.peerConnection.getReceivers(); const receivers = call.peerConnection.getReceivers();
for (const receiver of receivers) { for (const receiver of receivers) {
if (receiver.track && !call.transformedReceivers.has(receiver)) { if (receiver.track && !call.transformedReceivers.has(receiver)) {
console.log(`Applying decrypt transform to ${receiver.track.kind} in createE2EETransform`);
// @ts-ignore
receiver.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'decrypt', sessionId }); receiver.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'decrypt', sessionId });
call.transformedReceivers.add(receiver); call.transformedReceivers.add(receiver);
} }
@@ -701,8 +645,6 @@ async function createE2EETransform(call: WebRTCCall, sessionKey: Uint8Array, ses
const senders = call.peerConnection.getSenders(); const senders = call.peerConnection.getSenders();
for (const sender of senders) { for (const sender of senders) {
if (sender.track && !call.transformedSenders.has(sender)) { if (sender.track && !call.transformedSenders.has(sender)) {
console.log(`Applying encrypt transform to ${sender.track.kind} in createE2EETransform`);
// @ts-ignore
sender.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'encrypt', sessionId }); sender.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: 'encrypt', sessionId });
call.transformedSenders.add(sender); call.transformedSenders.add(sender);
} }
@@ -717,11 +659,8 @@ async function createE2EETransform(call: WebRTCCall, sessionKey: Uint8Array, ses
export async function handleCallOffer(userId: number, offer: RTCSessionDescriptionInit): Promise<void> { export async function handleCallOffer(userId: number, offer: RTCSessionDescriptionInit): Promise<void> {
let call = calls.get(userId); let call = calls.get(userId);
console.log("handleCallOffer called for user", userId, "offer type:", offer.type);
// Handle race condition - offer might arrive before peer connection is created // Handle race condition - offer might arrive before peer connection is created
if (!call) { if (!call) {
console.log("No call found for offer, creating peer connection (race condition handling)");
await createPeerConnection(userId); await createPeerConnection(userId);
call = calls.get(userId); call = calls.get(userId);
if (!call) { if (!call) {
@@ -732,7 +671,6 @@ export async function handleCallOffer(userId: number, offer: RTCSessionDescripti
try { try {
// Ensure we have local media before answering // Ensure we have local media before answering
if (!call.localStream) { if (!call.localStream) {
console.log("Getting local media for answer");
try { try {
const localStream = await navigator.mediaDevices.getUserMedia({ audio: true, video: false }); const localStream = await navigator.mediaDevices.getUserMedia({ audio: true, video: false });
call.localStream = localStream; call.localStream = localStream;
@@ -743,24 +681,17 @@ export async function handleCallOffer(userId: number, offer: RTCSessionDescripti
} }
} }
console.log("Setting remote description with", offer.sdp?.split('\n').filter(l => l.includes('m=')).join(', '));
// Set remote description // Set remote description
await call.peerConnection.setRemoteDescription(offer); await call.peerConnection.setRemoteDescription(offer);
console.log("Creating answer...");
// Create answer // Create answer
const answer = await call.peerConnection.createAnswer(); const answer = await call.peerConnection.createAnswer();
await call.peerConnection.setLocalDescription(answer); await call.peerConnection.setLocalDescription(answer);
console.log("Answer created with", answer.sdp?.split('\n').filter(l => l.includes('m=')).join(', '));
// Attach transforms on callee side if session key is available // Attach transforms on callee side if session key is available
// If not available yet, setSessionKey will apply them when it arrives // If not available yet, setSessionKey will apply them when it arrives
if (call.sessionKey) { if (call.sessionKey) {
await createE2EETransform(call, call.sessionKey, call.sessionId); await createE2EETransform(call, call.sessionKey, call.sessionId);
} else {
console.log("Session key not yet available in handleCallOffer - will apply transforms when key arrives");
} }
// Send answer to remote peer // Send answer to remote peer
@@ -770,8 +701,6 @@ export async function handleCallOffer(userId: number, offer: RTCSessionDescripti
toUserId: userId, toUserId: userId,
data: answer data: answer
}); });
console.log("Answer sent successfully");
} catch (error) { } catch (error) {
console.error("Failed to handle offer:", error); console.error("Failed to handle offer:", error);
throw error; throw error;
@@ -785,15 +714,8 @@ export async function handleCallAnswer(userId: number, answer: RTCSessionDescrip
} }
try { try {
console.log("handleCallAnswer - setting remote description");
console.log("Answer SDP media lines:", answer.sdp?.split('\n').filter(l => l.includes('m=')).join(', '));
await call.peerConnection.setRemoteDescription(answer); await call.peerConnection.setRemoteDescription(answer);
console.log("Remote description set successfully");
console.log("Current receivers after answer:", call.peerConnection.getReceivers().map(r => r.track?.kind));
console.log("Current senders after answer:", call.peerConnection.getSenders().map(s => s.track?.kind));
// Reset negotiating flag // Reset negotiating flag
call.isNegotiating = false; call.isNegotiating = false;
@@ -801,8 +723,6 @@ export async function handleCallAnswer(userId: number, answer: RTCSessionDescrip
// If not available yet, setSessionKey will apply them when it arrives // If not available yet, setSessionKey will apply them when it arrives
if (call.sessionKey) { if (call.sessionKey) {
await createE2EETransform(call, call.sessionKey, call.sessionId); await createE2EETransform(call, call.sessionKey, call.sessionId);
} else {
console.log("Session key not yet available in handleCallAnswer - will apply transforms when key arrives");
} }
// Check if there are new receivers with tracks that haven't been notified yet // Check if there are new receivers with tracks that haven't been notified yet
@@ -811,7 +731,6 @@ export async function handleCallAnswer(userId: number, answer: RTCSessionDescrip
for (const receiver of receivers) { for (const receiver of receivers) {
if (receiver.track) { if (receiver.track) {
const track = receiver.track; const track = receiver.track;
console.log("Checking receiver track:", track.kind);
// Find the stream for this track // Find the stream for this track
const transceiver = call.peerConnection.getTransceivers().find(t => t.receiver === receiver); const transceiver = call.peerConnection.getTransceivers().find(t => t.receiver === receiver);
@@ -820,7 +739,6 @@ export async function handleCallAnswer(userId: number, answer: RTCSessionDescrip
if (track.kind === "video" && onRemoteVideoStream) { if (track.kind === "video" && onRemoteVideoStream) {
// Create a MediaStream from the track // Create a MediaStream from the track
const stream = new MediaStream([track]); const stream = new MediaStream([track]);
console.log("Manually notifying remote video stream handler");
onRemoteVideoStream(userId, stream); onRemoteVideoStream(userId, stream);
} }
} }
@@ -835,13 +753,11 @@ export async function handleCallAnswer(userId: number, answer: RTCSessionDescrip
export async function handleIceCandidate(userId: number, candidate: RTCIceCandidateInit): Promise<void> { export async function handleIceCandidate(userId: number, candidate: RTCIceCandidateInit): Promise<void> {
let call = calls.get(userId); let call = calls.get(userId);
if (!call) { if (!call) {
console.warn("No call found for ICE candidate from user", userId, "- might arrive before connection setup");
// Don't create peer connection here - ICE candidates will be gathered again after connection is established // Don't create peer connection here - ICE candidates will be gathered again after connection is established
return; return;
} }
try { try {
console.log("Adding ICE candidate from user", userId, ":", candidate);
await call.peerConnection.addIceCandidate(candidate); await call.peerConnection.addIceCandidate(candidate);
} catch (error) { } catch (error) {
console.error("Failed to add ICE candidate:", error); console.error("Failed to add ICE candidate:", error);
@@ -863,7 +779,8 @@ export function toggleMute(userId: number): boolean {
} }
// Create a silent audio track using Web Audio API // Create a silent audio track using Web Audio API
const audioContext = new (window.AudioContext || (window as any).webkitAudioContext)(); const AudioContextClass = window.AudioContext || (window as unknown as { webkitAudioContext: typeof AudioContext }).webkitAudioContext;
const audioContext = new AudioContextClass();
const oscillator = audioContext.createOscillator(); const oscillator = audioContext.createOscillator();
const gainNode = audioContext.createGain(); const gainNode = audioContext.createGain();
@@ -949,25 +866,14 @@ export async function toggleVideo(userId: number): Promise<boolean> {
const sender = call.peerConnection.addTrack(videoTrack, videoStream); const sender = call.peerConnection.addTrack(videoTrack, videoStream);
call.videoSender = sender; call.videoSender = sender;
console.log("Video track added successfully");
console.log("Current senders:", call.peerConnection.getSenders().map(s => s.track?.kind));
console.log("Current transceivers:", call.peerConnection.getTransceivers().map(t => ({
sender: t.sender.track?.kind,
receiver: t.receiver.track?.kind,
direction: t.direction,
mid: t.mid
})));
// Apply E2EE transform with header-preserving encryption for video // Apply E2EE transform with header-preserving encryption for video
if (call.sessionKey && window.RTCRtpScriptTransform) { if (call.sessionKey && window.RTCRtpScriptTransform) {
try { try {
const key = await importAesGcmKey(call.sessionKey); const key = await importAesGcmKey(call.sessionKey);
console.log("Applying E2EE to video sender with header preservation");
const sender = call.peerConnection.getSenders().find(s => s.track === videoTrack); const sender = call.peerConnection.getSenders().find(s => s.track === videoTrack);
if (sender && !call.transformedSenders.has(sender)) { if (sender && !call.transformedSenders.has(sender)) {
sender.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: "encrypt", sessionId: call.sessionId }); sender.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: "encrypt", sessionId: call.sessionId });
call.transformedSenders.add(sender); call.transformedSenders.add(sender);
console.log("E2EE applied to video sender successfully");
} }
} catch (error) { } catch (error) {
console.error("Failed to apply E2EE to video:", error); console.error("Failed to apply E2EE to video:", error);
@@ -977,14 +883,10 @@ export async function toggleVideo(userId: number): Promise<boolean> {
// Notify local video stream handler // Notify local video stream handler
if (onLocalVideoStream) { if (onLocalVideoStream) {
console.log("Calling onLocalVideoStream handler with stream:", videoStream);
onLocalVideoStream(userId, videoStream); onLocalVideoStream(userId, videoStream);
} else {
console.warn("onLocalVideoStream handler is not set!");
} }
// Send signaling message to notify remote peer // Send signaling message to notify remote peer
console.log("Sending call_video_toggle with enabled: true");
await sendSignalingMessage({ await sendSignalingMessage({
type: "call_video_toggle", type: "call_video_toggle",
fromUserId: 0, fromUserId: 0,
@@ -992,7 +894,6 @@ export async function toggleVideo(userId: number): Promise<boolean> {
data: { enabled: true } data: { enabled: true }
}); });
console.log("Video enabled successfully");
return true; return true;
} catch (error) { } catch (error) {
console.error("Failed to enable video:", error); console.error("Failed to enable video:", error);
@@ -1062,9 +963,8 @@ export async function toggleScreenShare(userId: number): Promise<boolean> {
writable: false, writable: false,
configurable: true configurable: true
}); });
console.log("Set screen share stream ID to:", screenStream.id);
} catch (e) { } catch (e) {
console.warn("Failed to set custom stream ID, using default:", screenStream.id); // Use default stream ID
} }
call.screenShareStream = screenStream; call.screenShareStream = screenStream;
@@ -1075,7 +975,6 @@ export async function toggleScreenShare(userId: number): Promise<boolean> {
// Handle when user stops sharing via browser UI // Handle when user stops sharing via browser UI
videoTrack.addEventListener("ended", async () => { videoTrack.addEventListener("ended", async () => {
console.log("Screen share track ended by browser controls");
// Clean up screen share state // Clean up screen share state
if (call.screenShareStream) { if (call.screenShareStream) {
@@ -1118,7 +1017,6 @@ export async function toggleScreenShare(userId: number): Promise<boolean> {
// Send signaling message FIRST to notify remote peer before adding track // Send signaling message FIRST to notify remote peer before adding track
// This ensures the receiver knows it's screen share before the track arrives // This ensures the receiver knows it's screen share before the track arrives
console.log("Sending screen share toggle BEFORE adding track");
await sendSignalingMessage({ await sendSignalingMessage({
type: "call_screen_share_toggle", type: "call_screen_share_toggle",
fromUserId: 0, fromUserId: 0,
@@ -1127,56 +1025,32 @@ export async function toggleScreenShare(userId: number): Promise<boolean> {
}); });
// Small delay to ensure signaling message is processed before track arrives // Small delay to ensure signaling message is processed before track arrives
await new Promise(resolve => setTimeout(resolve, 100)); await delay(NEGOTIATION_DELAY);
const sender = call.peerConnection.addTrack(videoTrack, screenStream); const sender = call.peerConnection.addTrack(videoTrack, screenStream);
call.screenShareSender = sender; call.screenShareSender = sender;
console.log("Screen share track added, immediately applying E2EE transform");
// CRITICAL: Apply E2EE transform IMMEDIATELY after track is added // CRITICAL: Apply E2EE transform IMMEDIATELY after track is added
if (call.sessionKey && window.RTCRtpScriptTransform) { if (call.sessionKey && window.RTCRtpScriptTransform) {
try { try {
const key = await importAesGcmKey(call.sessionKey); const key = await importAesGcmKey(call.sessionKey);
console.log("Applying E2EE to screen share sender:");
console.log("- sessionId:", call.sessionId);
console.log("- sessionKey (first 8 bytes):", Array.from(new Uint8Array(call.sessionKey).slice(0, 8)));
console.log("Available senders:", call.peerConnection.getSenders().map(s => ({
track: s.track?.kind,
id: s.track?.id
})));
console.log("Looking for screen share track:", videoTrack.id);
const sender = call.peerConnection.getSenders().find(s => s.track === videoTrack); const sender = call.peerConnection.getSenders().find(s => s.track === videoTrack);
console.log("Found screen share sender:", !!sender);
if (sender && !call.transformedSenders.has(sender)) { if (sender && !call.transformedSenders.has(sender)) {
sender.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: "encrypt", sessionId: call.sessionId }); sender.transform = new RTCRtpScriptTransform(new E2EEWorker(), { key, mode: "encrypt", sessionId: call.sessionId });
call.transformedSenders.add(sender); call.transformedSenders.add(sender);
console.log("E2EE applied to screen share sender successfully");
} else {
console.log("Screen share sender not found or already transformed");
} }
} catch (error) { } catch (error) {
console.error("Failed to apply E2EE to screen share:", error); console.error("Failed to apply E2EE to screen share:", error);
throw error; // Fail securely throw error; // Fail securely
} }
} else {
console.log("Skipping E2EE for screen share - session key not available yet");
console.log("Session key exists:", !!call.sessionKey);
console.log("RTCRtpScriptTransform available:", !!window.RTCRtpScriptTransform);
} }
// Let browser handle screen share settings naturally
// Avoid applying constraints that might cause glitches
// Notify local screen share handler // Notify local screen share handler
if (onLocalScreenShare) { if (onLocalScreenShare) {
onLocalScreenShare(userId, screenStream); onLocalScreenShare(userId, screenStream);
} }
console.log("Screen share setup complete - using existing session key:", call.sessionId);
return true; return true;
} catch (error) { } catch (error) {
console.error("Failed to enable screen sharing:", error); console.error("Failed to enable screen sharing:", error);
@@ -1259,12 +1133,10 @@ export function cleanupCall(userId: number): void {
export function setRemoteVideoEnabled(userId: number, enabled: boolean): void { export function setRemoteVideoEnabled(userId: number, enabled: boolean): void {
const call = calls.get(userId); const call = calls.get(userId);
if (call) { if (call) {
console.log(`Setting remote video enabled to ${enabled} for user ${userId}`);
call.isRemoteVideoEnabled = enabled; call.isRemoteVideoEnabled = enabled;
// Reset counter when feature is disabled // Reset counter when feature is disabled
if (!enabled) { if (!enabled) {
call.receivedVideoTrackCount = 0; call.receivedVideoTrackCount = 0;
console.log("Reset video track counter");
} }
} }
} }
@@ -1275,12 +1147,10 @@ export function setRemoteVideoEnabled(userId: number, enabled: boolean): void {
export function setRemoteScreenSharing(userId: number, enabled: boolean): void { export function setRemoteScreenSharing(userId: number, enabled: boolean): void {
const call = calls.get(userId); const call = calls.get(userId);
if (call) { if (call) {
console.log(`Setting remote screen sharing to ${enabled} for user ${userId}`);
call.isRemoteScreenSharing = enabled; call.isRemoteScreenSharing = enabled;
// Reset counter when feature is disabled // Reset counter when feature is disabled
if (!enabled) { if (!enabled) {
call.receivedScreenShareTrackCount = 0; call.receivedScreenShareTrackCount = 0;
console.log("Reset screen share track counter");
} }
} }
} }
@@ -1,5 +1,5 @@
import { Navigate } from "react-router-dom"; import { Navigate } from "react-router-dom";
import { MINIMUM_WIDTH } from "../config"; import { MINIMUM_WIDTH } from "@/core/config";
import useWindowSize from "./useWindowSize"; import useWindowSize from "./useWindowSize";
export default function useDownloadAppScreen() { export default function useDownloadAppScreen() {
+73 -10
View File
@@ -435,11 +435,17 @@ export interface DialogProps {
} }
// Call types // Call types
export interface CallInvite extends CallSignalingData { export interface CallSignalingData {
type: "call_invite";
fromUserId: number; fromUserId: number;
toUserId: number;
}
export interface CallInviteData {
fromUsername: string;
}
export interface CallInviteMessageData {
fromUsername: string; fromUsername: string;
timestamp: string;
} }
export type CallSignalingDataType = "call_offer" | "call_answer" | "call_ice_candidate" | "call_end" | "call_invite" | "call_accept" | "call_reject" | "call_session_key" | "call_signaling" | "call_video_toggle" | "call_screen_share_toggle"; export type CallSignalingDataType = "call_offer" | "call_answer" | "call_ice_candidate" | "call_end" | "call_invite" | "call_accept" | "call_reject" | "call_session_key" | "call_signaling" | "call_video_toggle" | "call_screen_share_toggle";
@@ -449,19 +455,76 @@ export interface CallSignalingMessage extends WebSocketMessage {
fromUserId: number; fromUserId: number;
toUserId: number; toUserId: number;
sessionKeyHash?: string; sessionKeyHash?: string;
data: any; data: CallSignalingMessageData;
}
export type CallSignalingMessageData =
| CallInviteMessageData
| CallAcceptData
| CallRejectData
| CallOfferData
| CallAnswerData
| CallIceCandidateData
| CallEndData
| CallSessionKeyData
| CallVideoToggleData
| CallScreenShareToggleData;
export interface CallAcceptData {
fromUserId: number;
}
export interface CallRejectData {
fromUserId: number;
}
export interface CallOfferData extends RTCSessionDescriptionInit {
}
export interface CallAnswerData extends RTCSessionDescriptionInit {
}
export interface CallIceCandidateData extends RTCIceCandidateInit {
}
export interface CallEndData {
fromUserId: number;
}
export interface CallSessionKeyData {
wrappedSessionKey?: WrappedSessionKeyPayload;
}
export interface CallVideoToggleData {
enabled: boolean;
}
export interface CallScreenShareToggleData {
enabled: boolean;
}
export interface CallVideoToggleMessageData {
fromUserId: number;
data: CallVideoToggleData;
}
export interface CallScreenShareToggleMessageData {
fromUserId: number;
data: CallScreenShareToggleData;
}
export interface WrappedSessionKeyPayload {
salt: string;
iv2: string;
wrapped: string;
} }
export interface CallVideoToggleMessage extends CallSignalingMessage { export interface CallVideoToggleMessage extends CallSignalingMessage {
type: "call_video_toggle"; type: "call_video_toggle";
data: { data: CallVideoToggleData;
enabled: boolean;
};
} }
export interface CallScreenShareToggleMessage extends CallSignalingMessage { export interface CallScreenShareToggleMessage extends CallSignalingMessage {
type: "call_screen_share_toggle"; type: "call_screen_share_toggle";
data: { data: CallScreenShareToggleData;
enabled: boolean;
};
} }
+5 -5
View File
@@ -6,9 +6,9 @@
*/ */
import { API_WS_BASE_URL } from "./config"; import { API_WS_BASE_URL } from "./config";
import type { CallSignalingMessage, WebSocketMessage } from "./types"; import type { WebSocketMessage } from "./types";
import { delay } from "@/utils/utils"; import { delay } from "@/utils/utils";
import { CallSignalingHandler } from "@/core/calls/signaling"; import { CallSignalingHandler } from "./calls/signaling";
/** /**
* Creates a new WebSocket connection to the chat server * Creates a new WebSocket connection to the chat server
@@ -109,11 +109,11 @@ async function onError() {
websocket.addEventListener("message", (e) => { websocket.addEventListener("message", (e) => {
try { try {
const response: CallSignalingMessage = JSON.parse(e.data); const response: WebSocketMessage<any> = JSON.parse(e.data);
// Handle call signaling messages // Handle call signaling messages
if (callSignalingHandler && response.type === "call_signaling") { if (callSignalingHandler && response.type === "call_signaling" && response.data) {
callSignalingHandler.handleWebSocketMessage(response); callSignalingHandler.handleWebSocketMessage(response.data);
} }
// Route message to global handler if set // Route message to global handler if set
+6 -35
View File
@@ -4,6 +4,7 @@ import { CallSignalingHandler } from "@/core/calls/signaling";
import { setCallSignalingHandler } from "@/core/websocket"; import { setCallSignalingHandler } from "@/core/websocket";
import { generateCallSessionKey, generateCallEmojis } from "@/core/calls/encryption"; import { generateCallSessionKey, generateCallEmojis } from "@/core/calls/encryption";
import { createRef, useEffect } from "react"; import { createRef, useEffect } from "react";
import { doAfterInteraction } from "@/utils/utils";
// Global refs shared across all instances // Global refs shared across all instances
let globalRemoteAudioRef = createRef<HTMLAudioElement>(); let globalRemoteAudioRef = createRef<HTMLAudioElement>();
@@ -91,14 +92,7 @@ export default function useCall() {
}); });
el.play().catch(() => { el.play().catch(() => {
// Try to play after user interaction if autoplay is blocked doAfterInteraction(() => el.play());
const playAfterInteraction = () => {
el.play().catch(() => {});
document.removeEventListener("click", playAfterInteraction);
document.removeEventListener("touchstart", playAfterInteraction);
};
document.addEventListener("click", playAfterInteraction);
document.addEventListener("touchstart", playAfterInteraction);
}); });
} catch (e) { } catch (e) {
console.warn("failed to attach remote stream:", e); console.warn("failed to attach remote stream:", e);
@@ -107,9 +101,7 @@ export default function useCall() {
// Set up local video stream handler // Set up local video stream handler
WebRTC.setLocalVideoStreamHandler((_userId: number, stream: MediaStream | null) => { WebRTC.setLocalVideoStreamHandler((_userId: number, stream: MediaStream | null) => {
console.log("Local video stream handler called, stream:", stream, "ref exists:", !!localVideoRef.current);
if (!localVideoRef.current) { if (!localVideoRef.current) {
console.warn("Local video ref not available yet");
return; return;
} }
const el = localVideoRef.current; const el = localVideoRef.current;
@@ -118,9 +110,9 @@ export default function useCall() {
el.muted = true; // Always mute local video to avoid feedback el.muted = true; // Always mute local video to avoid feedback
el.autoplay = true; el.autoplay = true;
if (stream) { if (stream) {
console.log("Playing local video stream");
el.play().catch((err) => { el.play().catch((err) => {
console.error("Failed to play local video:", err); console.error("Failed to play local video:", err);
doAfterInteraction(() => el.play()).catch(() => {});
}); });
} }
} catch (e) { } catch (e) {
@@ -130,9 +122,7 @@ export default function useCall() {
// Set up remote video stream handler // Set up remote video stream handler
WebRTC.setRemoteVideoStreamHandler((_userId: number, stream: MediaStream | null) => { WebRTC.setRemoteVideoStreamHandler((_userId: number, stream: MediaStream | null) => {
console.log("Remote video stream handler called, stream:", stream, "ref exists:", !!remoteVideoRef.current);
if (!remoteVideoRef.current) { if (!remoteVideoRef.current) {
console.warn("Remote video ref not available yet");
return; return;
} }
const el = remoteVideoRef.current; const el = remoteVideoRef.current;
@@ -141,16 +131,9 @@ export default function useCall() {
el.muted = false; el.muted = false;
el.autoplay = true; el.autoplay = true;
if (stream) { if (stream) {
console.log("Playing remote video stream");
el.play().catch((err) => { el.play().catch((err) => {
console.error("Failed to play remote video:", err); console.error("Failed to play remote video:", err);
const playAfterInteraction = () => { doAfterInteraction(() => el.play()).catch(() => {});
el.play().catch(() => {});
document.removeEventListener("click", playAfterInteraction);
document.removeEventListener("touchstart", playAfterInteraction);
};
document.addEventListener("click", playAfterInteraction);
document.addEventListener("touchstart", playAfterInteraction);
}); });
} }
} catch (e) { } catch (e) {
@@ -160,9 +143,7 @@ export default function useCall() {
// Set up local screen share handler // Set up local screen share handler
WebRTC.setLocalScreenShareHandler((_userId: number, stream: MediaStream | null) => { WebRTC.setLocalScreenShareHandler((_userId: number, stream: MediaStream | null) => {
console.log("Local screen share handler called, stream:", stream, "ref exists:", !!localScreenShareRef.current);
if (!localScreenShareRef.current) { if (!localScreenShareRef.current) {
console.warn("Local screen share ref not available yet");
return; return;
} }
const el = localScreenShareRef.current; const el = localScreenShareRef.current;
@@ -171,9 +152,9 @@ export default function useCall() {
el.muted = true; el.muted = true;
el.autoplay = true; el.autoplay = true;
if (stream) { if (stream) {
console.log("Playing local screen share");
el.play().catch((err) => { el.play().catch((err) => {
console.error("Failed to play local screen share:", err); console.error("Failed to play local screen share:", err);
doAfterInteraction(() => el.play()).catch(() => {});
}); });
} }
} catch (e) { } catch (e) {
@@ -183,9 +164,7 @@ export default function useCall() {
// Set up remote screen share handler // Set up remote screen share handler
WebRTC.setRemoteScreenShareHandler((_userId: number, stream: MediaStream | null) => { WebRTC.setRemoteScreenShareHandler((_userId: number, stream: MediaStream | null) => {
console.log("Remote screen share handler called, stream:", stream, "ref exists:", !!remoteScreenShareRef.current);
if (!remoteScreenShareRef.current) { if (!remoteScreenShareRef.current) {
console.warn("Remote screen share ref not available yet");
return; return;
} }
const el = remoteScreenShareRef.current; const el = remoteScreenShareRef.current;
@@ -194,16 +173,9 @@ export default function useCall() {
el.muted = false; el.muted = false;
el.autoplay = true; el.autoplay = true;
if (stream) { if (stream) {
console.log("Playing remote screen share");
el.play().catch((err) => { el.play().catch((err) => {
console.error("Failed to play remote screen share:", err); console.error("Failed to play remote screen share:", err);
const playAfterInteraction = () => { doAfterInteraction(() => el.play()).catch(() => {});
el.play().catch(() => {});
document.removeEventListener("click", playAfterInteraction);
document.removeEventListener("touchstart", playAfterInteraction);
};
document.addEventListener("click", playAfterInteraction);
document.addEventListener("touchstart", playAfterInteraction);
}); });
} }
} catch (e) { } catch (e) {
@@ -245,7 +217,6 @@ export default function useCall() {
const hasPermission = await requestAudioPermissions(); const hasPermission = await requestAudioPermissions();
if (!hasPermission) { if (!hasPermission) {
console.log("Audio permission denied");
return; return;
} }
+1 -1
View File
@@ -6,7 +6,7 @@ import {
fetchDMHistory, fetchDMHistory,
decryptDm, decryptDm,
sendDMViaWebSocket sendDMViaWebSocket
} from "../../../core/api/dmApi"; } from "@/core/api/dmApi";
import type { User, Message, DmEncryptedJSON } from "@/core/types"; import type { User, Message, DmEncryptedJSON } from "@/core/types";
import { websocket } from "@/core/websocket"; import { websocket } from "@/core/websocket";
@@ -88,11 +88,9 @@ export function CallWindow() {
} else { } else {
if (shouldRender) { if (shouldRender) {
// Call ended - start exit animation // Call ended - start exit animation
console.log(`Call ended - starting exit animation. Was minimized: ${wasMinimized}`);
setIsVisible(false); setIsVisible(false);
// After animation completes, stop rendering // After animation completes, stop rendering
const timer = setTimeout(() => { const timer = setTimeout(() => {
console.log("Exit animation complete, removing window");
setShouldRender(false); setShouldRender(false);
setCallData(null); setCallData(null);
setWasMinimized(false); setWasMinimized(false);
@@ -195,12 +193,6 @@ export function CallWindow() {
left: pipPosition.x, left: pipPosition.x,
top: pipPosition.y top: pipPosition.y
} : undefined} } : undefined}
ref={(el) => {
if (el && !isVisible) {
console.log(`Window classes: ${el.className}`);
console.log(`Window state - was minimized: ${wasMinimized}, visible: ${isVisible}`);
}
}}
onMouseDown={(e) => { onMouseDown={(e) => {
if (call.isMinimized) { if (call.isMinimized) {
// Only start dragging if not clicking on a button // Only start dragging if not clicking on a button
@@ -6,7 +6,7 @@ import {
sendDmWithFiles, sendDmWithFiles,
editDmEnvelope, editDmEnvelope,
deleteDmEnvelope deleteDmEnvelope
} from "../../../../../core/api/dmApi"; } from "@/core/api/dmApi";
import type { DmEncryptedJSON, DmEnvelope, DMWebSocketMessage, EncryptedMessageJson, Message } from "@/core/types"; import type { DmEncryptedJSON, DmEnvelope, DMWebSocketMessage, EncryptedMessageJson, Message } from "@/core/types";
import type { UserState } from "@/pages/chat/state"; import type { UserState } from "@/pages/chat/state";
+30
View File
@@ -43,4 +43,34 @@ export function ub64(s: string): Uint8Array {
export function id<T extends Element = HTMLElement>(id: string): T { export function id<T extends Element = HTMLElement>(id: string): T {
return document.getElementById(id) as unknown as T return document.getElementById(id) as unknown as T
}
/**
* Runs the specified callback after `click` or `touchstart` event is triggered.
*
* @param action The action to perform after interaction
* @returns A function to clean up the event listeners.
*/
export function doAfterInteraction<T>(action?: () => (T | Promise<T>)): Promise<T> {
return new Promise((resolve, reject) => {
function doAfterInteractionInner() {
document.removeEventListener("click", doAfterInteractionInner);
document.removeEventListener("touchstart", doAfterInteractionInner);
try {
const result = action?.();
if (result instanceof Promise) {
result.then(resolve);
} else {
resolve(result as T);
}
} catch (error) {
reject(error);
}
}
document.addEventListener("click", doAfterInteractionInner);
document.addEventListener("touchstart", doAfterInteractionInner);
setTimeout(reject, 10000);
});
} }