s

.gitignore

+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+bin/*
+
+# Test binary, build with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# dep
+vendor
+
+# GoLand
+.idea/*
+.vscode
+
+# macOS file
+.DS_Store
+nali
\ No newline at end of file

Dockerfile

+FROM golang:alpine as builder
+
+RUN apk add --no-cache make git
+WORKDIR /nali-src
+COPY . /nali-src
+RUN go mod download && \
+    make docker && \
+    mv ./bin/nali-docker /nali
+
+FROM alpine:latest
+
+RUN apk add --no-cache ca-certificates
+COPY --from=builder /nali /
+ENTRYPOINT ["/nali"]
\ No newline at end of file

LICENSE

+The MIT License (MIT)
+
+Copyright © 2020-2022 zu1k i@zu1k.com
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

Makefile

+NAME=nali
+BINDIR=bin
+VERSION=$(shell git describe --tags || echo "unknown version")
+GOBUILD=CGO_ENABLED=0 go build -trimpath -ldflags '-X "github.com/zu1k/nali/internal/constant.Version=$(VERSION)" -w -s'
+
+PLATFORM_LIST = \
+	darwin-arm64 \
+	darwin-amd64 \
+	linux-386 \
+	linux-amd64 \
+	linux-armv5 \
+	linux-armv6 \
+	linux-armv7 \
+	linux-armv8 \
+	linux-mips-softfloat \
+	linux-mips-hardfloat \
+	linux-mipsle-softfloat \
+	linux-mipsle-hardfloat \
+	linux-mips64 \
+	linux-mips64le \
+	freebsd-386 \
+	freebsd-amd64
+
+WINDOWS_ARCH_LIST = \
+	windows-386 \
+	windows-amd64
+
+all: linux-amd64 darwin-amd64 windows-amd64 # Most used
+
+docker:
+	$(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+darwin-arm64:
+	GOARCH=arm64 GOOS=darwin $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+darwin-amd64:
+	GOARCH=amd64 GOOS=darwin $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-386:
+	GOARCH=386 GOOS=linux $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-amd64:
+	GOARCH=amd64 GOOS=linux $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-armv5:
+	GOARCH=arm GOOS=linux GOARM=5 $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-armv6:
+	GOARCH=arm GOOS=linux GOARM=6 $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-armv7:
+	GOARCH=arm GOOS=linux GOARM=7 $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-armv8:
+	GOARCH=arm64 GOOS=linux $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-mips-softfloat:
+	GOARCH=mips GOMIPS=softfloat GOOS=linux $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-mips-hardfloat:
+	GOARCH=mips GOMIPS=hardfloat GOOS=linux $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-mipsle-softfloat:
+	GOARCH=mipsle GOMIPS=softfloat GOOS=linux $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-mipsle-hardfloat:
+	GOARCH=mipsle GOMIPS=hardfloat GOOS=linux $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-mips64:
+	GOARCH=mips64 GOOS=linux $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+linux-mips64le:
+	GOARCH=mips64le GOOS=linux $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+freebsd-386:
+	GOARCH=386 GOOS=freebsd $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+freebsd-amd64:
+	GOARCH=amd64 GOOS=freebsd $(GOBUILD) -o $(BINDIR)/$(NAME)-$@
+
+windows-386:
+	GOARCH=386 GOOS=windows $(GOBUILD) -o $(BINDIR)/$(NAME)-$@.exe
+
+windows-amd64:
+	GOARCH=amd64 GOOS=windows $(GOBUILD) -o $(BINDIR)/$(NAME)-$@.exe
+
+gz_releases=$(addsuffix .gz, $(PLATFORM_LIST))
+zip_releases=$(addsuffix .zip, $(WINDOWS_ARCH_LIST))
+
+$(gz_releases): %.gz : %
+	chmod +x $(BINDIR)/$(NAME)-$(basename $@)
+	gzip -f -S -$(VERSION).gz $(BINDIR)/$(NAME)-$(basename $@)
+
+$(zip_releases): %.zip : %
+	zip -m -j $(BINDIR)/$(NAME)-$(basename $@)-$(VERSION).zip $(BINDIR)/$(NAME)-$(basename $@).exe
+
+all-arch: $(PLATFORM_LIST) $(WINDOWS_ARCH_LIST)
+
+releases: $(gz_releases) $(zip_releases)
+
+sha256sum:
+	cd $(BINDIR); for file in *; do sha256sum $$file > $$file.sha256; done
+
+clean:
+	rm $(BINDIR)/*

README.md

+<h1 align="center">
+  <br>Nali<br>
+</h1>
+
+<h4 align="center">一个查询IP地理信息和CDN提供商的离线终端工具.</h4>
+
+<p align="center">
+  <a href="https://github.com/zu1k/nali/actions">
+    <img src="https://img.shields.io/github/workflow/status/zu1k/nali/Go?style=flat-square" alt="Github Actions">
+  </a>
+  <a href="https://goreportcard.com/report/github.com/zu1k/nali">
+    <img src="https://goreportcard.com/badge/github.com/zu1k/nali?style=flat-square">
+  </a>
+  <a href="https://github.com/zu1k/nali/releases">
+    <img src="https://img.shields.io/github/release/zu1k/nali/all.svg?style=flat-square">
+  </a>
+</p>
+
+#### [English](https://github.com/zu1k/nali/blob/master/README_en.md)
+
+## 功能
+
+- 支持多种数据库
+  - 纯真 IPv4 离线数据库
+  - ZX IPv6 离线数据库
+  - Geoip2 城市数据库 (可选)
+  - IPIP 数据库 (可选)
+  - ip2region 数据库 (可选)
+  - DB-IP 数据库 (可选)
+  - IP2Location DB3 LITE 数据库 (可选)
+- CDN 服务提供商查询
+- 支持管道处理
+- 支持交互式查询
+- 同时支持IPv4和IPv6
+- 支持多语言
+- 查询完全离线
+- 全平台支持
+- 支持彩色输出
+
+## 安装
+
+### 从源码安装
+
+Nali 需要预先安装 Go >= 1.18. 安装后可以从源码安装软件:
+
+```sh
+$ go install github.com/zu1k/nali@latest
+```
+
+### 下载预编译的可执行程序
+
+可以从Release页面下载预编译好的可执行程序: [Release](https://github.com/zu1k/nali/releases)
+
+你需要选择适合你系统和硬件架构的版本下载，解压后可直接运行
+
+### Arch 系 Linux
+
+我们在 Aur 中发布了 3 个相关的包:
+
+- `nali-go`: Release 版本，安装时编译
+- `nali-go-bin`: Release 版本，预编译的二进制文件
+- `nali-go-git`: 最新的 master 分支版本，安装时编译
+  
+## 使用说明
+
+### 查询一个IP的地理信息
+
+```
+$ nali 1.2.3.4
+1.2.3.4 [澳大利亚 APNIC Debogon-prefix网络]
+```
+
+#### 或者 使用 `管道`
+
+```
+$ echo IP 6.6.6.6 | nali
+IP 6.6.6.6 [美国 亚利桑那州华楚卡堡市美国国防部网络中心]
+```
+
+### 同时查询多个IP的地理信息
+
+```
+$ nali 1.2.3.4 4.3.2.1 123.23.3.0
+1.2.3.4 [澳大利亚 APNIC Debogon-prefix网络]
+4.3.2.1 [美国 新泽西州纽瓦克市Level3Communications]
+123.23.3.0 [越南 越南邮电集团公司]
+```
+
+### 交互式查询
+
+使用 `exit` 或  `quit` 退出查询
+
+```
+$ nali
+123.23.23.23
+123.23.23.23 [越南 越南邮电集团公司]
+1.0.0.1
+1.0.0.1 [美国 APNIC&CloudFlare公共DNS服务器]
+8.8.8.8
+8.8.8.8 [美国 加利福尼亚州圣克拉拉县山景市谷歌公司DNS服务器]
+quit
+```
+
+### 与 `dig` 命令配合使用
+
+需要你系统中已经安装好 dig 程序
+
+```
+$ dig nali.zu1k.com +short | nali
+104.28.2.115 [美国 CloudFlare公司CDN节点]
+104.28.3.115 [美国 CloudFlare公司CDN节点]
+172.67.135.48 [美国 CloudFlare节点]
+```
+
+### 与 `nslookup` 命令配合使用
+
+需要你系统中已经安装好 nslookup 程序
+
+```
+$ nslookup nali.zu1k.com 8.8.8.8 | nali
+Server:         8.8.8.8 [美国 加利福尼亚州圣克拉拉县山景市谷歌公司DNS服务器]
+Address:        8.8.8.8 [美国 加利福尼亚州圣克拉拉县山景市谷歌公司DNS服务器]#53
+
+Non-authoritative answer:
+Name:   nali.zu1k.com
+Address: 104.28.3.115 [美国 CloudFlare公司CDN节点]
+Name:   nali.zu1k.com
+Address: 104.28.2.115 [美国 CloudFlare公司CDN节点]
+Name:   nali.zu1k.com
+Address: 172.67.135.48 [美国 CloudFlare节点]
+```
+
+### 与任意程序配合使用
+
+因为 nali 支持管道处理，所以可以和任意程序配合使用
+
+```
+bash abc.sh | nali
+```
+
+Nali 将在 IP后面插入IP地理信息，CDN域名后面插入CDN服务提供商信息
+
+### 支持IPv6
+
+和 IPv4 用法完全相同
+
+```
+$ nslookup google.com | nali
+Server:         127.0.0.53 [局域网 IP]
+Address:        127.0.0.53 [局域网 IP]#53
+
+Non-authoritative answer:
+Name:   google.com
+Address: 216.58.211.110 [美国 Google全球边缘网络]
+Name:   google.com
+Address: 2a00:1450:400e:809::200e [荷兰Amsterdam Google Inc. 服务器网段]
+```
+
+### 查询 CDN 服务提供商
+
+因为 CDN 服务通常使用 CNAME 的域名解析方式，所以推荐与 `nslookup` 或者 `dig` 配合使用，在已经知道 CNAME 后可单独使用
+
+```
+$ nslookup www.gov.cn | nali
+Server:         127.0.0.53 [局域网 IP]
+Address:        127.0.0.53 [局域网 IP]#53
+
+Non-authoritative answer:
+www.gov.cn      canonical name = www.gov.cn.bsgslb.cn [白山云 CDN].
+www.gov.cn.bsgslb.cn [白山云 CDN]       canonical name = zgovweb.v.bsgslb.cn [白山云 CDN].
+Name:   zgovweb.v.bsgslb.cn [白山云 CDN]
+Address: 103.104.170.25 [新加坡 ]
+Name:   zgovweb.v.bsgslb.cn [白山云 CDN]
+Address: 2001:428:6402:21b::5 [美国Louisiana州Monroe Qwest Communications Company, LLC (CenturyLink)]
+Name:   zgovweb.v.bsgslb.cn [白山云 CDN]
+Address: 2001:428:6402:21b::6 [美国Louisiana州Monroe Qwest Communications Company, LLC (CenturyLink)]
+```
+
+## 用户交互
+
+程序第一次运行后，会在工作目录生成配置文件 `config.yaml` (默认`~/.nali/config.yaml`)，配置文件定义了数据库信息，默认用户无需进行修改
+
+数据库格式默认如下：
+
+```yaml
+- name: geoip
+  name-alias:
+  - geolite
+  - geolite2
+  format: mmdb
+  file: GeoLite2-City.mmdb
+  languages:
+  - ALL
+  types:
+  - IPv4
+  - IPv6
+```
+
+其中，`languages` 和 `types` 表示该数据库支持的语言和查询类型。 如果你需要增加数据库，需小心修改配置文件，如果有任何问题，欢迎提 issue 询问。
+
+### 查看帮助
+
+```
+$ nali --help
+Usage:
+  nali [flags]
+  nali [command]
+
+Available Commands:
+  help        Help about any command
+  update      update qqwry, zxipv6wry, ip2region ip database and cdn
+
+Flags:
+  -h, --help     help for nali
+  -t, --toggle   Help message for toggle
+
+Use "nali [command] --help" for more information about a command.
+```
+
+### 更新数据库
+
+更新所有可以自动更新的数据库
+
+```
+$ nali update
+2020/07/17 12:53:46 正在下载最新纯真 IP 库...
+2020/07/17 12:54:05 已将最新的纯真 IP 库保存到本地 /root/.nali/qqwry.dat
+```
+
+或者指定数据库
+
+```
+$ nali update --db qqwry,cdn
+2020/07/17 12:53:46 正在下载最新纯真 IP 库...
+2020/07/17 12:54:05 已将最新的纯真 IP 库保存到本地 /root/.nali/qqwry.dat
+```
+
+### 自选数据库
+
+用户可以指定使用哪个数据库，需要设置环境变量： `NALI_DB_IP4`、`NALI_DB_IP6` 或者两个同时设置
+
+支持的变量内容:
+
+- Geoip2 `['geoip', 'geoip2']`
+- Chunzhen `['chunzhen', 'qqwry']`
+- IPIP `['ipip']`
+- Ip2Resion `['ip2region', 'i2r']`
+- DBIP `['dbip', 'db-ip']`
+- IP2Location `['ip2location']`
+
+#### Windows平台
+
+##### 使用geoip数据库
+
+```
+set NALI_DB_IP4=geoip
+
+或者使用 powershell
+
+$env:NALI_DB_IP4="geoip"
+```
+
+##### 使用ipip数据库
+
+```
+set NALI_DB_IP6=ipip
+
+或者使用 powershell
+
+$env:NALI_DB_IP6="ipip"
+```
+
+#### Linux平台
+
+##### 使用geoip数据库
+
+```
+export NALI_DB_IP4=geoip
+```
+
+##### 使用ipip数据库
+
+```
+export NALI_DB_IP4=ipip
+```
+
+### 多语言支持
+
+通过修改环境变量 `NALI_LANG` 来指定使用的语言，当使用非中文语言时仅支持GeoIP2这个数据库
+
+该参数可设置的值见 GeoIP2 这个数据库的支持列表
+
+```
+# NALI_LANG=en nali 1.1.1.1
+1.1.1.1 [Australia]
+```
+
+### 更换工作目录
+
+如果未指定数据库存放目录，数据库默认将存放在 `~/.nali`
+
+设置环境变量 `NALI_HOME` 来指定工作目录，数据库存放在工作目录下。也可在配置文件中使用绝对路径指定其他数据库路径。
+
+```
+set NALI_HOME=D:\nali
+
+or
+
+export NALI_HOME=/var/nali
+```
+
+## 感谢列表
+
+- [纯真QQIP离线数据库](http://www.cz88.net/fox/ipdat.shtml)
+- [qqwry mirror](https://qqwry.mirror.noc.one/)
+- [qqwry纯真数据库解析](https://github.com/yinheli/qqwry)
+- [ZX公网ipv6数据库](https://ip.zxinc.org/ipquery/)
+- [Geoip2 city数据库](https://www.maxmind.com/en/geoip2-precision-city-service)
+- [geoip2-golang解析器](https://github.com/oschwald/geoip2-golang)
+- [CDN provider数据库](https://github.com/SukkaLab/cdn)
+- [IPIP数据库](https://www.ipip.net/product/ip.html)
+- [IPIP数据库解析](https://github.com/ipipdotnet/ipdb-go)
+- [ip2region数据库](https://github.com/lionsoul2014/ip2region)
+- [IP2Location DB3 LITE](https://lite.ip2location.com/database/db3-ip-country-region-city)
+- [Cobra CLI库](https://github.com/spf13/cobra)
+- [Nali-cli](https://github.com/SukkaW/nali-cli)
+
+感谢 JetBrains 提供开源项目免费License 
+
+<a href="https://www.jetbrains.com/?from=nali">
+  <img src="assets/GoLand.svg">
+</a>
+
+## 作者
+
+**Nali** © [zu1k](https://github.com/zu1k), 遵循 [MIT](./LICENSE) 证书.<br>
+
+> Blog [zu1k.com](https://zu1k.com) · GitHub [@zu1k](https://github.com/zu1k) · Twitter [@zu1k_lv](https://twitter.com/zu1k_lv) · Telegram Channel [@peekfun](https://t.me/peekfun)
+
+## Star统计
+
+[![Stargazers over time](https://starchart.cc/zu1k/nali.svg)](https://starchart.cc/zu1k/nali)
\ No newline at end of file

README_en.md

+<h1 align="center">
+  <br>Nali<br>
+</h1>
+
+<h4 align="center">An offline tool for querying IP geographic information and CDN provider.</h4>
+
+<p align="center">
+  <a href="https://github.com/zu1k/nali/actions">
+    <img src="https://img.shields.io/github/workflow/status/zu1k/nali/Go?style=flat-square" alt="Github Actions">
+  </a>
+  <a href="https://goreportcard.com/report/github.com/zu1k/nali">
+    <img src="https://goreportcard.com/badge/github.com/zu1k/nali?style=flat-square">
+  </a>
+  <a href="https://github.com/zu1k/nali/releases">
+    <img src="https://img.shields.io/github/release/zu1k/nali/all.svg?style=flat-square">
+  </a>
+</p>
+
+#### [中文文档](https://github.com/zu1k/nali/blob/master/README.md)
+
+## Feature
+
+- Multi database support
+  - Chunzhen qqip database
+  - ZX ipv6 database
+  - Geoip2 city database
+  - IPIP free database
+  - ip2region database
+  - db-ip database
+  - IP2Location DB3 LITE database
+- Pipeline support
+- Interactive query
+- Offline query
+- Both ipv4 and ipv6 supported
+- Multilingual support
+- CDN provider query
+- Full platform support
+- Color print
+
+## Install
+
+### Install from source
+
+Nali Requires Go >= 1.18. You can build it from source:
+
+```sh
+$ go install github.com/zu1k/nali@latest
+```
+
+### Install pre-build binariy
+
+Pre-built binaries are available here: [release](https://github.com/zu1k/nali/releases)
+
+Download the binary compatible with your platform, unpack and copy to the directory in path
+
+### Arch Linux
+
+We have published 3 package in Aur:
+
+- `nali-go`: release version, compile when installing
+- `nali-go-bin`: release version, pre-compiled binary
+- `nali-go-git`: the latest master branch version, compile when installing
+
+## Usage
+
+### Query a simple IP address
+
+```
+$ nali 1.2.3.4
+1.2.3.4 [澳大利亚 APNIC Debogon-prefix网络]
+```
+
+#### or use `pipe`
+
+```
+$ echo IP 6.6.6.6 | nali
+IP 6.6.6.6 [美国 亚利桑那州华楚卡堡市美国国防部网络中心]
+```
+
+### Query multiple IP addresses
+
+```
+$ nali 1.2.3.4 4.3.2.1 123.23.3.0
+1.2.3.4 [澳大利亚 APNIC Debogon-prefix网络]
+4.3.2.1 [美国 新泽西州纽瓦克市Level3Communications]
+123.23.3.0 [越南 越南邮电集团公司]
+```
+
+### Interactive query
+
+use `exit` or  `quit` to quit
+
+```
+$ nali
+123.23.23.23
+123.23.23.23 [越南 越南邮电集团公司]
+1.0.0.1
+1.0.0.1 [美国 APNIC&CloudFlare公共DNS服务器]
+8.8.8.8
+8.8.8.8 [美国 加利福尼亚州圣克拉拉县山景市谷歌公司DNS服务器]
+quit
+```
+
+### Use with dig
+
+```
+$ dig nali.zu1k.com +short | nali
+104.28.2.115 [美国 CloudFlare公司CDN节点]
+104.28.3.115 [美国 CloudFlare公司CDN节点]
+172.67.135.48 [美国 CloudFlare节点]
+```
+
+### Use with nslookup
+
+```
+$ nslookup nali.zu1k.com 8.8.8.8 | nali
+Server:         8.8.8.8 [美国 加利福尼亚州圣克拉拉县山景市谷歌公司DNS服务器]
+Address:        8.8.8.8 [美国 加利福尼亚州圣克拉拉县山景市谷歌公司DNS服务器]#53
+
+Non-authoritative answer:
+Name:   nali.zu1k.com
+Address: 104.28.3.115 [美国 CloudFlare公司CDN节点]
+Name:   nali.zu1k.com
+Address: 104.28.2.115 [美国 CloudFlare公司CDN节点]
+Name:   nali.zu1k.com
+Address: 172.67.135.48 [美国 CloudFlare节点]
+```
+
+### Use with any other program
+
+Because nali can read the contents of the `stdin` pipeline, it can be used with any program
+
+```
+bash abc.sh | nali
+```
+
+Nali will insert ip information after ip
+
+### IPV6 support
+
+Use like ipv4
+
+```
+$ nslookup google.com | nali
+Server:         127.0.0.53 [局域网 IP]
+Address:        127.0.0.53 [局域网 IP]#53
+
+Non-authoritative answer:
+Name:   google.com
+Address: 216.58.211.110 [美国 Google全球边缘网络]
+Name:   google.com
+Address: 2a00:1450:400e:809::200e [荷兰Amsterdam Google Inc. 服务器网段]
+```
+
+### Query CDN provider
+
+```
+$ nslookup www.gov.cn | nali
+Server:         127.0.0.53 [局域网 IP]
+Address:        127.0.0.53 [局域网 IP]#53
+
+Non-authoritative answer:
+www.gov.cn      canonical name = www.gov.cn.bsgslb.cn [白山云 CDN].
+www.gov.cn.bsgslb.cn [白山云 CDN]       canonical name = zgovweb.v.bsgslb.cn [白山云 CDN].
+Name:   zgovweb.v.bsgslb.cn [白山云 CDN]
+Address: 103.104.170.25 [新加坡 ]
+Name:   zgovweb.v.bsgslb.cn [白山云 CDN]
+Address: 2001:428:6402:21b::5 [美国Louisiana州Monroe Qwest Communications Company, LLC (CenturyLink)]
+Name:   zgovweb.v.bsgslb.cn [白山云 CDN]
+Address: 2001:428:6402:21b::6 [美国Louisiana州Monroe Qwest Communications Company, LLC (CenturyLink)]
+```
+
+## Interface
+
+After nali runs for the first time, a configuration file `config.yaml` will be generated in the working directory (default `~/.nali/config.yaml`), the configuration file defines the database information.
+
+A database is defined as the follows:
+
+```yaml
+- name: geoip
+  name-alias:
+  - geolite
+  - geolite2
+  format: mmdb
+  file: GeoLite2-City.mmdb
+  languages:
+  - ALL
+  types:
+  - IPv4
+  - IPv6
+```
+
+### Help
+
+```
+$ nali --help
+Usage:
+  nali [flags]
+  nali [command]
+
+Available Commands:
+  completion  generate the autocompletion script for the specified shell
+  help        Help about any command
+  update      update chunzhen ip database
+
+Flags:
+      --gbk    Use GBK decoder
+  -h, --help   help for nali
+
+Use "nali [command] --help" for more information about a command.
+```
+
+### Update database
+
+Update all databases if avaliable:
+
+```
+$ nali update
+2020/07/17 12:53:46 正在下载最新纯真 IP 库...
+2020/07/17 12:54:05 已将最新的纯真 IP 库保存到本地 /root/.nali/qqwry.dat
+```
+
+Updata selected databases:
+
+```
+$ nali update --db qqwry,cdn
+2020/07/17 12:53:46 正在下载最新纯真 IP 库...
+2020/07/17 12:54:05 已将最新的纯真 IP 库保存到本地 /root/.nali/qqwry.dat
+```
+
+### Select database
+
+Users can specify which database to use， set environment variables `NALI_DB_IP4`, `NALI_DB_IP6` or both.
+
+supported database:
+
+- Geoip2 `['geoip', 'geoip2']`
+- Chunzhen `['chunzhen', 'qqwry']`
+- IPIP `['ipip']`
+- Ip2Resion `['ip2region', 'i2r']`
+- DBIP `['dbip', 'db-ip']`
+- IP2Location `['ip2location']`
+
+#### Windows
+
+##### Use geoip db
+
+```
+set NALI_DB_IP4=geoip
+
+or use powershell
+
+$env:NALI_DB_IP4="geoip"
+```
+
+##### Use ipip db
+
+```
+set NALI_DB_IP6=ipip
+
+or use powershell
+
+$env:NALI_DB_IP6="ipip"
+```
+
+#### Linux
+
+##### Use geoip db
+
+```
+export NALI_DB_IP4=geoip
+```
+
+##### Use ipip db
+
+```
+export NALI_DB_IP6=ipip
+```
+
+### Multilingual support
+
+Specify the language to be used by modifying the environment variable `NALI_LANG`, when using a non-Chinese language only the GeoIP2 database is supported
+
+The values that can be set for this parameter can be found in the list of supported databases for GeoIP2
+
+```
+# NALI_LANG=en nali 1.1.1.1
+1.1.1.1 [Australia]
+```
+
+### Change database directory
+
+If the database directory is not specified, the database will be placed in `~/.nali`
+
+Set environment variables `NALI_HOME` to specify the working directory
+
+```
+set NALI_HOME=D:\nalidb
+
+or
+
+export NALI_HOME=/home/nali
+```
+
+## Thanks
+
+- [纯真QQIP离线数据库](http://www.cz88.net/fox/ipdat.shtml)
+- [qqwry mirror](https://qqwry.mirror.noc.one/)
+- [qqwry纯真数据库解析](https://github.com/yinheli/qqwry)
+- [ZX公网ipv6数据库](https://ip.zxinc.org/ipquery/)
+- [Geoip2 city数据库](https://www.maxmind.com/en/geoip2-precision-city-service)
+- [geoip2-golang解析器](https://github.com/oschwald/geoip2-golang)
+- [CDN provider数据库](https://github.com/SukkaLab/cdn)
+- [IPIP数据库](https://www.ipip.net/product/ip.html)
+- [IPIP数据库解析](https://github.com/ipipdotnet/ipdb-go)
+- [ip2region数据库](https://github.com/lionsoul2014/ip2region)
+- [IP2Location DB3 LITE](https://lite.ip2location.com/database/db3-ip-country-region-city) *use the IPv6 BIN as it contains both IPv4 & IPv6*
+- [Cobra CLI库](https://github.com/spf13/cobra)
+- [Nali-cli](https://github.com/SukkaW/nali-cli)
+
+Thanks to JetBrains for the Open Source License 
+
+<a href="https://www.jetbrains.com/?from=nali">
+  <img src="assets/GoLand.svg">
+</a>
+
+## Author
+
+**Nali** © [zu1k](https://github.com/zu1k), Released under the [MIT](./LICENSE) License.<br>
+
+> Blog [zu1k.com](https://zu1k.com) · GitHub [@zu1k](https://github.com/zu1k) · Twitter [@zu1k_lv](https://twitter.com/zu1k_lv) · Telegram Channel [@peekfun](https://t.me/peekfun)
+
+
+## Stargazers over time
+
+[![Stargazers over time](https://starchart.cc/zu1k/nali.svg)](https://starchart.cc/zu1k/nali)
\ No newline at end of file

assets/GoLand.svg

+<svg height="105" viewBox="0 0 105 105" width="105" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink"><linearGradient id="a" gradientUnits="userSpaceOnUse" x1="99.03" x2="61.54" y1="59.18" y2="90.91"><stop offset="0" stop-color="#087cfa"/><stop offset=".02" stop-color="#0d7bfa"/><stop offset=".37" stop-color="#5566f9"/><stop offset=".66" stop-color="#8a57f8"/><stop offset=".88" stop-color="#ab4ef7"/><stop offset="1" stop-color="#b74af7"/></linearGradient><linearGradient id="b" x1="37.54" x2="60.33" xlink:href="#a" y1="34.26" y2="8.33"/><linearGradient id="c" gradientUnits="userSpaceOnUse" x1="17.84" x2="87.09" y1="88.36" y2="43.86"><stop offset="0" stop-color="#087cfa"/><stop offset=".1" stop-color="#1598d3"/><stop offset=".22" stop-color="#23b6aa"/><stop offset=".35" stop-color="#2dcc8b"/><stop offset=".46" stop-color="#35dd74"/><stop offset=".57" stop-color="#39e767"/><stop offset=".67" stop-color="#3bea62"/></linearGradient><path d="m89.06 41.63 11.44 25.37-19.58 33.5-12.06-29.28z" fill="url(#a)"/><path d="m68.86 71.22 12.06 29.28-30.34-10.43z" fill="#b74af7"/><path d="m72.13 31.61-6.22-27.11h-34.49l-26.92 41.18 7.79 18.92-7.79 17.27 55.75-47.43z" fill="url(#b)"/><path d="m100.5 24.88-40.25 9.56-55.75 47.43 35.96 18.63 28.4-29.28z" fill="url(#c)"/><path d="m22.5 22.5h60v60h-60z"/><g fill="#fff"><path d="m28.13 71.25h22.5v3.75h-22.5z"/><path d="m26.44 42.81v-.07a12.7 12.7 0 0 1 12.97-12.74 13.62 13.62 0 0 1 9.73 3.39l-3.43 4.14a9.13 9.13 0 0 0 -6.47-2.53c-4 0-7.11 3.5-7.11 7.7v.07c0 4.53 3.11 7.85 7.5 7.85a8.67 8.67 0 0 0 5.12-1.49v-3.56h-5.48v-4.7h10.73v10.74a16.07 16.07 0 0 1 -10.57 3.93c-7.69 0-12.99-5.41-12.99-12.73z"/><path d="m52.3 42.81v-.07a12.79 12.79 0 0 1 13.19-12.74c7.63 0 13.11 5.69 13.11 12.72v.07a12.79 12.79 0 0 1 -13.18 12.8c-7.64-.05-13.12-5.74-13.12-12.78zm20.61 0v-.07a7.51 7.51 0 0 0 -7.49-7.74 7.4 7.4 0 0 0 -7.42 7.67v.07a7.52 7.52 0 0 0 7.5 7.78 7.4 7.4 0 0 0 7.41-7.71z"/></g></svg>
\ No newline at end of file

cmd/root.go

+package cmd
+
+import (
+	"bufio"
+	"fmt"
+	"log"
+	"os"
+	"strings"
+
+	"github.com/fatih/color"
+	"github.com/spf13/cobra"
+	"golang.org/x/text/encoding/simplifiedchinese"
+	"golang.org/x/text/transform"
+
+	"github.com/zu1k/nali/internal/constant"
+	"github.com/zu1k/nali/pkg/entity"
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "nali",
+	Short: "An offline tool for querying IP geographic information",
+	Long: `An offline tool for querying IP geographic information.
+
+Find document on: https://github.com/zu1k/nali
+
+#1 Query a simple IP address
+
+	$ nali 1.2.3.4
+
+  or use pipe
+
+	$ echo IP 6.6.6.6 | nali
+
+#2 Query multiple IP addresses
+
+	$ nali 1.2.3.4 4.3.2.1 123.23.3.0
+
+#3 Interactive query
+
+	$ nali
+	123.23.23.23
+	123.23.23.23 [越南 越南邮电集团公司]
+	quit
+
+#4 Use with dig
+
+	$ dig nali.zu1k.com +short | nali
+
+#5 Use with nslookup
+
+	$ nslookup nali.zu1k.com 8.8.8.8 | nali
+
+#6 Use with any other program
+
+	bash abc.sh | nali
+
+#7 IPV6 support
+`,
+	Version: constant.Version,
+	Args:    cobra.MinimumNArgs(0),
+	Run: func(cmd *cobra.Command, args []string) {
+		gbk, _ := cmd.Flags().GetBool("gbk")
+
+		if len(args) == 0 {
+			stdin := bufio.NewScanner(os.Stdin)
+			for stdin.Scan() {
+				line := stdin.Text()
+				if gbk {
+					line, _, _ = transform.String(simplifiedchinese.GBK.NewDecoder(), line)
+				}
+				if line == "quit" || line == "exit" {
+					return
+				}
+				_, _ = fmt.Fprintf(color.Output, "%s\n", entity.ParseLine(line).ColorString())
+			}
+		} else {
+			_, _ = fmt.Fprintf(color.Output, "%s\n", entity.ParseLine(strings.Join(args, " ")).ColorString())
+		}
+	},
+}
+
+// Execute parse subcommand and run
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		log.Fatal(err.Error())
+	}
+}
+
+func init() {
+	rootCmd.Flags().Bool("gbk", false, "Use GBK decoder")
+}

cmd/update.go

+package cmd
+
+import (
+	"strings"
+
+	"github.com/zu1k/nali/internal/db"
+
+	"github.com/spf13/cobra"
+)
+
+// updateCmd represents the update command
+var updateCmd = &cobra.Command{
+	Use:     "update",
+	Short:   "update qqwry, zxipv6wry, ip2region ip database and cdn",
+	Long:    `update qqwry, zxipv6wry, ip2region ip database and cdn. Use commas to separate`,
+	Example: "nali update --db qqwry,cdn",
+	Run: func(cmd *cobra.Command, args []string) {
+		DBs, _ := cmd.Flags().GetString("db")
+		var DBNameArray []string
+		if DBs != "" {
+			DBNameArray = strings.Split(DBs, ",")
+		}
+		db.UpdateDB(DBNameArray...)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(updateCmd)
+	rootCmd.PersistentFlags().String("db", "", "choose db you want to update")
+}

go.mod

+module github.com/zu1k/nali
+
+go 1.18
+
+require (
+	github.com/fatih/color v1.13.0
+	github.com/ip2location/ip2location-go/v9 v9.4.0
+	github.com/ipipdotnet/ipdb-go v1.3.1
+	github.com/lionsoul2014/ip2region v2.2.0-release+incompatible
+	github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20220705082050-0f0c0cb4a5a6
+	github.com/oschwald/geoip2-golang v1.7.0
+	github.com/saracen/go7z v0.0.0-20191010121135-9c09b6bd7fda
+	github.com/spf13/cobra v1.5.0
+	github.com/spf13/viper v1.12.0
+	golang.org/x/text v0.3.7
+	gopkg.in/yaml.v2 v2.4.0
+)
+
+require (
+	github.com/fsnotify/fsnotify v1.5.4 // indirect
+	github.com/hashicorp/hcl v1.0.0 // indirect
+	github.com/inconshreveable/mousetrap v1.0.0 // indirect
+	github.com/magiconair/properties v1.8.6 // indirect
+	github.com/mattn/go-colorable v0.1.12 // indirect
+	github.com/mattn/go-isatty v0.0.14 // indirect
+	github.com/mitchellh/mapstructure v1.5.0 // indirect
+	github.com/oschwald/maxminddb-golang v1.9.0 // indirect
+	github.com/pelletier/go-toml v1.9.5 // indirect
+	github.com/pelletier/go-toml/v2 v2.0.2 // indirect
+	github.com/saracen/go7z-fixtures v0.0.0-20190623165746-aa6b8fba1d2f // indirect
+	github.com/saracen/solidblock v0.0.0-20190426153529-45df20abab6f // indirect
+	github.com/spf13/afero v1.8.2 // indirect
+	github.com/spf13/cast v1.5.0 // indirect
+	github.com/spf13/jwalterweatherman v1.1.0 // indirect
+	github.com/spf13/pflag v1.0.5 // indirect
+	github.com/subosito/gotenv v1.4.0 // indirect
+	github.com/ulikunitz/xz v0.5.10 // indirect
+	golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e // indirect
+	golang.org/x/tools v0.1.7 // indirect
+	gopkg.in/ini.v1 v1.66.6 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)

internal/config/config.go

+package config
+
+import (
+	"log"
+
+	"github.com/spf13/viper"
+	"github.com/zu1k/nali/internal/db"
+)
+
+func ReadConfig(basePath string) {
+	viper.SetDefault("databases", db.GetDefaultDBList())
+	viper.SetDefault("selected.ipv4", "qqwry")
+	viper.SetDefault("selected.ipv6", "zxipv6wry")
+	viper.SetDefault("selected.cdn", "cdn")
+	viper.SetDefault("selected.lang", "zh-CN")
+
+	viper.SetConfigName("config")
+	viper.SetConfigType("yaml")
+	viper.AddConfigPath(basePath)
+	err := viper.ReadInConfig()
+	if err != nil {
+		err = viper.SafeWriteConfig()
+		if err != nil {
+			panic(err)
+		}
+	}
+
+	_ = viper.BindEnv("selected.ipv4", "NALI_DB_IP4")
+	_ = viper.BindEnv("selected.ipv6", "NALI_DB_IP6")
+	_ = viper.BindEnv("selected.cdn", "NALI_DB_CDN")
+	_ = viper.BindEnv("selected.lang", "NALI_LANG")
+
+	dbList := db.List{}
+	err = viper.UnmarshalKey("databases", &dbList)
+	if err != nil {
+		log.Fatalln("Config invalid:", err)
+	}
+
+	db.NameDBMap.From(dbList)
+	db.TypeDBMap.From(dbList)
+}

internal/constant/path.go

+package constant
+
+import (
+	"log"
+	"os"
+	"path/filepath"
+)
+
+var (
+	// WorkDirPath database home path
+	WorkDirPath string
+)
+
+func init() {
+	WorkDirPath = os.Getenv("NALI_HOME")
+	if WorkDirPath == "" {
+		WorkDirPath = os.Getenv("NALI_DB_HOME")
+	}
+	if WorkDirPath == "" {
+		homeDir, err := os.UserHomeDir()
+		if err != nil {
+			panic(err)
+		}
+		WorkDirPath = filepath.Join(homeDir, ".nali")
+	}
+	if _, err := os.Stat(WorkDirPath); os.IsNotExist(err) {
+		if err := os.MkdirAll(WorkDirPath, 0777); err != nil {
+			log.Fatal("can not create", WorkDirPath, ", use bin dir instead")
+		}
+	}
+
+	_ = os.Chdir(WorkDirPath)
+}

internal/constant/version.go

+package constant
+
+var (
+	// Version like 1.0.1
+	Version = "unknown version"
+)

internal/db/cache.go

+package db
+
+import "github.com/zu1k/nali/pkg/dbif"
+
+var (
+	dbNameCache = make(map[string]dbif.DB)
+	dbTypeCache = make(map[dbif.QueryType]dbif.DB)
+	queryCache  = make(map[string]string)
+)
+
+var (
+	NameDBMap = make(NameMap)
+	TypeDBMap = make(TypeMap)
+)

internal/db/db.go

+package db
+
+import (
+	"log"
+	"strings"
+
+	"github.com/spf13/viper"
+
+	"github.com/zu1k/nali/pkg/cdn"
+	"github.com/zu1k/nali/pkg/dbif"
+	"github.com/zu1k/nali/pkg/geoip"
+	"github.com/zu1k/nali/pkg/qqwry"
+	"github.com/zu1k/nali/pkg/zxipv6wry"
+)
+
+func GetDB(typ dbif.QueryType) (db dbif.DB) {
+	if db, found := dbTypeCache[typ]; found {
+		return db
+	}
+
+	lang := viper.GetString("selected.lang")
+	if lang == "" {
+		lang = "zh-CN"
+	}
+
+	var err error
+	switch typ {
+	case dbif.TypeIPv4:
+		selected := viper.GetString("selected.ipv4")
+		if selected != "" {
+			db = getDbByName(selected).get()
+			break
+		}
+
+		if lang == "zh-CN" {
+			db, err = qqwry.NewQQwry(getDbByName("qqwry").File)
+		} else {
+			db, err = geoip.NewGeoIP(getDbByName("geoip").File)
+		}
+	case dbif.TypeIPv6:
+		selected := viper.GetString("selected.ipv6")
+		if selected != "" {
+			db = getDbByName(selected).get()
+			break
+		}
+
+		if lang == "zh-CN" {
+			db, err = zxipv6wry.NewZXwry(getDbByName("zxipv6wry").File)
+		} else {
+			db, err = geoip.NewGeoIP(getDbByName("geoip").File)
+		}
+	case dbif.TypeDomain:
+		selected := viper.GetString("selected.cdn")
+		if selected != "" {
+			db = getDbByName(selected).get()
+			break
+		}
+
+		db, err = cdn.NewCDN(getDbByName("cdn").File)
+	default:
+		panic("Query type not supported!")
+	}
+
+	if err != nil || db == nil {
+		log.Fatalln("Database init failed:", err)
+	}
+
+	dbTypeCache[typ] = db
+	return
+}
+
+func Find(typ dbif.QueryType, query string) string {
+	if result, found := queryCache[query]; found {
+		return result
+	}
+	result, err := GetDB(typ).Find(query)
+	if err != nil {
+		return ""
+	}
+	r := strings.Trim(result.String(), " ")
+	queryCache[query] = r
+	return r
+}

internal/db/default.go

+package db
+
+import (
+	"github.com/zu1k/nali/pkg/cdn"
+	"github.com/zu1k/nali/pkg/ip2region"
+)
+
+func GetDefaultDBList() List {
+	return List{
+		&DB{
+			Name: "qqwry",
+			NameAlias: []string{
+				"chunzhen",
+			},
+			Format:    FormatQQWry,
+			File:      "qqwry.dat",
+			Languages: LanguagesZH,
+			Types:     TypesIPv4,
+		},
+		&DB{
+			Name: "zxipv6wry",
+			NameAlias: []string{
+				"zxipv6",
+				"zx",
+			},
+			Format:    FormatZXIPv6Wry,
+			File:      "zxipv6wry.db",
+			Languages: LanguagesZH,
+			Types:     TypesIPv6,
+		},
+		&DB{
+			Name: "geoip",
+			NameAlias: []string{
+				"geoip2",
+				"geolite",
+				"geolite2",
+			},
+			Format:    FormatMMDB,
+			File:      "GeoLite2-City.mmdb",
+			Languages: LanguagesAll,
+			Types:     TypesIP,
+		},
+		&DB{
+			Name: "dbip",
+			NameAlias: []string{
+				"db-ip",
+			},
+			Format:    FormatMMDB,
+			File:      "dbip.mmdb",
+			Languages: LanguagesAll,
+			Types:     TypesIP,
+		},
+		&DB{
+			Name:      "ipip",
+			Format:    FormatIPIP,
+			File:      "ipipfree.ipdb",
+			Languages: LanguagesZH,
+			Types:     TypesIP,
+		},
+		&DB{
+			Name: "ip2region",
+			NameAlias: []string{
+				"i2r",
+			},
+			Format:       FormatIP2Region,
+			File:         "ip2region.xdb",
+			Languages:    LanguagesZH,
+			Types:        TypesIPv4,
+			DownloadUrls: ip2region.DownloadUrls,
+		},
+		&DB{
+			Name:      "ip2location",
+			Format:    FormatIP2Location,
+			File:      "IP2LOCATION-LITE-DB3.IPV6.BIN",
+			Languages: LanguagesEN,
+			Types:     TypesIP,
+		},
+
+		&DB{
+			Name:         "cdn",
+			Format:       FormatCDNYml,
+			File:         "cdn.yml",
+			Languages:    LanguagesZH,
+			Types:        TypesCDN,
+			DownloadUrls: cdn.DownloadUrls,
+		},
+	}
+}

internal/db/type.go

+package db
+
+import (
+	"log"
+
+	"github.com/zu1k/nali/pkg/cdn"
+	"github.com/zu1k/nali/pkg/dbif"
+	"github.com/zu1k/nali/pkg/geoip"
+	"github.com/zu1k/nali/pkg/ip2location"
+	"github.com/zu1k/nali/pkg/ip2region"
+	"github.com/zu1k/nali/pkg/ipip"
+	"github.com/zu1k/nali/pkg/qqwry"
+	"github.com/zu1k/nali/pkg/zxipv6wry"
+)
+
+type DB struct {
+	Name      string
+	NameAlias []string `yaml:"name-alias,omitempty" mapstructure:"name-alias"`
+	Format    Format
+	File      string
+
+	Languages []string
+	Types     []Type
+
+	DownloadUrls []string `yaml:"download-urls,omitempty" mapstructure:"download-urls"`
+}
+
+func (d *DB) get() (db dbif.DB) {
+	if db, found := dbNameCache[d.Name]; found {
+		return db
+	}
+
+	filePath := d.File
+
+	var err error
+	switch d.Format {
+	case FormatQQWry:
+		db, err = qqwry.NewQQwry(filePath)
+	case FormatZXIPv6Wry:
+		db, err = zxipv6wry.NewZXwry(filePath)
+	case FormatIPIP:
+		db, err = ipip.NewIPIP(filePath)
+	case FormatMMDB:
+		db, err = geoip.NewGeoIP(filePath)
+	case FormatIP2Region:
+		db, err = ip2region.NewIp2Region(filePath)
+	case FormatIP2Location:
+		db, err = ip2location.NewIP2Location(filePath)
+	case FormatCDNYml:
+		db, err = cdn.NewCDN(filePath)
+	default:
+		panic("DB format not supported!")
+	}
+
+	if err != nil || db == nil {
+		log.Fatalln("Database init failed:", err)
+	}
+
+	dbNameCache[d.Name] = db
+	return
+}
+
+type Format string
+
+const (
+	FormatMMDB        Format = "mmdb"
+	FormatQQWry              = "qqwry"
+	FormatZXIPv6Wry          = "zxipv6wry"
+	FormatIPIP               = "ipip"
+	FormatIP2Region          = "ip2region"
+	FormatIP2Location        = "ip2location"
+
+	FormatCDNYml = "cdn-yml"
+)
+
+var (
+	LanguagesAll = []string{"ALL"}
+	LanguagesZH  = []string{"zh-CN"}
+	LanguagesEN  = []string{"en"}
+)
+
+type Type string
+
+const (
+	TypeIPv4 Type = "IPv4"
+	TypeIPv6      = "IPv6"
+	TypeCDN       = "CDN"
+)
+
+var (
+	TypesAll  = []Type{TypeIPv4, TypeIPv6, TypeCDN}
+	TypesIP   = []Type{TypeIPv4, TypeIPv6}
+	TypesIPv4 = []Type{TypeIPv4}
+	TypesIPv6 = []Type{TypeIPv6}
+	TypesCDN  = []Type{TypeCDN}
+)
+
+type List []*DB
+type NameMap map[string]*DB
+type TypeMap map[Type][]*DB
+
+func (m *NameMap) From(dbs List) {
+	for _, db := range dbs {
+		(*m)[db.Name] = db
+
+		if alias := db.NameAlias; alias != nil {
+			for _, aName := range alias {
+				(*m)[aName] = db
+			}
+		}
+	}
+}
+
+func (m *TypeMap) From(dbs List) {
+	for _, db := range dbs {
+		for _, typ := range db.Types {
+			dbsInType := (*m)[typ]
+			if dbsInType == nil {
+				dbsInType = []*DB{db}
+			} else {
+				dbsInType = append(dbsInType, db)
+			}
+			(*m)[typ] = dbsInType
+		}
+	}
+}
+
+func getDbByName(name string) (db *DB) {
+	if dbInfo, found := NameDBMap[name]; found {
+		return dbInfo
+	}
+
+	defaultNameDBMap := NameMap{}
+	defaultNameDBMap.From(GetDefaultDBList())
+	if dbInfo, found := defaultNameDBMap[name]; found {
+		return dbInfo
+	}
+
+	log.Fatalf("DB with name %s not found!\n", name)
+	return
+}

internal/db/update.go

+package db
+
+import (
+	"log"
+	"strings"
+	"time"
+
+	"github.com/zu1k/nali/pkg/cdn"
+	"github.com/zu1k/nali/pkg/download"
+	"github.com/zu1k/nali/pkg/ip2region"
+	"github.com/zu1k/nali/pkg/qqwry"
+	"github.com/zu1k/nali/pkg/zxipv6wry"
+)
+
+func UpdateDB(dbNames ...string) {
+	if len(dbNames) == 0 {
+		dbNames = DbNameListForUpdate
+	}
+
+	done := make(map[string]struct{})
+	for _, dbName := range dbNames {
+		update, name := getUpdateFuncByName(dbName)
+		if _, found := done[name]; !found {
+			done[name] = struct{}{}
+			if err := update(); err != nil {
+				continue
+			}
+		}
+	}
+}
+
+var DbNameListForUpdate = []string{
+	"qqwry",
+	"zxipv6wry",
+	"ip2region",
+	"cdn",
+}
+
+func getUpdateFuncByName(name string) (func() error, string) {
+	name = strings.TrimSpace(name)
+	if db := getDbByName(name); db != nil {
+		// direct download if download-url not null
+		if len(db.DownloadUrls) > 0 {
+			return func() error {
+				log.Printf("正在下载最新 %s 数据库...\n", db.Name)
+				_, err := download.Download(db.File, db.DownloadUrls...)
+				if err != nil {
+					log.Printf("%s 数据库下载失败: %s\n", db.Name, db.File)
+					log.Println("error:", err)
+					return err
+				} else {
+					log.Printf("%s 数据库下载成功: %s\n", db.Name, db.File)
+					return nil
+				}
+			}, string(db.Format)
+		}
+
+		// intenel download func
+		switch db.Format {
+		case FormatQQWry:
+			return func() error {
+				log.Println("正在下载最新 纯真 IPv4数据库...")
+				_, err := qqwry.Download(getDbByName("qqwry").File)
+				if err != nil {
+					log.Println("数据库 QQWry 下载失败:", err)
+				}
+				return err
+			}, FormatQQWry
+		case FormatZXIPv6Wry:
+			return func() error {
+				log.Println("正在下载最新 ZX IPv6数据库...")
+				_, err := zxipv6wry.Download(getDbByName("zxipv6wry").File)
+				if err != nil {
+					log.Println("数据库 ZXIPv6Wry 下载失败:", err)
+				}
+				return err
+			}, FormatZXIPv6Wry
+		case FormatIP2Region:
+			return func() error {
+				log.Println("正在下载最新 Ip2Region 数据库...")
+				_, err := ip2region.Download(getDbByName("ip2region").File)
+				if err != nil {
+					log.Println("数据库 Ip2Region 下载失败:", err)
+				}
+				return err
+			}, FormatZXIPv6Wry
+		case FormatCDNYml:
+			return func() error {
+				log.Println("正在下载最新 CDN服务提供商数据库...")
+				_, err := cdn.Download(getDbByName("cdn").File)
+				if err != nil {
+					log.Println("数据库 CDN 下载失败:", err)
+				}
+				return err
+			}, FormatZXIPv6Wry
+		default:
+			return func() error {
+				log.Println("暂不支持该类型数据库的自动更新")
+				log.Println("可通过指定数据库的 download-urls 从特定链接下载数据库文件")
+				return nil
+			}, time.Now().String()
+		}
+	} else {
+		return func() error {
+			log.Fatalln("该名称的数据库未找到：", name)
+			return nil
+		}, time.Now().String()
+	}
+}

internal/migration/v4.go

+package migration
+
+import (
+	"log"
+
+	"github.com/spf13/viper"
+	"github.com/zu1k/nali/internal/constant"
+	"github.com/zu1k/nali/internal/db"
+	"github.com/zu1k/nali/pkg/cdn"
+	"github.com/zu1k/nali/pkg/ip2region"
+)
+
+func init() {
+	viper.SetConfigName("config")
+	viper.SetConfigType("yaml")
+	viper.AddConfigPath(constant.WorkDirPath)
+
+	err := viper.ReadInConfig()
+	if err != nil {
+		err = viper.SafeWriteConfig()
+		if err != nil {
+			panic(err)
+		}
+	}
+
+	dbList := db.List{}
+	err = viper.UnmarshalKey("databases", &dbList)
+	if err != nil {
+		log.Fatalln("Config invalid:", err)
+	}
+
+	needOverwrite := false
+	for _, adb := range dbList {
+		if adb.Name == "ip2region" && adb.File != "ip2region.xdb" {
+			needOverwrite = true
+			adb.File = "ip2region.xdb"
+			adb.DownloadUrls = ip2region.DownloadUrls
+		}
+
+		if adb.Name == "cdn" && adb.Format != "cdn-yml" {
+			needOverwrite = true
+			adb.Format = "cdn-yml"
+			adb.DownloadUrls = cdn.DownloadUrls
+		}
+	}
+
+	if needOverwrite {
+		viper.Set("databases", dbList)
+		err = viper.WriteConfig()
+		if err != nil {
+			log.Println(err)
+		}
+	}
+}

main.go

+package main
+
+import (
+	"github.com/zu1k/nali/cmd"
+	"github.com/zu1k/nali/internal/config"
+	"github.com/zu1k/nali/internal/constant"
+	_ "github.com/zu1k/nali/internal/migration"
+)
+
+func main() {
+	config.ReadConfig(constant.WorkDirPath)
+	cmd.Execute()
+}

pkg/cdn/cdn.go

+package cdn
+
+import (
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"os"
+	"regexp"
+	"strings"
+
+	"gopkg.in/yaml.v2"
+
+	"github.com/zu1k/nali/pkg/re"
+)
+
+type CDN struct {
+	Map   map[string]CDNResult
+	ReMap []CDNReTuple
+}
+
+type CDNReTuple struct {
+	*regexp.Regexp
+	CDNResult
+}
+
+type CDNResult struct {
+	Name string `yaml:"name"`
+	Link string `yaml:"link"`
+}
+
+func (r CDNResult) String() string {
+	return r.Name
+}
+
+func NewCDN(filePath string) (*CDN, error) {
+	fileData := make([]byte, 0)
+	_, err := os.Stat(filePath)
+	if err != nil && os.IsNotExist(err) {
+		log.Println("文件不存在，尝试从网络获取最新CDN数据库")
+		fileData, err = Download(filePath)
+		if err != nil {
+			return nil, err
+		}
+	} else {
+		cdnFile, err := os.OpenFile(filePath, os.O_RDONLY, 0400)
+		if err != nil {
+			return nil, err
+		}
+		defer cdnFile.Close()
+
+		fileData, err = ioutil.ReadAll(cdnFile)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	cdnMap := make(map[string]CDNResult)
+	err = yaml.Unmarshal(fileData, &cdnMap)
+	if err != nil {
+		return nil, err
+	}
+	cdnReMap := make([]CDNReTuple, 0)
+	for k, v := range cdnMap {
+		if re.MaybeRegexp(k) {
+			rex, err := regexp.Compile(k)
+			if err != nil {
+				log.Printf("[CDN Database] entry %s not a valid regexp", k)
+			}
+			cdnReMap = append(cdnReMap, CDNReTuple{
+				Regexp:    rex,
+				CDNResult: v,
+			})
+		}
+	}
+
+	return &CDN{Map: cdnMap, ReMap: cdnReMap}, nil
+}
+
+func (db CDN) Find(query string, params ...string) (result fmt.Stringer, err error) {
+	baseCname := parseBaseCname(query)
+	for _, domain := range baseCname {
+		if domain != "" {
+			cdnResult, found := db.Map[domain]
+			if found {
+				return cdnResult, nil
+			}
+		}
+
+		for _, entry := range db.ReMap {
+			if entry.Regexp.MatchString(domain) {
+				return entry.CDNResult, nil
+			}
+		}
+	}
+
+	return nil, errors.New("not found")
+}
+
+func parseBaseCname(domain string) (result []string) {
+	parts := strings.Split(domain, ".")
+	size := len(parts)
+	if size == 0 {
+		return []string{}
+	}
+	domain = parts[size-1]
+	result = append(result, domain)
+	for i := len(parts) - 2; i >= 0; i-- {
+		domain = parts[i] + "." + domain
+		result = append(result, domain)
+	}
+	return result
+}

pkg/cdn/update.go

+package cdn
+
+import (
+	"log"
+
+	"github.com/zu1k/nali/pkg/common"
+)
+
+var DownloadUrls = []string{
+	"https://cdn.jsdelivr.net/gh/4ft35t/cdn/src/cdn.yml",
+	"https://raw.githubusercontent.com/4ft35t/cdn/master/src/cdn.yml",
+	"https://raw.githubusercontent.com/SukkaLab/cdn/master/src/cdn.yml",
+}
+
+// Deprecated: This will be removed from 0.5.0, use package download instead
+func Download(filePath ...string) (data []byte, err error) {
+	data, err = common.GetHttpClient().Get(DownloadUrls...)
+	if err != nil {
+		log.Printf("CDN数据库下载失败，请手动下载解压后保存到本地: %s \n", filePath)
+		log.Println("下载链接：", DownloadUrls)
+		return
+	}
+
+	if len(filePath) == 1 {
+		if err := common.SaveFile(filePath[0], data); err == nil {
+			log.Printf("已将最新的 CDN数据库 保存到本地: %s \n", filePath)
+		}
+	}
+	return
+}

pkg/common/const.go

+package common
+
+const (
+	// RedirectMode1 [IP][0x01][国家和地区信息的绝对偏移地址]
+	RedirectMode1 = 0x01
+	// RedirectMode2 [IP][0x02][信息的绝对偏移][...] or [IP][国家][...]
+	RedirectMode2 = 0x02
+)

pkg/common/dbtool.go

+package common
+
+func ByteToUInt32(data []byte) uint32 {
+	i := uint32(data[0]) & 0xff
+	i |= (uint32(data[1]) << 8) & 0xff00
+	i |= (uint32(data[2]) << 16) & 0xff0000
+	return i
+}

pkg/common/httpclient.go

+package common
+
+import (
+	"io/ioutil"
+	"log"
+	"net/http"
+	"time"
+)
+
+const UserAgent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.0.0 Safari/537.36"
+
+type HttpClient struct {
+	*http.Client
+}
+
+var httpClient *HttpClient
+
+func init() {
+	httpClient = &HttpClient{http.DefaultClient}
+	httpClient.Timeout = time.Second * 60
+	httpClient.Transport = &http.Transport{
+		TLSHandshakeTimeout:   time.Second * 5,
+		IdleConnTimeout:       time.Second * 10,
+		ResponseHeaderTimeout: time.Second * 10,
+		ExpectContinueTimeout: time.Second * 20,
+	}
+}
+
+func GetHttpClient() *HttpClient {
+	c := *httpClient
+	return &c
+}
+
+func (c *HttpClient) Get(urls ...string) (body []byte, err error) {
+	var req *http.Request
+	var resp *http.Response
+
+	for _, url := range urls {
+		req, err = http.NewRequest(http.MethodGet, url, nil)
+		if err != nil {
+			log.Println(err)
+			continue
+		}
+		req.Header.Set("User-Agent", UserAgent)
+		resp, err = c.Do(req)
+
+		if err == nil && resp != nil && resp.StatusCode == 200 {
+			defer resp.Body.Close()
+			body, err = ioutil.ReadAll(resp.Body)
+			if err != nil {
+				continue
+			}
+			return
+		}
+	}
+
+	return nil, err
+}

pkg/common/savefile.go

+package common
+
+import (
+	"io/ioutil"
+	"log"
+	"os"
+)
+
+func SaveFile(path string, data []byte) (err error) {
+	// Remove file if exist
+	_, err = os.Stat(path)
+	if err == nil {
+		err = os.Remove(path)
+		if err != nil {
+			log.Fatalln("旧文件删除失败", err.Error())
+		}
+	}
+
+	// save file
+	return ioutil.WriteFile(path, data, 0644)
+}

pkg/common/struct.go

+package common
+
+import (
+	"fmt"
+	"os"
+)
+
+// FileData: info of database file
+type FileData struct {
+	Data     []byte
+	FilePath string
+	FileBase *os.File
+}
+
+// IPDB common ip database
+type IPDB struct {
+	Data   *FileData
+	Offset uint32
+	IPNum  uint32
+}
+
+// setOffset 设置偏移量
+func (db *IPDB) SetOffset(offset uint32) {
+	db.Offset = offset
+}
+
+// readString 获取字符串
+func (db *IPDB) ReadString(offset uint32) []byte {
+	db.SetOffset(offset)
+	data := make([]byte, 0, 30)
+	buf := make([]byte, 1)
+	for {
+		buf = db.ReadData(1)
+		if buf[0] == 0 {
+			break
+		}
+		data = append(data, buf[0])
+	}
+	return data
+}
+
+// readData 从文件中读取数据
+func (db *IPDB) ReadData(length uint32, offset ...uint32) (rs []byte) {
+	if len(offset) > 0 {
+		db.SetOffset(offset[0])
+	}
+
+	end := db.Offset + length
+	dataNum := uint32(len(db.Data.Data))
+	if db.Offset > dataNum {
+		return nil
+	}
+
+	if end > dataNum {
+		end = dataNum
+	}
+	rs = db.Data.Data[db.Offset:end]
+	db.Offset = end
+	return
+}
+
+// readMode 获取偏移值类型
+func (db *IPDB) ReadMode(offset uint32) byte {
+	mode := db.ReadData(1, offset)
+	return mode[0]
+}
+
+// ReadUInt24
+func (db *IPDB) ReadUInt24() uint32 {
+	buf := db.ReadData(3)
+	return ByteToUInt32(buf)
+}
+
+// readArea 读取区域
+func (db *IPDB) ReadArea(offset uint32) []byte {
+	mode := db.ReadMode(offset)
+	if mode == RedirectMode1 || mode == RedirectMode2 {
+		areaOffset := db.ReadUInt24()
+		if areaOffset == 0 {
+			return []byte("")
+		}
+		return db.ReadString(areaOffset)
+	}
+	return db.ReadString(offset)
+}
+
+func GetMiddleOffset(start uint32, end uint32, indexLen uint32) uint32 {
+	records := ((end - start) / indexLen) >> 1
+	return start + records*indexLen
+}
+
+type Result struct {
+	Country string
+	Area    string
+}
+
+func (r Result) String() string {
+	return fmt.Sprintf("%s %s", r.Country, r.Area)
+}

pkg/dbif/db.go

+package dbif
+
+import (
+	"fmt"
+
+	"github.com/zu1k/nali/pkg/cdn"
+	"github.com/zu1k/nali/pkg/geoip"
+	"github.com/zu1k/nali/pkg/ip2location"
+	"github.com/zu1k/nali/pkg/ip2region"
+	"github.com/zu1k/nali/pkg/ipip"
+	"github.com/zu1k/nali/pkg/qqwry"
+	"github.com/zu1k/nali/pkg/zxipv6wry"
+)
+
+type QueryType uint
+
+const (
+	TypeIPv4 = iota
+	TypeIPv6
+	TypeDomain
+)
+
+type DB interface {
+	Find(query string, params ...string) (result fmt.Stringer, err error)
+}
+
+var (
+	_ DB = &qqwry.QQwry{}
+	_ DB = &zxipv6wry.ZXwry{}
+	_ DB = &ipip.IPIPFree{}
+	_ DB = &geoip.GeoIP{}
+	_ DB = &ip2region.Ip2Region{}
+	_ DB = &ip2location.IP2Location{}
+	_ DB = &cdn.CDN{}
+)

pkg/download/download.go

+package download
+
+import (
+	"log"
+
+	"github.com/zu1k/nali/pkg/common"
+)
+
+func Download(filePath string, urls ...string) (data []byte, err error) {
+	_ = urls[0]
+
+	data, err = common.GetHttpClient().Get(urls...)
+	if err != nil {
+		log.Printf("文件下载失败，请手动下载解压后保存到本地: %s \n", filePath)
+		log.Println("下载链接：", urls)
+		return
+	}
+
+	if err := common.SaveFile(filePath, data); err == nil {
+		log.Println("文件下载成功:", filePath)
+	}
+	return
+}

pkg/entity/entity.go

+package entity
+
+import (
+	"strings"
+
+	"github.com/fatih/color"
+	"github.com/zu1k/nali/pkg/dbif"
+)
+
+type EntityType uint
+
+const (
+	TypeIPv4   = dbif.TypeIPv4
+	TypeIPv6   = dbif.TypeIPv6
+	TypeDomain = dbif.TypeDomain
+
+	TypePlain = 100
+)
+
+type Entity struct {
+	Loc  []int // s[Loc[0]:Loc[1]]
+	Type EntityType
+
+	Text string
+	Info string
+}
+
+func (e Entity) ParseInfo() error {
+	return nil
+}
+
+type Entities []*Entity
+
+func (es Entities) Len() int {
+	return len(es)
+}
+
+func (es Entities) Less(i, j int) bool {
+	return es[i].Loc[0] < es[j].Loc[0]
+}
+
+func (es Entities) Swap(i, j int) {
+	es[i], es[j] = es[j], es[i]
+}
+
+func (es Entities) String() string {
+	var result strings.Builder
+	for _, entity := range es {
+		result.WriteString(entity.Text)
+		if entity.Type != TypePlain && len(entity.Info) > 0 {
+			result.WriteString("[" + entity.Info + "] ")
+		}
+	}
+	return result.String()
+}
+
+func (es Entities) ColorString() string {
+	var line strings.Builder
+	for _, e := range es {
+		s := e.Text
+		switch e.Type {
+		case TypeIPv4:
+			s = color.GreenString(e.Text)
+		case TypeIPv6:
+			s = color.BlueString(e.Text)
+		case TypeDomain:
+			s = color.YellowString(e.Text)
+		}
+		if e.Type != TypePlain && len(e.Info) > 0 {
+			s += " [" + color.RedString(e.Info) + "] "
+		}
+		line.WriteString(s)
+	}
+	return line.String()
+}

pkg/entity/parse.go

+package entity
+
+import (
+	"net/netip"
+	"sort"
+
+	"github.com/zu1k/nali/internal/db"
+	"github.com/zu1k/nali/pkg/dbif"
+	"github.com/zu1k/nali/pkg/re"
+)
+
+// ParseLine parse a line into entities
+func ParseLine(line string) Entities {
+	ip4sLoc := re.IPv4Re.FindAllStringIndex(line, -1)
+	ip6sLoc := re.IPv6Re.FindAllStringIndex(line, -1)
+	domainsLoc := re.DomainRe.FindAllStringIndex(line, -1)
+
+	tmp := make(Entities, 0, len(ip4sLoc)+len(ip6sLoc)+len(domainsLoc))
+	for _, e := range ip4sLoc {
+		tmp = append(tmp, &Entity{
+			Loc:  e,
+			Type: TypeIPv4,
+			Text: line[e[0]:e[1]],
+		})
+	}
+	for _, e := range ip6sLoc {
+		text := line[e[0]:e[1]]
+		if ip, _ := netip.ParseAddr(text); !ip.Is4In6() {
+			tmp = append(tmp, &Entity{
+				Loc:  e,
+				Type: TypeIPv6,
+				Text: text,
+			})
+		}
+	}
+	for _, e := range domainsLoc {
+		tmp = append(tmp, &Entity{
+			Loc:  e,
+			Type: TypeDomain,
+			Text: line[e[0]:e[1]],
+		})
+	}
+
+	sort.Sort(tmp)
+	es := make(Entities, 0, len(tmp))
+
+	idx := 0
+	for _, e := range tmp {
+		start := e.Loc[0]
+		if start >= idx {
+			if start > idx {
+				es = append(es, &Entity{
+					Loc:  []int{idx, start},
+					Type: TypePlain,
+					Text: line[idx:start],
+				})
+			}
+			e.Info = db.Find(dbif.QueryType(e.Type), e.Text)
+			es = append(es, e)
+			idx = e.Loc[1]
+		}
+	}
+	if total := len(line); idx < total {
+		es = append(es, &Entity{
+			Loc:  []int{idx, total},
+			Type: TypePlain,
+			Text: line[idx:total],
+		})
+	}
+
+	return es
+}

pkg/entity/parse_test.go

+package entity
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestParse(t *testing.T) {
+	fmt.Println(ParseLine("2001:0db8:85a3:0000:0000:8a2e:0370:7334 baidu.com 1.2.3.4 baidu.com"))
+	fmt.Println(ParseLine("a.cn.b.com.c.org d.com"))
+}
+
+func TestColorPrint(t *testing.T) {
+	fmt.Println(ParseLine("2001:0db8:85a3:0000:0000:8a2e:0370:7334 baidu.com 1.2.3.4 baidu.com").ColorString())
+}

pkg/geoip/geoip.go

+package geoip
+
+import (
+	"errors"
+	"fmt"
+	"log"
+	"net"
+	"os"
+
+	"github.com/spf13/viper"
+
+	"github.com/oschwald/geoip2-golang"
+)
+
+// GeoIP2
+type GeoIP struct {
+	db *geoip2.Reader
+}
+
+// new geoip from database file
+func NewGeoIP(filePath string) (*GeoIP, error) {
+	// 判断文件是否存在
+	_, err := os.Stat(filePath)
+	if err != nil && os.IsNotExist(err) {
+		log.Println("文件不存在，请自行下载 Geoip2 City库，并保存在", filePath)
+		return nil, err
+	} else {
+		db, err := geoip2.Open(filePath)
+		if err != nil {
+			log.Fatal(err)
+		}
+		return &GeoIP{db: db}, nil
+	}
+}
+
+func (g GeoIP) Find(query string, params ...string) (result fmt.Stringer, err error) {
+	ip := net.ParseIP(query)
+	if ip == nil {
+		return nil, errors.New("Query should be valid IP")
+	}
+	record, err := g.db.City(ip)
+	if err != nil {
+		return
+	}
+
+	lang := viper.GetString("selected.lang")
+	if lang == "" {
+		lang = "zh-CN"
+	}
+
+	result = Result{
+		Country: record.Country.Names[lang],
+		City:    record.City.Names[lang],
+	}
+	return
+}
+
+type Result struct {
+	Country string
+	City    string
+}
+
+func (r Result) String() string {
+	if r.City == "" {
+		return r.Country
+	} else {
+		return fmt.Sprintf("%s %s", r.Country, r.City)
+	}
+}

pkg/ip2location/ip2location.go

+package ip2location
+
+import (
+	"errors"
+	"fmt"
+	"log"
+	"net"
+	"os"
+
+	"github.com/ip2location/ip2location-go/v9"
+)
+
+// IP2Location
+type IP2Location struct {
+	db *ip2location.DB
+}
+
+// new IP2Location from database file
+func NewIP2Location(filePath string) (*IP2Location, error) {
+	_, err := os.Stat(filePath)
+	if err != nil && os.IsNotExist(err) {
+		log.Println("文件不存在，请自行下载 IP2Location 库，并保存在", filePath)
+		return nil, err
+	} else {
+		db, err := ip2location.OpenDB(filePath)
+
+		if err != nil {
+			log.Fatal(err)
+		}
+		return &IP2Location{db: db}, nil
+	}
+}
+
+func (x IP2Location) Find(query string, params ...string) (result fmt.Stringer, err error) {
+	ip := net.ParseIP(query)
+	if ip == nil {
+		return nil, errors.New("Query should be valid IP")
+	}
+	record, err := x.db.Get_all(ip.String())
+
+	if err != nil {
+		return
+	}
+
+	result = Result{
+		Country: record.Country_long,
+		Region:  record.Region,
+		City:    record.City,
+	}
+	return
+}
+
+type Result struct {
+	Country string
+	Region  string
+	City    string
+}
+
+func (r Result) String() string {
+	return fmt.Sprintf("%s %s %s", r.Country, r.Region, r.City)
+}

pkg/ip2region/ip2region.go

+package ip2region
+
+import (
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"os"
+	"strings"
+
+	"github.com/lionsoul2014/ip2region/binding/golang/ip2region"
+	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
+	"github.com/zu1k/nali/pkg/common"
+)
+
+type Ip2Region struct {
+	seacher *xdb.Searcher
+	db_old  *ip2region.Ip2Region
+}
+
+func NewIp2Region(filePath string) (*Ip2Region, error) {
+	_, err := os.Stat(filePath)
+	if err != nil && os.IsNotExist(err) {
+		log.Println("文件不存在，尝试从网络获取最新 ip2region 库")
+		_, err = Download(filePath)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	switch {
+	case strings.HasSuffix(filePath, ".xdb"):
+		f, err := os.OpenFile(filePath, os.O_RDONLY, 0400)
+		if err != nil {
+			return nil, err
+		}
+		defer f.Close()
+
+		data, err := ioutil.ReadAll(f)
+		if err != nil {
+			return nil, err
+		}
+		searcher, err := xdb.NewWithBuffer(data)
+		if err != nil {
+			fmt.Printf("无法解析 ip2region xdb 数据库: %s\n", err)
+			return nil, err
+		}
+		return &Ip2Region{
+			seacher: searcher,
+		}, nil
+	default:
+		region, err := ip2region.New(filePath)
+		if err != nil {
+			return nil, err
+		}
+		return &Ip2Region{
+			db_old: region,
+		}, nil
+	}
+}
+
+func (db Ip2Region) Find(query string, params ...string) (result fmt.Stringer, err error) {
+	if db.seacher != nil {
+		res, err := db.seacher.SearchByStr(query)
+		if err != nil {
+			return nil, err
+		} else {
+			return common.Result{
+				Country: strings.ReplaceAll(res, "|0", ""),
+			}, nil
+		}
+	} else if db.db_old != nil {
+		ip, err := db.db_old.MemorySearch(query)
+		if err != nil {
+			return nil, err
+		}
+
+		area := ""
+		if ip.Province != "0" {
+			area = ip.Province
+		}
+		if ip.City != "0" && strings.EqualFold(ip.City, ip.Province) {
+			area = area + " " + ip.Province
+		}
+		if ip.ISP != "0" {
+			area = area + " " + ip.ISP
+		}
+
+		return common.Result{
+			Country: ip.Country,
+			Area:    area,
+		}, nil
+	}
+
+	return nil, errors.New("ip2region 未初始化")
+}

pkg/ip2region/update.go

+package ip2region
+
+import (
+	"log"
+
+	"github.com/zu1k/nali/pkg/common"
+)
+
+var DownloadUrls = []string{
+	"https://cdn.jsdelivr.net/gh/lionsoul2014/ip2region/data/ip2region.xdb",
+	"https://raw.githubusercontent.com/lionsoul2014/ip2region/master/data/ip2region.xdb",
+}
+
+// Deprecated: This will be removed from 0.5.0, use package download instead
+func Download(filePath ...string) (data []byte, err error) {
+	data, err = common.GetHttpClient().Get(DownloadUrls...)
+	if err != nil {
+		log.Printf("CDN数据库下载失败，请手动下载解压后保存到本地: %s \n", filePath)
+		log.Println("下载链接：", DownloadUrls)
+		return
+	}
+
+	if len(filePath) == 1 {
+		if err := common.SaveFile(filePath[0], data); err == nil {
+			log.Println("已将最新的 ip2region 保存到本地:", filePath)
+		}
+	}
+	return
+}

pkg/ipip/ipip.go

+package ipip
+
+import (
+	"fmt"
+	"log"
+	"os"
+
+	"github.com/ipipdotnet/ipdb-go"
+)
+
+type IPIPFree struct {
+	*ipdb.City
+}
+
+func NewIPIP(filePath string) (*IPIPFree, error) {
+	_, err := os.Stat(filePath)
+	if err != nil && os.IsNotExist(err) {
+		log.Printf("IPIP数据库不存在，请手动下载解压后保存到本地: %s \n", filePath)
+		log.Println("下载链接： https://www.ipip.net/product/ip.html")
+		return nil, err
+	} else {
+		db, err := ipdb.NewCity(filePath)
+		if err != nil {
+			return nil, err
+		}
+		return &IPIPFree{City: db}, nil
+	}
+}
+
+type Result struct {
+	Country string
+	Region  string
+	City    string
+}
+
+func (r Result) String() string {
+	if r.City == "" {
+		return fmt.Sprintf("%s %s", r.Country, r.Region)
+	}
+	return fmt.Sprintf("%s %s %s", r.Country, r.Region, r.City)
+}
+
+func (db IPIPFree) Find(query string, params ...string) (result fmt.Stringer, err error) {
+	info, err := db.FindInfo(query, "CN")
+	if err != nil || info == nil {
+		return nil, err
+	} else {
+		// info contains more info
+		result = Result{
+			Country: info.CountryName,
+			Region:  info.RegionName,
+			City:    info.CityName,
+		}
+		return
+	}
+}

pkg/qqwry/qqwry.go

+package qqwry
+
+import (
+	"encoding/binary"
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"net"
+	"os"
+	"strings"
+
+	"github.com/zu1k/nali/pkg/common"
+	"golang.org/x/text/encoding/simplifiedchinese"
+)
+
+type QQwry struct {
+	common.IPDB
+}
+
+// NewQQwry new database from path
+func NewQQwry(filePath string) (*QQwry, error) {
+	var fileData []byte
+	var fileInfo common.FileData
+
+	_, err := os.Stat(filePath)
+	if err != nil && os.IsNotExist(err) {
+		log.Println("文件不存在，尝试从网络获取最新纯真 IP 库")
+		fileData, err = Download(filePath)
+		if err != nil {
+			return nil, err
+		}
+	} else {
+		fileInfo.FileBase, err = os.OpenFile(filePath, os.O_RDONLY, 0400)
+		if err != nil {
+			return nil, err
+		}
+		defer fileInfo.FileBase.Close()
+
+		fileData, err = ioutil.ReadAll(fileInfo.FileBase)
+		if err != nil {
+			return nil, err
+		}
+	}
+	fileInfo.Data = fileData
+
+	buf := fileInfo.Data[0:8]
+	start := binary.LittleEndian.Uint32(buf[:4])
+	end := binary.LittleEndian.Uint32(buf[4:])
+
+	return &QQwry{
+		IPDB: common.IPDB{
+			Data:  &fileInfo,
+			IPNum: (end-start)/7 + 1,
+		},
+	}, nil
+}
+
+func (db QQwry) Find(query string, params ...string) (result fmt.Stringer, err error) {
+	ip := net.ParseIP(query)
+	if ip == nil {
+		return nil, errors.New("Query should be IPv4")
+	}
+	ip4 := ip.To4()
+	if ip4 == nil {
+		return nil, errors.New("Query should be IPv4")
+	}
+	ip4uint := binary.BigEndian.Uint32(ip4)
+
+	offset := db.searchIndex(ip4uint)
+	if offset <= 0 {
+		return nil, errors.New("Query not valid")
+	}
+
+	var gbkCountry []byte
+	var gbkArea []byte
+
+	mode := db.ReadMode(offset + 4)
+	switch mode {
+	case common.RedirectMode1: // [IP][0x01][国家和地区信息的绝对偏移地址]
+		countryOffset := db.ReadUInt24()
+		mode = db.ReadMode(countryOffset)
+		if mode == common.RedirectMode2 {
+			c := db.ReadUInt24()
+			gbkCountry = db.ReadString(c)
+			countryOffset += 4
+		} else {
+			gbkCountry = db.ReadString(countryOffset)
+			countryOffset += uint32(len(gbkCountry) + 1)
+		}
+		gbkArea = db.ReadArea(countryOffset)
+	case common.RedirectMode2:
+		countryOffset := db.ReadUInt24()
+		gbkCountry = db.ReadString(countryOffset)
+		gbkArea = db.ReadArea(offset + 8)
+	default:
+		gbkCountry = db.ReadString(offset + 4)
+		gbkArea = db.ReadArea(offset + uint32(5+len(gbkCountry)))
+	}
+
+	enc := simplifiedchinese.GBK.NewDecoder()
+	country, _ := enc.String(string(gbkCountry))
+	area, _ := enc.String(string(gbkArea))
+
+	result = common.Result{
+		Country: strings.ReplaceAll(country, " CZ88.NET", ""),
+		Area:    strings.ReplaceAll(area, " CZ88.NET", ""),
+	}
+	return result, nil
+}
+
+// searchIndex 查找索引位置
+func (db *QQwry) searchIndex(ip uint32) uint32 {
+	header := db.ReadData(8, 0)
+
+	start := binary.LittleEndian.Uint32(header[:4])
+	end := binary.LittleEndian.Uint32(header[4:])
+
+	buf := make([]byte, 7)
+	mid := uint32(0)
+	ipUint := uint32(0)
+
+	for {
+		mid = common.GetMiddleOffset(start, end, 7)
+		buf = db.ReadData(7, mid)
+		ipUint = binary.LittleEndian.Uint32(buf[:4])
+
+		if end-start == 7 {
+			offset := common.ByteToUInt32(buf[4:])
+			buf = db.ReadData(7)
+			if ip < binary.LittleEndian.Uint32(buf[:4]) {
+				return offset
+			}
+			return 0
+		}
+
+		if ipUint > ip {
+			end = mid
+		} else if ipUint < ip {
+			start = mid
+		} else if ipUint == ip {
+			return common.ByteToUInt32(buf[4:])
+		}
+	}
+}

pkg/qqwry/update.go

+package qqwry
+
+import (
+	"bytes"
+	"compress/zlib"
+	"encoding/binary"
+	"fmt"
+	"io/ioutil"
+	"log"
+
+	"github.com/zu1k/nali/pkg/common"
+)
+
+func Download(filePath ...string) (data []byte, err error) {
+	fmt.Println("此方式更新的 QQWry 数据库版本过旧，请手动下载最新版纯真免费IP库: https://www.cz88.net/help")
+
+	data, err = downloadAndDecrypt()
+	if err != nil {
+		log.Printf("纯真IP库下载失败，请手动下载解压后保存到本地: %s \n", filePath)
+		log.Println("下载链接： https://qqwry.mirror.noc.one/qqwry.rar")
+		return
+	}
+
+	if len(filePath) == 1 {
+		if err := common.SaveFile(filePath[0], data); err == nil {
+			log.Println("已将最新的 纯真IP库 保存到本地:", filePath)
+		}
+	}
+	return
+}
+
+const (
+	mirror = "https://qqwry.mirror.noc.one/qqwry.rar"
+	key    = "https://qqwry.mirror.noc.one/copywrite.rar"
+)
+
+func downloadAndDecrypt() (data []byte, err error) {
+	data, err = common.GetHttpClient().Get(mirror)
+	if err != nil {
+		return nil, err
+	}
+
+	key, err := getCopyWriteKey()
+	if err != nil {
+		return nil, err
+	}
+
+	return unRar(data, key)
+}
+
+func unRar(data []byte, key uint32) ([]byte, error) {
+	for i := 0; i < 0x200; i++ {
+		key = key * 0x805
+		key++
+		key = key & 0xff
+
+		data[i] = byte(uint32(data[i]) ^ key)
+	}
+
+	reader, err := zlib.NewReader(bytes.NewReader(data))
+	if err != nil {
+		return nil, err
+	}
+
+	return ioutil.ReadAll(reader)
+}
+
+func getCopyWriteKey() (uint32, error) {
+	body, err := common.GetHttpClient().Get(key)
+	if err != nil {
+		return 0, err
+	}
+
+	return binary.LittleEndian.Uint32(body[5*4:]), nil
+}

pkg/re/re.go

+package re
+
+import (
+	"regexp"
+	"strings"
+)
+
+var (
+	DomainRe = regexp.MustCompile(`([a-zA-Z0-9][-a-zA-Z0-9]{0,62}\.)+([a-zA-Z][-a-zA-Z]{0,62})`)
+
+	IPv4Re = regexp.MustCompile(`(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)(\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)){3}`)
+	IPv6Re = regexp.MustCompile(`fe80:(:[0-9a-fA-F]{1,4}){0,4}(%\w+)?|([0-9a-fA-F]{1,4}:){7}[0-9a-fA-F]{1,4}|::[fF]{4}:(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)(\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)){3}|(([0-9a-fA-F]{1,4}:){0,6}[0-9a-fA-F]{1,4})?::(([0-9a-fA-F]{1,4}:){0,6}[0-9a-fA-F]{1,4})?`)
+)
+
+func MaybeRegexp(s string) bool {
+	return strings.ContainsAny(s, "[]{}()?")
+}

pkg/re/re_test.go

+package re
+
+import (
+	"fmt"
+	"testing"
+)
+
+var domainList = []string{
+	"a.a.qiniudns.com",
+	"a.com.qiniudns.com",
+	"a.com.cn.qiniudns.com",
+	"看这里：a.com.cn.qiniudns.com行不行",
+}
+
+func TestDomainRe(t *testing.T) {
+	for _, domain := range domainList {
+		if !DomainRe.MatchString(domain) {
+			t.Error(domain)
+			t.Fail()
+		}
+		fmt.Println(DomainRe.FindAllString(domain, -1))
+	}
+}
+
+var validIPv6List = []string{
+	"::ffff:104.26.11.119",
+}
+
+func TestIPv6Re(t *testing.T) {
+	for _, ip := range validIPv6List {
+		if !IPv6Re.MatchString(ip) {
+			t.Error(ip)
+			t.Fail()
+		}
+		fmt.Println(IPv6Re.FindAllString(ip, -1))
+	}
+}
+
+var maybeRegexList = []string{
+	"[a-z]*\\.example.com",
+	"kunlun[^.]+.com",
+	"gtm-a[1-7]b[1-9].com",
+}
+
+func TestMaybeRegexp(t *testing.T) {
+	if MaybeRegexp("abc.com") {
+		t.Fail()
+	}
+
+	for _, str := range maybeRegexList {
+		if !MaybeRegexp(str) {
+			t.Fail()
+		}
+	}
+}

pkg/zxipv6wry/update.go

+package zxipv6wry
+
+import (
+	"io"
+	"io/ioutil"
+	"log"
+	"os"
+
+	"github.com/saracen/go7z"
+	"github.com/zu1k/nali/pkg/common"
+)
+
+func Download(filePath ...string) (data []byte, err error) {
+	data, err = getData()
+	if err != nil {
+		log.Printf("ZX IPv6数据库下载失败，请手动下载解压后保存到本地: %s \n", filePath)
+		log.Println("下载链接： https://ip.zxinc.org/ip.7z")
+		return
+	}
+
+	if len(filePath) == 1 {
+		if err := common.SaveFile(filePath[0], data); err == nil {
+			log.Println("已将最新的 ZX IPv6数据库 保存到本地:", filePath)
+		}
+	}
+	return
+}
+
+const (
+	zx = "https://ip.zxinc.org/ip.7z"
+)
+
+func getData() (data []byte, err error) {
+	data, err = common.GetHttpClient().Get(zx)
+
+	file7z, err := ioutil.TempFile("", "*")
+	if err != nil {
+		return nil, err
+	}
+	defer os.Remove(file7z.Name())
+	if err := ioutil.WriteFile(file7z.Name(), data, 0644); err == nil {
+		return Un7z(file7z.Name())
+	}
+	return
+}
+
+func Un7z(filePath string) (data []byte, err error) {
+	sz, err := go7z.OpenReader(filePath)
+	if err != nil {
+		return nil, err
+	}
+	defer sz.Close()
+
+	fileNoNeed, err := ioutil.TempFile("", "*")
+	if err != nil {
+		return nil, err
+	}
+	fileNeed, err := ioutil.TempFile("", "*")
+	if err != nil {
+		return nil, err
+	}
+
+	if err != nil {
+		return nil, err
+	}
+	for {
+		hdr, err := sz.Next()
+		if err == io.EOF {
+			break // End of archive
+		}
+		if err != nil {
+			return nil, err
+		}
+
+		if hdr.Name == "ipv6wry.db" {
+			if _, err := io.Copy(fileNeed, sz); err != nil {
+				log.Fatalln("ZX ipv6数据库解压出错：", err.Error())
+			}
+		} else {
+			if _, err := io.Copy(fileNoNeed, sz); err != nil {
+				log.Fatalln("ZX ipv6数据库解压出错：", err.Error())
+			}
+		}
+	}
+	err = fileNoNeed.Close()
+	if err != nil {
+		return nil, err
+	}
+	defer os.Remove(fileNoNeed.Name())
+	defer os.Remove(fileNeed.Name())
+	return ioutil.ReadFile(fileNeed.Name())
+}

pkg/zxipv6wry/zxipv6wry.go

+package zxipv6wry
+
+import (
+	"encoding/binary"
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"math/big"
+	"net"
+	"os"
+	"strings"
+
+	"github.com/zu1k/nali/pkg/common"
+)
+
+type ZXwry struct {
+	common.IPDB
+}
+
+func NewZXwry(filePath string) (*ZXwry, error) {
+	var fileData []byte
+	var fileInfo common.FileData
+
+	_, err := os.Stat(filePath)
+	if err != nil && os.IsNotExist(err) {
+		log.Println("文件不存在，尝试从网络获取最新ZX IPv6数据库")
+		fileData, err = Download(filePath)
+		if err != nil {
+			return nil, err
+		}
+	} else {
+		fileInfo.FileBase, err = os.OpenFile(filePath, os.O_RDONLY, 0400)
+		if err != nil {
+			return nil, err
+		}
+		defer fileInfo.FileBase.Close()
+
+		fileData, err = ioutil.ReadAll(fileInfo.FileBase)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	fileInfo.Data = fileData
+
+	return &ZXwry{
+		IPDB: common.IPDB{
+			Data: &fileInfo,
+		},
+	}, nil
+}
+
+func (db ZXwry) Find(query string, params ...string) (result fmt.Stringer, err error) {
+	ip := net.ParseIP(query)
+	if ip == nil {
+		return nil, errors.New("Query should be IPv6")
+	}
+	ip6 := ip.To16()
+	if ip6 == nil {
+		return nil, errors.New("Query should be IPv6")
+	}
+
+	tp := big.NewInt(0)
+	op := big.NewInt(0)
+	tp.SetBytes(ip6)
+	op.SetString("18446744073709551616", 10)
+	op.Div(tp, op)
+	tp.SetString("FFFFFFFFFFFFFFFF", 16)
+	op.And(op, tp)
+
+	ipv6 := op.Uint64()
+	offset := db.searchIndex(ipv6)
+	country, area := db.getAddr(offset)
+
+	result = common.Result{
+		Country: strings.ReplaceAll(country, " CZ88.NET", ""),
+		Area:    strings.ReplaceAll(area, " CZ88.NET", ""),
+	}
+	return result, nil
+}
+
+func (db *ZXwry) getAddr(offset uint32) (string, string) {
+	mode := db.ReadMode(offset)
+	if mode == common.RedirectMode1 {
+		offset = db.ReadUInt24()
+		return db.getAddr(offset)
+	}
+	realOffset := db.Offset - 1
+	c1 := db.ReadArea(realOffset)
+	if mode == common.RedirectMode2 {
+		db.Offset = 4 + realOffset
+	} else {
+		db.Offset = realOffset + uint32(1+len(c1))
+	}
+	c2 := db.ReadArea(db.Offset)
+	return string(c1), string(c2)
+}
+
+func (db *ZXwry) searchIndex(ip uint64) uint32 {
+	header := db.ReadData(16, 8)
+	start := binary.LittleEndian.Uint32(header[8:])
+	counts := binary.LittleEndian.Uint32(header[:8])
+	end := start + counts*11
+
+	buf := make([]byte, 11)
+
+	for {
+		mid := common.GetMiddleOffset(start, end, 11)
+		buf = db.ReadData(11, mid)
+		ipBytes := binary.LittleEndian.Uint64(buf[:8])
+
+		if end-start == 11 {
+			if ip >= binary.LittleEndian.Uint64(db.ReadData(8, end)) {
+				buf = db.ReadData(11, end)
+			}
+			return common.ByteToUInt32(buf[8:])
+		}
+
+		if ipBytes > ip {
+			end = mid
+		} else if ipBytes < ip {
+			start = mid
+		} else if ipBytes == ip {
+			return common.ByteToUInt32(buf[8:])
+		}
+	}
+}