mirror of
https://github.com/Macrame-App/Macrame
synced 2025-12-29 15:29:26 +00:00
b598a090bc
This is the first iteration of the registration of devices and authentication for a couple of endpoints. This needs to be refactored, the code is a bit of a mess. Also because of testing some endpoints are available for remotes that shouldn't be.
102 lines
2.1 KiB
Go
102 lines
2.1 KiB
Go
package helper
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"be/app/structs"
|
|
. "be/app/structs"
|
|
)
|
|
|
|
func EndpointAccess(w http.ResponseWriter, r *http.Request) (bool, string) {
|
|
ip, _, err := net.SplitHostPort(r.RemoteAddr)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if (isLocal(ip) && isEndpointAllowed("Local", r.URL.Path)) ||
|
|
(isLanRemote(ip) && isEndpointAllowed("Remote", r.URL.Path)) {
|
|
log.Println(r.URL.Path, "endpoint access: accessible")
|
|
return true, ""
|
|
} else if isLanRemote(ip) && isEndpointAllowed("Auth", r.URL.Path) {
|
|
log.Println(r.URL.Path, "endpoint access: authorized")
|
|
|
|
data := decryptAuth(r)
|
|
|
|
return data != "", data
|
|
}
|
|
|
|
log.Println(r.URL.Path, "endpoint access: not authorized or accessible")
|
|
|
|
return false, ""
|
|
}
|
|
|
|
func isLocal(ip string) bool {
|
|
return ip == "127.0.0.1" || ip == "::1"
|
|
}
|
|
|
|
func isLanRemote(ip string) bool {
|
|
return strings.HasPrefix(ip, "192.168.")
|
|
}
|
|
|
|
func isEndpointAllowed(source string, endpoint string) bool {
|
|
var endpoints, err = getAllowedEndpoints(source)
|
|
if err != "" {
|
|
log.Println(err)
|
|
}
|
|
|
|
if (endpoints != nil) && (len(endpoints) > 0) {
|
|
for _, e := range endpoints {
|
|
if e == endpoint {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func getAllowedEndpoints(source string) (endpoints []string, err string) {
|
|
if source == "Local" {
|
|
return Endpoints.Local, ""
|
|
}
|
|
if source == "Remote" {
|
|
return Endpoints.Remote, ""
|
|
}
|
|
if source == "Auth" {
|
|
return Endpoints.Auth, ""
|
|
}
|
|
|
|
return []string{}, "No allowed endpoints"
|
|
}
|
|
|
|
func decryptAuth(r *http.Request) string {
|
|
var req structs.Authcall
|
|
|
|
err := json.NewDecoder(r.Body).Decode(&req)
|
|
|
|
if err != nil || req.Uuid == "" || req.Data == "" {
|
|
return ""
|
|
}
|
|
|
|
deviceKey, errKey := GetKeyByUuid(req.Uuid)
|
|
decryptData, errDec := DecryptAES(deviceKey, req.Data)
|
|
|
|
if errKey != nil && errDec != nil || decryptData == "" {
|
|
return ""
|
|
}
|
|
|
|
return decryptData
|
|
}
|
|
|
|
func ParseRequest(req interface{}, data string, r *http.Request) (d interface{}, err error) {
|
|
if data != "" {
|
|
dataBytes := []byte(data)
|
|
return req, json.Unmarshal(dataBytes, &req)
|
|
} else {
|
|
return req, json.NewDecoder(r.Body).Decode(&req)
|
|
}
|
|
}
|