feat: adds certificates

open-amt-cloud-toolkit · May 14, 2024 · 8664c16 · 8664c16
1 parent f44527f
commit 8664c16
Show file tree

Hide file tree

Showing 5 changed files with 148 additions and 2 deletions.
diff --git a/internal/controller/http/v1/devicemanagement.go b/internal/controller/http/v1/devicemanagement.go
@@ -45,6 +45,7 @@ func newAmtRoutes(handler *gin.RouterGroup, d devices.Feature, l logger.Interfac
 		h.POST("userConsentCode/:guid", r.sendConsentCode)
 
 		h.GET("networkSettings/:guid", r.getNetworkSettings)
+		h.GET("certificates/:guid", r.getCertificates)
 	}
 }
 
@@ -350,3 +351,17 @@ func (r *deviceManagementRoutes) getNetworkSettings(c *gin.Context) {
 
 	c.JSON(http.StatusOK, network)
 }
+
+func (r *deviceManagementRoutes) getCertificates(c *gin.Context) {
+	guid := c.Param("guid")
+
+	network, err := r.d.GetCertificates(c, guid)
+	if err != nil {
+		r.l.Error(err, "http - v1 - getCertificates")
+		errorResponse(c, http.StatusInternalServerError, "problems")
+
+		return
+	}
+
+	c.JSON(http.StatusOK, network)
+}
diff --git a/internal/usecase/devices/certificates.go b/internal/usecase/devices/certificates.go
@@ -0,0 +1,23 @@
+package devices
+
+import (
+	"context"
+
+	"github.com/open-amt-cloud-toolkit/console/internal/usecase/utils"
+)
+
+func (uc *UseCase) GetCertificates(c context.Context, guid string) (interface{}, error) {
+	item, err := uc.repo.GetByID(c, guid, "")
+	if err != nil || item.GUID == "" {
+		return nil, utils.ErrNotFound
+	}
+
+	uc.device.SetupWsmanClient(item, false, true)
+
+	response, err := uc.device.GetCertificates()
+	if err != nil {
+		return nil, err
+	}
+
+	return response, nil
+}
diff --git a/internal/usecase/devices/interfaces.go b/internal/usecase/devices/interfaces.go
@@ -47,6 +47,7 @@ type (
 		GetAuditLog(startIndex int) (auditlog.Response, error)
 		GetEventLog() (messagelog.GetRecordsResponse, error)
 		GetNetworkSettings() (interface{}, error)
+		GetCertificates() (interface{}, error)
 	}
 	Redirection interface {
 		SetupWsmanClient(device entity.Device, isRedirection, logAMTMessages bool) wsman.Messages
@@ -95,5 +96,6 @@ type (
 		GetEventLog(ctx context.Context, guid string) (messagelog.GetRecordsResponse, error)
 		Redirect(ctx context.Context, conn *websocket.Conn, guid, mode string) error
 		GetNetworkSettings(c context.Context, guid string) (interface{}, error)
+		GetCertificates(c context.Context, guid string) (interface{}, error)
 	}
 )
diff --git a/internal/usecase/devices/mocks_test.go b/internal/usecase/devices/mocks_test.go
diff --git a/internal/usecase/devices/wsman/message.go b/internal/usecase/devices/wsman/message.go
@@ -3,6 +3,7 @@ package wsman
 import (
 	"strings"
 	"time"
+	"fmt"
 
 	"github.com/open-amt-cloud-toolkit/go-wsman-messages/v2/pkg/wsman"
 	amtAlarmClock "github.com/open-amt-cloud-toolkit/go-wsman-messages/v2/pkg/wsman/amt/alarmclock"
@@ -41,6 +42,7 @@ import (
 	ipsAlarmClock "github.com/open-amt-cloud-toolkit/go-wsman-messages/v2/pkg/wsman/ips/alarmclock"
 	ipsIEEE8021x "github.com/open-amt-cloud-toolkit/go-wsman-messages/v2/pkg/wsman/ips/ieee8021x"
 	"github.com/open-amt-cloud-toolkit/go-wsman-messages/v2/pkg/wsman/ips/optin"
+	// "github.com/rs/zerolog/log"
 
 	"github.com/open-amt-cloud-toolkit/console/internal/entity"
 	"github.com/open-amt-cloud-toolkit/console/internal/entity/dto"
@@ -196,9 +198,9 @@ func (g *GoWSMANMessages) SetFeatures(features dto.Features) (dto.Features, erro
 }
 
 func configureKVM(features dto.Features, listenerEnabled int, g *GoWSMANMessages) (int, error) {
-	kvmRequestedState := kvm.RedirectionSAP_Disable
+	kvmRequestedState := kvm.RedirectionSAPDisable
 	if features.EnableKVM {
-		kvmRequestedState = kvm.RedirectionSAP_Enable
+		kvmRequestedState = kvm.RedirectionSAPEnable
 		listenerEnabled = 1
 	}
 
@@ -902,3 +904,77 @@ func (g *GoWSMANMessages) GetNetworkSettings() (interface{}, error) {
 
 	return networkResults, nil
 }
+
+func (g *GoWSMANMessages) GetCertificates() (interface{}, error) {
+	// certificates := map[string]interface{}{}
+	credCntxtTlsEnumResp, err := g.wsmanMessages.AMT.TLSCredentialContext.Enumerate()
+	if err != nil {
+		return nil, err
+	}
+
+	credCntxtTlsResponse, err := g.wsmanMessages.AMT.TLSCredentialContext.Pull(credCntxtTlsEnumResp.Body.EnumerateResponse.EnumerationContext)
+	if err != nil {
+		return nil, err
+	}
+	fmt.Println(credCntxtTlsResponse)
+
+	concreteDepEnumResp, err := g.wsmanMessages.CIM.ConcreteDependency.Enumerate()
+	if err != nil {
+		return nil, err
+	}
+
+	concreteDepResponse, err := g.wsmanMessages.CIM.ConcreteDependency.Pull(concreteDepEnumResp.Body.EnumerateResponse.EnumerationContext)
+	if err != nil {
+		return nil, err
+	}
+	fmt.Println(concreteDepResponse)
+
+
+	pubKeyCertEnumResp, err := g.wsmanMessages.AMT.PublicKeyCertificate.Enumerate()
+	if err != nil {
+		return nil, err
+	}
+
+	pubKeyCertResponse, err := g.wsmanMessages.AMT.PublicKeyCertificate.Pull(pubKeyCertEnumResp.Body.EnumerateResponse.EnumerationContext)
+	if err != nil {
+		return nil, err
+	}
+	fmt.Println(pubKeyCertResponse)
+
+	pubPrivKeyPairEnumResp, err := g.wsmanMessages.AMT.PublicPrivateKeyPair.Enumerate()
+	if err != nil {
+		return nil, err
+	}
+
+	pubPrivKeyPairResponse, err := g.wsmanMessages.AMT.PublicPrivateKeyPair.Pull(pubPrivKeyPairEnumResp.Body.EnumerateResponse.EnumerationContext)
+	if err != nil {
+		return nil, err
+	}
+	fmt.Println(pubPrivKeyPairResponse)
+
+
+	credCntxt8021xEnumResp, err := g.wsmanMessages.IPS.IEEE8021xCredentialContext.Enumerate()
+	if err != nil {
+		return nil, err
+	}
+
+	credCntxt8021xResponse, err := g.wsmanMessages.IPS.IEEE8021xCredentialContext.Pull(credCntxt8021xEnumResp.Body.EnumerateResponse.EnumerationContext)
+	if err != nil {
+		return nil, err
+	}
+	fmt.Println(credCntxt8021xResponse)
+
+
+	cimCredCntxtEnumResp, err := g.wsmanMessages.CIM.CredentialContext.Enumerate()
+	if err != nil {
+		return nil, err
+	}
+
+	cimCredCntxtResponse, err := g.wsmanMessages.CIM.CredentialContext.Pull(cimCredCntxtEnumResp.Body.EnumerateResponse.EnumerationContext)
+	if err != nil {
+		return nil, err
+	}
+	fmt.Println(cimCredCntxtResponse)
+
+	return credCntxtTlsResponse, nil
+}