utils.go

package main

import (
	"crypto/ecdsa"
	"crypto/elliptic"
	"crypto/rand"
	"crypto/x509"
	"encoding/pem"
	"fmt"
	"io/ioutil"
	"os"
	"path"
	"sync"
)

func GenerateKeyPair(dir string, wg *sync.WaitGroup) {
	if _, err := os.Stat(dir); os.IsNotExist(err) {
		err := os.Mkdir(dir, 0700)
		if err != nil {
			panic(err)
		}
	}

	k, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
	if err != nil {
		_, err := fmt.Fprint(os.Stderr, err)
		if err != nil {
			panic(err)
		}
		return
	}
	pub := fmt.Sprintf("0x%X", FromECDSAPub(&k.PublicKey))
	err = ioutil.WriteFile(path.Join(dir, "pub_key.pub"), []byte(pub), 0600)
	if err != nil {
		panic(err)
	}

	b, err := x509.MarshalECPrivateKey(k)
	if err != nil {
		_, err := fmt.Fprint(os.Stderr, err)
		if err != nil {
			panic(err)
		}
		return
	}
	pemBlock := &pem.Block{Type: "EC PRIVATE KEY", Bytes: b}
	data := pem.EncodeToMemory(pemBlock)
	err = ioutil.WriteFile(path.Join(dir, "priv_key.pem"), data, 0600)
	if err != nil {
		panic(err)
	}

	wg.Done()
}

func FromECDSAPub(pub *ecdsa.PublicKey) []byte {
	return elliptic.Marshal(elliptic.P256(), pub.X, pub.Y)
}

func visitF(evm bool, file *os.File, genesisFile *os.File, seen uint64, hosts arrayHosts) func(string, os.FileInfo, error) error {
	return func(p string, f os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		if !f.IsDir() && path.Base(p) == "pub_key.pub" {
			var evmAccount []byte
			pubKeyHex, e := ioutil.ReadFile(p)
			if e != nil {
				return e
			}
			if evm {
				evmAccount, e = ioutil.ReadFile(path.Join(path.Dir(p), "eth", "evm-address"))
				if e != nil {
					return e
				}
			}

			idx := uint64(len(hosts)) - seen
			seen--

			var endl string

			if seen > 0 {
				endl = ",\n"
			} else {
				endl = "\n"
			}

			if _, err = fmt.Fprintf(file, "  {\n    \"NetAddr\": \"%s\",\n    \"PubKeyHex\": \"%s\"\n  }%s",
				hosts[idx], pubKeyHex, endl); err != nil {
				panic(err)
			}
			if evm {
				evmlFile, err := os.OpenFile(path.Join(path.Dir(p), "eth", "evml.toml"), os.O_APPEND|os.O_WRONLY, 0600)
				if err != nil {
					panic(err)
				}
				defer func() {
					if err = evmlFile.Close(); err != nil {
						panic(err)
					}
				}()

				if _, err = fmt.Fprintf(evmlFile, "listen = \"%s\"", hosts[idx]); err != nil {
					panic(err)
				} else if _, err = fmt.Fprintf(genesisFile, "\t\t\"%s\": {\n", evmAccount); err != nil {
					panic(err)
				} else if _, err = fmt.Fprintf(genesisFile, "\t\t\t\"balance\": \"2019000000000000000000\"\n\t\t}%s", endl); err != nil {
					panic(err)
				}
			}
		}
		return nil
	}
}