2024-12-10 15:47:37 +01:00

66 lines
2.3 KiB
Nim

import pkg/protobuf_serialization
import ../basics
import ./transaction
export protobuf_serialization
type
TransactionMessage* {.proto3.} = object
version* {.fieldNumber: 1, pint.}: uint32
kind* {.fieldNumber: 2, pint.}: uint32
requestId* {.fieldNumber: 3.}: seq[byte]
slotIndex* {.fieldNumber: 4, pint.}: uint32
period* {.fieldNumber: 5, pint.}: uint64
merkleRoot* {.fieldNumber: 6.}: seq[byte]
challenge* {.fieldNumber: 7.}: seq[byte]
proof* {.fieldNumber: 8.}: Groth16ProofMessage
Groth16ProofMessage* {.proto3.} = object
a* {.fieldNumber: 1.}: G1PointMessage
b* {.fieldNumber: 2.}: G2PointMessage
c* {.fieldNumber: 3.}: G1PointMessage
G1PointMessage* {.proto3.} = object
x* {.fieldNumber: 1.}: seq[byte]
y* {.fieldNumber: 2.}: seq[byte]
G2PointMessage* {.proto3.} = object
x* {.fieldNumber: 1.}: Fp2ElementMessage
y* {.fieldNumber: 2.}: Fp2ElementMessage
Fp2ElementMessage* {.proto3.} = object
real* {.fieldNumber: 1.}: seq[byte]
imag* {.fieldNumber: 2.}: seq[byte]
func init*(_: type TransactionMessage, transaction: Transaction): TransactionMessage =
var message = TransactionMessage(
version: transaction.version.uint32,
kind: transaction.kind.uint32,
requestId: @(array[32, byte](transaction.requestId)),
slotIndex: transaction.slotIndex,
period: transaction.period.uint64,
merkleRoot: @(transaction.merkleRoot),
challenge: @(transaction.challenge)
)
if transaction.kind == TransactionKind.storageProof:
message.proof = Groth16ProofMessage(
a: G1PointMessage(
x: @(transaction.proof.a.x.toBytesBE()),
y: @(transaction.proof.a.y.toBytesBE())
),
b: G2PointMessage(
x: Fp2ElementMessage(
imag: @(transaction.proof.b.x.imag.toBytesBE()),
real: @(transaction.proof.b.x.real.toBytesBE())
),
y: Fp2ElementMessage(
imag: @(transaction.proof.b.y.imag.toBytesBE()),
real: @(transaction.proof.b.y.real.toBytesBE())
)
),
c: G1PointMessage(
x: @(transaction.proof.c.x.toBytesBE()),
y: @(transaction.proof.c.y.toBytesBE())
)
)
message
func toBytes*(transaction: Transaction): seq[byte] =
ProtoBuf.encode(TransactionMessage.init(transaction))