Servlet Filter

WhiteList.java

package com.toowards.sbr;

import com.google.common.base.Charsets;
import com.google.common.io.Files;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.List;

/**
 * White list class
 * User: Jason@toowards
 * Date: 06/09/2013
 */
public class WhiteList {
    private static String WHITE_LIST_FILENAME;

    private List<String> whiteList;   // white list data
    private static WhiteList instance;
    private static Object lock = new Object();

    public static WhiteList getInstance() {
    	try
    	{
    		// finds resource relative to the class location
		URL url = new WhiteList().getClass().getResource("whitelist.txt");
    		WHITE_LIST_FILENAME= url.getPath();
    	}catch(Exception ioe)
    	{
    		ioe.printStackTrace();
    	}
        if (instance == null) {
            synchronized (lock) {
                if (instance == null) {
                    instance = new WhiteList();
                    File file = new File(WHITE_LIST_FILENAME);
                    try {
                        instance.whiteList = Files.readLines(file, Charsets.UTF_8);
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
        return instance;
    }

    public List<String> getWhiteList() {
        return whiteList;
    }
}

WhiteListFilter

package com.toowards.sbr;

import com.google.common.io.Files;

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.IOException;
import java.util.List;

/**
 * White list class
 * User: Jason@toowards
 * Date: 09/09/2013
 */
public class WhiteListFilter implements Filter {
    private static String IP = "ip";

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
    }

    /**
     * whitelist filter
     *
     * @param servletRequest
     * @param servletResponse
     * @param filterChain
     * @throws IOException
     * @throws ServletException
     */
    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        HttpServletRequest req = (HttpServletRequest) servletRequest;
        HttpServletResponse resp = (HttpServletResponse) servletResponse;
        String ip = req.getRemoteAddr();
        WhiteList whiteList = WhiteList.getInstance();
        if (!whiteList.getWhiteList().contains(ip)) {  // Surely we can do preg match to validate ip based on pattern here
            resp.setCharacterEncoding("UTF-8");
            resp.getWriter().println("IP restricted");
        } else {
            resp.setCharacterEncoding("UTF-8");
            filterChain.doFilter(req, resp);
        }
    }

    @Override
    public void destroy() {
    }
}


whitelist.txt

192.168.1.45
192.168.1.47


web.xml

    <!-- WhiteList filter config -->
    <filter>
        <filter-name>WhiteListFilter</filter-name>
        <filter-class>com.toowards.sbr.WhiteListFilter</filter-class>
    </filter>
    <filter-mapping>
        <filter-name>WhiteListFilter</filter-name>
        <url-pattern>/*</url-pattern>
    </filter-mapping>
    <!-- Put this before any servlet to restrict ip first -->






  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值