package com.terra.proxy.servlet;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.alibaba.fastjson.serializer.SerializerFeature;
|
import com.terra.proxy.bean.TokenCheckException;
|
import com.terra.proxy.bean.VistorBean;
|
import com.terra.proxy.mapper.ProxyServletMapper;
|
import com.terra.proxy.proxy.SolrProxyServletConfiguration;
|
import com.terra.proxy.util.CrpUtil;
|
import com.terra.proxy.util.HttpContextUtils;
|
import com.terra.proxy.util.JedisUtils;
|
import org.apache.http.*;
|
import org.apache.http.client.utils.URIUtils;
|
import org.apache.http.entity.BasicHttpEntity;
|
import org.mitre.dsmiley.httpproxy.ProxyServlet;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.web.context.WebApplicationContext;
|
import org.springframework.web.context.support.WebApplicationContextUtils;
|
import redis.clients.jedis.Jedis;
|
|
import javax.annotation.PostConstruct;
|
import javax.servlet.ServletException;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.net.URI;
|
import java.util.*;
|
import java.util.regex.Pattern;
|
import java.util.zip.GZIPInputStream;
|
import java.util.zip.GZIPOutputStream;
|
|
public class SubServlet extends ProxyServlet {
|
private Logger log = LoggerFactory.getLogger(SubServlet.class);
|
@Autowired
|
JedisUtils jedisUtils;
|
|
@Value("${fromsys}")
|
private String fromsys;
|
|
ProxyServletMapper service;
|
|
|
@PostConstruct
|
public void init() {
|
try {
|
super.init();
|
|
} catch (ServletException e) {
|
e.printStackTrace();
|
}
|
WebApplicationContext wac = WebApplicationContextUtils.getRequiredWebApplicationContext(getServletContext());
|
service = wac.getBean(ProxyServletMapper.class);
|
}
|
|
public static boolean isdigit(String str) {
|
|
return str.matches("[0-9]+");
|
}
|
|
protected String rewriteUrlFromRequest(HttpServletRequest servletRequest) {
|
StringBuilder uri = new StringBuilder(1000);
|
String fragment = null;
|
StringBuilder queryString2 = new StringBuilder(1000);
|
if (servletRequest.getPathInfo() != null) {// ex: /my/path.html
|
System.out.println(servletRequest.getPathInfo());
|
uri.append(encodeUriQuery(servletRequest.getPathInfo()));
|
}
|
URI errorurl = URI.create(
|
"http://" + "localhost" + ":" + servletRequest.getServerPort() + "/404.html");
|
servletRequest.setAttribute(ATTR_TARGET_HOST, URIUtils.extractHost(errorurl));
|
String queryString = servletRequest.getQueryString();// ex:(following//
|
// name=value&foo=bar#fragment
|
|
List<Map> serverlist = new ArrayList<>();
|
Map<String, Object> param = new HashMap();
|
String code = servletRequest.getPathInfo().split("/")[1];
|
Integer id = CrpUtil.decode(code);
|
|
param.put("id", id);
|
param.put("enable", 1);
|
serverlist = service.queryServer(param);
|
param.put("URL", serverlist.get(0).get("URL").toString());
|
serverlist.add(param);
|
|
String url = serverlist.get(0).get("URL").toString();
|
Integer ispublic = Integer.valueOf(serverlist.get(0).get("ISPUBLIC").toString());
|
VistorBean vb = new VistorBean(servletRequest.getRemoteAddr(), url, id, fromsys);
|
URI hostpath = URI.create(url);
|
servletRequest.setAttribute(ATTR_TARGET_HOST, URIUtils.extractHost(hostpath));
|
checkToken(queryString, queryString2, ispublic, vb, fragment);
|
|
if (!url.isEmpty()) {
|
StringBuilder returnuri = new StringBuilder(url);
|
String path = servletRequest.getPathInfo().substring(servletRequest.getPathInfo().indexOf(code.toString()) + code.toString().length() + 1);
|
returnuri.append(path);
|
if (queryString2 != null && queryString2.length() > 0) {
|
if (!(returnuri.indexOf("?") > -1)) {
|
returnuri.append('?');
|
}
|
|
returnuri.append(encodeUriQuery(queryString2));
|
}
|
|
if (doSendUrlFragment && fragment != null) {
|
returnuri.append('#');
|
returnuri.append(encodeUriQuery(fragment));
|
}
|
|
vb.setResult("success");
|
visitlogsaveinredis(vb);
|
return returnuri.toString();
|
}
|
return null;
|
|
}
|
|
|
protected void copyRequestHeaders(HttpServletRequest servletRequest, HttpRequest proxyRequest) {
|
Enumeration enumerationOfHeaderNames = servletRequest.getHeaderNames();
|
while (enumerationOfHeaderNames.hasMoreElements()) {
|
String headerName = (String) enumerationOfHeaderNames.nextElement();
|
if ((!headerName.equalsIgnoreCase("Content-Length")) &&
|
|
(!hopByHopHeaders.containsHeader(headerName))) {
|
Enumeration headers = servletRequest.getHeaders(headerName);
|
while (headers.hasMoreElements()) {
|
String headerValue = (String) headers.nextElement();
|
if (headerName.equalsIgnoreCase("Host")) {
|
HttpHost host = getTargetHost(servletRequest);
|
headerValue = host.getHostName();
|
if (host.getPort() != -1) {
|
headerValue = headerValue + ":" + host.getPort();
|
}
|
} else if (headerName.equalsIgnoreCase("Cookie")) {
|
headerValue = getRealCookie(headerValue);
|
}
|
proxyRequest.addHeader(headerName, headerValue);
|
}
|
}
|
}
|
}
|
|
|
private HttpHost getTargetHost(HttpServletRequest servletRequest) {
|
return (HttpHost) servletRequest.getAttribute(ATTR_TARGET_HOST);
|
}
|
|
public static boolean isInteger(String str) {
|
Pattern pattern = Pattern.compile("^[1-9]\\d*|0$");
|
return pattern.matcher(str).matches();
|
|
}
|
|
/**
|
* 验证token
|
*
|
* @param queryString
|
* @param ispublic
|
* @param vb
|
*/
|
public void checkToken(String queryString, StringBuilder queryString2, Integer ispublic, VistorBean vb, String fragment) {
|
// split off fragment from queryString, updating queryString if found
|
if (queryString != null) {
|
// 针对请求地址的参数做操作
|
// 判断是否带了token
|
if (queryString.indexOf("token") == -1 && ispublic != 1) {
|
|
vb.setResult("fail");
|
visitlogsaveinredis(vb);
|
throw new TokenCheckException("未传入token", "500");
|
}
|
String par[] = queryString.split("&");
|
for (int i = 0; i < par.length; i++) {
|
String a = par[i].split("=")[0];
|
String v = par[i].split("=").length >= 2 ? par[i].split("=")[1] : "";
|
if ("token".equals(a)) {
|
// 判断内存里有咩有匹配的token,咩有去redis库里查询，库里没有，则在内存和库中存储，库里有，存入内存
|
if (!SolrProxyServletConfiguration.tokens.contains(v)) {
|
Jedis jedis = null;
|
try {
|
jedis = JedisUtils.getJedis();
|
if (!jedis.exists(v)) {
|
vb.setResult("fail");
|
visitlogsaveinredis(vb);
|
throw new TokenCheckException("token不存在", "500");
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
} finally {
|
jedis.close();
|
}
|
|
}
|
|
} else {
|
|
queryString2.append("&").append(par[i]);
|
|
}
|
|
}
|
|
int fragIdx = queryString.indexOf('#');
|
if (fragIdx >= 0) {
|
fragment = queryString.substring(fragIdx + 1);
|
queryString = queryString.substring(0, fragIdx);
|
}
|
} else {
|
if (ispublic != 1) {
|
vb.setResult("fail");
|
visitlogsaveinredis(vb);
|
throw new TokenCheckException("未传入token", "500");
|
}
|
}
|
}
|
|
/**
|
* 统计不同ip调用服务次数,并将其存入入redis
|
*
|
* @param vb 日志bean
|
*/
|
private void visitlogsaveinredis(VistorBean vb) {
|
String fromip = vb.getRequestip();
|
vb.setDate(new Date());
|
Jedis jedis = null;
|
int count = 0;
|
try {
|
jedis = JedisUtils.getJedis();
|
jedis.sadd("visitlog",
|
JSONObject.toJSONStringWithDateFormat(vb, "yyyy-MM-dd HH:mm:ss", SerializerFeature.PrettyFormat));
|
} catch (Exception e) {
|
// TODO: handle exception
|
log.error("redis包错");
|
} finally {
|
jedis.close();
|
}
|
|
}
|
|
|
protected void copyResponseHeaders(HttpResponse proxyResponse, HttpServletRequest servletRequest, HttpServletResponse servletResponse) {
|
for (Header header : proxyResponse.getAllHeaders()) {
|
if (!hopByHopHeaders.containsHeader(header.getName())) {
|
if ((header.getName().equalsIgnoreCase("Set-Cookie")) || (header.getName().equalsIgnoreCase("Set-Cookie2"))) {
|
copyProxyCookie(servletRequest, servletResponse, header);
|
} else {
|
servletResponse.addHeader(header.getName(), header.getValue());
|
System.out.print(header.getName() + header.getValue());
|
}
|
}
|
}
|
System.out.print(servletResponse.containsHeader("Content-Length"));
|
}
|
|
public boolean isgzip(HttpEntity entity) {
|
return entity.getContentEncoding() != null
|
&& "Content-Encoding: gzip".equals(entity.getContentEncoding().toString()) ? true : false;
|
}
|
|
public static byte[] compress(String str) {
|
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
|
GZIPOutputStream gzip = null;
|
try {
|
gzip = new GZIPOutputStream(out);
|
gzip.write(str.getBytes());
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
if (gzip != null) {
|
try {
|
gzip.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
return out.toByteArray();
|
}
|
|
|
protected void copyResponseEntity(HttpResponse proxyResponse, HttpServletResponse servletResponse)
|
throws IOException {
|
HttpEntity entity = proxyResponse.getEntity();
|
BufferedReader reader;
|
InputStream is = entity.getContent();
|
servletResponse.setCharacterEncoding("UTF-8");
|
|
if (isgzip(entity)) {
|
reader = new BufferedReader(new InputStreamReader(new GZIPInputStream(is)));
|
StringBuffer sb = new StringBuffer();
|
String s = null;
|
while ((s = reader.readLine()) != null) {
|
|
sb.append(s);
|
}
|
String result = sb.toString();
|
|
String repx = "<a[^>]+>([^<]+</a>)";//a标签过滤
|
result = result.replaceAll(repx, "");
|
result = result.replaceAll("(http|https)://[^/]+/", "/");
|
|
HttpServletRequest hsr = HttpContextUtils.getHttpServletRequest();
|
|
BasicHttpEntity entity2 = new BasicHttpEntity();
|
byte[] bt;
|
|
bt = compress(result);
|
|
ByteArrayInputStream btinput = new ByteArrayInputStream(bt);
|
entity2.setContent(btinput);
|
entity2.setContentEncoding("gzip");
|
entity2.setContentType(entity.getContentType());
|
entity2.setChunked(false);
|
|
|
if (entity2 != null) {
|
servletResponse.addHeader("Content-Length", String.valueOf(bt.length));
|
OutputStream servletOutputStream = servletResponse.getOutputStream();
|
entity2.writeTo(servletOutputStream);
|
}
|
|
} else {
|
|
if (entity != null && "application/xml; charset=utf-8".equals(entity.getContentType().getValue())) {
|
reader = new BufferedReader(new InputStreamReader(is));
|
StringBuffer sb = new StringBuffer();
|
String s = null;
|
while ((s = reader.readLine()) != null) {
|
|
sb.append(s);
|
}
|
String result = sb.toString();
|
|
|
result = result.replaceAll("(http|https)://[^/]+/", "/");
|
BasicHttpEntity entity2 = new BasicHttpEntity();
|
byte[] bt;
|
|
bt = result.getBytes();
|
|
ByteArrayInputStream btinput = new ByteArrayInputStream(bt);
|
entity2.setContent(btinput);
|
entity2.setContentType(entity.getContentType());
|
entity2.setChunked(false);
|
servletResponse.addHeader("Content-Length", String.valueOf(bt.length));
|
OutputStream servletOutputStream = servletResponse.getOutputStream();
|
entity2.writeTo(servletOutputStream);
|
} else {
|
if (entity != null) {
|
|
OutputStream servletOutputStream = servletResponse.getOutputStream();
|
entity.writeTo(servletOutputStream);
|
}
|
}
|
}
|
}
|
|
}
|
