From 4f1dacae66a091d1466a9c91852ebe109998f3f0 Mon Sep 17 00:00:00 2001 From: JeffMboya Date: Thu, 9 May 2024 16:50:30 +0300 Subject: [PATCH] Add service.go Signed-off-by: JeffMboya --- coap/api/transport.go | 173 +-------------------------------------- coap/service.go | 186 ++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 187 insertions(+), 172 deletions(-) create mode 100644 coap/service.go diff --git a/coap/api/transport.go b/coap/api/transport.go index 64d8fabb44..93549cb80b 100644 --- a/coap/api/transport.go +++ b/coap/api/transport.go @@ -4,45 +4,17 @@ package api import ( - "context" - "fmt" - "io" "log/slog" "net/http" - "net/url" - "regexp" - "strings" - "time" "github.com/absmach/magistrala" "github.com/absmach/magistrala/coap" - "github.com/absmach/magistrala/pkg/errors" - svcerr "github.com/absmach/magistrala/pkg/errors/service" - "github.com/absmach/magistrala/pkg/messaging" "github.com/go-chi/chi/v5" - "github.com/plgd-dev/go-coap/v2/message" - "github.com/plgd-dev/go-coap/v2/message/codes" "github.com/plgd-dev/go-coap/v2/mux" "github.com/prometheus/client_golang/prometheus/promhttp" ) -const ( - protocol = "coap" - authQuery = "auth" - startObserve = 0 // observe option value that indicates start of observation -) - -var channelPartRegExp = regexp.MustCompile(`^/channels/([\w\-]+)/messages(/[^?]*)?(\?.*)?$`) - -const ( - numGroups = 3 // entire expression + channel group + subtopic group - channelGroup = 2 // channel group is second in channel regexp -) - -var ( - errMalformedSubtopic = errors.New("malformed subtopic") - errBadOptions = errors.New("bad options") -) +const protocol = "coap" var ( logger *slog.Logger @@ -65,146 +37,3 @@ func MakeCoAPHandler(svc coap.Service, l *slog.Logger) mux.HandlerFunc { return handler } - -func sendResp(w mux.ResponseWriter, resp *message.Message) { - if err := w.Client().WriteMessage(resp); err != nil { - logger.Warn(fmt.Sprintf("Can't set response: %s", err)) - } -} - -func handler(w mux.ResponseWriter, m *mux.Message) { - resp := message.Message{ - Code: codes.Content, - Token: m.Token, - Context: m.Context, - Options: make(message.Options, 0, 16), - } - defer sendResp(w, &resp) - msg, err := decodeMessage(m) - if err != nil { - logger.Warn(fmt.Sprintf("Error decoding message: %s", err)) - resp.Code = codes.BadRequest - return - } - key, err := parseKey(m) - if err != nil { - logger.Warn(fmt.Sprintf("Error parsing auth: %s", err)) - resp.Code = codes.Unauthorized - return - } - switch m.Code { - case codes.GET: - err = handleGet(m.Context, m, w.Client(), msg, key) - case codes.POST: - resp.Code = codes.Created - err = service.Publish(m.Context, key, msg) - default: - err = svcerr.ErrNotFound - } - if err != nil { - switch { - case err == errBadOptions: - resp.Code = codes.BadOption - case err == svcerr.ErrNotFound: - resp.Code = codes.NotFound - case errors.Contains(err, svcerr.ErrAuthorization), - errors.Contains(err, svcerr.ErrAuthentication): - resp.Code = codes.Unauthorized - default: - resp.Code = codes.InternalServerError - } - } -} - -func handleGet(ctx context.Context, m *mux.Message, c mux.Client, msg *messaging.Message, key string) error { - var obs uint32 - obs, err := m.Options.Observe() - if err != nil { - logger.Warn(fmt.Sprintf("Error reading observe option: %s", err)) - return errBadOptions - } - if obs == startObserve { - c := coap.NewClient(c, m.Token, logger) - return service.Subscribe(ctx, key, msg.GetChannel(), msg.GetSubtopic(), c) - } - return service.Unsubscribe(ctx, key, msg.GetChannel(), msg.GetSubtopic(), m.Token.String()) -} - -func decodeMessage(msg *mux.Message) (*messaging.Message, error) { - if msg.Options == nil { - return &messaging.Message{}, errBadOptions - } - path, err := msg.Options.Path() - if err != nil { - return &messaging.Message{}, err - } - channelParts := channelPartRegExp.FindStringSubmatch(path) - if len(channelParts) < numGroups { - return &messaging.Message{}, errMalformedSubtopic - } - - st, err := parseSubtopic(channelParts[channelGroup]) - if err != nil { - return &messaging.Message{}, err - } - ret := &messaging.Message{ - Protocol: protocol, - Channel: channelParts[1], - Subtopic: st, - Payload: []byte{}, - Created: time.Now().UnixNano(), - } - - if msg.Body != nil { - buff, err := io.ReadAll(msg.Body) - if err != nil { - return ret, err - } - ret.Payload = buff - } - return ret, nil -} - -func parseKey(msg *mux.Message) (string, error) { - if obs, _ := msg.Options.Observe(); obs != 0 && msg.Code == codes.GET { - return "", nil - } - authKey, err := msg.Options.GetString(message.URIQuery) - if err != nil { - return "", err - } - vars := strings.Split(authKey, "=") - if len(vars) != 2 || vars[0] != authQuery { - return "", svcerr.ErrAuthorization - } - return vars[1], nil -} - -func parseSubtopic(subtopic string) (string, error) { - if subtopic == "" { - return subtopic, nil - } - - subtopic, err := url.QueryUnescape(subtopic) - if err != nil { - return "", errMalformedSubtopic - } - subtopic = strings.ReplaceAll(subtopic, "/", ".") - - elems := strings.Split(subtopic, ".") - filteredElems := []string{} - for _, elem := range elems { - if elem == "" { - continue - } - - if len(elem) > 1 && (strings.Contains(elem, "*") || strings.Contains(elem, ">")) { - return "", errMalformedSubtopic - } - - filteredElems = append(filteredElems, elem) - } - - subtopic = strings.Join(filteredElems, ".") - return subtopic, nil -} diff --git a/coap/service.go b/coap/service.go new file mode 100644 index 0000000000..dc3969f746 --- /dev/null +++ b/coap/service.go @@ -0,0 +1,186 @@ +// Copyright (c) Abstract Machines +// SPDX-License-Identifier: Apache-2.0 + +package coap + +import ( + "context" + "fmt" + "io" + "log/slog" + "net/url" + "regexp" + "strings" + "time" + + "github.com/absmach/magistrala/pkg/errors" + svcerr "github.com/absmach/magistrala/pkg/errors/service" + "github.com/absmach/magistrala/pkg/messaging" + "github.com/mainflux/mainflux/coap" + "github.com/plgd-dev/go-coap/v2/message" + "github.com/plgd-dev/go-coap/v2/message/codes" + mux "github.com/plgd-dev/go-coap/v2/mux" +) + +const ( + protocol = "coap" + authQuery = "auth" + startObserve = 0 // observe option value that indicates start of observation + numGroups = 3 // entire expression + channel group + subtopic group + channelGroup = 2 // channel group is second in channel regexp +) + +var ( + errMalformedSubtopic = errors.New("malformed subtopic") + errBadOptions = errors.New("bad options") + channelPartRegExp = regexp.MustCompile(`^/channels/([\w\-]+)/messages(/[^?]*)?(\?.*)?$`) +) + +var ( + logger *slog.Logger + service coap.Service +) + +func sendResp(w mux.ResponseWriter, resp *message.Message) { + if err := w.Client().WriteMessage(resp); err != nil { + logger.Warn(fmt.Sprintf("Can't set response: %s", err)) + } +} + +// MakeCoAPHandler creates handler for CoAP messages. +func handler(w mux.ResponseWriter, m *mux.Message) { + resp := message.Message{ + Code: codes.Content, + Token: m.Token, + Context: m.Context, + Options: make(message.Options, 0, 16), + } + defer sendResp(w, &resp) + msg, err := decodeMessage(m) + if err != nil { + logger.Warn(fmt.Sprintf("Error decoding message: %s", err)) + resp.Code = codes.BadRequest + return + } + key, err := parseKey(m) + if err != nil { + logger.Warn(fmt.Sprintf("Error parsing auth: %s", err)) + resp.Code = codes.Unauthorized + return + } + switch m.Code { + case codes.GET: + err = handleGet(m.Context, m, w.Client(), msg, key) + case codes.POST: + resp.Code = codes.Created + err = service.Publish(m.Context, key, msg) + default: + err = svcerr.ErrNotFound + } + if err != nil { + switch { + case err == errBadOptions: + resp.Code = codes.BadOption + case err == svcerr.ErrNotFound: + resp.Code = codes.NotFound + case errors.Contains(err, svcerr.ErrAuthorization), + errors.Contains(err, svcerr.ErrAuthentication): + resp.Code = codes.Unauthorized + default: + resp.Code = codes.InternalServerError + } + } +} + +func handleGet(ctx context.Context, m *mux.Message, c mux.Client, msg *messaging.Message, key string) error { + var obs uint32 + obs, err := m.Options.Observe() + if err != nil { + logger.Warn(fmt.Sprintf("Error reading observe option: %s", err)) + return errBadOptions + } + if obs == startObserve { + c := coap.NewClient(c, m.Token, logger) + return service.Subscribe(ctx, key, msg.GetChannel(), msg.GetSubtopic(), c) + } + return service.Unsubscribe(ctx, key, msg.GetChannel(), msg.GetSubtopic(), m.Token.String()) +} + +func decodeMessage(msg *mux.Message) (*messaging.Message, error) { + if msg.Options == nil { + return &messaging.Message{}, errBadOptions + } + path, err := msg.Options.Path() + if err != nil { + return &messaging.Message{}, err + } + channelParts := channelPartRegExp.FindStringSubmatch(path) + if len(channelParts) < numGroups { + return &messaging.Message{}, errMalformedSubtopic + } + + st, err := parseSubtopic(channelParts[channelGroup]) + if err != nil { + return &messaging.Message{}, err + } + ret := &messaging.Message{ + Protocol: protocol, + Channel: channelParts[1], + Subtopic: st, + Payload: []byte{}, + Created: time.Now().UnixNano(), + } + + if msg.Body != nil { + buff, err := io.ReadAll(msg.Body) + if err != nil { + return ret, err + } + ret.Payload = buff + } + return ret, nil +} + +func parseKey(msg *mux.Message) (string, error) { + if obs, _ := msg.Options.Observe(); obs != 0 && msg.Code == codes.GET { + return "", nil + } + authKey, err := msg.Options.GetString(message.URIQuery) + if err != nil { + return "", err + } + vars := strings.Split(authKey, "=") + if len(vars) != 2 || vars[0] != authQuery { + return "", svcerr.ErrAuthorization + } + return vars[1], nil +} + +func parseSubtopic(subtopic string) (string, error) { + if subtopic == "" { + return subtopic, nil + } + + subtopic, err := url.QueryUnescape(subtopic) + if err != nil { + return "", errMalformedSubtopic + } + subtopic = strings.ReplaceAll(subtopic, "/", ".") + + elems := strings.Split(subtopic, ".") + filteredElems := []string{} + for _, elem := range elems { + if elem == "" { + continue + } + + if len(elem) > 1 && (strings.Contains(elem, "*") || strings.Contains(elem, ">")) { + return "", errMalformedSubtopic + } + + filteredElems = append(filteredElems, elem) + } + + subtopic = strings.Join(filteredElems, ".") + return subtopic, nil +}