Web 版 IM 即时通讯消息加密算法实现方案

• 描述：用于验证消息的来源和完整性。发送方使用自己的私钥对消息的哈希值进行加密，得到签名。接收方使用发送方的公钥对签名进行解密，并将结果与自己对消息计算的哈希值对比。如果匹配，则证明消息确实来自该发送者且未被篡改。
• 作用：提供身份验证和不可否认性。

第二部分：方案一：静态非对称加密（基础方案）

2.1 方案概述与流程

这是最直观的 E2EE 方案。每个用户拥有一对固定的长期密钥。发送者使用接收者的公钥直接加密每一条消息。

流程：

1. 密钥生成与上传：用户注册/登录时，前端生成 RSA 密钥对。私钥本地保存，公钥上传至服务器。
2. 获取公钥：A 要给 B 发消息时，A 的前端从服务器获取 B 的公钥。
3. 加密与发送：A 使用 B 的公钥加密消息，将密文发送给服务器。
4. 中继与接收：服务器将密文转发给 B。
5. 解密与展示：B 使用自己的私钥解密消息。

序列图：

+---------+ +-------------+ +---------+ +----------+
| 用户 A | | 前端 A | | 后端服务器 | | 前端 B |
+---------+ +-------------+ +---------+ +----------+
| | | | | | | |
| 1. 生成密钥对 | | | |------------------>| |
| 2. 上传公钥 | | | |------------------>| |
| 3. 存储公钥 | | | |<------------------| |
... |
| 4. 输入消息 | | | |------------------>| |
| 5. 获取 B 的公钥请求 | | | |------------------>| |
| 6. 返回 B 的公钥 | | | |<------------------| |
| 7. 用 B 的公钥加密消息 | | | |---(CPU Intensive)--| |
| 8. 发送密文给 B | | | |------------------>| |
| 9. 转发密文给 B | | | |------------------>| |
| 10. 用 B 的私钥解密 | | | |---(CPU Intensive)-- |
| 11. 显示明文 | | | |

2.2 前端 Vue 实现（使用 node-forge）

1. 安装依赖

npm install node-forge

2. 核心工具类 crypto.js

// utils/crypto.js
import forge from 'node-forge';

// 生成 RSA 密钥对
export function generateRSAKeyPair() {
    return new Promise((resolve, reject) => {
        forge.pki.rsa.generateKeyPair({
            bits: 2048,
            workers: 2
        }, (err, keypair) => {
            if (err) {
                reject(err);
                return;
            }
            const publicKey = forge.pki.publicKeyToPem(keypair.publicKey);
            const privateKey = forge.pki.privateKeyToPem(keypair.privateKey);
            resolve({ publicKey, privateKey });
        });
    });
}

// 使用公钥加密消息 (RSA-OAEP padding，比 PKCS#1 更安全)
export function encryptMessageWithPublicKey(publicKeyPem, message) {
    try {
        const publicKey = forge.pki.publicKeyFromPem(publicKeyPem);
        // 将字符串转换为字节缓冲区
        const encodedMessage = forge.util.encodeUtf8(message);
        // 使用 OAEP 填充进行加密，SHA-256 作为摘要算法
        const encrypted = publicKey.encrypt(encodedMessage, 'RSA-OAEP', {
            md: forge.md.sha256.create(),
        });
        // 将二进制数据转换为 Base64 字符串以便网络传输
        return forge.util.encode64(encrypted);
    } catch (error) {
        console.error('加密失败:', error);
        throw new Error('消息加密失败');
    }
}

// 使用私钥解密消息
export function decryptMessageWithPrivateKey(privateKeyPem, encryptedMessageBase64) {
    try {
        const privateKey = forge.pki.privateKeyFromPem(privateKeyPem);
        // 将 Base64 字符串解码为二进制数据
        const encryptedData = forge.util.decode64(encryptedMessageBase64);
        // 使用 OAEP 填充进行解密
        const decrypted = privateKey.decrypt(encryptedData, 'RSA-OAEP', {
            md: forge.md.sha256.create(),
        });
        // 将字节缓冲区转换回字符串
        return forge.util.decodeUtf8(decrypted);
    } catch (error) {
        console.error('解密失败:', error);
        throw new Error('消息解密失败，可能是密钥不匹配或消息已损坏');
    }
}

// 安全地存储私钥到本地存储 (提示：这并不绝对安全，可考虑使用浏览器安全 API 或用户密码二次加密)
export function savePrivateKeySecurely(userId, privateKey) {
    // 示例：简单存储。生产环境应使用更安全的方式，例如用用户密码派生出的密钥进行加密后再存储。
    localStorage.setItem(`im_private_key_${userId}`, privateKey);
}

// 从本地存储获取私钥
export function getPrivateKey(userId) {
    return localStorage.getItem(`im_private_key_${userId}`);
}

3. Vue 组件中使用

<!-- components/Chat.vue -->
<template>
    <div>
        <div v-for="msg in messages" :key="msg.id" :class="['message', msg.sender === currentUser.id ? 'sent' : 'received']">
            <p><strong>{{ msg.senderName }}:</strong> {{ msg.decryptedContent || '**加密消息**' }}</p>
            <small>{{ msg.timestamp }}</small>
            <button v-if="!msg.decryptedContent && msg.sender !== currentUser.id" @click="decryptMessage(msg)"> 解密 </button>
        </div>
        <div>
            <textarea v-model="newMessage" @keyup.enter="sendMessage" placeholder="输入消息..."></textarea>
            <button @click="sendMessage" :disabled="!newMessage.trim()">发送</button>
        </div>
    </div>
</template>
<script>
import { encryptMessageWithPublicKey, decryptMessageWithPrivateKey, getPrivateKey } from '@/utils/crypto';
import { apiGetUserPublicKey, apiSendMessage } from '@/api/chat';

export default {
    name: 'Chat',
    props: ['currentUser', 'targetUser'],
    data() {
        return {
            newMessage: '',
            messages: [], // 消息格式：{ id, sender, senderName, encryptedContent, decryptedContent, timestamp }
            websocket: null,
        };
    },
    async mounted() {
        this.connectWebSocket();
        // ... 加载历史消息 (历史消息也是密文，需要手动解密)
    },
    methods: {
        async sendMessage() {
            if (!this.newMessage.trim()) return;
            try {
                // 1. 从服务器获取目标用户的公钥
                const publicKeyResponse = await apiGetUserPublicKey(this.targetUser.id);
                const receiverPublicKey = publicKeyResponse.data;
                // 2. 加密消息
                const encryptedContent = encryptMessageWithPublicKey(receiverPublicKey, this.newMessage.trim());
                // 3. 构建消息对象并通过 WebSocket 或 API 发送
                const messagePayload = {
                    receiverId: this.targetUser.id,
                    type: 'text',
                    content: encryptedContent, // 发送的是密文
                    isEncrypted: true, // 标记此消息已加密
                    timestamp: new Date().toISOString(),
                };
                // 4. 通过 WebSocket 发送
                this.websocket.send(JSON.stringify(messagePayload));
                // 5. 乐观更新 UI (显示为'已发送，加密中')
                this.messages.push({
                    id: Date.now(), // 临时 ID
                    sender: this.currentUser.id,
                    senderName: '我',
                    encryptedContent: '**消息已加密发送**',
                    decryptedContent: null,
                    timestamp: '刚刚',
                });
                this.newMessage = '';
            } catch (error) {
                console.error('发送消息失败:', error);
                this.$notify({ type: 'error', title: '发送失败', text: '加密或发送消息时出错' });
            }
        },
        async decryptMessage(message) {
            if (message.decryptedContent) return; // 已解密
            try {
                // 获取自己的私钥
                const privateKey = getPrivateKey(this.currentUser.id);
                if (!privateKey) {
                    throw new Error('未找到解密密钥，请重新登录');
                }
                // 解密消息
                const decryptedContent = decryptMessageWithPrivateKey(privateKey, message.encryptedContent);
                // 更新消息状态
                message.decryptedContent = decryptedContent;
                // 可选：通知后端该消息已成功解密阅读（已读回执）
            } catch (error) {
                console.error('解密消息失败:', error);
                this.$notify({ type: 'error', title: '解密失败', text: '无法解密此消息' });
            }
        },
        connectWebSocket() {
            // ... WebSocket 连接逻辑，用于实时接收消息
            this.websocket.onmessage = (event) => {
                const messageData = JSON.parse(event.data);
                this.handleIncomingMessage(messageData);
            };
        },
        handleIncomingMessage(messageData) {
            // 处理接收到的消息
            const newMsg = {
                id: messageData.id,
                sender: messageData.senderId,
                senderName: messageData.senderName,
                encryptedContent: messageData.content, // 收到的是密文
                decryptedContent: null, // 初始化为 null，等待用户点击解密
                timestamp: new Date(messageData.timestamp).toLocaleTimeString(),
            };
            this.messages.push(newMsg);
            // 可选：如果是当前会话的消息，可以提示用户有新消息
            if (this.targetUser.id === messageData.senderId) {
                this.$notify({ type: 'info', title: '新消息', text: `来自 ${messageData.senderName} 的加密消息` });
            }
        },
    },
    beforeUnmount() {
        if (this.websocket) {
            this.websocket.close();
        }
    },
};
</script>
<style scoped>
/* ... 聊天样式 ... */
.message.received button {
    margin-left: 10px;
    font-size: 0.8em;
}
</style>

2.3 后端 Java 实现（Spring Boot）

后端在此方案中角色较轻，主要负责公钥的存储查询和消息的转发。

1. 实体类

// User 实体，增加公钥字段
@Entity
@Table(name = "im_users")
@Data
public class User {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String username;
    private String nickname;
    
    @Column(columnDefinition = "TEXT") // 公钥是长文本
    private String rsaPublicKey; // 用户的 RSA 公钥 (PEM 格式)
    
    // ... 其他字段如密码、创建时间等
}

// 消息实体
@Entity
@Table(name = "im_messages")
@Data
public class ChatMessage {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private Long senderId;
    private Long receiverId;
    
    @Column(columnDefinition = "TEXT") // 消息内容可能是长文本密文
    private String content;
    private Boolean isEncrypted; // 标记消息是否加密
    private String messageType; // "text", "image", "file" etc.
    private Instant timestamp;
    
    // ... 其他字段如状态（已发送、已送达、已读）
}

2. Controller 层

@RestController
@RequestMapping("/api/chat")
public class ChatController {
    @Autowired
    private UserRepository userRepository;
    
    @Autowired
    private SimpMessagingTemplate messagingTemplate; // Spring WebSocket 消息发送模板
    
    @Autowired
    private ChatMessageRepository messageRepository;

    // 获取指定用户的公钥
    @GetMapping("/user/{userId}/public-key")
    public ResponseEntity<?> getUserPublicKey(@PathVariable Long userId) {
        User user = userRepository.findById(userId).orElseThrow(() -> new ResourceNotFoundException("User not found"));
        if (user.getRsaPublicKey() == null) {
            return ResponseEntity.badRequest().body("User does not have a public key");
        }
        // 返回公钥字符串
        return ResponseEntity.ok().body(Collections.singletonMap("publicKey", user.getRsaPublicKey()));
    }

    // 接收并转发加密消息 (通过 HTTP API，也可通过 WebSocket 接收)
    @PostMapping("/message")
    public ResponseEntity<Void> sendEncryptedMessage(@RequestBody EncryptedMessageRequest request) {
        // 1. 验证发送者身份 (可以从 JWT token 中获取当前用户 ID，并与 request.getSenderId() 对比)
        // 2. 将消息保存到数据库（存密文）
        ChatMessage message = new ChatMessage();
        message.setSenderId(request.getSenderId());
        message.setReceiverId(request.getReceiverId());
        message.setContent(request.getContent()); // 这里已经是加密后的密文
        message.setIsEncrypted(true);
        message.setMessageType(request.getType());
        message.setTimestamp(Instant.now());
        messageRepository.save(message);

        // 3. 通过 WebSocket 实时转发给接收者
        // 目的地格式： /user/{userId}/queue/messages
        String destination = "/queue/messages";
        MessageDeliveryDto deliveryDto = new MessageDeliveryDto();
        deliveryDto.setId(message.getId());
        deliveryDto.setSenderId(request.getSenderId());
        deliveryDto.setSenderName(""); // 需要查询发送者姓名
        deliveryDto.setContent(request.getContent());
        deliveryDto.setEncrypted(true);
        deliveryDto.setType(request.getType());
        deliveryDto.setTimestamp(message.getTimestamp());
        messagingTemplate.convertAndSendToUser(
            request.getReceiverId().toString(),
            destination,
            deliveryDto
        );

        // 4. 可选：发送送达回执给发送者
        return ResponseEntity.ok().build();
    }

    // 请求体定义
    public static class EncryptedMessageRequest {
        private Long senderId;
        private Long receiverId;
        private String content;
        private String type;
        // getters and setters
    }

    // WebSocket 转发消息 DTO
    public static class MessageDeliveryDto {
        private Long id;
        private Long senderId;
        private String senderName;
        private String content;
        private Boolean isEncrypted;
        private String type;
        private Instant timestamp;
        // getters and setters
    }
}

3. WebSocket 配置

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").setAllowedOriginPatterns("*").withSockJS();
    }

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        registry.setApplicationDestinationPrefixes("/app");
        registry.enableSimpleBroker("/topic", "/queue");
        registry.setUserDestinationPrefix("/user");
    }

    // 可选：配置身份验证拦截器
    @Override
    public void configureClientInboundChannel(ChannelRegistration registration) {
        registration.interceptors(new AuthChannelInterceptor());
    }
}

2.4 密钥管理、注册与登录集成

1. 用户注册/登录时生成密钥

// Vuex Store (store/auth.js) 或登录组件中
import { generateRSAKeyPair, savePrivateKeySecurely } from '@/utils/crypto';
import { apiRegister, apiLogin, apiUpdatePublicKey } from '@/api/auth';

const actions = {
    async register({ commit }, userData) {
        try {
            // 1. 生成密钥对
            const { publicKey, privateKey } = await generateRSAKeyPair();
            // 2. 将公钥包含在注册数据中
            const registrationData = {
                ...userData,
                publicKey: publicKey
            };
            // 3. 调用注册 API
            const response = await apiRegister(registrationData);
            // 4. 注册成功，在本地安全存储私钥
            if (response.data.userId) {
                savePrivateKeySecurely(response.data.userId, privateKey);
                commit('SET_USER', response.data);
                commit('SET_PRIVATE_KEY', privateKey); // 也可存入 Vuex state
            }
            return response;
        } catch (error) {
            console.error('注册失败:', error);
            throw error;
        }
    },
    async login({ commit }, credentials) {
        try {
            const response = await apiLogin(credentials);
            const user = response.data;
            // 检查本地是否已有该用户的私钥
            let privateKey = getPrivateKey(user.id);
            // 如果没有，说明可能是新设备登录，需要引导用户重新生成密钥对并上传公钥
            if (!privateKey) {
                this.$router.push('/key-setup'); // 跳转到密钥设置页面
                return;
            }
            commit('SET_USER', user);
            commit('SET_PRIVATE_KEY', privateKey);
            return response;
        } catch (error) {
            console.error('登录失败:', error);
            throw error;
        }
    }
};

2. 密钥设置页面

<!-- views/KeySetup.vue -->
<template>
    <div>
        <h2>安全密钥设置</h2>
        <p>检测到您在新设备登录，需要为您生成新的加密密钥。</p>
        <p>生成后，您将无法在其他设备上解密之前的消息。</p>
        <!-- 这是该方案的一个缺点 -->
        <button @click="generateNewKeys">生成新密钥</button>
    </div>
</template>
<script>
import { generateRSAKeyPair, savePrivateKeySecurely } from '@/utils/crypto';
import { apiUpdatePublicKey } from '@/api/auth';

export default {
    methods: {
        async generateNewKeys() {
            try {
                const { publicKey, privateKey } = await generateRSAKeyPair();
                const userId = this.$store.state.auth.user.id;
                // 上传新公钥到服务器
                await apiUpdatePublicKey(userId, publicKey);
                // 保存新私钥到本地
                savePrivateKeySecurely(userId, privateKey);
                this.$store.commit('auth/SET_PRIVATE_KEY', privateKey);
                this.$router.go(-1); // 返回上一页
            } catch (error) {
                console.error('密钥生成失败:', error);
            }
        }
    }
};
</script>

2.5 方案一优缺点总结

• 优点：
  • 概念简单：易于理解和实现。
  • 符合 E2EE：服务器从未接触明文。
  • 无需状态管理：服务器无需管理会话密钥。
• 缺点：
  • 性能极差：RSA 加密非常慢，尤其是对长消息。频繁发送消息会导致前端界面卡顿。
  • 无前向保密性 (Forward Secrecy)：如果用户的长期私钥将来被泄露，攻击者可以解密该用户所有过去和未来的通信记录。这是致命的缺点。
  • 密钥管理复杂：用户更换设备后，旧设备上的私钥无法同步，导致无法解密新消息，且旧消息也无法在新设备上解密。解决方案（如用主密码加密私钥然后同步）会引入新的复杂性。
  • 无法认证发送者：接收者知道消息是用自己的公钥加密的，但无法密码学上证实发送者是谁。（方案二通过数字签名解决此问题）。

结论：方案一适用于学习原理或对安全要求不高、消息频率极低的场景，不推荐用于生产环境。

第三部分：方案二：非对称加密 + 数字签名（增强身份验证）

3.1 方案概述与流程

此方案在方案一的基础上增加了数字签名，解决了发送者身份验证和消息完整性的问题。

流程（A 发消息给 B）：

1. A 生成消息明文 M。
2. A 使用 B 的公钥加密 M，得到密文 C。
3. A 使用自己的私钥对密文 C（或明文 M 的哈希）进行签名，得到签名 S。
   • 签名明文哈希更常见，因为性能更好且符合标准。
4. A 将 { cipherText: C, signature: S, senderId: A } 发送给服务器。
5. 服务器转发给 B。
6. B 收到后，使用 A 的公钥验证签名 S。如果验证失败，则丢弃消息。
7. 验证通过后，B 使用自己的私钥解密 C，得到明文 M。

序列图：

+---------+ +-------------+ +----------+ +----------+
| 用户 A | | 前端 A | | 后端服务器 | | 前端 B |
+---------+ +-------------+ +----------+ +----------+
| | | | | | | |
| 1. 输入消息 | | | |------------------>| |
| 2. 获取 B 的公钥 | | | |------------------>| |
| 3. 返回 B 的公钥 | | | |<------------------| |
| 4. 用 B 的公钥加密消息 | | | |---(CPU Intensive)--| |
| 5. 用 A 的私钥签名 | | | |---(CPU Intensive)--| |
| 6. 发送 (密文 + 签名) | | | |------------------>| |
| 7. 转发给 B | | | |------------------>| |
| 8. 获取 A 的公钥 | | | |<------------------? |
| 9. 验证签名 | | | |---(CPU Intensive)-- |
| 10. 验证成功？ | | | |---Yes------------| |
| | | | | | | | |
| 11. 用 B 的私钥解密 | | | |---(CPU Intensive)-- |
| 12. 显示明文 | | | |---No-------------| |
| 13. 丢弃消息，报错 | | | |

3.2 前端 Vue 实现增强

在原有 crypto.js 工具类中添加签名和验证功能。

// utils/crypto.js (新增函数)

// 使用私钥对消息进行签名 (通常是对消息的哈希值进行签名)
export function signMessageWithPrivateKey(privateKeyPem, message) {
    try {
        const privateKey = forge.pki.privateKeyFromPem(privateKeyPem);
        const md = forge.md.sha256.create(); // 创建 SHA-256 哈希上下文
        md.update(message, 'utf8'); // 更新哈希内容
        const signature = privateKey.sign(md); // 使用私钥对哈希值进行签名
        return forge.util.encode64(signature); // 返回 Base64 编码的签名
    } catch (error) {
        console.error('签名失败:', error);
        throw new Error('消息签名失败');
    }
}

// 使用公钥验证签名
export function verifySignatureWithPublicKey(publicKeyPem, message, signatureBase64) {
    try {
        const publicKey = forge.pki.publicKeyFromPem(publicKeyPem);
        const md = forge.md.sha256.create();
        md.update(message, 'utf8');
        const signature = forge.util.decode64(signatureBase64);
        const isVerified = publicKey.verify(md.digest().getBytes(), signature);
        return isVerified;
    } catch (error) {
        console.error('验证签名失败:', error);
        return false; // 验证过程中出现异常，视为验证失败
    }
}

// 注意：以上是对明文签名。更常见的做法是对加密后的密文进行签名，以避免接收方先解密再验证的逻辑循环。
// 另一种标准做法是：签名明文，然后将（密文 + 签名）一起发送。
// 以下是签名明文的版本（更常见）：
export function signPlainTextWithPrivateKey(privateKeyPem, plainText) {
    // ... 实现同上，参数 plainText 代替 message
}

export function verifyPlainTextSignature(publicKeyPem, plainText, signatureBase64) {
    // ... 实现同上
}

修改发送和接收消息的逻辑：

发送消息（增加签名）

// components/Chat.vue - sendMessage 方法修改
async sendMessage() {
    // ... 获取接收者公钥、加密消息 (得到 encryptedContent) ...
    // 使用发送者 (A) 的私钥对【明文】进行签名
    const privateKey = getPrivateKey(this.currentUser.id);
    const signature = signPlainTextWithPrivateKey(privateKey, this.newMessage.trim());
    const messagePayload = {
        receiverId: this.targetUser.id,
        type: 'text',
        content: encryptedContent,
        signature: signature, // 新增签名字段
        isEncrypted: true,
        timestamp: new Date().toISOString(),
    };
    // ... 发送 messagePayload ...
}

接收消息（增加验证）

// components/Chat.vue - handleIncomingMessage 或 decryptMessage 方法修改
async decryptMessage(message) {
    // ... 获取自己的私钥解密 (得到 decryptedContent) ...
    // 获取发送者的公钥来验证签名
    try {
        const senderPublicKeyResponse = await apiGetUserPublicKey(message.sender);
        const senderPublicKey = senderPublicKeyResponse.data.publicKey;
        // 验证签名：用发送者公钥验证解密出的明文 against 消息中的签名
        const isSignatureValid = verifyPlainTextSignature(senderPublicKey, decryptedContent, message.signature);
        if (isSignatureValid) {
            message.decryptedContent = decryptedContent;
            message.signatureStatus = 'verified';
            this.$notify({ type: 'success', title: '解密成功', text: '消息签名验证通过' });
        } else {
            message.decryptedContent = decryptedContent + " [警告：签名验证失败，消息可能被篡改或来源不可信]";
            message.signatureStatus = 'invalid';
            this.$notify({ type: 'warning', title: '安全警告', text: '消息解密成功，但签名验证失败' });
        }
    } catch (error) {
        console.error('验证签名时出错:', error);
        message.decryptedContent = decryptedContent + " [错误：签名验证过程出错]";
        message.signatureStatus = 'error';
    }
}

3.3 后端 Java 实现增强

后端需要存储和转发签名。

1. 修改消息实体和 DTO

// ChatMessage 实体增加字段
public class ChatMessage {
    // ... 其他字段 ...
    @Column(columnDefinition = "TEXT")
    private String signature; // 存储消息的数字签名 (Base64 编码)
}

// Controller 中的 DTO 也要增加 signature 字段
public static class EncryptedMessageRequest {
    private Long senderId;
    private Long receiverId;
    private String content;
    private String signature; // 新增
    private String type; // getters and setters
}

public static class MessageDeliveryDto {
    // ... 其他字段 ...
    private String signature; // 新增
    // getters and setters
}

2. 后端签名验证（可选）

后端原则上不处理明文，但可以选择性地验证签名以增加一层安全防护（防止恶意客户端发送伪造签名的消息）。

// 在 ChatController 中，接收消息时可选验证签名
@PostMapping("/message")
public ResponseEntity<Void> sendEncryptedMessage(@RequestBody EncryptedMessageRequest request) {
    // 1. (可选) 验证发送者身份后，验证消息签名
    User sender = userRepository.findById(request.getSenderId()).orElseThrow(...);
    // 假设签名是对明文哈希的签名，但后端没有明文，无法验证。
    // 如果签名是对密文的签名，后端可以验证。
    // 通常后端不验证，因为它是密文。验证工作主要在客户端进行。
    // 2. 存储和转发消息（包括签名）
    // ... 原有逻辑 ...
}

3.4 方案二优缺点总结

• 优点：
  • 提供身份验证和不可否认性：接收者可以确信消息来自特定的发送者。
  • 提供完整性保护：签名验证失败意味着消息或签名在传输过程中被篡改。
  • 继承方案一的 E2EE 优点。
• 缺点：
  • 性能进一步下降：每次发送消息都需要进行两次昂贵的非对称加密操作（加密和签名）。
  • 依然没有前向保密性：长期私钥泄露的风险依然存在。
  • 密钥管理问题依旧。

结论：方案二解决了身份验证问题，但加剧了性能问题，且仍未解决前向保密这一核心安全缺陷。仍不推荐用于高频生产环境。

第四部分：方案三：混合加密系统（推荐生产方案）

4.1 方案概述与流程

这是现代安全通信的标准模型，完美结合了非对称加密和对称加密的优点。核心思想是：使用非对称加密安全地交换一个临时的对称密钥，然后使用这个对称密钥来加密实际的消息。

核心概念：前向保密 (Forward Secrecy)

• 每次会话或定期更换对称密钥（称为会话密钥）。
• 即使攻击者破解了用户的长期私钥，也无法解密过去的通信记录，因为过去的会话密钥早已丢弃，且是用当时的临时密钥加密的。
• 这是生产级 E2EE 系统的必备特性。

流程（A 发起与 B 的会话）：

1. 会话初始化：
   • A 生成一个随机的对称会话密钥SK 和初始化向量 IV。
   • A 获取 B 的长期公钥。
   • A 使用 B 的公钥加密(SK, IV)，得到密钥信封Envelope。
   • A 可选地用自已的私钥对Envelope签名。
   • A 将Envelope和签名发送给 B。这个过程称为密钥交换。
2. 发送消息：
   • A 使用会话密钥SK和IV，通过 AES 算法对消息明文进行对称加密，得到密文 C。
   • A 将密文 C 发送给 B。
3. 接收消息：
   • B 收到Envelope后，用自己的长期私钥解密，得到SK和IV。
   • B 收到密文 C 后，使用SK和IV进行对称解密，得到明文。

序列图：

+---------+ +-------------+ +----------+ +----------+
| 用户 A | | 前端 A | | 后端服务器 | | 前端 B |
+---------+ +-------------+ +----------+ +----------+
| | | | | | | |
| 1. 开始会话 | | | |------------------>| |
| 2. 生成会话密钥 SK | | | |---(Generate SK)---| |
| 3. 获取 B 的公钥 | | | |------------------>| |
| 4. 返回 B 的公钥 | | | |<------------------| |
| 5. 用 B 的公钥加密 SK | | | |---(CPU Intensive)--| |
| 6. 发送密钥信封 | | | |------------------>| |
| 7. 转发给 B | | | |------------------>| |
| 8. 用 B 的私钥解密得 SK | | | |---(CPU Intensive)-- |
| 9. 存储 SK | | | |---(Now ready)----| |
... |
| | | | | | | | |
| 10. 输入消息 | | | | |------------------>| |
| | | | | | | | |
| 11. 用 SK 加密消息 | | | | |---(Very Fast)-----| |
| | | | | | | | |
| 12. 发送消息密文 | | | | |------------------>| |
| | | | | | | | |
| 13. 转发给 B | | | | |------------------>| |
| | | | | | | | |
| 14. 用 SK 解密消息 | | | |---(Very Fast)----| |
| | | | | | | | |
| 15. 显示明文 | | | |

注：密钥交换通常只在会话开始时进行一次，后续所有消息都使用高效的对称加密。

4.2 前端 Vue 实现（重大修改）

我们需要一个全面的会话密钥管理机制。

1. 扩展加密工具类 crypto.js

// utils/crypto.js (新增混合加密函数)

// 生成随机的对称密钥和 IV (用于 AES-CBC 模式)
export function generateSymmetricKey() {
    const key = forge.random.getBytesSync(32); // AES-256 需要 32 字节的密钥
    const iv = forge.random.getBytesSync(16); // AES-CBC 需要 16 字节的 IV
    return {
        key: forge.util.encode64(key), // 编码为 Base64 便于存储
        iv: forge.util.encode64(iv)
    };
}

// 使用对称密钥加密消息 (AES-CBC)
export function encryptWithSymmetricKey(keyBase64, ivBase64, message) {
    try {
        const key = forge.util.decode64(keyBase64);
        const iv = forge.util.decode64(ivBase64);
        const cipher = forge.cipher.createCipher('AES-CBC', key);
        cipher.start({ iv: iv });
        cipher.update(forge.util.createBuffer(message, 'utf8'));
        cipher.finish();
        const encrypted = cipher.output;
        return forge.util.encode64(encrypted.data);
    } catch (error) {
        console.error('对称加密失败:', error);
        throw new Error('消息加密失败');
    }
}

// 使用对称密钥解密消息 (AES-CBC)
export function decryptWithSymmetricKey(keyBase64, ivBase64, encryptedMessageBase64) {
    try {
        const key = forge.util.decode64(keyBase64);
        const iv = forge.util.decode64(ivBase64);
        const encryptedData = forge.util.decode64(encryptedMessageBase64);
        const decipher = forge.cipher.createDecipher('AES-CBC', key);
        decipher.start({ iv: iv });
        decipher.update(forge.util.createBuffer(encryptedData));
        const result = decipher.finish();
        if (result) {
            return decipher.output.toString('utf8');
        } else {
            throw new Error('解密失败：可能密钥或 IV 不正确');
        }
    } catch (error) {
        console.error('对称解密失败:', error);
        throw new Error('消息解密失败');
    }
}

// 使用公钥加密对称密钥（封装信封）
export function encryptSymmetricKeyWithPublicKey(publicKeyPem, symmetricKeyObj) {
    // 将对称密钥对象转换为字符串以便加密
    const keyDataStr = JSON.stringify(symmetricKeyObj);
    return encryptMessageWithPublicKey(publicKeyPem, keyDataStr); // 复用之前的 RSA 加密函数
}

// 使用私钥解密对称密钥（解封装信封）
export function decryptSymmetricKeyWithPrivateKey(privateKeyPem, encryptedEnvelope) {
    const decryptedKeyStr = decryptMessageWithPrivateKey(privateKeyPem, encryptedEnvelope);
    return JSON.parse(decryptedKeyStr);
}

2. 会话密钥管理 (Vuex Store)

我们需要一个地方来存储和管理与不同用户的会话密钥。

// store/modules/chatSession.js
const state = {
    // 会话密钥库：{ [targetUserId]: { key: '...', iv: '...', timestamp: ... } }
    sessionKeys: {},
};

const mutations = {
    SET_SESSION_KEY(state, { userId, keyData }) {
        state.sessionKeys[userId] = {
            ...keyData,
            timestamp: Date.now(), // 记录密钥生成时间，可用于过期策略
        };
    },
    CLEAR_SESSION_KEY(state, userId) {
        delete state.sessionKeys[userId];
    },
    CLEAR_ALL_SESSIONS(state) {
        state.sessionKeys = {};
    },
};

const actions = {
    // 为与特定用户的会话生成并交换密钥
    async establishSession({ commit, rootState }, targetUserId) {
        try {
            // 1. 生成对称密钥
            const symmetricKey = generateSymmetricKey();
            // 2. 获取目标用户的公钥
            const publicKeyResponse = await apiGetUserPublicKey(targetUserId);
            const receiverPublicKey = publicKeyResponse.data.publicKey;
            // 3. 用对方的公钥加密我们的对称密钥（创建信封）
            const encryptedEnvelope = encryptSymmetricKeyWithPublicKey(receiverPublicKey, symmetricKey);
            // 4. 发送密钥交换消息
            await apiSendKeyExchange({
                receiverId: targetUserId,
                encryptedEnvelope: encryptedEnvelope, // 可以在这里添加签名
            });
            // 5. 将会话密钥保存在本地 Store 中
            commit('SET_SESSION_KEY', { userId: targetUserId, keyData: symmetricKey });
            console.log(`会话密钥已建立并发送给用户 ${targetUserId}`);
            return symmetricKey;
        } catch (error) {
            console.error('建立会话失败:', error);
            throw new Error('无法建立安全会话');
        }
    },
    // 获取与某个用户的会话密钥，如果没有则建立
    async getOrCreateSessionKey({ state, dispatch }, targetUserId) {
        let keyData = state.sessionKeys[targetUserId];
        // 简单检查密钥是否存在以及是否过期（例如，设置 1 小时过期）
        const isExpired = keyData && (Date.now() - keyData.timestamp > 3600000); // 1 小时
        if (!keyData || isExpired) {
            keyData = await dispatch('establishSession', targetUserId);
        }
        return keyData;
    },
};

export default {
    namespaced: true,
    state,
    mutations,
    actions,
};

3. 修改聊天组件

<!-- components/Chat.vue (重大修改) -->
<script>
import { encryptWithSymmetricKey, decryptWithSymmetricKey } from '@/utils/crypto';
import { mapActions } from 'vuex';

export default {
    // ... methods: {
    async sendMessage() {
        if (!this.newMessage.trim()) return;
        try {
            // 1. 获取或创建与目标用户的会话密钥
            const sessionKeyData = await this.$store.dispatch('chatSession/getOrCreateSessionKey', this.targetUser.id);
            // 2. 使用对称密钥加密消息 (非常快)
            const encryptedContent = encryptWithSymmetricKey(
                sessionKeyData.key,
                sessionKeyData.iv,
                this.newMessage.trim()
            );
            // 3. 发送消息
            const messagePayload = {
                receiverId: this.targetUser.id,
                type: 'text',
                content: encryptedContent,
                isEncrypted: true,
                isSymmetric: true, // 新增字段，表明是对称加密
                timestamp: new Date().toISOString(),
            };
            this.websocket.send(JSON.stringify(messagePayload));
            // 4. 乐观更新 UI
            this.messages.push({
                id: Date.now(),
                sender: this.currentUser.id,
                senderName: '我',
                encryptedContent: '**消息已加密发送**',
                decryptedContent: this.newMessage.trim(), // 乐观显示明文
                timestamp: '刚刚',
            });
            this.newMessage = '';
        } catch (error) {
            console.error('发送消息失败:', error);
            this.$notify({ type: 'error', title: '发送失败', text: '加密或发送消息时出错' });
        }
    },
    async handleIncomingMessage(messageData) {
        // 处理接收到的消息
        const newMsg = {
            id: messageData.id,
            sender: messageData.senderId,
            senderName: messageData.senderName,
            encryptedContent: messageData.content,
            decryptedContent: null,
            isSymmetric: messageData.isSymmetric, // 检查是否是对称加密
            timestamp: new Date(messageData.timestamp).toLocaleTimeString(),
        };
        // 如果是对称加密消息，并且我们有会话密钥，尝试自动解密
        if (newMsg.isSymmetric) {
            const sessionKey = this.$store.state.chatSession.sessionKeys[newMsg.sender];
            if (sessionKey) {
                try {
                    newMsg.decryptedContent = decryptWithSymmetricKey(
                        sessionKey.key,
                        sessionKey.iv,
                        newMsg.encryptedContent
                    );
                } catch (decryptError) {
                    console.error('自动解密失败:', decryptError);
                    newMsg.decryptedContent = null; // 解密失败，等待手动重试
                }
            }
        }
        this.messages.push(newMsg);
        // ... 其他逻辑 ...
    },
    // 处理接收到的密钥交换消息（来自 B 的回复或主动发起）
    async handleKeyExchangeMessage(messageData) {
        if (messageData.type === 'key_exchange') {
            try {
                const privateKey = getPrivateKey(this.currentUser.id);
                const decryptedKeyData = decryptSymmetricKeyWithPrivateKey(privateKey, messageData.encryptedEnvelope);
                // 存储发送方（B）的会话密钥，用于解密他发来的消息
                this.$store.commit('chatSession/SET_SESSION_KEY', {
                    userId: messageData.senderId,
                    keyData: decryptedKeyData
                });
                console.log(`收到来自 ${messageData.senderId} 的会话密钥并已保存`);
                // 可以发送一个确认消息
            } catch (error) {
                console.error('处理密钥交换消息失败:', error);
            }
        }
    },
    // ... };
</script>

4.3 后端 Java 实现

后端需要处理两种类型的消息：key_exchange和普通的text消息。

1. 修改 Controller

@PostMapping("/message")
public ResponseEntity<Void> sendMessage(@RequestBody MessageRequest request) {
    // ... 身份验证 ...
    ChatMessage message = new ChatMessage();
    message.setSenderId(request.getSenderId());
    message.setReceiverId(request.getReceiverId());
    message.setContent(request.getContent());
    message.setSignature(request.getSignature());
    message.setIsEncrypted(request.getIsEncrypted());
    message.setMessageType(request.getType()); // "text" or "key_exchange"
    message.setTimestamp(Instant.now());
    messageRepository.save(message);

    MessageDeliveryDto deliveryDto = new MessageDeliveryDto();
    // ... 填充字段 ...
    deliveryDto.setType(request.getType()); // 设置消息类型
    messagingTemplate.convertAndSendToUser(
        request.getReceiverId().toString(),
        "/queue/messages",
        deliveryDto
    );
    return ResponseEntity.ok().build();
}

// 统一的请求体，支持多种消息类型
public static class MessageRequest {
    private Long senderId;
    private Long receiverId;
    private String content;
    private String signature;
    private Boolean isEncrypted;
    private String type; // "text", "key_exchange", "image", etc.
    // getters and setters
}

4.4 高级特性：双工密钥协商与 Perfect Forward Secrecy

上面的实现是 A 生成密钥给 B，是单向的。更安全的做法是双方各自生成一个密钥种子，通过 Diffie-Hellman 密钥交换协议协商出一个共享的会话密钥。这可以实现完全的前向保密，即使双方的长期私钥都泄露，过去的会话也无法解密。

使用 ECDH（椭圆曲线 Diffie-Hellman）

1. A 和 B 各自生成临时的 ECC 密钥对。
2. A 将自己的临时公钥发送给 B。
3. B 将自己的临时公钥发送给 A。
4. A 用自己的临时私钥和 B 的临时公钥计算共享密钥。
5. B 用自己的临时私钥和 A 的临时公钥计算共享密钥。（根据 ECC 数学原理，两者计算出的共享密钥相同）
6. 双方用这个共享密钥派生出的对称密钥进行通信。
7. 会话结束后，双方立即销毁临时的 ECC 密钥对和会话密钥。

这样，每次会话的密钥都是独立的。实现此机制复杂度较高，通常借助libsodium.js等库。以下是概念性代码：

// 概念性代码，使用 libsodium.js
import sodium from 'libsodium-wrappers';
await sodium.ready;

// A 和 B 各自生成临时密钥对
let keyPairA = sodium.crypto_kx_keypair();
let keyPairB = sodium.crypto_kx_keypair();

// A 计算共享密钥
let sharedKeyA = sodium.crypto_kx_client_session_keys(keyPairA.publicKey, keyPairA.privateKey, keyPairB.publicKey);

// B 计算共享密钥
let sharedKeyB = sodium.crypto_kx_server_session_keys(keyPairB.publicKey, keyPairB.privateKey, keyPairA.publicKey);

// sharedKeyA.rx === sharedKeyB.tx
// sharedKeyA.tx === sharedKeyB.rx

// 双方现在有两个密钥：一个用于发送，一个用于接收

4.5 方案三优缺点总结

• 优点：
  • 高性能：消息通信使用高效的对称加密，用户体验流畅。
  • 前向保密性：会话密钥是临时的，定期更换可使过去的通信在长期私钥泄露后依然安全。
  • 高安全性：结合了非对称加密的安全密钥分发和对称加密的效率。
• 缺点：
  • 实现复杂度最高：需要管理会话状态、密钥的生命周期、密钥交换协议等。
  • 状态管理：需要在前端维护会话密钥的状态，页面刷新可能导致密钥丢失（需要重新协商）。可以考虑将会话密钥安全地存储在IndexedDB中。

结论：方案三是唯一推荐用于生产环境的方案。它提供了最佳的性能和安全特性平衡，是现代 E2EE 应用的标准做法。

第五部分：部署、测试与安全最佳实践

5.1 密钥安全存储指南

• 长期私钥：
  • 不要明文存储在localStorage中。容易被 XSS 攻击窃取。
  • 推荐方案：使用用户密码通过PBKDF2等 KDF 派生出一个密钥，用这个密钥对长期私钥进行加密后再存储。解密时要求用户输入密码。
  • 替代方案：使用浏览器的window.crypto.subtle API 生成非提取式密钥，并存储在IndexedDB中。
• 会话密钥：
  • 可以存储在 Vuex 内存中，页面刷新即丢失（需要重新协商，增强了前向保密性）。
  • 如果希望刷新后保持会话，可以加密后存入sessionStorage或IndexedDB。

5.2 传输安全

• 必须使用 HTTPS：所有 API 和 WebSocket 连接都必须通过 TLS 加密，防止中间人攻击窃取公钥或密文。
• 安全的 WebSocket：使用wss://协议。

5.3 后端安全考虑

• 身份验证：所有 API 调用都必须有严格的身份验证（如 JWT），确保用户不能冒充他人发送消息或获取他人的公钥。
• 权限检查：在转发消息前，验证发送者senderId确实属于当前登录用户。
• 速率限制：对密钥交换和消息发送接口实施速率限制，防止滥用。

5.4 测试策略

1. 单元测试：测试每个加密/解密函数，确保其正确性。
2. 集成测试：
   • 测试两个客户端能否成功完成密钥交换。
   • 测试加密消息能否被正确解密。
   • 测试签名验证能否正确识别有效和无效签名。
3. 负载测试：模拟大量用户同时发送加密消息，测试系统的性能表现。

5.5 处理常见问题

• '无法解密'错误：引导用户检查密码（如果用了加密存储）或尝试重新建立会话。
• '签名无效'警告：明确告知用户消息可能不可信，并建议通过其他渠道验证消息内容。
• 新设备登录：设计清晰的流程引导用户生成新密钥对，并告知其对旧消息的影响。

第六部分：总结与方案选择

最终强烈推荐：

对于任何严肃的、面向用户的 Web 版 IM 应用，请务必选择方案三（混合加密系统）。它虽然是实现起来最复杂的方案，但它是唯一能同时满足安全性（E2EE、前向保密）、性能和用户体验要求的方案。方案一和方案二仅适用于理解概念或极其特殊的低频场景。