From bd815cb70c6ba0a2dbc76166f28ccee53e66e248 Mon Sep 17 00:00:00 2001
From: Mikio Hara <mikioh.mikioh@gmail.com>
Date: Fri, 24 Oct 2014 17:37:57 +0900
Subject: [PATCH] go.net/ipv4: make use of go generate to create system
 adaptation files

LGTM=iant
R=iant
CC=golang-codereviews
https://golang.org/cl/160800043
---
 ipv4/gen.go | 68 +++++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 55 insertions(+), 13 deletions(-)

diff --git a/ipv4/gen.go b/ipv4/gen.go
index f2721a57..a1d65ef9 100644
--- a/ipv4/gen.go
+++ b/ipv4/gen.go
@@ -6,8 +6,9 @@
 
 //go:generate go run gen.go
 
-// This program generates internet protocol constants and tables by
-// reading IANA protocol registries.
+// This program generates system adaptation constants and types,
+// internet protocol constants and tables by reading template files
+// and IANA protocol registries.
 package main
 
 import (
@@ -19,10 +20,55 @@ import (
 	"io/ioutil"
 	"net/http"
 	"os"
+	"os/exec"
+	"runtime"
 	"strconv"
 	"strings"
 )
 
+func main() {
+	if err := genzsys(); err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+	if err := geniana(); err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+}
+
+func genzsys() error {
+	defs := "defs_" + runtime.GOOS + ".go"
+	f, err := os.Open(defs)
+	if err != nil {
+		if os.IsNotExist(err) {
+			return nil
+		}
+		return err
+	}
+	f.Close()
+	cmd := exec.Command("go", "tool", "cgo", "-godefs", defs)
+	b, err := cmd.Output()
+	if err != nil {
+		return err
+	}
+	switch runtime.GOOS {
+	case "dragonfly", "solaris":
+		// The ipv4 pacakge still supports go1.2, and so we
+		// need to take care of additional platforms in go1.3
+		// and above for working with go1.2.
+		b = bytes.Replace(b, []byte("package ipv4\n"), []byte("// +build "+runtime.GOOS+"\n\npackage ipv4\n"), 1)
+	}
+	b, err = format.Source(b)
+	if err != nil {
+		return err
+	}
+	if err := ioutil.WriteFile("zsys_"+runtime.GOOS+".go", b, 0644); err != nil {
+		return err
+	}
+	return nil
+}
+
 var registries = []struct {
 	url   string
 	parse func(io.Writer, io.Reader) error
@@ -33,7 +79,7 @@ var registries = []struct {
 	},
 }
 
-func main() {
+func geniana() error {
 	var bb bytes.Buffer
 	fmt.Fprintf(&bb, "// go generate gen.go\n")
 	fmt.Fprintf(&bb, "// GENERATED BY THE COMMAND ABOVE; DO NOT EDIT\n\n")
@@ -41,29 +87,25 @@ func main() {
 	for _, r := range registries {
 		resp, err := http.Get(r.url)
 		if err != nil {
-			fmt.Fprintln(os.Stderr, err)
-			os.Exit(1)
+			return err
 		}
 		defer resp.Body.Close()
 		if resp.StatusCode != http.StatusOK {
-			fmt.Fprintf(os.Stderr, "got HTTP status code %v for %v\n", resp.StatusCode, r.url)
-			os.Exit(1)
+			return fmt.Errorf("got HTTP status code %v for %v\n", resp.StatusCode, r.url)
 		}
 		if err := r.parse(&bb, resp.Body); err != nil {
-			fmt.Fprintln(os.Stderr, err)
-			os.Exit(1)
+			return err
 		}
 		fmt.Fprintf(&bb, "\n")
 	}
 	b, err := format.Source(bb.Bytes())
 	if err != nil {
-		fmt.Fprintln(os.Stderr, err)
-		os.Exit(1)
+		return err
 	}
 	if err := ioutil.WriteFile("iana.go", b, 0644); err != nil {
-		fmt.Fprintln(os.Stderr, err)
-		os.Exit(1)
+		return err
 	}
+	return nil
 }
 
 func parseICMPv4Parameters(w io.Writer, r io.Reader) error {