/* * Created on 2004-8-4 * */ package com.bupticet.ip; import java.io.UnsupportedEncodingException; /** * @author LJ-silver */ public class Utils { /** * 從ip的字符串形式獲得字節數組形式 * @param ip 字符串形式的ip * @return 字節數組形式的ip */ public static byte[] getIpByteArrayFromString(String ip) { byte[] ret = new byte[4]; java.util.StringTokenizer st = new java.util.StringTokenizer(ip, "."); try { ret[0] = (byte)(Integer.parseInt(st.nextToken()) & 0xFF); ret[1] = (byte)(Integer.parseInt(st.nextToken()) & 0xFF); ret[2] = (byte)(Integer.parseInt(st.nextToken()) & 0xFF); ret[3] = (byte)(Integer.parseInt(st.nextToken()) & 0xFF); } catch (Exception e) { System.out.println(e.getMessage()); } return ret; } /** * 對原始字符串進行編碼轉換,若是失敗,返回原始的字符串 * @param s 原始字符串 * @param srcEncoding 源編碼方式 * @param destEncoding 目標編碼方式 * @return 轉換編碼後的字符串,失敗返回原始字符串 */ public static String getString(String s, String srcEncoding, String destEncoding) { try { return new String(s.getBytes(srcEncoding), destEncoding); } catch (UnsupportedEncodingException e) { return s; } } /** * 根據某種編碼方式將字節數組轉換成字符串 * @param b 字節數組 * @param encoding 編碼方式 * @return 若是encoding不支持,返回一個缺省編碼的字符串 */ public static String getString(byte[] b, String encoding) { try { return new String(b, encoding); } catch (UnsupportedEncodingException e) { return new String(b); } } /** * 根據某種編碼方式將字節數組轉換成字符串 * @param b 字節數組 * @param offset 要轉換的起始位置 * @param len 要轉換的長度 * @param encoding 編碼方式 * @return 若是encoding不支持,返回一個缺省編碼的字符串 */ public static String getString(byte[] b, int offset, int len, String encoding) { try { return new String(b, offset, len, encoding); } catch (UnsupportedEncodingException e) { return new String(b, offset, len); } } /** * @param ip ip的字節數組形式 * @return 字符串形式的ip */ public static String getIpStringFromBytes(byte[] ip) { StringBuffer sb = new StringBuffer(); sb.append(ip[0] & 0xFF); sb.append('.'); sb.append(ip[1] & 0xFF); sb.append('.'); sb.append(ip[2] & 0xFF); sb.append('.'); sb.append(ip[3] & 0xFF); return sb.toString(); } }
代碼:
|
|
![]() |
/* * LumaQQ - Java QQ Client * * Copyright (C) 2004 luma <[email]stubma@163.com[/email]> * * 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 */ package com.bupticet.ip; import java.io.FileNotFoundException; import java.io.IOException; import java.io.RandomAccessFile; import java.nio.ByteOrder; import java.nio.MappedByteBuffer; import java.nio.channels.FileChannel; import java.util.ArrayList; import java.util.Hashtable; import java.util.List; /** * <pre> * 用來讀取QQwry.dat文件,以根據ip得到好友位置,QQwry.dat的格式是 * 一. 文件頭,共8字節 * 1. 第一個起始IP的絕對偏移, 4字節 * 2. 最後一個起始IP的絕對偏移, 4字節 * 二. "結束地址/國家/區域"記錄區 * 四字節ip地址後跟的每一條記錄分紅兩個部分 * 1. 國家記錄 * 2. 地區記錄 * 可是地區記錄是不必定有的。並且國家記錄和地區記錄都有兩種形式 * 1. 以0結束的字符串 * 2. 4個字節,一個字節可能爲0x1或0x2 * a. 爲0x1時,表示在絕對偏移後還跟着一個區域的記錄,注意是絕對偏移以後,而不是這四個字節以後 * b. 爲0x2時,表示在絕對偏移後沒有區域記錄 * 無論爲0x1仍是0x2,後三個字節都是實際國家名的文件內絕對偏移 * 若是是地區記錄,0x1和0x2的含義不明,可是若是出現這兩個字節,也確定是跟着3個字節偏移,若是不是 * 則爲0結尾字符串 * 三. "起始地址/結束地址偏移"記錄區 * 1. 每條記錄7字節,按照起始地址從小到大排列 * a. 起始IP地址,4字節 * b. 結束ip地址的絕對偏移,3字節 * * 注意,這個文件裏的ip地址和全部的偏移量均採用little-endian格式,而java是採用 * big-endian格式的,要注意轉換 * </pre> * * @author 馬若劼 */ public class IPSeeker { /** * <pre> * 用來封裝ip相關信息,目前只有兩個字段,ip所在的國家和地區 * </pre> * * @author 馬若劼 */ 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 String IP_FILE = IPSeeker.class.getResource("/QQWry.dat").toString().substring(5); // 一些固定常量,好比記錄長度等等 private static final int IP_RECORD_LENGTH = 7; private static final byte AREA_FOLLOWED = 0x01; private static final byte NO_AREA = 0x2; // 用來作爲cache,查詢一個ip時首先查看cache,以減小沒必要要的重複查找 private Hashtable ipCache; // 隨機文件訪問類 private RandomAccessFile ipFile; // 內存映射文件 private MappedByteBuffer mbb; // 單一模式實例 private static IPSeeker instance = new IPSeeker(); // 起始地區的開始和結束的絕對偏移 private long ipBegin, ipEnd; // 爲提升效率而採用的臨時變量 private IPLocation loc; private byte[] buf; private byte[] b4; private byte[] b3; /** * 私有構造函數 */ private IPSeeker() { ipCache = new Hashtable(); loc = new IPLocation(); buf = new byte[100]; b4 = new byte[4]; b3 = new byte[3]; try { ipFile = new RandomAccessFile(IP_FILE, "r"); } catch (FileNotFoundException e) { System.out.println(IPSeeker.class.getResource("/QQWry.dat").toString()); System.out.println(IP_FILE); System.out.println("IP地址信息文件沒有找到,IP顯示功能將沒法使用"); ipFile = null; } // 若是打開文件成功,讀取文件頭信息 if(ipFile != null) { try { ipBegin = readLong4(0); ipEnd = readLong4(4); if(ipBegin == -1 || ipEnd == -1) { ipFile.close(); ipFile = null; } } catch (IOException e) { System.out.println("IP地址信息文件格式有錯誤,IP顯示功能將沒法使用"); ipFile = null; } } } /** * @return 單一實例 */ public static IPSeeker getInstance() { return instance; } /** * 給定一個地點的不徹底名字,獲得一系列包含s子串的IP範圍記錄 * @param s 地點子串 * @return 包含IPEntry類型的List */ public List getIPEntriesDebug(String s) { List ret = new ArrayList(); 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 loc = getIPLocation(temp); // 判斷是否這個地點裏麪包含了s子串,若是包含了,添加這個記錄到List中,若是沒有,繼續 if(loc.country.indexOf(s) != -1 || loc.area.indexOf(s) != -1) { IPEntry entry = new IPEntry(); entry.country = loc.country; entry.area = loc.area; // 獲得起始IP readIP(offset - 4, b4); entry.beginIp = Utils.getIpStringFromBytes(b4); // 獲得結束IP readIP(temp, b4); entry.endIp = Utils.getIpStringFromBytes(b4); // 添加該記錄 ret.add(entry); } } } return ret; } /** * 給定一個地點的不徹底名字,獲得一系列包含s子串的IP範圍記錄 * @param s 地點子串 * @return 包含IPEntry類型的List */ public List getIPEntries(String s) { List ret = new ArrayList(); 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 loc = getIPLocation(temp); // 判斷是否這個地點裏麪包含了s子串,若是包含了,添加這個記錄到List中,若是沒有,繼續 if(loc.country.indexOf(s) != -1 || loc.area.indexOf(s) != -1) { IPEntry entry = new IPEntry(); entry.country = loc.country; entry.area = loc.area; // 獲得起始IP readIP(offset - 4, b4); entry.beginIp = Utils.getIpStringFromBytes(b4); // 獲得結束IP readIP(temp, b4); entry.endIp = Utils.getIpStringFromBytes(b4); // 添加該記錄 ret.add(entry); } } } } catch (IOException e) { System.out.println(e.getMessage()); } 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 = Utils.getIpStringFromBytes(ip); // 先檢查cache中是否已經包含有這個ip的結果,沒有再搜索文件 if(ipCache.containsKey(ipStr)) { IPLocation loc = (IPLocation)ipCache.get(ipStr); return loc.country; } else { IPLocation loc = getIPLocation(ip); ipCache.put(ipStr, loc.getCopy()); return loc.country; } } /** * 根據IP獲得國家名 * @param ip IP的字符串形式 * @return 國家名字符串 */ public String getCountry(String ip) { return getCountry(Utils.getIpByteArrayFromString(ip)); } /** * 根據IP獲得地區名 * @param ip ip的字節數組形式 * @return 地區名字符串 */ public String getArea(byte[] ip) { // 檢查ip地址文件是否正常 if(ipFile == null) return "錯誤的IP數據庫文件"; // 保存ip,轉換ip字節數組爲字符串形式 String ipStr = Utils.getIpStringFromBytes(ip); // 先檢查cache中是否已經包含有這個ip的結果,沒有再搜索文件 if(ipCache.containsKey(ipStr)) { IPLocation loc = (IPLocation)ipCache.get(ipStr); return loc.area; } else { IPLocation loc = getIPLocation(ip); ipCache.put(ipStr, loc.getCopy()); return loc.area; } } /** * 根據IP獲得地區名 * @param ip IP的字符串形式 * @return 地區名字符串 */ public String getArea(String ip) { return getArea(Utils.getIpByteArrayFromString(ip)); } /** * 根據ip搜索ip信息文件,獲得IPLocation結構,所搜索的ip參數從類成員ip中獲得 * @param ip 要查詢的IP * @return IPLocation結構 */ private IPLocation getIPLocation(byte[] ip) { IPLocation info = null; long offset = locateIP(ip); if(offset != -1) info = getIPLocation(offset); if(info == null) { info = new IPLocation(); info.country = "未知國家"; info.area = "未知地區"; } return info; } /** * 從offset位置讀取4個字節爲一個long,由於java爲big-endian格式,因此沒辦法 * 用了這麼一個函數來作轉換 * @param offset * @return 讀取的long值,返回-1表示讀取文件失敗 */ private long readLong4(long offset) { long ret = 0; try { ipFile.seek(offset); ret |= (ipFile.readByte() & 0xFF); ret |= ((ipFile.readByte() << 8) & 0xFF00); ret |= ((ipFile.readByte() << 16) & 0xFF0000); ret |= ((ipFile.readByte() << 24) & 0xFF000000); return ret; } catch (IOException e) { return -1; } } /** * 從offset位置讀取3個字節爲一個long,由於java爲big-endian格式,因此沒辦法 * 用了這麼一個函數來作轉換 * @param offset * @return 讀取的long值,返回-1表示讀取文件失敗 */ private long readLong3(long offset) { long ret = 0; try { ipFile.seek(offset); ipFile.readFully(b3); ret |= (b3[0] & 0xFF); ret |= ((b3[1] << 8) & 0xFF00); ret |= ((b3[2] << 16) & 0xFF0000); return ret; } catch (IOException e) { return -1; } } /** * 從當前位置讀取3個字節轉換成long * @return */ private long readLong3() { long ret = 0; try { ipFile.readFully(b3); ret |= (b3[0] & 0xFF); ret |= ((b3[1] << 8) & 0xFF00); ret |= ((b3[2] << 16) & 0xFF0000); return ret; } catch (IOException e) { return -1; } } /** * 從offset位置讀取四個字節的ip地址放入ip數組中,讀取後的ip爲big-endian格式,可是 * 文件中是little-endian形式,將會進行轉換 * @param offset * @param ip */ private void readIP(long offset, byte[] ip) { try { ipFile.seek(offset); ipFile.readFully(ip); byte temp = ip[0]; ip[0] = ip[3]; ip[3] = temp; temp = ip[1]; ip[1] = ip[2]; ip[2] = temp; } catch (IOException e) { System.out.println(e.getMessage()); } } /** * 從offset位置讀取四個字節的ip地址放入ip數組中,讀取後的ip爲big-endian格式,可是 * 文件中是little-endian形式,將會進行轉換 * @param offset * @param ip */ private void readIP(int offset, byte[] ip) { mbb.position(offset); mbb.get(ip); byte temp = ip[0]; ip[0] = ip[3]; ip[3] = temp; temp = ip[1]; ip[1] = ip[2]; ip[2] = temp; } /** * 把類成員ip和beginIp比較,注意這個beginIp是big-endian的 * @param ip 要查詢的IP * @param beginIp 和被查詢IP相比較的IP * @return 相等返回0,ip大於beginIp則返回1,小於返回-1。 */ private int compareIP(byte[] ip, byte[] beginIp) { for(int i = 0; i < 4; i++) { int r = compareByte(ip[i], beginIp[i]); if(r != 0) return r; } return 0; } /** * 把兩個byte看成無符號數進行比較 * @param b1 * @param b2 * @return 若b1大於b2則返回1,相等返回0,小於返回-1 */ private int compareByte(byte b1, byte b2) { if((b1 & 0xFF) > (b2 & 0xFF)) // 比較是否大於 return 1; else if((b1 ^ b2) == 0)// 判斷是否相等 return 0; else return -1; } /** * 這個方法將根據ip的內容,定位到包含這個ip國家地區的記錄處,返回一個絕對偏移 * 方法使用二分法查找。 * @param ip 要查詢的IP * @return 若是找到了,返回結束IP的偏移,若是沒有找到,返回-1 */ private long locateIP(byte[] ip) { long m = 0; int r; // 比較第一個ip項 readIP(ipBegin, b4); r = compareIP(ip, b4); if(r == 0) return ipBegin; else if(r < 0) return -1; // 開始二分搜索 for(long i = ipBegin, j = ipEnd; i < j; ) { m = getMiddleOffset(i, j); readIP(m, b4); r = compareIP(ip, b4); // log.debug(Utils.getIpStringFromBytes(b)); if(r > 0) i = m; else if(r < 0) { if(m == j) { j -= IP_RECORD_LENGTH; m = j; } else j = m; } else return readLong3(m + 4); } // 若是循環結束了,那麼i和j一定是相等的,這個記錄爲最可能的記錄,可是並不是 // 確定就是,還要檢查一下,若是是,就返回結束地址區的絕對偏移 m = readLong3(m + 4); readIP(m, b4); r = compareIP(ip, b4); if(r <= 0) return m; else return -1; } /** * 獲得begin偏移和end偏移中間位置記錄的偏移 * @param begin * @param end * @return */ private long getMiddleOffset(long begin, long end) { long records = (end - begin) / IP_RECORD_LENGTH; records >>= 1; if(records == 0) records = 1; return begin + records * IP_RECORD_LENGTH; } /** * 給定一個ip國家地區記錄的偏移,返回一個IPLocation結構 * @param offset * @return */ private IPLocation getIPLocation(long offset) { try { // 跳過4字節ip ipFile.seek(offset + 4); // 讀取第一個字節判斷是否標誌字節 byte b = ipFile.readByte(); if(b == AREA_FOLLOWED) { // 讀取國家偏移 long countryOffset = readLong3(); // 跳轉至偏移處 ipFile.seek(countryOffset); // 再檢查一次標誌字節,由於這個時候這個地方仍然多是個重定向 b = ipFile.readByte(); if(b == NO_AREA) { loc.country = readString(readLong3()); ipFile.seek(countryOffset + 4); } else loc.country = readString(countryOffset); // 讀取地區標誌 loc.area = readArea(ipFile.getFilePointer()); } else if(b == NO_AREA) { loc.country = readString(readLong3()); loc.area = readArea(offset + 8); } else { loc.country = readString(ipFile.getFilePointer() - 1); loc.area = readArea(ipFile.getFilePointer()); } return loc; } catch (IOException e) { return null; } } /** * @param offset * @return */ private IPLocation getIPLocation(int offset) { // 跳過4字節ip mbb.position(offset + 4); // 讀取第一個字節判斷是否標誌字節 byte b = mbb.get(); if(b == AREA_FOLLOWED) { // 讀取國家偏移 int countryOffset = readInt3(); // 跳轉至偏移處 mbb.position(countryOffset); // 再檢查一次標誌字節,由於這個時候這個地方仍然多是個重定向 b = mbb.get(); if(b == NO_AREA) { loc.country = readString(readInt3()); mbb.position(countryOffset + 4); } else loc.country = readString(countryOffset); // 讀取地區標誌 loc.area = readArea(mbb.position()); } else if(b == NO_AREA) { loc.country = readString(readInt3()); loc.area = readArea(offset + 8); } else { loc.country = readString(mbb.position() - 1); loc.area = readArea(mbb.position()); } return loc; } /** * 從offset偏移開始解析後面的字節,讀出一個地區名 * @param offset * @return 地區名字符串 * @throws IOException */ private String readArea(long offset) throws IOException { ipFile.seek(offset); byte b = ipFile.readByte(); if(b == 0x01 || b == 0x02) { long areaOffset = readLong3(offset + 1); if(areaOffset == 0) return "未知地區"; else return readString(areaOffset); } else return readString(offset); } /** * @param offset * @return */ private String readArea(int offset) { mbb.position(offset); byte b = mbb.get(); if(b == 0x01 || b == 0x02) { int areaOffset = readInt3(); if(areaOffset == 0) return "未知地區"; else return readString(areaOffset); } else return readString(offset); } /** * 從offset偏移處讀取一個以0結束的字符串 * @param offset * @return 讀取的字符串,出錯返回空字符串 */ private String readString(long offset) { try { ipFile.seek(offset); int i; for(i = 0, buf[i] = ipFile.readByte(); buf[i] != 0; buf[++i] = ipFile.readByte()); if(i != 0) return Utils.getString(buf, 0, i, "GBK"); } catch (IOException e) { System.out.println(e.getMessage()); } return ""; } /** * 從內存映射文件的offset位置獲得一個0結尾字符串 * @param offset * @return */ private String readString(int offset) { try { mbb.position(offset); int i; for(i = 0, buf[i] = mbb.get(); buf[i] != 0; buf[++i] = mbb.get()); if(i != 0) return Utils.getString(buf, 0, i, "GBK"); } catch (IllegalArgumentException e) { System.out.println(e.getMessage()); } return ""; } public String getAddress(String ip){ String country = getCountry(ip).equals(" CZ88.NET")?"":getCountry(ip); String area = getArea(ip).equals(" CZ88.NET")?"":getArea(ip); String address = country+" "+area; return address.trim(); } }
/** * @author LJ-silver */ public class Test { public static void main(String[] args) { com.ljsilver.ip2address.IPSeeker seeker = com.ljsilver.ip2address.IPSeeker.getInstance(); if(args.length==2){ if("ip".equals(args[0])){ System.out.println(args[0]+"的所在地址是:"+seeker.getAddress(args[1])); }else if("address".equals(args[0])){ List a = seeker.getIPEntries(args[1]); System.out.println(args[0]+":"); for(int i=0;i<a.size();i++){ System.out.println(a.get(i).toString()); } }else{ System.out.println("usage:java Test ip/address yourIpString/yourAddressString"); } }else{ System.out.println("usage:java Test ip/address yourIpString/yourAddressString");