From 707b18bc428cc4b10418f0fdc4255af59c2fd460 Mon Sep 17 00:00:00 2001
From: lifegpc <root@lifegpc.com>
Date: Mon, 21 Jul 2025 15:09:17 +0800
Subject: [PATCH] Add some check

---
 src/scripts/will_plus/ws2.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/scripts/will_plus/ws2.rs b/src/scripts/will_plus/ws2.rs
index f5a4f0b..e80ed63 100644
--- a/src/scripts/will_plus/ws2.rs
+++ b/src/scripts/will_plus/ws2.rs
@@ -300,6 +300,9 @@ impl Script for Ws2Script {
             file.write_all_at(str.pos, &encoded)?;
             m = mes.next();
         }
+        if m.is_some() || mes.next().is_some() {
+            return Err(anyhow::anyhow!("Too many messages provided."));
+        }
         Ok(())
     }
 }