From 56f84d850d3e9e021922a94e9db076a92f694125 Mon Sep 17 00:00:00 2001
From: CounterFire2023 <136581895+CounterFire2023@users.noreply.github.com>
Date: Wed, 23 Aug 2023 14:24:28 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3nft=E5=90=88=E7=BA=A6lock?=
 =?UTF-8?q?=E5=8A=9F=E8=83=BD=E6=97=A0=E6=95=88=E7=9A=84bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 contracts/tokens/erc721/BEBadge.sol | 1 +
 contracts/tokens/erc721/NFT.sol     | 1 +
 2 files changed, 2 insertions(+)

diff --git a/contracts/tokens/erc721/BEBadge.sol b/contracts/tokens/erc721/BEBadge.sol
index 713f969..bba0f5e 100644
--- a/contracts/tokens/erc721/BEBadge.sol
+++ b/contracts/tokens/erc721/BEBadge.sol
@@ -161,6 +161,7 @@ contract BEBadge is AccessControl, ERC721Enumerable {
     uint256 firstTokenId,
     uint256 batchSize
   ) internal virtual override {
+    require(!lockedTokens[firstTokenId], "Can not transfer locked token");
     super._beforeTokenTransfer(from, to, firstTokenId, batchSize);
   }
 
diff --git a/contracts/tokens/erc721/NFT.sol b/contracts/tokens/erc721/NFT.sol
index f5747eb..031acaa 100644
--- a/contracts/tokens/erc721/NFT.sol
+++ b/contracts/tokens/erc721/NFT.sol
@@ -126,6 +126,7 @@ contract NFT is AccessControl, ERC721Enumerable {
     uint256 firstTokenId,
     uint256 batchSize
   ) internal virtual override {
+    require(!lockedTokens[firstTokenId], "Can not transfer locked token");
     super._beforeTokenTransfer(from, to, firstTokenId, batchSize);
   }