github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-07 13:32:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

new: implemented POST /api/session route (ref #5).

Browse source
This commit is contained in:
evilsocket 2018-01-08 04:26:19 +01:00
parent c2826c9094
commit ee4b783015
2 changed files with 46 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

55
session/modules/api_rest.go
View file

@ -57,25 +57,60 @@ func NewRestAPI(s *session.Session) *RestAPI {
return api.Stop()
}))
http.HandleFunc("/api/session", api.getSession)
http.HandleFunc("/api/session", api.sessRoute)
return api
}
func (api *RestAPI) getSession(w http.ResponseWriter, r *http.Request) {
type JSSessionRequest struct {
Command string `json:"cmd"`
}
type JSSessionResponse struct {
Error string `json:"error"`
}
func (api *RestAPI) sessRoute(w http.ResponseWriter, r *http.Request) {
if api.checkAuth(w, r) == false {
return
}
js, err := json.Marshal(api.Session)
if err != nil {
log.Errorf("Error while returning session: %s", err)
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
if r.Method == "GET" {
js, err := json.Marshal(api.Session)
if err != nil {
log.Errorf("Error while returning session: %s", err)
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
w.Write(js)
w.Header().Set("Content-Type", "application/json")
w.Write(js)
} else if r.Method == "POST" && r.Body != nil {
var req JSSessionRequest
var res JSSessionResponse
err := json.NewDecoder(r.Body).Decode(&req)
if err != nil {
http.Error(w, err.Error(), 400)
return
}
err = api.Session.Run(req.Command)
if err != nil {
res.Error = err.Error()
}
js, err := json.Marshal(res)
if err != nil {
log.Errorf("Error while returning response: %s", err)
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
w.Write(js)
} else {
http.Error(w, "Not Found", 404)
}
}
func (api RestAPI) checkAuth(w http.ResponseWriter, r *http.Request) bool {