Formatting auto-generated code.

dh/sidh/internal/templates/gen.go

@@ -8,7 +8,9 @@
 package main
 
 import (
+	"bytes"
 	"fmt"
+	"go/format"
 	"os"
 	"strings"
 	"text/template"
@@ -47,22 +49,36 @@ var opt_arm = map[string]bool{
 // Generates an 'fileNameBase.go' from 'fileNameBase.gotemp' file
 // for a given finite 'field'. Maps placeholders to 'values'.
 func gen(field, fileNameBase string, values interface{}) {
+	// Template files are located in ../templates and have
+	// extension .gotemp
+	templateFile := "../templates/" + fileNameBase + ".gotemp"
+	t, err := template.ParseFiles(templateFile)
+	if err != nil {
+		panic(fmt.Sprintf("Cannot open template file %s", templateFile))
+	}
+	var buf bytes.Buffer
+	err = t.Execute(&buf, values)
+	if err != nil {
+		panic("bad template execution")
+	}
+
+	// Formating output code
+	code, err := format.Source(buf.Bytes())
+	if err != nil {
+		panic("error formating code")
+	}
+
 	// name of the output .go file
 	outFileName := fileNameBase + ".go"
 	out, err := os.Create(outFileName)
 	if err != nil {
 		panic("Cannot open file")
 	}
-
-	// Template files are located in ../templates and have
-	// extension .gotemp
-	templateFile := "../templates/" + fileNameBase + ".gotemp"
-	t, err := template.ParseFiles(templateFile)
+	_, err = out.Write(code)
 	if err != nil {
-		panic(fmt.Sprintf("Cannot open template file %s", templateFile))
+		panic("error writing code")
 	}
 
-	t.Execute(out, values)
 	err = out.Close()
 	if err != nil {
 		panic("Cant close generated file")

kem/kyber/gen.go

@@ -7,6 +7,7 @@ package main
 
 import (
 	"bytes"
+	"go/format"
 	"io/ioutil"
 	"strings"
 	"text/template"
@@ -47,7 +48,13 @@ func generatePackageFiles() {
 			panic(err)
 		}
 
-		res := string(buf.Bytes())
+		// Formating output code
+		code, err := format.Source(buf.Bytes())
+		if err != nil {
+			panic("error formating code")
+		}
+
+		res := string(code)
 		offset := strings.Index(res, TemplateWarning)
 		if offset == -1 {
 			panic("Missing template warning in pkg.templ.go")

kem/sike/gen.go

@@ -8,6 +8,7 @@ package main
 import (
 	"bytes"
 	"fmt"
+	"go/format"
 	"io/ioutil"
 	"strings"
 	"text/template"
@@ -56,7 +57,13 @@ func generatePackageFiles() {
 			panic(err)
 		}
 
-		res := string(buf.Bytes())
+		// Formating output code
+		code, err := format.Source(buf.Bytes())
+		if err != nil {
+			panic("error formating code")
+		}
+
+		res := string(code)
 		offset := strings.Index(res, TemplateWarning)
 		if offset == -1 {
 			panic("Missing template warning in pkg.templ.go")

pke/kyber/gen.go

@@ -8,6 +8,7 @@ package main
 import (
 	"bytes"
 	"fmt"
+	"go/format"
 	"io/ioutil"
 	"os"
 	"path"
@@ -82,7 +83,13 @@ func generateParamsFiles() {
 			panic(err)
 		}
 
-		res := string(buf.Bytes())
+		// Formating output code
+		code, err := format.Source(buf.Bytes())
+		if err != nil {
+			panic("error formating code")
+		}
+
+		res := string(code)
 		offset := strings.Index(res, TemplateWarning)
 		if offset == -1 {
 			panic("Missing template warning in params.templ.go")

sign/dilithium/gen.go

@@ -8,6 +8,7 @@ package main
 import (
 	"bytes"
 	"fmt"
+	"go/format"
 	"io/ioutil"
 	"os"
 	"path"
@@ -142,7 +143,13 @@ func generateParamsFiles() {
 			panic(err)
 		}
 
-		res := string(buf.Bytes())
+		// Formating output code
+		code, err := format.Source(buf.Bytes())
+		if err != nil {
+			panic("error formating code")
+		}
+
+		res := string(code)
 		offset := strings.Index(res, TemplateWarning)
 		if offset == -1 {
 			panic("Missing template warning in params.templ.go")