Refactor AUSF app, sbi, processor, consumer

cmd/main.go

@@ -9,9 +9,12 @@
 package main
 
 import (
+	"context"
 	"os"
+	"os/signal"
 	"path/filepath"
 	"runtime/debug"
+	"syscall"
 
 	"github.com/urfave/cli"
 
@@ -60,19 +63,31 @@ func action(cliCtx *cli.Context) error {
 
 	logger.MainLog.Infoln("AUSF version: ", version.GetVersion())
 
+	ctx, cancel := context.WithCancel(context.Background())
+	sigCh := make(chan os.Signal, 1)
+	signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
+
+	go func() {
+		<-sigCh  // Wait for interrupt signal to gracefully shutdown
+		cancel() // Notify each goroutine and wait them stopped
+	}()
+
 	cfg, err := factory.ReadConfig(cliCtx.String("config"))
 	if err != nil {
+		sigCh <- nil
 		return err
 	}
 	factory.AusfConfig = cfg
 
-	ausf, err := service.NewApp(cfg)
+	ausf, err := service.NewApp(ctx, cfg, tlsKeyLogPath)
 	if err != nil {
+		sigCh <- nil
 		return err
 	}
 	AUSF = ausf
 
-	ausf.Start(tlsKeyLogPath)
+	ausf.Start()
+	AUSF.WaitRoutineStopped()
 
 	return nil
 }

internal/logger/logger.go

@@ -13,6 +13,7 @@ var (
 	InitLog      *logrus.Entry
 	CfgLog       *logrus.Entry
 	CtxLog       *logrus.Entry
+	SBILog       *logrus.Entry
 	GinLog       *logrus.Entry
 	ConsumerLog  *logrus.Entry
 	UeAuthLog    *logrus.Entry
@@ -33,6 +34,7 @@ func init() {
 	InitLog = NfLog.WithField(logger_util.FieldCategory, "Init")
 	CfgLog = NfLog.WithField(logger_util.FieldCategory, "CFG")
 	CtxLog = NfLog.WithField(logger_util.FieldCategory, "CTX")
+	SBILog = NfLog.WithField(logger_util.FieldCategory, "SBI")
 	GinLog = NfLog.WithField(logger_util.FieldCategory, "GIN")
 	ConsumerLog = NfLog.WithField(logger_util.FieldCategory, "Consumer")
 	UeAuthLog = NfLog.WithField(logger_util.FieldCategory, "UeAuth")

internal/sbi/api_ueauthentication.go

@@ -0,0 +1,153 @@
+/*
+ * Nausf_UeAuthentication
+ *
+ * UeAuthentication Service
+ * © 2021, 3GPP Organizational Partners (ARIB, ATIS, CCSA, ETSI, TSDSI, TTA, TTC).
+ * All rights reserved.
+ *
+ * API version: 3.0.3
+ * Generated by: OpenAPI Generator (https://openapi-generator.tech)
+ */
+
+package sbi
+
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+
+	"github.com/free5gc/ausf/internal/logger"
+	"github.com/free5gc/openapi"
+	"github.com/free5gc/openapi/models"
+)
+
+// Index is the index handler.
+func Index(c *gin.Context) {
+	c.String(http.StatusOK, "Hello World!")
+}
+
+func (s *Server) getUeAuthenticationRoutes() []Route {
+	return []Route{
+		{
+			Method:  http.MethodGet,
+			Pattern: "/",
+			APIFunc: Index,
+		},
+		{
+			Method:  http.MethodPost,
+			Pattern: "/ue-authentications/:authCtxId/eap-session",
+			APIFunc: s.EapAuthMethodPost,
+		},
+		{
+			Method:  http.MethodPost,
+			Pattern: "/ue-authentications",
+			APIFunc: s.UeAuthenticationsPost,
+		},
+		{
+			Method:  http.MethodPut,
+			Pattern: "/ue-authentications/:authCtxId/5g-aka-confirmation",
+			APIFunc: s.UeAuthenticationsAuthCtxID5gAkaConfirmationPut,
+		},
+	}
+}
+
+// EapAuthMethodPost -
+func (s *Server) EapAuthMethodPost(c *gin.Context) {
+	var eapSessionReq models.EapSession
+
+	requestBody, err := c.GetRawData()
+	if err != nil {
+		problemDetail := models.ProblemDetails{
+			Title:  "System failure",
+			Status: http.StatusInternalServerError,
+			Detail: err.Error(),
+			Cause:  "SYSTEM_FAILURE",
+		}
+		logger.Auth5gAkaLog.Errorf("Get Request Body error: %+v", err)
+		c.JSON(http.StatusInternalServerError, problemDetail)
+		return
+	}
+
+	err = openapi.Deserialize(&eapSessionReq, requestBody, "application/json")
+	if err != nil {
+		problemDetail := "[Request Body] " + err.Error()
+		rsp := models.ProblemDetails{
+			Title:  "Malformed request syntax",
+			Status: http.StatusBadRequest,
+			Detail: problemDetail,
+		}
+		logger.Auth5gAkaLog.Errorln(problemDetail)
+		c.JSON(http.StatusBadRequest, rsp)
+		return
+	}
+	eapSessionId := c.Param("authCtxId")
+
+	s.Processor().HandleEapAuthComfirmRequest(c, eapSessionReq, eapSessionId)
+}
+
+// UeAuthenticationsPost
+func (s *Server) UeAuthenticationsPost(c *gin.Context) {
+	var authInfo models.AuthenticationInfo
+
+	requestBody, err := c.GetRawData()
+	if err != nil {
+		problemDetail := models.ProblemDetails{
+			Title:  "System failure",
+			Status: http.StatusInternalServerError,
+			Detail: err.Error(),
+			Cause:  "SYSTEM_FAILURE",
+		}
+		logger.UeAuthLog.Errorf("Get Request Body error: %+v", err)
+		c.JSON(http.StatusInternalServerError, problemDetail)
+		return
+	}
+
+	err = openapi.Deserialize(&authInfo, requestBody, "application/json")
+	if err != nil {
+		problemDetail := "[Request Body] " + err.Error()
+		rsp := models.ProblemDetails{
+			Title:  "Malformed request syntax",
+			Status: http.StatusBadRequest,
+			Detail: problemDetail,
+		}
+		logger.UeAuthLog.Errorln(problemDetail)
+		c.JSON(http.StatusBadRequest, rsp)
+		return
+	}
+
+	s.Processor().HandleUeAuthPostRequest(c, authInfo)
+}
+
+// UeAuthenticationsAuthCtxID5gAkaConfirmationPut
+func (s *Server) UeAuthenticationsAuthCtxID5gAkaConfirmationPut(c *gin.Context) {
+	var confirmationData models.ConfirmationData
+
+	requestBody, err := c.GetRawData()
+	if err != nil {
+		problemDetail := models.ProblemDetails{
+			Title:  "System failure",
+			Status: http.StatusInternalServerError,
+			Detail: err.Error(),
+			Cause:  "SYSTEM_FAILURE",
+		}
+		logger.Auth5gAkaLog.Errorf("Get Request Body error: %+v", err)
+		c.JSON(http.StatusInternalServerError, problemDetail)
+		return
+	}
+
+	err = openapi.Deserialize(&confirmationData, requestBody, "application/json")
+	if err != nil {
+		problemDetail := "[Request Body] " + err.Error()
+		rsp := models.ProblemDetails{
+			Title:  "Malformed request syntax",
+			Status: http.StatusBadRequest,
+			Detail: problemDetail,
+		}
+		logger.Auth5gAkaLog.Errorln(problemDetail)
+		c.JSON(http.StatusBadRequest, rsp)
+		return
+	}
+	confirmationDataResponseId := c.Param("authCtxId")
+
+	s.Processor().HandleAuth5gAkaComfirmRequest(c, confirmationData, confirmationDataResponseId)
+}

internal/sbi/consumer/consumer.go

@@ -0,0 +1,36 @@
+package consumer
+
+import (
+	"github.com/free5gc/ausf/pkg/app"
+	"github.com/free5gc/openapi/Nnrf_NFDiscovery"
+	"github.com/free5gc/openapi/Nnrf_NFManagement"
+)
+
+var consumer *Consumer
+
+type ConsumerAusf interface {
+	app.App
+}
+
+type Consumer struct {
+	ConsumerAusf
+
+	*nnrfService
+}
+
+func GetConsumer() *Consumer {
+	return consumer
+}
+
+func NewConsumer(ausf ConsumerAusf) (*Consumer, error) {
+	c := &Consumer{
+		ConsumerAusf: ausf,
+	}
+
+	c.nnrfService = &nnrfService{
+		consumer:        c,
+		nfMngmntClients: make(map[string]*Nnrf_NFManagement.APIClient),
+		nfDiscClients:   make(map[string]*Nnrf_NFDiscovery.APIClient),
+	}
+	return c, nil
+}