手机号码归属地库lua解析脚本

phone2.lua

#!/usr/bin/env lua

local phone_dat = "phone.dat"

function getPhoneType(a)
	if a == 1 then
		return "移动"
	elseif a == 2 then
		return "联通"
	elseif a==3 then
		return "电信"
	elseif a==4  then
		return "电信虚拟运营商"
	elseif a==5 then
		return "联通虚拟运营商"
	elseif a ==6 then
		return "移动虚拟运营商"
	else
		return ""
	end
end

local file = io.open(phone_dat)
if file == nil then
	buf = nil
else
	buf = file:read("*all")
end

function phoneFind(phone)
	
	if buf == nil then
		print("phone.dat not exists")
		return
	end
	local phone = string.sub(phone, 0, 7)
	local int_phone = tonumber(phone)
	--> print(phone)
	--> print("buf length:",string.len(buf))
	-- local version = string.sub(buf, 1, 4)
	local firstIndexOffset = string.unpack("<i", string.sub(buf,5,8))
	-- print("first index offset:",firstIndexOffset)
	local phoneRecordCount = math.floor((string.len(buf) - firstIndexOffset) / 9)
	local phoneIndexLength = 9
	-->print("phone record count:", phoneRecordCount)
	local left = 0
	local right = phoneRecordCount
	while (left <= right) do
		local middle = math.floor((left + right ) / 2)
		local curOffset = firstIndexOffset + middle * phoneIndexLength
		if curOffset >= string.len(buf) then
			return
		end
		local indexRecord = string.sub(buf, curOffset+1, curOffset + 9)
		local curPhone = string.unpack("<i", string.sub(indexRecord, 1,4))
		local phone_type = string.byte(indexRecord, 9)
		--> print(curPhone,phone_type)
		if curPhone > int_phone then
			right = middle - 1
		elseif curPhone < int_phone then
			left = middle + 1
		else 
			local dataOffset = string.unpack("<i", string.sub(indexRecord, 5, 8))
			--> print(dataOffset)
			local tmp = string.sub(buf, dataOffset + 1, dataOffset + 1024)
			local t = string.find(tmp, '\0')
			if t then
				return string.format("%s%s%s",string.sub(tmp, 1, t-1), "|" ,getPhoneType(phone_type))
			end
			break
		end
	end
end

print(phoneFind("1890303"))