storj/pkg/peertls/io_util.go

// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.

package peertls

import (
	"crypto/ecdsa"
	"encoding/pem"
	"io"
	"os"

	"github.com/zeebo/errs"
)

func writePem(block *pem.Block, file io.WriteCloser) error {
	if err := pem.Encode(file, block); err != nil {
		return errs.New("unable to PEM-encode/write bytes to file", err)
	}

	return nil
}

func writeCerts(certs [][]byte, path string) error {
	file, err := os.Create(path)

	if err != nil {
		return errs.New("unable to open file \"%s\" for writing", path, err)
	}

	defer file.Close()

	for _, cert := range certs {
		if err := writePem(newCertBlock(cert), file); err != nil {
			return err
		}
	}

	if err := file.Close(); err != nil {
		return errs.New("unable to close cert file \"%s\"", path, err)
	}

	return nil
}

func writeKey(key *ecdsa.PrivateKey, path string) error {
	file, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600)

	if err != nil {
		return errs.New("unable to open \"%s\" for writing", path, err)
	}

	defer file.Close()

	block, err := keyToBlock(key)
	if err != nil {
		return err
	}

	if err := writePem(block, file); err != nil {
		return err
	}

	if err := file.Close(); err != nil {
		return errs.New("unable to close key filei \"%s\"", path, err)
	}

	return nil
}