[PD1] Certs done

Projs/PD1/internal/utils/cryptoUtils/cryptoUtils.go

@@ -2,31 +2,41 @@ package cryptoUtils
 
 import (
 	"PD1/internal/client"
-	"PD1/internal/protocol"
 	"crypto/rsa"
+	"crypto/tls"
 	"crypto/x509"
-	"fmt"
+	"encoding/pem"
 	"log"
 	"os"
 
 	"software.sslmate.com/src/go-pkcs12"
 )
 
-func Print() {
-	fmt.Println("crypto package")
+type KeyStore struct {
+	cert        *x509.Certificate
+	caCertChain []*x509.Certificate
+	privKey     rsa.PrivateKey
 }
 
-func getUserInfo(certFilename string) (
-	rsa.PrivateKey,
-	*x509.Certificate,
-	[]*x509.Certificate,
-	error) {
+func (k KeyStore) GetCert() *x509.Certificate {
+	return k.cert
+}
+
+func (k KeyStore) GetCACertChain() []*x509.Certificate {
+	return k.caCertChain
+}
+
+func (k KeyStore) GetPrivKey() rsa.PrivateKey {
+	return k.privKey
+}
+
+func LoadKeyStore(keyStorePath string) KeyStore {
+
 	var privKey rsa.PrivateKey
 
-	certFile, err := os.ReadFile(certFilename)
+	certFile, err := os.ReadFile(keyStorePath)
 	if err != nil {
-		log.Panicln("Provided certificate %v couldn't be opened", certFilename)
-		return rsa.PrivateKey{}, nil, nil, err
+		log.Panicln("Provided certificate %v couldn't be opened", keyStorePath)
 	}
 
 	password := client.AskUserPassword()
@@ -34,16 +44,49 @@ func getUserInfo(certFilename string) (
 	privKey = privKeyInterface.(rsa.PrivateKey)
 	if err != nil {
 		log.Panicln("PKCS12 key store couldn't be decoded")
-		return rsa.PrivateKey{}, nil, nil, err
 	}
 	if err := privKey.Validate(); err != nil {
 		log.Panicln("Private key is not valid")
-		return rsa.PrivateKey{}, nil, nil, err
 	}
-	return privKey, cert, caCerts, nil
+	return KeyStore{cert: cert, caCertChain: caCerts, privKey: privKey}
 }
 
-func encryptMessageContent(privKey rsa.PrivateKey, peerPubKey rsa.PublicKey, content []byte) []byte {
+func (k KeyStore)GetTLSConfig() *tls.Config {
+    certificate ,err := tls.X509KeyPair(k.cert.Raw, pem.EncodeToMemory(&pem.Block{Type: "RSA PRIVATE KEY", Bytes: x509.MarshalPKCS1PrivateKey(&k.privKey)}))
+    if err!=nil{
+        log.Panicln("Could not load certificate and privkey to TLS")
+    }
+    
+    //Add the CA certificate chain to a CertPool
+    caCertPool := x509.NewCertPool()
+	for _, caCert := range k.caCertChain {
+		caCertPool.AddCert(caCert)
+	}
+
+    config := &tls.Config{
+		Certificates: []tls.Certificate{certificate},
+		ClientCAs:    caCertPool,
+	}
+    return config
+}
+
+func (k KeyStore)GetTLSConfigServer() *tls.Config {
+	config := k.GetTLSConfig()
+
+	config.ClientAuth = tls.RequireAndVerifyClientCert
+
+	return config
+}
+
+func (k KeyStore)GetTLSConfigClient() *tls.Config {
+	config:= k.GetTLSConfig()
+
+	config.ServerName = "SERVER"
+
+	return config
+}
+
+func (k KeyStore)EncryptMessageContent(peerPubKey rsa.PublicKey, content []byte) []byte {
 	// Digital envolope
-
+    return nil
 }