keccak/gen.go

// +build ignore

package main

import (
	"fmt"
	"os"
	"text/template"
)

var roundc [24]uint64
var rotc [5][5]int

func main() {
	// Round constants
	rc := uint8(1)
	for i := 0; i < 24; i++ {
		for j := 0; j <= 6; j++ {
			roundc[i] |= uint64(rc&1) << (1<<uint(j) - 1)
			rc = rc<<1 ^ 0x71&uint8(int8(rc)>>7)
		}
	}

	// Rotation constants
	x, y := 1, 0
	for i := 1; i < 25; i++ {
		rotc[x][y] = (i * (i + 1) / 2) % 64
		x, y = y, (2*x+3*y)%5
	}

	var ctx = struct {
		Rotc [5][5]int
	}{
		rotc,
	}
	err := tmpl.Execute(os.Stdout, &ctx)
	if err != nil {
		fmt.Println(err)
	}
}

func count(n int) []int {
	var out = make([]int, n)
	for i := 0; i < n; i++ {
		out[i] = i
	}
	return out
}

func add(a, b int) int { return a + b }
func sub(a, b int) int { return a - b }
func mul(a, b int) int { return a * b }
func mod(a, b int) int { return a % b }

func afunc(x, y int) string {
	return fmt.Sprintf("a[%d][%d]", y%5, x%5)
}

func bfunc(x, y int) string {
	return fmt.Sprintf("a%d%d", x%5, y%5)
}

var funcs = template.FuncMap{
	"count": count,
	"add":   add,
	"sub":   sub,
	"mul":   mul,
	"mod":   mod,
	"a":     afunc,
	"b":     bfunc,
}

var tmpl = template.Must(template.New("keccak").Funcs(funcs).Parse(`
// Generated from go run gen.go
// DO NOT EDIT

package keccak

// round implements one round of the keccak-f[1600] permutation.
func roundGo(a *[5][5]uint64) {
	{{ range $x := count 5 }}
		var a{{$x}}0, a{{$x}}1, a{{$x}}2, a{{$x}}3, a{{$x}}4 uint64
	{{ end }}

	// Theta
	var c0, c1, c2, c3, c4 uint64
	{{ range $y := count 5 }}
		{{ range $x := count 5 }}
			{{ if eq $y 0 }}
				c{{$x}} = {{a $x $y}}
			{{ else }}
				c{{$x}} ^= {{a $x $y}}
			{{ end }}
		{{ end }}
	{{ end }}
	var d uint64
	{{ range $x := count 5 }}
		{{ $x0 := mod (add $x 4) 5 }}
		{{ $x1 := mod (add $x 1) 5 }}
		d = c{{$x0}} ^ (c{{$x1}}<<1 | c{{$x1}}>>63)
		{{ range $y := count 5 }}
			{{b $x $y}} = {{a $x $y}} ^ d
		{{ end }}
	{{ end }}

	// Rho
	{{ range $y := count 5 }}
		{{ range $x := count 5 }}
			{{ $b := b $x $y }}
			{{ $r := index $.Rotc $x $y }}
			{{$b}} = {{$b}}<<{{$r}} | {{$b}}>>{{sub 64 $r}}
		{{ end }}
	{{ end }}

	// Pi / Chi / output
	{{ range $y := count 5 }}
		{{ range $x := count 5 }}
			{{ $x0 := add $x (mul $y 3) }}
			{{ $y0 := $x }}
			{{ $x1 := add (add $x 1) (mul $y 3) }}
			{{ $y1 := add $x 1 }}
			{{ $x2 := add (add $x 2) (mul $y 3) }}
			{{ $y2 := add $x 2 }}
			{{a $x $y}} = {{b $x0 $y0}} ^ ({{b $x2 $y2}} &^ {{b $x1 $y1}})
		{{ end }}
	{{ end }}
}
`))