src/main/java/com/att/nsa/dmaap/util/ContentLengthInterceptor.java

   1 /*******************************************************************************
   2  *  ============LICENSE_START=======================================================
   3  *  org.onap.dmaap
   4  *  ================================================================================
   5  *  Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  *  ================================================================================
   7  *  Licensed under the Apache License, Version 2.0 (the "License");
   8  *  you may not use this file except in compliance with the License.
   9  *  You may obtain a copy of the License at
  10  *        http://www.apache.org/licenses/LICENSE-2.0
  11 *
  12  *  Unless required by applicable law or agreed to in writing, software
  13  *  distributed under the License is distributed on an "AS IS" BASIS,
  14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  *  See the License for the specific language governing permissions and
  16  *  limitations under the License.
  17  *  ============LICENSE_END=========================================================
  18  *
  19  *  ECOMP is a trademark and service mark of AT&T Intellectual Property.
  20  *
  21  *******************************************************************************/
  22 package com.att.nsa.dmaap.util;\r
  23 \r
  24 import java.util.Map;\r
  25 import javax.servlet.http.HttpServletRequest;\r
  26 import javax.servlet.http.HttpServletResponse;\r
  27 import org.apache.http.HttpStatus;\r
  28 import com.att.eelf.configuration.EELFLogger;\r
  29 import com.att.eelf.configuration.EELFManager;\r
  30 import org.json.JSONException;\r
  31 import org.json.JSONObject;\r
  32 import org.springframework.stereotype.Component;\r
  33 import org.onap.dmaap.dmf.mr.CambriaApiException;\r
  34 import org.onap.dmaap.dmf.mr.exception.DMaaPResponseCode;\r
  35 import org.onap.dmaap.dmf.mr.exception.ErrorResponse;\r
  36 import ajsc.beans.interceptors.AjscInterceptor;\r
  37 \r
  38 /**\r
  39  * AJSC Intercepter implementation of ContentLengthFilter\r
  40  */\r
  41 @Component\r
  42 public class ContentLengthInterceptor implements AjscInterceptor{\r
  43 \r
  44         \r
  45         private String defLength;\r
  46         //private Logger log = Logger.getLogger(ContentLengthInterceptor.class.toString());\r
  47         private static final EELFLogger log = EELFManager.getInstance().getLogger(ContentLengthInterceptor.class);\r
  48 \r
  49 \r
  50         /**\r
  51          * Intercepter method to intercept requests before processing\r
  52          */\r
  53         @Override\r
  54         public boolean allowOrReject(HttpServletRequest httpservletrequest, HttpServletResponse httpservletresponse,\r
  55                         Map map) throws Exception {\r
  56                                 \r
  57                 log.info("inside Interceptor allowOrReject content length checking before pub/sub");\r
  58                 \r
  59                 JSONObject jsonObj = null;\r
  60                 int requestLength = 0;\r
  61                 setDefLength(System.getProperty("maxcontentlength"));\r
  62                 try {\r
  63                         // retrieving content length from message header\r
  64 \r
  65                         if (null != httpservletrequest.getHeader("Content-Length")) {\r
  66                                 requestLength = Integer.parseInt(httpservletrequest.getHeader("Content-Length"));\r
  67                         }\r
  68                         // retrieving encoding from message header\r
  69                         String transferEncoding = httpservletrequest.getHeader("Transfer-Encoding");\r
  70                         // checking for no encoding, chunked and requestLength greater then\r
  71                         // default length\r
  72                                 if (null != transferEncoding && !(transferEncoding.contains("chunked"))\r
  73                                                 && (requestLength > Integer.parseInt(getDefLength()))) {\r
  74                                         jsonObj = new JSONObject().append("defaultlength", getDefLength())\r
  75                                                         .append("requestlength", requestLength);\r
  76                                         log.error("message length is greater than default");\r
  77                                         throw new CambriaApiException(jsonObj);\r
  78                                 } \r
  79                                 else if (null == transferEncoding && (requestLength > Integer.parseInt(getDefLength()))) \r
  80                                 {\r
  81                                         jsonObj = new JSONObject().append("defaultlength", getDefLength()).append(\r
  82                                                         "requestlength", requestLength);\r
  83                                         log.error("Request message is not chunked or request length is greater than default length");\r
  84                                         throw new CambriaApiException(jsonObj);\r
  85                                 \r
  86                                 \r
  87                                 } \r
  88                                 else \r
  89                                 {\r
  90                                 //chain.doFilter(req, res);\r
  91                                 return true;\r
  92                                 }\r
  93                         \r
  94                 } catch (CambriaApiException | NumberFormatException | JSONException e) {\r
  95                         \r
  96                         log.info("Exception obj--"+e);\r
  97                         log.error("message size is greater then default"+e.getMessage());
  98                         String messg=e.toString();
  99                         if(jsonObj!=null){
 100                          messg=jsonObj.toString();
 101                         }\r
 102                         ErrorResponse errRes = new ErrorResponse(HttpStatus.SC_REQUEST_TOO_LONG,\r
 103                                         DMaaPResponseCode.MSG_SIZE_EXCEEDS_MSG_LIMIT.getResponseCode(), System.getProperty("msg_size_exceeds")\r
 104                                                         + messg);\r
 105                         log.info(errRes.toString());\r
 106                         \r
 107                         \r
 108                         map.put(HttpServletResponse.SC_INTERNAL_SERVER_ERROR,"test");\r
 109                         httpservletresponse.setStatus(HttpStatus.SC_REQUEST_TOO_LONG);
 110                         if(httpservletresponse.getOutputStream()!=null){\r
 111                         httpservletresponse.getOutputStream().write(errRes.toString().getBytes());
 112                         }\r
 113                         return false;\r
 114                 }\r
 115 \r
 116                 \r
 117                 \r
 118         }\r
 119 \r
 120         \r
 121         /**\r
 122          * Get Default Content Length\r
 123          * @return defLength\r
 124          */\r
 125         public String getDefLength() {\r
 126                 return defLength;\r
 127         }\r
 128         /**\r
 129          * Set Default Content Length\r
 130          * @param defLength\r
 131          */\r
 132         public void setDefLength(String defLength) {\r
 133                 this.defLength = defLength;\r
 134         }\r
 135 \r
 136         \r
 137 \r
 138 }\r