From 381d2254ebbc69a92a73ff55285a0bed1aa18c3d Mon Sep 17 00:00:00 2001
From: samkim-crypto <skim13@cs.stanford.edu>
Date: Tue, 24 Oct 2023 11:11:28 -0700
Subject: [PATCH] [zk-token-sdk] Check that discrete log compression batch size
 is greater than 0 (#33699)

check that discrete log compression batch size is greater than 0
---
 zk-token-sdk/src/encryption/discrete_log.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/zk-token-sdk/src/encryption/discrete_log.rs b/zk-token-sdk/src/encryption/discrete_log.rs
index 7f9891882..b3e02a746 100644
--- a/zk-token-sdk/src/encryption/discrete_log.rs
+++ b/zk-token-sdk/src/encryption/discrete_log.rs
@@ -130,7 +130,7 @@ impl DiscreteLog {
         &mut self,
         compression_batch_size: usize,
     ) -> Result<(), DiscreteLogError> {
-        if compression_batch_size >= TWO16 as usize {
+        if compression_batch_size >= TWO16 as usize || compression_batch_size == 0 {
             return Err(DiscreteLogError::DiscreteLogBatchSize);
         }
         self.compression_batch_size = compression_batch_size;