mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package checksum
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"errors"
|
|
"fmt"
|
|
"hash/crc32"
|
|
|
|
enumsspb "github.com/hanzoai/tasks/api/enums/v1"
|
|
persistencespb "github.com/hanzoai/tasks/api/persistence/v1"
|
|
)
|
|
|
|
// ErrMismatch indicates a checksum verification failure due to
|
|
// a derived checksum not being equal to expected checksum
|
|
var ErrMismatch = errors.New("checksum mismatch error")
|
|
|
|
type Marshaler interface {
|
|
Marshal() ([]byte, error)
|
|
}
|
|
|
|
// GenerateCRC32 generates an IEEE crc32 checksum on the
|
|
// serilized byte array of the given thrift object. The
|
|
// serialization proto used will be of type thriftRW
|
|
func GenerateCRC32(
|
|
payload Marshaler,
|
|
payloadVersion int32,
|
|
) (*persistencespb.Checksum, error) {
|
|
|
|
payloadBytes, err := payload.Marshal()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
crc := crc32.ChecksumIEEE(payloadBytes)
|
|
checksum := make([]byte, 4)
|
|
binary.BigEndian.PutUint32(checksum, crc)
|
|
return &persistencespb.Checksum{
|
|
Value: checksum,
|
|
Version: payloadVersion,
|
|
Flavor: enumsspb.CHECKSUM_FLAVOR_IEEE_CRC32_OVER_PROTO3_BINARY,
|
|
}, nil
|
|
}
|
|
|
|
// Verify verifies that the checksum generated from the
|
|
// given thrift object matches the specified expected checksum
|
|
// Return ErrMismatch when checksums mismatch
|
|
func Verify(
|
|
payload Marshaler,
|
|
checksum *persistencespb.Checksum,
|
|
) error {
|
|
|
|
if checksum.Flavor != enumsspb.CHECKSUM_FLAVOR_IEEE_CRC32_OVER_PROTO3_BINARY {
|
|
return fmt.Errorf("unknown checksum flavor %v", checksum.Flavor)
|
|
}
|
|
|
|
expected, err := GenerateCRC32(payload, checksum.Version)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !bytes.Equal(expected.GetValue(), checksum.GetValue()) {
|
|
return ErrMismatch
|
|
}
|
|
|
|
return nil
|
|
}
|