summaryrefslogtreecommitdiffstats
path: root/bridge/api
diff options
context:
space:
mode:
authorWim <wim@42.be>2023-03-11 23:10:41 +0100
committerGitHub <noreply@github.com>2023-03-11 23:10:41 +0100
commitc2b8e298d8c0a15a0e6489b658efe58030a18164 (patch)
treeb990b42b8cb646706eee6a280c3ecc22a1893682 /bridge/api
parent0917e173831e20f54fe08b1bb6442a0fdeb07776 (diff)
downloadmatterbridge-msglm-c2b8e298d8c0a15a0e6489b658efe58030a18164.tar.gz
matterbridge-msglm-c2b8e298d8c0a15a0e6489b658efe58030a18164.tar.bz2
matterbridge-msglm-c2b8e298d8c0a15a0e6489b658efe58030a18164.zip
Add extra error checking for fileinfo (api) (#2015)
Diffstat (limited to 'bridge/api')
-rw-r--r--bridge/api/api.go28
1 files changed, 25 insertions, 3 deletions
diff --git a/bridge/api/api.go b/bridge/api/api.go
index c0bef0e2..cb348f16 100644
--- a/bridge/api/api.go
+++ b/bridge/api/api.go
@@ -4,6 +4,7 @@ import (
"encoding/base64"
"encoding/json"
"net/http"
+ "strings"
"sync"
"time"
@@ -139,12 +140,33 @@ func (b *API) handlePostMessage(c echo.Context) error {
message.Account = b.Account
message.ID = ""
message.Timestamp = time.Now()
+
+ var (
+ fm map[string]interface{}
+ ds string
+ ok bool
+ )
+
for i, f := range message.Extra["file"] {
fi := config.FileInfo{}
- mapstructure.Decode(f.(map[string]interface{}), &fi)
- var data []byte
+ if fm, ok = f.(map[string]interface{}); !ok {
+ return echo.NewHTTPError(http.StatusInternalServerError, "invalid format for extra")
+ }
+ err := mapstructure.Decode(fm, &fi)
+ if err != nil {
+ if !strings.Contains(err.Error(), "got string") {
+ return err
+ }
+ }
// mapstructure doesn't decode base64 into []byte, so it must be done manually for fi.Data
- data, _ = base64.StdEncoding.DecodeString(f.(map[string]interface{})["Data"].(string))
+ if ds, ok = fm["Data"].(string); !ok {
+ return echo.NewHTTPError(http.StatusInternalServerError, "invalid format for data")
+ }
+
+ data, err := base64.StdEncoding.DecodeString(ds)
+ if err != nil {
+ return err
+ }
fi.Data = &data
message.Extra["file"][i] = fi
}