Ejemplo n.º 1
0
func (h *IPRangeHolder) FindCountry(ip string) (string, error) {
	v, err := helper.IPv42Int(ip)
	if nil != err {
		log.Printf("Failed to convert ip to int for reason:%v", err)
		return "z2", err
	}

	compare := func(i int) bool {
		return h.ranges[i].Start >= uint64(v)
	}
	index := sort.Search(len(h.ranges), compare)
	if index == len(h.ranges) {
		//log.Printf("####%d\n", len(h.ranges))
		return "zz", errIPRangeNotMatch
	}
	if index > 0 {
		if h.ranges[index].Start == uint64(v) && h.ranges[index].End >= uint64(v) {
			log.Printf("%s match ip range %s-%s", ip, helper.Long2IPv4(h.ranges[index].Start), helper.Long2IPv4(h.ranges[index].End))
			return h.ranges[index].Country, nil
		}
		if index > 0 {
			if h.ranges[index-1].Start <= uint64(v) && h.ranges[index-1].End >= uint64(v) {
				log.Printf("%s match ip range %s-%s", ip, helper.Long2IPv4(h.ranges[index-1].Start), helper.Long2IPv4(h.ranges[index-1].End))
				return h.ranges[index-1].Country, nil
			}
		}
		// if h.ranges[index].Start > uint64(v) || h.ranges[index].End < uint64(v) {
		// 	log.Printf("Got start:%d - %d for %d", h.ranges[index].Start, h.ranges[index].End, v)
		// 	return "z1", errIPRangeNotMatch
		// }
		// return h.ranges[index].Country, nil
	}
	return "", errIPRangeNotMatch
}
Ejemplo n.º 2
0
func parseApnicIPReader(rc io.ReadCloser, persist bool) (*IPRangeHolder, error) {
	var err error
	reader := bufio.NewReader(rc)
	var buffer bytes.Buffer
	var (
		part   []byte
		prefix bool
	)
	defer rc.Close()
	holder := new(IPRangeHolder)
	var perststFile *os.File
	if persist {
		var err error
		perststFile, err = os.Create(proxyHome + "/" + cnIPFile)
		if nil == err {
			defer perststFile.Close()
		} else {
			log.Printf("[ERROR]Failed to open CN IP file:%v", err)
		}
	}
	for {
		if part, prefix, err = reader.ReadLine(); err != nil {
			if err != io.EOF {
				return nil, err
			}
			break
		}
		buffer.Write(part)
		if !prefix {
			line := buffer.String()
			buffer.Reset()
			if strings.HasPrefix(line, "#") {
				continue
			}
			sp := strings.Split(line, "|")
			if len(sp) >= 6 {
				if sp[1] == "CN" && sp[2] == "ipv4" {
					startip, _ := helper.IPv42Int(sp[3])
					ipcount, _ := strconv.ParseUint(sp[4], 10, 32)
					tmp := &IPRange{uint64(startip), uint64(startip) + uint64(ipcount-1), sp[1]}
					holder.ranges = append(holder.ranges, tmp)
					if nil != perststFile {
						perststFile.Write([]byte(line))
						perststFile.Write([]byte("\r\n"))
					}
				}
			}
		}
	}
	holder.sort()
	return holder, nil
}