From 55ece585b5ee34967048053138b93004d0bd81c4 Mon Sep 17 00:00:00 2001 From: Chun-Yeow Yeoh Date: Fri, 14 Sep 2018 11:17:11 +0800 Subject: [PATCH] add missing API limits on CoAP POST method Signed-off-by: Chun-Yeow Yeoh --- .../server/transport/coap/CoapTransportResource.java | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/transport/coap/src/main/java/org/thingsboard/server/transport/coap/CoapTransportResource.java b/transport/coap/src/main/java/org/thingsboard/server/transport/coap/CoapTransportResource.java index bac68ffc58..8a596d7119 100644 --- a/transport/coap/src/main/java/org/thingsboard/server/transport/coap/CoapTransportResource.java +++ b/transport/coap/src/main/java/org/thingsboard/server/transport/coap/CoapTransportResource.java @@ -113,6 +113,12 @@ public class CoapTransportResource extends CoapResource { @Override public void handlePOST(CoapExchange exchange) { + if(quotaService.isQuotaExceeded(exchange.getSourceAddress().getHostAddress())) { + log.warn("COAP Quota exceeded for [{}:{}] . Disconnect", exchange.getSourceAddress().getHostAddress(), exchange.getSourcePort()); + exchange.respond(ResponseCode.BAD_REQUEST); + return; + } + Optional featureType = getFeatureType(exchange.advanced().getRequest()); if (!featureType.isPresent()) { log.trace("Missing feature type parameter");