summaryrefslogtreecommitdiffstats
path: root/src/main/java/com/att/nsa/dmaap/util/ContentLengthInterceptor.java
blob: 4ed932202853dbb35287ff326556cc2096ad7432 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
/*******************************************************************************
 *  ============LICENSE_START=======================================================
 *  org.onap.dmaap
 *  ================================================================================
 *  Copyright © 2017 AT&T Intellectual Property. All rights reserved.
 *  ================================================================================
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *        http://www.apache.org/licenses/LICENSE-2.0
*  
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *  ============LICENSE_END=========================================================
 *  
 *  ECOMP is a trademark and service mark of AT&T Intellectual Property.
 *  
 *******************************************************************************/
package com.att.nsa.dmaap.util;

import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.http.HttpStatus;
import com.att.eelf.configuration.EELFLogger;
import com.att.eelf.configuration.EELFManager;
import org.json.JSONException;
import org.json.JSONObject;
import org.springframework.stereotype.Component;
import com.att.dmf.mr.CambriaApiException;
import com.att.dmf.mr.exception.DMaaPResponseCode;
import com.att.dmf.mr.exception.ErrorResponse;
import ajsc.beans.interceptors.AjscInterceptor;

/**
 * AJSC Intercepter implementation of ContentLengthFilter
 */
@Component
public class ContentLengthInterceptor implements AjscInterceptor{

	
	private String defLength;
	//private Logger log = Logger.getLogger(ContentLengthInterceptor.class.toString());
	private static final EELFLogger log = EELFManager.getInstance().getLogger(ContentLengthInterceptor.class);


	/**
	 * Intercepter method to intercept requests before processing
	 */
	@Override
	public boolean allowOrReject(HttpServletRequest httpservletrequest, HttpServletResponse httpservletresponse,
			Map map) throws Exception {
				
		log.info("inside Interceptor allowOrReject content length checking before pub/sub");
		
		JSONObject jsonObj = null;
		int requestLength = 0;
		setDefLength(System.getProperty("maxcontentlength"));
		try {
			// retrieving content length from message header

			if (null != httpservletrequest.getHeader("Content-Length")) {
				requestLength = Integer.parseInt(httpservletrequest.getHeader("Content-Length"));
			}
			// retrieving encoding from message header
			String transferEncoding = httpservletrequest.getHeader("Transfer-Encoding");
			// checking for no encoding, chunked and requestLength greater then
			// default length
				if (null != transferEncoding && !(transferEncoding.contains("chunked"))
						&& (requestLength > Integer.parseInt(getDefLength()))) {
					jsonObj = new JSONObject().append("defaultlength", getDefLength())
							.append("requestlength", requestLength);
					log.error("message length is greater than default");
					throw new CambriaApiException(jsonObj);
				} 
				else if (null == transferEncoding && (requestLength > Integer.parseInt(getDefLength()))) 
				{
					jsonObj = new JSONObject().append("defaultlength", getDefLength()).append(
							"requestlength", requestLength);
					log.error("Request message is not chunked or request length is greater than default length");
					throw new CambriaApiException(jsonObj);
				
				
				} 
				else 
				{
				//chain.doFilter(req, res);
				return true;
				}
			
		} catch (CambriaApiException | NumberFormatException | JSONException e) {
			
			log.info("Exception obj--"+e);
			log.error("message size is greater then default"+e.getMessage());
			String messg=e.toString();
			if(jsonObj!=null){
			 messg=jsonObj.toString();
			}
			ErrorResponse errRes = new ErrorResponse(HttpStatus.SC_REQUEST_TOO_LONG,
					DMaaPResponseCode.MSG_SIZE_EXCEEDS_MSG_LIMIT.getResponseCode(), System.getProperty("msg_size_exceeds")
							+ messg);
			log.info(errRes.toString());
			
			
			map.put(HttpServletResponse.SC_INTERNAL_SERVER_ERROR,"test");
			httpservletresponse.setStatus(HttpStatus.SC_REQUEST_TOO_LONG);
			if(httpservletresponse.getOutputStream()!=null){
			httpservletresponse.getOutputStream().write(errRes.toString().getBytes());
			}
			return false;
		}

		
		
	}

	
	/**
	 * Get Default Content Length
	 * @return defLength
	 */
	public String getDefLength() {
		return defLength;
	}
	/**
	 * Set Default Content Length
	 * @param defLength
	 */
	public void setDefLength(String defLength) {
		this.defLength = defLength;
	}

	

}