mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-09 21:36:09 +08:00
changed encoding of structs
This commit is contained in:
parent
7524f08146
commit
1cf9ac77e9
3 changed files with 63 additions and 8 deletions
|
@ -176,10 +176,30 @@ func register(w http.ResponseWriter, r *http.Request) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
caBytes, err := config.ConvertCertToBytes(*ca)
|
||||||
|
if err != nil {
|
||||||
|
logger.Log(0, "failed to encode CA cert ", err.Error())
|
||||||
|
errorResponse := models.ErrorResponse{
|
||||||
|
Code: http.StatusInternalServerError, Message: err.Error(),
|
||||||
|
}
|
||||||
|
returnErrorResponse(w, r, errorResponse)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
certBytes, err := config.ConvertCertToBytes(*cert)
|
||||||
|
if err != nil {
|
||||||
|
logger.Log(0, "failed to encode CA cert ", err.Error())
|
||||||
|
errorResponse := models.ErrorResponse{
|
||||||
|
Code: http.StatusInternalServerError, Message: err.Error(),
|
||||||
|
}
|
||||||
|
returnErrorResponse(w, r, errorResponse)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
response := config.RegisterResponse{
|
response := config.RegisterResponse{
|
||||||
Key: *key,
|
Key: *key,
|
||||||
CA: *ca,
|
CABytes: caBytes,
|
||||||
Cert: *cert,
|
CertBytes: certBytes,
|
||||||
}
|
}
|
||||||
w.WriteHeader(http.StatusOK)
|
w.WriteHeader(http.StatusOK)
|
||||||
json.NewEncoder(w).Encode(response)
|
json.NewEncoder(w).Encode(response)
|
||||||
|
|
|
@ -2,9 +2,11 @@ package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
//"github.com/davecgh/go-spew/spew"
|
//"github.com/davecgh/go-spew/spew"
|
||||||
|
"bytes"
|
||||||
"crypto/ed25519"
|
"crypto/ed25519"
|
||||||
"crypto/x509"
|
"crypto/x509"
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
|
"encoding/gob"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -46,10 +48,33 @@ type RegisterRequest struct {
|
||||||
Key ed25519.PublicKey
|
Key ed25519.PublicKey
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ConvertStructToByte - util to convert bytes to a key to use elsewhere
|
||||||
|
func ConvertCertToBytes(cert x509.Certificate) ([]byte, error) {
|
||||||
|
var buffer bytes.Buffer
|
||||||
|
var enc = gob.NewEncoder(&buffer)
|
||||||
|
if err := enc.Encode(cert); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return buffer.Bytes(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ConvertStructToByte - util to convert bytes to a key to use elsewhere
|
||||||
|
func ConvertBytesToCert(data []byte) (x509.Certificate, error) {
|
||||||
|
var buffer = bytes.NewBuffer(data)
|
||||||
|
var dec = gob.NewDecoder(buffer)
|
||||||
|
var result = new(x509.Certificate)
|
||||||
|
var err = dec.Decode(result)
|
||||||
|
if err != nil {
|
||||||
|
return *result, err
|
||||||
|
}
|
||||||
|
return *result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RegisterResponse - the response to register function
|
||||||
type RegisterResponse struct {
|
type RegisterResponse struct {
|
||||||
Key ed25519.PrivateKey
|
Key ed25519.PrivateKey
|
||||||
CA x509.Certificate
|
CABytes []byte
|
||||||
Cert x509.Certificate
|
CertBytes []byte
|
||||||
}
|
}
|
||||||
|
|
||||||
// Write - writes the config of a client to disk
|
// Write - writes the config of a client to disk
|
||||||
|
|
|
@ -16,6 +16,7 @@ import (
|
||||||
"github.com/gravitl/netmaker/tls"
|
"github.com/gravitl/netmaker/tls"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Register - the function responsible for registering with the server and acquiring certs
|
||||||
func Register(cfg *config.ClientConfig) error {
|
func Register(cfg *config.ClientConfig) error {
|
||||||
if cfg.Server.Server == "" {
|
if cfg.Server.Server == "" {
|
||||||
return errors.New("no server provided")
|
return errors.New("no server provided")
|
||||||
|
@ -62,10 +63,19 @@ func Register(cfg *config.ClientConfig) error {
|
||||||
if err := json.NewDecoder(response.Body).Decode(&resp); err != nil {
|
if err := json.NewDecoder(response.Body).Decode(&resp); err != nil {
|
||||||
return errors.New("unmarshal cert error " + err.Error())
|
return errors.New("unmarshal cert error " + err.Error())
|
||||||
}
|
}
|
||||||
if err := tls.SaveCert(ncutils.GetNetclientPath()+cfg.Server.Server+"/", "root.pem", &resp.CA); err != nil {
|
responseCA, err := config.ConvertBytesToCert(resp.CABytes)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("could not acquire CA from response " + err.Error())
|
||||||
|
}
|
||||||
|
responseCert, err := config.ConvertBytesToCert(resp.CertBytes)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("could not acquire client certificate from response " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := tls.SaveCert(ncutils.GetNetclientPath()+cfg.Server.Server+"/", "root.pem", &responseCA); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if err := tls.SaveCert(ncutils.GetNetclientPath()+cfg.Server.Server+"/", "client.pem", &resp.Cert); err != nil {
|
if err := tls.SaveCert(ncutils.GetNetclientPath()+cfg.Server.Server+"/", "client.pem", &responseCert); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if err := tls.SaveKey(ncutils.GetNetclientPath(), "client.key", private); err != nil {
|
if err := tls.SaveKey(ncutils.GetNetclientPath(), "client.key", private); err != nil {
|
||||||
|
|
Loading…
Add table
Reference in a new issue