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

Merge pull request #170 from KhasMek/api

Browse source 
api: switch request router and add more paths
This commit is contained in:
Simone Margaritelli 2018-03-13 13:33:12 +01:00 • committed by GitHub
commit 0b51b6ebe4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 117 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

17
modules/api_rest.go
View file

@ -11,6 +11,7 @@ import (
"github.com/bettercap/bettercap/session"
"github.com/bettercap/bettercap/tls"
"github.com/gorilla/mux"
"github.com/gorilla/websocket"
)
@ -146,10 +147,22 @@ func (api *RestAPI) Configure() error {
api.server.Addr = fmt.Sprintf("%s:%d", ip, port)
router := http.NewServeMux()
router := mux.NewRouter()
router.HandleFunc("/api/session", api.sessionRoute)
router.HandleFunc("/api/events", api.eventsRoute)
router.HandleFunc("/api/session", api.sessionRoute)
router.HandleFunc("/api/session/ble", api.sessionRoute)
router.HandleFunc("/api/session/ble/{mac}", api.sessionRoute)
router.HandleFunc("/api/session/env", api.sessionRoute)
router.HandleFunc("/api/session/gateway", api.sessionRoute)
router.HandleFunc("/api/session/interface", api.sessionRoute)
router.HandleFunc("/api/session/lan", api.sessionRoute)
router.HandleFunc("/api/session/lan/{mac}", api.sessionRoute)
router.HandleFunc("/api/session/options", api.sessionRoute)
router.HandleFunc("/api/session/packets", api.sessionRoute)
router.HandleFunc("/api/session/started-at", api.sessionRoute)
router.HandleFunc("/api/session/wifi", api.sessionRoute)
router.HandleFunc("/api/session/wifi/{mac}", api.sessionRoute)
api.server.Handler = router

103
modules/api_rest_controller.go
View file

@ -11,6 +11,7 @@ import (
"github.com/bettercap/bettercap/log"
"github.com/bettercap/bettercap/session"
"github.com/gorilla/mux"
"github.com/gorilla/websocket"
)
@ -67,6 +68,73 @@ func (api *RestAPI) showSession(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I)
}
func (api *RestAPI) showBle(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I.BLE)
}
func (api *RestAPI) showBleEndpoint(w http.ResponseWriter, r *http.Request) {
params := mux.Vars(r)
mac := strings.ToLower(params["mac"])
if dev, found := session.I.BLE.Get(mac); found == true {
toJSON(w, dev)
}
}
func (api *RestAPI) showEnv(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I.Env)
}
func (api *RestAPI) showGateway(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I.Gateway)
}
func (api *RestAPI) showInterface(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I.Interface)
}
func (api *RestAPI) showLan(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I.Lan)
}
func (api *RestAPI) showLanEndpoint(w http.ResponseWriter, r *http.Request) {
params := mux.Vars(r)
mac := strings.ToLower(params["mac"])
if host, found := session.I.Lan.Get(mac); found == true {
toJSON(w, host)
}
}
func (api *RestAPI) showOptions(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I.Options)
}
func (api *RestAPI) showPackets(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I.Queue)
}
func (api *RestAPI) showStartedAt(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I.StartedAt)
}
func (api *RestAPI) showWiFi(w http.ResponseWriter, r *http.Request) {
toJSON(w, session.I.WiFi)
}
func (api *RestAPI) showWiFiEndpoint(w http.ResponseWriter, r *http.Request) {
params := mux.Vars(r)
mac := strings.ToLower(params["mac"])
if station, found := session.I.WiFi.Get(mac); found == true {
toJSON(w, station)
// cycle through station clients if not a station.
} else {
for _, ap := range session.I.WiFi.List() {
if client, found := ap.Get(mac); found == true {
toJSON(w, client)
}
}
}
}
func (api *RestAPI) runSessionCommand(w http.ResponseWriter, r *http.Request) {
var err error
var cmd CommandRequest
@ -211,7 +279,40 @@ func (api *RestAPI) sessionRoute(w http.ResponseWriter, r *http.Request) {
if api.checkAuth(r) == false {
setAuthFailed(w, r)
} else if r.Method == "GET" {
api.showSession(w, r)
params := mux.Vars(r)
if r.URL.String() == "/api/session" {
api.showSession(w, r)
} else if strings.HasPrefix(r.URL.String(), "/api/session/ble") {
if params["mac"] != "" {
api.showBleEndpoint(w, r)
} else {
api.showBle(w, r)
}
} else if r.URL.String() == "/api/session/env" {
api.showEnv(w, r)
} else if r.URL.String() == "/api/session/gateway" {
api.showGateway(w, r)
} else if r.URL.String() == "/api/session/interface" {
api.showInterface(w, r)
} else if strings.HasPrefix(r.URL.String(), "/api/session/lan") {
if params["mac"] != "" {
api.showLanEndpoint(w, r)
} else {
api.showLan(w, r)
}
} else if r.URL.String() == "/api/session/options" {
api.showOptions(w, r)
} else if r.URL.String() == "/api/session/packets" {
api.showPackets(w, r)
} else if r.URL.String() == "/api/session/started-at" {
api.showStartedAt(w, r)
} else if strings.HasPrefix(r.URL.String(), "/api/session/wifi") {
if params["mac"] != "" {
api.showWiFiEndpoint(w, r)
} else {
api.showWiFi(w, r)
}
}
} else if r.Method == "POST" {
api.runSessionCommand(w, r)
} else {