shikigrid/api/unit_report.go

package api

import (
	"encoding/json"
	"fmt"
	"github.com/evilsocket/islazy/log"
	"github.com/andatoshiki/shikigrid/models"
	"io/ioutil"
	"net"
	"net/http"
)

type apReport struct {
	ESSID string `json:"essid"`
	BSSID string `json:"bssid"`
}

func (api *API) unitReport(client string, unit *models.Unit, ap apReport) error {
	if parsed, err := net.ParseMAC(ap.BSSID); err != nil {
		return fmt.Errorf("error while parsing wifi ap bssid %s from %s: %v", ap.BSSID, client, err)
	} else {
		// normalize
		ap.BSSID = parsed.String()
	}

	if existing := unit.FindAccessPoint(ap.ESSID, ap.BSSID); existing == nil {
		log.Debug("unit %s (%s %s) reporting new wifi access point %v", unit.Identity(), unit.Address,
			unit.Country, ap)

		newAP := models.AccessPoint{
			Name:   ap.ESSID,
			Mac:    ap.BSSID,
			UnitID: unit.ID,
		}

		if err := models.Create(&newAP).Error; err != nil {
			return fmt.Errorf("error creating ap %v: %v", newAP, err)
		}
	} else if err := models.Update(existing).Error; err != nil {
		return fmt.Errorf("error updating ap %v: %v", existing, err)
	}

	return nil
}

func (api *API) UnitReportAP(w http.ResponseWriter, r *http.Request) {
	unit := Authenticate(w, r)
	if unit == nil {
		return
	}

	client := clientIP(r)
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		ERROR(w, http.StatusUnprocessableEntity, ErrEmpty)
		return
	}

	var ap apReport
	if err = json.Unmarshal(body, &ap); err != nil {
		log.Warning("error while reading wifi ap from %s: %v", client, err)
		ERROR(w, http.StatusUnprocessableEntity, ErrEmpty)
		return
	}

	if err := api.unitReport(client, unit, ap); err != nil {
		log.Warning("%v", err)
		ERROR(w, http.StatusUnprocessableEntity, ErrEmpty)
		return
	}

	JSON(w, http.StatusOK, map[string]interface{}{
		"success": true,
	})
}

func (api *API) UnitReportMultipleAP(w http.ResponseWriter, r *http.Request) {
	unit := Authenticate(w, r)
	if unit == nil {
		return
	}

	client := clientIP(r)
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		ERROR(w, http.StatusUnprocessableEntity, ErrEmpty)
		return
	}

	var aps []apReport
	if err = json.Unmarshal(body, &aps); err != nil {
		log.Warning("error while reading wifi ap list from %s: %v", client, err)
		ERROR(w, http.StatusUnprocessableEntity, ErrEmpty)
		return
	}

	for _, ap := range aps {
		if err := api.unitReport(client, unit, ap); err != nil {
			log.Warning("%v", err)
			ERROR(w, http.StatusUnprocessableEntity, ErrEmpty)
			return
		}
	}

	JSON(w, http.StatusOK, map[string]interface{}{
		"success": true,
	})
}