common-app-api/src/main/java/org/openecomp/sdc/common/rest/impl/validator/RequestHeadersValidator.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 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  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.common.rest.impl.validator;
  22
  23 import org.apache.commons.codec.binary.Base64;
  24 import org.openecomp.sdc.common.api.Constants;
  25 import org.openecomp.sdc.common.log.enums.EcompLoggerErrorCode;
  26 import org.openecomp.sdc.common.log.wrappers.Logger;
  27
  28 import javax.servlet.http.HttpServletRequest;
  29 import javax.ws.rs.core.MediaType;
  30 import java.util.List;
  31 import java.util.Map;
  32
  33 public class RequestHeadersValidator {
  34     private static Logger log = Logger.getLogger(RequestHeadersValidator.class.getName());
  35
  36     public static void validateContentType(HttpServletRequest request, MediaType expectedContentType,
  37                                            Map<String, String> headersMap) throws RestRequestValidationException {
  38
  39         log.debug("validateContentType - expected: {}", expectedContentType);
  40         if (request == null || expectedContentType == null) {
  41             throw new RestRequestValidationException("request or media-type are null");
  42         }
  43         String contentType = request.getHeader(Constants.CONTENT_TYPE_HEADER);
  44         if (contentType == null || !contentType.contains(MediaType.APPLICATION_JSON)) {
  45             throw new RestRequestValidationException(
  46                     "Content-Type of requset is different then " + expectedContentType);
  47         } else {
  48             headersMap.put(Constants.CONTENT_TYPE_HEADER, contentType);
  49         }
  50     }
  51
  52     public static void validateIdentificationHeaders(HttpServletRequest request, List<String> identificationList,
  53                                                      Map<String, String> headersMap) throws RestRequestValidationException {
  54
  55         log.debug("validateIdentificationHeaders");
  56         for (String requiredHeader : identificationList) {
  57             String headerVal = request.getHeader(requiredHeader);
  58             if (headerVal != null && !headerVal.isEmpty()) {
  59                 headersMap.put(requiredHeader, headerVal);
  60                 log.debug("found header - {} : {}", requiredHeader, headerVal);
  61             } else {
  62                 log.error(EcompLoggerErrorCode.UNKNOWN_ERROR, "", "", "missing identification header: {}", requiredHeader);
  63                 throw new RestRequestValidationException("missing identification header: " + requiredHeader);
  64             }
  65         }
  66
  67     }
  68
  69     public static void validateMd5(byte[] encodedData, HttpServletRequest request, Map<String, String> headersMap)
  70             throws RestRequestValidationException {
  71
  72         // validate parameters
  73         if (encodedData == null || request == null) {
  74             throw new RestRequestValidationException("encoded data or request are not valid");
  75         }
  76
  77         // calculate MD5 on the data
  78         String md5 = org.apache.commons.codec.digest.DigestUtils.md5Hex(encodedData);
  79         byte[] encodedMd5 = Base64.encodeBase64(md5.getBytes());
  80
  81         // read the Content-MD5 header
  82         String origMd5 = request.getHeader(Constants.MD5_HEADER);
  83         if ((origMd5 == null) || origMd5.isEmpty()) {
  84             throw new RestRequestValidationException("missing Content-MD5 header ");
  85         }
  86
  87         // verify MD5 value
  88         if (!origMd5.equals(new String(encodedMd5))) {
  89             throw new RestRequestValidationException("uploaded file failed MD5 validation");
  90         }
  91         headersMap.put(Constants.MD5_HEADER, origMd5);
  92     }
  93
  94 }