Skip to content

Commit 2e4ed64

Browse files
committed
fix(api): add input check
1 parent 6ae8610 commit 2e4ed64

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

framework/src/main/java/org/tron/core/Wallet.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -913,6 +913,9 @@ public long calcCanDelegatedEnergyMaxSize(ByteString ownerAddress) {
913913
}
914914

915915
public DelegatedResourceAccountIndex getDelegatedResourceAccountIndex(ByteString address) {
916+
if (address == null || address.size() != DecodeUtil.ADDRESS_SIZE / 2) {
917+
return DelegatedResourceAccountIndex.getDefaultInstance();
918+
}
916919
DelegatedResourceAccountIndexCapsule accountIndexCapsule =
917920
chainBaseManager.getDelegatedResourceAccountIndexStore().getIndex(address.toByteArray());
918921
if (accountIndexCapsule != null) {
@@ -923,6 +926,9 @@ public DelegatedResourceAccountIndex getDelegatedResourceAccountIndex(ByteString
923926
}
924927

925928
public DelegatedResourceAccountIndex getDelegatedResourceAccountIndexV2(ByteString address) {
929+
if (address == null || address.size() != DecodeUtil.ADDRESS_SIZE / 2) {
930+
return DelegatedResourceAccountIndex.getDefaultInstance();
931+
}
926932
DelegatedResourceAccountIndexCapsule accountIndexCapsule = chainBaseManager
927933
.getDelegatedResourceAccountIndexStore().getV2Index(address.toByteArray());
928934
if (accountIndexCapsule != null) {

0 commit comments

Comments
 (0)