`

基于SpringMVC的HTTP+JSON签名校验

 
阅读更多

客户端

 

Client.java

/**
     * 向服务器发送请求并解析返回响应结果
     * @param url - 请求的地址
     * @param baseClientDTO - 请求的参数
     * @param type - 返回参数的类型
     * @return 返回参数实例
     * @throws IOException
     */
    public static <T> T sendAndReturn(String url, BaseClientDTO baseClientDTO, TypeReference<T> type) throws IOException {
        //TODO 基于hmac256(baseClientDTO,secretKey)签名，服务器同样签名后比对，相同为真，不同为否
        String param = JSON.toJSONString(baseClientDTO);
        URL realUrl = new URL(url);
        // 打开和URL之间的连接
        HttpURLConnection connection = (HttpURLConnection)realUrl.openConnection();
        // 设置通用的请求属性
        connection.setConnectTimeout(10000);
        connection.setReadTimeout(20000);
        connection.setDoOutput(true);
        connection.setDoInput(true);
        connection.setRequestMethod("POST");
        connection.setUseCaches(false);
        connection.setInstanceFollowRedirects(true);
        connection.setRequestProperty("Content-Type","application/json;charset=utf-8");
        //签名设置
        String secretKey = OccDownloadService.getInstance().secretKey;
        String signature = null;
        if (secretKey != null) {
            signature = SecretUtil.hmac256(param,secretKey);
        }
        connection.setRequestProperty("signature", signature);

        connection.setRequestProperty("envCode", baseClientDTO.getEnvCode());
        connection.setRequestProperty("nsCode", baseClientDTO.getNsCode());
        connection.setRequestProperty("clientLocalIp",_CLIENT_LOCAL_IP);
        connection.connect();
        PrintWriter out = null;
    	BufferedReader br = null;
    	InputStreamReader ir = null;
        try {
            // 获取URLConnection对象对应的输出流
            out = new PrintWriter(connection.getOutputStream());
            // 发送请求参数
            out.print(param);
            // flush输出流的缓冲
            out.flush();
            //是否请求成功
            InputStream inputStream = (connection.getResponseCode() == HttpURLConnection.HTTP_OK) ?
                                        connection.getInputStream() :
                                        connection.getErrorStream();
            //获取数据
            br = new BufferedReader(ir = new InputStreamReader(inputStream,"UTF-8"));
            String line;
            StringBuilder result = new StringBuilder();
            while ((line = br.readLine()) != null) {
                result.append(line);
            }
            return JSON.parseObject(result.toString(), type);
        }finally {
            close(out,br,ir);
            connection.disconnect();
        }
    }

 

 

SecretUtil.java

 

import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;

/**
 * Created by Think on 2017/3/20.
 */
public class SecretUtil {

    /**
     * 签名加密
     *
     * @param str
     * @param stal
     * @return
     */
    public static String hmac256(String str, String stal) {
        Mac hmac;
        try {
            hmac = Mac.getInstance("HmacSHA256");
            SecretKeySpec macKey = new SecretKeySpec(stal.getBytes(), "RAW");
            hmac.init(macKey);
            byte[] key = hmac.doFinal(str.getBytes());
            return byteToHexString(key);
        } catch (Exception e) {
            return null;
        }
    }

    /**
     * Hex转换
     *
     * @param bt
     * @return
     */
    public static String byteToHexString(byte[] bt) {
        StringBuilder stringBuilder = new StringBuilder("");
        for (byte aSrc : bt) {
            int v = aSrc & 0xFF;
            String hv = Integer.toHexString(v);
            if (hv.length() < 2) {
                stringBuilder.append(0);
            }
            stringBuilder.append(hv);
        }
        return stringBuilder.toString();
    }

}

 

 

 

服务端

 

SecretKeyFilter.java

 

import com.odianyun.cc.business.utils.SpringContextUtil;
import com.odianyun.cc.business.write.manage.IOccSecretKeyManage;
import com.odianyun.cc.model.dto.OccSecretKeyDTO;
import com.odianyun.cc.model.util.SecretUtil;
import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;

public class SecretKeyFilter implements Filter{

	private static final Logger logger = LoggerFactory.getLogger(SecretKeyFilter.class);

	private static final String EXCLUDE_CONFIG_NAME = "excludeURL";

	private SecretKeyFilterConfig secretKeyFilterConfig;
	
	@Override
	public void init(FilterConfig filterConfig) throws ServletException {
		if (logger.isDebugEnabled()) {
			logger.debug("--------------------SecretKeyFilter-------------init ");
		}
		String excludeConfig = filterConfig.getInitParameter(EXCLUDE_CONFIG_NAME);
		this.secretKeyFilterConfig = SecretKeyFilterConfig.getInstance(excludeConfig);
	}

	private String getRequetUriExcludeContextPath(HttpServletRequest req) {
		String requestUrl = req.getRequestURI();
		String contextPath = req.getContextPath();
		String reqPath = requestUrl.replace(contextPath, "");
		while(reqPath.startsWith("//")){
			reqPath = reqPath.replaceFirst("//","/");
		}
		if (logger.isDebugEnabled()) {
			logger.debug("requestUrl={},contextPath={},reqPath={}", requestUrl, contextPath, reqPath);
		}
		return reqPath;
	}

	@Override
	public void doFilter(ServletRequest request, ServletResponse response,
			FilterChain chain) throws IOException, ServletException {
		HttpServletRequest req = (HttpServletRequest)request;
		String requestUri = this.getRequetUriExcludeContextPath(req);
		if (this.secretKeyFilterConfig.match(requestUri)){
			chain.doFilter(request, response);
			return;
		}
	    String envCode = req.getHeader("envCode");
	    String nsCode  = req.getHeader("nsCode");

        if(StringUtils.isNotBlank(envCode) && StringUtils.isNotBlank(nsCode)){
        	OccSecretKeyDTO dto = new OccSecretKeyDTO();
        	dto.setEnvCode(envCode);
        	dto.setNsCode(nsCode);
        	IOccSecretKeyManage occSecretKeyManage = SpringContextUtil.getBean(IOccSecretKeyManage.class);
        	try {
        		dto = occSecretKeyManage.selectOneSecretkey(dto);
			} catch (Exception e) {
				logger.error(e.getMessage(), e);
			}

			if(dto == null){
				chain.doFilter(request, response);
				return;
			}

			//签名校验 add by jdkleo @ 2017-03-21 11:06
			ServletRequest requestWrapper = new BodyReaderHttpServletRequestWrapper(req);
			String body = HttpHelper.getBodyString(requestWrapper);
			String signature = req.getHeader("signature");//签名
			String signatureString = SecretUtil.hmac256(body, dto.getSecretKey());//按原文进行签名计算
			if (signatureString.equals(signature)){//比较服务器签名和客户端签名
				chain.doFilter(requestWrapper, response);
				return;
			}
        }
	}

	@Override
	public void destroy() {
		
	}

}

 

 

BodyReaderHttpServletRequestWrapper.java

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.util.Enumeration;

import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;


public class BodyReaderHttpServletRequestWrapper extends HttpServletRequestWrapper {

    private final byte[] body;

    public BodyReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        body = HttpHelper.getBodyString(request).getBytes(Charset.forName("UTF-8"));
    }

    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(getInputStream()));
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {

        final ByteArrayInputStream bais = new ByteArrayInputStream(body);

        return new ServletInputStream() {
            @Override
            public int read() throws IOException {
                return bais.read();
            }
        };
    }

    @Override
    public String getHeader(String name) {
        return super.getHeader(name);
    }

    @Override
    public Enumeration<String> getHeaderNames() {
        return super.getHeaderNames();
    }

    @Override
    public Enumeration<String> getHeaders(String name) {
        return super.getHeaders(name);
    }

}

 


HttpHelper.java

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;

import javax.servlet.ServletRequest;

public class HttpHelper {
    /**
     * 获取请求Body
     *
     * @param request
     * @return
     */
    public static String getBodyString(ServletRequest request) {
        StringBuilder sb = new StringBuilder();
        InputStream inputStream = null;
        BufferedReader reader = null;
        try {
            inputStream = request.getInputStream();
            reader = new BufferedReader(new InputStreamReader(inputStream, Charset.forName("UTF-8")));
            String line = "";
            while ((line = reader.readLine()) != null) {
                sb.append(line);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (inputStream != null) {
                try {
                    inputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (reader != null) {
                try {
                    reader.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return sb.toString();
    }
}

 

 

分享到：
| 简单发布tomcat带日志输出
评论
发表评论

您还没有登录,请您登录后再发表评论

相关推荐

Magicbox
Global site tag (gtag.js) - Google Analytics