Rainnny/LicenseServer
1
0
Fork 0
Code Issues 1 Pull Requests 10 Actions Packages Projects Releases Activity

Request body validation

Browse Source
This commit is contained in:
Braydon 2023-05-31 02:19:48 -04:00
parent 3412668ec1
commit f1e3570546
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/main/java/me/braydon/license/controller/LicenseController.java
View File

@ -1,5 +1,6 @@
package me.braydon.license.controller;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import jakarta.servlet.http.HttpServletRequest;
import lombok.NonNull;
@ -8,6 +9,7 @@ import me.braydon.license.exception.APIException;
import me.braydon.license.model.License;
import me.braydon.license.service.LicenseService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
@ -45,11 +47,21 @@ public final class LicenseController {
String ip = request.getRemoteAddr(); // The IP of the requester
JsonObject jsonObject = LicenseServer.GSON.fromJson(body, JsonObject.class);
String key = jsonObject.get("key").getAsString(); // Get the key
String product = jsonObject.get("product").getAsString(); // Get the product
String hwid = jsonObject.get("hwid").getAsString(); // Get the hwid
JsonElement key = jsonObject.get("key"); // Get the key
JsonElement product = jsonObject.get("product"); // Get the product
JsonElement hwid = jsonObject.get("hwid"); // Get the hwid
service.check(key, product, ip, hwid); // Check the license
// Ensure the body keys aren't null
if (key.isJsonNull() || product.isJsonNull() || hwid.isJsonNull()) {
throw new APIException(HttpStatus.BAD_REQUEST, "Invalid request body");
}
// Check the license
service.check(
key.getAsString(),
product.getAsString(),
ip,
hwid.getAsString()
);
return ResponseEntity.ok().build(); // Return OK
} catch (APIException ex) { // Handle the exception
return ResponseEntity.status(ex.getStatus())