first draft
This commit is contained in:
commit
0fec126ac4
GPG Key ID:
86D4882C6C6CA48B
68
functions.go
Normal file
68
functions.go
Normal file
@ -0,0 +1,68 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
|
||||
"git.nils.zone/clientserverapi/server/storage"
|
||||
t "git.nils.zone/clientserverapi/server/types"
|
||||
"git.nils.zone/nils/prettify"
|
||||
"github.com/elastic/go-sysinfo/types"
|
||||
)
|
||||
|
||||
func check(err error) {
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
storage.GetComputers()
|
||||
}
|
||||
|
||||
func homePage(w http.ResponseWriter, r *http.Request) {
|
||||
fmt.Fprintf(w, "Nothing here...")
|
||||
}
|
||||
|
||||
func commands(w http.ResponseWriter, r *http.Request) {
|
||||
// Client requests instructions
|
||||
log.Println("Client requesting instructions, sending the following:")
|
||||
var o t.OSInfo
|
||||
err := json.NewDecoder(r.Body).Decode(&o)
|
||||
check(err)
|
||||
log.Println("Received the following OS info:")
|
||||
prettify.Print(o)
|
||||
|
||||
var i t.Instruction
|
||||
|
||||
// check the OS and send the appropriate command
|
||||
switch o.OSType {
|
||||
case "linux":
|
||||
// i = instruction{Command: "uname", Args: []string{"-a"}}
|
||||
i = t.Instruction{Command: "poweroff"}
|
||||
case "windows":
|
||||
i = t.Instruction{Command: "winver"}
|
||||
default:
|
||||
return
|
||||
}
|
||||
|
||||
log.Println("Sending the following instructions:")
|
||||
prettify.Print(i)
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(i)
|
||||
}
|
||||
|
||||
func response(w http.ResponseWriter, r *http.Request) {
|
||||
log.Println("Got the following output from client:")
|
||||
var p t.CmdResponse
|
||||
err := json.NewDecoder(r.Body).Decode(&p)
|
||||
check(err)
|
||||
prettify.Print(p)
|
||||
}
|
||||
|
||||
func register(w http.ResponseWriter, r *http.Request) {
|
||||
log.Println("Got the following output from client:")
|
||||
var h types.HostInfo
|
||||
err := json.NewDecoder(r.Body).Decode(&h)
|
||||
check(err)
|
||||
prettify.Print(h)
|
||||
}
|
||||
9
go.mod
Normal file
9
go.mod
Normal file
@ -0,0 +1,9 @@
|
||||
module git.nils.zone/clientserverapi/server
|
||||
|
||||
go 1.15
|
||||
|
||||
require (
|
||||
git.nils.zone/nils/prettify v0.0.4
|
||||
github.com/elastic/go-sysinfo v1.4.0
|
||||
github.com/gorilla/mux v1.8.0
|
||||
)
|
||||
39
go.sum
Normal file
39
go.sum
Normal file
@ -0,0 +1,39 @@
|
||||
git.nils.zone/nils/prettify v0.0.4 h1:y014Ejp0yhFAc9vu5U0nJRFxtXN6k6eWW6LiUkwn4NQ=
|
||||
git.nils.zone/nils/prettify v0.0.4/go.mod h1:q4ydEhSvXuywHe5U6uSEoEeS9f2Upz/n8z0MJylLz5c=
|
||||
github.com/TylerBrock/colorjson v0.0.0-20200706003622-8a50f05110d2 h1:ZBbLwSJqkHBuFDA6DUhhse0IGJ7T5bemHyNILUjvOq4=
|
||||
github.com/TylerBrock/colorjson v0.0.0-20200706003622-8a50f05110d2/go.mod h1:VSw57q4QFiWDbRnjdX8Cb3Ow0SFncRw+bA/ofY6Q83w=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/elastic/go-sysinfo v1.4.0 h1:LUnK6TNOuy8JEByuDzTAQH3iQ6bIywy55+Z+QlKNSWk=
|
||||
github.com/elastic/go-sysinfo v1.4.0/go.mod h1:i1ZYdU10oLNfRzq4vq62BEwD2fH8KaWh6eh0ikPT9F0=
|
||||
github.com/elastic/go-windows v1.0.0/go.mod h1:TsU0Nrp7/y3+VwE82FoZF8gC/XFg/Elz6CcloAxnPgU=
|
||||
github.com/fatih/color v1.9.0 h1:8xPHl4/q1VyqGIPif1F+1V3Y3lSmrq01EabUW3CoW5s=
|
||||
github.com/fatih/color v1.9.0/go.mod h1:eQcE1qtQxscV5RaZvpXrrb8Drkc3/DdQ+uUYCNjL+zU=
|
||||
github.com/gorilla/mux v1.8.0 h1:i40aqfkR1h2SlN9hojwV5ZA91wcXFOvkdNIeFDP5koI=
|
||||
github.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So=
|
||||
github.com/hokaccha/go-prettyjson v0.0.0-20190818114111-108c894c2c0e h1:0aewS5NTyxftZHSnFaJmWE5oCCrj4DyEXkAiMa1iZJM=
|
||||
github.com/hokaccha/go-prettyjson v0.0.0-20190818114111-108c894c2c0e/go.mod h1:pFlLw2CfqZiIBOx6BuCeRLCrfxBJipTY0nIOF/VbGcI=
|
||||
github.com/jessevdk/go-flags v1.4.0/go.mod h1:4FA24M0QyGHXBuZZK/XkWh8h0e1EYbRYJSGM75WSRxI=
|
||||
github.com/joeshaw/multierror v0.0.0-20140124173710-69b34d4ec901/go.mod h1:Z86h9688Y0wesXCyonoVr47MasHilkuLMqGhRZ4Hpak=
|
||||
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||
github.com/mattn/go-colorable v0.1.4 h1:snbPLB8fVfU9iwbbo30TPtbLRzwWu6aJS6Xh4eaaviA=
|
||||
github.com/mattn/go-colorable v0.1.4/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
|
||||
github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
|
||||
github.com/mattn/go-isatty v0.0.11 h1:FxPOTFNqGkuDUGi3H/qkUbQO4ZiBa2brKq5r0l8TGeM=
|
||||
github.com/mattn/go-isatty v0.0.11/go.mod h1:PhnuNfih5lzO57/f3n+odYbM4JtupLOxQOAqxQCu2WE=
|
||||
github.com/pkg/errors v0.8.1 h1:iURUrRGxPUNPdy5/HRSm+Yj6okJ6UtLINN0Q9M4+h3I=
|
||||
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/prometheus/procfs v0.0.0-20190425082905-87a4384529e0/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||
golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20191025021431-6c3a3bfe00ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20191026070338-33540a1f6037 h1:YyJpGZS1sBuBCzLAR1VEpK193GlqGZbnPFnPV/5Rsb4=
|
||||
golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
howett.net/plist v0.0.0-20181124034731-591f970eefbb/go.mod h1:vMygbs4qMhSZSc4lCUl2OEE+rDiIIJAIdR4m7MiMcm0=
|
||||
42
main.go
Normal file
42
main.go
Normal file
@ -0,0 +1,42 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"log"
|
||||
"net/http"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
)
|
||||
|
||||
func handleRequests(runmode string) {
|
||||
// creates a new instance of a mux router
|
||||
router := mux.NewRouter().StrictSlash(true)
|
||||
// replace http.HandleFunc with myRouter.HandleFunc
|
||||
router.HandleFunc("/", homePage)
|
||||
router.HandleFunc("/commands", commands)
|
||||
router.HandleFunc("/response", response)
|
||||
if runmode == "reg" {
|
||||
router.HandleFunc("/register", register)
|
||||
}
|
||||
|
||||
// serve the router
|
||||
log.Fatal(http.ListenAndServe(":10000", router))
|
||||
}
|
||||
|
||||
func main() {
|
||||
// Flags
|
||||
registerPtr := flag.Bool("register", false, "allow new clients to register with the server")
|
||||
|
||||
// parse all flags
|
||||
flag.Parse()
|
||||
|
||||
if *registerPtr == true {
|
||||
log.Println("Starting the server in registration mode...")
|
||||
handleRequests("reg")
|
||||
} else {
|
||||
log.Println("Starting the server...")
|
||||
handleRequests("")
|
||||
}
|
||||
|
||||
// start server
|
||||
}
|
||||
7
storage/storageFunctions.go
Normal file
7
storage/storageFunctions.go
Normal file
@ -0,0 +1,7 @@
|
||||
package storage
|
||||
|
||||
import "log"
|
||||
|
||||
func GetComputers() {
|
||||
log.Println("No computers here!")
|
||||
}
|
||||
20
types/types.go
Normal file
20
types/types.go
Normal file
@ -0,0 +1,20 @@
|
||||
package types
|
||||
|
||||
// instruction is a structured way of sending commands
|
||||
type Instruction struct {
|
||||
Command string `json:"command"`
|
||||
Args []string `json:"args"`
|
||||
}
|
||||
|
||||
// cmdResponse is a response from the client to the server
|
||||
type CmdResponse struct {
|
||||
Stdout string `json:"stdout"`
|
||||
Stderr string `json:"stderr"`
|
||||
ExitCode int `json:"exitcode"`
|
||||
}
|
||||
|
||||
// osInfo needs a comment
|
||||
type OSInfo struct {
|
||||
OSType string `json:"os_type"`
|
||||
OSArch string `json:"os_arch"`
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user