YkLib 0.1.0

data/lib/YkLib/Yk/ioctl.rb

@@ -0,0 +1,60 @@
+# generated by 'ruby-ioctl-0.1.4-0tz1.src.rpm' on Thu Feb 11 19:33:56 +0800 2010
+
+module Ioctl
+	FIOSETOWN = 0x8901
+	SIOCSPGRP = 0x8902
+	FIOGETOWN = 0x8903
+	SIOCGPGRP = 0x8904
+	SIOCGSTAMP = 0x8906
+	TCGETS = 0x5401
+	TCSETS = 0x5402
+	TCSETSW = 0x5403
+	TCSETSF = 0x5404
+	TCGETA = 0x5405
+	TCSETA = 0x5406
+	TCSETAW = 0x5407
+	TCSETAF = 0x5408
+	TCSBRK = 0x5409
+	TCXONC = 0x540a
+	TCFLSH = 0x540b
+	TIOCEXCL = 0x540c
+	TIOCNXCL = 0x540d
+	TIOCSCTTY = 0x540e
+	TIOCGPGRP = 0x540f
+	TIOCSPGRP = 0x5410
+	TIOCOUTQ = 0x5411
+	TIOCSTI = 0x5412
+	TIOCGWINSZ = 0x5413
+	TIOCSWINSZ = 0x5414
+	TIOCMGET = 0x5415
+	TIOCMBIS = 0x5416
+	TIOCMBIC = 0x5417
+	TIOCMSET = 0x5418
+	TIOCGSOFTCAR = 0x5419
+	TIOCSSOFTCAR = 0x541a
+	FIONREAD = 0x541b
+	TIOCINQ = 0x541b
+	TIOCLINUX = 0x541c
+	TIOCCONS = 0x541d
+	TIOCGSERIAL = 0x541e
+	TIOCSSERIAL = 0x541f
+	TIOCPKT = 0x5420
+	FIONBIO = 0x5421
+	TIOCNOTTY = 0x5422
+	TIOCSETD = 0x5423
+	TIOCGETD = 0x5424
+	TCSBRKP = 0x5425
+	FIONCLEX = 0x5450
+	FIOCLEX = 0x5451
+	FIOASYNC = 0x5452
+	TIOCSERCONFIG = 0x5453
+	TIOCSERGWILD = 0x5454
+	TIOCSERSWILD = 0x5455
+	TIOCGLCKTRMIOS = 0x5456
+	TIOCSLCKTRMIOS = 0x5457
+	TIOCSERGSTRUCT = 0x5458
+	TIOCSERGETLSR = 0x5459
+	TIOCSERGETMULTI = 0x545a
+	TIOCSERSETMULTI = 0x545b
+end
+

data/lib/YkLib/Yk/ipcc.rb

@@ -0,0 +1,87 @@
+#!/usr/bin/env ruby
+
+
+require 'Yk/path_aux'
+require 'Yk/ipv4adr'
+#require 'path_aux'
+#require 'socket'
+
+
+class String
+	ICONF = "/etc/ipcc.conf"
+	RCONF = "/etc/resolv.conf"
+	port = 53535
+	srv = nil
+	if ICONF.readable_file?
+		ICONF.read_each_line do |ln|
+			if((srv = ln.strip_comment) != "")
+				cfile = ICONF
+				if(srv =~ /:/)
+					port = $'
+					srv = $`
+				end
+				break;
+			end
+		end
+	end
+	if !srv && RCONF.readable_file?
+		RCONF.read_each_line do |ln|
+			arr = ln.strip_comment.split
+			if(arr && arr[0] == "nameserver" && arr[1])
+				cfile = ICONF
+				srv = arr[1]
+				break
+			end
+		end
+	end
+	if !srv
+		cfile = "default setting"
+		srv = "127.0.0.1"
+	end
+	sockaddr = nil
+	begin
+		sockaddr = Socket.sockaddr_in(port, srv)
+	rescue SocketError
+		STDERR.write "cannot create socket for #{srv}:#{port} (from #{ICONF})\n"
+	end
+	S = Socket.open(Socket::AF_INET, Socket::SOCK_DGRAM, 0) rescue nil
+	S.connect(sockaddr) if S
+	def to_ipcc
+		ipadr = nil
+		if self =~ /^(\d|[1-9]\d|[12]\d\d)\.(\d|[1-9]\d|[12]\d\d)\.(\d|[1-9]\d|[12]\d\d)\.(\d|[1-9]\d|[12]\d\d)$/
+			if $1.to_i <= 255 && $2.to_i <= 255 && $3.to_i <= 255 && $4.to_i <= 255
+				ipadr = self
+			end
+		end
+		if !ipadr
+			ipadr = to_ipadr
+		end
+		if !ipadr
+			STDERR.write "cannot get ip address for #{self}\n"
+			nil
+		else
+			sin_addr = ipadr.split(/\./).map{|e| e.to_i}.pack("CCCC")
+			10.times do
+				S.write sin_addr
+				sret = IO.select [S], [], [], 0.1
+				if sret && sret[0][0] == S
+					buff = S.recv(6) rescue nil
+					if buff && buff.size == 6 && buff[0..3] == sin_addr
+						if buff[4] == 0 || buff[5] == 0
+							return ""
+						else
+							return buff[4..5]
+						end
+					end
+				end
+			end
+			nil
+		end
+	end
+end
+
+
+if __FILE__.normalize_path == $0.normalize_path
+	print ARGV[0].to_ipcc, "\n" if ARGV[0]
+end
+

data/lib/YkLib/Yk/ipcountry.rb

@@ -0,0 +1,207 @@
+#!/usr/bin/env ruby
+
+
+require 'Yk/path_aux'
+require 'Yk/auto_pstore'
+require 'Yk/bsearch'
+require 'Yk/ipv4adr'
+require 'timeout'
+
+module IPCountry
+
+IPListURLs = %w{
+					ftp://ftp.ripe.net/pub/stats/apnic/delegated-apnic-latest
+					ftp://ftp.arin.net/pub/stats/arin/delegated-arin-extended-latest
+					ftp://ftp.lacnic.net/pub/stats/lacnic/delegated-lacnic-latest
+					ftp://ftp.ripe.net/ripe/stats/delegated-ripencc-latest
+					ftp://ftp.afrinic.net/pub/stats/afrinic/delegated-afrinic-latest
+				}
+
+
+class IPInfo
+	attr :country, true
+	attr :start, true
+	attr :ed, true
+	attr :nxt, true
+	def initialize c, s, e, n = nil
+		@country = c
+		if c == ""
+			@country = "  "
+		end
+		@start = s
+		@ed = e
+		@nxt = n
+	end
+	def inspect
+		[@country, @start.to_ipadr, @start].inspect
+	end
+end
+
+
+class AllocList < Array
+	attr :date
+	def refresh
+		tmp = []
+		IPListURLs.each do |u|
+			u.timeout = 3600 * 12
+		end
+		IPListURLs.each do |u|
+			u.read_each_line do |ln|
+				arr = ln.split /\|/
+				if arr[2] == "ipv4" && arr[1] != "*" && arr[1] != ""
+					tmp.push IPInfo.new(arr[1], arr[3].to_i, arr[3].to_i + arr[4].to_i)
+				end
+			end
+		end
+		tmp = tmp.sort_by do |a|
+			a.start
+		end
+		tmp2 = []
+		(0..tmp.size - 2).each do |i|
+			if tmp[i].country == tmp[i + 1].country && tmp[i].ed == tmp[i + 1].start
+				tmp[i + 1].start = tmp[i].start
+				next
+			end
+			tmp[i].nxt = tmp[i + 1].start
+			tmp2.push tmp[i]
+		end
+		tmp = tmp2
+		tmp[-1].nxt = tmp[-1].ed
+		clear
+		tmp.each do |e|
+			push e
+		end
+		("/var/data".check_dir / "ipcountry").open "w" do |fw|
+			(0..tmp.size - 2).each do |i|
+				if tmp[i].start >= tmp[i].ed
+					next
+				end
+				if i == 0 && tmp[i].start != 0
+					fw.write([0, 0, 0, 0, 0].pack("ICCCC"))
+				end
+				fw.write([tmp[i].start, tmp[i].country[0..0], tmp[i].country[1..1], 0, 0].pack("IAACC"))
+				if tmp[i].ed < tmp[i + 1].start
+					fw.write([tmp[i].ed, 0, 0, 0, 0].pack("ICCCC"))
+				elsif tmp[i].ed > tmp[i + 1].start
+					tmp[i + 1].start = tmp[i].ed
+					STDERR.write "error #{tmp[i].country}:#{tmp[i].start.to_ipadr}-#{tmp[i].ed.to_ipadr}, #{tmp[i + 1].country}:#{tmp[i + 1].start.to_ipadr}-#{tmp[i + 1].ed.to_ipadr}\n"
+				end
+				if i == tmp.size - 2
+					fw.write([tmp[i + 1].start, tmp[i + 1].country[0..0], tmp[i + 1].country[1..1], 0, 0].pack("IAACC"))
+					if tmp[i + 1].ed != 0
+						fw.write([tmp[i + 1].ed, 0, 0, 0, 0].pack("ICCCC"))
+					end
+				end
+			end
+		end
+		@date = Time.now
+	end
+	def initialize
+		refresh
+	end
+	def getCountry ipi
+		if ipi < self[0].start || self[-1].ed < ipi
+			return nil
+		end
+		fst = bsearch_first do |e|
+			if ipi < e.start
+				1
+			elsif ipi < e.nxt
+				0
+			else
+				-1
+			end
+		end
+		if fst
+			fst = self[fst]
+			if ipi < fst.ed
+				return fst.country
+			else
+				return nil
+			end
+		else
+			return nil
+		end
+	end
+end
+
+end #module IPCountry
+
+
+if $0 == __FILE__
+
+if ARGV.delete "--refreshdb"
+	AList = AutoPstore.allocList = IPCountry::AllocList.new
+	exit 0
+else
+	AList = AutoPstore.setReadOnly.allocList || (AutoPstore.allocList = IPCountry::AllocList.new)
+end
+
+
+longFormat = ARGV.delete "-l"
+
+
+if ARGV[0]
+	if a = ARGV[0].to_ipadr
+		if ret = AList.getCountry(a.to_i)
+			if longFormat
+				println "#{ret} : " + "ccode #{ret}".read_p
+			else
+				println ret
+			end
+			exit 0
+		else
+			STDERR.println "'#{ARGV[0]}' is not allocated"
+			if STDIN.tty?
+				print "\n"
+			end
+			exit 1
+		end
+	end
+else
+	STDERR.write "usage : ipcountry HOST\n        ipcountry --refreshdb (for refreshing its database)\n"
+	exit 1
+end
+
+else
+	require 'thread'
+	module IPCountry
+		@@mutex = Mutex.new
+		def refresh
+			@@mutex.synchronize do
+				@@list = AutoPstore.readTemp.allocList
+			end
+		end
+		def getCountry ip
+			@@mutex.synchronize do
+				@@list ||= AutoPstore.readTemp.allocList
+				@@list.getCountry(ip.to_i)
+			end
+		end
+		module_function :refresh, :getCountry
+	end
+	AutoPstore.setFileName "ipcountry"
+	IPCountry::List = AutoPstore.readTemp.allocList
+	class String
+		def ipcountry is_cc = false
+			TopLevelMethod.timeout 5 do
+				if ip = to_ipadr
+					if cc = IPCountry::getCountry(ip)
+						if is_cc
+							if cc == is_cc
+								return cc
+							end
+						else
+							return cc
+						end
+					else
+						return nil
+					end
+				else
+					return nil
+				end
+			end
+		end
+	end
+end
+

data/lib/YkLib/Yk/ipv4adr.rb

@@ -0,0 +1,318 @@
+
+
+require "resolv"
+require "Yk/ranger"
+
+
+class Integer
+	def to_ipadr
+		d1 = (0xff000000 & self) >> 24
+		d2 = (0x00ff0000 & self) >> 16
+		d3 = (0x0000ff00 & self) >> 8
+		d4 = 0x000000ff & self
+		"#{d1}.#{d2}.#{d3}.#{d4}"
+	end
+	def to_mask
+		if 0 <= self && self <= 32
+			String::IPMaskList[self]
+		else
+			nil
+		end
+	end
+end
+
+
+class String
+	IPMaskList = []
+	33.times do |i|
+		IPMaskList.unshift(((1 << 32) - (1 << i)).to_ipadr)
+	end
+	def __getMaskInfo
+		adr = self
+		if adr =~ /^\d|[1-9]\d$/ && adr.to_i <= 32
+			[IPMaskList[adr.to_i], adr.to_i]
+		elsif i = IPMaskList.index(adr)
+			[adr, i]
+		else
+			nil
+		end
+	end
+	def ipmask?
+		if self =~ /^(\d|[1-9]\d)$/ && self.__org__to_i___ <= 32
+			return true
+		elsif IPMaskList.include? self
+			return true
+		end
+		return false
+	end
+	def to_mask
+		if ipmask?
+			if self =~ /\./
+				self.clone
+			else
+				IPMaskList[self.__org__to_i___]
+			end
+		elsif self =~ /\// && $`.ipadr? && $'.ipmask?
+			$` + "/" + $'.to_mask
+		else
+			nil
+		end
+	end
+	def to_maskbit
+		if ipmask?
+			if self =~ /\./
+				IPMaskList.index self
+			else
+				self.__org__to_i___
+			end
+		elsif self =~ /\//
+			if $`.ipadr? && $'.ipmask?
+				$` + "/" + $'.to_maskbit.to_s
+			else
+				nil
+			end
+		else
+			nil
+		end
+	end
+	def ipadr?
+		if self =~ /^(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})$/
+			(1..4).each do |i|
+				a = $~[i].to_i
+				a > 255 and return(false)
+			end
+			true
+		else
+			false
+		end
+	end
+	def ipexpr?
+		if !ipadr?
+			if self !~ /\-/ || !$`.ipadr? || !$'.ipadr?
+				if self !~ /\// || !$`.ipadr? || !$'.ipmask?
+					return false
+				end
+			end
+		end
+		return true
+	end
+	def to_nets
+		if !ipexpr?
+			return nil
+		end
+		ipexpr = self
+		if ipexpr !~ /\-/
+			[ipexpr]
+		else
+			possibleMask = Proc.new do |i|
+				j2 = nil
+				32.times do |j|
+					if i & (1 << j) != 0
+						j2 = j
+						break
+					end
+				end
+				j2
+			end
+			ret = []
+			arr = ipexpr.split /\-/
+			bg = ip2int(arr[0].strip)
+			ed = ip2int(arr[1].strip)
+			while bg <= ed
+				bgNext = nil
+				possibleMask.call(bg).downto 0 do |m|
+					if (bgNext = bg + (1 << m) - 1) <= ed
+						bgNext += 1
+						if m != 0
+							ret.push "#{int2ip bg}/#{32 - m}"
+						else
+							ret.push int2ip(bg)
+						end
+						break
+					end
+				end
+				bg = bgNext
+			end
+			ret
+		end
+	end
+	def to_djbs
+		ret = []
+		expr = self
+		if expr !~ /\//
+			if expr =~ /\-/
+				to_nets.each do |expr|
+					ret.push *expr.to_djbs
+				end
+			end
+		end
+		net = $`
+		if net.ipaddr?
+			
+		end
+		mask = $'.to_i
+		case mask
+		when 25..32
+			base = (ip2int(net) >> (32 - mask)) << (32 - mask)
+			(0 ... 1 << (32 -mask)).each do |i|
+				ret.push int2ip(base + i)
+			end
+		when 24
+			net =~ /^(\d{1,3}\.\d{1,3}\.\d{1,3})\./
+			ret.push $1
+		when 17..23
+			base = (ip2int(net) >> (32 - mask)) << (24 - mask)
+			(0 ... 1 << (24 -mask)).each do |i|
+				ret.push int2ip((base + i) << 8)[/^(\d{1,3}\.\d{1,3}.\d{1,3})\./, 1]
+			end
+		when 16
+			net =~ /^(\d{1,3}\.\d{1,3})\./
+			ret.push $1
+		when 9..15
+			base = (ip2int(net) >> (32 - mask)) << (16 - mask)
+			(0 ... 1 << (16 -mask)).each do |i|
+				ret.push int2ip((base + i) << 16)[/^(\d{1,3}\.\d{1,3})\./, 1]
+			end
+		when 8
+			net =~ /^(\d{1,3})\./
+			ret.push $1
+		when 0..7
+			base = (ip2int(net) >> (32 - mask)) << (8 - mask)
+			(0 ... 1 << (8 -mask)).each do |i|
+				ret.push((base + i).to_s)
+			end
+		end
+		ret
+	end
+	def to_range
+		if ipadr?
+			a = __ip2int
+			return a .. a
+		elsif self =~ /\-/ && $`.ipadr? && $'.ipadr?
+			a = $'.__ip2int
+			b = $`.__ip2int
+			if a < b
+				return a .. b
+			elsif a > b
+				return b .. a
+			else
+				return a .. a
+			end
+		elsif self =~ /\// && $`.ipadr? && m = $'.__getMaskInfo
+			st = $`.__ip2int & m[0].__ip2int
+			ed = st + (1 << (32 - m[1])) - 1
+			return st .. ed
+		else
+			nil
+		end
+	end
+	def __ip2int
+		ret = 0
+		self.split(".").each do |e|
+			ret = ret * 256 + e.to_i
+		end
+		ret
+	end
+	alias_method :__org__to_i___, :to_i
+	def to_i base = 10
+		if ipadr?
+			__ip2int
+		else
+			__org__to_i___ base
+		end
+	end
+	RESOLVERS = Hash.new{|h, k| h[k] = Resolv::DNS.new(:nameserver => [k])}
+	HOST_RESOLVER = Resolv::Hosts.new
+	def to_ipadr server = nil
+		if !ipexpr?
+			ip2 = nil
+			resolver = nil
+			if server && (server = server.to_ipadr)
+				resolver = RESOLVERS[server]
+			else
+				resolver = Resolv
+			end
+			[HOST_RESOLVER, resolver].each do |r|
+				r.each_address(self) do |adr|
+					adr = adr.to_s
+					if !ip2
+					   	ip2 = adr.to_i
+					elsif ip2 < (ip3 = adr.to_i)
+					    ip2 = ip3
+					end
+				end
+				break if ip2
+			end
+			ip2.to_ipadr
+		else
+			self
+		end
+	end
+	def to_ranger
+		IPRanger.new self
+	end
+	def each_ip
+		r = to_range
+		r.each do |i|
+			yield i.to_ipadr
+		end
+	end
+end
+
+
+class IPRanger < Ranger
+	def initialize (*args)
+		super()
+		add *args
+	end
+	def inspect
+		ns = []
+		eachNetRange do |n|
+			ns.push n
+		end
+		"[" + ns.join(",") + "]"
+	end
+	def modargs (*args)
+		nargs = []
+		args.each do |a|
+			if a.is_a? String
+				if tmp = a.to_range
+					nargs.push tmp
+				else
+					nargs.push a.to_ipadr.to_range rescue next
+				end
+			elsif a.is_a? Ranger
+				a.eachRange do |e|
+					nargs.push e
+				end
+			else
+				nargs.push a
+			end
+		end
+		nargs
+	end
+	def add (*args)
+		super *modargs(*args)
+	end
+	def del (*args)
+		super *modargs(*args)
+	end
+	def include? (*args)
+		super *modargs(*args)
+	end
+	def eachNetRange
+		each do |e|
+			if e.is_a? Range
+				if e.exclude_end?
+					yield e.first.to_ipadr + "-" + (e.last - 1).to_ipadr
+				else
+					yield e.first.to_ipadr + "-" + e.last.to_ipadr
+				end
+			else
+				yield e.to_ipadr
+			end
+		end
+	end
+end
+
+