mirror of https://github.com/wh1te909/rmmagent.git
115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package agent
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
)
|
|
|
|
type WindowsUpdate struct {
|
|
PatchID int64 `json:"id"`
|
|
KB string `json:"kb"`
|
|
GUID string `json:"guid"`
|
|
}
|
|
|
|
type WindowsUpdates []WindowsUpdate
|
|
|
|
type SaltDownloadRet struct {
|
|
Updates interface{} `json:"Updates"`
|
|
Message string `json:"Message"`
|
|
Success bool `json:"Success"`
|
|
}
|
|
|
|
type SaltInstallRet struct {
|
|
Updates interface{} `json:"Updates"`
|
|
Message string `json:"Message"`
|
|
Success bool `json:"Success"`
|
|
NeedsReboot bool `json:"NeedsReboot"`
|
|
}
|
|
|
|
type SaltUpdateOutput struct {
|
|
SaltDownload SaltDownloadRet `json:"Download"`
|
|
SaltInstall SaltInstallRet `json:"Install"`
|
|
}
|
|
|
|
type LocalSaltUpdate struct {
|
|
Local SaltUpdateOutput `json:"local"`
|
|
}
|
|
|
|
func (a *WindowsAgent) InstallPatches() {
|
|
data := WindowsUpdates{}
|
|
url := fmt.Sprintf("%s/api/v3/%s/winupdater/", a.Server, a.AgentID)
|
|
|
|
r := &APIRequest{
|
|
URL: url,
|
|
Method: "GET",
|
|
Headers: a.Headers,
|
|
Timeout: 30,
|
|
LocalCert: a.DB.Cert,
|
|
Debug: a.Debug,
|
|
}
|
|
|
|
r1, err := r.MakeRequest()
|
|
if err != nil {
|
|
a.Logger.Debugln(err)
|
|
return
|
|
}
|
|
|
|
if r1.IsError() {
|
|
a.Logger.Debugln("Install Patches:", r1.String())
|
|
return
|
|
}
|
|
|
|
if err := json.Unmarshal(r1.Body(), &data); err != nil {
|
|
a.Logger.Debugln(err)
|
|
return
|
|
}
|
|
|
|
// no patches
|
|
if len(data) == 0 {
|
|
return
|
|
}
|
|
|
|
saltdata := LocalSaltUpdate{}
|
|
|
|
for _, patch := range data {
|
|
out, err := a.LocalSaltCall("win_wua.get", []string{patch.KB, "download=True", "install=True"}, 7200)
|
|
if err != nil {
|
|
a.Logger.Debugln(err)
|
|
continue
|
|
}
|
|
|
|
if err := json.Unmarshal(out, &saltdata); err != nil {
|
|
a.Logger.Debugln(err)
|
|
continue
|
|
}
|
|
|
|
a.Logger.Infoln(saltdata)
|
|
payload := map[string]string{"agent_id": a.AgentID, "kb": patch.KB}
|
|
|
|
if saltdata.Local.SaltInstall.Updates == "Nothing to install" {
|
|
payload["results"] = "alreadyinstalled"
|
|
} else {
|
|
if saltdata.Local.SaltInstall.Success {
|
|
payload["results"] = "success"
|
|
} else {
|
|
payload["results"] = "failed"
|
|
}
|
|
}
|
|
|
|
r := &APIRequest{
|
|
URL: url,
|
|
Headers: a.Headers,
|
|
Method: "PATCH",
|
|
Payload: payload,
|
|
Timeout: 30,
|
|
LocalCert: a.DB.Cert,
|
|
Debug: a.Debug,
|
|
}
|
|
|
|
_, rerr := r.MakeRequest()
|
|
if rerr != nil {
|
|
a.Logger.Debugln(rerr)
|
|
}
|
|
}
|
|
}
|