纯真IP地址查询（Java版源代码）

package edu.pku.ipseeker; /* * LumaQQ - Java QQ Client * * Copyright (C) 2004 luma <stubma@163.com> * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ import java.io.File; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.io.RandomAccessFile; import java.nio.ByteOrder; import java.nio.MappedByteBuffer; import java.nio.channels.FileChannel; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Random; /** * <pre> * 关于IP数据库文件格式，请参考LumaQQ主页文档“纯真IP数据库格式详解”一文。 * </pre> * * @author luma */ public class IPSeeker { /** * <pre> * 用来封装ip相关信息，目前只有两个字段，ip所在的国家和地区 * </pre> * * @author luma */ private class IPLocation { public String country; public String area; public IPLocation() { country = area = ""; } public IPLocation getCopy() { IPLocation ret = new IPLocation(); ret.country = country; ret.area = area; return ret; } } // 一些固定常量，比如记录长度等等 private static final int IP_RECORD_LENGTH = 7; private static final byte REDIRECT_MODE_1 = 0x01; private static final byte REDIRECT_MODE_2 = 0x02; // 用来做为cache，查询一个ip时首先查看cache，以减少不必要的重复查找 private Map<String, IPLocation> ipCache; // 随机文件访问类 private RandomAccessFile ipFile; // 内存映射文件 private MappedByteBuffer mbb; // 起始地区的开始和结束的绝对偏移 private long ipBegin, ipEnd; // 为提高效率而采用的临时变量 private IPLocation loc; private byte[] buf; private byte[] b4; private byte[] b3; private static Random random = new Random(System.currentTimeMillis()); private boolean extractResource(File tempFile) { boolean re = false; InputStream input = null; FileOutputStream output = null; try { input = IPSeeker.class.getResourceAsStream("QQWry.Dat"); output = new FileOutputStream(tempFile); byte[] buff = new byte[1024]; int len; while ((len = input.read(buff)) >= 0) { output.write(buff, 0, len); } re = true; } catch (IOException e) { } finally { if (input != null) { try { input.close(); } catch (IOException e) { re = false; } } if (output != null) { try { output.close(); } catch (IOException e) { re = false; } } } return re; } /** * 私有构造函数 */ public IPSeeker() { ipCache = new HashMap<String, IPLocation>(); loc = new IPLocation(); buf = new byte[100]; b4 = new byte[4]; b3 = new byte[3]; try { File tempFile = File.createTempFile(Long.toString(System .currentTimeMillis() + random.nextLong()), null); tempFile.deleteOnExit(); if (extractResource(tempFile)) { ipFile = new RandomAccessFile(tempFile, "r"); } } catch (FileNotFoundException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); } // 如果打开文件成功，读取文件头信息 if (ipFile != null) { try { ipBegin = readLong4(0); ipEnd = readLong4(4); if (ipBegin == -1 || ipEnd == -1) { ipFile.close(); ipFile = null; } } catch (IOException e) { ipFile = null; } } } /** * 给定一个地点的不完全名字，得到一系列包含s子串的IP范围记录 * * @param s * 地点子串 * @return 包含IPEntry类型的List */ public List<IPEntry> getIPEntriesDebug(String s) { List<IPEntry> ret = new ArrayList<IPEntry>(); long endOffset = ipEnd + 4; for (long offset = ipBegin + 4; offset <= endOffset; offset += IP_RECORD_LENGTH) { // 读取结束IP偏移 long temp = readLong3(offset); // 如果temp不等于-1，读取IP的地点信息 if (temp != -1) { IPLocation ipLoc = getIPLocation(temp); // 判断是否这个地点里面包含了s子串，如果包含了，添加这个记录到List中，如果没有，继续 if (ipLoc.country.indexOf(s) != -1 || ipLoc.area.indexOf(s) != -1) { IPEntry entry = new IPEntry(); entry.country = ipLoc.country; entry.area = ipLoc.area; // 得到起始IP readIP(offset - 4, b4); entry.beginIp = Util.getIpStringFromBytes(b4); // 得到结束IP readIP(temp, b4); entry.endIp = Util.getIpStringFromBytes(b4); // 添加该记录 ret.add(entry); } } } return ret; } /** * 给定一个地点的不完全名字，得到一系列包含s子串的IP范围记录 * * @param s * 地点子串 * @return 包含IPEntry类型的List */ public List<IPEntry> getIPEntries(String s) { List<IPEntry> ret = new ArrayList<IPEntry>(); try { // 映射IP信息文件到内存中 if (mbb == null) { FileChannel fc = ipFile.getChannel(); mbb = fc.map(FileChannel.MapMode.READ_ONLY, 0, ipFile.length()); mbb.order(ByteOrder.LITTLE_ENDIAN); } int endOffset = (int) ipEnd; for (int offset = (int) ipBegin + 4; offset <= endOffset; offset += IP_RECORD_LENGTH) { int temp = readInt3(offset); if (temp != -1) { IPLocation ipLoc = getIPLocation(temp); // 判断是否这个地点里面包含了s子串，如果包含了，添加这个记录到List中，如果没有，继续 if (ipLoc.country.indexOf(s) != -1 || ipLoc.area.indexOf(s) != -1) { IPEntry entry = new IPEntry(); entry.country = ipLoc.country; entry.area = ipLoc.area; // 得到起始IP readIP(offset - 4, b4); entry.beginIp = Util.getIpStringFromBytes(b4); // 得到结束IP readIP(temp, b4); entry.endIp = Util.getIpStringFromBytes(b4); // 添加该记录 ret.add(entry); } } } } catch (IOException e) { e.printStackTrace(); } return ret; } /** * 从内存映射文件的offset位置开始的3个字节读取一个int * * @param offset * @return */ private int readInt3(int offset) { mbb.position(offset); return mbb.getInt() & 0x00FFFFFF; } /** * 从内存映射文件的当前位置开始的3个字节读取一个int * * @return */ private int readInt3() { return mbb.getInt() & 0x00FFFFFF; } /** * 根据IP得到国家名 * * @param ip * ip的字节数组形式 * @return 国家名字符串 */ public String getCountry(byte[] ip) { // 检查ip地址文件是否正常 if (ipFile == null) return "非法的IP文件"; // 保存ip，转换ip字节数组为字符串形式 String ipStr = Util.getIpStringFromBytes(ip); // 先检查cache中是否已经包含有这个ip的结果，没有再搜索文件 if (ipCache.containsKey(ipStr)) { IPLocation ipLoc = ipCache.get(ipStr); return ipLoc.country; } else { IPLocation ipLoc = getIPLocation(ip); ipCache.put(ipStr, ipLoc.getCopy()); return ipLoc.country; } } /** * 根据IP得到国家�