support execute a command remote

Signed-off-by: Jianhui Zhao <jianhuizhao329@gmail.com>
2026-02-27 09:53:24 +08:00 · 2018-03-13 22:25:31 +08:00
parent 954b0ff2c1
commit f10a0c74e0
4 changed files with 133 additions and 18 deletions
@@ -0,0 +1,102 @@
+/*
+ * Copyright (C) 2017 Jianhui Zhao <jianhuizhao329@gmail.com>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301
+ * USA
+ */
+
+package main
+
+import (
+    "net/http"
+    "io/ioutil"
+    "encoding/json"
+    "github.com/gorilla/websocket"
+)
+
+const (
+    COMMAND_OK = iota
+    COMMAND_ERR_TIMEOUT
+    COMMAND_ERR_NOTFOUND
+    COMMAND_ERR_READ
+    COMMAND_ERR_LOGIN
+    COMMAND_ERR_SYS
+    COMMAND_PARAMETER_ERR
+    COMMAND_DEV_OFFLINE
+)
+
+type CommandReq struct {
+    ID uint32 `json:"id"`
+    Username string `json:"username"`
+    Password string `json:"password"`
+    Devid string `json:"devid"`
+    Cmd string `json:"cmd"`
+    Params []string `json:"params"`
+    Env []string `json:"env"`
+}
+
+type CommandResult struct {
+    ID uint32 `json:"id,omitempty"`
+    Err int `json:err`
+    Msg string `json:"msg"`
+    Code int `json:"code"`
+    Stdout string `json:"stdout"`
+    Stderr string `json:"stderr"`
+}
+
+func serveCmd(br *Bridge, w http.ResponseWriter, r *http.Request) {
+    err := COMMAND_OK
+    msg := "OK"
+
+    body, _ := ioutil.ReadAll(r.Body)
+    r.Body.Close()
+
+    req := CommandReq{}
+    json.Unmarshal(body, &req)
+
+    if req.Devid == "" || req.Cmd == "" {
+        err = COMMAND_PARAMETER_ERR
+        msg = "devid and cmd required"
+    } else if dev, ok := br.devices[req.Devid]; !ok {
+        err = COMMAND_DEV_OFFLINE
+        msg = "Device is offline"
+    } else {
+        req.ID = dev.cmdid
+        dev.cmd[req.ID] = make(chan *wsMessage)
+        dev.cmdid = dev.cmdid + 1
+        if dev.cmdid == 1024 {
+            dev.cmdid = 0
+        }
+        js, _ := json.Marshal(req)
+        dev.wsWrite(websocket.TextMessage, js)
+
+        wsMsg := <- dev.cmd[req.ID]
+
+        res := CommandResult{}
+        json.Unmarshal(wsMsg.data, &res)
+
+        delete(dev.cmd, res.ID)
+        res.ID = 0
+
+        js, _ = json.Marshal(res)
+
+        w.Write(js)
+        return
+    }
+
+    res := CommandResult{Err: err, Msg: msg}
+    js, _ := json.Marshal(res)
+    w.Write(js)
+}