WIP JSONRPC2 implementation

2017-10-31 10:23:17 +01:00
parent 5a8417cf18
commit 6ea4f6e123
4 changed files with 207 additions and 62 deletions
@@ -0,0 +1,110 @@
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package service
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/EtixLabs/cameradar"
+	"github.com/EtixLabs/cameradar/server/adaptor/jsonrpc2"
+	v "gopkg.in/go-playground/validator.v9"
+)
+
+func (c *Cameradar) handleRequest(message string) {
+	var ret []cmrdr.Stream
+	var JSONRPCErr jsonrpc2.Error
+
+	var request jsonrpc2.Request
+	err := json.Unmarshal([]byte(message), &request)
+	if err != nil {
+		JSONRPCErr = jsonrpc2.Error{
+			Code:    jsonrpc2.ParseError,
+			Message: jsonrpc2.ParseErrorMessage,
+			Data:    err.Error(),
+		}
+	}
+
+	validate := v.New()
+	err = validate.Struct(request)
+	if err != nil {
+		JSONRPCErr = jsonrpc2.Error{
+			Code:    jsonrpc2.InvalidRequest,
+			Message: jsonrpc2.InvalidRequestMessage,
+			Data:    err.Error(),
+		}
+	}
+
+	var options Options
+	err = json.Unmarshal([]byte(request.Params), &options)
+	if err != nil {
+		JSONRPCErr = jsonrpc2.Error{
+			Code:    jsonrpc2.InvalidParams,
+			Message: jsonrpc2.InvalidParamsMessage,
+			Data:    err.Error(),
+		}
+	}
+
+	c.SetOptions(options)
+
+	switch request.Method {
+	case "discover":
+		ret, err = c.Discover()
+	case "attack_credentials":
+		ret, err = c.Discover()
+	case "attack_routes":
+		ret, err = c.Discover()
+	case "discover_and_attack":
+		ret, err = c.DiscoverAndAttack()
+	default:
+		JSONRPCErr = jsonrpc2.Error{
+			Code:    jsonrpc2.MethodNotFound,
+			Message: jsonrpc2.MethodNotFoundMessage,
+			Data:    err.Error(),
+		}
+	}
+	if err != nil {
+		JSONRPCErr = jsonrpc2.Error{
+			Code:    jsonrpc2.InternalError,
+			Message: jsonrpc2.InternalErrorMessage,
+			Data:    err.Error(),
+		}
+	}
+
+	result, err := json.Marshal(ret)
+	if err != nil {
+		JSONRPCErr = jsonrpc2.Error{
+			Code:    jsonrpc2.InternalError,
+			Message: jsonrpc2.InternalErrorMessage,
+			Data:    err.Error(),
+		}
+	}
+	c.respondToClient(string(result), request.ID, JSONRPCErr)
+}
+
+func (c *Cameradar) respondToClient(result, ID string, JSONRPCErr jsonrpc2.Error) {
+	println(result)
+	r := jsonrpc2.Response{
+		JSONRPC: "2.0",
+		Result:  result,
+		Error:   JSONRPCErr,
+		ID:      ID,
+	}
+
+	response, err := json.Marshal(r)
+	if err != nil {
+		c.toClient <- "{\"jsonrpc\": \"2.0\",\"result\":null,\"error\":{\"code\":" + fmt.Sprint(jsonrpc2.InternalError) + ",\"" + jsonrpc2.InternalErrorMessage + "\",\"data\":\"could not marshal response\"},\"id\":\"" + ID + "\"}"
+	}
+
+	c.toClient <- string(response)
+}