blake3/gen/gen.go

package main

import (
	"fmt"
)

func permute(m *[16]uint32) {
	*m = [16]uint32{
		m[2], m[6], m[3], m[10],
		m[7], m[0], m[4], m[13],
		m[1], m[11], m[12], m[5],
		m[9], m[14], m[15], m[8],
	}
}

func main() {

	var m [16]uint32

	for i := range m {
		m[i] = uint32(i)
	}

	fmt.Printf(`	// round %d rather than init state and mix, do both.
	// Mix the columns.
	s0, s4, s8, s12 := g(n.cv[0], n.cv[4], iv[0], uint32(n.counter), n.block[%d], n.block[%d])
	s1, s5, s9, s13 := g(n.cv[1], n.cv[5], iv[1], uint32(n.counter>>32), n.block[%d], n.block[%d])
	s2, s6, s10, s14 := g(n.cv[2], n.cv[6], iv[2], n.blockLen, n.block[%d], n.block[%d])
	s3, s7, s11, s15 := g(n.cv[3], n.cv[7], iv[3], n.flags, n.block[%d], n.block[%d])

	// Mix the diagonals.
	s0, s5, s10, s15 = g(s0, s5, s10, s15, n.block[%d], n.block[%d])
	s1, s6, s11, s12 = g(s1, s6, s11, s12, n.block[%d], n.block[%d])
	s2, s7, s8, s13 = g(s2, s7, s8, s13, n.block[%d], n.block[%d])
	s3, s4, s9, s14 = g(s3, s4, s9, s14, n.block[%d], n.block[%d])

`, 1, m[0], m[1], m[2], m[3], m[4], m[5], m[6], m[7], m[8], m[9], m[10], m[11], m[12], m[13], m[14], m[15])

	permute(&m)

	for x := 2; x < 8; x++ {
		fmt.Printf(`	// round %d
	// Mix the columns.
	s0, s4, s8, s12 = g(s0, s4, s8, s12, n.block[%d], n.block[%d])
	s1, s5, s9, s13 = g(s1, s5, s9, s13, n.block[%d], n.block[%d])
	s2, s6, s10, s14 = g(s2, s6, s10, s14, n.block[%d], n.block[%d])
	s3, s7, s11, s15 = g(s3, s7, s11, s15, n.block[%d], n.block[%d])

	// Mix the diagonals.
	s0, s5, s10, s15 = g(s0, s5, s10, s15, n.block[%d], n.block[%d])
	s1, s6, s11, s12 = g(s1, s6, s11, s12, n.block[%d], n.block[%d])
	s2, s7, s8, s13 = g(s2, s7, s8, s13, n.block[%d], n.block[%d])
	s3, s4, s9, s14 = g(s3, s4, s9, s14, n.block[%d], n.block[%d])

`, x,
			m[0], m[1], m[2], m[3], m[4], m[5], m[6], m[7], m[8], m[9], m[10], m[11], m[12], m[13], m[14], m[15])
		permute(&m)
	}
}