fix pprf

osu-crypto · Nov 1, 2023 · cfa4726 · cfa4726
1 parent 95f0d86
commit cfa4726
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/libOTe/Tools/Subfield/Subfield.h b/libOTe/Tools/Subfield/Subfield.h
@@ -147,6 +147,7 @@ struct TypeTraitVec {
   static constexpr size_t bitsF = sizeof(F) * 8;
   static constexpr size_t bytesF = sizeof(F);
   static constexpr size_t sizeBlocks = (bytesF + sizeof(block) - 1) / sizeof(block);
+  static constexpr size_t size = N;
   union Buf {
     F f;
     block b[sizeBlocks];

diff --git a/libOTe/Tools/Subfield/SubfieldPprf.h b/libOTe/Tools/Subfield/SubfieldPprf.h
@@ -508,6 +508,10 @@ namespace osuCrypto::Subfield
                                     sum[6] = sum[6] ^ child[6];
                                     sum[7] = sum[7] ^ child[7];
                                 } else {
+                                    if (getLastLevel(pprf.mDepth, treeIdx).size() <= childIdx) {
+                                        childIdx = width;
+                                        break;
+                                    }
                                     auto& realChild = getLastLevel(pprf.mDepth, treeIdx)[childIdx];
                                     auto& lastSum = lastSums[keep];
                                     realChild[0] = TypeTrait::fromBlock(child[0]);
@@ -1246,6 +1250,10 @@ namespace osuCrypto::Subfield
                                 sum[6] = sum[6] ^ child[6];
                                 sum[7] = sum[7] ^ child[7];
                             } else {
+                                if (getLastLevel(pprf.mDepth, treeIdx).size() <= childIdx) {
+                                    childIdx = width;
+                                    break;
+                                }
                                 auto& realChild = getLastLevel(pprf.mDepth, treeIdx)[childIdx];
                                 auto& lastSum = lastSums[keep];
                                 realChild[0] = TypeTrait::fromBlock(child[0]);