From 87a92484f736baaec5537f2e6d118d208c13d3d1 Mon Sep 17 00:00:00 2001 From: Jianhui Zhao Date: Sat, 27 Apr 2019 14:25:39 +0800 Subject: [PATCH] Handle some error Signed-off-by: Jianhui Zhao --- command.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/command.go b/command.go index 6bf89a8..32bfcb4 100644 --- a/command.go +++ b/command.go @@ -22,6 +22,7 @@ package main import ( "fmt" jsoniter "github.com/json-iterator/go" + log "github.com/sirupsen/logrus" "io" "io/ioutil" "net/http" @@ -94,10 +95,15 @@ func serveCmd(br *Broker, w http.ResponseWriter, r *http.Request) { return } - body, _ := ioutil.ReadAll(r.Body) + body, err := ioutil.ReadAll(r.Body) + if err != nil { + log.Error(err) + cmdErrReply(RTTY_CMD_ERR_INVALID, w) + return + } cmdInfo := CommandInfo{} - err := jsoniter.Unmarshal(body, &cmdInfo) + err = jsoniter.Unmarshal(body, &cmdInfo) if err != nil || cmdInfo.Cmd == "" || cmdInfo.Devid == "" { cmdErrReply(RTTY_CMD_ERR_INVALID, w) return