mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-28 05:46:36 +08:00
ryan
3328d3d121
Agent ships without City/Country MMDB in the binary; Docker images COPY databases into data_dir, bare installs seed via download on first start. Server keeps Country-only embed for optional MaxMind control-plane use. Also harden fetch script nonempty check and reject non-file MMDB paths.
33 lines
1.1 KiB
Bash
Executable File
33 lines
1.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Download MaxMind GeoLite2 Country/City databases for packaging (Docker image COPY),
|
|
# not for go:embed into the agent binary.
|
|
set -euo pipefail
|
|
|
|
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
OUT_DIR="${GEOIP_OUT_DIR:-${ROOT}/dist/geoip}"
|
|
COUNTRY_URL="${GEOIP_MMDB_URL:-https://github.com/FyraLabs/geolite2/releases/latest/download/GeoLite2-Country.mmdb}"
|
|
CITY_URL="${GEOIP_CITY_MMDB_URL:-https://github.com/FyraLabs/geolite2/releases/latest/download/GeoLite2-City.mmdb}"
|
|
|
|
mkdir -p "${OUT_DIR}"
|
|
|
|
download_one() {
|
|
local url="$1"
|
|
local dest="$2"
|
|
local name="$3"
|
|
if curl -fsSL -o "${dest}.tmp" "$url" && [ -s "${dest}.tmp" ]; then
|
|
mv "${dest}.tmp" "$dest"
|
|
echo "GeoIP ${name} downloaded: $dest ($(wc -c <"$dest" | tr -d ' ') bytes)"
|
|
return 0
|
|
fi
|
|
rm -f "${dest}.tmp"
|
|
if [ -s "$dest" ]; then
|
|
echo "GeoIP ${name} download failed, keeping existing: $dest" >&2
|
|
return 0
|
|
fi
|
|
echo "GeoIP ${name} missing and download failed: $url" >&2
|
|
return 1
|
|
}
|
|
|
|
download_one "$COUNTRY_URL" "${OUT_DIR}/GeoLite2-Country.mmdb" "Country"
|
|
download_one "$CITY_URL" "${OUT_DIR}/GeoLite2-City.mmdb" "City"
|