packagecom.worthtech.app.util.ip;
importjava.io.FileNotFoundException;
importjava.io.IOException;
importjava.io.RandomAccessFile;
importjava.nio.ByteOrder;
importjava.nio.MappedByteBuffer;
importjava.nio.channels.FileChannel;
importjava.util.ArrayList;
importjava.util.Hashtable;
importjava.util.List;
/**
*
* 用来读取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格式的,要注意转换
*
*
* @author 马若劼
*/
publicclassIPSeeker {
/**
*
* 用来封装ip相关信息,目前只有两个字段,ip所在的国家和地区
*
*
* @author 马若劼
*/
privateclassIPLocation {
publicString country;
publicString area;
publicIPLocation() {
country = area = "";
}
publicIPLocation getCopy() {
IPLocation ret = newIPLocation();
ret.country = country;
ret.area = area;
returnret;
}
}
privatestaticfinalString IP_FILE = IPSeeker.class.getResource(
"QQWry.dat").toString().substring(5);
// 一些固定常量,比如记录长度等等
privatestaticfinalintIP_RECORD_LENGTH =7;
privatestaticfinalbyteAREA_FOLLOWED =0x01;
privatestaticfinalbyteNO_AREA =0x2;
// 用来做为cache,查询一个ip时首先查看cache,以减少不必要的重复查找
privatefinalHashtable ipCache;
// 随机文件访问类
privateRandomAccessFile ipFile;
// 内存映射文件
privateMappedByteBuffer mbb;
// 单一模式实例
privatestaticIPSeeker instance =newIPSeeker();
// 起始地区的开始和结束的绝对偏移
privatelongipBegin, ipEnd;
// 为提高效率而采用的临时变量
privatefinalIPLocation loc;
// private final byte[] buf;//不需要了
privatefinalbyte[] b4;
privatefinalbyte[] b3;
/**
* 私有构造函数
*/
privateIPSeeker() {
ipCache = newHashtable();
loc = newIPLocation();
// buf = new byte[100];//不需要了,这里初始化不好!
b4 = newbyte[4];
b3 = newbyte[3];
try{
ipFile = newRandomAccessFile(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 单一实例
*/
publicstaticIPSeeker getInstance() {
returninstance;
}
/**
* 给定一个地点的不完全名字,得到一系列包含s子串的IP范围记录
*
* @param s
* 地点子串
* @return 包含IPEntry类型的List
*/
publicList getIPEntriesDebug(String s) {
List ret = newArrayList();
longendOffset = ipEnd +4;
for(longoffset = ipBegin +4; offset <= endOffset; offset += IP_RECORD_LENGTH) {
// 读取结束IP偏移
longtemp = 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 = newIPEntry();
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);
}
}
}
returnret;
}
/**
* 给定一个地点的不完全名字,得到一系列包含s子串的IP范围记录
*
* @param s
* 地点子串
* @return 包含IPEntry类型的List
*/
publicList getIPEntries(String s) {
List ret = newArrayList();
try{
// 映射IP信息文件到内存中
if(mbb ==null) {
FileChannel fc = ipFile.getChannel();
mbb = fc.map(FileChannel.MapMode.READ_ONLY, 0, ipFile.length());
mbb.order(ByteOrder.LITTLE_ENDIAN);
}
intendOffset = (int) ipEnd;
for(intoffset = (int) ipBegin +4; offset <= endOffset; offset += IP_RECORD_LENGTH) {
inttemp = readInt3(offset);
if(temp != -1) {
IPLocation loc = getIPLocation(temp);
// 判断是否这个地点里面包含了s子串,如果包含了,添加这个记录到List中,如果没有,继续
if(loc.country.indexOf(s) != -1
|| loc.area.indexOf(s) != -1) {
IPEntry entry = newIPEntry();
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());
}
returnret;
}
/**
* 从内存映射文件的offset位置开始的3个字节读取一个int
*
* @param offset
* @return
*/
privateintreadInt3(intoffset) {
mbb.position(offset);
returnmbb.getInt() &0x00FFFFFF;
}
/**
* 从内存映射文件的当前位置开始的3个字节读取一个int
*
* @return
*/
privateintreadInt3() {
returnmbb.getInt() &0x00FFFFFF;
}
/**
* 根据IP得到国家名
*
* @param ip
* ip的字节数组形式
* @return 国家名字符串
*/
publicString 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);
returnloc.country;
} else{
IPLocation loc = getIPLocation(ip);
ipCache.put(ipStr, loc.getCopy());
returnloc.country;
}
}
/**
* 根据IP得到国家名
*
* @param ip
* IP的字符串形式
* @return 国家名字符串
*/
publicString getCountry(String ip) {
returngetCountry(Utils.getIpByteArrayFromString(ip));
}
/**
* 根据IP得到地区名
*
* @param ip
* ip的字节数组形式
* @return 地区名字符串
*/
publicString 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);
returnloc.area;
} else{
IPLocation loc = getIPLocation(ip);
ipCache.put(ipStr, loc.getCopy());
returnloc.area;
}
}
/**
* 根据IP得到地区名
*
* @param ip
* IP的字符串形式
* @return 地区名字符串
*/
publicString getArea(String ip) {
returngetArea(Utils.getIpByteArrayFromString(ip));
}
/**
* 根据ip搜索ip信息文件,得到IPLocation结构,所搜索的ip参数从类成员ip中得到
*
* @param ip
* 要查询的IP
* @return IPLocation结构
*/
privateIPLocation getIPLocation(byte[] ip) {
IPLocation info = null;
longoffset = locateIP(ip);
if(offset != -1)
info = getIPLocation(offset);
if(info ==null) {
info = newIPLocation();
info.country = "未知国家";
info.area = "未知地区";
}
returninfo;
}
/**
* 从offset位置读取4个字节为一个long,因为java为big-endian格式,所以没办法 用了这么一个函数来做转换
*
* @param offset
* @return 读取的long值,返回-1表示读取文件失败
*/
privatelongreadLong4(longoffset) {
longret =0;
try{
ipFile.seek(offset);
ret |= (ipFile.readByte() & 0xFF);
ret |= ((ipFile.readByte() << 8) &0xFF00);
ret |= ((ipFile.readByte() << 16) &0xFF0000);
ret |= ((ipFile.readByte() << 24) &0xFF000000);
returnret;
} catch(IOException e) {
return-1;
}
}
/**
* 从offset位置读取3个字节为一个long,因为java为big-endian格式,所以没办法 用了这么一个函数来做转换
*
* @param offset
* @return 读取的long值,返回-1表示读取文件失败
*/
privatelongreadLong3(longoffset) {
longret =0;
try{
ipFile.seek(offset);
ipFile.readFully(b3);
ret |= (b3[0] &0xFF);
ret |= ((b3[1] <<8) &0xFF00);
ret |= ((b3[2] <<16) &0xFF0000);
returnret;
} catch(IOException e) {
return-1;
}
}
/**
* 从当前位置读取3个字节转换成long
*
* @return
*/
privatelongreadLong3() {
longret =0;
try{
ipFile.readFully(b3);
ret |= (b3[0] &0xFF);
ret |= ((b3[1] <<8) &0xFF00);
ret |= ((b3[2] <<16) &0xFF0000);
returnret;
} catch(IOException e) {
return-1;
}
}
/**
* 从offset位置读取四个字节的ip地址放入ip数组中,读取后的ip为big-endian格式,但是
* 文件中是little-endian形式,将会进行转换
*
* @param offset
* @param ip
*/
privatevoidreadIP(longoffset,byte[] ip) {
try{
ipFile.seek(offset);
ipFile.readFully(ip);
bytetemp = 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
*/
privatevoidreadIP(intoffset,byte[] ip) {
mbb.position(offset);
mbb.get(ip);
bytetemp = 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。
*/
privateintcompareIP(byte[] ip,byte[] beginIp) {
for(inti =0; i <4; i++) {
intr = compareByte(ip[i], beginIp[i]);
if(r !=0)
returnr;
}
return0;
}
/**
* 把两个byte当作无符号数进行比较
*
* @param b1
* @param b2
* @return 若b1大于b2则返回1,相等返回0,小于返回-1
*/
privateintcompareByte(byteb1,byteb2) {
if((b1 &0xFF) > (b2 &0xFF))// 比较是否大于
return1;
elseif((b1 ^ b2) ==0)// 判断是否相等
return0;
else
return-1;
}
/**
* 这个方法将根据ip的内容,定位到包含这个ip国家地区的记录处,返回一个绝对偏移 方法使用二分法查找。
*
* @param ip
* 要查询的IP
* @return 如果找到了,返回结束IP的偏移,如果没有找到,返回-1
*/
privatelonglocateIP(byte[] ip) {
longm =0;
intr;
// 比较第一个ip项
readIP(ipBegin, b4);
r = compareIP(ip, b4);
if(r ==0)
returnipBegin;
elseif(r <0)
return-1;
// 开始二分搜索
for(longi = 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;
elseif(r <0) {
if(m == j) {
j -= IP_RECORD_LENGTH;
m = j;
} else
j = m;
} else
returnreadLong3(m +4);
}
// 如果循环结束了,那么i和j必定是相等的,这个记录为最可能的记录,但是并非
// 肯定就是,还要检查一下,如果是,就返回结束地址区的绝对偏移
m = readLong3(m + 4);
readIP(m, b4);
r = compareIP(ip, b4);
if(r <=0)
returnm;
else
return-1;
}
/**
* 得到begin偏移和end偏移中间位置记录的偏移
*
* @param begin
* @param end
* @return
*/
privatelonggetMiddleOffset(longbegin,longend) {
longrecords = (end - begin) / IP_RECORD_LENGTH;
records >>= 1;
if(records ==0)
records = 1;
returnbegin + records * IP_RECORD_LENGTH;
}
/**
* 给定一个ip国家地区记录的偏移,返回一个IPLocation结构
*
* @param offset
* @return
*/
privateIPLocation getIPLocation(longoffset) {
try{
// 跳过4字节ip
ipFile.seek(offset + 4);
// 读取第一个字节判断是否标志字节
byteb = ipFile.readByte();
if(b == AREA_FOLLOWED) {
// 读取国家偏移
longcountryOffset = 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());
} elseif(b == NO_AREA) {
loc.country = readString(readLong3());
loc.area = readArea(offset + 8);
} else{
loc.country = readString(ipFile.getFilePointer() - 1);
loc.area = readArea(ipFile.getFilePointer());
}
returnloc;
} catch(IOException e) {
returnnull;
}
}
/**
* @param offset
* @return
*/
privateIPLocation getIPLocation(intoffset) {
// 跳过4字节ip
mbb.position(offset + 4);
// 读取第一个字节判断是否标志字节
byteb = mbb.get();
if(b == AREA_FOLLOWED) {
// 读取国家偏移
intcountryOffset = 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());
} elseif(b == NO_AREA) {
loc.country = readString(readInt3());
loc.area = readArea(offset + 8);
} else{
loc.country = readString(mbb.position() - 1);
loc.area = readArea(mbb.position());
}
returnloc;
}
/**
* 从offset偏移开始解析后面的字节,读出一个地区名
*
* @param offset
* @return 地区名字符串
* @throws IOException
*/
privateString readArea(longoffset)throwsIOException {
ipFile.seek(offset);
byteb = ipFile.readByte();
if(b ==0x01|| b ==0x02) {
longareaOffset = readLong3(offset +1);
if(areaOffset ==0)
return"未知地区";
else
returnreadString(areaOffset);
} else
returnreadString(offset);
}
/**
* @param offset
* @return
*/
privateString readArea(intoffset) {
mbb.position(offset);
byteb = mbb.get();
if(b ==0x01|| b ==0x02) {
intareaOffset = readInt3();
if(areaOffset ==0)
return"未知地区";
else
returnreadString(areaOffset);
} else
returnreadString(offset);
}
/**
* 从offset偏移处读取一个以0结束的字符串
*
* @param offset
* @return 读取的字符串,出错返回空字符串
*/
privateString readString(longoffset) {
try{
ipFile.seek(offset);
//int i;
//for (i = 0, buf[i] = ipFile.readByte(); buf[i] != 0; buf[++i] = ipFile.readByte());
//上面的写法读取数据如果超过100个字节就会报数组越界异常
inti =0;
byte[] buf =newbyte[256];
while((buf[i] = ipFile.readByte()) !=0) {
++ i;
if(i >= buf.length) {
byte[] tmp =newbyte[i +100];
System.arraycopy(buf, 0, tmp,0, i);
buf = tmp;
}
}
/
if(i !=0)
returnUtils.getString(buf,0, i,"GBK");
} catch(IOException e) {
System.out.println(e.getMessage());
}
return"";
}
/**
* 从内存映射文件的offset位置得到一个0结尾字符串
*
* @param offset
* @return
*/
privateString readString(intoffset) {
try{
mbb.position(offset);
//int i;
//for (i = 0, buf[i] = mbb.get(); buf[i] != 0; buf[++i] = mbb.get());
inti =0;
byte[] buf =newbyte[256];
while((buf[i] = mbb.get()) !=0) {
++ i;
if(i >= buf.length) {
byte[] tmp =newbyte[i +100];
System.arraycopy(buf, 0, tmp,0, i);
buf = tmp;
}
}
if(i !=0)
returnUtils.getString(buf,0, i,"GBK");
} catch(IllegalArgumentException e) {
System.out.println(e.getMessage());
}
return"";
}
publicString 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;
returnaddress.trim();
}
}