Packing
class Packing
using Org.BouncyCastle.Utilities;
using System;
namespace Org.BouncyCastle.Pqc.Crypto.Crystals.Dilithium
{
internal class Packing
{
public static byte[] PackPublicKey(PolyVecK t1, DilithiumEngine Engine)
{
byte[] array = new byte[Engine.CryptoPublicKeyBytes - 32];
for (int i = 0; i < Engine.K; i++) {
Array.Copy(t1.Vec[i].PolyT1Pack(), 0, array, i * 320, 320);
}
return array;
}
public static PolyVecK UnpackPublicKey(PolyVecK t1, byte[] pk, DilithiumEngine Engine)
{
for (int i = 0; i < Engine.K; i++) {
t1.Vec[i].PolyT1Unpack(Arrays.CopyOfRange(pk, i * 320, 32 + (i + 1) * 320));
}
return t1;
}
public static void PackSecretKey(byte[] t0_, byte[] s1_, byte[] s2_, PolyVecK t0, PolyVecL s1, PolyVecK s2, DilithiumEngine Engine)
{
for (int i = 0; i < Engine.L; i++) {
s1.Vec[i].PolyEtaPack(s1_, i * Engine.PolyEtaPackedBytes);
}
for (int i = 0; i < Engine.K; i++) {
s2.Vec[i].PolyEtaPack(s2_, i * Engine.PolyEtaPackedBytes);
}
for (int i = 0; i < Engine.K; i++) {
t0.Vec[i].PolyT0Pack(t0_, i * 416);
}
}
public static void UnpackSecretKey(PolyVecK t0, PolyVecL s1, PolyVecK s2, byte[] t0Enc, byte[] s1Enc, byte[] s2Enc, DilithiumEngine Engine)
{
for (int i = 0; i < Engine.L; i++) {
s1.Vec[i].PolyEtaUnpack(s1Enc, i * Engine.PolyEtaPackedBytes);
}
for (int i = 0; i < Engine.K; i++) {
s2.Vec[i].PolyEtaUnpack(s2Enc, i * Engine.PolyEtaPackedBytes);
}
for (int i = 0; i < Engine.K; i++) {
t0.Vec[i].PolyT0Unpack(t0Enc, i * 416);
}
}
public static void PackSignature(byte[] sig, byte[] c, PolyVecL z, PolyVecK h, DilithiumEngine engine)
{
int num = 0;
Array.Copy(c, 0, sig, 0, engine.CTilde);
num += engine.CTilde;
for (int i = 0; i < engine.L; i++) {
z.Vec[i].PackZ(sig, num + i * engine.PolyZPackedBytes);
}
num += engine.L * engine.PolyZPackedBytes;
for (int i = 0; i < engine.Omega + engine.K; i++) {
sig[num + i] = 0;
}
int num2 = 0;
for (int i = 0; i < engine.K; i++) {
for (int j = 0; j < 256; j++) {
if (h.Vec[i].Coeffs[j] != 0)
sig[num + num2++] = (byte)j;
}
sig[num + engine.Omega + i] = (byte)num2;
}
}
public static bool UnpackSignature(PolyVecL z, PolyVecK h, byte[] sig, DilithiumEngine engine)
{
int cTilde = engine.CTilde;
for (int i = 0; i < engine.L; i++) {
z.Vec[i].UnpackZ(Arrays.CopyOfRange(sig, cTilde + i * engine.PolyZPackedBytes, cTilde + (i + 1) * engine.PolyZPackedBytes));
}
cTilde += engine.L * engine.PolyZPackedBytes;
int num = 0;
for (int i = 0; i < engine.K; i++) {
for (int j = 0; j < 256; j++) {
h.Vec[i].Coeffs[j] = 0;
}
if ((sig[cTilde + engine.Omega + i] & 255) < num || (sig[cTilde + engine.Omega + i] & 255) > engine.Omega)
return false;
for (int j = num; j < (sig[cTilde + engine.Omega + i] & 255); j++) {
if (j > num && (sig[cTilde + j] & 255) <= (sig[cTilde + j - 1] & 255))
return false;
h.Vec[i].Coeffs[sig[cTilde + j] & 255] = 1;
}
num = sig[cTilde + engine.Omega + i];
}
for (int j = num; j < engine.Omega; j++) {
if ((sig[cTilde + j] & 255) != 0)
return false;
}
return true;
}
}
}